o ‡ZŽh`ã@s<ddlZddlmZdedefdd„Zdedefdd „ZdS) éN)ÚArgumentÚvÚreturncCs2t|tjƒrd|j›d|j›d|j›dSt|ƒS)zD Helper function to print out debug info in a friendly way. zTensor(z, grad=z, dtype=ú))Ú isinstanceÚtorchZTensorÚshapeZ requires_gradZdtypeÚstr)r©r úR/var/www/auris/lib/python3.10/site-packages/torch/distributed/pipelining/_debug.pyÚfriendly_debug_infos r ÚacCstjj |t¡S)zp Helper function to apply `friendly_debug_info` to items in `a`. `a` may be a list, tuple, or dict. )rZfxÚnodeZ map_aggregater )r r r r Úmap_debug_infosr)rZ torch.fx.noderÚobjectr rr r r r Ús