1//===-- InstCount.cpp - Collects the count of all instructions ------------===// 2// 3// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. 4// See https://llvm.org/LICENSE.txt for license information. 5// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception 6// 7//===----------------------------------------------------------------------===// 8// 9// This pass collects the count of all instructions and reports them 10// 11//===----------------------------------------------------------------------===// 12 13#include "llvm/ADT/Statistic.h" 14#include "llvm/Analysis/Passes.h" 15#include "llvm/IR/Function.h" 16#include "llvm/IR/InstVisitor.h" 17#include "llvm/InitializePasses.h" 18#include "llvm/Pass.h" 19#include "llvm/Support/Debug.h" 20#include "llvm/Support/ErrorHandling.h" 21#include "llvm/Support/raw_ostream.h" 22using namespace llvm; 23 24#define DEBUG_TYPE "instcount" 25 26STATISTIC(TotalInsts , "Number of instructions (of all types)"); 27STATISTIC(TotalBlocks, "Number of basic blocks"); 28STATISTIC(TotalFuncs , "Number of non-external functions"); 29 30#define HANDLE_INST(N, OPCODE, CLASS) \ 31 STATISTIC(Num ## OPCODE ## Inst, "Number of " #OPCODE " insts"); 32 33#include "llvm/IR/Instruction.def" 34 35namespace { 36 class InstCount : public FunctionPass, public InstVisitor<InstCount> { 37 friend class InstVisitor<InstCount>; 38 39 void visitFunction (Function &F) { ++TotalFuncs; } 40 void visitBasicBlock(BasicBlock &BB) { ++TotalBlocks; } 41 42#define HANDLE_INST(N, OPCODE, CLASS) \ 43 void visit##OPCODE(CLASS &) { ++Num##OPCODE##Inst; ++TotalInsts; } 44 45#include "llvm/IR/Instruction.def" 46 47 void visitInstruction(Instruction &I) { 48 errs() << "Instruction Count does not know about " << I; 49 llvm_unreachable(nullptr); 50 } 51 public: 52 static char ID; // Pass identification, replacement for typeid 53 InstCount() : FunctionPass(ID) { 54 initializeInstCountPass(*PassRegistry::getPassRegistry()); 55 } 56 57 bool runOnFunction(Function &F) override; 58 59 void getAnalysisUsage(AnalysisUsage &AU) const override { 60 AU.setPreservesAll(); 61 } 62 void print(raw_ostream &O, const Module *M) const override {} 63 64 }; 65} 66 67char InstCount::ID = 0; 68INITIALIZE_PASS(InstCount, "instcount", 69 "Counts the various types of Instructions", false, true) 70 71FunctionPass *llvm::createInstCountPass() { return new InstCount(); } 72 73// InstCount::run - This is the main Analysis entry point for a 74// function. 75// 76bool InstCount::runOnFunction(Function &F) { 77 visit(F); 78 return false; 79} 80