blob: 108d82e6831376055020562477d9bb6447e822f4 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
|
def if_mkl_open_source_only(if_true, if_false = []):
"""Shorthand for select()'ing on whether we're building with
MKL-DNN open source lib only, without depending on MKL binary form.
Returns a select statement which evaluates to if_true if we're building
with MKL-DNN open source lib only. Otherwise,
the select statement evaluates to if_false.
"""
return select({
str(Label("//third_party/mkl_dnn:using_mkl_dnn_only")): if_true,
"//conditions:default": if_false
})
|