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

This commit is contained in:
David Testé
2024-07-05 15:25:19 +02:00
committed by David Testé
parent 8ec24d1bb7
commit cd329729d7

View File

@@ -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