diff options
| author |  Jeremy Lau <lauj@google.com> | 2018-01-22 17:26:43 -0800 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2018-01-22 17:30:59 -0800 |
| commit | 6042b5d267f42d004087b44c29525951700579f9 (patch) | |
| tree | c4aa81597cc062a64b200758ce6dec5a1f5f16e5 /tensorflow/core/distributed_runtime/rpc/BUILD | |
| parent | 2968447d32bdfd0dd6fafabfcd1aafd6dc261803 (diff) | |
Reject retried RecvTensor requests.
Retried RecvTensorRequests are problematic because a RecvTensor with no
corresponding sender will wait forever, and the tensor may have been delivered
to a previous retry.
This change adds a unique request_id to each RecvTensor request, and we check
these request_ids against a set of recent request_ids. If a request_id is in the
recent set, we reject the RecvTensor request.
PiperOrigin-RevId: 182863245
Diffstat (limited to 'tensorflow/core/distributed_runtime/rpc/BUILD')
| -rw-r--r-- | tensorflow/core/distributed_runtime/rpc/BUILD | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/tensorflow/core/distributed_runtime/rpc/BUILD b/tensorflow/core/distributed_runtime/rpc/BUILD index 80640c806d..dade26abc6 100644 --- a/tensorflow/core/distributed_runtime/rpc/BUILD +++ b/tensorflow/core/distributed_runtime/rpc/BUILD @@ -186,6 +186,7 @@ tf_cuda_library( "//tensorflow/core:lib_internal", "//tensorflow/core:worker_proto_cc", "//tensorflow/core/distributed_runtime:graph_mgr", + "//tensorflow/core/distributed_runtime:recent_request_ids", "//tensorflow/core/distributed_runtime:rendezvous_mgr_interface", "//tensorflow/core/distributed_runtime:worker", "//tensorflow/core/distributed_runtime:worker_cache", @@ -270,6 +271,7 @@ cc_library( "//tensorflow/core:framework", "//tensorflow/core:lib", "//tensorflow/core/distributed_runtime:base_rendezvous_mgr", + "//tensorflow/core/distributed_runtime:request_id", "//tensorflow/core/distributed_runtime:tensor_coding", "//tensorflow/core/distributed_runtime:worker_cache", "//tensorflow/core/distributed_runtime:worker_env", |