From 47f40c870895a0bdee50715dd3aab246915edbfe Mon Sep 17 00:00:00 2001
From: xutianming <tianmingxu.tmxu@gmail.com>
Date: Mon, 13 Apr 2020 15:30:48 +0800
Subject: [PATCH] Fix reduce op ut failure: root rank=-1

---
 tensorflow/core/nccl/nccl_manager.cc | 1 +
 1 file changed, 1 insertion(+)

diff --git a/tensorflow/core/nccl/nccl_manager.cc b/tensorflow/core/nccl/nccl_manager.cc
index 9f9fa28be0b..68af9fbb2f0 100644
--- a/tensorflow/core/nccl/nccl_manager.cc
+++ b/tensorflow/core/nccl/nccl_manager.cc
@@ -452,6 +452,7 @@ void NcclManager::AddReduceSend(std::unique_ptr<Participant> participant,
 void NcclManager::AddReduceRecv(std::unique_ptr<Participant> participant,
                                 const Context& context,
                                 ncclRedOp_t reduction_op) {
+  participant->root = true;
   AddParticipant(std::move(participant), context, kReduce, reduction_op);
 }