diff options
author | Jeremy Johnson <jeremy.johnson@arm.com> | 2022-09-12 17:27:37 +0100 |
---|---|---|
committer | Eric Kunze <eric.kunze@arm.com> | 2022-09-21 20:47:22 +0000 |
commit | 00423436471c23c70d89d434271de34bafd5986b (patch) | |
tree | 474bfae46808c0a16617aa5a67be78382c1ce3c3 /verif/generator | |
parent | 49ced1e0e79783f79a2597820b08562c34696b05 (diff) | |
download | reference_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.py | 4 | ||||
-rw-r--r-- | verif/generator/tosa_test_gen.py | 4 | ||||
-rw-r--r-- | verif/generator/tosa_verif_build_tests.py | 16 |
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) |