o ߗiB @sddlmZmZddlmZddlmZddlmZddl m Z ddl m Z gdZ edd Gd d d e Zedd  ddededededeeeeefff ddZedd dededefddZdS))DictTuple) compatibility)Graph) GraphModule)SubgraphMatcher)Module) HolderModulelift_subgraph_as_modulecompare_graphsF)is_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] \}}|||q dS)N)super__init__items add_module)selfdkv __class__Z/home/ubuntu/transcripts/venv/lib/python3.10/site-packages/torch/fx/passes/utils/common.pyrs zHolderModule.__init__)__name__ __module__ __qualname____doc__r __classcell__rrrrr sr rgmsubgraph comp_name class_namereturncCsti}i}|jD]S}|jdvrq |j}t|tsJ|d}|} |} |ddD]} t| | s9| | tit | | } t | | } q*|d} t | | } |d|||<t | | | q t ||||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 ) call_moduleget_attr.N) r nodesoptarget isinstancestrsplithasattrrgetattrsetattrr)rr r!r" submoduleorig_to_split_fqn_mappingnr*target_name_partscurrorig_gmnameleaf_node_name leaf_noderrrr s(       r leftrightcCs$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) match_outputmatch_placeholderr)rmatchlen)r:r;matchermatchesrrrr Ss  r N)rr)typingrrtorch.fx._compatibilityrtorch.fx.graphrtorch.fx.graph_moduler#torch.fx.passes.utils.matcher_utilsrtorch.nnr__all__r r,r boolr rrrrs2       7