From c3d88f657ff5a1c4af24045d8a3d581d09a700a5 Mon Sep 17 00:00:00 2001 From: Agnes Leroy Date: Fri, 2 Aug 2024 12:13:54 +0200 Subject: [PATCH] fix(gpu): fix tree_add_chunks after 5547d92c79147a --- .../tfhe-cuda-backend/cuda/src/integer/multiplication.cuh | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/backends/tfhe-cuda-backend/cuda/src/integer/multiplication.cuh b/backends/tfhe-cuda-backend/cuda/src/integer/multiplication.cuh index fc81a409c5..7c3dd985ea 100644 --- a/backends/tfhe-cuda-backend/cuda/src/integer/multiplication.cuh +++ b/backends/tfhe-cuda-backend/cuda/src/integer/multiplication.cuh @@ -234,7 +234,9 @@ __host__ void host_integer_sum_ciphertexts_vec_kb( int32_t h_smart_copy_in[r * num_blocks]; int32_t h_smart_copy_out[r * num_blocks]; - auto max_shared_memory = cuda_get_max_shared_memory(gpu_indexes[0]); + auto max_shared_memory = 0; + cudaDeviceGetAttribute(&max_shared_memory, cudaDevAttrMaxSharedMemoryPerBlock, + 0); // create lut object for message and carry // we allocate luts_message_carry in the host function (instead of scratch)