diff options
author | Nina Drozd <nina.drozd@arm.com> | 2020-05-05 12:16:06 +0100 |
---|---|---|
committer | Nina Drozd <nina.drozd@arm.com> | 2020-05-05 13:31:53 +0000 |
commit | b6390dd58a860e8e05947e9108da9e45098ddb0e (patch) | |
tree | 5ef57c335e4d5574c374a2ea6672300808bb9ecd /python/pyarmnn | |
parent | 81770986ac19d75c8b2ffebc87f97aaa81ad714f (diff) | |
download | armnn-b6390dd58a860e8e05947e9108da9e45098ddb0e.tar.gz |
MLECO-763: fixes to PyArmNN failing unit tests
Change-Id: Iac17cb6cd2136d5fd2f213561402f700329ac6c2
Signed-off-by: Nina Drozd <nina.drozd@arm.com>
Diffstat (limited to 'python/pyarmnn')
-rw-r--r-- | python/pyarmnn/test/test_profiling_utilities.py | 8 |
1 files changed, 6 insertions, 2 deletions
diff --git a/python/pyarmnn/test/test_profiling_utilities.py b/python/pyarmnn/test/test_profiling_utilities.py index b7b91b5613..9fb7fd8b3d 100644 --- a/python/pyarmnn/test/test_profiling_utilities.py +++ b/python/pyarmnn/test/test_profiling_utilities.py @@ -24,7 +24,9 @@ def mock_profiler(shared_data_folder): 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) @@ -60,7 +62,9 @@ def test_inference_exec(mock_profiler): '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 |