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, SmallVectorImpl<int> &ShuffleMask); 100 101 /// \brief Decode a zero extension instruction as a shuffle mask. 102 void DecodeZeroExtendMask(MVT SrcVT, MVT DstVT, 103 SmallVectorImpl<int> &ShuffleMask); 104 105 /// \brief Decode a move lower and zero upper instruction as a shuffle mask. 106 void DecodeZeroMoveLowMask(MVT VT, SmallVectorImpl<int> &ShuffleMask); 107 108 /// \brief Decode a scalar float move instruction as a shuffle mask. 109 void DecodeScalarMoveMask(MVT VT, bool IsLoad, 110 SmallVectorImpl<int> &ShuffleMask); 111 112 /// \brief Decode a SSE4A EXTRQ instruction as a v16i8 shuffle mask. 113 void DecodeEXTRQIMask(int Len, int Idx, 114 SmallVectorImpl<int> &ShuffleMask); 115 116 /// \brief Decode a SSE4A INSERTQ instruction as a v16i8 shuffle mask. 117 void DecodeINSERTQIMask(int Len, int Idx, 118 SmallVectorImpl<int> &ShuffleMask); 119 120 /// \brief Decode a VPERM W/D/Q/PS/PD mask from an IR-level vector constant. 121 void DecodeVPERMVMask(const Constant *C, MVT VT, 122 SmallVectorImpl<int> &ShuffleMask); 123 124 /// \brief Decode a VPERM W/D/Q/PS/PD mask from a raw array of constants. 125 void DecodeVPERMVMask(ArrayRef<uint64_t> RawMask, 126 SmallVectorImpl<int> &ShuffleMask); 127 128 /// \brief Decode a VPERMT2 W/D/Q/PS/PD mask from an IR-level vector constant. 129 void DecodeVPERMV3Mask(const Constant *C, MVT VT, 130 SmallVectorImpl<int> &ShuffleMask); 131 132 /// \brief Decode a VPERMT2 W/D/Q/PS/PD mask from a raw array of constants. 133 void DecodeVPERMV3Mask(ArrayRef<uint64_t> RawMask, 134 SmallVectorImpl<int> &ShuffleMask); 135 } // llvm namespace 136 137 #endif 138