48 lines
1.2 KiB
Rust
48 lines
1.2 KiB
Rust
// 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|1|2}}, 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|1|2}}, 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|0|1}}, 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|0|1}}, float %j, i32 %i
|
|
simd_insert(v, i, j)
|
|
}
|