From 03431e41a9911b17cde885e0803ec703235792c0 Mon Sep 17 00:00:00 2001 From: Pedro Alves Date: Fri, 27 Sep 2024 11:29:14 -0300 Subject: [PATCH] chore(gpu): change index array type in decompression --- backends/tfhe-cuda-backend/cuda/include/compression.h | 2 +- .../cuda/src/integer/compression/compression.cu | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/backends/tfhe-cuda-backend/cuda/include/compression.h b/backends/tfhe-cuda-backend/cuda/include/compression.h index ba44dd3983..537113dcae 100644 --- a/backends/tfhe-cuda-backend/cuda/include/compression.h +++ b/backends/tfhe-cuda-backend/cuda/include/compression.h @@ -28,7 +28,7 @@ void cuda_integer_compress_radix_ciphertext_64( void cuda_integer_decompress_radix_ciphertext_64( void **streams, uint32_t *gpu_indexes, uint32_t gpu_count, - void *lwe_array_out, void *glwe_in, void *indexes_array, + void *lwe_array_out, void *glwe_in, uint32_t *indexes_array, uint32_t indexes_array_size, void **bsks, int8_t *mem_ptr); void cleanup_cuda_integer_compress_radix_ciphertext_64(void **streams, diff --git a/backends/tfhe-cuda-backend/cuda/src/integer/compression/compression.cu b/backends/tfhe-cuda-backend/cuda/src/integer/compression/compression.cu index 43c35dc435..6087b89746 100644 --- a/backends/tfhe-cuda-backend/cuda/src/integer/compression/compression.cu +++ b/backends/tfhe-cuda-backend/cuda/src/integer/compression/compression.cu @@ -58,13 +58,13 @@ void cuda_integer_compress_radix_ciphertext_64( } void cuda_integer_decompress_radix_ciphertext_64( void **streams, uint32_t *gpu_indexes, uint32_t gpu_count, - void *lwe_array_out, void *glwe_in, void *indexes_array, + void *lwe_array_out, void *glwe_in, uint32_t *indexes_array, uint32_t indexes_array_size, void **bsks, int8_t *mem_ptr) { host_integer_decompress( (cudaStream_t *)(streams), gpu_indexes, gpu_count, static_cast(lwe_array_out), static_cast(glwe_in), - static_cast(indexes_array), indexes_array_size, bsks, + indexes_array, indexes_array_size, bsks, (int_decompression *)mem_ptr); }