
    sg-                         d dl Z d Zd Zy)    Nc                     t        | t        j                        r*d| j                   d| j                   d| j
                   dS t        |       S )zD
    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)vs    V/var/www/html/venv/lib/python3.12/site-packages/torch/distributed/pipelining/_debug.pyfriendly_debug_infor      sC     !U\\"	(9!''!LL1v    c                 ^    t         j                  j                  j                  | t              S )zp
    Helper function to apply `friendly_debug_info` to items in `a`.
    `a` may be a list, tuple, or dict.
    )r   fxnodemap_aggregater   )as    r   map_debug_infor      s     
 88==&&q*=>>r   )r   r   r    r   r   <module>r      s    ?r   