fix ut new_group_api (#44764)

d8fedcb9 · kuizhiqing · GitHub · 1bd6e28f · d8fedcb9
隐藏空白更改
内联并排

Showing with 1 addition and 1 deletion

python/paddle/fluid/tests/unittests/collective_allreduce_new_group_api.py ...uid/tests/unittests/collective_allreduce_new_group_api.py +1 -1

未找到文件。
--- 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]