From dde378ae034784879d0d1a4ad16f2765fdd4a5fd Mon Sep 17 00:00:00 2001 From: "A. Unique TensorFlower" Date: Tue, 20 Jun 2017 06:15:01 -0700 Subject: Creating configuration artifacts to use Bazel remote execution service PiperOrigin-RevId: 159552253 --- third_party/toolchains/gpus/cuda/build_defs.bzl | 37 +++++++++++++++++++++++++ 1 file changed, 37 insertions(+) create mode 100644 third_party/toolchains/gpus/cuda/build_defs.bzl (limited to 'third_party/toolchains/gpus/cuda/build_defs.bzl') diff --git a/third_party/toolchains/gpus/cuda/build_defs.bzl b/third_party/toolchains/gpus/cuda/build_defs.bzl new file mode 100644 index 0000000000..badaf43019 --- /dev/null +++ b/third_party/toolchains/gpus/cuda/build_defs.bzl @@ -0,0 +1,37 @@ +# Macros for building CUDA code used with Bazel remote +# execution service. +# DO NOT EDIT: automatically generated file + +def if_cuda(if_true, if_false = []): + """Shorthand for select()'ing on whether we're building with CUDA. + + Returns a select statement which evaluates to if_true if we're building + with CUDA enabled. Otherwise, the select statement evaluates to if_false. + + """ + return select({ + "@local_config_cuda//cuda:using_nvcc": if_true, + "@local_config_cuda//cuda:using_clang": if_true, + "//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 + +def if_cuda_is_configured(x): + """Tests if the CUDA was enabled during the configure process. + + Unlike if_cuda(), this does not require that we are building with + --config=cuda. Used to allow non-CUDA code to depend on CUDA libraries. + """ + if cuda_is_configured(): + return x + return [] + -- cgit v1.2.3