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

Question

我已经实现了自己定义的chainer Link，但因为它太慢了。 我已经实现了我的代码的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开发人员之一）。