o Ni@sddlZddlmZmZeejgdejdejdgejdejddgejdejgd ejdgejgd ejdejgd ejdd Zejgd ejdZGdddeZdS)N)_ButcherTableauRKAdaptiveStepsizeODESolver)??g?)dtyperr)qq?UUUUUU?qq?)r r r r)gtqDZgTUUUUU?gqq?g)alphabetac_solc_error)rrrrc@seZdZdZeZeZdS) Bosh3SolverN)__name__ __module__ __qualname__order_BOGACKI_SHAMPINE_TABLEAUtableau _BS_C_MIDmidrrK/home/ubuntu/.local/lib/python3.10/site-packages/torchdiffeq/_impl/bosh3.pyrsr) torch rk_commonrrtensorfloat64rrrrrrrs