VhJUddlmZmZddlZddlmZddlmZddlm Z gdZ edde d e d e fd Z ej jjj d d dge fej jj"j$d gde fej jj&j(dgde fej jj*j,d ge fej jj*j.d gde fej jj0j2d dge fiZeeee ee ee e ge fffed<eddej@d ee effdZ!edde d dfdZ"y))AnyCallableN) compatibility) GraphModule)default_matchingextract_attrs_for_loweringlift_lowering_attrs_to_nodesF)is_backward_compatiblenametarget_versionreturnc|S)zDefault matching method)r r s K/home/dcms/DCMS/lib/python3.12/site-packages/torch/fx/passes/param_fetch.pyrrs  Kweightbias)rr kernel_sizestridepaddingdilationgroups padding_mode)rr running_mean running_vareps)rrrrreturn_indices ceil_modeinplacemodule_fetch_bookmodc i}tj||d<t|tvrtt|\}}}||jkr2t d|dtj|d|jd|D]"}t ||||j||<$|St tj|d)zIf `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 zFetcher 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.)torchtypenametyper"_version RuntimeErrorgetattr)r#attrs_for_loweringversionparam_to_fetchmatching_methodattrs rrr9s *,!&!4v Cy%%3DT#Y3O0 S\\ !"7)>%..:M9NiX[XdXdWef::  # YD'.sOD#,,4W'X t $ Y  ~~c"#$6 6  r fx_modulec2t|j}|jjD]e}|jdk(st ||j trt||j It||j |_ gy)zjRecursively traverse all `fx_module` nodes and fetch the module's attributes if the node is a leaf module. call_moduleN) dict named_modulesgraphnodesop isinstancetargetrr rr+)r0 submodulesnodes rr r Tsyi--/0J%% 77m #*T[[1;?,Z -DE*Dt{{++' r)#typingrrr%torch.nnnntorch.fx._compatibilityrtorch.fx.graph_moduler__all__strintrmoduleslinearLinearconvConv2d batchnorm BatchNorm2dpoolingAdaptiveAvgPool2d MaxPool2d activationReLUr"r3r'tuplelist__annotations__Modulerr rrrrTs 1- e,3- HH""Q6(:e,BII$sCx.-4e, K D - r