diff options
Diffstat (limited to 'verif/generator/tosa_utils.py')
-rw-r--r-- | verif/generator/tosa_utils.py | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/verif/generator/tosa_utils.py b/verif/generator/tosa_utils.py index d79ab3c..29ae898 100644 --- a/verif/generator/tosa_utils.py +++ b/verif/generator/tosa_utils.py @@ -142,6 +142,9 @@ def get_wrong_output_type(op_name, rng, input_dtype): DType.INT32, DType.INT48, ) + else: + # Assume all types but the input type are incorrect + incorrect_types = list(usableDTypes(excludes=(input_dtype,))) return rng.choice(a=incorrect_types) |