aboutsummaryrefslogtreecommitdiff
path: root/verif/generator
diff options
context:
space:
mode:
authorJeremy Johnson <jeremy.johnson@arm.com>2022-09-15 15:01:30 +0100
committerJeremy Johnson <jeremy.johnson@arm.com>2022-09-26 10:12:10 +0100
commita0848c6edbf37034e280a670bdd2f990fdf796da (patch)
tree435f5a440a40c50edf5a9c696c32e8275fc0df15 /verif/generator
parentf7008da16ed36fce2866e0a4a2595acc8f0a27d6 (diff)
downloadreference_model-a0848c6edbf37034e280a670bdd2f990fdf796da.tar.gz
Add CONST testing to Numpy refmodel tests
Signed-off-by: Jeremy Johnson <jeremy.johnson@arm.com> Change-Id: Id32f3b7c287c9e8b0fd42dd96922be5a255598e2
Diffstat (limited to 'verif/generator')
-rw-r--r--verif/generator/tosa_test_gen.py2
-rw-r--r--verif/generator/tosa_verif_build_tests.py7
2 files changed, 8 insertions, 1 deletions
diff --git a/verif/generator/tosa_test_gen.py b/verif/generator/tosa_test_gen.py
index fe05b57..b76b656 100644
--- a/verif/generator/tosa_test_gen.py
+++ b/verif/generator/tosa_test_gen.py
@@ -40,7 +40,7 @@ class TosaTestGen:
fullPath = os.path.join(self.basePath, self.testPath)
os.makedirs(fullPath, exist_ok=True)
- self.ser = ts.TosaSerializer(fullPath)
+ self.ser = ts.TosaSerializer(fullPath, saveConstsToFile=self.args.dump_consts)
def getSerializer(self):
return self.ser
diff --git a/verif/generator/tosa_verif_build_tests.py b/verif/generator/tosa_verif_build_tests.py
index fee551b..2fafacb 100644
--- a/verif/generator/tosa_verif_build_tests.py
+++ b/verif/generator/tosa_verif_build_tests.py
@@ -192,6 +192,13 @@ def parseArgs(argv):
help="set a particular zero point for all valid positive tests",
)
+ parser.add_argument(
+ "--dump-const-tensors",
+ dest="dump_consts",
+ action="store_true",
+ help="output const tensors as numpy files for inspection",
+ )
+
args = parser.parse_args(argv)
return args