Filter params

tfhe/benches/core_crypto/ks_pbs_bench.rs

@@ -647,6 +647,9 @@ mod cuda {
                 println!("[WARNING] polynomial size is too large for parameters set '{}' (max: {}, got: {})", name, GPU_MAX_SUPPORTED_POLYNOMIAL_SIZE, params.polynomial_size.unwrap().0);
                 continue;
             }
+            if params.message_modulus.unwrap().ilog2() != params.carry_modulus.unwrap().ilog2() {
+                continue;
+            }
 
             // Create the LweSecretKey
             let input_lwe_secret_key = allocate_and_generate_new_binary_lwe_secret_key(
@@ -956,6 +959,12 @@ mod cuda {
                 println!("[WARNING] polynomial size is too large for parameters set '{}' (max: {}, got: {})", name, GPU_MAX_SUPPORTED_POLYNOMIAL_SIZE, params.polynomial_size.unwrap().0);
                 continue;
             }
+            if params.message_modulus.unwrap().ilog2() != params.carry_modulus.unwrap().ilog2() {
+                continue;
+            }
+            if grouping_factor.0 != 4 {
+                continue;
+            }
 
             // Create the LweSecretKey
             let input_lwe_secret_key = allocate_and_generate_new_binary_lwe_secret_key(

tfhe/benches/core_crypto/pbs_bench.rs

@@ -1109,6 +1109,9 @@ mod cuda {
                 println!("[WARNING] polynomial size is too large for parameters set '{}' (max: {}, got: {})", name, GPU_MAX_SUPPORTED_POLYNOMIAL_SIZE, params.polynomial_size.unwrap().0);
                 continue;
             }
+            if params.message_modulus.unwrap().ilog2() != params.carry_modulus.unwrap().ilog2() {
+                continue;
+            }
 
             // Create the LweSecretKey
             let input_lwe_secret_key = allocate_and_generate_new_binary_lwe_secret_key(
@@ -1368,6 +1371,12 @@ mod cuda {
                 println!("[WARNING] polynomial size is too large for parameters set '{}' (max: {}, got: {})", name, GPU_MAX_SUPPORTED_POLYNOMIAL_SIZE, params.polynomial_size.unwrap().0);
                 continue;
             }
+            if params.message_modulus.unwrap().ilog2() != params.carry_modulus.unwrap().ilog2() {
+                continue;
+            }
+            if grouping_factor.0 != 4 {
+                continue;
+            }
 
             // Create the LweSecretKey
             let input_lwe_secret_key = allocate_and_generate_new_binary_lwe_secret_key(