xref: /aosp_15_r20/external/pytorch/tools/config/BUILD (revision da0073e96a02ea20f0ac840b70461e3646d07c45)

load("@bazel_skylib//lib:selects.bzl", "selects")

config_setting(
    name = "cuda",
    define_values = {
        "cuda": "true",
    },
)

# Even when building with --config=cuda, host targets should be built with cuda disabled
# as these targets will run on CI machines that have no GPUs.
selects.config_setting_group(
    name = "cuda_enabled_and_capable",
    match_all = [
        ":cuda",
    ],
)

# Configures the system to build with cuda using clang.
config_setting(
    name = "cuda_clang",
    define_values = {
        "cuda_clang": "true",
    },
)

# Indicates that cuda code should be compiled with nvcc
# Mostly exists to support _analysis_ of tensorflow; more work is needed to actually make this
# setting work.
config_setting(
    name = "cuda_nvcc",
    define_values = {
        "cuda_nvcc": "true",
    },
)

config_setting(
    name = "thread_sanitizer",
    define_values = {"thread_sanitizer": "1"},
    visibility = ["//visibility:public"],
)