[fluid_ops] test_comm_cost.py modify c_allreduce_sum (#72112)

co63oc · web-flow · commit 75d488066e1c · 2025-04-10T11:01:13.000+08:00
diff --git a/test/auto_parallel/test_comm_cost.py b/test/auto_parallel/test_comm_cost.py
@@ -23,7 +23,7 @@
 from paddle.distributed.auto_parallel.static.cluster import Cluster
 from paddle.distributed.auto_parallel.static.cost import (
     AllgatherOpCost,
-    AllreduceSumOpCost,
+    AllReduceOpCost,
     BroadcastOpCost,
     CommContext,
     IdentityOpCost,
@@ -58,12 +58,13 @@ def test_comm_cost(self):
 
         # Check AllreduceSumCost 128MB ring cost
         allreduce_sum_op_desc = build_comm_desc(
-            "c_allreduce_sum",
+            "all_reduce",
             [0, 1, 2, 3, 4, 5, 6, 7],
             paddle.float32,
             [1, 32 * (10**6)],
+            {"reduce_type": paddle.distributed.ReduceOp.SUM},
         )
-        allreduce_sum_op_cost = AllreduceSumOpCost(
+        allreduce_sum_op_cost = AllReduceOpCost(
             op_desc=allreduce_sum_op_desc, comm_context=comm_context
         )
 
@@ -138,12 +139,13 @@ def test_cross_machine_comm_cost(self):
 
         # Check AllreduceSumCost 128MB ring cost
         allreduce_sum_op_desc = build_comm_desc(
-            "c_allreduce_sum",
+            "all_reduce",
             [0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15],
             paddle.float32,
             [1, 32 * (10**6)],
+            {"reduce_type": paddle.distributed.ReduceOp.SUM},
         )
-        allreduce_sum_op_cost = AllreduceSumOpCost(
+        allreduce_sum_op_cost = AllReduceOpCost(
             op_desc=allreduce_sum_op_desc, comm_context=comm_context
         )
 
