[llvm-commits] CVS: llvm/lib/Transforms/Instrumentation/TraceBasicBlocks.cpp
Vikram Adve
vadve at cs.uiuc.edu
Tue Jun 29 09:21:01 PDT 2004
Changes in directory llvm/lib/Transforms/Instrumentation:
TraceBasicBlocks.cpp updated: 1.3 -> 1.4
---
Log message:
Restoring this file.
---
Diffs of the changes: (+76 -0)
Index: llvm/lib/Transforms/Instrumentation/TraceBasicBlocks.cpp
diff -u /dev/null llvm/lib/Transforms/Instrumentation/TraceBasicBlocks.cpp:1.4
--- /dev/null Tue Jun 29 09:20:38 2004
+++ llvm/lib/Transforms/Instrumentation/TraceBasicBlocks.cpp Tue Jun 29 09:20:27 2004
@@ -0,0 +1,76 @@
+//===- TraceBasicBlocks.cpp - Insert basic-block trace instrumentation ----===//
+//
+// The LLVM Compiler Infrastructure
+//
+// This file was developed by the LLVM research group and is distributed under
+// the University of Illinois Open Source License. See LICENSE.TXT for details.
+//
+//===----------------------------------------------------------------------===//
+//
+// This pass instruments the specified program with calls into a runtime
+// library that cause it to output a trace of basic blocks as a side effect
+// of normal execution.
+//
+//===----------------------------------------------------------------------===//
+
+#include "llvm/Constants.h"
+#include "llvm/DerivedTypes.h"
+#include "llvm/Module.h"
+#include "llvm/Pass.h"
+#include "llvm/Transforms/Utils/BasicBlockUtils.h"
+#include "llvm/iOther.h"
+#include "llvm/iMemory.h"
+#include "llvm/iPHINode.h"
+#include "ProfilingUtils.h"
+#include "Support/Debug.h"
+#include <set>
+using namespace llvm;
+
+namespace {
+ class TraceBasicBlocks : public Pass {
+ bool run(Module &M);
+ };
+
+ RegisterOpt<TraceBasicBlocks> X("trace-basic-blocks",
+ "Insert instrumentation for basic block tracing");
+}
+
+static void InsertInstrumentationCall (BasicBlock *BB,
+ const std::string FnName,
+ unsigned BBNumber) {
+ DEBUG (std::cerr << "InsertInstrumentationCall (\"" << BB->getName ()
+ << "\", \"" << FnName << "\", " << BBNumber << ")\n");
+ Module &M = *BB->getParent ()->getParent ();
+ Function *InstrFn = M.getOrInsertFunction (FnName, Type::VoidTy,
+ Type::UIntTy, 0);
+ std::vector<Value*> Args (1);
+ Args[0] = ConstantUInt::get (Type::UIntTy, BBNumber);
+
+ // Insert the call after any alloca or PHI instructions...
+ BasicBlock::iterator InsertPos = BB->begin();
+ while (isa<AllocaInst>(InsertPos) || isa<PHINode>(InsertPos))
+ ++InsertPos;
+
+ Instruction *InstrCall = new CallInst (InstrFn, Args, "", InsertPos);
+}
+
+bool TraceBasicBlocks::run(Module &M) {
+ Function *Main = M.getMainFunction();
+ if (Main == 0) {
+ std::cerr << "WARNING: cannot insert basic-block trace instrumentation"
+ << " into a module with no main function!\n";
+ return false; // No main, no instrumentation!
+ }
+
+ unsigned BBNumber = 0;
+ for (Module::iterator F = M.begin(), E = M.end(); F != E; ++F)
+ for (Function::iterator BB = F->begin(), E = F->end(); BB != E; ++BB) {
+ InsertInstrumentationCall (BB, "llvm_trace_basic_block", BBNumber);
+ ++BBNumber;
+ }
+
+ // Add the initialization call to main.
+ InsertProfilingInitCall(Main, "llvm_start_basic_block_tracing");
+ return true;
+}
+
More information about the llvm-commits
mailing list