python - PyCUDA 按值将变量传递给内核

Question

应该足够简单；我确实想将一个 int 发送到 SourceModule 内核声明，其中 C 函数

__global__......(int value,.....)

随着值的声明和调用...

value = 256
...
...
func(value,...)

但是我从 pycuda 收到了各种错误，表明我使用了错误的类型。

Answer 1

标准 PyCUDA 函数接口(interface)要求参数具有 numpy 数据类型，因为它在内部确实映射到引擎盖下的 C 类型。因此，对于按值传递的标量参数，您需要先“转换”为合适的 numpy dtype。有点像

value = 256
va = numpy.int32(value)

func(va)

应该可以。如果您要传递单精度浮点值或数组，请确保明确使用 np.float32 数据类型，因为 numpy 默认使用 double ，您最终会遇到类似的错误。