diff options
author | Tarindu Jayatilaka <tarindujayatilaka@gmail.com> | 2020-07-22 09:52:53 -0700 |
---|---|---|
committer | Mircea Trofin <mtrofin@google.com> | 2020-07-22 10:07:35 -0700 |
commit | 418121c30a84b38288d615d1cd3443e0f63cce62 (patch) | |
tree | 1305963ce3ac99fdad34283ea1f58e80d597c30e /llvm/lib/Analysis/FunctionPropertiesAnalysis.cpp | |
parent | 274b6b0c7a8b584662595762eaeff57d61c6807f (diff) | |
download | llvm-418121c30a84b38288d615d1cd3443e0f63cce62.zip llvm-418121c30a84b38288d615d1cd3443e0f63cce62.tar.gz llvm-418121c30a84b38288d615d1cd3443e0f63cce62.tar.bz2 |
Reapply "Rename InlineFeatureAnalysis to FunctionPropertiesAnalysis"
(This reverts commit a5e0194709c40212694370e0ea789a1ca14548b5, and
corrects author).
Rename the pass to be able to extend it to function properties other than inliner features.
Reviewed By: mtrofin
Differential Revision: https://reviews.llvm.org/D82044
Diffstat (limited to 'llvm/lib/Analysis/FunctionPropertiesAnalysis.cpp')
-rw-r--r-- | llvm/lib/Analysis/FunctionPropertiesAnalysis.cpp | 42 |
1 files changed, 42 insertions, 0 deletions
diff --git a/llvm/lib/Analysis/FunctionPropertiesAnalysis.cpp b/llvm/lib/Analysis/FunctionPropertiesAnalysis.cpp new file mode 100644 index 0000000..a0fc017 --- /dev/null +++ b/llvm/lib/Analysis/FunctionPropertiesAnalysis.cpp @@ -0,0 +1,42 @@ +//===- FunctionPropertiesAnalysis.cpp - Function properties extraction ----===// +// +// 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 implements an analysis extracting function features, which may be +// used by ML-driven policies, for example. +// +//===----------------------------------------------------------------------===// + +#include "llvm/Analysis/FunctionPropertiesAnalysis.h" +#include "llvm/IR/Instructions.h" + +using namespace llvm; + +AnalysisKey FunctionPropertiesAnalysis::Key; + +FunctionPropertiesAnalysis::Result +FunctionPropertiesAnalysis::run(const Function &F, + FunctionAnalysisManager &FAM) { + Result Ret; + Ret.Uses = ((!F.hasLocalLinkage()) ? 1 : 0) + F.getNumUses(); + for (const auto &BB : F) { + ++Ret.BasicBlockCount; + if (const auto *BI = dyn_cast<BranchInst>(BB.getTerminator())) { + if (BI->isConditional()) + Ret.BlocksReachedFromConditionalInstruction += BI->getNumSuccessors(); + } else if (const auto *SI = dyn_cast<SwitchInst>(BB.getTerminator())) + Ret.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()) + ++Ret.DirectCallsToDefinedFunctions; + } + } + return Ret; +}
\ No newline at end of file |