
    sg                     X    d dl mZ d dlZd dlmZ dej
                  j                  fdZy)    )DictN)_ModuleFrame
orig_graphc                 ~   t         j                  j                  t         j                  j	                         t         j                  j                               }i }i }t        | t        | j                        ||d dgdi |	      j                          |j                  j                          |j                          |S )N )module)torchfxGraphModulennModuleGraphr   tuplenodes	run_outergraphlint	recompile)r   
new_module
seen_nodesseen_moduless       Z/var/www/html/venv/lib/python3.12/site-packages/torch/distributed/pipelining/_unflatten.py_outline_submodulesr   	   s    %%ehhoo&79IJJ+-J/1Lj	


 ik    )typingr   r	   torch.export.unflattenr   r
   r   r    r   r   <module>r      s$      /EHHNN r   