o ߗi-@sddlZddZddZdS)NcCs2t|tjrd|jd|jd|jdSt|S)zD Helper function to print out debug info in a friendly way. zTensor(z, grad=z, dtype=)) isinstancetorchTensorshape requires_graddtypestr)vr a/home/ubuntu/transcripts/venv/lib/python3.10/site-packages/torch/distributed/pipelining/_debug.pyfriendly_debug_infos r cCstjj|tS)zp Helper function to apply `friendly_debug_info` to items in `a`. `a` may be a list, tuple, or dict. )rfxnode map_aggregater )ar r r map_debug_infosr)rr rr r r r s