Skip to content

Commit

Permalink
fix(gpu): fix compression bench name
Browse files Browse the repository at this point in the history
  • Loading branch information
pdroalves committed Aug 19, 2024
1 parent e1a25a1 commit ca58157
Showing 1 changed file with 12 additions and 8 deletions.
20 changes: 12 additions & 8 deletions tfhe/benches/integer/bench.rs
Original file line number Diff line number Diff line change
Expand Up @@ -1404,8 +1404,8 @@ mod cuda {
bench_group.finish()
}

fn cuda_compress(c: &mut Criterion) {
let bench_name = "integer::cuda::compression";
fn cuda_compress_unsigned(c: &mut Criterion) {
let bench_name = "integer::cuda::unsigned::compression";
let mut bench_group = c.benchmark_group(bench_name);
bench_group
.sample_size(15)
Expand Down Expand Up @@ -1459,8 +1459,8 @@ mod cuda {
}
}

fn cuda_decompress(c: &mut Criterion) {
let bench_name = "integer::cuda::compression";
fn cuda_decompress_unsigned(c: &mut Criterion) {
let bench_name = "integer::cuda::unsigned::compression";
let mut bench_group = c.benchmark_group(bench_name);
bench_group
.sample_size(15)
Expand Down Expand Up @@ -2171,7 +2171,11 @@ mod cuda {
cuda_unsigned_overflowing_scalar_add,
);

criterion_group!(cuda_compress_ops, cuda_compress, cuda_decompress);
criterion_group!(
cuda_compress_unsigned_ops,
cuda_compress_unsigned,
cuda_decompress_unsigned
);

fn cuda_bench_server_key_cast_function<F>(
c: &mut Criterion,
Expand Down Expand Up @@ -2263,7 +2267,7 @@ mod cuda {

#[cfg(feature = "gpu")]
use cuda::{
cuda_cast_ops, cuda_compress_ops, default_cuda_dedup_ops, default_cuda_ops,
cuda_cast_ops, cuda_compress_unsigned_ops, default_cuda_dedup_ops, default_cuda_ops,
default_scalar_cuda_ops, unchecked_cuda_ops, unchecked_scalar_cuda_ops,
};

Expand Down Expand Up @@ -2616,13 +2620,13 @@ criterion_group!(oprf, oprf::unsigned_oprf);
fn go_through_gpu_bench_groups(val: &str) {
match val.to_lowercase().as_str() {
"default" => {
cuda_compress_ops();
cuda_compress_unsigned_ops();
default_cuda_ops();
default_scalar_cuda_ops();
cuda_cast_ops();
}
"fast_default" => {
cuda_compress_ops();
cuda_compress_unsigned_ops();
default_cuda_dedup_ops();
}
"unchecked" => {
Expand Down

0 comments on commit ca58157

Please sign in to comment.