From 203a7a8d523b7fa62e39612c97abc365d51114a1 Mon Sep 17 00:00:00 2001
From: Clayne Robison <clayne.b.robison@intel.com>
Date: Tue, 25 Jun 2019 12:11:20 -0700
Subject: [PATCH] Adding support for eigen unit tests in run_mkl.sh. This will
 allow us to triage which issues are unique to MKL.

---
 tensorflow/tools/ci_build/linux/cpu/run_mkl.sh | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

diff --git a/tensorflow/tools/ci_build/linux/cpu/run_mkl.sh b/tensorflow/tools/ci_build/linux/cpu/run_mkl.sh
index a8b73cbe0cf..f0a89cdb7e7 100755
--- a/tensorflow/tools/ci_build/linux/cpu/run_mkl.sh
+++ b/tensorflow/tools/ci_build/linux/cpu/run_mkl.sh
@@ -17,6 +17,7 @@
 
 set -e
 set -x
+MODE=${1:-"mkl"}
 
 N_JOBS=$(grep -c ^processor /proc/cpuinfo)
 
@@ -28,6 +29,11 @@ echo ""
 export TF_NEED_CUDA=0
 export PYTHON_BIN_PATH=`which python2`
 yes "" | $PYTHON_BIN_PATH configure.py
+if [[ "$MODE" == "eigen" ]]; then
+    CONFIG=""
+else
+    CONFIG="--config=mkl"
+fi
 
 # Run bazel test command. Double test timeouts to avoid flakes.
 # Setting KMP_BLOCKTIME to 0 lets OpenMP threads to sleep right after parallel execution
@@ -35,5 +41,5 @@ yes "" | $PYTHON_BIN_PATH configure.py
 # caused by executing multiple tests concurrently.
 bazel test --test_tag_filters=-no_oss,-oss_serial,-gpu,-benchmark-test --test_lang_filters=cc,py -k \
     --jobs=${N_JOBS} --test_timeout 300,450,1200,3600 --build_tests_only \
-    --config=mkl --test_env=KMP_BLOCKTIME=0 --config=opt --test_output=errors -- \
+    ${CONFIG} --test_env=KMP_BLOCKTIME=0 --config=opt --test_output=errors -- \
     //tensorflow/... -//tensorflow/compiler/... -//tensorflow/contrib/... -//tensorflow/lite/...