Wird Cupy Cython unterstützen (zB gepufferter Index)?

Question

Ich habe selbst einen Chainer Link implementiert, aber weil es zu langsam ist. Ich habe Cython-CPU-Version meines Codes implementiert. Aber ich möchte Geschwindigkeit über GPU weiter steigern. So teste ich den folgenden Code, aber es ist fehlgeschlagen:

%%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)) 

Berichterstattung:

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

Sorry, aber Cupy bietet keine cython Schnittstelle zur Zeit (ich bin einer von Cupy Entwickler).