• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1# Copyright © 2020 Arm Ltd. All rights reserved.
2# Copyright 2020 NXP
3# SPDX-License-Identifier: MIT
4import os
5
6version_info = (32, 1, 0)
7
8__dev_version_env = os.getenv("PYARMNN_DEV_VER", "")
9
10if __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)
20__arm_ml_version__ = '{}.{}.{}'.format(version_info[0], version_info[1], version_info[2])
21
22
23def 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:
27        installed_armnn_version (str): Arm NN version used to generate the package (e.g. 32.0.0)
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]), \
38        "Expected ArmNN version is {} but installed ArmNN version is {}".format(expected_armnn_version, installed_armnn_version)
39