src/test/codegen/simd-intrinsic/simd-intrinsic-generic-scatter.rs

   1 //
   2
   3 // compile-flags: -C no-prepopulate-passes
   4
   5 #![crate_type = "lib"]
   6
   7 #![feature(repr_simd, platform_intrinsics)]
   8 #![allow(non_camel_case_types)]
   9
  10 #[repr(simd)]
  11 #[derive(Copy, Clone, PartialEq, Debug)]
  12 pub struct Vec2<T>(pub T, pub T);
  13
  14 #[repr(simd)]
  15 #[derive(Copy, Clone, PartialEq, Debug)]
  16 pub struct Vec4<T>(pub T, pub T, pub T, pub T);
  17
  18 extern "platform-intrinsic" {
  19     fn simd_scatter<T, P, M>(value: T, pointers: P, mask: M);
  20 }
  21
  22 // CHECK-LABEL: @scatter_f32x2
  23 #[no_mangle]
  24 pub unsafe fn scatter_f32x2(pointers: Vec2<*mut f32>, mask: Vec2<i32>,
  25                             values: Vec2<f32>) {
  26     // CHECK: call void @llvm.masked.scatter.v2f32.v2p0{{.*}}(<2 x float> {{.*}}, <2 x {{float\*|ptr}}> {{.*}}, i32 {{.*}}, <2 x i1> {{.*}})
  27     simd_scatter(values, pointers, mask)
  28 }
  29
  30
  31 // CHECK-LABEL: @scatter_pf32x2
  32 #[no_mangle]
  33 pub unsafe fn scatter_pf32x2(pointers: Vec2<*mut *const f32>, mask: Vec2<i32>,
  34                              values: Vec2<*const f32>) {
  35     // CHECK: call void @llvm.masked.scatter.v2p0{{.*}}.v2p0{{.*}}(<2 x {{float\*|ptr}}> {{.*}}, <2 x {{float\*\*|ptr}}> {{.*}}, i32 {{.*}}, <2 x i1> {{.*}})
  36     simd_scatter(values, pointers, mask)
  37 }