o
    Ni                     @   s  d dl Z ddlmZmZ g dZdgddgg dg d	g d
g dg dg dg dg dg dg dg dgZg dZg dZdZdgd Z	ded  ded   ded   ded   d e  d! de  e	d < d"ed  d#ed   d$ed   d%ed   d&e  de  e	d< d'ed  d(ed   d)ed   d*ed   d+e  de  e	d,< d-ed  d.ed   d/ed   d0ed   d1e  de  e	d2< d3ed  d4ed   d5ed   d6ed   d7e  de  e	d8< d9ed  d:ed   d;ed   d<ed   d=e  de  e	d>< d?ed  d@ed   dAed   dBed   dCe  de  e	dD< dEed  dFed   dGed   dHed   dIe  de  e	dJ< dKed  dLed   dMed   dNed   dOe  de  e	dP< dQed  dRed   dSed   dTed   dUe  de  e	dV< e j
ee jdWZdXdY eD Ze j
ee jdWZe j
ee jdWZe j
e	e jdWZeeeeedZZG d[d\ d\eZdS )]    N   )_ButcherTableauRKAdaptiveStepsizeODESolver)qq?gUUUUUU?g      ?      ?g      ?gzG?g(\?gvw
`?g?g9Ȗ?r   r   r   r   gUUUUUU?g      ?)g      ?r   g      ?)r   r   g     g     ?)g333333?r   r   g      ?g333333?)gp?r   r   gfT?g\(g1oO?)gfR?r   r   g?gm5k?g+Q6?g8/|ƿ)g_`B?r   r   gXiKg_DĿg\ZK?g
$S?gXq.?)	g!0~?r   r   g2Z<{	g)-;qҿgPg飰_@gg@?g龭?)
gdKarr   r   gH>00@g$M?g99g2 0gײ-@g6,?*g'}-g@)g'`-?r   r   gp?g8^ۿg?gWeg\EO@g#P@gCX?g2]*)@gD};?)g'wSQ?r   r   gŃ=Q'ge>p>g7Mk#?gb&(@g:$rgN?gA^ͿgHU7؃?r   )-IZ_?r   r   r   r   uc?d8T͡?2(?LdO\@U#?"?|??獓_zο      ?)r   r   r   r   r   r   r	   r
   r   r   r   r   r   r   )g6OJ7P?r   r   r   r   g;Y?gDBt&igB8gj9U1L?gEʷ7god*?gIKM]ҿr   r   g      ?g           gia   g,"#6@   gk>@   geK3@   g_D@g      ?gn.Cg$k?M]@gc_^@gJ@gWu@g3}I`4@gV<IP@g:T@gE3Q*rG@ggct%@   g/V@g|50@gRC%#@g44?g/k?   gyyp@@gfvV@gL"|U@g׎A@g'cm@   gmt7[Q$gn6@gL}@Kj1@g?9X6@g|5?	   gf=P)gz
<@@g|T<@gWM֡$@g} ]8?
   gG&(=@gJZUT@gIA&pjT@gP
aA@g;Qu@   g#')kDg淦	]@g?^6z \@gNƕGvG@gj @   gf0L4@g!HwJ@gV"v? I@g3@gpʿm@   dtypec                 C   s   g | ]
}t j|t jd qS )r   )torchtensorfloat64).0B_ r$   L/home/ubuntu/.local/lib/python3.10/site-packages/torchdiffeq/_impl/dopri8.py
<listcomp>A   s    r&   )alphabetac_solc_errorc                   @   s   e Zd ZdZeZeZdS )Dopri8Solverr   N)__name__
__module____qualname__order_DOPRI8_TABLEAUtableau_C_midmidr$   r$   r$   r%   r+   I   s    r+   )r   	rk_commonr   r   ABC_solC_errhC_midr    r!   r2   r0   r+   r$   r$   r$   r%   <module>   sL    
HDDDDDDDDD