PyCUDA通过值将变量传递给内核

应该足够简单;我从字面上想将int发送到SourceModule内核声明,其中C函数
__global__......(int value,.....)
值被声明并调用...
value = 256
...
...
func(value,...)
但是我从pycuda中收到各种错误,这些错误是我使用了错误的类型。
已邀请:
标准的PyCUDA函数接口要求参数具有numpy dtypes,因为它在内部确实映射到C类型。因此,对于按值传递的标量参数,您需要先“投射”到合适的numpy dtype。就像是
value = 256
va = numpy.int32(value)

func(va)
应该管用。如果要传递单精度浮点值或数组,请确保显式使用np.float32的dtype,因为numpy默认情况下使用双精度,并且最终会出现类似错误。

要回复问题请先登录注册