mikros/rust
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Merge pull request #120 from miguelraz/simd_fsqrt

Browse Source 
add simd_fsqrt intrinsic
This commit is contained in:
Caleb Zulawski 2021-05-24 19:07:12 -04:00 committed by GitHub
commit 20c3b8e13f
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
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! (