github/tfhe-rs
1
1
Fork 0
mirror of https://github.com/zama-ai/tfhe-rs.git synced 2026-01-10 07:08:03 -05:00
Code Issues Packages Projects Releases Wiki Activity

chore(gpu): define higher values for the sm size based on compute capability

Browse Source
This commit is contained in:
Agnes Leroy
2024-08-01 10:15:48 +02:00
committed by Agnès Leroy
parent 80fe45f354
commit d69dd20079
6 changed files with 31 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
scripts/integer-tests.sh
View File

@@ -129,8 +129,13 @@ fi
# Override test-threads number to avoid Out-of-memory issues on GPU instances
if [[ "${backend}" == "gpu" ]]; then
test_threads=5
doctest_threads=5
if [[ "${BIG_TESTS_INSTANCE}" == TRUE ]]; then
test_threads=5
doctest_threads=5
else
test_threads=3
doctest_threads=3
fi
fi
filter_expression=$(/usr/bin/python3 scripts/test_filtering.py --layer integer --backend "${backend}" ${fast_tests_argument} ${nightly_tests_argument} ${multi_bit_argument} ${sign_argument} ${no_big_params_argument})