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

Convert _mm_extract_epi16 to const generics

Browse source
This commit is contained in:
Rémy Rakic 2021-03-01 18:53:36 +01:00 • committed by Amanieu d'Antras
parent 2d48c702be
commit 0c2974c04e
1 changed files with 7 additions and 12 deletions
Download patch file Download diff file Expand all files Collapse all files

19
library/stdarch/crates/core_arch/src/x86/sse2.rs
View file

@ -1348,17 +1348,12 @@ pub unsafe fn _mm_packus_epi16(a: __m128i, b: __m128i) -> __m128i {
/// [Intel's documentation](https://software.intel.com/sites/landingpage/IntrinsicsGuide/#text=_mm_extract_epi16)
#[inline]
#[target_feature(enable = "sse2")]
#[cfg_attr(test, assert_instr(pextrw, imm8 = 9))]
#[rustc_args_required_const(1)]
#[cfg_attr(test, assert_instr(pextrw, IMM8 = 7))]
#[rustc_legacy_const_generics(1)]
#[stable(feature = "simd_x86", since = "1.27.0")]
pub unsafe fn _mm_extract_epi16(a: __m128i, imm8: i32) -> i32 {
let a = a.as_u16x8();
macro_rules! call {
($imm3:expr) => {
simd_extract::<_, u16>(a, $imm3) as i32
};
}
constify_imm3!(imm8, call)
pub unsafe fn _mm_extract_epi16<const IMM8: i32>(a: __m128i) -> i32 {
static_assert_imm3!(IMM8);
simd_extract::<_, u16>(a.as_u16x8(), IMM8 as u32) as i32
}
/// Returns a new vector where the `imm8` element of `a` is replaced with `i`.
@ -3884,8 +3879,8 @@ mod tests {
#[simd_test(enable = "sse2")]
unsafe fn test_mm_extract_epi16() {
let a = _mm_setr_epi16(-1, 1, 2, 3, 4, 5, 6, 7);
let r1 = _mm_extract_epi16(a, 0);
let r2 = _mm_extract_epi16(a, 11);
let r1 = _mm_extract_epi16::<0>(a);
let r2 = _mm_extract_epi16::<3>(a);
assert_eq!(r1, 0xFFFF);
assert_eq!(r2, 3);
}