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): remove constraint in pcc to not use trivial name in bench

Browse Source
This commit is contained in:
Agnes Leroy
2025-12-11 13:50:21 +01:00
committed by Agnès Leroy
parent 8e4bec0b2a
commit b7a706a3db
2 changed files with 5 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
tfhe-benchmark/benches/core_crypto/ks_bench.rs
View File

@@ -449,9 +449,9 @@ mod cuda {
let gpu_count = get_number_of_gpus() as usize;
for uses_gemm_ks in [false, true] {
for uses_simple_indices in [false, true] {
let indices_str = if uses_simple_indices {
"simple"
for uses_trivial_indices in [false, true] {
let indices_str = if uses_trivial_indices {
"trivial"
} else {
"complex"
};
@@ -516,7 +516,7 @@ mod cuda {
})
.collect::<Vec<_>>();
let indexes_range: Vec<u64> = if uses_simple_indices {
let indexes_range: Vec<u64> = if uses_trivial_indices {
(0..(elements / gpu_count) as u64).collect()
} else {
(0..(elements / gpu_count) as u64).rev().collect()
@@ -555,7 +555,7 @@ mod cuda {
output_ct,
&cuda_indexes_vec[i].d_input,
&cuda_indexes_vec[i].d_output,
uses_simple_indices,
uses_trivial_indices,
local_stream,
uses_gemm_ks,
);