chore(ci): fix test filtering for gpu multi-bit parameters set

zama-ai · Jul 16, 2024 · ebf1fd9 · ebf1fd9
1 parent cef055b
commit ebf1fd9
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/scripts/test_filtering.py b/scripts/test_filtering.py
@@ -100,6 +100,9 @@ def filter_integer_tests(input_args):
     backend_filter = ""
     if input_args.backend == "gpu":
         backend_filter = "gpu::"
+        if multi_bit_filter:
+            # For now, GPU only has specific parameters set for multi-bit
+            multi_bit_filter = "_gpu_multi_bit"
 
     filter_expression = [f"test(/^integer::{backend_filter}.*/)"]