blob: 4c8aa4b0c2a02f355325558182ac6d89f567b48a [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"
Jan Veselyf97de002016-05-13 20:39:29 +000021#include "llvm/Analysis/ValueTracking.h"
Tom Stellard58ac7442014-04-29 23:12:48 +000022#include "llvm/CodeGen/FunctionLoweringInfo.h"
Matt Arsenaultf43c2a02016-03-23 21:49:25 +000023#include "llvm/CodeGen/MachineFrameInfo.h"
Matt Arsenaultd9d659a2015-11-03 22:30:08 +000024#include "llvm/CodeGen/PseudoSourceValue.h"
Benjamin Kramerd78bb462013-05-23 17:10:37 +000025#include "llvm/CodeGen/SelectionDAG.h"
Tom Stellard75aadc22012-12-11 21:25:42 +000026#include "llvm/CodeGen/SelectionDAGISel.h"
Oliver Stannard7e7d9832016-02-02 13:52:43 +000027#include "llvm/IR/DiagnosticInfo.h"
Tom Stellard75aadc22012-12-11 21:25:42 +000028
29using namespace llvm;
30
Matt Arsenaultd2759212016-02-13 01:24:08 +000031namespace llvm {
32class R600InstrInfo;
33}
34
Tom Stellard75aadc22012-12-11 21:25:42 +000035//===----------------------------------------------------------------------===//
36// Instruction Selector Implementation
37//===----------------------------------------------------------------------===//
38
39namespace {
Tom Stellardbc4497b2016-02-12 23:45:29 +000040
41static bool isCBranchSCC(const SDNode *N) {
42 assert(N->getOpcode() == ISD::BRCOND);
43 if (!N->hasOneUse())
44 return false;
45
46 SDValue Cond = N->getOperand(1);
47 if (Cond.getOpcode() == ISD::CopyToReg)
48 Cond = Cond.getOperand(2);
49 return Cond.getOpcode() == ISD::SETCC &&
50 Cond.getOperand(0).getValueType() == MVT::i32 &&
51 Cond.hasOneUse();
52}
53
Tom Stellard75aadc22012-12-11 21:25:42 +000054/// AMDGPU specific code to select AMDGPU machine instructions for
55/// SelectionDAG operations.
56class AMDGPUDAGToDAGISel : public SelectionDAGISel {
57 // Subtarget - Keep a pointer to the AMDGPU Subtarget around so that we can
58 // make the right decision when generating code for different targets.
Eric Christopher7792e322015-01-30 23:24:40 +000059 const AMDGPUSubtarget *Subtarget;
NAKAMURA Takumia9cb5382015-09-22 11:14:39 +000060
Tom Stellard75aadc22012-12-11 21:25:42 +000061public:
62 AMDGPUDAGToDAGISel(TargetMachine &TM);
63 virtual ~AMDGPUDAGToDAGISel();
Eric Christopher7792e322015-01-30 23:24:40 +000064 bool runOnMachineFunction(MachineFunction &MF) override;
Justin Bogner95927c02016-05-12 21:03:32 +000065 void Select(SDNode *N) override;
Craig Topper5656db42014-04-29 07:57:24 +000066 const char *getPassName() const override;
Matt Arsenault4bf43d42015-09-25 17:27:08 +000067 void PreprocessISelDAG() override;
Craig Topper5656db42014-04-29 07:57:24 +000068 void PostprocessISelDAG() override;
Tom Stellard75aadc22012-12-11 21:25:42 +000069
70private:
Tom Stellard7ed0b522014-04-03 20:19:27 +000071 bool isInlineImmediate(SDNode *N) const;
Vincent Lejeunec6896792013-06-04 23:17:15 +000072 bool FoldOperand(SDValue &Src, SDValue &Sel, SDValue &Neg, SDValue &Abs,
Tom Stellard84021442013-07-23 01:48:24 +000073 const R600InstrInfo *TII);
Tom Stellard365366f2013-01-23 02:09:06 +000074 bool FoldOperands(unsigned, const R600InstrInfo *, std::vector<SDValue> &);
Vincent Lejeunec6896792013-06-04 23:17:15 +000075 bool FoldDotOperands(unsigned, const R600InstrInfo *, std::vector<SDValue> &);
Tom Stellard75aadc22012-12-11 21:25:42 +000076
77 // Complex pattern selectors
78 bool SelectADDRParam(SDValue Addr, SDValue& R1, SDValue& R2);
79 bool SelectADDR(SDValue N, SDValue &R1, SDValue &R2);
80 bool SelectADDR64(SDValue N, SDValue &R1, SDValue &R2);
81
82 static bool checkType(const Value *ptr, unsigned int addrspace);
Nick Lewyckyaad475b2014-04-15 07:22:52 +000083 static bool checkPrivateAddress(const MachineMemOperand *Op);
Tom Stellard75aadc22012-12-11 21:25:42 +000084
Jan Vesely43b7b5b2016-04-07 19:23:11 +000085 static bool isGlobalStore(const MemSDNode *N);
86 static bool isFlatStore(const MemSDNode *N);
Tom Stellard75aadc22012-12-11 21:25:42 +000087 static bool isPrivateStore(const StoreSDNode *N);
88 static bool isLocalStore(const StoreSDNode *N);
89 static bool isRegionStore(const StoreSDNode *N);
90
Matt Arsenault2aabb062013-06-18 23:37:58 +000091 bool isCPLoad(const LoadSDNode *N) const;
Jan Vesely43b7b5b2016-04-07 19:23:11 +000092 bool isConstantLoad(const MemSDNode *N, int cbID) const;
93 bool isGlobalLoad(const MemSDNode *N) const;
94 bool isFlatLoad(const MemSDNode *N) const;
Matt Arsenault2aabb062013-06-18 23:37:58 +000095 bool isParamLoad(const LoadSDNode *N) const;
96 bool isPrivateLoad(const LoadSDNode *N) const;
97 bool isLocalLoad(const LoadSDNode *N) const;
98 bool isRegionLoad(const LoadSDNode *N) const;
Tom Stellard75aadc22012-12-11 21:25:42 +000099
Tom Stellardbc4497b2016-02-12 23:45:29 +0000100 bool isUniformBr(const SDNode *N) const;
101
Tom Stellard381a94a2015-05-12 15:00:49 +0000102 SDNode *glueCopyToM0(SDNode *N) const;
103
Tom Stellarddf94dc32013-08-14 23:24:24 +0000104 const TargetRegisterClass *getOperandRegClass(SDNode *N, unsigned OpNo) const;
Tom Stellard365366f2013-01-23 02:09:06 +0000105 bool SelectGlobalValueConstantOffset(SDValue Addr, SDValue& IntPtr);
Matt Arsenault209a7b92014-04-18 07:40:20 +0000106 bool SelectGlobalValueVariableOffset(SDValue Addr, SDValue &BaseReg,
107 SDValue& Offset);
Tom Stellard75aadc22012-12-11 21:25:42 +0000108 bool SelectADDRVTX_READ(SDValue Addr, SDValue &Base, SDValue &Offset);
Tom Stellardf3b2a1e2013-02-06 17:32:29 +0000109 bool SelectADDRIndirect(SDValue Addr, SDValue &Base, SDValue &Offset);
Tom Stellard85e8b6d2014-08-22 18:49:33 +0000110 bool isDSOffsetLegal(const SDValue &Base, unsigned Offset,
111 unsigned OffsetBits) const;
112 bool SelectDS1Addr1Offset(SDValue Ptr, SDValue &Base, SDValue &Offset) const;
Tom Stellardf3fc5552014-08-22 18:49:35 +0000113 bool SelectDS64Bit4ByteAligned(SDValue Ptr, SDValue &Base, SDValue &Offset0,
114 SDValue &Offset1) const;
Changpeng Fangb41574a2015-12-22 20:55:23 +0000115 bool SelectMUBUF(SDValue Addr, SDValue &SRsrc, SDValue &VAddr,
Tom Stellard155bbb72014-08-11 22:18:17 +0000116 SDValue &SOffset, SDValue &Offset, SDValue &Offen,
117 SDValue &Idxen, SDValue &Addr64, SDValue &GLC, SDValue &SLC,
118 SDValue &TFE) const;
119 bool SelectMUBUFAddr64(SDValue Addr, SDValue &SRsrc, SDValue &VAddr,
Tom Stellard1f9939f2015-02-27 14:59:41 +0000120 SDValue &SOffset, SDValue &Offset, SDValue &GLC,
121 SDValue &SLC, SDValue &TFE) const;
Tom Stellard7980fc82014-09-25 18:30:26 +0000122 bool SelectMUBUFAddr64(SDValue Addr, SDValue &SRsrc,
Tom Stellardc53861a2015-02-11 00:34:32 +0000123 SDValue &VAddr, SDValue &SOffset, SDValue &Offset,
Tom Stellard7980fc82014-09-25 18:30:26 +0000124 SDValue &SLC) const;
Tom Stellardb02094e2014-07-21 15:45:01 +0000125 bool SelectMUBUFScratch(SDValue Addr, SDValue &RSrc, SDValue &VAddr,
126 SDValue &SOffset, SDValue &ImmOffset) const;
Tom Stellard155bbb72014-08-11 22:18:17 +0000127 bool SelectMUBUFOffset(SDValue Addr, SDValue &SRsrc, SDValue &SOffset,
128 SDValue &Offset, SDValue &GLC, SDValue &SLC,
Tom Stellardb02094e2014-07-21 15:45:01 +0000129 SDValue &TFE) const;
Tom Stellard7980fc82014-09-25 18:30:26 +0000130 bool SelectMUBUFOffset(SDValue Addr, SDValue &SRsrc, SDValue &Soffset,
Matt Arsenault88701812016-06-09 23:42:48 +0000131 SDValue &Offset, SDValue &SLC) const;
Jan Vesely43b7b5b2016-04-07 19:23:11 +0000132 bool SelectMUBUFOffset(SDValue Addr, SDValue &SRsrc, SDValue &Soffset,
133 SDValue &Offset) const;
Nicolai Haehnle3003ba02016-03-18 16:24:20 +0000134 void SelectMUBUFConstant(SDValue Constant,
135 SDValue &SOffset,
136 SDValue &ImmOffset) const;
137 bool SelectMUBUFIntrinsicOffset(SDValue Offset, SDValue &SOffset,
138 SDValue &ImmOffset) const;
139 bool SelectMUBUFIntrinsicVOffset(SDValue Offset, SDValue &SOffset,
140 SDValue &ImmOffset, SDValue &VOffset) const;
Matt Arsenault7757c592016-06-09 23:42:54 +0000141
142 bool SelectFlat(SDValue Addr, SDValue &VAddr,
143 SDValue &SLC, SDValue &TFE) const;
144
Tom Stellarddee26a22015-08-06 19:28:30 +0000145 bool SelectSMRDOffset(SDValue ByteOffsetNode, SDValue &Offset,
146 bool &Imm) const;
147 bool SelectSMRD(SDValue Addr, SDValue &SBase, SDValue &Offset,
148 bool &Imm) const;
149 bool SelectSMRDImm(SDValue Addr, SDValue &SBase, SDValue &Offset) const;
Tom Stellard217361c2015-08-06 19:28:38 +0000150 bool SelectSMRDImm32(SDValue Addr, SDValue &SBase, SDValue &Offset) const;
Tom Stellarddee26a22015-08-06 19:28:30 +0000151 bool SelectSMRDSgpr(SDValue Addr, SDValue &SBase, SDValue &Offset) const;
152 bool SelectSMRDBufferImm(SDValue Addr, SDValue &Offset) const;
Tom Stellard217361c2015-08-06 19:28:38 +0000153 bool SelectSMRDBufferImm32(SDValue Addr, SDValue &Offset) const;
Tom Stellarddee26a22015-08-06 19:28:30 +0000154 bool SelectSMRDBufferSgpr(SDValue Addr, SDValue &Offset) const;
Tom Stellardb4a313a2014-08-01 00:32:39 +0000155 bool SelectVOP3Mods(SDValue In, SDValue &Src, SDValue &SrcMods) const;
Tom Stellarddb5a11f2015-07-13 15:47:57 +0000156 bool SelectVOP3NoMods(SDValue In, SDValue &Src, SDValue &SrcMods) const;
Tom Stellardb4a313a2014-08-01 00:32:39 +0000157 bool SelectVOP3Mods0(SDValue In, SDValue &Src, SDValue &SrcMods,
158 SDValue &Clamp, SDValue &Omod) const;
Tom Stellarddb5a11f2015-07-13 15:47:57 +0000159 bool SelectVOP3NoMods0(SDValue In, SDValue &Src, SDValue &SrcMods,
160 SDValue &Clamp, SDValue &Omod) const;
Tom Stellard75aadc22012-12-11 21:25:42 +0000161
Matt Arsenault1cffa4c2014-11-13 19:49:04 +0000162 bool SelectVOP3Mods0Clamp(SDValue In, SDValue &Src, SDValue &SrcMods,
163 SDValue &Omod) const;
Matt Arsenault4831ce52015-01-06 23:00:37 +0000164 bool SelectVOP3Mods0Clamp0OMod(SDValue In, SDValue &Src, SDValue &SrcMods,
165 SDValue &Clamp,
166 SDValue &Omod) const;
Matt Arsenault1cffa4c2014-11-13 19:49:04 +0000167
Justin Bogner95927c02016-05-12 21:03:32 +0000168 void SelectADD_SUB_I64(SDNode *N);
169 void SelectDIV_SCALE(SDNode *N);
Matt Arsenault9fa3f932014-06-23 18:00:34 +0000170
Marek Olsak9b728682015-03-24 13:40:27 +0000171 SDNode *getS_BFE(unsigned Opcode, SDLoc DL, SDValue Val,
172 uint32_t Offset, uint32_t Width);
Justin Bogner95927c02016-05-12 21:03:32 +0000173 void SelectS_BFEFromShifts(SDNode *N);
174 void SelectS_BFE(SDNode *N);
175 void SelectBRCOND(SDNode *N);
Matt Arsenault88701812016-06-09 23:42:48 +0000176 void SelectATOMIC_CMP_SWAP(SDNode *N);
Marek Olsak9b728682015-03-24 13:40:27 +0000177
Tom Stellard75aadc22012-12-11 21:25:42 +0000178 // Include the pieces autogenerated from the target description.
179#include "AMDGPUGenDAGISel.inc"
180};
181} // end anonymous namespace
182
183/// \brief This pass converts a legalized DAG into a AMDGPU-specific
184// DAG, ready for instruction scheduling.
Matt Arsenault209a7b92014-04-18 07:40:20 +0000185FunctionPass *llvm::createAMDGPUISelDag(TargetMachine &TM) {
Tom Stellard75aadc22012-12-11 21:25:42 +0000186 return new AMDGPUDAGToDAGISel(TM);
187}
188
Bill Wendlinga3cd3502013-06-19 21:36:55 +0000189AMDGPUDAGToDAGISel::AMDGPUDAGToDAGISel(TargetMachine &TM)
Eric Christopher7792e322015-01-30 23:24:40 +0000190 : SelectionDAGISel(TM) {}
191
192bool AMDGPUDAGToDAGISel::runOnMachineFunction(MachineFunction &MF) {
193 Subtarget = &static_cast<const AMDGPUSubtarget &>(MF.getSubtarget());
194 return SelectionDAGISel::runOnMachineFunction(MF);
Tom Stellard75aadc22012-12-11 21:25:42 +0000195}
196
197AMDGPUDAGToDAGISel::~AMDGPUDAGToDAGISel() {
198}
199
Tom Stellard7ed0b522014-04-03 20:19:27 +0000200bool AMDGPUDAGToDAGISel::isInlineImmediate(SDNode *N) const {
201 const SITargetLowering *TL
202 = static_cast<const SITargetLowering *>(getTargetLowering());
203 return TL->analyzeImmediate(N) == 0;
204}
205
Tom Stellarddf94dc32013-08-14 23:24:24 +0000206/// \brief Determine the register class for \p OpNo
207/// \returns The register class of the virtual register that will be used for
208/// the given operand number \OpNo or NULL if the register class cannot be
209/// determined.
210const TargetRegisterClass *AMDGPUDAGToDAGISel::getOperandRegClass(SDNode *N,
211 unsigned OpNo) const {
Matt Arsenault209a7b92014-04-18 07:40:20 +0000212 if (!N->isMachineOpcode())
213 return nullptr;
214
Tom Stellarddf94dc32013-08-14 23:24:24 +0000215 switch (N->getMachineOpcode()) {
216 default: {
Eric Christopherd9134482014-08-04 21:25:23 +0000217 const MCInstrDesc &Desc =
Eric Christopher7792e322015-01-30 23:24:40 +0000218 Subtarget->getInstrInfo()->get(N->getMachineOpcode());
Alexey Samsonov3186eb32013-08-15 07:11:34 +0000219 unsigned OpIdx = Desc.getNumDefs() + OpNo;
220 if (OpIdx >= Desc.getNumOperands())
Matt Arsenault209a7b92014-04-18 07:40:20 +0000221 return nullptr;
Alexey Samsonov3186eb32013-08-15 07:11:34 +0000222 int RegClass = Desc.OpInfo[OpIdx].RegClass;
Matt Arsenault209a7b92014-04-18 07:40:20 +0000223 if (RegClass == -1)
224 return nullptr;
225
Eric Christopher7792e322015-01-30 23:24:40 +0000226 return Subtarget->getRegisterInfo()->getRegClass(RegClass);
Tom Stellarddf94dc32013-08-14 23:24:24 +0000227 }
228 case AMDGPU::REG_SEQUENCE: {
Matt Arsenault209a7b92014-04-18 07:40:20 +0000229 unsigned RCID = cast<ConstantSDNode>(N->getOperand(0))->getZExtValue();
Eric Christopherd9134482014-08-04 21:25:23 +0000230 const TargetRegisterClass *SuperRC =
Eric Christopher7792e322015-01-30 23:24:40 +0000231 Subtarget->getRegisterInfo()->getRegClass(RCID);
Matt Arsenault209a7b92014-04-18 07:40:20 +0000232
233 SDValue SubRegOp = N->getOperand(OpNo + 1);
234 unsigned SubRegIdx = cast<ConstantSDNode>(SubRegOp)->getZExtValue();
Eric Christopher7792e322015-01-30 23:24:40 +0000235 return Subtarget->getRegisterInfo()->getSubClassWithSubReg(SuperRC,
236 SubRegIdx);
Tom Stellarddf94dc32013-08-14 23:24:24 +0000237 }
238 }
239}
240
Tom Stellard75aadc22012-12-11 21:25:42 +0000241bool AMDGPUDAGToDAGISel::SelectADDRParam(
Matt Arsenault209a7b92014-04-18 07:40:20 +0000242 SDValue Addr, SDValue& R1, SDValue& R2) {
Tom Stellard75aadc22012-12-11 21:25:42 +0000243
244 if (Addr.getOpcode() == ISD::FrameIndex) {
245 if (FrameIndexSDNode *FIN = dyn_cast<FrameIndexSDNode>(Addr)) {
246 R1 = CurDAG->getTargetFrameIndex(FIN->getIndex(), MVT::i32);
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +0000247 R2 = CurDAG->getTargetConstant(0, SDLoc(Addr), MVT::i32);
Tom Stellard75aadc22012-12-11 21:25:42 +0000248 } 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 } else if (Addr.getOpcode() == ISD::ADD) {
253 R1 = Addr.getOperand(0);
254 R2 = Addr.getOperand(1);
255 } else {
256 R1 = Addr;
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +0000257 R2 = CurDAG->getTargetConstant(0, SDLoc(Addr), MVT::i32);
Tom Stellard75aadc22012-12-11 21:25:42 +0000258 }
259 return true;
260}
261
262bool AMDGPUDAGToDAGISel::SelectADDR(SDValue Addr, SDValue& R1, SDValue& R2) {
263 if (Addr.getOpcode() == ISD::TargetExternalSymbol ||
264 Addr.getOpcode() == ISD::TargetGlobalAddress) {
265 return false;
266 }
267 return SelectADDRParam(Addr, R1, R2);
268}
269
270
271bool AMDGPUDAGToDAGISel::SelectADDR64(SDValue Addr, SDValue& R1, SDValue& R2) {
272 if (Addr.getOpcode() == ISD::TargetExternalSymbol ||
273 Addr.getOpcode() == ISD::TargetGlobalAddress) {
274 return false;
275 }
276
277 if (Addr.getOpcode() == ISD::FrameIndex) {
278 if (FrameIndexSDNode *FIN = dyn_cast<FrameIndexSDNode>(Addr)) {
279 R1 = CurDAG->getTargetFrameIndex(FIN->getIndex(), MVT::i64);
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +0000280 R2 = CurDAG->getTargetConstant(0, SDLoc(Addr), MVT::i64);
Tom Stellard75aadc22012-12-11 21:25:42 +0000281 } 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 } else if (Addr.getOpcode() == ISD::ADD) {
286 R1 = Addr.getOperand(0);
287 R2 = Addr.getOperand(1);
288 } else {
289 R1 = Addr;
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +0000290 R2 = CurDAG->getTargetConstant(0, SDLoc(Addr), MVT::i64);
Tom Stellard75aadc22012-12-11 21:25:42 +0000291 }
292 return true;
293}
294
Tom Stellard381a94a2015-05-12 15:00:49 +0000295SDNode *AMDGPUDAGToDAGISel::glueCopyToM0(SDNode *N) const {
296 if (Subtarget->getGeneration() < AMDGPUSubtarget::SOUTHERN_ISLANDS ||
297 !checkType(cast<MemSDNode>(N)->getMemOperand()->getValue(),
298 AMDGPUAS::LOCAL_ADDRESS))
299 return N;
300
301 const SITargetLowering& Lowering =
302 *static_cast<const SITargetLowering*>(getTargetLowering());
303
304 // Write max value to m0 before each load operation
305
306 SDValue M0 = Lowering.copyToM0(*CurDAG, CurDAG->getEntryNode(), SDLoc(N),
307 CurDAG->getTargetConstant(-1, SDLoc(N), MVT::i32));
308
309 SDValue Glue = M0.getValue(1);
310
311 SmallVector <SDValue, 8> Ops;
312 for (unsigned i = 0, e = N->getNumOperands(); i != e; ++i) {
313 Ops.push_back(N->getOperand(i));
314 }
315 Ops.push_back(Glue);
316 CurDAG->MorphNodeTo(N, N->getOpcode(), N->getVTList(), Ops);
317
318 return N;
319}
320
Matt Arsenault61cb6fa2015-11-11 00:01:36 +0000321static unsigned selectSGPRVectorRegClassID(unsigned NumVectorElts) {
Matt Arsenaultf1aebbf2015-11-02 23:30:48 +0000322 switch (NumVectorElts) {
323 case 1:
324 return AMDGPU::SReg_32RegClassID;
325 case 2:
326 return AMDGPU::SReg_64RegClassID;
327 case 4:
328 return AMDGPU::SReg_128RegClassID;
329 case 8:
330 return AMDGPU::SReg_256RegClassID;
331 case 16:
332 return AMDGPU::SReg_512RegClassID;
333 }
334
335 llvm_unreachable("invalid vector size");
336}
337
Justin Bogner95927c02016-05-12 21:03:32 +0000338void AMDGPUDAGToDAGISel::Select(SDNode *N) {
Tom Stellard75aadc22012-12-11 21:25:42 +0000339 unsigned int Opc = N->getOpcode();
340 if (N->isMachineOpcode()) {
Tim Northover31d093c2013-09-22 08:21:56 +0000341 N->setNodeId(-1);
Justin Bogner95927c02016-05-12 21:03:32 +0000342 return; // Already selected.
Tom Stellard75aadc22012-12-11 21:25:42 +0000343 }
Matt Arsenault78b86702014-04-18 05:19:26 +0000344
Matt Arsenaulta9dbdca2016-04-12 14:05:04 +0000345 if (isa<AtomicSDNode>(N) ||
346 (Opc == AMDGPUISD::ATOMIC_INC || Opc == AMDGPUISD::ATOMIC_DEC))
Tom Stellard381a94a2015-05-12 15:00:49 +0000347 N = glueCopyToM0(N);
348
Tom Stellard75aadc22012-12-11 21:25:42 +0000349 switch (Opc) {
350 default: break;
Tom Stellard1f15bff2014-02-25 21:36:18 +0000351 // We are selecting i64 ADD here instead of custom lower it during
352 // DAG legalization, so we can fold some i64 ADDs used for address
353 // calculation into the LOAD and STORE instructions.
Matt Arsenaultb8b51532014-06-23 18:00:38 +0000354 case ISD::ADD:
355 case ISD::SUB: {
Tom Stellard1f15bff2014-02-25 21:36:18 +0000356 if (N->getValueType(0) != MVT::i64 ||
Eric Christopher7792e322015-01-30 23:24:40 +0000357 Subtarget->getGeneration() < AMDGPUSubtarget::SOUTHERN_ISLANDS)
Tom Stellard1f15bff2014-02-25 21:36:18 +0000358 break;
359
Justin Bogner95927c02016-05-12 21:03:32 +0000360 SelectADD_SUB_I64(N);
361 return;
Tom Stellard1f15bff2014-02-25 21:36:18 +0000362 }
Matt Arsenault064c2062014-06-11 17:40:32 +0000363 case ISD::SCALAR_TO_VECTOR:
Tom Stellard880a80a2014-06-17 16:53:14 +0000364 case AMDGPUISD::BUILD_VERTICAL_VECTOR:
Vincent Lejeune3b6f20e2013-03-05 15:04:49 +0000365 case ISD::BUILD_VECTOR: {
Tom Stellard8e5da412013-08-14 23:24:32 +0000366 unsigned RegClassID;
Eric Christopher7792e322015-01-30 23:24:40 +0000367 const AMDGPURegisterInfo *TRI = Subtarget->getRegisterInfo();
Tom Stellard8e5da412013-08-14 23:24:32 +0000368 EVT VT = N->getValueType(0);
369 unsigned NumVectorElts = VT.getVectorNumElements();
Matt Arsenault064c2062014-06-11 17:40:32 +0000370 EVT EltVT = VT.getVectorElementType();
371 assert(EltVT.bitsEq(MVT::i32));
Eric Christopher7792e322015-01-30 23:24:40 +0000372 if (Subtarget->getGeneration() >= AMDGPUSubtarget::SOUTHERN_ISLANDS) {
Matt Arsenault61cb6fa2015-11-11 00:01:36 +0000373 RegClassID = selectSGPRVectorRegClassID(NumVectorElts);
Tom Stellard8e5da412013-08-14 23:24:32 +0000374 } else {
375 // BUILD_VECTOR was lowered into an IMPLICIT_DEF + 4 INSERT_SUBREG
376 // that adds a 128 bits reg copy when going through TwoAddressInstructions
377 // pass. We want to avoid 128 bits copies as much as possible because they
378 // can't be bundled by our scheduler.
379 switch(NumVectorElts) {
380 case 2: RegClassID = AMDGPU::R600_Reg64RegClassID; break;
Tom Stellard880a80a2014-06-17 16:53:14 +0000381 case 4:
382 if (Opc == AMDGPUISD::BUILD_VERTICAL_VECTOR)
383 RegClassID = AMDGPU::R600_Reg128VerticalRegClassID;
384 else
385 RegClassID = AMDGPU::R600_Reg128RegClassID;
386 break;
Tom Stellard8e5da412013-08-14 23:24:32 +0000387 default: llvm_unreachable("Do not know how to lower this BUILD_VECTOR");
388 }
Vincent Lejeune3b6f20e2013-03-05 15:04:49 +0000389 }
Tom Stellard0344cdf2013-08-01 15:23:42 +0000390
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +0000391 SDLoc DL(N);
392 SDValue RegClass = CurDAG->getTargetConstant(RegClassID, DL, MVT::i32);
Tom Stellard8e5da412013-08-14 23:24:32 +0000393
394 if (NumVectorElts == 1) {
Justin Bogner95927c02016-05-12 21:03:32 +0000395 CurDAG->SelectNodeTo(N, AMDGPU::COPY_TO_REGCLASS, EltVT, N->getOperand(0),
396 RegClass);
397 return;
Tom Stellard0344cdf2013-08-01 15:23:42 +0000398 }
Tom Stellard8e5da412013-08-14 23:24:32 +0000399
400 assert(NumVectorElts <= 16 && "Vectors with more than 16 elements not "
401 "supported yet");
402 // 16 = Max Num Vector Elements
403 // 2 = 2 REG_SEQUENCE operands per element (value, subreg index)
404 // 1 = Vector Register Class
Matt Arsenault064c2062014-06-11 17:40:32 +0000405 SmallVector<SDValue, 16 * 2 + 1> RegSeqArgs(NumVectorElts * 2 + 1);
Tom Stellard8e5da412013-08-14 23:24:32 +0000406
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +0000407 RegSeqArgs[0] = CurDAG->getTargetConstant(RegClassID, DL, MVT::i32);
Vincent Lejeune3b6f20e2013-03-05 15:04:49 +0000408 bool IsRegSeq = true;
Matt Arsenault064c2062014-06-11 17:40:32 +0000409 unsigned NOps = N->getNumOperands();
410 for (unsigned i = 0; i < NOps; i++) {
Tom Stellard8e5da412013-08-14 23:24:32 +0000411 // XXX: Why is this here?
Benjamin Kramer619c4e52015-04-10 11:24:51 +0000412 if (isa<RegisterSDNode>(N->getOperand(i))) {
Vincent Lejeune3b6f20e2013-03-05 15:04:49 +0000413 IsRegSeq = false;
414 break;
415 }
Tom Stellard8e5da412013-08-14 23:24:32 +0000416 RegSeqArgs[1 + (2 * i)] = N->getOperand(i);
417 RegSeqArgs[1 + (2 * i) + 1] =
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +0000418 CurDAG->getTargetConstant(TRI->getSubRegFromChannel(i), DL,
419 MVT::i32);
Vincent Lejeune3b6f20e2013-03-05 15:04:49 +0000420 }
Matt Arsenault064c2062014-06-11 17:40:32 +0000421
422 if (NOps != NumVectorElts) {
423 // Fill in the missing undef elements if this was a scalar_to_vector.
424 assert(Opc == ISD::SCALAR_TO_VECTOR && NOps < NumVectorElts);
425
426 MachineSDNode *ImpDef = CurDAG->getMachineNode(TargetOpcode::IMPLICIT_DEF,
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +0000427 DL, EltVT);
Matt Arsenault064c2062014-06-11 17:40:32 +0000428 for (unsigned i = NOps; i < NumVectorElts; ++i) {
429 RegSeqArgs[1 + (2 * i)] = SDValue(ImpDef, 0);
430 RegSeqArgs[1 + (2 * i) + 1] =
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +0000431 CurDAG->getTargetConstant(TRI->getSubRegFromChannel(i), DL, MVT::i32);
Matt Arsenault064c2062014-06-11 17:40:32 +0000432 }
433 }
434
Vincent Lejeune3b6f20e2013-03-05 15:04:49 +0000435 if (!IsRegSeq)
436 break;
Justin Bogner95927c02016-05-12 21:03:32 +0000437 CurDAG->SelectNodeTo(N, AMDGPU::REG_SEQUENCE, N->getVTList(), RegSeqArgs);
438 return;
Vincent Lejeune3b6f20e2013-03-05 15:04:49 +0000439 }
Tom Stellard754f80f2013-04-05 23:31:51 +0000440 case ISD::BUILD_PAIR: {
441 SDValue RC, SubReg0, SubReg1;
Eric Christopher7792e322015-01-30 23:24:40 +0000442 if (Subtarget->getGeneration() <= AMDGPUSubtarget::NORTHERN_ISLANDS) {
Tom Stellard754f80f2013-04-05 23:31:51 +0000443 break;
444 }
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +0000445 SDLoc DL(N);
Tom Stellard754f80f2013-04-05 23:31:51 +0000446 if (N->getValueType(0) == MVT::i128) {
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +0000447 RC = CurDAG->getTargetConstant(AMDGPU::SReg_128RegClassID, DL, MVT::i32);
448 SubReg0 = CurDAG->getTargetConstant(AMDGPU::sub0_sub1, DL, MVT::i32);
449 SubReg1 = CurDAG->getTargetConstant(AMDGPU::sub2_sub3, DL, MVT::i32);
Tom Stellard754f80f2013-04-05 23:31:51 +0000450 } else if (N->getValueType(0) == MVT::i64) {
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +0000451 RC = CurDAG->getTargetConstant(AMDGPU::SReg_64RegClassID, DL, MVT::i32);
452 SubReg0 = CurDAG->getTargetConstant(AMDGPU::sub0, DL, MVT::i32);
453 SubReg1 = CurDAG->getTargetConstant(AMDGPU::sub1, DL, MVT::i32);
Tom Stellard754f80f2013-04-05 23:31:51 +0000454 } else {
455 llvm_unreachable("Unhandled value type for BUILD_PAIR");
456 }
457 const SDValue Ops[] = { RC, N->getOperand(0), SubReg0,
458 N->getOperand(1), SubReg1 };
Justin Bogner95927c02016-05-12 21:03:32 +0000459 ReplaceNode(N, CurDAG->getMachineNode(TargetOpcode::REG_SEQUENCE, DL,
460 N->getValueType(0), Ops));
461 return;
Tom Stellard754f80f2013-04-05 23:31:51 +0000462 }
Tom Stellard7ed0b522014-04-03 20:19:27 +0000463
464 case ISD::Constant:
465 case ISD::ConstantFP: {
Eric Christopher7792e322015-01-30 23:24:40 +0000466 if (Subtarget->getGeneration() < AMDGPUSubtarget::SOUTHERN_ISLANDS ||
Tom Stellard7ed0b522014-04-03 20:19:27 +0000467 N->getValueType(0).getSizeInBits() != 64 || isInlineImmediate(N))
468 break;
469
470 uint64_t Imm;
471 if (ConstantFPSDNode *FP = dyn_cast<ConstantFPSDNode>(N))
472 Imm = FP->getValueAPF().bitcastToAPInt().getZExtValue();
473 else {
Tom Stellard3cbe0142014-04-07 19:31:13 +0000474 ConstantSDNode *C = cast<ConstantSDNode>(N);
Tom Stellard7ed0b522014-04-03 20:19:27 +0000475 Imm = C->getZExtValue();
476 }
477
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +0000478 SDLoc DL(N);
479 SDNode *Lo = CurDAG->getMachineNode(AMDGPU::S_MOV_B32, DL, MVT::i32,
480 CurDAG->getConstant(Imm & 0xFFFFFFFF, DL,
481 MVT::i32));
482 SDNode *Hi = CurDAG->getMachineNode(AMDGPU::S_MOV_B32, DL, MVT::i32,
483 CurDAG->getConstant(Imm >> 32, DL, MVT::i32));
Tom Stellard7ed0b522014-04-03 20:19:27 +0000484 const SDValue Ops[] = {
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +0000485 CurDAG->getTargetConstant(AMDGPU::SReg_64RegClassID, DL, MVT::i32),
486 SDValue(Lo, 0), CurDAG->getTargetConstant(AMDGPU::sub0, DL, MVT::i32),
487 SDValue(Hi, 0), CurDAG->getTargetConstant(AMDGPU::sub1, DL, MVT::i32)
Tom Stellard7ed0b522014-04-03 20:19:27 +0000488 };
489
Justin Bogner95927c02016-05-12 21:03:32 +0000490 ReplaceNode(N, CurDAG->getMachineNode(TargetOpcode::REG_SEQUENCE, DL,
491 N->getValueType(0), Ops));
492 return;
Tom Stellard7ed0b522014-04-03 20:19:27 +0000493 }
Matt Arsenault4bf43d42015-09-25 17:27:08 +0000494 case ISD::LOAD:
Tom Stellard096b8c12015-02-04 20:49:49 +0000495 case ISD::STORE: {
Tom Stellard381a94a2015-05-12 15:00:49 +0000496 N = glueCopyToM0(N);
Tom Stellard096b8c12015-02-04 20:49:49 +0000497 break;
498 }
Matt Arsenault78b86702014-04-18 05:19:26 +0000499
500 case AMDGPUISD::BFE_I32:
501 case AMDGPUISD::BFE_U32: {
Eric Christopher7792e322015-01-30 23:24:40 +0000502 if (Subtarget->getGeneration() < AMDGPUSubtarget::SOUTHERN_ISLANDS)
Matt Arsenault78b86702014-04-18 05:19:26 +0000503 break;
504
505 // There is a scalar version available, but unlike the vector version which
506 // has a separate operand for the offset and width, the scalar version packs
507 // the width and offset into a single operand. Try to move to the scalar
508 // version if the offsets are constant, so that we can try to keep extended
509 // loads of kernel arguments in SGPRs.
510
511 // TODO: Technically we could try to pattern match scalar bitshifts of
512 // dynamic values, but it's probably not useful.
513 ConstantSDNode *Offset = dyn_cast<ConstantSDNode>(N->getOperand(1));
514 if (!Offset)
515 break;
516
517 ConstantSDNode *Width = dyn_cast<ConstantSDNode>(N->getOperand(2));
518 if (!Width)
519 break;
520
521 bool Signed = Opc == AMDGPUISD::BFE_I32;
522
Matt Arsenault78b86702014-04-18 05:19:26 +0000523 uint32_t OffsetVal = Offset->getZExtValue();
524 uint32_t WidthVal = Width->getZExtValue();
525
Justin Bogner95927c02016-05-12 21:03:32 +0000526 ReplaceNode(N, getS_BFE(Signed ? AMDGPU::S_BFE_I32 : AMDGPU::S_BFE_U32,
527 SDLoc(N), N->getOperand(0), OffsetVal, WidthVal));
528 return;
Matt Arsenault78b86702014-04-18 05:19:26 +0000529 }
Matt Arsenaultf2b0aeb2014-06-23 18:28:28 +0000530 case AMDGPUISD::DIV_SCALE: {
Justin Bogner95927c02016-05-12 21:03:32 +0000531 SelectDIV_SCALE(N);
532 return;
Matt Arsenaultf2b0aeb2014-06-23 18:28:28 +0000533 }
Tom Stellard3457a842014-10-09 19:06:00 +0000534 case ISD::CopyToReg: {
535 const SITargetLowering& Lowering =
536 *static_cast<const SITargetLowering*>(getTargetLowering());
537 Lowering.legalizeTargetIndependentNode(N, *CurDAG);
538 break;
539 }
Marek Olsak9b728682015-03-24 13:40:27 +0000540 case ISD::AND:
541 case ISD::SRL:
542 case ISD::SRA:
Matt Arsenault7e8de012016-04-22 22:59:16 +0000543 case ISD::SIGN_EXTEND_INREG:
Marek Olsak9b728682015-03-24 13:40:27 +0000544 if (N->getValueType(0) != MVT::i32 ||
545 Subtarget->getGeneration() < AMDGPUSubtarget::SOUTHERN_ISLANDS)
546 break;
547
Justin Bogner95927c02016-05-12 21:03:32 +0000548 SelectS_BFE(N);
549 return;
Tom Stellardbc4497b2016-02-12 23:45:29 +0000550 case ISD::BRCOND:
Justin Bogner95927c02016-05-12 21:03:32 +0000551 SelectBRCOND(N);
552 return;
Matt Arsenault88701812016-06-09 23:42:48 +0000553
554 case AMDGPUISD::ATOMIC_CMP_SWAP:
555 SelectATOMIC_CMP_SWAP(N);
556 return;
Tom Stellard75aadc22012-12-11 21:25:42 +0000557 }
Tom Stellard3457a842014-10-09 19:06:00 +0000558
Justin Bogner95927c02016-05-12 21:03:32 +0000559 SelectCode(N);
Tom Stellard365366f2013-01-23 02:09:06 +0000560}
561
Matt Arsenault209a7b92014-04-18 07:40:20 +0000562bool AMDGPUDAGToDAGISel::checkType(const Value *Ptr, unsigned AS) {
563 assert(AS != 0 && "Use checkPrivateAddress instead.");
564 if (!Ptr)
Tom Stellard75aadc22012-12-11 21:25:42 +0000565 return false;
Matt Arsenault209a7b92014-04-18 07:40:20 +0000566
567 return Ptr->getType()->getPointerAddressSpace() == AS;
Tom Stellard75aadc22012-12-11 21:25:42 +0000568}
569
Nick Lewyckyaad475b2014-04-15 07:22:52 +0000570bool AMDGPUDAGToDAGISel::checkPrivateAddress(const MachineMemOperand *Op) {
Matt Arsenault209a7b92014-04-18 07:40:20 +0000571 if (Op->getPseudoValue())
572 return true;
573
574 if (PointerType *PT = dyn_cast<PointerType>(Op->getValue()->getType()))
575 return PT->getAddressSpace() == AMDGPUAS::PRIVATE_ADDRESS;
576
577 return false;
Nick Lewyckyaad475b2014-04-15 07:22:52 +0000578}
579
Jan Vesely43b7b5b2016-04-07 19:23:11 +0000580bool AMDGPUDAGToDAGISel::isGlobalStore(const MemSDNode *N) {
581 if (!N->writeMem())
582 return false;
Nick Lewyckyaad475b2014-04-15 07:22:52 +0000583 return checkType(N->getMemOperand()->getValue(), AMDGPUAS::GLOBAL_ADDRESS);
Tom Stellard75aadc22012-12-11 21:25:42 +0000584}
585
586bool AMDGPUDAGToDAGISel::isPrivateStore(const StoreSDNode *N) {
Matt Arsenault209a7b92014-04-18 07:40:20 +0000587 const Value *MemVal = N->getMemOperand()->getValue();
588 return (!checkType(MemVal, AMDGPUAS::LOCAL_ADDRESS) &&
589 !checkType(MemVal, AMDGPUAS::GLOBAL_ADDRESS) &&
590 !checkType(MemVal, AMDGPUAS::REGION_ADDRESS));
Tom Stellard75aadc22012-12-11 21:25:42 +0000591}
592
593bool AMDGPUDAGToDAGISel::isLocalStore(const StoreSDNode *N) {
Nick Lewyckyaad475b2014-04-15 07:22:52 +0000594 return checkType(N->getMemOperand()->getValue(), AMDGPUAS::LOCAL_ADDRESS);
Tom Stellard75aadc22012-12-11 21:25:42 +0000595}
596
Jan Vesely43b7b5b2016-04-07 19:23:11 +0000597bool AMDGPUDAGToDAGISel::isFlatStore(const MemSDNode *N) {
598 if (!N->writeMem())
599 return false;
Matt Arsenault3f981402014-09-15 15:41:53 +0000600 return checkType(N->getMemOperand()->getValue(), AMDGPUAS::FLAT_ADDRESS);
601}
602
Tom Stellard75aadc22012-12-11 21:25:42 +0000603bool AMDGPUDAGToDAGISel::isRegionStore(const StoreSDNode *N) {
Nick Lewyckyaad475b2014-04-15 07:22:52 +0000604 return checkType(N->getMemOperand()->getValue(), AMDGPUAS::REGION_ADDRESS);
Tom Stellard75aadc22012-12-11 21:25:42 +0000605}
606
Jan Vesely43b7b5b2016-04-07 19:23:11 +0000607bool AMDGPUDAGToDAGISel::isConstantLoad(const MemSDNode *N, int CbId) const {
608 if (!N->readMem())
609 return false;
Matt Arsenault209a7b92014-04-18 07:40:20 +0000610 const Value *MemVal = N->getMemOperand()->getValue();
611 if (CbId == -1)
612 return checkType(MemVal, AMDGPUAS::CONSTANT_ADDRESS);
613
614 return checkType(MemVal, AMDGPUAS::CONSTANT_BUFFER_0 + CbId);
Tom Stellard75aadc22012-12-11 21:25:42 +0000615}
616
Jan Vesely43b7b5b2016-04-07 19:23:11 +0000617bool AMDGPUDAGToDAGISel::isGlobalLoad(const MemSDNode *N) const {
618 if (!N->readMem())
619 return false;
Jan Veselyf97de002016-05-13 20:39:29 +0000620 if (N->getAddressSpace() == AMDGPUAS::CONSTANT_ADDRESS) {
621 if (Subtarget->getGeneration() < AMDGPUSubtarget::SOUTHERN_ISLANDS)
622 return !isa<GlobalValue>(
623 GetUnderlyingObject(N->getMemOperand()->getValue(),
624 CurDAG->getDataLayout()));
625
626 //TODO: Why do we need this?
627 if (N->getMemoryVT().bitsLT(MVT::i32))
Tom Stellard8cb0e472013-07-23 23:54:56 +0000628 return true;
Jan Veselyf97de002016-05-13 20:39:29 +0000629 }
Eric Christopher7792e322015-01-30 23:24:40 +0000630
Nick Lewyckyaad475b2014-04-15 07:22:52 +0000631 return checkType(N->getMemOperand()->getValue(), AMDGPUAS::GLOBAL_ADDRESS);
Tom Stellard75aadc22012-12-11 21:25:42 +0000632}
633
Matt Arsenault2aabb062013-06-18 23:37:58 +0000634bool AMDGPUDAGToDAGISel::isParamLoad(const LoadSDNode *N) const {
Nick Lewyckyaad475b2014-04-15 07:22:52 +0000635 return checkType(N->getMemOperand()->getValue(), AMDGPUAS::PARAM_I_ADDRESS);
Tom Stellard75aadc22012-12-11 21:25:42 +0000636}
637
Matt Arsenault2aabb062013-06-18 23:37:58 +0000638bool AMDGPUDAGToDAGISel::isLocalLoad(const LoadSDNode *N) const {
Nick Lewyckyaad475b2014-04-15 07:22:52 +0000639 return checkType(N->getMemOperand()->getValue(), AMDGPUAS::LOCAL_ADDRESS);
Tom Stellard75aadc22012-12-11 21:25:42 +0000640}
641
Jan Vesely43b7b5b2016-04-07 19:23:11 +0000642bool AMDGPUDAGToDAGISel::isFlatLoad(const MemSDNode *N) const {
643 if (!N->readMem())
644 return false;
Matt Arsenault3f981402014-09-15 15:41:53 +0000645 return checkType(N->getMemOperand()->getValue(), AMDGPUAS::FLAT_ADDRESS);
646}
647
Matt Arsenault2aabb062013-06-18 23:37:58 +0000648bool AMDGPUDAGToDAGISel::isRegionLoad(const LoadSDNode *N) const {
Nick Lewyckyaad475b2014-04-15 07:22:52 +0000649 return checkType(N->getMemOperand()->getValue(), AMDGPUAS::REGION_ADDRESS);
Tom Stellard75aadc22012-12-11 21:25:42 +0000650}
651
Matt Arsenault2aabb062013-06-18 23:37:58 +0000652bool AMDGPUDAGToDAGISel::isCPLoad(const LoadSDNode *N) const {
Tom Stellard75aadc22012-12-11 21:25:42 +0000653 MachineMemOperand *MMO = N->getMemOperand();
Nick Lewyckyaad475b2014-04-15 07:22:52 +0000654 if (checkPrivateAddress(N->getMemOperand())) {
Tom Stellard75aadc22012-12-11 21:25:42 +0000655 if (MMO) {
Nick Lewyckyaad475b2014-04-15 07:22:52 +0000656 const PseudoSourceValue *PSV = MMO->getPseudoValue();
Alex Lorenze40c8a22015-08-11 23:09:45 +0000657 if (PSV && PSV->isConstantPool()) {
Tom Stellard75aadc22012-12-11 21:25:42 +0000658 return true;
659 }
660 }
661 }
662 return false;
663}
664
Matt Arsenault2aabb062013-06-18 23:37:58 +0000665bool AMDGPUDAGToDAGISel::isPrivateLoad(const LoadSDNode *N) const {
Nick Lewyckyaad475b2014-04-15 07:22:52 +0000666 if (checkPrivateAddress(N->getMemOperand())) {
Tom Stellard75aadc22012-12-11 21:25:42 +0000667 // Check to make sure we are not a constant pool load or a constant load
668 // that is marked as a private load
669 if (isCPLoad(N) || isConstantLoad(N, -1)) {
670 return false;
671 }
672 }
Matt Arsenault209a7b92014-04-18 07:40:20 +0000673
674 const Value *MemVal = N->getMemOperand()->getValue();
Matt Arsenault8226fc42016-03-02 23:00:21 +0000675 return !checkType(MemVal, AMDGPUAS::LOCAL_ADDRESS) &&
676 !checkType(MemVal, AMDGPUAS::GLOBAL_ADDRESS) &&
677 !checkType(MemVal, AMDGPUAS::FLAT_ADDRESS) &&
678 !checkType(MemVal, AMDGPUAS::REGION_ADDRESS) &&
679 !checkType(MemVal, AMDGPUAS::CONSTANT_ADDRESS) &&
680 !checkType(MemVal, AMDGPUAS::PARAM_D_ADDRESS) &&
681 !checkType(MemVal, AMDGPUAS::PARAM_I_ADDRESS);
Tom Stellard75aadc22012-12-11 21:25:42 +0000682}
683
Tom Stellardbc4497b2016-02-12 23:45:29 +0000684bool AMDGPUDAGToDAGISel::isUniformBr(const SDNode *N) const {
685 const BasicBlock *BB = FuncInfo->MBB->getBasicBlock();
Nicolai Haehnle05b127d2016-04-14 17:42:35 +0000686 const Instruction *Term = BB->getTerminator();
687 return Term->getMetadata("amdgpu.uniform") ||
688 Term->getMetadata("structurizecfg.uniform");
Tom Stellardbc4497b2016-02-12 23:45:29 +0000689}
690
Tom Stellard75aadc22012-12-11 21:25:42 +0000691const char *AMDGPUDAGToDAGISel::getPassName() const {
692 return "AMDGPU DAG->DAG Pattern Instruction Selection";
693}
694
Tom Stellard41fc7852013-07-23 01:48:42 +0000695//===----------------------------------------------------------------------===//
696// Complex Patterns
697//===----------------------------------------------------------------------===//
Tom Stellard75aadc22012-12-11 21:25:42 +0000698
Tom Stellard365366f2013-01-23 02:09:06 +0000699bool AMDGPUDAGToDAGISel::SelectGlobalValueConstantOffset(SDValue Addr,
Matt Arsenault209a7b92014-04-18 07:40:20 +0000700 SDValue& IntPtr) {
Tom Stellard365366f2013-01-23 02:09:06 +0000701 if (ConstantSDNode *Cst = dyn_cast<ConstantSDNode>(Addr)) {
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +0000702 IntPtr = CurDAG->getIntPtrConstant(Cst->getZExtValue() / 4, SDLoc(Addr),
703 true);
Tom Stellard365366f2013-01-23 02:09:06 +0000704 return true;
705 }
706 return false;
707}
708
709bool AMDGPUDAGToDAGISel::SelectGlobalValueVariableOffset(SDValue Addr,
710 SDValue& BaseReg, SDValue &Offset) {
Matt Arsenault209a7b92014-04-18 07:40:20 +0000711 if (!isa<ConstantSDNode>(Addr)) {
Tom Stellard365366f2013-01-23 02:09:06 +0000712 BaseReg = Addr;
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +0000713 Offset = CurDAG->getIntPtrConstant(0, SDLoc(Addr), true);
Tom Stellard365366f2013-01-23 02:09:06 +0000714 return true;
715 }
716 return false;
717}
718
Tom Stellard75aadc22012-12-11 21:25:42 +0000719bool AMDGPUDAGToDAGISel::SelectADDRVTX_READ(SDValue Addr, SDValue &Base,
720 SDValue &Offset) {
Matt Arsenault209a7b92014-04-18 07:40:20 +0000721 ConstantSDNode *IMMOffset;
Tom Stellard75aadc22012-12-11 21:25:42 +0000722
723 if (Addr.getOpcode() == ISD::ADD
724 && (IMMOffset = dyn_cast<ConstantSDNode>(Addr.getOperand(1)))
725 && isInt<16>(IMMOffset->getZExtValue())) {
726
727 Base = Addr.getOperand(0);
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +0000728 Offset = CurDAG->getTargetConstant(IMMOffset->getZExtValue(), SDLoc(Addr),
729 MVT::i32);
Tom Stellard75aadc22012-12-11 21:25:42 +0000730 return true;
731 // If the pointer address is constant, we can move it to the offset field.
732 } else if ((IMMOffset = dyn_cast<ConstantSDNode>(Addr))
733 && isInt<16>(IMMOffset->getZExtValue())) {
734 Base = CurDAG->getCopyFromReg(CurDAG->getEntryNode(),
Andrew Trickef9de2a2013-05-25 02:42:55 +0000735 SDLoc(CurDAG->getEntryNode()),
Tom Stellard75aadc22012-12-11 21:25:42 +0000736 AMDGPU::ZERO, MVT::i32);
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +0000737 Offset = CurDAG->getTargetConstant(IMMOffset->getZExtValue(), SDLoc(Addr),
738 MVT::i32);
Tom Stellard75aadc22012-12-11 21:25:42 +0000739 return true;
740 }
741
742 // Default case, no offset
743 Base = Addr;
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +0000744 Offset = CurDAG->getTargetConstant(0, SDLoc(Addr), MVT::i32);
Tom Stellard75aadc22012-12-11 21:25:42 +0000745 return true;
746}
747
Tom Stellardf3b2a1e2013-02-06 17:32:29 +0000748bool AMDGPUDAGToDAGISel::SelectADDRIndirect(SDValue Addr, SDValue &Base,
749 SDValue &Offset) {
750 ConstantSDNode *C;
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +0000751 SDLoc DL(Addr);
Tom Stellardf3b2a1e2013-02-06 17:32:29 +0000752
753 if ((C = dyn_cast<ConstantSDNode>(Addr))) {
754 Base = CurDAG->getRegister(AMDGPU::INDIRECT_BASE_ADDR, MVT::i32);
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +0000755 Offset = CurDAG->getTargetConstant(C->getZExtValue(), DL, MVT::i32);
Tom Stellardf3b2a1e2013-02-06 17:32:29 +0000756 } else if ((Addr.getOpcode() == ISD::ADD || Addr.getOpcode() == ISD::OR) &&
757 (C = dyn_cast<ConstantSDNode>(Addr.getOperand(1)))) {
758 Base = Addr.getOperand(0);
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +0000759 Offset = CurDAG->getTargetConstant(C->getZExtValue(), DL, MVT::i32);
Tom Stellardf3b2a1e2013-02-06 17:32:29 +0000760 } else {
761 Base = Addr;
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +0000762 Offset = CurDAG->getTargetConstant(0, DL, MVT::i32);
Tom Stellardf3b2a1e2013-02-06 17:32:29 +0000763 }
764
765 return true;
766}
Christian Konigd910b7d2013-02-26 17:52:16 +0000767
Justin Bogner95927c02016-05-12 21:03:32 +0000768void AMDGPUDAGToDAGISel::SelectADD_SUB_I64(SDNode *N) {
Matt Arsenault9fa3f932014-06-23 18:00:34 +0000769 SDLoc DL(N);
770 SDValue LHS = N->getOperand(0);
771 SDValue RHS = N->getOperand(1);
772
Matt Arsenaultb8b51532014-06-23 18:00:38 +0000773 bool IsAdd = (N->getOpcode() == ISD::ADD);
774
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +0000775 SDValue Sub0 = CurDAG->getTargetConstant(AMDGPU::sub0, DL, MVT::i32);
776 SDValue Sub1 = CurDAG->getTargetConstant(AMDGPU::sub1, DL, MVT::i32);
Matt Arsenault9fa3f932014-06-23 18:00:34 +0000777
778 SDNode *Lo0 = CurDAG->getMachineNode(TargetOpcode::EXTRACT_SUBREG,
779 DL, MVT::i32, LHS, Sub0);
780 SDNode *Hi0 = CurDAG->getMachineNode(TargetOpcode::EXTRACT_SUBREG,
781 DL, MVT::i32, LHS, Sub1);
782
783 SDNode *Lo1 = CurDAG->getMachineNode(TargetOpcode::EXTRACT_SUBREG,
784 DL, MVT::i32, RHS, Sub0);
785 SDNode *Hi1 = CurDAG->getMachineNode(TargetOpcode::EXTRACT_SUBREG,
786 DL, MVT::i32, RHS, Sub1);
787
788 SDVTList VTList = CurDAG->getVTList(MVT::i32, MVT::Glue);
Matt Arsenault9fa3f932014-06-23 18:00:34 +0000789 SDValue AddLoArgs[] = { SDValue(Lo0, 0), SDValue(Lo1, 0) };
790
Matt Arsenaultb8b51532014-06-23 18:00:38 +0000791
Tom Stellard80942a12014-09-05 14:07:59 +0000792 unsigned Opc = IsAdd ? AMDGPU::S_ADD_U32 : AMDGPU::S_SUB_U32;
Matt Arsenaultb8b51532014-06-23 18:00:38 +0000793 unsigned CarryOpc = IsAdd ? AMDGPU::S_ADDC_U32 : AMDGPU::S_SUBB_U32;
794
Matt Arsenaultb8b51532014-06-23 18:00:38 +0000795 SDNode *AddLo = CurDAG->getMachineNode( Opc, DL, VTList, AddLoArgs);
796 SDValue Carry(AddLo, 1);
797 SDNode *AddHi
798 = CurDAG->getMachineNode(CarryOpc, DL, MVT::i32,
799 SDValue(Hi0, 0), SDValue(Hi1, 0), Carry);
Matt Arsenault9fa3f932014-06-23 18:00:34 +0000800
801 SDValue Args[5] = {
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +0000802 CurDAG->getTargetConstant(AMDGPU::SReg_64RegClassID, DL, MVT::i32),
Matt Arsenault9fa3f932014-06-23 18:00:34 +0000803 SDValue(AddLo,0),
804 Sub0,
805 SDValue(AddHi,0),
806 Sub1,
807 };
Justin Bogner95927c02016-05-12 21:03:32 +0000808 CurDAG->SelectNodeTo(N, AMDGPU::REG_SEQUENCE, MVT::i64, Args);
Matt Arsenault9fa3f932014-06-23 18:00:34 +0000809}
810
Matt Arsenault044f1d12015-02-14 04:24:28 +0000811// We need to handle this here because tablegen doesn't support matching
812// instructions with multiple outputs.
Justin Bogner95927c02016-05-12 21:03:32 +0000813void AMDGPUDAGToDAGISel::SelectDIV_SCALE(SDNode *N) {
Matt Arsenaultf2b0aeb2014-06-23 18:28:28 +0000814 SDLoc SL(N);
815 EVT VT = N->getValueType(0);
816
817 assert(VT == MVT::f32 || VT == MVT::f64);
818
819 unsigned Opc
820 = (VT == MVT::f64) ? AMDGPU::V_DIV_SCALE_F64 : AMDGPU::V_DIV_SCALE_F32;
821
NAKAMURA Takumi84965032015-09-22 11:14:12 +0000822 // src0_modifiers, src0, src1_modifiers, src1, src2_modifiers, src2, clamp,
823 // omod
Matt Arsenault044f1d12015-02-14 04:24:28 +0000824 SDValue Ops[8];
Matt Arsenaultf2b0aeb2014-06-23 18:28:28 +0000825
Matt Arsenault044f1d12015-02-14 04:24:28 +0000826 SelectVOP3Mods0(N->getOperand(0), Ops[1], Ops[0], Ops[6], Ops[7]);
827 SelectVOP3Mods(N->getOperand(1), Ops[3], Ops[2]);
828 SelectVOP3Mods(N->getOperand(2), Ops[5], Ops[4]);
Justin Bogner95927c02016-05-12 21:03:32 +0000829 CurDAG->SelectNodeTo(N, Opc, VT, MVT::i1, Ops);
Matt Arsenaultf2b0aeb2014-06-23 18:28:28 +0000830}
831
Tom Stellard85e8b6d2014-08-22 18:49:33 +0000832bool AMDGPUDAGToDAGISel::isDSOffsetLegal(const SDValue &Base, unsigned Offset,
833 unsigned OffsetBits) const {
Tom Stellard85e8b6d2014-08-22 18:49:33 +0000834 if ((OffsetBits == 16 && !isUInt<16>(Offset)) ||
835 (OffsetBits == 8 && !isUInt<8>(Offset)))
836 return false;
837
Matt Arsenault706f9302015-07-06 16:01:58 +0000838 if (Subtarget->getGeneration() >= AMDGPUSubtarget::SEA_ISLANDS ||
839 Subtarget->unsafeDSOffsetFoldingEnabled())
Tom Stellard85e8b6d2014-08-22 18:49:33 +0000840 return true;
841
842 // On Southern Islands instruction with a negative base value and an offset
843 // don't seem to work.
844 return CurDAG->SignBitIsZero(Base);
845}
846
847bool AMDGPUDAGToDAGISel::SelectDS1Addr1Offset(SDValue Addr, SDValue &Base,
848 SDValue &Offset) const {
Tom Stellard92b24f32016-04-29 14:34:26 +0000849 SDLoc DL(Addr);
Tom Stellard85e8b6d2014-08-22 18:49:33 +0000850 if (CurDAG->isBaseWithConstantOffset(Addr)) {
851 SDValue N0 = Addr.getOperand(0);
852 SDValue N1 = Addr.getOperand(1);
853 ConstantSDNode *C1 = cast<ConstantSDNode>(N1);
854 if (isDSOffsetLegal(N0, C1->getSExtValue(), 16)) {
855 // (add n0, c0)
856 Base = N0;
Tom Stellard92b24f32016-04-29 14:34:26 +0000857 Offset = CurDAG->getTargetConstant(C1->getZExtValue(), DL, MVT::i16);
Tom Stellard85e8b6d2014-08-22 18:49:33 +0000858 return true;
859 }
Matt Arsenault966a94f2015-09-08 19:34:22 +0000860 } else if (Addr.getOpcode() == ISD::SUB) {
861 // sub C, x -> add (sub 0, x), C
862 if (const ConstantSDNode *C = dyn_cast<ConstantSDNode>(Addr.getOperand(0))) {
863 int64_t ByteOffset = C->getSExtValue();
864 if (isUInt<16>(ByteOffset)) {
Matt Arsenault966a94f2015-09-08 19:34:22 +0000865 SDValue Zero = CurDAG->getTargetConstant(0, DL, MVT::i32);
Tom Stellard85e8b6d2014-08-22 18:49:33 +0000866
Matt Arsenault966a94f2015-09-08 19:34:22 +0000867 // XXX - This is kind of hacky. Create a dummy sub node so we can check
868 // the known bits in isDSOffsetLegal. We need to emit the selected node
869 // here, so this is thrown away.
870 SDValue Sub = CurDAG->getNode(ISD::SUB, DL, MVT::i32,
871 Zero, Addr.getOperand(1));
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +0000872
Matt Arsenault966a94f2015-09-08 19:34:22 +0000873 if (isDSOffsetLegal(Sub, ByteOffset, 16)) {
874 MachineSDNode *MachineSub
875 = CurDAG->getMachineNode(AMDGPU::V_SUB_I32_e32, DL, MVT::i32,
876 Zero, Addr.getOperand(1));
877
878 Base = SDValue(MachineSub, 0);
879 Offset = Addr.getOperand(0);
880 return true;
881 }
882 }
883 }
884 } else if (const ConstantSDNode *CAddr = dyn_cast<ConstantSDNode>(Addr)) {
885 // If we have a constant address, prefer to put the constant into the
886 // offset. This can save moves to load the constant address since multiple
887 // operations can share the zero base address register, and enables merging
888 // into read2 / write2 instructions.
889
890 SDLoc DL(Addr);
891
Matt Arsenaulte775f5f2014-10-14 17:21:19 +0000892 if (isUInt<16>(CAddr->getZExtValue())) {
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +0000893 SDValue Zero = CurDAG->getTargetConstant(0, DL, MVT::i32);
Tom Stellardc8d79202014-10-15 21:08:59 +0000894 MachineSDNode *MovZero = CurDAG->getMachineNode(AMDGPU::V_MOV_B32_e32,
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +0000895 DL, MVT::i32, Zero);
Tom Stellardc8d79202014-10-15 21:08:59 +0000896 Base = SDValue(MovZero, 0);
Matt Arsenaulte775f5f2014-10-14 17:21:19 +0000897 Offset = Addr;
898 return true;
899 }
900 }
901
Tom Stellard85e8b6d2014-08-22 18:49:33 +0000902 // default case
903 Base = Addr;
Matt Arsenault966a94f2015-09-08 19:34:22 +0000904 Offset = CurDAG->getTargetConstant(0, SDLoc(Addr), MVT::i16);
Tom Stellard85e8b6d2014-08-22 18:49:33 +0000905 return true;
906}
907
Matt Arsenault966a94f2015-09-08 19:34:22 +0000908// TODO: If offset is too big, put low 16-bit into offset.
Tom Stellardf3fc5552014-08-22 18:49:35 +0000909bool AMDGPUDAGToDAGISel::SelectDS64Bit4ByteAligned(SDValue Addr, SDValue &Base,
910 SDValue &Offset0,
911 SDValue &Offset1) const {
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +0000912 SDLoc DL(Addr);
913
Tom Stellardf3fc5552014-08-22 18:49:35 +0000914 if (CurDAG->isBaseWithConstantOffset(Addr)) {
915 SDValue N0 = Addr.getOperand(0);
916 SDValue N1 = Addr.getOperand(1);
917 ConstantSDNode *C1 = cast<ConstantSDNode>(N1);
918 unsigned DWordOffset0 = C1->getZExtValue() / 4;
919 unsigned DWordOffset1 = DWordOffset0 + 1;
920 // (add n0, c0)
921 if (isDSOffsetLegal(N0, DWordOffset1, 8)) {
922 Base = N0;
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +0000923 Offset0 = CurDAG->getTargetConstant(DWordOffset0, DL, MVT::i8);
924 Offset1 = CurDAG->getTargetConstant(DWordOffset1, DL, MVT::i8);
Tom Stellardf3fc5552014-08-22 18:49:35 +0000925 return true;
926 }
Matt Arsenault966a94f2015-09-08 19:34:22 +0000927 } else if (Addr.getOpcode() == ISD::SUB) {
928 // sub C, x -> add (sub 0, x), C
929 if (const ConstantSDNode *C = dyn_cast<ConstantSDNode>(Addr.getOperand(0))) {
930 unsigned DWordOffset0 = C->getZExtValue() / 4;
931 unsigned DWordOffset1 = DWordOffset0 + 1;
Tom Stellardf3fc5552014-08-22 18:49:35 +0000932
Matt Arsenault966a94f2015-09-08 19:34:22 +0000933 if (isUInt<8>(DWordOffset0)) {
934 SDLoc DL(Addr);
935 SDValue Zero = CurDAG->getTargetConstant(0, DL, MVT::i32);
936
937 // XXX - This is kind of hacky. Create a dummy sub node so we can check
938 // the known bits in isDSOffsetLegal. We need to emit the selected node
939 // here, so this is thrown away.
940 SDValue Sub = CurDAG->getNode(ISD::SUB, DL, MVT::i32,
941 Zero, Addr.getOperand(1));
942
943 if (isDSOffsetLegal(Sub, DWordOffset1, 8)) {
944 MachineSDNode *MachineSub
945 = CurDAG->getMachineNode(AMDGPU::V_SUB_I32_e32, DL, MVT::i32,
946 Zero, Addr.getOperand(1));
947
948 Base = SDValue(MachineSub, 0);
949 Offset0 = CurDAG->getTargetConstant(DWordOffset0, DL, MVT::i8);
950 Offset1 = CurDAG->getTargetConstant(DWordOffset1, DL, MVT::i8);
951 return true;
952 }
953 }
954 }
955 } else if (const ConstantSDNode *CAddr = dyn_cast<ConstantSDNode>(Addr)) {
Matt Arsenault1a74aff2014-10-15 18:06:43 +0000956 unsigned DWordOffset0 = CAddr->getZExtValue() / 4;
957 unsigned DWordOffset1 = DWordOffset0 + 1;
958 assert(4 * DWordOffset0 == CAddr->getZExtValue());
959
960 if (isUInt<8>(DWordOffset0) && isUInt<8>(DWordOffset1)) {
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +0000961 SDValue Zero = CurDAG->getTargetConstant(0, DL, MVT::i32);
Matt Arsenault1a74aff2014-10-15 18:06:43 +0000962 MachineSDNode *MovZero
963 = CurDAG->getMachineNode(AMDGPU::V_MOV_B32_e32,
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +0000964 DL, MVT::i32, Zero);
Matt Arsenault1a74aff2014-10-15 18:06:43 +0000965 Base = SDValue(MovZero, 0);
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +0000966 Offset0 = CurDAG->getTargetConstant(DWordOffset0, DL, MVT::i8);
967 Offset1 = CurDAG->getTargetConstant(DWordOffset1, DL, MVT::i8);
Matt Arsenault1a74aff2014-10-15 18:06:43 +0000968 return true;
969 }
970 }
971
Tom Stellardf3fc5552014-08-22 18:49:35 +0000972 // default case
973 Base = Addr;
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +0000974 Offset0 = CurDAG->getTargetConstant(0, DL, MVT::i8);
975 Offset1 = CurDAG->getTargetConstant(1, DL, MVT::i8);
Tom Stellardf3fc5552014-08-22 18:49:35 +0000976 return true;
977}
978
Tom Stellardb02094e2014-07-21 15:45:01 +0000979static bool isLegalMUBUFImmOffset(const ConstantSDNode *Imm) {
980 return isUInt<12>(Imm->getZExtValue());
981}
982
Changpeng Fangb41574a2015-12-22 20:55:23 +0000983bool AMDGPUDAGToDAGISel::SelectMUBUF(SDValue Addr, SDValue &Ptr,
Tom Stellard155bbb72014-08-11 22:18:17 +0000984 SDValue &VAddr, SDValue &SOffset,
985 SDValue &Offset, SDValue &Offen,
986 SDValue &Idxen, SDValue &Addr64,
987 SDValue &GLC, SDValue &SLC,
988 SDValue &TFE) const {
Changpeng Fangb41574a2015-12-22 20:55:23 +0000989 // Subtarget prefers to use flat instruction
990 if (Subtarget->useFlatForGlobal())
991 return false;
992
Tom Stellardb02c2682014-06-24 23:33:07 +0000993 SDLoc DL(Addr);
994
Jan Vesely43b7b5b2016-04-07 19:23:11 +0000995 if (!GLC.getNode())
996 GLC = CurDAG->getTargetConstant(0, DL, MVT::i1);
997 if (!SLC.getNode())
998 SLC = CurDAG->getTargetConstant(0, DL, MVT::i1);
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +0000999 TFE = CurDAG->getTargetConstant(0, DL, MVT::i1);
Tom Stellard155bbb72014-08-11 22:18:17 +00001000
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +00001001 Idxen = CurDAG->getTargetConstant(0, DL, MVT::i1);
1002 Offen = CurDAG->getTargetConstant(0, DL, MVT::i1);
1003 Addr64 = CurDAG->getTargetConstant(0, DL, MVT::i1);
1004 SOffset = CurDAG->getTargetConstant(0, DL, MVT::i32);
Tom Stellard155bbb72014-08-11 22:18:17 +00001005
Tom Stellardb02c2682014-06-24 23:33:07 +00001006 if (CurDAG->isBaseWithConstantOffset(Addr)) {
1007 SDValue N0 = Addr.getOperand(0);
1008 SDValue N1 = Addr.getOperand(1);
1009 ConstantSDNode *C1 = cast<ConstantSDNode>(N1);
1010
Tom Stellard94b72312015-02-11 00:34:35 +00001011 if (N0.getOpcode() == ISD::ADD) {
1012 // (add (add N2, N3), C1) -> addr64
1013 SDValue N2 = N0.getOperand(0);
1014 SDValue N3 = N0.getOperand(1);
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +00001015 Addr64 = CurDAG->getTargetConstant(1, DL, MVT::i1);
Tom Stellard94b72312015-02-11 00:34:35 +00001016 Ptr = N2;
1017 VAddr = N3;
1018 } else {
Tom Stellardb02c2682014-06-24 23:33:07 +00001019
Tom Stellard155bbb72014-08-11 22:18:17 +00001020 // (add N0, C1) -> offset
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +00001021 VAddr = CurDAG->getTargetConstant(0, DL, MVT::i32);
Tom Stellard155bbb72014-08-11 22:18:17 +00001022 Ptr = N0;
Tom Stellard94b72312015-02-11 00:34:35 +00001023 }
1024
1025 if (isLegalMUBUFImmOffset(C1)) {
Matt Arsenault88701812016-06-09 23:42:48 +00001026 Offset = CurDAG->getTargetConstant(C1->getZExtValue(), DL, MVT::i16);
1027 return true;
1028 }
1029
1030 if (isUInt<32>(C1->getZExtValue())) {
Tom Stellard94b72312015-02-11 00:34:35 +00001031 // Illegal offset, store it in soffset.
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +00001032 Offset = CurDAG->getTargetConstant(0, DL, MVT::i16);
Tom Stellard94b72312015-02-11 00:34:35 +00001033 SOffset = SDValue(CurDAG->getMachineNode(AMDGPU::S_MOV_B32, DL, MVT::i32,
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +00001034 CurDAG->getTargetConstant(C1->getZExtValue(), DL, MVT::i32)),
1035 0);
Changpeng Fangb41574a2015-12-22 20:55:23 +00001036 return true;
Tom Stellardb02c2682014-06-24 23:33:07 +00001037 }
1038 }
Tom Stellard94b72312015-02-11 00:34:35 +00001039
Tom Stellardb02c2682014-06-24 23:33:07 +00001040 if (Addr.getOpcode() == ISD::ADD) {
Tom Stellard155bbb72014-08-11 22:18:17 +00001041 // (add N0, N1) -> addr64
Tom Stellardb02c2682014-06-24 23:33:07 +00001042 SDValue N0 = Addr.getOperand(0);
1043 SDValue N1 = Addr.getOperand(1);
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +00001044 Addr64 = CurDAG->getTargetConstant(1, DL, MVT::i1);
Tom Stellard155bbb72014-08-11 22:18:17 +00001045 Ptr = N0;
1046 VAddr = N1;
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +00001047 Offset = CurDAG->getTargetConstant(0, DL, MVT::i16);
Changpeng Fangb41574a2015-12-22 20:55:23 +00001048 return true;
Tom Stellardb02c2682014-06-24 23:33:07 +00001049 }
1050
Tom Stellard155bbb72014-08-11 22:18:17 +00001051 // default case -> offset
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +00001052 VAddr = CurDAG->getTargetConstant(0, DL, MVT::i32);
Tom Stellard155bbb72014-08-11 22:18:17 +00001053 Ptr = Addr;
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +00001054 Offset = CurDAG->getTargetConstant(0, DL, MVT::i16);
Changpeng Fangb41574a2015-12-22 20:55:23 +00001055
1056 return true;
Tom Stellard155bbb72014-08-11 22:18:17 +00001057}
1058
1059bool AMDGPUDAGToDAGISel::SelectMUBUFAddr64(SDValue Addr, SDValue &SRsrc,
Tom Stellardc53861a2015-02-11 00:34:32 +00001060 SDValue &VAddr, SDValue &SOffset,
Tom Stellard1f9939f2015-02-27 14:59:41 +00001061 SDValue &Offset, SDValue &GLC,
1062 SDValue &SLC, SDValue &TFE) const {
1063 SDValue Ptr, Offen, Idxen, Addr64;
Tom Stellard155bbb72014-08-11 22:18:17 +00001064
Tom Stellard70580f82015-07-20 14:28:41 +00001065 // addr64 bit was removed for volcanic islands.
1066 if (Subtarget->getGeneration() >= AMDGPUSubtarget::VOLCANIC_ISLANDS)
1067 return false;
1068
Changpeng Fangb41574a2015-12-22 20:55:23 +00001069 if (!SelectMUBUF(Addr, Ptr, VAddr, SOffset, Offset, Offen, Idxen, Addr64,
1070 GLC, SLC, TFE))
1071 return false;
Tom Stellard155bbb72014-08-11 22:18:17 +00001072
1073 ConstantSDNode *C = cast<ConstantSDNode>(Addr64);
1074 if (C->getSExtValue()) {
1075 SDLoc DL(Addr);
Matt Arsenault485defe2014-11-05 19:01:17 +00001076
1077 const SITargetLowering& Lowering =
1078 *static_cast<const SITargetLowering*>(getTargetLowering());
1079
1080 SRsrc = SDValue(Lowering.wrapAddr64Rsrc(*CurDAG, DL, Ptr), 0);
Tom Stellard155bbb72014-08-11 22:18:17 +00001081 return true;
1082 }
Matt Arsenault485defe2014-11-05 19:01:17 +00001083
Tom Stellard155bbb72014-08-11 22:18:17 +00001084 return false;
1085}
1086
Tom Stellard7980fc82014-09-25 18:30:26 +00001087bool AMDGPUDAGToDAGISel::SelectMUBUFAddr64(SDValue Addr, SDValue &SRsrc,
Tom Stellardc53861a2015-02-11 00:34:32 +00001088 SDValue &VAddr, SDValue &SOffset,
NAKAMURA Takumi0a7d0ad2015-09-22 11:15:07 +00001089 SDValue &Offset,
1090 SDValue &SLC) const {
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +00001091 SLC = CurDAG->getTargetConstant(0, SDLoc(Addr), MVT::i1);
Tom Stellard1f9939f2015-02-27 14:59:41 +00001092 SDValue GLC, TFE;
Tom Stellard7980fc82014-09-25 18:30:26 +00001093
Tom Stellard1f9939f2015-02-27 14:59:41 +00001094 return SelectMUBUFAddr64(Addr, SRsrc, VAddr, SOffset, Offset, GLC, SLC, TFE);
Tom Stellard7980fc82014-09-25 18:30:26 +00001095}
1096
Tom Stellardb02094e2014-07-21 15:45:01 +00001097bool AMDGPUDAGToDAGISel::SelectMUBUFScratch(SDValue Addr, SDValue &Rsrc,
1098 SDValue &VAddr, SDValue &SOffset,
1099 SDValue &ImmOffset) const {
1100
1101 SDLoc DL(Addr);
1102 MachineFunction &MF = CurDAG->getMachineFunction();
Matt Arsenault0e3d3892015-11-30 21:15:53 +00001103 const SIMachineFunctionInfo *Info = MF.getInfo<SIMachineFunctionInfo>();
Tom Stellardb02094e2014-07-21 15:45:01 +00001104
Matt Arsenault0e3d3892015-11-30 21:15:53 +00001105 Rsrc = CurDAG->getRegister(Info->getScratchRSrcReg(), MVT::v4i32);
Matt Arsenault26f8f3d2015-11-30 21:16:03 +00001106 SOffset = CurDAG->getRegister(Info->getScratchWaveOffsetReg(), MVT::i32);
Tom Stellardb02094e2014-07-21 15:45:01 +00001107
1108 // (add n0, c1)
1109 if (CurDAG->isBaseWithConstantOffset(Addr)) {
Tom Stellard78655fc2015-07-16 19:40:09 +00001110 SDValue N0 = Addr.getOperand(0);
Tom Stellardb02094e2014-07-21 15:45:01 +00001111 SDValue N1 = Addr.getOperand(1);
Matt Arsenaultcd099612016-02-24 04:55:29 +00001112
Tom Stellard78655fc2015-07-16 19:40:09 +00001113 // Offsets in vaddr must be positive.
Matt Arsenaultcd099612016-02-24 04:55:29 +00001114 ConstantSDNode *C1 = cast<ConstantSDNode>(N1);
Matt Arsenaultcb38a6b2016-03-21 18:02:18 +00001115 if (isLegalMUBUFImmOffset(C1)) {
Matt Arsenaultcd099612016-02-24 04:55:29 +00001116 VAddr = N0;
1117 ImmOffset = CurDAG->getTargetConstant(C1->getZExtValue(), DL, MVT::i16);
1118 return true;
Tom Stellardb02094e2014-07-21 15:45:01 +00001119 }
1120 }
1121
Tom Stellardb02094e2014-07-21 15:45:01 +00001122 // (node)
1123 VAddr = Addr;
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +00001124 ImmOffset = CurDAG->getTargetConstant(0, DL, MVT::i16);
Tom Stellardb02094e2014-07-21 15:45:01 +00001125 return true;
1126}
1127
Tom Stellard155bbb72014-08-11 22:18:17 +00001128bool AMDGPUDAGToDAGISel::SelectMUBUFOffset(SDValue Addr, SDValue &SRsrc,
1129 SDValue &SOffset, SDValue &Offset,
1130 SDValue &GLC, SDValue &SLC,
1131 SDValue &TFE) const {
1132 SDValue Ptr, VAddr, Offen, Idxen, Addr64;
Tom Stellard794c8c02014-12-02 17:05:41 +00001133 const SIInstrInfo *TII =
Eric Christopher7792e322015-01-30 23:24:40 +00001134 static_cast<const SIInstrInfo *>(Subtarget->getInstrInfo());
Tom Stellardb02094e2014-07-21 15:45:01 +00001135
Changpeng Fangb41574a2015-12-22 20:55:23 +00001136 if (!SelectMUBUF(Addr, Ptr, VAddr, SOffset, Offset, Offen, Idxen, Addr64,
1137 GLC, SLC, TFE))
1138 return false;
Tom Stellardb02094e2014-07-21 15:45:01 +00001139
Tom Stellard155bbb72014-08-11 22:18:17 +00001140 if (!cast<ConstantSDNode>(Offen)->getSExtValue() &&
1141 !cast<ConstantSDNode>(Idxen)->getSExtValue() &&
1142 !cast<ConstantSDNode>(Addr64)->getSExtValue()) {
Tom Stellard794c8c02014-12-02 17:05:41 +00001143 uint64_t Rsrc = TII->getDefaultRsrcDataFormat() |
Tom Stellard155bbb72014-08-11 22:18:17 +00001144 APInt::getAllOnesValue(32).getZExtValue(); // Size
1145 SDLoc DL(Addr);
Matt Arsenaultf3cd4512014-11-05 19:01:19 +00001146
1147 const SITargetLowering& Lowering =
1148 *static_cast<const SITargetLowering*>(getTargetLowering());
1149
1150 SRsrc = SDValue(Lowering.buildRSRC(*CurDAG, DL, Ptr, 0, Rsrc), 0);
Tom Stellard155bbb72014-08-11 22:18:17 +00001151 return true;
1152 }
1153 return false;
Tom Stellardb02094e2014-07-21 15:45:01 +00001154}
1155
Tom Stellard7980fc82014-09-25 18:30:26 +00001156bool AMDGPUDAGToDAGISel::SelectMUBUFOffset(SDValue Addr, SDValue &SRsrc,
Jan Vesely43b7b5b2016-04-07 19:23:11 +00001157 SDValue &Soffset, SDValue &Offset
1158 ) const {
1159 SDValue GLC, SLC, TFE;
1160
1161 return SelectMUBUFOffset(Addr, SRsrc, Soffset, Offset, GLC, SLC, TFE);
1162}
1163bool AMDGPUDAGToDAGISel::SelectMUBUFOffset(SDValue Addr, SDValue &SRsrc,
Tom Stellard7980fc82014-09-25 18:30:26 +00001164 SDValue &Soffset, SDValue &Offset,
Matt Arsenault88701812016-06-09 23:42:48 +00001165 SDValue &SLC) const {
1166 SDValue GLC, TFE;
Tom Stellard7980fc82014-09-25 18:30:26 +00001167
1168 return SelectMUBUFOffset(Addr, SRsrc, Soffset, Offset, GLC, SLC, TFE);
1169}
1170
Nicolai Haehnle3003ba02016-03-18 16:24:20 +00001171void AMDGPUDAGToDAGISel::SelectMUBUFConstant(SDValue Constant,
1172 SDValue &SOffset,
1173 SDValue &ImmOffset) const {
1174 SDLoc DL(Constant);
1175 uint32_t Imm = cast<ConstantSDNode>(Constant)->getZExtValue();
1176 uint32_t Overflow = 0;
1177
1178 if (Imm >= 4096) {
1179 if (Imm <= 4095 + 64) {
1180 // Use an SOffset inline constant for 1..64
1181 Overflow = Imm - 4095;
1182 Imm = 4095;
1183 } else {
1184 // Try to keep the same value in SOffset for adjacent loads, so that
1185 // the corresponding register contents can be re-used.
1186 //
1187 // Load values with all low-bits set into SOffset, so that a larger
1188 // range of values can be covered using s_movk_i32
1189 uint32_t High = (Imm + 1) & ~4095;
1190 uint32_t Low = (Imm + 1) & 4095;
1191 Imm = Low;
1192 Overflow = High - 1;
1193 }
1194 }
1195
1196 ImmOffset = CurDAG->getTargetConstant(Imm, DL, MVT::i16);
1197
1198 if (Overflow <= 64)
1199 SOffset = CurDAG->getTargetConstant(Overflow, DL, MVT::i32);
1200 else
1201 SOffset = SDValue(CurDAG->getMachineNode(AMDGPU::S_MOV_B32, DL, MVT::i32,
1202 CurDAG->getTargetConstant(Overflow, DL, MVT::i32)),
1203 0);
1204}
1205
1206bool AMDGPUDAGToDAGISel::SelectMUBUFIntrinsicOffset(SDValue Offset,
1207 SDValue &SOffset,
1208 SDValue &ImmOffset) const {
1209 SDLoc DL(Offset);
1210
1211 if (!isa<ConstantSDNode>(Offset))
1212 return false;
1213
1214 SelectMUBUFConstant(Offset, SOffset, ImmOffset);
1215
1216 return true;
1217}
1218
1219bool AMDGPUDAGToDAGISel::SelectMUBUFIntrinsicVOffset(SDValue Offset,
1220 SDValue &SOffset,
1221 SDValue &ImmOffset,
1222 SDValue &VOffset) const {
1223 SDLoc DL(Offset);
1224
1225 // Don't generate an unnecessary voffset for constant offsets.
1226 if (isa<ConstantSDNode>(Offset))
1227 return false;
1228
1229 if (CurDAG->isBaseWithConstantOffset(Offset)) {
1230 SDValue N0 = Offset.getOperand(0);
1231 SDValue N1 = Offset.getOperand(1);
1232 SelectMUBUFConstant(N1, SOffset, ImmOffset);
1233 VOffset = N0;
1234 } else {
1235 SOffset = CurDAG->getTargetConstant(0, DL, MVT::i32);
1236 ImmOffset = CurDAG->getTargetConstant(0, DL, MVT::i16);
1237 VOffset = Offset;
1238 }
1239
1240 return true;
1241}
1242
Matt Arsenault7757c592016-06-09 23:42:54 +00001243bool AMDGPUDAGToDAGISel::SelectFlat(SDValue Addr,
1244 SDValue &VAddr,
1245 SDValue &SLC,
1246 SDValue &TFE) const {
1247 VAddr = Addr;
1248 TFE = SLC = CurDAG->getTargetConstant(0, SDLoc(), MVT::i1);
1249 return true;
1250}
1251
Tom Stellarddee26a22015-08-06 19:28:30 +00001252///
1253/// \param EncodedOffset This is the immediate value that will be encoded
1254/// directly into the instruction. On SI/CI the \p EncodedOffset
1255/// will be in units of dwords and on VI+ it will be units of bytes.
1256static bool isLegalSMRDImmOffset(const AMDGPUSubtarget *ST,
1257 int64_t EncodedOffset) {
1258 return ST->getGeneration() < AMDGPUSubtarget::VOLCANIC_ISLANDS ?
1259 isUInt<8>(EncodedOffset) : isUInt<20>(EncodedOffset);
1260}
1261
1262bool AMDGPUDAGToDAGISel::SelectSMRDOffset(SDValue ByteOffsetNode,
1263 SDValue &Offset, bool &Imm) const {
1264
1265 // FIXME: Handle non-constant offsets.
1266 ConstantSDNode *C = dyn_cast<ConstantSDNode>(ByteOffsetNode);
1267 if (!C)
1268 return false;
1269
1270 SDLoc SL(ByteOffsetNode);
1271 AMDGPUSubtarget::Generation Gen = Subtarget->getGeneration();
1272 int64_t ByteOffset = C->getSExtValue();
1273 int64_t EncodedOffset = Gen < AMDGPUSubtarget::VOLCANIC_ISLANDS ?
1274 ByteOffset >> 2 : ByteOffset;
1275
1276 if (isLegalSMRDImmOffset(Subtarget, EncodedOffset)) {
1277 Offset = CurDAG->getTargetConstant(EncodedOffset, SL, MVT::i32);
1278 Imm = true;
1279 return true;
1280 }
1281
Tom Stellard217361c2015-08-06 19:28:38 +00001282 if (!isUInt<32>(EncodedOffset) || !isUInt<32>(ByteOffset))
1283 return false;
1284
1285 if (Gen == AMDGPUSubtarget::SEA_ISLANDS && isUInt<32>(EncodedOffset)) {
1286 // 32-bit Immediates are supported on Sea Islands.
1287 Offset = CurDAG->getTargetConstant(EncodedOffset, SL, MVT::i32);
1288 } else {
Tom Stellarddee26a22015-08-06 19:28:30 +00001289 SDValue C32Bit = CurDAG->getTargetConstant(ByteOffset, SL, MVT::i32);
1290 Offset = SDValue(CurDAG->getMachineNode(AMDGPU::S_MOV_B32, SL, MVT::i32,
1291 C32Bit), 0);
Tom Stellarddee26a22015-08-06 19:28:30 +00001292 }
Tom Stellard217361c2015-08-06 19:28:38 +00001293 Imm = false;
1294 return true;
Tom Stellarddee26a22015-08-06 19:28:30 +00001295}
1296
1297bool AMDGPUDAGToDAGISel::SelectSMRD(SDValue Addr, SDValue &SBase,
1298 SDValue &Offset, bool &Imm) const {
1299
1300 SDLoc SL(Addr);
1301 if (CurDAG->isBaseWithConstantOffset(Addr)) {
1302 SDValue N0 = Addr.getOperand(0);
1303 SDValue N1 = Addr.getOperand(1);
1304
1305 if (SelectSMRDOffset(N1, Offset, Imm)) {
1306 SBase = N0;
1307 return true;
1308 }
1309 }
1310 SBase = Addr;
1311 Offset = CurDAG->getTargetConstant(0, SL, MVT::i32);
1312 Imm = true;
1313 return true;
1314}
1315
1316bool AMDGPUDAGToDAGISel::SelectSMRDImm(SDValue Addr, SDValue &SBase,
1317 SDValue &Offset) const {
1318 bool Imm;
1319 return SelectSMRD(Addr, SBase, Offset, Imm) && Imm;
1320}
1321
Tom Stellard217361c2015-08-06 19:28:38 +00001322bool AMDGPUDAGToDAGISel::SelectSMRDImm32(SDValue Addr, SDValue &SBase,
1323 SDValue &Offset) const {
1324
1325 if (Subtarget->getGeneration() != AMDGPUSubtarget::SEA_ISLANDS)
1326 return false;
1327
1328 bool Imm;
1329 if (!SelectSMRD(Addr, SBase, Offset, Imm))
1330 return false;
1331
1332 return !Imm && isa<ConstantSDNode>(Offset);
1333}
1334
Tom Stellarddee26a22015-08-06 19:28:30 +00001335bool AMDGPUDAGToDAGISel::SelectSMRDSgpr(SDValue Addr, SDValue &SBase,
1336 SDValue &Offset) const {
1337 bool Imm;
Tom Stellard217361c2015-08-06 19:28:38 +00001338 return SelectSMRD(Addr, SBase, Offset, Imm) && !Imm &&
1339 !isa<ConstantSDNode>(Offset);
Tom Stellarddee26a22015-08-06 19:28:30 +00001340}
1341
1342bool AMDGPUDAGToDAGISel::SelectSMRDBufferImm(SDValue Addr,
1343 SDValue &Offset) const {
1344 bool Imm;
1345 return SelectSMRDOffset(Addr, Offset, Imm) && Imm;
1346}
1347
Tom Stellard217361c2015-08-06 19:28:38 +00001348bool AMDGPUDAGToDAGISel::SelectSMRDBufferImm32(SDValue Addr,
1349 SDValue &Offset) const {
1350 if (Subtarget->getGeneration() != AMDGPUSubtarget::SEA_ISLANDS)
1351 return false;
1352
1353 bool Imm;
1354 if (!SelectSMRDOffset(Addr, Offset, Imm))
1355 return false;
1356
1357 return !Imm && isa<ConstantSDNode>(Offset);
1358}
1359
Tom Stellarddee26a22015-08-06 19:28:30 +00001360bool AMDGPUDAGToDAGISel::SelectSMRDBufferSgpr(SDValue Addr,
1361 SDValue &Offset) const {
1362 bool Imm;
Tom Stellard217361c2015-08-06 19:28:38 +00001363 return SelectSMRDOffset(Addr, Offset, Imm) && !Imm &&
1364 !isa<ConstantSDNode>(Offset);
Tom Stellarddee26a22015-08-06 19:28:30 +00001365}
1366
Marek Olsak9b728682015-03-24 13:40:27 +00001367SDNode *AMDGPUDAGToDAGISel::getS_BFE(unsigned Opcode, SDLoc DL, SDValue Val,
1368 uint32_t Offset, uint32_t Width) {
1369 // Transformation function, pack the offset and width of a BFE into
1370 // the format expected by the S_BFE_I32 / S_BFE_U32. In the second
1371 // source, bits [5:0] contain the offset and bits [22:16] the width.
1372 uint32_t PackedVal = Offset | (Width << 16);
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +00001373 SDValue PackedConst = CurDAG->getTargetConstant(PackedVal, DL, MVT::i32);
Marek Olsak9b728682015-03-24 13:40:27 +00001374
1375 return CurDAG->getMachineNode(Opcode, DL, MVT::i32, Val, PackedConst);
1376}
1377
Justin Bogner95927c02016-05-12 21:03:32 +00001378void AMDGPUDAGToDAGISel::SelectS_BFEFromShifts(SDNode *N) {
Marek Olsak9b728682015-03-24 13:40:27 +00001379 // "(a << b) srl c)" ---> "BFE_U32 a, (c-b), (32-c)
1380 // "(a << b) sra c)" ---> "BFE_I32 a, (c-b), (32-c)
1381 // Predicate: 0 < b <= c < 32
1382
1383 const SDValue &Shl = N->getOperand(0);
1384 ConstantSDNode *B = dyn_cast<ConstantSDNode>(Shl->getOperand(1));
1385 ConstantSDNode *C = dyn_cast<ConstantSDNode>(N->getOperand(1));
1386
1387 if (B && C) {
1388 uint32_t BVal = B->getZExtValue();
1389 uint32_t CVal = C->getZExtValue();
1390
1391 if (0 < BVal && BVal <= CVal && CVal < 32) {
1392 bool Signed = N->getOpcode() == ISD::SRA;
1393 unsigned Opcode = Signed ? AMDGPU::S_BFE_I32 : AMDGPU::S_BFE_U32;
1394
Justin Bogner95927c02016-05-12 21:03:32 +00001395 ReplaceNode(N, getS_BFE(Opcode, SDLoc(N), Shl.getOperand(0), CVal - BVal,
1396 32 - CVal));
1397 return;
Marek Olsak9b728682015-03-24 13:40:27 +00001398 }
1399 }
Justin Bogner95927c02016-05-12 21:03:32 +00001400 SelectCode(N);
Marek Olsak9b728682015-03-24 13:40:27 +00001401}
1402
Justin Bogner95927c02016-05-12 21:03:32 +00001403void AMDGPUDAGToDAGISel::SelectS_BFE(SDNode *N) {
Marek Olsak9b728682015-03-24 13:40:27 +00001404 switch (N->getOpcode()) {
1405 case ISD::AND:
1406 if (N->getOperand(0).getOpcode() == ISD::SRL) {
1407 // "(a srl b) & mask" ---> "BFE_U32 a, b, popcount(mask)"
1408 // Predicate: isMask(mask)
1409 const SDValue &Srl = N->getOperand(0);
1410 ConstantSDNode *Shift = dyn_cast<ConstantSDNode>(Srl.getOperand(1));
1411 ConstantSDNode *Mask = dyn_cast<ConstantSDNode>(N->getOperand(1));
1412
1413 if (Shift && Mask) {
1414 uint32_t ShiftVal = Shift->getZExtValue();
1415 uint32_t MaskVal = Mask->getZExtValue();
1416
1417 if (isMask_32(MaskVal)) {
1418 uint32_t WidthVal = countPopulation(MaskVal);
1419
Justin Bogner95927c02016-05-12 21:03:32 +00001420 ReplaceNode(N, getS_BFE(AMDGPU::S_BFE_U32, SDLoc(N),
1421 Srl.getOperand(0), ShiftVal, WidthVal));
1422 return;
Marek Olsak9b728682015-03-24 13:40:27 +00001423 }
1424 }
1425 }
1426 break;
1427 case ISD::SRL:
1428 if (N->getOperand(0).getOpcode() == ISD::AND) {
1429 // "(a & mask) srl b)" ---> "BFE_U32 a, b, popcount(mask >> b)"
1430 // Predicate: isMask(mask >> b)
1431 const SDValue &And = N->getOperand(0);
1432 ConstantSDNode *Shift = dyn_cast<ConstantSDNode>(N->getOperand(1));
1433 ConstantSDNode *Mask = dyn_cast<ConstantSDNode>(And->getOperand(1));
1434
1435 if (Shift && Mask) {
1436 uint32_t ShiftVal = Shift->getZExtValue();
1437 uint32_t MaskVal = Mask->getZExtValue() >> ShiftVal;
1438
1439 if (isMask_32(MaskVal)) {
1440 uint32_t WidthVal = countPopulation(MaskVal);
1441
Justin Bogner95927c02016-05-12 21:03:32 +00001442 ReplaceNode(N, getS_BFE(AMDGPU::S_BFE_U32, SDLoc(N),
1443 And.getOperand(0), ShiftVal, WidthVal));
1444 return;
Marek Olsak9b728682015-03-24 13:40:27 +00001445 }
1446 }
Justin Bogner95927c02016-05-12 21:03:32 +00001447 } else if (N->getOperand(0).getOpcode() == ISD::SHL) {
1448 SelectS_BFEFromShifts(N);
1449 return;
1450 }
Marek Olsak9b728682015-03-24 13:40:27 +00001451 break;
1452 case ISD::SRA:
Justin Bogner95927c02016-05-12 21:03:32 +00001453 if (N->getOperand(0).getOpcode() == ISD::SHL) {
1454 SelectS_BFEFromShifts(N);
1455 return;
1456 }
Marek Olsak9b728682015-03-24 13:40:27 +00001457 break;
Matt Arsenault7e8de012016-04-22 22:59:16 +00001458
1459 case ISD::SIGN_EXTEND_INREG: {
1460 // sext_inreg (srl x, 16), i8 -> bfe_i32 x, 16, 8
1461 SDValue Src = N->getOperand(0);
1462 if (Src.getOpcode() != ISD::SRL)
1463 break;
1464
1465 const ConstantSDNode *Amt = dyn_cast<ConstantSDNode>(Src.getOperand(1));
1466 if (!Amt)
1467 break;
1468
1469 unsigned Width = cast<VTSDNode>(N->getOperand(1))->getVT().getSizeInBits();
Justin Bogner95927c02016-05-12 21:03:32 +00001470 ReplaceNode(N, getS_BFE(AMDGPU::S_BFE_I32, SDLoc(N), Src.getOperand(0),
1471 Amt->getZExtValue(), Width));
1472 return;
Matt Arsenault7e8de012016-04-22 22:59:16 +00001473 }
Marek Olsak9b728682015-03-24 13:40:27 +00001474 }
1475
Justin Bogner95927c02016-05-12 21:03:32 +00001476 SelectCode(N);
Marek Olsak9b728682015-03-24 13:40:27 +00001477}
1478
Justin Bogner95927c02016-05-12 21:03:32 +00001479void AMDGPUDAGToDAGISel::SelectBRCOND(SDNode *N) {
Tom Stellardbc4497b2016-02-12 23:45:29 +00001480 SDValue Cond = N->getOperand(1);
1481
1482 if (isCBranchSCC(N)) {
1483 // This brcond will use S_CBRANCH_SCC*, so let tablegen handle it.
Justin Bogner95927c02016-05-12 21:03:32 +00001484 SelectCode(N);
1485 return;
Tom Stellardbc4497b2016-02-12 23:45:29 +00001486 }
1487
1488 // The result of VOPC instructions is or'd against ~EXEC before it is
1489 // written to vcc or another SGPR. This means that the value '1' is always
1490 // written to the corresponding bit for results that are masked. In order
1491 // to correctly check against vccz, we need to and VCC with the EXEC
1492 // register in order to clear the value from the masked bits.
1493
1494 SDLoc SL(N);
1495
1496 SDNode *MaskedCond =
1497 CurDAG->getMachineNode(AMDGPU::S_AND_B64, SL, MVT::i1,
1498 CurDAG->getRegister(AMDGPU::EXEC, MVT::i1),
1499 Cond);
1500 SDValue VCC = CurDAG->getCopyToReg(N->getOperand(0), SL, AMDGPU::VCC,
1501 SDValue(MaskedCond, 0),
1502 SDValue()); // Passing SDValue() adds a
1503 // glue output.
Justin Bogner95927c02016-05-12 21:03:32 +00001504 CurDAG->SelectNodeTo(N, AMDGPU::S_CBRANCH_VCCNZ, MVT::Other,
1505 N->getOperand(2), // Basic Block
1506 VCC.getValue(0), // Chain
1507 VCC.getValue(1)); // Glue
1508 return;
Tom Stellardbc4497b2016-02-12 23:45:29 +00001509}
1510
Matt Arsenault88701812016-06-09 23:42:48 +00001511// This is here because there isn't a way to use the generated sub0_sub1 as the
1512// subreg index to EXTRACT_SUBREG in tablegen.
1513void AMDGPUDAGToDAGISel::SelectATOMIC_CMP_SWAP(SDNode *N) {
1514 MemSDNode *Mem = cast<MemSDNode>(N);
1515 unsigned AS = Mem->getAddressSpace();
Matt Arsenault7757c592016-06-09 23:42:54 +00001516 if (AS == AMDGPUAS::FLAT_ADDRESS) {
1517 SelectCode(N);
1518 return;
1519 }
Matt Arsenault88701812016-06-09 23:42:48 +00001520
1521 MVT VT = N->getSimpleValueType(0);
1522 bool Is32 = (VT == MVT::i32);
1523 SDLoc SL(N);
1524
1525 MachineSDNode *CmpSwap = nullptr;
1526 if (Subtarget->hasAddr64()) {
1527 SDValue SRsrc, VAddr, SOffset, Offset, GLC, SLC;
1528
1529 if (SelectMUBUFAddr64(Mem->getBasePtr(), SRsrc, VAddr, SOffset, Offset, SLC)) {
1530 unsigned Opcode = Is32 ? AMDGPU::BUFFER_ATOMIC_CMPSWAP_RTN_ADDR64 :
1531 AMDGPU::BUFFER_ATOMIC_CMPSWAP_X2_RTN_ADDR64;
1532 SDValue CmpVal = Mem->getOperand(2);
1533
1534 // XXX - Do we care about glue operands?
1535
1536 SDValue Ops[] = {
1537 CmpVal, VAddr, SRsrc, SOffset, Offset, SLC, Mem->getChain()
1538 };
1539
1540 CmpSwap = CurDAG->getMachineNode(Opcode, SL, Mem->getVTList(), Ops);
1541 }
1542 }
1543
1544 if (!CmpSwap) {
1545 SDValue SRsrc, SOffset, Offset, SLC;
1546 if (SelectMUBUFOffset(Mem->getBasePtr(), SRsrc, SOffset, Offset, SLC)) {
1547 unsigned Opcode = Is32 ? AMDGPU::BUFFER_ATOMIC_CMPSWAP_RTN_OFFSET :
1548 AMDGPU::BUFFER_ATOMIC_CMPSWAP_X2_RTN_OFFSET;
1549
1550 SDValue CmpVal = Mem->getOperand(2);
1551 SDValue Ops[] = {
1552 CmpVal, SRsrc, SOffset, Offset, SLC, Mem->getChain()
1553 };
1554
1555 CmpSwap = CurDAG->getMachineNode(Opcode, SL, Mem->getVTList(), Ops);
1556 }
1557 }
1558
1559 if (!CmpSwap) {
1560 SelectCode(N);
1561 return;
1562 }
1563
1564 MachineSDNode::mmo_iterator MMOs = MF->allocateMemRefsArray(1);
1565 *MMOs = Mem->getMemOperand();
1566 CmpSwap->setMemRefs(MMOs, MMOs + 1);
1567
1568 unsigned SubReg = Is32 ? AMDGPU::sub0 : AMDGPU::sub0_sub1;
1569 SDValue Extract
1570 = CurDAG->getTargetExtractSubreg(SubReg, SL, VT, SDValue(CmpSwap, 0));
1571
1572 ReplaceUses(SDValue(N, 0), Extract);
1573 ReplaceUses(SDValue(N, 1), SDValue(CmpSwap, 1));
1574 CurDAG->RemoveDeadNode(N);
1575}
1576
Tom Stellardb4a313a2014-08-01 00:32:39 +00001577bool AMDGPUDAGToDAGISel::SelectVOP3Mods(SDValue In, SDValue &Src,
1578 SDValue &SrcMods) const {
1579
1580 unsigned Mods = 0;
1581
1582 Src = In;
1583
1584 if (Src.getOpcode() == ISD::FNEG) {
1585 Mods |= SISrcMods::NEG;
1586 Src = Src.getOperand(0);
1587 }
1588
1589 if (Src.getOpcode() == ISD::FABS) {
1590 Mods |= SISrcMods::ABS;
1591 Src = Src.getOperand(0);
1592 }
1593
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +00001594 SrcMods = CurDAG->getTargetConstant(Mods, SDLoc(In), MVT::i32);
Tom Stellardb4a313a2014-08-01 00:32:39 +00001595
1596 return true;
1597}
1598
Tom Stellarddb5a11f2015-07-13 15:47:57 +00001599bool AMDGPUDAGToDAGISel::SelectVOP3NoMods(SDValue In, SDValue &Src,
1600 SDValue &SrcMods) const {
1601 bool Res = SelectVOP3Mods(In, Src, SrcMods);
1602 return Res && cast<ConstantSDNode>(SrcMods)->isNullValue();
1603}
1604
Tom Stellardb4a313a2014-08-01 00:32:39 +00001605bool AMDGPUDAGToDAGISel::SelectVOP3Mods0(SDValue In, SDValue &Src,
1606 SDValue &SrcMods, SDValue &Clamp,
1607 SDValue &Omod) const {
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +00001608 SDLoc DL(In);
Tom Stellardb4a313a2014-08-01 00:32:39 +00001609 // FIXME: Handle Clamp and Omod
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +00001610 Clamp = CurDAG->getTargetConstant(0, DL, MVT::i32);
1611 Omod = CurDAG->getTargetConstant(0, DL, MVT::i32);
Tom Stellardb4a313a2014-08-01 00:32:39 +00001612
1613 return SelectVOP3Mods(In, Src, SrcMods);
1614}
1615
Tom Stellarddb5a11f2015-07-13 15:47:57 +00001616bool AMDGPUDAGToDAGISel::SelectVOP3NoMods0(SDValue In, SDValue &Src,
1617 SDValue &SrcMods, SDValue &Clamp,
1618 SDValue &Omod) const {
1619 bool Res = SelectVOP3Mods0(In, Src, SrcMods, Clamp, Omod);
1620
1621 return Res && cast<ConstantSDNode>(SrcMods)->isNullValue() &&
1622 cast<ConstantSDNode>(Clamp)->isNullValue() &&
1623 cast<ConstantSDNode>(Omod)->isNullValue();
1624}
1625
Matt Arsenault1cffa4c2014-11-13 19:49:04 +00001626bool AMDGPUDAGToDAGISel::SelectVOP3Mods0Clamp(SDValue In, SDValue &Src,
1627 SDValue &SrcMods,
1628 SDValue &Omod) const {
1629 // FIXME: Handle Omod
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +00001630 Omod = CurDAG->getTargetConstant(0, SDLoc(In), MVT::i32);
Matt Arsenault1cffa4c2014-11-13 19:49:04 +00001631
1632 return SelectVOP3Mods(In, Src, SrcMods);
1633}
1634
Matt Arsenault4831ce52015-01-06 23:00:37 +00001635bool AMDGPUDAGToDAGISel::SelectVOP3Mods0Clamp0OMod(SDValue In, SDValue &Src,
1636 SDValue &SrcMods,
1637 SDValue &Clamp,
1638 SDValue &Omod) const {
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +00001639 Clamp = Omod = CurDAG->getTargetConstant(0, SDLoc(In), MVT::i32);
Matt Arsenault4831ce52015-01-06 23:00:37 +00001640 return SelectVOP3Mods(In, Src, SrcMods);
1641}
1642
Matt Arsenault4bf43d42015-09-25 17:27:08 +00001643void AMDGPUDAGToDAGISel::PreprocessISelDAG() {
Matt Arsenaultf43c2a02016-03-23 21:49:25 +00001644 MachineFrameInfo *MFI = CurDAG->getMachineFunction().getFrameInfo();
1645
1646 // Handle the perverse case where a frame index is being stored. We don't
1647 // want to see multiple frame index operands on the same instruction since
1648 // it complicates things and violates some assumptions about frame index
1649 // lowering.
1650 for (int I = MFI->getObjectIndexBegin(), E = MFI->getObjectIndexEnd();
1651 I != E; ++I) {
1652 SDValue FI = CurDAG->getTargetFrameIndex(I, MVT::i32);
1653
1654 // It's possible that we have a frame index defined in the function that
1655 // isn't used in this block.
1656 if (FI.use_empty())
1657 continue;
1658
1659 // Skip over the AssertZext inserted during lowering.
1660 SDValue EffectiveFI = FI;
1661 auto It = FI->use_begin();
1662 if (It->getOpcode() == ISD::AssertZext && FI->hasOneUse()) {
1663 EffectiveFI = SDValue(*It, 0);
1664 It = EffectiveFI->use_begin();
1665 }
1666
1667 for (auto It = EffectiveFI->use_begin(); !It.atEnd(); ) {
1668 SDUse &Use = It.getUse();
1669 SDNode *User = Use.getUser();
1670 unsigned OpIdx = It.getOperandNo();
1671 ++It;
1672
1673 if (MemSDNode *M = dyn_cast<MemSDNode>(User)) {
1674 unsigned PtrIdx = M->getOpcode() == ISD::STORE ? 2 : 1;
1675 if (OpIdx == PtrIdx)
1676 continue;
1677
Vasileios Kalintirisb8a37202016-03-24 10:53:28 +00001678 unsigned OpN = M->getNumOperands();
Matt Arsenaultf43c2a02016-03-23 21:49:25 +00001679 SDValue NewOps[8];
1680
1681 assert(OpN < array_lengthof(NewOps));
1682 for (unsigned Op = 0; Op != OpN; ++Op) {
1683 if (Op != OpIdx) {
1684 NewOps[Op] = M->getOperand(Op);
1685 continue;
1686 }
1687
1688 MachineSDNode *Mov = CurDAG->getMachineNode(AMDGPU::V_MOV_B32_e32,
1689 SDLoc(M), MVT::i32, FI);
1690 NewOps[Op] = SDValue(Mov, 0);
1691 }
1692
1693 CurDAG->UpdateNodeOperands(M, makeArrayRef(NewOps, OpN));
Matt Arsenaultf43c2a02016-03-23 21:49:25 +00001694 }
1695 }
1696 }
Matt Arsenault4bf43d42015-09-25 17:27:08 +00001697}
1698
Christian Konigd910b7d2013-02-26 17:52:16 +00001699void AMDGPUDAGToDAGISel::PostprocessISelDAG() {
Bill Wendlinga3cd3502013-06-19 21:36:55 +00001700 const AMDGPUTargetLowering& Lowering =
Matt Arsenault209a7b92014-04-18 07:40:20 +00001701 *static_cast<const AMDGPUTargetLowering*>(getTargetLowering());
Vincent Lejeuneab3baf82013-09-12 23:44:44 +00001702 bool IsModified = false;
1703 do {
1704 IsModified = false;
1705 // Go over all selected nodes and try to fold them a bit more
Pete Cooper65c69402015-07-14 22:10:54 +00001706 for (SDNode &Node : CurDAG->allnodes()) {
1707 MachineSDNode *MachineNode = dyn_cast<MachineSDNode>(&Node);
Vincent Lejeuneab3baf82013-09-12 23:44:44 +00001708 if (!MachineNode)
1709 continue;
Christian Konigd910b7d2013-02-26 17:52:16 +00001710
Vincent Lejeuneab3baf82013-09-12 23:44:44 +00001711 SDNode *ResNode = Lowering.PostISelFolding(MachineNode, *CurDAG);
Pete Cooper65c69402015-07-14 22:10:54 +00001712 if (ResNode != &Node) {
1713 ReplaceUses(&Node, ResNode);
Vincent Lejeuneab3baf82013-09-12 23:44:44 +00001714 IsModified = true;
1715 }
Tom Stellard2183b702013-06-03 17:39:46 +00001716 }
Vincent Lejeuneab3baf82013-09-12 23:44:44 +00001717 CurDAG->RemoveDeadNodes();
1718 } while (IsModified);
Christian Konigd910b7d2013-02-26 17:52:16 +00001719}