chore(gpu): fix unsigned integer tests

zama-ai · Jul 12, 2024 · 1df8854 · 1df8854
1 parent 0370bf6
commit 1df8854
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/scripts/integer-tests.sh b/scripts/integer-tests.sh
@@ -124,7 +124,8 @@ fi
 
 # Override test-threads number to avoid Out-of-memory issues on GPU instances
 if [[ "${backend}" == "gpu" ]]; then
-    test_threads=6
+    test_threads=5
+    doctest_threads=5
 fi
 
 filter_expression=$(/usr/bin/python3 scripts/test_filtering.py --layer integer --backend "${backend}" ${fast_tests_argument} ${multi_bit_argument} ${sign_argument} ${no_big_params_argument})