Richard Burton | dc0c6ed | 2020-04-08 16:39:05 +0100 | [diff] [blame] | 1 | # Copyright © 2020 Arm Ltd. All rights reserved. |
Pavel Macenauer | 59e057f | 2020-04-15 14:17:26 +0000 | [diff] [blame] | 2 | # Copyright 2020 NXP |
Richard Burton | dc0c6ed | 2020-04-08 16:39:05 +0100 | [diff] [blame] | 3 | # SPDX-License-Identifier: MIT |
| 4 | import os |
| 5 | |
Cathal Corbett | 5aa9fd7 | 2022-02-25 15:33:28 +0000 | [diff] [blame^] | 6 | version_info = (29, 0, 0) |
Richard Burton | dc0c6ed | 2020-04-08 16:39:05 +0100 | [diff] [blame] | 7 | |
| 8 | __dev_version_env = os.getenv("PYARMNN_DEV_VER", "") |
| 9 | |
| 10 | if __dev_version_env: |
| 11 | __dev_version = "dev0" |
| 12 | try: |
| 13 | __dev_version = "dev{}".format(int(__dev_version_env)) |
| 14 | except ValueError: |
| 15 | __dev_version = str(__dev_version_env) |
| 16 | |
| 17 | version_info = (*version_info, __dev_version) |
| 18 | |
| 19 | __version__ = '.'.join(str(c) for c in version_info) |
Pavel Macenauer | 59e057f | 2020-04-15 14:17:26 +0000 | [diff] [blame] | 20 | __arm_ml_version__ = '{}.{}.{}'.format(version_info[0], version_info[1], version_info[2]) |
Richard Burton | dc0c6ed | 2020-04-08 16:39:05 +0100 | [diff] [blame] | 21 | |
| 22 | |
Pavel Macenauer | 59e057f | 2020-04-15 14:17:26 +0000 | [diff] [blame] | 23 | def check_armnn_version(installed_armnn_version: str, expected_armnn_version: str = __arm_ml_version__): |
| 24 | """Compares expected Arm NN version and Arm NN version used to build the package. |
| 25 | |
| 26 | Args: |
Cathal Corbett | 5aa9fd7 | 2022-02-25 15:33:28 +0000 | [diff] [blame^] | 27 | installed_armnn_version (str): Arm NN version used to generate the package (e.g. 29.0.0) |
Pavel Macenauer | 59e057f | 2020-04-15 14:17:26 +0000 | [diff] [blame] | 28 | expected_armnn_version (str): Expected Arm NN version |
| 29 | |
| 30 | Returns: |
| 31 | None |
| 32 | """ |
| 33 | expected = expected_armnn_version.split('.', 2) |
| 34 | installed = installed_armnn_version.split('.', 2) |
| 35 | |
| 36 | # only compare major and minor versions, not patch |
| 37 | assert (expected[0] == installed[0]) and (expected[1] == installed[1]), \ |
Richard Burton | dc0c6ed | 2020-04-08 16:39:05 +0100 | [diff] [blame] | 38 | "Expected ArmNN version is {} but installed ArmNN version is {}".format(expected_armnn_version, installed_armnn_version) |