| Kshitij Sisodia | f9efe0d | 2022-09-30 16:42:50 +0100 | [diff] [blame^] | 1 | # |
| 2 | # SPDX-FileCopyrightText: Copyright 2021-2022 Arm Limited and/or its affiliates <open-source-office@arm.com> |
| 3 | # SPDX-License-Identifier: Apache-2.0 |
| 4 | # |
| 5 | from ethosu_driver._generated.driver import SemanticVersion |
| 6 | from ethosu_driver._generated.driver import HardwareId |
| 7 | from ethosu_driver._generated.driver import HardwareConfiguration |
| 8 | from ethosu_driver._generated.driver import Capabilities |
| 9 | |
| 10 | |
| 11 | def check_semantic_version(ma, mi, pa, sv): |
| 12 | assert ma == sv.major |
| 13 | assert mi == sv.minor |
| 14 | assert pa == sv.patch |
| 15 | |
| 16 | |
| 17 | def test_semantic_version(): |
| 18 | sv = SemanticVersion(1, 2, 3) |
| 19 | assert '{ major=1, minor=2, patch=3 }' == sv.__str__() |
| 20 | check_semantic_version(1, 2, 3, sv) |
| 21 | |
| 22 | |
| 23 | def test_hardware_id(): |
| 24 | version = SemanticVersion(1, 2, 3) |
| 25 | product = SemanticVersion(4, 5, 6) |
| 26 | architecture = SemanticVersion(7, 8, 9) |
| 27 | hw_id = HardwareId(1, version, product, architecture) |
| 28 | |
| 29 | assert 1 == hw_id.versionStatus |
| 30 | |
| 31 | check_semantic_version(1, 2, 3, hw_id.version) |
| 32 | check_semantic_version(4, 5, 6, hw_id.product) |
| 33 | check_semantic_version(7, 8, 9, hw_id.architecture) |
| 34 | |
| 35 | assert '{versionStatus=1, version={ major=1, minor=2, patch=3 }, product={ major=4, minor=5, patch=6 }, ' \ |
| 36 | 'architecture={ major=7, minor=8, patch=9 }}' == hw_id.__str__() |
| 37 | |
| 38 | |
| 39 | def test_hw_configuration(): |
| 40 | hw_cfg = HardwareConfiguration(128, 1, True) |
| 41 | |
| 42 | assert 1 == hw_cfg.cmdStreamVersion |
| 43 | assert 128 == hw_cfg.macsPerClockCycle |
| 44 | assert hw_cfg.customDma |
| 45 | |
| 46 | assert "{macsPerClockCycle=128, cmdStreamVersion=1, customDma=True}" == hw_cfg.__str__() |
| 47 | |
| 48 | |
| 49 | def test_capabilities(): |
| 50 | version = SemanticVersion(100, 200, 300) |
| 51 | product = SemanticVersion(400, 500, 600) |
| 52 | architecture = SemanticVersion(700, 800, 900) |
| 53 | hw_id = HardwareId(1, version, product, architecture) |
| 54 | hw_cfg = HardwareConfiguration(256, 1000, False) |
| 55 | driver_v = SemanticVersion(10, 20, 30) |
| 56 | |
| 57 | cap = Capabilities(hw_id, hw_cfg, driver_v) |
| 58 | |
| 59 | check_semantic_version(10, 20, 30, cap.driver) |
| 60 | |
| 61 | check_semantic_version(100, 200, 300, cap.hwId.version) |
| 62 | check_semantic_version(400, 500, 600, cap.hwId.product) |
| 63 | check_semantic_version(700, 800, 900, cap.hwId.architecture) |
| 64 | |
| 65 | assert 1000 == cap.hwCfg.cmdStreamVersion |
| 66 | assert 256 == cap.hwCfg.macsPerClockCycle |
| 67 | assert not cap.hwCfg.customDma |
| 68 | |
| 69 | assert '{hwId={versionStatus=1, version={ major=100, minor=200, patch=300 }, ' \ |
| 70 | 'product={ major=400, minor=500, patch=600 }, ' \ |
| 71 | 'architecture={ major=700, minor=800, patch=900 }}, ' \ |
| 72 | 'hwCfg={macsPerClockCycle=256, cmdStreamVersion=1000, customDma=False}, ' \ |
| 73 | 'driver={ major=10, minor=20, patch=30 }}' == cap.__str__() |