o
    ÜÜÓhµ  ã                   @   sŽ  U d dl Z d dlZd dlZd dlmZ d dlmZ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ef  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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j9d'f‘ej:d(f‘ej;d)f‘ej0df‘ej1df‘ej<d*f‘ej=d+f‘ej>d,f‘ej?d-f‘ej@d.f‘ejAd/f‘ejBd0f‘ejCd1f‘e j/df‘e jDd2f‘e jAd/f‘e jEd3f‘e jFd4f‘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 j9d'f‘e j6d$f‘e j5d#f‘e j7d%f‘e j3d!f‘e j2d f‘e j4d"f‘e j0df‘e j<d*f‘e j1df‘ejGd5f‘ejHd6f‘ejIjJd7f‘ejIjKd8f‘ejjLd9f‘ejMjNjOd:f‘ejMjNjPd;f‘ejMjQjRd<f‘ejMjQjSd=f‘ejMjQjTd>f‘ejMjQjUd?f‘ejjVd@f‘ejjWdAf‘ejXdBf‘ejYjZdCf‘ejYj[dDf‘ejYj\dEf‘ejYj]dFf‘ejYj^dGf‘ejYj_dHf‘ejYj`dIf‘ejYjadJf‘ejYjbdKf‘ZcdLdM„ Zdedƒ ZedNdO„ ZfdPdQ„ ZgdRdS„ ZhdTdU„ ZidS )Vé    N)ÚOrderedDict)ÚDictÚOptionalé   )Ú_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   úI/var/www/html/ai/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_opst   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„ } tD ]}| |ƒ qt tjdf¡ t tjdf¡ t tjdf¡ dd lm	  m
} | ¡ rJt |jdf¡ t |jdf¡ tD ]
\}}|t t|ƒ< qLt S )	Nc                 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   Ú_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>   c                 C   s   |t ƒ t| ƒ< d S r   )r>   r:   )r"   Úopr   r   r   Ú_register_builtin¶   s   r@   c                 C   s   t ƒ  t| ƒ¡S r   )r>   Úgetr:   r!   r   r   r   Ú_find_builtinº   s   rB   )jÚcmathr0   ÚwarningsÚcollectionsr   Útypingr   r   r   Útorch.backends.cudnnÚbackendsÚcudnnÚnn.modules.utilsr   r   r   r	   r
   r   ÚintÚstrÚ__annotations__Ú_CÚ_nnÚ_fftÚ_linalgÚ_nestedÚ_sparseÚ_specialr/   ÚdictÚ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Úradiansr2   ÚphaseÚrectÚldexpÚ_assertr6   Úgradr9   Ú_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@   rB   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·J¶KµL´M³N²O±
P°
Q¯R®
S­
T¬
U«
Vª
W©
X¨
Y§
Z¦
[¥c(