P/jC ddlmZmZddlmZddlmZddlmZddl m Z ddl m Z gdZ ed Gd d e Zed  dd ed edededeeeeefff dZed dededefdZy))DictTuple) compatibility)Graph) GraphModule)SubgraphMatcher)Module) HolderModulelift_subgraph_as_modulecompare_graphsF)is_backward_compatiblec"eZdZdZfdZxZS)r zy 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 _/root/repos/pocs/video-evals/.venv/lib/python3.12/site-packages/torch/fx/passes/utils/common.pyrzHolderModule.__init__s5 GGI "DAq OOAq ! ")__name__ __module__ __qualname____doc__r __classcell__)rs@rr r s ""rr gmsubgraph 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) r nodesoptarget isinstancestrsplithasattrrgetattrsetattrr)r r!r"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 Ts/dNGmmE"G wrOs1 -? He, "6 "- "e,# 4S4S4S4S 4S  ;S#X &' 4S-4Sne,  u  - r