diff options
| author |  A. Unique TensorFlower <gardener@tensorflow.org> | 2018-08-14 14:10:20 -0700 |
|---|---|---|
| committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-08-14 14:15:42 -0700 |
| commit | 7253bd04a4bdeaa22aeafc61af0f1f611dfda77c (patch) | |
| tree | fd4a52af3ee3dfaa35294fece401bd9ab6c26978 /third_party/mkl_dnn | |
| parent | ab53f852b70250286cefb3192445775d7a5491e0 (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 'third_party/mkl_dnn')
| -rw-r--r-- | third_party/mkl_dnn/BUILD | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/third_party/mkl_dnn/BUILD b/third_party/mkl_dnn/BUILD index d075809ee9..3e567fa9fc 100644 --- a/third_party/mkl_dnn/BUILD +++ b/third_party/mkl_dnn/BUILD @@ -4,8 +4,9 @@ exports_files(["LICENSE"]) config_setting( name = "using_mkl_dnn_only", - values = { - "define": "using_mkl_dnn_only=true", + define_values = { + "using_mkl": "true", + "using_mkl_dnn_only": "true", }, visibility = ["//visibility:public"], ) |