aboutsummaryrefslogtreecommitdiff
path: root/verif/generator
diff options
context:
space:
mode:
authorJeremy Johnson <jeremy.johnson@arm.com>2022-09-12 17:27:37 +0100
committerEric Kunze <eric.kunze@arm.com>2022-09-21 20:47:22 +0000
commit00423436471c23c70d89d434271de34bafd5986b (patch)
tree474bfae46808c0a16617aa5a67be78382c1ce3c3 /verif/generator
parent49ced1e0e79783f79a2597820b08562c34696b05 (diff)
downloadreference_model-00423436471c23c70d89d434271de34bafd5986b.tar.gz
Add simple post commit reference model testing against Numpy
Only perform testing of this after build of ref model using postcommit pytest mark. Signed-off-by: Jeremy Johnson <jeremy.johnson@arm.com> Change-Id: I771a18d2c9cd4051fecafad3e6079b44f2ed62fa
Diffstat (limited to 'verif/generator')
-rw-r--r--verif/generator/tosa_arg_gen.py4
-rw-r--r--verif/generator/tosa_test_gen.py4
-rw-r--r--verif/generator/tosa_verif_build_tests.py16
3 files changed, 20 insertions, 4 deletions
diff --git a/verif/generator/tosa_arg_gen.py b/verif/generator/tosa_arg_gen.py
index ef84762..a65e220 100644
--- a/verif/generator/tosa_arg_gen.py
+++ b/verif/generator/tosa_arg_gen.py
@@ -29,8 +29,12 @@ class TosaQuantGen:
def getZeroPoint(testGen, dtype, error_name=None):
if dtype == DType.INT8:
+ if testGen.args.zeropoint is not None:
+ return min(127, max(-128, testGen.args.zeropoint))
return testGen.randInt(-128, 128)
elif dtype == DType.UINT8:
+ if testGen.args.zeropoint is not None:
+ return min(255, max(0, testGen.args.zeropoint))
return testGen.randInt(0, 256)
elif error_name in [
ErrorIf.InputZeroPointNotZero,
diff --git a/verif/generator/tosa_test_gen.py b/verif/generator/tosa_test_gen.py
index 56a34e5..fe05b57 100644
--- a/verif/generator/tosa_test_gen.py
+++ b/verif/generator/tosa_test_gen.py
@@ -2252,6 +2252,10 @@ class TosaTestGen:
def createDynamicOpLists(self):
+ if "conv2d_TEMPLATE" not in self.TOSA_OP_LIST:
+ # Already created these lists (can occur when class is initialized more than once)
+ return
+
# Dynamically create op lists for convolutions with a list of kernel sizes
KERNELS_2D = [[1, 1], [2, 2], [3, 3], [5, 5], [3, 1], [1, 3]]
diff --git a/verif/generator/tosa_verif_build_tests.py b/verif/generator/tosa_verif_build_tests.py
index 6ee873f..fee551b 100644
--- a/verif/generator/tosa_verif_build_tests.py
+++ b/verif/generator/tosa_verif_build_tests.py
@@ -23,7 +23,7 @@ def auto_int(x):
return int(x, 0)
-def parseArgs():
+def parseArgs(argv):
parser = argparse.ArgumentParser()
parser.add_argument(
@@ -184,14 +184,22 @@ def parseArgs():
help="allow oversize padding, stride and kernel tests",
)
- args = parser.parse_args()
+ parser.add_argument(
+ "--zero-point",
+ dest="zeropoint",
+ default=None,
+ type=int,
+ help="set a particular zero point for all valid positive tests",
+ )
+
+ args = parser.parse_args(argv)
return args
-def main():
+def main(argv=None):
- args = parseArgs()
+ args = parseArgs(argv)
ttg = TosaTestGen(args)