blob: 49b977e93e8038d65e7f3689acb8334c93b8d9e2 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
|
# Macros for building CUDA code.
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({
"//third_party/gpus/cuda:using_nvcc": if_true,
"//third_party/gpus/cuda:using_gcudacc": if_true,
"//conditions:default": if_false
})
|
