Update

[ghstack-poisoned]
pytorch · May 17, 2024 · 3f99204 · 3f99204
1 parent e014c20
commit 3f99204
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/torch/distributed/distributed_c10d.py b/torch/distributed/distributed_c10d.py
@@ -422,7 +422,7 @@ def __repr__(self):
         else:
             return super().__repr__()
 
-        return f"P2POp({op_name} pg={group_name}, s={s}, d={d},  numel={self.tensor.numel()}, {self.tensor.dtype})"
+        return f"P2POp({op_name} pg={group_name}, s={s}, d={d},  {self.tensor.shape}, {self.tensor.dtype})"
 
 class _CollOp:
     """