diff options
| author | Caleb Zulawski <caleb.zulawski@gmail.com> | 2024-02-17 11:33:13 -0500 |
|---|---|---|
| committer | Caleb Zulawski <caleb.zulawski@gmail.com> | 2024-02-17 11:33:13 -0500 |
| commit | aebf6f156056ed803afed2ad055094d2ff4fc0cb (patch) | |
| tree | 1fee65c9cdfd2836bc02e92594abfe9382915799 | |
| parent | 44b4d26273e6d18dcd6bbadd66e793c229e89a59 (diff) | |
| download | rust-aebf6f156056ed803afed2ad055094d2ff4fc0cb.tar.gz rust-aebf6f156056ed803afed2ad055094d2ff4fc0cb.zip | |
Use intrinsics directly to avoid recursion
| -rw-r--r-- | crates/core_simd/src/masks.rs | 13 |
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] |
