diff --git a/python/paddle/fluid/tests/unittests/collective_allreduce_new_group_api.py b/python/paddle/fluid/tests/unittests/collective_allreduce_new_group_api.py index 859161af456e25d6517c97903bf6ceb5d1e32e38..9d4e21aefff56c975825b1758ec4031568a32fff 100644 --- a/python/paddle/fluid/tests/unittests/collective_allreduce_new_group_api.py +++ b/python/paddle/fluid/tests/unittests/collective_allreduce_new_group_api.py @@ -51,7 +51,7 @@ class TestCollectiveAllreduceNewGroupAPI(TestCollectiveAPIRunnerBase): gp = paddle.distributed.new_group([0, 1]) paddle.distributed.all_reduce(tindata, group=gp, - use_calc_stream=False) + use_calc_stream=True) return [tindata]