aboutsummaryrefslogtreecommitdiffhomepage
path: root/third_party/toolchains/gpus/cuda/build_defs.bzl
diff options
context:
space:
mode:
Diffstat (limited to 'third_party/toolchains/gpus/cuda/build_defs.bzl')
-rw-r--r--third_party/toolchains/gpus/cuda/build_defs.bzl8
1 files changed, 3 insertions, 5 deletions
diff --git a/third_party/toolchains/gpus/cuda/build_defs.bzl b/third_party/toolchains/gpus/cuda/build_defs.bzl
index badaf43019..9210bfe016 100644
--- a/third_party/toolchains/gpus/cuda/build_defs.bzl
+++ b/third_party/toolchains/gpus/cuda/build_defs.bzl
@@ -2,6 +2,7 @@
# execution service.
# DO NOT EDIT: automatically generated file
+# Macros for building CUDA code.
def if_cuda(if_true, if_false = []):
"""Shorthand for select()'ing on whether we're building with CUDA.
@@ -12,15 +13,13 @@ 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,
- "//conditions:default": if_false
+ "//conditions:default": if_false,
})
-
def cuda_default_copts():
"""Default options for all CUDA compilations."""
return if_cuda(["-x", "cuda", "-DGOOGLE_CUDA=1"] + ["--cuda-gpu-arch=sm_30"])
-
def cuda_is_configured():
"""Returns true if CUDA was enabled during the configure process."""
return True
@@ -32,6 +31,5 @@ def if_cuda_is_configured(x):
--config=cuda. Used to allow non-CUDA code to depend on CUDA libraries.
"""
if cuda_is_configured():
- return x
+ return x
return []
-