about summary refs log tree commit diff
diff options
context:
space:
mode:
authorCaleb Zulawski <caleb.zulawski@gmail.com>2024-02-17 11:33:13 -0500
committerCaleb Zulawski <caleb.zulawski@gmail.com>2024-02-17 11:33:13 -0500
commitaebf6f156056ed803afed2ad055094d2ff4fc0cb (patch)
tree1fee65c9cdfd2836bc02e92594abfe9382915799
parent44b4d26273e6d18dcd6bbadd66e793c229e89a59 (diff)
downloadrust-aebf6f156056ed803afed2ad055094d2ff4fc0cb.tar.gz
rust-aebf6f156056ed803afed2ad055094d2ff4fc0cb.zip
Use intrinsics directly to avoid recursion
-rw-r--r--crates/core_simd/src/masks.rs13
1 files changed, 11 insertions, 2 deletions
diff --git a/crates/core_simd/src/masks.rs b/crates/core_simd/src/masks.rs
index ad7c68fc7f6..e480c25a51e 100644
--- a/crates/core_simd/src/masks.rs
+++ b/crates/core_simd/src/masks.rs
@@ -12,7 +12,7 @@
 )]
 mod mask_impl;
 
-use crate::simd::{cmp::SimdPartialEq, LaneCount, Simd, SimdCast, SimdElement, SupportedLaneCount};
+use crate::simd::{LaneCount, Simd, SimdCast, SimdElement, SupportedLaneCount};
 use core::cmp::Ordering;
 use core::{fmt, mem};
 
@@ -58,7 +58,16 @@ macro_rules! impl_element {
             where
                 LaneCount<N>: SupportedLaneCount,
             {
-                (value.simd_eq(Simd::splat(0 as _)) | value.simd_eq(Simd::splat(-1 as _))).all()
+                // We can't use `Simd` directly, because `Simd`'s functions call this function and
+                // we will end up with an infinite loop.
+                // Safety: `value` is an integer vector
+                unsafe {
+                    use core::intrinsics::simd;
+                    let falses: Simd<Self, N> = simd::simd_eq(value, Simd::splat(0 as _));
+                    let trues: Simd<Self, N> = simd::simd_eq(value, Simd::splat(-1 as _));
+                    let valid: Simd<Self, N> = simd::simd_or(falses, trues);
+                    simd::simd_reduce_all(valid)
+                }
             }
 
             #[inline]