diff options
Diffstat (limited to 'lib/oeqa/runtime/miutils/tests')
4 files changed, 5 insertions, 103 deletions
diff --git a/lib/oeqa/runtime/miutils/tests/dldt_inference_engine_test.py b/lib/oeqa/runtime/miutils/tests/dldt_inference_engine_test.py deleted file mode 100644 index a44f9027..00000000 --- a/lib/oeqa/runtime/miutils/tests/dldt_inference_engine_test.py +++ /dev/null | |||
@@ -1,48 +0,0 @@ | |||
1 | import os | ||
2 | script_path = os.path.dirname(os.path.realpath(__file__)) | ||
3 | files_path = os.path.join(script_path, '../../files/') | ||
4 | |||
5 | class DldtInferenceEngineTest(object): | ||
6 | ie_input_files = {'ie_python_sample': 'classification_sample.py', | ||
7 | 'input': 'chicky_512.png', | ||
8 | 'input_download': 'https://raw.githubusercontent.com/opencv/opencv/master/samples/data/chicky_512.png', | ||
9 | 'model': 'squeezenet_v1.1.xml'} | ||
10 | |||
11 | def __init__(self, target, work_dir): | ||
12 | self.target = target | ||
13 | self.work_dir = work_dir | ||
14 | |||
15 | def setup(self): | ||
16 | self.target.run('mkdir -p %s' % self.work_dir) | ||
17 | self.target.copy_to(os.path.join(files_path, 'dldt-inference-engine', self.ie_input_files['ie_python_sample']), | ||
18 | self.work_dir) | ||
19 | |||
20 | def tear_down(self): | ||
21 | self.target.run('rm -rf %s' % self.work_dir) | ||
22 | |||
23 | def test_can_download_input_file(self, proxy_port): | ||
24 | return self.target.run('cd %s; wget %s -e https_proxy=%s' % | ||
25 | (self.work_dir, | ||
26 | self.ie_input_files['input_download'], | ||
27 | proxy_port)) | ||
28 | |||
29 | def test_dldt_ie_classification_with_device(self, device, ir_files_dir): | ||
30 | return self.target.run('classification_sample_async -d %s -i %s -m %s' % | ||
31 | (device, | ||
32 | os.path.join(self.work_dir, self.ie_input_files['input']), | ||
33 | os.path.join(ir_files_dir, self.ie_input_files['model']))) | ||
34 | |||
35 | def test_dldt_ie_classification_python_api_with_device(self, device, ir_files_dir, extension=''): | ||
36 | if extension: | ||
37 | return self.target.run('python3 %s -d %s -i %s -m %s -l %s' % | ||
38 | (os.path.join(self.work_dir, self.ie_input_files['ie_python_sample']), | ||
39 | device, | ||
40 | os.path.join(self.work_dir, self.ie_input_files['input']), | ||
41 | os.path.join(ir_files_dir, self.ie_input_files['model']), | ||
42 | extension)) | ||
43 | else: | ||
44 | return self.target.run('python3 %s -d %s -i %s -m %s' % | ||
45 | (os.path.join(self.work_dir, self.ie_input_files['ie_python_sample']), | ||
46 | device, | ||
47 | os.path.join(self.work_dir, self.ie_input_files['input']), | ||
48 | os.path.join(ir_files_dir, self.ie_input_files['model']))) | ||
diff --git a/lib/oeqa/runtime/miutils/tests/dldt_model_optimizer_test.py b/lib/oeqa/runtime/miutils/tests/dldt_model_optimizer_test.py deleted file mode 100644 index 7d3db15b..00000000 --- a/lib/oeqa/runtime/miutils/tests/dldt_model_optimizer_test.py +++ /dev/null | |||
@@ -1,23 +0,0 @@ | |||
1 | import os | ||
2 | |||
3 | class DldtModelOptimizerTest(object): | ||
4 | mo_input_files = {'model': 'squeezenet_v1.1.caffemodel', | ||
5 | 'prototxt': 'deploy.prototxt'} | ||
6 | mo_exe = 'mo.py' | ||
7 | |||
8 | def __init__(self, target, work_dir): | ||
9 | self.target = target | ||
10 | self.work_dir = work_dir | ||
11 | |||
12 | def setup(self): | ||
13 | self.target.run('mkdir -p %s' % self.work_dir) | ||
14 | |||
15 | def tear_down(self): | ||
16 | self.target.run('rm -rf %s' % self.work_dir) | ||
17 | |||
18 | def test_dldt_mo_can_create_ir(self, mo_exe_dir, mo_files_dir): | ||
19 | return self.target.run('python3 %s --input_model %s --input_proto %s --output_dir %s --data_type FP16' % | ||
20 | (os.path.join(mo_exe_dir, self.mo_exe), | ||
21 | os.path.join(mo_files_dir, self.mo_input_files['model']), | ||
22 | os.path.join(mo_files_dir, self.mo_input_files['prototxt']), | ||
23 | self.work_dir)) | ||
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 |
diff --git a/lib/oeqa/runtime/miutils/tests/squeezenet_model_download_test.py b/lib/oeqa/runtime/miutils/tests/squeezenet_model_download_test.py deleted file mode 100644 index a3e46a0a..00000000 --- a/lib/oeqa/runtime/miutils/tests/squeezenet_model_download_test.py +++ /dev/null | |||
@@ -1,25 +0,0 @@ | |||
1 | class SqueezenetModelDownloadTest(object): | ||
2 | download_files = {'squeezenet1.1.prototxt': 'https://raw.githubusercontent.com/DeepScale/SqueezeNet/a47b6f13d30985279789d08053d37013d67d131b/SqueezeNet_v1.1/deploy.prototxt', | ||
3 | 'squeezenet1.1.caffemodel': 'https://github.com/DeepScale/SqueezeNet/raw/a47b6f13d30985279789d08053d37013d67d131b/SqueezeNet_v1.1/squeezenet_v1.1.caffemodel'} | ||
4 | |||
5 | def __init__(self, target, work_dir): | ||
6 | self.target = target | ||
7 | self.work_dir = work_dir | ||
8 | |||
9 | def setup(self): | ||
10 | self.target.run('mkdir -p %s' % self.work_dir) | ||
11 | |||
12 | def tear_down(self): | ||
13 | self.target.run('rm -rf %s' % self.work_dir) | ||
14 | |||
15 | def test_can_download_squeezenet_model(self, proxy_port): | ||
16 | return self.target.run('cd %s; wget %s -e https_proxy=%s' % | ||
17 | (self.work_dir, | ||
18 | self.download_files['squeezenet1.1.caffemodel'], | ||
19 | proxy_port)) | ||
20 | |||
21 | def test_can_download_squeezenet_prototxt(self, proxy_port): | ||
22 | return self.target.run('cd %s; wget %s -e https_proxy=%s' % | ||
23 | (self.work_dir, | ||
24 | self.download_files['squeezenet1.1.prototxt'], | ||
25 | proxy_port)) | ||