125 lines
		
	
	
		
			4.7 KiB
		
	
	
	
		
			C++
		
	
	
	
			
		
		
	
	
			125 lines
		
	
	
		
			4.7 KiB
		
	
	
	
		
			C++
		
	
	
	
//===-- MachineFunctionPass.cpp -------------------------------------------===//
 | 
						|
//
 | 
						|
// 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 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);
 | 
						|
}
 |