LLVM 24.0.0git
SPIRVLegalizerInfo.cpp
Go to the documentation of this file.
1//===- SPIRVLegalizerInfo.cpp --- SPIR-V Legalization Rules ------*- C++ -*-==//
2//
3// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
4// See https://llvm.org/LICENSE.txt for license information.
5// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
6//
7//===----------------------------------------------------------------------===//
8//
9// This file implements the targeting of the Machinelegalizer class for SPIR-V.
10//
11//===----------------------------------------------------------------------===//
12
13#include "SPIRVLegalizerInfo.h"
14#include "SPIRV.h"
15#include "SPIRVGlobalRegistry.h"
16#include "SPIRVSubtarget.h"
17#include "SPIRVUtils.h"
24#include "llvm/IR/IntrinsicsSPIRV.h"
25#include "llvm/Support/Debug.h"
27
28using namespace llvm;
29using namespace llvm::LegalizeActions;
30using namespace llvm::LegalityPredicates;
31
32#define DEBUG_TYPE "spirv-legalizer"
33
34LegalityPredicate typeOfExtendedScalars(unsigned TypeIdx, bool IsExtendedInts) {
35 return [IsExtendedInts, TypeIdx](const LegalityQuery &Query) {
36 const LLT Ty = Query.Types[TypeIdx];
37 return IsExtendedInts && Ty.isValid() && Ty.isScalar();
38 };
39}
40
41LegalityPredicate typeOfLongVectors(unsigned TypeIdx, bool IsLongVecs) {
42 return [TypeIdx, IsLongVecs](const LegalityQuery &Query) {
43 const LLT Ty = Query.Types[TypeIdx];
44 return IsLongVecs && Ty.isValid() && Ty.isVector();
45 };
46}
47
49 using namespace TargetOpcode;
50
51 this->ST = &ST;
52 GR = ST.getSPIRVGlobalRegistry();
53
54 const LLT s1 = LLT::scalar(1);
55 const LLT s8 = LLT::scalar(8);
56 const LLT s16 = LLT::scalar(16);
57 const LLT s32 = LLT::scalar(32);
58 const LLT s64 = LLT::scalar(64);
59 const LLT s128 = LLT::scalar(128);
60
61 const LLT v16s64 = LLT::fixed_vector(16, 64);
62 const LLT v16s32 = LLT::fixed_vector(16, 32);
63 const LLT v16s16 = LLT::fixed_vector(16, 16);
64 const LLT v16s8 = LLT::fixed_vector(16, 8);
65 const LLT v16s1 = LLT::fixed_vector(16, 1);
66
67 const LLT v8s64 = LLT::fixed_vector(8, 64);
68 const LLT v8s32 = LLT::fixed_vector(8, 32);
69 const LLT v8s16 = LLT::fixed_vector(8, 16);
70 const LLT v8s8 = LLT::fixed_vector(8, 8);
71 const LLT v8s1 = LLT::fixed_vector(8, 1);
72
73 const LLT v4s64 = LLT::fixed_vector(4, 64);
74 const LLT v4s32 = LLT::fixed_vector(4, 32);
75 const LLT v4s16 = LLT::fixed_vector(4, 16);
76 const LLT v4s8 = LLT::fixed_vector(4, 8);
77 const LLT v4s1 = LLT::fixed_vector(4, 1);
78
79 const LLT v3s64 = LLT::fixed_vector(3, 64);
80 const LLT v3s32 = LLT::fixed_vector(3, 32);
81 const LLT v3s16 = LLT::fixed_vector(3, 16);
82 const LLT v3s8 = LLT::fixed_vector(3, 8);
83 const LLT v3s1 = LLT::fixed_vector(3, 1);
84
85 const LLT v2s64 = LLT::fixed_vector(2, 64);
86 const LLT v2s32 = LLT::fixed_vector(2, 32);
87 const LLT v2s16 = LLT::fixed_vector(2, 16);
88 const LLT v2s8 = LLT::fixed_vector(2, 8);
89 const LLT v2s1 = LLT::fixed_vector(2, 1);
90
91 const unsigned PSize = ST.getPointerSize();
92 const LLT p0 = LLT::pointer(0, PSize); // Function
93 const LLT p1 = LLT::pointer(1, PSize); // CrossWorkgroup
94 const LLT p2 = LLT::pointer(2, PSize); // UniformConstant
95 const LLT p3 = LLT::pointer(3, PSize); // Workgroup
96 const LLT p4 = LLT::pointer(4, PSize); // Generic
97 const LLT p5 =
98 LLT::pointer(5, PSize); // Input, SPV_INTEL_usm_storage_classes (Device)
99 const LLT p6 = LLT::pointer(6, PSize); // SPV_INTEL_usm_storage_classes (Host)
100 const LLT p7 = LLT::pointer(7, PSize); // Input
101 const LLT p8 = LLT::pointer(8, PSize); // Output
102 const LLT p9 =
103 LLT::pointer(9, PSize); // CodeSectionINTEL, SPV_INTEL_function_pointers
104 const LLT p10 = LLT::pointer(10, PSize); // Private
105 const LLT p11 = LLT::pointer(11, PSize); // StorageBuffer
106 const LLT p12 = LLT::pointer(12, PSize); // Uniform
107 const LLT p13 = LLT::pointer(13, PSize); // PushConstant
108
109 // TODO: remove copy-pasting here by using concatenation in some way.
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};
116
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};
121
122 auto allShaderVectors = {v2s1, v2s8, v2s16, v2s32, v2s64,
123 v3s1, v3s8, v3s16, v3s32, v3s64,
124 v4s1, v4s8, v4s16, v4s32, v4s64};
125
126 auto allScalars = {s1, s8, s16, s32, s64};
127
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};
133
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};
137
138 auto &allowedScalarsAndVectors =
139 ST.isShader() ? allShaderScalarsAndVectors : allScalarsAndVectors;
140
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};
145
146 auto allBoolScalarsAndVectors = {s1, v2s1, v3s1, v4s1, v8s1, v16s1};
147 auto allBoolVectors = {v2s1, v3s1, v4s1, v8s1, v16s1};
148
149 auto allIntScalars = {s8, s16, s32, s64, s128};
150
151 auto allShaderIntVectors = {v2s8, v2s16, v2s32, v2s64, v3s8, v3s16,
152 v3s32, v3s64, v4s8, v4s16, v4s32, v4s64};
153
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};
157
158 auto &allowedIntVectorTypes =
159 ST.isShader() ? allShaderIntVectors : allIntVectors;
160
161 auto allFloatScalarsAndF16Vector2AndVector4s = {s16, s32, s64, v2s16, v4s16};
162
163 auto allFloatScalars = {s16, s32, s64};
164
165 auto allFloatScalarsAndVectors = {
166 s16, s32, s64, v2s16, v2s32, v2s64, v3s16, v3s32, v3s64,
167 v4s16, v4s32, v4s64, v8s16, v8s32, v8s64, v16s16, v16s32, v16s64};
168
169 auto allShaderFloatVectors = {v2s16, v2s32, v2s64, v3s16, v3s32,
170 v3s64, v4s16, v4s32, v4s64};
171
172 auto allFloatVectors = {v2s16, v2s32, v2s64, v3s16, v3s32,
173 v3s64, v4s16, v4s32, v4s64, v8s16,
174 v8s32, v8s64, v16s16, v16s32, v16s64};
175
176 auto &allowedFloatVectorTypes =
177 ST.isShader() ? allShaderFloatVectors : allFloatVectors;
178
179 auto allFloatAndIntScalarsAndPtrs = {s8, s16, s32, s64, p0, p1,
180 p2, p3, p4, p5, p6, p7,
181 p8, p9, p10, p11, p12, p13};
182
183 auto allPtrs = {p0, p1, p2, p3, p4, p5, p6, p7, p8, p9, p10, p11, p12, p13};
184
185 auto &allowedVectorTypes = ST.isShader() ? allShaderVectors : allVectors;
186
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 =
195 [IsExtendedInts](const LegalityQuery &Query) {
196 const LLT Ty = Query.Types[0];
197 return IsExtendedInts && Ty.isValid() &&
198 !Ty.isPointerOrPointerVector() && Ty.getScalarSizeInBits() > 1;
199 };
200 auto ExtendedScalarsAndVectorsProduct = [IsExtendedInts](
201 const LegalityQuery &Query) {
202 const LLT Ty1 = Query.Types[0], Ty2 = Query.Types[1];
203 return IsExtendedInts && Ty1.isValid() && Ty2.isValid() &&
204 !Ty1.isPointerOrPointerVector() && !Ty2.isPointerOrPointerVector();
205 };
206 auto ExtendedPtrsScalarsAndVectors =
207 [IsExtendedInts](const LegalityQuery &Query) {
208 const LLT Ty = Query.Types[0];
209 return IsExtendedInts && Ty.isValid();
210 };
211
212 // The universal validation rules in the SPIR-V specification state that
213 // vector sizes are typically limited to 2, 3, or 4. However, larger vector
214 // sizes (8 and 16) are enabled when the Kernel capability is present. For
215 // shader execution models, vector sizes are strictly limited to 4. In
216 // non-shader contexts, vector sizes of 8 and 16 are also permitted, but
217 // arbitrary sizes (e.g., 6 or 11) are not.
218 uint32_t MaxVectorSize = ST.isShader() ? 4 : 16;
219 LLVM_DEBUG(dbgs() << "MaxVectorSize: " << MaxVectorSize << "\n");
220
221 for (auto Opc : getTypeFoldingSupportedOpcodes()) {
222 switch (Opc) {
223 case G_EXTRACT_VECTOR_ELT:
224 case G_UREM:
225 case G_SREM:
226 case G_UDIV:
227 case G_SDIV:
228 case G_FREM:
229 break;
230 default:
232 .customFor(allScalars)
233 .customFor(allowedVectorTypes)
234 .customIf(typeOfLongVectors(0, IsLongVecs))
238 0, ElementCount::getFixed(MaxVectorSize)))
239 .custom();
240 break;
241 }
242 }
243
244 getActionDefinitionsBuilder({G_UREM, G_SREM, G_SDIV, G_UDIV, G_FREM})
245 .customFor(allScalars)
246 .customFor(allowedVectorTypes)
247 .customIf(typeOfLongVectors(0, IsLongVecs))
251 0, ElementCount::getFixed(MaxVectorSize)))
252 .custom();
253
254 getActionDefinitionsBuilder({G_FMA, G_STRICT_FMA})
255 .legalFor(allScalars)
256 .legalFor(allowedVectorTypes)
257 .legalIf(typeOfLongVectors(0, IsLongVecs))
261 0, ElementCount::getFixed(MaxVectorSize)))
262 .alwaysLegal();
263
264 getActionDefinitionsBuilder(G_INTRINSIC_W_SIDE_EFFECTS).custom();
265
266 getActionDefinitionsBuilder(G_SHUFFLE_VECTOR)
267 .legalForCartesianProduct(allowedVectorTypes, allowedVectorTypes)
268 .legalIf(typeOfLongVectors(0, IsLongVecs))
269 .legalIf(typeOfLongVectors(1, IsLongVecs))
271 .lowerIf(vectorElementCountIsGreaterThan(0, MaxVectorSize))
273 .lowerIf(vectorElementCountIsGreaterThan(1, MaxVectorSize));
274
275 getActionDefinitionsBuilder(G_EXTRACT_VECTOR_ELT)
276 .customIf(typeOfLongVectors(1, IsLongVecs))
280 1, ElementCount::getFixed(MaxVectorSize)))
281 .custom();
282
283 getActionDefinitionsBuilder(G_INSERT_VECTOR_ELT)
284 .customIf(typeOfLongVectors(0, IsLongVecs))
288 0, ElementCount::getFixed(MaxVectorSize)))
289 .custom();
290
291 // Illegal G_UNMERGE_VALUES instructions should be handled
292 // during the combine phase.
293 getActionDefinitionsBuilder(G_BUILD_VECTOR)
294 .legalIf(typeOfLongVectors(0, IsLongVecs))
298 0, ElementCount::getFixed(MaxVectorSize)));
299
300 // When entering the legalizer, there should be no G_BITCAST instructions.
301 // They should all be calls to the `spv_bitcast` intrinsic. The call to
302 // the intrinsic will be converted to a G_BITCAST during legalization if
303 // the vectors are not legal. After using the rules to legalize a G_BITCAST,
304 // we turn it back into a call to the intrinsic with a custom rule to avoid
305 // potential machine verifier failures.
307 .customIf(typeOfLongVectors(0, IsLongVecs))
312 0, ElementCount::getFixed(MaxVectorSize)))
313 .lowerIf(vectorElementCountIsGreaterThan(1, MaxVectorSize))
314 .custom();
315
316 // If the result is still illegal, the combiner should be able to remove it.
317 getActionDefinitionsBuilder(G_CONCAT_VECTORS)
318 .legalForCartesianProduct(allowedVectorTypes, allowedVectorTypes)
320 typeOfLongVectors(1, IsLongVecs)));
321
322 getActionDefinitionsBuilder(G_SPLAT_VECTOR)
323 .legalFor(allowedVectorTypes)
324 .legalIf(typeOfLongVectors(0, IsLongVecs))
328 .alwaysLegal();
329
330 // Vector Reduction Operations
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)
337 .legalIf(typeOfLongVectors(0, IsLongVecs))
338 .scalarize(1)
339 .lower();
340
341 getActionDefinitionsBuilder({G_VECREDUCE_SEQ_FADD, G_VECREDUCE_SEQ_FMUL})
342 .scalarize(2)
343 .lower();
344
345 // Illegal G_UNMERGE_VALUES instructions should be handled
346 // during the combine phase.
347 getActionDefinitionsBuilder(G_UNMERGE_VALUES)
349 typeOfLongVectors(1, IsLongVecs)))
351
352 getActionDefinitionsBuilder({G_MEMCPY, G_MEMCPY_INLINE, G_MEMMOVE})
353 .unsupportedIf(LegalityPredicates::any(typeIs(0, p9), typeIs(1, p9)))
354 .legalIf(all(typeInSet(0, allPtrs), typeInSet(1, allPtrs)));
355
356 getActionDefinitionsBuilder({G_MEMSET, G_MEMSET_INLINE})
357 .unsupportedIf(typeIs(0, p9))
358 .legalIf(all(typeInSet(0, allPtrs), typeInSet(1, allIntScalars)));
359
360 getActionDefinitionsBuilder(G_ADDRSPACE_CAST)
361 .legalForCartesianProduct(allPtrs, allPtrs);
362
363 // Should we be legalizing bad scalar sizes like s5 here instead
364 // of handling them in the instruction selector?
365 getActionDefinitionsBuilder({G_LOAD, G_STORE})
366 .unsupportedIf(typeIs(1, p9))
367 .legalForCartesianProduct(allowedVectorTypes, allPtrs)
368 .legalForCartesianProduct(allPtrs, allPtrs)
369 .legalIf(isScalar(0))
370 .legalIf(typeOfLongVectors(0, IsLongVecs))
371 .custom();
372
373 getActionDefinitionsBuilder({G_SMIN, G_SMAX, G_UMIN, G_UMAX, G_ABS,
374 G_BITREVERSE, G_SADDSAT, G_UADDSAT, G_SSUBSAT,
375 G_USUBSAT, G_SCMP, G_UCMP})
376 .legalFor(allIntScalars)
377 .legalFor(allowedIntVectorTypes)
378 .legalIf(ExtendedIntScalarsAndVectors)
379 // LLVM i1 maps to OpTypeBool, not OpTypeInt.
380 .scalarizeIf(typeInSet(0, allBoolVectors), 0)
381 .minScalar(0, s32)
384 0, ElementCount::getFixed(MaxVectorSize)))
386
387 getActionDefinitionsBuilder({G_SSHLSAT, G_USHLSAT}).lower();
388
389 getActionDefinitionsBuilder({G_FLDEXP, G_STRICT_FLDEXP})
390 .legalForCartesianProduct(allFloatScalarsAndVectors, allIntScalars);
391
392 getActionDefinitionsBuilder({G_FPTOSI, G_FPTOUI})
393 .legalForCartesianProduct(allIntScalarsAndVectors,
394 allFloatScalarsAndVectors);
395
396 getActionDefinitionsBuilder({G_FPTOSI_SAT, G_FPTOUI_SAT})
397 .legalForCartesianProduct(allIntScalarsAndVectors,
398 allFloatScalarsAndVectors);
399
400 getActionDefinitionsBuilder({G_SITOFP, G_UITOFP})
401 .legalForCartesianProduct(allFloatScalarsAndVectors,
402 allScalarsAndVectors);
403
405 .legalForCartesianProduct(allIntScalarsAndVectors)
406 .legalIf(ExtendedScalarsAndVectorsProduct)
407 .legalIf(typeOfLongVectors(0, IsLongVecs));
408
409 getActionDefinitionsBuilder({G_TRUNC, G_ZEXT, G_SEXT, G_ANYEXT})
410 .legalForCartesianProduct(allowedScalarsAndVectors)
411 .legalIf(ExtendedScalarsAndVectorsProduct)
412 .legalIf(typeOfLongVectors(0, IsLongVecs))
416 0, ElementCount::getFixed(MaxVectorSize)));
417
418 getActionDefinitionsBuilder(G_SEXT_INREG)
419 .lowerIf(typeOfLongVectors(0, IsLongVecs))
423 0, ElementCount::getFixed(MaxVectorSize)))
424 .lower();
425
427 .legalIf(typeOfLongVectors(0, IsLongVecs))
430 0, ElementCount::getFixed(MaxVectorSize)))
431 .legalFor(allPtrsScalarsAndVectors)
432 .legalIf(ExtendedPtrsScalarsAndVectors)
434
436 all(typeInSet(0, allPtrsScalarsAndVectors),
437 typeInSet(1, allPtrsScalarsAndVectors)));
438
439 getActionDefinitionsBuilder({G_IMPLICIT_DEF, G_FREEZE})
440 .legalFor({s1, s128})
441 .legalFor(allFloatAndIntScalarsAndPtrs)
442 .legalFor(allowedVectorTypes)
443 .legalIf([](const LegalityQuery &Query) {
444 return Query.Types[0].isPointerVector();
445 })
446 .legalIf(typeOfLongVectors(0, IsLongVecs))
450 0, ElementCount::getFixed(MaxVectorSize)));
451
452 getActionDefinitionsBuilder({G_STACKSAVE, G_STACKRESTORE}).alwaysLegal();
453
455 .legalForCartesianProduct(allPtrs, allIntScalars)
456 .legalIf(
457 all(typeInSet(0, allPtrs), typeOfExtendedScalars(1, IsExtendedInts)))
458 .legalIf([](const LegalityQuery &Query) {
459 const LLT DstTy = Query.Types[0];
460 const LLT SrcTy = Query.Types[1];
461 return DstTy.isPointerVector() && SrcTy.isVector() &&
462 !SrcTy.isPointer() &&
463 DstTy.getNumElements() == SrcTy.getNumElements();
464 });
466 .legalForCartesianProduct(allIntScalars, allPtrs)
467 .legalIf(
468 all(typeOfExtendedScalars(0, IsExtendedInts), typeInSet(1, allPtrs)))
469 .legalIf([](const LegalityQuery &Query) {
470 const LLT DstTy = Query.Types[0];
471 const LLT SrcTy = Query.Types[1];
472 return SrcTy.isPointerVector() && DstTy.isVector() &&
473 !DstTy.isPointer() &&
474 DstTy.getNumElements() == SrcTy.getNumElements();
475 });
477 .legalForCartesianProduct(allPtrs, allIntScalars)
478 .legalIf(
479 all(typeInSet(0, allPtrs), typeOfExtendedScalars(1, IsExtendedInts)));
480
482 .legalForCartesianProduct(allPtrs, allIntScalars)
483 .legalIf(
484 all(typeInSet(0, allPtrs), typeOfExtendedScalars(1, IsExtendedInts)))
485 .legalIf([](const LegalityQuery &Query) {
486 const LLT PtrTy = Query.Types[0];
487 const LLT MaskTy = Query.Types[1];
488 return PtrTy.isPointerVector() && MaskTy.isVector() &&
489 !MaskTy.isPointer() &&
490 PtrTy.getNumElements() == MaskTy.getNumElements();
491 });
492
493 // ST.canDirectlyComparePointers() for pointer args is supported in
494 // legalizeCustom().
497 all(typeIs(0, p9), typeInSet(1, allPtrs), typeIsNot(1, p9)),
498 all(typeInSet(0, allPtrs), typeIsNot(0, p9), typeIs(1, p9))))
501 1, ElementCount::getFixed(MaxVectorSize)))
502 .legalIf([IsExtendedInts](const LegalityQuery &Query) {
503 const LLT Ty = Query.Types[1];
504 return IsExtendedInts && Ty.isValid() && !Ty.isPointerOrPointerVector();
505 })
506 .customIf(all(typeInSet(0, allBoolScalarsAndVectors),
507 typeInSet(1, allPtrsScalarsAndVectors)));
508
512 1, ElementCount::getFixed(MaxVectorSize)))
513 .legalIf(all(typeInSet(0, allBoolScalarsAndVectors),
514 typeInSet(1, allFloatScalarsAndVectors)));
515
516 getActionDefinitionsBuilder({G_ATOMICRMW_OR, G_ATOMICRMW_ADD, G_ATOMICRMW_AND,
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);
521
523 {G_ATOMICRMW_FADD, G_ATOMICRMW_FSUB, G_ATOMICRMW_FMIN, G_ATOMICRMW_FMAX})
524 .legalForCartesianProduct(allFloatScalarsAndF16Vector2AndVector4s,
525 allPtrs);
526
527 getActionDefinitionsBuilder(G_ATOMICRMW_XCHG)
528 .legalForCartesianProduct(allFloatAndIntScalarsAndPtrs, allPtrs);
529
530 getActionDefinitionsBuilder(G_ATOMIC_CMPXCHG_WITH_SUCCESS).lower();
531 // TODO: add proper legalization rules.
532 getActionDefinitionsBuilder(G_ATOMIC_CMPXCHG).alwaysLegal();
534
535 getActionDefinitionsBuilder({G_UADDO, G_USUBO, G_UMULO, G_SMULO})
536 .alwaysLegal();
537
538 getActionDefinitionsBuilder({G_SADDO, G_SSUBO}).lower();
539
540 // Lowering widens s64 to s128, which needs
541 // SPV_ALTERA_arbitrary_precision_integers. Mark s64 unsupported otherwise.
542 auto &MulFix = getActionDefinitionsBuilder({G_SMULFIX, G_UMULFIX});
543 if (!HasArbitraryPrecisionInts)
544 MulFix.unsupportedFor({s64});
545 MulFix.lower();
546
547 getActionDefinitionsBuilder({G_LROUND, G_LLROUND})
548 .legalForCartesianProduct(allFloatScalarsAndVectors,
549 allIntScalarsAndVectors);
550
551 // FP conversions.
552 getActionDefinitionsBuilder({G_FPTRUNC, G_FPEXT})
553 .legalForCartesianProduct(allFloatScalarsAndVectors);
554
555 // Pointer-handling.
556 getActionDefinitionsBuilder(G_FRAME_INDEX).legalFor({p0});
557
558 getActionDefinitionsBuilder(G_GLOBAL_VALUE).legalFor(allPtrs);
559
560 // Control-flow. In some cases (e.g. constants) s1 may be promoted to s32.
562 getActionDefinitionsBuilder(G_BRCOND).legalFor({s1, s32});
563
565 allFloatScalarsAndVectors, {s32, v2s32, v3s32, v4s32, v8s32, v16s32});
566
567 // TODO: Review the target OpenCL and GLSL Extended Instruction Set specs to
568 // tighten these requirements. Many of these math functions are only legal on
569 // specific bitwidths, so they are not selectable for
570 // allFloatScalarsAndVectors.
571 // clang-format off
572 getActionDefinitionsBuilder({G_STRICT_FSQRT,
573 G_FPOW,
574 G_FEXP,
575 G_FMODF,
576 G_FSINCOS,
577 G_FEXP2,
578 G_FEXP10,
579 G_FLOG,
580 G_FLOG2,
581 G_FLOG10,
582 G_FABS,
583 G_FMINNUM,
584 G_FMAXNUM,
585 G_FCEIL,
586 G_FCOS,
587 G_FSIN,
588 G_FTAN,
589 G_FACOS,
590 G_FASIN,
591 G_FATAN,
592 G_FATAN2,
593 G_FCOSH,
594 G_FSINH,
595 G_FTANH,
596 G_FSQRT,
597 G_FFLOOR,
598 G_FRINT,
599 G_FNEARBYINT,
600 G_INTRINSIC_ROUND,
601 G_INTRINSIC_TRUNC,
602 G_FMINIMUM,
603 G_FMAXIMUM,
604 G_INTRINSIC_ROUNDEVEN})
605 .legalFor(allFloatScalars)
606 .legalFor(allowedFloatVectorTypes)
609 0, ElementCount::getFixed(MaxVectorSize)))
611 // clang-format on
612
613 getActionDefinitionsBuilder(G_FCOPYSIGN)
614 .legalForCartesianProduct(allFloatScalarsAndVectors,
615 allFloatScalarsAndVectors);
616
618 allFloatScalarsAndVectors, allIntScalarsAndVectors);
619
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);
625
626 // Struct return types become a single scalar, so cannot easily legalize.
627 getActionDefinitionsBuilder({G_SMULH, G_UMULH}).alwaysLegal();
628 }
629
630 getActionDefinitionsBuilder(G_IS_FPCLASS).custom();
631
632 getActionDefinitionsBuilder({G_INTRINSIC, G_INTRINSIC_CONVERGENT,
633 G_INTRINSIC_CONVERGENT_W_SIDE_EFFECTS})
634 .alwaysLegal();
636 getActionDefinitionsBuilder({G_TRAP, G_DEBUGTRAP, G_UBSANTRAP}).alwaysLegal();
637
638 verify(*ST.getInstrInfo());
639}
640
642 MachineInstr &MI) {
643 MachineIRBuilder &MIRBuilder = Helper.MIRBuilder;
644 Register DstReg = MI.getOperand(0).getReg();
645 Register SrcReg = MI.getOperand(1).getReg();
646 Register IdxReg = MI.getOperand(2).getReg();
647
648 MIRBuilder
649 .buildIntrinsic(Intrinsic::spv_extractelt, ArrayRef<Register>{DstReg})
650 .addUse(SrcReg)
651 .addUse(IdxReg);
652 MI.eraseFromParent();
653 return true;
654}
655
657 MachineIRBuilder &MIRBuilder = Helper.MIRBuilder;
658 Register DstReg = MI.getOperand(0).getReg();
659 Register SrcReg = MI.getOperand(1).getReg();
660 Register ValReg = MI.getOperand(2).getReg();
661 Register IdxReg = MI.getOperand(3).getReg();
662
663 MIRBuilder
664 .buildIntrinsic(Intrinsic::spv_insertelt, ArrayRef<Register>{DstReg})
665 .addUse(SrcReg)
666 .addUse(ValReg)
667 .addUse(IdxReg);
668 MI.eraseFromParent();
669 return true;
670}
671
673 LegalizerHelper &Helper,
676 Register ConvReg = MRI.createGenericVirtualRegister(ConvTy);
677 MRI.setRegClass(ConvReg, GR->getRegClass(SpvType));
678 GR->assignSPIRVTypeToVReg(SpvType, ConvReg, Helper.MIRBuilder.getMF());
679 Helper.MIRBuilder.buildInstr(TargetOpcode::G_PTRTOINT)
680 .addDef(ConvReg)
681 .addUse(Reg);
682 return ConvReg;
683}
684
685static bool needsVectorLegalization(const LLT &Ty, const SPIRVSubtarget &ST) {
686 if (!Ty.isVector() ||
687 ST.canUseExtension(SPIRV::Extension::SPV_EXT_long_vector))
688 return false;
689 unsigned NumElements = Ty.getNumElements();
690 unsigned MaxVectorSize = ST.isShader() ? 4 : 16;
691 return (NumElements > 4 && !isPowerOf2_32(NumElements)) ||
692 NumElements > MaxVectorSize;
693}
694
697 MachineRegisterInfo &MRI = MI.getMF()->getRegInfo();
698 MachineIRBuilder &MIRBuilder = Helper.MIRBuilder;
699 Register DstReg = MI.getOperand(0).getReg();
700 Register PtrReg = MI.getOperand(1).getReg();
701 LLT DstTy = MRI.getType(DstReg);
702
703 if (!DstTy.isVector())
704 return true;
705
706 const SPIRVSubtarget &ST = MI.getMF()->getSubtarget<SPIRVSubtarget>();
707 if (!needsVectorLegalization(DstTy, ST))
708 return true;
709
710 SmallVector<Register, 8> SplitRegs;
711 LLT EltTy = DstTy.getElementType();
712 unsigned NumElts = DstTy.getNumElements();
713
714 LLT PtrTy = MRI.getType(PtrReg);
715 auto Zero = MIRBuilder.buildConstant(LLT::scalar(32), 0);
716
717 for (unsigned i = 0; i < NumElts; ++i) {
718 auto Idx = MIRBuilder.buildConstant(LLT::scalar(32), i);
719 Register EltPtr = MRI.createGenericVirtualRegister(PtrTy);
720
721 MIRBuilder.buildIntrinsic(Intrinsic::spv_gep, ArrayRef<Register>{EltPtr})
722 .addImm(1) // InBounds
723 .addUse(PtrReg)
724 .addUse(Zero.getReg(0))
725 .addUse(Idx.getReg(0));
726
727 MachinePointerInfo EltPtrInfo;
728 Align EltAlign = Align(1);
729 if (!MI.memoperands_empty()) {
730 MachineMemOperand *MMO = *MI.memoperands_begin();
731 EltPtrInfo =
732 MMO->getPointerInfo().getWithOffset(i * EltTy.getSizeInBytes());
733 EltAlign = commonAlignment(MMO->getAlign(), i * EltTy.getSizeInBytes());
734 }
735
736 Register EltReg = MRI.createGenericVirtualRegister(EltTy);
737 MIRBuilder.buildLoad(EltReg, EltPtr, EltPtrInfo, EltAlign);
738 SplitRegs.push_back(EltReg);
739 }
740
741 MIRBuilder.buildBuildVector(DstReg, SplitRegs);
742 MI.eraseFromParent();
743 return true;
744}
745
748 MachineRegisterInfo &MRI = MI.getMF()->getRegInfo();
749 MachineIRBuilder &MIRBuilder = Helper.MIRBuilder;
750 Register ValReg = MI.getOperand(0).getReg();
751 Register PtrReg = MI.getOperand(1).getReg();
752 LLT ValTy = MRI.getType(ValReg);
753
754 assert(ValTy.isVector() && "Expected vector store");
755
756 SmallVector<Register, 8> SplitRegs;
757 LLT EltTy = ValTy.getElementType();
758 unsigned NumElts = ValTy.getNumElements();
759
760 for (unsigned i = 0; i < NumElts; ++i)
761 SplitRegs.push_back(MRI.createGenericVirtualRegister(EltTy));
762
763 MIRBuilder.buildUnmerge(SplitRegs, ValReg);
764
765 LLT PtrTy = MRI.getType(PtrReg);
766 auto Zero = MIRBuilder.buildConstant(LLT::scalar(32), 0);
767
768 for (unsigned i = 0; i < NumElts; ++i) {
769 auto Idx = MIRBuilder.buildConstant(LLT::scalar(32), i);
770 Register EltPtr = MRI.createGenericVirtualRegister(PtrTy);
771
772 MIRBuilder.buildIntrinsic(Intrinsic::spv_gep, ArrayRef<Register>{EltPtr})
773 .addImm(1) // InBounds
774 .addUse(PtrReg)
775 .addUse(Zero.getReg(0))
776 .addUse(Idx.getReg(0));
777
778 MachinePointerInfo EltPtrInfo;
779 Align EltAlign = Align(1);
780 if (!MI.memoperands_empty()) {
781 MachineMemOperand *MMO = *MI.memoperands_begin();
782 EltPtrInfo =
783 MMO->getPointerInfo().getWithOffset(i * EltTy.getSizeInBytes());
784 EltAlign = commonAlignment(MMO->getAlign(), i * EltTy.getSizeInBytes());
785 }
786
787 MIRBuilder.buildStore(SplitRegs[i], EltPtr, EltPtrInfo, EltAlign);
788 }
789
790 MI.eraseFromParent();
791 return true;
792}
793
796 LostDebugLocObserver &LocObserver) const {
797 MachineRegisterInfo &MRI = MI.getMF()->getRegInfo();
798 switch (MI.getOpcode()) {
799 default:
800 // TODO: implement legalization for other opcodes.
801 return true;
802 case TargetOpcode::G_BITCAST:
803 return legalizeBitcast(Helper, MI);
804 case TargetOpcode::G_EXTRACT_VECTOR_ELT:
805 return legalizeExtractVectorElt(Helper, MI);
806 case TargetOpcode::G_INSERT_VECTOR_ELT:
807 return legalizeInsertVectorElt(Helper, MI);
808 case TargetOpcode::G_INTRINSIC:
809 case TargetOpcode::G_INTRINSIC_W_SIDE_EFFECTS:
810 return legalizeIntrinsic(Helper, MI);
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);
816 Register Reg0 = Op0.getReg();
817 Register Reg1 = Op1.getReg();
819 static_cast<CmpInst::Predicate>(MI.getOperand(1).getPredicate());
820 if ((!ST->canDirectlyComparePointers() ||
822 MRI.getType(Reg0).isPointer() && MRI.getType(Reg1).isPointer()) {
823 LLT ConvT = LLT::scalar(ST->getPointerSize());
824 Type *LLVMTy = IntegerType::get(MI.getMF()->getFunction().getContext(),
825 ST->getPointerSize());
826 SPIRVTypeInst SpirvTy = GR->getOrCreateSPIRVType(
827 LLVMTy, Helper.MIRBuilder, SPIRV::AccessQualifier::ReadWrite, true);
828 Op0.setReg(convertPtrToInt(Reg0, ConvT, SpirvTy, Helper, MRI, GR));
829 Op1.setReg(convertPtrToInt(Reg1, ConvT, SpirvTy, Helper, MRI, GR));
830 }
831 return true;
832 }
833 case TargetOpcode::G_LOAD:
834 return legalizeLoad(Helper, MI, GR);
835 case TargetOpcode::G_STORE:
836 return legalizeStore(Helper, MI, GR);
837 }
838}
839
842 Register SrcReg, LLT SrcTy,
843 MachinePointerInfo &PtrInfo, Align &VecAlign) {
844 MachineIRBuilder &MIRBuilder = Helper.MIRBuilder;
845 MachineRegisterInfo &MRI = *MIRBuilder.getMRI();
846
847 VecAlign = Helper.getStackTemporaryAlignment(SrcTy);
848 auto StackTemp = Helper.createStackTemporary(
849 TypeSize::getFixed(SrcTy.getSizeInBytes()), VecAlign, PtrInfo);
850
851 // Set the type of StackTemp to a pointer to an array of the element type.
852 SPIRVTypeInst SpvSrcTy = GR->getSPIRVTypeForVReg(SrcReg);
853 SPIRVTypeInst EltSpvTy = GR->getScalarOrVectorComponentType(SpvSrcTy);
854 const Type *LLVMEltTy = GR->getTypeForSPIRVType(EltSpvTy);
855 const Type *LLVMArrTy =
856 ArrayType::get(const_cast<Type *>(LLVMEltTy), SrcTy.getNumElements());
857 SPIRVTypeInst ArrSpvTy = GR->getOrCreateSPIRVType(
858 LLVMArrTy, MIRBuilder, SPIRV::AccessQualifier::ReadWrite, true);
859 SPIRVTypeInst PtrToArrSpvTy = GR->getOrCreateSPIRVPointerType(
860 ArrSpvTy, MIRBuilder, SPIRV::StorageClass::Function);
861
862 Register StackReg = StackTemp.getReg(0);
863 MRI.setRegClass(StackReg, GR->getRegClass(PtrToArrSpvTy));
864 GR->assignSPIRVTypeToVReg(PtrToArrSpvTy, StackReg, MIRBuilder.getMF());
865
866 return StackTemp;
867}
868
871 LLVM_DEBUG(dbgs() << "Found a bitcast instruction\n");
872 MachineIRBuilder &MIRBuilder = Helper.MIRBuilder;
873 MachineRegisterInfo &MRI = *MIRBuilder.getMRI();
874 const SPIRVSubtarget &ST = MI.getMF()->getSubtarget<SPIRVSubtarget>();
875
876 Register DstReg = MI.getOperand(0).getReg();
877 Register SrcReg = MI.getOperand(2).getReg();
878 LLT DstTy = MRI.getType(DstReg);
879 LLT SrcTy = MRI.getType(SrcReg);
880
881 // If an spv_bitcast needs to be legalized, we convert it to G_BITCAST to
882 // allow using the generic legalization rules.
883 if (needsVectorLegalization(DstTy, ST) ||
884 needsVectorLegalization(SrcTy, ST)) {
885 LLVM_DEBUG(dbgs() << "Replacing with a G_BITCAST\n");
886 MIRBuilder.buildBitcast(DstReg, SrcReg);
887 MI.eraseFromParent();
888 }
889 return true;
890}
891
894 MachineIRBuilder &MIRBuilder = Helper.MIRBuilder;
895 MachineRegisterInfo &MRI = *MIRBuilder.getMRI();
896 const SPIRVSubtarget &ST = MI.getMF()->getSubtarget<SPIRVSubtarget>();
897
898 Register DstReg = MI.getOperand(0).getReg();
899 LLT DstTy = MRI.getType(DstReg);
900
901 if (needsVectorLegalization(DstTy, ST)) {
902 Register SrcReg = MI.getOperand(2).getReg();
903 Register ValReg = MI.getOperand(3).getReg();
904 LLT SrcTy = MRI.getType(SrcReg);
905 MachineOperand &IdxOperand = MI.getOperand(4);
906
907 if (getImm(IdxOperand, &MRI)) {
908 uint64_t IdxVal = foldImm(IdxOperand, &MRI);
909 if (IdxVal < SrcTy.getNumElements()) {
911 SPIRVTypeInst ElementType =
913 LLT ElementLLTTy = GR->getRegType(ElementType);
914 for (unsigned I = 0, E = SrcTy.getNumElements(); I < E; ++I) {
915 Register Reg = MRI.createGenericVirtualRegister(ElementLLTTy);
916 MRI.setRegClass(Reg, GR->getRegClass(ElementType));
917 GR->assignSPIRVTypeToVReg(ElementType, Reg, *MI.getMF());
918 Regs.push_back(Reg);
919 }
920 MIRBuilder.buildUnmerge(Regs, SrcReg);
921 Regs[IdxVal] = ValReg;
922 MIRBuilder.buildBuildVector(DstReg, Regs);
923 MI.eraseFromParent();
924 return true;
925 }
926 }
927
928 LLT EltTy = SrcTy.getElementType();
929 Align VecAlign;
930 MachinePointerInfo PtrInfo;
931 auto StackTemp = createStackTemporaryForVector(Helper, GR, SrcReg, SrcTy,
932 PtrInfo, VecAlign);
933
934 MIRBuilder.buildStore(SrcReg, StackTemp, PtrInfo, VecAlign);
935
936 Register IdxReg = IdxOperand.getReg();
937 LLT PtrTy = MRI.getType(StackTemp.getReg(0));
938 Register EltPtr = MRI.createGenericVirtualRegister(PtrTy);
939 auto Zero = MIRBuilder.buildConstant(LLT::scalar(32), 0);
940
941 MIRBuilder.buildIntrinsic(Intrinsic::spv_gep, ArrayRef<Register>{EltPtr})
942 .addImm(1) // InBounds
943 .addUse(StackTemp.getReg(0))
944 .addUse(Zero.getReg(0))
945 .addUse(IdxReg);
946
948 Align EltAlign = Helper.getStackTemporaryAlignment(EltTy);
949 MIRBuilder.buildStore(ValReg, EltPtr, EltPtrInfo, EltAlign);
950
951 MIRBuilder.buildLoad(DstReg, StackTemp, PtrInfo, VecAlign);
952 MI.eraseFromParent();
953 return true;
954 }
955 return true;
956}
957
960 MachineIRBuilder &MIRBuilder = Helper.MIRBuilder;
961 MachineRegisterInfo &MRI = *MIRBuilder.getMRI();
962 const SPIRVSubtarget &ST = MI.getMF()->getSubtarget<SPIRVSubtarget>();
963
964 Register SrcReg = MI.getOperand(2).getReg();
965 LLT SrcTy = MRI.getType(SrcReg);
966
967 if (needsVectorLegalization(SrcTy, ST)) {
968 Register DstReg = MI.getOperand(0).getReg();
969 MachineOperand &IdxOperand = MI.getOperand(3);
970
971 if (getImm(IdxOperand, &MRI)) {
972 uint64_t IdxVal = foldImm(IdxOperand, &MRI);
973 if (IdxVal < SrcTy.getNumElements()) {
974 LLT DstTy = MRI.getType(DstReg);
976 SPIRVTypeInst DstSpvTy = GR->getSPIRVTypeForVReg(DstReg);
977 for (unsigned I = 0, E = SrcTy.getNumElements(); I < E; ++I) {
978 if (I == IdxVal) {
979 Regs.push_back(DstReg);
980 } else {
982 MRI.setRegClass(Reg, GR->getRegClass(DstSpvTy));
983 GR->assignSPIRVTypeToVReg(DstSpvTy, Reg, *MI.getMF());
984 Regs.push_back(Reg);
985 }
986 }
987 MIRBuilder.buildUnmerge(Regs, SrcReg);
988 MI.eraseFromParent();
989 return true;
990 }
991 }
992
993 LLT EltTy = SrcTy.getElementType();
994 Align VecAlign;
995 MachinePointerInfo PtrInfo;
996 auto StackTemp = createStackTemporaryForVector(Helper, GR, SrcReg, SrcTy,
997 PtrInfo, VecAlign);
998
999 MIRBuilder.buildStore(SrcReg, StackTemp, PtrInfo, VecAlign);
1000
1001 Register IdxReg = IdxOperand.getReg();
1002 LLT PtrTy = MRI.getType(StackTemp.getReg(0));
1003 Register EltPtr = MRI.createGenericVirtualRegister(PtrTy);
1004 auto Zero = MIRBuilder.buildConstant(LLT::scalar(32), 0);
1005
1006 MIRBuilder.buildIntrinsic(Intrinsic::spv_gep, ArrayRef<Register>{EltPtr})
1007 .addImm(1) // InBounds
1008 .addUse(StackTemp.getReg(0))
1009 .addUse(Zero.getReg(0))
1010 .addUse(IdxReg);
1011
1013 Align EltAlign = Helper.getStackTemporaryAlignment(EltTy);
1014 MIRBuilder.buildLoad(DstReg, EltPtr, EltPtrInfo, EltAlign);
1015
1016 MI.eraseFromParent();
1017 return true;
1018 }
1019 return true;
1020}
1021
1023 SPIRVGlobalRegistry *GR) {
1024 MachineIRBuilder &MIRBuilder = Helper.MIRBuilder;
1025 MachineRegisterInfo &MRI = *MIRBuilder.getMRI();
1026 const SPIRVSubtarget &ST = MI.getMF()->getSubtarget<SPIRVSubtarget>();
1027
1028 Register DstReg = MI.getOperand(0).getReg();
1029 LLT DstTy = MRI.getType(DstReg);
1030
1031 if (!needsVectorLegalization(DstTy, ST))
1032 return true;
1033
1035 if (MI.getNumOperands() == 2) {
1036 // The "null" case: no values are attached.
1037 LLT EltTy = DstTy.getElementType();
1038 auto Zero = MIRBuilder.buildConstant(EltTy, 0);
1039 SPIRVTypeInst SpvDstTy = GR->getSPIRVTypeForVReg(DstReg);
1040 SPIRVTypeInst SpvEltTy = GR->getScalarOrVectorComponentType(SpvDstTy);
1041 GR->assignSPIRVTypeToVReg(SpvEltTy, Zero.getReg(0), MIRBuilder.getMF());
1042 for (unsigned i = 0; i < DstTy.getNumElements(); ++i)
1043 SrcRegs.push_back(Zero.getReg(0));
1044 } else {
1045 for (unsigned i = 2; i < MI.getNumOperands(); ++i) {
1046 SrcRegs.push_back(MI.getOperand(i).getReg());
1047 }
1048 }
1049 MIRBuilder.buildBuildVector(DstReg, SrcRegs);
1050 MI.eraseFromParent();
1051 return true;
1052}
1053
1055 MachineInstr &MI) const {
1056 LLVM_DEBUG(dbgs() << "legalizeIntrinsic: " << MI);
1057 auto IntrinsicID = cast<GIntrinsic>(MI).getIntrinsicID();
1058 switch (IntrinsicID) {
1059 case Intrinsic::spv_bitcast:
1060 return legalizeSpvBitcast(Helper, MI, GR);
1061 case Intrinsic::spv_insertelt:
1062 return legalizeSpvInsertElt(Helper, MI, GR);
1063 case Intrinsic::spv_extractelt:
1064 return legalizeSpvExtractElt(Helper, MI, GR);
1065 case Intrinsic::spv_const_composite:
1066 return legalizeSpvConstComposite(Helper, MI, GR);
1067 }
1068 return true;
1069}
1070
1071bool SPIRVLegalizerInfo::legalizeBitcast(LegalizerHelper &Helper,
1072 MachineInstr &MI) const {
1073 // Once the G_BITCAST is using vectors that are allowed, we turn it back into
1074 // an spv_bitcast to avoid verifier problems when the register types are the
1075 // same for the source and the result. Note that the SPIR-V types associated
1076 // with the bitcast can be different even if the register types are the same.
1077 MachineIRBuilder &MIRBuilder = Helper.MIRBuilder;
1078 Register DstReg = MI.getOperand(0).getReg();
1079 Register SrcReg = MI.getOperand(1).getReg();
1080 SmallVector<Register, 1> DstRegs = {DstReg};
1081 MIRBuilder.buildIntrinsic(Intrinsic::spv_bitcast, DstRegs).addUse(SrcReg);
1082 MI.eraseFromParent();
1083 return true;
1084}
1085
1086// Note this code was copied from LegalizerHelper::lowerISFPCLASS and adjusted
1087// to ensure that all instructions created during the lowering have SPIR-V types
1088// assigned to them.
1089bool SPIRVLegalizerInfo::legalizeIsFPClass(
1091 LostDebugLocObserver &LocObserver) const {
1092 auto [DstReg, DstTy, SrcReg, SrcTy] = MI.getFirst2RegLLTs();
1093 FPClassTest Mask = static_cast<FPClassTest>(MI.getOperand(2).getImm());
1094
1095 auto &MIRBuilder = Helper.MIRBuilder;
1096 auto &MF = MIRBuilder.getMF();
1097 MachineRegisterInfo &MRI = MF.getRegInfo();
1098
1099 Type *LLVMDstTy =
1100 IntegerType::get(MIRBuilder.getContext(), DstTy.getScalarSizeInBits());
1101 if (DstTy.isVector())
1102 LLVMDstTy = VectorType::get(LLVMDstTy, DstTy.getElementCount());
1103 SPIRVTypeInst SPIRVDstTy = GR->getOrCreateSPIRVType(
1104 LLVMDstTy, MIRBuilder, SPIRV::AccessQualifier::ReadWrite,
1105 /*EmitIR*/ true);
1106
1107 unsigned BitSize = SrcTy.getScalarSizeInBits();
1108 const fltSemantics &Semantics = getFltSemanticForLLT(SrcTy.getScalarType());
1109
1110 LLT IntTy = LLT::scalar(BitSize);
1111 Type *LLVMIntTy = IntegerType::get(MIRBuilder.getContext(), BitSize);
1112 if (SrcTy.isVector()) {
1113 IntTy = LLT::vector(SrcTy.getElementCount(), IntTy);
1114 LLVMIntTy = VectorType::get(LLVMIntTy, SrcTy.getElementCount());
1115 }
1116 SPIRVTypeInst SPIRVIntTy = GR->getOrCreateSPIRVType(
1117 LLVMIntTy, MIRBuilder, SPIRV::AccessQualifier::ReadWrite,
1118 /*EmitIR*/ true);
1119
1120 // Clang doesn't support capture of structured bindings:
1121 LLT DstTyCopy = DstTy;
1122 const auto assignSPIRVTy = [&](MachineInstrBuilder &&MI) {
1123 // Assign this MI's (assumed only) destination to one of the two types we
1124 // expect: either the G_IS_FPCLASS's destination type, or the integer type
1125 // bitcast from the source type.
1126 LLT MITy = MRI.getType(MI.getReg(0));
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);
1131 return MI;
1132 };
1133
1134 // Helper to build and assign a constant in one go
1135 const auto buildSPIRVConstant = [&](LLT Ty, auto &&C) -> MachineInstrBuilder {
1136 if (!Ty.isFixedVector())
1137 return assignSPIRVTy(MIRBuilder.buildConstant(Ty, C));
1138 auto ScalarC = MIRBuilder.buildConstant(Ty.getScalarType(), C);
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,
1144 /*EmitIR*/ true);
1145 GR->assignSPIRVTypeToVReg(VecEltTy, ScalarC.getReg(0), MF);
1146 return assignSPIRVTy(MIRBuilder.buildSplatBuildVector(Ty, ScalarC));
1147 };
1148
1149 if (Mask == fcNone) {
1150 MIRBuilder.buildCopy(DstReg, buildSPIRVConstant(DstTy, 0));
1151 MI.eraseFromParent();
1152 return true;
1153 }
1154 if (Mask == fcAllFlags) {
1155 MIRBuilder.buildCopy(DstReg, buildSPIRVConstant(DstTy, 1));
1156 MI.eraseFromParent();
1157 return true;
1158 }
1159
1160 // Note that rather than creating a COPY here (between a floating-point and
1161 // integer type of the same size) we create a SPIR-V bitcast immediately. We
1162 // can't create a G_BITCAST because the LLTs are the same, and we can't seem
1163 // to correctly lower COPYs to SPIR-V bitcasts at this moment.
1164 Register ResVReg = MRI.createGenericVirtualRegister(IntTy);
1165 MRI.setRegClass(ResVReg, GR->getRegClass(SPIRVIntTy));
1166 GR->assignSPIRVTypeToVReg(SPIRVIntTy, ResVReg, Helper.MIRBuilder.getMF());
1167 auto AsInt = MIRBuilder.buildInstr(SPIRV::OpBitcast)
1168 .addDef(ResVReg)
1169 .addUse(GR->getSPIRVTypeID(SPIRVIntTy))
1170 .addUse(SrcReg);
1171 AsInt = assignSPIRVTy(std::move(AsInt));
1172
1173 // Various masks.
1174 APInt SignBit = APInt::getSignMask(BitSize);
1175 APInt ValueMask = APInt::getSignedMaxValue(BitSize); // All bits but sign.
1176 APInt Inf = APFloat::getInf(Semantics).bitcastToAPInt(); // Exp and int bit.
1177 APInt ExpMask = Inf;
1178 APInt AllOneMantissa = APFloat::getLargest(Semantics).bitcastToAPInt() & ~Inf;
1179 APInt QNaNBitMask =
1180 APInt::getOneBitSet(BitSize, AllOneMantissa.getActiveBits() - 1);
1181 APInt InversionMask = APInt::getAllOnes(DstTy.getScalarSizeInBits());
1182
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);
1188
1189 auto Abs = assignSPIRVTy(MIRBuilder.buildAnd(IntTy, AsInt, ValueMaskC));
1190 auto Sign = assignSPIRVTy(
1191 MIRBuilder.buildICmp(CmpInst::Predicate::ICMP_NE, DstTy, AsInt, Abs));
1192
1193 auto Res = buildSPIRVConstant(DstTy, 0);
1194
1195 const auto appendToRes = [&](MachineInstrBuilder &&ToAppend) {
1196 Res = assignSPIRVTy(
1197 MIRBuilder.buildOr(DstTyCopy, Res, assignSPIRVTy(std::move(ToAppend))));
1198 };
1199
1200 // Tests that involve more than one class should be processed first.
1201 if ((Mask & fcFinite) == fcFinite) {
1202 // finite(V) ==> abs(V) u< exp_mask
1203 appendToRes(MIRBuilder.buildICmp(CmpInst::Predicate::ICMP_ULT, DstTy, Abs,
1204 ExpMaskC));
1205 Mask &= ~fcFinite;
1206 } else if ((Mask & fcFinite) == fcPosFinite) {
1207 // finite(V) && V > 0 ==> V u< exp_mask
1208 appendToRes(MIRBuilder.buildICmp(CmpInst::Predicate::ICMP_ULT, DstTy, AsInt,
1209 ExpMaskC));
1210 Mask &= ~fcPosFinite;
1211 } else if ((Mask & fcFinite) == fcNegFinite) {
1212 // finite(V) && V < 0 ==> abs(V) u< exp_mask && signbit == 1
1213 auto Cmp = assignSPIRVTy(MIRBuilder.buildICmp(CmpInst::Predicate::ICMP_ULT,
1214 DstTy, Abs, ExpMaskC));
1215 appendToRes(MIRBuilder.buildAnd(DstTy, Cmp, Sign));
1216 Mask &= ~fcNegFinite;
1217 }
1218
1219 if (FPClassTest PartialCheck = Mask & (fcZero | fcSubnormal)) {
1220 // fcZero | fcSubnormal => test all exponent bits are 0
1221 // TODO: Handle sign bit specific cases
1222 // TODO: Handle inverted case
1223 if (PartialCheck == (fcZero | fcSubnormal)) {
1224 auto ExpBits = assignSPIRVTy(MIRBuilder.buildAnd(IntTy, AsInt, ExpMaskC));
1225 appendToRes(MIRBuilder.buildICmp(CmpInst::Predicate::ICMP_EQ, DstTy,
1226 ExpBits, ZeroC));
1227 Mask &= ~PartialCheck;
1228 }
1229 }
1230
1231 // Check for individual classes.
1232 if (FPClassTest PartialCheck = Mask & fcZero) {
1233 if (PartialCheck == fcPosZero)
1234 appendToRes(MIRBuilder.buildICmp(CmpInst::Predicate::ICMP_EQ, DstTy,
1235 AsInt, ZeroC));
1236 else if (PartialCheck == fcZero)
1237 appendToRes(
1238 MIRBuilder.buildICmp(CmpInst::Predicate::ICMP_EQ, DstTy, Abs, ZeroC));
1239 else // fcNegZero
1240 appendToRes(MIRBuilder.buildICmp(CmpInst::Predicate::ICMP_EQ, DstTy,
1241 AsInt, SignBitC));
1242 }
1243
1244 if (FPClassTest PartialCheck = Mask & fcSubnormal) {
1245 // issubnormal(V) ==> unsigned(abs(V) - 1) u< (all mantissa bits set)
1246 // issubnormal(V) && V>0 ==> unsigned(V - 1) u< (all mantissa bits set)
1247 auto V = (PartialCheck == fcPosSubnormal) ? AsInt : Abs;
1248 auto OneC = buildSPIRVConstant(IntTy, 1);
1249 auto VMinusOne = MIRBuilder.buildSub(IntTy, V, OneC);
1250 auto SubnormalRes = assignSPIRVTy(
1251 MIRBuilder.buildICmp(CmpInst::Predicate::ICMP_ULT, DstTy, VMinusOne,
1252 buildSPIRVConstant(IntTy, AllOneMantissa)));
1253 if (PartialCheck == fcNegSubnormal)
1254 SubnormalRes = MIRBuilder.buildAnd(DstTy, SubnormalRes, Sign);
1255 appendToRes(std::move(SubnormalRes));
1256 }
1257
1258 if (FPClassTest PartialCheck = Mask & fcInf) {
1259 if (PartialCheck == fcPosInf)
1260 appendToRes(MIRBuilder.buildICmp(CmpInst::Predicate::ICMP_EQ, DstTy,
1261 AsInt, InfC));
1262 else if (PartialCheck == fcInf)
1263 appendToRes(
1264 MIRBuilder.buildICmp(CmpInst::Predicate::ICMP_EQ, DstTy, Abs, InfC));
1265 else { // fcNegInf
1266 APInt NegInf = APFloat::getInf(Semantics, true).bitcastToAPInt();
1267 auto NegInfC = buildSPIRVConstant(IntTy, NegInf);
1268 appendToRes(MIRBuilder.buildICmp(CmpInst::Predicate::ICMP_EQ, DstTy,
1269 AsInt, NegInfC));
1270 }
1271 }
1272
1273 if (FPClassTest PartialCheck = Mask & fcNan) {
1274 auto InfWithQnanBitC =
1275 buildSPIRVConstant(IntTy, std::move(Inf) | QNaNBitMask);
1276 if (PartialCheck == fcNan) {
1277 // isnan(V) ==> abs(V) u> int(inf)
1278 appendToRes(
1279 MIRBuilder.buildICmp(CmpInst::Predicate::ICMP_UGT, DstTy, Abs, InfC));
1280 } else if (PartialCheck == fcQNan) {
1281 // isquiet(V) ==> abs(V) u>= (unsigned(Inf) | quiet_bit)
1282 appendToRes(MIRBuilder.buildICmp(CmpInst::Predicate::ICMP_UGE, DstTy, Abs,
1283 InfWithQnanBitC));
1284 } else { // fcSNan
1285 // issignaling(V) ==> abs(V) u> unsigned(Inf) &&
1286 // abs(V) u< (unsigned(Inf) | quiet_bit)
1287 auto IsNan = assignSPIRVTy(
1288 MIRBuilder.buildICmp(CmpInst::Predicate::ICMP_UGT, DstTy, Abs, InfC));
1289 auto IsNotQnan = assignSPIRVTy(MIRBuilder.buildICmp(
1290 CmpInst::Predicate::ICMP_ULT, DstTy, Abs, InfWithQnanBitC));
1291 appendToRes(MIRBuilder.buildAnd(DstTy, IsNan, IsNotQnan));
1292 }
1293 }
1294
1295 if (FPClassTest PartialCheck = Mask & fcNormal) {
1296 // isnormal(V) ==> (0 u< exp u< max_exp) ==> (unsigned(exp-1) u<
1297 // (max_exp-1))
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(
1303 MIRBuilder.buildICmp(CmpInst::Predicate::ICMP_ULT, DstTy, ExpMinusOne,
1304 buildSPIRVConstant(IntTy, MaxExpMinusOne)));
1305 if (PartialCheck == fcNegNormal)
1306 NormalRes = MIRBuilder.buildAnd(DstTy, NormalRes, Sign);
1307 else if (PartialCheck == fcPosNormal) {
1308 auto PosSign = assignSPIRVTy(MIRBuilder.buildXor(
1309 DstTy, Sign, buildSPIRVConstant(DstTy, InversionMask)));
1310 NormalRes = MIRBuilder.buildAnd(DstTy, NormalRes, PosSign);
1311 }
1312 appendToRes(std::move(NormalRes));
1313 }
1314
1315 MIRBuilder.buildCopy(DstReg, Res);
1316 MI.eraseFromParent();
1317 return true;
1318}
assert(UImm &&(UImm !=~static_cast< T >(0)) &&"Invalid immediate!")
unsigned uint64_t
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...
IRTranslator LLVM IR MI
#define I(x, y, z)
Definition MD5.cpp:57
This file declares the MachineIRBuilder class.
Register Reg
Promote Memory to Register
Definition Mem2Reg.cpp:110
ppc ctr loops verify
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)
#define LLVM_DEBUG(...)
Definition Debug.h:119
APInt bitcastToAPInt() const
Definition APFloat.h:1475
static APFloat getLargest(const fltSemantics &Sem, bool Negative=false)
Returns the largest finite number in the given semantics.
Definition APFloat.h:1242
static APFloat getInf(const fltSemantics &Sem, bool Negative=false)
Factory for Positive and Negative Infinity.
Definition APFloat.h:1202
static APInt getAllOnes(unsigned numBits)
Return an APInt of a specified width with all bits set.
Definition APInt.h:231
static APInt getSignMask(unsigned BitWidth)
Get the SignMask for a specific bit width.
Definition APInt.h:226
unsigned getActiveBits() const
Compute the number of active bits in the value.
Definition APInt.h:1533
static APInt getSignedMaxValue(unsigned numBits)
Gets maximum signed value of APInt for a specific bit width.
Definition APInt.h:206
APInt shl(unsigned shiftAmt) const
Left-shift function.
Definition APInt.h:876
static APInt getOneBitSet(unsigned numBits, unsigned BitNo)
Return an APInt with exactly one bit set in the result.
Definition APInt.h:236
Represent a constant reference to an array (0 or more elements consecutively in memory),...
Definition ArrayRef.h:40
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.
Definition InstrTypes.h:740
@ ICMP_UGE
unsigned greater or equal
Definition InstrTypes.h:764
@ ICMP_UGT
unsigned greater than
Definition InstrTypes.h:763
@ ICMP_ULT
unsigned less than
Definition InstrTypes.h:765
@ ICMP_NE
not equal
Definition InstrTypes.h:762
static constexpr ElementCount getFixed(ScalarTy MinVal)
Definition TypeSize.h:309
static LLVM_ABI IntegerType * get(LLVMContext &C, unsigned NumBits)
This static method is the primary way of constructing an IntegerType.
Definition Type.cpp:348
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.
Definition Register.h:20
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)
Definition TypeSize.h:343
The instances of the Type class are immutable: once they are created, they are never changed.
Definition Type.h:46
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.
Definition MathExtras.h:280
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.
Definition Debug.cpp:209
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.
Definition Casting.h:559
Align commonAlignment(Align A, uint64_t Offset)
Returns the alignment that satisfies both alignments.
Definition Alignment.h:201
This struct is a compact representation of a valid (non-zero power of two) alignment.
Definition Alignment.h:39
The LegalityQuery object bundles together all the information that's needed to decide whether a given...
ArrayRef< LLT > Types
This class contains a discriminated union of information about pointers in memory operands,...
MachinePointerInfo getWithOffset(int64_t O) const