pytcuda学习笔记（一）_pycuda_pogevip111的博客-程序员秘密

技术标签： cuda

1.为什么不直接使用pycuda.autoinit？

import pycuda.autoinit

答：自动初始化很多时候不好使，比如多线程等。

2.能正常运行，退出时候报错，什么原因？

PyCUDA ERROR: The context stack was not empty upon module cleanup.

A context was still active when the context stack was being
cleaned up. At this point in our execution, CUDA may already
have been deinitialized, so there is no way we can finish
cleanly. The program will be aborted now.
Use Context.pop() to avoid this problem.

Aborted (core dumped)
在这里插入图片描述

答：没有正常退出，需要先删除上下文

ctx  = cuda.Device(0).make_context()
...
ctx.pop()

完整代码如下

# import pycuda.autoinit
import pycuda.driver as cuda
import numpy


from pycuda.compiler import SourceModule
# export PATH=/usr/local/cuda/bin:$PATH


cuda.init()     


def dott(array_a, array_b):
    #ctx.push()
    mod = SourceModule("""
    __global__ void dot(float *dest, float *a, float *b)
    {
    const int i = threadIdx.x;
    dest[i] = a[i] * b[i];
    }
    """)
    dot = mod.get_function("dot")

    dest = numpy.zeros_like(array_a)
    dot(cuda.Out(dest), cuda.In(array_a), cuda.In(array_b),block=(len(a),1,1), grid=(1,1))
    # 
    return dest



if __name__ == "__main__":
    a = numpy.random.normal(size=40).astype(numpy.float32)
    b = numpy.random.normal(size=40).astype(numpy.float32)
    ctx  = cuda.Device(0).make_context()
    print(dott(a, b))
    ctx.pop()
    # del ctx

本文链接：https://blog.csdn.net/Andrew_megemeta/article/details/115712499

原作者删帖不实内容删帖广告或垃圾文章投诉

智能推荐

js检测数据类型的四种方法_nvnv_yezi的博客-程序员秘密

typeof只能检测出undefined，string， number， boolean， symbol， function，object返回值基本类型，除 null 以外，均可以返回正确的结果。引用类型，除 function 以外，一律返回 object 类型。null ，返回 object 类型。function 返回 function 类型。原理js 在底层存储变量的时候...

DECLARE_DYNAMIC和IMPLEMENT_DYNAMIC宏_dll declare_dynamic 只有头文件_zhjf14的博客-程序员秘密

转自：http://blog.csdn.net/aofengdaxia/article/details/6926699PLEMENT_DYNAMIC是实现“运行时类型识别”宏，与之相对应的是DECLARE_DYNAMIC(声明“运行时类型识别”宏)。也就是说你在.CPP文件中如果看见有IMPLEMENT_DYNAMIC，则在.H文件中必定有DECLARE_DYNAMIC的声明。DECL

java线程Thread和Runnable区别和联系_大数据之夏的博客-程序员秘密

我们都晓得java实现线程2种方式，一个是继承Thread，另一个是实现Runnable。模拟窗口买票，第一例子继承thread，代码如下[code="java"]package thread;public class ThreadTest { public static void main(String[] args) { Thread1 t1 = n...

osg探究补充：osg数据加载原理(插件机制简介)_osg 加载 glb 的插件_小王_123的博客-程序员秘密

前言我们接着昨天的继续，昨天主要是讲解了DatabasePager类中的特定的成员变量以及run函数的第一部分，对所要请求加载的数据按照是否是网络数据进行分类加载模式。今天我们就看看数据是怎们加载到osg环境中的。还是在DatabasePager::DatabaseThread::run()函数中，首先我们保证了databaseRequest是线程安全的，因为run()函数是创建一个唯...

14通信工程光健+电赛FPGA作业_通信工程fpga复习csdn_燕子矶的博客-程序员秘密

14通信工程光健+电赛FPGA作业第一题电路生成的RTL图仿真波形图 HDL代码module cnt3(clk,rst,cnt);input clk,rst;output [2:0] cnt;reg [2:0] cnt;reg [2:0] q=5;always @(posedge clk) begin if(!rst) begin cnt<=3

vue新增、修改设置默认值_lxxy623的博客-程序员秘密

1. 在Controller中将值传入前台页面service实现层保存数据Controller将数据存入前台2.页面的处理在页面中循环下拉宽中的值（另外两个也是一样的循环）到现在基本完成了下拉选项，在进行更改操作时vue将floorInfoDate.aId进行判断；新增和修改页面，如果你想将数据库为未选择下拉选项的默认设为请选择的话，只要将floorInfoDate.aId设置为flo...