llvm-project

1
// RUN: mlir-opt -allow-unregistered-dialect %s -split-input-file -verify-diagnostics | FileCheck %s
2

3
//===----------------------------------------------------------------------===//
4
// Test the number of regions
5
//===----------------------------------------------------------------------===//
6

7
func.func @correct_number_of_regions() {
8
    // CHECK: test.two_region_op
9
    "test.two_region_op"()(
10
      {"work"() : () -> ()},
11
      {"work"() : () -> ()}
12
    ) : () -> ()
13
    return
14
}
15

16
// -----
17

18
func.func @missing_regions() {
19
    // expected-error@+1 {{expected 2 regions}}
20
    "test.two_region_op"()(
21
      {"work"() : () -> ()}
22
    ) : () -> ()
23
    return
24
}
25

26
// -----
27

28
func.func @extra_regions() {
29
    // expected-error@+1 {{expected 2 regions}}
30
    "test.two_region_op"()(
31
      {"work"() : () -> ()},
32
      {"work"() : () -> ()},
33
      {"work"() : () -> ()}
34
    ) : () -> ()
35
    return
36
}
37

38
// -----
39

40
//===----------------------------------------------------------------------===//
41
// Test SizedRegion
42
//===----------------------------------------------------------------------===//
43

44
func.func @unnamed_region_has_wrong_number_of_blocks() {
45
    // expected-error@+1 {{region #1 failed to verify constraint: region with 1 blocks}}
46
    "test.sized_region_op"() (
47
    {
48
        "work"() : () -> ()
49
        cf.br ^next1
50
      ^next1:
51
        "work"() : () -> ()
52
    },
53
    {
54
        "work"() : () -> ()
55
        cf.br ^next2
56
      ^next2:
57
        "work"() : () -> ()
58
    }) : () -> ()
59
    return
60
}
61

62
// -----
63

64
// Test region name in error message
65
func.func @named_region_has_wrong_number_of_blocks() {
66
    // expected-error@+1 {{region #0 ('my_region') failed to verify constraint: region with 2 blocks}}
67
    "test.sized_region_op"() (
68
    {
69
        "work"() : () -> ()
70
    },
71
    {
72
        "work"() : () -> ()
73
    }) : () -> ()
74
    return
75
}
76

77
// -----
78

79
// Region with single block and not terminator.
80
// CHECK: unregistered_without_terminator
81
"test.unregistered_without_terminator"() ({
82
  ^bb0:
83
}) : () -> ()
84

85
// -----
86

87
// CHECK: test.single_no_terminator_op
88
"test.single_no_terminator_op"() (
89
  {
90
    %foo = arith.constant 1 : i32
91
  }
92
) : () -> ()
93

94
// CHECK: test.variadic_no_terminator_op
95
"test.variadic_no_terminator_op"() (
96
  {
97
    %foo = arith.constant 1 : i32
98
  },
99
  {
100
    %bar = arith.constant 1 : i32
101
  }
102
) : () -> ()
103

104
// CHECK: test.single_no_terminator_custom_asm_op
105
// CHECK-NEXT: important_dont_drop
106
test.single_no_terminator_custom_asm_op {
107
  "important_dont_drop"() : () -> ()
108
}
109