diff options
author | Jeremy Johnson <jeremy.johnson@arm.com> | 2022-08-24 17:09:09 +0100 |
---|---|---|
committer | Eric Kunze <eric.kunze@arm.com> | 2022-08-25 15:27:20 +0000 |
commit | d32c6da91647dd09d3f22483ded8157941a5ade9 (patch) | |
tree | 45c944b94bc62ae9117bb32bb368d675ebaa7fe4 /verif/generator/tosa_test_gen.py | |
parent | d511f9e604c3e2b915d6f6b7a4975b23ac06041d (diff) | |
download | reference_model-d32c6da91647dd09d3f22483ded8157941a5ade9.tar.gz |
Add PAD ERROR_IF test for output shape
Signed-off-by: Jeremy Johnson <jeremy.johnson@arm.com>
Change-Id: I25a13540734fa30c0c21b46708dfabbec8c4d1e5
Diffstat (limited to 'verif/generator/tosa_test_gen.py')
-rw-r--r-- | verif/generator/tosa_test_gen.py | 5 |
1 files changed, 5 insertions, 0 deletions
diff --git a/verif/generator/tosa_test_gen.py b/verif/generator/tosa_test_gen.py index eeb0ac7..53d38dd 100644 --- a/verif/generator/tosa_test_gen.py +++ b/verif/generator/tosa_test_gen.py @@ -3428,6 +3428,7 @@ class TosaTestGen: "error_if_validators": ( TosaErrorValidator.evWrongInputType, TosaErrorValidator.evPadSmallerZero, + TosaErrorValidator.evPadOutputShapeMismatch, TosaErrorValidator.evWrongOutputType, TosaErrorValidator.evWrongInputList, TosaErrorValidator.evWrongOutputList, @@ -4262,6 +4263,10 @@ class OutputShaper: for i in range(len(output_shape)): output_shape[i] = padding[i][0] + padding[i][1] + output_shape[i] + if error_name == ErrorIf.PadOutputShapeMismatch: + bad_dim = rng.choice(range(len(output_shape))) + output_shape[bad_dim] -= rng.choice([1, 2]) + # Fix negative output shape if error_if test causes it if error_name == ErrorIf.PadSmallerZero and min(output_shape) < 1: output_shape = [i if i >= 1 else 1 for i in output_shape] |