diff options
author | Nina Drozd <nina.drozd@arm.com> | 2020-05-15 17:54:19 +0100 |
---|---|---|
committer | Nina Drozd <nina.drozd@arm.com> | 2020-05-15 17:11:36 +0000 |
commit | 63b234c6f5dd1340d8a7c5f64d7e1bbbe3e2d8df (patch) | |
tree | 9ac0bb59ac2d3170632131ab7ffc407ab6c24632 /python/pyarmnn/test | |
parent | 56b465d8d8e2cd7d9534b2e24fb7afad595bc992 (diff) | |
download | armnn-63b234c6f5dd1340d8a7c5f64d7e1bbbe3e2d8df.tar.gz |
Remove parsing of EthosN profiling data, too specific for pyarmnn
Signed-off-by: Nina Drozd <nina.drozd@arm.com>
Change-Id: I04d82dd3a9759c660fdb908433becda915a3f5b5
Diffstat (limited to 'python/pyarmnn/test')
-rw-r--r-- | python/pyarmnn/test/test_profiling_utilities.py | 10 |
1 files changed, 3 insertions, 7 deletions
diff --git a/python/pyarmnn/test/test_profiling_utilities.py b/python/pyarmnn/test/test_profiling_utilities.py index 9fb7fd8b3d..642d293932 100644 --- a/python/pyarmnn/test/test_profiling_utilities.py +++ b/python/pyarmnn/test/test_profiling_utilities.py @@ -24,9 +24,7 @@ def mock_profiler(shared_data_folder): def test_inference_exec(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) + profiling_data_obj = ann.get_profiling_data(mock_profiler) assert (len(profiling_data_obj.inference_data) > 0) assert (len(profiling_data_obj.per_workload_execution_data) > 0) @@ -62,11 +60,9 @@ def test_inference_exec(mock_profiler): 'EthosNSomeMock4dWorkload_Execute_#8') ]) def test_profiler_workloads(mock_profiler, exec_times, unit, backend, workload): - 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) + profiling_data_obj = ann.get_profiling_data(mock_profiler) work_load_exec = profiling_data_obj.per_workload_execution_data[workload] assert work_load_exec["execution_time"] == exec_times assert work_load_exec["time_unit"] == unit - assert work_load_exec["backend"] == backend + assert work_load_exec["backend"] == backend
\ No newline at end of file |