github/concrete
1
1
Fork 0
mirror of https://github.com/zama-ai/concrete.git synced 2026-02-16 23:51:36 -05:00
Code Issues Packages Projects Releases Wiki Activity

perf(compiler): Use parallel implementation to decompress bs and ks keys

Browse Source
This commit is contained in:
Bourgerie Quentin
2024-03-11 10:16:23 +01:00
committed by Quentin Bourgerie
parent 23a521e5d1
commit 0fdc7d78ff
4 changed files with 39 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
compilers/concrete-compiler/compiler/lib/Common/Keys.cpp
View File

@@ -233,7 +233,7 @@ void LweBootstrapKey::decompress() {
concrete_cpu_decompress_seeded_lwe_bootstrap_key_u64(
buffer->data(), seededBuffer->data() + 2, params.getInputLweDimension(),
params.getPolynomialSize(), params.getGlweDimension(),
params.getLevelCount(), params.getBaseLog(), seed);
params.getLevelCount(), params.getBaseLog(), seed, Parallelism::Rayon);
decompressed = true;
return;
}
@@ -353,7 +353,7 @@ void LweKeyswitchKey::decompress() {
concrete_cpu_decompress_seeded_lwe_keyswitch_key_u64(
buffer->data(), seededBuffer->data() + 2, params.getInputLweDimension(),
params.getOutputLweDimension(), params.getLevelCount(),
params.getBaseLog(), seed);
params.getBaseLog(), seed, Parallelism::Rayon);
decompressed = true;
return;
}