blob: 69c8433eef290040c5ba3e7e19fddc72cbc58f64 [file] [log] [blame]
Justin Holewinskiae556d32012-05-04 20:18:50 +00001//
2// The LLVM Compiler Infrastructure
3//
4// This file is distributed under the University of Illinois Open Source
5// License. See LICENSE.TXT for details.
6//
7//===----------------------------------------------------------------------===//
8//
9// This file defines the interfaces that NVPTX uses to lower LLVM code into a
10// selection DAG.
11//
12//===----------------------------------------------------------------------===//
13
Justin Holewinskiae556d32012-05-04 20:18:50 +000014#include "NVPTXISelLowering.h"
Chandler Carruthed0881b2012-12-03 16:50:05 +000015#include "NVPTX.h"
Justin Holewinskiae556d32012-05-04 20:18:50 +000016#include "NVPTXTargetMachine.h"
17#include "NVPTXTargetObjectFile.h"
18#include "NVPTXUtilities.h"
Justin Holewinskiae556d32012-05-04 20:18:50 +000019#include "llvm/CodeGen/Analysis.h"
20#include "llvm/CodeGen/MachineFrameInfo.h"
21#include "llvm/CodeGen/MachineFunction.h"
22#include "llvm/CodeGen/MachineInstrBuilder.h"
23#include "llvm/CodeGen/MachineRegisterInfo.h"
Justin Holewinskiae556d32012-05-04 20:18:50 +000024#include "llvm/CodeGen/TargetLoweringObjectFileImpl.h"
Chandler Carruth219b89b2014-03-04 11:01:28 +000025#include "llvm/IR/CallSite.h"
Chandler Carruth9fb823b2013-01-02 11:36:10 +000026#include "llvm/IR/DerivedTypes.h"
27#include "llvm/IR/Function.h"
28#include "llvm/IR/GlobalValue.h"
29#include "llvm/IR/IntrinsicInst.h"
30#include "llvm/IR/Intrinsics.h"
31#include "llvm/IR/Module.h"
Justin Holewinskiae556d32012-05-04 20:18:50 +000032#include "llvm/MC/MCSectionELF.h"
Chandler Carruthed0881b2012-12-03 16:50:05 +000033#include "llvm/Support/CommandLine.h"
34#include "llvm/Support/Debug.h"
35#include "llvm/Support/ErrorHandling.h"
Justin Holewinski9982f062014-06-27 19:36:25 +000036#include "llvm/Support/MathExtras.h"
Chandler Carruthed0881b2012-12-03 16:50:05 +000037#include "llvm/Support/raw_ostream.h"
Justin Holewinskiae556d32012-05-04 20:18:50 +000038#include <sstream>
39
40#undef DEBUG_TYPE
41#define DEBUG_TYPE "nvptx-lower"
42
43using namespace llvm;
44
45static unsigned int uniqueCallSite = 0;
46
Justin Holewinski0497ab12013-03-30 14:29:21 +000047static cl::opt<bool> sched4reg(
48 "nvptx-sched4reg",
49 cl::desc("NVPTX Specific: schedule for register pressue"), cl::init(false));
Justin Holewinskiae556d32012-05-04 20:18:50 +000050
Justin Holewinski428cf0e2014-07-17 18:10:09 +000051static cl::opt<unsigned>
52FMAContractLevelOpt("nvptx-fma-level", cl::ZeroOrMore, cl::Hidden,
53 cl::desc("NVPTX Specific: FMA contraction (0: don't do it"
54 " 1: do it 2: do it aggressively"),
55 cl::init(2));
56
Justin Holewinskibe8dc642013-02-12 14:18:49 +000057static bool IsPTXVectorType(MVT VT) {
58 switch (VT.SimpleTy) {
Justin Holewinski0497ab12013-03-30 14:29:21 +000059 default:
60 return false;
Justin Holewinskif8f70912013-06-28 17:57:59 +000061 case MVT::v2i1:
62 case MVT::v4i1:
Justin Holewinskibe8dc642013-02-12 14:18:49 +000063 case MVT::v2i8:
64 case MVT::v4i8:
65 case MVT::v2i16:
66 case MVT::v4i16:
67 case MVT::v2i32:
68 case MVT::v4i32:
69 case MVT::v2i64:
70 case MVT::v2f32:
71 case MVT::v4f32:
72 case MVT::v2f64:
Justin Holewinski0497ab12013-03-30 14:29:21 +000073 return true;
Justin Holewinskibe8dc642013-02-12 14:18:49 +000074 }
75}
76
Justin Holewinskif8f70912013-06-28 17:57:59 +000077/// ComputePTXValueVTs - For the given Type \p Ty, returns the set of primitive
78/// EVTs that compose it. Unlike ComputeValueVTs, this will break apart vectors
79/// into their primitive components.
80/// NOTE: This is a band-aid for code that expects ComputeValueVTs to return the
81/// same number of types as the Ins/Outs arrays in LowerFormalArguments,
82/// LowerCall, and LowerReturn.
Mehdi Amini56228da2015-07-09 01:57:34 +000083static void ComputePTXValueVTs(const TargetLowering &TLI, const DataLayout &DL,
84 Type *Ty, SmallVectorImpl<EVT> &ValueVTs,
Craig Topper062a2ba2014-04-25 05:30:21 +000085 SmallVectorImpl<uint64_t> *Offsets = nullptr,
Justin Holewinskif8f70912013-06-28 17:57:59 +000086 uint64_t StartingOffset = 0) {
87 SmallVector<EVT, 16> TempVTs;
88 SmallVector<uint64_t, 16> TempOffsets;
89
Mehdi Amini56228da2015-07-09 01:57:34 +000090 ComputeValueVTs(TLI, DL, Ty, TempVTs, &TempOffsets, StartingOffset);
Justin Holewinskif8f70912013-06-28 17:57:59 +000091 for (unsigned i = 0, e = TempVTs.size(); i != e; ++i) {
92 EVT VT = TempVTs[i];
93 uint64_t Off = TempOffsets[i];
94 if (VT.isVector())
95 for (unsigned j = 0, je = VT.getVectorNumElements(); j != je; ++j) {
96 ValueVTs.push_back(VT.getVectorElementType());
97 if (Offsets)
98 Offsets->push_back(Off+j*VT.getVectorElementType().getStoreSize());
99 }
100 else {
101 ValueVTs.push_back(VT);
102 if (Offsets)
103 Offsets->push_back(Off);
104 }
105 }
106}
107
Justin Holewinskiae556d32012-05-04 20:18:50 +0000108// NVPTXTargetLowering Constructor.
Eric Christopherbef0a372015-01-30 01:50:07 +0000109NVPTXTargetLowering::NVPTXTargetLowering(const NVPTXTargetMachine &TM,
110 const NVPTXSubtarget &STI)
111 : TargetLowering(TM), nvTM(&TM), STI(STI) {
Justin Holewinskiae556d32012-05-04 20:18:50 +0000112
113 // always lower memset, memcpy, and memmove intrinsics to load/store
114 // instructions, rather
115 // then generating calls to memset, mempcy or memmove.
Justin Holewinski0497ab12013-03-30 14:29:21 +0000116 MaxStoresPerMemset = (unsigned) 0xFFFFFFFF;
117 MaxStoresPerMemcpy = (unsigned) 0xFFFFFFFF;
118 MaxStoresPerMemmove = (unsigned) 0xFFFFFFFF;
Justin Holewinskiae556d32012-05-04 20:18:50 +0000119
120 setBooleanContents(ZeroOrNegativeOneBooleanContent);
Justin Holewinskid7d8fe02014-06-27 18:35:42 +0000121 setBooleanVectorContents(ZeroOrNegativeOneBooleanContent);
Justin Holewinskiae556d32012-05-04 20:18:50 +0000122
123 // Jump is Expensive. Don't create extra control flow for 'and', 'or'
124 // condition branches.
125 setJumpIsExpensive(true);
126
127 // By default, use the Source scheduling
128 if (sched4reg)
129 setSchedulingPreference(Sched::RegPressure);
130 else
131 setSchedulingPreference(Sched::Source);
132
133 addRegisterClass(MVT::i1, &NVPTX::Int1RegsRegClass);
Justin Holewinskiae556d32012-05-04 20:18:50 +0000134 addRegisterClass(MVT::i16, &NVPTX::Int16RegsRegClass);
135 addRegisterClass(MVT::i32, &NVPTX::Int32RegsRegClass);
136 addRegisterClass(MVT::i64, &NVPTX::Int64RegsRegClass);
137 addRegisterClass(MVT::f32, &NVPTX::Float32RegsRegClass);
138 addRegisterClass(MVT::f64, &NVPTX::Float64RegsRegClass);
139
Justin Holewinskiae556d32012-05-04 20:18:50 +0000140 // Operations not directly supported by NVPTX.
Tom Stellard3787b122014-06-10 16:01:29 +0000141 setOperationAction(ISD::SELECT_CC, MVT::f32, Expand);
142 setOperationAction(ISD::SELECT_CC, MVT::f64, Expand);
143 setOperationAction(ISD::SELECT_CC, MVT::i1, Expand);
144 setOperationAction(ISD::SELECT_CC, MVT::i8, Expand);
145 setOperationAction(ISD::SELECT_CC, MVT::i16, Expand);
146 setOperationAction(ISD::SELECT_CC, MVT::i32, Expand);
147 setOperationAction(ISD::SELECT_CC, MVT::i64, Expand);
Justin Holewinski0497ab12013-03-30 14:29:21 +0000148 setOperationAction(ISD::BR_CC, MVT::f32, Expand);
149 setOperationAction(ISD::BR_CC, MVT::f64, Expand);
150 setOperationAction(ISD::BR_CC, MVT::i1, Expand);
151 setOperationAction(ISD::BR_CC, MVT::i8, Expand);
152 setOperationAction(ISD::BR_CC, MVT::i16, Expand);
153 setOperationAction(ISD::BR_CC, MVT::i32, Expand);
154 setOperationAction(ISD::BR_CC, MVT::i64, Expand);
Justin Holewinski318c6252013-07-01 12:58:56 +0000155 // Some SIGN_EXTEND_INREG can be done using cvt instruction.
156 // For others we will expand to a SHL/SRA pair.
157 setOperationAction(ISD::SIGN_EXTEND_INREG, MVT::i64, Legal);
158 setOperationAction(ISD::SIGN_EXTEND_INREG, MVT::i32, Legal);
159 setOperationAction(ISD::SIGN_EXTEND_INREG, MVT::i16, Legal);
160 setOperationAction(ISD::SIGN_EXTEND_INREG, MVT::i8 , Legal);
Justin Holewinski0497ab12013-03-30 14:29:21 +0000161 setOperationAction(ISD::SIGN_EXTEND_INREG, MVT::i1, Expand);
Justin Holewinskiae556d32012-05-04 20:18:50 +0000162
Justin Holewinski360a5cf2014-06-27 18:35:40 +0000163 setOperationAction(ISD::SHL_PARTS, MVT::i32 , Custom);
164 setOperationAction(ISD::SRA_PARTS, MVT::i32 , Custom);
165 setOperationAction(ISD::SRL_PARTS, MVT::i32 , Custom);
166 setOperationAction(ISD::SHL_PARTS, MVT::i64 , Custom);
167 setOperationAction(ISD::SRA_PARTS, MVT::i64 , Custom);
168 setOperationAction(ISD::SRL_PARTS, MVT::i64 , Custom);
169
Eric Christopherbef0a372015-01-30 01:50:07 +0000170 if (STI.hasROT64()) {
Justin Holewinski0497ab12013-03-30 14:29:21 +0000171 setOperationAction(ISD::ROTL, MVT::i64, Legal);
172 setOperationAction(ISD::ROTR, MVT::i64, Legal);
173 } else {
174 setOperationAction(ISD::ROTL, MVT::i64, Expand);
175 setOperationAction(ISD::ROTR, MVT::i64, Expand);
Justin Holewinskiae556d32012-05-04 20:18:50 +0000176 }
Eric Christopherbef0a372015-01-30 01:50:07 +0000177 if (STI.hasROT32()) {
Justin Holewinski0497ab12013-03-30 14:29:21 +0000178 setOperationAction(ISD::ROTL, MVT::i32, Legal);
179 setOperationAction(ISD::ROTR, MVT::i32, Legal);
180 } else {
181 setOperationAction(ISD::ROTL, MVT::i32, Expand);
182 setOperationAction(ISD::ROTR, MVT::i32, Expand);
Justin Holewinskiae556d32012-05-04 20:18:50 +0000183 }
184
Justin Holewinski0497ab12013-03-30 14:29:21 +0000185 setOperationAction(ISD::ROTL, MVT::i16, Expand);
186 setOperationAction(ISD::ROTR, MVT::i16, Expand);
187 setOperationAction(ISD::ROTL, MVT::i8, Expand);
188 setOperationAction(ISD::ROTR, MVT::i8, Expand);
189 setOperationAction(ISD::BSWAP, MVT::i16, Expand);
190 setOperationAction(ISD::BSWAP, MVT::i32, Expand);
191 setOperationAction(ISD::BSWAP, MVT::i64, Expand);
Justin Holewinskiae556d32012-05-04 20:18:50 +0000192
193 // Indirect branch is not supported.
194 // This also disables Jump Table creation.
Justin Holewinski0497ab12013-03-30 14:29:21 +0000195 setOperationAction(ISD::BR_JT, MVT::Other, Expand);
196 setOperationAction(ISD::BRIND, MVT::Other, Expand);
Justin Holewinskiae556d32012-05-04 20:18:50 +0000197
Justin Holewinski0497ab12013-03-30 14:29:21 +0000198 setOperationAction(ISD::GlobalAddress, MVT::i32, Custom);
199 setOperationAction(ISD::GlobalAddress, MVT::i64, Custom);
Justin Holewinskiae556d32012-05-04 20:18:50 +0000200
201 // We want to legalize constant related memmove and memcopy
202 // intrinsics.
203 setOperationAction(ISD::INTRINSIC_W_CHAIN, MVT::Other, Custom);
204
205 // Turn FP extload into load/fextend
Ahmed Bougacha2b6917b2015-01-08 00:51:32 +0000206 setLoadExtAction(ISD::EXTLOAD, MVT::f32, MVT::f16, Expand);
207 setLoadExtAction(ISD::EXTLOAD, MVT::f64, MVT::f16, Expand);
208 setLoadExtAction(ISD::EXTLOAD, MVT::f64, MVT::f32, Expand);
Jingyue Wua0a56602015-07-01 21:32:42 +0000209 setLoadExtAction(ISD::EXTLOAD, MVT::v2f32, MVT::v2f16, Expand);
210 setLoadExtAction(ISD::EXTLOAD, MVT::v2f64, MVT::v2f16, Expand);
211 setLoadExtAction(ISD::EXTLOAD, MVT::v2f64, MVT::v2f32, Expand);
212 setLoadExtAction(ISD::EXTLOAD, MVT::v4f32, MVT::v4f16, Expand);
213 setLoadExtAction(ISD::EXTLOAD, MVT::v4f64, MVT::v4f16, Expand);
214 setLoadExtAction(ISD::EXTLOAD, MVT::v4f64, MVT::v4f32, Expand);
Justin Holewinskiae556d32012-05-04 20:18:50 +0000215 // Turn FP truncstore into trunc + store.
Jingyue Wua0a56602015-07-01 21:32:42 +0000216 // FIXME: vector types should also be expanded
Tim Northover9e108a02014-07-18 13:01:43 +0000217 setTruncStoreAction(MVT::f32, MVT::f16, Expand);
218 setTruncStoreAction(MVT::f64, MVT::f16, Expand);
Justin Holewinskiae556d32012-05-04 20:18:50 +0000219 setTruncStoreAction(MVT::f64, MVT::f32, Expand);
220
221 // PTX does not support load / store predicate registers
Justin Holewinskic6462aa2012-11-14 19:19:16 +0000222 setOperationAction(ISD::LOAD, MVT::i1, Custom);
223 setOperationAction(ISD::STORE, MVT::i1, Custom);
224
Ahmed Bougacha2b6917b2015-01-08 00:51:32 +0000225 for (MVT VT : MVT::integer_valuetypes()) {
226 setLoadExtAction(ISD::SEXTLOAD, VT, MVT::i1, Promote);
227 setLoadExtAction(ISD::ZEXTLOAD, VT, MVT::i1, Promote);
228 setTruncStoreAction(VT, MVT::i1, Expand);
229 }
Justin Holewinskiae556d32012-05-04 20:18:50 +0000230
231 // This is legal in NVPTX
Justin Holewinski0497ab12013-03-30 14:29:21 +0000232 setOperationAction(ISD::ConstantFP, MVT::f64, Legal);
233 setOperationAction(ISD::ConstantFP, MVT::f32, Legal);
Justin Holewinskiae556d32012-05-04 20:18:50 +0000234
235 // TRAP can be lowered to PTX trap
Justin Holewinski0497ab12013-03-30 14:29:21 +0000236 setOperationAction(ISD::TRAP, MVT::Other, Legal);
Justin Holewinskiae556d32012-05-04 20:18:50 +0000237
Justin Holewinski51cb1342013-07-01 12:59:04 +0000238 setOperationAction(ISD::ADDC, MVT::i64, Expand);
239 setOperationAction(ISD::ADDE, MVT::i64, Expand);
240
Justin Holewinskibe8dc642013-02-12 14:18:49 +0000241 // Register custom handling for vector loads/stores
Ahmed Bougacha67dd2d22015-01-07 21:27:10 +0000242 for (MVT VT : MVT::vector_valuetypes()) {
Justin Holewinskibe8dc642013-02-12 14:18:49 +0000243 if (IsPTXVectorType(VT)) {
244 setOperationAction(ISD::LOAD, VT, Custom);
245 setOperationAction(ISD::STORE, VT, Custom);
246 setOperationAction(ISD::INTRINSIC_W_CHAIN, VT, Custom);
247 }
248 }
Justin Holewinskiae556d32012-05-04 20:18:50 +0000249
Justin Holewinskif8f70912013-06-28 17:57:59 +0000250 // Custom handling for i8 intrinsics
251 setOperationAction(ISD::INTRINSIC_W_CHAIN, MVT::i8, Custom);
252
Justin Holewinskidc372df2013-06-28 17:58:07 +0000253 setOperationAction(ISD::CTLZ, MVT::i16, Legal);
254 setOperationAction(ISD::CTLZ, MVT::i32, Legal);
255 setOperationAction(ISD::CTLZ, MVT::i64, Legal);
256 setOperationAction(ISD::CTLZ_ZERO_UNDEF, MVT::i16, Legal);
257 setOperationAction(ISD::CTLZ_ZERO_UNDEF, MVT::i32, Legal);
258 setOperationAction(ISD::CTLZ_ZERO_UNDEF, MVT::i64, Legal);
259 setOperationAction(ISD::CTTZ, MVT::i16, Expand);
260 setOperationAction(ISD::CTTZ, MVT::i32, Expand);
261 setOperationAction(ISD::CTTZ, MVT::i64, Expand);
262 setOperationAction(ISD::CTTZ_ZERO_UNDEF, MVT::i16, Expand);
263 setOperationAction(ISD::CTTZ_ZERO_UNDEF, MVT::i32, Expand);
264 setOperationAction(ISD::CTTZ_ZERO_UNDEF, MVT::i64, Expand);
265 setOperationAction(ISD::CTPOP, MVT::i16, Legal);
266 setOperationAction(ISD::CTPOP, MVT::i32, Legal);
267 setOperationAction(ISD::CTPOP, MVT::i64, Legal);
268
Justin Holewinskid4d2e9b2015-01-26 19:52:20 +0000269 // PTX does not directly support SELP of i1, so promote to i32 first
270 setOperationAction(ISD::SELECT, MVT::i1, Custom);
271
Justin Holewinskieafe26d2014-06-27 18:35:37 +0000272 // We have some custom DAG combine patterns for these nodes
273 setTargetDAGCombine(ISD::ADD);
274 setTargetDAGCombine(ISD::AND);
275 setTargetDAGCombine(ISD::FADD);
276 setTargetDAGCombine(ISD::MUL);
277 setTargetDAGCombine(ISD::SHL);
278
Justin Holewinskiae556d32012-05-04 20:18:50 +0000279 // Now deduce the information based on the above mentioned
280 // actions
Eric Christopher23a3a7c2015-02-26 00:00:24 +0000281 computeRegisterProperties(STI.getRegisterInfo());
Justin Holewinskiae556d32012-05-04 20:18:50 +0000282}
283
Justin Holewinskiae556d32012-05-04 20:18:50 +0000284const char *NVPTXTargetLowering::getTargetNodeName(unsigned Opcode) const {
Matthias Braund04893f2015-05-07 21:33:59 +0000285 switch ((NVPTXISD::NodeType)Opcode) {
286 case NVPTXISD::FIRST_NUMBER:
287 break;
Justin Holewinski0497ab12013-03-30 14:29:21 +0000288 case NVPTXISD::CALL:
289 return "NVPTXISD::CALL";
290 case NVPTXISD::RET_FLAG:
291 return "NVPTXISD::RET_FLAG";
Matthias Braund04893f2015-05-07 21:33:59 +0000292 case NVPTXISD::LOAD_PARAM:
293 return "NVPTXISD::LOAD_PARAM";
Justin Holewinski0497ab12013-03-30 14:29:21 +0000294 case NVPTXISD::Wrapper:
295 return "NVPTXISD::Wrapper";
Justin Holewinski0497ab12013-03-30 14:29:21 +0000296 case NVPTXISD::DeclareParam:
297 return "NVPTXISD::DeclareParam";
Justin Holewinskiae556d32012-05-04 20:18:50 +0000298 case NVPTXISD::DeclareScalarParam:
299 return "NVPTXISD::DeclareScalarParam";
Justin Holewinski0497ab12013-03-30 14:29:21 +0000300 case NVPTXISD::DeclareRet:
301 return "NVPTXISD::DeclareRet";
Matthias Braund04893f2015-05-07 21:33:59 +0000302 case NVPTXISD::DeclareScalarRet:
303 return "NVPTXISD::DeclareScalarRet";
Justin Holewinski0497ab12013-03-30 14:29:21 +0000304 case NVPTXISD::DeclareRetParam:
305 return "NVPTXISD::DeclareRetParam";
306 case NVPTXISD::PrintCall:
307 return "NVPTXISD::PrintCall";
Matthias Braund04893f2015-05-07 21:33:59 +0000308 case NVPTXISD::PrintCallUni:
309 return "NVPTXISD::PrintCallUni";
Justin Holewinski0497ab12013-03-30 14:29:21 +0000310 case NVPTXISD::LoadParam:
311 return "NVPTXISD::LoadParam";
Justin Holewinskife44314f2013-06-28 17:57:51 +0000312 case NVPTXISD::LoadParamV2:
313 return "NVPTXISD::LoadParamV2";
314 case NVPTXISD::LoadParamV4:
315 return "NVPTXISD::LoadParamV4";
Justin Holewinski0497ab12013-03-30 14:29:21 +0000316 case NVPTXISD::StoreParam:
317 return "NVPTXISD::StoreParam";
Justin Holewinskife44314f2013-06-28 17:57:51 +0000318 case NVPTXISD::StoreParamV2:
319 return "NVPTXISD::StoreParamV2";
320 case NVPTXISD::StoreParamV4:
321 return "NVPTXISD::StoreParamV4";
Justin Holewinski0497ab12013-03-30 14:29:21 +0000322 case NVPTXISD::StoreParamS32:
323 return "NVPTXISD::StoreParamS32";
324 case NVPTXISD::StoreParamU32:
325 return "NVPTXISD::StoreParamU32";
Justin Holewinski0497ab12013-03-30 14:29:21 +0000326 case NVPTXISD::CallArgBegin:
327 return "NVPTXISD::CallArgBegin";
328 case NVPTXISD::CallArg:
329 return "NVPTXISD::CallArg";
330 case NVPTXISD::LastCallArg:
331 return "NVPTXISD::LastCallArg";
332 case NVPTXISD::CallArgEnd:
333 return "NVPTXISD::CallArgEnd";
334 case NVPTXISD::CallVoid:
335 return "NVPTXISD::CallVoid";
336 case NVPTXISD::CallVal:
337 return "NVPTXISD::CallVal";
338 case NVPTXISD::CallSymbol:
339 return "NVPTXISD::CallSymbol";
340 case NVPTXISD::Prototype:
341 return "NVPTXISD::Prototype";
342 case NVPTXISD::MoveParam:
343 return "NVPTXISD::MoveParam";
Justin Holewinski0497ab12013-03-30 14:29:21 +0000344 case NVPTXISD::StoreRetval:
345 return "NVPTXISD::StoreRetval";
Justin Holewinskife44314f2013-06-28 17:57:51 +0000346 case NVPTXISD::StoreRetvalV2:
347 return "NVPTXISD::StoreRetvalV2";
348 case NVPTXISD::StoreRetvalV4:
349 return "NVPTXISD::StoreRetvalV4";
Justin Holewinski0497ab12013-03-30 14:29:21 +0000350 case NVPTXISD::PseudoUseParam:
351 return "NVPTXISD::PseudoUseParam";
352 case NVPTXISD::RETURN:
353 return "NVPTXISD::RETURN";
354 case NVPTXISD::CallSeqBegin:
355 return "NVPTXISD::CallSeqBegin";
356 case NVPTXISD::CallSeqEnd:
357 return "NVPTXISD::CallSeqEnd";
Justin Holewinski3d49e5c2013-11-15 12:30:04 +0000358 case NVPTXISD::CallPrototype:
359 return "NVPTXISD::CallPrototype";
Justin Holewinski0497ab12013-03-30 14:29:21 +0000360 case NVPTXISD::LoadV2:
361 return "NVPTXISD::LoadV2";
362 case NVPTXISD::LoadV4:
363 return "NVPTXISD::LoadV4";
364 case NVPTXISD::LDGV2:
365 return "NVPTXISD::LDGV2";
366 case NVPTXISD::LDGV4:
367 return "NVPTXISD::LDGV4";
368 case NVPTXISD::LDUV2:
369 return "NVPTXISD::LDUV2";
370 case NVPTXISD::LDUV4:
371 return "NVPTXISD::LDUV4";
372 case NVPTXISD::StoreV2:
373 return "NVPTXISD::StoreV2";
374 case NVPTXISD::StoreV4:
375 return "NVPTXISD::StoreV4";
Justin Holewinskieafe26d2014-06-27 18:35:37 +0000376 case NVPTXISD::FUN_SHFL_CLAMP:
377 return "NVPTXISD::FUN_SHFL_CLAMP";
378 case NVPTXISD::FUN_SHFR_CLAMP:
379 return "NVPTXISD::FUN_SHFR_CLAMP";
Justin Holewinski360a5cf2014-06-27 18:35:40 +0000380 case NVPTXISD::IMAD:
381 return "NVPTXISD::IMAD";
Matthias Braund04893f2015-05-07 21:33:59 +0000382 case NVPTXISD::Dummy:
383 return "NVPTXISD::Dummy";
Justin Holewinski360a5cf2014-06-27 18:35:40 +0000384 case NVPTXISD::MUL_WIDE_SIGNED:
385 return "NVPTXISD::MUL_WIDE_SIGNED";
386 case NVPTXISD::MUL_WIDE_UNSIGNED:
387 return "NVPTXISD::MUL_WIDE_UNSIGNED";
Justin Holewinski9a2350e2014-07-17 11:59:04 +0000388 case NVPTXISD::Tex1DFloatS32: return "NVPTXISD::Tex1DFloatS32";
Justin Holewinski30d56a72014-04-09 15:39:15 +0000389 case NVPTXISD::Tex1DFloatFloat: return "NVPTXISD::Tex1DFloatFloat";
390 case NVPTXISD::Tex1DFloatFloatLevel:
391 return "NVPTXISD::Tex1DFloatFloatLevel";
392 case NVPTXISD::Tex1DFloatFloatGrad:
393 return "NVPTXISD::Tex1DFloatFloatGrad";
Justin Holewinski9a2350e2014-07-17 11:59:04 +0000394 case NVPTXISD::Tex1DS32S32: return "NVPTXISD::Tex1DS32S32";
395 case NVPTXISD::Tex1DS32Float: return "NVPTXISD::Tex1DS32Float";
396 case NVPTXISD::Tex1DS32FloatLevel:
397 return "NVPTXISD::Tex1DS32FloatLevel";
398 case NVPTXISD::Tex1DS32FloatGrad:
399 return "NVPTXISD::Tex1DS32FloatGrad";
400 case NVPTXISD::Tex1DU32S32: return "NVPTXISD::Tex1DU32S32";
401 case NVPTXISD::Tex1DU32Float: return "NVPTXISD::Tex1DU32Float";
402 case NVPTXISD::Tex1DU32FloatLevel:
403 return "NVPTXISD::Tex1DU32FloatLevel";
404 case NVPTXISD::Tex1DU32FloatGrad:
405 return "NVPTXISD::Tex1DU32FloatGrad";
406 case NVPTXISD::Tex1DArrayFloatS32: return "NVPTXISD::Tex1DArrayFloatS32";
407 case NVPTXISD::Tex1DArrayFloatFloat: return "NVPTXISD::Tex1DArrayFloatFloat";
Justin Holewinski30d56a72014-04-09 15:39:15 +0000408 case NVPTXISD::Tex1DArrayFloatFloatLevel:
Justin Holewinski9a2350e2014-07-17 11:59:04 +0000409 return "NVPTXISD::Tex1DArrayFloatFloatLevel";
Justin Holewinski30d56a72014-04-09 15:39:15 +0000410 case NVPTXISD::Tex1DArrayFloatFloatGrad:
Justin Holewinski9a2350e2014-07-17 11:59:04 +0000411 return "NVPTXISD::Tex1DArrayFloatFloatGrad";
412 case NVPTXISD::Tex1DArrayS32S32: return "NVPTXISD::Tex1DArrayS32S32";
413 case NVPTXISD::Tex1DArrayS32Float: return "NVPTXISD::Tex1DArrayS32Float";
414 case NVPTXISD::Tex1DArrayS32FloatLevel:
415 return "NVPTXISD::Tex1DArrayS32FloatLevel";
416 case NVPTXISD::Tex1DArrayS32FloatGrad:
417 return "NVPTXISD::Tex1DArrayS32FloatGrad";
418 case NVPTXISD::Tex1DArrayU32S32: return "NVPTXISD::Tex1DArrayU32S32";
419 case NVPTXISD::Tex1DArrayU32Float: return "NVPTXISD::Tex1DArrayU32Float";
420 case NVPTXISD::Tex1DArrayU32FloatLevel:
421 return "NVPTXISD::Tex1DArrayU32FloatLevel";
422 case NVPTXISD::Tex1DArrayU32FloatGrad:
423 return "NVPTXISD::Tex1DArrayU32FloatGrad";
424 case NVPTXISD::Tex2DFloatS32: return "NVPTXISD::Tex2DFloatS32";
Justin Holewinski30d56a72014-04-09 15:39:15 +0000425 case NVPTXISD::Tex2DFloatFloat: return "NVPTXISD::Tex2DFloatFloat";
426 case NVPTXISD::Tex2DFloatFloatLevel:
427 return "NVPTXISD::Tex2DFloatFloatLevel";
428 case NVPTXISD::Tex2DFloatFloatGrad:
429 return "NVPTXISD::Tex2DFloatFloatGrad";
Justin Holewinski9a2350e2014-07-17 11:59:04 +0000430 case NVPTXISD::Tex2DS32S32: return "NVPTXISD::Tex2DS32S32";
431 case NVPTXISD::Tex2DS32Float: return "NVPTXISD::Tex2DS32Float";
432 case NVPTXISD::Tex2DS32FloatLevel:
433 return "NVPTXISD::Tex2DS32FloatLevel";
434 case NVPTXISD::Tex2DS32FloatGrad:
435 return "NVPTXISD::Tex2DS32FloatGrad";
436 case NVPTXISD::Tex2DU32S32: return "NVPTXISD::Tex2DU32S32";
437 case NVPTXISD::Tex2DU32Float: return "NVPTXISD::Tex2DU32Float";
438 case NVPTXISD::Tex2DU32FloatLevel:
439 return "NVPTXISD::Tex2DU32FloatLevel";
440 case NVPTXISD::Tex2DU32FloatGrad:
441 return "NVPTXISD::Tex2DU32FloatGrad";
442 case NVPTXISD::Tex2DArrayFloatS32: return "NVPTXISD::Tex2DArrayFloatS32";
Justin Holewinski30d56a72014-04-09 15:39:15 +0000443 case NVPTXISD::Tex2DArrayFloatFloat: return "NVPTXISD::Tex2DArrayFloatFloat";
444 case NVPTXISD::Tex2DArrayFloatFloatLevel:
445 return "NVPTXISD::Tex2DArrayFloatFloatLevel";
446 case NVPTXISD::Tex2DArrayFloatFloatGrad:
447 return "NVPTXISD::Tex2DArrayFloatFloatGrad";
Justin Holewinski9a2350e2014-07-17 11:59:04 +0000448 case NVPTXISD::Tex2DArrayS32S32: return "NVPTXISD::Tex2DArrayS32S32";
449 case NVPTXISD::Tex2DArrayS32Float: return "NVPTXISD::Tex2DArrayS32Float";
450 case NVPTXISD::Tex2DArrayS32FloatLevel:
451 return "NVPTXISD::Tex2DArrayS32FloatLevel";
452 case NVPTXISD::Tex2DArrayS32FloatGrad:
453 return "NVPTXISD::Tex2DArrayS32FloatGrad";
454 case NVPTXISD::Tex2DArrayU32S32: return "NVPTXISD::Tex2DArrayU32S32";
455 case NVPTXISD::Tex2DArrayU32Float: return "NVPTXISD::Tex2DArrayU32Float";
456 case NVPTXISD::Tex2DArrayU32FloatLevel:
457 return "NVPTXISD::Tex2DArrayU32FloatLevel";
458 case NVPTXISD::Tex2DArrayU32FloatGrad:
459 return "NVPTXISD::Tex2DArrayU32FloatGrad";
460 case NVPTXISD::Tex3DFloatS32: return "NVPTXISD::Tex3DFloatS32";
Justin Holewinski30d56a72014-04-09 15:39:15 +0000461 case NVPTXISD::Tex3DFloatFloat: return "NVPTXISD::Tex3DFloatFloat";
462 case NVPTXISD::Tex3DFloatFloatLevel:
463 return "NVPTXISD::Tex3DFloatFloatLevel";
464 case NVPTXISD::Tex3DFloatFloatGrad:
465 return "NVPTXISD::Tex3DFloatFloatGrad";
Justin Holewinski9a2350e2014-07-17 11:59:04 +0000466 case NVPTXISD::Tex3DS32S32: return "NVPTXISD::Tex3DS32S32";
467 case NVPTXISD::Tex3DS32Float: return "NVPTXISD::Tex3DS32Float";
468 case NVPTXISD::Tex3DS32FloatLevel:
469 return "NVPTXISD::Tex3DS32FloatLevel";
470 case NVPTXISD::Tex3DS32FloatGrad:
471 return "NVPTXISD::Tex3DS32FloatGrad";
472 case NVPTXISD::Tex3DU32S32: return "NVPTXISD::Tex3DU32S32";
473 case NVPTXISD::Tex3DU32Float: return "NVPTXISD::Tex3DU32Float";
474 case NVPTXISD::Tex3DU32FloatLevel:
475 return "NVPTXISD::Tex3DU32FloatLevel";
476 case NVPTXISD::Tex3DU32FloatGrad:
477 return "NVPTXISD::Tex3DU32FloatGrad";
478 case NVPTXISD::TexCubeFloatFloat: return "NVPTXISD::TexCubeFloatFloat";
479 case NVPTXISD::TexCubeFloatFloatLevel:
480 return "NVPTXISD::TexCubeFloatFloatLevel";
481 case NVPTXISD::TexCubeS32Float: return "NVPTXISD::TexCubeS32Float";
482 case NVPTXISD::TexCubeS32FloatLevel:
483 return "NVPTXISD::TexCubeS32FloatLevel";
484 case NVPTXISD::TexCubeU32Float: return "NVPTXISD::TexCubeU32Float";
485 case NVPTXISD::TexCubeU32FloatLevel:
486 return "NVPTXISD::TexCubeU32FloatLevel";
487 case NVPTXISD::TexCubeArrayFloatFloat:
488 return "NVPTXISD::TexCubeArrayFloatFloat";
489 case NVPTXISD::TexCubeArrayFloatFloatLevel:
490 return "NVPTXISD::TexCubeArrayFloatFloatLevel";
491 case NVPTXISD::TexCubeArrayS32Float:
492 return "NVPTXISD::TexCubeArrayS32Float";
493 case NVPTXISD::TexCubeArrayS32FloatLevel:
494 return "NVPTXISD::TexCubeArrayS32FloatLevel";
495 case NVPTXISD::TexCubeArrayU32Float:
496 return "NVPTXISD::TexCubeArrayU32Float";
497 case NVPTXISD::TexCubeArrayU32FloatLevel:
498 return "NVPTXISD::TexCubeArrayU32FloatLevel";
499 case NVPTXISD::Tld4R2DFloatFloat:
500 return "NVPTXISD::Tld4R2DFloatFloat";
501 case NVPTXISD::Tld4G2DFloatFloat:
502 return "NVPTXISD::Tld4G2DFloatFloat";
503 case NVPTXISD::Tld4B2DFloatFloat:
504 return "NVPTXISD::Tld4B2DFloatFloat";
505 case NVPTXISD::Tld4A2DFloatFloat:
506 return "NVPTXISD::Tld4A2DFloatFloat";
507 case NVPTXISD::Tld4R2DS64Float:
508 return "NVPTXISD::Tld4R2DS64Float";
509 case NVPTXISD::Tld4G2DS64Float:
510 return "NVPTXISD::Tld4G2DS64Float";
511 case NVPTXISD::Tld4B2DS64Float:
512 return "NVPTXISD::Tld4B2DS64Float";
513 case NVPTXISD::Tld4A2DS64Float:
514 return "NVPTXISD::Tld4A2DS64Float";
515 case NVPTXISD::Tld4R2DU64Float:
516 return "NVPTXISD::Tld4R2DU64Float";
517 case NVPTXISD::Tld4G2DU64Float:
518 return "NVPTXISD::Tld4G2DU64Float";
519 case NVPTXISD::Tld4B2DU64Float:
520 return "NVPTXISD::Tld4B2DU64Float";
521 case NVPTXISD::Tld4A2DU64Float:
522 return "NVPTXISD::Tld4A2DU64Float";
523
524 case NVPTXISD::TexUnified1DFloatS32:
525 return "NVPTXISD::TexUnified1DFloatS32";
526 case NVPTXISD::TexUnified1DFloatFloat:
527 return "NVPTXISD::TexUnified1DFloatFloat";
528 case NVPTXISD::TexUnified1DFloatFloatLevel:
529 return "NVPTXISD::TexUnified1DFloatFloatLevel";
530 case NVPTXISD::TexUnified1DFloatFloatGrad:
531 return "NVPTXISD::TexUnified1DFloatFloatGrad";
532 case NVPTXISD::TexUnified1DS32S32:
533 return "NVPTXISD::TexUnified1DS32S32";
534 case NVPTXISD::TexUnified1DS32Float:
535 return "NVPTXISD::TexUnified1DS32Float";
536 case NVPTXISD::TexUnified1DS32FloatLevel:
537 return "NVPTXISD::TexUnified1DS32FloatLevel";
538 case NVPTXISD::TexUnified1DS32FloatGrad:
539 return "NVPTXISD::TexUnified1DS32FloatGrad";
540 case NVPTXISD::TexUnified1DU32S32:
541 return "NVPTXISD::TexUnified1DU32S32";
542 case NVPTXISD::TexUnified1DU32Float:
543 return "NVPTXISD::TexUnified1DU32Float";
544 case NVPTXISD::TexUnified1DU32FloatLevel:
545 return "NVPTXISD::TexUnified1DU32FloatLevel";
546 case NVPTXISD::TexUnified1DU32FloatGrad:
547 return "NVPTXISD::TexUnified1DU32FloatGrad";
548 case NVPTXISD::TexUnified1DArrayFloatS32:
549 return "NVPTXISD::TexUnified1DArrayFloatS32";
550 case NVPTXISD::TexUnified1DArrayFloatFloat:
551 return "NVPTXISD::TexUnified1DArrayFloatFloat";
552 case NVPTXISD::TexUnified1DArrayFloatFloatLevel:
553 return "NVPTXISD::TexUnified1DArrayFloatFloatLevel";
554 case NVPTXISD::TexUnified1DArrayFloatFloatGrad:
555 return "NVPTXISD::TexUnified1DArrayFloatFloatGrad";
556 case NVPTXISD::TexUnified1DArrayS32S32:
557 return "NVPTXISD::TexUnified1DArrayS32S32";
558 case NVPTXISD::TexUnified1DArrayS32Float:
559 return "NVPTXISD::TexUnified1DArrayS32Float";
560 case NVPTXISD::TexUnified1DArrayS32FloatLevel:
561 return "NVPTXISD::TexUnified1DArrayS32FloatLevel";
562 case NVPTXISD::TexUnified1DArrayS32FloatGrad:
563 return "NVPTXISD::TexUnified1DArrayS32FloatGrad";
564 case NVPTXISD::TexUnified1DArrayU32S32:
565 return "NVPTXISD::TexUnified1DArrayU32S32";
566 case NVPTXISD::TexUnified1DArrayU32Float:
567 return "NVPTXISD::TexUnified1DArrayU32Float";
568 case NVPTXISD::TexUnified1DArrayU32FloatLevel:
569 return "NVPTXISD::TexUnified1DArrayU32FloatLevel";
570 case NVPTXISD::TexUnified1DArrayU32FloatGrad:
571 return "NVPTXISD::TexUnified1DArrayU32FloatGrad";
572 case NVPTXISD::TexUnified2DFloatS32:
573 return "NVPTXISD::TexUnified2DFloatS32";
574 case NVPTXISD::TexUnified2DFloatFloat:
575 return "NVPTXISD::TexUnified2DFloatFloat";
576 case NVPTXISD::TexUnified2DFloatFloatLevel:
577 return "NVPTXISD::TexUnified2DFloatFloatLevel";
578 case NVPTXISD::TexUnified2DFloatFloatGrad:
579 return "NVPTXISD::TexUnified2DFloatFloatGrad";
580 case NVPTXISD::TexUnified2DS32S32:
581 return "NVPTXISD::TexUnified2DS32S32";
582 case NVPTXISD::TexUnified2DS32Float:
583 return "NVPTXISD::TexUnified2DS32Float";
584 case NVPTXISD::TexUnified2DS32FloatLevel:
585 return "NVPTXISD::TexUnified2DS32FloatLevel";
586 case NVPTXISD::TexUnified2DS32FloatGrad:
587 return "NVPTXISD::TexUnified2DS32FloatGrad";
588 case NVPTXISD::TexUnified2DU32S32:
589 return "NVPTXISD::TexUnified2DU32S32";
590 case NVPTXISD::TexUnified2DU32Float:
591 return "NVPTXISD::TexUnified2DU32Float";
592 case NVPTXISD::TexUnified2DU32FloatLevel:
593 return "NVPTXISD::TexUnified2DU32FloatLevel";
594 case NVPTXISD::TexUnified2DU32FloatGrad:
595 return "NVPTXISD::TexUnified2DU32FloatGrad";
596 case NVPTXISD::TexUnified2DArrayFloatS32:
597 return "NVPTXISD::TexUnified2DArrayFloatS32";
598 case NVPTXISD::TexUnified2DArrayFloatFloat:
599 return "NVPTXISD::TexUnified2DArrayFloatFloat";
600 case NVPTXISD::TexUnified2DArrayFloatFloatLevel:
601 return "NVPTXISD::TexUnified2DArrayFloatFloatLevel";
602 case NVPTXISD::TexUnified2DArrayFloatFloatGrad:
603 return "NVPTXISD::TexUnified2DArrayFloatFloatGrad";
604 case NVPTXISD::TexUnified2DArrayS32S32:
605 return "NVPTXISD::TexUnified2DArrayS32S32";
606 case NVPTXISD::TexUnified2DArrayS32Float:
607 return "NVPTXISD::TexUnified2DArrayS32Float";
608 case NVPTXISD::TexUnified2DArrayS32FloatLevel:
609 return "NVPTXISD::TexUnified2DArrayS32FloatLevel";
610 case NVPTXISD::TexUnified2DArrayS32FloatGrad:
611 return "NVPTXISD::TexUnified2DArrayS32FloatGrad";
612 case NVPTXISD::TexUnified2DArrayU32S32:
613 return "NVPTXISD::TexUnified2DArrayU32S32";
614 case NVPTXISD::TexUnified2DArrayU32Float:
615 return "NVPTXISD::TexUnified2DArrayU32Float";
616 case NVPTXISD::TexUnified2DArrayU32FloatLevel:
617 return "NVPTXISD::TexUnified2DArrayU32FloatLevel";
618 case NVPTXISD::TexUnified2DArrayU32FloatGrad:
619 return "NVPTXISD::TexUnified2DArrayU32FloatGrad";
620 case NVPTXISD::TexUnified3DFloatS32:
621 return "NVPTXISD::TexUnified3DFloatS32";
622 case NVPTXISD::TexUnified3DFloatFloat:
623 return "NVPTXISD::TexUnified3DFloatFloat";
624 case NVPTXISD::TexUnified3DFloatFloatLevel:
625 return "NVPTXISD::TexUnified3DFloatFloatLevel";
626 case NVPTXISD::TexUnified3DFloatFloatGrad:
627 return "NVPTXISD::TexUnified3DFloatFloatGrad";
628 case NVPTXISD::TexUnified3DS32S32:
629 return "NVPTXISD::TexUnified3DS32S32";
630 case NVPTXISD::TexUnified3DS32Float:
631 return "NVPTXISD::TexUnified3DS32Float";
632 case NVPTXISD::TexUnified3DS32FloatLevel:
633 return "NVPTXISD::TexUnified3DS32FloatLevel";
634 case NVPTXISD::TexUnified3DS32FloatGrad:
635 return "NVPTXISD::TexUnified3DS32FloatGrad";
636 case NVPTXISD::TexUnified3DU32S32:
637 return "NVPTXISD::TexUnified3DU32S32";
638 case NVPTXISD::TexUnified3DU32Float:
639 return "NVPTXISD::TexUnified3DU32Float";
640 case NVPTXISD::TexUnified3DU32FloatLevel:
641 return "NVPTXISD::TexUnified3DU32FloatLevel";
642 case NVPTXISD::TexUnified3DU32FloatGrad:
643 return "NVPTXISD::TexUnified3DU32FloatGrad";
644 case NVPTXISD::TexUnifiedCubeFloatFloat:
645 return "NVPTXISD::TexUnifiedCubeFloatFloat";
646 case NVPTXISD::TexUnifiedCubeFloatFloatLevel:
647 return "NVPTXISD::TexUnifiedCubeFloatFloatLevel";
648 case NVPTXISD::TexUnifiedCubeS32Float:
649 return "NVPTXISD::TexUnifiedCubeS32Float";
650 case NVPTXISD::TexUnifiedCubeS32FloatLevel:
651 return "NVPTXISD::TexUnifiedCubeS32FloatLevel";
652 case NVPTXISD::TexUnifiedCubeU32Float:
653 return "NVPTXISD::TexUnifiedCubeU32Float";
654 case NVPTXISD::TexUnifiedCubeU32FloatLevel:
655 return "NVPTXISD::TexUnifiedCubeU32FloatLevel";
656 case NVPTXISD::TexUnifiedCubeArrayFloatFloat:
657 return "NVPTXISD::TexUnifiedCubeArrayFloatFloat";
658 case NVPTXISD::TexUnifiedCubeArrayFloatFloatLevel:
659 return "NVPTXISD::TexUnifiedCubeArrayFloatFloatLevel";
660 case NVPTXISD::TexUnifiedCubeArrayS32Float:
661 return "NVPTXISD::TexUnifiedCubeArrayS32Float";
662 case NVPTXISD::TexUnifiedCubeArrayS32FloatLevel:
663 return "NVPTXISD::TexUnifiedCubeArrayS32FloatLevel";
664 case NVPTXISD::TexUnifiedCubeArrayU32Float:
665 return "NVPTXISD::TexUnifiedCubeArrayU32Float";
666 case NVPTXISD::TexUnifiedCubeArrayU32FloatLevel:
667 return "NVPTXISD::TexUnifiedCubeArrayU32FloatLevel";
668 case NVPTXISD::Tld4UnifiedR2DFloatFloat:
669 return "NVPTXISD::Tld4UnifiedR2DFloatFloat";
670 case NVPTXISD::Tld4UnifiedG2DFloatFloat:
671 return "NVPTXISD::Tld4UnifiedG2DFloatFloat";
672 case NVPTXISD::Tld4UnifiedB2DFloatFloat:
673 return "NVPTXISD::Tld4UnifiedB2DFloatFloat";
674 case NVPTXISD::Tld4UnifiedA2DFloatFloat:
675 return "NVPTXISD::Tld4UnifiedA2DFloatFloat";
676 case NVPTXISD::Tld4UnifiedR2DS64Float:
677 return "NVPTXISD::Tld4UnifiedR2DS64Float";
678 case NVPTXISD::Tld4UnifiedG2DS64Float:
679 return "NVPTXISD::Tld4UnifiedG2DS64Float";
680 case NVPTXISD::Tld4UnifiedB2DS64Float:
681 return "NVPTXISD::Tld4UnifiedB2DS64Float";
682 case NVPTXISD::Tld4UnifiedA2DS64Float:
683 return "NVPTXISD::Tld4UnifiedA2DS64Float";
684 case NVPTXISD::Tld4UnifiedR2DU64Float:
685 return "NVPTXISD::Tld4UnifiedR2DU64Float";
686 case NVPTXISD::Tld4UnifiedG2DU64Float:
687 return "NVPTXISD::Tld4UnifiedG2DU64Float";
688 case NVPTXISD::Tld4UnifiedB2DU64Float:
689 return "NVPTXISD::Tld4UnifiedB2DU64Float";
690 case NVPTXISD::Tld4UnifiedA2DU64Float:
691 return "NVPTXISD::Tld4UnifiedA2DU64Float";
692
693 case NVPTXISD::Suld1DI8Clamp: return "NVPTXISD::Suld1DI8Clamp";
694 case NVPTXISD::Suld1DI16Clamp: return "NVPTXISD::Suld1DI16Clamp";
695 case NVPTXISD::Suld1DI32Clamp: return "NVPTXISD::Suld1DI32Clamp";
696 case NVPTXISD::Suld1DI64Clamp: return "NVPTXISD::Suld1DI64Clamp";
697 case NVPTXISD::Suld1DV2I8Clamp: return "NVPTXISD::Suld1DV2I8Clamp";
698 case NVPTXISD::Suld1DV2I16Clamp: return "NVPTXISD::Suld1DV2I16Clamp";
699 case NVPTXISD::Suld1DV2I32Clamp: return "NVPTXISD::Suld1DV2I32Clamp";
700 case NVPTXISD::Suld1DV2I64Clamp: return "NVPTXISD::Suld1DV2I64Clamp";
701 case NVPTXISD::Suld1DV4I8Clamp: return "NVPTXISD::Suld1DV4I8Clamp";
702 case NVPTXISD::Suld1DV4I16Clamp: return "NVPTXISD::Suld1DV4I16Clamp";
703 case NVPTXISD::Suld1DV4I32Clamp: return "NVPTXISD::Suld1DV4I32Clamp";
704
705 case NVPTXISD::Suld1DArrayI8Clamp: return "NVPTXISD::Suld1DArrayI8Clamp";
706 case NVPTXISD::Suld1DArrayI16Clamp: return "NVPTXISD::Suld1DArrayI16Clamp";
707 case NVPTXISD::Suld1DArrayI32Clamp: return "NVPTXISD::Suld1DArrayI32Clamp";
708 case NVPTXISD::Suld1DArrayI64Clamp: return "NVPTXISD::Suld1DArrayI64Clamp";
709 case NVPTXISD::Suld1DArrayV2I8Clamp: return "NVPTXISD::Suld1DArrayV2I8Clamp";
710 case NVPTXISD::Suld1DArrayV2I16Clamp:return "NVPTXISD::Suld1DArrayV2I16Clamp";
711 case NVPTXISD::Suld1DArrayV2I32Clamp:return "NVPTXISD::Suld1DArrayV2I32Clamp";
712 case NVPTXISD::Suld1DArrayV2I64Clamp:return "NVPTXISD::Suld1DArrayV2I64Clamp";
713 case NVPTXISD::Suld1DArrayV4I8Clamp: return "NVPTXISD::Suld1DArrayV4I8Clamp";
714 case NVPTXISD::Suld1DArrayV4I16Clamp:return "NVPTXISD::Suld1DArrayV4I16Clamp";
715 case NVPTXISD::Suld1DArrayV4I32Clamp:return "NVPTXISD::Suld1DArrayV4I32Clamp";
716
717 case NVPTXISD::Suld2DI8Clamp: return "NVPTXISD::Suld2DI8Clamp";
718 case NVPTXISD::Suld2DI16Clamp: return "NVPTXISD::Suld2DI16Clamp";
719 case NVPTXISD::Suld2DI32Clamp: return "NVPTXISD::Suld2DI32Clamp";
720 case NVPTXISD::Suld2DI64Clamp: return "NVPTXISD::Suld2DI64Clamp";
721 case NVPTXISD::Suld2DV2I8Clamp: return "NVPTXISD::Suld2DV2I8Clamp";
722 case NVPTXISD::Suld2DV2I16Clamp: return "NVPTXISD::Suld2DV2I16Clamp";
723 case NVPTXISD::Suld2DV2I32Clamp: return "NVPTXISD::Suld2DV2I32Clamp";
724 case NVPTXISD::Suld2DV2I64Clamp: return "NVPTXISD::Suld2DV2I64Clamp";
725 case NVPTXISD::Suld2DV4I8Clamp: return "NVPTXISD::Suld2DV4I8Clamp";
726 case NVPTXISD::Suld2DV4I16Clamp: return "NVPTXISD::Suld2DV4I16Clamp";
727 case NVPTXISD::Suld2DV4I32Clamp: return "NVPTXISD::Suld2DV4I32Clamp";
728
729 case NVPTXISD::Suld2DArrayI8Clamp: return "NVPTXISD::Suld2DArrayI8Clamp";
730 case NVPTXISD::Suld2DArrayI16Clamp: return "NVPTXISD::Suld2DArrayI16Clamp";
731 case NVPTXISD::Suld2DArrayI32Clamp: return "NVPTXISD::Suld2DArrayI32Clamp";
732 case NVPTXISD::Suld2DArrayI64Clamp: return "NVPTXISD::Suld2DArrayI64Clamp";
733 case NVPTXISD::Suld2DArrayV2I8Clamp: return "NVPTXISD::Suld2DArrayV2I8Clamp";
734 case NVPTXISD::Suld2DArrayV2I16Clamp:return "NVPTXISD::Suld2DArrayV2I16Clamp";
735 case NVPTXISD::Suld2DArrayV2I32Clamp:return "NVPTXISD::Suld2DArrayV2I32Clamp";
736 case NVPTXISD::Suld2DArrayV2I64Clamp:return "NVPTXISD::Suld2DArrayV2I64Clamp";
737 case NVPTXISD::Suld2DArrayV4I8Clamp: return "NVPTXISD::Suld2DArrayV4I8Clamp";
738 case NVPTXISD::Suld2DArrayV4I16Clamp:return "NVPTXISD::Suld2DArrayV4I16Clamp";
739 case NVPTXISD::Suld2DArrayV4I32Clamp:return "NVPTXISD::Suld2DArrayV4I32Clamp";
740
741 case NVPTXISD::Suld3DI8Clamp: return "NVPTXISD::Suld3DI8Clamp";
742 case NVPTXISD::Suld3DI16Clamp: return "NVPTXISD::Suld3DI16Clamp";
743 case NVPTXISD::Suld3DI32Clamp: return "NVPTXISD::Suld3DI32Clamp";
744 case NVPTXISD::Suld3DI64Clamp: return "NVPTXISD::Suld3DI64Clamp";
745 case NVPTXISD::Suld3DV2I8Clamp: return "NVPTXISD::Suld3DV2I8Clamp";
746 case NVPTXISD::Suld3DV2I16Clamp: return "NVPTXISD::Suld3DV2I16Clamp";
747 case NVPTXISD::Suld3DV2I32Clamp: return "NVPTXISD::Suld3DV2I32Clamp";
748 case NVPTXISD::Suld3DV2I64Clamp: return "NVPTXISD::Suld3DV2I64Clamp";
749 case NVPTXISD::Suld3DV4I8Clamp: return "NVPTXISD::Suld3DV4I8Clamp";
750 case NVPTXISD::Suld3DV4I16Clamp: return "NVPTXISD::Suld3DV4I16Clamp";
751 case NVPTXISD::Suld3DV4I32Clamp: return "NVPTXISD::Suld3DV4I32Clamp";
Justin Holewinski30d56a72014-04-09 15:39:15 +0000752
753 case NVPTXISD::Suld1DI8Trap: return "NVPTXISD::Suld1DI8Trap";
754 case NVPTXISD::Suld1DI16Trap: return "NVPTXISD::Suld1DI16Trap";
755 case NVPTXISD::Suld1DI32Trap: return "NVPTXISD::Suld1DI32Trap";
Justin Holewinski9a2350e2014-07-17 11:59:04 +0000756 case NVPTXISD::Suld1DI64Trap: return "NVPTXISD::Suld1DI64Trap";
Justin Holewinski30d56a72014-04-09 15:39:15 +0000757 case NVPTXISD::Suld1DV2I8Trap: return "NVPTXISD::Suld1DV2I8Trap";
758 case NVPTXISD::Suld1DV2I16Trap: return "NVPTXISD::Suld1DV2I16Trap";
759 case NVPTXISD::Suld1DV2I32Trap: return "NVPTXISD::Suld1DV2I32Trap";
Justin Holewinski9a2350e2014-07-17 11:59:04 +0000760 case NVPTXISD::Suld1DV2I64Trap: return "NVPTXISD::Suld1DV2I64Trap";
Justin Holewinski30d56a72014-04-09 15:39:15 +0000761 case NVPTXISD::Suld1DV4I8Trap: return "NVPTXISD::Suld1DV4I8Trap";
762 case NVPTXISD::Suld1DV4I16Trap: return "NVPTXISD::Suld1DV4I16Trap";
763 case NVPTXISD::Suld1DV4I32Trap: return "NVPTXISD::Suld1DV4I32Trap";
764
765 case NVPTXISD::Suld1DArrayI8Trap: return "NVPTXISD::Suld1DArrayI8Trap";
766 case NVPTXISD::Suld1DArrayI16Trap: return "NVPTXISD::Suld1DArrayI16Trap";
767 case NVPTXISD::Suld1DArrayI32Trap: return "NVPTXISD::Suld1DArrayI32Trap";
Justin Holewinski9a2350e2014-07-17 11:59:04 +0000768 case NVPTXISD::Suld1DArrayI64Trap: return "NVPTXISD::Suld1DArrayI64Trap";
Justin Holewinski30d56a72014-04-09 15:39:15 +0000769 case NVPTXISD::Suld1DArrayV2I8Trap: return "NVPTXISD::Suld1DArrayV2I8Trap";
770 case NVPTXISD::Suld1DArrayV2I16Trap: return "NVPTXISD::Suld1DArrayV2I16Trap";
771 case NVPTXISD::Suld1DArrayV2I32Trap: return "NVPTXISD::Suld1DArrayV2I32Trap";
Justin Holewinski9a2350e2014-07-17 11:59:04 +0000772 case NVPTXISD::Suld1DArrayV2I64Trap: return "NVPTXISD::Suld1DArrayV2I64Trap";
Justin Holewinski30d56a72014-04-09 15:39:15 +0000773 case NVPTXISD::Suld1DArrayV4I8Trap: return "NVPTXISD::Suld1DArrayV4I8Trap";
774 case NVPTXISD::Suld1DArrayV4I16Trap: return "NVPTXISD::Suld1DArrayV4I16Trap";
775 case NVPTXISD::Suld1DArrayV4I32Trap: return "NVPTXISD::Suld1DArrayV4I32Trap";
776
777 case NVPTXISD::Suld2DI8Trap: return "NVPTXISD::Suld2DI8Trap";
778 case NVPTXISD::Suld2DI16Trap: return "NVPTXISD::Suld2DI16Trap";
779 case NVPTXISD::Suld2DI32Trap: return "NVPTXISD::Suld2DI32Trap";
Justin Holewinski9a2350e2014-07-17 11:59:04 +0000780 case NVPTXISD::Suld2DI64Trap: return "NVPTXISD::Suld2DI64Trap";
Justin Holewinski30d56a72014-04-09 15:39:15 +0000781 case NVPTXISD::Suld2DV2I8Trap: return "NVPTXISD::Suld2DV2I8Trap";
782 case NVPTXISD::Suld2DV2I16Trap: return "NVPTXISD::Suld2DV2I16Trap";
783 case NVPTXISD::Suld2DV2I32Trap: return "NVPTXISD::Suld2DV2I32Trap";
Justin Holewinski9a2350e2014-07-17 11:59:04 +0000784 case NVPTXISD::Suld2DV2I64Trap: return "NVPTXISD::Suld2DV2I64Trap";
Justin Holewinski30d56a72014-04-09 15:39:15 +0000785 case NVPTXISD::Suld2DV4I8Trap: return "NVPTXISD::Suld2DV4I8Trap";
786 case NVPTXISD::Suld2DV4I16Trap: return "NVPTXISD::Suld2DV4I16Trap";
787 case NVPTXISD::Suld2DV4I32Trap: return "NVPTXISD::Suld2DV4I32Trap";
788
789 case NVPTXISD::Suld2DArrayI8Trap: return "NVPTXISD::Suld2DArrayI8Trap";
790 case NVPTXISD::Suld2DArrayI16Trap: return "NVPTXISD::Suld2DArrayI16Trap";
791 case NVPTXISD::Suld2DArrayI32Trap: return "NVPTXISD::Suld2DArrayI32Trap";
Justin Holewinski9a2350e2014-07-17 11:59:04 +0000792 case NVPTXISD::Suld2DArrayI64Trap: return "NVPTXISD::Suld2DArrayI64Trap";
Justin Holewinski30d56a72014-04-09 15:39:15 +0000793 case NVPTXISD::Suld2DArrayV2I8Trap: return "NVPTXISD::Suld2DArrayV2I8Trap";
794 case NVPTXISD::Suld2DArrayV2I16Trap: return "NVPTXISD::Suld2DArrayV2I16Trap";
795 case NVPTXISD::Suld2DArrayV2I32Trap: return "NVPTXISD::Suld2DArrayV2I32Trap";
Justin Holewinski9a2350e2014-07-17 11:59:04 +0000796 case NVPTXISD::Suld2DArrayV2I64Trap: return "NVPTXISD::Suld2DArrayV2I64Trap";
Justin Holewinski30d56a72014-04-09 15:39:15 +0000797 case NVPTXISD::Suld2DArrayV4I8Trap: return "NVPTXISD::Suld2DArrayV4I8Trap";
798 case NVPTXISD::Suld2DArrayV4I16Trap: return "NVPTXISD::Suld2DArrayV4I16Trap";
799 case NVPTXISD::Suld2DArrayV4I32Trap: return "NVPTXISD::Suld2DArrayV4I32Trap";
800
801 case NVPTXISD::Suld3DI8Trap: return "NVPTXISD::Suld3DI8Trap";
802 case NVPTXISD::Suld3DI16Trap: return "NVPTXISD::Suld3DI16Trap";
803 case NVPTXISD::Suld3DI32Trap: return "NVPTXISD::Suld3DI32Trap";
Justin Holewinski9a2350e2014-07-17 11:59:04 +0000804 case NVPTXISD::Suld3DI64Trap: return "NVPTXISD::Suld3DI64Trap";
Justin Holewinski30d56a72014-04-09 15:39:15 +0000805 case NVPTXISD::Suld3DV2I8Trap: return "NVPTXISD::Suld3DV2I8Trap";
806 case NVPTXISD::Suld3DV2I16Trap: return "NVPTXISD::Suld3DV2I16Trap";
807 case NVPTXISD::Suld3DV2I32Trap: return "NVPTXISD::Suld3DV2I32Trap";
Justin Holewinski9a2350e2014-07-17 11:59:04 +0000808 case NVPTXISD::Suld3DV2I64Trap: return "NVPTXISD::Suld3DV2I64Trap";
Justin Holewinski30d56a72014-04-09 15:39:15 +0000809 case NVPTXISD::Suld3DV4I8Trap: return "NVPTXISD::Suld3DV4I8Trap";
810 case NVPTXISD::Suld3DV4I16Trap: return "NVPTXISD::Suld3DV4I16Trap";
811 case NVPTXISD::Suld3DV4I32Trap: return "NVPTXISD::Suld3DV4I32Trap";
Justin Holewinski9a2350e2014-07-17 11:59:04 +0000812
813 case NVPTXISD::Suld1DI8Zero: return "NVPTXISD::Suld1DI8Zero";
814 case NVPTXISD::Suld1DI16Zero: return "NVPTXISD::Suld1DI16Zero";
815 case NVPTXISD::Suld1DI32Zero: return "NVPTXISD::Suld1DI32Zero";
816 case NVPTXISD::Suld1DI64Zero: return "NVPTXISD::Suld1DI64Zero";
817 case NVPTXISD::Suld1DV2I8Zero: return "NVPTXISD::Suld1DV2I8Zero";
818 case NVPTXISD::Suld1DV2I16Zero: return "NVPTXISD::Suld1DV2I16Zero";
819 case NVPTXISD::Suld1DV2I32Zero: return "NVPTXISD::Suld1DV2I32Zero";
820 case NVPTXISD::Suld1DV2I64Zero: return "NVPTXISD::Suld1DV2I64Zero";
821 case NVPTXISD::Suld1DV4I8Zero: return "NVPTXISD::Suld1DV4I8Zero";
822 case NVPTXISD::Suld1DV4I16Zero: return "NVPTXISD::Suld1DV4I16Zero";
823 case NVPTXISD::Suld1DV4I32Zero: return "NVPTXISD::Suld1DV4I32Zero";
824
825 case NVPTXISD::Suld1DArrayI8Zero: return "NVPTXISD::Suld1DArrayI8Zero";
826 case NVPTXISD::Suld1DArrayI16Zero: return "NVPTXISD::Suld1DArrayI16Zero";
827 case NVPTXISD::Suld1DArrayI32Zero: return "NVPTXISD::Suld1DArrayI32Zero";
828 case NVPTXISD::Suld1DArrayI64Zero: return "NVPTXISD::Suld1DArrayI64Zero";
829 case NVPTXISD::Suld1DArrayV2I8Zero: return "NVPTXISD::Suld1DArrayV2I8Zero";
830 case NVPTXISD::Suld1DArrayV2I16Zero: return "NVPTXISD::Suld1DArrayV2I16Zero";
831 case NVPTXISD::Suld1DArrayV2I32Zero: return "NVPTXISD::Suld1DArrayV2I32Zero";
832 case NVPTXISD::Suld1DArrayV2I64Zero: return "NVPTXISD::Suld1DArrayV2I64Zero";
833 case NVPTXISD::Suld1DArrayV4I8Zero: return "NVPTXISD::Suld1DArrayV4I8Zero";
834 case NVPTXISD::Suld1DArrayV4I16Zero: return "NVPTXISD::Suld1DArrayV4I16Zero";
835 case NVPTXISD::Suld1DArrayV4I32Zero: return "NVPTXISD::Suld1DArrayV4I32Zero";
836
837 case NVPTXISD::Suld2DI8Zero: return "NVPTXISD::Suld2DI8Zero";
838 case NVPTXISD::Suld2DI16Zero: return "NVPTXISD::Suld2DI16Zero";
839 case NVPTXISD::Suld2DI32Zero: return "NVPTXISD::Suld2DI32Zero";
840 case NVPTXISD::Suld2DI64Zero: return "NVPTXISD::Suld2DI64Zero";
841 case NVPTXISD::Suld2DV2I8Zero: return "NVPTXISD::Suld2DV2I8Zero";
842 case NVPTXISD::Suld2DV2I16Zero: return "NVPTXISD::Suld2DV2I16Zero";
843 case NVPTXISD::Suld2DV2I32Zero: return "NVPTXISD::Suld2DV2I32Zero";
844 case NVPTXISD::Suld2DV2I64Zero: return "NVPTXISD::Suld2DV2I64Zero";
845 case NVPTXISD::Suld2DV4I8Zero: return "NVPTXISD::Suld2DV4I8Zero";
846 case NVPTXISD::Suld2DV4I16Zero: return "NVPTXISD::Suld2DV4I16Zero";
847 case NVPTXISD::Suld2DV4I32Zero: return "NVPTXISD::Suld2DV4I32Zero";
848
849 case NVPTXISD::Suld2DArrayI8Zero: return "NVPTXISD::Suld2DArrayI8Zero";
850 case NVPTXISD::Suld2DArrayI16Zero: return "NVPTXISD::Suld2DArrayI16Zero";
851 case NVPTXISD::Suld2DArrayI32Zero: return "NVPTXISD::Suld2DArrayI32Zero";
852 case NVPTXISD::Suld2DArrayI64Zero: return "NVPTXISD::Suld2DArrayI64Zero";
853 case NVPTXISD::Suld2DArrayV2I8Zero: return "NVPTXISD::Suld2DArrayV2I8Zero";
854 case NVPTXISD::Suld2DArrayV2I16Zero: return "NVPTXISD::Suld2DArrayV2I16Zero";
855 case NVPTXISD::Suld2DArrayV2I32Zero: return "NVPTXISD::Suld2DArrayV2I32Zero";
856 case NVPTXISD::Suld2DArrayV2I64Zero: return "NVPTXISD::Suld2DArrayV2I64Zero";
857 case NVPTXISD::Suld2DArrayV4I8Zero: return "NVPTXISD::Suld2DArrayV4I8Zero";
858 case NVPTXISD::Suld2DArrayV4I16Zero: return "NVPTXISD::Suld2DArrayV4I16Zero";
859 case NVPTXISD::Suld2DArrayV4I32Zero: return "NVPTXISD::Suld2DArrayV4I32Zero";
860
861 case NVPTXISD::Suld3DI8Zero: return "NVPTXISD::Suld3DI8Zero";
862 case NVPTXISD::Suld3DI16Zero: return "NVPTXISD::Suld3DI16Zero";
863 case NVPTXISD::Suld3DI32Zero: return "NVPTXISD::Suld3DI32Zero";
864 case NVPTXISD::Suld3DI64Zero: return "NVPTXISD::Suld3DI64Zero";
865 case NVPTXISD::Suld3DV2I8Zero: return "NVPTXISD::Suld3DV2I8Zero";
866 case NVPTXISD::Suld3DV2I16Zero: return "NVPTXISD::Suld3DV2I16Zero";
867 case NVPTXISD::Suld3DV2I32Zero: return "NVPTXISD::Suld3DV2I32Zero";
868 case NVPTXISD::Suld3DV2I64Zero: return "NVPTXISD::Suld3DV2I64Zero";
869 case NVPTXISD::Suld3DV4I8Zero: return "NVPTXISD::Suld3DV4I8Zero";
870 case NVPTXISD::Suld3DV4I16Zero: return "NVPTXISD::Suld3DV4I16Zero";
871 case NVPTXISD::Suld3DV4I32Zero: return "NVPTXISD::Suld3DV4I32Zero";
Justin Holewinskiae556d32012-05-04 20:18:50 +0000872 }
Matthias Braund04893f2015-05-07 21:33:59 +0000873 return nullptr;
Justin Holewinskiae556d32012-05-04 20:18:50 +0000874}
875
Chandler Carruth9d010ff2014-07-03 00:23:43 +0000876TargetLoweringBase::LegalizeTypeAction
877NVPTXTargetLowering::getPreferredVectorAction(EVT VT) const {
878 if (VT.getVectorNumElements() != 1 && VT.getScalarType() == MVT::i1)
879 return TypeSplitVector;
880
881 return TargetLoweringBase::getPreferredVectorAction(VT);
Justin Holewinskibc451192012-11-29 14:26:24 +0000882}
Justin Holewinskiae556d32012-05-04 20:18:50 +0000883
884SDValue
885NVPTXTargetLowering::LowerGlobalAddress(SDValue Op, SelectionDAG &DAG) const {
Andrew Trickef9de2a2013-05-25 02:42:55 +0000886 SDLoc dl(Op);
Justin Holewinskiae556d32012-05-04 20:18:50 +0000887 const GlobalValue *GV = cast<GlobalAddressSDNode>(Op)->getGlobal();
Mehdi Amini44ede332015-07-09 02:09:04 +0000888 auto PtrVT = getPointerTy(DAG.getDataLayout());
889 Op = DAG.getTargetGlobalAddress(GV, dl, PtrVT);
890 return DAG.getNode(NVPTXISD::Wrapper, dl, PtrVT, Op);
Justin Holewinskiae556d32012-05-04 20:18:50 +0000891}
892
Justin Holewinskif8f70912013-06-28 17:57:59 +0000893std::string
894NVPTXTargetLowering::getPrototype(Type *retTy, const ArgListTy &Args,
895 const SmallVectorImpl<ISD::OutputArg> &Outs,
896 unsigned retAlignment,
897 const ImmutableCallSite *CS) const {
Mehdi Amini44ede332015-07-09 02:09:04 +0000898 auto PtrVT = getPointerTy(*getDataLayout());
Eric Christopherbef0a372015-01-30 01:50:07 +0000899 bool isABI = (STI.getSmVersion() >= 20);
Justin Holewinskif8f70912013-06-28 17:57:59 +0000900 assert(isABI && "Non-ABI compilation is not supported");
901 if (!isABI)
902 return "";
903
904 std::stringstream O;
905 O << "prototype_" << uniqueCallSite << " : .callprototype ";
906
907 if (retTy->getTypeID() == Type::VoidTyID) {
908 O << "()";
909 } else {
910 O << "(";
Rafael Espindola08013342013-12-07 19:34:20 +0000911 if (retTy->isFloatingPointTy() || retTy->isIntegerTy()) {
Justin Holewinskif8f70912013-06-28 17:57:59 +0000912 unsigned size = 0;
913 if (const IntegerType *ITy = dyn_cast<IntegerType>(retTy)) {
914 size = ITy->getBitWidth();
915 if (size < 32)
916 size = 32;
917 } else {
918 assert(retTy->isFloatingPointTy() &&
919 "Floating point type expected here");
920 size = retTy->getPrimitiveSizeInBits();
921 }
922
923 O << ".param .b" << size << " _";
924 } else if (isa<PointerType>(retTy)) {
Mehdi Amini44ede332015-07-09 02:09:04 +0000925 O << ".param .b" << PtrVT.getSizeInBits() << " _";
Craig Topperd3c02f12015-01-05 10:15:49 +0000926 } else if ((retTy->getTypeID() == Type::StructTyID) ||
927 isa<VectorType>(retTy)) {
928 O << ".param .align "
929 << retAlignment
930 << " .b8 _["
931 << getDataLayout()->getTypeAllocSize(retTy) << "]";
Justin Holewinskif8f70912013-06-28 17:57:59 +0000932 } else {
Craig Topperd3c02f12015-01-05 10:15:49 +0000933 llvm_unreachable("Unknown return type");
Justin Holewinskif8f70912013-06-28 17:57:59 +0000934 }
935 O << ") ";
936 }
937 O << "_ (";
938
939 bool first = true;
Justin Holewinskif8f70912013-06-28 17:57:59 +0000940
941 unsigned OIdx = 0;
942 for (unsigned i = 0, e = Args.size(); i != e; ++i, ++OIdx) {
943 Type *Ty = Args[i].Ty;
944 if (!first) {
945 O << ", ";
946 }
947 first = false;
948
Eli Bendersky3e840192015-03-23 16:26:23 +0000949 if (!Outs[OIdx].Flags.isByVal()) {
Mehdi Amini44ede332015-07-09 02:09:04 +0000950 const DataLayout *TD = getDataLayout();
Justin Holewinskif8f70912013-06-28 17:57:59 +0000951 if (Ty->isAggregateType() || Ty->isVectorTy()) {
952 unsigned align = 0;
953 const CallInst *CallI = cast<CallInst>(CS->getInstruction());
Justin Holewinskif8f70912013-06-28 17:57:59 +0000954 // +1 because index 0 is reserved for return type alignment
955 if (!llvm::getAlign(*CallI, i + 1, align))
956 align = TD->getABITypeAlignment(Ty);
957 unsigned sz = TD->getTypeAllocSize(Ty);
958 O << ".param .align " << align << " .b8 ";
959 O << "_";
960 O << "[" << sz << "]";
961 // update the index for Outs
962 SmallVector<EVT, 16> vtparts;
Mehdi Amini56228da2015-07-09 01:57:34 +0000963 ComputeValueVTs(*this, *TD, Ty, vtparts);
Justin Holewinskif8f70912013-06-28 17:57:59 +0000964 if (unsigned len = vtparts.size())
965 OIdx += len - 1;
966 continue;
967 }
Justin Holewinskidff28d22013-07-01 12:59:01 +0000968 // i8 types in IR will be i16 types in SDAG
Mehdi Amini44ede332015-07-09 02:09:04 +0000969 assert(
970 (getValueType(*TD, Ty) == Outs[OIdx].VT ||
971 (getValueType(*TD, Ty) == MVT::i8 && Outs[OIdx].VT == MVT::i16)) &&
972 "type mismatch between callee prototype and arguments");
Justin Holewinskif8f70912013-06-28 17:57:59 +0000973 // scalar type
974 unsigned sz = 0;
975 if (isa<IntegerType>(Ty)) {
976 sz = cast<IntegerType>(Ty)->getBitWidth();
977 if (sz < 32)
978 sz = 32;
979 } else if (isa<PointerType>(Ty))
Mehdi Amini44ede332015-07-09 02:09:04 +0000980 sz = PtrVT.getSizeInBits();
Justin Holewinskif8f70912013-06-28 17:57:59 +0000981 else
982 sz = Ty->getPrimitiveSizeInBits();
983 O << ".param .b" << sz << " ";
984 O << "_";
985 continue;
986 }
987 const PointerType *PTy = dyn_cast<PointerType>(Ty);
988 assert(PTy && "Param with byval attribute should be a pointer type");
989 Type *ETy = PTy->getElementType();
990
991 unsigned align = Outs[OIdx].Flags.getByValAlign();
992 unsigned sz = getDataLayout()->getTypeAllocSize(ETy);
993 O << ".param .align " << align << " .b8 ";
994 O << "_";
995 O << "[" << sz << "]";
996 }
997 O << ");";
998 return O.str();
999}
1000
1001unsigned
1002NVPTXTargetLowering::getArgumentAlignment(SDValue Callee,
1003 const ImmutableCallSite *CS,
1004 Type *Ty,
1005 unsigned Idx) const {
1006 const DataLayout *TD = getDataLayout();
Justin Holewinski124e93d2013-11-11 19:28:19 +00001007 unsigned Align = 0;
1008 const Value *DirectCallee = CS->getCalledFunction();
Justin Holewinskif8f70912013-06-28 17:57:59 +00001009
Justin Holewinski124e93d2013-11-11 19:28:19 +00001010 if (!DirectCallee) {
1011 // We don't have a direct function symbol, but that may be because of
1012 // constant cast instructions in the call.
1013 const Instruction *CalleeI = CS->getInstruction();
1014 assert(CalleeI && "Call target is not a function or derived value?");
1015
1016 // With bitcast'd call targets, the instruction will be the call
1017 if (isa<CallInst>(CalleeI)) {
1018 // Check if we have call alignment metadata
1019 if (llvm::getAlign(*cast<CallInst>(CalleeI), Idx, Align))
1020 return Align;
1021
1022 const Value *CalleeV = cast<CallInst>(CalleeI)->getCalledValue();
1023 // Ignore any bitcast instructions
1024 while(isa<ConstantExpr>(CalleeV)) {
1025 const ConstantExpr *CE = cast<ConstantExpr>(CalleeV);
1026 if (!CE->isCast())
1027 break;
1028 // Look through the bitcast
1029 CalleeV = cast<ConstantExpr>(CalleeV)->getOperand(0);
1030 }
1031
1032 // We have now looked past all of the bitcasts. Do we finally have a
1033 // Function?
1034 if (isa<Function>(CalleeV))
1035 DirectCallee = CalleeV;
1036 }
Justin Holewinskif8f70912013-06-28 17:57:59 +00001037 }
1038
Justin Holewinski124e93d2013-11-11 19:28:19 +00001039 // Check for function alignment information if we found that the
1040 // ultimate target is a Function
1041 if (DirectCallee)
1042 if (llvm::getAlign(*cast<Function>(DirectCallee), Idx, Align))
1043 return Align;
1044
1045 // Call is indirect or alignment information is not available, fall back to
1046 // the ABI type alignment
1047 return TD->getABITypeAlignment(Ty);
Justin Holewinskiae556d32012-05-04 20:18:50 +00001048}
1049
Justin Holewinski0497ab12013-03-30 14:29:21 +00001050SDValue NVPTXTargetLowering::LowerCall(TargetLowering::CallLoweringInfo &CLI,
1051 SmallVectorImpl<SDValue> &InVals) const {
1052 SelectionDAG &DAG = CLI.DAG;
Andrew Trickef9de2a2013-05-25 02:42:55 +00001053 SDLoc dl = CLI.DL;
Craig Topperb94011f2013-07-14 04:42:23 +00001054 SmallVectorImpl<ISD::OutputArg> &Outs = CLI.Outs;
1055 SmallVectorImpl<SDValue> &OutVals = CLI.OutVals;
1056 SmallVectorImpl<ISD::InputArg> &Ins = CLI.Ins;
Justin Holewinski0497ab12013-03-30 14:29:21 +00001057 SDValue Chain = CLI.Chain;
1058 SDValue Callee = CLI.Callee;
1059 bool &isTailCall = CLI.IsTailCall;
Saleem Abdulrasool9f664c12014-05-17 21:50:01 +00001060 ArgListTy &Args = CLI.getArgs();
Justin Holewinski0497ab12013-03-30 14:29:21 +00001061 Type *retTy = CLI.RetTy;
1062 ImmutableCallSite *CS = CLI.CS;
Justin Holewinskiaa583972012-05-25 16:35:28 +00001063
Eric Christopherbef0a372015-01-30 01:50:07 +00001064 bool isABI = (STI.getSmVersion() >= 20);
Justin Holewinskif8f70912013-06-28 17:57:59 +00001065 assert(isABI && "Non-ABI compilation is not supported");
1066 if (!isABI)
1067 return Chain;
Justin Holewinskif8f70912013-06-28 17:57:59 +00001068 MachineFunction &MF = DAG.getMachineFunction();
1069 const Function *F = MF.getFunction();
Mehdi Amini56228da2015-07-09 01:57:34 +00001070 auto &DL = MF.getDataLayout();
Justin Holewinskiae556d32012-05-04 20:18:50 +00001071
1072 SDValue tempChain = Chain;
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +00001073 Chain = DAG.getCALLSEQ_START(Chain,
1074 DAG.getIntPtrConstant(uniqueCallSite, dl, true),
1075 dl);
Justin Holewinskiae556d32012-05-04 20:18:50 +00001076 SDValue InFlag = Chain.getValue(1);
1077
Justin Holewinskiae556d32012-05-04 20:18:50 +00001078 unsigned paramCount = 0;
Justin Holewinskif8f70912013-06-28 17:57:59 +00001079 // Args.size() and Outs.size() need not match.
1080 // Outs.size() will be larger
1081 // * if there is an aggregate argument with multiple fields (each field
1082 // showing up separately in Outs)
1083 // * if there is a vector argument with more than typical vector-length
1084 // elements (generally if more than 4) where each vector element is
1085 // individually present in Outs.
1086 // So a different index should be used for indexing into Outs/OutVals.
1087 // See similar issue in LowerFormalArguments.
1088 unsigned OIdx = 0;
Justin Holewinskiae556d32012-05-04 20:18:50 +00001089 // Declare the .params or .reg need to pass values
1090 // to the function
Justin Holewinskif8f70912013-06-28 17:57:59 +00001091 for (unsigned i = 0, e = Args.size(); i != e; ++i, ++OIdx) {
1092 EVT VT = Outs[OIdx].VT;
1093 Type *Ty = Args[i].Ty;
Justin Holewinskiae556d32012-05-04 20:18:50 +00001094
Eli Bendersky3e840192015-03-23 16:26:23 +00001095 if (!Outs[OIdx].Flags.isByVal()) {
Justin Holewinskif8f70912013-06-28 17:57:59 +00001096 if (Ty->isAggregateType()) {
1097 // aggregate
1098 SmallVector<EVT, 16> vtparts;
Justin Holewinski6e40f632014-06-27 18:35:44 +00001099 SmallVector<uint64_t, 16> Offsets;
Mehdi Amini56228da2015-07-09 01:57:34 +00001100 ComputePTXValueVTs(*this, DL, Ty, vtparts, &Offsets, 0);
Justin Holewinskif8f70912013-06-28 17:57:59 +00001101
1102 unsigned align = getArgumentAlignment(Callee, CS, Ty, paramCount + 1);
1103 // declare .param .align <align> .b8 .param<n>[<size>];
Mehdi Amini56228da2015-07-09 01:57:34 +00001104 unsigned sz = DL.getTypeAllocSize(Ty);
Justin Holewinskif8f70912013-06-28 17:57:59 +00001105 SDVTList DeclareParamVTs = DAG.getVTList(MVT::Other, MVT::Glue);
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +00001106 SDValue DeclareParamOps[] = { Chain, DAG.getConstant(align, dl,
1107 MVT::i32),
1108 DAG.getConstant(paramCount, dl, MVT::i32),
1109 DAG.getConstant(sz, dl, MVT::i32),
1110 InFlag };
Justin Holewinskif8f70912013-06-28 17:57:59 +00001111 Chain = DAG.getNode(NVPTXISD::DeclareParam, dl, DeclareParamVTs,
Craig Topper48d114b2014-04-26 18:35:24 +00001112 DeclareParamOps);
Justin Holewinskif8f70912013-06-28 17:57:59 +00001113 InFlag = Chain.getValue(1);
Justin Holewinskif8f70912013-06-28 17:57:59 +00001114 for (unsigned j = 0, je = vtparts.size(); j != je; ++j) {
Justin Holewinskif8f70912013-06-28 17:57:59 +00001115 EVT elemtype = vtparts[j];
Justin Holewinski9982f062014-06-27 19:36:25 +00001116 unsigned ArgAlign = GreatestCommonDivisor64(align, Offsets[j]);
Justin Holewinski6e40f632014-06-27 18:35:44 +00001117 if (elemtype.isInteger() && (sz < 8))
1118 sz = 8;
1119 SDValue StVal = OutVals[OIdx];
1120 if (elemtype.getSizeInBits() < 16) {
1121 StVal = DAG.getNode(ISD::ANY_EXTEND, dl, MVT::i16, StVal);
Justin Holewinskif8f70912013-06-28 17:57:59 +00001122 }
Justin Holewinski6e40f632014-06-27 18:35:44 +00001123 SDVTList CopyParamVTs = DAG.getVTList(MVT::Other, MVT::Glue);
1124 SDValue CopyParamOps[] = { Chain,
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +00001125 DAG.getConstant(paramCount, dl, MVT::i32),
1126 DAG.getConstant(Offsets[j], dl, MVT::i32),
Justin Holewinski6e40f632014-06-27 18:35:44 +00001127 StVal, InFlag };
1128 Chain = DAG.getMemIntrinsicNode(NVPTXISD::StoreParam, dl,
1129 CopyParamVTs, CopyParamOps,
1130 elemtype, MachinePointerInfo(),
1131 ArgAlign);
1132 InFlag = Chain.getValue(1);
1133 ++OIdx;
Justin Holewinskif8f70912013-06-28 17:57:59 +00001134 }
1135 if (vtparts.size() > 0)
1136 --OIdx;
1137 ++paramCount;
1138 continue;
1139 }
1140 if (Ty->isVectorTy()) {
Mehdi Amini44ede332015-07-09 02:09:04 +00001141 EVT ObjectVT = getValueType(DL, Ty);
Justin Holewinskif8f70912013-06-28 17:57:59 +00001142 unsigned align = getArgumentAlignment(Callee, CS, Ty, paramCount + 1);
1143 // declare .param .align <align> .b8 .param<n>[<size>];
Mehdi Amini56228da2015-07-09 01:57:34 +00001144 unsigned sz = DL.getTypeAllocSize(Ty);
Justin Holewinskif8f70912013-06-28 17:57:59 +00001145 SDVTList DeclareParamVTs = DAG.getVTList(MVT::Other, MVT::Glue);
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +00001146 SDValue DeclareParamOps[] = { Chain,
1147 DAG.getConstant(align, dl, MVT::i32),
1148 DAG.getConstant(paramCount, dl, MVT::i32),
1149 DAG.getConstant(sz, dl, MVT::i32),
1150 InFlag };
Justin Holewinskif8f70912013-06-28 17:57:59 +00001151 Chain = DAG.getNode(NVPTXISD::DeclareParam, dl, DeclareParamVTs,
Craig Topper48d114b2014-04-26 18:35:24 +00001152 DeclareParamOps);
Justin Holewinskif8f70912013-06-28 17:57:59 +00001153 InFlag = Chain.getValue(1);
1154 unsigned NumElts = ObjectVT.getVectorNumElements();
1155 EVT EltVT = ObjectVT.getVectorElementType();
1156 EVT MemVT = EltVT;
1157 bool NeedExtend = false;
1158 if (EltVT.getSizeInBits() < 16) {
1159 NeedExtend = true;
1160 EltVT = MVT::i16;
1161 }
1162
1163 // V1 store
1164 if (NumElts == 1) {
1165 SDValue Elt = OutVals[OIdx++];
1166 if (NeedExtend)
1167 Elt = DAG.getNode(ISD::ZERO_EXTEND, dl, MVT::i16, Elt);
1168
1169 SDVTList CopyParamVTs = DAG.getVTList(MVT::Other, MVT::Glue);
1170 SDValue CopyParamOps[] = { Chain,
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +00001171 DAG.getConstant(paramCount, dl, MVT::i32),
1172 DAG.getConstant(0, dl, MVT::i32), Elt,
Justin Holewinskif8f70912013-06-28 17:57:59 +00001173 InFlag };
1174 Chain = DAG.getMemIntrinsicNode(NVPTXISD::StoreParam, dl,
Craig Topper206fcd42014-04-26 19:29:41 +00001175 CopyParamVTs, CopyParamOps,
Justin Holewinskif8f70912013-06-28 17:57:59 +00001176 MemVT, MachinePointerInfo());
1177 InFlag = Chain.getValue(1);
1178 } else if (NumElts == 2) {
1179 SDValue Elt0 = OutVals[OIdx++];
1180 SDValue Elt1 = OutVals[OIdx++];
1181 if (NeedExtend) {
1182 Elt0 = DAG.getNode(ISD::ZERO_EXTEND, dl, MVT::i16, Elt0);
1183 Elt1 = DAG.getNode(ISD::ZERO_EXTEND, dl, MVT::i16, Elt1);
1184 }
1185
1186 SDVTList CopyParamVTs = DAG.getVTList(MVT::Other, MVT::Glue);
1187 SDValue CopyParamOps[] = { Chain,
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +00001188 DAG.getConstant(paramCount, dl, MVT::i32),
1189 DAG.getConstant(0, dl, MVT::i32), Elt0,
1190 Elt1, InFlag };
Justin Holewinskif8f70912013-06-28 17:57:59 +00001191 Chain = DAG.getMemIntrinsicNode(NVPTXISD::StoreParamV2, dl,
Craig Topper206fcd42014-04-26 19:29:41 +00001192 CopyParamVTs, CopyParamOps,
Justin Holewinskif8f70912013-06-28 17:57:59 +00001193 MemVT, MachinePointerInfo());
1194 InFlag = Chain.getValue(1);
1195 } else {
1196 unsigned curOffset = 0;
1197 // V4 stores
1198 // We have at least 4 elements (<3 x Ty> expands to 4 elements) and
1199 // the
1200 // vector will be expanded to a power of 2 elements, so we know we can
1201 // always round up to the next multiple of 4 when creating the vector
1202 // stores.
1203 // e.g. 4 elem => 1 st.v4
1204 // 6 elem => 2 st.v4
1205 // 8 elem => 2 st.v4
1206 // 11 elem => 3 st.v4
1207 unsigned VecSize = 4;
1208 if (EltVT.getSizeInBits() == 64)
1209 VecSize = 2;
1210
1211 // This is potentially only part of a vector, so assume all elements
1212 // are packed together.
1213 unsigned PerStoreOffset = MemVT.getStoreSizeInBits() / 8 * VecSize;
1214
1215 for (unsigned i = 0; i < NumElts; i += VecSize) {
1216 // Get values
1217 SDValue StoreVal;
1218 SmallVector<SDValue, 8> Ops;
1219 Ops.push_back(Chain);
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +00001220 Ops.push_back(DAG.getConstant(paramCount, dl, MVT::i32));
1221 Ops.push_back(DAG.getConstant(curOffset, dl, MVT::i32));
Justin Holewinskif8f70912013-06-28 17:57:59 +00001222
1223 unsigned Opc = NVPTXISD::StoreParamV2;
1224
1225 StoreVal = OutVals[OIdx++];
1226 if (NeedExtend)
1227 StoreVal = DAG.getNode(ISD::ZERO_EXTEND, dl, MVT::i16, StoreVal);
1228 Ops.push_back(StoreVal);
1229
1230 if (i + 1 < NumElts) {
1231 StoreVal = OutVals[OIdx++];
1232 if (NeedExtend)
1233 StoreVal =
1234 DAG.getNode(ISD::ZERO_EXTEND, dl, MVT::i16, StoreVal);
1235 } else {
1236 StoreVal = DAG.getUNDEF(EltVT);
1237 }
1238 Ops.push_back(StoreVal);
1239
1240 if (VecSize == 4) {
1241 Opc = NVPTXISD::StoreParamV4;
1242 if (i + 2 < NumElts) {
1243 StoreVal = OutVals[OIdx++];
1244 if (NeedExtend)
1245 StoreVal =
1246 DAG.getNode(ISD::ZERO_EXTEND, dl, MVT::i16, StoreVal);
1247 } else {
1248 StoreVal = DAG.getUNDEF(EltVT);
1249 }
1250 Ops.push_back(StoreVal);
1251
1252 if (i + 3 < NumElts) {
1253 StoreVal = OutVals[OIdx++];
1254 if (NeedExtend)
1255 StoreVal =
1256 DAG.getNode(ISD::ZERO_EXTEND, dl, MVT::i16, StoreVal);
1257 } else {
1258 StoreVal = DAG.getUNDEF(EltVT);
1259 }
1260 Ops.push_back(StoreVal);
1261 }
1262
Justin Holewinskidff28d22013-07-01 12:59:01 +00001263 Ops.push_back(InFlag);
1264
Justin Holewinskif8f70912013-06-28 17:57:59 +00001265 SDVTList CopyParamVTs = DAG.getVTList(MVT::Other, MVT::Glue);
Craig Topper206fcd42014-04-26 19:29:41 +00001266 Chain = DAG.getMemIntrinsicNode(Opc, dl, CopyParamVTs, Ops,
1267 MemVT, MachinePointerInfo());
Justin Holewinskif8f70912013-06-28 17:57:59 +00001268 InFlag = Chain.getValue(1);
1269 curOffset += PerStoreOffset;
1270 }
1271 }
1272 ++paramCount;
1273 --OIdx;
1274 continue;
1275 }
Justin Holewinskiae556d32012-05-04 20:18:50 +00001276 // Plain scalar
1277 // for ABI, declare .param .b<size> .param<n>;
Justin Holewinskiae556d32012-05-04 20:18:50 +00001278 unsigned sz = VT.getSizeInBits();
Justin Holewinskif8f70912013-06-28 17:57:59 +00001279 bool needExtend = false;
1280 if (VT.isInteger()) {
1281 if (sz < 16)
1282 needExtend = true;
1283 if (sz < 32)
1284 sz = 32;
1285 }
Justin Holewinskiae556d32012-05-04 20:18:50 +00001286 SDVTList DeclareParamVTs = DAG.getVTList(MVT::Other, MVT::Glue);
1287 SDValue DeclareParamOps[] = { Chain,
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +00001288 DAG.getConstant(paramCount, dl, MVT::i32),
1289 DAG.getConstant(sz, dl, MVT::i32),
1290 DAG.getConstant(0, dl, MVT::i32), InFlag };
Justin Holewinskiae556d32012-05-04 20:18:50 +00001291 Chain = DAG.getNode(NVPTXISD::DeclareScalarParam, dl, DeclareParamVTs,
Craig Topper48d114b2014-04-26 18:35:24 +00001292 DeclareParamOps);
Justin Holewinskiae556d32012-05-04 20:18:50 +00001293 InFlag = Chain.getValue(1);
Justin Holewinskif8f70912013-06-28 17:57:59 +00001294 SDValue OutV = OutVals[OIdx];
1295 if (needExtend) {
1296 // zext/sext i1 to i16
1297 unsigned opc = ISD::ZERO_EXTEND;
1298 if (Outs[OIdx].Flags.isSExt())
1299 opc = ISD::SIGN_EXTEND;
1300 OutV = DAG.getNode(opc, dl, MVT::i16, OutV);
1301 }
Justin Holewinskiae556d32012-05-04 20:18:50 +00001302 SDVTList CopyParamVTs = DAG.getVTList(MVT::Other, MVT::Glue);
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +00001303 SDValue CopyParamOps[] = { Chain,
1304 DAG.getConstant(paramCount, dl, MVT::i32),
1305 DAG.getConstant(0, dl, MVT::i32), OutV,
1306 InFlag };
Justin Holewinskiae556d32012-05-04 20:18:50 +00001307
1308 unsigned opcode = NVPTXISD::StoreParam;
Justin Holewinskif8f70912013-06-28 17:57:59 +00001309 if (Outs[OIdx].Flags.isZExt())
1310 opcode = NVPTXISD::StoreParamU32;
1311 else if (Outs[OIdx].Flags.isSExt())
1312 opcode = NVPTXISD::StoreParamS32;
Craig Topper206fcd42014-04-26 19:29:41 +00001313 Chain = DAG.getMemIntrinsicNode(opcode, dl, CopyParamVTs, CopyParamOps,
Justin Holewinskif8f70912013-06-28 17:57:59 +00001314 VT, MachinePointerInfo());
Justin Holewinskiae556d32012-05-04 20:18:50 +00001315
1316 InFlag = Chain.getValue(1);
1317 ++paramCount;
1318 continue;
1319 }
1320 // struct or vector
1321 SmallVector<EVT, 16> vtparts;
Justin Holewinski6e40f632014-06-27 18:35:44 +00001322 SmallVector<uint64_t, 16> Offsets;
Justin Holewinskiae556d32012-05-04 20:18:50 +00001323 const PointerType *PTy = dyn_cast<PointerType>(Args[i].Ty);
Justin Holewinski0497ab12013-03-30 14:29:21 +00001324 assert(PTy && "Type of a byval parameter should be pointer");
Mehdi Amini56228da2015-07-09 01:57:34 +00001325 ComputePTXValueVTs(*this, DL, PTy->getElementType(), vtparts, &Offsets, 0);
Justin Holewinskiae556d32012-05-04 20:18:50 +00001326
Justin Holewinskif8f70912013-06-28 17:57:59 +00001327 // declare .param .align <align> .b8 .param<n>[<size>];
1328 unsigned sz = Outs[OIdx].Flags.getByValSize();
1329 SDVTList DeclareParamVTs = DAG.getVTList(MVT::Other, MVT::Glue);
Justin Holewinski6e40f632014-06-27 18:35:44 +00001330 unsigned ArgAlign = Outs[OIdx].Flags.getByValAlign();
Justin Holewinskif8f70912013-06-28 17:57:59 +00001331 // The ByValAlign in the Outs[OIdx].Flags is alway set at this point,
1332 // so we don't need to worry about natural alignment or not.
1333 // See TargetLowering::LowerCallTo().
1334 SDValue DeclareParamOps[] = {
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +00001335 Chain, DAG.getConstant(Outs[OIdx].Flags.getByValAlign(), dl, MVT::i32),
1336 DAG.getConstant(paramCount, dl, MVT::i32),
1337 DAG.getConstant(sz, dl, MVT::i32), InFlag
Justin Holewinskif8f70912013-06-28 17:57:59 +00001338 };
1339 Chain = DAG.getNode(NVPTXISD::DeclareParam, dl, DeclareParamVTs,
Craig Topper48d114b2014-04-26 18:35:24 +00001340 DeclareParamOps);
Justin Holewinskif8f70912013-06-28 17:57:59 +00001341 InFlag = Chain.getValue(1);
Justin Holewinski0497ab12013-03-30 14:29:21 +00001342 for (unsigned j = 0, je = vtparts.size(); j != je; ++j) {
Justin Holewinskiae556d32012-05-04 20:18:50 +00001343 EVT elemtype = vtparts[j];
Justin Holewinski6e40f632014-06-27 18:35:44 +00001344 int curOffset = Offsets[j];
Justin Holewinski9982f062014-06-27 19:36:25 +00001345 unsigned PartAlign = GreatestCommonDivisor64(ArgAlign, curOffset);
Mehdi Amini44ede332015-07-09 02:09:04 +00001346 auto PtrVT = getPointerTy(DAG.getDataLayout());
1347 SDValue srcAddr = DAG.getNode(ISD::ADD, dl, PtrVT, OutVals[OIdx],
1348 DAG.getConstant(curOffset, dl, PtrVT));
Justin Holewinski6e40f632014-06-27 18:35:44 +00001349 SDValue theVal = DAG.getLoad(elemtype, dl, tempChain, srcAddr,
1350 MachinePointerInfo(), false, false, false,
1351 PartAlign);
1352 if (elemtype.getSizeInBits() < 16) {
1353 theVal = DAG.getNode(ISD::ANY_EXTEND, dl, MVT::i16, theVal);
Justin Holewinskiae556d32012-05-04 20:18:50 +00001354 }
Justin Holewinski6e40f632014-06-27 18:35:44 +00001355 SDVTList CopyParamVTs = DAG.getVTList(MVT::Other, MVT::Glue);
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +00001356 SDValue CopyParamOps[] = { Chain,
1357 DAG.getConstant(paramCount, dl, MVT::i32),
1358 DAG.getConstant(curOffset, dl, MVT::i32),
1359 theVal, InFlag };
Justin Holewinski6e40f632014-06-27 18:35:44 +00001360 Chain = DAG.getMemIntrinsicNode(NVPTXISD::StoreParam, dl, CopyParamVTs,
1361 CopyParamOps, elemtype,
1362 MachinePointerInfo());
Justin Holewinskif8f70912013-06-28 17:57:59 +00001363
Justin Holewinski6e40f632014-06-27 18:35:44 +00001364 InFlag = Chain.getValue(1);
Justin Holewinskiae556d32012-05-04 20:18:50 +00001365 }
Justin Holewinskif8f70912013-06-28 17:57:59 +00001366 ++paramCount;
Justin Holewinskiae556d32012-05-04 20:18:50 +00001367 }
1368
1369 GlobalAddressSDNode *Func = dyn_cast<GlobalAddressSDNode>(Callee.getNode());
1370 unsigned retAlignment = 0;
1371
1372 // Handle Result
Justin Holewinskiae556d32012-05-04 20:18:50 +00001373 if (Ins.size() > 0) {
1374 SmallVector<EVT, 16> resvtparts;
Mehdi Amini56228da2015-07-09 01:57:34 +00001375 ComputeValueVTs(*this, DL, retTy, resvtparts);
Justin Holewinskiae556d32012-05-04 20:18:50 +00001376
Justin Holewinskif8f70912013-06-28 17:57:59 +00001377 // Declare
1378 // .param .align 16 .b8 retval0[<size-in-bytes>], or
1379 // .param .b<size-in-bits> retval0
Mehdi Amini56228da2015-07-09 01:57:34 +00001380 unsigned resultsz = DL.getTypeAllocSizeInBits(retTy);
Jingyue Wuea511612014-10-25 03:46:16 +00001381 // Emit ".param .b<size-in-bits> retval0" instead of byte arrays only for
1382 // these three types to match the logic in
1383 // NVPTXAsmPrinter::printReturnValStr and NVPTXTargetLowering::getPrototype.
1384 // Plus, this behavior is consistent with nvcc's.
1385 if (retTy->isFloatingPointTy() || retTy->isIntegerTy() ||
1386 retTy->isPointerTy()) {
Justin Holewinskif8f70912013-06-28 17:57:59 +00001387 // Scalar needs to be at least 32bit wide
1388 if (resultsz < 32)
1389 resultsz = 32;
1390 SDVTList DeclareRetVTs = DAG.getVTList(MVT::Other, MVT::Glue);
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +00001391 SDValue DeclareRetOps[] = { Chain, DAG.getConstant(1, dl, MVT::i32),
1392 DAG.getConstant(resultsz, dl, MVT::i32),
1393 DAG.getConstant(0, dl, MVT::i32), InFlag };
Justin Holewinskif8f70912013-06-28 17:57:59 +00001394 Chain = DAG.getNode(NVPTXISD::DeclareRet, dl, DeclareRetVTs,
Craig Topper48d114b2014-04-26 18:35:24 +00001395 DeclareRetOps);
Justin Holewinskif8f70912013-06-28 17:57:59 +00001396 InFlag = Chain.getValue(1);
1397 } else {
1398 retAlignment = getArgumentAlignment(Callee, CS, retTy, 0);
1399 SDVTList DeclareRetVTs = DAG.getVTList(MVT::Other, MVT::Glue);
1400 SDValue DeclareRetOps[] = { Chain,
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +00001401 DAG.getConstant(retAlignment, dl, MVT::i32),
1402 DAG.getConstant(resultsz / 8, dl, MVT::i32),
1403 DAG.getConstant(0, dl, MVT::i32), InFlag };
Justin Holewinskif8f70912013-06-28 17:57:59 +00001404 Chain = DAG.getNode(NVPTXISD::DeclareRetParam, dl, DeclareRetVTs,
Craig Topper48d114b2014-04-26 18:35:24 +00001405 DeclareRetOps);
Justin Holewinskif8f70912013-06-28 17:57:59 +00001406 InFlag = Chain.getValue(1);
Justin Holewinskiae556d32012-05-04 20:18:50 +00001407 }
1408 }
1409
1410 if (!Func) {
1411 // This is indirect function call case : PTX requires a prototype of the
1412 // form
1413 // proto_0 : .callprototype(.param .b32 _) _ (.param .b32 _);
1414 // to be emitted, and the label has to used as the last arg of call
1415 // instruction.
Justin Holewinski3d49e5c2013-11-15 12:30:04 +00001416 // The prototype is embedded in a string and put as the operand for a
1417 // CallPrototype SDNode which will print out to the value of the string.
1418 SDVTList ProtoVTs = DAG.getVTList(MVT::Other, MVT::Glue);
1419 std::string Proto = getPrototype(retTy, Args, Outs, retAlignment, CS);
1420 const char *ProtoStr =
1421 nvTM->getManagedStrPool()->getManagedString(Proto.c_str())->c_str();
1422 SDValue ProtoOps[] = {
1423 Chain, DAG.getTargetExternalSymbol(ProtoStr, MVT::i32), InFlag,
Justin Holewinski0497ab12013-03-30 14:29:21 +00001424 };
Craig Topper48d114b2014-04-26 18:35:24 +00001425 Chain = DAG.getNode(NVPTXISD::CallPrototype, dl, ProtoVTs, ProtoOps);
Justin Holewinskiae556d32012-05-04 20:18:50 +00001426 InFlag = Chain.getValue(1);
1427 }
1428 // Op to just print "call"
1429 SDVTList PrintCallVTs = DAG.getVTList(MVT::Other, MVT::Glue);
Justin Holewinski0497ab12013-03-30 14:29:21 +00001430 SDValue PrintCallOps[] = {
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +00001431 Chain, DAG.getConstant((Ins.size() == 0) ? 0 : 1, dl, MVT::i32), InFlag
Justin Holewinski0497ab12013-03-30 14:29:21 +00001432 };
1433 Chain = DAG.getNode(Func ? (NVPTXISD::PrintCallUni) : (NVPTXISD::PrintCall),
Craig Topper48d114b2014-04-26 18:35:24 +00001434 dl, PrintCallVTs, PrintCallOps);
Justin Holewinskiae556d32012-05-04 20:18:50 +00001435 InFlag = Chain.getValue(1);
1436
1437 // Ops to print out the function name
1438 SDVTList CallVoidVTs = DAG.getVTList(MVT::Other, MVT::Glue);
1439 SDValue CallVoidOps[] = { Chain, Callee, InFlag };
Craig Topper48d114b2014-04-26 18:35:24 +00001440 Chain = DAG.getNode(NVPTXISD::CallVoid, dl, CallVoidVTs, CallVoidOps);
Justin Holewinskiae556d32012-05-04 20:18:50 +00001441 InFlag = Chain.getValue(1);
1442
1443 // Ops to print out the param list
1444 SDVTList CallArgBeginVTs = DAG.getVTList(MVT::Other, MVT::Glue);
1445 SDValue CallArgBeginOps[] = { Chain, InFlag };
1446 Chain = DAG.getNode(NVPTXISD::CallArgBegin, dl, CallArgBeginVTs,
Craig Topper48d114b2014-04-26 18:35:24 +00001447 CallArgBeginOps);
Justin Holewinskiae556d32012-05-04 20:18:50 +00001448 InFlag = Chain.getValue(1);
1449
Justin Holewinski0497ab12013-03-30 14:29:21 +00001450 for (unsigned i = 0, e = paramCount; i != e; ++i) {
Justin Holewinskiae556d32012-05-04 20:18:50 +00001451 unsigned opcode;
Justin Holewinski0497ab12013-03-30 14:29:21 +00001452 if (i == (e - 1))
Justin Holewinskiae556d32012-05-04 20:18:50 +00001453 opcode = NVPTXISD::LastCallArg;
1454 else
1455 opcode = NVPTXISD::CallArg;
1456 SDVTList CallArgVTs = DAG.getVTList(MVT::Other, MVT::Glue);
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +00001457 SDValue CallArgOps[] = { Chain, DAG.getConstant(1, dl, MVT::i32),
1458 DAG.getConstant(i, dl, MVT::i32), InFlag };
Craig Topper48d114b2014-04-26 18:35:24 +00001459 Chain = DAG.getNode(opcode, dl, CallArgVTs, CallArgOps);
Justin Holewinskiae556d32012-05-04 20:18:50 +00001460 InFlag = Chain.getValue(1);
1461 }
1462 SDVTList CallArgEndVTs = DAG.getVTList(MVT::Other, MVT::Glue);
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +00001463 SDValue CallArgEndOps[] = { Chain,
1464 DAG.getConstant(Func ? 1 : 0, dl, MVT::i32),
Justin Holewinskiae556d32012-05-04 20:18:50 +00001465 InFlag };
Craig Topper48d114b2014-04-26 18:35:24 +00001466 Chain = DAG.getNode(NVPTXISD::CallArgEnd, dl, CallArgEndVTs, CallArgEndOps);
Justin Holewinskiae556d32012-05-04 20:18:50 +00001467 InFlag = Chain.getValue(1);
1468
1469 if (!Func) {
1470 SDVTList PrototypeVTs = DAG.getVTList(MVT::Other, MVT::Glue);
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +00001471 SDValue PrototypeOps[] = { Chain,
1472 DAG.getConstant(uniqueCallSite, dl, MVT::i32),
Justin Holewinskiae556d32012-05-04 20:18:50 +00001473 InFlag };
Craig Topper48d114b2014-04-26 18:35:24 +00001474 Chain = DAG.getNode(NVPTXISD::Prototype, dl, PrototypeVTs, PrototypeOps);
Justin Holewinskiae556d32012-05-04 20:18:50 +00001475 InFlag = Chain.getValue(1);
1476 }
1477
1478 // Generate loads from param memory/moves from registers for result
1479 if (Ins.size() > 0) {
Justin Holewinskif8f70912013-06-28 17:57:59 +00001480 if (retTy && retTy->isVectorTy()) {
Mehdi Amini44ede332015-07-09 02:09:04 +00001481 EVT ObjectVT = getValueType(DL, retTy);
Justin Holewinskif8f70912013-06-28 17:57:59 +00001482 unsigned NumElts = ObjectVT.getVectorNumElements();
1483 EVT EltVT = ObjectVT.getVectorElementType();
Eric Christopherbef0a372015-01-30 01:50:07 +00001484 assert(STI.getTargetLowering()->getNumRegisters(F->getContext(),
1485 ObjectVT) == NumElts &&
Justin Holewinskif8f70912013-06-28 17:57:59 +00001486 "Vector was not scalarized");
1487 unsigned sz = EltVT.getSizeInBits();
Eli Bendersky3e840192015-03-23 16:26:23 +00001488 bool needTruncate = sz < 8;
Justin Holewinskif8f70912013-06-28 17:57:59 +00001489
1490 if (NumElts == 1) {
1491 // Just a simple load
Craig Topper59f626d2014-04-26 19:29:47 +00001492 SmallVector<EVT, 4> LoadRetVTs;
Justin Holewinski6e40f632014-06-27 18:35:44 +00001493 if (EltVT == MVT::i1 || EltVT == MVT::i8) {
1494 // If loading i1/i8 result, generate
1495 // load.b8 i16
1496 // if i1
Justin Holewinskif8f70912013-06-28 17:57:59 +00001497 // trunc i16 to i1
1498 LoadRetVTs.push_back(MVT::i16);
1499 } else
1500 LoadRetVTs.push_back(EltVT);
1501 LoadRetVTs.push_back(MVT::Other);
1502 LoadRetVTs.push_back(MVT::Glue);
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +00001503 SDValue LoadRetOps[] = {Chain, DAG.getConstant(1, dl, MVT::i32),
1504 DAG.getConstant(0, dl, MVT::i32), InFlag};
Justin Holewinskif8f70912013-06-28 17:57:59 +00001505 SDValue retval = DAG.getMemIntrinsicNode(
1506 NVPTXISD::LoadParam, dl,
Craig Topper206fcd42014-04-26 19:29:41 +00001507 DAG.getVTList(LoadRetVTs), LoadRetOps, EltVT, MachinePointerInfo());
Justin Holewinskiae556d32012-05-04 20:18:50 +00001508 Chain = retval.getValue(1);
1509 InFlag = retval.getValue(2);
Justin Holewinskif8f70912013-06-28 17:57:59 +00001510 SDValue Ret0 = retval;
1511 if (needTruncate)
1512 Ret0 = DAG.getNode(ISD::TRUNCATE, dl, EltVT, Ret0);
1513 InVals.push_back(Ret0);
1514 } else if (NumElts == 2) {
1515 // LoadV2
Craig Topper59f626d2014-04-26 19:29:47 +00001516 SmallVector<EVT, 4> LoadRetVTs;
Justin Holewinski6e40f632014-06-27 18:35:44 +00001517 if (EltVT == MVT::i1 || EltVT == MVT::i8) {
1518 // If loading i1/i8 result, generate
1519 // load.b8 i16
1520 // if i1
Justin Holewinskif8f70912013-06-28 17:57:59 +00001521 // trunc i16 to i1
1522 LoadRetVTs.push_back(MVT::i16);
1523 LoadRetVTs.push_back(MVT::i16);
1524 } else {
1525 LoadRetVTs.push_back(EltVT);
1526 LoadRetVTs.push_back(EltVT);
1527 }
1528 LoadRetVTs.push_back(MVT::Other);
1529 LoadRetVTs.push_back(MVT::Glue);
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +00001530 SDValue LoadRetOps[] = {Chain, DAG.getConstant(1, dl, MVT::i32),
1531 DAG.getConstant(0, dl, MVT::i32), InFlag};
Justin Holewinskif8f70912013-06-28 17:57:59 +00001532 SDValue retval = DAG.getMemIntrinsicNode(
1533 NVPTXISD::LoadParamV2, dl,
Craig Topper206fcd42014-04-26 19:29:41 +00001534 DAG.getVTList(LoadRetVTs), LoadRetOps, EltVT, MachinePointerInfo());
Justin Holewinskif8f70912013-06-28 17:57:59 +00001535 Chain = retval.getValue(2);
1536 InFlag = retval.getValue(3);
1537 SDValue Ret0 = retval.getValue(0);
1538 SDValue Ret1 = retval.getValue(1);
1539 if (needTruncate) {
1540 Ret0 = DAG.getNode(ISD::TRUNCATE, dl, MVT::i1, Ret0);
1541 InVals.push_back(Ret0);
1542 Ret1 = DAG.getNode(ISD::TRUNCATE, dl, MVT::i1, Ret1);
1543 InVals.push_back(Ret1);
1544 } else {
1545 InVals.push_back(Ret0);
1546 InVals.push_back(Ret1);
1547 }
1548 } else {
1549 // Split into N LoadV4
1550 unsigned Ofst = 0;
1551 unsigned VecSize = 4;
1552 unsigned Opc = NVPTXISD::LoadParamV4;
1553 if (EltVT.getSizeInBits() == 64) {
1554 VecSize = 2;
1555 Opc = NVPTXISD::LoadParamV2;
1556 }
1557 EVT VecVT = EVT::getVectorVT(F->getContext(), EltVT, VecSize);
1558 for (unsigned i = 0; i < NumElts; i += VecSize) {
1559 SmallVector<EVT, 8> LoadRetVTs;
Justin Holewinski6e40f632014-06-27 18:35:44 +00001560 if (EltVT == MVT::i1 || EltVT == MVT::i8) {
1561 // If loading i1/i8 result, generate
1562 // load.b8 i16
1563 // if i1
Justin Holewinskif8f70912013-06-28 17:57:59 +00001564 // trunc i16 to i1
1565 for (unsigned j = 0; j < VecSize; ++j)
1566 LoadRetVTs.push_back(MVT::i16);
1567 } else {
1568 for (unsigned j = 0; j < VecSize; ++j)
1569 LoadRetVTs.push_back(EltVT);
1570 }
1571 LoadRetVTs.push_back(MVT::Other);
1572 LoadRetVTs.push_back(MVT::Glue);
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +00001573 SDValue LoadRetOps[] = {Chain, DAG.getConstant(1, dl, MVT::i32),
1574 DAG.getConstant(Ofst, dl, MVT::i32), InFlag};
Justin Holewinskif8f70912013-06-28 17:57:59 +00001575 SDValue retval = DAG.getMemIntrinsicNode(
Craig Topperabb4ac72014-04-16 06:10:51 +00001576 Opc, dl, DAG.getVTList(LoadRetVTs),
Craig Topper206fcd42014-04-26 19:29:41 +00001577 LoadRetOps, EltVT, MachinePointerInfo());
Justin Holewinskif8f70912013-06-28 17:57:59 +00001578 if (VecSize == 2) {
1579 Chain = retval.getValue(2);
1580 InFlag = retval.getValue(3);
1581 } else {
1582 Chain = retval.getValue(4);
1583 InFlag = retval.getValue(5);
1584 }
1585
1586 for (unsigned j = 0; j < VecSize; ++j) {
1587 if (i + j >= NumElts)
1588 break;
1589 SDValue Elt = retval.getValue(j);
1590 if (needTruncate)
1591 Elt = DAG.getNode(ISD::TRUNCATE, dl, EltVT, Elt);
1592 InVals.push_back(Elt);
1593 }
Mehdi Amini56228da2015-07-09 01:57:34 +00001594 Ofst += DL.getTypeAllocSize(VecVT.getTypeForEVT(F->getContext()));
Justin Holewinskif8f70912013-06-28 17:57:59 +00001595 }
Justin Holewinskiae556d32012-05-04 20:18:50 +00001596 }
Justin Holewinski0497ab12013-03-30 14:29:21 +00001597 } else {
Justin Holewinskif8f70912013-06-28 17:57:59 +00001598 SmallVector<EVT, 16> VTs;
Justin Holewinski6e40f632014-06-27 18:35:44 +00001599 SmallVector<uint64_t, 16> Offsets;
Mehdi Amini56228da2015-07-09 01:57:34 +00001600 ComputePTXValueVTs(*this, DL, retTy, VTs, &Offsets, 0);
Justin Holewinskif8f70912013-06-28 17:57:59 +00001601 assert(VTs.size() == Ins.size() && "Bad value decomposition");
Justin Holewinski6e40f632014-06-27 18:35:44 +00001602 unsigned RetAlign = getArgumentAlignment(Callee, CS, retTy, 0);
Justin Holewinski0497ab12013-03-30 14:29:21 +00001603 for (unsigned i = 0, e = Ins.size(); i != e; ++i) {
Justin Holewinskif8f70912013-06-28 17:57:59 +00001604 unsigned sz = VTs[i].getSizeInBits();
Justin Holewinski9982f062014-06-27 19:36:25 +00001605 unsigned AlignI = GreatestCommonDivisor64(RetAlign, Offsets[i]);
Eli Bendersky3e840192015-03-23 16:26:23 +00001606 bool needTruncate = sz < 8;
Justin Holewinskif8f70912013-06-28 17:57:59 +00001607 if (VTs[i].isInteger() && (sz < 8))
1608 sz = 8;
1609
1610 SmallVector<EVT, 4> LoadRetVTs;
Justin Holewinskie04e4bd2013-06-28 17:58:10 +00001611 EVT TheLoadType = VTs[i];
Mehdi Amini56228da2015-07-09 01:57:34 +00001612 if (retTy->isIntegerTy() && DL.getTypeAllocSizeInBits(retTy) < 32) {
Justin Holewinskie04e4bd2013-06-28 17:58:10 +00001613 // This is for integer types only, and specifically not for
1614 // aggregates.
1615 LoadRetVTs.push_back(MVT::i32);
1616 TheLoadType = MVT::i32;
1617 } else if (sz < 16) {
Justin Holewinskif8f70912013-06-28 17:57:59 +00001618 // If loading i1/i8 result, generate
1619 // load i8 (-> i16)
1620 // trunc i16 to i1/i8
1621 LoadRetVTs.push_back(MVT::i16);
1622 } else
1623 LoadRetVTs.push_back(Ins[i].VT);
1624 LoadRetVTs.push_back(MVT::Other);
1625 LoadRetVTs.push_back(MVT::Glue);
1626
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +00001627 SDValue LoadRetOps[] = {Chain, DAG.getConstant(1, dl, MVT::i32),
1628 DAG.getConstant(Offsets[i], dl, MVT::i32),
1629 InFlag};
Justin Holewinskif8f70912013-06-28 17:57:59 +00001630 SDValue retval = DAG.getMemIntrinsicNode(
1631 NVPTXISD::LoadParam, dl,
Craig Topper206fcd42014-04-26 19:29:41 +00001632 DAG.getVTList(LoadRetVTs), LoadRetOps,
Justin Holewinski6e40f632014-06-27 18:35:44 +00001633 TheLoadType, MachinePointerInfo(), AlignI);
Justin Holewinskif8f70912013-06-28 17:57:59 +00001634 Chain = retval.getValue(1);
1635 InFlag = retval.getValue(2);
1636 SDValue Ret0 = retval.getValue(0);
1637 if (needTruncate)
1638 Ret0 = DAG.getNode(ISD::TRUNCATE, dl, Ins[i].VT, Ret0);
1639 InVals.push_back(Ret0);
Justin Holewinskiae556d32012-05-04 20:18:50 +00001640 }
1641 }
1642 }
Justin Holewinskif8f70912013-06-28 17:57:59 +00001643
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +00001644 Chain = DAG.getCALLSEQ_END(Chain,
1645 DAG.getIntPtrConstant(uniqueCallSite, dl, true),
1646 DAG.getIntPtrConstant(uniqueCallSite + 1, dl,
1647 true),
Andrew Trickad6d08a2013-05-29 22:03:55 +00001648 InFlag, dl);
Justin Holewinskiae556d32012-05-04 20:18:50 +00001649 uniqueCallSite++;
1650
1651 // set isTailCall to false for now, until we figure out how to express
1652 // tail call optimization in PTX
1653 isTailCall = false;
1654 return Chain;
1655}
Justin Holewinskiae556d32012-05-04 20:18:50 +00001656
1657// By default CONCAT_VECTORS is lowered by ExpandVectorBuildThroughStack()
1658// (see LegalizeDAG.cpp). This is slow and uses local memory.
1659// We use extract/insert/build vector just as what LegalizeOp() does in llvm 2.5
Justin Holewinski0497ab12013-03-30 14:29:21 +00001660SDValue
1661NVPTXTargetLowering::LowerCONCAT_VECTORS(SDValue Op, SelectionDAG &DAG) const {
Justin Holewinskiae556d32012-05-04 20:18:50 +00001662 SDNode *Node = Op.getNode();
Andrew Trickef9de2a2013-05-25 02:42:55 +00001663 SDLoc dl(Node);
Justin Holewinskiae556d32012-05-04 20:18:50 +00001664 SmallVector<SDValue, 8> Ops;
1665 unsigned NumOperands = Node->getNumOperands();
Justin Holewinski0497ab12013-03-30 14:29:21 +00001666 for (unsigned i = 0; i < NumOperands; ++i) {
Justin Holewinskiae556d32012-05-04 20:18:50 +00001667 SDValue SubOp = Node->getOperand(i);
1668 EVT VVT = SubOp.getNode()->getValueType(0);
1669 EVT EltVT = VVT.getVectorElementType();
1670 unsigned NumSubElem = VVT.getVectorNumElements();
Justin Holewinski0497ab12013-03-30 14:29:21 +00001671 for (unsigned j = 0; j < NumSubElem; ++j) {
Justin Holewinskiae556d32012-05-04 20:18:50 +00001672 Ops.push_back(DAG.getNode(ISD::EXTRACT_VECTOR_ELT, dl, EltVT, SubOp,
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +00001673 DAG.getIntPtrConstant(j, dl)));
Justin Holewinskiae556d32012-05-04 20:18:50 +00001674 }
1675 }
Craig Topper48d114b2014-04-26 18:35:24 +00001676 return DAG.getNode(ISD::BUILD_VECTOR, dl, Node->getValueType(0), Ops);
Justin Holewinskiae556d32012-05-04 20:18:50 +00001677}
1678
Justin Holewinski360a5cf2014-06-27 18:35:40 +00001679/// LowerShiftRightParts - Lower SRL_PARTS, SRA_PARTS, which
1680/// 1) returns two i32 values and take a 2 x i32 value to shift plus a shift
1681/// amount, or
1682/// 2) returns two i64 values and take a 2 x i64 value to shift plus a shift
1683/// amount.
1684SDValue NVPTXTargetLowering::LowerShiftRightParts(SDValue Op,
1685 SelectionDAG &DAG) const {
1686 assert(Op.getNumOperands() == 3 && "Not a double-shift!");
1687 assert(Op.getOpcode() == ISD::SRA_PARTS || Op.getOpcode() == ISD::SRL_PARTS);
1688
1689 EVT VT = Op.getValueType();
1690 unsigned VTBits = VT.getSizeInBits();
1691 SDLoc dl(Op);
1692 SDValue ShOpLo = Op.getOperand(0);
1693 SDValue ShOpHi = Op.getOperand(1);
1694 SDValue ShAmt = Op.getOperand(2);
1695 unsigned Opc = (Op.getOpcode() == ISD::SRA_PARTS) ? ISD::SRA : ISD::SRL;
1696
Eric Christopherbef0a372015-01-30 01:50:07 +00001697 if (VTBits == 32 && STI.getSmVersion() >= 35) {
Justin Holewinski360a5cf2014-06-27 18:35:40 +00001698
1699 // For 32bit and sm35, we can use the funnel shift 'shf' instruction.
1700 // {dHi, dLo} = {aHi, aLo} >> Amt
1701 // dHi = aHi >> Amt
1702 // dLo = shf.r.clamp aLo, aHi, Amt
1703
1704 SDValue Hi = DAG.getNode(Opc, dl, VT, ShOpHi, ShAmt);
1705 SDValue Lo = DAG.getNode(NVPTXISD::FUN_SHFR_CLAMP, dl, VT, ShOpLo, ShOpHi,
1706 ShAmt);
1707
1708 SDValue Ops[2] = { Lo, Hi };
1709 return DAG.getMergeValues(Ops, dl);
1710 }
1711 else {
1712
1713 // {dHi, dLo} = {aHi, aLo} >> Amt
1714 // - if (Amt>=size) then
1715 // dLo = aHi >> (Amt-size)
1716 // dHi = aHi >> Amt (this is either all 0 or all 1)
1717 // else
1718 // dLo = (aLo >>logic Amt) | (aHi << (size-Amt))
1719 // dHi = aHi >> Amt
1720
1721 SDValue RevShAmt = DAG.getNode(ISD::SUB, dl, MVT::i32,
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +00001722 DAG.getConstant(VTBits, dl, MVT::i32),
1723 ShAmt);
Justin Holewinski360a5cf2014-06-27 18:35:40 +00001724 SDValue Tmp1 = DAG.getNode(ISD::SRL, dl, VT, ShOpLo, ShAmt);
1725 SDValue ExtraShAmt = DAG.getNode(ISD::SUB, dl, MVT::i32, ShAmt,
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +00001726 DAG.getConstant(VTBits, dl, MVT::i32));
Justin Holewinski360a5cf2014-06-27 18:35:40 +00001727 SDValue Tmp2 = DAG.getNode(ISD::SHL, dl, VT, ShOpHi, RevShAmt);
1728 SDValue FalseVal = DAG.getNode(ISD::OR, dl, VT, Tmp1, Tmp2);
1729 SDValue TrueVal = DAG.getNode(Opc, dl, VT, ShOpHi, ExtraShAmt);
1730
1731 SDValue Cmp = DAG.getSetCC(dl, MVT::i1, ShAmt,
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +00001732 DAG.getConstant(VTBits, dl, MVT::i32),
1733 ISD::SETGE);
Justin Holewinski360a5cf2014-06-27 18:35:40 +00001734 SDValue Hi = DAG.getNode(Opc, dl, VT, ShOpHi, ShAmt);
1735 SDValue Lo = DAG.getNode(ISD::SELECT, dl, VT, Cmp, TrueVal, FalseVal);
1736
1737 SDValue Ops[2] = { Lo, Hi };
1738 return DAG.getMergeValues(Ops, dl);
1739 }
1740}
1741
1742/// LowerShiftLeftParts - Lower SHL_PARTS, which
1743/// 1) returns two i32 values and take a 2 x i32 value to shift plus a shift
1744/// amount, or
1745/// 2) returns two i64 values and take a 2 x i64 value to shift plus a shift
1746/// amount.
1747SDValue NVPTXTargetLowering::LowerShiftLeftParts(SDValue Op,
1748 SelectionDAG &DAG) const {
1749 assert(Op.getNumOperands() == 3 && "Not a double-shift!");
1750 assert(Op.getOpcode() == ISD::SHL_PARTS);
1751
1752 EVT VT = Op.getValueType();
1753 unsigned VTBits = VT.getSizeInBits();
1754 SDLoc dl(Op);
1755 SDValue ShOpLo = Op.getOperand(0);
1756 SDValue ShOpHi = Op.getOperand(1);
1757 SDValue ShAmt = Op.getOperand(2);
1758
Eric Christopherbef0a372015-01-30 01:50:07 +00001759 if (VTBits == 32 && STI.getSmVersion() >= 35) {
Justin Holewinski360a5cf2014-06-27 18:35:40 +00001760
1761 // For 32bit and sm35, we can use the funnel shift 'shf' instruction.
1762 // {dHi, dLo} = {aHi, aLo} << Amt
1763 // dHi = shf.l.clamp aLo, aHi, Amt
1764 // dLo = aLo << Amt
1765
1766 SDValue Hi = DAG.getNode(NVPTXISD::FUN_SHFL_CLAMP, dl, VT, ShOpLo, ShOpHi,
1767 ShAmt);
1768 SDValue Lo = DAG.getNode(ISD::SHL, dl, VT, ShOpLo, ShAmt);
1769
1770 SDValue Ops[2] = { Lo, Hi };
1771 return DAG.getMergeValues(Ops, dl);
1772 }
1773 else {
1774
1775 // {dHi, dLo} = {aHi, aLo} << Amt
1776 // - if (Amt>=size) then
1777 // dLo = aLo << Amt (all 0)
1778 // dLo = aLo << (Amt-size)
1779 // else
1780 // dLo = aLo << Amt
1781 // dHi = (aHi << Amt) | (aLo >> (size-Amt))
1782
1783 SDValue RevShAmt = DAG.getNode(ISD::SUB, dl, MVT::i32,
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +00001784 DAG.getConstant(VTBits, dl, MVT::i32),
1785 ShAmt);
Justin Holewinski360a5cf2014-06-27 18:35:40 +00001786 SDValue Tmp1 = DAG.getNode(ISD::SHL, dl, VT, ShOpHi, ShAmt);
1787 SDValue ExtraShAmt = DAG.getNode(ISD::SUB, dl, MVT::i32, ShAmt,
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +00001788 DAG.getConstant(VTBits, dl, MVT::i32));
Justin Holewinski360a5cf2014-06-27 18:35:40 +00001789 SDValue Tmp2 = DAG.getNode(ISD::SRL, dl, VT, ShOpLo, RevShAmt);
1790 SDValue FalseVal = DAG.getNode(ISD::OR, dl, VT, Tmp1, Tmp2);
1791 SDValue TrueVal = DAG.getNode(ISD::SHL, dl, VT, ShOpLo, ExtraShAmt);
1792
1793 SDValue Cmp = DAG.getSetCC(dl, MVT::i1, ShAmt,
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +00001794 DAG.getConstant(VTBits, dl, MVT::i32),
1795 ISD::SETGE);
Justin Holewinski360a5cf2014-06-27 18:35:40 +00001796 SDValue Lo = DAG.getNode(ISD::SHL, dl, VT, ShOpLo, ShAmt);
1797 SDValue Hi = DAG.getNode(ISD::SELECT, dl, VT, Cmp, TrueVal, FalseVal);
1798
1799 SDValue Ops[2] = { Lo, Hi };
1800 return DAG.getMergeValues(Ops, dl);
1801 }
1802}
1803
Justin Holewinski0497ab12013-03-30 14:29:21 +00001804SDValue
1805NVPTXTargetLowering::LowerOperation(SDValue Op, SelectionDAG &DAG) const {
Justin Holewinskiae556d32012-05-04 20:18:50 +00001806 switch (Op.getOpcode()) {
Justin Holewinski0497ab12013-03-30 14:29:21 +00001807 case ISD::RETURNADDR:
1808 return SDValue();
1809 case ISD::FRAMEADDR:
1810 return SDValue();
1811 case ISD::GlobalAddress:
1812 return LowerGlobalAddress(Op, DAG);
1813 case ISD::INTRINSIC_W_CHAIN:
1814 return Op;
Justin Holewinskiae556d32012-05-04 20:18:50 +00001815 case ISD::BUILD_VECTOR:
1816 case ISD::EXTRACT_SUBVECTOR:
1817 return Op;
Justin Holewinski0497ab12013-03-30 14:29:21 +00001818 case ISD::CONCAT_VECTORS:
1819 return LowerCONCAT_VECTORS(Op, DAG);
1820 case ISD::STORE:
1821 return LowerSTORE(Op, DAG);
1822 case ISD::LOAD:
1823 return LowerLOAD(Op, DAG);
Justin Holewinski360a5cf2014-06-27 18:35:40 +00001824 case ISD::SHL_PARTS:
1825 return LowerShiftLeftParts(Op, DAG);
1826 case ISD::SRA_PARTS:
1827 case ISD::SRL_PARTS:
1828 return LowerShiftRightParts(Op, DAG);
Justin Holewinskid4d2e9b2015-01-26 19:52:20 +00001829 case ISD::SELECT:
1830 return LowerSelect(Op, DAG);
Justin Holewinskiae556d32012-05-04 20:18:50 +00001831 default:
David Blaikie891d0a32012-05-04 22:34:16 +00001832 llvm_unreachable("Custom lowering not defined for operation");
Justin Holewinskiae556d32012-05-04 20:18:50 +00001833 }
1834}
1835
Justin Holewinskid4d2e9b2015-01-26 19:52:20 +00001836SDValue NVPTXTargetLowering::LowerSelect(SDValue Op, SelectionDAG &DAG) const {
1837 SDValue Op0 = Op->getOperand(0);
1838 SDValue Op1 = Op->getOperand(1);
1839 SDValue Op2 = Op->getOperand(2);
1840 SDLoc DL(Op.getNode());
1841
1842 assert(Op.getValueType() == MVT::i1 && "Custom lowering enabled only for i1");
1843
1844 Op1 = DAG.getNode(ISD::ANY_EXTEND, DL, MVT::i32, Op1);
1845 Op2 = DAG.getNode(ISD::ANY_EXTEND, DL, MVT::i32, Op2);
1846 SDValue Select = DAG.getNode(ISD::SELECT, DL, MVT::i32, Op0, Op1, Op2);
1847 SDValue Trunc = DAG.getNode(ISD::TRUNCATE, DL, MVT::i1, Select);
1848
1849 return Trunc;
1850}
1851
Justin Holewinskibe8dc642013-02-12 14:18:49 +00001852SDValue NVPTXTargetLowering::LowerLOAD(SDValue Op, SelectionDAG &DAG) const {
1853 if (Op.getValueType() == MVT::i1)
1854 return LowerLOADi1(Op, DAG);
1855 else
1856 return SDValue();
1857}
1858
Justin Holewinskic6462aa2012-11-14 19:19:16 +00001859// v = ld i1* addr
1860// =>
Justin Holewinskif8f70912013-06-28 17:57:59 +00001861// v1 = ld i8* addr (-> i16)
1862// v = trunc i16 to i1
Justin Holewinski0497ab12013-03-30 14:29:21 +00001863SDValue NVPTXTargetLowering::LowerLOADi1(SDValue Op, SelectionDAG &DAG) const {
Justin Holewinskic6462aa2012-11-14 19:19:16 +00001864 SDNode *Node = Op.getNode();
1865 LoadSDNode *LD = cast<LoadSDNode>(Node);
Andrew Trickef9de2a2013-05-25 02:42:55 +00001866 SDLoc dl(Node);
Justin Holewinski0497ab12013-03-30 14:29:21 +00001867 assert(LD->getExtensionType() == ISD::NON_EXTLOAD);
NAKAMURA Takumi5bbe0e12012-11-14 23:46:15 +00001868 assert(Node->getValueType(0) == MVT::i1 &&
1869 "Custom lowering for i1 load only");
Justin Holewinski0497ab12013-03-30 14:29:21 +00001870 SDValue newLD =
Justin Holewinskif8f70912013-06-28 17:57:59 +00001871 DAG.getLoad(MVT::i16, dl, LD->getChain(), LD->getBasePtr(),
Justin Holewinski0497ab12013-03-30 14:29:21 +00001872 LD->getPointerInfo(), LD->isVolatile(), LD->isNonTemporal(),
1873 LD->isInvariant(), LD->getAlignment());
Justin Holewinskic6462aa2012-11-14 19:19:16 +00001874 SDValue result = DAG.getNode(ISD::TRUNCATE, dl, MVT::i1, newLD);
1875 // The legalizer (the caller) is expecting two values from the legalized
1876 // load, so we build a MergeValues node for it. See ExpandUnalignedLoad()
1877 // in LegalizeDAG.cpp which also uses MergeValues.
Justin Holewinski0497ab12013-03-30 14:29:21 +00001878 SDValue Ops[] = { result, LD->getChain() };
Craig Topper64941d92014-04-27 19:20:57 +00001879 return DAG.getMergeValues(Ops, dl);
Justin Holewinskic6462aa2012-11-14 19:19:16 +00001880}
1881
Justin Holewinskibe8dc642013-02-12 14:18:49 +00001882SDValue NVPTXTargetLowering::LowerSTORE(SDValue Op, SelectionDAG &DAG) const {
1883 EVT ValVT = Op.getOperand(1).getValueType();
1884 if (ValVT == MVT::i1)
1885 return LowerSTOREi1(Op, DAG);
1886 else if (ValVT.isVector())
1887 return LowerSTOREVector(Op, DAG);
1888 else
1889 return SDValue();
1890}
1891
1892SDValue
1893NVPTXTargetLowering::LowerSTOREVector(SDValue Op, SelectionDAG &DAG) const {
1894 SDNode *N = Op.getNode();
1895 SDValue Val = N->getOperand(1);
Andrew Trickef9de2a2013-05-25 02:42:55 +00001896 SDLoc DL(N);
Justin Holewinskibe8dc642013-02-12 14:18:49 +00001897 EVT ValVT = Val.getValueType();
1898
1899 if (ValVT.isVector()) {
1900 // We only handle "native" vector sizes for now, e.g. <4 x double> is not
1901 // legal. We can (and should) split that into 2 stores of <2 x double> here
1902 // but I'm leaving that as a TODO for now.
1903 if (!ValVT.isSimple())
1904 return SDValue();
1905 switch (ValVT.getSimpleVT().SimpleTy) {
Justin Holewinski0497ab12013-03-30 14:29:21 +00001906 default:
1907 return SDValue();
Justin Holewinskibe8dc642013-02-12 14:18:49 +00001908 case MVT::v2i8:
1909 case MVT::v2i16:
1910 case MVT::v2i32:
1911 case MVT::v2i64:
1912 case MVT::v2f32:
1913 case MVT::v2f64:
1914 case MVT::v4i8:
1915 case MVT::v4i16:
1916 case MVT::v4i32:
1917 case MVT::v4f32:
1918 // This is a "native" vector type
1919 break;
1920 }
1921
Justin Holewinskiac451062014-07-16 19:45:35 +00001922 MemSDNode *MemSD = cast<MemSDNode>(N);
1923 const DataLayout *TD = getDataLayout();
1924
1925 unsigned Align = MemSD->getAlignment();
1926 unsigned PrefAlign =
1927 TD->getPrefTypeAlignment(ValVT.getTypeForEVT(*DAG.getContext()));
1928 if (Align < PrefAlign) {
1929 // This store is not sufficiently aligned, so bail out and let this vector
1930 // store be scalarized. Note that we may still be able to emit smaller
1931 // vector stores. For example, if we are storing a <4 x float> with an
1932 // alignment of 8, this check will fail but the legalizer will try again
1933 // with 2 x <2 x float>, which will succeed with an alignment of 8.
1934 return SDValue();
1935 }
1936
Justin Holewinskibe8dc642013-02-12 14:18:49 +00001937 unsigned Opcode = 0;
1938 EVT EltVT = ValVT.getVectorElementType();
1939 unsigned NumElts = ValVT.getVectorNumElements();
1940
1941 // Since StoreV2 is a target node, we cannot rely on DAG type legalization.
1942 // Therefore, we must ensure the type is legal. For i1 and i8, we set the
Alp Tokercb402912014-01-24 17:20:08 +00001943 // stored type to i16 and propagate the "real" type as the memory type.
Justin Holewinskia2911282013-07-01 12:58:58 +00001944 bool NeedExt = false;
Justin Holewinskibe8dc642013-02-12 14:18:49 +00001945 if (EltVT.getSizeInBits() < 16)
Justin Holewinskia2911282013-07-01 12:58:58 +00001946 NeedExt = true;
Justin Holewinskibe8dc642013-02-12 14:18:49 +00001947
1948 switch (NumElts) {
Justin Holewinski0497ab12013-03-30 14:29:21 +00001949 default:
1950 return SDValue();
Justin Holewinskibe8dc642013-02-12 14:18:49 +00001951 case 2:
1952 Opcode = NVPTXISD::StoreV2;
1953 break;
1954 case 4: {
1955 Opcode = NVPTXISD::StoreV4;
1956 break;
1957 }
1958 }
1959
1960 SmallVector<SDValue, 8> Ops;
1961
1962 // First is the chain
1963 Ops.push_back(N->getOperand(0));
1964
1965 // Then the split values
1966 for (unsigned i = 0; i < NumElts; ++i) {
1967 SDValue ExtVal = DAG.getNode(ISD::EXTRACT_VECTOR_ELT, DL, EltVT, Val,
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +00001968 DAG.getIntPtrConstant(i, DL));
Justin Holewinskia2911282013-07-01 12:58:58 +00001969 if (NeedExt)
1970 ExtVal = DAG.getNode(ISD::ANY_EXTEND, DL, MVT::i16, ExtVal);
Justin Holewinskibe8dc642013-02-12 14:18:49 +00001971 Ops.push_back(ExtVal);
1972 }
1973
1974 // Then any remaining arguments
Benjamin Kramer5fbfe2f2015-02-28 13:20:15 +00001975 Ops.append(N->op_begin() + 2, N->op_end());
Justin Holewinskibe8dc642013-02-12 14:18:49 +00001976
Justin Holewinski0497ab12013-03-30 14:29:21 +00001977 SDValue NewSt = DAG.getMemIntrinsicNode(
Craig Topper206fcd42014-04-26 19:29:41 +00001978 Opcode, DL, DAG.getVTList(MVT::Other), Ops,
Justin Holewinski0497ab12013-03-30 14:29:21 +00001979 MemSD->getMemoryVT(), MemSD->getMemOperand());
Justin Holewinskibe8dc642013-02-12 14:18:49 +00001980
1981 //return DCI.CombineTo(N, NewSt, true);
1982 return NewSt;
1983 }
1984
1985 return SDValue();
1986}
1987
Justin Holewinskic6462aa2012-11-14 19:19:16 +00001988// st i1 v, addr
1989// =>
Justin Holewinskif8f70912013-06-28 17:57:59 +00001990// v1 = zxt v to i16
1991// st.u8 i16, addr
Justin Holewinski0497ab12013-03-30 14:29:21 +00001992SDValue NVPTXTargetLowering::LowerSTOREi1(SDValue Op, SelectionDAG &DAG) const {
Justin Holewinskic6462aa2012-11-14 19:19:16 +00001993 SDNode *Node = Op.getNode();
Andrew Trickef9de2a2013-05-25 02:42:55 +00001994 SDLoc dl(Node);
Justin Holewinskic6462aa2012-11-14 19:19:16 +00001995 StoreSDNode *ST = cast<StoreSDNode>(Node);
1996 SDValue Tmp1 = ST->getChain();
1997 SDValue Tmp2 = ST->getBasePtr();
1998 SDValue Tmp3 = ST->getValue();
NAKAMURA Takumi5bbe0e12012-11-14 23:46:15 +00001999 assert(Tmp3.getValueType() == MVT::i1 && "Custom lowering for i1 store only");
Justin Holewinskic6462aa2012-11-14 19:19:16 +00002000 unsigned Alignment = ST->getAlignment();
2001 bool isVolatile = ST->isVolatile();
2002 bool isNonTemporal = ST->isNonTemporal();
Justin Holewinskif8f70912013-06-28 17:57:59 +00002003 Tmp3 = DAG.getNode(ISD::ZERO_EXTEND, dl, MVT::i16, Tmp3);
2004 SDValue Result = DAG.getTruncStore(Tmp1, dl, Tmp3, Tmp2,
2005 ST->getPointerInfo(), MVT::i8, isNonTemporal,
2006 isVolatile, Alignment);
Justin Holewinskic6462aa2012-11-14 19:19:16 +00002007 return Result;
2008}
2009
Justin Holewinski0497ab12013-03-30 14:29:21 +00002010SDValue NVPTXTargetLowering::getExtSymb(SelectionDAG &DAG, const char *inname,
2011 int idx, EVT v) const {
Justin Holewinskiae556d32012-05-04 20:18:50 +00002012 std::string *name = nvTM->getManagedStrPool()->getManagedString(inname);
2013 std::stringstream suffix;
2014 suffix << idx;
2015 *name += suffix.str();
2016 return DAG.getTargetExternalSymbol(name->c_str(), v);
2017}
2018
2019SDValue
2020NVPTXTargetLowering::getParamSymbol(SelectionDAG &DAG, int idx, EVT v) const {
Justin Holewinskia2a63d22013-08-06 14:13:27 +00002021 std::string ParamSym;
2022 raw_string_ostream ParamStr(ParamSym);
2023
2024 ParamStr << DAG.getMachineFunction().getName() << "_param_" << idx;
2025 ParamStr.flush();
2026
2027 std::string *SavedStr =
2028 nvTM->getManagedStrPool()->getManagedString(ParamSym.c_str());
2029 return DAG.getTargetExternalSymbol(SavedStr->c_str(), v);
Justin Holewinskiae556d32012-05-04 20:18:50 +00002030}
2031
Justin Holewinski0497ab12013-03-30 14:29:21 +00002032SDValue NVPTXTargetLowering::getParamHelpSymbol(SelectionDAG &DAG, int idx) {
Justin Holewinskiae556d32012-05-04 20:18:50 +00002033 return getExtSymb(DAG, ".HLPPARAM", idx);
2034}
2035
2036// Check to see if the kernel argument is image*_t or sampler_t
2037
2038bool llvm::isImageOrSamplerVal(const Value *arg, const Module *context) {
Justin Holewinski0497ab12013-03-30 14:29:21 +00002039 static const char *const specialTypes[] = { "struct._image2d_t",
2040 "struct._image3d_t",
2041 "struct._sampler_t" };
Justin Holewinskiae556d32012-05-04 20:18:50 +00002042
2043 const Type *Ty = arg->getType();
2044 const PointerType *PTy = dyn_cast<PointerType>(Ty);
2045
2046 if (!PTy)
2047 return false;
2048
2049 if (!context)
2050 return false;
2051
2052 const StructType *STy = dyn_cast<StructType>(PTy->getElementType());
Justin Holewinskifb711152012-12-05 20:50:28 +00002053 const std::string TypeName = STy && !STy->isLiteral() ? STy->getName() : "";
Justin Holewinskiae556d32012-05-04 20:18:50 +00002054
Craig Toppere4260f92012-05-24 04:22:05 +00002055 for (int i = 0, e = array_lengthof(specialTypes); i != e; ++i)
Justin Holewinskiae556d32012-05-04 20:18:50 +00002056 if (TypeName == specialTypes[i])
2057 return true;
2058
2059 return false;
2060}
2061
Justin Holewinski0497ab12013-03-30 14:29:21 +00002062SDValue NVPTXTargetLowering::LowerFormalArguments(
2063 SDValue Chain, CallingConv::ID CallConv, bool isVarArg,
Andrew Trickef9de2a2013-05-25 02:42:55 +00002064 const SmallVectorImpl<ISD::InputArg> &Ins, SDLoc dl, SelectionDAG &DAG,
Justin Holewinski0497ab12013-03-30 14:29:21 +00002065 SmallVectorImpl<SDValue> &InVals) const {
Justin Holewinskiae556d32012-05-04 20:18:50 +00002066 MachineFunction &MF = DAG.getMachineFunction();
Mehdi Amini56228da2015-07-09 01:57:34 +00002067 const DataLayout &DL = MF.getDataLayout();
Mehdi Amini44ede332015-07-09 02:09:04 +00002068 auto PtrVT = getPointerTy(DL);
Justin Holewinskiae556d32012-05-04 20:18:50 +00002069
2070 const Function *F = MF.getFunction();
Bill Wendlinge94d8432012-12-07 23:16:57 +00002071 const AttributeSet &PAL = F->getAttributes();
Eric Christopherbef0a372015-01-30 01:50:07 +00002072 const TargetLowering *TLI = STI.getTargetLowering();
Justin Holewinskiae556d32012-05-04 20:18:50 +00002073
2074 SDValue Root = DAG.getRoot();
2075 std::vector<SDValue> OutChains;
2076
2077 bool isKernel = llvm::isKernelFunction(*F);
Eric Christopherbef0a372015-01-30 01:50:07 +00002078 bool isABI = (STI.getSmVersion() >= 20);
Justin Holewinski44f5c602013-06-28 17:57:53 +00002079 assert(isABI && "Non-ABI compilation is not supported");
2080 if (!isABI)
2081 return Chain;
Justin Holewinskiae556d32012-05-04 20:18:50 +00002082
2083 std::vector<Type *> argTypes;
2084 std::vector<const Argument *> theArgs;
2085 for (Function::const_arg_iterator I = F->arg_begin(), E = F->arg_end();
Justin Holewinski0497ab12013-03-30 14:29:21 +00002086 I != E; ++I) {
Justin Holewinskiae556d32012-05-04 20:18:50 +00002087 theArgs.push_back(I);
2088 argTypes.push_back(I->getType());
2089 }
Justin Holewinski44f5c602013-06-28 17:57:53 +00002090 // argTypes.size() (or theArgs.size()) and Ins.size() need not match.
2091 // Ins.size() will be larger
2092 // * if there is an aggregate argument with multiple fields (each field
2093 // showing up separately in Ins)
2094 // * if there is a vector argument with more than typical vector-length
2095 // elements (generally if more than 4) where each vector element is
2096 // individually present in Ins.
2097 // So a different index should be used for indexing into Ins.
2098 // See similar issue in LowerCall.
2099 unsigned InsIdx = 0;
Justin Holewinskiae556d32012-05-04 20:18:50 +00002100
2101 int idx = 0;
Justin Holewinski44f5c602013-06-28 17:57:53 +00002102 for (unsigned i = 0, e = theArgs.size(); i != e; ++i, ++idx, ++InsIdx) {
Justin Holewinskiae556d32012-05-04 20:18:50 +00002103 Type *Ty = argTypes[i];
Justin Holewinskiae556d32012-05-04 20:18:50 +00002104
2105 // If the kernel argument is image*_t or sampler_t, convert it to
2106 // a i32 constant holding the parameter position. This can later
2107 // matched in the AsmPrinter to output the correct mangled name.
Justin Holewinski0497ab12013-03-30 14:29:21 +00002108 if (isImageOrSamplerVal(
2109 theArgs[i],
2110 (theArgs[i]->getParent() ? theArgs[i]->getParent()->getParent()
Craig Topper062a2ba2014-04-25 05:30:21 +00002111 : nullptr))) {
Justin Holewinskiae556d32012-05-04 20:18:50 +00002112 assert(isKernel && "Only kernels can have image/sampler params");
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +00002113 InVals.push_back(DAG.getConstant(i + 1, dl, MVT::i32));
Justin Holewinskiae556d32012-05-04 20:18:50 +00002114 continue;
2115 }
2116
2117 if (theArgs[i]->use_empty()) {
2118 // argument is dead
Justin Holewinski44f5c602013-06-28 17:57:53 +00002119 if (Ty->isAggregateType()) {
2120 SmallVector<EVT, 16> vtparts;
2121
Mehdi Amini56228da2015-07-09 01:57:34 +00002122 ComputePTXValueVTs(*this, DL, Ty, vtparts);
Justin Holewinski44f5c602013-06-28 17:57:53 +00002123 assert(vtparts.size() > 0 && "empty aggregate type not expected");
2124 for (unsigned parti = 0, parte = vtparts.size(); parti != parte;
2125 ++parti) {
Justin Holewinskib5db95e2014-06-27 18:36:04 +00002126 InVals.push_back(DAG.getNode(ISD::UNDEF, dl, Ins[InsIdx].VT));
Justin Holewinski44f5c602013-06-28 17:57:53 +00002127 ++InsIdx;
Justin Holewinskie9884092013-03-24 21:17:47 +00002128 }
Justin Holewinski44f5c602013-06-28 17:57:53 +00002129 if (vtparts.size() > 0)
2130 --InsIdx;
2131 continue;
Justin Holewinskie9884092013-03-24 21:17:47 +00002132 }
Justin Holewinski44f5c602013-06-28 17:57:53 +00002133 if (Ty->isVectorTy()) {
Mehdi Amini44ede332015-07-09 02:09:04 +00002134 EVT ObjectVT = getValueType(DL, Ty);
Justin Holewinski44f5c602013-06-28 17:57:53 +00002135 unsigned NumRegs = TLI->getNumRegisters(F->getContext(), ObjectVT);
2136 for (unsigned parti = 0; parti < NumRegs; ++parti) {
2137 InVals.push_back(DAG.getNode(ISD::UNDEF, dl, Ins[InsIdx].VT));
2138 ++InsIdx;
2139 }
2140 if (NumRegs > 0)
2141 --InsIdx;
2142 continue;
2143 }
2144 InVals.push_back(DAG.getNode(ISD::UNDEF, dl, Ins[InsIdx].VT));
Justin Holewinskiae556d32012-05-04 20:18:50 +00002145 continue;
2146 }
2147
2148 // In the following cases, assign a node order of "idx+1"
Justin Holewinski44f5c602013-06-28 17:57:53 +00002149 // to newly created nodes. The SDNodes for params have to
Justin Holewinskiae556d32012-05-04 20:18:50 +00002150 // appear in the same order as their order of appearance
2151 // in the original function. "idx+1" holds that order.
Eli Bendersky3e840192015-03-23 16:26:23 +00002152 if (!PAL.hasAttribute(i + 1, Attribute::ByVal)) {
Justin Holewinski44f5c602013-06-28 17:57:53 +00002153 if (Ty->isAggregateType()) {
2154 SmallVector<EVT, 16> vtparts;
2155 SmallVector<uint64_t, 16> offsets;
2156
Justin Holewinskif8f70912013-06-28 17:57:59 +00002157 // NOTE: Here, we lose the ability to issue vector loads for vectors
2158 // that are a part of a struct. This should be investigated in the
2159 // future.
Mehdi Amini56228da2015-07-09 01:57:34 +00002160 ComputePTXValueVTs(*this, DL, Ty, vtparts, &offsets, 0);
Justin Holewinski44f5c602013-06-28 17:57:53 +00002161 assert(vtparts.size() > 0 && "empty aggregate type not expected");
2162 bool aggregateIsPacked = false;
2163 if (StructType *STy = llvm::dyn_cast<StructType>(Ty))
2164 aggregateIsPacked = STy->isPacked();
2165
Mehdi Amini44ede332015-07-09 02:09:04 +00002166 SDValue Arg = getParamSymbol(DAG, idx, PtrVT);
Justin Holewinski44f5c602013-06-28 17:57:53 +00002167 for (unsigned parti = 0, parte = vtparts.size(); parti != parte;
2168 ++parti) {
2169 EVT partVT = vtparts[parti];
2170 Value *srcValue = Constant::getNullValue(
2171 PointerType::get(partVT.getTypeForEVT(F->getContext()),
2172 llvm::ADDRESS_SPACE_PARAM));
2173 SDValue srcAddr =
Mehdi Amini44ede332015-07-09 02:09:04 +00002174 DAG.getNode(ISD::ADD, dl, PtrVT, Arg,
2175 DAG.getConstant(offsets[parti], dl, PtrVT));
Mehdi Amini56228da2015-07-09 01:57:34 +00002176 unsigned partAlign = aggregateIsPacked
2177 ? 1
2178 : DL.getABITypeAlignment(
2179 partVT.getTypeForEVT(F->getContext()));
Justin Holewinskia2911282013-07-01 12:58:58 +00002180 SDValue p;
2181 if (Ins[InsIdx].VT.getSizeInBits() > partVT.getSizeInBits()) {
2182 ISD::LoadExtType ExtOp = Ins[InsIdx].Flags.isSExt() ?
2183 ISD::SEXTLOAD : ISD::ZEXTLOAD;
2184 p = DAG.getExtLoad(ExtOp, dl, Ins[InsIdx].VT, Root, srcAddr,
Justin Holewinskif8f70912013-06-28 17:57:59 +00002185 MachinePointerInfo(srcValue), partVT, false,
Louis Gerbarg67474e32014-07-31 21:45:05 +00002186 false, false, partAlign);
Justin Holewinskia2911282013-07-01 12:58:58 +00002187 } else {
Justin Holewinskif8f70912013-06-28 17:57:59 +00002188 p = DAG.getLoad(partVT, dl, Root, srcAddr,
2189 MachinePointerInfo(srcValue), false, false, false,
2190 partAlign);
Justin Holewinskia2911282013-07-01 12:58:58 +00002191 }
Justin Holewinski44f5c602013-06-28 17:57:53 +00002192 if (p.getNode())
2193 p.getNode()->setIROrder(idx + 1);
2194 InVals.push_back(p);
2195 ++InsIdx;
Justin Holewinskie9884092013-03-24 21:17:47 +00002196 }
Justin Holewinski44f5c602013-06-28 17:57:53 +00002197 if (vtparts.size() > 0)
2198 --InsIdx;
Justin Holewinskie9884092013-03-24 21:17:47 +00002199 continue;
2200 }
Justin Holewinski44f5c602013-06-28 17:57:53 +00002201 if (Ty->isVectorTy()) {
Mehdi Amini44ede332015-07-09 02:09:04 +00002202 EVT ObjectVT = getValueType(DL, Ty);
2203 SDValue Arg = getParamSymbol(DAG, idx, PtrVT);
Justin Holewinski44f5c602013-06-28 17:57:53 +00002204 unsigned NumElts = ObjectVT.getVectorNumElements();
2205 assert(TLI->getNumRegisters(F->getContext(), ObjectVT) == NumElts &&
2206 "Vector was not scalarized");
Justin Holewinski44f5c602013-06-28 17:57:53 +00002207 EVT EltVT = ObjectVT.getVectorElementType();
2208
2209 // V1 load
2210 // f32 = load ...
2211 if (NumElts == 1) {
2212 // We only have one element, so just directly load it
2213 Value *SrcValue = Constant::getNullValue(PointerType::get(
2214 EltVT.getTypeForEVT(F->getContext()), llvm::ADDRESS_SPACE_PARAM));
Justin Holewinski44f5c602013-06-28 17:57:53 +00002215 SDValue P = DAG.getLoad(
Mehdi Amini56228da2015-07-09 01:57:34 +00002216 EltVT, dl, Root, Arg, MachinePointerInfo(SrcValue), false, false,
2217 true,
2218 DL.getABITypeAlignment(EltVT.getTypeForEVT(F->getContext())));
Justin Holewinski44f5c602013-06-28 17:57:53 +00002219 if (P.getNode())
2220 P.getNode()->setIROrder(idx + 1);
2221
Justin Holewinskif8f70912013-06-28 17:57:59 +00002222 if (Ins[InsIdx].VT.getSizeInBits() > EltVT.getSizeInBits())
Justin Holewinskia2911282013-07-01 12:58:58 +00002223 P = DAG.getNode(ISD::ANY_EXTEND, dl, Ins[InsIdx].VT, P);
Justin Holewinski44f5c602013-06-28 17:57:53 +00002224 InVals.push_back(P);
Justin Holewinski44f5c602013-06-28 17:57:53 +00002225 ++InsIdx;
2226 } else if (NumElts == 2) {
2227 // V2 load
2228 // f32,f32 = load ...
2229 EVT VecVT = EVT::getVectorVT(F->getContext(), EltVT, 2);
2230 Value *SrcValue = Constant::getNullValue(PointerType::get(
2231 VecVT.getTypeForEVT(F->getContext()), llvm::ADDRESS_SPACE_PARAM));
Justin Holewinski44f5c602013-06-28 17:57:53 +00002232 SDValue P = DAG.getLoad(
Mehdi Amini56228da2015-07-09 01:57:34 +00002233 VecVT, dl, Root, Arg, MachinePointerInfo(SrcValue), false, false,
2234 true,
2235 DL.getABITypeAlignment(VecVT.getTypeForEVT(F->getContext())));
Justin Holewinski44f5c602013-06-28 17:57:53 +00002236 if (P.getNode())
2237 P.getNode()->setIROrder(idx + 1);
2238
2239 SDValue Elt0 = DAG.getNode(ISD::EXTRACT_VECTOR_ELT, dl, EltVT, P,
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +00002240 DAG.getIntPtrConstant(0, dl));
Justin Holewinski44f5c602013-06-28 17:57:53 +00002241 SDValue Elt1 = DAG.getNode(ISD::EXTRACT_VECTOR_ELT, dl, EltVT, P,
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +00002242 DAG.getIntPtrConstant(1, dl));
Justin Holewinskif8f70912013-06-28 17:57:59 +00002243
2244 if (Ins[InsIdx].VT.getSizeInBits() > EltVT.getSizeInBits()) {
Justin Holewinskia2911282013-07-01 12:58:58 +00002245 Elt0 = DAG.getNode(ISD::ANY_EXTEND, dl, Ins[InsIdx].VT, Elt0);
2246 Elt1 = DAG.getNode(ISD::ANY_EXTEND, dl, Ins[InsIdx].VT, Elt1);
Justin Holewinskif8f70912013-06-28 17:57:59 +00002247 }
2248
Justin Holewinski44f5c602013-06-28 17:57:53 +00002249 InVals.push_back(Elt0);
2250 InVals.push_back(Elt1);
Justin Holewinski44f5c602013-06-28 17:57:53 +00002251 InsIdx += 2;
2252 } else {
2253 // V4 loads
2254 // We have at least 4 elements (<3 x Ty> expands to 4 elements) and
2255 // the
2256 // vector will be expanded to a power of 2 elements, so we know we can
2257 // always round up to the next multiple of 4 when creating the vector
2258 // loads.
2259 // e.g. 4 elem => 1 ld.v4
2260 // 6 elem => 2 ld.v4
2261 // 8 elem => 2 ld.v4
2262 // 11 elem => 3 ld.v4
2263 unsigned VecSize = 4;
2264 if (EltVT.getSizeInBits() == 64) {
2265 VecSize = 2;
2266 }
2267 EVT VecVT = EVT::getVectorVT(F->getContext(), EltVT, VecSize);
Tilmann Scheller383b4ff2014-10-02 15:12:48 +00002268 unsigned Ofst = 0;
Justin Holewinski44f5c602013-06-28 17:57:53 +00002269 for (unsigned i = 0; i < NumElts; i += VecSize) {
2270 Value *SrcValue = Constant::getNullValue(
2271 PointerType::get(VecVT.getTypeForEVT(F->getContext()),
2272 llvm::ADDRESS_SPACE_PARAM));
Mehdi Amini44ede332015-07-09 02:09:04 +00002273 SDValue SrcAddr = DAG.getNode(ISD::ADD, dl, PtrVT, Arg,
2274 DAG.getConstant(Ofst, dl, PtrVT));
Justin Holewinski44f5c602013-06-28 17:57:53 +00002275 SDValue P = DAG.getLoad(
2276 VecVT, dl, Root, SrcAddr, MachinePointerInfo(SrcValue), false,
2277 false, true,
Mehdi Amini56228da2015-07-09 01:57:34 +00002278 DL.getABITypeAlignment(VecVT.getTypeForEVT(F->getContext())));
Justin Holewinski44f5c602013-06-28 17:57:53 +00002279 if (P.getNode())
2280 P.getNode()->setIROrder(idx + 1);
2281
2282 for (unsigned j = 0; j < VecSize; ++j) {
2283 if (i + j >= NumElts)
2284 break;
2285 SDValue Elt = DAG.getNode(ISD::EXTRACT_VECTOR_ELT, dl, EltVT, P,
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +00002286 DAG.getIntPtrConstant(j, dl));
Justin Holewinskif8f70912013-06-28 17:57:59 +00002287 if (Ins[InsIdx].VT.getSizeInBits() > EltVT.getSizeInBits())
Justin Holewinskia2911282013-07-01 12:58:58 +00002288 Elt = DAG.getNode(ISD::ANY_EXTEND, dl, Ins[InsIdx].VT, Elt);
Justin Holewinski44f5c602013-06-28 17:57:53 +00002289 InVals.push_back(Elt);
2290 }
Mehdi Amini56228da2015-07-09 01:57:34 +00002291 Ofst += DL.getTypeAllocSize(VecVT.getTypeForEVT(F->getContext()));
Justin Holewinski44f5c602013-06-28 17:57:53 +00002292 }
Justin Holewinski4f5bc9b2013-11-11 19:28:16 +00002293 InsIdx += NumElts;
Justin Holewinski44f5c602013-06-28 17:57:53 +00002294 }
2295
2296 if (NumElts > 0)
2297 --InsIdx;
2298 continue;
Justin Holewinskiae556d32012-05-04 20:18:50 +00002299 }
Justin Holewinski44f5c602013-06-28 17:57:53 +00002300 // A plain scalar.
Mehdi Amini44ede332015-07-09 02:09:04 +00002301 EVT ObjectVT = getValueType(DL, Ty);
Justin Holewinski44f5c602013-06-28 17:57:53 +00002302 // If ABI, load from the param symbol
Mehdi Amini44ede332015-07-09 02:09:04 +00002303 SDValue Arg = getParamSymbol(DAG, idx, PtrVT);
Justin Holewinski44f5c602013-06-28 17:57:53 +00002304 Value *srcValue = Constant::getNullValue(PointerType::get(
2305 ObjectVT.getTypeForEVT(F->getContext()), llvm::ADDRESS_SPACE_PARAM));
Justin Holewinskif8f70912013-06-28 17:57:59 +00002306 SDValue p;
Justin Holewinskia2911282013-07-01 12:58:58 +00002307 if (ObjectVT.getSizeInBits() < Ins[InsIdx].VT.getSizeInBits()) {
2308 ISD::LoadExtType ExtOp = Ins[InsIdx].Flags.isSExt() ?
2309 ISD::SEXTLOAD : ISD::ZEXTLOAD;
Mehdi Amini56228da2015-07-09 01:57:34 +00002310 p = DAG.getExtLoad(
2311 ExtOp, dl, Ins[InsIdx].VT, Root, Arg, MachinePointerInfo(srcValue),
2312 ObjectVT, false, false, false,
2313 DL.getABITypeAlignment(ObjectVT.getTypeForEVT(F->getContext())));
Justin Holewinskia2911282013-07-01 12:58:58 +00002314 } else {
Mehdi Amini56228da2015-07-09 01:57:34 +00002315 p = DAG.getLoad(
2316 Ins[InsIdx].VT, dl, Root, Arg, MachinePointerInfo(srcValue), false,
2317 false, false,
2318 DL.getABITypeAlignment(ObjectVT.getTypeForEVT(F->getContext())));
Justin Holewinskia2911282013-07-01 12:58:58 +00002319 }
Justin Holewinski44f5c602013-06-28 17:57:53 +00002320 if (p.getNode())
2321 p.getNode()->setIROrder(idx + 1);
2322 InVals.push_back(p);
Justin Holewinskiae556d32012-05-04 20:18:50 +00002323 continue;
2324 }
2325
2326 // Param has ByVal attribute
Justin Holewinski44f5c602013-06-28 17:57:53 +00002327 // Return MoveParam(param symbol).
2328 // Ideally, the param symbol can be returned directly,
2329 // but when SDNode builder decides to use it in a CopyToReg(),
2330 // machine instruction fails because TargetExternalSymbol
2331 // (not lowered) is target dependent, and CopyToReg assumes
2332 // the source is lowered.
Mehdi Amini44ede332015-07-09 02:09:04 +00002333 EVT ObjectVT = getValueType(DL, Ty);
Justin Holewinski44f5c602013-06-28 17:57:53 +00002334 assert(ObjectVT == Ins[InsIdx].VT &&
2335 "Ins type did not match function type");
Mehdi Amini44ede332015-07-09 02:09:04 +00002336 SDValue Arg = getParamSymbol(DAG, idx, PtrVT);
Justin Holewinski44f5c602013-06-28 17:57:53 +00002337 SDValue p = DAG.getNode(NVPTXISD::MoveParam, dl, ObjectVT, Arg);
2338 if (p.getNode())
2339 p.getNode()->setIROrder(idx + 1);
2340 if (isKernel)
2341 InVals.push_back(p);
2342 else {
2343 SDValue p2 = DAG.getNode(
2344 ISD::INTRINSIC_WO_CHAIN, dl, ObjectVT,
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +00002345 DAG.getConstant(Intrinsic::nvvm_ptr_local_to_gen, dl, MVT::i32), p);
Justin Holewinski44f5c602013-06-28 17:57:53 +00002346 InVals.push_back(p2);
Justin Holewinskiae556d32012-05-04 20:18:50 +00002347 }
2348 }
2349
2350 // Clang will check explicit VarArg and issue error if any. However, Clang
2351 // will let code with
Justin Holewinski44f5c602013-06-28 17:57:53 +00002352 // implicit var arg like f() pass. See bug 617733.
Justin Holewinskiae556d32012-05-04 20:18:50 +00002353 // We treat this case as if the arg list is empty.
Justin Holewinski44f5c602013-06-28 17:57:53 +00002354 // if (F.isVarArg()) {
Justin Holewinskiae556d32012-05-04 20:18:50 +00002355 // assert(0 && "VarArg not supported yet!");
2356 //}
2357
2358 if (!OutChains.empty())
Craig Topper48d114b2014-04-26 18:35:24 +00002359 DAG.setRoot(DAG.getNode(ISD::TokenFactor, dl, MVT::Other, OutChains));
Justin Holewinskiae556d32012-05-04 20:18:50 +00002360
2361 return Chain;
2362}
2363
Justin Holewinski44f5c602013-06-28 17:57:53 +00002364
Justin Holewinski120baee2013-06-28 17:57:55 +00002365SDValue
2366NVPTXTargetLowering::LowerReturn(SDValue Chain, CallingConv::ID CallConv,
2367 bool isVarArg,
2368 const SmallVectorImpl<ISD::OutputArg> &Outs,
2369 const SmallVectorImpl<SDValue> &OutVals,
2370 SDLoc dl, SelectionDAG &DAG) const {
2371 MachineFunction &MF = DAG.getMachineFunction();
2372 const Function *F = MF.getFunction();
Justin Holewinskie04e4bd2013-06-28 17:58:10 +00002373 Type *RetTy = F->getReturnType();
Mehdi Amini44ede332015-07-09 02:09:04 +00002374 const DataLayout &TD = DAG.getDataLayout();
Justin Holewinskiae556d32012-05-04 20:18:50 +00002375
Eric Christopherbef0a372015-01-30 01:50:07 +00002376 bool isABI = (STI.getSmVersion() >= 20);
Justin Holewinski120baee2013-06-28 17:57:55 +00002377 assert(isABI && "Non-ABI compilation is not supported");
2378 if (!isABI)
2379 return Chain;
Justin Holewinskiae556d32012-05-04 20:18:50 +00002380
Justin Holewinskie04e4bd2013-06-28 17:58:10 +00002381 if (VectorType *VTy = dyn_cast<VectorType>(RetTy)) {
Justin Holewinski120baee2013-06-28 17:57:55 +00002382 // If we have a vector type, the OutVals array will be the scalarized
2383 // components and we have combine them into 1 or more vector stores.
2384 unsigned NumElts = VTy->getNumElements();
2385 assert(NumElts == Outs.size() && "Bad scalarization of return value");
2386
Justin Holewinskif8f70912013-06-28 17:57:59 +00002387 // const_cast can be removed in later LLVM versions
Mehdi Amini44ede332015-07-09 02:09:04 +00002388 EVT EltVT = getValueType(TD, RetTy).getVectorElementType();
Justin Holewinskif8f70912013-06-28 17:57:59 +00002389 bool NeedExtend = false;
2390 if (EltVT.getSizeInBits() < 16)
2391 NeedExtend = true;
2392
Justin Holewinski120baee2013-06-28 17:57:55 +00002393 // V1 store
2394 if (NumElts == 1) {
2395 SDValue StoreVal = OutVals[0];
2396 // We only have one element, so just directly store it
Justin Holewinskif8f70912013-06-28 17:57:59 +00002397 if (NeedExtend)
2398 StoreVal = DAG.getNode(ISD::ZERO_EXTEND, dl, MVT::i16, StoreVal);
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +00002399 SDValue Ops[] = { Chain, DAG.getConstant(0, dl, MVT::i32), StoreVal };
Justin Holewinskif8f70912013-06-28 17:57:59 +00002400 Chain = DAG.getMemIntrinsicNode(NVPTXISD::StoreRetval, dl,
Craig Topper206fcd42014-04-26 19:29:41 +00002401 DAG.getVTList(MVT::Other), Ops,
Justin Holewinskif8f70912013-06-28 17:57:59 +00002402 EltVT, MachinePointerInfo());
2403
Justin Holewinski120baee2013-06-28 17:57:55 +00002404 } else if (NumElts == 2) {
2405 // V2 store
2406 SDValue StoreVal0 = OutVals[0];
2407 SDValue StoreVal1 = OutVals[1];
2408
Justin Holewinskif8f70912013-06-28 17:57:59 +00002409 if (NeedExtend) {
2410 StoreVal0 = DAG.getNode(ISD::ZERO_EXTEND, dl, MVT::i16, StoreVal0);
2411 StoreVal1 = DAG.getNode(ISD::ZERO_EXTEND, dl, MVT::i16, StoreVal1);
Justin Holewinski120baee2013-06-28 17:57:55 +00002412 }
2413
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +00002414 SDValue Ops[] = { Chain, DAG.getConstant(0, dl, MVT::i32), StoreVal0,
Justin Holewinskif8f70912013-06-28 17:57:59 +00002415 StoreVal1 };
2416 Chain = DAG.getMemIntrinsicNode(NVPTXISD::StoreRetvalV2, dl,
Craig Topper206fcd42014-04-26 19:29:41 +00002417 DAG.getVTList(MVT::Other), Ops,
Justin Holewinskif8f70912013-06-28 17:57:59 +00002418 EltVT, MachinePointerInfo());
Justin Holewinski120baee2013-06-28 17:57:55 +00002419 } else {
2420 // V4 stores
2421 // We have at least 4 elements (<3 x Ty> expands to 4 elements) and the
2422 // vector will be expanded to a power of 2 elements, so we know we can
2423 // always round up to the next multiple of 4 when creating the vector
2424 // stores.
2425 // e.g. 4 elem => 1 st.v4
2426 // 6 elem => 2 st.v4
2427 // 8 elem => 2 st.v4
2428 // 11 elem => 3 st.v4
2429
2430 unsigned VecSize = 4;
2431 if (OutVals[0].getValueType().getSizeInBits() == 64)
2432 VecSize = 2;
2433
2434 unsigned Offset = 0;
2435
2436 EVT VecVT =
Justin Holewinskib5db95e2014-06-27 18:36:04 +00002437 EVT::getVectorVT(F->getContext(), EltVT, VecSize);
Justin Holewinski120baee2013-06-28 17:57:55 +00002438 unsigned PerStoreOffset =
Mehdi Amini44ede332015-07-09 02:09:04 +00002439 TD.getTypeAllocSize(VecVT.getTypeForEVT(F->getContext()));
Justin Holewinski120baee2013-06-28 17:57:55 +00002440
Justin Holewinski120baee2013-06-28 17:57:55 +00002441 for (unsigned i = 0; i < NumElts; i += VecSize) {
2442 // Get values
2443 SDValue StoreVal;
2444 SmallVector<SDValue, 8> Ops;
2445 Ops.push_back(Chain);
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +00002446 Ops.push_back(DAG.getConstant(Offset, dl, MVT::i32));
Justin Holewinski120baee2013-06-28 17:57:55 +00002447 unsigned Opc = NVPTXISD::StoreRetvalV2;
Justin Holewinskif8f70912013-06-28 17:57:59 +00002448 EVT ExtendedVT = (NeedExtend) ? MVT::i16 : OutVals[0].getValueType();
Justin Holewinski120baee2013-06-28 17:57:55 +00002449
2450 StoreVal = OutVals[i];
Justin Holewinskif8f70912013-06-28 17:57:59 +00002451 if (NeedExtend)
2452 StoreVal = DAG.getNode(ISD::ZERO_EXTEND, dl, ExtendedVT, StoreVal);
Justin Holewinski120baee2013-06-28 17:57:55 +00002453 Ops.push_back(StoreVal);
2454
2455 if (i + 1 < NumElts) {
2456 StoreVal = OutVals[i + 1];
Justin Holewinskif8f70912013-06-28 17:57:59 +00002457 if (NeedExtend)
2458 StoreVal = DAG.getNode(ISD::ZERO_EXTEND, dl, ExtendedVT, StoreVal);
Justin Holewinski120baee2013-06-28 17:57:55 +00002459 } else {
2460 StoreVal = DAG.getUNDEF(ExtendedVT);
2461 }
2462 Ops.push_back(StoreVal);
2463
2464 if (VecSize == 4) {
2465 Opc = NVPTXISD::StoreRetvalV4;
2466 if (i + 2 < NumElts) {
2467 StoreVal = OutVals[i + 2];
Justin Holewinskif8f70912013-06-28 17:57:59 +00002468 if (NeedExtend)
2469 StoreVal =
2470 DAG.getNode(ISD::ZERO_EXTEND, dl, ExtendedVT, StoreVal);
Justin Holewinski120baee2013-06-28 17:57:55 +00002471 } else {
2472 StoreVal = DAG.getUNDEF(ExtendedVT);
2473 }
2474 Ops.push_back(StoreVal);
2475
2476 if (i + 3 < NumElts) {
2477 StoreVal = OutVals[i + 3];
Justin Holewinskif8f70912013-06-28 17:57:59 +00002478 if (NeedExtend)
2479 StoreVal =
2480 DAG.getNode(ISD::ZERO_EXTEND, dl, ExtendedVT, StoreVal);
Justin Holewinski120baee2013-06-28 17:57:55 +00002481 } else {
2482 StoreVal = DAG.getUNDEF(ExtendedVT);
2483 }
2484 Ops.push_back(StoreVal);
2485 }
2486
Justin Holewinskif8f70912013-06-28 17:57:59 +00002487 // Chain = DAG.getNode(Opc, dl, MVT::Other, &Ops[0], Ops.size());
2488 Chain =
Craig Topper206fcd42014-04-26 19:29:41 +00002489 DAG.getMemIntrinsicNode(Opc, dl, DAG.getVTList(MVT::Other), Ops,
2490 EltVT, MachinePointerInfo());
Justin Holewinski120baee2013-06-28 17:57:55 +00002491 Offset += PerStoreOffset;
2492 }
2493 }
2494 } else {
Justin Holewinskif8f70912013-06-28 17:57:59 +00002495 SmallVector<EVT, 16> ValVTs;
Justin Holewinskib5db95e2014-06-27 18:36:04 +00002496 SmallVector<uint64_t, 16> Offsets;
Mehdi Amini56228da2015-07-09 01:57:34 +00002497 ComputePTXValueVTs(*this, DAG.getDataLayout(), RetTy, ValVTs, &Offsets, 0);
Justin Holewinskif8f70912013-06-28 17:57:59 +00002498 assert(ValVTs.size() == OutVals.size() && "Bad return value decomposition");
2499
Justin Holewinski120baee2013-06-28 17:57:55 +00002500 for (unsigned i = 0, e = Outs.size(); i != e; ++i) {
2501 SDValue theVal = OutVals[i];
Justin Holewinskie04e4bd2013-06-28 17:58:10 +00002502 EVT TheValType = theVal.getValueType();
Justin Holewinski120baee2013-06-28 17:57:55 +00002503 unsigned numElems = 1;
Justin Holewinskie04e4bd2013-06-28 17:58:10 +00002504 if (TheValType.isVector())
2505 numElems = TheValType.getVectorNumElements();
Justin Holewinski120baee2013-06-28 17:57:55 +00002506 for (unsigned j = 0, je = numElems; j != je; ++j) {
Justin Holewinskie04e4bd2013-06-28 17:58:10 +00002507 SDValue TmpVal = theVal;
2508 if (TheValType.isVector())
2509 TmpVal = DAG.getNode(ISD::EXTRACT_VECTOR_ELT, dl,
2510 TheValType.getVectorElementType(), TmpVal,
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +00002511 DAG.getIntPtrConstant(j, dl));
Justin Holewinskie04e4bd2013-06-28 17:58:10 +00002512 EVT TheStoreType = ValVTs[i];
Mehdi Amini44ede332015-07-09 02:09:04 +00002513 if (RetTy->isIntegerTy() && TD.getTypeAllocSizeInBits(RetTy) < 32) {
Justin Holewinskie04e4bd2013-06-28 17:58:10 +00002514 // The following zero-extension is for integer types only, and
2515 // specifically not for aggregates.
2516 TmpVal = DAG.getNode(ISD::ZERO_EXTEND, dl, MVT::i32, TmpVal);
2517 TheStoreType = MVT::i32;
2518 }
2519 else if (TmpVal.getValueType().getSizeInBits() < 16)
2520 TmpVal = DAG.getNode(ISD::ANY_EXTEND, dl, MVT::i16, TmpVal);
2521
Justin Holewinskib5db95e2014-06-27 18:36:04 +00002522 SDValue Ops[] = {
2523 Chain,
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +00002524 DAG.getConstant(Offsets[i], dl, MVT::i32),
Justin Holewinskib5db95e2014-06-27 18:36:04 +00002525 TmpVal };
Justin Holewinskif8f70912013-06-28 17:57:59 +00002526 Chain = DAG.getMemIntrinsicNode(NVPTXISD::StoreRetval, dl,
Craig Topper206fcd42014-04-26 19:29:41 +00002527 DAG.getVTList(MVT::Other), Ops,
2528 TheStoreType,
Justin Holewinskie04e4bd2013-06-28 17:58:10 +00002529 MachinePointerInfo());
Justin Holewinski120baee2013-06-28 17:57:55 +00002530 }
Justin Holewinskiae556d32012-05-04 20:18:50 +00002531 }
2532 }
2533
2534 return DAG.getNode(NVPTXISD::RET_FLAG, dl, MVT::Other, Chain);
2535}
2536
Justin Holewinskif8f70912013-06-28 17:57:59 +00002537
Justin Holewinski0497ab12013-03-30 14:29:21 +00002538void NVPTXTargetLowering::LowerAsmOperandForConstraint(
2539 SDValue Op, std::string &Constraint, std::vector<SDValue> &Ops,
2540 SelectionDAG &DAG) const {
Justin Holewinskiae556d32012-05-04 20:18:50 +00002541 if (Constraint.length() > 1)
2542 return;
2543 else
2544 TargetLowering::LowerAsmOperandForConstraint(Op, Constraint, Ops, DAG);
2545}
2546
2547// NVPTX suuport vector of legal types of any length in Intrinsics because the
2548// NVPTX specific type legalizer
2549// will legalize them to the PTX supported length.
Justin Holewinski0497ab12013-03-30 14:29:21 +00002550bool NVPTXTargetLowering::isTypeSupportedInIntrinsic(MVT VT) const {
Justin Holewinskiae556d32012-05-04 20:18:50 +00002551 if (isTypeLegal(VT))
2552 return true;
2553 if (VT.isVector()) {
2554 MVT eVT = VT.getVectorElementType();
2555 if (isTypeLegal(eVT))
2556 return true;
2557 }
2558 return false;
2559}
2560
Justin Holewinski30d56a72014-04-09 15:39:15 +00002561static unsigned getOpcForTextureInstr(unsigned Intrinsic) {
2562 switch (Intrinsic) {
2563 default:
2564 return 0;
2565
Justin Holewinski9a2350e2014-07-17 11:59:04 +00002566 case Intrinsic::nvvm_tex_1d_v4f32_s32:
2567 return NVPTXISD::Tex1DFloatS32;
Justin Holewinski30d56a72014-04-09 15:39:15 +00002568 case Intrinsic::nvvm_tex_1d_v4f32_f32:
2569 return NVPTXISD::Tex1DFloatFloat;
2570 case Intrinsic::nvvm_tex_1d_level_v4f32_f32:
2571 return NVPTXISD::Tex1DFloatFloatLevel;
2572 case Intrinsic::nvvm_tex_1d_grad_v4f32_f32:
2573 return NVPTXISD::Tex1DFloatFloatGrad;
Justin Holewinski9a2350e2014-07-17 11:59:04 +00002574 case Intrinsic::nvvm_tex_1d_v4s32_s32:
2575 return NVPTXISD::Tex1DS32S32;
2576 case Intrinsic::nvvm_tex_1d_v4s32_f32:
2577 return NVPTXISD::Tex1DS32Float;
2578 case Intrinsic::nvvm_tex_1d_level_v4s32_f32:
2579 return NVPTXISD::Tex1DS32FloatLevel;
2580 case Intrinsic::nvvm_tex_1d_grad_v4s32_f32:
2581 return NVPTXISD::Tex1DS32FloatGrad;
2582 case Intrinsic::nvvm_tex_1d_v4u32_s32:
2583 return NVPTXISD::Tex1DU32S32;
2584 case Intrinsic::nvvm_tex_1d_v4u32_f32:
2585 return NVPTXISD::Tex1DU32Float;
2586 case Intrinsic::nvvm_tex_1d_level_v4u32_f32:
2587 return NVPTXISD::Tex1DU32FloatLevel;
2588 case Intrinsic::nvvm_tex_1d_grad_v4u32_f32:
2589 return NVPTXISD::Tex1DU32FloatGrad;
Justin Holewinski30d56a72014-04-09 15:39:15 +00002590
Justin Holewinski9a2350e2014-07-17 11:59:04 +00002591 case Intrinsic::nvvm_tex_1d_array_v4f32_s32:
2592 return NVPTXISD::Tex1DArrayFloatS32;
Justin Holewinski30d56a72014-04-09 15:39:15 +00002593 case Intrinsic::nvvm_tex_1d_array_v4f32_f32:
2594 return NVPTXISD::Tex1DArrayFloatFloat;
2595 case Intrinsic::nvvm_tex_1d_array_level_v4f32_f32:
2596 return NVPTXISD::Tex1DArrayFloatFloatLevel;
2597 case Intrinsic::nvvm_tex_1d_array_grad_v4f32_f32:
2598 return NVPTXISD::Tex1DArrayFloatFloatGrad;
Justin Holewinski9a2350e2014-07-17 11:59:04 +00002599 case Intrinsic::nvvm_tex_1d_array_v4s32_s32:
2600 return NVPTXISD::Tex1DArrayS32S32;
2601 case Intrinsic::nvvm_tex_1d_array_v4s32_f32:
2602 return NVPTXISD::Tex1DArrayS32Float;
2603 case Intrinsic::nvvm_tex_1d_array_level_v4s32_f32:
2604 return NVPTXISD::Tex1DArrayS32FloatLevel;
2605 case Intrinsic::nvvm_tex_1d_array_grad_v4s32_f32:
2606 return NVPTXISD::Tex1DArrayS32FloatGrad;
2607 case Intrinsic::nvvm_tex_1d_array_v4u32_s32:
2608 return NVPTXISD::Tex1DArrayU32S32;
2609 case Intrinsic::nvvm_tex_1d_array_v4u32_f32:
2610 return NVPTXISD::Tex1DArrayU32Float;
2611 case Intrinsic::nvvm_tex_1d_array_level_v4u32_f32:
2612 return NVPTXISD::Tex1DArrayU32FloatLevel;
2613 case Intrinsic::nvvm_tex_1d_array_grad_v4u32_f32:
2614 return NVPTXISD::Tex1DArrayU32FloatGrad;
Justin Holewinski30d56a72014-04-09 15:39:15 +00002615
Justin Holewinski9a2350e2014-07-17 11:59:04 +00002616 case Intrinsic::nvvm_tex_2d_v4f32_s32:
2617 return NVPTXISD::Tex2DFloatS32;
Justin Holewinski30d56a72014-04-09 15:39:15 +00002618 case Intrinsic::nvvm_tex_2d_v4f32_f32:
2619 return NVPTXISD::Tex2DFloatFloat;
2620 case Intrinsic::nvvm_tex_2d_level_v4f32_f32:
2621 return NVPTXISD::Tex2DFloatFloatLevel;
2622 case Intrinsic::nvvm_tex_2d_grad_v4f32_f32:
2623 return NVPTXISD::Tex2DFloatFloatGrad;
Justin Holewinski9a2350e2014-07-17 11:59:04 +00002624 case Intrinsic::nvvm_tex_2d_v4s32_s32:
2625 return NVPTXISD::Tex2DS32S32;
2626 case Intrinsic::nvvm_tex_2d_v4s32_f32:
2627 return NVPTXISD::Tex2DS32Float;
2628 case Intrinsic::nvvm_tex_2d_level_v4s32_f32:
2629 return NVPTXISD::Tex2DS32FloatLevel;
2630 case Intrinsic::nvvm_tex_2d_grad_v4s32_f32:
2631 return NVPTXISD::Tex2DS32FloatGrad;
2632 case Intrinsic::nvvm_tex_2d_v4u32_s32:
2633 return NVPTXISD::Tex2DU32S32;
2634 case Intrinsic::nvvm_tex_2d_v4u32_f32:
2635 return NVPTXISD::Tex2DU32Float;
2636 case Intrinsic::nvvm_tex_2d_level_v4u32_f32:
2637 return NVPTXISD::Tex2DU32FloatLevel;
2638 case Intrinsic::nvvm_tex_2d_grad_v4u32_f32:
2639 return NVPTXISD::Tex2DU32FloatGrad;
Justin Holewinski30d56a72014-04-09 15:39:15 +00002640
Justin Holewinski9a2350e2014-07-17 11:59:04 +00002641 case Intrinsic::nvvm_tex_2d_array_v4f32_s32:
2642 return NVPTXISD::Tex2DArrayFloatS32;
Justin Holewinski30d56a72014-04-09 15:39:15 +00002643 case Intrinsic::nvvm_tex_2d_array_v4f32_f32:
2644 return NVPTXISD::Tex2DArrayFloatFloat;
2645 case Intrinsic::nvvm_tex_2d_array_level_v4f32_f32:
2646 return NVPTXISD::Tex2DArrayFloatFloatLevel;
2647 case Intrinsic::nvvm_tex_2d_array_grad_v4f32_f32:
2648 return NVPTXISD::Tex2DArrayFloatFloatGrad;
Justin Holewinski9a2350e2014-07-17 11:59:04 +00002649 case Intrinsic::nvvm_tex_2d_array_v4s32_s32:
2650 return NVPTXISD::Tex2DArrayS32S32;
2651 case Intrinsic::nvvm_tex_2d_array_v4s32_f32:
2652 return NVPTXISD::Tex2DArrayS32Float;
2653 case Intrinsic::nvvm_tex_2d_array_level_v4s32_f32:
2654 return NVPTXISD::Tex2DArrayS32FloatLevel;
2655 case Intrinsic::nvvm_tex_2d_array_grad_v4s32_f32:
2656 return NVPTXISD::Tex2DArrayS32FloatGrad;
2657 case Intrinsic::nvvm_tex_2d_array_v4u32_s32:
2658 return NVPTXISD::Tex2DArrayU32S32;
2659 case Intrinsic::nvvm_tex_2d_array_v4u32_f32:
2660 return NVPTXISD::Tex2DArrayU32Float;
2661 case Intrinsic::nvvm_tex_2d_array_level_v4u32_f32:
2662 return NVPTXISD::Tex2DArrayU32FloatLevel;
2663 case Intrinsic::nvvm_tex_2d_array_grad_v4u32_f32:
2664 return NVPTXISD::Tex2DArrayU32FloatGrad;
Justin Holewinski30d56a72014-04-09 15:39:15 +00002665
Justin Holewinski9a2350e2014-07-17 11:59:04 +00002666 case Intrinsic::nvvm_tex_3d_v4f32_s32:
2667 return NVPTXISD::Tex3DFloatS32;
Justin Holewinski30d56a72014-04-09 15:39:15 +00002668 case Intrinsic::nvvm_tex_3d_v4f32_f32:
2669 return NVPTXISD::Tex3DFloatFloat;
2670 case Intrinsic::nvvm_tex_3d_level_v4f32_f32:
2671 return NVPTXISD::Tex3DFloatFloatLevel;
2672 case Intrinsic::nvvm_tex_3d_grad_v4f32_f32:
2673 return NVPTXISD::Tex3DFloatFloatGrad;
Justin Holewinski9a2350e2014-07-17 11:59:04 +00002674 case Intrinsic::nvvm_tex_3d_v4s32_s32:
2675 return NVPTXISD::Tex3DS32S32;
2676 case Intrinsic::nvvm_tex_3d_v4s32_f32:
2677 return NVPTXISD::Tex3DS32Float;
2678 case Intrinsic::nvvm_tex_3d_level_v4s32_f32:
2679 return NVPTXISD::Tex3DS32FloatLevel;
2680 case Intrinsic::nvvm_tex_3d_grad_v4s32_f32:
2681 return NVPTXISD::Tex3DS32FloatGrad;
2682 case Intrinsic::nvvm_tex_3d_v4u32_s32:
2683 return NVPTXISD::Tex3DU32S32;
2684 case Intrinsic::nvvm_tex_3d_v4u32_f32:
2685 return NVPTXISD::Tex3DU32Float;
2686 case Intrinsic::nvvm_tex_3d_level_v4u32_f32:
2687 return NVPTXISD::Tex3DU32FloatLevel;
2688 case Intrinsic::nvvm_tex_3d_grad_v4u32_f32:
2689 return NVPTXISD::Tex3DU32FloatGrad;
2690
2691 case Intrinsic::nvvm_tex_cube_v4f32_f32:
2692 return NVPTXISD::TexCubeFloatFloat;
2693 case Intrinsic::nvvm_tex_cube_level_v4f32_f32:
2694 return NVPTXISD::TexCubeFloatFloatLevel;
2695 case Intrinsic::nvvm_tex_cube_v4s32_f32:
2696 return NVPTXISD::TexCubeS32Float;
2697 case Intrinsic::nvvm_tex_cube_level_v4s32_f32:
2698 return NVPTXISD::TexCubeS32FloatLevel;
2699 case Intrinsic::nvvm_tex_cube_v4u32_f32:
2700 return NVPTXISD::TexCubeU32Float;
2701 case Intrinsic::nvvm_tex_cube_level_v4u32_f32:
2702 return NVPTXISD::TexCubeU32FloatLevel;
2703
2704 case Intrinsic::nvvm_tex_cube_array_v4f32_f32:
2705 return NVPTXISD::TexCubeArrayFloatFloat;
2706 case Intrinsic::nvvm_tex_cube_array_level_v4f32_f32:
2707 return NVPTXISD::TexCubeArrayFloatFloatLevel;
2708 case Intrinsic::nvvm_tex_cube_array_v4s32_f32:
2709 return NVPTXISD::TexCubeArrayS32Float;
2710 case Intrinsic::nvvm_tex_cube_array_level_v4s32_f32:
2711 return NVPTXISD::TexCubeArrayS32FloatLevel;
2712 case Intrinsic::nvvm_tex_cube_array_v4u32_f32:
2713 return NVPTXISD::TexCubeArrayU32Float;
2714 case Intrinsic::nvvm_tex_cube_array_level_v4u32_f32:
2715 return NVPTXISD::TexCubeArrayU32FloatLevel;
2716
2717 case Intrinsic::nvvm_tld4_r_2d_v4f32_f32:
2718 return NVPTXISD::Tld4R2DFloatFloat;
2719 case Intrinsic::nvvm_tld4_g_2d_v4f32_f32:
2720 return NVPTXISD::Tld4G2DFloatFloat;
2721 case Intrinsic::nvvm_tld4_b_2d_v4f32_f32:
2722 return NVPTXISD::Tld4B2DFloatFloat;
2723 case Intrinsic::nvvm_tld4_a_2d_v4f32_f32:
2724 return NVPTXISD::Tld4A2DFloatFloat;
2725 case Intrinsic::nvvm_tld4_r_2d_v4s32_f32:
2726 return NVPTXISD::Tld4R2DS64Float;
2727 case Intrinsic::nvvm_tld4_g_2d_v4s32_f32:
2728 return NVPTXISD::Tld4G2DS64Float;
2729 case Intrinsic::nvvm_tld4_b_2d_v4s32_f32:
2730 return NVPTXISD::Tld4B2DS64Float;
2731 case Intrinsic::nvvm_tld4_a_2d_v4s32_f32:
2732 return NVPTXISD::Tld4A2DS64Float;
2733 case Intrinsic::nvvm_tld4_r_2d_v4u32_f32:
2734 return NVPTXISD::Tld4R2DU64Float;
2735 case Intrinsic::nvvm_tld4_g_2d_v4u32_f32:
2736 return NVPTXISD::Tld4G2DU64Float;
2737 case Intrinsic::nvvm_tld4_b_2d_v4u32_f32:
2738 return NVPTXISD::Tld4B2DU64Float;
2739 case Intrinsic::nvvm_tld4_a_2d_v4u32_f32:
2740 return NVPTXISD::Tld4A2DU64Float;
2741
2742 case Intrinsic::nvvm_tex_unified_1d_v4f32_s32:
2743 return NVPTXISD::TexUnified1DFloatS32;
2744 case Intrinsic::nvvm_tex_unified_1d_v4f32_f32:
2745 return NVPTXISD::TexUnified1DFloatFloat;
2746 case Intrinsic::nvvm_tex_unified_1d_level_v4f32_f32:
2747 return NVPTXISD::TexUnified1DFloatFloatLevel;
2748 case Intrinsic::nvvm_tex_unified_1d_grad_v4f32_f32:
2749 return NVPTXISD::TexUnified1DFloatFloatGrad;
2750 case Intrinsic::nvvm_tex_unified_1d_v4s32_s32:
2751 return NVPTXISD::TexUnified1DS32S32;
2752 case Intrinsic::nvvm_tex_unified_1d_v4s32_f32:
2753 return NVPTXISD::TexUnified1DS32Float;
2754 case Intrinsic::nvvm_tex_unified_1d_level_v4s32_f32:
2755 return NVPTXISD::TexUnified1DS32FloatLevel;
2756 case Intrinsic::nvvm_tex_unified_1d_grad_v4s32_f32:
2757 return NVPTXISD::TexUnified1DS32FloatGrad;
2758 case Intrinsic::nvvm_tex_unified_1d_v4u32_s32:
2759 return NVPTXISD::TexUnified1DU32S32;
2760 case Intrinsic::nvvm_tex_unified_1d_v4u32_f32:
2761 return NVPTXISD::TexUnified1DU32Float;
2762 case Intrinsic::nvvm_tex_unified_1d_level_v4u32_f32:
2763 return NVPTXISD::TexUnified1DU32FloatLevel;
2764 case Intrinsic::nvvm_tex_unified_1d_grad_v4u32_f32:
2765 return NVPTXISD::TexUnified1DU32FloatGrad;
2766
2767 case Intrinsic::nvvm_tex_unified_1d_array_v4f32_s32:
2768 return NVPTXISD::TexUnified1DArrayFloatS32;
2769 case Intrinsic::nvvm_tex_unified_1d_array_v4f32_f32:
2770 return NVPTXISD::TexUnified1DArrayFloatFloat;
2771 case Intrinsic::nvvm_tex_unified_1d_array_level_v4f32_f32:
2772 return NVPTXISD::TexUnified1DArrayFloatFloatLevel;
2773 case Intrinsic::nvvm_tex_unified_1d_array_grad_v4f32_f32:
2774 return NVPTXISD::TexUnified1DArrayFloatFloatGrad;
2775 case Intrinsic::nvvm_tex_unified_1d_array_v4s32_s32:
2776 return NVPTXISD::TexUnified1DArrayS32S32;
2777 case Intrinsic::nvvm_tex_unified_1d_array_v4s32_f32:
2778 return NVPTXISD::TexUnified1DArrayS32Float;
2779 case Intrinsic::nvvm_tex_unified_1d_array_level_v4s32_f32:
2780 return NVPTXISD::TexUnified1DArrayS32FloatLevel;
2781 case Intrinsic::nvvm_tex_unified_1d_array_grad_v4s32_f32:
2782 return NVPTXISD::TexUnified1DArrayS32FloatGrad;
2783 case Intrinsic::nvvm_tex_unified_1d_array_v4u32_s32:
2784 return NVPTXISD::TexUnified1DArrayU32S32;
2785 case Intrinsic::nvvm_tex_unified_1d_array_v4u32_f32:
2786 return NVPTXISD::TexUnified1DArrayU32Float;
2787 case Intrinsic::nvvm_tex_unified_1d_array_level_v4u32_f32:
2788 return NVPTXISD::TexUnified1DArrayU32FloatLevel;
2789 case Intrinsic::nvvm_tex_unified_1d_array_grad_v4u32_f32:
2790 return NVPTXISD::TexUnified1DArrayU32FloatGrad;
2791
2792 case Intrinsic::nvvm_tex_unified_2d_v4f32_s32:
2793 return NVPTXISD::TexUnified2DFloatS32;
2794 case Intrinsic::nvvm_tex_unified_2d_v4f32_f32:
2795 return NVPTXISD::TexUnified2DFloatFloat;
2796 case Intrinsic::nvvm_tex_unified_2d_level_v4f32_f32:
2797 return NVPTXISD::TexUnified2DFloatFloatLevel;
2798 case Intrinsic::nvvm_tex_unified_2d_grad_v4f32_f32:
2799 return NVPTXISD::TexUnified2DFloatFloatGrad;
2800 case Intrinsic::nvvm_tex_unified_2d_v4s32_s32:
2801 return NVPTXISD::TexUnified2DS32S32;
2802 case Intrinsic::nvvm_tex_unified_2d_v4s32_f32:
2803 return NVPTXISD::TexUnified2DS32Float;
2804 case Intrinsic::nvvm_tex_unified_2d_level_v4s32_f32:
2805 return NVPTXISD::TexUnified2DS32FloatLevel;
2806 case Intrinsic::nvvm_tex_unified_2d_grad_v4s32_f32:
2807 return NVPTXISD::TexUnified2DS32FloatGrad;
2808 case Intrinsic::nvvm_tex_unified_2d_v4u32_s32:
2809 return NVPTXISD::TexUnified2DU32S32;
2810 case Intrinsic::nvvm_tex_unified_2d_v4u32_f32:
2811 return NVPTXISD::TexUnified2DU32Float;
2812 case Intrinsic::nvvm_tex_unified_2d_level_v4u32_f32:
2813 return NVPTXISD::TexUnified2DU32FloatLevel;
2814 case Intrinsic::nvvm_tex_unified_2d_grad_v4u32_f32:
2815 return NVPTXISD::TexUnified2DU32FloatGrad;
2816
2817 case Intrinsic::nvvm_tex_unified_2d_array_v4f32_s32:
2818 return NVPTXISD::TexUnified2DArrayFloatS32;
2819 case Intrinsic::nvvm_tex_unified_2d_array_v4f32_f32:
2820 return NVPTXISD::TexUnified2DArrayFloatFloat;
2821 case Intrinsic::nvvm_tex_unified_2d_array_level_v4f32_f32:
2822 return NVPTXISD::TexUnified2DArrayFloatFloatLevel;
2823 case Intrinsic::nvvm_tex_unified_2d_array_grad_v4f32_f32:
2824 return NVPTXISD::TexUnified2DArrayFloatFloatGrad;
2825 case Intrinsic::nvvm_tex_unified_2d_array_v4s32_s32:
2826 return NVPTXISD::TexUnified2DArrayS32S32;
2827 case Intrinsic::nvvm_tex_unified_2d_array_v4s32_f32:
2828 return NVPTXISD::TexUnified2DArrayS32Float;
2829 case Intrinsic::nvvm_tex_unified_2d_array_level_v4s32_f32:
2830 return NVPTXISD::TexUnified2DArrayS32FloatLevel;
2831 case Intrinsic::nvvm_tex_unified_2d_array_grad_v4s32_f32:
2832 return NVPTXISD::TexUnified2DArrayS32FloatGrad;
2833 case Intrinsic::nvvm_tex_unified_2d_array_v4u32_s32:
2834 return NVPTXISD::TexUnified2DArrayU32S32;
2835 case Intrinsic::nvvm_tex_unified_2d_array_v4u32_f32:
2836 return NVPTXISD::TexUnified2DArrayU32Float;
2837 case Intrinsic::nvvm_tex_unified_2d_array_level_v4u32_f32:
2838 return NVPTXISD::TexUnified2DArrayU32FloatLevel;
2839 case Intrinsic::nvvm_tex_unified_2d_array_grad_v4u32_f32:
2840 return NVPTXISD::TexUnified2DArrayU32FloatGrad;
2841
2842 case Intrinsic::nvvm_tex_unified_3d_v4f32_s32:
2843 return NVPTXISD::TexUnified3DFloatS32;
2844 case Intrinsic::nvvm_tex_unified_3d_v4f32_f32:
2845 return NVPTXISD::TexUnified3DFloatFloat;
2846 case Intrinsic::nvvm_tex_unified_3d_level_v4f32_f32:
2847 return NVPTXISD::TexUnified3DFloatFloatLevel;
2848 case Intrinsic::nvvm_tex_unified_3d_grad_v4f32_f32:
2849 return NVPTXISD::TexUnified3DFloatFloatGrad;
2850 case Intrinsic::nvvm_tex_unified_3d_v4s32_s32:
2851 return NVPTXISD::TexUnified3DS32S32;
2852 case Intrinsic::nvvm_tex_unified_3d_v4s32_f32:
2853 return NVPTXISD::TexUnified3DS32Float;
2854 case Intrinsic::nvvm_tex_unified_3d_level_v4s32_f32:
2855 return NVPTXISD::TexUnified3DS32FloatLevel;
2856 case Intrinsic::nvvm_tex_unified_3d_grad_v4s32_f32:
2857 return NVPTXISD::TexUnified3DS32FloatGrad;
2858 case Intrinsic::nvvm_tex_unified_3d_v4u32_s32:
2859 return NVPTXISD::TexUnified3DU32S32;
2860 case Intrinsic::nvvm_tex_unified_3d_v4u32_f32:
2861 return NVPTXISD::TexUnified3DU32Float;
2862 case Intrinsic::nvvm_tex_unified_3d_level_v4u32_f32:
2863 return NVPTXISD::TexUnified3DU32FloatLevel;
2864 case Intrinsic::nvvm_tex_unified_3d_grad_v4u32_f32:
2865 return NVPTXISD::TexUnified3DU32FloatGrad;
2866
2867 case Intrinsic::nvvm_tex_unified_cube_v4f32_f32:
2868 return NVPTXISD::TexUnifiedCubeFloatFloat;
2869 case Intrinsic::nvvm_tex_unified_cube_level_v4f32_f32:
2870 return NVPTXISD::TexUnifiedCubeFloatFloatLevel;
2871 case Intrinsic::nvvm_tex_unified_cube_v4s32_f32:
2872 return NVPTXISD::TexUnifiedCubeS32Float;
2873 case Intrinsic::nvvm_tex_unified_cube_level_v4s32_f32:
2874 return NVPTXISD::TexUnifiedCubeS32FloatLevel;
2875 case Intrinsic::nvvm_tex_unified_cube_v4u32_f32:
2876 return NVPTXISD::TexUnifiedCubeU32Float;
2877 case Intrinsic::nvvm_tex_unified_cube_level_v4u32_f32:
2878 return NVPTXISD::TexUnifiedCubeU32FloatLevel;
2879
2880 case Intrinsic::nvvm_tex_unified_cube_array_v4f32_f32:
2881 return NVPTXISD::TexUnifiedCubeArrayFloatFloat;
2882 case Intrinsic::nvvm_tex_unified_cube_array_level_v4f32_f32:
2883 return NVPTXISD::TexUnifiedCubeArrayFloatFloatLevel;
2884 case Intrinsic::nvvm_tex_unified_cube_array_v4s32_f32:
2885 return NVPTXISD::TexUnifiedCubeArrayS32Float;
2886 case Intrinsic::nvvm_tex_unified_cube_array_level_v4s32_f32:
2887 return NVPTXISD::TexUnifiedCubeArrayS32FloatLevel;
2888 case Intrinsic::nvvm_tex_unified_cube_array_v4u32_f32:
2889 return NVPTXISD::TexUnifiedCubeArrayU32Float;
2890 case Intrinsic::nvvm_tex_unified_cube_array_level_v4u32_f32:
2891 return NVPTXISD::TexUnifiedCubeArrayU32FloatLevel;
2892
2893 case Intrinsic::nvvm_tld4_unified_r_2d_v4f32_f32:
2894 return NVPTXISD::Tld4UnifiedR2DFloatFloat;
2895 case Intrinsic::nvvm_tld4_unified_g_2d_v4f32_f32:
2896 return NVPTXISD::Tld4UnifiedG2DFloatFloat;
2897 case Intrinsic::nvvm_tld4_unified_b_2d_v4f32_f32:
2898 return NVPTXISD::Tld4UnifiedB2DFloatFloat;
2899 case Intrinsic::nvvm_tld4_unified_a_2d_v4f32_f32:
2900 return NVPTXISD::Tld4UnifiedA2DFloatFloat;
2901 case Intrinsic::nvvm_tld4_unified_r_2d_v4s32_f32:
2902 return NVPTXISD::Tld4UnifiedR2DS64Float;
2903 case Intrinsic::nvvm_tld4_unified_g_2d_v4s32_f32:
2904 return NVPTXISD::Tld4UnifiedG2DS64Float;
2905 case Intrinsic::nvvm_tld4_unified_b_2d_v4s32_f32:
2906 return NVPTXISD::Tld4UnifiedB2DS64Float;
2907 case Intrinsic::nvvm_tld4_unified_a_2d_v4s32_f32:
2908 return NVPTXISD::Tld4UnifiedA2DS64Float;
2909 case Intrinsic::nvvm_tld4_unified_r_2d_v4u32_f32:
2910 return NVPTXISD::Tld4UnifiedR2DU64Float;
2911 case Intrinsic::nvvm_tld4_unified_g_2d_v4u32_f32:
2912 return NVPTXISD::Tld4UnifiedG2DU64Float;
2913 case Intrinsic::nvvm_tld4_unified_b_2d_v4u32_f32:
2914 return NVPTXISD::Tld4UnifiedB2DU64Float;
2915 case Intrinsic::nvvm_tld4_unified_a_2d_v4u32_f32:
2916 return NVPTXISD::Tld4UnifiedA2DU64Float;
Justin Holewinski30d56a72014-04-09 15:39:15 +00002917 }
2918}
2919
2920static unsigned getOpcForSurfaceInstr(unsigned Intrinsic) {
2921 switch (Intrinsic) {
2922 default:
2923 return 0;
Justin Holewinski9a2350e2014-07-17 11:59:04 +00002924 case Intrinsic::nvvm_suld_1d_i8_clamp:
2925 return NVPTXISD::Suld1DI8Clamp;
2926 case Intrinsic::nvvm_suld_1d_i16_clamp:
2927 return NVPTXISD::Suld1DI16Clamp;
2928 case Intrinsic::nvvm_suld_1d_i32_clamp:
2929 return NVPTXISD::Suld1DI32Clamp;
2930 case Intrinsic::nvvm_suld_1d_i64_clamp:
2931 return NVPTXISD::Suld1DI64Clamp;
2932 case Intrinsic::nvvm_suld_1d_v2i8_clamp:
2933 return NVPTXISD::Suld1DV2I8Clamp;
2934 case Intrinsic::nvvm_suld_1d_v2i16_clamp:
2935 return NVPTXISD::Suld1DV2I16Clamp;
2936 case Intrinsic::nvvm_suld_1d_v2i32_clamp:
2937 return NVPTXISD::Suld1DV2I32Clamp;
2938 case Intrinsic::nvvm_suld_1d_v2i64_clamp:
2939 return NVPTXISD::Suld1DV2I64Clamp;
2940 case Intrinsic::nvvm_suld_1d_v4i8_clamp:
2941 return NVPTXISD::Suld1DV4I8Clamp;
2942 case Intrinsic::nvvm_suld_1d_v4i16_clamp:
2943 return NVPTXISD::Suld1DV4I16Clamp;
2944 case Intrinsic::nvvm_suld_1d_v4i32_clamp:
2945 return NVPTXISD::Suld1DV4I32Clamp;
2946 case Intrinsic::nvvm_suld_1d_array_i8_clamp:
2947 return NVPTXISD::Suld1DArrayI8Clamp;
2948 case Intrinsic::nvvm_suld_1d_array_i16_clamp:
2949 return NVPTXISD::Suld1DArrayI16Clamp;
2950 case Intrinsic::nvvm_suld_1d_array_i32_clamp:
2951 return NVPTXISD::Suld1DArrayI32Clamp;
2952 case Intrinsic::nvvm_suld_1d_array_i64_clamp:
2953 return NVPTXISD::Suld1DArrayI64Clamp;
2954 case Intrinsic::nvvm_suld_1d_array_v2i8_clamp:
2955 return NVPTXISD::Suld1DArrayV2I8Clamp;
2956 case Intrinsic::nvvm_suld_1d_array_v2i16_clamp:
2957 return NVPTXISD::Suld1DArrayV2I16Clamp;
2958 case Intrinsic::nvvm_suld_1d_array_v2i32_clamp:
2959 return NVPTXISD::Suld1DArrayV2I32Clamp;
2960 case Intrinsic::nvvm_suld_1d_array_v2i64_clamp:
2961 return NVPTXISD::Suld1DArrayV2I64Clamp;
2962 case Intrinsic::nvvm_suld_1d_array_v4i8_clamp:
2963 return NVPTXISD::Suld1DArrayV4I8Clamp;
2964 case Intrinsic::nvvm_suld_1d_array_v4i16_clamp:
2965 return NVPTXISD::Suld1DArrayV4I16Clamp;
2966 case Intrinsic::nvvm_suld_1d_array_v4i32_clamp:
2967 return NVPTXISD::Suld1DArrayV4I32Clamp;
2968 case Intrinsic::nvvm_suld_2d_i8_clamp:
2969 return NVPTXISD::Suld2DI8Clamp;
2970 case Intrinsic::nvvm_suld_2d_i16_clamp:
2971 return NVPTXISD::Suld2DI16Clamp;
2972 case Intrinsic::nvvm_suld_2d_i32_clamp:
2973 return NVPTXISD::Suld2DI32Clamp;
2974 case Intrinsic::nvvm_suld_2d_i64_clamp:
2975 return NVPTXISD::Suld2DI64Clamp;
2976 case Intrinsic::nvvm_suld_2d_v2i8_clamp:
2977 return NVPTXISD::Suld2DV2I8Clamp;
2978 case Intrinsic::nvvm_suld_2d_v2i16_clamp:
2979 return NVPTXISD::Suld2DV2I16Clamp;
2980 case Intrinsic::nvvm_suld_2d_v2i32_clamp:
2981 return NVPTXISD::Suld2DV2I32Clamp;
2982 case Intrinsic::nvvm_suld_2d_v2i64_clamp:
2983 return NVPTXISD::Suld2DV2I64Clamp;
2984 case Intrinsic::nvvm_suld_2d_v4i8_clamp:
2985 return NVPTXISD::Suld2DV4I8Clamp;
2986 case Intrinsic::nvvm_suld_2d_v4i16_clamp:
2987 return NVPTXISD::Suld2DV4I16Clamp;
2988 case Intrinsic::nvvm_suld_2d_v4i32_clamp:
2989 return NVPTXISD::Suld2DV4I32Clamp;
2990 case Intrinsic::nvvm_suld_2d_array_i8_clamp:
2991 return NVPTXISD::Suld2DArrayI8Clamp;
2992 case Intrinsic::nvvm_suld_2d_array_i16_clamp:
2993 return NVPTXISD::Suld2DArrayI16Clamp;
2994 case Intrinsic::nvvm_suld_2d_array_i32_clamp:
2995 return NVPTXISD::Suld2DArrayI32Clamp;
2996 case Intrinsic::nvvm_suld_2d_array_i64_clamp:
2997 return NVPTXISD::Suld2DArrayI64Clamp;
2998 case Intrinsic::nvvm_suld_2d_array_v2i8_clamp:
2999 return NVPTXISD::Suld2DArrayV2I8Clamp;
3000 case Intrinsic::nvvm_suld_2d_array_v2i16_clamp:
3001 return NVPTXISD::Suld2DArrayV2I16Clamp;
3002 case Intrinsic::nvvm_suld_2d_array_v2i32_clamp:
3003 return NVPTXISD::Suld2DArrayV2I32Clamp;
3004 case Intrinsic::nvvm_suld_2d_array_v2i64_clamp:
3005 return NVPTXISD::Suld2DArrayV2I64Clamp;
3006 case Intrinsic::nvvm_suld_2d_array_v4i8_clamp:
3007 return NVPTXISD::Suld2DArrayV4I8Clamp;
3008 case Intrinsic::nvvm_suld_2d_array_v4i16_clamp:
3009 return NVPTXISD::Suld2DArrayV4I16Clamp;
3010 case Intrinsic::nvvm_suld_2d_array_v4i32_clamp:
3011 return NVPTXISD::Suld2DArrayV4I32Clamp;
3012 case Intrinsic::nvvm_suld_3d_i8_clamp:
3013 return NVPTXISD::Suld3DI8Clamp;
3014 case Intrinsic::nvvm_suld_3d_i16_clamp:
3015 return NVPTXISD::Suld3DI16Clamp;
3016 case Intrinsic::nvvm_suld_3d_i32_clamp:
3017 return NVPTXISD::Suld3DI32Clamp;
3018 case Intrinsic::nvvm_suld_3d_i64_clamp:
3019 return NVPTXISD::Suld3DI64Clamp;
3020 case Intrinsic::nvvm_suld_3d_v2i8_clamp:
3021 return NVPTXISD::Suld3DV2I8Clamp;
3022 case Intrinsic::nvvm_suld_3d_v2i16_clamp:
3023 return NVPTXISD::Suld3DV2I16Clamp;
3024 case Intrinsic::nvvm_suld_3d_v2i32_clamp:
3025 return NVPTXISD::Suld3DV2I32Clamp;
3026 case Intrinsic::nvvm_suld_3d_v2i64_clamp:
3027 return NVPTXISD::Suld3DV2I64Clamp;
3028 case Intrinsic::nvvm_suld_3d_v4i8_clamp:
3029 return NVPTXISD::Suld3DV4I8Clamp;
3030 case Intrinsic::nvvm_suld_3d_v4i16_clamp:
3031 return NVPTXISD::Suld3DV4I16Clamp;
3032 case Intrinsic::nvvm_suld_3d_v4i32_clamp:
3033 return NVPTXISD::Suld3DV4I32Clamp;
Justin Holewinski30d56a72014-04-09 15:39:15 +00003034 case Intrinsic::nvvm_suld_1d_i8_trap:
3035 return NVPTXISD::Suld1DI8Trap;
3036 case Intrinsic::nvvm_suld_1d_i16_trap:
3037 return NVPTXISD::Suld1DI16Trap;
3038 case Intrinsic::nvvm_suld_1d_i32_trap:
3039 return NVPTXISD::Suld1DI32Trap;
Justin Holewinski9a2350e2014-07-17 11:59:04 +00003040 case Intrinsic::nvvm_suld_1d_i64_trap:
3041 return NVPTXISD::Suld1DI64Trap;
Justin Holewinski30d56a72014-04-09 15:39:15 +00003042 case Intrinsic::nvvm_suld_1d_v2i8_trap:
3043 return NVPTXISD::Suld1DV2I8Trap;
3044 case Intrinsic::nvvm_suld_1d_v2i16_trap:
3045 return NVPTXISD::Suld1DV2I16Trap;
3046 case Intrinsic::nvvm_suld_1d_v2i32_trap:
3047 return NVPTXISD::Suld1DV2I32Trap;
Justin Holewinski9a2350e2014-07-17 11:59:04 +00003048 case Intrinsic::nvvm_suld_1d_v2i64_trap:
3049 return NVPTXISD::Suld1DV2I64Trap;
Justin Holewinski30d56a72014-04-09 15:39:15 +00003050 case Intrinsic::nvvm_suld_1d_v4i8_trap:
3051 return NVPTXISD::Suld1DV4I8Trap;
3052 case Intrinsic::nvvm_suld_1d_v4i16_trap:
3053 return NVPTXISD::Suld1DV4I16Trap;
3054 case Intrinsic::nvvm_suld_1d_v4i32_trap:
3055 return NVPTXISD::Suld1DV4I32Trap;
3056 case Intrinsic::nvvm_suld_1d_array_i8_trap:
3057 return NVPTXISD::Suld1DArrayI8Trap;
3058 case Intrinsic::nvvm_suld_1d_array_i16_trap:
3059 return NVPTXISD::Suld1DArrayI16Trap;
3060 case Intrinsic::nvvm_suld_1d_array_i32_trap:
3061 return NVPTXISD::Suld1DArrayI32Trap;
Justin Holewinski9a2350e2014-07-17 11:59:04 +00003062 case Intrinsic::nvvm_suld_1d_array_i64_trap:
3063 return NVPTXISD::Suld1DArrayI64Trap;
Justin Holewinski30d56a72014-04-09 15:39:15 +00003064 case Intrinsic::nvvm_suld_1d_array_v2i8_trap:
3065 return NVPTXISD::Suld1DArrayV2I8Trap;
3066 case Intrinsic::nvvm_suld_1d_array_v2i16_trap:
3067 return NVPTXISD::Suld1DArrayV2I16Trap;
3068 case Intrinsic::nvvm_suld_1d_array_v2i32_trap:
3069 return NVPTXISD::Suld1DArrayV2I32Trap;
Justin Holewinski9a2350e2014-07-17 11:59:04 +00003070 case Intrinsic::nvvm_suld_1d_array_v2i64_trap:
3071 return NVPTXISD::Suld1DArrayV2I64Trap;
Justin Holewinski30d56a72014-04-09 15:39:15 +00003072 case Intrinsic::nvvm_suld_1d_array_v4i8_trap:
3073 return NVPTXISD::Suld1DArrayV4I8Trap;
3074 case Intrinsic::nvvm_suld_1d_array_v4i16_trap:
3075 return NVPTXISD::Suld1DArrayV4I16Trap;
3076 case Intrinsic::nvvm_suld_1d_array_v4i32_trap:
3077 return NVPTXISD::Suld1DArrayV4I32Trap;
3078 case Intrinsic::nvvm_suld_2d_i8_trap:
3079 return NVPTXISD::Suld2DI8Trap;
3080 case Intrinsic::nvvm_suld_2d_i16_trap:
3081 return NVPTXISD::Suld2DI16Trap;
3082 case Intrinsic::nvvm_suld_2d_i32_trap:
3083 return NVPTXISD::Suld2DI32Trap;
Justin Holewinski9a2350e2014-07-17 11:59:04 +00003084 case Intrinsic::nvvm_suld_2d_i64_trap:
3085 return NVPTXISD::Suld2DI64Trap;
Justin Holewinski30d56a72014-04-09 15:39:15 +00003086 case Intrinsic::nvvm_suld_2d_v2i8_trap:
3087 return NVPTXISD::Suld2DV2I8Trap;
3088 case Intrinsic::nvvm_suld_2d_v2i16_trap:
3089 return NVPTXISD::Suld2DV2I16Trap;
3090 case Intrinsic::nvvm_suld_2d_v2i32_trap:
3091 return NVPTXISD::Suld2DV2I32Trap;
Justin Holewinski9a2350e2014-07-17 11:59:04 +00003092 case Intrinsic::nvvm_suld_2d_v2i64_trap:
3093 return NVPTXISD::Suld2DV2I64Trap;
Justin Holewinski30d56a72014-04-09 15:39:15 +00003094 case Intrinsic::nvvm_suld_2d_v4i8_trap:
3095 return NVPTXISD::Suld2DV4I8Trap;
3096 case Intrinsic::nvvm_suld_2d_v4i16_trap:
3097 return NVPTXISD::Suld2DV4I16Trap;
3098 case Intrinsic::nvvm_suld_2d_v4i32_trap:
3099 return NVPTXISD::Suld2DV4I32Trap;
3100 case Intrinsic::nvvm_suld_2d_array_i8_trap:
3101 return NVPTXISD::Suld2DArrayI8Trap;
3102 case Intrinsic::nvvm_suld_2d_array_i16_trap:
3103 return NVPTXISD::Suld2DArrayI16Trap;
3104 case Intrinsic::nvvm_suld_2d_array_i32_trap:
3105 return NVPTXISD::Suld2DArrayI32Trap;
Justin Holewinski9a2350e2014-07-17 11:59:04 +00003106 case Intrinsic::nvvm_suld_2d_array_i64_trap:
3107 return NVPTXISD::Suld2DArrayI64Trap;
Justin Holewinski30d56a72014-04-09 15:39:15 +00003108 case Intrinsic::nvvm_suld_2d_array_v2i8_trap:
3109 return NVPTXISD::Suld2DArrayV2I8Trap;
3110 case Intrinsic::nvvm_suld_2d_array_v2i16_trap:
3111 return NVPTXISD::Suld2DArrayV2I16Trap;
3112 case Intrinsic::nvvm_suld_2d_array_v2i32_trap:
3113 return NVPTXISD::Suld2DArrayV2I32Trap;
Justin Holewinski9a2350e2014-07-17 11:59:04 +00003114 case Intrinsic::nvvm_suld_2d_array_v2i64_trap:
3115 return NVPTXISD::Suld2DArrayV2I64Trap;
Justin Holewinski30d56a72014-04-09 15:39:15 +00003116 case Intrinsic::nvvm_suld_2d_array_v4i8_trap:
3117 return NVPTXISD::Suld2DArrayV4I8Trap;
3118 case Intrinsic::nvvm_suld_2d_array_v4i16_trap:
3119 return NVPTXISD::Suld2DArrayV4I16Trap;
3120 case Intrinsic::nvvm_suld_2d_array_v4i32_trap:
3121 return NVPTXISD::Suld2DArrayV4I32Trap;
3122 case Intrinsic::nvvm_suld_3d_i8_trap:
3123 return NVPTXISD::Suld3DI8Trap;
3124 case Intrinsic::nvvm_suld_3d_i16_trap:
3125 return NVPTXISD::Suld3DI16Trap;
3126 case Intrinsic::nvvm_suld_3d_i32_trap:
3127 return NVPTXISD::Suld3DI32Trap;
Justin Holewinski9a2350e2014-07-17 11:59:04 +00003128 case Intrinsic::nvvm_suld_3d_i64_trap:
3129 return NVPTXISD::Suld3DI64Trap;
Justin Holewinski30d56a72014-04-09 15:39:15 +00003130 case Intrinsic::nvvm_suld_3d_v2i8_trap:
3131 return NVPTXISD::Suld3DV2I8Trap;
3132 case Intrinsic::nvvm_suld_3d_v2i16_trap:
3133 return NVPTXISD::Suld3DV2I16Trap;
3134 case Intrinsic::nvvm_suld_3d_v2i32_trap:
3135 return NVPTXISD::Suld3DV2I32Trap;
Justin Holewinski9a2350e2014-07-17 11:59:04 +00003136 case Intrinsic::nvvm_suld_3d_v2i64_trap:
3137 return NVPTXISD::Suld3DV2I64Trap;
Justin Holewinski30d56a72014-04-09 15:39:15 +00003138 case Intrinsic::nvvm_suld_3d_v4i8_trap:
3139 return NVPTXISD::Suld3DV4I8Trap;
3140 case Intrinsic::nvvm_suld_3d_v4i16_trap:
3141 return NVPTXISD::Suld3DV4I16Trap;
3142 case Intrinsic::nvvm_suld_3d_v4i32_trap:
3143 return NVPTXISD::Suld3DV4I32Trap;
Justin Holewinski9a2350e2014-07-17 11:59:04 +00003144 case Intrinsic::nvvm_suld_1d_i8_zero:
3145 return NVPTXISD::Suld1DI8Zero;
3146 case Intrinsic::nvvm_suld_1d_i16_zero:
3147 return NVPTXISD::Suld1DI16Zero;
3148 case Intrinsic::nvvm_suld_1d_i32_zero:
3149 return NVPTXISD::Suld1DI32Zero;
3150 case Intrinsic::nvvm_suld_1d_i64_zero:
3151 return NVPTXISD::Suld1DI64Zero;
3152 case Intrinsic::nvvm_suld_1d_v2i8_zero:
3153 return NVPTXISD::Suld1DV2I8Zero;
3154 case Intrinsic::nvvm_suld_1d_v2i16_zero:
3155 return NVPTXISD::Suld1DV2I16Zero;
3156 case Intrinsic::nvvm_suld_1d_v2i32_zero:
3157 return NVPTXISD::Suld1DV2I32Zero;
3158 case Intrinsic::nvvm_suld_1d_v2i64_zero:
3159 return NVPTXISD::Suld1DV2I64Zero;
3160 case Intrinsic::nvvm_suld_1d_v4i8_zero:
3161 return NVPTXISD::Suld1DV4I8Zero;
3162 case Intrinsic::nvvm_suld_1d_v4i16_zero:
3163 return NVPTXISD::Suld1DV4I16Zero;
3164 case Intrinsic::nvvm_suld_1d_v4i32_zero:
3165 return NVPTXISD::Suld1DV4I32Zero;
3166 case Intrinsic::nvvm_suld_1d_array_i8_zero:
3167 return NVPTXISD::Suld1DArrayI8Zero;
3168 case Intrinsic::nvvm_suld_1d_array_i16_zero:
3169 return NVPTXISD::Suld1DArrayI16Zero;
3170 case Intrinsic::nvvm_suld_1d_array_i32_zero:
3171 return NVPTXISD::Suld1DArrayI32Zero;
3172 case Intrinsic::nvvm_suld_1d_array_i64_zero:
3173 return NVPTXISD::Suld1DArrayI64Zero;
3174 case Intrinsic::nvvm_suld_1d_array_v2i8_zero:
3175 return NVPTXISD::Suld1DArrayV2I8Zero;
3176 case Intrinsic::nvvm_suld_1d_array_v2i16_zero:
3177 return NVPTXISD::Suld1DArrayV2I16Zero;
3178 case Intrinsic::nvvm_suld_1d_array_v2i32_zero:
3179 return NVPTXISD::Suld1DArrayV2I32Zero;
3180 case Intrinsic::nvvm_suld_1d_array_v2i64_zero:
3181 return NVPTXISD::Suld1DArrayV2I64Zero;
3182 case Intrinsic::nvvm_suld_1d_array_v4i8_zero:
3183 return NVPTXISD::Suld1DArrayV4I8Zero;
3184 case Intrinsic::nvvm_suld_1d_array_v4i16_zero:
3185 return NVPTXISD::Suld1DArrayV4I16Zero;
3186 case Intrinsic::nvvm_suld_1d_array_v4i32_zero:
3187 return NVPTXISD::Suld1DArrayV4I32Zero;
3188 case Intrinsic::nvvm_suld_2d_i8_zero:
3189 return NVPTXISD::Suld2DI8Zero;
3190 case Intrinsic::nvvm_suld_2d_i16_zero:
3191 return NVPTXISD::Suld2DI16Zero;
3192 case Intrinsic::nvvm_suld_2d_i32_zero:
3193 return NVPTXISD::Suld2DI32Zero;
3194 case Intrinsic::nvvm_suld_2d_i64_zero:
3195 return NVPTXISD::Suld2DI64Zero;
3196 case Intrinsic::nvvm_suld_2d_v2i8_zero:
3197 return NVPTXISD::Suld2DV2I8Zero;
3198 case Intrinsic::nvvm_suld_2d_v2i16_zero:
3199 return NVPTXISD::Suld2DV2I16Zero;
3200 case Intrinsic::nvvm_suld_2d_v2i32_zero:
3201 return NVPTXISD::Suld2DV2I32Zero;
3202 case Intrinsic::nvvm_suld_2d_v2i64_zero:
3203 return NVPTXISD::Suld2DV2I64Zero;
3204 case Intrinsic::nvvm_suld_2d_v4i8_zero:
3205 return NVPTXISD::Suld2DV4I8Zero;
3206 case Intrinsic::nvvm_suld_2d_v4i16_zero:
3207 return NVPTXISD::Suld2DV4I16Zero;
3208 case Intrinsic::nvvm_suld_2d_v4i32_zero:
3209 return NVPTXISD::Suld2DV4I32Zero;
3210 case Intrinsic::nvvm_suld_2d_array_i8_zero:
3211 return NVPTXISD::Suld2DArrayI8Zero;
3212 case Intrinsic::nvvm_suld_2d_array_i16_zero:
3213 return NVPTXISD::Suld2DArrayI16Zero;
3214 case Intrinsic::nvvm_suld_2d_array_i32_zero:
3215 return NVPTXISD::Suld2DArrayI32Zero;
3216 case Intrinsic::nvvm_suld_2d_array_i64_zero:
3217 return NVPTXISD::Suld2DArrayI64Zero;
3218 case Intrinsic::nvvm_suld_2d_array_v2i8_zero:
3219 return NVPTXISD::Suld2DArrayV2I8Zero;
3220 case Intrinsic::nvvm_suld_2d_array_v2i16_zero:
3221 return NVPTXISD::Suld2DArrayV2I16Zero;
3222 case Intrinsic::nvvm_suld_2d_array_v2i32_zero:
3223 return NVPTXISD::Suld2DArrayV2I32Zero;
3224 case Intrinsic::nvvm_suld_2d_array_v2i64_zero:
3225 return NVPTXISD::Suld2DArrayV2I64Zero;
3226 case Intrinsic::nvvm_suld_2d_array_v4i8_zero:
3227 return NVPTXISD::Suld2DArrayV4I8Zero;
3228 case Intrinsic::nvvm_suld_2d_array_v4i16_zero:
3229 return NVPTXISD::Suld2DArrayV4I16Zero;
3230 case Intrinsic::nvvm_suld_2d_array_v4i32_zero:
3231 return NVPTXISD::Suld2DArrayV4I32Zero;
3232 case Intrinsic::nvvm_suld_3d_i8_zero:
3233 return NVPTXISD::Suld3DI8Zero;
3234 case Intrinsic::nvvm_suld_3d_i16_zero:
3235 return NVPTXISD::Suld3DI16Zero;
3236 case Intrinsic::nvvm_suld_3d_i32_zero:
3237 return NVPTXISD::Suld3DI32Zero;
3238 case Intrinsic::nvvm_suld_3d_i64_zero:
3239 return NVPTXISD::Suld3DI64Zero;
3240 case Intrinsic::nvvm_suld_3d_v2i8_zero:
3241 return NVPTXISD::Suld3DV2I8Zero;
3242 case Intrinsic::nvvm_suld_3d_v2i16_zero:
3243 return NVPTXISD::Suld3DV2I16Zero;
3244 case Intrinsic::nvvm_suld_3d_v2i32_zero:
3245 return NVPTXISD::Suld3DV2I32Zero;
3246 case Intrinsic::nvvm_suld_3d_v2i64_zero:
3247 return NVPTXISD::Suld3DV2I64Zero;
3248 case Intrinsic::nvvm_suld_3d_v4i8_zero:
3249 return NVPTXISD::Suld3DV4I8Zero;
3250 case Intrinsic::nvvm_suld_3d_v4i16_zero:
3251 return NVPTXISD::Suld3DV4I16Zero;
3252 case Intrinsic::nvvm_suld_3d_v4i32_zero:
3253 return NVPTXISD::Suld3DV4I32Zero;
Justin Holewinski30d56a72014-04-09 15:39:15 +00003254 }
3255}
3256
Justin Holewinskiae556d32012-05-04 20:18:50 +00003257// llvm.ptx.memcpy.const and llvm.ptx.memmove.const need to be modeled as
3258// TgtMemIntrinsic
3259// because we need the information that is only available in the "Value" type
3260// of destination
3261// pointer. In particular, the address space information.
Justin Holewinski0497ab12013-03-30 14:29:21 +00003262bool NVPTXTargetLowering::getTgtMemIntrinsic(
3263 IntrinsicInfo &Info, const CallInst &I, unsigned Intrinsic) const {
Justin Holewinskiae556d32012-05-04 20:18:50 +00003264 switch (Intrinsic) {
3265 default:
3266 return false;
3267
3268 case Intrinsic::nvvm_atomic_load_add_f32:
3269 Info.opc = ISD::INTRINSIC_W_CHAIN;
3270 Info.memVT = MVT::f32;
3271 Info.ptrVal = I.getArgOperand(0);
3272 Info.offset = 0;
3273 Info.vol = 0;
3274 Info.readMem = true;
3275 Info.writeMem = true;
3276 Info.align = 0;
3277 return true;
3278
3279 case Intrinsic::nvvm_atomic_load_inc_32:
3280 case Intrinsic::nvvm_atomic_load_dec_32:
3281 Info.opc = ISD::INTRINSIC_W_CHAIN;
3282 Info.memVT = MVT::i32;
3283 Info.ptrVal = I.getArgOperand(0);
3284 Info.offset = 0;
3285 Info.vol = 0;
3286 Info.readMem = true;
3287 Info.writeMem = true;
3288 Info.align = 0;
3289 return true;
3290
3291 case Intrinsic::nvvm_ldu_global_i:
3292 case Intrinsic::nvvm_ldu_global_f:
Justin Holewinskib926d9d2014-06-27 18:35:51 +00003293 case Intrinsic::nvvm_ldu_global_p: {
Mehdi Amini44ede332015-07-09 02:09:04 +00003294 auto &DL = I.getModule()->getDataLayout();
Justin Holewinskiae556d32012-05-04 20:18:50 +00003295 Info.opc = ISD::INTRINSIC_W_CHAIN;
3296 if (Intrinsic == Intrinsic::nvvm_ldu_global_i)
Mehdi Amini44ede332015-07-09 02:09:04 +00003297 Info.memVT = getValueType(DL, I.getType());
Justin Holewinskib926d9d2014-06-27 18:35:51 +00003298 else if(Intrinsic == Intrinsic::nvvm_ldu_global_p)
Mehdi Amini44ede332015-07-09 02:09:04 +00003299 Info.memVT = getPointerTy(DL);
Justin Holewinskiae556d32012-05-04 20:18:50 +00003300 else
Mehdi Amini44ede332015-07-09 02:09:04 +00003301 Info.memVT = getValueType(DL, I.getType());
Justin Holewinskiae556d32012-05-04 20:18:50 +00003302 Info.ptrVal = I.getArgOperand(0);
3303 Info.offset = 0;
3304 Info.vol = 0;
3305 Info.readMem = true;
3306 Info.writeMem = false;
Jingyue Wucb83a152014-08-29 15:30:20 +00003307 Info.align = cast<ConstantInt>(I.getArgOperand(1))->getZExtValue();
Justin Holewinskib926d9d2014-06-27 18:35:51 +00003308
Justin Holewinskiae556d32012-05-04 20:18:50 +00003309 return true;
Justin Holewinskib926d9d2014-06-27 18:35:51 +00003310 }
3311 case Intrinsic::nvvm_ldg_global_i:
3312 case Intrinsic::nvvm_ldg_global_f:
3313 case Intrinsic::nvvm_ldg_global_p: {
Mehdi Amini44ede332015-07-09 02:09:04 +00003314 auto &DL = I.getModule()->getDataLayout();
Justin Holewinskib926d9d2014-06-27 18:35:51 +00003315
3316 Info.opc = ISD::INTRINSIC_W_CHAIN;
3317 if (Intrinsic == Intrinsic::nvvm_ldg_global_i)
Mehdi Amini44ede332015-07-09 02:09:04 +00003318 Info.memVT = getValueType(DL, I.getType());
Justin Holewinskib926d9d2014-06-27 18:35:51 +00003319 else if(Intrinsic == Intrinsic::nvvm_ldg_global_p)
Mehdi Amini44ede332015-07-09 02:09:04 +00003320 Info.memVT = getPointerTy(DL);
Justin Holewinskib926d9d2014-06-27 18:35:51 +00003321 else
Mehdi Amini44ede332015-07-09 02:09:04 +00003322 Info.memVT = getValueType(DL, I.getType());
Justin Holewinskib926d9d2014-06-27 18:35:51 +00003323 Info.ptrVal = I.getArgOperand(0);
3324 Info.offset = 0;
3325 Info.vol = 0;
3326 Info.readMem = true;
3327 Info.writeMem = false;
Jingyue Wucb83a152014-08-29 15:30:20 +00003328 Info.align = cast<ConstantInt>(I.getArgOperand(1))->getZExtValue();
Justin Holewinskib926d9d2014-06-27 18:35:51 +00003329
3330 return true;
3331 }
Justin Holewinskiae556d32012-05-04 20:18:50 +00003332
Justin Holewinski9a2350e2014-07-17 11:59:04 +00003333 case Intrinsic::nvvm_tex_1d_v4f32_s32:
Justin Holewinski30d56a72014-04-09 15:39:15 +00003334 case Intrinsic::nvvm_tex_1d_v4f32_f32:
3335 case Intrinsic::nvvm_tex_1d_level_v4f32_f32:
3336 case Intrinsic::nvvm_tex_1d_grad_v4f32_f32:
Justin Holewinski9a2350e2014-07-17 11:59:04 +00003337 case Intrinsic::nvvm_tex_1d_array_v4f32_s32:
Justin Holewinski30d56a72014-04-09 15:39:15 +00003338 case Intrinsic::nvvm_tex_1d_array_v4f32_f32:
3339 case Intrinsic::nvvm_tex_1d_array_level_v4f32_f32:
3340 case Intrinsic::nvvm_tex_1d_array_grad_v4f32_f32:
Justin Holewinski9a2350e2014-07-17 11:59:04 +00003341 case Intrinsic::nvvm_tex_2d_v4f32_s32:
Justin Holewinski30d56a72014-04-09 15:39:15 +00003342 case Intrinsic::nvvm_tex_2d_v4f32_f32:
3343 case Intrinsic::nvvm_tex_2d_level_v4f32_f32:
3344 case Intrinsic::nvvm_tex_2d_grad_v4f32_f32:
Justin Holewinski9a2350e2014-07-17 11:59:04 +00003345 case Intrinsic::nvvm_tex_2d_array_v4f32_s32:
Justin Holewinski30d56a72014-04-09 15:39:15 +00003346 case Intrinsic::nvvm_tex_2d_array_v4f32_f32:
3347 case Intrinsic::nvvm_tex_2d_array_level_v4f32_f32:
3348 case Intrinsic::nvvm_tex_2d_array_grad_v4f32_f32:
Justin Holewinski9a2350e2014-07-17 11:59:04 +00003349 case Intrinsic::nvvm_tex_3d_v4f32_s32:
Justin Holewinski30d56a72014-04-09 15:39:15 +00003350 case Intrinsic::nvvm_tex_3d_v4f32_f32:
3351 case Intrinsic::nvvm_tex_3d_level_v4f32_f32:
Justin Holewinski9a2350e2014-07-17 11:59:04 +00003352 case Intrinsic::nvvm_tex_3d_grad_v4f32_f32:
3353 case Intrinsic::nvvm_tex_cube_v4f32_f32:
3354 case Intrinsic::nvvm_tex_cube_level_v4f32_f32:
3355 case Intrinsic::nvvm_tex_cube_array_v4f32_f32:
3356 case Intrinsic::nvvm_tex_cube_array_level_v4f32_f32:
3357 case Intrinsic::nvvm_tld4_r_2d_v4f32_f32:
3358 case Intrinsic::nvvm_tld4_g_2d_v4f32_f32:
3359 case Intrinsic::nvvm_tld4_b_2d_v4f32_f32:
3360 case Intrinsic::nvvm_tld4_a_2d_v4f32_f32:
3361 case Intrinsic::nvvm_tex_unified_1d_v4f32_s32:
3362 case Intrinsic::nvvm_tex_unified_1d_v4f32_f32:
3363 case Intrinsic::nvvm_tex_unified_1d_level_v4f32_f32:
3364 case Intrinsic::nvvm_tex_unified_1d_grad_v4f32_f32:
3365 case Intrinsic::nvvm_tex_unified_1d_array_v4f32_s32:
3366 case Intrinsic::nvvm_tex_unified_1d_array_v4f32_f32:
3367 case Intrinsic::nvvm_tex_unified_1d_array_level_v4f32_f32:
3368 case Intrinsic::nvvm_tex_unified_1d_array_grad_v4f32_f32:
3369 case Intrinsic::nvvm_tex_unified_2d_v4f32_s32:
3370 case Intrinsic::nvvm_tex_unified_2d_v4f32_f32:
3371 case Intrinsic::nvvm_tex_unified_2d_level_v4f32_f32:
3372 case Intrinsic::nvvm_tex_unified_2d_grad_v4f32_f32:
3373 case Intrinsic::nvvm_tex_unified_2d_array_v4f32_s32:
3374 case Intrinsic::nvvm_tex_unified_2d_array_v4f32_f32:
3375 case Intrinsic::nvvm_tex_unified_2d_array_level_v4f32_f32:
3376 case Intrinsic::nvvm_tex_unified_2d_array_grad_v4f32_f32:
3377 case Intrinsic::nvvm_tex_unified_3d_v4f32_s32:
3378 case Intrinsic::nvvm_tex_unified_3d_v4f32_f32:
3379 case Intrinsic::nvvm_tex_unified_3d_level_v4f32_f32:
3380 case Intrinsic::nvvm_tex_unified_3d_grad_v4f32_f32:
3381 case Intrinsic::nvvm_tex_unified_cube_v4f32_f32:
3382 case Intrinsic::nvvm_tex_unified_cube_level_v4f32_f32:
3383 case Intrinsic::nvvm_tex_unified_cube_array_v4f32_f32:
3384 case Intrinsic::nvvm_tex_unified_cube_array_level_v4f32_f32:
3385 case Intrinsic::nvvm_tld4_unified_r_2d_v4f32_f32:
3386 case Intrinsic::nvvm_tld4_unified_g_2d_v4f32_f32:
3387 case Intrinsic::nvvm_tld4_unified_b_2d_v4f32_f32:
3388 case Intrinsic::nvvm_tld4_unified_a_2d_v4f32_f32: {
Justin Holewinski30d56a72014-04-09 15:39:15 +00003389 Info.opc = getOpcForTextureInstr(Intrinsic);
Justin Holewinski9a2350e2014-07-17 11:59:04 +00003390 Info.memVT = MVT::v4f32;
Craig Topper062a2ba2014-04-25 05:30:21 +00003391 Info.ptrVal = nullptr;
Justin Holewinski30d56a72014-04-09 15:39:15 +00003392 Info.offset = 0;
3393 Info.vol = 0;
3394 Info.readMem = true;
3395 Info.writeMem = false;
3396 Info.align = 16;
3397 return true;
3398 }
Justin Holewinski9a2350e2014-07-17 11:59:04 +00003399 case Intrinsic::nvvm_tex_1d_v4s32_s32:
3400 case Intrinsic::nvvm_tex_1d_v4s32_f32:
3401 case Intrinsic::nvvm_tex_1d_level_v4s32_f32:
3402 case Intrinsic::nvvm_tex_1d_grad_v4s32_f32:
3403 case Intrinsic::nvvm_tex_1d_array_v4s32_s32:
3404 case Intrinsic::nvvm_tex_1d_array_v4s32_f32:
3405 case Intrinsic::nvvm_tex_1d_array_level_v4s32_f32:
3406 case Intrinsic::nvvm_tex_1d_array_grad_v4s32_f32:
3407 case Intrinsic::nvvm_tex_2d_v4s32_s32:
3408 case Intrinsic::nvvm_tex_2d_v4s32_f32:
3409 case Intrinsic::nvvm_tex_2d_level_v4s32_f32:
3410 case Intrinsic::nvvm_tex_2d_grad_v4s32_f32:
3411 case Intrinsic::nvvm_tex_2d_array_v4s32_s32:
3412 case Intrinsic::nvvm_tex_2d_array_v4s32_f32:
3413 case Intrinsic::nvvm_tex_2d_array_level_v4s32_f32:
3414 case Intrinsic::nvvm_tex_2d_array_grad_v4s32_f32:
3415 case Intrinsic::nvvm_tex_3d_v4s32_s32:
3416 case Intrinsic::nvvm_tex_3d_v4s32_f32:
3417 case Intrinsic::nvvm_tex_3d_level_v4s32_f32:
3418 case Intrinsic::nvvm_tex_3d_grad_v4s32_f32:
3419 case Intrinsic::nvvm_tex_cube_v4s32_f32:
3420 case Intrinsic::nvvm_tex_cube_level_v4s32_f32:
3421 case Intrinsic::nvvm_tex_cube_array_v4s32_f32:
3422 case Intrinsic::nvvm_tex_cube_array_level_v4s32_f32:
3423 case Intrinsic::nvvm_tex_cube_v4u32_f32:
3424 case Intrinsic::nvvm_tex_cube_level_v4u32_f32:
3425 case Intrinsic::nvvm_tex_cube_array_v4u32_f32:
3426 case Intrinsic::nvvm_tex_cube_array_level_v4u32_f32:
3427 case Intrinsic::nvvm_tex_1d_v4u32_s32:
3428 case Intrinsic::nvvm_tex_1d_v4u32_f32:
3429 case Intrinsic::nvvm_tex_1d_level_v4u32_f32:
3430 case Intrinsic::nvvm_tex_1d_grad_v4u32_f32:
3431 case Intrinsic::nvvm_tex_1d_array_v4u32_s32:
3432 case Intrinsic::nvvm_tex_1d_array_v4u32_f32:
3433 case Intrinsic::nvvm_tex_1d_array_level_v4u32_f32:
3434 case Intrinsic::nvvm_tex_1d_array_grad_v4u32_f32:
3435 case Intrinsic::nvvm_tex_2d_v4u32_s32:
3436 case Intrinsic::nvvm_tex_2d_v4u32_f32:
3437 case Intrinsic::nvvm_tex_2d_level_v4u32_f32:
3438 case Intrinsic::nvvm_tex_2d_grad_v4u32_f32:
3439 case Intrinsic::nvvm_tex_2d_array_v4u32_s32:
3440 case Intrinsic::nvvm_tex_2d_array_v4u32_f32:
3441 case Intrinsic::nvvm_tex_2d_array_level_v4u32_f32:
3442 case Intrinsic::nvvm_tex_2d_array_grad_v4u32_f32:
3443 case Intrinsic::nvvm_tex_3d_v4u32_s32:
3444 case Intrinsic::nvvm_tex_3d_v4u32_f32:
3445 case Intrinsic::nvvm_tex_3d_level_v4u32_f32:
3446 case Intrinsic::nvvm_tex_3d_grad_v4u32_f32:
3447 case Intrinsic::nvvm_tld4_r_2d_v4s32_f32:
3448 case Intrinsic::nvvm_tld4_g_2d_v4s32_f32:
3449 case Intrinsic::nvvm_tld4_b_2d_v4s32_f32:
3450 case Intrinsic::nvvm_tld4_a_2d_v4s32_f32:
3451 case Intrinsic::nvvm_tld4_r_2d_v4u32_f32:
3452 case Intrinsic::nvvm_tld4_g_2d_v4u32_f32:
3453 case Intrinsic::nvvm_tld4_b_2d_v4u32_f32:
3454 case Intrinsic::nvvm_tld4_a_2d_v4u32_f32:
3455 case Intrinsic::nvvm_tex_unified_1d_v4s32_s32:
3456 case Intrinsic::nvvm_tex_unified_1d_v4s32_f32:
3457 case Intrinsic::nvvm_tex_unified_1d_level_v4s32_f32:
3458 case Intrinsic::nvvm_tex_unified_1d_grad_v4s32_f32:
3459 case Intrinsic::nvvm_tex_unified_1d_array_v4s32_s32:
3460 case Intrinsic::nvvm_tex_unified_1d_array_v4s32_f32:
3461 case Intrinsic::nvvm_tex_unified_1d_array_level_v4s32_f32:
3462 case Intrinsic::nvvm_tex_unified_1d_array_grad_v4s32_f32:
3463 case Intrinsic::nvvm_tex_unified_2d_v4s32_s32:
3464 case Intrinsic::nvvm_tex_unified_2d_v4s32_f32:
3465 case Intrinsic::nvvm_tex_unified_2d_level_v4s32_f32:
3466 case Intrinsic::nvvm_tex_unified_2d_grad_v4s32_f32:
3467 case Intrinsic::nvvm_tex_unified_2d_array_v4s32_s32:
3468 case Intrinsic::nvvm_tex_unified_2d_array_v4s32_f32:
3469 case Intrinsic::nvvm_tex_unified_2d_array_level_v4s32_f32:
3470 case Intrinsic::nvvm_tex_unified_2d_array_grad_v4s32_f32:
3471 case Intrinsic::nvvm_tex_unified_3d_v4s32_s32:
3472 case Intrinsic::nvvm_tex_unified_3d_v4s32_f32:
3473 case Intrinsic::nvvm_tex_unified_3d_level_v4s32_f32:
3474 case Intrinsic::nvvm_tex_unified_3d_grad_v4s32_f32:
3475 case Intrinsic::nvvm_tex_unified_1d_v4u32_s32:
3476 case Intrinsic::nvvm_tex_unified_1d_v4u32_f32:
3477 case Intrinsic::nvvm_tex_unified_1d_level_v4u32_f32:
3478 case Intrinsic::nvvm_tex_unified_1d_grad_v4u32_f32:
3479 case Intrinsic::nvvm_tex_unified_1d_array_v4u32_s32:
3480 case Intrinsic::nvvm_tex_unified_1d_array_v4u32_f32:
3481 case Intrinsic::nvvm_tex_unified_1d_array_level_v4u32_f32:
3482 case Intrinsic::nvvm_tex_unified_1d_array_grad_v4u32_f32:
3483 case Intrinsic::nvvm_tex_unified_2d_v4u32_s32:
3484 case Intrinsic::nvvm_tex_unified_2d_v4u32_f32:
3485 case Intrinsic::nvvm_tex_unified_2d_level_v4u32_f32:
3486 case Intrinsic::nvvm_tex_unified_2d_grad_v4u32_f32:
3487 case Intrinsic::nvvm_tex_unified_2d_array_v4u32_s32:
3488 case Intrinsic::nvvm_tex_unified_2d_array_v4u32_f32:
3489 case Intrinsic::nvvm_tex_unified_2d_array_level_v4u32_f32:
3490 case Intrinsic::nvvm_tex_unified_2d_array_grad_v4u32_f32:
3491 case Intrinsic::nvvm_tex_unified_3d_v4u32_s32:
3492 case Intrinsic::nvvm_tex_unified_3d_v4u32_f32:
3493 case Intrinsic::nvvm_tex_unified_3d_level_v4u32_f32:
3494 case Intrinsic::nvvm_tex_unified_3d_grad_v4u32_f32:
3495 case Intrinsic::nvvm_tex_unified_cube_v4s32_f32:
3496 case Intrinsic::nvvm_tex_unified_cube_level_v4s32_f32:
3497 case Intrinsic::nvvm_tex_unified_cube_array_v4s32_f32:
3498 case Intrinsic::nvvm_tex_unified_cube_array_level_v4s32_f32:
3499 case Intrinsic::nvvm_tex_unified_cube_v4u32_f32:
3500 case Intrinsic::nvvm_tex_unified_cube_level_v4u32_f32:
3501 case Intrinsic::nvvm_tex_unified_cube_array_v4u32_f32:
3502 case Intrinsic::nvvm_tex_unified_cube_array_level_v4u32_f32:
3503 case Intrinsic::nvvm_tld4_unified_r_2d_v4s32_f32:
3504 case Intrinsic::nvvm_tld4_unified_g_2d_v4s32_f32:
3505 case Intrinsic::nvvm_tld4_unified_b_2d_v4s32_f32:
3506 case Intrinsic::nvvm_tld4_unified_a_2d_v4s32_f32:
3507 case Intrinsic::nvvm_tld4_unified_r_2d_v4u32_f32:
3508 case Intrinsic::nvvm_tld4_unified_g_2d_v4u32_f32:
3509 case Intrinsic::nvvm_tld4_unified_b_2d_v4u32_f32:
3510 case Intrinsic::nvvm_tld4_unified_a_2d_v4u32_f32: {
Justin Holewinski30d56a72014-04-09 15:39:15 +00003511 Info.opc = getOpcForTextureInstr(Intrinsic);
Justin Holewinski9a2350e2014-07-17 11:59:04 +00003512 Info.memVT = MVT::v4i32;
Craig Topper062a2ba2014-04-25 05:30:21 +00003513 Info.ptrVal = nullptr;
Justin Holewinski30d56a72014-04-09 15:39:15 +00003514 Info.offset = 0;
3515 Info.vol = 0;
3516 Info.readMem = true;
3517 Info.writeMem = false;
3518 Info.align = 16;
3519 return true;
3520 }
Justin Holewinski9a2350e2014-07-17 11:59:04 +00003521 case Intrinsic::nvvm_suld_1d_i8_clamp:
3522 case Intrinsic::nvvm_suld_1d_v2i8_clamp:
3523 case Intrinsic::nvvm_suld_1d_v4i8_clamp:
3524 case Intrinsic::nvvm_suld_1d_array_i8_clamp:
3525 case Intrinsic::nvvm_suld_1d_array_v2i8_clamp:
3526 case Intrinsic::nvvm_suld_1d_array_v4i8_clamp:
3527 case Intrinsic::nvvm_suld_2d_i8_clamp:
3528 case Intrinsic::nvvm_suld_2d_v2i8_clamp:
3529 case Intrinsic::nvvm_suld_2d_v4i8_clamp:
3530 case Intrinsic::nvvm_suld_2d_array_i8_clamp:
3531 case Intrinsic::nvvm_suld_2d_array_v2i8_clamp:
3532 case Intrinsic::nvvm_suld_2d_array_v4i8_clamp:
3533 case Intrinsic::nvvm_suld_3d_i8_clamp:
3534 case Intrinsic::nvvm_suld_3d_v2i8_clamp:
3535 case Intrinsic::nvvm_suld_3d_v4i8_clamp:
Justin Holewinski30d56a72014-04-09 15:39:15 +00003536 case Intrinsic::nvvm_suld_1d_i8_trap:
3537 case Intrinsic::nvvm_suld_1d_v2i8_trap:
3538 case Intrinsic::nvvm_suld_1d_v4i8_trap:
3539 case Intrinsic::nvvm_suld_1d_array_i8_trap:
3540 case Intrinsic::nvvm_suld_1d_array_v2i8_trap:
3541 case Intrinsic::nvvm_suld_1d_array_v4i8_trap:
3542 case Intrinsic::nvvm_suld_2d_i8_trap:
3543 case Intrinsic::nvvm_suld_2d_v2i8_trap:
3544 case Intrinsic::nvvm_suld_2d_v4i8_trap:
3545 case Intrinsic::nvvm_suld_2d_array_i8_trap:
3546 case Intrinsic::nvvm_suld_2d_array_v2i8_trap:
3547 case Intrinsic::nvvm_suld_2d_array_v4i8_trap:
3548 case Intrinsic::nvvm_suld_3d_i8_trap:
3549 case Intrinsic::nvvm_suld_3d_v2i8_trap:
Justin Holewinski9a2350e2014-07-17 11:59:04 +00003550 case Intrinsic::nvvm_suld_3d_v4i8_trap:
3551 case Intrinsic::nvvm_suld_1d_i8_zero:
3552 case Intrinsic::nvvm_suld_1d_v2i8_zero:
3553 case Intrinsic::nvvm_suld_1d_v4i8_zero:
3554 case Intrinsic::nvvm_suld_1d_array_i8_zero:
3555 case Intrinsic::nvvm_suld_1d_array_v2i8_zero:
3556 case Intrinsic::nvvm_suld_1d_array_v4i8_zero:
3557 case Intrinsic::nvvm_suld_2d_i8_zero:
3558 case Intrinsic::nvvm_suld_2d_v2i8_zero:
3559 case Intrinsic::nvvm_suld_2d_v4i8_zero:
3560 case Intrinsic::nvvm_suld_2d_array_i8_zero:
3561 case Intrinsic::nvvm_suld_2d_array_v2i8_zero:
3562 case Intrinsic::nvvm_suld_2d_array_v4i8_zero:
3563 case Intrinsic::nvvm_suld_3d_i8_zero:
3564 case Intrinsic::nvvm_suld_3d_v2i8_zero:
3565 case Intrinsic::nvvm_suld_3d_v4i8_zero: {
Justin Holewinski30d56a72014-04-09 15:39:15 +00003566 Info.opc = getOpcForSurfaceInstr(Intrinsic);
3567 Info.memVT = MVT::i8;
Craig Topper062a2ba2014-04-25 05:30:21 +00003568 Info.ptrVal = nullptr;
Justin Holewinski30d56a72014-04-09 15:39:15 +00003569 Info.offset = 0;
3570 Info.vol = 0;
3571 Info.readMem = true;
3572 Info.writeMem = false;
3573 Info.align = 16;
3574 return true;
3575 }
Justin Holewinski9a2350e2014-07-17 11:59:04 +00003576 case Intrinsic::nvvm_suld_1d_i16_clamp:
3577 case Intrinsic::nvvm_suld_1d_v2i16_clamp:
3578 case Intrinsic::nvvm_suld_1d_v4i16_clamp:
3579 case Intrinsic::nvvm_suld_1d_array_i16_clamp:
3580 case Intrinsic::nvvm_suld_1d_array_v2i16_clamp:
3581 case Intrinsic::nvvm_suld_1d_array_v4i16_clamp:
3582 case Intrinsic::nvvm_suld_2d_i16_clamp:
3583 case Intrinsic::nvvm_suld_2d_v2i16_clamp:
3584 case Intrinsic::nvvm_suld_2d_v4i16_clamp:
3585 case Intrinsic::nvvm_suld_2d_array_i16_clamp:
3586 case Intrinsic::nvvm_suld_2d_array_v2i16_clamp:
3587 case Intrinsic::nvvm_suld_2d_array_v4i16_clamp:
3588 case Intrinsic::nvvm_suld_3d_i16_clamp:
3589 case Intrinsic::nvvm_suld_3d_v2i16_clamp:
3590 case Intrinsic::nvvm_suld_3d_v4i16_clamp:
Justin Holewinski30d56a72014-04-09 15:39:15 +00003591 case Intrinsic::nvvm_suld_1d_i16_trap:
3592 case Intrinsic::nvvm_suld_1d_v2i16_trap:
3593 case Intrinsic::nvvm_suld_1d_v4i16_trap:
3594 case Intrinsic::nvvm_suld_1d_array_i16_trap:
3595 case Intrinsic::nvvm_suld_1d_array_v2i16_trap:
3596 case Intrinsic::nvvm_suld_1d_array_v4i16_trap:
3597 case Intrinsic::nvvm_suld_2d_i16_trap:
3598 case Intrinsic::nvvm_suld_2d_v2i16_trap:
3599 case Intrinsic::nvvm_suld_2d_v4i16_trap:
3600 case Intrinsic::nvvm_suld_2d_array_i16_trap:
3601 case Intrinsic::nvvm_suld_2d_array_v2i16_trap:
3602 case Intrinsic::nvvm_suld_2d_array_v4i16_trap:
3603 case Intrinsic::nvvm_suld_3d_i16_trap:
3604 case Intrinsic::nvvm_suld_3d_v2i16_trap:
Justin Holewinski9a2350e2014-07-17 11:59:04 +00003605 case Intrinsic::nvvm_suld_3d_v4i16_trap:
3606 case Intrinsic::nvvm_suld_1d_i16_zero:
3607 case Intrinsic::nvvm_suld_1d_v2i16_zero:
3608 case Intrinsic::nvvm_suld_1d_v4i16_zero:
3609 case Intrinsic::nvvm_suld_1d_array_i16_zero:
3610 case Intrinsic::nvvm_suld_1d_array_v2i16_zero:
3611 case Intrinsic::nvvm_suld_1d_array_v4i16_zero:
3612 case Intrinsic::nvvm_suld_2d_i16_zero:
3613 case Intrinsic::nvvm_suld_2d_v2i16_zero:
3614 case Intrinsic::nvvm_suld_2d_v4i16_zero:
3615 case Intrinsic::nvvm_suld_2d_array_i16_zero:
3616 case Intrinsic::nvvm_suld_2d_array_v2i16_zero:
3617 case Intrinsic::nvvm_suld_2d_array_v4i16_zero:
3618 case Intrinsic::nvvm_suld_3d_i16_zero:
3619 case Intrinsic::nvvm_suld_3d_v2i16_zero:
3620 case Intrinsic::nvvm_suld_3d_v4i16_zero: {
Justin Holewinski30d56a72014-04-09 15:39:15 +00003621 Info.opc = getOpcForSurfaceInstr(Intrinsic);
3622 Info.memVT = MVT::i16;
Craig Topper062a2ba2014-04-25 05:30:21 +00003623 Info.ptrVal = nullptr;
Justin Holewinski30d56a72014-04-09 15:39:15 +00003624 Info.offset = 0;
3625 Info.vol = 0;
3626 Info.readMem = true;
3627 Info.writeMem = false;
3628 Info.align = 16;
3629 return true;
3630 }
Justin Holewinski9a2350e2014-07-17 11:59:04 +00003631 case Intrinsic::nvvm_suld_1d_i32_clamp:
3632 case Intrinsic::nvvm_suld_1d_v2i32_clamp:
3633 case Intrinsic::nvvm_suld_1d_v4i32_clamp:
3634 case Intrinsic::nvvm_suld_1d_array_i32_clamp:
3635 case Intrinsic::nvvm_suld_1d_array_v2i32_clamp:
3636 case Intrinsic::nvvm_suld_1d_array_v4i32_clamp:
3637 case Intrinsic::nvvm_suld_2d_i32_clamp:
3638 case Intrinsic::nvvm_suld_2d_v2i32_clamp:
3639 case Intrinsic::nvvm_suld_2d_v4i32_clamp:
3640 case Intrinsic::nvvm_suld_2d_array_i32_clamp:
3641 case Intrinsic::nvvm_suld_2d_array_v2i32_clamp:
3642 case Intrinsic::nvvm_suld_2d_array_v4i32_clamp:
3643 case Intrinsic::nvvm_suld_3d_i32_clamp:
3644 case Intrinsic::nvvm_suld_3d_v2i32_clamp:
3645 case Intrinsic::nvvm_suld_3d_v4i32_clamp:
Justin Holewinski30d56a72014-04-09 15:39:15 +00003646 case Intrinsic::nvvm_suld_1d_i32_trap:
3647 case Intrinsic::nvvm_suld_1d_v2i32_trap:
3648 case Intrinsic::nvvm_suld_1d_v4i32_trap:
3649 case Intrinsic::nvvm_suld_1d_array_i32_trap:
3650 case Intrinsic::nvvm_suld_1d_array_v2i32_trap:
3651 case Intrinsic::nvvm_suld_1d_array_v4i32_trap:
3652 case Intrinsic::nvvm_suld_2d_i32_trap:
3653 case Intrinsic::nvvm_suld_2d_v2i32_trap:
3654 case Intrinsic::nvvm_suld_2d_v4i32_trap:
3655 case Intrinsic::nvvm_suld_2d_array_i32_trap:
3656 case Intrinsic::nvvm_suld_2d_array_v2i32_trap:
3657 case Intrinsic::nvvm_suld_2d_array_v4i32_trap:
3658 case Intrinsic::nvvm_suld_3d_i32_trap:
3659 case Intrinsic::nvvm_suld_3d_v2i32_trap:
Justin Holewinski9a2350e2014-07-17 11:59:04 +00003660 case Intrinsic::nvvm_suld_3d_v4i32_trap:
3661 case Intrinsic::nvvm_suld_1d_i32_zero:
3662 case Intrinsic::nvvm_suld_1d_v2i32_zero:
3663 case Intrinsic::nvvm_suld_1d_v4i32_zero:
3664 case Intrinsic::nvvm_suld_1d_array_i32_zero:
3665 case Intrinsic::nvvm_suld_1d_array_v2i32_zero:
3666 case Intrinsic::nvvm_suld_1d_array_v4i32_zero:
3667 case Intrinsic::nvvm_suld_2d_i32_zero:
3668 case Intrinsic::nvvm_suld_2d_v2i32_zero:
3669 case Intrinsic::nvvm_suld_2d_v4i32_zero:
3670 case Intrinsic::nvvm_suld_2d_array_i32_zero:
3671 case Intrinsic::nvvm_suld_2d_array_v2i32_zero:
3672 case Intrinsic::nvvm_suld_2d_array_v4i32_zero:
3673 case Intrinsic::nvvm_suld_3d_i32_zero:
3674 case Intrinsic::nvvm_suld_3d_v2i32_zero:
3675 case Intrinsic::nvvm_suld_3d_v4i32_zero: {
Justin Holewinski30d56a72014-04-09 15:39:15 +00003676 Info.opc = getOpcForSurfaceInstr(Intrinsic);
3677 Info.memVT = MVT::i32;
Craig Topper062a2ba2014-04-25 05:30:21 +00003678 Info.ptrVal = nullptr;
Justin Holewinski30d56a72014-04-09 15:39:15 +00003679 Info.offset = 0;
3680 Info.vol = 0;
3681 Info.readMem = true;
3682 Info.writeMem = false;
3683 Info.align = 16;
3684 return true;
3685 }
Justin Holewinski9a2350e2014-07-17 11:59:04 +00003686 case Intrinsic::nvvm_suld_1d_i64_clamp:
3687 case Intrinsic::nvvm_suld_1d_v2i64_clamp:
3688 case Intrinsic::nvvm_suld_1d_array_i64_clamp:
3689 case Intrinsic::nvvm_suld_1d_array_v2i64_clamp:
3690 case Intrinsic::nvvm_suld_2d_i64_clamp:
3691 case Intrinsic::nvvm_suld_2d_v2i64_clamp:
3692 case Intrinsic::nvvm_suld_2d_array_i64_clamp:
3693 case Intrinsic::nvvm_suld_2d_array_v2i64_clamp:
3694 case Intrinsic::nvvm_suld_3d_i64_clamp:
3695 case Intrinsic::nvvm_suld_3d_v2i64_clamp:
3696 case Intrinsic::nvvm_suld_1d_i64_trap:
3697 case Intrinsic::nvvm_suld_1d_v2i64_trap:
3698 case Intrinsic::nvvm_suld_1d_array_i64_trap:
3699 case Intrinsic::nvvm_suld_1d_array_v2i64_trap:
3700 case Intrinsic::nvvm_suld_2d_i64_trap:
3701 case Intrinsic::nvvm_suld_2d_v2i64_trap:
3702 case Intrinsic::nvvm_suld_2d_array_i64_trap:
3703 case Intrinsic::nvvm_suld_2d_array_v2i64_trap:
3704 case Intrinsic::nvvm_suld_3d_i64_trap:
3705 case Intrinsic::nvvm_suld_3d_v2i64_trap:
3706 case Intrinsic::nvvm_suld_1d_i64_zero:
3707 case Intrinsic::nvvm_suld_1d_v2i64_zero:
3708 case Intrinsic::nvvm_suld_1d_array_i64_zero:
3709 case Intrinsic::nvvm_suld_1d_array_v2i64_zero:
3710 case Intrinsic::nvvm_suld_2d_i64_zero:
3711 case Intrinsic::nvvm_suld_2d_v2i64_zero:
3712 case Intrinsic::nvvm_suld_2d_array_i64_zero:
3713 case Intrinsic::nvvm_suld_2d_array_v2i64_zero:
3714 case Intrinsic::nvvm_suld_3d_i64_zero:
3715 case Intrinsic::nvvm_suld_3d_v2i64_zero: {
3716 Info.opc = getOpcForSurfaceInstr(Intrinsic);
3717 Info.memVT = MVT::i64;
3718 Info.ptrVal = nullptr;
3719 Info.offset = 0;
3720 Info.vol = 0;
3721 Info.readMem = true;
3722 Info.writeMem = false;
3723 Info.align = 16;
3724 return true;
3725 }
Justin Holewinskiae556d32012-05-04 20:18:50 +00003726 }
3727 return false;
3728}
3729
3730/// isLegalAddressingMode - Return true if the addressing mode represented
3731/// by AM is legal for this target, for a load/store of the specified type.
3732/// Used to guide target specific optimizations, like loop strength reduction
3733/// (LoopStrengthReduce.cpp) and memory optimization for address mode
3734/// (CodeGenPrepare.cpp)
Mehdi Amini0cdec1e2015-07-09 02:09:40 +00003735bool NVPTXTargetLowering::isLegalAddressingMode(const DataLayout &DL,
3736 const AddrMode &AM, Type *Ty,
Matt Arsenaultbd7d80a2015-06-01 05:31:59 +00003737 unsigned AS) const {
Justin Holewinskiae556d32012-05-04 20:18:50 +00003738
3739 // AddrMode - This represents an addressing mode of:
3740 // BaseGV + BaseOffs + BaseReg + Scale*ScaleReg
3741 //
3742 // The legal address modes are
3743 // - [avar]
3744 // - [areg]
3745 // - [areg+immoff]
3746 // - [immAddr]
3747
3748 if (AM.BaseGV) {
3749 if (AM.BaseOffs || AM.HasBaseReg || AM.Scale)
3750 return false;
3751 return true;
3752 }
3753
3754 switch (AM.Scale) {
Justin Holewinski0497ab12013-03-30 14:29:21 +00003755 case 0: // "r", "r+i" or "i" is allowed
Justin Holewinskiae556d32012-05-04 20:18:50 +00003756 break;
3757 case 1:
Justin Holewinski0497ab12013-03-30 14:29:21 +00003758 if (AM.HasBaseReg) // "r+r+i" or "r+r" is not allowed.
Justin Holewinskiae556d32012-05-04 20:18:50 +00003759 return false;
3760 // Otherwise we have r+i.
3761 break;
3762 default:
3763 // No scale > 1 is allowed
3764 return false;
3765 }
3766 return true;
3767}
3768
3769//===----------------------------------------------------------------------===//
3770// NVPTX Inline Assembly Support
3771//===----------------------------------------------------------------------===//
3772
3773/// getConstraintType - Given a constraint letter, return the type of
3774/// constraint it is for this target.
3775NVPTXTargetLowering::ConstraintType
Benjamin Kramer9bfb6272015-07-05 19:29:18 +00003776NVPTXTargetLowering::getConstraintType(StringRef Constraint) const {
Justin Holewinskiae556d32012-05-04 20:18:50 +00003777 if (Constraint.size() == 1) {
3778 switch (Constraint[0]) {
3779 default:
3780 break;
Justin Holewinski2739c012014-06-27 18:36:06 +00003781 case 'b':
Justin Holewinskiae556d32012-05-04 20:18:50 +00003782 case 'r':
3783 case 'h':
3784 case 'c':
3785 case 'l':
3786 case 'f':
3787 case 'd':
3788 case '0':
3789 case 'N':
3790 return C_RegisterClass;
3791 }
3792 }
3793 return TargetLowering::getConstraintType(Constraint);
3794}
3795
Justin Holewinski0497ab12013-03-30 14:29:21 +00003796std::pair<unsigned, const TargetRegisterClass *>
Eric Christopher11e4df72015-02-26 22:38:43 +00003797NVPTXTargetLowering::getRegForInlineAsmConstraint(const TargetRegisterInfo *TRI,
Benjamin Kramer9bfb6272015-07-05 19:29:18 +00003798 StringRef Constraint,
Chad Rosier295bd432013-06-22 18:37:38 +00003799 MVT VT) const {
Justin Holewinskiae556d32012-05-04 20:18:50 +00003800 if (Constraint.size() == 1) {
3801 switch (Constraint[0]) {
Justin Holewinski2739c012014-06-27 18:36:06 +00003802 case 'b':
3803 return std::make_pair(0U, &NVPTX::Int1RegsRegClass);
Justin Holewinskiae556d32012-05-04 20:18:50 +00003804 case 'c':
Justin Holewinskif8f70912013-06-28 17:57:59 +00003805 return std::make_pair(0U, &NVPTX::Int16RegsRegClass);
Justin Holewinskiae556d32012-05-04 20:18:50 +00003806 case 'h':
3807 return std::make_pair(0U, &NVPTX::Int16RegsRegClass);
3808 case 'r':
3809 return std::make_pair(0U, &NVPTX::Int32RegsRegClass);
3810 case 'l':
3811 case 'N':
3812 return std::make_pair(0U, &NVPTX::Int64RegsRegClass);
3813 case 'f':
3814 return std::make_pair(0U, &NVPTX::Float32RegsRegClass);
3815 case 'd':
3816 return std::make_pair(0U, &NVPTX::Float64RegsRegClass);
3817 }
3818 }
Eric Christopher11e4df72015-02-26 22:38:43 +00003819 return TargetLowering::getRegForInlineAsmConstraint(TRI, Constraint, VT);
Justin Holewinskiae556d32012-05-04 20:18:50 +00003820}
3821
Justin Holewinskiae556d32012-05-04 20:18:50 +00003822/// getFunctionAlignment - Return the Log2 alignment of this function.
3823unsigned NVPTXTargetLowering::getFunctionAlignment(const Function *) const {
3824 return 4;
3825}
Justin Holewinskibe8dc642013-02-12 14:18:49 +00003826
Justin Holewinskieafe26d2014-06-27 18:35:37 +00003827//===----------------------------------------------------------------------===//
3828// NVPTX DAG Combining
3829//===----------------------------------------------------------------------===//
3830
Justin Holewinski428cf0e2014-07-17 18:10:09 +00003831bool NVPTXTargetLowering::allowFMA(MachineFunction &MF,
3832 CodeGenOpt::Level OptLevel) const {
3833 const Function *F = MF.getFunction();
3834 const TargetOptions &TO = MF.getTarget().Options;
3835
3836 // Always honor command-line argument
3837 if (FMAContractLevelOpt.getNumOccurrences() > 0) {
3838 return FMAContractLevelOpt > 0;
3839 } else if (OptLevel == 0) {
3840 // Do not contract if we're not optimizing the code
3841 return false;
3842 } else if (TO.AllowFPOpFusion == FPOpFusion::Fast || TO.UnsafeFPMath) {
3843 // Honor TargetOptions flags that explicitly say fusion is okay
3844 return true;
3845 } else if (F->hasFnAttribute("unsafe-fp-math")) {
3846 // Check for unsafe-fp-math=true coming from Clang
3847 Attribute Attr = F->getFnAttribute("unsafe-fp-math");
3848 StringRef Val = Attr.getValueAsString();
3849 if (Val == "true")
3850 return true;
3851 }
3852
3853 // We did not have a clear indication that fusion is allowed, so assume not
3854 return false;
3855}
Justin Holewinskieafe26d2014-06-27 18:35:37 +00003856
3857/// PerformADDCombineWithOperands - Try DAG combinations for an ADD with
3858/// operands N0 and N1. This is a helper for PerformADDCombine that is
3859/// called with the default operands, and if that fails, with commuted
3860/// operands.
3861static SDValue PerformADDCombineWithOperands(SDNode *N, SDValue N0, SDValue N1,
3862 TargetLowering::DAGCombinerInfo &DCI,
3863 const NVPTXSubtarget &Subtarget,
3864 CodeGenOpt::Level OptLevel) {
3865 SelectionDAG &DAG = DCI.DAG;
3866 // Skip non-integer, non-scalar case
3867 EVT VT=N0.getValueType();
3868 if (VT.isVector())
3869 return SDValue();
3870
3871 // fold (add (mul a, b), c) -> (mad a, b, c)
3872 //
3873 if (N0.getOpcode() == ISD::MUL) {
3874 assert (VT.isInteger());
3875 // For integer:
3876 // Since integer multiply-add costs the same as integer multiply
3877 // but is more costly than integer add, do the fusion only when
3878 // the mul is only used in the add.
3879 if (OptLevel==CodeGenOpt::None || VT != MVT::i32 ||
3880 !N0.getNode()->hasOneUse())
3881 return SDValue();
3882
3883 // Do the folding
3884 return DAG.getNode(NVPTXISD::IMAD, SDLoc(N), VT,
3885 N0.getOperand(0), N0.getOperand(1), N1);
3886 }
3887 else if (N0.getOpcode() == ISD::FMUL) {
3888 if (VT == MVT::f32 || VT == MVT::f64) {
Aaron Ballman53201af2014-07-31 12:55:49 +00003889 const auto *TLI = static_cast<const NVPTXTargetLowering *>(
3890 &DAG.getTargetLoweringInfo());
Justin Holewinski428cf0e2014-07-17 18:10:09 +00003891 if (!TLI->allowFMA(DAG.getMachineFunction(), OptLevel))
Justin Holewinskieafe26d2014-06-27 18:35:37 +00003892 return SDValue();
3893
3894 // For floating point:
3895 // Do the fusion only when the mul has less than 5 uses and all
3896 // are add.
3897 // The heuristic is that if a use is not an add, then that use
3898 // cannot be fused into fma, therefore mul is still needed anyway.
3899 // If there are more than 4 uses, even if they are all add, fusing
3900 // them will increase register pressue.
3901 //
3902 int numUses = 0;
3903 int nonAddCount = 0;
3904 for (SDNode::use_iterator UI = N0.getNode()->use_begin(),
3905 UE = N0.getNode()->use_end();
3906 UI != UE; ++UI) {
3907 numUses++;
3908 SDNode *User = *UI;
3909 if (User->getOpcode() != ISD::FADD)
3910 ++nonAddCount;
3911 }
3912 if (numUses >= 5)
3913 return SDValue();
3914 if (nonAddCount) {
3915 int orderNo = N->getIROrder();
3916 int orderNo2 = N0.getNode()->getIROrder();
3917 // simple heuristics here for considering potential register
3918 // pressure, the logics here is that the differnce are used
3919 // to measure the distance between def and use, the longer distance
3920 // more likely cause register pressure.
3921 if (orderNo - orderNo2 < 500)
3922 return SDValue();
3923
3924 // Now, check if at least one of the FMUL's operands is live beyond the node N,
3925 // which guarantees that the FMA will not increase register pressure at node N.
3926 bool opIsLive = false;
3927 const SDNode *left = N0.getOperand(0).getNode();
3928 const SDNode *right = N0.getOperand(1).getNode();
3929
Benjamin Kramer619c4e52015-04-10 11:24:51 +00003930 if (isa<ConstantSDNode>(left) || isa<ConstantSDNode>(right))
Justin Holewinskieafe26d2014-06-27 18:35:37 +00003931 opIsLive = true;
3932
3933 if (!opIsLive)
3934 for (SDNode::use_iterator UI = left->use_begin(), UE = left->use_end(); UI != UE; ++UI) {
3935 SDNode *User = *UI;
3936 int orderNo3 = User->getIROrder();
3937 if (orderNo3 > orderNo) {
3938 opIsLive = true;
3939 break;
3940 }
3941 }
3942
3943 if (!opIsLive)
3944 for (SDNode::use_iterator UI = right->use_begin(), UE = right->use_end(); UI != UE; ++UI) {
3945 SDNode *User = *UI;
3946 int orderNo3 = User->getIROrder();
3947 if (orderNo3 > orderNo) {
3948 opIsLive = true;
3949 break;
3950 }
3951 }
3952
3953 if (!opIsLive)
3954 return SDValue();
3955 }
3956
3957 return DAG.getNode(ISD::FMA, SDLoc(N), VT,
3958 N0.getOperand(0), N0.getOperand(1), N1);
3959 }
3960 }
3961
3962 return SDValue();
3963}
3964
3965/// PerformADDCombine - Target-specific dag combine xforms for ISD::ADD.
3966///
3967static SDValue PerformADDCombine(SDNode *N,
3968 TargetLowering::DAGCombinerInfo &DCI,
3969 const NVPTXSubtarget &Subtarget,
3970 CodeGenOpt::Level OptLevel) {
3971 SDValue N0 = N->getOperand(0);
3972 SDValue N1 = N->getOperand(1);
3973
3974 // First try with the default operand order.
3975 SDValue Result = PerformADDCombineWithOperands(N, N0, N1, DCI, Subtarget,
3976 OptLevel);
3977 if (Result.getNode())
3978 return Result;
3979
3980 // If that didn't work, try again with the operands commuted.
3981 return PerformADDCombineWithOperands(N, N1, N0, DCI, Subtarget, OptLevel);
3982}
3983
3984static SDValue PerformANDCombine(SDNode *N,
3985 TargetLowering::DAGCombinerInfo &DCI) {
3986 // The type legalizer turns a vector load of i8 values into a zextload to i16
3987 // registers, optionally ANY_EXTENDs it (if target type is integer),
3988 // and ANDs off the high 8 bits. Since we turn this load into a
3989 // target-specific DAG node, the DAG combiner fails to eliminate these AND
3990 // nodes. Do that here.
3991 SDValue Val = N->getOperand(0);
3992 SDValue Mask = N->getOperand(1);
3993
3994 if (isa<ConstantSDNode>(Val)) {
3995 std::swap(Val, Mask);
3996 }
3997
3998 SDValue AExt;
3999 // Generally, we will see zextload -> IMOV16rr -> ANY_EXTEND -> and
4000 if (Val.getOpcode() == ISD::ANY_EXTEND) {
4001 AExt = Val;
4002 Val = Val->getOperand(0);
4003 }
4004
4005 if (Val->isMachineOpcode() && Val->getMachineOpcode() == NVPTX::IMOV16rr) {
4006 Val = Val->getOperand(0);
4007 }
4008
4009 if (Val->getOpcode() == NVPTXISD::LoadV2 ||
4010 Val->getOpcode() == NVPTXISD::LoadV4) {
4011 ConstantSDNode *MaskCnst = dyn_cast<ConstantSDNode>(Mask);
4012 if (!MaskCnst) {
4013 // Not an AND with a constant
4014 return SDValue();
4015 }
4016
4017 uint64_t MaskVal = MaskCnst->getZExtValue();
4018 if (MaskVal != 0xff) {
4019 // Not an AND that chops off top 8 bits
4020 return SDValue();
4021 }
4022
4023 MemSDNode *Mem = dyn_cast<MemSDNode>(Val);
4024 if (!Mem) {
4025 // Not a MemSDNode?!?
4026 return SDValue();
4027 }
4028
4029 EVT MemVT = Mem->getMemoryVT();
4030 if (MemVT != MVT::v2i8 && MemVT != MVT::v4i8) {
4031 // We only handle the i8 case
4032 return SDValue();
4033 }
4034
4035 unsigned ExtType =
4036 cast<ConstantSDNode>(Val->getOperand(Val->getNumOperands()-1))->
4037 getZExtValue();
4038 if (ExtType == ISD::SEXTLOAD) {
4039 // If for some reason the load is a sextload, the and is needed to zero
4040 // out the high 8 bits
4041 return SDValue();
4042 }
4043
4044 bool AddTo = false;
4045 if (AExt.getNode() != 0) {
4046 // Re-insert the ext as a zext.
4047 Val = DCI.DAG.getNode(ISD::ZERO_EXTEND, SDLoc(N),
4048 AExt.getValueType(), Val);
4049 AddTo = true;
4050 }
4051
4052 // If we get here, the AND is unnecessary. Just replace it with the load
4053 DCI.CombineTo(N, Val, AddTo);
4054 }
4055
4056 return SDValue();
4057}
4058
4059enum OperandSignedness {
4060 Signed = 0,
4061 Unsigned,
4062 Unknown
4063};
4064
4065/// IsMulWideOperandDemotable - Checks if the provided DAG node is an operand
4066/// that can be demoted to \p OptSize bits without loss of information. The
4067/// signedness of the operand, if determinable, is placed in \p S.
4068static bool IsMulWideOperandDemotable(SDValue Op,
4069 unsigned OptSize,
4070 OperandSignedness &S) {
4071 S = Unknown;
4072
4073 if (Op.getOpcode() == ISD::SIGN_EXTEND ||
4074 Op.getOpcode() == ISD::SIGN_EXTEND_INREG) {
4075 EVT OrigVT = Op.getOperand(0).getValueType();
Justin Holewinskiecca7152014-07-23 18:46:03 +00004076 if (OrigVT.getSizeInBits() <= OptSize) {
Justin Holewinskieafe26d2014-06-27 18:35:37 +00004077 S = Signed;
4078 return true;
4079 }
4080 } else if (Op.getOpcode() == ISD::ZERO_EXTEND) {
4081 EVT OrigVT = Op.getOperand(0).getValueType();
Justin Holewinskiecca7152014-07-23 18:46:03 +00004082 if (OrigVT.getSizeInBits() <= OptSize) {
Justin Holewinskieafe26d2014-06-27 18:35:37 +00004083 S = Unsigned;
4084 return true;
4085 }
4086 }
4087
4088 return false;
4089}
4090
4091/// AreMulWideOperandsDemotable - Checks if the given LHS and RHS operands can
4092/// be demoted to \p OptSize bits without loss of information. If the operands
4093/// contain a constant, it should appear as the RHS operand. The signedness of
4094/// the operands is placed in \p IsSigned.
4095static bool AreMulWideOperandsDemotable(SDValue LHS, SDValue RHS,
4096 unsigned OptSize,
4097 bool &IsSigned) {
4098
4099 OperandSignedness LHSSign;
4100
4101 // The LHS operand must be a demotable op
4102 if (!IsMulWideOperandDemotable(LHS, OptSize, LHSSign))
4103 return false;
4104
4105 // We should have been able to determine the signedness from the LHS
4106 if (LHSSign == Unknown)
4107 return false;
4108
4109 IsSigned = (LHSSign == Signed);
4110
4111 // The RHS can be a demotable op or a constant
4112 if (ConstantSDNode *CI = dyn_cast<ConstantSDNode>(RHS)) {
4113 APInt Val = CI->getAPIntValue();
4114 if (LHSSign == Unsigned) {
4115 if (Val.isIntN(OptSize)) {
4116 return true;
4117 }
4118 return false;
4119 } else {
4120 if (Val.isSignedIntN(OptSize)) {
4121 return true;
4122 }
4123 return false;
4124 }
4125 } else {
4126 OperandSignedness RHSSign;
4127 if (!IsMulWideOperandDemotable(RHS, OptSize, RHSSign))
4128 return false;
4129
4130 if (LHSSign != RHSSign)
4131 return false;
4132
4133 return true;
4134 }
4135}
4136
4137/// TryMULWIDECombine - Attempt to replace a multiply of M bits with a multiply
4138/// of M/2 bits that produces an M-bit result (i.e. mul.wide). This transform
4139/// works on both multiply DAG nodes and SHL DAG nodes with a constant shift
4140/// amount.
4141static SDValue TryMULWIDECombine(SDNode *N,
4142 TargetLowering::DAGCombinerInfo &DCI) {
4143 EVT MulType = N->getValueType(0);
4144 if (MulType != MVT::i32 && MulType != MVT::i64) {
4145 return SDValue();
4146 }
4147
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +00004148 SDLoc DL(N);
Justin Holewinskieafe26d2014-06-27 18:35:37 +00004149 unsigned OptSize = MulType.getSizeInBits() >> 1;
4150 SDValue LHS = N->getOperand(0);
4151 SDValue RHS = N->getOperand(1);
4152
4153 // Canonicalize the multiply so the constant (if any) is on the right
4154 if (N->getOpcode() == ISD::MUL) {
4155 if (isa<ConstantSDNode>(LHS)) {
4156 std::swap(LHS, RHS);
4157 }
4158 }
4159
4160 // If we have a SHL, determine the actual multiply amount
4161 if (N->getOpcode() == ISD::SHL) {
4162 ConstantSDNode *ShlRHS = dyn_cast<ConstantSDNode>(RHS);
4163 if (!ShlRHS) {
4164 return SDValue();
4165 }
4166
4167 APInt ShiftAmt = ShlRHS->getAPIntValue();
4168 unsigned BitWidth = MulType.getSizeInBits();
4169 if (ShiftAmt.sge(0) && ShiftAmt.slt(BitWidth)) {
4170 APInt MulVal = APInt(BitWidth, 1) << ShiftAmt;
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +00004171 RHS = DCI.DAG.getConstant(MulVal, DL, MulType);
Justin Holewinskieafe26d2014-06-27 18:35:37 +00004172 } else {
4173 return SDValue();
4174 }
4175 }
4176
4177 bool Signed;
4178 // Verify that our operands are demotable
4179 if (!AreMulWideOperandsDemotable(LHS, RHS, OptSize, Signed)) {
4180 return SDValue();
4181 }
4182
4183 EVT DemotedVT;
4184 if (MulType == MVT::i32) {
4185 DemotedVT = MVT::i16;
4186 } else {
4187 DemotedVT = MVT::i32;
4188 }
4189
4190 // Truncate the operands to the correct size. Note that these are just for
4191 // type consistency and will (likely) be eliminated in later phases.
4192 SDValue TruncLHS =
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +00004193 DCI.DAG.getNode(ISD::TRUNCATE, DL, DemotedVT, LHS);
Justin Holewinskieafe26d2014-06-27 18:35:37 +00004194 SDValue TruncRHS =
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +00004195 DCI.DAG.getNode(ISD::TRUNCATE, DL, DemotedVT, RHS);
Justin Holewinskieafe26d2014-06-27 18:35:37 +00004196
4197 unsigned Opc;
4198 if (Signed) {
4199 Opc = NVPTXISD::MUL_WIDE_SIGNED;
4200 } else {
4201 Opc = NVPTXISD::MUL_WIDE_UNSIGNED;
4202 }
4203
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +00004204 return DCI.DAG.getNode(Opc, DL, MulType, TruncLHS, TruncRHS);
Justin Holewinskieafe26d2014-06-27 18:35:37 +00004205}
4206
4207/// PerformMULCombine - Runs PTX-specific DAG combine patterns on MUL nodes.
4208static SDValue PerformMULCombine(SDNode *N,
4209 TargetLowering::DAGCombinerInfo &DCI,
4210 CodeGenOpt::Level OptLevel) {
4211 if (OptLevel > 0) {
4212 // Try mul.wide combining at OptLevel > 0
4213 SDValue Ret = TryMULWIDECombine(N, DCI);
4214 if (Ret.getNode())
4215 return Ret;
4216 }
4217
4218 return SDValue();
4219}
4220
4221/// PerformSHLCombine - Runs PTX-specific DAG combine patterns on SHL nodes.
4222static SDValue PerformSHLCombine(SDNode *N,
4223 TargetLowering::DAGCombinerInfo &DCI,
4224 CodeGenOpt::Level OptLevel) {
4225 if (OptLevel > 0) {
4226 // Try mul.wide combining at OptLevel > 0
4227 SDValue Ret = TryMULWIDECombine(N, DCI);
4228 if (Ret.getNode())
4229 return Ret;
4230 }
4231
4232 return SDValue();
4233}
4234
4235SDValue NVPTXTargetLowering::PerformDAGCombine(SDNode *N,
4236 DAGCombinerInfo &DCI) const {
Justin Holewinski511664d2014-07-23 17:40:45 +00004237 CodeGenOpt::Level OptLevel = getTargetMachine().getOptLevel();
Justin Holewinskieafe26d2014-06-27 18:35:37 +00004238 switch (N->getOpcode()) {
4239 default: break;
4240 case ISD::ADD:
4241 case ISD::FADD:
Eric Christopherbef0a372015-01-30 01:50:07 +00004242 return PerformADDCombine(N, DCI, STI, OptLevel);
Justin Holewinskieafe26d2014-06-27 18:35:37 +00004243 case ISD::MUL:
4244 return PerformMULCombine(N, DCI, OptLevel);
4245 case ISD::SHL:
4246 return PerformSHLCombine(N, DCI, OptLevel);
4247 case ISD::AND:
4248 return PerformANDCombine(N, DCI);
4249 }
4250 return SDValue();
4251}
4252
Justin Holewinskibe8dc642013-02-12 14:18:49 +00004253/// ReplaceVectorLoad - Convert vector loads into multi-output scalar loads.
4254static void ReplaceLoadVector(SDNode *N, SelectionDAG &DAG,
Justin Holewinskiac451062014-07-16 19:45:35 +00004255 const DataLayout *TD,
Justin Holewinski0497ab12013-03-30 14:29:21 +00004256 SmallVectorImpl<SDValue> &Results) {
Justin Holewinskibe8dc642013-02-12 14:18:49 +00004257 EVT ResVT = N->getValueType(0);
Andrew Trickef9de2a2013-05-25 02:42:55 +00004258 SDLoc DL(N);
Justin Holewinskibe8dc642013-02-12 14:18:49 +00004259
4260 assert(ResVT.isVector() && "Vector load must have vector type");
4261
4262 // We only handle "native" vector sizes for now, e.g. <4 x double> is not
4263 // legal. We can (and should) split that into 2 loads of <2 x double> here
4264 // but I'm leaving that as a TODO for now.
4265 assert(ResVT.isSimple() && "Can only handle simple types");
4266 switch (ResVT.getSimpleVT().SimpleTy) {
Justin Holewinski0497ab12013-03-30 14:29:21 +00004267 default:
4268 return;
Justin Holewinskibe8dc642013-02-12 14:18:49 +00004269 case MVT::v2i8:
4270 case MVT::v2i16:
4271 case MVT::v2i32:
4272 case MVT::v2i64:
4273 case MVT::v2f32:
4274 case MVT::v2f64:
4275 case MVT::v4i8:
4276 case MVT::v4i16:
4277 case MVT::v4i32:
4278 case MVT::v4f32:
4279 // This is a "native" vector type
4280 break;
4281 }
4282
Justin Holewinskiac451062014-07-16 19:45:35 +00004283 LoadSDNode *LD = cast<LoadSDNode>(N);
4284
4285 unsigned Align = LD->getAlignment();
4286 unsigned PrefAlign =
4287 TD->getPrefTypeAlignment(ResVT.getTypeForEVT(*DAG.getContext()));
4288 if (Align < PrefAlign) {
4289 // This load is not sufficiently aligned, so bail out and let this vector
4290 // load be scalarized. Note that we may still be able to emit smaller
4291 // vector loads. For example, if we are loading a <4 x float> with an
4292 // alignment of 8, this check will fail but the legalizer will try again
4293 // with 2 x <2 x float>, which will succeed with an alignment of 8.
4294 return;
4295 }
4296
Justin Holewinskibe8dc642013-02-12 14:18:49 +00004297 EVT EltVT = ResVT.getVectorElementType();
4298 unsigned NumElts = ResVT.getVectorNumElements();
4299
4300 // Since LoadV2 is a target node, we cannot rely on DAG type legalization.
4301 // Therefore, we must ensure the type is legal. For i1 and i8, we set the
Alp Tokercb402912014-01-24 17:20:08 +00004302 // loaded type to i16 and propagate the "real" type as the memory type.
Justin Holewinskibe8dc642013-02-12 14:18:49 +00004303 bool NeedTrunc = false;
4304 if (EltVT.getSizeInBits() < 16) {
4305 EltVT = MVT::i16;
4306 NeedTrunc = true;
4307 }
4308
4309 unsigned Opcode = 0;
4310 SDVTList LdResVTs;
4311
4312 switch (NumElts) {
Justin Holewinski0497ab12013-03-30 14:29:21 +00004313 default:
4314 return;
Justin Holewinskibe8dc642013-02-12 14:18:49 +00004315 case 2:
4316 Opcode = NVPTXISD::LoadV2;
4317 LdResVTs = DAG.getVTList(EltVT, EltVT, MVT::Other);
4318 break;
4319 case 4: {
4320 Opcode = NVPTXISD::LoadV4;
4321 EVT ListVTs[] = { EltVT, EltVT, EltVT, EltVT, MVT::Other };
Craig Topperabb4ac72014-04-16 06:10:51 +00004322 LdResVTs = DAG.getVTList(ListVTs);
Justin Holewinskibe8dc642013-02-12 14:18:49 +00004323 break;
4324 }
4325 }
4326
Justin Holewinskibe8dc642013-02-12 14:18:49 +00004327 // Copy regular operands
Benjamin Kramerea68a942015-02-19 15:26:17 +00004328 SmallVector<SDValue, 8> OtherOps(N->op_begin(), N->op_end());
Justin Holewinskibe8dc642013-02-12 14:18:49 +00004329
Justin Holewinskibe8dc642013-02-12 14:18:49 +00004330 // The select routine does not have access to the LoadSDNode instance, so
4331 // pass along the extension information
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +00004332 OtherOps.push_back(DAG.getIntPtrConstant(LD->getExtensionType(), DL));
Justin Holewinskibe8dc642013-02-12 14:18:49 +00004333
Craig Topper206fcd42014-04-26 19:29:41 +00004334 SDValue NewLD = DAG.getMemIntrinsicNode(Opcode, DL, LdResVTs, OtherOps,
4335 LD->getMemoryVT(),
Justin Holewinskibe8dc642013-02-12 14:18:49 +00004336 LD->getMemOperand());
4337
4338 SmallVector<SDValue, 4> ScalarRes;
4339
4340 for (unsigned i = 0; i < NumElts; ++i) {
4341 SDValue Res = NewLD.getValue(i);
4342 if (NeedTrunc)
4343 Res = DAG.getNode(ISD::TRUNCATE, DL, ResVT.getVectorElementType(), Res);
4344 ScalarRes.push_back(Res);
4345 }
4346
4347 SDValue LoadChain = NewLD.getValue(NumElts);
4348
Craig Topper48d114b2014-04-26 18:35:24 +00004349 SDValue BuildVec = DAG.getNode(ISD::BUILD_VECTOR, DL, ResVT, ScalarRes);
Justin Holewinskibe8dc642013-02-12 14:18:49 +00004350
4351 Results.push_back(BuildVec);
4352 Results.push_back(LoadChain);
4353}
4354
Justin Holewinski0497ab12013-03-30 14:29:21 +00004355static void ReplaceINTRINSIC_W_CHAIN(SDNode *N, SelectionDAG &DAG,
Justin Holewinskibe8dc642013-02-12 14:18:49 +00004356 SmallVectorImpl<SDValue> &Results) {
4357 SDValue Chain = N->getOperand(0);
4358 SDValue Intrin = N->getOperand(1);
Andrew Trickef9de2a2013-05-25 02:42:55 +00004359 SDLoc DL(N);
Justin Holewinskibe8dc642013-02-12 14:18:49 +00004360
4361 // Get the intrinsic ID
4362 unsigned IntrinNo = cast<ConstantSDNode>(Intrin.getNode())->getZExtValue();
Justin Holewinski0497ab12013-03-30 14:29:21 +00004363 switch (IntrinNo) {
4364 default:
4365 return;
Justin Holewinskibe8dc642013-02-12 14:18:49 +00004366 case Intrinsic::nvvm_ldg_global_i:
4367 case Intrinsic::nvvm_ldg_global_f:
4368 case Intrinsic::nvvm_ldg_global_p:
4369 case Intrinsic::nvvm_ldu_global_i:
4370 case Intrinsic::nvvm_ldu_global_f:
4371 case Intrinsic::nvvm_ldu_global_p: {
4372 EVT ResVT = N->getValueType(0);
4373
4374 if (ResVT.isVector()) {
4375 // Vector LDG/LDU
4376
4377 unsigned NumElts = ResVT.getVectorNumElements();
4378 EVT EltVT = ResVT.getVectorElementType();
4379
Justin Holewinskif8f70912013-06-28 17:57:59 +00004380 // Since LDU/LDG are target nodes, we cannot rely on DAG type
4381 // legalization.
Justin Holewinskibe8dc642013-02-12 14:18:49 +00004382 // Therefore, we must ensure the type is legal. For i1 and i8, we set the
Alp Tokercb402912014-01-24 17:20:08 +00004383 // loaded type to i16 and propagate the "real" type as the memory type.
Justin Holewinskibe8dc642013-02-12 14:18:49 +00004384 bool NeedTrunc = false;
4385 if (EltVT.getSizeInBits() < 16) {
4386 EltVT = MVT::i16;
4387 NeedTrunc = true;
4388 }
4389
4390 unsigned Opcode = 0;
4391 SDVTList LdResVTs;
4392
4393 switch (NumElts) {
Justin Holewinski0497ab12013-03-30 14:29:21 +00004394 default:
4395 return;
Justin Holewinskibe8dc642013-02-12 14:18:49 +00004396 case 2:
Justin Holewinski0497ab12013-03-30 14:29:21 +00004397 switch (IntrinNo) {
4398 default:
4399 return;
Justin Holewinskibe8dc642013-02-12 14:18:49 +00004400 case Intrinsic::nvvm_ldg_global_i:
4401 case Intrinsic::nvvm_ldg_global_f:
4402 case Intrinsic::nvvm_ldg_global_p:
4403 Opcode = NVPTXISD::LDGV2;
4404 break;
4405 case Intrinsic::nvvm_ldu_global_i:
4406 case Intrinsic::nvvm_ldu_global_f:
4407 case Intrinsic::nvvm_ldu_global_p:
4408 Opcode = NVPTXISD::LDUV2;
4409 break;
4410 }
4411 LdResVTs = DAG.getVTList(EltVT, EltVT, MVT::Other);
4412 break;
4413 case 4: {
Justin Holewinski0497ab12013-03-30 14:29:21 +00004414 switch (IntrinNo) {
4415 default:
4416 return;
Justin Holewinskibe8dc642013-02-12 14:18:49 +00004417 case Intrinsic::nvvm_ldg_global_i:
4418 case Intrinsic::nvvm_ldg_global_f:
4419 case Intrinsic::nvvm_ldg_global_p:
4420 Opcode = NVPTXISD::LDGV4;
4421 break;
4422 case Intrinsic::nvvm_ldu_global_i:
4423 case Intrinsic::nvvm_ldu_global_f:
4424 case Intrinsic::nvvm_ldu_global_p:
4425 Opcode = NVPTXISD::LDUV4;
4426 break;
4427 }
4428 EVT ListVTs[] = { EltVT, EltVT, EltVT, EltVT, MVT::Other };
Craig Topperabb4ac72014-04-16 06:10:51 +00004429 LdResVTs = DAG.getVTList(ListVTs);
Justin Holewinskibe8dc642013-02-12 14:18:49 +00004430 break;
4431 }
4432 }
4433
4434 SmallVector<SDValue, 8> OtherOps;
4435
4436 // Copy regular operands
4437
4438 OtherOps.push_back(Chain); // Chain
Justin Holewinski0497ab12013-03-30 14:29:21 +00004439 // Skip operand 1 (intrinsic ID)
Justin Holewinskif8f70912013-06-28 17:57:59 +00004440 // Others
Benjamin Kramerea68a942015-02-19 15:26:17 +00004441 OtherOps.append(N->op_begin() + 2, N->op_end());
Justin Holewinskibe8dc642013-02-12 14:18:49 +00004442
4443 MemIntrinsicSDNode *MemSD = cast<MemIntrinsicSDNode>(N);
4444
Craig Topper206fcd42014-04-26 19:29:41 +00004445 SDValue NewLD = DAG.getMemIntrinsicNode(Opcode, DL, LdResVTs, OtherOps,
4446 MemSD->getMemoryVT(),
4447 MemSD->getMemOperand());
Justin Holewinskibe8dc642013-02-12 14:18:49 +00004448
4449 SmallVector<SDValue, 4> ScalarRes;
4450
4451 for (unsigned i = 0; i < NumElts; ++i) {
4452 SDValue Res = NewLD.getValue(i);
4453 if (NeedTrunc)
Justin Holewinski0497ab12013-03-30 14:29:21 +00004454 Res =
4455 DAG.getNode(ISD::TRUNCATE, DL, ResVT.getVectorElementType(), Res);
Justin Holewinskibe8dc642013-02-12 14:18:49 +00004456 ScalarRes.push_back(Res);
4457 }
4458
4459 SDValue LoadChain = NewLD.getValue(NumElts);
4460
Justin Holewinski0497ab12013-03-30 14:29:21 +00004461 SDValue BuildVec =
Craig Topper48d114b2014-04-26 18:35:24 +00004462 DAG.getNode(ISD::BUILD_VECTOR, DL, ResVT, ScalarRes);
Justin Holewinskibe8dc642013-02-12 14:18:49 +00004463
4464 Results.push_back(BuildVec);
4465 Results.push_back(LoadChain);
4466 } else {
4467 // i8 LDG/LDU
4468 assert(ResVT.isSimple() && ResVT.getSimpleVT().SimpleTy == MVT::i8 &&
4469 "Custom handling of non-i8 ldu/ldg?");
4470
4471 // Just copy all operands as-is
Benjamin Kramerea68a942015-02-19 15:26:17 +00004472 SmallVector<SDValue, 4> Ops(N->op_begin(), N->op_end());
Justin Holewinskibe8dc642013-02-12 14:18:49 +00004473
4474 // Force output to i16
4475 SDVTList LdResVTs = DAG.getVTList(MVT::i16, MVT::Other);
4476
4477 MemIntrinsicSDNode *MemSD = cast<MemIntrinsicSDNode>(N);
4478
4479 // We make sure the memory type is i8, which will be used during isel
4480 // to select the proper instruction.
Justin Holewinski0497ab12013-03-30 14:29:21 +00004481 SDValue NewLD =
Craig Topper206fcd42014-04-26 19:29:41 +00004482 DAG.getMemIntrinsicNode(ISD::INTRINSIC_W_CHAIN, DL, LdResVTs, Ops,
4483 MVT::i8, MemSD->getMemOperand());
Justin Holewinskibe8dc642013-02-12 14:18:49 +00004484
Justin Holewinskie8c93e32013-07-01 12:58:48 +00004485 Results.push_back(DAG.getNode(ISD::TRUNCATE, DL, MVT::i8,
4486 NewLD.getValue(0)));
Justin Holewinskibe8dc642013-02-12 14:18:49 +00004487 Results.push_back(NewLD.getValue(1));
4488 }
4489 }
4490 }
4491}
4492
Justin Holewinski0497ab12013-03-30 14:29:21 +00004493void NVPTXTargetLowering::ReplaceNodeResults(
4494 SDNode *N, SmallVectorImpl<SDValue> &Results, SelectionDAG &DAG) const {
Justin Holewinskibe8dc642013-02-12 14:18:49 +00004495 switch (N->getOpcode()) {
Justin Holewinski0497ab12013-03-30 14:29:21 +00004496 default:
4497 report_fatal_error("Unhandled custom legalization");
Justin Holewinskibe8dc642013-02-12 14:18:49 +00004498 case ISD::LOAD:
Justin Holewinskiac451062014-07-16 19:45:35 +00004499 ReplaceLoadVector(N, DAG, getDataLayout(), Results);
Justin Holewinskibe8dc642013-02-12 14:18:49 +00004500 return;
4501 case ISD::INTRINSIC_W_CHAIN:
4502 ReplaceINTRINSIC_W_CHAIN(N, DAG, Results);
4503 return;
4504 }
4505}
Juergen Ributzkad12ccbd2013-11-19 00:57:56 +00004506
4507// Pin NVPTXSection's and NVPTXTargetObjectFile's vtables to this file.
4508void NVPTXSection::anchor() {}
4509
4510NVPTXTargetObjectFile::~NVPTXTargetObjectFile() {
4511 delete TextSection;
4512 delete DataSection;
4513 delete BSSSection;
4514 delete ReadOnlySection;
4515
4516 delete StaticCtorSection;
4517 delete StaticDtorSection;
4518 delete LSDASection;
4519 delete EHFrameSection;
4520 delete DwarfAbbrevSection;
4521 delete DwarfInfoSection;
4522 delete DwarfLineSection;
4523 delete DwarfFrameSection;
4524 delete DwarfPubTypesSection;
4525 delete DwarfDebugInlineSection;
4526 delete DwarfStrSection;
4527 delete DwarfLocSection;
4528 delete DwarfARangesSection;
4529 delete DwarfRangesSection;
Juergen Ributzkad12ccbd2013-11-19 00:57:56 +00004530}
Rafael Espindola35a12a82014-11-12 01:27:22 +00004531
Rafael Espindola0709a7b2015-05-21 19:20:38 +00004532MCSection *
Rafael Espindola35a12a82014-11-12 01:27:22 +00004533NVPTXTargetObjectFile::SelectSectionForGlobal(const GlobalValue *GV,
4534 SectionKind Kind, Mangler &Mang,
4535 const TargetMachine &TM) const {
4536 return getDataSection();
4537}