o NiA@sddlZddlmZmZeejdgejdejdgejdgejddgejdejddgejddZejdd gejdZGd d d eZdS) N)_ButcherTableauRKAdaptiveStepsizeODESolverg?)dtypeg?g)alphabetac_solc_errorgc@seZdZdZeZeZdS)AdaptiveHeunSolverN)__name__ __module__ __qualname__order_ADAPTIVE_HEUN_TABLEAUtableau _AH_C_MIDmidrrS/home/ubuntu/.local/lib/python3.10/site-packages/torchdiffeq/_impl/adaptive_heun.pyr sr ) torch rk_commonrrtensorfloat64rrr rrrrs(