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): 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

1
Makefile
View File

@@ -1343,7 +1343,6 @@ dieharder_csprng: install_dieharder build_tfhe_csprng
.PHONY: clippy_bench # Run clippy lints on tfhe-benchmark
clippy_bench: install_rs_check_toolchain
! (grep --recursive "trivial" tfhe-benchmark && echo "trivial found in benches")
RUSTFLAGS="$(RUSTFLAGS)" cargo "$(CARGO_RS_CHECK_TOOLCHAIN)" clippy --all-targets \
--features=boolean,shortint,integer,internal-keycache,pbs-stats,zk-pok \
-p tfhe-benchmark -- --no-deps -D warnings

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,
);