aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/contrib/nccl
diff options
context:
space:
mode:
authorGravatar Shanqing Cai <cais@google.com>2017-10-23 10:26:34 -0700
committerGravatar TensorFlower Gardener <gardener@tensorflow.org>2017-10-23 10:29:59 -0700
commit0e56ffb7b7cddaf3f0521747d2fade90c56b586f (patch)
tree77e1491c4fc9a5070c5be9b187867d472c724853 /tensorflow/contrib/nccl
parent3ed049b673e1a2b58e197fd8429ed81a015cd351 (diff)
Fix breakages in OSS builds
See example breakages logs at: http://ci.tensorflow.org/job/tensorflow-cl-cpu-python3-pip/10847/console http://ci.tensorflow.org/job/tensorflow-cl-gpu/11008/console 1. CL/172477381 added the no_oss tag to tests with oss_serial tags, which broke the logic of OSS_SERIAL tests in pip.sh and run_pip_test.sh. This CL fixes that. 2. The nccl_kernels BUILD target in contrib/nccl/BUILD was missing some dependencies. This CL adds the missing ones. Fixes: #13918 PiperOrigin-RevId: 173133914
Diffstat (limited to 'tensorflow/contrib/nccl')
-rw-r--r--tensorflow/contrib/nccl/BUILD2
1 files changed, 2 insertions, 0 deletions
diff --git a/tensorflow/contrib/nccl/BUILD b/tensorflow/contrib/nccl/BUILD
index 5e7263ff62..3aa3215a5f 100644
--- a/tensorflow/contrib/nccl/BUILD
+++ b/tensorflow/contrib/nccl/BUILD
@@ -74,8 +74,10 @@ tf_kernel_library(
"kernels/nccl_rewrite.cc",
],
deps = [
+ "//tensorflow/core:core_cpu",
"//tensorflow/core:framework",
"//tensorflow/core:gpu_headers_lib",
+ "//tensorflow/core:lib",
"//tensorflow/core:proto_text",
"@nccl_archive//:nccl",
],