diff options
Diffstat (limited to 'verif/frameworks/tosa_verif_framework_generator.py')
| -rwxr-xr-x | verif/frameworks/tosa_verif_framework_generator.py | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/verif/frameworks/tosa_verif_framework_generator.py b/verif/frameworks/tosa_verif_framework_generator.py index 8457d92..3b5d012 100755 --- a/verif/frameworks/tosa_verif_framework_generator.py +++ b/verif/frameworks/tosa_verif_framework_generator.py @@ -1147,7 +1147,9 @@ def build_const_net( addl_args_tuple = arg_gen_fcn(op, curr_shape, rng) for desc, addl_args in addl_args_tuple: - if not filter or filter.search(test_dir + desc): + # Only filter on the full test_name, not the output directory + _, test_name = os.path.split(test_dir + desc) + if not filter or filter.search(test_name): unit_test_args.append( [ op_name, |