[llvm] r313678 - Import all inlined indirect call targets for SamplePGO.

[Dehao Chen via llvm-commits]

Author: dehao
Date: Tue Sep 19 14:18:14 2017
New Revision: 313678

URL: http://llvm.org/viewvc/llvm-project?rev=313678&view=rev
Log:
Import all inlined indirect call targets for SamplePGO.

Summary: In the ThinLTO compilation, if a function is inlined in the profiling binary, we need to inline it before annotation. If the callee is not available in the primary module, a first step is needed to import that callee function. For the current implementation, if the call is an indirect call, which has been promoted to >1 targets and inlined, SamplePGO will only import one target with the largest sample count. This patch fixed the bug to import all targets instead.

Reviewers: tejohnson, davidxl

Reviewed By: tejohnson

Subscribers: sanjoy, llvm-commits, mehdi_amini

Differential Revision: https://reviews.llvm.org/D36637

Modified:
    llvm/trunk/lib/Transforms/IPO/SampleProfile.cpp
    llvm/trunk/test/Transforms/SampleProfile/Inputs/import.prof
    llvm/trunk/test/Transforms/SampleProfile/import.ll

Modified: llvm/trunk/lib/Transforms/IPO/SampleProfile.cpp
URL: http://llvm.org/viewvc/llvm-project/llvm/trunk/lib/Transforms/IPO/SampleProfile.cpp?rev=313678&r1=313677&r2=313678&view=diff
==============================================================================
--- llvm/trunk/lib/Transforms/IPO/SampleProfile.cpp (original)
+++ llvm/trunk/lib/Transforms/IPO/SampleProfile.cpp Tue Sep 19 14:18:14 2017
@@ -756,9 +756,11 @@ bool SampleProfileLoader::inlineHotFunct
           continue;
       }
       if (!CalledFunction || !CalledFunction->getSubprogram()) {
-        findCalleeFunctionSamples(*I)->findImportedFunctions(
-            ImportGUIDs, F.getParent(),
-            Samples->getTotalSamples() * SampleProfileHotThreshold / 100);
+        // Handles functions that are imported from other modules.
+        for (const FunctionSamples *FS : findIndirectCallFunctionSamples(*I))
+          FS->findImportedFunctions(
+              ImportGUIDs, F.getParent(),
+              Samples->getTotalSamples() * SampleProfileHotThreshold / 100);
         continue;
       }
       assert(isa<CallInst>(DI) || isa<InvokeInst>(DI));

Modified: llvm/trunk/test/Transforms/SampleProfile/Inputs/import.prof
URL: http://llvm.org/viewvc/llvm-project/llvm/trunk/test/Transforms/SampleProfile/Inputs/import.prof?rev=313678&r1=313677&r2=313678&view=diff
==============================================================================
--- llvm/trunk/test/Transforms/SampleProfile/Inputs/import.prof (original)
+++ llvm/trunk/test/Transforms/SampleProfile/Inputs/import.prof Tue Sep 19 14:18:14 2017
@@ -1,4 +1,8 @@
-main:10000:0
+test:10000:0
  3: foo:1000
   3: bar:200
    4: baz:10
+ 4: foo1:1000
+  1: 1000
+ 4: foo2:1000
+  1: 1000

Modified: llvm/trunk/test/Transforms/SampleProfile/import.ll
URL: http://llvm.org/viewvc/llvm-project/llvm/trunk/test/Transforms/SampleProfile/import.ll?rev=313678&r1=313677&r2=313678&view=diff
==============================================================================
--- llvm/trunk/test/Transforms/SampleProfile/import.ll (original)
+++ llvm/trunk/test/Transforms/SampleProfile/import.ll Tue Sep 19 14:18:14 2017
@@ -5,14 +5,19 @@
 
 declare void @foo()
 
-define void @main() !dbg !7 {
+define void @test(void ()*) !dbg !7 {
+  %2 = alloca void ()*
+  store void ()* %0, void ()** %2
+  %3 = load void ()*, void ()** %2
+  ; CHECK: call {{.*}}, !prof ![[PROF:[0-9]+]]
   call void @foo(), !dbg !18
+  call void %3(), !dbg !19
   ret void
 }
 
-; GUIDs of foo and bar should be included in the metadata to make sure hot
-; inline stacks are imported.
-; CHECK: !{!"function_entry_count", i64 1, i64 6699318081062747564, i64 -2012135647395072713}
+; GUIDs of foo, bar, foo1 and foo2 should be included in the metadata to make
+; sure hot inline stacks are imported.
+; CHECK: !{!"function_entry_count", i64 1, i64 2494702099028631698, i64 6699318081062747564, i64 7682762345278052905, i64 -2012135647395072713}
 
 !llvm.dbg.cu = !{!0}
 !llvm.module.flags = !{!8, !9}
@@ -22,10 +27,11 @@ define void @main() !dbg !7 {
 !1 = !DIFile(filename: "calls.cc", directory: ".")
 !2 = !{}
 !6 = !DISubroutineType(types: !2)
-!7 = distinct !DISubprogram(name: "main", line: 7, isLocal: false, isDefinition: true, virtualIndex: 6, flags: DIFlagPrototyped, isOptimized: false, unit: !0, scopeLine: 7, file: !1, scope: !1, type: !6, variables: !2)
+!7 = distinct !DISubprogram(name: "test", line: 7, isLocal: false, isDefinition: true, virtualIndex: 6, flags: DIFlagPrototyped, isOptimized: false, unit: !0, scopeLine: 7, file: !1, scope: !1, type: !6, variables: !2)
 !8 = !{i32 2, !"Dwarf Version", i32 4}
 !9 = !{i32 1, !"Debug Info Version", i32 3}
 !10 = !{!"clang version 3.5 "}
 !15 = !DILexicalBlockFile(discriminator: 1, file: !1, scope: !7)
 !17 = distinct !DILexicalBlock(line: 10, column: 0, file: !1, scope: !7)
 !18 = !DILocation(line: 10, scope: !17)
+!19 = !DILocation(line: 11, scope: !17)