From ec1235d7571b56ea8887afa25deaaed931e37b2e Mon Sep 17 00:00:00 2001
From: Roman Walch <9820846+rw0x0@users.noreply.github.com>
Date: Mon, 13 Jan 2025 15:21:38 +0100
Subject: [PATCH] .

---
 iris-mpc-gpu/src/threshold_ring/protocol.rs | 1 +
 iris-mpc-gpu/tests/bucket_threshold.rs      | 4 ++--
 2 files changed, 3 insertions(+), 2 deletions(-)

diff --git a/iris-mpc-gpu/src/threshold_ring/protocol.rs b/iris-mpc-gpu/src/threshold_ring/protocol.rs
index 826b0598c..dbaae5bc6 100644
--- a/iris-mpc-gpu/src/threshold_ring/protocol.rs
+++ b/iris-mpc-gpu/src/threshold_ring/protocol.rs
@@ -2406,6 +2406,7 @@ impl Circuits {
 
     // same as compare_threshold_masked_many, just via the functions used in the
     // bucketing
+    // Just here for testing
     pub fn compare_threshold_masked_many_bucket_functions(
         &mut self,
         code_dots: &[ChunkShareView<u16>],
diff --git a/iris-mpc-gpu/tests/bucket_threshold.rs b/iris-mpc-gpu/tests/bucket_threshold.rs
index eb8d5bf76..b51348be2 100644
--- a/iris-mpc-gpu/tests/bucket_threshold.rs
+++ b/iris-mpc-gpu/tests/bucket_threshold.rs
@@ -1,5 +1,5 @@
 #[cfg(feature = "gpu_dependent")]
-mod threshold_test {
+mod bucket_threshold_test {
     use cudarc::{
         driver::{CudaDevice, CudaStream},
         nccl::Id,
@@ -230,7 +230,7 @@ mod threshold_test {
     }
 
     #[tokio::test(flavor = "multi_thread", worker_threads = 3)]
-    async fn test_threshold() -> eyre::Result<()> {
+    async fn test_bucket_threshold() -> eyre::Result<()> {
         install_tracing();
         env::set_var("NCCL_P2P_LEVEL", "LOC");
         env::set_var("NCCL_NET", "Socket");