use crate::integer::gpu::server_key::radix::tests_unsigned::{
create_gpu_parameterized_test, GpuFunctionExecutor,
};
use crate::integer::gpu::CudaServerKey;
use crate::integer::server_key::radix_parallel::tests_unsigned::test_rotate::{
default_rotate_left_test, default_rotate_right_test, unchecked_rotate_left_test,
unchecked_rotate_right_test,
};
use crate::shortint::parameters::test_params::*;
use crate::shortint::parameters::*;
create_gpu_parameterized_test!(integer_unchecked_rotate_left);
create_gpu_parameterized_test!(integer_unchecked_rotate_right);
create_gpu_parameterized_test!(integer_rotate_left {
PARAM_MESSAGE_2_CARRY_2_KS_PBS_TUNIFORM_2M128,
TEST_PARAM_MESSAGE_3_CARRY_3_KS_PBS_GAUSSIAN_2M128,
TEST_PARAM_GPU_MULTI_BIT_GROUP_4_MESSAGE_2_CARRY_2_KS_PBS_TUNIFORM_2M128,
TEST_PARAM_GPU_MULTI_BIT_GROUP_3_MESSAGE_3_CARRY_3_KS_PBS_GAUSSIAN_2M64,
});
create_gpu_parameterized_test!(integer_rotate_right {
PARAM_MESSAGE_2_CARRY_2_KS_PBS_TUNIFORM_2M128,
TEST_PARAM_MESSAGE_3_CARRY_3_KS_PBS_GAUSSIAN_2M128,
TEST_PARAM_GPU_MULTI_BIT_GROUP_4_MESSAGE_2_CARRY_2_KS_PBS_TUNIFORM_2M128,
TEST_PARAM_GPU_MULTI_BIT_GROUP_3_MESSAGE_3_CARRY_3_KS_PBS_GAUSSIAN_2M64,
});
fn integer_unchecked_rotate_right<P>(param: P)
where
P: Into<TestParameters> + Copy,
{
let executor = GpuFunctionExecutor::new(&CudaServerKey::unchecked_rotate_right);
unchecked_rotate_right_test(param, executor);
}
fn integer_rotate_right<P>(param: P)
where
P: Into<TestParameters> + Copy,
{
let executor = GpuFunctionExecutor::new(&CudaServerKey::rotate_right);
default_rotate_right_test(param, executor);
}
fn integer_unchecked_rotate_left<P>(param: P)
where
P: Into<TestParameters> + Copy,
{
let executor = GpuFunctionExecutor::new(&CudaServerKey::unchecked_rotate_left);
unchecked_rotate_left_test(param, executor);
}
fn integer_rotate_left<P>(param: P)
where
P: Into<TestParameters> + Copy,
{
let executor = GpuFunctionExecutor::new(&CudaServerKey::rotate_left);
default_rotate_left_test(param, executor);
}