github/tfhe-rs
1
1
Fork 0
mirror of https://github.com/zama-ai/tfhe-rs.git synced 2026-01-09 22:57:59 -05:00
Code Issues Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
David Testé
2024-07-05 15:25:19 +02:00
committed by David Testé
parent 8ec24d1bb7
commit cd329729d7
1 changed files with 5 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
scripts/integer-tests.sh
View File

@@ -122,6 +122,11 @@ else
doctest_threads="${num_cpu_threads}" doctest_threads="${num_cpu_threads}"
fi 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}) 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 if [[ "${FAST_TESTS}" == "TRUE" ]]; then