From 252aa8b5661559377e86b4763980f59f214b9eee Mon Sep 17 00:00:00 2001 From: Bairen Yi Date: Thu, 17 Jan 2019 22:48:05 +0000 Subject: [PATCH] Address review comments in #24864. Signed-off-by: Bairen Yi --- tensorflow/contrib/gdr/gdr_worker.cc | 6 +++--- tensorflow/contrib/gdr/gdr_worker.h | 2 +- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/tensorflow/contrib/gdr/gdr_worker.cc b/tensorflow/contrib/gdr/gdr_worker.cc index 29ee4c93752..a2425557da4 100644 --- a/tensorflow/contrib/gdr/gdr_worker.cc +++ b/tensorflow/contrib/gdr/gdr_worker.cc @@ -44,13 +44,13 @@ GdrWorker::GdrWorker(WorkerEnv* worker_env, const ConfigProto& config, RemoteMemoryManager* remote_memory_manager) : GrpcWorker(worker_env, config), remote_memory_manager_(remote_memory_manager), - recv_tensor_recent_request_ids_(100000) {} + recent_request_ids_(100000) {} void GdrWorker::GrpcRecvTensorAsync(CallOptions* opts, const RecvTensorRequest* request, ::grpc::ByteBuffer* response, StatusCallback done) { - Status s = recv_tensor_recent_request_ids_.TrackUnique( + Status s = recent_request_ids_.TrackUnique( request->request_id(), "RecvTensor (GdrWorker)", *request); if (!s.ok()) { done(s); @@ -152,7 +152,7 @@ void GdrWorker::GrpcRecvTensorAsync(CallOptions* opts, void GdrWorker::RecvBufAsync(CallOptions* opts, const RecvBufRequest* request, RecvBufResponse* response, StatusCallback done) { // This is an RDMA enabled implementation augmenting grpc. - Status s = recv_tensor_recent_request_ids_.TrackUnique( + Status s = recent_request_ids_.TrackUnique( request->request_id(), "RecvBuf (GdrWorker)", *request); if (!s.ok()) { done(s); diff --git a/tensorflow/contrib/gdr/gdr_worker.h b/tensorflow/contrib/gdr/gdr_worker.h index 7af03d16fca..9a85cfd4263 100644 --- a/tensorflow/contrib/gdr/gdr_worker.h +++ b/tensorflow/contrib/gdr/gdr_worker.h @@ -44,7 +44,7 @@ class GdrWorker : public GrpcWorker { private: RemoteMemoryManager* remote_memory_manager_; // Not owned - RecentRequestIds recv_tensor_recent_request_ids_; + RecentRequestIds recent_request_ids_; }; } // namespace tensorflow