diff options
author | Patrik Gustavsson <patrik.gustavsson@arm.com> | 2020-11-18 17:05:50 +0100 |
---|---|---|
committer | patrik.gustavsson <patrik.gustavsson@arm.com> | 2020-11-19 12:12:07 +0000 |
commit | c8a22f198c4fb23735962fee00c062b0325ca6a6 (patch) | |
tree | e9a99cb102d43471a6454267170aa7e5ba3b541f /ethosu/vela/vela.py | |
parent | bdf09f9eca75fab14298a94a5402efc15f698e68 (diff) | |
download | ethos-u-vela-c8a22f198c4fb23735962fee00c062b0325ca6a6.tar.gz |
MLBEDSW-3251 Add version to external API
Added version to the external API
-Added CLI-option --api_version
-Added API function to get the API version
Signed-off-by: Patrik Gustavsson <patrik.gustavsson@arm.com>
Change-Id: I0143b50adf884a2b05145912a1c7bef8cecc5f02
Diffstat (limited to 'ethosu/vela/vela.py')
-rw-r--r-- | ethosu/vela/vela.py | 4 |
1 files changed, 4 insertions, 0 deletions
diff --git a/ethosu/vela/vela.py b/ethosu/vela/vela.py index 5df21f56..9f5c78c3 100644 --- a/ethosu/vela/vela.py +++ b/ethosu/vela/vela.py @@ -31,6 +31,7 @@ from . import scheduler from . import stats_writer from . import tflite_writer from ._version import __version__ +from .api import api_version from .debug_database import DebugDatabase from .errors import InputFileError from .nn_graph import PassPlacement @@ -190,6 +191,9 @@ def main(args=None): parser = argparse.ArgumentParser(prog="vela", description="Neural network model compiler for Ethos-U55") parser.add_argument("--version", action="version", version=__version__) parser.add_argument( + "--api-version", action="version", version=api_version, help="Displays the version of the external API." + ) + parser.add_argument( "--supported-ops-report", action="store_true", help="Generate the SUPPORTED_OPS.md file in the current working directory and exits.", |