python指针使用_python-如何调用使用指针返回numba @jit中的值...

例如,下面是一个简单的c函数,该函数使用指针返回值：

void add(double x, double y, double *r)

{

*r = x + y;

}

我想为两个数组中的每个元素调用add()函数,并通过numba @jit函数收集结果.

首先编译c代码：

!gcc -c -fpic func.c

!gcc -shared -o func.so func.o

并通过ctypes加载它：

lib = ctypes.cdll.LoadLibrary("./func.so")

add = lib.add

add.argtypes = ctypes.c_double, ctypes.c_double, ctypes.c_void_p

add.restype = None

然后是numba函数：

from numba import jit, float64

@jit(float64(float64[:], float64[:]))

def f(x, y):

z = np.zeros_like(x)

for i in range(x.shape[0]):

add(x[i], y[i], &z[i]) # here I want to pass the address of z[i]

return z

但是numba没有运算符或函数的地址.

目前,我正在使用以下方法.但是这种方法不能用于nopython模式,而且我不知道for循环中的代码是否具有python对象.

@jit(float64(float64[:], float64[:]))

def f(x, y):

z = np.zeros_like(x)

tmp = ctypes.c_double(0.0)

addr = intp(ctypes.addressof(tmp))

val = carray(ctypes.pointer(tmp), 1)

for i in range(x.shape[0]):

add(x[i], y[i], addr)

z[i] = val[0]

return z