o ei@s`UddlmZddlmZddlZddlmZddlmZddl m Z gdZ eddd e d e d e fd d ZejjjjdddgefejjjjdgdefejjjjdgdefejjjjdgefejjjjdgdefejjjjddgefiZeeee ee ee e ge fffe d<edddej!d ee effddZ"eddde d dfddZ#dS))Callable)AnyN) compatibility) GraphModule)default_matchingextract_attrs_for_loweringlift_lowering_attrs_to_nodesF)is_backward_compatiblenametarget_versionreturncCs|S)zDefault matching method)r r r r e/var/www/addictedbytheproject.nl/epg/venv/lib/python3.10/site-packages/torch/fx/passes/param_fetch.pyrsrweightbias)rr kernel_sizestridepaddingdilationgroups padding_mode)rr running_mean running_vareps)rrrrreturn_indices ceil_modeinplacemodule_fetch_bookmodcCsi}t||d<t|tvrAtt|\}}}||jkr/td|dt|d|jd|D] }t||||j||<q1|Stt|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_methodattrr r rr:s  r fx_modulecCsXt|}|jjD]}|jdkr)t||jtr!t||jq t ||j|_ q dS)zjRecursively traverse all `fx_module` nodes and fetch the module's attributes if the node is a leaf module. call_moduleN) dict named_modulesgraphnodesop isinstancetargetrrrr')r, submodulesnoder r rrUs   r)$collections.abcrtypingrr!torch.nnnntorch.fx._compatibilityrtorch.fx.graph_moduler__all__strintrmoduleslinearLinearconvConv2d batchnorm BatchNorm2dpoolingAdaptiveAvgPool2d MaxPool2d activationReLUrr.r#tuplelist__annotations__Modulerrr r r rs@        .