[X86] Fix shuffle decoding for variable VPERMIL to be tolerant of the Constant type...
[oota-llvm.git] / lib / Target / X86 / Utils / X86ShuffleDecode.h
1 //===-- X86ShuffleDecode.h - X86 shuffle decode logic -----------*-C++-*---===//
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 // Define several functions to decode x86 specific shuffle semantics into a
11 // generic vector mask.
12 //
13 //===----------------------------------------------------------------------===//
14
15 #ifndef LLVM_LIB_TARGET_X86_UTILS_X86SHUFFLEDECODE_H
16 #define LLVM_LIB_TARGET_X86_UTILS_X86SHUFFLEDECODE_H
17
18 #include "llvm/ADT/SmallVector.h"
19 #include "llvm/ADT/ArrayRef.h"
20
21 //===----------------------------------------------------------------------===//
22 //  Vector Mask Decoding
23 //===----------------------------------------------------------------------===//
24
25 namespace llvm {
26 class Constant;
27 class MVT;
28
29 enum { SM_SentinelUndef = -1, SM_SentinelZero = -2 };
30
31 void DecodeINSERTPSMask(unsigned Imm, SmallVectorImpl<int> &ShuffleMask);
32
33 // <3,1> or <6,7,2,3>
34 void DecodeMOVHLPSMask(unsigned NElts, SmallVectorImpl<int> &ShuffleMask);
35
36 // <0,2> or <0,1,4,5>
37 void DecodeMOVLHPSMask(unsigned NElts, SmallVectorImpl<int> &ShuffleMask);
38
39 void DecodeMOVSLDUPMask(MVT VT, SmallVectorImpl<int> &ShuffleMask);
40
41 void DecodeMOVSHDUPMask(MVT VT, SmallVectorImpl<int> &ShuffleMask);
42
43 void DecodeMOVDDUPMask(MVT VT, SmallVectorImpl<int> &ShuffleMask);
44
45 void DecodePSLLDQMask(MVT VT, unsigned Imm, SmallVectorImpl<int> &ShuffleMask);
46
47 void DecodePSRLDQMask(MVT VT, unsigned Imm, SmallVectorImpl<int> &ShuffleMask);
48
49 void DecodePALIGNRMask(MVT VT, unsigned Imm, SmallVectorImpl<int> &ShuffleMask);
50
51 void DecodePSHUFMask(MVT VT, unsigned Imm, SmallVectorImpl<int> &ShuffleMask);
52
53 void DecodePSHUFHWMask(MVT VT, unsigned Imm, SmallVectorImpl<int> &ShuffleMask);
54
55 void DecodePSHUFLWMask(MVT, unsigned Imm, SmallVectorImpl<int> &ShuffleMask);
56
57 /// \brief Decodes a PSWAPD 3DNow! instruction.
58 void DecodePSWAPMask(MVT VT, SmallVectorImpl<int> &ShuffleMask);
59
60 /// DecodeSHUFPMask - This decodes the shuffle masks for shufp*. VT indicates
61 /// the type of the vector allowing it to handle different datatypes and vector
62 /// widths.
63 void DecodeSHUFPMask(MVT VT, unsigned Imm, SmallVectorImpl<int> &ShuffleMask);
64
65 /// DecodeUNPCKHMask - This decodes the shuffle masks for unpckhps/unpckhpd
66 /// and punpckh*. VT indicates the type of the vector allowing it to handle
67 /// different datatypes and vector widths.
68 void DecodeUNPCKHMask(MVT VT, SmallVectorImpl<int> &ShuffleMask);
69
70 /// DecodeUNPCKLMask - This decodes the shuffle masks for unpcklps/unpcklpd
71 /// and punpckl*. VT indicates the type of the vector allowing it to handle
72 /// different datatypes and vector widths.
73 void DecodeUNPCKLMask(MVT VT, SmallVectorImpl<int> &ShuffleMask);
74
75 /// \brief Decode a PSHUFB mask from an IR-level vector constant.
76 void DecodePSHUFBMask(const Constant *C, SmallVectorImpl<int> &ShuffleMask);
77
78 /// \brief Decode a PSHUFB mask from a raw array of constants such as from
79 /// BUILD_VECTOR.
80 void DecodePSHUFBMask(ArrayRef<uint64_t> RawMask,
81                       SmallVectorImpl<int> &ShuffleMask);
82
83 /// \brief Decode a BLEND immediate mask into a shuffle mask.
84 void DecodeBLENDMask(MVT VT, unsigned Imm, SmallVectorImpl<int> &ShuffleMask);
85
86 void DecodeVPERM2X128Mask(MVT VT, unsigned Imm,
87                           SmallVectorImpl<int> &ShuffleMask);
88
89 /// \brief Decode a shuffle packed values at 128-bit granularity
90 /// immediate mask into a shuffle mask.
91 void decodeVSHUF64x2FamilyMask(MVT VT, unsigned Imm,
92                                SmallVectorImpl<int> &ShuffleMask);
93
94 /// DecodeVPERMMask - this decodes the shuffle masks for VPERMQ/VPERMPD.
95 /// No VT provided since it only works on 256-bit, 4 element vectors.
96 void DecodeVPERMMask(unsigned Imm, SmallVectorImpl<int> &ShuffleMask);
97
98 /// \brief Decode a VPERMILP variable mask from an IR-level vector constant.
99 void DecodeVPERMILPMask(const Constant *C, unsigned ElSize,
100                         SmallVectorImpl<int> &ShuffleMask);
101
102 /// \brief Decode a zero extension instruction as a shuffle mask.
103 void DecodeZeroExtendMask(MVT SrcVT, MVT DstVT,
104                           SmallVectorImpl<int> &ShuffleMask);
105
106 /// \brief Decode a move lower and zero upper instruction as a shuffle mask.
107 void DecodeZeroMoveLowMask(MVT VT, SmallVectorImpl<int> &ShuffleMask);
108
109 /// \brief Decode a scalar float move instruction as a shuffle mask.
110 void DecodeScalarMoveMask(MVT VT, bool IsLoad,
111                           SmallVectorImpl<int> &ShuffleMask);
112
113 /// \brief Decode a SSE4A EXTRQ instruction as a v16i8 shuffle mask.
114 void DecodeEXTRQIMask(int Len, int Idx,
115                       SmallVectorImpl<int> &ShuffleMask);
116
117 /// \brief Decode a SSE4A INSERTQ instruction as a v16i8 shuffle mask.
118 void DecodeINSERTQIMask(int Len, int Idx,
119                         SmallVectorImpl<int> &ShuffleMask);
120
121 /// \brief Decode a VPERM W/D/Q/PS/PD mask from an IR-level vector constant.
122 void DecodeVPERMVMask(const Constant *C, MVT VT,
123                       SmallVectorImpl<int> &ShuffleMask);
124
125 /// \brief Decode a VPERM W/D/Q/PS/PD mask from a raw array of constants.
126 void DecodeVPERMVMask(ArrayRef<uint64_t> RawMask,
127                       SmallVectorImpl<int> &ShuffleMask);
128
129 /// \brief Decode a VPERMT2 W/D/Q/PS/PD mask from an IR-level vector constant.
130 void DecodeVPERMV3Mask(const Constant *C, MVT VT,
131                       SmallVectorImpl<int> &ShuffleMask);
132
133 /// \brief Decode a VPERMT2 W/D/Q/PS/PD mask from a raw array of constants.
134 void DecodeVPERMV3Mask(ArrayRef<uint64_t> RawMask,
135                       SmallVectorImpl<int> &ShuffleMask);
136 } // llvm namespace
137
138 #endif