a
    HSic>  ã                Z   @   sÊ  U d dl Z d dlZd dl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 dlmZ d dl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fZ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 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 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 jBd1fej.dfejCd2fej@d/fejDd3fejEd4fej)dfej*dfej-dfej#dfej9d(fej:d)fej8d'fej5d$fej4d#fej6d%fej2d!fej1d fej3d"fej/dfej;d*fej0dfe jFd5fejGd6fejHjId7fejHjJd8fejjKd9fejLjMjNd:fejLjMjOd;fejLjPjQd<fejLjPjRd=fejLjPjSd>fejLjPjTd?fejjUd@fejVdAfejWjXdBfejWjYdCfejWjZdDfejWj[dEfejWj\dFfejWj]dGfejWj^dHfejWj_dIfejWj`dJfgYZadKdL„ Zbebƒ ZcdMdN„ ZddOdP„ ZedQdR„ ZfdSdT„ ZgdS )Ué    Né   )Ú_singleÚ_pairÚ_tripleÚ
_quadrupleÚ_list_with_default)ÚOrderedDict)ÚDictÚOptionalÚ_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::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¢} t dd„ | D ƒƒS )N)ÚstftÚistftÚluÚcdistÚnormÚuniqueÚunique_consecutiveÚ	tensordotc                 s   s   | ]}t tj|ƒV  qd S ©N)ÚgetattrÚtorchÚ
functional)Ú.0Úname© r   úO/var/www/html/django/DPS/env/lib/python3.9/site-packages/torch/jit/_builtins.pyÚ	<genexpr>x   ó    z7_gen_torch_functional_registered_ops.<locals>.<genexpr>)Úset)Úopsr   r   r   Ú$_gen_torch_functional_registered_opsr   s    r!   c                 C   s   | t v S r   )Ú_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
} | ¡ r”t |jdf¡ t |jdf¡ tD ]\}}|t t|ƒ< q˜t S )	Nc                 S   sR   t | ƒD ]D}t| |ƒ}t|ƒrt|ƒs|tjur|tjurt |d| f¡ qd S )Nz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ÚbuiltinZ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    rA   )hr0   ÚcmathÚwarningsr   Útorch.backends.cudnnÚbackendsÚcudnnZnn.modules.utilsr   r   r   r   r   Úcollectionsr   Útypingr	   r
   r   ÚintÚstrÚ__annotations__Ú_CÚ_nnÚ_fftÚ_linalgÚ_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ÚwarnÚ_VFr   r   r   r   Ú
unique_dimr   Únuclear_normÚfrobenius_normr   r*   r!   r"   r%   r=   r?   rA   r   r   r   r   Ú<module>   sÒ   
$












¦a