sse41.rs.html -- source
Description: Source to the Rust file `src\libcore\..\stdarch\crates\core_arch\src\x86_64\sse41.rs`.
Keywords: rust, rustlang, rust-lang

     1
     2
     3
     4
     5
     6
     7
     8
     9
    10
    11
    12
    13
    14
    15
    16
    17
    18
    19
    20
    21
    22
    23
    24
    25
    26
    27
    28
    29
    30
    31
    32
    33
    34
    35
    36
    37
    38
    39
    40
    41
    42
    43
    44
    45
    46
    47
    48
    49
    50
    51
    52
    53
    54
    55
    56
    57
    58
    59
    60

    //! `i686`'s Streaming SIMD Extensions 4.1 (SSE4.1)

    use crate::{
        core_arch::{simd_llvm::*, x86::*},
        mem::transmute,
    };

    #[cfg(test)]
    use stdarch_test::assert_instr;

    /// Extracts an 64-bit integer from `a` selected with `imm8`
    ///
    /// [Intel's documentation](https://software.intel.com/sites/landingpage/IntrinsicsGuide/#text=_mm_extract_epi64)
    #[inline]
    #[target_feature(enable = "sse4.1")]
    #[cfg_attr(all(test, not(target_os = "windows")), assert_instr(pextrq, imm8 = 1))]
    #[rustc_args_required_const(1)]
    #[stable(feature = "simd_x86", since = "1.27.0")]
    pub unsafe fn _mm_extract_epi64(a: __m128i, imm8: i32) -> i64 {
        let imm8 = (imm8 & 1) as u32;
        simd_extract(a.as_i64x2(), imm8)
    }

    /// Returns a copy of `a` with the 64-bit integer from `i` inserted at a
    /// location specified by `imm8`.
    ///
    /// [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)]
    #[stable(feature = "simd_x86", since = "1.27.0")]
    pub unsafe fn _mm_insert_epi64(a: __m128i, i: i64, imm8: i32) -> __m128i {
        transmute(simd_insert(a.as_i64x2(), (imm8 & 1) as u32, i))
    }

    #[cfg(test)]
    mod tests {
        use crate::core_arch::arch::x86_64::*;
        use stdarch_test::simd_test;

        #[simd_test(enable = "sse4.1")]
        unsafe fn test_mm_extract_epi64() {
            let a = _mm_setr_epi64x(0, 1);
            let r = _mm_extract_epi64(a, 1);
            assert_eq!(r, 1);
            let r = _mm_extract_epi64(a, 3);
            assert_eq!(r, 1);
        }

        #[simd_test(enable = "sse4.1")]
        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);
            assert_eq_m128i(r, e);
            let r = _mm_insert_epi64(a, 32, 3);
            assert_eq_m128i(r, e);
        }
    }