diff options
| author | William Isaksson <william.isaksson@arm.com> | 2023-07-03 20:31:42 +0000 |
|---|---|---|
| committer | tim.hall <tim.hall@arm.com> | 2023-08-04 08:47:27 +0000 |
| commit | ea8c5374989ff9622d20b663e7707378636e4405 (patch) | |
| tree | be492dfd4de4761413a3c0fe25cab009860279ad | |
| parent | e4d576772c0e0a1c7ab1e6bdcf5be8964994fbbc (diff) | |
| download | ethos-u-vela-ea8c5374989ff9622d20b663e7707378636e4405.tar.gz | |
MLBEDSW-7681: Add Vela version to output file
Adds Vela version to description and metadata
Change-Id: I75fccd1a05a396612a249b8ec1662d8cae940ee6
Signed-off-by: William Isaksson <william.isaksson@arm.com>
| -rw-r--r-- | ethosu/vela/tflite_writer.py | 9 |
1 files changed, 7 insertions, 2 deletions
diff --git a/ethosu/vela/tflite_writer.py b/ethosu/vela/tflite_writer.py index ae529a4..5aa38a7 100644 --- a/ethosu/vela/tflite_writer.py +++ b/ethosu/vela/tflite_writer.py @@ -22,6 +22,7 @@ import numpy as np from flatbuffers import encode from flatbuffers.builder import UOffsetTFlags +from ._version import __version__ from .errors import VelaError from .nn_graph import PassPlacement from .operation import Op @@ -427,7 +428,10 @@ class TFLiteSerialiser: builder = self.builder builder.assertNotNested() builder.nested = True - data = bytes(buf) + if isinstance(buf, str): + data = bytes(buf, "utf-8") + else: + data = bytes(buf) length_bytes = UOffsetTFlags.py_type(len(data)) builder.vectorNumElems = length_bytes builder.Prep(16, length_bytes) # Reserve aligned storage @@ -464,7 +468,8 @@ class TFLiteSerialiser: ] ) - description = builder.CreateString("Vela Optimised") + description = builder.CreateString(f"Vela {__version__} Optimised") + self.nng.metadata.append(("vela_version", __version__)) subgraph_offset = self.write_offset_vector( [self.serialise_subgraph(sg, builder.CreateString(sg.name)) for sg in self.subgraphs_to_write] |