79 lines
2.1 KiB
Rust
79 lines
2.1 KiB
Rust
// compile-flags: -O -Z merge-functions=disabled
|
|
// only-x86_64
|
|
// ignore-debug: the debug assertions get in the way
|
|
|
|
#![crate_type = "lib"]
|
|
|
|
use std::mem::swap;
|
|
|
|
type RGB48 = [u16; 3];
|
|
|
|
// CHECK-LABEL: @swap_rgb48_manually(
|
|
#[no_mangle]
|
|
pub fn swap_rgb48_manually(x: &mut RGB48, y: &mut RGB48) {
|
|
// CHECK-NOT: alloca
|
|
// CHECK: %[[TEMP0:.+]] = load <3 x i16>, ptr %x, align 2
|
|
// CHECK: %[[TEMP1:.+]] = load <3 x i16>, ptr %y, align 2
|
|
// CHECK: store <3 x i16> %[[TEMP1]], ptr %x, align 2
|
|
// CHECK: store <3 x i16> %[[TEMP0]], ptr %y, align 2
|
|
|
|
let temp = *x;
|
|
*x = *y;
|
|
*y = temp;
|
|
}
|
|
|
|
// CHECK-LABEL: @swap_rgb48
|
|
#[no_mangle]
|
|
pub fn swap_rgb48(x: &mut RGB48, y: &mut RGB48) {
|
|
// CHECK-NOT: alloca
|
|
// CHECK: load <3 x i16>
|
|
// CHECK: load <3 x i16>
|
|
// CHECK: store <3 x i16>
|
|
// CHECK: store <3 x i16>
|
|
swap(x, y)
|
|
}
|
|
|
|
// LLVM doesn't vectorize a loop over 3-byte elements,
|
|
// so we chunk it down to bytes and loop over those instead.
|
|
type RGB24 = [u8; 3];
|
|
|
|
// CHECK-LABEL: @swap_rgb24_slices
|
|
#[no_mangle]
|
|
pub fn swap_rgb24_slices(x: &mut [RGB24], y: &mut [RGB24]) {
|
|
// CHECK-NOT: alloca
|
|
// CHECK: load <{{[0-9]+}} x i8>
|
|
// CHECK: store <{{[0-9]+}} x i8>
|
|
if x.len() == y.len() {
|
|
x.swap_with_slice(y);
|
|
}
|
|
}
|
|
|
|
// This one has a power-of-two size, so we iterate over it directly
|
|
type RGBA32 = [u8; 4];
|
|
|
|
// CHECK-LABEL: @swap_rgba32_slices
|
|
#[no_mangle]
|
|
pub fn swap_rgba32_slices(x: &mut [RGBA32], y: &mut [RGBA32]) {
|
|
// CHECK-NOT: alloca
|
|
// CHECK: load <{{[0-9]+}} x i32>
|
|
// CHECK: store <{{[0-9]+}} x i32>
|
|
if x.len() == y.len() {
|
|
x.swap_with_slice(y);
|
|
}
|
|
}
|
|
|
|
// Strings have a non-power-of-two size, but have pointer alignment,
|
|
// so we swap usizes instead of dropping all the way down to bytes.
|
|
const _: () = assert!(!std::mem::size_of::<String>().is_power_of_two());
|
|
|
|
// CHECK-LABEL: @swap_string_slices
|
|
#[no_mangle]
|
|
pub fn swap_string_slices(x: &mut [String], y: &mut [String]) {
|
|
// CHECK-NOT: alloca
|
|
// CHECK: load <{{[0-9]+}} x i64>
|
|
// CHECK: store <{{[0-9]+}} x i64>
|
|
if x.len() == y.len() {
|
|
x.swap_with_slice(y);
|
|
}
|
|
}
|