blob: f6c074a4cd540410b411dbff4cd8fb6fb4688fe7 [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
Tom Stellarde3d4cbc2013-06-28 15:47:08 +0000249 const GlobalValue *GV = G->getGlobal();
Tom Stellarde3d4cbc2013-06-28 15:47:08 +0000250
Tom Stellard470c4512013-09-05 18:37:57 +0000251 unsigned Offset;
252 if (MFI->LocalMemoryObjects.count(GV) == 0) {
253 uint64_t Size = TD->getTypeAllocSize(GV->getType()->getElementType());
254 Offset = MFI->LDSSize;
255 MFI->LocalMemoryObjects[GV] = Offset;
256 // XXX: Account for alignment?
257 MFI->LDSSize += Size;
258 } else {
259 Offset = MFI->LocalMemoryObjects[GV];
260 }
Tom Stellarde3d4cbc2013-06-28 15:47:08 +0000261
Tom Stellardda25cd32013-08-26 15:05:36 +0000262 return DAG.getConstant(Offset, getPointerTy(G->getAddressSpace()));
Tom Stellarde3d4cbc2013-06-28 15:47:08 +0000263}
264
Tom Stellarda41520c2013-08-14 23:25:00 +0000265void AMDGPUTargetLowering::ExtractVectorElements(SDValue Op, SelectionDAG &DAG,
266 SmallVectorImpl<SDValue> &Args,
267 unsigned Start,
268 unsigned Count) const {
269 EVT VT = Op.getValueType();
270 for (unsigned i = Start, e = Start + Count; i != e; ++i) {
271 Args.push_back(DAG.getNode(ISD::EXTRACT_VECTOR_ELT, SDLoc(Op),
272 VT.getVectorElementType(),
273 Op, DAG.getConstant(i, MVT::i32)));
274 }
275}
276
277SDValue AMDGPUTargetLowering::LowerCONCAT_VECTORS(SDValue Op,
278 SelectionDAG &DAG) const {
279 SmallVector<SDValue, 8> Args;
280 SDValue A = Op.getOperand(0);
281 SDValue B = Op.getOperand(1);
282
283 ExtractVectorElements(A, DAG, Args, 0,
284 A.getValueType().getVectorNumElements());
285 ExtractVectorElements(B, DAG, Args, 0,
286 B.getValueType().getVectorNumElements());
287
288 return DAG.getNode(ISD::BUILD_VECTOR, SDLoc(Op), Op.getValueType(),
289 &Args[0], Args.size());
290}
291
292SDValue AMDGPUTargetLowering::LowerEXTRACT_SUBVECTOR(SDValue Op,
293 SelectionDAG &DAG) const {
294
295 SmallVector<SDValue, 8> Args;
296 EVT VT = Op.getValueType();
297 unsigned Start = cast<ConstantSDNode>(Op.getOperand(1))->getZExtValue();
298 ExtractVectorElements(Op.getOperand(0), DAG, Args, Start,
299 VT.getVectorNumElements());
300
301 return DAG.getNode(ISD::BUILD_VECTOR, SDLoc(Op), Op.getValueType(),
302 &Args[0], Args.size());
303}
304
305
Tom Stellardf98f2ce2012-12-11 21:25:42 +0000306SDValue AMDGPUTargetLowering::LowerINTRINSIC_WO_CHAIN(SDValue Op,
307 SelectionDAG &DAG) const {
308 unsigned IntrinsicID = cast<ConstantSDNode>(Op.getOperand(0))->getZExtValue();
Andrew Trickac6d9be2013-05-25 02:42:55 +0000309 SDLoc DL(Op);
Tom Stellardf98f2ce2012-12-11 21:25:42 +0000310 EVT VT = Op.getValueType();
311
312 switch (IntrinsicID) {
313 default: return Op;
314 case AMDGPUIntrinsic::AMDIL_abs:
315 return LowerIntrinsicIABS(Op, DAG);
316 case AMDGPUIntrinsic::AMDIL_exp:
317 return DAG.getNode(ISD::FEXP2, DL, VT, Op.getOperand(1));
318 case AMDGPUIntrinsic::AMDGPU_lrp:
319 return LowerIntrinsicLRP(Op, DAG);
320 case AMDGPUIntrinsic::AMDIL_fraction:
321 return DAG.getNode(AMDGPUISD::FRACT, DL, VT, Op.getOperand(1));
Tom Stellardf98f2ce2012-12-11 21:25:42 +0000322 case AMDGPUIntrinsic::AMDIL_max:
323 return DAG.getNode(AMDGPUISD::FMAX, DL, VT, Op.getOperand(1),
324 Op.getOperand(2));
325 case AMDGPUIntrinsic::AMDGPU_imax:
326 return DAG.getNode(AMDGPUISD::SMAX, DL, VT, Op.getOperand(1),
327 Op.getOperand(2));
328 case AMDGPUIntrinsic::AMDGPU_umax:
329 return DAG.getNode(AMDGPUISD::UMAX, DL, VT, Op.getOperand(1),
330 Op.getOperand(2));
331 case AMDGPUIntrinsic::AMDIL_min:
332 return DAG.getNode(AMDGPUISD::FMIN, DL, VT, Op.getOperand(1),
333 Op.getOperand(2));
334 case AMDGPUIntrinsic::AMDGPU_imin:
335 return DAG.getNode(AMDGPUISD::SMIN, DL, VT, Op.getOperand(1),
336 Op.getOperand(2));
337 case AMDGPUIntrinsic::AMDGPU_umin:
338 return DAG.getNode(AMDGPUISD::UMIN, DL, VT, Op.getOperand(1),
339 Op.getOperand(2));
340 case AMDGPUIntrinsic::AMDIL_round_nearest:
341 return DAG.getNode(ISD::FRINT, DL, VT, Op.getOperand(1));
342 }
343}
344
345///IABS(a) = SMAX(sub(0, a), a)
346SDValue AMDGPUTargetLowering::LowerIntrinsicIABS(SDValue Op,
347 SelectionDAG &DAG) const {
348
Andrew Trickac6d9be2013-05-25 02:42:55 +0000349 SDLoc DL(Op);
Tom Stellardf98f2ce2012-12-11 21:25:42 +0000350 EVT VT = Op.getValueType();
351 SDValue Neg = DAG.getNode(ISD::SUB, DL, VT, DAG.getConstant(0, VT),
352 Op.getOperand(1));
353
354 return DAG.getNode(AMDGPUISD::SMAX, DL, VT, Neg, Op.getOperand(1));
355}
356
357/// Linear Interpolation
358/// LRP(a, b, c) = muladd(a, b, (1 - a) * c)
359SDValue AMDGPUTargetLowering::LowerIntrinsicLRP(SDValue Op,
360 SelectionDAG &DAG) const {
Andrew Trickac6d9be2013-05-25 02:42:55 +0000361 SDLoc DL(Op);
Tom Stellardf98f2ce2012-12-11 21:25:42 +0000362 EVT VT = Op.getValueType();
363 SDValue OneSubA = DAG.getNode(ISD::FSUB, DL, VT,
364 DAG.getConstantFP(1.0f, MVT::f32),
365 Op.getOperand(1));
366 SDValue OneSubAC = DAG.getNode(ISD::FMUL, DL, VT, OneSubA,
367 Op.getOperand(3));
Vincent Lejeunee3111962013-02-18 14:11:28 +0000368 return DAG.getNode(ISD::FADD, DL, VT,
369 DAG.getNode(ISD::FMUL, DL, VT, Op.getOperand(1), Op.getOperand(2)),
370 OneSubAC);
Tom Stellardf98f2ce2012-12-11 21:25:42 +0000371}
372
373/// \brief Generate Min/Max node
374SDValue AMDGPUTargetLowering::LowerMinMax(SDValue Op,
375 SelectionDAG &DAG) const {
Andrew Trickac6d9be2013-05-25 02:42:55 +0000376 SDLoc DL(Op);
Tom Stellardf98f2ce2012-12-11 21:25:42 +0000377 EVT VT = Op.getValueType();
378
379 SDValue LHS = Op.getOperand(0);
380 SDValue RHS = Op.getOperand(1);
381 SDValue True = Op.getOperand(2);
382 SDValue False = Op.getOperand(3);
383 SDValue CC = Op.getOperand(4);
384
385 if (VT != MVT::f32 ||
386 !((LHS == True && RHS == False) || (LHS == False && RHS == True))) {
387 return SDValue();
388 }
389
390 ISD::CondCode CCOpcode = cast<CondCodeSDNode>(CC)->get();
391 switch (CCOpcode) {
392 case ISD::SETOEQ:
393 case ISD::SETONE:
394 case ISD::SETUNE:
395 case ISD::SETNE:
396 case ISD::SETUEQ:
397 case ISD::SETEQ:
398 case ISD::SETFALSE:
399 case ISD::SETFALSE2:
400 case ISD::SETTRUE:
401 case ISD::SETTRUE2:
402 case ISD::SETUO:
403 case ISD::SETO:
404 assert(0 && "Operation should already be optimised !");
405 case ISD::SETULE:
406 case ISD::SETULT:
407 case ISD::SETOLE:
408 case ISD::SETOLT:
409 case ISD::SETLE:
410 case ISD::SETLT: {
411 if (LHS == True)
412 return DAG.getNode(AMDGPUISD::FMIN, DL, VT, LHS, RHS);
413 else
414 return DAG.getNode(AMDGPUISD::FMAX, DL, VT, LHS, RHS);
415 }
416 case ISD::SETGT:
417 case ISD::SETGE:
418 case ISD::SETUGE:
419 case ISD::SETOGE:
420 case ISD::SETUGT:
421 case ISD::SETOGT: {
422 if (LHS == True)
423 return DAG.getNode(AMDGPUISD::FMAX, DL, VT, LHS, RHS);
424 else
425 return DAG.getNode(AMDGPUISD::FMIN, DL, VT, LHS, RHS);
426 }
427 case ISD::SETCC_INVALID:
428 assert(0 && "Invalid setcc condcode !");
429 }
430 return Op;
431}
432
Tom Stellardd08a9302013-08-26 15:06:04 +0000433SDValue AMDGPUTargetLowering::SplitVectorLoad(const SDValue &Op,
434 SelectionDAG &DAG) const {
435 LoadSDNode *Load = dyn_cast<LoadSDNode>(Op);
436 EVT MemEltVT = Load->getMemoryVT().getVectorElementType();
437 EVT EltVT = Op.getValueType().getVectorElementType();
438 EVT PtrVT = Load->getBasePtr().getValueType();
439 unsigned NumElts = Load->getMemoryVT().getVectorNumElements();
440 SmallVector<SDValue, 8> Loads;
441 SDLoc SL(Op);
442
443 for (unsigned i = 0, e = NumElts; i != e; ++i) {
444 SDValue Ptr = DAG.getNode(ISD::ADD, SL, PtrVT, Load->getBasePtr(),
445 DAG.getConstant(i * (MemEltVT.getSizeInBits() / 8), PtrVT));
446 Loads.push_back(DAG.getExtLoad(Load->getExtensionType(), SL, EltVT,
447 Load->getChain(), Ptr,
448 MachinePointerInfo(Load->getMemOperand()->getValue()),
449 MemEltVT, Load->isVolatile(), Load->isNonTemporal(),
450 Load->getAlignment()));
451 }
452 return DAG.getNode(ISD::BUILD_VECTOR, SL, Op.getValueType(), &Loads[0],
453 Loads.size());
454}
455
Tom Stellard7a0282d2013-08-26 15:05:44 +0000456SDValue AMDGPUTargetLowering::MergeVectorStore(const SDValue &Op,
457 SelectionDAG &DAG) const {
458 StoreSDNode *Store = dyn_cast<StoreSDNode>(Op);
459 EVT MemVT = Store->getMemoryVT();
460 unsigned MemBits = MemVT.getSizeInBits();
Tom Stellardf98f2ce2012-12-11 21:25:42 +0000461
Tom Stellard7a0282d2013-08-26 15:05:44 +0000462 // Byte stores are really expensive, so if possible, try to pack
463 // 32-bit vector truncatating store into an i32 store.
464 // XXX: We could also handle optimize other vector bitwidths
465 if (!MemVT.isVector() || MemBits > 32) {
466 return SDValue();
467 }
468
469 SDLoc DL(Op);
470 const SDValue &Value = Store->getValue();
471 EVT VT = Value.getValueType();
472 const SDValue &Ptr = Store->getBasePtr();
473 EVT MemEltVT = MemVT.getVectorElementType();
474 unsigned MemEltBits = MemEltVT.getSizeInBits();
475 unsigned MemNumElements = MemVT.getVectorNumElements();
476 EVT PackedVT = EVT::getIntegerVT(*DAG.getContext(), MemVT.getSizeInBits());
477 SDValue Mask;
478 switch(MemEltBits) {
479 case 8:
480 Mask = DAG.getConstant(0xFF, PackedVT);
481 break;
482 case 16:
483 Mask = DAG.getConstant(0xFFFF, PackedVT);
484 break;
485 default:
486 llvm_unreachable("Cannot lower this vector store");
487 }
488 SDValue PackedValue;
489 for (unsigned i = 0; i < MemNumElements; ++i) {
490 EVT ElemVT = VT.getVectorElementType();
491 SDValue Elt = DAG.getNode(ISD::EXTRACT_VECTOR_ELT, DL, ElemVT, Value,
492 DAG.getConstant(i, MVT::i32));
493 Elt = DAG.getZExtOrTrunc(Elt, DL, PackedVT);
494 Elt = DAG.getNode(ISD::AND, DL, PackedVT, Elt, Mask);
495 SDValue Shift = DAG.getConstant(MemEltBits * i, PackedVT);
496 Elt = DAG.getNode(ISD::SHL, DL, PackedVT, Elt, Shift);
497 if (i == 0) {
498 PackedValue = Elt;
499 } else {
500 PackedValue = DAG.getNode(ISD::OR, DL, PackedVT, PackedValue, Elt);
501 }
502 }
503 return DAG.getStore(Store->getChain(), DL, PackedValue, Ptr,
504 MachinePointerInfo(Store->getMemOperand()->getValue()),
505 Store->isVolatile(), Store->isNonTemporal(),
506 Store->getAlignment());
507}
508
509SDValue AMDGPUTargetLowering::SplitVectorStore(SDValue Op,
510 SelectionDAG &DAG) const {
511 StoreSDNode *Store = cast<StoreSDNode>(Op);
512 EVT MemEltVT = Store->getMemoryVT().getVectorElementType();
513 EVT EltVT = Store->getValue().getValueType().getVectorElementType();
514 EVT PtrVT = Store->getBasePtr().getValueType();
515 unsigned NumElts = Store->getMemoryVT().getVectorNumElements();
516 SDLoc SL(Op);
517
518 SmallVector<SDValue, 8> Chains;
519
520 for (unsigned i = 0, e = NumElts; i != e; ++i) {
521 SDValue Val = DAG.getNode(ISD::EXTRACT_VECTOR_ELT, SL, EltVT,
522 Store->getValue(), DAG.getConstant(i, MVT::i32));
523 SDValue Ptr = DAG.getNode(ISD::ADD, SL, PtrVT,
524 Store->getBasePtr(),
525 DAG.getConstant(i * (MemEltVT.getSizeInBits() / 8),
526 PtrVT));
Tom Stellard8e780122013-08-26 15:05:49 +0000527 Chains.push_back(DAG.getTruncStore(Store->getChain(), SL, Val, Ptr,
Tom Stellard7a0282d2013-08-26 15:05:44 +0000528 MachinePointerInfo(Store->getMemOperand()->getValue()),
Tom Stellard8e780122013-08-26 15:05:49 +0000529 MemEltVT, Store->isVolatile(), Store->isNonTemporal(),
Tom Stellard7a0282d2013-08-26 15:05:44 +0000530 Store->getAlignment()));
531 }
532 return DAG.getNode(ISD::TokenFactor, SL, MVT::Other, &Chains[0], NumElts);
533}
534
535SDValue AMDGPUTargetLowering::LowerSTORE(SDValue Op, SelectionDAG &DAG) const {
536 SDValue Result = AMDGPUTargetLowering::MergeVectorStore(Op, DAG);
537 if (Result.getNode()) {
538 return Result;
539 }
540
541 StoreSDNode *Store = cast<StoreSDNode>(Op);
542 if (Store->getAddressSpace() == AMDGPUAS::LOCAL_ADDRESS &&
543 Store->getValue().getValueType().isVector()) {
544 return SplitVectorStore(Op, DAG);
545 }
546 return SDValue();
547}
Tom Stellardf98f2ce2012-12-11 21:25:42 +0000548
549SDValue AMDGPUTargetLowering::LowerUDIVREM(SDValue Op,
550 SelectionDAG &DAG) const {
Andrew Trickac6d9be2013-05-25 02:42:55 +0000551 SDLoc DL(Op);
Tom Stellardf98f2ce2012-12-11 21:25:42 +0000552 EVT VT = Op.getValueType();
553
554 SDValue Num = Op.getOperand(0);
555 SDValue Den = Op.getOperand(1);
556
557 SmallVector<SDValue, 8> Results;
558
559 // RCP = URECIP(Den) = 2^32 / Den + e
560 // e is rounding error.
561 SDValue RCP = DAG.getNode(AMDGPUISD::URECIP, DL, VT, Den);
562
563 // RCP_LO = umulo(RCP, Den) */
564 SDValue RCP_LO = DAG.getNode(ISD::UMULO, DL, VT, RCP, Den);
565
566 // RCP_HI = mulhu (RCP, Den) */
567 SDValue RCP_HI = DAG.getNode(ISD::MULHU, DL, VT, RCP, Den);
568
569 // NEG_RCP_LO = -RCP_LO
570 SDValue NEG_RCP_LO = DAG.getNode(ISD::SUB, DL, VT, DAG.getConstant(0, VT),
571 RCP_LO);
572
573 // ABS_RCP_LO = (RCP_HI == 0 ? NEG_RCP_LO : RCP_LO)
574 SDValue ABS_RCP_LO = DAG.getSelectCC(DL, RCP_HI, DAG.getConstant(0, VT),
575 NEG_RCP_LO, RCP_LO,
576 ISD::SETEQ);
577 // Calculate the rounding error from the URECIP instruction
578 // E = mulhu(ABS_RCP_LO, RCP)
579 SDValue E = DAG.getNode(ISD::MULHU, DL, VT, ABS_RCP_LO, RCP);
580
581 // RCP_A_E = RCP + E
582 SDValue RCP_A_E = DAG.getNode(ISD::ADD, DL, VT, RCP, E);
583
584 // RCP_S_E = RCP - E
585 SDValue RCP_S_E = DAG.getNode(ISD::SUB, DL, VT, RCP, E);
586
587 // Tmp0 = (RCP_HI == 0 ? RCP_A_E : RCP_SUB_E)
588 SDValue Tmp0 = DAG.getSelectCC(DL, RCP_HI, DAG.getConstant(0, VT),
589 RCP_A_E, RCP_S_E,
590 ISD::SETEQ);
591 // Quotient = mulhu(Tmp0, Num)
592 SDValue Quotient = DAG.getNode(ISD::MULHU, DL, VT, Tmp0, Num);
593
594 // Num_S_Remainder = Quotient * Den
595 SDValue Num_S_Remainder = DAG.getNode(ISD::UMULO, DL, VT, Quotient, Den);
596
597 // Remainder = Num - Num_S_Remainder
598 SDValue Remainder = DAG.getNode(ISD::SUB, DL, VT, Num, Num_S_Remainder);
599
600 // Remainder_GE_Den = (Remainder >= Den ? -1 : 0)
601 SDValue Remainder_GE_Den = DAG.getSelectCC(DL, Remainder, Den,
602 DAG.getConstant(-1, VT),
603 DAG.getConstant(0, VT),
604 ISD::SETGE);
605 // Remainder_GE_Zero = (Remainder >= 0 ? -1 : 0)
606 SDValue Remainder_GE_Zero = DAG.getSelectCC(DL, Remainder,
607 DAG.getConstant(0, VT),
608 DAG.getConstant(-1, VT),
609 DAG.getConstant(0, VT),
610 ISD::SETGE);
611 // Tmp1 = Remainder_GE_Den & Remainder_GE_Zero
612 SDValue Tmp1 = DAG.getNode(ISD::AND, DL, VT, Remainder_GE_Den,
613 Remainder_GE_Zero);
614
615 // Calculate Division result:
616
617 // Quotient_A_One = Quotient + 1
618 SDValue Quotient_A_One = DAG.getNode(ISD::ADD, DL, VT, Quotient,
619 DAG.getConstant(1, VT));
620
621 // Quotient_S_One = Quotient - 1
622 SDValue Quotient_S_One = DAG.getNode(ISD::SUB, DL, VT, Quotient,
623 DAG.getConstant(1, VT));
624
625 // Div = (Tmp1 == 0 ? Quotient : Quotient_A_One)
626 SDValue Div = DAG.getSelectCC(DL, Tmp1, DAG.getConstant(0, VT),
627 Quotient, Quotient_A_One, ISD::SETEQ);
628
629 // Div = (Remainder_GE_Zero == 0 ? Quotient_S_One : Div)
630 Div = DAG.getSelectCC(DL, Remainder_GE_Zero, DAG.getConstant(0, VT),
631 Quotient_S_One, Div, ISD::SETEQ);
632
633 // Calculate Rem result:
634
635 // Remainder_S_Den = Remainder - Den
636 SDValue Remainder_S_Den = DAG.getNode(ISD::SUB, DL, VT, Remainder, Den);
637
638 // Remainder_A_Den = Remainder + Den
639 SDValue Remainder_A_Den = DAG.getNode(ISD::ADD, DL, VT, Remainder, Den);
640
641 // Rem = (Tmp1 == 0 ? Remainder : Remainder_S_Den)
642 SDValue Rem = DAG.getSelectCC(DL, Tmp1, DAG.getConstant(0, VT),
643 Remainder, Remainder_S_Den, ISD::SETEQ);
644
645 // Rem = (Remainder_GE_Zero == 0 ? Remainder_A_Den : Rem)
646 Rem = DAG.getSelectCC(DL, Remainder_GE_Zero, DAG.getConstant(0, VT),
647 Remainder_A_Den, Rem, ISD::SETEQ);
648 SDValue Ops[2];
649 Ops[0] = Div;
650 Ops[1] = Rem;
651 return DAG.getMergeValues(Ops, 2, DL);
652}
653
Tom Stellard4c52d452013-08-16 01:12:11 +0000654
Tom Stellardf98f2ce2012-12-11 21:25:42 +0000655//===----------------------------------------------------------------------===//
656// Helper functions
657//===----------------------------------------------------------------------===//
658
659bool AMDGPUTargetLowering::isHWTrueValue(SDValue Op) const {
660 if (ConstantFPSDNode * CFP = dyn_cast<ConstantFPSDNode>(Op)) {
661 return CFP->isExactlyValue(1.0);
662 }
663 if (ConstantSDNode *C = dyn_cast<ConstantSDNode>(Op)) {
664 return C->isAllOnesValue();
665 }
666 return false;
667}
668
669bool AMDGPUTargetLowering::isHWFalseValue(SDValue Op) const {
670 if (ConstantFPSDNode * CFP = dyn_cast<ConstantFPSDNode>(Op)) {
671 return CFP->getValueAPF().isZero();
672 }
673 if (ConstantSDNode *C = dyn_cast<ConstantSDNode>(Op)) {
674 return C->isNullValue();
675 }
676 return false;
677}
678
679SDValue AMDGPUTargetLowering::CreateLiveInRegister(SelectionDAG &DAG,
680 const TargetRegisterClass *RC,
681 unsigned Reg, EVT VT) const {
682 MachineFunction &MF = DAG.getMachineFunction();
683 MachineRegisterInfo &MRI = MF.getRegInfo();
684 unsigned VirtualRegister;
685 if (!MRI.isLiveIn(Reg)) {
686 VirtualRegister = MRI.createVirtualRegister(RC);
687 MRI.addLiveIn(Reg, VirtualRegister);
688 } else {
689 VirtualRegister = MRI.getLiveInVirtReg(Reg);
690 }
691 return DAG.getRegister(VirtualRegister, VT);
692}
693
694#define NODE_NAME_CASE(node) case AMDGPUISD::node: return #node;
695
696const char* AMDGPUTargetLowering::getTargetNodeName(unsigned Opcode) const {
697 switch (Opcode) {
698 default: return 0;
699 // AMDIL DAG nodes
Tom Stellardf98f2ce2012-12-11 21:25:42 +0000700 NODE_NAME_CASE(CALL);
701 NODE_NAME_CASE(UMUL);
702 NODE_NAME_CASE(DIV_INF);
703 NODE_NAME_CASE(RET_FLAG);
704 NODE_NAME_CASE(BRANCH_COND);
705
706 // AMDGPU DAG nodes
707 NODE_NAME_CASE(DWORDADDR)
708 NODE_NAME_CASE(FRACT)
709 NODE_NAME_CASE(FMAX)
710 NODE_NAME_CASE(SMAX)
711 NODE_NAME_CASE(UMAX)
712 NODE_NAME_CASE(FMIN)
713 NODE_NAME_CASE(SMIN)
714 NODE_NAME_CASE(UMIN)
715 NODE_NAME_CASE(URECIP)
Tom Stellardf98f2ce2012-12-11 21:25:42 +0000716 NODE_NAME_CASE(EXPORT)
Tom Stellardc7e18882013-01-23 02:09:03 +0000717 NODE_NAME_CASE(CONST_ADDRESS)
Tom Stellardc0b0c672013-02-06 17:32:29 +0000718 NODE_NAME_CASE(REGISTER_LOAD)
719 NODE_NAME_CASE(REGISTER_STORE)
Tom Stellard68db37b2013-08-14 23:24:45 +0000720 NODE_NAME_CASE(LOAD_CONSTANT)
721 NODE_NAME_CASE(LOAD_INPUT)
722 NODE_NAME_CASE(SAMPLE)
723 NODE_NAME_CASE(SAMPLEB)
724 NODE_NAME_CASE(SAMPLED)
725 NODE_NAME_CASE(SAMPLEL)
Tom Stellardec484272013-08-16 01:12:06 +0000726 NODE_NAME_CASE(STORE_MSKOR)
Tom Stellarda3c2bcf2013-09-12 02:55:14 +0000727 NODE_NAME_CASE(TBUFFER_STORE_FORMAT)
Tom Stellardf98f2ce2012-12-11 21:25:42 +0000728 }
729}