aboutsummaryrefslogtreecommitdiff
path: root/ethosu/vela/vela.py
diff options
context:
space:
mode:
authorPatrik Gustavsson <patrik.gustavsson@arm.com>2020-11-18 17:05:50 +0100
committerpatrik.gustavsson <patrik.gustavsson@arm.com>2020-11-19 12:12:07 +0000
commitc8a22f198c4fb23735962fee00c062b0325ca6a6 (patch)
treee9a99cb102d43471a6454267170aa7e5ba3b541f /ethosu/vela/vela.py
parentbdf09f9eca75fab14298a94a5402efc15f698e68 (diff)
downloadethos-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.py4
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.",