blob: 88867b6e9da9b7d290a43f8a509953e66f8899bb [file] [log] [blame]
Tom Stellardf98f2ce2012-12-11 21:25:42 +00001//===-- AMDGPUISelLowering.cpp - AMDGPU Common DAG lowering functions -----===//
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/// \file
11/// \brief This is the parent TargetLowering class for hardware code gen
12/// targets.
13//
14//===----------------------------------------------------------------------===//
15
16#include "AMDGPUISelLowering.h"
Tom Stellarde7397ee2013-06-03 17:40:11 +000017#include "AMDGPU.h"
Christian Konig90c64cb2013-03-07 09:03:52 +000018#include "AMDGPURegisterInfo.h"
Christian Konig90c64cb2013-03-07 09:03:52 +000019#include "AMDGPUSubtarget.h"
Benjamin Kramer5c352902013-05-23 17:10:37 +000020#include "AMDILIntrinsicInfo.h"
Tom Stellardf502c292013-07-23 01:48:05 +000021#include "R600MachineFunctionInfo.h"
Tom Stellarde7397ee2013-06-03 17:40:11 +000022#include "SIMachineFunctionInfo.h"
Christian Konig90c64cb2013-03-07 09:03:52 +000023#include "llvm/CodeGen/CallingConvLower.h"
Tom Stellardf98f2ce2012-12-11 21:25:42 +000024#include "llvm/CodeGen/MachineFunction.h"
25#include "llvm/CodeGen/MachineRegisterInfo.h"
26#include "llvm/CodeGen/SelectionDAG.h"
27#include "llvm/CodeGen/TargetLoweringObjectFileImpl.h"
Tom Stellarde3d4cbc2013-06-28 15:47:08 +000028#include "llvm/IR/DataLayout.h"
Tom Stellardf98f2ce2012-12-11 21:25:42 +000029
30using namespace llvm;
31
Christian Konig90c64cb2013-03-07 09:03:52 +000032#include "AMDGPUGenCallingConv.inc"
33
Tom Stellardf98f2ce2012-12-11 21:25:42 +000034AMDGPUTargetLowering::AMDGPUTargetLowering(TargetMachine &TM) :
35 TargetLowering(TM, new TargetLoweringObjectFileELF()) {
36
37 // Initialize target lowering borrowed from AMDIL
38 InitAMDILLowering();
39
40 // We need to custom lower some of the intrinsics
41 setOperationAction(ISD::INTRINSIC_WO_CHAIN, MVT::Other, Custom);
42
43 // Library functions. These default to Expand, but we have instructions
44 // for them.
45 setOperationAction(ISD::FCEIL, MVT::f32, Legal);
46 setOperationAction(ISD::FEXP2, MVT::f32, Legal);
47 setOperationAction(ISD::FPOW, MVT::f32, Legal);
48 setOperationAction(ISD::FLOG2, MVT::f32, Legal);
49 setOperationAction(ISD::FABS, MVT::f32, Legal);
50 setOperationAction(ISD::FFLOOR, MVT::f32, Legal);
51 setOperationAction(ISD::FRINT, MVT::f32, Legal);
52
Tom Stellardba534c22013-05-20 15:02:19 +000053 // The hardware supports ROTR, but not ROTL
54 setOperationAction(ISD::ROTL, MVT::i32, Expand);
55
Tom Stellardf98f2ce2012-12-11 21:25:42 +000056 // Lower floating point store/load to integer store/load to reduce the number
57 // of patterns in tablegen.
58 setOperationAction(ISD::STORE, MVT::f32, Promote);
59 AddPromotedToType(ISD::STORE, MVT::f32, MVT::i32);
60
Tom Stellardfc047272013-07-18 21:43:42 +000061 setOperationAction(ISD::STORE, MVT::v2f32, Promote);
62 AddPromotedToType(ISD::STORE, MVT::v2f32, MVT::v2i32);
63
Tom Stellardf98f2ce2012-12-11 21:25:42 +000064 setOperationAction(ISD::STORE, MVT::v4f32, Promote);
65 AddPromotedToType(ISD::STORE, MVT::v4f32, MVT::v4i32);
66
Tom Stellard68e13282013-07-12 18:14:56 +000067 setOperationAction(ISD::STORE, MVT::f64, Promote);
68 AddPromotedToType(ISD::STORE, MVT::f64, MVT::i64);
69
Tom Stellard7a0282d2013-08-26 15:05:44 +000070 // Custom lowering of vector stores is required for local address space
71 // stores.
72 setOperationAction(ISD::STORE, MVT::v4i32, Custom);
73 // XXX: Native v2i32 local address space stores are possible, but not
74 // currently implemented.
75 setOperationAction(ISD::STORE, MVT::v2i32, Custom);
76
Tom Stellard4c52d452013-08-16 01:12:11 +000077 setTruncStoreAction(MVT::v2i32, MVT::v2i16, Custom);
78 setTruncStoreAction(MVT::v2i32, MVT::v2i8, Custom);
79 setTruncStoreAction(MVT::v4i32, MVT::v4i8, Custom);
80 // XXX: This can be change to Custom, once ExpandVectorStores can
81 // handle 64-bit stores.
82 setTruncStoreAction(MVT::v4i32, MVT::v4i16, Expand);
83
Tom Stellardf98f2ce2012-12-11 21:25:42 +000084 setOperationAction(ISD::LOAD, MVT::f32, Promote);
85 AddPromotedToType(ISD::LOAD, MVT::f32, MVT::i32);
86
Tom Stellardac85f3f2013-07-18 21:43:48 +000087 setOperationAction(ISD::LOAD, MVT::v2f32, Promote);
88 AddPromotedToType(ISD::LOAD, MVT::v2f32, MVT::v2i32);
89
Tom Stellardf98f2ce2012-12-11 21:25:42 +000090 setOperationAction(ISD::LOAD, MVT::v4f32, Promote);
91 AddPromotedToType(ISD::LOAD, MVT::v4f32, MVT::v4i32);
92
Tom Stellard68e13282013-07-12 18:14:56 +000093 setOperationAction(ISD::LOAD, MVT::f64, Promote);
94 AddPromotedToType(ISD::LOAD, MVT::f64, MVT::i64);
95
Tom Stellarda41520c2013-08-14 23:25:00 +000096 setOperationAction(ISD::CONCAT_VECTORS, MVT::v4i32, Custom);
97 setOperationAction(ISD::CONCAT_VECTORS, MVT::v4f32, Custom);
98 setOperationAction(ISD::EXTRACT_SUBVECTOR, MVT::v2i32, Custom);
99 setOperationAction(ISD::EXTRACT_SUBVECTOR, MVT::v2f32, Custom);
Tom Stellard692ee102013-08-01 15:23:42 +0000100
Tom Stellard30d84d82013-08-16 01:12:16 +0000101 setLoadExtAction(ISD::EXTLOAD, MVT::v2i8, Expand);
102 setLoadExtAction(ISD::SEXTLOAD, MVT::v2i8, Expand);
103 setLoadExtAction(ISD::ZEXTLOAD, MVT::v2i8, Expand);
104 setLoadExtAction(ISD::EXTLOAD, MVT::v4i8, Expand);
105 setLoadExtAction(ISD::SEXTLOAD, MVT::v4i8, Expand);
106 setLoadExtAction(ISD::ZEXTLOAD, MVT::v4i8, Expand);
107 setLoadExtAction(ISD::EXTLOAD, MVT::v2i16, Expand);
108 setLoadExtAction(ISD::SEXTLOAD, MVT::v2i16, Expand);
109 setLoadExtAction(ISD::ZEXTLOAD, MVT::v2i16, Expand);
110 setLoadExtAction(ISD::EXTLOAD, MVT::v4i16, Expand);
111 setLoadExtAction(ISD::SEXTLOAD, MVT::v4i16, Expand);
112 setLoadExtAction(ISD::ZEXTLOAD, MVT::v4i16, Expand);
113
Tom Stellardd7a472c2013-07-23 01:47:46 +0000114 setOperationAction(ISD::FNEG, MVT::v2f32, Expand);
115 setOperationAction(ISD::FNEG, MVT::v4f32, Expand);
116
Christian Konig45b14e32013-03-27 09:12:51 +0000117 setOperationAction(ISD::MUL, MVT::i64, Expand);
118
Tom Stellardf98f2ce2012-12-11 21:25:42 +0000119 setOperationAction(ISD::UDIV, MVT::i32, Expand);
120 setOperationAction(ISD::UDIVREM, MVT::i32, Custom);
121 setOperationAction(ISD::UREM, MVT::i32, Expand);
Tom Stellardf5660aa2013-07-18 21:43:35 +0000122 setOperationAction(ISD::VSELECT, MVT::v2f32, Expand);
123 setOperationAction(ISD::VSELECT, MVT::v4f32, Expand);
Aaron Watryf97c7fe2013-06-25 13:55:57 +0000124
Tom Stellard5464a922013-08-21 22:14:17 +0000125 static const MVT::SimpleValueType IntTypes[] = {
126 MVT::v2i32, MVT::v4i32
Aaron Watryf97c7fe2013-06-25 13:55:57 +0000127 };
Tom Stellard0991c312013-08-16 23:51:24 +0000128 const size_t NumIntTypes = array_lengthof(IntTypes);
Aaron Watryf97c7fe2013-06-25 13:55:57 +0000129
Tom Stellard0991c312013-08-16 23:51:24 +0000130 for (unsigned int x = 0; x < NumIntTypes; ++x) {
Tom Stellard5464a922013-08-21 22:14:17 +0000131 MVT::SimpleValueType VT = IntTypes[x];
Aaron Watryf97c7fe2013-06-25 13:55:57 +0000132 //Expand the following operations for the current type by default
133 setOperationAction(ISD::ADD, VT, Expand);
134 setOperationAction(ISD::AND, VT, Expand);
Tom Stellarde3d60ac2013-07-30 14:31:03 +0000135 setOperationAction(ISD::FP_TO_SINT, VT, Expand);
136 setOperationAction(ISD::FP_TO_UINT, VT, Expand);
Aaron Watryf97c7fe2013-06-25 13:55:57 +0000137 setOperationAction(ISD::MUL, VT, Expand);
138 setOperationAction(ISD::OR, VT, Expand);
139 setOperationAction(ISD::SHL, VT, Expand);
Tom Stellarde3d60ac2013-07-30 14:31:03 +0000140 setOperationAction(ISD::SINT_TO_FP, VT, Expand);
Aaron Watryf97c7fe2013-06-25 13:55:57 +0000141 setOperationAction(ISD::SRL, VT, Expand);
142 setOperationAction(ISD::SRA, VT, Expand);
143 setOperationAction(ISD::SUB, VT, Expand);
144 setOperationAction(ISD::UDIV, VT, Expand);
Tom Stellarde3d60ac2013-07-30 14:31:03 +0000145 setOperationAction(ISD::UINT_TO_FP, VT, Expand);
Aaron Watryf97c7fe2013-06-25 13:55:57 +0000146 setOperationAction(ISD::UREM, VT, Expand);
Tom Stellardf5660aa2013-07-18 21:43:35 +0000147 setOperationAction(ISD::VSELECT, VT, Expand);
Aaron Watryf97c7fe2013-06-25 13:55:57 +0000148 setOperationAction(ISD::XOR, VT, Expand);
149 }
Tom Stellard0991c312013-08-16 23:51:24 +0000150
Tom Stellard5464a922013-08-21 22:14:17 +0000151 static const MVT::SimpleValueType FloatTypes[] = {
152 MVT::v2f32, MVT::v4f32
Tom Stellard0991c312013-08-16 23:51:24 +0000153 };
154 const size_t NumFloatTypes = array_lengthof(FloatTypes);
155
156 for (unsigned int x = 0; x < NumFloatTypes; ++x) {
Tom Stellard5464a922013-08-21 22:14:17 +0000157 MVT::SimpleValueType VT = FloatTypes[x];
Tom Stellard0991c312013-08-16 23:51:24 +0000158 setOperationAction(ISD::FADD, VT, Expand);
159 setOperationAction(ISD::FDIV, VT, Expand);
Tom Stellard84c0bd92013-08-16 23:51:29 +0000160 setOperationAction(ISD::FFLOOR, VT, Expand);
Tom Stellard0991c312013-08-16 23:51:24 +0000161 setOperationAction(ISD::FMUL, VT, Expand);
Tom Stellard3cae8232013-08-16 23:51:33 +0000162 setOperationAction(ISD::FRINT, VT, Expand);
Tom Stellard0991c312013-08-16 23:51:24 +0000163 setOperationAction(ISD::FSUB, VT, Expand);
164 }
Tom Stellardf98f2ce2012-12-11 21:25:42 +0000165}
166
Tom Stellard2b272a12013-08-05 22:22:07 +0000167//===----------------------------------------------------------------------===//
168// Target Information
169//===----------------------------------------------------------------------===//
170
171MVT AMDGPUTargetLowering::getVectorIdxTy() const {
172 return MVT::i32;
173}
174
175
Tom Stellardf98f2ce2012-12-11 21:25:42 +0000176//===---------------------------------------------------------------------===//
Tom Stellard1f67c632013-07-23 23:55:03 +0000177// Target Properties
178//===---------------------------------------------------------------------===//
179
180bool AMDGPUTargetLowering::isFAbsFree(EVT VT) const {
181 assert(VT.isFloatingPoint());
182 return VT == MVT::f32;
183}
184
185bool AMDGPUTargetLowering::isFNegFree(EVT VT) const {
186 assert(VT.isFloatingPoint());
187 return VT == MVT::f32;
188}
189
190//===---------------------------------------------------------------------===//
Tom Stellardf98f2ce2012-12-11 21:25:42 +0000191// TargetLowering Callbacks
192//===---------------------------------------------------------------------===//
193
Christian Konig90c64cb2013-03-07 09:03:52 +0000194void AMDGPUTargetLowering::AnalyzeFormalArguments(CCState &State,
195 const SmallVectorImpl<ISD::InputArg> &Ins) const {
196
197 State.AnalyzeFormalArguments(Ins, CC_AMDGPU);
Tom Stellardf98f2ce2012-12-11 21:25:42 +0000198}
199
200SDValue AMDGPUTargetLowering::LowerReturn(
201 SDValue Chain,
202 CallingConv::ID CallConv,
203 bool isVarArg,
204 const SmallVectorImpl<ISD::OutputArg> &Outs,
205 const SmallVectorImpl<SDValue> &OutVals,
Andrew Trickac6d9be2013-05-25 02:42:55 +0000206 SDLoc DL, SelectionDAG &DAG) const {
Tom Stellardf98f2ce2012-12-11 21:25:42 +0000207 return DAG.getNode(AMDGPUISD::RET_FLAG, DL, MVT::Other, Chain);
208}
209
210//===---------------------------------------------------------------------===//
211// Target specific lowering
212//===---------------------------------------------------------------------===//
213
214SDValue AMDGPUTargetLowering::LowerOperation(SDValue Op, SelectionDAG &DAG)
215 const {
216 switch (Op.getOpcode()) {
217 default:
218 Op.getNode()->dump();
219 assert(0 && "Custom lowering code for this"
220 "instruction is not implemented yet!");
221 break;
222 // AMDIL DAG lowering
223 case ISD::SDIV: return LowerSDIV(Op, DAG);
224 case ISD::SREM: return LowerSREM(Op, DAG);
225 case ISD::SIGN_EXTEND_INREG: return LowerSIGN_EXTEND_INREG(Op, DAG);
226 case ISD::BRCOND: return LowerBRCOND(Op, DAG);
227 // AMDGPU DAG lowering
Tom Stellarda41520c2013-08-14 23:25:00 +0000228 case ISD::CONCAT_VECTORS: return LowerCONCAT_VECTORS(Op, DAG);
229 case ISD::EXTRACT_SUBVECTOR: return LowerEXTRACT_SUBVECTOR(Op, DAG);
Tom Stellardf98f2ce2012-12-11 21:25:42 +0000230 case ISD::INTRINSIC_WO_CHAIN: return LowerINTRINSIC_WO_CHAIN(Op, DAG);
Tom Stellard7a0282d2013-08-26 15:05:44 +0000231 case ISD::STORE: return LowerSTORE(Op, DAG);
Tom Stellardf98f2ce2012-12-11 21:25:42 +0000232 case ISD::UDIVREM: return LowerUDIVREM(Op, DAG);
233 }
234 return Op;
235}
236
Tom Stellarde3d4cbc2013-06-28 15:47:08 +0000237SDValue AMDGPUTargetLowering::LowerGlobalAddress(AMDGPUMachineFunction* MFI,
238 SDValue Op,
239 SelectionDAG &DAG) const {
240
241 const DataLayout *TD = getTargetMachine().getDataLayout();
242 GlobalAddressSDNode *G = cast<GlobalAddressSDNode>(Op);
Tom Stellardda25cd32013-08-26 15:05:36 +0000243
244 assert(G->getAddressSpace() == AMDGPUAS::LOCAL_ADDRESS);
Tom Stellarde3d4cbc2013-06-28 15:47:08 +0000245 // XXX: What does the value of G->getOffset() mean?
246 assert(G->getOffset() == 0 &&
247 "Do not know what to do with an non-zero offset");
248
249 unsigned Offset = MFI->LDSSize;
250 const GlobalValue *GV = G->getGlobal();
251 uint64_t Size = TD->getTypeAllocSize(GV->getType()->getElementType());
252
253 // XXX: Account for alignment?
254 MFI->LDSSize += Size;
255
Tom Stellardda25cd32013-08-26 15:05:36 +0000256 return DAG.getConstant(Offset, getPointerTy(G->getAddressSpace()));
Tom Stellarde3d4cbc2013-06-28 15:47:08 +0000257}
258
Tom Stellarda41520c2013-08-14 23:25:00 +0000259void AMDGPUTargetLowering::ExtractVectorElements(SDValue Op, SelectionDAG &DAG,
260 SmallVectorImpl<SDValue> &Args,
261 unsigned Start,
262 unsigned Count) const {
263 EVT VT = Op.getValueType();
264 for (unsigned i = Start, e = Start + Count; i != e; ++i) {
265 Args.push_back(DAG.getNode(ISD::EXTRACT_VECTOR_ELT, SDLoc(Op),
266 VT.getVectorElementType(),
267 Op, DAG.getConstant(i, MVT::i32)));
268 }
269}
270
271SDValue AMDGPUTargetLowering::LowerCONCAT_VECTORS(SDValue Op,
272 SelectionDAG &DAG) const {
273 SmallVector<SDValue, 8> Args;
274 SDValue A = Op.getOperand(0);
275 SDValue B = Op.getOperand(1);
276
277 ExtractVectorElements(A, DAG, Args, 0,
278 A.getValueType().getVectorNumElements());
279 ExtractVectorElements(B, DAG, Args, 0,
280 B.getValueType().getVectorNumElements());
281
282 return DAG.getNode(ISD::BUILD_VECTOR, SDLoc(Op), Op.getValueType(),
283 &Args[0], Args.size());
284}
285
286SDValue AMDGPUTargetLowering::LowerEXTRACT_SUBVECTOR(SDValue Op,
287 SelectionDAG &DAG) const {
288
289 SmallVector<SDValue, 8> Args;
290 EVT VT = Op.getValueType();
291 unsigned Start = cast<ConstantSDNode>(Op.getOperand(1))->getZExtValue();
292 ExtractVectorElements(Op.getOperand(0), DAG, Args, Start,
293 VT.getVectorNumElements());
294
295 return DAG.getNode(ISD::BUILD_VECTOR, SDLoc(Op), Op.getValueType(),
296 &Args[0], Args.size());
297}
298
299
Tom Stellardf98f2ce2012-12-11 21:25:42 +0000300SDValue AMDGPUTargetLowering::LowerINTRINSIC_WO_CHAIN(SDValue Op,
301 SelectionDAG &DAG) const {
302 unsigned IntrinsicID = cast<ConstantSDNode>(Op.getOperand(0))->getZExtValue();
Andrew Trickac6d9be2013-05-25 02:42:55 +0000303 SDLoc DL(Op);
Tom Stellardf98f2ce2012-12-11 21:25:42 +0000304 EVT VT = Op.getValueType();
305
306 switch (IntrinsicID) {
307 default: return Op;
308 case AMDGPUIntrinsic::AMDIL_abs:
309 return LowerIntrinsicIABS(Op, DAG);
310 case AMDGPUIntrinsic::AMDIL_exp:
311 return DAG.getNode(ISD::FEXP2, DL, VT, Op.getOperand(1));
312 case AMDGPUIntrinsic::AMDGPU_lrp:
313 return LowerIntrinsicLRP(Op, DAG);
314 case AMDGPUIntrinsic::AMDIL_fraction:
315 return DAG.getNode(AMDGPUISD::FRACT, DL, VT, Op.getOperand(1));
Tom Stellardf98f2ce2012-12-11 21:25:42 +0000316 case AMDGPUIntrinsic::AMDIL_max:
317 return DAG.getNode(AMDGPUISD::FMAX, DL, VT, Op.getOperand(1),
318 Op.getOperand(2));
319 case AMDGPUIntrinsic::AMDGPU_imax:
320 return DAG.getNode(AMDGPUISD::SMAX, DL, VT, Op.getOperand(1),
321 Op.getOperand(2));
322 case AMDGPUIntrinsic::AMDGPU_umax:
323 return DAG.getNode(AMDGPUISD::UMAX, DL, VT, Op.getOperand(1),
324 Op.getOperand(2));
325 case AMDGPUIntrinsic::AMDIL_min:
326 return DAG.getNode(AMDGPUISD::FMIN, DL, VT, Op.getOperand(1),
327 Op.getOperand(2));
328 case AMDGPUIntrinsic::AMDGPU_imin:
329 return DAG.getNode(AMDGPUISD::SMIN, DL, VT, Op.getOperand(1),
330 Op.getOperand(2));
331 case AMDGPUIntrinsic::AMDGPU_umin:
332 return DAG.getNode(AMDGPUISD::UMIN, DL, VT, Op.getOperand(1),
333 Op.getOperand(2));
334 case AMDGPUIntrinsic::AMDIL_round_nearest:
335 return DAG.getNode(ISD::FRINT, DL, VT, Op.getOperand(1));
336 }
337}
338
339///IABS(a) = SMAX(sub(0, a), a)
340SDValue AMDGPUTargetLowering::LowerIntrinsicIABS(SDValue Op,
341 SelectionDAG &DAG) const {
342
Andrew Trickac6d9be2013-05-25 02:42:55 +0000343 SDLoc DL(Op);
Tom Stellardf98f2ce2012-12-11 21:25:42 +0000344 EVT VT = Op.getValueType();
345 SDValue Neg = DAG.getNode(ISD::SUB, DL, VT, DAG.getConstant(0, VT),
346 Op.getOperand(1));
347
348 return DAG.getNode(AMDGPUISD::SMAX, DL, VT, Neg, Op.getOperand(1));
349}
350
351/// Linear Interpolation
352/// LRP(a, b, c) = muladd(a, b, (1 - a) * c)
353SDValue AMDGPUTargetLowering::LowerIntrinsicLRP(SDValue Op,
354 SelectionDAG &DAG) const {
Andrew Trickac6d9be2013-05-25 02:42:55 +0000355 SDLoc DL(Op);
Tom Stellardf98f2ce2012-12-11 21:25:42 +0000356 EVT VT = Op.getValueType();
357 SDValue OneSubA = DAG.getNode(ISD::FSUB, DL, VT,
358 DAG.getConstantFP(1.0f, MVT::f32),
359 Op.getOperand(1));
360 SDValue OneSubAC = DAG.getNode(ISD::FMUL, DL, VT, OneSubA,
361 Op.getOperand(3));
Vincent Lejeunee3111962013-02-18 14:11:28 +0000362 return DAG.getNode(ISD::FADD, DL, VT,
363 DAG.getNode(ISD::FMUL, DL, VT, Op.getOperand(1), Op.getOperand(2)),
364 OneSubAC);
Tom Stellardf98f2ce2012-12-11 21:25:42 +0000365}
366
367/// \brief Generate Min/Max node
368SDValue AMDGPUTargetLowering::LowerMinMax(SDValue Op,
369 SelectionDAG &DAG) const {
Andrew Trickac6d9be2013-05-25 02:42:55 +0000370 SDLoc DL(Op);
Tom Stellardf98f2ce2012-12-11 21:25:42 +0000371 EVT VT = Op.getValueType();
372
373 SDValue LHS = Op.getOperand(0);
374 SDValue RHS = Op.getOperand(1);
375 SDValue True = Op.getOperand(2);
376 SDValue False = Op.getOperand(3);
377 SDValue CC = Op.getOperand(4);
378
379 if (VT != MVT::f32 ||
380 !((LHS == True && RHS == False) || (LHS == False && RHS == True))) {
381 return SDValue();
382 }
383
384 ISD::CondCode CCOpcode = cast<CondCodeSDNode>(CC)->get();
385 switch (CCOpcode) {
386 case ISD::SETOEQ:
387 case ISD::SETONE:
388 case ISD::SETUNE:
389 case ISD::SETNE:
390 case ISD::SETUEQ:
391 case ISD::SETEQ:
392 case ISD::SETFALSE:
393 case ISD::SETFALSE2:
394 case ISD::SETTRUE:
395 case ISD::SETTRUE2:
396 case ISD::SETUO:
397 case ISD::SETO:
398 assert(0 && "Operation should already be optimised !");
399 case ISD::SETULE:
400 case ISD::SETULT:
401 case ISD::SETOLE:
402 case ISD::SETOLT:
403 case ISD::SETLE:
404 case ISD::SETLT: {
405 if (LHS == True)
406 return DAG.getNode(AMDGPUISD::FMIN, DL, VT, LHS, RHS);
407 else
408 return DAG.getNode(AMDGPUISD::FMAX, DL, VT, LHS, RHS);
409 }
410 case ISD::SETGT:
411 case ISD::SETGE:
412 case ISD::SETUGE:
413 case ISD::SETOGE:
414 case ISD::SETUGT:
415 case ISD::SETOGT: {
416 if (LHS == True)
417 return DAG.getNode(AMDGPUISD::FMAX, DL, VT, LHS, RHS);
418 else
419 return DAG.getNode(AMDGPUISD::FMIN, DL, VT, LHS, RHS);
420 }
421 case ISD::SETCC_INVALID:
422 assert(0 && "Invalid setcc condcode !");
423 }
424 return Op;
425}
426
Tom Stellard7a0282d2013-08-26 15:05:44 +0000427SDValue AMDGPUTargetLowering::MergeVectorStore(const SDValue &Op,
428 SelectionDAG &DAG) const {
429 StoreSDNode *Store = dyn_cast<StoreSDNode>(Op);
430 EVT MemVT = Store->getMemoryVT();
431 unsigned MemBits = MemVT.getSizeInBits();
Tom Stellardf98f2ce2012-12-11 21:25:42 +0000432
Tom Stellard7a0282d2013-08-26 15:05:44 +0000433 // Byte stores are really expensive, so if possible, try to pack
434 // 32-bit vector truncatating store into an i32 store.
435 // XXX: We could also handle optimize other vector bitwidths
436 if (!MemVT.isVector() || MemBits > 32) {
437 return SDValue();
438 }
439
440 SDLoc DL(Op);
441 const SDValue &Value = Store->getValue();
442 EVT VT = Value.getValueType();
443 const SDValue &Ptr = Store->getBasePtr();
444 EVT MemEltVT = MemVT.getVectorElementType();
445 unsigned MemEltBits = MemEltVT.getSizeInBits();
446 unsigned MemNumElements = MemVT.getVectorNumElements();
447 EVT PackedVT = EVT::getIntegerVT(*DAG.getContext(), MemVT.getSizeInBits());
448 SDValue Mask;
449 switch(MemEltBits) {
450 case 8:
451 Mask = DAG.getConstant(0xFF, PackedVT);
452 break;
453 case 16:
454 Mask = DAG.getConstant(0xFFFF, PackedVT);
455 break;
456 default:
457 llvm_unreachable("Cannot lower this vector store");
458 }
459 SDValue PackedValue;
460 for (unsigned i = 0; i < MemNumElements; ++i) {
461 EVT ElemVT = VT.getVectorElementType();
462 SDValue Elt = DAG.getNode(ISD::EXTRACT_VECTOR_ELT, DL, ElemVT, Value,
463 DAG.getConstant(i, MVT::i32));
464 Elt = DAG.getZExtOrTrunc(Elt, DL, PackedVT);
465 Elt = DAG.getNode(ISD::AND, DL, PackedVT, Elt, Mask);
466 SDValue Shift = DAG.getConstant(MemEltBits * i, PackedVT);
467 Elt = DAG.getNode(ISD::SHL, DL, PackedVT, Elt, Shift);
468 if (i == 0) {
469 PackedValue = Elt;
470 } else {
471 PackedValue = DAG.getNode(ISD::OR, DL, PackedVT, PackedValue, Elt);
472 }
473 }
474 return DAG.getStore(Store->getChain(), DL, PackedValue, Ptr,
475 MachinePointerInfo(Store->getMemOperand()->getValue()),
476 Store->isVolatile(), Store->isNonTemporal(),
477 Store->getAlignment());
478}
479
480SDValue AMDGPUTargetLowering::SplitVectorStore(SDValue Op,
481 SelectionDAG &DAG) const {
482 StoreSDNode *Store = cast<StoreSDNode>(Op);
483 EVT MemEltVT = Store->getMemoryVT().getVectorElementType();
484 EVT EltVT = Store->getValue().getValueType().getVectorElementType();
485 EVT PtrVT = Store->getBasePtr().getValueType();
486 unsigned NumElts = Store->getMemoryVT().getVectorNumElements();
487 SDLoc SL(Op);
488
489 SmallVector<SDValue, 8> Chains;
490
491 for (unsigned i = 0, e = NumElts; i != e; ++i) {
492 SDValue Val = DAG.getNode(ISD::EXTRACT_VECTOR_ELT, SL, EltVT,
493 Store->getValue(), DAG.getConstant(i, MVT::i32));
494 SDValue Ptr = DAG.getNode(ISD::ADD, SL, PtrVT,
495 Store->getBasePtr(),
496 DAG.getConstant(i * (MemEltVT.getSizeInBits() / 8),
497 PtrVT));
Tom Stellard8e780122013-08-26 15:05:49 +0000498 Chains.push_back(DAG.getTruncStore(Store->getChain(), SL, Val, Ptr,
Tom Stellard7a0282d2013-08-26 15:05:44 +0000499 MachinePointerInfo(Store->getMemOperand()->getValue()),
Tom Stellard8e780122013-08-26 15:05:49 +0000500 MemEltVT, Store->isVolatile(), Store->isNonTemporal(),
Tom Stellard7a0282d2013-08-26 15:05:44 +0000501 Store->getAlignment()));
502 }
503 return DAG.getNode(ISD::TokenFactor, SL, MVT::Other, &Chains[0], NumElts);
504}
505
506SDValue AMDGPUTargetLowering::LowerSTORE(SDValue Op, SelectionDAG &DAG) const {
507 SDValue Result = AMDGPUTargetLowering::MergeVectorStore(Op, DAG);
508 if (Result.getNode()) {
509 return Result;
510 }
511
512 StoreSDNode *Store = cast<StoreSDNode>(Op);
513 if (Store->getAddressSpace() == AMDGPUAS::LOCAL_ADDRESS &&
514 Store->getValue().getValueType().isVector()) {
515 return SplitVectorStore(Op, DAG);
516 }
517 return SDValue();
518}
Tom Stellardf98f2ce2012-12-11 21:25:42 +0000519
520SDValue AMDGPUTargetLowering::LowerUDIVREM(SDValue Op,
521 SelectionDAG &DAG) const {
Andrew Trickac6d9be2013-05-25 02:42:55 +0000522 SDLoc DL(Op);
Tom Stellardf98f2ce2012-12-11 21:25:42 +0000523 EVT VT = Op.getValueType();
524
525 SDValue Num = Op.getOperand(0);
526 SDValue Den = Op.getOperand(1);
527
528 SmallVector<SDValue, 8> Results;
529
530 // RCP = URECIP(Den) = 2^32 / Den + e
531 // e is rounding error.
532 SDValue RCP = DAG.getNode(AMDGPUISD::URECIP, DL, VT, Den);
533
534 // RCP_LO = umulo(RCP, Den) */
535 SDValue RCP_LO = DAG.getNode(ISD::UMULO, DL, VT, RCP, Den);
536
537 // RCP_HI = mulhu (RCP, Den) */
538 SDValue RCP_HI = DAG.getNode(ISD::MULHU, DL, VT, RCP, Den);
539
540 // NEG_RCP_LO = -RCP_LO
541 SDValue NEG_RCP_LO = DAG.getNode(ISD::SUB, DL, VT, DAG.getConstant(0, VT),
542 RCP_LO);
543
544 // ABS_RCP_LO = (RCP_HI == 0 ? NEG_RCP_LO : RCP_LO)
545 SDValue ABS_RCP_LO = DAG.getSelectCC(DL, RCP_HI, DAG.getConstant(0, VT),
546 NEG_RCP_LO, RCP_LO,
547 ISD::SETEQ);
548 // Calculate the rounding error from the URECIP instruction
549 // E = mulhu(ABS_RCP_LO, RCP)
550 SDValue E = DAG.getNode(ISD::MULHU, DL, VT, ABS_RCP_LO, RCP);
551
552 // RCP_A_E = RCP + E
553 SDValue RCP_A_E = DAG.getNode(ISD::ADD, DL, VT, RCP, E);
554
555 // RCP_S_E = RCP - E
556 SDValue RCP_S_E = DAG.getNode(ISD::SUB, DL, VT, RCP, E);
557
558 // Tmp0 = (RCP_HI == 0 ? RCP_A_E : RCP_SUB_E)
559 SDValue Tmp0 = DAG.getSelectCC(DL, RCP_HI, DAG.getConstant(0, VT),
560 RCP_A_E, RCP_S_E,
561 ISD::SETEQ);
562 // Quotient = mulhu(Tmp0, Num)
563 SDValue Quotient = DAG.getNode(ISD::MULHU, DL, VT, Tmp0, Num);
564
565 // Num_S_Remainder = Quotient * Den
566 SDValue Num_S_Remainder = DAG.getNode(ISD::UMULO, DL, VT, Quotient, Den);
567
568 // Remainder = Num - Num_S_Remainder
569 SDValue Remainder = DAG.getNode(ISD::SUB, DL, VT, Num, Num_S_Remainder);
570
571 // Remainder_GE_Den = (Remainder >= Den ? -1 : 0)
572 SDValue Remainder_GE_Den = DAG.getSelectCC(DL, Remainder, Den,
573 DAG.getConstant(-1, VT),
574 DAG.getConstant(0, VT),
575 ISD::SETGE);
576 // Remainder_GE_Zero = (Remainder >= 0 ? -1 : 0)
577 SDValue Remainder_GE_Zero = DAG.getSelectCC(DL, Remainder,
578 DAG.getConstant(0, VT),
579 DAG.getConstant(-1, VT),
580 DAG.getConstant(0, VT),
581 ISD::SETGE);
582 // Tmp1 = Remainder_GE_Den & Remainder_GE_Zero
583 SDValue Tmp1 = DAG.getNode(ISD::AND, DL, VT, Remainder_GE_Den,
584 Remainder_GE_Zero);
585
586 // Calculate Division result:
587
588 // Quotient_A_One = Quotient + 1
589 SDValue Quotient_A_One = DAG.getNode(ISD::ADD, DL, VT, Quotient,
590 DAG.getConstant(1, VT));
591
592 // Quotient_S_One = Quotient - 1
593 SDValue Quotient_S_One = DAG.getNode(ISD::SUB, DL, VT, Quotient,
594 DAG.getConstant(1, VT));
595
596 // Div = (Tmp1 == 0 ? Quotient : Quotient_A_One)
597 SDValue Div = DAG.getSelectCC(DL, Tmp1, DAG.getConstant(0, VT),
598 Quotient, Quotient_A_One, ISD::SETEQ);
599
600 // Div = (Remainder_GE_Zero == 0 ? Quotient_S_One : Div)
601 Div = DAG.getSelectCC(DL, Remainder_GE_Zero, DAG.getConstant(0, VT),
602 Quotient_S_One, Div, ISD::SETEQ);
603
604 // Calculate Rem result:
605
606 // Remainder_S_Den = Remainder - Den
607 SDValue Remainder_S_Den = DAG.getNode(ISD::SUB, DL, VT, Remainder, Den);
608
609 // Remainder_A_Den = Remainder + Den
610 SDValue Remainder_A_Den = DAG.getNode(ISD::ADD, DL, VT, Remainder, Den);
611
612 // Rem = (Tmp1 == 0 ? Remainder : Remainder_S_Den)
613 SDValue Rem = DAG.getSelectCC(DL, Tmp1, DAG.getConstant(0, VT),
614 Remainder, Remainder_S_Den, ISD::SETEQ);
615
616 // Rem = (Remainder_GE_Zero == 0 ? Remainder_A_Den : Rem)
617 Rem = DAG.getSelectCC(DL, Remainder_GE_Zero, DAG.getConstant(0, VT),
618 Remainder_A_Den, Rem, ISD::SETEQ);
619 SDValue Ops[2];
620 Ops[0] = Div;
621 Ops[1] = Rem;
622 return DAG.getMergeValues(Ops, 2, DL);
623}
624
Tom Stellard4c52d452013-08-16 01:12:11 +0000625
Tom Stellardf98f2ce2012-12-11 21:25:42 +0000626//===----------------------------------------------------------------------===//
627// Helper functions
628//===----------------------------------------------------------------------===//
629
630bool AMDGPUTargetLowering::isHWTrueValue(SDValue Op) const {
631 if (ConstantFPSDNode * CFP = dyn_cast<ConstantFPSDNode>(Op)) {
632 return CFP->isExactlyValue(1.0);
633 }
634 if (ConstantSDNode *C = dyn_cast<ConstantSDNode>(Op)) {
635 return C->isAllOnesValue();
636 }
637 return false;
638}
639
640bool AMDGPUTargetLowering::isHWFalseValue(SDValue Op) const {
641 if (ConstantFPSDNode * CFP = dyn_cast<ConstantFPSDNode>(Op)) {
642 return CFP->getValueAPF().isZero();
643 }
644 if (ConstantSDNode *C = dyn_cast<ConstantSDNode>(Op)) {
645 return C->isNullValue();
646 }
647 return false;
648}
649
650SDValue AMDGPUTargetLowering::CreateLiveInRegister(SelectionDAG &DAG,
651 const TargetRegisterClass *RC,
652 unsigned Reg, EVT VT) const {
653 MachineFunction &MF = DAG.getMachineFunction();
654 MachineRegisterInfo &MRI = MF.getRegInfo();
655 unsigned VirtualRegister;
656 if (!MRI.isLiveIn(Reg)) {
657 VirtualRegister = MRI.createVirtualRegister(RC);
658 MRI.addLiveIn(Reg, VirtualRegister);
659 } else {
660 VirtualRegister = MRI.getLiveInVirtReg(Reg);
661 }
662 return DAG.getRegister(VirtualRegister, VT);
663}
664
665#define NODE_NAME_CASE(node) case AMDGPUISD::node: return #node;
666
667const char* AMDGPUTargetLowering::getTargetNodeName(unsigned Opcode) const {
668 switch (Opcode) {
669 default: return 0;
670 // AMDIL DAG nodes
Tom Stellardf98f2ce2012-12-11 21:25:42 +0000671 NODE_NAME_CASE(CALL);
672 NODE_NAME_CASE(UMUL);
673 NODE_NAME_CASE(DIV_INF);
674 NODE_NAME_CASE(RET_FLAG);
675 NODE_NAME_CASE(BRANCH_COND);
676
677 // AMDGPU DAG nodes
678 NODE_NAME_CASE(DWORDADDR)
679 NODE_NAME_CASE(FRACT)
680 NODE_NAME_CASE(FMAX)
681 NODE_NAME_CASE(SMAX)
682 NODE_NAME_CASE(UMAX)
683 NODE_NAME_CASE(FMIN)
684 NODE_NAME_CASE(SMIN)
685 NODE_NAME_CASE(UMIN)
686 NODE_NAME_CASE(URECIP)
Tom Stellardf98f2ce2012-12-11 21:25:42 +0000687 NODE_NAME_CASE(EXPORT)
Tom Stellardc7e18882013-01-23 02:09:03 +0000688 NODE_NAME_CASE(CONST_ADDRESS)
Tom Stellardc0b0c672013-02-06 17:32:29 +0000689 NODE_NAME_CASE(REGISTER_LOAD)
690 NODE_NAME_CASE(REGISTER_STORE)
Tom Stellard68db37b2013-08-14 23:24:45 +0000691 NODE_NAME_CASE(LOAD_CONSTANT)
692 NODE_NAME_CASE(LOAD_INPUT)
693 NODE_NAME_CASE(SAMPLE)
694 NODE_NAME_CASE(SAMPLEB)
695 NODE_NAME_CASE(SAMPLED)
696 NODE_NAME_CASE(SAMPLEL)
Tom Stellardec484272013-08-16 01:12:06 +0000697 NODE_NAME_CASE(STORE_MSKOR)
Tom Stellardf98f2ce2012-12-11 21:25:42 +0000698 }
699}