llvm/test/CodeGen/AArch64/fp-dp3.ll

; NOTE: Assertions have been autogenerated by utils/update_llc_test_checks.py UTC_ARGS: --version 6
; RUN: llc -mtriple=aarch64-none-linux < %s | FileCheck %s

declare float @llvm.fma.f32(float, float, float)
declare double @llvm.fma.f64(double, double, double)

define float @test_fmadd(float %a, float %b, float %c) {
; CHECK-LABEL: test_fmadd:
; CHECK:       // %bb.0:
; CHECK-NEXT:    fmadd s0, s0, s1, s2
; CHECK-NEXT:    ret
  %val = call float @llvm.fma.f32(float %a, float %b, float %c)
  ret float %val
}

define float @test_fmsub(float %a, float %b, float %c) {
; CHECK-LABEL: test_fmsub:
; CHECK:       // %bb.0:
; CHECK-NEXT:    fmsub s0, s0, s1, s2
; CHECK-NEXT:    ret
  %nega = fsub float -0.0, %a
  %val = call float @llvm.fma.f32(float %nega, float %b, float %c)
  ret float %val
}

define float @test_fnmadd(float %a, float %b, float %c) {
; CHECK-LABEL: test_fnmadd:
; CHECK:       // %bb.0:
; CHECK-NEXT:    fnmadd s0, s0, s1, s2
; CHECK-NEXT:    ret
  %nega = fsub float -0.0, %a
  %negc = fsub float -0.0, %c
  %val = call float @llvm.fma.f32(float %nega, float %b, float %negc)
  ret float %val
}

define float @test_fnmsub(float %a, float %b, float %c) {
; CHECK-LABEL: test_fnmsub:
; CHECK:       // %bb.0:
; CHECK-NEXT:    fnmsub s0, s0, s1, s2
; CHECK-NEXT:    ret
  %negc = fsub float -0.0, %c
  %val = call float @llvm.fma.f32(float %a, float %b, float %negc)
  ret float %val
}

define double @testd_fmadd(double %a, double %b, double %c) {
; CHECK-LABEL: testd_fmadd:
; CHECK:       // %bb.0:
; CHECK-NEXT:    fmadd d0, d0, d1, d2
; CHECK-NEXT:    ret
  %val = call double @llvm.fma.f64(double %a, double %b, double %c)
  ret double %val
}

define double @testd_fmsub(double %a, double %b, double %c) {
; CHECK-LABEL: testd_fmsub:
; CHECK:       // %bb.0:
; CHECK-NEXT:    fmsub d0, d0, d1, d2
; CHECK-NEXT:    ret
  %nega = fsub double -0.0, %a
  %val = call double @llvm.fma.f64(double %nega, double %b, double %c)
  ret double %val
}

define double @testd_fnmadd(double %a, double %b, double %c) {
; CHECK-LABEL: testd_fnmadd:
; CHECK:       // %bb.0:
; CHECK-NEXT:    fnmadd d0, d0, d1, d2
; CHECK-NEXT:    ret
  %nega = fsub double -0.0, %a
  %negc = fsub double -0.0, %c
  %val = call double @llvm.fma.f64(double %nega, double %b, double %negc)
  ret double %val
}

define double @testd_fnmsub(double %a, double %b, double %c) {
; CHECK-LABEL: testd_fnmsub:
; CHECK:       // %bb.0:
; CHECK-NEXT:    fnmsub d0, d0, d1, d2
; CHECK-NEXT:    ret
  %negc = fsub double -0.0, %c
  %val = call double @llvm.fma.f64(double %a, double %b, double %negc)
  ret double %val
}

define float @test_fmadd_unfused(float %a, float %b, float %c) {
; CHECK-LABEL: test_fmadd_unfused:
; CHECK:       // %bb.0:
; CHECK-NEXT:    fmul s1, s1, s2
; CHECK-NEXT:    fadd s0, s0, s1
; CHECK-NEXT:    ret
  %prod = fmul float %b, %c
  %sum = fadd float %a, %prod
  ret float %sum
}

define float @test_fmsub_unfused(float %a, float %b, float %c) {
; CHECK-LABEL: test_fmsub_unfused:
; CHECK:       // %bb.0:
; CHECK-NEXT:    fmul s1, s1, s2
; CHECK-NEXT:    fsub s0, s0, s1
; CHECK-NEXT:    ret
  %prod = fmul float %b, %c
  %diff = fsub float %a, %prod
  ret float %diff
}

define float @test_fnmadd_unfused(float %a, float %b, float %c) {
; CHECK-LABEL: test_fnmadd_unfused:
; CHECK:       // %bb.0:
; CHECK-NEXT:    fneg s0, s0
; CHECK-NEXT:    fmul s1, s1, s2
; CHECK-NEXT:    fsub s0, s0, s1
; CHECK-NEXT:    ret
  %nega = fsub float -0.0, %a
  %prod = fmul float %b, %c
  %diff = fsub float %nega, %prod
  ret float %diff
}

define float @test_fnmsub_unfused(float %a, float %b, float %c) {
; CHECK-LABEL: test_fnmsub_unfused:
; CHECK:       // %bb.0:
; CHECK-NEXT:    fmul s1, s1, s2
; CHECK-NEXT:    fsub s0, s1, s0
; CHECK-NEXT:    ret
  %nega = fsub float -0.0, %a
  %prod = fmul float %b, %c
  %sum = fadd float %nega, %prod
  ret float %sum
}

; With the contract flag the multiply and add/subtract fuse to a single op.

define float @test_fmadd_contract(float %a, float %b, float %c) {
; CHECK-LABEL: test_fmadd_contract:
; CHECK:       // %bb.0:
; CHECK-NEXT:    fmadd s0, s1, s2, s0
; CHECK-NEXT:    ret
  %prod = fmul contract float %b, %c
  %sum = fadd contract float %a, %prod
  ret float %sum
}

define float @test_fmsub_contract(float %a, float %b, float %c) {
; CHECK-LABEL: test_fmsub_contract:
; CHECK:       // %bb.0:
; CHECK-NEXT:    fmsub s0, s1, s2, s0
; CHECK-NEXT:    ret
  %prod = fmul contract float %b, %c
  %diff = fsub contract float %a, %prod
  ret float %diff
}

define float @test_fnmadd_contract(float %a, float %b, float %c) {
; CHECK-LABEL: test_fnmadd_contract:
; CHECK:       // %bb.0:
; CHECK-NEXT:    fnmadd s0, s1, s2, s0
; CHECK-NEXT:    ret
  %nega = fsub float -0.0, %a
  %prod = fmul contract float %b, %c
  %diff = fsub contract float %nega, %prod
  ret float %diff
}

define float @test_fnmsub_contract(float %a, float %b, float %c) {
; CHECK-LABEL: test_fnmsub_contract:
; CHECK:       // %bb.0:
; CHECK-NEXT:    fnmsub s0, s1, s2, s0
; CHECK-NEXT:    ret
  %nega = fsub float -0.0, %a
  %prod = fmul contract float %b, %c
  %sum = fadd contract float %nega, %prod
  ret float %sum
}

; Another set of tests that check for multiply single use

define float @test_fmadd_unfused_su(float %a, float %b, float %c) {
; CHECK-LABEL: test_fmadd_unfused_su:
; CHECK:       // %bb.0:
; CHECK-NEXT:    fmul s1, s1, s2
; CHECK-NEXT:    fadd s0, s0, s1
; CHECK-NEXT:    fadd s0, s0, s1
; CHECK-NEXT:    ret
  %prod = fmul float %b, %c
  %sum = fadd float %a, %prod
  %res = fadd float %sum, %prod
  ret float %res
}

define float @test_fmsub_unfused_su(float %a, float %b, float %c) {
; CHECK-LABEL: test_fmsub_unfused_su:
; CHECK:       // %bb.0:
; CHECK-NEXT:    fmul s1, s1, s2
; CHECK-NEXT:    fsub s0, s0, s1
; CHECK-NEXT:    fsub s0, s0, s1
; CHECK-NEXT:    ret
  %prod = fmul float %b, %c
  %diff = fsub float %a, %prod
  %res = fsub float %diff, %prod
  ret float %res
}