From b7c9ea172f2faedc9257dd406bb1e336c12284fa Mon Sep 17 00:00:00 2001 From: Mahmoud Abuzaina Date: Tue, 5 Feb 2019 17:57:44 -0800 Subject: [PATCH] Added #ifdef INTEL_MKL to the Quantized Concat test --- tensorflow/core/kernels/mkl_quantized_concat_op_test.cc | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/tensorflow/core/kernels/mkl_quantized_concat_op_test.cc b/tensorflow/core/kernels/mkl_quantized_concat_op_test.cc index dcd1de95e02..fc68480bbe8 100644 --- a/tensorflow/core/kernels/mkl_quantized_concat_op_test.cc +++ b/tensorflow/core/kernels/mkl_quantized_concat_op_test.cc @@ -13,6 +13,8 @@ See the License for the specific language governing permissions and limitations under the License. ==============================================================================*/ +#if defined(INTEL_MKL) && defined(ENABLE_MKL) + #define EIGEN_USE_THREADS #include @@ -228,3 +230,5 @@ void QuantizedConcatTest::TestSecondDim8Bit(float first_min, float first_max, } } // namespace tensorflow + +#endif // INTEL_MKL && ENABLE_MKL