
    IЦi                     r    S SK Jr  S SKJrJrJr  S SKrS SKJrJ	r	  S\R                  R                  4S jrg)    )defaultdict)DictListSetN)_ModuleFrame_SubmoduleEntry
orig_graphc                    [         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) r   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         f/var/www/html/ai-image-ml/venv/lib/python3.13/site-packages/torch/distributed/pipelining/_unflatten.py_outline_submodulesr!   
   s    %%ehhoo&79IJJ+-J5@5FL&1#&6J24Oj		

 ik    )collectionsr   typingr   r   r   r   torch.export.unflattenr   r   r   r   r!    r"   r    <module>r'      s*    $ " "  @EHHNN r"   