o
    k9:jx  ã                   @   s€  U d dl Z d dlZd dlZd dlmZ d dlZd dlm  mZ d dl	m
Z
mZmZmZmZ daeeef dB ed< eejjejjejjejjejjejjfZg edf‘edf‘edf‘edf‘e
d	f‘ed
f‘ed
f‘ejdf‘ejdf‘ejdf‘ejdf‘ej df‘ej!df‘ej"df‘ej#df‘ej$df‘ej%df‘ej&df‘ej'df‘ej(df‘ej)df‘ej*df‘ej+df‘ej,df‘ej-df‘ej.df‘ej/df‘ej0df‘ej1d f‘ej2d!f‘ej3d"f‘ej4d#f‘ej5d$f‘ej6d%f‘ej7d&f‘ej8d'f‘ej-df‘ej.df‘ej9d(f‘ej:d)f‘ej;d*f‘ej<d+f‘ej=d,f‘ej>d-f‘ej?d.f‘ej@d/f‘e j,df‘e jAd0f‘e j>d-f‘e jBd1f‘e jCd2f‘e j'df‘e j(df‘e j+df‘e j!df‘e j7d&f‘e j8d'f‘e j6d%f‘e j3d"f‘e j2d!f‘e j4d#f‘e j0df‘e j/df‘e j1d f‘e j-df‘e j9d(f‘e j.df‘ejDd3f‘ejEd4f‘ejFjGd5f‘ejFjHd6f‘ejjId7f‘ejJjKjLd8f‘ejJjKjMd9f‘ejJjNjOd:f‘ejJjNjPd;f‘ejJjNjQd<f‘ejJjNjRd=f‘ejjSd>f‘ejjTd?f‘ejUd@f‘ejVjWdAf‘ejVjXdBf‘ejVjYdCf‘ejVjZdDf‘ejVj[dEf‘ejVj\dFf‘ejVj]dGf‘ejVj^dHf‘ejVj_dIf‘Z`dJdK„ Zaeaƒ ZbdLdM„ ZcdNdO„ ZddUdQdR„ZedSdT„ ZfdS )Vé    N)ÚOrderedDict)Ú_list_with_defaultÚ_pairÚ
_quadrupleÚ_singleÚ_tripleÚ_builtin_tablezaten::_pairzaten::_quadruplezaten::_singlezaten::_triplezaten::list_with_defaultz
aten::dictzaten::cudnn_is_acceptablez
aten::ceilzaten::copysignz	aten::erfz
aten::erfcz	aten::expzaten::expm1z
aten::fabszaten::floorzaten::gammazaten::lgammaz	aten::logzaten::log10zaten::log1pz	aten::powz
aten::sqrtzaten::isnanzaten::asinhzaten::atanhz
aten::coshz
aten::sinhz
aten::tanhz
aten::acosz
aten::asinz
aten::atanzaten::atan2z	aten::cosz	aten::sinz	aten::tanzaten::acoshz
aten::fmodz
aten::modfzaten::factorialzaten::frexpzaten::isinfzaten::degreeszaten::radiansúaten::isfinitezaten::anglezaten::polarzaten::ldexpzaten::_assertz
aten::gradzaten::backwardzaten::_infer_sizez aten::_no_grad_embedding_renorm_zaten::_assert_int_or_pairzaten::_no_grad_fill_zaten::_no_grad_normal_zaten::_no_grad_uniform_zaten::_no_grad_zero_zaten::_get_tracing_statezaten::_get_cpu_capabilityz
aten::warnz
aten::stftzaten::istftzaten::cdistz
aten::normzaten::unique_dimzaten::unique_consecutivezaten::nuclear_normzaten::frobenius_normzaten::tensordotc                  C   s   g d¢} dd„ | D ƒS )N)ÚstftÚistftÚluÚcdistÚnormÚuniqueÚunique_consecutiveÚ	tensordotc                 S   s   h | ]}t tj|ƒ’qS © )ÚgetattrÚtorchÚ
functional)Ú.0Únamer   r   úZ/home/nk/hobo-godmode/plappi-mvp/.venv/lib/python3.10/site-packages/torch/jit/_builtins.pyÚ	<setcomp>“   s    z7_gen_torch_functional_registered_ops.<locals>.<setcomp>r   )Úopsr   r   r   Ú$_gen_torch_functional_registered_ops„   s   
r   c                 C   s   | t v S ©N)Ú_functional_registered_ops©Úfnr   r   r   Ú_is_special_functional_bound_op™   s   r    c                  C   s´   t d urt S i a d
dd„} tD ]}| |ƒ qt tjdf¡ t tjdf¡ t tjdf¡ dd lm	  m
} | ¡ rKt |jdf¡ t |jd	f¡ tD ]
\}}|t t|ƒ< qMt S )NÚreturnc                 S   sf   t | ƒD ],}t| |ƒ}t|ƒr0t|ƒs0|tjur0|tjur0|dkr'|dd … }t |d| f¡ qd S )NÚ_segment_reduceé   zaten::)	Údirr   Úcallabler    r   Úno_gradÚautocastÚ_builtin_opsÚappend)Úmodr   Úvr   r   r   Úregister_all¤   s   
ÿþ

€õz(_get_builtin_table.<locals>.register_allz	aten::gcdr	   zaten::mathremainderr   zaten::get_gradientszaten::dist_backward©r!   N)r   Ú_modules_containing_builtinsr(   r)   ÚmathÚgcdÚisfiniteÚ	remainderÚtorch.distributed.autogradÚdistributedÚautogradÚis_availableÚget_gradientsÚbackwardÚid)r,   r*   Údist_autogradÚbuiltinÚaten_opr   r   r   Ú_get_builtin_tablež   s    

r=   r!   c                 C   s   |t ƒ t| ƒ< d S r   )r=   r9   )r   Úopr   r   r   Ú_register_builtinÆ   s   r?   c                 C   s   t ƒ  t| ƒ¡S r   )r=   Úgetr9   r   r   r   r   Ú_find_builtinÊ   s   rA   r-   )gÚcmathr/   ÚwarningsÚcollectionsr   r   Útorch.backends.cudnnÚbackendsÚcudnnÚtorch.nn.modules.utilsr   r   r   r   r   r   ÚdictÚintÚstrÚ__annotations__Ú_CÚ_nnÚ_fftÚ_linalgÚ_nestedÚ_sparseÚ_specialr.   Úis_acceptableÚceilÚcopysignÚerfÚerfcÚexpÚexpm1ÚfabsÚfloorÚgammaÚlgammaÚlogÚlog10Úlog1pÚpowÚsqrtÚisnanÚasinhÚatanhÚcoshÚsinhÚtanhÚacosÚasinÚatanÚatan2ÚcosÚsinÚtanÚacoshÚfmodÚmodfÚ	factorialÚfrexpÚisinfÚdegreesÚradiansr1   ÚphaseÚrectÚldexpÚ_assertr5   Úgradr8   Ú_infer_sizeÚnnr   Ú_no_grad_embedding_renorm_Úassert_int_or_pairÚinitÚ_no_grad_fill_Ú_no_grad_normal_Ú_no_grad_uniform_Ú_no_grad_zero_Ú_get_tracing_stateÚ_get_cpu_capabilityÚwarnÚ_VFr
   r   r   r   Ú
unique_dimr   Únuclear_normÚfrobenius_normr   r(   r   r   r    r=   r?   rA   r   r   r   r   Ú<module>   sš  
	ù
þýüûúùø	÷
öõôóòñðïîíìëêéèçæåäãâá à!ß"Þ#Ý$Ü%Û&Ú'Ù(Ø)×*Ö+Õ,Ô-Ó.Ò/Ñ0Ð1Ï2Î3Í4Ì5Ë6Ê7É8È9Ç:Æ;Å<Ä=Ã>Â?Á@ÀA¿B¾C½D¼E»Fº
G¹
H¸
I·Kþ¶N²O±P°Q¯R®
S­
T¬U«
Vª
W©
X¨
Y§
Z¦
[¥
\¤
]£
^¢f
(