4d8cf2ae68
In D152399, we calculate BPI->BFI in MachineFunctionSplit pass just to use PSI->isFunctionHotInCallGraph, which is expensive. Instead, we can implement this directly with MBFI. Reviewer mentioned in the comment, that machine_size_opts already has isFunctionColdInCallGraph, isFunctionHotInCallGraphNthPercentile, etc implemented. These can be refactored and reused across MFS and machine size opts. 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
53 lines
2.0 KiB
C++
53 lines
2.0 KiB
C++
//===- MachineSizeOpts.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 machine IR code size optimization related
|
|
// code.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#include "llvm/CodeGen/MachineSizeOpts.h"
|
|
#include "llvm/CodeGen/MBFIWrapper.h"
|
|
#include "llvm/Analysis/ProfileSummaryInfo.h"
|
|
#include "llvm/CodeGen/MachineBlockFrequencyInfo.h"
|
|
|
|
using namespace llvm;
|
|
|
|
extern cl::opt<bool> EnablePGSO;
|
|
extern cl::opt<bool> PGSOLargeWorkingSetSizeOnly;
|
|
extern cl::opt<bool> ForcePGSO;
|
|
extern cl::opt<int> PgsoCutoffInstrProf;
|
|
extern cl::opt<int> PgsoCutoffSampleProf;
|
|
|
|
bool llvm::shouldOptimizeForSize(const MachineFunction *MF,
|
|
ProfileSummaryInfo *PSI,
|
|
const MachineBlockFrequencyInfo *MBFI,
|
|
PGSOQueryType QueryType) {
|
|
return shouldFuncOptimizeForSizeImpl(MF, PSI, MBFI, QueryType);
|
|
}
|
|
|
|
bool llvm::shouldOptimizeForSize(const MachineBasicBlock *MBB,
|
|
ProfileSummaryInfo *PSI,
|
|
const MachineBlockFrequencyInfo *MBFI,
|
|
PGSOQueryType QueryType) {
|
|
assert(MBB);
|
|
return shouldOptimizeForSizeImpl(MBB, PSI, MBFI, QueryType);
|
|
}
|
|
|
|
bool llvm::shouldOptimizeForSize(const MachineBasicBlock *MBB,
|
|
ProfileSummaryInfo *PSI,
|
|
MBFIWrapper *MBFIW,
|
|
PGSOQueryType QueryType) {
|
|
assert(MBB);
|
|
if (!PSI || !MBFIW)
|
|
return false;
|
|
BlockFrequency BlockFreq = MBFIW->getBlockFreq(MBB);
|
|
return shouldOptimizeForSizeImpl(BlockFreq, PSI, &MBFIW->getMBFI(),
|
|
QueryType);
|
|
}
|