From d8fedcb947688cde0f64b56240995363a80570f3 Mon Sep 17 00:00:00 2001
From: kuizhiqing <kuizhiqing@baidu.com>
Date: Tue, 2 Aug 2022 10:41:59 +0800
Subject: [PATCH] fix ut new_group_api (#44764)

---
 .../fluid/tests/unittests/collective_allreduce_new_group_api.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

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 859161af456..9d4e21aefff 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]
 
 
-- 
GitLab