1 //===- AMDGPUBaseInfo.cpp - AMDGPU Base encoding information --------------===//
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 #include "AMDGPUBaseInfo.h"
10 #include "AMDGPU.h"
11 #include "AMDGPUAsmUtils.h"
12 #include "AMDKernelCodeT.h"
13 #include "GCNSubtarget.h"
14 #include "MCTargetDesc/AMDGPUMCTargetDesc.h"
15 #include "llvm/BinaryFormat/ELF.h"
16 #include "llvm/IR/Attributes.h"
17 #include "llvm/IR/Function.h"
18 #include "llvm/IR/GlobalValue.h"
19 #include "llvm/IR/IntrinsicsAMDGPU.h"
20 #include "llvm/IR/IntrinsicsR600.h"
21 #include "llvm/IR/LLVMContext.h"
22 #include "llvm/MC/MCSubtargetInfo.h"
23 #include "llvm/Support/AMDHSAKernelDescriptor.h"
24 #include "llvm/Support/CommandLine.h"
25 #include "llvm/Support/TargetParser.h"
26 
27 #define GET_INSTRINFO_NAMED_OPS
28 #define GET_INSTRMAP_INFO
29 #include "AMDGPUGenInstrInfo.inc"
30 
31 static llvm::cl::opt<unsigned> AmdhsaCodeObjectVersion(
32   "amdhsa-code-object-version", llvm::cl::Hidden,
33   llvm::cl::desc("AMDHSA Code Object Version"), llvm::cl::init(3));
34 
35 namespace {
36 
37 /// \returns Bit mask for given bit \p Shift and bit \p Width.
38 unsigned getBitMask(unsigned Shift, unsigned Width) {
39   return ((1 << Width) - 1) << Shift;
40 }
41 
42 /// Packs \p Src into \p Dst for given bit \p Shift and bit \p Width.
43 ///
44 /// \returns Packed \p Dst.
45 unsigned packBits(unsigned Src, unsigned Dst, unsigned Shift, unsigned Width) {
46   Dst &= ~(1 << Shift) & ~getBitMask(Shift, Width);
47   Dst |= (Src << Shift) & getBitMask(Shift, Width);
48   return Dst;
49 }
50 
51 /// Unpacks bits from \p Src for given bit \p Shift and bit \p Width.
52 ///
53 /// \returns Unpacked bits.
54 unsigned unpackBits(unsigned Src, unsigned Shift, unsigned Width) {
55   return (Src & getBitMask(Shift, Width)) >> Shift;
56 }
57 
58 /// \returns Vmcnt bit shift (lower bits).
59 unsigned getVmcntBitShiftLo() { return 0; }
60 
61 /// \returns Vmcnt bit width (lower bits).
62 unsigned getVmcntBitWidthLo() { return 4; }
63 
64 /// \returns Expcnt bit shift.
65 unsigned getExpcntBitShift() { return 4; }
66 
67 /// \returns Expcnt bit width.
68 unsigned getExpcntBitWidth() { return 3; }
69 
70 /// \returns Lgkmcnt bit shift.
71 unsigned getLgkmcntBitShift() { return 8; }
72 
73 /// \returns Lgkmcnt bit width.
74 unsigned getLgkmcntBitWidth(unsigned VersionMajor) {
75   return (VersionMajor >= 10) ? 6 : 4;
76 }
77 
78 /// \returns Vmcnt bit shift (higher bits).
79 unsigned getVmcntBitShiftHi() { return 14; }
80 
81 /// \returns Vmcnt bit width (higher bits).
82 unsigned getVmcntBitWidthHi() { return 2; }
83 
84 } // end namespace anonymous
85 
86 namespace llvm {
87 
88 namespace AMDGPU {
89 
90 Optional<uint8_t> getHsaAbiVersion(const MCSubtargetInfo *STI) {
91   if (STI && STI->getTargetTriple().getOS() != Triple::AMDHSA)
92     return None;
93 
94   switch (AmdhsaCodeObjectVersion) {
95   case 2:
96     return ELF::ELFABIVERSION_AMDGPU_HSA_V2;
97   case 3:
98     return ELF::ELFABIVERSION_AMDGPU_HSA_V3;
99   default:
100     return ELF::ELFABIVERSION_AMDGPU_HSA_V3;
101   }
102 }
103 
104 bool isHsaAbiVersion2(const MCSubtargetInfo *STI) {
105   if (const auto &&HsaAbiVer = getHsaAbiVersion(STI))
106     return HsaAbiVer.getValue() == ELF::ELFABIVERSION_AMDGPU_HSA_V2;
107   return false;
108 }
109 
110 bool isHsaAbiVersion3(const MCSubtargetInfo *STI) {
111   if (const auto &&HsaAbiVer = getHsaAbiVersion(STI))
112     return HsaAbiVer.getValue() == ELF::ELFABIVERSION_AMDGPU_HSA_V3;
113   return false;
114 }
115 
116 #define GET_MIMGBaseOpcodesTable_IMPL
117 #define GET_MIMGDimInfoTable_IMPL
118 #define GET_MIMGInfoTable_IMPL
119 #define GET_MIMGLZMappingTable_IMPL
120 #define GET_MIMGMIPMappingTable_IMPL
121 #define GET_MIMGG16MappingTable_IMPL
122 #include "AMDGPUGenSearchableTables.inc"
123 
124 int getMIMGOpcode(unsigned BaseOpcode, unsigned MIMGEncoding,
125                   unsigned VDataDwords, unsigned VAddrDwords) {
126   const MIMGInfo *Info = getMIMGOpcodeHelper(BaseOpcode, MIMGEncoding,
127                                              VDataDwords, VAddrDwords);
128   return Info ? Info->Opcode : -1;
129 }
130 
131 const MIMGBaseOpcodeInfo *getMIMGBaseOpcode(unsigned Opc) {
132   const MIMGInfo *Info = getMIMGInfo(Opc);
133   return Info ? getMIMGBaseOpcodeInfo(Info->BaseOpcode) : nullptr;
134 }
135 
136 int getMaskedMIMGOp(unsigned Opc, unsigned NewChannels) {
137   const MIMGInfo *OrigInfo = getMIMGInfo(Opc);
138   const MIMGInfo *NewInfo =
139       getMIMGOpcodeHelper(OrigInfo->BaseOpcode, OrigInfo->MIMGEncoding,
140                           NewChannels, OrigInfo->VAddrDwords);
141   return NewInfo ? NewInfo->Opcode : -1;
142 }
143 
144 struct MUBUFInfo {
145   uint16_t Opcode;
146   uint16_t BaseOpcode;
147   uint8_t elements;
148   bool has_vaddr;
149   bool has_srsrc;
150   bool has_soffset;
151 };
152 
153 struct MTBUFInfo {
154   uint16_t Opcode;
155   uint16_t BaseOpcode;
156   uint8_t elements;
157   bool has_vaddr;
158   bool has_srsrc;
159   bool has_soffset;
160 };
161 
162 struct SMInfo {
163   uint16_t Opcode;
164   bool IsBuffer;
165 };
166 
167 #define GET_MTBUFInfoTable_DECL
168 #define GET_MTBUFInfoTable_IMPL
169 #define GET_MUBUFInfoTable_DECL
170 #define GET_MUBUFInfoTable_IMPL
171 #define GET_SMInfoTable_DECL
172 #define GET_SMInfoTable_IMPL
173 #include "AMDGPUGenSearchableTables.inc"
174 
175 int getMTBUFBaseOpcode(unsigned Opc) {
176   const MTBUFInfo *Info = getMTBUFInfoFromOpcode(Opc);
177   return Info ? Info->BaseOpcode : -1;
178 }
179 
180 int getMTBUFOpcode(unsigned BaseOpc, unsigned Elements) {
181   const MTBUFInfo *Info = getMTBUFInfoFromBaseOpcodeAndElements(BaseOpc, Elements);
182   return Info ? Info->Opcode : -1;
183 }
184 
185 int getMTBUFElements(unsigned Opc) {
186   const MTBUFInfo *Info = getMTBUFOpcodeHelper(Opc);
187   return Info ? Info->elements : 0;
188 }
189 
190 bool getMTBUFHasVAddr(unsigned Opc) {
191   const MTBUFInfo *Info = getMTBUFOpcodeHelper(Opc);
192   return Info ? Info->has_vaddr : false;
193 }
194 
195 bool getMTBUFHasSrsrc(unsigned Opc) {
196   const MTBUFInfo *Info = getMTBUFOpcodeHelper(Opc);
197   return Info ? Info->has_srsrc : false;
198 }
199 
200 bool getMTBUFHasSoffset(unsigned Opc) {
201   const MTBUFInfo *Info = getMTBUFOpcodeHelper(Opc);
202   return Info ? Info->has_soffset : false;
203 }
204 
205 int getMUBUFBaseOpcode(unsigned Opc) {
206   const MUBUFInfo *Info = getMUBUFInfoFromOpcode(Opc);
207   return Info ? Info->BaseOpcode : -1;
208 }
209 
210 int getMUBUFOpcode(unsigned BaseOpc, unsigned Elements) {
211   const MUBUFInfo *Info = getMUBUFInfoFromBaseOpcodeAndElements(BaseOpc, Elements);
212   return Info ? Info->Opcode : -1;
213 }
214 
215 int getMUBUFElements(unsigned Opc) {
216   const MUBUFInfo *Info = getMUBUFOpcodeHelper(Opc);
217   return Info ? Info->elements : 0;
218 }
219 
220 bool getMUBUFHasVAddr(unsigned Opc) {
221   const MUBUFInfo *Info = getMUBUFOpcodeHelper(Opc);
222   return Info ? Info->has_vaddr : false;
223 }
224 
225 bool getMUBUFHasSrsrc(unsigned Opc) {
226   const MUBUFInfo *Info = getMUBUFOpcodeHelper(Opc);
227   return Info ? Info->has_srsrc : false;
228 }
229 
230 bool getMUBUFHasSoffset(unsigned Opc) {
231   const MUBUFInfo *Info = getMUBUFOpcodeHelper(Opc);
232   return Info ? Info->has_soffset : false;
233 }
234 
235 bool getSMEMIsBuffer(unsigned Opc) {
236   const SMInfo *Info = getSMEMOpcodeHelper(Opc);
237   return Info ? Info->IsBuffer : false;
238 }
239 
240 // Wrapper for Tablegen'd function.  enum Subtarget is not defined in any
241 // header files, so we need to wrap it in a function that takes unsigned
242 // instead.
243 int getMCOpcode(uint16_t Opcode, unsigned Gen) {
244   return getMCOpcodeGen(Opcode, static_cast<Subtarget>(Gen));
245 }
246 
247 namespace IsaInfo {
248 
249 AMDGPUTargetID::AMDGPUTargetID(const MCSubtargetInfo &STI)
250     : XnackSetting(TargetIDSetting::Any), SramEccSetting(TargetIDSetting::Any) {
251   if (!STI.getFeatureBits().test(FeatureSupportsXNACK))
252     XnackSetting = TargetIDSetting::Unsupported;
253   if (!STI.getFeatureBits().test(FeatureSupportsSRAMECC))
254     SramEccSetting = TargetIDSetting::Unsupported;
255 }
256 
257 void AMDGPUTargetID::setTargetIDFromFeaturesString(StringRef FS) {
258   // Check if xnack or sramecc is explicitly enabled or disabled.  In the
259   // absence of the target features we assume we must generate code that can run
260   // in any environment.
261   SubtargetFeatures Features(FS);
262   Optional<bool> XnackRequested;
263   Optional<bool> SramEccRequested;
264 
265   for (const std::string &Feature : Features.getFeatures()) {
266     if (Feature == "+xnack")
267       XnackRequested = true;
268     else if (Feature == "-xnack")
269       XnackRequested = false;
270     else if (Feature == "+sramecc")
271       SramEccRequested = true;
272     else if (Feature == "-sramecc")
273       SramEccRequested = false;
274   }
275 
276   bool XnackSupported = isXnackSupported();
277   bool SramEccSupported = isSramEccSupported();
278 
279   if (XnackRequested) {
280     if (XnackSupported) {
281       XnackSetting =
282           *XnackRequested ? TargetIDSetting::On : TargetIDSetting::Off;
283     } else {
284       // If a specific xnack setting was requested and this GPU does not support
285       // xnack emit a warning. Setting will remain set to "Unsupported".
286       if (*XnackRequested) {
287         errs() << "warning: xnack 'On' was requested for a processor that does "
288                   "not support it!\n";
289       } else {
290         errs() << "warning: xnack 'Off' was requested for a processor that "
291                   "does not support it!\n";
292       }
293     }
294   }
295 
296   if (SramEccRequested) {
297     if (SramEccSupported) {
298       SramEccSetting =
299           *SramEccRequested ? TargetIDSetting::On : TargetIDSetting::Off;
300     } else {
301       // If a specific sramecc setting was requested and this GPU does not
302       // support sramecc emit a warning. Setting will remain set to
303       // "Unsupported".
304       if (*SramEccRequested) {
305         errs() << "warning: sramecc 'On' was requested for a processor that "
306                   "does not support it!\n";
307       } else {
308         errs() << "warning: sramecc 'Off' was requested for a processor that "
309                   "does not support it!\n";
310       }
311     }
312   }
313 }
314 
315 static TargetIDSetting
316 getTargetIDSettingFromFeatureString(StringRef FeatureString) {
317   if (FeatureString.endswith("-"))
318     return TargetIDSetting::Off;
319   if (FeatureString.endswith("+"))
320     return TargetIDSetting::On;
321 
322   llvm_unreachable("Malformed feature string");
323 }
324 
325 void AMDGPUTargetID::setTargetIDFromTargetIDStream(StringRef TargetID) {
326   SmallVector<StringRef, 3> TargetIDSplit;
327   TargetID.split(TargetIDSplit, ':');
328 
329   for (const auto &FeatureString : TargetIDSplit) {
330     if (FeatureString.startswith("xnack"))
331       XnackSetting = getTargetIDSettingFromFeatureString(FeatureString);
332     if (FeatureString.startswith("sramecc"))
333       SramEccSetting = getTargetIDSettingFromFeatureString(FeatureString);
334   }
335 }
336 
337 void streamIsaVersion(const MCSubtargetInfo *STI, raw_ostream &Stream) {
338   auto TargetTriple = STI->getTargetTriple();
339   auto Version = getIsaVersion(STI->getCPU());
340 
341   Stream << TargetTriple.getArchName() << '-'
342          << TargetTriple.getVendorName() << '-'
343          << TargetTriple.getOSName() << '-'
344          << TargetTriple.getEnvironmentName() << '-'
345          << "gfx"
346          << Version.Major
347          << Version.Minor
348          << hexdigit(Version.Stepping, true);
349 
350   if (hasXNACK(*STI))
351     Stream << "+xnack";
352   if (hasSRAMECC(*STI))
353     Stream << "+sramecc";
354 
355   Stream.flush();
356 }
357 
358 unsigned getWavefrontSize(const MCSubtargetInfo *STI) {
359   if (STI->getFeatureBits().test(FeatureWavefrontSize16))
360     return 16;
361   if (STI->getFeatureBits().test(FeatureWavefrontSize32))
362     return 32;
363 
364   return 64;
365 }
366 
367 unsigned getLocalMemorySize(const MCSubtargetInfo *STI) {
368   if (STI->getFeatureBits().test(FeatureLocalMemorySize32768))
369     return 32768;
370   if (STI->getFeatureBits().test(FeatureLocalMemorySize65536))
371     return 65536;
372 
373   return 0;
374 }
375 
376 unsigned getEUsPerCU(const MCSubtargetInfo *STI) {
377   // "Per CU" really means "per whatever functional block the waves of a
378   // workgroup must share". For gfx10 in CU mode this is the CU, which contains
379   // two SIMDs.
380   if (isGFX10Plus(*STI) && STI->getFeatureBits().test(FeatureCuMode))
381     return 2;
382   // Pre-gfx10 a CU contains four SIMDs. For gfx10 in WGP mode the WGP contains
383   // two CUs, so a total of four SIMDs.
384   return 4;
385 }
386 
387 unsigned getMaxWorkGroupsPerCU(const MCSubtargetInfo *STI,
388                                unsigned FlatWorkGroupSize) {
389   assert(FlatWorkGroupSize != 0);
390   if (STI->getTargetTriple().getArch() != Triple::amdgcn)
391     return 8;
392   unsigned N = getWavesPerWorkGroup(STI, FlatWorkGroupSize);
393   if (N == 1)
394     return 40;
395   N = 40 / N;
396   return std::min(N, 16u);
397 }
398 
399 unsigned getMinWavesPerEU(const MCSubtargetInfo *STI) {
400   return 1;
401 }
402 
403 unsigned getMaxWavesPerEU(const MCSubtargetInfo *STI) {
404   // FIXME: Need to take scratch memory into account.
405   if (isGFX90A(*STI))
406     return 8;
407   if (!isGFX10Plus(*STI))
408     return 10;
409   return hasGFX10_3Insts(*STI) ? 16 : 20;
410 }
411 
412 unsigned getWavesPerEUForWorkGroup(const MCSubtargetInfo *STI,
413                                    unsigned FlatWorkGroupSize) {
414   return divideCeil(getWavesPerWorkGroup(STI, FlatWorkGroupSize),
415                     getEUsPerCU(STI));
416 }
417 
418 unsigned getMinFlatWorkGroupSize(const MCSubtargetInfo *STI) {
419   return 1;
420 }
421 
422 unsigned getMaxFlatWorkGroupSize(const MCSubtargetInfo *STI) {
423   // Some subtargets allow encoding 2048, but this isn't tested or supported.
424   return 1024;
425 }
426 
427 unsigned getWavesPerWorkGroup(const MCSubtargetInfo *STI,
428                               unsigned FlatWorkGroupSize) {
429   return divideCeil(FlatWorkGroupSize, getWavefrontSize(STI));
430 }
431 
432 unsigned getSGPRAllocGranule(const MCSubtargetInfo *STI) {
433   IsaVersion Version = getIsaVersion(STI->getCPU());
434   if (Version.Major >= 10)
435     return getAddressableNumSGPRs(STI);
436   if (Version.Major >= 8)
437     return 16;
438   return 8;
439 }
440 
441 unsigned getSGPREncodingGranule(const MCSubtargetInfo *STI) {
442   return 8;
443 }
444 
445 unsigned getTotalNumSGPRs(const MCSubtargetInfo *STI) {
446   IsaVersion Version = getIsaVersion(STI->getCPU());
447   if (Version.Major >= 8)
448     return 800;
449   return 512;
450 }
451 
452 unsigned getAddressableNumSGPRs(const MCSubtargetInfo *STI) {
453   if (STI->getFeatureBits().test(FeatureSGPRInitBug))
454     return FIXED_NUM_SGPRS_FOR_INIT_BUG;
455 
456   IsaVersion Version = getIsaVersion(STI->getCPU());
457   if (Version.Major >= 10)
458     return 106;
459   if (Version.Major >= 8)
460     return 102;
461   return 104;
462 }
463 
464 unsigned getMinNumSGPRs(const MCSubtargetInfo *STI, unsigned WavesPerEU) {
465   assert(WavesPerEU != 0);
466 
467   IsaVersion Version = getIsaVersion(STI->getCPU());
468   if (Version.Major >= 10)
469     return 0;
470 
471   if (WavesPerEU >= getMaxWavesPerEU(STI))
472     return 0;
473 
474   unsigned MinNumSGPRs = getTotalNumSGPRs(STI) / (WavesPerEU + 1);
475   if (STI->getFeatureBits().test(FeatureTrapHandler))
476     MinNumSGPRs -= std::min(MinNumSGPRs, (unsigned)TRAP_NUM_SGPRS);
477   MinNumSGPRs = alignDown(MinNumSGPRs, getSGPRAllocGranule(STI)) + 1;
478   return std::min(MinNumSGPRs, getAddressableNumSGPRs(STI));
479 }
480 
481 unsigned getMaxNumSGPRs(const MCSubtargetInfo *STI, unsigned WavesPerEU,
482                         bool Addressable) {
483   assert(WavesPerEU != 0);
484 
485   unsigned AddressableNumSGPRs = getAddressableNumSGPRs(STI);
486   IsaVersion Version = getIsaVersion(STI->getCPU());
487   if (Version.Major >= 10)
488     return Addressable ? AddressableNumSGPRs : 108;
489   if (Version.Major >= 8 && !Addressable)
490     AddressableNumSGPRs = 112;
491   unsigned MaxNumSGPRs = getTotalNumSGPRs(STI) / WavesPerEU;
492   if (STI->getFeatureBits().test(FeatureTrapHandler))
493     MaxNumSGPRs -= std::min(MaxNumSGPRs, (unsigned)TRAP_NUM_SGPRS);
494   MaxNumSGPRs = alignDown(MaxNumSGPRs, getSGPRAllocGranule(STI));
495   return std::min(MaxNumSGPRs, AddressableNumSGPRs);
496 }
497 
498 unsigned getNumExtraSGPRs(const MCSubtargetInfo *STI, bool VCCUsed,
499                           bool FlatScrUsed, bool XNACKUsed) {
500   unsigned ExtraSGPRs = 0;
501   if (VCCUsed)
502     ExtraSGPRs = 2;
503 
504   IsaVersion Version = getIsaVersion(STI->getCPU());
505   if (Version.Major >= 10)
506     return ExtraSGPRs;
507 
508   if (Version.Major < 8) {
509     if (FlatScrUsed)
510       ExtraSGPRs = 4;
511   } else {
512     if (XNACKUsed)
513       ExtraSGPRs = 4;
514 
515     if (FlatScrUsed)
516       ExtraSGPRs = 6;
517   }
518 
519   return ExtraSGPRs;
520 }
521 
522 unsigned getNumExtraSGPRs(const MCSubtargetInfo *STI, bool VCCUsed,
523                           bool FlatScrUsed) {
524   return getNumExtraSGPRs(STI, VCCUsed, FlatScrUsed,
525                           STI->getFeatureBits().test(AMDGPU::FeatureXNACK));
526 }
527 
528 unsigned getNumSGPRBlocks(const MCSubtargetInfo *STI, unsigned NumSGPRs) {
529   NumSGPRs = alignTo(std::max(1u, NumSGPRs), getSGPREncodingGranule(STI));
530   // SGPRBlocks is actual number of SGPR blocks minus 1.
531   return NumSGPRs / getSGPREncodingGranule(STI) - 1;
532 }
533 
534 unsigned getVGPRAllocGranule(const MCSubtargetInfo *STI,
535                              Optional<bool> EnableWavefrontSize32) {
536   if (STI->getFeatureBits().test(FeatureGFX90AInsts))
537     return 8;
538 
539   bool IsWave32 = EnableWavefrontSize32 ?
540       *EnableWavefrontSize32 :
541       STI->getFeatureBits().test(FeatureWavefrontSize32);
542 
543   if (hasGFX10_3Insts(*STI))
544     return IsWave32 ? 16 : 8;
545 
546   return IsWave32 ? 8 : 4;
547 }
548 
549 unsigned getVGPREncodingGranule(const MCSubtargetInfo *STI,
550                                 Optional<bool> EnableWavefrontSize32) {
551   if (STI->getFeatureBits().test(FeatureGFX90AInsts))
552     return 8;
553 
554   bool IsWave32 = EnableWavefrontSize32 ?
555       *EnableWavefrontSize32 :
556       STI->getFeatureBits().test(FeatureWavefrontSize32);
557 
558   return IsWave32 ? 8 : 4;
559 }
560 
561 unsigned getTotalNumVGPRs(const MCSubtargetInfo *STI) {
562   if (STI->getFeatureBits().test(FeatureGFX90AInsts))
563     return 512;
564   if (!isGFX10Plus(*STI))
565     return 256;
566   return STI->getFeatureBits().test(FeatureWavefrontSize32) ? 1024 : 512;
567 }
568 
569 unsigned getAddressableNumVGPRs(const MCSubtargetInfo *STI) {
570   if (STI->getFeatureBits().test(FeatureGFX90AInsts))
571     return 512;
572   return 256;
573 }
574 
575 unsigned getMinNumVGPRs(const MCSubtargetInfo *STI, unsigned WavesPerEU) {
576   assert(WavesPerEU != 0);
577 
578   if (WavesPerEU >= getMaxWavesPerEU(STI))
579     return 0;
580   unsigned MinNumVGPRs =
581       alignDown(getTotalNumVGPRs(STI) / (WavesPerEU + 1),
582                 getVGPRAllocGranule(STI)) + 1;
583   return std::min(MinNumVGPRs, getAddressableNumVGPRs(STI));
584 }
585 
586 unsigned getMaxNumVGPRs(const MCSubtargetInfo *STI, unsigned WavesPerEU) {
587   assert(WavesPerEU != 0);
588 
589   unsigned MaxNumVGPRs = alignDown(getTotalNumVGPRs(STI) / WavesPerEU,
590                                    getVGPRAllocGranule(STI));
591   unsigned AddressableNumVGPRs = getAddressableNumVGPRs(STI);
592   return std::min(MaxNumVGPRs, AddressableNumVGPRs);
593 }
594 
595 unsigned getNumVGPRBlocks(const MCSubtargetInfo *STI, unsigned NumVGPRs,
596                           Optional<bool> EnableWavefrontSize32) {
597   NumVGPRs = alignTo(std::max(1u, NumVGPRs),
598                      getVGPREncodingGranule(STI, EnableWavefrontSize32));
599   // VGPRBlocks is actual number of VGPR blocks minus 1.
600   return NumVGPRs / getVGPREncodingGranule(STI, EnableWavefrontSize32) - 1;
601 }
602 
603 } // end namespace IsaInfo
604 
605 void initDefaultAMDKernelCodeT(amd_kernel_code_t &Header,
606                                const MCSubtargetInfo *STI) {
607   IsaVersion Version = getIsaVersion(STI->getCPU());
608 
609   memset(&Header, 0, sizeof(Header));
610 
611   Header.amd_kernel_code_version_major = 1;
612   Header.amd_kernel_code_version_minor = 2;
613   Header.amd_machine_kind = 1; // AMD_MACHINE_KIND_AMDGPU
614   Header.amd_machine_version_major = Version.Major;
615   Header.amd_machine_version_minor = Version.Minor;
616   Header.amd_machine_version_stepping = Version.Stepping;
617   Header.kernel_code_entry_byte_offset = sizeof(Header);
618   Header.wavefront_size = 6;
619 
620   // If the code object does not support indirect functions, then the value must
621   // be 0xffffffff.
622   Header.call_convention = -1;
623 
624   // These alignment values are specified in powers of two, so alignment =
625   // 2^n.  The minimum alignment is 2^4 = 16.
626   Header.kernarg_segment_alignment = 4;
627   Header.group_segment_alignment = 4;
628   Header.private_segment_alignment = 4;
629 
630   if (Version.Major >= 10) {
631     if (STI->getFeatureBits().test(FeatureWavefrontSize32)) {
632       Header.wavefront_size = 5;
633       Header.code_properties |= AMD_CODE_PROPERTY_ENABLE_WAVEFRONT_SIZE32;
634     }
635     Header.compute_pgm_resource_registers |=
636       S_00B848_WGP_MODE(STI->getFeatureBits().test(FeatureCuMode) ? 0 : 1) |
637       S_00B848_MEM_ORDERED(1);
638   }
639 }
640 
641 amdhsa::kernel_descriptor_t getDefaultAmdhsaKernelDescriptor(
642     const MCSubtargetInfo *STI) {
643   IsaVersion Version = getIsaVersion(STI->getCPU());
644 
645   amdhsa::kernel_descriptor_t KD;
646   memset(&KD, 0, sizeof(KD));
647 
648   AMDHSA_BITS_SET(KD.compute_pgm_rsrc1,
649                   amdhsa::COMPUTE_PGM_RSRC1_FLOAT_DENORM_MODE_16_64,
650                   amdhsa::FLOAT_DENORM_MODE_FLUSH_NONE);
651   AMDHSA_BITS_SET(KD.compute_pgm_rsrc1,
652                   amdhsa::COMPUTE_PGM_RSRC1_ENABLE_DX10_CLAMP, 1);
653   AMDHSA_BITS_SET(KD.compute_pgm_rsrc1,
654                   amdhsa::COMPUTE_PGM_RSRC1_ENABLE_IEEE_MODE, 1);
655   AMDHSA_BITS_SET(KD.compute_pgm_rsrc2,
656                   amdhsa::COMPUTE_PGM_RSRC2_ENABLE_SGPR_WORKGROUP_ID_X, 1);
657   if (Version.Major >= 10) {
658     AMDHSA_BITS_SET(KD.kernel_code_properties,
659                     amdhsa::KERNEL_CODE_PROPERTY_ENABLE_WAVEFRONT_SIZE32,
660                     STI->getFeatureBits().test(FeatureWavefrontSize32) ? 1 : 0);
661     AMDHSA_BITS_SET(KD.compute_pgm_rsrc1,
662                     amdhsa::COMPUTE_PGM_RSRC1_WGP_MODE,
663                     STI->getFeatureBits().test(FeatureCuMode) ? 0 : 1);
664     AMDHSA_BITS_SET(KD.compute_pgm_rsrc1,
665                     amdhsa::COMPUTE_PGM_RSRC1_MEM_ORDERED, 1);
666   }
667   if (AMDGPU::isGFX90A(*STI)) {
668     AMDHSA_BITS_SET(KD.compute_pgm_rsrc3,
669                     amdhsa::COMPUTE_PGM_RSRC3_GFX90A_TG_SPLIT,
670                     STI->getFeatureBits().test(FeatureTgSplit) ? 1 : 0);
671   }
672   return KD;
673 }
674 
675 bool isGroupSegment(const GlobalValue *GV) {
676   return GV->getAddressSpace() == AMDGPUAS::LOCAL_ADDRESS;
677 }
678 
679 bool isGlobalSegment(const GlobalValue *GV) {
680   return GV->getAddressSpace() == AMDGPUAS::GLOBAL_ADDRESS;
681 }
682 
683 bool isReadOnlySegment(const GlobalValue *GV) {
684   unsigned AS = GV->getAddressSpace();
685   return AS == AMDGPUAS::CONSTANT_ADDRESS ||
686          AS == AMDGPUAS::CONSTANT_ADDRESS_32BIT;
687 }
688 
689 bool shouldEmitConstantsToTextSection(const Triple &TT) {
690   return TT.getArch() == Triple::r600;
691 }
692 
693 int getIntegerAttribute(const Function &F, StringRef Name, int Default) {
694   Attribute A = F.getFnAttribute(Name);
695   int Result = Default;
696 
697   if (A.isStringAttribute()) {
698     StringRef Str = A.getValueAsString();
699     if (Str.getAsInteger(0, Result)) {
700       LLVMContext &Ctx = F.getContext();
701       Ctx.emitError("can't parse integer attribute " + Name);
702     }
703   }
704 
705   return Result;
706 }
707 
708 std::pair<int, int> getIntegerPairAttribute(const Function &F,
709                                             StringRef Name,
710                                             std::pair<int, int> Default,
711                                             bool OnlyFirstRequired) {
712   Attribute A = F.getFnAttribute(Name);
713   if (!A.isStringAttribute())
714     return Default;
715 
716   LLVMContext &Ctx = F.getContext();
717   std::pair<int, int> Ints = Default;
718   std::pair<StringRef, StringRef> Strs = A.getValueAsString().split(',');
719   if (Strs.first.trim().getAsInteger(0, Ints.first)) {
720     Ctx.emitError("can't parse first integer attribute " + Name);
721     return Default;
722   }
723   if (Strs.second.trim().getAsInteger(0, Ints.second)) {
724     if (!OnlyFirstRequired || !Strs.second.trim().empty()) {
725       Ctx.emitError("can't parse second integer attribute " + Name);
726       return Default;
727     }
728   }
729 
730   return Ints;
731 }
732 
733 unsigned getVmcntBitMask(const IsaVersion &Version) {
734   unsigned VmcntLo = (1 << getVmcntBitWidthLo()) - 1;
735   if (Version.Major < 9)
736     return VmcntLo;
737 
738   unsigned VmcntHi = ((1 << getVmcntBitWidthHi()) - 1) << getVmcntBitWidthLo();
739   return VmcntLo | VmcntHi;
740 }
741 
742 unsigned getExpcntBitMask(const IsaVersion &Version) {
743   return (1 << getExpcntBitWidth()) - 1;
744 }
745 
746 unsigned getLgkmcntBitMask(const IsaVersion &Version) {
747   return (1 << getLgkmcntBitWidth(Version.Major)) - 1;
748 }
749 
750 unsigned getWaitcntBitMask(const IsaVersion &Version) {
751   unsigned VmcntLo = getBitMask(getVmcntBitShiftLo(), getVmcntBitWidthLo());
752   unsigned Expcnt = getBitMask(getExpcntBitShift(), getExpcntBitWidth());
753   unsigned Lgkmcnt = getBitMask(getLgkmcntBitShift(),
754                                 getLgkmcntBitWidth(Version.Major));
755   unsigned Waitcnt = VmcntLo | Expcnt | Lgkmcnt;
756   if (Version.Major < 9)
757     return Waitcnt;
758 
759   unsigned VmcntHi = getBitMask(getVmcntBitShiftHi(), getVmcntBitWidthHi());
760   return Waitcnt | VmcntHi;
761 }
762 
763 unsigned decodeVmcnt(const IsaVersion &Version, unsigned Waitcnt) {
764   unsigned VmcntLo =
765       unpackBits(Waitcnt, getVmcntBitShiftLo(), getVmcntBitWidthLo());
766   if (Version.Major < 9)
767     return VmcntLo;
768 
769   unsigned VmcntHi =
770       unpackBits(Waitcnt, getVmcntBitShiftHi(), getVmcntBitWidthHi());
771   VmcntHi <<= getVmcntBitWidthLo();
772   return VmcntLo | VmcntHi;
773 }
774 
775 unsigned decodeExpcnt(const IsaVersion &Version, unsigned Waitcnt) {
776   return unpackBits(Waitcnt, getExpcntBitShift(), getExpcntBitWidth());
777 }
778 
779 unsigned decodeLgkmcnt(const IsaVersion &Version, unsigned Waitcnt) {
780   return unpackBits(Waitcnt, getLgkmcntBitShift(),
781                     getLgkmcntBitWidth(Version.Major));
782 }
783 
784 void decodeWaitcnt(const IsaVersion &Version, unsigned Waitcnt,
785                    unsigned &Vmcnt, unsigned &Expcnt, unsigned &Lgkmcnt) {
786   Vmcnt = decodeVmcnt(Version, Waitcnt);
787   Expcnt = decodeExpcnt(Version, Waitcnt);
788   Lgkmcnt = decodeLgkmcnt(Version, Waitcnt);
789 }
790 
791 Waitcnt decodeWaitcnt(const IsaVersion &Version, unsigned Encoded) {
792   Waitcnt Decoded;
793   Decoded.VmCnt = decodeVmcnt(Version, Encoded);
794   Decoded.ExpCnt = decodeExpcnt(Version, Encoded);
795   Decoded.LgkmCnt = decodeLgkmcnt(Version, Encoded);
796   return Decoded;
797 }
798 
799 unsigned encodeVmcnt(const IsaVersion &Version, unsigned Waitcnt,
800                      unsigned Vmcnt) {
801   Waitcnt =
802       packBits(Vmcnt, Waitcnt, getVmcntBitShiftLo(), getVmcntBitWidthLo());
803   if (Version.Major < 9)
804     return Waitcnt;
805 
806   Vmcnt >>= getVmcntBitWidthLo();
807   return packBits(Vmcnt, Waitcnt, getVmcntBitShiftHi(), getVmcntBitWidthHi());
808 }
809 
810 unsigned encodeExpcnt(const IsaVersion &Version, unsigned Waitcnt,
811                       unsigned Expcnt) {
812   return packBits(Expcnt, Waitcnt, getExpcntBitShift(), getExpcntBitWidth());
813 }
814 
815 unsigned encodeLgkmcnt(const IsaVersion &Version, unsigned Waitcnt,
816                        unsigned Lgkmcnt) {
817   return packBits(Lgkmcnt, Waitcnt, getLgkmcntBitShift(),
818                                     getLgkmcntBitWidth(Version.Major));
819 }
820 
821 unsigned encodeWaitcnt(const IsaVersion &Version,
822                        unsigned Vmcnt, unsigned Expcnt, unsigned Lgkmcnt) {
823   unsigned Waitcnt = getWaitcntBitMask(Version);
824   Waitcnt = encodeVmcnt(Version, Waitcnt, Vmcnt);
825   Waitcnt = encodeExpcnt(Version, Waitcnt, Expcnt);
826   Waitcnt = encodeLgkmcnt(Version, Waitcnt, Lgkmcnt);
827   return Waitcnt;
828 }
829 
830 unsigned encodeWaitcnt(const IsaVersion &Version, const Waitcnt &Decoded) {
831   return encodeWaitcnt(Version, Decoded.VmCnt, Decoded.ExpCnt, Decoded.LgkmCnt);
832 }
833 
834 //===----------------------------------------------------------------------===//
835 // hwreg
836 //===----------------------------------------------------------------------===//
837 
838 namespace Hwreg {
839 
840 int64_t getHwregId(const StringRef Name) {
841   for (int Id = ID_SYMBOLIC_FIRST_; Id < ID_SYMBOLIC_LAST_; ++Id) {
842     if (IdSymbolic[Id] && Name == IdSymbolic[Id])
843       return Id;
844   }
845   return ID_UNKNOWN_;
846 }
847 
848 static unsigned getLastSymbolicHwreg(const MCSubtargetInfo &STI) {
849   if (isSI(STI) || isCI(STI) || isVI(STI))
850     return ID_SYMBOLIC_FIRST_GFX9_;
851   else if (isGFX9(STI))
852     return ID_SYMBOLIC_FIRST_GFX10_;
853   else if (isGFX10(STI) && !isGFX10_BEncoding(STI))
854     return ID_SYMBOLIC_FIRST_GFX1030_;
855   else
856     return ID_SYMBOLIC_LAST_;
857 }
858 
859 bool isValidHwreg(int64_t Id, const MCSubtargetInfo &STI) {
860   return
861     ID_SYMBOLIC_FIRST_ <= Id && Id < getLastSymbolicHwreg(STI) &&
862     IdSymbolic[Id] && (Id != ID_XNACK_MASK || !AMDGPU::isGFX10_BEncoding(STI));
863 }
864 
865 bool isValidHwreg(int64_t Id) {
866   return 0 <= Id && isUInt<ID_WIDTH_>(Id);
867 }
868 
869 bool isValidHwregOffset(int64_t Offset) {
870   return 0 <= Offset && isUInt<OFFSET_WIDTH_>(Offset);
871 }
872 
873 bool isValidHwregWidth(int64_t Width) {
874   return 0 <= (Width - 1) && isUInt<WIDTH_M1_WIDTH_>(Width - 1);
875 }
876 
877 uint64_t encodeHwreg(uint64_t Id, uint64_t Offset, uint64_t Width) {
878   return (Id << ID_SHIFT_) |
879          (Offset << OFFSET_SHIFT_) |
880          ((Width - 1) << WIDTH_M1_SHIFT_);
881 }
882 
883 StringRef getHwreg(unsigned Id, const MCSubtargetInfo &STI) {
884   return isValidHwreg(Id, STI) ? IdSymbolic[Id] : "";
885 }
886 
887 void decodeHwreg(unsigned Val, unsigned &Id, unsigned &Offset, unsigned &Width) {
888   Id = (Val & ID_MASK_) >> ID_SHIFT_;
889   Offset = (Val & OFFSET_MASK_) >> OFFSET_SHIFT_;
890   Width = ((Val & WIDTH_M1_MASK_) >> WIDTH_M1_SHIFT_) + 1;
891 }
892 
893 } // namespace Hwreg
894 
895 //===----------------------------------------------------------------------===//
896 // exp tgt
897 //===----------------------------------------------------------------------===//
898 
899 namespace Exp {
900 
901 struct ExpTgt {
902   StringLiteral Name;
903   unsigned Tgt;
904   unsigned MaxIndex;
905 };
906 
907 static constexpr ExpTgt ExpTgtInfo[] = {
908   {{"null"},  ET_NULL,   ET_NULL_MAX_IDX},
909   {{"mrtz"},  ET_MRTZ,   ET_MRTZ_MAX_IDX},
910   {{"prim"},  ET_PRIM,   ET_PRIM_MAX_IDX},
911   {{"mrt"},   ET_MRT0,   ET_MRT_MAX_IDX},
912   {{"pos"},   ET_POS0,   ET_POS_MAX_IDX},
913   {{"param"}, ET_PARAM0, ET_PARAM_MAX_IDX},
914 };
915 
916 bool getTgtName(unsigned Id, StringRef &Name, int &Index) {
917   for (const ExpTgt &Val : ExpTgtInfo) {
918     if (Val.Tgt <= Id && Id <= Val.Tgt + Val.MaxIndex) {
919       Index = (Val.MaxIndex == 0) ? -1 : (Id - Val.Tgt);
920       Name = Val.Name;
921       return true;
922     }
923   }
924   return false;
925 }
926 
927 unsigned getTgtId(const StringRef Name) {
928 
929   for (const ExpTgt &Val : ExpTgtInfo) {
930     if (Val.MaxIndex == 0 && Name == Val.Name)
931       return Val.Tgt;
932 
933     if (Val.MaxIndex > 0 && Name.startswith(Val.Name)) {
934       StringRef Suffix = Name.drop_front(Val.Name.size());
935 
936       unsigned Id;
937       if (Suffix.getAsInteger(10, Id) || Id > Val.MaxIndex)
938         return ET_INVALID;
939 
940       // Disable leading zeroes
941       if (Suffix.size() > 1 && Suffix[0] == '0')
942         return ET_INVALID;
943 
944       return Val.Tgt + Id;
945     }
946   }
947   return ET_INVALID;
948 }
949 
950 bool isSupportedTgtId(unsigned Id, const MCSubtargetInfo &STI) {
951   return (Id != ET_POS4 && Id != ET_PRIM) || isGFX10Plus(STI);
952 }
953 
954 } // namespace Exp
955 
956 //===----------------------------------------------------------------------===//
957 // MTBUF Format
958 //===----------------------------------------------------------------------===//
959 
960 namespace MTBUFFormat {
961 
962 int64_t getDfmt(const StringRef Name) {
963   for (int Id = DFMT_MIN; Id <= DFMT_MAX; ++Id) {
964     if (Name == DfmtSymbolic[Id])
965       return Id;
966   }
967   return DFMT_UNDEF;
968 }
969 
970 StringRef getDfmtName(unsigned Id) {
971   assert(Id <= DFMT_MAX);
972   return DfmtSymbolic[Id];
973 }
974 
975 static StringLiteral const *getNfmtLookupTable(const MCSubtargetInfo &STI) {
976   if (isSI(STI) || isCI(STI))
977     return NfmtSymbolicSICI;
978   if (isVI(STI) || isGFX9(STI))
979     return NfmtSymbolicVI;
980   return NfmtSymbolicGFX10;
981 }
982 
983 int64_t getNfmt(const StringRef Name, const MCSubtargetInfo &STI) {
984   auto lookupTable = getNfmtLookupTable(STI);
985   for (int Id = NFMT_MIN; Id <= NFMT_MAX; ++Id) {
986     if (Name == lookupTable[Id])
987       return Id;
988   }
989   return NFMT_UNDEF;
990 }
991 
992 StringRef getNfmtName(unsigned Id, const MCSubtargetInfo &STI) {
993   assert(Id <= NFMT_MAX);
994   return getNfmtLookupTable(STI)[Id];
995 }
996 
997 bool isValidDfmtNfmt(unsigned Id, const MCSubtargetInfo &STI) {
998   unsigned Dfmt;
999   unsigned Nfmt;
1000   decodeDfmtNfmt(Id, Dfmt, Nfmt);
1001   return isValidNfmt(Nfmt, STI);
1002 }
1003 
1004 bool isValidNfmt(unsigned Id, const MCSubtargetInfo &STI) {
1005   return !getNfmtName(Id, STI).empty();
1006 }
1007 
1008 int64_t encodeDfmtNfmt(unsigned Dfmt, unsigned Nfmt) {
1009   return (Dfmt << DFMT_SHIFT) | (Nfmt << NFMT_SHIFT);
1010 }
1011 
1012 void decodeDfmtNfmt(unsigned Format, unsigned &Dfmt, unsigned &Nfmt) {
1013   Dfmt = (Format >> DFMT_SHIFT) & DFMT_MASK;
1014   Nfmt = (Format >> NFMT_SHIFT) & NFMT_MASK;
1015 }
1016 
1017 int64_t getUnifiedFormat(const StringRef Name) {
1018   for (int Id = UFMT_FIRST; Id <= UFMT_LAST; ++Id) {
1019     if (Name == UfmtSymbolic[Id])
1020       return Id;
1021   }
1022   return UFMT_UNDEF;
1023 }
1024 
1025 StringRef getUnifiedFormatName(unsigned Id) {
1026   return isValidUnifiedFormat(Id) ? UfmtSymbolic[Id] : "";
1027 }
1028 
1029 bool isValidUnifiedFormat(unsigned Id) {
1030   return Id <= UFMT_LAST;
1031 }
1032 
1033 int64_t convertDfmtNfmt2Ufmt(unsigned Dfmt, unsigned Nfmt) {
1034   int64_t Fmt = encodeDfmtNfmt(Dfmt, Nfmt);
1035   for (int Id = UFMT_FIRST; Id <= UFMT_LAST; ++Id) {
1036     if (Fmt == DfmtNfmt2UFmt[Id])
1037       return Id;
1038   }
1039   return UFMT_UNDEF;
1040 }
1041 
1042 bool isValidFormatEncoding(unsigned Val, const MCSubtargetInfo &STI) {
1043   return isGFX10Plus(STI) ? (Val <= UFMT_MAX) : (Val <= DFMT_NFMT_MAX);
1044 }
1045 
1046 unsigned getDefaultFormatEncoding(const MCSubtargetInfo &STI) {
1047   if (isGFX10Plus(STI))
1048     return UFMT_DEFAULT;
1049   return DFMT_NFMT_DEFAULT;
1050 }
1051 
1052 } // namespace MTBUFFormat
1053 
1054 //===----------------------------------------------------------------------===//
1055 // SendMsg
1056 //===----------------------------------------------------------------------===//
1057 
1058 namespace SendMsg {
1059 
1060 int64_t getMsgId(const StringRef Name) {
1061   for (int i = ID_GAPS_FIRST_; i < ID_GAPS_LAST_; ++i) {
1062     if (IdSymbolic[i] && Name == IdSymbolic[i])
1063       return i;
1064   }
1065   return ID_UNKNOWN_;
1066 }
1067 
1068 static bool isValidMsgId(int64_t MsgId) {
1069   return (ID_GAPS_FIRST_ <= MsgId && MsgId < ID_GAPS_LAST_) && IdSymbolic[MsgId];
1070 }
1071 
1072 bool isValidMsgId(int64_t MsgId, const MCSubtargetInfo &STI, bool Strict) {
1073   if (Strict) {
1074     if (MsgId == ID_GS_ALLOC_REQ || MsgId == ID_GET_DOORBELL)
1075       return isGFX9Plus(STI);
1076     else
1077       return isValidMsgId(MsgId);
1078   } else {
1079     return 0 <= MsgId && isUInt<ID_WIDTH_>(MsgId);
1080   }
1081 }
1082 
1083 StringRef getMsgName(int64_t MsgId) {
1084   return isValidMsgId(MsgId)? IdSymbolic[MsgId] : "";
1085 }
1086 
1087 int64_t getMsgOpId(int64_t MsgId, const StringRef Name) {
1088   const char* const *S = (MsgId == ID_SYSMSG) ? OpSysSymbolic : OpGsSymbolic;
1089   const int F = (MsgId == ID_SYSMSG) ? OP_SYS_FIRST_ : OP_GS_FIRST_;
1090   const int L = (MsgId == ID_SYSMSG) ? OP_SYS_LAST_ : OP_GS_LAST_;
1091   for (int i = F; i < L; ++i) {
1092     if (Name == S[i]) {
1093       return i;
1094     }
1095   }
1096   return OP_UNKNOWN_;
1097 }
1098 
1099 bool isValidMsgOp(int64_t MsgId, int64_t OpId, bool Strict) {
1100 
1101   if (!Strict)
1102     return 0 <= OpId && isUInt<OP_WIDTH_>(OpId);
1103 
1104   switch(MsgId)
1105   {
1106   case ID_GS:
1107     return (OP_GS_FIRST_ <= OpId && OpId < OP_GS_LAST_) && OpId != OP_GS_NOP;
1108   case ID_GS_DONE:
1109     return OP_GS_FIRST_ <= OpId && OpId < OP_GS_LAST_;
1110   case ID_SYSMSG:
1111     return OP_SYS_FIRST_ <= OpId && OpId < OP_SYS_LAST_;
1112   default:
1113     return OpId == OP_NONE_;
1114   }
1115 }
1116 
1117 StringRef getMsgOpName(int64_t MsgId, int64_t OpId) {
1118   assert(msgRequiresOp(MsgId));
1119   return (MsgId == ID_SYSMSG)? OpSysSymbolic[OpId] : OpGsSymbolic[OpId];
1120 }
1121 
1122 bool isValidMsgStream(int64_t MsgId, int64_t OpId, int64_t StreamId, bool Strict) {
1123 
1124   if (!Strict)
1125     return 0 <= StreamId && isUInt<STREAM_ID_WIDTH_>(StreamId);
1126 
1127   switch(MsgId)
1128   {
1129   case ID_GS:
1130     return STREAM_ID_FIRST_ <= StreamId && StreamId < STREAM_ID_LAST_;
1131   case ID_GS_DONE:
1132     return (OpId == OP_GS_NOP)?
1133            (StreamId == STREAM_ID_NONE_) :
1134            (STREAM_ID_FIRST_ <= StreamId && StreamId < STREAM_ID_LAST_);
1135   default:
1136     return StreamId == STREAM_ID_NONE_;
1137   }
1138 }
1139 
1140 bool msgRequiresOp(int64_t MsgId) {
1141   return MsgId == ID_GS || MsgId == ID_GS_DONE || MsgId == ID_SYSMSG;
1142 }
1143 
1144 bool msgSupportsStream(int64_t MsgId, int64_t OpId) {
1145   return (MsgId == ID_GS || MsgId == ID_GS_DONE) && OpId != OP_GS_NOP;
1146 }
1147 
1148 void decodeMsg(unsigned Val,
1149                uint16_t &MsgId,
1150                uint16_t &OpId,
1151                uint16_t &StreamId) {
1152   MsgId = Val & ID_MASK_;
1153   OpId = (Val & OP_MASK_) >> OP_SHIFT_;
1154   StreamId = (Val & STREAM_ID_MASK_) >> STREAM_ID_SHIFT_;
1155 }
1156 
1157 uint64_t encodeMsg(uint64_t MsgId,
1158                    uint64_t OpId,
1159                    uint64_t StreamId) {
1160   return (MsgId << ID_SHIFT_) |
1161          (OpId << OP_SHIFT_) |
1162          (StreamId << STREAM_ID_SHIFT_);
1163 }
1164 
1165 } // namespace SendMsg
1166 
1167 //===----------------------------------------------------------------------===//
1168 //
1169 //===----------------------------------------------------------------------===//
1170 
1171 unsigned getInitialPSInputAddr(const Function &F) {
1172   return getIntegerAttribute(F, "InitialPSInputAddr", 0);
1173 }
1174 
1175 bool isShader(CallingConv::ID cc) {
1176   switch(cc) {
1177     case CallingConv::AMDGPU_VS:
1178     case CallingConv::AMDGPU_LS:
1179     case CallingConv::AMDGPU_HS:
1180     case CallingConv::AMDGPU_ES:
1181     case CallingConv::AMDGPU_GS:
1182     case CallingConv::AMDGPU_PS:
1183     case CallingConv::AMDGPU_CS:
1184       return true;
1185     default:
1186       return false;
1187   }
1188 }
1189 
1190 bool isGraphics(CallingConv::ID cc) {
1191   return isShader(cc) || cc == CallingConv::AMDGPU_Gfx;
1192 }
1193 
1194 bool isCompute(CallingConv::ID cc) {
1195   return !isGraphics(cc) || cc == CallingConv::AMDGPU_CS;
1196 }
1197 
1198 bool isEntryFunctionCC(CallingConv::ID CC) {
1199   switch (CC) {
1200   case CallingConv::AMDGPU_KERNEL:
1201   case CallingConv::SPIR_KERNEL:
1202   case CallingConv::AMDGPU_VS:
1203   case CallingConv::AMDGPU_GS:
1204   case CallingConv::AMDGPU_PS:
1205   case CallingConv::AMDGPU_CS:
1206   case CallingConv::AMDGPU_ES:
1207   case CallingConv::AMDGPU_HS:
1208   case CallingConv::AMDGPU_LS:
1209     return true;
1210   default:
1211     return false;
1212   }
1213 }
1214 
1215 bool isModuleEntryFunctionCC(CallingConv::ID CC) {
1216   switch (CC) {
1217   case CallingConv::AMDGPU_Gfx:
1218     return true;
1219   default:
1220     return isEntryFunctionCC(CC);
1221   }
1222 }
1223 
1224 bool hasXNACK(const MCSubtargetInfo &STI) {
1225   return STI.getFeatureBits()[AMDGPU::FeatureXNACK];
1226 }
1227 
1228 bool hasSRAMECC(const MCSubtargetInfo &STI) {
1229   return STI.getFeatureBits()[AMDGPU::FeatureSRAMECC];
1230 }
1231 
1232 bool hasMIMG_R128(const MCSubtargetInfo &STI) {
1233   return STI.getFeatureBits()[AMDGPU::FeatureMIMG_R128] && !STI.getFeatureBits()[AMDGPU::FeatureR128A16];
1234 }
1235 
1236 bool hasGFX10A16(const MCSubtargetInfo &STI) {
1237   return STI.getFeatureBits()[AMDGPU::FeatureGFX10A16];
1238 }
1239 
1240 bool hasG16(const MCSubtargetInfo &STI) {
1241   return STI.getFeatureBits()[AMDGPU::FeatureG16];
1242 }
1243 
1244 bool hasPackedD16(const MCSubtargetInfo &STI) {
1245   return !STI.getFeatureBits()[AMDGPU::FeatureUnpackedD16VMem];
1246 }
1247 
1248 bool isSI(const MCSubtargetInfo &STI) {
1249   return STI.getFeatureBits()[AMDGPU::FeatureSouthernIslands];
1250 }
1251 
1252 bool isCI(const MCSubtargetInfo &STI) {
1253   return STI.getFeatureBits()[AMDGPU::FeatureSeaIslands];
1254 }
1255 
1256 bool isVI(const MCSubtargetInfo &STI) {
1257   return STI.getFeatureBits()[AMDGPU::FeatureVolcanicIslands];
1258 }
1259 
1260 bool isGFX9(const MCSubtargetInfo &STI) {
1261   return STI.getFeatureBits()[AMDGPU::FeatureGFX9];
1262 }
1263 
1264 bool isGFX9Plus(const MCSubtargetInfo &STI) {
1265   return isGFX9(STI) || isGFX10Plus(STI);
1266 }
1267 
1268 bool isGFX10(const MCSubtargetInfo &STI) {
1269   return STI.getFeatureBits()[AMDGPU::FeatureGFX10];
1270 }
1271 
1272 bool isGFX10Plus(const MCSubtargetInfo &STI) { return isGFX10(STI); }
1273 
1274 bool isGCN3Encoding(const MCSubtargetInfo &STI) {
1275   return STI.getFeatureBits()[AMDGPU::FeatureGCN3Encoding];
1276 }
1277 
1278 bool isGFX10_BEncoding(const MCSubtargetInfo &STI) {
1279   return STI.getFeatureBits()[AMDGPU::FeatureGFX10_BEncoding];
1280 }
1281 
1282 bool hasGFX10_3Insts(const MCSubtargetInfo &STI) {
1283   return STI.getFeatureBits()[AMDGPU::FeatureGFX10_3Insts];
1284 }
1285 
1286 bool isGFX90A(const MCSubtargetInfo &STI) {
1287   return STI.getFeatureBits()[AMDGPU::FeatureGFX90AInsts];
1288 }
1289 
1290 bool isSGPR(unsigned Reg, const MCRegisterInfo* TRI) {
1291   const MCRegisterClass SGPRClass = TRI->getRegClass(AMDGPU::SReg_32RegClassID);
1292   const unsigned FirstSubReg = TRI->getSubReg(Reg, AMDGPU::sub0);
1293   return SGPRClass.contains(FirstSubReg != 0 ? FirstSubReg : Reg) ||
1294     Reg == AMDGPU::SCC;
1295 }
1296 
1297 bool isRegIntersect(unsigned Reg0, unsigned Reg1, const MCRegisterInfo* TRI) {
1298   for (MCRegAliasIterator R(Reg0, TRI, true); R.isValid(); ++R) {
1299     if (*R == Reg1) return true;
1300   }
1301   return false;
1302 }
1303 
1304 #define MAP_REG2REG \
1305   using namespace AMDGPU; \
1306   switch(Reg) { \
1307   default: return Reg; \
1308   CASE_CI_VI(FLAT_SCR) \
1309   CASE_CI_VI(FLAT_SCR_LO) \
1310   CASE_CI_VI(FLAT_SCR_HI) \
1311   CASE_VI_GFX9PLUS(TTMP0) \
1312   CASE_VI_GFX9PLUS(TTMP1) \
1313   CASE_VI_GFX9PLUS(TTMP2) \
1314   CASE_VI_GFX9PLUS(TTMP3) \
1315   CASE_VI_GFX9PLUS(TTMP4) \
1316   CASE_VI_GFX9PLUS(TTMP5) \
1317   CASE_VI_GFX9PLUS(TTMP6) \
1318   CASE_VI_GFX9PLUS(TTMP7) \
1319   CASE_VI_GFX9PLUS(TTMP8) \
1320   CASE_VI_GFX9PLUS(TTMP9) \
1321   CASE_VI_GFX9PLUS(TTMP10) \
1322   CASE_VI_GFX9PLUS(TTMP11) \
1323   CASE_VI_GFX9PLUS(TTMP12) \
1324   CASE_VI_GFX9PLUS(TTMP13) \
1325   CASE_VI_GFX9PLUS(TTMP14) \
1326   CASE_VI_GFX9PLUS(TTMP15) \
1327   CASE_VI_GFX9PLUS(TTMP0_TTMP1) \
1328   CASE_VI_GFX9PLUS(TTMP2_TTMP3) \
1329   CASE_VI_GFX9PLUS(TTMP4_TTMP5) \
1330   CASE_VI_GFX9PLUS(TTMP6_TTMP7) \
1331   CASE_VI_GFX9PLUS(TTMP8_TTMP9) \
1332   CASE_VI_GFX9PLUS(TTMP10_TTMP11) \
1333   CASE_VI_GFX9PLUS(TTMP12_TTMP13) \
1334   CASE_VI_GFX9PLUS(TTMP14_TTMP15) \
1335   CASE_VI_GFX9PLUS(TTMP0_TTMP1_TTMP2_TTMP3) \
1336   CASE_VI_GFX9PLUS(TTMP4_TTMP5_TTMP6_TTMP7) \
1337   CASE_VI_GFX9PLUS(TTMP8_TTMP9_TTMP10_TTMP11) \
1338   CASE_VI_GFX9PLUS(TTMP12_TTMP13_TTMP14_TTMP15) \
1339   CASE_VI_GFX9PLUS(TTMP0_TTMP1_TTMP2_TTMP3_TTMP4_TTMP5_TTMP6_TTMP7) \
1340   CASE_VI_GFX9PLUS(TTMP4_TTMP5_TTMP6_TTMP7_TTMP8_TTMP9_TTMP10_TTMP11) \
1341   CASE_VI_GFX9PLUS(TTMP8_TTMP9_TTMP10_TTMP11_TTMP12_TTMP13_TTMP14_TTMP15) \
1342   CASE_VI_GFX9PLUS(TTMP0_TTMP1_TTMP2_TTMP3_TTMP4_TTMP5_TTMP6_TTMP7_TTMP8_TTMP9_TTMP10_TTMP11_TTMP12_TTMP13_TTMP14_TTMP15) \
1343   }
1344 
1345 #define CASE_CI_VI(node) \
1346   assert(!isSI(STI)); \
1347   case node: return isCI(STI) ? node##_ci : node##_vi;
1348 
1349 #define CASE_VI_GFX9PLUS(node) \
1350   case node: return isGFX9Plus(STI) ? node##_gfx9plus : node##_vi;
1351 
1352 unsigned getMCReg(unsigned Reg, const MCSubtargetInfo &STI) {
1353   if (STI.getTargetTriple().getArch() == Triple::r600)
1354     return Reg;
1355   MAP_REG2REG
1356 }
1357 
1358 #undef CASE_CI_VI
1359 #undef CASE_VI_GFX9PLUS
1360 
1361 #define CASE_CI_VI(node)   case node##_ci: case node##_vi:   return node;
1362 #define CASE_VI_GFX9PLUS(node) case node##_vi: case node##_gfx9plus: return node;
1363 
1364 unsigned mc2PseudoReg(unsigned Reg) {
1365   MAP_REG2REG
1366 }
1367 
1368 #undef CASE_CI_VI
1369 #undef CASE_VI_GFX9PLUS
1370 #undef MAP_REG2REG
1371 
1372 bool isSISrcOperand(const MCInstrDesc &Desc, unsigned OpNo) {
1373   assert(OpNo < Desc.NumOperands);
1374   unsigned OpType = Desc.OpInfo[OpNo].OperandType;
1375   return OpType >= AMDGPU::OPERAND_SRC_FIRST &&
1376          OpType <= AMDGPU::OPERAND_SRC_LAST;
1377 }
1378 
1379 bool isSISrcFPOperand(const MCInstrDesc &Desc, unsigned OpNo) {
1380   assert(OpNo < Desc.NumOperands);
1381   unsigned OpType = Desc.OpInfo[OpNo].OperandType;
1382   switch (OpType) {
1383   case AMDGPU::OPERAND_REG_IMM_FP32:
1384   case AMDGPU::OPERAND_REG_IMM_FP64:
1385   case AMDGPU::OPERAND_REG_IMM_FP16:
1386   case AMDGPU::OPERAND_REG_IMM_V2FP16:
1387   case AMDGPU::OPERAND_REG_IMM_V2INT16:
1388   case AMDGPU::OPERAND_REG_INLINE_C_FP32:
1389   case AMDGPU::OPERAND_REG_INLINE_C_FP64:
1390   case AMDGPU::OPERAND_REG_INLINE_C_FP16:
1391   case AMDGPU::OPERAND_REG_INLINE_C_V2FP16:
1392   case AMDGPU::OPERAND_REG_INLINE_C_V2INT16:
1393   case AMDGPU::OPERAND_REG_INLINE_AC_FP32:
1394   case AMDGPU::OPERAND_REG_INLINE_AC_FP16:
1395   case AMDGPU::OPERAND_REG_INLINE_AC_V2FP16:
1396   case AMDGPU::OPERAND_REG_INLINE_AC_V2INT16:
1397   case AMDGPU::OPERAND_REG_IMM_V2FP32:
1398   case AMDGPU::OPERAND_REG_INLINE_C_V2FP32:
1399   case AMDGPU::OPERAND_REG_INLINE_AC_FP64:
1400     return true;
1401   default:
1402     return false;
1403   }
1404 }
1405 
1406 bool isSISrcInlinableOperand(const MCInstrDesc &Desc, unsigned OpNo) {
1407   assert(OpNo < Desc.NumOperands);
1408   unsigned OpType = Desc.OpInfo[OpNo].OperandType;
1409   return OpType >= AMDGPU::OPERAND_REG_INLINE_C_FIRST &&
1410          OpType <= AMDGPU::OPERAND_REG_INLINE_C_LAST;
1411 }
1412 
1413 // Avoid using MCRegisterClass::getSize, since that function will go away
1414 // (move from MC* level to Target* level). Return size in bits.
1415 unsigned getRegBitWidth(unsigned RCID) {
1416   switch (RCID) {
1417   case AMDGPU::VGPR_LO16RegClassID:
1418   case AMDGPU::VGPR_HI16RegClassID:
1419   case AMDGPU::SGPR_LO16RegClassID:
1420   case AMDGPU::AGPR_LO16RegClassID:
1421     return 16;
1422   case AMDGPU::SGPR_32RegClassID:
1423   case AMDGPU::VGPR_32RegClassID:
1424   case AMDGPU::VRegOrLds_32RegClassID:
1425   case AMDGPU::AGPR_32RegClassID:
1426   case AMDGPU::VS_32RegClassID:
1427   case AMDGPU::AV_32RegClassID:
1428   case AMDGPU::SReg_32RegClassID:
1429   case AMDGPU::SReg_32_XM0RegClassID:
1430   case AMDGPU::SRegOrLds_32RegClassID:
1431     return 32;
1432   case AMDGPU::SGPR_64RegClassID:
1433   case AMDGPU::VS_64RegClassID:
1434   case AMDGPU::AV_64RegClassID:
1435   case AMDGPU::SReg_64RegClassID:
1436   case AMDGPU::VReg_64RegClassID:
1437   case AMDGPU::AReg_64RegClassID:
1438   case AMDGPU::SReg_64_XEXECRegClassID:
1439     return 64;
1440   case AMDGPU::SGPR_96RegClassID:
1441   case AMDGPU::SReg_96RegClassID:
1442   case AMDGPU::VReg_96RegClassID:
1443   case AMDGPU::AReg_96RegClassID:
1444   case AMDGPU::AV_96RegClassID:
1445     return 96;
1446   case AMDGPU::SGPR_128RegClassID:
1447   case AMDGPU::SReg_128RegClassID:
1448   case AMDGPU::VReg_128RegClassID:
1449   case AMDGPU::AReg_128RegClassID:
1450   case AMDGPU::AV_128RegClassID:
1451     return 128;
1452   case AMDGPU::SGPR_160RegClassID:
1453   case AMDGPU::SReg_160RegClassID:
1454   case AMDGPU::VReg_160RegClassID:
1455   case AMDGPU::AReg_160RegClassID:
1456   case AMDGPU::AV_160RegClassID:
1457     return 160;
1458   case AMDGPU::SGPR_192RegClassID:
1459   case AMDGPU::SReg_192RegClassID:
1460   case AMDGPU::VReg_192RegClassID:
1461   case AMDGPU::AReg_192RegClassID:
1462     return 192;
1463   case AMDGPU::SGPR_256RegClassID:
1464   case AMDGPU::SReg_256RegClassID:
1465   case AMDGPU::VReg_256RegClassID:
1466   case AMDGPU::AReg_256RegClassID:
1467     return 256;
1468   case AMDGPU::SGPR_512RegClassID:
1469   case AMDGPU::SReg_512RegClassID:
1470   case AMDGPU::VReg_512RegClassID:
1471   case AMDGPU::AReg_512RegClassID:
1472     return 512;
1473   case AMDGPU::SGPR_1024RegClassID:
1474   case AMDGPU::SReg_1024RegClassID:
1475   case AMDGPU::VReg_1024RegClassID:
1476   case AMDGPU::AReg_1024RegClassID:
1477     return 1024;
1478   default:
1479     llvm_unreachable("Unexpected register class");
1480   }
1481 }
1482 
1483 unsigned getRegBitWidth(const MCRegisterClass &RC) {
1484   return getRegBitWidth(RC.getID());
1485 }
1486 
1487 unsigned getRegOperandSize(const MCRegisterInfo *MRI, const MCInstrDesc &Desc,
1488                            unsigned OpNo) {
1489   assert(OpNo < Desc.NumOperands);
1490   unsigned RCID = Desc.OpInfo[OpNo].RegClass;
1491   return getRegBitWidth(MRI->getRegClass(RCID)) / 8;
1492 }
1493 
1494 bool isInlinableLiteral64(int64_t Literal, bool HasInv2Pi) {
1495   if (isInlinableIntLiteral(Literal))
1496     return true;
1497 
1498   uint64_t Val = static_cast<uint64_t>(Literal);
1499   return (Val == DoubleToBits(0.0)) ||
1500          (Val == DoubleToBits(1.0)) ||
1501          (Val == DoubleToBits(-1.0)) ||
1502          (Val == DoubleToBits(0.5)) ||
1503          (Val == DoubleToBits(-0.5)) ||
1504          (Val == DoubleToBits(2.0)) ||
1505          (Val == DoubleToBits(-2.0)) ||
1506          (Val == DoubleToBits(4.0)) ||
1507          (Val == DoubleToBits(-4.0)) ||
1508          (Val == 0x3fc45f306dc9c882 && HasInv2Pi);
1509 }
1510 
1511 bool isInlinableLiteral32(int32_t Literal, bool HasInv2Pi) {
1512   if (isInlinableIntLiteral(Literal))
1513     return true;
1514 
1515   // The actual type of the operand does not seem to matter as long
1516   // as the bits match one of the inline immediate values.  For example:
1517   //
1518   // -nan has the hexadecimal encoding of 0xfffffffe which is -2 in decimal,
1519   // so it is a legal inline immediate.
1520   //
1521   // 1065353216 has the hexadecimal encoding 0x3f800000 which is 1.0f in
1522   // floating-point, so it is a legal inline immediate.
1523 
1524   uint32_t Val = static_cast<uint32_t>(Literal);
1525   return (Val == FloatToBits(0.0f)) ||
1526          (Val == FloatToBits(1.0f)) ||
1527          (Val == FloatToBits(-1.0f)) ||
1528          (Val == FloatToBits(0.5f)) ||
1529          (Val == FloatToBits(-0.5f)) ||
1530          (Val == FloatToBits(2.0f)) ||
1531          (Val == FloatToBits(-2.0f)) ||
1532          (Val == FloatToBits(4.0f)) ||
1533          (Val == FloatToBits(-4.0f)) ||
1534          (Val == 0x3e22f983 && HasInv2Pi);
1535 }
1536 
1537 bool isInlinableLiteral16(int16_t Literal, bool HasInv2Pi) {
1538   if (!HasInv2Pi)
1539     return false;
1540 
1541   if (isInlinableIntLiteral(Literal))
1542     return true;
1543 
1544   uint16_t Val = static_cast<uint16_t>(Literal);
1545   return Val == 0x3C00 || // 1.0
1546          Val == 0xBC00 || // -1.0
1547          Val == 0x3800 || // 0.5
1548          Val == 0xB800 || // -0.5
1549          Val == 0x4000 || // 2.0
1550          Val == 0xC000 || // -2.0
1551          Val == 0x4400 || // 4.0
1552          Val == 0xC400 || // -4.0
1553          Val == 0x3118;   // 1/2pi
1554 }
1555 
1556 bool isInlinableLiteralV216(int32_t Literal, bool HasInv2Pi) {
1557   assert(HasInv2Pi);
1558 
1559   if (isInt<16>(Literal) || isUInt<16>(Literal)) {
1560     int16_t Trunc = static_cast<int16_t>(Literal);
1561     return AMDGPU::isInlinableLiteral16(Trunc, HasInv2Pi);
1562   }
1563   if (!(Literal & 0xffff))
1564     return AMDGPU::isInlinableLiteral16(Literal >> 16, HasInv2Pi);
1565 
1566   int16_t Lo16 = static_cast<int16_t>(Literal);
1567   int16_t Hi16 = static_cast<int16_t>(Literal >> 16);
1568   return Lo16 == Hi16 && isInlinableLiteral16(Lo16, HasInv2Pi);
1569 }
1570 
1571 bool isInlinableIntLiteralV216(int32_t Literal) {
1572   int16_t Lo16 = static_cast<int16_t>(Literal);
1573   if (isInt<16>(Literal) || isUInt<16>(Literal))
1574     return isInlinableIntLiteral(Lo16);
1575 
1576   int16_t Hi16 = static_cast<int16_t>(Literal >> 16);
1577   if (!(Literal & 0xffff))
1578     return isInlinableIntLiteral(Hi16);
1579   return Lo16 == Hi16 && isInlinableIntLiteral(Lo16);
1580 }
1581 
1582 bool isFoldableLiteralV216(int32_t Literal, bool HasInv2Pi) {
1583   assert(HasInv2Pi);
1584 
1585   int16_t Lo16 = static_cast<int16_t>(Literal);
1586   if (isInt<16>(Literal) || isUInt<16>(Literal))
1587     return true;
1588 
1589   int16_t Hi16 = static_cast<int16_t>(Literal >> 16);
1590   if (!(Literal & 0xffff))
1591     return true;
1592   return Lo16 == Hi16;
1593 }
1594 
1595 bool isArgPassedInSGPR(const Argument *A) {
1596   const Function *F = A->getParent();
1597 
1598   // Arguments to compute shaders are never a source of divergence.
1599   CallingConv::ID CC = F->getCallingConv();
1600   switch (CC) {
1601   case CallingConv::AMDGPU_KERNEL:
1602   case CallingConv::SPIR_KERNEL:
1603     return true;
1604   case CallingConv::AMDGPU_VS:
1605   case CallingConv::AMDGPU_LS:
1606   case CallingConv::AMDGPU_HS:
1607   case CallingConv::AMDGPU_ES:
1608   case CallingConv::AMDGPU_GS:
1609   case CallingConv::AMDGPU_PS:
1610   case CallingConv::AMDGPU_CS:
1611   case CallingConv::AMDGPU_Gfx:
1612     // For non-compute shaders, SGPR inputs are marked with either inreg or byval.
1613     // Everything else is in VGPRs.
1614     return F->getAttributes().hasParamAttribute(A->getArgNo(), Attribute::InReg) ||
1615            F->getAttributes().hasParamAttribute(A->getArgNo(), Attribute::ByVal);
1616   default:
1617     // TODO: Should calls support inreg for SGPR inputs?
1618     return false;
1619   }
1620 }
1621 
1622 static bool hasSMEMByteOffset(const MCSubtargetInfo &ST) {
1623   return isGCN3Encoding(ST) || isGFX10Plus(ST);
1624 }
1625 
1626 static bool hasSMRDSignedImmOffset(const MCSubtargetInfo &ST) {
1627   return isGFX9Plus(ST);
1628 }
1629 
1630 bool isLegalSMRDEncodedUnsignedOffset(const MCSubtargetInfo &ST,
1631                                       int64_t EncodedOffset) {
1632   return hasSMEMByteOffset(ST) ? isUInt<20>(EncodedOffset)
1633                                : isUInt<8>(EncodedOffset);
1634 }
1635 
1636 bool isLegalSMRDEncodedSignedOffset(const MCSubtargetInfo &ST,
1637                                     int64_t EncodedOffset,
1638                                     bool IsBuffer) {
1639   return !IsBuffer &&
1640          hasSMRDSignedImmOffset(ST) &&
1641          isInt<21>(EncodedOffset);
1642 }
1643 
1644 static bool isDwordAligned(uint64_t ByteOffset) {
1645   return (ByteOffset & 3) == 0;
1646 }
1647 
1648 uint64_t convertSMRDOffsetUnits(const MCSubtargetInfo &ST,
1649                                 uint64_t ByteOffset) {
1650   if (hasSMEMByteOffset(ST))
1651     return ByteOffset;
1652 
1653   assert(isDwordAligned(ByteOffset));
1654   return ByteOffset >> 2;
1655 }
1656 
1657 Optional<int64_t> getSMRDEncodedOffset(const MCSubtargetInfo &ST,
1658                                        int64_t ByteOffset, bool IsBuffer) {
1659   // The signed version is always a byte offset.
1660   if (!IsBuffer && hasSMRDSignedImmOffset(ST)) {
1661     assert(hasSMEMByteOffset(ST));
1662     return isInt<20>(ByteOffset) ? Optional<int64_t>(ByteOffset) : None;
1663   }
1664 
1665   if (!isDwordAligned(ByteOffset) && !hasSMEMByteOffset(ST))
1666     return None;
1667 
1668   int64_t EncodedOffset = convertSMRDOffsetUnits(ST, ByteOffset);
1669   return isLegalSMRDEncodedUnsignedOffset(ST, EncodedOffset)
1670              ? Optional<int64_t>(EncodedOffset)
1671              : None;
1672 }
1673 
1674 Optional<int64_t> getSMRDEncodedLiteralOffset32(const MCSubtargetInfo &ST,
1675                                                 int64_t ByteOffset) {
1676   if (!isCI(ST) || !isDwordAligned(ByteOffset))
1677     return None;
1678 
1679   int64_t EncodedOffset = convertSMRDOffsetUnits(ST, ByteOffset);
1680   return isUInt<32>(EncodedOffset) ? Optional<int64_t>(EncodedOffset) : None;
1681 }
1682 
1683 unsigned getNumFlatOffsetBits(const MCSubtargetInfo &ST, bool Signed) {
1684   // Address offset is 12-bit signed for GFX10, 13-bit for GFX9.
1685   if (AMDGPU::isGFX10(ST))
1686     return Signed ? 12 : 11;
1687 
1688   return Signed ? 13 : 12;
1689 }
1690 
1691 // Given Imm, split it into the values to put into the SOffset and ImmOffset
1692 // fields in an MUBUF instruction. Return false if it is not possible (due to a
1693 // hardware bug needing a workaround).
1694 //
1695 // The required alignment ensures that individual address components remain
1696 // aligned if they are aligned to begin with. It also ensures that additional
1697 // offsets within the given alignment can be added to the resulting ImmOffset.
1698 bool splitMUBUFOffset(uint32_t Imm, uint32_t &SOffset, uint32_t &ImmOffset,
1699                       const GCNSubtarget *Subtarget, Align Alignment) {
1700   const uint32_t MaxImm = alignDown(4095, Alignment.value());
1701   uint32_t Overflow = 0;
1702 
1703   if (Imm > MaxImm) {
1704     if (Imm <= MaxImm + 64) {
1705       // Use an SOffset inline constant for 4..64
1706       Overflow = Imm - MaxImm;
1707       Imm = MaxImm;
1708     } else {
1709       // Try to keep the same value in SOffset for adjacent loads, so that
1710       // the corresponding register contents can be re-used.
1711       //
1712       // Load values with all low-bits (except for alignment bits) set into
1713       // SOffset, so that a larger range of values can be covered using
1714       // s_movk_i32.
1715       //
1716       // Atomic operations fail to work correctly when individual address
1717       // components are unaligned, even if their sum is aligned.
1718       uint32_t High = (Imm + Alignment.value()) & ~4095;
1719       uint32_t Low = (Imm + Alignment.value()) & 4095;
1720       Imm = Low;
1721       Overflow = High - Alignment.value();
1722     }
1723   }
1724 
1725   // There is a hardware bug in SI and CI which prevents address clamping in
1726   // MUBUF instructions from working correctly with SOffsets. The immediate
1727   // offset is unaffected.
1728   if (Overflow > 0 &&
1729       Subtarget->getGeneration() <= AMDGPUSubtarget::SEA_ISLANDS)
1730     return false;
1731 
1732   ImmOffset = Imm;
1733   SOffset = Overflow;
1734   return true;
1735 }
1736 
1737 SIModeRegisterDefaults::SIModeRegisterDefaults(const Function &F) {
1738   *this = getDefaultForCallingConv(F.getCallingConv());
1739 
1740   StringRef IEEEAttr = F.getFnAttribute("amdgpu-ieee").getValueAsString();
1741   if (!IEEEAttr.empty())
1742     IEEE = IEEEAttr == "true";
1743 
1744   StringRef DX10ClampAttr
1745     = F.getFnAttribute("amdgpu-dx10-clamp").getValueAsString();
1746   if (!DX10ClampAttr.empty())
1747     DX10Clamp = DX10ClampAttr == "true";
1748 
1749   StringRef DenormF32Attr = F.getFnAttribute("denormal-fp-math-f32").getValueAsString();
1750   if (!DenormF32Attr.empty()) {
1751     DenormalMode DenormMode = parseDenormalFPAttribute(DenormF32Attr);
1752     FP32InputDenormals = DenormMode.Input == DenormalMode::IEEE;
1753     FP32OutputDenormals = DenormMode.Output == DenormalMode::IEEE;
1754   }
1755 
1756   StringRef DenormAttr = F.getFnAttribute("denormal-fp-math").getValueAsString();
1757   if (!DenormAttr.empty()) {
1758     DenormalMode DenormMode = parseDenormalFPAttribute(DenormAttr);
1759 
1760     if (DenormF32Attr.empty()) {
1761       FP32InputDenormals = DenormMode.Input == DenormalMode::IEEE;
1762       FP32OutputDenormals = DenormMode.Output == DenormalMode::IEEE;
1763     }
1764 
1765     FP64FP16InputDenormals = DenormMode.Input == DenormalMode::IEEE;
1766     FP64FP16OutputDenormals = DenormMode.Output == DenormalMode::IEEE;
1767   }
1768 }
1769 
1770 namespace {
1771 
1772 struct SourceOfDivergence {
1773   unsigned Intr;
1774 };
1775 const SourceOfDivergence *lookupSourceOfDivergence(unsigned Intr);
1776 
1777 #define GET_SourcesOfDivergence_IMPL
1778 #define GET_Gfx9BufferFormat_IMPL
1779 #define GET_Gfx10PlusBufferFormat_IMPL
1780 #include "AMDGPUGenSearchableTables.inc"
1781 
1782 } // end anonymous namespace
1783 
1784 bool isIntrinsicSourceOfDivergence(unsigned IntrID) {
1785   return lookupSourceOfDivergence(IntrID);
1786 }
1787 
1788 const GcnBufferFormatInfo *getGcnBufferFormatInfo(uint8_t BitsPerComp,
1789                                                   uint8_t NumComponents,
1790                                                   uint8_t NumFormat,
1791                                                   const MCSubtargetInfo &STI) {
1792   return isGFX10Plus(STI)
1793              ? getGfx10PlusBufferFormatInfo(BitsPerComp, NumComponents,
1794                                             NumFormat)
1795              : getGfx9BufferFormatInfo(BitsPerComp, NumComponents, NumFormat);
1796 }
1797 
1798 const GcnBufferFormatInfo *getGcnBufferFormatInfo(uint8_t Format,
1799                                                   const MCSubtargetInfo &STI) {
1800   return isGFX10Plus(STI) ? getGfx10PlusBufferFormatInfo(Format)
1801                           : getGfx9BufferFormatInfo(Format);
1802 }
1803 
1804 } // namespace AMDGPU
1805 
1806 raw_ostream &operator<<(raw_ostream &OS,
1807                         const AMDGPU::IsaInfo::TargetIDSetting S) {
1808   switch (S) {
1809   case (AMDGPU::IsaInfo::TargetIDSetting::Unsupported):
1810     OS << "Unsupported";
1811     break;
1812   case (AMDGPU::IsaInfo::TargetIDSetting::Any):
1813     OS << "Any";
1814     break;
1815   case (AMDGPU::IsaInfo::TargetIDSetting::Off):
1816     OS << "Off";
1817     break;
1818   case (AMDGPU::IsaInfo::TargetIDSetting::On):
1819     OS << "On";
1820     break;
1821   }
1822   return OS;
1823 }
1824 
1825 } // namespace llvm
1826