38 lines · plain
1// RUN: mlir-opt %s --split-input-file --math-uplift-to-fma | FileCheck %s2 3// No uplifting without fastmath flags.4// CHECK-LABEL: func @test5// CHECK-SAME: (%[[ARG1:.*]]: f32, %[[ARG2:.*]]: f32, %[[ARG3:.*]]: f32)6// CHECK: %[[V1:.*]] = arith.mulf %[[ARG1]], %[[ARG2]]7// CHECK: %[[V2:.*]] = arith.addf %[[V1]], %[[ARG3]]8// CHECK: return %[[V2]]9func.func @test(%arg1: f32, %arg2: f32, %arg3: f32) -> f32 {10 %1 = arith.mulf %arg1, %arg2 : f3211 %2 = arith.addf %1, %arg3 : f3212 return %2 : f3213}14 15// -----16 17// CHECK-LABEL: func @test18// CHECK-SAME: (%[[ARG1:.*]]: f32, %[[ARG2:.*]]: f32, %[[ARG3:.*]]: f32)19// CHECK: %[[RES:.*]] = math.fma %[[ARG1]], %[[ARG2]], %[[ARG3]] fastmath<fast> : f3220// CHECK: return %[[RES]]21func.func @test(%arg1: f32, %arg2: f32, %arg3: f32) -> f32 {22 %1 = arith.mulf %arg1, %arg2 fastmath<fast> : f3223 %2 = arith.addf %1, %arg3 fastmath<fast> : f3224 return %2 : f3225}26 27// -----28 29// CHECK-LABEL: func @test30// CHECK-SAME: (%[[ARG1:.*]]: f32, %[[ARG2:.*]]: f32, %[[ARG3:.*]]: f32)31// CHECK: %[[RES:.*]] = math.fma %[[ARG1]], %[[ARG2]], %[[ARG3]] fastmath<contract> : f3232// CHECK: return %[[RES]]33func.func @test(%arg1: f32, %arg2: f32, %arg3: f32) -> f32 {34 %1 = arith.mulf %arg1, %arg2 fastmath<fast> : f3235 %2 = arith.addf %arg3, %1 fastmath<contract> : f3236 return %2 : f3237}38