commit ea8c5374989ff9622d20b663e7707378636e4405
author William Isaksson <william.isaksson@arm.com> Mon Jul 03 20:31:42 2023 +0000
committer tim.hall <tim.hall@arm.com> Fri Aug 04 08:47:27 2023 +0000
tree be492dfd4de4761413a3c0fe25cab009860279ad
parent e4d576772c0e0a1c7ab1e6bdcf5be8964994fbbc

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>

ethosu/vela/tflite_writer.py

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 @@
 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 @@
         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 @@
             ]
         )
 
-        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]