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

bench(backend/cuda): restrict the number of inputs for large poly sizes

Browse Source
This commit is contained in:
Agnes Leroy
2023-03-30 11:18:22 +02:00
committed by Agnès Leroy
parent c236dffdd8
commit 32bcda3a93
1 changed files with 8 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
backends/concrete-cuda/implementation/test_and_benchmark/benchmark/benchmark_bootstrap.cpp
View File

@@ -230,11 +230,17 @@ BootstrapBenchmarkGenerateParams(benchmark::internal::Benchmark *b) {
};
// Add to the list of parameters to benchmark
for (auto x : params)
for (int num_samples = 1; num_samples <= 10000; num_samples *= 10) {
for (auto x : params) {
int max_num_samples = 10000;
if (x.polynomial_size >= 8192) {
max_num_samples = 100;
}
for (int num_samples = 1; num_samples <= max_num_samples;
num_samples *= 10) {
b->Args({x.lwe_dimension, x.glwe_dimension, x.polynomial_size,
x.pbs_base_log, x.pbs_level, num_samples});
}
}
}
BENCHMARK_REGISTER_F(Bootstrap_u64, ConcreteCuda_AmortizedPBS)