| 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 |
| 3 | import os |
| 4 | |
| 5 | version_info = (20, 2, 0) |
| 6 | |
| 7 | __dev_version_env = os.getenv("PYARMNN_DEV_VER", "") |
| 8 | |
| 9 | if __dev_version_env: |
| 10 | __dev_version = "dev0" |
| 11 | try: |
| 12 | __dev_version = "dev{}".format(int(__dev_version_env)) |
| 13 | except ValueError: |
| 14 | __dev_version = str(__dev_version_env) |
| 15 | |
| 16 | version_info = (*version_info, __dev_version) |
| 17 | |
| 18 | __version__ = '.'.join(str(c) for c in version_info) |
| 19 | __arm_ml_version__ = '2{:03d}{:02d}{:02d}'.format(version_info[0], version_info[1], version_info[2]) |
| 20 | |
| 21 | |
| 22 | def check_armnn_version(installed_armnn_version, expected_armnn_version=__arm_ml_version__): |
| 23 | expected_armnn_version = expected_armnn_version[:-2] # cut off minor patch version |
| 24 | installed_armnn_version = installed_armnn_version[:-2] # cut off minor patch version |
| 25 | assert expected_armnn_version == installed_armnn_version, \ |
| 26 | "Expected ArmNN version is {} but installed ArmNN version is {}".format(expected_armnn_version, installed_armnn_version) |