Disable most IR-level transform passes on functions marked 'optnone'.

[oota-llvm.git] / lib / Transforms / Scalar / TailRecursionElimination.cpp

diff --git a/lib/Transforms/Scalar/TailRecursionElimination.cpp b/lib/Transforms/Scalar/TailRecursionElimination.cpp
index 9de138892143725c1aa086c96bbcef0bd937fbba..351aee058ba846fa69740d18d871d5b54d53a428 100644 (file)
--- a/lib/Transforms/Scalar/TailRecursionElimination.cpp
+++ b/lib/Transforms/Scalar/TailRecursionElimination.cpp
@@ -145,19 +145,20 @@ static bool CanTRE(AllocaInst *AI) {
          isa<ConstantInt>(AI->getArraySize());
 }
 
+namespace {
 struct AllocaCaptureTracker : public CaptureTracker {
   AllocaCaptureTracker() : Captured(false) {}
 
-  void tooManyUses() { Captured = true; }
+  void tooManyUses() LLVM_OVERRIDE { Captured = true; }
 
-  bool shouldExplore(Use *U) {
+  bool shouldExplore(Use *U) LLVM_OVERRIDE {
     Value *V = U->getUser();
     if (isa<CallInst>(V) || isa<InvokeInst>(V))
       UsesAlloca.insert(V);
     return true;
   }
 
-  bool captured(Use *U) {
+  bool captured(Use *U) LLVM_OVERRIDE {
     if (isa<ReturnInst>(U->getUser()))
       return false;
     Captured = true;
@@ -165,10 +166,14 @@ struct AllocaCaptureTracker : public CaptureTracker {
   }
 
   bool Captured;
-  SmallPtrSet<const Value *, 64> UsesAlloca;
+  SmallPtrSet<const Value *, 16> UsesAlloca;
 };
+} // end anonymous namespace
 
 bool TailCallElim::runOnFunction(Function &F) {
+  if (skipOptnoneFunction(F))
+    return false;
+
   // If this function is a varargs function, we won't be able to PHI the args
   // right, so don't even try to convert it...
   if (F.getFunctionType()->isVarArg()) return false;