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 개발자 중 하나입니다) 현재 사이 썬 인터페이스를 제공하지 않습니다.