如何在NumbaPro函数中使用numpy函数?

时间:2014-09-08 14:40:05

标签: python numpy

我是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)

0 个答案:

没有答案