1//! `i686`'s Streaming SIMD Extensions 4.1 (SSE4.1)
2
3use crate::{core_arch::x86::*, mem::transmute};
4
5#[cfg(test)]
6use stdarch_test::assert_instr;
7
8/// Extracts an 64-bit integer from `a` selected with `IMM1`
9///
10/// [Intel's documentation](https://www.intel.com/content/www/us/en/docs/intrinsics-guide/index.html#text=_mm_extract_epi64)
11#[inline]
12#[target_feature(enable = "sse4.1")]
13#[cfg_attr(test, assert_instr(pextrq, IMM1 = 1))]
14#[rustc_legacy_const_generics(1)]
15#[stable(feature = "simd_x86", since = "1.27.0")]
16#[rustc_const_unstable(feature = "stdarch_const_x86", issue = "149298")]
17pub const fn _mm_extract_epi64<const IMM1: i32>(a: __m128i) -> i64 {
18 static_assert_uimm_bits!(IMM1, 1);
19 unsafe { simd_extract!(a.as_i64x2(), IMM1 as u32) }
20}
21
22/// Returns a copy of `a` with the 64-bit integer from `i` inserted at a
23/// location specified by `IMM1`.
24///
25/// [Intel's documentation](https://www.intel.com/content/www/us/en/docs/intrinsics-guide/index.html#text=_mm_insert_epi64)
26#[inline]
27#[target_feature(enable = "sse4.1")]
28#[cfg_attr(test, assert_instr(pinsrq, IMM1 = 0))]
29#[rustc_legacy_const_generics(2)]
30#[stable(feature = "simd_x86", since = "1.27.0")]
31#[rustc_const_unstable(feature = "stdarch_const_x86", issue = "149298")]
32pub const fn _mm_insert_epi64<const IMM1: i32>(a: __m128i, i: i64) -> __m128i {
33 static_assert_uimm_bits!(IMM1, 1);
34 unsafe { transmute(src:simd_insert!(a.as_i64x2(), IMM1 as u32, i)) }
35}
36
37#[cfg(test)]
38mod tests {
39 use crate::core_arch::arch::x86_64::*;
40 use crate::core_arch::assert_eq_const as assert_eq;
41 use stdarch_test::simd_test;
42
43 #[simd_test(enable = "sse4.1")]
44 const fn test_mm_extract_epi64() {
45 let a = _mm_setr_epi64x(0, 1);
46 let r = _mm_extract_epi64::<1>(a);
47 assert_eq!(r, 1);
48 let r = _mm_extract_epi64::<0>(a);
49 assert_eq!(r, 0);
50 }
51
52 #[simd_test(enable = "sse4.1")]
53 const fn test_mm_insert_epi64() {
54 let a = _mm_set1_epi64x(0);
55 let e = _mm_setr_epi64x(0, 32);
56 let r = _mm_insert_epi64::<1>(a, 32);
57 assert_eq_m128i(r, e);
58 let e = _mm_setr_epi64x(32, 0);
59 let r = _mm_insert_epi64::<0>(a, 32);
60 assert_eq_m128i(r, e);
61 }
62}
63