tj (ddlmZddlmZddlmZddlmZddlm Z gdZ edGd d e Z ed dd edede de de eee e fff dZeddededefdZdS)) compatibility)Graph) GraphModule)SubgraphMatcher)Module) HolderModulelift_subgraph_as_modulecompare_graphsF)is_backward_compatiblecBeZdZdZdeeedzfddffd ZxZS)rzy HolderModule is used to copy all the attributes from original module to submodules that uses the attributes dNreturnct|D]\}}|||dS)N)super__init__items add_module)selfr kv __class__s Z/var/www/html/CA-Chatbot/venv/lib/python3.11/site-packages/torch/fx/passes/utils/common.pyrzHolderModule.__init__sU GGII " "DAq OOAq ! ! ! ! " ") __name__ __module__ __qualname____doc__dictstrrr __classcell__)rs@rrr sc "$sFTM12"t""""""""""rrrgmsubgraph comp_name class_namerc@ti}i}|jD]}|jdvr |j}t |t st dt||d}|} |} |ddD]U} t| | s#| | tit| | } t| | } V|d} t| | } |d|||<t| | | t||||fS)a Create a GraphModule for subgraph, which copies the necessary attributes from the original parent graph_module. Args: gm (GraphModule): parent graph module subgraph (:class:`torch.fx.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_attrzExpected str target, got .N)rnodesoptarget isinstancerAssertionErrortypesplithasattrrgetattrsetattrr)r"r#r$r% submoduleorig_to_split_fqn_mappingnr-target_name_partscurrorig_gmnameleaf_node_name leaf_nodes rr r sQ:R  I02 ^11 42 2 2 &#&& M !KT&\\!K!KLL L"LL--%crc* - -D4&& 8l2&6&67774&&Dgt,,GG*2.G^44 /8,C,C6,C,C!&)ni0000 y(J 7 79R RRrleftrightcvt|dd}||}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?matchermatchess rr r Ss:dNNNGmmE""G w<rPs111111 ------?????? H G Ge,,, " " " " "6 " "-, "e,,,# 7S7S7S7S7S 7S  ;S#X &' 7S7S7S-,7Ste,,,  u    -,   r