24#include "llvm/IR/IntrinsicsSPIRV.h"
32#define DEBUG_TYPE "spirv-legalizer"
35 return [IsExtendedInts, TypeIdx](
const LegalityQuery &Query) {
36 const LLT Ty = Query.Types[TypeIdx];
37 return IsExtendedInts && Ty.isValid() && Ty.isScalar();
43 const LLT Ty = Query.Types[TypeIdx];
44 return IsLongVecs && Ty.isValid() && Ty.isVector();
91 const unsigned PSize = ST.getPointerSize();
110 auto allPtrsScalarsAndVectors = {
111 p0, p1, p2, p3, p4, p5, p6, p7, p8,
112 p9, p10, p11, p12, p13, s1, s8, s16, s32,
113 s64, s128, v2s1, v2s8, v2s16, v2s32, v2s64, v3s1, v3s8,
114 v3s16, v3s32, v3s64, v4s1, v4s8, v4s16, v4s32, v4s64, v8s1,
115 v8s8, v8s16, v8s32, v8s64, v16s1, v16s8, v16s16, v16s32, v16s64};
117 auto allVectors = {v2s1, v2s8, v2s16, v2s32, v2s64, v3s1, v3s8,
118 v3s16, v3s32, v3s64, v4s1, v4s8, v4s16, v4s32,
119 v4s64, v8s1, v8s8, v8s16, v8s32, v8s64, v16s1,
120 v16s8, v16s16, v16s32, v16s64};
122 auto allShaderVectors = {v2s1, v2s8, v2s16, v2s32, v2s64,
123 v3s1, v3s8, v3s16, v3s32, v3s64,
124 v4s1, v4s8, v4s16, v4s32, v4s64};
126 auto allScalars = {s1, s8, s16, s32, s64};
128 auto allScalarsAndVectors = {
129 s1, s8, s16, s32, s64, s128, v2s1, v2s8,
130 v2s16, v2s32, v2s64, v3s1, v3s8, v3s16, v3s32, v3s64,
131 v4s1, v4s8, v4s16, v4s32, v4s64, v8s1, v8s8, v8s16,
132 v8s32, v8s64, v16s1, v16s8, v16s16, v16s32, v16s64};
134 auto allShaderScalarsAndVectors = {
135 s1, s8, s16, s32, s64, s128, v2s1, v2s8, v2s16, v2s32, v2s64,
136 v3s1, v3s8, v3s16, v3s32, v3s64, v4s1, v4s8, v4s16, v4s32, v4s64};
138 auto &allowedScalarsAndVectors =
139 ST.isShader() ? allShaderScalarsAndVectors : allScalarsAndVectors;
141 auto allIntScalarsAndVectors = {
142 s8, s16, s32, s64, s128, v2s8, v2s16, v2s32, v2s64,
143 v3s8, v3s16, v3s32, v3s64, v4s8, v4s16, v4s32, v4s64, v8s8,
144 v8s16, v8s32, v8s64, v16s8, v16s16, v16s32, v16s64};
146 auto allBoolScalarsAndVectors = {s1, v2s1, v3s1, v4s1, v8s1, v16s1};
147 auto allBoolVectors = {v2s1, v3s1, v4s1, v8s1, v16s1};
149 auto allIntScalars = {s8, s16, s32, s64, s128};
151 auto allShaderIntVectors = {v2s8, v2s16, v2s32, v2s64, v3s8, v3s16,
152 v3s32, v3s64, v4s8, v4s16, v4s32, v4s64};
154 auto allIntVectors = {v2s8, v2s16, v2s32, v2s64, v3s8, v3s16, v3s32,
155 v3s64, v4s8, v4s16, v4s32, v4s64, v8s8, v8s16,
156 v8s32, v8s64, v16s8, v16s16, v16s32, v16s64};
158 auto &allowedIntVectorTypes =
159 ST.isShader() ? allShaderIntVectors : allIntVectors;
161 auto allFloatScalarsAndF16Vector2AndVector4s = {s16, s32, s64, v2s16, v4s16};
163 auto allFloatScalars = {s16, s32, s64};
165 auto allFloatScalarsAndVectors = {
166 s16, s32, s64, v2s16, v2s32, v2s64, v3s16, v3s32, v3s64,
167 v4s16, v4s32, v4s64, v8s16, v8s32, v8s64, v16s16, v16s32, v16s64};
169 auto allShaderFloatVectors = {v2s16, v2s32, v2s64, v3s16, v3s32,
170 v3s64, v4s16, v4s32, v4s64};
172 auto allFloatVectors = {v2s16, v2s32, v2s64, v3s16, v3s32,
173 v3s64, v4s16, v4s32, v4s64, v8s16,
174 v8s32, v8s64, v16s16, v16s32, v16s64};
176 auto &allowedFloatVectorTypes =
177 ST.isShader() ? allShaderFloatVectors : allFloatVectors;
179 auto allFloatAndIntScalarsAndPtrs = {s8, s16, s32, s64, p0, p1,
180 p2, p3, p4, p5, p6, p7,
181 p8, p9, p10, p11, p12, p13};
183 auto allPtrs = {p0, p1, p2, p3, p4, p5, p6, p7, p8, p9, p10, p11, p12, p13};
185 auto &allowedVectorTypes = ST.isShader() ? allShaderVectors : allVectors;
187 bool HasArbitraryPrecisionInts = ST.canUseExtension(
188 SPIRV::Extension::SPV_ALTERA_arbitrary_precision_integers);
189 bool IsExtendedInts =
190 HasArbitraryPrecisionInts ||
191 ST.canUseExtension(SPIRV::Extension::SPV_KHR_bit_instructions) ||
192 ST.canUseExtension(SPIRV::Extension::SPV_INTEL_int4);
193 bool IsLongVecs = ST.canUseExtension(SPIRV::Extension::SPV_EXT_long_vector);
194 auto ExtendedIntScalarsAndVectors =
196 const LLT Ty = Query.Types[0];
197 return IsExtendedInts && Ty.isValid() &&
198 !Ty.isPointerOrPointerVector() && Ty.getScalarSizeInBits() > 1;
200 auto ExtendedScalarsAndVectorsProduct = [IsExtendedInts](
202 const LLT Ty1 = Query.Types[0], Ty2 = Query.Types[1];
203 return IsExtendedInts && Ty1.
isValid() && Ty2.isValid() &&
206 auto ExtendedPtrsScalarsAndVectors =
208 const LLT Ty = Query.Types[0];
209 return IsExtendedInts && Ty.isValid();
218 uint32_t MaxVectorSize = ST.isShader() ? 4 : 16;
223 case G_EXTRACT_VECTOR_ELT:
245 .customFor(allScalars)
255 .legalFor(allScalars)
332 {G_VECREDUCE_SMIN, G_VECREDUCE_SMAX, G_VECREDUCE_UMIN, G_VECREDUCE_UMAX,
333 G_VECREDUCE_ADD, G_VECREDUCE_MUL, G_VECREDUCE_FMUL, G_VECREDUCE_FMIN,
334 G_VECREDUCE_FMAX, G_VECREDUCE_FMINIMUM, G_VECREDUCE_FMAXIMUM,
335 G_VECREDUCE_OR, G_VECREDUCE_AND, G_VECREDUCE_XOR})
336 .legalFor(allowedVectorTypes)
357 .unsupportedIf(
typeIs(0, p9))
366 .unsupportedIf(
typeIs(1, p9))
374 G_BITREVERSE, G_SADDSAT, G_UADDSAT, G_SSUBSAT,
375 G_USUBSAT, G_SCMP, G_UCMP})
376 .legalFor(allIntScalars)
378 .
legalIf(ExtendedIntScalarsAndVectors)
390 .legalForCartesianProduct(allFloatScalarsAndVectors, allIntScalars);
393 .legalForCartesianProduct(allIntScalarsAndVectors,
394 allFloatScalarsAndVectors);
397 .legalForCartesianProduct(allIntScalarsAndVectors,
398 allFloatScalarsAndVectors);
401 .legalForCartesianProduct(allFloatScalarsAndVectors,
402 allScalarsAndVectors);
406 .
legalIf(ExtendedScalarsAndVectorsProduct)
410 .legalForCartesianProduct(allowedScalarsAndVectors)
411 .
legalIf(ExtendedScalarsAndVectorsProduct)
432 .
legalIf(ExtendedPtrsScalarsAndVectors)
437 typeInSet(1, allPtrsScalarsAndVectors)));
440 .legalFor({s1, s128})
441 .legalFor(allFloatAndIntScalarsAndPtrs)
444 return Query.
Types[0].isPointerVector();
462 !SrcTy.isPointer() &&
472 return SrcTy.isPointerVector() && DstTy.
isVector() &&
504 return IsExtendedInts && Ty.isValid() && !Ty.isPointerOrPointerVector();
507 typeInSet(1, allPtrsScalarsAndVectors)));
514 typeInSet(1, allFloatScalarsAndVectors)));
517 G_ATOMICRMW_MAX, G_ATOMICRMW_MIN,
518 G_ATOMICRMW_SUB, G_ATOMICRMW_XOR,
519 G_ATOMICRMW_UMAX, G_ATOMICRMW_UMIN})
520 .legalForCartesianProduct(allIntScalars, allPtrs);
523 {G_ATOMICRMW_FADD, G_ATOMICRMW_FSUB, G_ATOMICRMW_FMIN, G_ATOMICRMW_FMAX})
524 .legalForCartesianProduct(allFloatScalarsAndF16Vector2AndVector4s,
543 if (!HasArbitraryPrecisionInts)
548 .legalForCartesianProduct(allFloatScalarsAndVectors,
549 allIntScalarsAndVectors);
553 .legalForCartesianProduct(allFloatScalarsAndVectors);
565 allFloatScalarsAndVectors, {s32, v2s32, v3s32, v4s32, v8s32, v16s32});
604 G_INTRINSIC_ROUNDEVEN})
605 .legalFor(allFloatScalars)
615 allFloatScalarsAndVectors);
618 allFloatScalarsAndVectors, allIntScalarsAndVectors);
620 if (ST.canUseExtInstSet(SPIRV::InstructionSet::OpenCL_std)) {
622 {G_CTTZ, G_CTTZ_ZERO_POISON, G_CTLZ, G_CTLZ_ZERO_POISON})
623 .legalForCartesianProduct(allIntScalarsAndVectors,
624 allIntScalarsAndVectors);
633 G_INTRINSIC_CONVERGENT_W_SIDE_EFFECTS})
638 verify(*ST.getInstrInfo());
652 MI.eraseFromParent();
668 MI.eraseFromParent();
686 if (!Ty.isVector() ||
687 ST.canUseExtension(SPIRV::Extension::SPV_EXT_long_vector))
689 unsigned NumElements = Ty.getNumElements();
690 unsigned MaxVectorSize = ST.isShader() ? 4 : 16;
692 NumElements > MaxVectorSize;
717 for (
unsigned i = 0; i < NumElts; ++i) {
729 if (!
MI.memoperands_empty()) {
737 MIRBuilder.
buildLoad(EltReg, EltPtr, EltPtrInfo, EltAlign);
742 MI.eraseFromParent();
760 for (
unsigned i = 0; i < NumElts; ++i)
768 for (
unsigned i = 0; i < NumElts; ++i) {
780 if (!
MI.memoperands_empty()) {
787 MIRBuilder.
buildStore(SplitRegs[i], EltPtr, EltPtrInfo, EltAlign);
790 MI.eraseFromParent();
798 switch (
MI.getOpcode()) {
802 case TargetOpcode::G_BITCAST:
803 return legalizeBitcast(Helper,
MI);
804 case TargetOpcode::G_EXTRACT_VECTOR_ELT:
806 case TargetOpcode::G_INSERT_VECTOR_ELT:
808 case TargetOpcode::G_INTRINSIC:
809 case TargetOpcode::G_INTRINSIC_W_SIDE_EFFECTS:
811 case TargetOpcode::G_IS_FPCLASS:
812 return legalizeIsFPClass(Helper,
MI, LocObserver);
813 case TargetOpcode::G_ICMP: {
814 auto &Op0 =
MI.getOperand(2);
815 auto &Op1 =
MI.getOperand(3);
820 if ((!ST->canDirectlyComparePointers() ||
825 ST->getPointerSize());
827 LLVMTy, Helper.
MIRBuilder, SPIRV::AccessQualifier::ReadWrite,
true);
833 case TargetOpcode::G_LOAD:
835 case TargetOpcode::G_STORE:
855 const Type *LLVMArrTy =
858 LLVMArrTy, MIRBuilder, SPIRV::AccessQualifier::ReadWrite,
true);
860 ArrSpvTy, MIRBuilder, SPIRV::StorageClass::Function);
862 Register StackReg = StackTemp.getReg(0);
887 MI.eraseFromParent();
907 if (
getImm(IdxOperand, &MRI)) {
909 if (IdxVal < SrcTy.getNumElements()) {
914 for (
unsigned I = 0,
E = SrcTy.getNumElements();
I <
E; ++
I) {
921 Regs[IdxVal] = ValReg;
923 MI.eraseFromParent();
934 MIRBuilder.
buildStore(SrcReg, StackTemp, PtrInfo, VecAlign);
943 .
addUse(StackTemp.getReg(0))
949 MIRBuilder.
buildStore(ValReg, EltPtr, EltPtrInfo, EltAlign);
951 MIRBuilder.
buildLoad(DstReg, StackTemp, PtrInfo, VecAlign);
952 MI.eraseFromParent();
971 if (
getImm(IdxOperand, &MRI)) {
973 if (IdxVal < SrcTy.getNumElements()) {
977 for (
unsigned I = 0,
E = SrcTy.getNumElements();
I <
E; ++
I) {
988 MI.eraseFromParent();
999 MIRBuilder.
buildStore(SrcReg, StackTemp, PtrInfo, VecAlign);
1002 LLT PtrTy = MRI.
getType(StackTemp.getReg(0));
1008 .
addUse(StackTemp.getReg(0))
1014 MIRBuilder.
buildLoad(DstReg, EltPtr, EltPtrInfo, EltAlign);
1016 MI.eraseFromParent();
1035 if (
MI.getNumOperands() == 2) {
1045 for (
unsigned i = 2; i <
MI.getNumOperands(); ++i) {
1050 MI.eraseFromParent();
1058 switch (IntrinsicID) {
1059 case Intrinsic::spv_bitcast:
1061 case Intrinsic::spv_insertelt:
1063 case Intrinsic::spv_extractelt:
1065 case Intrinsic::spv_const_composite:
1082 MI.eraseFromParent();
1089bool SPIRVLegalizerInfo::legalizeIsFPClass(
1092 auto [DstReg, DstTy, SrcReg, SrcTy] =
MI.getFirst2RegLLTs();
1096 auto &MF = MIRBuilder.
getMF();
1101 if (DstTy.isVector())
1103 SPIRVTypeInst SPIRVDstTy = GR->getOrCreateSPIRVType(
1104 LLVMDstTy, MIRBuilder, SPIRV::AccessQualifier::ReadWrite,
1107 unsigned BitSize = SrcTy.getScalarSizeInBits();
1112 if (SrcTy.isVector()) {
1113 IntTy =
LLT::vector(SrcTy.getElementCount(), IntTy);
1116 SPIRVTypeInst SPIRVIntTy = GR->getOrCreateSPIRVType(
1117 LLVMIntTy, MIRBuilder, SPIRV::AccessQualifier::ReadWrite,
1121 LLT DstTyCopy = DstTy;
1122 const auto assignSPIRVTy = [&](MachineInstrBuilder &&
MI) {
1127 assert((MITy == IntTy || MITy == DstTyCopy) &&
1128 "Unexpected LLT type while lowering G_IS_FPCLASS");
1129 SPIRVTypeInst SPVTy = MITy == IntTy ? SPIRVIntTy : SPIRVDstTy;
1130 GR->assignSPIRVTypeToVReg(SPVTy,
MI.getReg(0), MF);
1135 const auto buildSPIRVConstant = [&](LLT Ty,
auto &&
C) -> MachineInstrBuilder {
1139 assert((Ty == IntTy || Ty == DstTyCopy) &&
1140 "Unexpected LLT type while lowering constant for G_IS_FPCLASS");
1141 SPIRVTypeInst VecEltTy = GR->getOrCreateSPIRVType(
1142 (Ty == IntTy ? LLVMIntTy : LLVMDstTy)->getScalarType(), MIRBuilder,
1143 SPIRV::AccessQualifier::ReadWrite,
1145 GR->assignSPIRVTypeToVReg(VecEltTy, ScalarC.getReg(0), MF);
1150 MIRBuilder.
buildCopy(DstReg, buildSPIRVConstant(DstTy, 0));
1151 MI.eraseFromParent();
1155 MIRBuilder.
buildCopy(DstReg, buildSPIRVConstant(DstTy, 1));
1156 MI.eraseFromParent();
1165 MRI.
setRegClass(ResVReg, GR->getRegClass(SPIRVIntTy));
1166 GR->assignSPIRVTypeToVReg(SPIRVIntTy, ResVReg, Helper.
MIRBuilder.
getMF());
1167 auto AsInt = MIRBuilder.
buildInstr(SPIRV::OpBitcast)
1169 .
addUse(GR->getSPIRVTypeID(SPIRVIntTy))
1171 AsInt = assignSPIRVTy(std::move(AsInt));
1177 APInt ExpMask = Inf;
1183 auto SignBitC = buildSPIRVConstant(IntTy, SignBit);
1184 auto ValueMaskC = buildSPIRVConstant(IntTy, ValueMask);
1185 auto InfC = buildSPIRVConstant(IntTy, Inf);
1186 auto ExpMaskC = buildSPIRVConstant(IntTy, ExpMask);
1187 auto ZeroC = buildSPIRVConstant(IntTy, 0);
1189 auto Abs = assignSPIRVTy(MIRBuilder.
buildAnd(IntTy, AsInt, ValueMaskC));
1190 auto Sign = assignSPIRVTy(
1193 auto Res = buildSPIRVConstant(DstTy, 0);
1195 const auto appendToRes = [&](MachineInstrBuilder &&ToAppend) {
1196 Res = assignSPIRVTy(
1197 MIRBuilder.
buildOr(DstTyCopy, Res, assignSPIRVTy(std::move(ToAppend))));
1210 Mask &= ~fcPosFinite;
1214 DstTy, Abs, ExpMaskC));
1215 appendToRes(MIRBuilder.
buildAnd(DstTy, Cmp, Sign));
1216 Mask &= ~fcNegFinite;
1224 auto ExpBits = assignSPIRVTy(MIRBuilder.
buildAnd(IntTy, AsInt, ExpMaskC));
1227 Mask &= ~PartialCheck;
1236 else if (PartialCheck ==
fcZero)
1248 auto OneC = buildSPIRVConstant(IntTy, 1);
1249 auto VMinusOne = MIRBuilder.
buildSub(IntTy, V, OneC);
1250 auto SubnormalRes = assignSPIRVTy(
1252 buildSPIRVConstant(IntTy, AllOneMantissa)));
1254 SubnormalRes = MIRBuilder.
buildAnd(DstTy, SubnormalRes, Sign);
1255 appendToRes(std::move(SubnormalRes));
1262 else if (PartialCheck ==
fcInf)
1267 auto NegInfC = buildSPIRVConstant(IntTy, NegInf);
1274 auto InfWithQnanBitC =
1275 buildSPIRVConstant(IntTy, std::move(Inf) | QNaNBitMask);
1276 if (PartialCheck ==
fcNan) {
1280 }
else if (PartialCheck ==
fcQNan) {
1287 auto IsNan = assignSPIRVTy(
1289 auto IsNotQnan = assignSPIRVTy(MIRBuilder.
buildICmp(
1291 appendToRes(MIRBuilder.
buildAnd(DstTy, IsNan, IsNotQnan));
1298 APInt ExpLSB = ExpMask & ~(ExpMask.
shl(1));
1299 auto ExpMinusOne = assignSPIRVTy(
1300 MIRBuilder.
buildSub(IntTy, Abs, buildSPIRVConstant(IntTy, ExpLSB)));
1301 APInt MaxExpMinusOne = std::move(ExpMask) - ExpLSB;
1302 auto NormalRes = assignSPIRVTy(
1304 buildSPIRVConstant(IntTy, MaxExpMinusOne)));
1306 NormalRes = MIRBuilder.
buildAnd(DstTy, NormalRes, Sign);
1308 auto PosSign = assignSPIRVTy(MIRBuilder.
buildXor(
1309 DstTy, Sign, buildSPIRVConstant(DstTy, InversionMask)));
1310 NormalRes = MIRBuilder.
buildAnd(DstTy, NormalRes, PosSign);
1312 appendToRes(std::move(NormalRes));
1316 MI.eraseFromParent();
assert(UImm &&(UImm !=~static_cast< T >(0)) &&"Invalid immediate!")
static GCRegistry::Add< ShadowStackGC > C("shadow-stack", "Very portable GC for uncooperative code generators")
static GCRegistry::Add< CoreCLRGC > E("coreclr", "CoreCLR-compatible GC")
static void scalarize(Instruction *I, SmallVectorImpl< Instruction * > &Worklist)
Declares convenience wrapper classes for interpreting MachineInstr instances as specific generic oper...
This file declares the MachineIRBuilder class.
Promote Memory to Register
const SmallVectorImpl< MachineOperand > & Cond
static bool legalizeSpvInsertElt(LegalizerHelper &Helper, MachineInstr &MI, SPIRVGlobalRegistry *GR)
static bool needsVectorLegalization(const LLT &Ty, const SPIRVSubtarget &ST)
static bool legalizeInsertVectorElt(LegalizerHelper &Helper, MachineInstr &MI)
static MachineInstrBuilder createStackTemporaryForVector(LegalizerHelper &Helper, SPIRVGlobalRegistry *GR, Register SrcReg, LLT SrcTy, MachinePointerInfo &PtrInfo, Align &VecAlign)
static Register convertPtrToInt(Register Reg, LLT ConvTy, SPIRVTypeInst SpvType, LegalizerHelper &Helper, MachineRegisterInfo &MRI, SPIRVGlobalRegistry *GR)
LegalityPredicate typeOfExtendedScalars(unsigned TypeIdx, bool IsExtendedInts)
static bool legalizeStore(LegalizerHelper &Helper, MachineInstr &MI, SPIRVGlobalRegistry *GR)
static bool legalizeExtractVectorElt(LegalizerHelper &Helper, MachineInstr &MI)
static bool legalizeSpvExtractElt(LegalizerHelper &Helper, MachineInstr &MI, SPIRVGlobalRegistry *GR)
LegalityPredicate typeOfLongVectors(unsigned TypeIdx, bool IsLongVecs)
static bool legalizeSpvBitcast(LegalizerHelper &Helper, MachineInstr &MI, SPIRVGlobalRegistry *GR)
static bool legalizeSpvConstComposite(LegalizerHelper &Helper, MachineInstr &MI, SPIRVGlobalRegistry *GR)
static bool legalizeLoad(LegalizerHelper &Helper, MachineInstr &MI, SPIRVGlobalRegistry *GR)
APInt bitcastToAPInt() const
static APFloat getLargest(const fltSemantics &Sem, bool Negative=false)
Returns the largest finite number in the given semantics.
static APFloat getInf(const fltSemantics &Sem, bool Negative=false)
Factory for Positive and Negative Infinity.
static APInt getAllOnes(unsigned numBits)
Return an APInt of a specified width with all bits set.
static APInt getSignMask(unsigned BitWidth)
Get the SignMask for a specific bit width.
unsigned getActiveBits() const
Compute the number of active bits in the value.
static APInt getSignedMaxValue(unsigned numBits)
Gets maximum signed value of APInt for a specific bit width.
APInt shl(unsigned shiftAmt) const
Left-shift function.
static APInt getOneBitSet(unsigned numBits, unsigned BitNo)
Return an APInt with exactly one bit set in the result.
Represent a constant reference to an array (0 or more elements consecutively in memory),...
static LLVM_ABI ArrayType * get(Type *ElementType, uint64_t NumElements)
This static method is the primary way to construct an ArrayType.
Predicate
This enumeration lists the possible predicates for CmpInst subclasses.
@ ICMP_UGE
unsigned greater or equal
@ ICMP_UGT
unsigned greater than
@ ICMP_ULT
unsigned less than
static constexpr ElementCount getFixed(ScalarTy MinVal)
static LLVM_ABI IntegerType * get(LLVMContext &C, unsigned NumBits)
This static method is the primary way of constructing an IntegerType.
static constexpr LLT vector(ElementCount EC, unsigned ScalarSizeInBits)
Get a low-level vector of some number of elements and element width.
LLT getScalarType() const
constexpr bool isPointerVector() const
static constexpr LLT scalar(unsigned SizeInBits)
Get a low-level scalar or aggregate "bag of bits".
constexpr bool isValid() const
constexpr uint16_t getNumElements() const
Returns the number of elements in a vector LLT.
constexpr bool isVector() const
static constexpr LLT pointer(unsigned AddressSpace, unsigned SizeInBits)
Get a low-level pointer in the given address space.
constexpr bool isPointer() const
constexpr unsigned getAddressSpace() const
static constexpr LLT fixed_vector(unsigned NumElements, unsigned ScalarSizeInBits)
Get a low-level fixed-width vector of some number of elements and element width.
constexpr bool isPointerOrPointerVector() const
constexpr bool isFixedVector() const
Returns true if the LLT is a fixed vector.
constexpr TypeSize getSizeInBytes() const
Returns the total size of the type in bytes, i.e.
LLT getElementType() const
Returns the vector's element type. Only valid for vector types.
LegalizeRuleSet & minScalar(unsigned TypeIdx, const LLT Ty)
Ensure the scalar is at least as wide as Ty.
LegalizeRuleSet & legalFor(std::initializer_list< LLT > Types)
The instruction is legal when type index 0 is any type in the given list.
LegalizeRuleSet & fewerElementsIf(LegalityPredicate Predicate, LegalizeMutation Mutation)
Remove elements to reach the type selected by the mutation if the predicate is true.
LegalizeRuleSet & unsupportedFor(std::initializer_list< LLT > Types)
LegalizeRuleSet & moreElementsToNextPow2(unsigned TypeIdx)
Add more elements to the vector to reach the next power of two.
LegalizeRuleSet & lower()
The instruction is lowered.
LegalizeRuleSet & scalarizeIf(LegalityPredicate Predicate, unsigned TypeIdx)
LegalizeRuleSet & lowerIf(LegalityPredicate Predicate)
The instruction is lowered if predicate is true.
LegalizeRuleSet & custom()
Unconditionally custom lower.
LegalizeRuleSet & unsupportedIf(LegalityPredicate Predicate)
LegalizeRuleSet & alwaysLegal()
LegalizeRuleSet & customIf(LegalityPredicate Predicate)
LegalizeRuleSet & scalarize(unsigned TypeIdx)
LegalizeRuleSet & legalForCartesianProduct(std::initializer_list< LLT > Types)
The instruction is legal when type indexes 0 and 1 are both in the given list.
LegalizeRuleSet & legalIf(LegalityPredicate Predicate)
The instruction is legal if predicate is true.
LegalizeRuleSet & customFor(std::initializer_list< LLT > Types)
LLVM_ABI MachineInstrBuilder createStackTemporary(TypeSize Bytes, Align Alignment, MachinePointerInfo &PtrInfo)
Create a stack temporary based on the size in bytes and the alignment.
MachineIRBuilder & MIRBuilder
Expose MIRBuilder so clients can set their own RecordInsertInstruction functions.
LLVM_ABI Align getStackTemporaryAlignment(LLT Type, Align MinAlign=Align()) const
Return the alignment to use for a stack temporary object with the given type.
LegalizeRuleSet & getActionDefinitionsBuilder(unsigned Opcode)
Get the action definition builder for the given opcode.
MachineRegisterInfo & getRegInfo()
getRegInfo - Return information about the registers currently in use.
Helper class to build MachineInstr.
LLVMContext & getContext() const
MachineInstrBuilder buildUnmerge(ArrayRef< LLT > Res, const SrcOp &Op)
Build and insert Res0, ... = G_UNMERGE_VALUES Op.
MachineInstrBuilder buildAnd(const DstOp &Dst, const SrcOp &Src0, const SrcOp &Src1)
Build and insert Res = G_AND Op0, Op1.
MachineInstrBuilder buildICmp(CmpInst::Predicate Pred, const DstOp &Res, const SrcOp &Op0, const SrcOp &Op1, std::optional< unsigned > Flags=std::nullopt)
Build and insert a Res = G_ICMP Pred, Op0, Op1.
MachineInstrBuilder buildSub(const DstOp &Dst, const SrcOp &Src0, const SrcOp &Src1, std::optional< unsigned > Flags=std::nullopt)
Build and insert Res = G_SUB Op0, Op1.
MachineInstrBuilder buildIntrinsic(Intrinsic::ID ID, ArrayRef< Register > Res, bool HasSideEffects, bool isConvergent)
Build and insert a G_INTRINSIC instruction.
MachineInstrBuilder buildSplatBuildVector(const DstOp &Res, const SrcOp &Src)
Build and insert Res = G_BUILD_VECTOR with Src replicated to fill the number of elements.
MachineInstrBuilder buildBuildVector(const DstOp &Res, ArrayRef< Register > Ops)
Build and insert Res = G_BUILD_VECTOR Op0, ...
MachineInstrBuilder buildLoad(const DstOp &Res, const SrcOp &Addr, MachineMemOperand &MMO)
Build and insert Res = G_LOAD Addr, MMO.
MachineInstrBuilder buildStore(const SrcOp &Val, const SrcOp &Addr, MachineMemOperand &MMO)
Build and insert G_STORE Val, Addr, MMO.
MachineInstrBuilder buildInstr(unsigned Opcode)
Build and insert <empty> = Opcode <empty>.
MachineFunction & getMF()
Getter for the function we currently build.
MachineInstrBuilder buildBitcast(const DstOp &Dst, const SrcOp &Src)
Build and insert Dst = G_BITCAST Src.
MachineRegisterInfo * getMRI()
Getter for MRI.
MachineInstrBuilder buildOr(const DstOp &Dst, const SrcOp &Src0, const SrcOp &Src1, std::optional< unsigned > Flags=std::nullopt)
Build and insert Res = G_OR Op0, Op1.
MachineInstrBuilder buildCopy(const DstOp &Res, const SrcOp &Op)
Build and insert Res = COPY Op.
MachineInstrBuilder buildXor(const DstOp &Dst, const SrcOp &Src0, const SrcOp &Src1)
Build and insert Res = G_XOR Op0, Op1.
virtual MachineInstrBuilder buildConstant(const DstOp &Res, const ConstantInt &Val)
Build and insert Res = G_CONSTANT Val.
const MachineInstrBuilder & addUse(Register RegNo, RegState Flags={}, unsigned SubReg=0) const
Add a virtual register use operand.
const MachineInstrBuilder & addDef(Register RegNo, RegState Flags={}, unsigned SubReg=0) const
Add a virtual register definition operand.
Representation of each machine instruction.
A description of a memory reference used in the backend.
const MachinePointerInfo & getPointerInfo() const
LLVM_ABI Align getAlign() const
Return the minimum known alignment in bytes of the actual memory reference.
MachineOperand class - Representation of each machine instruction operand.
Register getReg() const
getReg - Returns the register number.
MachineRegisterInfo - Keep track of information for virtual and physical registers,...
LLT getType(Register Reg) const
Get the low-level type of Reg or LLT{} if Reg is not a generic (target independent) virtual register.
LLVM_ABI void setRegClass(Register Reg, const TargetRegisterClass *RC)
setRegClass - Set the register class of the specified virtual register.
LLVM_ABI Register createGenericVirtualRegister(LLT Ty, StringRef Name="")
Create and return a new generic virtual register with low-level type Ty.
Wrapper class representing virtual and physical registers.
void assignSPIRVTypeToVReg(SPIRVTypeInst Type, Register VReg, const MachineFunction &MF)
SPIRVTypeInst getOrCreateSPIRVPointerType(const Type *BaseType, MachineIRBuilder &MIRBuilder, SPIRV::StorageClass::StorageClass SC, bool ForceTyped=false)
const TargetRegisterClass * getRegClass(SPIRVTypeInst SpvType) const
const Type * getTypeForSPIRVType(SPIRVTypeInst Ty) const
LLT getRegType(SPIRVTypeInst SpvType) const
SPIRVTypeInst getScalarOrVectorComponentType(SPIRVTypeInst Type) const
SPIRVTypeInst getOrCreateSPIRVType(const Type *Type, MachineInstr &I, SPIRV::AccessQualifier::AccessQualifier AQ, bool EmitIR)
SPIRVTypeInst getSPIRVTypeForVReg(Register VReg, const MachineFunction *MF=nullptr) const
SPIRVLegalizerInfo(const SPIRVSubtarget &ST)
bool legalizeCustom(LegalizerHelper &Helper, MachineInstr &MI, LostDebugLocObserver &LocObserver) const override
Called for instructions with the Custom LegalizationAction.
bool legalizeIntrinsic(LegalizerHelper &Helper, MachineInstr &MI) const override
SPIRVGlobalRegistry * getSPIRVGlobalRegistry() const
void push_back(const T &Elt)
This is a 'vector' (really, a variable-sized array), optimized for the case when the array is small.
static constexpr TypeSize getFixed(ScalarTy ExactSize)
The instances of the Type class are immutable: once they are created, they are never changed.
static LLVM_ABI VectorType * get(Type *ElementType, ElementCount EC)
This static method is the primary way to construct an VectorType.
constexpr std::underlying_type_t< E > Mask()
Get a bitmask with 1s in all places up to the high-order bit of E's largest value.
LLVM_ABI LegalityPredicate isScalar(unsigned TypeIdx)
True iff the specified type index is a scalar.
LLVM_ABI LegalityPredicate numElementsNotPow2(unsigned TypeIdx)
True iff the specified type index is a vector whose element count is not a power of 2.
LLVM_ABI LegalityPredicate vectorElementCountIsLessThanOrEqualTo(unsigned TypeIdx, unsigned Size)
True iff the specified type index is a vector with a number of elements that's less than or equal to ...
LLVM_ABI LegalityPredicate typeInSet(unsigned TypeIdx, std::initializer_list< LLT > TypesInit)
True iff the given type index is one of the specified types.
LLVM_ABI LegalityPredicate vectorElementCountIsGreaterThan(unsigned TypeIdx, unsigned Size)
True iff the specified type index is a vector with a number of elements that's greater than the given...
Predicate any(Predicate P0, Predicate P1)
True iff P0 or P1 are true.
LegalityPredicate typeIsNot(unsigned TypeIdx, LLT Type)
True iff the given type index is not the specified type.
Predicate all(Predicate P0, Predicate P1)
True iff P0 and P1 are true.
LLVM_ABI LegalityPredicate typeIs(unsigned TypeIdx, LLT TypesInit)
True iff the given type index is the specified type.
LLVM_ABI LegalizeMutation changeElementCountTo(unsigned TypeIdx, unsigned FromTypeIdx)
Keep the same scalar or element type as TypeIdx, but take the number of elements from FromTypeIdx.
LLVM_ABI LegalizeMutation changeElementSizeTo(unsigned TypeIdx, unsigned FromTypeIdx)
Change the scalar size or element size to have the same scalar size as type index FromIndex.
Invariant opcodes: All instruction sets have these as their low opcodes.
This is an optimization pass for GlobalISel generic memory operations.
LLVM_ABI const llvm::fltSemantics & getFltSemanticForLLT(LLT Ty)
Get the appropriate floating point arithmetic semantic based on the bit size of the given scalar LLT.
std::function< bool(const LegalityQuery &)> LegalityPredicate
MachineInstr * getImm(const MachineOperand &MO, const MachineRegisterInfo *MRI)
constexpr bool isPowerOf2_32(uint32_t Value)
Return true if the argument is a power of two > 0.
FPClassTest
Floating-point class tests, supported by 'is_fpclass' intrinsic.
LLVM_ABI raw_ostream & dbgs()
dbgs() - This returns a reference to a raw_ostream for debugging messages.
const std::set< unsigned > & getTypeFoldingSupportedOpcodes()
int64_t foldImm(const MachineOperand &MO, const MachineRegisterInfo *MRI)
decltype(auto) cast(const From &Val)
cast<X> - Return the argument parameter cast to the specified type.
Align commonAlignment(Align A, uint64_t Offset)
Returns the alignment that satisfies both alignments.
This struct is a compact representation of a valid (non-zero power of two) alignment.
The LegalityQuery object bundles together all the information that's needed to decide whether a given...
This class contains a discriminated union of information about pointers in memory operands,...
MachinePointerInfo getWithOffset(int64_t O) const