diff options
Diffstat (limited to 'verif/generator/tosa_test_gen.py')
-rw-r--r-- | verif/generator/tosa_test_gen.py | 6 |
1 files changed, 5 insertions, 1 deletions
diff --git a/verif/generator/tosa_test_gen.py b/verif/generator/tosa_test_gen.py index bd371eb..b5e71ac 100644 --- a/verif/generator/tosa_test_gen.py +++ b/verif/generator/tosa_test_gen.py @@ -51,7 +51,11 @@ class TosaTestGen: fullPath = os.path.join(self.basePath, self.testPath) os.makedirs(fullPath, exist_ok=True) - self.ser = ts.TosaSerializer(fullPath, saveConstsToFile=self.args.dump_consts) + # Embed const data in the flatbuffer + constMode = ts.ConstMode.EMBED + if self.args.dump_consts: + constMode = ts.ConstMode.EMBED_DUMP + self.ser = ts.TosaSerializer(fullPath, constMode) def getSerializer(self): return self.ser |