From 62c1e95dc2f438e027f94e1dd2f486631489b189 Mon Sep 17 00:00:00 2001
From: Frederick Liardet <frederick.liardet@arm.com>
Date: Wed, 24 Aug 2022 10:07:25 +0100
Subject: Add tosa identification numbers

Signed-off-by: Frederick Liardet <frederick.liardet@arm.com>
Change-Id: Idc5668fb01dc8dd657beb911d2cb9295f9275ac6
---
 python/tosa_serializer.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

(limited to '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 @@ DTypeNames = [
     "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 @@ class TosaSerializer:
         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):
-- 
cgit v1.2.1