aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorJeremy Johnson <jeremy.johnson@arm.com>2021-12-08 16:13:21 +0000
committerJeremy Johnson <jeremy.johnson@arm.com>2021-12-08 16:13:21 +0000
commitf20a5a48d1df6b4d9c6da38533f77af5591e939e (patch)
treead39d41a564c530fbe734832de4ad95fb5fdf36b
parent729b035e447dda1f16df1aaee64b47c394e24f96 (diff)
downloadreference_model-f20a5a48d1df6b4d9c6da38533f77af5591e939e.tar.gz
Stop COND_IF neg tests creating bad arrays
Signed-off-by: Jeremy Johnson <jeremy.johnson@arm.com> Change-Id: I267fa624241f60f1fa613e27e4fe46fc43768f8a
-rw-r--r--verif/tosa_test_gen.py2
1 files changed, 1 insertions, 1 deletions
diff --git a/verif/tosa_test_gen.py b/verif/tosa_test_gen.py
index 655cdfc..0071b9f 100644
--- a/verif/tosa_test_gen.py
+++ b/verif/tosa_test_gen.py
@@ -4761,7 +4761,7 @@ class TosaTestGen:
if error_name in [ErrorIf.CondIfOutputListThenGraphMismatch, ErrorIf.CondIfOutputListElseGraphMismatch]:
incorrect_shape = deepcopy(then_tens.shape)
for i in range(len(incorrect_shape)):
- incorrect_shape[i] = incorrect_shape[i] + self.rng.choice([-3, -2, 2, 3])
+ incorrect_shape[i] += self.rng.choice([-3, -2, 2, 3]) if incorrect_shape[i] > 3 else self.rng.choice([1, 2, 4])
incorrect_arr = np.int32(self.rng.integers(0, 256, size=incorrect_shape))
then_arr = np.int32(self.rng.integers(0, 256, size=out_shape))