o h @sddlmZddlmZddlmZddlmZddlm Z gdZ e ddGd d d eZ e dddd ed ede defddZ e dddededefddZdS))Module) GraphModule)Graph)SubgraphMatcher) compatibility) HolderModulelift_subgraph_as_modulecompare_graphsF)is_backward_compatiblecs eZdZdZfddZZS)rzy 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__R/var/www/html/ai/venv/lib/python3.10/site-packages/torch/fx/passes/utils/common.pyr s zHolderModule.__init__)__name__ __module__ __qualname____doc__r __classcell__rrrrr srrgmsubgraph class_namereturnc Csti}|jD]J}|jdvrq|j}t|tsJ|d}|}|}|ddD]} t|| s7|| tit || }t || }q(|d} t || } t || | qt |||S)aB 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 class_name (str): name for the submodule ) call_moduleget_attr.N) rnodesoptarget isinstancestrsplithasattrrgetattrsetattrr) rrr submodulenr&target_name_partscurrorig_gmnameleaf_node_name leaf_noderrrrs$        rleftrightcCs$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)r5r6matchermatchesrrrr Gs  r N)torch.nnrtorch.fx.graph_modulertorch.fx.graphr#torch.fx.passes.utils.matcher_utilsrtorch.fx._compatibilityr__all__rr(rboolr rrrrs     .