blob: 75fd37f01a191e84f3581401d7a51358dcbb155f [file] [log] [blame]
Ulrich Weigand5f613df2013-05-06 16:15:19 +00001//===-- SystemZISelDAGToDAG.cpp - A dag to dag inst selector for SystemZ --===//
2//
3// The LLVM Compiler Infrastructure
4//
5// This file is distributed under the University of Illinois Open Source
6// License. See LICENSE.TXT for details.
7//
8//===----------------------------------------------------------------------===//
9//
10// This file defines an instruction selector for the SystemZ target.
11//
12//===----------------------------------------------------------------------===//
13
14#include "SystemZTargetMachine.h"
Richard Sandiford97846492013-07-09 09:46:39 +000015#include "llvm/Analysis/AliasAnalysis.h"
Ulrich Weigand5f613df2013-05-06 16:15:19 +000016#include "llvm/CodeGen/SelectionDAGISel.h"
17#include "llvm/Support/Debug.h"
18#include "llvm/Support/raw_ostream.h"
19
20using namespace llvm;
21
Chandler Carruthe96dd892014-04-21 22:55:11 +000022#define DEBUG_TYPE "systemz-isel"
23
Ulrich Weigand5f613df2013-05-06 16:15:19 +000024namespace {
25// Used to build addressing modes.
26struct SystemZAddressingMode {
27 // The shape of the address.
28 enum AddrForm {
29 // base+displacement
30 FormBD,
31
32 // base+displacement+index for load and store operands
33 FormBDXNormal,
34
35 // base+displacement+index for load address operands
36 FormBDXLA,
37
38 // base+displacement+index+ADJDYNALLOC
39 FormBDXDynAlloc
40 };
41 AddrForm Form;
42
43 // The type of displacement. The enum names here correspond directly
44 // to the definitions in SystemZOperand.td. We could split them into
45 // flags -- single/pair, 128-bit, etc. -- but it hardly seems worth it.
46 enum DispRange {
47 Disp12Only,
48 Disp12Pair,
49 Disp20Only,
50 Disp20Only128,
51 Disp20Pair
52 };
53 DispRange DR;
54
55 // The parts of the address. The address is equivalent to:
56 //
57 // Base + Disp + Index + (IncludesDynAlloc ? ADJDYNALLOC : 0)
58 SDValue Base;
59 int64_t Disp;
60 SDValue Index;
61 bool IncludesDynAlloc;
62
63 SystemZAddressingMode(AddrForm form, DispRange dr)
64 : Form(form), DR(dr), Base(), Disp(0), Index(),
65 IncludesDynAlloc(false) {}
66
67 // True if the address can have an index register.
68 bool hasIndexField() { return Form != FormBD; }
69
70 // True if the address can (and must) include ADJDYNALLOC.
71 bool isDynAlloc() { return Form == FormBDXDynAlloc; }
72
73 void dump() {
74 errs() << "SystemZAddressingMode " << this << '\n';
75
76 errs() << " Base ";
Craig Topper062a2ba2014-04-25 05:30:21 +000077 if (Base.getNode())
Ulrich Weigand5f613df2013-05-06 16:15:19 +000078 Base.getNode()->dump();
79 else
80 errs() << "null\n";
81
82 if (hasIndexField()) {
83 errs() << " Index ";
Craig Topper062a2ba2014-04-25 05:30:21 +000084 if (Index.getNode())
Ulrich Weigand5f613df2013-05-06 16:15:19 +000085 Index.getNode()->dump();
86 else
87 errs() << "null\n";
88 }
89
90 errs() << " Disp " << Disp;
91 if (IncludesDynAlloc)
92 errs() << " + ADJDYNALLOC";
93 errs() << '\n';
94 }
95};
96
Richard Sandiford82ec87d2013-07-16 11:02:24 +000097// Return a mask with Count low bits set.
98static uint64_t allOnes(unsigned int Count) {
Ulrich Weigand77884bc2015-06-25 11:52:36 +000099 assert(Count <= 64);
Justin Bognerc97c48a2015-06-24 05:59:19 +0000100 if (Count > 63)
101 return UINT64_MAX;
102 return (uint64_t(1) << Count) - 1;
Richard Sandiford82ec87d2013-07-16 11:02:24 +0000103}
104
Richard Sandiford51093212013-07-18 10:40:35 +0000105// Represents operands 2 to 5 of the ROTATE AND ... SELECTED BITS operation
106// given by Opcode. The operands are: Input (R2), Start (I3), End (I4) and
107// Rotate (I5). The combined operand value is effectively:
108//
109// (or (rotl Input, Rotate), ~Mask)
110//
111// for RNSBG and:
112//
113// (and (rotl Input, Rotate), Mask)
114//
Richard Sandiford3e382972013-10-16 13:35:13 +0000115// otherwise. The output value has BitSize bits, although Input may be
116// narrower (in which case the upper bits are don't care).
Richard Sandiford5cbac962013-07-18 09:45:08 +0000117struct RxSBGOperands {
Richard Sandiford51093212013-07-18 10:40:35 +0000118 RxSBGOperands(unsigned Op, SDValue N)
119 : Opcode(Op), BitSize(N.getValueType().getSizeInBits()),
120 Mask(allOnes(BitSize)), Input(N), Start(64 - BitSize), End(63),
121 Rotate(0) {}
Richard Sandiford82ec87d2013-07-16 11:02:24 +0000122
Richard Sandiford51093212013-07-18 10:40:35 +0000123 unsigned Opcode;
Richard Sandiford82ec87d2013-07-16 11:02:24 +0000124 unsigned BitSize;
125 uint64_t Mask;
126 SDValue Input;
127 unsigned Start;
128 unsigned End;
129 unsigned Rotate;
130};
131
Ulrich Weigand5f613df2013-05-06 16:15:19 +0000132class SystemZDAGToDAGISel : public SelectionDAGISel {
Eric Christophera6734172015-01-31 00:06:45 +0000133 const SystemZSubtarget *Subtarget;
Ulrich Weigand5f613df2013-05-06 16:15:19 +0000134
135 // Used by SystemZOperands.td to create integer constants.
Richard Sandiford54b36912013-09-27 15:14:04 +0000136 inline SDValue getImm(const SDNode *Node, uint64_t Imm) const {
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +0000137 return CurDAG->getTargetConstant(Imm, SDLoc(Node), Node->getValueType(0));
Ulrich Weigand5f613df2013-05-06 16:15:19 +0000138 }
139
Richard Sandiford6a06ba32013-07-31 11:36:35 +0000140 const SystemZTargetMachine &getTargetMachine() const {
141 return static_cast<const SystemZTargetMachine &>(TM);
142 }
143
144 const SystemZInstrInfo *getInstrInfo() const {
Eric Christophera6734172015-01-31 00:06:45 +0000145 return Subtarget->getInstrInfo();
Richard Sandiford6a06ba32013-07-31 11:36:35 +0000146 }
147
Ulrich Weigand5f613df2013-05-06 16:15:19 +0000148 // Try to fold more of the base or index of AM into AM, where IsBase
149 // selects between the base and index.
Richard Sandiford54b36912013-09-27 15:14:04 +0000150 bool expandAddress(SystemZAddressingMode &AM, bool IsBase) const;
Ulrich Weigand5f613df2013-05-06 16:15:19 +0000151
152 // Try to describe N in AM, returning true on success.
Richard Sandiford54b36912013-09-27 15:14:04 +0000153 bool selectAddress(SDValue N, SystemZAddressingMode &AM) const;
Ulrich Weigand5f613df2013-05-06 16:15:19 +0000154
155 // Extract individual target operands from matched address AM.
156 void getAddressOperands(const SystemZAddressingMode &AM, EVT VT,
Richard Sandiford54b36912013-09-27 15:14:04 +0000157 SDValue &Base, SDValue &Disp) const;
Ulrich Weigand5f613df2013-05-06 16:15:19 +0000158 void getAddressOperands(const SystemZAddressingMode &AM, EVT VT,
Richard Sandiford54b36912013-09-27 15:14:04 +0000159 SDValue &Base, SDValue &Disp, SDValue &Index) const;
Ulrich Weigand5f613df2013-05-06 16:15:19 +0000160
161 // Try to match Addr as a FormBD address with displacement type DR.
162 // Return true on success, storing the base and displacement in
163 // Base and Disp respectively.
164 bool selectBDAddr(SystemZAddressingMode::DispRange DR, SDValue Addr,
Richard Sandiford54b36912013-09-27 15:14:04 +0000165 SDValue &Base, SDValue &Disp) const;
Ulrich Weigand5f613df2013-05-06 16:15:19 +0000166
Richard Sandiforda481f582013-08-23 11:18:53 +0000167 // Try to match Addr as a FormBDX address with displacement type DR.
168 // Return true on success and if the result had no index. Store the
169 // base and displacement in Base and Disp respectively.
170 bool selectMVIAddr(SystemZAddressingMode::DispRange DR, SDValue Addr,
Richard Sandiford54b36912013-09-27 15:14:04 +0000171 SDValue &Base, SDValue &Disp) const;
Richard Sandiforda481f582013-08-23 11:18:53 +0000172
Ulrich Weigand5f613df2013-05-06 16:15:19 +0000173 // Try to match Addr as a FormBDX* address of form Form with
174 // displacement type DR. Return true on success, storing the base,
175 // displacement and index in Base, Disp and Index respectively.
176 bool selectBDXAddr(SystemZAddressingMode::AddrForm Form,
177 SystemZAddressingMode::DispRange DR, SDValue Addr,
Richard Sandiford54b36912013-09-27 15:14:04 +0000178 SDValue &Base, SDValue &Disp, SDValue &Index) const;
Ulrich Weigand5f613df2013-05-06 16:15:19 +0000179
180 // PC-relative address matching routines used by SystemZOperands.td.
Richard Sandiford54b36912013-09-27 15:14:04 +0000181 bool selectPCRelAddress(SDValue Addr, SDValue &Target) const {
182 if (SystemZISD::isPCREL(Addr.getOpcode())) {
Ulrich Weigand5f613df2013-05-06 16:15:19 +0000183 Target = Addr.getOperand(0);
184 return true;
185 }
186 return false;
187 }
188
189 // BD matching routines used by SystemZOperands.td.
Richard Sandiford54b36912013-09-27 15:14:04 +0000190 bool selectBDAddr12Only(SDValue Addr, SDValue &Base, SDValue &Disp) const {
Ulrich Weigand5f613df2013-05-06 16:15:19 +0000191 return selectBDAddr(SystemZAddressingMode::Disp12Only, Addr, Base, Disp);
192 }
Richard Sandiford54b36912013-09-27 15:14:04 +0000193 bool selectBDAddr12Pair(SDValue Addr, SDValue &Base, SDValue &Disp) const {
Ulrich Weigand5f613df2013-05-06 16:15:19 +0000194 return selectBDAddr(SystemZAddressingMode::Disp12Pair, Addr, Base, Disp);
195 }
Richard Sandiford54b36912013-09-27 15:14:04 +0000196 bool selectBDAddr20Only(SDValue Addr, SDValue &Base, SDValue &Disp) const {
Ulrich Weigand5f613df2013-05-06 16:15:19 +0000197 return selectBDAddr(SystemZAddressingMode::Disp20Only, Addr, Base, Disp);
198 }
Richard Sandiford54b36912013-09-27 15:14:04 +0000199 bool selectBDAddr20Pair(SDValue Addr, SDValue &Base, SDValue &Disp) const {
Ulrich Weigand5f613df2013-05-06 16:15:19 +0000200 return selectBDAddr(SystemZAddressingMode::Disp20Pair, Addr, Base, Disp);
201 }
202
Richard Sandiforda481f582013-08-23 11:18:53 +0000203 // MVI matching routines used by SystemZOperands.td.
Richard Sandiford54b36912013-09-27 15:14:04 +0000204 bool selectMVIAddr12Pair(SDValue Addr, SDValue &Base, SDValue &Disp) const {
Richard Sandiforda481f582013-08-23 11:18:53 +0000205 return selectMVIAddr(SystemZAddressingMode::Disp12Pair, Addr, Base, Disp);
206 }
Richard Sandiford54b36912013-09-27 15:14:04 +0000207 bool selectMVIAddr20Pair(SDValue Addr, SDValue &Base, SDValue &Disp) const {
Richard Sandiforda481f582013-08-23 11:18:53 +0000208 return selectMVIAddr(SystemZAddressingMode::Disp20Pair, Addr, Base, Disp);
209 }
210
Ulrich Weigand5f613df2013-05-06 16:15:19 +0000211 // BDX matching routines used by SystemZOperands.td.
212 bool selectBDXAddr12Only(SDValue Addr, SDValue &Base, SDValue &Disp,
Richard Sandiford54b36912013-09-27 15:14:04 +0000213 SDValue &Index) const {
Ulrich Weigand5f613df2013-05-06 16:15:19 +0000214 return selectBDXAddr(SystemZAddressingMode::FormBDXNormal,
215 SystemZAddressingMode::Disp12Only,
216 Addr, Base, Disp, Index);
217 }
218 bool selectBDXAddr12Pair(SDValue Addr, SDValue &Base, SDValue &Disp,
Richard Sandiford54b36912013-09-27 15:14:04 +0000219 SDValue &Index) const {
Ulrich Weigand5f613df2013-05-06 16:15:19 +0000220 return selectBDXAddr(SystemZAddressingMode::FormBDXNormal,
221 SystemZAddressingMode::Disp12Pair,
222 Addr, Base, Disp, Index);
223 }
224 bool selectDynAlloc12Only(SDValue Addr, SDValue &Base, SDValue &Disp,
Richard Sandiford54b36912013-09-27 15:14:04 +0000225 SDValue &Index) const {
Ulrich Weigand5f613df2013-05-06 16:15:19 +0000226 return selectBDXAddr(SystemZAddressingMode::FormBDXDynAlloc,
227 SystemZAddressingMode::Disp12Only,
228 Addr, Base, Disp, Index);
229 }
230 bool selectBDXAddr20Only(SDValue Addr, SDValue &Base, SDValue &Disp,
Richard Sandiford54b36912013-09-27 15:14:04 +0000231 SDValue &Index) const {
Ulrich Weigand5f613df2013-05-06 16:15:19 +0000232 return selectBDXAddr(SystemZAddressingMode::FormBDXNormal,
233 SystemZAddressingMode::Disp20Only,
234 Addr, Base, Disp, Index);
235 }
236 bool selectBDXAddr20Only128(SDValue Addr, SDValue &Base, SDValue &Disp,
Richard Sandiford54b36912013-09-27 15:14:04 +0000237 SDValue &Index) const {
Ulrich Weigand5f613df2013-05-06 16:15:19 +0000238 return selectBDXAddr(SystemZAddressingMode::FormBDXNormal,
239 SystemZAddressingMode::Disp20Only128,
240 Addr, Base, Disp, Index);
241 }
242 bool selectBDXAddr20Pair(SDValue Addr, SDValue &Base, SDValue &Disp,
Richard Sandiford54b36912013-09-27 15:14:04 +0000243 SDValue &Index) const {
Ulrich Weigand5f613df2013-05-06 16:15:19 +0000244 return selectBDXAddr(SystemZAddressingMode::FormBDXNormal,
245 SystemZAddressingMode::Disp20Pair,
246 Addr, Base, Disp, Index);
247 }
248 bool selectLAAddr12Pair(SDValue Addr, SDValue &Base, SDValue &Disp,
Richard Sandiford54b36912013-09-27 15:14:04 +0000249 SDValue &Index) const {
Ulrich Weigand5f613df2013-05-06 16:15:19 +0000250 return selectBDXAddr(SystemZAddressingMode::FormBDXLA,
251 SystemZAddressingMode::Disp12Pair,
252 Addr, Base, Disp, Index);
253 }
254 bool selectLAAddr20Pair(SDValue Addr, SDValue &Base, SDValue &Disp,
Richard Sandiford54b36912013-09-27 15:14:04 +0000255 SDValue &Index) const {
Ulrich Weigand5f613df2013-05-06 16:15:19 +0000256 return selectBDXAddr(SystemZAddressingMode::FormBDXLA,
257 SystemZAddressingMode::Disp20Pair,
258 Addr, Base, Disp, Index);
259 }
260
Ulrich Weigandce4c1092015-05-05 19:25:42 +0000261 // Try to match Addr as an address with a base, 12-bit displacement
262 // and index, where the index is element Elem of a vector.
263 // Return true on success, storing the base, displacement and vector
264 // in Base, Disp and Index respectively.
265 bool selectBDVAddr12Only(SDValue Addr, SDValue Elem, SDValue &Base,
266 SDValue &Disp, SDValue &Index) const;
267
Richard Sandiford885140c2013-07-16 11:55:57 +0000268 // Check whether (or Op (and X InsertMask)) is effectively an insertion
269 // of X into bits InsertMask of some Y != Op. Return true if so and
270 // set Op to that Y.
Richard Sandiford54b36912013-09-27 15:14:04 +0000271 bool detectOrAndInsertion(SDValue &Op, uint64_t InsertMask) const;
Richard Sandiford885140c2013-07-16 11:55:57 +0000272
Richard Sandiford6a06ba32013-07-31 11:36:35 +0000273 // Try to update RxSBG so that only the bits of RxSBG.Input in Mask are used.
274 // Return true on success.
Richard Sandiford54b36912013-09-27 15:14:04 +0000275 bool refineRxSBGMask(RxSBGOperands &RxSBG, uint64_t Mask) const;
Richard Sandiford6a06ba32013-07-31 11:36:35 +0000276
Richard Sandiford5cbac962013-07-18 09:45:08 +0000277 // Try to fold some of RxSBG.Input into other fields of RxSBG.
278 // Return true on success.
Richard Sandiford54b36912013-09-27 15:14:04 +0000279 bool expandRxSBG(RxSBGOperands &RxSBG) const;
Richard Sandiford82ec87d2013-07-16 11:02:24 +0000280
Richard Sandiford3ad5a152013-10-01 14:36:20 +0000281 // Return an undefined value of type VT.
282 SDValue getUNDEF(SDLoc DL, EVT VT) const;
Richard Sandiford84f54a32013-07-11 08:59:12 +0000283
284 // Convert N to VT, if it isn't already.
Richard Sandiford54b36912013-09-27 15:14:04 +0000285 SDValue convertTo(SDLoc DL, EVT VT, SDValue N) const;
Richard Sandiford84f54a32013-07-11 08:59:12 +0000286
Richard Sandiford82ec87d2013-07-16 11:02:24 +0000287 // Try to implement AND or shift node N using RISBG with the zero flag set.
288 // Return the selected node on success, otherwise return null.
289 SDNode *tryRISBGZero(SDNode *N);
Richard Sandiford84f54a32013-07-11 08:59:12 +0000290
Richard Sandiford7878b852013-07-18 10:06:15 +0000291 // Try to use RISBG or Opcode to implement OR or XOR node N.
292 // Return the selected node on success, otherwise return null.
293 SDNode *tryRxSBG(SDNode *N, unsigned Opcode);
Richard Sandiford885140c2013-07-16 11:55:57 +0000294
Ulrich Weigand5f613df2013-05-06 16:15:19 +0000295 // If Op0 is null, then Node is a constant that can be loaded using:
296 //
297 // (Opcode UpperVal LowerVal)
298 //
299 // If Op0 is nonnull, then Node can be implemented using:
300 //
301 // (Opcode (Opcode Op0 UpperVal) LowerVal)
302 SDNode *splitLargeImmediate(unsigned Opcode, SDNode *Node, SDValue Op0,
303 uint64_t UpperVal, uint64_t LowerVal);
304
Ulrich Weigandce4c1092015-05-05 19:25:42 +0000305 // Try to use gather instruction Opcode to implement vector insertion N.
306 SDNode *tryGather(SDNode *N, unsigned Opcode);
307
308 // Try to use scatter instruction Opcode to implement store Store.
309 SDNode *tryScatter(StoreSDNode *Store, unsigned Opcode);
310
Richard Sandiford067817e2013-09-27 15:29:20 +0000311 // Return true if Load and Store are loads and stores of the same size
312 // and are guaranteed not to overlap. Such operations can be implemented
313 // using block (SS-format) instructions.
314 //
315 // Partial overlap would lead to incorrect code, since the block operations
316 // are logically bytewise, even though they have a fast path for the
317 // non-overlapping case. We also need to avoid full overlap (i.e. two
318 // addresses that might be equal at run time) because although that case
319 // would be handled correctly, it might be implemented by millicode.
320 bool canUseBlockOperation(StoreSDNode *Store, LoadSDNode *Load) const;
321
Richard Sandiford178273a2013-09-05 10:36:45 +0000322 // N is a (store (load Y), X) pattern. Return true if it can use an MVC
323 // from Y to X.
Richard Sandiford97846492013-07-09 09:46:39 +0000324 bool storeLoadCanUseMVC(SDNode *N) const;
325
Richard Sandiford178273a2013-09-05 10:36:45 +0000326 // N is a (store (op (load A[0]), (load A[1])), X) pattern. Return true
327 // if A[1 - I] == X and if N can use a block operation like NC from A[I]
328 // to X.
329 bool storeLoadCanUseBlockBinary(SDNode *N, unsigned I) const;
330
Ulrich Weigand5f613df2013-05-06 16:15:19 +0000331public:
332 SystemZDAGToDAGISel(SystemZTargetMachine &TM, CodeGenOpt::Level OptLevel)
Eric Christophera6734172015-01-31 00:06:45 +0000333 : SelectionDAGISel(TM, OptLevel) {}
334
335 bool runOnMachineFunction(MachineFunction &MF) override {
336 Subtarget = &MF.getSubtarget<SystemZSubtarget>();
337 return SelectionDAGISel::runOnMachineFunction(MF);
338 }
Ulrich Weigand5f613df2013-05-06 16:15:19 +0000339
340 // Override MachineFunctionPass.
Richard Sandifordb4d67b52014-03-06 12:03:36 +0000341 const char *getPassName() const override {
Ulrich Weigand5f613df2013-05-06 16:15:19 +0000342 return "SystemZ DAG->DAG Pattern Instruction Selection";
343 }
344
345 // Override SelectionDAGISel.
Richard Sandifordb4d67b52014-03-06 12:03:36 +0000346 SDNode *Select(SDNode *Node) override;
Daniel Sanders60f1db02015-03-13 12:45:09 +0000347 bool SelectInlineAsmMemoryOperand(const SDValue &Op, unsigned ConstraintID,
Richard Sandifordb4d67b52014-03-06 12:03:36 +0000348 std::vector<SDValue> &OutOps) override;
Ulrich Weigand5f613df2013-05-06 16:15:19 +0000349
350 // Include the pieces autogenerated from the target description.
351 #include "SystemZGenDAGISel.inc"
352};
353} // end anonymous namespace
354
355FunctionPass *llvm::createSystemZISelDag(SystemZTargetMachine &TM,
356 CodeGenOpt::Level OptLevel) {
357 return new SystemZDAGToDAGISel(TM, OptLevel);
358}
359
360// Return true if Val should be selected as a displacement for an address
361// with range DR. Here we're interested in the range of both the instruction
362// described by DR and of any pairing instruction.
363static bool selectDisp(SystemZAddressingMode::DispRange DR, int64_t Val) {
364 switch (DR) {
365 case SystemZAddressingMode::Disp12Only:
366 return isUInt<12>(Val);
367
368 case SystemZAddressingMode::Disp12Pair:
369 case SystemZAddressingMode::Disp20Only:
370 case SystemZAddressingMode::Disp20Pair:
371 return isInt<20>(Val);
372
373 case SystemZAddressingMode::Disp20Only128:
374 return isInt<20>(Val) && isInt<20>(Val + 8);
375 }
376 llvm_unreachable("Unhandled displacement range");
377}
378
379// Change the base or index in AM to Value, where IsBase selects
380// between the base and index.
381static void changeComponent(SystemZAddressingMode &AM, bool IsBase,
382 SDValue Value) {
383 if (IsBase)
384 AM.Base = Value;
385 else
386 AM.Index = Value;
387}
388
389// The base or index of AM is equivalent to Value + ADJDYNALLOC,
390// where IsBase selects between the base and index. Try to fold the
391// ADJDYNALLOC into AM.
392static bool expandAdjDynAlloc(SystemZAddressingMode &AM, bool IsBase,
393 SDValue Value) {
394 if (AM.isDynAlloc() && !AM.IncludesDynAlloc) {
395 changeComponent(AM, IsBase, Value);
396 AM.IncludesDynAlloc = true;
397 return true;
398 }
399 return false;
400}
401
402// The base of AM is equivalent to Base + Index. Try to use Index as
403// the index register.
404static bool expandIndex(SystemZAddressingMode &AM, SDValue Base,
405 SDValue Index) {
406 if (AM.hasIndexField() && !AM.Index.getNode()) {
407 AM.Base = Base;
408 AM.Index = Index;
409 return true;
410 }
411 return false;
412}
413
414// The base or index of AM is equivalent to Op0 + Op1, where IsBase selects
415// between the base and index. Try to fold Op1 into AM's displacement.
416static bool expandDisp(SystemZAddressingMode &AM, bool IsBase,
Richard Sandiford54b36912013-09-27 15:14:04 +0000417 SDValue Op0, uint64_t Op1) {
Ulrich Weigand5f613df2013-05-06 16:15:19 +0000418 // First try adjusting the displacement.
Richard Sandiford54b36912013-09-27 15:14:04 +0000419 int64_t TestDisp = AM.Disp + Op1;
Ulrich Weigand5f613df2013-05-06 16:15:19 +0000420 if (selectDisp(AM.DR, TestDisp)) {
421 changeComponent(AM, IsBase, Op0);
422 AM.Disp = TestDisp;
423 return true;
424 }
425
426 // We could consider forcing the displacement into a register and
427 // using it as an index, but it would need to be carefully tuned.
428 return false;
429}
430
431bool SystemZDAGToDAGISel::expandAddress(SystemZAddressingMode &AM,
Richard Sandiford54b36912013-09-27 15:14:04 +0000432 bool IsBase) const {
Ulrich Weigand5f613df2013-05-06 16:15:19 +0000433 SDValue N = IsBase ? AM.Base : AM.Index;
434 unsigned Opcode = N.getOpcode();
435 if (Opcode == ISD::TRUNCATE) {
436 N = N.getOperand(0);
437 Opcode = N.getOpcode();
438 }
439 if (Opcode == ISD::ADD || CurDAG->isBaseWithConstantOffset(N)) {
440 SDValue Op0 = N.getOperand(0);
441 SDValue Op1 = N.getOperand(1);
442
443 unsigned Op0Code = Op0->getOpcode();
444 unsigned Op1Code = Op1->getOpcode();
445
446 if (Op0Code == SystemZISD::ADJDYNALLOC)
447 return expandAdjDynAlloc(AM, IsBase, Op1);
448 if (Op1Code == SystemZISD::ADJDYNALLOC)
449 return expandAdjDynAlloc(AM, IsBase, Op0);
450
451 if (Op0Code == ISD::Constant)
Richard Sandiford54b36912013-09-27 15:14:04 +0000452 return expandDisp(AM, IsBase, Op1,
453 cast<ConstantSDNode>(Op0)->getSExtValue());
Ulrich Weigand5f613df2013-05-06 16:15:19 +0000454 if (Op1Code == ISD::Constant)
Richard Sandiford54b36912013-09-27 15:14:04 +0000455 return expandDisp(AM, IsBase, Op0,
456 cast<ConstantSDNode>(Op1)->getSExtValue());
Ulrich Weigand5f613df2013-05-06 16:15:19 +0000457
458 if (IsBase && expandIndex(AM, Op0, Op1))
459 return true;
460 }
Richard Sandiford54b36912013-09-27 15:14:04 +0000461 if (Opcode == SystemZISD::PCREL_OFFSET) {
462 SDValue Full = N.getOperand(0);
463 SDValue Base = N.getOperand(1);
464 SDValue Anchor = Base.getOperand(0);
465 uint64_t Offset = (cast<GlobalAddressSDNode>(Full)->getOffset() -
466 cast<GlobalAddressSDNode>(Anchor)->getOffset());
467 return expandDisp(AM, IsBase, Base, Offset);
468 }
Ulrich Weigand5f613df2013-05-06 16:15:19 +0000469 return false;
470}
471
472// Return true if an instruction with displacement range DR should be
473// used for displacement value Val. selectDisp(DR, Val) must already hold.
474static bool isValidDisp(SystemZAddressingMode::DispRange DR, int64_t Val) {
475 assert(selectDisp(DR, Val) && "Invalid displacement");
476 switch (DR) {
477 case SystemZAddressingMode::Disp12Only:
478 case SystemZAddressingMode::Disp20Only:
479 case SystemZAddressingMode::Disp20Only128:
480 return true;
481
482 case SystemZAddressingMode::Disp12Pair:
483 // Use the other instruction if the displacement is too large.
484 return isUInt<12>(Val);
485
486 case SystemZAddressingMode::Disp20Pair:
487 // Use the other instruction if the displacement is small enough.
488 return !isUInt<12>(Val);
489 }
490 llvm_unreachable("Unhandled displacement range");
491}
492
493// Return true if Base + Disp + Index should be performed by LA(Y).
494static bool shouldUseLA(SDNode *Base, int64_t Disp, SDNode *Index) {
495 // Don't use LA(Y) for constants.
496 if (!Base)
497 return false;
498
499 // Always use LA(Y) for frame addresses, since we know that the destination
500 // register is almost always (perhaps always) going to be different from
501 // the frame register.
502 if (Base->getOpcode() == ISD::FrameIndex)
503 return true;
504
505 if (Disp) {
506 // Always use LA(Y) if there is a base, displacement and index.
507 if (Index)
508 return true;
509
510 // Always use LA if the displacement is small enough. It should always
511 // be no worse than AGHI (and better if it avoids a move).
512 if (isUInt<12>(Disp))
513 return true;
514
515 // For similar reasons, always use LAY if the constant is too big for AGHI.
516 // LAY should be no worse than AGFI.
517 if (!isInt<16>(Disp))
518 return true;
519 } else {
520 // Don't use LA for plain registers.
521 if (!Index)
522 return false;
523
524 // Don't use LA for plain addition if the index operand is only used
525 // once. It should be a natural two-operand addition in that case.
526 if (Index->hasOneUse())
527 return false;
528
529 // Prefer addition if the second operation is sign-extended, in the
530 // hope of using AGF.
531 unsigned IndexOpcode = Index->getOpcode();
532 if (IndexOpcode == ISD::SIGN_EXTEND ||
533 IndexOpcode == ISD::SIGN_EXTEND_INREG)
534 return false;
535 }
536
537 // Don't use LA for two-operand addition if either operand is only
538 // used once. The addition instructions are better in that case.
539 if (Base->hasOneUse())
540 return false;
541
542 return true;
543}
544
545// Return true if Addr is suitable for AM, updating AM if so.
546bool SystemZDAGToDAGISel::selectAddress(SDValue Addr,
Richard Sandiford54b36912013-09-27 15:14:04 +0000547 SystemZAddressingMode &AM) const {
Ulrich Weigand5f613df2013-05-06 16:15:19 +0000548 // Start out assuming that the address will need to be loaded separately,
549 // then try to extend it as much as we can.
550 AM.Base = Addr;
551
552 // First try treating the address as a constant.
553 if (Addr.getOpcode() == ISD::Constant &&
Richard Sandiford54b36912013-09-27 15:14:04 +0000554 expandDisp(AM, true, SDValue(),
555 cast<ConstantSDNode>(Addr)->getSExtValue()))
Ulrich Weigand5f613df2013-05-06 16:15:19 +0000556 ;
557 else
558 // Otherwise try expanding each component.
559 while (expandAddress(AM, true) ||
560 (AM.Index.getNode() && expandAddress(AM, false)))
561 continue;
562
563 // Reject cases where it isn't profitable to use LA(Y).
564 if (AM.Form == SystemZAddressingMode::FormBDXLA &&
565 !shouldUseLA(AM.Base.getNode(), AM.Disp, AM.Index.getNode()))
566 return false;
567
568 // Reject cases where the other instruction in a pair should be used.
569 if (!isValidDisp(AM.DR, AM.Disp))
570 return false;
571
572 // Make sure that ADJDYNALLOC is included where necessary.
573 if (AM.isDynAlloc() && !AM.IncludesDynAlloc)
574 return false;
575
576 DEBUG(AM.dump());
577 return true;
578}
579
580// Insert a node into the DAG at least before Pos. This will reposition
581// the node as needed, and will assign it a node ID that is <= Pos's ID.
582// Note that this does *not* preserve the uniqueness of node IDs!
583// The selection DAG must no longer depend on their uniqueness when this
584// function is used.
585static void insertDAGNode(SelectionDAG *DAG, SDNode *Pos, SDValue N) {
586 if (N.getNode()->getNodeId() == -1 ||
587 N.getNode()->getNodeId() > Pos->getNodeId()) {
588 DAG->RepositionNode(Pos, N.getNode());
589 N.getNode()->setNodeId(Pos->getNodeId());
590 }
591}
592
593void SystemZDAGToDAGISel::getAddressOperands(const SystemZAddressingMode &AM,
594 EVT VT, SDValue &Base,
Richard Sandiford54b36912013-09-27 15:14:04 +0000595 SDValue &Disp) const {
Ulrich Weigand5f613df2013-05-06 16:15:19 +0000596 Base = AM.Base;
597 if (!Base.getNode())
598 // Register 0 means "no base". This is mostly useful for shifts.
599 Base = CurDAG->getRegister(0, VT);
600 else if (Base.getOpcode() == ISD::FrameIndex) {
601 // Lower a FrameIndex to a TargetFrameIndex.
602 int64_t FrameIndex = cast<FrameIndexSDNode>(Base)->getIndex();
603 Base = CurDAG->getTargetFrameIndex(FrameIndex, VT);
604 } else if (Base.getValueType() != VT) {
605 // Truncate values from i64 to i32, for shifts.
606 assert(VT == MVT::i32 && Base.getValueType() == MVT::i64 &&
607 "Unexpected truncation");
Andrew Trickef9de2a2013-05-25 02:42:55 +0000608 SDLoc DL(Base);
Ulrich Weigand5f613df2013-05-06 16:15:19 +0000609 SDValue Trunc = CurDAG->getNode(ISD::TRUNCATE, DL, VT, Base);
610 insertDAGNode(CurDAG, Base.getNode(), Trunc);
611 Base = Trunc;
612 }
613
614 // Lower the displacement to a TargetConstant.
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +0000615 Disp = CurDAG->getTargetConstant(AM.Disp, SDLoc(Base), VT);
Ulrich Weigand5f613df2013-05-06 16:15:19 +0000616}
617
618void SystemZDAGToDAGISel::getAddressOperands(const SystemZAddressingMode &AM,
619 EVT VT, SDValue &Base,
Richard Sandiford54b36912013-09-27 15:14:04 +0000620 SDValue &Disp,
621 SDValue &Index) const {
Ulrich Weigand5f613df2013-05-06 16:15:19 +0000622 getAddressOperands(AM, VT, Base, Disp);
623
624 Index = AM.Index;
625 if (!Index.getNode())
626 // Register 0 means "no index".
627 Index = CurDAG->getRegister(0, VT);
628}
629
630bool SystemZDAGToDAGISel::selectBDAddr(SystemZAddressingMode::DispRange DR,
631 SDValue Addr, SDValue &Base,
Richard Sandiford54b36912013-09-27 15:14:04 +0000632 SDValue &Disp) const {
Ulrich Weigand5f613df2013-05-06 16:15:19 +0000633 SystemZAddressingMode AM(SystemZAddressingMode::FormBD, DR);
634 if (!selectAddress(Addr, AM))
635 return false;
636
637 getAddressOperands(AM, Addr.getValueType(), Base, Disp);
638 return true;
639}
640
Richard Sandiforda481f582013-08-23 11:18:53 +0000641bool SystemZDAGToDAGISel::selectMVIAddr(SystemZAddressingMode::DispRange DR,
642 SDValue Addr, SDValue &Base,
Richard Sandiford54b36912013-09-27 15:14:04 +0000643 SDValue &Disp) const {
Richard Sandiforda481f582013-08-23 11:18:53 +0000644 SystemZAddressingMode AM(SystemZAddressingMode::FormBDXNormal, DR);
645 if (!selectAddress(Addr, AM) || AM.Index.getNode())
646 return false;
647
648 getAddressOperands(AM, Addr.getValueType(), Base, Disp);
649 return true;
650}
651
Ulrich Weigand5f613df2013-05-06 16:15:19 +0000652bool SystemZDAGToDAGISel::selectBDXAddr(SystemZAddressingMode::AddrForm Form,
653 SystemZAddressingMode::DispRange DR,
654 SDValue Addr, SDValue &Base,
Richard Sandiford54b36912013-09-27 15:14:04 +0000655 SDValue &Disp, SDValue &Index) const {
Ulrich Weigand5f613df2013-05-06 16:15:19 +0000656 SystemZAddressingMode AM(Form, DR);
657 if (!selectAddress(Addr, AM))
658 return false;
659
660 getAddressOperands(AM, Addr.getValueType(), Base, Disp, Index);
661 return true;
662}
663
Ulrich Weigandce4c1092015-05-05 19:25:42 +0000664bool SystemZDAGToDAGISel::selectBDVAddr12Only(SDValue Addr, SDValue Elem,
665 SDValue &Base,
666 SDValue &Disp,
667 SDValue &Index) const {
668 SDValue Regs[2];
669 if (selectBDXAddr12Only(Addr, Regs[0], Disp, Regs[1]) &&
670 Regs[0].getNode() && Regs[1].getNode()) {
671 for (unsigned int I = 0; I < 2; ++I) {
672 Base = Regs[I];
673 Index = Regs[1 - I];
674 // We can't tell here whether the index vector has the right type
675 // for the access; the caller needs to do that instead.
676 if (Index.getOpcode() == ISD::ZERO_EXTEND)
677 Index = Index.getOperand(0);
678 if (Index.getOpcode() == ISD::EXTRACT_VECTOR_ELT &&
679 Index.getOperand(1) == Elem) {
680 Index = Index.getOperand(0);
681 return true;
682 }
683 }
684 }
685 return false;
686}
687
Richard Sandiford885140c2013-07-16 11:55:57 +0000688bool SystemZDAGToDAGISel::detectOrAndInsertion(SDValue &Op,
Richard Sandiford54b36912013-09-27 15:14:04 +0000689 uint64_t InsertMask) const {
Richard Sandiford885140c2013-07-16 11:55:57 +0000690 // We're only interested in cases where the insertion is into some operand
691 // of Op, rather than into Op itself. The only useful case is an AND.
692 if (Op.getOpcode() != ISD::AND)
693 return false;
694
695 // We need a constant mask.
Richard Sandiford21f5d682014-03-06 11:22:58 +0000696 auto *MaskNode = dyn_cast<ConstantSDNode>(Op.getOperand(1).getNode());
Richard Sandiford885140c2013-07-16 11:55:57 +0000697 if (!MaskNode)
698 return false;
699
700 // It's not an insertion of Op.getOperand(0) if the two masks overlap.
701 uint64_t AndMask = MaskNode->getZExtValue();
702 if (InsertMask & AndMask)
703 return false;
704
705 // It's only an insertion if all bits are covered or are known to be zero.
706 // The inner check covers all cases but is more expensive.
707 uint64_t Used = allOnes(Op.getValueType().getSizeInBits());
708 if (Used != (AndMask | InsertMask)) {
709 APInt KnownZero, KnownOne;
Jay Foada0653a32014-05-14 21:14:37 +0000710 CurDAG->computeKnownBits(Op.getOperand(0), KnownZero, KnownOne);
Richard Sandiford885140c2013-07-16 11:55:57 +0000711 if (Used != (AndMask | InsertMask | KnownZero.getZExtValue()))
712 return false;
713 }
714
715 Op = Op.getOperand(0);
716 return true;
717}
718
Richard Sandiford54b36912013-09-27 15:14:04 +0000719bool SystemZDAGToDAGISel::refineRxSBGMask(RxSBGOperands &RxSBG,
720 uint64_t Mask) const {
Richard Sandiford6a06ba32013-07-31 11:36:35 +0000721 const SystemZInstrInfo *TII = getInstrInfo();
Richard Sandiford5cbac962013-07-18 09:45:08 +0000722 if (RxSBG.Rotate != 0)
723 Mask = (Mask << RxSBG.Rotate) | (Mask >> (64 - RxSBG.Rotate));
724 Mask &= RxSBG.Mask;
Richard Sandiford6a06ba32013-07-31 11:36:35 +0000725 if (TII->isRxSBGMask(Mask, RxSBG.BitSize, RxSBG.Start, RxSBG.End)) {
Richard Sandiford5cbac962013-07-18 09:45:08 +0000726 RxSBG.Mask = Mask;
Richard Sandiford5cbac962013-07-18 09:45:08 +0000727 return true;
728 }
Richard Sandiford84f54a32013-07-11 08:59:12 +0000729 return false;
730}
731
Richard Sandiforddd7dd932013-11-26 10:53:16 +0000732// Return true if any bits of (RxSBG.Input & Mask) are significant.
733static bool maskMatters(RxSBGOperands &RxSBG, uint64_t Mask) {
734 // Rotate the mask in the same way as RxSBG.Input is rotated.
Richard Sandiford297f7d22013-07-18 10:14:55 +0000735 if (RxSBG.Rotate != 0)
Richard Sandiforddd7dd932013-11-26 10:53:16 +0000736 Mask = ((Mask << RxSBG.Rotate) | (Mask >> (64 - RxSBG.Rotate)));
737 return (Mask & RxSBG.Mask) != 0;
Richard Sandiford297f7d22013-07-18 10:14:55 +0000738}
739
Richard Sandiford54b36912013-09-27 15:14:04 +0000740bool SystemZDAGToDAGISel::expandRxSBG(RxSBGOperands &RxSBG) const {
Richard Sandiford5cbac962013-07-18 09:45:08 +0000741 SDValue N = RxSBG.Input;
Richard Sandiford297f7d22013-07-18 10:14:55 +0000742 unsigned Opcode = N.getOpcode();
743 switch (Opcode) {
Richard Sandiford82ec87d2013-07-16 11:02:24 +0000744 case ISD::AND: {
Richard Sandiford51093212013-07-18 10:40:35 +0000745 if (RxSBG.Opcode == SystemZ::RNSBG)
746 return false;
747
Richard Sandiford21f5d682014-03-06 11:22:58 +0000748 auto *MaskNode = dyn_cast<ConstantSDNode>(N.getOperand(1).getNode());
Richard Sandiford82ec87d2013-07-16 11:02:24 +0000749 if (!MaskNode)
750 return false;
751
752 SDValue Input = N.getOperand(0);
753 uint64_t Mask = MaskNode->getZExtValue();
Richard Sandiford5cbac962013-07-18 09:45:08 +0000754 if (!refineRxSBGMask(RxSBG, Mask)) {
Richard Sandiford82ec87d2013-07-16 11:02:24 +0000755 // If some bits of Input are already known zeros, those bits will have
756 // been removed from the mask. See if adding them back in makes the
757 // mask suitable.
758 APInt KnownZero, KnownOne;
Jay Foada0653a32014-05-14 21:14:37 +0000759 CurDAG->computeKnownBits(Input, KnownZero, KnownOne);
Richard Sandiford82ec87d2013-07-16 11:02:24 +0000760 Mask |= KnownZero.getZExtValue();
Richard Sandiford5cbac962013-07-18 09:45:08 +0000761 if (!refineRxSBGMask(RxSBG, Mask))
Richard Sandiford82ec87d2013-07-16 11:02:24 +0000762 return false;
763 }
Richard Sandiford5cbac962013-07-18 09:45:08 +0000764 RxSBG.Input = Input;
Richard Sandiford82ec87d2013-07-16 11:02:24 +0000765 return true;
766 }
767
Richard Sandiford51093212013-07-18 10:40:35 +0000768 case ISD::OR: {
769 if (RxSBG.Opcode != SystemZ::RNSBG)
770 return false;
771
Richard Sandiford21f5d682014-03-06 11:22:58 +0000772 auto *MaskNode = dyn_cast<ConstantSDNode>(N.getOperand(1).getNode());
Richard Sandiford51093212013-07-18 10:40:35 +0000773 if (!MaskNode)
774 return false;
775
776 SDValue Input = N.getOperand(0);
777 uint64_t Mask = ~MaskNode->getZExtValue();
778 if (!refineRxSBGMask(RxSBG, Mask)) {
779 // If some bits of Input are already known ones, those bits will have
780 // been removed from the mask. See if adding them back in makes the
781 // mask suitable.
782 APInt KnownZero, KnownOne;
Jay Foada0653a32014-05-14 21:14:37 +0000783 CurDAG->computeKnownBits(Input, KnownZero, KnownOne);
Richard Sandiford51093212013-07-18 10:40:35 +0000784 Mask &= ~KnownOne.getZExtValue();
785 if (!refineRxSBGMask(RxSBG, Mask))
786 return false;
787 }
788 RxSBG.Input = Input;
789 return true;
790 }
791
Richard Sandiford82ec87d2013-07-16 11:02:24 +0000792 case ISD::ROTL: {
Richard Sandiford5cbac962013-07-18 09:45:08 +0000793 // Any 64-bit rotate left can be merged into the RxSBG.
Richard Sandiford3e382972013-10-16 13:35:13 +0000794 if (RxSBG.BitSize != 64 || N.getValueType() != MVT::i64)
Richard Sandiford82ec87d2013-07-16 11:02:24 +0000795 return false;
Richard Sandiford21f5d682014-03-06 11:22:58 +0000796 auto *CountNode = dyn_cast<ConstantSDNode>(N.getOperand(1).getNode());
Richard Sandiford82ec87d2013-07-16 11:02:24 +0000797 if (!CountNode)
798 return false;
799
Richard Sandiford5cbac962013-07-18 09:45:08 +0000800 RxSBG.Rotate = (RxSBG.Rotate + CountNode->getZExtValue()) & 63;
801 RxSBG.Input = N.getOperand(0);
Richard Sandiford82ec87d2013-07-16 11:02:24 +0000802 return true;
803 }
804
Richard Sandiford220ee492013-12-20 11:49:48 +0000805 case ISD::ANY_EXTEND:
806 // Bits above the extended operand are don't-care.
807 RxSBG.Input = N.getOperand(0);
808 return true;
809
Richard Sandiford3875cb62014-01-09 11:28:53 +0000810 case ISD::ZERO_EXTEND:
811 if (RxSBG.Opcode != SystemZ::RNSBG) {
812 // Restrict the mask to the extended operand.
813 unsigned InnerBitSize = N.getOperand(0).getValueType().getSizeInBits();
814 if (!refineRxSBGMask(RxSBG, allOnes(InnerBitSize)))
815 return false;
Richard Sandiford220ee492013-12-20 11:49:48 +0000816
Richard Sandiford3875cb62014-01-09 11:28:53 +0000817 RxSBG.Input = N.getOperand(0);
818 return true;
819 }
820 // Fall through.
Richard Sandiford220ee492013-12-20 11:49:48 +0000821
822 case ISD::SIGN_EXTEND: {
Richard Sandiford3e382972013-10-16 13:35:13 +0000823 // Check that the extension bits are don't-care (i.e. are masked out
824 // by the final mask).
825 unsigned InnerBitSize = N.getOperand(0).getValueType().getSizeInBits();
Richard Sandiforddd7dd932013-11-26 10:53:16 +0000826 if (maskMatters(RxSBG, allOnes(RxSBG.BitSize) - allOnes(InnerBitSize)))
Richard Sandiford3e382972013-10-16 13:35:13 +0000827 return false;
828
829 RxSBG.Input = N.getOperand(0);
830 return true;
831 }
832
Richard Sandiford82ec87d2013-07-16 11:02:24 +0000833 case ISD::SHL: {
Richard Sandiford21f5d682014-03-06 11:22:58 +0000834 auto *CountNode = dyn_cast<ConstantSDNode>(N.getOperand(1).getNode());
Richard Sandiford82ec87d2013-07-16 11:02:24 +0000835 if (!CountNode)
836 return false;
837
838 uint64_t Count = CountNode->getZExtValue();
Richard Sandiford3e382972013-10-16 13:35:13 +0000839 unsigned BitSize = N.getValueType().getSizeInBits();
840 if (Count < 1 || Count >= BitSize)
Richard Sandiford82ec87d2013-07-16 11:02:24 +0000841 return false;
842
Richard Sandiford51093212013-07-18 10:40:35 +0000843 if (RxSBG.Opcode == SystemZ::RNSBG) {
844 // Treat (shl X, count) as (rotl X, size-count) as long as the bottom
845 // count bits from RxSBG.Input are ignored.
Richard Sandiforddd7dd932013-11-26 10:53:16 +0000846 if (maskMatters(RxSBG, allOnes(Count)))
Richard Sandiford51093212013-07-18 10:40:35 +0000847 return false;
848 } else {
849 // Treat (shl X, count) as (and (rotl X, count), ~0<<count).
Richard Sandiford3e382972013-10-16 13:35:13 +0000850 if (!refineRxSBGMask(RxSBG, allOnes(BitSize - Count) << Count))
Richard Sandiford51093212013-07-18 10:40:35 +0000851 return false;
852 }
853
Richard Sandiford5cbac962013-07-18 09:45:08 +0000854 RxSBG.Rotate = (RxSBG.Rotate + Count) & 63;
855 RxSBG.Input = N.getOperand(0);
Richard Sandiford82ec87d2013-07-16 11:02:24 +0000856 return true;
857 }
858
Richard Sandiford297f7d22013-07-18 10:14:55 +0000859 case ISD::SRL:
860 case ISD::SRA: {
Richard Sandiford21f5d682014-03-06 11:22:58 +0000861 auto *CountNode = dyn_cast<ConstantSDNode>(N.getOperand(1).getNode());
Richard Sandiford82ec87d2013-07-16 11:02:24 +0000862 if (!CountNode)
863 return false;
864
865 uint64_t Count = CountNode->getZExtValue();
Richard Sandiford3e382972013-10-16 13:35:13 +0000866 unsigned BitSize = N.getValueType().getSizeInBits();
867 if (Count < 1 || Count >= BitSize)
Richard Sandiford82ec87d2013-07-16 11:02:24 +0000868 return false;
869
Richard Sandiford51093212013-07-18 10:40:35 +0000870 if (RxSBG.Opcode == SystemZ::RNSBG || Opcode == ISD::SRA) {
871 // Treat (srl|sra X, count) as (rotl X, size-count) as long as the top
872 // count bits from RxSBG.Input are ignored.
Richard Sandiforddd7dd932013-11-26 10:53:16 +0000873 if (maskMatters(RxSBG, allOnes(Count) << (BitSize - Count)))
Richard Sandiford297f7d22013-07-18 10:14:55 +0000874 return false;
875 } else {
876 // Treat (srl X, count), mask) as (and (rotl X, size-count), ~0>>count),
877 // which is similar to SLL above.
Richard Sandiford3e382972013-10-16 13:35:13 +0000878 if (!refineRxSBGMask(RxSBG, allOnes(BitSize - Count)))
Richard Sandiford297f7d22013-07-18 10:14:55 +0000879 return false;
880 }
881
Richard Sandiford5cbac962013-07-18 09:45:08 +0000882 RxSBG.Rotate = (RxSBG.Rotate - Count) & 63;
883 RxSBG.Input = N.getOperand(0);
Richard Sandiford82ec87d2013-07-16 11:02:24 +0000884 return true;
885 }
Richard Sandiford82ec87d2013-07-16 11:02:24 +0000886 default:
887 return false;
888 }
889}
890
Richard Sandiford3ad5a152013-10-01 14:36:20 +0000891SDValue SystemZDAGToDAGISel::getUNDEF(SDLoc DL, EVT VT) const {
892 SDNode *N = CurDAG->getMachineNode(TargetOpcode::IMPLICIT_DEF, DL, VT);
Richard Sandiford84f54a32013-07-11 08:59:12 +0000893 return SDValue(N, 0);
894}
895
Richard Sandiford54b36912013-09-27 15:14:04 +0000896SDValue SystemZDAGToDAGISel::convertTo(SDLoc DL, EVT VT, SDValue N) const {
Richard Sandifordd8163202013-09-13 09:12:44 +0000897 if (N.getValueType() == MVT::i32 && VT == MVT::i64)
Richard Sandiford87a44362013-09-30 10:28:35 +0000898 return CurDAG->getTargetInsertSubreg(SystemZ::subreg_l32,
Richard Sandiford3ad5a152013-10-01 14:36:20 +0000899 DL, VT, getUNDEF(DL, MVT::i64), N);
Richard Sandifordd8163202013-09-13 09:12:44 +0000900 if (N.getValueType() == MVT::i64 && VT == MVT::i32)
Richard Sandiford87a44362013-09-30 10:28:35 +0000901 return CurDAG->getTargetExtractSubreg(SystemZ::subreg_l32, DL, VT, N);
Richard Sandiford84f54a32013-07-11 08:59:12 +0000902 assert(N.getValueType() == VT && "Unexpected value types");
903 return N;
904}
905
Richard Sandiford82ec87d2013-07-16 11:02:24 +0000906SDNode *SystemZDAGToDAGISel::tryRISBGZero(SDNode *N) {
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +0000907 SDLoc DL(N);
Richard Sandiford6a06ba32013-07-31 11:36:35 +0000908 EVT VT = N->getValueType(0);
Ulrich Weigand77884bc2015-06-25 11:52:36 +0000909 if (!VT.isInteger() || VT.getSizeInBits() > 64)
910 return nullptr;
Richard Sandiford51093212013-07-18 10:40:35 +0000911 RxSBGOperands RISBG(SystemZ::RISBG, SDValue(N, 0));
Richard Sandiford82ec87d2013-07-16 11:02:24 +0000912 unsigned Count = 0;
Richard Sandiford5cbac962013-07-18 09:45:08 +0000913 while (expandRxSBG(RISBG))
Richard Sandiford3e382972013-10-16 13:35:13 +0000914 if (RISBG.Input.getOpcode() != ISD::ANY_EXTEND)
915 Count += 1;
Richard Sandiford6a06ba32013-07-31 11:36:35 +0000916 if (Count == 0)
Craig Topper062a2ba2014-04-25 05:30:21 +0000917 return nullptr;
Richard Sandiford6a06ba32013-07-31 11:36:35 +0000918 if (Count == 1) {
919 // Prefer to use normal shift instructions over RISBG, since they can handle
920 // all cases and are sometimes shorter.
921 if (N->getOpcode() != ISD::AND)
Craig Topper062a2ba2014-04-25 05:30:21 +0000922 return nullptr;
Richard Sandiford82ec87d2013-07-16 11:02:24 +0000923
Richard Sandiford6a06ba32013-07-31 11:36:35 +0000924 // Prefer register extensions like LLC over RISBG. Also prefer to start
925 // out with normal ANDs if one instruction would be enough. We can convert
926 // these ANDs into an RISBG later if a three-address instruction is useful.
927 if (VT == MVT::i32 ||
928 RISBG.Mask == 0xff ||
929 RISBG.Mask == 0xffff ||
930 SystemZ::isImmLF(~RISBG.Mask) ||
931 SystemZ::isImmHF(~RISBG.Mask)) {
932 // Force the new mask into the DAG, since it may include known-one bits.
Richard Sandiford21f5d682014-03-06 11:22:58 +0000933 auto *MaskN = cast<ConstantSDNode>(N->getOperand(1).getNode());
Richard Sandiford6a06ba32013-07-31 11:36:35 +0000934 if (MaskN->getZExtValue() != RISBG.Mask) {
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +0000935 SDValue NewMask = CurDAG->getConstant(RISBG.Mask, DL, VT);
Richard Sandiford6a06ba32013-07-31 11:36:35 +0000936 N = CurDAG->UpdateNodeOperands(N, N->getOperand(0), NewMask);
937 return SelectCode(N);
938 }
Craig Topper062a2ba2014-04-25 05:30:21 +0000939 return nullptr;
Richard Sandiford6a06ba32013-07-31 11:36:35 +0000940 }
941 }
Richard Sandiford82ec87d2013-07-16 11:02:24 +0000942
Richard Sandiford3ad5a152013-10-01 14:36:20 +0000943 unsigned Opcode = SystemZ::RISBG;
Ulrich Weigand371d10a2015-03-31 12:58:17 +0000944 // Prefer RISBGN if available, since it does not clobber CC.
945 if (Subtarget->hasMiscellaneousExtensions())
946 Opcode = SystemZ::RISBGN;
Richard Sandiford3ad5a152013-10-01 14:36:20 +0000947 EVT OpcodeVT = MVT::i64;
Eric Christophera6734172015-01-31 00:06:45 +0000948 if (VT == MVT::i32 && Subtarget->hasHighWord()) {
Richard Sandiford3ad5a152013-10-01 14:36:20 +0000949 Opcode = SystemZ::RISBMux;
950 OpcodeVT = MVT::i32;
951 RISBG.Start &= 31;
952 RISBG.End &= 31;
953 }
Richard Sandiford84f54a32013-07-11 08:59:12 +0000954 SDValue Ops[5] = {
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +0000955 getUNDEF(DL, OpcodeVT),
956 convertTo(DL, OpcodeVT, RISBG.Input),
957 CurDAG->getTargetConstant(RISBG.Start, DL, MVT::i32),
958 CurDAG->getTargetConstant(RISBG.End | 128, DL, MVT::i32),
959 CurDAG->getTargetConstant(RISBG.Rotate, DL, MVT::i32)
Richard Sandiford84f54a32013-07-11 08:59:12 +0000960 };
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +0000961 N = CurDAG->getMachineNode(Opcode, DL, OpcodeVT, Ops);
962 return convertTo(DL, VT, SDValue(N, 0)).getNode();
Richard Sandiford84f54a32013-07-11 08:59:12 +0000963}
964
Richard Sandiford7878b852013-07-18 10:06:15 +0000965SDNode *SystemZDAGToDAGISel::tryRxSBG(SDNode *N, unsigned Opcode) {
Ulrich Weigand77884bc2015-06-25 11:52:36 +0000966 SDLoc DL(N);
967 EVT VT = N->getValueType(0);
968 if (!VT.isInteger() || VT.getSizeInBits() > 64)
969 return nullptr;
Richard Sandiford7878b852013-07-18 10:06:15 +0000970 // Try treating each operand of N as the second operand of the RxSBG
Richard Sandiford885140c2013-07-16 11:55:57 +0000971 // and see which goes deepest.
Richard Sandiford51093212013-07-18 10:40:35 +0000972 RxSBGOperands RxSBG[] = {
973 RxSBGOperands(Opcode, N->getOperand(0)),
974 RxSBGOperands(Opcode, N->getOperand(1))
975 };
Richard Sandiford885140c2013-07-16 11:55:57 +0000976 unsigned Count[] = { 0, 0 };
977 for (unsigned I = 0; I < 2; ++I)
Richard Sandiford5cbac962013-07-18 09:45:08 +0000978 while (expandRxSBG(RxSBG[I]))
Richard Sandiford3e382972013-10-16 13:35:13 +0000979 if (RxSBG[I].Input.getOpcode() != ISD::ANY_EXTEND)
980 Count[I] += 1;
Richard Sandiford885140c2013-07-16 11:55:57 +0000981
982 // Do nothing if neither operand is suitable.
983 if (Count[0] == 0 && Count[1] == 0)
Craig Topper062a2ba2014-04-25 05:30:21 +0000984 return nullptr;
Richard Sandiford885140c2013-07-16 11:55:57 +0000985
986 // Pick the deepest second operand.
987 unsigned I = Count[0] > Count[1] ? 0 : 1;
988 SDValue Op0 = N->getOperand(I ^ 1);
989
990 // Prefer IC for character insertions from memory.
Richard Sandiford7878b852013-07-18 10:06:15 +0000991 if (Opcode == SystemZ::ROSBG && (RxSBG[I].Mask & 0xff) == 0)
Richard Sandiford21f5d682014-03-06 11:22:58 +0000992 if (auto *Load = dyn_cast<LoadSDNode>(Op0.getNode()))
Richard Sandiford885140c2013-07-16 11:55:57 +0000993 if (Load->getMemoryVT() == MVT::i8)
Craig Topper062a2ba2014-04-25 05:30:21 +0000994 return nullptr;
Richard Sandiford885140c2013-07-16 11:55:57 +0000995
996 // See whether we can avoid an AND in the first operand by converting
997 // ROSBG to RISBG.
Ulrich Weigand371d10a2015-03-31 12:58:17 +0000998 if (Opcode == SystemZ::ROSBG && detectOrAndInsertion(Op0, RxSBG[I].Mask)) {
Richard Sandiford885140c2013-07-16 11:55:57 +0000999 Opcode = SystemZ::RISBG;
Ulrich Weigand371d10a2015-03-31 12:58:17 +00001000 // Prefer RISBGN if available, since it does not clobber CC.
1001 if (Subtarget->hasMiscellaneousExtensions())
1002 Opcode = SystemZ::RISBGN;
1003 }
1004
Richard Sandiford885140c2013-07-16 11:55:57 +00001005 SDValue Ops[5] = {
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +00001006 convertTo(DL, MVT::i64, Op0),
1007 convertTo(DL, MVT::i64, RxSBG[I].Input),
1008 CurDAG->getTargetConstant(RxSBG[I].Start, DL, MVT::i32),
1009 CurDAG->getTargetConstant(RxSBG[I].End, DL, MVT::i32),
1010 CurDAG->getTargetConstant(RxSBG[I].Rotate, DL, MVT::i32)
Richard Sandiford885140c2013-07-16 11:55:57 +00001011 };
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +00001012 N = CurDAG->getMachineNode(Opcode, DL, MVT::i64, Ops);
1013 return convertTo(DL, VT, SDValue(N, 0)).getNode();
Richard Sandiford885140c2013-07-16 11:55:57 +00001014}
1015
Ulrich Weigand5f613df2013-05-06 16:15:19 +00001016SDNode *SystemZDAGToDAGISel::splitLargeImmediate(unsigned Opcode, SDNode *Node,
1017 SDValue Op0, uint64_t UpperVal,
1018 uint64_t LowerVal) {
1019 EVT VT = Node->getValueType(0);
Andrew Trickef9de2a2013-05-25 02:42:55 +00001020 SDLoc DL(Node);
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +00001021 SDValue Upper = CurDAG->getConstant(UpperVal, DL, VT);
Ulrich Weigand5f613df2013-05-06 16:15:19 +00001022 if (Op0.getNode())
1023 Upper = CurDAG->getNode(Opcode, DL, VT, Op0, Upper);
1024 Upper = SDValue(Select(Upper.getNode()), 0);
1025
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +00001026 SDValue Lower = CurDAG->getConstant(LowerVal, DL, VT);
Ulrich Weigand5f613df2013-05-06 16:15:19 +00001027 SDValue Or = CurDAG->getNode(Opcode, DL, VT, Upper, Lower);
1028 return Or.getNode();
1029}
1030
Ulrich Weigandce4c1092015-05-05 19:25:42 +00001031SDNode *SystemZDAGToDAGISel::tryGather(SDNode *N, unsigned Opcode) {
1032 SDValue ElemV = N->getOperand(2);
1033 auto *ElemN = dyn_cast<ConstantSDNode>(ElemV);
1034 if (!ElemN)
1035 return 0;
1036
1037 unsigned Elem = ElemN->getZExtValue();
1038 EVT VT = N->getValueType(0);
1039 if (Elem >= VT.getVectorNumElements())
1040 return 0;
1041
1042 auto *Load = dyn_cast<LoadSDNode>(N->getOperand(1));
1043 if (!Load || !Load->hasOneUse())
1044 return 0;
1045 if (Load->getMemoryVT().getSizeInBits() !=
1046 Load->getValueType(0).getSizeInBits())
1047 return 0;
1048
1049 SDValue Base, Disp, Index;
1050 if (!selectBDVAddr12Only(Load->getBasePtr(), ElemV, Base, Disp, Index) ||
1051 Index.getValueType() != VT.changeVectorElementTypeToInteger())
1052 return 0;
1053
1054 SDLoc DL(Load);
1055 SDValue Ops[] = {
1056 N->getOperand(0), Base, Disp, Index,
1057 CurDAG->getTargetConstant(Elem, DL, MVT::i32), Load->getChain()
1058 };
1059 SDNode *Res = CurDAG->getMachineNode(Opcode, DL, VT, MVT::Other, Ops);
1060 ReplaceUses(SDValue(Load, 1), SDValue(Res, 1));
1061 return Res;
1062}
1063
1064SDNode *SystemZDAGToDAGISel::tryScatter(StoreSDNode *Store, unsigned Opcode) {
1065 SDValue Value = Store->getValue();
1066 if (Value.getOpcode() != ISD::EXTRACT_VECTOR_ELT)
1067 return 0;
1068 if (Store->getMemoryVT().getSizeInBits() !=
1069 Value.getValueType().getSizeInBits())
1070 return 0;
1071
1072 SDValue ElemV = Value.getOperand(1);
1073 auto *ElemN = dyn_cast<ConstantSDNode>(ElemV);
1074 if (!ElemN)
1075 return 0;
1076
1077 SDValue Vec = Value.getOperand(0);
1078 EVT VT = Vec.getValueType();
1079 unsigned Elem = ElemN->getZExtValue();
1080 if (Elem >= VT.getVectorNumElements())
1081 return 0;
1082
1083 SDValue Base, Disp, Index;
1084 if (!selectBDVAddr12Only(Store->getBasePtr(), ElemV, Base, Disp, Index) ||
1085 Index.getValueType() != VT.changeVectorElementTypeToInteger())
1086 return 0;
1087
1088 SDLoc DL(Store);
1089 SDValue Ops[] = {
1090 Vec, Base, Disp, Index, CurDAG->getTargetConstant(Elem, DL, MVT::i32),
1091 Store->getChain()
1092 };
1093 return CurDAG->getMachineNode(Opcode, DL, MVT::Other, Ops);
1094}
1095
Richard Sandiford067817e2013-09-27 15:29:20 +00001096bool SystemZDAGToDAGISel::canUseBlockOperation(StoreSDNode *Store,
1097 LoadSDNode *Load) const {
Richard Sandiford178273a2013-09-05 10:36:45 +00001098 // Check that the two memory operands have the same size.
1099 if (Load->getMemoryVT() != Store->getMemoryVT())
Richard Sandiford97846492013-07-09 09:46:39 +00001100 return false;
1101
Richard Sandiford178273a2013-09-05 10:36:45 +00001102 // Volatility stops an access from being decomposed.
1103 if (Load->isVolatile() || Store->isVolatile())
1104 return false;
Richard Sandiford97846492013-07-09 09:46:39 +00001105
1106 // There's no chance of overlap if the load is invariant.
1107 if (Load->isInvariant())
1108 return true;
1109
Richard Sandiford97846492013-07-09 09:46:39 +00001110 // Otherwise we need to check whether there's an alias.
Nick Lewyckyaad475b2014-04-15 07:22:52 +00001111 const Value *V1 = Load->getMemOperand()->getValue();
1112 const Value *V2 = Store->getMemOperand()->getValue();
Richard Sandiford97846492013-07-09 09:46:39 +00001113 if (!V1 || !V2)
1114 return false;
1115
Richard Sandiford067817e2013-09-27 15:29:20 +00001116 // Reject equality.
1117 uint64_t Size = Load->getMemoryVT().getStoreSize();
Richard Sandiford97846492013-07-09 09:46:39 +00001118 int64_t End1 = Load->getSrcValueOffset() + Size;
1119 int64_t End2 = Store->getSrcValueOffset() + Size;
Richard Sandiford067817e2013-09-27 15:29:20 +00001120 if (V1 == V2 && End1 == End2)
1121 return false;
1122
Chandler Carruthac80dc72015-06-17 07:18:54 +00001123 return !AA->alias(MemoryLocation(V1, End1, Load->getAAInfo()),
1124 MemoryLocation(V2, End2, Store->getAAInfo()));
Richard Sandiford97846492013-07-09 09:46:39 +00001125}
1126
Richard Sandiford178273a2013-09-05 10:36:45 +00001127bool SystemZDAGToDAGISel::storeLoadCanUseMVC(SDNode *N) const {
Richard Sandiford21f5d682014-03-06 11:22:58 +00001128 auto *Store = cast<StoreSDNode>(N);
1129 auto *Load = cast<LoadSDNode>(Store->getValue());
Richard Sandiford178273a2013-09-05 10:36:45 +00001130
1131 // Prefer not to use MVC if either address can use ... RELATIVE LONG
1132 // instructions.
1133 uint64_t Size = Load->getMemoryVT().getStoreSize();
1134 if (Size > 1 && Size <= 8) {
1135 // Prefer LHRL, LRL and LGRL.
Richard Sandiford54b36912013-09-27 15:14:04 +00001136 if (SystemZISD::isPCREL(Load->getBasePtr().getOpcode()))
Richard Sandiford178273a2013-09-05 10:36:45 +00001137 return false;
1138 // Prefer STHRL, STRL and STGRL.
Richard Sandiford54b36912013-09-27 15:14:04 +00001139 if (SystemZISD::isPCREL(Store->getBasePtr().getOpcode()))
Richard Sandiford178273a2013-09-05 10:36:45 +00001140 return false;
1141 }
1142
Richard Sandiford067817e2013-09-27 15:29:20 +00001143 return canUseBlockOperation(Store, Load);
Richard Sandiford178273a2013-09-05 10:36:45 +00001144}
1145
1146bool SystemZDAGToDAGISel::storeLoadCanUseBlockBinary(SDNode *N,
1147 unsigned I) const {
Richard Sandiford21f5d682014-03-06 11:22:58 +00001148 auto *StoreA = cast<StoreSDNode>(N);
1149 auto *LoadA = cast<LoadSDNode>(StoreA->getValue().getOperand(1 - I));
1150 auto *LoadB = cast<LoadSDNode>(StoreA->getValue().getOperand(I));
Richard Sandiford067817e2013-09-27 15:29:20 +00001151 return !LoadA->isVolatile() && canUseBlockOperation(StoreA, LoadB);
Richard Sandiford178273a2013-09-05 10:36:45 +00001152}
1153
Ulrich Weigand5f613df2013-05-06 16:15:19 +00001154SDNode *SystemZDAGToDAGISel::Select(SDNode *Node) {
1155 // Dump information about the Node being selected
1156 DEBUG(errs() << "Selecting: "; Node->dump(CurDAG); errs() << "\n");
1157
1158 // If we have a custom node, we already have selected!
1159 if (Node->isMachineOpcode()) {
1160 DEBUG(errs() << "== "; Node->dump(CurDAG); errs() << "\n");
Tim Northover31d093c2013-09-22 08:21:56 +00001161 Node->setNodeId(-1);
Craig Topper062a2ba2014-04-25 05:30:21 +00001162 return nullptr;
Ulrich Weigand5f613df2013-05-06 16:15:19 +00001163 }
1164
1165 unsigned Opcode = Node->getOpcode();
Craig Topper062a2ba2014-04-25 05:30:21 +00001166 SDNode *ResNode = nullptr;
Ulrich Weigand5f613df2013-05-06 16:15:19 +00001167 switch (Opcode) {
1168 case ISD::OR:
Richard Sandiford885140c2013-07-16 11:55:57 +00001169 if (Node->getOperand(1).getOpcode() != ISD::Constant)
Richard Sandiford7878b852013-07-18 10:06:15 +00001170 ResNode = tryRxSBG(Node, SystemZ::ROSBG);
1171 goto or_xor;
1172
Ulrich Weigand5f613df2013-05-06 16:15:19 +00001173 case ISD::XOR:
Richard Sandiford7878b852013-07-18 10:06:15 +00001174 if (Node->getOperand(1).getOpcode() != ISD::Constant)
1175 ResNode = tryRxSBG(Node, SystemZ::RXSBG);
1176 // Fall through.
1177 or_xor:
Ulrich Weigand5f613df2013-05-06 16:15:19 +00001178 // If this is a 64-bit operation in which both 32-bit halves are nonzero,
1179 // split the operation into two.
Richard Sandiford885140c2013-07-16 11:55:57 +00001180 if (!ResNode && Node->getValueType(0) == MVT::i64)
Richard Sandiford21f5d682014-03-06 11:22:58 +00001181 if (auto *Op1 = dyn_cast<ConstantSDNode>(Node->getOperand(1))) {
Ulrich Weigand5f613df2013-05-06 16:15:19 +00001182 uint64_t Val = Op1->getZExtValue();
1183 if (!SystemZ::isImmLF(Val) && !SystemZ::isImmHF(Val))
1184 Node = splitLargeImmediate(Opcode, Node, Node->getOperand(0),
1185 Val - uint32_t(Val), uint32_t(Val));
1186 }
1187 break;
1188
Richard Sandiford84f54a32013-07-11 08:59:12 +00001189 case ISD::AND:
Richard Sandiford51093212013-07-18 10:40:35 +00001190 if (Node->getOperand(1).getOpcode() != ISD::Constant)
1191 ResNode = tryRxSBG(Node, SystemZ::RNSBG);
1192 // Fall through.
Richard Sandiford82ec87d2013-07-16 11:02:24 +00001193 case ISD::ROTL:
1194 case ISD::SHL:
1195 case ISD::SRL:
Richard Sandiford220ee492013-12-20 11:49:48 +00001196 case ISD::ZERO_EXTEND:
Richard Sandiford7878b852013-07-18 10:06:15 +00001197 if (!ResNode)
1198 ResNode = tryRISBGZero(Node);
Richard Sandiford84f54a32013-07-11 08:59:12 +00001199 break;
1200
Ulrich Weigand5f613df2013-05-06 16:15:19 +00001201 case ISD::Constant:
1202 // If this is a 64-bit constant that is out of the range of LLILF,
1203 // LLIHF and LGFI, split it into two 32-bit pieces.
1204 if (Node->getValueType(0) == MVT::i64) {
1205 uint64_t Val = cast<ConstantSDNode>(Node)->getZExtValue();
1206 if (!SystemZ::isImmLF(Val) && !SystemZ::isImmHF(Val) && !isInt<32>(Val))
1207 Node = splitLargeImmediate(ISD::OR, Node, SDValue(),
1208 Val - uint32_t(Val), uint32_t(Val));
1209 }
1210 break;
1211
Richard Sandifordee834382013-07-31 12:38:08 +00001212 case SystemZISD::SELECT_CCMASK: {
1213 SDValue Op0 = Node->getOperand(0);
1214 SDValue Op1 = Node->getOperand(1);
1215 // Prefer to put any load first, so that it can be matched as a
1216 // conditional load.
1217 if (Op1.getOpcode() == ISD::LOAD && Op0.getOpcode() != ISD::LOAD) {
1218 SDValue CCValid = Node->getOperand(2);
1219 SDValue CCMask = Node->getOperand(3);
1220 uint64_t ConstCCValid =
1221 cast<ConstantSDNode>(CCValid.getNode())->getZExtValue();
1222 uint64_t ConstCCMask =
1223 cast<ConstantSDNode>(CCMask.getNode())->getZExtValue();
1224 // Invert the condition.
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +00001225 CCMask = CurDAG->getConstant(ConstCCValid ^ ConstCCMask, SDLoc(Node),
Richard Sandifordee834382013-07-31 12:38:08 +00001226 CCMask.getValueType());
1227 SDValue Op4 = Node->getOperand(4);
1228 Node = CurDAG->UpdateNodeOperands(Node, Op1, Op0, CCValid, CCMask, Op4);
1229 }
1230 break;
1231 }
Ulrich Weigandce4c1092015-05-05 19:25:42 +00001232
1233 case ISD::INSERT_VECTOR_ELT: {
1234 EVT VT = Node->getValueType(0);
1235 unsigned ElemBitSize = VT.getVectorElementType().getSizeInBits();
1236 if (ElemBitSize == 32)
1237 ResNode = tryGather(Node, SystemZ::VGEF);
1238 else if (ElemBitSize == 64)
1239 ResNode = tryGather(Node, SystemZ::VGEG);
1240 break;
1241 }
1242
1243 case ISD::STORE: {
1244 auto *Store = cast<StoreSDNode>(Node);
1245 unsigned ElemBitSize = Store->getValue().getValueType().getSizeInBits();
1246 if (ElemBitSize == 32)
1247 ResNode = tryScatter(Store, SystemZ::VSCEF);
1248 else if (ElemBitSize == 64)
1249 ResNode = tryScatter(Store, SystemZ::VSCEG);
1250 break;
1251 }
Ulrich Weigand5f613df2013-05-06 16:15:19 +00001252 }
1253
1254 // Select the default instruction
Richard Sandiford84f54a32013-07-11 08:59:12 +00001255 if (!ResNode)
1256 ResNode = SelectCode(Node);
Ulrich Weigand5f613df2013-05-06 16:15:19 +00001257
1258 DEBUG(errs() << "=> ";
Craig Topper062a2ba2014-04-25 05:30:21 +00001259 if (ResNode == nullptr || ResNode == Node)
Ulrich Weigand5f613df2013-05-06 16:15:19 +00001260 Node->dump(CurDAG);
1261 else
1262 ResNode->dump(CurDAG);
1263 errs() << "\n";
1264 );
1265 return ResNode;
1266}
1267
1268bool SystemZDAGToDAGISel::
1269SelectInlineAsmMemoryOperand(const SDValue &Op,
Daniel Sanders60f1db02015-03-13 12:45:09 +00001270 unsigned ConstraintID,
Ulrich Weigand5f613df2013-05-06 16:15:19 +00001271 std::vector<SDValue> &OutOps) {
Daniel Sanders2eeace22015-03-17 16:16:14 +00001272 switch(ConstraintID) {
1273 default:
1274 llvm_unreachable("Unexpected asm memory constraint");
1275 case InlineAsm::Constraint_i:
1276 case InlineAsm::Constraint_m:
1277 case InlineAsm::Constraint_Q:
1278 case InlineAsm::Constraint_R:
1279 case InlineAsm::Constraint_S:
1280 case InlineAsm::Constraint_T:
1281 // Accept addresses with short displacements, which are compatible
1282 // with Q, R, S and T. But keep the index operand for future expansion.
1283 SDValue Base, Disp, Index;
1284 if (selectBDXAddr(SystemZAddressingMode::FormBD,
1285 SystemZAddressingMode::Disp12Only,
1286 Op, Base, Disp, Index)) {
1287 OutOps.push_back(Base);
1288 OutOps.push_back(Disp);
1289 OutOps.push_back(Index);
1290 return false;
1291 }
1292 break;
1293 }
1294 return true;
Ulrich Weigand5f613df2013-05-06 16:15:19 +00001295}