test/CodeGen/AArch64/arm64-stp.ll

   1 ; RUN: llc < %s -mtriple=arm64-eabi -aarch64-enable-stp-suppress=false -verify-machineinstrs -mcpu=cyclone | FileCheck %s
   2
   3 ; CHECK-LABEL: stp_int
   4 ; CHECK: stp w0, w1, [x2]
   5 define void @stp_int(i32 %a, i32 %b, i32* nocapture %p) nounwind {
   6   store i32 %a, i32* %p, align 4
   7   %add.ptr = getelementptr inbounds i32, i32* %p, i64 1
   8   store i32 %b, i32* %add.ptr, align 4
   9   ret void
  10 }
  11
  12 ; CHECK-LABEL: stp_long
  13 ; CHECK: stp x0, x1, [x2]
  14 define void @stp_long(i64 %a, i64 %b, i64* nocapture %p) nounwind {
  15   store i64 %a, i64* %p, align 8
  16   %add.ptr = getelementptr inbounds i64, i64* %p, i64 1
  17   store i64 %b, i64* %add.ptr, align 8
  18   ret void
  19 }
  20
  21 ; CHECK-LABEL: stp_float
  22 ; CHECK: stp s0, s1, [x0]
  23 define void @stp_float(float %a, float %b, float* nocapture %p) nounwind {
  24   store float %a, float* %p, align 4
  25   %add.ptr = getelementptr inbounds float, float* %p, i64 1
  26   store float %b, float* %add.ptr, align 4
  27   ret void
  28 }
  29
  30 ; CHECK-LABEL: stp_double
  31 ; CHECK: stp d0, d1, [x0]
  32 define void @stp_double(double %a, double %b, double* nocapture %p) nounwind {
  33   store double %a, double* %p, align 8
  34   %add.ptr = getelementptr inbounds double, double* %p, i64 1
  35   store double %b, double* %add.ptr, align 8
  36   ret void
  37 }
  38
  39 ; CHECK-LABEL: stp_doublex2
  40 ; CHECK: stp q0, q1, [x0]
  41 define void @stp_doublex2(<2 x double> %a, <2 x double> %b, <2 x double>* nocapture %p) nounwind {
  42   store <2 x double> %a, <2 x double>* %p, align 16
  43   %add.ptr = getelementptr inbounds <2 x double>, <2 x double>* %p, i64 1
  44   store <2 x double> %b, <2 x double>* %add.ptr, align 16
  45   ret void
  46 }
  47
  48 ; Test the load/store optimizer---combine ldurs into a ldp, if appropriate
  49 define void @stur_int(i32 %a, i32 %b, i32* nocapture %p) nounwind {
  50 ; CHECK-LABEL: stur_int
  51 ; CHECK: stp w{{[0-9]+}}, {{w[0-9]+}}, [x{{[0-9]+}}, #-8]
  52 ; CHECK-NEXT: ret
  53   %p1 = getelementptr inbounds i32, i32* %p, i32 -1
  54   store i32 %a, i32* %p1, align 2
  55   %p2 = getelementptr inbounds i32, i32* %p, i32 -2
  56   store i32 %b, i32* %p2, align 2
  57   ret void
  58 }
  59
  60 define void @stur_long(i64 %a, i64 %b, i64* nocapture %p) nounwind {
  61 ; CHECK-LABEL: stur_long
  62 ; CHECK: stp x{{[0-9]+}}, {{x[0-9]+}}, [x{{[0-9]+}}, #-16]
  63 ; CHECK-NEXT: ret
  64   %p1 = getelementptr inbounds i64, i64* %p, i32 -1
  65   store i64 %a, i64* %p1, align 2
  66   %p2 = getelementptr inbounds i64, i64* %p, i32 -2
  67   store i64 %b, i64* %p2, align 2
  68   ret void
  69 }
  70
  71 define void @stur_float(float %a, float %b, float* nocapture %p) nounwind {
  72 ; CHECK-LABEL: stur_float
  73 ; CHECK: stp s{{[0-9]+}}, {{s[0-9]+}}, [x{{[0-9]+}}, #-8]
  74 ; CHECK-NEXT: ret
  75   %p1 = getelementptr inbounds float, float* %p, i32 -1
  76   store float %a, float* %p1, align 2
  77   %p2 = getelementptr inbounds float, float* %p, i32 -2
  78   store float %b, float* %p2, align 2
  79   ret void
  80 }
  81
  82 define void @stur_double(double %a, double %b, double* nocapture %p) nounwind {
  83 ; CHECK-LABEL: stur_double
  84 ; CHECK: stp d{{[0-9]+}}, {{d[0-9]+}}, [x{{[0-9]+}}, #-16]
  85 ; CHECK-NEXT: ret
  86   %p1 = getelementptr inbounds double, double* %p, i32 -1
  87   store double %a, double* %p1, align 2
  88   %p2 = getelementptr inbounds double, double* %p, i32 -2
  89   store double %b, double* %p2, align 2
  90   ret void
  91 }
  92
  93 define void @stur_doublex2(<2 x double> %a, <2 x double> %b, <2 x double>* nocapture %p) nounwind {
  94 ; CHECK-LABEL: stur_doublex2
  95 ; CHECK: stp q{{[0-9]+}}, q{{[0-9]+}}, [x{{[0-9]+}}, #-32]
  96 ; CHECK-NEXT: ret
  97   %p1 = getelementptr inbounds <2 x double>, <2 x double>* %p, i32 -1
  98   store <2 x double> %a, <2 x double>* %p1, align 2
  99   %p2 = getelementptr inbounds <2 x double>, <2 x double>* %p, i32 -2
 100   store <2 x double> %b, <2 x double>* %p2, align 2
 101   ret void
 102 }
 103
 104 define void @splat_v4i32(i32 %v, i32 *%p) {
 105 entry:
 106
 107 ; CHECK-LABEL: splat_v4i32
 108 ; CHECK-DAG: stp w0, w0, [x1]
 109 ; CHECK-DAG: stp w0, w0, [x1, #8]
 110 ; CHECK: ret
 111
 112   %p17 = insertelement <4 x i32> undef, i32 %v, i32 0
 113   %p18 = insertelement <4 x i32> %p17, i32 %v, i32 1
 114   %p19 = insertelement <4 x i32> %p18, i32 %v, i32 2
 115   %p20 = insertelement <4 x i32> %p19, i32 %v, i32 3
 116   %p21 = bitcast i32* %p to <4 x i32>*
 117   store <4 x i32> %p20, <4 x i32>* %p21, align 4
 118   ret void
 119 }
 120
 121 ; Check that a non-splat store that is storing a vector created by 4
 122 ; insertelements that is not a splat vector does not get split.
 123 define void @nosplat_v4i32(i32 %v, i32 *%p) {
 124 entry:
 125
 126 ; CHECK-LABEL: nosplat_v4i32:
 127 ; CHECK: str w0,
 128 ; CHECK: ldr q[[REG1:[0-9]+]],
 129 ; CHECK-DAG: mov v[[REG1]].s[1], w0
 130 ; CHECK-DAG: mov v[[REG1]].s[2], w0
 131 ; CHECK-DAG: mov v[[REG1]].s[3], w0
 132 ; CHECK: ext v[[REG2:[0-9]+]].16b, v[[REG1]].16b, v[[REG1]].16b, #8
 133 ; CHECK: stp d[[REG1]], d[[REG2]], [x1]
 134 ; CHECK: ret
 135
 136   %p17 = insertelement <4 x i32> undef, i32 %v, i32 %v
 137   %p18 = insertelement <4 x i32> %p17, i32 %v, i32 1
 138   %p19 = insertelement <4 x i32> %p18, i32 %v, i32 2
 139   %p20 = insertelement <4 x i32> %p19, i32 %v, i32 3
 140   %p21 = bitcast i32* %p to <4 x i32>*
 141   store <4 x i32> %p20, <4 x i32>* %p21, align 4
 142   ret void
 143 }
 144
 145 ; Check that a non-splat store that is storing a vector created by 4
 146 ; insertelements that is not a splat vector does not get split.
 147 define void @nosplat2_v4i32(i32 %v, i32 *%p, <4 x i32> %vin) {
 148 entry:
 149
 150 ; CHECK-LABEL: nosplat2_v4i32:
 151 ; CHECK: mov v[[REG1]].s[1], w0
 152 ; CHECK-DAG: mov v[[REG1]].s[2], w0
 153 ; CHECK-DAG: mov v[[REG1]].s[3], w0
 154 ; CHECK: ext v[[REG2:[0-9]+]].16b, v[[REG1]].16b, v[[REG1]].16b, #8
 155 ; CHECK: stp d[[REG1]], d[[REG2]], [x1]
 156 ; CHECK: ret
 157
 158   %p18 = insertelement <4 x i32> %vin, i32 %v, i32 1
 159   %p19 = insertelement <4 x i32> %p18, i32 %v, i32 2
 160   %p20 = insertelement <4 x i32> %p19, i32 %v, i32 3
 161   %p21 = bitcast i32* %p to <4 x i32>*
 162   store <4 x i32> %p20, <4 x i32>* %p21, align 4
 163   ret void
 164 }
 165
 166 ; Read of %b to compute %tmp2 shouldn't prevent formation of stp
 167 ; CHECK-LABEL: stp_int_rar_hazard
 168 ; CHECK: ldr [[REG:w[0-9]+]], [x2, #8]
 169 ; CHECK: add w8, [[REG]], w1
 170 ; CHECK: stp w0, w1, [x2]
 171 ; CHECK: ret
 172 define i32 @stp_int_rar_hazard(i32 %a, i32 %b, i32* nocapture %p) nounwind {
 173   store i32 %a, i32* %p, align 4
 174   %ld.ptr = getelementptr inbounds i32, i32* %p, i64 2
 175   %tmp = load i32, i32* %ld.ptr, align 4
 176   %tmp2 = add i32 %tmp, %b
 177   %add.ptr = getelementptr inbounds i32, i32* %p, i64 1
 178   store i32 %b, i32* %add.ptr, align 4
 179   ret i32 %tmp2
 180 }
 181
 182 ; Read of %b to compute %tmp2 shouldn't prevent formation of stp
 183 ; CHECK-LABEL: stp_int_rar_hazard_after
 184 ; CHECK: ldr [[REG:w[0-9]+]], [x3, #4]
 185 ; CHECK: add w0, [[REG]], w2
 186 ; CHECK: stp w1, w2, [x3]
 187 ; CHECK: ret
 188 define i32 @stp_int_rar_hazard_after(i32 %w0, i32 %a, i32 %b, i32* nocapture %p) nounwind {
 189   store i32 %a, i32* %p, align 4
 190   %ld.ptr = getelementptr inbounds i32, i32* %p, i64 1
 191   %tmp = load i32, i32* %ld.ptr, align 4
 192   %tmp2 = add i32 %tmp, %b
 193   %add.ptr = getelementptr inbounds i32, i32* %p, i64 1
 194   store i32 %b, i32* %add.ptr, align 4
 195   ret i32 %tmp2
 196 }