§
    Ÿ²Åjø  ã                   ó^   — d dl mZ d dlmZmZmZ d dlZd dlmZm	Z	 dej
        j        fd„ZdS )é    )Údefaultdict)ÚDictÚListÚSetN)Ú_ModuleFrameÚ_SubmoduleEntryÚ
orig_graphc                 óæ  — t           j                             t           j                             ¦   «         t           j                             ¦   «         ¦  «        }i }t          t          ¦  «        }t          t          ¦  «        }i }t          | t          | j        ¦  «        ||||d dgdi |¬¦  «                             ¦   «          |j                             ¦   «          |                     ¦   «          |S )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         úXC:\Projects\Historical\venv\Lib\site-packages\torch/distributed/pipelining/_unflatten.pyÚ_outline_submodulesr!   
   sÏ   € å”×%Ò%¥e¤h§o¢oÑ&7Ô&7½¼¿ºÑ9IÔ9IÑJÔJ€JØ+-€JÝ5@ÅÑ5FÔ5F€LÝ&1µ#Ñ&6Ô&6€JØ24€OÝØÝˆjÔÑÔØØØØØØ	ˆ	Ø
Ø
Øðñ ô ÷ ‚i�k„k€kØÔ×ÒÑÔÐØ×ÒÑÔÐØÐó    )Úcollectionsr   Útypingr   r   r   r   Útorch.export.unflattenr   r   r   r   r!   © r"   r    ú<module>r'      s�   ðð $Ð #Ð #Ð #Ð #Ð #Ø "Ð "Ð "Ð "Ð "Ð "Ð "Ð "Ð "Ð "à €€€Ø @Ð @Ð @Ð @Ð @Ð @Ð @Ð @ð E¤H¤Nð ð ð ð ð ð r"   