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

Update MKL+Horovod test case

Browse Source
This commit is contained in:
Abolfazl Shahbazi 2020-07-16 17:04:21 -07:00
parent a7580dc7f2
commit d2a72ca245
1 changed files with 13 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
tensorflow/tools/dockerfiles/tests/import-onednn-horovod.sh
View File

@ -15,12 +15,16 @@
# limitations under the License. # limitations under the License.
# ============================================================================ # ============================================================================
{ # try python -c 'from tensorflow.python import _pywrap_util_port; print(_pywrap_util_port.IsMklEnabled()); import horovod.tensorflow as hvd'
echo `python -c 'from tensorflow.python import _pywrap_util_port; print(_pywrap_util_port.IsMklEnabled()); import horovod.tensorflow as hvd'` new_mkl_horovod_enabled=$?
echo "PASS: Horovod with MKL is enabled"
} || { # catch python -c 'from tensorflow.python import pywrap_tensorflow; print(pywrap_tensorflow.IsMklEnabled()); import horovod.tensorflow as hvd'
echo `python -c 'from tensorflow.python import pywrap_tensorflow; print(pywrap_tensorflow.IsMklEnabled()); import horovod.tensorflow as hvd'` old_mkl_horovod_enabled=$?
echo "PASS: Horovod with Old MKL is detected"
} || { # finally if [[ $new_mkl_horovod_enabled -eq 0 ]]; then
die "FAIL: Horovod with MKL is not enabled" echo "PASS: Horovod with MKL is enabled"
} elif [[ $old_mkl_horovod_enabled -eq 0]]; then
echo "PASS: Horovod with Old MKL is detected"
else
die "FAIL: Horovod with MKL is not enabled"
fi