计算函数中LLVM指令的数量

时间:2017-11-09 20:14:58

标签: llvm

我一直在使用

opt -stats -analyze -instcount file.bc

获取代码的统计信息。现在我想在一个特定名称的函数中获取LLVM指令的数量,比如“bar”。

理想情况下,我希望选择opt,这样就可以了

opt -stats -analyze -instcount   funcname="bar"

使用什么是正确的选择?我google了很多,还没有得到答案。

1 个答案:

答案 0 :(得分:3)

创建function analysis pass。 (llvm::FunctionPass documentation

您的代码看起来像这样:

// This is a contrived example.

#include <iterator>
#include <string>

#include "llvm/Pass.h"
#include "llvm/IR/BasicBlock.h"
#include "llvm/IR/Function.h"
#include "llvm/IR/Instruction.h"
#include "llvm/Support/raw_ostream.h"

namespace
{

using namespace llvm;

cl::opt<std::string> functionName("funcname", cl::ValueRequired,
  cl::desc("Function name"), cl::NotHidden);

class FunctionInstCounter : public FunctionPass
{
public:
  static char ID;

  FunctionInstCounter()
    : FunctionPass(ID)
  {
    initializeFunctionInstCounterPass(*PassRegistry::getPassRegistry());
  }

  bool runOnFunction(Function& func) override
  {
    if (func.getName() != functionName)
    {
      return false;
    }

    unsigned int instCount = 0;
    for (BasicBlock& bb : func)
    {
      instCount += std::distance(bb.begin(), bb.end());
    }

    llvm::outs() << "Number of instructions in " << func.getName() << ": "
      << instCount << "\n";
    return false;
  }
};

} // namespace

char FunctionInstCounter::ID = 0;

INITIALIZE_PASS(FunctionInstCounter, "funcinstcount",
  "Function instruction counter", false, true)

llvm::Pass* llvm::createFunctionInstCounterPass()
{
  return new FunctionInstCounter();
}

你会这样称呼:

opt -funcinstcount -funcname=NameOfFunctionHere