Index: lib/Analysis/ModuleSummaryAnalysis.cpp =================================================================== --- lib/Analysis/ModuleSummaryAnalysis.cpp +++ lib/Analysis/ModuleSummaryAnalysis.cpp @@ -101,15 +101,26 @@ auto CS = ImmutableCallSite(&I); if (!CS) continue; + auto *CalledValue = CS.getCalledValue(); auto *CalledFunction = CS.getCalledFunction(); + // Check if this is an alias to a function. If so, get the + // called aliasee for the checks below. + if (auto *GA = dyn_cast(CalledValue)) { + assert(!CalledFunction && "Expected null called function in callsite for alias"); + CalledFunction = dyn_cast(GA->getBaseObject()); + } // Check if this is a direct call to a known function. if (CalledFunction) { // Skip nameless and intrinsics. if (!CalledFunction->hasName() || CalledFunction->isIntrinsic()) continue; auto ScaledCount = BFI ? BFI->getBlockProfileCount(&BB) : None; + // Use the original CalledValue, in case it was an alias. We want + // to record the call edge to the alias in that case. Eventually + // an alias summary will be created to associate the alias and + // aliasee. auto *CalleeId = - M.getValueSymbolTable().lookup(CalledFunction->getName()); + M.getValueSymbolTable().lookup(CalledValue->getName()); auto Hotness = ScaledCount ? getHotness(ScaledCount.getValue(), PSI) : CalleeInfo::HotnessType::Unknown; Index: test/Bitcode/thinlto-alias2.ll =================================================================== --- /dev/null +++ test/Bitcode/thinlto-alias2.ll @@ -0,0 +1,28 @@ +; Test to check the callgraph for call to alias in module. +; RUN: opt -module-summary %s -o %t.o +; RUN: llvm-bcanalyzer -dump %t.o | FileCheck %s + +; CHECK: +; CHECK-NEXT: +; CHECK-NEXT: + +; ModuleID = 'thinlto-alias2.ll' +target datalayout = "e-m:e-i64:64-f80:128-n8:16:32:64-S128" +target triple = "x86_64-unknown-linux-gnu" + +define i32 @main() { +entry: + call void (...) @analias() + ret i32 0 +} + +@analias = alias void (...), bitcast (void ()* @aliasee to void (...)*) + +define void @aliasee() #0 { +entry: + ret void +} +