[MLIR] Extend MPI dialect (#123255)
[llvm-project.git] / llvm / test / Transforms / IROutliner / phi-nodes-simple.ll
blob31cfe2ccea95260dfabe831a1242c69f3f1fbec7
1 ; NOTE: Assertions have been autogenerated by utils/update_test_checks.py UTC_ARGS: --include-generated-funcs
2 ; RUN: opt -S -passes=verify,iroutliner -ir-outlining-no-cost < %s | FileCheck %s
4 ; Show that we are able to outline the simple phi node case of constants when
5 ; the corresponding labels match.
7 define void @function1(ptr %a, ptr %b) {
8 entry:
9   br label %test
10 test:
11   br label %first
12 first:
13   %0 = phi i32 [ 0, %test ]
14   store i32 2, ptr %a, align 4
15   store i32 3, ptr %b, align 4
16   ret void
19 define void @function2(ptr %a, ptr %b) {
20 entry:
21   br label %test
22 test:
23   br label %first
24 first:
25   %0 = phi i32 [ 0, %test ]
26   store i32 2, ptr %a, align 4
27   store i32 3, ptr %b, align 4
28   ret void
30 ; CHECK-LABEL: @function1(
31 ; CHECK-NEXT:  entry:
32 ; CHECK-NEXT:    call void @outlined_ir_func_0(ptr [[A:%.*]], ptr [[B:%.*]])
33 ; CHECK-NEXT:    ret void
36 ; CHECK-LABEL: @function2(
37 ; CHECK-NEXT:  entry:
38 ; CHECK-NEXT:    call void @outlined_ir_func_0(ptr [[A:%.*]], ptr [[B:%.*]])
39 ; CHECK-NEXT:    ret void
42 ; CHECK: define internal void @outlined_ir_func_0(
43 ; CHECK-NEXT:  newFuncRoot:
44 ; CHECK-NEXT:    br label [[ENTRY_TO_OUTLINE:%.*]]
45 ; CHECK:       entry_to_outline:
46 ; CHECK-NEXT:    br label [[TEST:%.*]]
47 ; CHECK:       test:
48 ; CHECK-NEXT:    br label [[FIRST:%.*]]
49 ; CHECK:       first:
50 ; CHECK-NEXT:    [[TMP2:%.*]] = phi i32 [ 0, [[TEST]] ]
51 ; CHECK-NEXT:    store i32 2, ptr [[TMP0:%.*]], align 4
52 ; CHECK-NEXT:    store i32 3, ptr [[TMP1:%.*]], align 4
53 ; CHECK-NEXT:    br label [[ENTRY_AFTER_OUTLINE_EXITSTUB:%.*]]
54 ; CHECK:       entry_after_outline.exitStub:
55 ; CHECK-NEXT:    ret void