squiid/rust
1
0
Fork 0
Code Issues Pull requests Projects Activity

add simd_fsqrt intrinsic

Browse source
This commit is contained in:
miguel raz 2021-05-18 09:26:01 -05:00
parent ce92300a49
commit d6795814d4
3 changed files with 18 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
crates/core_simd/src/intrinsics.rs
View file

@ -45,6 +45,9 @@ extern "platform-intrinsic" {
/// fabs
pub(crate) fn simd_fabs<T>(x: T) -> T;
/// fsqrt
pub(crate) fn simd_fsqrt<T>(x: T) -> T;
pub(crate) fn simd_eq<T, U>(x: T, y: T) -> U;
pub(crate) fn simd_ne<T, U>(x: T, y: T) -> U;

8
crates/core_simd/src/vector/float.rs
View file

@ -35,6 +35,14 @@ macro_rules! impl_float_vector {
pub fn abs(self) -> Self {
unsafe { crate::intrinsics::simd_fabs(self) }
}
/// Produces a vector where every lane has the square root value
/// of the equivalently-indexed lane in `self`
#[inline]
#[cfg(feature = "std")]
pub fn sqrt(self) -> Self {
unsafe { crate::intrinsics::simd_fsqrt(self) }
}
}
impl<const LANES: usize> $name<LANES>

7
crates/core_simd/tests/ops_macros.rs
View file

@ -426,6 +426,13 @@ macro_rules! impl_float_tests {
)
}
fn sqrt<const LANES: usize>() {
test_helpers::test_unary_elementwise(
&Vector::<LANES>::sqrt,
&Scalar::sqrt,
&|_| true,
)
}
fn horizontal_sum<const LANES: usize>() {
test_helpers::test_1(&|x| {
test_helpers::prop_assert_biteq! (