diff options
| author |  Shrinidhi KL <klshrinidhi@users.noreply.github.com> | 2018-01-26 14:28:34 -0500 |
|---|---|---|
| committer |  Rasmus Munk Larsen <rmlarsen@google.com> | 2018-01-26 11:28:34 -0800 |
| commit | 62b54bfc4bf112eab2a60034fe0ff7b6cfb2ddd7 (patch) | |
| tree | 6a9d6adf28d7abf571a23dbf39b975882692a618 /tensorflow/contrib/mpi | |
| parent | a169db5d066efd696d80fd989091a54cc4fe5baf (diff) | |
Fix build errors in contrib/mpi introduced by commit 6042b5d267f (#16427)
* Add missing header.
* Fix typo. Should refer to incoming argument.
Diffstat (limited to 'tensorflow/contrib/mpi')
| -rw-r--r-- | tensorflow/contrib/mpi/mpi_rendezvous_mgr.cc | 2 | ||||
| -rw-r--r-- | tensorflow/contrib/mpi/mpi_rendezvous_mgr.h | 1 |
2 files changed, 2 insertions, 1 deletions
diff --git a/tensorflow/contrib/mpi/mpi_rendezvous_mgr.cc b/tensorflow/contrib/mpi/mpi_rendezvous_mgr.cc index 8d14a3ef04..0252bc7992 100644 --- a/tensorflow/contrib/mpi/mpi_rendezvous_mgr.cc +++ b/tensorflow/contrib/mpi/mpi_rendezvous_mgr.cc @@ -152,7 +152,7 @@ MPIRemoteRendezvous::~MPIRemoteRendezvous() {} void MPIRendezvousMgr::AddRequest(RecvTensorRequest request, const int mpi_dst) { TF_CHECK_OK(recv_tensor_recent_request_ids_.TrackUnique( - req.request_id(), "RecvTensor (MPIRendezvousMgr)", req)); + request.request_id(), "RecvTensor (MPIRendezvousMgr)", request)); const int64 step_id = request.step_id(); const std::string& key = request.rendezvous_key(); Rendezvous::ParsedKey parsed; diff --git a/tensorflow/contrib/mpi/mpi_rendezvous_mgr.h b/tensorflow/contrib/mpi/mpi_rendezvous_mgr.h index ca42ee2f6d..d35e65363f 100644 --- a/tensorflow/contrib/mpi/mpi_rendezvous_mgr.h +++ b/tensorflow/contrib/mpi/mpi_rendezvous_mgr.h @@ -33,6 +33,7 @@ limitations under the License. #include "tensorflow/contrib/mpi/mpi_msg.pb.h" #include "tensorflow/contrib/mpi/mpi_utils.h" #include "tensorflow/core/distributed_runtime/base_rendezvous_mgr.h" +#include "tensorflow/core/distributed_runtime/recent_request_ids.h" #include "tensorflow/core/distributed_runtime/request_id.h" #include "tensorflow/core/distributed_runtime/worker_env.h" #include "tensorflow/core/protobuf/worker.pb.h" |