blob: 72d26261136ddc272791da89f90ba5b4f5758faa [file] [log] [blame]
Eli Friedmanda90dd62009-05-23 12:35:30 +00001//===-- LegalizeVectorOps.cpp - Implement SelectionDAG::LegalizeVectors ---===//
2//
3// The LLVM Compiler Infrastructure
4//
5// This file is distributed under the University of Illinois Open Source
6// License. See LICENSE.TXT for details.
7//
8//===----------------------------------------------------------------------===//
9//
10// This file implements the SelectionDAG::LegalizeVectors method.
11//
12// The vector legalizer looks for vector operations which might need to be
Eli Friedman3b251702009-05-27 07:58:35 +000013// scalarized and legalizes them. This is a separate step from Legalize because
14// scalarizing can introduce illegal types. For example, suppose we have an
Eli Friedmanda90dd62009-05-23 12:35:30 +000015// ISD::SDIV of type v2i64 on x86-32. The type is legal (for example, addition
16// on a v2i64 is legal), but ISD::SDIV isn't legal, so we have to unroll the
17// operation, which introduces nodes with the illegal type i64 which must be
18// expanded. Similarly, suppose we have an ISD::SRA of type v16i8 on PowerPC;
19// the operation must be unrolled, which introduces nodes with the illegal
20// type i8 which must be promoted.
21//
22// This does not legalize vector manipulations like ISD::BUILD_VECTOR,
Dan Gohmanf9bbcd12009-08-05 01:29:28 +000023// or operations that happen to take a vector which are custom-lowered;
24// the legalization for such operations never produces nodes
Eli Friedmanda90dd62009-05-23 12:35:30 +000025// with illegal types, so it's okay to put off legalizing them until
26// SelectionDAG::Legalize runs.
27//
28//===----------------------------------------------------------------------===//
29
30#include "llvm/CodeGen/SelectionDAG.h"
31#include "llvm/Target/TargetLowering.h"
32using namespace llvm;
33
34namespace {
35class VectorLegalizer {
36 SelectionDAG& DAG;
Dan Gohman21cea8a2010-04-17 15:26:15 +000037 const TargetLowering &TLI;
Eli Friedmanda90dd62009-05-23 12:35:30 +000038 bool Changed; // Keep track of whether anything changed
39
Chandler Carruth68adf152014-07-02 02:16:57 +000040 /// For nodes that are of legal width, and that have more than one use, this
41 /// map indicates what regularized operand to use. This allows us to avoid
42 /// legalizing the same thing more than once.
Preston Gurd0959bb72013-01-25 15:18:54 +000043 SmallDenseMap<SDValue, SDValue, 64> LegalizedNodes;
Eli Friedmanda90dd62009-05-23 12:35:30 +000044
Chandler Carruth68adf152014-07-02 02:16:57 +000045 /// \brief Adds a node to the translation cache.
Eli Friedmanda90dd62009-05-23 12:35:30 +000046 void AddLegalizedOperand(SDValue From, SDValue To) {
47 LegalizedNodes.insert(std::make_pair(From, To));
48 // If someone requests legalization of the new node, return itself.
49 if (From != To)
50 LegalizedNodes.insert(std::make_pair(To, To));
51 }
52
Chandler Carruth68adf152014-07-02 02:16:57 +000053 /// \brief Legalizes the given node.
Eli Friedmanda90dd62009-05-23 12:35:30 +000054 SDValue LegalizeOp(SDValue Op);
Chandler Carruth68adf152014-07-02 02:16:57 +000055
56 /// \brief Assuming the node is legal, "legalize" the results.
Eli Friedmanda90dd62009-05-23 12:35:30 +000057 SDValue TranslateLegalizeResults(SDValue Op, SDValue Result);
Chandler Carruth68adf152014-07-02 02:16:57 +000058
59 /// \brief Implements unrolling a VSETCC.
Eli Friedmanda90dd62009-05-23 12:35:30 +000060 SDValue UnrollVSETCC(SDValue Op);
Chandler Carruth68adf152014-07-02 02:16:57 +000061
Chandler Carruthc1bedac2014-07-02 06:23:34 +000062 /// \brief Implement expand-based legalization of vector operations.
63 ///
64 /// This is just a high-level routine to dispatch to specific code paths for
65 /// operations to legalize them.
66 SDValue Expand(SDValue Op);
67
Chandler Carruth68adf152014-07-02 02:16:57 +000068 /// \brief Implements expansion for FNEG; falls back to UnrollVectorOp if
69 /// FSUB isn't legal.
70 ///
71 /// Implements expansion for UINT_TO_FLOAT; falls back to UnrollVectorOp if
72 /// SINT_TO_FLOAT and SHR on vectors isn't legal.
Nadav Roteme7a101c2011-03-19 13:09:10 +000073 SDValue ExpandUINT_TO_FLOAT(SDValue Op);
Chandler Carruth68adf152014-07-02 02:16:57 +000074
75 /// \brief Implement expansion for SIGN_EXTEND_INREG using SRL and SRA.
Nadav Rotemdbe5c722013-01-11 22:57:48 +000076 SDValue ExpandSEXTINREG(SDValue Op);
Chandler Carruth68adf152014-07-02 02:16:57 +000077
Chandler Carruth0b666e02014-07-10 12:32:32 +000078 /// \brief Implement expansion for ANY_EXTEND_VECTOR_INREG.
79 ///
80 /// Shuffles the low lanes of the operand into place and bitcasts to the proper
81 /// type. The contents of the bits in the extended part of each element are
82 /// undef.
83 SDValue ExpandANY_EXTEND_VECTOR_INREG(SDValue Op);
84
85 /// \brief Implement expansion for SIGN_EXTEND_VECTOR_INREG.
86 ///
87 /// Shuffles the low lanes of the operand into place, bitcasts to the proper
88 /// type, then shifts left and arithmetic shifts right to introduce a sign
89 /// extension.
90 SDValue ExpandSIGN_EXTEND_VECTOR_INREG(SDValue Op);
91
Chandler Carruthafe4b252014-07-09 10:58:18 +000092 /// \brief Implement expansion for ZERO_EXTEND_VECTOR_INREG.
93 ///
94 /// Shuffles the low lanes of the operand into place and blends zeros into
95 /// the remaining lanes, finally bitcasting to the proper type.
96 SDValue ExpandZERO_EXTEND_VECTOR_INREG(SDValue Op);
97
Chandler Carruth68adf152014-07-02 02:16:57 +000098 /// \brief Expand bswap of vectors into a shuffle if legal.
Benjamin Kramerf3ad2352014-05-19 13:12:38 +000099 SDValue ExpandBSWAP(SDValue Op);
Chandler Carruth68adf152014-07-02 02:16:57 +0000100
101 /// \brief Implement vselect in terms of XOR, AND, OR when blend is not
102 /// supported by the target.
Nadav Rotem52202fb2011-09-13 19:17:42 +0000103 SDValue ExpandVSELECT(SDValue Op);
Nadav Rotemea973bd2012-08-30 19:17:29 +0000104 SDValue ExpandSELECT(SDValue Op);
Nadav Rotemebe13bc2011-10-15 07:41:10 +0000105 SDValue ExpandLoad(SDValue Op);
106 SDValue ExpandStore(SDValue Op);
Eli Friedmanda90dd62009-05-23 12:35:30 +0000107 SDValue ExpandFNEG(SDValue Op);
Matt Arsenaultd0792852015-12-14 17:25:38 +0000108 SDValue ExpandBITREVERSE(SDValue Op);
Simon Pilgrimd02c5522016-11-08 14:10:28 +0000109 SDValue ExpandCTLZ(SDValue Op);
110 SDValue ExpandCTTZ_ZERO_UNDEF(SDValue Op);
Chandler Carruth68adf152014-07-02 02:16:57 +0000111
112 /// \brief Implements vector promotion.
113 ///
114 /// This is essentially just bitcasting the operands to a different type and
115 /// bitcasting the result back to the original type.
Chandler Carruth1cfa8952014-07-02 03:07:11 +0000116 SDValue Promote(SDValue Op);
Chandler Carruth68adf152014-07-02 02:16:57 +0000117
118 /// \brief Implements [SU]INT_TO_FP vector promotion.
119 ///
120 /// This is a [zs]ext of the input operand to the next size up.
Chandler Carruth1cfa8952014-07-02 03:07:11 +0000121 SDValue PromoteINT_TO_FP(SDValue Op);
Chandler Carruth68adf152014-07-02 02:16:57 +0000122
123 /// \brief Implements FP_TO_[SU]INT vector promotion of the result type.
124 ///
125 /// It is promoted to the next size up integer type. The result is then
126 /// truncated back to the original type.
Chandler Carruth1cfa8952014-07-02 03:07:11 +0000127 SDValue PromoteFP_TO_INT(SDValue Op, bool isSigned);
Eli Friedmanda90dd62009-05-23 12:35:30 +0000128
Chandler Carruth68adf152014-07-02 02:16:57 +0000129public:
130 /// \brief Begin legalizer the vector operations in the DAG.
Eli Friedmanda90dd62009-05-23 12:35:30 +0000131 bool Run();
132 VectorLegalizer(SelectionDAG& dag) :
133 DAG(dag), TLI(dag.getTargetLoweringInfo()), Changed(false) {}
134};
135
136bool VectorLegalizer::Run() {
Nadav Rotemb7f90bd2013-02-22 23:33:30 +0000137 // Before we start legalizing vector nodes, check if there are any vectors.
138 bool HasVectors = false;
139 for (SelectionDAG::allnodes_iterator I = DAG.allnodes_begin(),
Benjamin Kramerb6d0bd42014-03-02 12:27:27 +0000140 E = std::prev(DAG.allnodes_end()); I != std::next(E); ++I) {
Nadav Rotemb7f90bd2013-02-22 23:33:30 +0000141 // Check if the values of the nodes contain vectors. We don't need to check
142 // the operands because we are going to check their values at some point.
143 for (SDNode::value_iterator J = I->value_begin(), E = I->value_end();
144 J != E; ++J)
145 HasVectors |= J->isVector();
146
147 // If we found a vector node we can start the legalization.
148 if (HasVectors)
149 break;
150 }
151
152 // If this basic block has no vectors then no need to legalize vectors.
153 if (!HasVectors)
154 return false;
155
Eli Friedmanda90dd62009-05-23 12:35:30 +0000156 // The legalize process is inherently a bottom-up recursive process (users
157 // legalize their uses before themselves). Given infinite stack space, we
158 // could just start legalizing on the root and traverse the whole graph. In
159 // practice however, this causes us to run out of stack space on large basic
160 // blocks. To avoid this problem, compute an ordering of the nodes where each
161 // node is only legalized after all of its operands are legalized.
162 DAG.AssignTopologicalOrder();
163 for (SelectionDAG::allnodes_iterator I = DAG.allnodes_begin(),
Benjamin Kramerb6d0bd42014-03-02 12:27:27 +0000164 E = std::prev(DAG.allnodes_end()); I != std::next(E); ++I)
Duncan P. N. Exon Smithe400a7d2015-10-13 19:47:46 +0000165 LegalizeOp(SDValue(&*I, 0));
Eli Friedmanda90dd62009-05-23 12:35:30 +0000166
167 // Finally, it's possible the root changed. Get the new root.
168 SDValue OldRoot = DAG.getRoot();
169 assert(LegalizedNodes.count(OldRoot) && "Root didn't get legalized?");
170 DAG.setRoot(LegalizedNodes[OldRoot]);
171
172 LegalizedNodes.clear();
173
174 // Remove dead nodes now.
175 DAG.RemoveDeadNodes();
176
177 return Changed;
178}
179
180SDValue VectorLegalizer::TranslateLegalizeResults(SDValue Op, SDValue Result) {
181 // Generic legalization: just pass the operand through.
182 for (unsigned i = 0, e = Op.getNode()->getNumValues(); i != e; ++i)
183 AddLegalizedOperand(Op.getValue(i), Result.getValue(i));
184 return Result.getValue(Op.getResNo());
185}
186
187SDValue VectorLegalizer::LegalizeOp(SDValue Op) {
188 // Note that LegalizeOp may be reentered even from single-use nodes, which
189 // means that we always must cache transformed nodes.
190 DenseMap<SDValue, SDValue>::iterator I = LegalizedNodes.find(Op);
191 if (I != LegalizedNodes.end()) return I->second;
192
193 SDNode* Node = Op.getNode();
194
195 // Legalize the operands
196 SmallVector<SDValue, 8> Ops;
Pete Cooper8fc121d2015-06-26 19:08:33 +0000197 for (const SDValue &Op : Node->op_values())
198 Ops.push_back(LegalizeOp(Op));
Eli Friedmanda90dd62009-05-23 12:35:30 +0000199
Craig Topper8c0b4d02014-04-28 05:57:50 +0000200 SDValue Result = SDValue(DAG.UpdateNodeOperands(Op.getNode(), Ops), 0);
Eli Friedmanda90dd62009-05-23 12:35:30 +0000201
Elena Demikhovsky1b60ed72015-05-03 07:12:25 +0000202 bool HasVectorValue = false;
Nadav Rotemebe13bc2011-10-15 07:41:10 +0000203 if (Op.getOpcode() == ISD::LOAD) {
204 LoadSDNode *LD = cast<LoadSDNode>(Op.getNode());
205 ISD::LoadExtType ExtType = LD->getExtensionType();
Chandler Carruth80b86942014-07-24 22:09:56 +0000206 if (LD->getMemoryVT().isVector() && ExtType != ISD::NON_EXTLOAD)
Ahmed Bougacha2b6917b2015-01-08 00:51:32 +0000207 switch (TLI.getLoadExtAction(LD->getExtensionType(), LD->getValueType(0),
208 LD->getMemoryVT())) {
Chandler Carruth80b86942014-07-24 22:09:56 +0000209 default: llvm_unreachable("This action is not supported yet!");
210 case TargetLowering::Legal:
Nadav Rotemebe13bc2011-10-15 07:41:10 +0000211 return TranslateLegalizeResults(Op, Result);
Chandler Carruth80b86942014-07-24 22:09:56 +0000212 case TargetLowering::Custom:
213 if (SDValue Lowered = TLI.LowerOperation(Result, DAG)) {
Hal Finkelcec70132015-02-24 12:59:47 +0000214 if (Lowered == Result)
215 return TranslateLegalizeResults(Op, Lowered);
Chandler Carruth80b86942014-07-24 22:09:56 +0000216 Changed = true;
217 if (Lowered->getNumValues() != Op->getNumValues()) {
218 // This expanded to something other than the load. Assume the
219 // lowering code took care of any chain values, and just handle the
220 // returned value.
221 assert(Result.getValue(1).use_empty() &&
222 "There are still live users of the old chain!");
223 return LegalizeOp(Lowered);
Chandler Carruth80b86942014-07-24 22:09:56 +0000224 }
Mehdi Amini891c0972015-10-27 08:12:08 +0000225 return TranslateLegalizeResults(Op, Lowered);
Chandler Carruth80b86942014-07-24 22:09:56 +0000226 }
227 case TargetLowering::Expand:
228 Changed = true;
229 return LegalizeOp(ExpandLoad(Op));
230 }
Nadav Rotemebe13bc2011-10-15 07:41:10 +0000231 } else if (Op.getOpcode() == ISD::STORE) {
232 StoreSDNode *ST = cast<StoreSDNode>(Op.getNode());
233 EVT StVT = ST->getMemoryVT();
Patrik Hagglundd7cdcf82012-12-19 08:28:51 +0000234 MVT ValVT = ST->getValue().getSimpleValueType();
Nadav Rotemebe13bc2011-10-15 07:41:10 +0000235 if (StVT.isVector() && ST->isTruncatingStore())
Eric Christopher4675c432015-11-25 09:11:53 +0000236 switch (TLI.getTruncStoreAction(ValVT, StVT)) {
Craig Topperee4dab52012-02-05 08:31:47 +0000237 default: llvm_unreachable("This action is not supported yet!");
Nadav Rotemebe13bc2011-10-15 07:41:10 +0000238 case TargetLowering::Legal:
239 return TranslateLegalizeResults(Op, Result);
Hal Finkelcec70132015-02-24 12:59:47 +0000240 case TargetLowering::Custom: {
241 SDValue Lowered = TLI.LowerOperation(Result, DAG);
242 Changed = Lowered != Result;
243 return TranslateLegalizeResults(Op, Lowered);
244 }
Nadav Rotemebe13bc2011-10-15 07:41:10 +0000245 case TargetLowering::Expand:
246 Changed = true;
247 return LegalizeOp(ExpandStore(Op));
248 }
Elena Demikhovsky33e61ec2015-12-07 13:39:24 +0000249 } else if (Op.getOpcode() == ISD::MSCATTER || Op.getOpcode() == ISD::MSTORE)
Elena Demikhovsky1b60ed72015-05-03 07:12:25 +0000250 HasVectorValue = true;
Nadav Rotemebe13bc2011-10-15 07:41:10 +0000251
Eli Friedmanda90dd62009-05-23 12:35:30 +0000252 for (SDNode::value_iterator J = Node->value_begin(), E = Node->value_end();
253 J != E;
254 ++J)
255 HasVectorValue |= J->isVector();
256 if (!HasVectorValue)
257 return TranslateLegalizeResults(Op, Result);
258
Owen Anderson53aa7a92009-08-10 22:56:29 +0000259 EVT QueryType;
Eli Friedmanda90dd62009-05-23 12:35:30 +0000260 switch (Op.getOpcode()) {
261 default:
262 return TranslateLegalizeResults(Op, Result);
263 case ISD::ADD:
264 case ISD::SUB:
265 case ISD::MUL:
266 case ISD::SDIV:
267 case ISD::UDIV:
268 case ISD::SREM:
269 case ISD::UREM:
Artyom Skrobovb844fa72015-10-20 13:06:02 +0000270 case ISD::SDIVREM:
271 case ISD::UDIVREM:
Eli Friedmanda90dd62009-05-23 12:35:30 +0000272 case ISD::FADD:
273 case ISD::FSUB:
274 case ISD::FMUL:
275 case ISD::FDIV:
276 case ISD::FREM:
277 case ISD::AND:
278 case ISD::OR:
279 case ISD::XOR:
280 case ISD::SHL:
281 case ISD::SRA:
282 case ISD::SRL:
283 case ISD::ROTL:
284 case ISD::ROTR:
Hal Finkel5c968d92014-02-03 17:27:25 +0000285 case ISD::BSWAP:
Matt Arsenaultd0792852015-12-14 17:25:38 +0000286 case ISD::BITREVERSE:
Eli Friedmanda90dd62009-05-23 12:35:30 +0000287 case ISD::CTLZ:
Chandler Carruth637cc6a2011-12-13 01:56:10 +0000288 case ISD::CTTZ:
289 case ISD::CTLZ_ZERO_UNDEF:
290 case ISD::CTTZ_ZERO_UNDEF:
Eli Friedmanda90dd62009-05-23 12:35:30 +0000291 case ISD::CTPOP:
292 case ISD::SELECT:
Nadav Rotem52202fb2011-09-13 19:17:42 +0000293 case ISD::VSELECT:
Eli Friedmanda90dd62009-05-23 12:35:30 +0000294 case ISD::SELECT_CC:
Duncan Sandsf2641e12011-09-06 19:07:46 +0000295 case ISD::SETCC:
Eli Friedmanda90dd62009-05-23 12:35:30 +0000296 case ISD::ZERO_EXTEND:
297 case ISD::ANY_EXTEND:
298 case ISD::TRUNCATE:
299 case ISD::SIGN_EXTEND:
Eli Friedmanda90dd62009-05-23 12:35:30 +0000300 case ISD::FP_TO_SINT:
301 case ISD::FP_TO_UINT:
302 case ISD::FNEG:
303 case ISD::FABS:
Matt Arsenault7c936902014-10-21 23:01:01 +0000304 case ISD::FMINNUM:
305 case ISD::FMAXNUM:
James Molloy01cdecc2015-08-11 09:13:05 +0000306 case ISD::FMINNAN:
307 case ISD::FMAXNAN:
Hal Finkel0c5c01aa2013-08-19 23:35:46 +0000308 case ISD::FCOPYSIGN:
Eli Friedmanda90dd62009-05-23 12:35:30 +0000309 case ISD::FSQRT:
310 case ISD::FSIN:
311 case ISD::FCOS:
312 case ISD::FPOWI:
313 case ISD::FPOW:
314 case ISD::FLOG:
315 case ISD::FLOG2:
316 case ISD::FLOG10:
317 case ISD::FEXP:
318 case ISD::FEXP2:
319 case ISD::FCEIL:
320 case ISD::FTRUNC:
321 case ISD::FRINT:
322 case ISD::FNEARBYINT:
Hal Finkel171817e2013-08-07 22:49:12 +0000323 case ISD::FROUND:
Eli Friedmanda90dd62009-05-23 12:35:30 +0000324 case ISD::FFLOOR:
Eli Friedmane6385e62012-11-15 22:44:27 +0000325 case ISD::FP_ROUND:
Eli Friedman30834942012-11-17 01:52:46 +0000326 case ISD::FP_EXTEND:
Craig Topper2da13f92012-08-30 07:34:22 +0000327 case ISD::FMA:
Nadav Rotem771f2962011-07-14 11:11:14 +0000328 case ISD::SIGN_EXTEND_INREG:
Chandler Carruth0b666e02014-07-10 12:32:32 +0000329 case ISD::ANY_EXTEND_VECTOR_INREG:
330 case ISD::SIGN_EXTEND_VECTOR_INREG:
Chandler Carruthafe4b252014-07-09 10:58:18 +0000331 case ISD::ZERO_EXTEND_VECTOR_INREG:
James Molloy7e9776b2015-05-15 09:03:15 +0000332 case ISD::SMIN:
333 case ISD::SMAX:
334 case ISD::UMIN:
335 case ISD::UMAX:
Eli Friedmanaea9b652009-06-06 03:27:50 +0000336 QueryType = Node->getValueType(0);
337 break;
Dan Gohman6bd3ef82010-01-09 02:13:55 +0000338 case ISD::FP_ROUND_INREG:
339 QueryType = cast<VTSDNode>(Node->getOperand(1))->getVT();
340 break;
Eli Friedmanaea9b652009-06-06 03:27:50 +0000341 case ISD::SINT_TO_FP:
342 case ISD::UINT_TO_FP:
343 QueryType = Node->getOperand(0).getValueType();
Eli Friedmanda90dd62009-05-23 12:35:30 +0000344 break;
Elena Demikhovsky1b60ed72015-05-03 07:12:25 +0000345 case ISD::MSCATTER:
346 QueryType = cast<MaskedScatterSDNode>(Node)->getValue().getValueType();
347 break;
Elena Demikhovsky33e61ec2015-12-07 13:39:24 +0000348 case ISD::MSTORE:
349 QueryType = cast<MaskedStoreSDNode>(Node)->getValue().getValueType();
350 break;
Eli Friedmanda90dd62009-05-23 12:35:30 +0000351 }
352
Eli Friedmanaea9b652009-06-06 03:27:50 +0000353 switch (TLI.getOperationAction(Node->getOpcode(), QueryType)) {
Artyom Skrobovc7368632015-10-20 15:06:37 +0000354 default: llvm_unreachable("This action is not supported yet!");
Eli Friedmanda90dd62009-05-23 12:35:30 +0000355 case TargetLowering::Promote:
Chandler Carruth2746c282014-07-02 03:07:15 +0000356 Result = Promote(Op);
357 Changed = true;
Eli Friedmanda90dd62009-05-23 12:35:30 +0000358 break;
Chandler Carruth2746c282014-07-02 03:07:15 +0000359 case TargetLowering::Legal:
360 break;
Eli Friedmanda90dd62009-05-23 12:35:30 +0000361 case TargetLowering::Custom: {
Ahmed Bougachaf8dfb472016-02-09 22:54:12 +0000362 if (SDValue Tmp1 = TLI.LowerOperation(Op, DAG)) {
Eli Friedmanda90dd62009-05-23 12:35:30 +0000363 Result = Tmp1;
364 break;
365 }
Justin Bognercd1d5aa2016-08-17 20:30:52 +0000366 LLVM_FALLTHROUGH;
Eli Friedmanda90dd62009-05-23 12:35:30 +0000367 }
368 case TargetLowering::Expand:
Chandler Carruthc1bedac2014-07-02 06:23:34 +0000369 Result = Expand(Op);
Eli Friedmanda90dd62009-05-23 12:35:30 +0000370 }
371
372 // Make sure that the generated code is itself legal.
373 if (Result != Op) {
374 Result = LegalizeOp(Result);
375 Changed = true;
376 }
377
378 // Note that LegalizeOp may be reentered even from single-use nodes, which
379 // means that we always must cache transformed nodes.
380 AddLegalizedOperand(Op, Result);
381 return Result;
382}
383
Chandler Carruth1cfa8952014-07-02 03:07:11 +0000384SDValue VectorLegalizer::Promote(SDValue Op) {
Chandler Carruth2746c282014-07-02 03:07:15 +0000385 // For a few operations there is a specific concept for promotion based on
386 // the operand's type.
387 switch (Op.getOpcode()) {
388 case ISD::SINT_TO_FP:
389 case ISD::UINT_TO_FP:
390 // "Promote" the operation by extending the operand.
391 return PromoteINT_TO_FP(Op);
Chandler Carruth2746c282014-07-02 03:07:15 +0000392 case ISD::FP_TO_UINT:
393 case ISD::FP_TO_SINT:
394 // Promote the operation by extending the operand.
395 return PromoteFP_TO_INT(Op, Op->getOpcode() == ISD::FP_TO_SINT);
Chandler Carruth2746c282014-07-02 03:07:15 +0000396 }
397
Oliver Stannard89d15422014-08-27 16:16:04 +0000398 // There are currently two cases of vector promotion:
399 // 1) Bitcasting a vector of integers to a different type to a vector of the
Sanjay Patelf1765662015-03-27 21:45:18 +0000400 // same overall length. For example, x86 promotes ISD::AND v2i32 to v1i64.
401 // 2) Extending a vector of floats to a vector of the same number of larger
Oliver Stannard89d15422014-08-27 16:16:04 +0000402 // floats. For example, AArch64 promotes ISD::FADD on v4f16 to v4f32.
Patrik Hagglundfd41b5b2012-12-19 11:21:04 +0000403 MVT VT = Op.getSimpleValueType();
Eli Friedmanda90dd62009-05-23 12:35:30 +0000404 assert(Op.getNode()->getNumValues() == 1 &&
405 "Can't promote a vector with multiple results!");
Patrik Hagglundfd41b5b2012-12-19 11:21:04 +0000406 MVT NVT = TLI.getTypeToPromoteTo(Op.getOpcode(), VT);
Benjamin Kramer351d53c2013-05-28 16:31:26 +0000407 SDLoc dl(Op);
Eli Friedmanda90dd62009-05-23 12:35:30 +0000408 SmallVector<SDValue, 4> Operands(Op.getNumOperands());
409
410 for (unsigned j = 0; j != Op.getNumOperands(); ++j) {
411 if (Op.getOperand(j).getValueType().isVector())
Oliver Stannard89d15422014-08-27 16:16:04 +0000412 if (Op.getOperand(j)
413 .getValueType()
414 .getVectorElementType()
Hal Finkel271e9f22015-02-12 22:43:52 +0000415 .isFloatingPoint() &&
416 NVT.isVector() && NVT.getVectorElementType().isFloatingPoint())
Oliver Stannard89d15422014-08-27 16:16:04 +0000417 Operands[j] = DAG.getNode(ISD::FP_EXTEND, dl, NVT, Op.getOperand(j));
418 else
419 Operands[j] = DAG.getNode(ISD::BITCAST, dl, NVT, Op.getOperand(j));
Eli Friedmanda90dd62009-05-23 12:35:30 +0000420 else
421 Operands[j] = Op.getOperand(j);
422 }
Matt Arsenaultd0792852015-12-14 17:25:38 +0000423
Sanjay Patela2607012015-09-16 16:31:21 +0000424 Op = DAG.getNode(Op.getOpcode(), dl, NVT, Operands, Op.getNode()->getFlags());
Hal Finkel271e9f22015-02-12 22:43:52 +0000425 if ((VT.isFloatingPoint() && NVT.isFloatingPoint()) ||
426 (VT.isVector() && VT.getVectorElementType().isFloatingPoint() &&
427 NVT.isVector() && NVT.getVectorElementType().isFloatingPoint()))
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +0000428 return DAG.getNode(ISD::FP_ROUND, dl, VT, Op, DAG.getIntPtrConstant(0, dl));
Oliver Stannard89d15422014-08-27 16:16:04 +0000429 else
430 return DAG.getNode(ISD::BITCAST, dl, VT, Op);
Eli Friedmanda90dd62009-05-23 12:35:30 +0000431}
432
Chandler Carruth1cfa8952014-07-02 03:07:11 +0000433SDValue VectorLegalizer::PromoteINT_TO_FP(SDValue Op) {
Jim Grosbache0c10d82012-06-28 21:03:44 +0000434 // INT_TO_FP operations may require the input operand be promoted even
435 // when the type is otherwise legal.
436 EVT VT = Op.getOperand(0).getValueType();
437 assert(Op.getNode()->getNumValues() == 1 &&
438 "Can't promote a vector with multiple results!");
439
440 // Normal getTypeToPromoteTo() doesn't work here, as that will promote
441 // by widening the vector w/ the same element width and twice the number
442 // of elements. We want the other way around, the same number of elements,
443 // each twice the width.
444 //
445 // Increase the bitwidth of the element to the next pow-of-two
446 // (which is greater than 8 bits).
Jim Grosbache0c10d82012-06-28 21:03:44 +0000447
Adam Nemet24381f12014-03-17 17:06:14 +0000448 EVT NVT = VT.widenIntegerVectorElementType(*DAG.getContext());
449 assert(NVT.isSimple() && "Promoting to a non-simple vector type!");
Benjamin Kramer351d53c2013-05-28 16:31:26 +0000450 SDLoc dl(Op);
Jim Grosbache0c10d82012-06-28 21:03:44 +0000451 SmallVector<SDValue, 4> Operands(Op.getNumOperands());
452
453 unsigned Opc = Op.getOpcode() == ISD::UINT_TO_FP ? ISD::ZERO_EXTEND :
454 ISD::SIGN_EXTEND;
455 for (unsigned j = 0; j != Op.getNumOperands(); ++j) {
456 if (Op.getOperand(j).getValueType().isVector())
457 Operands[j] = DAG.getNode(Opc, dl, NVT, Op.getOperand(j));
458 else
459 Operands[j] = Op.getOperand(j);
460 }
461
Craig Topper48d114b2014-04-26 18:35:24 +0000462 return DAG.getNode(Op.getOpcode(), dl, Op.getValueType(), Operands);
Jim Grosbache0c10d82012-06-28 21:03:44 +0000463}
464
Adam Nemet24381f12014-03-17 17:06:14 +0000465// For FP_TO_INT we promote the result type to a vector type with wider
466// elements and then truncate the result. This is different from the default
467// PromoteVector which uses bitcast to promote thus assumning that the
468// promoted vector type has the same overall size.
Chandler Carruth1cfa8952014-07-02 03:07:11 +0000469SDValue VectorLegalizer::PromoteFP_TO_INT(SDValue Op, bool isSigned) {
Adam Nemet24381f12014-03-17 17:06:14 +0000470 assert(Op.getNode()->getNumValues() == 1 &&
471 "Can't promote a vector with multiple results!");
472 EVT VT = Op.getValueType();
473
474 EVT NewVT;
475 unsigned NewOpc;
476 while (1) {
477 NewVT = VT.widenIntegerVectorElementType(*DAG.getContext());
478 assert(NewVT.isSimple() && "Promoting to a non-simple vector type!");
479 if (TLI.isOperationLegalOrCustom(ISD::FP_TO_SINT, NewVT)) {
480 NewOpc = ISD::FP_TO_SINT;
481 break;
482 }
483 if (!isSigned && TLI.isOperationLegalOrCustom(ISD::FP_TO_UINT, NewVT)) {
484 NewOpc = ISD::FP_TO_UINT;
485 break;
486 }
487 }
488
489 SDLoc loc(Op);
490 SDValue promoted = DAG.getNode(NewOpc, SDLoc(Op), NewVT, Op.getOperand(0));
491 return DAG.getNode(ISD::TRUNCATE, SDLoc(Op), VT, promoted);
492}
493
Nadav Rotemebe13bc2011-10-15 07:41:10 +0000494
495SDValue VectorLegalizer::ExpandLoad(SDValue Op) {
Nadav Rotemebe13bc2011-10-15 07:41:10 +0000496 LoadSDNode *LD = cast<LoadSDNode>(Op.getNode());
Nadav Rotemebe13bc2011-10-15 07:41:10 +0000497
Matt Arsenaulta4b1b6e2016-03-30 21:15:10 +0000498 EVT SrcVT = LD->getMemoryVT();
499 EVT SrcEltVT = SrcVT.getScalarType();
Nadav Rotemebe13bc2011-10-15 07:41:10 +0000500 unsigned NumElem = SrcVT.getVectorNumElements();
Nadav Rotemebe13bc2011-10-15 07:41:10 +0000501
Nadav Rotemebe13bc2011-10-15 07:41:10 +0000502
Matt Arsenaulta4b1b6e2016-03-30 21:15:10 +0000503 SDValue NewChain;
504 SDValue Value;
Michael Liao7fb39662013-02-20 18:04:21 +0000505 if (SrcVT.getVectorNumElements() > 1 && !SrcEltVT.isByteSized()) {
Matt Arsenaulta4b1b6e2016-03-30 21:15:10 +0000506 SDLoc dl(Op);
507
508 SmallVector<SDValue, 8> Vals;
509 SmallVector<SDValue, 8> LoadChains;
510
511 EVT DstEltVT = LD->getValueType(0).getScalarType();
512 SDValue Chain = LD->getChain();
513 SDValue BasePTR = LD->getBasePtr();
514 ISD::LoadExtType ExtType = LD->getExtensionType();
515
Michael Liao7fb39662013-02-20 18:04:21 +0000516 // When elements in a vector is not byte-addressable, we cannot directly
517 // load each element by advancing pointer, which could only address bytes.
518 // Instead, we load all significant words, mask bits off, and concatenate
519 // them to form each element. Finally, they are extended to destination
520 // scalar type to build the destination vector.
Mehdi Amini44ede332015-07-09 02:09:04 +0000521 EVT WideVT = TLI.getPointerTy(DAG.getDataLayout());
Nadav Rotem75c22292011-10-18 22:32:43 +0000522
Michael Liao7fb39662013-02-20 18:04:21 +0000523 assert(WideVT.isRound() &&
524 "Could not handle the sophisticated case when the widest integer is"
525 " not power of 2.");
526 assert(WideVT.bitsGE(SrcEltVT) &&
527 "Type is not legalized?");
528
529 unsigned WideBytes = WideVT.getStoreSize();
530 unsigned Offset = 0;
531 unsigned RemainingBytes = SrcVT.getStoreSize();
532 SmallVector<SDValue, 8> LoadVals;
533
534 while (RemainingBytes > 0) {
535 SDValue ScalarLoad;
536 unsigned LoadBytes = WideBytes;
537
538 if (RemainingBytes >= LoadBytes) {
Justin Lebar9c375812016-07-15 18:27:10 +0000539 ScalarLoad =
540 DAG.getLoad(WideVT, dl, Chain, BasePTR,
541 LD->getPointerInfo().getWithOffset(Offset),
542 MinAlign(LD->getAlignment(), Offset),
543 LD->getMemOperand()->getFlags(), LD->getAAInfo());
Michael Liao7fb39662013-02-20 18:04:21 +0000544 } else {
545 EVT LoadVT = WideVT;
546 while (RemainingBytes < LoadBytes) {
547 LoadBytes >>= 1; // Reduce the load size by half.
548 LoadVT = EVT::getIntegerVT(*DAG.getContext(), LoadBytes << 3);
549 }
Justin Lebar9c375812016-07-15 18:27:10 +0000550 ScalarLoad =
551 DAG.getExtLoad(ISD::EXTLOAD, dl, WideVT, Chain, BasePTR,
552 LD->getPointerInfo().getWithOffset(Offset), LoadVT,
553 MinAlign(LD->getAlignment(), Offset),
554 LD->getMemOperand()->getFlags(), LD->getAAInfo());
Michael Liao7fb39662013-02-20 18:04:21 +0000555 }
556
557 RemainingBytes -= LoadBytes;
558 Offset += LoadBytes;
559 BasePTR = DAG.getNode(ISD::ADD, dl, BasePTR.getValueType(), BasePTR,
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +0000560 DAG.getConstant(LoadBytes, dl,
561 BasePTR.getValueType()));
Michael Liao7fb39662013-02-20 18:04:21 +0000562
563 LoadVals.push_back(ScalarLoad.getValue(0));
564 LoadChains.push_back(ScalarLoad.getValue(1));
565 }
566
567 // Extract bits, pack and extend/trunc them into destination type.
568 unsigned SrcEltBits = SrcEltVT.getSizeInBits();
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +0000569 SDValue SrcEltBitMask = DAG.getConstant((1U << SrcEltBits) - 1, dl, WideVT);
Michael Liao7fb39662013-02-20 18:04:21 +0000570
571 unsigned BitOffset = 0;
572 unsigned WideIdx = 0;
573 unsigned WideBits = WideVT.getSizeInBits();
574
575 for (unsigned Idx = 0; Idx != NumElem; ++Idx) {
576 SDValue Lo, Hi, ShAmt;
577
578 if (BitOffset < WideBits) {
Mehdi Amini9639d652015-07-09 02:09:20 +0000579 ShAmt = DAG.getConstant(
580 BitOffset, dl, TLI.getShiftAmountTy(WideVT, DAG.getDataLayout()));
Michael Liao7fb39662013-02-20 18:04:21 +0000581 Lo = DAG.getNode(ISD::SRL, dl, WideVT, LoadVals[WideIdx], ShAmt);
582 Lo = DAG.getNode(ISD::AND, dl, WideVT, Lo, SrcEltBitMask);
583 }
584
585 BitOffset += SrcEltBits;
586 if (BitOffset >= WideBits) {
587 WideIdx++;
Michael Kupersteincd63c5f2015-02-04 18:54:01 +0000588 BitOffset -= WideBits;
589 if (BitOffset > 0) {
Mehdi Amini9639d652015-07-09 02:09:20 +0000590 ShAmt = DAG.getConstant(
591 SrcEltBits - BitOffset, dl,
592 TLI.getShiftAmountTy(WideVT, DAG.getDataLayout()));
Michael Liao7fb39662013-02-20 18:04:21 +0000593 Hi = DAG.getNode(ISD::SHL, dl, WideVT, LoadVals[WideIdx], ShAmt);
594 Hi = DAG.getNode(ISD::AND, dl, WideVT, Hi, SrcEltBitMask);
595 }
596 }
597
598 if (Hi.getNode())
599 Lo = DAG.getNode(ISD::OR, dl, WideVT, Lo, Hi);
600
601 switch (ExtType) {
602 default: llvm_unreachable("Unknown extended-load op!");
603 case ISD::EXTLOAD:
604 Lo = DAG.getAnyExtOrTrunc(Lo, dl, DstEltVT);
605 break;
606 case ISD::ZEXTLOAD:
607 Lo = DAG.getZExtOrTrunc(Lo, dl, DstEltVT);
608 break;
609 case ISD::SEXTLOAD:
Mehdi Amini9639d652015-07-09 02:09:20 +0000610 ShAmt =
611 DAG.getConstant(WideBits - SrcEltBits, dl,
612 TLI.getShiftAmountTy(WideVT, DAG.getDataLayout()));
Michael Liao7fb39662013-02-20 18:04:21 +0000613 Lo = DAG.getNode(ISD::SHL, dl, WideVT, Lo, ShAmt);
614 Lo = DAG.getNode(ISD::SRA, dl, WideVT, Lo, ShAmt);
615 Lo = DAG.getSExtOrTrunc(Lo, dl, DstEltVT);
616 break;
617 }
618 Vals.push_back(Lo);
619 }
Matt Arsenaulta4b1b6e2016-03-30 21:15:10 +0000620
621 NewChain = DAG.getNode(ISD::TokenFactor, dl, MVT::Other, LoadChains);
622 Value = DAG.getNode(ISD::BUILD_VECTOR, dl,
623 Op.getNode()->getValueType(0), Vals);
Michael Liao7fb39662013-02-20 18:04:21 +0000624 } else {
Matt Arsenaulta4b1b6e2016-03-30 21:15:10 +0000625 SDValue Scalarized = TLI.scalarizeVectorLoad(LD, DAG);
Michael Liao7fb39662013-02-20 18:04:21 +0000626
Matt Arsenaulta4b1b6e2016-03-30 21:15:10 +0000627 NewChain = Scalarized.getValue(1);
628 Value = Scalarized.getValue(0);
Nadav Rotemebe13bc2011-10-15 07:41:10 +0000629 }
Nadav Rotem75c22292011-10-18 22:32:43 +0000630
Nadav Rotemebe13bc2011-10-15 07:41:10 +0000631 AddLegalizedOperand(Op.getValue(0), Value);
632 AddLegalizedOperand(Op.getValue(1), NewChain);
633
634 return (Op.getResNo() ? NewChain : Value);
635}
636
637SDValue VectorLegalizer::ExpandStore(SDValue Op) {
Nadav Rotemebe13bc2011-10-15 07:41:10 +0000638 StoreSDNode *ST = cast<StoreSDNode>(Op.getNode());
Matt Arsenault46ba3162016-03-30 21:15:18 +0000639
Nadav Rotemebe13bc2011-10-15 07:41:10 +0000640 EVT StVT = ST->getMemoryVT();
Nadav Rotemebe13bc2011-10-15 07:41:10 +0000641 EVT MemSclVT = StVT.getScalarType();
Nadav Rotemebe13bc2011-10-15 07:41:10 +0000642 unsigned ScalarSize = MemSclVT.getSizeInBits();
Nadav Rotemebe13bc2011-10-15 07:41:10 +0000643
644 // Round odd types to the next pow of two.
Matt Arsenault46ba3162016-03-30 21:15:18 +0000645 if (!isPowerOf2_32(ScalarSize)) {
646 // FIXME: This is completely broken and inconsistent with ExpandLoad
647 // handling.
Nadav Rotemebe13bc2011-10-15 07:41:10 +0000648
Matt Arsenault46ba3162016-03-30 21:15:18 +0000649 // For sub-byte element sizes, this ends up with 0 stride between elements,
650 // so the same element just gets re-written to the same location. There seem
651 // to be tests explicitly testing for this broken behavior though. tests
652 // for this broken behavior.
Nadav Rotemebe13bc2011-10-15 07:41:10 +0000653
Matt Arsenault46ba3162016-03-30 21:15:18 +0000654 LLVMContext &Ctx = *DAG.getContext();
Nadav Rotemebe13bc2011-10-15 07:41:10 +0000655
Matt Arsenault46ba3162016-03-30 21:15:18 +0000656 EVT NewMemVT
657 = EVT::getVectorVT(Ctx,
658 MemSclVT.getIntegerVT(Ctx, NextPowerOf2(ScalarSize)),
659 StVT.getVectorNumElements());
Nadav Rotem75c22292011-10-18 22:32:43 +0000660
Justin Lebar9c375812016-07-15 18:27:10 +0000661 SDValue NewVectorStore = DAG.getTruncStore(
662 ST->getChain(), SDLoc(Op), ST->getValue(), ST->getBasePtr(),
663 ST->getPointerInfo(), NewMemVT, ST->getAlignment(),
664 ST->getMemOperand()->getFlags(), ST->getAAInfo());
Matt Arsenault46ba3162016-03-30 21:15:18 +0000665 ST = cast<StoreSDNode>(NewVectorStore.getNode());
Nadav Rotemebe13bc2011-10-15 07:41:10 +0000666 }
Matt Arsenault46ba3162016-03-30 21:15:18 +0000667
668 SDValue TF = TLI.scalarizeVectorStore(ST, DAG);
Nadav Rotemebe13bc2011-10-15 07:41:10 +0000669 AddLegalizedOperand(Op, TF);
670 return TF;
671}
672
Chandler Carruthc1bedac2014-07-02 06:23:34 +0000673SDValue VectorLegalizer::Expand(SDValue Op) {
674 switch (Op->getOpcode()) {
675 case ISD::SIGN_EXTEND_INREG:
676 return ExpandSEXTINREG(Op);
Chandler Carruth0b666e02014-07-10 12:32:32 +0000677 case ISD::ANY_EXTEND_VECTOR_INREG:
678 return ExpandANY_EXTEND_VECTOR_INREG(Op);
679 case ISD::SIGN_EXTEND_VECTOR_INREG:
680 return ExpandSIGN_EXTEND_VECTOR_INREG(Op);
Chandler Carruthafe4b252014-07-09 10:58:18 +0000681 case ISD::ZERO_EXTEND_VECTOR_INREG:
682 return ExpandZERO_EXTEND_VECTOR_INREG(Op);
Chandler Carruthc1bedac2014-07-02 06:23:34 +0000683 case ISD::BSWAP:
684 return ExpandBSWAP(Op);
685 case ISD::VSELECT:
686 return ExpandVSELECT(Op);
687 case ISD::SELECT:
688 return ExpandSELECT(Op);
689 case ISD::UINT_TO_FP:
690 return ExpandUINT_TO_FLOAT(Op);
691 case ISD::FNEG:
692 return ExpandFNEG(Op);
693 case ISD::SETCC:
694 return UnrollVSETCC(Op);
Matt Arsenaultd0792852015-12-14 17:25:38 +0000695 case ISD::BITREVERSE:
696 return ExpandBITREVERSE(Op);
Simon Pilgrimd02c5522016-11-08 14:10:28 +0000697 case ISD::CTLZ:
Craig Topper4b1808d2015-12-27 21:33:47 +0000698 case ISD::CTLZ_ZERO_UNDEF:
Simon Pilgrimd02c5522016-11-08 14:10:28 +0000699 return ExpandCTLZ(Op);
Craig Topper4b1808d2015-12-27 21:33:47 +0000700 case ISD::CTTZ_ZERO_UNDEF:
Simon Pilgrimd02c5522016-11-08 14:10:28 +0000701 return ExpandCTTZ_ZERO_UNDEF(Op);
Chandler Carruthc1bedac2014-07-02 06:23:34 +0000702 default:
703 return DAG.UnrollVectorOp(Op.getNode());
704 }
705}
706
Nadav Rotemea973bd2012-08-30 19:17:29 +0000707SDValue VectorLegalizer::ExpandSELECT(SDValue Op) {
708 // Lower a select instruction where the condition is a scalar and the
709 // operands are vectors. Lower this select to VSELECT and implement it
Stephen Lincfe7f352013-07-08 00:37:03 +0000710 // using XOR AND OR. The selector bit is broadcasted.
Nadav Rotemea973bd2012-08-30 19:17:29 +0000711 EVT VT = Op.getValueType();
Benjamin Kramer351d53c2013-05-28 16:31:26 +0000712 SDLoc DL(Op);
Nadav Rotemea973bd2012-08-30 19:17:29 +0000713
714 SDValue Mask = Op.getOperand(0);
715 SDValue Op1 = Op.getOperand(1);
716 SDValue Op2 = Op.getOperand(2);
717
718 assert(VT.isVector() && !Mask.getValueType().isVector()
719 && Op1.getValueType() == Op2.getValueType() && "Invalid type");
720
721 unsigned NumElem = VT.getVectorNumElements();
722
723 // If we can't even use the basic vector operations of
724 // AND,OR,XOR, we will have to scalarize the op.
725 // Notice that the operation may be 'promoted' which means that it is
726 // 'bitcasted' to another type which is handled.
727 // Also, we need to be able to construct a splat vector using BUILD_VECTOR.
728 if (TLI.getOperationAction(ISD::AND, VT) == TargetLowering::Expand ||
729 TLI.getOperationAction(ISD::XOR, VT) == TargetLowering::Expand ||
730 TLI.getOperationAction(ISD::OR, VT) == TargetLowering::Expand ||
731 TLI.getOperationAction(ISD::BUILD_VECTOR, VT) == TargetLowering::Expand)
732 return DAG.UnrollVectorOp(Op.getNode());
733
734 // Generate a mask operand.
Matt Arsenaultd2322222013-09-10 00:41:56 +0000735 EVT MaskTy = VT.changeVectorElementTypeToInteger();
Nadav Rotemea973bd2012-08-30 19:17:29 +0000736
737 // What is the size of each element in the vector mask.
738 EVT BitTy = MaskTy.getScalarType();
739
Matt Arsenaultd2f03322013-06-14 22:04:37 +0000740 Mask = DAG.getSelect(DL, BitTy, Mask,
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +0000741 DAG.getConstant(APInt::getAllOnesValue(BitTy.getSizeInBits()), DL,
742 BitTy),
743 DAG.getConstant(0, DL, BitTy));
Nadav Rotemea973bd2012-08-30 19:17:29 +0000744
745 // Broadcast the mask so that the entire vector is all-one or all zero.
746 SmallVector<SDValue, 8> Ops(NumElem, Mask);
Craig Topper48d114b2014-04-26 18:35:24 +0000747 Mask = DAG.getNode(ISD::BUILD_VECTOR, DL, MaskTy, Ops);
Nadav Rotemea973bd2012-08-30 19:17:29 +0000748
749 // Bitcast the operands to be the same type as the mask.
750 // This is needed when we select between FP types because
751 // the mask is a vector of integers.
752 Op1 = DAG.getNode(ISD::BITCAST, DL, MaskTy, Op1);
753 Op2 = DAG.getNode(ISD::BITCAST, DL, MaskTy, Op2);
754
755 SDValue AllOnes = DAG.getConstant(
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +0000756 APInt::getAllOnesValue(BitTy.getSizeInBits()), DL, MaskTy);
Nadav Rotemea973bd2012-08-30 19:17:29 +0000757 SDValue NotMask = DAG.getNode(ISD::XOR, DL, MaskTy, Mask, AllOnes);
758
759 Op1 = DAG.getNode(ISD::AND, DL, MaskTy, Op1, Mask);
760 Op2 = DAG.getNode(ISD::AND, DL, MaskTy, Op2, NotMask);
761 SDValue Val = DAG.getNode(ISD::OR, DL, MaskTy, Op1, Op2);
762 return DAG.getNode(ISD::BITCAST, DL, Op.getValueType(), Val);
763}
764
Nadav Rotemdbe5c722013-01-11 22:57:48 +0000765SDValue VectorLegalizer::ExpandSEXTINREG(SDValue Op) {
766 EVT VT = Op.getValueType();
767
Benjamin Kramer5ea03492013-01-12 19:06:44 +0000768 // Make sure that the SRA and SHL instructions are available.
Nadav Rotemdbe5c722013-01-11 22:57:48 +0000769 if (TLI.getOperationAction(ISD::SRA, VT) == TargetLowering::Expand ||
Benjamin Kramer5ea03492013-01-12 19:06:44 +0000770 TLI.getOperationAction(ISD::SHL, VT) == TargetLowering::Expand)
Nadav Rotemdbe5c722013-01-11 22:57:48 +0000771 return DAG.UnrollVectorOp(Op.getNode());
772
Benjamin Kramer351d53c2013-05-28 16:31:26 +0000773 SDLoc DL(Op);
Nadav Rotemdbe5c722013-01-11 22:57:48 +0000774 EVT OrigTy = cast<VTSDNode>(Op->getOperand(1))->getVT();
775
Sanjay Patelbd6fca12016-09-14 15:21:00 +0000776 unsigned BW = VT.getScalarSizeInBits();
777 unsigned OrigBW = OrigTy.getScalarSizeInBits();
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +0000778 SDValue ShiftSz = DAG.getConstant(BW - OrigBW, DL, VT);
Nadav Rotemdbe5c722013-01-11 22:57:48 +0000779
780 Op = Op.getOperand(0);
Benjamin Kramer5ea03492013-01-12 19:06:44 +0000781 Op = DAG.getNode(ISD::SHL, DL, VT, Op, ShiftSz);
Nadav Rotemdbe5c722013-01-11 22:57:48 +0000782 return DAG.getNode(ISD::SRA, DL, VT, Op, ShiftSz);
783}
784
Chandler Carruth0b666e02014-07-10 12:32:32 +0000785// Generically expand a vector anyext in register to a shuffle of the relevant
786// lanes into the appropriate locations, with other lanes left undef.
787SDValue VectorLegalizer::ExpandANY_EXTEND_VECTOR_INREG(SDValue Op) {
788 SDLoc DL(Op);
789 EVT VT = Op.getValueType();
790 int NumElements = VT.getVectorNumElements();
791 SDValue Src = Op.getOperand(0);
792 EVT SrcVT = Src.getValueType();
793 int NumSrcElements = SrcVT.getVectorNumElements();
794
795 // Build a base mask of undef shuffles.
796 SmallVector<int, 16> ShuffleMask;
797 ShuffleMask.resize(NumSrcElements, -1);
798
799 // Place the extended lanes into the correct locations.
800 int ExtLaneScale = NumSrcElements / NumElements;
Mehdi Amini8ac7a9d2015-07-07 19:07:19 +0000801 int EndianOffset = DAG.getDataLayout().isBigEndian() ? ExtLaneScale - 1 : 0;
Chandler Carruth0b666e02014-07-10 12:32:32 +0000802 for (int i = 0; i < NumElements; ++i)
803 ShuffleMask[i * ExtLaneScale + EndianOffset] = i;
804
805 return DAG.getNode(
806 ISD::BITCAST, DL, VT,
807 DAG.getVectorShuffle(SrcVT, DL, Src, DAG.getUNDEF(SrcVT), ShuffleMask));
808}
809
810SDValue VectorLegalizer::ExpandSIGN_EXTEND_VECTOR_INREG(SDValue Op) {
811 SDLoc DL(Op);
812 EVT VT = Op.getValueType();
813 SDValue Src = Op.getOperand(0);
814 EVT SrcVT = Src.getValueType();
815
816 // First build an any-extend node which can be legalized above when we
817 // recurse through it.
818 Op = DAG.getAnyExtendVectorInReg(Src, DL, VT);
819
820 // Now we need sign extend. Do this by shifting the elements. Even if these
821 // aren't legal operations, they have a better chance of being legalized
822 // without full scalarization than the sign extension does.
Sanjay Patel1ed771f2016-09-14 16:37:15 +0000823 unsigned EltWidth = VT.getScalarSizeInBits();
824 unsigned SrcEltWidth = SrcVT.getScalarSizeInBits();
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +0000825 SDValue ShiftAmount = DAG.getConstant(EltWidth - SrcEltWidth, DL, VT);
Chandler Carruth0b666e02014-07-10 12:32:32 +0000826 return DAG.getNode(ISD::SRA, DL, VT,
827 DAG.getNode(ISD::SHL, DL, VT, Op, ShiftAmount),
828 ShiftAmount);
829}
830
Chandler Carruthafe4b252014-07-09 10:58:18 +0000831// Generically expand a vector zext in register to a shuffle of the relevant
832// lanes into the appropriate locations, a blend of zero into the high bits,
833// and a bitcast to the wider element type.
834SDValue VectorLegalizer::ExpandZERO_EXTEND_VECTOR_INREG(SDValue Op) {
835 SDLoc DL(Op);
836 EVT VT = Op.getValueType();
837 int NumElements = VT.getVectorNumElements();
838 SDValue Src = Op.getOperand(0);
839 EVT SrcVT = Src.getValueType();
840 int NumSrcElements = SrcVT.getVectorNumElements();
841
842 // Build up a zero vector to blend into this one.
Simon Pilgrim61eb49e2016-03-10 20:40:26 +0000843 SDValue Zero = DAG.getConstant(0, DL, SrcVT);
Chandler Carruthafe4b252014-07-09 10:58:18 +0000844
845 // Shuffle the incoming lanes into the correct position, and pull all other
846 // lanes from the zero vector.
847 SmallVector<int, 16> ShuffleMask;
848 ShuffleMask.reserve(NumSrcElements);
849 for (int i = 0; i < NumSrcElements; ++i)
850 ShuffleMask.push_back(i);
851
852 int ExtLaneScale = NumSrcElements / NumElements;
Mehdi Amini8ac7a9d2015-07-07 19:07:19 +0000853 int EndianOffset = DAG.getDataLayout().isBigEndian() ? ExtLaneScale - 1 : 0;
Chandler Carruthafe4b252014-07-09 10:58:18 +0000854 for (int i = 0; i < NumElements; ++i)
855 ShuffleMask[i * ExtLaneScale + EndianOffset] = NumSrcElements + i;
856
857 return DAG.getNode(ISD::BITCAST, DL, VT,
858 DAG.getVectorShuffle(SrcVT, DL, Zero, Src, ShuffleMask));
859}
860
Simon Pilgrim89b896502016-05-12 13:09:49 +0000861static void createBSWAPShuffleMask(EVT VT, SmallVectorImpl<int> &ShuffleMask) {
862 int ScalarSizeInBytes = VT.getScalarSizeInBits() / 8;
863 for (int I = 0, E = VT.getVectorNumElements(); I != E; ++I)
864 for (int J = ScalarSizeInBytes - 1; J >= 0; --J)
865 ShuffleMask.push_back((I * ScalarSizeInBytes) + J);
866}
867
Benjamin Kramerf3ad2352014-05-19 13:12:38 +0000868SDValue VectorLegalizer::ExpandBSWAP(SDValue Op) {
869 EVT VT = Op.getValueType();
870
871 // Generate a byte wise shuffle mask for the BSWAP.
872 SmallVector<int, 16> ShuffleMask;
Simon Pilgrim89b896502016-05-12 13:09:49 +0000873 createBSWAPShuffleMask(VT, ShuffleMask);
Benjamin Kramerf3ad2352014-05-19 13:12:38 +0000874 EVT ByteVT = EVT::getVectorVT(*DAG.getContext(), MVT::i8, ShuffleMask.size());
875
876 // Only emit a shuffle if the mask is legal.
877 if (!TLI.isShuffleMaskLegal(ShuffleMask, ByteVT))
878 return DAG.UnrollVectorOp(Op.getNode());
879
880 SDLoc DL(Op);
881 Op = DAG.getNode(ISD::BITCAST, DL, ByteVT, Op.getOperand(0));
Craig Topper2bd8b4b2016-07-01 06:54:47 +0000882 Op = DAG.getVectorShuffle(ByteVT, DL, Op, DAG.getUNDEF(ByteVT), ShuffleMask);
Benjamin Kramerf3ad2352014-05-19 13:12:38 +0000883 return DAG.getNode(ISD::BITCAST, DL, VT, Op);
884}
885
Matt Arsenaultd0792852015-12-14 17:25:38 +0000886SDValue VectorLegalizer::ExpandBITREVERSE(SDValue Op) {
887 EVT VT = Op.getValueType();
888
889 // If we have the scalar operation, it's probably cheaper to unroll it.
890 if (TLI.isOperationLegalOrCustom(ISD::BITREVERSE, VT.getScalarType()))
891 return DAG.UnrollVectorOp(Op.getNode());
892
Simon Pilgrim89b896502016-05-12 13:09:49 +0000893 // If the vector element width is a whole number of bytes, test if its legal
894 // to BSWAP shuffle the bytes and then perform the BITREVERSE on the byte
895 // vector. This greatly reduces the number of bit shifts necessary.
896 unsigned ScalarSizeInBits = VT.getScalarSizeInBits();
897 if (ScalarSizeInBits > 8 && (ScalarSizeInBits % 8) == 0) {
898 SmallVector<int, 16> BSWAPMask;
899 createBSWAPShuffleMask(VT, BSWAPMask);
900
901 EVT ByteVT = EVT::getVectorVT(*DAG.getContext(), MVT::i8, BSWAPMask.size());
902 if (TLI.isShuffleMaskLegal(BSWAPMask, ByteVT) &&
903 (TLI.isOperationLegalOrCustom(ISD::BITREVERSE, ByteVT) ||
904 (TLI.isOperationLegalOrCustom(ISD::SHL, ByteVT) &&
905 TLI.isOperationLegalOrCustom(ISD::SRL, ByteVT) &&
906 TLI.isOperationLegalOrCustomOrPromote(ISD::AND, ByteVT) &&
907 TLI.isOperationLegalOrCustomOrPromote(ISD::OR, ByteVT)))) {
908 SDLoc DL(Op);
909 Op = DAG.getNode(ISD::BITCAST, DL, ByteVT, Op.getOperand(0));
910 Op = DAG.getVectorShuffle(ByteVT, DL, Op, DAG.getUNDEF(ByteVT),
Craig Topper2bd8b4b2016-07-01 06:54:47 +0000911 BSWAPMask);
Simon Pilgrim89b896502016-05-12 13:09:49 +0000912 Op = DAG.getNode(ISD::BITREVERSE, DL, ByteVT, Op);
913 return DAG.getNode(ISD::BITCAST, DL, VT, Op);
914 }
915 }
916
Matt Arsenaultd0792852015-12-14 17:25:38 +0000917 // If we have the appropriate vector bit operations, it is better to use them
918 // than unrolling and expanding each component.
919 if (!TLI.isOperationLegalOrCustom(ISD::SHL, VT) ||
920 !TLI.isOperationLegalOrCustom(ISD::SRL, VT) ||
Simon Pilgrim1f5ad702016-05-04 22:08:51 +0000921 !TLI.isOperationLegalOrCustomOrPromote(ISD::AND, VT) ||
922 !TLI.isOperationLegalOrCustomOrPromote(ISD::OR, VT))
Matt Arsenaultd0792852015-12-14 17:25:38 +0000923 return DAG.UnrollVectorOp(Op.getNode());
924
925 // Let LegalizeDAG handle this later.
926 return Op;
927}
928
Nadav Rotem52202fb2011-09-13 19:17:42 +0000929SDValue VectorLegalizer::ExpandVSELECT(SDValue Op) {
930 // Implement VSELECT in terms of XOR, AND, OR
931 // on platforms which do not support blend natively.
Benjamin Kramer351d53c2013-05-28 16:31:26 +0000932 SDLoc DL(Op);
Nadav Rotem52202fb2011-09-13 19:17:42 +0000933
934 SDValue Mask = Op.getOperand(0);
935 SDValue Op1 = Op.getOperand(1);
936 SDValue Op2 = Op.getOperand(2);
937
Matt Arsenaulta5733dc2013-05-07 20:24:18 +0000938 EVT VT = Mask.getValueType();
939
Nadav Rotem52202fb2011-09-13 19:17:42 +0000940 // If we can't even use the basic vector operations of
941 // AND,OR,XOR, we will have to scalarize the op.
Nadav Rotem88244722011-10-19 20:43:16 +0000942 // Notice that the operation may be 'promoted' which means that it is
943 // 'bitcasted' to another type which is handled.
Pete Cooper2455e9c2012-09-01 22:27:48 +0000944 // This operation also isn't safe with AND, OR, XOR when the boolean
945 // type is 0/1 as we need an all ones vector constant to mask with.
946 // FIXME: Sign extend 1 to all ones if thats legal on the target.
Nadav Rotem88244722011-10-19 20:43:16 +0000947 if (TLI.getOperationAction(ISD::AND, VT) == TargetLowering::Expand ||
948 TLI.getOperationAction(ISD::XOR, VT) == TargetLowering::Expand ||
Daniel Sanderscbd44c52014-07-10 10:18:12 +0000949 TLI.getOperationAction(ISD::OR, VT) == TargetLowering::Expand ||
950 TLI.getBooleanContents(Op1.getValueType()) !=
951 TargetLowering::ZeroOrNegativeOneBooleanContent)
Nadav Rotem88244722011-10-19 20:43:16 +0000952 return DAG.UnrollVectorOp(Op.getNode());
Nadav Rotem52202fb2011-09-13 19:17:42 +0000953
Matt Arsenaulta5733dc2013-05-07 20:24:18 +0000954 // If the mask and the type are different sizes, unroll the vector op. This
955 // can occur when getSetCCResultType returns something that is different in
956 // size from the operand types. For example, v4i8 = select v4i32, v4i8, v4i8.
Sanjay Patelb1f0a0f2016-09-14 16:05:51 +0000957 if (VT.getSizeInBits() != Op1.getValueSizeInBits())
Matt Arsenaulta5733dc2013-05-07 20:24:18 +0000958 return DAG.UnrollVectorOp(Op.getNode());
959
Nadav Rotem52202fb2011-09-13 19:17:42 +0000960 // Bitcast the operands to be the same type as the mask.
961 // This is needed when we select between FP types because
962 // the mask is a vector of integers.
963 Op1 = DAG.getNode(ISD::BITCAST, DL, VT, Op1);
964 Op2 = DAG.getNode(ISD::BITCAST, DL, VT, Op2);
965
966 SDValue AllOnes = DAG.getConstant(
Sanjay Patelbd6fca12016-09-14 15:21:00 +0000967 APInt::getAllOnesValue(VT.getScalarSizeInBits()), DL, VT);
Nadav Rotem52202fb2011-09-13 19:17:42 +0000968 SDValue NotMask = DAG.getNode(ISD::XOR, DL, VT, Mask, AllOnes);
969
970 Op1 = DAG.getNode(ISD::AND, DL, VT, Op1, Mask);
971 Op2 = DAG.getNode(ISD::AND, DL, VT, Op2, NotMask);
Nadav Rotem02ef0c32012-04-15 15:08:09 +0000972 SDValue Val = DAG.getNode(ISD::OR, DL, VT, Op1, Op2);
973 return DAG.getNode(ISD::BITCAST, DL, Op.getValueType(), Val);
Nadav Rotem52202fb2011-09-13 19:17:42 +0000974}
975
Nadav Roteme7a101c2011-03-19 13:09:10 +0000976SDValue VectorLegalizer::ExpandUINT_TO_FLOAT(SDValue Op) {
Nadav Roteme7a101c2011-03-19 13:09:10 +0000977 EVT VT = Op.getOperand(0).getValueType();
Benjamin Kramer351d53c2013-05-28 16:31:26 +0000978 SDLoc DL(Op);
Nadav Roteme7a101c2011-03-19 13:09:10 +0000979
980 // Make sure that the SINT_TO_FP and SRL instructions are available.
Nadav Rotem88244722011-10-19 20:43:16 +0000981 if (TLI.getOperationAction(ISD::SINT_TO_FP, VT) == TargetLowering::Expand ||
982 TLI.getOperationAction(ISD::SRL, VT) == TargetLowering::Expand)
983 return DAG.UnrollVectorOp(Op.getNode());
Nadav Roteme7a101c2011-03-19 13:09:10 +0000984
985 EVT SVT = VT.getScalarType();
986 assert((SVT.getSizeInBits() == 64 || SVT.getSizeInBits() == 32) &&
987 "Elements in vector-UINT_TO_FP must be 32 or 64 bits wide");
988
989 unsigned BW = SVT.getSizeInBits();
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +0000990 SDValue HalfWord = DAG.getConstant(BW/2, DL, VT);
Nadav Roteme7a101c2011-03-19 13:09:10 +0000991
992 // Constants to clear the upper part of the word.
993 // Notice that we can also use SHL+SHR, but using a constant is slightly
994 // faster on x86.
995 uint64_t HWMask = (SVT.getSizeInBits()==64)?0x00000000FFFFFFFF:0x0000FFFF;
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +0000996 SDValue HalfWordMask = DAG.getConstant(HWMask, DL, VT);
Nadav Roteme7a101c2011-03-19 13:09:10 +0000997
998 // Two to the power of half-word-size.
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +0000999 SDValue TWOHW = DAG.getConstantFP(1 << (BW/2), DL, Op.getValueType());
Nadav Roteme7a101c2011-03-19 13:09:10 +00001000
1001 // Clear upper part of LO, lower HI
1002 SDValue HI = DAG.getNode(ISD::SRL, DL, VT, Op.getOperand(0), HalfWord);
1003 SDValue LO = DAG.getNode(ISD::AND, DL, VT, Op.getOperand(0), HalfWordMask);
1004
1005 // Convert hi and lo to floats
1006 // Convert the hi part back to the upper values
Sanjay Patela2607012015-09-16 16:31:21 +00001007 // TODO: Can any fast-math-flags be set on these nodes?
Nadav Roteme7a101c2011-03-19 13:09:10 +00001008 SDValue fHI = DAG.getNode(ISD::SINT_TO_FP, DL, Op.getValueType(), HI);
1009 fHI = DAG.getNode(ISD::FMUL, DL, Op.getValueType(), fHI, TWOHW);
1010 SDValue fLO = DAG.getNode(ISD::SINT_TO_FP, DL, Op.getValueType(), LO);
1011
1012 // Add the two halves
1013 return DAG.getNode(ISD::FADD, DL, Op.getValueType(), fHI, fLO);
1014}
1015
1016
Eli Friedmanda90dd62009-05-23 12:35:30 +00001017SDValue VectorLegalizer::ExpandFNEG(SDValue Op) {
1018 if (TLI.isOperationLegalOrCustom(ISD::FSUB, Op.getValueType())) {
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +00001019 SDLoc DL(Op);
1020 SDValue Zero = DAG.getConstantFP(-0.0, DL, Op.getValueType());
Sanjay Patela2607012015-09-16 16:31:21 +00001021 // TODO: If FNEG had fast-math-flags, they'd get propagated to this FSUB.
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +00001022 return DAG.getNode(ISD::FSUB, DL, Op.getValueType(),
Eli Friedmanda90dd62009-05-23 12:35:30 +00001023 Zero, Op.getOperand(0));
1024 }
Mon P Wang32f8bb92009-11-30 02:42:02 +00001025 return DAG.UnrollVectorOp(Op.getNode());
Eli Friedmanda90dd62009-05-23 12:35:30 +00001026}
1027
Simon Pilgrimd02c5522016-11-08 14:10:28 +00001028SDValue VectorLegalizer::ExpandCTLZ(SDValue Op) {
1029 EVT VT = Op.getValueType();
1030 unsigned NumBitsPerElt = VT.getScalarSizeInBits();
1031
Craig Topper52cb5ec2016-04-21 04:43:57 +00001032 // If the non-ZERO_UNDEF version is supported we can use that instead.
Simon Pilgrimd02c5522016-11-08 14:10:28 +00001033 if (Op.getOpcode() == ISD::CTLZ_ZERO_UNDEF &&
1034 TLI.isOperationLegalOrCustom(ISD::CTLZ, VT)) {
Craig Topper52cb5ec2016-04-21 04:43:57 +00001035 SDLoc DL(Op);
Simon Pilgrimd02c5522016-11-08 14:10:28 +00001036 return DAG.getNode(ISD::CTLZ, DL, Op.getValueType(), Op.getOperand(0));
1037 }
1038
1039 // If CTPOP is available we can lower with a CTPOP based method:
1040 // u16 ctlz(u16 x) {
1041 // x |= (x >> 1);
1042 // x |= (x >> 2);
1043 // x |= (x >> 4);
1044 // x |= (x >> 8);
1045 // return ctpop(~x);
1046 // }
1047 // Ref: "Hacker's Delight" by Henry Warren
1048 if (isPowerOf2_32(NumBitsPerElt) &&
1049 TLI.isOperationLegalOrCustom(ISD::CTPOP, VT) &&
1050 TLI.isOperationLegalOrCustom(ISD::SRL, VT) &&
1051 TLI.isOperationLegalOrCustomOrPromote(ISD::OR, VT) &&
1052 TLI.isOperationLegalOrCustomOrPromote(ISD::XOR, VT)) {
1053 SDLoc DL(Op);
1054 SDValue Res = Op.getOperand(0);
1055 EVT ShiftTy = TLI.getShiftAmountTy(VT, DAG.getDataLayout());
1056
1057 for (unsigned i = 1; i != NumBitsPerElt; i *= 2)
1058 Res = DAG.getNode(
1059 ISD::OR, DL, VT, Res,
1060 DAG.getNode(ISD::SRL, DL, VT, Res, DAG.getConstant(i, DL, ShiftTy)));
1061
1062 Res = DAG.getNOT(DL, Res, VT);
1063 return DAG.getNode(ISD::CTPOP, DL, VT, Res);
1064 }
1065
1066 // Otherwise go ahead and unroll.
1067 return DAG.UnrollVectorOp(Op.getNode());
1068}
1069
1070SDValue VectorLegalizer::ExpandCTTZ_ZERO_UNDEF(SDValue Op) {
1071 // If the non-ZERO_UNDEF version is supported we can use that instead.
1072 if (TLI.isOperationLegalOrCustom(ISD::CTTZ, Op.getValueType())) {
1073 SDLoc DL(Op);
1074 return DAG.getNode(ISD::CTTZ, DL, Op.getValueType(), Op.getOperand(0));
Craig Topper52cb5ec2016-04-21 04:43:57 +00001075 }
Craig Topper4b1808d2015-12-27 21:33:47 +00001076
1077 // Otherwise go ahead and unroll.
1078 return DAG.UnrollVectorOp(Op.getNode());
1079}
1080
Eli Friedmanda90dd62009-05-23 12:35:30 +00001081SDValue VectorLegalizer::UnrollVSETCC(SDValue Op) {
Owen Anderson53aa7a92009-08-10 22:56:29 +00001082 EVT VT = Op.getValueType();
Eli Friedmanda90dd62009-05-23 12:35:30 +00001083 unsigned NumElems = VT.getVectorNumElements();
Owen Anderson53aa7a92009-08-10 22:56:29 +00001084 EVT EltVT = VT.getVectorElementType();
Eli Friedmanda90dd62009-05-23 12:35:30 +00001085 SDValue LHS = Op.getOperand(0), RHS = Op.getOperand(1), CC = Op.getOperand(2);
Owen Anderson53aa7a92009-08-10 22:56:29 +00001086 EVT TmpEltVT = LHS.getValueType().getVectorElementType();
Benjamin Kramer351d53c2013-05-28 16:31:26 +00001087 SDLoc dl(Op);
Eli Friedmanda90dd62009-05-23 12:35:30 +00001088 SmallVector<SDValue, 8> Ops(NumElems);
1089 for (unsigned i = 0; i < NumElems; ++i) {
Mehdi Amini44ede332015-07-09 02:09:04 +00001090 SDValue LHSElem = DAG.getNode(
1091 ISD::EXTRACT_VECTOR_ELT, dl, TmpEltVT, LHS,
1092 DAG.getConstant(i, dl, TLI.getVectorIdxTy(DAG.getDataLayout())));
1093 SDValue RHSElem = DAG.getNode(
1094 ISD::EXTRACT_VECTOR_ELT, dl, TmpEltVT, RHS,
1095 DAG.getConstant(i, dl, TLI.getVectorIdxTy(DAG.getDataLayout())));
Matt Arsenault758659232013-05-18 00:21:46 +00001096 Ops[i] = DAG.getNode(ISD::SETCC, dl,
Mehdi Amini44ede332015-07-09 02:09:04 +00001097 TLI.getSetCCResultType(DAG.getDataLayout(),
1098 *DAG.getContext(), TmpEltVT),
Eli Friedmanda90dd62009-05-23 12:35:30 +00001099 LHSElem, RHSElem, CC);
Matt Arsenaultd2f03322013-06-14 22:04:37 +00001100 Ops[i] = DAG.getSelect(dl, EltVT, Ops[i],
1101 DAG.getConstant(APInt::getAllOnesValue
Sergey Dmitrouk842a51b2015-04-28 14:05:47 +00001102 (EltVT.getSizeInBits()), dl, EltVT),
1103 DAG.getConstant(0, dl, EltVT));
Eli Friedmanda90dd62009-05-23 12:35:30 +00001104 }
Craig Topper48d114b2014-04-26 18:35:24 +00001105 return DAG.getNode(ISD::BUILD_VECTOR, dl, VT, Ops);
Eli Friedmanda90dd62009-05-23 12:35:30 +00001106}
1107
Alexander Kornienkof00654e2015-06-23 09:49:53 +00001108}
Eli Friedmanda90dd62009-05-23 12:35:30 +00001109
1110bool SelectionDAG::LegalizeVectors() {
1111 return VectorLegalizer(*this).Run();
1112}