[fluid_ops] Modify legacy_C_ops.c_allreduce_sum (PaddlePaddle#70435)

enkilee · Dec 25, 2024 · 2df97ab · 2df97ab
1 parent 3c2d8c7
commit 2df97ab
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/python/paddle/distributed/fleet/layers/mpu/mp_ops.py b/python/paddle/distributed/fleet/layers/mpu/mp_ops.py
@@ -255,12 +255,10 @@ def forward(
             group.process_group.all_reduce_on_calc_stream(tensor, op_type)
             return tensor
         else:
-            return _legacy_C_ops.c_allreduce_sum_(
+            return _C_ops.all_reduce_(
                 tensor,
-                'use_calc_stream',
-                use_calc_stream,
-                'ring_id',
                 group.id,
+                paddle.distributed.ReduceOp.SUM,
             )
 
     @staticmethod