
    [Th                         S SK Jr  S SKrS SKJrJr  S\R                  R                  S\R                  R                  4S jr	g)    )defaultdictN)_ModuleFrame_SubmoduleEntry
orig_graphreturnc                    [         R                  R                  [         R                  R	                  5       [         R                  R                  5       5      n0 n[        [        5      n[        [        5      n0 n[        U [        U R                  5      UUUUS S/S0 US9R                  5         UR                  R                  5         UR                  5         U$ )N) Nr   r	   )module)torchfxGraphModulennModuleGraphr   listsetr   tuplenodes	run_outergraphlint	recompile)r   
new_module
seen_nodesseen_modules
seen_attrscreated_moduless         _/var/www/auris/envauris/lib/python3.13/site-packages/torch/distributed/pipelining/_unflatten.py_outline_submodulesr      s    %%ehhoo&79IJJ+-J5@5FL&1#&6J24Oj	

 ik    )
collectionsr   r   torch.export.unflattenr   r   r   r   r   r    r    r   <module>r$      s3    #  @EHHNN uxx7K7K r    