^j 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<eZdZdZdeeedzfddffd ZxZS)rzy HolderModule is used to copy all the attributes from original module to submodules that uses the attributes dNreturncvt||jD]\}}|j||y)N)super__init__items add_module)selfr kv __class__s g/var/www/ramen.bs-engineer-server.com/venv/lib/python3.12/site-packages/torch/fx/passes/utils/common.pyrzHolderModule.__init__s5 GGI "DAq OOAq ! ") __name__ __module__ __qualname____doc__dictstrrr __classcell__)rs@rrr s. "$sFTM12"t""rrgmsubgraph comp_name class_namercti}i}|jD]}|jdvr|j}t |t st dt||jd}|} |} |ddD]A} t| | s| j| tit| | } t| | } C|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 s:R I02 ^^1 442 2 &#& #matchermatchess rr r Ss/dNGmmE"G wrPs1 -? He, "6 "- "e,# 7S7S7S7S 7S  ;S#X &' 7S-7Ste,  u  - r