fix(gpu): fix rtx 4090 tests

zama-ai · Oct 7, 2024 · 78d1f29 · 78d1f29
1 parent 7c7e344
commit 78d1f29
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/backends/tfhe-cuda-backend/cuda/src/device.cu b/backends/tfhe-cuda-backend/cuda/src/device.cu
@@ -274,7 +274,7 @@ int cuda_get_max_shared_memory(uint32_t gpu_index) {
 #if CUDA_ARCH == 900
   max_shared_memory = 226000;
 #elif CUDA_ARCH == 890
-  max_shared_memory = 127000;
+  max_shared_memory = 110000;
 #elif CUDA_ARCH == 800
   max_shared_memory = 163000;
 #elif CUDA_ARCH == 700