我使用的简化代码基本上与文档中的示例完全相同。下面的代码应该返回2^3 + 2^3 = 16
,但是它返回9。我做错什么了?在
import numpy
import pycuda.reduction as reduct
import pycuda.gpuarray as gpuarray
import pycuda.autoinit
from pycuda.compiler import SourceModule as module
newzeros = [{1,2,3},{4,5,6}]
gpuSum = reduct.ReductionKernel(numpy.uint64, neutral="0", reduce_expr="a+b", map_expr="1 << x[i]", arguments="int* x")
mylengths = pycuda.gpuarray.to_gpu(numpy.array(map(len,newzeros),dtype = "uint64",))
sumfalse = gpuSum(mylengths).get()
print sumfalse
我刚想好了。定义内核时使用的参数列表应该是
unsigned long *x
,而不是int *x
。我在其他地方使用64位整数,结果搞砸了。在相关问题 更多 >
编程相关推荐