test/CodeGen/NVPTX/fma-assoc.ll

   1 ; RUN: llc < %s -march=nvptx -mcpu=sm_20 -fp-contract=fast | FileCheck %s
   2
   3 define ptx_device float @t1_f32(float %x, float %y, float %z,
   4                                 float %u, float %v) {
   5 ; CHECK: fma.rn.f32 %f{{[0-9]+}}, %f{{[0-9]+}}, %f{{[0-9]+}}, %f{{[0-9]+}};
   6 ; CHECK: fma.rn.f32 %f{{[0-9]+}}, %f{{[0-9]+}}, %f{{[0-9]+}}, %f{{[0-9]+}};
   7 ; CHECK: ret;
   8   %a = fmul float %x, %y
   9   %b = fmul float %u, %v
  10   %c = fadd float %a, %b
  11   %d = fadd float %c, %z
  12   ret float %d
  13 }
  14
  15 define ptx_device double @t1_f64(double %x, double %y, double %z,
  16                                  double %u, double %v) {
  17 ; CHECK: fma.rn.f64 %fd{{[0-9]+}}, %fd{{[0-9]+}}, %fd{{[0-9]+}}, %fd{{[0-9]+}};
  18 ; CHECK: fma.rn.f64 %fd{{[0-9]+}}, %fd{{[0-9]+}}, %fd{{[0-9]+}}, %fd{{[0-9]+}};
  19 ; CHECK: ret;
  20   %a = fmul double %x, %y
  21   %b = fmul double %u, %v
  22   %c = fadd double %a, %b
  23   %d = fadd double %c, %z
  24   ret double %d
  25 }
  26
  27 define double @two_choices(double %val1, double %val2) {
  28 ; CHECK-LABEL: two_choices(
  29 ; CHECK: mul.f64
  30 ; CHECK-NOT: mul.f64
  31 ; CHECK: fma.rn.f64
  32   %1 = fmul double %val1, %val2
  33   %2 = fmul double %1, %1
  34   %3 = fadd double %1, %2
  35
  36   ret double %3
  37 }
  38