U yh-@sddlZddZddZdS)NcCs6t|tjr*d|jd|jd|jdSt|SdS)zD Helper function to print out debug info in a friendly way. zTensor(z, grad=z, dtype=)N) isinstancetorchZTensorshapeZ requires_gradZdtypestr)vrU/var/www/html/venv/lib/python3.8/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. )rZfxnodeZ map_aggregater )arrr map_debug_infosr )rr r rrrr s