//===------- llvm/IR/OptBisect/Bisect.cpp - LLVM Bisect support --------===// // // The LLVM Compiler Infrastructure // // This file is distributed under the University of Illinois Open Source // License. See LICENSE.TXT for details. // //===----------------------------------------------------------------------===// /// /// \file /// This file implements support for a bisecting optimizations based on a /// command line option. /// //===----------------------------------------------------------------------===// #include "llvm/Analysis/CallGraphSCCPass.h" #include "llvm/Analysis/LazyCallGraph.h" #include "llvm/Analysis/LoopInfo.h" #include "llvm/IR/Module.h" #include "llvm/IR/OptBisect.h" #include "llvm/Pass.h" #include "llvm/Support/CommandLine.h" #include "llvm/Support/raw_ostream.h" using namespace llvm; static cl::opt<int> OptBisectLimit("opt-bisect-limit", cl::Hidden, cl::init(INT_MAX), cl::Optional, cl::desc("Maximum optimization to perform")); OptBisect::OptBisect() { BisectEnabled = OptBisectLimit != INT_MAX; } static void printPassMessage(const StringRef &Name, int PassNum, StringRef TargetDesc, bool Running) { StringRef Status = Running ? "" : "NOT "; errs() << "BISECT: " << Status << "running pass " << "(" << PassNum << ") " << Name << " on " << TargetDesc << "\n"; } static void printCaseMessage(int CaseNum, StringRef Msg, bool Running) { if (Running) errs() << "BISECT: running case ("; else errs() << "BISECT: NOT running case ("; errs() << CaseNum << "): " << Msg << "\n"; } static std::string getDescription(const Module &M) { return "module (" + M.getName().str() + ")"; } static std::string getDescription(const Function &F) { return "function (" + F.getName().str() + ")"; } static std::string getDescription(const BasicBlock &BB) { return "basic block (" + BB.getName().str() + ") in function (" + BB.getParent()->getName().str() + ")"; } static std::string getDescription(const Loop &L) { // FIXME: I'd like to be able to provide a better description here, but // calling L->getHeader() would introduce a new dependency on the // LLVMCore library. return "loop"; } static std::string getDescription(const CallGraphSCC &SCC) { std::string Desc = "SCC ("; bool First = true; for (CallGraphNode *CGN : SCC) { if (First) First = false; else Desc += ", "; Function *F = CGN->getFunction(); if (F) Desc += F->getName(); else Desc += "<<null function>>"; } Desc += ")"; return Desc; } // Force instantiations. template bool OptBisect::shouldRunPass(const Pass *, const Module &); template bool OptBisect::shouldRunPass(const Pass *, const Function &); template bool OptBisect::shouldRunPass(const Pass *, const BasicBlock &); template bool OptBisect::shouldRunPass(const Pass *, const Loop &); template bool OptBisect::shouldRunPass(const Pass *, const CallGraphSCC &); template <class UnitT> bool OptBisect::shouldRunPass(const Pass *P, const UnitT &U) { if (!BisectEnabled) return true; return checkPass(P->getPassName(), getDescription(U)); } bool OptBisect::checkPass(const StringRef PassName, const StringRef TargetDesc) { assert(BisectEnabled); int CurBisectNum = ++LastBisectNum; bool ShouldRun = (OptBisectLimit == -1 || CurBisectNum <= OptBisectLimit); printPassMessage(PassName, CurBisectNum, TargetDesc, ShouldRun); return ShouldRun; } bool OptBisect::shouldRunCase(const Twine &Msg) { if (!BisectEnabled) return true; int CurFuelNum = ++LastBisectNum; bool ShouldRun = (OptBisectLimit == -1 || CurFuelNum <= OptBisectLimit); printCaseMessage(CurFuelNum, Msg.str(), ShouldRun); return ShouldRun; }