llvm-project/llvm/test/CodeGen/AArch64/sve-streaming-mode-fixed-le...

342 lines
7.6 KiB
LLVM

; NOTE: Assertions have been autogenerated by utils/update_llc_test_checks.py
; RUN: llc -force-streaming-compatible-sve < %s | FileCheck %s
; Test we can code generater patterns of the form:
; fixed_length_vector = ISD::EXTRACT_SUBVECTOR scalable_vector, 0
; scalable_vector = ISD::INSERT_SUBVECTOR scalable_vector, fixed_length_vector, 0
;
; NOTE: Currently shufflevector does not support scalable vectors so it cannot
; be used to model the above operations. Instead these tests rely on knowing
; how fixed length operation are lowered to scalable ones, with multiple blocks
; ensuring insert/extract sequences are not folded away.
target datalayout = "e-m:o-i64:64-i128:128-n32:64-S128"
target triple = "aarch64-unknown-linux-gnu"
; i8
define void @subvector_v4i8(ptr %in, ptr %out) #0 {
; CHECK-LABEL: subvector_v4i8:
; CHECK: // %bb.0: // %bb1
; CHECK-NEXT: ldr s0, [x0]
; CHECK-NEXT: ptrue p0.h, vl4
; CHECK-NEXT: uunpklo z0.h, z0.b
; CHECK-NEXT: st1b { z0.h }, p0, [x1]
; CHECK-NEXT: ret
%a = load <4 x i8>, ptr %in
br label %bb1
bb1:
store <4 x i8> %a, ptr %out
ret void
}
define void @subvector_v8i8(ptr %in, ptr %out) #0 {
; CHECK-LABEL: subvector_v8i8:
; CHECK: // %bb.0: // %bb1
; CHECK-NEXT: ldr d0, [x0]
; CHECK-NEXT: str d0, [x1]
; CHECK-NEXT: ret
%a = load <8 x i8>, ptr %in
br label %bb1
bb1:
store <8 x i8> %a, ptr %out
ret void
}
define void @subvector_v16i8(ptr %in, ptr %out) #0 {
; CHECK-LABEL: subvector_v16i8:
; CHECK: // %bb.0: // %bb1
; CHECK-NEXT: ldr q0, [x0]
; CHECK-NEXT: str q0, [x1]
; CHECK-NEXT: ret
%a = load <16 x i8>, ptr %in
br label %bb1
bb1:
store <16 x i8> %a, ptr %out
ret void
}
define void @subvector_v32i8(ptr %in, ptr %out) #0 {
; CHECK-LABEL: subvector_v32i8:
; CHECK: // %bb.0: // %bb1
; CHECK-NEXT: ldp q0, q1, [x0]
; CHECK-NEXT: stp q0, q1, [x1]
; CHECK-NEXT: ret
%a = load <32 x i8>, ptr %in
br label %bb1
bb1:
store <32 x i8> %a, ptr %out
ret void
}
; i16
define void @subvector_v2i16(ptr %in, ptr %out) #0 {
; CHECK-LABEL: subvector_v2i16:
; CHECK: // %bb.0: // %bb1
; CHECK-NEXT: sub sp, sp, #16
; CHECK-NEXT: .cfi_def_cfa_offset 16
; CHECK-NEXT: ldrh w8, [x0, #2]
; CHECK-NEXT: ptrue p0.s, vl2
; CHECK-NEXT: str w8, [sp, #12]
; CHECK-NEXT: ldrh w8, [x0]
; CHECK-NEXT: str w8, [sp, #8]
; CHECK-NEXT: ldr d0, [sp, #8]
; CHECK-NEXT: st1h { z0.s }, p0, [x1]
; CHECK-NEXT: add sp, sp, #16
; CHECK-NEXT: ret
%a = load <2 x i16>, ptr %in
br label %bb1
bb1:
store <2 x i16> %a, ptr %out
ret void
}
define void @subvector_v4i16(ptr %in, ptr %out) #0 {
; CHECK-LABEL: subvector_v4i16:
; CHECK: // %bb.0: // %bb1
; CHECK-NEXT: ldr d0, [x0]
; CHECK-NEXT: str d0, [x1]
; CHECK-NEXT: ret
%a = load <4 x i16>, ptr %in
br label %bb1
bb1:
store <4 x i16> %a, ptr %out
ret void
}
define void @subvector_v8i16(ptr %in, ptr %out) #0 {
; CHECK-LABEL: subvector_v8i16:
; CHECK: // %bb.0: // %bb1
; CHECK-NEXT: ldr q0, [x0]
; CHECK-NEXT: str q0, [x1]
; CHECK-NEXT: ret
%a = load <8 x i16>, ptr %in
br label %bb1
bb1:
store <8 x i16> %a, ptr %out
ret void
}
define void @subvector_v16i16(ptr %in, ptr %out) #0 {
; CHECK-LABEL: subvector_v16i16:
; CHECK: // %bb.0: // %bb1
; CHECK-NEXT: ldp q0, q1, [x0]
; CHECK-NEXT: stp q0, q1, [x1]
; CHECK-NEXT: ret
%a = load <16 x i16>, ptr %in
br label %bb1
bb1:
store <16 x i16> %a, ptr %out
ret void
}
; i32
define void @subvector_v2i32(ptr %in, ptr %out) #0 {
; CHECK-LABEL: subvector_v2i32:
; CHECK: // %bb.0: // %bb1
; CHECK-NEXT: ldr d0, [x0]
; CHECK-NEXT: str d0, [x1]
; CHECK-NEXT: ret
%a = load <2 x i32>, ptr %in
br label %bb1
bb1:
store <2 x i32> %a, ptr %out
ret void
}
define void @subvector_v4i32(ptr %in, ptr %out) #0 {
; CHECK-LABEL: subvector_v4i32:
; CHECK: // %bb.0: // %bb1
; CHECK-NEXT: ldr q0, [x0]
; CHECK-NEXT: str q0, [x1]
; CHECK-NEXT: ret
%a = load <4 x i32>, ptr %in
br label %bb1
bb1:
store <4 x i32> %a, ptr %out
ret void
}
define void @subvector_v8i32(ptr %in, ptr %out) #0 {
; CHECK-LABEL: subvector_v8i32:
; CHECK: // %bb.0: // %bb1
; CHECK-NEXT: ldp q0, q1, [x0]
; CHECK-NEXT: stp q0, q1, [x1]
; CHECK-NEXT: ret
%a = load <8 x i32>, ptr %in
br label %bb1
bb1:
store <8 x i32> %a, ptr %out
ret void
}
; i64
define void @subvector_v2i64(ptr %in, ptr %out) #0 {
; CHECK-LABEL: subvector_v2i64:
; CHECK: // %bb.0: // %bb1
; CHECK-NEXT: ldr q0, [x0]
; CHECK-NEXT: str q0, [x1]
; CHECK-NEXT: ret
%a = load <2 x i64>, ptr %in
br label %bb1
bb1:
store <2 x i64> %a, ptr %out
ret void
}
define void @subvector_v4i64(ptr %in, ptr %out) #0 {
; CHECK-LABEL: subvector_v4i64:
; CHECK: // %bb.0: // %bb1
; CHECK-NEXT: ldp q0, q1, [x0]
; CHECK-NEXT: stp q0, q1, [x1]
; CHECK-NEXT: ret
%a = load <4 x i64>, ptr %in
br label %bb1
bb1:
store <4 x i64> %a, ptr %out
ret void
}
; f16
define void @subvector_v2f16(ptr %in, ptr %out) #0 {
; CHECK-LABEL: subvector_v2f16:
; CHECK: // %bb.0: // %bb1
; CHECK-NEXT: ldr w8, [x0]
; CHECK-NEXT: str w8, [x1]
; CHECK-NEXT: ret
%a = load <2 x half>, ptr %in
br label %bb1
bb1:
store <2 x half> %a, ptr %out
ret void
}
define void @subvector_v4f16(ptr %in, ptr %out) #0 {
; CHECK-LABEL: subvector_v4f16:
; CHECK: // %bb.0: // %bb1
; CHECK-NEXT: ldr d0, [x0]
; CHECK-NEXT: str d0, [x1]
; CHECK-NEXT: ret
%a = load <4 x half>, ptr %in
br label %bb1
bb1:
store <4 x half> %a, ptr %out
ret void
}
define void @subvector_v8f16(ptr %in, ptr %out) #0 {
; CHECK-LABEL: subvector_v8f16:
; CHECK: // %bb.0: // %bb1
; CHECK-NEXT: ldr q0, [x0]
; CHECK-NEXT: str q0, [x1]
; CHECK-NEXT: ret
%a = load <8 x half>, ptr %in
br label %bb1
bb1:
store <8 x half> %a, ptr %out
ret void
}
define void @subvector_v16f16(ptr %in, ptr %out) #0 {
; CHECK-LABEL: subvector_v16f16:
; CHECK: // %bb.0: // %bb1
; CHECK-NEXT: ldp q0, q1, [x0]
; CHECK-NEXT: stp q0, q1, [x1]
; CHECK-NEXT: ret
%a = load <16 x half>, ptr %in
br label %bb1
bb1:
store <16 x half> %a, ptr %out
ret void
}
; f32
define void @subvector_v2f32(ptr %in, ptr %out) #0 {
; CHECK-LABEL: subvector_v2f32:
; CHECK: // %bb.0: // %bb1
; CHECK-NEXT: ldr d0, [x0]
; CHECK-NEXT: str d0, [x1]
; CHECK-NEXT: ret
%a = load <2 x float>, ptr %in
br label %bb1
bb1:
store <2 x float> %a, ptr %out
ret void
}
define void @subvector_v4f32(ptr %in, ptr %out) #0 {
; CHECK-LABEL: subvector_v4f32:
; CHECK: // %bb.0: // %bb1
; CHECK-NEXT: ldr q0, [x0]
; CHECK-NEXT: str q0, [x1]
; CHECK-NEXT: ret
%a = load <4 x float>, ptr %in
br label %bb1
bb1:
store <4 x float> %a, ptr %out
ret void
}
define void @subvector_v8f32(ptr %in, ptr %out) #0 {
; CHECK-LABEL: subvector_v8f32:
; CHECK: // %bb.0: // %bb1
; CHECK-NEXT: ldp q0, q1, [x0]
; CHECK-NEXT: stp q0, q1, [x1]
; CHECK-NEXT: ret
%a = load <8 x float>,ptr %in
br label %bb1
bb1:
store <8 x float> %a, ptr %out
ret void
}
; f64
define void @subvector_v2f64(ptr %in, ptr %out) #0 {
; CHECK-LABEL: subvector_v2f64:
; CHECK: // %bb.0: // %bb1
; CHECK-NEXT: ldr q0, [x0]
; CHECK-NEXT: str q0, [x1]
; CHECK-NEXT: ret
%a = load <2 x double>, ptr %in
br label %bb1
bb1:
store <2 x double> %a, ptr %out
ret void
}
define void @subvector_v4f64(ptr %in, ptr %out) #0 {
; CHECK-LABEL: subvector_v4f64:
; CHECK: // %bb.0: // %bb1
; CHECK-NEXT: ldp q0, q1, [x0]
; CHECK-NEXT: stp q0, q1, [x1]
; CHECK-NEXT: ret
%a = load <4 x double>, ptr %in
br label %bb1
bb1:
store <4 x double> %a, ptr %out
ret void
}
attributes #0 = { "target-features"="+sve" }