89 lines
3.1 KiB
C++
89 lines
3.1 KiB
C++
|
//===- FunctionPropertiesAnalysis.cpp - Function Properties Analysis ------===//
|
||
|
//
|
||
|
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
|
||
|
// See https://llvm.org/LICENSE.txt for license information.
|
||
|
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
|
||
|
//
|
||
|
//===----------------------------------------------------------------------===//
|
||
|
//
|
||
|
// This file defines the FunctionPropertiesInfo and FunctionPropertiesAnalysis
|
||
|
// classes used to extract function properties.
|
||
|
//
|
||
|
//===----------------------------------------------------------------------===//
|
||
|
|
||
|
#include "llvm/Analysis/FunctionPropertiesAnalysis.h"
|
||
|
#include "llvm/IR/Instructions.h"
|
||
|
|
||
|
using namespace llvm;
|
||
|
|
||
|
FunctionPropertiesInfo
|
||
|
FunctionPropertiesInfo::getFunctionPropertiesInfo(const Function &F,
|
||
|
const LoopInfo &LI) {
|
||
|
|
||
|
FunctionPropertiesInfo FPI;
|
||
|
|
||
|
FPI.Uses = ((!F.hasLocalLinkage()) ? 1 : 0) + F.getNumUses();
|
||
|
|
||
|
for (const auto &BB : F) {
|
||
|
++FPI.BasicBlockCount;
|
||
|
|
||
|
if (const auto *BI = dyn_cast<BranchInst>(BB.getTerminator())) {
|
||
|
if (BI->isConditional())
|
||
|
FPI.BlocksReachedFromConditionalInstruction += BI->getNumSuccessors();
|
||
|
} else if (const auto *SI = dyn_cast<SwitchInst>(BB.getTerminator())) {
|
||
|
FPI.BlocksReachedFromConditionalInstruction +=
|
||
|
(SI->getNumCases() + (nullptr != SI->getDefaultDest()));
|
||
|
}
|
||
|
|
||
|
for (const auto &I : BB) {
|
||
|
if (auto *CS = dyn_cast<CallBase>(&I)) {
|
||
|
const auto *Callee = CS->getCalledFunction();
|
||
|
if (Callee && !Callee->isIntrinsic() && !Callee->isDeclaration())
|
||
|
++FPI.DirectCallsToDefinedFunctions;
|
||
|
}
|
||
|
if (I.getOpcode() == Instruction::Load) {
|
||
|
++FPI.LoadInstCount;
|
||
|
} else if (I.getOpcode() == Instruction::Store) {
|
||
|
++FPI.StoreInstCount;
|
||
|
}
|
||
|
}
|
||
|
// Loop Depth of the Basic Block
|
||
|
int64_t LoopDepth;
|
||
|
LoopDepth = LI.getLoopDepth(&BB);
|
||
|
if (FPI.MaxLoopDepth < LoopDepth)
|
||
|
FPI.MaxLoopDepth = LoopDepth;
|
||
|
}
|
||
|
FPI.TopLevelLoopCount += llvm::size(LI);
|
||
|
return FPI;
|
||
|
}
|
||
|
|
||
|
void FunctionPropertiesInfo::print(raw_ostream &OS) const {
|
||
|
OS << "BasicBlockCount: " << BasicBlockCount << "\n"
|
||
|
<< "BlocksReachedFromConditionalInstruction: "
|
||
|
<< BlocksReachedFromConditionalInstruction << "\n"
|
||
|
<< "Uses: " << Uses << "\n"
|
||
|
<< "DirectCallsToDefinedFunctions: " << DirectCallsToDefinedFunctions
|
||
|
<< "\n"
|
||
|
<< "LoadInstCount: " << LoadInstCount << "\n"
|
||
|
<< "StoreInstCount: " << StoreInstCount << "\n"
|
||
|
<< "MaxLoopDepth: " << MaxLoopDepth << "\n"
|
||
|
<< "TopLevelLoopCount: " << TopLevelLoopCount << "\n\n";
|
||
|
}
|
||
|
|
||
|
AnalysisKey FunctionPropertiesAnalysis::Key;
|
||
|
|
||
|
FunctionPropertiesInfo
|
||
|
FunctionPropertiesAnalysis::run(Function &F, FunctionAnalysisManager &FAM) {
|
||
|
return FunctionPropertiesInfo::getFunctionPropertiesInfo(
|
||
|
F, FAM.getResult<LoopAnalysis>(F));
|
||
|
}
|
||
|
|
||
|
PreservedAnalyses
|
||
|
FunctionPropertiesPrinterPass::run(Function &F, FunctionAnalysisManager &AM) {
|
||
|
OS << "Printing analysis results of CFA for function "
|
||
|
<< "'" << F.getName() << "':"
|
||
|
<< "\n";
|
||
|
AM.getResult<FunctionPropertiesAnalysis>(F).print(OS);
|
||
|
return PreservedAnalyses::all();
|
||
|
}
|