mirror of
git://git.yoctoproject.org/meta-intel.git
synced 2025-07-05 05:04:45 +02:00
runtime/miutils/tests/mkl_dnn_test: update test test_mkldnn_rnn_api
Update test_mkldnn_rnn_api to run batch test 'test_rnn_all'. Previously it was not ready and testcase was running 'test_rnn_small'. test_rnn_small is no longer available. Signed-off-by: sangeeta jain <sangeeta.jain@intel.com> Signed-off-by: Anuj Mittal <anuj.mittal@intel.com>
This commit is contained in:
parent
69e587c2cb
commit
37787d86a8
|
@ -49,9 +49,7 @@ class MkldnnTest(object):
|
|||
return self._run_mkldnn_benchdnn_test('./benchdnn --reorder --batch=inputs/reorder/test_reorder_bfloat16')
|
||||
|
||||
def test_mkldnn_rnn_api(self):
|
||||
# test_rnn_inference was not yet ready and was expected to fail
|
||||
# while waiting it to be ready, use test_rnn_small for now
|
||||
return self._run_mkldnn_benchdnn_test('./benchdnn --rnn --batch=inputs/rnn/test_rnn_small')
|
||||
return self._run_mkldnn_benchdnn_test('./benchdnn --rnn --batch=inputs/rnn/test_rnn_all')
|
||||
|
||||
def test_mkldnn_shuffle_api(self):
|
||||
return self._run_mkldnn_benchdnn_test('./benchdnn --shuffle --batch=inputs/shuffle/test_shuffle_bfloat16')
|
Loading…
Reference in New Issue
Block a user