Created this for instcount and func-properties-analysis to be able to see the change the optimization pipelines have on stats
106 lines
4.0 KiB
C++
106 lines
4.0 KiB
C++
//===-- InstCount.cpp - Collects the count of all instructions ------------===//
|
|
//
|
|
// 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 pass collects the count of all instructions and reports them
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#include "llvm/Analysis/InstCount.h"
|
|
#include "llvm/ADT/Statistic.h"
|
|
#include "llvm/IR/Function.h"
|
|
#include "llvm/IR/InstVisitor.h"
|
|
#include "llvm/Support/Debug.h"
|
|
#include "llvm/Support/ErrorHandling.h"
|
|
#include "llvm/Support/raw_ostream.h"
|
|
using namespace llvm;
|
|
|
|
#define DEBUG_TYPE "instcount"
|
|
|
|
STATISTIC(TotalInstsPreOptimization,
|
|
"Number of instructions of all types (before optimizations)");
|
|
STATISTIC(TotalInsts, "Number of instructions of all types");
|
|
STATISTIC(TotalBlocksPreOptimization,
|
|
"Number of basic blocks (before optimizations)");
|
|
STATISTIC(TotalBlocks, "Number of basic blocks");
|
|
STATISTIC(TotalFuncsPreOptimization,
|
|
"Number of non-external functions (before optimizations)");
|
|
STATISTIC(TotalFuncs, "Number of non-external functions");
|
|
STATISTIC(LargestFunctionSizePreOptimization,
|
|
"Largest number of instructions in a single function (before "
|
|
"optimizations)");
|
|
STATISTIC(LargestFunctionSize,
|
|
"Largest number of instructions in a single function");
|
|
STATISTIC(LargestFunctionBBCountPreOptimization,
|
|
"Largest number of basic blocks in a single function (before "
|
|
"optimizations)");
|
|
STATISTIC(LargestFunctionBBCount,
|
|
"Largest number of basic blocks in a single function");
|
|
|
|
#define HANDLE_INST(N, OPCODE, CLASS) \
|
|
STATISTIC(Num##OPCODE##InstPreOptimization, \
|
|
"Number of " #OPCODE " insts (before optimizations)"); \
|
|
STATISTIC(Num##OPCODE##Inst, "Number of " #OPCODE " insts");
|
|
|
|
#include "llvm/IR/Instruction.def"
|
|
|
|
namespace {
|
|
class InstCount : public InstVisitor<InstCount> {
|
|
friend class InstVisitor<InstCount>;
|
|
bool IsPreOptimization;
|
|
|
|
public:
|
|
InstCount(bool IsPreOptimization) : IsPreOptimization(IsPreOptimization) {}
|
|
|
|
void visitFunction(Function &F) {
|
|
if (IsPreOptimization) {
|
|
++TotalFuncsPreOptimization;
|
|
LargestFunctionSizePreOptimization.updateMax(F.getInstructionCount());
|
|
LargestFunctionBBCountPreOptimization.updateMax(F.size());
|
|
} else {
|
|
++TotalFuncs;
|
|
LargestFunctionSize.updateMax(F.getInstructionCount());
|
|
LargestFunctionBBCount.updateMax(F.size());
|
|
}
|
|
}
|
|
|
|
void visitBasicBlock(BasicBlock &BB) {
|
|
if (IsPreOptimization)
|
|
++TotalBlocksPreOptimization;
|
|
else
|
|
++TotalBlocks;
|
|
}
|
|
|
|
#define HANDLE_INST(N, OPCODE, CLASS) \
|
|
void visit##OPCODE(CLASS &) { \
|
|
if (IsPreOptimization) { \
|
|
++Num##OPCODE##InstPreOptimization; \
|
|
++TotalInstsPreOptimization; \
|
|
} else { \
|
|
++Num##OPCODE##Inst; \
|
|
++TotalInsts; \
|
|
} \
|
|
}
|
|
|
|
#include "llvm/IR/Instruction.def"
|
|
|
|
void visitInstruction(Instruction &I) {
|
|
errs() << "Instruction Count does not know about " << I;
|
|
llvm_unreachable(nullptr);
|
|
}
|
|
};
|
|
} // namespace
|
|
|
|
PreservedAnalyses InstCountPass::run(Function &F,
|
|
FunctionAnalysisManager &FAM) {
|
|
LLVM_DEBUG(dbgs() << "INSTCOUNT: running on function " << F.getName()
|
|
<< "\n");
|
|
InstCount(this->IsPreOptimization).visit(F);
|
|
|
|
return PreservedAnalyses::all();
|
|
}
|