diff options
Diffstat (limited to 'tensorflow/contrib/nccl/BUILD')
-rw-r--r-- | tensorflow/contrib/nccl/BUILD | 21 |
1 files changed, 11 insertions, 10 deletions
diff --git a/tensorflow/contrib/nccl/BUILD b/tensorflow/contrib/nccl/BUILD index 62996d1fd8..9a9d480260 100644 --- a/tensorflow/contrib/nccl/BUILD +++ b/tensorflow/contrib/nccl/BUILD @@ -31,9 +31,11 @@ tf_custom_op_library( "kernels/nccl_manager.h", "kernels/nccl_ops.cc", ]), - deps = if_cuda([ + deps = [] + if_cuda([ "@local_config_nccl//:nccl", "//tensorflow/core:gpu_headers_lib", + "//tensorflow/core:protos_all_cc", + "//tensorflow/core:protos_all_proto_text", ]), ) @@ -57,32 +59,31 @@ tf_cuda_cc_test( "notap", ], deps = - [ + if_cuda([ + "@local_config_nccl//:nccl", "//tensorflow/core:cuda", "//tensorflow/core:test", "//tensorflow/core:test_main", "//tensorflow/core:testlib", - "@local_config_nccl//:nccl", - ], + ]), ) tf_kernel_library( name = "nccl_kernels", - srcs = [ + srcs = if_cuda([ "kernels/nccl_manager.cc", "kernels/nccl_manager.h", "kernels/nccl_ops.cc", "kernels/nccl_rewrite.cc", - ], - deps = [ + ]), + deps = if_cuda([ + "@local_config_nccl//:nccl", "//tensorflow/core:core_cpu", "//tensorflow/core:framework", "//tensorflow/core:gpu_headers_lib", "//tensorflow/core:lib", - "//tensorflow/core:proto_text", "//tensorflow/core:stream_executor", - "@local_config_nccl//:nccl", - ], + ]), alwayslink = 1, ) |