Richard Burton | dc0c6ed | 2020-04-08 16:39:05 +0100 | [diff] [blame] | 1 | # Copyright © 2020 Arm Ltd. All rights reserved. |
| 2 | # SPDX-License-Identifier: MIT |
Richard Burton | dc0c6ed | 2020-04-08 16:39:05 +0100 | [diff] [blame] | 3 | import pytest |
| 4 | import pyarmnn as ann |
| 5 | |
| 6 | |
| 7 | @pytest.fixture() |
| 8 | def get_supported_backends_setup(shared_data_folder): |
| 9 | options = ann.CreationOptions() |
| 10 | runtime = ann.IRuntime(options) |
| 11 | |
| 12 | get_device_spec = runtime.GetDeviceSpec() |
| 13 | supported_backends = get_device_spec.GetSupportedBackends() |
| 14 | |
| 15 | yield supported_backends |
| 16 | |
| 17 | |
| 18 | def test_ownership(): |
| 19 | options = ann.CreationOptions() |
| 20 | runtime = ann.IRuntime(options) |
| 21 | |
| 22 | device_spec = runtime.GetDeviceSpec() |
| 23 | |
| 24 | assert not device_spec.thisown |
| 25 | |
| 26 | |
| 27 | def test_to_string(): |
| 28 | options = ann.CreationOptions() |
| 29 | runtime = ann.IRuntime(options) |
| 30 | |
| 31 | device_spec = runtime.GetDeviceSpec() |
| 32 | expected_str = "IDeviceSpec {{ supportedBackends: [" \ |
| 33 | "{}" \ |
| 34 | "]}}".format(', '.join(map(lambda b: str(b), device_spec.GetSupportedBackends()))) |
| 35 | |
| 36 | assert expected_str == str(device_spec) |
| 37 | |
| 38 | |
| 39 | def test_get_supported_backends_cpu_ref(get_supported_backends_setup): |
| 40 | assert "CpuRef" in map(lambda b: str(b), get_supported_backends_setup) |
| 41 | |
| 42 | |
| 43 | @pytest.mark.aarch64 |
| 44 | class TestNoneCpuRefBackends: |
| 45 | |
| 46 | @pytest.mark.parametrize("backend", ["CpuAcc"]) |
| 47 | def test_get_supported_backends_cpu_acc(self, get_supported_backends_setup, backend): |
| 48 | assert backend in map(lambda b: str(b), get_supported_backends_setup) |