!&h# ddlmZddlmZddlmZddlmZddlm Z gdZ edGd d e Z ed dd ed ed e de de eee e fff dZeddededefdZy)) compatibility)Graph) GraphModule)SubgraphMatcher)Module) HolderModulelift_subgraph_as_modulecompare_graphsF)is_backward_compatiblec"eZdZdZfdZxZS)rzy HolderModule is used to copy all the attributes from original module to submodules that uses the attributes cvt||jD]\}}|j||y)N)super__init__items add_module)selfdkv __class__s r/var/www/pru.catia.catastroantioquia-mas.com/valormas/lib/python3.12/site-packages/torch/fx/passes/utils/common.pyrzHolderModule.__init__s5 GGI "DAq OOAq ! ")__name__ __module__ __qualname____doc__r __classcell__)rs@rrr s ""rrgmsubgraph comp_name class_namereturncti}i}|jD]}|jdvr|j}t |t sJ|j d}|} |} |ddD]A} t| | s| j| tit| | } t| | } C|d} t| | } |d|||<t| | | 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) rnodesoptarget isinstancestrsplithasattrrgetattrsetattrr)rrr r! submoduleorig_to_split_fqn_mappingnr*target_name_partscurrorig_gmnameleaf_node_name leaf_nodes rr r s6R I02 ^^1 442 2 &#&&&"LL-%cr* -D4&l2&674&Dgt,G  -+2.G^4 /8k6(,C!&)ni0-10 y(J 79R RRrleftrightc\t|dd}|j|}t|dkDS)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;matchermatchess rr r Rs/dNGmmE"G wrNs2 -? He, "6 "- "e,# 4S4S4S4S 4S  ;S#X &' 4S-4Sne,  u  - r