aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/core/util
diff options
context:
space:
mode:
authorGravatar A. Unique TensorFlower <gardener@tensorflow.org>2018-09-25 22:57:54 -0700
committerGravatar TensorFlower Gardener <gardener@tensorflow.org>2018-09-25 23:02:36 -0700
commitc63d21b0bfc534b6377b332e9d2ba2abbdb7e0eb (patch)
tree6fb44dc9eb26de3ff6e25f57e4799ac2da066d95 /tensorflow/core/util
parentf2b17b22e12bd743b66945070f338f70b5fa3332 (diff)
Adds a build flag to enable MKL (mkl_enabled=true).
PiperOrigin-RevId: 214557082
Diffstat (limited to 'tensorflow/core/util')
-rw-r--r--tensorflow/core/util/port.cc4
1 files changed, 2 insertions, 2 deletions
diff --git a/tensorflow/core/util/port.cc b/tensorflow/core/util/port.cc
index c081ceae57..e01058dff6 100644
--- a/tensorflow/core/util/port.cc
+++ b/tensorflow/core/util/port.cc
@@ -38,10 +38,10 @@ bool CudaSupportsHalfMatMulAndConv() {
}
bool IsMklEnabled() {
-#ifdef INTEL_MKL
+#if defined(INTEL_MKL) && defined(ENABLE_MKL)
return true;
#else
return false;
-#endif
+#endif // INTEL_MKL && ENABLE_MKL
}
} // end namespace tensorflow