U ʹj @sddlmZddlmZddlmZddlmZddlm Z dddgZ e d d Gd ddeZ e d d deee ed ddZ e d d eeedddZdS))Module) GraphModule)Graph)SubgraphMatcher) compatibility HolderModulelift_subgraph_as_modulecompare_graphsF)Zis_backward_compatiblecs eZdZdZfddZZS)rzy HolderModule is used to copy all the attributes from original module to submodules that uses the attributes cs,t|D]\}}|||qdS)N)super__init__items add_module)selfdkv __class__[/var/www/html/bwave/python/venv/lib/python3.8/site-packages/torch/fx/passes/utils/common.pyr s zHolderModule.__init__)__name__ __module__ __qualname____doc__r __classcell__rrrrr sr)gmsubgraph class_namereturnc Csti}|jD]}|jdkrq|j}t|ts2t|d}|}|}|ddD]2} t|| sn| | tit || }t || }qP|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 )Z call_moduleZget_attr.N) rZnodesoptarget isinstancestrAssertionErrorsplithasattrr getattrsetattrr) rrr submodulenr"Ztarget_name_partscurrZorig_gmnameZleaf_node_nameZ leaf_noderrrrs$       )leftrightrcCs$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)Z match_outputZmatch_placeholderr)rmatchlen)r.r/Zmatchermatchesrrrr Gs  N)r)Ztorch.nnrZtorch.fx.graph_modulerZtorch.fx.graphrZ#torch.fx.passes.utils.matcher_utilsrZtorch.fx._compatibilityr__all__rr$rboolr rrrrs       .