add vec_cmpnrg

This commit is contained in:
Folkert de Vries 2025-03-06 16:32:16 +01:00 committed by Amanieu d'Antras
parent edbeee5b25
commit 511a8eb3cd

View file

@ -4720,6 +4720,7 @@ pub unsafe fn vec_srdb<T: sealed::VectorSrdb, const C: u32>(a: T, b: T) -> T {
a.vec_srdb::<C>(b)
}
/// Vector Compare Ranges
#[inline]
#[target_feature(enable = "vector")]
#[unstable(feature = "stdarch_s390x", issue = "135681")]
@ -4727,6 +4728,14 @@ pub unsafe fn vec_cmprg<T: sealed::VectorCompareRange>(a: T, b: T, c: T) -> T::R
a.vstrc::<{ FindImm::Eq as u32 }>(b, c)
}
/// Vector Compare Not in Ranges
#[inline]
#[target_feature(enable = "vector")]
#[unstable(feature = "stdarch_s390x", issue = "135681")]
pub unsafe fn vec_cmpnrg<T: sealed::VectorCompareRange>(a: T, b: T, c: T) -> T::Result {
a.vstrc::<{ FindImm::Ne as u32 }>(b, c)
}
#[cfg(test)]
mod tests {
use super::*;
@ -6005,12 +6014,12 @@ mod tests {
}
}
const GT: u32 = 0x20000000;
const LT: u32 = 0x40000000;
const EQ: u32 = 0x80000000;
#[simd_test(enable = "vector")]
fn test_vec_cmprg() {
const GT: u32 = 0x20000000;
const LT: u32 = 0x40000000;
const EQ: u32 = 0x80000000;
let a = vector_unsigned_int([11, 22, 33, 44]);
let b = vector_unsigned_int([10, 20, 30, 40]);
@ -6029,4 +6038,25 @@ mod tests {
let d = unsafe { vec_cmprg(a, b, c) };
assert_eq!(d.as_array(), &[!0, 0, 0, !0]);
}
#[simd_test(enable = "vector")]
fn test_vec_cmpnrg() {
let a = vector_unsigned_int([11, 22, 33, 44]);
let b = vector_unsigned_int([10, 20, 30, 40]);
let c = vector_unsigned_int([GT, LT, GT, LT]);
let d = unsafe { vec_cmpnrg(a, b, c) };
assert_eq!(d.as_array(), &[0, !0, 0, !0]);
let c = vector_unsigned_int([GT, LT, 0, 0]);
let d = unsafe { vec_cmpnrg(a, b, c) };
assert_eq!(d.as_array(), &[0, !0, !0, !0]);
let a = vector_unsigned_int([11, 22, 33, 30]);
let b = vector_unsigned_int([10, 20, 30, 30]);
let c = vector_unsigned_int([GT, LT, EQ, EQ]);
let d = unsafe { vec_cmpnrg(a, b, c) };
assert_eq!(d.as_array(), &[0, !0, !0, 0]);
}
}