1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
|
# Copyright © 2020 Arm Ltd. All rights reserved.
# SPDX-License-Identifier: MIT
import os
import pytest
import pyarmnn as ann
class MockIProfiler:
def __init__(self, json_string):
self._profile_json = json_string
def as_json(self):
return self._profile_json
@pytest.fixture()
def mock_profiler(shared_data_folder):
path_to_file = os.path.join(shared_data_folder, 'mock_profile_out.json')
with open(path_to_file, 'r') as file:
profiler_output = file.read()
return MockIProfiler(profiler_output)
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)
assert (len(profiling_data_obj.inference_data) > 0)
assert (len(profiling_data_obj.per_workload_execution_data) > 0)
# Check each total execution time
assert (profiling_data_obj.inference_data["execution_time"] == [1.1, 2.2, 3.3, 4.4, 5.5, 6.6])
assert (profiling_data_obj.inference_data["time_unit"] == "us")
@pytest.mark.parametrize("exec_times, unit, backend, workload", [([2, 2,
2, 2,
2, 2],
'us',
'CpuRef',
'RefSomeMock1dWorkload_Execute_#5'),
([2, 2,
2, 2,
2, 2],
'us',
'CpuAcc',
'NeonSomeMock2Workload_Execute_#6'),
([2, 2,
2, 2,
2, 2],
'us',
'GpuAcc',
'ClSomeMock3dWorkload_Execute_#7'),
([2, 2,
2, 2,
2, 2],
'us',
'EthosNAcc',
'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)
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
|
