1
0
mirror of https://github.com/RPCS3/llvm-mirror.git synced 2025-02-01 05:01:59 +01:00

Reland [NewPM] Add OptimizationLevel param to registerPipelineStartEPCallback

This allows targets to skip optional optimization passes at -O0.

Reviewed By: ychen

Differential Revision: https://reviews.llvm.org/D90777
This commit is contained in:
Arthur Eubanks 2020-11-04 10:14:37 -08:00
parent 3a6a4e9f83
commit b3f5096b36
4 changed files with 16 additions and 14 deletions

View File

@ -587,7 +587,7 @@ public:
/// pipeline. This does not apply to 'backend' compiles (LTO and ThinLTO /// pipeline. This does not apply to 'backend' compiles (LTO and ThinLTO
/// link-time pipelines). /// link-time pipelines).
void registerPipelineStartEPCallback( void registerPipelineStartEPCallback(
const std::function<void(ModulePassManager &)> &C) { const std::function<void(ModulePassManager &, OptimizationLevel)> &C) {
PipelineStartEPCallbacks.push_back(C); PipelineStartEPCallbacks.push_back(C);
} }
@ -722,7 +722,7 @@ private:
SmallVector<std::function<void(ModulePassManager &, OptimizationLevel)>, 2> SmallVector<std::function<void(ModulePassManager &, OptimizationLevel)>, 2>
OptimizerLastEPCallbacks; OptimizerLastEPCallbacks;
// Module callbacks // Module callbacks
SmallVector<std::function<void(ModulePassManager &)>, 2> SmallVector<std::function<void(ModulePassManager &, OptimizationLevel)>, 2>
PipelineStartEPCallbacks; PipelineStartEPCallbacks;
SmallVector<std::function<void(ModuleAnalysisManager &)>, 2> SmallVector<std::function<void(ModuleAnalysisManager &)>, 2>
ModuleAnalysisRegistrationCallbacks; ModuleAnalysisRegistrationCallbacks;

View File

@ -1319,7 +1319,7 @@ PassBuilder::buildPerModuleDefaultPipeline(OptimizationLevel Level,
// Apply module pipeline start EP callback. // Apply module pipeline start EP callback.
for (auto &C : PipelineStartEPCallbacks) for (auto &C : PipelineStartEPCallbacks)
C(MPM); C(MPM, Level);
if (PGOOpt && PGOOpt->SamplePGOSupport) if (PGOOpt && PGOOpt->SamplePGOSupport)
MPM.addPass(createModuleToFunctionPassAdaptor(AddDiscriminatorsPass())); MPM.addPass(createModuleToFunctionPassAdaptor(AddDiscriminatorsPass()));
@ -1348,7 +1348,7 @@ PassBuilder::buildThinLTOPreLinkDefaultPipeline(OptimizationLevel Level) {
// Apply module pipeline start EP callback. // Apply module pipeline start EP callback.
for (auto &C : PipelineStartEPCallbacks) for (auto &C : PipelineStartEPCallbacks)
C(MPM); C(MPM, Level);
// If we are planning to perform ThinLTO later, we don't bloat the code with // If we are planning to perform ThinLTO later, we don't bloat the code with
// unrolling/vectorization/... now. Just simplify the module as much as we // unrolling/vectorization/... now. Just simplify the module as much as we

View File

@ -125,7 +125,8 @@ void BPFTargetMachine::adjustPassManager(PassManagerBuilder &Builder) {
void BPFTargetMachine::registerPassBuilderCallbacks(PassBuilder &PB, void BPFTargetMachine::registerPassBuilderCallbacks(PassBuilder &PB,
bool DebugPassManager) { bool DebugPassManager) {
PB.registerPipelineStartEPCallback([=](ModulePassManager &MPM) { PB.registerPipelineStartEPCallback(
[=](ModulePassManager &MPM, PassBuilder::OptimizationLevel) {
FunctionPassManager FPM(DebugPassManager); FunctionPassManager FPM(DebugPassManager);
FPM.addPass(BPFAbstractMemberAccessPass(this)); FPM.addPass(BPFAbstractMemberAccessPass(this));
FPM.addPass(BPFPreserveDITypePass()); FPM.addPass(BPFPreserveDITypePass());

View File

@ -190,7 +190,8 @@ static void registerEPCallbacks(PassBuilder &PB) {
Err(PB.parsePassPipeline(PM, VectorizerStartEPPipeline)); Err(PB.parsePassPipeline(PM, VectorizerStartEPPipeline));
}); });
if (tryParsePipelineText<ModulePassManager>(PB, PipelineStartEPPipeline)) if (tryParsePipelineText<ModulePassManager>(PB, PipelineStartEPPipeline))
PB.registerPipelineStartEPCallback([&PB](ModulePassManager &PM) { PB.registerPipelineStartEPCallback(
[&PB](ModulePassManager &PM, PassBuilder::OptimizationLevel) {
ExitOnError Err("Unable to parse PipelineStartEP pipeline: "); ExitOnError Err("Unable to parse PipelineStartEP pipeline: ");
Err(PB.parsePassPipeline(PM, PipelineStartEPPipeline)); Err(PB.parsePassPipeline(PM, PipelineStartEPPipeline));
}); });