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 e6d7df8..00490fa 100644 --- a/verif/generator/tosa_arg_gen.py +++ b/verif/generator/tosa_arg_gen.py @@ -2851,8 +2851,17 @@ class TosaArgGen: multiples.append(2) else: multiples.append(testGen.randInt(1, 4)) - arg_list.append(("perm{}".format(p), [multiples])) + arg_list.append(("perm{}".format(p), {"multiples": multiples})) + # 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 |