blob: e3807f15e225d56b484019d1b5e09dda627a5ed8 [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
Adrian Prantl5f8f34e42018-05-01 15:54:18 +000011/// Defines an instruction selector for the AMDGPU target.
Tom Stellard75aadc22012-12-11 21:25:42 +000012//
13//===----------------------------------------------------------------------===//
Matt Arsenault592d0682015-12-01 23:04:05 +000014
Eugene Zelenko2bc2f332016-12-09 22:06:55 +000015#include "AMDGPU.h"
Matt Arsenault7016f132017-08-03 22:30:46 +000016#include "AMDGPUArgumentUsageInfo.h"
Chandler Carruth6bda14b2017-06-06 11:49:48 +000017#include "AMDGPUISelLowering.h" // For AMDGPUISD
Tom Stellard75aadc22012-12-11 21:25:42 +000018#include "AMDGPUInstrInfo.h"
Stanislav Mekhanoshin1c538422018-05-25 17:25:12 +000019#include "AMDGPUPerfHintAnalysis.h"
Eugene Zelenko2bc2f332016-12-09 22:06:55 +000020#include "AMDGPURegisterInfo.h"
Tom Stellard2e59a452014-06-13 01:32:00 +000021#include "AMDGPUSubtarget.h"
Matt Arsenaultcc852232017-10-10 20:22:07 +000022#include "AMDGPUTargetMachine.h"
Eugene Zelenko2bc2f332016-12-09 22:06:55 +000023#include "SIDefines.h"
Christian Konigf82901a2013-02-26 17:52:23 +000024#include "SIISelLowering.h"
Chandler Carruth6bda14b2017-06-06 11:49:48 +000025#include "SIInstrInfo.h"
Tom Stellardb02094e2014-07-21 15:45:01 +000026#include "SIMachineFunctionInfo.h"
Chandler Carruth6bda14b2017-06-06 11:49:48 +000027#include "SIRegisterInfo.h"
Tom Stellard44b30b42018-05-22 02:03:23 +000028#include "MCTargetDesc/AMDGPUMCTargetDesc.h"
Eugene Zelenko2bc2f332016-12-09 22:06:55 +000029#include "llvm/ADT/APInt.h"
30#include "llvm/ADT/SmallVector.h"
31#include "llvm/ADT/StringRef.h"
Alexander Timofeev2e5eece2018-03-05 15:12:21 +000032#include "llvm/Analysis/DivergenceAnalysis.h"
Jan Veselyf97de002016-05-13 20:39:29 +000033#include "llvm/Analysis/ValueTracking.h"
Tom Stellard58ac7442014-04-29 23:12:48 +000034#include "llvm/CodeGen/FunctionLoweringInfo.h"
Eugene Zelenko2bc2f332016-12-09 22:06:55 +000035#include "llvm/CodeGen/ISDOpcodes.h"
36#include "llvm/CodeGen/MachineFunction.h"
37#include "llvm/CodeGen/MachineRegisterInfo.h"
Benjamin Kramerd78bb462013-05-23 17:10:37 +000038#include "llvm/CodeGen/SelectionDAG.h"
Tom Stellard75aadc22012-12-11 21:25:42 +000039#include "llvm/CodeGen/SelectionDAGISel.h"
Eugene Zelenko2bc2f332016-12-09 22:06:55 +000040#include "llvm/CodeGen/SelectionDAGNodes.h"
Craig Topper2fa14362018-03-29 17:21:10 +000041#include "llvm/CodeGen/ValueTypes.h"
Eugene Zelenko2bc2f332016-12-09 22:06:55 +000042#include "llvm/IR/BasicBlock.h"
43#include "llvm/IR/Instruction.h"
44#include "llvm/MC/MCInstrDesc.h"
45#include "llvm/Support/Casting.h"
46#include "llvm/Support/CodeGen.h"
47#include "llvm/Support/ErrorHandling.h"
David Blaikie13e77db2018-03-23 23:58:25 +000048#include "llvm/Support/MachineValueType.h"
Eugene Zelenko2bc2f332016-12-09 22:06:55 +000049#include "llvm/Support/MathExtras.h"
50#include <cassert>
51#include <cstdint>
52#include <new>
53#include <vector>
Tom Stellard75aadc22012-12-11 21:25:42 +000054
55using namespace llvm;
56
Matt Arsenaultd2759212016-02-13 01:24:08 +000057namespace llvm {
Eugene Zelenko2bc2f332016-12-09 22:06:55 +000058
Matt Arsenaultd2759212016-02-13 01:24:08 +000059class R600InstrInfo;
Eugene Zelenko2bc2f332016-12-09 22:06:55 +000060
61} // end namespace llvm
Matt Arsenaultd2759212016-02-13 01:24:08 +000062
Tom Stellard75aadc22012-12-11 21:25:42 +000063//===----------------------------------------------------------------------===//
64// Instruction Selector Implementation
65//===----------------------------------------------------------------------===//
66
67namespace {
Tom Stellardbc4497b2016-02-12 23:45:29 +000068
Tom Stellard75aadc22012-12-11 21:25:42 +000069/// AMDGPU specific code to select AMDGPU machine instructions for
70/// SelectionDAG operations.
71class AMDGPUDAGToDAGISel : public SelectionDAGISel {
72 // Subtarget - Keep a pointer to the AMDGPU Subtarget around so that we can
73 // make the right decision when generating code for different targets.
Tom Stellard5bfbae52018-07-11 20:59:01 +000074 const GCNSubtarget *Subtarget;
Yaxun Liu1a14bfa2017-03-27 14:04:01 +000075 AMDGPUAS AMDGPUASI;
Matt Arsenaultcc852232017-10-10 20:22:07 +000076 bool EnableLateStructurizeCFG;
NAKAMURA Takumia9cb5382015-09-22 11:14:39 +000077
Tom Stellard75aadc22012-12-11 21:25:42 +000078public:
Matt Arsenault7016f132017-08-03 22:30:46 +000079 explicit AMDGPUDAGToDAGISel(TargetMachine *TM = nullptr,
80 CodeGenOpt::Level OptLevel = CodeGenOpt::Default)
81 : SelectionDAGISel(*TM, OptLevel) {
82 AMDGPUASI = AMDGPU::getAMDGPUAS(*TM);
Matt Arsenaultcc852232017-10-10 20:22:07 +000083 EnableLateStructurizeCFG = AMDGPUTargetMachine::EnableLateStructurizeCFG;
Yaxun Liu1a14bfa2017-03-27 14:04:01 +000084 }
Eugene Zelenko2bc2f332016-12-09 22:06:55 +000085 ~AMDGPUDAGToDAGISel() override = default;
Konstantin Zhuravlyov60a83732016-10-03 18:47:26 +000086
Matt Arsenault7016f132017-08-03 22:30:46 +000087 void getAnalysisUsage(AnalysisUsage &AU) const override {
88 AU.addRequired<AMDGPUArgumentUsageInfo>();
Stanislav Mekhanoshin1c538422018-05-25 17:25:12 +000089 AU.addRequired<AMDGPUPerfHintAnalysis>();
Alexander Timofeev2e5eece2018-03-05 15:12:21 +000090 AU.addRequired<DivergenceAnalysis>();
Matt Arsenault7016f132017-08-03 22:30:46 +000091 SelectionDAGISel::getAnalysisUsage(AU);
92 }
93
Eric Christopher7792e322015-01-30 23:24:40 +000094 bool runOnMachineFunction(MachineFunction &MF) override;
Justin Bogner95927c02016-05-12 21:03:32 +000095 void Select(SDNode *N) override;
Mehdi Amini117296c2016-10-01 02:56:57 +000096 StringRef getPassName() const override;
Craig Topper5656db42014-04-29 07:57:24 +000097 void PostprocessISelDAG() override;
Tom Stellard75aadc22012-12-11 21:25:42 +000098
Tom Stellard20287692017-08-08 04:57:55 +000099protected:
100 void SelectBuildVector(SDNode *N, unsigned RegClassID);
101
Tom Stellard75aadc22012-12-11 21:25:42 +0000102private:
Matt Arsenault156d3ae2017-05-17 21:02:58 +0000103 std::pair<SDValue, SDValue> foldFrameIndex(SDValue N) const;
Matt Arsenaultf84e5d92017-01-31 03:07:46 +0000104 bool isNoNanSrc(SDValue N) const;
Matt Arsenaultfe267752016-07-28 00:32:02 +0000105 bool isInlineImmediate(const SDNode *N) const;
Tom Stellard75aadc22012-12-11 21:25:42 +0000106
Tom Stellardbc4497b2016-02-12 23:45:29 +0000107 bool isUniformBr(const SDNode *N) const;
108
Tom Stellard381a94a2015-05-12 15:00:49 +0000109 SDNode *glueCopyToM0(SDNode *N) const;
110
Tom Stellarddf94dc32013-08-14 23:24:24 +0000111 const TargetRegisterClass *getOperandRegClass(SDNode *N, unsigned OpNo) const;
Tom Stellard20287692017-08-08 04:57:55 +0000112 virtual bool SelectADDRVTX_READ(SDValue Addr, SDValue &Base, SDValue &Offset);
113 virtual bool SelectADDRIndirect(SDValue Addr, SDValue &Base, SDValue &Offset);
Tom Stellard85e8b6d2014-08-22 18:49:33 +0000114 bool isDSOffsetLegal(const SDValue &Base, unsigned Offset,
115 unsigned OffsetBits) const;
116 bool SelectDS1Addr1Offset(SDValue Ptr, SDValue &Base, SDValue &Offset) const;
Tom Stellardf3fc5552014-08-22 18:49:35 +0000117 bool SelectDS64Bit4ByteAligned(SDValue Ptr, SDValue &Base, SDValue &Offset0,
118 SDValue &Offset1) const;
Changpeng Fangb41574a2015-12-22 20:55:23 +0000119 bool SelectMUBUF(SDValue Addr, SDValue &SRsrc, SDValue &VAddr,
Tom Stellard155bbb72014-08-11 22:18:17 +0000120 SDValue &SOffset, SDValue &Offset, SDValue &Offen,
121 SDValue &Idxen, SDValue &Addr64, SDValue &GLC, SDValue &SLC,
122 SDValue &TFE) const;
123 bool SelectMUBUFAddr64(SDValue Addr, SDValue &SRsrc, SDValue &VAddr,
Tom Stellard1f9939f2015-02-27 14:59:41 +0000124 SDValue &SOffset, SDValue &Offset, SDValue &GLC,
125 SDValue &SLC, SDValue &TFE) const;
Tom Stellard7980fc82014-09-25 18:30:26 +0000126 bool SelectMUBUFAddr64(SDValue Addr, SDValue &SRsrc,
Tom Stellardc53861a2015-02-11 00:34:32 +0000127 SDValue &VAddr, SDValue &SOffset, SDValue &Offset,
Tom Stellard7980fc82014-09-25 18:30:26 +0000128 SDValue &SLC) const;
Matt Arsenaultb81495d2017-09-20 05:01:53 +0000129 bool SelectMUBUFScratchOffen(SDNode *Parent,
Matt Arsenault156d3ae2017-05-17 21:02:58 +0000130 SDValue Addr, SDValue &RSrc, SDValue &VAddr,
Matt Arsenault0774ea22017-04-24 19:40:59 +0000131 SDValue &SOffset, SDValue &ImmOffset) const;
Matt Arsenaultb81495d2017-09-20 05:01:53 +0000132 bool SelectMUBUFScratchOffset(SDNode *Parent,
Matt Arsenault156d3ae2017-05-17 21:02:58 +0000133 SDValue Addr, SDValue &SRsrc, SDValue &Soffset,
Matt Arsenault0774ea22017-04-24 19:40:59 +0000134 SDValue &Offset) const;
135
Tom Stellard155bbb72014-08-11 22:18:17 +0000136 bool SelectMUBUFOffset(SDValue Addr, SDValue &SRsrc, SDValue &SOffset,
137 SDValue &Offset, SDValue &GLC, SDValue &SLC,
Tom Stellardb02094e2014-07-21 15:45:01 +0000138 SDValue &TFE) const;
Tom Stellard7980fc82014-09-25 18:30:26 +0000139 bool SelectMUBUFOffset(SDValue Addr, SDValue &SRsrc, SDValue &Soffset,
Matt Arsenault88701812016-06-09 23:42:48 +0000140 SDValue &Offset, SDValue &SLC) const;
Jan Vesely43b7b5b2016-04-07 19:23:11 +0000141 bool SelectMUBUFOffset(SDValue Addr, SDValue &SRsrc, SDValue &Soffset,
142 SDValue &Offset) const;
Nicolai Haehnlea6092592016-06-15 07:13:05 +0000143 bool SelectMUBUFConstant(SDValue Constant,
Nicolai Haehnle3003ba02016-03-18 16:24:20 +0000144 SDValue &SOffset,
145 SDValue &ImmOffset) const;
146 bool SelectMUBUFIntrinsicOffset(SDValue Offset, SDValue &SOffset,
147 SDValue &ImmOffset) const;
148 bool SelectMUBUFIntrinsicVOffset(SDValue Offset, SDValue &SOffset,
149 SDValue &ImmOffset, SDValue &VOffset) const;
Matt Arsenault7757c592016-06-09 23:42:54 +0000150
Matt Arsenaultdb7c6a82017-06-12 16:53:51 +0000151 bool SelectFlatAtomic(SDValue Addr, SDValue &VAddr,
152 SDValue &Offset, SDValue &SLC) const;
Matt Arsenault4e309b02017-07-29 01:03:53 +0000153 bool SelectFlatAtomicSigned(SDValue Addr, SDValue &VAddr,
154 SDValue &Offset, SDValue &SLC) const;
155
156 template <bool IsSigned>
Matt Arsenaultdb7c6a82017-06-12 16:53:51 +0000157 bool SelectFlatOffset(SDValue Addr, SDValue &VAddr,
158 SDValue &Offset, SDValue &SLC) const;
Matt Arsenault7757c592016-06-09 23:42:54 +0000159
Tom Stellarddee26a22015-08-06 19:28:30 +0000160 bool SelectSMRDOffset(SDValue ByteOffsetNode, SDValue &Offset,
161 bool &Imm) const;
Matt Arsenault923712b2018-02-09 16:57:57 +0000162 SDValue Expand32BitAddress(SDValue Addr) const;
Tom Stellarddee26a22015-08-06 19:28:30 +0000163 bool SelectSMRD(SDValue Addr, SDValue &SBase, SDValue &Offset,
164 bool &Imm) const;
165 bool SelectSMRDImm(SDValue Addr, SDValue &SBase, SDValue &Offset) const;
Marek Olsak8973a0a2017-05-24 14:53:50 +0000166 bool SelectSMRDImm32(SDValue Addr, SDValue &SBase, SDValue &Offset) const;
Tom Stellarddee26a22015-08-06 19:28:30 +0000167 bool SelectSMRDSgpr(SDValue Addr, SDValue &SBase, SDValue &Offset) const;
168 bool SelectSMRDBufferImm(SDValue Addr, SDValue &Offset) const;
Marek Olsak8973a0a2017-05-24 14:53:50 +0000169 bool SelectSMRDBufferImm32(SDValue Addr, SDValue &Offset) const;
Nicolai Haehnle7968c342016-07-12 08:12:16 +0000170 bool SelectMOVRELOffset(SDValue Index, SDValue &Base, SDValue &Offset) const;
Matt Arsenaultf84e5d92017-01-31 03:07:46 +0000171
172 bool SelectVOP3Mods_NNaN(SDValue In, SDValue &Src, SDValue &SrcMods) const;
Matt Arsenaultd7e23032017-09-07 18:05:07 +0000173 bool SelectVOP3ModsImpl(SDValue In, SDValue &Src, unsigned &SrcMods) const;
Tom Stellardb4a313a2014-08-01 00:32:39 +0000174 bool SelectVOP3Mods(SDValue In, SDValue &Src, SDValue &SrcMods) const;
Matt Arsenaultdf58e822017-04-25 21:17:38 +0000175 bool SelectVOP3NoMods(SDValue In, SDValue &Src) const;
Tom Stellardb4a313a2014-08-01 00:32:39 +0000176 bool SelectVOP3Mods0(SDValue In, SDValue &Src, SDValue &SrcMods,
177 SDValue &Clamp, SDValue &Omod) const;
Tom Stellarddb5a11f2015-07-13 15:47:57 +0000178 bool SelectVOP3NoMods0(SDValue In, SDValue &Src, SDValue &SrcMods,
179 SDValue &Clamp, SDValue &Omod) const;
Tom Stellard75aadc22012-12-11 21:25:42 +0000180
Matt Arsenault4831ce52015-01-06 23:00:37 +0000181 bool SelectVOP3Mods0Clamp0OMod(SDValue In, SDValue &Src, SDValue &SrcMods,
182 SDValue &Clamp,
183 SDValue &Omod) const;
Matt Arsenault1cffa4c2014-11-13 19:49:04 +0000184
Dmitry Preobrazhenskyc512d442017-03-27 15:57:17 +0000185 bool SelectVOP3OMods(SDValue In, SDValue &Src,
186 SDValue &Clamp, SDValue &Omod) const;
187
Matt Arsenaulteb522e62017-02-27 22:15:25 +0000188 bool SelectVOP3PMods(SDValue In, SDValue &Src, SDValue &SrcMods) const;
189 bool SelectVOP3PMods0(SDValue In, SDValue &Src, SDValue &SrcMods,
190 SDValue &Clamp) const;
191
Dmitry Preobrazhenskyabf28392017-07-21 13:54:11 +0000192 bool SelectVOP3OpSel(SDValue In, SDValue &Src, SDValue &SrcMods) const;
193 bool SelectVOP3OpSel0(SDValue In, SDValue &Src, SDValue &SrcMods,
194 SDValue &Clamp) const;
195
196 bool SelectVOP3OpSelMods(SDValue In, SDValue &Src, SDValue &SrcMods) const;
197 bool SelectVOP3OpSelMods0(SDValue In, SDValue &Src, SDValue &SrcMods,
198 SDValue &Clamp) const;
Matt Arsenaultd7e23032017-09-07 18:05:07 +0000199 bool SelectVOP3PMadMixModsImpl(SDValue In, SDValue &Src, unsigned &Mods) const;
Matt Arsenault76935122017-09-20 20:28:39 +0000200 bool SelectVOP3PMadMixMods(SDValue In, SDValue &Src, SDValue &SrcMods) const;
Dmitry Preobrazhenskyabf28392017-07-21 13:54:11 +0000201
Matt Arsenaulte1cd4822017-11-13 00:22:09 +0000202 bool SelectHi16Elt(SDValue In, SDValue &Src) const;
203
Justin Bogner95927c02016-05-12 21:03:32 +0000204 void SelectADD_SUB_I64(SDNode *N);
Matt Arsenaultee3f0ac2017-01-30 18:11:38 +0000205 void SelectUADDO_USUBO(SDNode *N);
Justin Bogner95927c02016-05-12 21:03:32 +0000206 void SelectDIV_SCALE(SDNode *N);
Matt Arsenault4f6318f2017-11-06 17:04:37 +0000207 void SelectMAD_64_32(SDNode *N);
Tom Stellard8485fa02016-12-07 02:42:15 +0000208 void SelectFMA_W_CHAIN(SDNode *N);
209 void SelectFMUL_W_CHAIN(SDNode *N);
Matt Arsenault9fa3f932014-06-23 18:00:34 +0000210
Benjamin Kramerbdc49562016-06-12 15:39:02 +0000211 SDNode *getS_BFE(unsigned Opcode, const SDLoc &DL, SDValue Val,
Marek Olsak9b728682015-03-24 13:40:27 +0000212 uint32_t Offset, uint32_t Width);
Justin Bogner95927c02016-05-12 21:03:32 +0000213 void SelectS_BFEFromShifts(SDNode *N);
214 void SelectS_BFE(SDNode *N);
Matt Arsenault7b1dc2c2016-09-17 02:02:19 +0000215 bool isCBranchSCC(const SDNode *N) const;
Justin Bogner95927c02016-05-12 21:03:32 +0000216 void SelectBRCOND(SDNode *N);
Matt Arsenault0084adc2018-04-30 19:08:16 +0000217 void SelectFMAD_FMA(SDNode *N);
Matt Arsenault88701812016-06-09 23:42:48 +0000218 void SelectATOMIC_CMP_SWAP(SDNode *N);
Marek Olsak9b728682015-03-24 13:40:27 +0000219
Tom Stellard20287692017-08-08 04:57:55 +0000220protected:
Tom Stellard75aadc22012-12-11 21:25:42 +0000221 // Include the pieces autogenerated from the target description.
222#include "AMDGPUGenDAGISel.inc"
223};
Eugene Zelenko2bc2f332016-12-09 22:06:55 +0000224
Tom Stellard20287692017-08-08 04:57:55 +0000225class R600DAGToDAGISel : public AMDGPUDAGToDAGISel {
Tom Stellardc5a154d2018-06-28 23:47:12 +0000226 const R600Subtarget *Subtarget;
227 AMDGPUAS AMDGPUASI;
228
229 bool isConstantLoad(const MemSDNode *N, int cbID) const;
230 bool SelectGlobalValueConstantOffset(SDValue Addr, SDValue& IntPtr);
231 bool SelectGlobalValueVariableOffset(SDValue Addr, SDValue &BaseReg,
232 SDValue& Offset);
Tom Stellard20287692017-08-08 04:57:55 +0000233public:
234 explicit R600DAGToDAGISel(TargetMachine *TM, CodeGenOpt::Level OptLevel) :
Tom Stellardc5a154d2018-06-28 23:47:12 +0000235 AMDGPUDAGToDAGISel(TM, OptLevel) {
236 AMDGPUASI = AMDGPU::getAMDGPUAS(*TM);
237 }
Tom Stellard20287692017-08-08 04:57:55 +0000238
239 void Select(SDNode *N) override;
240
241 bool SelectADDRIndirect(SDValue Addr, SDValue &Base,
242 SDValue &Offset) override;
243 bool SelectADDRVTX_READ(SDValue Addr, SDValue &Base,
244 SDValue &Offset) override;
Tom Stellardc5a154d2018-06-28 23:47:12 +0000245
246 bool runOnMachineFunction(MachineFunction &MF) override;
247protected:
248 // Include the pieces autogenerated from the target description.
249#include "R600GenDAGISel.inc"
Tom Stellard20287692017-08-08 04:57:55 +0000250};
251
Tom Stellard75aadc22012-12-11 21:25:42 +0000252} // end anonymous namespace
253
Matt Arsenault7016f132017-08-03 22:30:46 +0000254INITIALIZE_PASS_BEGIN(AMDGPUDAGToDAGISel, "isel",
255 "AMDGPU DAG->DAG Pattern Instruction Selection", false, false)
256INITIALIZE_PASS_DEPENDENCY(AMDGPUArgumentUsageInfo)
Stanislav Mekhanoshin1c538422018-05-25 17:25:12 +0000257INITIALIZE_PASS_DEPENDENCY(AMDGPUPerfHintAnalysis)
Stanislav Mekhanoshin9badad22018-05-21 18:18:52 +0000258INITIALIZE_PASS_DEPENDENCY(DivergenceAnalysis)
Matt Arsenault7016f132017-08-03 22:30:46 +0000259INITIALIZE_PASS_END(AMDGPUDAGToDAGISel, "isel",
260 "AMDGPU DAG->DAG Pattern Instruction Selection", false, false)
261
Adrian Prantl5f8f34e42018-05-01 15:54:18 +0000262/// This pass converts a legalized DAG into a AMDGPU-specific
Tom Stellard75aadc22012-12-11 21:25:42 +0000263// DAG, ready for instruction scheduling.
Matt Arsenault7016f132017-08-03 22:30:46 +0000264FunctionPass *llvm::createAMDGPUISelDag(TargetMachine *TM,
Konstantin Zhuravlyov60a83732016-10-03 18:47:26 +0000265 CodeGenOpt::Level OptLevel) {
266 return new AMDGPUDAGToDAGISel(TM, OptLevel);
Tom Stellard75aadc22012-12-11 21:25:42 +0000267}
268
Adrian Prantl5f8f34e42018-05-01 15:54:18 +0000269/// This pass converts a legalized DAG into a R600-specific
Tom Stellard20287692017-08-08 04:57:55 +0000270// DAG, ready for instruction scheduling.
271FunctionPass *llvm::createR600ISelDag(TargetMachine *TM,
272 CodeGenOpt::Level OptLevel) {
273 return new R600DAGToDAGISel(TM, OptLevel);
274}
275
Eric Christopher7792e322015-01-30 23:24:40 +0000276bool AMDGPUDAGToDAGISel::runOnMachineFunction(MachineFunction &MF) {
Tom Stellard5bfbae52018-07-11 20:59:01 +0000277 Subtarget = &MF.getSubtarget<GCNSubtarget>();
Eric Christopher7792e322015-01-30 23:24:40 +0000278 return SelectionDAGISel::runOnMachineFunction(MF);
Tom Stellard75aadc22012-12-11 21:25:42 +0000279}
280
Matt Arsenaultf84e5d92017-01-31 03:07:46 +0000281bool AMDGPUDAGToDAGISel::isNoNanSrc(SDValue N) const {
282 if (TM.Options.NoNaNsFPMath)
283 return true;
284
285 // TODO: Move into isKnownNeverNaN
Amara Emersond28f0cd42017-05-01 15:17:51 +0000286 if (N->getFlags().isDefined())
287 return N->getFlags().hasNoNaNs();
Matt Arsenaultf84e5d92017-01-31 03:07:46 +0000288
289 return CurDAG->isKnownNeverNaN(N);
290}
291
Matt Arsenaultfe267752016-07-28 00:32:02 +0000292bool AMDGPUDAGToDAGISel::isInlineImmediate(const SDNode *N) const {
Tom Stellardc5a154d2018-06-28 23:47:12 +0000293 const SIInstrInfo *TII = Subtarget->getInstrInfo();
Matt Arsenaultfe267752016-07-28 00:32:02 +0000294
295 if (const ConstantSDNode *C = dyn_cast<ConstantSDNode>(N))
296 return TII->isInlineConstant(C->getAPIntValue());
297
298 if (const ConstantFPSDNode *C = dyn_cast<ConstantFPSDNode>(N))
299 return TII->isInlineConstant(C->getValueAPF().bitcastToAPInt());
300
301 return false;
Tom Stellard7ed0b522014-04-03 20:19:27 +0000302}
303
Adrian Prantl5f8f34e42018-05-01 15:54:18 +0000304/// Determine the register class for \p OpNo
Tom Stellarddf94dc32013-08-14 23:24:24 +0000305/// \returns The register class of the virtual register that will be used for
306/// the given operand number \OpNo or NULL if the register class cannot be
307/// determined.
308const TargetRegisterClass *AMDGPUDAGToDAGISel::getOperandRegClass(SDNode *N,
309 unsigned OpNo) const {
Matt Arsenaultc507cdb2016-11-01 23:22:17 +0000310 if (!N->isMachineOpcode()) {
311 if (N->getOpcode() == ISD::CopyToReg) {
312 unsigned Reg = cast<RegisterSDNode>(N->getOperand(1))->getReg();
313 if (TargetRegisterInfo::isVirtualRegister(Reg)) {
314 MachineRegisterInfo &MRI = CurDAG->getMachineFunction().getRegInfo();
315 return MRI.getRegClass(Reg);
316 }
317
318 const SIRegisterInfo *TRI
Tom Stellard5bfbae52018-07-11 20:59:01 +0000319 = static_cast<const GCNSubtarget *>(Subtarget)->getRegisterInfo();
Matt Arsenaultc507cdb2016-11-01 23:22:17 +0000320 return TRI->getPhysRegClass(Reg);
321 }
322
Matt Arsenault209a7b92014-04-18 07:40:20 +0000323 return nullptr;
Matt Arsenaultc507cdb2016-11-01 23:22:17 +0000324 }
Matt Arsenault209a7b92014-04-18 07:40:20 +0000325
Tom Stellarddf94dc32013-08-14 23:24:24 +0000326 switch (N->getMachineOpcode()) {
327 default: {
Eric Christopherd9134482014-08-04 21:25:23 +0000328 const MCInstrDesc &Desc =
Eric Christopher7792e322015-01-30 23:24:40 +0000329 Subtarget->getInstrInfo()->get(N->getMachineOpcode());
Alexey Samsonov3186eb32013-08-15 07:11:34 +0000330 unsigned OpIdx = Desc.getNumDefs() + OpNo;
331 if (OpIdx >= Desc.getNumOperands())
Matt Arsenault209a7b92014-04-18 07:40:20 +0000332 return nullptr;
Alexey Samsonov3186eb32013-08-15 07:11:34 +0000333 int RegClass = Desc.OpInfo[OpIdx].RegClass;
Matt Arsenault209a7b92014-04-18 07:40:20 +0000334 if (RegClass == -1)
335 return nullptr;
336
Eric Christopher7792e322015-01-30 23:24:40 +0000337 return Subtarget->getRegisterInfo()->getRegClass(RegClass);
Tom Stellarddf94dc32013-08-14 23:24:24 +0000338 }
339 case AMDGPU::REG_SEQUENCE: {
Matt Arsenault209a7b92014-04-18 07:40:20 +0000340 unsigned RCID = cast<ConstantSDNode>(N->getOperand(0))->getZExtValue();
Eric Christopherd9134482014-08-04 21:25:23 +0000341 const TargetRegisterClass *SuperRC =
Eric Christopher7792e322015-01-30 23:24:40 +0000342 Subtarget->getRegisterInfo()->getRegClass(RCID);
Matt Arsenault209a7b92014-04-18 07:40:20 +0000343
344 SDValue SubRegOp = N->getOperand(OpNo + 1);
345 unsigned SubRegIdx = cast<ConstantSDNode>(SubRegOp)->getZExtValue();
Eric Christopher7792e322015-01-30 23:24:40 +0000346 return Subtarget->getRegisterInfo()->getSubClassWithSubReg(SuperRC,
347 SubRegIdx);
Tom Stellarddf94dc32013-08-14 23:24:24 +0000348 }
349 }
350}
351
Tom Stellard381a94a2015-05-12 15:00:49 +0000352SDNode *AMDGPUDAGToDAGISel::glueCopyToM0(SDNode *N) const {
Matt Arsenault3f71c0e2017-11-29 00:55:57 +0000353 if (cast<MemSDNode>(N)->getAddressSpace() != AMDGPUASI.LOCAL_ADDRESS ||
354 !Subtarget->ldsRequiresM0Init())
Tom Stellard381a94a2015-05-12 15:00:49 +0000355 return N;
356
357 const SITargetLowering& Lowering =
358 *static_cast<const SITargetLowering*>(getTargetLowering());
359
360 // Write max value to m0 before each load operation
361
362 SDValue M0 = Lowering.copyToM0(*CurDAG, CurDAG->getEntryNode(), SDLoc(N),
363 CurDAG->getTargetConstant(-1, SDLoc(N), MVT::i32));
364
365 SDValue Glue = M0.getValue(1);
366
367 SmallVector <SDValue, 8> Ops;
368 for (unsigned i = 0, e = N->getNumOperands(); i != e; ++i) {
369 Ops.push_back(N->getOperand(i));
370 }
371 Ops.push_back(Glue);
Matt Arsenaulte6667de2017-12-04 22:18:22 +0000372 return CurDAG->MorphNodeTo(N, N->getOpcode(), N->getVTList(), Ops);
Tom Stellard381a94a2015-05-12 15:00:49 +0000373}
374
Matt Arsenault61cb6fa2015-11-11 00:01:36 +0000375static unsigned selectSGPRVectorRegClassID(unsigned NumVectorElts) {
Matt Arsenaultf1aebbf2015-11-02 23:30:48 +0000376 switch (NumVectorElts) {
377 case 1:
Marek Olsak79c05872016-11-25 17:37:09 +0000378 return AMDGPU::SReg_32_XM0RegClassID;
Matt Arsenaultf1aebbf2015-11-02 23:30:48 +0000379 case 2:
380 return AMDGPU::SReg_64RegClassID;
381 case 4:
382 return AMDGPU::SReg_128RegClassID;
383 case 8:
384 return AMDGPU::SReg_256RegClassID;
385 case 16:
386 return AMDGPU::SReg_512RegClassID;
387 }
388
389 llvm_unreachable("invalid vector size");
390}
391
Matt Arsenaulteb522e62017-02-27 22:15:25 +0000392static bool getConstantValue(SDValue N, uint32_t &Out) {
393 if (const ConstantSDNode *C = dyn_cast<ConstantSDNode>(N)) {
394 Out = C->getAPIntValue().getZExtValue();
395 return true;
396 }
397
398 if (const ConstantFPSDNode *C = dyn_cast<ConstantFPSDNode>(N)) {
399 Out = C->getValueAPF().bitcastToAPInt().getZExtValue();
400 return true;
401 }
402
403 return false;
404}
405
Tom Stellard20287692017-08-08 04:57:55 +0000406void AMDGPUDAGToDAGISel::SelectBuildVector(SDNode *N, unsigned RegClassID) {
Tom Stellard20287692017-08-08 04:57:55 +0000407 EVT VT = N->getValueType(0);
408 unsigned NumVectorElts = VT.getVectorNumElements();
409 EVT EltVT = VT.getVectorElementType();
Tom Stellard20287692017-08-08 04:57:55 +0000410 SDLoc DL(N);
411 SDValue RegClass = CurDAG->getTargetConstant(RegClassID, DL, MVT::i32);
412
413 if (NumVectorElts == 1) {
414 CurDAG->SelectNodeTo(N, AMDGPU::COPY_TO_REGCLASS, EltVT, N->getOperand(0),
415 RegClass);
416 return;
417 }
418
419 assert(NumVectorElts <= 16 && "Vectors with more than 16 elements not "
420 "supported yet");
421 // 16 = Max Num Vector Elements
422 // 2 = 2 REG_SEQUENCE operands per element (value, subreg index)
423 // 1 = Vector Register Class
424 SmallVector<SDValue, 16 * 2 + 1> RegSeqArgs(NumVectorElts * 2 + 1);
425
426 RegSeqArgs[0] = CurDAG->getTargetConstant(RegClassID, DL, MVT::i32);
427 bool IsRegSeq = true;
428 unsigned NOps = N->getNumOperands();
429 for (unsigned i = 0; i < NOps; i++) {
430 // XXX: Why is this here?
431 if (isa<RegisterSDNode>(N->getOperand(i))) {
432 IsRegSeq = false;
433 break;
434 }
Simon Pilgrimede0e402018-05-19 12:46:02 +0000435 unsigned Sub = AMDGPURegisterInfo::getSubRegFromChannel(i);
Tom Stellard20287692017-08-08 04:57:55 +0000436 RegSeqArgs[1 + (2 * i)] = N->getOperand(i);
Simon Pilgrimede0e402018-05-19 12:46:02 +0000437 RegSeqArgs[1 + (2 * i) + 1] = CurDAG->getTargetConstant(Sub, DL, MVT::i32);
Tom Stellard20287692017-08-08 04:57:55 +0000438 }
439 if (NOps != NumVectorElts) {
440 // Fill in the missing undef elements if this was a scalar_to_vector.
Tom Stellard03aa3ae2017-08-08 05:52:00 +0000441 assert(N->getOpcode() == ISD::SCALAR_TO_VECTOR && NOps < NumVectorElts);
Tom Stellard20287692017-08-08 04:57:55 +0000442 MachineSDNode *ImpDef = CurDAG->getMachineNode(TargetOpcode::IMPLICIT_DEF,
443 DL, EltVT);
444 for (unsigned i = NOps; i < NumVectorElts; ++i) {
Simon Pilgrimede0e402018-05-19 12:46:02 +0000445 unsigned Sub = AMDGPURegisterInfo::getSubRegFromChannel(i);
Tom Stellard20287692017-08-08 04:57:55 +0000446 RegSeqArgs[1 + (2 * i)] = SDValue(ImpDef, 0);
447 RegSeqArgs[1 + (2 * i) + 1] =
Simon Pilgrimede0e402018-05-19 12:46:02 +0000448 CurDAG->getTargetConstant(Sub, DL, MVT::i32);
Tom Stellard20287692017-08-08 04:57:55 +0000449 }
450 }
451
452 if (!IsRegSeq)
453 SelectCode(N);
454 CurDAG->SelectNodeTo(N, AMDGPU::REG_SEQUENCE, N->getVTList(), RegSeqArgs);
455}
456
Justin Bogner95927c02016-05-12 21:03:32 +0000457void AMDGPUDAGToDAGISel::Select(SDNode *N) {
Tom Stellard75aadc22012-12-11 21:25:42 +0000458 unsigned int Opc = N->getOpcode();
459 if (N->isMachineOpcode()) {
Tim Northover31d093c2013-09-22 08:21:56 +0000460 N->setNodeId(-1);
Justin Bogner95927c02016-05-12 21:03:32 +0000461 return; // Already selected.
Tom Stellard75aadc22012-12-11 21:25:42 +0000462 }
Matt Arsenault78b86702014-04-18 05:19:26 +0000463
Matt Arsenaulta9dbdca2016-04-12 14:05:04 +0000464 if (isa<AtomicSDNode>(N) ||
Daniil Fukalovd5fca552018-01-17 14:05:05 +0000465 (Opc == AMDGPUISD::ATOMIC_INC || Opc == AMDGPUISD::ATOMIC_DEC ||
466 Opc == AMDGPUISD::ATOMIC_LOAD_FADD ||
467 Opc == AMDGPUISD::ATOMIC_LOAD_FMIN ||
468 Opc == AMDGPUISD::ATOMIC_LOAD_FMAX))
Tom Stellard381a94a2015-05-12 15:00:49 +0000469 N = glueCopyToM0(N);
470
Tom Stellard75aadc22012-12-11 21:25:42 +0000471 switch (Opc) {
Matt Arsenault84445dd2017-11-30 22:51:26 +0000472 default:
473 break;
Tom Stellard1f15bff2014-02-25 21:36:18 +0000474 // We are selecting i64 ADD here instead of custom lower it during
475 // DAG legalization, so we can fold some i64 ADDs used for address
476 // calculation into the LOAD and STORE instructions.
Nicolai Haehnle67624af2016-10-14 10:30:00 +0000477 case ISD::ADDC:
478 case ISD::ADDE:
Nicolai Haehnle67624af2016-10-14 10:30:00 +0000479 case ISD::SUBC:
480 case ISD::SUBE: {
Tom Stellard20287692017-08-08 04:57:55 +0000481 if (N->getValueType(0) != MVT::i64)
Tom Stellard1f15bff2014-02-25 21:36:18 +0000482 break;
483
Justin Bogner95927c02016-05-12 21:03:32 +0000484 SelectADD_SUB_I64(N);
485 return;
Tom Stellard1f15bff2014-02-25 21:36:18 +0000486 }
Matt Arsenaultee3f0ac2017-01-30 18:11:38 +0000487 case ISD::UADDO:
488 case ISD::USUBO: {
489 SelectUADDO_USUBO(N);
490 return;
491 }
Tom Stellard8485fa02016-12-07 02:42:15 +0000492 case AMDGPUISD::FMUL_W_CHAIN: {
493 SelectFMUL_W_CHAIN(N);
494 return;
495 }
496 case AMDGPUISD::FMA_W_CHAIN: {
497 SelectFMA_W_CHAIN(N);
498 return;
499 }
500
Matt Arsenault064c2062014-06-11 17:40:32 +0000501 case ISD::SCALAR_TO_VECTOR:
Vincent Lejeune3b6f20e2013-03-05 15:04:49 +0000502 case ISD::BUILD_VECTOR: {
Tom Stellard8e5da412013-08-14 23:24:32 +0000503 EVT VT = N->getValueType(0);
504 unsigned NumVectorElts = VT.getVectorNumElements();
Matt Arsenault5a4ec812018-06-20 19:45:48 +0000505 if (VT.getScalarSizeInBits() == 16) {
506 if (Opc == ISD::BUILD_VECTOR && NumVectorElts == 2) {
Matt Arsenaulteb522e62017-02-27 22:15:25 +0000507 uint32_t LHSVal, RHSVal;
508 if (getConstantValue(N->getOperand(0), LHSVal) &&
509 getConstantValue(N->getOperand(1), RHSVal)) {
510 uint32_t K = LHSVal | (RHSVal << 16);
511 CurDAG->SelectNodeTo(N, AMDGPU::S_MOV_B32, VT,
512 CurDAG->getTargetConstant(K, SDLoc(N), MVT::i32));
513 return;
514 }
515 }
516
517 break;
518 }
519
Tom Stellard03aa3ae2017-08-08 05:52:00 +0000520 assert(VT.getVectorElementType().bitsEq(MVT::i32));
Tom Stellard20287692017-08-08 04:57:55 +0000521 unsigned RegClassID = selectSGPRVectorRegClassID(NumVectorElts);
522 SelectBuildVector(N, RegClassID);
Justin Bogner95927c02016-05-12 21:03:32 +0000523 return;
Vincent Lejeune3b6f20e2013-03-05 15:04:49 +0000524 }
Tom Stellard754f80f2013-04-05 23:31:51 +0000525 case ISD::BUILD_PAIR: {
526 SDValue RC, SubReg0, SubReg1;
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +0000527 SDLoc DL(N);
Tom Stellard754f80f2013-04-05 23:31:51 +0000528 if (N->getValueType(0) == MVT::i128) {
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +0000529 RC = CurDAG->getTargetConstant(AMDGPU::SReg_128RegClassID, DL, MVT::i32);
530 SubReg0 = CurDAG->getTargetConstant(AMDGPU::sub0_sub1, DL, MVT::i32);
531 SubReg1 = CurDAG->getTargetConstant(AMDGPU::sub2_sub3, DL, MVT::i32);
Tom Stellard754f80f2013-04-05 23:31:51 +0000532 } else if (N->getValueType(0) == MVT::i64) {
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +0000533 RC = CurDAG->getTargetConstant(AMDGPU::SReg_64RegClassID, DL, MVT::i32);
534 SubReg0 = CurDAG->getTargetConstant(AMDGPU::sub0, DL, MVT::i32);
535 SubReg1 = CurDAG->getTargetConstant(AMDGPU::sub1, DL, MVT::i32);
Tom Stellard754f80f2013-04-05 23:31:51 +0000536 } else {
537 llvm_unreachable("Unhandled value type for BUILD_PAIR");
538 }
539 const SDValue Ops[] = { RC, N->getOperand(0), SubReg0,
540 N->getOperand(1), SubReg1 };
Justin Bogner95927c02016-05-12 21:03:32 +0000541 ReplaceNode(N, CurDAG->getMachineNode(TargetOpcode::REG_SEQUENCE, DL,
542 N->getValueType(0), Ops));
543 return;
Tom Stellard754f80f2013-04-05 23:31:51 +0000544 }
Tom Stellard7ed0b522014-04-03 20:19:27 +0000545
546 case ISD::Constant:
547 case ISD::ConstantFP: {
Tom Stellard20287692017-08-08 04:57:55 +0000548 if (N->getValueType(0).getSizeInBits() != 64 || isInlineImmediate(N))
Tom Stellard7ed0b522014-04-03 20:19:27 +0000549 break;
550
551 uint64_t Imm;
552 if (ConstantFPSDNode *FP = dyn_cast<ConstantFPSDNode>(N))
553 Imm = FP->getValueAPF().bitcastToAPInt().getZExtValue();
554 else {
Tom Stellard3cbe0142014-04-07 19:31:13 +0000555 ConstantSDNode *C = cast<ConstantSDNode>(N);
Tom Stellard7ed0b522014-04-03 20:19:27 +0000556 Imm = C->getZExtValue();
557 }
558
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +0000559 SDLoc DL(N);
560 SDNode *Lo = CurDAG->getMachineNode(AMDGPU::S_MOV_B32, DL, MVT::i32,
561 CurDAG->getConstant(Imm & 0xFFFFFFFF, DL,
562 MVT::i32));
563 SDNode *Hi = CurDAG->getMachineNode(AMDGPU::S_MOV_B32, DL, MVT::i32,
564 CurDAG->getConstant(Imm >> 32, DL, MVT::i32));
Tom Stellard7ed0b522014-04-03 20:19:27 +0000565 const SDValue Ops[] = {
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +0000566 CurDAG->getTargetConstant(AMDGPU::SReg_64RegClassID, DL, MVT::i32),
567 SDValue(Lo, 0), CurDAG->getTargetConstant(AMDGPU::sub0, DL, MVT::i32),
568 SDValue(Hi, 0), CurDAG->getTargetConstant(AMDGPU::sub1, DL, MVT::i32)
Tom Stellard7ed0b522014-04-03 20:19:27 +0000569 };
570
Justin Bogner95927c02016-05-12 21:03:32 +0000571 ReplaceNode(N, CurDAG->getMachineNode(TargetOpcode::REG_SEQUENCE, DL,
572 N->getValueType(0), Ops));
573 return;
Tom Stellard7ed0b522014-04-03 20:19:27 +0000574 }
Matt Arsenault4bf43d42015-09-25 17:27:08 +0000575 case ISD::LOAD:
Matt Arsenault3f8e7a32018-06-22 08:39:52 +0000576 case ISD::STORE:
577 case ISD::ATOMIC_LOAD:
578 case ISD::ATOMIC_STORE: {
Tom Stellard381a94a2015-05-12 15:00:49 +0000579 N = glueCopyToM0(N);
Tom Stellard096b8c12015-02-04 20:49:49 +0000580 break;
581 }
Matt Arsenault78b86702014-04-18 05:19:26 +0000582
583 case AMDGPUISD::BFE_I32:
584 case AMDGPUISD::BFE_U32: {
Matt Arsenault78b86702014-04-18 05:19:26 +0000585 // There is a scalar version available, but unlike the vector version which
586 // has a separate operand for the offset and width, the scalar version packs
587 // the width and offset into a single operand. Try to move to the scalar
588 // version if the offsets are constant, so that we can try to keep extended
589 // loads of kernel arguments in SGPRs.
590
591 // TODO: Technically we could try to pattern match scalar bitshifts of
592 // dynamic values, but it's probably not useful.
593 ConstantSDNode *Offset = dyn_cast<ConstantSDNode>(N->getOperand(1));
594 if (!Offset)
595 break;
596
597 ConstantSDNode *Width = dyn_cast<ConstantSDNode>(N->getOperand(2));
598 if (!Width)
599 break;
600
601 bool Signed = Opc == AMDGPUISD::BFE_I32;
602
Matt Arsenault78b86702014-04-18 05:19:26 +0000603 uint32_t OffsetVal = Offset->getZExtValue();
604 uint32_t WidthVal = Width->getZExtValue();
605
Justin Bogner95927c02016-05-12 21:03:32 +0000606 ReplaceNode(N, getS_BFE(Signed ? AMDGPU::S_BFE_I32 : AMDGPU::S_BFE_U32,
607 SDLoc(N), N->getOperand(0), OffsetVal, WidthVal));
608 return;
Matt Arsenault78b86702014-04-18 05:19:26 +0000609 }
Matt Arsenaultf2b0aeb2014-06-23 18:28:28 +0000610 case AMDGPUISD::DIV_SCALE: {
Justin Bogner95927c02016-05-12 21:03:32 +0000611 SelectDIV_SCALE(N);
612 return;
Matt Arsenaultf2b0aeb2014-06-23 18:28:28 +0000613 }
Matt Arsenault4f6318f2017-11-06 17:04:37 +0000614 case AMDGPUISD::MAD_I64_I32:
615 case AMDGPUISD::MAD_U64_U32: {
616 SelectMAD_64_32(N);
617 return;
618 }
Tom Stellard3457a842014-10-09 19:06:00 +0000619 case ISD::CopyToReg: {
620 const SITargetLowering& Lowering =
621 *static_cast<const SITargetLowering*>(getTargetLowering());
Matt Arsenault0d0d6c22017-04-12 21:58:23 +0000622 N = Lowering.legalizeTargetIndependentNode(N, *CurDAG);
Tom Stellard3457a842014-10-09 19:06:00 +0000623 break;
624 }
Marek Olsak9b728682015-03-24 13:40:27 +0000625 case ISD::AND:
626 case ISD::SRL:
627 case ISD::SRA:
Matt Arsenault7e8de012016-04-22 22:59:16 +0000628 case ISD::SIGN_EXTEND_INREG:
Tom Stellard20287692017-08-08 04:57:55 +0000629 if (N->getValueType(0) != MVT::i32)
Marek Olsak9b728682015-03-24 13:40:27 +0000630 break;
631
Justin Bogner95927c02016-05-12 21:03:32 +0000632 SelectS_BFE(N);
633 return;
Tom Stellardbc4497b2016-02-12 23:45:29 +0000634 case ISD::BRCOND:
Justin Bogner95927c02016-05-12 21:03:32 +0000635 SelectBRCOND(N);
636 return;
Matt Arsenaultd7e23032017-09-07 18:05:07 +0000637 case ISD::FMAD:
Matt Arsenault0084adc2018-04-30 19:08:16 +0000638 case ISD::FMA:
639 SelectFMAD_FMA(N);
Matt Arsenaultd7e23032017-09-07 18:05:07 +0000640 return;
Matt Arsenault88701812016-06-09 23:42:48 +0000641 case AMDGPUISD::ATOMIC_CMP_SWAP:
642 SelectATOMIC_CMP_SWAP(N);
643 return;
Matt Arsenault709374d2018-08-01 20:13:58 +0000644 case AMDGPUISD::CVT_PKRTZ_F16_F32:
645 case AMDGPUISD::CVT_PKNORM_I16_F32:
646 case AMDGPUISD::CVT_PKNORM_U16_F32:
647 case AMDGPUISD::CVT_PK_U16_U32:
648 case AMDGPUISD::CVT_PK_I16_I32: {
649 // Hack around using a legal type if f16 is illegal.
650 if (N->getValueType(0) == MVT::i32) {
651 MVT NewVT = Opc == AMDGPUISD::CVT_PKRTZ_F16_F32 ? MVT::v2f16 : MVT::v2i16;
652 N = CurDAG->MorphNodeTo(N, N->getOpcode(), CurDAG->getVTList(NewVT),
653 { N->getOperand(0), N->getOperand(1) });
654 SelectCode(N);
655 return;
656 }
657 }
Tom Stellard75aadc22012-12-11 21:25:42 +0000658 }
Tom Stellard3457a842014-10-09 19:06:00 +0000659
Justin Bogner95927c02016-05-12 21:03:32 +0000660 SelectCode(N);
Tom Stellard365366f2013-01-23 02:09:06 +0000661}
662
Tom Stellardbc4497b2016-02-12 23:45:29 +0000663bool AMDGPUDAGToDAGISel::isUniformBr(const SDNode *N) const {
664 const BasicBlock *BB = FuncInfo->MBB->getBasicBlock();
Nicolai Haehnle05b127d2016-04-14 17:42:35 +0000665 const Instruction *Term = BB->getTerminator();
666 return Term->getMetadata("amdgpu.uniform") ||
667 Term->getMetadata("structurizecfg.uniform");
Tom Stellardbc4497b2016-02-12 23:45:29 +0000668}
669
Mehdi Amini117296c2016-10-01 02:56:57 +0000670StringRef AMDGPUDAGToDAGISel::getPassName() const {
Tom Stellard75aadc22012-12-11 21:25:42 +0000671 return "AMDGPU DAG->DAG Pattern Instruction Selection";
672}
673
Tom Stellard41fc7852013-07-23 01:48:42 +0000674//===----------------------------------------------------------------------===//
675// Complex Patterns
676//===----------------------------------------------------------------------===//
Tom Stellard75aadc22012-12-11 21:25:42 +0000677
Tom Stellard75aadc22012-12-11 21:25:42 +0000678bool AMDGPUDAGToDAGISel::SelectADDRVTX_READ(SDValue Addr, SDValue &Base,
Tom Stellard20287692017-08-08 04:57:55 +0000679 SDValue &Offset) {
680 return false;
Tom Stellard75aadc22012-12-11 21:25:42 +0000681}
682
Tom Stellardf3b2a1e2013-02-06 17:32:29 +0000683bool AMDGPUDAGToDAGISel::SelectADDRIndirect(SDValue Addr, SDValue &Base,
684 SDValue &Offset) {
685 ConstantSDNode *C;
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +0000686 SDLoc DL(Addr);
Tom Stellardf3b2a1e2013-02-06 17:32:29 +0000687
688 if ((C = dyn_cast<ConstantSDNode>(Addr))) {
Tom Stellardc5a154d2018-06-28 23:47:12 +0000689 Base = CurDAG->getRegister(R600::INDIRECT_BASE_ADDR, MVT::i32);
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +0000690 Offset = CurDAG->getTargetConstant(C->getZExtValue(), DL, MVT::i32);
Jan Vesely06200bd2017-01-06 21:00:46 +0000691 } else if ((Addr.getOpcode() == AMDGPUISD::DWORDADDR) &&
692 (C = dyn_cast<ConstantSDNode>(Addr.getOperand(0)))) {
Tom Stellardc5a154d2018-06-28 23:47:12 +0000693 Base = CurDAG->getRegister(R600::INDIRECT_BASE_ADDR, MVT::i32);
Jan Vesely06200bd2017-01-06 21:00:46 +0000694 Offset = CurDAG->getTargetConstant(C->getZExtValue(), DL, MVT::i32);
Tom Stellardf3b2a1e2013-02-06 17:32:29 +0000695 } else if ((Addr.getOpcode() == ISD::ADD || Addr.getOpcode() == ISD::OR) &&
696 (C = dyn_cast<ConstantSDNode>(Addr.getOperand(1)))) {
697 Base = Addr.getOperand(0);
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +0000698 Offset = CurDAG->getTargetConstant(C->getZExtValue(), DL, MVT::i32);
Tom Stellardf3b2a1e2013-02-06 17:32:29 +0000699 } else {
700 Base = Addr;
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +0000701 Offset = CurDAG->getTargetConstant(0, DL, MVT::i32);
Tom Stellardf3b2a1e2013-02-06 17:32:29 +0000702 }
703
704 return true;
705}
Christian Konigd910b7d2013-02-26 17:52:16 +0000706
Matt Arsenault84445dd2017-11-30 22:51:26 +0000707// FIXME: Should only handle addcarry/subcarry
Justin Bogner95927c02016-05-12 21:03:32 +0000708void AMDGPUDAGToDAGISel::SelectADD_SUB_I64(SDNode *N) {
Matt Arsenault9fa3f932014-06-23 18:00:34 +0000709 SDLoc DL(N);
710 SDValue LHS = N->getOperand(0);
711 SDValue RHS = N->getOperand(1);
712
Nicolai Haehnle67624af2016-10-14 10:30:00 +0000713 unsigned Opcode = N->getOpcode();
714 bool ConsumeCarry = (Opcode == ISD::ADDE || Opcode == ISD::SUBE);
715 bool ProduceCarry =
716 ConsumeCarry || Opcode == ISD::ADDC || Opcode == ISD::SUBC;
Matt Arsenault84445dd2017-11-30 22:51:26 +0000717 bool IsAdd = Opcode == ISD::ADD || Opcode == ISD::ADDC || Opcode == ISD::ADDE;
Matt Arsenaultb8b51532014-06-23 18:00:38 +0000718
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +0000719 SDValue Sub0 = CurDAG->getTargetConstant(AMDGPU::sub0, DL, MVT::i32);
720 SDValue Sub1 = CurDAG->getTargetConstant(AMDGPU::sub1, DL, MVT::i32);
Matt Arsenault9fa3f932014-06-23 18:00:34 +0000721
722 SDNode *Lo0 = CurDAG->getMachineNode(TargetOpcode::EXTRACT_SUBREG,
723 DL, MVT::i32, LHS, Sub0);
724 SDNode *Hi0 = CurDAG->getMachineNode(TargetOpcode::EXTRACT_SUBREG,
725 DL, MVT::i32, LHS, Sub1);
726
727 SDNode *Lo1 = CurDAG->getMachineNode(TargetOpcode::EXTRACT_SUBREG,
728 DL, MVT::i32, RHS, Sub0);
729 SDNode *Hi1 = CurDAG->getMachineNode(TargetOpcode::EXTRACT_SUBREG,
730 DL, MVT::i32, RHS, Sub1);
731
732 SDVTList VTList = CurDAG->getVTList(MVT::i32, MVT::Glue);
Matt Arsenault9fa3f932014-06-23 18:00:34 +0000733
Tom Stellard80942a12014-09-05 14:07:59 +0000734 unsigned Opc = IsAdd ? AMDGPU::S_ADD_U32 : AMDGPU::S_SUB_U32;
Matt Arsenaultb8b51532014-06-23 18:00:38 +0000735 unsigned CarryOpc = IsAdd ? AMDGPU::S_ADDC_U32 : AMDGPU::S_SUBB_U32;
736
Nicolai Haehnle67624af2016-10-14 10:30:00 +0000737 SDNode *AddLo;
738 if (!ConsumeCarry) {
739 SDValue Args[] = { SDValue(Lo0, 0), SDValue(Lo1, 0) };
740 AddLo = CurDAG->getMachineNode(Opc, DL, VTList, Args);
741 } else {
742 SDValue Args[] = { SDValue(Lo0, 0), SDValue(Lo1, 0), N->getOperand(2) };
743 AddLo = CurDAG->getMachineNode(CarryOpc, DL, VTList, Args);
744 }
745 SDValue AddHiArgs[] = {
746 SDValue(Hi0, 0),
747 SDValue(Hi1, 0),
748 SDValue(AddLo, 1)
749 };
750 SDNode *AddHi = CurDAG->getMachineNode(CarryOpc, DL, VTList, AddHiArgs);
Matt Arsenault9fa3f932014-06-23 18:00:34 +0000751
Nicolai Haehnle67624af2016-10-14 10:30:00 +0000752 SDValue RegSequenceArgs[] = {
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +0000753 CurDAG->getTargetConstant(AMDGPU::SReg_64RegClassID, DL, MVT::i32),
Matt Arsenault9fa3f932014-06-23 18:00:34 +0000754 SDValue(AddLo,0),
755 Sub0,
756 SDValue(AddHi,0),
757 Sub1,
758 };
Nicolai Haehnle67624af2016-10-14 10:30:00 +0000759 SDNode *RegSequence = CurDAG->getMachineNode(AMDGPU::REG_SEQUENCE, DL,
760 MVT::i64, RegSequenceArgs);
761
762 if (ProduceCarry) {
763 // Replace the carry-use
Nirav Dave3264c1b2018-03-19 20:19:46 +0000764 ReplaceUses(SDValue(N, 1), SDValue(AddHi, 1));
Nicolai Haehnle67624af2016-10-14 10:30:00 +0000765 }
766
767 // Replace the remaining uses.
Nirav Dave3264c1b2018-03-19 20:19:46 +0000768 ReplaceNode(N, RegSequence);
Matt Arsenault9fa3f932014-06-23 18:00:34 +0000769}
770
Matt Arsenaultee3f0ac2017-01-30 18:11:38 +0000771void AMDGPUDAGToDAGISel::SelectUADDO_USUBO(SDNode *N) {
772 // The name of the opcodes are misleading. v_add_i32/v_sub_i32 have unsigned
773 // carry out despite the _i32 name. These were renamed in VI to _U32.
774 // FIXME: We should probably rename the opcodes here.
775 unsigned Opc = N->getOpcode() == ISD::UADDO ?
776 AMDGPU::V_ADD_I32_e64 : AMDGPU::V_SUB_I32_e64;
777
778 CurDAG->SelectNodeTo(N, Opc, N->getVTList(),
779 { N->getOperand(0), N->getOperand(1) });
780}
781
Tom Stellard8485fa02016-12-07 02:42:15 +0000782void AMDGPUDAGToDAGISel::SelectFMA_W_CHAIN(SDNode *N) {
783 SDLoc SL(N);
784 // src0_modifiers, src0, src1_modifiers, src1, src2_modifiers, src2, clamp, omod
785 SDValue Ops[10];
786
787 SelectVOP3Mods0(N->getOperand(1), Ops[1], Ops[0], Ops[6], Ops[7]);
788 SelectVOP3Mods(N->getOperand(2), Ops[3], Ops[2]);
789 SelectVOP3Mods(N->getOperand(3), Ops[5], Ops[4]);
790 Ops[8] = N->getOperand(0);
791 Ops[9] = N->getOperand(4);
792
793 CurDAG->SelectNodeTo(N, AMDGPU::V_FMA_F32, N->getVTList(), Ops);
794}
795
796void AMDGPUDAGToDAGISel::SelectFMUL_W_CHAIN(SDNode *N) {
797 SDLoc SL(N);
NAKAMURA Takumi6f43bd42017-10-18 13:31:28 +0000798 // src0_modifiers, src0, src1_modifiers, src1, clamp, omod
Tom Stellard8485fa02016-12-07 02:42:15 +0000799 SDValue Ops[8];
800
801 SelectVOP3Mods0(N->getOperand(1), Ops[1], Ops[0], Ops[4], Ops[5]);
802 SelectVOP3Mods(N->getOperand(2), Ops[3], Ops[2]);
803 Ops[6] = N->getOperand(0);
804 Ops[7] = N->getOperand(3);
805
806 CurDAG->SelectNodeTo(N, AMDGPU::V_MUL_F32_e64, N->getVTList(), Ops);
807}
808
Matt Arsenault044f1d12015-02-14 04:24:28 +0000809// We need to handle this here because tablegen doesn't support matching
810// instructions with multiple outputs.
Justin Bogner95927c02016-05-12 21:03:32 +0000811void AMDGPUDAGToDAGISel::SelectDIV_SCALE(SDNode *N) {
Matt Arsenaultf2b0aeb2014-06-23 18:28:28 +0000812 SDLoc SL(N);
813 EVT VT = N->getValueType(0);
814
815 assert(VT == MVT::f32 || VT == MVT::f64);
816
817 unsigned Opc
818 = (VT == MVT::f64) ? AMDGPU::V_DIV_SCALE_F64 : AMDGPU::V_DIV_SCALE_F32;
819
Matt Arsenault3b99f122017-01-19 06:04:12 +0000820 SDValue Ops[] = { N->getOperand(0), N->getOperand(1), N->getOperand(2) };
821 CurDAG->SelectNodeTo(N, Opc, N->getVTList(), Ops);
Matt Arsenaultf2b0aeb2014-06-23 18:28:28 +0000822}
823
Matt Arsenault4f6318f2017-11-06 17:04:37 +0000824// We need to handle this here because tablegen doesn't support matching
825// instructions with multiple outputs.
826void AMDGPUDAGToDAGISel::SelectMAD_64_32(SDNode *N) {
827 SDLoc SL(N);
828 bool Signed = N->getOpcode() == AMDGPUISD::MAD_I64_I32;
829 unsigned Opc = Signed ? AMDGPU::V_MAD_I64_I32 : AMDGPU::V_MAD_U64_U32;
830
831 SDValue Clamp = CurDAG->getTargetConstant(0, SL, MVT::i1);
832 SDValue Ops[] = { N->getOperand(0), N->getOperand(1), N->getOperand(2),
833 Clamp };
834 CurDAG->SelectNodeTo(N, Opc, N->getVTList(), Ops);
835}
836
Tom Stellard85e8b6d2014-08-22 18:49:33 +0000837bool AMDGPUDAGToDAGISel::isDSOffsetLegal(const SDValue &Base, unsigned Offset,
838 unsigned OffsetBits) const {
Tom Stellard85e8b6d2014-08-22 18:49:33 +0000839 if ((OffsetBits == 16 && !isUInt<16>(Offset)) ||
840 (OffsetBits == 8 && !isUInt<8>(Offset)))
841 return false;
842
Matt Arsenault706f9302015-07-06 16:01:58 +0000843 if (Subtarget->getGeneration() >= AMDGPUSubtarget::SEA_ISLANDS ||
844 Subtarget->unsafeDSOffsetFoldingEnabled())
Tom Stellard85e8b6d2014-08-22 18:49:33 +0000845 return true;
846
847 // On Southern Islands instruction with a negative base value and an offset
848 // don't seem to work.
849 return CurDAG->SignBitIsZero(Base);
850}
851
852bool AMDGPUDAGToDAGISel::SelectDS1Addr1Offset(SDValue Addr, SDValue &Base,
853 SDValue &Offset) const {
Tom Stellard92b24f32016-04-29 14:34:26 +0000854 SDLoc DL(Addr);
Tom Stellard85e8b6d2014-08-22 18:49:33 +0000855 if (CurDAG->isBaseWithConstantOffset(Addr)) {
856 SDValue N0 = Addr.getOperand(0);
857 SDValue N1 = Addr.getOperand(1);
858 ConstantSDNode *C1 = cast<ConstantSDNode>(N1);
859 if (isDSOffsetLegal(N0, C1->getSExtValue(), 16)) {
860 // (add n0, c0)
861 Base = N0;
Tom Stellard92b24f32016-04-29 14:34:26 +0000862 Offset = CurDAG->getTargetConstant(C1->getZExtValue(), DL, MVT::i16);
Tom Stellard85e8b6d2014-08-22 18:49:33 +0000863 return true;
864 }
Matt Arsenault966a94f2015-09-08 19:34:22 +0000865 } else if (Addr.getOpcode() == ISD::SUB) {
866 // sub C, x -> add (sub 0, x), C
867 if (const ConstantSDNode *C = dyn_cast<ConstantSDNode>(Addr.getOperand(0))) {
868 int64_t ByteOffset = C->getSExtValue();
869 if (isUInt<16>(ByteOffset)) {
Matt Arsenault966a94f2015-09-08 19:34:22 +0000870 SDValue Zero = CurDAG->getTargetConstant(0, DL, MVT::i32);
Tom Stellard85e8b6d2014-08-22 18:49:33 +0000871
Matt Arsenault966a94f2015-09-08 19:34:22 +0000872 // XXX - This is kind of hacky. Create a dummy sub node so we can check
873 // the known bits in isDSOffsetLegal. We need to emit the selected node
874 // here, so this is thrown away.
875 SDValue Sub = CurDAG->getNode(ISD::SUB, DL, MVT::i32,
876 Zero, Addr.getOperand(1));
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +0000877
Matt Arsenault966a94f2015-09-08 19:34:22 +0000878 if (isDSOffsetLegal(Sub, ByteOffset, 16)) {
Matt Arsenault84445dd2017-11-30 22:51:26 +0000879 // FIXME: Select to VOP3 version for with-carry.
880 unsigned SubOp = Subtarget->hasAddNoCarry() ?
881 AMDGPU::V_SUB_U32_e64 : AMDGPU::V_SUB_I32_e32;
882
Matt Arsenault966a94f2015-09-08 19:34:22 +0000883 MachineSDNode *MachineSub
Matt Arsenault84445dd2017-11-30 22:51:26 +0000884 = CurDAG->getMachineNode(SubOp, DL, MVT::i32,
Matt Arsenault966a94f2015-09-08 19:34:22 +0000885 Zero, Addr.getOperand(1));
886
887 Base = SDValue(MachineSub, 0);
Tom Stellard26a2ab72016-06-10 00:01:04 +0000888 Offset = CurDAG->getTargetConstant(ByteOffset, DL, MVT::i16);
Matt Arsenault966a94f2015-09-08 19:34:22 +0000889 return true;
890 }
891 }
892 }
893 } else if (const ConstantSDNode *CAddr = dyn_cast<ConstantSDNode>(Addr)) {
894 // If we have a constant address, prefer to put the constant into the
895 // offset. This can save moves to load the constant address since multiple
896 // operations can share the zero base address register, and enables merging
897 // into read2 / write2 instructions.
898
899 SDLoc DL(Addr);
900
Matt Arsenaulte775f5f2014-10-14 17:21:19 +0000901 if (isUInt<16>(CAddr->getZExtValue())) {
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +0000902 SDValue Zero = CurDAG->getTargetConstant(0, DL, MVT::i32);
Tom Stellardc8d79202014-10-15 21:08:59 +0000903 MachineSDNode *MovZero = CurDAG->getMachineNode(AMDGPU::V_MOV_B32_e32,
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +0000904 DL, MVT::i32, Zero);
Tom Stellardc8d79202014-10-15 21:08:59 +0000905 Base = SDValue(MovZero, 0);
Tom Stellard26a2ab72016-06-10 00:01:04 +0000906 Offset = CurDAG->getTargetConstant(CAddr->getZExtValue(), DL, MVT::i16);
Matt Arsenaulte775f5f2014-10-14 17:21:19 +0000907 return true;
908 }
909 }
910
Tom Stellard85e8b6d2014-08-22 18:49:33 +0000911 // default case
912 Base = Addr;
Matt Arsenault966a94f2015-09-08 19:34:22 +0000913 Offset = CurDAG->getTargetConstant(0, SDLoc(Addr), MVT::i16);
Tom Stellard85e8b6d2014-08-22 18:49:33 +0000914 return true;
915}
916
Matt Arsenault966a94f2015-09-08 19:34:22 +0000917// TODO: If offset is too big, put low 16-bit into offset.
Tom Stellardf3fc5552014-08-22 18:49:35 +0000918bool AMDGPUDAGToDAGISel::SelectDS64Bit4ByteAligned(SDValue Addr, SDValue &Base,
919 SDValue &Offset0,
920 SDValue &Offset1) const {
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +0000921 SDLoc DL(Addr);
922
Tom Stellardf3fc5552014-08-22 18:49:35 +0000923 if (CurDAG->isBaseWithConstantOffset(Addr)) {
924 SDValue N0 = Addr.getOperand(0);
925 SDValue N1 = Addr.getOperand(1);
926 ConstantSDNode *C1 = cast<ConstantSDNode>(N1);
927 unsigned DWordOffset0 = C1->getZExtValue() / 4;
928 unsigned DWordOffset1 = DWordOffset0 + 1;
929 // (add n0, c0)
930 if (isDSOffsetLegal(N0, DWordOffset1, 8)) {
931 Base = N0;
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +0000932 Offset0 = CurDAG->getTargetConstant(DWordOffset0, DL, MVT::i8);
933 Offset1 = CurDAG->getTargetConstant(DWordOffset1, DL, MVT::i8);
Tom Stellardf3fc5552014-08-22 18:49:35 +0000934 return true;
935 }
Matt Arsenault966a94f2015-09-08 19:34:22 +0000936 } else if (Addr.getOpcode() == ISD::SUB) {
937 // sub C, x -> add (sub 0, x), C
938 if (const ConstantSDNode *C = dyn_cast<ConstantSDNode>(Addr.getOperand(0))) {
939 unsigned DWordOffset0 = C->getZExtValue() / 4;
940 unsigned DWordOffset1 = DWordOffset0 + 1;
Tom Stellardf3fc5552014-08-22 18:49:35 +0000941
Matt Arsenault966a94f2015-09-08 19:34:22 +0000942 if (isUInt<8>(DWordOffset0)) {
943 SDLoc DL(Addr);
944 SDValue Zero = CurDAG->getTargetConstant(0, DL, MVT::i32);
945
946 // XXX - This is kind of hacky. Create a dummy sub node so we can check
947 // the known bits in isDSOffsetLegal. We need to emit the selected node
948 // here, so this is thrown away.
949 SDValue Sub = CurDAG->getNode(ISD::SUB, DL, MVT::i32,
950 Zero, Addr.getOperand(1));
951
952 if (isDSOffsetLegal(Sub, DWordOffset1, 8)) {
Matt Arsenault84445dd2017-11-30 22:51:26 +0000953 unsigned SubOp = Subtarget->hasAddNoCarry() ?
954 AMDGPU::V_SUB_U32_e64 : AMDGPU::V_SUB_I32_e32;
955
Matt Arsenault966a94f2015-09-08 19:34:22 +0000956 MachineSDNode *MachineSub
Matt Arsenault84445dd2017-11-30 22:51:26 +0000957 = CurDAG->getMachineNode(SubOp, DL, MVT::i32,
Matt Arsenault966a94f2015-09-08 19:34:22 +0000958 Zero, Addr.getOperand(1));
959
960 Base = SDValue(MachineSub, 0);
961 Offset0 = CurDAG->getTargetConstant(DWordOffset0, DL, MVT::i8);
962 Offset1 = CurDAG->getTargetConstant(DWordOffset1, DL, MVT::i8);
963 return true;
964 }
965 }
966 }
967 } else if (const ConstantSDNode *CAddr = dyn_cast<ConstantSDNode>(Addr)) {
Matt Arsenault1a74aff2014-10-15 18:06:43 +0000968 unsigned DWordOffset0 = CAddr->getZExtValue() / 4;
969 unsigned DWordOffset1 = DWordOffset0 + 1;
970 assert(4 * DWordOffset0 == CAddr->getZExtValue());
971
972 if (isUInt<8>(DWordOffset0) && isUInt<8>(DWordOffset1)) {
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +0000973 SDValue Zero = CurDAG->getTargetConstant(0, DL, MVT::i32);
Matt Arsenault1a74aff2014-10-15 18:06:43 +0000974 MachineSDNode *MovZero
975 = CurDAG->getMachineNode(AMDGPU::V_MOV_B32_e32,
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +0000976 DL, MVT::i32, Zero);
Matt Arsenault1a74aff2014-10-15 18:06:43 +0000977 Base = SDValue(MovZero, 0);
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +0000978 Offset0 = CurDAG->getTargetConstant(DWordOffset0, DL, MVT::i8);
979 Offset1 = CurDAG->getTargetConstant(DWordOffset1, DL, MVT::i8);
Matt Arsenault1a74aff2014-10-15 18:06:43 +0000980 return true;
981 }
982 }
983
Tom Stellardf3fc5552014-08-22 18:49:35 +0000984 // default case
Matt Arsenault0efdd062016-09-09 22:29:28 +0000985
986 // FIXME: This is broken on SI where we still need to check if the base
987 // pointer is positive here.
Tom Stellardf3fc5552014-08-22 18:49:35 +0000988 Base = Addr;
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +0000989 Offset0 = CurDAG->getTargetConstant(0, DL, MVT::i8);
990 Offset1 = CurDAG->getTargetConstant(1, DL, MVT::i8);
Tom Stellardf3fc5552014-08-22 18:49:35 +0000991 return true;
992}
993
Changpeng Fangb41574a2015-12-22 20:55:23 +0000994bool AMDGPUDAGToDAGISel::SelectMUBUF(SDValue Addr, SDValue &Ptr,
Tom Stellard155bbb72014-08-11 22:18:17 +0000995 SDValue &VAddr, SDValue &SOffset,
996 SDValue &Offset, SDValue &Offen,
997 SDValue &Idxen, SDValue &Addr64,
998 SDValue &GLC, SDValue &SLC,
999 SDValue &TFE) const {
Changpeng Fangb41574a2015-12-22 20:55:23 +00001000 // Subtarget prefers to use flat instruction
1001 if (Subtarget->useFlatForGlobal())
1002 return false;
1003
Tom Stellardb02c2682014-06-24 23:33:07 +00001004 SDLoc DL(Addr);
1005
Jan Vesely43b7b5b2016-04-07 19:23:11 +00001006 if (!GLC.getNode())
1007 GLC = CurDAG->getTargetConstant(0, DL, MVT::i1);
1008 if (!SLC.getNode())
1009 SLC = CurDAG->getTargetConstant(0, DL, MVT::i1);
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +00001010 TFE = CurDAG->getTargetConstant(0, DL, MVT::i1);
Tom Stellard155bbb72014-08-11 22:18:17 +00001011
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +00001012 Idxen = CurDAG->getTargetConstant(0, DL, MVT::i1);
1013 Offen = CurDAG->getTargetConstant(0, DL, MVT::i1);
1014 Addr64 = CurDAG->getTargetConstant(0, DL, MVT::i1);
1015 SOffset = CurDAG->getTargetConstant(0, DL, MVT::i32);
Tom Stellard155bbb72014-08-11 22:18:17 +00001016
Tom Stellardb02c2682014-06-24 23:33:07 +00001017 if (CurDAG->isBaseWithConstantOffset(Addr)) {
1018 SDValue N0 = Addr.getOperand(0);
1019 SDValue N1 = Addr.getOperand(1);
1020 ConstantSDNode *C1 = cast<ConstantSDNode>(N1);
1021
Tom Stellard94b72312015-02-11 00:34:35 +00001022 if (N0.getOpcode() == ISD::ADD) {
1023 // (add (add N2, N3), C1) -> addr64
1024 SDValue N2 = N0.getOperand(0);
1025 SDValue N3 = N0.getOperand(1);
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +00001026 Addr64 = CurDAG->getTargetConstant(1, DL, MVT::i1);
Tom Stellard94b72312015-02-11 00:34:35 +00001027 Ptr = N2;
1028 VAddr = N3;
1029 } else {
Tom Stellard155bbb72014-08-11 22:18:17 +00001030 // (add N0, C1) -> offset
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +00001031 VAddr = CurDAG->getTargetConstant(0, DL, MVT::i32);
Tom Stellard155bbb72014-08-11 22:18:17 +00001032 Ptr = N0;
Tom Stellard94b72312015-02-11 00:34:35 +00001033 }
1034
Marek Olsakffadcb72017-11-09 01:52:17 +00001035 if (SIInstrInfo::isLegalMUBUFImmOffset(C1->getZExtValue())) {
Matt Arsenault88701812016-06-09 23:42:48 +00001036 Offset = CurDAG->getTargetConstant(C1->getZExtValue(), DL, MVT::i16);
1037 return true;
1038 }
1039
1040 if (isUInt<32>(C1->getZExtValue())) {
Tom Stellard94b72312015-02-11 00:34:35 +00001041 // Illegal offset, store it in soffset.
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +00001042 Offset = CurDAG->getTargetConstant(0, DL, MVT::i16);
Tom Stellard94b72312015-02-11 00:34:35 +00001043 SOffset = SDValue(CurDAG->getMachineNode(AMDGPU::S_MOV_B32, DL, MVT::i32,
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +00001044 CurDAG->getTargetConstant(C1->getZExtValue(), DL, MVT::i32)),
1045 0);
Changpeng Fangb41574a2015-12-22 20:55:23 +00001046 return true;
Tom Stellardb02c2682014-06-24 23:33:07 +00001047 }
1048 }
Tom Stellard94b72312015-02-11 00:34:35 +00001049
Tom Stellardb02c2682014-06-24 23:33:07 +00001050 if (Addr.getOpcode() == ISD::ADD) {
Tom Stellard155bbb72014-08-11 22:18:17 +00001051 // (add N0, N1) -> addr64
Tom Stellardb02c2682014-06-24 23:33:07 +00001052 SDValue N0 = Addr.getOperand(0);
1053 SDValue N1 = Addr.getOperand(1);
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +00001054 Addr64 = CurDAG->getTargetConstant(1, DL, MVT::i1);
Tom Stellard155bbb72014-08-11 22:18:17 +00001055 Ptr = N0;
1056 VAddr = N1;
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +00001057 Offset = CurDAG->getTargetConstant(0, DL, MVT::i16);
Changpeng Fangb41574a2015-12-22 20:55:23 +00001058 return true;
Tom Stellardb02c2682014-06-24 23:33:07 +00001059 }
1060
Tom Stellard155bbb72014-08-11 22:18:17 +00001061 // default case -> offset
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +00001062 VAddr = CurDAG->getTargetConstant(0, DL, MVT::i32);
Tom Stellard155bbb72014-08-11 22:18:17 +00001063 Ptr = Addr;
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +00001064 Offset = CurDAG->getTargetConstant(0, DL, MVT::i16);
Changpeng Fangb41574a2015-12-22 20:55:23 +00001065
1066 return true;
Tom Stellard155bbb72014-08-11 22:18:17 +00001067}
1068
1069bool AMDGPUDAGToDAGISel::SelectMUBUFAddr64(SDValue Addr, SDValue &SRsrc,
Tom Stellardc53861a2015-02-11 00:34:32 +00001070 SDValue &VAddr, SDValue &SOffset,
Tom Stellard1f9939f2015-02-27 14:59:41 +00001071 SDValue &Offset, SDValue &GLC,
1072 SDValue &SLC, SDValue &TFE) const {
1073 SDValue Ptr, Offen, Idxen, Addr64;
Tom Stellard155bbb72014-08-11 22:18:17 +00001074
Tom Stellard70580f82015-07-20 14:28:41 +00001075 // addr64 bit was removed for volcanic islands.
1076 if (Subtarget->getGeneration() >= AMDGPUSubtarget::VOLCANIC_ISLANDS)
1077 return false;
1078
Changpeng Fangb41574a2015-12-22 20:55:23 +00001079 if (!SelectMUBUF(Addr, Ptr, VAddr, SOffset, Offset, Offen, Idxen, Addr64,
1080 GLC, SLC, TFE))
1081 return false;
Tom Stellard155bbb72014-08-11 22:18:17 +00001082
1083 ConstantSDNode *C = cast<ConstantSDNode>(Addr64);
1084 if (C->getSExtValue()) {
1085 SDLoc DL(Addr);
Matt Arsenault485defe2014-11-05 19:01:17 +00001086
1087 const SITargetLowering& Lowering =
1088 *static_cast<const SITargetLowering*>(getTargetLowering());
1089
1090 SRsrc = SDValue(Lowering.wrapAddr64Rsrc(*CurDAG, DL, Ptr), 0);
Tom Stellard155bbb72014-08-11 22:18:17 +00001091 return true;
1092 }
Matt Arsenault485defe2014-11-05 19:01:17 +00001093
Tom Stellard155bbb72014-08-11 22:18:17 +00001094 return false;
1095}
1096
Tom Stellard7980fc82014-09-25 18:30:26 +00001097bool AMDGPUDAGToDAGISel::SelectMUBUFAddr64(SDValue Addr, SDValue &SRsrc,
Tom Stellardc53861a2015-02-11 00:34:32 +00001098 SDValue &VAddr, SDValue &SOffset,
NAKAMURA Takumi0a7d0ad2015-09-22 11:15:07 +00001099 SDValue &Offset,
1100 SDValue &SLC) const {
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +00001101 SLC = CurDAG->getTargetConstant(0, SDLoc(Addr), MVT::i1);
Tom Stellard1f9939f2015-02-27 14:59:41 +00001102 SDValue GLC, TFE;
Tom Stellard7980fc82014-09-25 18:30:26 +00001103
Tom Stellard1f9939f2015-02-27 14:59:41 +00001104 return SelectMUBUFAddr64(Addr, SRsrc, VAddr, SOffset, Offset, GLC, SLC, TFE);
Tom Stellard7980fc82014-09-25 18:30:26 +00001105}
1106
Matt Arsenault156d3ae2017-05-17 21:02:58 +00001107static bool isStackPtrRelative(const MachinePointerInfo &PtrInfo) {
1108 auto PSV = PtrInfo.V.dyn_cast<const PseudoSourceValue *>();
1109 return PSV && PSV->isStack();
Matt Arsenaultac0fc842016-09-17 16:09:55 +00001110}
1111
Matt Arsenault156d3ae2017-05-17 21:02:58 +00001112std::pair<SDValue, SDValue> AMDGPUDAGToDAGISel::foldFrameIndex(SDValue N) const {
1113 const MachineFunction &MF = CurDAG->getMachineFunction();
1114 const SIMachineFunctionInfo *Info = MF.getInfo<SIMachineFunctionInfo>();
1115
1116 if (auto FI = dyn_cast<FrameIndexSDNode>(N)) {
1117 SDValue TFI = CurDAG->getTargetFrameIndex(FI->getIndex(),
1118 FI->getValueType(0));
1119
1120 // If we can resolve this to a frame index access, this is relative to the
1121 // frame pointer SGPR.
1122 return std::make_pair(TFI, CurDAG->getRegister(Info->getFrameOffsetReg(),
1123 MVT::i32));
1124 }
1125
1126 // If we don't know this private access is a local stack object, it needs to
1127 // be relative to the entry point's scratch wave offset register.
1128 return std::make_pair(N, CurDAG->getRegister(Info->getScratchWaveOffsetReg(),
1129 MVT::i32));
1130}
1131
Matt Arsenaultb81495d2017-09-20 05:01:53 +00001132bool AMDGPUDAGToDAGISel::SelectMUBUFScratchOffen(SDNode *Parent,
Matt Arsenault156d3ae2017-05-17 21:02:58 +00001133 SDValue Addr, SDValue &Rsrc,
Matt Arsenault0774ea22017-04-24 19:40:59 +00001134 SDValue &VAddr, SDValue &SOffset,
1135 SDValue &ImmOffset) const {
Tom Stellardb02094e2014-07-21 15:45:01 +00001136
1137 SDLoc DL(Addr);
1138 MachineFunction &MF = CurDAG->getMachineFunction();
Matt Arsenault0e3d3892015-11-30 21:15:53 +00001139 const SIMachineFunctionInfo *Info = MF.getInfo<SIMachineFunctionInfo>();
Tom Stellardb02094e2014-07-21 15:45:01 +00001140
Matt Arsenault0e3d3892015-11-30 21:15:53 +00001141 Rsrc = CurDAG->getRegister(Info->getScratchRSrcReg(), MVT::v4i32);
Tom Stellardb02094e2014-07-21 15:45:01 +00001142
Matt Arsenault0774ea22017-04-24 19:40:59 +00001143 if (ConstantSDNode *CAddr = dyn_cast<ConstantSDNode>(Addr)) {
1144 unsigned Imm = CAddr->getZExtValue();
Matt Arsenault0774ea22017-04-24 19:40:59 +00001145
1146 SDValue HighBits = CurDAG->getTargetConstant(Imm & ~4095, DL, MVT::i32);
1147 MachineSDNode *MovHighBits = CurDAG->getMachineNode(AMDGPU::V_MOV_B32_e32,
1148 DL, MVT::i32, HighBits);
1149 VAddr = SDValue(MovHighBits, 0);
Matt Arsenault156d3ae2017-05-17 21:02:58 +00001150
1151 // In a call sequence, stores to the argument stack area are relative to the
1152 // stack pointer.
Matt Arsenaultb81495d2017-09-20 05:01:53 +00001153 const MachinePointerInfo &PtrInfo = cast<MemSDNode>(Parent)->getPointerInfo();
Matt Arsenault156d3ae2017-05-17 21:02:58 +00001154 unsigned SOffsetReg = isStackPtrRelative(PtrInfo) ?
1155 Info->getStackPtrOffsetReg() : Info->getScratchWaveOffsetReg();
1156
1157 SOffset = CurDAG->getRegister(SOffsetReg, MVT::i32);
Matt Arsenault0774ea22017-04-24 19:40:59 +00001158 ImmOffset = CurDAG->getTargetConstant(Imm & 4095, DL, MVT::i16);
1159 return true;
1160 }
1161
Tom Stellardb02094e2014-07-21 15:45:01 +00001162 if (CurDAG->isBaseWithConstantOffset(Addr)) {
Matt Arsenault0774ea22017-04-24 19:40:59 +00001163 // (add n0, c1)
1164
Tom Stellard78655fc2015-07-16 19:40:09 +00001165 SDValue N0 = Addr.getOperand(0);
Tom Stellardb02094e2014-07-21 15:45:01 +00001166 SDValue N1 = Addr.getOperand(1);
Matt Arsenaultcd099612016-02-24 04:55:29 +00001167
Matt Arsenaultcaf0ed42017-11-30 00:52:40 +00001168 // Offsets in vaddr must be positive if range checking is enabled.
Matt Arsenault45b98182017-11-15 00:45:43 +00001169 //
Matt Arsenaultcaf0ed42017-11-30 00:52:40 +00001170 // The total computation of vaddr + soffset + offset must not overflow. If
1171 // vaddr is negative, even if offset is 0 the sgpr offset add will end up
Matt Arsenault45b98182017-11-15 00:45:43 +00001172 // overflowing.
Matt Arsenaultcaf0ed42017-11-30 00:52:40 +00001173 //
1174 // Prior to gfx9, MUBUF instructions with the vaddr offset enabled would
1175 // always perform a range check. If a negative vaddr base index was used,
1176 // this would fail the range check. The overall address computation would
1177 // compute a valid address, but this doesn't happen due to the range
1178 // check. For out-of-bounds MUBUF loads, a 0 is returned.
1179 //
1180 // Therefore it should be safe to fold any VGPR offset on gfx9 into the
1181 // MUBUF vaddr, but not on older subtargets which can only do this if the
1182 // sign bit is known 0.
Matt Arsenaultcd099612016-02-24 04:55:29 +00001183 ConstantSDNode *C1 = cast<ConstantSDNode>(N1);
Matt Arsenault45b98182017-11-15 00:45:43 +00001184 if (SIInstrInfo::isLegalMUBUFImmOffset(C1->getZExtValue()) &&
Matt Arsenaultcaf0ed42017-11-30 00:52:40 +00001185 (!Subtarget->privateMemoryResourceIsRangeChecked() ||
1186 CurDAG->SignBitIsZero(N0))) {
Matt Arsenault156d3ae2017-05-17 21:02:58 +00001187 std::tie(VAddr, SOffset) = foldFrameIndex(N0);
Matt Arsenaultcd099612016-02-24 04:55:29 +00001188 ImmOffset = CurDAG->getTargetConstant(C1->getZExtValue(), DL, MVT::i16);
1189 return true;
Tom Stellardb02094e2014-07-21 15:45:01 +00001190 }
1191 }
1192
Tom Stellardb02094e2014-07-21 15:45:01 +00001193 // (node)
Matt Arsenault156d3ae2017-05-17 21:02:58 +00001194 std::tie(VAddr, SOffset) = foldFrameIndex(Addr);
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +00001195 ImmOffset = CurDAG->getTargetConstant(0, DL, MVT::i16);
Tom Stellardb02094e2014-07-21 15:45:01 +00001196 return true;
1197}
1198
Matt Arsenaultb81495d2017-09-20 05:01:53 +00001199bool AMDGPUDAGToDAGISel::SelectMUBUFScratchOffset(SDNode *Parent,
Matt Arsenault156d3ae2017-05-17 21:02:58 +00001200 SDValue Addr,
Matt Arsenault0774ea22017-04-24 19:40:59 +00001201 SDValue &SRsrc,
1202 SDValue &SOffset,
1203 SDValue &Offset) const {
1204 ConstantSDNode *CAddr = dyn_cast<ConstantSDNode>(Addr);
Marek Olsakffadcb72017-11-09 01:52:17 +00001205 if (!CAddr || !SIInstrInfo::isLegalMUBUFImmOffset(CAddr->getZExtValue()))
Matt Arsenault0774ea22017-04-24 19:40:59 +00001206 return false;
1207
1208 SDLoc DL(Addr);
1209 MachineFunction &MF = CurDAG->getMachineFunction();
1210 const SIMachineFunctionInfo *Info = MF.getInfo<SIMachineFunctionInfo>();
1211
1212 SRsrc = CurDAG->getRegister(Info->getScratchRSrcReg(), MVT::v4i32);
Matt Arsenault156d3ae2017-05-17 21:02:58 +00001213
Matt Arsenaultb81495d2017-09-20 05:01:53 +00001214 const MachinePointerInfo &PtrInfo = cast<MemSDNode>(Parent)->getPointerInfo();
Matt Arsenault156d3ae2017-05-17 21:02:58 +00001215 unsigned SOffsetReg = isStackPtrRelative(PtrInfo) ?
1216 Info->getStackPtrOffsetReg() : Info->getScratchWaveOffsetReg();
1217
1218 // FIXME: Get from MachinePointerInfo? We should only be using the frame
1219 // offset if we know this is in a call sequence.
1220 SOffset = CurDAG->getRegister(SOffsetReg, MVT::i32);
1221
Matt Arsenault0774ea22017-04-24 19:40:59 +00001222 Offset = CurDAG->getTargetConstant(CAddr->getZExtValue(), DL, MVT::i16);
1223 return true;
1224}
1225
Tom Stellard155bbb72014-08-11 22:18:17 +00001226bool AMDGPUDAGToDAGISel::SelectMUBUFOffset(SDValue Addr, SDValue &SRsrc,
1227 SDValue &SOffset, SDValue &Offset,
1228 SDValue &GLC, SDValue &SLC,
1229 SDValue &TFE) const {
1230 SDValue Ptr, VAddr, Offen, Idxen, Addr64;
Tom Stellard794c8c02014-12-02 17:05:41 +00001231 const SIInstrInfo *TII =
Eric Christopher7792e322015-01-30 23:24:40 +00001232 static_cast<const SIInstrInfo *>(Subtarget->getInstrInfo());
Tom Stellardb02094e2014-07-21 15:45:01 +00001233
Changpeng Fangb41574a2015-12-22 20:55:23 +00001234 if (!SelectMUBUF(Addr, Ptr, VAddr, SOffset, Offset, Offen, Idxen, Addr64,
1235 GLC, SLC, TFE))
1236 return false;
Tom Stellardb02094e2014-07-21 15:45:01 +00001237
Tom Stellard155bbb72014-08-11 22:18:17 +00001238 if (!cast<ConstantSDNode>(Offen)->getSExtValue() &&
1239 !cast<ConstantSDNode>(Idxen)->getSExtValue() &&
1240 !cast<ConstantSDNode>(Addr64)->getSExtValue()) {
Tom Stellard794c8c02014-12-02 17:05:41 +00001241 uint64_t Rsrc = TII->getDefaultRsrcDataFormat() |
Tom Stellard155bbb72014-08-11 22:18:17 +00001242 APInt::getAllOnesValue(32).getZExtValue(); // Size
1243 SDLoc DL(Addr);
Matt Arsenaultf3cd4512014-11-05 19:01:19 +00001244
1245 const SITargetLowering& Lowering =
1246 *static_cast<const SITargetLowering*>(getTargetLowering());
1247
1248 SRsrc = SDValue(Lowering.buildRSRC(*CurDAG, DL, Ptr, 0, Rsrc), 0);
Tom Stellard155bbb72014-08-11 22:18:17 +00001249 return true;
1250 }
1251 return false;
Tom Stellardb02094e2014-07-21 15:45:01 +00001252}
1253
Tom Stellard7980fc82014-09-25 18:30:26 +00001254bool AMDGPUDAGToDAGISel::SelectMUBUFOffset(SDValue Addr, SDValue &SRsrc,
Jan Vesely43b7b5b2016-04-07 19:23:11 +00001255 SDValue &Soffset, SDValue &Offset
1256 ) const {
1257 SDValue GLC, SLC, TFE;
1258
1259 return SelectMUBUFOffset(Addr, SRsrc, Soffset, Offset, GLC, SLC, TFE);
1260}
1261bool AMDGPUDAGToDAGISel::SelectMUBUFOffset(SDValue Addr, SDValue &SRsrc,
Tom Stellard7980fc82014-09-25 18:30:26 +00001262 SDValue &Soffset, SDValue &Offset,
Matt Arsenault88701812016-06-09 23:42:48 +00001263 SDValue &SLC) const {
1264 SDValue GLC, TFE;
Tom Stellard7980fc82014-09-25 18:30:26 +00001265
1266 return SelectMUBUFOffset(Addr, SRsrc, Soffset, Offset, GLC, SLC, TFE);
1267}
1268
Nicolai Haehnlea6092592016-06-15 07:13:05 +00001269bool AMDGPUDAGToDAGISel::SelectMUBUFConstant(SDValue Constant,
Nicolai Haehnle3003ba02016-03-18 16:24:20 +00001270 SDValue &SOffset,
1271 SDValue &ImmOffset) const {
1272 SDLoc DL(Constant);
Nicolai Haehnle312b64f2017-10-10 12:22:23 +00001273 const uint32_t Align = 4;
1274 const uint32_t MaxImm = alignDown(4095, Align);
Nicolai Haehnle3003ba02016-03-18 16:24:20 +00001275 uint32_t Imm = cast<ConstantSDNode>(Constant)->getZExtValue();
1276 uint32_t Overflow = 0;
1277
Nicolai Haehnle312b64f2017-10-10 12:22:23 +00001278 if (Imm > MaxImm) {
1279 if (Imm <= MaxImm + 64) {
1280 // Use an SOffset inline constant for 4..64
1281 Overflow = Imm - MaxImm;
1282 Imm = MaxImm;
Nicolai Haehnle3003ba02016-03-18 16:24:20 +00001283 } else {
1284 // Try to keep the same value in SOffset for adjacent loads, so that
1285 // the corresponding register contents can be re-used.
1286 //
Nicolai Haehnle312b64f2017-10-10 12:22:23 +00001287 // Load values with all low-bits (except for alignment bits) set into
1288 // SOffset, so that a larger range of values can be covered using
1289 // s_movk_i32.
1290 //
1291 // Atomic operations fail to work correctly when individual address
1292 // components are unaligned, even if their sum is aligned.
1293 uint32_t High = (Imm + Align) & ~4095;
1294 uint32_t Low = (Imm + Align) & 4095;
Nicolai Haehnle3003ba02016-03-18 16:24:20 +00001295 Imm = Low;
Nicolai Haehnle312b64f2017-10-10 12:22:23 +00001296 Overflow = High - Align;
Nicolai Haehnle3003ba02016-03-18 16:24:20 +00001297 }
1298 }
1299
Nicolai Haehnlea6092592016-06-15 07:13:05 +00001300 // There is a hardware bug in SI and CI which prevents address clamping in
1301 // MUBUF instructions from working correctly with SOffsets. The immediate
1302 // offset is unaffected.
1303 if (Overflow > 0 &&
1304 Subtarget->getGeneration() <= AMDGPUSubtarget::SEA_ISLANDS)
1305 return false;
1306
Nicolai Haehnle3003ba02016-03-18 16:24:20 +00001307 ImmOffset = CurDAG->getTargetConstant(Imm, DL, MVT::i16);
1308
1309 if (Overflow <= 64)
1310 SOffset = CurDAG->getTargetConstant(Overflow, DL, MVT::i32);
1311 else
1312 SOffset = SDValue(CurDAG->getMachineNode(AMDGPU::S_MOV_B32, DL, MVT::i32,
1313 CurDAG->getTargetConstant(Overflow, DL, MVT::i32)),
1314 0);
Nicolai Haehnlea6092592016-06-15 07:13:05 +00001315
1316 return true;
Nicolai Haehnle3003ba02016-03-18 16:24:20 +00001317}
1318
1319bool AMDGPUDAGToDAGISel::SelectMUBUFIntrinsicOffset(SDValue Offset,
1320 SDValue &SOffset,
1321 SDValue &ImmOffset) const {
1322 SDLoc DL(Offset);
1323
1324 if (!isa<ConstantSDNode>(Offset))
1325 return false;
1326
Nicolai Haehnlea6092592016-06-15 07:13:05 +00001327 return SelectMUBUFConstant(Offset, SOffset, ImmOffset);
Nicolai Haehnle3003ba02016-03-18 16:24:20 +00001328}
1329
1330bool AMDGPUDAGToDAGISel::SelectMUBUFIntrinsicVOffset(SDValue Offset,
1331 SDValue &SOffset,
1332 SDValue &ImmOffset,
1333 SDValue &VOffset) const {
1334 SDLoc DL(Offset);
1335
1336 // Don't generate an unnecessary voffset for constant offsets.
Nicolai Haehnlea6092592016-06-15 07:13:05 +00001337 if (isa<ConstantSDNode>(Offset)) {
1338 SDValue Tmp1, Tmp2;
1339
1340 // When necessary, use a voffset in <= CI anyway to work around a hardware
1341 // bug.
1342 if (Subtarget->getGeneration() > AMDGPUSubtarget::SEA_ISLANDS ||
1343 SelectMUBUFConstant(Offset, Tmp1, Tmp2))
1344 return false;
1345 }
Nicolai Haehnle3003ba02016-03-18 16:24:20 +00001346
1347 if (CurDAG->isBaseWithConstantOffset(Offset)) {
1348 SDValue N0 = Offset.getOperand(0);
1349 SDValue N1 = Offset.getOperand(1);
Nicolai Haehnlea6092592016-06-15 07:13:05 +00001350 if (cast<ConstantSDNode>(N1)->getSExtValue() >= 0 &&
1351 SelectMUBUFConstant(N1, SOffset, ImmOffset)) {
1352 VOffset = N0;
1353 return true;
1354 }
Nicolai Haehnle3003ba02016-03-18 16:24:20 +00001355 }
1356
Nicolai Haehnlea6092592016-06-15 07:13:05 +00001357 SOffset = CurDAG->getTargetConstant(0, DL, MVT::i32);
1358 ImmOffset = CurDAG->getTargetConstant(0, DL, MVT::i16);
1359 VOffset = Offset;
1360
Nicolai Haehnle3003ba02016-03-18 16:24:20 +00001361 return true;
1362}
1363
Matt Arsenault4e309b02017-07-29 01:03:53 +00001364template <bool IsSigned>
Matt Arsenaultdb7c6a82017-06-12 16:53:51 +00001365bool AMDGPUDAGToDAGISel::SelectFlatOffset(SDValue Addr,
1366 SDValue &VAddr,
1367 SDValue &Offset,
1368 SDValue &SLC) const {
1369 int64_t OffsetVal = 0;
1370
1371 if (Subtarget->hasFlatInstOffsets() &&
1372 CurDAG->isBaseWithConstantOffset(Addr)) {
1373 SDValue N0 = Addr.getOperand(0);
1374 SDValue N1 = Addr.getOperand(1);
Matt Arsenault4e309b02017-07-29 01:03:53 +00001375 int64_t COffsetVal = cast<ConstantSDNode>(N1)->getSExtValue();
1376
1377 if ((IsSigned && isInt<13>(COffsetVal)) ||
1378 (!IsSigned && isUInt<12>(COffsetVal))) {
Matt Arsenaultdb7c6a82017-06-12 16:53:51 +00001379 Addr = N0;
1380 OffsetVal = COffsetVal;
1381 }
1382 }
1383
Matt Arsenault7757c592016-06-09 23:42:54 +00001384 VAddr = Addr;
Matt Arsenaultdb7c6a82017-06-12 16:53:51 +00001385 Offset = CurDAG->getTargetConstant(OffsetVal, SDLoc(), MVT::i16);
Matt Arsenault47ccafe2017-05-11 17:38:33 +00001386 SLC = CurDAG->getTargetConstant(0, SDLoc(), MVT::i1);
Matt Arsenaultdb7c6a82017-06-12 16:53:51 +00001387
Matt Arsenault7757c592016-06-09 23:42:54 +00001388 return true;
1389}
1390
Matt Arsenaultdb7c6a82017-06-12 16:53:51 +00001391bool AMDGPUDAGToDAGISel::SelectFlatAtomic(SDValue Addr,
1392 SDValue &VAddr,
1393 SDValue &Offset,
1394 SDValue &SLC) const {
Matt Arsenault4e309b02017-07-29 01:03:53 +00001395 return SelectFlatOffset<false>(Addr, VAddr, Offset, SLC);
1396}
1397
1398bool AMDGPUDAGToDAGISel::SelectFlatAtomicSigned(SDValue Addr,
1399 SDValue &VAddr,
1400 SDValue &Offset,
1401 SDValue &SLC) const {
1402 return SelectFlatOffset<true>(Addr, VAddr, Offset, SLC);
Matt Arsenaultdb7c6a82017-06-12 16:53:51 +00001403}
1404
Tom Stellarddee26a22015-08-06 19:28:30 +00001405bool AMDGPUDAGToDAGISel::SelectSMRDOffset(SDValue ByteOffsetNode,
1406 SDValue &Offset, bool &Imm) const {
1407
1408 // FIXME: Handle non-constant offsets.
1409 ConstantSDNode *C = dyn_cast<ConstantSDNode>(ByteOffsetNode);
1410 if (!C)
1411 return false;
1412
1413 SDLoc SL(ByteOffsetNode);
Tom Stellard5bfbae52018-07-11 20:59:01 +00001414 GCNSubtarget::Generation Gen = Subtarget->getGeneration();
Tom Stellarddee26a22015-08-06 19:28:30 +00001415 int64_t ByteOffset = C->getSExtValue();
Tom Stellard08efb7e2017-01-27 18:41:14 +00001416 int64_t EncodedOffset = AMDGPU::getSMRDEncodedOffset(*Subtarget, ByteOffset);
Tom Stellarddee26a22015-08-06 19:28:30 +00001417
Tom Stellard08efb7e2017-01-27 18:41:14 +00001418 if (AMDGPU::isLegalSMRDImmOffset(*Subtarget, ByteOffset)) {
Tom Stellarddee26a22015-08-06 19:28:30 +00001419 Offset = CurDAG->getTargetConstant(EncodedOffset, SL, MVT::i32);
1420 Imm = true;
1421 return true;
1422 }
1423
Tom Stellard217361c2015-08-06 19:28:38 +00001424 if (!isUInt<32>(EncodedOffset) || !isUInt<32>(ByteOffset))
1425 return false;
1426
Marek Olsak8973a0a2017-05-24 14:53:50 +00001427 if (Gen == AMDGPUSubtarget::SEA_ISLANDS && isUInt<32>(EncodedOffset)) {
1428 // 32-bit Immediates are supported on Sea Islands.
Tom Stellard217361c2015-08-06 19:28:38 +00001429 Offset = CurDAG->getTargetConstant(EncodedOffset, SL, MVT::i32);
1430 } else {
Tom Stellarddee26a22015-08-06 19:28:30 +00001431 SDValue C32Bit = CurDAG->getTargetConstant(ByteOffset, SL, MVT::i32);
1432 Offset = SDValue(CurDAG->getMachineNode(AMDGPU::S_MOV_B32, SL, MVT::i32,
1433 C32Bit), 0);
Tom Stellarddee26a22015-08-06 19:28:30 +00001434 }
Tom Stellard217361c2015-08-06 19:28:38 +00001435 Imm = false;
1436 return true;
Tom Stellarddee26a22015-08-06 19:28:30 +00001437}
1438
Matt Arsenault923712b2018-02-09 16:57:57 +00001439SDValue AMDGPUDAGToDAGISel::Expand32BitAddress(SDValue Addr) const {
1440 if (Addr.getValueType() != MVT::i32)
1441 return Addr;
1442
1443 // Zero-extend a 32-bit address.
1444 SDLoc SL(Addr);
1445
1446 const MachineFunction &MF = CurDAG->getMachineFunction();
1447 const SIMachineFunctionInfo *Info = MF.getInfo<SIMachineFunctionInfo>();
1448 unsigned AddrHiVal = Info->get32BitAddressHighBits();
1449 SDValue AddrHi = CurDAG->getTargetConstant(AddrHiVal, SL, MVT::i32);
1450
1451 const SDValue Ops[] = {
1452 CurDAG->getTargetConstant(AMDGPU::SReg_64_XEXECRegClassID, SL, MVT::i32),
1453 Addr,
1454 CurDAG->getTargetConstant(AMDGPU::sub0, SL, MVT::i32),
1455 SDValue(CurDAG->getMachineNode(AMDGPU::S_MOV_B32, SL, MVT::i32, AddrHi),
1456 0),
1457 CurDAG->getTargetConstant(AMDGPU::sub1, SL, MVT::i32),
1458 };
1459
1460 return SDValue(CurDAG->getMachineNode(AMDGPU::REG_SEQUENCE, SL, MVT::i64,
1461 Ops), 0);
1462}
1463
Tom Stellarddee26a22015-08-06 19:28:30 +00001464bool AMDGPUDAGToDAGISel::SelectSMRD(SDValue Addr, SDValue &SBase,
1465 SDValue &Offset, bool &Imm) const {
Tom Stellarddee26a22015-08-06 19:28:30 +00001466 SDLoc SL(Addr);
Matt Arsenault923712b2018-02-09 16:57:57 +00001467
Tom Stellarddee26a22015-08-06 19:28:30 +00001468 if (CurDAG->isBaseWithConstantOffset(Addr)) {
1469 SDValue N0 = Addr.getOperand(0);
1470 SDValue N1 = Addr.getOperand(1);
1471
1472 if (SelectSMRDOffset(N1, Offset, Imm)) {
Matt Arsenault923712b2018-02-09 16:57:57 +00001473 SBase = Expand32BitAddress(N0);
Tom Stellarddee26a22015-08-06 19:28:30 +00001474 return true;
1475 }
1476 }
Matt Arsenault923712b2018-02-09 16:57:57 +00001477 SBase = Expand32BitAddress(Addr);
Tom Stellarddee26a22015-08-06 19:28:30 +00001478 Offset = CurDAG->getTargetConstant(0, SL, MVT::i32);
1479 Imm = true;
1480 return true;
1481}
1482
1483bool AMDGPUDAGToDAGISel::SelectSMRDImm(SDValue Addr, SDValue &SBase,
1484 SDValue &Offset) const {
1485 bool Imm;
Marek Olsak8973a0a2017-05-24 14:53:50 +00001486 return SelectSMRD(Addr, SBase, Offset, Imm) && Imm;
1487}
Tom Stellarddee26a22015-08-06 19:28:30 +00001488
Marek Olsak8973a0a2017-05-24 14:53:50 +00001489bool AMDGPUDAGToDAGISel::SelectSMRDImm32(SDValue Addr, SDValue &SBase,
1490 SDValue &Offset) const {
1491
1492 if (Subtarget->getGeneration() != AMDGPUSubtarget::SEA_ISLANDS)
1493 return false;
1494
1495 bool Imm;
Tom Stellard217361c2015-08-06 19:28:38 +00001496 if (!SelectSMRD(Addr, SBase, Offset, Imm))
1497 return false;
1498
Marek Olsak8973a0a2017-05-24 14:53:50 +00001499 return !Imm && isa<ConstantSDNode>(Offset);
Tom Stellard217361c2015-08-06 19:28:38 +00001500}
1501
Tom Stellarddee26a22015-08-06 19:28:30 +00001502bool AMDGPUDAGToDAGISel::SelectSMRDSgpr(SDValue Addr, SDValue &SBase,
1503 SDValue &Offset) const {
1504 bool Imm;
Tom Stellard217361c2015-08-06 19:28:38 +00001505 return SelectSMRD(Addr, SBase, Offset, Imm) && !Imm &&
1506 !isa<ConstantSDNode>(Offset);
Tom Stellarddee26a22015-08-06 19:28:30 +00001507}
1508
1509bool AMDGPUDAGToDAGISel::SelectSMRDBufferImm(SDValue Addr,
1510 SDValue &Offset) const {
1511 bool Imm;
Marek Olsak8973a0a2017-05-24 14:53:50 +00001512 return SelectSMRDOffset(Addr, Offset, Imm) && Imm;
1513}
Tom Stellarddee26a22015-08-06 19:28:30 +00001514
Marek Olsak8973a0a2017-05-24 14:53:50 +00001515bool AMDGPUDAGToDAGISel::SelectSMRDBufferImm32(SDValue Addr,
1516 SDValue &Offset) const {
1517 if (Subtarget->getGeneration() != AMDGPUSubtarget::SEA_ISLANDS)
1518 return false;
1519
1520 bool Imm;
Tom Stellard217361c2015-08-06 19:28:38 +00001521 if (!SelectSMRDOffset(Addr, Offset, Imm))
1522 return false;
1523
Marek Olsak8973a0a2017-05-24 14:53:50 +00001524 return !Imm && isa<ConstantSDNode>(Offset);
Tom Stellard217361c2015-08-06 19:28:38 +00001525}
1526
Nicolai Haehnle7968c342016-07-12 08:12:16 +00001527bool AMDGPUDAGToDAGISel::SelectMOVRELOffset(SDValue Index,
1528 SDValue &Base,
1529 SDValue &Offset) const {
Matt Arsenault1322b6f2016-07-09 01:13:56 +00001530 SDLoc DL(Index);
1531
1532 if (CurDAG->isBaseWithConstantOffset(Index)) {
1533 SDValue N0 = Index.getOperand(0);
1534 SDValue N1 = Index.getOperand(1);
1535 ConstantSDNode *C1 = cast<ConstantSDNode>(N1);
1536
1537 // (add n0, c0)
1538 Base = N0;
1539 Offset = CurDAG->getTargetConstant(C1->getZExtValue(), DL, MVT::i32);
1540 return true;
1541 }
1542
Nicolai Haehnle7968c342016-07-12 08:12:16 +00001543 if (isa<ConstantSDNode>(Index))
1544 return false;
Matt Arsenault1322b6f2016-07-09 01:13:56 +00001545
1546 Base = Index;
1547 Offset = CurDAG->getTargetConstant(0, DL, MVT::i32);
1548 return true;
1549}
1550
Benjamin Kramerbdc49562016-06-12 15:39:02 +00001551SDNode *AMDGPUDAGToDAGISel::getS_BFE(unsigned Opcode, const SDLoc &DL,
1552 SDValue Val, uint32_t Offset,
1553 uint32_t Width) {
Marek Olsak9b728682015-03-24 13:40:27 +00001554 // Transformation function, pack the offset and width of a BFE into
1555 // the format expected by the S_BFE_I32 / S_BFE_U32. In the second
1556 // source, bits [5:0] contain the offset and bits [22:16] the width.
1557 uint32_t PackedVal = Offset | (Width << 16);
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +00001558 SDValue PackedConst = CurDAG->getTargetConstant(PackedVal, DL, MVT::i32);
Marek Olsak9b728682015-03-24 13:40:27 +00001559
1560 return CurDAG->getMachineNode(Opcode, DL, MVT::i32, Val, PackedConst);
1561}
1562
Justin Bogner95927c02016-05-12 21:03:32 +00001563void AMDGPUDAGToDAGISel::SelectS_BFEFromShifts(SDNode *N) {
Marek Olsak9b728682015-03-24 13:40:27 +00001564 // "(a << b) srl c)" ---> "BFE_U32 a, (c-b), (32-c)
1565 // "(a << b) sra c)" ---> "BFE_I32 a, (c-b), (32-c)
1566 // Predicate: 0 < b <= c < 32
1567
1568 const SDValue &Shl = N->getOperand(0);
1569 ConstantSDNode *B = dyn_cast<ConstantSDNode>(Shl->getOperand(1));
1570 ConstantSDNode *C = dyn_cast<ConstantSDNode>(N->getOperand(1));
1571
1572 if (B && C) {
1573 uint32_t BVal = B->getZExtValue();
1574 uint32_t CVal = C->getZExtValue();
1575
1576 if (0 < BVal && BVal <= CVal && CVal < 32) {
1577 bool Signed = N->getOpcode() == ISD::SRA;
1578 unsigned Opcode = Signed ? AMDGPU::S_BFE_I32 : AMDGPU::S_BFE_U32;
1579
Justin Bogner95927c02016-05-12 21:03:32 +00001580 ReplaceNode(N, getS_BFE(Opcode, SDLoc(N), Shl.getOperand(0), CVal - BVal,
1581 32 - CVal));
1582 return;
Marek Olsak9b728682015-03-24 13:40:27 +00001583 }
1584 }
Justin Bogner95927c02016-05-12 21:03:32 +00001585 SelectCode(N);
Marek Olsak9b728682015-03-24 13:40:27 +00001586}
1587
Justin Bogner95927c02016-05-12 21:03:32 +00001588void AMDGPUDAGToDAGISel::SelectS_BFE(SDNode *N) {
Marek Olsak9b728682015-03-24 13:40:27 +00001589 switch (N->getOpcode()) {
1590 case ISD::AND:
1591 if (N->getOperand(0).getOpcode() == ISD::SRL) {
1592 // "(a srl b) & mask" ---> "BFE_U32 a, b, popcount(mask)"
1593 // Predicate: isMask(mask)
1594 const SDValue &Srl = N->getOperand(0);
1595 ConstantSDNode *Shift = dyn_cast<ConstantSDNode>(Srl.getOperand(1));
1596 ConstantSDNode *Mask = dyn_cast<ConstantSDNode>(N->getOperand(1));
1597
1598 if (Shift && Mask) {
1599 uint32_t ShiftVal = Shift->getZExtValue();
1600 uint32_t MaskVal = Mask->getZExtValue();
1601
1602 if (isMask_32(MaskVal)) {
1603 uint32_t WidthVal = countPopulation(MaskVal);
1604
Justin Bogner95927c02016-05-12 21:03:32 +00001605 ReplaceNode(N, getS_BFE(AMDGPU::S_BFE_U32, SDLoc(N),
1606 Srl.getOperand(0), ShiftVal, WidthVal));
1607 return;
Marek Olsak9b728682015-03-24 13:40:27 +00001608 }
1609 }
1610 }
1611 break;
1612 case ISD::SRL:
1613 if (N->getOperand(0).getOpcode() == ISD::AND) {
1614 // "(a & mask) srl b)" ---> "BFE_U32 a, b, popcount(mask >> b)"
1615 // Predicate: isMask(mask >> b)
1616 const SDValue &And = N->getOperand(0);
1617 ConstantSDNode *Shift = dyn_cast<ConstantSDNode>(N->getOperand(1));
1618 ConstantSDNode *Mask = dyn_cast<ConstantSDNode>(And->getOperand(1));
1619
1620 if (Shift && Mask) {
1621 uint32_t ShiftVal = Shift->getZExtValue();
1622 uint32_t MaskVal = Mask->getZExtValue() >> ShiftVal;
1623
1624 if (isMask_32(MaskVal)) {
1625 uint32_t WidthVal = countPopulation(MaskVal);
1626
Justin Bogner95927c02016-05-12 21:03:32 +00001627 ReplaceNode(N, getS_BFE(AMDGPU::S_BFE_U32, SDLoc(N),
1628 And.getOperand(0), ShiftVal, WidthVal));
1629 return;
Marek Olsak9b728682015-03-24 13:40:27 +00001630 }
1631 }
Justin Bogner95927c02016-05-12 21:03:32 +00001632 } else if (N->getOperand(0).getOpcode() == ISD::SHL) {
1633 SelectS_BFEFromShifts(N);
1634 return;
1635 }
Marek Olsak9b728682015-03-24 13:40:27 +00001636 break;
1637 case ISD::SRA:
Justin Bogner95927c02016-05-12 21:03:32 +00001638 if (N->getOperand(0).getOpcode() == ISD::SHL) {
1639 SelectS_BFEFromShifts(N);
1640 return;
1641 }
Marek Olsak9b728682015-03-24 13:40:27 +00001642 break;
Matt Arsenault7e8de012016-04-22 22:59:16 +00001643
1644 case ISD::SIGN_EXTEND_INREG: {
1645 // sext_inreg (srl x, 16), i8 -> bfe_i32 x, 16, 8
1646 SDValue Src = N->getOperand(0);
1647 if (Src.getOpcode() != ISD::SRL)
1648 break;
1649
1650 const ConstantSDNode *Amt = dyn_cast<ConstantSDNode>(Src.getOperand(1));
1651 if (!Amt)
1652 break;
1653
1654 unsigned Width = cast<VTSDNode>(N->getOperand(1))->getVT().getSizeInBits();
Justin Bogner95927c02016-05-12 21:03:32 +00001655 ReplaceNode(N, getS_BFE(AMDGPU::S_BFE_I32, SDLoc(N), Src.getOperand(0),
1656 Amt->getZExtValue(), Width));
1657 return;
Matt Arsenault7e8de012016-04-22 22:59:16 +00001658 }
Marek Olsak9b728682015-03-24 13:40:27 +00001659 }
1660
Justin Bogner95927c02016-05-12 21:03:32 +00001661 SelectCode(N);
Marek Olsak9b728682015-03-24 13:40:27 +00001662}
1663
Matt Arsenault7b1dc2c2016-09-17 02:02:19 +00001664bool AMDGPUDAGToDAGISel::isCBranchSCC(const SDNode *N) const {
1665 assert(N->getOpcode() == ISD::BRCOND);
1666 if (!N->hasOneUse())
1667 return false;
1668
1669 SDValue Cond = N->getOperand(1);
1670 if (Cond.getOpcode() == ISD::CopyToReg)
1671 Cond = Cond.getOperand(2);
1672
1673 if (Cond.getOpcode() != ISD::SETCC || !Cond.hasOneUse())
1674 return false;
1675
1676 MVT VT = Cond.getOperand(0).getSimpleValueType();
1677 if (VT == MVT::i32)
1678 return true;
1679
1680 if (VT == MVT::i64) {
Tom Stellard5bfbae52018-07-11 20:59:01 +00001681 auto ST = static_cast<const GCNSubtarget *>(Subtarget);
Matt Arsenault7b1dc2c2016-09-17 02:02:19 +00001682
1683 ISD::CondCode CC = cast<CondCodeSDNode>(Cond.getOperand(2))->get();
1684 return (CC == ISD::SETEQ || CC == ISD::SETNE) && ST->hasScalarCompareEq64();
1685 }
1686
1687 return false;
1688}
1689
Justin Bogner95927c02016-05-12 21:03:32 +00001690void AMDGPUDAGToDAGISel::SelectBRCOND(SDNode *N) {
Tom Stellardbc4497b2016-02-12 23:45:29 +00001691 SDValue Cond = N->getOperand(1);
1692
Matt Arsenault327188a2016-12-15 21:57:11 +00001693 if (Cond.isUndef()) {
1694 CurDAG->SelectNodeTo(N, AMDGPU::SI_BR_UNDEF, MVT::Other,
1695 N->getOperand(2), N->getOperand(0));
1696 return;
1697 }
1698
Matt Arsenaultd674e0a2017-10-10 20:34:49 +00001699 bool UseSCCBr = isCBranchSCC(N) && isUniformBr(N);
1700 unsigned BrOp = UseSCCBr ? AMDGPU::S_CBRANCH_SCC1 : AMDGPU::S_CBRANCH_VCCNZ;
1701 unsigned CondReg = UseSCCBr ? AMDGPU::SCC : AMDGPU::VCC;
Tom Stellardbc4497b2016-02-12 23:45:29 +00001702 SDLoc SL(N);
1703
Tim Renouf6eaad1e2018-01-09 21:34:43 +00001704 if (!UseSCCBr) {
1705 // This is the case that we are selecting to S_CBRANCH_VCCNZ. We have not
1706 // analyzed what generates the vcc value, so we do not know whether vcc
1707 // bits for disabled lanes are 0. Thus we need to mask out bits for
1708 // disabled lanes.
1709 //
1710 // For the case that we select S_CBRANCH_SCC1 and it gets
1711 // changed to S_CBRANCH_VCCNZ in SIFixSGPRCopies, SIFixSGPRCopies calls
1712 // SIInstrInfo::moveToVALU which inserts the S_AND).
1713 //
1714 // We could add an analysis of what generates the vcc value here and omit
1715 // the S_AND when is unnecessary. But it would be better to add a separate
1716 // pass after SIFixSGPRCopies to do the unnecessary S_AND removal, so it
1717 // catches both cases.
1718 Cond = SDValue(CurDAG->getMachineNode(AMDGPU::S_AND_B64, SL, MVT::i1,
1719 CurDAG->getRegister(AMDGPU::EXEC, MVT::i1),
1720 Cond),
1721 0);
1722 }
1723
Matt Arsenaultd674e0a2017-10-10 20:34:49 +00001724 SDValue VCC = CurDAG->getCopyToReg(N->getOperand(0), SL, CondReg, Cond);
1725 CurDAG->SelectNodeTo(N, BrOp, MVT::Other,
Justin Bogner95927c02016-05-12 21:03:32 +00001726 N->getOperand(2), // Basic Block
Matt Arsenaultf530e8b2016-11-07 19:09:33 +00001727 VCC.getValue(0));
Tom Stellardbc4497b2016-02-12 23:45:29 +00001728}
1729
Matt Arsenault0084adc2018-04-30 19:08:16 +00001730void AMDGPUDAGToDAGISel::SelectFMAD_FMA(SDNode *N) {
Matt Arsenaultd7e23032017-09-07 18:05:07 +00001731 MVT VT = N->getSimpleValueType(0);
Matt Arsenault0084adc2018-04-30 19:08:16 +00001732 bool IsFMA = N->getOpcode() == ISD::FMA;
1733 if (VT != MVT::f32 || (!Subtarget->hasMadMixInsts() &&
1734 !Subtarget->hasFmaMixInsts()) ||
1735 ((IsFMA && Subtarget->hasMadMixInsts()) ||
1736 (!IsFMA && Subtarget->hasFmaMixInsts()))) {
Matt Arsenaultd7e23032017-09-07 18:05:07 +00001737 SelectCode(N);
1738 return;
1739 }
1740
1741 SDValue Src0 = N->getOperand(0);
1742 SDValue Src1 = N->getOperand(1);
1743 SDValue Src2 = N->getOperand(2);
1744 unsigned Src0Mods, Src1Mods, Src2Mods;
1745
Matt Arsenault0084adc2018-04-30 19:08:16 +00001746 // Avoid using v_mad_mix_f32/v_fma_mix_f32 unless there is actually an operand
1747 // using the conversion from f16.
Matt Arsenaultd7e23032017-09-07 18:05:07 +00001748 bool Sel0 = SelectVOP3PMadMixModsImpl(Src0, Src0, Src0Mods);
1749 bool Sel1 = SelectVOP3PMadMixModsImpl(Src1, Src1, Src1Mods);
1750 bool Sel2 = SelectVOP3PMadMixModsImpl(Src2, Src2, Src2Mods);
1751
Matt Arsenault0084adc2018-04-30 19:08:16 +00001752 assert((IsFMA || !Subtarget->hasFP32Denormals()) &&
Matt Arsenaultd7e23032017-09-07 18:05:07 +00001753 "fmad selected with denormals enabled");
1754 // TODO: We can select this with f32 denormals enabled if all the sources are
1755 // converted from f16 (in which case fmad isn't legal).
1756
1757 if (Sel0 || Sel1 || Sel2) {
1758 // For dummy operands.
1759 SDValue Zero = CurDAG->getTargetConstant(0, SDLoc(), MVT::i32);
1760 SDValue Ops[] = {
1761 CurDAG->getTargetConstant(Src0Mods, SDLoc(), MVT::i32), Src0,
1762 CurDAG->getTargetConstant(Src1Mods, SDLoc(), MVT::i32), Src1,
1763 CurDAG->getTargetConstant(Src2Mods, SDLoc(), MVT::i32), Src2,
1764 CurDAG->getTargetConstant(0, SDLoc(), MVT::i1),
1765 Zero, Zero
1766 };
1767
Matt Arsenault0084adc2018-04-30 19:08:16 +00001768 CurDAG->SelectNodeTo(N,
1769 IsFMA ? AMDGPU::V_FMA_MIX_F32 : AMDGPU::V_MAD_MIX_F32,
1770 MVT::f32, Ops);
Matt Arsenaultd7e23032017-09-07 18:05:07 +00001771 } else {
1772 SelectCode(N);
1773 }
1774}
1775
Matt Arsenault88701812016-06-09 23:42:48 +00001776// This is here because there isn't a way to use the generated sub0_sub1 as the
1777// subreg index to EXTRACT_SUBREG in tablegen.
1778void AMDGPUDAGToDAGISel::SelectATOMIC_CMP_SWAP(SDNode *N) {
1779 MemSDNode *Mem = cast<MemSDNode>(N);
1780 unsigned AS = Mem->getAddressSpace();
Yaxun Liu1a14bfa2017-03-27 14:04:01 +00001781 if (AS == AMDGPUASI.FLAT_ADDRESS) {
Matt Arsenault7757c592016-06-09 23:42:54 +00001782 SelectCode(N);
1783 return;
1784 }
Matt Arsenault88701812016-06-09 23:42:48 +00001785
1786 MVT VT = N->getSimpleValueType(0);
1787 bool Is32 = (VT == MVT::i32);
1788 SDLoc SL(N);
1789
1790 MachineSDNode *CmpSwap = nullptr;
1791 if (Subtarget->hasAddr64()) {
Vitaly Buka74503982017-10-15 05:35:02 +00001792 SDValue SRsrc, VAddr, SOffset, Offset, SLC;
Matt Arsenault88701812016-06-09 23:42:48 +00001793
1794 if (SelectMUBUFAddr64(Mem->getBasePtr(), SRsrc, VAddr, SOffset, Offset, SLC)) {
Matt Arsenaulte5456ce2017-07-20 21:06:04 +00001795 unsigned Opcode = Is32 ? AMDGPU::BUFFER_ATOMIC_CMPSWAP_ADDR64_RTN :
1796 AMDGPU::BUFFER_ATOMIC_CMPSWAP_X2_ADDR64_RTN;
Matt Arsenault88701812016-06-09 23:42:48 +00001797 SDValue CmpVal = Mem->getOperand(2);
1798
1799 // XXX - Do we care about glue operands?
1800
1801 SDValue Ops[] = {
1802 CmpVal, VAddr, SRsrc, SOffset, Offset, SLC, Mem->getChain()
1803 };
1804
1805 CmpSwap = CurDAG->getMachineNode(Opcode, SL, Mem->getVTList(), Ops);
1806 }
1807 }
1808
1809 if (!CmpSwap) {
1810 SDValue SRsrc, SOffset, Offset, SLC;
1811 if (SelectMUBUFOffset(Mem->getBasePtr(), SRsrc, SOffset, Offset, SLC)) {
Matt Arsenaulte5456ce2017-07-20 21:06:04 +00001812 unsigned Opcode = Is32 ? AMDGPU::BUFFER_ATOMIC_CMPSWAP_OFFSET_RTN :
1813 AMDGPU::BUFFER_ATOMIC_CMPSWAP_X2_OFFSET_RTN;
Matt Arsenault88701812016-06-09 23:42:48 +00001814
1815 SDValue CmpVal = Mem->getOperand(2);
1816 SDValue Ops[] = {
1817 CmpVal, SRsrc, SOffset, Offset, SLC, Mem->getChain()
1818 };
1819
1820 CmpSwap = CurDAG->getMachineNode(Opcode, SL, Mem->getVTList(), Ops);
1821 }
1822 }
1823
1824 if (!CmpSwap) {
1825 SelectCode(N);
1826 return;
1827 }
1828
1829 MachineSDNode::mmo_iterator MMOs = MF->allocateMemRefsArray(1);
1830 *MMOs = Mem->getMemOperand();
1831 CmpSwap->setMemRefs(MMOs, MMOs + 1);
1832
1833 unsigned SubReg = Is32 ? AMDGPU::sub0 : AMDGPU::sub0_sub1;
1834 SDValue Extract
1835 = CurDAG->getTargetExtractSubreg(SubReg, SL, VT, SDValue(CmpSwap, 0));
1836
1837 ReplaceUses(SDValue(N, 0), Extract);
1838 ReplaceUses(SDValue(N, 1), SDValue(CmpSwap, 1));
1839 CurDAG->RemoveDeadNode(N);
1840}
1841
Matt Arsenaultd7e23032017-09-07 18:05:07 +00001842bool AMDGPUDAGToDAGISel::SelectVOP3ModsImpl(SDValue In, SDValue &Src,
1843 unsigned &Mods) const {
1844 Mods = 0;
Tom Stellardb4a313a2014-08-01 00:32:39 +00001845 Src = In;
1846
1847 if (Src.getOpcode() == ISD::FNEG) {
1848 Mods |= SISrcMods::NEG;
1849 Src = Src.getOperand(0);
1850 }
1851
1852 if (Src.getOpcode() == ISD::FABS) {
1853 Mods |= SISrcMods::ABS;
1854 Src = Src.getOperand(0);
1855 }
1856
Tom Stellardb4a313a2014-08-01 00:32:39 +00001857 return true;
1858}
1859
Matt Arsenaultd7e23032017-09-07 18:05:07 +00001860bool AMDGPUDAGToDAGISel::SelectVOP3Mods(SDValue In, SDValue &Src,
1861 SDValue &SrcMods) const {
1862 unsigned Mods;
1863 if (SelectVOP3ModsImpl(In, Src, Mods)) {
1864 SrcMods = CurDAG->getTargetConstant(Mods, SDLoc(In), MVT::i32);
1865 return true;
1866 }
1867
1868 return false;
1869}
1870
Matt Arsenaultf84e5d92017-01-31 03:07:46 +00001871bool AMDGPUDAGToDAGISel::SelectVOP3Mods_NNaN(SDValue In, SDValue &Src,
1872 SDValue &SrcMods) const {
1873 SelectVOP3Mods(In, Src, SrcMods);
1874 return isNoNanSrc(Src);
1875}
1876
Matt Arsenaultdf58e822017-04-25 21:17:38 +00001877bool AMDGPUDAGToDAGISel::SelectVOP3NoMods(SDValue In, SDValue &Src) const {
1878 if (In.getOpcode() == ISD::FABS || In.getOpcode() == ISD::FNEG)
1879 return false;
1880
1881 Src = In;
1882 return true;
Tom Stellarddb5a11f2015-07-13 15:47:57 +00001883}
1884
Tom Stellardb4a313a2014-08-01 00:32:39 +00001885bool AMDGPUDAGToDAGISel::SelectVOP3Mods0(SDValue In, SDValue &Src,
1886 SDValue &SrcMods, SDValue &Clamp,
1887 SDValue &Omod) const {
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +00001888 SDLoc DL(In);
Matt Arsenaultdf58e822017-04-25 21:17:38 +00001889 Clamp = CurDAG->getTargetConstant(0, DL, MVT::i1);
1890 Omod = CurDAG->getTargetConstant(0, DL, MVT::i1);
Tom Stellardb4a313a2014-08-01 00:32:39 +00001891
1892 return SelectVOP3Mods(In, Src, SrcMods);
1893}
1894
Matt Arsenault4831ce52015-01-06 23:00:37 +00001895bool AMDGPUDAGToDAGISel::SelectVOP3Mods0Clamp0OMod(SDValue In, SDValue &Src,
1896 SDValue &SrcMods,
1897 SDValue &Clamp,
1898 SDValue &Omod) const {
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +00001899 Clamp = Omod = CurDAG->getTargetConstant(0, SDLoc(In), MVT::i32);
Matt Arsenault4831ce52015-01-06 23:00:37 +00001900 return SelectVOP3Mods(In, Src, SrcMods);
1901}
1902
Dmitry Preobrazhenskyc512d442017-03-27 15:57:17 +00001903bool AMDGPUDAGToDAGISel::SelectVOP3OMods(SDValue In, SDValue &Src,
1904 SDValue &Clamp, SDValue &Omod) const {
1905 Src = In;
1906
1907 SDLoc DL(In);
Matt Arsenaultdf58e822017-04-25 21:17:38 +00001908 Clamp = CurDAG->getTargetConstant(0, DL, MVT::i1);
1909 Omod = CurDAG->getTargetConstant(0, DL, MVT::i1);
Dmitry Preobrazhenskyc512d442017-03-27 15:57:17 +00001910
1911 return true;
1912}
1913
Matt Arsenault98f29462017-05-17 20:30:58 +00001914static SDValue stripBitcast(SDValue Val) {
1915 return Val.getOpcode() == ISD::BITCAST ? Val.getOperand(0) : Val;
1916}
1917
1918// Figure out if this is really an extract of the high 16-bits of a dword.
1919static bool isExtractHiElt(SDValue In, SDValue &Out) {
1920 In = stripBitcast(In);
1921 if (In.getOpcode() != ISD::TRUNCATE)
1922 return false;
1923
1924 SDValue Srl = In.getOperand(0);
1925 if (Srl.getOpcode() == ISD::SRL) {
1926 if (ConstantSDNode *ShiftAmt = dyn_cast<ConstantSDNode>(Srl.getOperand(1))) {
1927 if (ShiftAmt->getZExtValue() == 16) {
1928 Out = stripBitcast(Srl.getOperand(0));
1929 return true;
1930 }
1931 }
1932 }
1933
1934 return false;
1935}
1936
1937// Look through operations that obscure just looking at the low 16-bits of the
1938// same register.
1939static SDValue stripExtractLoElt(SDValue In) {
1940 if (In.getOpcode() == ISD::TRUNCATE) {
1941 SDValue Src = In.getOperand(0);
1942 if (Src.getValueType().getSizeInBits() == 32)
1943 return stripBitcast(Src);
1944 }
1945
1946 return In;
1947}
1948
Matt Arsenaulteb522e62017-02-27 22:15:25 +00001949bool AMDGPUDAGToDAGISel::SelectVOP3PMods(SDValue In, SDValue &Src,
1950 SDValue &SrcMods) const {
1951 unsigned Mods = 0;
1952 Src = In;
1953
Matt Arsenaulteb522e62017-02-27 22:15:25 +00001954 if (Src.getOpcode() == ISD::FNEG) {
Matt Arsenault786eeea2017-05-17 20:00:00 +00001955 Mods ^= (SISrcMods::NEG | SISrcMods::NEG_HI);
Matt Arsenaulteb522e62017-02-27 22:15:25 +00001956 Src = Src.getOperand(0);
1957 }
1958
Matt Arsenault786eeea2017-05-17 20:00:00 +00001959 if (Src.getOpcode() == ISD::BUILD_VECTOR) {
1960 unsigned VecMods = Mods;
1961
Matt Arsenault98f29462017-05-17 20:30:58 +00001962 SDValue Lo = stripBitcast(Src.getOperand(0));
1963 SDValue Hi = stripBitcast(Src.getOperand(1));
Matt Arsenault786eeea2017-05-17 20:00:00 +00001964
1965 if (Lo.getOpcode() == ISD::FNEG) {
Matt Arsenault98f29462017-05-17 20:30:58 +00001966 Lo = stripBitcast(Lo.getOperand(0));
Matt Arsenault786eeea2017-05-17 20:00:00 +00001967 Mods ^= SISrcMods::NEG;
1968 }
1969
1970 if (Hi.getOpcode() == ISD::FNEG) {
Matt Arsenault98f29462017-05-17 20:30:58 +00001971 Hi = stripBitcast(Hi.getOperand(0));
Matt Arsenault786eeea2017-05-17 20:00:00 +00001972 Mods ^= SISrcMods::NEG_HI;
1973 }
1974
Matt Arsenault98f29462017-05-17 20:30:58 +00001975 if (isExtractHiElt(Lo, Lo))
1976 Mods |= SISrcMods::OP_SEL_0;
1977
1978 if (isExtractHiElt(Hi, Hi))
1979 Mods |= SISrcMods::OP_SEL_1;
1980
1981 Lo = stripExtractLoElt(Lo);
1982 Hi = stripExtractLoElt(Hi);
1983
Matt Arsenault786eeea2017-05-17 20:00:00 +00001984 if (Lo == Hi && !isInlineImmediate(Lo.getNode())) {
1985 // Really a scalar input. Just select from the low half of the register to
1986 // avoid packing.
1987
1988 Src = Lo;
1989 SrcMods = CurDAG->getTargetConstant(Mods, SDLoc(In), MVT::i32);
1990 return true;
1991 }
1992
1993 Mods = VecMods;
1994 }
1995
Matt Arsenaulteb522e62017-02-27 22:15:25 +00001996 // Packed instructions do not have abs modifiers.
Matt Arsenaulteb522e62017-02-27 22:15:25 +00001997 Mods |= SISrcMods::OP_SEL_1;
1998
1999 SrcMods = CurDAG->getTargetConstant(Mods, SDLoc(In), MVT::i32);
2000 return true;
2001}
2002
2003bool AMDGPUDAGToDAGISel::SelectVOP3PMods0(SDValue In, SDValue &Src,
2004 SDValue &SrcMods,
2005 SDValue &Clamp) const {
2006 SDLoc SL(In);
2007
2008 // FIXME: Handle clamp and op_sel
2009 Clamp = CurDAG->getTargetConstant(0, SL, MVT::i32);
2010
2011 return SelectVOP3PMods(In, Src, SrcMods);
2012}
2013
Dmitry Preobrazhenskyabf28392017-07-21 13:54:11 +00002014bool AMDGPUDAGToDAGISel::SelectVOP3OpSel(SDValue In, SDValue &Src,
2015 SDValue &SrcMods) const {
2016 Src = In;
2017 // FIXME: Handle op_sel
2018 SrcMods = CurDAG->getTargetConstant(0, SDLoc(In), MVT::i32);
2019 return true;
2020}
2021
2022bool AMDGPUDAGToDAGISel::SelectVOP3OpSel0(SDValue In, SDValue &Src,
2023 SDValue &SrcMods,
2024 SDValue &Clamp) const {
2025 SDLoc SL(In);
2026
2027 // FIXME: Handle clamp
2028 Clamp = CurDAG->getTargetConstant(0, SL, MVT::i32);
2029
2030 return SelectVOP3OpSel(In, Src, SrcMods);
2031}
2032
2033bool AMDGPUDAGToDAGISel::SelectVOP3OpSelMods(SDValue In, SDValue &Src,
2034 SDValue &SrcMods) const {
2035 // FIXME: Handle op_sel
2036 return SelectVOP3Mods(In, Src, SrcMods);
2037}
2038
2039bool AMDGPUDAGToDAGISel::SelectVOP3OpSelMods0(SDValue In, SDValue &Src,
2040 SDValue &SrcMods,
2041 SDValue &Clamp) const {
2042 SDLoc SL(In);
2043
2044 // FIXME: Handle clamp
2045 Clamp = CurDAG->getTargetConstant(0, SL, MVT::i32);
2046
2047 return SelectVOP3OpSelMods(In, Src, SrcMods);
2048}
2049
Matt Arsenaultd7e23032017-09-07 18:05:07 +00002050// The return value is not whether the match is possible (which it always is),
2051// but whether or not it a conversion is really used.
2052bool AMDGPUDAGToDAGISel::SelectVOP3PMadMixModsImpl(SDValue In, SDValue &Src,
2053 unsigned &Mods) const {
2054 Mods = 0;
2055 SelectVOP3ModsImpl(In, Src, Mods);
2056
2057 if (Src.getOpcode() == ISD::FP_EXTEND) {
2058 Src = Src.getOperand(0);
2059 assert(Src.getValueType() == MVT::f16);
2060 Src = stripBitcast(Src);
2061
Matt Arsenault550c66d2017-10-13 20:45:49 +00002062 // Be careful about folding modifiers if we already have an abs. fneg is
2063 // applied last, so we don't want to apply an earlier fneg.
2064 if ((Mods & SISrcMods::ABS) == 0) {
2065 unsigned ModsTmp;
2066 SelectVOP3ModsImpl(Src, Src, ModsTmp);
2067
2068 if ((ModsTmp & SISrcMods::NEG) != 0)
2069 Mods ^= SISrcMods::NEG;
2070
2071 if ((ModsTmp & SISrcMods::ABS) != 0)
2072 Mods |= SISrcMods::ABS;
2073 }
2074
Matt Arsenaultd7e23032017-09-07 18:05:07 +00002075 // op_sel/op_sel_hi decide the source type and source.
2076 // If the source's op_sel_hi is set, it indicates to do a conversion from fp16.
2077 // If the sources's op_sel is set, it picks the high half of the source
2078 // register.
2079
2080 Mods |= SISrcMods::OP_SEL_1;
Matt Arsenault550c66d2017-10-13 20:45:49 +00002081 if (isExtractHiElt(Src, Src)) {
Matt Arsenaultd7e23032017-09-07 18:05:07 +00002082 Mods |= SISrcMods::OP_SEL_0;
2083
Matt Arsenault550c66d2017-10-13 20:45:49 +00002084 // TODO: Should we try to look for neg/abs here?
2085 }
2086
Matt Arsenaultd7e23032017-09-07 18:05:07 +00002087 return true;
2088 }
2089
2090 return false;
2091}
2092
Matt Arsenault76935122017-09-20 20:28:39 +00002093bool AMDGPUDAGToDAGISel::SelectVOP3PMadMixMods(SDValue In, SDValue &Src,
2094 SDValue &SrcMods) const {
2095 unsigned Mods = 0;
2096 SelectVOP3PMadMixModsImpl(In, Src, Mods);
2097 SrcMods = CurDAG->getTargetConstant(Mods, SDLoc(In), MVT::i32);
2098 return true;
2099}
2100
Matt Arsenaulte1cd4822017-11-13 00:22:09 +00002101// TODO: Can we identify things like v_mad_mixhi_f16?
2102bool AMDGPUDAGToDAGISel::SelectHi16Elt(SDValue In, SDValue &Src) const {
2103 if (In.isUndef()) {
2104 Src = In;
2105 return true;
2106 }
2107
2108 if (ConstantSDNode *C = dyn_cast<ConstantSDNode>(In)) {
2109 SDLoc SL(In);
2110 SDValue K = CurDAG->getTargetConstant(C->getZExtValue() << 16, SL, MVT::i32);
2111 MachineSDNode *MovK = CurDAG->getMachineNode(AMDGPU::V_MOV_B32_e32,
2112 SL, MVT::i32, K);
2113 Src = SDValue(MovK, 0);
2114 return true;
2115 }
2116
2117 if (ConstantFPSDNode *C = dyn_cast<ConstantFPSDNode>(In)) {
2118 SDLoc SL(In);
2119 SDValue K = CurDAG->getTargetConstant(
2120 C->getValueAPF().bitcastToAPInt().getZExtValue() << 16, SL, MVT::i32);
2121 MachineSDNode *MovK = CurDAG->getMachineNode(AMDGPU::V_MOV_B32_e32,
2122 SL, MVT::i32, K);
2123 Src = SDValue(MovK, 0);
2124 return true;
2125 }
2126
2127 return isExtractHiElt(In, Src);
2128}
2129
Christian Konigd910b7d2013-02-26 17:52:16 +00002130void AMDGPUDAGToDAGISel::PostprocessISelDAG() {
Bill Wendlinga3cd3502013-06-19 21:36:55 +00002131 const AMDGPUTargetLowering& Lowering =
Matt Arsenault209a7b92014-04-18 07:40:20 +00002132 *static_cast<const AMDGPUTargetLowering*>(getTargetLowering());
Vincent Lejeuneab3baf82013-09-12 23:44:44 +00002133 bool IsModified = false;
2134 do {
2135 IsModified = false;
Matt Arsenault68f05052017-12-04 22:18:27 +00002136
Vincent Lejeuneab3baf82013-09-12 23:44:44 +00002137 // Go over all selected nodes and try to fold them a bit more
Matt Arsenault68f05052017-12-04 22:18:27 +00002138 SelectionDAG::allnodes_iterator Position = CurDAG->allnodes_begin();
2139 while (Position != CurDAG->allnodes_end()) {
2140 SDNode *Node = &*Position++;
2141 MachineSDNode *MachineNode = dyn_cast<MachineSDNode>(Node);
Vincent Lejeuneab3baf82013-09-12 23:44:44 +00002142 if (!MachineNode)
2143 continue;
Christian Konigd910b7d2013-02-26 17:52:16 +00002144
Vincent Lejeuneab3baf82013-09-12 23:44:44 +00002145 SDNode *ResNode = Lowering.PostISelFolding(MachineNode, *CurDAG);
Matt Arsenault68f05052017-12-04 22:18:27 +00002146 if (ResNode != Node) {
2147 if (ResNode)
2148 ReplaceUses(Node, ResNode);
Vincent Lejeuneab3baf82013-09-12 23:44:44 +00002149 IsModified = true;
2150 }
Tom Stellard2183b702013-06-03 17:39:46 +00002151 }
Vincent Lejeuneab3baf82013-09-12 23:44:44 +00002152 CurDAG->RemoveDeadNodes();
2153 } while (IsModified);
Christian Konigd910b7d2013-02-26 17:52:16 +00002154}
Tom Stellard20287692017-08-08 04:57:55 +00002155
Tom Stellardc5a154d2018-06-28 23:47:12 +00002156bool R600DAGToDAGISel::runOnMachineFunction(MachineFunction &MF) {
2157 Subtarget = &MF.getSubtarget<R600Subtarget>();
2158 return SelectionDAGISel::runOnMachineFunction(MF);
2159}
2160
2161bool R600DAGToDAGISel::isConstantLoad(const MemSDNode *N, int CbId) const {
2162 if (!N->readMem())
2163 return false;
2164 if (CbId == -1)
2165 return N->getAddressSpace() == AMDGPUASI.CONSTANT_ADDRESS ||
2166 N->getAddressSpace() == AMDGPUASI.CONSTANT_ADDRESS_32BIT;
2167
2168 return N->getAddressSpace() == AMDGPUASI.CONSTANT_BUFFER_0 + CbId;
2169}
2170
2171bool R600DAGToDAGISel::SelectGlobalValueConstantOffset(SDValue Addr,
2172 SDValue& IntPtr) {
2173 if (ConstantSDNode *Cst = dyn_cast<ConstantSDNode>(Addr)) {
2174 IntPtr = CurDAG->getIntPtrConstant(Cst->getZExtValue() / 4, SDLoc(Addr),
2175 true);
2176 return true;
2177 }
2178 return false;
2179}
2180
2181bool R600DAGToDAGISel::SelectGlobalValueVariableOffset(SDValue Addr,
2182 SDValue& BaseReg, SDValue &Offset) {
2183 if (!isa<ConstantSDNode>(Addr)) {
2184 BaseReg = Addr;
2185 Offset = CurDAG->getIntPtrConstant(0, SDLoc(Addr), true);
2186 return true;
2187 }
2188 return false;
2189}
2190
Tom Stellard20287692017-08-08 04:57:55 +00002191void R600DAGToDAGISel::Select(SDNode *N) {
2192 unsigned int Opc = N->getOpcode();
2193 if (N->isMachineOpcode()) {
2194 N->setNodeId(-1);
2195 return; // Already selected.
2196 }
2197
2198 switch (Opc) {
2199 default: break;
2200 case AMDGPUISD::BUILD_VERTICAL_VECTOR:
2201 case ISD::SCALAR_TO_VECTOR:
2202 case ISD::BUILD_VECTOR: {
2203 EVT VT = N->getValueType(0);
2204 unsigned NumVectorElts = VT.getVectorNumElements();
2205 unsigned RegClassID;
2206 // BUILD_VECTOR was lowered into an IMPLICIT_DEF + 4 INSERT_SUBREG
2207 // that adds a 128 bits reg copy when going through TwoAddressInstructions
2208 // pass. We want to avoid 128 bits copies as much as possible because they
2209 // can't be bundled by our scheduler.
2210 switch(NumVectorElts) {
Tom Stellardc5a154d2018-06-28 23:47:12 +00002211 case 2: RegClassID = R600::R600_Reg64RegClassID; break;
Tom Stellard20287692017-08-08 04:57:55 +00002212 case 4:
2213 if (Opc == AMDGPUISD::BUILD_VERTICAL_VECTOR)
Tom Stellardc5a154d2018-06-28 23:47:12 +00002214 RegClassID = R600::R600_Reg128VerticalRegClassID;
Tom Stellard20287692017-08-08 04:57:55 +00002215 else
Tom Stellardc5a154d2018-06-28 23:47:12 +00002216 RegClassID = R600::R600_Reg128RegClassID;
Tom Stellard20287692017-08-08 04:57:55 +00002217 break;
2218 default: llvm_unreachable("Do not know how to lower this BUILD_VECTOR");
2219 }
2220 SelectBuildVector(N, RegClassID);
2221 return;
2222 }
2223 }
2224
2225 SelectCode(N);
2226}
2227
2228bool R600DAGToDAGISel::SelectADDRIndirect(SDValue Addr, SDValue &Base,
2229 SDValue &Offset) {
2230 ConstantSDNode *C;
2231 SDLoc DL(Addr);
2232
2233 if ((C = dyn_cast<ConstantSDNode>(Addr))) {
Tom Stellardc5a154d2018-06-28 23:47:12 +00002234 Base = CurDAG->getRegister(R600::INDIRECT_BASE_ADDR, MVT::i32);
Tom Stellard20287692017-08-08 04:57:55 +00002235 Offset = CurDAG->getTargetConstant(C->getZExtValue(), DL, MVT::i32);
2236 } else if ((Addr.getOpcode() == AMDGPUISD::DWORDADDR) &&
2237 (C = dyn_cast<ConstantSDNode>(Addr.getOperand(0)))) {
Tom Stellardc5a154d2018-06-28 23:47:12 +00002238 Base = CurDAG->getRegister(R600::INDIRECT_BASE_ADDR, MVT::i32);
Tom Stellard20287692017-08-08 04:57:55 +00002239 Offset = CurDAG->getTargetConstant(C->getZExtValue(), DL, MVT::i32);
2240 } else if ((Addr.getOpcode() == ISD::ADD || Addr.getOpcode() == ISD::OR) &&
2241 (C = dyn_cast<ConstantSDNode>(Addr.getOperand(1)))) {
2242 Base = Addr.getOperand(0);
2243 Offset = CurDAG->getTargetConstant(C->getZExtValue(), DL, MVT::i32);
2244 } else {
2245 Base = Addr;
2246 Offset = CurDAG->getTargetConstant(0, DL, MVT::i32);
2247 }
2248
2249 return true;
2250}
2251
2252bool R600DAGToDAGISel::SelectADDRVTX_READ(SDValue Addr, SDValue &Base,
2253 SDValue &Offset) {
2254 ConstantSDNode *IMMOffset;
2255
2256 if (Addr.getOpcode() == ISD::ADD
2257 && (IMMOffset = dyn_cast<ConstantSDNode>(Addr.getOperand(1)))
2258 && isInt<16>(IMMOffset->getZExtValue())) {
2259
2260 Base = Addr.getOperand(0);
2261 Offset = CurDAG->getTargetConstant(IMMOffset->getZExtValue(), SDLoc(Addr),
2262 MVT::i32);
2263 return true;
2264 // If the pointer address is constant, we can move it to the offset field.
2265 } else if ((IMMOffset = dyn_cast<ConstantSDNode>(Addr))
2266 && isInt<16>(IMMOffset->getZExtValue())) {
2267 Base = CurDAG->getCopyFromReg(CurDAG->getEntryNode(),
2268 SDLoc(CurDAG->getEntryNode()),
Tom Stellardc5a154d2018-06-28 23:47:12 +00002269 R600::ZERO, MVT::i32);
Tom Stellard20287692017-08-08 04:57:55 +00002270 Offset = CurDAG->getTargetConstant(IMMOffset->getZExtValue(), SDLoc(Addr),
2271 MVT::i32);
2272 return true;
2273 }
2274
2275 // Default case, no offset
2276 Base = Addr;
2277 Offset = CurDAG->getTargetConstant(0, SDLoc(Addr), MVT::i32);
2278 return true;
2279}