mirror of
https://github.com/zama-ai/concrete.git
synced 2026-02-09 03:55:04 -05:00
0aedb1a4f4
- Add FP-Keyswitch. - Add entry points for cuda fk ksk in the public API. - Add test for fp_ksk in cuda backend. - Add fixture for bit extract Co-authored-by: agnesLeroy <agnes.leroy@zama.ai>
38 lines
1.7 KiB
C++
38 lines
1.7 KiB
C++
#ifndef CNCRT_KS_H_
|
|
#define CNCRT_KS_H_
|
|
|
|
#include <cstdint>
|
|
|
|
extern "C" {
|
|
|
|
void cuda_keyswitch_lwe_ciphertext_vector_32(
|
|
void *v_stream, uint32_t gpu_index, void *lwe_array_out, void *lwe_array_in,
|
|
void *ksk, uint32_t lwe_dimension_in, uint32_t lwe_dimension_out,
|
|
uint32_t base_log, uint32_t level_count, uint32_t num_samples);
|
|
|
|
void cuda_keyswitch_lwe_ciphertext_vector_64(
|
|
void *v_stream, uint32_t gpu_index, void *lwe_array_out, void *lwe_array_in,
|
|
void *ksk, uint32_t lwe_dimension_in, uint32_t lwe_dimension_out,
|
|
uint32_t base_log, uint32_t level_count, uint32_t num_samples);
|
|
|
|
void cuda_fp_keyswitch_lwe_to_glwe_32(void *v_stream, void *glwe_array_out,
|
|
void *lwe_array_in, void *fp_ksk_array,
|
|
uint32_t input_lwe_dimension,
|
|
uint32_t output_glwe_dimension,
|
|
uint32_t output_polynomial_size,
|
|
uint32_t base_log, uint32_t level_count,
|
|
uint32_t number_of_input_lwe,
|
|
uint32_t number_of_keys);
|
|
|
|
void cuda_fp_keyswitch_lwe_to_glwe_64(void *v_stream, void *glwe_array_out,
|
|
void *lwe_array_in, void *fp_ksk_array,
|
|
uint32_t input_lwe_dimension,
|
|
uint32_t output_glwe_dimension,
|
|
uint32_t output_polynomial_size,
|
|
uint32_t base_log, uint32_t level_count,
|
|
uint32_t number_of_input_lwe,
|
|
uint32_t number_of_keys);
|
|
}
|
|
|
|
#endif // CNCRT_KS_H_
|