From ca581575faa862a3bdc7b389996192a168c91482 Mon Sep 17 00:00:00 2001
From: Pedro Alves <pedro.alves@zama.ai>
Date: Mon, 19 Aug 2024 12:44:29 -0300
Subject: [PATCH] fix(gpu): fix compression bench name

---
 tfhe/benches/integer/bench.rs | 20 ++++++++++++--------
 1 file changed, 12 insertions(+), 8 deletions(-)

diff --git a/tfhe/benches/integer/bench.rs b/tfhe/benches/integer/bench.rs
index c238ad7126..c2586c0ecb 100644
--- a/tfhe/benches/integer/bench.rs
+++ b/tfhe/benches/integer/bench.rs
@@ -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)
@@ -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)
@@ -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,
@@ -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,
 };
 
@@ -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" => {