chore(gpu): add missing underscore in comparison tests

zama-ai · Apr 3, 2024 · 1fc3297 · 1fc3297
1 parent cc72594
commit 1fc3297
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 5 deletions.
diff --git a/tfhe/src/integer/gpu/server_key/radix/comparison.rs b/tfhe/src/integer/gpu/server_key/radix/comparison.rs
@@ -3,7 +3,7 @@ use crate::core_crypto::gpu::CudaStream;
 use crate::core_crypto::prelude::{CiphertextModulus, LweCiphertextCount};
 use crate::integer::gpu::ciphertext::boolean_value::CudaBooleanBlock;
 use crate::integer::gpu::ciphertext::info::CudaRadixCiphertextInfo;
-use crate::integer::gpu::ciphertext::CudaIntegerRadixCiphertext;
+use crate::integer::gpu::ciphertext::{CudaIntegerRadixCiphertext, CudaRadixCiphertext};
 use crate::integer::gpu::server_key::CudaBootstrappingKey;
 use crate::integer::gpu::{ComparisonType, CudaServerKey};
 use crate::shortint::ciphertext::Degree;

diff --git a/tfhe/src/integer/gpu/server_key/radix/tests_signed/test_comparison.rs b/tfhe/src/integer/gpu/server_key/radix/tests_signed/test_comparison.rs
@@ -30,7 +30,7 @@ macro_rules! define_gpu_signed_comparison_test_functions {
                 )
             }
 
-            fn [<integer_signed_default_ $comparison_name $clear_type>]<P>(param: P) where P: Into<PBSParameters> {
+            fn [<integer_signed_default_ $comparison_name _ $clear_type>]<P>(param: P) where P: Into<PBSParameters> {
                 let num_tests = 1;
                 let executor = GpuFunctionExecutor::new(&CudaServerKey::[<$comparison_name>]);
                 test_signed_default_function(
@@ -44,7 +44,7 @@ macro_rules! define_gpu_signed_comparison_test_functions {
 
             // Then call our create_gpu_parametrized_test macro onto or specialized fns
             create_gpu_parametrized_test!([<integer_signed_unchecked_ $comparison_name _ $clear_type>]);
-            create_gpu_parametrized_test!([<integer_signed_default_ $comparison_name $clear_type>]);
+            create_gpu_parametrized_test!([<integer_signed_default_ $comparison_name _ $clear_type>]);
         }
     };
 }

diff --git a/tfhe/src/integer/gpu/server_key/radix/tests_unsigned/test_comparison.rs b/tfhe/src/integer/gpu/server_key/radix/tests_unsigned/test_comparison.rs
@@ -29,7 +29,7 @@ macro_rules! define_gpu_comparison_test_functions {
                 )
             }
 
-            fn [<integer_default_ $comparison_name $clear_type:lower>]<P>(param: P) where P: Into<PBSParameters> {
+            fn [<integer_default_ $comparison_name _ $clear_type:lower>]<P>(param: P) where P: Into<PBSParameters> {
                 let num_tests = 1;
                 let executor = GpuFunctionExecutor::new(&CudaServerKey::[<$comparison_name>]);
                 test_default_function(
@@ -41,7 +41,7 @@ macro_rules! define_gpu_comparison_test_functions {
             }
 
             create_gpu_parametrized_test!([<integer_unchecked_ $comparison_name _ $clear_type:lower>]);
-            create_gpu_parametrized_test!([<integer_default_ $comparison_name $clear_type:lower>]);
+            create_gpu_parametrized_test!([<integer_default_ $comparison_name _ $clear_type:lower>]);
         }
     };
 }