| //=== RISCVPostLegalizerCombiner.cpp --------------------------*- C++ -*-===// |
| // |
| // 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 |
| // |
| //===----------------------------------------------------------------------===// |
| /// |
| /// \file |
| /// Post-legalization combines on generic MachineInstrs. |
| /// |
| /// The combines here must preserve instruction legality. |
| /// |
| /// Combines which don't rely on instruction legality should go in the |
| /// RISCVPreLegalizerCombiner. |
| /// |
| //===----------------------------------------------------------------------===// |
| |
| #include "RISCVTargetMachine.h" |
| #include "llvm/CodeGen/GlobalISel/CSEInfo.h" |
| #include "llvm/CodeGen/GlobalISel/Combiner.h" |
| #include "llvm/CodeGen/GlobalISel/CombinerHelper.h" |
| #include "llvm/CodeGen/GlobalISel/CombinerInfo.h" |
| #include "llvm/CodeGen/GlobalISel/GIMatchTableExecutorImpl.h" |
| #include "llvm/CodeGen/GlobalISel/GISelKnownBits.h" |
| #include "llvm/CodeGen/GlobalISel/GenericMachineInstrs.h" |
| #include "llvm/CodeGen/GlobalISel/MachineIRBuilder.h" |
| #include "llvm/CodeGen/MachineDominators.h" |
| #include "llvm/CodeGen/MachineFunctionPass.h" |
| #include "llvm/CodeGen/MachineRegisterInfo.h" |
| #include "llvm/CodeGen/TargetPassConfig.h" |
| |
| #define GET_GICOMBINER_DEPS |
| #include "RISCVGenPostLegalizeGICombiner.inc" |
| #undef GET_GICOMBINER_DEPS |
| |
| #define DEBUG_TYPE "riscv-postlegalizer-combiner" |
| |
| using namespace llvm; |
| |
| namespace { |
| |
| #define GET_GICOMBINER_TYPES |
| #include "RISCVGenPostLegalizeGICombiner.inc" |
| #undef GET_GICOMBINER_TYPES |
| |
| class RISCVPostLegalizerCombinerImpl : public Combiner { |
| protected: |
| // TODO: Make CombinerHelper methods const. |
| mutable CombinerHelper Helper; |
| const RISCVPostLegalizerCombinerImplRuleConfig &RuleConfig; |
| const RISCVSubtarget &STI; |
| |
| public: |
| RISCVPostLegalizerCombinerImpl( |
| MachineFunction &MF, CombinerInfo &CInfo, const TargetPassConfig *TPC, |
| GISelKnownBits &KB, GISelCSEInfo *CSEInfo, |
| const RISCVPostLegalizerCombinerImplRuleConfig &RuleConfig, |
| const RISCVSubtarget &STI, MachineDominatorTree *MDT, |
| const LegalizerInfo *LI); |
| |
| static const char *getName() { return "RISCVPostLegalizerCombiner"; } |
| |
| bool tryCombineAll(MachineInstr &I) const override; |
| |
| private: |
| #define GET_GICOMBINER_CLASS_MEMBERS |
| #include "RISCVGenPostLegalizeGICombiner.inc" |
| #undef GET_GICOMBINER_CLASS_MEMBERS |
| }; |
| |
| #define GET_GICOMBINER_IMPL |
| #include "RISCVGenPostLegalizeGICombiner.inc" |
| #undef GET_GICOMBINER_IMPL |
| |
| RISCVPostLegalizerCombinerImpl::RISCVPostLegalizerCombinerImpl( |
| MachineFunction &MF, CombinerInfo &CInfo, const TargetPassConfig *TPC, |
| GISelKnownBits &KB, GISelCSEInfo *CSEInfo, |
| const RISCVPostLegalizerCombinerImplRuleConfig &RuleConfig, |
| const RISCVSubtarget &STI, MachineDominatorTree *MDT, |
| const LegalizerInfo *LI) |
| : Combiner(MF, CInfo, TPC, &KB, CSEInfo), |
| Helper(Observer, B, /*IsPreLegalize*/ false, &KB, MDT, LI), |
| RuleConfig(RuleConfig), STI(STI), |
| #define GET_GICOMBINER_CONSTRUCTOR_INITS |
| #include "RISCVGenPostLegalizeGICombiner.inc" |
| #undef GET_GICOMBINER_CONSTRUCTOR_INITS |
| { |
| } |
| |
| class RISCVPostLegalizerCombiner : public MachineFunctionPass { |
| public: |
| static char ID; |
| |
| RISCVPostLegalizerCombiner(); |
| |
| StringRef getPassName() const override { |
| return "RISCVPostLegalizerCombiner"; |
| } |
| |
| bool runOnMachineFunction(MachineFunction &MF) override; |
| void getAnalysisUsage(AnalysisUsage &AU) const override; |
| |
| private: |
| RISCVPostLegalizerCombinerImplRuleConfig RuleConfig; |
| }; |
| } // end anonymous namespace |
| |
| void RISCVPostLegalizerCombiner::getAnalysisUsage(AnalysisUsage &AU) const { |
| AU.addRequired<TargetPassConfig>(); |
| AU.setPreservesCFG(); |
| getSelectionDAGFallbackAnalysisUsage(AU); |
| AU.addRequired<GISelKnownBitsAnalysis>(); |
| AU.addPreserved<GISelKnownBitsAnalysis>(); |
| AU.addRequired<MachineDominatorTree>(); |
| AU.addPreserved<MachineDominatorTree>(); |
| AU.addRequired<GISelCSEAnalysisWrapperPass>(); |
| AU.addPreserved<GISelCSEAnalysisWrapperPass>(); |
| MachineFunctionPass::getAnalysisUsage(AU); |
| } |
| |
| RISCVPostLegalizerCombiner::RISCVPostLegalizerCombiner() |
| : MachineFunctionPass(ID) { |
| initializeRISCVPostLegalizerCombinerPass(*PassRegistry::getPassRegistry()); |
| |
| if (!RuleConfig.parseCommandLineOption()) |
| report_fatal_error("Invalid rule identifier"); |
| } |
| |
| bool RISCVPostLegalizerCombiner::runOnMachineFunction(MachineFunction &MF) { |
| if (MF.getProperties().hasProperty( |
| MachineFunctionProperties::Property::FailedISel)) |
| return false; |
| assert(MF.getProperties().hasProperty( |
| MachineFunctionProperties::Property::Legalized) && |
| "Expected a legalized function?"); |
| auto *TPC = &getAnalysis<TargetPassConfig>(); |
| const Function &F = MF.getFunction(); |
| bool EnableOpt = |
| MF.getTarget().getOptLevel() != CodeGenOptLevel::None && !skipFunction(F); |
| |
| const RISCVSubtarget &ST = MF.getSubtarget<RISCVSubtarget>(); |
| const auto *LI = ST.getLegalizerInfo(); |
| |
| GISelKnownBits *KB = &getAnalysis<GISelKnownBitsAnalysis>().get(MF); |
| MachineDominatorTree *MDT = &getAnalysis<MachineDominatorTree>(); |
| GISelCSEAnalysisWrapper &Wrapper = |
| getAnalysis<GISelCSEAnalysisWrapperPass>().getCSEWrapper(); |
| auto *CSEInfo = &Wrapper.get(TPC->getCSEConfig()); |
| |
| CombinerInfo CInfo(/*AllowIllegalOps*/ true, /*ShouldLegalizeIllegal*/ false, |
| /*LegalizerInfo*/ nullptr, EnableOpt, F.hasOptSize(), |
| F.hasMinSize()); |
| RISCVPostLegalizerCombinerImpl Impl(MF, CInfo, TPC, *KB, CSEInfo, |
| RuleConfig, ST, MDT, LI); |
| return Impl.combineMachineInstrs(); |
| } |
| |
| char RISCVPostLegalizerCombiner::ID = 0; |
| INITIALIZE_PASS_BEGIN(RISCVPostLegalizerCombiner, DEBUG_TYPE, |
| "Combine RISC-V MachineInstrs after legalization", false, |
| false) |
| INITIALIZE_PASS_DEPENDENCY(TargetPassConfig) |
| INITIALIZE_PASS_DEPENDENCY(GISelKnownBitsAnalysis) |
| INITIALIZE_PASS_END(RISCVPostLegalizerCombiner, DEBUG_TYPE, |
| "Combine RISC-V MachineInstrs after legalization", false, |
| false) |
| |
| namespace llvm { |
| FunctionPass *createRISCVPostLegalizerCombiner() { |
| return new RISCVPostLegalizerCombiner(); |
| } |
| } // end namespace llvm |