test/Transforms/FunctionImport/hotness_based_import.ll

   1 ; Test to check the callgraph in summary when there is PGO
   2 ; RUN: opt -module-summary %s -o %t.bc
   3 ; RUN: opt -module-summary %p/Inputs/hotness_based_import.ll -o %t2.bc
   4 ; RUN: llvm-lto -thinlto -o %t3 %t.bc %t2.bc
   5
   6 ; Test import with default hot multiplier (3) and default hot-evolution-factor (1.0)
   7 ; RUN: opt -function-import -summary-file %t3.thinlto.bc %t.bc -import-instr-limit=1 --S | FileCheck %s --check-prefix=CHECK --check-prefix=HOT-DEFAULT
   8 ; RUN: opt -function-import -summary-file %t3.thinlto.bc %t.bc -import-instr-limit=1 --S -import-hot-multiplier=3.0 -import-cold-multiplier=0.0 | FileCheck %s --check-prefix=CHECK --check-prefix=HOT-DEFAULT
   9 ; RUN: opt -function-import -summary-file %t3.thinlto.bc %t.bc -import-instr-limit=1 --S -import-hot-multiplier=3.0 | FileCheck %s --check-prefix=CHECK --check-prefix=HOT-DEFAULT
  10 ; RUN: opt -function-import -summary-file %t3.thinlto.bc %t.bc -import-instr-limit=1 --S -import-hot-multiplier=3.0 -import-instr-evolution-factor=0.0 | FileCheck %s --check-prefix=CHECK --check-prefix=HOT-DEFAULT
  11 ; RUN: opt -function-import -summary-file %t3.thinlto.bc %t.bc -import-instr-limit=1 --S -import-hot-multiplier=3.0 -import-hot-evolution-factor=1.0 | FileCheck %s --check-prefix=CHECK --check-prefix=HOT-DEFAULT
  12 ; HOT-DEFAULT-DAG: define available_externally void @hot1()
  13 ; HOT-DEFAULT-DAG: define available_externally void @hot2()
  14 ; HOT-DEFAULT-DAG: define available_externally void @calledFromHot()
  15 ; HOT-DEFAULT-DAG: define available_externally void @calledFromHot2()
  16 ; HOT-DEFAULT-DAG: define available_externally void @none1()
  17 ; HOT-DEFAULT-NOT: define available_externally void @cold()
  18 ; HOT-DEFAULT-NOT: define available_externally void @hot3()
  19 ; HOT-DEFAULT-NOT: define available_externally void @none2()
  20 ; HOT-DEFAULT-NOT: define available_externally void @none3()
  21 ; HOT-DEFAULT-NOT: define available_externally void @cold2()
  22 ; HOT-DEFAULT-NOT: define available_externally void @calledFromCold()
  23 ; HOT-DEFAULT-NOT: define available_externally void @calledFromNone()
  24
  25 ; This one tests if we decay threshold for hot callsites. With hot-evolution-factor of 0
  26 ; we should not import any of calledFromHot functions
  27 ; RUN: opt -function-import -summary-file %t3.thinlto.bc %t.bc -import-instr-limit=1 --S -import-hot-multiplier=3.0 -import-hot-evolution-factor=0.0 | FileCheck %s --check-prefix=CHECK --check-prefix=HOT-EVOLUTION
  28 ; HOT-EVOLUTION-DAG: define available_externally void @hot1()
  29 ; HOT-EVOLUTION-DAG: define available_externally void @hot2()
  30 ; HOT-EVOLUTION-DAG: define available_externally void @none1()
  31 ; HOT-EVOLUTION-NOT: define available_externally void @hot3()
  32 ; HOT-EVOLUTION-NOT: define available_externally void @cold()
  33 ; HOT-EVOLUTION-NOT: define available_externally void @none2()
  34 ; HOT-EVOLUTION-NOT: define available_externally void @none3()
  35 ; HOT-EVOLUTION-NOT: define available_externally void @cold2()
  36 ; HOT-EVOLUTION-NOT: define available_externally void @calledFromHot()
  37 ; HOT-EVOLUTION-NOT: define available_externally void @calledFromHot2()
  38
  39 ; Test import with hot multiplier 1.0 - treat hot callsites as normal.
  40 ; RUN: opt -function-import -summary-file %t3.thinlto.bc %t.bc -import-instr-limit=1 -import-hot-multiplier=1.0 --S | FileCheck %s --check-prefix=CHECK --check-prefix=HOT-ONE
  41 ; HOT-ONE-DAG: define available_externally void @hot1()
  42 ; HOT-ONE-DAG: define available_externally void @none1()
  43 ; HOT-ONE-NOT: define available_externally void @cold()
  44 ; HOT-ONE-NOT: define available_externally void @hot2()
  45 ; HOT-ONE-NOT: define available_externally void @hot3()
  46 ; HOT-ONE-NOT: define available_externally void @none2()
  47 ; HOT-ONE-NOT: define available_externally void @none3()
  48 ; HOT-ONE-NOT: define available_externally void @cold2()
  49
  50 ; RUN: opt -function-import -summary-file %t3.thinlto.bc %t.bc -import-instr-limit=1 -import-hot-multiplier=1.0 -import-cold-multiplier=1.0 --S | FileCheck %s --check-prefix=CHECK --check-prefix=HOT-COLD-ONE
  51 ; HOT-COLD-ONE-DAG: define available_externally void @hot1()
  52 ; HOT-COLD-ONE-DAG: define available_externally void @cold()
  53 ; HOT-COLD-ONE-DAG: define available_externally void @none1()
  54 ; HOT-COLD-ONE-NOT: define available_externally void @hot2()
  55 ; HOT-COLD-ONE-NOT: define available_externally void @hot3()
  56 ; HOT-COLD-ONE-NOT: define available_externally void @none2()
  57 ; HOT-COLD-ONE-NOT: define available_externally void @none3()
  58 ; HOT-COLD-ONE-NOT: define available_externally void @cold2()
  59
  60 ; Test import with hot multiplier 0.0 and high threshold - don't import functions called from hot callsite.
  61 ; RUN: opt -function-import -summary-file %t3.thinlto.bc %t.bc -import-instr-limit=10 -import-hot-multiplier=0.0 -import-cold-multiplier=1.0 --S | FileCheck %s --check-prefix=CHECK --check-prefix=HOT-ZERO
  62 ; HOT-ZERO-DAG: define available_externally void @cold()
  63 ; HOT-ZERO-DAG: define available_externally void @none1()
  64 ; HOT-ZERO-DAG: define available_externally void @none2()
  65 ; HOT-ZERO-DAG: define available_externally void @none3()
  66 ; HOT-ZERO-DAG: define available_externally void @cold2()
  67 ; HOT-ZERO-DAG: define available_externally void @calledFromCold()
  68 ; HOT-ZERO-DAG: define available_externally void @calledFromNone()
  69 ; HOT-ZERO-NOT: define available_externally void @hot2()
  70 ; HOT-ZERO-NOT: define available_externally void @hot1()
  71 ; HOT-ZERO-NOT: define available_externally void @hot3()
  72 ; HOT-ZERO-NOT: define available_externally void @calledFromHot()
  73 ; HOT-ZERO-NOT: define available_externally void @calledFromHot2()
  74
  75
  76 ; ModuleID = 'thinlto-function-summary-callgraph.ll'
  77 target datalayout = "e-m:e-p270:32:32-p271:32:32-p272:64:64-i64:64-f80:128-n8:16:32:64-S128"
  78 target triple = "x86_64-unknown-linux-gnu"
  79
  80 ; This function have high profile count, so entry block is hot.
  81 define void @hot_function(i1 %a, i1 %a2) !prof !20 {
  82 entry:
  83     call void @hot1()
  84     br i1 %a, label %Cold, label %Hot, !prof !41
  85 Cold:           ; 1/1000 goes here
  86   call void @cold()
  87   call void @cold2()
  88   call void @hot2()
  89   call void @none1()
  90   br label %exit
  91 Hot:            ; 999/1000 goes here
  92   call void @hot2()
  93   call void @hot3()
  94   br i1 %a2, label %None1, label %None2, !prof !42
  95 None1:          ; half goes here
  96   call void @none1()
  97   call void @none2()
  98   br label %exit
  99 None2:          ; half goes here
 100   call void @none3()
 101   br label %exit
 102 exit:
 103   ret void
 104 }
 105
 106 declare void @hot1() #1
 107 declare void @hot2() #1
 108 declare void @hot3() #1
 109 declare void @cold() #1
 110 declare void @cold2() #1
 111 declare void @none1() #1
 112 declare void @none2() #1
 113 declare void @none3() #1
 114
 115
 116 !41 = !{!"branch_weights", i32 1, i32 1000}
 117 !42 = !{!"branch_weights", i32 1, i32 1}
 118
 119
 120
 121 !llvm.module.flags = !{!1}
 122 !20 = !{!"function_entry_count", i64 110}
 123
 124 !1 = !{i32 1, !"ProfileSummary", !2}
 125 !2 = !{!3, !4, !5, !6, !7, !8, !9, !10}
 126 !3 = !{!"ProfileFormat", !"InstrProf"}
 127 !4 = !{!"TotalCount", i64 10000}
 128 !5 = !{!"MaxCount", i64 10}
 129 !6 = !{!"MaxInternalCount", i64 1}
 130 !7 = !{!"MaxFunctionCount", i64 1000}
 131 !8 = !{!"NumCounts", i64 3}
 132 !9 = !{!"NumFunctions", i64 3}
 133 !10 = !{!"DetailedSummary", !11}
 134 !11 = !{!12, !13, !14}
 135 !12 = !{i32 10000, i64 100, i32 1}
 136 !13 = !{i32 999000, i64 100, i32 1}
 137 !14 = !{i32 999999, i64 1, i32 2}