Do not unswitch if the function notes say we're optimizing this function for size.

llvm-svn: 55786

1 files changed, 7 insertions, 1 deletions

diff --git a/llvm/lib/Transforms/Scalar/LoopUnswitch.cpp b/llvm/lib/Transforms/Scalar/LoopUnswitch.cpp
index 372e6e4f834..f9046a1a055 100644
--- a/llvm/lib/Transforms/Scalar/LoopUnswitch.cpp
+++ b/llvm/lib/Transforms/Scalar/LoopUnswitch.cpp
@@ -417,6 +417,13 @@ unsigned LoopUnswitch::getLoopUnswitchCost(Value *LIC) {
 /// LoopCond == Val to simplify the loop.  If we decide that this is profitable,
 /// unswitch the loop, reprocess the pieces, then return true.
 bool LoopUnswitch::UnswitchIfProfitable(Value *LoopCond, Constant *Val){
+
+  Function *F = loopHeader->getParent();
+
+  // Do not unswitch if the function is optimized for size.
+  if (F->getNotes() & FN_NOTE_OptimizeForSize)
+    return false;
+
   // Check to see if it would be profitable to unswitch current loop.
   unsigned Cost = getLoopUnswitchCost(LoopCond);
 
@@ -445,7 +452,6 @@ bool LoopUnswitch::UnswitchIfProfitable(Value *LoopCond, Constant *Val){
   }
 
   // FIXME: Reconstruct dom info, because it is not preserved properly.
-  Function *F = loopHeader->getParent();
   if (DT)
     DT->runOnFunction(*F);
   if (DF)