revert process_group

PaddlePaddle · Jan 16, 2025 · 305f5af · 305f5af
1 parent 16723e9
commit 305f5af
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/python/paddle/distributed/auto_parallel/static/process_group.py b/python/paddle/distributed/auto_parallel/static/process_group.py
@@ -229,7 +229,9 @@ def instantiate(self):
                 alltoall_tmp = paddle.empty(
                     shape=[self.nranks, self.nranks], dtype="int32"
                 )
-                paddle._C_ops.all_to_all(alltoall_tmp, ring_id)
+                paddle._legacy_C_ops.all_to_all(
+                    alltoall_tmp, 'use_calc_stream', True, 'ring_id', ring_id
+                )
                 paddle.device.cuda.synchronize()
 
         if self.nranks > 1: