llvm/test/Transforms/IROutliner/region-inputs-in-phi-nodes.ll

   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
   3
   4 ; Show that we are able to propogate inputs to the region into the split PHINode
   5 ; outside of the region if necessary.
   6
   7 define void @function1(ptr %a, ptr %b) {
   8 entry:
   9   %0 = alloca i32, align 4
  10   %c = load i32, ptr %0, align 4
  11   %z = add i32 %c, %c
  12   br i1 true, label %test1, label %first
  13 test1:
  14   %e = load i32, ptr %0, align 4
  15   %1 = add i32 %c, %c
  16   br i1 true, label %first, label %test
  17 test:
  18   %d = load i32, ptr %0, align 4
  19   br i1 true, label %first, label %next
  20 dummy:
  21   ret void
  22 first:
  23   %2 = phi i32 [ %d, %test ], [ %e, %test1 ], [ %c, %entry ]
  24   ret void
  25 next:
  26   ret void
  27 }
  28
  29 define void @function2(ptr %a, ptr %b) {
  30 entry:
  31   %0 = alloca i32, align 4
  32   %c = load i32, ptr %0, align 4
  33   %z = mul i32 %c, %c
  34   br i1 true, label %test1, label %first
  35 test1:
  36   %e = load i32, ptr %0, align 4
  37   %1 = add i32 %c, %c
  38   br i1 true, label %first, label %test
  39 test:
  40   %d = load i32, ptr %0, align 4
  41   br i1 true, label %first, label %next
  42 dummy:
  43   ret void
  44 first:
  45   %2 = phi i32 [ %d, %test ], [ %e, %test1 ], [ %c, %entry ]
  46   ret void
  47 next:
  48   ret void
  49 }
  50 ; CHECK-LABEL: @function1(
  51 ; CHECK-NEXT:  entry:
  52 ; CHECK-NEXT:    [[DOTCE_LOC:%.*]] = alloca i32, align 4
  53 ; CHECK-NEXT:    [[TMP0:%.*]] = alloca i32, align 4
  54 ; CHECK-NEXT:    [[C:%.*]] = load i32, ptr [[TMP0]], align 4
  55 ; CHECK-NEXT:    [[Z:%.*]] = add i32 [[C]], [[C]]
  56 ; CHECK-NEXT:    call void @llvm.lifetime.start.p0(i64 -1, ptr [[DOTCE_LOC]])
  57 ; CHECK-NEXT:    [[TARGETBLOCK:%.*]] = call i1 @outlined_ir_func_0(ptr [[TMP0]], i32 [[C]], ptr [[DOTCE_LOC]])
  58 ; CHECK-NEXT:    [[DOTCE_RELOAD:%.*]] = load i32, ptr [[DOTCE_LOC]], align 4
  59 ; CHECK-NEXT:    call void @llvm.lifetime.end.p0(i64 -1, ptr [[DOTCE_LOC]])
  60 ; CHECK-NEXT:    br i1 [[TARGETBLOCK]], label [[FIRST:%.*]], label [[NEXT:%.*]]
  61 ; CHECK: dummy:
  62 ; CHECK-NEXT:  ret void
  63 ; CHECK:       first:
  64 ; CHECK-NEXT:    [[TMP1:%.*]] = phi i32 [ [[DOTCE_RELOAD]], [[ENTRY:%.*]] ]
  65 ; CHECK-NEXT:    ret void
  66 ; CHECK:       next:
  67 ; CHECK-NEXT:    ret void
  68 ;
  69 ;
  70 ; CHECK-LABEL: @function2(
  71 ; CHECK-NEXT:  entry:
  72 ; CHECK-NEXT:    [[DOTCE_LOC:%.*]] = alloca i32, align 4
  73 ; CHECK-NEXT:    [[TMP0:%.*]] = alloca i32, align 4
  74 ; CHECK-NEXT:    [[C:%.*]] = load i32, ptr [[TMP0]], align 4
  75 ; CHECK-NEXT:    [[Z:%.*]] = mul i32 [[C]], [[C]]
  76 ; CHECK-NEXT:    call void @llvm.lifetime.start.p0(i64 -1, ptr [[DOTCE_LOC]])
  77 ; CHECK-NEXT:    [[TARGETBLOCK:%.*]] = call i1 @outlined_ir_func_0(ptr [[TMP0]], i32 [[C]], ptr [[DOTCE_LOC]])
  78 ; CHECK-NEXT:    [[DOTCE_RELOAD:%.*]] = load i32, ptr [[DOTCE_LOC]], align 4
  79 ; CHECK-NEXT:    call void @llvm.lifetime.end.p0(i64 -1, ptr [[DOTCE_LOC]])
  80 ; CHECK-NEXT:    br i1 [[TARGETBLOCK]], label [[FIRST:%.*]], label [[NEXT:%.*]]
  81 ; CHECK: dummy:
  82 ; CHECK-NEXT:  ret void
  83 ; CHECK:       first:
  84 ; CHECK-NEXT:    [[TMP1:%.*]] = phi i32 [ [[DOTCE_RELOAD]], [[ENTRY:%.*]] ]
  85 ; CHECK-NEXT:    ret void
  86 ; CHECK:       next:
  87 ; CHECK-NEXT:    ret void
  88 ;
  89 ;
  90 ; CHECK-LABEL: define internal i1 @outlined_ir_func_0(
  91 ; CHECK-NEXT:  newFuncRoot:
  92 ; CHECK-NEXT:    br label [[ENTRY_TO_OUTLINE:%.*]]
  93 ; CHECK:       entry_to_outline:
  94 ; CHECK-NEXT:    br i1 true, label [[TEST1:%.*]], label [[FIRST_SPLIT:%.*]]
  95 ; CHECK:       test1:
  96 ; CHECK-NEXT:    [[E:%.*]] = load i32, ptr [[TMP0:%.*]], align 4
  97 ; CHECK-NEXT:    [[TMP3:%.*]] = add i32 [[TMP1:%.*]], [[TMP1]]
  98 ; CHECK-NEXT:    br i1 true, label [[FIRST_SPLIT]], label [[TEST:%.*]]
  99 ; CHECK:       test:
 100 ; CHECK-NEXT:    [[D:%.*]] = load i32, ptr [[TMP0]], align 4
 101 ; CHECK-NEXT:    br i1 true, label [[FIRST_SPLIT]], label [[NEXT_EXITSTUB:%.*]]
 102 ; CHECK:       first.split:
 103 ; CHECK-NEXT:    [[DOTCE:%.*]] = phi i32 [ [[D]], [[TEST]] ], [ [[E]], [[TEST1]] ], [ [[TMP1]], [[ENTRY_TO_OUTLINE]] ]
 104 ; CHECK-NEXT:    br label [[FIRST_EXITSTUB:%.*]]
 105 ; CHECK:       first.exitStub:
 106 ; CHECK-NEXT:    store i32 [[DOTCE]], ptr [[TMP2:%.*]], align 4
 107 ; CHECK-NEXT:    ret i1 true
 108 ; CHECK:       next.exitStub:
 109 ; CHECK-NEXT:    ret i1 false
 110 ;