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 19 using namespace llvm; 20 21 extern cl::opt<bool> EnablePGSO; 22 extern cl::opt<bool> PGSOLargeWorkingSetSizeOnly; 23 extern cl::opt<bool> ForcePGSO; 24 extern cl::opt<int> PgsoCutoffInstrProf; 25 extern cl::opt<int> PgsoCutoffSampleProf; 26 27 bool 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(MF, PSI, MBFI, QueryType); 34 } 35 36 bool 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(MBB, PSI, MBFI, QueryType); 44 } 45 46 bool 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(BlockFreq, PSI, &MBFIW->getMBFI(), 57 QueryType); 58 } 59