commit 62c1e95dc2f438e027f94e1dd2f486631489b189
author Frederick Liardet <frederick.liardet@arm.com> Wed Aug 24 10:07:25 2022 +0100
committer Frederick Liardet <frederick.liardet@arm.com> Wed Aug 31 16:47:30 2022 +0100
tree 4460d880bcd6bbfe155a279c869de659e0e22116
parent 545a508429afe1d22760563d252839e13ecd12a3

Add tosa identification numbers

Signed-off-by: Frederick Liardet <frederick.liardet@arm.com>
Change-Id: Idc5668fb01dc8dd657beb911d2cb9295f9275ac6

python/tosa_serializer.py

diff --git a/python/tosa_serializer.py b/python/tosa_serializer.py
index 6915c83..b11f9cd 100644
--- a/python/tosa_serializer.py
+++ b/python/tosa_serializer.py
@@ -60,6 +60,9 @@
     "FLOAT",
 ]
 
+# File identifier needs to be kept in sync with schema/tosa.fbs
+TOSA_GRAPH_IDENTIFIER = b"\x54\x4F\x53\x41"
+
 ByteMask = np.uint64(0xFF)
 
 
@@ -713,7 +716,7 @@
         TosaGraph.TosaGraphAddBlocks(builder, fbv_bb)
         graph = TosaGraph.TosaGraphEnd(builder)
 
-        self.builder.Finish(graph)
+        self.builder.Finish(graph, TOSA_GRAPH_IDENTIFIER)
         return self.builder.Output()
 
     def writeJson(self, tosa_filename):