llvm/test/Transforms/LoopFusion/sink_store.ll

   1 ; RUN: opt -S -passes=loop-simplify,loop-fusion -debug-only=loop-fusion < %s 2>&1 | FileCheck %s
   2 ; REQUIRES: asserts
   3
   4 @A = common global [100 x i32] zeroinitializer, align 16
   5 define void @sink_preheader(i32 %N) {
   6 ; CHECK-LABEL: @sink_preheader(
   7 ; CHECK-NEXT:  pre1:
   8 ; CHECK-NEXT:    [[PTR:%.*]] = alloca i32, align 4
   9 ; CHECK-NEXT:    br label [[BODY1:%.*]]
  10 ; CHECK:       body1:
  11 ; CHECK-NEXT:    [[I:%.*]] = phi i32 [ [[I_NEXT:%.*]], [[BODY1]] ], [ 0, [[PRE1:%.*]] ]
  12 ; CHECK-NEXT:    [[I2:%.*]] = phi i32 [ [[I_NEXT2:%.*]], [[BODY1]] ], [ 0, [[PRE1]] ]
  13 ; CHECK-NEXT:    [[I_NEXT]] = add i32 1, [[I]]
  14 ; CHECK-NEXT:    [[COND:%.*]] = icmp ne i32 [[I]], [[N:%.*]]
  15 ; CHECK-NEXT:    [[B:%.*]] = load i32, ptr [[PTR]], align 4
  16 ; CHECK-NEXT:    [[I_NEXT2]] = add i32 1, [[I2]]
  17 ; CHECK-NEXT:    [[COND2:%.*]] = icmp ne i32 [[I2]], [[N]]
  18 ; CHECK-NEXT:    br i1 [[COND2]], label [[BODY1]], label [[EXIT:%.*]]
  19 ; CHECK:       exit:
  20 ; CHECK-NEXT:    store i32 3, ptr [[PTR]], align 4
  21 ; CHECK-NEXT:    ret void
  22 ;
  23 pre1:
  24   %ptr = alloca i32
  25   br label %body1
  26
  27 body1:  ; preds = %pre1, %body1
  28   %i = phi i32 [%i_next, %body1], [0, %pre1]
  29   %i_next = add i32 1, %i
  30   %cond = icmp ne i32 %i, %N
  31   %b = load i32, ptr %ptr
  32   br i1 %cond, label %body1, label %pre2
  33
  34 pre2:
  35   store i32 3, ptr %ptr
  36   br label %body2
  37
  38 body2:  ; preds = %pre2, %body2
  39   %i2 = phi i32 [%i_next2, %body2], [0, %pre2]
  40   %i_next2 = add i32 1, %i2
  41   %cond2 = icmp ne i32 %i2, %N
  42   br i1 %cond2, label %body2, label %exit
  43
  44 exit:
  45   ret void
  46 }