chore(gpu): fix unsigned integer tests

This commit is contained in:
Agnes Leroy
2024-07-12 16:29:39 +02:00
committed by Agnès Leroy
parent 74cafd0e9d
commit 6a3ff21de2

View File

@@ -129,7 +129,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} ${nightly_tests_argument} ${multi_bit_argument} ${sign_argument} ${no_big_params_argument})