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
|
// RUN: mlir-opt -llvm-legalize-for-export --split-input-file %s | FileCheck %s
// Verifies that duplicate successor with different arguments are deduplicated
// by introducing a new block that forwards its arguments to the original
// successor through an unconditional branch.
// CHECK-LABEL: @repeated_successor_different_args
llvm.func @repeated_successor_different_args(%arg0: i1, %arg1: i32, %arg2: i32) {
// CHECK: llvm.cond_br %{{.*}}, ^[[BB1:.*]]({{.*}}), ^[[BB2:.*]]({{.*}})
llvm.cond_br %arg0, ^bb1(%arg1: i32), ^bb1(%arg2: i32)
// CHECK: ^[[BB1]]({{.*}}):
^bb1(%arg3: i32):
llvm.return
// CHECK: ^[[BB2]](%[[ARG:.*]]: i32):
// CHECK: llvm.br ^[[BB1]](%[[ARG]] : i32)
}
// Verifies that duplicate successors without arguments do not lead to the
// introduction of new blocks during legalization.
// CHECK-LABEL: @repeated_successor_no_args
llvm.func @repeated_successor_no_args(%arg0: i1) {
// CHECK: llvm.cond_br
llvm.cond_br %arg0, ^bb1, ^bb1
// CHECK: ^{{.*}}:
^bb1:
llvm.return
// CHECK-NOT: ^{{.*}}:
}
// CHECK: @repeated_successor_openmp
llvm.func @repeated_successor_openmp(%arg0: i64, %arg1: i64, %arg2: i64, %arg3: i1) {
omp.wsloop for (%arg4) : i64 = (%arg0) to (%arg1) step (%arg2) {
// CHECK: llvm.cond_br %{{.*}}, ^[[BB1:.*]]({{.*}}), ^[[BB2:.*]]({{.*}})
llvm.cond_br %arg3, ^bb1(%arg0 : i64), ^bb1(%arg1 : i64)
// CHECK: ^[[BB1]]
^bb1(%0: i64): // 2 preds: ^bb0, ^bb0
omp.yield
// CHECK: ^[[BB2]](%[[ARG:.*]]: i64):
// CHECK: llvm.br ^[[BB1]](%[[ARG]] : i64)
}
llvm.return
}
// -----
// This module won't have any LLVM dialect entities as input.
// Check that we don't crash when building LLVM entities.
// CHECK-LABEL:func @func1
func.func @func1(%arg0: i1, %arg1 : i1) {
// CHECK: llvm.br
cf.cond_br %arg0, ^bb40(%arg0 : i1), ^bb40(%arg1 : i1)
^bb40(%47: i1):
return
}
|
