forked from pytorch/pytorch
-
Notifications
You must be signed in to change notification settings - Fork 0
/
BUILD
41 lines (36 loc) · 972 Bytes
/
BUILD
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
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"],
)