File: uplift-to-fma.mlir

package info (click to toggle)
swiftlang 6.1.3-2
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid
  • size: 2,791,604 kB
  • sloc: cpp: 9,901,740; ansic: 2,201,431; asm: 1,091,827; python: 308,252; objc: 82,166; f90: 80,126; lisp: 38,358; pascal: 25,559; sh: 20,429; ml: 5,058; perl: 4,745; makefile: 4,484; awk: 3,535; javascript: 3,018; xml: 918; fortran: 664; cs: 573; ruby: 396
file content (37 lines) | stat: -rw-r--r-- 1,364 bytes parent folder | download | duplicates (13) 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
 
// RUN: mlir-opt %s --split-input-file --math-uplift-to-fma | FileCheck %s

// No uplifting without fastmath flags.
// CHECK-LABEL: func @test
//  CHECK-SAME: (%[[ARG1:.*]]: f32, %[[ARG2:.*]]: f32, %[[ARG3:.*]]: f32)
//       CHECK: %[[V1:.*]] = arith.mulf %[[ARG1]], %[[ARG2]]
//       CHECK: %[[V2:.*]] = arith.addf %[[V1]], %[[ARG3]]
//       CHECK: return %[[V2]]
func.func @test(%arg1: f32, %arg2: f32, %arg3: f32) -> f32 {
  %1 = arith.mulf %arg1, %arg2 : f32
  %2 = arith.addf %1, %arg3 : f32
  return %2 : f32
}

// -----

// CHECK-LABEL: func @test
//  CHECK-SAME: (%[[ARG1:.*]]: f32, %[[ARG2:.*]]: f32, %[[ARG3:.*]]: f32)
//       CHECK: %[[RES:.*]] = math.fma %[[ARG1]], %[[ARG2]], %[[ARG3]] fastmath<fast> : f32
//       CHECK: return %[[RES]]
func.func @test(%arg1: f32, %arg2: f32, %arg3: f32) -> f32 {
  %1 = arith.mulf %arg1, %arg2 fastmath<fast> : f32
  %2 = arith.addf %1, %arg3 fastmath<fast> : f32
  return %2 : f32
}

// -----

// CHECK-LABEL: func @test
//  CHECK-SAME: (%[[ARG1:.*]]: f32, %[[ARG2:.*]]: f32, %[[ARG3:.*]]: f32)
//       CHECK: %[[RES:.*]] = math.fma %[[ARG1]], %[[ARG2]], %[[ARG3]] fastmath<contract> : f32
//       CHECK: return %[[RES]]
func.func @test(%arg1: f32, %arg2: f32, %arg3: f32) -> f32 {
  %1 = arith.mulf %arg1, %arg2 fastmath<fast> : f32
  %2 = arith.addf %arg3, %1 fastmath<contract> : f32
  return %2 : f32
}