| Richard Sandiford | bdbb8af | 2013-08-05 10:58:53 +0000 | [diff] [blame] | 1 | //===-- SystemZElimCompare.cpp - Eliminate comparison instructions --------===// |
| 2 | // |
| Chandler Carruth | 2946cd7 | 2019-01-19 08:50:56 +0000 | [diff] [blame] | 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 |
| Richard Sandiford | bdbb8af | 2013-08-05 10:58:53 +0000 | [diff] [blame] | 6 | // |
| 7 | //===----------------------------------------------------------------------===// |
| 8 | // |
| 9 | // This pass: |
| 10 | // (1) tries to remove compares if CC already contains the required information |
| 11 | // (2) fuses compares and branches into COMPARE AND BRANCH instructions |
| 12 | // |
| 13 | //===----------------------------------------------------------------------===// |
| 14 | |
| Eugene Zelenko | 3943d2b | 2017-01-24 22:10:43 +0000 | [diff] [blame] | 15 | #include "SystemZ.h" |
| 16 | #include "SystemZInstrInfo.h" |
| Richard Sandiford | bdbb8af | 2013-08-05 10:58:53 +0000 | [diff] [blame] | 17 | #include "SystemZTargetMachine.h" |
| Eugene Zelenko | 3943d2b | 2017-01-24 22:10:43 +0000 | [diff] [blame] | 18 | #include "llvm/ADT/SmallVector.h" |
| Richard Sandiford | bdbb8af | 2013-08-05 10:58:53 +0000 | [diff] [blame] | 19 | #include "llvm/ADT/Statistic.h" |
| Eugene Zelenko | 3943d2b | 2017-01-24 22:10:43 +0000 | [diff] [blame] | 20 | #include "llvm/ADT/StringRef.h" |
| 21 | #include "llvm/CodeGen/MachineBasicBlock.h" |
| 22 | #include "llvm/CodeGen/MachineFunction.h" |
| Richard Sandiford | bdbb8af | 2013-08-05 10:58:53 +0000 | [diff] [blame] | 23 | #include "llvm/CodeGen/MachineFunctionPass.h" |
| Eugene Zelenko | 3943d2b | 2017-01-24 22:10:43 +0000 | [diff] [blame] | 24 | #include "llvm/CodeGen/MachineInstr.h" |
| Richard Sandiford | bdbb8af | 2013-08-05 10:58:53 +0000 | [diff] [blame] | 25 | #include "llvm/CodeGen/MachineInstrBuilder.h" |
| Eugene Zelenko | 3943d2b | 2017-01-24 22:10:43 +0000 | [diff] [blame] | 26 | #include "llvm/CodeGen/MachineOperand.h" |
| David Blaikie | b3bde2e | 2017-11-17 01:07:10 +0000 | [diff] [blame] | 27 | #include "llvm/CodeGen/TargetRegisterInfo.h" |
| 28 | #include "llvm/CodeGen/TargetSubtargetInfo.h" |
| Eugene Zelenko | 3943d2b | 2017-01-24 22:10:43 +0000 | [diff] [blame] | 29 | #include "llvm/MC/MCInstrDesc.h" |
| Eugene Zelenko | 3943d2b | 2017-01-24 22:10:43 +0000 | [diff] [blame] | 30 | #include <cassert> |
| 31 | #include <cstdint> |
| Richard Sandiford | bdbb8af | 2013-08-05 10:58:53 +0000 | [diff] [blame] | 32 | |
| 33 | using namespace llvm; |
| 34 | |
| Chandler Carruth | 84e68b2 | 2014-04-22 02:41:26 +0000 | [diff] [blame] | 35 | #define DEBUG_TYPE "systemz-elim-compare" |
| 36 | |
| Richard Sandiford | c212125 | 2013-08-05 11:23:46 +0000 | [diff] [blame] | 37 | STATISTIC(BranchOnCounts, "Number of branch-on-count instructions"); |
| Ulrich Weigand | 2d9e3d9 | 2016-11-28 13:59:22 +0000 | [diff] [blame] | 38 | STATISTIC(LoadAndTraps, "Number of load-and-trap instructions"); |
| Richard Sandiford | bdbb8af | 2013-08-05 10:58:53 +0000 | [diff] [blame] | 39 | STATISTIC(EliminatedComparisons, "Number of eliminated comparisons"); |
| 40 | STATISTIC(FusedComparisons, "Number of fused compare-and-branch instructions"); |
| 41 | |
| 42 | namespace { |
| Eugene Zelenko | 3943d2b | 2017-01-24 22:10:43 +0000 | [diff] [blame] | 43 | |
| Richard Sandiford | c231269 | 2014-03-06 10:38:30 +0000 | [diff] [blame] | 44 | // Represents the references to a particular register in one or more |
| 45 | // instructions. |
| 46 | struct Reference { |
| Eugene Zelenko | 3943d2b | 2017-01-24 22:10:43 +0000 | [diff] [blame] | 47 | Reference() = default; |
| Richard Sandiford | c212125 | 2013-08-05 11:23:46 +0000 | [diff] [blame] | 48 | |
| Richard Sandiford | c231269 | 2014-03-06 10:38:30 +0000 | [diff] [blame] | 49 | Reference &operator|=(const Reference &Other) { |
| 50 | Def |= Other.Def; |
| Richard Sandiford | c231269 | 2014-03-06 10:38:30 +0000 | [diff] [blame] | 51 | Use |= Other.Use; |
| Richard Sandiford | c231269 | 2014-03-06 10:38:30 +0000 | [diff] [blame] | 52 | return *this; |
| 53 | } |
| Richard Sandiford | c212125 | 2013-08-05 11:23:46 +0000 | [diff] [blame] | 54 | |
| Aaron Ballman | b46962f | 2015-02-15 22:00:20 +0000 | [diff] [blame] | 55 | explicit operator bool() const { return Def || Use; } |
| Richard Sandiford | c212125 | 2013-08-05 11:23:46 +0000 | [diff] [blame] | 56 | |
| Richard Sandiford | c231269 | 2014-03-06 10:38:30 +0000 | [diff] [blame] | 57 | // True if the register is defined or used in some form, either directly or |
| 58 | // via a sub- or super-register. |
| Eugene Zelenko | 3943d2b | 2017-01-24 22:10:43 +0000 | [diff] [blame] | 59 | bool Def = false; |
| 60 | bool Use = false; |
| Richard Sandiford | c231269 | 2014-03-06 10:38:30 +0000 | [diff] [blame] | 61 | }; |
| Richard Sandiford | c212125 | 2013-08-05 11:23:46 +0000 | [diff] [blame] | 62 | |
| Richard Sandiford | c231269 | 2014-03-06 10:38:30 +0000 | [diff] [blame] | 63 | class SystemZElimCompare : public MachineFunctionPass { |
| 64 | public: |
| 65 | static char ID; |
| Eugene Zelenko | 3943d2b | 2017-01-24 22:10:43 +0000 | [diff] [blame] | 66 | |
| Richard Sandiford | c231269 | 2014-03-06 10:38:30 +0000 | [diff] [blame] | 67 | SystemZElimCompare(const SystemZTargetMachine &tm) |
| Eugene Zelenko | 3943d2b | 2017-01-24 22:10:43 +0000 | [diff] [blame] | 68 | : MachineFunctionPass(ID) {} |
| Richard Sandiford | bdbb8af | 2013-08-05 10:58:53 +0000 | [diff] [blame] | 69 | |
| Mehdi Amini | 117296c | 2016-10-01 02:56:57 +0000 | [diff] [blame] | 70 | StringRef getPassName() const override { |
| Richard Sandiford | c231269 | 2014-03-06 10:38:30 +0000 | [diff] [blame] | 71 | return "SystemZ Comparison Elimination"; |
| 72 | } |
| Richard Sandiford | bdbb8af | 2013-08-05 10:58:53 +0000 | [diff] [blame] | 73 | |
| Richard Sandiford | 28c111e | 2014-03-06 11:00:15 +0000 | [diff] [blame] | 74 | bool processBlock(MachineBasicBlock &MBB); |
| Craig Topper | 9d74a5a | 2014-04-29 07:58:41 +0000 | [diff] [blame] | 75 | bool runOnMachineFunction(MachineFunction &F) override; |
| Eugene Zelenko | 3943d2b | 2017-01-24 22:10:43 +0000 | [diff] [blame] | 76 | |
| Derek Schuff | 1dbf7a5 | 2016-04-04 17:09:25 +0000 | [diff] [blame] | 77 | MachineFunctionProperties getRequiredProperties() const override { |
| 78 | return MachineFunctionProperties().set( |
| Matthias Braun | 1eb4736 | 2016-08-25 01:27:13 +0000 | [diff] [blame] | 79 | MachineFunctionProperties::Property::NoVRegs); |
| Derek Schuff | 1dbf7a5 | 2016-04-04 17:09:25 +0000 | [diff] [blame] | 80 | } |
| Richard Sandiford | bdbb8af | 2013-08-05 10:58:53 +0000 | [diff] [blame] | 81 | |
| Richard Sandiford | c231269 | 2014-03-06 10:38:30 +0000 | [diff] [blame] | 82 | private: |
| Duncan P. N. Exon Smith | 4565ec0 | 2016-07-12 01:39:01 +0000 | [diff] [blame] | 83 | Reference getRegReferences(MachineInstr &MI, unsigned Reg); |
| 84 | bool convertToBRCT(MachineInstr &MI, MachineInstr &Compare, |
| Richard Sandiford | c231269 | 2014-03-06 10:38:30 +0000 | [diff] [blame] | 85 | SmallVectorImpl<MachineInstr *> &CCUsers); |
| Ulrich Weigand | 2d9e3d9 | 2016-11-28 13:59:22 +0000 | [diff] [blame] | 86 | bool convertToLoadAndTrap(MachineInstr &MI, MachineInstr &Compare, |
| 87 | SmallVectorImpl<MachineInstr *> &CCUsers); |
| Jonas Paulsson | 776a81a | 2018-01-15 15:41:26 +0000 | [diff] [blame] | 88 | bool convertToLoadAndTest(MachineInstr &MI, MachineInstr &Compare, |
| 89 | SmallVectorImpl<MachineInstr *> &CCUsers); |
| Duncan P. N. Exon Smith | 4565ec0 | 2016-07-12 01:39:01 +0000 | [diff] [blame] | 90 | bool adjustCCMasksForInstr(MachineInstr &MI, MachineInstr &Compare, |
| Jonas Paulsson | 776a81a | 2018-01-15 15:41:26 +0000 | [diff] [blame] | 91 | SmallVectorImpl<MachineInstr *> &CCUsers, |
| 92 | unsigned ConvOpc = 0); |
| Duncan P. N. Exon Smith | 4565ec0 | 2016-07-12 01:39:01 +0000 | [diff] [blame] | 93 | bool optimizeCompareZero(MachineInstr &Compare, |
| Richard Sandiford | c231269 | 2014-03-06 10:38:30 +0000 | [diff] [blame] | 94 | SmallVectorImpl<MachineInstr *> &CCUsers); |
| Duncan P. N. Exon Smith | 4565ec0 | 2016-07-12 01:39:01 +0000 | [diff] [blame] | 95 | bool fuseCompareOperations(MachineInstr &Compare, |
| Zhan Jun Liau | ab42cbc | 2016-06-10 19:58:10 +0000 | [diff] [blame] | 96 | SmallVectorImpl<MachineInstr *> &CCUsers); |
| Richard Sandiford | bdbb8af | 2013-08-05 10:58:53 +0000 | [diff] [blame] | 97 | |
| Eugene Zelenko | 3943d2b | 2017-01-24 22:10:43 +0000 | [diff] [blame] | 98 | const SystemZInstrInfo *TII = nullptr; |
| 99 | const TargetRegisterInfo *TRI = nullptr; |
| Richard Sandiford | c231269 | 2014-03-06 10:38:30 +0000 | [diff] [blame] | 100 | }; |
| Richard Sandiford | bdbb8af | 2013-08-05 10:58:53 +0000 | [diff] [blame] | 101 | |
| Richard Sandiford | c231269 | 2014-03-06 10:38:30 +0000 | [diff] [blame] | 102 | char SystemZElimCompare::ID = 0; |
| Richard Sandiford | bdbb8af | 2013-08-05 10:58:53 +0000 | [diff] [blame] | 103 | |
| Eugene Zelenko | 3943d2b | 2017-01-24 22:10:43 +0000 | [diff] [blame] | 104 | } // end anonymous namespace |
| Richard Sandiford | bdbb8af | 2013-08-05 10:58:53 +0000 | [diff] [blame] | 105 | |
| 106 | // Return true if CC is live out of MBB. |
| Richard Sandiford | 28c111e | 2014-03-06 11:00:15 +0000 | [diff] [blame] | 107 | static bool isCCLiveOut(MachineBasicBlock &MBB) { |
| 108 | for (auto SI = MBB.succ_begin(), SE = MBB.succ_end(); SI != SE; ++SI) |
| Richard Sandiford | bdbb8af | 2013-08-05 10:58:53 +0000 | [diff] [blame] | 109 | if ((*SI)->isLiveIn(SystemZ::CC)) |
| 110 | return true; |
| 111 | return false; |
| 112 | } |
| 113 | |
| Jonas Paulsson | b0e8a2e | 2017-09-21 13:52:24 +0000 | [diff] [blame] | 114 | // Returns true if MI is an instruction whose output equals the value in Reg. |
| 115 | static bool preservesValueOf(MachineInstr &MI, unsigned Reg) { |
| Duncan P. N. Exon Smith | 4565ec0 | 2016-07-12 01:39:01 +0000 | [diff] [blame] | 116 | switch (MI.getOpcode()) { |
| Richard Sandiford | b49a3ab | 2013-08-05 11:03:20 +0000 | [diff] [blame] | 117 | case SystemZ::LR: |
| 118 | case SystemZ::LGR: |
| 119 | case SystemZ::LGFR: |
| 120 | case SystemZ::LTR: |
| 121 | case SystemZ::LTGR: |
| 122 | case SystemZ::LTGFR: |
| Richard Sandiford | 0897fce | 2013-08-07 11:10:06 +0000 | [diff] [blame] | 123 | case SystemZ::LER: |
| 124 | case SystemZ::LDR: |
| 125 | case SystemZ::LXR: |
| 126 | case SystemZ::LTEBR: |
| 127 | case SystemZ::LTDBR: |
| 128 | case SystemZ::LTXBR: |
| Duncan P. N. Exon Smith | 4565ec0 | 2016-07-12 01:39:01 +0000 | [diff] [blame] | 129 | if (MI.getOperand(1).getReg() == Reg) |
| Richard Sandiford | b49a3ab | 2013-08-05 11:03:20 +0000 | [diff] [blame] | 130 | return true; |
| 131 | } |
| 132 | |
| Richard Sandiford | bdbb8af | 2013-08-05 10:58:53 +0000 | [diff] [blame] | 133 | return false; |
| 134 | } |
| 135 | |
| Jonas Paulsson | b0e8a2e | 2017-09-21 13:52:24 +0000 | [diff] [blame] | 136 | // Return true if any CC result of MI would (perhaps after conversion) |
| 137 | // reflect the value of Reg. |
| 138 | static bool resultTests(MachineInstr &MI, unsigned Reg) { |
| 139 | if (MI.getNumOperands() > 0 && MI.getOperand(0).isReg() && |
| 140 | MI.getOperand(0).isDef() && MI.getOperand(0).getReg() == Reg) |
| 141 | return true; |
| 142 | |
| 143 | return (preservesValueOf(MI, Reg)); |
| 144 | } |
| 145 | |
| Jonas Paulsson | ee3685f | 2015-10-09 11:27:44 +0000 | [diff] [blame] | 146 | // Describe the references to Reg or any of its aliases in MI. |
| Duncan P. N. Exon Smith | 4565ec0 | 2016-07-12 01:39:01 +0000 | [diff] [blame] | 147 | Reference SystemZElimCompare::getRegReferences(MachineInstr &MI, unsigned Reg) { |
| Richard Sandiford | c212125 | 2013-08-05 11:23:46 +0000 | [diff] [blame] | 148 | Reference Ref; |
| Jonas Paulsson | 961c47e | 2019-01-23 07:42:26 +0000 | [diff] [blame] | 149 | if (MI.isDebugInstr()) |
| 150 | return Ref; |
| 151 | |
| Duncan P. N. Exon Smith | 4565ec0 | 2016-07-12 01:39:01 +0000 | [diff] [blame] | 152 | for (unsigned I = 0, E = MI.getNumOperands(); I != E; ++I) { |
| 153 | const MachineOperand &MO = MI.getOperand(I); |
| Richard Sandiford | c212125 | 2013-08-05 11:23:46 +0000 | [diff] [blame] | 154 | if (MO.isReg()) { |
| 155 | if (unsigned MOReg = MO.getReg()) { |
| Jonas Paulsson | ee3685f | 2015-10-09 11:27:44 +0000 | [diff] [blame] | 156 | if (TRI->regsOverlap(MOReg, Reg)) { |
| 157 | if (MO.isUse()) |
| Richard Sandiford | c212125 | 2013-08-05 11:23:46 +0000 | [diff] [blame] | 158 | Ref.Use = true; |
| Jonas Paulsson | ee3685f | 2015-10-09 11:27:44 +0000 | [diff] [blame] | 159 | else if (MO.isDef()) |
| Richard Sandiford | c212125 | 2013-08-05 11:23:46 +0000 | [diff] [blame] | 160 | Ref.Def = true; |
| Richard Sandiford | c212125 | 2013-08-05 11:23:46 +0000 | [diff] [blame] | 161 | } |
| 162 | } |
| 163 | } |
| 164 | } |
| 165 | return Ref; |
| 166 | } |
| 167 | |
| Jonas Paulsson | 5d3fbd3 | 2015-10-08 07:40:23 +0000 | [diff] [blame] | 168 | // Return true if this is a load and test which can be optimized the |
| 169 | // same way as compare instruction. |
| Duncan P. N. Exon Smith | 4565ec0 | 2016-07-12 01:39:01 +0000 | [diff] [blame] | 170 | static bool isLoadAndTestAsCmp(MachineInstr &MI) { |
| Jonas Paulsson | 5d3fbd3 | 2015-10-08 07:40:23 +0000 | [diff] [blame] | 171 | // If we during isel used a load-and-test as a compare with 0, the |
| 172 | // def operand is dead. |
| Duncan P. N. Exon Smith | 4565ec0 | 2016-07-12 01:39:01 +0000 | [diff] [blame] | 173 | return (MI.getOpcode() == SystemZ::LTEBR || |
| 174 | MI.getOpcode() == SystemZ::LTDBR || |
| 175 | MI.getOpcode() == SystemZ::LTXBR) && |
| 176 | MI.getOperand(0).isDead(); |
| Jonas Paulsson | 5d3fbd3 | 2015-10-08 07:40:23 +0000 | [diff] [blame] | 177 | } |
| 178 | |
| 179 | // Return the source register of Compare, which is the unknown value |
| 180 | // being tested. |
| Duncan P. N. Exon Smith | 4565ec0 | 2016-07-12 01:39:01 +0000 | [diff] [blame] | 181 | static unsigned getCompareSourceReg(MachineInstr &Compare) { |
| Jonas Paulsson | 5d3fbd3 | 2015-10-08 07:40:23 +0000 | [diff] [blame] | 182 | unsigned reg = 0; |
| Duncan P. N. Exon Smith | 4565ec0 | 2016-07-12 01:39:01 +0000 | [diff] [blame] | 183 | if (Compare.isCompare()) |
| 184 | reg = Compare.getOperand(0).getReg(); |
| Jonas Paulsson | 5d3fbd3 | 2015-10-08 07:40:23 +0000 | [diff] [blame] | 185 | else if (isLoadAndTestAsCmp(Compare)) |
| Duncan P. N. Exon Smith | 4565ec0 | 2016-07-12 01:39:01 +0000 | [diff] [blame] | 186 | reg = Compare.getOperand(1).getReg(); |
| Eugene Zelenko | 3943d2b | 2017-01-24 22:10:43 +0000 | [diff] [blame] | 187 | assert(reg); |
| Jonas Paulsson | 5d3fbd3 | 2015-10-08 07:40:23 +0000 | [diff] [blame] | 188 | |
| 189 | return reg; |
| 190 | } |
| 191 | |
| Richard Sandiford | c212125 | 2013-08-05 11:23:46 +0000 | [diff] [blame] | 192 | // Compare compares the result of MI against zero. If MI is an addition |
| 193 | // of -1 and if CCUsers is a single branch on nonzero, eliminate the addition |
| Ulrich Weigand | 7583991 | 2016-11-28 13:40:08 +0000 | [diff] [blame] | 194 | // and convert the branch to a BRCT(G) or BRCTH. Return true on success. |
| Duncan P. N. Exon Smith | 4565ec0 | 2016-07-12 01:39:01 +0000 | [diff] [blame] | 195 | bool SystemZElimCompare::convertToBRCT( |
| 196 | MachineInstr &MI, MachineInstr &Compare, |
| 197 | SmallVectorImpl<MachineInstr *> &CCUsers) { |
| Richard Sandiford | c212125 | 2013-08-05 11:23:46 +0000 | [diff] [blame] | 198 | // Check whether we have an addition of -1. |
| Duncan P. N. Exon Smith | 4565ec0 | 2016-07-12 01:39:01 +0000 | [diff] [blame] | 199 | unsigned Opcode = MI.getOpcode(); |
| Richard Sandiford | c212125 | 2013-08-05 11:23:46 +0000 | [diff] [blame] | 200 | unsigned BRCT; |
| 201 | if (Opcode == SystemZ::AHI) |
| 202 | BRCT = SystemZ::BRCT; |
| 203 | else if (Opcode == SystemZ::AGHI) |
| 204 | BRCT = SystemZ::BRCTG; |
| Ulrich Weigand | 7583991 | 2016-11-28 13:40:08 +0000 | [diff] [blame] | 205 | else if (Opcode == SystemZ::AIH) |
| 206 | BRCT = SystemZ::BRCTH; |
| Richard Sandiford | c212125 | 2013-08-05 11:23:46 +0000 | [diff] [blame] | 207 | else |
| 208 | return false; |
| Duncan P. N. Exon Smith | 4565ec0 | 2016-07-12 01:39:01 +0000 | [diff] [blame] | 209 | if (MI.getOperand(2).getImm() != -1) |
| Richard Sandiford | c212125 | 2013-08-05 11:23:46 +0000 | [diff] [blame] | 210 | return false; |
| 211 | |
| 212 | // Check whether we have a single JLH. |
| 213 | if (CCUsers.size() != 1) |
| 214 | return false; |
| 215 | MachineInstr *Branch = CCUsers[0]; |
| 216 | if (Branch->getOpcode() != SystemZ::BRC || |
| 217 | Branch->getOperand(0).getImm() != SystemZ::CCMASK_ICMP || |
| 218 | Branch->getOperand(1).getImm() != SystemZ::CCMASK_CMP_NE) |
| 219 | return false; |
| 220 | |
| 221 | // We already know that there are no references to the register between |
| 222 | // MI and Compare. Make sure that there are also no references between |
| 223 | // Compare and Branch. |
| Jonas Paulsson | 5d3fbd3 | 2015-10-08 07:40:23 +0000 | [diff] [blame] | 224 | unsigned SrcReg = getCompareSourceReg(Compare); |
| Richard Sandiford | c212125 | 2013-08-05 11:23:46 +0000 | [diff] [blame] | 225 | MachineBasicBlock::iterator MBBI = Compare, MBBE = Branch; |
| 226 | for (++MBBI; MBBI != MBBE; ++MBBI) |
| Duncan P. N. Exon Smith | 4565ec0 | 2016-07-12 01:39:01 +0000 | [diff] [blame] | 227 | if (getRegReferences(*MBBI, SrcReg)) |
| Richard Sandiford | c212125 | 2013-08-05 11:23:46 +0000 | [diff] [blame] | 228 | return false; |
| 229 | |
| Ulrich Weigand | 7583991 | 2016-11-28 13:40:08 +0000 | [diff] [blame] | 230 | // The transformation is OK. Rebuild Branch as a BRCT(G) or BRCTH. |
| Richard Sandiford | c212125 | 2013-08-05 11:23:46 +0000 | [diff] [blame] | 231 | MachineOperand Target(Branch->getOperand(2)); |
| Jonas Paulsson | 63a2b68 | 2015-10-10 07:14:24 +0000 | [diff] [blame] | 232 | while (Branch->getNumOperands()) |
| 233 | Branch->RemoveOperand(0); |
| Richard Sandiford | c212125 | 2013-08-05 11:23:46 +0000 | [diff] [blame] | 234 | Branch->setDesc(TII->get(BRCT)); |
| Ulrich Weigand | 7583991 | 2016-11-28 13:40:08 +0000 | [diff] [blame] | 235 | MachineInstrBuilder MIB(*Branch->getParent()->getParent(), Branch); |
| Diana Picus | 116bbab | 2017-01-13 09:58:52 +0000 | [diff] [blame] | 236 | MIB.add(MI.getOperand(0)).add(MI.getOperand(1)).add(Target); |
| Ulrich Weigand | 7583991 | 2016-11-28 13:40:08 +0000 | [diff] [blame] | 237 | // Add a CC def to BRCT(G), since we may have to split them again if the |
| 238 | // branch displacement overflows. BRCTH has a 32-bit displacement, so |
| 239 | // this is not necessary there. |
| 240 | if (BRCT != SystemZ::BRCTH) |
| 241 | MIB.addReg(SystemZ::CC, RegState::ImplicitDefine | RegState::Dead); |
| Duncan P. N. Exon Smith | 4565ec0 | 2016-07-12 01:39:01 +0000 | [diff] [blame] | 242 | MI.eraseFromParent(); |
| Richard Sandiford | c212125 | 2013-08-05 11:23:46 +0000 | [diff] [blame] | 243 | return true; |
| 244 | } |
| 245 | |
| Ulrich Weigand | 2d9e3d9 | 2016-11-28 13:59:22 +0000 | [diff] [blame] | 246 | // Compare compares the result of MI against zero. If MI is a suitable load |
| 247 | // instruction and if CCUsers is a single conditional trap on zero, eliminate |
| 248 | // the load and convert the branch to a load-and-trap. Return true on success. |
| 249 | bool SystemZElimCompare::convertToLoadAndTrap( |
| 250 | MachineInstr &MI, MachineInstr &Compare, |
| 251 | SmallVectorImpl<MachineInstr *> &CCUsers) { |
| 252 | unsigned LATOpcode = TII->getLoadAndTrap(MI.getOpcode()); |
| 253 | if (!LATOpcode) |
| 254 | return false; |
| 255 | |
| 256 | // Check whether we have a single CondTrap that traps on zero. |
| 257 | if (CCUsers.size() != 1) |
| 258 | return false; |
| 259 | MachineInstr *Branch = CCUsers[0]; |
| 260 | if (Branch->getOpcode() != SystemZ::CondTrap || |
| 261 | Branch->getOperand(0).getImm() != SystemZ::CCMASK_ICMP || |
| 262 | Branch->getOperand(1).getImm() != SystemZ::CCMASK_CMP_EQ) |
| 263 | return false; |
| 264 | |
| 265 | // We already know that there are no references to the register between |
| 266 | // MI and Compare. Make sure that there are also no references between |
| 267 | // Compare and Branch. |
| 268 | unsigned SrcReg = getCompareSourceReg(Compare); |
| 269 | MachineBasicBlock::iterator MBBI = Compare, MBBE = Branch; |
| 270 | for (++MBBI; MBBI != MBBE; ++MBBI) |
| 271 | if (getRegReferences(*MBBI, SrcReg)) |
| 272 | return false; |
| 273 | |
| 274 | // The transformation is OK. Rebuild Branch as a load-and-trap. |
| Ulrich Weigand | 2d9e3d9 | 2016-11-28 13:59:22 +0000 | [diff] [blame] | 275 | while (Branch->getNumOperands()) |
| 276 | Branch->RemoveOperand(0); |
| 277 | Branch->setDesc(TII->get(LATOpcode)); |
| 278 | MachineInstrBuilder(*Branch->getParent()->getParent(), Branch) |
| Diana Picus | 116bbab | 2017-01-13 09:58:52 +0000 | [diff] [blame] | 279 | .add(MI.getOperand(0)) |
| 280 | .add(MI.getOperand(1)) |
| 281 | .add(MI.getOperand(2)) |
| 282 | .add(MI.getOperand(3)); |
| Ulrich Weigand | 2d9e3d9 | 2016-11-28 13:59:22 +0000 | [diff] [blame] | 283 | MI.eraseFromParent(); |
| 284 | return true; |
| 285 | } |
| 286 | |
| Richard Sandiford | b49a3ab | 2013-08-05 11:03:20 +0000 | [diff] [blame] | 287 | // If MI is a load instruction, try to convert it into a LOAD AND TEST. |
| 288 | // Return true on success. |
| Jonas Paulsson | 776a81a | 2018-01-15 15:41:26 +0000 | [diff] [blame] | 289 | bool SystemZElimCompare::convertToLoadAndTest( |
| 290 | MachineInstr &MI, MachineInstr &Compare, |
| 291 | SmallVectorImpl<MachineInstr *> &CCUsers) { |
| 292 | |
| 293 | // Try to adjust CC masks for the LOAD AND TEST opcode that could replace MI. |
| Duncan P. N. Exon Smith | 4565ec0 | 2016-07-12 01:39:01 +0000 | [diff] [blame] | 294 | unsigned Opcode = TII->getLoadAndTest(MI.getOpcode()); |
| Jonas Paulsson | 776a81a | 2018-01-15 15:41:26 +0000 | [diff] [blame] | 295 | if (!Opcode || !adjustCCMasksForInstr(MI, Compare, CCUsers, Opcode)) |
| Richard Sandiford | b49a3ab | 2013-08-05 11:03:20 +0000 | [diff] [blame] | 296 | return false; |
| 297 | |
| Jonas Paulsson | e80d405 | 2018-06-07 05:59:07 +0000 | [diff] [blame] | 298 | // Rebuild to get the CC operand in the right place. |
| Chandler Carruth | c73c030 | 2018-08-16 21:30:05 +0000 | [diff] [blame] | 299 | auto MIB = BuildMI(*MI.getParent(), MI, MI.getDebugLoc(), TII->get(Opcode)); |
| Jonas Paulsson | e80d405 | 2018-06-07 05:59:07 +0000 | [diff] [blame] | 300 | for (const auto &MO : MI.operands()) |
| Chandler Carruth | c73c030 | 2018-08-16 21:30:05 +0000 | [diff] [blame] | 301 | MIB.add(MO); |
| 302 | MIB.setMemRefs(MI.memoperands()); |
| Jonas Paulsson | e80d405 | 2018-06-07 05:59:07 +0000 | [diff] [blame] | 303 | MI.eraseFromParent(); |
| 304 | |
| Richard Sandiford | b49a3ab | 2013-08-05 11:03:20 +0000 | [diff] [blame] | 305 | return true; |
| 306 | } |
| 307 | |
| Richard Sandiford | bdbb8af | 2013-08-05 10:58:53 +0000 | [diff] [blame] | 308 | // The CC users in CCUsers are testing the result of a comparison of some |
| Jonas Paulsson | 776a81a | 2018-01-15 15:41:26 +0000 | [diff] [blame] | 309 | // value X against zero and we know that any CC value produced by MI would |
| 310 | // also reflect the value of X. ConvOpc may be used to pass the transfomed |
| 311 | // opcode MI will have if this succeeds. Try to adjust CCUsers so that they |
| 312 | // test the result of MI directly, returning true on success. Leave |
| 313 | // everything unchanged on failure. |
| Duncan P. N. Exon Smith | 4565ec0 | 2016-07-12 01:39:01 +0000 | [diff] [blame] | 314 | bool SystemZElimCompare::adjustCCMasksForInstr( |
| 315 | MachineInstr &MI, MachineInstr &Compare, |
| Jonas Paulsson | 776a81a | 2018-01-15 15:41:26 +0000 | [diff] [blame] | 316 | SmallVectorImpl<MachineInstr *> &CCUsers, |
| 317 | unsigned ConvOpc) { |
| 318 | int Opcode = (ConvOpc ? ConvOpc : MI.getOpcode()); |
| Richard Sandiford | bdbb8af | 2013-08-05 10:58:53 +0000 | [diff] [blame] | 319 | const MCInstrDesc &Desc = TII->get(Opcode); |
| 320 | unsigned MIFlags = Desc.TSFlags; |
| 321 | |
| 322 | // See which compare-style condition codes are available. |
| Richard Sandiford | 0897fce | 2013-08-07 11:10:06 +0000 | [diff] [blame] | 323 | unsigned ReusableCCMask = SystemZII::getCompareZeroCCMask(MIFlags); |
| Richard Sandiford | bdbb8af | 2013-08-05 10:58:53 +0000 | [diff] [blame] | 324 | |
| 325 | // For unsigned comparisons with zero, only equality makes sense. |
| Duncan P. N. Exon Smith | 4565ec0 | 2016-07-12 01:39:01 +0000 | [diff] [blame] | 326 | unsigned CompareFlags = Compare.getDesc().TSFlags; |
| Richard Sandiford | 0897fce | 2013-08-07 11:10:06 +0000 | [diff] [blame] | 327 | if (CompareFlags & SystemZII::IsLogical) |
| 328 | ReusableCCMask &= SystemZ::CCMASK_CMP_EQ; |
| Richard Sandiford | bdbb8af | 2013-08-05 10:58:53 +0000 | [diff] [blame] | 329 | |
| 330 | if (ReusableCCMask == 0) |
| 331 | return false; |
| 332 | |
| 333 | unsigned CCValues = SystemZII::getCCValues(MIFlags); |
| 334 | assert((ReusableCCMask & ~CCValues) == 0 && "Invalid CCValues"); |
| 335 | |
| Jonas Paulsson | 776a81a | 2018-01-15 15:41:26 +0000 | [diff] [blame] | 336 | bool MIEquivalentToCmp = |
| 337 | (ReusableCCMask == CCValues && |
| 338 | CCValues == SystemZII::getCCValues(CompareFlags)); |
| Richard Sandiford | bdbb8af | 2013-08-05 10:58:53 +0000 | [diff] [blame] | 339 | |
| Jonas Paulsson | 776a81a | 2018-01-15 15:41:26 +0000 | [diff] [blame] | 340 | if (!MIEquivalentToCmp) { |
| 341 | // Now check whether these flags are enough for all users. |
| 342 | SmallVector<MachineOperand *, 4> AlterMasks; |
| 343 | for (unsigned int I = 0, E = CCUsers.size(); I != E; ++I) { |
| 344 | MachineInstr *MI = CCUsers[I]; |
| Richard Sandiford | bdbb8af | 2013-08-05 10:58:53 +0000 | [diff] [blame] | 345 | |
| Jonas Paulsson | 776a81a | 2018-01-15 15:41:26 +0000 | [diff] [blame] | 346 | // Fail if this isn't a use of CC that we understand. |
| 347 | unsigned Flags = MI->getDesc().TSFlags; |
| 348 | unsigned FirstOpNum; |
| 349 | if (Flags & SystemZII::CCMaskFirst) |
| 350 | FirstOpNum = 0; |
| 351 | else if (Flags & SystemZII::CCMaskLast) |
| 352 | FirstOpNum = MI->getNumExplicitOperands() - 2; |
| 353 | else |
| 354 | return false; |
| Richard Sandiford | bdbb8af | 2013-08-05 10:58:53 +0000 | [diff] [blame] | 355 | |
| Jonas Paulsson | 776a81a | 2018-01-15 15:41:26 +0000 | [diff] [blame] | 356 | // Check whether the instruction predicate treats all CC values |
| 357 | // outside of ReusableCCMask in the same way. In that case it |
| 358 | // doesn't matter what those CC values mean. |
| 359 | unsigned CCValid = MI->getOperand(FirstOpNum).getImm(); |
| 360 | unsigned CCMask = MI->getOperand(FirstOpNum + 1).getImm(); |
| 361 | unsigned OutValid = ~ReusableCCMask & CCValid; |
| 362 | unsigned OutMask = ~ReusableCCMask & CCMask; |
| 363 | if (OutMask != 0 && OutMask != OutValid) |
| 364 | return false; |
| Richard Sandiford | bdbb8af | 2013-08-05 10:58:53 +0000 | [diff] [blame] | 365 | |
| Jonas Paulsson | 776a81a | 2018-01-15 15:41:26 +0000 | [diff] [blame] | 366 | AlterMasks.push_back(&MI->getOperand(FirstOpNum)); |
| 367 | AlterMasks.push_back(&MI->getOperand(FirstOpNum + 1)); |
| 368 | } |
| 369 | |
| 370 | // All users are OK. Adjust the masks for MI. |
| 371 | for (unsigned I = 0, E = AlterMasks.size(); I != E; I += 2) { |
| 372 | AlterMasks[I]->setImm(CCValues); |
| 373 | unsigned CCMask = AlterMasks[I + 1]->getImm(); |
| 374 | if (CCMask & ~ReusableCCMask) |
| 375 | AlterMasks[I + 1]->setImm((CCMask & ReusableCCMask) | |
| 376 | (CCValues & ~ReusableCCMask)); |
| 377 | } |
| Richard Sandiford | bdbb8af | 2013-08-05 10:58:53 +0000 | [diff] [blame] | 378 | } |
| 379 | |
| 380 | // CC is now live after MI. |
| Jonas Paulsson | 776a81a | 2018-01-15 15:41:26 +0000 | [diff] [blame] | 381 | if (!ConvOpc) { |
| 382 | int CCDef = MI.findRegisterDefOperandIdx(SystemZ::CC, false, true, TRI); |
| 383 | assert(CCDef >= 0 && "Couldn't find CC set"); |
| 384 | MI.getOperand(CCDef).setIsDead(false); |
| 385 | } |
| 386 | |
| 387 | // Check if MI lies before Compare. |
| 388 | bool BeforeCmp = false; |
| 389 | MachineBasicBlock::iterator MBBI = MI, MBBE = MI.getParent()->end(); |
| 390 | for (++MBBI; MBBI != MBBE; ++MBBI) |
| 391 | if (MBBI == Compare) { |
| 392 | BeforeCmp = true; |
| 393 | break; |
| 394 | } |
| Richard Sandiford | bdbb8af | 2013-08-05 10:58:53 +0000 | [diff] [blame] | 395 | |
| 396 | // Clear any intervening kills of CC. |
| Jonas Paulsson | 776a81a | 2018-01-15 15:41:26 +0000 | [diff] [blame] | 397 | if (BeforeCmp) { |
| 398 | MachineBasicBlock::iterator MBBI = MI, MBBE = Compare; |
| 399 | for (++MBBI; MBBI != MBBE; ++MBBI) |
| 400 | MBBI->clearRegisterKills(SystemZ::CC, TRI); |
| 401 | } |
| Richard Sandiford | bdbb8af | 2013-08-05 10:58:53 +0000 | [diff] [blame] | 402 | |
| 403 | return true; |
| 404 | } |
| 405 | |
| Richard Sandiford | 0897fce | 2013-08-07 11:10:06 +0000 | [diff] [blame] | 406 | // Return true if Compare is a comparison against zero. |
| Duncan P. N. Exon Smith | 4565ec0 | 2016-07-12 01:39:01 +0000 | [diff] [blame] | 407 | static bool isCompareZero(MachineInstr &Compare) { |
| 408 | switch (Compare.getOpcode()) { |
| Richard Sandiford | 0897fce | 2013-08-07 11:10:06 +0000 | [diff] [blame] | 409 | case SystemZ::LTEBRCompare: |
| 410 | case SystemZ::LTDBRCompare: |
| 411 | case SystemZ::LTXBRCompare: |
| 412 | return true; |
| 413 | |
| 414 | default: |
| Jonas Paulsson | 5d3fbd3 | 2015-10-08 07:40:23 +0000 | [diff] [blame] | 415 | if (isLoadAndTestAsCmp(Compare)) |
| 416 | return true; |
| Duncan P. N. Exon Smith | 4565ec0 | 2016-07-12 01:39:01 +0000 | [diff] [blame] | 417 | return Compare.getNumExplicitOperands() == 2 && |
| 418 | Compare.getOperand(1).isImm() && Compare.getOperand(1).getImm() == 0; |
| Richard Sandiford | 0897fce | 2013-08-07 11:10:06 +0000 | [diff] [blame] | 419 | } |
| 420 | } |
| 421 | |
| Richard Sandiford | bdbb8af | 2013-08-05 10:58:53 +0000 | [diff] [blame] | 422 | // Try to optimize cases where comparison instruction Compare is testing |
| 423 | // a value against zero. Return true on success and if Compare should be |
| 424 | // deleted as dead. CCUsers is the list of instructions that use the CC |
| 425 | // value produced by Compare. |
| Duncan P. N. Exon Smith | 4565ec0 | 2016-07-12 01:39:01 +0000 | [diff] [blame] | 426 | bool SystemZElimCompare::optimizeCompareZero( |
| 427 | MachineInstr &Compare, SmallVectorImpl<MachineInstr *> &CCUsers) { |
| Richard Sandiford | 0897fce | 2013-08-07 11:10:06 +0000 | [diff] [blame] | 428 | if (!isCompareZero(Compare)) |
| Richard Sandiford | bdbb8af | 2013-08-05 10:58:53 +0000 | [diff] [blame] | 429 | return false; |
| 430 | |
| 431 | // Search back for CC results that are based on the first operand. |
| Jonas Paulsson | 5d3fbd3 | 2015-10-08 07:40:23 +0000 | [diff] [blame] | 432 | unsigned SrcReg = getCompareSourceReg(Compare); |
| Duncan P. N. Exon Smith | 4565ec0 | 2016-07-12 01:39:01 +0000 | [diff] [blame] | 433 | MachineBasicBlock &MBB = *Compare.getParent(); |
| Richard Sandiford | c212125 | 2013-08-05 11:23:46 +0000 | [diff] [blame] | 434 | Reference CCRefs; |
| 435 | Reference SrcRefs; |
| Jonas Paulsson | e80d405 | 2018-06-07 05:59:07 +0000 | [diff] [blame] | 436 | for (MachineBasicBlock::reverse_iterator MBBI = |
| 437 | std::next(MachineBasicBlock::reverse_iterator(&Compare)), |
| 438 | MBBE = MBB.rend(); MBBI != MBBE;) { |
| 439 | MachineInstr &MI = *MBBI++; |
| Jonas Paulsson | 2c96dd6 | 2015-10-08 07:40:11 +0000 | [diff] [blame] | 440 | if (resultTests(MI, SrcReg)) { |
| Richard Sandiford | c212125 | 2013-08-05 11:23:46 +0000 | [diff] [blame] | 441 | // Try to remove both MI and Compare by converting a branch to BRCT(G). |
| Ulrich Weigand | 2d9e3d9 | 2016-11-28 13:59:22 +0000 | [diff] [blame] | 442 | // or a load-and-trap instruction. We don't care in this case whether |
| 443 | // CC is modified between MI and Compare. |
| 444 | if (!CCRefs.Use && !SrcRefs) { |
| 445 | if (convertToBRCT(MI, Compare, CCUsers)) { |
| 446 | BranchOnCounts += 1; |
| 447 | return true; |
| 448 | } |
| 449 | if (convertToLoadAndTrap(MI, Compare, CCUsers)) { |
| 450 | LoadAndTraps += 1; |
| 451 | return true; |
| 452 | } |
| Richard Sandiford | c212125 | 2013-08-05 11:23:46 +0000 | [diff] [blame] | 453 | } |
| 454 | // Try to eliminate Compare by reusing a CC result from MI. |
| Jonas Paulsson | 776a81a | 2018-01-15 15:41:26 +0000 | [diff] [blame] | 455 | if ((!CCRefs && convertToLoadAndTest(MI, Compare, CCUsers)) || |
| Richard Sandiford | c212125 | 2013-08-05 11:23:46 +0000 | [diff] [blame] | 456 | (!CCRefs.Def && adjustCCMasksForInstr(MI, Compare, CCUsers))) { |
| 457 | EliminatedComparisons += 1; |
| 458 | return true; |
| 459 | } |
| Richard Sandiford | bdbb8af | 2013-08-05 10:58:53 +0000 | [diff] [blame] | 460 | } |
| Richard Sandiford | c212125 | 2013-08-05 11:23:46 +0000 | [diff] [blame] | 461 | SrcRefs |= getRegReferences(MI, SrcReg); |
| 462 | if (SrcRefs.Def) |
| Jonas Paulsson | b0e8a2e | 2017-09-21 13:52:24 +0000 | [diff] [blame] | 463 | break; |
| Richard Sandiford | c212125 | 2013-08-05 11:23:46 +0000 | [diff] [blame] | 464 | CCRefs |= getRegReferences(MI, SystemZ::CC); |
| 465 | if (CCRefs.Use && CCRefs.Def) |
| Jonas Paulsson | b0e8a2e | 2017-09-21 13:52:24 +0000 | [diff] [blame] | 466 | break; |
| 467 | } |
| 468 | |
| 469 | // Also do a forward search to handle cases where an instruction after the |
| Jonas Paulsson | 22f208f | 2018-01-08 12:52:40 +0000 | [diff] [blame] | 470 | // compare can be converted, like |
| 471 | // LTEBRCompare %f0s, %f0s; %f2s = LER %f0s => LTEBRCompare %f2s, %f0s |
| Jonas Paulsson | e80d405 | 2018-06-07 05:59:07 +0000 | [diff] [blame] | 472 | for (MachineBasicBlock::iterator MBBI = |
| 473 | std::next(MachineBasicBlock::iterator(&Compare)), MBBE = MBB.end(); |
| 474 | MBBI != MBBE;) { |
| 475 | MachineInstr &MI = *MBBI++; |
| Jonas Paulsson | b0e8a2e | 2017-09-21 13:52:24 +0000 | [diff] [blame] | 476 | if (preservesValueOf(MI, SrcReg)) { |
| 477 | // Try to eliminate Compare by reusing a CC result from MI. |
| Jonas Paulsson | 776a81a | 2018-01-15 15:41:26 +0000 | [diff] [blame] | 478 | if (convertToLoadAndTest(MI, Compare, CCUsers)) { |
| Jonas Paulsson | b0e8a2e | 2017-09-21 13:52:24 +0000 | [diff] [blame] | 479 | EliminatedComparisons += 1; |
| 480 | return true; |
| 481 | } |
| 482 | } |
| 483 | if (getRegReferences(MI, SrcReg).Def) |
| 484 | return false; |
| 485 | if (getRegReferences(MI, SystemZ::CC)) |
| Richard Sandiford | c212125 | 2013-08-05 11:23:46 +0000 | [diff] [blame] | 486 | return false; |
| Richard Sandiford | bdbb8af | 2013-08-05 10:58:53 +0000 | [diff] [blame] | 487 | } |
| Jonas Paulsson | b0e8a2e | 2017-09-21 13:52:24 +0000 | [diff] [blame] | 488 | |
| Richard Sandiford | bdbb8af | 2013-08-05 10:58:53 +0000 | [diff] [blame] | 489 | return false; |
| 490 | } |
| 491 | |
| 492 | // Try to fuse comparison instruction Compare into a later branch. |
| 493 | // Return true on success and if Compare is therefore redundant. |
| Duncan P. N. Exon Smith | 4565ec0 | 2016-07-12 01:39:01 +0000 | [diff] [blame] | 494 | bool SystemZElimCompare::fuseCompareOperations( |
| 495 | MachineInstr &Compare, SmallVectorImpl<MachineInstr *> &CCUsers) { |
| Richard Sandiford | bdbb8af | 2013-08-05 10:58:53 +0000 | [diff] [blame] | 496 | // See whether we have a single branch with which to fuse. |
| 497 | if (CCUsers.size() != 1) |
| 498 | return false; |
| 499 | MachineInstr *Branch = CCUsers[0]; |
| Zhan Jun Liau | ab42cbc | 2016-06-10 19:58:10 +0000 | [diff] [blame] | 500 | SystemZII::FusedCompareType Type; |
| Ulrich Weigand | 2eb027d | 2016-04-07 16:11:44 +0000 | [diff] [blame] | 501 | switch (Branch->getOpcode()) { |
| 502 | case SystemZ::BRC: |
| 503 | Type = SystemZII::CompareAndBranch; |
| 504 | break; |
| 505 | case SystemZ::CondReturn: |
| 506 | Type = SystemZII::CompareAndReturn; |
| 507 | break; |
| Ulrich Weigand | 848a513 | 2016-04-11 12:12:32 +0000 | [diff] [blame] | 508 | case SystemZ::CallBCR: |
| 509 | Type = SystemZII::CompareAndSibcall; |
| 510 | break; |
| Zhan Jun Liau | ab42cbc | 2016-06-10 19:58:10 +0000 | [diff] [blame] | 511 | case SystemZ::CondTrap: |
| 512 | Type = SystemZII::CompareAndTrap; |
| 513 | break; |
| Ulrich Weigand | 2eb027d | 2016-04-07 16:11:44 +0000 | [diff] [blame] | 514 | default: |
| 515 | return false; |
| 516 | } |
| 517 | |
| 518 | // See whether we have a comparison that can be fused. |
| Duncan P. N. Exon Smith | 4565ec0 | 2016-07-12 01:39:01 +0000 | [diff] [blame] | 519 | unsigned FusedOpcode = |
| 520 | TII->getFusedCompare(Compare.getOpcode(), Type, &Compare); |
| Ulrich Weigand | 2eb027d | 2016-04-07 16:11:44 +0000 | [diff] [blame] | 521 | if (!FusedOpcode) |
| Richard Sandiford | bdbb8af | 2013-08-05 10:58:53 +0000 | [diff] [blame] | 522 | return false; |
| 523 | |
| 524 | // Make sure that the operands are available at the branch. |
| Ulrich Weigand | a0e7325 | 2016-11-11 12:48:26 +0000 | [diff] [blame] | 525 | // SrcReg2 is the register if the source operand is a register, |
| 526 | // 0 if the source operand is immediate, and the base register |
| 527 | // if the source operand is memory (index is not supported). |
| Matt Arsenault | e3a676e | 2019-06-24 15:50:29 +0000 | [diff] [blame] | 528 | Register SrcReg = Compare.getOperand(0).getReg(); |
| 529 | Register SrcReg2 = |
| 530 | Compare.getOperand(1).isReg() ? Compare.getOperand(1).getReg() : Register(); |
| Richard Sandiford | bdbb8af | 2013-08-05 10:58:53 +0000 | [diff] [blame] | 531 | MachineBasicBlock::iterator MBBI = Compare, MBBE = Branch; |
| 532 | for (++MBBI; MBBI != MBBE; ++MBBI) |
| 533 | if (MBBI->modifiesRegister(SrcReg, TRI) || |
| 534 | (SrcReg2 && MBBI->modifiesRegister(SrcReg2, TRI))) |
| 535 | return false; |
| 536 | |
| Ulrich Weigand | 848a513 | 2016-04-11 12:12:32 +0000 | [diff] [blame] | 537 | // Read the branch mask, target (if applicable), regmask (if applicable). |
| Richard Sandiford | bdbb8af | 2013-08-05 10:58:53 +0000 | [diff] [blame] | 538 | MachineOperand CCMask(MBBI->getOperand(1)); |
| Richard Sandiford | bdbb8af | 2013-08-05 10:58:53 +0000 | [diff] [blame] | 539 | assert((CCMask.getImm() & ~SystemZ::CCMASK_ICMP) == 0 && |
| 540 | "Invalid condition-code mask for integer comparison"); |
| Ulrich Weigand | 2eb027d | 2016-04-07 16:11:44 +0000 | [diff] [blame] | 541 | // This is only valid for CompareAndBranch. |
| 542 | MachineOperand Target(MBBI->getOperand( |
| 543 | Type == SystemZII::CompareAndBranch ? 2 : 0)); |
| Ulrich Weigand | 848a513 | 2016-04-11 12:12:32 +0000 | [diff] [blame] | 544 | const uint32_t *RegMask; |
| 545 | if (Type == SystemZII::CompareAndSibcall) |
| 546 | RegMask = MBBI->getOperand(2).getRegMask(); |
| Richard Sandiford | bdbb8af | 2013-08-05 10:58:53 +0000 | [diff] [blame] | 547 | |
| 548 | // Clear out all current operands. |
| 549 | int CCUse = MBBI->findRegisterUseOperandIdx(SystemZ::CC, false, TRI); |
| Ulrich Weigand | 2eb027d | 2016-04-07 16:11:44 +0000 | [diff] [blame] | 550 | assert(CCUse >= 0 && "BRC/BCR must use CC"); |
| Richard Sandiford | bdbb8af | 2013-08-05 10:58:53 +0000 | [diff] [blame] | 551 | Branch->RemoveOperand(CCUse); |
| Ulrich Weigand | 848a513 | 2016-04-11 12:12:32 +0000 | [diff] [blame] | 552 | // Remove target (branch) or regmask (sibcall). |
| 553 | if (Type == SystemZII::CompareAndBranch || |
| 554 | Type == SystemZII::CompareAndSibcall) |
| Ulrich Weigand | 2eb027d | 2016-04-07 16:11:44 +0000 | [diff] [blame] | 555 | Branch->RemoveOperand(2); |
| Richard Sandiford | bdbb8af | 2013-08-05 10:58:53 +0000 | [diff] [blame] | 556 | Branch->RemoveOperand(1); |
| 557 | Branch->RemoveOperand(0); |
| 558 | |
| 559 | // Rebuild Branch as a fused compare and branch. |
| Ulrich Weigand | a0e7325 | 2016-11-11 12:48:26 +0000 | [diff] [blame] | 560 | // SrcNOps is the number of MI operands of the compare instruction |
| 561 | // that we need to copy over. |
| 562 | unsigned SrcNOps = 2; |
| 563 | if (FusedOpcode == SystemZ::CLT || FusedOpcode == SystemZ::CLGT) |
| 564 | SrcNOps = 3; |
| Richard Sandiford | bdbb8af | 2013-08-05 10:58:53 +0000 | [diff] [blame] | 565 | Branch->setDesc(TII->get(FusedOpcode)); |
| Ulrich Weigand | 2eb027d | 2016-04-07 16:11:44 +0000 | [diff] [blame] | 566 | MachineInstrBuilder MIB(*Branch->getParent()->getParent(), Branch); |
| Ulrich Weigand | a0e7325 | 2016-11-11 12:48:26 +0000 | [diff] [blame] | 567 | for (unsigned I = 0; I < SrcNOps; I++) |
| Diana Picus | 116bbab | 2017-01-13 09:58:52 +0000 | [diff] [blame] | 568 | MIB.add(Compare.getOperand(I)); |
| 569 | MIB.add(CCMask); |
| Ulrich Weigand | 2eb027d | 2016-04-07 16:11:44 +0000 | [diff] [blame] | 570 | |
| 571 | if (Type == SystemZII::CompareAndBranch) { |
| 572 | // Only conditional branches define CC, as they may be converted back |
| 573 | // to a non-fused branch because of a long displacement. Conditional |
| 574 | // returns don't have that problem. |
| Diana Picus | 116bbab | 2017-01-13 09:58:52 +0000 | [diff] [blame] | 575 | MIB.add(Target).addReg(SystemZ::CC, |
| 576 | RegState::ImplicitDefine | RegState::Dead); |
| Ulrich Weigand | 2eb027d | 2016-04-07 16:11:44 +0000 | [diff] [blame] | 577 | } |
| Richard Sandiford | bdbb8af | 2013-08-05 10:58:53 +0000 | [diff] [blame] | 578 | |
| Ulrich Weigand | 848a513 | 2016-04-11 12:12:32 +0000 | [diff] [blame] | 579 | if (Type == SystemZII::CompareAndSibcall) |
| 580 | MIB.addRegMask(RegMask); |
| 581 | |
| Richard Sandiford | bdbb8af | 2013-08-05 10:58:53 +0000 | [diff] [blame] | 582 | // Clear any intervening kills of SrcReg and SrcReg2. |
| 583 | MBBI = Compare; |
| 584 | for (++MBBI; MBBI != MBBE; ++MBBI) { |
| 585 | MBBI->clearRegisterKills(SrcReg, TRI); |
| 586 | if (SrcReg2) |
| 587 | MBBI->clearRegisterKills(SrcReg2, TRI); |
| 588 | } |
| 589 | FusedComparisons += 1; |
| 590 | return true; |
| 591 | } |
| 592 | |
| 593 | // Process all comparison instructions in MBB. Return true if something |
| 594 | // changed. |
| Richard Sandiford | 28c111e | 2014-03-06 11:00:15 +0000 | [diff] [blame] | 595 | bool SystemZElimCompare::processBlock(MachineBasicBlock &MBB) { |
| Richard Sandiford | bdbb8af | 2013-08-05 10:58:53 +0000 | [diff] [blame] | 596 | bool Changed = false; |
| 597 | |
| 598 | // Walk backwards through the block looking for comparisons, recording |
| 599 | // all CC users as we go. The subroutines can delete Compare and |
| 600 | // instructions before it. |
| 601 | bool CompleteCCUsers = !isCCLiveOut(MBB); |
| 602 | SmallVector<MachineInstr *, 4> CCUsers; |
| Richard Sandiford | 28c111e | 2014-03-06 11:00:15 +0000 | [diff] [blame] | 603 | MachineBasicBlock::iterator MBBI = MBB.end(); |
| 604 | while (MBBI != MBB.begin()) { |
| Duncan P. N. Exon Smith | 4565ec0 | 2016-07-12 01:39:01 +0000 | [diff] [blame] | 605 | MachineInstr &MI = *--MBBI; |
| 606 | if (CompleteCCUsers && (MI.isCompare() || isLoadAndTestAsCmp(MI)) && |
| Richard Sandiford | bdbb8af | 2013-08-05 10:58:53 +0000 | [diff] [blame] | 607 | (optimizeCompareZero(MI, CCUsers) || |
| Zhan Jun Liau | ab42cbc | 2016-06-10 19:58:10 +0000 | [diff] [blame] | 608 | fuseCompareOperations(MI, CCUsers))) { |
| Richard Sandiford | bdbb8af | 2013-08-05 10:58:53 +0000 | [diff] [blame] | 609 | ++MBBI; |
| Duncan P. N. Exon Smith | 4565ec0 | 2016-07-12 01:39:01 +0000 | [diff] [blame] | 610 | MI.eraseFromParent(); |
| Richard Sandiford | bdbb8af | 2013-08-05 10:58:53 +0000 | [diff] [blame] | 611 | Changed = true; |
| 612 | CCUsers.clear(); |
| Richard Sandiford | bdbb8af | 2013-08-05 10:58:53 +0000 | [diff] [blame] | 613 | continue; |
| 614 | } |
| 615 | |
| Duncan P. N. Exon Smith | 4565ec0 | 2016-07-12 01:39:01 +0000 | [diff] [blame] | 616 | if (MI.definesRegister(SystemZ::CC)) { |
| Richard Sandiford | bdbb8af | 2013-08-05 10:58:53 +0000 | [diff] [blame] | 617 | CCUsers.clear(); |
| Jonas Paulsson | 9e1f3bd | 2015-10-08 07:39:55 +0000 | [diff] [blame] | 618 | CompleteCCUsers = true; |
| Richard Sandiford | c212125 | 2013-08-05 11:23:46 +0000 | [diff] [blame] | 619 | } |
| Duncan P. N. Exon Smith | 4565ec0 | 2016-07-12 01:39:01 +0000 | [diff] [blame] | 620 | if (MI.readsRegister(SystemZ::CC) && CompleteCCUsers) |
| 621 | CCUsers.push_back(&MI); |
| Richard Sandiford | bdbb8af | 2013-08-05 10:58:53 +0000 | [diff] [blame] | 622 | } |
| 623 | return Changed; |
| 624 | } |
| 625 | |
| 626 | bool SystemZElimCompare::runOnMachineFunction(MachineFunction &F) { |
| Matthias Braun | f1caa28 | 2017-12-15 22:22:58 +0000 | [diff] [blame] | 627 | if (skipFunction(F.getFunction())) |
| Andrew Kaylor | d9974cc | 2016-04-26 23:49:41 +0000 | [diff] [blame] | 628 | return false; |
| 629 | |
| Eric Christopher | fc6de42 | 2014-08-05 02:39:49 +0000 | [diff] [blame] | 630 | TII = static_cast<const SystemZInstrInfo *>(F.getSubtarget().getInstrInfo()); |
| Richard Sandiford | bdbb8af | 2013-08-05 10:58:53 +0000 | [diff] [blame] | 631 | TRI = &TII->getRegisterInfo(); |
| 632 | |
| 633 | bool Changed = false; |
| Richard Sandiford | 28c111e | 2014-03-06 11:00:15 +0000 | [diff] [blame] | 634 | for (auto &MBB : F) |
| 635 | Changed |= processBlock(MBB); |
| Richard Sandiford | bdbb8af | 2013-08-05 10:58:53 +0000 | [diff] [blame] | 636 | |
| 637 | return Changed; |
| 638 | } |
| Eugene Zelenko | 3943d2b | 2017-01-24 22:10:43 +0000 | [diff] [blame] | 639 | |
| 640 | FunctionPass *llvm::createSystemZElimComparePass(SystemZTargetMachine &TM) { |
| 641 | return new SystemZElimCompare(TM); |
| 642 | } |