update code, P2POp add iherit parent class

lvyufeng · Dec 18, 2024 · cf78f50 · cf78f50
1 parent 3479faa
commit cf78f50
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/torch/distributed/distributed_c10d.py b/torch/distributed/distributed_c10d.py
@@ -409,7 +409,7 @@ def get_device_backend_map(self) -> Dict[str, Backend]:
         return self.device_backend_map
 
 
-class P2POp:
+class P2POp(mindspore.mint.distributed.P2POp):
     """
     A class to build point-to-point operations for ``batch_isend_irecv``.