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は（私はCuPyの開発者の一人）は現在cythonインタフェースを提供していません。