Vh`8ddlZddlmZdedefdZdedefdZy)N)Argumentvreturnct|tjr*d|jd|jd|j dSt |S)zD Helper function to print out debug info in a friendly way. zTensor(z, grad=z, dtype=)) isinstancetorchTensorshape requires_graddtypestr)rs S/home/dcms/DCMS/lib/python3.12/site-packages/torch/distributed/pipelining/_debug.pyfriendly_debug_inforsC!U\\" (9!''!LL1v ac^tjjj|tS)zp Helper function to apply `friendly_debug_info` to items in `a`. `a` may be a list, tuple, or dict. )r fxnode map_aggregater)rs rmap_debug_infors 88== & &q*= >>r)r torch.fx.noderobjectrrrrrs3 "6h?h?8?r