github/tfhe-rs
1
1
Fork 0
mirror of https://github.com/zama-ai/tfhe-rs.git synced 2026-01-09 22:57:59 -05:00
Code Issues Packages Projects Releases Wiki Activity

fix(gpu): reduces number of elements in the ZK throughput benchmark

Browse Source
This commit is contained in:
Pedro Alves
2025-07-09 09:18:19 -03:00
committed by Agnès Leroy
parent 15762623d1
commit d3dd010deb
1 changed files with 6 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
tfhe-benchmark/benches/integer/zk_pke.rs
View File

@@ -602,7 +602,12 @@ mod cuda {
});
}
BenchmarkType::Throughput => {
let elements = 100 * get_number_of_gpus() as u64; // This value, found empirically, ensure saturation of 8XH100 SXM5
let mut elements_per_gpu = 100;
if bits == 4096 {
elements_per_gpu /= 5;
}
// This value, found empirically, ensure saturation of 8XH100 SXM5
let elements = elements_per_gpu * get_number_of_gpus() as u64;
bench_group.throughput(Throughput::Elements(elements));
bench_id_verify = format!(