From 1d2c99c336c156e502a081b89e8f6937839e86e5 Mon Sep 17 00:00:00 2001 From: "Yang Yang(Tony)" Date: Wed, 24 Jan 2018 19:08:22 -0800 Subject: [PATCH] fix nccl root number (#7842) * fix call once logic * clean up * further clean up * fix root number --- paddle/operators/nccl_op_test.cu.cc | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/paddle/operators/nccl_op_test.cu.cc b/paddle/operators/nccl_op_test.cu.cc index 6546096069d..072e4eb2eff 100644 --- a/paddle/operators/nccl_op_test.cu.cc +++ b/paddle/operators/nccl_op_test.cu.cc @@ -241,7 +241,7 @@ TEST_F(NCCLTester, ncclReduceOp) { // ncclBcastOp with desc TEST_F(NCCLTester, ncclBcastOp) { std::unique_ptr op2(new f::OpDesc); - const int kRoot = 5; + const int kRoot = 0; op2->SetType("ncclBcast"); op2->SetInput("X", {"st"}); op2->SetInput("Communicator", {"comm"}); -- GitLab