tests/codegen/simd-intrinsic/simd-intrinsic-generic-gather.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_gather<T, P, M>(value: T, pointers: P, mask: M) -> T;
  20 }
  21
  22 // CHECK-LABEL: @gather_f32x2
  23 #[no_mangle]
  24 pub unsafe fn gather_f32x2(pointers: Vec2<*const f32>, mask: Vec2<i32>,
  25                            values: Vec2<f32>) -> Vec2<f32> {
  26     // CHECK: call <2 x float> @llvm.masked.gather.v2f32.{{.+}}(<2 x {{float\*|ptr}}> {{.*}}, i32 {{.*}}, <2 x i1> {{.*}}, <2 x float> {{.*}})
  27     simd_gather(values, pointers, mask)
  28 }
  29
  30 // CHECK-LABEL: @gather_pf32x2
  31 #[no_mangle]
  32 pub unsafe fn gather_pf32x2(pointers: Vec2<*const *const f32>, mask: Vec2<i32>,
  33                            values: Vec2<*const f32>) -> Vec2<*const f32> {
  34     // CHECK: call <2 x {{float\*|ptr}}> @llvm.masked.gather.{{.+}}(<2 x {{float\*\*|ptr}}> {{.*}}, i32 {{.*}}, <2 x i1> {{.*}}, <2 x {{float\*|ptr}}> {{.*}})
  35     simd_gather(values, pointers, mask)
  36 }