llvm/test/CodeGen/AArch64/sve-st1-addressing-mode-reg-imm.ll

   1 ; NOTE: Assertions have been autogenerated by utils/update_llc_test_checks.py
   2 ; RUN: llc -mtriple=aarch64-linux-gnu -mattr=+sve < %s | FileCheck %s
   3
   4 ; ST1B
   5
   6 define void @st1b_lower_bound(<vscale x 16 x i8> %data, <vscale x 16 x i8>* %a) {
   7 ; CHECK-LABEL: st1b_lower_bound:
   8 ; CHECK:       // %bb.0:
   9 ; CHECK-NEXT:    ptrue p0.b
  10 ; CHECK-NEXT:    st1b { z0.b }, p0, [x0, #-8, mul vl]
  11 ; CHECK-NEXT:    ret
  12   %base = getelementptr <vscale x 16 x i8>, <vscale x 16 x i8>* %a, i64 -8
  13   store <vscale x 16 x i8> %data, <vscale x 16 x i8>* %base
  14   ret void
  15 }
  16
  17 define void @st1b_inbound(<vscale x 16 x i8> %data, <vscale x 16 x i8>* %a) {
  18 ; CHECK-LABEL: st1b_inbound:
  19 ; CHECK:       // %bb.0:
  20 ; CHECK-NEXT:    ptrue p0.b
  21 ; CHECK-NEXT:    st1b { z0.b }, p0, [x0, #1, mul vl]
  22 ; CHECK-NEXT:    ret
  23   %base = getelementptr <vscale x 16 x i8>, <vscale x 16 x i8>* %a, i64 1
  24   store <vscale x 16 x i8> %data, <vscale x 16 x i8>* %base
  25   ret void
  26 }
  27
  28 define void @st1b_upper_bound(<vscale x 16 x i8> %data, <vscale x 16 x i8>* %a) {
  29 ; CHECK-LABEL: st1b_upper_bound:
  30 ; CHECK:       // %bb.0:
  31 ; CHECK-NEXT:    ptrue p0.b
  32 ; CHECK-NEXT:    st1b { z0.b }, p0, [x0, #7, mul vl]
  33 ; CHECK-NEXT:    ret
  34   %base = getelementptr <vscale x 16 x i8>, <vscale x 16 x i8>* %a, i64 7
  35   store <vscale x 16 x i8> %data, <vscale x 16 x i8>* %base
  36   ret void
  37 }
  38
  39 define void @st1b_out_of_upper_bound(<vscale x 16 x i8> %data, <vscale x 16 x i8>* %a) {
  40 ; CHECK-LABEL: st1b_out_of_upper_bound:
  41 ; CHECK:       // %bb.0:
  42 ; CHECK-NEXT:    ptrue p0.b
  43 ; CHECK-NEXT:    rdvl x8, #8
  44 ; CHECK-NEXT:    st1b { z0.b }, p0, [x0, x8]
  45 ; CHECK-NEXT:    ret
  46   %base = getelementptr <vscale x 16 x i8>, <vscale x 16 x i8>* %a, i64 8
  47   store <vscale x 16 x i8> %data, <vscale x 16 x i8>* %base
  48   ret void
  49 }
  50
  51 define void @st1b_out_of_lower_bound(<vscale x 16 x i8> %data, <vscale x 16 x i8>* %a) {
  52 ; CHECK-LABEL: st1b_out_of_lower_bound:
  53 ; CHECK:       // %bb.0:
  54 ; CHECK-NEXT:    ptrue p0.b
  55 ; CHECK-NEXT:    rdvl x8, #-9
  56 ; CHECK-NEXT:    st1b { z0.b }, p0, [x0, x8]
  57 ; CHECK-NEXT:    ret
  58   %base = getelementptr <vscale x 16 x i8>, <vscale x 16 x i8>* %a, i64 -9
  59   store <vscale x 16 x i8> %data, <vscale x 16 x i8>* %base
  60   ret void
  61 }
  62
  63 ; ST1H
  64
  65 define void @st1h_inbound(<vscale x 8 x i16> %data, <vscale x 8 x i16>* %a) {
  66 ; CHECK-LABEL: st1h_inbound:
  67 ; CHECK:       // %bb.0:
  68 ; CHECK-NEXT:    ptrue p0.h
  69 ; CHECK-NEXT:    st1h { z0.h }, p0, [x0, #-6, mul vl]
  70 ; CHECK-NEXT:    ret
  71   %base = getelementptr <vscale x 8 x i16>, <vscale x 8 x i16>* %a, i64 -6
  72   store <vscale x 8 x i16> %data, <vscale x 8 x i16>* %base
  73   ret void
  74 }
  75
  76 ; ST1W
  77
  78 define void @st1w_inbound(<vscale x 4 x i32> %data, <vscale x 4 x i32>* %a) {
  79 ; CHECK-LABEL: st1w_inbound:
  80 ; CHECK:       // %bb.0:
  81 ; CHECK-NEXT:    ptrue p0.s
  82 ; CHECK-NEXT:    st1w { z0.s }, p0, [x0, #2, mul vl]
  83 ; CHECK-NEXT:    ret
  84   %base = getelementptr <vscale x 4 x i32>, <vscale x 4 x i32>* %a, i64 2
  85   store <vscale x 4 x i32> %data, <vscale x 4 x i32>* %base
  86   ret void
  87 }
  88
  89 ; ST1D
  90
  91 define void @st1d_inbound(<vscale x 2 x i64> %data, <vscale x 2 x i64>* %a) {
  92 ; CHECK-LABEL: st1d_inbound:
  93 ; CHECK:       // %bb.0:
  94 ; CHECK-NEXT:    ptrue p0.d
  95 ; CHECK-NEXT:    st1d { z0.d }, p0, [x0, #5, mul vl]
  96 ; CHECK-NEXT:    ret
  97   %base = getelementptr <vscale x 2 x i64>, <vscale x 2 x i64>* %a, i64 5
  98   store <vscale x 2 x i64> %data, <vscale x 2 x i64>* %base
  99   ret void
 100 }
 101
 102
 103 ; Splat stores of unpacked FP scalable vectors
 104
 105 define void @store_nxv2f32(<vscale x 2 x float>* %out) {
 106 ; CHECK-LABEL: store_nxv2f32:
 107 ; CHECK:       // %bb.0:
 108 ; CHECK-NEXT:    ptrue p0.d
 109 ; CHECK-NEXT:    fmov z0.s, #1.00000000
 110 ; CHECK-NEXT:    st1w { z0.d }, p0, [x0]
 111 ; CHECK-NEXT:    ret
 112   %ins = insertelement <vscale x 2 x float> undef, float 1.0, i32 0
 113   %splat = shufflevector <vscale x 2 x float> %ins, <vscale x 2 x float> undef, <vscale x 2 x i32> zeroinitializer
 114   store <vscale x 2 x float> %splat, <vscale x 2 x float>* %out
 115   ret void
 116 }
 117
 118 define void @store_nxv4f16(<vscale x 4 x half>* %out) {
 119 ; CHECK-LABEL: store_nxv4f16:
 120 ; CHECK:       // %bb.0:
 121 ; CHECK-NEXT:    ptrue p0.s
 122 ; CHECK-NEXT:    fmov z0.h, #1.00000000
 123 ; CHECK-NEXT:    st1h { z0.s }, p0, [x0]
 124 ; CHECK-NEXT:    ret
 125   %ins = insertelement <vscale x 4 x half> undef, half 1.0, i32 0
 126   %splat = shufflevector <vscale x 4 x half> %ins, <vscale x 4 x half> undef, <vscale x 4 x i32> zeroinitializer
 127   store <vscale x 4 x half> %splat, <vscale x 4 x half>* %out
 128   ret void
 129 }
 130
 131 ; Splat stores of unusual FP scalable vector types
 132
 133 define void @store_nxv6f32(<vscale x 6 x float>* %out) {
 134 ; CHECK-LABEL: store_nxv6f32:
 135 ; CHECK:       // %bb.0:
 136 ; CHECK-NEXT:    ptrue p0.d
 137 ; CHECK-NEXT:    ptrue p1.s
 138 ; CHECK-NEXT:    fmov z0.s, #1.00000000
 139 ; CHECK-NEXT:    st1w { z0.d }, p0, [x0, #2, mul vl]
 140 ; CHECK-NEXT:    st1w { z0.s }, p1, [x0]
 141 ; CHECK-NEXT:    ret
 142   %ins = insertelement <vscale x 6 x float> undef, float 1.0, i32 0
 143   %splat = shufflevector <vscale x 6 x float> %ins, <vscale x 6 x float> undef, <vscale x 6 x i32> zeroinitializer
 144   store <vscale x 6 x float> %splat, <vscale x 6 x float>* %out
 145   ret void
 146 }
 147
 148 define void @store_nxv12f16(<vscale x 12 x half>* %out) {
 149 ; CHECK-LABEL: store_nxv12f16:
 150 ; CHECK:       // %bb.0:
 151 ; CHECK-NEXT:    ptrue p0.s
 152 ; CHECK-NEXT:    ptrue p1.h
 153 ; CHECK-NEXT:    fmov z0.h, #1.00000000
 154 ; CHECK-NEXT:    st1h { z0.s }, p0, [x0, #2, mul vl]
 155 ; CHECK-NEXT:    st1h { z0.h }, p1, [x0]
 156 ; CHECK-NEXT:    ret
 157   %ins = insertelement <vscale x 12 x half> undef, half 1.0, i32 0
 158   %splat = shufflevector <vscale x 12 x half> %ins, <vscale x 12 x half> undef, <vscale x 12 x i32> zeroinitializer
 159   store <vscale x 12 x half> %splat, <vscale x 12 x half>* %out
 160   ret void
 161 }