Skip to content

Commit 7850cf3

Browse files
test in performancee scope demos with dynamic shape
1 parent f5386a7 commit 7850cf3

File tree

2 files changed

+20
-1
lines changed

2 files changed

+20
-1
lines changed

demos/tests/cases.py

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -41,6 +41,7 @@ def __init__(self, name, implementation, model_keys=None, device_keys=None, test
4141

4242
self._exec_name = self.subdirectory.replace('/', '_')
4343
self.parser = None
44+
self.supported_devices = None
4445

4546
Demo.IMPLEMENTATION_TYPES.add(implementation)
4647

@@ -50,6 +51,8 @@ def models_lst_path(self, source_dir):
5051
def device_args(self, device_list):
5152
if len(self.device_keys) == 0:
5253
return {'CPU': []}
54+
if self.supported_devices:
55+
device_list = self.supported_devices
5356
return {device: [arg for key in self.device_keys for arg in [key, device]] for device in device_list}
5457

5558
def get_models(self, case):
@@ -99,6 +102,10 @@ def only_models(self, models):
99102
continue
100103
return self
101104

105+
def only_devices(self, devices):
106+
self.supported_devices = devices
107+
return self
108+
102109
def set_precisions(self, precisions, model_info):
103110
for case in self.test_cases[:]:
104111
updated_options = {p: {} for p in precisions}

demos/tests/performance_cases.py

Lines changed: 13 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -69,5 +69,17 @@ def write_to_csv(self, result, test_case, device):
6969
.add_test_cases(single_option_cases('-nireq', '3', '5'),
7070
single_option_cases('-nstreams', '3', '4'),
7171
single_option_cases('-nthreads', str(THREADS_NUM), str(THREADS_NUM - 2)))
72-
.add_parser(PerformanceParser)
72+
.add_parser(PerformanceParser),
73+
74+
deepcopy(BASE['bert_named_entity_recognition_demo/python'])
75+
.update_option({'--dynamic_shape': None})
76+
.only_devices(['CPU']),
77+
78+
deepcopy(BASE['gpt2_text_prediction_demo/python'])
79+
.update_option({'--dynamic_shape': None})
80+
.only_devices(['CPU']),
81+
82+
deepcopy(BASE['speech_recognition_wav2vec_demo/python'])
83+
.update_option({'--dynamic_shape': None})
84+
.only_devices(['CPU'])
7385
]

0 commit comments

Comments
 (0)