Fix

Author: co63oc

python/paddle/distributed/auto_parallel/static/helper.py

@@ -456,9 +456,7 @@ def init_pir(self, main_program, place):
             barrier_tensor = paddle.full([1], 1, dtype="int32")
             # barrier is not available in xpu for now
             if not paddle.framework.core.is_compiled_with_xpu():
-                paddle._legacy_C_ops.barrier(
-                    barrier_tensor, barrier_tensor, 'ring_id', 0
-                )
+                paddle._C_ops.barrier(barrier_tensor, 0)
             paddle.enable_static()
 
     def init(self, main_program, place, dist_context):
@@ -600,9 +598,7 @@ def init(self, main_program, place, dist_context):
             barrier_tensor = paddle.full([1], 1, dtype="int32")
             # barrier is not available in xpu for now
             if not paddle.framework.core.is_compiled_with_xpu():
-                paddle._legacy_C_ops.barrier(
-                    barrier_tensor, barrier_tensor, 'ring_id', 0
-                )
+                paddle._C_ops.barrier(barrier_tensor, 0)
             paddle.enable_static()
 
     def cache_whole_graph_dist_attr(self, all_params):

python/paddle/distributed/auto_parallel/static/process_group.py

@@ -222,9 +222,7 @@ def instantiate(self):
             barrier_tensor = paddle.full([1], 1, dtype="int32")
             # barrier is not available in xpu for now
             if not paddle.framework.core.is_compiled_with_xpu():
-                paddle._legacy_C_ops.barrier(
-                    barrier_tensor, barrier_tensor, 'ring_id', ring_id
-                )
+                paddle._C_ops.barrier(barrier_tensor, ring_id)
 
             # NOTE(zhiqiu): to avoid send/recv hang in lazy init
             if self._group_type == 'p2p':
@@ -240,9 +238,7 @@ def instantiate(self):
             barrier_tensor = paddle.full([1], 1, dtype="int32")
             # barrier is not available in xpu for now
             if not paddle.framework.core.is_compiled_with_xpu():
-                paddle._legacy_C_ops.barrier(
-                    barrier_tensor, barrier_tensor, 'ring_id', 0
-                )
+                paddle._C_ops.barrier(barrier_tensor, 0)
 
         self._is_instantiate = True
 

python/paddle/distributed/communication/group.py

@@ -347,9 +347,7 @@ def barrier(group: Group | None = None) -> None:
     if framework.in_dynamic_mode():
         # barrier is not available in xpu for now
         if not paddle.framework.core.is_compiled_with_xpu():
-            return paddle._legacy_C_ops.barrier(
-                barrier_tensor, barrier_tensor, 'ring_id', ring_id
-            )
+            return paddle._C_ops.barrier(barrier_tensor, ring_id)
     else:
         op_type = 'barrier'
         if not isinstance(ring_id, int):