From ea8c5374989ff9622d20b663e7707378636e4405 Mon Sep 17 00:00:00 2001 From: William Isaksson Date: Mon, 3 Jul 2023 20:31:42 +0000 Subject: MLBEDSW-7681: Add Vela version to output file Adds Vela version to description and metadata Change-Id: I75fccd1a05a396612a249b8ec1662d8cae940ee6 Signed-off-by: William Isaksson --- ethosu/vela/tflite_writer.py | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) diff --git a/ethosu/vela/tflite_writer.py b/ethosu/vela/tflite_writer.py index ae529a43..5aa38a75 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] -- cgit v1.2.1