
    [Th`                     <    S SK r S SKJr  S\S\4S jrS\S\4S jrg)    N)Argumentvreturnc                     [        U [        R                  5      (       a*  SU R                   SU R                   SU R
                   S3$ [        U 5      $ )z<
Helper function to print out debug info in a friendly way.
zTensor(z, grad=z, dtype=))
isinstancetorchTensorshaperequires_graddtypestr)r   s    [/var/www/auris/envauris/lib/python3.13/site-packages/torch/distributed/pipelining/_debug.pyfriendly_debug_infor      sF     !U\\""	(9!''!LL1v    ac                 ^    [         R                  R                  R                  U [        5      $ )zd
Helper function to apply `friendly_debug_info` to items in `a`.
`a` may be a list, tuple, or dict.
)r	   fxnodemap_aggregater   )r   s    r   map_debug_infor      s     
 88==&&q*=>>r   )r	   torch.fx.noder   objectr   r    r   r   <module>r      s3     "6 h ?h ?8 ?r   