forked from OSchip/llvm-project
				
			
		
			
				
	
	
		
			126 lines
		
	
	
		
			4.7 KiB
		
	
	
	
		
			C++
		
	
	
	
			
		
		
	
	
			126 lines
		
	
	
		
			4.7 KiB
		
	
	
	
		
			C++
		
	
	
	
| //===-- MachineFunctionPass.cpp -------------------------------------------===//
 | |
| //
 | |
| //                     The LLVM Compiler Infrastructure
 | |
| //
 | |
| // This file is distributed under the University of Illinois Open Source
 | |
| // License. See LICENSE.TXT for details.
 | |
| //
 | |
| //===----------------------------------------------------------------------===//
 | |
| //
 | |
| // This file contains the definitions of the MachineFunctionPass members.
 | |
| //
 | |
| //===----------------------------------------------------------------------===//
 | |
| 
 | |
| #include "llvm/CodeGen/MachineFunctionPass.h"
 | |
| #include "llvm/Analysis/AliasAnalysis.h"
 | |
| #include "llvm/Analysis/BasicAliasAnalysis.h"
 | |
| #include "llvm/Analysis/DominanceFrontier.h"
 | |
| #include "llvm/Analysis/GlobalsModRef.h"
 | |
| #include "llvm/Analysis/IVUsers.h"
 | |
| #include "llvm/Analysis/LoopInfo.h"
 | |
| #include "llvm/Analysis/MemoryDependenceAnalysis.h"
 | |
| #include "llvm/Analysis/ScalarEvolution.h"
 | |
| #include "llvm/Analysis/ScalarEvolutionAliasAnalysis.h"
 | |
| #include "llvm/CodeGen/MachineFunction.h"
 | |
| #include "llvm/CodeGen/MachineModuleInfo.h"
 | |
| #include "llvm/CodeGen/MachineOptimizationRemarkEmitter.h"
 | |
| #include "llvm/CodeGen/Passes.h"
 | |
| #include "llvm/IR/Dominators.h"
 | |
| #include "llvm/IR/Function.h"
 | |
| 
 | |
| using namespace llvm;
 | |
| using namespace ore;
 | |
| 
 | |
| Pass *MachineFunctionPass::createPrinterPass(raw_ostream &O,
 | |
|                                              const std::string &Banner) const {
 | |
|   return createMachineFunctionPrinterPass(O, Banner);
 | |
| }
 | |
| 
 | |
| bool MachineFunctionPass::runOnFunction(Function &F) {
 | |
|   // Do not codegen any 'available_externally' functions at all, they have
 | |
|   // definitions outside the translation unit.
 | |
|   if (F.hasAvailableExternallyLinkage())
 | |
|     return false;
 | |
| 
 | |
|   MachineModuleInfo &MMI = getAnalysis<MachineModuleInfo>();
 | |
|   MachineFunction &MF = MMI.getOrCreateMachineFunction(F);
 | |
| 
 | |
|   MachineFunctionProperties &MFProps = MF.getProperties();
 | |
| 
 | |
| #ifndef NDEBUG
 | |
|   if (!MFProps.verifyRequiredProperties(RequiredProperties)) {
 | |
|     errs() << "MachineFunctionProperties required by " << getPassName()
 | |
|            << " pass are not met by function " << F.getName() << ".\n"
 | |
|            << "Required properties: ";
 | |
|     RequiredProperties.print(errs());
 | |
|     errs() << "\nCurrent properties: ";
 | |
|     MFProps.print(errs());
 | |
|     errs() << "\n";
 | |
|     llvm_unreachable("MachineFunctionProperties check failed");
 | |
|   }
 | |
| #endif
 | |
|   // Collect the MI count of the function before the pass.
 | |
|   unsigned CountBefore, CountAfter;
 | |
| 
 | |
|   // Check if the user asked for size remarks.
 | |
|   bool ShouldEmitSizeRemarks =
 | |
|       F.getParent()->shouldEmitInstrCountChangedRemark();
 | |
| 
 | |
|   // If we want size remarks, collect the number of MachineInstrs in our
 | |
|   // MachineFunction before the pass runs.
 | |
|   if (ShouldEmitSizeRemarks)
 | |
|     CountBefore = MF.getInstructionCount();
 | |
| 
 | |
|   bool RV = runOnMachineFunction(MF);
 | |
| 
 | |
|   if (ShouldEmitSizeRemarks) {
 | |
|     // We wanted size remarks. Check if there was a change to the number of
 | |
|     // MachineInstrs in the module. Emit a remark if there was a change.
 | |
|     CountAfter = MF.getInstructionCount();
 | |
|     if (CountBefore != CountAfter) {
 | |
|       MachineOptimizationRemarkEmitter MORE(MF, nullptr);
 | |
|       MORE.emit([&]() {
 | |
|         int64_t Delta = static_cast<int64_t>(CountAfter) -
 | |
|                         static_cast<int64_t>(CountBefore);
 | |
|         MachineOptimizationRemarkAnalysis R("size-info", "FunctionMISizeChange",
 | |
|                                             MF.getFunction().getSubprogram(),
 | |
|                                             &MF.front());
 | |
|         R << NV("Pass", getPassName())
 | |
|           << ": Function: " << NV("Function", F.getName()) << ": "
 | |
|           << "MI Instruction count changed from "
 | |
|           << NV("MIInstrsBefore", CountBefore) << " to "
 | |
|           << NV("MIInstrsAfter", CountAfter)
 | |
|           << "; Delta: " << NV("Delta", Delta);
 | |
|         return R;
 | |
|       });
 | |
|     }
 | |
|   }
 | |
| 
 | |
|   MFProps.set(SetProperties);
 | |
|   MFProps.reset(ClearedProperties);
 | |
|   return RV;
 | |
| }
 | |
| 
 | |
| void MachineFunctionPass::getAnalysisUsage(AnalysisUsage &AU) const {
 | |
|   AU.addRequired<MachineModuleInfo>();
 | |
|   AU.addPreserved<MachineModuleInfo>();
 | |
| 
 | |
|   // MachineFunctionPass preserves all LLVM IR passes, but there's no
 | |
|   // high-level way to express this. Instead, just list a bunch of
 | |
|   // passes explicitly. This does not include setPreservesCFG,
 | |
|   // because CodeGen overloads that to mean preserving the MachineBasicBlock
 | |
|   // CFG in addition to the LLVM IR CFG.
 | |
|   AU.addPreserved<BasicAAWrapperPass>();
 | |
|   AU.addPreserved<DominanceFrontierWrapperPass>();
 | |
|   AU.addPreserved<DominatorTreeWrapperPass>();
 | |
|   AU.addPreserved<AAResultsWrapperPass>();
 | |
|   AU.addPreserved<GlobalsAAWrapperPass>();
 | |
|   AU.addPreserved<IVUsersWrapperPass>();
 | |
|   AU.addPreserved<LoopInfoWrapperPass>();
 | |
|   AU.addPreserved<MemoryDependenceWrapperPass>();
 | |
|   AU.addPreserved<ScalarEvolutionWrapperPass>();
 | |
|   AU.addPreserved<SCEVAAWrapperPass>();
 | |
| 
 | |
|   FunctionPass::getAnalysisUsage(AU);
 | |
| }
 |