| ; NOTE: Assertions have been autogenerated by utils/update_llc_test_checks.py |
| ; RUN: llc -mattr=+sve -force-streaming-compatible-sve < %s | FileCheck %s |
| |
| target triple = "aarch64-unknown-linux-gnu" |
| |
| declare void @def(ptr) |
| |
| define void @alloc_v4i8(ptr %st_ptr) nounwind { |
| ; CHECK-LABEL: alloc_v4i8: |
| ; CHECK: // %bb.0: |
| ; CHECK-NEXT: sub sp, sp, #48 |
| ; CHECK-NEXT: stp x20, x19, [sp, #32] // 16-byte Folded Spill |
| ; CHECK-NEXT: mov x19, x0 |
| ; CHECK-NEXT: add x0, sp, #28 |
| ; CHECK-NEXT: str x30, [sp, #16] // 8-byte Folded Spill |
| ; CHECK-NEXT: add x20, sp, #28 |
| ; CHECK-NEXT: bl def |
| ; CHECK-NEXT: ptrue p0.b, vl2 |
| ; CHECK-NEXT: ld2b { z0.b, z1.b }, p0/z, [x20] |
| ; CHECK-NEXT: ptrue p0.s, vl2 |
| ; CHECK-NEXT: ldr x30, [sp, #16] // 8-byte Folded Reload |
| ; CHECK-NEXT: mov z2.b, z0.b[1] |
| ; CHECK-NEXT: fmov w8, s0 |
| ; CHECK-NEXT: fmov w9, s2 |
| ; CHECK-NEXT: stp w8, w9, [sp, #8] |
| ; CHECK-NEXT: ldr d0, [sp, #8] |
| ; CHECK-NEXT: st1b { z0.s }, p0, [x19] |
| ; CHECK-NEXT: ldp x20, x19, [sp, #32] // 16-byte Folded Reload |
| ; CHECK-NEXT: add sp, sp, #48 |
| ; CHECK-NEXT: ret |
| %alloc = alloca [4 x i8] |
| call void @def(ptr %alloc) |
| %load = load <4 x i8>, ptr %alloc |
| %strided.vec = shufflevector <4 x i8> %load, <4 x i8> poison, <2 x i32> <i32 0, i32 2> |
| store <2 x i8> %strided.vec, ptr %st_ptr |
| ret void |
| } |
| |
| define void @alloc_v6i8(ptr %st_ptr) nounwind { |
| ; CHECK-LABEL: alloc_v6i8: |
| ; CHECK: // %bb.0: |
| ; CHECK-NEXT: sub sp, sp, #48 |
| ; CHECK-NEXT: stp x20, x19, [sp, #32] // 16-byte Folded Spill |
| ; CHECK-NEXT: mov x19, x0 |
| ; CHECK-NEXT: add x0, sp, #24 |
| ; CHECK-NEXT: str x30, [sp, #16] // 8-byte Folded Spill |
| ; CHECK-NEXT: add x20, sp, #24 |
| ; CHECK-NEXT: bl def |
| ; CHECK-NEXT: ptrue p0.b, vl3 |
| ; CHECK-NEXT: ld2b { z0.b, z1.b }, p0/z, [x20] |
| ; CHECK-NEXT: ptrue p0.h, vl4 |
| ; CHECK-NEXT: mov z2.b, z1.b[3] |
| ; CHECK-NEXT: fmov w8, s1 |
| ; CHECK-NEXT: mov z3.b, z1.b[2] |
| ; CHECK-NEXT: mov z4.b, z1.b[1] |
| ; CHECK-NEXT: strh w8, [sp] |
| ; CHECK-NEXT: fmov w8, s2 |
| ; CHECK-NEXT: fmov w9, s3 |
| ; CHECK-NEXT: strh w8, [sp, #6] |
| ; CHECK-NEXT: fmov w8, s4 |
| ; CHECK-NEXT: strh w9, [sp, #4] |
| ; CHECK-NEXT: strh w8, [sp, #2] |
| ; CHECK-NEXT: add x8, sp, #12 |
| ; CHECK-NEXT: ldr d0, [sp] |
| ; CHECK-NEXT: st1b { z0.h }, p0, [x8] |
| ; CHECK-NEXT: ldrh w8, [sp, #12] |
| ; CHECK-NEXT: strb w9, [x19, #2] |
| ; CHECK-NEXT: ldr x30, [sp, #16] // 8-byte Folded Reload |
| ; CHECK-NEXT: strh w8, [x19] |
| ; CHECK-NEXT: ldp x20, x19, [sp, #32] // 16-byte Folded Reload |
| ; CHECK-NEXT: add sp, sp, #48 |
| ; CHECK-NEXT: ret |
| %alloc = alloca [6 x i8] |
| call void @def(ptr %alloc) |
| %load = load <6 x i8>, ptr %alloc |
| %strided.vec = shufflevector <6 x i8> %load, <6 x i8> poison, <3 x i32> <i32 1, i32 3, i32 5> |
| store <3 x i8> %strided.vec, ptr %st_ptr |
| ret void |
| } |
| |
| define void @alloc_v32i8(ptr %st_ptr) nounwind { |
| ; CHECK-LABEL: alloc_v32i8: |
| ; CHECK: // %bb.0: |
| ; CHECK-NEXT: sub sp, sp, #48 |
| ; CHECK-NEXT: stp x30, x19, [sp, #32] // 16-byte Folded Spill |
| ; CHECK-NEXT: mov x19, x0 |
| ; CHECK-NEXT: mov x0, sp |
| ; CHECK-NEXT: bl def |
| ; CHECK-NEXT: adrp x8, .LCPI2_0 |
| ; CHECK-NEXT: ldr q0, [sp] |
| ; CHECK-NEXT: ldr q1, [x8, :lo12:.LCPI2_0] |
| ; CHECK-NEXT: tbl z0.b, { z0.b }, z1.b |
| ; CHECK-NEXT: ldr q1, [sp, #16] |
| ; CHECK-NEXT: fmov w8, s1 |
| ; CHECK-NEXT: strb w8, [x19, #8] |
| ; CHECK-NEXT: fmov x8, d0 |
| ; CHECK-NEXT: str x8, [x19] |
| ; CHECK-NEXT: ldp x30, x19, [sp, #32] // 16-byte Folded Reload |
| ; CHECK-NEXT: add sp, sp, #48 |
| ; CHECK-NEXT: ret |
| %alloc = alloca [32 x i8] |
| call void @def(ptr %alloc) |
| %load = load <32 x i8>, ptr %alloc |
| %strided.vec = shufflevector <32 x i8> %load, <32 x i8> poison, <9 x i32> <i32 0, i32 2, i32 4, i32 6, i32 8, i32 10, i32 12, i32 14, i32 16> |
| store <9 x i8> %strided.vec, ptr %st_ptr |
| ret void |
| } |
| |
| |
| define void @alloc_v8f64(ptr %st_ptr) nounwind { |
| ; CHECK-LABEL: alloc_v8f64: |
| ; CHECK: // %bb.0: |
| ; CHECK-NEXT: sub sp, sp, #96 |
| ; CHECK-NEXT: stp x20, x19, [sp, #80] // 16-byte Folded Spill |
| ; CHECK-NEXT: mov x19, x0 |
| ; CHECK-NEXT: mov x0, sp |
| ; CHECK-NEXT: str x30, [sp, #64] // 8-byte Folded Spill |
| ; CHECK-NEXT: mov x20, sp |
| ; CHECK-NEXT: bl def |
| ; CHECK-NEXT: ptrue p0.d, vl2 |
| ; CHECK-NEXT: mov x8, #4 // =0x4 |
| ; CHECK-NEXT: ld2d { z0.d, z1.d }, p0/z, [x20] |
| ; CHECK-NEXT: ld2d { z2.d, z3.d }, p0/z, [x20, x8, lsl #3] |
| ; CHECK-NEXT: ldr x30, [sp, #64] // 8-byte Folded Reload |
| ; CHECK-NEXT: stp q0, q2, [x19] |
| ; CHECK-NEXT: ldp x20, x19, [sp, #80] // 16-byte Folded Reload |
| ; CHECK-NEXT: add sp, sp, #96 |
| ; CHECK-NEXT: ret |
| %alloc = alloca [8 x double] |
| call void @def(ptr %alloc) |
| %load = load <8 x double>, ptr %alloc |
| %strided.vec = shufflevector <8 x double> %load, <8 x double> poison, <4 x i32> <i32 0, i32 2, i32 4, i32 6> |
| store <4 x double> %strided.vec, ptr %st_ptr |
| ret void |
| } |