diff options
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 5df21f5..9f5c78c 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.", |