cython - Cupy 会支持 cython（例如缓冲索引）吗？

Question

我已经实现了自己定义的链接器链接，但是因为它太慢了。我已经实现了我的代码的 cython CPU 版本。但我想通过 GPU 进一步提高速度。所以我测试了以下代码，但它失败了：

%%cython

import numpy as np
cimport numpy as np
import cupy as cp
cimport cupy as cp
cdef class A:
    def __init__(self):
        pass

    cdef cp_test(self, cp.ndarray[cp.float_t, ndim=2] arr):
        return cp.sum(arr)

a = A()
arr = cp.arange(100).reshape(20,50)
print(a.cp_test(arr))

报告：

    cdef cp_test(self, cp.ndarray[cp.float_t, ndim=2] arr):
                      ^
------------------------------------------------------------

C:\Users\.ipython\cython\_cython_magic_d4940a274af88f0257c368b8a5d0e3f5.pyx:13:23: 'ndarray' is not a type identifier

Answer 1

抱歉，CuPy 目前不提供 cython 接口（我是 CuPy 开发人员之一）。