קg-ddlZdZdZdS)Nct|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)vs _/var/www/html/ai-engine/env/lib/python3.11/site-packages/torch/distributed/pipelining/_debug.pyfriendly_debug_inforsJ!U\""LLLLL!'LLLL1vv cVtjj|tS)zp Helper function to apply `friendly_debug_info` to items in `a`. `a` may be a list, tuple, or dict. )rfxnode map_aggregater)as r map_debug_infors 8= & &q*= > >>r)rrrrr rs8 ?????r