From 11274f0148fffd5c985317a4349b4c6c6bb9c52d Mon Sep 17 00:00:00 2001 From: Agnes Leroy Date: Fri, 2 Aug 2024 15:13:43 +0200 Subject: [PATCH] fix(gpu): fix argument in scratch mul --- backends/tfhe-cuda-backend/cuda/include/integer.h | 2 +- backends/tfhe-cuda-backend/cuda/src/integer/multiplication.cu | 3 +-- 2 files changed, 2 insertions(+), 3 deletions(-) diff --git a/backends/tfhe-cuda-backend/cuda/include/integer.h b/backends/tfhe-cuda-backend/cuda/include/integer.h index a32f6e6320..c1874aed98 100644 --- a/backends/tfhe-cuda-backend/cuda/include/integer.h +++ b/backends/tfhe-cuda-backend/cuda/include/integer.h @@ -101,7 +101,7 @@ void scratch_cuda_integer_mult_radix_ciphertext_kb_64( uint32_t lwe_dimension, uint32_t polynomial_size, uint32_t pbs_base_log, uint32_t pbs_level, uint32_t ks_base_log, uint32_t ks_level, uint32_t grouping_factor, uint32_t num_blocks, PBS_TYPE pbs_type, - uint32_t max_shared_memory, bool allocate_gpu_memory); + bool allocate_gpu_memory); void cuda_integer_mult_radix_ciphertext_kb_64( void **streams, uint32_t *gpu_indexes, uint32_t gpu_count, diff --git a/backends/tfhe-cuda-backend/cuda/src/integer/multiplication.cu b/backends/tfhe-cuda-backend/cuda/src/integer/multiplication.cu index d0eeae2817..49bb1e4dca 100644 --- a/backends/tfhe-cuda-backend/cuda/src/integer/multiplication.cu +++ b/backends/tfhe-cuda-backend/cuda/src/integer/multiplication.cu @@ -71,7 +71,7 @@ void scratch_cuda_integer_mult_radix_ciphertext_kb_64( uint32_t lwe_dimension, uint32_t polynomial_size, uint32_t pbs_base_log, uint32_t pbs_level, uint32_t ks_base_log, uint32_t ks_level, uint32_t grouping_factor, uint32_t num_radix_blocks, PBS_TYPE pbs_type, - uint32_t max_shared_memory, bool allocate_gpu_memory) { + bool allocate_gpu_memory) { int_radix_params params(pbs_type, glwe_dimension, polynomial_size, polynomial_size * glwe_dimension, lwe_dimension, @@ -123,7 +123,6 @@ void scratch_cuda_integer_mult_radix_ciphertext_kb_64( * - 'num_blocks' is the number of big lwe ciphertext blocks inside radix * ciphertext * - 'pbs_type' selects which PBS implementation should be used - * - 'max_shared_memory' maximum shared memory per cuda block */ void cuda_integer_mult_radix_ciphertext_kb_64( void **streams, uint32_t *gpu_indexes, uint32_t gpu_count,