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

Update 'import-onednn.sh' test case per code review

Browse Source
This commit is contained in:
Abolfazl Shahbazi 2020-07-08 18:01:34 -07:00
parent 6879366d63
commit 6e98b7f78d
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.sh
View File

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