[llvm] r357795 - [LCG] Add aliased functions as LCG roots

[Guozhi Wei via llvm-commits]

Author: carrot
Date: Fri Apr  5 11:51:08 2019
New Revision: 357795

URL: http://llvm.org/viewvc/llvm-project?rev=357795&view=rev
Log:
[LCG] Add aliased functions as LCG roots

Current LCG doesn't check aliased functions. So if an internal function has a public alias it will not be added to CG SCC, but it is still reachable from outside through the alias.
So this patch adds aliased functions to SCC.

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


Added:
    llvm/trunk/test/Analysis/LazyCallGraph/alias.ll
Modified:
    llvm/trunk/lib/Analysis/LazyCallGraph.cpp

Modified: llvm/trunk/lib/Analysis/LazyCallGraph.cpp
URL: http://llvm.org/viewvc/llvm-project/llvm/trunk/lib/Analysis/LazyCallGraph.cpp?rev=357795&r1=357794&r2=357795&view=diff
==============================================================================
--- llvm/trunk/lib/Analysis/LazyCallGraph.cpp (original)
+++ llvm/trunk/lib/Analysis/LazyCallGraph.cpp Fri Apr  5 11:51:08 2019
@@ -172,6 +172,19 @@ LazyCallGraph::LazyCallGraph(Module &M,
     addEdge(EntryEdges.Edges, EntryEdges.EdgeIndexMap, get(F), Edge::Ref);
   }
 
+  // Externally visible aliases of internal functions are also viable entry
+  // edges to the module.
+  for (auto &A : M.aliases()) {
+    if (A.hasLocalLinkage())
+      continue;
+    if (Function* F = dyn_cast<Function>(A.getAliasee())) {
+      LLVM_DEBUG(dbgs() << "  Adding '" << F->getName()
+                        << "' with alias '" << A.getName()
+                        << "' to entry set of the graph.\n");
+      addEdge(EntryEdges.Edges, EntryEdges.EdgeIndexMap, get(*F), Edge::Ref);
+    }
+  }
+
   // Now add entry nodes for functions reachable via initializers to globals.
   SmallVector<Constant *, 16> Worklist;
   SmallPtrSet<Constant *, 16> Visited;

Added: llvm/trunk/test/Analysis/LazyCallGraph/alias.ll
URL: http://llvm.org/viewvc/llvm-project/llvm/trunk/test/Analysis/LazyCallGraph/alias.ll?rev=357795&view=auto
==============================================================================
--- llvm/trunk/test/Analysis/LazyCallGraph/alias.ll (added)
+++ llvm/trunk/test/Analysis/LazyCallGraph/alias.ll Fri Apr  5 11:51:08 2019
@@ -0,0 +1,38 @@
+; RUN: opt -disable-output -passes=print-lcg %s 2>&1 | FileCheck %s
+;
+; Aliased function should be reachable in CGSCC.
+
+target triple = "x86_64-grtev4-linux-gnu"
+
+; CHECK:        Edges in function: foo
+; CHECK:        Edges in function: bar
+; CHECK:        Edges in function: baz
+
+; CHECK:        RefSCC with 1 call SCCs:
+; CHECK-NEXT:     SCC with 1 functions:
+; CHECK-NEXT:       bar
+
+; CHECK:       RefSCC with 1 call SCCs:
+; CHECK-NEXT:    SCC with 1 functions:
+; CHECK-NEXT:      foo
+
+; CHECK-NOT:       baz
+
+ at alias1 = weak dso_local alias i8* (i8*), i8* (i8*)* @foo
+
+define dso_local i8* @foo(i8* %returned) {
+  ret i8* %returned
+}
+
+ at alias2 = weak dso_local alias i8* (i8*), i8* (i8*)* @bar
+
+define internal i8* @bar(i8* %returned) {
+  ret i8* %returned
+}
+
+; Internal alias is not reachable.
+ at alias3 = internal alias i8* (i8*), i8* (i8*)* @baz
+
+define internal i8* @baz(i8* %returned) {
+  ret i8* %returned
+}