From 6a3ff21de27629649f34ca5f45593edacd0c2d97 Mon Sep 17 00:00:00 2001
From: Agnes Leroy <agnes.leroy@zama.ai>
Date: Fri, 12 Jul 2024 16:29:39 +0200
Subject: [PATCH] chore(gpu): fix unsigned integer tests

---
 scripts/integer-tests.sh | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/scripts/integer-tests.sh b/scripts/integer-tests.sh
index 98258edaef..2c44dd9b10 100755
--- a/scripts/integer-tests.sh
+++ b/scripts/integer-tests.sh
@@ -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})