itsscb/rust
1
0
Fork 0
mirror of https://github.com/rust-lang/rust.git synced 2025-11-25 01:07:56 +00:00
Code Issues Packages Projects Releases Wiki Activity

add vec_cmprg_idx and vec_cmpnrg_idx

Browse Source
This commit is contained in:
Folkert de Vries 2025-03-06 16:35:33 +01:00 committed by Amanieu d'Antras
parent 511a8eb3cd
commit 895d11f085
1 changed files with 36 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

36
library/stdarch/crates/core_arch/src/s390x/vector.rs
View File

@ -4736,6 +4736,22 @@ pub unsafe fn vec_cmpnrg<T: sealed::VectorCompareRange>(a: T, b: T, c: T) -> T::
a.vstrc::<{ FindImm::Ne as u32 }>(b, c)
}
/// Vector Compare Ranges Index
#[inline]
#[target_feature(enable = "vector")]
#[unstable(feature = "stdarch_s390x", issue = "135681")]
pub unsafe fn vec_cmprg_idx<T: sealed::VectorCompareRange>(a: T, b: T, c: T) -> T::Result {
a.vstrc::<{ FindImm::EqIdx as u32 }>(b, c)
}
/// Vector Compare Not in Ranges Index
#[inline]
#[target_feature(enable = "vector")]
#[unstable(feature = "stdarch_s390x", issue = "135681")]
pub unsafe fn vec_cmpnrg_idx<T: sealed::VectorCompareRange>(a: T, b: T, c: T) -> T::Result {
a.vstrc::<{ FindImm::NeIdx as u32 }>(b, c)
}
#[cfg(test)]
mod tests {
use super::*;
@ -6059,4 +6075,24 @@ mod tests {
let d = unsafe { vec_cmpnrg(a, b, c) };
assert_eq!(d.as_array(), &[0, !0, !0, 0]);
}
#[simd_test(enable = "vector")]
fn test_vec_cmprg_idx() {
let a = vector_unsigned_int([1, 11, 22, 33]);
let b = vector_unsigned_int([10, 20, 30, 40]);
let c = vector_unsigned_int([GT, LT, GT, LT]);
let d = unsafe { vec_cmprg_idx(a, b, c) };
assert_eq!(d.as_array(), &[0, 4, 0, 0]);
}
#[simd_test(enable = "vector")]
fn test_vec_cmpnrg_idx() {
let a = vector_unsigned_int([1, 11, 22, 33]);
let b = vector_unsigned_int([10, 20, 30, 40]);
let c = vector_unsigned_int([GT, LT, GT, LT]);
let d = unsafe { vec_cmpnrg_idx(a, b, c) };
assert_eq!(d.as_array(), &[0, 0, 0, 0]);
}
}