Numba cfunc工厂结合NumPy数组的ctypes指针传递错误排查

错误原因 #

你编写的工厂函数中，sum_array_2这个Numba cfunc直接引用了ctypes.POINTER类型的arr_ptr，但Numba的nopython编译模式无法处理ctypes对象，导致类型推断失败，抛出unsupported PEP 3118 format错误。ctypes指针是Python层面的对象，Numba无法将其编译为原生C类型。

修正方案 #

核心思路是避免在Numba cfunc中直接使用ctypes对象，改用Numba能识别的内存地址（整数），再在cfunc内部转换为Numba的CPointer类型。

完整修正代码 #

import numpy as np
from numba import cfunc, carray
from numba.types import intc, CPointer, float64
import ctypes

class SimpleExample():
    def __init__(self,array):
        self.array = array
        self.n = array.size

    # C签名定义：double func(double *input_array, int n)
    sig = float64(CPointer(float64), intc)
    sig2 = float64()

    @staticmethod
    @cfunc(sig)
    def sum_array(ptr, n):
        arr = carray(ptr, n)  # 将原生指针转为NumPy数组
        return np.sum(arr)

    def sum_factory(self):
        arr = self.array
        size = self.n
        sum_fun = type(self).sum_array
        # 获取数组内存地址（整数类型，Numba可识别）
        arr_addr = arr.ctypes.data
        @cfunc(sig2)
        def sum_array_2():
            # 将整数地址转为Numba的CPointer类型
            ptr = CPointer(float64)(arr_addr)
            return sum_fun(ptr, size)
        return sum_array_2.address


a = np.array([1.0, 2.0, 3.0, 4.0], dtype=np.float64)
example = SimpleExample(a)

# 获取工厂生成的函数指针
func_ptr = example.sum_factory()

# 创建无参的ctypes函数指针
cfunc_type = ctypes.CFUNCTYPE(ctypes.c_double)
cfunc_instance = cfunc_type(func_ptr)

print("Sum from SimpleExample sum_factory:", cfunc_instance())

关键修改点 #

1. 替换arr.ctypes.data_as(ctypes.POINTER(ctypes.c_double))为arr.ctypes.data，获取数组内存的整数地址
2. 在sum_array_2内部，用CPointer(float64)(arr_addr)将整数地址转为Numba可识别的原生指针类型
3. 保持原有cfunc的调用逻辑不变，确保编译通过

运行修正后的代码会输出：Sum from SimpleExample sum_factory: 10.0，正确计算数组元素的和。

内容的提问来源于stack exchange，提问作者Miguel Madeira