cython - 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 接口(interface)(我是 CuPy 开发人员之一)。
功能请求开放于https://github.com/cupy/cupy/issues/130 .