diff options
Diffstat (limited to 'verif/generator/tosa_arg_gen.py')
| -rw-r--r-- | verif/generator/tosa_arg_gen.py | 11 |
1 files changed, 10 insertions, 1 deletions
diff --git a/verif/generator/tosa_arg_gen.py b/verif/generator/tosa_arg_gen.py index 7825445..b4939da 100644 --- a/verif/generator/tosa_arg_gen.py +++ b/verif/generator/tosa_arg_gen.py @@ -2907,9 +2907,18 @@ class TosaArgGen: # Create list of required amount of permutations arg_list = [ - ("perm{}".format(p), [random_permutations[p].tolist()]) + ("perm{}".format(p), {"perms": random_permutations[p].tolist()}) for p in range(limit) ] + # Now add data generator types + arg_list = TosaArgGen._add_data_generators( + testGen, + opName, + dtype, + arg_list, + error_name, + ) + # Return list of tuples: (arg_str, args_dict) return arg_list @staticmethod |