llvm/test/Transforms/PGOProfile/split-indirectbr-critical-edges.ll

   1 ; RUN: opt < %s -passes=pgo-instr-gen -S | FileCheck %s
   2
   3 ; Function Attrs: norecurse nounwind readnone uwtable
   4 define i32 @bar(i32 %v) local_unnamed_addr #0 {
   5 entry:
   6   %mul = shl nsw i32 %v, 1
   7   ret i32 %mul
   8 }
   9
  10 ; Function Attrs: norecurse nounwind readonly uwtable
  11 define i32 @foo(ptr nocapture readonly %p) #1 {
  12 entry:
  13   %targets = alloca [256 x ptr], align 16
  14   %arrayidx1 = getelementptr inbounds [256 x ptr], ptr %targets, i64 0, i64 93
  15   store ptr blockaddress(@foo, %if.end), ptr %arrayidx1, align 8
  16   br label %for.cond2
  17
  18 for.cond2:                                        ; preds = %if.end, %for.cond2, %entry
  19 ; CHECK: for.cond2:                                        ; preds = %.split1
  20   %p.addr.0 = phi ptr [ %p, %entry ], [ %incdec.ptr5, %if.end ], [ %incdec.ptr, %for.cond2 ]
  21   %incdec.ptr = getelementptr inbounds i8, ptr %p.addr.0, i64 1
  22   %0 = load i8, ptr %p.addr.0, align 1
  23   %cond = icmp eq i8 %0, 93
  24   br i1 %cond, label %if.end.preheader, label %for.cond2
  25
  26 if.end.preheader:                                 ; preds = %for.cond2
  27   br label %if.end
  28
  29 if.end:                                           ; preds = %if.end.preheader, %if.end
  30 ; CHECK: if.end:                                           ; preds = %.split1
  31   %p.addr.1 = phi ptr [ %incdec.ptr5, %if.end ], [ %incdec.ptr, %if.end.preheader ]
  32   %incdec.ptr5 = getelementptr inbounds i8, ptr %p.addr.1, i64 1
  33   %1 = load i8, ptr %p.addr.1, align 1
  34   %idxprom6 = zext i8 %1 to i64
  35   %arrayidx7 = getelementptr inbounds [256 x ptr], ptr %targets, i64 0, i64 %idxprom6
  36   %2 = load ptr, ptr %arrayidx7, align 8
  37   indirectbr ptr %2, [label %for.cond2, label %if.end]
  38 ; CHECK: indirectbr ptr %2, [label %for.cond2, label %if.end]
  39 }
  40
  41 ;; If an indirectbr critical edge cannot be split, ignore it.
  42 ;; The edge will not be profiled.
  43 ; CHECK-LABEL: @cannot_split(
  44 ; CHECK-NEXT:  entry:
  45 ; CHECK-NEXT:    call void @llvm.instrprof.increment
  46 ; CHECK: indirect:
  47 ; CHECK-NOT:     call void @llvm.instrprof.increment
  48 ; CHECK: indirect2:
  49 ; CHECK-NEXT:    call void @llvm.instrprof.increment
  50 define i32 @cannot_split(ptr nocapture readonly %p) {
  51 entry:
  52   %targets = alloca <2 x ptr>, align 16
  53   store <2 x ptr> <ptr blockaddress(@cannot_split, %indirect), ptr blockaddress(@cannot_split, %end)>, ptr %targets, align 16
  54   %arrayidx2 = getelementptr inbounds i8, ptr %p, i64 1
  55   %0 = load i8, ptr %arrayidx2
  56   %idxprom = sext i8 %0 to i64
  57   %arrayidx3 = getelementptr inbounds <2 x ptr>, ptr %targets, i64 0, i64 %idxprom
  58   %1 = load ptr, ptr %arrayidx3, align 8
  59   br label %indirect
  60
  61 indirect:                                         ; preds = %entry, %indirect
  62   indirectbr ptr %1, [label %indirect, label %end, label %indirect2]
  63
  64 indirect2:
  65   ; For this test we do not want critical edges split. Adding a 2nd `indirectbr`
  66   ; does the trick.
  67   indirectbr ptr %1, [label %indirect, label %end]
  68
  69 end:                                              ; preds = %indirect
  70   ret i32 0
  71 }