Back to PBT run

test.mlir

// RUN: loom-raise-opt --loom-materialize-fmuladd=shape=fused %s | FileCheck %s

llvm.func @envelope_0(%x: f64, %y: f64, %z: f64) -> f64
    attributes {fp_contract = "off"} {
  %r0 = llvm.intr.fmuladd(%x, %y, %z) {fastmathFlags = #llvm.fastmath<nnan, ninf, nsz>} : (f64, f64, f64) -> f64
  %r1 = llvm.intr.fmuladd(%r0, %y, %z) {fastmathFlags = #llvm.fastmath<nnan, ninf, nsz>} : (f64, f64, f64) -> f64
  llvm.return %r1 : f64
}

llvm.func @envelope_1(%x: f32, %y: f32, %z: f32) -> f32
    attributes {denormal_fpenv = #llvm.denormal_fpenv<default_output_mode = ieee, default_input_mode = ieee, float_output_mode = ieee, float_input_mode = ieee>} {
  %r0 = llvm.intr.fmuladd(%x, %y, %z) {fastmathFlags = #llvm.fastmath<nnan, contract>} : (f32, f32, f32) -> f32
  llvm.return %r0 : f32
}

llvm.func @envelope_2(%x: f32, %y: f32, %z: f32) -> f32
    attributes {denormal_fpenv = #llvm.denormal_fpenv<default_output_mode = ieee, default_input_mode = ieee, float_output_mode = ieee, float_input_mode = ieee>} {
  %r0 = llvm.intr.fmuladd(%x, %y, %z) {fastmathFlags = #llvm.fastmath<nnan, contract>} : (f32, f32, f32) -> f32
  llvm.return %r0 : f32
}


// CHECK: module {
// CHECK-NEXT:   llvm.func @envelope_0(%arg0: f64, %arg1: f64, %arg2: f64) -> f64 attributes {fp_contract = "off"} {
// CHECK-NEXT:     %0 = math.fma %arg0, %arg1, %arg2 fastmath<nnan,ninf,nsz> : f64
// CHECK-NEXT:     %1 = math.fma %0, %arg1, %arg2 fastmath<nnan,ninf,nsz> : f64
// CHECK-NEXT:     llvm.return %1 : f64
// CHECK-NEXT:   }
// CHECK-NEXT:   llvm.func @envelope_1(%arg0: f32, %arg1: f32, %arg2: f32) -> f32 attributes {denormal_fpenv = #llvm.denormal_fpenv<default_output_mode = ieee, default_input_mode = ieee, float_output_mode = ieee, float_input_mode = ieee>} {
// CHECK-NEXT:     %0 = math.fma %arg0, %arg1, %arg2 fastmath<nnan,contract> : f32
// CHECK-NEXT:     llvm.return %0 : f32
// CHECK-NEXT:   }
// CHECK-NEXT:   llvm.func @envelope_2(%arg0: f32, %arg1: f32, %arg2: f32) -> f32 attributes {denormal_fpenv = #llvm.denormal_fpenv<default_output_mode = ieee, default_input_mode = ieee, float_output_mode = ieee, float_input_mode = ieee>} {
// CHECK-NEXT:     %0 = math.fma %arg0, %arg1, %arg2 fastmath<nnan,contract> : f32
// CHECK-NEXT:     llvm.return %0 : f32
// CHECK-NEXT:   }
// CHECK-NEXT: }
// CHECK-EMPTY: