[sse] _mm_stream_{ps,pi} (#219)

Author: gnzlbg

coresimd/src/lib.rs

@@ -13,7 +13,7 @@
 #![allow(unused_features)]
 #![feature(const_fn, link_llvm_intrinsics, platform_intrinsics, repr_simd,
            simd_ffi, target_feature, cfg_target_feature, i128_type, asm,
-           const_atomic_usize_new, stmt_expr_attributes)]
+           const_atomic_usize_new, stmt_expr_attributes, core_intrinsics)]
 #![cfg_attr(test, feature(proc_macro, test, repr_align, attr_literals))]
 #![cfg_attr(feature = "cargo-clippy",
             allow(inline_always, too_many_arguments, cast_sign_loss,

coresimd/src/x86/i586/sse.rs

@@ -1680,6 +1680,30 @@ extern "C" {
     fn cmpss(a: f32x4, b: f32x4, imm8: i8) -> f32x4;
 }
 
+/// Stores `a` into the memory at `mem_addr` using a non-temporal memory hint.
+///
+/// `mem_addr` must be aligned on a 16-byte boundary or a general-protection
+/// exception _may_ be generated.
+#[inline(always)]
+#[target_feature = "+sse"]
+#[cfg_attr(test, assert_instr(movntps))]
+pub unsafe fn _mm_stream_ps(mem_addr: *mut f32, a: f32x4) {
+    ::core::intrinsics::nontemporal_store(mem::transmute(mem_addr), a);
+}
+
+/// Store 64-bits of integer data from a into memory using a non-temporal
+/// memory hint.
+#[inline(always)]
+#[target_feature = "+sse"]
+// generates movnti on i686 and x86_64 but just a mov on i586
+#[cfg_attr(all(test,
+               any(target_arch = "x86_64",
+                   all(target_arch = "x86", target_feature = "sse2"))),
+           assert_instr(movnti))]
+pub unsafe fn _mm_stream_pi(mem_addr: *mut i64, a: i64) {
+    ::core::intrinsics::nontemporal_store(mem_addr, a);
+}
+
 #[cfg(test)]
 mod tests {
     use v128::*;
@@ -3261,4 +3285,28 @@ mod tests {
         assert_eq!(c, f32x4::new(3.0, 7.0, 11.0, 15.0));
         assert_eq!(d, f32x4::new(4.0, 8.0, 12.0, 16.0));
     }
+
+    #[repr(align(16))]
+    struct Memory {
+        pub data: [f32; 4],
+    }
+
+    #[simd_test = "sse"]
+    unsafe fn _mm_stream_ps() {
+        let a = f32x4::splat(7.0);
+        let mut mem = Memory { data: [-1.0; 4] };
+
+        sse::_mm_stream_ps(&mut mem.data[0] as *mut f32, a);
+        for i in 0..4 {
+            assert_eq!(mem.data[i], a.extract(i as u32));
+        }
+    }
+
+    #[simd_test = "sse"]
+    unsafe fn _mm_stream_pi() {
+        let a: i64 = 7;
+        let mut mem = ::std::boxed::Box::<i64>::new(-1);
+        sse::_mm_stream_pi(&mut *mem as *mut i64, a);
+        assert_eq!(a, *mem);
+    }
 }