Chris Lattner | 5930d3d | 2005-11-16 22:59:19 +0000 | [diff] [blame] | 1 | //===- X86ISelDAGToDAG.cpp - A DAG pattern matching inst selector for X86 -===// |
Chris Lattner | 655e7df | 2005-11-16 01:54:32 +0000 | [diff] [blame] | 2 | // |
| 3 | // The LLVM Compiler Infrastructure |
| 4 | // |
Chris Lattner | f3ebc3f | 2007-12-29 20:36:04 +0000 | [diff] [blame] | 5 | // This file is distributed under the University of Illinois Open Source |
| 6 | // License. See LICENSE.TXT for details. |
Chris Lattner | 655e7df | 2005-11-16 01:54:32 +0000 | [diff] [blame] | 7 | // |
| 8 | //===----------------------------------------------------------------------===// |
| 9 | // |
| 10 | // This file defines a DAG pattern matching instruction selector for X86, |
| 11 | // converting from a legalized dag to a X86 dag. |
| 12 | // |
| 13 | //===----------------------------------------------------------------------===// |
| 14 | |
| 15 | #include "X86.h" |
Evan Cheng | bc7a0f44 | 2006-01-11 06:09:51 +0000 | [diff] [blame] | 16 | #include "X86InstrBuilder.h" |
Evan Cheng | f55b738 | 2008-01-05 00:41:47 +0000 | [diff] [blame] | 17 | #include "X86MachineFunctionInfo.h" |
Chris Lattner | 7c55126 | 2006-01-11 01:15:34 +0000 | [diff] [blame] | 18 | #include "X86RegisterInfo.h" |
Chris Lattner | 655e7df | 2005-11-16 01:54:32 +0000 | [diff] [blame] | 19 | #include "X86Subtarget.h" |
Evan Cheng | 2dd2c65 | 2006-03-13 23:20:37 +0000 | [diff] [blame] | 20 | #include "X86TargetMachine.h" |
Chandler Carruth | ed0881b | 2012-12-03 16:50:05 +0000 | [diff] [blame] | 21 | #include "llvm/ADT/Statistic.h" |
Evan Cheng | 73a1ad9 | 2006-01-10 20:26:56 +0000 | [diff] [blame] | 22 | #include "llvm/CodeGen/MachineFrameInfo.h" |
Chandler Carruth | ed0881b | 2012-12-03 16:50:05 +0000 | [diff] [blame] | 23 | #include "llvm/CodeGen/MachineFunction.h" |
Chris Lattner | 7c55126 | 2006-01-11 01:15:34 +0000 | [diff] [blame] | 24 | #include "llvm/CodeGen/MachineInstrBuilder.h" |
Chris Lattner | a10fff5 | 2007-12-31 04:13:23 +0000 | [diff] [blame] | 25 | #include "llvm/CodeGen/MachineRegisterInfo.h" |
Chris Lattner | 655e7df | 2005-11-16 01:54:32 +0000 | [diff] [blame] | 26 | #include "llvm/CodeGen/SelectionDAGISel.h" |
Eric Christopher | 79cc1e3 | 2014-09-02 22:28:02 +0000 | [diff] [blame] | 27 | #include "llvm/IR/Function.h" |
Chandler Carruth | 9fb823b | 2013-01-02 11:36:10 +0000 | [diff] [blame] | 28 | #include "llvm/IR/Instructions.h" |
| 29 | #include "llvm/IR/Intrinsics.h" |
| 30 | #include "llvm/IR/Type.h" |
Evan Cheng | 11b0a5d | 2006-09-08 06:48:29 +0000 | [diff] [blame] | 31 | #include "llvm/Support/Debug.h" |
Torok Edwin | fb8d6d5 | 2009-07-08 20:53:28 +0000 | [diff] [blame] | 32 | #include "llvm/Support/ErrorHandling.h" |
Evan Cheng | 11b0a5d | 2006-09-08 06:48:29 +0000 | [diff] [blame] | 33 | #include "llvm/Support/MathExtras.h" |
Torok Edwin | fb8d6d5 | 2009-07-08 20:53:28 +0000 | [diff] [blame] | 34 | #include "llvm/Support/raw_ostream.h" |
Chandler Carruth | ed0881b | 2012-12-03 16:50:05 +0000 | [diff] [blame] | 35 | #include "llvm/Target/TargetMachine.h" |
| 36 | #include "llvm/Target/TargetOptions.h" |
Robin Morisset | 880580b | 2014-10-07 23:53:57 +0000 | [diff] [blame] | 37 | #include <stdint.h> |
Chris Lattner | 655e7df | 2005-11-16 01:54:32 +0000 | [diff] [blame] | 38 | using namespace llvm; |
| 39 | |
Chandler Carruth | 84e68b2 | 2014-04-22 02:41:26 +0000 | [diff] [blame] | 40 | #define DEBUG_TYPE "x86-isel" |
| 41 | |
Chris Lattner | 1ef9cd4 | 2006-12-19 22:59:26 +0000 | [diff] [blame] | 42 | STATISTIC(NumLoadMoved, "Number of loads moved below TokenFactor"); |
| 43 | |
Chris Lattner | 655e7df | 2005-11-16 01:54:32 +0000 | [diff] [blame] | 44 | //===----------------------------------------------------------------------===// |
| 45 | // Pattern Matcher Implementation |
| 46 | //===----------------------------------------------------------------------===// |
| 47 | |
| 48 | namespace { |
Chris Lattner | 3f0f71b | 2005-11-19 02:11:08 +0000 | [diff] [blame] | 49 | /// X86ISelAddressMode - This corresponds to X86AddressMode, but uses |
Dan Gohman | 2ce6f2a | 2008-07-27 21:46:04 +0000 | [diff] [blame] | 50 | /// SDValue's instead of register numbers for the leaves of the matched |
Chris Lattner | 3f0f71b | 2005-11-19 02:11:08 +0000 | [diff] [blame] | 51 | /// tree. |
| 52 | struct X86ISelAddressMode { |
| 53 | enum { |
| 54 | RegBase, |
Chris Lattner | aa237256 | 2006-05-24 17:04:05 +0000 | [diff] [blame] | 55 | FrameIndexBase |
Chris Lattner | 3f0f71b | 2005-11-19 02:11:08 +0000 | [diff] [blame] | 56 | } BaseType; |
| 57 | |
Dan Gohman | 0fd54fb | 2010-04-29 23:30:41 +0000 | [diff] [blame] | 58 | // This is really a union, discriminated by BaseType! |
| 59 | SDValue Base_Reg; |
| 60 | int Base_FrameIndex; |
Chris Lattner | 3f0f71b | 2005-11-19 02:11:08 +0000 | [diff] [blame] | 61 | |
| 62 | unsigned Scale; |
Chad Rosier | 24c19d2 | 2012-08-01 18:39:17 +0000 | [diff] [blame] | 63 | SDValue IndexReg; |
Dan Gohman | 059c4fa | 2008-11-11 15:52:29 +0000 | [diff] [blame] | 64 | int32_t Disp; |
Rafael Espindola | 3b2df10 | 2009-04-08 21:14:34 +0000 | [diff] [blame] | 65 | SDValue Segment; |
Dan Gohman | bcaf681 | 2010-04-15 01:51:59 +0000 | [diff] [blame] | 66 | const GlobalValue *GV; |
| 67 | const Constant *CP; |
| 68 | const BlockAddress *BlockAddr; |
Evan Cheng | 11b0a5d | 2006-09-08 06:48:29 +0000 | [diff] [blame] | 69 | const char *ES; |
| 70 | int JT; |
Evan Cheng | 77d86ff | 2006-02-25 10:09:08 +0000 | [diff] [blame] | 71 | unsigned Align; // CP alignment. |
Chris Lattner | bd7e26d | 2009-06-26 05:51:45 +0000 | [diff] [blame] | 72 | unsigned char SymbolFlags; // X86II::MO_* |
Chris Lattner | 3f0f71b | 2005-11-19 02:11:08 +0000 | [diff] [blame] | 73 | |
| 74 | X86ISelAddressMode() |
Dan Gohman | 0fd54fb | 2010-04-29 23:30:41 +0000 | [diff] [blame] | 75 | : BaseType(RegBase), Base_FrameIndex(0), Scale(1), IndexReg(), Disp(0), |
Craig Topper | 062a2ba | 2014-04-25 05:30:21 +0000 | [diff] [blame] | 76 | Segment(), GV(nullptr), CP(nullptr), BlockAddr(nullptr), ES(nullptr), |
| 77 | JT(-1), Align(0), SymbolFlags(X86II::MO_NO_FLAG) { |
Chris Lattner | 3f0f71b | 2005-11-19 02:11:08 +0000 | [diff] [blame] | 78 | } |
Dan Gohman | 4e3e3de | 2009-02-07 00:43:41 +0000 | [diff] [blame] | 79 | |
| 80 | bool hasSymbolicDisplacement() const { |
Craig Topper | 062a2ba | 2014-04-25 05:30:21 +0000 | [diff] [blame] | 81 | return GV != nullptr || CP != nullptr || ES != nullptr || |
| 82 | JT != -1 || BlockAddr != nullptr; |
Dan Gohman | 4e3e3de | 2009-02-07 00:43:41 +0000 | [diff] [blame] | 83 | } |
Chad Rosier | 24c19d2 | 2012-08-01 18:39:17 +0000 | [diff] [blame] | 84 | |
Chris Lattner | fea81da | 2009-06-27 04:16:01 +0000 | [diff] [blame] | 85 | bool hasBaseOrIndexReg() const { |
Tim Northover | 97347a8 | 2013-09-19 11:33:53 +0000 | [diff] [blame] | 86 | return BaseType == FrameIndexBase || |
Craig Topper | 062a2ba | 2014-04-25 05:30:21 +0000 | [diff] [blame] | 87 | IndexReg.getNode() != nullptr || Base_Reg.getNode() != nullptr; |
Chris Lattner | fea81da | 2009-06-27 04:16:01 +0000 | [diff] [blame] | 88 | } |
Chad Rosier | 24c19d2 | 2012-08-01 18:39:17 +0000 | [diff] [blame] | 89 | |
Chris Lattner | fea81da | 2009-06-27 04:16:01 +0000 | [diff] [blame] | 90 | /// isRIPRelative - Return true if this addressing mode is already RIP |
| 91 | /// relative. |
| 92 | bool isRIPRelative() const { |
| 93 | if (BaseType != RegBase) return false; |
| 94 | if (RegisterSDNode *RegNode = |
Dan Gohman | 0fd54fb | 2010-04-29 23:30:41 +0000 | [diff] [blame] | 95 | dyn_cast_or_null<RegisterSDNode>(Base_Reg.getNode())) |
Chris Lattner | fea81da | 2009-06-27 04:16:01 +0000 | [diff] [blame] | 96 | return RegNode->getReg() == X86::RIP; |
| 97 | return false; |
| 98 | } |
Chad Rosier | 24c19d2 | 2012-08-01 18:39:17 +0000 | [diff] [blame] | 99 | |
Chris Lattner | fea81da | 2009-06-27 04:16:01 +0000 | [diff] [blame] | 100 | void setBaseReg(SDValue Reg) { |
| 101 | BaseType = RegBase; |
Dan Gohman | 0fd54fb | 2010-04-29 23:30:41 +0000 | [diff] [blame] | 102 | Base_Reg = Reg; |
Chris Lattner | fea81da | 2009-06-27 04:16:01 +0000 | [diff] [blame] | 103 | } |
Dan Gohman | 4e3e3de | 2009-02-07 00:43:41 +0000 | [diff] [blame] | 104 | |
Manman Ren | 19f49ac | 2012-09-11 22:23:19 +0000 | [diff] [blame] | 105 | #if !defined(NDEBUG) || defined(LLVM_ENABLE_DUMP) |
Dale Johannesen | dafdbf7 | 2008-08-11 23:46:25 +0000 | [diff] [blame] | 106 | void dump() { |
David Greene | dbdb1b2 | 2010-01-05 01:29:08 +0000 | [diff] [blame] | 107 | dbgs() << "X86ISelAddressMode " << this << '\n'; |
Dan Gohman | 0fd54fb | 2010-04-29 23:30:41 +0000 | [diff] [blame] | 108 | dbgs() << "Base_Reg "; |
Craig Topper | e73658d | 2014-04-28 04:05:08 +0000 | [diff] [blame] | 109 | if (Base_Reg.getNode()) |
Chad Rosier | 24c19d2 | 2012-08-01 18:39:17 +0000 | [diff] [blame] | 110 | Base_Reg.getNode()->dump(); |
Bill Wendling | fe3bdb4 | 2009-08-07 21:33:25 +0000 | [diff] [blame] | 111 | else |
David Greene | dbdb1b2 | 2010-01-05 01:29:08 +0000 | [diff] [blame] | 112 | dbgs() << "nul"; |
Dan Gohman | 0fd54fb | 2010-04-29 23:30:41 +0000 | [diff] [blame] | 113 | dbgs() << " Base.FrameIndex " << Base_FrameIndex << '\n' |
Benjamin Kramer | 940fbb0 | 2009-08-23 11:52:17 +0000 | [diff] [blame] | 114 | << " Scale" << Scale << '\n' |
| 115 | << "IndexReg "; |
Craig Topper | e73658d | 2014-04-28 04:05:08 +0000 | [diff] [blame] | 116 | if (IndexReg.getNode()) |
Bill Wendling | fe3bdb4 | 2009-08-07 21:33:25 +0000 | [diff] [blame] | 117 | IndexReg.getNode()->dump(); |
| 118 | else |
Chad Rosier | 24c19d2 | 2012-08-01 18:39:17 +0000 | [diff] [blame] | 119 | dbgs() << "nul"; |
David Greene | dbdb1b2 | 2010-01-05 01:29:08 +0000 | [diff] [blame] | 120 | dbgs() << " Disp " << Disp << '\n' |
Benjamin Kramer | 940fbb0 | 2009-08-23 11:52:17 +0000 | [diff] [blame] | 121 | << "GV "; |
Bill Wendling | fe3bdb4 | 2009-08-07 21:33:25 +0000 | [diff] [blame] | 122 | if (GV) |
| 123 | GV->dump(); |
| 124 | else |
David Greene | dbdb1b2 | 2010-01-05 01:29:08 +0000 | [diff] [blame] | 125 | dbgs() << "nul"; |
| 126 | dbgs() << " CP "; |
Bill Wendling | fe3bdb4 | 2009-08-07 21:33:25 +0000 | [diff] [blame] | 127 | if (CP) |
| 128 | CP->dump(); |
| 129 | else |
David Greene | dbdb1b2 | 2010-01-05 01:29:08 +0000 | [diff] [blame] | 130 | dbgs() << "nul"; |
| 131 | dbgs() << '\n' |
Benjamin Kramer | 940fbb0 | 2009-08-23 11:52:17 +0000 | [diff] [blame] | 132 | << "ES "; |
Bill Wendling | fe3bdb4 | 2009-08-07 21:33:25 +0000 | [diff] [blame] | 133 | if (ES) |
David Greene | dbdb1b2 | 2010-01-05 01:29:08 +0000 | [diff] [blame] | 134 | dbgs() << ES; |
Bill Wendling | fe3bdb4 | 2009-08-07 21:33:25 +0000 | [diff] [blame] | 135 | else |
David Greene | dbdb1b2 | 2010-01-05 01:29:08 +0000 | [diff] [blame] | 136 | dbgs() << "nul"; |
| 137 | dbgs() << " JT" << JT << " Align" << Align << '\n'; |
Dale Johannesen | dafdbf7 | 2008-08-11 23:46:25 +0000 | [diff] [blame] | 138 | } |
Manman Ren | 742534c | 2012-09-06 19:06:06 +0000 | [diff] [blame] | 139 | #endif |
Chris Lattner | 3f0f71b | 2005-11-19 02:11:08 +0000 | [diff] [blame] | 140 | }; |
| 141 | } |
| 142 | |
| 143 | namespace { |
Chris Lattner | 655e7df | 2005-11-16 01:54:32 +0000 | [diff] [blame] | 144 | //===--------------------------------------------------------------------===// |
| 145 | /// ISel - X86 specific code to select X86 machine instructions for |
| 146 | /// SelectionDAG operations. |
| 147 | /// |
Craig Topper | 26eec09 | 2014-03-31 06:22:15 +0000 | [diff] [blame] | 148 | class X86DAGToDAGISel final : public SelectionDAGISel { |
Chris Lattner | 655e7df | 2005-11-16 01:54:32 +0000 | [diff] [blame] | 149 | /// Subtarget - Keep a pointer to the X86Subtarget around so that we can |
| 150 | /// make the right decision when generating code for different targets. |
| 151 | const X86Subtarget *Subtarget; |
Evan Cheng | 5588de9 | 2006-02-18 00:15:05 +0000 | [diff] [blame] | 152 | |
Evan Cheng | 7d6fa97 | 2008-09-26 23:41:32 +0000 | [diff] [blame] | 153 | /// OptForSize - If true, selector should try to optimize for code size |
| 154 | /// instead of performance. |
| 155 | bool OptForSize; |
| 156 | |
Chris Lattner | 655e7df | 2005-11-16 01:54:32 +0000 | [diff] [blame] | 157 | public: |
Bill Wendling | 026e5d7 | 2009-04-29 23:29:43 +0000 | [diff] [blame] | 158 | explicit X86DAGToDAGISel(X86TargetMachine &tm, CodeGenOpt::Level OptLevel) |
Eric Christopher | 05b8197 | 2015-02-02 17:38:43 +0000 | [diff] [blame] | 159 | : SelectionDAGISel(tm, OptLevel), OptForSize(false) {} |
Chris Lattner | 655e7df | 2005-11-16 01:54:32 +0000 | [diff] [blame] | 160 | |
Craig Topper | 2d9361e | 2014-03-09 07:44:38 +0000 | [diff] [blame] | 161 | const char *getPassName() const override { |
Chris Lattner | 655e7df | 2005-11-16 01:54:32 +0000 | [diff] [blame] | 162 | return "X86 DAG->DAG Instruction Selection"; |
| 163 | } |
| 164 | |
Eric Christopher | 4f09c59 | 2014-05-22 01:53:26 +0000 | [diff] [blame] | 165 | bool runOnMachineFunction(MachineFunction &MF) override { |
| 166 | // Reset the subtarget each time through. |
Eric Christopher | 05b8197 | 2015-02-02 17:38:43 +0000 | [diff] [blame] | 167 | Subtarget = &MF.getSubtarget<X86Subtarget>(); |
Eric Christopher | 4f09c59 | 2014-05-22 01:53:26 +0000 | [diff] [blame] | 168 | SelectionDAGISel::runOnMachineFunction(MF); |
| 169 | return true; |
| 170 | } |
| 171 | |
Craig Topper | 2d9361e | 2014-03-09 07:44:38 +0000 | [diff] [blame] | 172 | void EmitFunctionEntryCode() override; |
Anton Korobeynikov | 9091074 | 2007-09-25 21:52:30 +0000 | [diff] [blame] | 173 | |
Craig Topper | 2d9361e | 2014-03-09 07:44:38 +0000 | [diff] [blame] | 174 | bool IsProfitableToFold(SDValue N, SDNode *U, SDNode *Root) const override; |
Evan Cheng | 5e73ff2 | 2010-02-15 19:41:07 +0000 | [diff] [blame] | 175 | |
Craig Topper | 2d9361e | 2014-03-09 07:44:38 +0000 | [diff] [blame] | 176 | void PreprocessISelDAG() override; |
Chris Lattner | f98f124 | 2010-03-02 06:34:30 +0000 | [diff] [blame] | 177 | |
Jakob Stoklund Olesen | 08aede2 | 2010-09-03 00:35:18 +0000 | [diff] [blame] | 178 | inline bool immSext8(SDNode *N) const { |
| 179 | return isInt<8>(cast<ConstantSDNode>(N)->getSExtValue()); |
| 180 | } |
| 181 | |
| 182 | // i64immSExt32 predicate - True if the 64-bit immediate fits in a 32-bit |
| 183 | // sign extended field. |
| 184 | inline bool i64immSExt32(SDNode *N) const { |
| 185 | uint64_t v = cast<ConstantSDNode>(N)->getZExtValue(); |
| 186 | return (int64_t)v == (int32_t)v; |
| 187 | } |
| 188 | |
Chris Lattner | 655e7df | 2005-11-16 01:54:32 +0000 | [diff] [blame] | 189 | // Include the pieces autogenerated from the target description. |
| 190 | #include "X86GenDAGISel.inc" |
| 191 | |
| 192 | private: |
Craig Topper | 2d9361e | 2014-03-09 07:44:38 +0000 | [diff] [blame] | 193 | SDNode *Select(SDNode *N) override; |
Manman Ren | a098204 | 2012-06-26 19:47:59 +0000 | [diff] [blame] | 194 | SDNode *SelectGather(SDNode *N, unsigned Opc); |
Craig Topper | 83e042a | 2013-08-15 05:57:07 +0000 | [diff] [blame] | 195 | SDNode *SelectAtomicLoadArith(SDNode *Node, MVT NVT); |
Chris Lattner | 655e7df | 2005-11-16 01:54:32 +0000 | [diff] [blame] | 196 | |
Eli Friedman | ef67e7d | 2011-07-13 20:44:23 +0000 | [diff] [blame] | 197 | bool FoldOffsetIntoAddress(uint64_t Offset, X86ISelAddressMode &AM); |
Chris Lattner | 8a236b6 | 2010-09-22 04:39:11 +0000 | [diff] [blame] | 198 | bool MatchLoadInAddress(LoadSDNode *N, X86ISelAddressMode &AM); |
Rafael Espindola | 6688b0a | 2009-04-12 21:55:03 +0000 | [diff] [blame] | 199 | bool MatchWrapper(SDValue N, X86ISelAddressMode &AM); |
Dan Gohman | 824ab40 | 2009-07-22 23:26:55 +0000 | [diff] [blame] | 200 | bool MatchAddress(SDValue N, X86ISelAddressMode &AM); |
| 201 | bool MatchAddressRecursively(SDValue N, X86ISelAddressMode &AM, |
| 202 | unsigned Depth); |
Rafael Espindola | 9277379 | 2009-03-31 16:16:57 +0000 | [diff] [blame] | 203 | bool MatchAddressBase(SDValue N, X86ISelAddressMode &AM); |
Chris Lattner | d58d7c1 | 2010-09-21 22:07:31 +0000 | [diff] [blame] | 204 | bool SelectAddr(SDNode *Parent, SDValue N, SDValue &Base, |
Rafael Espindola | 3b2df10 | 2009-04-08 21:14:34 +0000 | [diff] [blame] | 205 | SDValue &Scale, SDValue &Index, SDValue &Disp, |
| 206 | SDValue &Segment); |
Elena Demikhovsky | e1eda8a | 2015-04-30 08:38:48 +0000 | [diff] [blame] | 207 | bool SelectVectorAddr(SDNode *Parent, SDValue N, SDValue &Base, |
| 208 | SDValue &Scale, SDValue &Index, SDValue &Disp, |
| 209 | SDValue &Segment); |
Tim Northover | 3a1fd4c | 2013-06-01 09:55:14 +0000 | [diff] [blame] | 210 | bool SelectMOV64Imm32(SDValue N, SDValue &Imm); |
Chris Lattner | 0e023ea | 2010-09-21 20:31:19 +0000 | [diff] [blame] | 211 | bool SelectLEAAddr(SDValue N, SDValue &Base, |
Chris Lattner | f469307 | 2010-07-08 23:46:44 +0000 | [diff] [blame] | 212 | SDValue &Scale, SDValue &Index, SDValue &Disp, |
| 213 | SDValue &Segment); |
Tim Northover | 6833e3f | 2013-06-10 20:43:49 +0000 | [diff] [blame] | 214 | bool SelectLEA64_32Addr(SDValue N, SDValue &Base, |
| 215 | SDValue &Scale, SDValue &Index, SDValue &Disp, |
| 216 | SDValue &Segment); |
Chris Lattner | 0e023ea | 2010-09-21 20:31:19 +0000 | [diff] [blame] | 217 | bool SelectTLSADDRAddr(SDValue N, SDValue &Base, |
Chris Lattner | f469307 | 2010-07-08 23:46:44 +0000 | [diff] [blame] | 218 | SDValue &Scale, SDValue &Index, SDValue &Disp, |
| 219 | SDValue &Segment); |
Chris Lattner | bd6e193 | 2010-03-01 22:51:11 +0000 | [diff] [blame] | 220 | bool SelectScalarSSELoad(SDNode *Root, SDValue N, |
Chris Lattner | afac7dad | 2010-02-16 22:35:06 +0000 | [diff] [blame] | 221 | SDValue &Base, SDValue &Scale, |
Dan Gohman | 2ce6f2a | 2008-07-27 21:46:04 +0000 | [diff] [blame] | 222 | SDValue &Index, SDValue &Disp, |
Rafael Espindola | 3b2df10 | 2009-04-08 21:14:34 +0000 | [diff] [blame] | 223 | SDValue &Segment, |
Chris Lattner | 18a32ce | 2010-02-21 03:17:59 +0000 | [diff] [blame] | 224 | SDValue &NodeWithChain); |
Chad Rosier | 24c19d2 | 2012-08-01 18:39:17 +0000 | [diff] [blame] | 225 | |
Dan Gohman | ea6f91f | 2010-01-05 01:24:18 +0000 | [diff] [blame] | 226 | bool TryFoldLoad(SDNode *P, SDValue N, |
Dan Gohman | 2ce6f2a | 2008-07-27 21:46:04 +0000 | [diff] [blame] | 227 | SDValue &Base, SDValue &Scale, |
Rafael Espindola | 3b2df10 | 2009-04-08 21:14:34 +0000 | [diff] [blame] | 228 | SDValue &Index, SDValue &Disp, |
| 229 | SDValue &Segment); |
Chad Rosier | 24c19d2 | 2012-08-01 18:39:17 +0000 | [diff] [blame] | 230 | |
Chris Lattner | ba1ed58 | 2006-06-08 18:03:49 +0000 | [diff] [blame] | 231 | /// SelectInlineAsmMemoryOperand - Implement addressing mode selection for |
| 232 | /// inline asm expressions. |
Craig Topper | 2d9361e | 2014-03-09 07:44:38 +0000 | [diff] [blame] | 233 | bool SelectInlineAsmMemoryOperand(const SDValue &Op, |
Daniel Sanders | 60f1db0 | 2015-03-13 12:45:09 +0000 | [diff] [blame] | 234 | unsigned ConstraintID, |
Craig Topper | 2d9361e | 2014-03-09 07:44:38 +0000 | [diff] [blame] | 235 | std::vector<SDValue> &OutOps) override; |
Chad Rosier | 24c19d2 | 2012-08-01 18:39:17 +0000 | [diff] [blame] | 236 | |
David Majnemer | d5ab35f | 2015-02-21 05:49:45 +0000 | [diff] [blame] | 237 | void EmitSpecialCodeForMain(); |
Anton Korobeynikov | 9091074 | 2007-09-25 21:52:30 +0000 | [diff] [blame] | 238 | |
Sergey Dmitrouk | 842a51b | 2015-04-28 14:05:47 +0000 | [diff] [blame] | 239 | inline void getAddressOperands(X86ISelAddressMode &AM, SDLoc DL, |
| 240 | SDValue &Base, SDValue &Scale, |
| 241 | SDValue &Index, SDValue &Disp, |
| 242 | SDValue &Segment) { |
Eric Christopher | b17140d | 2014-10-08 07:32:17 +0000 | [diff] [blame] | 243 | Base = (AM.BaseType == X86ISelAddressMode::FrameIndexBase) |
| 244 | ? CurDAG->getTargetFrameIndex(AM.Base_FrameIndex, |
| 245 | TLI->getPointerTy()) |
| 246 | : AM.Base_Reg; |
Sergey Dmitrouk | 842a51b | 2015-04-28 14:05:47 +0000 | [diff] [blame] | 247 | Scale = getI8Imm(AM.Scale, DL); |
Evan Cheng | 67ed58e | 2005-12-12 21:49:40 +0000 | [diff] [blame] | 248 | Index = AM.IndexReg; |
Evan Cheng | 11b0a5d | 2006-09-08 06:48:29 +0000 | [diff] [blame] | 249 | // These are 32-bit even in 64-bit mode since RIP relative offset |
| 250 | // is 32-bit. |
| 251 | if (AM.GV) |
Andrew Trick | ef9de2a | 2013-05-25 02:42:55 +0000 | [diff] [blame] | 252 | Disp = CurDAG->getTargetGlobalAddress(AM.GV, SDLoc(), |
Devang Patel | a3ca21b | 2010-07-06 22:08:15 +0000 | [diff] [blame] | 253 | MVT::i32, AM.Disp, |
Chris Lattner | bd7e26d | 2009-06-26 05:51:45 +0000 | [diff] [blame] | 254 | AM.SymbolFlags); |
Evan Cheng | 11b0a5d | 2006-09-08 06:48:29 +0000 | [diff] [blame] | 255 | else if (AM.CP) |
Owen Anderson | 9f94459 | 2009-08-11 20:47:22 +0000 | [diff] [blame] | 256 | Disp = CurDAG->getTargetConstantPool(AM.CP, MVT::i32, |
Chris Lattner | bd7e26d | 2009-06-26 05:51:45 +0000 | [diff] [blame] | 257 | AM.Align, AM.Disp, AM.SymbolFlags); |
Michael Liao | abb87d4 | 2012-09-12 21:43:09 +0000 | [diff] [blame] | 258 | else if (AM.ES) { |
| 259 | assert(!AM.Disp && "Non-zero displacement is ignored with ES."); |
Owen Anderson | 9f94459 | 2009-08-11 20:47:22 +0000 | [diff] [blame] | 260 | Disp = CurDAG->getTargetExternalSymbol(AM.ES, MVT::i32, AM.SymbolFlags); |
Michael Liao | abb87d4 | 2012-09-12 21:43:09 +0000 | [diff] [blame] | 261 | } else if (AM.JT != -1) { |
| 262 | assert(!AM.Disp && "Non-zero displacement is ignored with JT."); |
Owen Anderson | 9f94459 | 2009-08-11 20:47:22 +0000 | [diff] [blame] | 263 | Disp = CurDAG->getTargetJumpTable(AM.JT, MVT::i32, AM.SymbolFlags); |
Michael Liao | abb87d4 | 2012-09-12 21:43:09 +0000 | [diff] [blame] | 264 | } else if (AM.BlockAddr) |
| 265 | Disp = CurDAG->getTargetBlockAddress(AM.BlockAddr, MVT::i32, AM.Disp, |
| 266 | AM.SymbolFlags); |
Evan Cheng | 11b0a5d | 2006-09-08 06:48:29 +0000 | [diff] [blame] | 267 | else |
Sergey Dmitrouk | 842a51b | 2015-04-28 14:05:47 +0000 | [diff] [blame] | 268 | Disp = CurDAG->getTargetConstant(AM.Disp, DL, MVT::i32); |
Rafael Espindola | 3b2df10 | 2009-04-08 21:14:34 +0000 | [diff] [blame] | 269 | |
| 270 | if (AM.Segment.getNode()) |
| 271 | Segment = AM.Segment; |
| 272 | else |
Owen Anderson | 9f94459 | 2009-08-11 20:47:22 +0000 | [diff] [blame] | 273 | Segment = CurDAG->getRegister(0, MVT::i32); |
Evan Cheng | 67ed58e | 2005-12-12 21:49:40 +0000 | [diff] [blame] | 274 | } |
| 275 | |
Chris Lattner | 3f0f71b | 2005-11-19 02:11:08 +0000 | [diff] [blame] | 276 | /// getI8Imm - Return a target constant with the specified value, of type |
| 277 | /// i8. |
Sergey Dmitrouk | 842a51b | 2015-04-28 14:05:47 +0000 | [diff] [blame] | 278 | inline SDValue getI8Imm(unsigned Imm, SDLoc DL) { |
| 279 | return CurDAG->getTargetConstant(Imm, DL, MVT::i8); |
Chris Lattner | 3f0f71b | 2005-11-19 02:11:08 +0000 | [diff] [blame] | 280 | } |
| 281 | |
Chris Lattner | 655e7df | 2005-11-16 01:54:32 +0000 | [diff] [blame] | 282 | /// getI32Imm - Return a target constant with the specified value, of type |
| 283 | /// i32. |
Sergey Dmitrouk | 842a51b | 2015-04-28 14:05:47 +0000 | [diff] [blame] | 284 | inline SDValue getI32Imm(unsigned Imm, SDLoc DL) { |
| 285 | return CurDAG->getTargetConstant(Imm, DL, MVT::i32); |
Chris Lattner | 655e7df | 2005-11-16 01:54:32 +0000 | [diff] [blame] | 286 | } |
Evan Cheng | d49cc36 | 2006-02-10 22:24:32 +0000 | [diff] [blame] | 287 | |
Dan Gohman | 2430073 | 2008-09-23 18:22:58 +0000 | [diff] [blame] | 288 | /// getGlobalBaseReg - Return an SDNode that returns the value of |
| 289 | /// the global base register. Output instructions required to |
| 290 | /// initialize the global base register, if necessary. |
| 291 | /// |
Evan Cheng | 61413a3 | 2006-08-26 05:34:46 +0000 | [diff] [blame] | 292 | SDNode *getGlobalBaseReg(); |
Evan Cheng | 5588de9 | 2006-02-18 00:15:05 +0000 | [diff] [blame] | 293 | |
Dan Gohman | 4751bb9 | 2009-06-03 20:20:00 +0000 | [diff] [blame] | 294 | /// getTargetMachine - Return a reference to the TargetMachine, casted |
| 295 | /// to the target-specific type. |
Jakub Staszak | e167cf5 | 2013-02-19 21:54:59 +0000 | [diff] [blame] | 296 | const X86TargetMachine &getTargetMachine() const { |
Dan Gohman | 4751bb9 | 2009-06-03 20:20:00 +0000 | [diff] [blame] | 297 | return static_cast<const X86TargetMachine &>(TM); |
| 298 | } |
| 299 | |
| 300 | /// getInstrInfo - Return a reference to the TargetInstrInfo, casted |
| 301 | /// to the target-specific type. |
Jakub Staszak | e167cf5 | 2013-02-19 21:54:59 +0000 | [diff] [blame] | 302 | const X86InstrInfo *getInstrInfo() const { |
Eric Christopher | 05b8197 | 2015-02-02 17:38:43 +0000 | [diff] [blame] | 303 | return Subtarget->getInstrInfo(); |
Dan Gohman | 4751bb9 | 2009-06-03 20:20:00 +0000 | [diff] [blame] | 304 | } |
Adam Nemet | ff63a2d | 2014-10-03 20:00:34 +0000 | [diff] [blame] | 305 | |
| 306 | /// \brief Address-mode matching performs shift-of-and to and-of-shift |
| 307 | /// reassociation in order to expose more scaled addressing |
| 308 | /// opportunities. |
| 309 | bool ComplexPatternFuncMutatesDAG() const override { |
| 310 | return true; |
| 311 | } |
Chris Lattner | 655e7df | 2005-11-16 01:54:32 +0000 | [diff] [blame] | 312 | }; |
| 313 | } |
| 314 | |
Evan Cheng | 72bb66a | 2006-08-08 00:31:00 +0000 | [diff] [blame] | 315 | |
Evan Cheng | 5e73ff2 | 2010-02-15 19:41:07 +0000 | [diff] [blame] | 316 | bool |
| 317 | X86DAGToDAGISel::IsProfitableToFold(SDValue N, SDNode *U, SDNode *Root) const { |
Bill Wendling | 026e5d7 | 2009-04-29 23:29:43 +0000 | [diff] [blame] | 318 | if (OptLevel == CodeGenOpt::None) return false; |
Evan Cheng | b86375c | 2006-10-14 08:33:25 +0000 | [diff] [blame] | 319 | |
Evan Cheng | 5e73ff2 | 2010-02-15 19:41:07 +0000 | [diff] [blame] | 320 | if (!N.hasOneUse()) |
| 321 | return false; |
| 322 | |
| 323 | if (N.getOpcode() != ISD::LOAD) |
| 324 | return true; |
| 325 | |
| 326 | // If N is a load, do additional profitability checks. |
| 327 | if (U == Root) { |
Evan Cheng | 83bdb38 | 2008-11-27 00:49:46 +0000 | [diff] [blame] | 328 | switch (U->getOpcode()) { |
| 329 | default: break; |
Dan Gohman | 85d4fdf | 2010-01-04 20:51:50 +0000 | [diff] [blame] | 330 | case X86ISD::ADD: |
| 331 | case X86ISD::SUB: |
| 332 | case X86ISD::AND: |
| 333 | case X86ISD::XOR: |
| 334 | case X86ISD::OR: |
Evan Cheng | 83bdb38 | 2008-11-27 00:49:46 +0000 | [diff] [blame] | 335 | case ISD::ADD: |
| 336 | case ISD::ADDC: |
| 337 | case ISD::ADDE: |
| 338 | case ISD::AND: |
| 339 | case ISD::OR: |
| 340 | case ISD::XOR: { |
Rafael Espindola | bb834f0 | 2009-04-10 10:09:34 +0000 | [diff] [blame] | 341 | SDValue Op1 = U->getOperand(1); |
| 342 | |
Evan Cheng | 83bdb38 | 2008-11-27 00:49:46 +0000 | [diff] [blame] | 343 | // If the other operand is a 8-bit immediate we should fold the immediate |
| 344 | // instead. This reduces code size. |
| 345 | // e.g. |
| 346 | // movl 4(%esp), %eax |
| 347 | // addl $4, %eax |
| 348 | // vs. |
| 349 | // movl $4, %eax |
| 350 | // addl 4(%esp), %eax |
| 351 | // The former is 2 bytes shorter. In case where the increment is 1, then |
| 352 | // the saving can be 4 bytes (by using incl %eax). |
Rafael Espindola | bb834f0 | 2009-04-10 10:09:34 +0000 | [diff] [blame] | 353 | if (ConstantSDNode *Imm = dyn_cast<ConstantSDNode>(Op1)) |
Dan Gohman | 2293eb6 | 2009-03-14 02:07:16 +0000 | [diff] [blame] | 354 | if (Imm->getAPIntValue().isSignedIntN(8)) |
| 355 | return false; |
Rafael Espindola | bb834f0 | 2009-04-10 10:09:34 +0000 | [diff] [blame] | 356 | |
| 357 | // If the other operand is a TLS address, we should fold it instead. |
| 358 | // This produces |
| 359 | // movl %gs:0, %eax |
| 360 | // leal i@NTPOFF(%eax), %eax |
| 361 | // instead of |
| 362 | // movl $i@NTPOFF, %eax |
| 363 | // addl %gs:0, %eax |
| 364 | // if the block also has an access to a second TLS address this will save |
| 365 | // a load. |
Alp Toker | f907b89 | 2013-12-05 05:44:44 +0000 | [diff] [blame] | 366 | // FIXME: This is probably also true for non-TLS addresses. |
Rafael Espindola | bb834f0 | 2009-04-10 10:09:34 +0000 | [diff] [blame] | 367 | if (Op1.getOpcode() == X86ISD::Wrapper) { |
| 368 | SDValue Val = Op1.getOperand(0); |
| 369 | if (Val.getOpcode() == ISD::TargetGlobalTLSAddress) |
| 370 | return false; |
| 371 | } |
Evan Cheng | 83bdb38 | 2008-11-27 00:49:46 +0000 | [diff] [blame] | 372 | } |
| 373 | } |
Evan Cheng | 5e73ff2 | 2010-02-15 19:41:07 +0000 | [diff] [blame] | 374 | } |
| 375 | |
| 376 | return true; |
| 377 | } |
| 378 | |
Evan Cheng | d703df6 | 2010-03-14 03:48:46 +0000 | [diff] [blame] | 379 | /// MoveBelowCallOrigChain - Replace the original chain operand of the call with |
| 380 | /// load's chain operand and move load below the call's chain operand. |
| 381 | static void MoveBelowOrigChain(SelectionDAG *CurDAG, SDValue Load, |
Evan Cheng | 214156c | 2012-10-02 23:49:13 +0000 | [diff] [blame] | 382 | SDValue Call, SDValue OrigChain) { |
Evan Cheng | f00f1e5 | 2008-08-25 21:27:18 +0000 | [diff] [blame] | 383 | SmallVector<SDValue, 8> Ops; |
Evan Cheng | d703df6 | 2010-03-14 03:48:46 +0000 | [diff] [blame] | 384 | SDValue Chain = OrigChain.getOperand(0); |
Evan Cheng | 6c7e851 | 2009-01-26 18:43:34 +0000 | [diff] [blame] | 385 | if (Chain.getNode() == Load.getNode()) |
| 386 | Ops.push_back(Load.getOperand(0)); |
| 387 | else { |
| 388 | assert(Chain.getOpcode() == ISD::TokenFactor && |
Evan Cheng | d703df6 | 2010-03-14 03:48:46 +0000 | [diff] [blame] | 389 | "Unexpected chain operand"); |
Evan Cheng | 6c7e851 | 2009-01-26 18:43:34 +0000 | [diff] [blame] | 390 | for (unsigned i = 0, e = Chain.getNumOperands(); i != e; ++i) |
| 391 | if (Chain.getOperand(i).getNode() == Load.getNode()) |
| 392 | Ops.push_back(Load.getOperand(0)); |
| 393 | else |
| 394 | Ops.push_back(Chain.getOperand(i)); |
| 395 | SDValue NewChain = |
Craig Topper | 48d114b | 2014-04-26 18:35:24 +0000 | [diff] [blame] | 396 | CurDAG->getNode(ISD::TokenFactor, SDLoc(Load), MVT::Other, Ops); |
Evan Cheng | 6c7e851 | 2009-01-26 18:43:34 +0000 | [diff] [blame] | 397 | Ops.clear(); |
| 398 | Ops.push_back(NewChain); |
| 399 | } |
Benjamin Kramer | 6cd780f | 2015-02-17 15:29:18 +0000 | [diff] [blame] | 400 | Ops.append(OrigChain->op_begin() + 1, OrigChain->op_end()); |
Craig Topper | 8c0b4d0 | 2014-04-28 05:57:50 +0000 | [diff] [blame] | 401 | CurDAG->UpdateNodeOperands(OrigChain.getNode(), Ops); |
Dan Gohman | 92c11ac | 2010-06-18 15:30:29 +0000 | [diff] [blame] | 402 | CurDAG->UpdateNodeOperands(Load.getNode(), Call.getOperand(0), |
Evan Cheng | f00f1e5 | 2008-08-25 21:27:18 +0000 | [diff] [blame] | 403 | Load.getOperand(1), Load.getOperand(2)); |
Evan Cheng | 214156c | 2012-10-02 23:49:13 +0000 | [diff] [blame] | 404 | |
Evan Cheng | f00f1e5 | 2008-08-25 21:27:18 +0000 | [diff] [blame] | 405 | Ops.clear(); |
Gabor Greif | f304a7a | 2008-08-28 21:40:38 +0000 | [diff] [blame] | 406 | Ops.push_back(SDValue(Load.getNode(), 1)); |
Benjamin Kramer | 6cd780f | 2015-02-17 15:29:18 +0000 | [diff] [blame] | 407 | Ops.append(Call->op_begin() + 1, Call->op_end()); |
Craig Topper | 8c0b4d0 | 2014-04-28 05:57:50 +0000 | [diff] [blame] | 408 | CurDAG->UpdateNodeOperands(Call.getNode(), Ops); |
Evan Cheng | f00f1e5 | 2008-08-25 21:27:18 +0000 | [diff] [blame] | 409 | } |
| 410 | |
| 411 | /// isCalleeLoad - Return true if call address is a load and it can be |
| 412 | /// moved below CALLSEQ_START and the chains leading up to the call. |
| 413 | /// Return the CALLSEQ_START by reference as a second output. |
Evan Cheng | d703df6 | 2010-03-14 03:48:46 +0000 | [diff] [blame] | 414 | /// In the case of a tail call, there isn't a callseq node between the call |
| 415 | /// chain and the load. |
| 416 | static bool isCalleeLoad(SDValue Callee, SDValue &Chain, bool HasCallSeq) { |
Evan Cheng | 847ad44 | 2012-10-05 01:48:22 +0000 | [diff] [blame] | 417 | // The transformation is somewhat dangerous if the call's chain was glued to |
| 418 | // the call. After MoveBelowOrigChain the load is moved between the call and |
| 419 | // the chain, this can create a cycle if the load is not folded. So it is |
| 420 | // *really* important that we are sure the load will be folded. |
Gabor Greif | f304a7a | 2008-08-28 21:40:38 +0000 | [diff] [blame] | 421 | if (Callee.getNode() == Chain.getNode() || !Callee.hasOneUse()) |
Evan Cheng | f00f1e5 | 2008-08-25 21:27:18 +0000 | [diff] [blame] | 422 | return false; |
Gabor Greif | f304a7a | 2008-08-28 21:40:38 +0000 | [diff] [blame] | 423 | LoadSDNode *LD = dyn_cast<LoadSDNode>(Callee.getNode()); |
Evan Cheng | f00f1e5 | 2008-08-25 21:27:18 +0000 | [diff] [blame] | 424 | if (!LD || |
| 425 | LD->isVolatile() || |
| 426 | LD->getAddressingMode() != ISD::UNINDEXED || |
| 427 | LD->getExtensionType() != ISD::NON_EXTLOAD) |
| 428 | return false; |
| 429 | |
| 430 | // Now let's find the callseq_start. |
Evan Cheng | d703df6 | 2010-03-14 03:48:46 +0000 | [diff] [blame] | 431 | while (HasCallSeq && Chain.getOpcode() != ISD::CALLSEQ_START) { |
Evan Cheng | f00f1e5 | 2008-08-25 21:27:18 +0000 | [diff] [blame] | 432 | if (!Chain.hasOneUse()) |
| 433 | return false; |
| 434 | Chain = Chain.getOperand(0); |
| 435 | } |
Evan Cheng | d703df6 | 2010-03-14 03:48:46 +0000 | [diff] [blame] | 436 | |
| 437 | if (!Chain.getNumOperands()) |
| 438 | return false; |
Evan Cheng | 3fb03e2 | 2013-01-06 19:00:15 +0000 | [diff] [blame] | 439 | // Since we are not checking for AA here, conservatively abort if the chain |
| 440 | // writes to memory. It's not safe to move the callee (a load) across a store. |
| 441 | if (isa<MemSDNode>(Chain.getNode()) && |
| 442 | cast<MemSDNode>(Chain.getNode())->writeMem()) |
| 443 | return false; |
Evan Cheng | 6c7e851 | 2009-01-26 18:43:34 +0000 | [diff] [blame] | 444 | if (Chain.getOperand(0).getNode() == Callee.getNode()) |
| 445 | return true; |
| 446 | if (Chain.getOperand(0).getOpcode() == ISD::TokenFactor && |
Dan Gohman | 520a685 | 2009-09-15 01:22:01 +0000 | [diff] [blame] | 447 | Callee.getValue(1).isOperandOf(Chain.getOperand(0).getNode()) && |
| 448 | Callee.getValue(1).hasOneUse()) |
Evan Cheng | 6c7e851 | 2009-01-26 18:43:34 +0000 | [diff] [blame] | 449 | return true; |
| 450 | return false; |
Evan Cheng | f00f1e5 | 2008-08-25 21:27:18 +0000 | [diff] [blame] | 451 | } |
| 452 | |
Chris Lattner | 8d63704 | 2010-03-02 23:12:51 +0000 | [diff] [blame] | 453 | void X86DAGToDAGISel::PreprocessISelDAG() { |
Chris Lattner | 82cc533 | 2010-03-04 01:43:43 +0000 | [diff] [blame] | 454 | // OptForSize is used in pattern predicates that isel is matching. |
Duncan P. N. Exon Smith | 5975a70 | 2015-02-14 01:59:52 +0000 | [diff] [blame] | 455 | OptForSize = MF->getFunction()->hasFnAttribute(Attribute::OptimizeForSize); |
Chad Rosier | 24c19d2 | 2012-08-01 18:39:17 +0000 | [diff] [blame] | 456 | |
Dan Gohman | eb0cee9 | 2008-08-23 02:25:05 +0000 | [diff] [blame] | 457 | for (SelectionDAG::allnodes_iterator I = CurDAG->allnodes_begin(), |
| 458 | E = CurDAG->allnodes_end(); I != E; ) { |
Chris Lattner | a91f77e | 2008-01-24 08:07:48 +0000 | [diff] [blame] | 459 | SDNode *N = I++; // Preincrement iterator to avoid invalidation issues. |
Chris Lattner | 8d63704 | 2010-03-02 23:12:51 +0000 | [diff] [blame] | 460 | |
Evan Cheng | d703df6 | 2010-03-14 03:48:46 +0000 | [diff] [blame] | 461 | if (OptLevel != CodeGenOpt::None && |
Michael Liao | 96b4260 | 2013-03-28 23:13:21 +0000 | [diff] [blame] | 462 | // Only does this when target favors doesn't favor register indirect |
| 463 | // call. |
| 464 | ((N->getOpcode() == X86ISD::CALL && !Subtarget->callRegIndirect()) || |
Evan Cheng | 847ad44 | 2012-10-05 01:48:22 +0000 | [diff] [blame] | 465 | (N->getOpcode() == X86ISD::TC_RETURN && |
Nick Lewycky | f41a80e | 2013-01-13 19:03:55 +0000 | [diff] [blame] | 466 | // Only does this if load can be folded into TC_RETURN. |
Evan Cheng | 847ad44 | 2012-10-05 01:48:22 +0000 | [diff] [blame] | 467 | (Subtarget->is64Bit() || |
| 468 | getTargetMachine().getRelocationModel() != Reloc::PIC_)))) { |
Chris Lattner | 8d63704 | 2010-03-02 23:12:51 +0000 | [diff] [blame] | 469 | /// Also try moving call address load from outside callseq_start to just |
| 470 | /// before the call to allow it to be folded. |
| 471 | /// |
| 472 | /// [Load chain] |
| 473 | /// ^ |
| 474 | /// | |
| 475 | /// [Load] |
| 476 | /// ^ ^ |
| 477 | /// | | |
| 478 | /// / \-- |
| 479 | /// / | |
| 480 | ///[CALLSEQ_START] | |
| 481 | /// ^ | |
| 482 | /// | | |
| 483 | /// [LOAD/C2Reg] | |
| 484 | /// | | |
| 485 | /// \ / |
| 486 | /// \ / |
| 487 | /// [CALL] |
Evan Cheng | d703df6 | 2010-03-14 03:48:46 +0000 | [diff] [blame] | 488 | bool HasCallSeq = N->getOpcode() == X86ISD::CALL; |
Chris Lattner | 8d63704 | 2010-03-02 23:12:51 +0000 | [diff] [blame] | 489 | SDValue Chain = N->getOperand(0); |
| 490 | SDValue Load = N->getOperand(1); |
Evan Cheng | d703df6 | 2010-03-14 03:48:46 +0000 | [diff] [blame] | 491 | if (!isCalleeLoad(Load, Chain, HasCallSeq)) |
Chris Lattner | 8d63704 | 2010-03-02 23:12:51 +0000 | [diff] [blame] | 492 | continue; |
Evan Cheng | d703df6 | 2010-03-14 03:48:46 +0000 | [diff] [blame] | 493 | MoveBelowOrigChain(CurDAG, Load, SDValue(N, 0), Chain); |
Chris Lattner | 8d63704 | 2010-03-02 23:12:51 +0000 | [diff] [blame] | 494 | ++NumLoadMoved; |
| 495 | continue; |
| 496 | } |
Chad Rosier | 24c19d2 | 2012-08-01 18:39:17 +0000 | [diff] [blame] | 497 | |
Chris Lattner | 8d63704 | 2010-03-02 23:12:51 +0000 | [diff] [blame] | 498 | // Lower fpround and fpextend nodes that target the FP stack to be store and |
| 499 | // load to the stack. This is a gross hack. We would like to simply mark |
| 500 | // these as being illegal, but when we do that, legalize produces these when |
| 501 | // it expands calls, then expands these in the same legalize pass. We would |
| 502 | // like dag combine to be able to hack on these between the call expansion |
| 503 | // and the node legalization. As such this pass basically does "really |
| 504 | // late" legalization of these inline with the X86 isel pass. |
| 505 | // FIXME: This should only happen when not compiled with -O0. |
Chris Lattner | a91f77e | 2008-01-24 08:07:48 +0000 | [diff] [blame] | 506 | if (N->getOpcode() != ISD::FP_ROUND && N->getOpcode() != ISD::FP_EXTEND) |
| 507 | continue; |
Chad Rosier | 24c19d2 | 2012-08-01 18:39:17 +0000 | [diff] [blame] | 508 | |
Craig Topper | 83e042a | 2013-08-15 05:57:07 +0000 | [diff] [blame] | 509 | MVT SrcVT = N->getOperand(0).getSimpleValueType(); |
| 510 | MVT DstVT = N->getSimpleValueType(0); |
Bruno Cardoso Lopes | 616fe60 | 2011-08-01 21:54:05 +0000 | [diff] [blame] | 511 | |
| 512 | // If any of the sources are vectors, no fp stack involved. |
| 513 | if (SrcVT.isVector() || DstVT.isVector()) |
| 514 | continue; |
| 515 | |
| 516 | // If the source and destination are SSE registers, then this is a legal |
| 517 | // conversion that should not be lowered. |
Benjamin Kramer | 02ff1cd | 2013-06-27 11:07:42 +0000 | [diff] [blame] | 518 | const X86TargetLowering *X86Lowering = |
Eric Christopher | b17140d | 2014-10-08 07:32:17 +0000 | [diff] [blame] | 519 | static_cast<const X86TargetLowering *>(TLI); |
Bill Wendling | a3cd350 | 2013-06-19 21:36:55 +0000 | [diff] [blame] | 520 | bool SrcIsSSE = X86Lowering->isScalarFPTypeInSSEReg(SrcVT); |
| 521 | bool DstIsSSE = X86Lowering->isScalarFPTypeInSSEReg(DstVT); |
Chris Lattner | a91f77e | 2008-01-24 08:07:48 +0000 | [diff] [blame] | 522 | if (SrcIsSSE && DstIsSSE) |
| 523 | continue; |
| 524 | |
Chris Lattner | d587e58 | 2008-03-09 07:05:32 +0000 | [diff] [blame] | 525 | if (!SrcIsSSE && !DstIsSSE) { |
| 526 | // If this is an FPStack extension, it is a noop. |
| 527 | if (N->getOpcode() == ISD::FP_EXTEND) |
| 528 | continue; |
| 529 | // If this is a value-preserving FPStack truncation, it is a noop. |
| 530 | if (N->getConstantOperandVal(1)) |
| 531 | continue; |
| 532 | } |
Chad Rosier | 24c19d2 | 2012-08-01 18:39:17 +0000 | [diff] [blame] | 533 | |
Chris Lattner | a91f77e | 2008-01-24 08:07:48 +0000 | [diff] [blame] | 534 | // Here we could have an FP stack truncation or an FPStack <-> SSE convert. |
| 535 | // FPStack has extload and truncstore. SSE can fold direct loads into other |
| 536 | // operations. Based on this, decide what we want to do. |
Craig Topper | 83e042a | 2013-08-15 05:57:07 +0000 | [diff] [blame] | 537 | MVT MemVT; |
Chris Lattner | a91f77e | 2008-01-24 08:07:48 +0000 | [diff] [blame] | 538 | if (N->getOpcode() == ISD::FP_ROUND) |
| 539 | MemVT = DstVT; // FP_ROUND must use DstVT, we can't do a 'trunc load'. |
| 540 | else |
| 541 | MemVT = SrcIsSSE ? SrcVT : DstVT; |
Chad Rosier | 24c19d2 | 2012-08-01 18:39:17 +0000 | [diff] [blame] | 542 | |
Dan Gohman | eb0cee9 | 2008-08-23 02:25:05 +0000 | [diff] [blame] | 543 | SDValue MemTmp = CurDAG->CreateStackTemporary(MemVT); |
Andrew Trick | ef9de2a | 2013-05-25 02:42:55 +0000 | [diff] [blame] | 544 | SDLoc dl(N); |
Chad Rosier | 24c19d2 | 2012-08-01 18:39:17 +0000 | [diff] [blame] | 545 | |
Chris Lattner | a91f77e | 2008-01-24 08:07:48 +0000 | [diff] [blame] | 546 | // FIXME: optimize the case where the src/dest is a load or store? |
Dale Johannesen | 14f2d9d | 2009-02-03 21:48:12 +0000 | [diff] [blame] | 547 | SDValue Store = CurDAG->getTruncStore(CurDAG->getEntryNode(), dl, |
Dan Gohman | eb0cee9 | 2008-08-23 02:25:05 +0000 | [diff] [blame] | 548 | N->getOperand(0), |
Chris Lattner | 3d178ed | 2010-09-21 17:04:51 +0000 | [diff] [blame] | 549 | MemTmp, MachinePointerInfo(), MemVT, |
David Greene | cbd39c5 | 2010-02-15 16:57:43 +0000 | [diff] [blame] | 550 | false, false, 0); |
Stuart Hastings | 81c4306 | 2011-02-16 16:23:55 +0000 | [diff] [blame] | 551 | SDValue Result = CurDAG->getExtLoad(ISD::EXTLOAD, dl, DstVT, Store, MemTmp, |
Chris Lattner | 3d178ed | 2010-09-21 17:04:51 +0000 | [diff] [blame] | 552 | MachinePointerInfo(), |
Louis Gerbarg | 67474e3 | 2014-07-31 21:45:05 +0000 | [diff] [blame] | 553 | MemVT, false, false, false, 0); |
Chris Lattner | a91f77e | 2008-01-24 08:07:48 +0000 | [diff] [blame] | 554 | |
| 555 | // We're about to replace all uses of the FP_ROUND/FP_EXTEND with the |
| 556 | // extload we created. This will cause general havok on the dag because |
| 557 | // anything below the conversion could be folded into other existing nodes. |
| 558 | // To avoid invalidating 'I', back it up to the convert node. |
| 559 | --I; |
Dan Gohman | eb0cee9 | 2008-08-23 02:25:05 +0000 | [diff] [blame] | 560 | CurDAG->ReplaceAllUsesOfValueWith(SDValue(N, 0), Result); |
Chad Rosier | 24c19d2 | 2012-08-01 18:39:17 +0000 | [diff] [blame] | 561 | |
Chris Lattner | a91f77e | 2008-01-24 08:07:48 +0000 | [diff] [blame] | 562 | // Now that we did that, the node is dead. Increment the iterator to the |
| 563 | // next node to process, then delete N. |
| 564 | ++I; |
Dan Gohman | eb0cee9 | 2008-08-23 02:25:05 +0000 | [diff] [blame] | 565 | CurDAG->DeleteNode(N); |
Chad Rosier | 24c19d2 | 2012-08-01 18:39:17 +0000 | [diff] [blame] | 566 | } |
Chris Lattner | a91f77e | 2008-01-24 08:07:48 +0000 | [diff] [blame] | 567 | } |
| 568 | |
Chris Lattner | 655e7df | 2005-11-16 01:54:32 +0000 | [diff] [blame] | 569 | |
Anton Korobeynikov | 9091074 | 2007-09-25 21:52:30 +0000 | [diff] [blame] | 570 | /// EmitSpecialCodeForMain - Emit any code that needs to be executed only in |
| 571 | /// the main function. |
David Majnemer | d5ab35f | 2015-02-21 05:49:45 +0000 | [diff] [blame] | 572 | void X86DAGToDAGISel::EmitSpecialCodeForMain() { |
Bill Wendling | 81d4071 | 2011-01-06 00:47:10 +0000 | [diff] [blame] | 573 | if (Subtarget->isTargetCygMing()) { |
David Majnemer | d5ab35f | 2015-02-21 05:49:45 +0000 | [diff] [blame] | 574 | TargetLowering::ArgListTy Args; |
| 575 | |
| 576 | TargetLowering::CallLoweringInfo CLI(*CurDAG); |
| 577 | CLI.setChain(CurDAG->getRoot()) |
| 578 | .setCallee(CallingConv::C, Type::getVoidTy(*CurDAG->getContext()), |
| 579 | CurDAG->getExternalSymbol("__main", TLI->getPointerTy()), |
| 580 | std::move(Args), 0); |
| 581 | const TargetLowering &TLI = CurDAG->getTargetLoweringInfo(); |
| 582 | std::pair<SDValue, SDValue> Result = TLI.LowerCallTo(CLI); |
| 583 | CurDAG->setRoot(Result.second); |
Bill Wendling | 81d4071 | 2011-01-06 00:47:10 +0000 | [diff] [blame] | 584 | } |
Anton Korobeynikov | 9091074 | 2007-09-25 21:52:30 +0000 | [diff] [blame] | 585 | } |
| 586 | |
Dan Gohman | c87b74d | 2010-04-14 20:17:22 +0000 | [diff] [blame] | 587 | void X86DAGToDAGISel::EmitFunctionEntryCode() { |
Anton Korobeynikov | 9091074 | 2007-09-25 21:52:30 +0000 | [diff] [blame] | 588 | // If this is main, emit special code for main. |
Dan Gohman | c87b74d | 2010-04-14 20:17:22 +0000 | [diff] [blame] | 589 | if (const Function *Fn = MF->getFunction()) |
| 590 | if (Fn->hasExternalLinkage() && Fn->getName() == "main") |
David Majnemer | d5ab35f | 2015-02-21 05:49:45 +0000 | [diff] [blame] | 591 | EmitSpecialCodeForMain(); |
Anton Korobeynikov | 9091074 | 2007-09-25 21:52:30 +0000 | [diff] [blame] | 592 | } |
| 593 | |
Eli Friedman | 344ec79 | 2011-07-13 21:29:53 +0000 | [diff] [blame] | 594 | static bool isDispSafeForFrameIndex(int64_t Val) { |
| 595 | // On 64-bit platforms, we can run into an issue where a frame index |
| 596 | // includes a displacement that, when added to the explicit displacement, |
| 597 | // will overflow the displacement field. Assuming that the frame index |
| 598 | // displacement fits into a 31-bit integer (which is only slightly more |
| 599 | // aggressive than the current fundamental assumption that it fits into |
| 600 | // a 32-bit integer), a 31-bit disp should always be safe. |
| 601 | return isInt<31>(Val); |
| 602 | } |
| 603 | |
Eli Friedman | ef67e7d | 2011-07-13 20:44:23 +0000 | [diff] [blame] | 604 | bool X86DAGToDAGISel::FoldOffsetIntoAddress(uint64_t Offset, |
| 605 | X86ISelAddressMode &AM) { |
| 606 | int64_t Val = AM.Disp + Offset; |
| 607 | CodeModel::Model M = TM.getCodeModel(); |
Eli Friedman | 344ec79 | 2011-07-13 21:29:53 +0000 | [diff] [blame] | 608 | if (Subtarget->is64Bit()) { |
| 609 | if (!X86::isOffsetSuitableForCodeModel(Val, M, |
| 610 | AM.hasSymbolicDisplacement())) |
| 611 | return true; |
| 612 | // In addition to the checks required for a register base, check that |
| 613 | // we do not try to use an unsafe Disp with a frame index. |
| 614 | if (AM.BaseType == X86ISelAddressMode::FrameIndexBase && |
| 615 | !isDispSafeForFrameIndex(Val)) |
| 616 | return true; |
Eli Friedman | ef67e7d | 2011-07-13 20:44:23 +0000 | [diff] [blame] | 617 | } |
Eli Friedman | 344ec79 | 2011-07-13 21:29:53 +0000 | [diff] [blame] | 618 | AM.Disp = Val; |
| 619 | return false; |
| 620 | |
Eli Friedman | ef67e7d | 2011-07-13 20:44:23 +0000 | [diff] [blame] | 621 | } |
Rafael Espindola | 3b2df10 | 2009-04-08 21:14:34 +0000 | [diff] [blame] | 622 | |
Chris Lattner | 8a236b6 | 2010-09-22 04:39:11 +0000 | [diff] [blame] | 623 | bool X86DAGToDAGISel::MatchLoadInAddress(LoadSDNode *N, X86ISelAddressMode &AM){ |
| 624 | SDValue Address = N->getOperand(1); |
Chad Rosier | 24c19d2 | 2012-08-01 18:39:17 +0000 | [diff] [blame] | 625 | |
Chris Lattner | 8a236b6 | 2010-09-22 04:39:11 +0000 | [diff] [blame] | 626 | // load gs:0 -> GS segment register. |
| 627 | // load fs:0 -> FS segment register. |
| 628 | // |
Rafael Espindola | 3b2df10 | 2009-04-08 21:14:34 +0000 | [diff] [blame] | 629 | // This optimization is valid because the GNU TLS model defines that |
| 630 | // gs:0 (or fs:0 on X86-64) contains its own address. |
| 631 | // For more information see http://people.redhat.com/drepper/tls.pdf |
Chris Lattner | 8a236b6 | 2010-09-22 04:39:11 +0000 | [diff] [blame] | 632 | if (ConstantSDNode *C = dyn_cast<ConstantSDNode>(Address)) |
Craig Topper | 062a2ba | 2014-04-25 05:30:21 +0000 | [diff] [blame] | 633 | if (C->getSExtValue() == 0 && AM.Segment.getNode() == nullptr && |
David Chisnall | 5b8c168 | 2012-07-24 20:04:16 +0000 | [diff] [blame] | 634 | Subtarget->isTargetLinux()) |
Chris Lattner | 8a236b6 | 2010-09-22 04:39:11 +0000 | [diff] [blame] | 635 | switch (N->getPointerInfo().getAddrSpace()) { |
| 636 | case 256: |
| 637 | AM.Segment = CurDAG->getRegister(X86::GS, MVT::i16); |
| 638 | return false; |
| 639 | case 257: |
| 640 | AM.Segment = CurDAG->getRegister(X86::FS, MVT::i16); |
| 641 | return false; |
| 642 | } |
Chad Rosier | 24c19d2 | 2012-08-01 18:39:17 +0000 | [diff] [blame] | 643 | |
Rafael Espindola | 3b2df10 | 2009-04-08 21:14:34 +0000 | [diff] [blame] | 644 | return true; |
| 645 | } |
| 646 | |
Chris Lattner | fea81da | 2009-06-27 04:16:01 +0000 | [diff] [blame] | 647 | /// MatchWrapper - Try to match X86ISD::Wrapper and X86ISD::WrapperRIP nodes |
| 648 | /// into an addressing mode. These wrap things that will resolve down into a |
| 649 | /// symbol reference. If no match is possible, this returns true, otherwise it |
Anton Korobeynikov | 741ea0d | 2009-08-05 23:01:26 +0000 | [diff] [blame] | 650 | /// returns false. |
Rafael Espindola | 6688b0a | 2009-04-12 21:55:03 +0000 | [diff] [blame] | 651 | bool X86DAGToDAGISel::MatchWrapper(SDValue N, X86ISelAddressMode &AM) { |
Chris Lattner | fea81da | 2009-06-27 04:16:01 +0000 | [diff] [blame] | 652 | // If the addressing mode already has a symbol as the displacement, we can |
| 653 | // never match another symbol. |
Rafael Espindola | 6688b0a | 2009-04-12 21:55:03 +0000 | [diff] [blame] | 654 | if (AM.hasSymbolicDisplacement()) |
| 655 | return true; |
Rafael Espindola | 6688b0a | 2009-04-12 21:55:03 +0000 | [diff] [blame] | 656 | |
| 657 | SDValue N0 = N.getOperand(0); |
Anton Korobeynikov | 741ea0d | 2009-08-05 23:01:26 +0000 | [diff] [blame] | 658 | CodeModel::Model M = TM.getCodeModel(); |
| 659 | |
Chris Lattner | fea81da | 2009-06-27 04:16:01 +0000 | [diff] [blame] | 660 | // Handle X86-64 rip-relative addresses. We check this before checking direct |
| 661 | // folding because RIP is preferable to non-RIP accesses. |
Chandler Carruth | 3779ac1 | 2012-04-09 02:13:06 +0000 | [diff] [blame] | 662 | if (Subtarget->is64Bit() && N.getOpcode() == X86ISD::WrapperRIP && |
Chris Lattner | fea81da | 2009-06-27 04:16:01 +0000 | [diff] [blame] | 663 | // Under X86-64 non-small code model, GV (and friends) are 64-bits, so |
| 664 | // they cannot be folded into immediate fields. |
| 665 | // FIXME: This can be improved for kernel and other models? |
Chandler Carruth | 3779ac1 | 2012-04-09 02:13:06 +0000 | [diff] [blame] | 666 | (M == CodeModel::Small || M == CodeModel::Kernel)) { |
| 667 | // Base and index reg must be 0 in order to use %rip as base. |
| 668 | if (AM.hasBaseOrIndexReg()) |
| 669 | return true; |
Chris Lattner | fea81da | 2009-06-27 04:16:01 +0000 | [diff] [blame] | 670 | if (GlobalAddressSDNode *G = dyn_cast<GlobalAddressSDNode>(N0)) { |
Eli Friedman | ef67e7d | 2011-07-13 20:44:23 +0000 | [diff] [blame] | 671 | X86ISelAddressMode Backup = AM; |
Chris Lattner | fea81da | 2009-06-27 04:16:01 +0000 | [diff] [blame] | 672 | AM.GV = G->getGlobal(); |
Chris Lattner | bd7e26d | 2009-06-26 05:51:45 +0000 | [diff] [blame] | 673 | AM.SymbolFlags = G->getTargetFlags(); |
Eli Friedman | ef67e7d | 2011-07-13 20:44:23 +0000 | [diff] [blame] | 674 | if (FoldOffsetIntoAddress(G->getOffset(), AM)) { |
| 675 | AM = Backup; |
| 676 | return true; |
| 677 | } |
Chris Lattner | fea81da | 2009-06-27 04:16:01 +0000 | [diff] [blame] | 678 | } else if (ConstantPoolSDNode *CP = dyn_cast<ConstantPoolSDNode>(N0)) { |
Eli Friedman | ef67e7d | 2011-07-13 20:44:23 +0000 | [diff] [blame] | 679 | X86ISelAddressMode Backup = AM; |
Rafael Espindola | 6688b0a | 2009-04-12 21:55:03 +0000 | [diff] [blame] | 680 | AM.CP = CP->getConstVal(); |
| 681 | AM.Align = CP->getAlignment(); |
Chris Lattner | 1d3b65a | 2009-06-26 05:56:49 +0000 | [diff] [blame] | 682 | AM.SymbolFlags = CP->getTargetFlags(); |
Eli Friedman | ef67e7d | 2011-07-13 20:44:23 +0000 | [diff] [blame] | 683 | if (FoldOffsetIntoAddress(CP->getOffset(), AM)) { |
| 684 | AM = Backup; |
| 685 | return true; |
| 686 | } |
Chris Lattner | fea81da | 2009-06-27 04:16:01 +0000 | [diff] [blame] | 687 | } else if (ExternalSymbolSDNode *S = dyn_cast<ExternalSymbolSDNode>(N0)) { |
| 688 | AM.ES = S->getSymbol(); |
| 689 | AM.SymbolFlags = S->getTargetFlags(); |
Chris Lattner | 50ba5c3 | 2009-11-01 03:25:03 +0000 | [diff] [blame] | 690 | } else if (JumpTableSDNode *J = dyn_cast<JumpTableSDNode>(N0)) { |
Chris Lattner | fea81da | 2009-06-27 04:16:01 +0000 | [diff] [blame] | 691 | AM.JT = J->getIndex(); |
| 692 | AM.SymbolFlags = J->getTargetFlags(); |
Michael Liao | abb87d4 | 2012-09-12 21:43:09 +0000 | [diff] [blame] | 693 | } else if (BlockAddressSDNode *BA = dyn_cast<BlockAddressSDNode>(N0)) { |
| 694 | X86ISelAddressMode Backup = AM; |
| 695 | AM.BlockAddr = BA->getBlockAddress(); |
| 696 | AM.SymbolFlags = BA->getTargetFlags(); |
| 697 | if (FoldOffsetIntoAddress(BA->getOffset(), AM)) { |
| 698 | AM = Backup; |
| 699 | return true; |
| 700 | } |
| 701 | } else |
| 702 | llvm_unreachable("Unhandled symbol reference node."); |
Anton Korobeynikov | 741ea0d | 2009-08-05 23:01:26 +0000 | [diff] [blame] | 703 | |
Chris Lattner | fea81da | 2009-06-27 04:16:01 +0000 | [diff] [blame] | 704 | if (N.getOpcode() == X86ISD::WrapperRIP) |
Owen Anderson | 9f94459 | 2009-08-11 20:47:22 +0000 | [diff] [blame] | 705 | AM.setBaseReg(CurDAG->getRegister(X86::RIP, MVT::i64)); |
Rafael Espindola | 6688b0a | 2009-04-12 21:55:03 +0000 | [diff] [blame] | 706 | return false; |
Chris Lattner | fea81da | 2009-06-27 04:16:01 +0000 | [diff] [blame] | 707 | } |
| 708 | |
| 709 | // Handle the case when globals fit in our immediate field: This is true for |
Chandler Carruth | 3779ac1 | 2012-04-09 02:13:06 +0000 | [diff] [blame] | 710 | // X86-32 always and X86-64 when in -mcmodel=small mode. In 64-bit |
| 711 | // mode, this only applies to a non-RIP-relative computation. |
Chris Lattner | fea81da | 2009-06-27 04:16:01 +0000 | [diff] [blame] | 712 | if (!Subtarget->is64Bit() || |
Chandler Carruth | 3779ac1 | 2012-04-09 02:13:06 +0000 | [diff] [blame] | 713 | M == CodeModel::Small || M == CodeModel::Kernel) { |
| 714 | assert(N.getOpcode() != X86ISD::WrapperRIP && |
| 715 | "RIP-relative addressing already handled"); |
Chris Lattner | fea81da | 2009-06-27 04:16:01 +0000 | [diff] [blame] | 716 | if (GlobalAddressSDNode *G = dyn_cast<GlobalAddressSDNode>(N0)) { |
| 717 | AM.GV = G->getGlobal(); |
| 718 | AM.Disp += G->getOffset(); |
| 719 | AM.SymbolFlags = G->getTargetFlags(); |
| 720 | } else if (ConstantPoolSDNode *CP = dyn_cast<ConstantPoolSDNode>(N0)) { |
| 721 | AM.CP = CP->getConstVal(); |
| 722 | AM.Align = CP->getAlignment(); |
| 723 | AM.Disp += CP->getOffset(); |
| 724 | AM.SymbolFlags = CP->getTargetFlags(); |
| 725 | } else if (ExternalSymbolSDNode *S = dyn_cast<ExternalSymbolSDNode>(N0)) { |
| 726 | AM.ES = S->getSymbol(); |
| 727 | AM.SymbolFlags = S->getTargetFlags(); |
Chris Lattner | 50ba5c3 | 2009-11-01 03:25:03 +0000 | [diff] [blame] | 728 | } else if (JumpTableSDNode *J = dyn_cast<JumpTableSDNode>(N0)) { |
Chris Lattner | fea81da | 2009-06-27 04:16:01 +0000 | [diff] [blame] | 729 | AM.JT = J->getIndex(); |
| 730 | AM.SymbolFlags = J->getTargetFlags(); |
Michael Liao | abb87d4 | 2012-09-12 21:43:09 +0000 | [diff] [blame] | 731 | } else if (BlockAddressSDNode *BA = dyn_cast<BlockAddressSDNode>(N0)) { |
| 732 | AM.BlockAddr = BA->getBlockAddress(); |
| 733 | AM.Disp += BA->getOffset(); |
| 734 | AM.SymbolFlags = BA->getTargetFlags(); |
| 735 | } else |
| 736 | llvm_unreachable("Unhandled symbol reference node."); |
Rafael Espindola | 6688b0a | 2009-04-12 21:55:03 +0000 | [diff] [blame] | 737 | return false; |
| 738 | } |
| 739 | |
| 740 | return true; |
| 741 | } |
| 742 | |
Chris Lattner | 3f0f71b | 2005-11-19 02:11:08 +0000 | [diff] [blame] | 743 | /// MatchAddress - Add the specified node to the specified addressing mode, |
| 744 | /// returning true if it cannot be done. This just pattern matches for the |
Chris Lattner | ff87f05e | 2007-12-08 07:22:58 +0000 | [diff] [blame] | 745 | /// addressing mode. |
Dan Gohman | 824ab40 | 2009-07-22 23:26:55 +0000 | [diff] [blame] | 746 | bool X86DAGToDAGISel::MatchAddress(SDValue N, X86ISelAddressMode &AM) { |
Dan Gohman | 99ba4da | 2010-06-18 01:24:29 +0000 | [diff] [blame] | 747 | if (MatchAddressRecursively(N, AM, 0)) |
Dan Gohman | 824ab40 | 2009-07-22 23:26:55 +0000 | [diff] [blame] | 748 | return true; |
| 749 | |
| 750 | // Post-processing: Convert lea(,%reg,2) to lea(%reg,%reg), which has |
| 751 | // a smaller encoding and avoids a scaled-index. |
| 752 | if (AM.Scale == 2 && |
| 753 | AM.BaseType == X86ISelAddressMode::RegBase && |
Craig Topper | 062a2ba | 2014-04-25 05:30:21 +0000 | [diff] [blame] | 754 | AM.Base_Reg.getNode() == nullptr) { |
Dan Gohman | 0fd54fb | 2010-04-29 23:30:41 +0000 | [diff] [blame] | 755 | AM.Base_Reg = AM.IndexReg; |
Dan Gohman | 824ab40 | 2009-07-22 23:26:55 +0000 | [diff] [blame] | 756 | AM.Scale = 1; |
| 757 | } |
| 758 | |
Dan Gohman | 0504608 | 2009-08-20 18:23:44 +0000 | [diff] [blame] | 759 | // Post-processing: Convert foo to foo(%rip), even in non-PIC mode, |
| 760 | // because it has a smaller encoding. |
| 761 | // TODO: Which other code models can use this? |
| 762 | if (TM.getCodeModel() == CodeModel::Small && |
| 763 | Subtarget->is64Bit() && |
| 764 | AM.Scale == 1 && |
| 765 | AM.BaseType == X86ISelAddressMode::RegBase && |
Craig Topper | 062a2ba | 2014-04-25 05:30:21 +0000 | [diff] [blame] | 766 | AM.Base_Reg.getNode() == nullptr && |
| 767 | AM.IndexReg.getNode() == nullptr && |
Dan Gohman | 0f6bf2d | 2009-08-25 17:47:44 +0000 | [diff] [blame] | 768 | AM.SymbolFlags == X86II::MO_NO_FLAG && |
Dan Gohman | 0504608 | 2009-08-20 18:23:44 +0000 | [diff] [blame] | 769 | AM.hasSymbolicDisplacement()) |
Dan Gohman | 0fd54fb | 2010-04-29 23:30:41 +0000 | [diff] [blame] | 770 | AM.Base_Reg = CurDAG->getRegister(X86::RIP, MVT::i64); |
Dan Gohman | 0504608 | 2009-08-20 18:23:44 +0000 | [diff] [blame] | 771 | |
Dan Gohman | 824ab40 | 2009-07-22 23:26:55 +0000 | [diff] [blame] | 772 | return false; |
| 773 | } |
| 774 | |
Chandler Carruth | 3eacfb8 | 2012-01-11 11:04:36 +0000 | [diff] [blame] | 775 | // Insert a node into the DAG at least before the Pos node's position. This |
| 776 | // will reposition the node as needed, and will assign it a node ID that is <= |
| 777 | // the Pos node's ID. Note that this does *not* preserve the uniqueness of node |
| 778 | // IDs! The selection DAG must no longer depend on their uniqueness when this |
| 779 | // is used. |
| 780 | static void InsertDAGNode(SelectionDAG &DAG, SDValue Pos, SDValue N) { |
| 781 | if (N.getNode()->getNodeId() == -1 || |
| 782 | N.getNode()->getNodeId() > Pos.getNode()->getNodeId()) { |
| 783 | DAG.RepositionNode(Pos.getNode(), N.getNode()); |
| 784 | N.getNode()->setNodeId(Pos.getNode()->getNodeId()); |
| 785 | } |
| 786 | } |
| 787 | |
Adam Nemet | 0c7caf4 | 2014-09-16 17:14:10 +0000 | [diff] [blame] | 788 | // Transform "(X >> (8-C1)) & (0xff << C1)" to "((X >> 8) & 0xff) << C1" if |
| 789 | // safe. This allows us to convert the shift and and into an h-register |
| 790 | // extract and a scaled index. Returns false if the simplification is |
| 791 | // performed. |
Chandler Carruth | 51d3076 | 2012-01-11 08:48:20 +0000 | [diff] [blame] | 792 | static bool FoldMaskAndShiftToExtract(SelectionDAG &DAG, SDValue N, |
| 793 | uint64_t Mask, |
| 794 | SDValue Shift, SDValue X, |
| 795 | X86ISelAddressMode &AM) { |
| 796 | if (Shift.getOpcode() != ISD::SRL || |
| 797 | !isa<ConstantSDNode>(Shift.getOperand(1)) || |
| 798 | !Shift.hasOneUse()) |
| 799 | return true; |
| 800 | |
| 801 | int ScaleLog = 8 - Shift.getConstantOperandVal(1); |
| 802 | if (ScaleLog <= 0 || ScaleLog >= 4 || |
| 803 | Mask != (0xffu << ScaleLog)) |
| 804 | return true; |
| 805 | |
Craig Topper | 83e042a | 2013-08-15 05:57:07 +0000 | [diff] [blame] | 806 | MVT VT = N.getSimpleValueType(); |
Andrew Trick | ef9de2a | 2013-05-25 02:42:55 +0000 | [diff] [blame] | 807 | SDLoc DL(N); |
Sergey Dmitrouk | 842a51b | 2015-04-28 14:05:47 +0000 | [diff] [blame] | 808 | SDValue Eight = DAG.getConstant(8, DL, MVT::i8); |
| 809 | SDValue NewMask = DAG.getConstant(0xff, DL, VT); |
Chandler Carruth | 51d3076 | 2012-01-11 08:48:20 +0000 | [diff] [blame] | 810 | SDValue Srl = DAG.getNode(ISD::SRL, DL, VT, X, Eight); |
| 811 | SDValue And = DAG.getNode(ISD::AND, DL, VT, Srl, NewMask); |
Sergey Dmitrouk | 842a51b | 2015-04-28 14:05:47 +0000 | [diff] [blame] | 812 | SDValue ShlCount = DAG.getConstant(ScaleLog, DL, MVT::i8); |
Chandler Carruth | 51d3076 | 2012-01-11 08:48:20 +0000 | [diff] [blame] | 813 | SDValue Shl = DAG.getNode(ISD::SHL, DL, VT, And, ShlCount); |
| 814 | |
Chandler Carruth | eb21da0 | 2012-01-12 01:34:44 +0000 | [diff] [blame] | 815 | // Insert the new nodes into the topological ordering. We must do this in |
| 816 | // a valid topological ordering as nothing is going to go back and re-sort |
| 817 | // these nodes. We continually insert before 'N' in sequence as this is |
| 818 | // essentially a pre-flattened and pre-sorted sequence of nodes. There is no |
| 819 | // hierarchy left to express. |
| 820 | InsertDAGNode(DAG, N, Eight); |
| 821 | InsertDAGNode(DAG, N, Srl); |
| 822 | InsertDAGNode(DAG, N, NewMask); |
Chandler Carruth | 3eacfb8 | 2012-01-11 11:04:36 +0000 | [diff] [blame] | 823 | InsertDAGNode(DAG, N, And); |
Chandler Carruth | eb21da0 | 2012-01-12 01:34:44 +0000 | [diff] [blame] | 824 | InsertDAGNode(DAG, N, ShlCount); |
Chandler Carruth | 3eacfb8 | 2012-01-11 11:04:36 +0000 | [diff] [blame] | 825 | InsertDAGNode(DAG, N, Shl); |
Chandler Carruth | 51d3076 | 2012-01-11 08:48:20 +0000 | [diff] [blame] | 826 | DAG.ReplaceAllUsesWith(N, Shl); |
| 827 | AM.IndexReg = And; |
| 828 | AM.Scale = (1 << ScaleLog); |
| 829 | return false; |
| 830 | } |
| 831 | |
Chandler Carruth | aa01e66 | 2012-01-11 09:35:00 +0000 | [diff] [blame] | 832 | // Transforms "(X << C1) & C2" to "(X & (C2>>C1)) << C1" if safe and if this |
| 833 | // allows us to fold the shift into this addressing mode. Returns false if the |
| 834 | // transform succeeded. |
| 835 | static bool FoldMaskedShiftToScaledMask(SelectionDAG &DAG, SDValue N, |
| 836 | uint64_t Mask, |
| 837 | SDValue Shift, SDValue X, |
| 838 | X86ISelAddressMode &AM) { |
| 839 | if (Shift.getOpcode() != ISD::SHL || |
| 840 | !isa<ConstantSDNode>(Shift.getOperand(1))) |
| 841 | return true; |
| 842 | |
| 843 | // Not likely to be profitable if either the AND or SHIFT node has more |
| 844 | // than one use (unless all uses are for address computation). Besides, |
| 845 | // isel mechanism requires their node ids to be reused. |
| 846 | if (!N.hasOneUse() || !Shift.hasOneUse()) |
| 847 | return true; |
| 848 | |
| 849 | // Verify that the shift amount is something we can fold. |
| 850 | unsigned ShiftAmt = Shift.getConstantOperandVal(1); |
| 851 | if (ShiftAmt != 1 && ShiftAmt != 2 && ShiftAmt != 3) |
| 852 | return true; |
| 853 | |
Craig Topper | 83e042a | 2013-08-15 05:57:07 +0000 | [diff] [blame] | 854 | MVT VT = N.getSimpleValueType(); |
Andrew Trick | ef9de2a | 2013-05-25 02:42:55 +0000 | [diff] [blame] | 855 | SDLoc DL(N); |
Sergey Dmitrouk | 842a51b | 2015-04-28 14:05:47 +0000 | [diff] [blame] | 856 | SDValue NewMask = DAG.getConstant(Mask >> ShiftAmt, DL, VT); |
Chandler Carruth | aa01e66 | 2012-01-11 09:35:00 +0000 | [diff] [blame] | 857 | SDValue NewAnd = DAG.getNode(ISD::AND, DL, VT, X, NewMask); |
| 858 | SDValue NewShift = DAG.getNode(ISD::SHL, DL, VT, NewAnd, Shift.getOperand(1)); |
| 859 | |
Chandler Carruth | eb21da0 | 2012-01-12 01:34:44 +0000 | [diff] [blame] | 860 | // Insert the new nodes into the topological ordering. We must do this in |
| 861 | // a valid topological ordering as nothing is going to go back and re-sort |
| 862 | // these nodes. We continually insert before 'N' in sequence as this is |
| 863 | // essentially a pre-flattened and pre-sorted sequence of nodes. There is no |
| 864 | // hierarchy left to express. |
| 865 | InsertDAGNode(DAG, N, NewMask); |
| 866 | InsertDAGNode(DAG, N, NewAnd); |
Chandler Carruth | 3eacfb8 | 2012-01-11 11:04:36 +0000 | [diff] [blame] | 867 | InsertDAGNode(DAG, N, NewShift); |
Chandler Carruth | aa01e66 | 2012-01-11 09:35:00 +0000 | [diff] [blame] | 868 | DAG.ReplaceAllUsesWith(N, NewShift); |
| 869 | |
| 870 | AM.Scale = 1 << ShiftAmt; |
| 871 | AM.IndexReg = NewAnd; |
| 872 | return false; |
| 873 | } |
| 874 | |
Chandler Carruth | 55b2cde | 2012-01-11 08:41:08 +0000 | [diff] [blame] | 875 | // Implement some heroics to detect shifts of masked values where the mask can |
| 876 | // be replaced by extending the shift and undoing that in the addressing mode |
| 877 | // scale. Patterns such as (shl (srl x, c1), c2) are canonicalized into (and |
| 878 | // (srl x, SHIFT), MASK) by DAGCombines that don't know the shl can be done in |
| 879 | // the addressing mode. This results in code such as: |
| 880 | // |
| 881 | // int f(short *y, int *lookup_table) { |
| 882 | // ... |
| 883 | // return *y + lookup_table[*y >> 11]; |
| 884 | // } |
| 885 | // |
| 886 | // Turning into: |
| 887 | // movzwl (%rdi), %eax |
| 888 | // movl %eax, %ecx |
| 889 | // shrl $11, %ecx |
| 890 | // addl (%rsi,%rcx,4), %eax |
| 891 | // |
| 892 | // Instead of: |
| 893 | // movzwl (%rdi), %eax |
| 894 | // movl %eax, %ecx |
| 895 | // shrl $9, %ecx |
| 896 | // andl $124, %rcx |
| 897 | // addl (%rsi,%rcx), %eax |
| 898 | // |
Chandler Carruth | 3dbcda8 | 2012-01-11 09:35:02 +0000 | [diff] [blame] | 899 | // Note that this function assumes the mask is provided as a mask *after* the |
| 900 | // value is shifted. The input chain may or may not match that, but computing |
| 901 | // such a mask is trivial. |
Chandler Carruth | 55b2cde | 2012-01-11 08:41:08 +0000 | [diff] [blame] | 902 | static bool FoldMaskAndShiftToScale(SelectionDAG &DAG, SDValue N, |
Chandler Carruth | 3dbcda8 | 2012-01-11 09:35:02 +0000 | [diff] [blame] | 903 | uint64_t Mask, |
| 904 | SDValue Shift, SDValue X, |
Chandler Carruth | 55b2cde | 2012-01-11 08:41:08 +0000 | [diff] [blame] | 905 | X86ISelAddressMode &AM) { |
Chandler Carruth | 3dbcda8 | 2012-01-11 09:35:02 +0000 | [diff] [blame] | 906 | if (Shift.getOpcode() != ISD::SRL || !Shift.hasOneUse() || |
| 907 | !isa<ConstantSDNode>(Shift.getOperand(1))) |
Chandler Carruth | 55b2cde | 2012-01-11 08:41:08 +0000 | [diff] [blame] | 908 | return true; |
Chandler Carruth | 55b2cde | 2012-01-11 08:41:08 +0000 | [diff] [blame] | 909 | |
Chandler Carruth | 55b2cde | 2012-01-11 08:41:08 +0000 | [diff] [blame] | 910 | unsigned ShiftAmt = Shift.getConstantOperandVal(1); |
Michael J. Spencer | df1ecbd7 | 2013-05-24 22:23:49 +0000 | [diff] [blame] | 911 | unsigned MaskLZ = countLeadingZeros(Mask); |
| 912 | unsigned MaskTZ = countTrailingZeros(Mask); |
Chandler Carruth | 55b2cde | 2012-01-11 08:41:08 +0000 | [diff] [blame] | 913 | |
| 914 | // The amount of shift we're trying to fit into the addressing mode is taken |
Chandler Carruth | 3dbcda8 | 2012-01-11 09:35:02 +0000 | [diff] [blame] | 915 | // from the trailing zeros of the mask. |
| 916 | unsigned AMShiftAmt = MaskTZ; |
Chandler Carruth | 55b2cde | 2012-01-11 08:41:08 +0000 | [diff] [blame] | 917 | |
| 918 | // There is nothing we can do here unless the mask is removing some bits. |
| 919 | // Also, the addressing mode can only represent shifts of 1, 2, or 3 bits. |
| 920 | if (AMShiftAmt <= 0 || AMShiftAmt > 3) return true; |
| 921 | |
| 922 | // We also need to ensure that mask is a continuous run of bits. |
Benjamin Kramer | 5f6a907 | 2015-02-12 15:35:40 +0000 | [diff] [blame] | 923 | if (countTrailingOnes(Mask >> MaskTZ) + MaskTZ + MaskLZ != 64) return true; |
Chandler Carruth | 55b2cde | 2012-01-11 08:41:08 +0000 | [diff] [blame] | 924 | |
| 925 | // Scale the leading zero count down based on the actual size of the value. |
Chandler Carruth | 3dbcda8 | 2012-01-11 09:35:02 +0000 | [diff] [blame] | 926 | // Also scale it down based on the size of the shift. |
Craig Topper | 83e042a | 2013-08-15 05:57:07 +0000 | [diff] [blame] | 927 | MaskLZ -= (64 - X.getSimpleValueType().getSizeInBits()) + ShiftAmt; |
Chandler Carruth | 55b2cde | 2012-01-11 08:41:08 +0000 | [diff] [blame] | 928 | |
| 929 | // The final check is to ensure that any masked out high bits of X are |
| 930 | // already known to be zero. Otherwise, the mask has a semantic impact |
| 931 | // other than masking out a couple of low bits. Unfortunately, because of |
| 932 | // the mask, zero extensions will be removed from operands in some cases. |
| 933 | // This code works extra hard to look through extensions because we can |
| 934 | // replace them with zero extensions cheaply if necessary. |
| 935 | bool ReplacingAnyExtend = false; |
| 936 | if (X.getOpcode() == ISD::ANY_EXTEND) { |
Craig Topper | 83e042a | 2013-08-15 05:57:07 +0000 | [diff] [blame] | 937 | unsigned ExtendBits = X.getSimpleValueType().getSizeInBits() - |
| 938 | X.getOperand(0).getSimpleValueType().getSizeInBits(); |
Chandler Carruth | 55b2cde | 2012-01-11 08:41:08 +0000 | [diff] [blame] | 939 | // Assume that we'll replace the any-extend with a zero-extend, and |
| 940 | // narrow the search to the extended value. |
| 941 | X = X.getOperand(0); |
| 942 | MaskLZ = ExtendBits > MaskLZ ? 0 : MaskLZ - ExtendBits; |
| 943 | ReplacingAnyExtend = true; |
| 944 | } |
Craig Topper | 83e042a | 2013-08-15 05:57:07 +0000 | [diff] [blame] | 945 | APInt MaskedHighBits = |
| 946 | APInt::getHighBitsSet(X.getSimpleValueType().getSizeInBits(), MaskLZ); |
Chandler Carruth | 55b2cde | 2012-01-11 08:41:08 +0000 | [diff] [blame] | 947 | APInt KnownZero, KnownOne; |
Jay Foad | a0653a3 | 2014-05-14 21:14:37 +0000 | [diff] [blame] | 948 | DAG.computeKnownBits(X, KnownZero, KnownOne); |
Chandler Carruth | 55b2cde | 2012-01-11 08:41:08 +0000 | [diff] [blame] | 949 | if (MaskedHighBits != KnownZero) return true; |
| 950 | |
| 951 | // We've identified a pattern that can be transformed into a single shift |
| 952 | // and an addressing mode. Make it so. |
Craig Topper | 83e042a | 2013-08-15 05:57:07 +0000 | [diff] [blame] | 953 | MVT VT = N.getSimpleValueType(); |
Chandler Carruth | 55b2cde | 2012-01-11 08:41:08 +0000 | [diff] [blame] | 954 | if (ReplacingAnyExtend) { |
| 955 | assert(X.getValueType() != VT); |
| 956 | // We looked through an ANY_EXTEND node, insert a ZERO_EXTEND. |
Andrew Trick | ef9de2a | 2013-05-25 02:42:55 +0000 | [diff] [blame] | 957 | SDValue NewX = DAG.getNode(ISD::ZERO_EXTEND, SDLoc(X), VT, X); |
Chandler Carruth | 3eacfb8 | 2012-01-11 11:04:36 +0000 | [diff] [blame] | 958 | InsertDAGNode(DAG, N, NewX); |
Chandler Carruth | 55b2cde | 2012-01-11 08:41:08 +0000 | [diff] [blame] | 959 | X = NewX; |
| 960 | } |
Andrew Trick | ef9de2a | 2013-05-25 02:42:55 +0000 | [diff] [blame] | 961 | SDLoc DL(N); |
Sergey Dmitrouk | 842a51b | 2015-04-28 14:05:47 +0000 | [diff] [blame] | 962 | SDValue NewSRLAmt = DAG.getConstant(ShiftAmt + AMShiftAmt, DL, MVT::i8); |
Chandler Carruth | 55b2cde | 2012-01-11 08:41:08 +0000 | [diff] [blame] | 963 | SDValue NewSRL = DAG.getNode(ISD::SRL, DL, VT, X, NewSRLAmt); |
Sergey Dmitrouk | 842a51b | 2015-04-28 14:05:47 +0000 | [diff] [blame] | 964 | SDValue NewSHLAmt = DAG.getConstant(AMShiftAmt, DL, MVT::i8); |
Chandler Carruth | 55b2cde | 2012-01-11 08:41:08 +0000 | [diff] [blame] | 965 | SDValue NewSHL = DAG.getNode(ISD::SHL, DL, VT, NewSRL, NewSHLAmt); |
Chandler Carruth | eb21da0 | 2012-01-12 01:34:44 +0000 | [diff] [blame] | 966 | |
| 967 | // Insert the new nodes into the topological ordering. We must do this in |
| 968 | // a valid topological ordering as nothing is going to go back and re-sort |
| 969 | // these nodes. We continually insert before 'N' in sequence as this is |
| 970 | // essentially a pre-flattened and pre-sorted sequence of nodes. There is no |
| 971 | // hierarchy left to express. |
Chandler Carruth | 3eacfb8 | 2012-01-11 11:04:36 +0000 | [diff] [blame] | 972 | InsertDAGNode(DAG, N, NewSRLAmt); |
| 973 | InsertDAGNode(DAG, N, NewSRL); |
| 974 | InsertDAGNode(DAG, N, NewSHLAmt); |
| 975 | InsertDAGNode(DAG, N, NewSHL); |
Chandler Carruth | 55b2cde | 2012-01-11 08:41:08 +0000 | [diff] [blame] | 976 | DAG.ReplaceAllUsesWith(N, NewSHL); |
| 977 | |
| 978 | AM.Scale = 1 << AMShiftAmt; |
| 979 | AM.IndexReg = NewSRL; |
| 980 | return false; |
| 981 | } |
| 982 | |
Dan Gohman | 824ab40 | 2009-07-22 23:26:55 +0000 | [diff] [blame] | 983 | bool X86DAGToDAGISel::MatchAddressRecursively(SDValue N, X86ISelAddressMode &AM, |
| 984 | unsigned Depth) { |
Andrew Trick | ef9de2a | 2013-05-25 02:42:55 +0000 | [diff] [blame] | 985 | SDLoc dl(N); |
Bill Wendling | fe3bdb4 | 2009-08-07 21:33:25 +0000 | [diff] [blame] | 986 | DEBUG({ |
David Greene | dbdb1b2 | 2010-01-05 01:29:08 +0000 | [diff] [blame] | 987 | dbgs() << "MatchAddress: "; |
Bill Wendling | fe3bdb4 | 2009-08-07 21:33:25 +0000 | [diff] [blame] | 988 | AM.dump(); |
| 989 | }); |
Dan Gohman | ccb3611 | 2007-08-13 20:03:06 +0000 | [diff] [blame] | 990 | // Limit recursion. |
| 991 | if (Depth > 5) |
Rafael Espindola | 9277379 | 2009-03-31 16:16:57 +0000 | [diff] [blame] | 992 | return MatchAddressBase(N, AM); |
Anton Korobeynikov | 741ea0d | 2009-08-05 23:01:26 +0000 | [diff] [blame] | 993 | |
Chris Lattner | fea81da | 2009-06-27 04:16:01 +0000 | [diff] [blame] | 994 | // If this is already a %rip relative address, we can only merge immediates |
| 995 | // into it. Instead of handling this in every case, we handle it here. |
Evan Cheng | 11b0a5d | 2006-09-08 06:48:29 +0000 | [diff] [blame] | 996 | // RIP relative addressing: %rip + 32-bit displacement! |
Chris Lattner | fea81da | 2009-06-27 04:16:01 +0000 | [diff] [blame] | 997 | if (AM.isRIPRelative()) { |
| 998 | // FIXME: JumpTable and ExternalSymbol address currently don't like |
| 999 | // displacements. It isn't very important, but this should be fixed for |
| 1000 | // consistency. |
| 1001 | if (!AM.ES && AM.JT != -1) return true; |
Anton Korobeynikov | 741ea0d | 2009-08-05 23:01:26 +0000 | [diff] [blame] | 1002 | |
Eli Friedman | ef67e7d | 2011-07-13 20:44:23 +0000 | [diff] [blame] | 1003 | if (ConstantSDNode *Cst = dyn_cast<ConstantSDNode>(N)) |
| 1004 | if (!FoldOffsetIntoAddress(Cst->getSExtValue(), AM)) |
Evan Cheng | 11b0a5d | 2006-09-08 06:48:29 +0000 | [diff] [blame] | 1005 | return false; |
Evan Cheng | 11b0a5d | 2006-09-08 06:48:29 +0000 | [diff] [blame] | 1006 | return true; |
| 1007 | } |
| 1008 | |
Chris Lattner | 3f0f71b | 2005-11-19 02:11:08 +0000 | [diff] [blame] | 1009 | switch (N.getOpcode()) { |
| 1010 | default: break; |
David Majnemer | 71b9b6b | 2015-03-05 18:50:12 +0000 | [diff] [blame] | 1011 | case ISD::FRAME_ALLOC_RECOVER: { |
| 1012 | if (!AM.hasSymbolicDisplacement()) |
| 1013 | if (const auto *ESNode = dyn_cast<ExternalSymbolSDNode>(N.getOperand(0))) |
| 1014 | if (ESNode->getOpcode() == ISD::TargetExternalSymbol) { |
Reid Kleckner | c695471 | 2015-04-29 16:46:01 +0000 | [diff] [blame] | 1015 | // Use the symbol and don't prefix it. |
David Majnemer | 71b9b6b | 2015-03-05 18:50:12 +0000 | [diff] [blame] | 1016 | AM.ES = ESNode->getSymbol(); |
Reid Kleckner | c695471 | 2015-04-29 16:46:01 +0000 | [diff] [blame] | 1017 | AM.SymbolFlags = X86II::MO_NOPREFIX; |
David Majnemer | 71b9b6b | 2015-03-05 18:50:12 +0000 | [diff] [blame] | 1018 | return false; |
| 1019 | } |
| 1020 | break; |
| 1021 | } |
Evan Cheng | 11b0a5d | 2006-09-08 06:48:29 +0000 | [diff] [blame] | 1022 | case ISD::Constant: { |
Dan Gohman | 059c4fa | 2008-11-11 15:52:29 +0000 | [diff] [blame] | 1023 | uint64_t Val = cast<ConstantSDNode>(N)->getSExtValue(); |
Eli Friedman | ef67e7d | 2011-07-13 20:44:23 +0000 | [diff] [blame] | 1024 | if (!FoldOffsetIntoAddress(Val, AM)) |
Evan Cheng | 11b0a5d | 2006-09-08 06:48:29 +0000 | [diff] [blame] | 1025 | return false; |
Evan Cheng | 11b0a5d | 2006-09-08 06:48:29 +0000 | [diff] [blame] | 1026 | break; |
| 1027 | } |
Evan Cheng | 77d86ff | 2006-02-25 10:09:08 +0000 | [diff] [blame] | 1028 | |
Rafael Espindola | 6688b0a | 2009-04-12 21:55:03 +0000 | [diff] [blame] | 1029 | case X86ISD::Wrapper: |
Chris Lattner | fea81da | 2009-06-27 04:16:01 +0000 | [diff] [blame] | 1030 | case X86ISD::WrapperRIP: |
Rafael Espindola | 6688b0a | 2009-04-12 21:55:03 +0000 | [diff] [blame] | 1031 | if (!MatchWrapper(N, AM)) |
| 1032 | return false; |
Evan Cheng | 77d86ff | 2006-02-25 10:09:08 +0000 | [diff] [blame] | 1033 | break; |
| 1034 | |
Rafael Espindola | 3b2df10 | 2009-04-08 21:14:34 +0000 | [diff] [blame] | 1035 | case ISD::LOAD: |
Chris Lattner | 8a236b6 | 2010-09-22 04:39:11 +0000 | [diff] [blame] | 1036 | if (!MatchLoadInAddress(cast<LoadSDNode>(N), AM)) |
Rafael Espindola | 3b2df10 | 2009-04-08 21:14:34 +0000 | [diff] [blame] | 1037 | return false; |
| 1038 | break; |
| 1039 | |
Chris Lattner | 3f0f71b | 2005-11-19 02:11:08 +0000 | [diff] [blame] | 1040 | case ISD::FrameIndex: |
Eli Friedman | 344ec79 | 2011-07-13 21:29:53 +0000 | [diff] [blame] | 1041 | if (AM.BaseType == X86ISelAddressMode::RegBase && |
Craig Topper | 062a2ba | 2014-04-25 05:30:21 +0000 | [diff] [blame] | 1042 | AM.Base_Reg.getNode() == nullptr && |
Eli Friedman | 344ec79 | 2011-07-13 21:29:53 +0000 | [diff] [blame] | 1043 | (!Subtarget->is64Bit() || isDispSafeForFrameIndex(AM.Disp))) { |
Chris Lattner | 3f0f71b | 2005-11-19 02:11:08 +0000 | [diff] [blame] | 1044 | AM.BaseType = X86ISelAddressMode::FrameIndexBase; |
Dan Gohman | 0fd54fb | 2010-04-29 23:30:41 +0000 | [diff] [blame] | 1045 | AM.Base_FrameIndex = cast<FrameIndexSDNode>(N)->getIndex(); |
Chris Lattner | 3f0f71b | 2005-11-19 02:11:08 +0000 | [diff] [blame] | 1046 | return false; |
| 1047 | } |
| 1048 | break; |
Evan Cheng | c9fab31 | 2005-12-08 02:01:35 +0000 | [diff] [blame] | 1049 | |
Chris Lattner | 3f0f71b | 2005-11-19 02:11:08 +0000 | [diff] [blame] | 1050 | case ISD::SHL: |
Craig Topper | 062a2ba | 2014-04-25 05:30:21 +0000 | [diff] [blame] | 1051 | if (AM.IndexReg.getNode() != nullptr || AM.Scale != 1) |
Chris Lattner | ff87f05e | 2007-12-08 07:22:58 +0000 | [diff] [blame] | 1052 | break; |
Chad Rosier | 24c19d2 | 2012-08-01 18:39:17 +0000 | [diff] [blame] | 1053 | |
Gabor Greif | 81d6a38 | 2008-08-31 15:37:04 +0000 | [diff] [blame] | 1054 | if (ConstantSDNode |
| 1055 | *CN = dyn_cast<ConstantSDNode>(N.getNode()->getOperand(1))) { |
Dan Gohman | effb894 | 2008-09-12 16:56:44 +0000 | [diff] [blame] | 1056 | unsigned Val = CN->getZExtValue(); |
Dan Gohman | 824ab40 | 2009-07-22 23:26:55 +0000 | [diff] [blame] | 1057 | // Note that we handle x<<1 as (,x,2) rather than (x,x) here so |
| 1058 | // that the base operand remains free for further matching. If |
| 1059 | // the base doesn't end up getting used, a post-processing step |
| 1060 | // in MatchAddress turns (,x,2) into (x,x), which is cheaper. |
Chris Lattner | ff87f05e | 2007-12-08 07:22:58 +0000 | [diff] [blame] | 1061 | if (Val == 1 || Val == 2 || Val == 3) { |
| 1062 | AM.Scale = 1 << Val; |
Gabor Greif | f304a7a | 2008-08-28 21:40:38 +0000 | [diff] [blame] | 1063 | SDValue ShVal = N.getNode()->getOperand(0); |
Chris Lattner | 3f0f71b | 2005-11-19 02:11:08 +0000 | [diff] [blame] | 1064 | |
Chris Lattner | ff87f05e | 2007-12-08 07:22:58 +0000 | [diff] [blame] | 1065 | // Okay, we know that we have a scale by now. However, if the scaled |
| 1066 | // value is an add of something and a constant, we can fold the |
| 1067 | // constant into the disp field here. |
Chris Lattner | 46c01a3 | 2011-02-13 22:25:43 +0000 | [diff] [blame] | 1068 | if (CurDAG->isBaseWithConstantOffset(ShVal)) { |
Gabor Greif | f304a7a | 2008-08-28 21:40:38 +0000 | [diff] [blame] | 1069 | AM.IndexReg = ShVal.getNode()->getOperand(0); |
Chris Lattner | ff87f05e | 2007-12-08 07:22:58 +0000 | [diff] [blame] | 1070 | ConstantSDNode *AddVal = |
Gabor Greif | f304a7a | 2008-08-28 21:40:38 +0000 | [diff] [blame] | 1071 | cast<ConstantSDNode>(ShVal.getNode()->getOperand(1)); |
Richard Smith | 228e6d4 | 2012-08-24 23:29:28 +0000 | [diff] [blame] | 1072 | uint64_t Disp = (uint64_t)AddVal->getSExtValue() << Val; |
Eli Friedman | ef67e7d | 2011-07-13 20:44:23 +0000 | [diff] [blame] | 1073 | if (!FoldOffsetIntoAddress(Disp, AM)) |
| 1074 | return false; |
Chris Lattner | 3f0f71b | 2005-11-19 02:11:08 +0000 | [diff] [blame] | 1075 | } |
Eli Friedman | ef67e7d | 2011-07-13 20:44:23 +0000 | [diff] [blame] | 1076 | |
| 1077 | AM.IndexReg = ShVal; |
Chris Lattner | ff87f05e | 2007-12-08 07:22:58 +0000 | [diff] [blame] | 1078 | return false; |
Chris Lattner | 3f0f71b | 2005-11-19 02:11:08 +0000 | [diff] [blame] | 1079 | } |
Chris Lattner | ff87f05e | 2007-12-08 07:22:58 +0000 | [diff] [blame] | 1080 | } |
Jakub Staszak | 43fafaf | 2013-01-04 23:01:26 +0000 | [diff] [blame] | 1081 | break; |
Evan Cheng | c9fab31 | 2005-12-08 02:01:35 +0000 | [diff] [blame] | 1082 | |
Chandler Carruth | 3dbcda8 | 2012-01-11 09:35:02 +0000 | [diff] [blame] | 1083 | case ISD::SRL: { |
| 1084 | // Scale must not be used already. |
Craig Topper | 062a2ba | 2014-04-25 05:30:21 +0000 | [diff] [blame] | 1085 | if (AM.IndexReg.getNode() != nullptr || AM.Scale != 1) break; |
Chandler Carruth | 3dbcda8 | 2012-01-11 09:35:02 +0000 | [diff] [blame] | 1086 | |
| 1087 | SDValue And = N.getOperand(0); |
| 1088 | if (And.getOpcode() != ISD::AND) break; |
| 1089 | SDValue X = And.getOperand(0); |
| 1090 | |
| 1091 | // We only handle up to 64-bit values here as those are what matter for |
| 1092 | // addressing mode optimizations. |
Craig Topper | 83e042a | 2013-08-15 05:57:07 +0000 | [diff] [blame] | 1093 | if (X.getSimpleValueType().getSizeInBits() > 64) break; |
Chandler Carruth | 3dbcda8 | 2012-01-11 09:35:02 +0000 | [diff] [blame] | 1094 | |
| 1095 | // The mask used for the transform is expected to be post-shift, but we |
| 1096 | // found the shift first so just apply the shift to the mask before passing |
| 1097 | // it down. |
| 1098 | if (!isa<ConstantSDNode>(N.getOperand(1)) || |
| 1099 | !isa<ConstantSDNode>(And.getOperand(1))) |
| 1100 | break; |
| 1101 | uint64_t Mask = And.getConstantOperandVal(1) >> N.getConstantOperandVal(1); |
| 1102 | |
Chandler Carruth | 55b2cde | 2012-01-11 08:41:08 +0000 | [diff] [blame] | 1103 | // Try to fold the mask and shift into the scale, and return false if we |
| 1104 | // succeed. |
Chandler Carruth | 3dbcda8 | 2012-01-11 09:35:02 +0000 | [diff] [blame] | 1105 | if (!FoldMaskAndShiftToScale(*CurDAG, N, Mask, N, X, AM)) |
Chandler Carruth | 55b2cde | 2012-01-11 08:41:08 +0000 | [diff] [blame] | 1106 | return false; |
| 1107 | break; |
Chandler Carruth | 3dbcda8 | 2012-01-11 09:35:02 +0000 | [diff] [blame] | 1108 | } |
Chandler Carruth | 55b2cde | 2012-01-11 08:41:08 +0000 | [diff] [blame] | 1109 | |
Dan Gohman | bf47495 | 2007-10-22 20:22:24 +0000 | [diff] [blame] | 1110 | case ISD::SMUL_LOHI: |
| 1111 | case ISD::UMUL_LOHI: |
| 1112 | // A mul_lohi where we need the low part can be folded as a plain multiply. |
Gabor Greif | abfdf92 | 2008-08-26 22:36:50 +0000 | [diff] [blame] | 1113 | if (N.getResNo() != 0) break; |
Dan Gohman | bf47495 | 2007-10-22 20:22:24 +0000 | [diff] [blame] | 1114 | // FALL THROUGH |
Chris Lattner | 3f0f71b | 2005-11-19 02:11:08 +0000 | [diff] [blame] | 1115 | case ISD::MUL: |
Evan Cheng | a84a318 | 2009-03-30 21:36:47 +0000 | [diff] [blame] | 1116 | case X86ISD::MUL_IMM: |
Chris Lattner | 3f0f71b | 2005-11-19 02:11:08 +0000 | [diff] [blame] | 1117 | // X*[3,5,9] -> X+X*[2,4,8] |
Dan Gohman | f14b77e | 2008-11-05 04:14:16 +0000 | [diff] [blame] | 1118 | if (AM.BaseType == X86ISelAddressMode::RegBase && |
Craig Topper | 062a2ba | 2014-04-25 05:30:21 +0000 | [diff] [blame] | 1119 | AM.Base_Reg.getNode() == nullptr && |
| 1120 | AM.IndexReg.getNode() == nullptr) { |
Gabor Greif | 81d6a38 | 2008-08-31 15:37:04 +0000 | [diff] [blame] | 1121 | if (ConstantSDNode |
| 1122 | *CN = dyn_cast<ConstantSDNode>(N.getNode()->getOperand(1))) |
Dan Gohman | effb894 | 2008-09-12 16:56:44 +0000 | [diff] [blame] | 1123 | if (CN->getZExtValue() == 3 || CN->getZExtValue() == 5 || |
| 1124 | CN->getZExtValue() == 9) { |
| 1125 | AM.Scale = unsigned(CN->getZExtValue())-1; |
Chris Lattner | 3f0f71b | 2005-11-19 02:11:08 +0000 | [diff] [blame] | 1126 | |
Gabor Greif | f304a7a | 2008-08-28 21:40:38 +0000 | [diff] [blame] | 1127 | SDValue MulVal = N.getNode()->getOperand(0); |
Dan Gohman | 2ce6f2a | 2008-07-27 21:46:04 +0000 | [diff] [blame] | 1128 | SDValue Reg; |
Chris Lattner | 3f0f71b | 2005-11-19 02:11:08 +0000 | [diff] [blame] | 1129 | |
| 1130 | // Okay, we know that we have a scale by now. However, if the scaled |
| 1131 | // value is an add of something and a constant, we can fold the |
| 1132 | // constant into the disp field here. |
Gabor Greif | f304a7a | 2008-08-28 21:40:38 +0000 | [diff] [blame] | 1133 | if (MulVal.getNode()->getOpcode() == ISD::ADD && MulVal.hasOneUse() && |
| 1134 | isa<ConstantSDNode>(MulVal.getNode()->getOperand(1))) { |
| 1135 | Reg = MulVal.getNode()->getOperand(0); |
Chris Lattner | 3f0f71b | 2005-11-19 02:11:08 +0000 | [diff] [blame] | 1136 | ConstantSDNode *AddVal = |
Gabor Greif | f304a7a | 2008-08-28 21:40:38 +0000 | [diff] [blame] | 1137 | cast<ConstantSDNode>(MulVal.getNode()->getOperand(1)); |
Eli Friedman | ef67e7d | 2011-07-13 20:44:23 +0000 | [diff] [blame] | 1138 | uint64_t Disp = AddVal->getSExtValue() * CN->getZExtValue(); |
| 1139 | if (FoldOffsetIntoAddress(Disp, AM)) |
Gabor Greif | f304a7a | 2008-08-28 21:40:38 +0000 | [diff] [blame] | 1140 | Reg = N.getNode()->getOperand(0); |
Chris Lattner | 3f0f71b | 2005-11-19 02:11:08 +0000 | [diff] [blame] | 1141 | } else { |
Gabor Greif | f304a7a | 2008-08-28 21:40:38 +0000 | [diff] [blame] | 1142 | Reg = N.getNode()->getOperand(0); |
Chris Lattner | 3f0f71b | 2005-11-19 02:11:08 +0000 | [diff] [blame] | 1143 | } |
| 1144 | |
Dan Gohman | 0fd54fb | 2010-04-29 23:30:41 +0000 | [diff] [blame] | 1145 | AM.IndexReg = AM.Base_Reg = Reg; |
Chris Lattner | 3f0f71b | 2005-11-19 02:11:08 +0000 | [diff] [blame] | 1146 | return false; |
| 1147 | } |
Chris Lattner | fe8c530 | 2007-02-04 20:18:17 +0000 | [diff] [blame] | 1148 | } |
Chris Lattner | 3f0f71b | 2005-11-19 02:11:08 +0000 | [diff] [blame] | 1149 | break; |
| 1150 | |
Dan Gohman | faf75c8 | 2009-05-11 18:02:53 +0000 | [diff] [blame] | 1151 | case ISD::SUB: { |
| 1152 | // Given A-B, if A can be completely folded into the address and |
| 1153 | // the index field with the index field unused, use -B as the index. |
| 1154 | // This is a win if a has multiple parts that can be folded into |
| 1155 | // the address. Also, this saves a mov if the base register has |
| 1156 | // other uses, since it avoids a two-address sub instruction, however |
| 1157 | // it costs an additional mov if the index register has other uses. |
| 1158 | |
Dan Gohman | 99ba4da | 2010-06-18 01:24:29 +0000 | [diff] [blame] | 1159 | // Add an artificial use to this node so that we can keep track of |
| 1160 | // it if it gets CSE'd with a different node. |
| 1161 | HandleSDNode Handle(N); |
| 1162 | |
Dan Gohman | faf75c8 | 2009-05-11 18:02:53 +0000 | [diff] [blame] | 1163 | // Test if the LHS of the sub can be folded. |
| 1164 | X86ISelAddressMode Backup = AM; |
Dan Gohman | 99ba4da | 2010-06-18 01:24:29 +0000 | [diff] [blame] | 1165 | if (MatchAddressRecursively(N.getNode()->getOperand(0), AM, Depth+1)) { |
Dan Gohman | faf75c8 | 2009-05-11 18:02:53 +0000 | [diff] [blame] | 1166 | AM = Backup; |
| 1167 | break; |
| 1168 | } |
| 1169 | // Test if the index field is free for use. |
Chris Lattner | fea81da | 2009-06-27 04:16:01 +0000 | [diff] [blame] | 1170 | if (AM.IndexReg.getNode() || AM.isRIPRelative()) { |
Dan Gohman | faf75c8 | 2009-05-11 18:02:53 +0000 | [diff] [blame] | 1171 | AM = Backup; |
| 1172 | break; |
| 1173 | } |
Evan Cheng | 68333f5 | 2010-03-17 23:58:35 +0000 | [diff] [blame] | 1174 | |
Dan Gohman | faf75c8 | 2009-05-11 18:02:53 +0000 | [diff] [blame] | 1175 | int Cost = 0; |
Dan Gohman | 99ba4da | 2010-06-18 01:24:29 +0000 | [diff] [blame] | 1176 | SDValue RHS = Handle.getValue().getNode()->getOperand(1); |
Dan Gohman | faf75c8 | 2009-05-11 18:02:53 +0000 | [diff] [blame] | 1177 | // If the RHS involves a register with multiple uses, this |
| 1178 | // transformation incurs an extra mov, due to the neg instruction |
| 1179 | // clobbering its operand. |
| 1180 | if (!RHS.getNode()->hasOneUse() || |
| 1181 | RHS.getNode()->getOpcode() == ISD::CopyFromReg || |
| 1182 | RHS.getNode()->getOpcode() == ISD::TRUNCATE || |
| 1183 | RHS.getNode()->getOpcode() == ISD::ANY_EXTEND || |
| 1184 | (RHS.getNode()->getOpcode() == ISD::ZERO_EXTEND && |
Owen Anderson | 9f94459 | 2009-08-11 20:47:22 +0000 | [diff] [blame] | 1185 | RHS.getNode()->getOperand(0).getValueType() == MVT::i32)) |
Dan Gohman | faf75c8 | 2009-05-11 18:02:53 +0000 | [diff] [blame] | 1186 | ++Cost; |
| 1187 | // If the base is a register with multiple uses, this |
| 1188 | // transformation may save a mov. |
| 1189 | if ((AM.BaseType == X86ISelAddressMode::RegBase && |
Dan Gohman | 0fd54fb | 2010-04-29 23:30:41 +0000 | [diff] [blame] | 1190 | AM.Base_Reg.getNode() && |
| 1191 | !AM.Base_Reg.getNode()->hasOneUse()) || |
Dan Gohman | faf75c8 | 2009-05-11 18:02:53 +0000 | [diff] [blame] | 1192 | AM.BaseType == X86ISelAddressMode::FrameIndexBase) |
| 1193 | --Cost; |
| 1194 | // If the folded LHS was interesting, this transformation saves |
| 1195 | // address arithmetic. |
| 1196 | if ((AM.hasSymbolicDisplacement() && !Backup.hasSymbolicDisplacement()) + |
| 1197 | ((AM.Disp != 0) && (Backup.Disp == 0)) + |
| 1198 | (AM.Segment.getNode() && !Backup.Segment.getNode()) >= 2) |
| 1199 | --Cost; |
| 1200 | // If it doesn't look like it may be an overall win, don't do it. |
| 1201 | if (Cost >= 0) { |
| 1202 | AM = Backup; |
| 1203 | break; |
| 1204 | } |
| 1205 | |
| 1206 | // Ok, the transformation is legal and appears profitable. Go for it. |
Sergey Dmitrouk | 842a51b | 2015-04-28 14:05:47 +0000 | [diff] [blame] | 1207 | SDValue Zero = CurDAG->getConstant(0, dl, N.getValueType()); |
Dan Gohman | faf75c8 | 2009-05-11 18:02:53 +0000 | [diff] [blame] | 1208 | SDValue Neg = CurDAG->getNode(ISD::SUB, dl, N.getValueType(), Zero, RHS); |
| 1209 | AM.IndexReg = Neg; |
| 1210 | AM.Scale = 1; |
| 1211 | |
| 1212 | // Insert the new nodes into the topological ordering. |
Chandler Carruth | 3eacfb8 | 2012-01-11 11:04:36 +0000 | [diff] [blame] | 1213 | InsertDAGNode(*CurDAG, N, Zero); |
| 1214 | InsertDAGNode(*CurDAG, N, Neg); |
Dan Gohman | faf75c8 | 2009-05-11 18:02:53 +0000 | [diff] [blame] | 1215 | return false; |
| 1216 | } |
| 1217 | |
Evan Cheng | bf38a5e | 2009-01-17 07:09:27 +0000 | [diff] [blame] | 1218 | case ISD::ADD: { |
Dan Gohman | 99ba4da | 2010-06-18 01:24:29 +0000 | [diff] [blame] | 1219 | // Add an artificial use to this node so that we can keep track of |
| 1220 | // it if it gets CSE'd with a different node. |
| 1221 | HandleSDNode Handle(N); |
Dan Gohman | 99ba4da | 2010-06-18 01:24:29 +0000 | [diff] [blame] | 1222 | |
Evan Cheng | bf38a5e | 2009-01-17 07:09:27 +0000 | [diff] [blame] | 1223 | X86ISelAddressMode Backup = AM; |
Chris Lattner | 35a2e65 | 2011-01-16 08:48:11 +0000 | [diff] [blame] | 1224 | if (!MatchAddressRecursively(N.getOperand(0), AM, Depth+1) && |
| 1225 | !MatchAddressRecursively(Handle.getValue().getOperand(1), AM, Depth+1)) |
Dan Gohman | 99ba4da | 2010-06-18 01:24:29 +0000 | [diff] [blame] | 1226 | return false; |
| 1227 | AM = Backup; |
Chad Rosier | 24c19d2 | 2012-08-01 18:39:17 +0000 | [diff] [blame] | 1228 | |
Evan Cheng | 68333f5 | 2010-03-17 23:58:35 +0000 | [diff] [blame] | 1229 | // Try again after commuting the operands. |
Chris Lattner | 35a2e65 | 2011-01-16 08:48:11 +0000 | [diff] [blame] | 1230 | if (!MatchAddressRecursively(Handle.getValue().getOperand(1), AM, Depth+1)&& |
| 1231 | !MatchAddressRecursively(Handle.getValue().getOperand(0), AM, Depth+1)) |
Dan Gohman | 99ba4da | 2010-06-18 01:24:29 +0000 | [diff] [blame] | 1232 | return false; |
Evan Cheng | bf38a5e | 2009-01-17 07:09:27 +0000 | [diff] [blame] | 1233 | AM = Backup; |
Dan Gohman | a1d9242 | 2009-03-13 02:25:09 +0000 | [diff] [blame] | 1234 | |
| 1235 | // If we couldn't fold both operands into the address at the same time, |
| 1236 | // see if we can just put each operand into a register and fold at least |
| 1237 | // the add. |
| 1238 | if (AM.BaseType == X86ISelAddressMode::RegBase && |
Dan Gohman | 0fd54fb | 2010-04-29 23:30:41 +0000 | [diff] [blame] | 1239 | !AM.Base_Reg.getNode() && |
Chris Lattner | fea81da | 2009-06-27 04:16:01 +0000 | [diff] [blame] | 1240 | !AM.IndexReg.getNode()) { |
Chris Lattner | 35a2e65 | 2011-01-16 08:48:11 +0000 | [diff] [blame] | 1241 | N = Handle.getValue(); |
| 1242 | AM.Base_Reg = N.getOperand(0); |
| 1243 | AM.IndexReg = N.getOperand(1); |
Dan Gohman | a1d9242 | 2009-03-13 02:25:09 +0000 | [diff] [blame] | 1244 | AM.Scale = 1; |
| 1245 | return false; |
| 1246 | } |
Chris Lattner | 35a2e65 | 2011-01-16 08:48:11 +0000 | [diff] [blame] | 1247 | N = Handle.getValue(); |
Chris Lattner | 3f0f71b | 2005-11-19 02:11:08 +0000 | [diff] [blame] | 1248 | break; |
Evan Cheng | bf38a5e | 2009-01-17 07:09:27 +0000 | [diff] [blame] | 1249 | } |
Evan Cheng | 734e1e2 | 2006-05-30 06:59:36 +0000 | [diff] [blame] | 1250 | |
Chris Lattner | fe8c530 | 2007-02-04 20:18:17 +0000 | [diff] [blame] | 1251 | case ISD::OR: |
Sylvestre Ledru | 91ce36c | 2012-09-27 10:14:43 +0000 | [diff] [blame] | 1252 | // Handle "X | C" as "X + C" iff X is known to have C bits clear. |
Chris Lattner | 46c01a3 | 2011-02-13 22:25:43 +0000 | [diff] [blame] | 1253 | if (CurDAG->isBaseWithConstantOffset(N)) { |
Chris Lattner | ff87f05e | 2007-12-08 07:22:58 +0000 | [diff] [blame] | 1254 | X86ISelAddressMode Backup = AM; |
Chris Lattner | 8477678 | 2010-04-20 23:18:40 +0000 | [diff] [blame] | 1255 | ConstantSDNode *CN = cast<ConstantSDNode>(N.getOperand(1)); |
Evan Cheng | 68333f5 | 2010-03-17 23:58:35 +0000 | [diff] [blame] | 1256 | |
Chris Lattner | ff87f05e | 2007-12-08 07:22:58 +0000 | [diff] [blame] | 1257 | // Start with the LHS as an addr mode. |
Dan Gohman | 99ba4da | 2010-06-18 01:24:29 +0000 | [diff] [blame] | 1258 | if (!MatchAddressRecursively(N.getOperand(0), AM, Depth+1) && |
Eli Friedman | ef67e7d | 2011-07-13 20:44:23 +0000 | [diff] [blame] | 1259 | !FoldOffsetIntoAddress(CN->getSExtValue(), AM)) |
Chris Lattner | ff87f05e | 2007-12-08 07:22:58 +0000 | [diff] [blame] | 1260 | return false; |
Chris Lattner | ff87f05e | 2007-12-08 07:22:58 +0000 | [diff] [blame] | 1261 | AM = Backup; |
Evan Cheng | 734e1e2 | 2006-05-30 06:59:36 +0000 | [diff] [blame] | 1262 | } |
| 1263 | break; |
Chad Rosier | 24c19d2 | 2012-08-01 18:39:17 +0000 | [diff] [blame] | 1264 | |
Evan Cheng | 827d30d | 2007-12-13 00:43:27 +0000 | [diff] [blame] | 1265 | case ISD::AND: { |
Dan Gohman | 57d6bd3 | 2009-04-13 16:09:41 +0000 | [diff] [blame] | 1266 | // Perform some heroic transforms on an and of a constant-count shift |
| 1267 | // with a constant to enable use of the scaled offset field. |
| 1268 | |
Evan Cheng | 827d30d | 2007-12-13 00:43:27 +0000 | [diff] [blame] | 1269 | // Scale must not be used already. |
Craig Topper | 062a2ba | 2014-04-25 05:30:21 +0000 | [diff] [blame] | 1270 | if (AM.IndexReg.getNode() != nullptr || AM.Scale != 1) break; |
Evan Cheng | a20a773 | 2008-02-07 08:53:49 +0000 | [diff] [blame] | 1271 | |
Chandler Carruth | aa01e66 | 2012-01-11 09:35:00 +0000 | [diff] [blame] | 1272 | SDValue Shift = N.getOperand(0); |
| 1273 | if (Shift.getOpcode() != ISD::SRL && Shift.getOpcode() != ISD::SHL) break; |
Dan Gohman | 57d6bd3 | 2009-04-13 16:09:41 +0000 | [diff] [blame] | 1274 | SDValue X = Shift.getOperand(0); |
Chandler Carruth | aa01e66 | 2012-01-11 09:35:00 +0000 | [diff] [blame] | 1275 | |
| 1276 | // We only handle up to 64-bit values here as those are what matter for |
| 1277 | // addressing mode optimizations. |
Craig Topper | 83e042a | 2013-08-15 05:57:07 +0000 | [diff] [blame] | 1278 | if (X.getSimpleValueType().getSizeInBits() > 64) break; |
Chandler Carruth | aa01e66 | 2012-01-11 09:35:00 +0000 | [diff] [blame] | 1279 | |
Chandler Carruth | b0049f4 | 2012-01-11 09:35:04 +0000 | [diff] [blame] | 1280 | if (!isa<ConstantSDNode>(N.getOperand(1))) |
| 1281 | break; |
| 1282 | uint64_t Mask = N.getConstantOperandVal(1); |
Evan Cheng | 827d30d | 2007-12-13 00:43:27 +0000 | [diff] [blame] | 1283 | |
Chandler Carruth | 51d3076 | 2012-01-11 08:48:20 +0000 | [diff] [blame] | 1284 | // Try to fold the mask and shift into an extract and scale. |
Chandler Carruth | b0049f4 | 2012-01-11 09:35:04 +0000 | [diff] [blame] | 1285 | if (!FoldMaskAndShiftToExtract(*CurDAG, N, Mask, Shift, X, AM)) |
Chandler Carruth | 51d3076 | 2012-01-11 08:48:20 +0000 | [diff] [blame] | 1286 | return false; |
Dan Gohman | 57d6bd3 | 2009-04-13 16:09:41 +0000 | [diff] [blame] | 1287 | |
Chandler Carruth | 51d3076 | 2012-01-11 08:48:20 +0000 | [diff] [blame] | 1288 | // Try to fold the mask and shift directly into the scale. |
Chandler Carruth | b0049f4 | 2012-01-11 09:35:04 +0000 | [diff] [blame] | 1289 | if (!FoldMaskAndShiftToScale(*CurDAG, N, Mask, Shift, X, AM)) |
Chandler Carruth | 55b2cde | 2012-01-11 08:41:08 +0000 | [diff] [blame] | 1290 | return false; |
| 1291 | |
Chandler Carruth | aa01e66 | 2012-01-11 09:35:00 +0000 | [diff] [blame] | 1292 | // Try to swap the mask and shift to place shifts which can be done as |
| 1293 | // a scale on the outside of the mask. |
Chandler Carruth | b0049f4 | 2012-01-11 09:35:04 +0000 | [diff] [blame] | 1294 | if (!FoldMaskedShiftToScaledMask(*CurDAG, N, Mask, Shift, X, AM)) |
Chandler Carruth | aa01e66 | 2012-01-11 09:35:00 +0000 | [diff] [blame] | 1295 | return false; |
| 1296 | break; |
Evan Cheng | 827d30d | 2007-12-13 00:43:27 +0000 | [diff] [blame] | 1297 | } |
Evan Cheng | 734e1e2 | 2006-05-30 06:59:36 +0000 | [diff] [blame] | 1298 | } |
Chris Lattner | 3f0f71b | 2005-11-19 02:11:08 +0000 | [diff] [blame] | 1299 | |
Rafael Espindola | 9277379 | 2009-03-31 16:16:57 +0000 | [diff] [blame] | 1300 | return MatchAddressBase(N, AM); |
Dan Gohman | ccb3611 | 2007-08-13 20:03:06 +0000 | [diff] [blame] | 1301 | } |
| 1302 | |
| 1303 | /// MatchAddressBase - Helper for MatchAddress. Add the specified node to the |
| 1304 | /// specified addressing mode without any further recursion. |
Rafael Espindola | 9277379 | 2009-03-31 16:16:57 +0000 | [diff] [blame] | 1305 | bool X86DAGToDAGISel::MatchAddressBase(SDValue N, X86ISelAddressMode &AM) { |
Chris Lattner | 3f0f71b | 2005-11-19 02:11:08 +0000 | [diff] [blame] | 1306 | // Is the base register already occupied? |
Dan Gohman | 0fd54fb | 2010-04-29 23:30:41 +0000 | [diff] [blame] | 1307 | if (AM.BaseType != X86ISelAddressMode::RegBase || AM.Base_Reg.getNode()) { |
Chris Lattner | 3f0f71b | 2005-11-19 02:11:08 +0000 | [diff] [blame] | 1308 | // If so, check to see if the scale index register is set. |
Craig Topper | 062a2ba | 2014-04-25 05:30:21 +0000 | [diff] [blame] | 1309 | if (!AM.IndexReg.getNode()) { |
Chris Lattner | 3f0f71b | 2005-11-19 02:11:08 +0000 | [diff] [blame] | 1310 | AM.IndexReg = N; |
| 1311 | AM.Scale = 1; |
| 1312 | return false; |
| 1313 | } |
| 1314 | |
| 1315 | // Otherwise, we cannot select it. |
| 1316 | return true; |
| 1317 | } |
| 1318 | |
| 1319 | // Default, generate it as a register. |
| 1320 | AM.BaseType = X86ISelAddressMode::RegBase; |
Dan Gohman | 0fd54fb | 2010-04-29 23:30:41 +0000 | [diff] [blame] | 1321 | AM.Base_Reg = N; |
Chris Lattner | 3f0f71b | 2005-11-19 02:11:08 +0000 | [diff] [blame] | 1322 | return false; |
| 1323 | } |
| 1324 | |
Elena Demikhovsky | e1eda8a | 2015-04-30 08:38:48 +0000 | [diff] [blame] | 1325 | bool X86DAGToDAGISel::SelectVectorAddr(SDNode *Parent, SDValue N, SDValue &Base, |
| 1326 | SDValue &Scale, SDValue &Index, |
| 1327 | SDValue &Disp, SDValue &Segment) { |
| 1328 | |
| 1329 | MaskedGatherScatterSDNode *Mgs = dyn_cast<MaskedGatherScatterSDNode>(Parent); |
| 1330 | if (!Mgs) |
| 1331 | return false; |
| 1332 | X86ISelAddressMode AM; |
| 1333 | unsigned AddrSpace = Mgs->getPointerInfo().getAddrSpace(); |
| 1334 | // AddrSpace 256 -> GS, 257 -> FS. |
| 1335 | if (AddrSpace == 256) |
| 1336 | AM.Segment = CurDAG->getRegister(X86::GS, MVT::i16); |
| 1337 | if (AddrSpace == 257) |
| 1338 | AM.Segment = CurDAG->getRegister(X86::FS, MVT::i16); |
| 1339 | |
| 1340 | SDLoc DL(N); |
| 1341 | Base = Mgs->getBasePtr(); |
| 1342 | Index = Mgs->getIndex(); |
| 1343 | unsigned ScalarSize = Mgs->getValue().getValueType().getScalarSizeInBits(); |
| 1344 | Scale = getI8Imm(ScalarSize/8, DL); |
| 1345 | |
| 1346 | // If Base is 0, the whole address is in index and the Scale is 1 |
Daniel Jasper | 232778a | 2015-04-30 09:01:21 +0000 | [diff] [blame^] | 1347 | if (isa<ConstantSDNode>(Base)) { |
| 1348 | assert(dyn_cast<ConstantSDNode>(Base)->isNullValue() && |
| 1349 | "Unexpected base in gather/scatter"); |
Elena Demikhovsky | e1eda8a | 2015-04-30 08:38:48 +0000 | [diff] [blame] | 1350 | Scale = getI8Imm(1, DL); |
| 1351 | Base = CurDAG->getRegister(0, MVT::i32); |
| 1352 | } |
| 1353 | if (AM.Segment.getNode()) |
| 1354 | Segment = AM.Segment; |
| 1355 | else |
| 1356 | Segment = CurDAG->getRegister(0, MVT::i32); |
| 1357 | Disp = CurDAG->getTargetConstant(0, DL, MVT::i32); |
| 1358 | return true; |
| 1359 | } |
| 1360 | |
Evan Cheng | c9fab31 | 2005-12-08 02:01:35 +0000 | [diff] [blame] | 1361 | /// SelectAddr - returns true if it is able pattern match an addressing mode. |
| 1362 | /// It returns the operands which make up the maximal addressing mode it can |
| 1363 | /// match by reference. |
Chris Lattner | d58d7c1 | 2010-09-21 22:07:31 +0000 | [diff] [blame] | 1364 | /// |
| 1365 | /// Parent is the parent node of the addr operand that is being matched. It |
| 1366 | /// is always a load, store, atomic node, or null. It is only null when |
| 1367 | /// checking memory operands for inline asm nodes. |
| 1368 | bool X86DAGToDAGISel::SelectAddr(SDNode *Parent, SDValue N, SDValue &Base, |
Dan Gohman | 2ce6f2a | 2008-07-27 21:46:04 +0000 | [diff] [blame] | 1369 | SDValue &Scale, SDValue &Index, |
Rafael Espindola | 3b2df10 | 2009-04-08 21:14:34 +0000 | [diff] [blame] | 1370 | SDValue &Disp, SDValue &Segment) { |
Evan Cheng | c9fab31 | 2005-12-08 02:01:35 +0000 | [diff] [blame] | 1371 | X86ISelAddressMode AM; |
Chad Rosier | 24c19d2 | 2012-08-01 18:39:17 +0000 | [diff] [blame] | 1372 | |
Chris Lattner | 8a236b6 | 2010-09-22 04:39:11 +0000 | [diff] [blame] | 1373 | if (Parent && |
| 1374 | // This list of opcodes are all the nodes that have an "addr:$ptr" operand |
| 1375 | // that are not a MemSDNode, and thus don't have proper addrspace info. |
Chris Lattner | 8a236b6 | 2010-09-22 04:39:11 +0000 | [diff] [blame] | 1376 | Parent->getOpcode() != ISD::INTRINSIC_W_CHAIN && // unaligned loads, fixme |
Eric Christopher | c1b3e07 | 2010-09-22 20:42:08 +0000 | [diff] [blame] | 1377 | Parent->getOpcode() != ISD::INTRINSIC_VOID && // nontemporal stores |
Michael Liao | 97bf363 | 2012-10-15 22:39:43 +0000 | [diff] [blame] | 1378 | Parent->getOpcode() != X86ISD::TLSCALL && // Fixme |
| 1379 | Parent->getOpcode() != X86ISD::EH_SJLJ_SETJMP && // setjmp |
| 1380 | Parent->getOpcode() != X86ISD::EH_SJLJ_LONGJMP) { // longjmp |
Chris Lattner | 8a236b6 | 2010-09-22 04:39:11 +0000 | [diff] [blame] | 1381 | unsigned AddrSpace = |
| 1382 | cast<MemSDNode>(Parent)->getPointerInfo().getAddrSpace(); |
| 1383 | // AddrSpace 256 -> GS, 257 -> FS. |
| 1384 | if (AddrSpace == 256) |
| 1385 | AM.Segment = CurDAG->getRegister(X86::GS, MVT::i16); |
| 1386 | if (AddrSpace == 257) |
| 1387 | AM.Segment = CurDAG->getRegister(X86::FS, MVT::i16); |
| 1388 | } |
Chad Rosier | 24c19d2 | 2012-08-01 18:39:17 +0000 | [diff] [blame] | 1389 | |
Evan Cheng | 3dfd04e | 2009-12-18 01:59:21 +0000 | [diff] [blame] | 1390 | if (MatchAddress(N, AM)) |
Evan Cheng | bc7a0f44 | 2006-01-11 06:09:51 +0000 | [diff] [blame] | 1391 | return false; |
Evan Cheng | c9fab31 | 2005-12-08 02:01:35 +0000 | [diff] [blame] | 1392 | |
Craig Topper | 83e042a | 2013-08-15 05:57:07 +0000 | [diff] [blame] | 1393 | MVT VT = N.getSimpleValueType(); |
Evan Cheng | bc7a0f44 | 2006-01-11 06:09:51 +0000 | [diff] [blame] | 1394 | if (AM.BaseType == X86ISelAddressMode::RegBase) { |
Dan Gohman | 0fd54fb | 2010-04-29 23:30:41 +0000 | [diff] [blame] | 1395 | if (!AM.Base_Reg.getNode()) |
| 1396 | AM.Base_Reg = CurDAG->getRegister(0, VT); |
Evan Cheng | c9fab31 | 2005-12-08 02:01:35 +0000 | [diff] [blame] | 1397 | } |
Evan Cheng | bc7a0f44 | 2006-01-11 06:09:51 +0000 | [diff] [blame] | 1398 | |
Gabor Greif | f304a7a | 2008-08-28 21:40:38 +0000 | [diff] [blame] | 1399 | if (!AM.IndexReg.getNode()) |
Evan Cheng | 11b0a5d | 2006-09-08 06:48:29 +0000 | [diff] [blame] | 1400 | AM.IndexReg = CurDAG->getRegister(0, VT); |
Evan Cheng | bc7a0f44 | 2006-01-11 06:09:51 +0000 | [diff] [blame] | 1401 | |
Sergey Dmitrouk | 842a51b | 2015-04-28 14:05:47 +0000 | [diff] [blame] | 1402 | getAddressOperands(AM, SDLoc(N), Base, Scale, Index, Disp, Segment); |
Evan Cheng | bc7a0f44 | 2006-01-11 06:09:51 +0000 | [diff] [blame] | 1403 | return true; |
Evan Cheng | c9fab31 | 2005-12-08 02:01:35 +0000 | [diff] [blame] | 1404 | } |
| 1405 | |
Chris Lattner | 398195e | 2006-10-07 21:55:32 +0000 | [diff] [blame] | 1406 | /// SelectScalarSSELoad - Match a scalar SSE load. In particular, we want to |
| 1407 | /// match a load whose top elements are either undef or zeros. The load flavor |
| 1408 | /// is derived from the type of N, which is either v4f32 or v2f64. |
Chris Lattner | 3f48215 | 2010-02-17 06:07:47 +0000 | [diff] [blame] | 1409 | /// |
| 1410 | /// We also return: |
Chris Lattner | 18a32ce | 2010-02-21 03:17:59 +0000 | [diff] [blame] | 1411 | /// PatternChainNode: this is the matched node that has a chain input and |
| 1412 | /// output. |
Chris Lattner | bd6e193 | 2010-03-01 22:51:11 +0000 | [diff] [blame] | 1413 | bool X86DAGToDAGISel::SelectScalarSSELoad(SDNode *Root, |
Dan Gohman | 2ce6f2a | 2008-07-27 21:46:04 +0000 | [diff] [blame] | 1414 | SDValue N, SDValue &Base, |
| 1415 | SDValue &Scale, SDValue &Index, |
Rafael Espindola | 3b2df10 | 2009-04-08 21:14:34 +0000 | [diff] [blame] | 1416 | SDValue &Disp, SDValue &Segment, |
Chris Lattner | 18a32ce | 2010-02-21 03:17:59 +0000 | [diff] [blame] | 1417 | SDValue &PatternNodeWithChain) { |
Chris Lattner | 398195e | 2006-10-07 21:55:32 +0000 | [diff] [blame] | 1418 | if (N.getOpcode() == ISD::SCALAR_TO_VECTOR) { |
Chris Lattner | 18a32ce | 2010-02-21 03:17:59 +0000 | [diff] [blame] | 1419 | PatternNodeWithChain = N.getOperand(0); |
| 1420 | if (ISD::isNON_EXTLoad(PatternNodeWithChain.getNode()) && |
| 1421 | PatternNodeWithChain.hasOneUse() && |
Chris Lattner | 3c29aff | 2010-02-21 04:53:34 +0000 | [diff] [blame] | 1422 | IsProfitableToFold(N.getOperand(0), N.getNode(), Root) && |
Dan Gohman | 21cea8a | 2010-04-17 15:26:15 +0000 | [diff] [blame] | 1423 | IsLegalToFold(N.getOperand(0), N.getNode(), Root, OptLevel)) { |
Chris Lattner | 18a32ce | 2010-02-21 03:17:59 +0000 | [diff] [blame] | 1424 | LoadSDNode *LD = cast<LoadSDNode>(PatternNodeWithChain); |
Chris Lattner | d58d7c1 | 2010-09-21 22:07:31 +0000 | [diff] [blame] | 1425 | if (!SelectAddr(LD, LD->getBasePtr(), Base, Scale, Index, Disp, Segment)) |
Chris Lattner | 398195e | 2006-10-07 21:55:32 +0000 | [diff] [blame] | 1426 | return false; |
| 1427 | return true; |
| 1428 | } |
| 1429 | } |
Chris Lattner | d5fcfaa | 2006-10-11 22:09:58 +0000 | [diff] [blame] | 1430 | |
| 1431 | // Also handle the case where we explicitly require zeros in the top |
Chris Lattner | 398195e | 2006-10-07 21:55:32 +0000 | [diff] [blame] | 1432 | // elements. This is a vector shuffle from the zero vector. |
Gabor Greif | f304a7a | 2008-08-28 21:40:38 +0000 | [diff] [blame] | 1433 | if (N.getOpcode() == X86ISD::VZEXT_MOVL && N.getNode()->hasOneUse() && |
Chris Lattner | 5728bdd | 2007-11-25 00:24:49 +0000 | [diff] [blame] | 1434 | // Check to see if the top elements are all zeros (or bitcast of zeros). |
Chad Rosier | 24c19d2 | 2012-08-01 18:39:17 +0000 | [diff] [blame] | 1435 | N.getOperand(0).getOpcode() == ISD::SCALAR_TO_VECTOR && |
Gabor Greif | f304a7a | 2008-08-28 21:40:38 +0000 | [diff] [blame] | 1436 | N.getOperand(0).getNode()->hasOneUse() && |
| 1437 | ISD::isNON_EXTLoad(N.getOperand(0).getOperand(0).getNode()) && |
Chris Lattner | afac7dad | 2010-02-16 22:35:06 +0000 | [diff] [blame] | 1438 | N.getOperand(0).getOperand(0).hasOneUse() && |
| 1439 | IsProfitableToFold(N.getOperand(0), N.getNode(), Root) && |
Dan Gohman | 21cea8a | 2010-04-17 15:26:15 +0000 | [diff] [blame] | 1440 | IsLegalToFold(N.getOperand(0), N.getNode(), Root, OptLevel)) { |
Evan Cheng | 78af38c | 2008-05-08 00:57:18 +0000 | [diff] [blame] | 1441 | // Okay, this is a zero extending load. Fold it. |
| 1442 | LoadSDNode *LD = cast<LoadSDNode>(N.getOperand(0).getOperand(0)); |
Chris Lattner | d58d7c1 | 2010-09-21 22:07:31 +0000 | [diff] [blame] | 1443 | if (!SelectAddr(LD, LD->getBasePtr(), Base, Scale, Index, Disp, Segment)) |
Evan Cheng | 78af38c | 2008-05-08 00:57:18 +0000 | [diff] [blame] | 1444 | return false; |
Chris Lattner | 18a32ce | 2010-02-21 03:17:59 +0000 | [diff] [blame] | 1445 | PatternNodeWithChain = SDValue(LD, 0); |
Evan Cheng | 78af38c | 2008-05-08 00:57:18 +0000 | [diff] [blame] | 1446 | return true; |
Chris Lattner | d5fcfaa | 2006-10-11 22:09:58 +0000 | [diff] [blame] | 1447 | } |
Chris Lattner | 398195e | 2006-10-07 21:55:32 +0000 | [diff] [blame] | 1448 | return false; |
| 1449 | } |
| 1450 | |
| 1451 | |
Tim Northover | 3a1fd4c | 2013-06-01 09:55:14 +0000 | [diff] [blame] | 1452 | bool X86DAGToDAGISel::SelectMOV64Imm32(SDValue N, SDValue &Imm) { |
| 1453 | if (const ConstantSDNode *CN = dyn_cast<ConstantSDNode>(N)) { |
| 1454 | uint64_t ImmVal = CN->getZExtValue(); |
| 1455 | if ((uint32_t)ImmVal != (uint64_t)ImmVal) |
| 1456 | return false; |
| 1457 | |
Sergey Dmitrouk | 842a51b | 2015-04-28 14:05:47 +0000 | [diff] [blame] | 1458 | Imm = CurDAG->getTargetConstant(ImmVal, SDLoc(N), MVT::i64); |
Tim Northover | 3a1fd4c | 2013-06-01 09:55:14 +0000 | [diff] [blame] | 1459 | return true; |
| 1460 | } |
| 1461 | |
| 1462 | // In static codegen with small code model, we can get the address of a label |
| 1463 | // into a register with 'movl'. TableGen has already made sure we're looking |
| 1464 | // at a label of some kind. |
Tim Northover | 6833e3f | 2013-06-10 20:43:49 +0000 | [diff] [blame] | 1465 | assert(N->getOpcode() == X86ISD::Wrapper && |
| 1466 | "Unexpected node type for MOV32ri64"); |
Tim Northover | 3a1fd4c | 2013-06-01 09:55:14 +0000 | [diff] [blame] | 1467 | N = N.getOperand(0); |
| 1468 | |
| 1469 | if (N->getOpcode() != ISD::TargetConstantPool && |
| 1470 | N->getOpcode() != ISD::TargetJumpTable && |
| 1471 | N->getOpcode() != ISD::TargetGlobalAddress && |
| 1472 | N->getOpcode() != ISD::TargetExternalSymbol && |
| 1473 | N->getOpcode() != ISD::TargetBlockAddress) |
| 1474 | return false; |
| 1475 | |
| 1476 | Imm = N; |
| 1477 | return TM.getCodeModel() == CodeModel::Small; |
| 1478 | } |
| 1479 | |
Tim Northover | 6833e3f | 2013-06-10 20:43:49 +0000 | [diff] [blame] | 1480 | bool X86DAGToDAGISel::SelectLEA64_32Addr(SDValue N, SDValue &Base, |
| 1481 | SDValue &Scale, SDValue &Index, |
| 1482 | SDValue &Disp, SDValue &Segment) { |
| 1483 | if (!SelectLEAAddr(N, Base, Scale, Index, Disp, Segment)) |
| 1484 | return false; |
| 1485 | |
| 1486 | SDLoc DL(N); |
| 1487 | RegisterSDNode *RN = dyn_cast<RegisterSDNode>(Base); |
| 1488 | if (RN && RN->getReg() == 0) |
| 1489 | Base = CurDAG->getRegister(0, MVT::i64); |
Pavel Chupin | 01a4e0a | 2014-08-20 11:59:22 +0000 | [diff] [blame] | 1490 | else if (Base.getValueType() == MVT::i32 && !dyn_cast<FrameIndexSDNode>(Base)) { |
Tim Northover | 6833e3f | 2013-06-10 20:43:49 +0000 | [diff] [blame] | 1491 | // Base could already be %rip, particularly in the x32 ABI. |
| 1492 | Base = SDValue(CurDAG->getMachineNode( |
| 1493 | TargetOpcode::SUBREG_TO_REG, DL, MVT::i64, |
Sergey Dmitrouk | 842a51b | 2015-04-28 14:05:47 +0000 | [diff] [blame] | 1494 | CurDAG->getTargetConstant(0, DL, MVT::i64), |
Tim Northover | 6833e3f | 2013-06-10 20:43:49 +0000 | [diff] [blame] | 1495 | Base, |
Sergey Dmitrouk | 842a51b | 2015-04-28 14:05:47 +0000 | [diff] [blame] | 1496 | CurDAG->getTargetConstant(X86::sub_32bit, DL, MVT::i32)), |
Tim Northover | 6833e3f | 2013-06-10 20:43:49 +0000 | [diff] [blame] | 1497 | 0); |
| 1498 | } |
| 1499 | |
| 1500 | RN = dyn_cast<RegisterSDNode>(Index); |
| 1501 | if (RN && RN->getReg() == 0) |
| 1502 | Index = CurDAG->getRegister(0, MVT::i64); |
| 1503 | else { |
| 1504 | assert(Index.getValueType() == MVT::i32 && |
| 1505 | "Expect to be extending 32-bit registers for use in LEA"); |
| 1506 | Index = SDValue(CurDAG->getMachineNode( |
| 1507 | TargetOpcode::SUBREG_TO_REG, DL, MVT::i64, |
Sergey Dmitrouk | 842a51b | 2015-04-28 14:05:47 +0000 | [diff] [blame] | 1508 | CurDAG->getTargetConstant(0, DL, MVT::i64), |
Tim Northover | 6833e3f | 2013-06-10 20:43:49 +0000 | [diff] [blame] | 1509 | Index, |
Sergey Dmitrouk | 842a51b | 2015-04-28 14:05:47 +0000 | [diff] [blame] | 1510 | CurDAG->getTargetConstant(X86::sub_32bit, DL, |
| 1511 | MVT::i32)), |
Tim Northover | 6833e3f | 2013-06-10 20:43:49 +0000 | [diff] [blame] | 1512 | 0); |
| 1513 | } |
| 1514 | |
| 1515 | return true; |
| 1516 | } |
| 1517 | |
Evan Cheng | 77d86ff | 2006-02-25 10:09:08 +0000 | [diff] [blame] | 1518 | /// SelectLEAAddr - it calls SelectAddr and determines if the maximal addressing |
| 1519 | /// mode it matches can be cost effectively emitted as an LEA instruction. |
Chris Lattner | 0e023ea | 2010-09-21 20:31:19 +0000 | [diff] [blame] | 1520 | bool X86DAGToDAGISel::SelectLEAAddr(SDValue N, |
Dan Gohman | 2ce6f2a | 2008-07-27 21:46:04 +0000 | [diff] [blame] | 1521 | SDValue &Base, SDValue &Scale, |
Chris Lattner | f469307 | 2010-07-08 23:46:44 +0000 | [diff] [blame] | 1522 | SDValue &Index, SDValue &Disp, |
| 1523 | SDValue &Segment) { |
Evan Cheng | 77d86ff | 2006-02-25 10:09:08 +0000 | [diff] [blame] | 1524 | X86ISelAddressMode AM; |
Rafael Espindola | bb834f0 | 2009-04-10 10:09:34 +0000 | [diff] [blame] | 1525 | |
| 1526 | // Set AM.Segment to prevent MatchAddress from using one. LEA doesn't support |
| 1527 | // segments. |
| 1528 | SDValue Copy = AM.Segment; |
Owen Anderson | 9f94459 | 2009-08-11 20:47:22 +0000 | [diff] [blame] | 1529 | SDValue T = CurDAG->getRegister(0, MVT::i32); |
Rafael Espindola | bb834f0 | 2009-04-10 10:09:34 +0000 | [diff] [blame] | 1530 | AM.Segment = T; |
Evan Cheng | 77d86ff | 2006-02-25 10:09:08 +0000 | [diff] [blame] | 1531 | if (MatchAddress(N, AM)) |
| 1532 | return false; |
Rafael Espindola | bb834f0 | 2009-04-10 10:09:34 +0000 | [diff] [blame] | 1533 | assert (T == AM.Segment); |
| 1534 | AM.Segment = Copy; |
Rafael Espindola | 3b2df10 | 2009-04-08 21:14:34 +0000 | [diff] [blame] | 1535 | |
Craig Topper | 83e042a | 2013-08-15 05:57:07 +0000 | [diff] [blame] | 1536 | MVT VT = N.getSimpleValueType(); |
Evan Cheng | 77d86ff | 2006-02-25 10:09:08 +0000 | [diff] [blame] | 1537 | unsigned Complexity = 0; |
| 1538 | if (AM.BaseType == X86ISelAddressMode::RegBase) |
Dan Gohman | 0fd54fb | 2010-04-29 23:30:41 +0000 | [diff] [blame] | 1539 | if (AM.Base_Reg.getNode()) |
Evan Cheng | 77d86ff | 2006-02-25 10:09:08 +0000 | [diff] [blame] | 1540 | Complexity = 1; |
| 1541 | else |
Dan Gohman | 0fd54fb | 2010-04-29 23:30:41 +0000 | [diff] [blame] | 1542 | AM.Base_Reg = CurDAG->getRegister(0, VT); |
Evan Cheng | 77d86ff | 2006-02-25 10:09:08 +0000 | [diff] [blame] | 1543 | else if (AM.BaseType == X86ISelAddressMode::FrameIndexBase) |
| 1544 | Complexity = 4; |
| 1545 | |
Gabor Greif | f304a7a | 2008-08-28 21:40:38 +0000 | [diff] [blame] | 1546 | if (AM.IndexReg.getNode()) |
Evan Cheng | 77d86ff | 2006-02-25 10:09:08 +0000 | [diff] [blame] | 1547 | Complexity++; |
| 1548 | else |
Evan Cheng | 11b0a5d | 2006-09-08 06:48:29 +0000 | [diff] [blame] | 1549 | AM.IndexReg = CurDAG->getRegister(0, VT); |
Evan Cheng | 77d86ff | 2006-02-25 10:09:08 +0000 | [diff] [blame] | 1550 | |
Chris Lattner | 3e1d917 | 2007-03-20 06:08:29 +0000 | [diff] [blame] | 1551 | // Don't match just leal(,%reg,2). It's cheaper to do addl %reg, %reg, or with |
| 1552 | // a simple shift. |
| 1553 | if (AM.Scale > 1) |
Evan Cheng | 990c360 | 2006-02-28 21:13:57 +0000 | [diff] [blame] | 1554 | Complexity++; |
Evan Cheng | 77d86ff | 2006-02-25 10:09:08 +0000 | [diff] [blame] | 1555 | |
| 1556 | // FIXME: We are artificially lowering the criteria to turn ADD %reg, $GA |
| 1557 | // to a LEA. This is determined with some expermentation but is by no means |
| 1558 | // optimal (especially for code size consideration). LEA is nice because of |
| 1559 | // its three-address nature. Tweak the cost function again when we can run |
| 1560 | // convertToThreeAddress() at register allocation time. |
Dan Gohman | 4e3e3de | 2009-02-07 00:43:41 +0000 | [diff] [blame] | 1561 | if (AM.hasSymbolicDisplacement()) { |
Evan Cheng | 11b0a5d | 2006-09-08 06:48:29 +0000 | [diff] [blame] | 1562 | // For X86-64, we should always use lea to materialize RIP relative |
| 1563 | // addresses. |
Evan Cheng | 47e181c | 2006-12-05 22:03:40 +0000 | [diff] [blame] | 1564 | if (Subtarget->is64Bit()) |
Evan Cheng | 11b0a5d | 2006-09-08 06:48:29 +0000 | [diff] [blame] | 1565 | Complexity = 4; |
| 1566 | else |
| 1567 | Complexity += 2; |
| 1568 | } |
Evan Cheng | 77d86ff | 2006-02-25 10:09:08 +0000 | [diff] [blame] | 1569 | |
Dan Gohman | 0fd54fb | 2010-04-29 23:30:41 +0000 | [diff] [blame] | 1570 | if (AM.Disp && (AM.Base_Reg.getNode() || AM.IndexReg.getNode())) |
Evan Cheng | 77d86ff | 2006-02-25 10:09:08 +0000 | [diff] [blame] | 1571 | Complexity++; |
| 1572 | |
Chris Lattner | 4d10f1a | 2009-07-11 22:50:33 +0000 | [diff] [blame] | 1573 | // If it isn't worth using an LEA, reject it. |
Chris Lattner | 48cee9b | 2009-07-11 23:07:30 +0000 | [diff] [blame] | 1574 | if (Complexity <= 2) |
Chris Lattner | 4d10f1a | 2009-07-11 22:50:33 +0000 | [diff] [blame] | 1575 | return false; |
Chad Rosier | 24c19d2 | 2012-08-01 18:39:17 +0000 | [diff] [blame] | 1576 | |
Sergey Dmitrouk | 842a51b | 2015-04-28 14:05:47 +0000 | [diff] [blame] | 1577 | getAddressOperands(AM, SDLoc(N), Base, Scale, Index, Disp, Segment); |
Chris Lattner | 4d10f1a | 2009-07-11 22:50:33 +0000 | [diff] [blame] | 1578 | return true; |
Evan Cheng | 77d86ff | 2006-02-25 10:09:08 +0000 | [diff] [blame] | 1579 | } |
| 1580 | |
Chris Lattner | 7d2b049 | 2009-06-20 20:38:48 +0000 | [diff] [blame] | 1581 | /// SelectTLSADDRAddr - This is only run on TargetGlobalTLSAddress nodes. |
Chris Lattner | 0e023ea | 2010-09-21 20:31:19 +0000 | [diff] [blame] | 1582 | bool X86DAGToDAGISel::SelectTLSADDRAddr(SDValue N, SDValue &Base, |
Chris Lattner | 7d2b049 | 2009-06-20 20:38:48 +0000 | [diff] [blame] | 1583 | SDValue &Scale, SDValue &Index, |
Chris Lattner | f469307 | 2010-07-08 23:46:44 +0000 | [diff] [blame] | 1584 | SDValue &Disp, SDValue &Segment) { |
Chris Lattner | 7d2b049 | 2009-06-20 20:38:48 +0000 | [diff] [blame] | 1585 | assert(N.getOpcode() == ISD::TargetGlobalTLSAddress); |
| 1586 | const GlobalAddressSDNode *GA = cast<GlobalAddressSDNode>(N); |
Chad Rosier | 24c19d2 | 2012-08-01 18:39:17 +0000 | [diff] [blame] | 1587 | |
Chris Lattner | 7d2b049 | 2009-06-20 20:38:48 +0000 | [diff] [blame] | 1588 | X86ISelAddressMode AM; |
| 1589 | AM.GV = GA->getGlobal(); |
| 1590 | AM.Disp += GA->getOffset(); |
Dan Gohman | 0fd54fb | 2010-04-29 23:30:41 +0000 | [diff] [blame] | 1591 | AM.Base_Reg = CurDAG->getRegister(0, N.getValueType()); |
Chris Lattner | 899abc4 | 2009-06-26 21:18:37 +0000 | [diff] [blame] | 1592 | AM.SymbolFlags = GA->getTargetFlags(); |
| 1593 | |
Owen Anderson | 9f94459 | 2009-08-11 20:47:22 +0000 | [diff] [blame] | 1594 | if (N.getValueType() == MVT::i32) { |
Chris Lattner | 7d2b049 | 2009-06-20 20:38:48 +0000 | [diff] [blame] | 1595 | AM.Scale = 1; |
Owen Anderson | 9f94459 | 2009-08-11 20:47:22 +0000 | [diff] [blame] | 1596 | AM.IndexReg = CurDAG->getRegister(X86::EBX, MVT::i32); |
Chris Lattner | 7d2b049 | 2009-06-20 20:38:48 +0000 | [diff] [blame] | 1597 | } else { |
Owen Anderson | 9f94459 | 2009-08-11 20:47:22 +0000 | [diff] [blame] | 1598 | AM.IndexReg = CurDAG->getRegister(0, MVT::i64); |
Chris Lattner | 7d2b049 | 2009-06-20 20:38:48 +0000 | [diff] [blame] | 1599 | } |
Chad Rosier | 24c19d2 | 2012-08-01 18:39:17 +0000 | [diff] [blame] | 1600 | |
Sergey Dmitrouk | 842a51b | 2015-04-28 14:05:47 +0000 | [diff] [blame] | 1601 | getAddressOperands(AM, SDLoc(N), Base, Scale, Index, Disp, Segment); |
Chris Lattner | 7d2b049 | 2009-06-20 20:38:48 +0000 | [diff] [blame] | 1602 | return true; |
| 1603 | } |
| 1604 | |
| 1605 | |
Dan Gohman | ea6f91f | 2010-01-05 01:24:18 +0000 | [diff] [blame] | 1606 | bool X86DAGToDAGISel::TryFoldLoad(SDNode *P, SDValue N, |
Dan Gohman | 2ce6f2a | 2008-07-27 21:46:04 +0000 | [diff] [blame] | 1607 | SDValue &Base, SDValue &Scale, |
Rafael Espindola | 3b2df10 | 2009-04-08 21:14:34 +0000 | [diff] [blame] | 1608 | SDValue &Index, SDValue &Disp, |
| 1609 | SDValue &Segment) { |
Chris Lattner | dd03070 | 2010-03-02 22:20:06 +0000 | [diff] [blame] | 1610 | if (!ISD::isNON_EXTLoad(N.getNode()) || |
| 1611 | !IsProfitableToFold(N, P, P) || |
Dan Gohman | 21cea8a | 2010-04-17 15:26:15 +0000 | [diff] [blame] | 1612 | !IsLegalToFold(N, P, P, OptLevel)) |
Chris Lattner | dd03070 | 2010-03-02 22:20:06 +0000 | [diff] [blame] | 1613 | return false; |
Chad Rosier | 24c19d2 | 2012-08-01 18:39:17 +0000 | [diff] [blame] | 1614 | |
Chris Lattner | d58d7c1 | 2010-09-21 22:07:31 +0000 | [diff] [blame] | 1615 | return SelectAddr(N.getNode(), |
| 1616 | N.getOperand(1), Base, Scale, Index, Disp, Segment); |
Evan Cheng | 10d2790 | 2006-01-06 20:36:21 +0000 | [diff] [blame] | 1617 | } |
| 1618 | |
Dan Gohman | 2430073 | 2008-09-23 18:22:58 +0000 | [diff] [blame] | 1619 | /// getGlobalBaseReg - Return an SDNode that returns the value of |
| 1620 | /// the global base register. Output instructions required to |
| 1621 | /// initialize the global base register, if necessary. |
Evan Cheng | 5588de9 | 2006-02-18 00:15:05 +0000 | [diff] [blame] | 1622 | /// |
Evan Cheng | 61413a3 | 2006-08-26 05:34:46 +0000 | [diff] [blame] | 1623 | SDNode *X86DAGToDAGISel::getGlobalBaseReg() { |
Dan Gohman | 4751bb9 | 2009-06-03 20:20:00 +0000 | [diff] [blame] | 1624 | unsigned GlobalBaseReg = getInstrInfo()->getGlobalBaseReg(MF); |
Eric Christopher | b17140d | 2014-10-08 07:32:17 +0000 | [diff] [blame] | 1625 | return CurDAG->getRegister(GlobalBaseReg, TLI->getPointerTy()).getNode(); |
Evan Cheng | 5588de9 | 2006-02-18 00:15:05 +0000 | [diff] [blame] | 1626 | } |
| 1627 | |
Michael Liao | 8372539 | 2012-09-19 19:36:58 +0000 | [diff] [blame] | 1628 | /// Atomic opcode table |
| 1629 | /// |
Eric Christopher | eb47a2a | 2011-05-17 07:47:55 +0000 | [diff] [blame] | 1630 | enum AtomicOpc { |
Michael Liao | 8372539 | 2012-09-19 19:36:58 +0000 | [diff] [blame] | 1631 | ADD, |
| 1632 | SUB, |
| 1633 | INC, |
| 1634 | DEC, |
Eric Christopher | abfe313 | 2011-05-17 07:50:41 +0000 | [diff] [blame] | 1635 | OR, |
Eric Christopher | a1d9e29 | 2011-05-17 08:10:18 +0000 | [diff] [blame] | 1636 | AND, |
| 1637 | XOR, |
Eric Christopher | abfe313 | 2011-05-17 07:50:41 +0000 | [diff] [blame] | 1638 | AtomicOpcEnd |
Eric Christopher | eb47a2a | 2011-05-17 07:47:55 +0000 | [diff] [blame] | 1639 | }; |
| 1640 | |
| 1641 | enum AtomicSz { |
| 1642 | ConstantI8, |
| 1643 | I8, |
| 1644 | SextConstantI16, |
| 1645 | ConstantI16, |
| 1646 | I16, |
| 1647 | SextConstantI32, |
| 1648 | ConstantI32, |
| 1649 | I32, |
| 1650 | SextConstantI64, |
| 1651 | ConstantI64, |
Eric Christopher | abfe313 | 2011-05-17 07:50:41 +0000 | [diff] [blame] | 1652 | I64, |
| 1653 | AtomicSzEnd |
Eric Christopher | eb47a2a | 2011-05-17 07:47:55 +0000 | [diff] [blame] | 1654 | }; |
| 1655 | |
Craig Topper | 2dac962 | 2012-03-09 07:45:21 +0000 | [diff] [blame] | 1656 | static const uint16_t AtomicOpcTbl[AtomicOpcEnd][AtomicSzEnd] = { |
Eric Christopher | 2a9dbbb | 2011-05-11 21:44:58 +0000 | [diff] [blame] | 1657 | { |
Michael Liao | 8372539 | 2012-09-19 19:36:58 +0000 | [diff] [blame] | 1658 | X86::LOCK_ADD8mi, |
| 1659 | X86::LOCK_ADD8mr, |
| 1660 | X86::LOCK_ADD16mi8, |
| 1661 | X86::LOCK_ADD16mi, |
| 1662 | X86::LOCK_ADD16mr, |
| 1663 | X86::LOCK_ADD32mi8, |
| 1664 | X86::LOCK_ADD32mi, |
| 1665 | X86::LOCK_ADD32mr, |
| 1666 | X86::LOCK_ADD64mi8, |
| 1667 | X86::LOCK_ADD64mi32, |
| 1668 | X86::LOCK_ADD64mr, |
| 1669 | }, |
| 1670 | { |
| 1671 | X86::LOCK_SUB8mi, |
| 1672 | X86::LOCK_SUB8mr, |
| 1673 | X86::LOCK_SUB16mi8, |
| 1674 | X86::LOCK_SUB16mi, |
| 1675 | X86::LOCK_SUB16mr, |
| 1676 | X86::LOCK_SUB32mi8, |
| 1677 | X86::LOCK_SUB32mi, |
| 1678 | X86::LOCK_SUB32mr, |
| 1679 | X86::LOCK_SUB64mi8, |
| 1680 | X86::LOCK_SUB64mi32, |
| 1681 | X86::LOCK_SUB64mr, |
| 1682 | }, |
| 1683 | { |
| 1684 | 0, |
| 1685 | X86::LOCK_INC8m, |
| 1686 | 0, |
| 1687 | 0, |
| 1688 | X86::LOCK_INC16m, |
| 1689 | 0, |
| 1690 | 0, |
| 1691 | X86::LOCK_INC32m, |
| 1692 | 0, |
| 1693 | 0, |
| 1694 | X86::LOCK_INC64m, |
| 1695 | }, |
| 1696 | { |
| 1697 | 0, |
| 1698 | X86::LOCK_DEC8m, |
| 1699 | 0, |
| 1700 | 0, |
| 1701 | X86::LOCK_DEC16m, |
| 1702 | 0, |
| 1703 | 0, |
| 1704 | X86::LOCK_DEC32m, |
| 1705 | 0, |
| 1706 | 0, |
| 1707 | X86::LOCK_DEC64m, |
| 1708 | }, |
| 1709 | { |
Eric Christopher | 2a9dbbb | 2011-05-11 21:44:58 +0000 | [diff] [blame] | 1710 | X86::LOCK_OR8mi, |
| 1711 | X86::LOCK_OR8mr, |
| 1712 | X86::LOCK_OR16mi8, |
| 1713 | X86::LOCK_OR16mi, |
| 1714 | X86::LOCK_OR16mr, |
| 1715 | X86::LOCK_OR32mi8, |
| 1716 | X86::LOCK_OR32mi, |
| 1717 | X86::LOCK_OR32mr, |
| 1718 | X86::LOCK_OR64mi8, |
| 1719 | X86::LOCK_OR64mi32, |
Michael Liao | 8372539 | 2012-09-19 19:36:58 +0000 | [diff] [blame] | 1720 | X86::LOCK_OR64mr, |
Eric Christopher | a1d9e29 | 2011-05-17 08:10:18 +0000 | [diff] [blame] | 1721 | }, |
| 1722 | { |
| 1723 | X86::LOCK_AND8mi, |
| 1724 | X86::LOCK_AND8mr, |
| 1725 | X86::LOCK_AND16mi8, |
| 1726 | X86::LOCK_AND16mi, |
| 1727 | X86::LOCK_AND16mr, |
| 1728 | X86::LOCK_AND32mi8, |
| 1729 | X86::LOCK_AND32mi, |
| 1730 | X86::LOCK_AND32mr, |
| 1731 | X86::LOCK_AND64mi8, |
| 1732 | X86::LOCK_AND64mi32, |
Michael Liao | 8372539 | 2012-09-19 19:36:58 +0000 | [diff] [blame] | 1733 | X86::LOCK_AND64mr, |
Eric Christopher | a1d9e29 | 2011-05-17 08:10:18 +0000 | [diff] [blame] | 1734 | }, |
| 1735 | { |
| 1736 | X86::LOCK_XOR8mi, |
| 1737 | X86::LOCK_XOR8mr, |
| 1738 | X86::LOCK_XOR16mi8, |
| 1739 | X86::LOCK_XOR16mi, |
| 1740 | X86::LOCK_XOR16mr, |
| 1741 | X86::LOCK_XOR32mi8, |
| 1742 | X86::LOCK_XOR32mi, |
| 1743 | X86::LOCK_XOR32mr, |
| 1744 | X86::LOCK_XOR64mi8, |
| 1745 | X86::LOCK_XOR64mi32, |
Michael Liao | 8372539 | 2012-09-19 19:36:58 +0000 | [diff] [blame] | 1746 | X86::LOCK_XOR64mr, |
Eric Christopher | 2a9dbbb | 2011-05-11 21:44:58 +0000 | [diff] [blame] | 1747 | } |
| 1748 | }; |
| 1749 | |
Michael Liao | 8372539 | 2012-09-19 19:36:58 +0000 | [diff] [blame] | 1750 | // Return the target constant operand for atomic-load-op and do simple |
| 1751 | // translations, such as from atomic-load-add to lock-sub. The return value is |
| 1752 | // one of the following 3 cases: |
| 1753 | // + target-constant, the operand could be supported as a target constant. |
| 1754 | // + empty, the operand is not needed any more with the new op selected. |
| 1755 | // + non-empty, otherwise. |
| 1756 | static SDValue getAtomicLoadArithTargetConstant(SelectionDAG *CurDAG, |
Andrew Trick | ef9de2a | 2013-05-25 02:42:55 +0000 | [diff] [blame] | 1757 | SDLoc dl, |
Craig Topper | 83e042a | 2013-08-15 05:57:07 +0000 | [diff] [blame] | 1758 | enum AtomicOpc &Op, MVT NVT, |
Robin Morisset | 6f3d04e | 2014-10-08 23:16:23 +0000 | [diff] [blame] | 1759 | SDValue Val, |
| 1760 | const X86Subtarget *Subtarget) { |
Michael Liao | 8372539 | 2012-09-19 19:36:58 +0000 | [diff] [blame] | 1761 | if (ConstantSDNode *CN = dyn_cast<ConstantSDNode>(Val)) { |
| 1762 | int64_t CNVal = CN->getSExtValue(); |
| 1763 | // Quit if not 32-bit imm. |
| 1764 | if ((int32_t)CNVal != CNVal) |
| 1765 | return Val; |
Robin Morisset | 880580b | 2014-10-07 23:53:57 +0000 | [diff] [blame] | 1766 | // Quit if INT32_MIN: it would be negated as it is negative and overflow, |
| 1767 | // producing an immediate that does not fit in the 32 bits available for |
| 1768 | // an immediate operand to sub. However, it still fits in 32 bits for the |
| 1769 | // add (since it is not negated) so we can return target-constant. |
| 1770 | if (CNVal == INT32_MIN) |
Sergey Dmitrouk | 842a51b | 2015-04-28 14:05:47 +0000 | [diff] [blame] | 1771 | return CurDAG->getTargetConstant(CNVal, dl, NVT); |
Michael Liao | 8372539 | 2012-09-19 19:36:58 +0000 | [diff] [blame] | 1772 | // For atomic-load-add, we could do some optimizations. |
| 1773 | if (Op == ADD) { |
| 1774 | // Translate to INC/DEC if ADD by 1 or -1. |
Robin Morisset | 6f3d04e | 2014-10-08 23:16:23 +0000 | [diff] [blame] | 1775 | if (((CNVal == 1) || (CNVal == -1)) && !Subtarget->slowIncDec()) { |
Michael Liao | 8372539 | 2012-09-19 19:36:58 +0000 | [diff] [blame] | 1776 | Op = (CNVal == 1) ? INC : DEC; |
| 1777 | // No more constant operand after being translated into INC/DEC. |
| 1778 | return SDValue(); |
| 1779 | } |
| 1780 | // Translate to SUB if ADD by negative value. |
| 1781 | if (CNVal < 0) { |
| 1782 | Op = SUB; |
| 1783 | CNVal = -CNVal; |
| 1784 | } |
| 1785 | } |
Sergey Dmitrouk | 842a51b | 2015-04-28 14:05:47 +0000 | [diff] [blame] | 1786 | return CurDAG->getTargetConstant(CNVal, dl, NVT); |
Michael Liao | 8372539 | 2012-09-19 19:36:58 +0000 | [diff] [blame] | 1787 | } |
| 1788 | |
| 1789 | // If the value operand is single-used, try to optimize it. |
| 1790 | if (Op == ADD && Val.hasOneUse()) { |
| 1791 | // Translate (atomic-load-add ptr (sub 0 x)) back to (lock-sub x). |
| 1792 | if (Val.getOpcode() == ISD::SUB && X86::isZeroNode(Val.getOperand(0))) { |
| 1793 | Op = SUB; |
| 1794 | return Val.getOperand(1); |
| 1795 | } |
| 1796 | // A special case for i16, which needs truncating as, in most cases, it's |
| 1797 | // promoted to i32. We will translate |
| 1798 | // (atomic-load-add (truncate (sub 0 x))) to (lock-sub (EXTRACT_SUBREG x)) |
| 1799 | if (Val.getOpcode() == ISD::TRUNCATE && NVT == MVT::i16 && |
| 1800 | Val.getOperand(0).getOpcode() == ISD::SUB && |
| 1801 | X86::isZeroNode(Val.getOperand(0).getOperand(0))) { |
| 1802 | Op = SUB; |
| 1803 | Val = Val.getOperand(0); |
| 1804 | return CurDAG->getTargetExtractSubreg(X86::sub_16bit, dl, NVT, |
| 1805 | Val.getOperand(1)); |
| 1806 | } |
| 1807 | } |
| 1808 | |
| 1809 | return Val; |
| 1810 | } |
| 1811 | |
Craig Topper | 83e042a | 2013-08-15 05:57:07 +0000 | [diff] [blame] | 1812 | SDNode *X86DAGToDAGISel::SelectAtomicLoadArith(SDNode *Node, MVT NVT) { |
Eric Christopher | 4a34e61 | 2011-05-10 23:57:45 +0000 | [diff] [blame] | 1813 | if (Node->hasAnyUseOfValue(0)) |
Craig Topper | 062a2ba | 2014-04-25 05:30:21 +0000 | [diff] [blame] | 1814 | return nullptr; |
Chad Rosier | 24c19d2 | 2012-08-01 18:39:17 +0000 | [diff] [blame] | 1815 | |
Andrew Trick | ef9de2a | 2013-05-25 02:42:55 +0000 | [diff] [blame] | 1816 | SDLoc dl(Node); |
Michael Liao | 8372539 | 2012-09-19 19:36:58 +0000 | [diff] [blame] | 1817 | |
Eric Christopher | 56a42eb | 2011-05-17 08:16:14 +0000 | [diff] [blame] | 1818 | // Optimize common patterns for __sync_or_and_fetch and similar arith |
| 1819 | // operations where the result is not used. This allows us to use the "lock" |
| 1820 | // version of the arithmetic instruction. |
Eric Christopher | 4a34e61 | 2011-05-10 23:57:45 +0000 | [diff] [blame] | 1821 | SDValue Chain = Node->getOperand(0); |
| 1822 | SDValue Ptr = Node->getOperand(1); |
| 1823 | SDValue Val = Node->getOperand(2); |
Robin Morisset | 5ce0ce4 | 2014-08-29 20:19:23 +0000 | [diff] [blame] | 1824 | SDValue Base, Scale, Index, Disp, Segment; |
| 1825 | if (!SelectAddr(Node, Ptr, Base, Scale, Index, Disp, Segment)) |
Craig Topper | 062a2ba | 2014-04-25 05:30:21 +0000 | [diff] [blame] | 1826 | return nullptr; |
Eric Christopher | 4a34e61 | 2011-05-10 23:57:45 +0000 | [diff] [blame] | 1827 | |
Eric Christopher | a1d9e29 | 2011-05-17 08:10:18 +0000 | [diff] [blame] | 1828 | // Which index into the table. |
| 1829 | enum AtomicOpc Op; |
| 1830 | switch (Node->getOpcode()) { |
Michael Liao | 8372539 | 2012-09-19 19:36:58 +0000 | [diff] [blame] | 1831 | default: |
Craig Topper | 062a2ba | 2014-04-25 05:30:21 +0000 | [diff] [blame] | 1832 | return nullptr; |
Eric Christopher | a1d9e29 | 2011-05-17 08:10:18 +0000 | [diff] [blame] | 1833 | case ISD::ATOMIC_LOAD_OR: |
| 1834 | Op = OR; |
| 1835 | break; |
| 1836 | case ISD::ATOMIC_LOAD_AND: |
| 1837 | Op = AND; |
| 1838 | break; |
| 1839 | case ISD::ATOMIC_LOAD_XOR: |
| 1840 | Op = XOR; |
| 1841 | break; |
Michael Liao | 8372539 | 2012-09-19 19:36:58 +0000 | [diff] [blame] | 1842 | case ISD::ATOMIC_LOAD_ADD: |
| 1843 | Op = ADD; |
| 1844 | break; |
Eric Christopher | a1d9e29 | 2011-05-17 08:10:18 +0000 | [diff] [blame] | 1845 | } |
Andrew Trick | 52b8387 | 2013-04-13 06:07:36 +0000 | [diff] [blame] | 1846 | |
Robin Morisset | 6f3d04e | 2014-10-08 23:16:23 +0000 | [diff] [blame] | 1847 | Val = getAtomicLoadArithTargetConstant(CurDAG, dl, Op, NVT, Val, Subtarget); |
Michael Liao | 8372539 | 2012-09-19 19:36:58 +0000 | [diff] [blame] | 1848 | bool isUnOp = !Val.getNode(); |
| 1849 | bool isCN = Val.getNode() && (Val.getOpcode() == ISD::TargetConstant); |
Chad Rosier | 24c19d2 | 2012-08-01 18:39:17 +0000 | [diff] [blame] | 1850 | |
Eric Christopher | 4a34e61 | 2011-05-10 23:57:45 +0000 | [diff] [blame] | 1851 | unsigned Opc = 0; |
Craig Topper | 83e042a | 2013-08-15 05:57:07 +0000 | [diff] [blame] | 1852 | switch (NVT.SimpleTy) { |
Craig Topper | 062a2ba | 2014-04-25 05:30:21 +0000 | [diff] [blame] | 1853 | default: return nullptr; |
Eric Christopher | 4a34e61 | 2011-05-10 23:57:45 +0000 | [diff] [blame] | 1854 | case MVT::i8: |
| 1855 | if (isCN) |
Eric Christopher | eb47a2a | 2011-05-17 07:47:55 +0000 | [diff] [blame] | 1856 | Opc = AtomicOpcTbl[Op][ConstantI8]; |
Eric Christopher | 4a34e61 | 2011-05-10 23:57:45 +0000 | [diff] [blame] | 1857 | else |
Eric Christopher | eb47a2a | 2011-05-17 07:47:55 +0000 | [diff] [blame] | 1858 | Opc = AtomicOpcTbl[Op][I8]; |
Eric Christopher | 4a34e61 | 2011-05-10 23:57:45 +0000 | [diff] [blame] | 1859 | break; |
| 1860 | case MVT::i16: |
| 1861 | if (isCN) { |
| 1862 | if (immSext8(Val.getNode())) |
Eric Christopher | eb47a2a | 2011-05-17 07:47:55 +0000 | [diff] [blame] | 1863 | Opc = AtomicOpcTbl[Op][SextConstantI16]; |
Eric Christopher | 4a34e61 | 2011-05-10 23:57:45 +0000 | [diff] [blame] | 1864 | else |
Eric Christopher | eb47a2a | 2011-05-17 07:47:55 +0000 | [diff] [blame] | 1865 | Opc = AtomicOpcTbl[Op][ConstantI16]; |
Eric Christopher | 4a34e61 | 2011-05-10 23:57:45 +0000 | [diff] [blame] | 1866 | } else |
Eric Christopher | eb47a2a | 2011-05-17 07:47:55 +0000 | [diff] [blame] | 1867 | Opc = AtomicOpcTbl[Op][I16]; |
Eric Christopher | 4a34e61 | 2011-05-10 23:57:45 +0000 | [diff] [blame] | 1868 | break; |
| 1869 | case MVT::i32: |
| 1870 | if (isCN) { |
| 1871 | if (immSext8(Val.getNode())) |
Eric Christopher | eb47a2a | 2011-05-17 07:47:55 +0000 | [diff] [blame] | 1872 | Opc = AtomicOpcTbl[Op][SextConstantI32]; |
Eric Christopher | 4a34e61 | 2011-05-10 23:57:45 +0000 | [diff] [blame] | 1873 | else |
Eric Christopher | eb47a2a | 2011-05-17 07:47:55 +0000 | [diff] [blame] | 1874 | Opc = AtomicOpcTbl[Op][ConstantI32]; |
Eric Christopher | 4a34e61 | 2011-05-10 23:57:45 +0000 | [diff] [blame] | 1875 | } else |
Eric Christopher | eb47a2a | 2011-05-17 07:47:55 +0000 | [diff] [blame] | 1876 | Opc = AtomicOpcTbl[Op][I32]; |
Eric Christopher | 4a34e61 | 2011-05-10 23:57:45 +0000 | [diff] [blame] | 1877 | break; |
| 1878 | case MVT::i64: |
| 1879 | if (isCN) { |
| 1880 | if (immSext8(Val.getNode())) |
Eric Christopher | eb47a2a | 2011-05-17 07:47:55 +0000 | [diff] [blame] | 1881 | Opc = AtomicOpcTbl[Op][SextConstantI64]; |
Eric Christopher | 4a34e61 | 2011-05-10 23:57:45 +0000 | [diff] [blame] | 1882 | else if (i64immSExt32(Val.getNode())) |
Eric Christopher | eb47a2a | 2011-05-17 07:47:55 +0000 | [diff] [blame] | 1883 | Opc = AtomicOpcTbl[Op][ConstantI64]; |
Robin Morisset | 880580b | 2014-10-07 23:53:57 +0000 | [diff] [blame] | 1884 | else |
| 1885 | llvm_unreachable("True 64 bits constant in SelectAtomicLoadArith"); |
Robin Morisset | 5ce0ce4 | 2014-08-29 20:19:23 +0000 | [diff] [blame] | 1886 | } else |
| 1887 | Opc = AtomicOpcTbl[Op][I64]; |
Eric Christopher | 4a34e61 | 2011-05-10 23:57:45 +0000 | [diff] [blame] | 1888 | break; |
| 1889 | } |
Chad Rosier | 24c19d2 | 2012-08-01 18:39:17 +0000 | [diff] [blame] | 1890 | |
Eric Christopher | c9321737 | 2011-06-30 00:48:30 +0000 | [diff] [blame] | 1891 | assert(Opc != 0 && "Invalid arith lock transform!"); |
| 1892 | |
Robin Morisset | 5ce0ce4 | 2014-08-29 20:19:23 +0000 | [diff] [blame] | 1893 | // Building the new node. |
Michael Liao | 8372539 | 2012-09-19 19:36:58 +0000 | [diff] [blame] | 1894 | SDValue Ret; |
Michael Liao | 8372539 | 2012-09-19 19:36:58 +0000 | [diff] [blame] | 1895 | if (isUnOp) { |
Robin Morisset | 5ce0ce4 | 2014-08-29 20:19:23 +0000 | [diff] [blame] | 1896 | SDValue Ops[] = { Base, Scale, Index, Disp, Segment, Chain }; |
Michael Liao | b53d896 | 2013-04-19 22:22:57 +0000 | [diff] [blame] | 1897 | Ret = SDValue(CurDAG->getMachineNode(Opc, dl, MVT::Other, Ops), 0); |
Michael Liao | 8372539 | 2012-09-19 19:36:58 +0000 | [diff] [blame] | 1898 | } else { |
Robin Morisset | 5ce0ce4 | 2014-08-29 20:19:23 +0000 | [diff] [blame] | 1899 | SDValue Ops[] = { Base, Scale, Index, Disp, Segment, Val, Chain }; |
Michael Liao | b53d896 | 2013-04-19 22:22:57 +0000 | [diff] [blame] | 1900 | Ret = SDValue(CurDAG->getMachineNode(Opc, dl, MVT::Other, Ops), 0); |
Michael Liao | 8372539 | 2012-09-19 19:36:58 +0000 | [diff] [blame] | 1901 | } |
Robin Morisset | 5ce0ce4 | 2014-08-29 20:19:23 +0000 | [diff] [blame] | 1902 | |
| 1903 | // Copying the MachineMemOperand. |
| 1904 | MachineSDNode::mmo_iterator MemOp = MF->allocateMemRefsArray(1); |
| 1905 | MemOp[0] = cast<MemSDNode>(Node)->getMemOperand(); |
Eric Christopher | 4a34e61 | 2011-05-10 23:57:45 +0000 | [diff] [blame] | 1906 | cast<MachineSDNode>(Ret)->setMemRefs(MemOp, MemOp + 1); |
Robin Morisset | 5ce0ce4 | 2014-08-29 20:19:23 +0000 | [diff] [blame] | 1907 | |
| 1908 | // We need to have two outputs as that is what the original instruction had. |
| 1909 | // So we add a dummy, undefined output. This is safe as we checked first |
| 1910 | // that no-one uses our output anyway. |
| 1911 | SDValue Undef = SDValue(CurDAG->getMachineNode(TargetOpcode::IMPLICIT_DEF, |
| 1912 | dl, NVT), 0); |
Eric Christopher | 4a34e61 | 2011-05-10 23:57:45 +0000 | [diff] [blame] | 1913 | SDValue RetVals[] = { Undef, Ret }; |
Craig Topper | 64941d9 | 2014-04-27 19:20:57 +0000 | [diff] [blame] | 1914 | return CurDAG->getMergeValues(RetVals, dl).getNode(); |
Eric Christopher | 4a34e61 | 2011-05-10 23:57:45 +0000 | [diff] [blame] | 1915 | } |
| 1916 | |
Dan Gohman | 7d9dffb | 2009-10-09 20:35:19 +0000 | [diff] [blame] | 1917 | /// HasNoSignedComparisonUses - Test whether the given X86ISD::CMP node has |
| 1918 | /// any uses which require the SF or OF bits to be accurate. |
| 1919 | static bool HasNoSignedComparisonUses(SDNode *N) { |
| 1920 | // Examine each user of the node. |
| 1921 | for (SDNode::use_iterator UI = N->use_begin(), |
| 1922 | UE = N->use_end(); UI != UE; ++UI) { |
| 1923 | // Only examine CopyToReg uses. |
| 1924 | if (UI->getOpcode() != ISD::CopyToReg) |
| 1925 | return false; |
| 1926 | // Only examine CopyToReg uses that copy to EFLAGS. |
| 1927 | if (cast<RegisterSDNode>(UI->getOperand(1))->getReg() != |
| 1928 | X86::EFLAGS) |
| 1929 | return false; |
| 1930 | // Examine each user of the CopyToReg use. |
| 1931 | for (SDNode::use_iterator FlagUI = UI->use_begin(), |
| 1932 | FlagUE = UI->use_end(); FlagUI != FlagUE; ++FlagUI) { |
| 1933 | // Only examine the Flag result. |
| 1934 | if (FlagUI.getUse().getResNo() != 1) continue; |
| 1935 | // Anything unusual: assume conservatively. |
| 1936 | if (!FlagUI->isMachineOpcode()) return false; |
| 1937 | // Examine the opcode of the user. |
| 1938 | switch (FlagUI->getMachineOpcode()) { |
| 1939 | // These comparisons don't treat the most significant bit specially. |
| 1940 | case X86::SETAr: case X86::SETAEr: case X86::SETBr: case X86::SETBEr: |
| 1941 | case X86::SETEr: case X86::SETNEr: case X86::SETPr: case X86::SETNPr: |
| 1942 | case X86::SETAm: case X86::SETAEm: case X86::SETBm: case X86::SETBEm: |
| 1943 | case X86::SETEm: case X86::SETNEm: case X86::SETPm: case X86::SETNPm: |
Craig Topper | 49758aa | 2015-01-06 04:23:53 +0000 | [diff] [blame] | 1944 | case X86::JA_1: case X86::JAE_1: case X86::JB_1: case X86::JBE_1: |
| 1945 | case X86::JE_1: case X86::JNE_1: case X86::JP_1: case X86::JNP_1: |
Dan Gohman | 7d9dffb | 2009-10-09 20:35:19 +0000 | [diff] [blame] | 1946 | case X86::CMOVA16rr: case X86::CMOVA16rm: |
| 1947 | case X86::CMOVA32rr: case X86::CMOVA32rm: |
| 1948 | case X86::CMOVA64rr: case X86::CMOVA64rm: |
| 1949 | case X86::CMOVAE16rr: case X86::CMOVAE16rm: |
| 1950 | case X86::CMOVAE32rr: case X86::CMOVAE32rm: |
| 1951 | case X86::CMOVAE64rr: case X86::CMOVAE64rm: |
| 1952 | case X86::CMOVB16rr: case X86::CMOVB16rm: |
| 1953 | case X86::CMOVB32rr: case X86::CMOVB32rm: |
| 1954 | case X86::CMOVB64rr: case X86::CMOVB64rm: |
Chris Lattner | 1a1c600 | 2010-10-05 23:00:14 +0000 | [diff] [blame] | 1955 | case X86::CMOVBE16rr: case X86::CMOVBE16rm: |
| 1956 | case X86::CMOVBE32rr: case X86::CMOVBE32rm: |
| 1957 | case X86::CMOVBE64rr: case X86::CMOVBE64rm: |
Dan Gohman | 7d9dffb | 2009-10-09 20:35:19 +0000 | [diff] [blame] | 1958 | case X86::CMOVE16rr: case X86::CMOVE16rm: |
| 1959 | case X86::CMOVE32rr: case X86::CMOVE32rm: |
| 1960 | case X86::CMOVE64rr: case X86::CMOVE64rm: |
| 1961 | case X86::CMOVNE16rr: case X86::CMOVNE16rm: |
| 1962 | case X86::CMOVNE32rr: case X86::CMOVNE32rm: |
| 1963 | case X86::CMOVNE64rr: case X86::CMOVNE64rm: |
| 1964 | case X86::CMOVNP16rr: case X86::CMOVNP16rm: |
| 1965 | case X86::CMOVNP32rr: case X86::CMOVNP32rm: |
| 1966 | case X86::CMOVNP64rr: case X86::CMOVNP64rm: |
| 1967 | case X86::CMOVP16rr: case X86::CMOVP16rm: |
| 1968 | case X86::CMOVP32rr: case X86::CMOVP32rm: |
| 1969 | case X86::CMOVP64rr: case X86::CMOVP64rm: |
| 1970 | continue; |
| 1971 | // Anything else: assume conservatively. |
| 1972 | default: return false; |
| 1973 | } |
| 1974 | } |
| 1975 | } |
| 1976 | return true; |
| 1977 | } |
| 1978 | |
Joel Jones | 68d59e8 | 2012-03-29 05:45:48 +0000 | [diff] [blame] | 1979 | /// isLoadIncOrDecStore - Check whether or not the chain ending in StoreNode |
| 1980 | /// is suitable for doing the {load; increment or decrement; store} to modify |
| 1981 | /// transformation. |
Chad Rosier | 24c19d2 | 2012-08-01 18:39:17 +0000 | [diff] [blame] | 1982 | static bool isLoadIncOrDecStore(StoreSDNode *StoreNode, unsigned Opc, |
Evan Cheng | 3e869f0 | 2012-04-12 19:14:21 +0000 | [diff] [blame] | 1983 | SDValue StoredVal, SelectionDAG *CurDAG, |
| 1984 | LoadSDNode* &LoadNode, SDValue &InputChain) { |
Joel Jones | 68d59e8 | 2012-03-29 05:45:48 +0000 | [diff] [blame] | 1985 | |
| 1986 | // is the value stored the result of a DEC or INC? |
| 1987 | if (!(Opc == X86ISD::DEC || Opc == X86ISD::INC)) return false; |
| 1988 | |
Joel Jones | 68d59e8 | 2012-03-29 05:45:48 +0000 | [diff] [blame] | 1989 | // is the stored value result 0 of the load? |
| 1990 | if (StoredVal.getResNo() != 0) return false; |
| 1991 | |
| 1992 | // are there other uses of the loaded value than the inc or dec? |
| 1993 | if (!StoredVal.getNode()->hasNUsesOfValue(1, 0)) return false; |
| 1994 | |
Joel Jones | 68d59e8 | 2012-03-29 05:45:48 +0000 | [diff] [blame] | 1995 | // is the store non-extending and non-indexed? |
Evan Cheng | 3e869f0 | 2012-04-12 19:14:21 +0000 | [diff] [blame] | 1996 | if (!ISD::isNormalStore(StoreNode) || StoreNode->isNonTemporal()) |
Joel Jones | 68d59e8 | 2012-03-29 05:45:48 +0000 | [diff] [blame] | 1997 | return false; |
| 1998 | |
Evan Cheng | 3e869f0 | 2012-04-12 19:14:21 +0000 | [diff] [blame] | 1999 | SDValue Load = StoredVal->getOperand(0); |
| 2000 | // Is the stored value a non-extending and non-indexed load? |
| 2001 | if (!ISD::isNormalLoad(Load.getNode())) return false; |
| 2002 | |
| 2003 | // Return LoadNode by reference. |
| 2004 | LoadNode = cast<LoadSDNode>(Load); |
| 2005 | // is the size of the value one that we can handle? (i.e. 64, 32, 16, or 8) |
Chad Rosier | 24c19d2 | 2012-08-01 18:39:17 +0000 | [diff] [blame] | 2006 | EVT LdVT = LoadNode->getMemoryVT(); |
| 2007 | if (LdVT != MVT::i64 && LdVT != MVT::i32 && LdVT != MVT::i16 && |
Evan Cheng | 3e869f0 | 2012-04-12 19:14:21 +0000 | [diff] [blame] | 2008 | LdVT != MVT::i8) |
| 2009 | return false; |
| 2010 | |
| 2011 | // Is store the only read of the loaded value? |
| 2012 | if (!Load.hasOneUse()) |
| 2013 | return false; |
Chad Rosier | 24c19d2 | 2012-08-01 18:39:17 +0000 | [diff] [blame] | 2014 | |
Evan Cheng | 3e869f0 | 2012-04-12 19:14:21 +0000 | [diff] [blame] | 2015 | // Is the address of the store the same as the load? |
| 2016 | if (LoadNode->getBasePtr() != StoreNode->getBasePtr() || |
| 2017 | LoadNode->getOffset() != StoreNode->getOffset()) |
| 2018 | return false; |
| 2019 | |
| 2020 | // Check if the chain is produced by the load or is a TokenFactor with |
| 2021 | // the load output chain as an operand. Return InputChain by reference. |
| 2022 | SDValue Chain = StoreNode->getChain(); |
| 2023 | |
| 2024 | bool ChainCheck = false; |
| 2025 | if (Chain == Load.getValue(1)) { |
| 2026 | ChainCheck = true; |
| 2027 | InputChain = LoadNode->getChain(); |
| 2028 | } else if (Chain.getOpcode() == ISD::TokenFactor) { |
| 2029 | SmallVector<SDValue, 4> ChainOps; |
| 2030 | for (unsigned i = 0, e = Chain.getNumOperands(); i != e; ++i) { |
| 2031 | SDValue Op = Chain.getOperand(i); |
| 2032 | if (Op == Load.getValue(1)) { |
| 2033 | ChainCheck = true; |
| 2034 | continue; |
| 2035 | } |
Evan Cheng | 58a95f0 | 2012-05-16 01:54:27 +0000 | [diff] [blame] | 2036 | |
| 2037 | // Make sure using Op as part of the chain would not cause a cycle here. |
| 2038 | // In theory, we could check whether the chain node is a predecessor of |
| 2039 | // the load. But that can be very expensive. Instead visit the uses and |
| 2040 | // make sure they all have smaller node id than the load. |
| 2041 | int LoadId = LoadNode->getNodeId(); |
| 2042 | for (SDNode::use_iterator UI = Op.getNode()->use_begin(), |
| 2043 | UE = UI->use_end(); UI != UE; ++UI) { |
| 2044 | if (UI.getUse().getResNo() != 0) |
| 2045 | continue; |
| 2046 | if (UI->getNodeId() > LoadId) |
| 2047 | return false; |
| 2048 | } |
| 2049 | |
Evan Cheng | 3e869f0 | 2012-04-12 19:14:21 +0000 | [diff] [blame] | 2050 | ChainOps.push_back(Op); |
| 2051 | } |
| 2052 | |
| 2053 | if (ChainCheck) |
| 2054 | // Make a new TokenFactor with all the other input chains except |
| 2055 | // for the load. |
Andrew Trick | ef9de2a | 2013-05-25 02:42:55 +0000 | [diff] [blame] | 2056 | InputChain = CurDAG->getNode(ISD::TokenFactor, SDLoc(Chain), |
Craig Topper | 48d114b | 2014-04-26 18:35:24 +0000 | [diff] [blame] | 2057 | MVT::Other, ChainOps); |
Evan Cheng | 3e869f0 | 2012-04-12 19:14:21 +0000 | [diff] [blame] | 2058 | } |
| 2059 | if (!ChainCheck) |
Joel Jones | 68d59e8 | 2012-03-29 05:45:48 +0000 | [diff] [blame] | 2060 | return false; |
| 2061 | |
| 2062 | return true; |
| 2063 | } |
| 2064 | |
Benjamin Kramer | 8619c37 | 2012-03-29 12:37:26 +0000 | [diff] [blame] | 2065 | /// getFusedLdStOpcode - Get the appropriate X86 opcode for an in memory |
| 2066 | /// increment or decrement. Opc should be X86ISD::DEC or X86ISD::INC. |
Joel Jones | 68d59e8 | 2012-03-29 05:45:48 +0000 | [diff] [blame] | 2067 | static unsigned getFusedLdStOpcode(EVT &LdVT, unsigned Opc) { |
| 2068 | if (Opc == X86ISD::DEC) { |
| 2069 | if (LdVT == MVT::i64) return X86::DEC64m; |
| 2070 | if (LdVT == MVT::i32) return X86::DEC32m; |
| 2071 | if (LdVT == MVT::i16) return X86::DEC16m; |
| 2072 | if (LdVT == MVT::i8) return X86::DEC8m; |
Benjamin Kramer | 8619c37 | 2012-03-29 12:37:26 +0000 | [diff] [blame] | 2073 | } else { |
| 2074 | assert(Opc == X86ISD::INC && "unrecognized opcode"); |
Joel Jones | 68d59e8 | 2012-03-29 05:45:48 +0000 | [diff] [blame] | 2075 | if (LdVT == MVT::i64) return X86::INC64m; |
| 2076 | if (LdVT == MVT::i32) return X86::INC32m; |
| 2077 | if (LdVT == MVT::i16) return X86::INC16m; |
| 2078 | if (LdVT == MVT::i8) return X86::INC8m; |
Joel Jones | 68d59e8 | 2012-03-29 05:45:48 +0000 | [diff] [blame] | 2079 | } |
Benjamin Kramer | 8619c37 | 2012-03-29 12:37:26 +0000 | [diff] [blame] | 2080 | llvm_unreachable("unrecognized size for LdVT"); |
Joel Jones | 68d59e8 | 2012-03-29 05:45:48 +0000 | [diff] [blame] | 2081 | } |
| 2082 | |
Manman Ren | a098204 | 2012-06-26 19:47:59 +0000 | [diff] [blame] | 2083 | /// SelectGather - Customized ISel for GATHER operations. |
| 2084 | /// |
| 2085 | SDNode *X86DAGToDAGISel::SelectGather(SDNode *Node, unsigned Opc) { |
| 2086 | // Operands of Gather: VSrc, Base, VIdx, VMask, Scale |
| 2087 | SDValue Chain = Node->getOperand(0); |
| 2088 | SDValue VSrc = Node->getOperand(2); |
| 2089 | SDValue Base = Node->getOperand(3); |
| 2090 | SDValue VIdx = Node->getOperand(4); |
| 2091 | SDValue VMask = Node->getOperand(5); |
| 2092 | ConstantSDNode *Scale = dyn_cast<ConstantSDNode>(Node->getOperand(6)); |
Craig Topper | fbb954f7 | 2012-07-01 02:17:08 +0000 | [diff] [blame] | 2093 | if (!Scale) |
Craig Topper | 062a2ba | 2014-04-25 05:30:21 +0000 | [diff] [blame] | 2094 | return nullptr; |
Manman Ren | a098204 | 2012-06-26 19:47:59 +0000 | [diff] [blame] | 2095 | |
Craig Topper | f7755df | 2012-07-12 06:52:41 +0000 | [diff] [blame] | 2096 | SDVTList VTs = CurDAG->getVTList(VSrc.getValueType(), VSrc.getValueType(), |
| 2097 | MVT::Other); |
| 2098 | |
Sergey Dmitrouk | 842a51b | 2015-04-28 14:05:47 +0000 | [diff] [blame] | 2099 | SDLoc DL(Node); |
| 2100 | |
Manman Ren | a098204 | 2012-06-26 19:47:59 +0000 | [diff] [blame] | 2101 | // Memory Operands: Base, Scale, Index, Disp, Segment |
Sergey Dmitrouk | 842a51b | 2015-04-28 14:05:47 +0000 | [diff] [blame] | 2102 | SDValue Disp = CurDAG->getTargetConstant(0, DL, MVT::i32); |
Manman Ren | a098204 | 2012-06-26 19:47:59 +0000 | [diff] [blame] | 2103 | SDValue Segment = CurDAG->getRegister(0, MVT::i32); |
Sergey Dmitrouk | 842a51b | 2015-04-28 14:05:47 +0000 | [diff] [blame] | 2104 | const SDValue Ops[] = { VSrc, Base, getI8Imm(Scale->getSExtValue(), DL), VIdx, |
Manman Ren | a098204 | 2012-06-26 19:47:59 +0000 | [diff] [blame] | 2105 | Disp, Segment, VMask, Chain}; |
Sergey Dmitrouk | 842a51b | 2015-04-28 14:05:47 +0000 | [diff] [blame] | 2106 | SDNode *ResNode = CurDAG->getMachineNode(Opc, DL, VTs, Ops); |
Craig Topper | f7755df | 2012-07-12 06:52:41 +0000 | [diff] [blame] | 2107 | // Node has 2 outputs: VDst and MVT::Other. |
| 2108 | // ResNode has 3 outputs: VDst, VMask_wb, and MVT::Other. |
| 2109 | // We replace VDst of Node with VDst of ResNode, and Other of Node with Other |
| 2110 | // of ResNode. |
| 2111 | ReplaceUses(SDValue(Node, 0), SDValue(ResNode, 0)); |
| 2112 | ReplaceUses(SDValue(Node, 1), SDValue(ResNode, 2)); |
Manman Ren | a098204 | 2012-06-26 19:47:59 +0000 | [diff] [blame] | 2113 | return ResNode; |
| 2114 | } |
| 2115 | |
Dan Gohman | ea6f91f | 2010-01-05 01:24:18 +0000 | [diff] [blame] | 2116 | SDNode *X86DAGToDAGISel::Select(SDNode *Node) { |
Craig Topper | 83e042a | 2013-08-15 05:57:07 +0000 | [diff] [blame] | 2117 | MVT NVT = Node->getSimpleValueType(0); |
Evan Cheng | 10d2790 | 2006-01-06 20:36:21 +0000 | [diff] [blame] | 2118 | unsigned Opc, MOpc; |
| 2119 | unsigned Opcode = Node->getOpcode(); |
Andrew Trick | ef9de2a | 2013-05-25 02:42:55 +0000 | [diff] [blame] | 2120 | SDLoc dl(Node); |
Chad Rosier | 24c19d2 | 2012-08-01 18:39:17 +0000 | [diff] [blame] | 2121 | |
Chris Lattner | f98f124 | 2010-03-02 06:34:30 +0000 | [diff] [blame] | 2122 | DEBUG(dbgs() << "Selecting: "; Node->dump(CurDAG); dbgs() << '\n'); |
Evan Cheng | d49cc36 | 2006-02-10 22:24:32 +0000 | [diff] [blame] | 2123 | |
Dan Gohman | 1705968 | 2008-07-17 19:10:17 +0000 | [diff] [blame] | 2124 | if (Node->isMachineOpcode()) { |
Chris Lattner | f98f124 | 2010-03-02 06:34:30 +0000 | [diff] [blame] | 2125 | DEBUG(dbgs() << "== "; Node->dump(CurDAG); dbgs() << '\n'); |
Tim Northover | 31d093c | 2013-09-22 08:21:56 +0000 | [diff] [blame] | 2126 | Node->setNodeId(-1); |
Craig Topper | 062a2ba | 2014-04-25 05:30:21 +0000 | [diff] [blame] | 2127 | return nullptr; // Already selected. |
Evan Cheng | 6dc90ca | 2006-02-09 00:37:58 +0000 | [diff] [blame] | 2128 | } |
Evan Cheng | 2ae799a | 2006-01-11 22:15:18 +0000 | [diff] [blame] | 2129 | |
Evan Cheng | 10d2790 | 2006-01-06 20:36:21 +0000 | [diff] [blame] | 2130 | switch (Opcode) { |
Dan Gohman | 757eee8 | 2009-08-02 16:10:52 +0000 | [diff] [blame] | 2131 | default: break; |
Manman Ren | a098204 | 2012-06-26 19:47:59 +0000 | [diff] [blame] | 2132 | case ISD::INTRINSIC_W_CHAIN: { |
| 2133 | unsigned IntNo = cast<ConstantSDNode>(Node->getOperand(1))->getZExtValue(); |
| 2134 | switch (IntNo) { |
| 2135 | default: break; |
| 2136 | case Intrinsic::x86_avx2_gather_d_pd: |
Manman Ren | a098204 | 2012-06-26 19:47:59 +0000 | [diff] [blame] | 2137 | case Intrinsic::x86_avx2_gather_d_pd_256: |
Manman Ren | a098204 | 2012-06-26 19:47:59 +0000 | [diff] [blame] | 2138 | case Intrinsic::x86_avx2_gather_q_pd: |
Manman Ren | a098204 | 2012-06-26 19:47:59 +0000 | [diff] [blame] | 2139 | case Intrinsic::x86_avx2_gather_q_pd_256: |
Manman Ren | a098204 | 2012-06-26 19:47:59 +0000 | [diff] [blame] | 2140 | case Intrinsic::x86_avx2_gather_d_ps: |
Manman Ren | a098204 | 2012-06-26 19:47:59 +0000 | [diff] [blame] | 2141 | case Intrinsic::x86_avx2_gather_d_ps_256: |
Manman Ren | a098204 | 2012-06-26 19:47:59 +0000 | [diff] [blame] | 2142 | case Intrinsic::x86_avx2_gather_q_ps: |
Manman Ren | a098204 | 2012-06-26 19:47:59 +0000 | [diff] [blame] | 2143 | case Intrinsic::x86_avx2_gather_q_ps_256: |
Manman Ren | 98a5bf2 | 2012-06-29 00:54:20 +0000 | [diff] [blame] | 2144 | case Intrinsic::x86_avx2_gather_d_q: |
Manman Ren | 98a5bf2 | 2012-06-29 00:54:20 +0000 | [diff] [blame] | 2145 | case Intrinsic::x86_avx2_gather_d_q_256: |
Manman Ren | 98a5bf2 | 2012-06-29 00:54:20 +0000 | [diff] [blame] | 2146 | case Intrinsic::x86_avx2_gather_q_q: |
Manman Ren | 98a5bf2 | 2012-06-29 00:54:20 +0000 | [diff] [blame] | 2147 | case Intrinsic::x86_avx2_gather_q_q_256: |
Manman Ren | 98a5bf2 | 2012-06-29 00:54:20 +0000 | [diff] [blame] | 2148 | case Intrinsic::x86_avx2_gather_d_d: |
Manman Ren | 98a5bf2 | 2012-06-29 00:54:20 +0000 | [diff] [blame] | 2149 | case Intrinsic::x86_avx2_gather_d_d_256: |
Manman Ren | 98a5bf2 | 2012-06-29 00:54:20 +0000 | [diff] [blame] | 2150 | case Intrinsic::x86_avx2_gather_q_d: |
Craig Topper | def044b | 2012-07-01 02:05:52 +0000 | [diff] [blame] | 2151 | case Intrinsic::x86_avx2_gather_q_d_256: { |
Michael Liao | 00b20cc | 2013-06-05 18:12:26 +0000 | [diff] [blame] | 2152 | if (!Subtarget->hasAVX2()) |
| 2153 | break; |
Craig Topper | def044b | 2012-07-01 02:05:52 +0000 | [diff] [blame] | 2154 | unsigned Opc; |
| 2155 | switch (IntNo) { |
Craig Topper | 3af251d | 2012-07-01 02:55:34 +0000 | [diff] [blame] | 2156 | default: llvm_unreachable("Impossible intrinsic"); |
Craig Topper | def044b | 2012-07-01 02:05:52 +0000 | [diff] [blame] | 2157 | case Intrinsic::x86_avx2_gather_d_pd: Opc = X86::VGATHERDPDrm; break; |
| 2158 | case Intrinsic::x86_avx2_gather_d_pd_256: Opc = X86::VGATHERDPDYrm; break; |
| 2159 | case Intrinsic::x86_avx2_gather_q_pd: Opc = X86::VGATHERQPDrm; break; |
| 2160 | case Intrinsic::x86_avx2_gather_q_pd_256: Opc = X86::VGATHERQPDYrm; break; |
| 2161 | case Intrinsic::x86_avx2_gather_d_ps: Opc = X86::VGATHERDPSrm; break; |
| 2162 | case Intrinsic::x86_avx2_gather_d_ps_256: Opc = X86::VGATHERDPSYrm; break; |
| 2163 | case Intrinsic::x86_avx2_gather_q_ps: Opc = X86::VGATHERQPSrm; break; |
| 2164 | case Intrinsic::x86_avx2_gather_q_ps_256: Opc = X86::VGATHERQPSYrm; break; |
| 2165 | case Intrinsic::x86_avx2_gather_d_q: Opc = X86::VPGATHERDQrm; break; |
| 2166 | case Intrinsic::x86_avx2_gather_d_q_256: Opc = X86::VPGATHERDQYrm; break; |
| 2167 | case Intrinsic::x86_avx2_gather_q_q: Opc = X86::VPGATHERQQrm; break; |
| 2168 | case Intrinsic::x86_avx2_gather_q_q_256: Opc = X86::VPGATHERQQYrm; break; |
| 2169 | case Intrinsic::x86_avx2_gather_d_d: Opc = X86::VPGATHERDDrm; break; |
| 2170 | case Intrinsic::x86_avx2_gather_d_d_256: Opc = X86::VPGATHERDDYrm; break; |
| 2171 | case Intrinsic::x86_avx2_gather_q_d: Opc = X86::VPGATHERQDrm; break; |
| 2172 | case Intrinsic::x86_avx2_gather_q_d_256: Opc = X86::VPGATHERQDYrm; break; |
| 2173 | } |
Craig Topper | fbb954f7 | 2012-07-01 02:17:08 +0000 | [diff] [blame] | 2174 | SDNode *RetVal = SelectGather(Node, Opc); |
| 2175 | if (RetVal) |
Craig Topper | f7755df | 2012-07-12 06:52:41 +0000 | [diff] [blame] | 2176 | // We already called ReplaceUses inside SelectGather. |
Craig Topper | 062a2ba | 2014-04-25 05:30:21 +0000 | [diff] [blame] | 2177 | return nullptr; |
Craig Topper | e15e5f7 | 2012-07-01 02:18:18 +0000 | [diff] [blame] | 2178 | break; |
Craig Topper | def044b | 2012-07-01 02:05:52 +0000 | [diff] [blame] | 2179 | } |
Manman Ren | a098204 | 2012-06-26 19:47:59 +0000 | [diff] [blame] | 2180 | } |
| 2181 | break; |
| 2182 | } |
Dan Gohman | 757eee8 | 2009-08-02 16:10:52 +0000 | [diff] [blame] | 2183 | case X86ISD::GlobalBaseReg: |
| 2184 | return getGlobalBaseReg(); |
Evan Cheng | e0ed6ec | 2006-02-23 20:41:18 +0000 | [diff] [blame] | 2185 | |
Quentin Colombet | dbe33e7 | 2014-11-06 02:25:03 +0000 | [diff] [blame] | 2186 | case X86ISD::SHRUNKBLEND: { |
| 2187 | // SHRUNKBLEND selects like a regular VSELECT. |
| 2188 | SDValue VSelect = CurDAG->getNode( |
| 2189 | ISD::VSELECT, SDLoc(Node), Node->getValueType(0), Node->getOperand(0), |
| 2190 | Node->getOperand(1), Node->getOperand(2)); |
| 2191 | ReplaceUses(SDValue(Node, 0), VSelect); |
| 2192 | SelectCode(VSelect.getNode()); |
| 2193 | // We already called ReplaceUses. |
| 2194 | return nullptr; |
| 2195 | } |
Craig Topper | 3af251d | 2012-07-01 02:55:34 +0000 | [diff] [blame] | 2196 | |
Eric Christopher | a1d9e29 | 2011-05-17 08:10:18 +0000 | [diff] [blame] | 2197 | case ISD::ATOMIC_LOAD_XOR: |
| 2198 | case ISD::ATOMIC_LOAD_AND: |
Michael Liao | 8372539 | 2012-09-19 19:36:58 +0000 | [diff] [blame] | 2199 | case ISD::ATOMIC_LOAD_OR: |
| 2200 | case ISD::ATOMIC_LOAD_ADD: { |
Eric Christopher | a1d9e29 | 2011-05-17 08:10:18 +0000 | [diff] [blame] | 2201 | SDNode *RetVal = SelectAtomicLoadArith(Node, NVT); |
Eric Christopher | 4a34e61 | 2011-05-10 23:57:45 +0000 | [diff] [blame] | 2202 | if (RetVal) |
| 2203 | return RetVal; |
| 2204 | break; |
| 2205 | } |
Benjamin Kramer | 4c81624 | 2011-04-22 15:30:40 +0000 | [diff] [blame] | 2206 | case ISD::AND: |
| 2207 | case ISD::OR: |
| 2208 | case ISD::XOR: { |
| 2209 | // For operations of the form (x << C1) op C2, check if we can use a smaller |
| 2210 | // encoding for C2 by transforming it into (x op (C2>>C1)) << C1. |
| 2211 | SDValue N0 = Node->getOperand(0); |
| 2212 | SDValue N1 = Node->getOperand(1); |
| 2213 | |
| 2214 | if (N0->getOpcode() != ISD::SHL || !N0->hasOneUse()) |
| 2215 | break; |
| 2216 | |
| 2217 | // i8 is unshrinkable, i16 should be promoted to i32. |
| 2218 | if (NVT != MVT::i32 && NVT != MVT::i64) |
| 2219 | break; |
| 2220 | |
| 2221 | ConstantSDNode *Cst = dyn_cast<ConstantSDNode>(N1); |
| 2222 | ConstantSDNode *ShlCst = dyn_cast<ConstantSDNode>(N0->getOperand(1)); |
| 2223 | if (!Cst || !ShlCst) |
| 2224 | break; |
| 2225 | |
| 2226 | int64_t Val = Cst->getSExtValue(); |
| 2227 | uint64_t ShlVal = ShlCst->getZExtValue(); |
| 2228 | |
| 2229 | // Make sure that we don't change the operation by removing bits. |
| 2230 | // This only matters for OR and XOR, AND is unaffected. |
Richard Smith | 228e6d4 | 2012-08-24 23:29:28 +0000 | [diff] [blame] | 2231 | uint64_t RemovedBitsMask = (1ULL << ShlVal) - 1; |
| 2232 | if (Opcode != ISD::AND && (Val & RemovedBitsMask) != 0) |
Benjamin Kramer | 4c81624 | 2011-04-22 15:30:40 +0000 | [diff] [blame] | 2233 | break; |
| 2234 | |
Benjamin Kramer | 3a16a36 | 2015-04-01 19:01:09 +0000 | [diff] [blame] | 2235 | unsigned ShlOp, AddOp, Op; |
Craig Topper | 83e042a | 2013-08-15 05:57:07 +0000 | [diff] [blame] | 2236 | MVT CstVT = NVT; |
Benjamin Kramer | 4c81624 | 2011-04-22 15:30:40 +0000 | [diff] [blame] | 2237 | |
| 2238 | // Check the minimum bitwidth for the new constant. |
| 2239 | // TODO: AND32ri is the same as AND64ri32 with zext imm. |
| 2240 | // TODO: MOV32ri+OR64r is cheaper than MOV64ri64+OR64rr |
| 2241 | // TODO: Using 16 and 8 bit operations is also possible for or32 & xor32. |
| 2242 | if (!isInt<8>(Val) && isInt<8>(Val >> ShlVal)) |
| 2243 | CstVT = MVT::i8; |
| 2244 | else if (!isInt<32>(Val) && isInt<32>(Val >> ShlVal)) |
| 2245 | CstVT = MVT::i32; |
| 2246 | |
| 2247 | // Bail if there is no smaller encoding. |
| 2248 | if (NVT == CstVT) |
| 2249 | break; |
| 2250 | |
Craig Topper | 83e042a | 2013-08-15 05:57:07 +0000 | [diff] [blame] | 2251 | switch (NVT.SimpleTy) { |
Benjamin Kramer | 4c81624 | 2011-04-22 15:30:40 +0000 | [diff] [blame] | 2252 | default: llvm_unreachable("Unsupported VT!"); |
| 2253 | case MVT::i32: |
| 2254 | assert(CstVT == MVT::i8); |
| 2255 | ShlOp = X86::SHL32ri; |
Benjamin Kramer | 3a16a36 | 2015-04-01 19:01:09 +0000 | [diff] [blame] | 2256 | AddOp = X86::ADD32rr; |
Benjamin Kramer | 4c81624 | 2011-04-22 15:30:40 +0000 | [diff] [blame] | 2257 | |
| 2258 | switch (Opcode) { |
Craig Topper | 22cb0c5 | 2012-08-11 17:44:14 +0000 | [diff] [blame] | 2259 | default: llvm_unreachable("Impossible opcode"); |
Benjamin Kramer | 4c81624 | 2011-04-22 15:30:40 +0000 | [diff] [blame] | 2260 | case ISD::AND: Op = X86::AND32ri8; break; |
| 2261 | case ISD::OR: Op = X86::OR32ri8; break; |
| 2262 | case ISD::XOR: Op = X86::XOR32ri8; break; |
| 2263 | } |
| 2264 | break; |
| 2265 | case MVT::i64: |
| 2266 | assert(CstVT == MVT::i8 || CstVT == MVT::i32); |
| 2267 | ShlOp = X86::SHL64ri; |
Benjamin Kramer | 3a16a36 | 2015-04-01 19:01:09 +0000 | [diff] [blame] | 2268 | AddOp = X86::ADD64rr; |
Benjamin Kramer | 4c81624 | 2011-04-22 15:30:40 +0000 | [diff] [blame] | 2269 | |
| 2270 | switch (Opcode) { |
Craig Topper | 22cb0c5 | 2012-08-11 17:44:14 +0000 | [diff] [blame] | 2271 | default: llvm_unreachable("Impossible opcode"); |
Benjamin Kramer | 4c81624 | 2011-04-22 15:30:40 +0000 | [diff] [blame] | 2272 | case ISD::AND: Op = CstVT==MVT::i8? X86::AND64ri8 : X86::AND64ri32; break; |
| 2273 | case ISD::OR: Op = CstVT==MVT::i8? X86::OR64ri8 : X86::OR64ri32; break; |
| 2274 | case ISD::XOR: Op = CstVT==MVT::i8? X86::XOR64ri8 : X86::XOR64ri32; break; |
| 2275 | } |
| 2276 | break; |
| 2277 | } |
| 2278 | |
| 2279 | // Emit the smaller op and the shift. |
Sergey Dmitrouk | 842a51b | 2015-04-28 14:05:47 +0000 | [diff] [blame] | 2280 | SDValue NewCst = CurDAG->getTargetConstant(Val >> ShlVal, dl, CstVT); |
Benjamin Kramer | 4c81624 | 2011-04-22 15:30:40 +0000 | [diff] [blame] | 2281 | SDNode *New = CurDAG->getMachineNode(Op, dl, NVT, N0->getOperand(0),NewCst); |
Benjamin Kramer | 3a16a36 | 2015-04-01 19:01:09 +0000 | [diff] [blame] | 2282 | if (ShlVal == 1) |
| 2283 | return CurDAG->SelectNodeTo(Node, AddOp, NVT, SDValue(New, 0), |
| 2284 | SDValue(New, 0)); |
Benjamin Kramer | 4c81624 | 2011-04-22 15:30:40 +0000 | [diff] [blame] | 2285 | return CurDAG->SelectNodeTo(Node, ShlOp, NVT, SDValue(New, 0), |
Sergey Dmitrouk | 842a51b | 2015-04-28 14:05:47 +0000 | [diff] [blame] | 2286 | getI8Imm(ShlVal, dl)); |
Benjamin Kramer | 4c81624 | 2011-04-22 15:30:40 +0000 | [diff] [blame] | 2287 | } |
Ahmed Bougacha | 5175bcf | 2014-10-23 21:55:31 +0000 | [diff] [blame] | 2288 | case X86ISD::UMUL8: |
| 2289 | case X86ISD::SMUL8: { |
| 2290 | SDValue N0 = Node->getOperand(0); |
| 2291 | SDValue N1 = Node->getOperand(1); |
| 2292 | |
| 2293 | Opc = (Opcode == X86ISD::SMUL8 ? X86::IMUL8r : X86::MUL8r); |
| 2294 | |
| 2295 | SDValue InFlag = CurDAG->getCopyToReg(CurDAG->getEntryNode(), dl, X86::AL, |
| 2296 | N0, SDValue()).getValue(1); |
| 2297 | |
| 2298 | SDVTList VTs = CurDAG->getVTList(NVT, MVT::i32); |
| 2299 | SDValue Ops[] = {N1, InFlag}; |
| 2300 | SDNode *CNode = CurDAG->getMachineNode(Opc, dl, VTs, Ops); |
| 2301 | |
| 2302 | ReplaceUses(SDValue(Node, 0), SDValue(CNode, 0)); |
| 2303 | ReplaceUses(SDValue(Node, 1), SDValue(CNode, 1)); |
| 2304 | return nullptr; |
| 2305 | } |
| 2306 | |
Chris Lattner | 364bb0a | 2010-12-05 07:30:36 +0000 | [diff] [blame] | 2307 | case X86ISD::UMUL: { |
| 2308 | SDValue N0 = Node->getOperand(0); |
| 2309 | SDValue N1 = Node->getOperand(1); |
Chad Rosier | 24c19d2 | 2012-08-01 18:39:17 +0000 | [diff] [blame] | 2310 | |
Ted Kremenek | b5241b2 | 2011-01-14 22:34:13 +0000 | [diff] [blame] | 2311 | unsigned LoReg; |
Craig Topper | 83e042a | 2013-08-15 05:57:07 +0000 | [diff] [blame] | 2312 | switch (NVT.SimpleTy) { |
Chris Lattner | 364bb0a | 2010-12-05 07:30:36 +0000 | [diff] [blame] | 2313 | default: llvm_unreachable("Unsupported VT!"); |
Ted Kremenek | b5241b2 | 2011-01-14 22:34:13 +0000 | [diff] [blame] | 2314 | case MVT::i8: LoReg = X86::AL; Opc = X86::MUL8r; break; |
| 2315 | case MVT::i16: LoReg = X86::AX; Opc = X86::MUL16r; break; |
| 2316 | case MVT::i32: LoReg = X86::EAX; Opc = X86::MUL32r; break; |
| 2317 | case MVT::i64: LoReg = X86::RAX; Opc = X86::MUL64r; break; |
Chris Lattner | 364bb0a | 2010-12-05 07:30:36 +0000 | [diff] [blame] | 2318 | } |
Chad Rosier | 24c19d2 | 2012-08-01 18:39:17 +0000 | [diff] [blame] | 2319 | |
Chris Lattner | 364bb0a | 2010-12-05 07:30:36 +0000 | [diff] [blame] | 2320 | SDValue InFlag = CurDAG->getCopyToReg(CurDAG->getEntryNode(), dl, LoReg, |
| 2321 | N0, SDValue()).getValue(1); |
Chad Rosier | 24c19d2 | 2012-08-01 18:39:17 +0000 | [diff] [blame] | 2322 | |
Chris Lattner | 364bb0a | 2010-12-05 07:30:36 +0000 | [diff] [blame] | 2323 | SDVTList VTs = CurDAG->getVTList(NVT, NVT, MVT::i32); |
| 2324 | SDValue Ops[] = {N1, InFlag}; |
Michael Liao | b53d896 | 2013-04-19 22:22:57 +0000 | [diff] [blame] | 2325 | SDNode *CNode = CurDAG->getMachineNode(Opc, dl, VTs, Ops); |
Chad Rosier | 24c19d2 | 2012-08-01 18:39:17 +0000 | [diff] [blame] | 2326 | |
Chris Lattner | 364bb0a | 2010-12-05 07:30:36 +0000 | [diff] [blame] | 2327 | ReplaceUses(SDValue(Node, 0), SDValue(CNode, 0)); |
| 2328 | ReplaceUses(SDValue(Node, 1), SDValue(CNode, 1)); |
| 2329 | ReplaceUses(SDValue(Node, 2), SDValue(CNode, 2)); |
Craig Topper | 062a2ba | 2014-04-25 05:30:21 +0000 | [diff] [blame] | 2330 | return nullptr; |
Chris Lattner | 364bb0a | 2010-12-05 07:30:36 +0000 | [diff] [blame] | 2331 | } |
Chad Rosier | 24c19d2 | 2012-08-01 18:39:17 +0000 | [diff] [blame] | 2332 | |
Dan Gohman | 757eee8 | 2009-08-02 16:10:52 +0000 | [diff] [blame] | 2333 | case ISD::SMUL_LOHI: |
| 2334 | case ISD::UMUL_LOHI: { |
| 2335 | SDValue N0 = Node->getOperand(0); |
| 2336 | SDValue N1 = Node->getOperand(1); |
| 2337 | |
| 2338 | bool isSigned = Opcode == ISD::SMUL_LOHI; |
Michael Liao | f9f7b55 | 2012-09-26 08:22:37 +0000 | [diff] [blame] | 2339 | bool hasBMI2 = Subtarget->hasBMI2(); |
Bill Wendling | fe3bdb4 | 2009-08-07 21:33:25 +0000 | [diff] [blame] | 2340 | if (!isSigned) { |
Craig Topper | 83e042a | 2013-08-15 05:57:07 +0000 | [diff] [blame] | 2341 | switch (NVT.SimpleTy) { |
Dan Gohman | 757eee8 | 2009-08-02 16:10:52 +0000 | [diff] [blame] | 2342 | default: llvm_unreachable("Unsupported VT!"); |
Owen Anderson | 9f94459 | 2009-08-11 20:47:22 +0000 | [diff] [blame] | 2343 | case MVT::i8: Opc = X86::MUL8r; MOpc = X86::MUL8m; break; |
| 2344 | case MVT::i16: Opc = X86::MUL16r; MOpc = X86::MUL16m; break; |
Michael Liao | f9f7b55 | 2012-09-26 08:22:37 +0000 | [diff] [blame] | 2345 | case MVT::i32: Opc = hasBMI2 ? X86::MULX32rr : X86::MUL32r; |
| 2346 | MOpc = hasBMI2 ? X86::MULX32rm : X86::MUL32m; break; |
| 2347 | case MVT::i64: Opc = hasBMI2 ? X86::MULX64rr : X86::MUL64r; |
| 2348 | MOpc = hasBMI2 ? X86::MULX64rm : X86::MUL64m; break; |
Dan Gohman | 757eee8 | 2009-08-02 16:10:52 +0000 | [diff] [blame] | 2349 | } |
Bill Wendling | fe3bdb4 | 2009-08-07 21:33:25 +0000 | [diff] [blame] | 2350 | } else { |
Craig Topper | 83e042a | 2013-08-15 05:57:07 +0000 | [diff] [blame] | 2351 | switch (NVT.SimpleTy) { |
Dan Gohman | 757eee8 | 2009-08-02 16:10:52 +0000 | [diff] [blame] | 2352 | default: llvm_unreachable("Unsupported VT!"); |
Owen Anderson | 9f94459 | 2009-08-11 20:47:22 +0000 | [diff] [blame] | 2353 | case MVT::i8: Opc = X86::IMUL8r; MOpc = X86::IMUL8m; break; |
| 2354 | case MVT::i16: Opc = X86::IMUL16r; MOpc = X86::IMUL16m; break; |
| 2355 | case MVT::i32: Opc = X86::IMUL32r; MOpc = X86::IMUL32m; break; |
| 2356 | case MVT::i64: Opc = X86::IMUL64r; MOpc = X86::IMUL64m; break; |
Dan Gohman | 757eee8 | 2009-08-02 16:10:52 +0000 | [diff] [blame] | 2357 | } |
Bill Wendling | fe3bdb4 | 2009-08-07 21:33:25 +0000 | [diff] [blame] | 2358 | } |
Dan Gohman | 757eee8 | 2009-08-02 16:10:52 +0000 | [diff] [blame] | 2359 | |
Michael Liao | f9f7b55 | 2012-09-26 08:22:37 +0000 | [diff] [blame] | 2360 | unsigned SrcReg, LoReg, HiReg; |
| 2361 | switch (Opc) { |
| 2362 | default: llvm_unreachable("Unknown MUL opcode!"); |
| 2363 | case X86::IMUL8r: |
| 2364 | case X86::MUL8r: |
| 2365 | SrcReg = LoReg = X86::AL; HiReg = X86::AH; |
| 2366 | break; |
| 2367 | case X86::IMUL16r: |
| 2368 | case X86::MUL16r: |
| 2369 | SrcReg = LoReg = X86::AX; HiReg = X86::DX; |
| 2370 | break; |
| 2371 | case X86::IMUL32r: |
| 2372 | case X86::MUL32r: |
| 2373 | SrcReg = LoReg = X86::EAX; HiReg = X86::EDX; |
| 2374 | break; |
| 2375 | case X86::IMUL64r: |
| 2376 | case X86::MUL64r: |
| 2377 | SrcReg = LoReg = X86::RAX; HiReg = X86::RDX; |
| 2378 | break; |
| 2379 | case X86::MULX32rr: |
| 2380 | SrcReg = X86::EDX; LoReg = HiReg = 0; |
| 2381 | break; |
| 2382 | case X86::MULX64rr: |
| 2383 | SrcReg = X86::RDX; LoReg = HiReg = 0; |
| 2384 | break; |
Dan Gohman | 757eee8 | 2009-08-02 16:10:52 +0000 | [diff] [blame] | 2385 | } |
| 2386 | |
| 2387 | SDValue Tmp0, Tmp1, Tmp2, Tmp3, Tmp4; |
Dan Gohman | ea6f91f | 2010-01-05 01:24:18 +0000 | [diff] [blame] | 2388 | bool foldedLoad = TryFoldLoad(Node, N1, Tmp0, Tmp1, Tmp2, Tmp3, Tmp4); |
Bill Wendling | fe3bdb4 | 2009-08-07 21:33:25 +0000 | [diff] [blame] | 2389 | // Multiply is commmutative. |
Dan Gohman | 757eee8 | 2009-08-02 16:10:52 +0000 | [diff] [blame] | 2390 | if (!foldedLoad) { |
Dan Gohman | ea6f91f | 2010-01-05 01:24:18 +0000 | [diff] [blame] | 2391 | foldedLoad = TryFoldLoad(Node, N0, Tmp0, Tmp1, Tmp2, Tmp3, Tmp4); |
Dan Gohman | 757eee8 | 2009-08-02 16:10:52 +0000 | [diff] [blame] | 2392 | if (foldedLoad) |
| 2393 | std::swap(N0, N1); |
| 2394 | } |
| 2395 | |
Michael Liao | f9f7b55 | 2012-09-26 08:22:37 +0000 | [diff] [blame] | 2396 | SDValue InFlag = CurDAG->getCopyToReg(CurDAG->getEntryNode(), dl, SrcReg, |
Craig Topper | a4fd6d6 | 2012-05-23 05:44:51 +0000 | [diff] [blame] | 2397 | N0, SDValue()).getValue(1); |
Michael Liao | f9f7b55 | 2012-09-26 08:22:37 +0000 | [diff] [blame] | 2398 | SDValue ResHi, ResLo; |
Dan Gohman | 757eee8 | 2009-08-02 16:10:52 +0000 | [diff] [blame] | 2399 | |
| 2400 | if (foldedLoad) { |
Michael Liao | f9f7b55 | 2012-09-26 08:22:37 +0000 | [diff] [blame] | 2401 | SDValue Chain; |
Dan Gohman | 757eee8 | 2009-08-02 16:10:52 +0000 | [diff] [blame] | 2402 | SDValue Ops[] = { Tmp0, Tmp1, Tmp2, Tmp3, Tmp4, N1.getOperand(0), |
| 2403 | InFlag }; |
Michael Liao | f9f7b55 | 2012-09-26 08:22:37 +0000 | [diff] [blame] | 2404 | if (MOpc == X86::MULX32rm || MOpc == X86::MULX64rm) { |
| 2405 | SDVTList VTs = CurDAG->getVTList(NVT, NVT, MVT::Other, MVT::Glue); |
Michael Liao | b53d896 | 2013-04-19 22:22:57 +0000 | [diff] [blame] | 2406 | SDNode *CNode = CurDAG->getMachineNode(MOpc, dl, VTs, Ops); |
Michael Liao | f9f7b55 | 2012-09-26 08:22:37 +0000 | [diff] [blame] | 2407 | ResHi = SDValue(CNode, 0); |
| 2408 | ResLo = SDValue(CNode, 1); |
| 2409 | Chain = SDValue(CNode, 2); |
| 2410 | InFlag = SDValue(CNode, 3); |
| 2411 | } else { |
| 2412 | SDVTList VTs = CurDAG->getVTList(MVT::Other, MVT::Glue); |
Michael Liao | b53d896 | 2013-04-19 22:22:57 +0000 | [diff] [blame] | 2413 | SDNode *CNode = CurDAG->getMachineNode(MOpc, dl, VTs, Ops); |
Michael Liao | f9f7b55 | 2012-09-26 08:22:37 +0000 | [diff] [blame] | 2414 | Chain = SDValue(CNode, 0); |
| 2415 | InFlag = SDValue(CNode, 1); |
| 2416 | } |
Chris Lattner | 364bb0a | 2010-12-05 07:30:36 +0000 | [diff] [blame] | 2417 | |
Dan Gohman | 757eee8 | 2009-08-02 16:10:52 +0000 | [diff] [blame] | 2418 | // Update the chain. |
Michael Liao | f9f7b55 | 2012-09-26 08:22:37 +0000 | [diff] [blame] | 2419 | ReplaceUses(N1.getValue(1), Chain); |
Dan Gohman | 757eee8 | 2009-08-02 16:10:52 +0000 | [diff] [blame] | 2420 | } else { |
Michael Liao | f9f7b55 | 2012-09-26 08:22:37 +0000 | [diff] [blame] | 2421 | SDValue Ops[] = { N1, InFlag }; |
| 2422 | if (Opc == X86::MULX32rr || Opc == X86::MULX64rr) { |
| 2423 | SDVTList VTs = CurDAG->getVTList(NVT, NVT, MVT::Glue); |
Michael Liao | b53d896 | 2013-04-19 22:22:57 +0000 | [diff] [blame] | 2424 | SDNode *CNode = CurDAG->getMachineNode(Opc, dl, VTs, Ops); |
Michael Liao | f9f7b55 | 2012-09-26 08:22:37 +0000 | [diff] [blame] | 2425 | ResHi = SDValue(CNode, 0); |
| 2426 | ResLo = SDValue(CNode, 1); |
| 2427 | InFlag = SDValue(CNode, 2); |
| 2428 | } else { |
| 2429 | SDVTList VTs = CurDAG->getVTList(MVT::Glue); |
Michael Liao | b53d896 | 2013-04-19 22:22:57 +0000 | [diff] [blame] | 2430 | SDNode *CNode = CurDAG->getMachineNode(Opc, dl, VTs, Ops); |
Michael Liao | f9f7b55 | 2012-09-26 08:22:37 +0000 | [diff] [blame] | 2431 | InFlag = SDValue(CNode, 0); |
| 2432 | } |
Dan Gohman | 757eee8 | 2009-08-02 16:10:52 +0000 | [diff] [blame] | 2433 | } |
| 2434 | |
Jakob Stoklund Olesen | d7d0d4e | 2010-06-26 00:39:23 +0000 | [diff] [blame] | 2435 | // Prevent use of AH in a REX instruction by referencing AX instead. |
| 2436 | if (HiReg == X86::AH && Subtarget->is64Bit() && |
| 2437 | !SDValue(Node, 1).use_empty()) { |
| 2438 | SDValue Result = CurDAG->getCopyFromReg(CurDAG->getEntryNode(), dl, |
| 2439 | X86::AX, MVT::i16, InFlag); |
| 2440 | InFlag = Result.getValue(2); |
| 2441 | // Get the low part if needed. Don't use getCopyFromReg for aliasing |
| 2442 | // registers. |
| 2443 | if (!SDValue(Node, 0).use_empty()) |
| 2444 | ReplaceUses(SDValue(Node, 1), |
| 2445 | CurDAG->getTargetExtractSubreg(X86::sub_8bit, dl, MVT::i8, Result)); |
| 2446 | |
| 2447 | // Shift AX down 8 bits. |
| 2448 | Result = SDValue(CurDAG->getMachineNode(X86::SHR16ri, dl, MVT::i16, |
| 2449 | Result, |
Sergey Dmitrouk | 842a51b | 2015-04-28 14:05:47 +0000 | [diff] [blame] | 2450 | CurDAG->getTargetConstant(8, dl, MVT::i8)), |
| 2451 | 0); |
Jakob Stoklund Olesen | d7d0d4e | 2010-06-26 00:39:23 +0000 | [diff] [blame] | 2452 | // Then truncate it down to i8. |
| 2453 | ReplaceUses(SDValue(Node, 1), |
| 2454 | CurDAG->getTargetExtractSubreg(X86::sub_8bit, dl, MVT::i8, Result)); |
| 2455 | } |
Dan Gohman | 757eee8 | 2009-08-02 16:10:52 +0000 | [diff] [blame] | 2456 | // Copy the low half of the result, if it is needed. |
Dan Gohman | ea6f91f | 2010-01-05 01:24:18 +0000 | [diff] [blame] | 2457 | if (!SDValue(Node, 0).use_empty()) { |
Craig Topper | 062a2ba | 2014-04-25 05:30:21 +0000 | [diff] [blame] | 2458 | if (!ResLo.getNode()) { |
Michael Liao | f9f7b55 | 2012-09-26 08:22:37 +0000 | [diff] [blame] | 2459 | assert(LoReg && "Register for low half is not defined!"); |
| 2460 | ResLo = CurDAG->getCopyFromReg(CurDAG->getEntryNode(), dl, LoReg, NVT, |
| 2461 | InFlag); |
| 2462 | InFlag = ResLo.getValue(2); |
| 2463 | } |
| 2464 | ReplaceUses(SDValue(Node, 0), ResLo); |
| 2465 | DEBUG(dbgs() << "=> "; ResLo.getNode()->dump(CurDAG); dbgs() << '\n'); |
Dan Gohman | 757eee8 | 2009-08-02 16:10:52 +0000 | [diff] [blame] | 2466 | } |
| 2467 | // Copy the high half of the result, if it is needed. |
Dan Gohman | ea6f91f | 2010-01-05 01:24:18 +0000 | [diff] [blame] | 2468 | if (!SDValue(Node, 1).use_empty()) { |
Craig Topper | 062a2ba | 2014-04-25 05:30:21 +0000 | [diff] [blame] | 2469 | if (!ResHi.getNode()) { |
Michael Liao | f9f7b55 | 2012-09-26 08:22:37 +0000 | [diff] [blame] | 2470 | assert(HiReg && "Register for high half is not defined!"); |
| 2471 | ResHi = CurDAG->getCopyFromReg(CurDAG->getEntryNode(), dl, HiReg, NVT, |
| 2472 | InFlag); |
| 2473 | InFlag = ResHi.getValue(2); |
| 2474 | } |
| 2475 | ReplaceUses(SDValue(Node, 1), ResHi); |
| 2476 | DEBUG(dbgs() << "=> "; ResHi.getNode()->dump(CurDAG); dbgs() << '\n'); |
Dan Gohman | 757eee8 | 2009-08-02 16:10:52 +0000 | [diff] [blame] | 2477 | } |
Chad Rosier | 24c19d2 | 2012-08-01 18:39:17 +0000 | [diff] [blame] | 2478 | |
Craig Topper | 062a2ba | 2014-04-25 05:30:21 +0000 | [diff] [blame] | 2479 | return nullptr; |
Dan Gohman | 757eee8 | 2009-08-02 16:10:52 +0000 | [diff] [blame] | 2480 | } |
| 2481 | |
| 2482 | case ISD::SDIVREM: |
Ahmed Bougacha | 12eb558 | 2014-11-03 20:26:35 +0000 | [diff] [blame] | 2483 | case ISD::UDIVREM: |
| 2484 | case X86ISD::SDIVREM8_SEXT_HREG: |
| 2485 | case X86ISD::UDIVREM8_ZEXT_HREG: { |
Dan Gohman | 757eee8 | 2009-08-02 16:10:52 +0000 | [diff] [blame] | 2486 | SDValue N0 = Node->getOperand(0); |
| 2487 | SDValue N1 = Node->getOperand(1); |
| 2488 | |
Ahmed Bougacha | 12eb558 | 2014-11-03 20:26:35 +0000 | [diff] [blame] | 2489 | bool isSigned = (Opcode == ISD::SDIVREM || |
| 2490 | Opcode == X86ISD::SDIVREM8_SEXT_HREG); |
Bill Wendling | fe3bdb4 | 2009-08-07 21:33:25 +0000 | [diff] [blame] | 2491 | if (!isSigned) { |
Craig Topper | 83e042a | 2013-08-15 05:57:07 +0000 | [diff] [blame] | 2492 | switch (NVT.SimpleTy) { |
Dan Gohman | 757eee8 | 2009-08-02 16:10:52 +0000 | [diff] [blame] | 2493 | default: llvm_unreachable("Unsupported VT!"); |
Owen Anderson | 9f94459 | 2009-08-11 20:47:22 +0000 | [diff] [blame] | 2494 | case MVT::i8: Opc = X86::DIV8r; MOpc = X86::DIV8m; break; |
| 2495 | case MVT::i16: Opc = X86::DIV16r; MOpc = X86::DIV16m; break; |
| 2496 | case MVT::i32: Opc = X86::DIV32r; MOpc = X86::DIV32m; break; |
| 2497 | case MVT::i64: Opc = X86::DIV64r; MOpc = X86::DIV64m; break; |
Dan Gohman | 757eee8 | 2009-08-02 16:10:52 +0000 | [diff] [blame] | 2498 | } |
Bill Wendling | fe3bdb4 | 2009-08-07 21:33:25 +0000 | [diff] [blame] | 2499 | } else { |
Craig Topper | 83e042a | 2013-08-15 05:57:07 +0000 | [diff] [blame] | 2500 | switch (NVT.SimpleTy) { |
Dan Gohman | 757eee8 | 2009-08-02 16:10:52 +0000 | [diff] [blame] | 2501 | default: llvm_unreachable("Unsupported VT!"); |
Owen Anderson | 9f94459 | 2009-08-11 20:47:22 +0000 | [diff] [blame] | 2502 | case MVT::i8: Opc = X86::IDIV8r; MOpc = X86::IDIV8m; break; |
| 2503 | case MVT::i16: Opc = X86::IDIV16r; MOpc = X86::IDIV16m; break; |
| 2504 | case MVT::i32: Opc = X86::IDIV32r; MOpc = X86::IDIV32m; break; |
| 2505 | case MVT::i64: Opc = X86::IDIV64r; MOpc = X86::IDIV64m; break; |
Dan Gohman | 757eee8 | 2009-08-02 16:10:52 +0000 | [diff] [blame] | 2506 | } |
Bill Wendling | fe3bdb4 | 2009-08-07 21:33:25 +0000 | [diff] [blame] | 2507 | } |
Dan Gohman | 757eee8 | 2009-08-02 16:10:52 +0000 | [diff] [blame] | 2508 | |
Chris Lattner | 518b037 | 2009-12-23 01:45:04 +0000 | [diff] [blame] | 2509 | unsigned LoReg, HiReg, ClrReg; |
Tim Northover | 64ec0ff | 2013-05-30 13:19:42 +0000 | [diff] [blame] | 2510 | unsigned SExtOpcode; |
Craig Topper | 83e042a | 2013-08-15 05:57:07 +0000 | [diff] [blame] | 2511 | switch (NVT.SimpleTy) { |
Dan Gohman | 757eee8 | 2009-08-02 16:10:52 +0000 | [diff] [blame] | 2512 | default: llvm_unreachable("Unsupported VT!"); |
Owen Anderson | 9f94459 | 2009-08-11 20:47:22 +0000 | [diff] [blame] | 2513 | case MVT::i8: |
Chris Lattner | 518b037 | 2009-12-23 01:45:04 +0000 | [diff] [blame] | 2514 | LoReg = X86::AL; ClrReg = HiReg = X86::AH; |
Dan Gohman | 757eee8 | 2009-08-02 16:10:52 +0000 | [diff] [blame] | 2515 | SExtOpcode = X86::CBW; |
| 2516 | break; |
Owen Anderson | 9f94459 | 2009-08-11 20:47:22 +0000 | [diff] [blame] | 2517 | case MVT::i16: |
Dan Gohman | 757eee8 | 2009-08-02 16:10:52 +0000 | [diff] [blame] | 2518 | LoReg = X86::AX; HiReg = X86::DX; |
Tim Northover | 64ec0ff | 2013-05-30 13:19:42 +0000 | [diff] [blame] | 2519 | ClrReg = X86::DX; |
Dan Gohman | 757eee8 | 2009-08-02 16:10:52 +0000 | [diff] [blame] | 2520 | SExtOpcode = X86::CWD; |
| 2521 | break; |
Owen Anderson | 9f94459 | 2009-08-11 20:47:22 +0000 | [diff] [blame] | 2522 | case MVT::i32: |
Chris Lattner | 518b037 | 2009-12-23 01:45:04 +0000 | [diff] [blame] | 2523 | LoReg = X86::EAX; ClrReg = HiReg = X86::EDX; |
Dan Gohman | 757eee8 | 2009-08-02 16:10:52 +0000 | [diff] [blame] | 2524 | SExtOpcode = X86::CDQ; |
| 2525 | break; |
Owen Anderson | 9f94459 | 2009-08-11 20:47:22 +0000 | [diff] [blame] | 2526 | case MVT::i64: |
Chris Lattner | 518b037 | 2009-12-23 01:45:04 +0000 | [diff] [blame] | 2527 | LoReg = X86::RAX; ClrReg = HiReg = X86::RDX; |
Dan Gohman | 757eee8 | 2009-08-02 16:10:52 +0000 | [diff] [blame] | 2528 | SExtOpcode = X86::CQO; |
Evan Cheng | e62288f | 2009-07-30 08:33:02 +0000 | [diff] [blame] | 2529 | break; |
| 2530 | } |
| 2531 | |
Dan Gohman | 757eee8 | 2009-08-02 16:10:52 +0000 | [diff] [blame] | 2532 | SDValue Tmp0, Tmp1, Tmp2, Tmp3, Tmp4; |
Dan Gohman | ea6f91f | 2010-01-05 01:24:18 +0000 | [diff] [blame] | 2533 | bool foldedLoad = TryFoldLoad(Node, N1, Tmp0, Tmp1, Tmp2, Tmp3, Tmp4); |
Dan Gohman | 757eee8 | 2009-08-02 16:10:52 +0000 | [diff] [blame] | 2534 | bool signBitIsZero = CurDAG->SignBitIsZero(N0); |
Dan Gohman | a160361 | 2007-10-08 18:33:35 +0000 | [diff] [blame] | 2535 | |
Dan Gohman | 757eee8 | 2009-08-02 16:10:52 +0000 | [diff] [blame] | 2536 | SDValue InFlag; |
Owen Anderson | 9f94459 | 2009-08-11 20:47:22 +0000 | [diff] [blame] | 2537 | if (NVT == MVT::i8 && (!isSigned || signBitIsZero)) { |
Dan Gohman | 757eee8 | 2009-08-02 16:10:52 +0000 | [diff] [blame] | 2538 | // Special case for div8, just use a move with zero extension to AX to |
| 2539 | // clear the upper 8 bits (AH). |
| 2540 | SDValue Tmp0, Tmp1, Tmp2, Tmp3, Tmp4, Move, Chain; |
Dan Gohman | ea6f91f | 2010-01-05 01:24:18 +0000 | [diff] [blame] | 2541 | if (TryFoldLoad(Node, N0, Tmp0, Tmp1, Tmp2, Tmp3, Tmp4)) { |
Dan Gohman | 757eee8 | 2009-08-02 16:10:52 +0000 | [diff] [blame] | 2542 | SDValue Ops[] = { Tmp0, Tmp1, Tmp2, Tmp3, Tmp4, N0.getOperand(0) }; |
| 2543 | Move = |
Stuart Hastings | 91f1d24 | 2011-05-20 19:04:40 +0000 | [diff] [blame] | 2544 | SDValue(CurDAG->getMachineNode(X86::MOVZX32rm8, dl, MVT::i32, |
Michael Liao | b53d896 | 2013-04-19 22:22:57 +0000 | [diff] [blame] | 2545 | MVT::Other, Ops), 0); |
Dan Gohman | 757eee8 | 2009-08-02 16:10:52 +0000 | [diff] [blame] | 2546 | Chain = Move.getValue(1); |
| 2547 | ReplaceUses(N0.getValue(1), Chain); |
Evan Cheng | 10d2790 | 2006-01-06 20:36:21 +0000 | [diff] [blame] | 2548 | } else { |
Dan Gohman | 757eee8 | 2009-08-02 16:10:52 +0000 | [diff] [blame] | 2549 | Move = |
Stuart Hastings | 91f1d24 | 2011-05-20 19:04:40 +0000 | [diff] [blame] | 2550 | SDValue(CurDAG->getMachineNode(X86::MOVZX32rr8, dl, MVT::i32, N0),0); |
Dan Gohman | 757eee8 | 2009-08-02 16:10:52 +0000 | [diff] [blame] | 2551 | Chain = CurDAG->getEntryNode(); |
| 2552 | } |
Stuart Hastings | 91f1d24 | 2011-05-20 19:04:40 +0000 | [diff] [blame] | 2553 | Chain = CurDAG->getCopyToReg(Chain, dl, X86::EAX, Move, SDValue()); |
Dan Gohman | 757eee8 | 2009-08-02 16:10:52 +0000 | [diff] [blame] | 2554 | InFlag = Chain.getValue(1); |
| 2555 | } else { |
| 2556 | InFlag = |
| 2557 | CurDAG->getCopyToReg(CurDAG->getEntryNode(), dl, |
| 2558 | LoReg, N0, SDValue()).getValue(1); |
| 2559 | if (isSigned && !signBitIsZero) { |
| 2560 | // Sign extend the low part into the high part. |
Evan Cheng | d1b82d8 | 2006-02-09 07:17:49 +0000 | [diff] [blame] | 2561 | InFlag = |
Chris Lattner | 3e5fbd7 | 2010-12-21 02:38:05 +0000 | [diff] [blame] | 2562 | SDValue(CurDAG->getMachineNode(SExtOpcode, dl, MVT::Glue, InFlag),0); |
Dan Gohman | 757eee8 | 2009-08-02 16:10:52 +0000 | [diff] [blame] | 2563 | } else { |
| 2564 | // Zero out the high part, effectively zero extending the input. |
Michael Liao | 5bf9578 | 2014-12-04 05:20:33 +0000 | [diff] [blame] | 2565 | SDValue ClrNode = SDValue(CurDAG->getMachineNode(X86::MOV32r0, dl, NVT), 0); |
Craig Topper | 83e042a | 2013-08-15 05:57:07 +0000 | [diff] [blame] | 2566 | switch (NVT.SimpleTy) { |
Tim Northover | 64ec0ff | 2013-05-30 13:19:42 +0000 | [diff] [blame] | 2567 | case MVT::i16: |
| 2568 | ClrNode = |
| 2569 | SDValue(CurDAG->getMachineNode( |
| 2570 | TargetOpcode::EXTRACT_SUBREG, dl, MVT::i16, ClrNode, |
Sergey Dmitrouk | 842a51b | 2015-04-28 14:05:47 +0000 | [diff] [blame] | 2571 | CurDAG->getTargetConstant(X86::sub_16bit, dl, |
| 2572 | MVT::i32)), |
Tim Northover | 64ec0ff | 2013-05-30 13:19:42 +0000 | [diff] [blame] | 2573 | 0); |
| 2574 | break; |
| 2575 | case MVT::i32: |
| 2576 | break; |
| 2577 | case MVT::i64: |
| 2578 | ClrNode = |
| 2579 | SDValue(CurDAG->getMachineNode( |
| 2580 | TargetOpcode::SUBREG_TO_REG, dl, MVT::i64, |
Sergey Dmitrouk | 842a51b | 2015-04-28 14:05:47 +0000 | [diff] [blame] | 2581 | CurDAG->getTargetConstant(0, dl, MVT::i64), ClrNode, |
| 2582 | CurDAG->getTargetConstant(X86::sub_32bit, dl, |
| 2583 | MVT::i32)), |
Tim Northover | 64ec0ff | 2013-05-30 13:19:42 +0000 | [diff] [blame] | 2584 | 0); |
| 2585 | break; |
| 2586 | default: |
| 2587 | llvm_unreachable("Unexpected division source"); |
| 2588 | } |
| 2589 | |
Chris Lattner | 518b037 | 2009-12-23 01:45:04 +0000 | [diff] [blame] | 2590 | InFlag = CurDAG->getCopyToReg(CurDAG->getEntryNode(), dl, ClrReg, |
Dan Gohman | 757eee8 | 2009-08-02 16:10:52 +0000 | [diff] [blame] | 2591 | ClrNode, InFlag).getValue(1); |
Dan Gohman | a160361 | 2007-10-08 18:33:35 +0000 | [diff] [blame] | 2592 | } |
Evan Cheng | 92e2797 | 2006-01-06 23:19:29 +0000 | [diff] [blame] | 2593 | } |
Dan Gohman | a160361 | 2007-10-08 18:33:35 +0000 | [diff] [blame] | 2594 | |
Dan Gohman | 757eee8 | 2009-08-02 16:10:52 +0000 | [diff] [blame] | 2595 | if (foldedLoad) { |
| 2596 | SDValue Ops[] = { Tmp0, Tmp1, Tmp2, Tmp3, Tmp4, N1.getOperand(0), |
| 2597 | InFlag }; |
| 2598 | SDNode *CNode = |
Michael Liao | b53d896 | 2013-04-19 22:22:57 +0000 | [diff] [blame] | 2599 | CurDAG->getMachineNode(MOpc, dl, MVT::Other, MVT::Glue, Ops); |
Dan Gohman | 757eee8 | 2009-08-02 16:10:52 +0000 | [diff] [blame] | 2600 | InFlag = SDValue(CNode, 1); |
| 2601 | // Update the chain. |
| 2602 | ReplaceUses(N1.getValue(1), SDValue(CNode, 0)); |
| 2603 | } else { |
| 2604 | InFlag = |
Chris Lattner | 3e5fbd7 | 2010-12-21 02:38:05 +0000 | [diff] [blame] | 2605 | SDValue(CurDAG->getMachineNode(Opc, dl, MVT::Glue, N1, InFlag), 0); |
Dan Gohman | 757eee8 | 2009-08-02 16:10:52 +0000 | [diff] [blame] | 2606 | } |
Evan Cheng | 92e2797 | 2006-01-06 23:19:29 +0000 | [diff] [blame] | 2607 | |
Ahmed Bougacha | 12eb558 | 2014-11-03 20:26:35 +0000 | [diff] [blame] | 2608 | // Prevent use of AH in a REX instruction by explicitly copying it to |
| 2609 | // an ABCD_L register. |
Jim Grosbach | 340b6da | 2013-07-09 02:07:28 +0000 | [diff] [blame] | 2610 | // |
| 2611 | // The current assumption of the register allocator is that isel |
Ahmed Bougacha | 12eb558 | 2014-11-03 20:26:35 +0000 | [diff] [blame] | 2612 | // won't generate explicit references to the GR8_ABCD_H registers. If |
Jim Grosbach | 340b6da | 2013-07-09 02:07:28 +0000 | [diff] [blame] | 2613 | // the allocator and/or the backend get enhanced to be more robust in |
| 2614 | // that regard, this can be, and should be, removed. |
Ahmed Bougacha | 12eb558 | 2014-11-03 20:26:35 +0000 | [diff] [blame] | 2615 | if (HiReg == X86::AH && !SDValue(Node, 1).use_empty()) { |
| 2616 | SDValue AHCopy = CurDAG->getRegister(X86::AH, MVT::i8); |
| 2617 | unsigned AHExtOpcode = |
| 2618 | isSigned ? X86::MOVSX32_NOREXrr8 : X86::MOVZX32_NOREXrr8; |
Jakob Stoklund Olesen | d7d0d4e | 2010-06-26 00:39:23 +0000 | [diff] [blame] | 2619 | |
Ahmed Bougacha | 12eb558 | 2014-11-03 20:26:35 +0000 | [diff] [blame] | 2620 | SDNode *RNode = CurDAG->getMachineNode(AHExtOpcode, dl, MVT::i32, |
| 2621 | MVT::Glue, AHCopy, InFlag); |
| 2622 | SDValue Result(RNode, 0); |
| 2623 | InFlag = SDValue(RNode, 1); |
Jakob Stoklund Olesen | d7d0d4e | 2010-06-26 00:39:23 +0000 | [diff] [blame] | 2624 | |
Ahmed Bougacha | 12eb558 | 2014-11-03 20:26:35 +0000 | [diff] [blame] | 2625 | if (Opcode == X86ISD::UDIVREM8_ZEXT_HREG || |
| 2626 | Opcode == X86ISD::SDIVREM8_SEXT_HREG) { |
| 2627 | if (Node->getValueType(1) == MVT::i64) { |
| 2628 | // It's not possible to directly movsx AH to a 64bit register, because |
| 2629 | // the latter needs the REX prefix, but the former can't have it. |
| 2630 | assert(Opcode != X86ISD::SDIVREM8_SEXT_HREG && |
| 2631 | "Unexpected i64 sext of h-register"); |
| 2632 | Result = |
| 2633 | SDValue(CurDAG->getMachineNode( |
| 2634 | TargetOpcode::SUBREG_TO_REG, dl, MVT::i64, |
Sergey Dmitrouk | 842a51b | 2015-04-28 14:05:47 +0000 | [diff] [blame] | 2635 | CurDAG->getTargetConstant(0, dl, MVT::i64), Result, |
| 2636 | CurDAG->getTargetConstant(X86::sub_32bit, dl, |
| 2637 | MVT::i32)), |
Ahmed Bougacha | 12eb558 | 2014-11-03 20:26:35 +0000 | [diff] [blame] | 2638 | 0); |
| 2639 | } |
| 2640 | } else { |
| 2641 | Result = |
| 2642 | CurDAG->getTargetExtractSubreg(X86::sub_8bit, dl, MVT::i8, Result); |
| 2643 | } |
| 2644 | ReplaceUses(SDValue(Node, 1), Result); |
| 2645 | DEBUG(dbgs() << "=> "; Result.getNode()->dump(CurDAG); dbgs() << '\n'); |
Jakob Stoklund Olesen | d7d0d4e | 2010-06-26 00:39:23 +0000 | [diff] [blame] | 2646 | } |
Dan Gohman | 757eee8 | 2009-08-02 16:10:52 +0000 | [diff] [blame] | 2647 | // Copy the division (low) result, if it is needed. |
Dan Gohman | ea6f91f | 2010-01-05 01:24:18 +0000 | [diff] [blame] | 2648 | if (!SDValue(Node, 0).use_empty()) { |
Dan Gohman | 757eee8 | 2009-08-02 16:10:52 +0000 | [diff] [blame] | 2649 | SDValue Result = CurDAG->getCopyFromReg(CurDAG->getEntryNode(), dl, |
| 2650 | LoReg, NVT, InFlag); |
| 2651 | InFlag = Result.getValue(2); |
Dan Gohman | ea6f91f | 2010-01-05 01:24:18 +0000 | [diff] [blame] | 2652 | ReplaceUses(SDValue(Node, 0), Result); |
Chris Lattner | f98f124 | 2010-03-02 06:34:30 +0000 | [diff] [blame] | 2653 | DEBUG(dbgs() << "=> "; Result.getNode()->dump(CurDAG); dbgs() << '\n'); |
Dan Gohman | 757eee8 | 2009-08-02 16:10:52 +0000 | [diff] [blame] | 2654 | } |
| 2655 | // Copy the remainder (high) result, if it is needed. |
Dan Gohman | ea6f91f | 2010-01-05 01:24:18 +0000 | [diff] [blame] | 2656 | if (!SDValue(Node, 1).use_empty()) { |
Jakob Stoklund Olesen | d7d0d4e | 2010-06-26 00:39:23 +0000 | [diff] [blame] | 2657 | SDValue Result = CurDAG->getCopyFromReg(CurDAG->getEntryNode(), dl, |
| 2658 | HiReg, NVT, InFlag); |
| 2659 | InFlag = Result.getValue(2); |
Dan Gohman | ea6f91f | 2010-01-05 01:24:18 +0000 | [diff] [blame] | 2660 | ReplaceUses(SDValue(Node, 1), Result); |
Chris Lattner | f98f124 | 2010-03-02 06:34:30 +0000 | [diff] [blame] | 2661 | DEBUG(dbgs() << "=> "; Result.getNode()->dump(CurDAG); dbgs() << '\n'); |
Dan Gohman | 757eee8 | 2009-08-02 16:10:52 +0000 | [diff] [blame] | 2662 | } |
Craig Topper | 062a2ba | 2014-04-25 05:30:21 +0000 | [diff] [blame] | 2663 | return nullptr; |
Dan Gohman | 757eee8 | 2009-08-02 16:10:52 +0000 | [diff] [blame] | 2664 | } |
| 2665 | |
Manman Ren | 1be131b | 2012-08-08 00:51:41 +0000 | [diff] [blame] | 2666 | case X86ISD::CMP: |
| 2667 | case X86ISD::SUB: { |
| 2668 | // Sometimes a SUB is used to perform comparison. |
| 2669 | if (Opcode == X86ISD::SUB && Node->hasAnyUseOfValue(0)) |
| 2670 | // This node is not a CMP. |
| 2671 | break; |
Dan Gohman | ac33a90 | 2009-08-19 18:16:17 +0000 | [diff] [blame] | 2672 | SDValue N0 = Node->getOperand(0); |
| 2673 | SDValue N1 = Node->getOperand(1); |
| 2674 | |
Elena Demikhovsky | 34d2d76 | 2014-08-18 11:59:06 +0000 | [diff] [blame] | 2675 | if (N0.getOpcode() == ISD::TRUNCATE && N0.hasOneUse() && |
Elena Demikhovsky | d2cb3c8 | 2015-02-12 08:40:34 +0000 | [diff] [blame] | 2676 | HasNoSignedComparisonUses(Node)) |
Elena Demikhovsky | 34d2d76 | 2014-08-18 11:59:06 +0000 | [diff] [blame] | 2677 | N0 = N0.getOperand(0); |
Elena Demikhovsky | d2cb3c8 | 2015-02-12 08:40:34 +0000 | [diff] [blame] | 2678 | |
Dan Gohman | ac33a90 | 2009-08-19 18:16:17 +0000 | [diff] [blame] | 2679 | // Look for (X86cmp (and $op, $imm), 0) and see if we can convert it to |
| 2680 | // use a smaller encoding. |
Elena Demikhovsky | 34d2d76 | 2014-08-18 11:59:06 +0000 | [diff] [blame] | 2681 | // Look past the truncate if CMP is the only use of it. |
Dan Gohman | 198b7ff | 2011-11-03 21:49:52 +0000 | [diff] [blame] | 2682 | if ((N0.getNode()->getOpcode() == ISD::AND || |
| 2683 | (N0.getResNo() == 0 && N0.getNode()->getOpcode() == X86ISD::AND)) && |
| 2684 | N0.getNode()->hasOneUse() && |
Dan Gohman | ac33a90 | 2009-08-19 18:16:17 +0000 | [diff] [blame] | 2685 | N0.getValueType() != MVT::i8 && |
| 2686 | X86::isZeroNode(N1)) { |
| 2687 | ConstantSDNode *C = dyn_cast<ConstantSDNode>(N0.getNode()->getOperand(1)); |
| 2688 | if (!C) break; |
| 2689 | |
| 2690 | // For example, convert "testl %eax, $8" to "testb %al, $8" |
Dan Gohman | 7d9dffb | 2009-10-09 20:35:19 +0000 | [diff] [blame] | 2691 | if ((C->getZExtValue() & ~UINT64_C(0xff)) == 0 && |
| 2692 | (!(C->getZExtValue() & 0x80) || |
| 2693 | HasNoSignedComparisonUses(Node))) { |
Sergey Dmitrouk | 842a51b | 2015-04-28 14:05:47 +0000 | [diff] [blame] | 2694 | SDValue Imm = CurDAG->getTargetConstant(C->getZExtValue(), dl, MVT::i8); |
Dan Gohman | ac33a90 | 2009-08-19 18:16:17 +0000 | [diff] [blame] | 2695 | SDValue Reg = N0.getNode()->getOperand(0); |
| 2696 | |
| 2697 | // On x86-32, only the ABCD registers have 8-bit subregisters. |
| 2698 | if (!Subtarget->is64Bit()) { |
Craig Topper | cc830f8 | 2012-02-22 07:28:11 +0000 | [diff] [blame] | 2699 | const TargetRegisterClass *TRC; |
Craig Topper | 5671010 | 2013-08-15 02:33:50 +0000 | [diff] [blame] | 2700 | switch (N0.getSimpleValueType().SimpleTy) { |
Dan Gohman | ac33a90 | 2009-08-19 18:16:17 +0000 | [diff] [blame] | 2701 | case MVT::i32: TRC = &X86::GR32_ABCDRegClass; break; |
| 2702 | case MVT::i16: TRC = &X86::GR16_ABCDRegClass; break; |
| 2703 | default: llvm_unreachable("Unsupported TEST operand type!"); |
| 2704 | } |
Sergey Dmitrouk | 842a51b | 2015-04-28 14:05:47 +0000 | [diff] [blame] | 2705 | SDValue RC = CurDAG->getTargetConstant(TRC->getID(), dl, MVT::i32); |
Dan Gohman | 32f71d7 | 2009-09-25 18:54:59 +0000 | [diff] [blame] | 2706 | Reg = SDValue(CurDAG->getMachineNode(X86::COPY_TO_REGCLASS, dl, |
| 2707 | Reg.getValueType(), Reg, RC), 0); |
Dan Gohman | ac33a90 | 2009-08-19 18:16:17 +0000 | [diff] [blame] | 2708 | } |
| 2709 | |
| 2710 | // Extract the l-register. |
Jakob Stoklund Olesen | 9340ea5 | 2010-05-24 14:48:17 +0000 | [diff] [blame] | 2711 | SDValue Subreg = CurDAG->getTargetExtractSubreg(X86::sub_8bit, dl, |
Dan Gohman | ac33a90 | 2009-08-19 18:16:17 +0000 | [diff] [blame] | 2712 | MVT::i8, Reg); |
| 2713 | |
| 2714 | // Emit a testb. |
Manman Ren | 511c6d0 | 2012-09-28 18:53:24 +0000 | [diff] [blame] | 2715 | SDNode *NewNode = CurDAG->getMachineNode(X86::TEST8ri, dl, MVT::i32, |
| 2716 | Subreg, Imm); |
| 2717 | // Replace SUB|CMP with TEST, since SUB has two outputs while TEST has |
| 2718 | // one, do not call ReplaceAllUsesWith. |
| 2719 | ReplaceUses(SDValue(Node, (Opcode == X86ISD::SUB ? 1 : 0)), |
| 2720 | SDValue(NewNode, 0)); |
Craig Topper | 062a2ba | 2014-04-25 05:30:21 +0000 | [diff] [blame] | 2721 | return nullptr; |
Dan Gohman | ac33a90 | 2009-08-19 18:16:17 +0000 | [diff] [blame] | 2722 | } |
| 2723 | |
| 2724 | // For example, "testl %eax, $2048" to "testb %ah, $8". |
Dan Gohman | 7d9dffb | 2009-10-09 20:35:19 +0000 | [diff] [blame] | 2725 | if ((C->getZExtValue() & ~UINT64_C(0xff00)) == 0 && |
| 2726 | (!(C->getZExtValue() & 0x8000) || |
| 2727 | HasNoSignedComparisonUses(Node))) { |
Dan Gohman | ac33a90 | 2009-08-19 18:16:17 +0000 | [diff] [blame] | 2728 | // Shift the immediate right by 8 bits. |
| 2729 | SDValue ShiftedImm = CurDAG->getTargetConstant(C->getZExtValue() >> 8, |
Sergey Dmitrouk | 842a51b | 2015-04-28 14:05:47 +0000 | [diff] [blame] | 2730 | dl, MVT::i8); |
Dan Gohman | ac33a90 | 2009-08-19 18:16:17 +0000 | [diff] [blame] | 2731 | SDValue Reg = N0.getNode()->getOperand(0); |
| 2732 | |
| 2733 | // Put the value in an ABCD register. |
Craig Topper | cc830f8 | 2012-02-22 07:28:11 +0000 | [diff] [blame] | 2734 | const TargetRegisterClass *TRC; |
Craig Topper | 5671010 | 2013-08-15 02:33:50 +0000 | [diff] [blame] | 2735 | switch (N0.getSimpleValueType().SimpleTy) { |
Dan Gohman | ac33a90 | 2009-08-19 18:16:17 +0000 | [diff] [blame] | 2736 | case MVT::i64: TRC = &X86::GR64_ABCDRegClass; break; |
| 2737 | case MVT::i32: TRC = &X86::GR32_ABCDRegClass; break; |
| 2738 | case MVT::i16: TRC = &X86::GR16_ABCDRegClass; break; |
| 2739 | default: llvm_unreachable("Unsupported TEST operand type!"); |
| 2740 | } |
Sergey Dmitrouk | 842a51b | 2015-04-28 14:05:47 +0000 | [diff] [blame] | 2741 | SDValue RC = CurDAG->getTargetConstant(TRC->getID(), dl, MVT::i32); |
Dan Gohman | 32f71d7 | 2009-09-25 18:54:59 +0000 | [diff] [blame] | 2742 | Reg = SDValue(CurDAG->getMachineNode(X86::COPY_TO_REGCLASS, dl, |
| 2743 | Reg.getValueType(), Reg, RC), 0); |
Dan Gohman | ac33a90 | 2009-08-19 18:16:17 +0000 | [diff] [blame] | 2744 | |
| 2745 | // Extract the h-register. |
Jakob Stoklund Olesen | 9340ea5 | 2010-05-24 14:48:17 +0000 | [diff] [blame] | 2746 | SDValue Subreg = CurDAG->getTargetExtractSubreg(X86::sub_8bit_hi, dl, |
Dan Gohman | ac33a90 | 2009-08-19 18:16:17 +0000 | [diff] [blame] | 2747 | MVT::i8, Reg); |
| 2748 | |
Jakob Stoklund Olesen | 729abd3 | 2011-10-08 18:28:28 +0000 | [diff] [blame] | 2749 | // Emit a testb. The EXTRACT_SUBREG becomes a COPY that can only |
| 2750 | // target GR8_NOREX registers, so make sure the register class is |
| 2751 | // forced. |
Manman Ren | 511c6d0 | 2012-09-28 18:53:24 +0000 | [diff] [blame] | 2752 | SDNode *NewNode = CurDAG->getMachineNode(X86::TEST8ri_NOREX, dl, |
| 2753 | MVT::i32, Subreg, ShiftedImm); |
| 2754 | // Replace SUB|CMP with TEST, since SUB has two outputs while TEST has |
| 2755 | // one, do not call ReplaceAllUsesWith. |
| 2756 | ReplaceUses(SDValue(Node, (Opcode == X86ISD::SUB ? 1 : 0)), |
| 2757 | SDValue(NewNode, 0)); |
Craig Topper | 062a2ba | 2014-04-25 05:30:21 +0000 | [diff] [blame] | 2758 | return nullptr; |
Dan Gohman | ac33a90 | 2009-08-19 18:16:17 +0000 | [diff] [blame] | 2759 | } |
| 2760 | |
| 2761 | // For example, "testl %eax, $32776" to "testw %ax, $32776". |
| 2762 | if ((C->getZExtValue() & ~UINT64_C(0xffff)) == 0 && |
Dan Gohman | 7d9dffb | 2009-10-09 20:35:19 +0000 | [diff] [blame] | 2763 | N0.getValueType() != MVT::i16 && |
| 2764 | (!(C->getZExtValue() & 0x8000) || |
| 2765 | HasNoSignedComparisonUses(Node))) { |
Sergey Dmitrouk | 842a51b | 2015-04-28 14:05:47 +0000 | [diff] [blame] | 2766 | SDValue Imm = CurDAG->getTargetConstant(C->getZExtValue(), dl, |
| 2767 | MVT::i16); |
Dan Gohman | ac33a90 | 2009-08-19 18:16:17 +0000 | [diff] [blame] | 2768 | SDValue Reg = N0.getNode()->getOperand(0); |
| 2769 | |
| 2770 | // Extract the 16-bit subregister. |
Jakob Stoklund Olesen | 9340ea5 | 2010-05-24 14:48:17 +0000 | [diff] [blame] | 2771 | SDValue Subreg = CurDAG->getTargetExtractSubreg(X86::sub_16bit, dl, |
Dan Gohman | ac33a90 | 2009-08-19 18:16:17 +0000 | [diff] [blame] | 2772 | MVT::i16, Reg); |
| 2773 | |
| 2774 | // Emit a testw. |
Manman Ren | 511c6d0 | 2012-09-28 18:53:24 +0000 | [diff] [blame] | 2775 | SDNode *NewNode = CurDAG->getMachineNode(X86::TEST16ri, dl, MVT::i32, |
| 2776 | Subreg, Imm); |
| 2777 | // Replace SUB|CMP with TEST, since SUB has two outputs while TEST has |
| 2778 | // one, do not call ReplaceAllUsesWith. |
| 2779 | ReplaceUses(SDValue(Node, (Opcode == X86ISD::SUB ? 1 : 0)), |
| 2780 | SDValue(NewNode, 0)); |
Craig Topper | 062a2ba | 2014-04-25 05:30:21 +0000 | [diff] [blame] | 2781 | return nullptr; |
Dan Gohman | ac33a90 | 2009-08-19 18:16:17 +0000 | [diff] [blame] | 2782 | } |
| 2783 | |
| 2784 | // For example, "testq %rax, $268468232" to "testl %eax, $268468232". |
| 2785 | if ((C->getZExtValue() & ~UINT64_C(0xffffffff)) == 0 && |
Dan Gohman | 7d9dffb | 2009-10-09 20:35:19 +0000 | [diff] [blame] | 2786 | N0.getValueType() == MVT::i64 && |
| 2787 | (!(C->getZExtValue() & 0x80000000) || |
| 2788 | HasNoSignedComparisonUses(Node))) { |
Sergey Dmitrouk | 842a51b | 2015-04-28 14:05:47 +0000 | [diff] [blame] | 2789 | SDValue Imm = CurDAG->getTargetConstant(C->getZExtValue(), dl, |
| 2790 | MVT::i32); |
Dan Gohman | ac33a90 | 2009-08-19 18:16:17 +0000 | [diff] [blame] | 2791 | SDValue Reg = N0.getNode()->getOperand(0); |
| 2792 | |
| 2793 | // Extract the 32-bit subregister. |
Jakob Stoklund Olesen | 9340ea5 | 2010-05-24 14:48:17 +0000 | [diff] [blame] | 2794 | SDValue Subreg = CurDAG->getTargetExtractSubreg(X86::sub_32bit, dl, |
Dan Gohman | ac33a90 | 2009-08-19 18:16:17 +0000 | [diff] [blame] | 2795 | MVT::i32, Reg); |
| 2796 | |
| 2797 | // Emit a testl. |
Manman Ren | 511c6d0 | 2012-09-28 18:53:24 +0000 | [diff] [blame] | 2798 | SDNode *NewNode = CurDAG->getMachineNode(X86::TEST32ri, dl, MVT::i32, |
| 2799 | Subreg, Imm); |
| 2800 | // Replace SUB|CMP with TEST, since SUB has two outputs while TEST has |
| 2801 | // one, do not call ReplaceAllUsesWith. |
| 2802 | ReplaceUses(SDValue(Node, (Opcode == X86ISD::SUB ? 1 : 0)), |
| 2803 | SDValue(NewNode, 0)); |
Craig Topper | 062a2ba | 2014-04-25 05:30:21 +0000 | [diff] [blame] | 2804 | return nullptr; |
Dan Gohman | ac33a90 | 2009-08-19 18:16:17 +0000 | [diff] [blame] | 2805 | } |
| 2806 | } |
| 2807 | break; |
| 2808 | } |
Pete Cooper | 7c7ba1b | 2011-11-15 21:57:53 +0000 | [diff] [blame] | 2809 | case ISD::STORE: { |
Joel Jones | 68d59e8 | 2012-03-29 05:45:48 +0000 | [diff] [blame] | 2810 | // Change a chain of {load; incr or dec; store} of the same value into |
| 2811 | // a simple increment or decrement through memory of that value, if the |
| 2812 | // uses of the modified value and its address are suitable. |
Pete Cooper | 48784ed | 2011-11-16 19:03:23 +0000 | [diff] [blame] | 2813 | // The DEC64m tablegen pattern is currently not able to match the case where |
Chad Rosier | 24c19d2 | 2012-08-01 18:39:17 +0000 | [diff] [blame] | 2814 | // the EFLAGS on the original DEC are used. (This also applies to |
Joel Jones | 68d59e8 | 2012-03-29 05:45:48 +0000 | [diff] [blame] | 2815 | // {INC,DEC}X{64,32,16,8}.) |
| 2816 | // We'll need to improve tablegen to allow flags to be transferred from a |
Pete Cooper | 48784ed | 2011-11-16 19:03:23 +0000 | [diff] [blame] | 2817 | // node in the pattern to the result node. probably with a new keyword |
| 2818 | // for example, we have this |
| 2819 | // def DEC64m : RI<0xFF, MRM1m, (outs), (ins i64mem:$dst), "dec{q}\t$dst", |
| 2820 | // [(store (add (loadi64 addr:$dst), -1), addr:$dst), |
| 2821 | // (implicit EFLAGS)]>; |
| 2822 | // but maybe need something like this |
| 2823 | // def DEC64m : RI<0xFF, MRM1m, (outs), (ins i64mem:$dst), "dec{q}\t$dst", |
| 2824 | // [(store (add (loadi64 addr:$dst), -1), addr:$dst), |
| 2825 | // (transferrable EFLAGS)]>; |
Joel Jones | 68d59e8 | 2012-03-29 05:45:48 +0000 | [diff] [blame] | 2826 | |
Pete Cooper | 7c7ba1b | 2011-11-15 21:57:53 +0000 | [diff] [blame] | 2827 | StoreSDNode *StoreNode = cast<StoreSDNode>(Node); |
Pete Cooper | 7c7ba1b | 2011-11-15 21:57:53 +0000 | [diff] [blame] | 2828 | SDValue StoredVal = StoreNode->getOperand(1); |
Joel Jones | 68d59e8 | 2012-03-29 05:45:48 +0000 | [diff] [blame] | 2829 | unsigned Opc = StoredVal->getOpcode(); |
Pete Cooper | 7c7ba1b | 2011-11-15 21:57:53 +0000 | [diff] [blame] | 2830 | |
Craig Topper | 062a2ba | 2014-04-25 05:30:21 +0000 | [diff] [blame] | 2831 | LoadSDNode *LoadNode = nullptr; |
Evan Cheng | 3e869f0 | 2012-04-12 19:14:21 +0000 | [diff] [blame] | 2832 | SDValue InputChain; |
| 2833 | if (!isLoadIncOrDecStore(StoreNode, Opc, StoredVal, CurDAG, |
| 2834 | LoadNode, InputChain)) |
| 2835 | break; |
Pete Cooper | 7c7ba1b | 2011-11-15 21:57:53 +0000 | [diff] [blame] | 2836 | |
| 2837 | SDValue Base, Scale, Index, Disp, Segment; |
| 2838 | if (!SelectAddr(LoadNode, LoadNode->getBasePtr(), |
| 2839 | Base, Scale, Index, Disp, Segment)) |
| 2840 | break; |
| 2841 | |
| 2842 | MachineSDNode::mmo_iterator MemOp = MF->allocateMemRefsArray(2); |
| 2843 | MemOp[0] = StoreNode->getMemOperand(); |
| 2844 | MemOp[1] = LoadNode->getMemOperand(); |
| 2845 | const SDValue Ops[] = { Base, Scale, Index, Disp, Segment, InputChain }; |
Chad Rosier | 24c19d2 | 2012-08-01 18:39:17 +0000 | [diff] [blame] | 2846 | EVT LdVT = LoadNode->getMemoryVT(); |
Joel Jones | 68d59e8 | 2012-03-29 05:45:48 +0000 | [diff] [blame] | 2847 | unsigned newOpc = getFusedLdStOpcode(LdVT, Opc); |
| 2848 | MachineSDNode *Result = CurDAG->getMachineNode(newOpc, |
Andrew Trick | ef9de2a | 2013-05-25 02:42:55 +0000 | [diff] [blame] | 2849 | SDLoc(Node), |
Michael Liao | b53d896 | 2013-04-19 22:22:57 +0000 | [diff] [blame] | 2850 | MVT::i32, MVT::Other, Ops); |
Pete Cooper | 7c7ba1b | 2011-11-15 21:57:53 +0000 | [diff] [blame] | 2851 | Result->setMemRefs(MemOp, MemOp + 2); |
| 2852 | |
| 2853 | ReplaceUses(SDValue(StoreNode, 0), SDValue(Result, 1)); |
| 2854 | ReplaceUses(SDValue(StoredVal.getNode(), 1), SDValue(Result, 0)); |
| 2855 | |
| 2856 | return Result; |
| 2857 | } |
Chris Lattner | 655e7df | 2005-11-16 01:54:32 +0000 | [diff] [blame] | 2858 | } |
| 2859 | |
Dan Gohman | ea6f91f | 2010-01-05 01:24:18 +0000 | [diff] [blame] | 2860 | SDNode *ResNode = SelectCode(Node); |
Evan Cheng | bd1c5a8 | 2006-08-11 09:08:15 +0000 | [diff] [blame] | 2861 | |
Chris Lattner | f98f124 | 2010-03-02 06:34:30 +0000 | [diff] [blame] | 2862 | DEBUG(dbgs() << "=> "; |
Craig Topper | e73658d | 2014-04-28 04:05:08 +0000 | [diff] [blame] | 2863 | if (ResNode == nullptr || ResNode == Node) |
Chris Lattner | f98f124 | 2010-03-02 06:34:30 +0000 | [diff] [blame] | 2864 | Node->dump(CurDAG); |
| 2865 | else |
| 2866 | ResNode->dump(CurDAG); |
| 2867 | dbgs() << '\n'); |
Evan Cheng | bd1c5a8 | 2006-08-11 09:08:15 +0000 | [diff] [blame] | 2868 | |
| 2869 | return ResNode; |
Chris Lattner | 655e7df | 2005-11-16 01:54:32 +0000 | [diff] [blame] | 2870 | } |
| 2871 | |
Chris Lattner | ba1ed58 | 2006-06-08 18:03:49 +0000 | [diff] [blame] | 2872 | bool X86DAGToDAGISel:: |
Daniel Sanders | 60f1db0 | 2015-03-13 12:45:09 +0000 | [diff] [blame] | 2873 | SelectInlineAsmMemoryOperand(const SDValue &Op, unsigned ConstraintID, |
Dan Gohman | eb0cee9 | 2008-08-23 02:25:05 +0000 | [diff] [blame] | 2874 | std::vector<SDValue> &OutOps) { |
Rafael Espindola | 3b2df10 | 2009-04-08 21:14:34 +0000 | [diff] [blame] | 2875 | SDValue Op0, Op1, Op2, Op3, Op4; |
Daniel Sanders | 60f1db0 | 2015-03-13 12:45:09 +0000 | [diff] [blame] | 2876 | switch (ConstraintID) { |
| 2877 | case InlineAsm::Constraint_o: // offsetable ?? |
| 2878 | case InlineAsm::Constraint_v: // not offsetable ?? |
Chris Lattner | ba1ed58 | 2006-06-08 18:03:49 +0000 | [diff] [blame] | 2879 | default: return true; |
Daniel Sanders | 60f1db0 | 2015-03-13 12:45:09 +0000 | [diff] [blame] | 2880 | case InlineAsm::Constraint_m: // memory |
Craig Topper | 062a2ba | 2014-04-25 05:30:21 +0000 | [diff] [blame] | 2881 | if (!SelectAddr(nullptr, Op, Op0, Op1, Op2, Op3, Op4)) |
Chris Lattner | ba1ed58 | 2006-06-08 18:03:49 +0000 | [diff] [blame] | 2882 | return true; |
| 2883 | break; |
| 2884 | } |
Chad Rosier | 24c19d2 | 2012-08-01 18:39:17 +0000 | [diff] [blame] | 2885 | |
Evan Cheng | 2d48722 | 2006-08-26 01:05:16 +0000 | [diff] [blame] | 2886 | OutOps.push_back(Op0); |
| 2887 | OutOps.push_back(Op1); |
| 2888 | OutOps.push_back(Op2); |
| 2889 | OutOps.push_back(Op3); |
Rafael Espindola | 3b2df10 | 2009-04-08 21:14:34 +0000 | [diff] [blame] | 2890 | OutOps.push_back(Op4); |
Chris Lattner | ba1ed58 | 2006-06-08 18:03:49 +0000 | [diff] [blame] | 2891 | return false; |
| 2892 | } |
| 2893 | |
Chad Rosier | 24c19d2 | 2012-08-01 18:39:17 +0000 | [diff] [blame] | 2894 | /// createX86ISelDag - This pass converts a legalized DAG into a |
Chris Lattner | 655e7df | 2005-11-16 01:54:32 +0000 | [diff] [blame] | 2895 | /// X86-specific DAG, ready for instruction scheduling. |
| 2896 | /// |
Bill Wendling | 026e5d7 | 2009-04-29 23:29:43 +0000 | [diff] [blame] | 2897 | FunctionPass *llvm::createX86ISelDag(X86TargetMachine &TM, |
Craig Topper | f6e7e12 | 2012-03-27 07:21:54 +0000 | [diff] [blame] | 2898 | CodeGenOpt::Level OptLevel) { |
Bill Wendling | 084669a | 2009-04-29 00:15:41 +0000 | [diff] [blame] | 2899 | return new X86DAGToDAGISel(TM, OptLevel); |
Chris Lattner | 655e7df | 2005-11-16 01:54:32 +0000 | [diff] [blame] | 2900 | } |