Do not unswitch if the function notes say we're optimizing this function for size.
git-svn-id: https://llvm.org/svn/llvm-project/llvm/trunk@55786 91177308-0d34-0410-b5e6-96231b3b80d8
diff --git a/lib/Transforms/Scalar/LoopUnswitch.cpp b/lib/Transforms/Scalar/LoopUnswitch.cpp
index 372e6e4..f9046a1 100644
--- a/lib/Transforms/Scalar/LoopUnswitch.cpp
+++ b/lib/Transforms/Scalar/LoopUnswitch.cpp
@@ -417,6 +417,13 @@
/// 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 @@
}
// FIXME: Reconstruct dom info, because it is not preserved properly.
- Function *F = loopHeader->getParent();
if (DT)
DT->runOnFunction(*F);
if (DF)