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

Address review comments in #24864.

Browse Source 
Signed-off-by: Bairen Yi <byronyi@clustar.ai>
This commit is contained in:
Bairen Yi 2019-01-17 22:48:05 +00:00
parent a423ca4725
commit 252aa8b566
2 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
tensorflow/contrib/gdr/gdr_worker.cc
View File

@ -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);

2
tensorflow/contrib/gdr/gdr_worker.h
View File

@ -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