llvm/llvm/test/CodeGen/AArch64/arm64-fma-combines.ll

; RUN: llc < %s -O=3 -mtriple=arm64-apple-ios -mcpu=cyclone -mattr=+fullfp16 -enable-unsafe-fp-math -verify-machineinstrs | FileCheck %s

define void @foo_2d(ptr %src) {
; CHECK-LABEL: %entry
; CHECK: fmul {{d[0-9]+}}, {{d[0-9]+}}, {{d[0-9]+}}
; CHECK: fmadd {{d[0-9]+}}, {{d[0-9]+}}, {{d[0-9]+}}, {{d[0-9]+}}
entry:
  %arrayidx1 = getelementptr inbounds double, ptr %src, i64 5
  %arrayidx2 = getelementptr inbounds double, ptr %src, i64 11
  %tmp1 = load double, ptr %arrayidx2, align 8
  %tmp2 = load double, ptr %arrayidx1, align 8
  %fmul = fmul fast double %tmp1, %tmp1
  %fmul2 = fmul fast double %tmp2, 0x3F94AFD6A052BF5B
  %fadd = fadd fast double %fmul, %fmul2
  br label %for.body

; CHECK-LABEL: %for.body
; CHECK: fmla.2d {{v[0-9]+}}, {{v[0-9]+}}, {{v[0-9]+}}
; CHECK: fmla.2d {{v[0-9]+}}, {{v[0-9]+}}, {{v[0-9]+}}[0]
; CHECK: fmadd {{d[0-9]+}}, {{d[0-9]+}}, {{d[0-9]+}}, {{d[0-9]+}}
for.body:                                         ; preds = %for.body, %entry
  %indvars.iv = phi i64 [ 0, %entry ], [ %indvars.iv.next, %for.body ]
  %indvars.iv.next = add nuw nsw i64 %indvars.iv, 1
  %arrayidx3 = getelementptr inbounds double, ptr %src, i64 %indvars.iv.next
  %tmp3 = load double, ptr %arrayidx3, align 8
  %add = fadd fast double %tmp3, %tmp3
  %mul = fmul fast double %add, %fadd
  %e1 = insertelement <2 x double> undef, double %add, i32 0
  %e2 = insertelement <2 x double> %e1, double %add, i32 1
  %add2 = fadd fast <2 x double> %e2, <double 3.000000e+00, double -3.000000e+00>
  %e3 = insertelement <2 x double> undef, double %mul, i32 0
  %e4 = insertelement <2 x double> %e3, double %mul, i32 1
  %mul2 = fmul fast <2 x double> %add2,<double 3.000000e+00, double -3.000000e+00>
  %e5 = insertelement <2 x double> undef, double %add, i32 0
  %e6 = insertelement <2 x double> %e5, double %add, i32 1
  %add3 = fadd fast  <2 x double> %mul2, <double 3.000000e+00, double -3.000000e+00>
  %mulx = fmul fast <2 x double> %add2, %e2
  %addx = fadd fast  <2 x double> %mulx, %e4
  %e7 = insertelement <2 x double> undef, double %mul, i32 0
  %e8 = insertelement <2 x double> %e7, double %mul, i32 1
  %e9 = fmul fast <2 x double>  %addx, %add3
  store <2 x double> %e9, ptr %arrayidx1, align 8
  %e10 = extractelement <2 x double> %add3, i32 0
  %mul3 = fmul fast double %mul, %e10
  %add4 = fadd fast double %mul3, %mul
  store double %add4, ptr %arrayidx2, align 8
  %exitcond = icmp eq i64 %indvars.iv.next, 25
  br i1 %exitcond, label %for.end, label %for.body

for.end:                                          ; preds = %for.body
  ret void
}
define void @foo_2s(ptr %src) {
entry:
  %arrayidx1 = getelementptr inbounds float, ptr %src, i64 5
  %arrayidx2 = getelementptr inbounds float, ptr %src, i64 11
  br label %for.body

; CHECK-LABEL: %for.body
; CHECK: fmla.2s {{v[0-9]+}}, {{v[0-9]+}}, {{v[0-9]+}}
; CHECK: fmla.2s {{v[0-9]+}}, {{v[0-9]+}}, {{v[0-9]+}}[0]
; CHECK: fmadd {{s[0-9]+}}, {{s[0-9]+}}, {{s[0-9]+}}, {{s[0-9]+}}
for.body:                                         ; preds = %for.body, %entry
  %indvars.iv = phi i64 [ 0, %entry ], [ %indvars.iv.next, %for.body ]
  %indvars.iv.next = add nuw nsw i64 %indvars.iv, 1
  %arrayidx3 = getelementptr inbounds float, ptr %src, i64 %indvars.iv.next
  %tmp1 = load float, ptr %arrayidx3, align 8
  %add = fadd fast float %tmp1, %tmp1
  %mul = fmul fast float %add, %add
  %e1 = insertelement <2 x float> undef, float %add, i32 0
  %e2 = insertelement <2 x float> %e1, float %add, i32 1
  %add2 = fadd fast <2 x float> %e2, <float 3.000000e+00, float -3.000000e+00>
  %e3 = insertelement <2 x float> undef, float %mul, i32 0
  %e4 = insertelement <2 x float> %e3, float %mul, i32 1
  %mul2 = fmul fast <2 x float> %add2,<float 3.000000e+00, float -3.000000e+00>
  %e5 = insertelement <2 x float> undef, float %add, i32 0
  %e6 = insertelement <2 x float> %e5, float %add, i32 1
  %add3 = fadd fast  <2 x float> %mul2, <float 3.000000e+00, float -3.000000e+00>
  %mulx = fmul fast <2 x float> %add2, %e2
  %addx = fadd fast  <2 x float> %mulx, %e4
  %e7 = insertelement <2 x float> undef, float %mul, i32 0
  %e8 = insertelement <2 x float> %e7, float %mul, i32 1
  %e9 = fmul fast <2 x float>  %addx, %add3
  store <2 x float> %e9, ptr %arrayidx1, align 8
  %e10 = extractelement <2 x float> %add3, i32 0
  %mul3 = fmul fast float %mul, %e10
  %add4 = fadd fast float %mul3, %mul
  store float %add4, ptr %arrayidx2, align 8
  %exitcond = icmp eq i64 %indvars.iv.next, 25
  br i1 %exitcond, label %for.end, label %for.body

for.end:                                          ; preds = %for.body
  ret void
}
define void @foo_4s(ptr %src) {
entry:
  %arrayidx1 = getelementptr inbounds float, ptr %src, i64 5
  %arrayidx2 = getelementptr inbounds float, ptr %src, i64 11
  br label %for.body

; CHECK-LABEL: %for.body
; CHECK: fmla.4s {{v[0-9]+}}, {{v[0-9]+}}, {{v[0-9]+}}
; CHECK: fmla.4s {{v[0-9]+}}, {{v[0-9]+}}, {{v[0-9]+}}[0]
for.body:                                         ; preds = %for.body, %entry
  %indvars.iv = phi i64 [ 0, %entry ], [ %indvars.iv.next, %for.body ]
  %indvars.iv.next = add nuw nsw i64 %indvars.iv, 1
  %arrayidx3 = getelementptr inbounds float, ptr %src, i64 %indvars.iv.next
  %tmp1 = load float, ptr %arrayidx3, align 8
  %add = fadd fast float %tmp1, %tmp1
  %mul = fmul fast float %add, %add
  %e1 = insertelement <4 x float> undef, float %add, i32 0
  %e2 = insertelement <4 x float> %e1, float %add, i32 1
  %add2 = fadd fast <4 x float> %e2, <float 3.000000e+00, float -3.000000e+00, float 5.000000e+00, float 7.000000e+00>
  %e3 = insertelement <4 x float> undef, float %mul, i32 0
  %e4 = insertelement <4 x float> %e3, float %mul, i32 1
  %mul2 = fmul fast <4 x float> %add2,<float 3.000000e+00, float -3.000000e+00, float 5.000000e+00, float 7.000000e+00>
  %e5 = insertelement <4 x float> undef, float %add, i32 0
  %e6 = insertelement <4 x float> %e5, float %add, i32 1
  %add3 = fadd fast  <4 x float> %mul2, <float 3.000000e+00, float -3.000000e+00, float 5.000000e+00, float 7.000000e+00> 
  %mulx = fmul fast <4 x float> %add2, %e2
  %addx = fadd fast  <4 x float> %mulx, %e4
  %e7 = insertelement <4 x float> undef, float %mul, i32 0
  %e8 = insertelement <4 x float> %e7, float %mul, i32 1
  %e9 = fmul fast <4 x float>  %addx, %add3
  store <4 x float> %e9, ptr %arrayidx1, align 8
  %e10 = extractelement <4 x float> %add3, i32 0
  %mul3 = fmul fast float %mul, %e10
  store float %mul3, ptr %arrayidx2, align 8
  %exitcond = icmp eq i64 %indvars.iv.next, 25
  br i1 %exitcond, label %for.end, label %for.body

for.end:                                          ; preds = %for.body
  ret void
}

define void @indexed_2s(<2 x float> %shuf, <2 x float> %add,
                        ptr %pmul, ptr %pret) {
; CHECK-LABEL: %entry
; CHECK: for.body
; CHECK: fmla.2s {{v[0-9]+}}, {{v[0-9]+}}, {{v[0-9]+}}[0]
;
entry:
  %shuffle = shufflevector <2 x float> %shuf, <2 x float> undef, <2 x i32> zeroinitializer
  br label %for.body

for.body:
  %i = phi i64 [ 0, %entry ], [ %inext, %for.body ]
  %pmul_i = getelementptr inbounds <2 x float>, ptr %pmul, i64 %i
  %pret_i = getelementptr inbounds <2 x float>, ptr %pret, i64 %i

  %mul_i = load <2 x float>, ptr %pmul_i

  %mul = fmul fast <2 x float> %mul_i, %shuffle
  %muladd = fadd fast <2 x float> %mul, %add

  store <2 x float> %muladd, ptr %pret_i, align 16
  %inext = add i64 %i, 1
  br label %for.body
}

define void @indexed_2d(<2 x double> %shuf, <2 x double> %add,
                        ptr %pmul, ptr %pret) {
; CHECK-LABEL: %entry
; CHECK: for.body
; CHECK: fmla.2d {{v[0-9]+}}, {{v[0-9]+}}, {{v[0-9]+}}[0]
;
entry:
  %shuffle = shufflevector <2 x double> %shuf, <2 x double> undef, <2 x i32> zeroinitializer
  br label %for.body

for.body:
  %i = phi i64 [ 0, %entry ], [ %inext, %for.body ]
  %pmul_i = getelementptr inbounds <2 x double>, ptr %pmul, i64 %i
  %pret_i = getelementptr inbounds <2 x double>, ptr %pret, i64 %i

  %mul_i = load <2 x double>, ptr %pmul_i

  %mul = fmul fast <2 x double> %mul_i, %shuffle
  %muladd = fadd fast <2 x double> %mul, %add

  store <2 x double> %muladd, ptr %pret_i, align 16
  %inext = add i64 %i, 1
  br label %for.body
}

define void @indexed_4s(<4 x float> %shuf, <4 x float> %add,
                        ptr %pmul, ptr %pret) {
; CHECK-LABEL: %entry
; CHECK: for.body
; CHECK: fmla.4s {{v[0-9]+}}, {{v[0-9]+}}, {{v[0-9]+}}[0]
;
entry:
  %shuffle = shufflevector <4 x float> %shuf, <4 x float> undef, <4 x i32> zeroinitializer
  br label %for.body

for.body:
  %i = phi i64 [ 0, %entry ], [ %inext, %for.body ]
  %pmul_i = getelementptr inbounds <4 x float>, ptr %pmul, i64 %i
  %pret_i = getelementptr inbounds <4 x float>, ptr %pret, i64 %i

  %mul_i = load <4 x float>, ptr %pmul_i

  %mul = fmul fast <4 x float> %mul_i, %shuffle
  %muladd = fadd fast <4 x float> %mul, %add

  store <4 x float> %muladd, ptr %pret_i, align 16
  %inext = add i64 %i, 1
  br label %for.body
}

define void @indexed_4h(<4 x half> %shuf, <4 x half> %add,
                        ptr %pmul, ptr %pret) {
; CHECK-LABEL: %entry
; CHECK: for.body
; CHECK: fmla.4h {{v[0-9]+}}, {{v[0-9]+}}, {{v[0-9]+}}[0]
;
entry:
  %shuffle = shufflevector <4 x half> %shuf, <4 x half> undef, <4 x i32> zeroinitializer
  br label %for.body

for.body:
  %i = phi i64 [ 0, %entry ], [ %inext, %for.body ]
  %pmul_i = getelementptr inbounds <4 x half>, ptr %pmul, i64 %i
  %pret_i = getelementptr inbounds <4 x half>, ptr %pret, i64 %i

  %mul_i = load <4 x half>, ptr %pmul_i

  %mul = fmul fast <4 x half> %mul_i, %shuffle
  %muladd = fadd fast <4 x half> %mul, %add

  store <4 x half> %muladd, ptr %pret_i, align 16
  %inext = add i64 %i, 1
  br label %for.body
}

define void @indexed_8h(<8 x half> %shuf, <8 x half> %add,
                        ptr %pmul, ptr %pret) {
; CHECK-LABEL: %entry
; CHECK: for.body
; CHECK: fmla.8h {{v[0-9]+}}, {{v[0-9]+}}, {{v[0-9]+}}[0]
;
entry:
  %shuffle = shufflevector <8 x half> %shuf, <8 x half> undef, <8 x i32> zeroinitializer
  br label %for.body

for.body:
  %i = phi i64 [ 0, %entry ], [ %inext, %for.body ]
  %pmul_i = getelementptr inbounds <8 x half>, ptr %pmul, i64 %i
  %pret_i = getelementptr inbounds <8 x half>, ptr %pret, i64 %i

  %mul_i = load <8 x half>, ptr %pmul_i

  %mul = fmul fast <8 x half> %mul_i, %shuffle
  %muladd = fadd fast <8 x half> %mul, %add

  store <8 x half> %muladd, ptr %pret_i, align 16
  %inext = add i64 %i, 1
  br label %for.body
}