mirror of
https://github.com/zama-ai/concrete.git
synced 2026-02-18 08:31:31 -05:00
c67fac4082
- added --compress-input compiler option which forces the use of seeded bootstrap keys and keyswitch keys - replaced the concrete-cpu FHE implementation with tfhe-rs Co-authored-by: Nikita Frolov <nf@mkmks.org>
90 lines
3.0 KiB
C++
90 lines
3.0 KiB
C++
// Part of the Concrete Compiler Project, under the BSD3 License with Zama
|
|
// Exceptions. See
|
|
// https://github.com/zama-ai/concrete-compiler-internal/blob/main/LICENSE.txt
|
|
// for license information.
|
|
|
|
#include "concretelang/Runtime/context.h"
|
|
#include "concretelang/Common/Error.h"
|
|
#include "concretelang/Common/Keysets.h"
|
|
#include <assert.h>
|
|
#include <stdio.h>
|
|
|
|
namespace mlir {
|
|
namespace concretelang {
|
|
|
|
FFT::FFT(size_t polynomial_size)
|
|
: fft(nullptr), polynomial_size(polynomial_size) {
|
|
fft = (struct Fft *)aligned_alloc(CONCRETE_FFT_ALIGN, CONCRETE_FFT_SIZE);
|
|
concrete_cpu_construct_concrete_fft(fft, polynomial_size);
|
|
}
|
|
|
|
FFT::FFT(FFT &&other) : fft(other.fft), polynomial_size(other.polynomial_size) {
|
|
other.fft = nullptr;
|
|
}
|
|
|
|
FFT::~FFT() {
|
|
if (fft != nullptr) {
|
|
concrete_cpu_destroy_concrete_fft(fft);
|
|
free(fft);
|
|
}
|
|
}
|
|
|
|
RuntimeContext::RuntimeContext(ServerKeyset serverKeyset)
|
|
: serverKeyset(serverKeyset) {
|
|
{
|
|
|
|
// Initialize for each bootstrap key the fourier one
|
|
for (size_t i = 0; i < serverKeyset.lweBootstrapKeys.size(); i++) {
|
|
|
|
auto bsk = serverKeyset.lweBootstrapKeys[i];
|
|
auto info = bsk.getInfo().asReader();
|
|
|
|
size_t decomposition_level_count = info.getParams().getLevelCount();
|
|
size_t decomposition_base_log = info.getParams().getBaseLog();
|
|
size_t glwe_dimension = info.getParams().getGlweDimension();
|
|
size_t polynomial_size = info.getParams().getPolynomialSize();
|
|
size_t input_lwe_dimension = info.getParams().getInputLweDimension();
|
|
|
|
// Create the FFT
|
|
FFT fft(polynomial_size);
|
|
|
|
// Allocate scratch for key conversion
|
|
size_t scratch_size;
|
|
size_t scratch_align;
|
|
concrete_cpu_bootstrap_key_convert_u64_to_fourier_scratch(
|
|
&scratch_size, &scratch_align, fft.fft);
|
|
auto scratch = (uint8_t *)aligned_alloc(scratch_align, scratch_size);
|
|
|
|
// Allocate the fourier_bootstrap_key
|
|
auto &bsk_buffer = bsk.getBuffer();
|
|
auto fourier_data = std::make_shared<std::vector<std::complex<double>>>();
|
|
fourier_data->resize(bsk_buffer.size() / 2);
|
|
auto bsk_data = bsk_buffer.data();
|
|
|
|
// Convert bootstrap_key to the fourier domain
|
|
concrete_cpu_bootstrap_key_convert_u64_to_fourier(
|
|
bsk_data, fourier_data->data(), decomposition_level_count,
|
|
decomposition_base_log, glwe_dimension, polynomial_size,
|
|
input_lwe_dimension, fft.fft, scratch, scratch_size);
|
|
|
|
// Store the fourier_bootstrap_key in the context
|
|
fourier_bootstrap_keys.push_back(fourier_data);
|
|
ffts.push_back(std::move(fft));
|
|
free(scratch);
|
|
}
|
|
|
|
#ifdef CONCRETELANG_CUDA_SUPPORT
|
|
assert(cudaGetDeviceCount(&num_devices) == cudaSuccess);
|
|
bsk_gpu.resize(num_devices, nullptr);
|
|
ksk_gpu.resize(num_devices, nullptr);
|
|
for (int i = 0; i < num_devices; ++i) {
|
|
bsk_gpu_mutex.push_back(std::make_unique<std::mutex>());
|
|
ksk_gpu_mutex.push_back(std::make_unique<std::mutex>());
|
|
}
|
|
#endif
|
|
}
|
|
}
|
|
|
|
} // namespace concretelang
|
|
} // namespace mlir
|