o Zh@sBddlmZddlZddlmZmZdejjdejjfddZ dS)) defaultdictN) _ModuleFrame_SubmoduleEntry orig_graphreturnc Csrtjtjtj}i}tt}tt}i}t |t |j ||||ddgdi|d |j ||S)N)Nrr)module)torchfx GraphModulennModuleGraphrlistsetrtuplenodesZ run_outergraphZlintZ recompile)rZ new_moduleZ seen_nodesZ seen_modulesZ seen_attrsZcreated_modulesrV/var/www/auris/lib/python3.10/site-packages/torch/distributed/pipelining/_unflatten.py_outline_submoduless, r) collectionsrr Ztorch.export.unflattenrrr rr rrrrrs