`i`:ddlZddlmZdedefdZdedefdZdS)N)Argumentvreturnct|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)rs w/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/torch/distributed/pipelining/_debug.pyfriendly_debug_inforsJ!U\""LLLLL!'LLLL1vv acVtjj|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 8= & &q*= > >>r)r torch.fx.noderobjectrrrrrsp """"""6h?h?8??????r