llvm/test/CodeGen/AArch64/GlobalISel/sve-formal-argument.ll

   1 ; NOTE: Assertions have been autogenerated by utils/update_llc_test_checks.py UTC_ARGS: --version 5
   2 ; RUN: llc -mtriple=aarch64-linux-gnu -mattr=+sve -global-isel -global-isel-abort=1 -aarch64-enable-gisel-sve=1 %s -o - | FileCheck %s
   3 ;; vscale x 128-bit
   4
   5 define void @formal_argument_nxv16i8(<vscale x 16 x i8> %0, ptr %p) {
   6 ; CHECK-LABEL: formal_argument_nxv16i8:
   7 ; CHECK:       // %bb.0:
   8 ; CHECK-NEXT:    ptrue p0.b
   9 ; CHECK-NEXT:    st1b { z0.b }, p0, [x0]
  10 ; CHECK-NEXT:    ret
  11   store <vscale x 16 x i8> %0, ptr %p, align 16
  12   ret void
  13 }
  14
  15 define void @formal_argument_nxv8i16(<vscale x 8 x i16> %0, ptr %p) {
  16 ; CHECK-LABEL: formal_argument_nxv8i16:
  17 ; CHECK:       // %bb.0:
  18 ; CHECK-NEXT:    ptrue p0.h
  19 ; CHECK-NEXT:    st1h { z0.h }, p0, [x0]
  20 ; CHECK-NEXT:    ret
  21   store <vscale x 8 x i16> %0, ptr %p, align 16
  22   ret void
  23 }
  24
  25 define void @formal_argument_nxv4i32(<vscale x 4 x i32> %0, ptr %p) {
  26 ; CHECK-LABEL: formal_argument_nxv4i32:
  27 ; CHECK:       // %bb.0:
  28 ; CHECK-NEXT:    ptrue p0.s
  29 ; CHECK-NEXT:    st1w { z0.s }, p0, [x0]
  30 ; CHECK-NEXT:    ret
  31   store <vscale x 4 x i32> %0, ptr %p, align 16
  32   ret void
  33 }
  34
  35 define void @formal_argument_nxv2i64(<vscale x 2 x i64> %0, ptr %p) {
  36 ; CHECK-LABEL: formal_argument_nxv2i64:
  37 ; CHECK:       // %bb.0:
  38 ; CHECK-NEXT:    ptrue p0.d
  39 ; CHECK-NEXT:    st1d { z0.d }, p0, [x0]
  40 ; CHECK-NEXT:    ret
  41   store <vscale x 2 x i64> %0, ptr %p, align 16
  42   ret void
  43 }
  44
  45 ;; TODO: Add tests for other types when store is supported for them.