From 00423436471c23c70d89d434271de34bafd5986b Mon Sep 17 00:00:00 2001 From: Jeremy Johnson Date: Mon, 12 Sep 2022 17:27:37 +0100 Subject: 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 Change-Id: I771a18d2c9cd4051fecafad3e6079b44f2ed62fa --- verif/generator/tosa_arg_gen.py | 4 ++++ 1 file changed, 4 insertions(+) (limited to 'verif/generator/tosa_arg_gen.py') 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, -- cgit v1.2.1