Skip to content

Commit 1d2c99c

Browse files
author
Yang Yang(Tony)
authored
fix nccl root number (#7842)
* fix call once logic * clean up * further clean up * fix root number
1 parent 25cb906 commit 1d2c99c

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

paddle/operators/nccl_op_test.cu.cc

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -241,7 +241,7 @@ TEST_F(NCCLTester, ncclReduceOp) {
241241
// ncclBcastOp with desc
242242
TEST_F(NCCLTester, ncclBcastOp) {
243243
std::unique_ptr<f::OpDesc> op2(new f::OpDesc);
244-
const int kRoot = 5;
244+
const int kRoot = 0;
245245
op2->SetType("ncclBcast");
246246
op2->SetInput("X", {"st"});
247247
op2->SetInput("Communicator", {"comm"});

0 commit comments

Comments
 (0)