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

chore(bench): write params to file for each zk benchmark on gpu

Browse Source 
To be parsable each benchmark criterion ID must have their crypto
details written to a file.
This commit is contained in:
David Testé
2025-08-07 11:44:04 +02:00
committed by David Testé
parent afd8f58a8d
commit 3b42f9873a
1 changed files with 15 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -710,15 +710,21 @@ mod cuda {
}
}
write_to_json::<u64, _>(
&bench_id_verify_and_expand,
shortint_params,
param_name,
"pke_zk_verify_and_expand",
&OperatorType::Atomic,
shortint_params.message_modulus().0 as u32,
vec![shortint_params.message_modulus().0.ilog2(); num_block],
);
for (bench_id, display_name) in [
(bench_id_verify, "pke_zk_verify"),
(bench_id_expand_without_verify, "pke_zk_verify_only_expand"),
(bench_id_verify_and_expand, "pke_zk_verify_and_expand"),
] {
write_to_json::<u64, _>(
&bench_id,
shortint_params,
param_name,
display_name,
&OperatorType::Atomic,
shortint_params.message_modulus().0 as u32,
vec![shortint_params.message_modulus().0.ilog2(); num_block],
);
}
}
}
}