diff --git a/third_party/gpus/cuda/build_defs.bzl.tpl b/third_party/gpus/cuda/build_defs.bzl.tpl index e38e285423f..f53da0b341e 100644 --- a/third_party/gpus/cuda/build_defs.bzl.tpl +++ b/third_party/gpus/cuda/build_defs.bzl.tpl @@ -7,8 +7,7 @@ def if_cuda(if_true, if_false = []): """ return select({ - "@local_config_cuda//cuda:using_nvcc": if_true, - "@local_config_cuda//cuda:using_clang": if_true, + "@local_config_cuda//:is_cuda_enabled": if_true, "//conditions:default": if_false, }) diff --git a/third_party/nccl/build_defs.bzl.tpl b/third_party/nccl/build_defs.bzl.tpl index 7e698f589a7..ceb65e3b632 100644 --- a/third_party/nccl/build_defs.bzl.tpl +++ b/third_party/nccl/build_defs.bzl.tpl @@ -46,13 +46,13 @@ def _rdc_copts(): maxrregcount = "-maxrregcount=96" return cuda_default_copts() + select({ - "@local_config_cuda//cuda:using_nvcc": [ + "@local_config_cuda//:is_cuda_compiler_nvcc": [ "-nvcc_options", "relocatable-device-code=true", "-nvcc_options", "ptxas-options=" + maxrregcount, ], - "@local_config_cuda//cuda:using_clang": [ + "@local_config_cuda//:is_cuda_compiler_clang": [ "-fcuda-rdc", "-Xcuda-ptxas", maxrregcount,