chore(ci): force test-threads value on gpu integer tests

zama-ai · Jul 5, 2024 · 97bcf02 · 97bcf02
1 parent aa92adc
commit 97bcf02
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/scripts/integer-tests.sh b/scripts/integer-tests.sh
@@ -122,6 +122,11 @@ else
     doctest_threads="${num_cpu_threads}"
 fi
 
+# Override test-threads number to avoid Out-of-memory issues on GPU instances
+if [[ "${backend}" == "gpu" ]]; then
+    test_threads=6
+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})
 
 if [[ "${FAST_TESTS}" == "TRUE" ]]; then