experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Merge pull request #30146 from Intel-tensorflow:eigen-on-public-ci

Browse Source 
PiperOrigin-RevId: 255056769
This commit is contained in:
TensorFlower Gardener 2019-06-26 00:13:38 -07:00
commit 80fa1b4cce
1 changed files with 7 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
tensorflow/tools/ci_build/linux/cpu/run_mkl.sh
View File

@ -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/...