llvm/test/Transforms/LoopLoadElim/opt-size.ll

   1 ; RUN: opt -basic-aa -loop-load-elim -S < %s | FileCheck %s
   2 ; RUN: opt -basic-aa -loop-load-elim -pgso -S < %s -enable-new-pm=0 | FileCheck %s -check-prefix=PGSO
   3 ; RUN: opt -aa-pipeline=basic-aa -passes='require<profile-summary>,function(loop-load-elim)' -S < %s | FileCheck %s -check-prefix=PGSO
   4 ; RUN: opt -basic-aa -loop-load-elim -pgso=false -S < %s | FileCheck %s -check-prefix=NPGSO
   5
   6 ; When optimizing for size don't eliminate in this loop because the loop would
   7 ; have to be versioned first because A and C may alias.
   8 ;
   9 ;   for (unsigned i = 0; i < 100; i++) {
  10 ;     A[i+1] = B[i] + 2;
  11 ;     C[i] = A[i] * 2;
  12 ;   }
  13
  14 target datalayout = "e-m:o-i64:64-f80:128-n8:16:32:64-S128"
  15
  16 ; CHECK-LABEL: @f(
  17 define void @f(i32* %A, i32* %B, i32* %C, i64 %N) optsize {
  18
  19 entry:
  20   br label %for.body
  21
  22 for.body:                                         ; preds = %for.body, %entry
  23   %indvars.iv = phi i64 [ 0, %entry ], [ %indvars.iv.next, %for.body ]
  24   %indvars.iv.next = add nuw nsw i64 %indvars.iv, 1
  25
  26   %Aidx_next = getelementptr inbounds i32, i32* %A, i64 %indvars.iv.next
  27   %Bidx = getelementptr inbounds i32, i32* %B, i64 %indvars.iv
  28   %Cidx = getelementptr inbounds i32, i32* %C, i64 %indvars.iv
  29   %Aidx = getelementptr inbounds i32, i32* %A, i64 %indvars.iv
  30
  31   %b = load i32, i32* %Bidx, align 4
  32   %a_p1 = add i32 %b, 2
  33   store i32 %a_p1, i32* %Aidx_next, align 4
  34
  35   %a = load i32, i32* %Aidx, align 4
  36 ; CHECK: %c = mul i32 %a, 2
  37   %c = mul i32 %a, 2
  38   store i32 %c, i32* %Cidx, align 4
  39
  40   %exitcond = icmp eq i64 %indvars.iv.next, %N
  41   br i1 %exitcond, label %for.end, label %for.body
  42
  43 for.end:                                          ; preds = %for.body
  44   ret void
  45 }
  46
  47 ; Same loop but with noalias on %A and %C.  In this case load-eliminate even
  48 ; with -Os.
  49
  50 ; CHECK-LABEL: @g(
  51 define void @g(i32* noalias %A, i32* %B, i32* noalias %C, i64 %N) optsize {
  52
  53 entry:
  54   br label %for.body
  55
  56 for.body:                                         ; preds = %for.body, %entry
  57   %indvars.iv = phi i64 [ 0, %entry ], [ %indvars.iv.next, %for.body ]
  58   %indvars.iv.next = add nuw nsw i64 %indvars.iv, 1
  59
  60   %Aidx_next = getelementptr inbounds i32, i32* %A, i64 %indvars.iv.next
  61   %Bidx = getelementptr inbounds i32, i32* %B, i64 %indvars.iv
  62   %Cidx = getelementptr inbounds i32, i32* %C, i64 %indvars.iv
  63   %Aidx = getelementptr inbounds i32, i32* %A, i64 %indvars.iv
  64
  65   %b = load i32, i32* %Bidx, align 4
  66   %a_p1 = add i32 %b, 2
  67   store i32 %a_p1, i32* %Aidx_next, align 4
  68
  69   %a = load i32, i32* %Aidx, align 4
  70 ; CHECK: %c = mul i32 %store_forwarded, 2
  71   %c = mul i32 %a, 2
  72   store i32 %c, i32* %Cidx, align 4
  73
  74   %exitcond = icmp eq i64 %indvars.iv.next, %N
  75   br i1 %exitcond, label %for.end, label %for.body
  76
  77 for.end:                                          ; preds = %for.body
  78   ret void
  79 }
  80
  81
  82 ; PGSO-LABEL: @f_pgso(
  83 ; NPGSO-LABEL: @f_pgso(
  84 define void @f_pgso(i32* %A, i32* %B, i32* %C, i64 %N) !prof !14 {
  85
  86 entry:
  87   br label %for.body
  88
  89 for.body:                                         ; preds = %for.body, %entry
  90   %indvars.iv = phi i64 [ 0, %entry ], [ %indvars.iv.next, %for.body ]
  91   %indvars.iv.next = add nuw nsw i64 %indvars.iv, 1
  92
  93   %Aidx_next = getelementptr inbounds i32, i32* %A, i64 %indvars.iv.next
  94   %Bidx = getelementptr inbounds i32, i32* %B, i64 %indvars.iv
  95   %Cidx = getelementptr inbounds i32, i32* %C, i64 %indvars.iv
  96   %Aidx = getelementptr inbounds i32, i32* %A, i64 %indvars.iv
  97
  98   %b = load i32, i32* %Bidx, align 4
  99   %a_p1 = add i32 %b, 2
 100   store i32 %a_p1, i32* %Aidx_next, align 4
 101
 102   %a = load i32, i32* %Aidx, align 4
 103 ; PGSO: %c = mul i32 %a, 2
 104 ; NPGSO-NOT: %c = mul i32 %a, 2
 105   %c = mul i32 %a, 2
 106   store i32 %c, i32* %Cidx, align 4
 107
 108   %exitcond = icmp eq i64 %indvars.iv.next, %N
 109   br i1 %exitcond, label %for.end, label %for.body
 110
 111 for.end:                                          ; preds = %for.body
 112   ret void
 113 }
 114
 115 !llvm.module.flags = !{!0}
 116 !0 = !{i32 1, !"ProfileSummary", !1}
 117 !1 = !{!2, !3, !4, !5, !6, !7, !8, !9}
 118 !2 = !{!"ProfileFormat", !"InstrProf"}
 119 !3 = !{!"TotalCount", i64 10000}
 120 !4 = !{!"MaxCount", i64 10}
 121 !5 = !{!"MaxInternalCount", i64 1}
 122 !6 = !{!"MaxFunctionCount", i64 1000}
 123 !7 = !{!"NumCounts", i64 3}
 124 !8 = !{!"NumFunctions", i64 3}
 125 !9 = !{!"DetailedSummary", !10}
 126 !10 = !{!11, !12, !13}
 127 !11 = !{i32 10000, i64 100, i32 1}
 128 !12 = !{i32 999000, i64 100, i32 1}
 129 !13 = !{i32 999999, i64 1, i32 2}
 130 !14 = !{!"function_entry_count", i64 0}