diff options
author | Matthew Bentham <Matthew.Bentham@arm.com> | 2019-12-02 12:59:43 +0000 |
---|---|---|
committer | Matthew Bentham <Matthew.Bentham@arm.com> | 2019-12-02 16:23:45 +0000 |
commit | 245d64c60d0ea30f5080ff53225b5169927e24d6 (patch) | |
tree | d623e46d7d5ddb34ef3bb84c45df3ada9209ce82 /python/pyarmnn/src/pyarmnn/_version.py | |
parent | 88d5f9f1615fa956464b8932b574d85c37cec937 (diff) | |
download | armnn-245d64c60d0ea30f5080ff53225b5169927e24d6.tar.gz |
Work in progress of python bindings for Arm NNexperimental/pyarmnn
Not built or tested in any way
Signed-off-by: Matthew Bentham <Matthew.Bentham@arm.com>
Change-Id: Ie7f92b529aa5087130f0c5cc8c17db1581373236
Diffstat (limited to 'python/pyarmnn/src/pyarmnn/_version.py')
-rw-r--r-- | python/pyarmnn/src/pyarmnn/_version.py | 26 |
1 files changed, 26 insertions, 0 deletions
diff --git a/python/pyarmnn/src/pyarmnn/_version.py b/python/pyarmnn/src/pyarmnn/_version.py new file mode 100644 index 0000000000..2bcb888819 --- /dev/null +++ b/python/pyarmnn/src/pyarmnn/_version.py @@ -0,0 +1,26 @@ +# Copyright © 2019 Arm Ltd. All rights reserved. +# SPDX-License-Identifier: MIT +import os + +version_info = (19, 11, 0) + +__dev_version_env = os.getenv("PYARMNN_DEV_VER", "") + +if __dev_version_env: + __dev_version = "dev0" + try: + __dev_version = "dev{}".format(int(__dev_version_env)) + except ValueError: + __dev_version = str(__dev_version_env) + + version_info = (*version_info, __dev_version) + +__version__ = '.'.join(str(c) for c in version_info) +__arm_ml_version__ = '2{:03d}{:02d}{:02d}'.format(version_info[0], version_info[1], version_info[2]) + + +def check_armnn_version(installed_armnn_version, expected_armnn_version=__arm_ml_version__): + expected_armnn_version = expected_armnn_version[:-2] # cut off minor patch version + installed_armnn_version = installed_armnn_version[:-2] # cut off minor patch version + assert expected_armnn_version == installed_armnn_version, \ + "Expected ArmNN version is {} but installed ArmNN version is {}".format(expected_armnn_version, installed_armnn_version) |