diff options
| author |  Ayush Dubey <ayushd@google.com> | 2018-08-31 16:24:01 -0700 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2018-08-31 16:33:51 -0700 |
| commit | c72dc92100be6169ed5ae5b59ba7a91b34cb2db4 (patch) | |
| tree | b4184ffebef6a41fb55566c897bd1a8dce81eacf /tensorflow/core/BUILD | |
| parent | f66aa92e0e6a8ce83637f3012fb37e6a93a4e9a0 (diff) | |
BEGIN_PUBLIC
Rollback of rollback. Fix: make access to collective_graph_key thread-safe.
The original change introduced a collective_graph_key_ integer to DirectSession, but it did not protect accesses to this integer. This change protects access with a mutex.
END_PUBLIC
Automated rollback of commit cb9443831283c2366e3dd91001db6362d6594f66
PiperOrigin-RevId: 211161961
Diffstat (limited to 'tensorflow/core/BUILD')
| -rw-r--r-- | tensorflow/core/BUILD | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/tensorflow/core/BUILD b/tensorflow/core/BUILD index 599db6bb3e..5c314f359c 100644 --- a/tensorflow/core/BUILD +++ b/tensorflow/core/BUILD @@ -4066,6 +4066,7 @@ tf_cuda_cc_test( ":testlib", "//third_party/eigen3", "//tensorflow/cc:cc_ops", + "//tensorflow/core/kernels:collective_ops", "//tensorflow/core/kernels:control_flow_ops", "//tensorflow/core/kernels:cwise_op", "//tensorflow/core/kernels:dense_update_ops", @@ -4107,6 +4108,7 @@ tf_cc_test( "//tensorflow/cc:cc_ops", # Link with support for TensorFlow Debugger (tfdbg). "//tensorflow/core/debug", + "//tensorflow/core/kernels:collective_ops", "//tensorflow/core/kernels:control_flow_ops", "//tensorflow/core/kernels:cwise_op", "//tensorflow/core/kernels:dense_update_ops", |