U ʹj @s|UddlmZddlmZmZmZmZmZmZddl Z ddl m Z ddl m Z dddgZe dd eeed d dZe j jjjd d dgefe j jjjd d dddddddgefe j jjjdd ddddgefe j jjjd gefe j jjjd ddddddgefe j jjjd dgefiZeeeeeeeeegefffed<e dd e j eeefdddZ!e dd eddd dZ"dS)!) GraphModule)AnyCallableDictListTupleTypeN) compatibilitydefault_matchingextract_attrs_for_loweringlift_lowering_attrs_to_nodesF)Zis_backward_compatible)nametarget_versionreturncCs|S)zDefault matching method )r rrrZ/var/www/html/bwave/python/venv/lib/python3.8/site-packages/torch/fx/passes/param_fetch.pyr sweightZbiasZ kernel_sizeZstridepaddingZdilationgroupsZ padding_modeZ running_meanZ running_varepsZreturn_indicesZ ceil_modeZinplacemodule_fetch_book)modrcCsi}t||d<t|tkrtt|\}}}||jkr^td|dt|d|jd|D]}t||||j||<qbntt|d|S)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)rattrs_for_loweringversionZparam_to_fetchZmatching_methodattrrrrr !s  $) fx_modulercCsXt|}|jjD]>}|jdkrt||jtrBt||jqt ||j|_ qdS)zoRecursively traverse all `fx_module` nodes and fetch the module's attributes if the node is a leaf module. Z call_moduleN) dictZ named_modulesgraphZnodesop isinstancetargetrr r r )r#Z submodulesnoderrrr 7s    )#Ztorch.fx.graph_modulertypingrrrrrrrZtorch.nnnnZtorch.fx._compatibilityr __all__strintr modulesZlinearZLinearconvZConv2dZ batchnormZ BatchNorm2dZpoolingZAdaptiveAvgPool2dZ MaxPool2dZ activationZReLUr__annotations__Moduler r rrrrs>           .