blob: 3847187b2eff9cbeadf997072dfebf094c4e971e [file] [log] [blame]
Ulrich Weigand5f613df2013-05-06 16:15:19 +00001//===-- SystemZISelLowering.cpp - SystemZ DAG lowering implementation -----===//
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 implements the SystemZTargetLowering class.
11//
12//===----------------------------------------------------------------------===//
13
Ulrich Weigand5f613df2013-05-06 16:15:19 +000014#include "SystemZISelLowering.h"
15#include "SystemZCallingConv.h"
16#include "SystemZConstantPoolValue.h"
17#include "SystemZMachineFunctionInfo.h"
18#include "SystemZTargetMachine.h"
19#include "llvm/CodeGen/CallingConvLower.h"
20#include "llvm/CodeGen/MachineInstrBuilder.h"
21#include "llvm/CodeGen/MachineRegisterInfo.h"
22#include "llvm/CodeGen/TargetLoweringObjectFileImpl.h"
Will Dietz981af002013-10-12 00:55:57 +000023#include <cctype>
24
Ulrich Weigand5f613df2013-05-06 16:15:19 +000025using namespace llvm;
26
Chandler Carruth84e68b22014-04-22 02:41:26 +000027#define DEBUG_TYPE "systemz-lower"
28
Richard Sandifordf722a8e302013-10-16 11:10:55 +000029namespace {
30// Represents a sequence for extracting a 0/1 value from an IPM result:
31// (((X ^ XORValue) + AddValue) >> Bit)
32struct IPMConversion {
33 IPMConversion(unsigned xorValue, int64_t addValue, unsigned bit)
34 : XORValue(xorValue), AddValue(addValue), Bit(bit) {}
35
36 int64_t XORValue;
37 int64_t AddValue;
38 unsigned Bit;
39};
Richard Sandifordd420f732013-12-13 15:28:45 +000040
41// Represents information about a comparison.
42struct Comparison {
43 Comparison(SDValue Op0In, SDValue Op1In)
44 : Op0(Op0In), Op1(Op1In), Opcode(0), ICmpType(0), CCValid(0), CCMask(0) {}
45
46 // The operands to the comparison.
47 SDValue Op0, Op1;
48
49 // The opcode that should be used to compare Op0 and Op1.
50 unsigned Opcode;
51
52 // A SystemZICMP value. Only used for integer comparisons.
53 unsigned ICmpType;
54
55 // The mask of CC values that Opcode can produce.
56 unsigned CCValid;
57
58 // The mask of CC values for which the original condition is true.
59 unsigned CCMask;
60};
Richard Sandifordc2312692014-03-06 10:38:30 +000061} // end anonymous namespace
Richard Sandifordf722a8e302013-10-16 11:10:55 +000062
Ulrich Weigand5f613df2013-05-06 16:15:19 +000063// Classify VT as either 32 or 64 bit.
64static bool is32Bit(EVT VT) {
65 switch (VT.getSimpleVT().SimpleTy) {
66 case MVT::i32:
67 return true;
68 case MVT::i64:
69 return false;
70 default:
71 llvm_unreachable("Unsupported type");
72 }
73}
74
75// Return a version of MachineOperand that can be safely used before the
76// final use.
77static MachineOperand earlyUseOperand(MachineOperand Op) {
78 if (Op.isReg())
79 Op.setIsKill(false);
80 return Op;
81}
82
Eric Christophera6734172015-01-31 00:06:45 +000083SystemZTargetLowering::SystemZTargetLowering(const TargetMachine &tm,
84 const SystemZSubtarget &STI)
85 : TargetLowering(tm), Subtarget(STI) {
Ulrich Weigand5f613df2013-05-06 16:15:19 +000086 MVT PtrVT = getPointerTy();
87
88 // Set up the register classes.
Richard Sandiford0755c932013-10-01 11:26:28 +000089 if (Subtarget.hasHighWord())
90 addRegisterClass(MVT::i32, &SystemZ::GRX32BitRegClass);
91 else
92 addRegisterClass(MVT::i32, &SystemZ::GR32BitRegClass);
Ulrich Weigand5f613df2013-05-06 16:15:19 +000093 addRegisterClass(MVT::i64, &SystemZ::GR64BitRegClass);
94 addRegisterClass(MVT::f32, &SystemZ::FP32BitRegClass);
95 addRegisterClass(MVT::f64, &SystemZ::FP64BitRegClass);
96 addRegisterClass(MVT::f128, &SystemZ::FP128BitRegClass);
97
98 // Compute derived properties from the register classes
99 computeRegisterProperties();
100
101 // Set up special registers.
102 setExceptionPointerRegister(SystemZ::R6D);
103 setExceptionSelectorRegister(SystemZ::R7D);
104 setStackPointerRegisterToSaveRestore(SystemZ::R15D);
105
106 // TODO: It may be better to default to latency-oriented scheduling, however
107 // LLVM's current latency-oriented scheduler can't handle physreg definitions
Richard Sandiford14a44492013-05-22 13:38:45 +0000108 // such as SystemZ has with CC, so set this to the register-pressure
Ulrich Weigand5f613df2013-05-06 16:15:19 +0000109 // scheduler, because it can.
110 setSchedulingPreference(Sched::RegPressure);
111
112 setBooleanContents(ZeroOrOneBooleanContent);
113 setBooleanVectorContents(ZeroOrOneBooleanContent); // FIXME: Is this correct?
114
115 // Instructions are strings of 2-byte aligned 2-byte values.
116 setMinFunctionAlignment(2);
117
118 // Handle operations that are handled in a similar way for all types.
119 for (unsigned I = MVT::FIRST_INTEGER_VALUETYPE;
120 I <= MVT::LAST_FP_VALUETYPE;
121 ++I) {
122 MVT VT = MVT::SimpleValueType(I);
123 if (isTypeLegal(VT)) {
Richard Sandifordf722a8e302013-10-16 11:10:55 +0000124 // Lower SET_CC into an IPM-based sequence.
125 setOperationAction(ISD::SETCC, VT, Custom);
Ulrich Weigand5f613df2013-05-06 16:15:19 +0000126
127 // Expand SELECT(C, A, B) into SELECT_CC(X, 0, A, B, NE).
128 setOperationAction(ISD::SELECT, VT, Expand);
129
130 // Lower SELECT_CC and BR_CC into separate comparisons and branches.
131 setOperationAction(ISD::SELECT_CC, VT, Custom);
132 setOperationAction(ISD::BR_CC, VT, Custom);
133 }
134 }
135
136 // Expand jump table branches as address arithmetic followed by an
137 // indirect jump.
138 setOperationAction(ISD::BR_JT, MVT::Other, Expand);
139
140 // Expand BRCOND into a BR_CC (see above).
141 setOperationAction(ISD::BRCOND, MVT::Other, Expand);
142
143 // Handle integer types.
144 for (unsigned I = MVT::FIRST_INTEGER_VALUETYPE;
145 I <= MVT::LAST_INTEGER_VALUETYPE;
146 ++I) {
147 MVT VT = MVT::SimpleValueType(I);
148 if (isTypeLegal(VT)) {
149 // Expand individual DIV and REMs into DIVREMs.
150 setOperationAction(ISD::SDIV, VT, Expand);
151 setOperationAction(ISD::UDIV, VT, Expand);
152 setOperationAction(ISD::SREM, VT, Expand);
153 setOperationAction(ISD::UREM, VT, Expand);
154 setOperationAction(ISD::SDIVREM, VT, Custom);
155 setOperationAction(ISD::UDIVREM, VT, Custom);
156
Richard Sandifordbef3d7a2013-12-10 10:49:34 +0000157 // Lower ATOMIC_LOAD and ATOMIC_STORE into normal volatile loads and
158 // stores, putting a serialization instruction after the stores.
159 setOperationAction(ISD::ATOMIC_LOAD, VT, Custom);
160 setOperationAction(ISD::ATOMIC_STORE, VT, Custom);
Ulrich Weigand5f613df2013-05-06 16:15:19 +0000161
Richard Sandiford41350a52013-12-24 15:18:04 +0000162 // Lower ATOMIC_LOAD_SUB into ATOMIC_LOAD_ADD if LAA and LAAG are
163 // available, or if the operand is constant.
164 setOperationAction(ISD::ATOMIC_LOAD_SUB, VT, Custom);
165
Ulrich Weigand5f613df2013-05-06 16:15:19 +0000166 // No special instructions for these.
167 setOperationAction(ISD::CTPOP, VT, Expand);
168 setOperationAction(ISD::CTTZ, VT, Expand);
169 setOperationAction(ISD::CTTZ_ZERO_UNDEF, VT, Expand);
170 setOperationAction(ISD::CTLZ_ZERO_UNDEF, VT, Expand);
171 setOperationAction(ISD::ROTR, VT, Expand);
172
Richard Sandiford7d86e472013-08-21 09:34:56 +0000173 // Use *MUL_LOHI where possible instead of MULH*.
Ulrich Weigand5f613df2013-05-06 16:15:19 +0000174 setOperationAction(ISD::MULHS, VT, Expand);
175 setOperationAction(ISD::MULHU, VT, Expand);
Richard Sandiford7d86e472013-08-21 09:34:56 +0000176 setOperationAction(ISD::SMUL_LOHI, VT, Custom);
177 setOperationAction(ISD::UMUL_LOHI, VT, Custom);
Ulrich Weigand5f613df2013-05-06 16:15:19 +0000178
Richard Sandiforddc6c2c92014-03-21 10:56:30 +0000179 // Only z196 and above have native support for conversions to unsigned.
180 if (!Subtarget.hasFPExtension())
181 setOperationAction(ISD::FP_TO_UINT, VT, Expand);
Ulrich Weigand5f613df2013-05-06 16:15:19 +0000182 }
183 }
184
185 // Type legalization will convert 8- and 16-bit atomic operations into
186 // forms that operate on i32s (but still keeping the original memory VT).
187 // Lower them into full i32 operations.
188 setOperationAction(ISD::ATOMIC_SWAP, MVT::i32, Custom);
189 setOperationAction(ISD::ATOMIC_LOAD_ADD, MVT::i32, Custom);
190 setOperationAction(ISD::ATOMIC_LOAD_SUB, MVT::i32, Custom);
191 setOperationAction(ISD::ATOMIC_LOAD_AND, MVT::i32, Custom);
192 setOperationAction(ISD::ATOMIC_LOAD_OR, MVT::i32, Custom);
193 setOperationAction(ISD::ATOMIC_LOAD_XOR, MVT::i32, Custom);
194 setOperationAction(ISD::ATOMIC_LOAD_NAND, MVT::i32, Custom);
195 setOperationAction(ISD::ATOMIC_LOAD_MIN, MVT::i32, Custom);
196 setOperationAction(ISD::ATOMIC_LOAD_MAX, MVT::i32, Custom);
197 setOperationAction(ISD::ATOMIC_LOAD_UMIN, MVT::i32, Custom);
198 setOperationAction(ISD::ATOMIC_LOAD_UMAX, MVT::i32, Custom);
199 setOperationAction(ISD::ATOMIC_CMP_SWAP, MVT::i32, Custom);
200
Richard Sandiforddc6c2c92014-03-21 10:56:30 +0000201 // z10 has instructions for signed but not unsigned FP conversion.
Ulrich Weigand5f613df2013-05-06 16:15:19 +0000202 // Handle unsigned 32-bit types as signed 64-bit types.
Richard Sandiforddc6c2c92014-03-21 10:56:30 +0000203 if (!Subtarget.hasFPExtension()) {
204 setOperationAction(ISD::UINT_TO_FP, MVT::i32, Promote);
205 setOperationAction(ISD::UINT_TO_FP, MVT::i64, Expand);
206 }
Ulrich Weigand5f613df2013-05-06 16:15:19 +0000207
208 // We have native support for a 64-bit CTLZ, via FLOGR.
209 setOperationAction(ISD::CTLZ, MVT::i32, Promote);
210 setOperationAction(ISD::CTLZ, MVT::i64, Legal);
211
212 // Give LowerOperation the chance to replace 64-bit ORs with subregs.
213 setOperationAction(ISD::OR, MVT::i64, Custom);
214
Ulrich Weigand5f613df2013-05-06 16:15:19 +0000215 // FIXME: Can we support these natively?
216 setOperationAction(ISD::SRL_PARTS, MVT::i64, Expand);
217 setOperationAction(ISD::SHL_PARTS, MVT::i64, Expand);
218 setOperationAction(ISD::SRA_PARTS, MVT::i64, Expand);
219
220 // We have native instructions for i8, i16 and i32 extensions, but not i1.
Ulrich Weigand5f613df2013-05-06 16:15:19 +0000221 setOperationAction(ISD::SIGN_EXTEND_INREG, MVT::i1, Expand);
Ahmed Bougacha2b6917b2015-01-08 00:51:32 +0000222 for (MVT VT : MVT::integer_valuetypes()) {
223 setLoadExtAction(ISD::SEXTLOAD, VT, MVT::i1, Promote);
224 setLoadExtAction(ISD::ZEXTLOAD, VT, MVT::i1, Promote);
225 setLoadExtAction(ISD::EXTLOAD, VT, MVT::i1, Promote);
226 }
Ulrich Weigand5f613df2013-05-06 16:15:19 +0000227
228 // Handle the various types of symbolic address.
229 setOperationAction(ISD::ConstantPool, PtrVT, Custom);
230 setOperationAction(ISD::GlobalAddress, PtrVT, Custom);
231 setOperationAction(ISD::GlobalTLSAddress, PtrVT, Custom);
232 setOperationAction(ISD::BlockAddress, PtrVT, Custom);
233 setOperationAction(ISD::JumpTable, PtrVT, Custom);
234
235 // We need to handle dynamic allocations specially because of the
236 // 160-byte area at the bottom of the stack.
237 setOperationAction(ISD::DYNAMIC_STACKALLOC, PtrVT, Custom);
238
239 // Use custom expanders so that we can force the function to use
240 // a frame pointer.
241 setOperationAction(ISD::STACKSAVE, MVT::Other, Custom);
242 setOperationAction(ISD::STACKRESTORE, MVT::Other, Custom);
243
Richard Sandiford03481332013-08-23 11:36:42 +0000244 // Handle prefetches with PFD or PFDRL.
245 setOperationAction(ISD::PREFETCH, MVT::Other, Custom);
246
Ulrich Weigand5f613df2013-05-06 16:15:19 +0000247 // Handle floating-point types.
248 for (unsigned I = MVT::FIRST_FP_VALUETYPE;
249 I <= MVT::LAST_FP_VALUETYPE;
250 ++I) {
251 MVT VT = MVT::SimpleValueType(I);
252 if (isTypeLegal(VT)) {
253 // We can use FI for FRINT.
254 setOperationAction(ISD::FRINT, VT, Legal);
255
Richard Sandifordaf5f66a2013-08-21 09:04:20 +0000256 // We can use the extended form of FI for other rounding operations.
257 if (Subtarget.hasFPExtension()) {
258 setOperationAction(ISD::FNEARBYINT, VT, Legal);
259 setOperationAction(ISD::FFLOOR, VT, Legal);
260 setOperationAction(ISD::FCEIL, VT, Legal);
261 setOperationAction(ISD::FTRUNC, VT, Legal);
262 setOperationAction(ISD::FROUND, VT, Legal);
263 }
264
Ulrich Weigand5f613df2013-05-06 16:15:19 +0000265 // No special instructions for these.
266 setOperationAction(ISD::FSIN, VT, Expand);
267 setOperationAction(ISD::FCOS, VT, Expand);
268 setOperationAction(ISD::FREM, VT, Expand);
269 }
270 }
271
272 // We have fused multiply-addition for f32 and f64 but not f128.
273 setOperationAction(ISD::FMA, MVT::f32, Legal);
274 setOperationAction(ISD::FMA, MVT::f64, Legal);
275 setOperationAction(ISD::FMA, MVT::f128, Expand);
276
277 // Needed so that we don't try to implement f128 constant loads using
278 // a load-and-extend of a f80 constant (in cases where the constant
279 // would fit in an f80).
Ahmed Bougacha2b6917b2015-01-08 00:51:32 +0000280 for (MVT VT : MVT::fp_valuetypes())
281 setLoadExtAction(ISD::EXTLOAD, VT, MVT::f80, Expand);
Ulrich Weigand5f613df2013-05-06 16:15:19 +0000282
283 // Floating-point truncation and stores need to be done separately.
284 setTruncStoreAction(MVT::f64, MVT::f32, Expand);
285 setTruncStoreAction(MVT::f128, MVT::f32, Expand);
286 setTruncStoreAction(MVT::f128, MVT::f64, Expand);
287
288 // We have 64-bit FPR<->GPR moves, but need special handling for
289 // 32-bit forms.
290 setOperationAction(ISD::BITCAST, MVT::i32, Custom);
291 setOperationAction(ISD::BITCAST, MVT::f32, Custom);
292
293 // VASTART and VACOPY need to deal with the SystemZ-specific varargs
294 // structure, but VAEND is a no-op.
295 setOperationAction(ISD::VASTART, MVT::Other, Custom);
296 setOperationAction(ISD::VACOPY, MVT::Other, Custom);
297 setOperationAction(ISD::VAEND, MVT::Other, Expand);
Richard Sandifordd131ff82013-07-08 09:35:23 +0000298
Richard Sandiford95bc5f92014-03-07 11:34:35 +0000299 // Codes for which we want to perform some z-specific combinations.
300 setTargetDAGCombine(ISD::SIGN_EXTEND);
301
Richard Sandifordd131ff82013-07-08 09:35:23 +0000302 // We want to use MVC in preference to even a single load/store pair.
303 MaxStoresPerMemcpy = 0;
304 MaxStoresPerMemcpyOptSize = 0;
Richard Sandiford47660c12013-07-09 09:32:42 +0000305
306 // The main memset sequence is a byte store followed by an MVC.
307 // Two STC or MV..I stores win over that, but the kind of fused stores
308 // generated by target-independent code don't when the byte value is
309 // variable. E.g. "STC <reg>;MHI <reg>,257;STH <reg>" is not better
310 // than "STC;MVC". Handle the choice in target-specific code instead.
311 MaxStoresPerMemset = 0;
312 MaxStoresPerMemsetOptSize = 0;
Ulrich Weigand5f613df2013-05-06 16:15:19 +0000313}
314
Richard Sandifordabc010b2013-11-06 12:16:02 +0000315EVT SystemZTargetLowering::getSetCCResultType(LLVMContext &, EVT VT) const {
316 if (!VT.isVector())
317 return MVT::i32;
318 return VT.changeVectorElementTypeToInteger();
319}
320
321bool SystemZTargetLowering::isFMAFasterThanFMulAndFAdd(EVT VT) const {
Stephen Lin73de7bf2013-07-09 18:16:56 +0000322 VT = VT.getScalarType();
323
324 if (!VT.isSimple())
325 return false;
326
327 switch (VT.getSimpleVT().SimpleTy) {
328 case MVT::f32:
329 case MVT::f64:
330 return true;
331 case MVT::f128:
332 return false;
333 default:
334 break;
335 }
336
337 return false;
338}
339
Ulrich Weigand5f613df2013-05-06 16:15:19 +0000340bool SystemZTargetLowering::isFPImmLegal(const APFloat &Imm, EVT VT) const {
341 // We can load zero using LZ?R and negative zero using LZ?R;LC?BR.
342 return Imm.isZero() || Imm.isNegZero();
343}
344
Matt Arsenault6f2a5262014-07-27 17:46:40 +0000345bool SystemZTargetLowering::allowsMisalignedMemoryAccesses(EVT VT,
346 unsigned,
347 unsigned,
348 bool *Fast) const {
Richard Sandiford46af5a22013-05-30 09:45:42 +0000349 // Unaligned accesses should never be slower than the expanded version.
350 // We check specifically for aligned accesses in the few cases where
351 // they are required.
352 if (Fast)
353 *Fast = true;
354 return true;
355}
356
Richard Sandiford791bea42013-07-31 12:58:26 +0000357bool SystemZTargetLowering::isLegalAddressingMode(const AddrMode &AM,
358 Type *Ty) const {
359 // Punt on globals for now, although they can be used in limited
360 // RELATIVE LONG cases.
361 if (AM.BaseGV)
362 return false;
363
364 // Require a 20-bit signed offset.
365 if (!isInt<20>(AM.BaseOffs))
366 return false;
367
368 // Indexing is OK but no scale factor can be applied.
369 return AM.Scale == 0 || AM.Scale == 1;
370}
371
Richard Sandiford709bda62013-08-19 12:42:31 +0000372bool SystemZTargetLowering::isTruncateFree(Type *FromType, Type *ToType) const {
373 if (!FromType->isIntegerTy() || !ToType->isIntegerTy())
374 return false;
375 unsigned FromBits = FromType->getPrimitiveSizeInBits();
376 unsigned ToBits = ToType->getPrimitiveSizeInBits();
377 return FromBits > ToBits;
378}
379
380bool SystemZTargetLowering::isTruncateFree(EVT FromVT, EVT ToVT) const {
381 if (!FromVT.isInteger() || !ToVT.isInteger())
382 return false;
383 unsigned FromBits = FromVT.getSizeInBits();
384 unsigned ToBits = ToVT.getSizeInBits();
385 return FromBits > ToBits;
386}
387
Ulrich Weigand5f613df2013-05-06 16:15:19 +0000388//===----------------------------------------------------------------------===//
389// Inline asm support
390//===----------------------------------------------------------------------===//
391
392TargetLowering::ConstraintType
393SystemZTargetLowering::getConstraintType(const std::string &Constraint) const {
394 if (Constraint.size() == 1) {
395 switch (Constraint[0]) {
396 case 'a': // Address register
397 case 'd': // Data register (equivalent to 'r')
398 case 'f': // Floating-point register
Richard Sandiford0755c932013-10-01 11:26:28 +0000399 case 'h': // High-part register
Ulrich Weigand5f613df2013-05-06 16:15:19 +0000400 case 'r': // General-purpose register
401 return C_RegisterClass;
402
403 case 'Q': // Memory with base and unsigned 12-bit displacement
404 case 'R': // Likewise, plus an index
405 case 'S': // Memory with base and signed 20-bit displacement
406 case 'T': // Likewise, plus an index
407 case 'm': // Equivalent to 'T'.
408 return C_Memory;
409
410 case 'I': // Unsigned 8-bit constant
411 case 'J': // Unsigned 12-bit constant
412 case 'K': // Signed 16-bit constant
413 case 'L': // Signed 20-bit displacement (on all targets we support)
414 case 'M': // 0x7fffffff
415 return C_Other;
416
417 default:
418 break;
419 }
420 }
421 return TargetLowering::getConstraintType(Constraint);
422}
423
424TargetLowering::ConstraintWeight SystemZTargetLowering::
425getSingleConstraintMatchWeight(AsmOperandInfo &info,
426 const char *constraint) const {
427 ConstraintWeight weight = CW_Invalid;
428 Value *CallOperandVal = info.CallOperandVal;
429 // If we don't have a value, we can't do a match,
430 // but allow it at the lowest weight.
Craig Topper062a2ba2014-04-25 05:30:21 +0000431 if (!CallOperandVal)
Ulrich Weigand5f613df2013-05-06 16:15:19 +0000432 return CW_Default;
433 Type *type = CallOperandVal->getType();
434 // Look at the constraint type.
435 switch (*constraint) {
436 default:
437 weight = TargetLowering::getSingleConstraintMatchWeight(info, constraint);
438 break;
439
440 case 'a': // Address register
441 case 'd': // Data register (equivalent to 'r')
Richard Sandiford0755c932013-10-01 11:26:28 +0000442 case 'h': // High-part register
Ulrich Weigand5f613df2013-05-06 16:15:19 +0000443 case 'r': // General-purpose register
444 if (CallOperandVal->getType()->isIntegerTy())
445 weight = CW_Register;
446 break;
447
448 case 'f': // Floating-point register
449 if (type->isFloatingPointTy())
450 weight = CW_Register;
451 break;
452
453 case 'I': // Unsigned 8-bit constant
Richard Sandiford21f5d682014-03-06 11:22:58 +0000454 if (auto *C = dyn_cast<ConstantInt>(CallOperandVal))
Ulrich Weigand5f613df2013-05-06 16:15:19 +0000455 if (isUInt<8>(C->getZExtValue()))
456 weight = CW_Constant;
457 break;
458
459 case 'J': // Unsigned 12-bit constant
Richard Sandiford21f5d682014-03-06 11:22:58 +0000460 if (auto *C = dyn_cast<ConstantInt>(CallOperandVal))
Ulrich Weigand5f613df2013-05-06 16:15:19 +0000461 if (isUInt<12>(C->getZExtValue()))
462 weight = CW_Constant;
463 break;
464
465 case 'K': // Signed 16-bit constant
Richard Sandiford21f5d682014-03-06 11:22:58 +0000466 if (auto *C = dyn_cast<ConstantInt>(CallOperandVal))
Ulrich Weigand5f613df2013-05-06 16:15:19 +0000467 if (isInt<16>(C->getSExtValue()))
468 weight = CW_Constant;
469 break;
470
471 case 'L': // Signed 20-bit displacement (on all targets we support)
Richard Sandiford21f5d682014-03-06 11:22:58 +0000472 if (auto *C = dyn_cast<ConstantInt>(CallOperandVal))
Ulrich Weigand5f613df2013-05-06 16:15:19 +0000473 if (isInt<20>(C->getSExtValue()))
474 weight = CW_Constant;
475 break;
476
477 case 'M': // 0x7fffffff
Richard Sandiford21f5d682014-03-06 11:22:58 +0000478 if (auto *C = dyn_cast<ConstantInt>(CallOperandVal))
Ulrich Weigand5f613df2013-05-06 16:15:19 +0000479 if (C->getZExtValue() == 0x7fffffff)
480 weight = CW_Constant;
481 break;
482 }
483 return weight;
484}
485
Richard Sandifordb8204052013-07-12 09:08:12 +0000486// Parse a "{tNNN}" register constraint for which the register type "t"
487// has already been verified. MC is the class associated with "t" and
488// Map maps 0-based register numbers to LLVM register numbers.
489static std::pair<unsigned, const TargetRegisterClass *>
490parseRegisterNumber(const std::string &Constraint,
491 const TargetRegisterClass *RC, const unsigned *Map) {
492 assert(*(Constraint.end()-1) == '}' && "Missing '}'");
493 if (isdigit(Constraint[2])) {
494 std::string Suffix(Constraint.data() + 2, Constraint.size() - 2);
495 unsigned Index = atoi(Suffix.c_str());
496 if (Index < 16 && Map[Index])
497 return std::make_pair(Map[Index], RC);
498 }
Craig Topper062a2ba2014-04-25 05:30:21 +0000499 return std::make_pair(0U, nullptr);
Richard Sandifordb8204052013-07-12 09:08:12 +0000500}
501
Ulrich Weigand5f613df2013-05-06 16:15:19 +0000502std::pair<unsigned, const TargetRegisterClass *> SystemZTargetLowering::
Chad Rosier295bd432013-06-22 18:37:38 +0000503getRegForInlineAsmConstraint(const std::string &Constraint, MVT VT) const {
Ulrich Weigand5f613df2013-05-06 16:15:19 +0000504 if (Constraint.size() == 1) {
505 // GCC Constraint Letters
506 switch (Constraint[0]) {
507 default: break;
508 case 'd': // Data register (equivalent to 'r')
509 case 'r': // General-purpose register
510 if (VT == MVT::i64)
511 return std::make_pair(0U, &SystemZ::GR64BitRegClass);
512 else if (VT == MVT::i128)
513 return std::make_pair(0U, &SystemZ::GR128BitRegClass);
514 return std::make_pair(0U, &SystemZ::GR32BitRegClass);
515
516 case 'a': // Address register
517 if (VT == MVT::i64)
518 return std::make_pair(0U, &SystemZ::ADDR64BitRegClass);
519 else if (VT == MVT::i128)
520 return std::make_pair(0U, &SystemZ::ADDR128BitRegClass);
521 return std::make_pair(0U, &SystemZ::ADDR32BitRegClass);
522
Richard Sandiford0755c932013-10-01 11:26:28 +0000523 case 'h': // High-part register (an LLVM extension)
524 return std::make_pair(0U, &SystemZ::GRH32BitRegClass);
525
Ulrich Weigand5f613df2013-05-06 16:15:19 +0000526 case 'f': // Floating-point register
527 if (VT == MVT::f64)
528 return std::make_pair(0U, &SystemZ::FP64BitRegClass);
529 else if (VT == MVT::f128)
530 return std::make_pair(0U, &SystemZ::FP128BitRegClass);
531 return std::make_pair(0U, &SystemZ::FP32BitRegClass);
532 }
533 }
Richard Sandifordb8204052013-07-12 09:08:12 +0000534 if (Constraint[0] == '{') {
535 // We need to override the default register parsing for GPRs and FPRs
536 // because the interpretation depends on VT. The internal names of
537 // the registers are also different from the external names
538 // (F0D and F0S instead of F0, etc.).
539 if (Constraint[1] == 'r') {
540 if (VT == MVT::i32)
541 return parseRegisterNumber(Constraint, &SystemZ::GR32BitRegClass,
542 SystemZMC::GR32Regs);
543 if (VT == MVT::i128)
544 return parseRegisterNumber(Constraint, &SystemZ::GR128BitRegClass,
545 SystemZMC::GR128Regs);
546 return parseRegisterNumber(Constraint, &SystemZ::GR64BitRegClass,
547 SystemZMC::GR64Regs);
548 }
549 if (Constraint[1] == 'f') {
550 if (VT == MVT::f32)
551 return parseRegisterNumber(Constraint, &SystemZ::FP32BitRegClass,
552 SystemZMC::FP32Regs);
553 if (VT == MVT::f128)
554 return parseRegisterNumber(Constraint, &SystemZ::FP128BitRegClass,
555 SystemZMC::FP128Regs);
556 return parseRegisterNumber(Constraint, &SystemZ::FP64BitRegClass,
557 SystemZMC::FP64Regs);
558 }
559 }
Ulrich Weigand5f613df2013-05-06 16:15:19 +0000560 return TargetLowering::getRegForInlineAsmConstraint(Constraint, VT);
561}
562
563void SystemZTargetLowering::
564LowerAsmOperandForConstraint(SDValue Op, std::string &Constraint,
565 std::vector<SDValue> &Ops,
566 SelectionDAG &DAG) const {
567 // Only support length 1 constraints for now.
568 if (Constraint.length() == 1) {
569 switch (Constraint[0]) {
570 case 'I': // Unsigned 8-bit constant
Richard Sandiford21f5d682014-03-06 11:22:58 +0000571 if (auto *C = dyn_cast<ConstantSDNode>(Op))
Ulrich Weigand5f613df2013-05-06 16:15:19 +0000572 if (isUInt<8>(C->getZExtValue()))
573 Ops.push_back(DAG.getTargetConstant(C->getZExtValue(),
574 Op.getValueType()));
575 return;
576
577 case 'J': // Unsigned 12-bit constant
Richard Sandiford21f5d682014-03-06 11:22:58 +0000578 if (auto *C = dyn_cast<ConstantSDNode>(Op))
Ulrich Weigand5f613df2013-05-06 16:15:19 +0000579 if (isUInt<12>(C->getZExtValue()))
580 Ops.push_back(DAG.getTargetConstant(C->getZExtValue(),
581 Op.getValueType()));
582 return;
583
584 case 'K': // Signed 16-bit constant
Richard Sandiford21f5d682014-03-06 11:22:58 +0000585 if (auto *C = dyn_cast<ConstantSDNode>(Op))
Ulrich Weigand5f613df2013-05-06 16:15:19 +0000586 if (isInt<16>(C->getSExtValue()))
587 Ops.push_back(DAG.getTargetConstant(C->getSExtValue(),
588 Op.getValueType()));
589 return;
590
591 case 'L': // Signed 20-bit displacement (on all targets we support)
Richard Sandiford21f5d682014-03-06 11:22:58 +0000592 if (auto *C = dyn_cast<ConstantSDNode>(Op))
Ulrich Weigand5f613df2013-05-06 16:15:19 +0000593 if (isInt<20>(C->getSExtValue()))
594 Ops.push_back(DAG.getTargetConstant(C->getSExtValue(),
595 Op.getValueType()));
596 return;
597
598 case 'M': // 0x7fffffff
Richard Sandiford21f5d682014-03-06 11:22:58 +0000599 if (auto *C = dyn_cast<ConstantSDNode>(Op))
Ulrich Weigand5f613df2013-05-06 16:15:19 +0000600 if (C->getZExtValue() == 0x7fffffff)
601 Ops.push_back(DAG.getTargetConstant(C->getZExtValue(),
602 Op.getValueType()));
603 return;
604 }
605 }
606 TargetLowering::LowerAsmOperandForConstraint(Op, Constraint, Ops, DAG);
607}
608
609//===----------------------------------------------------------------------===//
610// Calling conventions
611//===----------------------------------------------------------------------===//
612
613#include "SystemZGenCallingConv.inc"
614
Richard Sandiford709bda62013-08-19 12:42:31 +0000615bool SystemZTargetLowering::allowTruncateForTailCall(Type *FromType,
616 Type *ToType) const {
617 return isTruncateFree(FromType, ToType);
618}
619
620bool SystemZTargetLowering::mayBeEmittedAsTailCall(CallInst *CI) const {
621 if (!CI->isTailCall())
622 return false;
623 return true;
624}
625
Ulrich Weigand5f613df2013-05-06 16:15:19 +0000626// Value is a value that has been passed to us in the location described by VA
627// (and so has type VA.getLocVT()). Convert Value to VA.getValVT(), chaining
628// any loads onto Chain.
Andrew Trickef9de2a2013-05-25 02:42:55 +0000629static SDValue convertLocVTToValVT(SelectionDAG &DAG, SDLoc DL,
Ulrich Weigand5f613df2013-05-06 16:15:19 +0000630 CCValAssign &VA, SDValue Chain,
631 SDValue Value) {
632 // If the argument has been promoted from a smaller type, insert an
633 // assertion to capture this.
634 if (VA.getLocInfo() == CCValAssign::SExt)
635 Value = DAG.getNode(ISD::AssertSext, DL, VA.getLocVT(), Value,
636 DAG.getValueType(VA.getValVT()));
637 else if (VA.getLocInfo() == CCValAssign::ZExt)
638 Value = DAG.getNode(ISD::AssertZext, DL, VA.getLocVT(), Value,
639 DAG.getValueType(VA.getValVT()));
640
641 if (VA.isExtInLoc())
642 Value = DAG.getNode(ISD::TRUNCATE, DL, VA.getValVT(), Value);
643 else if (VA.getLocInfo() == CCValAssign::Indirect)
644 Value = DAG.getLoad(VA.getValVT(), DL, Chain, Value,
645 MachinePointerInfo(), false, false, false, 0);
646 else
647 assert(VA.getLocInfo() == CCValAssign::Full && "Unsupported getLocInfo");
648 return Value;
649}
650
651// Value is a value of type VA.getValVT() that we need to copy into
652// the location described by VA. Return a copy of Value converted to
653// VA.getValVT(). The caller is responsible for handling indirect values.
Andrew Trickef9de2a2013-05-25 02:42:55 +0000654static SDValue convertValVTToLocVT(SelectionDAG &DAG, SDLoc DL,
Ulrich Weigand5f613df2013-05-06 16:15:19 +0000655 CCValAssign &VA, SDValue Value) {
656 switch (VA.getLocInfo()) {
657 case CCValAssign::SExt:
658 return DAG.getNode(ISD::SIGN_EXTEND, DL, VA.getLocVT(), Value);
659 case CCValAssign::ZExt:
660 return DAG.getNode(ISD::ZERO_EXTEND, DL, VA.getLocVT(), Value);
661 case CCValAssign::AExt:
662 return DAG.getNode(ISD::ANY_EXTEND, DL, VA.getLocVT(), Value);
663 case CCValAssign::Full:
664 return Value;
665 default:
666 llvm_unreachable("Unhandled getLocInfo()");
667 }
668}
669
670SDValue SystemZTargetLowering::
671LowerFormalArguments(SDValue Chain, CallingConv::ID CallConv, bool IsVarArg,
672 const SmallVectorImpl<ISD::InputArg> &Ins,
Andrew Trickef9de2a2013-05-25 02:42:55 +0000673 SDLoc DL, SelectionDAG &DAG,
Ulrich Weigand5f613df2013-05-06 16:15:19 +0000674 SmallVectorImpl<SDValue> &InVals) const {
675 MachineFunction &MF = DAG.getMachineFunction();
676 MachineFrameInfo *MFI = MF.getFrameInfo();
677 MachineRegisterInfo &MRI = MF.getRegInfo();
678 SystemZMachineFunctionInfo *FuncInfo =
Eric Christophera6734172015-01-31 00:06:45 +0000679 MF.getInfo<SystemZMachineFunctionInfo>();
680 auto *TFL =
681 static_cast<const SystemZFrameLowering *>(Subtarget.getFrameLowering());
Ulrich Weigand5f613df2013-05-06 16:15:19 +0000682
683 // Assign locations to all of the incoming arguments.
684 SmallVector<CCValAssign, 16> ArgLocs;
Eric Christopherb5217502014-08-06 18:45:26 +0000685 CCState CCInfo(CallConv, IsVarArg, MF, ArgLocs, *DAG.getContext());
Ulrich Weigand5f613df2013-05-06 16:15:19 +0000686 CCInfo.AnalyzeFormalArguments(Ins, CC_SystemZ);
687
688 unsigned NumFixedGPRs = 0;
689 unsigned NumFixedFPRs = 0;
690 for (unsigned I = 0, E = ArgLocs.size(); I != E; ++I) {
691 SDValue ArgValue;
692 CCValAssign &VA = ArgLocs[I];
693 EVT LocVT = VA.getLocVT();
694 if (VA.isRegLoc()) {
695 // Arguments passed in registers
696 const TargetRegisterClass *RC;
697 switch (LocVT.getSimpleVT().SimpleTy) {
698 default:
699 // Integers smaller than i64 should be promoted to i64.
700 llvm_unreachable("Unexpected argument type");
701 case MVT::i32:
702 NumFixedGPRs += 1;
703 RC = &SystemZ::GR32BitRegClass;
704 break;
705 case MVT::i64:
706 NumFixedGPRs += 1;
707 RC = &SystemZ::GR64BitRegClass;
708 break;
709 case MVT::f32:
710 NumFixedFPRs += 1;
711 RC = &SystemZ::FP32BitRegClass;
712 break;
713 case MVT::f64:
714 NumFixedFPRs += 1;
715 RC = &SystemZ::FP64BitRegClass;
716 break;
717 }
718
719 unsigned VReg = MRI.createVirtualRegister(RC);
720 MRI.addLiveIn(VA.getLocReg(), VReg);
721 ArgValue = DAG.getCopyFromReg(Chain, DL, VReg, LocVT);
722 } else {
723 assert(VA.isMemLoc() && "Argument not register or memory");
724
725 // Create the frame index object for this incoming parameter.
726 int FI = MFI->CreateFixedObject(LocVT.getSizeInBits() / 8,
727 VA.getLocMemOffset(), true);
728
729 // Create the SelectionDAG nodes corresponding to a load
730 // from this parameter. Unpromoted ints and floats are
731 // passed as right-justified 8-byte values.
732 EVT PtrVT = getPointerTy();
733 SDValue FIN = DAG.getFrameIndex(FI, PtrVT);
734 if (VA.getLocVT() == MVT::i32 || VA.getLocVT() == MVT::f32)
735 FIN = DAG.getNode(ISD::ADD, DL, PtrVT, FIN, DAG.getIntPtrConstant(4));
736 ArgValue = DAG.getLoad(LocVT, DL, Chain, FIN,
737 MachinePointerInfo::getFixedStack(FI),
738 false, false, false, 0);
739 }
740
741 // Convert the value of the argument register into the value that's
742 // being passed.
743 InVals.push_back(convertLocVTToValVT(DAG, DL, VA, Chain, ArgValue));
744 }
745
746 if (IsVarArg) {
747 // Save the number of non-varargs registers for later use by va_start, etc.
748 FuncInfo->setVarArgsFirstGPR(NumFixedGPRs);
749 FuncInfo->setVarArgsFirstFPR(NumFixedFPRs);
750
751 // Likewise the address (in the form of a frame index) of where the
752 // first stack vararg would be. The 1-byte size here is arbitrary.
753 int64_t StackSize = CCInfo.getNextStackOffset();
754 FuncInfo->setVarArgsFrameIndex(MFI->CreateFixedObject(1, StackSize, true));
755
756 // ...and a similar frame index for the caller-allocated save area
757 // that will be used to store the incoming registers.
758 int64_t RegSaveOffset = TFL->getOffsetOfLocalArea();
759 unsigned RegSaveIndex = MFI->CreateFixedObject(1, RegSaveOffset, true);
760 FuncInfo->setRegSaveFrameIndex(RegSaveIndex);
761
762 // Store the FPR varargs in the reserved frame slots. (We store the
763 // GPRs as part of the prologue.)
764 if (NumFixedFPRs < SystemZ::NumArgFPRs) {
765 SDValue MemOps[SystemZ::NumArgFPRs];
766 for (unsigned I = NumFixedFPRs; I < SystemZ::NumArgFPRs; ++I) {
767 unsigned Offset = TFL->getRegSpillOffset(SystemZ::ArgFPRs[I]);
768 int FI = MFI->CreateFixedObject(8, RegSaveOffset + Offset, true);
769 SDValue FIN = DAG.getFrameIndex(FI, getPointerTy());
770 unsigned VReg = MF.addLiveIn(SystemZ::ArgFPRs[I],
771 &SystemZ::FP64BitRegClass);
772 SDValue ArgValue = DAG.getCopyFromReg(Chain, DL, VReg, MVT::f64);
773 MemOps[I] = DAG.getStore(ArgValue.getValue(1), DL, ArgValue, FIN,
774 MachinePointerInfo::getFixedStack(FI),
775 false, false, 0);
776
777 }
778 // Join the stores, which are independent of one another.
779 Chain = DAG.getNode(ISD::TokenFactor, DL, MVT::Other,
Craig Topper2d2aa0c2014-04-30 07:17:30 +0000780 makeArrayRef(&MemOps[NumFixedFPRs],
781 SystemZ::NumArgFPRs-NumFixedFPRs));
Ulrich Weigand5f613df2013-05-06 16:15:19 +0000782 }
783 }
784
785 return Chain;
786}
787
Benjamin Kramerc6cc58e2014-10-04 16:55:56 +0000788static bool canUseSiblingCall(const CCState &ArgCCInfo,
Richard Sandiford709bda62013-08-19 12:42:31 +0000789 SmallVectorImpl<CCValAssign> &ArgLocs) {
790 // Punt if there are any indirect or stack arguments, or if the call
791 // needs the call-saved argument register R6.
792 for (unsigned I = 0, E = ArgLocs.size(); I != E; ++I) {
793 CCValAssign &VA = ArgLocs[I];
794 if (VA.getLocInfo() == CCValAssign::Indirect)
795 return false;
796 if (!VA.isRegLoc())
797 return false;
798 unsigned Reg = VA.getLocReg();
Richard Sandiford0755c932013-10-01 11:26:28 +0000799 if (Reg == SystemZ::R6H || Reg == SystemZ::R6L || Reg == SystemZ::R6D)
Richard Sandiford709bda62013-08-19 12:42:31 +0000800 return false;
801 }
802 return true;
803}
804
Ulrich Weigand5f613df2013-05-06 16:15:19 +0000805SDValue
806SystemZTargetLowering::LowerCall(CallLoweringInfo &CLI,
807 SmallVectorImpl<SDValue> &InVals) const {
808 SelectionDAG &DAG = CLI.DAG;
Andrew Trickef9de2a2013-05-25 02:42:55 +0000809 SDLoc &DL = CLI.DL;
Craig Topperb94011f2013-07-14 04:42:23 +0000810 SmallVectorImpl<ISD::OutputArg> &Outs = CLI.Outs;
811 SmallVectorImpl<SDValue> &OutVals = CLI.OutVals;
812 SmallVectorImpl<ISD::InputArg> &Ins = CLI.Ins;
Ulrich Weigand5f613df2013-05-06 16:15:19 +0000813 SDValue Chain = CLI.Chain;
814 SDValue Callee = CLI.Callee;
Richard Sandiford709bda62013-08-19 12:42:31 +0000815 bool &IsTailCall = CLI.IsTailCall;
Ulrich Weigand5f613df2013-05-06 16:15:19 +0000816 CallingConv::ID CallConv = CLI.CallConv;
817 bool IsVarArg = CLI.IsVarArg;
818 MachineFunction &MF = DAG.getMachineFunction();
819 EVT PtrVT = getPointerTy();
820
Ulrich Weigand5f613df2013-05-06 16:15:19 +0000821 // Analyze the operands of the call, assigning locations to each operand.
822 SmallVector<CCValAssign, 16> ArgLocs;
Eric Christopherb5217502014-08-06 18:45:26 +0000823 CCState ArgCCInfo(CallConv, IsVarArg, MF, ArgLocs, *DAG.getContext());
Ulrich Weigand5f613df2013-05-06 16:15:19 +0000824 ArgCCInfo.AnalyzeCallOperands(Outs, CC_SystemZ);
825
Richard Sandiford709bda62013-08-19 12:42:31 +0000826 // We don't support GuaranteedTailCallOpt, only automatically-detected
827 // sibling calls.
828 if (IsTailCall && !canUseSiblingCall(ArgCCInfo, ArgLocs))
829 IsTailCall = false;
830
Ulrich Weigand5f613df2013-05-06 16:15:19 +0000831 // Get a count of how many bytes are to be pushed on the stack.
832 unsigned NumBytes = ArgCCInfo.getNextStackOffset();
833
834 // Mark the start of the call.
Richard Sandiford709bda62013-08-19 12:42:31 +0000835 if (!IsTailCall)
836 Chain = DAG.getCALLSEQ_START(Chain, DAG.getConstant(NumBytes, PtrVT, true),
837 DL);
Ulrich Weigand5f613df2013-05-06 16:15:19 +0000838
839 // Copy argument values to their designated locations.
840 SmallVector<std::pair<unsigned, SDValue>, 9> RegsToPass;
841 SmallVector<SDValue, 8> MemOpChains;
842 SDValue StackPtr;
843 for (unsigned I = 0, E = ArgLocs.size(); I != E; ++I) {
844 CCValAssign &VA = ArgLocs[I];
845 SDValue ArgValue = OutVals[I];
846
847 if (VA.getLocInfo() == CCValAssign::Indirect) {
848 // Store the argument in a stack slot and pass its address.
849 SDValue SpillSlot = DAG.CreateStackTemporary(VA.getValVT());
850 int FI = cast<FrameIndexSDNode>(SpillSlot)->getIndex();
851 MemOpChains.push_back(DAG.getStore(Chain, DL, ArgValue, SpillSlot,
852 MachinePointerInfo::getFixedStack(FI),
853 false, false, 0));
854 ArgValue = SpillSlot;
855 } else
856 ArgValue = convertValVTToLocVT(DAG, DL, VA, ArgValue);
857
858 if (VA.isRegLoc())
859 // Queue up the argument copies and emit them at the end.
860 RegsToPass.push_back(std::make_pair(VA.getLocReg(), ArgValue));
861 else {
862 assert(VA.isMemLoc() && "Argument not register or memory");
863
864 // Work out the address of the stack slot. Unpromoted ints and
865 // floats are passed as right-justified 8-byte values.
866 if (!StackPtr.getNode())
867 StackPtr = DAG.getCopyFromReg(Chain, DL, SystemZ::R15D, PtrVT);
868 unsigned Offset = SystemZMC::CallFrameSize + VA.getLocMemOffset();
869 if (VA.getLocVT() == MVT::i32 || VA.getLocVT() == MVT::f32)
870 Offset += 4;
871 SDValue Address = DAG.getNode(ISD::ADD, DL, PtrVT, StackPtr,
872 DAG.getIntPtrConstant(Offset));
873
874 // Emit the store.
875 MemOpChains.push_back(DAG.getStore(Chain, DL, ArgValue, Address,
876 MachinePointerInfo(),
877 false, false, 0));
878 }
879 }
880
881 // Join the stores, which are independent of one another.
882 if (!MemOpChains.empty())
Craig Topper48d114b2014-04-26 18:35:24 +0000883 Chain = DAG.getNode(ISD::TokenFactor, DL, MVT::Other, MemOpChains);
Ulrich Weigand5f613df2013-05-06 16:15:19 +0000884
Ulrich Weigand5f613df2013-05-06 16:15:19 +0000885 // Accept direct calls by converting symbolic call addresses to the
Richard Sandiford709bda62013-08-19 12:42:31 +0000886 // associated Target* opcodes. Force %r1 to be used for indirect
887 // tail calls.
888 SDValue Glue;
Richard Sandiford21f5d682014-03-06 11:22:58 +0000889 if (auto *G = dyn_cast<GlobalAddressSDNode>(Callee)) {
Ulrich Weigand5f613df2013-05-06 16:15:19 +0000890 Callee = DAG.getTargetGlobalAddress(G->getGlobal(), DL, PtrVT);
891 Callee = DAG.getNode(SystemZISD::PCREL_WRAPPER, DL, PtrVT, Callee);
Richard Sandiford21f5d682014-03-06 11:22:58 +0000892 } else if (auto *E = dyn_cast<ExternalSymbolSDNode>(Callee)) {
Ulrich Weigand5f613df2013-05-06 16:15:19 +0000893 Callee = DAG.getTargetExternalSymbol(E->getSymbol(), PtrVT);
894 Callee = DAG.getNode(SystemZISD::PCREL_WRAPPER, DL, PtrVT, Callee);
Richard Sandiford709bda62013-08-19 12:42:31 +0000895 } else if (IsTailCall) {
896 Chain = DAG.getCopyToReg(Chain, DL, SystemZ::R1D, Callee, Glue);
897 Glue = Chain.getValue(1);
898 Callee = DAG.getRegister(SystemZ::R1D, Callee.getValueType());
899 }
900
901 // Build a sequence of copy-to-reg nodes, chained and glued together.
902 for (unsigned I = 0, E = RegsToPass.size(); I != E; ++I) {
903 Chain = DAG.getCopyToReg(Chain, DL, RegsToPass[I].first,
904 RegsToPass[I].second, Glue);
905 Glue = Chain.getValue(1);
Ulrich Weigand5f613df2013-05-06 16:15:19 +0000906 }
907
908 // The first call operand is the chain and the second is the target address.
909 SmallVector<SDValue, 8> Ops;
910 Ops.push_back(Chain);
911 Ops.push_back(Callee);
912
913 // Add argument registers to the end of the list so that they are
914 // known live into the call.
915 for (unsigned I = 0, E = RegsToPass.size(); I != E; ++I)
916 Ops.push_back(DAG.getRegister(RegsToPass[I].first,
917 RegsToPass[I].second.getValueType()));
918
Richard Sandiford02bb0ec2014-07-10 11:44:37 +0000919 // Add a register mask operand representing the call-preserved registers.
Eric Christophera6734172015-01-31 00:06:45 +0000920 const TargetRegisterInfo *TRI = Subtarget.getRegisterInfo();
Richard Sandiford02bb0ec2014-07-10 11:44:37 +0000921 const uint32_t *Mask = TRI->getCallPreservedMask(CallConv);
922 assert(Mask && "Missing call preserved mask for calling convention");
923 Ops.push_back(DAG.getRegisterMask(Mask));
924
Ulrich Weigand5f613df2013-05-06 16:15:19 +0000925 // Glue the call to the argument copies, if any.
926 if (Glue.getNode())
927 Ops.push_back(Glue);
928
929 // Emit the call.
930 SDVTList NodeTys = DAG.getVTList(MVT::Other, MVT::Glue);
Richard Sandiford709bda62013-08-19 12:42:31 +0000931 if (IsTailCall)
Craig Topper48d114b2014-04-26 18:35:24 +0000932 return DAG.getNode(SystemZISD::SIBCALL, DL, NodeTys, Ops);
933 Chain = DAG.getNode(SystemZISD::CALL, DL, NodeTys, Ops);
Ulrich Weigand5f613df2013-05-06 16:15:19 +0000934 Glue = Chain.getValue(1);
935
936 // Mark the end of the call, which is glued to the call itself.
937 Chain = DAG.getCALLSEQ_END(Chain,
938 DAG.getConstant(NumBytes, PtrVT, true),
939 DAG.getConstant(0, PtrVT, true),
Andrew Trickad6d08a2013-05-29 22:03:55 +0000940 Glue, DL);
Ulrich Weigand5f613df2013-05-06 16:15:19 +0000941 Glue = Chain.getValue(1);
942
943 // Assign locations to each value returned by this call.
944 SmallVector<CCValAssign, 16> RetLocs;
Eric Christopherb5217502014-08-06 18:45:26 +0000945 CCState RetCCInfo(CallConv, IsVarArg, MF, RetLocs, *DAG.getContext());
Ulrich Weigand5f613df2013-05-06 16:15:19 +0000946 RetCCInfo.AnalyzeCallResult(Ins, RetCC_SystemZ);
947
948 // Copy all of the result registers out of their specified physreg.
949 for (unsigned I = 0, E = RetLocs.size(); I != E; ++I) {
950 CCValAssign &VA = RetLocs[I];
951
952 // Copy the value out, gluing the copy to the end of the call sequence.
953 SDValue RetValue = DAG.getCopyFromReg(Chain, DL, VA.getLocReg(),
954 VA.getLocVT(), Glue);
955 Chain = RetValue.getValue(1);
956 Glue = RetValue.getValue(2);
957
958 // Convert the value of the return register into the value that's
959 // being returned.
960 InVals.push_back(convertLocVTToValVT(DAG, DL, VA, Chain, RetValue));
961 }
962
963 return Chain;
964}
965
966SDValue
967SystemZTargetLowering::LowerReturn(SDValue Chain,
968 CallingConv::ID CallConv, bool IsVarArg,
969 const SmallVectorImpl<ISD::OutputArg> &Outs,
970 const SmallVectorImpl<SDValue> &OutVals,
Andrew Trickef9de2a2013-05-25 02:42:55 +0000971 SDLoc DL, SelectionDAG &DAG) const {
Ulrich Weigand5f613df2013-05-06 16:15:19 +0000972 MachineFunction &MF = DAG.getMachineFunction();
973
974 // Assign locations to each returned value.
975 SmallVector<CCValAssign, 16> RetLocs;
Eric Christopherb5217502014-08-06 18:45:26 +0000976 CCState RetCCInfo(CallConv, IsVarArg, MF, RetLocs, *DAG.getContext());
Ulrich Weigand5f613df2013-05-06 16:15:19 +0000977 RetCCInfo.AnalyzeReturn(Outs, RetCC_SystemZ);
978
979 // Quick exit for void returns
980 if (RetLocs.empty())
981 return DAG.getNode(SystemZISD::RET_FLAG, DL, MVT::Other, Chain);
982
983 // Copy the result values into the output registers.
984 SDValue Glue;
985 SmallVector<SDValue, 4> RetOps;
986 RetOps.push_back(Chain);
987 for (unsigned I = 0, E = RetLocs.size(); I != E; ++I) {
988 CCValAssign &VA = RetLocs[I];
989 SDValue RetValue = OutVals[I];
990
991 // Make the return register live on exit.
992 assert(VA.isRegLoc() && "Can only return in registers!");
993
994 // Promote the value as required.
995 RetValue = convertValVTToLocVT(DAG, DL, VA, RetValue);
996
997 // Chain and glue the copies together.
998 unsigned Reg = VA.getLocReg();
999 Chain = DAG.getCopyToReg(Chain, DL, Reg, RetValue, Glue);
1000 Glue = Chain.getValue(1);
1001 RetOps.push_back(DAG.getRegister(Reg, VA.getLocVT()));
1002 }
1003
1004 // Update chain and glue.
1005 RetOps[0] = Chain;
1006 if (Glue.getNode())
1007 RetOps.push_back(Glue);
1008
Craig Topper48d114b2014-04-26 18:35:24 +00001009 return DAG.getNode(SystemZISD::RET_FLAG, DL, MVT::Other, RetOps);
Ulrich Weigand5f613df2013-05-06 16:15:19 +00001010}
1011
Richard Sandiford9afe6132013-12-10 10:36:34 +00001012SDValue SystemZTargetLowering::
1013prepareVolatileOrAtomicLoad(SDValue Chain, SDLoc DL, SelectionDAG &DAG) const {
1014 return DAG.getNode(SystemZISD::SERIALIZE, DL, MVT::Other, Chain);
1015}
1016
Ulrich Weigand5f613df2013-05-06 16:15:19 +00001017// CC is a comparison that will be implemented using an integer or
1018// floating-point comparison. Return the condition code mask for
1019// a branch on true. In the integer case, CCMASK_CMP_UO is set for
1020// unsigned comparisons and clear for signed ones. In the floating-point
1021// case, CCMASK_CMP_UO has its normal mask meaning (unordered).
1022static unsigned CCMaskForCondCode(ISD::CondCode CC) {
1023#define CONV(X) \
1024 case ISD::SET##X: return SystemZ::CCMASK_CMP_##X; \
1025 case ISD::SETO##X: return SystemZ::CCMASK_CMP_##X; \
1026 case ISD::SETU##X: return SystemZ::CCMASK_CMP_UO | SystemZ::CCMASK_CMP_##X
1027
1028 switch (CC) {
1029 default:
1030 llvm_unreachable("Invalid integer condition!");
1031
1032 CONV(EQ);
1033 CONV(NE);
1034 CONV(GT);
1035 CONV(GE);
1036 CONV(LT);
1037 CONV(LE);
1038
1039 case ISD::SETO: return SystemZ::CCMASK_CMP_O;
1040 case ISD::SETUO: return SystemZ::CCMASK_CMP_UO;
1041 }
1042#undef CONV
1043}
1044
Richard Sandifordf722a8e302013-10-16 11:10:55 +00001045// Return a sequence for getting a 1 from an IPM result when CC has a
1046// value in CCMask and a 0 when CC has a value in CCValid & ~CCMask.
1047// The handling of CC values outside CCValid doesn't matter.
1048static IPMConversion getIPMConversion(unsigned CCValid, unsigned CCMask) {
1049 // Deal with cases where the result can be taken directly from a bit
1050 // of the IPM result.
1051 if (CCMask == (CCValid & (SystemZ::CCMASK_1 | SystemZ::CCMASK_3)))
1052 return IPMConversion(0, 0, SystemZ::IPM_CC);
1053 if (CCMask == (CCValid & (SystemZ::CCMASK_2 | SystemZ::CCMASK_3)))
1054 return IPMConversion(0, 0, SystemZ::IPM_CC + 1);
1055
1056 // Deal with cases where we can add a value to force the sign bit
1057 // to contain the right value. Putting the bit in 31 means we can
1058 // use SRL rather than RISBG(L), and also makes it easier to get a
1059 // 0/-1 value, so it has priority over the other tests below.
1060 //
1061 // These sequences rely on the fact that the upper two bits of the
1062 // IPM result are zero.
1063 uint64_t TopBit = uint64_t(1) << 31;
1064 if (CCMask == (CCValid & SystemZ::CCMASK_0))
1065 return IPMConversion(0, -(1 << SystemZ::IPM_CC), 31);
1066 if (CCMask == (CCValid & (SystemZ::CCMASK_0 | SystemZ::CCMASK_1)))
1067 return IPMConversion(0, -(2 << SystemZ::IPM_CC), 31);
1068 if (CCMask == (CCValid & (SystemZ::CCMASK_0
1069 | SystemZ::CCMASK_1
1070 | SystemZ::CCMASK_2)))
1071 return IPMConversion(0, -(3 << SystemZ::IPM_CC), 31);
1072 if (CCMask == (CCValid & SystemZ::CCMASK_3))
1073 return IPMConversion(0, TopBit - (3 << SystemZ::IPM_CC), 31);
1074 if (CCMask == (CCValid & (SystemZ::CCMASK_1
1075 | SystemZ::CCMASK_2
1076 | SystemZ::CCMASK_3)))
1077 return IPMConversion(0, TopBit - (1 << SystemZ::IPM_CC), 31);
1078
1079 // Next try inverting the value and testing a bit. 0/1 could be
1080 // handled this way too, but we dealt with that case above.
1081 if (CCMask == (CCValid & (SystemZ::CCMASK_0 | SystemZ::CCMASK_2)))
1082 return IPMConversion(-1, 0, SystemZ::IPM_CC);
1083
1084 // Handle cases where adding a value forces a non-sign bit to contain
1085 // the right value.
1086 if (CCMask == (CCValid & (SystemZ::CCMASK_1 | SystemZ::CCMASK_2)))
1087 return IPMConversion(0, 1 << SystemZ::IPM_CC, SystemZ::IPM_CC + 1);
1088 if (CCMask == (CCValid & (SystemZ::CCMASK_0 | SystemZ::CCMASK_3)))
1089 return IPMConversion(0, -(1 << SystemZ::IPM_CC), SystemZ::IPM_CC + 1);
1090
Alp Tokercb402912014-01-24 17:20:08 +00001091 // The remaining cases are 1, 2, 0/1/3 and 0/2/3. All these are
Richard Sandifordf722a8e302013-10-16 11:10:55 +00001092 // can be done by inverting the low CC bit and applying one of the
1093 // sign-based extractions above.
1094 if (CCMask == (CCValid & SystemZ::CCMASK_1))
1095 return IPMConversion(1 << SystemZ::IPM_CC, -(1 << SystemZ::IPM_CC), 31);
1096 if (CCMask == (CCValid & SystemZ::CCMASK_2))
1097 return IPMConversion(1 << SystemZ::IPM_CC,
1098 TopBit - (3 << SystemZ::IPM_CC), 31);
1099 if (CCMask == (CCValid & (SystemZ::CCMASK_0
1100 | SystemZ::CCMASK_1
1101 | SystemZ::CCMASK_3)))
1102 return IPMConversion(1 << SystemZ::IPM_CC, -(3 << SystemZ::IPM_CC), 31);
1103 if (CCMask == (CCValid & (SystemZ::CCMASK_0
1104 | SystemZ::CCMASK_2
1105 | SystemZ::CCMASK_3)))
1106 return IPMConversion(1 << SystemZ::IPM_CC,
1107 TopBit - (1 << SystemZ::IPM_CC), 31);
1108
1109 llvm_unreachable("Unexpected CC combination");
1110}
1111
Richard Sandifordd420f732013-12-13 15:28:45 +00001112// If C can be converted to a comparison against zero, adjust the operands
Richard Sandiforda0757082013-08-01 10:29:45 +00001113// as necessary.
Richard Sandifordd420f732013-12-13 15:28:45 +00001114static void adjustZeroCmp(SelectionDAG &DAG, Comparison &C) {
1115 if (C.ICmpType == SystemZICMP::UnsignedOnly)
Richard Sandiforda0757082013-08-01 10:29:45 +00001116 return;
1117
Richard Sandiford21f5d682014-03-06 11:22:58 +00001118 auto *ConstOp1 = dyn_cast<ConstantSDNode>(C.Op1.getNode());
Richard Sandiforda0757082013-08-01 10:29:45 +00001119 if (!ConstOp1)
1120 return;
1121
1122 int64_t Value = ConstOp1->getSExtValue();
Richard Sandifordd420f732013-12-13 15:28:45 +00001123 if ((Value == -1 && C.CCMask == SystemZ::CCMASK_CMP_GT) ||
1124 (Value == -1 && C.CCMask == SystemZ::CCMASK_CMP_LE) ||
1125 (Value == 1 && C.CCMask == SystemZ::CCMASK_CMP_LT) ||
1126 (Value == 1 && C.CCMask == SystemZ::CCMASK_CMP_GE)) {
1127 C.CCMask ^= SystemZ::CCMASK_CMP_EQ;
1128 C.Op1 = DAG.getConstant(0, C.Op1.getValueType());
Richard Sandiforda0757082013-08-01 10:29:45 +00001129 }
1130}
1131
Richard Sandifordd420f732013-12-13 15:28:45 +00001132// If a comparison described by C is suitable for CLI(Y), CHHSI or CLHHSI,
1133// adjust the operands as necessary.
1134static void adjustSubwordCmp(SelectionDAG &DAG, Comparison &C) {
Ulrich Weigand5f613df2013-05-06 16:15:19 +00001135 // For us to make any changes, it must a comparison between a single-use
1136 // load and a constant.
Richard Sandifordd420f732013-12-13 15:28:45 +00001137 if (!C.Op0.hasOneUse() ||
1138 C.Op0.getOpcode() != ISD::LOAD ||
1139 C.Op1.getOpcode() != ISD::Constant)
Ulrich Weigand5f613df2013-05-06 16:15:19 +00001140 return;
1141
1142 // We must have an 8- or 16-bit load.
Richard Sandiford21f5d682014-03-06 11:22:58 +00001143 auto *Load = cast<LoadSDNode>(C.Op0);
Ulrich Weigand5f613df2013-05-06 16:15:19 +00001144 unsigned NumBits = Load->getMemoryVT().getStoreSizeInBits();
1145 if (NumBits != 8 && NumBits != 16)
1146 return;
1147
1148 // The load must be an extending one and the constant must be within the
1149 // range of the unextended value.
Richard Sandiford21f5d682014-03-06 11:22:58 +00001150 auto *ConstOp1 = cast<ConstantSDNode>(C.Op1);
Richard Sandifordd420f732013-12-13 15:28:45 +00001151 uint64_t Value = ConstOp1->getZExtValue();
Ulrich Weigand5f613df2013-05-06 16:15:19 +00001152 uint64_t Mask = (1 << NumBits) - 1;
1153 if (Load->getExtensionType() == ISD::SEXTLOAD) {
Richard Sandifordd420f732013-12-13 15:28:45 +00001154 // Make sure that ConstOp1 is in range of C.Op0.
1155 int64_t SignedValue = ConstOp1->getSExtValue();
1156 if (uint64_t(SignedValue) + (uint64_t(1) << (NumBits - 1)) > Mask)
Ulrich Weigand5f613df2013-05-06 16:15:19 +00001157 return;
Richard Sandifordd420f732013-12-13 15:28:45 +00001158 if (C.ICmpType != SystemZICMP::SignedOnly) {
1159 // Unsigned comparison between two sign-extended values is equivalent
1160 // to unsigned comparison between two zero-extended values.
Ulrich Weigand5f613df2013-05-06 16:15:19 +00001161 Value &= Mask;
Richard Sandifordd420f732013-12-13 15:28:45 +00001162 } else if (NumBits == 8) {
Ulrich Weigand5f613df2013-05-06 16:15:19 +00001163 // Try to treat the comparison as unsigned, so that we can use CLI.
1164 // Adjust CCMask and Value as necessary.
Richard Sandifordd420f732013-12-13 15:28:45 +00001165 if (Value == 0 && C.CCMask == SystemZ::CCMASK_CMP_LT)
Ulrich Weigand5f613df2013-05-06 16:15:19 +00001166 // Test whether the high bit of the byte is set.
Richard Sandifordd420f732013-12-13 15:28:45 +00001167 Value = 127, C.CCMask = SystemZ::CCMASK_CMP_GT;
1168 else if (Value == 0 && C.CCMask == SystemZ::CCMASK_CMP_GE)
Ulrich Weigand5f613df2013-05-06 16:15:19 +00001169 // Test whether the high bit of the byte is clear.
Richard Sandifordd420f732013-12-13 15:28:45 +00001170 Value = 128, C.CCMask = SystemZ::CCMASK_CMP_LT;
Ulrich Weigand5f613df2013-05-06 16:15:19 +00001171 else
1172 // No instruction exists for this combination.
1173 return;
Richard Sandifordd420f732013-12-13 15:28:45 +00001174 C.ICmpType = SystemZICMP::UnsignedOnly;
Ulrich Weigand5f613df2013-05-06 16:15:19 +00001175 }
1176 } else if (Load->getExtensionType() == ISD::ZEXTLOAD) {
1177 if (Value > Mask)
1178 return;
Richard Sandifordd420f732013-12-13 15:28:45 +00001179 assert(C.ICmpType == SystemZICMP::Any &&
1180 "Signedness shouldn't matter here.");
Ulrich Weigand5f613df2013-05-06 16:15:19 +00001181 } else
1182 return;
1183
1184 // Make sure that the first operand is an i32 of the right extension type.
Richard Sandifordd420f732013-12-13 15:28:45 +00001185 ISD::LoadExtType ExtType = (C.ICmpType == SystemZICMP::SignedOnly ?
1186 ISD::SEXTLOAD :
1187 ISD::ZEXTLOAD);
1188 if (C.Op0.getValueType() != MVT::i32 ||
Ulrich Weigand5f613df2013-05-06 16:15:19 +00001189 Load->getExtensionType() != ExtType)
Richard Sandifordd420f732013-12-13 15:28:45 +00001190 C.Op0 = DAG.getExtLoad(ExtType, SDLoc(Load), MVT::i32,
1191 Load->getChain(), Load->getBasePtr(),
1192 Load->getPointerInfo(), Load->getMemoryVT(),
1193 Load->isVolatile(), Load->isNonTemporal(),
Louis Gerbarg67474e32014-07-31 21:45:05 +00001194 Load->isInvariant(), Load->getAlignment());
Ulrich Weigand5f613df2013-05-06 16:15:19 +00001195
1196 // Make sure that the second operand is an i32 with the right value.
Richard Sandifordd420f732013-12-13 15:28:45 +00001197 if (C.Op1.getValueType() != MVT::i32 ||
1198 Value != ConstOp1->getZExtValue())
1199 C.Op1 = DAG.getConstant(Value, MVT::i32);
Ulrich Weigand5f613df2013-05-06 16:15:19 +00001200}
1201
Richard Sandiford5bc670b2013-09-06 11:51:39 +00001202// Return true if Op is either an unextended load, or a load suitable
1203// for integer register-memory comparisons of type ICmpType.
1204static bool isNaturalMemoryOperand(SDValue Op, unsigned ICmpType) {
Richard Sandiford21f5d682014-03-06 11:22:58 +00001205 auto *Load = dyn_cast<LoadSDNode>(Op.getNode());
Richard Sandiford5bc670b2013-09-06 11:51:39 +00001206 if (Load) {
1207 // There are no instructions to compare a register with a memory byte.
1208 if (Load->getMemoryVT() == MVT::i8)
1209 return false;
1210 // Otherwise decide on extension type.
Richard Sandiford24e597b2013-08-23 11:27:19 +00001211 switch (Load->getExtensionType()) {
1212 case ISD::NON_EXTLOAD:
Richard Sandiford24e597b2013-08-23 11:27:19 +00001213 return true;
1214 case ISD::SEXTLOAD:
Richard Sandiford5bc670b2013-09-06 11:51:39 +00001215 return ICmpType != SystemZICMP::UnsignedOnly;
Richard Sandiford24e597b2013-08-23 11:27:19 +00001216 case ISD::ZEXTLOAD:
Richard Sandiford5bc670b2013-09-06 11:51:39 +00001217 return ICmpType != SystemZICMP::SignedOnly;
Richard Sandiford24e597b2013-08-23 11:27:19 +00001218 default:
1219 break;
1220 }
Richard Sandiford5bc670b2013-09-06 11:51:39 +00001221 }
Richard Sandiford24e597b2013-08-23 11:27:19 +00001222 return false;
1223}
1224
Richard Sandifordd420f732013-12-13 15:28:45 +00001225// Return true if it is better to swap the operands of C.
1226static bool shouldSwapCmpOperands(const Comparison &C) {
Richard Sandiford24e597b2013-08-23 11:27:19 +00001227 // Leave f128 comparisons alone, since they have no memory forms.
Richard Sandifordd420f732013-12-13 15:28:45 +00001228 if (C.Op0.getValueType() == MVT::f128)
Richard Sandiford24e597b2013-08-23 11:27:19 +00001229 return false;
1230
1231 // Always keep a floating-point constant second, since comparisons with
1232 // zero can use LOAD TEST and comparisons with other constants make a
1233 // natural memory operand.
Richard Sandifordd420f732013-12-13 15:28:45 +00001234 if (isa<ConstantFPSDNode>(C.Op1))
Richard Sandiford24e597b2013-08-23 11:27:19 +00001235 return false;
1236
1237 // Never swap comparisons with zero since there are many ways to optimize
1238 // those later.
Richard Sandiford21f5d682014-03-06 11:22:58 +00001239 auto *ConstOp1 = dyn_cast<ConstantSDNode>(C.Op1);
Richard Sandifordd420f732013-12-13 15:28:45 +00001240 if (ConstOp1 && ConstOp1->getZExtValue() == 0)
Richard Sandiford24e597b2013-08-23 11:27:19 +00001241 return false;
1242
Richard Sandiford7b4118a2013-12-06 09:56:50 +00001243 // Also keep natural memory operands second if the loaded value is
1244 // only used here. Several comparisons have memory forms.
Richard Sandifordd420f732013-12-13 15:28:45 +00001245 if (isNaturalMemoryOperand(C.Op1, C.ICmpType) && C.Op1.hasOneUse())
Richard Sandiford7b4118a2013-12-06 09:56:50 +00001246 return false;
1247
Richard Sandiford24e597b2013-08-23 11:27:19 +00001248 // Look for cases where Cmp0 is a single-use load and Cmp1 isn't.
1249 // In that case we generally prefer the memory to be second.
Richard Sandifordd420f732013-12-13 15:28:45 +00001250 if (isNaturalMemoryOperand(C.Op0, C.ICmpType) && C.Op0.hasOneUse()) {
Richard Sandiford24e597b2013-08-23 11:27:19 +00001251 // The only exceptions are when the second operand is a constant and
1252 // we can use things like CHHSI.
Richard Sandifordd420f732013-12-13 15:28:45 +00001253 if (!ConstOp1)
Richard Sandiford24e597b2013-08-23 11:27:19 +00001254 return true;
Richard Sandiford5bc670b2013-09-06 11:51:39 +00001255 // The unsigned memory-immediate instructions can handle 16-bit
1256 // unsigned integers.
Richard Sandifordd420f732013-12-13 15:28:45 +00001257 if (C.ICmpType != SystemZICMP::SignedOnly &&
1258 isUInt<16>(ConstOp1->getZExtValue()))
Richard Sandiford5bc670b2013-09-06 11:51:39 +00001259 return false;
1260 // The signed memory-immediate instructions can handle 16-bit
1261 // signed integers.
Richard Sandifordd420f732013-12-13 15:28:45 +00001262 if (C.ICmpType != SystemZICMP::UnsignedOnly &&
1263 isInt<16>(ConstOp1->getSExtValue()))
Richard Sandiford5bc670b2013-09-06 11:51:39 +00001264 return false;
Richard Sandiford24e597b2013-08-23 11:27:19 +00001265 return true;
1266 }
Richard Sandiford7b4118a2013-12-06 09:56:50 +00001267
1268 // Try to promote the use of CGFR and CLGFR.
Richard Sandifordd420f732013-12-13 15:28:45 +00001269 unsigned Opcode0 = C.Op0.getOpcode();
1270 if (C.ICmpType != SystemZICMP::UnsignedOnly && Opcode0 == ISD::SIGN_EXTEND)
Richard Sandiford7b4118a2013-12-06 09:56:50 +00001271 return true;
Richard Sandifordd420f732013-12-13 15:28:45 +00001272 if (C.ICmpType != SystemZICMP::SignedOnly && Opcode0 == ISD::ZERO_EXTEND)
Richard Sandiford7b4118a2013-12-06 09:56:50 +00001273 return true;
Richard Sandifordd420f732013-12-13 15:28:45 +00001274 if (C.ICmpType != SystemZICMP::SignedOnly &&
Richard Sandiford7b4118a2013-12-06 09:56:50 +00001275 Opcode0 == ISD::AND &&
Richard Sandifordd420f732013-12-13 15:28:45 +00001276 C.Op0.getOperand(1).getOpcode() == ISD::Constant &&
1277 cast<ConstantSDNode>(C.Op0.getOperand(1))->getZExtValue() == 0xffffffff)
Richard Sandiford7b4118a2013-12-06 09:56:50 +00001278 return true;
1279
Richard Sandiford24e597b2013-08-23 11:27:19 +00001280 return false;
1281}
1282
Richard Sandiford73170f82013-12-11 11:45:08 +00001283// Return a version of comparison CC mask CCMask in which the LT and GT
1284// actions are swapped.
1285static unsigned reverseCCMask(unsigned CCMask) {
1286 return ((CCMask & SystemZ::CCMASK_CMP_EQ) |
1287 (CCMask & SystemZ::CCMASK_CMP_GT ? SystemZ::CCMASK_CMP_LT : 0) |
1288 (CCMask & SystemZ::CCMASK_CMP_LT ? SystemZ::CCMASK_CMP_GT : 0) |
1289 (CCMask & SystemZ::CCMASK_CMP_UO));
1290}
1291
Richard Sandiford0847c452013-12-13 15:50:30 +00001292// Check whether C tests for equality between X and Y and whether X - Y
1293// or Y - X is also computed. In that case it's better to compare the
1294// result of the subtraction against zero.
1295static void adjustForSubtraction(SelectionDAG &DAG, Comparison &C) {
1296 if (C.CCMask == SystemZ::CCMASK_CMP_EQ ||
1297 C.CCMask == SystemZ::CCMASK_CMP_NE) {
Richard Sandiford28c111e2014-03-06 11:00:15 +00001298 for (auto I = C.Op0->use_begin(), E = C.Op0->use_end(); I != E; ++I) {
Richard Sandiford0847c452013-12-13 15:50:30 +00001299 SDNode *N = *I;
1300 if (N->getOpcode() == ISD::SUB &&
1301 ((N->getOperand(0) == C.Op0 && N->getOperand(1) == C.Op1) ||
1302 (N->getOperand(0) == C.Op1 && N->getOperand(1) == C.Op0))) {
1303 C.Op0 = SDValue(N, 0);
1304 C.Op1 = DAG.getConstant(0, N->getValueType(0));
1305 return;
1306 }
1307 }
1308 }
1309}
1310
Richard Sandifordd420f732013-12-13 15:28:45 +00001311// Check whether C compares a floating-point value with zero and if that
1312// floating-point value is also negated. In this case we can use the
1313// negation to set CC, so avoiding separate LOAD AND TEST and
1314// LOAD (NEGATIVE/COMPLEMENT) instructions.
1315static void adjustForFNeg(Comparison &C) {
Richard Sandiford21f5d682014-03-06 11:22:58 +00001316 auto *C1 = dyn_cast<ConstantFPSDNode>(C.Op1);
Richard Sandiford73170f82013-12-11 11:45:08 +00001317 if (C1 && C1->isZero()) {
Richard Sandiford28c111e2014-03-06 11:00:15 +00001318 for (auto I = C.Op0->use_begin(), E = C.Op0->use_end(); I != E; ++I) {
Richard Sandiford73170f82013-12-11 11:45:08 +00001319 SDNode *N = *I;
1320 if (N->getOpcode() == ISD::FNEG) {
Richard Sandifordd420f732013-12-13 15:28:45 +00001321 C.Op0 = SDValue(N, 0);
1322 C.CCMask = reverseCCMask(C.CCMask);
Richard Sandiford73170f82013-12-11 11:45:08 +00001323 return;
1324 }
1325 }
1326 }
1327}
1328
Richard Sandifordd420f732013-12-13 15:28:45 +00001329// Check whether C compares (shl X, 32) with 0 and whether X is
Richard Sandifordbd2f0e92013-12-13 15:07:39 +00001330// also sign-extended. In that case it is better to test the result
1331// of the sign extension using LTGFR.
1332//
1333// This case is important because InstCombine transforms a comparison
1334// with (sext (trunc X)) into a comparison with (shl X, 32).
Richard Sandifordd420f732013-12-13 15:28:45 +00001335static void adjustForLTGFR(Comparison &C) {
Richard Sandifordbd2f0e92013-12-13 15:07:39 +00001336 // Check for a comparison between (shl X, 32) and 0.
Richard Sandifordd420f732013-12-13 15:28:45 +00001337 if (C.Op0.getOpcode() == ISD::SHL &&
1338 C.Op0.getValueType() == MVT::i64 &&
1339 C.Op1.getOpcode() == ISD::Constant &&
1340 cast<ConstantSDNode>(C.Op1)->getZExtValue() == 0) {
Richard Sandiford21f5d682014-03-06 11:22:58 +00001341 auto *C1 = dyn_cast<ConstantSDNode>(C.Op0.getOperand(1));
Richard Sandifordbd2f0e92013-12-13 15:07:39 +00001342 if (C1 && C1->getZExtValue() == 32) {
Richard Sandifordd420f732013-12-13 15:28:45 +00001343 SDValue ShlOp0 = C.Op0.getOperand(0);
Richard Sandifordbd2f0e92013-12-13 15:07:39 +00001344 // See whether X has any SIGN_EXTEND_INREG uses.
Richard Sandiford28c111e2014-03-06 11:00:15 +00001345 for (auto I = ShlOp0->use_begin(), E = ShlOp0->use_end(); I != E; ++I) {
Richard Sandifordbd2f0e92013-12-13 15:07:39 +00001346 SDNode *N = *I;
1347 if (N->getOpcode() == ISD::SIGN_EXTEND_INREG &&
1348 cast<VTSDNode>(N->getOperand(1))->getVT() == MVT::i32) {
Richard Sandifordd420f732013-12-13 15:28:45 +00001349 C.Op0 = SDValue(N, 0);
Richard Sandifordbd2f0e92013-12-13 15:07:39 +00001350 return;
1351 }
1352 }
1353 }
1354 }
1355}
1356
Richard Sandiford83a0b6a2013-12-20 11:56:02 +00001357// If C compares the truncation of an extending load, try to compare
1358// the untruncated value instead. This exposes more opportunities to
1359// reuse CC.
1360static void adjustICmpTruncate(SelectionDAG &DAG, Comparison &C) {
1361 if (C.Op0.getOpcode() == ISD::TRUNCATE &&
1362 C.Op0.getOperand(0).getOpcode() == ISD::LOAD &&
1363 C.Op1.getOpcode() == ISD::Constant &&
1364 cast<ConstantSDNode>(C.Op1)->getZExtValue() == 0) {
Richard Sandiford21f5d682014-03-06 11:22:58 +00001365 auto *L = cast<LoadSDNode>(C.Op0.getOperand(0));
Richard Sandiford83a0b6a2013-12-20 11:56:02 +00001366 if (L->getMemoryVT().getStoreSizeInBits()
1367 <= C.Op0.getValueType().getSizeInBits()) {
1368 unsigned Type = L->getExtensionType();
1369 if ((Type == ISD::ZEXTLOAD && C.ICmpType != SystemZICMP::SignedOnly) ||
1370 (Type == ISD::SEXTLOAD && C.ICmpType != SystemZICMP::UnsignedOnly)) {
1371 C.Op0 = C.Op0.getOperand(0);
1372 C.Op1 = DAG.getConstant(0, C.Op0.getValueType());
1373 }
1374 }
1375 }
1376}
1377
Richard Sandiford030c1652013-09-13 09:09:50 +00001378// Return true if shift operation N has an in-range constant shift value.
1379// Store it in ShiftVal if so.
1380static bool isSimpleShift(SDValue N, unsigned &ShiftVal) {
Richard Sandiford21f5d682014-03-06 11:22:58 +00001381 auto *Shift = dyn_cast<ConstantSDNode>(N.getOperand(1));
Richard Sandiford030c1652013-09-13 09:09:50 +00001382 if (!Shift)
1383 return false;
1384
1385 uint64_t Amount = Shift->getZExtValue();
1386 if (Amount >= N.getValueType().getSizeInBits())
1387 return false;
1388
1389 ShiftVal = Amount;
1390 return true;
1391}
1392
1393// Check whether an AND with Mask is suitable for a TEST UNDER MASK
1394// instruction and whether the CC value is descriptive enough to handle
1395// a comparison of type Opcode between the AND result and CmpVal.
1396// CCMask says which comparison result is being tested and BitSize is
1397// the number of bits in the operands. If TEST UNDER MASK can be used,
1398// return the corresponding CC mask, otherwise return 0.
Richard Sandiford5bc670b2013-09-06 11:51:39 +00001399static unsigned getTestUnderMaskCond(unsigned BitSize, unsigned CCMask,
1400 uint64_t Mask, uint64_t CmpVal,
1401 unsigned ICmpType) {
Richard Sandiford113c8702013-09-03 15:38:35 +00001402 assert(Mask != 0 && "ANDs with zero should have been removed by now");
1403
Richard Sandiford030c1652013-09-13 09:09:50 +00001404 // Check whether the mask is suitable for TMHH, TMHL, TMLH or TMLL.
1405 if (!SystemZ::isImmLL(Mask) && !SystemZ::isImmLH(Mask) &&
1406 !SystemZ::isImmHL(Mask) && !SystemZ::isImmHH(Mask))
1407 return 0;
1408
Richard Sandiford113c8702013-09-03 15:38:35 +00001409 // Work out the masks for the lowest and highest bits.
1410 unsigned HighShift = 63 - countLeadingZeros(Mask);
1411 uint64_t High = uint64_t(1) << HighShift;
1412 uint64_t Low = uint64_t(1) << countTrailingZeros(Mask);
1413
1414 // Signed ordered comparisons are effectively unsigned if the sign
1415 // bit is dropped.
Richard Sandiford5bc670b2013-09-06 11:51:39 +00001416 bool EffectivelyUnsigned = (ICmpType != SystemZICMP::SignedOnly);
Richard Sandiford113c8702013-09-03 15:38:35 +00001417
1418 // Check for equality comparisons with 0, or the equivalent.
1419 if (CmpVal == 0) {
1420 if (CCMask == SystemZ::CCMASK_CMP_EQ)
1421 return SystemZ::CCMASK_TM_ALL_0;
1422 if (CCMask == SystemZ::CCMASK_CMP_NE)
1423 return SystemZ::CCMASK_TM_SOME_1;
1424 }
1425 if (EffectivelyUnsigned && CmpVal <= Low) {
1426 if (CCMask == SystemZ::CCMASK_CMP_LT)
1427 return SystemZ::CCMASK_TM_ALL_0;
1428 if (CCMask == SystemZ::CCMASK_CMP_GE)
1429 return SystemZ::CCMASK_TM_SOME_1;
1430 }
1431 if (EffectivelyUnsigned && CmpVal < Low) {
1432 if (CCMask == SystemZ::CCMASK_CMP_LE)
1433 return SystemZ::CCMASK_TM_ALL_0;
1434 if (CCMask == SystemZ::CCMASK_CMP_GT)
1435 return SystemZ::CCMASK_TM_SOME_1;
1436 }
1437
1438 // Check for equality comparisons with the mask, or the equivalent.
1439 if (CmpVal == Mask) {
1440 if (CCMask == SystemZ::CCMASK_CMP_EQ)
1441 return SystemZ::CCMASK_TM_ALL_1;
1442 if (CCMask == SystemZ::CCMASK_CMP_NE)
1443 return SystemZ::CCMASK_TM_SOME_0;
1444 }
1445 if (EffectivelyUnsigned && CmpVal >= Mask - Low && CmpVal < Mask) {
1446 if (CCMask == SystemZ::CCMASK_CMP_GT)
1447 return SystemZ::CCMASK_TM_ALL_1;
1448 if (CCMask == SystemZ::CCMASK_CMP_LE)
1449 return SystemZ::CCMASK_TM_SOME_0;
1450 }
1451 if (EffectivelyUnsigned && CmpVal > Mask - Low && CmpVal <= Mask) {
1452 if (CCMask == SystemZ::CCMASK_CMP_GE)
1453 return SystemZ::CCMASK_TM_ALL_1;
1454 if (CCMask == SystemZ::CCMASK_CMP_LT)
1455 return SystemZ::CCMASK_TM_SOME_0;
1456 }
1457
1458 // Check for ordered comparisons with the top bit.
1459 if (EffectivelyUnsigned && CmpVal >= Mask - High && CmpVal < High) {
1460 if (CCMask == SystemZ::CCMASK_CMP_LE)
1461 return SystemZ::CCMASK_TM_MSB_0;
1462 if (CCMask == SystemZ::CCMASK_CMP_GT)
1463 return SystemZ::CCMASK_TM_MSB_1;
1464 }
1465 if (EffectivelyUnsigned && CmpVal > Mask - High && CmpVal <= High) {
1466 if (CCMask == SystemZ::CCMASK_CMP_LT)
1467 return SystemZ::CCMASK_TM_MSB_0;
1468 if (CCMask == SystemZ::CCMASK_CMP_GE)
1469 return SystemZ::CCMASK_TM_MSB_1;
1470 }
1471
1472 // If there are just two bits, we can do equality checks for Low and High
1473 // as well.
1474 if (Mask == Low + High) {
1475 if (CCMask == SystemZ::CCMASK_CMP_EQ && CmpVal == Low)
1476 return SystemZ::CCMASK_TM_MIXED_MSB_0;
1477 if (CCMask == SystemZ::CCMASK_CMP_NE && CmpVal == Low)
1478 return SystemZ::CCMASK_TM_MIXED_MSB_0 ^ SystemZ::CCMASK_ANY;
1479 if (CCMask == SystemZ::CCMASK_CMP_EQ && CmpVal == High)
1480 return SystemZ::CCMASK_TM_MIXED_MSB_1;
1481 if (CCMask == SystemZ::CCMASK_CMP_NE && CmpVal == High)
1482 return SystemZ::CCMASK_TM_MIXED_MSB_1 ^ SystemZ::CCMASK_ANY;
1483 }
1484
1485 // Looks like we've exhausted our options.
1486 return 0;
1487}
1488
Richard Sandifordd420f732013-12-13 15:28:45 +00001489// See whether C can be implemented as a TEST UNDER MASK instruction.
1490// Update the arguments with the TM version if so.
1491static void adjustForTestUnderMask(SelectionDAG &DAG, Comparison &C) {
Richard Sandiford113c8702013-09-03 15:38:35 +00001492 // Check that we have a comparison with a constant.
Richard Sandiford21f5d682014-03-06 11:22:58 +00001493 auto *ConstOp1 = dyn_cast<ConstantSDNode>(C.Op1);
Richard Sandifordd420f732013-12-13 15:28:45 +00001494 if (!ConstOp1)
Richard Sandiford35b9be22013-08-28 10:31:43 +00001495 return;
Richard Sandifordd420f732013-12-13 15:28:45 +00001496 uint64_t CmpVal = ConstOp1->getZExtValue();
Richard Sandiford35b9be22013-08-28 10:31:43 +00001497
1498 // Check whether the nonconstant input is an AND with a constant mask.
Richard Sandifordc3dc4472013-12-13 15:46:55 +00001499 Comparison NewC(C);
1500 uint64_t MaskVal;
Craig Topper062a2ba2014-04-25 05:30:21 +00001501 ConstantSDNode *Mask = nullptr;
Richard Sandifordc3dc4472013-12-13 15:46:55 +00001502 if (C.Op0.getOpcode() == ISD::AND) {
1503 NewC.Op0 = C.Op0.getOperand(0);
1504 NewC.Op1 = C.Op0.getOperand(1);
1505 Mask = dyn_cast<ConstantSDNode>(NewC.Op1);
1506 if (!Mask)
1507 return;
1508 MaskVal = Mask->getZExtValue();
1509 } else {
1510 // There is no instruction to compare with a 64-bit immediate
1511 // so use TMHH instead if possible. We need an unsigned ordered
1512 // comparison with an i64 immediate.
1513 if (NewC.Op0.getValueType() != MVT::i64 ||
1514 NewC.CCMask == SystemZ::CCMASK_CMP_EQ ||
1515 NewC.CCMask == SystemZ::CCMASK_CMP_NE ||
1516 NewC.ICmpType == SystemZICMP::SignedOnly)
1517 return;
1518 // Convert LE and GT comparisons into LT and GE.
1519 if (NewC.CCMask == SystemZ::CCMASK_CMP_LE ||
1520 NewC.CCMask == SystemZ::CCMASK_CMP_GT) {
1521 if (CmpVal == uint64_t(-1))
1522 return;
1523 CmpVal += 1;
1524 NewC.CCMask ^= SystemZ::CCMASK_CMP_EQ;
1525 }
1526 // If the low N bits of Op1 are zero than the low N bits of Op0 can
1527 // be masked off without changing the result.
1528 MaskVal = -(CmpVal & -CmpVal);
1529 NewC.ICmpType = SystemZICMP::UnsignedOnly;
1530 }
Richard Sandiford35b9be22013-08-28 10:31:43 +00001531
Richard Sandiford113c8702013-09-03 15:38:35 +00001532 // Check whether the combination of mask, comparison value and comparison
1533 // type are suitable.
Richard Sandifordc3dc4472013-12-13 15:46:55 +00001534 unsigned BitSize = NewC.Op0.getValueType().getSizeInBits();
Richard Sandiford030c1652013-09-13 09:09:50 +00001535 unsigned NewCCMask, ShiftVal;
Richard Sandifordc3dc4472013-12-13 15:46:55 +00001536 if (NewC.ICmpType != SystemZICMP::SignedOnly &&
1537 NewC.Op0.getOpcode() == ISD::SHL &&
1538 isSimpleShift(NewC.Op0, ShiftVal) &&
1539 (NewCCMask = getTestUnderMaskCond(BitSize, NewC.CCMask,
1540 MaskVal >> ShiftVal,
Richard Sandiford030c1652013-09-13 09:09:50 +00001541 CmpVal >> ShiftVal,
1542 SystemZICMP::Any))) {
Richard Sandifordc3dc4472013-12-13 15:46:55 +00001543 NewC.Op0 = NewC.Op0.getOperand(0);
1544 MaskVal >>= ShiftVal;
1545 } else if (NewC.ICmpType != SystemZICMP::SignedOnly &&
1546 NewC.Op0.getOpcode() == ISD::SRL &&
1547 isSimpleShift(NewC.Op0, ShiftVal) &&
1548 (NewCCMask = getTestUnderMaskCond(BitSize, NewC.CCMask,
Richard Sandiford030c1652013-09-13 09:09:50 +00001549 MaskVal << ShiftVal,
1550 CmpVal << ShiftVal,
1551 SystemZICMP::UnsignedOnly))) {
Richard Sandifordc3dc4472013-12-13 15:46:55 +00001552 NewC.Op0 = NewC.Op0.getOperand(0);
1553 MaskVal <<= ShiftVal;
Richard Sandiford030c1652013-09-13 09:09:50 +00001554 } else {
Richard Sandifordc3dc4472013-12-13 15:46:55 +00001555 NewCCMask = getTestUnderMaskCond(BitSize, NewC.CCMask, MaskVal, CmpVal,
1556 NewC.ICmpType);
Richard Sandiford030c1652013-09-13 09:09:50 +00001557 if (!NewCCMask)
1558 return;
1559 }
Richard Sandiford113c8702013-09-03 15:38:35 +00001560
Richard Sandiford35b9be22013-08-28 10:31:43 +00001561 // Go ahead and make the change.
Richard Sandifordd420f732013-12-13 15:28:45 +00001562 C.Opcode = SystemZISD::TM;
Richard Sandifordc3dc4472013-12-13 15:46:55 +00001563 C.Op0 = NewC.Op0;
1564 if (Mask && Mask->getZExtValue() == MaskVal)
1565 C.Op1 = SDValue(Mask, 0);
1566 else
1567 C.Op1 = DAG.getConstant(MaskVal, C.Op0.getValueType());
Richard Sandifordd420f732013-12-13 15:28:45 +00001568 C.CCValid = SystemZ::CCMASK_TM;
1569 C.CCMask = NewCCMask;
Richard Sandiford35b9be22013-08-28 10:31:43 +00001570}
1571
Richard Sandifordd420f732013-12-13 15:28:45 +00001572// Decide how to implement a comparison of type Cond between CmpOp0 with CmpOp1.
1573static Comparison getCmp(SelectionDAG &DAG, SDValue CmpOp0, SDValue CmpOp1,
1574 ISD::CondCode Cond) {
1575 Comparison C(CmpOp0, CmpOp1);
1576 C.CCMask = CCMaskForCondCode(Cond);
1577 if (C.Op0.getValueType().isFloatingPoint()) {
1578 C.CCValid = SystemZ::CCMASK_FCMP;
1579 C.Opcode = SystemZISD::FCMP;
Richard Sandiford83a0b6a2013-12-20 11:56:02 +00001580 adjustForFNeg(C);
Richard Sandiford5bc670b2013-09-06 11:51:39 +00001581 } else {
Richard Sandifordd420f732013-12-13 15:28:45 +00001582 C.CCValid = SystemZ::CCMASK_ICMP;
1583 C.Opcode = SystemZISD::ICMP;
Richard Sandiford5bc670b2013-09-06 11:51:39 +00001584 // Choose the type of comparison. Equality and inequality tests can
1585 // use either signed or unsigned comparisons. The choice also doesn't
1586 // matter if both sign bits are known to be clear. In those cases we
1587 // want to give the main isel code the freedom to choose whichever
1588 // form fits best.
Richard Sandifordd420f732013-12-13 15:28:45 +00001589 if (C.CCMask == SystemZ::CCMASK_CMP_EQ ||
1590 C.CCMask == SystemZ::CCMASK_CMP_NE ||
1591 (DAG.SignBitIsZero(C.Op0) && DAG.SignBitIsZero(C.Op1)))
1592 C.ICmpType = SystemZICMP::Any;
1593 else if (C.CCMask & SystemZ::CCMASK_CMP_UO)
1594 C.ICmpType = SystemZICMP::UnsignedOnly;
Richard Sandiford5bc670b2013-09-06 11:51:39 +00001595 else
Richard Sandifordd420f732013-12-13 15:28:45 +00001596 C.ICmpType = SystemZICMP::SignedOnly;
1597 C.CCMask &= ~SystemZ::CCMASK_CMP_UO;
1598 adjustZeroCmp(DAG, C);
1599 adjustSubwordCmp(DAG, C);
Richard Sandiford0847c452013-12-13 15:50:30 +00001600 adjustForSubtraction(DAG, C);
Richard Sandiford83a0b6a2013-12-20 11:56:02 +00001601 adjustForLTGFR(C);
1602 adjustICmpTruncate(DAG, C);
Ulrich Weigand5f613df2013-05-06 16:15:19 +00001603 }
1604
Richard Sandifordd420f732013-12-13 15:28:45 +00001605 if (shouldSwapCmpOperands(C)) {
1606 std::swap(C.Op0, C.Op1);
1607 C.CCMask = reverseCCMask(C.CCMask);
Richard Sandiford24e597b2013-08-23 11:27:19 +00001608 }
1609
Richard Sandifordd420f732013-12-13 15:28:45 +00001610 adjustForTestUnderMask(DAG, C);
Richard Sandifordd420f732013-12-13 15:28:45 +00001611 return C;
1612}
1613
1614// Emit the comparison instruction described by C.
1615static SDValue emitCmp(SelectionDAG &DAG, SDLoc DL, Comparison &C) {
1616 if (C.Opcode == SystemZISD::ICMP)
1617 return DAG.getNode(SystemZISD::ICMP, DL, MVT::Glue, C.Op0, C.Op1,
1618 DAG.getConstant(C.ICmpType, MVT::i32));
1619 if (C.Opcode == SystemZISD::TM) {
1620 bool RegisterOnly = (bool(C.CCMask & SystemZ::CCMASK_TM_MIXED_MSB_0) !=
1621 bool(C.CCMask & SystemZ::CCMASK_TM_MIXED_MSB_1));
1622 return DAG.getNode(SystemZISD::TM, DL, MVT::Glue, C.Op0, C.Op1,
1623 DAG.getConstant(RegisterOnly, MVT::i32));
1624 }
1625 return DAG.getNode(C.Opcode, DL, MVT::Glue, C.Op0, C.Op1);
Ulrich Weigand5f613df2013-05-06 16:15:19 +00001626}
1627
Richard Sandiford7d86e472013-08-21 09:34:56 +00001628// Implement a 32-bit *MUL_LOHI operation by extending both operands to
1629// 64 bits. Extend is the extension type to use. Store the high part
1630// in Hi and the low part in Lo.
1631static void lowerMUL_LOHI32(SelectionDAG &DAG, SDLoc DL,
1632 unsigned Extend, SDValue Op0, SDValue Op1,
1633 SDValue &Hi, SDValue &Lo) {
1634 Op0 = DAG.getNode(Extend, DL, MVT::i64, Op0);
1635 Op1 = DAG.getNode(Extend, DL, MVT::i64, Op1);
1636 SDValue Mul = DAG.getNode(ISD::MUL, DL, MVT::i64, Op0, Op1);
1637 Hi = DAG.getNode(ISD::SRL, DL, MVT::i64, Mul, DAG.getConstant(32, MVT::i64));
1638 Hi = DAG.getNode(ISD::TRUNCATE, DL, MVT::i32, Hi);
1639 Lo = DAG.getNode(ISD::TRUNCATE, DL, MVT::i32, Mul);
1640}
1641
Ulrich Weigand5f613df2013-05-06 16:15:19 +00001642// Lower a binary operation that produces two VT results, one in each
1643// half of a GR128 pair. Op0 and Op1 are the VT operands to the operation,
1644// Extend extends Op0 to a GR128, and Opcode performs the GR128 operation
1645// on the extended Op0 and (unextended) Op1. Store the even register result
1646// in Even and the odd register result in Odd.
Andrew Trickef9de2a2013-05-25 02:42:55 +00001647static void lowerGR128Binary(SelectionDAG &DAG, SDLoc DL, EVT VT,
Ulrich Weigand5f613df2013-05-06 16:15:19 +00001648 unsigned Extend, unsigned Opcode,
1649 SDValue Op0, SDValue Op1,
1650 SDValue &Even, SDValue &Odd) {
1651 SDNode *In128 = DAG.getMachineNode(Extend, DL, MVT::Untyped, Op0);
1652 SDValue Result = DAG.getNode(Opcode, DL, MVT::Untyped,
1653 SDValue(In128, 0), Op1);
1654 bool Is32Bit = is32Bit(VT);
Richard Sandifordd8163202013-09-13 09:12:44 +00001655 Even = DAG.getTargetExtractSubreg(SystemZ::even128(Is32Bit), DL, VT, Result);
1656 Odd = DAG.getTargetExtractSubreg(SystemZ::odd128(Is32Bit), DL, VT, Result);
Ulrich Weigand5f613df2013-05-06 16:15:19 +00001657}
1658
Richard Sandiford48ef6ab2013-12-06 09:53:09 +00001659// Return an i32 value that is 1 if the CC value produced by Glue is
1660// in the mask CCMask and 0 otherwise. CC is known to have a value
1661// in CCValid, so other values can be ignored.
1662static SDValue emitSETCC(SelectionDAG &DAG, SDLoc DL, SDValue Glue,
1663 unsigned CCValid, unsigned CCMask) {
Richard Sandifordf722a8e302013-10-16 11:10:55 +00001664 IPMConversion Conversion = getIPMConversion(CCValid, CCMask);
1665 SDValue Result = DAG.getNode(SystemZISD::IPM, DL, MVT::i32, Glue);
1666
1667 if (Conversion.XORValue)
1668 Result = DAG.getNode(ISD::XOR, DL, MVT::i32, Result,
1669 DAG.getConstant(Conversion.XORValue, MVT::i32));
1670
1671 if (Conversion.AddValue)
1672 Result = DAG.getNode(ISD::ADD, DL, MVT::i32, Result,
1673 DAG.getConstant(Conversion.AddValue, MVT::i32));
1674
1675 // The SHR/AND sequence should get optimized to an RISBG.
1676 Result = DAG.getNode(ISD::SRL, DL, MVT::i32, Result,
1677 DAG.getConstant(Conversion.Bit, MVT::i32));
1678 if (Conversion.Bit != 31)
1679 Result = DAG.getNode(ISD::AND, DL, MVT::i32, Result,
1680 DAG.getConstant(1, MVT::i32));
1681 return Result;
1682}
1683
Richard Sandiford48ef6ab2013-12-06 09:53:09 +00001684SDValue SystemZTargetLowering::lowerSETCC(SDValue Op,
1685 SelectionDAG &DAG) const {
1686 SDValue CmpOp0 = Op.getOperand(0);
1687 SDValue CmpOp1 = Op.getOperand(1);
1688 ISD::CondCode CC = cast<CondCodeSDNode>(Op.getOperand(2))->get();
1689 SDLoc DL(Op);
1690
Richard Sandifordd420f732013-12-13 15:28:45 +00001691 Comparison C(getCmp(DAG, CmpOp0, CmpOp1, CC));
1692 SDValue Glue = emitCmp(DAG, DL, C);
1693 return emitSETCC(DAG, DL, Glue, C.CCValid, C.CCMask);
Richard Sandiford48ef6ab2013-12-06 09:53:09 +00001694}
1695
Ulrich Weigand5f613df2013-05-06 16:15:19 +00001696SDValue SystemZTargetLowering::lowerBR_CC(SDValue Op, SelectionDAG &DAG) const {
1697 SDValue Chain = Op.getOperand(0);
1698 ISD::CondCode CC = cast<CondCodeSDNode>(Op.getOperand(1))->get();
1699 SDValue CmpOp0 = Op.getOperand(2);
1700 SDValue CmpOp1 = Op.getOperand(3);
1701 SDValue Dest = Op.getOperand(4);
Andrew Trickef9de2a2013-05-25 02:42:55 +00001702 SDLoc DL(Op);
Ulrich Weigand5f613df2013-05-06 16:15:19 +00001703
Richard Sandifordd420f732013-12-13 15:28:45 +00001704 Comparison C(getCmp(DAG, CmpOp0, CmpOp1, CC));
1705 SDValue Glue = emitCmp(DAG, DL, C);
Ulrich Weigand5f613df2013-05-06 16:15:19 +00001706 return DAG.getNode(SystemZISD::BR_CCMASK, DL, Op.getValueType(),
Richard Sandifordd420f732013-12-13 15:28:45 +00001707 Chain, DAG.getConstant(C.CCValid, MVT::i32),
1708 DAG.getConstant(C.CCMask, MVT::i32), Dest, Glue);
Ulrich Weigand5f613df2013-05-06 16:15:19 +00001709}
1710
Richard Sandiford57485472013-12-13 15:35:00 +00001711// Return true if Pos is CmpOp and Neg is the negative of CmpOp,
1712// allowing Pos and Neg to be wider than CmpOp.
1713static bool isAbsolute(SDValue CmpOp, SDValue Pos, SDValue Neg) {
1714 return (Neg.getOpcode() == ISD::SUB &&
1715 Neg.getOperand(0).getOpcode() == ISD::Constant &&
1716 cast<ConstantSDNode>(Neg.getOperand(0))->getZExtValue() == 0 &&
1717 Neg.getOperand(1) == Pos &&
1718 (Pos == CmpOp ||
1719 (Pos.getOpcode() == ISD::SIGN_EXTEND &&
1720 Pos.getOperand(0) == CmpOp)));
1721}
1722
1723// Return the absolute or negative absolute of Op; IsNegative decides which.
1724static SDValue getAbsolute(SelectionDAG &DAG, SDLoc DL, SDValue Op,
1725 bool IsNegative) {
1726 Op = DAG.getNode(SystemZISD::IABS, DL, Op.getValueType(), Op);
1727 if (IsNegative)
1728 Op = DAG.getNode(ISD::SUB, DL, Op.getValueType(),
1729 DAG.getConstant(0, Op.getValueType()), Op);
1730 return Op;
1731}
1732
Ulrich Weigand5f613df2013-05-06 16:15:19 +00001733SDValue SystemZTargetLowering::lowerSELECT_CC(SDValue Op,
1734 SelectionDAG &DAG) const {
1735 SDValue CmpOp0 = Op.getOperand(0);
1736 SDValue CmpOp1 = Op.getOperand(1);
1737 SDValue TrueOp = Op.getOperand(2);
1738 SDValue FalseOp = Op.getOperand(3);
1739 ISD::CondCode CC = cast<CondCodeSDNode>(Op.getOperand(4))->get();
Andrew Trickef9de2a2013-05-25 02:42:55 +00001740 SDLoc DL(Op);
Ulrich Weigand5f613df2013-05-06 16:15:19 +00001741
Richard Sandifordd420f732013-12-13 15:28:45 +00001742 Comparison C(getCmp(DAG, CmpOp0, CmpOp1, CC));
Richard Sandiford57485472013-12-13 15:35:00 +00001743
1744 // Check for absolute and negative-absolute selections, including those
1745 // where the comparison value is sign-extended (for LPGFR and LNGFR).
1746 // This check supplements the one in DAGCombiner.
1747 if (C.Opcode == SystemZISD::ICMP &&
1748 C.CCMask != SystemZ::CCMASK_CMP_EQ &&
1749 C.CCMask != SystemZ::CCMASK_CMP_NE &&
1750 C.Op1.getOpcode() == ISD::Constant &&
1751 cast<ConstantSDNode>(C.Op1)->getZExtValue() == 0) {
1752 if (isAbsolute(C.Op0, TrueOp, FalseOp))
1753 return getAbsolute(DAG, DL, TrueOp, C.CCMask & SystemZ::CCMASK_CMP_LT);
1754 if (isAbsolute(C.Op0, FalseOp, TrueOp))
1755 return getAbsolute(DAG, DL, FalseOp, C.CCMask & SystemZ::CCMASK_CMP_GT);
1756 }
1757
Richard Sandifordd420f732013-12-13 15:28:45 +00001758 SDValue Glue = emitCmp(DAG, DL, C);
Richard Sandiford48ef6ab2013-12-06 09:53:09 +00001759
1760 // Special case for handling -1/0 results. The shifts we use here
1761 // should get optimized with the IPM conversion sequence.
Richard Sandiford21f5d682014-03-06 11:22:58 +00001762 auto *TrueC = dyn_cast<ConstantSDNode>(TrueOp);
1763 auto *FalseC = dyn_cast<ConstantSDNode>(FalseOp);
Richard Sandiford48ef6ab2013-12-06 09:53:09 +00001764 if (TrueC && FalseC) {
1765 int64_t TrueVal = TrueC->getSExtValue();
1766 int64_t FalseVal = FalseC->getSExtValue();
1767 if ((TrueVal == -1 && FalseVal == 0) || (TrueVal == 0 && FalseVal == -1)) {
1768 // Invert the condition if we want -1 on false.
1769 if (TrueVal == 0)
Richard Sandifordd420f732013-12-13 15:28:45 +00001770 C.CCMask ^= C.CCValid;
1771 SDValue Result = emitSETCC(DAG, DL, Glue, C.CCValid, C.CCMask);
Richard Sandiford48ef6ab2013-12-06 09:53:09 +00001772 EVT VT = Op.getValueType();
1773 // Extend the result to VT. Upper bits are ignored.
1774 if (!is32Bit(VT))
1775 Result = DAG.getNode(ISD::ANY_EXTEND, DL, VT, Result);
1776 // Sign-extend from the low bit.
1777 SDValue ShAmt = DAG.getConstant(VT.getSizeInBits() - 1, MVT::i32);
1778 SDValue Shl = DAG.getNode(ISD::SHL, DL, VT, Result, ShAmt);
1779 return DAG.getNode(ISD::SRA, DL, VT, Shl, ShAmt);
1780 }
1781 }
Ulrich Weigand5f613df2013-05-06 16:15:19 +00001782
Richard Sandiford3d768e32013-07-31 12:30:20 +00001783 SmallVector<SDValue, 5> Ops;
Ulrich Weigand5f613df2013-05-06 16:15:19 +00001784 Ops.push_back(TrueOp);
1785 Ops.push_back(FalseOp);
Richard Sandifordd420f732013-12-13 15:28:45 +00001786 Ops.push_back(DAG.getConstant(C.CCValid, MVT::i32));
1787 Ops.push_back(DAG.getConstant(C.CCMask, MVT::i32));
Richard Sandiford48ef6ab2013-12-06 09:53:09 +00001788 Ops.push_back(Glue);
Ulrich Weigand5f613df2013-05-06 16:15:19 +00001789
1790 SDVTList VTs = DAG.getVTList(Op.getValueType(), MVT::Glue);
Craig Topper48d114b2014-04-26 18:35:24 +00001791 return DAG.getNode(SystemZISD::SELECT_CCMASK, DL, VTs, Ops);
Ulrich Weigand5f613df2013-05-06 16:15:19 +00001792}
1793
1794SDValue SystemZTargetLowering::lowerGlobalAddress(GlobalAddressSDNode *Node,
1795 SelectionDAG &DAG) const {
Andrew Trickef9de2a2013-05-25 02:42:55 +00001796 SDLoc DL(Node);
Ulrich Weigand5f613df2013-05-06 16:15:19 +00001797 const GlobalValue *GV = Node->getGlobal();
1798 int64_t Offset = Node->getOffset();
1799 EVT PtrVT = getPointerTy();
Eric Christopher93bf97c2014-06-27 07:38:01 +00001800 Reloc::Model RM = DAG.getTarget().getRelocationModel();
1801 CodeModel::Model CM = DAG.getTarget().getCodeModel();
Ulrich Weigand5f613df2013-05-06 16:15:19 +00001802
1803 SDValue Result;
1804 if (Subtarget.isPC32DBLSymbol(GV, RM, CM)) {
Richard Sandiford54b36912013-09-27 15:14:04 +00001805 // Assign anchors at 1<<12 byte boundaries.
1806 uint64_t Anchor = Offset & ~uint64_t(0xfff);
1807 Result = DAG.getTargetGlobalAddress(GV, DL, PtrVT, Anchor);
1808 Result = DAG.getNode(SystemZISD::PCREL_WRAPPER, DL, PtrVT, Result);
1809
1810 // The offset can be folded into the address if it is aligned to a halfword.
1811 Offset -= Anchor;
1812 if (Offset != 0 && (Offset & 1) == 0) {
1813 SDValue Full = DAG.getTargetGlobalAddress(GV, DL, PtrVT, Anchor + Offset);
1814 Result = DAG.getNode(SystemZISD::PCREL_OFFSET, DL, PtrVT, Full, Result);
Ulrich Weigand5f613df2013-05-06 16:15:19 +00001815 Offset = 0;
1816 }
Ulrich Weigand5f613df2013-05-06 16:15:19 +00001817 } else {
1818 Result = DAG.getTargetGlobalAddress(GV, DL, PtrVT, 0, SystemZII::MO_GOT);
1819 Result = DAG.getNode(SystemZISD::PCREL_WRAPPER, DL, PtrVT, Result);
1820 Result = DAG.getLoad(PtrVT, DL, DAG.getEntryNode(), Result,
1821 MachinePointerInfo::getGOT(), false, false, false, 0);
1822 }
1823
1824 // If there was a non-zero offset that we didn't fold, create an explicit
1825 // addition for it.
1826 if (Offset != 0)
1827 Result = DAG.getNode(ISD::ADD, DL, PtrVT, Result,
1828 DAG.getConstant(Offset, PtrVT));
1829
1830 return Result;
1831}
1832
Ulrich Weigand7db69182015-02-18 09:13:27 +00001833SDValue SystemZTargetLowering::lowerTLSGetOffset(GlobalAddressSDNode *Node,
1834 SelectionDAG &DAG,
1835 unsigned Opcode,
1836 SDValue GOTOffset) const {
1837 SDLoc DL(Node);
1838 EVT PtrVT = getPointerTy();
1839 SDValue Chain = DAG.getEntryNode();
1840 SDValue Glue;
1841
1842 // __tls_get_offset takes the GOT offset in %r2 and the GOT in %r12.
1843 SDValue GOT = DAG.getGLOBAL_OFFSET_TABLE(PtrVT);
1844 Chain = DAG.getCopyToReg(Chain, DL, SystemZ::R12D, GOT, Glue);
1845 Glue = Chain.getValue(1);
1846 Chain = DAG.getCopyToReg(Chain, DL, SystemZ::R2D, GOTOffset, Glue);
1847 Glue = Chain.getValue(1);
1848
1849 // The first call operand is the chain and the second is the TLS symbol.
1850 SmallVector<SDValue, 8> Ops;
1851 Ops.push_back(Chain);
1852 Ops.push_back(DAG.getTargetGlobalAddress(Node->getGlobal(), DL,
1853 Node->getValueType(0),
1854 0, 0));
1855
1856 // Add argument registers to the end of the list so that they are
1857 // known live into the call.
1858 Ops.push_back(DAG.getRegister(SystemZ::R2D, PtrVT));
1859 Ops.push_back(DAG.getRegister(SystemZ::R12D, PtrVT));
1860
1861 // Add a register mask operand representing the call-preserved registers.
1862 const TargetRegisterInfo *TRI = Subtarget.getRegisterInfo();
1863 const uint32_t *Mask = TRI->getCallPreservedMask(CallingConv::C);
1864 assert(Mask && "Missing call preserved mask for calling convention");
1865 Ops.push_back(DAG.getRegisterMask(Mask));
1866
1867 // Glue the call to the argument copies.
1868 Ops.push_back(Glue);
1869
1870 // Emit the call.
1871 SDVTList NodeTys = DAG.getVTList(MVT::Other, MVT::Glue);
1872 Chain = DAG.getNode(Opcode, DL, NodeTys, Ops);
1873 Glue = Chain.getValue(1);
1874
1875 // Copy the return value from %r2.
1876 return DAG.getCopyFromReg(Chain, DL, SystemZ::R2D, PtrVT, Glue);
1877}
1878
Ulrich Weigand5f613df2013-05-06 16:15:19 +00001879SDValue SystemZTargetLowering::lowerGlobalTLSAddress(GlobalAddressSDNode *Node,
1880 SelectionDAG &DAG) const {
Andrew Trickef9de2a2013-05-25 02:42:55 +00001881 SDLoc DL(Node);
Ulrich Weigand5f613df2013-05-06 16:15:19 +00001882 const GlobalValue *GV = Node->getGlobal();
1883 EVT PtrVT = getPointerTy();
Eric Christopher93bf97c2014-06-27 07:38:01 +00001884 TLSModel::Model model = DAG.getTarget().getTLSModel(GV);
Ulrich Weigand5f613df2013-05-06 16:15:19 +00001885
Ulrich Weigand5f613df2013-05-06 16:15:19 +00001886 // The high part of the thread pointer is in access register 0.
1887 SDValue TPHi = DAG.getNode(SystemZISD::EXTRACT_ACCESS, DL, MVT::i32,
1888 DAG.getConstant(0, MVT::i32));
1889 TPHi = DAG.getNode(ISD::ANY_EXTEND, DL, PtrVT, TPHi);
1890
1891 // The low part of the thread pointer is in access register 1.
1892 SDValue TPLo = DAG.getNode(SystemZISD::EXTRACT_ACCESS, DL, MVT::i32,
1893 DAG.getConstant(1, MVT::i32));
1894 TPLo = DAG.getNode(ISD::ZERO_EXTEND, DL, PtrVT, TPLo);
1895
1896 // Merge them into a single 64-bit address.
1897 SDValue TPHiShifted = DAG.getNode(ISD::SHL, DL, PtrVT, TPHi,
1898 DAG.getConstant(32, PtrVT));
1899 SDValue TP = DAG.getNode(ISD::OR, DL, PtrVT, TPHiShifted, TPLo);
1900
Ulrich Weigand7db69182015-02-18 09:13:27 +00001901 // Get the offset of GA from the thread pointer, based on the TLS model.
1902 SDValue Offset;
1903 switch (model) {
1904 case TLSModel::GeneralDynamic: {
1905 // Load the GOT offset of the tls_index (module ID / per-symbol offset).
1906 SystemZConstantPoolValue *CPV =
1907 SystemZConstantPoolValue::Create(GV, SystemZCP::TLSGD);
Ulrich Weigand5f613df2013-05-06 16:15:19 +00001908
Ulrich Weigand7db69182015-02-18 09:13:27 +00001909 Offset = DAG.getConstantPool(CPV, PtrVT, 8);
1910 Offset = DAG.getLoad(PtrVT, DL, DAG.getEntryNode(),
1911 Offset, MachinePointerInfo::getConstantPool(),
1912 false, false, false, 0);
1913
1914 // Call __tls_get_offset to retrieve the offset.
1915 Offset = lowerTLSGetOffset(Node, DAG, SystemZISD::TLS_GDCALL, Offset);
1916 break;
1917 }
1918
1919 case TLSModel::LocalDynamic: {
1920 // Load the GOT offset of the module ID.
1921 SystemZConstantPoolValue *CPV =
1922 SystemZConstantPoolValue::Create(GV, SystemZCP::TLSLDM);
1923
1924 Offset = DAG.getConstantPool(CPV, PtrVT, 8);
1925 Offset = DAG.getLoad(PtrVT, DL, DAG.getEntryNode(),
1926 Offset, MachinePointerInfo::getConstantPool(),
1927 false, false, false, 0);
1928
1929 // Call __tls_get_offset to retrieve the module base offset.
1930 Offset = lowerTLSGetOffset(Node, DAG, SystemZISD::TLS_LDCALL, Offset);
1931
1932 // Note: The SystemZLDCleanupPass will remove redundant computations
1933 // of the module base offset. Count total number of local-dynamic
1934 // accesses to trigger execution of that pass.
1935 SystemZMachineFunctionInfo* MFI =
1936 DAG.getMachineFunction().getInfo<SystemZMachineFunctionInfo>();
1937 MFI->incNumLocalDynamicTLSAccesses();
1938
1939 // Add the per-symbol offset.
1940 CPV = SystemZConstantPoolValue::Create(GV, SystemZCP::DTPOFF);
1941
1942 SDValue DTPOffset = DAG.getConstantPool(CPV, PtrVT, 8);
1943 DTPOffset = DAG.getLoad(PtrVT, DL, DAG.getEntryNode(),
1944 DTPOffset, MachinePointerInfo::getConstantPool(),
1945 false, false, false, 0);
1946
1947 Offset = DAG.getNode(ISD::ADD, DL, PtrVT, Offset, DTPOffset);
1948 break;
1949 }
1950
1951 case TLSModel::InitialExec: {
1952 // Load the offset from the GOT.
1953 Offset = DAG.getTargetGlobalAddress(GV, DL, PtrVT, 0,
1954 SystemZII::MO_INDNTPOFF);
1955 Offset = DAG.getNode(SystemZISD::PCREL_WRAPPER, DL, PtrVT, Offset);
1956 Offset = DAG.getLoad(PtrVT, DL, DAG.getEntryNode(),
1957 Offset, MachinePointerInfo::getGOT(),
1958 false, false, false, 0);
1959 break;
1960 }
1961
1962 case TLSModel::LocalExec: {
1963 // Force the offset into the constant pool and load it from there.
1964 SystemZConstantPoolValue *CPV =
1965 SystemZConstantPoolValue::Create(GV, SystemZCP::NTPOFF);
1966
1967 Offset = DAG.getConstantPool(CPV, PtrVT, 8);
1968 Offset = DAG.getLoad(PtrVT, DL, DAG.getEntryNode(),
1969 Offset, MachinePointerInfo::getConstantPool(),
1970 false, false, false, 0);
1971 break;
Ulrich Weigandb7e59092015-02-18 09:42:23 +00001972 }
Ulrich Weigand7db69182015-02-18 09:13:27 +00001973 }
Ulrich Weigand5f613df2013-05-06 16:15:19 +00001974
1975 // Add the base and offset together.
1976 return DAG.getNode(ISD::ADD, DL, PtrVT, TP, Offset);
1977}
1978
1979SDValue SystemZTargetLowering::lowerBlockAddress(BlockAddressSDNode *Node,
1980 SelectionDAG &DAG) const {
Andrew Trickef9de2a2013-05-25 02:42:55 +00001981 SDLoc DL(Node);
Ulrich Weigand5f613df2013-05-06 16:15:19 +00001982 const BlockAddress *BA = Node->getBlockAddress();
1983 int64_t Offset = Node->getOffset();
1984 EVT PtrVT = getPointerTy();
1985
1986 SDValue Result = DAG.getTargetBlockAddress(BA, PtrVT, Offset);
1987 Result = DAG.getNode(SystemZISD::PCREL_WRAPPER, DL, PtrVT, Result);
1988 return Result;
1989}
1990
1991SDValue SystemZTargetLowering::lowerJumpTable(JumpTableSDNode *JT,
1992 SelectionDAG &DAG) const {
Andrew Trickef9de2a2013-05-25 02:42:55 +00001993 SDLoc DL(JT);
Ulrich Weigand5f613df2013-05-06 16:15:19 +00001994 EVT PtrVT = getPointerTy();
1995 SDValue Result = DAG.getTargetJumpTable(JT->getIndex(), PtrVT);
1996
1997 // Use LARL to load the address of the table.
1998 return DAG.getNode(SystemZISD::PCREL_WRAPPER, DL, PtrVT, Result);
1999}
2000
2001SDValue SystemZTargetLowering::lowerConstantPool(ConstantPoolSDNode *CP,
2002 SelectionDAG &DAG) const {
Andrew Trickef9de2a2013-05-25 02:42:55 +00002003 SDLoc DL(CP);
Ulrich Weigand5f613df2013-05-06 16:15:19 +00002004 EVT PtrVT = getPointerTy();
2005
2006 SDValue Result;
2007 if (CP->isMachineConstantPoolEntry())
2008 Result = DAG.getTargetConstantPool(CP->getMachineCPVal(), PtrVT,
2009 CP->getAlignment());
2010 else
2011 Result = DAG.getTargetConstantPool(CP->getConstVal(), PtrVT,
2012 CP->getAlignment(), CP->getOffset());
2013
2014 // Use LARL to load the address of the constant pool entry.
2015 return DAG.getNode(SystemZISD::PCREL_WRAPPER, DL, PtrVT, Result);
2016}
2017
2018SDValue SystemZTargetLowering::lowerBITCAST(SDValue Op,
2019 SelectionDAG &DAG) const {
Andrew Trickef9de2a2013-05-25 02:42:55 +00002020 SDLoc DL(Op);
Ulrich Weigand5f613df2013-05-06 16:15:19 +00002021 SDValue In = Op.getOperand(0);
2022 EVT InVT = In.getValueType();
2023 EVT ResVT = Op.getValueType();
2024
Ulrich Weigand5f613df2013-05-06 16:15:19 +00002025 if (InVT == MVT::i32 && ResVT == MVT::f32) {
Richard Sandifordf6377fb2013-10-01 14:31:11 +00002026 SDValue In64;
2027 if (Subtarget.hasHighWord()) {
2028 SDNode *U64 = DAG.getMachineNode(TargetOpcode::IMPLICIT_DEF, DL,
2029 MVT::i64);
2030 In64 = DAG.getTargetInsertSubreg(SystemZ::subreg_h32, DL,
2031 MVT::i64, SDValue(U64, 0), In);
2032 } else {
2033 In64 = DAG.getNode(ISD::ANY_EXTEND, DL, MVT::i64, In);
2034 In64 = DAG.getNode(ISD::SHL, DL, MVT::i64, In64,
2035 DAG.getConstant(32, MVT::i64));
2036 }
2037 SDValue Out64 = DAG.getNode(ISD::BITCAST, DL, MVT::f64, In64);
Richard Sandiford87a44362013-09-30 10:28:35 +00002038 return DAG.getTargetExtractSubreg(SystemZ::subreg_h32,
Richard Sandifordd8163202013-09-13 09:12:44 +00002039 DL, MVT::f32, Out64);
Ulrich Weigand5f613df2013-05-06 16:15:19 +00002040 }
2041 if (InVT == MVT::f32 && ResVT == MVT::i32) {
2042 SDNode *U64 = DAG.getMachineNode(TargetOpcode::IMPLICIT_DEF, DL, MVT::f64);
Richard Sandiford87a44362013-09-30 10:28:35 +00002043 SDValue In64 = DAG.getTargetInsertSubreg(SystemZ::subreg_h32, DL,
Richard Sandifordd8163202013-09-13 09:12:44 +00002044 MVT::f64, SDValue(U64, 0), In);
2045 SDValue Out64 = DAG.getNode(ISD::BITCAST, DL, MVT::i64, In64);
Richard Sandifordf6377fb2013-10-01 14:31:11 +00002046 if (Subtarget.hasHighWord())
2047 return DAG.getTargetExtractSubreg(SystemZ::subreg_h32, DL,
2048 MVT::i32, Out64);
2049 SDValue Shift = DAG.getNode(ISD::SRL, DL, MVT::i64, Out64,
2050 DAG.getConstant(32, MVT::i64));
2051 return DAG.getNode(ISD::TRUNCATE, DL, MVT::i32, Shift);
Ulrich Weigand5f613df2013-05-06 16:15:19 +00002052 }
2053 llvm_unreachable("Unexpected bitcast combination");
2054}
2055
2056SDValue SystemZTargetLowering::lowerVASTART(SDValue Op,
2057 SelectionDAG &DAG) const {
2058 MachineFunction &MF = DAG.getMachineFunction();
2059 SystemZMachineFunctionInfo *FuncInfo =
2060 MF.getInfo<SystemZMachineFunctionInfo>();
2061 EVT PtrVT = getPointerTy();
2062
2063 SDValue Chain = Op.getOperand(0);
2064 SDValue Addr = Op.getOperand(1);
2065 const Value *SV = cast<SrcValueSDNode>(Op.getOperand(2))->getValue();
Andrew Trickef9de2a2013-05-25 02:42:55 +00002066 SDLoc DL(Op);
Ulrich Weigand5f613df2013-05-06 16:15:19 +00002067
2068 // The initial values of each field.
2069 const unsigned NumFields = 4;
2070 SDValue Fields[NumFields] = {
2071 DAG.getConstant(FuncInfo->getVarArgsFirstGPR(), PtrVT),
2072 DAG.getConstant(FuncInfo->getVarArgsFirstFPR(), PtrVT),
2073 DAG.getFrameIndex(FuncInfo->getVarArgsFrameIndex(), PtrVT),
2074 DAG.getFrameIndex(FuncInfo->getRegSaveFrameIndex(), PtrVT)
2075 };
2076
2077 // Store each field into its respective slot.
2078 SDValue MemOps[NumFields];
2079 unsigned Offset = 0;
2080 for (unsigned I = 0; I < NumFields; ++I) {
2081 SDValue FieldAddr = Addr;
2082 if (Offset != 0)
2083 FieldAddr = DAG.getNode(ISD::ADD, DL, PtrVT, FieldAddr,
2084 DAG.getIntPtrConstant(Offset));
2085 MemOps[I] = DAG.getStore(Chain, DL, Fields[I], FieldAddr,
2086 MachinePointerInfo(SV, Offset),
2087 false, false, 0);
2088 Offset += 8;
2089 }
Craig Topper48d114b2014-04-26 18:35:24 +00002090 return DAG.getNode(ISD::TokenFactor, DL, MVT::Other, MemOps);
Ulrich Weigand5f613df2013-05-06 16:15:19 +00002091}
2092
2093SDValue SystemZTargetLowering::lowerVACOPY(SDValue Op,
2094 SelectionDAG &DAG) const {
2095 SDValue Chain = Op.getOperand(0);
2096 SDValue DstPtr = Op.getOperand(1);
2097 SDValue SrcPtr = Op.getOperand(2);
2098 const Value *DstSV = cast<SrcValueSDNode>(Op.getOperand(3))->getValue();
2099 const Value *SrcSV = cast<SrcValueSDNode>(Op.getOperand(4))->getValue();
Andrew Trickef9de2a2013-05-25 02:42:55 +00002100 SDLoc DL(Op);
Ulrich Weigand5f613df2013-05-06 16:15:19 +00002101
2102 return DAG.getMemcpy(Chain, DL, DstPtr, SrcPtr, DAG.getIntPtrConstant(32),
2103 /*Align*/8, /*isVolatile*/false, /*AlwaysInline*/false,
2104 MachinePointerInfo(DstSV), MachinePointerInfo(SrcSV));
2105}
2106
2107SDValue SystemZTargetLowering::
2108lowerDYNAMIC_STACKALLOC(SDValue Op, SelectionDAG &DAG) const {
2109 SDValue Chain = Op.getOperand(0);
2110 SDValue Size = Op.getOperand(1);
Andrew Trickef9de2a2013-05-25 02:42:55 +00002111 SDLoc DL(Op);
Ulrich Weigand5f613df2013-05-06 16:15:19 +00002112
2113 unsigned SPReg = getStackPointerRegisterToSaveRestore();
2114
2115 // Get a reference to the stack pointer.
2116 SDValue OldSP = DAG.getCopyFromReg(Chain, DL, SPReg, MVT::i64);
2117
2118 // Get the new stack pointer value.
2119 SDValue NewSP = DAG.getNode(ISD::SUB, DL, MVT::i64, OldSP, Size);
2120
2121 // Copy the new stack pointer back.
2122 Chain = DAG.getCopyToReg(Chain, DL, SPReg, NewSP);
2123
2124 // The allocated data lives above the 160 bytes allocated for the standard
2125 // frame, plus any outgoing stack arguments. We don't know how much that
2126 // amounts to yet, so emit a special ADJDYNALLOC placeholder.
2127 SDValue ArgAdjust = DAG.getNode(SystemZISD::ADJDYNALLOC, DL, MVT::i64);
2128 SDValue Result = DAG.getNode(ISD::ADD, DL, MVT::i64, NewSP, ArgAdjust);
2129
2130 SDValue Ops[2] = { Result, Chain };
Craig Topper64941d92014-04-27 19:20:57 +00002131 return DAG.getMergeValues(Ops, DL);
Ulrich Weigand5f613df2013-05-06 16:15:19 +00002132}
2133
Richard Sandiford7d86e472013-08-21 09:34:56 +00002134SDValue SystemZTargetLowering::lowerSMUL_LOHI(SDValue Op,
2135 SelectionDAG &DAG) const {
2136 EVT VT = Op.getValueType();
2137 SDLoc DL(Op);
2138 SDValue Ops[2];
2139 if (is32Bit(VT))
2140 // Just do a normal 64-bit multiplication and extract the results.
2141 // We define this so that it can be used for constant division.
2142 lowerMUL_LOHI32(DAG, DL, ISD::SIGN_EXTEND, Op.getOperand(0),
2143 Op.getOperand(1), Ops[1], Ops[0]);
2144 else {
2145 // Do a full 128-bit multiplication based on UMUL_LOHI64:
2146 //
2147 // (ll * rl) + ((lh * rl) << 64) + ((ll * rh) << 64)
2148 //
2149 // but using the fact that the upper halves are either all zeros
2150 // or all ones:
2151 //
2152 // (ll * rl) - ((lh & rl) << 64) - ((ll & rh) << 64)
2153 //
2154 // and grouping the right terms together since they are quicker than the
2155 // multiplication:
2156 //
2157 // (ll * rl) - (((lh & rl) + (ll & rh)) << 64)
2158 SDValue C63 = DAG.getConstant(63, MVT::i64);
2159 SDValue LL = Op.getOperand(0);
2160 SDValue RL = Op.getOperand(1);
2161 SDValue LH = DAG.getNode(ISD::SRA, DL, VT, LL, C63);
2162 SDValue RH = DAG.getNode(ISD::SRA, DL, VT, RL, C63);
2163 // UMUL_LOHI64 returns the low result in the odd register and the high
2164 // result in the even register. SMUL_LOHI is defined to return the
2165 // low half first, so the results are in reverse order.
2166 lowerGR128Binary(DAG, DL, VT, SystemZ::AEXT128_64, SystemZISD::UMUL_LOHI64,
2167 LL, RL, Ops[1], Ops[0]);
2168 SDValue NegLLTimesRH = DAG.getNode(ISD::AND, DL, VT, LL, RH);
2169 SDValue NegLHTimesRL = DAG.getNode(ISD::AND, DL, VT, LH, RL);
2170 SDValue NegSum = DAG.getNode(ISD::ADD, DL, VT, NegLLTimesRH, NegLHTimesRL);
2171 Ops[1] = DAG.getNode(ISD::SUB, DL, VT, Ops[1], NegSum);
2172 }
Craig Topper64941d92014-04-27 19:20:57 +00002173 return DAG.getMergeValues(Ops, DL);
Richard Sandiford7d86e472013-08-21 09:34:56 +00002174}
2175
Ulrich Weigand5f613df2013-05-06 16:15:19 +00002176SDValue SystemZTargetLowering::lowerUMUL_LOHI(SDValue Op,
2177 SelectionDAG &DAG) const {
2178 EVT VT = Op.getValueType();
Andrew Trickef9de2a2013-05-25 02:42:55 +00002179 SDLoc DL(Op);
Ulrich Weigand5f613df2013-05-06 16:15:19 +00002180 SDValue Ops[2];
Richard Sandiford7d86e472013-08-21 09:34:56 +00002181 if (is32Bit(VT))
2182 // Just do a normal 64-bit multiplication and extract the results.
2183 // We define this so that it can be used for constant division.
2184 lowerMUL_LOHI32(DAG, DL, ISD::ZERO_EXTEND, Op.getOperand(0),
2185 Op.getOperand(1), Ops[1], Ops[0]);
2186 else
2187 // UMUL_LOHI64 returns the low result in the odd register and the high
2188 // result in the even register. UMUL_LOHI is defined to return the
2189 // low half first, so the results are in reverse order.
2190 lowerGR128Binary(DAG, DL, VT, SystemZ::AEXT128_64, SystemZISD::UMUL_LOHI64,
2191 Op.getOperand(0), Op.getOperand(1), Ops[1], Ops[0]);
Craig Topper64941d92014-04-27 19:20:57 +00002192 return DAG.getMergeValues(Ops, DL);
Ulrich Weigand5f613df2013-05-06 16:15:19 +00002193}
2194
2195SDValue SystemZTargetLowering::lowerSDIVREM(SDValue Op,
2196 SelectionDAG &DAG) const {
2197 SDValue Op0 = Op.getOperand(0);
2198 SDValue Op1 = Op.getOperand(1);
2199 EVT VT = Op.getValueType();
Andrew Trickef9de2a2013-05-25 02:42:55 +00002200 SDLoc DL(Op);
Richard Sandiforde6e78852013-07-02 15:40:22 +00002201 unsigned Opcode;
Ulrich Weigand5f613df2013-05-06 16:15:19 +00002202
2203 // We use DSGF for 32-bit division.
2204 if (is32Bit(VT)) {
2205 Op0 = DAG.getNode(ISD::SIGN_EXTEND, DL, MVT::i64, Op0);
Richard Sandiforde6e78852013-07-02 15:40:22 +00002206 Opcode = SystemZISD::SDIVREM32;
2207 } else if (DAG.ComputeNumSignBits(Op1) > 32) {
2208 Op1 = DAG.getNode(ISD::TRUNCATE, DL, MVT::i32, Op1);
2209 Opcode = SystemZISD::SDIVREM32;
2210 } else
2211 Opcode = SystemZISD::SDIVREM64;
Ulrich Weigand5f613df2013-05-06 16:15:19 +00002212
2213 // DSG(F) takes a 64-bit dividend, so the even register in the GR128
2214 // input is "don't care". The instruction returns the remainder in
2215 // the even register and the quotient in the odd register.
2216 SDValue Ops[2];
Richard Sandiforde6e78852013-07-02 15:40:22 +00002217 lowerGR128Binary(DAG, DL, VT, SystemZ::AEXT128_64, Opcode,
Ulrich Weigand5f613df2013-05-06 16:15:19 +00002218 Op0, Op1, Ops[1], Ops[0]);
Craig Topper64941d92014-04-27 19:20:57 +00002219 return DAG.getMergeValues(Ops, DL);
Ulrich Weigand5f613df2013-05-06 16:15:19 +00002220}
2221
2222SDValue SystemZTargetLowering::lowerUDIVREM(SDValue Op,
2223 SelectionDAG &DAG) const {
2224 EVT VT = Op.getValueType();
Andrew Trickef9de2a2013-05-25 02:42:55 +00002225 SDLoc DL(Op);
Ulrich Weigand5f613df2013-05-06 16:15:19 +00002226
2227 // DL(G) uses a double-width dividend, so we need to clear the even
2228 // register in the GR128 input. The instruction returns the remainder
2229 // in the even register and the quotient in the odd register.
2230 SDValue Ops[2];
2231 if (is32Bit(VT))
2232 lowerGR128Binary(DAG, DL, VT, SystemZ::ZEXT128_32, SystemZISD::UDIVREM32,
2233 Op.getOperand(0), Op.getOperand(1), Ops[1], Ops[0]);
2234 else
2235 lowerGR128Binary(DAG, DL, VT, SystemZ::ZEXT128_64, SystemZISD::UDIVREM64,
2236 Op.getOperand(0), Op.getOperand(1), Ops[1], Ops[0]);
Craig Topper64941d92014-04-27 19:20:57 +00002237 return DAG.getMergeValues(Ops, DL);
Ulrich Weigand5f613df2013-05-06 16:15:19 +00002238}
2239
2240SDValue SystemZTargetLowering::lowerOR(SDValue Op, SelectionDAG &DAG) const {
2241 assert(Op.getValueType() == MVT::i64 && "Should be 64-bit operation");
2242
2243 // Get the known-zero masks for each operand.
2244 SDValue Ops[] = { Op.getOperand(0), Op.getOperand(1) };
2245 APInt KnownZero[2], KnownOne[2];
Jay Foada0653a32014-05-14 21:14:37 +00002246 DAG.computeKnownBits(Ops[0], KnownZero[0], KnownOne[0]);
2247 DAG.computeKnownBits(Ops[1], KnownZero[1], KnownOne[1]);
Ulrich Weigand5f613df2013-05-06 16:15:19 +00002248
2249 // See if the upper 32 bits of one operand and the lower 32 bits of the
2250 // other are known zero. They are the low and high operands respectively.
2251 uint64_t Masks[] = { KnownZero[0].getZExtValue(),
2252 KnownZero[1].getZExtValue() };
2253 unsigned High, Low;
2254 if ((Masks[0] >> 32) == 0xffffffff && uint32_t(Masks[1]) == 0xffffffff)
2255 High = 1, Low = 0;
2256 else if ((Masks[1] >> 32) == 0xffffffff && uint32_t(Masks[0]) == 0xffffffff)
2257 High = 0, Low = 1;
2258 else
2259 return Op;
2260
2261 SDValue LowOp = Ops[Low];
2262 SDValue HighOp = Ops[High];
2263
2264 // If the high part is a constant, we're better off using IILH.
2265 if (HighOp.getOpcode() == ISD::Constant)
2266 return Op;
2267
2268 // If the low part is a constant that is outside the range of LHI,
2269 // then we're better off using IILF.
2270 if (LowOp.getOpcode() == ISD::Constant) {
2271 int64_t Value = int32_t(cast<ConstantSDNode>(LowOp)->getZExtValue());
2272 if (!isInt<16>(Value))
2273 return Op;
2274 }
2275
2276 // Check whether the high part is an AND that doesn't change the
2277 // high 32 bits and just masks out low bits. We can skip it if so.
2278 if (HighOp.getOpcode() == ISD::AND &&
2279 HighOp.getOperand(1).getOpcode() == ISD::Constant) {
Richard Sandifordccc2a7c2013-12-03 11:01:54 +00002280 SDValue HighOp0 = HighOp.getOperand(0);
2281 uint64_t Mask = cast<ConstantSDNode>(HighOp.getOperand(1))->getZExtValue();
2282 if (DAG.MaskedValueIsZero(HighOp0, APInt(64, ~(Mask | 0xffffffff))))
2283 HighOp = HighOp0;
Ulrich Weigand5f613df2013-05-06 16:15:19 +00002284 }
2285
2286 // Take advantage of the fact that all GR32 operations only change the
2287 // low 32 bits by truncating Low to an i32 and inserting it directly
2288 // using a subreg. The interesting cases are those where the truncation
2289 // can be folded.
Andrew Trickef9de2a2013-05-25 02:42:55 +00002290 SDLoc DL(Op);
Ulrich Weigand5f613df2013-05-06 16:15:19 +00002291 SDValue Low32 = DAG.getNode(ISD::TRUNCATE, DL, MVT::i32, LowOp);
Richard Sandiford87a44362013-09-30 10:28:35 +00002292 return DAG.getTargetInsertSubreg(SystemZ::subreg_l32, DL,
Richard Sandifordd8163202013-09-13 09:12:44 +00002293 MVT::i64, HighOp, Low32);
Ulrich Weigand5f613df2013-05-06 16:15:19 +00002294}
2295
Richard Sandifordbef3d7a2013-12-10 10:49:34 +00002296// Op is an atomic load. Lower it into a normal volatile load.
2297SDValue SystemZTargetLowering::lowerATOMIC_LOAD(SDValue Op,
2298 SelectionDAG &DAG) const {
Richard Sandiford21f5d682014-03-06 11:22:58 +00002299 auto *Node = cast<AtomicSDNode>(Op.getNode());
Richard Sandifordbef3d7a2013-12-10 10:49:34 +00002300 return DAG.getExtLoad(ISD::EXTLOAD, SDLoc(Op), Op.getValueType(),
2301 Node->getChain(), Node->getBasePtr(),
2302 Node->getMemoryVT(), Node->getMemOperand());
2303}
2304
2305// Op is an atomic store. Lower it into a normal volatile store followed
2306// by a serialization.
2307SDValue SystemZTargetLowering::lowerATOMIC_STORE(SDValue Op,
2308 SelectionDAG &DAG) const {
Richard Sandiford21f5d682014-03-06 11:22:58 +00002309 auto *Node = cast<AtomicSDNode>(Op.getNode());
Richard Sandifordbef3d7a2013-12-10 10:49:34 +00002310 SDValue Chain = DAG.getTruncStore(Node->getChain(), SDLoc(Op), Node->getVal(),
2311 Node->getBasePtr(), Node->getMemoryVT(),
2312 Node->getMemOperand());
2313 return SDValue(DAG.getMachineNode(SystemZ::Serialize, SDLoc(Op), MVT::Other,
2314 Chain), 0);
2315}
2316
Ulrich Weigand5f613df2013-05-06 16:15:19 +00002317// Op is an 8-, 16-bit or 32-bit ATOMIC_LOAD_* operation. Lower the first
2318// two into the fullword ATOMIC_LOADW_* operation given by Opcode.
Richard Sandifordbef3d7a2013-12-10 10:49:34 +00002319SDValue SystemZTargetLowering::lowerATOMIC_LOAD_OP(SDValue Op,
2320 SelectionDAG &DAG,
2321 unsigned Opcode) const {
Richard Sandiford21f5d682014-03-06 11:22:58 +00002322 auto *Node = cast<AtomicSDNode>(Op.getNode());
Ulrich Weigand5f613df2013-05-06 16:15:19 +00002323
2324 // 32-bit operations need no code outside the main loop.
2325 EVT NarrowVT = Node->getMemoryVT();
2326 EVT WideVT = MVT::i32;
2327 if (NarrowVT == WideVT)
2328 return Op;
2329
2330 int64_t BitSize = NarrowVT.getSizeInBits();
2331 SDValue ChainIn = Node->getChain();
2332 SDValue Addr = Node->getBasePtr();
2333 SDValue Src2 = Node->getVal();
2334 MachineMemOperand *MMO = Node->getMemOperand();
Andrew Trickef9de2a2013-05-25 02:42:55 +00002335 SDLoc DL(Node);
Ulrich Weigand5f613df2013-05-06 16:15:19 +00002336 EVT PtrVT = Addr.getValueType();
2337
2338 // Convert atomic subtracts of constants into additions.
2339 if (Opcode == SystemZISD::ATOMIC_LOADW_SUB)
Richard Sandiford21f5d682014-03-06 11:22:58 +00002340 if (auto *Const = dyn_cast<ConstantSDNode>(Src2)) {
Ulrich Weigand5f613df2013-05-06 16:15:19 +00002341 Opcode = SystemZISD::ATOMIC_LOADW_ADD;
2342 Src2 = DAG.getConstant(-Const->getSExtValue(), Src2.getValueType());
2343 }
2344
2345 // Get the address of the containing word.
2346 SDValue AlignedAddr = DAG.getNode(ISD::AND, DL, PtrVT, Addr,
2347 DAG.getConstant(-4, PtrVT));
2348
2349 // Get the number of bits that the word must be rotated left in order
2350 // to bring the field to the top bits of a GR32.
2351 SDValue BitShift = DAG.getNode(ISD::SHL, DL, PtrVT, Addr,
2352 DAG.getConstant(3, PtrVT));
2353 BitShift = DAG.getNode(ISD::TRUNCATE, DL, WideVT, BitShift);
2354
2355 // Get the complementing shift amount, for rotating a field in the top
2356 // bits back to its proper position.
2357 SDValue NegBitShift = DAG.getNode(ISD::SUB, DL, WideVT,
2358 DAG.getConstant(0, WideVT), BitShift);
2359
2360 // Extend the source operand to 32 bits and prepare it for the inner loop.
2361 // ATOMIC_SWAPW uses RISBG to rotate the field left, but all other
2362 // operations require the source to be shifted in advance. (This shift
2363 // can be folded if the source is constant.) For AND and NAND, the lower
2364 // bits must be set, while for other opcodes they should be left clear.
2365 if (Opcode != SystemZISD::ATOMIC_SWAPW)
2366 Src2 = DAG.getNode(ISD::SHL, DL, WideVT, Src2,
2367 DAG.getConstant(32 - BitSize, WideVT));
2368 if (Opcode == SystemZISD::ATOMIC_LOADW_AND ||
2369 Opcode == SystemZISD::ATOMIC_LOADW_NAND)
2370 Src2 = DAG.getNode(ISD::OR, DL, WideVT, Src2,
2371 DAG.getConstant(uint32_t(-1) >> BitSize, WideVT));
2372
2373 // Construct the ATOMIC_LOADW_* node.
2374 SDVTList VTList = DAG.getVTList(WideVT, MVT::Other);
2375 SDValue Ops[] = { ChainIn, AlignedAddr, Src2, BitShift, NegBitShift,
2376 DAG.getConstant(BitSize, WideVT) };
2377 SDValue AtomicOp = DAG.getMemIntrinsicNode(Opcode, DL, VTList, Ops,
Ulrich Weigand5f613df2013-05-06 16:15:19 +00002378 NarrowVT, MMO);
2379
2380 // Rotate the result of the final CS so that the field is in the lower
2381 // bits of a GR32, then truncate it.
2382 SDValue ResultShift = DAG.getNode(ISD::ADD, DL, WideVT, BitShift,
2383 DAG.getConstant(BitSize, WideVT));
2384 SDValue Result = DAG.getNode(ISD::ROTL, DL, WideVT, AtomicOp, ResultShift);
2385
2386 SDValue RetOps[2] = { Result, AtomicOp.getValue(1) };
Craig Topper64941d92014-04-27 19:20:57 +00002387 return DAG.getMergeValues(RetOps, DL);
Ulrich Weigand5f613df2013-05-06 16:15:19 +00002388}
2389
Richard Sandiford41350a52013-12-24 15:18:04 +00002390// Op is an ATOMIC_LOAD_SUB operation. Lower 8- and 16-bit operations
Richard Sandiford002019a2013-12-24 15:22:39 +00002391// into ATOMIC_LOADW_SUBs and decide whether to convert 32- and 64-bit
Richard Sandiford41350a52013-12-24 15:18:04 +00002392// operations into additions.
2393SDValue SystemZTargetLowering::lowerATOMIC_LOAD_SUB(SDValue Op,
2394 SelectionDAG &DAG) const {
Richard Sandiford21f5d682014-03-06 11:22:58 +00002395 auto *Node = cast<AtomicSDNode>(Op.getNode());
Richard Sandiford41350a52013-12-24 15:18:04 +00002396 EVT MemVT = Node->getMemoryVT();
2397 if (MemVT == MVT::i32 || MemVT == MVT::i64) {
2398 // A full-width operation.
2399 assert(Op.getValueType() == MemVT && "Mismatched VTs");
2400 SDValue Src2 = Node->getVal();
2401 SDValue NegSrc2;
2402 SDLoc DL(Src2);
2403
Richard Sandiford21f5d682014-03-06 11:22:58 +00002404 if (auto *Op2 = dyn_cast<ConstantSDNode>(Src2)) {
Richard Sandiford41350a52013-12-24 15:18:04 +00002405 // Use an addition if the operand is constant and either LAA(G) is
2406 // available or the negative value is in the range of A(G)FHI.
2407 int64_t Value = (-Op2->getAPIntValue()).getSExtValue();
Eric Christopher93bf97c2014-06-27 07:38:01 +00002408 if (isInt<32>(Value) || Subtarget.hasInterlockedAccess1())
Richard Sandiford41350a52013-12-24 15:18:04 +00002409 NegSrc2 = DAG.getConstant(Value, MemVT);
Eric Christopher93bf97c2014-06-27 07:38:01 +00002410 } else if (Subtarget.hasInterlockedAccess1())
Richard Sandiford41350a52013-12-24 15:18:04 +00002411 // Use LAA(G) if available.
2412 NegSrc2 = DAG.getNode(ISD::SUB, DL, MemVT, DAG.getConstant(0, MemVT),
2413 Src2);
2414
2415 if (NegSrc2.getNode())
2416 return DAG.getAtomic(ISD::ATOMIC_LOAD_ADD, DL, MemVT,
2417 Node->getChain(), Node->getBasePtr(), NegSrc2,
2418 Node->getMemOperand(), Node->getOrdering(),
2419 Node->getSynchScope());
2420
2421 // Use the node as-is.
2422 return Op;
2423 }
2424
2425 return lowerATOMIC_LOAD_OP(Op, DAG, SystemZISD::ATOMIC_LOADW_SUB);
2426}
2427
Ulrich Weigand5f613df2013-05-06 16:15:19 +00002428// Node is an 8- or 16-bit ATOMIC_CMP_SWAP operation. Lower the first two
2429// into a fullword ATOMIC_CMP_SWAPW operation.
2430SDValue SystemZTargetLowering::lowerATOMIC_CMP_SWAP(SDValue Op,
2431 SelectionDAG &DAG) const {
Richard Sandiford21f5d682014-03-06 11:22:58 +00002432 auto *Node = cast<AtomicSDNode>(Op.getNode());
Ulrich Weigand5f613df2013-05-06 16:15:19 +00002433
2434 // We have native support for 32-bit compare and swap.
2435 EVT NarrowVT = Node->getMemoryVT();
2436 EVT WideVT = MVT::i32;
2437 if (NarrowVT == WideVT)
2438 return Op;
2439
2440 int64_t BitSize = NarrowVT.getSizeInBits();
2441 SDValue ChainIn = Node->getOperand(0);
2442 SDValue Addr = Node->getOperand(1);
2443 SDValue CmpVal = Node->getOperand(2);
2444 SDValue SwapVal = Node->getOperand(3);
2445 MachineMemOperand *MMO = Node->getMemOperand();
Andrew Trickef9de2a2013-05-25 02:42:55 +00002446 SDLoc DL(Node);
Ulrich Weigand5f613df2013-05-06 16:15:19 +00002447 EVT PtrVT = Addr.getValueType();
2448
2449 // Get the address of the containing word.
2450 SDValue AlignedAddr = DAG.getNode(ISD::AND, DL, PtrVT, Addr,
2451 DAG.getConstant(-4, PtrVT));
2452
2453 // Get the number of bits that the word must be rotated left in order
2454 // to bring the field to the top bits of a GR32.
2455 SDValue BitShift = DAG.getNode(ISD::SHL, DL, PtrVT, Addr,
2456 DAG.getConstant(3, PtrVT));
2457 BitShift = DAG.getNode(ISD::TRUNCATE, DL, WideVT, BitShift);
2458
2459 // Get the complementing shift amount, for rotating a field in the top
2460 // bits back to its proper position.
2461 SDValue NegBitShift = DAG.getNode(ISD::SUB, DL, WideVT,
2462 DAG.getConstant(0, WideVT), BitShift);
2463
2464 // Construct the ATOMIC_CMP_SWAPW node.
2465 SDVTList VTList = DAG.getVTList(WideVT, MVT::Other);
2466 SDValue Ops[] = { ChainIn, AlignedAddr, CmpVal, SwapVal, BitShift,
2467 NegBitShift, DAG.getConstant(BitSize, WideVT) };
2468 SDValue AtomicOp = DAG.getMemIntrinsicNode(SystemZISD::ATOMIC_CMP_SWAPW, DL,
Craig Topper206fcd42014-04-26 19:29:41 +00002469 VTList, Ops, NarrowVT, MMO);
Ulrich Weigand5f613df2013-05-06 16:15:19 +00002470 return AtomicOp;
2471}
2472
2473SDValue SystemZTargetLowering::lowerSTACKSAVE(SDValue Op,
2474 SelectionDAG &DAG) const {
2475 MachineFunction &MF = DAG.getMachineFunction();
2476 MF.getInfo<SystemZMachineFunctionInfo>()->setManipulatesSP(true);
Andrew Trickef9de2a2013-05-25 02:42:55 +00002477 return DAG.getCopyFromReg(Op.getOperand(0), SDLoc(Op),
Ulrich Weigand5f613df2013-05-06 16:15:19 +00002478 SystemZ::R15D, Op.getValueType());
2479}
2480
2481SDValue SystemZTargetLowering::lowerSTACKRESTORE(SDValue Op,
2482 SelectionDAG &DAG) const {
2483 MachineFunction &MF = DAG.getMachineFunction();
2484 MF.getInfo<SystemZMachineFunctionInfo>()->setManipulatesSP(true);
Andrew Trickef9de2a2013-05-25 02:42:55 +00002485 return DAG.getCopyToReg(Op.getOperand(0), SDLoc(Op),
Ulrich Weigand5f613df2013-05-06 16:15:19 +00002486 SystemZ::R15D, Op.getOperand(1));
2487}
2488
Richard Sandiford03481332013-08-23 11:36:42 +00002489SDValue SystemZTargetLowering::lowerPREFETCH(SDValue Op,
2490 SelectionDAG &DAG) const {
2491 bool IsData = cast<ConstantSDNode>(Op.getOperand(4))->getZExtValue();
2492 if (!IsData)
2493 // Just preserve the chain.
2494 return Op.getOperand(0);
2495
2496 bool IsWrite = cast<ConstantSDNode>(Op.getOperand(2))->getZExtValue();
2497 unsigned Code = IsWrite ? SystemZ::PFD_WRITE : SystemZ::PFD_READ;
Richard Sandiford21f5d682014-03-06 11:22:58 +00002498 auto *Node = cast<MemIntrinsicSDNode>(Op.getNode());
Richard Sandiford03481332013-08-23 11:36:42 +00002499 SDValue Ops[] = {
2500 Op.getOperand(0),
2501 DAG.getConstant(Code, MVT::i32),
2502 Op.getOperand(1)
2503 };
2504 return DAG.getMemIntrinsicNode(SystemZISD::PREFETCH, SDLoc(Op),
Craig Topper206fcd42014-04-26 19:29:41 +00002505 Node->getVTList(), Ops,
Richard Sandiford03481332013-08-23 11:36:42 +00002506 Node->getMemoryVT(), Node->getMemOperand());
2507}
2508
Ulrich Weigand5f613df2013-05-06 16:15:19 +00002509SDValue SystemZTargetLowering::LowerOperation(SDValue Op,
2510 SelectionDAG &DAG) const {
2511 switch (Op.getOpcode()) {
2512 case ISD::BR_CC:
2513 return lowerBR_CC(Op, DAG);
2514 case ISD::SELECT_CC:
2515 return lowerSELECT_CC(Op, DAG);
Richard Sandifordf722a8e302013-10-16 11:10:55 +00002516 case ISD::SETCC:
2517 return lowerSETCC(Op, DAG);
Ulrich Weigand5f613df2013-05-06 16:15:19 +00002518 case ISD::GlobalAddress:
2519 return lowerGlobalAddress(cast<GlobalAddressSDNode>(Op), DAG);
2520 case ISD::GlobalTLSAddress:
2521 return lowerGlobalTLSAddress(cast<GlobalAddressSDNode>(Op), DAG);
2522 case ISD::BlockAddress:
2523 return lowerBlockAddress(cast<BlockAddressSDNode>(Op), DAG);
2524 case ISD::JumpTable:
2525 return lowerJumpTable(cast<JumpTableSDNode>(Op), DAG);
2526 case ISD::ConstantPool:
2527 return lowerConstantPool(cast<ConstantPoolSDNode>(Op), DAG);
2528 case ISD::BITCAST:
2529 return lowerBITCAST(Op, DAG);
2530 case ISD::VASTART:
2531 return lowerVASTART(Op, DAG);
2532 case ISD::VACOPY:
2533 return lowerVACOPY(Op, DAG);
2534 case ISD::DYNAMIC_STACKALLOC:
2535 return lowerDYNAMIC_STACKALLOC(Op, DAG);
Richard Sandiford7d86e472013-08-21 09:34:56 +00002536 case ISD::SMUL_LOHI:
2537 return lowerSMUL_LOHI(Op, DAG);
Ulrich Weigand5f613df2013-05-06 16:15:19 +00002538 case ISD::UMUL_LOHI:
2539 return lowerUMUL_LOHI(Op, DAG);
2540 case ISD::SDIVREM:
2541 return lowerSDIVREM(Op, DAG);
2542 case ISD::UDIVREM:
2543 return lowerUDIVREM(Op, DAG);
2544 case ISD::OR:
2545 return lowerOR(Op, DAG);
2546 case ISD::ATOMIC_SWAP:
Richard Sandifordbef3d7a2013-12-10 10:49:34 +00002547 return lowerATOMIC_LOAD_OP(Op, DAG, SystemZISD::ATOMIC_SWAPW);
2548 case ISD::ATOMIC_STORE:
2549 return lowerATOMIC_STORE(Op, DAG);
2550 case ISD::ATOMIC_LOAD:
2551 return lowerATOMIC_LOAD(Op, DAG);
Ulrich Weigand5f613df2013-05-06 16:15:19 +00002552 case ISD::ATOMIC_LOAD_ADD:
Richard Sandifordbef3d7a2013-12-10 10:49:34 +00002553 return lowerATOMIC_LOAD_OP(Op, DAG, SystemZISD::ATOMIC_LOADW_ADD);
Ulrich Weigand5f613df2013-05-06 16:15:19 +00002554 case ISD::ATOMIC_LOAD_SUB:
Richard Sandiford41350a52013-12-24 15:18:04 +00002555 return lowerATOMIC_LOAD_SUB(Op, DAG);
Ulrich Weigand5f613df2013-05-06 16:15:19 +00002556 case ISD::ATOMIC_LOAD_AND:
Richard Sandifordbef3d7a2013-12-10 10:49:34 +00002557 return lowerATOMIC_LOAD_OP(Op, DAG, SystemZISD::ATOMIC_LOADW_AND);
Ulrich Weigand5f613df2013-05-06 16:15:19 +00002558 case ISD::ATOMIC_LOAD_OR:
Richard Sandifordbef3d7a2013-12-10 10:49:34 +00002559 return lowerATOMIC_LOAD_OP(Op, DAG, SystemZISD::ATOMIC_LOADW_OR);
Ulrich Weigand5f613df2013-05-06 16:15:19 +00002560 case ISD::ATOMIC_LOAD_XOR:
Richard Sandifordbef3d7a2013-12-10 10:49:34 +00002561 return lowerATOMIC_LOAD_OP(Op, DAG, SystemZISD::ATOMIC_LOADW_XOR);
Ulrich Weigand5f613df2013-05-06 16:15:19 +00002562 case ISD::ATOMIC_LOAD_NAND:
Richard Sandifordbef3d7a2013-12-10 10:49:34 +00002563 return lowerATOMIC_LOAD_OP(Op, DAG, SystemZISD::ATOMIC_LOADW_NAND);
Ulrich Weigand5f613df2013-05-06 16:15:19 +00002564 case ISD::ATOMIC_LOAD_MIN:
Richard Sandifordbef3d7a2013-12-10 10:49:34 +00002565 return lowerATOMIC_LOAD_OP(Op, DAG, SystemZISD::ATOMIC_LOADW_MIN);
Ulrich Weigand5f613df2013-05-06 16:15:19 +00002566 case ISD::ATOMIC_LOAD_MAX:
Richard Sandifordbef3d7a2013-12-10 10:49:34 +00002567 return lowerATOMIC_LOAD_OP(Op, DAG, SystemZISD::ATOMIC_LOADW_MAX);
Ulrich Weigand5f613df2013-05-06 16:15:19 +00002568 case ISD::ATOMIC_LOAD_UMIN:
Richard Sandifordbef3d7a2013-12-10 10:49:34 +00002569 return lowerATOMIC_LOAD_OP(Op, DAG, SystemZISD::ATOMIC_LOADW_UMIN);
Ulrich Weigand5f613df2013-05-06 16:15:19 +00002570 case ISD::ATOMIC_LOAD_UMAX:
Richard Sandifordbef3d7a2013-12-10 10:49:34 +00002571 return lowerATOMIC_LOAD_OP(Op, DAG, SystemZISD::ATOMIC_LOADW_UMAX);
Ulrich Weigand5f613df2013-05-06 16:15:19 +00002572 case ISD::ATOMIC_CMP_SWAP:
2573 return lowerATOMIC_CMP_SWAP(Op, DAG);
2574 case ISD::STACKSAVE:
2575 return lowerSTACKSAVE(Op, DAG);
2576 case ISD::STACKRESTORE:
2577 return lowerSTACKRESTORE(Op, DAG);
Richard Sandiford03481332013-08-23 11:36:42 +00002578 case ISD::PREFETCH:
2579 return lowerPREFETCH(Op, DAG);
Ulrich Weigand5f613df2013-05-06 16:15:19 +00002580 default:
2581 llvm_unreachable("Unexpected node to lower");
2582 }
2583}
2584
2585const char *SystemZTargetLowering::getTargetNodeName(unsigned Opcode) const {
2586#define OPCODE(NAME) case SystemZISD::NAME: return "SystemZISD::" #NAME
2587 switch (Opcode) {
2588 OPCODE(RET_FLAG);
2589 OPCODE(CALL);
Richard Sandiford709bda62013-08-19 12:42:31 +00002590 OPCODE(SIBCALL);
Ulrich Weigand5f613df2013-05-06 16:15:19 +00002591 OPCODE(PCREL_WRAPPER);
Richard Sandiford54b36912013-09-27 15:14:04 +00002592 OPCODE(PCREL_OFFSET);
Richard Sandiford57485472013-12-13 15:35:00 +00002593 OPCODE(IABS);
Richard Sandiford5bc670b2013-09-06 11:51:39 +00002594 OPCODE(ICMP);
2595 OPCODE(FCMP);
Richard Sandiford35b9be22013-08-28 10:31:43 +00002596 OPCODE(TM);
Ulrich Weigand5f613df2013-05-06 16:15:19 +00002597 OPCODE(BR_CCMASK);
2598 OPCODE(SELECT_CCMASK);
2599 OPCODE(ADJDYNALLOC);
2600 OPCODE(EXTRACT_ACCESS);
2601 OPCODE(UMUL_LOHI64);
2602 OPCODE(SDIVREM64);
2603 OPCODE(UDIVREM32);
2604 OPCODE(UDIVREM64);
Richard Sandifordd131ff82013-07-08 09:35:23 +00002605 OPCODE(MVC);
Richard Sandiford5e318f02013-08-27 09:54:29 +00002606 OPCODE(MVC_LOOP);
Richard Sandiford178273a2013-09-05 10:36:45 +00002607 OPCODE(NC);
2608 OPCODE(NC_LOOP);
2609 OPCODE(OC);
2610 OPCODE(OC_LOOP);
2611 OPCODE(XC);
2612 OPCODE(XC_LOOP);
Richard Sandiford761703a2013-08-12 10:17:33 +00002613 OPCODE(CLC);
Richard Sandiford5e318f02013-08-27 09:54:29 +00002614 OPCODE(CLC_LOOP);
Richard Sandifordca232712013-08-16 11:21:54 +00002615 OPCODE(STRCMP);
Richard Sandifordbb83a502013-08-16 11:29:37 +00002616 OPCODE(STPCPY);
Richard Sandiford0dec06a2013-08-16 11:41:43 +00002617 OPCODE(SEARCH_STRING);
Richard Sandiford564681c2013-08-12 10:28:10 +00002618 OPCODE(IPM);
Richard Sandiford9afe6132013-12-10 10:36:34 +00002619 OPCODE(SERIALIZE);
Ulrich Weigand5f613df2013-05-06 16:15:19 +00002620 OPCODE(ATOMIC_SWAPW);
2621 OPCODE(ATOMIC_LOADW_ADD);
2622 OPCODE(ATOMIC_LOADW_SUB);
2623 OPCODE(ATOMIC_LOADW_AND);
2624 OPCODE(ATOMIC_LOADW_OR);
2625 OPCODE(ATOMIC_LOADW_XOR);
2626 OPCODE(ATOMIC_LOADW_NAND);
2627 OPCODE(ATOMIC_LOADW_MIN);
2628 OPCODE(ATOMIC_LOADW_MAX);
2629 OPCODE(ATOMIC_LOADW_UMIN);
2630 OPCODE(ATOMIC_LOADW_UMAX);
2631 OPCODE(ATOMIC_CMP_SWAPW);
Richard Sandiford03481332013-08-23 11:36:42 +00002632 OPCODE(PREFETCH);
Ulrich Weigand5f613df2013-05-06 16:15:19 +00002633 }
Craig Topper062a2ba2014-04-25 05:30:21 +00002634 return nullptr;
Ulrich Weigand5f613df2013-05-06 16:15:19 +00002635#undef OPCODE
2636}
2637
Richard Sandiford95bc5f92014-03-07 11:34:35 +00002638SDValue SystemZTargetLowering::PerformDAGCombine(SDNode *N,
2639 DAGCombinerInfo &DCI) const {
2640 SelectionDAG &DAG = DCI.DAG;
2641 unsigned Opcode = N->getOpcode();
2642 if (Opcode == ISD::SIGN_EXTEND) {
2643 // Convert (sext (ashr (shl X, C1), C2)) to
2644 // (ashr (shl (anyext X), C1'), C2')), since wider shifts are as
2645 // cheap as narrower ones.
2646 SDValue N0 = N->getOperand(0);
2647 EVT VT = N->getValueType(0);
2648 if (N0.hasOneUse() && N0.getOpcode() == ISD::SRA) {
2649 auto *SraAmt = dyn_cast<ConstantSDNode>(N0.getOperand(1));
2650 SDValue Inner = N0.getOperand(0);
2651 if (SraAmt && Inner.hasOneUse() && Inner.getOpcode() == ISD::SHL) {
2652 if (auto *ShlAmt = dyn_cast<ConstantSDNode>(Inner.getOperand(1))) {
2653 unsigned Extra = (VT.getSizeInBits() -
2654 N0.getValueType().getSizeInBits());
2655 unsigned NewShlAmt = ShlAmt->getZExtValue() + Extra;
2656 unsigned NewSraAmt = SraAmt->getZExtValue() + Extra;
2657 EVT ShiftVT = N0.getOperand(1).getValueType();
2658 SDValue Ext = DAG.getNode(ISD::ANY_EXTEND, SDLoc(Inner), VT,
2659 Inner.getOperand(0));
2660 SDValue Shl = DAG.getNode(ISD::SHL, SDLoc(Inner), VT, Ext,
2661 DAG.getConstant(NewShlAmt, ShiftVT));
2662 return DAG.getNode(ISD::SRA, SDLoc(N0), VT, Shl,
2663 DAG.getConstant(NewSraAmt, ShiftVT));
2664 }
2665 }
2666 }
2667 }
2668 return SDValue();
2669}
2670
Ulrich Weigand5f613df2013-05-06 16:15:19 +00002671//===----------------------------------------------------------------------===//
2672// Custom insertion
2673//===----------------------------------------------------------------------===//
2674
2675// Create a new basic block after MBB.
2676static MachineBasicBlock *emitBlockAfter(MachineBasicBlock *MBB) {
2677 MachineFunction &MF = *MBB->getParent();
2678 MachineBasicBlock *NewMBB = MF.CreateMachineBasicBlock(MBB->getBasicBlock());
Benjamin Kramerb6d0bd42014-03-02 12:27:27 +00002679 MF.insert(std::next(MachineFunction::iterator(MBB)), NewMBB);
Ulrich Weigand5f613df2013-05-06 16:15:19 +00002680 return NewMBB;
2681}
2682
Richard Sandifordbe133a82013-08-28 09:01:51 +00002683// Split MBB after MI and return the new block (the one that contains
2684// instructions after MI).
2685static MachineBasicBlock *splitBlockAfter(MachineInstr *MI,
2686 MachineBasicBlock *MBB) {
2687 MachineBasicBlock *NewMBB = emitBlockAfter(MBB);
2688 NewMBB->splice(NewMBB->begin(), MBB,
Benjamin Kramerb6d0bd42014-03-02 12:27:27 +00002689 std::next(MachineBasicBlock::iterator(MI)), MBB->end());
Richard Sandifordbe133a82013-08-28 09:01:51 +00002690 NewMBB->transferSuccessorsAndUpdatePHIs(MBB);
2691 return NewMBB;
2692}
2693
Richard Sandiford5e318f02013-08-27 09:54:29 +00002694// Split MBB before MI and return the new block (the one that contains MI).
2695static MachineBasicBlock *splitBlockBefore(MachineInstr *MI,
2696 MachineBasicBlock *MBB) {
Ulrich Weigand5f613df2013-05-06 16:15:19 +00002697 MachineBasicBlock *NewMBB = emitBlockAfter(MBB);
Richard Sandiford5e318f02013-08-27 09:54:29 +00002698 NewMBB->splice(NewMBB->begin(), MBB, MI, MBB->end());
Ulrich Weigand5f613df2013-05-06 16:15:19 +00002699 NewMBB->transferSuccessorsAndUpdatePHIs(MBB);
2700 return NewMBB;
2701}
2702
Richard Sandiford5e318f02013-08-27 09:54:29 +00002703// Force base value Base into a register before MI. Return the register.
2704static unsigned forceReg(MachineInstr *MI, MachineOperand &Base,
2705 const SystemZInstrInfo *TII) {
2706 if (Base.isReg())
2707 return Base.getReg();
2708
2709 MachineBasicBlock *MBB = MI->getParent();
2710 MachineFunction &MF = *MBB->getParent();
2711 MachineRegisterInfo &MRI = MF.getRegInfo();
2712
2713 unsigned Reg = MRI.createVirtualRegister(&SystemZ::ADDR64BitRegClass);
2714 BuildMI(*MBB, MI, MI->getDebugLoc(), TII->get(SystemZ::LA), Reg)
2715 .addOperand(Base).addImm(0).addReg(0);
2716 return Reg;
2717}
2718
Ulrich Weigand5f613df2013-05-06 16:15:19 +00002719// Implement EmitInstrWithCustomInserter for pseudo Select* instruction MI.
2720MachineBasicBlock *
2721SystemZTargetLowering::emitSelect(MachineInstr *MI,
2722 MachineBasicBlock *MBB) const {
Eric Christophera6734172015-01-31 00:06:45 +00002723 const SystemZInstrInfo *TII =
2724 static_cast<const SystemZInstrInfo *>(Subtarget.getInstrInfo());
Ulrich Weigand5f613df2013-05-06 16:15:19 +00002725
2726 unsigned DestReg = MI->getOperand(0).getReg();
2727 unsigned TrueReg = MI->getOperand(1).getReg();
2728 unsigned FalseReg = MI->getOperand(2).getReg();
Richard Sandiford3d768e32013-07-31 12:30:20 +00002729 unsigned CCValid = MI->getOperand(3).getImm();
2730 unsigned CCMask = MI->getOperand(4).getImm();
Ulrich Weigand5f613df2013-05-06 16:15:19 +00002731 DebugLoc DL = MI->getDebugLoc();
2732
2733 MachineBasicBlock *StartMBB = MBB;
Richard Sandiford5e318f02013-08-27 09:54:29 +00002734 MachineBasicBlock *JoinMBB = splitBlockBefore(MI, MBB);
Ulrich Weigand5f613df2013-05-06 16:15:19 +00002735 MachineBasicBlock *FalseMBB = emitBlockAfter(StartMBB);
2736
2737 // StartMBB:
Richard Sandiford0fb90ab2013-05-28 10:41:11 +00002738 // BRC CCMask, JoinMBB
Ulrich Weigand5f613df2013-05-06 16:15:19 +00002739 // # fallthrough to FalseMBB
2740 MBB = StartMBB;
Richard Sandiford3d768e32013-07-31 12:30:20 +00002741 BuildMI(MBB, DL, TII->get(SystemZ::BRC))
2742 .addImm(CCValid).addImm(CCMask).addMBB(JoinMBB);
Ulrich Weigand5f613df2013-05-06 16:15:19 +00002743 MBB->addSuccessor(JoinMBB);
2744 MBB->addSuccessor(FalseMBB);
2745
2746 // FalseMBB:
2747 // # fallthrough to JoinMBB
2748 MBB = FalseMBB;
2749 MBB->addSuccessor(JoinMBB);
2750
2751 // JoinMBB:
2752 // %Result = phi [ %FalseReg, FalseMBB ], [ %TrueReg, StartMBB ]
2753 // ...
2754 MBB = JoinMBB;
Richard Sandiford5e318f02013-08-27 09:54:29 +00002755 BuildMI(*MBB, MI, DL, TII->get(SystemZ::PHI), DestReg)
Ulrich Weigand5f613df2013-05-06 16:15:19 +00002756 .addReg(TrueReg).addMBB(StartMBB)
2757 .addReg(FalseReg).addMBB(FalseMBB);
2758
2759 MI->eraseFromParent();
2760 return JoinMBB;
2761}
2762
Richard Sandifordb86a8342013-06-27 09:27:40 +00002763// Implement EmitInstrWithCustomInserter for pseudo CondStore* instruction MI.
2764// StoreOpcode is the store to use and Invert says whether the store should
Richard Sandiforda68e6f52013-07-25 08:57:02 +00002765// happen when the condition is false rather than true. If a STORE ON
2766// CONDITION is available, STOCOpcode is its opcode, otherwise it is 0.
Richard Sandifordb86a8342013-06-27 09:27:40 +00002767MachineBasicBlock *
2768SystemZTargetLowering::emitCondStore(MachineInstr *MI,
2769 MachineBasicBlock *MBB,
Richard Sandiforda68e6f52013-07-25 08:57:02 +00002770 unsigned StoreOpcode, unsigned STOCOpcode,
2771 bool Invert) const {
Eric Christophera6734172015-01-31 00:06:45 +00002772 const SystemZInstrInfo *TII =
2773 static_cast<const SystemZInstrInfo *>(Subtarget.getInstrInfo());
Richard Sandifordb86a8342013-06-27 09:27:40 +00002774
Richard Sandiforda68e6f52013-07-25 08:57:02 +00002775 unsigned SrcReg = MI->getOperand(0).getReg();
2776 MachineOperand Base = MI->getOperand(1);
2777 int64_t Disp = MI->getOperand(2).getImm();
2778 unsigned IndexReg = MI->getOperand(3).getReg();
Richard Sandiford3d768e32013-07-31 12:30:20 +00002779 unsigned CCValid = MI->getOperand(4).getImm();
2780 unsigned CCMask = MI->getOperand(5).getImm();
Richard Sandifordb86a8342013-06-27 09:27:40 +00002781 DebugLoc DL = MI->getDebugLoc();
2782
2783 StoreOpcode = TII->getOpcodeForOffset(StoreOpcode, Disp);
2784
Richard Sandiforda68e6f52013-07-25 08:57:02 +00002785 // Use STOCOpcode if possible. We could use different store patterns in
2786 // order to avoid matching the index register, but the performance trade-offs
2787 // might be more complicated in that case.
Eric Christopher93bf97c2014-06-27 07:38:01 +00002788 if (STOCOpcode && !IndexReg && Subtarget.hasLoadStoreOnCond()) {
Richard Sandiforda68e6f52013-07-25 08:57:02 +00002789 if (Invert)
Richard Sandiford3d768e32013-07-31 12:30:20 +00002790 CCMask ^= CCValid;
Richard Sandiforda68e6f52013-07-25 08:57:02 +00002791 BuildMI(*MBB, MI, DL, TII->get(STOCOpcode))
Richard Sandifordfd7f4ae2013-08-01 10:39:40 +00002792 .addReg(SrcReg).addOperand(Base).addImm(Disp)
2793 .addImm(CCValid).addImm(CCMask);
Richard Sandiforda68e6f52013-07-25 08:57:02 +00002794 MI->eraseFromParent();
2795 return MBB;
2796 }
2797
Richard Sandifordb86a8342013-06-27 09:27:40 +00002798 // Get the condition needed to branch around the store.
2799 if (!Invert)
Richard Sandiford3d768e32013-07-31 12:30:20 +00002800 CCMask ^= CCValid;
Richard Sandifordb86a8342013-06-27 09:27:40 +00002801
2802 MachineBasicBlock *StartMBB = MBB;
Richard Sandiford5e318f02013-08-27 09:54:29 +00002803 MachineBasicBlock *JoinMBB = splitBlockBefore(MI, MBB);
Richard Sandifordb86a8342013-06-27 09:27:40 +00002804 MachineBasicBlock *FalseMBB = emitBlockAfter(StartMBB);
2805
2806 // StartMBB:
2807 // BRC CCMask, JoinMBB
2808 // # fallthrough to FalseMBB
Richard Sandifordb86a8342013-06-27 09:27:40 +00002809 MBB = StartMBB;
Richard Sandiford3d768e32013-07-31 12:30:20 +00002810 BuildMI(MBB, DL, TII->get(SystemZ::BRC))
2811 .addImm(CCValid).addImm(CCMask).addMBB(JoinMBB);
Richard Sandifordb86a8342013-06-27 09:27:40 +00002812 MBB->addSuccessor(JoinMBB);
2813 MBB->addSuccessor(FalseMBB);
2814
2815 // FalseMBB:
2816 // store %SrcReg, %Disp(%Index,%Base)
2817 // # fallthrough to JoinMBB
2818 MBB = FalseMBB;
2819 BuildMI(MBB, DL, TII->get(StoreOpcode))
2820 .addReg(SrcReg).addOperand(Base).addImm(Disp).addReg(IndexReg);
2821 MBB->addSuccessor(JoinMBB);
2822
2823 MI->eraseFromParent();
2824 return JoinMBB;
2825}
2826
Ulrich Weigand5f613df2013-05-06 16:15:19 +00002827// Implement EmitInstrWithCustomInserter for pseudo ATOMIC_LOAD{,W}_*
2828// or ATOMIC_SWAP{,W} instruction MI. BinOpcode is the instruction that
2829// performs the binary operation elided by "*", or 0 for ATOMIC_SWAP{,W}.
2830// BitSize is the width of the field in bits, or 0 if this is a partword
2831// ATOMIC_LOADW_* or ATOMIC_SWAPW instruction, in which case the bitsize
2832// is one of the operands. Invert says whether the field should be
2833// inverted after performing BinOpcode (e.g. for NAND).
2834MachineBasicBlock *
2835SystemZTargetLowering::emitAtomicLoadBinary(MachineInstr *MI,
2836 MachineBasicBlock *MBB,
2837 unsigned BinOpcode,
2838 unsigned BitSize,
2839 bool Invert) const {
Ulrich Weigand5f613df2013-05-06 16:15:19 +00002840 MachineFunction &MF = *MBB->getParent();
Eric Christopherfc6de422014-08-05 02:39:49 +00002841 const SystemZInstrInfo *TII =
Eric Christophera6734172015-01-31 00:06:45 +00002842 static_cast<const SystemZInstrInfo *>(Subtarget.getInstrInfo());
Ulrich Weigand5f613df2013-05-06 16:15:19 +00002843 MachineRegisterInfo &MRI = MF.getRegInfo();
Ulrich Weigand5f613df2013-05-06 16:15:19 +00002844 bool IsSubWord = (BitSize < 32);
2845
2846 // Extract the operands. Base can be a register or a frame index.
2847 // Src2 can be a register or immediate.
2848 unsigned Dest = MI->getOperand(0).getReg();
2849 MachineOperand Base = earlyUseOperand(MI->getOperand(1));
2850 int64_t Disp = MI->getOperand(2).getImm();
2851 MachineOperand Src2 = earlyUseOperand(MI->getOperand(3));
2852 unsigned BitShift = (IsSubWord ? MI->getOperand(4).getReg() : 0);
2853 unsigned NegBitShift = (IsSubWord ? MI->getOperand(5).getReg() : 0);
2854 DebugLoc DL = MI->getDebugLoc();
2855 if (IsSubWord)
2856 BitSize = MI->getOperand(6).getImm();
2857
2858 // Subword operations use 32-bit registers.
2859 const TargetRegisterClass *RC = (BitSize <= 32 ?
2860 &SystemZ::GR32BitRegClass :
2861 &SystemZ::GR64BitRegClass);
2862 unsigned LOpcode = BitSize <= 32 ? SystemZ::L : SystemZ::LG;
2863 unsigned CSOpcode = BitSize <= 32 ? SystemZ::CS : SystemZ::CSG;
2864
2865 // Get the right opcodes for the displacement.
2866 LOpcode = TII->getOpcodeForOffset(LOpcode, Disp);
2867 CSOpcode = TII->getOpcodeForOffset(CSOpcode, Disp);
2868 assert(LOpcode && CSOpcode && "Displacement out of range");
2869
2870 // Create virtual registers for temporary results.
2871 unsigned OrigVal = MRI.createVirtualRegister(RC);
2872 unsigned OldVal = MRI.createVirtualRegister(RC);
2873 unsigned NewVal = (BinOpcode || IsSubWord ?
2874 MRI.createVirtualRegister(RC) : Src2.getReg());
2875 unsigned RotatedOldVal = (IsSubWord ? MRI.createVirtualRegister(RC) : OldVal);
2876 unsigned RotatedNewVal = (IsSubWord ? MRI.createVirtualRegister(RC) : NewVal);
2877
2878 // Insert a basic block for the main loop.
2879 MachineBasicBlock *StartMBB = MBB;
Richard Sandiford5e318f02013-08-27 09:54:29 +00002880 MachineBasicBlock *DoneMBB = splitBlockBefore(MI, MBB);
Ulrich Weigand5f613df2013-05-06 16:15:19 +00002881 MachineBasicBlock *LoopMBB = emitBlockAfter(StartMBB);
2882
2883 // StartMBB:
2884 // ...
2885 // %OrigVal = L Disp(%Base)
2886 // # fall through to LoopMMB
2887 MBB = StartMBB;
2888 BuildMI(MBB, DL, TII->get(LOpcode), OrigVal)
2889 .addOperand(Base).addImm(Disp).addReg(0);
2890 MBB->addSuccessor(LoopMBB);
2891
2892 // LoopMBB:
2893 // %OldVal = phi [ %OrigVal, StartMBB ], [ %Dest, LoopMBB ]
2894 // %RotatedOldVal = RLL %OldVal, 0(%BitShift)
2895 // %RotatedNewVal = OP %RotatedOldVal, %Src2
2896 // %NewVal = RLL %RotatedNewVal, 0(%NegBitShift)
2897 // %Dest = CS %OldVal, %NewVal, Disp(%Base)
2898 // JNE LoopMBB
2899 // # fall through to DoneMMB
2900 MBB = LoopMBB;
2901 BuildMI(MBB, DL, TII->get(SystemZ::PHI), OldVal)
2902 .addReg(OrigVal).addMBB(StartMBB)
2903 .addReg(Dest).addMBB(LoopMBB);
2904 if (IsSubWord)
2905 BuildMI(MBB, DL, TII->get(SystemZ::RLL), RotatedOldVal)
2906 .addReg(OldVal).addReg(BitShift).addImm(0);
2907 if (Invert) {
2908 // Perform the operation normally and then invert every bit of the field.
2909 unsigned Tmp = MRI.createVirtualRegister(RC);
2910 BuildMI(MBB, DL, TII->get(BinOpcode), Tmp)
2911 .addReg(RotatedOldVal).addOperand(Src2);
Alexey Samsonovfffd56ec2014-08-20 21:56:43 +00002912 if (BitSize <= 32)
Ulrich Weigand5f613df2013-05-06 16:15:19 +00002913 // XILF with the upper BitSize bits set.
Richard Sandiford652784e2013-09-25 11:11:53 +00002914 BuildMI(MBB, DL, TII->get(SystemZ::XILF), RotatedNewVal)
Alexey Samsonovfffd56ec2014-08-20 21:56:43 +00002915 .addReg(Tmp).addImm(-1U << (32 - BitSize));
Ulrich Weigand5f613df2013-05-06 16:15:19 +00002916 else {
2917 // Use LCGR and add -1 to the result, which is more compact than
2918 // an XILF, XILH pair.
2919 unsigned Tmp2 = MRI.createVirtualRegister(RC);
2920 BuildMI(MBB, DL, TII->get(SystemZ::LCGR), Tmp2).addReg(Tmp);
2921 BuildMI(MBB, DL, TII->get(SystemZ::AGHI), RotatedNewVal)
2922 .addReg(Tmp2).addImm(-1);
2923 }
2924 } else if (BinOpcode)
2925 // A simply binary operation.
2926 BuildMI(MBB, DL, TII->get(BinOpcode), RotatedNewVal)
2927 .addReg(RotatedOldVal).addOperand(Src2);
2928 else if (IsSubWord)
2929 // Use RISBG to rotate Src2 into position and use it to replace the
2930 // field in RotatedOldVal.
2931 BuildMI(MBB, DL, TII->get(SystemZ::RISBG32), RotatedNewVal)
2932 .addReg(RotatedOldVal).addReg(Src2.getReg())
2933 .addImm(32).addImm(31 + BitSize).addImm(32 - BitSize);
2934 if (IsSubWord)
2935 BuildMI(MBB, DL, TII->get(SystemZ::RLL), NewVal)
2936 .addReg(RotatedNewVal).addReg(NegBitShift).addImm(0);
2937 BuildMI(MBB, DL, TII->get(CSOpcode), Dest)
2938 .addReg(OldVal).addReg(NewVal).addOperand(Base).addImm(Disp);
Richard Sandiford3d768e32013-07-31 12:30:20 +00002939 BuildMI(MBB, DL, TII->get(SystemZ::BRC))
2940 .addImm(SystemZ::CCMASK_CS).addImm(SystemZ::CCMASK_CS_NE).addMBB(LoopMBB);
Ulrich Weigand5f613df2013-05-06 16:15:19 +00002941 MBB->addSuccessor(LoopMBB);
2942 MBB->addSuccessor(DoneMBB);
2943
2944 MI->eraseFromParent();
2945 return DoneMBB;
2946}
2947
2948// Implement EmitInstrWithCustomInserter for pseudo
2949// ATOMIC_LOAD{,W}_{,U}{MIN,MAX} instruction MI. CompareOpcode is the
2950// instruction that should be used to compare the current field with the
2951// minimum or maximum value. KeepOldMask is the BRC condition-code mask
2952// for when the current field should be kept. BitSize is the width of
2953// the field in bits, or 0 if this is a partword ATOMIC_LOADW_* instruction.
2954MachineBasicBlock *
2955SystemZTargetLowering::emitAtomicLoadMinMax(MachineInstr *MI,
2956 MachineBasicBlock *MBB,
2957 unsigned CompareOpcode,
2958 unsigned KeepOldMask,
2959 unsigned BitSize) const {
Ulrich Weigand5f613df2013-05-06 16:15:19 +00002960 MachineFunction &MF = *MBB->getParent();
Eric Christopherfc6de422014-08-05 02:39:49 +00002961 const SystemZInstrInfo *TII =
Eric Christophera6734172015-01-31 00:06:45 +00002962 static_cast<const SystemZInstrInfo *>(Subtarget.getInstrInfo());
Ulrich Weigand5f613df2013-05-06 16:15:19 +00002963 MachineRegisterInfo &MRI = MF.getRegInfo();
Ulrich Weigand5f613df2013-05-06 16:15:19 +00002964 bool IsSubWord = (BitSize < 32);
2965
2966 // Extract the operands. Base can be a register or a frame index.
2967 unsigned Dest = MI->getOperand(0).getReg();
2968 MachineOperand Base = earlyUseOperand(MI->getOperand(1));
2969 int64_t Disp = MI->getOperand(2).getImm();
2970 unsigned Src2 = MI->getOperand(3).getReg();
2971 unsigned BitShift = (IsSubWord ? MI->getOperand(4).getReg() : 0);
2972 unsigned NegBitShift = (IsSubWord ? MI->getOperand(5).getReg() : 0);
2973 DebugLoc DL = MI->getDebugLoc();
2974 if (IsSubWord)
2975 BitSize = MI->getOperand(6).getImm();
2976
2977 // Subword operations use 32-bit registers.
2978 const TargetRegisterClass *RC = (BitSize <= 32 ?
2979 &SystemZ::GR32BitRegClass :
2980 &SystemZ::GR64BitRegClass);
2981 unsigned LOpcode = BitSize <= 32 ? SystemZ::L : SystemZ::LG;
2982 unsigned CSOpcode = BitSize <= 32 ? SystemZ::CS : SystemZ::CSG;
2983
2984 // Get the right opcodes for the displacement.
2985 LOpcode = TII->getOpcodeForOffset(LOpcode, Disp);
2986 CSOpcode = TII->getOpcodeForOffset(CSOpcode, Disp);
2987 assert(LOpcode && CSOpcode && "Displacement out of range");
2988
2989 // Create virtual registers for temporary results.
2990 unsigned OrigVal = MRI.createVirtualRegister(RC);
2991 unsigned OldVal = MRI.createVirtualRegister(RC);
2992 unsigned NewVal = MRI.createVirtualRegister(RC);
2993 unsigned RotatedOldVal = (IsSubWord ? MRI.createVirtualRegister(RC) : OldVal);
2994 unsigned RotatedAltVal = (IsSubWord ? MRI.createVirtualRegister(RC) : Src2);
2995 unsigned RotatedNewVal = (IsSubWord ? MRI.createVirtualRegister(RC) : NewVal);
2996
2997 // Insert 3 basic blocks for the loop.
2998 MachineBasicBlock *StartMBB = MBB;
Richard Sandiford5e318f02013-08-27 09:54:29 +00002999 MachineBasicBlock *DoneMBB = splitBlockBefore(MI, MBB);
Ulrich Weigand5f613df2013-05-06 16:15:19 +00003000 MachineBasicBlock *LoopMBB = emitBlockAfter(StartMBB);
3001 MachineBasicBlock *UseAltMBB = emitBlockAfter(LoopMBB);
3002 MachineBasicBlock *UpdateMBB = emitBlockAfter(UseAltMBB);
3003
3004 // StartMBB:
3005 // ...
3006 // %OrigVal = L Disp(%Base)
3007 // # fall through to LoopMMB
3008 MBB = StartMBB;
3009 BuildMI(MBB, DL, TII->get(LOpcode), OrigVal)
3010 .addOperand(Base).addImm(Disp).addReg(0);
3011 MBB->addSuccessor(LoopMBB);
3012
3013 // LoopMBB:
3014 // %OldVal = phi [ %OrigVal, StartMBB ], [ %Dest, UpdateMBB ]
3015 // %RotatedOldVal = RLL %OldVal, 0(%BitShift)
3016 // CompareOpcode %RotatedOldVal, %Src2
Richard Sandiford312425f2013-05-20 14:23:08 +00003017 // BRC KeepOldMask, UpdateMBB
Ulrich Weigand5f613df2013-05-06 16:15:19 +00003018 MBB = LoopMBB;
3019 BuildMI(MBB, DL, TII->get(SystemZ::PHI), OldVal)
3020 .addReg(OrigVal).addMBB(StartMBB)
3021 .addReg(Dest).addMBB(UpdateMBB);
3022 if (IsSubWord)
3023 BuildMI(MBB, DL, TII->get(SystemZ::RLL), RotatedOldVal)
3024 .addReg(OldVal).addReg(BitShift).addImm(0);
Richard Sandiford8a757bb2013-07-31 12:11:07 +00003025 BuildMI(MBB, DL, TII->get(CompareOpcode))
3026 .addReg(RotatedOldVal).addReg(Src2);
3027 BuildMI(MBB, DL, TII->get(SystemZ::BRC))
Richard Sandiford3d768e32013-07-31 12:30:20 +00003028 .addImm(SystemZ::CCMASK_ICMP).addImm(KeepOldMask).addMBB(UpdateMBB);
Ulrich Weigand5f613df2013-05-06 16:15:19 +00003029 MBB->addSuccessor(UpdateMBB);
3030 MBB->addSuccessor(UseAltMBB);
3031
3032 // UseAltMBB:
3033 // %RotatedAltVal = RISBG %RotatedOldVal, %Src2, 32, 31 + BitSize, 0
3034 // # fall through to UpdateMMB
3035 MBB = UseAltMBB;
3036 if (IsSubWord)
3037 BuildMI(MBB, DL, TII->get(SystemZ::RISBG32), RotatedAltVal)
3038 .addReg(RotatedOldVal).addReg(Src2)
3039 .addImm(32).addImm(31 + BitSize).addImm(0);
3040 MBB->addSuccessor(UpdateMBB);
3041
3042 // UpdateMBB:
3043 // %RotatedNewVal = PHI [ %RotatedOldVal, LoopMBB ],
3044 // [ %RotatedAltVal, UseAltMBB ]
3045 // %NewVal = RLL %RotatedNewVal, 0(%NegBitShift)
3046 // %Dest = CS %OldVal, %NewVal, Disp(%Base)
3047 // JNE LoopMBB
3048 // # fall through to DoneMMB
3049 MBB = UpdateMBB;
3050 BuildMI(MBB, DL, TII->get(SystemZ::PHI), RotatedNewVal)
3051 .addReg(RotatedOldVal).addMBB(LoopMBB)
3052 .addReg(RotatedAltVal).addMBB(UseAltMBB);
3053 if (IsSubWord)
3054 BuildMI(MBB, DL, TII->get(SystemZ::RLL), NewVal)
3055 .addReg(RotatedNewVal).addReg(NegBitShift).addImm(0);
3056 BuildMI(MBB, DL, TII->get(CSOpcode), Dest)
3057 .addReg(OldVal).addReg(NewVal).addOperand(Base).addImm(Disp);
Richard Sandiford3d768e32013-07-31 12:30:20 +00003058 BuildMI(MBB, DL, TII->get(SystemZ::BRC))
3059 .addImm(SystemZ::CCMASK_CS).addImm(SystemZ::CCMASK_CS_NE).addMBB(LoopMBB);
Ulrich Weigand5f613df2013-05-06 16:15:19 +00003060 MBB->addSuccessor(LoopMBB);
3061 MBB->addSuccessor(DoneMBB);
3062
3063 MI->eraseFromParent();
3064 return DoneMBB;
3065}
3066
3067// Implement EmitInstrWithCustomInserter for pseudo ATOMIC_CMP_SWAPW
3068// instruction MI.
3069MachineBasicBlock *
3070SystemZTargetLowering::emitAtomicCmpSwapW(MachineInstr *MI,
3071 MachineBasicBlock *MBB) const {
Ulrich Weigand5f613df2013-05-06 16:15:19 +00003072 MachineFunction &MF = *MBB->getParent();
Eric Christopherfc6de422014-08-05 02:39:49 +00003073 const SystemZInstrInfo *TII =
Eric Christophera6734172015-01-31 00:06:45 +00003074 static_cast<const SystemZInstrInfo *>(Subtarget.getInstrInfo());
Ulrich Weigand5f613df2013-05-06 16:15:19 +00003075 MachineRegisterInfo &MRI = MF.getRegInfo();
Ulrich Weigand5f613df2013-05-06 16:15:19 +00003076
3077 // Extract the operands. Base can be a register or a frame index.
3078 unsigned Dest = MI->getOperand(0).getReg();
3079 MachineOperand Base = earlyUseOperand(MI->getOperand(1));
3080 int64_t Disp = MI->getOperand(2).getImm();
3081 unsigned OrigCmpVal = MI->getOperand(3).getReg();
3082 unsigned OrigSwapVal = MI->getOperand(4).getReg();
3083 unsigned BitShift = MI->getOperand(5).getReg();
3084 unsigned NegBitShift = MI->getOperand(6).getReg();
3085 int64_t BitSize = MI->getOperand(7).getImm();
3086 DebugLoc DL = MI->getDebugLoc();
3087
3088 const TargetRegisterClass *RC = &SystemZ::GR32BitRegClass;
3089
3090 // Get the right opcodes for the displacement.
3091 unsigned LOpcode = TII->getOpcodeForOffset(SystemZ::L, Disp);
3092 unsigned CSOpcode = TII->getOpcodeForOffset(SystemZ::CS, Disp);
3093 assert(LOpcode && CSOpcode && "Displacement out of range");
3094
3095 // Create virtual registers for temporary results.
3096 unsigned OrigOldVal = MRI.createVirtualRegister(RC);
3097 unsigned OldVal = MRI.createVirtualRegister(RC);
3098 unsigned CmpVal = MRI.createVirtualRegister(RC);
3099 unsigned SwapVal = MRI.createVirtualRegister(RC);
3100 unsigned StoreVal = MRI.createVirtualRegister(RC);
3101 unsigned RetryOldVal = MRI.createVirtualRegister(RC);
3102 unsigned RetryCmpVal = MRI.createVirtualRegister(RC);
3103 unsigned RetrySwapVal = MRI.createVirtualRegister(RC);
3104
3105 // Insert 2 basic blocks for the loop.
3106 MachineBasicBlock *StartMBB = MBB;
Richard Sandiford5e318f02013-08-27 09:54:29 +00003107 MachineBasicBlock *DoneMBB = splitBlockBefore(MI, MBB);
Ulrich Weigand5f613df2013-05-06 16:15:19 +00003108 MachineBasicBlock *LoopMBB = emitBlockAfter(StartMBB);
3109 MachineBasicBlock *SetMBB = emitBlockAfter(LoopMBB);
3110
3111 // StartMBB:
3112 // ...
3113 // %OrigOldVal = L Disp(%Base)
3114 // # fall through to LoopMMB
3115 MBB = StartMBB;
3116 BuildMI(MBB, DL, TII->get(LOpcode), OrigOldVal)
3117 .addOperand(Base).addImm(Disp).addReg(0);
3118 MBB->addSuccessor(LoopMBB);
3119
3120 // LoopMBB:
3121 // %OldVal = phi [ %OrigOldVal, EntryBB ], [ %RetryOldVal, SetMBB ]
3122 // %CmpVal = phi [ %OrigCmpVal, EntryBB ], [ %RetryCmpVal, SetMBB ]
3123 // %SwapVal = phi [ %OrigSwapVal, EntryBB ], [ %RetrySwapVal, SetMBB ]
3124 // %Dest = RLL %OldVal, BitSize(%BitShift)
3125 // ^^ The low BitSize bits contain the field
3126 // of interest.
3127 // %RetryCmpVal = RISBG32 %CmpVal, %Dest, 32, 63-BitSize, 0
3128 // ^^ Replace the upper 32-BitSize bits of the
3129 // comparison value with those that we loaded,
3130 // so that we can use a full word comparison.
Richard Sandiford8a757bb2013-07-31 12:11:07 +00003131 // CR %Dest, %RetryCmpVal
3132 // JNE DoneMBB
Ulrich Weigand5f613df2013-05-06 16:15:19 +00003133 // # Fall through to SetMBB
3134 MBB = LoopMBB;
3135 BuildMI(MBB, DL, TII->get(SystemZ::PHI), OldVal)
3136 .addReg(OrigOldVal).addMBB(StartMBB)
3137 .addReg(RetryOldVal).addMBB(SetMBB);
3138 BuildMI(MBB, DL, TII->get(SystemZ::PHI), CmpVal)
3139 .addReg(OrigCmpVal).addMBB(StartMBB)
3140 .addReg(RetryCmpVal).addMBB(SetMBB);
3141 BuildMI(MBB, DL, TII->get(SystemZ::PHI), SwapVal)
3142 .addReg(OrigSwapVal).addMBB(StartMBB)
3143 .addReg(RetrySwapVal).addMBB(SetMBB);
3144 BuildMI(MBB, DL, TII->get(SystemZ::RLL), Dest)
3145 .addReg(OldVal).addReg(BitShift).addImm(BitSize);
3146 BuildMI(MBB, DL, TII->get(SystemZ::RISBG32), RetryCmpVal)
3147 .addReg(CmpVal).addReg(Dest).addImm(32).addImm(63 - BitSize).addImm(0);
Richard Sandiford8a757bb2013-07-31 12:11:07 +00003148 BuildMI(MBB, DL, TII->get(SystemZ::CR))
3149 .addReg(Dest).addReg(RetryCmpVal);
3150 BuildMI(MBB, DL, TII->get(SystemZ::BRC))
Richard Sandiford3d768e32013-07-31 12:30:20 +00003151 .addImm(SystemZ::CCMASK_ICMP)
3152 .addImm(SystemZ::CCMASK_CMP_NE).addMBB(DoneMBB);
Ulrich Weigand5f613df2013-05-06 16:15:19 +00003153 MBB->addSuccessor(DoneMBB);
3154 MBB->addSuccessor(SetMBB);
3155
3156 // SetMBB:
3157 // %RetrySwapVal = RISBG32 %SwapVal, %Dest, 32, 63-BitSize, 0
3158 // ^^ Replace the upper 32-BitSize bits of the new
3159 // value with those that we loaded.
3160 // %StoreVal = RLL %RetrySwapVal, -BitSize(%NegBitShift)
3161 // ^^ Rotate the new field to its proper position.
3162 // %RetryOldVal = CS %Dest, %StoreVal, Disp(%Base)
3163 // JNE LoopMBB
3164 // # fall through to ExitMMB
3165 MBB = SetMBB;
3166 BuildMI(MBB, DL, TII->get(SystemZ::RISBG32), RetrySwapVal)
3167 .addReg(SwapVal).addReg(Dest).addImm(32).addImm(63 - BitSize).addImm(0);
3168 BuildMI(MBB, DL, TII->get(SystemZ::RLL), StoreVal)
3169 .addReg(RetrySwapVal).addReg(NegBitShift).addImm(-BitSize);
3170 BuildMI(MBB, DL, TII->get(CSOpcode), RetryOldVal)
3171 .addReg(OldVal).addReg(StoreVal).addOperand(Base).addImm(Disp);
Richard Sandiford3d768e32013-07-31 12:30:20 +00003172 BuildMI(MBB, DL, TII->get(SystemZ::BRC))
3173 .addImm(SystemZ::CCMASK_CS).addImm(SystemZ::CCMASK_CS_NE).addMBB(LoopMBB);
Ulrich Weigand5f613df2013-05-06 16:15:19 +00003174 MBB->addSuccessor(LoopMBB);
3175 MBB->addSuccessor(DoneMBB);
3176
3177 MI->eraseFromParent();
3178 return DoneMBB;
3179}
3180
3181// Emit an extension from a GR32 or GR64 to a GR128. ClearEven is true
3182// if the high register of the GR128 value must be cleared or false if
Richard Sandiford87a44362013-09-30 10:28:35 +00003183// it's "don't care". SubReg is subreg_l32 when extending a GR32
3184// and subreg_l64 when extending a GR64.
Ulrich Weigand5f613df2013-05-06 16:15:19 +00003185MachineBasicBlock *
3186SystemZTargetLowering::emitExt128(MachineInstr *MI,
3187 MachineBasicBlock *MBB,
3188 bool ClearEven, unsigned SubReg) const {
Ulrich Weigand5f613df2013-05-06 16:15:19 +00003189 MachineFunction &MF = *MBB->getParent();
Eric Christopherfc6de422014-08-05 02:39:49 +00003190 const SystemZInstrInfo *TII =
Eric Christophera6734172015-01-31 00:06:45 +00003191 static_cast<const SystemZInstrInfo *>(Subtarget.getInstrInfo());
Ulrich Weigand5f613df2013-05-06 16:15:19 +00003192 MachineRegisterInfo &MRI = MF.getRegInfo();
3193 DebugLoc DL = MI->getDebugLoc();
3194
3195 unsigned Dest = MI->getOperand(0).getReg();
3196 unsigned Src = MI->getOperand(1).getReg();
3197 unsigned In128 = MRI.createVirtualRegister(&SystemZ::GR128BitRegClass);
3198
3199 BuildMI(*MBB, MI, DL, TII->get(TargetOpcode::IMPLICIT_DEF), In128);
3200 if (ClearEven) {
3201 unsigned NewIn128 = MRI.createVirtualRegister(&SystemZ::GR128BitRegClass);
3202 unsigned Zero64 = MRI.createVirtualRegister(&SystemZ::GR64BitRegClass);
3203
3204 BuildMI(*MBB, MI, DL, TII->get(SystemZ::LLILL), Zero64)
3205 .addImm(0);
3206 BuildMI(*MBB, MI, DL, TII->get(TargetOpcode::INSERT_SUBREG), NewIn128)
Richard Sandiford87a44362013-09-30 10:28:35 +00003207 .addReg(In128).addReg(Zero64).addImm(SystemZ::subreg_h64);
Ulrich Weigand5f613df2013-05-06 16:15:19 +00003208 In128 = NewIn128;
3209 }
3210 BuildMI(*MBB, MI, DL, TII->get(TargetOpcode::INSERT_SUBREG), Dest)
3211 .addReg(In128).addReg(Src).addImm(SubReg);
3212
3213 MI->eraseFromParent();
3214 return MBB;
3215}
3216
Richard Sandifordd131ff82013-07-08 09:35:23 +00003217MachineBasicBlock *
Richard Sandiford564681c2013-08-12 10:28:10 +00003218SystemZTargetLowering::emitMemMemWrapper(MachineInstr *MI,
3219 MachineBasicBlock *MBB,
3220 unsigned Opcode) const {
Richard Sandiford5e318f02013-08-27 09:54:29 +00003221 MachineFunction &MF = *MBB->getParent();
Eric Christopherfc6de422014-08-05 02:39:49 +00003222 const SystemZInstrInfo *TII =
Eric Christophera6734172015-01-31 00:06:45 +00003223 static_cast<const SystemZInstrInfo *>(Subtarget.getInstrInfo());
Richard Sandiford5e318f02013-08-27 09:54:29 +00003224 MachineRegisterInfo &MRI = MF.getRegInfo();
Richard Sandifordd131ff82013-07-08 09:35:23 +00003225 DebugLoc DL = MI->getDebugLoc();
3226
Richard Sandiford5e318f02013-08-27 09:54:29 +00003227 MachineOperand DestBase = earlyUseOperand(MI->getOperand(0));
Richard Sandifordd131ff82013-07-08 09:35:23 +00003228 uint64_t DestDisp = MI->getOperand(1).getImm();
Richard Sandiford5e318f02013-08-27 09:54:29 +00003229 MachineOperand SrcBase = earlyUseOperand(MI->getOperand(2));
Richard Sandifordd131ff82013-07-08 09:35:23 +00003230 uint64_t SrcDisp = MI->getOperand(3).getImm();
3231 uint64_t Length = MI->getOperand(4).getImm();
3232
Richard Sandifordbe133a82013-08-28 09:01:51 +00003233 // When generating more than one CLC, all but the last will need to
3234 // branch to the end when a difference is found.
3235 MachineBasicBlock *EndMBB = (Length > 256 && Opcode == SystemZ::CLC ?
Craig Topper062a2ba2014-04-25 05:30:21 +00003236 splitBlockAfter(MI, MBB) : nullptr);
Richard Sandifordbe133a82013-08-28 09:01:51 +00003237
Richard Sandiford5e318f02013-08-27 09:54:29 +00003238 // Check for the loop form, in which operand 5 is the trip count.
3239 if (MI->getNumExplicitOperands() > 5) {
3240 bool HaveSingleBase = DestBase.isIdenticalTo(SrcBase);
3241
3242 uint64_t StartCountReg = MI->getOperand(5).getReg();
3243 uint64_t StartSrcReg = forceReg(MI, SrcBase, TII);
3244 uint64_t StartDestReg = (HaveSingleBase ? StartSrcReg :
3245 forceReg(MI, DestBase, TII));
3246
3247 const TargetRegisterClass *RC = &SystemZ::ADDR64BitRegClass;
3248 uint64_t ThisSrcReg = MRI.createVirtualRegister(RC);
3249 uint64_t ThisDestReg = (HaveSingleBase ? ThisSrcReg :
3250 MRI.createVirtualRegister(RC));
3251 uint64_t NextSrcReg = MRI.createVirtualRegister(RC);
3252 uint64_t NextDestReg = (HaveSingleBase ? NextSrcReg :
3253 MRI.createVirtualRegister(RC));
3254
3255 RC = &SystemZ::GR64BitRegClass;
3256 uint64_t ThisCountReg = MRI.createVirtualRegister(RC);
3257 uint64_t NextCountReg = MRI.createVirtualRegister(RC);
3258
3259 MachineBasicBlock *StartMBB = MBB;
3260 MachineBasicBlock *DoneMBB = splitBlockBefore(MI, MBB);
3261 MachineBasicBlock *LoopMBB = emitBlockAfter(StartMBB);
Richard Sandifordbe133a82013-08-28 09:01:51 +00003262 MachineBasicBlock *NextMBB = (EndMBB ? emitBlockAfter(LoopMBB) : LoopMBB);
Richard Sandiford5e318f02013-08-27 09:54:29 +00003263
3264 // StartMBB:
3265 // # fall through to LoopMMB
3266 MBB->addSuccessor(LoopMBB);
3267
3268 // LoopMBB:
3269 // %ThisDestReg = phi [ %StartDestReg, StartMBB ],
Richard Sandifordbe133a82013-08-28 09:01:51 +00003270 // [ %NextDestReg, NextMBB ]
Richard Sandiford5e318f02013-08-27 09:54:29 +00003271 // %ThisSrcReg = phi [ %StartSrcReg, StartMBB ],
Richard Sandifordbe133a82013-08-28 09:01:51 +00003272 // [ %NextSrcReg, NextMBB ]
Richard Sandiford5e318f02013-08-27 09:54:29 +00003273 // %ThisCountReg = phi [ %StartCountReg, StartMBB ],
Richard Sandifordbe133a82013-08-28 09:01:51 +00003274 // [ %NextCountReg, NextMBB ]
3275 // ( PFD 2, 768+DestDisp(%ThisDestReg) )
Richard Sandiford5e318f02013-08-27 09:54:29 +00003276 // Opcode DestDisp(256,%ThisDestReg), SrcDisp(%ThisSrcReg)
Richard Sandifordbe133a82013-08-28 09:01:51 +00003277 // ( JLH EndMBB )
3278 //
3279 // The prefetch is used only for MVC. The JLH is used only for CLC.
3280 MBB = LoopMBB;
3281
3282 BuildMI(MBB, DL, TII->get(SystemZ::PHI), ThisDestReg)
3283 .addReg(StartDestReg).addMBB(StartMBB)
3284 .addReg(NextDestReg).addMBB(NextMBB);
3285 if (!HaveSingleBase)
3286 BuildMI(MBB, DL, TII->get(SystemZ::PHI), ThisSrcReg)
3287 .addReg(StartSrcReg).addMBB(StartMBB)
3288 .addReg(NextSrcReg).addMBB(NextMBB);
3289 BuildMI(MBB, DL, TII->get(SystemZ::PHI), ThisCountReg)
3290 .addReg(StartCountReg).addMBB(StartMBB)
3291 .addReg(NextCountReg).addMBB(NextMBB);
3292 if (Opcode == SystemZ::MVC)
3293 BuildMI(MBB, DL, TII->get(SystemZ::PFD))
3294 .addImm(SystemZ::PFD_WRITE)
3295 .addReg(ThisDestReg).addImm(DestDisp + 768).addReg(0);
3296 BuildMI(MBB, DL, TII->get(Opcode))
3297 .addReg(ThisDestReg).addImm(DestDisp).addImm(256)
3298 .addReg(ThisSrcReg).addImm(SrcDisp);
3299 if (EndMBB) {
3300 BuildMI(MBB, DL, TII->get(SystemZ::BRC))
3301 .addImm(SystemZ::CCMASK_ICMP).addImm(SystemZ::CCMASK_CMP_NE)
3302 .addMBB(EndMBB);
3303 MBB->addSuccessor(EndMBB);
3304 MBB->addSuccessor(NextMBB);
3305 }
3306
3307 // NextMBB:
Richard Sandiford5e318f02013-08-27 09:54:29 +00003308 // %NextDestReg = LA 256(%ThisDestReg)
3309 // %NextSrcReg = LA 256(%ThisSrcReg)
3310 // %NextCountReg = AGHI %ThisCountReg, -1
3311 // CGHI %NextCountReg, 0
3312 // JLH LoopMBB
3313 // # fall through to DoneMMB
3314 //
3315 // The AGHI, CGHI and JLH should be converted to BRCTG by later passes.
Richard Sandifordbe133a82013-08-28 09:01:51 +00003316 MBB = NextMBB;
Richard Sandiford5e318f02013-08-27 09:54:29 +00003317
Richard Sandiford5e318f02013-08-27 09:54:29 +00003318 BuildMI(MBB, DL, TII->get(SystemZ::LA), NextDestReg)
3319 .addReg(ThisDestReg).addImm(256).addReg(0);
3320 if (!HaveSingleBase)
3321 BuildMI(MBB, DL, TII->get(SystemZ::LA), NextSrcReg)
3322 .addReg(ThisSrcReg).addImm(256).addReg(0);
3323 BuildMI(MBB, DL, TII->get(SystemZ::AGHI), NextCountReg)
3324 .addReg(ThisCountReg).addImm(-1);
3325 BuildMI(MBB, DL, TII->get(SystemZ::CGHI))
3326 .addReg(NextCountReg).addImm(0);
3327 BuildMI(MBB, DL, TII->get(SystemZ::BRC))
3328 .addImm(SystemZ::CCMASK_ICMP).addImm(SystemZ::CCMASK_CMP_NE)
3329 .addMBB(LoopMBB);
3330 MBB->addSuccessor(LoopMBB);
3331 MBB->addSuccessor(DoneMBB);
3332
3333 DestBase = MachineOperand::CreateReg(NextDestReg, false);
3334 SrcBase = MachineOperand::CreateReg(NextSrcReg, false);
3335 Length &= 255;
3336 MBB = DoneMBB;
3337 }
3338 // Handle any remaining bytes with straight-line code.
3339 while (Length > 0) {
3340 uint64_t ThisLength = std::min(Length, uint64_t(256));
3341 // The previous iteration might have created out-of-range displacements.
3342 // Apply them using LAY if so.
3343 if (!isUInt<12>(DestDisp)) {
3344 unsigned Reg = MRI.createVirtualRegister(&SystemZ::ADDR64BitRegClass);
3345 BuildMI(*MBB, MI, MI->getDebugLoc(), TII->get(SystemZ::LAY), Reg)
3346 .addOperand(DestBase).addImm(DestDisp).addReg(0);
3347 DestBase = MachineOperand::CreateReg(Reg, false);
3348 DestDisp = 0;
3349 }
3350 if (!isUInt<12>(SrcDisp)) {
3351 unsigned Reg = MRI.createVirtualRegister(&SystemZ::ADDR64BitRegClass);
3352 BuildMI(*MBB, MI, MI->getDebugLoc(), TII->get(SystemZ::LAY), Reg)
3353 .addOperand(SrcBase).addImm(SrcDisp).addReg(0);
3354 SrcBase = MachineOperand::CreateReg(Reg, false);
3355 SrcDisp = 0;
3356 }
3357 BuildMI(*MBB, MI, DL, TII->get(Opcode))
3358 .addOperand(DestBase).addImm(DestDisp).addImm(ThisLength)
3359 .addOperand(SrcBase).addImm(SrcDisp);
3360 DestDisp += ThisLength;
3361 SrcDisp += ThisLength;
3362 Length -= ThisLength;
Richard Sandifordbe133a82013-08-28 09:01:51 +00003363 // If there's another CLC to go, branch to the end if a difference
3364 // was found.
3365 if (EndMBB && Length > 0) {
3366 MachineBasicBlock *NextMBB = splitBlockBefore(MI, MBB);
3367 BuildMI(MBB, DL, TII->get(SystemZ::BRC))
3368 .addImm(SystemZ::CCMASK_ICMP).addImm(SystemZ::CCMASK_CMP_NE)
3369 .addMBB(EndMBB);
3370 MBB->addSuccessor(EndMBB);
3371 MBB->addSuccessor(NextMBB);
3372 MBB = NextMBB;
3373 }
3374 }
3375 if (EndMBB) {
3376 MBB->addSuccessor(EndMBB);
3377 MBB = EndMBB;
3378 MBB->addLiveIn(SystemZ::CC);
Richard Sandiford5e318f02013-08-27 09:54:29 +00003379 }
Richard Sandifordd131ff82013-07-08 09:35:23 +00003380
3381 MI->eraseFromParent();
3382 return MBB;
3383}
3384
Richard Sandifordca232712013-08-16 11:21:54 +00003385// Decompose string pseudo-instruction MI into a loop that continually performs
3386// Opcode until CC != 3.
3387MachineBasicBlock *
3388SystemZTargetLowering::emitStringWrapper(MachineInstr *MI,
3389 MachineBasicBlock *MBB,
3390 unsigned Opcode) const {
Richard Sandifordca232712013-08-16 11:21:54 +00003391 MachineFunction &MF = *MBB->getParent();
Eric Christopherfc6de422014-08-05 02:39:49 +00003392 const SystemZInstrInfo *TII =
Eric Christophera6734172015-01-31 00:06:45 +00003393 static_cast<const SystemZInstrInfo *>(Subtarget.getInstrInfo());
Richard Sandifordca232712013-08-16 11:21:54 +00003394 MachineRegisterInfo &MRI = MF.getRegInfo();
3395 DebugLoc DL = MI->getDebugLoc();
3396
3397 uint64_t End1Reg = MI->getOperand(0).getReg();
3398 uint64_t Start1Reg = MI->getOperand(1).getReg();
3399 uint64_t Start2Reg = MI->getOperand(2).getReg();
3400 uint64_t CharReg = MI->getOperand(3).getReg();
3401
3402 const TargetRegisterClass *RC = &SystemZ::GR64BitRegClass;
3403 uint64_t This1Reg = MRI.createVirtualRegister(RC);
3404 uint64_t This2Reg = MRI.createVirtualRegister(RC);
3405 uint64_t End2Reg = MRI.createVirtualRegister(RC);
3406
3407 MachineBasicBlock *StartMBB = MBB;
Richard Sandiford5e318f02013-08-27 09:54:29 +00003408 MachineBasicBlock *DoneMBB = splitBlockBefore(MI, MBB);
Richard Sandifordca232712013-08-16 11:21:54 +00003409 MachineBasicBlock *LoopMBB = emitBlockAfter(StartMBB);
3410
3411 // StartMBB:
Richard Sandifordca232712013-08-16 11:21:54 +00003412 // # fall through to LoopMMB
Richard Sandifordca232712013-08-16 11:21:54 +00003413 MBB->addSuccessor(LoopMBB);
3414
3415 // LoopMBB:
3416 // %This1Reg = phi [ %Start1Reg, StartMBB ], [ %End1Reg, LoopMBB ]
3417 // %This2Reg = phi [ %Start2Reg, StartMBB ], [ %End2Reg, LoopMBB ]
Richard Sandiford7789b082013-09-30 08:48:38 +00003418 // R0L = %CharReg
3419 // %End1Reg, %End2Reg = CLST %This1Reg, %This2Reg -- uses R0L
Richard Sandifordca232712013-08-16 11:21:54 +00003420 // JO LoopMBB
3421 // # fall through to DoneMMB
Richard Sandiford6f6d5512013-08-20 09:38:48 +00003422 //
Richard Sandiford7789b082013-09-30 08:48:38 +00003423 // The load of R0L can be hoisted by post-RA LICM.
Richard Sandifordca232712013-08-16 11:21:54 +00003424 MBB = LoopMBB;
Richard Sandifordca232712013-08-16 11:21:54 +00003425
3426 BuildMI(MBB, DL, TII->get(SystemZ::PHI), This1Reg)
3427 .addReg(Start1Reg).addMBB(StartMBB)
3428 .addReg(End1Reg).addMBB(LoopMBB);
3429 BuildMI(MBB, DL, TII->get(SystemZ::PHI), This2Reg)
3430 .addReg(Start2Reg).addMBB(StartMBB)
3431 .addReg(End2Reg).addMBB(LoopMBB);
Richard Sandiford7789b082013-09-30 08:48:38 +00003432 BuildMI(MBB, DL, TII->get(TargetOpcode::COPY), SystemZ::R0L).addReg(CharReg);
Richard Sandifordca232712013-08-16 11:21:54 +00003433 BuildMI(MBB, DL, TII->get(Opcode))
3434 .addReg(End1Reg, RegState::Define).addReg(End2Reg, RegState::Define)
3435 .addReg(This1Reg).addReg(This2Reg);
3436 BuildMI(MBB, DL, TII->get(SystemZ::BRC))
3437 .addImm(SystemZ::CCMASK_ANY).addImm(SystemZ::CCMASK_3).addMBB(LoopMBB);
3438 MBB->addSuccessor(LoopMBB);
3439 MBB->addSuccessor(DoneMBB);
3440
3441 DoneMBB->addLiveIn(SystemZ::CC);
3442
3443 MI->eraseFromParent();
3444 return DoneMBB;
3445}
3446
Ulrich Weigand5f613df2013-05-06 16:15:19 +00003447MachineBasicBlock *SystemZTargetLowering::
3448EmitInstrWithCustomInserter(MachineInstr *MI, MachineBasicBlock *MBB) const {
3449 switch (MI->getOpcode()) {
Richard Sandiford7c5c0ea2013-10-01 13:10:16 +00003450 case SystemZ::Select32Mux:
Ulrich Weigand5f613df2013-05-06 16:15:19 +00003451 case SystemZ::Select32:
3452 case SystemZ::SelectF32:
3453 case SystemZ::Select64:
3454 case SystemZ::SelectF64:
3455 case SystemZ::SelectF128:
3456 return emitSelect(MI, MBB);
3457
Richard Sandiford2896d042013-10-01 14:33:55 +00003458 case SystemZ::CondStore8Mux:
3459 return emitCondStore(MI, MBB, SystemZ::STCMux, 0, false);
3460 case SystemZ::CondStore8MuxInv:
3461 return emitCondStore(MI, MBB, SystemZ::STCMux, 0, true);
3462 case SystemZ::CondStore16Mux:
3463 return emitCondStore(MI, MBB, SystemZ::STHMux, 0, false);
3464 case SystemZ::CondStore16MuxInv:
3465 return emitCondStore(MI, MBB, SystemZ::STHMux, 0, true);
Richard Sandifordb86a8342013-06-27 09:27:40 +00003466 case SystemZ::CondStore8:
Richard Sandiforda68e6f52013-07-25 08:57:02 +00003467 return emitCondStore(MI, MBB, SystemZ::STC, 0, false);
Richard Sandifordb86a8342013-06-27 09:27:40 +00003468 case SystemZ::CondStore8Inv:
Richard Sandiforda68e6f52013-07-25 08:57:02 +00003469 return emitCondStore(MI, MBB, SystemZ::STC, 0, true);
Richard Sandifordb86a8342013-06-27 09:27:40 +00003470 case SystemZ::CondStore16:
Richard Sandiforda68e6f52013-07-25 08:57:02 +00003471 return emitCondStore(MI, MBB, SystemZ::STH, 0, false);
Richard Sandifordb86a8342013-06-27 09:27:40 +00003472 case SystemZ::CondStore16Inv:
Richard Sandiforda68e6f52013-07-25 08:57:02 +00003473 return emitCondStore(MI, MBB, SystemZ::STH, 0, true);
Richard Sandifordb86a8342013-06-27 09:27:40 +00003474 case SystemZ::CondStore32:
Richard Sandiforda68e6f52013-07-25 08:57:02 +00003475 return emitCondStore(MI, MBB, SystemZ::ST, SystemZ::STOC, false);
Richard Sandifordb86a8342013-06-27 09:27:40 +00003476 case SystemZ::CondStore32Inv:
Richard Sandiforda68e6f52013-07-25 08:57:02 +00003477 return emitCondStore(MI, MBB, SystemZ::ST, SystemZ::STOC, true);
Richard Sandifordb86a8342013-06-27 09:27:40 +00003478 case SystemZ::CondStore64:
Richard Sandiforda68e6f52013-07-25 08:57:02 +00003479 return emitCondStore(MI, MBB, SystemZ::STG, SystemZ::STOCG, false);
Richard Sandifordb86a8342013-06-27 09:27:40 +00003480 case SystemZ::CondStore64Inv:
Richard Sandiforda68e6f52013-07-25 08:57:02 +00003481 return emitCondStore(MI, MBB, SystemZ::STG, SystemZ::STOCG, true);
Richard Sandifordb86a8342013-06-27 09:27:40 +00003482 case SystemZ::CondStoreF32:
Richard Sandiforda68e6f52013-07-25 08:57:02 +00003483 return emitCondStore(MI, MBB, SystemZ::STE, 0, false);
Richard Sandifordb86a8342013-06-27 09:27:40 +00003484 case SystemZ::CondStoreF32Inv:
Richard Sandiforda68e6f52013-07-25 08:57:02 +00003485 return emitCondStore(MI, MBB, SystemZ::STE, 0, true);
Richard Sandifordb86a8342013-06-27 09:27:40 +00003486 case SystemZ::CondStoreF64:
Richard Sandiforda68e6f52013-07-25 08:57:02 +00003487 return emitCondStore(MI, MBB, SystemZ::STD, 0, false);
Richard Sandifordb86a8342013-06-27 09:27:40 +00003488 case SystemZ::CondStoreF64Inv:
Richard Sandiforda68e6f52013-07-25 08:57:02 +00003489 return emitCondStore(MI, MBB, SystemZ::STD, 0, true);
Richard Sandifordb86a8342013-06-27 09:27:40 +00003490
Ulrich Weigand5f613df2013-05-06 16:15:19 +00003491 case SystemZ::AEXT128_64:
Richard Sandiford87a44362013-09-30 10:28:35 +00003492 return emitExt128(MI, MBB, false, SystemZ::subreg_l64);
Ulrich Weigand5f613df2013-05-06 16:15:19 +00003493 case SystemZ::ZEXT128_32:
Richard Sandiford87a44362013-09-30 10:28:35 +00003494 return emitExt128(MI, MBB, true, SystemZ::subreg_l32);
Ulrich Weigand5f613df2013-05-06 16:15:19 +00003495 case SystemZ::ZEXT128_64:
Richard Sandiford87a44362013-09-30 10:28:35 +00003496 return emitExt128(MI, MBB, true, SystemZ::subreg_l64);
Ulrich Weigand5f613df2013-05-06 16:15:19 +00003497
3498 case SystemZ::ATOMIC_SWAPW:
3499 return emitAtomicLoadBinary(MI, MBB, 0, 0);
3500 case SystemZ::ATOMIC_SWAP_32:
3501 return emitAtomicLoadBinary(MI, MBB, 0, 32);
3502 case SystemZ::ATOMIC_SWAP_64:
3503 return emitAtomicLoadBinary(MI, MBB, 0, 64);
3504
3505 case SystemZ::ATOMIC_LOADW_AR:
3506 return emitAtomicLoadBinary(MI, MBB, SystemZ::AR, 0);
3507 case SystemZ::ATOMIC_LOADW_AFI:
3508 return emitAtomicLoadBinary(MI, MBB, SystemZ::AFI, 0);
3509 case SystemZ::ATOMIC_LOAD_AR:
3510 return emitAtomicLoadBinary(MI, MBB, SystemZ::AR, 32);
3511 case SystemZ::ATOMIC_LOAD_AHI:
3512 return emitAtomicLoadBinary(MI, MBB, SystemZ::AHI, 32);
3513 case SystemZ::ATOMIC_LOAD_AFI:
3514 return emitAtomicLoadBinary(MI, MBB, SystemZ::AFI, 32);
3515 case SystemZ::ATOMIC_LOAD_AGR:
3516 return emitAtomicLoadBinary(MI, MBB, SystemZ::AGR, 64);
3517 case SystemZ::ATOMIC_LOAD_AGHI:
3518 return emitAtomicLoadBinary(MI, MBB, SystemZ::AGHI, 64);
3519 case SystemZ::ATOMIC_LOAD_AGFI:
3520 return emitAtomicLoadBinary(MI, MBB, SystemZ::AGFI, 64);
3521
3522 case SystemZ::ATOMIC_LOADW_SR:
3523 return emitAtomicLoadBinary(MI, MBB, SystemZ::SR, 0);
3524 case SystemZ::ATOMIC_LOAD_SR:
3525 return emitAtomicLoadBinary(MI, MBB, SystemZ::SR, 32);
3526 case SystemZ::ATOMIC_LOAD_SGR:
3527 return emitAtomicLoadBinary(MI, MBB, SystemZ::SGR, 64);
3528
3529 case SystemZ::ATOMIC_LOADW_NR:
3530 return emitAtomicLoadBinary(MI, MBB, SystemZ::NR, 0);
3531 case SystemZ::ATOMIC_LOADW_NILH:
Richard Sandiford652784e2013-09-25 11:11:53 +00003532 return emitAtomicLoadBinary(MI, MBB, SystemZ::NILH, 0);
Ulrich Weigand5f613df2013-05-06 16:15:19 +00003533 case SystemZ::ATOMIC_LOAD_NR:
3534 return emitAtomicLoadBinary(MI, MBB, SystemZ::NR, 32);
Richard Sandiford652784e2013-09-25 11:11:53 +00003535 case SystemZ::ATOMIC_LOAD_NILL:
3536 return emitAtomicLoadBinary(MI, MBB, SystemZ::NILL, 32);
3537 case SystemZ::ATOMIC_LOAD_NILH:
3538 return emitAtomicLoadBinary(MI, MBB, SystemZ::NILH, 32);
3539 case SystemZ::ATOMIC_LOAD_NILF:
3540 return emitAtomicLoadBinary(MI, MBB, SystemZ::NILF, 32);
Ulrich Weigand5f613df2013-05-06 16:15:19 +00003541 case SystemZ::ATOMIC_LOAD_NGR:
3542 return emitAtomicLoadBinary(MI, MBB, SystemZ::NGR, 64);
Richard Sandiford652784e2013-09-25 11:11:53 +00003543 case SystemZ::ATOMIC_LOAD_NILL64:
3544 return emitAtomicLoadBinary(MI, MBB, SystemZ::NILL64, 64);
3545 case SystemZ::ATOMIC_LOAD_NILH64:
3546 return emitAtomicLoadBinary(MI, MBB, SystemZ::NILH64, 64);
Richard Sandiford70284282013-10-01 14:20:41 +00003547 case SystemZ::ATOMIC_LOAD_NIHL64:
3548 return emitAtomicLoadBinary(MI, MBB, SystemZ::NIHL64, 64);
3549 case SystemZ::ATOMIC_LOAD_NIHH64:
3550 return emitAtomicLoadBinary(MI, MBB, SystemZ::NIHH64, 64);
Richard Sandiford652784e2013-09-25 11:11:53 +00003551 case SystemZ::ATOMIC_LOAD_NILF64:
3552 return emitAtomicLoadBinary(MI, MBB, SystemZ::NILF64, 64);
Richard Sandiford70284282013-10-01 14:20:41 +00003553 case SystemZ::ATOMIC_LOAD_NIHF64:
3554 return emitAtomicLoadBinary(MI, MBB, SystemZ::NIHF64, 64);
Ulrich Weigand5f613df2013-05-06 16:15:19 +00003555
3556 case SystemZ::ATOMIC_LOADW_OR:
3557 return emitAtomicLoadBinary(MI, MBB, SystemZ::OR, 0);
3558 case SystemZ::ATOMIC_LOADW_OILH:
Richard Sandiford652784e2013-09-25 11:11:53 +00003559 return emitAtomicLoadBinary(MI, MBB, SystemZ::OILH, 0);
Ulrich Weigand5f613df2013-05-06 16:15:19 +00003560 case SystemZ::ATOMIC_LOAD_OR:
3561 return emitAtomicLoadBinary(MI, MBB, SystemZ::OR, 32);
Richard Sandiford652784e2013-09-25 11:11:53 +00003562 case SystemZ::ATOMIC_LOAD_OILL:
3563 return emitAtomicLoadBinary(MI, MBB, SystemZ::OILL, 32);
3564 case SystemZ::ATOMIC_LOAD_OILH:
3565 return emitAtomicLoadBinary(MI, MBB, SystemZ::OILH, 32);
3566 case SystemZ::ATOMIC_LOAD_OILF:
3567 return emitAtomicLoadBinary(MI, MBB, SystemZ::OILF, 32);
Ulrich Weigand5f613df2013-05-06 16:15:19 +00003568 case SystemZ::ATOMIC_LOAD_OGR:
3569 return emitAtomicLoadBinary(MI, MBB, SystemZ::OGR, 64);
Richard Sandiford652784e2013-09-25 11:11:53 +00003570 case SystemZ::ATOMIC_LOAD_OILL64:
3571 return emitAtomicLoadBinary(MI, MBB, SystemZ::OILL64, 64);
3572 case SystemZ::ATOMIC_LOAD_OILH64:
3573 return emitAtomicLoadBinary(MI, MBB, SystemZ::OILH64, 64);
Richard Sandiford6e96ac62013-10-01 13:22:41 +00003574 case SystemZ::ATOMIC_LOAD_OIHL64:
3575 return emitAtomicLoadBinary(MI, MBB, SystemZ::OIHL64, 64);
3576 case SystemZ::ATOMIC_LOAD_OIHH64:
3577 return emitAtomicLoadBinary(MI, MBB, SystemZ::OIHH64, 64);
Richard Sandiford652784e2013-09-25 11:11:53 +00003578 case SystemZ::ATOMIC_LOAD_OILF64:
3579 return emitAtomicLoadBinary(MI, MBB, SystemZ::OILF64, 64);
Richard Sandiford6e96ac62013-10-01 13:22:41 +00003580 case SystemZ::ATOMIC_LOAD_OIHF64:
3581 return emitAtomicLoadBinary(MI, MBB, SystemZ::OIHF64, 64);
Ulrich Weigand5f613df2013-05-06 16:15:19 +00003582
3583 case SystemZ::ATOMIC_LOADW_XR:
3584 return emitAtomicLoadBinary(MI, MBB, SystemZ::XR, 0);
3585 case SystemZ::ATOMIC_LOADW_XILF:
Richard Sandiford652784e2013-09-25 11:11:53 +00003586 return emitAtomicLoadBinary(MI, MBB, SystemZ::XILF, 0);
Ulrich Weigand5f613df2013-05-06 16:15:19 +00003587 case SystemZ::ATOMIC_LOAD_XR:
3588 return emitAtomicLoadBinary(MI, MBB, SystemZ::XR, 32);
Richard Sandiford652784e2013-09-25 11:11:53 +00003589 case SystemZ::ATOMIC_LOAD_XILF:
3590 return emitAtomicLoadBinary(MI, MBB, SystemZ::XILF, 32);
Ulrich Weigand5f613df2013-05-06 16:15:19 +00003591 case SystemZ::ATOMIC_LOAD_XGR:
3592 return emitAtomicLoadBinary(MI, MBB, SystemZ::XGR, 64);
Richard Sandiford652784e2013-09-25 11:11:53 +00003593 case SystemZ::ATOMIC_LOAD_XILF64:
3594 return emitAtomicLoadBinary(MI, MBB, SystemZ::XILF64, 64);
Richard Sandiford5718dac2013-10-01 14:08:44 +00003595 case SystemZ::ATOMIC_LOAD_XIHF64:
3596 return emitAtomicLoadBinary(MI, MBB, SystemZ::XIHF64, 64);
Ulrich Weigand5f613df2013-05-06 16:15:19 +00003597
3598 case SystemZ::ATOMIC_LOADW_NRi:
3599 return emitAtomicLoadBinary(MI, MBB, SystemZ::NR, 0, true);
3600 case SystemZ::ATOMIC_LOADW_NILHi:
Richard Sandiford652784e2013-09-25 11:11:53 +00003601 return emitAtomicLoadBinary(MI, MBB, SystemZ::NILH, 0, true);
Ulrich Weigand5f613df2013-05-06 16:15:19 +00003602 case SystemZ::ATOMIC_LOAD_NRi:
3603 return emitAtomicLoadBinary(MI, MBB, SystemZ::NR, 32, true);
Richard Sandiford652784e2013-09-25 11:11:53 +00003604 case SystemZ::ATOMIC_LOAD_NILLi:
3605 return emitAtomicLoadBinary(MI, MBB, SystemZ::NILL, 32, true);
3606 case SystemZ::ATOMIC_LOAD_NILHi:
3607 return emitAtomicLoadBinary(MI, MBB, SystemZ::NILH, 32, true);
3608 case SystemZ::ATOMIC_LOAD_NILFi:
3609 return emitAtomicLoadBinary(MI, MBB, SystemZ::NILF, 32, true);
Ulrich Weigand5f613df2013-05-06 16:15:19 +00003610 case SystemZ::ATOMIC_LOAD_NGRi:
3611 return emitAtomicLoadBinary(MI, MBB, SystemZ::NGR, 64, true);
Richard Sandiford652784e2013-09-25 11:11:53 +00003612 case SystemZ::ATOMIC_LOAD_NILL64i:
3613 return emitAtomicLoadBinary(MI, MBB, SystemZ::NILL64, 64, true);
3614 case SystemZ::ATOMIC_LOAD_NILH64i:
3615 return emitAtomicLoadBinary(MI, MBB, SystemZ::NILH64, 64, true);
Richard Sandiford70284282013-10-01 14:20:41 +00003616 case SystemZ::ATOMIC_LOAD_NIHL64i:
3617 return emitAtomicLoadBinary(MI, MBB, SystemZ::NIHL64, 64, true);
3618 case SystemZ::ATOMIC_LOAD_NIHH64i:
3619 return emitAtomicLoadBinary(MI, MBB, SystemZ::NIHH64, 64, true);
Richard Sandiford652784e2013-09-25 11:11:53 +00003620 case SystemZ::ATOMIC_LOAD_NILF64i:
3621 return emitAtomicLoadBinary(MI, MBB, SystemZ::NILF64, 64, true);
Richard Sandiford70284282013-10-01 14:20:41 +00003622 case SystemZ::ATOMIC_LOAD_NIHF64i:
3623 return emitAtomicLoadBinary(MI, MBB, SystemZ::NIHF64, 64, true);
Ulrich Weigand5f613df2013-05-06 16:15:19 +00003624
3625 case SystemZ::ATOMIC_LOADW_MIN:
3626 return emitAtomicLoadMinMax(MI, MBB, SystemZ::CR,
3627 SystemZ::CCMASK_CMP_LE, 0);
3628 case SystemZ::ATOMIC_LOAD_MIN_32:
3629 return emitAtomicLoadMinMax(MI, MBB, SystemZ::CR,
3630 SystemZ::CCMASK_CMP_LE, 32);
3631 case SystemZ::ATOMIC_LOAD_MIN_64:
3632 return emitAtomicLoadMinMax(MI, MBB, SystemZ::CGR,
3633 SystemZ::CCMASK_CMP_LE, 64);
3634
3635 case SystemZ::ATOMIC_LOADW_MAX:
3636 return emitAtomicLoadMinMax(MI, MBB, SystemZ::CR,
3637 SystemZ::CCMASK_CMP_GE, 0);
3638 case SystemZ::ATOMIC_LOAD_MAX_32:
3639 return emitAtomicLoadMinMax(MI, MBB, SystemZ::CR,
3640 SystemZ::CCMASK_CMP_GE, 32);
3641 case SystemZ::ATOMIC_LOAD_MAX_64:
3642 return emitAtomicLoadMinMax(MI, MBB, SystemZ::CGR,
3643 SystemZ::CCMASK_CMP_GE, 64);
3644
3645 case SystemZ::ATOMIC_LOADW_UMIN:
3646 return emitAtomicLoadMinMax(MI, MBB, SystemZ::CLR,
3647 SystemZ::CCMASK_CMP_LE, 0);
3648 case SystemZ::ATOMIC_LOAD_UMIN_32:
3649 return emitAtomicLoadMinMax(MI, MBB, SystemZ::CLR,
3650 SystemZ::CCMASK_CMP_LE, 32);
3651 case SystemZ::ATOMIC_LOAD_UMIN_64:
3652 return emitAtomicLoadMinMax(MI, MBB, SystemZ::CLGR,
3653 SystemZ::CCMASK_CMP_LE, 64);
3654
3655 case SystemZ::ATOMIC_LOADW_UMAX:
3656 return emitAtomicLoadMinMax(MI, MBB, SystemZ::CLR,
3657 SystemZ::CCMASK_CMP_GE, 0);
3658 case SystemZ::ATOMIC_LOAD_UMAX_32:
3659 return emitAtomicLoadMinMax(MI, MBB, SystemZ::CLR,
3660 SystemZ::CCMASK_CMP_GE, 32);
3661 case SystemZ::ATOMIC_LOAD_UMAX_64:
3662 return emitAtomicLoadMinMax(MI, MBB, SystemZ::CLGR,
3663 SystemZ::CCMASK_CMP_GE, 64);
3664
3665 case SystemZ::ATOMIC_CMP_SWAPW:
3666 return emitAtomicCmpSwapW(MI, MBB);
Richard Sandiford5e318f02013-08-27 09:54:29 +00003667 case SystemZ::MVCSequence:
3668 case SystemZ::MVCLoop:
Richard Sandiford564681c2013-08-12 10:28:10 +00003669 return emitMemMemWrapper(MI, MBB, SystemZ::MVC);
Richard Sandiford178273a2013-09-05 10:36:45 +00003670 case SystemZ::NCSequence:
3671 case SystemZ::NCLoop:
3672 return emitMemMemWrapper(MI, MBB, SystemZ::NC);
3673 case SystemZ::OCSequence:
3674 case SystemZ::OCLoop:
3675 return emitMemMemWrapper(MI, MBB, SystemZ::OC);
3676 case SystemZ::XCSequence:
3677 case SystemZ::XCLoop:
3678 return emitMemMemWrapper(MI, MBB, SystemZ::XC);
Richard Sandiford5e318f02013-08-27 09:54:29 +00003679 case SystemZ::CLCSequence:
3680 case SystemZ::CLCLoop:
Richard Sandiford564681c2013-08-12 10:28:10 +00003681 return emitMemMemWrapper(MI, MBB, SystemZ::CLC);
Richard Sandifordca232712013-08-16 11:21:54 +00003682 case SystemZ::CLSTLoop:
3683 return emitStringWrapper(MI, MBB, SystemZ::CLST);
Richard Sandifordbb83a502013-08-16 11:29:37 +00003684 case SystemZ::MVSTLoop:
3685 return emitStringWrapper(MI, MBB, SystemZ::MVST);
Richard Sandiford0dec06a2013-08-16 11:41:43 +00003686 case SystemZ::SRSTLoop:
3687 return emitStringWrapper(MI, MBB, SystemZ::SRST);
Ulrich Weigand5f613df2013-05-06 16:15:19 +00003688 default:
3689 llvm_unreachable("Unexpected instr type to insert");
3690 }
3691}