mirror of
https://github.com/zama-ai/concrete.git
synced 2026-02-09 03:55:04 -05:00
3cd48f0de2
- FFT can work for any polynomial size, as long as twiddles are provided. - All the twiddles fit in the constant memory. - Bit reverse is not used anymore, no more sw1 and sw2 arrays in constant memory. - Real to complex compression algorithm is changed. - Twiddle initialization functions are removed.
146 lines
7.8 KiB
C++
146 lines
7.8 KiB
C++
#ifndef CUDA_BOOTSTRAP_H
|
|
#define CUDA_BOOTSTRAP_H
|
|
|
|
#include <cstdint>
|
|
|
|
extern "C" {
|
|
|
|
void cuda_convert_lwe_bootstrap_key_32(void *dest, void *src, void *v_stream,
|
|
uint32_t gpu_index,
|
|
uint32_t input_lwe_dim,
|
|
uint32_t glwe_dim, uint32_t level_count,
|
|
uint32_t polynomial_size);
|
|
|
|
void cuda_convert_lwe_bootstrap_key_64(void *dest, void *src, void *v_stream,
|
|
uint32_t gpu_index,
|
|
uint32_t input_lwe_dim,
|
|
uint32_t glwe_dim, uint32_t level_count,
|
|
uint32_t polynomial_size);
|
|
|
|
void cuda_bootstrap_amortized_lwe_ciphertext_vector_32(
|
|
void *v_stream, uint32_t gpu_index, void *lwe_array_out, void *test_vector,
|
|
void *test_vector_indexes, void *lwe_array_in, void *bootstrapping_key,
|
|
uint32_t lwe_dimension, uint32_t glwe_dimension, uint32_t polynomial_size,
|
|
uint32_t base_log, uint32_t level_count, uint32_t num_samples,
|
|
uint32_t num_test_vectors, uint32_t lwe_idx, uint32_t max_shared_memory);
|
|
|
|
void cuda_bootstrap_amortized_lwe_ciphertext_vector_64(
|
|
void *v_stream, uint32_t gpu_index, void *lwe_array_out, void *test_vector,
|
|
void *test_vector_indexes, void *lwe_array_in, void *bootstrapping_key,
|
|
uint32_t lwe_dimension, uint32_t glwe_dimension, uint32_t polynomial_size,
|
|
uint32_t base_log, uint32_t level_count, uint32_t num_samples,
|
|
uint32_t num_test_vectors, uint32_t lwe_idx, uint32_t max_shared_memory);
|
|
|
|
void cuda_bootstrap_low_latency_lwe_ciphertext_vector_32(
|
|
void *v_stream, uint32_t gpu_index, void *lwe_array_out, void *test_vector,
|
|
void *test_vector_indexes, void *lwe_array_in, void *bootstrapping_key,
|
|
uint32_t lwe_dimension, uint32_t glwe_dimension, uint32_t polynomial_size,
|
|
uint32_t base_log, uint32_t level_count, uint32_t num_samples,
|
|
uint32_t num_test_vectors, uint32_t lwe_idx, uint32_t max_shared_memory);
|
|
|
|
void cuda_bootstrap_low_latency_lwe_ciphertext_vector_64(
|
|
void *v_stream, uint32_t gpu_index, void *lwe_array_out, void *test_vector,
|
|
void *test_vector_indexes, void *lwe_array_in, void *bootstrapping_key,
|
|
uint32_t lwe_dimension, uint32_t glwe_dimension, uint32_t polynomial_size,
|
|
uint32_t base_log, uint32_t level_count, uint32_t num_samples,
|
|
uint32_t num_test_vectors, uint32_t lwe_idx, uint32_t max_shared_memory);
|
|
|
|
void cuda_cmux_tree_32(void *v_stream, uint32_t gpu_index, void *glwe_array_out,
|
|
void *ggsw_in, void *lut_vector, uint32_t glwe_dimension,
|
|
uint32_t polynomial_size, uint32_t base_log,
|
|
uint32_t level_count, uint32_t r, uint32_t tau,
|
|
uint32_t max_shared_memory);
|
|
|
|
void cuda_cmux_tree_64(void *v_stream, uint32_t gpu_index, void *glwe_array_out,
|
|
void *ggsw_in, void *lut_vector, uint32_t glwe_dimension,
|
|
uint32_t polynomial_size, uint32_t base_log,
|
|
uint32_t level_count, uint32_t r, uint32_t tau,
|
|
uint32_t max_shared_memory);
|
|
|
|
void cuda_blind_rotate_and_sample_extraction_64(
|
|
void *v_stream, uint32_t gpu_index, void *lwe_out, void *ggsw_in,
|
|
void *lut_vector, uint32_t mbr_size, uint32_t tau, uint32_t glwe_dimension,
|
|
uint32_t polynomial_size, uint32_t base_log, uint32_t l_gadget,
|
|
uint32_t max_shared_memory);
|
|
|
|
void cuda_extract_bits_32(
|
|
void *v_stream, uint32_t gpu_index, void *list_lwe_array_out,
|
|
void *lwe_array_in, void *lwe_array_in_buffer,
|
|
void *lwe_array_in_shifted_buffer, void *lwe_array_out_ks_buffer,
|
|
void *lwe_array_out_pbs_buffer, void *lut_pbs, void *lut_vector_indexes,
|
|
void *ksk, void *fourier_bsk, uint32_t number_of_bits, uint32_t delta_log,
|
|
uint32_t lwe_dimension_in, uint32_t lwe_dimension_out,
|
|
uint32_t glwe_dimension, uint32_t base_log_bsk, uint32_t level_count_bsk,
|
|
uint32_t base_log_ksk, uint32_t level_count_ksk, uint32_t number_of_samples,
|
|
uint32_t max_shared_memory);
|
|
|
|
void cuda_extract_bits_64(
|
|
void *v_stream, uint32_t gpu_index, void *list_lwe_array_out,
|
|
void *lwe_array_in, void *lwe_array_in_buffer,
|
|
void *lwe_array_in_shifted_buffer, void *lwe_array_out_ks_buffer,
|
|
void *lwe_array_out_pbs_buffer, void *lut_pbs, void *lut_vector_indexes,
|
|
void *ksk, void *fourier_bsk, uint32_t number_of_bits, uint32_t delta_log,
|
|
uint32_t lwe_dimension_in, uint32_t lwe_dimension_out,
|
|
uint32_t glwe_dimension, uint32_t base_log_bsk, uint32_t level_count_bsk,
|
|
uint32_t base_log_ksk, uint32_t level_count_ksk, uint32_t number_of_samples,
|
|
uint32_t max_shared_memory);
|
|
|
|
void cuda_circuit_bootstrap_32(
|
|
void *v_stream, uint32_t gpu_index, void *ggsw_out, void *lwe_array_in,
|
|
void *fourier_bsk, void *fp_ksk_array, void *lwe_array_in_shifted_buffer,
|
|
void *lut_vector, void *lut_vector_indexes, void *lwe_array_out_pbs_buffer,
|
|
void *lwe_array_in_fp_ks_buffer, uint32_t delta_log,
|
|
uint32_t polynomial_size, uint32_t glwe_dimension, uint32_t lwe_dimension,
|
|
uint32_t level_bsk, uint32_t base_log_bsk, uint32_t level_pksk,
|
|
uint32_t base_log_pksk, uint32_t level_cbs, uint32_t base_log_cbs,
|
|
uint32_t number_of_samples, uint32_t max_shared_memory);
|
|
|
|
void cuda_circuit_bootstrap_64(
|
|
void *v_stream, uint32_t gpu_index, void *ggsw_out, void *lwe_array_in,
|
|
void *fourier_bsk, void *fp_ksk_array, void *lwe_array_in_shifted_buffer,
|
|
void *lut_vector, void *lut_vector_indexes, void *lwe_array_out_pbs_buffer,
|
|
void *lwe_array_in_fp_ks_buffer, uint32_t delta_log,
|
|
uint32_t polynomial_size, uint32_t glwe_dimension, uint32_t lwe_dimension,
|
|
uint32_t level_bsk, uint32_t base_log_bsk, uint32_t level_pksk,
|
|
uint32_t base_log_pksk, uint32_t level_cbs, uint32_t base_log_cbs,
|
|
uint32_t number_of_samples, uint32_t max_shared_memory);
|
|
|
|
void cuda_circuit_bootstrap_vertical_packing_64(
|
|
void *v_stream, uint32_t gpu_index, void *lwe_array_out, void *lwe_array_in,
|
|
void *fourier_bsk, void *cbs_fpksk, void *lut_vector,
|
|
uint32_t polynomial_size, uint32_t glwe_dimension, uint32_t lwe_dimension,
|
|
uint32_t level_count_bsk, uint32_t base_log_bsk, uint32_t level_count_pksk,
|
|
uint32_t base_log_pksk, uint32_t level_count_cbs, uint32_t base_log_cbs,
|
|
uint32_t number_of_inputs, uint32_t lut_number, uint32_t max_shared_memory);
|
|
|
|
void cuda_wop_pbs_64(void *v_stream, uint32_t gpu_index, void *lwe_array_out,
|
|
void *lwe_array_in, void *lut_vector, void *fourier_bsk,
|
|
void *ksk, void *cbs_fpksk, uint32_t glwe_dimension,
|
|
uint32_t lwe_dimension, uint32_t polynomial_size,
|
|
uint32_t base_log_bsk, uint32_t level_count_bsk,
|
|
uint32_t base_log_ksk, uint32_t level_count_ksk,
|
|
uint32_t base_log_pksk, uint32_t level_count_pksk,
|
|
uint32_t base_log_cbs, uint32_t level_count_cbs,
|
|
uint32_t number_of_bits_of_message_including_padding,
|
|
uint32_t number_of_bits_to_extract,
|
|
uint32_t number_of_inputs, uint32_t max_shared_memory);
|
|
}
|
|
|
|
#ifdef __CUDACC__
|
|
__device__ inline int get_start_ith_ggsw(int i, uint32_t polynomial_size,
|
|
int glwe_dimension,
|
|
uint32_t level_count);
|
|
|
|
template <typename T>
|
|
__device__ T *get_ith_mask_kth_block(T *ptr, int i, int k, int level,
|
|
uint32_t polynomial_size,
|
|
int glwe_dimension, uint32_t level_count);
|
|
|
|
template <typename T>
|
|
__device__ T *get_ith_body_kth_block(T *ptr, int i, int k, int level,
|
|
uint32_t polynomial_size,
|
|
int glwe_dimension, uint32_t level_count);
|
|
#endif
|
|
|
|
#endif // CUDA_BOOTSTRAP_H
|