aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/core/platform
diff options
context:
space:
mode:
authorGravatar A. Unique TensorFlower <gardener@tensorflow.org>2018-08-14 14:10:20 -0700
committerGravatar TensorFlower Gardener <gardener@tensorflow.org>2018-08-14 14:15:42 -0700
commit7253bd04a4bdeaa22aeafc61af0f1f611dfda77c (patch)
treefd4a52af3ee3dfaa35294fece401bd9ab6c26978 /tensorflow/core/platform
parentab53f852b70250286cefb3192445775d7a5491e0 (diff)
Apply MKL-{ML,DNN}-only config settings to dependencies, not just code.
Previously, specifying --define=using_mkl_dnn_only=true would cause MKL-ML-dependent code to be #ifdef'd out, but dependencies on MKL-ML itself would still be present. This change makes all library dependencies on MKL properly select MKL-ML, MKL-DNN, or both, depending on the selected configuration. PiperOrigin-RevId: 208710102
Diffstat (limited to 'tensorflow/core/platform')
-rw-r--r--tensorflow/core/platform/default/build_config.bzl6
1 files changed, 3 insertions, 3 deletions
diff --git a/tensorflow/core/platform/default/build_config.bzl b/tensorflow/core/platform/default/build_config.bzl
index fb4ee1c33c..6383180e94 100644
--- a/tensorflow/core/platform/default/build_config.bzl
+++ b/tensorflow/core/platform/default/build_config.bzl
@@ -8,7 +8,7 @@ load("//tensorflow/core:platform/default/build_config_root.bzl", "if_static")
load("@local_config_cuda//cuda:build_defs.bzl", "if_cuda")
load(
"//third_party/mkl:build_defs.bzl",
- "if_mkl",
+ "if_mkl_ml",
)
# Appends a suffix to a list of deps.
@@ -700,8 +700,8 @@ def tf_additional_binary_deps():
# core).
"//tensorflow/core/kernels:lookup_util",
"//tensorflow/core/util/tensor_bundle",
- ] + if_mkl(
+ ] + if_mkl_ml(
[
- "//third_party/mkl:intel_binary_blob",
+ "//third_party/intel_mkl_ml",
],
)