blob: 66291d0be4e6b67148396acb5214f8ea8213b8c6 [file] [log] [blame]
Tom Stellard75aadc22012-12-11 21:25:42 +00001//===-- R600ISelLowering.cpp - R600 DAG Lowering Implementation -----------===//
2//
3// The LLVM Compiler Infrastructure
4//
5// This file is distributed under the University of Illinois Open Source
6// License. See LICENSE.TXT for details.
7//
8//===----------------------------------------------------------------------===//
9//
10/// \file
11/// \brief Custom DAG lowering for R600
12//
13//===----------------------------------------------------------------------===//
14
15#include "R600ISelLowering.h"
Tom Stellard2e59a452014-06-13 01:32:00 +000016#include "AMDGPUFrameLowering.h"
Matt Arsenaultc791f392014-06-23 18:00:31 +000017#include "AMDGPUIntrinsicInfo.h"
Tom Stellard2e59a452014-06-13 01:32:00 +000018#include "AMDGPUSubtarget.h"
Tom Stellard75aadc22012-12-11 21:25:42 +000019#include "R600Defines.h"
Eugene Zelenko2bc2f332016-12-09 22:06:55 +000020#include "R600FrameLowering.h"
Tom Stellard75aadc22012-12-11 21:25:42 +000021#include "R600InstrInfo.h"
22#include "R600MachineFunctionInfo.h"
Eugene Zelenko2bc2f332016-12-09 22:06:55 +000023#include "Utils/AMDGPUBaseInfo.h"
24#include "llvm/ADT/APFloat.h"
25#include "llvm/ADT/APInt.h"
26#include "llvm/ADT/ArrayRef.h"
27#include "llvm/ADT/DenseMap.h"
28#include "llvm/ADT/SmallVector.h"
Tom Stellardacfeebf2013-07-23 01:48:05 +000029#include "llvm/CodeGen/CallingConvLower.h"
Eugene Zelenko2bc2f332016-12-09 22:06:55 +000030#include "llvm/CodeGen/DAGCombine.h"
31#include "llvm/CodeGen/ISDOpcodes.h"
32#include "llvm/CodeGen/MachineBasicBlock.h"
33#include "llvm/CodeGen/MachineFunction.h"
34#include "llvm/CodeGen/MachineInstr.h"
Tom Stellard75aadc22012-12-11 21:25:42 +000035#include "llvm/CodeGen/MachineInstrBuilder.h"
Eugene Zelenko2bc2f332016-12-09 22:06:55 +000036#include "llvm/CodeGen/MachineMemOperand.h"
Tom Stellard75aadc22012-12-11 21:25:42 +000037#include "llvm/CodeGen/MachineRegisterInfo.h"
Eugene Zelenko2bc2f332016-12-09 22:06:55 +000038#include "llvm/CodeGen/MachineValueType.h"
Tom Stellard75aadc22012-12-11 21:25:42 +000039#include "llvm/CodeGen/SelectionDAG.h"
Eugene Zelenko2bc2f332016-12-09 22:06:55 +000040#include "llvm/IR/Constants.h"
41#include "llvm/IR/DerivedTypes.h"
42#include "llvm/Support/Casting.h"
43#include "llvm/Support/Compiler.h"
44#include "llvm/Support/ErrorHandling.h"
45#include <cassert>
46#include <cstdint>
47#include <iterator>
48#include <utility>
49#include <vector>
Tom Stellard75aadc22012-12-11 21:25:42 +000050
51using namespace llvm;
52
Matt Arsenault43e92fe2016-06-24 06:30:11 +000053R600TargetLowering::R600TargetLowering(const TargetMachine &TM,
54 const R600Subtarget &STI)
Eric Christopher7792e322015-01-30 23:24:40 +000055 : AMDGPUTargetLowering(TM, STI), Gen(STI.getGeneration()) {
Tom Stellard75aadc22012-12-11 21:25:42 +000056 addRegisterClass(MVT::f32, &AMDGPU::R600_Reg32RegClass);
Tom Stellard75aadc22012-12-11 21:25:42 +000057 addRegisterClass(MVT::i32, &AMDGPU::R600_Reg32RegClass);
Tom Stellard0344cdf2013-08-01 15:23:42 +000058 addRegisterClass(MVT::v2f32, &AMDGPU::R600_Reg64RegClass);
59 addRegisterClass(MVT::v2i32, &AMDGPU::R600_Reg64RegClass);
Matt Arsenault71e66762016-05-21 02:27:49 +000060 addRegisterClass(MVT::v4f32, &AMDGPU::R600_Reg128RegClass);
61 addRegisterClass(MVT::v4i32, &AMDGPU::R600_Reg128RegClass);
Tom Stellard0344cdf2013-08-01 15:23:42 +000062
Eric Christopher23a3a7c2015-02-26 00:00:24 +000063 computeRegisterProperties(STI.getRegisterInfo());
Tom Stellard75aadc22012-12-11 21:25:42 +000064
Matt Arsenault71e66762016-05-21 02:27:49 +000065 // Legalize loads and stores to the private address space.
66 setOperationAction(ISD::LOAD, MVT::i32, Custom);
67 setOperationAction(ISD::LOAD, MVT::v2i32, Custom);
68 setOperationAction(ISD::LOAD, MVT::v4i32, Custom);
69
70 // EXTLOAD should be the same as ZEXTLOAD. It is legal for some address
71 // spaces, so it is custom lowered to handle those where it isn't.
72 for (MVT VT : MVT::integer_valuetypes()) {
73 setLoadExtAction(ISD::SEXTLOAD, VT, MVT::i1, Promote);
74 setLoadExtAction(ISD::SEXTLOAD, VT, MVT::i8, Custom);
75 setLoadExtAction(ISD::SEXTLOAD, VT, MVT::i16, Custom);
76
77 setLoadExtAction(ISD::ZEXTLOAD, VT, MVT::i1, Promote);
78 setLoadExtAction(ISD::ZEXTLOAD, VT, MVT::i8, Custom);
79 setLoadExtAction(ISD::ZEXTLOAD, VT, MVT::i16, Custom);
80
81 setLoadExtAction(ISD::EXTLOAD, VT, MVT::i1, Promote);
82 setLoadExtAction(ISD::EXTLOAD, VT, MVT::i8, Custom);
83 setLoadExtAction(ISD::EXTLOAD, VT, MVT::i16, Custom);
84 }
85
Matt Arsenaultd1097a32016-06-02 19:54:26 +000086 // Workaround for LegalizeDAG asserting on expansion of i1 vector loads.
87 setLoadExtAction(ISD::EXTLOAD, MVT::v2i32, MVT::v2i1, Expand);
88 setLoadExtAction(ISD::SEXTLOAD, MVT::v2i32, MVT::v2i1, Expand);
89 setLoadExtAction(ISD::ZEXTLOAD, MVT::v2i32, MVT::v2i1, Expand);
90
91 setLoadExtAction(ISD::EXTLOAD, MVT::v4i32, MVT::v4i1, Expand);
92 setLoadExtAction(ISD::SEXTLOAD, MVT::v4i32, MVT::v4i1, Expand);
93 setLoadExtAction(ISD::ZEXTLOAD, MVT::v4i32, MVT::v4i1, Expand);
94
Matt Arsenault71e66762016-05-21 02:27:49 +000095 setOperationAction(ISD::STORE, MVT::i8, Custom);
96 setOperationAction(ISD::STORE, MVT::i32, Custom);
97 setOperationAction(ISD::STORE, MVT::v2i32, Custom);
98 setOperationAction(ISD::STORE, MVT::v4i32, Custom);
99
100 setTruncStoreAction(MVT::i32, MVT::i8, Custom);
101 setTruncStoreAction(MVT::i32, MVT::i16, Custom);
Jan Vesely06200bd2017-01-06 21:00:46 +0000102 // We need to include these since trunc STORES to PRIVATE need
103 // special handling to accommodate RMW
104 setTruncStoreAction(MVT::v2i32, MVT::v2i16, Custom);
105 setTruncStoreAction(MVT::v4i32, MVT::v4i16, Custom);
106 setTruncStoreAction(MVT::v8i32, MVT::v8i16, Custom);
107 setTruncStoreAction(MVT::v16i32, MVT::v16i16, Custom);
108 setTruncStoreAction(MVT::v32i32, MVT::v32i16, Custom);
109 setTruncStoreAction(MVT::v2i32, MVT::v2i8, Custom);
110 setTruncStoreAction(MVT::v4i32, MVT::v4i8, Custom);
111 setTruncStoreAction(MVT::v8i32, MVT::v8i8, Custom);
112 setTruncStoreAction(MVT::v16i32, MVT::v16i8, Custom);
113 setTruncStoreAction(MVT::v32i32, MVT::v32i8, Custom);
Matt Arsenault71e66762016-05-21 02:27:49 +0000114
Matt Arsenaultd1097a32016-06-02 19:54:26 +0000115 // Workaround for LegalizeDAG asserting on expansion of i1 vector stores.
116 setTruncStoreAction(MVT::v2i32, MVT::v2i1, Expand);
117 setTruncStoreAction(MVT::v4i32, MVT::v4i1, Expand);
118
Tom Stellard0351ea22013-09-28 02:50:50 +0000119 // Set condition code actions
120 setCondCodeAction(ISD::SETO, MVT::f32, Expand);
121 setCondCodeAction(ISD::SETUO, MVT::f32, Expand);
Tom Stellardcd428182013-09-28 02:50:38 +0000122 setCondCodeAction(ISD::SETLT, MVT::f32, Expand);
Tom Stellard0351ea22013-09-28 02:50:50 +0000123 setCondCodeAction(ISD::SETLE, MVT::f32, Expand);
Tom Stellardcd428182013-09-28 02:50:38 +0000124 setCondCodeAction(ISD::SETOLT, MVT::f32, Expand);
125 setCondCodeAction(ISD::SETOLE, MVT::f32, Expand);
Tom Stellard0351ea22013-09-28 02:50:50 +0000126 setCondCodeAction(ISD::SETONE, MVT::f32, Expand);
127 setCondCodeAction(ISD::SETUEQ, MVT::f32, Expand);
128 setCondCodeAction(ISD::SETUGE, MVT::f32, Expand);
129 setCondCodeAction(ISD::SETUGT, MVT::f32, Expand);
Tom Stellardcd428182013-09-28 02:50:38 +0000130 setCondCodeAction(ISD::SETULT, MVT::f32, Expand);
131 setCondCodeAction(ISD::SETULE, MVT::f32, Expand);
132
133 setCondCodeAction(ISD::SETLE, MVT::i32, Expand);
134 setCondCodeAction(ISD::SETLT, MVT::i32, Expand);
135 setCondCodeAction(ISD::SETULE, MVT::i32, Expand);
136 setCondCodeAction(ISD::SETULT, MVT::i32, Expand);
137
Vincent Lejeuneb55940c2013-07-09 15:03:11 +0000138 setOperationAction(ISD::FCOS, MVT::f32, Custom);
139 setOperationAction(ISD::FSIN, MVT::f32, Custom);
140
Tom Stellard75aadc22012-12-11 21:25:42 +0000141 setOperationAction(ISD::SETCC, MVT::v4i32, Expand);
Tom Stellard0344cdf2013-08-01 15:23:42 +0000142 setOperationAction(ISD::SETCC, MVT::v2i32, Expand);
Tom Stellard75aadc22012-12-11 21:25:42 +0000143
Tom Stellard492ebea2013-03-08 15:37:07 +0000144 setOperationAction(ISD::BR_CC, MVT::i32, Expand);
145 setOperationAction(ISD::BR_CC, MVT::f32, Expand);
Matt Arsenault1d555c42014-06-23 18:00:55 +0000146 setOperationAction(ISD::BRCOND, MVT::Other, Custom);
Tom Stellard75aadc22012-12-11 21:25:42 +0000147
148 setOperationAction(ISD::FSUB, MVT::f32, Expand);
149
Tom Stellard75aadc22012-12-11 21:25:42 +0000150 setOperationAction(ISD::SELECT_CC, MVT::f32, Custom);
151 setOperationAction(ISD::SELECT_CC, MVT::i32, Custom);
152
Tom Stellarde8f9f282013-03-08 15:37:05 +0000153 setOperationAction(ISD::SETCC, MVT::i32, Expand);
154 setOperationAction(ISD::SETCC, MVT::f32, Expand);
Tom Stellard75aadc22012-12-11 21:25:42 +0000155 setOperationAction(ISD::FP_TO_UINT, MVT::i1, Custom);
Matt Arsenault7fb961f2016-07-22 17:01:21 +0000156 setOperationAction(ISD::FP_TO_SINT, MVT::i1, Custom);
Jan Vesely2cb62ce2014-07-10 22:40:21 +0000157 setOperationAction(ISD::FP_TO_SINT, MVT::i64, Custom);
158 setOperationAction(ISD::FP_TO_UINT, MVT::i64, Custom);
Tom Stellard75aadc22012-12-11 21:25:42 +0000159
Tom Stellard53f2f902013-09-05 18:38:03 +0000160 setOperationAction(ISD::SELECT, MVT::i32, Expand);
161 setOperationAction(ISD::SELECT, MVT::f32, Expand);
162 setOperationAction(ISD::SELECT, MVT::v2i32, Expand);
Tom Stellard53f2f902013-09-05 18:38:03 +0000163 setOperationAction(ISD::SELECT, MVT::v4i32, Expand);
Tom Stellard75aadc22012-12-11 21:25:42 +0000164
Jan Vesely808fff52015-04-30 17:15:56 +0000165 // ADD, SUB overflow.
166 // TODO: turn these into Legal?
167 if (Subtarget->hasCARRY())
168 setOperationAction(ISD::UADDO, MVT::i32, Custom);
169
170 if (Subtarget->hasBORROW())
171 setOperationAction(ISD::USUBO, MVT::i32, Custom);
172
Matt Arsenault4e466652014-04-16 01:41:30 +0000173 // Expand sign extension of vectors
174 if (!Subtarget->hasBFE())
175 setOperationAction(ISD::SIGN_EXTEND_INREG, MVT::i1, Expand);
176
177 setOperationAction(ISD::SIGN_EXTEND_INREG, MVT::v2i1, Expand);
178 setOperationAction(ISD::SIGN_EXTEND_INREG, MVT::v4i1, Expand);
179
180 if (!Subtarget->hasBFE())
181 setOperationAction(ISD::SIGN_EXTEND_INREG, MVT::i8, Expand);
182 setOperationAction(ISD::SIGN_EXTEND_INREG, MVT::v2i8, Expand);
183 setOperationAction(ISD::SIGN_EXTEND_INREG, MVT::v4i8, Expand);
184
185 if (!Subtarget->hasBFE())
186 setOperationAction(ISD::SIGN_EXTEND_INREG, MVT::i16, Expand);
187 setOperationAction(ISD::SIGN_EXTEND_INREG, MVT::v2i16, Expand);
188 setOperationAction(ISD::SIGN_EXTEND_INREG, MVT::v4i16, Expand);
189
190 setOperationAction(ISD::SIGN_EXTEND_INREG, MVT::i32, Legal);
191 setOperationAction(ISD::SIGN_EXTEND_INREG, MVT::v2i32, Expand);
192 setOperationAction(ISD::SIGN_EXTEND_INREG, MVT::v4i32, Expand);
193
194 setOperationAction(ISD::SIGN_EXTEND_INREG, MVT::Other, Expand);
195
Tom Stellardf3b2a1e2013-02-06 17:32:29 +0000196 setOperationAction(ISD::FrameIndex, MVT::i32, Custom);
197
Tom Stellard880a80a2014-06-17 16:53:14 +0000198 setOperationAction(ISD::EXTRACT_VECTOR_ELT, MVT::v2i32, Custom);
199 setOperationAction(ISD::EXTRACT_VECTOR_ELT, MVT::v2f32, Custom);
200 setOperationAction(ISD::EXTRACT_VECTOR_ELT, MVT::v4i32, Custom);
201 setOperationAction(ISD::EXTRACT_VECTOR_ELT, MVT::v4f32, Custom);
202
203 setOperationAction(ISD::INSERT_VECTOR_ELT, MVT::v2i32, Custom);
204 setOperationAction(ISD::INSERT_VECTOR_ELT, MVT::v2f32, Custom);
205 setOperationAction(ISD::INSERT_VECTOR_ELT, MVT::v4i32, Custom);
206 setOperationAction(ISD::INSERT_VECTOR_ELT, MVT::v4f32, Custom);
207
Jan Vesely25f36272014-06-18 12:27:13 +0000208 // We don't have 64-bit shifts. Thus we need either SHX i64 or SHX_PARTS i32
209 // to be Legal/Custom in order to avoid library calls.
210 setOperationAction(ISD::SHL_PARTS, MVT::i32, Custom);
Jan Vesely900ff2e2014-06-18 12:27:15 +0000211 setOperationAction(ISD::SRL_PARTS, MVT::i32, Custom);
Jan Veselyecf51332014-06-18 12:27:17 +0000212 setOperationAction(ISD::SRA_PARTS, MVT::i32, Custom);
Jan Vesely25f36272014-06-18 12:27:13 +0000213
Jan Vesely39aeab42017-12-04 23:07:28 +0000214 if (!Subtarget->hasFMA()) {
215 setOperationAction(ISD::FMA, MVT::f32, Expand);
216 setOperationAction(ISD::FMA, MVT::f64, Expand);
217 }
218
Michel Danzer49812b52013-07-10 16:37:07 +0000219 setOperationAction(ISD::GlobalAddress, MVT::i32, Custom);
220
Matt Arsenaultc4d3d3a2014-06-23 18:00:49 +0000221 const MVT ScalarIntVTs[] = { MVT::i32, MVT::i64 };
222 for (MVT VT : ScalarIntVTs) {
223 setOperationAction(ISD::ADDC, VT, Expand);
224 setOperationAction(ISD::SUBC, VT, Expand);
225 setOperationAction(ISD::ADDE, VT, Expand);
226 setOperationAction(ISD::SUBE, VT, Expand);
227 }
228
Jan Vesely334f51a2017-01-16 21:20:13 +0000229 // LLVM will expand these to atomic_cmp_swap(0)
230 // and atomic_swap, respectively.
231 setOperationAction(ISD::ATOMIC_LOAD, MVT::i32, Expand);
232 setOperationAction(ISD::ATOMIC_STORE, MVT::i32, Expand);
233
Matt Arsenault754dd3e2017-04-03 18:08:08 +0000234 // We need to custom lower some of the intrinsics
235 setOperationAction(ISD::INTRINSIC_VOID, MVT::Other, Custom);
236 setOperationAction(ISD::INTRINSIC_WO_CHAIN, MVT::Other, Custom);
237
Tom Stellardfc455472013-08-12 22:33:21 +0000238 setSchedulingPreference(Sched::Source);
Matt Arsenault71e66762016-05-21 02:27:49 +0000239
Matt Arsenault71e66762016-05-21 02:27:49 +0000240 setTargetDAGCombine(ISD::FP_ROUND);
241 setTargetDAGCombine(ISD::FP_TO_SINT);
242 setTargetDAGCombine(ISD::EXTRACT_VECTOR_ELT);
243 setTargetDAGCombine(ISD::SELECT_CC);
244 setTargetDAGCombine(ISD::INSERT_VECTOR_ELT);
Jan Vesely38814fa2016-08-27 19:09:43 +0000245 setTargetDAGCombine(ISD::LOAD);
Tom Stellard75aadc22012-12-11 21:25:42 +0000246}
247
Matt Arsenault43e92fe2016-06-24 06:30:11 +0000248const R600Subtarget *R600TargetLowering::getSubtarget() const {
249 return static_cast<const R600Subtarget *>(Subtarget);
250}
251
Tom Stellardc0f0fba2015-10-01 17:51:29 +0000252static inline bool isEOP(MachineBasicBlock::iterator I) {
Hans Wennborg0dd9ed12016-08-13 01:12:49 +0000253 if (std::next(I) == I->getParent()->end())
254 return false;
Tom Stellardc0f0fba2015-10-01 17:51:29 +0000255 return std::next(I)->getOpcode() == AMDGPU::RETURN;
256}
257
Duncan P. N. Exon Smithe4f5e4f2016-06-30 22:52:52 +0000258MachineBasicBlock *
259R600TargetLowering::EmitInstrWithCustomInserter(MachineInstr &MI,
260 MachineBasicBlock *BB) const {
Eugene Zelenko2bc2f332016-12-09 22:06:55 +0000261 MachineFunction *MF = BB->getParent();
Tom Stellard75aadc22012-12-11 21:25:42 +0000262 MachineRegisterInfo &MRI = MF->getRegInfo();
Duncan P. N. Exon Smithe4f5e4f2016-06-30 22:52:52 +0000263 MachineBasicBlock::iterator I = MI;
Matt Arsenault43e92fe2016-06-24 06:30:11 +0000264 const R600InstrInfo *TII = getSubtarget()->getInstrInfo();
Tom Stellard75aadc22012-12-11 21:25:42 +0000265
Duncan P. N. Exon Smithe4f5e4f2016-06-30 22:52:52 +0000266 switch (MI.getOpcode()) {
Tom Stellardc6f4a292013-08-26 15:05:59 +0000267 default:
Tom Stellard8f9fc202013-11-15 00:12:45 +0000268 // Replace LDS_*_RET instruction that don't have any uses with the
269 // equivalent LDS_*_NORET instruction.
Duncan P. N. Exon Smithe4f5e4f2016-06-30 22:52:52 +0000270 if (TII->isLDSRetInstr(MI.getOpcode())) {
271 int DstIdx = TII->getOperandIdx(MI.getOpcode(), AMDGPU::OpName::dst);
Tom Stellard13c68ef2013-09-05 18:38:09 +0000272 assert(DstIdx != -1);
273 MachineInstrBuilder NewMI;
Aaron Watry1885e532014-09-11 15:02:54 +0000274 // FIXME: getLDSNoRetOp method only handles LDS_1A1D LDS ops. Add
275 // LDS_1A2D support and remove this special case.
Duncan P. N. Exon Smithe4f5e4f2016-06-30 22:52:52 +0000276 if (!MRI.use_empty(MI.getOperand(DstIdx).getReg()) ||
277 MI.getOpcode() == AMDGPU::LDS_CMPST_RET)
Tom Stellard8f9fc202013-11-15 00:12:45 +0000278 return BB;
279
280 NewMI = BuildMI(*BB, I, BB->findDebugLoc(I),
Duncan P. N. Exon Smithe4f5e4f2016-06-30 22:52:52 +0000281 TII->get(AMDGPU::getLDSNoRetOp(MI.getOpcode())));
282 for (unsigned i = 1, e = MI.getNumOperands(); i < e; ++i) {
Diana Picus116bbab2017-01-13 09:58:52 +0000283 NewMI.add(MI.getOperand(i));
Tom Stellardc6f4a292013-08-26 15:05:59 +0000284 }
Tom Stellardc6f4a292013-08-26 15:05:59 +0000285 } else {
286 return AMDGPUTargetLowering::EmitInstrWithCustomInserter(MI, BB);
287 }
288 break;
Tom Stellard75aadc22012-12-11 21:25:42 +0000289 case AMDGPU::CLAMP_R600: {
Duncan P. N. Exon Smithe4f5e4f2016-06-30 22:52:52 +0000290 MachineInstr *NewMI = TII->buildDefaultInstruction(
291 *BB, I, AMDGPU::MOV, MI.getOperand(0).getReg(),
292 MI.getOperand(1).getReg());
Duncan P. N. Exon Smith9cfc75c2016-06-30 00:01:54 +0000293 TII->addFlag(*NewMI, 0, MO_FLAG_CLAMP);
Tom Stellard75aadc22012-12-11 21:25:42 +0000294 break;
295 }
296
297 case AMDGPU::FABS_R600: {
Duncan P. N. Exon Smithe4f5e4f2016-06-30 22:52:52 +0000298 MachineInstr *NewMI = TII->buildDefaultInstruction(
299 *BB, I, AMDGPU::MOV, MI.getOperand(0).getReg(),
300 MI.getOperand(1).getReg());
Duncan P. N. Exon Smith9cfc75c2016-06-30 00:01:54 +0000301 TII->addFlag(*NewMI, 0, MO_FLAG_ABS);
Tom Stellard75aadc22012-12-11 21:25:42 +0000302 break;
303 }
304
305 case AMDGPU::FNEG_R600: {
Duncan P. N. Exon Smithe4f5e4f2016-06-30 22:52:52 +0000306 MachineInstr *NewMI = TII->buildDefaultInstruction(
307 *BB, I, AMDGPU::MOV, MI.getOperand(0).getReg(),
308 MI.getOperand(1).getReg());
Duncan P. N. Exon Smith9cfc75c2016-06-30 00:01:54 +0000309 TII->addFlag(*NewMI, 0, MO_FLAG_NEG);
Tom Stellard75aadc22012-12-11 21:25:42 +0000310 break;
311 }
312
Tom Stellard75aadc22012-12-11 21:25:42 +0000313 case AMDGPU::MASK_WRITE: {
Duncan P. N. Exon Smithe4f5e4f2016-06-30 22:52:52 +0000314 unsigned maskedRegister = MI.getOperand(0).getReg();
Tom Stellard75aadc22012-12-11 21:25:42 +0000315 assert(TargetRegisterInfo::isVirtualRegister(maskedRegister));
316 MachineInstr * defInstr = MRI.getVRegDef(maskedRegister);
Duncan P. N. Exon Smith9cfc75c2016-06-30 00:01:54 +0000317 TII->addFlag(*defInstr, 0, MO_FLAG_MASK);
Tom Stellard75aadc22012-12-11 21:25:42 +0000318 break;
319 }
320
321 case AMDGPU::MOV_IMM_F32:
Duncan P. N. Exon Smithe4f5e4f2016-06-30 22:52:52 +0000322 TII->buildMovImm(*BB, I, MI.getOperand(0).getReg(), MI.getOperand(1)
323 .getFPImm()
324 ->getValueAPF()
325 .bitcastToAPInt()
326 .getZExtValue());
Tom Stellard75aadc22012-12-11 21:25:42 +0000327 break;
Eugene Zelenko2bc2f332016-12-09 22:06:55 +0000328
Tom Stellard75aadc22012-12-11 21:25:42 +0000329 case AMDGPU::MOV_IMM_I32:
Duncan P. N. Exon Smithe4f5e4f2016-06-30 22:52:52 +0000330 TII->buildMovImm(*BB, I, MI.getOperand(0).getReg(),
331 MI.getOperand(1).getImm());
Tom Stellard75aadc22012-12-11 21:25:42 +0000332 break;
Eugene Zelenko2bc2f332016-12-09 22:06:55 +0000333
Jan Veselyf97de002016-05-13 20:39:29 +0000334 case AMDGPU::MOV_IMM_GLOBAL_ADDR: {
335 //TODO: Perhaps combine this instruction with the next if possible
Duncan P. N. Exon Smithe4f5e4f2016-06-30 22:52:52 +0000336 auto MIB = TII->buildDefaultInstruction(
337 *BB, MI, AMDGPU::MOV, MI.getOperand(0).getReg(), AMDGPU::ALU_LITERAL_X);
Jan Veselyf97de002016-05-13 20:39:29 +0000338 int Idx = TII->getOperandIdx(*MIB, AMDGPU::OpName::literal);
339 //TODO: Ugh this is rather ugly
Duncan P. N. Exon Smithe4f5e4f2016-06-30 22:52:52 +0000340 MIB->getOperand(Idx) = MI.getOperand(1);
Jan Veselyf97de002016-05-13 20:39:29 +0000341 break;
342 }
Eugene Zelenko2bc2f332016-12-09 22:06:55 +0000343
Vincent Lejeune0b72f102013-03-05 15:04:55 +0000344 case AMDGPU::CONST_COPY: {
Duncan P. N. Exon Smithe4f5e4f2016-06-30 22:52:52 +0000345 MachineInstr *NewMI = TII->buildDefaultInstruction(
346 *BB, MI, AMDGPU::MOV, MI.getOperand(0).getReg(), AMDGPU::ALU_CONST);
Duncan P. N. Exon Smith9cfc75c2016-06-30 00:01:54 +0000347 TII->setImmOperand(*NewMI, AMDGPU::OpName::src0_sel,
Duncan P. N. Exon Smithe4f5e4f2016-06-30 22:52:52 +0000348 MI.getOperand(1).getImm());
Vincent Lejeune0b72f102013-03-05 15:04:55 +0000349 break;
350 }
Tom Stellard75aadc22012-12-11 21:25:42 +0000351
352 case AMDGPU::RAT_WRITE_CACHELESS_32_eg:
Tom Stellard0344cdf2013-08-01 15:23:42 +0000353 case AMDGPU::RAT_WRITE_CACHELESS_64_eg:
Eugene Zelenko2bc2f332016-12-09 22:06:55 +0000354 case AMDGPU::RAT_WRITE_CACHELESS_128_eg:
Duncan P. N. Exon Smithe4f5e4f2016-06-30 22:52:52 +0000355 BuildMI(*BB, I, BB->findDebugLoc(I), TII->get(MI.getOpcode()))
Diana Picus116bbab2017-01-13 09:58:52 +0000356 .add(MI.getOperand(0))
357 .add(MI.getOperand(1))
Duncan P. N. Exon Smithe4f5e4f2016-06-30 22:52:52 +0000358 .addImm(isEOP(I)); // Set End of program bit
Tom Stellard75aadc22012-12-11 21:25:42 +0000359 break;
Eugene Zelenko2bc2f332016-12-09 22:06:55 +0000360
361 case AMDGPU::RAT_STORE_TYPED_eg:
Duncan P. N. Exon Smithe4f5e4f2016-06-30 22:52:52 +0000362 BuildMI(*BB, I, BB->findDebugLoc(I), TII->get(MI.getOpcode()))
Diana Picus116bbab2017-01-13 09:58:52 +0000363 .add(MI.getOperand(0))
364 .add(MI.getOperand(1))
365 .add(MI.getOperand(2))
Duncan P. N. Exon Smithe4f5e4f2016-06-30 22:52:52 +0000366 .addImm(isEOP(I)); // Set End of program bit
Tom Stellarde0e582c2015-10-01 17:51:34 +0000367 break;
Eugene Zelenko2bc2f332016-12-09 22:06:55 +0000368
Tom Stellard75aadc22012-12-11 21:25:42 +0000369 case AMDGPU::BRANCH:
Duncan P. N. Exon Smithe4f5e4f2016-06-30 22:52:52 +0000370 BuildMI(*BB, I, BB->findDebugLoc(I), TII->get(AMDGPU::JUMP))
Diana Picus116bbab2017-01-13 09:58:52 +0000371 .add(MI.getOperand(0));
Duncan P. N. Exon Smithe4f5e4f2016-06-30 22:52:52 +0000372 break;
Tom Stellard75aadc22012-12-11 21:25:42 +0000373
374 case AMDGPU::BRANCH_COND_f32: {
375 MachineInstr *NewMI =
Duncan P. N. Exon Smithe4f5e4f2016-06-30 22:52:52 +0000376 BuildMI(*BB, I, BB->findDebugLoc(I), TII->get(AMDGPU::PRED_X),
377 AMDGPU::PREDICATE_BIT)
Diana Picus116bbab2017-01-13 09:58:52 +0000378 .add(MI.getOperand(1))
Matt Arsenault44f6d692016-08-13 01:43:46 +0000379 .addImm(AMDGPU::PRED_SETNE)
Duncan P. N. Exon Smithe4f5e4f2016-06-30 22:52:52 +0000380 .addImm(0); // Flags
Duncan P. N. Exon Smith9cfc75c2016-06-30 00:01:54 +0000381 TII->addFlag(*NewMI, 0, MO_FLAG_PUSH);
Vincent Lejeunee5ecf102013-03-11 18:15:06 +0000382 BuildMI(*BB, I, BB->findDebugLoc(I), TII->get(AMDGPU::JUMP_COND))
Diana Picus116bbab2017-01-13 09:58:52 +0000383 .add(MI.getOperand(0))
Duncan P. N. Exon Smithe4f5e4f2016-06-30 22:52:52 +0000384 .addReg(AMDGPU::PREDICATE_BIT, RegState::Kill);
Tom Stellard75aadc22012-12-11 21:25:42 +0000385 break;
386 }
387
388 case AMDGPU::BRANCH_COND_i32: {
389 MachineInstr *NewMI =
Duncan P. N. Exon Smithe4f5e4f2016-06-30 22:52:52 +0000390 BuildMI(*BB, I, BB->findDebugLoc(I), TII->get(AMDGPU::PRED_X),
391 AMDGPU::PREDICATE_BIT)
Diana Picus116bbab2017-01-13 09:58:52 +0000392 .add(MI.getOperand(1))
Matt Arsenault44f6d692016-08-13 01:43:46 +0000393 .addImm(AMDGPU::PRED_SETNE_INT)
Tom Stellard75aadc22012-12-11 21:25:42 +0000394 .addImm(0); // Flags
Duncan P. N. Exon Smith9cfc75c2016-06-30 00:01:54 +0000395 TII->addFlag(*NewMI, 0, MO_FLAG_PUSH);
Vincent Lejeunee5ecf102013-03-11 18:15:06 +0000396 BuildMI(*BB, I, BB->findDebugLoc(I), TII->get(AMDGPU::JUMP_COND))
Diana Picus116bbab2017-01-13 09:58:52 +0000397 .add(MI.getOperand(0))
Duncan P. N. Exon Smithe4f5e4f2016-06-30 22:52:52 +0000398 .addReg(AMDGPU::PREDICATE_BIT, RegState::Kill);
Tom Stellard75aadc22012-12-11 21:25:42 +0000399 break;
400 }
401
Tom Stellard75aadc22012-12-11 21:25:42 +0000402 case AMDGPU::EG_ExportSwz:
403 case AMDGPU::R600_ExportSwz: {
Tom Stellard6f1b8652013-01-23 21:39:49 +0000404 // Instruction is left unmodified if its not the last one of its type
405 bool isLastInstructionOfItsType = true;
Duncan P. N. Exon Smithe4f5e4f2016-06-30 22:52:52 +0000406 unsigned InstExportType = MI.getOperand(1).getImm();
Benjamin Kramerb6d0bd42014-03-02 12:27:27 +0000407 for (MachineBasicBlock::iterator NextExportInst = std::next(I),
Tom Stellard6f1b8652013-01-23 21:39:49 +0000408 EndBlock = BB->end(); NextExportInst != EndBlock;
Benjamin Kramerb6d0bd42014-03-02 12:27:27 +0000409 NextExportInst = std::next(NextExportInst)) {
Tom Stellard6f1b8652013-01-23 21:39:49 +0000410 if (NextExportInst->getOpcode() == AMDGPU::EG_ExportSwz ||
411 NextExportInst->getOpcode() == AMDGPU::R600_ExportSwz) {
412 unsigned CurrentInstExportType = NextExportInst->getOperand(1)
413 .getImm();
414 if (CurrentInstExportType == InstExportType) {
415 isLastInstructionOfItsType = false;
416 break;
417 }
418 }
419 }
Tom Stellardc0f0fba2015-10-01 17:51:29 +0000420 bool EOP = isEOP(I);
Tom Stellard6f1b8652013-01-23 21:39:49 +0000421 if (!EOP && !isLastInstructionOfItsType)
Tom Stellard75aadc22012-12-11 21:25:42 +0000422 return BB;
Duncan P. N. Exon Smithe4f5e4f2016-06-30 22:52:52 +0000423 unsigned CfInst = (MI.getOpcode() == AMDGPU::EG_ExportSwz) ? 84 : 40;
424 BuildMI(*BB, I, BB->findDebugLoc(I), TII->get(MI.getOpcode()))
Diana Picus116bbab2017-01-13 09:58:52 +0000425 .add(MI.getOperand(0))
426 .add(MI.getOperand(1))
427 .add(MI.getOperand(2))
428 .add(MI.getOperand(3))
429 .add(MI.getOperand(4))
430 .add(MI.getOperand(5))
431 .add(MI.getOperand(6))
Duncan P. N. Exon Smithe4f5e4f2016-06-30 22:52:52 +0000432 .addImm(CfInst)
433 .addImm(EOP);
Tom Stellard75aadc22012-12-11 21:25:42 +0000434 break;
435 }
Jakob Stoklund Olesenfdc37672013-02-05 17:53:52 +0000436 case AMDGPU::RETURN: {
Jakob Stoklund Olesenfdc37672013-02-05 17:53:52 +0000437 return BB;
438 }
Tom Stellard75aadc22012-12-11 21:25:42 +0000439 }
440
Duncan P. N. Exon Smithe4f5e4f2016-06-30 22:52:52 +0000441 MI.eraseFromParent();
Tom Stellard75aadc22012-12-11 21:25:42 +0000442 return BB;
443}
444
445//===----------------------------------------------------------------------===//
446// Custom DAG Lowering Operations
447//===----------------------------------------------------------------------===//
448
Tom Stellard75aadc22012-12-11 21:25:42 +0000449SDValue R600TargetLowering::LowerOperation(SDValue Op, SelectionDAG &DAG) const {
Tom Stellardc026e8b2013-06-28 15:47:08 +0000450 MachineFunction &MF = DAG.getMachineFunction();
451 R600MachineFunctionInfo *MFI = MF.getInfo<R600MachineFunctionInfo>();
Tom Stellard75aadc22012-12-11 21:25:42 +0000452 switch (Op.getOpcode()) {
453 default: return AMDGPUTargetLowering::LowerOperation(Op, DAG);
Tom Stellard880a80a2014-06-17 16:53:14 +0000454 case ISD::EXTRACT_VECTOR_ELT: return LowerEXTRACT_VECTOR_ELT(Op, DAG);
455 case ISD::INSERT_VECTOR_ELT: return LowerINSERT_VECTOR_ELT(Op, DAG);
Jan Vesely25f36272014-06-18 12:27:13 +0000456 case ISD::SHL_PARTS: return LowerSHLParts(Op, DAG);
Jan Veselyecf51332014-06-18 12:27:17 +0000457 case ISD::SRA_PARTS:
Jan Vesely900ff2e2014-06-18 12:27:15 +0000458 case ISD::SRL_PARTS: return LowerSRXParts(Op, DAG);
Jan Vesely808fff52015-04-30 17:15:56 +0000459 case ISD::UADDO: return LowerUADDSUBO(Op, DAG, ISD::ADD, AMDGPUISD::CARRY);
460 case ISD::USUBO: return LowerUADDSUBO(Op, DAG, ISD::SUB, AMDGPUISD::BORROW);
Vincent Lejeuneb55940c2013-07-09 15:03:11 +0000461 case ISD::FCOS:
462 case ISD::FSIN: return LowerTrig(Op, DAG);
Tom Stellard75aadc22012-12-11 21:25:42 +0000463 case ISD::SELECT_CC: return LowerSELECT_CC(Op, DAG);
Tom Stellard75aadc22012-12-11 21:25:42 +0000464 case ISD::STORE: return LowerSTORE(Op, DAG);
Matt Arsenaultd2c9e082014-07-07 18:34:45 +0000465 case ISD::LOAD: {
466 SDValue Result = LowerLOAD(Op, DAG);
467 assert((!Result.getNode() ||
468 Result.getNode()->getNumValues() == 2) &&
469 "Load should return a value and a chain");
470 return Result;
471 }
472
Matt Arsenault1d555c42014-06-23 18:00:55 +0000473 case ISD::BRCOND: return LowerBRCOND(Op, DAG);
Tom Stellardc026e8b2013-06-28 15:47:08 +0000474 case ISD::GlobalAddress: return LowerGlobalAddress(MFI, Op, DAG);
Matt Arsenault81d06012016-03-07 21:10:13 +0000475 case ISD::FrameIndex: return lowerFrameIndex(Op, DAG);
Tom Stellard75aadc22012-12-11 21:25:42 +0000476 case ISD::INTRINSIC_VOID: {
477 SDValue Chain = Op.getOperand(0);
478 unsigned IntrinsicID =
479 cast<ConstantSDNode>(Op.getOperand(1))->getZExtValue();
480 switch (IntrinsicID) {
Matt Arsenault82e5e1e2016-07-15 21:27:08 +0000481 case AMDGPUIntrinsic::r600_store_swizzle: {
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +0000482 SDLoc DL(Op);
Vincent Lejeuned80bc152013-02-14 16:55:06 +0000483 const SDValue Args[8] = {
484 Chain,
485 Op.getOperand(2), // Export Value
486 Op.getOperand(3), // ArrayBase
487 Op.getOperand(4), // Type
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +0000488 DAG.getConstant(0, DL, MVT::i32), // SWZ_X
489 DAG.getConstant(1, DL, MVT::i32), // SWZ_Y
490 DAG.getConstant(2, DL, MVT::i32), // SWZ_Z
491 DAG.getConstant(3, DL, MVT::i32) // SWZ_W
Vincent Lejeuned80bc152013-02-14 16:55:06 +0000492 };
Matt Arsenault7bee6ac2016-12-05 20:23:10 +0000493 return DAG.getNode(AMDGPUISD::R600_EXPORT, DL, Op.getValueType(), Args);
Tom Stellard75aadc22012-12-11 21:25:42 +0000494 }
Tom Stellard75aadc22012-12-11 21:25:42 +0000495
Tom Stellard75aadc22012-12-11 21:25:42 +0000496 // default for switch(IntrinsicID)
497 default: break;
498 }
499 // break out of case ISD::INTRINSIC_VOID in switch(Op.getOpcode())
500 break;
501 }
502 case ISD::INTRINSIC_WO_CHAIN: {
503 unsigned IntrinsicID =
504 cast<ConstantSDNode>(Op.getOperand(0))->getZExtValue();
505 EVT VT = Op.getValueType();
Andrew Trickef9de2a2013-05-25 02:42:55 +0000506 SDLoc DL(Op);
Matt Arsenault754dd3e2017-04-03 18:08:08 +0000507 switch (IntrinsicID) {
Matt Arsenault59bd3012016-01-22 19:00:09 +0000508 case AMDGPUIntrinsic::r600_tex:
Matt Arsenaultf9245b72016-07-22 17:01:25 +0000509 case AMDGPUIntrinsic::r600_texc: {
Vincent Lejeuned3eed662013-05-17 16:50:20 +0000510 unsigned TextureOp;
511 switch (IntrinsicID) {
Matt Arsenault59bd3012016-01-22 19:00:09 +0000512 case AMDGPUIntrinsic::r600_tex:
Vincent Lejeuned3eed662013-05-17 16:50:20 +0000513 TextureOp = 0;
514 break;
Matt Arsenault59bd3012016-01-22 19:00:09 +0000515 case AMDGPUIntrinsic::r600_texc:
Vincent Lejeuned3eed662013-05-17 16:50:20 +0000516 TextureOp = 1;
517 break;
Vincent Lejeuned3eed662013-05-17 16:50:20 +0000518 default:
Matt Arsenault60a750f2016-07-26 21:03:38 +0000519 llvm_unreachable("unhandled texture operation");
Vincent Lejeuned3eed662013-05-17 16:50:20 +0000520 }
521
522 SDValue TexArgs[19] = {
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +0000523 DAG.getConstant(TextureOp, DL, MVT::i32),
Vincent Lejeuned3eed662013-05-17 16:50:20 +0000524 Op.getOperand(1),
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +0000525 DAG.getConstant(0, DL, MVT::i32),
526 DAG.getConstant(1, DL, MVT::i32),
527 DAG.getConstant(2, DL, MVT::i32),
528 DAG.getConstant(3, DL, MVT::i32),
Vincent Lejeuned3eed662013-05-17 16:50:20 +0000529 Op.getOperand(2),
530 Op.getOperand(3),
531 Op.getOperand(4),
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +0000532 DAG.getConstant(0, DL, MVT::i32),
533 DAG.getConstant(1, DL, MVT::i32),
534 DAG.getConstant(2, DL, MVT::i32),
535 DAG.getConstant(3, DL, MVT::i32),
Vincent Lejeuned3eed662013-05-17 16:50:20 +0000536 Op.getOperand(5),
537 Op.getOperand(6),
538 Op.getOperand(7),
539 Op.getOperand(8),
540 Op.getOperand(9),
541 Op.getOperand(10)
542 };
Craig Topper48d114b2014-04-26 18:35:24 +0000543 return DAG.getNode(AMDGPUISD::TEXTURE_FETCH, DL, MVT::v4f32, TexArgs);
Vincent Lejeuned3eed662013-05-17 16:50:20 +0000544 }
Matt Arsenaultca7f5702016-07-14 05:47:17 +0000545 case AMDGPUIntrinsic::r600_dot4: {
Vincent Lejeune519f21e2013-05-17 16:50:32 +0000546 SDValue Args[8] = {
547 DAG.getNode(ISD::EXTRACT_VECTOR_ELT, DL, MVT::f32, Op.getOperand(1),
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +0000548 DAG.getConstant(0, DL, MVT::i32)),
Vincent Lejeune519f21e2013-05-17 16:50:32 +0000549 DAG.getNode(ISD::EXTRACT_VECTOR_ELT, DL, MVT::f32, Op.getOperand(2),
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +0000550 DAG.getConstant(0, DL, MVT::i32)),
Vincent Lejeune519f21e2013-05-17 16:50:32 +0000551 DAG.getNode(ISD::EXTRACT_VECTOR_ELT, DL, MVT::f32, Op.getOperand(1),
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +0000552 DAG.getConstant(1, DL, MVT::i32)),
Vincent Lejeune519f21e2013-05-17 16:50:32 +0000553 DAG.getNode(ISD::EXTRACT_VECTOR_ELT, DL, MVT::f32, Op.getOperand(2),
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +0000554 DAG.getConstant(1, DL, MVT::i32)),
Vincent Lejeune519f21e2013-05-17 16:50:32 +0000555 DAG.getNode(ISD::EXTRACT_VECTOR_ELT, DL, MVT::f32, Op.getOperand(1),
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +0000556 DAG.getConstant(2, DL, MVT::i32)),
Vincent Lejeune519f21e2013-05-17 16:50:32 +0000557 DAG.getNode(ISD::EXTRACT_VECTOR_ELT, DL, MVT::f32, Op.getOperand(2),
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +0000558 DAG.getConstant(2, DL, MVT::i32)),
Vincent Lejeune519f21e2013-05-17 16:50:32 +0000559 DAG.getNode(ISD::EXTRACT_VECTOR_ELT, DL, MVT::f32, Op.getOperand(1),
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +0000560 DAG.getConstant(3, DL, MVT::i32)),
Vincent Lejeune519f21e2013-05-17 16:50:32 +0000561 DAG.getNode(ISD::EXTRACT_VECTOR_ELT, DL, MVT::f32, Op.getOperand(2),
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +0000562 DAG.getConstant(3, DL, MVT::i32))
Vincent Lejeune519f21e2013-05-17 16:50:32 +0000563 };
Craig Topper48d114b2014-04-26 18:35:24 +0000564 return DAG.getNode(AMDGPUISD::DOT4, DL, MVT::f32, Args);
Vincent Lejeune519f21e2013-05-17 16:50:32 +0000565 }
Tom Stellard75aadc22012-12-11 21:25:42 +0000566
Jan Vesely2fa28c32016-07-10 21:20:29 +0000567 case Intrinsic::r600_implicitarg_ptr: {
Yaxun Liu1a14bfa2017-03-27 14:04:01 +0000568 MVT PtrVT = getPointerTy(DAG.getDataLayout(), AMDGPUASI.PARAM_I_ADDRESS);
Jan Vesely2fa28c32016-07-10 21:20:29 +0000569 uint32_t ByteOffset = getImplicitParameterOffset(MFI, FIRST_IMPLICIT);
570 return DAG.getConstant(ByteOffset, DL, PtrVT);
571 }
NAKAMURA Takumi4f328e12013-05-22 06:37:31 +0000572 case Intrinsic::r600_read_ngroups_x:
Tom Stellard75aadc22012-12-11 21:25:42 +0000573 return LowerImplicitParameter(DAG, VT, DL, 0);
NAKAMURA Takumi4f328e12013-05-22 06:37:31 +0000574 case Intrinsic::r600_read_ngroups_y:
Tom Stellard75aadc22012-12-11 21:25:42 +0000575 return LowerImplicitParameter(DAG, VT, DL, 1);
NAKAMURA Takumi4f328e12013-05-22 06:37:31 +0000576 case Intrinsic::r600_read_ngroups_z:
Tom Stellard75aadc22012-12-11 21:25:42 +0000577 return LowerImplicitParameter(DAG, VT, DL, 2);
NAKAMURA Takumi4f328e12013-05-22 06:37:31 +0000578 case Intrinsic::r600_read_global_size_x:
Tom Stellard75aadc22012-12-11 21:25:42 +0000579 return LowerImplicitParameter(DAG, VT, DL, 3);
NAKAMURA Takumi4f328e12013-05-22 06:37:31 +0000580 case Intrinsic::r600_read_global_size_y:
Tom Stellard75aadc22012-12-11 21:25:42 +0000581 return LowerImplicitParameter(DAG, VT, DL, 4);
NAKAMURA Takumi4f328e12013-05-22 06:37:31 +0000582 case Intrinsic::r600_read_global_size_z:
Tom Stellard75aadc22012-12-11 21:25:42 +0000583 return LowerImplicitParameter(DAG, VT, DL, 5);
NAKAMURA Takumi4f328e12013-05-22 06:37:31 +0000584 case Intrinsic::r600_read_local_size_x:
Tom Stellard75aadc22012-12-11 21:25:42 +0000585 return LowerImplicitParameter(DAG, VT, DL, 6);
NAKAMURA Takumi4f328e12013-05-22 06:37:31 +0000586 case Intrinsic::r600_read_local_size_y:
Tom Stellard75aadc22012-12-11 21:25:42 +0000587 return LowerImplicitParameter(DAG, VT, DL, 7);
NAKAMURA Takumi4f328e12013-05-22 06:37:31 +0000588 case Intrinsic::r600_read_local_size_z:
Tom Stellard75aadc22012-12-11 21:25:42 +0000589 return LowerImplicitParameter(DAG, VT, DL, 8);
590
NAKAMURA Takumi4f328e12013-05-22 06:37:31 +0000591 case Intrinsic::r600_read_tgid_x:
Matt Arsenaulte0e68a72017-06-19 21:52:45 +0000592 return CreateLiveInRegisterRaw(DAG, &AMDGPU::R600_TReg32RegClass,
593 AMDGPU::T1_X, VT);
NAKAMURA Takumi4f328e12013-05-22 06:37:31 +0000594 case Intrinsic::r600_read_tgid_y:
Matt Arsenaulte0e68a72017-06-19 21:52:45 +0000595 return CreateLiveInRegisterRaw(DAG, &AMDGPU::R600_TReg32RegClass,
596 AMDGPU::T1_Y, VT);
NAKAMURA Takumi4f328e12013-05-22 06:37:31 +0000597 case Intrinsic::r600_read_tgid_z:
Matt Arsenaulte0e68a72017-06-19 21:52:45 +0000598 return CreateLiveInRegisterRaw(DAG, &AMDGPU::R600_TReg32RegClass,
599 AMDGPU::T1_Z, VT);
NAKAMURA Takumi4f328e12013-05-22 06:37:31 +0000600 case Intrinsic::r600_read_tidig_x:
Matt Arsenaulte0e68a72017-06-19 21:52:45 +0000601 return CreateLiveInRegisterRaw(DAG, &AMDGPU::R600_TReg32RegClass,
602 AMDGPU::T0_X, VT);
NAKAMURA Takumi4f328e12013-05-22 06:37:31 +0000603 case Intrinsic::r600_read_tidig_y:
Matt Arsenaulte0e68a72017-06-19 21:52:45 +0000604 return CreateLiveInRegisterRaw(DAG, &AMDGPU::R600_TReg32RegClass,
605 AMDGPU::T0_Y, VT);
NAKAMURA Takumi4f328e12013-05-22 06:37:31 +0000606 case Intrinsic::r600_read_tidig_z:
Matt Arsenaulte0e68a72017-06-19 21:52:45 +0000607 return CreateLiveInRegisterRaw(DAG, &AMDGPU::R600_TReg32RegClass,
608 AMDGPU::T0_Z, VT);
Matt Arsenaultbef34e22016-01-22 21:30:34 +0000609
Matt Arsenault09b2c4a2016-07-15 21:26:52 +0000610 case Intrinsic::r600_recipsqrt_ieee:
611 return DAG.getNode(AMDGPUISD::RSQ, DL, VT, Op.getOperand(1));
Matt Arsenaultbef34e22016-01-22 21:30:34 +0000612
Matt Arsenault09b2c4a2016-07-15 21:26:52 +0000613 case Intrinsic::r600_recipsqrt_clamped:
614 return DAG.getNode(AMDGPUISD::RSQ_CLAMP, DL, VT, Op.getOperand(1));
Matt Arsenault754dd3e2017-04-03 18:08:08 +0000615 default:
616 return Op;
Tom Stellard75aadc22012-12-11 21:25:42 +0000617 }
Matt Arsenault09b2c4a2016-07-15 21:26:52 +0000618
Tom Stellard75aadc22012-12-11 21:25:42 +0000619 // break out of case ISD::INTRINSIC_WO_CHAIN in switch(Op.getOpcode())
620 break;
621 }
622 } // end switch(Op.getOpcode())
623 return SDValue();
624}
625
626void R600TargetLowering::ReplaceNodeResults(SDNode *N,
627 SmallVectorImpl<SDValue> &Results,
628 SelectionDAG &DAG) const {
629 switch (N->getOpcode()) {
Matt Arsenaultd125d742014-03-27 17:23:24 +0000630 default:
631 AMDGPUTargetLowering::ReplaceNodeResults(N, Results, DAG);
632 return;
Jan Vesely2cb62ce2014-07-10 22:40:21 +0000633 case ISD::FP_TO_UINT:
634 if (N->getValueType(0) == MVT::i1) {
Matt Arsenault7fb961f2016-07-22 17:01:21 +0000635 Results.push_back(lowerFP_TO_UINT(N->getOperand(0), DAG));
Jan Vesely2cb62ce2014-07-10 22:40:21 +0000636 return;
637 }
Justin Bognerb03fd122016-08-17 05:10:15 +0000638 // Since we don't care about out of bounds values we can use FP_TO_SINT for
639 // uints too. The DAGLegalizer code for uint considers some extra cases
640 // which are not necessary here.
641 LLVM_FALLTHROUGH;
Jan Vesely2cb62ce2014-07-10 22:40:21 +0000642 case ISD::FP_TO_SINT: {
Matt Arsenault7fb961f2016-07-22 17:01:21 +0000643 if (N->getValueType(0) == MVT::i1) {
644 Results.push_back(lowerFP_TO_SINT(N->getOperand(0), DAG));
645 return;
646 }
647
Jan Vesely2cb62ce2014-07-10 22:40:21 +0000648 SDValue Result;
649 if (expandFP_TO_SINT(N, Result, DAG))
650 Results.push_back(Result);
Tom Stellard365366f2013-01-23 02:09:06 +0000651 return;
Jan Vesely2cb62ce2014-07-10 22:40:21 +0000652 }
Jan Vesely343cd6f02014-06-22 21:43:01 +0000653 case ISD::SDIVREM: {
654 SDValue Op = SDValue(N, 1);
655 SDValue RES = LowerSDIVREM(Op, DAG);
656 Results.push_back(RES);
657 Results.push_back(RES.getValue(1));
658 break;
659 }
660 case ISD::UDIVREM: {
661 SDValue Op = SDValue(N, 0);
Tom Stellardbf69d762014-11-15 01:07:53 +0000662 LowerUDIVREM64(Op, DAG, Results);
Jan Vesely343cd6f02014-06-22 21:43:01 +0000663 break;
664 }
665 }
Tom Stellard75aadc22012-12-11 21:25:42 +0000666}
667
Tom Stellard880a80a2014-06-17 16:53:14 +0000668SDValue R600TargetLowering::vectorToVerticalVector(SelectionDAG &DAG,
669 SDValue Vector) const {
Tom Stellard880a80a2014-06-17 16:53:14 +0000670 SDLoc DL(Vector);
671 EVT VecVT = Vector.getValueType();
672 EVT EltVT = VecVT.getVectorElementType();
673 SmallVector<SDValue, 8> Args;
674
Eugene Zelenko2bc2f332016-12-09 22:06:55 +0000675 for (unsigned i = 0, e = VecVT.getVectorNumElements(); i != e; ++i) {
Mehdi Amini44ede332015-07-09 02:09:04 +0000676 Args.push_back(DAG.getNode(
677 ISD::EXTRACT_VECTOR_ELT, DL, EltVT, Vector,
678 DAG.getConstant(i, DL, getVectorIdxTy(DAG.getDataLayout()))));
Tom Stellard880a80a2014-06-17 16:53:14 +0000679 }
680
681 return DAG.getNode(AMDGPUISD::BUILD_VERTICAL_VECTOR, DL, VecVT, Args);
682}
683
684SDValue R600TargetLowering::LowerEXTRACT_VECTOR_ELT(SDValue Op,
685 SelectionDAG &DAG) const {
Tom Stellard880a80a2014-06-17 16:53:14 +0000686 SDLoc DL(Op);
687 SDValue Vector = Op.getOperand(0);
688 SDValue Index = Op.getOperand(1);
689
690 if (isa<ConstantSDNode>(Index) ||
691 Vector.getOpcode() == AMDGPUISD::BUILD_VERTICAL_VECTOR)
692 return Op;
693
694 Vector = vectorToVerticalVector(DAG, Vector);
695 return DAG.getNode(ISD::EXTRACT_VECTOR_ELT, DL, Op.getValueType(),
696 Vector, Index);
697}
698
699SDValue R600TargetLowering::LowerINSERT_VECTOR_ELT(SDValue Op,
700 SelectionDAG &DAG) const {
701 SDLoc DL(Op);
702 SDValue Vector = Op.getOperand(0);
703 SDValue Value = Op.getOperand(1);
704 SDValue Index = Op.getOperand(2);
705
706 if (isa<ConstantSDNode>(Index) ||
707 Vector.getOpcode() == AMDGPUISD::BUILD_VERTICAL_VECTOR)
708 return Op;
709
710 Vector = vectorToVerticalVector(DAG, Vector);
711 SDValue Insert = DAG.getNode(ISD::INSERT_VECTOR_ELT, DL, Op.getValueType(),
712 Vector, Value, Index);
713 return vectorToVerticalVector(DAG, Insert);
714}
715
Tom Stellard27233b72016-05-02 18:05:17 +0000716SDValue R600TargetLowering::LowerGlobalAddress(AMDGPUMachineFunction *MFI,
717 SDValue Op,
718 SelectionDAG &DAG) const {
Tom Stellard27233b72016-05-02 18:05:17 +0000719 GlobalAddressSDNode *GSD = cast<GlobalAddressSDNode>(Op);
Yaxun Liu1a14bfa2017-03-27 14:04:01 +0000720 if (GSD->getAddressSpace() != AMDGPUASI.CONSTANT_ADDRESS)
Tom Stellard27233b72016-05-02 18:05:17 +0000721 return AMDGPUTargetLowering::LowerGlobalAddress(MFI, Op, DAG);
722
723 const DataLayout &DL = DAG.getDataLayout();
724 const GlobalValue *GV = GSD->getGlobal();
Yaxun Liu1a14bfa2017-03-27 14:04:01 +0000725 MVT ConstPtrVT = getPointerTy(DL, AMDGPUASI.CONSTANT_ADDRESS);
Tom Stellard27233b72016-05-02 18:05:17 +0000726
Jan Veselyf97de002016-05-13 20:39:29 +0000727 SDValue GA = DAG.getTargetGlobalAddress(GV, SDLoc(GSD), ConstPtrVT);
728 return DAG.getNode(AMDGPUISD::CONST_DATA_PTR, SDLoc(GSD), ConstPtrVT, GA);
Tom Stellard27233b72016-05-02 18:05:17 +0000729}
730
Vincent Lejeuneb55940c2013-07-09 15:03:11 +0000731SDValue R600TargetLowering::LowerTrig(SDValue Op, SelectionDAG &DAG) const {
732 // On hw >= R700, COS/SIN input must be between -1. and 1.
733 // Thus we lower them to TRIG ( FRACT ( x / 2Pi + 0.5) - 0.5)
734 EVT VT = Op.getValueType();
735 SDValue Arg = Op.getOperand(0);
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +0000736 SDLoc DL(Op);
Sanjay Patela2607012015-09-16 16:31:21 +0000737
738 // TODO: Should this propagate fast-math-flags?
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +0000739 SDValue FractPart = DAG.getNode(AMDGPUISD::FRACT, DL, VT,
740 DAG.getNode(ISD::FADD, DL, VT,
741 DAG.getNode(ISD::FMUL, DL, VT, Arg,
742 DAG.getConstantFP(0.15915494309, DL, MVT::f32)),
743 DAG.getConstantFP(0.5, DL, MVT::f32)));
Vincent Lejeuneb55940c2013-07-09 15:03:11 +0000744 unsigned TrigNode;
745 switch (Op.getOpcode()) {
746 case ISD::FCOS:
747 TrigNode = AMDGPUISD::COS_HW;
748 break;
749 case ISD::FSIN:
750 TrigNode = AMDGPUISD::SIN_HW;
751 break;
752 default:
753 llvm_unreachable("Wrong trig opcode");
754 }
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +0000755 SDValue TrigVal = DAG.getNode(TrigNode, DL, VT,
756 DAG.getNode(ISD::FADD, DL, VT, FractPart,
757 DAG.getConstantFP(-0.5, DL, MVT::f32)));
Matt Arsenault43e92fe2016-06-24 06:30:11 +0000758 if (Gen >= R600Subtarget::R700)
Vincent Lejeuneb55940c2013-07-09 15:03:11 +0000759 return TrigVal;
760 // On R600 hw, COS/SIN input must be between -Pi and Pi.
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +0000761 return DAG.getNode(ISD::FMUL, DL, VT, TrigVal,
762 DAG.getConstantFP(3.14159265359, DL, MVT::f32));
Vincent Lejeuneb55940c2013-07-09 15:03:11 +0000763}
764
Jan Vesely25f36272014-06-18 12:27:13 +0000765SDValue R600TargetLowering::LowerSHLParts(SDValue Op, SelectionDAG &DAG) const {
766 SDLoc DL(Op);
767 EVT VT = Op.getValueType();
768
769 SDValue Lo = Op.getOperand(0);
770 SDValue Hi = Op.getOperand(1);
771 SDValue Shift = Op.getOperand(2);
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +0000772 SDValue Zero = DAG.getConstant(0, DL, VT);
773 SDValue One = DAG.getConstant(1, DL, VT);
Jan Vesely25f36272014-06-18 12:27:13 +0000774
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +0000775 SDValue Width = DAG.getConstant(VT.getSizeInBits(), DL, VT);
776 SDValue Width1 = DAG.getConstant(VT.getSizeInBits() - 1, DL, VT);
Jan Vesely25f36272014-06-18 12:27:13 +0000777 SDValue BigShift = DAG.getNode(ISD::SUB, DL, VT, Shift, Width);
778 SDValue CompShift = DAG.getNode(ISD::SUB, DL, VT, Width1, Shift);
779
780 // The dance around Width1 is necessary for 0 special case.
781 // Without it the CompShift might be 32, producing incorrect results in
782 // Overflow. So we do the shift in two steps, the alternative is to
783 // add a conditional to filter the special case.
784
785 SDValue Overflow = DAG.getNode(ISD::SRL, DL, VT, Lo, CompShift);
786 Overflow = DAG.getNode(ISD::SRL, DL, VT, Overflow, One);
787
788 SDValue HiSmall = DAG.getNode(ISD::SHL, DL, VT, Hi, Shift);
789 HiSmall = DAG.getNode(ISD::OR, DL, VT, HiSmall, Overflow);
790 SDValue LoSmall = DAG.getNode(ISD::SHL, DL, VT, Lo, Shift);
791
792 SDValue HiBig = DAG.getNode(ISD::SHL, DL, VT, Lo, BigShift);
793 SDValue LoBig = Zero;
794
795 Hi = DAG.getSelectCC(DL, Shift, Width, HiSmall, HiBig, ISD::SETULT);
796 Lo = DAG.getSelectCC(DL, Shift, Width, LoSmall, LoBig, ISD::SETULT);
797
798 return DAG.getNode(ISD::MERGE_VALUES, DL, DAG.getVTList(VT,VT), Lo, Hi);
799}
800
Jan Vesely900ff2e2014-06-18 12:27:15 +0000801SDValue R600TargetLowering::LowerSRXParts(SDValue Op, SelectionDAG &DAG) const {
802 SDLoc DL(Op);
803 EVT VT = Op.getValueType();
804
805 SDValue Lo = Op.getOperand(0);
806 SDValue Hi = Op.getOperand(1);
807 SDValue Shift = Op.getOperand(2);
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +0000808 SDValue Zero = DAG.getConstant(0, DL, VT);
809 SDValue One = DAG.getConstant(1, DL, VT);
Jan Vesely900ff2e2014-06-18 12:27:15 +0000810
Jan Veselyecf51332014-06-18 12:27:17 +0000811 const bool SRA = Op.getOpcode() == ISD::SRA_PARTS;
812
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +0000813 SDValue Width = DAG.getConstant(VT.getSizeInBits(), DL, VT);
814 SDValue Width1 = DAG.getConstant(VT.getSizeInBits() - 1, DL, VT);
Jan Vesely900ff2e2014-06-18 12:27:15 +0000815 SDValue BigShift = DAG.getNode(ISD::SUB, DL, VT, Shift, Width);
816 SDValue CompShift = DAG.getNode(ISD::SUB, DL, VT, Width1, Shift);
817
818 // The dance around Width1 is necessary for 0 special case.
819 // Without it the CompShift might be 32, producing incorrect results in
820 // Overflow. So we do the shift in two steps, the alternative is to
821 // add a conditional to filter the special case.
822
823 SDValue Overflow = DAG.getNode(ISD::SHL, DL, VT, Hi, CompShift);
824 Overflow = DAG.getNode(ISD::SHL, DL, VT, Overflow, One);
825
Jan Veselyecf51332014-06-18 12:27:17 +0000826 SDValue HiSmall = DAG.getNode(SRA ? ISD::SRA : ISD::SRL, DL, VT, Hi, Shift);
Jan Vesely900ff2e2014-06-18 12:27:15 +0000827 SDValue LoSmall = DAG.getNode(ISD::SRL, DL, VT, Lo, Shift);
828 LoSmall = DAG.getNode(ISD::OR, DL, VT, LoSmall, Overflow);
829
Jan Veselyecf51332014-06-18 12:27:17 +0000830 SDValue LoBig = DAG.getNode(SRA ? ISD::SRA : ISD::SRL, DL, VT, Hi, BigShift);
831 SDValue HiBig = SRA ? DAG.getNode(ISD::SRA, DL, VT, Hi, Width1) : Zero;
Jan Vesely900ff2e2014-06-18 12:27:15 +0000832
833 Hi = DAG.getSelectCC(DL, Shift, Width, HiSmall, HiBig, ISD::SETULT);
834 Lo = DAG.getSelectCC(DL, Shift, Width, LoSmall, LoBig, ISD::SETULT);
835
836 return DAG.getNode(ISD::MERGE_VALUES, DL, DAG.getVTList(VT,VT), Lo, Hi);
837}
838
Jan Vesely808fff52015-04-30 17:15:56 +0000839SDValue R600TargetLowering::LowerUADDSUBO(SDValue Op, SelectionDAG &DAG,
840 unsigned mainop, unsigned ovf) const {
841 SDLoc DL(Op);
842 EVT VT = Op.getValueType();
843
844 SDValue Lo = Op.getOperand(0);
845 SDValue Hi = Op.getOperand(1);
846
847 SDValue OVF = DAG.getNode(ovf, DL, VT, Lo, Hi);
848 // Extend sign.
849 OVF = DAG.getNode(ISD::SIGN_EXTEND_INREG, DL, VT, OVF,
850 DAG.getValueType(MVT::i1));
851
852 SDValue Res = DAG.getNode(mainop, DL, VT, Lo, Hi);
853
854 return DAG.getNode(ISD::MERGE_VALUES, DL, DAG.getVTList(VT, VT), Res, OVF);
855}
856
Matt Arsenault7fb961f2016-07-22 17:01:21 +0000857SDValue R600TargetLowering::lowerFP_TO_UINT(SDValue Op, SelectionDAG &DAG) const {
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +0000858 SDLoc DL(Op);
Tom Stellard75aadc22012-12-11 21:25:42 +0000859 return DAG.getNode(
860 ISD::SETCC,
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +0000861 DL,
Tom Stellard75aadc22012-12-11 21:25:42 +0000862 MVT::i1,
Matt Arsenault7fb961f2016-07-22 17:01:21 +0000863 Op, DAG.getConstantFP(1.0f, DL, MVT::f32),
864 DAG.getCondCode(ISD::SETEQ));
865}
866
867SDValue R600TargetLowering::lowerFP_TO_SINT(SDValue Op, SelectionDAG &DAG) const {
868 SDLoc DL(Op);
869 return DAG.getNode(
870 ISD::SETCC,
871 DL,
872 MVT::i1,
873 Op, DAG.getConstantFP(-1.0f, DL, MVT::f32),
874 DAG.getCondCode(ISD::SETEQ));
Tom Stellard75aadc22012-12-11 21:25:42 +0000875}
876
Tom Stellard75aadc22012-12-11 21:25:42 +0000877SDValue R600TargetLowering::LowerImplicitParameter(SelectionDAG &DAG, EVT VT,
Benjamin Kramerbdc49562016-06-12 15:39:02 +0000878 const SDLoc &DL,
Tom Stellard75aadc22012-12-11 21:25:42 +0000879 unsigned DwordOffset) const {
880 unsigned ByteOffset = DwordOffset * 4;
881 PointerType * PtrType = PointerType::get(VT.getTypeForEVT(*DAG.getContext()),
Yaxun Liu1a14bfa2017-03-27 14:04:01 +0000882 AMDGPUASI.CONSTANT_BUFFER_0);
Tom Stellard75aadc22012-12-11 21:25:42 +0000883
884 // We shouldn't be using an offset wider than 16-bits for implicit parameters.
885 assert(isInt<16>(ByteOffset));
886
887 return DAG.getLoad(VT, DL, DAG.getEntryNode(),
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +0000888 DAG.getConstant(ByteOffset, DL, MVT::i32), // PTR
Justin Lebar9c375812016-07-15 18:27:10 +0000889 MachinePointerInfo(ConstantPointerNull::get(PtrType)));
Tom Stellard75aadc22012-12-11 21:25:42 +0000890}
891
Tom Stellard75aadc22012-12-11 21:25:42 +0000892bool R600TargetLowering::isZero(SDValue Op) const {
893 if(ConstantSDNode *Cst = dyn_cast<ConstantSDNode>(Op)) {
894 return Cst->isNullValue();
895 } else if(ConstantFPSDNode *CstFP = dyn_cast<ConstantFPSDNode>(Op)){
896 return CstFP->isZero();
897 } else {
898 return false;
899 }
900}
901
Matt Arsenault6b6a2c32016-03-11 08:00:27 +0000902bool R600TargetLowering::isHWTrueValue(SDValue Op) const {
903 if (ConstantFPSDNode * CFP = dyn_cast<ConstantFPSDNode>(Op)) {
904 return CFP->isExactlyValue(1.0);
905 }
906 return isAllOnesConstant(Op);
907}
908
909bool R600TargetLowering::isHWFalseValue(SDValue Op) const {
910 if (ConstantFPSDNode * CFP = dyn_cast<ConstantFPSDNode>(Op)) {
911 return CFP->getValueAPF().isZero();
912 }
913 return isNullConstant(Op);
914}
915
Tom Stellard75aadc22012-12-11 21:25:42 +0000916SDValue R600TargetLowering::LowerSELECT_CC(SDValue Op, SelectionDAG &DAG) const {
Andrew Trickef9de2a2013-05-25 02:42:55 +0000917 SDLoc DL(Op);
Tom Stellard75aadc22012-12-11 21:25:42 +0000918 EVT VT = Op.getValueType();
919
920 SDValue LHS = Op.getOperand(0);
921 SDValue RHS = Op.getOperand(1);
922 SDValue True = Op.getOperand(2);
923 SDValue False = Op.getOperand(3);
924 SDValue CC = Op.getOperand(4);
925 SDValue Temp;
926
Matt Arsenault1e3a4eb2014-12-12 02:30:37 +0000927 if (VT == MVT::f32) {
928 DAGCombinerInfo DCI(DAG, AfterLegalizeVectorOps, true, nullptr);
Matt Arsenaultda7a6562017-02-01 00:42:40 +0000929 SDValue MinMax = combineFMinMaxLegacy(DL, VT, LHS, RHS, True, False, CC, DCI);
Matt Arsenault1e3a4eb2014-12-12 02:30:37 +0000930 if (MinMax)
931 return MinMax;
932 }
933
Tom Stellard75aadc22012-12-11 21:25:42 +0000934 // LHS and RHS are guaranteed to be the same value type
935 EVT CompareVT = LHS.getValueType();
936
937 // Check if we can lower this to a native operation.
938
Tom Stellard2add82d2013-03-08 15:37:09 +0000939 // Try to lower to a SET* instruction:
940 //
941 // SET* can match the following patterns:
942 //
Tom Stellardcd428182013-09-28 02:50:38 +0000943 // select_cc f32, f32, -1, 0, cc_supported
944 // select_cc f32, f32, 1.0f, 0.0f, cc_supported
945 // select_cc i32, i32, -1, 0, cc_supported
Tom Stellard2add82d2013-03-08 15:37:09 +0000946 //
947
948 // Move hardware True/False values to the correct operand.
Tom Stellardcd428182013-09-28 02:50:38 +0000949 ISD::CondCode CCOpcode = cast<CondCodeSDNode>(CC)->get();
950 ISD::CondCode InverseCC =
951 ISD::getSetCCInverse(CCOpcode, CompareVT == MVT::i32);
Tom Stellard5694d302013-09-28 02:50:43 +0000952 if (isHWTrueValue(False) && isHWFalseValue(True)) {
953 if (isCondCodeLegal(InverseCC, CompareVT.getSimpleVT())) {
954 std::swap(False, True);
955 CC = DAG.getCondCode(InverseCC);
956 } else {
957 ISD::CondCode SwapInvCC = ISD::getSetCCSwappedOperands(InverseCC);
958 if (isCondCodeLegal(SwapInvCC, CompareVT.getSimpleVT())) {
959 std::swap(False, True);
960 std::swap(LHS, RHS);
961 CC = DAG.getCondCode(SwapInvCC);
962 }
963 }
Tom Stellard2add82d2013-03-08 15:37:09 +0000964 }
965
966 if (isHWTrueValue(True) && isHWFalseValue(False) &&
967 (CompareVT == VT || VT == MVT::i32)) {
968 // This can be matched by a SET* instruction.
969 return DAG.getNode(ISD::SELECT_CC, DL, VT, LHS, RHS, True, False, CC);
970 }
971
Tom Stellard75aadc22012-12-11 21:25:42 +0000972 // Try to lower to a CND* instruction:
Tom Stellard2add82d2013-03-08 15:37:09 +0000973 //
974 // CND* can match the following patterns:
975 //
Tom Stellardcd428182013-09-28 02:50:38 +0000976 // select_cc f32, 0.0, f32, f32, cc_supported
977 // select_cc f32, 0.0, i32, i32, cc_supported
978 // select_cc i32, 0, f32, f32, cc_supported
979 // select_cc i32, 0, i32, i32, cc_supported
Tom Stellard2add82d2013-03-08 15:37:09 +0000980 //
Tom Stellardcd428182013-09-28 02:50:38 +0000981
982 // Try to move the zero value to the RHS
983 if (isZero(LHS)) {
984 ISD::CondCode CCOpcode = cast<CondCodeSDNode>(CC)->get();
985 // Try swapping the operands
986 ISD::CondCode CCSwapped = ISD::getSetCCSwappedOperands(CCOpcode);
987 if (isCondCodeLegal(CCSwapped, CompareVT.getSimpleVT())) {
988 std::swap(LHS, RHS);
989 CC = DAG.getCondCode(CCSwapped);
990 } else {
991 // Try inverting the conditon and then swapping the operands
992 ISD::CondCode CCInv = ISD::getSetCCInverse(CCOpcode, CompareVT.isInteger());
993 CCSwapped = ISD::getSetCCSwappedOperands(CCInv);
994 if (isCondCodeLegal(CCSwapped, CompareVT.getSimpleVT())) {
995 std::swap(True, False);
996 std::swap(LHS, RHS);
997 CC = DAG.getCondCode(CCSwapped);
998 }
999 }
1000 }
1001 if (isZero(RHS)) {
1002 SDValue Cond = LHS;
1003 SDValue Zero = RHS;
Tom Stellard75aadc22012-12-11 21:25:42 +00001004 ISD::CondCode CCOpcode = cast<CondCodeSDNode>(CC)->get();
1005 if (CompareVT != VT) {
1006 // Bitcast True / False to the correct types. This will end up being
1007 // a nop, but it allows us to define only a single pattern in the
1008 // .TD files for each CND* instruction rather than having to have
1009 // one pattern for integer True/False and one for fp True/False
1010 True = DAG.getNode(ISD::BITCAST, DL, CompareVT, True);
1011 False = DAG.getNode(ISD::BITCAST, DL, CompareVT, False);
1012 }
Tom Stellard75aadc22012-12-11 21:25:42 +00001013
1014 switch (CCOpcode) {
1015 case ISD::SETONE:
1016 case ISD::SETUNE:
1017 case ISD::SETNE:
Tom Stellard75aadc22012-12-11 21:25:42 +00001018 CCOpcode = ISD::getSetCCInverse(CCOpcode, CompareVT == MVT::i32);
1019 Temp = True;
1020 True = False;
1021 False = Temp;
1022 break;
1023 default:
1024 break;
1025 }
1026 SDValue SelectNode = DAG.getNode(ISD::SELECT_CC, DL, CompareVT,
1027 Cond, Zero,
1028 True, False,
1029 DAG.getCondCode(CCOpcode));
1030 return DAG.getNode(ISD::BITCAST, DL, VT, SelectNode);
1031 }
1032
Tom Stellard75aadc22012-12-11 21:25:42 +00001033 // If we make it this for it means we have no native instructions to handle
1034 // this SELECT_CC, so we must lower it.
1035 SDValue HWTrue, HWFalse;
1036
1037 if (CompareVT == MVT::f32) {
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +00001038 HWTrue = DAG.getConstantFP(1.0f, DL, CompareVT);
1039 HWFalse = DAG.getConstantFP(0.0f, DL, CompareVT);
Tom Stellard75aadc22012-12-11 21:25:42 +00001040 } else if (CompareVT == MVT::i32) {
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +00001041 HWTrue = DAG.getConstant(-1, DL, CompareVT);
1042 HWFalse = DAG.getConstant(0, DL, CompareVT);
Tom Stellard75aadc22012-12-11 21:25:42 +00001043 }
1044 else {
Matt Arsenaulteaa3a7e2013-12-10 21:37:42 +00001045 llvm_unreachable("Unhandled value type in LowerSELECT_CC");
Tom Stellard75aadc22012-12-11 21:25:42 +00001046 }
1047
1048 // Lower this unsupported SELECT_CC into a combination of two supported
1049 // SELECT_CC operations.
1050 SDValue Cond = DAG.getNode(ISD::SELECT_CC, DL, CompareVT, LHS, RHS, HWTrue, HWFalse, CC);
1051
1052 return DAG.getNode(ISD::SELECT_CC, DL, VT,
1053 Cond, HWFalse,
1054 True, False,
1055 DAG.getCondCode(ISD::SETNE));
1056}
1057
Alp Tokercb402912014-01-24 17:20:08 +00001058/// LLVM generates byte-addressed pointers. For indirect addressing, we need to
Tom Stellardf3b2a1e2013-02-06 17:32:29 +00001059/// convert these pointers to a register index. Each register holds
1060/// 16 bytes, (4 x 32bit sub-register), but we need to take into account the
1061/// \p StackWidth, which tells us how many of the 4 sub-registrers will be used
1062/// for indirect addressing.
1063SDValue R600TargetLowering::stackPtrToRegIndex(SDValue Ptr,
1064 unsigned StackWidth,
1065 SelectionDAG &DAG) const {
1066 unsigned SRLPad;
1067 switch(StackWidth) {
1068 case 1:
1069 SRLPad = 2;
1070 break;
1071 case 2:
1072 SRLPad = 3;
1073 break;
1074 case 4:
1075 SRLPad = 4;
1076 break;
1077 default: llvm_unreachable("Invalid stack width");
1078 }
1079
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +00001080 SDLoc DL(Ptr);
1081 return DAG.getNode(ISD::SRL, DL, Ptr.getValueType(), Ptr,
1082 DAG.getConstant(SRLPad, DL, MVT::i32));
Tom Stellardf3b2a1e2013-02-06 17:32:29 +00001083}
1084
1085void R600TargetLowering::getStackAddress(unsigned StackWidth,
1086 unsigned ElemIdx,
1087 unsigned &Channel,
1088 unsigned &PtrIncr) const {
1089 switch (StackWidth) {
1090 default:
1091 case 1:
1092 Channel = 0;
1093 if (ElemIdx > 0) {
1094 PtrIncr = 1;
1095 } else {
1096 PtrIncr = 0;
1097 }
1098 break;
1099 case 2:
1100 Channel = ElemIdx % 2;
1101 if (ElemIdx == 2) {
1102 PtrIncr = 1;
1103 } else {
1104 PtrIncr = 0;
1105 }
1106 break;
1107 case 4:
1108 Channel = ElemIdx;
1109 PtrIncr = 0;
1110 break;
1111 }
1112}
1113
Matt Arsenault95245662016-02-11 05:32:46 +00001114SDValue R600TargetLowering::lowerPrivateTruncStore(StoreSDNode *Store,
1115 SelectionDAG &DAG) const {
1116 SDLoc DL(Store);
Jan Vesely06200bd2017-01-06 21:00:46 +00001117 //TODO: Who creates the i8 stores?
1118 assert(Store->isTruncatingStore()
1119 || Store->getValue().getValueType() == MVT::i8);
Yaxun Liu1a14bfa2017-03-27 14:04:01 +00001120 assert(Store->getAddressSpace() == AMDGPUASI.PRIVATE_ADDRESS);
Tom Stellard75aadc22012-12-11 21:25:42 +00001121
Jan Vesely06200bd2017-01-06 21:00:46 +00001122 SDValue Mask;
Matt Arsenault95245662016-02-11 05:32:46 +00001123 if (Store->getMemoryVT() == MVT::i8) {
Jan Vesely06200bd2017-01-06 21:00:46 +00001124 assert(Store->getAlignment() >= 1);
1125 Mask = DAG.getConstant(0xff, DL, MVT::i32);
Matt Arsenault95245662016-02-11 05:32:46 +00001126 } else if (Store->getMemoryVT() == MVT::i16) {
Jan Vesely06200bd2017-01-06 21:00:46 +00001127 assert(Store->getAlignment() >= 2);
Mandeep Singh Grang5e1697e2017-06-06 05:08:36 +00001128 Mask = DAG.getConstant(0xffff, DL, MVT::i32);
Jan Vesely06200bd2017-01-06 21:00:46 +00001129 } else {
1130 llvm_unreachable("Unsupported private trunc store");
Matt Arsenault95245662016-02-11 05:32:46 +00001131 }
1132
Jan Veselyf1705042017-01-20 21:24:26 +00001133 SDValue OldChain = Store->getChain();
1134 bool VectorTrunc = (OldChain.getOpcode() == AMDGPUISD::DUMMY_CHAIN);
1135 // Skip dummy
1136 SDValue Chain = VectorTrunc ? OldChain->getOperand(0) : OldChain;
Matt Arsenault95245662016-02-11 05:32:46 +00001137 SDValue BasePtr = Store->getBasePtr();
Jan Vesely06200bd2017-01-06 21:00:46 +00001138 SDValue Offset = Store->getOffset();
Matt Arsenault95245662016-02-11 05:32:46 +00001139 EVT MemVT = Store->getMemoryVT();
1140
Jan Vesely06200bd2017-01-06 21:00:46 +00001141 SDValue LoadPtr = BasePtr;
1142 if (!Offset.isUndef()) {
1143 LoadPtr = DAG.getNode(ISD::ADD, DL, MVT::i32, BasePtr, Offset);
1144 }
Matt Arsenault95245662016-02-11 05:32:46 +00001145
Jan Vesely06200bd2017-01-06 21:00:46 +00001146 // Get dword location
1147 // TODO: this should be eliminated by the future SHR ptr, 2
1148 SDValue Ptr = DAG.getNode(ISD::AND, DL, MVT::i32, LoadPtr,
1149 DAG.getConstant(0xfffffffc, DL, MVT::i32));
1150
1151 // Load dword
1152 // TODO: can we be smarter about machine pointer info?
Yaxun Liu35845f02017-11-10 02:03:28 +00001153 MachinePointerInfo PtrInfo(UndefValue::get(
1154 Type::getInt32PtrTy(*DAG.getContext(), AMDGPUASI.PRIVATE_ADDRESS)));
1155 SDValue Dst = DAG.getLoad(MVT::i32, DL, Chain, Ptr, PtrInfo);
Jan Vesely06200bd2017-01-06 21:00:46 +00001156
1157 Chain = Dst.getValue(1);
1158
1159 // Get offset in dword
1160 SDValue ByteIdx = DAG.getNode(ISD::AND, DL, MVT::i32, LoadPtr,
Matt Arsenault95245662016-02-11 05:32:46 +00001161 DAG.getConstant(0x3, DL, MVT::i32));
1162
Jan Vesely06200bd2017-01-06 21:00:46 +00001163 // Convert byte offset to bit shift
Matt Arsenault95245662016-02-11 05:32:46 +00001164 SDValue ShiftAmt = DAG.getNode(ISD::SHL, DL, MVT::i32, ByteIdx,
1165 DAG.getConstant(3, DL, MVT::i32));
1166
Jan Vesely06200bd2017-01-06 21:00:46 +00001167 // TODO: Contrary to the name of the functiom,
1168 // it also handles sub i32 non-truncating stores (like i1)
Matt Arsenault95245662016-02-11 05:32:46 +00001169 SDValue SExtValue = DAG.getNode(ISD::SIGN_EXTEND, DL, MVT::i32,
1170 Store->getValue());
1171
Jan Vesely06200bd2017-01-06 21:00:46 +00001172 // Mask the value to the right type
Matt Arsenault95245662016-02-11 05:32:46 +00001173 SDValue MaskedValue = DAG.getZeroExtendInReg(SExtValue, DL, MemVT);
1174
Jan Vesely06200bd2017-01-06 21:00:46 +00001175 // Shift the value in place
Matt Arsenault95245662016-02-11 05:32:46 +00001176 SDValue ShiftedValue = DAG.getNode(ISD::SHL, DL, MVT::i32,
1177 MaskedValue, ShiftAmt);
1178
Jan Vesely06200bd2017-01-06 21:00:46 +00001179 // Shift the mask in place
1180 SDValue DstMask = DAG.getNode(ISD::SHL, DL, MVT::i32, Mask, ShiftAmt);
1181
1182 // Invert the mask. NOTE: if we had native ROL instructions we could
1183 // use inverted mask
1184 DstMask = DAG.getNOT(DL, DstMask, MVT::i32);
1185
1186 // Cleanup the target bits
Matt Arsenault95245662016-02-11 05:32:46 +00001187 Dst = DAG.getNode(ISD::AND, DL, MVT::i32, Dst, DstMask);
1188
Jan Vesely06200bd2017-01-06 21:00:46 +00001189 // Add the new bits
Matt Arsenault95245662016-02-11 05:32:46 +00001190 SDValue Value = DAG.getNode(ISD::OR, DL, MVT::i32, Dst, ShiftedValue);
Jan Vesely06200bd2017-01-06 21:00:46 +00001191
1192 // Store dword
1193 // TODO: Can we be smarter about MachinePointerInfo?
Yaxun Liu35845f02017-11-10 02:03:28 +00001194 SDValue NewStore = DAG.getStore(Chain, DL, Value, Ptr, PtrInfo);
Jan Veselyf1705042017-01-20 21:24:26 +00001195
1196 // If we are part of expanded vector, make our neighbors depend on this store
1197 if (VectorTrunc) {
1198 // Make all other vector elements depend on this store
1199 Chain = DAG.getNode(AMDGPUISD::DUMMY_CHAIN, DL, MVT::Other, NewStore);
1200 DAG.ReplaceAllUsesOfValueWith(OldChain, Chain);
1201 }
1202 return NewStore;
Matt Arsenault95245662016-02-11 05:32:46 +00001203}
1204
1205SDValue R600TargetLowering::LowerSTORE(SDValue Op, SelectionDAG &DAG) const {
Matt Arsenault95245662016-02-11 05:32:46 +00001206 StoreSDNode *StoreNode = cast<StoreSDNode>(Op);
1207 unsigned AS = StoreNode->getAddressSpace();
Matt Arsenault95245662016-02-11 05:32:46 +00001208
Jan Vesely06200bd2017-01-06 21:00:46 +00001209 SDValue Chain = StoreNode->getChain();
1210 SDValue Ptr = StoreNode->getBasePtr();
1211 SDValue Value = StoreNode->getValue();
1212
1213 EVT VT = Value.getValueType();
1214 EVT MemVT = StoreNode->getMemoryVT();
1215 EVT PtrVT = Ptr.getValueType();
1216
1217 SDLoc DL(Op);
1218
1219 // Neither LOCAL nor PRIVATE can do vectors at the moment
Yaxun Liu1a14bfa2017-03-27 14:04:01 +00001220 if ((AS == AMDGPUASI.LOCAL_ADDRESS || AS == AMDGPUASI.PRIVATE_ADDRESS) &&
Jan Vesely06200bd2017-01-06 21:00:46 +00001221 VT.isVector()) {
Yaxun Liu1a14bfa2017-03-27 14:04:01 +00001222 if ((AS == AMDGPUASI.PRIVATE_ADDRESS) &&
1223 StoreNode->isTruncatingStore()) {
Jan Veselyf1705042017-01-20 21:24:26 +00001224 // Add an extra level of chain to isolate this vector
1225 SDValue NewChain = DAG.getNode(AMDGPUISD::DUMMY_CHAIN, DL, MVT::Other, Chain);
1226 // TODO: can the chain be replaced without creating a new store?
1227 SDValue NewStore = DAG.getTruncStore(
1228 NewChain, DL, Value, Ptr, StoreNode->getPointerInfo(),
1229 MemVT, StoreNode->getAlignment(),
1230 StoreNode->getMemOperand()->getFlags(), StoreNode->getAAInfo());
1231 StoreNode = cast<StoreSDNode>(NewStore);
1232 }
1233
Jan Vesely06200bd2017-01-06 21:00:46 +00001234 return scalarizeVectorStore(StoreNode, DAG);
Matt Arsenault95245662016-02-11 05:32:46 +00001235 }
1236
Jan Vesely06200bd2017-01-06 21:00:46 +00001237 unsigned Align = StoreNode->getAlignment();
1238 if (Align < MemVT.getStoreSize() &&
Eugene Zelenko2bc2f332016-12-09 22:06:55 +00001239 !allowsMisalignedMemoryAccesses(MemVT, AS, Align, nullptr)) {
Jan Vesely00864882016-09-02 19:07:06 +00001240 return expandUnalignedStore(StoreNode, DAG);
1241 }
1242
Jan Vesely06200bd2017-01-06 21:00:46 +00001243 SDValue DWordAddr = DAG.getNode(ISD::SRL, DL, PtrVT, Ptr,
1244 DAG.getConstant(2, DL, PtrVT));
Matt Arsenault95245662016-02-11 05:32:46 +00001245
Yaxun Liu1a14bfa2017-03-27 14:04:01 +00001246 if (AS == AMDGPUASI.GLOBAL_ADDRESS) {
Jan Vesely00864882016-09-02 19:07:06 +00001247 // It is beneficial to create MSKOR here instead of combiner to avoid
1248 // artificial dependencies introduced by RMW
Tom Stellardd3ee8c12013-08-16 01:12:06 +00001249 if (StoreNode->isTruncatingStore()) {
Tom Stellardfbab8272013-08-16 01:12:11 +00001250 assert(VT.bitsLE(MVT::i32));
Tom Stellardd3ee8c12013-08-16 01:12:06 +00001251 SDValue MaskConstant;
1252 if (MemVT == MVT::i8) {
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +00001253 MaskConstant = DAG.getConstant(0xFF, DL, MVT::i32);
Tom Stellardd3ee8c12013-08-16 01:12:06 +00001254 } else {
1255 assert(MemVT == MVT::i16);
Jan Vesely00864882016-09-02 19:07:06 +00001256 assert(StoreNode->getAlignment() >= 2);
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +00001257 MaskConstant = DAG.getConstant(0xFFFF, DL, MVT::i32);
Tom Stellardd3ee8c12013-08-16 01:12:06 +00001258 }
Jan Vesely06200bd2017-01-06 21:00:46 +00001259
1260 SDValue ByteIndex = DAG.getNode(ISD::AND, DL, PtrVT, Ptr,
1261 DAG.getConstant(0x00000003, DL, PtrVT));
1262 SDValue BitShift = DAG.getNode(ISD::SHL, DL, VT, ByteIndex,
1263 DAG.getConstant(3, DL, VT));
1264
1265 // Put the mask in correct place
1266 SDValue Mask = DAG.getNode(ISD::SHL, DL, VT, MaskConstant, BitShift);
1267
Jan Veselyf1705042017-01-20 21:24:26 +00001268 // Put the value bits in correct place
Tom Stellardd3ee8c12013-08-16 01:12:06 +00001269 SDValue TruncValue = DAG.getNode(ISD::AND, DL, VT, Value, MaskConstant);
Jan Vesely06200bd2017-01-06 21:00:46 +00001270 SDValue ShiftedValue = DAG.getNode(ISD::SHL, DL, VT, TruncValue, BitShift);
1271
Tom Stellardd3ee8c12013-08-16 01:12:06 +00001272 // XXX: If we add a 64-bit ZW register class, then we could use a 2 x i32
1273 // vector instead.
1274 SDValue Src[4] = {
1275 ShiftedValue,
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +00001276 DAG.getConstant(0, DL, MVT::i32),
1277 DAG.getConstant(0, DL, MVT::i32),
Tom Stellardd3ee8c12013-08-16 01:12:06 +00001278 Mask
1279 };
Ahmed Bougacha128f8732016-04-26 21:15:30 +00001280 SDValue Input = DAG.getBuildVector(MVT::v4i32, DL, Src);
Tom Stellardd3ee8c12013-08-16 01:12:06 +00001281 SDValue Args[3] = { Chain, Input, DWordAddr };
1282 return DAG.getMemIntrinsicNode(AMDGPUISD::STORE_MSKOR, DL,
Craig Topper206fcd42014-04-26 19:29:41 +00001283 Op->getVTList(), Args, MemVT,
Tom Stellardd3ee8c12013-08-16 01:12:06 +00001284 StoreNode->getMemOperand());
Jan Vesely06200bd2017-01-06 21:00:46 +00001285 } else if (Ptr->getOpcode() != AMDGPUISD::DWORDADDR && VT.bitsGE(MVT::i32)) {
Tom Stellardd3ee8c12013-08-16 01:12:06 +00001286 // Convert pointer from byte address to dword address.
Jan Vesely06200bd2017-01-06 21:00:46 +00001287 Ptr = DAG.getNode(AMDGPUISD::DWORDADDR, DL, PtrVT, DWordAddr);
Tom Stellard75aadc22012-12-11 21:25:42 +00001288
Tom Stellardd3ee8c12013-08-16 01:12:06 +00001289 if (StoreNode->isTruncatingStore() || StoreNode->isIndexed()) {
Matt Arsenaulteaa3a7e2013-12-10 21:37:42 +00001290 llvm_unreachable("Truncated and indexed stores not supported yet");
Tom Stellardd3ee8c12013-08-16 01:12:06 +00001291 } else {
1292 Chain = DAG.getStore(Chain, DL, Value, Ptr, StoreNode->getMemOperand());
1293 }
1294 return Chain;
Tom Stellard75aadc22012-12-11 21:25:42 +00001295 }
Tom Stellard75aadc22012-12-11 21:25:42 +00001296 }
Tom Stellardf3b2a1e2013-02-06 17:32:29 +00001297
Jan Vesely06200bd2017-01-06 21:00:46 +00001298 // GLOBAL_ADDRESS has been handled above, LOCAL_ADDRESS allows all sizes
Yaxun Liu1a14bfa2017-03-27 14:04:01 +00001299 if (AS != AMDGPUASI.PRIVATE_ADDRESS)
Tom Stellardf3b2a1e2013-02-06 17:32:29 +00001300 return SDValue();
Tom Stellardf3b2a1e2013-02-06 17:32:29 +00001301
Matt Arsenault95245662016-02-11 05:32:46 +00001302 if (MemVT.bitsLT(MVT::i32))
1303 return lowerPrivateTruncStore(StoreNode, DAG);
Tom Stellardf3b2a1e2013-02-06 17:32:29 +00001304
Jan Vesely06200bd2017-01-06 21:00:46 +00001305 // Standard i32+ store, tag it with DWORDADDR to note that the address
1306 // has been shifted
1307 if (Ptr.getOpcode() != AMDGPUISD::DWORDADDR) {
1308 Ptr = DAG.getNode(AMDGPUISD::DWORDADDR, DL, PtrVT, DWordAddr);
1309 return DAG.getStore(Chain, DL, Value, Ptr, StoreNode->getMemOperand());
Tom Stellardf3b2a1e2013-02-06 17:32:29 +00001310 }
1311
Jan Vesely06200bd2017-01-06 21:00:46 +00001312 // Tagged i32+ stores will be matched by patterns
1313 return SDValue();
Tom Stellard75aadc22012-12-11 21:25:42 +00001314}
1315
Tom Stellard365366f2013-01-23 02:09:06 +00001316// return (512 + (kc_bank << 12)
1317static int
Konstantin Zhuravlyov435151a2017-11-01 19:12:38 +00001318ConstantAddressBlock(unsigned AddressSpace) {
Tom Stellard365366f2013-01-23 02:09:06 +00001319 switch (AddressSpace) {
Konstantin Zhuravlyov435151a2017-11-01 19:12:38 +00001320 case AMDGPUAS::CONSTANT_BUFFER_0:
Tom Stellard365366f2013-01-23 02:09:06 +00001321 return 512;
Konstantin Zhuravlyov435151a2017-11-01 19:12:38 +00001322 case AMDGPUAS::CONSTANT_BUFFER_1:
Tom Stellard365366f2013-01-23 02:09:06 +00001323 return 512 + 4096;
Konstantin Zhuravlyov435151a2017-11-01 19:12:38 +00001324 case AMDGPUAS::CONSTANT_BUFFER_2:
Tom Stellard365366f2013-01-23 02:09:06 +00001325 return 512 + 4096 * 2;
Konstantin Zhuravlyov435151a2017-11-01 19:12:38 +00001326 case AMDGPUAS::CONSTANT_BUFFER_3:
Tom Stellard365366f2013-01-23 02:09:06 +00001327 return 512 + 4096 * 3;
Konstantin Zhuravlyov435151a2017-11-01 19:12:38 +00001328 case AMDGPUAS::CONSTANT_BUFFER_4:
Tom Stellard365366f2013-01-23 02:09:06 +00001329 return 512 + 4096 * 4;
Konstantin Zhuravlyov435151a2017-11-01 19:12:38 +00001330 case AMDGPUAS::CONSTANT_BUFFER_5:
Tom Stellard365366f2013-01-23 02:09:06 +00001331 return 512 + 4096 * 5;
Konstantin Zhuravlyov435151a2017-11-01 19:12:38 +00001332 case AMDGPUAS::CONSTANT_BUFFER_6:
Tom Stellard365366f2013-01-23 02:09:06 +00001333 return 512 + 4096 * 6;
Konstantin Zhuravlyov435151a2017-11-01 19:12:38 +00001334 case AMDGPUAS::CONSTANT_BUFFER_7:
Tom Stellard365366f2013-01-23 02:09:06 +00001335 return 512 + 4096 * 7;
Konstantin Zhuravlyov435151a2017-11-01 19:12:38 +00001336 case AMDGPUAS::CONSTANT_BUFFER_8:
Tom Stellard365366f2013-01-23 02:09:06 +00001337 return 512 + 4096 * 8;
Konstantin Zhuravlyov435151a2017-11-01 19:12:38 +00001338 case AMDGPUAS::CONSTANT_BUFFER_9:
Tom Stellard365366f2013-01-23 02:09:06 +00001339 return 512 + 4096 * 9;
Konstantin Zhuravlyov435151a2017-11-01 19:12:38 +00001340 case AMDGPUAS::CONSTANT_BUFFER_10:
Tom Stellard365366f2013-01-23 02:09:06 +00001341 return 512 + 4096 * 10;
Konstantin Zhuravlyov435151a2017-11-01 19:12:38 +00001342 case AMDGPUAS::CONSTANT_BUFFER_11:
Tom Stellard365366f2013-01-23 02:09:06 +00001343 return 512 + 4096 * 11;
Konstantin Zhuravlyov435151a2017-11-01 19:12:38 +00001344 case AMDGPUAS::CONSTANT_BUFFER_12:
Tom Stellard365366f2013-01-23 02:09:06 +00001345 return 512 + 4096 * 12;
Konstantin Zhuravlyov435151a2017-11-01 19:12:38 +00001346 case AMDGPUAS::CONSTANT_BUFFER_13:
Tom Stellard365366f2013-01-23 02:09:06 +00001347 return 512 + 4096 * 13;
Konstantin Zhuravlyov435151a2017-11-01 19:12:38 +00001348 case AMDGPUAS::CONSTANT_BUFFER_14:
Tom Stellard365366f2013-01-23 02:09:06 +00001349 return 512 + 4096 * 14;
Konstantin Zhuravlyov435151a2017-11-01 19:12:38 +00001350 case AMDGPUAS::CONSTANT_BUFFER_15:
Tom Stellard365366f2013-01-23 02:09:06 +00001351 return 512 + 4096 * 15;
1352 default:
1353 return -1;
1354 }
1355}
1356
Matt Arsenault6dfda962016-02-10 18:21:39 +00001357SDValue R600TargetLowering::lowerPrivateExtLoad(SDValue Op,
1358 SelectionDAG &DAG) const {
Andrew Trickef9de2a2013-05-25 02:42:55 +00001359 SDLoc DL(Op);
Matt Arsenault6dfda962016-02-10 18:21:39 +00001360 LoadSDNode *Load = cast<LoadSDNode>(Op);
1361 ISD::LoadExtType ExtType = Load->getExtensionType();
1362 EVT MemVT = Load->getMemoryVT();
Jan Vesely06200bd2017-01-06 21:00:46 +00001363 assert(Load->getAlignment() >= MemVT.getStoreSize());
Tom Stellard365366f2013-01-23 02:09:06 +00001364
Jan Vesely06200bd2017-01-06 21:00:46 +00001365 SDValue BasePtr = Load->getBasePtr();
1366 SDValue Chain = Load->getChain();
1367 SDValue Offset = Load->getOffset();
Matt Arsenault6dfda962016-02-10 18:21:39 +00001368
Jan Vesely06200bd2017-01-06 21:00:46 +00001369 SDValue LoadPtr = BasePtr;
1370 if (!Offset.isUndef()) {
1371 LoadPtr = DAG.getNode(ISD::ADD, DL, MVT::i32, BasePtr, Offset);
1372 }
1373
1374 // Get dword location
1375 // NOTE: this should be eliminated by the future SHR ptr, 2
1376 SDValue Ptr = DAG.getNode(ISD::AND, DL, MVT::i32, LoadPtr,
1377 DAG.getConstant(0xfffffffc, DL, MVT::i32));
1378
1379 // Load dword
1380 // TODO: can we be smarter about machine pointer info?
Yaxun Liu35845f02017-11-10 02:03:28 +00001381 MachinePointerInfo PtrInfo(UndefValue::get(
1382 Type::getInt32PtrTy(*DAG.getContext(), AMDGPUASI.PRIVATE_ADDRESS)));
1383 SDValue Read = DAG.getLoad(MVT::i32, DL, Chain, Ptr, PtrInfo);
Matt Arsenault6dfda962016-02-10 18:21:39 +00001384
1385 // Get offset within the register.
1386 SDValue ByteIdx = DAG.getNode(ISD::AND, DL, MVT::i32,
Jan Vesely06200bd2017-01-06 21:00:46 +00001387 LoadPtr, DAG.getConstant(0x3, DL, MVT::i32));
Matt Arsenault6dfda962016-02-10 18:21:39 +00001388
1389 // Bit offset of target byte (byteIdx * 8).
1390 SDValue ShiftAmt = DAG.getNode(ISD::SHL, DL, MVT::i32, ByteIdx,
1391 DAG.getConstant(3, DL, MVT::i32));
1392
1393 // Shift to the right.
Jan Vesely06200bd2017-01-06 21:00:46 +00001394 SDValue Ret = DAG.getNode(ISD::SRL, DL, MVT::i32, Read, ShiftAmt);
Matt Arsenault6dfda962016-02-10 18:21:39 +00001395
1396 // Eliminate the upper bits by setting them to ...
1397 EVT MemEltVT = MemVT.getScalarType();
1398
Jan Vesely06200bd2017-01-06 21:00:46 +00001399 if (ExtType == ISD::SEXTLOAD) { // ... ones.
Matt Arsenault6dfda962016-02-10 18:21:39 +00001400 SDValue MemEltVTNode = DAG.getValueType(MemEltVT);
Jan Vesely06200bd2017-01-06 21:00:46 +00001401 Ret = DAG.getNode(ISD::SIGN_EXTEND_INREG, DL, MVT::i32, Ret, MemEltVTNode);
1402 } else { // ... or zeros.
1403 Ret = DAG.getZeroExtendInReg(Ret, DL, MemEltVT);
Matt Arsenault6dfda962016-02-10 18:21:39 +00001404 }
1405
Matt Arsenault6dfda962016-02-10 18:21:39 +00001406 SDValue Ops[] = {
Jan Vesely06200bd2017-01-06 21:00:46 +00001407 Ret,
1408 Read.getValue(1) // This should be our output chain
Matt Arsenault6dfda962016-02-10 18:21:39 +00001409 };
1410
1411 return DAG.getMergeValues(Ops, DL);
1412}
1413
1414SDValue R600TargetLowering::LowerLOAD(SDValue Op, SelectionDAG &DAG) const {
1415 LoadSDNode *LoadNode = cast<LoadSDNode>(Op);
1416 unsigned AS = LoadNode->getAddressSpace();
1417 EVT MemVT = LoadNode->getMemoryVT();
1418 ISD::LoadExtType ExtType = LoadNode->getExtensionType();
1419
Yaxun Liu1a14bfa2017-03-27 14:04:01 +00001420 if (AS == AMDGPUASI.PRIVATE_ADDRESS &&
Matt Arsenault6dfda962016-02-10 18:21:39 +00001421 ExtType != ISD::NON_EXTLOAD && MemVT.bitsLT(MVT::i32)) {
1422 return lowerPrivateExtLoad(Op, DAG);
1423 }
1424
1425 SDLoc DL(Op);
1426 EVT VT = Op.getValueType();
1427 SDValue Chain = LoadNode->getChain();
1428 SDValue Ptr = LoadNode->getBasePtr();
Tom Stellarde9373602014-01-22 19:24:14 +00001429
Yaxun Liu1a14bfa2017-03-27 14:04:01 +00001430 if ((LoadNode->getAddressSpace() == AMDGPUASI.LOCAL_ADDRESS ||
1431 LoadNode->getAddressSpace() == AMDGPUASI.PRIVATE_ADDRESS) &&
Jan Vesely06200bd2017-01-06 21:00:46 +00001432 VT.isVector()) {
1433 return scalarizeVectorLoad(LoadNode, DAG);
Tom Stellard35bb18c2013-08-26 15:06:04 +00001434 }
1435
Konstantin Zhuravlyov435151a2017-11-01 19:12:38 +00001436 int ConstantBlock = ConstantAddressBlock(LoadNode->getAddressSpace());
Matt Arsenault00a0d6f2013-11-13 02:39:07 +00001437 if (ConstantBlock > -1 &&
1438 ((LoadNode->getExtensionType() == ISD::NON_EXTLOAD) ||
1439 (LoadNode->getExtensionType() == ISD::ZEXTLOAD))) {
Tom Stellard365366f2013-01-23 02:09:06 +00001440 SDValue Result;
Nick Lewyckyaad475b2014-04-15 07:22:52 +00001441 if (isa<ConstantExpr>(LoadNode->getMemOperand()->getValue()) ||
1442 isa<Constant>(LoadNode->getMemOperand()->getValue()) ||
Matt Arsenaultef1a9502013-11-01 17:39:26 +00001443 isa<ConstantSDNode>(Ptr)) {
Tom Stellard365366f2013-01-23 02:09:06 +00001444 SDValue Slots[4];
1445 for (unsigned i = 0; i < 4; i++) {
1446 // We want Const position encoded with the following formula :
1447 // (((512 + (kc_bank << 12) + const_index) << 2) + chan)
1448 // const_index is Ptr computed by llvm using an alignment of 16.
1449 // Thus we add (((512 + (kc_bank << 12)) + chan ) * 4 here and
1450 // then div by 4 at the ISel step
1451 SDValue NewPtr = DAG.getNode(ISD::ADD, DL, Ptr.getValueType(), Ptr,
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +00001452 DAG.getConstant(4 * i + ConstantBlock * 16, DL, MVT::i32));
Tom Stellard365366f2013-01-23 02:09:06 +00001453 Slots[i] = DAG.getNode(AMDGPUISD::CONST_ADDRESS, DL, MVT::i32, NewPtr);
1454 }
Tom Stellard0344cdf2013-08-01 15:23:42 +00001455 EVT NewVT = MVT::v4i32;
1456 unsigned NumElements = 4;
1457 if (VT.isVector()) {
1458 NewVT = VT;
1459 NumElements = VT.getVectorNumElements();
1460 }
Ahmed Bougacha128f8732016-04-26 21:15:30 +00001461 Result = DAG.getBuildVector(NewVT, DL, makeArrayRef(Slots, NumElements));
Tom Stellard365366f2013-01-23 02:09:06 +00001462 } else {
Alp Tokerf907b892013-12-05 05:44:44 +00001463 // non-constant ptr can't be folded, keeps it as a v4f32 load
Tom Stellard365366f2013-01-23 02:09:06 +00001464 Result = DAG.getNode(AMDGPUISD::CONST_ADDRESS, DL, MVT::v4i32,
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +00001465 DAG.getNode(ISD::SRL, DL, MVT::i32, Ptr,
1466 DAG.getConstant(4, DL, MVT::i32)),
1467 DAG.getConstant(LoadNode->getAddressSpace() -
Yaxun Liu1a14bfa2017-03-27 14:04:01 +00001468 AMDGPUASI.CONSTANT_BUFFER_0, DL, MVT::i32)
Tom Stellard365366f2013-01-23 02:09:06 +00001469 );
1470 }
1471
1472 if (!VT.isVector()) {
1473 Result = DAG.getNode(ISD::EXTRACT_VECTOR_ELT, DL, MVT::i32, Result,
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +00001474 DAG.getConstant(0, DL, MVT::i32));
Tom Stellard365366f2013-01-23 02:09:06 +00001475 }
1476
1477 SDValue MergedValues[2] = {
Matt Arsenault7939acd2014-04-07 16:44:24 +00001478 Result,
1479 Chain
Tom Stellard365366f2013-01-23 02:09:06 +00001480 };
Craig Topper64941d92014-04-27 19:20:57 +00001481 return DAG.getMergeValues(MergedValues, DL);
Tom Stellard365366f2013-01-23 02:09:06 +00001482 }
1483
Matt Arsenault909d0c02013-10-30 23:43:29 +00001484 // For most operations returning SDValue() will result in the node being
1485 // expanded by the DAG Legalizer. This is not the case for ISD::LOAD, so we
1486 // need to manually expand loads that may be legal in some address spaces and
1487 // illegal in others. SEXT loads from CONSTANT_BUFFER_0 are supported for
1488 // compute shaders, since the data is sign extended when it is uploaded to the
1489 // buffer. However SEXT loads from other address spaces are not supported, so
1490 // we need to expand them here.
Tom Stellard84021442013-07-23 01:48:24 +00001491 if (LoadNode->getExtensionType() == ISD::SEXTLOAD) {
1492 EVT MemVT = LoadNode->getMemoryVT();
1493 assert(!MemVT.isVector() && (MemVT == MVT::i16 || MemVT == MVT::i8));
Justin Lebar9c375812016-07-15 18:27:10 +00001494 SDValue NewLoad = DAG.getExtLoad(
1495 ISD::EXTLOAD, DL, VT, Chain, Ptr, LoadNode->getPointerInfo(), MemVT,
1496 LoadNode->getAlignment(), LoadNode->getMemOperand()->getFlags());
Jan Veselyb670d372015-05-26 18:07:22 +00001497 SDValue Res = DAG.getNode(ISD::SIGN_EXTEND_INREG, DL, VT, NewLoad,
1498 DAG.getValueType(MemVT));
Tom Stellard84021442013-07-23 01:48:24 +00001499
Jan Veselyb670d372015-05-26 18:07:22 +00001500 SDValue MergedValues[2] = { Res, Chain };
Craig Topper64941d92014-04-27 19:20:57 +00001501 return DAG.getMergeValues(MergedValues, DL);
Tom Stellard84021442013-07-23 01:48:24 +00001502 }
1503
Yaxun Liu1a14bfa2017-03-27 14:04:01 +00001504 if (LoadNode->getAddressSpace() != AMDGPUASI.PRIVATE_ADDRESS) {
Tom Stellardf3b2a1e2013-02-06 17:32:29 +00001505 return SDValue();
1506 }
1507
Jan Vesely06200bd2017-01-06 21:00:46 +00001508 // DWORDADDR ISD marks already shifted address
1509 if (Ptr.getOpcode() != AMDGPUISD::DWORDADDR) {
1510 assert(VT == MVT::i32);
1511 Ptr = DAG.getNode(ISD::SRL, DL, MVT::i32, Ptr, DAG.getConstant(2, DL, MVT::i32));
1512 Ptr = DAG.getNode(AMDGPUISD::DWORDADDR, DL, MVT::i32, Ptr);
1513 return DAG.getLoad(MVT::i32, DL, Chain, Ptr, LoadNode->getMemOperand());
Tom Stellardf3b2a1e2013-02-06 17:32:29 +00001514 }
Jan Vesely06200bd2017-01-06 21:00:46 +00001515 return SDValue();
Tom Stellard365366f2013-01-23 02:09:06 +00001516}
Tom Stellard75aadc22012-12-11 21:25:42 +00001517
Matt Arsenault1d555c42014-06-23 18:00:55 +00001518SDValue R600TargetLowering::LowerBRCOND(SDValue Op, SelectionDAG &DAG) const {
1519 SDValue Chain = Op.getOperand(0);
1520 SDValue Cond = Op.getOperand(1);
1521 SDValue Jump = Op.getOperand(2);
1522
1523 return DAG.getNode(AMDGPUISD::BRANCH_COND, SDLoc(Op), Op.getValueType(),
1524 Chain, Jump, Cond);
1525}
1526
Matt Arsenault81d06012016-03-07 21:10:13 +00001527SDValue R600TargetLowering::lowerFrameIndex(SDValue Op,
1528 SelectionDAG &DAG) const {
1529 MachineFunction &MF = DAG.getMachineFunction();
Matt Arsenault43e92fe2016-06-24 06:30:11 +00001530 const R600FrameLowering *TFL = getSubtarget()->getFrameLowering();
Matt Arsenault81d06012016-03-07 21:10:13 +00001531
1532 FrameIndexSDNode *FIN = cast<FrameIndexSDNode>(Op);
1533
1534 unsigned FrameIndex = FIN->getIndex();
1535 unsigned IgnoredFrameReg;
1536 unsigned Offset =
1537 TFL->getFrameIndexReference(MF, FrameIndex, IgnoredFrameReg);
1538 return DAG.getConstant(Offset * 4 * TFL->getStackWidth(MF), SDLoc(Op),
1539 Op.getValueType());
1540}
1541
Tom Stellard75aadc22012-12-11 21:25:42 +00001542/// XXX Only kernel functions are supported, so we can assume for now that
1543/// every function is a kernel function, but in the future we should use
1544/// separate calling conventions for kernel and non-kernel functions.
1545SDValue R600TargetLowering::LowerFormalArguments(
Benjamin Kramerbdc49562016-06-12 15:39:02 +00001546 SDValue Chain, CallingConv::ID CallConv, bool isVarArg,
1547 const SmallVectorImpl<ISD::InputArg> &Ins, const SDLoc &DL,
1548 SelectionDAG &DAG, SmallVectorImpl<SDValue> &InVals) const {
Tom Stellardacfeebf2013-07-23 01:48:05 +00001549 SmallVector<CCValAssign, 16> ArgLocs;
Eric Christopherb5217502014-08-06 18:45:26 +00001550 CCState CCInfo(CallConv, isVarArg, DAG.getMachineFunction(), ArgLocs,
1551 *DAG.getContext());
Vincent Lejeunef143af32013-11-11 22:10:24 +00001552 MachineFunction &MF = DAG.getMachineFunction();
Jan Veselye5121f32014-10-14 20:05:26 +00001553 R600MachineFunctionInfo *MFI = MF.getInfo<R600MachineFunctionInfo>();
Tom Stellardacfeebf2013-07-23 01:48:05 +00001554
Tom Stellardaf775432013-10-23 00:44:32 +00001555 SmallVector<ISD::InputArg, 8> LocalIns;
1556
Tom Stellardbbeb45a2016-09-16 21:53:00 +00001557 if (AMDGPU::isShader(CallConv)) {
Matt Arsenaulte622dc32017-04-11 22:29:24 +00001558 CCInfo.AnalyzeFormalArguments(Ins, CCAssignFnForCall(CallConv, isVarArg));
Tom Stellardbbeb45a2016-09-16 21:53:00 +00001559 } else {
1560 analyzeFormalArgumentsCompute(CCInfo, Ins);
1561 }
Tom Stellardacfeebf2013-07-23 01:48:05 +00001562
Tom Stellard1e803092013-07-23 01:48:18 +00001563 for (unsigned i = 0, e = Ins.size(); i < e; ++i) {
Tom Stellardacfeebf2013-07-23 01:48:05 +00001564 CCValAssign &VA = ArgLocs[i];
Matt Arsenault74ef2772014-08-13 18:14:11 +00001565 const ISD::InputArg &In = Ins[i];
1566 EVT VT = In.VT;
1567 EVT MemVT = VA.getLocVT();
1568 if (!VT.isVector() && MemVT.isVector()) {
1569 // Get load source type if scalarized.
1570 MemVT = MemVT.getVectorElementType();
1571 }
Tom Stellard78e01292013-07-23 01:47:58 +00001572
Nicolai Haehnledf3a20c2016-04-06 19:40:20 +00001573 if (AMDGPU::isShader(CallConv)) {
Vincent Lejeunef143af32013-11-11 22:10:24 +00001574 unsigned Reg = MF.addLiveIn(VA.getLocReg(), &AMDGPU::R600_Reg128RegClass);
1575 SDValue Register = DAG.getCopyFromReg(Chain, DL, Reg, VT);
1576 InVals.push_back(Register);
1577 continue;
1578 }
1579
Tom Stellard75aadc22012-12-11 21:25:42 +00001580 PointerType *PtrTy = PointerType::get(VT.getTypeForEVT(*DAG.getContext()),
Yaxun Liu1a14bfa2017-03-27 14:04:01 +00001581 AMDGPUASI.CONSTANT_BUFFER_0);
Tom Stellardacfeebf2013-07-23 01:48:05 +00001582
Matt Arsenaultfae02982014-03-17 18:58:11 +00001583 // i64 isn't a legal type, so the register type used ends up as i32, which
1584 // isn't expected here. It attempts to create this sextload, but it ends up
1585 // being invalid. Somehow this seems to work with i64 arguments, but breaks
1586 // for <1 x i64>.
1587
Tom Stellardacfeebf2013-07-23 01:48:05 +00001588 // The first 36 bytes of the input buffer contains information about
1589 // thread group and global sizes.
Matt Arsenault74ef2772014-08-13 18:14:11 +00001590 ISD::LoadExtType Ext = ISD::NON_EXTLOAD;
1591 if (MemVT.getScalarSizeInBits() != VT.getScalarSizeInBits()) {
1592 // FIXME: This should really check the extload type, but the handling of
1593 // extload vector parameters seems to be broken.
Matt Arsenaulte1f030c2014-04-11 20:59:54 +00001594
Matt Arsenault74ef2772014-08-13 18:14:11 +00001595 // Ext = In.Flags.isSExt() ? ISD::SEXTLOAD : ISD::ZEXTLOAD;
1596 Ext = ISD::SEXTLOAD;
1597 }
1598
1599 // Compute the offset from the value.
1600 // XXX - I think PartOffset should give you this, but it seems to give the
1601 // size of the register which isn't useful.
1602
Andrew Trick05938a52015-02-16 18:10:47 +00001603 unsigned ValBase = ArgLocs[In.getOrigArgIndex()].getLocMemOffset();
Matt Arsenault74ef2772014-08-13 18:14:11 +00001604 unsigned PartOffset = VA.getLocMemOffset();
Tom Stellard2f3f9852017-01-25 01:25:13 +00001605 unsigned Offset = Subtarget->getExplicitKernelArgOffset(MF) + VA.getLocMemOffset();
Matt Arsenault74ef2772014-08-13 18:14:11 +00001606
1607 MachinePointerInfo PtrInfo(UndefValue::get(PtrTy), PartOffset - ValBase);
Justin Lebar9c375812016-07-15 18:27:10 +00001608 SDValue Arg = DAG.getLoad(
1609 ISD::UNINDEXED, Ext, VT, DL, Chain,
1610 DAG.getConstant(Offset, DL, MVT::i32), DAG.getUNDEF(MVT::i32), PtrInfo,
Justin Lebaradbf09e2016-09-11 01:38:58 +00001611 MemVT, /* Alignment = */ 4, MachineMemOperand::MONonTemporal |
1612 MachineMemOperand::MODereferenceable |
1613 MachineMemOperand::MOInvariant);
Matt Arsenault209a7b92014-04-18 07:40:20 +00001614
1615 // 4 is the preferred alignment for the CONSTANT memory space.
Tom Stellard75aadc22012-12-11 21:25:42 +00001616 InVals.push_back(Arg);
Matt Arsenault52ef4012016-07-26 16:45:58 +00001617 MFI->setABIArgOffset(Offset + MemVT.getStoreSize());
Tom Stellard75aadc22012-12-11 21:25:42 +00001618 }
1619 return Chain;
1620}
1621
Mehdi Amini44ede332015-07-09 02:09:04 +00001622EVT R600TargetLowering::getSetCCResultType(const DataLayout &DL, LLVMContext &,
1623 EVT VT) const {
Matt Arsenault209a7b92014-04-18 07:40:20 +00001624 if (!VT.isVector())
1625 return MVT::i32;
Tom Stellard75aadc22012-12-11 21:25:42 +00001626 return VT.changeVectorElementTypeToInteger();
1627}
1628
Nirav Dave4dcad5d2017-07-10 20:25:54 +00001629bool R600TargetLowering::canMergeStoresTo(unsigned AS, EVT MemVT,
1630 const SelectionDAG &DAG) const {
Nirav Daved20066c2017-05-24 15:59:09 +00001631 // Local and Private addresses do not handle vectors. Limit to i32
1632 if ((AS == AMDGPUASI.LOCAL_ADDRESS || AS == AMDGPUASI.PRIVATE_ADDRESS)) {
1633 return (MemVT.getSizeInBits() <= 32);
1634 }
1635 return true;
1636}
1637
Matt Arsenaultfa67bdb2016-02-22 21:04:16 +00001638bool R600TargetLowering::allowsMisalignedMemoryAccesses(EVT VT,
1639 unsigned AddrSpace,
1640 unsigned Align,
1641 bool *IsFast) const {
1642 if (IsFast)
1643 *IsFast = false;
1644
1645 if (!VT.isSimple() || VT == MVT::Other)
1646 return false;
1647
1648 if (VT.bitsLT(MVT::i32))
1649 return false;
1650
1651 // TODO: This is a rough estimate.
1652 if (IsFast)
1653 *IsFast = true;
1654
1655 return VT.bitsGT(MVT::i32) && Align % 4 == 0;
1656}
1657
Matt Arsenault209a7b92014-04-18 07:40:20 +00001658static SDValue CompactSwizzlableVector(
1659 SelectionDAG &DAG, SDValue VectorEntry,
1660 DenseMap<unsigned, unsigned> &RemapSwizzle) {
Vincent Lejeune276ceb82013-06-04 15:04:53 +00001661 assert(VectorEntry.getOpcode() == ISD::BUILD_VECTOR);
1662 assert(RemapSwizzle.empty());
1663 SDValue NewBldVec[4] = {
Matt Arsenault209a7b92014-04-18 07:40:20 +00001664 VectorEntry.getOperand(0),
1665 VectorEntry.getOperand(1),
1666 VectorEntry.getOperand(2),
1667 VectorEntry.getOperand(3)
Vincent Lejeune276ceb82013-06-04 15:04:53 +00001668 };
1669
1670 for (unsigned i = 0; i < 4; i++) {
Sanjay Patel57195842016-03-14 17:28:46 +00001671 if (NewBldVec[i].isUndef())
Vincent Lejeunefa58a5f2013-10-13 17:56:10 +00001672 // We mask write here to teach later passes that the ith element of this
1673 // vector is undef. Thus we can use it to reduce 128 bits reg usage,
1674 // break false dependencies and additionnaly make assembly easier to read.
1675 RemapSwizzle[i] = 7; // SEL_MASK_WRITE
Vincent Lejeune276ceb82013-06-04 15:04:53 +00001676 if (ConstantFPSDNode *C = dyn_cast<ConstantFPSDNode>(NewBldVec[i])) {
1677 if (C->isZero()) {
1678 RemapSwizzle[i] = 4; // SEL_0
1679 NewBldVec[i] = DAG.getUNDEF(MVT::f32);
1680 } else if (C->isExactlyValue(1.0)) {
1681 RemapSwizzle[i] = 5; // SEL_1
1682 NewBldVec[i] = DAG.getUNDEF(MVT::f32);
1683 }
1684 }
1685
Sanjay Patel57195842016-03-14 17:28:46 +00001686 if (NewBldVec[i].isUndef())
Vincent Lejeune276ceb82013-06-04 15:04:53 +00001687 continue;
1688 for (unsigned j = 0; j < i; j++) {
1689 if (NewBldVec[i] == NewBldVec[j]) {
1690 NewBldVec[i] = DAG.getUNDEF(NewBldVec[i].getValueType());
1691 RemapSwizzle[i] = j;
1692 break;
1693 }
1694 }
1695 }
1696
Ahmed Bougacha128f8732016-04-26 21:15:30 +00001697 return DAG.getBuildVector(VectorEntry.getValueType(), SDLoc(VectorEntry),
1698 NewBldVec);
Vincent Lejeune276ceb82013-06-04 15:04:53 +00001699}
1700
Benjamin Kramer193960c2013-06-11 13:32:25 +00001701static SDValue ReorganizeVector(SelectionDAG &DAG, SDValue VectorEntry,
1702 DenseMap<unsigned, unsigned> &RemapSwizzle) {
Vincent Lejeune276ceb82013-06-04 15:04:53 +00001703 assert(VectorEntry.getOpcode() == ISD::BUILD_VECTOR);
1704 assert(RemapSwizzle.empty());
1705 SDValue NewBldVec[4] = {
1706 VectorEntry.getOperand(0),
1707 VectorEntry.getOperand(1),
1708 VectorEntry.getOperand(2),
1709 VectorEntry.getOperand(3)
1710 };
1711 bool isUnmovable[4] = { false, false, false, false };
Vincent Lejeunecc0ea742013-12-10 14:43:31 +00001712 for (unsigned i = 0; i < 4; i++) {
Vincent Lejeuneb8aac8d2013-07-09 15:03:25 +00001713 RemapSwizzle[i] = i;
Vincent Lejeunecc0ea742013-12-10 14:43:31 +00001714 if (NewBldVec[i].getOpcode() == ISD::EXTRACT_VECTOR_ELT) {
1715 unsigned Idx = dyn_cast<ConstantSDNode>(NewBldVec[i].getOperand(1))
1716 ->getZExtValue();
1717 if (i == Idx)
1718 isUnmovable[Idx] = true;
1719 }
1720 }
Vincent Lejeune276ceb82013-06-04 15:04:53 +00001721
1722 for (unsigned i = 0; i < 4; i++) {
1723 if (NewBldVec[i].getOpcode() == ISD::EXTRACT_VECTOR_ELT) {
1724 unsigned Idx = dyn_cast<ConstantSDNode>(NewBldVec[i].getOperand(1))
1725 ->getZExtValue();
Vincent Lejeune301beb82013-10-13 17:56:04 +00001726 if (isUnmovable[Idx])
1727 continue;
1728 // Swap i and Idx
1729 std::swap(NewBldVec[Idx], NewBldVec[i]);
1730 std::swap(RemapSwizzle[i], RemapSwizzle[Idx]);
1731 break;
Vincent Lejeune276ceb82013-06-04 15:04:53 +00001732 }
1733 }
1734
Ahmed Bougacha128f8732016-04-26 21:15:30 +00001735 return DAG.getBuildVector(VectorEntry.getValueType(), SDLoc(VectorEntry),
1736 NewBldVec);
Vincent Lejeune276ceb82013-06-04 15:04:53 +00001737}
1738
Benjamin Kramerbdc49562016-06-12 15:39:02 +00001739SDValue R600TargetLowering::OptimizeSwizzle(SDValue BuildVector, SDValue Swz[4],
1740 SelectionDAG &DAG,
1741 const SDLoc &DL) const {
Vincent Lejeune276ceb82013-06-04 15:04:53 +00001742 assert(BuildVector.getOpcode() == ISD::BUILD_VECTOR);
1743 // Old -> New swizzle values
1744 DenseMap<unsigned, unsigned> SwizzleRemap;
1745
1746 BuildVector = CompactSwizzlableVector(DAG, BuildVector, SwizzleRemap);
1747 for (unsigned i = 0; i < 4; i++) {
Benjamin Kramer619c4e52015-04-10 11:24:51 +00001748 unsigned Idx = cast<ConstantSDNode>(Swz[i])->getZExtValue();
Vincent Lejeune276ceb82013-06-04 15:04:53 +00001749 if (SwizzleRemap.find(Idx) != SwizzleRemap.end())
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +00001750 Swz[i] = DAG.getConstant(SwizzleRemap[Idx], DL, MVT::i32);
Vincent Lejeune276ceb82013-06-04 15:04:53 +00001751 }
1752
1753 SwizzleRemap.clear();
1754 BuildVector = ReorganizeVector(DAG, BuildVector, SwizzleRemap);
1755 for (unsigned i = 0; i < 4; i++) {
Benjamin Kramer619c4e52015-04-10 11:24:51 +00001756 unsigned Idx = cast<ConstantSDNode>(Swz[i])->getZExtValue();
Vincent Lejeune276ceb82013-06-04 15:04:53 +00001757 if (SwizzleRemap.find(Idx) != SwizzleRemap.end())
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +00001758 Swz[i] = DAG.getConstant(SwizzleRemap[Idx], DL, MVT::i32);
Vincent Lejeune276ceb82013-06-04 15:04:53 +00001759 }
1760
1761 return BuildVector;
1762}
1763
Tom Stellard75aadc22012-12-11 21:25:42 +00001764//===----------------------------------------------------------------------===//
1765// Custom DAG Optimizations
1766//===----------------------------------------------------------------------===//
1767
1768SDValue R600TargetLowering::PerformDAGCombine(SDNode *N,
1769 DAGCombinerInfo &DCI) const {
1770 SelectionDAG &DAG = DCI.DAG;
Jan Vesely89876672016-08-29 23:21:46 +00001771 SDLoc DL(N);
Tom Stellard75aadc22012-12-11 21:25:42 +00001772
1773 switch (N->getOpcode()) {
1774 // (f32 fp_round (f64 uint_to_fp a)) -> (f32 uint_to_fp a)
1775 case ISD::FP_ROUND: {
1776 SDValue Arg = N->getOperand(0);
1777 if (Arg.getOpcode() == ISD::UINT_TO_FP && Arg.getValueType() == MVT::f64) {
Jan Vesely89876672016-08-29 23:21:46 +00001778 return DAG.getNode(ISD::UINT_TO_FP, DL, N->getValueType(0),
Tom Stellard75aadc22012-12-11 21:25:42 +00001779 Arg.getOperand(0));
1780 }
1781 break;
1782 }
Tom Stellarde06163a2013-02-07 14:02:35 +00001783
1784 // (i32 fp_to_sint (fneg (select_cc f32, f32, 1.0, 0.0 cc))) ->
1785 // (i32 select_cc f32, f32, -1, 0 cc)
1786 //
1787 // Mesa's GLSL frontend generates the above pattern a lot and we can lower
1788 // this to one of the SET*_DX10 instructions.
1789 case ISD::FP_TO_SINT: {
1790 SDValue FNeg = N->getOperand(0);
1791 if (FNeg.getOpcode() != ISD::FNEG) {
1792 return SDValue();
1793 }
1794 SDValue SelectCC = FNeg.getOperand(0);
1795 if (SelectCC.getOpcode() != ISD::SELECT_CC ||
1796 SelectCC.getOperand(0).getValueType() != MVT::f32 || // LHS
1797 SelectCC.getOperand(2).getValueType() != MVT::f32 || // True
1798 !isHWTrueValue(SelectCC.getOperand(2)) ||
1799 !isHWFalseValue(SelectCC.getOperand(3))) {
1800 return SDValue();
1801 }
1802
Jan Vesely89876672016-08-29 23:21:46 +00001803 return DAG.getNode(ISD::SELECT_CC, DL, N->getValueType(0),
Tom Stellarde06163a2013-02-07 14:02:35 +00001804 SelectCC.getOperand(0), // LHS
1805 SelectCC.getOperand(1), // RHS
Jan Vesely89876672016-08-29 23:21:46 +00001806 DAG.getConstant(-1, DL, MVT::i32), // True
1807 DAG.getConstant(0, DL, MVT::i32), // False
Tom Stellarde06163a2013-02-07 14:02:35 +00001808 SelectCC.getOperand(4)); // CC
1809
1810 break;
1811 }
Quentin Colombete2e05482013-07-30 00:27:16 +00001812
NAKAMURA Takumi8a046432013-10-28 04:07:38 +00001813 // insert_vector_elt (build_vector elt0, ... , eltN), NewEltIdx, idx
1814 // => build_vector elt0, ... , NewEltIdx, ... , eltN
Quentin Colombete2e05482013-07-30 00:27:16 +00001815 case ISD::INSERT_VECTOR_ELT: {
1816 SDValue InVec = N->getOperand(0);
1817 SDValue InVal = N->getOperand(1);
1818 SDValue EltNo = N->getOperand(2);
Quentin Colombete2e05482013-07-30 00:27:16 +00001819
1820 // If the inserted element is an UNDEF, just use the input vector.
Sanjay Patel57195842016-03-14 17:28:46 +00001821 if (InVal.isUndef())
Quentin Colombete2e05482013-07-30 00:27:16 +00001822 return InVec;
1823
1824 EVT VT = InVec.getValueType();
1825
1826 // If we can't generate a legal BUILD_VECTOR, exit
1827 if (!isOperationLegal(ISD::BUILD_VECTOR, VT))
1828 return SDValue();
1829
1830 // Check that we know which element is being inserted
1831 if (!isa<ConstantSDNode>(EltNo))
1832 return SDValue();
1833 unsigned Elt = cast<ConstantSDNode>(EltNo)->getZExtValue();
1834
1835 // Check that the operand is a BUILD_VECTOR (or UNDEF, which can essentially
1836 // be converted to a BUILD_VECTOR). Fill in the Ops vector with the
1837 // vector elements.
1838 SmallVector<SDValue, 8> Ops;
1839 if (InVec.getOpcode() == ISD::BUILD_VECTOR) {
1840 Ops.append(InVec.getNode()->op_begin(),
1841 InVec.getNode()->op_end());
Sanjay Patel57195842016-03-14 17:28:46 +00001842 } else if (InVec.isUndef()) {
Quentin Colombete2e05482013-07-30 00:27:16 +00001843 unsigned NElts = VT.getVectorNumElements();
1844 Ops.append(NElts, DAG.getUNDEF(InVal.getValueType()));
1845 } else {
1846 return SDValue();
1847 }
1848
1849 // Insert the element
1850 if (Elt < Ops.size()) {
1851 // All the operands of BUILD_VECTOR must have the same type;
1852 // we enforce that here.
1853 EVT OpVT = Ops[0].getValueType();
1854 if (InVal.getValueType() != OpVT)
1855 InVal = OpVT.bitsGT(InVal.getValueType()) ?
Jan Vesely89876672016-08-29 23:21:46 +00001856 DAG.getNode(ISD::ANY_EXTEND, DL, OpVT, InVal) :
1857 DAG.getNode(ISD::TRUNCATE, DL, OpVT, InVal);
Quentin Colombete2e05482013-07-30 00:27:16 +00001858 Ops[Elt] = InVal;
1859 }
1860
1861 // Return the new vector
Jan Vesely89876672016-08-29 23:21:46 +00001862 return DAG.getBuildVector(VT, DL, Ops);
Quentin Colombete2e05482013-07-30 00:27:16 +00001863 }
1864
Tom Stellard365366f2013-01-23 02:09:06 +00001865 // Extract_vec (Build_vector) generated by custom lowering
1866 // also needs to be customly combined
1867 case ISD::EXTRACT_VECTOR_ELT: {
1868 SDValue Arg = N->getOperand(0);
1869 if (Arg.getOpcode() == ISD::BUILD_VECTOR) {
1870 if (ConstantSDNode *Const = dyn_cast<ConstantSDNode>(N->getOperand(1))) {
1871 unsigned Element = Const->getZExtValue();
1872 return Arg->getOperand(Element);
1873 }
1874 }
Tom Stellarddd04c832013-01-31 22:11:53 +00001875 if (Arg.getOpcode() == ISD::BITCAST &&
Jan Veselyea457462016-09-02 20:13:19 +00001876 Arg.getOperand(0).getOpcode() == ISD::BUILD_VECTOR &&
1877 (Arg.getOperand(0).getValueType().getVectorNumElements() ==
1878 Arg.getValueType().getVectorNumElements())) {
Tom Stellarddd04c832013-01-31 22:11:53 +00001879 if (ConstantSDNode *Const = dyn_cast<ConstantSDNode>(N->getOperand(1))) {
1880 unsigned Element = Const->getZExtValue();
Jan Vesely89876672016-08-29 23:21:46 +00001881 return DAG.getNode(ISD::BITCAST, DL, N->getVTList(),
1882 Arg->getOperand(0).getOperand(Element));
Tom Stellarddd04c832013-01-31 22:11:53 +00001883 }
1884 }
Mehdi Aminie029eae2015-07-16 06:23:12 +00001885 break;
Tom Stellard365366f2013-01-23 02:09:06 +00001886 }
Tom Stellarde06163a2013-02-07 14:02:35 +00001887
1888 case ISD::SELECT_CC: {
Tom Stellardafa8b532014-05-09 16:42:16 +00001889 // Try common optimizations
Ahmed Bougachaf8dfb472016-02-09 22:54:12 +00001890 if (SDValue Ret = AMDGPUTargetLowering::PerformDAGCombine(N, DCI))
Tom Stellardafa8b532014-05-09 16:42:16 +00001891 return Ret;
1892
Tom Stellarde06163a2013-02-07 14:02:35 +00001893 // fold selectcc (selectcc x, y, a, b, cc), b, a, b, seteq ->
1894 // selectcc x, y, a, b, inv(cc)
Tom Stellard5e524892013-03-08 15:37:11 +00001895 //
1896 // fold selectcc (selectcc x, y, a, b, cc), b, a, b, setne ->
1897 // selectcc x, y, a, b, cc
Tom Stellarde06163a2013-02-07 14:02:35 +00001898 SDValue LHS = N->getOperand(0);
1899 if (LHS.getOpcode() != ISD::SELECT_CC) {
1900 return SDValue();
1901 }
1902
1903 SDValue RHS = N->getOperand(1);
1904 SDValue True = N->getOperand(2);
1905 SDValue False = N->getOperand(3);
Tom Stellard5e524892013-03-08 15:37:11 +00001906 ISD::CondCode NCC = cast<CondCodeSDNode>(N->getOperand(4))->get();
Tom Stellarde06163a2013-02-07 14:02:35 +00001907
1908 if (LHS.getOperand(2).getNode() != True.getNode() ||
1909 LHS.getOperand(3).getNode() != False.getNode() ||
Tom Stellard5e524892013-03-08 15:37:11 +00001910 RHS.getNode() != False.getNode()) {
Tom Stellarde06163a2013-02-07 14:02:35 +00001911 return SDValue();
1912 }
1913
Tom Stellard5e524892013-03-08 15:37:11 +00001914 switch (NCC) {
1915 default: return SDValue();
1916 case ISD::SETNE: return LHS;
1917 case ISD::SETEQ: {
1918 ISD::CondCode LHSCC = cast<CondCodeSDNode>(LHS.getOperand(4))->get();
1919 LHSCC = ISD::getSetCCInverse(LHSCC,
1920 LHS.getOperand(0).getValueType().isInteger());
Tom Stellardcd428182013-09-28 02:50:38 +00001921 if (DCI.isBeforeLegalizeOps() ||
1922 isCondCodeLegal(LHSCC, LHS.getOperand(0).getSimpleValueType()))
Jan Vesely89876672016-08-29 23:21:46 +00001923 return DAG.getSelectCC(DL,
Tom Stellardcd428182013-09-28 02:50:38 +00001924 LHS.getOperand(0),
1925 LHS.getOperand(1),
1926 LHS.getOperand(2),
1927 LHS.getOperand(3),
1928 LHSCC);
1929 break;
Vincent Lejeuned80bc152013-02-14 16:55:06 +00001930 }
Tom Stellard5e524892013-03-08 15:37:11 +00001931 }
Tom Stellardcd428182013-09-28 02:50:38 +00001932 return SDValue();
Tom Stellard5e524892013-03-08 15:37:11 +00001933 }
Tom Stellardfbab8272013-08-16 01:12:11 +00001934
Matt Arsenault7bee6ac2016-12-05 20:23:10 +00001935 case AMDGPUISD::R600_EXPORT: {
Vincent Lejeuned80bc152013-02-14 16:55:06 +00001936 SDValue Arg = N->getOperand(1);
1937 if (Arg.getOpcode() != ISD::BUILD_VECTOR)
1938 break;
Vincent Lejeune276ceb82013-06-04 15:04:53 +00001939
Vincent Lejeuned80bc152013-02-14 16:55:06 +00001940 SDValue NewArgs[8] = {
1941 N->getOperand(0), // Chain
1942 SDValue(),
1943 N->getOperand(2), // ArrayBase
1944 N->getOperand(3), // Type
1945 N->getOperand(4), // SWZ_X
1946 N->getOperand(5), // SWZ_Y
1947 N->getOperand(6), // SWZ_Z
1948 N->getOperand(7) // SWZ_W
1949 };
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +00001950 NewArgs[1] = OptimizeSwizzle(N->getOperand(1), &NewArgs[4], DAG, DL);
Matt Arsenault7bee6ac2016-12-05 20:23:10 +00001951 return DAG.getNode(AMDGPUISD::R600_EXPORT, DL, N->getVTList(), NewArgs);
Tom Stellarde06163a2013-02-07 14:02:35 +00001952 }
Vincent Lejeune276ceb82013-06-04 15:04:53 +00001953 case AMDGPUISD::TEXTURE_FETCH: {
1954 SDValue Arg = N->getOperand(1);
1955 if (Arg.getOpcode() != ISD::BUILD_VECTOR)
1956 break;
1957
1958 SDValue NewArgs[19] = {
1959 N->getOperand(0),
1960 N->getOperand(1),
1961 N->getOperand(2),
1962 N->getOperand(3),
1963 N->getOperand(4),
1964 N->getOperand(5),
1965 N->getOperand(6),
1966 N->getOperand(7),
1967 N->getOperand(8),
1968 N->getOperand(9),
1969 N->getOperand(10),
1970 N->getOperand(11),
1971 N->getOperand(12),
1972 N->getOperand(13),
1973 N->getOperand(14),
1974 N->getOperand(15),
1975 N->getOperand(16),
1976 N->getOperand(17),
1977 N->getOperand(18),
1978 };
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +00001979 NewArgs[1] = OptimizeSwizzle(N->getOperand(1), &NewArgs[2], DAG, DL);
1980 return DAG.getNode(AMDGPUISD::TEXTURE_FETCH, DL, N->getVTList(), NewArgs);
Vincent Lejeune276ceb82013-06-04 15:04:53 +00001981 }
Jan Vesely89876672016-08-29 23:21:46 +00001982 default: break;
Tom Stellard75aadc22012-12-11 21:25:42 +00001983 }
Matt Arsenault5565f65e2014-05-22 18:09:07 +00001984
1985 return AMDGPUTargetLowering::PerformDAGCombine(N, DCI);
Tom Stellard75aadc22012-12-11 21:25:42 +00001986}
Vincent Lejeuneab3baf82013-09-12 23:44:44 +00001987
Matt Arsenault43e92fe2016-06-24 06:30:11 +00001988bool R600TargetLowering::FoldOperand(SDNode *ParentNode, unsigned SrcIdx,
1989 SDValue &Src, SDValue &Neg, SDValue &Abs,
1990 SDValue &Sel, SDValue &Imm,
1991 SelectionDAG &DAG) const {
1992 const R600InstrInfo *TII = getSubtarget()->getInstrInfo();
Vincent Lejeuneab3baf82013-09-12 23:44:44 +00001993 if (!Src.isMachineOpcode())
1994 return false;
Matt Arsenault43e92fe2016-06-24 06:30:11 +00001995
Vincent Lejeuneab3baf82013-09-12 23:44:44 +00001996 switch (Src.getMachineOpcode()) {
1997 case AMDGPU::FNEG_R600:
1998 if (!Neg.getNode())
1999 return false;
2000 Src = Src.getOperand(0);
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +00002001 Neg = DAG.getTargetConstant(1, SDLoc(ParentNode), MVT::i32);
Vincent Lejeuneab3baf82013-09-12 23:44:44 +00002002 return true;
2003 case AMDGPU::FABS_R600:
2004 if (!Abs.getNode())
2005 return false;
2006 Src = Src.getOperand(0);
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +00002007 Abs = DAG.getTargetConstant(1, SDLoc(ParentNode), MVT::i32);
Vincent Lejeuneab3baf82013-09-12 23:44:44 +00002008 return true;
2009 case AMDGPU::CONST_COPY: {
2010 unsigned Opcode = ParentNode->getMachineOpcode();
2011 bool HasDst = TII->getOperandIdx(Opcode, AMDGPU::OpName::dst) > -1;
2012
2013 if (!Sel.getNode())
2014 return false;
2015
2016 SDValue CstOffset = Src.getOperand(0);
2017 if (ParentNode->getValueType(0).isVector())
2018 return false;
2019
2020 // Gather constants values
2021 int SrcIndices[] = {
2022 TII->getOperandIdx(Opcode, AMDGPU::OpName::src0),
2023 TII->getOperandIdx(Opcode, AMDGPU::OpName::src1),
2024 TII->getOperandIdx(Opcode, AMDGPU::OpName::src2),
2025 TII->getOperandIdx(Opcode, AMDGPU::OpName::src0_X),
2026 TII->getOperandIdx(Opcode, AMDGPU::OpName::src0_Y),
2027 TII->getOperandIdx(Opcode, AMDGPU::OpName::src0_Z),
2028 TII->getOperandIdx(Opcode, AMDGPU::OpName::src0_W),
2029 TII->getOperandIdx(Opcode, AMDGPU::OpName::src1_X),
2030 TII->getOperandIdx(Opcode, AMDGPU::OpName::src1_Y),
2031 TII->getOperandIdx(Opcode, AMDGPU::OpName::src1_Z),
2032 TII->getOperandIdx(Opcode, AMDGPU::OpName::src1_W)
2033 };
2034 std::vector<unsigned> Consts;
Matt Arsenault4d64f962014-05-12 19:23:21 +00002035 for (int OtherSrcIdx : SrcIndices) {
Vincent Lejeuneab3baf82013-09-12 23:44:44 +00002036 int OtherSelIdx = TII->getSelIdx(Opcode, OtherSrcIdx);
2037 if (OtherSrcIdx < 0 || OtherSelIdx < 0)
2038 continue;
2039 if (HasDst) {
2040 OtherSrcIdx--;
2041 OtherSelIdx--;
2042 }
2043 if (RegisterSDNode *Reg =
2044 dyn_cast<RegisterSDNode>(ParentNode->getOperand(OtherSrcIdx))) {
2045 if (Reg->getReg() == AMDGPU::ALU_CONST) {
Matt Arsenaultb3ee3882014-05-12 19:26:38 +00002046 ConstantSDNode *Cst
2047 = cast<ConstantSDNode>(ParentNode->getOperand(OtherSelIdx));
Vincent Lejeuneab3baf82013-09-12 23:44:44 +00002048 Consts.push_back(Cst->getZExtValue());
2049 }
2050 }
2051 }
2052
Matt Arsenault37c12d72014-05-12 20:42:57 +00002053 ConstantSDNode *Cst = cast<ConstantSDNode>(CstOffset);
Vincent Lejeuneab3baf82013-09-12 23:44:44 +00002054 Consts.push_back(Cst->getZExtValue());
2055 if (!TII->fitsConstReadLimitations(Consts)) {
2056 return false;
2057 }
2058
2059 Sel = CstOffset;
2060 Src = DAG.getRegister(AMDGPU::ALU_CONST, MVT::f32);
2061 return true;
2062 }
Jan Vesely16800392016-05-13 20:39:31 +00002063 case AMDGPU::MOV_IMM_GLOBAL_ADDR:
2064 // Check if the Imm slot is used. Taken from below.
2065 if (cast<ConstantSDNode>(Imm)->getZExtValue())
2066 return false;
2067 Imm = Src.getOperand(0);
2068 Src = DAG.getRegister(AMDGPU::ALU_LITERAL_X, MVT::i32);
2069 return true;
Vincent Lejeune9a248e52013-09-12 23:44:53 +00002070 case AMDGPU::MOV_IMM_I32:
2071 case AMDGPU::MOV_IMM_F32: {
2072 unsigned ImmReg = AMDGPU::ALU_LITERAL_X;
2073 uint64_t ImmValue = 0;
2074
Vincent Lejeune9a248e52013-09-12 23:44:53 +00002075 if (Src.getMachineOpcode() == AMDGPU::MOV_IMM_F32) {
2076 ConstantFPSDNode *FPC = dyn_cast<ConstantFPSDNode>(Src.getOperand(0));
2077 float FloatValue = FPC->getValueAPF().convertToFloat();
2078 if (FloatValue == 0.0) {
2079 ImmReg = AMDGPU::ZERO;
2080 } else if (FloatValue == 0.5) {
2081 ImmReg = AMDGPU::HALF;
2082 } else if (FloatValue == 1.0) {
2083 ImmReg = AMDGPU::ONE;
2084 } else {
2085 ImmValue = FPC->getValueAPF().bitcastToAPInt().getZExtValue();
2086 }
2087 } else {
2088 ConstantSDNode *C = dyn_cast<ConstantSDNode>(Src.getOperand(0));
2089 uint64_t Value = C->getZExtValue();
2090 if (Value == 0) {
2091 ImmReg = AMDGPU::ZERO;
2092 } else if (Value == 1) {
2093 ImmReg = AMDGPU::ONE_INT;
2094 } else {
2095 ImmValue = Value;
2096 }
2097 }
2098
2099 // Check that we aren't already using an immediate.
2100 // XXX: It's possible for an instruction to have more than one
2101 // immediate operand, but this is not supported yet.
2102 if (ImmReg == AMDGPU::ALU_LITERAL_X) {
2103 if (!Imm.getNode())
2104 return false;
2105 ConstantSDNode *C = dyn_cast<ConstantSDNode>(Imm);
2106 assert(C);
2107 if (C->getZExtValue())
2108 return false;
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +00002109 Imm = DAG.getTargetConstant(ImmValue, SDLoc(ParentNode), MVT::i32);
Vincent Lejeune9a248e52013-09-12 23:44:53 +00002110 }
2111 Src = DAG.getRegister(ImmReg, MVT::i32);
2112 return true;
2113 }
Vincent Lejeuneab3baf82013-09-12 23:44:44 +00002114 default:
2115 return false;
2116 }
2117}
2118
Vincent Lejeuneab3baf82013-09-12 23:44:44 +00002119/// \brief Fold the instructions after selecting them
2120SDNode *R600TargetLowering::PostISelFolding(MachineSDNode *Node,
2121 SelectionDAG &DAG) const {
Matt Arsenault43e92fe2016-06-24 06:30:11 +00002122 const R600InstrInfo *TII = getSubtarget()->getInstrInfo();
Vincent Lejeuneab3baf82013-09-12 23:44:44 +00002123 if (!Node->isMachineOpcode())
2124 return Node;
Matt Arsenault43e92fe2016-06-24 06:30:11 +00002125
Vincent Lejeuneab3baf82013-09-12 23:44:44 +00002126 unsigned Opcode = Node->getMachineOpcode();
2127 SDValue FakeOp;
2128
Benjamin Kramer6cd780f2015-02-17 15:29:18 +00002129 std::vector<SDValue> Ops(Node->op_begin(), Node->op_end());
Vincent Lejeuneab3baf82013-09-12 23:44:44 +00002130
2131 if (Opcode == AMDGPU::DOT_4) {
2132 int OperandIdx[] = {
2133 TII->getOperandIdx(Opcode, AMDGPU::OpName::src0_X),
2134 TII->getOperandIdx(Opcode, AMDGPU::OpName::src0_Y),
2135 TII->getOperandIdx(Opcode, AMDGPU::OpName::src0_Z),
2136 TII->getOperandIdx(Opcode, AMDGPU::OpName::src0_W),
2137 TII->getOperandIdx(Opcode, AMDGPU::OpName::src1_X),
2138 TII->getOperandIdx(Opcode, AMDGPU::OpName::src1_Y),
2139 TII->getOperandIdx(Opcode, AMDGPU::OpName::src1_Z),
2140 TII->getOperandIdx(Opcode, AMDGPU::OpName::src1_W)
NAKAMURA Takumi4bb85f92013-10-28 04:07:23 +00002141 };
Vincent Lejeuneab3baf82013-09-12 23:44:44 +00002142 int NegIdx[] = {
2143 TII->getOperandIdx(Opcode, AMDGPU::OpName::src0_neg_X),
2144 TII->getOperandIdx(Opcode, AMDGPU::OpName::src0_neg_Y),
2145 TII->getOperandIdx(Opcode, AMDGPU::OpName::src0_neg_Z),
2146 TII->getOperandIdx(Opcode, AMDGPU::OpName::src0_neg_W),
2147 TII->getOperandIdx(Opcode, AMDGPU::OpName::src1_neg_X),
2148 TII->getOperandIdx(Opcode, AMDGPU::OpName::src1_neg_Y),
2149 TII->getOperandIdx(Opcode, AMDGPU::OpName::src1_neg_Z),
2150 TII->getOperandIdx(Opcode, AMDGPU::OpName::src1_neg_W)
2151 };
2152 int AbsIdx[] = {
2153 TII->getOperandIdx(Opcode, AMDGPU::OpName::src0_abs_X),
2154 TII->getOperandIdx(Opcode, AMDGPU::OpName::src0_abs_Y),
2155 TII->getOperandIdx(Opcode, AMDGPU::OpName::src0_abs_Z),
2156 TII->getOperandIdx(Opcode, AMDGPU::OpName::src0_abs_W),
2157 TII->getOperandIdx(Opcode, AMDGPU::OpName::src1_abs_X),
2158 TII->getOperandIdx(Opcode, AMDGPU::OpName::src1_abs_Y),
2159 TII->getOperandIdx(Opcode, AMDGPU::OpName::src1_abs_Z),
2160 TII->getOperandIdx(Opcode, AMDGPU::OpName::src1_abs_W)
2161 };
2162 for (unsigned i = 0; i < 8; i++) {
2163 if (OperandIdx[i] < 0)
2164 return Node;
2165 SDValue &Src = Ops[OperandIdx[i] - 1];
2166 SDValue &Neg = Ops[NegIdx[i] - 1];
2167 SDValue &Abs = Ops[AbsIdx[i] - 1];
2168 bool HasDst = TII->getOperandIdx(Opcode, AMDGPU::OpName::dst) > -1;
2169 int SelIdx = TII->getSelIdx(Opcode, OperandIdx[i]);
2170 if (HasDst)
2171 SelIdx--;
2172 SDValue &Sel = (SelIdx > -1) ? Ops[SelIdx] : FakeOp;
Vincent Lejeune9a248e52013-09-12 23:44:53 +00002173 if (FoldOperand(Node, i, Src, Neg, Abs, Sel, FakeOp, DAG))
2174 return DAG.getMachineNode(Opcode, SDLoc(Node), Node->getVTList(), Ops);
2175 }
2176 } else if (Opcode == AMDGPU::REG_SEQUENCE) {
2177 for (unsigned i = 1, e = Node->getNumOperands(); i < e; i += 2) {
2178 SDValue &Src = Ops[i];
2179 if (FoldOperand(Node, i, Src, FakeOp, FakeOp, FakeOp, FakeOp, DAG))
Vincent Lejeuneab3baf82013-09-12 23:44:44 +00002180 return DAG.getMachineNode(Opcode, SDLoc(Node), Node->getVTList(), Ops);
2181 }
Vincent Lejeune0167a312013-09-12 23:45:00 +00002182 } else if (Opcode == AMDGPU::CLAMP_R600) {
2183 SDValue Src = Node->getOperand(0);
2184 if (!Src.isMachineOpcode() ||
2185 !TII->hasInstrModifiers(Src.getMachineOpcode()))
2186 return Node;
2187 int ClampIdx = TII->getOperandIdx(Src.getMachineOpcode(),
2188 AMDGPU::OpName::clamp);
2189 if (ClampIdx < 0)
2190 return Node;
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +00002191 SDLoc DL(Node);
Benjamin Kramer6cd780f2015-02-17 15:29:18 +00002192 std::vector<SDValue> Ops(Src->op_begin(), Src->op_end());
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +00002193 Ops[ClampIdx - 1] = DAG.getTargetConstant(1, DL, MVT::i32);
2194 return DAG.getMachineNode(Src.getMachineOpcode(), DL,
2195 Node->getVTList(), Ops);
Vincent Lejeuneab3baf82013-09-12 23:44:44 +00002196 } else {
2197 if (!TII->hasInstrModifiers(Opcode))
2198 return Node;
2199 int OperandIdx[] = {
2200 TII->getOperandIdx(Opcode, AMDGPU::OpName::src0),
2201 TII->getOperandIdx(Opcode, AMDGPU::OpName::src1),
2202 TII->getOperandIdx(Opcode, AMDGPU::OpName::src2)
2203 };
2204 int NegIdx[] = {
2205 TII->getOperandIdx(Opcode, AMDGPU::OpName::src0_neg),
2206 TII->getOperandIdx(Opcode, AMDGPU::OpName::src1_neg),
2207 TII->getOperandIdx(Opcode, AMDGPU::OpName::src2_neg)
2208 };
2209 int AbsIdx[] = {
2210 TII->getOperandIdx(Opcode, AMDGPU::OpName::src0_abs),
2211 TII->getOperandIdx(Opcode, AMDGPU::OpName::src1_abs),
2212 -1
2213 };
2214 for (unsigned i = 0; i < 3; i++) {
2215 if (OperandIdx[i] < 0)
2216 return Node;
2217 SDValue &Src = Ops[OperandIdx[i] - 1];
2218 SDValue &Neg = Ops[NegIdx[i] - 1];
2219 SDValue FakeAbs;
2220 SDValue &Abs = (AbsIdx[i] > -1) ? Ops[AbsIdx[i] - 1] : FakeAbs;
2221 bool HasDst = TII->getOperandIdx(Opcode, AMDGPU::OpName::dst) > -1;
2222 int SelIdx = TII->getSelIdx(Opcode, OperandIdx[i]);
Vincent Lejeune9a248e52013-09-12 23:44:53 +00002223 int ImmIdx = TII->getOperandIdx(Opcode, AMDGPU::OpName::literal);
2224 if (HasDst) {
Vincent Lejeuneab3baf82013-09-12 23:44:44 +00002225 SelIdx--;
Vincent Lejeune9a248e52013-09-12 23:44:53 +00002226 ImmIdx--;
2227 }
Vincent Lejeuneab3baf82013-09-12 23:44:44 +00002228 SDValue &Sel = (SelIdx > -1) ? Ops[SelIdx] : FakeOp;
Vincent Lejeune9a248e52013-09-12 23:44:53 +00002229 SDValue &Imm = Ops[ImmIdx];
2230 if (FoldOperand(Node, i, Src, Neg, Abs, Sel, Imm, DAG))
Vincent Lejeuneab3baf82013-09-12 23:44:44 +00002231 return DAG.getMachineNode(Opcode, SDLoc(Node), Node->getVTList(), Ops);
2232 }
2233 }
2234
2235 return Node;
2236}