Ë
    M²j�  ã                   óR  — U d dl mZ d dlmZ d dlZd dlmZ d dlmZ d dl	m
Z
 g d¢Z ed¬«      d	ed
edefd„«       Zej                  j                  j                   j"                  dddgefej                  j                  j$                  j&                  dg d¢efej                  j                  j(                  j*                  dg d¢efej                  j                  j,                  j.                  dg efej                  j                  j,                  j0                  dg d¢efej                  j                  j2                  j4                  ddgefiZeeeeee   eeegef   f   f   e d<    ed¬«      dejB                  deeef   fd„«       Z" ed¬«      de
ddfd„«       Z#y)é    )ÚCallable)ÚAnyN)Úcompatibility)ÚGraphModule)Údefault_matchingÚextract_attrs_for_loweringÚlift_lowering_attrs_to_nodesF)Úis_backward_compatibleÚnameÚtarget_versionÚreturnc                 ó   — | S )zDefault matching method© )r   r   s     úZ/var/www/html/ia-enhanced/venv/lib/python3.12/site-packages/torch/fx/passes/param_fetch.pyr   r      s	   € ð €Kó    é   ÚweightÚbias)r   r   Úkernel_sizeÚstrideÚpaddingÚdilationÚgroupsÚpadding_modeé   )r   r   Úrunning_meanÚrunning_varÚeps)r   r   r   r   Úreturn_indicesÚ	ceil_modeÚinplaceÚmodule_fetch_bookÚmodc           	      óš  — i }t        j                  | «      |d<   t        | «      t        v r€t        t        | «         \  }}}|| j                  k  r2t        d|› dt        j                  | «      › d| j                  › d�«      ‚|D ]"  }t        |  ||| j                  «      «      ||<   Œ$ |S t        t        j                  | «      › d�«      ‚)z³If `mod` is in `module_fetch_book`, fetch the mod's attributes that in the `module_fetch_book`
    after checking module's version is compatible with the `module_fetch_book`.
    r   zFetcher version z try to fetch z	 version zl, please upgrade the module_fetch_book, open an issue and @842974287 or report a bug to AIACC team directly.z’ is not in the module_fetch_book yet, please add it to the module_fetch_book, open an issue and @842974287 or report a bug to AIACC team directly.)ÚtorchÚtypenameÚtyper"   Ú_versionÚRuntimeErrorÚgetattr)r#   Úattrs_for_loweringÚversionÚparam_to_fetchÚmatching_methodÚattrs         r   r   r   :   sé   € ð
 *,ÐÜ!&§¡°Ó!4Ð�vÑäˆCƒyÔ%Ñ%Ü3DÄTÈ#ÃYÑ3OÑ0ˆ� Ø�S—\‘\Ò!ÜØ" 7 )¨>¼%¿.¹.ÈÓ:MÐ9NÈiÐX[×XdÑXdÐWeð f:ð :óð ð
 #ò 	YˆDÜ'.¨s±OÀDÈ#Ï,É,Ó4WÓ'XÐ˜tÒ$ð	Yð Ðô Ü�~‰~˜cÓ"Ð#ð $6ð 6ó
ð 	
r   Ú	fx_modulec                 ó2  — t        | j                  «       «      }| j                  j                  D ]e  }|j                  dk(  sŒt        ||j                     t        «      rt        ||j                     «       ŒIt        ||j                     «      |_
        Œg y)zjRecursively traverse all `fx_module` nodes and fetch the module's attributes if the node is a leaf module.Úcall_moduleN)ÚdictÚnamed_modulesÚgraphÚnodesÚopÚ
isinstanceÚtargetr   r	   r   r+   )r0   Ú
submodulesÚnodes      r   r	   r	   U   sy   € ô �i×-Ñ-Ó/Ó0€Jà—‘×%Ñ%ò ˆØ�7‰7�mÓ#Ü˜* T§[¡[Ñ1´;Ô?Ü,¨Z¸¿¹Ñ-DÕEä*DØ˜tŸ{™{Ñ+ó+�Õ'ñr   )$Úcollections.abcr   Útypingr   r%   Útorch.nnÚnnÚtorch.fx._compatibilityr   Útorch.fx.graph_moduler   Ú__all__ÚstrÚintr   ÚmodulesÚlinearÚLinearÚconvÚConv2dÚ	batchnormÚBatchNorm2dÚpoolingÚAdaptiveAvgPool2dÚ	MaxPool2dÚ
activationÚReLUr"   r3   r'   ÚtupleÚlistÚ__annotations__ÚModuler   r	   r   r   r   ú<module>rU      sÎ  ðÞ $Ý ã Ý Ý 1Ý -ò€ñ  eÔ,ð˜3ð °ð ¸ò ó -ðð 
‡H�H×Ñ×Ñ×"Ñ" Q¨°6Ð(:Ð<LÐ$MØ	‡H�H×Ñ×Ñ× Ñ Ø	ò		
ð 	ð#ð 
‡H�H×Ñ×Ñ×*Ñ*Ø	Ú@Øð-ð
 
‡H�H×Ñ×Ñ×.Ñ.°°BÐ8HÐ0IØ	‡H�H×Ñ×Ñ×&Ñ&Ø	ÚWØð)ð
 
‡H�H×Ñ×Ñ×$Ñ$ q¨9¨+Ð7GÐ&Hð7SÐ �4˜˜e C¨¨c©°H¸cÀ3¸ZÈ¸_Ñ4MÐ$MÑNÐNÑOó ñ>  eÔ,ð B§I¡Ið °$°s¸C°x±.ò ó -ðñ4  eÔ,ð¨Kð ¸Dò ó -ñr   