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
74
75
76
77
78
79
80
81
82
83
|
// RUN: mlir-opt %s -convert-cf-to-llvm -split-input-file | FileCheck %s
// Unstructured control flow is converted, but the enclosing op is not
// converted.
// CHECK-LABEL: func.func @cf_br(
// CHECK-SAME: %[[arg0:.*]]: index) -> index {
// CHECK: %[[cast0:.*]] = builtin.unrealized_conversion_cast %[[arg0]] : index to i64
// CHECK: llvm.br ^[[bb1:.*]](%[[cast0]] : i64)
// CHECK: ^[[bb1]](%[[arg1:.*]]: i64):
// CHECK: %[[cast1:.*]] = builtin.unrealized_conversion_cast %[[arg1]] : i64 to index
// CHECK: return %[[cast1]] : index
// CHECK: }
func.func @cf_br(%arg0: index) -> index {
cf.br ^bb1(%arg0 : index)
^bb1(%arg1: index):
return %arg1 : index
}
// -----
// func.func and func.return types match. No unrealized_conversion_cast is
// needed.
// CHECK-LABEL: func.func @cf_br_type_match(
// CHECK-SAME: %[[arg0:.*]]: i64) -> i64 {
// CHECK: llvm.br ^[[bb1:.*]](%[[arg0:.*]] : i64)
// CHECK: ^[[bb1]](%[[arg1:.*]]: i64):
// CHECK: return %[[arg1]] : i64
// CHECK: }
func.func @cf_br_type_match(%arg0: i64) -> i64 {
cf.br ^bb1(%arg0 : i64)
^bb1(%arg1: i64):
return %arg1 : i64
}
// -----
// Test case for cf.cond_br.
// CHECK-LABEL: func.func @cf_cond_br
// CHECK-COUNT-2: unrealized_conversion_cast {{.*}} : index to i64
// CHECK: llvm.cond_br %{{.*}}, ^{{.*}}(%{{.*}} : i64), ^{{.*}}(%{{.*}} : i64)
// CHECK: ^{{.*}}(%{{.*}}: i64):
// CHECK: unrealized_conversion_cast {{.*}} : i64 to index
// CHECK: ^{{.*}}(%{{.*}}: i64):
// CHECK: unrealized_conversion_cast {{.*}} : i64 to index
func.func @cf_cond_br(%cond: i1, %a: index, %b: index) -> index {
cf.cond_br %cond, ^bb1(%a : index), ^bb2(%b : index)
^bb1(%arg1: index):
return %arg1 : index
^bb2(%arg2: index):
return %arg2 : index
}
// -----
// Unreachable block (and IR in general) is not converted during a dialect
// conversion.
// CHECK-LABEL: func.func @unreachable_block()
// CHECK: return
// CHECK: ^[[bb1:.*]](%[[arg0:.*]]: index):
// CHECK: cf.br ^[[bb1]](%[[arg0]] : index)
func.func @unreachable_block() {
return
^bb1(%arg0: index):
cf.br ^bb1(%arg0 : index)
}
// -----
// Test case for cf.cond_br with weights.
// CHECK-LABEL: func.func @cf_cond_br_with_weights(
func.func @cf_cond_br_with_weights(%cond: i1, %a: index, %b: index) -> index {
// CHECK: llvm.cond_br %{{.*}} weights([90, 10]), ^bb1(%{{.*}} : i64), ^bb2(%{{.*}} : i64)
cf.cond_br %cond, ^bb1(%a : index), ^bb2(%b : index) {branch_weights = array<i32: 90, 10>}
^bb1(%arg1: index):
return %arg1 : index
^bb2(%arg2: index):
return %arg2 : index
}
|
