aboutsummaryrefslogtreecommitdiffhomepage
path: root/third_party/mkl_dnn
diff options
context:
space:
mode:
Diffstat (limited to 'third_party/mkl_dnn')
-rw-r--r--third_party/mkl_dnn/build_defs.bzl13
1 files changed, 13 insertions, 0 deletions
diff --git a/third_party/mkl_dnn/build_defs.bzl b/third_party/mkl_dnn/build_defs.bzl
new file mode 100644
index 0000000000..108d82e683
--- /dev/null
+++ b/third_party/mkl_dnn/build_defs.bzl
@@ -0,0 +1,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
+ }) \ No newline at end of file