1//=== lib/CodeGen/GlobalISel/MipsPreLegalizerCombiner.cpp --------------===//
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 pass does combining of machine instructions at the generic MI level,
10// before the legalizer.
11//
12//===----------------------------------------------------------------------===//
13
14#include "MipsLegalizerInfo.h"
15#include "MipsTargetMachine.h"
16#include "llvm/CodeGen/GlobalISel/Combiner.h"
17#include "llvm/CodeGen/GlobalISel/CombinerHelper.h"
18#include "llvm/CodeGen/GlobalISel/CombinerInfo.h"
19#include "llvm/CodeGen/GlobalISel/GISelValueTracking.h"
20#include "llvm/CodeGen/MachineFunctionPass.h"
21#include "llvm/CodeGen/TargetPassConfig.h"
22
23#define DEBUG_TYPE "mips-prelegalizer-combiner"
24
25using namespace llvm;
26
27namespace {
28struct MipsPreLegalizerCombinerInfo : public CombinerInfo {
29public:
30 MipsPreLegalizerCombinerInfo()
31 : CombinerInfo(/*AllowIllegalOps*/ true, /*ShouldLegalizeIllegal*/ false,
32 /*LegalizerInfo*/ nullptr, /*EnableOpt*/ false,
33 /*EnableOptSize*/ false, /*EnableMinSize*/ false) {}
34};
35
36class MipsPreLegalizerCombinerImpl : public Combiner {
37protected:
38 const MipsSubtarget &STI;
39 const CombinerHelper Helper;
40
41public:
42 MipsPreLegalizerCombinerImpl(MachineFunction &MF, CombinerInfo &CInfo,
43 GISelValueTracking &VT, GISelCSEInfo *CSEInfo,
44 const MipsSubtarget &STI,
45 MachineDominatorTree *MDT,
46 const LegalizerInfo *LI)
47 : Combiner(MF, CInfo, &VT, CSEInfo), STI(STI),
48 Helper(Observer, B, /*IsPreLegalize*/ true, &VT, MDT, LI) {}
49
50 static const char *getName() { return "MipsPreLegalizerCombiner"; }
51
52 void setupGeneratedPerFunctionState(MachineFunction &MF) override {
53 // TODO: TableGen-erate this class' impl.
54 }
55
56 bool tryCombineAll(MachineInstr &MI) const override {
57 switch (MI.getOpcode()) {
58 default:
59 return false;
60 case TargetOpcode::G_MEMCPY_INLINE:
61 case TargetOpcode::G_MEMSET_INLINE:
62 return Helper.tryCombineMemCpyFamily(MI);
63 case TargetOpcode::G_LOAD:
64 case TargetOpcode::G_SEXTLOAD:
65 case TargetOpcode::G_ZEXTLOAD: {
66 // Don't attempt to combine non power of 2 loads or unaligned loads when
67 // subtarget doesn't support them.
68 auto MMO = *MI.memoperands_begin();
69 const MipsSubtarget &STI = MI.getMF()->getSubtarget<MipsSubtarget>();
70 if (!MMO->getSize().hasValue() ||
71 !isPowerOf2_64(Value: MMO->getSize().getValue()))
72 return false;
73 bool isUnaligned = MMO->getAlign() < MMO->getSize().getValue();
74 if (!STI.systemSupportsUnalignedAccess() && isUnaligned)
75 return false;
76
77 return Helper.tryCombineExtendingLoads(MI);
78 }
79 }
80
81 return false;
82 }
83};
84
85// Pass boilerplate
86// ================
87
88class MipsPreLegalizerCombiner : public MachineFunctionPass {
89public:
90 static char ID;
91
92 MipsPreLegalizerCombiner();
93
94 StringRef getPassName() const override { return "MipsPreLegalizerCombiner"; }
95
96 bool runOnMachineFunction(MachineFunction &MF) override;
97
98 void getAnalysisUsage(AnalysisUsage &AU) const override;
99};
100} // end anonymous namespace
101
102void MipsPreLegalizerCombiner::getAnalysisUsage(AnalysisUsage &AU) const {
103 AU.addRequired<GISelValueTrackingAnalysisLegacy>();
104 AU.addPreserved<GISelValueTrackingAnalysisLegacy>();
105 AU.setPreservesCFG();
106 getSelectionDAGFallbackAnalysisUsage(AU);
107 MachineFunctionPass::getAnalysisUsage(AU);
108}
109
110MipsPreLegalizerCombiner::MipsPreLegalizerCombiner()
111 : MachineFunctionPass(ID) {}
112
113bool MipsPreLegalizerCombiner::runOnMachineFunction(MachineFunction &MF) {
114 if (MF.getProperties().hasFailedISel())
115 return false;
116
117 const MipsSubtarget &ST = MF.getSubtarget<MipsSubtarget>();
118 const MipsLegalizerInfo *LI =
119 static_cast<const MipsLegalizerInfo *>(ST.getLegalizerInfo());
120
121 GISelValueTracking *VT =
122 &getAnalysis<GISelValueTrackingAnalysisLegacy>().get(MF);
123 MipsPreLegalizerCombinerInfo PCInfo;
124 MipsPreLegalizerCombinerImpl Impl(MF, PCInfo, *VT, /*CSEInfo*/ nullptr, ST,
125 /*MDT*/ nullptr, LI);
126 return Impl.combineMachineInstrs();
127}
128
129char MipsPreLegalizerCombiner::ID = 0;
130INITIALIZE_PASS_BEGIN(MipsPreLegalizerCombiner, DEBUG_TYPE,
131 "Combine Mips machine instrs before legalization", false,
132 false)
133INITIALIZE_PASS_DEPENDENCY(GISelValueTrackingAnalysisLegacy)
134INITIALIZE_PASS_END(MipsPreLegalizerCombiner, DEBUG_TYPE,
135 "Combine Mips machine instrs before legalization", false,
136 false)
137
138FunctionPass *llvm::createMipsPreLegalizeCombiner() {
139 return new MipsPreLegalizerCombiner();
140}
141