commit b6390dd58a860e8e05947e9108da9e45098ddb0e
author Nina Drozd <nina.drozd@arm.com> Tue May 05 12:16:06 2020 +0100
committer Nina Drozd <nina.drozd@arm.com> Tue May 05 13:31:53 2020 +0000
tree 5ef57c335e4d5574c374a2ea6672300808bb9ecd
parent 81770986ac19d75c8b2ffebc87f97aaa81ad714f

MLECO-763: fixes to PyArmNN failing unit tests

Change-Id: Iac17cb6cd2136d5fd2f213561402f700329ac6c2
Signed-off-by: Nina Drozd <nina.drozd@arm.com>

python/pyarmnn/test/test_profiling_utilities.py

diff --git a/python/pyarmnn/test/test_profiling_utilities.py b/python/pyarmnn/test/test_profiling_utilities.py
index b7b91b5..9fb7fd8 100644
--- a/python/pyarmnn/test/test_profiling_utilities.py
+++ b/python/pyarmnn/test/test_profiling_utilities.py
@@ -24,7 +24,9 @@
 
 
 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 @@
                                                                   '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