llvm/test/Transforms/SLPVectorizer/RISCV/floating-point.ll

   1 ; NOTE: Assertions have been autogenerated by utils/update_test_checks.py UTC_ARGS: --version 2
   2 ; RUN: opt < %s -passes=slp-vectorizer -S -mtriple=riscv64 -mattr=+v,+f \
   3 ; RUN:     -riscv-v-vector-bits-min=-1 -riscv-v-slp-max-vf=0 \
   4 ; RUN:     | FileCheck %s
   5 ; RUN: opt < %s -passes=slp-vectorizer -S -mtriple=riscv64 -mattr=+v,+f \
   6 ; RUN:     | FileCheck %s --check-prefix=DEFAULT
   7
   8 define void @fp_add(ptr %dst, ptr %p, ptr %q) {
   9 ; CHECK-LABEL: define void @fp_add
  10 ; CHECK-SAME: (ptr [[DST:%.*]], ptr [[P:%.*]], ptr [[Q:%.*]]) #[[ATTR0:[0-9]+]] {
  11 ; CHECK-NEXT:  entry:
  12 ; CHECK-NEXT:    [[TMP0:%.*]] = load <4 x float>, ptr [[P]], align 4
  13 ; CHECK-NEXT:    [[TMP1:%.*]] = load <4 x float>, ptr [[Q]], align 4
  14 ; CHECK-NEXT:    [[TMP2:%.*]] = fadd <4 x float> [[TMP0]], [[TMP1]]
  15 ; CHECK-NEXT:    store <4 x float> [[TMP2]], ptr [[DST]], align 4
  16 ; CHECK-NEXT:    ret void
  17 ;
  18 ; DEFAULT-LABEL: define void @fp_add
  19 ; DEFAULT-SAME: (ptr [[DST:%.*]], ptr [[P:%.*]], ptr [[Q:%.*]]) #[[ATTR0:[0-9]+]] {
  20 ; DEFAULT-NEXT:  entry:
  21 ; DEFAULT-NEXT:    [[TMP0:%.*]] = load <4 x float>, ptr [[P]], align 4
  22 ; DEFAULT-NEXT:    [[TMP1:%.*]] = load <4 x float>, ptr [[Q]], align 4
  23 ; DEFAULT-NEXT:    [[TMP2:%.*]] = fadd <4 x float> [[TMP0]], [[TMP1]]
  24 ; DEFAULT-NEXT:    store <4 x float> [[TMP2]], ptr [[DST]], align 4
  25 ; DEFAULT-NEXT:    ret void
  26 ;
  27 entry:
  28   %e0 = load float, ptr %p, align 4
  29   %pe1 = getelementptr inbounds float, ptr %p, i64 1
  30   %e1 = load float, ptr %pe1, align 4
  31   %pe2 = getelementptr inbounds float, ptr %p, i64 2
  32   %e2 = load float, ptr %pe2, align 4
  33   %pe3 = getelementptr inbounds float, ptr %p, i64 3
  34   %e3 = load float, ptr %pe3, align 4
  35
  36   %f0 = load float, ptr %q, align 4
  37   %pf1 = getelementptr inbounds float, ptr %q, i64 1
  38   %f1 = load float, ptr %pf1, align 4
  39   %pf2 = getelementptr inbounds float, ptr %q, i64 2
  40   %f2 = load float, ptr %pf2, align 4
  41   %pf3 = getelementptr inbounds float, ptr %q, i64 3
  42   %f3 = load float, ptr %pf3, align 4
  43
  44   %a0 = fadd float %e0, %f0
  45   %a1 = fadd float %e1, %f1
  46   %a2 = fadd float %e2, %f2
  47   %a3 = fadd float %e3, %f3
  48
  49   store float %a0, ptr %dst, align 4
  50   %pa1 = getelementptr inbounds float, ptr %dst, i64 1
  51   store float %a1, ptr %pa1, align 4
  52   %pa2 = getelementptr inbounds float, ptr %dst, i64 2
  53   store float %a2, ptr %pa2, align 4
  54   %pa3 = getelementptr inbounds float, ptr %dst, i64 3
  55   store float %a3, ptr %pa3, align 4
  56
  57   ret void
  58 }
  59
  60 define void @fp_sub(ptr %dst, ptr %p) {
  61 ; CHECK-LABEL: define void @fp_sub
  62 ; CHECK-SAME: (ptr [[DST:%.*]], ptr [[P:%.*]]) #[[ATTR0]] {
  63 ; CHECK-NEXT:  entry:
  64 ; CHECK-NEXT:    [[TMP0:%.*]] = load <4 x float>, ptr [[P]], align 4
  65 ; CHECK-NEXT:    [[TMP1:%.*]] = fsub <4 x float> [[TMP0]], <float 3.000000e+00, float 3.000000e+00, float 3.000000e+00, float 3.000000e+00>
  66 ; CHECK-NEXT:    store <4 x float> [[TMP1]], ptr [[DST]], align 4
  67 ; CHECK-NEXT:    ret void
  68 ;
  69 ; DEFAULT-LABEL: define void @fp_sub
  70 ; DEFAULT-SAME: (ptr [[DST:%.*]], ptr [[P:%.*]]) #[[ATTR0]] {
  71 ; DEFAULT-NEXT:  entry:
  72 ; DEFAULT-NEXT:    [[TMP0:%.*]] = load <4 x float>, ptr [[P]], align 4
  73 ; DEFAULT-NEXT:    [[TMP1:%.*]] = fsub <4 x float> [[TMP0]], <float 3.000000e+00, float 3.000000e+00, float 3.000000e+00, float 3.000000e+00>
  74 ; DEFAULT-NEXT:    store <4 x float> [[TMP1]], ptr [[DST]], align 4
  75 ; DEFAULT-NEXT:    ret void
  76 ;
  77 entry:
  78   %e0 = load float, ptr %p, align 4
  79   %pe1 = getelementptr inbounds float, ptr %p, i64 1
  80   %e1 = load float, ptr %pe1, align 4
  81   %pe2 = getelementptr inbounds float, ptr %p, i64 2
  82   %e2 = load float, ptr %pe2, align 4
  83   %pe3 = getelementptr inbounds float, ptr %p, i64 3
  84   %e3 = load float, ptr %pe3, align 4
  85
  86   %a0 = fsub float %e0, 3.0
  87   %a1 = fsub float %e1, 3.0
  88   %a2 = fsub float %e2, 3.0
  89   %a3 = fsub float %e3, 3.0
  90
  91   store float %a0, ptr %dst, align 4
  92   %pa1 = getelementptr inbounds float, ptr %dst, i64 1
  93   store float %a1, ptr %pa1, align 4
  94   %pa2 = getelementptr inbounds float, ptr %dst, i64 2
  95   store float %a2, ptr %pa2, align 4
  96   %pa3 = getelementptr inbounds float, ptr %dst, i64 3
  97   store float %a3, ptr %pa3, align 4
  98
  99   ret void
 100 }
 101
 102 define void @fp_mul(ptr %dst, ptr %p, ptr %q) {
 103 ; CHECK-LABEL: define void @fp_mul
 104 ; CHECK-SAME: (ptr [[DST:%.*]], ptr [[P:%.*]], ptr [[Q:%.*]]) #[[ATTR0]] {
 105 ; CHECK-NEXT:  entry:
 106 ; CHECK-NEXT:    [[TMP0:%.*]] = load <4 x float>, ptr [[P]], align 4
 107 ; CHECK-NEXT:    [[TMP1:%.*]] = load <4 x float>, ptr [[Q]], align 4
 108 ; CHECK-NEXT:    [[TMP2:%.*]] = fmul <4 x float> [[TMP0]], [[TMP1]]
 109 ; CHECK-NEXT:    store <4 x float> [[TMP2]], ptr [[DST]], align 4
 110 ; CHECK-NEXT:    ret void
 111 ;
 112 ; DEFAULT-LABEL: define void @fp_mul
 113 ; DEFAULT-SAME: (ptr [[DST:%.*]], ptr [[P:%.*]], ptr [[Q:%.*]]) #[[ATTR0]] {
 114 ; DEFAULT-NEXT:  entry:
 115 ; DEFAULT-NEXT:    [[TMP0:%.*]] = load <4 x float>, ptr [[P]], align 4
 116 ; DEFAULT-NEXT:    [[TMP1:%.*]] = load <4 x float>, ptr [[Q]], align 4
 117 ; DEFAULT-NEXT:    [[TMP2:%.*]] = fmul <4 x float> [[TMP0]], [[TMP1]]
 118 ; DEFAULT-NEXT:    store <4 x float> [[TMP2]], ptr [[DST]], align 4
 119 ; DEFAULT-NEXT:    ret void
 120 ;
 121 entry:
 122   %e0 = load float, ptr %p, align 4
 123   %pe1 = getelementptr inbounds float, ptr %p, i64 1
 124   %e1 = load float, ptr %pe1, align 4
 125   %pe2 = getelementptr inbounds float, ptr %p, i64 2
 126   %e2 = load float, ptr %pe2, align 4
 127   %pe3 = getelementptr inbounds float, ptr %p, i64 3
 128   %e3 = load float, ptr %pe3, align 4
 129
 130   %f0 = load float, ptr %q, align 4
 131   %pf1 = getelementptr inbounds float, ptr %q, i64 1
 132   %f1 = load float, ptr %pf1, align 4
 133   %pf2 = getelementptr inbounds float, ptr %q, i64 2
 134   %f2 = load float, ptr %pf2, align 4
 135   %pf3 = getelementptr inbounds float, ptr %q, i64 3
 136   %f3 = load float, ptr %pf3, align 4
 137
 138   %a0 = fmul float %e0, %f0
 139   %a1 = fmul float %e1, %f1
 140   %a2 = fmul float %e2, %f2
 141   %a3 = fmul float %e3, %f3
 142
 143   store float %a0, ptr %dst, align 4
 144   %pa1 = getelementptr inbounds float, ptr %dst, i64 1
 145   store float %a1, ptr %pa1, align 4
 146   %pa2 = getelementptr inbounds float, ptr %dst, i64 2
 147   store float %a2, ptr %pa2, align 4
 148   %pa3 = getelementptr inbounds float, ptr %dst, i64 3
 149   store float %a3, ptr %pa3, align 4
 150
 151   ret void
 152 }
 153
 154 define void @fp_div(ptr %dst, ptr %p) {
 155 ; CHECK-LABEL: define void @fp_div
 156 ; CHECK-SAME: (ptr [[DST:%.*]], ptr [[P:%.*]]) #[[ATTR0]] {
 157 ; CHECK-NEXT:  entry:
 158 ; CHECK-NEXT:    [[TMP0:%.*]] = load <4 x float>, ptr [[P]], align 4
 159 ; CHECK-NEXT:    [[TMP1:%.*]] = fdiv <4 x float> [[TMP0]], <float 1.050000e+01, float 1.050000e+01, float 1.050000e+01, float 1.050000e+01>
 160 ; CHECK-NEXT:    store <4 x float> [[TMP1]], ptr [[DST]], align 4
 161 ; CHECK-NEXT:    ret void
 162 ;
 163 ; DEFAULT-LABEL: define void @fp_div
 164 ; DEFAULT-SAME: (ptr [[DST:%.*]], ptr [[P:%.*]]) #[[ATTR0]] {
 165 ; DEFAULT-NEXT:  entry:
 166 ; DEFAULT-NEXT:    [[TMP0:%.*]] = load <4 x float>, ptr [[P]], align 4
 167 ; DEFAULT-NEXT:    [[TMP1:%.*]] = fdiv <4 x float> [[TMP0]], <float 1.050000e+01, float 1.050000e+01, float 1.050000e+01, float 1.050000e+01>
 168 ; DEFAULT-NEXT:    store <4 x float> [[TMP1]], ptr [[DST]], align 4
 169 ; DEFAULT-NEXT:    ret void
 170 ;
 171 entry:
 172   %e0 = load float, ptr %p, align 4
 173   %pe1 = getelementptr inbounds float, ptr %p, i64 1
 174   %e1 = load float, ptr %pe1, align 4
 175   %pe2 = getelementptr inbounds float, ptr %p, i64 2
 176   %e2 = load float, ptr %pe2, align 4
 177   %pe3 = getelementptr inbounds float, ptr %p, i64 3
 178   %e3 = load float, ptr %pe3, align 4
 179
 180   %a0 = fdiv float %e0, 10.5
 181   %a1 = fdiv float %e1, 10.5
 182   %a2 = fdiv float %e2, 10.5
 183   %a3 = fdiv float %e3, 10.5
 184
 185   store float %a0, ptr %dst, align 4
 186   %pa1 = getelementptr inbounds float, ptr %dst, i64 1
 187   store float %a1, ptr %pa1, align 4
 188   %pa2 = getelementptr inbounds float, ptr %dst, i64 2
 189   store float %a2, ptr %pa2, align 4
 190   %pa3 = getelementptr inbounds float, ptr %dst, i64 3
 191   store float %a3, ptr %pa3, align 4
 192
 193   ret void
 194 }
 195
 196 declare float @llvm.maxnum.f32(float, float)
 197
 198 define void @fp_max(ptr %dst, ptr %p, ptr %q) {
 199 ; CHECK-LABEL: define void @fp_max
 200 ; CHECK-SAME: (ptr [[DST:%.*]], ptr [[P:%.*]], ptr [[Q:%.*]]) #[[ATTR0]] {
 201 ; CHECK-NEXT:  entry:
 202 ; CHECK-NEXT:    [[TMP0:%.*]] = load <4 x float>, ptr [[P]], align 4
 203 ; CHECK-NEXT:    [[TMP1:%.*]] = load <4 x float>, ptr [[Q]], align 4
 204 ; CHECK-NEXT:    [[TMP2:%.*]] = call <4 x float> @llvm.maxnum.v4f32(<4 x float> [[TMP0]], <4 x float> [[TMP1]])
 205 ; CHECK-NEXT:    store <4 x float> [[TMP2]], ptr [[DST]], align 4
 206 ; CHECK-NEXT:    ret void
 207 ;
 208 ; DEFAULT-LABEL: define void @fp_max
 209 ; DEFAULT-SAME: (ptr [[DST:%.*]], ptr [[P:%.*]], ptr [[Q:%.*]]) #[[ATTR0]] {
 210 ; DEFAULT-NEXT:  entry:
 211 ; DEFAULT-NEXT:    [[TMP0:%.*]] = load <4 x float>, ptr [[P]], align 4
 212 ; DEFAULT-NEXT:    [[TMP1:%.*]] = load <4 x float>, ptr [[Q]], align 4
 213 ; DEFAULT-NEXT:    [[TMP2:%.*]] = call <4 x float> @llvm.maxnum.v4f32(<4 x float> [[TMP0]], <4 x float> [[TMP1]])
 214 ; DEFAULT-NEXT:    store <4 x float> [[TMP2]], ptr [[DST]], align 4
 215 ; DEFAULT-NEXT:    ret void
 216 ;
 217 entry:
 218   %e0 = load float, ptr %p, align 4
 219   %pe1 = getelementptr inbounds float, ptr %p, i64 1
 220   %e1 = load float, ptr %pe1, align 4
 221   %pe2 = getelementptr inbounds float, ptr %p, i64 2
 222   %e2 = load float, ptr %pe2, align 4
 223   %pe3 = getelementptr inbounds float, ptr %p, i64 3
 224   %e3 = load float, ptr %pe3, align 4
 225
 226   %f0 = load float, ptr %q, align 4
 227   %pf1 = getelementptr inbounds float, ptr %q, i64 1
 228   %f1 = load float, ptr %pf1, align 4
 229   %pf2 = getelementptr inbounds float, ptr %q, i64 2
 230   %f2 = load float, ptr %pf2, align 4
 231   %pf3 = getelementptr inbounds float, ptr %q, i64 3
 232   %f3 = load float, ptr %pf3, align 4
 233
 234   %a0 = tail call float @llvm.maxnum.f32(float %e0, float %f0)
 235   %a1 = tail call float @llvm.maxnum.f32(float %e1, float %f1)
 236   %a2 = tail call float @llvm.maxnum.f32(float %e2, float %f2)
 237   %a3 = tail call float @llvm.maxnum.f32(float %e3, float %f3)
 238
 239   store float %a0, ptr %dst, align 4
 240   %pa1 = getelementptr inbounds float, ptr %dst, i64 1
 241   store float %a1, ptr %pa1, align 4
 242   %pa2 = getelementptr inbounds float, ptr %dst, i64 2
 243   store float %a2, ptr %pa2, align 4
 244   %pa3 = getelementptr inbounds float, ptr %dst, i64 3
 245   store float %a3, ptr %pa3, align 4
 246
 247   ret void
 248 }
 249
 250 declare float @llvm.minnum.f32(float, float)
 251
 252 define void @fp_min(ptr %dst, ptr %p) {
 253 ; CHECK-LABEL: define void @fp_min
 254 ; CHECK-SAME: (ptr [[DST:%.*]], ptr [[P:%.*]]) #[[ATTR0]] {
 255 ; CHECK-NEXT:  entry:
 256 ; CHECK-NEXT:    [[TMP0:%.*]] = load <4 x float>, ptr [[P]], align 4
 257 ; CHECK-NEXT:    [[TMP1:%.*]] = call <4 x float> @llvm.minnum.v4f32(<4 x float> [[TMP0]], <4 x float> <float 1.250000e+00, float 1.250000e+00, float 1.250000e+00, float 1.250000e+00>)
 258 ; CHECK-NEXT:    store <4 x float> [[TMP1]], ptr [[DST]], align 4
 259 ; CHECK-NEXT:    ret void
 260 ;
 261 ; DEFAULT-LABEL: define void @fp_min
 262 ; DEFAULT-SAME: (ptr [[DST:%.*]], ptr [[P:%.*]]) #[[ATTR0]] {
 263 ; DEFAULT-NEXT:  entry:
 264 ; DEFAULT-NEXT:    [[TMP0:%.*]] = load <4 x float>, ptr [[P]], align 4
 265 ; DEFAULT-NEXT:    [[TMP1:%.*]] = call <4 x float> @llvm.minnum.v4f32(<4 x float> [[TMP0]], <4 x float> <float 1.250000e+00, float 1.250000e+00, float 1.250000e+00, float 1.250000e+00>)
 266 ; DEFAULT-NEXT:    store <4 x float> [[TMP1]], ptr [[DST]], align 4
 267 ; DEFAULT-NEXT:    ret void
 268 ;
 269 entry:
 270   %e0 = load float, ptr %p, align 4
 271   %pe1 = getelementptr inbounds float, ptr %p, i64 1
 272   %e1 = load float, ptr %pe1, align 4
 273   %pe2 = getelementptr inbounds float, ptr %p, i64 2
 274   %e2 = load float, ptr %pe2, align 4
 275   %pe3 = getelementptr inbounds float, ptr %p, i64 3
 276   %e3 = load float, ptr %pe3, align 4
 277
 278   %a0 = tail call float @llvm.minnum.f32(float %e0, float 1.25)
 279   %a1 = tail call float @llvm.minnum.f32(float %e1, float 1.25)
 280   %a2 = tail call float @llvm.minnum.f32(float %e2, float 1.25)
 281   %a3 = tail call float @llvm.minnum.f32(float %e3, float 1.25)
 282
 283   store float %a0, ptr %dst, align 4
 284   %pa1 = getelementptr inbounds float, ptr %dst, i64 1
 285   store float %a1, ptr %pa1, align 4
 286   %pa2 = getelementptr inbounds float, ptr %dst, i64 2
 287   store float %a2, ptr %pa2, align 4
 288   %pa3 = getelementptr inbounds float, ptr %dst, i64 3
 289   store float %a3, ptr %pa3, align 4
 290
 291   ret void
 292 }
 293
 294 declare i32 @llvm.fptosi.sat.i32.f32(float)
 295
 296 define void @fp_convert(ptr %dst, ptr %p) {
 297 ; CHECK-LABEL: define void @fp_convert
 298 ; CHECK-SAME: (ptr [[DST:%.*]], ptr [[P:%.*]]) #[[ATTR0]] {
 299 ; CHECK-NEXT:  entry:
 300 ; CHECK-NEXT:    [[TMP0:%.*]] = load <4 x float>, ptr [[P]], align 4
 301 ; CHECK-NEXT:    [[TMP1:%.*]] = call <4 x i32> @llvm.fptosi.sat.v4i32.v4f32(<4 x float> [[TMP0]])
 302 ; CHECK-NEXT:    store <4 x i32> [[TMP1]], ptr [[DST]], align 4
 303 ; CHECK-NEXT:    ret void
 304 ;
 305 ; DEFAULT-LABEL: define void @fp_convert
 306 ; DEFAULT-SAME: (ptr [[DST:%.*]], ptr [[P:%.*]]) #[[ATTR0]] {
 307 ; DEFAULT-NEXT:  entry:
 308 ; DEFAULT-NEXT:    [[TMP0:%.*]] = load <4 x float>, ptr [[P]], align 4
 309 ; DEFAULT-NEXT:    [[TMP1:%.*]] = call <4 x i32> @llvm.fptosi.sat.v4i32.v4f32(<4 x float> [[TMP0]])
 310 ; DEFAULT-NEXT:    store <4 x i32> [[TMP1]], ptr [[DST]], align 4
 311 ; DEFAULT-NEXT:    ret void
 312 ;
 313 entry:
 314   %e0 = load float, ptr %p, align 4
 315   %pe1 = getelementptr inbounds float, ptr %p, i64 1
 316   %e1 = load float, ptr %pe1, align 4
 317   %pe2 = getelementptr inbounds float, ptr %p, i64 2
 318   %e2 = load float, ptr %pe2, align 4
 319   %pe3 = getelementptr inbounds float, ptr %p, i64 3
 320   %e3 = load float, ptr %pe3, align 4
 321
 322   %a0 = tail call i32 @llvm.fptosi.sat.i32.f32(float %e0)
 323   %a1 = tail call i32 @llvm.fptosi.sat.i32.f32(float %e1)
 324   %a2 = tail call i32 @llvm.fptosi.sat.i32.f32(float %e2)
 325   %a3 = tail call i32 @llvm.fptosi.sat.i32.f32(float %e3)
 326
 327   store i32 %a0, ptr %dst, align 4
 328   %pa1 = getelementptr inbounds i32, ptr %dst, i64 1
 329   store i32 %a1, ptr %pa1, align 4
 330   %pa2 = getelementptr inbounds i32, ptr %dst, i64 2
 331   store i32 %a2, ptr %pa2, align 4
 332   %pa3 = getelementptr inbounds i32, ptr %dst, i64 3
 333   store i32 %a3, ptr %pa3, align 4
 334
 335   ret void
 336 }