1//===- MachineSizeOpts.cpp - code size optimization related code ----------===//
2//
3// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
4// See https://llvm.org/LICENSE.txt for license information.
5// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
6//
7//===----------------------------------------------------------------------===//
8//
9// This file contains some shared machine IR code size optimization related
10// code.
11//
12//===----------------------------------------------------------------------===//
13
14#include "llvm/CodeGen/MachineSizeOpts.h"
15#include "llvm/CodeGen/MBFIWrapper.h"
16#include "llvm/Analysis/ProfileSummaryInfo.h"
17#include "llvm/CodeGen/MachineBlockFrequencyInfo.h"
18
19using namespace llvm;
20
21extern cl::opt<bool> EnablePGSO;
22extern cl::opt<bool> PGSOLargeWorkingSetSizeOnly;
23extern cl::opt<bool> ForcePGSO;
24extern cl::opt<int> PgsoCutoffInstrProf;
25extern cl::opt<int> PgsoCutoffSampleProf;
26
27bool llvm::shouldOptimizeForSize(const MachineFunction *MF,
28 ProfileSummaryInfo *PSI,
29 const MachineBlockFrequencyInfo *MBFI,
30 PGSOQueryType QueryType) {
31 if (MF->getFunction().hasOptSize())
32 return true;
33 return shouldFuncOptimizeForSizeImpl(F: MF, PSI, BFI: MBFI, QueryType);
34}
35
36bool llvm::shouldOptimizeForSize(const MachineBasicBlock *MBB,
37 ProfileSummaryInfo *PSI,
38 const MachineBlockFrequencyInfo *MBFI,
39 PGSOQueryType QueryType) {
40 assert(MBB);
41 if (MBB->getParent()->getFunction().hasOptSize())
42 return true;
43 return shouldOptimizeForSizeImpl(BBOrBlockFreq: MBB, PSI, BFI: MBFI, QueryType);
44}
45
46bool llvm::shouldOptimizeForSize(const MachineBasicBlock *MBB,
47 ProfileSummaryInfo *PSI,
48 MBFIWrapper *MBFIW,
49 PGSOQueryType QueryType) {
50 assert(MBB);
51 if (MBB->getParent()->getFunction().hasOptSize())
52 return true;
53 if (!MBFIW)
54 return false;
55 BlockFrequency BlockFreq = MBFIW->getBlockFreq(MBB);
56 return shouldOptimizeForSizeImpl(BBOrBlockFreq: BlockFreq, PSI, BFI: &MBFIW->getMBFI(),
57 QueryType);
58}
59