Fix

co63oc · co63oc · commit 945fd20a27e6 · 2025-04-09T14:19:34.000+08:00
diff --git a/test/auto_parallel/test_comm_cost.py b/test/auto_parallel/test_comm_cost.py
@@ -143,6 +143,7 @@ def test_cross_machine_comm_cost(self):
             [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 = AllReduceOpCost(
             op_desc=allreduce_sum_op_desc, comm_context=comm_context

Original file line number	Diff line number	Diff line change
`@@ -143,6 +143,7 @@ def test_cross_machine_comm_cost(self):`
`143`	`143`	`[0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15],`
`144`	`144`	`paddle.float32,`
`145`	`145`	`[1, 32 * (10**6)],`
	`146`	`+ {"reduce_type": paddle.distributed.ReduceOp.SUM},`
`146`	`147`	`)`
`147`	`148`	`allreduce_sum_op_cost = AllReduceOpCost(`
`148`	`149`	`op_desc=allreduce_sum_op_desc, comm_context=comm_context`