summaryrefslogtreecommitdiffstats
path: root/tests/codegen/simd-intrinsic/simd-intrinsic-generic-extract-insert.rs
blob: b5b0b1330a629f875ead37fb2785300d777fa248 (plain)
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
// compile-flags: -C no-prepopulate-passes

#![crate_type = "lib"]

#![feature(repr_simd, platform_intrinsics, min_const_generics)]
#![allow(non_camel_case_types)]

#[repr(simd)]
#[derive(Copy, Clone)]
pub struct M(pub f32, pub f32, pub f32, pub f32);

#[repr(simd)]
#[derive(Copy, Clone)]
pub struct S<const N: usize>([f32; N]);

extern "platform-intrinsic" {
    fn simd_extract<T, U>(x: T, idx: u32) -> U;
    fn simd_insert<T, U>(x: T, idx: u32, b: U) -> T;
}

// CHECK-LABEL: @extract_m
#[no_mangle]
pub unsafe fn extract_m(v: M, i: u32) -> f32  {
    // CHECK: extractelement <4 x float> %{{v|_3}}, i32 %i
    simd_extract(v, i)
}

// CHECK-LABEL: @extract_s
#[no_mangle]
pub unsafe fn extract_s(v: S<4>, i: u32) -> f32  {
    // CHECK: extractelement <4 x float> %{{v|_3}}, i32 %i
    simd_extract(v, i)
}

// CHECK-LABEL: @insert_m
#[no_mangle]
pub unsafe fn insert_m(v: M, i: u32, j: f32) -> M  {
    // CHECK: insertelement <4 x float> %{{v|_4}}, float %j, i32 %i
    simd_insert(v, i, j)
}

// CHECK-LABEL: @insert_s
#[no_mangle]
pub unsafe fn insert_s(v: S<4>, i: u32, j: f32) -> S<4>  {
    // CHECK: insertelement <4 x float> %{{v|_4}}, float %j, i32 %i
    simd_insert(v, i, j)
}