test/CodeGen/SystemZ/cond-store-05.ll

   1 ; Test f32 conditional stores that are presented as selects.
   2 ;
   3 ; RUN: llc < %s -mtriple=s390x-linux-gnu | FileCheck %s
   4
   5 declare void @foo(float *)
   6
   7 ; Test with the loaded value first.
   8 define void @f1(float *%ptr, float %alt, i32 %limit) {
   9 ; CHECK-LABEL: f1:
  10 ; CHECK-NOT: %r2
  11 ; CHECK: blr %r14
  12 ; CHECK-NOT: %r2
  13 ; CHECK: ste %f0, 0(%r2)
  14 ; CHECK: br %r14
  15   %cond = icmp ult i32 %limit, 420
  16   %orig = load float, float *%ptr
  17   %res = select i1 %cond, float %orig, float %alt
  18   store float %res, float *%ptr
  19   ret void
  20 }
  21
  22 ; ...and with the loaded value second
  23 define void @f2(float *%ptr, float %alt, i32 %limit) {
  24 ; CHECK-LABEL: f2:
  25 ; CHECK-NOT: %r2
  26 ; CHECK: bher %r14
  27 ; CHECK-NOT: %r2
  28 ; CHECK: ste %f0, 0(%r2)
  29 ; CHECK: br %r14
  30   %cond = icmp ult i32 %limit, 420
  31   %orig = load float, float *%ptr
  32   %res = select i1 %cond, float %alt, float %orig
  33   store float %res, float *%ptr
  34   ret void
  35 }
  36
  37 ; Check the high end of the aligned STE range.
  38 define void @f3(float *%base, float %alt, i32 %limit) {
  39 ; CHECK-LABEL: f3:
  40 ; CHECK-NOT: %r2
  41 ; CHECK: blr %r14
  42 ; CHECK-NOT: %r2
  43 ; CHECK: ste %f0, 4092(%r2)
  44 ; CHECK: br %r14
  45   %ptr = getelementptr float, float *%base, i64 1023
  46   %cond = icmp ult i32 %limit, 420
  47   %orig = load float, float *%ptr
  48   %res = select i1 %cond, float %orig, float %alt
  49   store float %res, float *%ptr
  50   ret void
  51 }
  52
  53 ; Check the next word up, which should use STEY instead of STE.
  54 define void @f4(float *%base, float %alt, i32 %limit) {
  55 ; CHECK-LABEL: f4:
  56 ; CHECK-NOT: %r2
  57 ; CHECK: blr %r14
  58 ; CHECK-NOT: %r2
  59 ; CHECK: stey %f0, 4096(%r2)
  60 ; CHECK: br %r14
  61   %ptr = getelementptr float, float *%base, i64 1024
  62   %cond = icmp ult i32 %limit, 420
  63   %orig = load float, float *%ptr
  64   %res = select i1 %cond, float %orig, float %alt
  65   store float %res, float *%ptr
  66   ret void
  67 }
  68
  69 ; Check the high end of the aligned STEY range.
  70 define void @f5(float *%base, float %alt, i32 %limit) {
  71 ; CHECK-LABEL: f5:
  72 ; CHECK-NOT: %r2
  73 ; CHECK: blr %r14
  74 ; CHECK-NOT: %r2
  75 ; CHECK: stey %f0, 524284(%r2)
  76 ; CHECK: br %r14
  77   %ptr = getelementptr float, float *%base, i64 131071
  78   %cond = icmp ult i32 %limit, 420
  79   %orig = load float, float *%ptr
  80   %res = select i1 %cond, float %orig, float %alt
  81   store float %res, float *%ptr
  82   ret void
  83 }
  84
  85 ; Check the next word up, which needs separate address logic.
  86 ; Other sequences besides this one would be OK.
  87 define void @f6(float *%base, float %alt, i32 %limit) {
  88 ; CHECK-LABEL: f6:
  89 ; CHECK-NOT: %r2
  90 ; CHECK: blr %r14
  91 ; CHECK-NOT: %r2
  92 ; CHECK: agfi %r2, 524288
  93 ; CHECK: ste %f0, 0(%r2)
  94 ; CHECK: br %r14
  95   %ptr = getelementptr float, float *%base, i64 131072
  96   %cond = icmp ult i32 %limit, 420
  97   %orig = load float, float *%ptr
  98   %res = select i1 %cond, float %orig, float %alt
  99   store float %res, float *%ptr
 100   ret void
 101 }
 102
 103 ; Check the low end of the STEY range.
 104 define void @f7(float *%base, float %alt, i32 %limit) {
 105 ; CHECK-LABEL: f7:
 106 ; CHECK-NOT: %r2
 107 ; CHECK: blr %r14
 108 ; CHECK-NOT: %r2
 109 ; CHECK: stey %f0, -524288(%r2)
 110 ; CHECK: br %r14
 111   %ptr = getelementptr float, float *%base, i64 -131072
 112   %cond = icmp ult i32 %limit, 420
 113   %orig = load float, float *%ptr
 114   %res = select i1 %cond, float %orig, float %alt
 115   store float %res, float *%ptr
 116   ret void
 117 }
 118
 119 ; Check the next word down, which needs separate address logic.
 120 ; Other sequences besides this one would be OK.
 121 define void @f8(float *%base, float %alt, i32 %limit) {
 122 ; CHECK-LABEL: f8:
 123 ; CHECK-NOT: %r2
 124 ; CHECK: blr %r14
 125 ; CHECK-NOT: %r2
 126 ; CHECK: agfi %r2, -524292
 127 ; CHECK: ste %f0, 0(%r2)
 128 ; CHECK: br %r14
 129   %ptr = getelementptr float, float *%base, i64 -131073
 130   %cond = icmp ult i32 %limit, 420
 131   %orig = load float, float *%ptr
 132   %res = select i1 %cond, float %orig, float %alt
 133   store float %res, float *%ptr
 134   ret void
 135 }
 136
 137 ; Check that STEY allows an index.
 138 define void @f9(i64 %base, i64 %index, float %alt, i32 %limit) {
 139 ; CHECK-LABEL: f9:
 140 ; CHECK-NOT: %r2
 141 ; CHECK: blr %r14
 142 ; CHECK-NOT: %r2
 143 ; CHECK: stey %f0, 4096(%r3,%r2)
 144 ; CHECK: br %r14
 145   %add1 = add i64 %base, %index
 146   %add2 = add i64 %add1, 4096
 147   %ptr = inttoptr i64 %add2 to float *
 148   %cond = icmp ult i32 %limit, 420
 149   %orig = load float, float *%ptr
 150   %res = select i1 %cond, float %orig, float %alt
 151   store float %res, float *%ptr
 152   ret void
 153 }
 154
 155 ; Check that volatile loads are not matched.
 156 define void @f10(float *%ptr, float %alt, i32 %limit) {
 157 ; CHECK-LABEL: f10:
 158 ; CHECK: le {{%f[0-5]}}, 0(%r2)
 159 ; CHECK: {{jl|jnl}} [[LABEL:[^ ]*]]
 160 ; CHECK: [[LABEL]]:
 161 ; CHECK: ste {{%f[0-5]}}, 0(%r2)
 162 ; CHECK: br %r14
 163   %cond = icmp ult i32 %limit, 420
 164   %orig = load volatile float, float *%ptr
 165   %res = select i1 %cond, float %orig, float %alt
 166   store float %res, float *%ptr
 167   ret void
 168 }
 169
 170 ; ...likewise stores.  In this case we should have a conditional load into %f0.
 171 define void @f11(float *%ptr, float %alt, i32 %limit) {
 172 ; CHECK-LABEL: f11:
 173 ; CHECK: jhe [[LABEL:[^ ]*]]
 174 ; CHECK: le %f0, 0(%r2)
 175 ; CHECK: [[LABEL]]:
 176 ; CHECK: ste %f0, 0(%r2)
 177 ; CHECK: br %r14
 178   %cond = icmp ult i32 %limit, 420
 179   %orig = load float, float *%ptr
 180   %res = select i1 %cond, float %orig, float %alt
 181   store volatile float %res, float *%ptr
 182   ret void
 183 }
 184
 185 ; Try a frame index base.
 186 define void @f12(float %alt, i32 %limit) {
 187 ; CHECK-LABEL: f12:
 188 ; CHECK: brasl %r14, foo@PLT
 189 ; CHECK-NOT: %r15
 190 ; CHECK: jl [[LABEL:[^ ]*]]
 191 ; CHECK-NOT: %r15
 192 ; CHECK: ste {{%f[0-9]+}}, {{[0-9]+}}(%r15)
 193 ; CHECK: [[LABEL]]:
 194 ; CHECK: brasl %r14, foo@PLT
 195 ; CHECK: br %r14
 196   %ptr = alloca float
 197   call void @foo(float *%ptr)
 198   %cond = icmp ult i32 %limit, 420
 199   %orig = load float, float *%ptr
 200   %res = select i1 %cond, float %orig, float %alt
 201   store float %res, float *%ptr
 202   call void @foo(float *%ptr)
 203   ret void
 204 }