95 lines
2.6 KiB
Rust
95 lines
2.6 KiB
Rust
#![feature(portable_simd)]
|
|
|
|
macro_rules! unary_test {
|
|
{ $scalar:tt, $($func:tt),+ } => {
|
|
test_helpers::test_lanes! {
|
|
$(
|
|
fn $func<const LANES: usize>() {
|
|
test_helpers::test_unary_elementwise(
|
|
&core_simd::simd::Simd::<$scalar, LANES>::$func,
|
|
&$scalar::$func,
|
|
&|_| true,
|
|
)
|
|
}
|
|
)*
|
|
}
|
|
}
|
|
}
|
|
|
|
macro_rules! unary_approx_test {
|
|
{ $scalar:tt, $($func:tt),+ } => {
|
|
test_helpers::test_lanes! {
|
|
$(
|
|
fn $func<const LANES: usize>() {
|
|
test_helpers::test_unary_elementwise_approx(
|
|
&core_simd::simd::Simd::<$scalar, LANES>::$func,
|
|
&$scalar::$func,
|
|
&|_| true,
|
|
8,
|
|
)
|
|
}
|
|
)*
|
|
}
|
|
}
|
|
}
|
|
|
|
macro_rules! binary_approx_test {
|
|
{ $scalar:tt, $($func:tt),+ } => {
|
|
test_helpers::test_lanes! {
|
|
$(
|
|
fn $func<const LANES: usize>() {
|
|
test_helpers::test_binary_elementwise_approx(
|
|
&core_simd::simd::Simd::<$scalar, LANES>::$func,
|
|
&$scalar::$func,
|
|
&|_, _| true,
|
|
16,
|
|
)
|
|
}
|
|
)*
|
|
}
|
|
}
|
|
}
|
|
|
|
macro_rules! ternary_test {
|
|
{ $scalar:tt, $($func:tt),+ } => {
|
|
test_helpers::test_lanes! {
|
|
$(
|
|
fn $func<const LANES: usize>() {
|
|
test_helpers::test_ternary_elementwise(
|
|
&core_simd::simd::Simd::<$scalar, LANES>::$func,
|
|
&$scalar::$func,
|
|
&|_, _, _| true,
|
|
)
|
|
}
|
|
)*
|
|
}
|
|
}
|
|
}
|
|
|
|
macro_rules! impl_tests {
|
|
{ $scalar:tt } => {
|
|
mod $scalar {
|
|
use std_float::StdFloat;
|
|
|
|
unary_test! { $scalar, sqrt, ceil, floor, round, trunc }
|
|
ternary_test! { $scalar, mul_add }
|
|
|
|
// https://github.com/rust-lang/miri/issues/3555
|
|
unary_approx_test! { $scalar, sin, cos, exp, exp2, ln, log2, log10 }
|
|
binary_approx_test! { $scalar, log }
|
|
|
|
test_helpers::test_lanes! {
|
|
fn fract<const LANES: usize>() {
|
|
test_helpers::test_unary_elementwise_flush_subnormals(
|
|
&core_simd::simd::Simd::<$scalar, LANES>::fract,
|
|
&$scalar::fract,
|
|
&|_| true,
|
|
)
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
impl_tests! { f32 }
|
|
impl_tests! { f64 }
|