80155cbf0b
Currently, to use PSI->isFunctionHotInCallGraph, we first need to calculate BPI->BFI, which is expensive. Instead, we can implement this directly with MBFI. Also as @wenlei mentioned in another patch review, that MachineSizeOpts already has isFunctionColdInCallGraph, isFunctionHotInCallGraphNthPercentile, etc implemented. These can be refactored and so they can be reused across MachineFunctionSplitting and MachineSizeOpts passes. This CL does this - it refactors out those internal static functions into PSI as templated functions, so they can be accessed easily. Differential Revision: https://reviews.llvm.org/D153927
110 lines
4.7 KiB
C++
110 lines
4.7 KiB
C++
//===-- SizeOpts.cpp - code size optimization related code ----------------===//
|
|
//
|
|
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
|
|
// See https://llvm.org/LICENSE.txt for license information.
|
|
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
//
|
|
// This file contains some shared code size optimization related code.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#include "llvm/Transforms/Utils/SizeOpts.h"
|
|
|
|
using namespace llvm;
|
|
|
|
cl::opt<bool> llvm::EnablePGSO(
|
|
"pgso", cl::Hidden, cl::init(true),
|
|
cl::desc("Enable the profile guided size optimizations. "));
|
|
|
|
cl::opt<bool> llvm::PGSOLargeWorkingSetSizeOnly(
|
|
"pgso-lwss-only", cl::Hidden, cl::init(true),
|
|
cl::desc("Apply the profile guided size optimizations only "
|
|
"if the working set size is large (except for cold code.)"));
|
|
|
|
cl::opt<bool> llvm::PGSOColdCodeOnly(
|
|
"pgso-cold-code-only", cl::Hidden, cl::init(false),
|
|
cl::desc("Apply the profile guided size optimizations only "
|
|
"to cold code."));
|
|
|
|
cl::opt<bool> llvm::PGSOColdCodeOnlyForInstrPGO(
|
|
"pgso-cold-code-only-for-instr-pgo", cl::Hidden, cl::init(false),
|
|
cl::desc("Apply the profile guided size optimizations only "
|
|
"to cold code under instrumentation PGO."));
|
|
|
|
cl::opt<bool> llvm::PGSOColdCodeOnlyForSamplePGO(
|
|
"pgso-cold-code-only-for-sample-pgo", cl::Hidden, cl::init(false),
|
|
cl::desc("Apply the profile guided size optimizations only "
|
|
"to cold code under sample PGO."));
|
|
|
|
cl::opt<bool> llvm::PGSOColdCodeOnlyForPartialSamplePGO(
|
|
"pgso-cold-code-only-for-partial-sample-pgo", cl::Hidden, cl::init(false),
|
|
cl::desc("Apply the profile guided size optimizations only "
|
|
"to cold code under partial-profile sample PGO."));
|
|
|
|
cl::opt<bool> llvm::ForcePGSO(
|
|
"force-pgso", cl::Hidden, cl::init(false),
|
|
cl::desc("Force the (profiled-guided) size optimizations. "));
|
|
|
|
cl::opt<int> llvm::PgsoCutoffInstrProf(
|
|
"pgso-cutoff-instr-prof", cl::Hidden, cl::init(950000),
|
|
cl::desc("The profile guided size optimization profile summary cutoff "
|
|
"for instrumentation profile."));
|
|
|
|
cl::opt<int> llvm::PgsoCutoffSampleProf(
|
|
"pgso-cutoff-sample-prof", cl::Hidden, cl::init(990000),
|
|
cl::desc("The profile guided size optimization profile summary cutoff "
|
|
"for sample profile."));
|
|
|
|
namespace {
|
|
struct BasicBlockBFIAdapter {
|
|
static bool isFunctionColdInCallGraph(const Function *F,
|
|
ProfileSummaryInfo *PSI,
|
|
BlockFrequencyInfo &BFI) {
|
|
return PSI->isFunctionColdInCallGraph(F, BFI);
|
|
}
|
|
static bool isFunctionHotInCallGraphNthPercentile(int CutOff,
|
|
const Function *F,
|
|
ProfileSummaryInfo *PSI,
|
|
BlockFrequencyInfo &BFI) {
|
|
return PSI->isFunctionHotInCallGraphNthPercentile(CutOff, F, BFI);
|
|
}
|
|
static bool isFunctionColdInCallGraphNthPercentile(int CutOff,
|
|
const Function *F,
|
|
ProfileSummaryInfo *PSI,
|
|
BlockFrequencyInfo &BFI) {
|
|
return PSI->isFunctionColdInCallGraphNthPercentile(CutOff, F, BFI);
|
|
}
|
|
static bool isColdBlock(const BasicBlock *BB,
|
|
ProfileSummaryInfo *PSI,
|
|
BlockFrequencyInfo *BFI) {
|
|
return PSI->isColdBlock(BB, BFI);
|
|
}
|
|
static bool isHotBlockNthPercentile(int CutOff,
|
|
const BasicBlock *BB,
|
|
ProfileSummaryInfo *PSI,
|
|
BlockFrequencyInfo *BFI) {
|
|
return PSI->isHotBlockNthPercentile(CutOff, BB, BFI);
|
|
}
|
|
static bool isColdBlockNthPercentile(int CutOff, const BasicBlock *BB,
|
|
ProfileSummaryInfo *PSI,
|
|
BlockFrequencyInfo *BFI) {
|
|
return PSI->isColdBlockNthPercentile(CutOff, BB, BFI);
|
|
}
|
|
};
|
|
} // end anonymous namespace
|
|
|
|
bool llvm::shouldOptimizeForSize(const Function *F, ProfileSummaryInfo *PSI,
|
|
BlockFrequencyInfo *BFI,
|
|
PGSOQueryType QueryType) {
|
|
return shouldFuncOptimizeForSizeImpl(F, PSI, BFI, QueryType);
|
|
}
|
|
|
|
bool llvm::shouldOptimizeForSize(const BasicBlock *BB, ProfileSummaryInfo *PSI,
|
|
BlockFrequencyInfo *BFI,
|
|
PGSOQueryType QueryType) {
|
|
assert(BB);
|
|
return shouldOptimizeForSizeImpl(BB, PSI, BFI, QueryType);
|
|
}
|