import numpy
import numpy as np
from numbapro import cuda
@cuda.autojit
def foo(aryA, aryB,out):
d_ary1 = cuda.to_device(aryA)
d_ary2 = cuda.to_device(aryB)
#dd = numpy.empty(10, dtype=np.int32)
d_ary1.copy_to_host(out)
griddim = 1, 2
blockdim = 3, 4
aryA = numpy.arange(10, dtype=np.int32)
aryB = numpy.arange(10, dtype=np.int32)
out = numpy.empty(10, dtype=np.int32)
foo[griddim, blockdim](aryA, aryB,out)
Exception: Caused by input line 11: can only get attribute from globals, complex numbers or arrays
我是新手,需要提示!在
@cuda.autotjit
标记并编译foo()
作为CUDA内核。内存传输操作应该放在内核之外。它应该类似于以下代码:我建议新的NumbaPro用户查看https://github.com/ContinuumIO/numbapro-examples中的示例。在
相关问题 更多 >
编程相关推荐