Wie numba jitclass angeben, wenn das Attribut der Klasse eine andere Klasseninstanz enthält?

Question

Ich versuche mit Numba die Python-Leistung von scipy.integrate.odeint zu steigern. Zu diesem Zweck muss ich @ nb.jit (nopython = True) für die Funktion verwenden, die das ODE-System definiert. Diese Funktion muss jedoch eine andere Python-Klasseninstanz als Argument in meinem Programm verwenden. Ich musste die Klasse auch mit @ nb.jitclass (spec) mit entsprechenden Spezifikationen jit. Dies funktionierte gut, bis ich ein ernstes Problem fand, wenn die Spezifikationen der Klasse einen anderen Typ von Klasseninstanzen als Methode enthielten. Mein Code folgt.

import numba as nb 
from scipy.integrate import odeint 


spec1=[("hi", nb.i4)] 
@nb.jitclass(spec1) 
class Hi(object): 
    def __init__(self): 
     self.hi = 0 

spec2=[("dummy", nb.i4), ("dummy1", nb.i4)] 
@nb.jitclass(spec2) 
class Dummy(object): 
    def __init__(self, anotherClassInstance): 
     self.dummy = 0 
     self.dummy1 = anotherClassInstance 

class A: 
    def __init__(self, someClassInstance): 
     self.a1 = someClassInstance 

    def odeSystem(self, x, t): 
     return _odeSystem(x, t, self.a1) 

    def odeSolve(self, iValues, ts): 
     sol = odeint(self.odeSystem, iValues, ts) 
     return sol 

@nb.jit(nopython=True) 
def _odeSystem(x, t, someClassInstance): 
    return 1-x 



if __name__ == "__main__": 
    c = Hi() 
    b = Dummy(c) 
    a = A(b) 
    print a.odeSolve(0.5, range(0, 10)) 

Zusammengefasst: So hier "Klasse A" ist meine Ode Löser.

1. Um die Methode "odeSystem" mit numba zu kompilieren, muss es keine Klassenmethode sein. Also habe ich eine andere Funktion außerhalb der Klasse "_odeSystem" gemacht.

2. Leider muss mein odeSystem eine Klasseninstanz als Argument haben. Daher habe ich @jitclass verwendet, um das Klasseninstanzargument korrekt zu kompilieren.

3. Ich stieß wieder auf ein anderes Problem, wo diese Klasse "Dummy" auch eine andere Art von Klasseninstanz als eines seiner Attribute nimmt. Ich habe keine Ahnung, wie man "spec" für diese Klasse setzt. Ich habe den Typ für "dummy1" mit "nb.typeof (Hi)" versucht, aber es hat nicht funktioniert.

Bitte helfen Sie mir. Danke im Voraus.

Answer 1

Sie können in Ihrer Spezifikationsdefinition .class_type.instance_type verwenden, um eine Instanz eines anderen Typs zu halten. Siehe Beispiel im Numba-Quellenbaum here

spec2=[("dummy", nb.i4), ("dummy1", Hi.class_type.instance_type)] 
@nb.jitclass(spec2) 
class Dummy(object): 
    def __init__(self, anotherClassInstance): 
     self.dummy = 0 
     self.dummy1 = anotherClassInstance