xref: /llvm-project/mlir/test/Dialect/Math/uplift-to-fma.mlir (revision ee8b8d6b58e1ff45d1bcbad38c6cf458f872b38d) 
1// RUN: mlir-opt %s --split-input-file --math-uplift-to-fma | FileCheck %s
2
3// No uplifting without fastmath flags.
4// CHECK-LABEL: func @test
5//  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 : f32
11  %2 = arith.addf %1, %arg3 : f32
12  return %2 : f32
13}
14
15// -----
16
17// CHECK-LABEL: func @test
18//  CHECK-SAME: (%[[ARG1:.*]]: f32, %[[ARG2:.*]]: f32, %[[ARG3:.*]]: f32)
19//       CHECK: %[[RES:.*]] = math.fma %[[ARG1]], %[[ARG2]], %[[ARG3]] fastmath<fast> : f32
20//       CHECK: return %[[RES]]
21func.func @test(%arg1: f32, %arg2: f32, %arg3: f32) -> f32 {
22  %1 = arith.mulf %arg1, %arg2 fastmath<fast> : f32
23  %2 = arith.addf %1, %arg3 fastmath<fast> : f32
24  return %2 : f32
25}
26
27// -----
28
29// CHECK-LABEL: func @test
30//  CHECK-SAME: (%[[ARG1:.*]]: f32, %[[ARG2:.*]]: f32, %[[ARG3:.*]]: f32)
31//       CHECK: %[[RES:.*]] = math.fma %[[ARG1]], %[[ARG2]], %[[ARG3]] fastmath<contract> : f32
32//       CHECK: return %[[RES]]
33func.func @test(%arg1: f32, %arg2: f32, %arg3: f32) -> f32 {
34  %1 = arith.mulf %arg1, %arg2 fastmath<fast> : f32
35  %2 = arith.addf %arg3, %1 fastmath<contract> : f32
36  return %2 : f32
37}
38