user0/rust
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

add vec_rl_mask

Browse source
This commit is contained in:
Folkert de Vries 2025-02-19 16:35:33 +01:00 • committed by Amanieu d'Antras
parent dc0a5c88fb
commit 4341a8b8b3
2 changed files with 85 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

36
library/stdarch/crates/core_arch/src/s390x/macros.rs
View file

@ -213,6 +213,41 @@ macro_rules! s_t_l {
};
}
macro_rules! l_t_t {
(vector_signed_long_long) => {
i64
};
(vector_signed_int) => {
i32
};
(vector_signed_short) => {
i16
};
(vector_signed_char) => {
i8
};
(vector_unsigned_long_long ) => {
u64
};
(vector_unsigned_int ) => {
u32
};
(vector_unsigned_short ) => {
u16
};
(vector_unsigned_char ) => {
u8
};
(vector_float) => {
f32
};
(vector_double) => {
f64
};
}
macro_rules! t_t_l {
(i64) => {
vector_signed_long_long
@ -401,6 +436,7 @@ macro_rules! impl_neg {
pub(crate) use impl_from;
pub(crate) use impl_neg;
pub(crate) use impl_vec_trait;
pub(crate) use l_t_t;
pub(crate) use s_t_l;
pub(crate) use t_b;
pub(crate) use t_t_l;

49
library/stdarch/crates/core_arch/src/s390x/vector.rs
View file

@ -95,6 +95,11 @@ unsafe extern "unadjusted" {
#[link_name = "llvm.fshl.v8i16"] fn fshlh(a: vector_unsigned_short, b: vector_unsigned_short, c: vector_unsigned_short) -> vector_unsigned_short;
#[link_name = "llvm.fshl.v4i32"] fn fshlf(a: vector_unsigned_int, b: vector_unsigned_int, c: vector_unsigned_int) -> vector_unsigned_int;
#[link_name = "llvm.fshl.v2i64"] fn fshlg(a: vector_unsigned_long_long, b: vector_unsigned_long_long, c: vector_unsigned_long_long) -> vector_unsigned_long_long;
#[link_name = "llvm.s390.verimb"] fn verimb(a: vector_signed_char, b: vector_signed_char, c: vector_signed_char, d: i32) -> vector_signed_char;
#[link_name = "llvm.s390.verimh"] fn verimh(a: vector_signed_short, b: vector_signed_short, c: vector_signed_short, d: i32) -> vector_signed_short;
#[link_name = "llvm.s390.verimf"] fn verimf(a: vector_signed_int, b: vector_signed_int, c: vector_signed_int, d: i32) -> vector_signed_int;
#[link_name = "llvm.s390.verimg"] fn verimg(a: vector_signed_long_long, b: vector_signed_long_long, c: vector_signed_long_long, d: i32) -> vector_signed_long_long;
}
impl_from! { i8x16, u8x16, i16x8, u16x8, i32x4, u32x4, i64x2, u64x2, f32x4, f64x2 }
@ -948,6 +953,50 @@ mod sealed {
vector_signed_int, verllvf_imm,
vector_signed_long_long, verllvg_imm
}
#[unstable(feature = "stdarch_s390x", issue = "135681")]
pub trait VectorRlMask<Other> {
unsafe fn vec_rl_mask<const IMM8: u8>(self, other: Other) -> Self;
}
macro_rules! impl_rl_mask {
($($ty:ident, $intr:ident, $fun:ident),*) => {
$(
#[inline]
#[target_feature(enable = "vector")]
#[cfg_attr(test, assert_instr($intr, IMM8 = 6))]
unsafe fn $fun<const IMM8: u8>(a: $ty, b: t_u!($ty)) -> $ty {
// mod by the number of bits in a's element type to prevent UB
$intr(a, a, transmute(b), const { (IMM8 % <l_t_t!($ty)>::BITS as u8) as i32 })
}
#[unstable(feature = "stdarch_s390x", issue = "135681")]
impl VectorRlMask<t_u!($ty)> for $ty {
#[inline]
#[target_feature(enable = "vector")]
unsafe fn vec_rl_mask<const IMM8: u8>(self, other: t_u!($ty)) -> Self {
$fun::<IMM8>(self, other)
}
}
#[unstable(feature = "stdarch_s390x", issue = "135681")]
impl VectorRlMask<t_u!($ty)> for t_u!($ty) {
#[inline]
#[target_feature(enable = "vector")]
unsafe fn vec_rl_mask<const IMM8: u8>(self, other: t_u!($ty)) -> Self {
transmute($fun::<IMM8>(transmute(self), transmute(other)))
}
}
)*
}
}
impl_rl_mask! {
vector_signed_char, verimb, test_verimb,
vector_signed_short, verimh, test_verimh,
vector_signed_int, verimf, test_verimf,
vector_signed_long_long, verimg, test_verimg
}
}
/// Vector element-wise addition.