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 allFloatScalarsAndF16Vector2AndVector4s = {s16, s32, s64, v2s16, v4s16};
153 auto allFloatScalars = {s16, s32, s64};
155 auto allFloatScalarsAndVectors = {
156 s16, s32, s64, v2s16, v2s32, v2s64, v3s16, v3s32, v3s64,
157 v4s16, v4s32, v4s64, v8s16, v8s32, v8s64, v16s16, v16s32, v16s64};
159 auto allShaderFloatVectors = {v2s16, v2s32, v2s64, v3s16, v3s32,
160 v3s64, v4s16, v4s32, v4s64};
162 auto allFloatVectors = {v2s16, v2s32, v2s64, v3s16, v3s32,
163 v3s64, v4s16, v4s32, v4s64, v8s16,
164 v8s32, v8s64, v16s16, v16s32, v16s64};
166 auto &allowedFloatVectorTypes =
167 ST.isShader() ? allShaderFloatVectors : allFloatVectors;
169 auto allFloatAndIntScalarsAndPtrs = {s8, s16, s32, s64, p0, p1,
170 p2, p3, p4, p5, p6, p7,
171 p8, p9, p10, p11, p12, p13};
173 auto allPtrs = {p0, p1, p2, p3, p4, p5, p6, p7, p8, p9, p10, p11, p12, p13};
175 auto &allowedVectorTypes = ST.isShader() ? allShaderVectors : allVectors;
177 bool HasArbitraryPrecisionInts = ST.canUseExtension(
178 SPIRV::Extension::SPV_ALTERA_arbitrary_precision_integers);
179 bool IsExtendedInts =
180 HasArbitraryPrecisionInts ||
181 ST.canUseExtension(SPIRV::Extension::SPV_KHR_bit_instructions) ||
182 ST.canUseExtension(SPIRV::Extension::SPV_INTEL_int4);
183 bool IsLongVecs = ST.canUseExtension(SPIRV::Extension::SPV_EXT_long_vector);
184 auto ExtendedIntScalarsAndVectors =
186 const LLT Ty = Query.Types[0];
187 return IsExtendedInts && Ty.isValid() &&
188 !Ty.isPointerOrPointerVector() && Ty.getScalarSizeInBits() > 1;
190 auto ExtendedScalarsAndVectorsProduct = [IsExtendedInts](
192 const LLT Ty1 = Query.Types[0], Ty2 = Query.Types[1];
193 return IsExtendedInts && Ty1.
isValid() && Ty2.isValid() &&
196 auto ExtendedPtrsScalarsAndVectors =
198 const LLT Ty = Query.Types[0];
199 return IsExtendedInts && Ty.isValid();
208 uint32_t MaxVectorSize = ST.isShader() ? 4 : 16;
213 case G_EXTRACT_VECTOR_ELT:
235 .customFor(allScalars)
245 .legalFor(allScalars)
322 {G_VECREDUCE_SMIN, G_VECREDUCE_SMAX, G_VECREDUCE_UMIN, G_VECREDUCE_UMAX,
323 G_VECREDUCE_ADD, G_VECREDUCE_MUL, G_VECREDUCE_FMUL, G_VECREDUCE_FMIN,
324 G_VECREDUCE_FMAX, G_VECREDUCE_FMINIMUM, G_VECREDUCE_FMAXIMUM,
325 G_VECREDUCE_OR, G_VECREDUCE_AND, G_VECREDUCE_XOR})
326 .legalFor(allowedVectorTypes)
347 .unsupportedIf(
typeIs(0, p9))
356 .unsupportedIf(
typeIs(1, p9))
364 G_BITREVERSE, G_SADDSAT, G_UADDSAT, G_SSUBSAT,
365 G_USUBSAT, G_SCMP, G_UCMP})
366 .legalFor(allIntScalarsAndVectors)
367 .
legalIf(ExtendedIntScalarsAndVectors)
375 .legalForCartesianProduct(allFloatScalarsAndVectors, allIntScalars);
378 .legalForCartesianProduct(allIntScalarsAndVectors,
379 allFloatScalarsAndVectors);
382 .legalForCartesianProduct(allIntScalarsAndVectors,
383 allFloatScalarsAndVectors);
386 .legalForCartesianProduct(allFloatScalarsAndVectors,
387 allScalarsAndVectors);
391 .
legalIf(ExtendedScalarsAndVectorsProduct)
395 .legalForCartesianProduct(allowedScalarsAndVectors)
396 .
legalIf(ExtendedScalarsAndVectorsProduct)
417 .
legalIf(ExtendedPtrsScalarsAndVectors)
422 typeInSet(1, allPtrsScalarsAndVectors)));
425 .legalFor({s1, s128})
426 .legalFor(allFloatAndIntScalarsAndPtrs)
429 return Query.
Types[0].isPointerVector();
447 !SrcTy.isPointer() &&
457 return SrcTy.isPointerVector() && DstTy.
isVector() &&
489 return IsExtendedInts && Ty.isValid() && !Ty.isPointerOrPointerVector();
492 typeInSet(1, allPtrsScalarsAndVectors)));
499 typeInSet(1, allFloatScalarsAndVectors)));
502 G_ATOMICRMW_MAX, G_ATOMICRMW_MIN,
503 G_ATOMICRMW_SUB, G_ATOMICRMW_XOR,
504 G_ATOMICRMW_UMAX, G_ATOMICRMW_UMIN})
505 .legalForCartesianProduct(allIntScalars, allPtrs);
508 {G_ATOMICRMW_FADD, G_ATOMICRMW_FSUB, G_ATOMICRMW_FMIN, G_ATOMICRMW_FMAX})
509 .legalForCartesianProduct(allFloatScalarsAndF16Vector2AndVector4s,
528 if (!HasArbitraryPrecisionInts)
533 .legalForCartesianProduct(allFloatScalarsAndVectors,
534 allIntScalarsAndVectors);
538 .legalForCartesianProduct(allFloatScalarsAndVectors);
550 allFloatScalarsAndVectors, {s32, v2s32, v3s32, v4s32, v8s32, v16s32});
589 G_INTRINSIC_ROUNDEVEN})
590 .legalFor(allFloatScalars)
600 allFloatScalarsAndVectors);
603 allFloatScalarsAndVectors, allIntScalarsAndVectors);
605 if (ST.canUseExtInstSet(SPIRV::InstructionSet::OpenCL_std)) {
607 {G_CTTZ, G_CTTZ_ZERO_POISON, G_CTLZ, G_CTLZ_ZERO_POISON})
608 .legalForCartesianProduct(allIntScalarsAndVectors,
609 allIntScalarsAndVectors);
618 G_INTRINSIC_CONVERGENT_W_SIDE_EFFECTS})
623 verify(*ST.getInstrInfo());
637 MI.eraseFromParent();
653 MI.eraseFromParent();
671 if (!Ty.isVector() ||
672 ST.canUseExtension(SPIRV::Extension::SPV_EXT_long_vector))
674 unsigned NumElements = Ty.getNumElements();
675 unsigned MaxVectorSize = ST.isShader() ? 4 : 16;
677 NumElements > MaxVectorSize;
702 for (
unsigned i = 0; i < NumElts; ++i) {
714 if (!
MI.memoperands_empty()) {
722 MIRBuilder.
buildLoad(EltReg, EltPtr, EltPtrInfo, EltAlign);
727 MI.eraseFromParent();
745 for (
unsigned i = 0; i < NumElts; ++i)
753 for (
unsigned i = 0; i < NumElts; ++i) {
765 if (!
MI.memoperands_empty()) {
772 MIRBuilder.
buildStore(SplitRegs[i], EltPtr, EltPtrInfo, EltAlign);
775 MI.eraseFromParent();
783 switch (
MI.getOpcode()) {
787 case TargetOpcode::G_BITCAST:
788 return legalizeBitcast(Helper,
MI);
789 case TargetOpcode::G_EXTRACT_VECTOR_ELT:
791 case TargetOpcode::G_INSERT_VECTOR_ELT:
793 case TargetOpcode::G_INTRINSIC:
794 case TargetOpcode::G_INTRINSIC_W_SIDE_EFFECTS:
796 case TargetOpcode::G_IS_FPCLASS:
797 return legalizeIsFPClass(Helper,
MI, LocObserver);
798 case TargetOpcode::G_ICMP: {
799 auto &Op0 =
MI.getOperand(2);
800 auto &Op1 =
MI.getOperand(3);
805 if ((!ST->canDirectlyComparePointers() ||
810 ST->getPointerSize());
812 LLVMTy, Helper.
MIRBuilder, SPIRV::AccessQualifier::ReadWrite,
true);
818 case TargetOpcode::G_LOAD:
820 case TargetOpcode::G_STORE:
840 const Type *LLVMArrTy =
843 LLVMArrTy, MIRBuilder, SPIRV::AccessQualifier::ReadWrite,
true);
845 ArrSpvTy, MIRBuilder, SPIRV::StorageClass::Function);
847 Register StackReg = StackTemp.getReg(0);
872 MI.eraseFromParent();
892 if (
getImm(IdxOperand, &MRI)) {
894 if (IdxVal < SrcTy.getNumElements()) {
899 for (
unsigned I = 0,
E = SrcTy.getNumElements();
I <
E; ++
I) {
906 Regs[IdxVal] = ValReg;
908 MI.eraseFromParent();
919 MIRBuilder.
buildStore(SrcReg, StackTemp, PtrInfo, VecAlign);
928 .
addUse(StackTemp.getReg(0))
934 MIRBuilder.
buildStore(ValReg, EltPtr, EltPtrInfo, EltAlign);
936 MIRBuilder.
buildLoad(DstReg, StackTemp, PtrInfo, VecAlign);
937 MI.eraseFromParent();
956 if (
getImm(IdxOperand, &MRI)) {
958 if (IdxVal < SrcTy.getNumElements()) {
962 for (
unsigned I = 0,
E = SrcTy.getNumElements();
I <
E; ++
I) {
973 MI.eraseFromParent();
984 MIRBuilder.
buildStore(SrcReg, StackTemp, PtrInfo, VecAlign);
993 .
addUse(StackTemp.getReg(0))
999 MIRBuilder.
buildLoad(DstReg, EltPtr, EltPtrInfo, EltAlign);
1001 MI.eraseFromParent();
1020 if (
MI.getNumOperands() == 2) {
1030 for (
unsigned i = 2; i <
MI.getNumOperands(); ++i) {
1035 MI.eraseFromParent();
1043 switch (IntrinsicID) {
1044 case Intrinsic::spv_bitcast:
1046 case Intrinsic::spv_insertelt:
1048 case Intrinsic::spv_extractelt:
1050 case Intrinsic::spv_const_composite:
1067 MI.eraseFromParent();
1074bool SPIRVLegalizerInfo::legalizeIsFPClass(
1077 auto [DstReg, DstTy, SrcReg, SrcTy] =
MI.getFirst2RegLLTs();
1081 auto &MF = MIRBuilder.
getMF();
1086 if (DstTy.isVector())
1088 SPIRVTypeInst SPIRVDstTy = GR->getOrCreateSPIRVType(
1089 LLVMDstTy, MIRBuilder, SPIRV::AccessQualifier::ReadWrite,
1092 unsigned BitSize = SrcTy.getScalarSizeInBits();
1097 if (SrcTy.isVector()) {
1098 IntTy =
LLT::vector(SrcTy.getElementCount(), IntTy);
1101 SPIRVTypeInst SPIRVIntTy = GR->getOrCreateSPIRVType(
1102 LLVMIntTy, MIRBuilder, SPIRV::AccessQualifier::ReadWrite,
1106 LLT DstTyCopy = DstTy;
1107 const auto assignSPIRVTy = [&](MachineInstrBuilder &&
MI) {
1112 assert((MITy == IntTy || MITy == DstTyCopy) &&
1113 "Unexpected LLT type while lowering G_IS_FPCLASS");
1114 SPIRVTypeInst SPVTy = MITy == IntTy ? SPIRVIntTy : SPIRVDstTy;
1115 GR->assignSPIRVTypeToVReg(SPVTy,
MI.getReg(0), MF);
1120 const auto buildSPIRVConstant = [&](LLT Ty,
auto &&
C) -> MachineInstrBuilder {
1124 assert((Ty == IntTy || Ty == DstTyCopy) &&
1125 "Unexpected LLT type while lowering constant for G_IS_FPCLASS");
1126 SPIRVTypeInst VecEltTy = GR->getOrCreateSPIRVType(
1127 (Ty == IntTy ? LLVMIntTy : LLVMDstTy)->getScalarType(), MIRBuilder,
1128 SPIRV::AccessQualifier::ReadWrite,
1130 GR->assignSPIRVTypeToVReg(VecEltTy, ScalarC.getReg(0), MF);
1135 MIRBuilder.
buildCopy(DstReg, buildSPIRVConstant(DstTy, 0));
1136 MI.eraseFromParent();
1140 MIRBuilder.
buildCopy(DstReg, buildSPIRVConstant(DstTy, 1));
1141 MI.eraseFromParent();
1150 MRI.
setRegClass(ResVReg, GR->getRegClass(SPIRVIntTy));
1151 GR->assignSPIRVTypeToVReg(SPIRVIntTy, ResVReg, Helper.
MIRBuilder.
getMF());
1152 auto AsInt = MIRBuilder.
buildInstr(SPIRV::OpBitcast)
1154 .
addUse(GR->getSPIRVTypeID(SPIRVIntTy))
1156 AsInt = assignSPIRVTy(std::move(AsInt));
1162 APInt ExpMask = Inf;
1168 auto SignBitC = buildSPIRVConstant(IntTy, SignBit);
1169 auto ValueMaskC = buildSPIRVConstant(IntTy, ValueMask);
1170 auto InfC = buildSPIRVConstant(IntTy, Inf);
1171 auto ExpMaskC = buildSPIRVConstant(IntTy, ExpMask);
1172 auto ZeroC = buildSPIRVConstant(IntTy, 0);
1174 auto Abs = assignSPIRVTy(MIRBuilder.
buildAnd(IntTy, AsInt, ValueMaskC));
1175 auto Sign = assignSPIRVTy(
1178 auto Res = buildSPIRVConstant(DstTy, 0);
1180 const auto appendToRes = [&](MachineInstrBuilder &&ToAppend) {
1181 Res = assignSPIRVTy(
1182 MIRBuilder.
buildOr(DstTyCopy, Res, assignSPIRVTy(std::move(ToAppend))));
1195 Mask &= ~fcPosFinite;
1199 DstTy, Abs, ExpMaskC));
1200 appendToRes(MIRBuilder.
buildAnd(DstTy, Cmp, Sign));
1201 Mask &= ~fcNegFinite;
1209 auto ExpBits = assignSPIRVTy(MIRBuilder.
buildAnd(IntTy, AsInt, ExpMaskC));
1212 Mask &= ~PartialCheck;
1221 else if (PartialCheck ==
fcZero)
1233 auto OneC = buildSPIRVConstant(IntTy, 1);
1234 auto VMinusOne = MIRBuilder.
buildSub(IntTy, V, OneC);
1235 auto SubnormalRes = assignSPIRVTy(
1237 buildSPIRVConstant(IntTy, AllOneMantissa)));
1239 SubnormalRes = MIRBuilder.
buildAnd(DstTy, SubnormalRes, Sign);
1240 appendToRes(std::move(SubnormalRes));
1247 else if (PartialCheck ==
fcInf)
1252 auto NegInfC = buildSPIRVConstant(IntTy, NegInf);
1259 auto InfWithQnanBitC =
1260 buildSPIRVConstant(IntTy, std::move(Inf) | QNaNBitMask);
1261 if (PartialCheck ==
fcNan) {
1265 }
else if (PartialCheck ==
fcQNan) {
1272 auto IsNan = assignSPIRVTy(
1274 auto IsNotQnan = assignSPIRVTy(MIRBuilder.
buildICmp(
1276 appendToRes(MIRBuilder.
buildAnd(DstTy, IsNan, IsNotQnan));
1283 APInt ExpLSB = ExpMask & ~(ExpMask.
shl(1));
1284 auto ExpMinusOne = assignSPIRVTy(
1285 MIRBuilder.
buildSub(IntTy, Abs, buildSPIRVConstant(IntTy, ExpLSB)));
1286 APInt MaxExpMinusOne = std::move(ExpMask) - ExpLSB;
1287 auto NormalRes = assignSPIRVTy(
1289 buildSPIRVConstant(IntTy, MaxExpMinusOne)));
1291 NormalRes = MIRBuilder.
buildAnd(DstTy, NormalRes, Sign);
1293 auto PosSign = assignSPIRVTy(MIRBuilder.
buildXor(
1294 DstTy, Sign, buildSPIRVConstant(DstTy, InversionMask)));
1295 NormalRes = MIRBuilder.
buildAnd(DstTy, NormalRes, PosSign);
1297 appendToRes(std::move(NormalRes));
1301 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