diff options
| -rw-r--r-- | llvm/include/llvm/Transforms/Utils/SizeOpts.h | 7 | ||||
| -rw-r--r-- | llvm/lib/Transforms/Utils/SizeOpts.cpp | 5 |
2 files changed, 10 insertions, 2 deletions
diff --git a/llvm/include/llvm/Transforms/Utils/SizeOpts.h b/llvm/include/llvm/Transforms/Utils/SizeOpts.h index 1c56da08ef8..4614007a645 100644 --- a/llvm/include/llvm/Transforms/Utils/SizeOpts.h +++ b/llvm/include/llvm/Transforms/Utils/SizeOpts.h @@ -21,6 +21,7 @@ using namespace llvm; extern cl::opt<bool> EnablePGSO; extern cl::opt<bool> PGSOLargeWorkingSetSizeOnly; +extern cl::opt<bool> PGSOColdCodeOnly; extern cl::opt<bool> ForcePGSO; extern cl::opt<int> PgsoCutoffInstrProf; extern cl::opt<int> PgsoCutoffSampleProf; @@ -42,7 +43,8 @@ bool shouldFuncOptimizeForSizeImpl(const FuncT *F, ProfileSummaryInfo *PSI, return true; if (!EnablePGSO) return false; - if (PGSOLargeWorkingSetSizeOnly && !PSI->hasLargeWorkingSetSize()) { + if (PGSOColdCodeOnly || + (PGSOLargeWorkingSetSizeOnly && !PSI->hasLargeWorkingSetSize())) { // Even if the working set size isn't large, size-optimize cold code. return AdapterT::isFunctionColdInCallGraph(F, PSI, *BFI); } @@ -61,7 +63,8 @@ bool shouldOptimizeForSizeImpl(const BlockT *BB, ProfileSummaryInfo *PSI, return true; if (!EnablePGSO) return false; - if (PGSOLargeWorkingSetSizeOnly && !PSI->hasLargeWorkingSetSize()) { + if (PGSOColdCodeOnly || + (PGSOLargeWorkingSetSizeOnly && !PSI->hasLargeWorkingSetSize())) { // Even if the working set size isn't large, size-optimize cold code. return AdapterT::isColdBlock(BB, PSI, BFI); } diff --git a/llvm/lib/Transforms/Utils/SizeOpts.cpp b/llvm/lib/Transforms/Utils/SizeOpts.cpp index f819c67d69d..f1200471cb4 100644 --- a/llvm/lib/Transforms/Utils/SizeOpts.cpp +++ b/llvm/lib/Transforms/Utils/SizeOpts.cpp @@ -23,6 +23,11 @@ cl::opt<bool> PGSOLargeWorkingSetSizeOnly( cl::desc("Apply the profile guided size optimizations only " "if the working set size is large (except for cold code.)")); +cl::opt<bool> PGSOColdCodeOnly( + "pgso-cold-code-only", cl::Hidden, cl::init(true), + cl::desc("Apply the profile guided size optimizations only " + "to cold code.")); + cl::opt<bool> ForcePGSO( "force-pgso", cl::Hidden, cl::init(false), cl::desc("Force the (profiled-guided) size optimizations. ")); |
