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

chore(bench): fix benchmark id pattern for aes and aes256

Browse Source
This commit is contained in:
David Testé
2025-11-19 16:08:04 +01:00
committed by David Testé
parent 19c2146b2d
commit 071e70c037
2 changed files with 8 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -43,7 +43,7 @@ pub mod cuda {
{
const NUM_AES_INPUTS: usize = 1;
const SBOX_PARALLELISM: usize = 16;
let bench_id = format!("{param_name}::{NUM_AES_INPUTS}_input_encryption");
let bench_id = format!("{bench_name}::{param_name}::{NUM_AES_INPUTS}_input_encryption");
let round_keys = sks.key_expansion(&d_key, &streams);
@@ -72,7 +72,7 @@ pub mod cuda {
}
{
let bench_id = format!("{param_name}::key_expansion");
let bench_id = format!("{bench_name}::{param_name}::key_expansion");
bench_group.bench_function(&bench_id, |b| {
b.iter(|| {
black_box(sks.key_expansion(&d_key, &streams));
@@ -93,7 +93,8 @@ pub mod cuda {
{
const NUM_AES_INPUTS: usize = 192;
const SBOX_PARALLELISM: usize = 16;
let bench_id = format!("{param_name}::{NUM_AES_INPUTS}_inputs_encryption");
let bench_id =
format!("{bench_name}::{param_name}::{NUM_AES_INPUTS}_inputs_encryption");
let streams = CudaStreams::new_multi_gpu();
let (cpu_cks, _) = KEY_CACHE.get_from_params(atomic_param, IntegerKeyKind::Radix);

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

@@ -47,7 +47,7 @@ pub mod cuda {
{
const NUM_AES_INPUTS: usize = 1;
const SBOX_PARALLELISM: usize = 16;
let bench_id = format!("{param_name}::{NUM_AES_INPUTS}_input_encryption");
let bench_id = format!("{bench_name}::{param_name}::{NUM_AES_INPUTS}_input_encryption");
let round_keys = sks.key_expansion_256(&d_key, &streams);
@@ -76,7 +76,7 @@ pub mod cuda {
}
{
let bench_id = format!("{param_name}::key_expansion");
let bench_id = format!("{bench_name}::{param_name}::key_expansion");
bench_group.bench_function(&bench_id, |b| {
b.iter(|| {
black_box(sks.key_expansion_256(&d_key, &streams));
@@ -97,7 +97,8 @@ pub mod cuda {
{
const NUM_AES_INPUTS: usize = 192;
const SBOX_PARALLELISM: usize = 16;
let bench_id = format!("{param_name}::{NUM_AES_INPUTS}_inputs_encryption");
let bench_id =
format!("{bench_name}::{param_name}::{NUM_AES_INPUTS}_inputs_encryption");
let streams = CudaStreams::new_multi_gpu();
let (cpu_cks, _) = KEY_CACHE.get_from_params(atomic_param, IntegerKeyKind::Radix);