diff options
Diffstat (limited to 'lib/oeqa/runtime/miutils/tests/mkl_dnn_test.py')
-rw-r--r-- | lib/oeqa/runtime/miutils/tests/mkl_dnn_test.py | 12 |
1 files changed, 5 insertions, 7 deletions
diff --git a/lib/oeqa/runtime/miutils/tests/mkl_dnn_test.py b/lib/oeqa/runtime/miutils/tests/mkl_dnn_test.py index 13afd1a4..869a4cbe 100644 --- a/lib/oeqa/runtime/miutils/tests/mkl_dnn_test.py +++ b/lib/oeqa/runtime/miutils/tests/mkl_dnn_test.py | |||
@@ -9,13 +9,13 @@ class MkldnnTest(object): | |||
9 | self.target.run('rm /tmp/%s' % self.mkldnn_target_test_filename) | 9 | self.target.run('rm /tmp/%s' % self.mkldnn_target_test_filename) |
10 | 10 | ||
11 | def test_mkldnn_can_compile_and_execute(self): | 11 | def test_mkldnn_can_compile_and_execute(self): |
12 | mkldnn_src_dir = '/usr/src/debug/mkl-dnn/' | 12 | mkldnn_src_dir = '/usr/src/debug/onednn/' |
13 | mkldnn_src_test_filename = 'api.c' | 13 | mkldnn_src_test_filename = 'api.c' |
14 | mkldnn_src_test_file = '' | 14 | mkldnn_src_test_file = '' |
15 | 15 | ||
16 | (status, output) = self.target.run('cd %s; find -name %s' % (mkldnn_src_dir, mkldnn_src_test_filename)) | 16 | (__, output) = self.target.run('cd %s; find -name %s' % (mkldnn_src_dir, mkldnn_src_test_filename)) |
17 | if status: | 17 | if 'No such file or directory' in output: |
18 | return status, output | 18 | return -1, output |
19 | 19 | ||
20 | mkldnn_src_test_file = os.path.join(mkldnn_src_dir, output) | 20 | mkldnn_src_test_file = os.path.join(mkldnn_src_dir, output) |
21 | (status, output) = self.target.run('gcc %s -o /tmp/%s -ldnnl' % (mkldnn_src_test_file, self.mkldnn_target_test_filename)) | 21 | (status, output) = self.target.run('gcc %s -o /tmp/%s -ldnnl' % (mkldnn_src_test_file, self.mkldnn_target_test_filename)) |
@@ -49,9 +49,7 @@ class MkldnnTest(object): | |||
49 | return self._run_mkldnn_benchdnn_test('./benchdnn --reorder --batch=inputs/reorder/test_reorder_bfloat16') | 49 | return self._run_mkldnn_benchdnn_test('./benchdnn --reorder --batch=inputs/reorder/test_reorder_bfloat16') |
50 | 50 | ||
51 | def test_mkldnn_rnn_api(self): | 51 | def test_mkldnn_rnn_api(self): |
52 | # test_rnn_inference was not yet ready and was expected to fail | 52 | return self._run_mkldnn_benchdnn_test('./benchdnn --rnn --batch=inputs/rnn/test_rnn_all') |
53 | # while waiting it to be ready, use test_rnn_small for now | ||
54 | return self._run_mkldnn_benchdnn_test('./benchdnn --rnn --batch=inputs/rnn/test_rnn_small') | ||
55 | 53 | ||
56 | def test_mkldnn_shuffle_api(self): | 54 | def test_mkldnn_shuffle_api(self): |
57 | return self._run_mkldnn_benchdnn_test('./benchdnn --shuffle --batch=inputs/shuffle/test_shuffle_bfloat16') \ No newline at end of file | 55 | return self._run_mkldnn_benchdnn_test('./benchdnn --shuffle --batch=inputs/shuffle/test_shuffle_bfloat16') \ No newline at end of file |