aboutsummaryrefslogtreecommitdiff
path: root/mlir/test/Transforms/inlining-dce.mlir
blob: 06504ec39e71d6ff93c6ad15e29bf9662029efb8 (plain)
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
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
// RUN: mlir-opt -allow-unregistered-dialect %s -inline -split-input-file | FileCheck %s

// This file tests the callgraph dead code elimination performed by the inliner.

// Function is already dead.
// CHECK-NOT: func @dead_function
func @dead_function() attributes {sym_visibility = "private"} {
  return
}

// Function becomes dead after inlining.
// CHECK-NOT: func @dead_function_b
func @dead_function_b() attributes {sym_visibility = "private"} {
  return
}

// CHECK: func @live_function()
func @live_function() {
  call @dead_function_b() : () -> ()
  return
}

// Same as above, but a transitive example.

// CHECK: func @live_function_b
func @live_function_b() {
  return
}
// CHECK-NOT: func @dead_function_c
func @dead_function_c() attributes {sym_visibility = "private"} {
  call @live_function_b() : () -> ()
  return
}
// CHECK-NOT: func @dead_function_d
func @dead_function_d() attributes {sym_visibility = "private"} {
  call @dead_function_c() : () -> ()
  call @dead_function_c() : () -> ()
  return
}
// CHECK: func @live_function_c
func @live_function_c() {
  call @dead_function_c() : () -> ()
  call @dead_function_d() : () -> ()
  return
}

// Function is referenced by non-callable top-level user.
// CHECK: func @live_function_d
func @live_function_d() attributes {sym_visibility = "private"} {
  return
}

"live.user"() {use = @live_function_d} : () -> ()

// -----

// This test checks that the inliner can properly handle the deletion of
// functions in different SCCs that are referenced by calls materialized during
// canonicalization.
// CHECK: func @live_function_e
func @live_function_e() {
  call @dead_function_e() : () -> ()
  return
}
// CHECK-NOT: func @dead_function_e
func @dead_function_e() -> () attributes {sym_visibility = "private"} {
  "test.fold_to_call_op"() {callee=@dead_function_f} : () -> ()
  return
}
// CHECK-NOT: func @dead_function_f
func @dead_function_f() attributes {sym_visibility = "private"} {
  return
}