Как использовать numba для функции-члена класса?

Question

Я использую стабильную версию Numba 0.30.1.

Я могу это сделать:

import numba as nb 
@nb.jit("void(f8[:])",nopython=True)        
def complicated(x):         
    for a in x: 
     b = a**2.+a**3. 

в качестве тестового примера, и убыстрение огромна. Но я не знаю, как действовать, если мне нужно ускорить функцию внутри класса.

import numba as nb 
def myClass(object): 
    def __init__(self): 
     self.k = 1 
    #@nb.jit(???,nopython=True)        
    def complicated(self,x):         
     for a in x: 
      b = a**2.+a**3.+self.k 

Какой тип numba я использую для объекта self? Мне нужно иметь эту функцию внутри класса, так как ей нужно получить доступ к переменной-члену.

Answer 1

У вас есть несколько вариантов:

Используйте jitclass (http://numba.pydata.org/numba-doc/0.30.1/user/jitclass.html) к "Numba-Изе" все это.

Или сделать функцию-член обертку и передать переменные-члены через:

import numba as nb 

@nb.jit 
def _complicated(x, k): 
    for a in x: 
     b = a**2.+a**3.+k 

def myClass(object): 
    def __init__(self): 
     self.k = 1 

    def complicated(self,x):         
     _complicated(x, self.k)