Est-ce que le cython supportera le support (par exemple, l'index tamponné)?

Question

J'ai implémenté moi même un chainer Link, mais parce que c'est trop lent. J'ai implémenté la version CPU de Cython de mon code. Mais je veux encore augmenter la vitesse via GPU. Donc, je teste le code suivant, mais il a échoué:

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

rapports:

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

Désolé, mais cupy ne fournit pas l'interface cython (je suis l'un des développeurs cupy).