U yhC @sddlmZmZddlmZddlmZddlmZddl m Z ddl m Z ddd gZ ed d Gd dde Zed d deeeeeeeeeffdddZed d eeeddd ZdS))DictTuple) compatibility)Graph) GraphModule)SubgraphMatcher)Module HolderModulelift_subgraph_as_modulecompare_graphsF)Zis_backward_compatiblecs eZdZdZfddZZS)r zy HolderModule is used to copy all the attributes from original module to submodules that uses the attributes cs,t|D]\}}|||qdS)N)super__init__items add_module)selfdkv __class__N/var/www/html/venv/lib/python3.8/site-packages/torch/fx/passes/utils/common.pyr s zHolderModule.__init__)__name__ __module__ __qualname____doc__r __classcell__rrrrr sr)gmsubgraph comp_name class_namereturncCsti}i}|jD]}|jdkr"q|j}t|ts6t|d}|} |} |ddD]2} t| | sr| | tit | | } t | | } qT|d} t | | } |d|||<t | | | qt ||||fS)aw Create a GraphModule for subgraph, which copies the necessary attributes from the original parent graph_module. Args: gm (GraphModule): parent graph module subgraph (Graph): a valid subgraph that contains copied nodes from the parent graph comp_name (str): name for the new component class_name (str): name for the submodule )Z call_moduleZget_attr.N) r Znodesoptarget isinstancestrAssertionErrorsplithasattrrgetattrsetattrr)rrr r! submoduleZorig_to_split_fqn_mappingnr&Ztarget_name_partscurrZorig_gmnameZleaf_node_nameZ leaf_noderrrr s(       )leftrightr"cCs$t|ddd}||}t|dkS)z Return True if two graphs are identical, i.e they - have the same number of outputs in the same order - have the same number of inputs in the same order - have the same set of nodes, and identical connectivity T)Z match_outputZmatch_placeholderr)rmatchlen)r2r3Zmatchermatchesrrrr Ts  N)rr)typingrrZtorch.fx._compatibilityrZtorch.fx.graphrZtorch.fx.graph_modulerZ#torch.fx.passes.utils.matcher_utilsrZtorch.nnr__all__r r(r boolr rrrrs(       7