mirror of
https://github.com/zama-ai/concrete.git
synced 2026-02-09 03:55:04 -05:00
64521f6747
- a new crate concrete-cuda is added to the repository, containing some Cuda implementations for the bootstrap and keyswitch and a Rust wrapping to call them - a new backend_cuda is added to concrete-core, with dedicated entities whose memory is located on the GPU and engines that call the Cuda accelerated functions
25 lines
788 B
C++
25 lines
788 B
C++
#ifndef CNCRT_KS_H_
|
|
#define CNCRT_KS_H_
|
|
|
|
#include <cstdint>
|
|
|
|
extern "C" {
|
|
|
|
void cuda_keyswitch_lwe_ciphertext_vector_32(void *v_stream, void *lwe_out, void *lwe_in,
|
|
void *ksk,
|
|
uint32_t lwe_dimension_before,
|
|
uint32_t lwe_dimension_after,
|
|
uint32_t base_log, uint32_t l_gadget,
|
|
uint32_t num_samples);
|
|
|
|
void cuda_keyswitch_lwe_ciphertext_vector_64(void *v_stream, void *lwe_out, void *lwe_in,
|
|
void *ksk,
|
|
uint32_t lwe_dimension_before,
|
|
uint32_t lwe_dimension_after,
|
|
uint32_t base_log, uint32_t l_gadget,
|
|
uint32_t num_samples);
|
|
|
|
}
|
|
|
|
#endif // CNCRT_KS_H_
|