experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Fix reduce op ut failure: root rank=-1

Browse Source
This commit is contained in:
xutianming 2020-04-13 15:30:48 +08:00
parent 716bcea5a2
commit 47f40c8708
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
tensorflow/core/nccl/nccl_manager.cc
View File

@ -452,6 +452,7 @@ void NcclManager::AddReduceSend(std::unique_ptr<Participant> participant,
void NcclManager::AddReduceRecv(std::unique_ptr<Participant> participant, void NcclManager::AddReduceRecv(std::unique_ptr<Participant> participant,
const Context& context, const Context& context,
ncclRedOp_t reduction_op) { ncclRedOp_t reduction_op) {
participant->root = true;
AddParticipant(std::move(participant), context, kReduce, reduction_op); AddParticipant(std::move(participant), context, kReduce, reduction_op);
} }