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

convert _mm_insert_epi64 to const generics

Browse source
This commit is contained in:
Rémy Rakic 2021-03-04 21:15:59 +01:00 • committed by Amanieu d'Antras
parent 6d8acf67de
commit 318f4ef207
3 changed files with 14 additions and 14 deletions
Download patch file Download diff file Expand all files Collapse all files

2
library/stdarch/crates/core_arch/src/x86/avx2.rs
View file

@ -5184,7 +5184,7 @@ mod tests {
#[simd_test(enable = "avx2")]
unsafe fn test_mm256_sll_epi64() {
let a = _mm256_set1_epi64x(0xFFFFFFFF);
let b = _mm_insert_epi64(_mm_set1_epi64x(0), 4, 0);
let b = _mm_insert_epi64::<0>(_mm_set1_epi64x(0), 4);
let r = _mm256_sll_epi64(a, b);
assert_eq_m256i(r, _mm256_set1_epi64x(0xFFFFFFFF0));
}

6
library/stdarch/crates/core_arch/src/x86/test.rs
View file

@ -92,14 +92,16 @@ pub unsafe fn get_m512i(a: __m512i, idx: usize) -> i64 {
mod x86_polyfill {
use crate::core_arch::x86::*;
pub unsafe fn _mm_insert_epi64(a: __m128i, val: i64, idx: i32) -> __m128i {
#[rustc_legacy_const_generics(2)]
pub unsafe fn _mm_insert_epi64<const INDEX: i32>(a: __m128i, val: i64) -> __m128i {
static_assert_imm1!(INDEX);
#[repr(C)]
union A {
a: __m128i,
b: [i64; 2],
}
let mut a = A { a };
a.b[idx as usize] = val;
a.b[INDEX as usize] = val;
a.a
}

20
library/stdarch/crates/core_arch/src/x86_64/sse41.rs
View file

@ -22,20 +22,17 @@ pub unsafe fn _mm_extract_epi64<const IMM1: i32>(a: __m128i) -> i64 {
}
/// Returns a copy of `a` with the 64-bit integer from `i` inserted at a
/// location specified by `imm8`.
/// location specified by `IMM1`.
///
/// [Intel's documentation](https://software.intel.com/sites/landingpage/IntrinsicsGuide/#text=_mm_insert_epi64)
#[inline]
#[target_feature(enable = "sse4.1")]
#[cfg_attr(test, assert_instr(pinsrq, imm8 = 0))]
#[rustc_args_required_const(2)]
#[cfg_attr(test, assert_instr(pinsrq, IMM1 = 0))]
#[rustc_legacy_const_generics(2)]
#[stable(feature = "simd_x86", since = "1.27.0")]
pub unsafe fn _mm_insert_epi64(a: __m128i, i: i64, imm8: i32) -> __m128i {
let a = a.as_i64x2();
match imm8 & 1 {
0 => transmute(simd_insert(a, 0, i)),
_ => transmute(simd_insert(a, 1, i)),
}
pub unsafe fn _mm_insert_epi64<const IMM1: i32>(a: __m128i, i: i64) -> __m128i {
static_assert_imm1!(IMM1);
transmute(simd_insert(a.as_i64x2(), IMM1 as u32, i))
}
#[cfg(test)]
@ -56,9 +53,10 @@ mod tests {
unsafe fn test_mm_insert_epi64() {
let a = _mm_set1_epi64x(0);
let e = _mm_setr_epi64x(0, 32);
let r = _mm_insert_epi64(a, 32, 1);
let r = _mm_insert_epi64::<1>(a, 32);
assert_eq_m128i(r, e);
let r = _mm_insert_epi64(a, 32, 3);
let e = _mm_setr_epi64x(32, 0);
let r = _mm_insert_epi64::<0>(a, 32);
assert_eq_m128i(r, e);
}
}