blob: 156423500af98dd9a9848b26a51a400032cfa970 [file] [log] [blame]
Matt Arsenault7836f892016-01-20 21:22:21 +00001//===-- AMDGPUISelDAGToDAG.cpp - A dag to dag inst selector for AMDGPU ----===//
Tom Stellard75aadc22012-12-11 21:25:42 +00002//
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 Defines an instruction selector for the AMDGPU target.
12//
13//===----------------------------------------------------------------------===//
Matt Arsenault592d0682015-12-01 23:04:05 +000014
Tom Stellard75aadc22012-12-11 21:25:42 +000015#include "AMDGPUInstrInfo.h"
Tom Stellardbc4497b2016-02-12 23:45:29 +000016#include "AMDGPUIntrinsicInfo.h"
Tom Stellard75aadc22012-12-11 21:25:42 +000017#include "AMDGPUISelLowering.h" // For AMDGPUISD
Tom Stellard2e59a452014-06-13 01:32:00 +000018#include "AMDGPUSubtarget.h"
Christian Konigf82901a2013-02-26 17:52:23 +000019#include "SIISelLowering.h"
Tom Stellardb02094e2014-07-21 15:45:01 +000020#include "SIMachineFunctionInfo.h"
Tom Stellard58ac7442014-04-29 23:12:48 +000021#include "llvm/CodeGen/FunctionLoweringInfo.h"
Matt Arsenaultf43c2a02016-03-23 21:49:25 +000022#include "llvm/CodeGen/MachineFrameInfo.h"
Matt Arsenaultd9d659a2015-11-03 22:30:08 +000023#include "llvm/CodeGen/PseudoSourceValue.h"
Benjamin Kramerd78bb462013-05-23 17:10:37 +000024#include "llvm/CodeGen/SelectionDAG.h"
Tom Stellard75aadc22012-12-11 21:25:42 +000025#include "llvm/CodeGen/SelectionDAGISel.h"
Oliver Stannard7e7d9832016-02-02 13:52:43 +000026#include "llvm/IR/DiagnosticInfo.h"
Tom Stellard75aadc22012-12-11 21:25:42 +000027
28using namespace llvm;
29
Matt Arsenaultd2759212016-02-13 01:24:08 +000030namespace llvm {
31class R600InstrInfo;
32}
33
Tom Stellard75aadc22012-12-11 21:25:42 +000034//===----------------------------------------------------------------------===//
35// Instruction Selector Implementation
36//===----------------------------------------------------------------------===//
37
38namespace {
Tom Stellardbc4497b2016-02-12 23:45:29 +000039
40static bool isCBranchSCC(const SDNode *N) {
41 assert(N->getOpcode() == ISD::BRCOND);
42 if (!N->hasOneUse())
43 return false;
44
45 SDValue Cond = N->getOperand(1);
46 if (Cond.getOpcode() == ISD::CopyToReg)
47 Cond = Cond.getOperand(2);
48 return Cond.getOpcode() == ISD::SETCC &&
49 Cond.getOperand(0).getValueType() == MVT::i32 &&
50 Cond.hasOneUse();
51}
52
Tom Stellard75aadc22012-12-11 21:25:42 +000053/// AMDGPU specific code to select AMDGPU machine instructions for
54/// SelectionDAG operations.
55class AMDGPUDAGToDAGISel : public SelectionDAGISel {
56 // Subtarget - Keep a pointer to the AMDGPU Subtarget around so that we can
57 // make the right decision when generating code for different targets.
Eric Christopher7792e322015-01-30 23:24:40 +000058 const AMDGPUSubtarget *Subtarget;
NAKAMURA Takumia9cb5382015-09-22 11:14:39 +000059
Tom Stellard75aadc22012-12-11 21:25:42 +000060public:
61 AMDGPUDAGToDAGISel(TargetMachine &TM);
62 virtual ~AMDGPUDAGToDAGISel();
Eric Christopher7792e322015-01-30 23:24:40 +000063 bool runOnMachineFunction(MachineFunction &MF) override;
Craig Topper5656db42014-04-29 07:57:24 +000064 SDNode *Select(SDNode *N) override;
65 const char *getPassName() const override;
Matt Arsenault4bf43d42015-09-25 17:27:08 +000066 void PreprocessISelDAG() override;
Craig Topper5656db42014-04-29 07:57:24 +000067 void PostprocessISelDAG() override;
Tom Stellard75aadc22012-12-11 21:25:42 +000068
69private:
Tom Stellard7ed0b522014-04-03 20:19:27 +000070 bool isInlineImmediate(SDNode *N) const;
Vincent Lejeunec6896792013-06-04 23:17:15 +000071 bool FoldOperand(SDValue &Src, SDValue &Sel, SDValue &Neg, SDValue &Abs,
Tom Stellard84021442013-07-23 01:48:24 +000072 const R600InstrInfo *TII);
Tom Stellard365366f2013-01-23 02:09:06 +000073 bool FoldOperands(unsigned, const R600InstrInfo *, std::vector<SDValue> &);
Vincent Lejeunec6896792013-06-04 23:17:15 +000074 bool FoldDotOperands(unsigned, const R600InstrInfo *, std::vector<SDValue> &);
Tom Stellard75aadc22012-12-11 21:25:42 +000075
76 // Complex pattern selectors
77 bool SelectADDRParam(SDValue Addr, SDValue& R1, SDValue& R2);
78 bool SelectADDR(SDValue N, SDValue &R1, SDValue &R2);
79 bool SelectADDR64(SDValue N, SDValue &R1, SDValue &R2);
80
81 static bool checkType(const Value *ptr, unsigned int addrspace);
Nick Lewyckyaad475b2014-04-15 07:22:52 +000082 static bool checkPrivateAddress(const MachineMemOperand *Op);
Tom Stellard75aadc22012-12-11 21:25:42 +000083
84 static bool isGlobalStore(const StoreSDNode *N);
Matt Arsenault3f981402014-09-15 15:41:53 +000085 static bool isFlatStore(const StoreSDNode *N);
Tom Stellard75aadc22012-12-11 21:25:42 +000086 static bool isPrivateStore(const StoreSDNode *N);
87 static bool isLocalStore(const StoreSDNode *N);
88 static bool isRegionStore(const StoreSDNode *N);
89
Matt Arsenault2aabb062013-06-18 23:37:58 +000090 bool isCPLoad(const LoadSDNode *N) const;
91 bool isConstantLoad(const LoadSDNode *N, int cbID) const;
92 bool isGlobalLoad(const LoadSDNode *N) const;
Matt Arsenault3f981402014-09-15 15:41:53 +000093 bool isFlatLoad(const LoadSDNode *N) const;
Matt Arsenault2aabb062013-06-18 23:37:58 +000094 bool isParamLoad(const LoadSDNode *N) const;
95 bool isPrivateLoad(const LoadSDNode *N) const;
96 bool isLocalLoad(const LoadSDNode *N) const;
97 bool isRegionLoad(const LoadSDNode *N) const;
Tom Stellard75aadc22012-12-11 21:25:42 +000098
Tom Stellardbc4497b2016-02-12 23:45:29 +000099 bool isUniformBr(const SDNode *N) const;
100
Tom Stellard381a94a2015-05-12 15:00:49 +0000101 SDNode *glueCopyToM0(SDNode *N) const;
102
Tom Stellarddf94dc32013-08-14 23:24:24 +0000103 const TargetRegisterClass *getOperandRegClass(SDNode *N, unsigned OpNo) const;
Tom Stellard365366f2013-01-23 02:09:06 +0000104 bool SelectGlobalValueConstantOffset(SDValue Addr, SDValue& IntPtr);
Matt Arsenault209a7b92014-04-18 07:40:20 +0000105 bool SelectGlobalValueVariableOffset(SDValue Addr, SDValue &BaseReg,
106 SDValue& Offset);
Tom Stellard75aadc22012-12-11 21:25:42 +0000107 bool SelectADDRVTX_READ(SDValue Addr, SDValue &Base, SDValue &Offset);
Tom Stellardf3b2a1e2013-02-06 17:32:29 +0000108 bool SelectADDRIndirect(SDValue Addr, SDValue &Base, SDValue &Offset);
Tom Stellard85e8b6d2014-08-22 18:49:33 +0000109 bool isDSOffsetLegal(const SDValue &Base, unsigned Offset,
110 unsigned OffsetBits) const;
111 bool SelectDS1Addr1Offset(SDValue Ptr, SDValue &Base, SDValue &Offset) const;
Tom Stellardf3fc5552014-08-22 18:49:35 +0000112 bool SelectDS64Bit4ByteAligned(SDValue Ptr, SDValue &Base, SDValue &Offset0,
113 SDValue &Offset1) const;
Changpeng Fangb41574a2015-12-22 20:55:23 +0000114 bool SelectMUBUF(SDValue Addr, SDValue &SRsrc, SDValue &VAddr,
Tom Stellard155bbb72014-08-11 22:18:17 +0000115 SDValue &SOffset, SDValue &Offset, SDValue &Offen,
116 SDValue &Idxen, SDValue &Addr64, SDValue &GLC, SDValue &SLC,
117 SDValue &TFE) const;
118 bool SelectMUBUFAddr64(SDValue Addr, SDValue &SRsrc, SDValue &VAddr,
Tom Stellard1f9939f2015-02-27 14:59:41 +0000119 SDValue &SOffset, SDValue &Offset, SDValue &GLC,
120 SDValue &SLC, SDValue &TFE) const;
Tom Stellard7980fc82014-09-25 18:30:26 +0000121 bool SelectMUBUFAddr64(SDValue Addr, SDValue &SRsrc,
Tom Stellardc53861a2015-02-11 00:34:32 +0000122 SDValue &VAddr, SDValue &SOffset, SDValue &Offset,
Tom Stellard7980fc82014-09-25 18:30:26 +0000123 SDValue &SLC) const;
Tom Stellardb02094e2014-07-21 15:45:01 +0000124 bool SelectMUBUFScratch(SDValue Addr, SDValue &RSrc, SDValue &VAddr,
125 SDValue &SOffset, SDValue &ImmOffset) const;
Tom Stellard155bbb72014-08-11 22:18:17 +0000126 bool SelectMUBUFOffset(SDValue Addr, SDValue &SRsrc, SDValue &SOffset,
127 SDValue &Offset, SDValue &GLC, SDValue &SLC,
Tom Stellardb02094e2014-07-21 15:45:01 +0000128 SDValue &TFE) const;
Tom Stellard7980fc82014-09-25 18:30:26 +0000129 bool SelectMUBUFOffset(SDValue Addr, SDValue &SRsrc, SDValue &Soffset,
130 SDValue &Offset, SDValue &GLC) const;
Nicolai Haehnle3003ba02016-03-18 16:24:20 +0000131 void SelectMUBUFConstant(SDValue Constant,
132 SDValue &SOffset,
133 SDValue &ImmOffset) const;
134 bool SelectMUBUFIntrinsicOffset(SDValue Offset, SDValue &SOffset,
135 SDValue &ImmOffset) const;
136 bool SelectMUBUFIntrinsicVOffset(SDValue Offset, SDValue &SOffset,
137 SDValue &ImmOffset, SDValue &VOffset) const;
Tom Stellarddee26a22015-08-06 19:28:30 +0000138 bool SelectSMRDOffset(SDValue ByteOffsetNode, SDValue &Offset,
139 bool &Imm) const;
140 bool SelectSMRD(SDValue Addr, SDValue &SBase, SDValue &Offset,
141 bool &Imm) const;
142 bool SelectSMRDImm(SDValue Addr, SDValue &SBase, SDValue &Offset) const;
Tom Stellard217361c2015-08-06 19:28:38 +0000143 bool SelectSMRDImm32(SDValue Addr, SDValue &SBase, SDValue &Offset) const;
Tom Stellarddee26a22015-08-06 19:28:30 +0000144 bool SelectSMRDSgpr(SDValue Addr, SDValue &SBase, SDValue &Offset) const;
145 bool SelectSMRDBufferImm(SDValue Addr, SDValue &Offset) const;
Tom Stellard217361c2015-08-06 19:28:38 +0000146 bool SelectSMRDBufferImm32(SDValue Addr, SDValue &Offset) const;
Tom Stellarddee26a22015-08-06 19:28:30 +0000147 bool SelectSMRDBufferSgpr(SDValue Addr, SDValue &Offset) const;
Matt Arsenault3f981402014-09-15 15:41:53 +0000148 SDNode *SelectAddrSpaceCast(SDNode *N);
Tom Stellardb4a313a2014-08-01 00:32:39 +0000149 bool SelectVOP3Mods(SDValue In, SDValue &Src, SDValue &SrcMods) const;
Tom Stellarddb5a11f2015-07-13 15:47:57 +0000150 bool SelectVOP3NoMods(SDValue In, SDValue &Src, SDValue &SrcMods) const;
Tom Stellardb4a313a2014-08-01 00:32:39 +0000151 bool SelectVOP3Mods0(SDValue In, SDValue &Src, SDValue &SrcMods,
152 SDValue &Clamp, SDValue &Omod) const;
Tom Stellarddb5a11f2015-07-13 15:47:57 +0000153 bool SelectVOP3NoMods0(SDValue In, SDValue &Src, SDValue &SrcMods,
154 SDValue &Clamp, SDValue &Omod) const;
Tom Stellard75aadc22012-12-11 21:25:42 +0000155
Matt Arsenault1cffa4c2014-11-13 19:49:04 +0000156 bool SelectVOP3Mods0Clamp(SDValue In, SDValue &Src, SDValue &SrcMods,
157 SDValue &Omod) const;
Matt Arsenault4831ce52015-01-06 23:00:37 +0000158 bool SelectVOP3Mods0Clamp0OMod(SDValue In, SDValue &Src, SDValue &SrcMods,
159 SDValue &Clamp,
160 SDValue &Omod) const;
Matt Arsenault1cffa4c2014-11-13 19:49:04 +0000161
Matt Arsenaultb8b51532014-06-23 18:00:38 +0000162 SDNode *SelectADD_SUB_I64(SDNode *N);
Matt Arsenaultf2b0aeb2014-06-23 18:28:28 +0000163 SDNode *SelectDIV_SCALE(SDNode *N);
Matt Arsenault9fa3f932014-06-23 18:00:34 +0000164
Marek Olsak9b728682015-03-24 13:40:27 +0000165 SDNode *getS_BFE(unsigned Opcode, SDLoc DL, SDValue Val,
166 uint32_t Offset, uint32_t Width);
167 SDNode *SelectS_BFEFromShifts(SDNode *N);
168 SDNode *SelectS_BFE(SDNode *N);
Tom Stellardbc4497b2016-02-12 23:45:29 +0000169 SDNode *SelectBRCOND(SDNode *N);
Marek Olsak9b728682015-03-24 13:40:27 +0000170
Tom Stellard75aadc22012-12-11 21:25:42 +0000171 // Include the pieces autogenerated from the target description.
172#include "AMDGPUGenDAGISel.inc"
173};
174} // end anonymous namespace
175
176/// \brief This pass converts a legalized DAG into a AMDGPU-specific
177// DAG, ready for instruction scheduling.
Matt Arsenault209a7b92014-04-18 07:40:20 +0000178FunctionPass *llvm::createAMDGPUISelDag(TargetMachine &TM) {
Tom Stellard75aadc22012-12-11 21:25:42 +0000179 return new AMDGPUDAGToDAGISel(TM);
180}
181
Bill Wendlinga3cd3502013-06-19 21:36:55 +0000182AMDGPUDAGToDAGISel::AMDGPUDAGToDAGISel(TargetMachine &TM)
Eric Christopher7792e322015-01-30 23:24:40 +0000183 : SelectionDAGISel(TM) {}
184
185bool AMDGPUDAGToDAGISel::runOnMachineFunction(MachineFunction &MF) {
186 Subtarget = &static_cast<const AMDGPUSubtarget &>(MF.getSubtarget());
187 return SelectionDAGISel::runOnMachineFunction(MF);
Tom Stellard75aadc22012-12-11 21:25:42 +0000188}
189
190AMDGPUDAGToDAGISel::~AMDGPUDAGToDAGISel() {
191}
192
Tom Stellard7ed0b522014-04-03 20:19:27 +0000193bool AMDGPUDAGToDAGISel::isInlineImmediate(SDNode *N) const {
194 const SITargetLowering *TL
195 = static_cast<const SITargetLowering *>(getTargetLowering());
196 return TL->analyzeImmediate(N) == 0;
197}
198
Tom Stellarddf94dc32013-08-14 23:24:24 +0000199/// \brief Determine the register class for \p OpNo
200/// \returns The register class of the virtual register that will be used for
201/// the given operand number \OpNo or NULL if the register class cannot be
202/// determined.
203const TargetRegisterClass *AMDGPUDAGToDAGISel::getOperandRegClass(SDNode *N,
204 unsigned OpNo) const {
Matt Arsenault209a7b92014-04-18 07:40:20 +0000205 if (!N->isMachineOpcode())
206 return nullptr;
207
Tom Stellarddf94dc32013-08-14 23:24:24 +0000208 switch (N->getMachineOpcode()) {
209 default: {
Eric Christopherd9134482014-08-04 21:25:23 +0000210 const MCInstrDesc &Desc =
Eric Christopher7792e322015-01-30 23:24:40 +0000211 Subtarget->getInstrInfo()->get(N->getMachineOpcode());
Alexey Samsonov3186eb32013-08-15 07:11:34 +0000212 unsigned OpIdx = Desc.getNumDefs() + OpNo;
213 if (OpIdx >= Desc.getNumOperands())
Matt Arsenault209a7b92014-04-18 07:40:20 +0000214 return nullptr;
Alexey Samsonov3186eb32013-08-15 07:11:34 +0000215 int RegClass = Desc.OpInfo[OpIdx].RegClass;
Matt Arsenault209a7b92014-04-18 07:40:20 +0000216 if (RegClass == -1)
217 return nullptr;
218
Eric Christopher7792e322015-01-30 23:24:40 +0000219 return Subtarget->getRegisterInfo()->getRegClass(RegClass);
Tom Stellarddf94dc32013-08-14 23:24:24 +0000220 }
221 case AMDGPU::REG_SEQUENCE: {
Matt Arsenault209a7b92014-04-18 07:40:20 +0000222 unsigned RCID = cast<ConstantSDNode>(N->getOperand(0))->getZExtValue();
Eric Christopherd9134482014-08-04 21:25:23 +0000223 const TargetRegisterClass *SuperRC =
Eric Christopher7792e322015-01-30 23:24:40 +0000224 Subtarget->getRegisterInfo()->getRegClass(RCID);
Matt Arsenault209a7b92014-04-18 07:40:20 +0000225
226 SDValue SubRegOp = N->getOperand(OpNo + 1);
227 unsigned SubRegIdx = cast<ConstantSDNode>(SubRegOp)->getZExtValue();
Eric Christopher7792e322015-01-30 23:24:40 +0000228 return Subtarget->getRegisterInfo()->getSubClassWithSubReg(SuperRC,
229 SubRegIdx);
Tom Stellarddf94dc32013-08-14 23:24:24 +0000230 }
231 }
232}
233
Tom Stellard75aadc22012-12-11 21:25:42 +0000234bool AMDGPUDAGToDAGISel::SelectADDRParam(
Matt Arsenault209a7b92014-04-18 07:40:20 +0000235 SDValue Addr, SDValue& R1, SDValue& R2) {
Tom Stellard75aadc22012-12-11 21:25:42 +0000236
237 if (Addr.getOpcode() == ISD::FrameIndex) {
238 if (FrameIndexSDNode *FIN = dyn_cast<FrameIndexSDNode>(Addr)) {
239 R1 = CurDAG->getTargetFrameIndex(FIN->getIndex(), MVT::i32);
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +0000240 R2 = CurDAG->getTargetConstant(0, SDLoc(Addr), MVT::i32);
Tom Stellard75aadc22012-12-11 21:25:42 +0000241 } else {
242 R1 = Addr;
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +0000243 R2 = CurDAG->getTargetConstant(0, SDLoc(Addr), MVT::i32);
Tom Stellard75aadc22012-12-11 21:25:42 +0000244 }
245 } else if (Addr.getOpcode() == ISD::ADD) {
246 R1 = Addr.getOperand(0);
247 R2 = Addr.getOperand(1);
248 } else {
249 R1 = Addr;
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +0000250 R2 = CurDAG->getTargetConstant(0, SDLoc(Addr), MVT::i32);
Tom Stellard75aadc22012-12-11 21:25:42 +0000251 }
252 return true;
253}
254
255bool AMDGPUDAGToDAGISel::SelectADDR(SDValue Addr, SDValue& R1, SDValue& R2) {
256 if (Addr.getOpcode() == ISD::TargetExternalSymbol ||
257 Addr.getOpcode() == ISD::TargetGlobalAddress) {
258 return false;
259 }
260 return SelectADDRParam(Addr, R1, R2);
261}
262
263
264bool AMDGPUDAGToDAGISel::SelectADDR64(SDValue Addr, SDValue& R1, SDValue& R2) {
265 if (Addr.getOpcode() == ISD::TargetExternalSymbol ||
266 Addr.getOpcode() == ISD::TargetGlobalAddress) {
267 return false;
268 }
269
270 if (Addr.getOpcode() == ISD::FrameIndex) {
271 if (FrameIndexSDNode *FIN = dyn_cast<FrameIndexSDNode>(Addr)) {
272 R1 = CurDAG->getTargetFrameIndex(FIN->getIndex(), MVT::i64);
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +0000273 R2 = CurDAG->getTargetConstant(0, SDLoc(Addr), MVT::i64);
Tom Stellard75aadc22012-12-11 21:25:42 +0000274 } else {
275 R1 = Addr;
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +0000276 R2 = CurDAG->getTargetConstant(0, SDLoc(Addr), MVT::i64);
Tom Stellard75aadc22012-12-11 21:25:42 +0000277 }
278 } else if (Addr.getOpcode() == ISD::ADD) {
279 R1 = Addr.getOperand(0);
280 R2 = Addr.getOperand(1);
281 } else {
282 R1 = Addr;
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +0000283 R2 = CurDAG->getTargetConstant(0, SDLoc(Addr), MVT::i64);
Tom Stellard75aadc22012-12-11 21:25:42 +0000284 }
285 return true;
286}
287
Tom Stellard381a94a2015-05-12 15:00:49 +0000288SDNode *AMDGPUDAGToDAGISel::glueCopyToM0(SDNode *N) const {
289 if (Subtarget->getGeneration() < AMDGPUSubtarget::SOUTHERN_ISLANDS ||
290 !checkType(cast<MemSDNode>(N)->getMemOperand()->getValue(),
291 AMDGPUAS::LOCAL_ADDRESS))
292 return N;
293
294 const SITargetLowering& Lowering =
295 *static_cast<const SITargetLowering*>(getTargetLowering());
296
297 // Write max value to m0 before each load operation
298
299 SDValue M0 = Lowering.copyToM0(*CurDAG, CurDAG->getEntryNode(), SDLoc(N),
300 CurDAG->getTargetConstant(-1, SDLoc(N), MVT::i32));
301
302 SDValue Glue = M0.getValue(1);
303
304 SmallVector <SDValue, 8> Ops;
305 for (unsigned i = 0, e = N->getNumOperands(); i != e; ++i) {
306 Ops.push_back(N->getOperand(i));
307 }
308 Ops.push_back(Glue);
309 CurDAG->MorphNodeTo(N, N->getOpcode(), N->getVTList(), Ops);
310
311 return N;
312}
313
Matt Arsenault61cb6fa2015-11-11 00:01:36 +0000314static unsigned selectSGPRVectorRegClassID(unsigned NumVectorElts) {
Matt Arsenaultf1aebbf2015-11-02 23:30:48 +0000315 switch (NumVectorElts) {
316 case 1:
317 return AMDGPU::SReg_32RegClassID;
318 case 2:
319 return AMDGPU::SReg_64RegClassID;
320 case 4:
321 return AMDGPU::SReg_128RegClassID;
322 case 8:
323 return AMDGPU::SReg_256RegClassID;
324 case 16:
325 return AMDGPU::SReg_512RegClassID;
326 }
327
328 llvm_unreachable("invalid vector size");
329}
330
Tom Stellard75aadc22012-12-11 21:25:42 +0000331SDNode *AMDGPUDAGToDAGISel::Select(SDNode *N) {
332 unsigned int Opc = N->getOpcode();
333 if (N->isMachineOpcode()) {
Tim Northover31d093c2013-09-22 08:21:56 +0000334 N->setNodeId(-1);
Matt Arsenault209a7b92014-04-18 07:40:20 +0000335 return nullptr; // Already selected.
Tom Stellard75aadc22012-12-11 21:25:42 +0000336 }
Matt Arsenault78b86702014-04-18 05:19:26 +0000337
Tom Stellard381a94a2015-05-12 15:00:49 +0000338 if (isa<AtomicSDNode>(N))
339 N = glueCopyToM0(N);
340
Tom Stellard75aadc22012-12-11 21:25:42 +0000341 switch (Opc) {
342 default: break;
Tom Stellard1f15bff2014-02-25 21:36:18 +0000343 // We are selecting i64 ADD here instead of custom lower it during
344 // DAG legalization, so we can fold some i64 ADDs used for address
345 // calculation into the LOAD and STORE instructions.
Matt Arsenaultb8b51532014-06-23 18:00:38 +0000346 case ISD::ADD:
347 case ISD::SUB: {
Tom Stellard1f15bff2014-02-25 21:36:18 +0000348 if (N->getValueType(0) != MVT::i64 ||
Eric Christopher7792e322015-01-30 23:24:40 +0000349 Subtarget->getGeneration() < AMDGPUSubtarget::SOUTHERN_ISLANDS)
Tom Stellard1f15bff2014-02-25 21:36:18 +0000350 break;
351
Matt Arsenaultb8b51532014-06-23 18:00:38 +0000352 return SelectADD_SUB_I64(N);
Tom Stellard1f15bff2014-02-25 21:36:18 +0000353 }
Matt Arsenault064c2062014-06-11 17:40:32 +0000354 case ISD::SCALAR_TO_VECTOR:
Tom Stellard880a80a2014-06-17 16:53:14 +0000355 case AMDGPUISD::BUILD_VERTICAL_VECTOR:
Vincent Lejeune3b6f20e2013-03-05 15:04:49 +0000356 case ISD::BUILD_VECTOR: {
Tom Stellard8e5da412013-08-14 23:24:32 +0000357 unsigned RegClassID;
Eric Christopher7792e322015-01-30 23:24:40 +0000358 const AMDGPURegisterInfo *TRI = Subtarget->getRegisterInfo();
Tom Stellard8e5da412013-08-14 23:24:32 +0000359 EVT VT = N->getValueType(0);
360 unsigned NumVectorElts = VT.getVectorNumElements();
Matt Arsenault064c2062014-06-11 17:40:32 +0000361 EVT EltVT = VT.getVectorElementType();
362 assert(EltVT.bitsEq(MVT::i32));
Eric Christopher7792e322015-01-30 23:24:40 +0000363 if (Subtarget->getGeneration() >= AMDGPUSubtarget::SOUTHERN_ISLANDS) {
Matt Arsenault61cb6fa2015-11-11 00:01:36 +0000364 RegClassID = selectSGPRVectorRegClassID(NumVectorElts);
Tom Stellard8e5da412013-08-14 23:24:32 +0000365 } else {
366 // BUILD_VECTOR was lowered into an IMPLICIT_DEF + 4 INSERT_SUBREG
367 // that adds a 128 bits reg copy when going through TwoAddressInstructions
368 // pass. We want to avoid 128 bits copies as much as possible because they
369 // can't be bundled by our scheduler.
370 switch(NumVectorElts) {
371 case 2: RegClassID = AMDGPU::R600_Reg64RegClassID; break;
Tom Stellard880a80a2014-06-17 16:53:14 +0000372 case 4:
373 if (Opc == AMDGPUISD::BUILD_VERTICAL_VECTOR)
374 RegClassID = AMDGPU::R600_Reg128VerticalRegClassID;
375 else
376 RegClassID = AMDGPU::R600_Reg128RegClassID;
377 break;
Tom Stellard8e5da412013-08-14 23:24:32 +0000378 default: llvm_unreachable("Do not know how to lower this BUILD_VECTOR");
379 }
Vincent Lejeune3b6f20e2013-03-05 15:04:49 +0000380 }
Tom Stellard0344cdf2013-08-01 15:23:42 +0000381
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +0000382 SDLoc DL(N);
383 SDValue RegClass = CurDAG->getTargetConstant(RegClassID, DL, MVT::i32);
Tom Stellard8e5da412013-08-14 23:24:32 +0000384
385 if (NumVectorElts == 1) {
Matt Arsenault064c2062014-06-11 17:40:32 +0000386 return CurDAG->SelectNodeTo(N, AMDGPU::COPY_TO_REGCLASS, EltVT,
Tom Stellard8e5da412013-08-14 23:24:32 +0000387 N->getOperand(0), RegClass);
Tom Stellard0344cdf2013-08-01 15:23:42 +0000388 }
Tom Stellard8e5da412013-08-14 23:24:32 +0000389
390 assert(NumVectorElts <= 16 && "Vectors with more than 16 elements not "
391 "supported yet");
392 // 16 = Max Num Vector Elements
393 // 2 = 2 REG_SEQUENCE operands per element (value, subreg index)
394 // 1 = Vector Register Class
Matt Arsenault064c2062014-06-11 17:40:32 +0000395 SmallVector<SDValue, 16 * 2 + 1> RegSeqArgs(NumVectorElts * 2 + 1);
Tom Stellard8e5da412013-08-14 23:24:32 +0000396
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +0000397 RegSeqArgs[0] = CurDAG->getTargetConstant(RegClassID, DL, MVT::i32);
Vincent Lejeune3b6f20e2013-03-05 15:04:49 +0000398 bool IsRegSeq = true;
Matt Arsenault064c2062014-06-11 17:40:32 +0000399 unsigned NOps = N->getNumOperands();
400 for (unsigned i = 0; i < NOps; i++) {
Tom Stellard8e5da412013-08-14 23:24:32 +0000401 // XXX: Why is this here?
Benjamin Kramer619c4e52015-04-10 11:24:51 +0000402 if (isa<RegisterSDNode>(N->getOperand(i))) {
Vincent Lejeune3b6f20e2013-03-05 15:04:49 +0000403 IsRegSeq = false;
404 break;
405 }
Tom Stellard8e5da412013-08-14 23:24:32 +0000406 RegSeqArgs[1 + (2 * i)] = N->getOperand(i);
407 RegSeqArgs[1 + (2 * i) + 1] =
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +0000408 CurDAG->getTargetConstant(TRI->getSubRegFromChannel(i), DL,
409 MVT::i32);
Vincent Lejeune3b6f20e2013-03-05 15:04:49 +0000410 }
Matt Arsenault064c2062014-06-11 17:40:32 +0000411
412 if (NOps != NumVectorElts) {
413 // Fill in the missing undef elements if this was a scalar_to_vector.
414 assert(Opc == ISD::SCALAR_TO_VECTOR && NOps < NumVectorElts);
415
416 MachineSDNode *ImpDef = CurDAG->getMachineNode(TargetOpcode::IMPLICIT_DEF,
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +0000417 DL, EltVT);
Matt Arsenault064c2062014-06-11 17:40:32 +0000418 for (unsigned i = NOps; i < NumVectorElts; ++i) {
419 RegSeqArgs[1 + (2 * i)] = SDValue(ImpDef, 0);
420 RegSeqArgs[1 + (2 * i) + 1] =
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +0000421 CurDAG->getTargetConstant(TRI->getSubRegFromChannel(i), DL, MVT::i32);
Matt Arsenault064c2062014-06-11 17:40:32 +0000422 }
423 }
424
Vincent Lejeune3b6f20e2013-03-05 15:04:49 +0000425 if (!IsRegSeq)
426 break;
427 return CurDAG->SelectNodeTo(N, AMDGPU::REG_SEQUENCE, N->getVTList(),
Craig Topper481fb282014-04-27 19:21:11 +0000428 RegSeqArgs);
Vincent Lejeune3b6f20e2013-03-05 15:04:49 +0000429 }
Tom Stellard754f80f2013-04-05 23:31:51 +0000430 case ISD::BUILD_PAIR: {
431 SDValue RC, SubReg0, SubReg1;
Eric Christopher7792e322015-01-30 23:24:40 +0000432 if (Subtarget->getGeneration() <= AMDGPUSubtarget::NORTHERN_ISLANDS) {
Tom Stellard754f80f2013-04-05 23:31:51 +0000433 break;
434 }
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +0000435 SDLoc DL(N);
Tom Stellard754f80f2013-04-05 23:31:51 +0000436 if (N->getValueType(0) == MVT::i128) {
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +0000437 RC = CurDAG->getTargetConstant(AMDGPU::SReg_128RegClassID, DL, MVT::i32);
438 SubReg0 = CurDAG->getTargetConstant(AMDGPU::sub0_sub1, DL, MVT::i32);
439 SubReg1 = CurDAG->getTargetConstant(AMDGPU::sub2_sub3, DL, MVT::i32);
Tom Stellard754f80f2013-04-05 23:31:51 +0000440 } else if (N->getValueType(0) == MVT::i64) {
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +0000441 RC = CurDAG->getTargetConstant(AMDGPU::SReg_64RegClassID, DL, MVT::i32);
442 SubReg0 = CurDAG->getTargetConstant(AMDGPU::sub0, DL, MVT::i32);
443 SubReg1 = CurDAG->getTargetConstant(AMDGPU::sub1, DL, MVT::i32);
Tom Stellard754f80f2013-04-05 23:31:51 +0000444 } else {
445 llvm_unreachable("Unhandled value type for BUILD_PAIR");
446 }
447 const SDValue Ops[] = { RC, N->getOperand(0), SubReg0,
448 N->getOperand(1), SubReg1 };
449 return CurDAG->getMachineNode(TargetOpcode::REG_SEQUENCE,
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +0000450 DL, N->getValueType(0), Ops);
Tom Stellard754f80f2013-04-05 23:31:51 +0000451 }
Tom Stellard7ed0b522014-04-03 20:19:27 +0000452
453 case ISD::Constant:
454 case ISD::ConstantFP: {
Eric Christopher7792e322015-01-30 23:24:40 +0000455 if (Subtarget->getGeneration() < AMDGPUSubtarget::SOUTHERN_ISLANDS ||
Tom Stellard7ed0b522014-04-03 20:19:27 +0000456 N->getValueType(0).getSizeInBits() != 64 || isInlineImmediate(N))
457 break;
458
459 uint64_t Imm;
460 if (ConstantFPSDNode *FP = dyn_cast<ConstantFPSDNode>(N))
461 Imm = FP->getValueAPF().bitcastToAPInt().getZExtValue();
462 else {
Tom Stellard3cbe0142014-04-07 19:31:13 +0000463 ConstantSDNode *C = cast<ConstantSDNode>(N);
Tom Stellard7ed0b522014-04-03 20:19:27 +0000464 Imm = C->getZExtValue();
465 }
466
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +0000467 SDLoc DL(N);
468 SDNode *Lo = CurDAG->getMachineNode(AMDGPU::S_MOV_B32, DL, MVT::i32,
469 CurDAG->getConstant(Imm & 0xFFFFFFFF, DL,
470 MVT::i32));
471 SDNode *Hi = CurDAG->getMachineNode(AMDGPU::S_MOV_B32, DL, MVT::i32,
472 CurDAG->getConstant(Imm >> 32, DL, MVT::i32));
Tom Stellard7ed0b522014-04-03 20:19:27 +0000473 const SDValue Ops[] = {
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +0000474 CurDAG->getTargetConstant(AMDGPU::SReg_64RegClassID, DL, MVT::i32),
475 SDValue(Lo, 0), CurDAG->getTargetConstant(AMDGPU::sub0, DL, MVT::i32),
476 SDValue(Hi, 0), CurDAG->getTargetConstant(AMDGPU::sub1, DL, MVT::i32)
Tom Stellard7ed0b522014-04-03 20:19:27 +0000477 };
478
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +0000479 return CurDAG->getMachineNode(TargetOpcode::REG_SEQUENCE, DL,
Tom Stellard7ed0b522014-04-03 20:19:27 +0000480 N->getValueType(0), Ops);
481 }
Matt Arsenault4bf43d42015-09-25 17:27:08 +0000482 case ISD::LOAD:
Tom Stellard096b8c12015-02-04 20:49:49 +0000483 case ISD::STORE: {
Tom Stellard381a94a2015-05-12 15:00:49 +0000484 N = glueCopyToM0(N);
Tom Stellard096b8c12015-02-04 20:49:49 +0000485 break;
486 }
Matt Arsenault78b86702014-04-18 05:19:26 +0000487
488 case AMDGPUISD::BFE_I32:
489 case AMDGPUISD::BFE_U32: {
Eric Christopher7792e322015-01-30 23:24:40 +0000490 if (Subtarget->getGeneration() < AMDGPUSubtarget::SOUTHERN_ISLANDS)
Matt Arsenault78b86702014-04-18 05:19:26 +0000491 break;
492
493 // There is a scalar version available, but unlike the vector version which
494 // has a separate operand for the offset and width, the scalar version packs
495 // the width and offset into a single operand. Try to move to the scalar
496 // version if the offsets are constant, so that we can try to keep extended
497 // loads of kernel arguments in SGPRs.
498
499 // TODO: Technically we could try to pattern match scalar bitshifts of
500 // dynamic values, but it's probably not useful.
501 ConstantSDNode *Offset = dyn_cast<ConstantSDNode>(N->getOperand(1));
502 if (!Offset)
503 break;
504
505 ConstantSDNode *Width = dyn_cast<ConstantSDNode>(N->getOperand(2));
506 if (!Width)
507 break;
508
509 bool Signed = Opc == AMDGPUISD::BFE_I32;
510
Matt Arsenault78b86702014-04-18 05:19:26 +0000511 uint32_t OffsetVal = Offset->getZExtValue();
512 uint32_t WidthVal = Width->getZExtValue();
513
Marek Olsak9b728682015-03-24 13:40:27 +0000514 return getS_BFE(Signed ? AMDGPU::S_BFE_I32 : AMDGPU::S_BFE_U32, SDLoc(N),
515 N->getOperand(0), OffsetVal, WidthVal);
Matt Arsenault78b86702014-04-18 05:19:26 +0000516 }
Matt Arsenaultf2b0aeb2014-06-23 18:28:28 +0000517 case AMDGPUISD::DIV_SCALE: {
518 return SelectDIV_SCALE(N);
519 }
Tom Stellard3457a842014-10-09 19:06:00 +0000520 case ISD::CopyToReg: {
521 const SITargetLowering& Lowering =
522 *static_cast<const SITargetLowering*>(getTargetLowering());
523 Lowering.legalizeTargetIndependentNode(N, *CurDAG);
524 break;
525 }
Matt Arsenault3f981402014-09-15 15:41:53 +0000526 case ISD::ADDRSPACECAST:
527 return SelectAddrSpaceCast(N);
Marek Olsak9b728682015-03-24 13:40:27 +0000528 case ISD::AND:
529 case ISD::SRL:
530 case ISD::SRA:
531 if (N->getValueType(0) != MVT::i32 ||
532 Subtarget->getGeneration() < AMDGPUSubtarget::SOUTHERN_ISLANDS)
533 break;
534
535 return SelectS_BFE(N);
Tom Stellardbc4497b2016-02-12 23:45:29 +0000536 case ISD::BRCOND:
537 return SelectBRCOND(N);
Tom Stellard75aadc22012-12-11 21:25:42 +0000538 }
Tom Stellard3457a842014-10-09 19:06:00 +0000539
Vincent Lejeune0167a312013-09-12 23:45:00 +0000540 return SelectCode(N);
Tom Stellard365366f2013-01-23 02:09:06 +0000541}
542
Matt Arsenault209a7b92014-04-18 07:40:20 +0000543bool AMDGPUDAGToDAGISel::checkType(const Value *Ptr, unsigned AS) {
544 assert(AS != 0 && "Use checkPrivateAddress instead.");
545 if (!Ptr)
Tom Stellard75aadc22012-12-11 21:25:42 +0000546 return false;
Matt Arsenault209a7b92014-04-18 07:40:20 +0000547
548 return Ptr->getType()->getPointerAddressSpace() == AS;
Tom Stellard75aadc22012-12-11 21:25:42 +0000549}
550
Nick Lewyckyaad475b2014-04-15 07:22:52 +0000551bool AMDGPUDAGToDAGISel::checkPrivateAddress(const MachineMemOperand *Op) {
Matt Arsenault209a7b92014-04-18 07:40:20 +0000552 if (Op->getPseudoValue())
553 return true;
554
555 if (PointerType *PT = dyn_cast<PointerType>(Op->getValue()->getType()))
556 return PT->getAddressSpace() == AMDGPUAS::PRIVATE_ADDRESS;
557
558 return false;
Nick Lewyckyaad475b2014-04-15 07:22:52 +0000559}
560
Tom Stellard75aadc22012-12-11 21:25:42 +0000561bool AMDGPUDAGToDAGISel::isGlobalStore(const StoreSDNode *N) {
Nick Lewyckyaad475b2014-04-15 07:22:52 +0000562 return checkType(N->getMemOperand()->getValue(), AMDGPUAS::GLOBAL_ADDRESS);
Tom Stellard75aadc22012-12-11 21:25:42 +0000563}
564
565bool AMDGPUDAGToDAGISel::isPrivateStore(const StoreSDNode *N) {
Matt Arsenault209a7b92014-04-18 07:40:20 +0000566 const Value *MemVal = N->getMemOperand()->getValue();
567 return (!checkType(MemVal, AMDGPUAS::LOCAL_ADDRESS) &&
568 !checkType(MemVal, AMDGPUAS::GLOBAL_ADDRESS) &&
569 !checkType(MemVal, AMDGPUAS::REGION_ADDRESS));
Tom Stellard75aadc22012-12-11 21:25:42 +0000570}
571
572bool AMDGPUDAGToDAGISel::isLocalStore(const StoreSDNode *N) {
Nick Lewyckyaad475b2014-04-15 07:22:52 +0000573 return checkType(N->getMemOperand()->getValue(), AMDGPUAS::LOCAL_ADDRESS);
Tom Stellard75aadc22012-12-11 21:25:42 +0000574}
575
Matt Arsenault3f981402014-09-15 15:41:53 +0000576bool AMDGPUDAGToDAGISel::isFlatStore(const StoreSDNode *N) {
577 return checkType(N->getMemOperand()->getValue(), AMDGPUAS::FLAT_ADDRESS);
578}
579
Tom Stellard75aadc22012-12-11 21:25:42 +0000580bool AMDGPUDAGToDAGISel::isRegionStore(const StoreSDNode *N) {
Nick Lewyckyaad475b2014-04-15 07:22:52 +0000581 return checkType(N->getMemOperand()->getValue(), AMDGPUAS::REGION_ADDRESS);
Tom Stellard75aadc22012-12-11 21:25:42 +0000582}
583
Tom Stellard1e803092013-07-23 01:48:18 +0000584bool AMDGPUDAGToDAGISel::isConstantLoad(const LoadSDNode *N, int CbId) const {
Matt Arsenault209a7b92014-04-18 07:40:20 +0000585 const Value *MemVal = N->getMemOperand()->getValue();
586 if (CbId == -1)
587 return checkType(MemVal, AMDGPUAS::CONSTANT_ADDRESS);
588
589 return checkType(MemVal, AMDGPUAS::CONSTANT_BUFFER_0 + CbId);
Tom Stellard75aadc22012-12-11 21:25:42 +0000590}
591
Matt Arsenault2aabb062013-06-18 23:37:58 +0000592bool AMDGPUDAGToDAGISel::isGlobalLoad(const LoadSDNode *N) const {
Eric Christopher7792e322015-01-30 23:24:40 +0000593 if (N->getAddressSpace() == AMDGPUAS::CONSTANT_ADDRESS)
594 if (Subtarget->getGeneration() < AMDGPUSubtarget::SOUTHERN_ISLANDS ||
595 N->getMemoryVT().bitsLT(MVT::i32))
Tom Stellard8cb0e472013-07-23 23:54:56 +0000596 return true;
Eric Christopher7792e322015-01-30 23:24:40 +0000597
Nick Lewyckyaad475b2014-04-15 07:22:52 +0000598 return checkType(N->getMemOperand()->getValue(), AMDGPUAS::GLOBAL_ADDRESS);
Tom Stellard75aadc22012-12-11 21:25:42 +0000599}
600
Matt Arsenault2aabb062013-06-18 23:37:58 +0000601bool AMDGPUDAGToDAGISel::isParamLoad(const LoadSDNode *N) const {
Nick Lewyckyaad475b2014-04-15 07:22:52 +0000602 return checkType(N->getMemOperand()->getValue(), AMDGPUAS::PARAM_I_ADDRESS);
Tom Stellard75aadc22012-12-11 21:25:42 +0000603}
604
Matt Arsenault2aabb062013-06-18 23:37:58 +0000605bool AMDGPUDAGToDAGISel::isLocalLoad(const LoadSDNode *N) const {
Nick Lewyckyaad475b2014-04-15 07:22:52 +0000606 return checkType(N->getMemOperand()->getValue(), AMDGPUAS::LOCAL_ADDRESS);
Tom Stellard75aadc22012-12-11 21:25:42 +0000607}
608
Matt Arsenault3f981402014-09-15 15:41:53 +0000609bool AMDGPUDAGToDAGISel::isFlatLoad(const LoadSDNode *N) const {
610 return checkType(N->getMemOperand()->getValue(), AMDGPUAS::FLAT_ADDRESS);
611}
612
Matt Arsenault2aabb062013-06-18 23:37:58 +0000613bool AMDGPUDAGToDAGISel::isRegionLoad(const LoadSDNode *N) const {
Nick Lewyckyaad475b2014-04-15 07:22:52 +0000614 return checkType(N->getMemOperand()->getValue(), AMDGPUAS::REGION_ADDRESS);
Tom Stellard75aadc22012-12-11 21:25:42 +0000615}
616
Matt Arsenault2aabb062013-06-18 23:37:58 +0000617bool AMDGPUDAGToDAGISel::isCPLoad(const LoadSDNode *N) const {
Tom Stellard75aadc22012-12-11 21:25:42 +0000618 MachineMemOperand *MMO = N->getMemOperand();
Nick Lewyckyaad475b2014-04-15 07:22:52 +0000619 if (checkPrivateAddress(N->getMemOperand())) {
Tom Stellard75aadc22012-12-11 21:25:42 +0000620 if (MMO) {
Nick Lewyckyaad475b2014-04-15 07:22:52 +0000621 const PseudoSourceValue *PSV = MMO->getPseudoValue();
Alex Lorenze40c8a22015-08-11 23:09:45 +0000622 if (PSV && PSV->isConstantPool()) {
Tom Stellard75aadc22012-12-11 21:25:42 +0000623 return true;
624 }
625 }
626 }
627 return false;
628}
629
Matt Arsenault2aabb062013-06-18 23:37:58 +0000630bool AMDGPUDAGToDAGISel::isPrivateLoad(const LoadSDNode *N) const {
Nick Lewyckyaad475b2014-04-15 07:22:52 +0000631 if (checkPrivateAddress(N->getMemOperand())) {
Tom Stellard75aadc22012-12-11 21:25:42 +0000632 // Check to make sure we are not a constant pool load or a constant load
633 // that is marked as a private load
634 if (isCPLoad(N) || isConstantLoad(N, -1)) {
635 return false;
636 }
637 }
Matt Arsenault209a7b92014-04-18 07:40:20 +0000638
639 const Value *MemVal = N->getMemOperand()->getValue();
Matt Arsenault8226fc42016-03-02 23:00:21 +0000640 return !checkType(MemVal, AMDGPUAS::LOCAL_ADDRESS) &&
641 !checkType(MemVal, AMDGPUAS::GLOBAL_ADDRESS) &&
642 !checkType(MemVal, AMDGPUAS::FLAT_ADDRESS) &&
643 !checkType(MemVal, AMDGPUAS::REGION_ADDRESS) &&
644 !checkType(MemVal, AMDGPUAS::CONSTANT_ADDRESS) &&
645 !checkType(MemVal, AMDGPUAS::PARAM_D_ADDRESS) &&
646 !checkType(MemVal, AMDGPUAS::PARAM_I_ADDRESS);
Tom Stellard75aadc22012-12-11 21:25:42 +0000647}
648
Tom Stellardbc4497b2016-02-12 23:45:29 +0000649bool AMDGPUDAGToDAGISel::isUniformBr(const SDNode *N) const {
650 const BasicBlock *BB = FuncInfo->MBB->getBasicBlock();
651 return BB->getTerminator()->getMetadata("amdgpu.uniform");
652}
653
Tom Stellard75aadc22012-12-11 21:25:42 +0000654const char *AMDGPUDAGToDAGISel::getPassName() const {
655 return "AMDGPU DAG->DAG Pattern Instruction Selection";
656}
657
Tom Stellard41fc7852013-07-23 01:48:42 +0000658//===----------------------------------------------------------------------===//
659// Complex Patterns
660//===----------------------------------------------------------------------===//
Tom Stellard75aadc22012-12-11 21:25:42 +0000661
Tom Stellard365366f2013-01-23 02:09:06 +0000662bool AMDGPUDAGToDAGISel::SelectGlobalValueConstantOffset(SDValue Addr,
Matt Arsenault209a7b92014-04-18 07:40:20 +0000663 SDValue& IntPtr) {
Tom Stellard365366f2013-01-23 02:09:06 +0000664 if (ConstantSDNode *Cst = dyn_cast<ConstantSDNode>(Addr)) {
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +0000665 IntPtr = CurDAG->getIntPtrConstant(Cst->getZExtValue() / 4, SDLoc(Addr),
666 true);
Tom Stellard365366f2013-01-23 02:09:06 +0000667 return true;
668 }
669 return false;
670}
671
672bool AMDGPUDAGToDAGISel::SelectGlobalValueVariableOffset(SDValue Addr,
673 SDValue& BaseReg, SDValue &Offset) {
Matt Arsenault209a7b92014-04-18 07:40:20 +0000674 if (!isa<ConstantSDNode>(Addr)) {
Tom Stellard365366f2013-01-23 02:09:06 +0000675 BaseReg = Addr;
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +0000676 Offset = CurDAG->getIntPtrConstant(0, SDLoc(Addr), true);
Tom Stellard365366f2013-01-23 02:09:06 +0000677 return true;
678 }
679 return false;
680}
681
Tom Stellard75aadc22012-12-11 21:25:42 +0000682bool AMDGPUDAGToDAGISel::SelectADDRVTX_READ(SDValue Addr, SDValue &Base,
683 SDValue &Offset) {
Matt Arsenault209a7b92014-04-18 07:40:20 +0000684 ConstantSDNode *IMMOffset;
Tom Stellard75aadc22012-12-11 21:25:42 +0000685
686 if (Addr.getOpcode() == ISD::ADD
687 && (IMMOffset = dyn_cast<ConstantSDNode>(Addr.getOperand(1)))
688 && isInt<16>(IMMOffset->getZExtValue())) {
689
690 Base = Addr.getOperand(0);
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +0000691 Offset = CurDAG->getTargetConstant(IMMOffset->getZExtValue(), SDLoc(Addr),
692 MVT::i32);
Tom Stellard75aadc22012-12-11 21:25:42 +0000693 return true;
694 // If the pointer address is constant, we can move it to the offset field.
695 } else if ((IMMOffset = dyn_cast<ConstantSDNode>(Addr))
696 && isInt<16>(IMMOffset->getZExtValue())) {
697 Base = CurDAG->getCopyFromReg(CurDAG->getEntryNode(),
Andrew Trickef9de2a2013-05-25 02:42:55 +0000698 SDLoc(CurDAG->getEntryNode()),
Tom Stellard75aadc22012-12-11 21:25:42 +0000699 AMDGPU::ZERO, MVT::i32);
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +0000700 Offset = CurDAG->getTargetConstant(IMMOffset->getZExtValue(), SDLoc(Addr),
701 MVT::i32);
Tom Stellard75aadc22012-12-11 21:25:42 +0000702 return true;
703 }
704
705 // Default case, no offset
706 Base = Addr;
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +0000707 Offset = CurDAG->getTargetConstant(0, SDLoc(Addr), MVT::i32);
Tom Stellard75aadc22012-12-11 21:25:42 +0000708 return true;
709}
710
Tom Stellardf3b2a1e2013-02-06 17:32:29 +0000711bool AMDGPUDAGToDAGISel::SelectADDRIndirect(SDValue Addr, SDValue &Base,
712 SDValue &Offset) {
713 ConstantSDNode *C;
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +0000714 SDLoc DL(Addr);
Tom Stellardf3b2a1e2013-02-06 17:32:29 +0000715
716 if ((C = dyn_cast<ConstantSDNode>(Addr))) {
717 Base = CurDAG->getRegister(AMDGPU::INDIRECT_BASE_ADDR, MVT::i32);
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +0000718 Offset = CurDAG->getTargetConstant(C->getZExtValue(), DL, MVT::i32);
Tom Stellardf3b2a1e2013-02-06 17:32:29 +0000719 } else if ((Addr.getOpcode() == ISD::ADD || Addr.getOpcode() == ISD::OR) &&
720 (C = dyn_cast<ConstantSDNode>(Addr.getOperand(1)))) {
721 Base = Addr.getOperand(0);
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +0000722 Offset = CurDAG->getTargetConstant(C->getZExtValue(), DL, MVT::i32);
Tom Stellardf3b2a1e2013-02-06 17:32:29 +0000723 } else {
724 Base = Addr;
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +0000725 Offset = CurDAG->getTargetConstant(0, DL, MVT::i32);
Tom Stellardf3b2a1e2013-02-06 17:32:29 +0000726 }
727
728 return true;
729}
Christian Konigd910b7d2013-02-26 17:52:16 +0000730
Matt Arsenaultb8b51532014-06-23 18:00:38 +0000731SDNode *AMDGPUDAGToDAGISel::SelectADD_SUB_I64(SDNode *N) {
Matt Arsenault9fa3f932014-06-23 18:00:34 +0000732 SDLoc DL(N);
733 SDValue LHS = N->getOperand(0);
734 SDValue RHS = N->getOperand(1);
735
Matt Arsenaultb8b51532014-06-23 18:00:38 +0000736 bool IsAdd = (N->getOpcode() == ISD::ADD);
737
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +0000738 SDValue Sub0 = CurDAG->getTargetConstant(AMDGPU::sub0, DL, MVT::i32);
739 SDValue Sub1 = CurDAG->getTargetConstant(AMDGPU::sub1, DL, MVT::i32);
Matt Arsenault9fa3f932014-06-23 18:00:34 +0000740
741 SDNode *Lo0 = CurDAG->getMachineNode(TargetOpcode::EXTRACT_SUBREG,
742 DL, MVT::i32, LHS, Sub0);
743 SDNode *Hi0 = CurDAG->getMachineNode(TargetOpcode::EXTRACT_SUBREG,
744 DL, MVT::i32, LHS, Sub1);
745
746 SDNode *Lo1 = CurDAG->getMachineNode(TargetOpcode::EXTRACT_SUBREG,
747 DL, MVT::i32, RHS, Sub0);
748 SDNode *Hi1 = CurDAG->getMachineNode(TargetOpcode::EXTRACT_SUBREG,
749 DL, MVT::i32, RHS, Sub1);
750
751 SDVTList VTList = CurDAG->getVTList(MVT::i32, MVT::Glue);
Matt Arsenault9fa3f932014-06-23 18:00:34 +0000752 SDValue AddLoArgs[] = { SDValue(Lo0, 0), SDValue(Lo1, 0) };
753
Matt Arsenaultb8b51532014-06-23 18:00:38 +0000754
Tom Stellard80942a12014-09-05 14:07:59 +0000755 unsigned Opc = IsAdd ? AMDGPU::S_ADD_U32 : AMDGPU::S_SUB_U32;
Matt Arsenaultb8b51532014-06-23 18:00:38 +0000756 unsigned CarryOpc = IsAdd ? AMDGPU::S_ADDC_U32 : AMDGPU::S_SUBB_U32;
757
Matt Arsenaultb8b51532014-06-23 18:00:38 +0000758 SDNode *AddLo = CurDAG->getMachineNode( Opc, DL, VTList, AddLoArgs);
759 SDValue Carry(AddLo, 1);
760 SDNode *AddHi
761 = CurDAG->getMachineNode(CarryOpc, DL, MVT::i32,
762 SDValue(Hi0, 0), SDValue(Hi1, 0), Carry);
Matt Arsenault9fa3f932014-06-23 18:00:34 +0000763
764 SDValue Args[5] = {
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +0000765 CurDAG->getTargetConstant(AMDGPU::SReg_64RegClassID, DL, MVT::i32),
Matt Arsenault9fa3f932014-06-23 18:00:34 +0000766 SDValue(AddLo,0),
767 Sub0,
768 SDValue(AddHi,0),
769 Sub1,
770 };
771 return CurDAG->SelectNodeTo(N, AMDGPU::REG_SEQUENCE, MVT::i64, Args);
772}
773
Matt Arsenault044f1d12015-02-14 04:24:28 +0000774// We need to handle this here because tablegen doesn't support matching
775// instructions with multiple outputs.
Matt Arsenaultf2b0aeb2014-06-23 18:28:28 +0000776SDNode *AMDGPUDAGToDAGISel::SelectDIV_SCALE(SDNode *N) {
777 SDLoc SL(N);
778 EVT VT = N->getValueType(0);
779
780 assert(VT == MVT::f32 || VT == MVT::f64);
781
782 unsigned Opc
783 = (VT == MVT::f64) ? AMDGPU::V_DIV_SCALE_F64 : AMDGPU::V_DIV_SCALE_F32;
784
NAKAMURA Takumi84965032015-09-22 11:14:12 +0000785 // src0_modifiers, src0, src1_modifiers, src1, src2_modifiers, src2, clamp,
786 // omod
Matt Arsenault044f1d12015-02-14 04:24:28 +0000787 SDValue Ops[8];
Matt Arsenaultf2b0aeb2014-06-23 18:28:28 +0000788
Matt Arsenault044f1d12015-02-14 04:24:28 +0000789 SelectVOP3Mods0(N->getOperand(0), Ops[1], Ops[0], Ops[6], Ops[7]);
790 SelectVOP3Mods(N->getOperand(1), Ops[3], Ops[2]);
791 SelectVOP3Mods(N->getOperand(2), Ops[5], Ops[4]);
Matt Arsenaultf2b0aeb2014-06-23 18:28:28 +0000792 return CurDAG->SelectNodeTo(N, Opc, VT, MVT::i1, Ops);
793}
794
Tom Stellard85e8b6d2014-08-22 18:49:33 +0000795bool AMDGPUDAGToDAGISel::isDSOffsetLegal(const SDValue &Base, unsigned Offset,
796 unsigned OffsetBits) const {
Tom Stellard85e8b6d2014-08-22 18:49:33 +0000797 if ((OffsetBits == 16 && !isUInt<16>(Offset)) ||
798 (OffsetBits == 8 && !isUInt<8>(Offset)))
799 return false;
800
Matt Arsenault706f9302015-07-06 16:01:58 +0000801 if (Subtarget->getGeneration() >= AMDGPUSubtarget::SEA_ISLANDS ||
802 Subtarget->unsafeDSOffsetFoldingEnabled())
Tom Stellard85e8b6d2014-08-22 18:49:33 +0000803 return true;
804
805 // On Southern Islands instruction with a negative base value and an offset
806 // don't seem to work.
807 return CurDAG->SignBitIsZero(Base);
808}
809
810bool AMDGPUDAGToDAGISel::SelectDS1Addr1Offset(SDValue Addr, SDValue &Base,
811 SDValue &Offset) const {
812 if (CurDAG->isBaseWithConstantOffset(Addr)) {
813 SDValue N0 = Addr.getOperand(0);
814 SDValue N1 = Addr.getOperand(1);
815 ConstantSDNode *C1 = cast<ConstantSDNode>(N1);
816 if (isDSOffsetLegal(N0, C1->getSExtValue(), 16)) {
817 // (add n0, c0)
818 Base = N0;
819 Offset = N1;
820 return true;
821 }
Matt Arsenault966a94f2015-09-08 19:34:22 +0000822 } else if (Addr.getOpcode() == ISD::SUB) {
823 // sub C, x -> add (sub 0, x), C
824 if (const ConstantSDNode *C = dyn_cast<ConstantSDNode>(Addr.getOperand(0))) {
825 int64_t ByteOffset = C->getSExtValue();
826 if (isUInt<16>(ByteOffset)) {
827 SDLoc DL(Addr);
828 SDValue Zero = CurDAG->getTargetConstant(0, DL, MVT::i32);
Tom Stellard85e8b6d2014-08-22 18:49:33 +0000829
Matt Arsenault966a94f2015-09-08 19:34:22 +0000830 // XXX - This is kind of hacky. Create a dummy sub node so we can check
831 // the known bits in isDSOffsetLegal. We need to emit the selected node
832 // here, so this is thrown away.
833 SDValue Sub = CurDAG->getNode(ISD::SUB, DL, MVT::i32,
834 Zero, Addr.getOperand(1));
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +0000835
Matt Arsenault966a94f2015-09-08 19:34:22 +0000836 if (isDSOffsetLegal(Sub, ByteOffset, 16)) {
837 MachineSDNode *MachineSub
838 = CurDAG->getMachineNode(AMDGPU::V_SUB_I32_e32, DL, MVT::i32,
839 Zero, Addr.getOperand(1));
840
841 Base = SDValue(MachineSub, 0);
842 Offset = Addr.getOperand(0);
843 return true;
844 }
845 }
846 }
847 } else if (const ConstantSDNode *CAddr = dyn_cast<ConstantSDNode>(Addr)) {
848 // If we have a constant address, prefer to put the constant into the
849 // offset. This can save moves to load the constant address since multiple
850 // operations can share the zero base address register, and enables merging
851 // into read2 / write2 instructions.
852
853 SDLoc DL(Addr);
854
Matt Arsenaulte775f5f2014-10-14 17:21:19 +0000855 if (isUInt<16>(CAddr->getZExtValue())) {
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +0000856 SDValue Zero = CurDAG->getTargetConstant(0, DL, MVT::i32);
Tom Stellardc8d79202014-10-15 21:08:59 +0000857 MachineSDNode *MovZero = CurDAG->getMachineNode(AMDGPU::V_MOV_B32_e32,
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +0000858 DL, MVT::i32, Zero);
Tom Stellardc8d79202014-10-15 21:08:59 +0000859 Base = SDValue(MovZero, 0);
Matt Arsenaulte775f5f2014-10-14 17:21:19 +0000860 Offset = Addr;
861 return true;
862 }
863 }
864
Tom Stellard85e8b6d2014-08-22 18:49:33 +0000865 // default case
866 Base = Addr;
Matt Arsenault966a94f2015-09-08 19:34:22 +0000867 Offset = CurDAG->getTargetConstant(0, SDLoc(Addr), MVT::i16);
Tom Stellard85e8b6d2014-08-22 18:49:33 +0000868 return true;
869}
870
Matt Arsenault966a94f2015-09-08 19:34:22 +0000871// TODO: If offset is too big, put low 16-bit into offset.
Tom Stellardf3fc5552014-08-22 18:49:35 +0000872bool AMDGPUDAGToDAGISel::SelectDS64Bit4ByteAligned(SDValue Addr, SDValue &Base,
873 SDValue &Offset0,
874 SDValue &Offset1) const {
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +0000875 SDLoc DL(Addr);
876
Tom Stellardf3fc5552014-08-22 18:49:35 +0000877 if (CurDAG->isBaseWithConstantOffset(Addr)) {
878 SDValue N0 = Addr.getOperand(0);
879 SDValue N1 = Addr.getOperand(1);
880 ConstantSDNode *C1 = cast<ConstantSDNode>(N1);
881 unsigned DWordOffset0 = C1->getZExtValue() / 4;
882 unsigned DWordOffset1 = DWordOffset0 + 1;
883 // (add n0, c0)
884 if (isDSOffsetLegal(N0, DWordOffset1, 8)) {
885 Base = N0;
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +0000886 Offset0 = CurDAG->getTargetConstant(DWordOffset0, DL, MVT::i8);
887 Offset1 = CurDAG->getTargetConstant(DWordOffset1, DL, MVT::i8);
Tom Stellardf3fc5552014-08-22 18:49:35 +0000888 return true;
889 }
Matt Arsenault966a94f2015-09-08 19:34:22 +0000890 } else if (Addr.getOpcode() == ISD::SUB) {
891 // sub C, x -> add (sub 0, x), C
892 if (const ConstantSDNode *C = dyn_cast<ConstantSDNode>(Addr.getOperand(0))) {
893 unsigned DWordOffset0 = C->getZExtValue() / 4;
894 unsigned DWordOffset1 = DWordOffset0 + 1;
Tom Stellardf3fc5552014-08-22 18:49:35 +0000895
Matt Arsenault966a94f2015-09-08 19:34:22 +0000896 if (isUInt<8>(DWordOffset0)) {
897 SDLoc DL(Addr);
898 SDValue Zero = CurDAG->getTargetConstant(0, DL, MVT::i32);
899
900 // XXX - This is kind of hacky. Create a dummy sub node so we can check
901 // the known bits in isDSOffsetLegal. We need to emit the selected node
902 // here, so this is thrown away.
903 SDValue Sub = CurDAG->getNode(ISD::SUB, DL, MVT::i32,
904 Zero, Addr.getOperand(1));
905
906 if (isDSOffsetLegal(Sub, DWordOffset1, 8)) {
907 MachineSDNode *MachineSub
908 = CurDAG->getMachineNode(AMDGPU::V_SUB_I32_e32, DL, MVT::i32,
909 Zero, Addr.getOperand(1));
910
911 Base = SDValue(MachineSub, 0);
912 Offset0 = CurDAG->getTargetConstant(DWordOffset0, DL, MVT::i8);
913 Offset1 = CurDAG->getTargetConstant(DWordOffset1, DL, MVT::i8);
914 return true;
915 }
916 }
917 }
918 } else if (const ConstantSDNode *CAddr = dyn_cast<ConstantSDNode>(Addr)) {
Matt Arsenault1a74aff2014-10-15 18:06:43 +0000919 unsigned DWordOffset0 = CAddr->getZExtValue() / 4;
920 unsigned DWordOffset1 = DWordOffset0 + 1;
921 assert(4 * DWordOffset0 == CAddr->getZExtValue());
922
923 if (isUInt<8>(DWordOffset0) && isUInt<8>(DWordOffset1)) {
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +0000924 SDValue Zero = CurDAG->getTargetConstant(0, DL, MVT::i32);
Matt Arsenault1a74aff2014-10-15 18:06:43 +0000925 MachineSDNode *MovZero
926 = CurDAG->getMachineNode(AMDGPU::V_MOV_B32_e32,
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +0000927 DL, MVT::i32, Zero);
Matt Arsenault1a74aff2014-10-15 18:06:43 +0000928 Base = SDValue(MovZero, 0);
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +0000929 Offset0 = CurDAG->getTargetConstant(DWordOffset0, DL, MVT::i8);
930 Offset1 = CurDAG->getTargetConstant(DWordOffset1, DL, MVT::i8);
Matt Arsenault1a74aff2014-10-15 18:06:43 +0000931 return true;
932 }
933 }
934
Tom Stellardf3fc5552014-08-22 18:49:35 +0000935 // default case
936 Base = Addr;
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +0000937 Offset0 = CurDAG->getTargetConstant(0, DL, MVT::i8);
938 Offset1 = CurDAG->getTargetConstant(1, DL, MVT::i8);
Tom Stellardf3fc5552014-08-22 18:49:35 +0000939 return true;
940}
941
Tom Stellardb02094e2014-07-21 15:45:01 +0000942static bool isLegalMUBUFImmOffset(const ConstantSDNode *Imm) {
943 return isUInt<12>(Imm->getZExtValue());
944}
945
Changpeng Fangb41574a2015-12-22 20:55:23 +0000946bool AMDGPUDAGToDAGISel::SelectMUBUF(SDValue Addr, SDValue &Ptr,
Tom Stellard155bbb72014-08-11 22:18:17 +0000947 SDValue &VAddr, SDValue &SOffset,
948 SDValue &Offset, SDValue &Offen,
949 SDValue &Idxen, SDValue &Addr64,
950 SDValue &GLC, SDValue &SLC,
951 SDValue &TFE) const {
Changpeng Fangb41574a2015-12-22 20:55:23 +0000952 // Subtarget prefers to use flat instruction
953 if (Subtarget->useFlatForGlobal())
954 return false;
955
Tom Stellardb02c2682014-06-24 23:33:07 +0000956 SDLoc DL(Addr);
957
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +0000958 GLC = CurDAG->getTargetConstant(0, DL, MVT::i1);
959 SLC = CurDAG->getTargetConstant(0, DL, MVT::i1);
960 TFE = CurDAG->getTargetConstant(0, DL, MVT::i1);
Tom Stellard155bbb72014-08-11 22:18:17 +0000961
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +0000962 Idxen = CurDAG->getTargetConstant(0, DL, MVT::i1);
963 Offen = CurDAG->getTargetConstant(0, DL, MVT::i1);
964 Addr64 = CurDAG->getTargetConstant(0, DL, MVT::i1);
965 SOffset = CurDAG->getTargetConstant(0, DL, MVT::i32);
Tom Stellard155bbb72014-08-11 22:18:17 +0000966
Tom Stellardb02c2682014-06-24 23:33:07 +0000967 if (CurDAG->isBaseWithConstantOffset(Addr)) {
968 SDValue N0 = Addr.getOperand(0);
969 SDValue N1 = Addr.getOperand(1);
970 ConstantSDNode *C1 = cast<ConstantSDNode>(N1);
971
Tom Stellard94b72312015-02-11 00:34:35 +0000972 if (N0.getOpcode() == ISD::ADD) {
973 // (add (add N2, N3), C1) -> addr64
974 SDValue N2 = N0.getOperand(0);
975 SDValue N3 = N0.getOperand(1);
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +0000976 Addr64 = CurDAG->getTargetConstant(1, DL, MVT::i1);
Tom Stellard94b72312015-02-11 00:34:35 +0000977 Ptr = N2;
978 VAddr = N3;
979 } else {
Tom Stellardb02c2682014-06-24 23:33:07 +0000980
Tom Stellard155bbb72014-08-11 22:18:17 +0000981 // (add N0, C1) -> offset
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +0000982 VAddr = CurDAG->getTargetConstant(0, DL, MVT::i32);
Tom Stellard155bbb72014-08-11 22:18:17 +0000983 Ptr = N0;
Tom Stellard94b72312015-02-11 00:34:35 +0000984 }
985
986 if (isLegalMUBUFImmOffset(C1)) {
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +0000987 Offset = CurDAG->getTargetConstant(C1->getZExtValue(), DL, MVT::i16);
Changpeng Fangb41574a2015-12-22 20:55:23 +0000988 return true;
Tom Stellard94b72312015-02-11 00:34:35 +0000989 } else if (isUInt<32>(C1->getZExtValue())) {
990 // Illegal offset, store it in soffset.
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +0000991 Offset = CurDAG->getTargetConstant(0, DL, MVT::i16);
Tom Stellard94b72312015-02-11 00:34:35 +0000992 SOffset = SDValue(CurDAG->getMachineNode(AMDGPU::S_MOV_B32, DL, MVT::i32,
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +0000993 CurDAG->getTargetConstant(C1->getZExtValue(), DL, MVT::i32)),
994 0);
Changpeng Fangb41574a2015-12-22 20:55:23 +0000995 return true;
Tom Stellardb02c2682014-06-24 23:33:07 +0000996 }
997 }
Tom Stellard94b72312015-02-11 00:34:35 +0000998
Tom Stellardb02c2682014-06-24 23:33:07 +0000999 if (Addr.getOpcode() == ISD::ADD) {
Tom Stellard155bbb72014-08-11 22:18:17 +00001000 // (add N0, N1) -> addr64
Tom Stellardb02c2682014-06-24 23:33:07 +00001001 SDValue N0 = Addr.getOperand(0);
1002 SDValue N1 = Addr.getOperand(1);
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +00001003 Addr64 = CurDAG->getTargetConstant(1, DL, MVT::i1);
Tom Stellard155bbb72014-08-11 22:18:17 +00001004 Ptr = N0;
1005 VAddr = N1;
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +00001006 Offset = CurDAG->getTargetConstant(0, DL, MVT::i16);
Changpeng Fangb41574a2015-12-22 20:55:23 +00001007 return true;
Tom Stellardb02c2682014-06-24 23:33:07 +00001008 }
1009
Tom Stellard155bbb72014-08-11 22:18:17 +00001010 // default case -> offset
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +00001011 VAddr = CurDAG->getTargetConstant(0, DL, MVT::i32);
Tom Stellard155bbb72014-08-11 22:18:17 +00001012 Ptr = Addr;
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +00001013 Offset = CurDAG->getTargetConstant(0, DL, MVT::i16);
Changpeng Fangb41574a2015-12-22 20:55:23 +00001014
1015 return true;
Tom Stellard155bbb72014-08-11 22:18:17 +00001016}
1017
1018bool AMDGPUDAGToDAGISel::SelectMUBUFAddr64(SDValue Addr, SDValue &SRsrc,
Tom Stellardc53861a2015-02-11 00:34:32 +00001019 SDValue &VAddr, SDValue &SOffset,
Tom Stellard1f9939f2015-02-27 14:59:41 +00001020 SDValue &Offset, SDValue &GLC,
1021 SDValue &SLC, SDValue &TFE) const {
1022 SDValue Ptr, Offen, Idxen, Addr64;
Tom Stellard155bbb72014-08-11 22:18:17 +00001023
Tom Stellard70580f82015-07-20 14:28:41 +00001024 // addr64 bit was removed for volcanic islands.
1025 if (Subtarget->getGeneration() >= AMDGPUSubtarget::VOLCANIC_ISLANDS)
1026 return false;
1027
Changpeng Fangb41574a2015-12-22 20:55:23 +00001028 if (!SelectMUBUF(Addr, Ptr, VAddr, SOffset, Offset, Offen, Idxen, Addr64,
1029 GLC, SLC, TFE))
1030 return false;
Tom Stellard155bbb72014-08-11 22:18:17 +00001031
1032 ConstantSDNode *C = cast<ConstantSDNode>(Addr64);
1033 if (C->getSExtValue()) {
1034 SDLoc DL(Addr);
Matt Arsenault485defe2014-11-05 19:01:17 +00001035
1036 const SITargetLowering& Lowering =
1037 *static_cast<const SITargetLowering*>(getTargetLowering());
1038
1039 SRsrc = SDValue(Lowering.wrapAddr64Rsrc(*CurDAG, DL, Ptr), 0);
Tom Stellard155bbb72014-08-11 22:18:17 +00001040 return true;
1041 }
Matt Arsenault485defe2014-11-05 19:01:17 +00001042
Tom Stellard155bbb72014-08-11 22:18:17 +00001043 return false;
1044}
1045
Tom Stellard7980fc82014-09-25 18:30:26 +00001046bool AMDGPUDAGToDAGISel::SelectMUBUFAddr64(SDValue Addr, SDValue &SRsrc,
Tom Stellardc53861a2015-02-11 00:34:32 +00001047 SDValue &VAddr, SDValue &SOffset,
NAKAMURA Takumi0a7d0ad2015-09-22 11:15:07 +00001048 SDValue &Offset,
1049 SDValue &SLC) const {
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +00001050 SLC = CurDAG->getTargetConstant(0, SDLoc(Addr), MVT::i1);
Tom Stellard1f9939f2015-02-27 14:59:41 +00001051 SDValue GLC, TFE;
Tom Stellard7980fc82014-09-25 18:30:26 +00001052
Tom Stellard1f9939f2015-02-27 14:59:41 +00001053 return SelectMUBUFAddr64(Addr, SRsrc, VAddr, SOffset, Offset, GLC, SLC, TFE);
Tom Stellard7980fc82014-09-25 18:30:26 +00001054}
1055
Tom Stellardb02094e2014-07-21 15:45:01 +00001056bool AMDGPUDAGToDAGISel::SelectMUBUFScratch(SDValue Addr, SDValue &Rsrc,
1057 SDValue &VAddr, SDValue &SOffset,
1058 SDValue &ImmOffset) const {
1059
1060 SDLoc DL(Addr);
1061 MachineFunction &MF = CurDAG->getMachineFunction();
Matt Arsenault0e3d3892015-11-30 21:15:53 +00001062 const SIMachineFunctionInfo *Info = MF.getInfo<SIMachineFunctionInfo>();
Tom Stellardb02094e2014-07-21 15:45:01 +00001063
Matt Arsenault0e3d3892015-11-30 21:15:53 +00001064 Rsrc = CurDAG->getRegister(Info->getScratchRSrcReg(), MVT::v4i32);
Matt Arsenault26f8f3d2015-11-30 21:16:03 +00001065 SOffset = CurDAG->getRegister(Info->getScratchWaveOffsetReg(), MVT::i32);
Tom Stellardb02094e2014-07-21 15:45:01 +00001066
1067 // (add n0, c1)
1068 if (CurDAG->isBaseWithConstantOffset(Addr)) {
Tom Stellard78655fc2015-07-16 19:40:09 +00001069 SDValue N0 = Addr.getOperand(0);
Tom Stellardb02094e2014-07-21 15:45:01 +00001070 SDValue N1 = Addr.getOperand(1);
Matt Arsenaultcd099612016-02-24 04:55:29 +00001071
Tom Stellard78655fc2015-07-16 19:40:09 +00001072 // Offsets in vaddr must be positive.
Matt Arsenaultcd099612016-02-24 04:55:29 +00001073 ConstantSDNode *C1 = cast<ConstantSDNode>(N1);
Matt Arsenaultcb38a6b2016-03-21 18:02:18 +00001074 if (isLegalMUBUFImmOffset(C1)) {
Matt Arsenaultcd099612016-02-24 04:55:29 +00001075 VAddr = N0;
1076 ImmOffset = CurDAG->getTargetConstant(C1->getZExtValue(), DL, MVT::i16);
1077 return true;
Tom Stellardb02094e2014-07-21 15:45:01 +00001078 }
1079 }
1080
Tom Stellardb02094e2014-07-21 15:45:01 +00001081 // (node)
1082 VAddr = Addr;
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +00001083 ImmOffset = CurDAG->getTargetConstant(0, DL, MVT::i16);
Tom Stellardb02094e2014-07-21 15:45:01 +00001084 return true;
1085}
1086
Tom Stellard155bbb72014-08-11 22:18:17 +00001087bool AMDGPUDAGToDAGISel::SelectMUBUFOffset(SDValue Addr, SDValue &SRsrc,
1088 SDValue &SOffset, SDValue &Offset,
1089 SDValue &GLC, SDValue &SLC,
1090 SDValue &TFE) const {
1091 SDValue Ptr, VAddr, Offen, Idxen, Addr64;
Tom Stellard794c8c02014-12-02 17:05:41 +00001092 const SIInstrInfo *TII =
Eric Christopher7792e322015-01-30 23:24:40 +00001093 static_cast<const SIInstrInfo *>(Subtarget->getInstrInfo());
Tom Stellardb02094e2014-07-21 15:45:01 +00001094
Changpeng Fangb41574a2015-12-22 20:55:23 +00001095 if (!SelectMUBUF(Addr, Ptr, VAddr, SOffset, Offset, Offen, Idxen, Addr64,
1096 GLC, SLC, TFE))
1097 return false;
Tom Stellardb02094e2014-07-21 15:45:01 +00001098
Tom Stellard155bbb72014-08-11 22:18:17 +00001099 if (!cast<ConstantSDNode>(Offen)->getSExtValue() &&
1100 !cast<ConstantSDNode>(Idxen)->getSExtValue() &&
1101 !cast<ConstantSDNode>(Addr64)->getSExtValue()) {
Tom Stellard794c8c02014-12-02 17:05:41 +00001102 uint64_t Rsrc = TII->getDefaultRsrcDataFormat() |
Tom Stellard155bbb72014-08-11 22:18:17 +00001103 APInt::getAllOnesValue(32).getZExtValue(); // Size
1104 SDLoc DL(Addr);
Matt Arsenaultf3cd4512014-11-05 19:01:19 +00001105
1106 const SITargetLowering& Lowering =
1107 *static_cast<const SITargetLowering*>(getTargetLowering());
1108
1109 SRsrc = SDValue(Lowering.buildRSRC(*CurDAG, DL, Ptr, 0, Rsrc), 0);
Tom Stellard155bbb72014-08-11 22:18:17 +00001110 return true;
1111 }
1112 return false;
Tom Stellardb02094e2014-07-21 15:45:01 +00001113}
1114
Tom Stellard7980fc82014-09-25 18:30:26 +00001115bool AMDGPUDAGToDAGISel::SelectMUBUFOffset(SDValue Addr, SDValue &SRsrc,
1116 SDValue &Soffset, SDValue &Offset,
1117 SDValue &GLC) const {
1118 SDValue SLC, TFE;
1119
1120 return SelectMUBUFOffset(Addr, SRsrc, Soffset, Offset, GLC, SLC, TFE);
1121}
1122
Nicolai Haehnle3003ba02016-03-18 16:24:20 +00001123void AMDGPUDAGToDAGISel::SelectMUBUFConstant(SDValue Constant,
1124 SDValue &SOffset,
1125 SDValue &ImmOffset) const {
1126 SDLoc DL(Constant);
1127 uint32_t Imm = cast<ConstantSDNode>(Constant)->getZExtValue();
1128 uint32_t Overflow = 0;
1129
1130 if (Imm >= 4096) {
1131 if (Imm <= 4095 + 64) {
1132 // Use an SOffset inline constant for 1..64
1133 Overflow = Imm - 4095;
1134 Imm = 4095;
1135 } else {
1136 // Try to keep the same value in SOffset for adjacent loads, so that
1137 // the corresponding register contents can be re-used.
1138 //
1139 // Load values with all low-bits set into SOffset, so that a larger
1140 // range of values can be covered using s_movk_i32
1141 uint32_t High = (Imm + 1) & ~4095;
1142 uint32_t Low = (Imm + 1) & 4095;
1143 Imm = Low;
1144 Overflow = High - 1;
1145 }
1146 }
1147
1148 ImmOffset = CurDAG->getTargetConstant(Imm, DL, MVT::i16);
1149
1150 if (Overflow <= 64)
1151 SOffset = CurDAG->getTargetConstant(Overflow, DL, MVT::i32);
1152 else
1153 SOffset = SDValue(CurDAG->getMachineNode(AMDGPU::S_MOV_B32, DL, MVT::i32,
1154 CurDAG->getTargetConstant(Overflow, DL, MVT::i32)),
1155 0);
1156}
1157
1158bool AMDGPUDAGToDAGISel::SelectMUBUFIntrinsicOffset(SDValue Offset,
1159 SDValue &SOffset,
1160 SDValue &ImmOffset) const {
1161 SDLoc DL(Offset);
1162
1163 if (!isa<ConstantSDNode>(Offset))
1164 return false;
1165
1166 SelectMUBUFConstant(Offset, SOffset, ImmOffset);
1167
1168 return true;
1169}
1170
1171bool AMDGPUDAGToDAGISel::SelectMUBUFIntrinsicVOffset(SDValue Offset,
1172 SDValue &SOffset,
1173 SDValue &ImmOffset,
1174 SDValue &VOffset) const {
1175 SDLoc DL(Offset);
1176
1177 // Don't generate an unnecessary voffset for constant offsets.
1178 if (isa<ConstantSDNode>(Offset))
1179 return false;
1180
1181 if (CurDAG->isBaseWithConstantOffset(Offset)) {
1182 SDValue N0 = Offset.getOperand(0);
1183 SDValue N1 = Offset.getOperand(1);
1184 SelectMUBUFConstant(N1, SOffset, ImmOffset);
1185 VOffset = N0;
1186 } else {
1187 SOffset = CurDAG->getTargetConstant(0, DL, MVT::i32);
1188 ImmOffset = CurDAG->getTargetConstant(0, DL, MVT::i16);
1189 VOffset = Offset;
1190 }
1191
1192 return true;
1193}
1194
Tom Stellarddee26a22015-08-06 19:28:30 +00001195///
1196/// \param EncodedOffset This is the immediate value that will be encoded
1197/// directly into the instruction. On SI/CI the \p EncodedOffset
1198/// will be in units of dwords and on VI+ it will be units of bytes.
1199static bool isLegalSMRDImmOffset(const AMDGPUSubtarget *ST,
1200 int64_t EncodedOffset) {
1201 return ST->getGeneration() < AMDGPUSubtarget::VOLCANIC_ISLANDS ?
1202 isUInt<8>(EncodedOffset) : isUInt<20>(EncodedOffset);
1203}
1204
1205bool AMDGPUDAGToDAGISel::SelectSMRDOffset(SDValue ByteOffsetNode,
1206 SDValue &Offset, bool &Imm) const {
1207
1208 // FIXME: Handle non-constant offsets.
1209 ConstantSDNode *C = dyn_cast<ConstantSDNode>(ByteOffsetNode);
1210 if (!C)
1211 return false;
1212
1213 SDLoc SL(ByteOffsetNode);
1214 AMDGPUSubtarget::Generation Gen = Subtarget->getGeneration();
1215 int64_t ByteOffset = C->getSExtValue();
1216 int64_t EncodedOffset = Gen < AMDGPUSubtarget::VOLCANIC_ISLANDS ?
1217 ByteOffset >> 2 : ByteOffset;
1218
1219 if (isLegalSMRDImmOffset(Subtarget, EncodedOffset)) {
1220 Offset = CurDAG->getTargetConstant(EncodedOffset, SL, MVT::i32);
1221 Imm = true;
1222 return true;
1223 }
1224
Tom Stellard217361c2015-08-06 19:28:38 +00001225 if (!isUInt<32>(EncodedOffset) || !isUInt<32>(ByteOffset))
1226 return false;
1227
1228 if (Gen == AMDGPUSubtarget::SEA_ISLANDS && isUInt<32>(EncodedOffset)) {
1229 // 32-bit Immediates are supported on Sea Islands.
1230 Offset = CurDAG->getTargetConstant(EncodedOffset, SL, MVT::i32);
1231 } else {
Tom Stellarddee26a22015-08-06 19:28:30 +00001232 SDValue C32Bit = CurDAG->getTargetConstant(ByteOffset, SL, MVT::i32);
1233 Offset = SDValue(CurDAG->getMachineNode(AMDGPU::S_MOV_B32, SL, MVT::i32,
1234 C32Bit), 0);
Tom Stellarddee26a22015-08-06 19:28:30 +00001235 }
Tom Stellard217361c2015-08-06 19:28:38 +00001236 Imm = false;
1237 return true;
Tom Stellarddee26a22015-08-06 19:28:30 +00001238}
1239
1240bool AMDGPUDAGToDAGISel::SelectSMRD(SDValue Addr, SDValue &SBase,
1241 SDValue &Offset, bool &Imm) const {
1242
1243 SDLoc SL(Addr);
1244 if (CurDAG->isBaseWithConstantOffset(Addr)) {
1245 SDValue N0 = Addr.getOperand(0);
1246 SDValue N1 = Addr.getOperand(1);
1247
1248 if (SelectSMRDOffset(N1, Offset, Imm)) {
1249 SBase = N0;
1250 return true;
1251 }
1252 }
1253 SBase = Addr;
1254 Offset = CurDAG->getTargetConstant(0, SL, MVT::i32);
1255 Imm = true;
1256 return true;
1257}
1258
1259bool AMDGPUDAGToDAGISel::SelectSMRDImm(SDValue Addr, SDValue &SBase,
1260 SDValue &Offset) const {
1261 bool Imm;
1262 return SelectSMRD(Addr, SBase, Offset, Imm) && Imm;
1263}
1264
Tom Stellard217361c2015-08-06 19:28:38 +00001265bool AMDGPUDAGToDAGISel::SelectSMRDImm32(SDValue Addr, SDValue &SBase,
1266 SDValue &Offset) const {
1267
1268 if (Subtarget->getGeneration() != AMDGPUSubtarget::SEA_ISLANDS)
1269 return false;
1270
1271 bool Imm;
1272 if (!SelectSMRD(Addr, SBase, Offset, Imm))
1273 return false;
1274
1275 return !Imm && isa<ConstantSDNode>(Offset);
1276}
1277
Tom Stellarddee26a22015-08-06 19:28:30 +00001278bool AMDGPUDAGToDAGISel::SelectSMRDSgpr(SDValue Addr, SDValue &SBase,
1279 SDValue &Offset) const {
1280 bool Imm;
Tom Stellard217361c2015-08-06 19:28:38 +00001281 return SelectSMRD(Addr, SBase, Offset, Imm) && !Imm &&
1282 !isa<ConstantSDNode>(Offset);
Tom Stellarddee26a22015-08-06 19:28:30 +00001283}
1284
1285bool AMDGPUDAGToDAGISel::SelectSMRDBufferImm(SDValue Addr,
1286 SDValue &Offset) const {
1287 bool Imm;
1288 return SelectSMRDOffset(Addr, Offset, Imm) && Imm;
1289}
1290
Tom Stellard217361c2015-08-06 19:28:38 +00001291bool AMDGPUDAGToDAGISel::SelectSMRDBufferImm32(SDValue Addr,
1292 SDValue &Offset) const {
1293 if (Subtarget->getGeneration() != AMDGPUSubtarget::SEA_ISLANDS)
1294 return false;
1295
1296 bool Imm;
1297 if (!SelectSMRDOffset(Addr, Offset, Imm))
1298 return false;
1299
1300 return !Imm && isa<ConstantSDNode>(Offset);
1301}
1302
Tom Stellarddee26a22015-08-06 19:28:30 +00001303bool AMDGPUDAGToDAGISel::SelectSMRDBufferSgpr(SDValue Addr,
1304 SDValue &Offset) const {
1305 bool Imm;
Tom Stellard217361c2015-08-06 19:28:38 +00001306 return SelectSMRDOffset(Addr, Offset, Imm) && !Imm &&
1307 !isa<ConstantSDNode>(Offset);
Tom Stellarddee26a22015-08-06 19:28:30 +00001308}
1309
Matt Arsenault3f981402014-09-15 15:41:53 +00001310// FIXME: This is incorrect and only enough to be able to compile.
1311SDNode *AMDGPUDAGToDAGISel::SelectAddrSpaceCast(SDNode *N) {
1312 AddrSpaceCastSDNode *ASC = cast<AddrSpaceCastSDNode>(N);
1313 SDLoc DL(N);
1314
Matt Arsenault592d0682015-12-01 23:04:05 +00001315 const MachineFunction &MF = CurDAG->getMachineFunction();
Oliver Stannard7e7d9832016-02-02 13:52:43 +00001316 DiagnosticInfoUnsupported NotImplemented(
1317 *MF.getFunction(), "addrspacecast not implemented", DL.getDebugLoc());
Matt Arsenault592d0682015-12-01 23:04:05 +00001318 CurDAG->getContext()->diagnose(NotImplemented);
1319
Eric Christopher7792e322015-01-30 23:24:40 +00001320 assert(Subtarget->hasFlatAddressSpace() &&
Matt Arsenault3f981402014-09-15 15:41:53 +00001321 "addrspacecast only supported with flat address space!");
1322
Matt Arsenault3f981402014-09-15 15:41:53 +00001323 assert((ASC->getSrcAddressSpace() == AMDGPUAS::FLAT_ADDRESS ||
1324 ASC->getDestAddressSpace() == AMDGPUAS::FLAT_ADDRESS) &&
1325 "Can only cast to / from flat address space!");
1326
1327 // The flat instructions read the address as the index of the VGPR holding the
1328 // address, so casting should just be reinterpreting the base VGPR, so just
1329 // insert trunc / bitcast / zext.
1330
1331 SDValue Src = ASC->getOperand(0);
1332 EVT DestVT = ASC->getValueType(0);
1333 EVT SrcVT = Src.getValueType();
1334
1335 unsigned SrcSize = SrcVT.getSizeInBits();
1336 unsigned DestSize = DestVT.getSizeInBits();
1337
1338 if (SrcSize > DestSize) {
1339 assert(SrcSize == 64 && DestSize == 32);
1340 return CurDAG->getMachineNode(
1341 TargetOpcode::EXTRACT_SUBREG,
1342 DL,
1343 DestVT,
1344 Src,
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +00001345 CurDAG->getTargetConstant(AMDGPU::sub0, DL, MVT::i32));
Matt Arsenault3f981402014-09-15 15:41:53 +00001346 }
1347
Matt Arsenault3f981402014-09-15 15:41:53 +00001348 if (DestSize > SrcSize) {
1349 assert(SrcSize == 32 && DestSize == 64);
1350
Tom Stellardb6550522015-01-12 19:33:18 +00001351 // FIXME: This is probably wrong, we should never be defining
1352 // a register class with both VGPRs and SGPRs
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +00001353 SDValue RC = CurDAG->getTargetConstant(AMDGPU::VS_64RegClassID, DL,
1354 MVT::i32);
Matt Arsenault3f981402014-09-15 15:41:53 +00001355
1356 const SDValue Ops[] = {
1357 RC,
1358 Src,
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +00001359 CurDAG->getTargetConstant(AMDGPU::sub0, DL, MVT::i32),
1360 SDValue(CurDAG->getMachineNode(AMDGPU::S_MOV_B32, DL, MVT::i32,
1361 CurDAG->getConstant(0, DL, MVT::i32)), 0),
1362 CurDAG->getTargetConstant(AMDGPU::sub1, DL, MVT::i32)
Matt Arsenault3f981402014-09-15 15:41:53 +00001363 };
1364
1365 return CurDAG->getMachineNode(TargetOpcode::REG_SEQUENCE,
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +00001366 DL, N->getValueType(0), Ops);
Matt Arsenault3f981402014-09-15 15:41:53 +00001367 }
1368
1369 assert(SrcSize == 64 && DestSize == 64);
1370 return CurDAG->getNode(ISD::BITCAST, DL, DestVT, Src).getNode();
1371}
1372
Marek Olsak9b728682015-03-24 13:40:27 +00001373SDNode *AMDGPUDAGToDAGISel::getS_BFE(unsigned Opcode, SDLoc DL, SDValue Val,
1374 uint32_t Offset, uint32_t Width) {
1375 // Transformation function, pack the offset and width of a BFE into
1376 // the format expected by the S_BFE_I32 / S_BFE_U32. In the second
1377 // source, bits [5:0] contain the offset and bits [22:16] the width.
1378 uint32_t PackedVal = Offset | (Width << 16);
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +00001379 SDValue PackedConst = CurDAG->getTargetConstant(PackedVal, DL, MVT::i32);
Marek Olsak9b728682015-03-24 13:40:27 +00001380
1381 return CurDAG->getMachineNode(Opcode, DL, MVT::i32, Val, PackedConst);
1382}
1383
1384SDNode *AMDGPUDAGToDAGISel::SelectS_BFEFromShifts(SDNode *N) {
1385 // "(a << b) srl c)" ---> "BFE_U32 a, (c-b), (32-c)
1386 // "(a << b) sra c)" ---> "BFE_I32 a, (c-b), (32-c)
1387 // Predicate: 0 < b <= c < 32
1388
1389 const SDValue &Shl = N->getOperand(0);
1390 ConstantSDNode *B = dyn_cast<ConstantSDNode>(Shl->getOperand(1));
1391 ConstantSDNode *C = dyn_cast<ConstantSDNode>(N->getOperand(1));
1392
1393 if (B && C) {
1394 uint32_t BVal = B->getZExtValue();
1395 uint32_t CVal = C->getZExtValue();
1396
1397 if (0 < BVal && BVal <= CVal && CVal < 32) {
1398 bool Signed = N->getOpcode() == ISD::SRA;
1399 unsigned Opcode = Signed ? AMDGPU::S_BFE_I32 : AMDGPU::S_BFE_U32;
1400
1401 return getS_BFE(Opcode, SDLoc(N), Shl.getOperand(0),
1402 CVal - BVal, 32 - CVal);
1403 }
1404 }
1405 return SelectCode(N);
1406}
1407
1408SDNode *AMDGPUDAGToDAGISel::SelectS_BFE(SDNode *N) {
1409 switch (N->getOpcode()) {
1410 case ISD::AND:
1411 if (N->getOperand(0).getOpcode() == ISD::SRL) {
1412 // "(a srl b) & mask" ---> "BFE_U32 a, b, popcount(mask)"
1413 // Predicate: isMask(mask)
1414 const SDValue &Srl = N->getOperand(0);
1415 ConstantSDNode *Shift = dyn_cast<ConstantSDNode>(Srl.getOperand(1));
1416 ConstantSDNode *Mask = dyn_cast<ConstantSDNode>(N->getOperand(1));
1417
1418 if (Shift && Mask) {
1419 uint32_t ShiftVal = Shift->getZExtValue();
1420 uint32_t MaskVal = Mask->getZExtValue();
1421
1422 if (isMask_32(MaskVal)) {
1423 uint32_t WidthVal = countPopulation(MaskVal);
1424
1425 return getS_BFE(AMDGPU::S_BFE_U32, SDLoc(N), Srl.getOperand(0),
1426 ShiftVal, WidthVal);
1427 }
1428 }
1429 }
1430 break;
1431 case ISD::SRL:
1432 if (N->getOperand(0).getOpcode() == ISD::AND) {
1433 // "(a & mask) srl b)" ---> "BFE_U32 a, b, popcount(mask >> b)"
1434 // Predicate: isMask(mask >> b)
1435 const SDValue &And = N->getOperand(0);
1436 ConstantSDNode *Shift = dyn_cast<ConstantSDNode>(N->getOperand(1));
1437 ConstantSDNode *Mask = dyn_cast<ConstantSDNode>(And->getOperand(1));
1438
1439 if (Shift && Mask) {
1440 uint32_t ShiftVal = Shift->getZExtValue();
1441 uint32_t MaskVal = Mask->getZExtValue() >> ShiftVal;
1442
1443 if (isMask_32(MaskVal)) {
1444 uint32_t WidthVal = countPopulation(MaskVal);
1445
1446 return getS_BFE(AMDGPU::S_BFE_U32, SDLoc(N), And.getOperand(0),
1447 ShiftVal, WidthVal);
1448 }
1449 }
1450 } else if (N->getOperand(0).getOpcode() == ISD::SHL)
1451 return SelectS_BFEFromShifts(N);
1452 break;
1453 case ISD::SRA:
1454 if (N->getOperand(0).getOpcode() == ISD::SHL)
1455 return SelectS_BFEFromShifts(N);
1456 break;
1457 }
1458
1459 return SelectCode(N);
1460}
1461
Tom Stellardbc4497b2016-02-12 23:45:29 +00001462SDNode *AMDGPUDAGToDAGISel::SelectBRCOND(SDNode *N) {
1463 SDValue Cond = N->getOperand(1);
1464
1465 if (isCBranchSCC(N)) {
1466 // This brcond will use S_CBRANCH_SCC*, so let tablegen handle it.
1467 return SelectCode(N);
1468 }
1469
1470 // The result of VOPC instructions is or'd against ~EXEC before it is
1471 // written to vcc or another SGPR. This means that the value '1' is always
1472 // written to the corresponding bit for results that are masked. In order
1473 // to correctly check against vccz, we need to and VCC with the EXEC
1474 // register in order to clear the value from the masked bits.
1475
1476 SDLoc SL(N);
1477
1478 SDNode *MaskedCond =
1479 CurDAG->getMachineNode(AMDGPU::S_AND_B64, SL, MVT::i1,
1480 CurDAG->getRegister(AMDGPU::EXEC, MVT::i1),
1481 Cond);
1482 SDValue VCC = CurDAG->getCopyToReg(N->getOperand(0), SL, AMDGPU::VCC,
1483 SDValue(MaskedCond, 0),
1484 SDValue()); // Passing SDValue() adds a
1485 // glue output.
1486 return CurDAG->SelectNodeTo(N, AMDGPU::S_CBRANCH_VCCNZ, MVT::Other,
1487 N->getOperand(2), // Basic Block
1488 VCC.getValue(0), // Chain
1489 VCC.getValue(1)); // Glue
1490}
1491
Tom Stellardb4a313a2014-08-01 00:32:39 +00001492bool AMDGPUDAGToDAGISel::SelectVOP3Mods(SDValue In, SDValue &Src,
1493 SDValue &SrcMods) const {
1494
1495 unsigned Mods = 0;
1496
1497 Src = In;
1498
1499 if (Src.getOpcode() == ISD::FNEG) {
1500 Mods |= SISrcMods::NEG;
1501 Src = Src.getOperand(0);
1502 }
1503
1504 if (Src.getOpcode() == ISD::FABS) {
1505 Mods |= SISrcMods::ABS;
1506 Src = Src.getOperand(0);
1507 }
1508
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +00001509 SrcMods = CurDAG->getTargetConstant(Mods, SDLoc(In), MVT::i32);
Tom Stellardb4a313a2014-08-01 00:32:39 +00001510
1511 return true;
1512}
1513
Tom Stellarddb5a11f2015-07-13 15:47:57 +00001514bool AMDGPUDAGToDAGISel::SelectVOP3NoMods(SDValue In, SDValue &Src,
1515 SDValue &SrcMods) const {
1516 bool Res = SelectVOP3Mods(In, Src, SrcMods);
1517 return Res && cast<ConstantSDNode>(SrcMods)->isNullValue();
1518}
1519
Tom Stellardb4a313a2014-08-01 00:32:39 +00001520bool AMDGPUDAGToDAGISel::SelectVOP3Mods0(SDValue In, SDValue &Src,
1521 SDValue &SrcMods, SDValue &Clamp,
1522 SDValue &Omod) const {
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +00001523 SDLoc DL(In);
Tom Stellardb4a313a2014-08-01 00:32:39 +00001524 // FIXME: Handle Clamp and Omod
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +00001525 Clamp = CurDAG->getTargetConstant(0, DL, MVT::i32);
1526 Omod = CurDAG->getTargetConstant(0, DL, MVT::i32);
Tom Stellardb4a313a2014-08-01 00:32:39 +00001527
1528 return SelectVOP3Mods(In, Src, SrcMods);
1529}
1530
Tom Stellarddb5a11f2015-07-13 15:47:57 +00001531bool AMDGPUDAGToDAGISel::SelectVOP3NoMods0(SDValue In, SDValue &Src,
1532 SDValue &SrcMods, SDValue &Clamp,
1533 SDValue &Omod) const {
1534 bool Res = SelectVOP3Mods0(In, Src, SrcMods, Clamp, Omod);
1535
1536 return Res && cast<ConstantSDNode>(SrcMods)->isNullValue() &&
1537 cast<ConstantSDNode>(Clamp)->isNullValue() &&
1538 cast<ConstantSDNode>(Omod)->isNullValue();
1539}
1540
Matt Arsenault1cffa4c2014-11-13 19:49:04 +00001541bool AMDGPUDAGToDAGISel::SelectVOP3Mods0Clamp(SDValue In, SDValue &Src,
1542 SDValue &SrcMods,
1543 SDValue &Omod) const {
1544 // FIXME: Handle Omod
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +00001545 Omod = CurDAG->getTargetConstant(0, SDLoc(In), MVT::i32);
Matt Arsenault1cffa4c2014-11-13 19:49:04 +00001546
1547 return SelectVOP3Mods(In, Src, SrcMods);
1548}
1549
Matt Arsenault4831ce52015-01-06 23:00:37 +00001550bool AMDGPUDAGToDAGISel::SelectVOP3Mods0Clamp0OMod(SDValue In, SDValue &Src,
1551 SDValue &SrcMods,
1552 SDValue &Clamp,
1553 SDValue &Omod) const {
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +00001554 Clamp = Omod = CurDAG->getTargetConstant(0, SDLoc(In), MVT::i32);
Matt Arsenault4831ce52015-01-06 23:00:37 +00001555 return SelectVOP3Mods(In, Src, SrcMods);
1556}
1557
Matt Arsenault4bf43d42015-09-25 17:27:08 +00001558void AMDGPUDAGToDAGISel::PreprocessISelDAG() {
1559 bool Modified = false;
1560
Matt Arsenaultf43c2a02016-03-23 21:49:25 +00001561 MachineFrameInfo *MFI = CurDAG->getMachineFunction().getFrameInfo();
1562
1563 // Handle the perverse case where a frame index is being stored. We don't
1564 // want to see multiple frame index operands on the same instruction since
1565 // it complicates things and violates some assumptions about frame index
1566 // lowering.
1567 for (int I = MFI->getObjectIndexBegin(), E = MFI->getObjectIndexEnd();
1568 I != E; ++I) {
1569 SDValue FI = CurDAG->getTargetFrameIndex(I, MVT::i32);
1570
1571 // It's possible that we have a frame index defined in the function that
1572 // isn't used in this block.
1573 if (FI.use_empty())
1574 continue;
1575
1576 // Skip over the AssertZext inserted during lowering.
1577 SDValue EffectiveFI = FI;
1578 auto It = FI->use_begin();
1579 if (It->getOpcode() == ISD::AssertZext && FI->hasOneUse()) {
1580 EffectiveFI = SDValue(*It, 0);
1581 It = EffectiveFI->use_begin();
1582 }
1583
1584 for (auto It = EffectiveFI->use_begin(); !It.atEnd(); ) {
1585 SDUse &Use = It.getUse();
1586 SDNode *User = Use.getUser();
1587 unsigned OpIdx = It.getOperandNo();
1588 ++It;
1589
1590 if (MemSDNode *M = dyn_cast<MemSDNode>(User)) {
1591 unsigned PtrIdx = M->getOpcode() == ISD::STORE ? 2 : 1;
1592 if (OpIdx == PtrIdx)
1593 continue;
1594
1595 unsigned OpN = OpN = M->getNumOperands();
1596 SDValue NewOps[8];
1597
1598 assert(OpN < array_lengthof(NewOps));
1599 for (unsigned Op = 0; Op != OpN; ++Op) {
1600 if (Op != OpIdx) {
1601 NewOps[Op] = M->getOperand(Op);
1602 continue;
1603 }
1604
1605 MachineSDNode *Mov = CurDAG->getMachineNode(AMDGPU::V_MOV_B32_e32,
1606 SDLoc(M), MVT::i32, FI);
1607 NewOps[Op] = SDValue(Mov, 0);
1608 }
1609
1610 CurDAG->UpdateNodeOperands(M, makeArrayRef(NewOps, OpN));
1611 Modified = true;
1612 }
1613 }
1614 }
1615
Matt Arsenault4bf43d42015-09-25 17:27:08 +00001616 // XXX - Other targets seem to be able to do this without a worklist.
1617 SmallVector<LoadSDNode *, 8> LoadsToReplace;
1618 SmallVector<StoreSDNode *, 8> StoresToReplace;
1619
1620 for (SDNode &Node : CurDAG->allnodes()) {
1621 if (LoadSDNode *LD = dyn_cast<LoadSDNode>(&Node)) {
1622 EVT VT = LD->getValueType(0);
1623 if (VT != MVT::i64 || LD->getExtensionType() != ISD::NON_EXTLOAD)
1624 continue;
1625
1626 // To simplify the TableGen patters, we replace all i64 loads with v2i32
1627 // loads. Alternatively, we could promote i64 loads to v2i32 during DAG
1628 // legalization, however, so places (ExpandUnalignedLoad) in the DAG
1629 // legalizer assume that if i64 is legal, so doing this promotion early
1630 // can cause problems.
1631 LoadsToReplace.push_back(LD);
1632 } else if (StoreSDNode *ST = dyn_cast<StoreSDNode>(&Node)) {
1633 // Handle i64 stores here for the same reason mentioned above for loads.
1634 SDValue Value = ST->getValue();
1635 if (Value.getValueType() != MVT::i64 || ST->isTruncatingStore())
1636 continue;
1637 StoresToReplace.push_back(ST);
1638 }
1639 }
1640
1641 for (LoadSDNode *LD : LoadsToReplace) {
1642 SDLoc SL(LD);
1643
1644 SDValue NewLoad = CurDAG->getLoad(MVT::v2i32, SL, LD->getChain(),
1645 LD->getBasePtr(), LD->getMemOperand());
1646 SDValue BitCast = CurDAG->getNode(ISD::BITCAST, SL,
1647 MVT::i64, NewLoad);
1648 CurDAG->ReplaceAllUsesOfValueWith(SDValue(LD, 1), NewLoad.getValue(1));
1649 CurDAG->ReplaceAllUsesOfValueWith(SDValue(LD, 0), BitCast);
1650 Modified = true;
1651 }
1652
1653 for (StoreSDNode *ST : StoresToReplace) {
1654 SDValue NewValue = CurDAG->getNode(ISD::BITCAST, SDLoc(ST),
1655 MVT::v2i32, ST->getValue());
1656 const SDValue StoreOps[] = {
1657 ST->getChain(),
1658 NewValue,
1659 ST->getBasePtr(),
1660 ST->getOffset()
1661 };
1662
1663 CurDAG->UpdateNodeOperands(ST, StoreOps);
1664 Modified = true;
1665 }
1666
1667 // XXX - Is this necessary?
1668 if (Modified)
1669 CurDAG->RemoveDeadNodes();
1670}
1671
Christian Konigd910b7d2013-02-26 17:52:16 +00001672void AMDGPUDAGToDAGISel::PostprocessISelDAG() {
Bill Wendlinga3cd3502013-06-19 21:36:55 +00001673 const AMDGPUTargetLowering& Lowering =
Matt Arsenault209a7b92014-04-18 07:40:20 +00001674 *static_cast<const AMDGPUTargetLowering*>(getTargetLowering());
Vincent Lejeuneab3baf82013-09-12 23:44:44 +00001675 bool IsModified = false;
1676 do {
1677 IsModified = false;
1678 // Go over all selected nodes and try to fold them a bit more
Pete Cooper65c69402015-07-14 22:10:54 +00001679 for (SDNode &Node : CurDAG->allnodes()) {
1680 MachineSDNode *MachineNode = dyn_cast<MachineSDNode>(&Node);
Vincent Lejeuneab3baf82013-09-12 23:44:44 +00001681 if (!MachineNode)
1682 continue;
Christian Konigd910b7d2013-02-26 17:52:16 +00001683
Vincent Lejeuneab3baf82013-09-12 23:44:44 +00001684 SDNode *ResNode = Lowering.PostISelFolding(MachineNode, *CurDAG);
Pete Cooper65c69402015-07-14 22:10:54 +00001685 if (ResNode != &Node) {
1686 ReplaceUses(&Node, ResNode);
Vincent Lejeuneab3baf82013-09-12 23:44:44 +00001687 IsModified = true;
1688 }
Tom Stellard2183b702013-06-03 17:39:46 +00001689 }
Vincent Lejeuneab3baf82013-09-12 23:44:44 +00001690 CurDAG->RemoveDeadNodes();
1691 } while (IsModified);
Christian Konigd910b7d2013-02-26 17:52:16 +00001692}