github/tfhe-rs
1
1
Fork 0
mirror of https://github.com/zama-ai/tfhe-rs.git synced 2026-01-10 15:18:33 -05:00
Code Issues Packages Projects Releases Wiki Activity

chore(gpu): add corner case test for mul

Browse Source
This commit is contained in:
Agnes Leroy
2025-07-30 17:13:46 +02:00
committed by Agnès Leroy
parent 82a5cc7f2d
commit b6e6abb066
1 changed files with 10 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
tfhe/src/integer/gpu/server_key/radix/tests_unsigned/test_mul.rs
View File

@@ -3,12 +3,13 @@ use crate::integer::gpu::server_key::radix::tests_unsigned::{
};
use crate::integer::gpu::CudaServerKey;
use crate::integer::server_key::radix_parallel::tests_cases_unsigned::{
default_mul_test, unchecked_mul_test,
default_mul_test, unchecked_mul_corner_cases_test, unchecked_mul_test,
};
use crate::shortint::parameters::test_params::*;
use crate::shortint::parameters::*;
create_gpu_parameterized_test!(integer_unchecked_mul);
create_gpu_parameterized_test!(integer_unchecked_mul_corner_cases);
create_gpu_parameterized_test!(integer_mul {
PARAM_MESSAGE_2_CARRY_2_KS_PBS_TUNIFORM_2M128,
TEST_PARAM_MESSAGE_3_CARRY_3_KS_PBS_GAUSSIAN_2M128,
@@ -24,6 +25,14 @@ where
unchecked_mul_test(param, executor);
}
fn integer_unchecked_mul_corner_cases<P>(param: P)
where
P: Into<TestParameters>,
{
let executor = GpuFunctionExecutor::new(&CudaServerKey::unchecked_mul);
unchecked_mul_corner_cases_test(param, executor);
}
fn integer_mul<P>(param: P)
where
P: Into<TestParameters>,