blob: 6488b9aee0ab96bfcd2d2ee03551405767365cd9 [file] [log] [blame]
Krzysztof Parzyszek7d37dd82017-12-06 16:40:37 +00001//===-- HexagonISelLoweringHVX.cpp --- Lowering HVX operations ------------===//
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#include "HexagonISelLowering.h"
11#include "HexagonRegisterInfo.h"
12#include "HexagonSubtarget.h"
13
14using namespace llvm;
15
16SDValue
17HexagonTargetLowering::getInt(unsigned IntId, MVT ResTy, ArrayRef<SDValue> Ops,
18 const SDLoc &dl, SelectionDAG &DAG) const {
19 SmallVector<SDValue,4> IntOps;
20 IntOps.push_back(DAG.getConstant(IntId, dl, MVT::i32));
21 for (const SDValue &Op : Ops)
22 IntOps.push_back(Op);
23 return DAG.getNode(ISD::INTRINSIC_WO_CHAIN, dl, ResTy, IntOps);
24}
25
26MVT
27HexagonTargetLowering::typeJoin(const TypePair &Tys) const {
28 assert(Tys.first.getVectorElementType() == Tys.second.getVectorElementType());
29
30 MVT ElemTy = Tys.first.getVectorElementType();
31 return MVT::getVectorVT(ElemTy, Tys.first.getVectorNumElements() +
32 Tys.second.getVectorNumElements());
33}
34
35HexagonTargetLowering::TypePair
36HexagonTargetLowering::typeSplit(MVT VecTy) const {
37 assert(VecTy.isVector());
38 unsigned NumElem = VecTy.getVectorNumElements();
39 assert((NumElem % 2) == 0 && "Expecting even-sized vector type");
40 MVT HalfTy = MVT::getVectorVT(VecTy.getVectorElementType(), NumElem/2);
41 return { HalfTy, HalfTy };
42}
43
44MVT
45HexagonTargetLowering::typeExtElem(MVT VecTy, unsigned Factor) const {
46 MVT ElemTy = VecTy.getVectorElementType();
47 MVT NewElemTy = MVT::getIntegerVT(ElemTy.getSizeInBits() * Factor);
48 return MVT::getVectorVT(NewElemTy, VecTy.getVectorNumElements());
49}
50
51MVT
52HexagonTargetLowering::typeTruncElem(MVT VecTy, unsigned Factor) const {
53 MVT ElemTy = VecTy.getVectorElementType();
54 MVT NewElemTy = MVT::getIntegerVT(ElemTy.getSizeInBits() / Factor);
55 return MVT::getVectorVT(NewElemTy, VecTy.getVectorNumElements());
56}
57
58SDValue
59HexagonTargetLowering::opCastElem(SDValue Vec, MVT ElemTy,
60 SelectionDAG &DAG) const {
61 if (ty(Vec).getVectorElementType() == ElemTy)
62 return Vec;
63 MVT CastTy = tyVector(Vec.getValueType().getSimpleVT(), ElemTy);
64 return DAG.getBitcast(CastTy, Vec);
65}
66
67SDValue
68HexagonTargetLowering::opJoin(const VectorPair &Ops, const SDLoc &dl,
69 SelectionDAG &DAG) const {
70 return DAG.getNode(ISD::CONCAT_VECTORS, dl, typeJoin(ty(Ops)),
71 Ops.second, Ops.first);
72}
73
74HexagonTargetLowering::VectorPair
75HexagonTargetLowering::opSplit(SDValue Vec, const SDLoc &dl,
76 SelectionDAG &DAG) const {
77 TypePair Tys = typeSplit(ty(Vec));
78 return DAG.SplitVector(Vec, dl, Tys.first, Tys.second);
79}
80
81SDValue
82HexagonTargetLowering::convertToByteIndex(SDValue ElemIdx, MVT ElemTy,
83 SelectionDAG &DAG) const {
84 if (ElemIdx.getValueType().getSimpleVT() != MVT::i32)
85 ElemIdx = DAG.getBitcast(MVT::i32, ElemIdx);
86
87 unsigned ElemWidth = ElemTy.getSizeInBits();
88 if (ElemWidth == 8)
89 return ElemIdx;
90
91 unsigned L = Log2_32(ElemWidth/8);
92 const SDLoc &dl(ElemIdx);
93 return DAG.getNode(ISD::SHL, dl, MVT::i32,
94 {ElemIdx, DAG.getConstant(L, dl, MVT::i32)});
95}
96
97SDValue
98HexagonTargetLowering::getIndexInWord32(SDValue Idx, MVT ElemTy,
99 SelectionDAG &DAG) const {
100 unsigned ElemWidth = ElemTy.getSizeInBits();
101 assert(ElemWidth >= 8 && ElemWidth <= 32);
102 if (ElemWidth == 32)
103 return Idx;
104
105 if (ty(Idx) != MVT::i32)
106 Idx = DAG.getBitcast(MVT::i32, Idx);
107 const SDLoc &dl(Idx);
108 SDValue Mask = DAG.getConstant(32/ElemWidth - 1, dl, MVT::i32);
109 SDValue SubIdx = DAG.getNode(ISD::AND, dl, MVT::i32, {Idx, Mask});
110 return SubIdx;
111}
112
113SDValue
Krzysztof Parzyszek039d4d92017-12-07 17:37:28 +0000114HexagonTargetLowering::getByteShuffle(const SDLoc &dl, SDValue Op0,
115 SDValue Op1, ArrayRef<int> Mask,
116 SelectionDAG &DAG) const {
117 MVT OpTy = ty(Op0);
118 assert(OpTy == ty(Op1));
119
120 MVT ElemTy = OpTy.getVectorElementType();
121 if (ElemTy == MVT::i8)
122 return DAG.getVectorShuffle(OpTy, dl, Op0, Op1, Mask);
123 assert(ElemTy.getSizeInBits() >= 8);
124
125 MVT ResTy = tyVector(OpTy, MVT::i8);
126 unsigned ElemSize = ElemTy.getSizeInBits() / 8;
127
128 SmallVector<int,128> ByteMask;
129 for (int M : Mask) {
130 if (M < 0) {
131 for (unsigned I = 0; I != ElemSize; ++I)
132 ByteMask.push_back(-1);
133 } else {
134 int NewM = M*ElemSize;
135 for (unsigned I = 0; I != ElemSize; ++I)
136 ByteMask.push_back(NewM+I);
137 }
138 }
139 assert(ResTy.getVectorNumElements() == ByteMask.size());
140 return DAG.getVectorShuffle(ResTy, dl, opCastElem(Op0, MVT::i8, DAG),
141 opCastElem(Op1, MVT::i8, DAG), ByteMask);
142}
143
Krzysztof Parzyszeke4ce92c2017-12-20 20:49:43 +0000144MVT
145HexagonTargetLowering::getVecBoolVT() const {
146 return MVT::getVectorVT(MVT::i1, 8*Subtarget.getVectorLength());
147}
Krzysztof Parzyszek7d37dd82017-12-06 16:40:37 +0000148
Krzysztof Parzyszeke4ce92c2017-12-20 20:49:43 +0000149SDValue
150HexagonTargetLowering::buildHvxVectorSingle(ArrayRef<SDValue> Values,
151 const SDLoc &dl, MVT VecTy,
152 SelectionDAG &DAG) const {
153 unsigned VecLen = Values.size();
154 MachineFunction &MF = DAG.getMachineFunction();
155 MVT ElemTy = VecTy.getVectorElementType();
156 unsigned ElemWidth = ElemTy.getSizeInBits();
157 unsigned HwLen = Subtarget.getVectorLength();
158
159 SmallVector<ConstantInt*, 128> Consts(VecLen);
160 bool AllConst = getBuildVectorConstInts(Values, VecTy, DAG, Consts);
161 if (AllConst) {
162 if (llvm::all_of(Consts, [](ConstantInt *CI) { return CI->isZero(); }))
163 return getZero(dl, VecTy, DAG);
164
165 ArrayRef<Constant*> Tmp((Constant**)Consts.begin(),
166 (Constant**)Consts.end());
167 Constant *CV = ConstantVector::get(Tmp);
168 unsigned Align = HwLen;
Krzysztof Parzyszek7d37dd82017-12-06 16:40:37 +0000169 SDValue CP = LowerConstantPool(DAG.getConstantPool(CV, VecTy, Align), DAG);
170 return DAG.getLoad(VecTy, dl, DAG.getEntryNode(), CP,
171 MachinePointerInfo::getConstantPool(MF), Align);
172 }
173
Krzysztof Parzyszeke4ce92c2017-12-20 20:49:43 +0000174 unsigned ElemSize = ElemWidth / 8;
175 assert(ElemSize*VecLen == HwLen);
Krzysztof Parzyszek7d37dd82017-12-06 16:40:37 +0000176 SmallVector<SDValue,32> Words;
Krzysztof Parzyszek7d37dd82017-12-06 16:40:37 +0000177
178 if (VecTy.getVectorElementType() != MVT::i32) {
Krzysztof Parzyszek7d37dd82017-12-06 16:40:37 +0000179 assert((ElemSize == 1 || ElemSize == 2) && "Invalid element size");
180 unsigned OpsPerWord = (ElemSize == 1) ? 4 : 2;
181 MVT PartVT = MVT::getVectorVT(VecTy.getVectorElementType(), OpsPerWord);
Krzysztof Parzyszeke4ce92c2017-12-20 20:49:43 +0000182 for (unsigned i = 0; i != VecLen; i += OpsPerWord) {
183 SDValue W = buildVector32(Values.slice(i, OpsPerWord), dl, PartVT, DAG);
Krzysztof Parzyszek7d37dd82017-12-06 16:40:37 +0000184 Words.push_back(DAG.getBitcast(MVT::i32, W));
185 }
186 } else {
Krzysztof Parzyszeke4ce92c2017-12-20 20:49:43 +0000187 Words.assign(Values.begin(), Values.end());
Krzysztof Parzyszek7d37dd82017-12-06 16:40:37 +0000188 }
189
190 // Construct two halves in parallel, then or them together.
191 assert(4*Words.size() == Subtarget.getVectorLength());
192 SDValue HalfV0 = getNode(Hexagon::V6_vd0, dl, VecTy, {}, DAG);
193 SDValue HalfV1 = getNode(Hexagon::V6_vd0, dl, VecTy, {}, DAG);
194 SDValue S = DAG.getConstant(4, dl, MVT::i32);
195 unsigned NumWords = Words.size();
196 for (unsigned i = 0; i != NumWords/2; ++i) {
197 SDValue N = DAG.getNode(HexagonISD::VINSERTW0, dl, VecTy,
198 {HalfV0, Words[i]});
199 SDValue M = DAG.getNode(HexagonISD::VINSERTW0, dl, VecTy,
200 {HalfV1, Words[i+NumWords/2]});
201 HalfV0 = DAG.getNode(HexagonISD::VROR, dl, VecTy, {N, S});
202 HalfV1 = DAG.getNode(HexagonISD::VROR, dl, VecTy, {M, S});
203 }
204
205 HalfV0 = DAG.getNode(HexagonISD::VROR, dl, VecTy,
206 {HalfV0, DAG.getConstant(HwLen/2, dl, MVT::i32)});
207 SDValue DstV = DAG.getNode(ISD::OR, dl, VecTy, {HalfV0, HalfV1});
208 return DstV;
209}
210
211SDValue
Krzysztof Parzyszeke4ce92c2017-12-20 20:49:43 +0000212HexagonTargetLowering::buildHvxVectorPred(ArrayRef<SDValue> Values,
213 const SDLoc &dl, MVT VecTy,
214 SelectionDAG &DAG) const {
215 // Construct a vector V of bytes, such that a comparison V >u 0 would
216 // produce the required vector predicate.
217 unsigned VecLen = Values.size();
218 unsigned HwLen = Subtarget.getVectorLength();
219 assert(VecLen <= HwLen || VecLen == 8*HwLen);
220 SmallVector<SDValue,128> Bytes;
221
222 if (VecLen <= HwLen) {
223 // In the hardware, each bit of a vector predicate corresponds to a byte
224 // of a vector register. Calculate how many bytes does a bit of VecTy
225 // correspond to.
226 assert(HwLen % VecLen == 0);
227 unsigned BitBytes = HwLen / VecLen;
228 for (SDValue V : Values) {
229 SDValue Ext = !V.isUndef() ? DAG.getZExtOrTrunc(V, dl, MVT::i8)
230 : DAG.getConstant(0, dl, MVT::i8);
231 for (unsigned B = 0; B != BitBytes; ++B)
232 Bytes.push_back(Ext);
233 }
234 } else {
235 // There are as many i1 values, as there are bits in a vector register.
236 // Divide the values into groups of 8 and check that each group consists
237 // of the same value (ignoring undefs).
238 for (unsigned I = 0; I != VecLen; I += 8) {
239 unsigned B = 0;
240 // Find the first non-undef value in this group.
241 for (; B != 8; ++B) {
242 if (!Values[I+B].isUndef())
243 break;
244 }
245 SDValue F = Values[I+B];
246 SDValue Ext = (B < 8) ? DAG.getZExtOrTrunc(F, dl, MVT::i8)
247 : DAG.getConstant(0, dl, MVT::i8);
248 Bytes.push_back(Ext);
249 // Verify that the rest of values in the group are the same as the
250 // first.
251 for (; B != 8; ++B)
252 assert(Values[I+B].isUndef() || Values[I+B] == F);
253 }
254 }
255
256 MVT ByteTy = MVT::getVectorVT(MVT::i8, HwLen);
257 SDValue ByteVec = buildHvxVectorSingle(Bytes, dl, ByteTy, DAG);
258 SDValue Cmp = DAG.getSetCC(dl, VecTy, ByteVec, getZero(dl, ByteTy, DAG),
259 ISD::SETUGT);
260 return Cmp;
261}
262
263SDValue
264HexagonTargetLowering::LowerHvxBuildVector(SDValue Op, SelectionDAG &DAG)
265 const {
266 const SDLoc &dl(Op);
267 MVT VecTy = ty(Op);
268
269 unsigned Size = Op.getNumOperands();
270 SmallVector<SDValue,128> Ops;
271 for (unsigned i = 0; i != Size; ++i)
272 Ops.push_back(Op.getOperand(i));
273
274 if (VecTy.getVectorElementType() == MVT::i1)
275 return buildHvxVectorPred(Ops, dl, VecTy, DAG);
276
277 if (VecTy.getSizeInBits() == 16*Subtarget.getVectorLength()) {
278 ArrayRef<SDValue> A(Ops);
279 MVT SingleTy = typeSplit(VecTy).first;
280 SDValue V0 = buildHvxVectorSingle(A.take_front(Size/2), dl, SingleTy, DAG);
281 SDValue V1 = buildHvxVectorSingle(A.drop_front(Size/2), dl, SingleTy, DAG);
282 return DAG.getNode(ISD::CONCAT_VECTORS, dl, VecTy, V0, V1);
283 }
284
285 return buildHvxVectorSingle(Ops, dl, VecTy, DAG);
286}
287
288SDValue
Krzysztof Parzyszek7d37dd82017-12-06 16:40:37 +0000289HexagonTargetLowering::LowerHvxExtractElement(SDValue Op, SelectionDAG &DAG)
290 const {
291 // Change the type of the extracted element to i32.
292 SDValue VecV = Op.getOperand(0);
293 MVT ElemTy = ty(VecV).getVectorElementType();
294 unsigned ElemWidth = ElemTy.getSizeInBits();
Krzysztof Parzyszek708c9f52017-12-14 18:35:24 +0000295 assert(ElemWidth >= 8 && ElemWidth <= 32);
Tim Shen7654ed032017-12-06 19:22:19 +0000296 (void)ElemWidth;
Krzysztof Parzyszek7d37dd82017-12-06 16:40:37 +0000297
298 const SDLoc &dl(Op);
299 SDValue IdxV = Op.getOperand(1);
300 if (ty(IdxV) != MVT::i32)
301 IdxV = DAG.getBitcast(MVT::i32, IdxV);
302
303 SDValue ByteIdx = convertToByteIndex(IdxV, ElemTy, DAG);
304 SDValue ExWord = DAG.getNode(HexagonISD::VEXTRACTW, dl, MVT::i32,
305 {VecV, ByteIdx});
306 if (ElemTy == MVT::i32)
307 return ExWord;
308
309 // Have an extracted word, need to extract the smaller element out of it.
310 // 1. Extract the bits of (the original) IdxV that correspond to the index
311 // of the desired element in the 32-bit word.
312 SDValue SubIdx = getIndexInWord32(IdxV, ElemTy, DAG);
313 // 2. Extract the element from the word.
314 SDValue ExVec = DAG.getBitcast(tyVector(ty(ExWord), ElemTy), ExWord);
315 return extractVector(ExVec, SubIdx, dl, ElemTy, MVT::i32, DAG);
316}
317
318SDValue
319HexagonTargetLowering::LowerHvxInsertElement(SDValue Op, SelectionDAG &DAG)
320 const {
321 const SDLoc &dl(Op);
322 SDValue VecV = Op.getOperand(0);
323 SDValue ValV = Op.getOperand(1);
324 SDValue IdxV = Op.getOperand(2);
325 MVT ElemTy = ty(VecV).getVectorElementType();
326 unsigned ElemWidth = ElemTy.getSizeInBits();
Krzysztof Parzyszek708c9f52017-12-14 18:35:24 +0000327 assert(ElemWidth >= 8 && ElemWidth <= 32);
Tim Shen7654ed032017-12-06 19:22:19 +0000328 (void)ElemWidth;
Krzysztof Parzyszek7d37dd82017-12-06 16:40:37 +0000329
330 auto InsertWord = [&DAG,&dl,this] (SDValue VecV, SDValue ValV,
331 SDValue ByteIdxV) {
332 MVT VecTy = ty(VecV);
333 unsigned HwLen = Subtarget.getVectorLength();
334 SDValue MaskV = DAG.getNode(ISD::AND, dl, MVT::i32,
335 {ByteIdxV, DAG.getConstant(-4, dl, MVT::i32)});
336 SDValue RotV = DAG.getNode(HexagonISD::VROR, dl, VecTy, {VecV, MaskV});
337 SDValue InsV = DAG.getNode(HexagonISD::VINSERTW0, dl, VecTy, {RotV, ValV});
338 SDValue SubV = DAG.getNode(ISD::SUB, dl, MVT::i32,
339 {DAG.getConstant(HwLen/4, dl, MVT::i32), MaskV});
340 SDValue TorV = DAG.getNode(HexagonISD::VROR, dl, VecTy, {InsV, SubV});
341 return TorV;
342 };
343
344 SDValue ByteIdx = convertToByteIndex(IdxV, ElemTy, DAG);
345 if (ElemTy == MVT::i32)
346 return InsertWord(VecV, ValV, ByteIdx);
347
348 // If this is not inserting a 32-bit word, convert it into such a thing.
349 // 1. Extract the existing word from the target vector.
350 SDValue WordIdx = DAG.getNode(ISD::SRL, dl, MVT::i32,
351 {ByteIdx, DAG.getConstant(2, dl, MVT::i32)});
352 SDValue Ex0 = DAG.getNode(ISD::EXTRACT_VECTOR_ELT, dl, MVT::i32,
353 {opCastElem(VecV, MVT::i32, DAG), WordIdx});
354 SDValue Ext = LowerHvxExtractElement(Ex0, DAG);
355
356 // 2. Treating the extracted word as a 32-bit vector, insert the given
357 // value into it.
358 SDValue SubIdx = getIndexInWord32(IdxV, ElemTy, DAG);
359 MVT SubVecTy = tyVector(ty(Ext), ElemTy);
360 SDValue Ins = insertVector(DAG.getBitcast(SubVecTy, Ext),
Krzysztof Parzyszek15241452017-12-11 14:46:06 +0000361 ValV, SubIdx, dl, ElemTy, DAG);
Krzysztof Parzyszek7d37dd82017-12-06 16:40:37 +0000362
363 // 3. Insert the 32-bit word back into the original vector.
364 return InsertWord(VecV, Ins, ByteIdx);
365}
366
367SDValue
368HexagonTargetLowering::LowerHvxExtractSubvector(SDValue Op, SelectionDAG &DAG)
369 const {
370 SDValue SrcV = Op.getOperand(0);
371 MVT SrcTy = ty(SrcV);
372 unsigned SrcElems = SrcTy.getVectorNumElements();
373 SDValue IdxV = Op.getOperand(1);
374 unsigned Idx = cast<ConstantSDNode>(IdxV.getNode())->getZExtValue();
375 MVT DstTy = ty(Op);
376 assert(Idx == 0 || DstTy.getVectorNumElements() % Idx == 0);
377 const SDLoc &dl(Op);
378 if (Idx == 0)
379 return DAG.getTargetExtractSubreg(Hexagon::vsub_lo, dl, DstTy, SrcV);
380 if (Idx == SrcElems/2)
381 return DAG.getTargetExtractSubreg(Hexagon::vsub_hi, dl, DstTy, SrcV);
382 return SDValue();
383}
384
385SDValue
386HexagonTargetLowering::LowerHvxInsertSubvector(SDValue Op, SelectionDAG &DAG)
387 const {
Krzysztof Parzyszek039d4d92017-12-07 17:37:28 +0000388 // Idx may be variable.
Krzysztof Parzyszek7d37dd82017-12-06 16:40:37 +0000389 SDValue IdxV = Op.getOperand(2);
390 auto *IdxN = dyn_cast<ConstantSDNode>(IdxV.getNode());
391 if (!IdxN)
392 return SDValue();
393 unsigned Idx = IdxN->getZExtValue();
394
395 SDValue DstV = Op.getOperand(0);
396 SDValue SrcV = Op.getOperand(1);
397 MVT DstTy = ty(DstV);
398 MVT SrcTy = ty(SrcV);
399 unsigned DstElems = DstTy.getVectorNumElements();
400 unsigned SrcElems = SrcTy.getVectorNumElements();
401 if (2*SrcElems != DstElems)
402 return SDValue();
403
404 const SDLoc &dl(Op);
405 if (Idx == 0)
406 return DAG.getTargetInsertSubreg(Hexagon::vsub_lo, dl, DstTy, DstV, SrcV);
407 if (Idx == SrcElems)
408 return DAG.getTargetInsertSubreg(Hexagon::vsub_hi, dl, DstTy, DstV, SrcV);
409 return SDValue();
410}
Krzysztof Parzyszek039d4d92017-12-07 17:37:28 +0000411
412SDValue
413HexagonTargetLowering::LowerHvxMul(SDValue Op, SelectionDAG &DAG) const {
414 MVT ResTy = ty(Op);
415 if (!ResTy.isVector())
416 return SDValue();
417 const SDLoc &dl(Op);
418 SmallVector<int,256> ShuffMask;
419
420 MVT ElemTy = ResTy.getVectorElementType();
421 unsigned VecLen = ResTy.getVectorNumElements();
422 SDValue Vs = Op.getOperand(0);
423 SDValue Vt = Op.getOperand(1);
424
425 switch (ElemTy.SimpleTy) {
426 case MVT::i8:
427 case MVT::i16: {
428 // For i8 vectors Vs = (a0, a1, ...), Vt = (b0, b1, ...),
429 // V6_vmpybv Vs, Vt produces a pair of i16 vectors Hi:Lo,
430 // where Lo = (a0*b0, a2*b2, ...), Hi = (a1*b1, a3*b3, ...).
431 // For i16, use V6_vmpyhv, which behaves in an analogous way to
432 // V6_vmpybv: results Lo and Hi are products of even/odd elements
433 // respectively.
434 MVT ExtTy = typeExtElem(ResTy, 2);
435 unsigned MpyOpc = ElemTy == MVT::i8 ? Hexagon::V6_vmpybv
436 : Hexagon::V6_vmpyhv;
437 SDValue M = getNode(MpyOpc, dl, ExtTy, {Vs, Vt}, DAG);
438
439 // Discard high halves of the resulting values, collect the low halves.
440 for (unsigned I = 0; I < VecLen; I += 2) {
441 ShuffMask.push_back(I); // Pick even element.
442 ShuffMask.push_back(I+VecLen); // Pick odd element.
443 }
444 VectorPair P = opSplit(opCastElem(M, ElemTy, DAG), dl, DAG);
Krzysztof Parzyszek0f5d9762018-01-05 20:45:34 +0000445 SDValue BS = getByteShuffle(dl, P.first, P.second, ShuffMask, DAG);
446 return DAG.getBitcast(ResTy, BS);
Krzysztof Parzyszek039d4d92017-12-07 17:37:28 +0000447 }
448 case MVT::i32: {
449 // Use the following sequence for signed word multiply:
450 // T0 = V6_vmpyiowh Vs, Vt
451 // T1 = V6_vaslw T0, 16
452 // T2 = V6_vmpyiewuh_acc T1, Vs, Vt
453 SDValue S16 = DAG.getConstant(16, dl, MVT::i32);
454 SDValue T0 = getNode(Hexagon::V6_vmpyiowh, dl, ResTy, {Vs, Vt}, DAG);
455 SDValue T1 = getNode(Hexagon::V6_vaslw, dl, ResTy, {T0, S16}, DAG);
456 SDValue T2 = getNode(Hexagon::V6_vmpyiewuh_acc, dl, ResTy,
457 {T1, Vs, Vt}, DAG);
458 return T2;
459 }
460 default:
461 break;
462 }
463 return SDValue();
464}
Krzysztof Parzyszek47076052017-12-14 21:28:48 +0000465
466SDValue
467HexagonTargetLowering::LowerHvxSetCC(SDValue Op, SelectionDAG &DAG) const {
468 MVT VecTy = ty(Op.getOperand(0));
469 assert(VecTy == ty(Op.getOperand(1)));
470
471 SDValue Cmp = Op.getOperand(2);
472 ISD::CondCode CC = cast<CondCodeSDNode>(Cmp)->get();
473 bool Negate = false, Swap = false;
474
475 // HVX has instructions for SETEQ, SETGT, SETUGT. The other comparisons
476 // can be arranged as operand-swapped/negated versions of these. Since
477 // the generated code will have the original CC expressed as
478 // (negate (swap-op NewCmp)),
479 // the condition code for the NewCmp should be calculated from the original
480 // CC by applying these operations in the reverse order.
Krzysztof Parzyszeke4ce92c2017-12-20 20:49:43 +0000481 //
482 // This could also be done through setCondCodeAction, but for negation it
483 // uses a xor with a vector of -1s, which it obtains from BUILD_VECTOR.
484 // That is far too expensive for what can be done with a single instruction.
Krzysztof Parzyszek47076052017-12-14 21:28:48 +0000485
486 switch (CC) {
487 case ISD::SETNE: // !eq
488 case ISD::SETLE: // !gt
489 case ISD::SETGE: // !lt
490 case ISD::SETULE: // !ugt
491 case ISD::SETUGE: // !ult
492 CC = ISD::getSetCCInverse(CC, true);
493 Negate = true;
494 break;
495 default:
496 break;
497 }
498
499 switch (CC) {
500 case ISD::SETLT: // swap gt
501 case ISD::SETULT: // swap ugt
502 CC = ISD::getSetCCSwappedOperands(CC);
503 Swap = true;
504 break;
505 default:
506 break;
507 }
508
509 assert(CC == ISD::SETEQ || CC == ISD::SETGT || CC == ISD::SETUGT);
510
511 MVT ElemTy = VecTy.getVectorElementType();
512 unsigned ElemWidth = ElemTy.getSizeInBits();
513 assert(isPowerOf2_32(ElemWidth));
514
515 auto getIdx = [] (unsigned Code) {
516 static const unsigned Idx[] = { ISD::SETEQ, ISD::SETGT, ISD::SETUGT };
517 for (unsigned I = 0, E = array_lengthof(Idx); I != E; ++I)
518 if (Code == Idx[I])
519 return I;
520 llvm_unreachable("Unhandled CondCode");
521 };
522
523 static unsigned OpcTable[3][3] = {
524 // SETEQ SETGT, SETUGT
525 /* Byte */ { Hexagon::V6_veqb, Hexagon::V6_vgtb, Hexagon::V6_vgtub },
526 /* Half */ { Hexagon::V6_veqh, Hexagon::V6_vgth, Hexagon::V6_vgtuh },
527 /* Word */ { Hexagon::V6_veqw, Hexagon::V6_vgtw, Hexagon::V6_vgtuw }
528 };
529
530 unsigned CmpOpc = OpcTable[Log2_32(ElemWidth)-3][getIdx(CC)];
531
532 MVT ResTy = ty(Op);
533 const SDLoc &dl(Op);
534 SDValue OpL = Swap ? Op.getOperand(1) : Op.getOperand(0);
535 SDValue OpR = Swap ? Op.getOperand(0) : Op.getOperand(1);
536 SDValue CmpV = getNode(CmpOpc, dl, ResTy, {OpL, OpR}, DAG);
537 return Negate ? getNode(Hexagon::V6_pred_not, dl, ResTy, {CmpV}, DAG)
538 : CmpV;
539}
Krzysztof Parzyszek6b589e52017-12-18 18:32:27 +0000540
541SDValue
542HexagonTargetLowering::LowerHvxExtend(SDValue Op, SelectionDAG &DAG) const {
543 // Sign- and zero-extends are legal.
544 assert(Op.getOpcode() == ISD::ANY_EXTEND_VECTOR_INREG);
545 return DAG.getZeroExtendVectorInReg(Op.getOperand(0), SDLoc(Op), ty(Op));
546}