diff --git a/configure.py b/configure.py index dfb87550b1c5ba..0b16fe13141ca8 100644 --- a/configure.py +++ b/configure.py @@ -1694,7 +1694,6 @@ def main(): config_info_line('nohdfs', 'Disable HDFS support.') config_info_line('noignite', 'Disable Apacha Ignite support.') config_info_line('nokafka', 'Disable Apache Kafka support.') - config_info_line('nonccl', 'Disable NVIDIA NCCL support.') if __name__ == '__main__': diff --git a/tensorflow/BUILD b/tensorflow/BUILD index fd4b94202aad24..17577afecb74b7 100644 --- a/tensorflow/BUILD +++ b/tensorflow/BUILD @@ -246,12 +246,6 @@ config_setting( visibility = ["//visibility:public"], ) -config_setting( - name = "no_nccl_support", - define_values = {"no_nccl_support": "true"}, - visibility = ["//visibility:public"], -) - # Crosses between platforms and file system libraries not supported on those # platforms due to limitations in nested select() statements. config_setting( diff --git a/tensorflow/core/BUILD b/tensorflow/core/BUILD index 1caca0028ed8e3..2a8c2718edd7fa 100644 --- a/tensorflow/core/BUILD +++ b/tensorflow/core/BUILD @@ -95,6 +95,7 @@ load("//tensorflow:tensorflow.bzl", "tf_cc_test_gpu") load("//tensorflow:tensorflow.bzl", "tf_cc_tests_gpu") load("//tensorflow:tensorflow.bzl", "tf_cuda_cc_test") load("//tensorflow:tensorflow.bzl", "tf_version_info_genrule") +load("//tensorflow:tensorflow.bzl", "if_not_tx2_llvm_or_windows_cuda") load("//tensorflow:tensorflow.bzl", "tf_cuda_only_cc_test") # For platform specific build config @@ -1402,11 +1403,9 @@ cc_library( "//tensorflow/core/kernels:summary_kernels", "//tensorflow/core/kernels:training_ops", "//tensorflow/core/kernels:word2vec_kernels", - ] + tf_additional_cloud_kernel_deps() + select({ - "//tensorflow:no_nccl_support": [], - "//tensorflow:windows": [], - "//conditions:default": ["//tensorflow/core/kernels:nccl_kernels"], - }) + if_not_windows([ + ] + tf_additional_cloud_kernel_deps() + if_not_tx2_llvm_or_windows_cuda([ + "//tensorflow/core/kernels:nccl_kernels", + ]) + if_not_windows([ "//tensorflow/core/kernels:fact_op", "//tensorflow/core/kernels:array_not_windows", "//tensorflow/core/kernels:math_not_windows", diff --git a/tools/bazel.rc b/tools/bazel.rc index 1fdf51f53e29c7..8c2052ee8ad6e6 100644 --- a/tools/bazel.rc +++ b/tools/bazel.rc @@ -72,7 +72,6 @@ build:nogcp --define=no_gcp_support=true build:nohdfs --define=no_hdfs_support=true build:nokafka --define=no_kafka_support=true build:noignite --define=no_ignite_support=true -build:nonccl --define=no_nccl_support=true build --define=use_fast_cpp_protos=true build --define=allow_oversize_protos=true