我是CUDA的新手,我正在尝试使用NumbaPro的cuda扩展来计算一些指数,但是我似乎无法在cuda编译函数中使用numpy?我知道我在做一些愚蠢的事,但我无法理解。
import numpy as np
from pylab import imshow, show, plot
from timeit import default_timer as timer
from numbapro import cuda
from numba import *
import math
def Deng_test(xx,Expre):
ExTest = (np.abs(np.sum(Expre)))**2
return ExTest
Deng_test_gpu = cuda.jit(argtypes=(double,complex128), restype=float64, device=True, inline=True)(Deng_test)