diff options
author | Jeremy Johnson <jeremy.johnson@arm.com> | 2023-11-29 15:57:58 +0000 |
---|---|---|
committer | Jeremy Johnson <jeremy.johnson@arm.com> | 2023-12-04 10:01:48 +0000 |
commit | fe79accba2c220036c7b5ea0aa27bff5ef74ec73 (patch) | |
tree | aa7374aa4132ebbf7925d4ecde1174b612b49b83 /verif/generator/tosa_arg_gen.py | |
parent | 708da823504b9a7f4e2ffc10e00f06bb092ce637 (diff) | |
download | reference_model-fe79accba2c220036c7b5ea0aa27bff5ef74ec73.tar.gz |
Main Compliance testing support for RESHAPE
Signed-off-by: Jeremy Johnson <jeremy.johnson@arm.com>
Change-Id: I750ef5d3c95daca10e7717437ed375a81b434582
Diffstat (limited to 'verif/generator/tosa_arg_gen.py')
-rw-r--r-- | verif/generator/tosa_arg_gen.py | 20 |
1 files changed, 16 insertions, 4 deletions
diff --git a/verif/generator/tosa_arg_gen.py b/verif/generator/tosa_arg_gen.py index c557207..8e88390 100644 --- a/verif/generator/tosa_arg_gen.py +++ b/verif/generator/tosa_arg_gen.py @@ -2584,8 +2584,8 @@ class TosaArgGen: # Check for duplicates found = False - for name, other_shape in arg_list: - if other_shape[0] == newShape: + for name, args_dict in arg_list: + if args_dict["new_shape"] == newShape: found = True break @@ -2620,16 +2620,28 @@ class TosaArgGen: new_shape_inferred[extra_dim] = -1 else: arg_list.append( - ("perm{}_rank{}_outdefined".format(p, newRank), [newShape]) + ( + "perm{}_rank{}_outdefined".format(p, newRank), + {"new_shape": newShape}, + ) ) if error_name != ErrorIf.TensorSizeInputOutputMismatch: arg_list.append( ( "perm{}_rank{}_outinferred".format(p, newRank), - [new_shape_inferred], + {"new_shape": new_shape_inferred}, ) ) + # Now add data generator types + arg_list = TosaArgGen._add_data_generators( + testGen, + opName, + dtype, + arg_list, + error_name, + ) + return arg_list @staticmethod |