Change-Id: Iac17cb6cd2136d5fd2f213561402f700329ac6c2
Signed-off-by: Nina Drozd <nina.drozd@arm.com>
def test_inference_exec(mock_profiler):
def test_inference_exec(mock_profiler):
- profiling_data_obj = ann.get_profiling_data(mock_profiler)
+ preferred_backends = [ann.BackendId('CpuRef'), ann.BackendId('CpuAcc'),
+ ann.BackendId('GpuAcc'), ann.BackendId('EthosNAcc')]
+ profiling_data_obj = ann.get_profiling_data(mock_profiler, preferred_backends)
assert (len(profiling_data_obj.inference_data) > 0)
assert (len(profiling_data_obj.per_workload_execution_data) > 0)
assert (len(profiling_data_obj.inference_data) > 0)
assert (len(profiling_data_obj.per_workload_execution_data) > 0)
'EthosNSomeMock4dWorkload_Execute_#8')
])
def test_profiler_workloads(mock_profiler, exec_times, unit, backend, workload):
'EthosNSomeMock4dWorkload_Execute_#8')
])
def test_profiler_workloads(mock_profiler, exec_times, unit, backend, workload):
- profiling_data_obj = ann.get_profiling_data(mock_profiler)
+ preferred_backends = [ann.BackendId('CpuRef'), ann.BackendId('CpuAcc'),
+ ann.BackendId('GpuAcc'), ann.BackendId('EthosNAcc')]
+ profiling_data_obj = ann.get_profiling_data(mock_profiler, preferred_backends)
work_load_exec = profiling_data_obj.per_workload_execution_data[workload]
assert work_load_exec["execution_time"] == exec_times
work_load_exec = profiling_data_obj.per_workload_execution_data[workload]
assert work_load_exec["execution_time"] == exec_times