llvm/test/Transforms/LoopUnroll/peel-loop2.ll

   1 ; RUN: opt -S -passes=loop-unroll -unroll-force-peel-count=1 -verify-dom-info <%s
   2
   3 ; Check if loop composed of several BBs is peeled correctly.
   4
   5 declare void @funcb()
   6 @Comma = external global i8
   7 define void @funca(ptr readnone %b, ptr readnone %e) {
   8 entry:
   9   %cmp2 = icmp eq ptr %b, %e
  10   br i1 %cmp2, label %for.end, label %for.body.preheader
  11
  12 for.body.preheader:
  13   br label %for.body
  14
  15 for.body:
  16   %b.addr.03 = phi ptr [ %incdec.ptr, %for.inc ], [ %b, %for.body.preheader ]
  17   %0 = load i8, ptr @Comma
  18   %tobool = icmp eq i8 %0, 0
  19   br i1 %tobool, label %for.inc, label %if.then
  20
  21 if.then:
  22   tail call void @funcb()
  23   store i8 1, ptr @Comma
  24   br label %for.inc
  25
  26 for.inc:
  27   %incdec.ptr = getelementptr inbounds i8, ptr %b.addr.03, i64 1
  28   %cmp = icmp eq ptr %incdec.ptr, %e
  29   br i1 %cmp, label %for.end, label %for.body
  30
  31 for.end:
  32   ret void
  33 }
  34
  35 ; CHECK_LABEL: @funca
  36
  37 ; Peeled iteration
  38 ; CHECK: %[[REG1:[0-9]+]] = load i8, ptr @Comma
  39 ; CHECK: %[[REG2:.*]] = icmp eq i8 %[[REG1]], 0
  40 ; CHECK: br i1 %[[REG2]], label %{{.*}}, label %[[IFTHEN:.*]]
  41 ; CHECK: [[IFTHEN]]:
  42 ; CHECK: call void @funcb()
  43 ; CHECK: store i8 1, ptr @Comma
  44 ; CHECK: br label %[[FORINC]]
  45 ; CHECK: [[FORINC]]:
  46 ; CHECK: %[[REG3:.*]] = getelementptr inbounds i8, ptr %b, i64 1
  47 ; CHECK: %[[REG4:.*]] = icmp eq ptr %[[REG3]], %e
  48 ; CHECK: br i1 %[[REG4]]
  49
  50 ; main body
  51 ; CHECK: %[[REG1b:.*]] = load i8, ptr @Comma
  52 ; CHECK: %[[REG2b:.*]] = icmp eq i8 %[[REG1b]], 0
  53 ; CHECK: br i1 %[[REG2b]], label %{{.*}}, label %[[IFTHENb:.*]]
  54 ; CHECK: [[IFTHENb]]:
  55 ; CHECK: call void @funcb()
  56 ; CHECK: store i8 1, ptr @Comma
  57 ; CHECK: br label %[[FORINCb]]
  58 ; CHECK: [[FORINCb]]:
  59 ; CHECK: %[[REG3b:.*]] = getelementptr inbounds i8, ptr %b, i64 1
  60 ; CHECK: %[[REG4b:.*]] = icmp eq ptr %[[REG3b]], %e
  61 ; CHECK: br i1 %[[REG4b]]