diff options
| author | Luke Hutton <luke.hutton@arm.com> | 2023-02-22 11:53:48 +0000 |
|---|---|---|
| committer | Eric Kunze <eric.kunze@arm.com> | 2023-02-28 20:08:57 +0000 |
| commit | a4e48ca7b032992ca0110900935c08d7cf860cd3 (patch) | |
| tree | a58c8617390225ecc107721d9b5ff87c2bdb01b0 /verif/generator/tosa_error_if.py | |
| parent | 2226f90d5a6c48a975045bc9e0419113ce764aaf (diff) | |
| download | reference_model-a4e48ca7b032992ca0110900935c08d7cf860cd3.tar.gz | |
Update rank limits for SLICE, TILE and TRANSPOSE
Updated to align with corresponding changes to the
spec.
In addition, some ERROR_IF tests have been updated to
match the checks specified by the spec, including:
PAD, SLICE, TILE, TRANSPOSE.
Signed-off-by: Luke Hutton <luke.hutton@arm.com>
Change-Id: Ie2c5f48e79a5610eb82739170e25057a63dac1d8
Diffstat (limited to 'verif/generator/tosa_error_if.py')
| -rw-r--r-- | verif/generator/tosa_error_if.py | 16 |
1 files changed, 10 insertions, 6 deletions
diff --git a/verif/generator/tosa_error_if.py b/verif/generator/tosa_error_if.py index ee227b3..b19d5e9 100644 --- a/verif/generator/tosa_error_if.py +++ b/verif/generator/tosa_error_if.py @@ -1067,7 +1067,9 @@ class TosaErrorValidator: if check: input1_shape = kwargs["input1"].shape - input2_shape = kwargs["input2"].shape + input2_shape = ( + kwargs["input2"].shape if "input2" in kwargs else input1_shape + ) # In case of SELECT op input3_shape = ( kwargs["input3"].shape if "input3" in kwargs else input2_shape @@ -1921,11 +1923,13 @@ class TosaErrorValidator: input_shape = kwargs["input_shape"] output_shape = kwargs["output_shape"] size = kwargs["size"] - rank = len(input_shape) - if len(size) == rank: - for index in range(rank): - if size[index] != output_shape[index]: - error_result = True + + if len(input_shape) == len(output_shape): + rank = len(input_shape) + if len(size) == rank: + for index in range(rank): + if size[index] != output_shape[index]: + error_result = True info_dict = { "error_name": error_name, |