10b57cec5SDimitry Andric //===- AMDGPUBaseInfo.cpp - AMDGPU Base encoding information --------------===//
20b57cec5SDimitry Andric //
30b57cec5SDimitry Andric // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
40b57cec5SDimitry Andric // See https://llvm.org/LICENSE.txt for license information.
50b57cec5SDimitry Andric // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
60b57cec5SDimitry Andric //
70b57cec5SDimitry Andric //===----------------------------------------------------------------------===//
80b57cec5SDimitry Andric 
90b57cec5SDimitry Andric #include "AMDGPUBaseInfo.h"
100b57cec5SDimitry Andric #include "AMDGPU.h"
110b57cec5SDimitry Andric #include "AMDGPUAsmUtils.h"
12e8d8bef9SDimitry Andric #include "AMDKernelCodeT.h"
13e8d8bef9SDimitry Andric #include "MCTargetDesc/AMDGPUMCTargetDesc.h"
140b57cec5SDimitry Andric #include "llvm/BinaryFormat/ELF.h"
150b57cec5SDimitry Andric #include "llvm/IR/Attributes.h"
1606c3fb27SDimitry Andric #include "llvm/IR/Constants.h"
170b57cec5SDimitry Andric #include "llvm/IR/Function.h"
180b57cec5SDimitry Andric #include "llvm/IR/GlobalValue.h"
19480093f4SDimitry Andric #include "llvm/IR/IntrinsicsAMDGPU.h"
20480093f4SDimitry Andric #include "llvm/IR/IntrinsicsR600.h"
210b57cec5SDimitry Andric #include "llvm/IR/LLVMContext.h"
2206c3fb27SDimitry Andric #include "llvm/MC/MCInstrInfo.h"
2306c3fb27SDimitry Andric #include "llvm/MC/MCRegisterInfo.h"
240b57cec5SDimitry Andric #include "llvm/MC/MCSubtargetInfo.h"
25e8d8bef9SDimitry Andric #include "llvm/Support/AMDHSAKernelDescriptor.h"
26e8d8bef9SDimitry Andric #include "llvm/Support/CommandLine.h"
2706c3fb27SDimitry Andric #include "llvm/TargetParser/TargetParser.h"
28bdd1243dSDimitry Andric #include <optional>
290b57cec5SDimitry Andric 
300b57cec5SDimitry Andric #define GET_INSTRINFO_NAMED_OPS
310b57cec5SDimitry Andric #define GET_INSTRMAP_INFO
320b57cec5SDimitry Andric #include "AMDGPUGenInstrInfo.inc"
33e8d8bef9SDimitry Andric 
347a6dacacSDimitry Andric static llvm::cl::opt<unsigned> DefaultAMDHSACodeObjectVersion(
357a6dacacSDimitry Andric     "amdhsa-code-object-version", llvm::cl::Hidden,
367a6dacacSDimitry Andric     llvm::cl::init(llvm::AMDGPU::AMDHSA_COV5),
377a6dacacSDimitry Andric     llvm::cl::desc("Set default AMDHSA Code Object Version (module flag "
387a6dacacSDimitry Andric                    "or asm directive still take priority if present)"));
3981ad6265SDimitry Andric 
400b57cec5SDimitry Andric namespace {
410b57cec5SDimitry Andric 
420b57cec5SDimitry Andric /// \returns Bit mask for given bit \p Shift and bit \p Width.
getBitMask(unsigned Shift,unsigned Width)430b57cec5SDimitry Andric unsigned getBitMask(unsigned Shift, unsigned Width) {
440b57cec5SDimitry Andric   return ((1 << Width) - 1) << Shift;
450b57cec5SDimitry Andric }
460b57cec5SDimitry Andric 
470b57cec5SDimitry Andric /// Packs \p Src into \p Dst for given bit \p Shift and bit \p Width.
480b57cec5SDimitry Andric ///
490b57cec5SDimitry Andric /// \returns Packed \p Dst.
packBits(unsigned Src,unsigned Dst,unsigned Shift,unsigned Width)500b57cec5SDimitry Andric unsigned packBits(unsigned Src, unsigned Dst, unsigned Shift, unsigned Width) {
5181ad6265SDimitry Andric   unsigned Mask = getBitMask(Shift, Width);
5281ad6265SDimitry Andric   return ((Src << Shift) & Mask) | (Dst & ~Mask);
530b57cec5SDimitry Andric }
540b57cec5SDimitry Andric 
550b57cec5SDimitry Andric /// Unpacks bits from \p Src for given bit \p Shift and bit \p Width.
560b57cec5SDimitry Andric ///
570b57cec5SDimitry Andric /// \returns Unpacked bits.
unpackBits(unsigned Src,unsigned Shift,unsigned Width)580b57cec5SDimitry Andric unsigned unpackBits(unsigned Src, unsigned Shift, unsigned Width) {
590b57cec5SDimitry Andric   return (Src & getBitMask(Shift, Width)) >> Shift;
600b57cec5SDimitry Andric }
610b57cec5SDimitry Andric 
620b57cec5SDimitry Andric /// \returns Vmcnt bit shift (lower bits).
getVmcntBitShiftLo(unsigned VersionMajor)6381ad6265SDimitry Andric unsigned getVmcntBitShiftLo(unsigned VersionMajor) {
6481ad6265SDimitry Andric   return VersionMajor >= 11 ? 10 : 0;
6581ad6265SDimitry Andric }
660b57cec5SDimitry Andric 
670b57cec5SDimitry Andric /// \returns Vmcnt bit width (lower bits).
getVmcntBitWidthLo(unsigned VersionMajor)6881ad6265SDimitry Andric unsigned getVmcntBitWidthLo(unsigned VersionMajor) {
6981ad6265SDimitry Andric   return VersionMajor >= 11 ? 6 : 4;
7081ad6265SDimitry Andric }
710b57cec5SDimitry Andric 
720b57cec5SDimitry Andric /// \returns Expcnt bit shift.
getExpcntBitShift(unsigned VersionMajor)7381ad6265SDimitry Andric unsigned getExpcntBitShift(unsigned VersionMajor) {
7481ad6265SDimitry Andric   return VersionMajor >= 11 ? 0 : 4;
7581ad6265SDimitry Andric }
760b57cec5SDimitry Andric 
770b57cec5SDimitry Andric /// \returns Expcnt bit width.
getExpcntBitWidth(unsigned VersionMajor)7881ad6265SDimitry Andric unsigned getExpcntBitWidth(unsigned VersionMajor) { return 3; }
790b57cec5SDimitry Andric 
800b57cec5SDimitry Andric /// \returns Lgkmcnt bit shift.
getLgkmcntBitShift(unsigned VersionMajor)8181ad6265SDimitry Andric unsigned getLgkmcntBitShift(unsigned VersionMajor) {
8281ad6265SDimitry Andric   return VersionMajor >= 11 ? 4 : 8;
8381ad6265SDimitry Andric }
840b57cec5SDimitry Andric 
850b57cec5SDimitry Andric /// \returns Lgkmcnt bit width.
getLgkmcntBitWidth(unsigned VersionMajor)860b57cec5SDimitry Andric unsigned getLgkmcntBitWidth(unsigned VersionMajor) {
8781ad6265SDimitry Andric   return VersionMajor >= 10 ? 6 : 4;
880b57cec5SDimitry Andric }
890b57cec5SDimitry Andric 
900b57cec5SDimitry Andric /// \returns Vmcnt bit shift (higher bits).
getVmcntBitShiftHi(unsigned VersionMajor)9181ad6265SDimitry Andric unsigned getVmcntBitShiftHi(unsigned VersionMajor) { return 14; }
920b57cec5SDimitry Andric 
930b57cec5SDimitry Andric /// \returns Vmcnt bit width (higher bits).
getVmcntBitWidthHi(unsigned VersionMajor)9481ad6265SDimitry Andric unsigned getVmcntBitWidthHi(unsigned VersionMajor) {
9581ad6265SDimitry Andric   return (VersionMajor == 9 || VersionMajor == 10) ? 2 : 0;
9681ad6265SDimitry Andric }
970b57cec5SDimitry Andric 
987a6dacacSDimitry Andric /// \returns Loadcnt bit width
getLoadcntBitWidth(unsigned VersionMajor)997a6dacacSDimitry Andric unsigned getLoadcntBitWidth(unsigned VersionMajor) {
1007a6dacacSDimitry Andric   return VersionMajor >= 12 ? 6 : 0;
1017a6dacacSDimitry Andric }
1027a6dacacSDimitry Andric 
1037a6dacacSDimitry Andric /// \returns Samplecnt bit width.
getSamplecntBitWidth(unsigned VersionMajor)1047a6dacacSDimitry Andric unsigned getSamplecntBitWidth(unsigned VersionMajor) {
1057a6dacacSDimitry Andric   return VersionMajor >= 12 ? 6 : 0;
1067a6dacacSDimitry Andric }
1077a6dacacSDimitry Andric 
1087a6dacacSDimitry Andric /// \returns Bvhcnt bit width.
getBvhcntBitWidth(unsigned VersionMajor)1097a6dacacSDimitry Andric unsigned getBvhcntBitWidth(unsigned VersionMajor) {
1107a6dacacSDimitry Andric   return VersionMajor >= 12 ? 3 : 0;
1117a6dacacSDimitry Andric }
1127a6dacacSDimitry Andric 
1137a6dacacSDimitry Andric /// \returns Dscnt bit width.
getDscntBitWidth(unsigned VersionMajor)1147a6dacacSDimitry Andric unsigned getDscntBitWidth(unsigned VersionMajor) {
1157a6dacacSDimitry Andric   return VersionMajor >= 12 ? 6 : 0;
1167a6dacacSDimitry Andric }
1177a6dacacSDimitry Andric 
1187a6dacacSDimitry Andric /// \returns Dscnt bit shift in combined S_WAIT instructions.
getDscntBitShift(unsigned VersionMajor)1197a6dacacSDimitry Andric unsigned getDscntBitShift(unsigned VersionMajor) { return 0; }
1207a6dacacSDimitry Andric 
1217a6dacacSDimitry Andric /// \returns Storecnt or Vscnt bit width, depending on VersionMajor.
getStorecntBitWidth(unsigned VersionMajor)1227a6dacacSDimitry Andric unsigned getStorecntBitWidth(unsigned VersionMajor) {
1237a6dacacSDimitry Andric   return VersionMajor >= 10 ? 6 : 0;
1247a6dacacSDimitry Andric }
1257a6dacacSDimitry Andric 
1267a6dacacSDimitry Andric /// \returns Kmcnt bit width.
getKmcntBitWidth(unsigned VersionMajor)1277a6dacacSDimitry Andric unsigned getKmcntBitWidth(unsigned VersionMajor) {
1287a6dacacSDimitry Andric   return VersionMajor >= 12 ? 5 : 0;
1297a6dacacSDimitry Andric }
1307a6dacacSDimitry Andric 
1317a6dacacSDimitry Andric /// \returns shift for Loadcnt/Storecnt in combined S_WAIT instructions.
getLoadcntStorecntBitShift(unsigned VersionMajor)1327a6dacacSDimitry Andric unsigned getLoadcntStorecntBitShift(unsigned VersionMajor) {
1337a6dacacSDimitry Andric   return VersionMajor >= 12 ? 8 : 0;
1347a6dacacSDimitry Andric }
1357a6dacacSDimitry Andric 
13606c3fb27SDimitry Andric /// \returns VmVsrc bit width
getVmVsrcBitWidth()13706c3fb27SDimitry Andric inline unsigned getVmVsrcBitWidth() { return 3; }
13806c3fb27SDimitry Andric 
13906c3fb27SDimitry Andric /// \returns VmVsrc bit shift
getVmVsrcBitShift()14006c3fb27SDimitry Andric inline unsigned getVmVsrcBitShift() { return 2; }
14106c3fb27SDimitry Andric 
14206c3fb27SDimitry Andric /// \returns VaVdst bit width
getVaVdstBitWidth()14306c3fb27SDimitry Andric inline unsigned getVaVdstBitWidth() { return 4; }
14406c3fb27SDimitry Andric 
14506c3fb27SDimitry Andric /// \returns VaVdst bit shift
getVaVdstBitShift()14606c3fb27SDimitry Andric inline unsigned getVaVdstBitShift() { return 12; }
14706c3fb27SDimitry Andric 
14806c3fb27SDimitry Andric /// \returns SaSdst bit width
getSaSdstBitWidth()14906c3fb27SDimitry Andric inline unsigned getSaSdstBitWidth() { return 1; }
15006c3fb27SDimitry Andric 
15106c3fb27SDimitry Andric /// \returns SaSdst bit shift
getSaSdstBitShift()15206c3fb27SDimitry Andric inline unsigned getSaSdstBitShift() { return 0; }
15306c3fb27SDimitry Andric 
1540b57cec5SDimitry Andric } // end namespace anonymous
1550b57cec5SDimitry Andric 
1560b57cec5SDimitry Andric namespace llvm {
1570b57cec5SDimitry Andric 
1580b57cec5SDimitry Andric namespace AMDGPU {
1590b57cec5SDimitry Andric 
1605f757f3fSDimitry Andric /// \returns True if \p STI is AMDHSA.
isHsaAbi(const MCSubtargetInfo & STI)1615f757f3fSDimitry Andric bool isHsaAbi(const MCSubtargetInfo &STI) {
1625f757f3fSDimitry Andric   return STI.getTargetTriple().getOS() == Triple::AMDHSA;
1635f757f3fSDimitry Andric }
1645f757f3fSDimitry Andric 
getAMDHSACodeObjectVersion(const Module & M)1657a6dacacSDimitry Andric unsigned getAMDHSACodeObjectVersion(const Module &M) {
16606c3fb27SDimitry Andric   if (auto Ver = mdconst::extract_or_null<ConstantInt>(
16706c3fb27SDimitry Andric           M.getModuleFlag("amdgpu_code_object_version"))) {
16806c3fb27SDimitry Andric     return (unsigned)Ver->getZExtValue() / 100;
16906c3fb27SDimitry Andric   }
17006c3fb27SDimitry Andric 
1717a6dacacSDimitry Andric   return getDefaultAMDHSACodeObjectVersion();
1727a6dacacSDimitry Andric }
1737a6dacacSDimitry Andric 
getDefaultAMDHSACodeObjectVersion()1747a6dacacSDimitry Andric unsigned getDefaultAMDHSACodeObjectVersion() {
1757a6dacacSDimitry Andric   return DefaultAMDHSACodeObjectVersion;
1767a6dacacSDimitry Andric }
1777a6dacacSDimitry Andric 
getELFABIVersion(const Triple & T,unsigned CodeObjectVersion)1787a6dacacSDimitry Andric uint8_t getELFABIVersion(const Triple &T, unsigned CodeObjectVersion) {
1797a6dacacSDimitry Andric   if (T.getOS() != Triple::AMDHSA)
1807a6dacacSDimitry Andric     return 0;
1817a6dacacSDimitry Andric 
1827a6dacacSDimitry Andric   switch (CodeObjectVersion) {
1837a6dacacSDimitry Andric   case 4:
1847a6dacacSDimitry Andric     return ELF::ELFABIVERSION_AMDGPU_HSA_V4;
1857a6dacacSDimitry Andric   case 5:
1867a6dacacSDimitry Andric     return ELF::ELFABIVERSION_AMDGPU_HSA_V5;
1877a6dacacSDimitry Andric   default:
1887a6dacacSDimitry Andric     report_fatal_error("Unsupported AMDHSA Code Object Version " +
1897a6dacacSDimitry Andric                        Twine(CodeObjectVersion));
1907a6dacacSDimitry Andric   }
19106c3fb27SDimitry Andric }
19206c3fb27SDimitry Andric 
getMultigridSyncArgImplicitArgPosition(unsigned CodeObjectVersion)19306c3fb27SDimitry Andric unsigned getMultigridSyncArgImplicitArgPosition(unsigned CodeObjectVersion) {
19406c3fb27SDimitry Andric   switch (CodeObjectVersion) {
19506c3fb27SDimitry Andric   case AMDHSA_COV4:
19681ad6265SDimitry Andric     return 48;
19706c3fb27SDimitry Andric   case AMDHSA_COV5:
19881ad6265SDimitry Andric   default:
19906c3fb27SDimitry Andric     return AMDGPU::ImplicitArg::MULTIGRID_SYNC_ARG_OFFSET;
20081ad6265SDimitry Andric   }
20181ad6265SDimitry Andric }
20281ad6265SDimitry Andric 
20381ad6265SDimitry Andric 
20481ad6265SDimitry Andric // FIXME: All such magic numbers about the ABI should be in a
20581ad6265SDimitry Andric // central TD file.
getHostcallImplicitArgPosition(unsigned CodeObjectVersion)20606c3fb27SDimitry Andric unsigned getHostcallImplicitArgPosition(unsigned CodeObjectVersion) {
20706c3fb27SDimitry Andric   switch (CodeObjectVersion) {
20806c3fb27SDimitry Andric   case AMDHSA_COV4:
20981ad6265SDimitry Andric     return 24;
21006c3fb27SDimitry Andric   case AMDHSA_COV5:
21181ad6265SDimitry Andric   default:
21206c3fb27SDimitry Andric     return AMDGPU::ImplicitArg::HOSTCALL_PTR_OFFSET;
21381ad6265SDimitry Andric   }
21481ad6265SDimitry Andric }
21581ad6265SDimitry Andric 
getDefaultQueueImplicitArgPosition(unsigned CodeObjectVersion)21606c3fb27SDimitry Andric unsigned getDefaultQueueImplicitArgPosition(unsigned CodeObjectVersion) {
21706c3fb27SDimitry Andric   switch (CodeObjectVersion) {
21806c3fb27SDimitry Andric   case AMDHSA_COV4:
219bdd1243dSDimitry Andric     return 32;
22006c3fb27SDimitry Andric   case AMDHSA_COV5:
221bdd1243dSDimitry Andric   default:
222bdd1243dSDimitry Andric     return AMDGPU::ImplicitArg::DEFAULT_QUEUE_OFFSET;
223bdd1243dSDimitry Andric   }
224bdd1243dSDimitry Andric }
225bdd1243dSDimitry Andric 
getCompletionActionImplicitArgPosition(unsigned CodeObjectVersion)22606c3fb27SDimitry Andric unsigned getCompletionActionImplicitArgPosition(unsigned CodeObjectVersion) {
22706c3fb27SDimitry Andric   switch (CodeObjectVersion) {
22806c3fb27SDimitry Andric   case AMDHSA_COV4:
229bdd1243dSDimitry Andric     return 40;
23006c3fb27SDimitry Andric   case AMDHSA_COV5:
231bdd1243dSDimitry Andric   default:
232bdd1243dSDimitry Andric     return AMDGPU::ImplicitArg::COMPLETION_ACTION_OFFSET;
233bdd1243dSDimitry Andric   }
234bdd1243dSDimitry Andric }
235bdd1243dSDimitry Andric 
2360b57cec5SDimitry Andric #define GET_MIMGBaseOpcodesTable_IMPL
2370b57cec5SDimitry Andric #define GET_MIMGDimInfoTable_IMPL
2380b57cec5SDimitry Andric #define GET_MIMGInfoTable_IMPL
2390b57cec5SDimitry Andric #define GET_MIMGLZMappingTable_IMPL
2400b57cec5SDimitry Andric #define GET_MIMGMIPMappingTable_IMPL
24104eeddc0SDimitry Andric #define GET_MIMGBiasMappingTable_IMPL
24204eeddc0SDimitry Andric #define GET_MIMGOffsetMappingTable_IMPL
2435ffd83dbSDimitry Andric #define GET_MIMGG16MappingTable_IMPL
24481ad6265SDimitry Andric #define GET_MAIInstInfoTable_IMPL
2450b57cec5SDimitry Andric #include "AMDGPUGenSearchableTables.inc"
2460b57cec5SDimitry Andric 
getMIMGOpcode(unsigned BaseOpcode,unsigned MIMGEncoding,unsigned VDataDwords,unsigned VAddrDwords)2470b57cec5SDimitry Andric int getMIMGOpcode(unsigned BaseOpcode, unsigned MIMGEncoding,
2480b57cec5SDimitry Andric                   unsigned VDataDwords, unsigned VAddrDwords) {
2490b57cec5SDimitry Andric   const MIMGInfo *Info = getMIMGOpcodeHelper(BaseOpcode, MIMGEncoding,
2500b57cec5SDimitry Andric                                              VDataDwords, VAddrDwords);
2510b57cec5SDimitry Andric   return Info ? Info->Opcode : -1;
2520b57cec5SDimitry Andric }
2530b57cec5SDimitry Andric 
getMIMGBaseOpcode(unsigned Opc)2540b57cec5SDimitry Andric const MIMGBaseOpcodeInfo *getMIMGBaseOpcode(unsigned Opc) {
2550b57cec5SDimitry Andric   const MIMGInfo *Info = getMIMGInfo(Opc);
2560b57cec5SDimitry Andric   return Info ? getMIMGBaseOpcodeInfo(Info->BaseOpcode) : nullptr;
2570b57cec5SDimitry Andric }
2580b57cec5SDimitry Andric 
getMaskedMIMGOp(unsigned Opc,unsigned NewChannels)2590b57cec5SDimitry Andric int getMaskedMIMGOp(unsigned Opc, unsigned NewChannels) {
2600b57cec5SDimitry Andric   const MIMGInfo *OrigInfo = getMIMGInfo(Opc);
2610b57cec5SDimitry Andric   const MIMGInfo *NewInfo =
2620b57cec5SDimitry Andric       getMIMGOpcodeHelper(OrigInfo->BaseOpcode, OrigInfo->MIMGEncoding,
2630b57cec5SDimitry Andric                           NewChannels, OrigInfo->VAddrDwords);
2640b57cec5SDimitry Andric   return NewInfo ? NewInfo->Opcode : -1;
2650b57cec5SDimitry Andric }
2660b57cec5SDimitry Andric 
getAddrSizeMIMGOp(const MIMGBaseOpcodeInfo * BaseOpcode,const MIMGDimInfo * Dim,bool IsA16,bool IsG16Supported)267fe6060f1SDimitry Andric unsigned getAddrSizeMIMGOp(const MIMGBaseOpcodeInfo *BaseOpcode,
268fe6060f1SDimitry Andric                            const MIMGDimInfo *Dim, bool IsA16,
269fe6060f1SDimitry Andric                            bool IsG16Supported) {
270fe6060f1SDimitry Andric   unsigned AddrWords = BaseOpcode->NumExtraArgs;
271fe6060f1SDimitry Andric   unsigned AddrComponents = (BaseOpcode->Coordinates ? Dim->NumCoords : 0) +
272fe6060f1SDimitry Andric                             (BaseOpcode->LodOrClampOrMip ? 1 : 0);
273fe6060f1SDimitry Andric   if (IsA16)
274fe6060f1SDimitry Andric     AddrWords += divideCeil(AddrComponents, 2);
275fe6060f1SDimitry Andric   else
276fe6060f1SDimitry Andric     AddrWords += AddrComponents;
277fe6060f1SDimitry Andric 
278fe6060f1SDimitry Andric   // Note: For subtargets that support A16 but not G16, enabling A16 also
279fe6060f1SDimitry Andric   // enables 16 bit gradients.
280fe6060f1SDimitry Andric   // For subtargets that support A16 (operand) and G16 (done with a different
281fe6060f1SDimitry Andric   // instruction encoding), they are independent.
282fe6060f1SDimitry Andric 
283fe6060f1SDimitry Andric   if (BaseOpcode->Gradients) {
284fe6060f1SDimitry Andric     if ((IsA16 && !IsG16Supported) || BaseOpcode->G16)
285fe6060f1SDimitry Andric       // There are two gradients per coordinate, we pack them separately.
286fe6060f1SDimitry Andric       // For the 3d case,
287fe6060f1SDimitry Andric       // we get (dy/du, dx/du) (-, dz/du) (dy/dv, dx/dv) (-, dz/dv)
288fe6060f1SDimitry Andric       AddrWords += alignTo<2>(Dim->NumGradients / 2);
289fe6060f1SDimitry Andric     else
290fe6060f1SDimitry Andric       AddrWords += Dim->NumGradients;
291fe6060f1SDimitry Andric   }
292fe6060f1SDimitry Andric   return AddrWords;
293fe6060f1SDimitry Andric }
294fe6060f1SDimitry Andric 
2950b57cec5SDimitry Andric struct MUBUFInfo {
2960b57cec5SDimitry Andric   uint16_t Opcode;
2970b57cec5SDimitry Andric   uint16_t BaseOpcode;
2988bcb0991SDimitry Andric   uint8_t elements;
2990b57cec5SDimitry Andric   bool has_vaddr;
3000b57cec5SDimitry Andric   bool has_srsrc;
3010b57cec5SDimitry Andric   bool has_soffset;
302fe6060f1SDimitry Andric   bool IsBufferInv;
3030b57cec5SDimitry Andric };
3040b57cec5SDimitry Andric 
3058bcb0991SDimitry Andric struct MTBUFInfo {
3068bcb0991SDimitry Andric   uint16_t Opcode;
3078bcb0991SDimitry Andric   uint16_t BaseOpcode;
3088bcb0991SDimitry Andric   uint8_t elements;
3098bcb0991SDimitry Andric   bool has_vaddr;
3108bcb0991SDimitry Andric   bool has_srsrc;
3118bcb0991SDimitry Andric   bool has_soffset;
3128bcb0991SDimitry Andric };
3138bcb0991SDimitry Andric 
3145ffd83dbSDimitry Andric struct SMInfo {
3155ffd83dbSDimitry Andric   uint16_t Opcode;
3165ffd83dbSDimitry Andric   bool IsBuffer;
3175ffd83dbSDimitry Andric };
3185ffd83dbSDimitry Andric 
319fe6060f1SDimitry Andric struct VOPInfo {
320fe6060f1SDimitry Andric   uint16_t Opcode;
321fe6060f1SDimitry Andric   bool IsSingle;
322fe6060f1SDimitry Andric };
323fe6060f1SDimitry Andric 
32481ad6265SDimitry Andric struct VOPC64DPPInfo {
32581ad6265SDimitry Andric   uint16_t Opcode;
32681ad6265SDimitry Andric };
32781ad6265SDimitry Andric 
328753f127fSDimitry Andric struct VOPDComponentInfo {
329753f127fSDimitry Andric   uint16_t BaseVOP;
330753f127fSDimitry Andric   uint16_t VOPDOp;
331753f127fSDimitry Andric   bool CanBeVOPDX;
332753f127fSDimitry Andric };
333753f127fSDimitry Andric 
334753f127fSDimitry Andric struct VOPDInfo {
335753f127fSDimitry Andric   uint16_t Opcode;
336753f127fSDimitry Andric   uint16_t OpX;
337753f127fSDimitry Andric   uint16_t OpY;
3385f757f3fSDimitry Andric   uint16_t Subtarget;
339753f127fSDimitry Andric };
340753f127fSDimitry Andric 
341bdd1243dSDimitry Andric struct VOPTrue16Info {
342bdd1243dSDimitry Andric   uint16_t Opcode;
343bdd1243dSDimitry Andric   bool IsTrue16;
344bdd1243dSDimitry Andric };
345bdd1243dSDimitry Andric 
3468bcb0991SDimitry Andric #define GET_MTBUFInfoTable_DECL
3478bcb0991SDimitry Andric #define GET_MTBUFInfoTable_IMPL
3480b57cec5SDimitry Andric #define GET_MUBUFInfoTable_DECL
3490b57cec5SDimitry Andric #define GET_MUBUFInfoTable_IMPL
3505ffd83dbSDimitry Andric #define GET_SMInfoTable_DECL
3515ffd83dbSDimitry Andric #define GET_SMInfoTable_IMPL
352fe6060f1SDimitry Andric #define GET_VOP1InfoTable_DECL
353fe6060f1SDimitry Andric #define GET_VOP1InfoTable_IMPL
354fe6060f1SDimitry Andric #define GET_VOP2InfoTable_DECL
355fe6060f1SDimitry Andric #define GET_VOP2InfoTable_IMPL
356fe6060f1SDimitry Andric #define GET_VOP3InfoTable_DECL
357fe6060f1SDimitry Andric #define GET_VOP3InfoTable_IMPL
35881ad6265SDimitry Andric #define GET_VOPC64DPPTable_DECL
35981ad6265SDimitry Andric #define GET_VOPC64DPPTable_IMPL
36081ad6265SDimitry Andric #define GET_VOPC64DPP8Table_DECL
36181ad6265SDimitry Andric #define GET_VOPC64DPP8Table_IMPL
362753f127fSDimitry Andric #define GET_VOPDComponentTable_DECL
363753f127fSDimitry Andric #define GET_VOPDComponentTable_IMPL
364753f127fSDimitry Andric #define GET_VOPDPairs_DECL
365753f127fSDimitry Andric #define GET_VOPDPairs_IMPL
366bdd1243dSDimitry Andric #define GET_VOPTrue16Table_DECL
367bdd1243dSDimitry Andric #define GET_VOPTrue16Table_IMPL
36881ad6265SDimitry Andric #define GET_WMMAOpcode2AddrMappingTable_DECL
36981ad6265SDimitry Andric #define GET_WMMAOpcode2AddrMappingTable_IMPL
37081ad6265SDimitry Andric #define GET_WMMAOpcode3AddrMappingTable_DECL
37181ad6265SDimitry Andric #define GET_WMMAOpcode3AddrMappingTable_IMPL
3720b57cec5SDimitry Andric #include "AMDGPUGenSearchableTables.inc"
3730b57cec5SDimitry Andric 
getMTBUFBaseOpcode(unsigned Opc)3748bcb0991SDimitry Andric int getMTBUFBaseOpcode(unsigned Opc) {
3758bcb0991SDimitry Andric   const MTBUFInfo *Info = getMTBUFInfoFromOpcode(Opc);
3768bcb0991SDimitry Andric   return Info ? Info->BaseOpcode : -1;
3778bcb0991SDimitry Andric }
3788bcb0991SDimitry Andric 
getMTBUFOpcode(unsigned BaseOpc,unsigned Elements)3798bcb0991SDimitry Andric int getMTBUFOpcode(unsigned BaseOpc, unsigned Elements) {
3808bcb0991SDimitry Andric   const MTBUFInfo *Info = getMTBUFInfoFromBaseOpcodeAndElements(BaseOpc, Elements);
3818bcb0991SDimitry Andric   return Info ? Info->Opcode : -1;
3828bcb0991SDimitry Andric }
3838bcb0991SDimitry Andric 
getMTBUFElements(unsigned Opc)3848bcb0991SDimitry Andric int getMTBUFElements(unsigned Opc) {
3858bcb0991SDimitry Andric   const MTBUFInfo *Info = getMTBUFOpcodeHelper(Opc);
3868bcb0991SDimitry Andric   return Info ? Info->elements : 0;
3878bcb0991SDimitry Andric }
3888bcb0991SDimitry Andric 
getMTBUFHasVAddr(unsigned Opc)3898bcb0991SDimitry Andric bool getMTBUFHasVAddr(unsigned Opc) {
3908bcb0991SDimitry Andric   const MTBUFInfo *Info = getMTBUFOpcodeHelper(Opc);
3918bcb0991SDimitry Andric   return Info ? Info->has_vaddr : false;
3928bcb0991SDimitry Andric }
3938bcb0991SDimitry Andric 
getMTBUFHasSrsrc(unsigned Opc)3948bcb0991SDimitry Andric bool getMTBUFHasSrsrc(unsigned Opc) {
3958bcb0991SDimitry Andric   const MTBUFInfo *Info = getMTBUFOpcodeHelper(Opc);
3968bcb0991SDimitry Andric   return Info ? Info->has_srsrc : false;
3978bcb0991SDimitry Andric }
3988bcb0991SDimitry Andric 
getMTBUFHasSoffset(unsigned Opc)3998bcb0991SDimitry Andric bool getMTBUFHasSoffset(unsigned Opc) {
4008bcb0991SDimitry Andric   const MTBUFInfo *Info = getMTBUFOpcodeHelper(Opc);
4018bcb0991SDimitry Andric   return Info ? Info->has_soffset : false;
4028bcb0991SDimitry Andric }
4038bcb0991SDimitry Andric 
getMUBUFBaseOpcode(unsigned Opc)4040b57cec5SDimitry Andric int getMUBUFBaseOpcode(unsigned Opc) {
4050b57cec5SDimitry Andric   const MUBUFInfo *Info = getMUBUFInfoFromOpcode(Opc);
4060b57cec5SDimitry Andric   return Info ? Info->BaseOpcode : -1;
4070b57cec5SDimitry Andric }
4080b57cec5SDimitry Andric 
getMUBUFOpcode(unsigned BaseOpc,unsigned Elements)4098bcb0991SDimitry Andric int getMUBUFOpcode(unsigned BaseOpc, unsigned Elements) {
4108bcb0991SDimitry Andric   const MUBUFInfo *Info = getMUBUFInfoFromBaseOpcodeAndElements(BaseOpc, Elements);
4110b57cec5SDimitry Andric   return Info ? Info->Opcode : -1;
4120b57cec5SDimitry Andric }
4130b57cec5SDimitry Andric 
getMUBUFElements(unsigned Opc)4148bcb0991SDimitry Andric int getMUBUFElements(unsigned Opc) {
4150b57cec5SDimitry Andric   const MUBUFInfo *Info = getMUBUFOpcodeHelper(Opc);
4168bcb0991SDimitry Andric   return Info ? Info->elements : 0;
4170b57cec5SDimitry Andric }
4180b57cec5SDimitry Andric 
getMUBUFHasVAddr(unsigned Opc)4190b57cec5SDimitry Andric bool getMUBUFHasVAddr(unsigned Opc) {
4200b57cec5SDimitry Andric   const MUBUFInfo *Info = getMUBUFOpcodeHelper(Opc);
4210b57cec5SDimitry Andric   return Info ? Info->has_vaddr : false;
4220b57cec5SDimitry Andric }
4230b57cec5SDimitry Andric 
getMUBUFHasSrsrc(unsigned Opc)4240b57cec5SDimitry Andric bool getMUBUFHasSrsrc(unsigned Opc) {
4250b57cec5SDimitry Andric   const MUBUFInfo *Info = getMUBUFOpcodeHelper(Opc);
4260b57cec5SDimitry Andric   return Info ? Info->has_srsrc : false;
4270b57cec5SDimitry Andric }
4280b57cec5SDimitry Andric 
getMUBUFHasSoffset(unsigned Opc)4290b57cec5SDimitry Andric bool getMUBUFHasSoffset(unsigned Opc) {
4300b57cec5SDimitry Andric   const MUBUFInfo *Info = getMUBUFOpcodeHelper(Opc);
4310b57cec5SDimitry Andric   return Info ? Info->has_soffset : false;
4320b57cec5SDimitry Andric }
4330b57cec5SDimitry Andric 
getMUBUFIsBufferInv(unsigned Opc)434fe6060f1SDimitry Andric bool getMUBUFIsBufferInv(unsigned Opc) {
435fe6060f1SDimitry Andric   const MUBUFInfo *Info = getMUBUFOpcodeHelper(Opc);
436fe6060f1SDimitry Andric   return Info ? Info->IsBufferInv : false;
437fe6060f1SDimitry Andric }
438fe6060f1SDimitry Andric 
getSMEMIsBuffer(unsigned Opc)4395ffd83dbSDimitry Andric bool getSMEMIsBuffer(unsigned Opc) {
4405ffd83dbSDimitry Andric   const SMInfo *Info = getSMEMOpcodeHelper(Opc);
4415ffd83dbSDimitry Andric   return Info ? Info->IsBuffer : false;
4425ffd83dbSDimitry Andric }
4435ffd83dbSDimitry Andric 
getVOP1IsSingle(unsigned Opc)444fe6060f1SDimitry Andric bool getVOP1IsSingle(unsigned Opc) {
445fe6060f1SDimitry Andric   const VOPInfo *Info = getVOP1OpcodeHelper(Opc);
446fe6060f1SDimitry Andric   return Info ? Info->IsSingle : false;
447fe6060f1SDimitry Andric }
448fe6060f1SDimitry Andric 
getVOP2IsSingle(unsigned Opc)449fe6060f1SDimitry Andric bool getVOP2IsSingle(unsigned Opc) {
450fe6060f1SDimitry Andric   const VOPInfo *Info = getVOP2OpcodeHelper(Opc);
451fe6060f1SDimitry Andric   return Info ? Info->IsSingle : false;
452fe6060f1SDimitry Andric }
453fe6060f1SDimitry Andric 
getVOP3IsSingle(unsigned Opc)454fe6060f1SDimitry Andric bool getVOP3IsSingle(unsigned Opc) {
455fe6060f1SDimitry Andric   const VOPInfo *Info = getVOP3OpcodeHelper(Opc);
456fe6060f1SDimitry Andric   return Info ? Info->IsSingle : false;
457fe6060f1SDimitry Andric }
458fe6060f1SDimitry Andric 
isVOPC64DPP(unsigned Opc)45981ad6265SDimitry Andric bool isVOPC64DPP(unsigned Opc) {
46081ad6265SDimitry Andric   return isVOPC64DPPOpcodeHelper(Opc) || isVOPC64DPP8OpcodeHelper(Opc);
46181ad6265SDimitry Andric }
46281ad6265SDimitry Andric 
getMAIIsDGEMM(unsigned Opc)46381ad6265SDimitry Andric bool getMAIIsDGEMM(unsigned Opc) {
46481ad6265SDimitry Andric   const MAIInstInfo *Info = getMAIInstInfoHelper(Opc);
46581ad6265SDimitry Andric   return Info ? Info->is_dgemm : false;
46681ad6265SDimitry Andric }
46781ad6265SDimitry Andric 
getMAIIsGFX940XDL(unsigned Opc)46881ad6265SDimitry Andric bool getMAIIsGFX940XDL(unsigned Opc) {
46981ad6265SDimitry Andric   const MAIInstInfo *Info = getMAIInstInfoHelper(Opc);
47081ad6265SDimitry Andric   return Info ? Info->is_gfx940_xdl : false;
47181ad6265SDimitry Andric }
47281ad6265SDimitry Andric 
getVOPDEncodingFamily(const MCSubtargetInfo & ST)4735f757f3fSDimitry Andric unsigned getVOPDEncodingFamily(const MCSubtargetInfo &ST) {
4745f757f3fSDimitry Andric   if (ST.hasFeature(AMDGPU::FeatureGFX12Insts))
4755f757f3fSDimitry Andric     return SIEncodingFamily::GFX12;
4765f757f3fSDimitry Andric   if (ST.hasFeature(AMDGPU::FeatureGFX11Insts))
4775f757f3fSDimitry Andric     return SIEncodingFamily::GFX11;
4785f757f3fSDimitry Andric   llvm_unreachable("Subtarget generation does not support VOPD!");
4795f757f3fSDimitry Andric }
4805f757f3fSDimitry Andric 
getCanBeVOPD(unsigned Opc)481753f127fSDimitry Andric CanBeVOPD getCanBeVOPD(unsigned Opc) {
482753f127fSDimitry Andric   const VOPDComponentInfo *Info = getVOPDComponentHelper(Opc);
483753f127fSDimitry Andric   if (Info)
484fcaf7f86SDimitry Andric     return {Info->CanBeVOPDX, true};
485753f127fSDimitry Andric   else
486fcaf7f86SDimitry Andric     return {false, false};
487753f127fSDimitry Andric }
488753f127fSDimitry Andric 
getVOPDOpcode(unsigned Opc)489753f127fSDimitry Andric unsigned getVOPDOpcode(unsigned Opc) {
490753f127fSDimitry Andric   const VOPDComponentInfo *Info = getVOPDComponentHelper(Opc);
491753f127fSDimitry Andric   return Info ? Info->VOPDOp : ~0u;
492753f127fSDimitry Andric }
493753f127fSDimitry Andric 
isVOPD(unsigned Opc)494bdd1243dSDimitry Andric bool isVOPD(unsigned Opc) {
495bdd1243dSDimitry Andric   return AMDGPU::hasNamedOperand(Opc, AMDGPU::OpName::src0X);
496bdd1243dSDimitry Andric }
497bdd1243dSDimitry Andric 
isMAC(unsigned Opc)498bdd1243dSDimitry Andric bool isMAC(unsigned Opc) {
499bdd1243dSDimitry Andric   return Opc == AMDGPU::V_MAC_F32_e64_gfx6_gfx7 ||
500bdd1243dSDimitry Andric          Opc == AMDGPU::V_MAC_F32_e64_gfx10 ||
501bdd1243dSDimitry Andric          Opc == AMDGPU::V_MAC_F32_e64_vi ||
502bdd1243dSDimitry Andric          Opc == AMDGPU::V_MAC_LEGACY_F32_e64_gfx6_gfx7 ||
503bdd1243dSDimitry Andric          Opc == AMDGPU::V_MAC_LEGACY_F32_e64_gfx10 ||
504bdd1243dSDimitry Andric          Opc == AMDGPU::V_MAC_F16_e64_vi ||
505bdd1243dSDimitry Andric          Opc == AMDGPU::V_FMAC_F64_e64_gfx90a ||
506bdd1243dSDimitry Andric          Opc == AMDGPU::V_FMAC_F32_e64_gfx10 ||
507bdd1243dSDimitry Andric          Opc == AMDGPU::V_FMAC_F32_e64_gfx11 ||
5085f757f3fSDimitry Andric          Opc == AMDGPU::V_FMAC_F32_e64_gfx12 ||
509bdd1243dSDimitry Andric          Opc == AMDGPU::V_FMAC_F32_e64_vi ||
510bdd1243dSDimitry Andric          Opc == AMDGPU::V_FMAC_LEGACY_F32_e64_gfx10 ||
511bdd1243dSDimitry Andric          Opc == AMDGPU::V_FMAC_DX9_ZERO_F32_e64_gfx11 ||
512bdd1243dSDimitry Andric          Opc == AMDGPU::V_FMAC_F16_e64_gfx10 ||
513bdd1243dSDimitry Andric          Opc == AMDGPU::V_FMAC_F16_t16_e64_gfx11 ||
5145f757f3fSDimitry Andric          Opc == AMDGPU::V_FMAC_F16_t16_e64_gfx12 ||
515bdd1243dSDimitry Andric          Opc == AMDGPU::V_DOT2C_F32_F16_e64_vi ||
516bdd1243dSDimitry Andric          Opc == AMDGPU::V_DOT2C_I32_I16_e64_vi ||
517bdd1243dSDimitry Andric          Opc == AMDGPU::V_DOT4C_I32_I8_e64_vi ||
518bdd1243dSDimitry Andric          Opc == AMDGPU::V_DOT8C_I32_I4_e64_vi;
519bdd1243dSDimitry Andric }
520bdd1243dSDimitry Andric 
isPermlane16(unsigned Opc)521bdd1243dSDimitry Andric bool isPermlane16(unsigned Opc) {
522bdd1243dSDimitry Andric   return Opc == AMDGPU::V_PERMLANE16_B32_gfx10 ||
523bdd1243dSDimitry Andric          Opc == AMDGPU::V_PERMLANEX16_B32_gfx10 ||
524bdd1243dSDimitry Andric          Opc == AMDGPU::V_PERMLANE16_B32_e64_gfx11 ||
5255f757f3fSDimitry Andric          Opc == AMDGPU::V_PERMLANEX16_B32_e64_gfx11 ||
5265f757f3fSDimitry Andric          Opc == AMDGPU::V_PERMLANE16_B32_e64_gfx12 ||
5275f757f3fSDimitry Andric          Opc == AMDGPU::V_PERMLANEX16_B32_e64_gfx12 ||
5285f757f3fSDimitry Andric          Opc == AMDGPU::V_PERMLANE16_VAR_B32_e64_gfx12 ||
5295f757f3fSDimitry Andric          Opc == AMDGPU::V_PERMLANEX16_VAR_B32_e64_gfx12;
5305f757f3fSDimitry Andric }
5315f757f3fSDimitry Andric 
isCvt_F32_Fp8_Bf8_e64(unsigned Opc)532b3edf446SDimitry Andric bool isCvt_F32_Fp8_Bf8_e64(unsigned Opc) {
533b3edf446SDimitry Andric   return Opc == AMDGPU::V_CVT_F32_BF8_e64_gfx12 ||
534b3edf446SDimitry Andric          Opc == AMDGPU::V_CVT_F32_FP8_e64_gfx12 ||
535b3edf446SDimitry Andric          Opc == AMDGPU::V_CVT_F32_BF8_e64_dpp_gfx12 ||
536b3edf446SDimitry Andric          Opc == AMDGPU::V_CVT_F32_FP8_e64_dpp_gfx12 ||
537b3edf446SDimitry Andric          Opc == AMDGPU::V_CVT_F32_BF8_e64_dpp8_gfx12 ||
538b3edf446SDimitry Andric          Opc == AMDGPU::V_CVT_F32_FP8_e64_dpp8_gfx12 ||
539b3edf446SDimitry Andric          Opc == AMDGPU::V_CVT_PK_F32_BF8_e64_gfx12 ||
540b3edf446SDimitry Andric          Opc == AMDGPU::V_CVT_PK_F32_FP8_e64_gfx12;
541b3edf446SDimitry Andric }
542b3edf446SDimitry Andric 
isGenericAtomic(unsigned Opc)5435f757f3fSDimitry Andric bool isGenericAtomic(unsigned Opc) {
5445f757f3fSDimitry Andric   return Opc == AMDGPU::G_AMDGPU_ATOMIC_FMIN ||
5455f757f3fSDimitry Andric          Opc == AMDGPU::G_AMDGPU_ATOMIC_FMAX ||
5465f757f3fSDimitry Andric          Opc == AMDGPU::G_AMDGPU_BUFFER_ATOMIC_SWAP ||
5475f757f3fSDimitry Andric          Opc == AMDGPU::G_AMDGPU_BUFFER_ATOMIC_ADD ||
5485f757f3fSDimitry Andric          Opc == AMDGPU::G_AMDGPU_BUFFER_ATOMIC_SUB ||
5495f757f3fSDimitry Andric          Opc == AMDGPU::G_AMDGPU_BUFFER_ATOMIC_SMIN ||
5505f757f3fSDimitry Andric          Opc == AMDGPU::G_AMDGPU_BUFFER_ATOMIC_UMIN ||
5515f757f3fSDimitry Andric          Opc == AMDGPU::G_AMDGPU_BUFFER_ATOMIC_SMAX ||
5525f757f3fSDimitry Andric          Opc == AMDGPU::G_AMDGPU_BUFFER_ATOMIC_UMAX ||
5535f757f3fSDimitry Andric          Opc == AMDGPU::G_AMDGPU_BUFFER_ATOMIC_AND ||
5545f757f3fSDimitry Andric          Opc == AMDGPU::G_AMDGPU_BUFFER_ATOMIC_OR ||
5555f757f3fSDimitry Andric          Opc == AMDGPU::G_AMDGPU_BUFFER_ATOMIC_XOR ||
5565f757f3fSDimitry Andric          Opc == AMDGPU::G_AMDGPU_BUFFER_ATOMIC_INC ||
5575f757f3fSDimitry Andric          Opc == AMDGPU::G_AMDGPU_BUFFER_ATOMIC_DEC ||
5585f757f3fSDimitry Andric          Opc == AMDGPU::G_AMDGPU_BUFFER_ATOMIC_FADD ||
5595f757f3fSDimitry Andric          Opc == AMDGPU::G_AMDGPU_BUFFER_ATOMIC_FMIN ||
5605f757f3fSDimitry Andric          Opc == AMDGPU::G_AMDGPU_BUFFER_ATOMIC_FMAX ||
5615f757f3fSDimitry Andric          Opc == AMDGPU::G_AMDGPU_BUFFER_ATOMIC_CMPSWAP ||
5625f757f3fSDimitry Andric          Opc == AMDGPU::G_AMDGPU_ATOMIC_CMPXCHG;
563bdd1243dSDimitry Andric }
564bdd1243dSDimitry Andric 
isTrue16Inst(unsigned Opc)565bdd1243dSDimitry Andric bool isTrue16Inst(unsigned Opc) {
566bdd1243dSDimitry Andric   const VOPTrue16Info *Info = getTrue16OpcodeHelper(Opc);
567bdd1243dSDimitry Andric   return Info ? Info->IsTrue16 : false;
568bdd1243dSDimitry Andric }
569bdd1243dSDimitry Andric 
mapWMMA2AddrTo3AddrOpcode(unsigned Opc)57081ad6265SDimitry Andric unsigned mapWMMA2AddrTo3AddrOpcode(unsigned Opc) {
57181ad6265SDimitry Andric   const WMMAOpcodeMappingInfo *Info = getWMMAMappingInfoFrom2AddrOpcode(Opc);
57281ad6265SDimitry Andric   return Info ? Info->Opcode3Addr : ~0u;
57381ad6265SDimitry Andric }
57481ad6265SDimitry Andric 
mapWMMA3AddrTo2AddrOpcode(unsigned Opc)57581ad6265SDimitry Andric unsigned mapWMMA3AddrTo2AddrOpcode(unsigned Opc) {
57681ad6265SDimitry Andric   const WMMAOpcodeMappingInfo *Info = getWMMAMappingInfoFrom3AddrOpcode(Opc);
57781ad6265SDimitry Andric   return Info ? Info->Opcode2Addr : ~0u;
57881ad6265SDimitry Andric }
57981ad6265SDimitry Andric 
5800b57cec5SDimitry Andric // Wrapper for Tablegen'd function.  enum Subtarget is not defined in any
5810b57cec5SDimitry Andric // header files, so we need to wrap it in a function that takes unsigned
5820b57cec5SDimitry Andric // instead.
getMCOpcode(uint16_t Opcode,unsigned Gen)5830b57cec5SDimitry Andric int getMCOpcode(uint16_t Opcode, unsigned Gen) {
5840b57cec5SDimitry Andric   return getMCOpcodeGen(Opcode, static_cast<Subtarget>(Gen));
5850b57cec5SDimitry Andric }
5860b57cec5SDimitry Andric 
getVOPDFull(unsigned OpX,unsigned OpY,unsigned EncodingFamily)5875f757f3fSDimitry Andric int getVOPDFull(unsigned OpX, unsigned OpY, unsigned EncodingFamily) {
5885f757f3fSDimitry Andric   const VOPDInfo *Info =
5895f757f3fSDimitry Andric       getVOPDInfoFromComponentOpcodes(OpX, OpY, EncodingFamily);
590753f127fSDimitry Andric   return Info ? Info->Opcode : -1;
591753f127fSDimitry Andric }
592753f127fSDimitry Andric 
getVOPDComponents(unsigned VOPDOpcode)593bdd1243dSDimitry Andric std::pair<unsigned, unsigned> getVOPDComponents(unsigned VOPDOpcode) {
594bdd1243dSDimitry Andric   const VOPDInfo *Info = getVOPDOpcodeHelper(VOPDOpcode);
595bdd1243dSDimitry Andric   assert(Info);
596bdd1243dSDimitry Andric   auto OpX = getVOPDBaseFromComponent(Info->OpX);
597bdd1243dSDimitry Andric   auto OpY = getVOPDBaseFromComponent(Info->OpY);
598bdd1243dSDimitry Andric   assert(OpX && OpY);
599bdd1243dSDimitry Andric   return {OpX->BaseVOP, OpY->BaseVOP};
600bdd1243dSDimitry Andric }
601bdd1243dSDimitry Andric 
602bdd1243dSDimitry Andric namespace VOPD {
603bdd1243dSDimitry Andric 
ComponentProps(const MCInstrDesc & OpDesc)604bdd1243dSDimitry Andric ComponentProps::ComponentProps(const MCInstrDesc &OpDesc) {
605bdd1243dSDimitry Andric   assert(OpDesc.getNumDefs() == Component::DST_NUM);
606bdd1243dSDimitry Andric 
607bdd1243dSDimitry Andric   assert(OpDesc.getOperandConstraint(Component::SRC0, MCOI::TIED_TO) == -1);
608bdd1243dSDimitry Andric   assert(OpDesc.getOperandConstraint(Component::SRC1, MCOI::TIED_TO) == -1);
609bdd1243dSDimitry Andric   auto TiedIdx = OpDesc.getOperandConstraint(Component::SRC2, MCOI::TIED_TO);
610bdd1243dSDimitry Andric   assert(TiedIdx == -1 || TiedIdx == Component::DST);
611bdd1243dSDimitry Andric   HasSrc2Acc = TiedIdx != -1;
612bdd1243dSDimitry Andric 
613bdd1243dSDimitry Andric   SrcOperandsNum = OpDesc.getNumOperands() - OpDesc.getNumDefs();
614bdd1243dSDimitry Andric   assert(SrcOperandsNum <= Component::MAX_SRC_NUM);
615bdd1243dSDimitry Andric 
616bdd1243dSDimitry Andric   auto OperandsNum = OpDesc.getNumOperands();
617bdd1243dSDimitry Andric   unsigned CompOprIdx;
618bdd1243dSDimitry Andric   for (CompOprIdx = Component::SRC1; CompOprIdx < OperandsNum; ++CompOprIdx) {
619bdd1243dSDimitry Andric     if (OpDesc.operands()[CompOprIdx].OperandType == AMDGPU::OPERAND_KIMM32) {
620bdd1243dSDimitry Andric       MandatoryLiteralIdx = CompOprIdx;
621bdd1243dSDimitry Andric       break;
622bdd1243dSDimitry Andric     }
623bdd1243dSDimitry Andric   }
624bdd1243dSDimitry Andric }
625bdd1243dSDimitry Andric 
getIndexInParsedOperands(unsigned CompOprIdx) const626bdd1243dSDimitry Andric unsigned ComponentInfo::getIndexInParsedOperands(unsigned CompOprIdx) const {
627bdd1243dSDimitry Andric   assert(CompOprIdx < Component::MAX_OPR_NUM);
628bdd1243dSDimitry Andric 
629bdd1243dSDimitry Andric   if (CompOprIdx == Component::DST)
630bdd1243dSDimitry Andric     return getIndexOfDstInParsedOperands();
631bdd1243dSDimitry Andric 
632bdd1243dSDimitry Andric   auto CompSrcIdx = CompOprIdx - Component::DST_NUM;
633bdd1243dSDimitry Andric   if (CompSrcIdx < getCompParsedSrcOperandsNum())
634bdd1243dSDimitry Andric     return getIndexOfSrcInParsedOperands(CompSrcIdx);
635bdd1243dSDimitry Andric 
636bdd1243dSDimitry Andric   // The specified operand does not exist.
637bdd1243dSDimitry Andric   return 0;
638bdd1243dSDimitry Andric }
639bdd1243dSDimitry Andric 
getInvalidCompOperandIndex(std::function<unsigned (unsigned,unsigned)> GetRegIdx,bool SkipSrc) const640bdd1243dSDimitry Andric std::optional<unsigned> InstInfo::getInvalidCompOperandIndex(
6415f757f3fSDimitry Andric     std::function<unsigned(unsigned, unsigned)> GetRegIdx, bool SkipSrc) const {
642bdd1243dSDimitry Andric 
643bdd1243dSDimitry Andric   auto OpXRegs = getRegIndices(ComponentIndex::X, GetRegIdx);
644bdd1243dSDimitry Andric   auto OpYRegs = getRegIndices(ComponentIndex::Y, GetRegIdx);
645bdd1243dSDimitry Andric 
6465f757f3fSDimitry Andric   const unsigned CompOprNum =
6475f757f3fSDimitry Andric       SkipSrc ? Component::DST_NUM : Component::MAX_OPR_NUM;
648bdd1243dSDimitry Andric   unsigned CompOprIdx;
6495f757f3fSDimitry Andric   for (CompOprIdx = 0; CompOprIdx < CompOprNum; ++CompOprIdx) {
65006c3fb27SDimitry Andric     unsigned BanksMasks = VOPD_VGPR_BANK_MASKS[CompOprIdx];
651bdd1243dSDimitry Andric     if (OpXRegs[CompOprIdx] && OpYRegs[CompOprIdx] &&
65206c3fb27SDimitry Andric         ((OpXRegs[CompOprIdx] & BanksMasks) ==
65306c3fb27SDimitry Andric          (OpYRegs[CompOprIdx] & BanksMasks)))
654bdd1243dSDimitry Andric       return CompOprIdx;
655bdd1243dSDimitry Andric   }
656bdd1243dSDimitry Andric 
657bdd1243dSDimitry Andric   return {};
658bdd1243dSDimitry Andric }
659bdd1243dSDimitry Andric 
660bdd1243dSDimitry Andric // Return an array of VGPR registers [DST,SRC0,SRC1,SRC2] used
661bdd1243dSDimitry Andric // by the specified component. If an operand is unused
662bdd1243dSDimitry Andric // or is not a VGPR, the corresponding value is 0.
663bdd1243dSDimitry Andric //
664bdd1243dSDimitry Andric // GetRegIdx(Component, MCOperandIdx) must return a VGPR register index
665bdd1243dSDimitry Andric // for the specified component and MC operand. The callback must return 0
666bdd1243dSDimitry Andric // if the operand is not a register or not a VGPR.
getRegIndices(unsigned CompIdx,std::function<unsigned (unsigned,unsigned)> GetRegIdx) const667bdd1243dSDimitry Andric InstInfo::RegIndices InstInfo::getRegIndices(
668bdd1243dSDimitry Andric     unsigned CompIdx,
669bdd1243dSDimitry Andric     std::function<unsigned(unsigned, unsigned)> GetRegIdx) const {
670bdd1243dSDimitry Andric   assert(CompIdx < COMPONENTS_NUM);
671bdd1243dSDimitry Andric 
672bdd1243dSDimitry Andric   const auto &Comp = CompInfo[CompIdx];
673bdd1243dSDimitry Andric   InstInfo::RegIndices RegIndices;
674bdd1243dSDimitry Andric 
675bdd1243dSDimitry Andric   RegIndices[DST] = GetRegIdx(CompIdx, Comp.getIndexOfDstInMCOperands());
676bdd1243dSDimitry Andric 
677bdd1243dSDimitry Andric   for (unsigned CompOprIdx : {SRC0, SRC1, SRC2}) {
678bdd1243dSDimitry Andric     unsigned CompSrcIdx = CompOprIdx - DST_NUM;
679bdd1243dSDimitry Andric     RegIndices[CompOprIdx] =
680bdd1243dSDimitry Andric         Comp.hasRegSrcOperand(CompSrcIdx)
681bdd1243dSDimitry Andric             ? GetRegIdx(CompIdx, Comp.getIndexOfSrcInMCOperands(CompSrcIdx))
682bdd1243dSDimitry Andric             : 0;
683bdd1243dSDimitry Andric   }
684bdd1243dSDimitry Andric   return RegIndices;
685bdd1243dSDimitry Andric }
686bdd1243dSDimitry Andric 
687bdd1243dSDimitry Andric } // namespace VOPD
688bdd1243dSDimitry Andric 
getVOPDInstInfo(const MCInstrDesc & OpX,const MCInstrDesc & OpY)689bdd1243dSDimitry Andric VOPD::InstInfo getVOPDInstInfo(const MCInstrDesc &OpX, const MCInstrDesc &OpY) {
690bdd1243dSDimitry Andric   return VOPD::InstInfo(OpX, OpY);
691bdd1243dSDimitry Andric }
692bdd1243dSDimitry Andric 
getVOPDInstInfo(unsigned VOPDOpcode,const MCInstrInfo * InstrInfo)693bdd1243dSDimitry Andric VOPD::InstInfo getVOPDInstInfo(unsigned VOPDOpcode,
694bdd1243dSDimitry Andric                                const MCInstrInfo *InstrInfo) {
695bdd1243dSDimitry Andric   auto [OpX, OpY] = getVOPDComponents(VOPDOpcode);
696bdd1243dSDimitry Andric   const auto &OpXDesc = InstrInfo->get(OpX);
697bdd1243dSDimitry Andric   const auto &OpYDesc = InstrInfo->get(OpY);
698bdd1243dSDimitry Andric   VOPD::ComponentInfo OpXInfo(OpXDesc, VOPD::ComponentKind::COMPONENT_X);
699bdd1243dSDimitry Andric   VOPD::ComponentInfo OpYInfo(OpYDesc, OpXInfo);
700bdd1243dSDimitry Andric   return VOPD::InstInfo(OpXInfo, OpYInfo);
701bdd1243dSDimitry Andric }
702bdd1243dSDimitry Andric 
7030b57cec5SDimitry Andric namespace IsaInfo {
7040b57cec5SDimitry Andric 
AMDGPUTargetID(const MCSubtargetInfo & STI)705e8d8bef9SDimitry Andric AMDGPUTargetID::AMDGPUTargetID(const MCSubtargetInfo &STI)
706fe6060f1SDimitry Andric     : STI(STI), XnackSetting(TargetIDSetting::Any),
7077a6dacacSDimitry Andric       SramEccSetting(TargetIDSetting::Any) {
708e8d8bef9SDimitry Andric   if (!STI.getFeatureBits().test(FeatureSupportsXNACK))
709e8d8bef9SDimitry Andric     XnackSetting = TargetIDSetting::Unsupported;
710e8d8bef9SDimitry Andric   if (!STI.getFeatureBits().test(FeatureSupportsSRAMECC))
711e8d8bef9SDimitry Andric     SramEccSetting = TargetIDSetting::Unsupported;
712e8d8bef9SDimitry Andric }
713e8d8bef9SDimitry Andric 
setTargetIDFromFeaturesString(StringRef FS)714e8d8bef9SDimitry Andric void AMDGPUTargetID::setTargetIDFromFeaturesString(StringRef FS) {
715e8d8bef9SDimitry Andric   // Check if xnack or sramecc is explicitly enabled or disabled.  In the
716e8d8bef9SDimitry Andric   // absence of the target features we assume we must generate code that can run
717e8d8bef9SDimitry Andric   // in any environment.
718e8d8bef9SDimitry Andric   SubtargetFeatures Features(FS);
719bdd1243dSDimitry Andric   std::optional<bool> XnackRequested;
720bdd1243dSDimitry Andric   std::optional<bool> SramEccRequested;
721e8d8bef9SDimitry Andric 
722e8d8bef9SDimitry Andric   for (const std::string &Feature : Features.getFeatures()) {
723e8d8bef9SDimitry Andric     if (Feature == "+xnack")
724e8d8bef9SDimitry Andric       XnackRequested = true;
725e8d8bef9SDimitry Andric     else if (Feature == "-xnack")
726e8d8bef9SDimitry Andric       XnackRequested = false;
727e8d8bef9SDimitry Andric     else if (Feature == "+sramecc")
728e8d8bef9SDimitry Andric       SramEccRequested = true;
729e8d8bef9SDimitry Andric     else if (Feature == "-sramecc")
730e8d8bef9SDimitry Andric       SramEccRequested = false;
731e8d8bef9SDimitry Andric   }
732e8d8bef9SDimitry Andric 
733e8d8bef9SDimitry Andric   bool XnackSupported = isXnackSupported();
734e8d8bef9SDimitry Andric   bool SramEccSupported = isSramEccSupported();
735e8d8bef9SDimitry Andric 
736e8d8bef9SDimitry Andric   if (XnackRequested) {
737e8d8bef9SDimitry Andric     if (XnackSupported) {
738e8d8bef9SDimitry Andric       XnackSetting =
739e8d8bef9SDimitry Andric           *XnackRequested ? TargetIDSetting::On : TargetIDSetting::Off;
740e8d8bef9SDimitry Andric     } else {
741e8d8bef9SDimitry Andric       // If a specific xnack setting was requested and this GPU does not support
742e8d8bef9SDimitry Andric       // xnack emit a warning. Setting will remain set to "Unsupported".
743e8d8bef9SDimitry Andric       if (*XnackRequested) {
744e8d8bef9SDimitry Andric         errs() << "warning: xnack 'On' was requested for a processor that does "
745e8d8bef9SDimitry Andric                   "not support it!\n";
746e8d8bef9SDimitry Andric       } else {
747e8d8bef9SDimitry Andric         errs() << "warning: xnack 'Off' was requested for a processor that "
748e8d8bef9SDimitry Andric                   "does not support it!\n";
749e8d8bef9SDimitry Andric       }
750e8d8bef9SDimitry Andric     }
751e8d8bef9SDimitry Andric   }
752e8d8bef9SDimitry Andric 
753e8d8bef9SDimitry Andric   if (SramEccRequested) {
754e8d8bef9SDimitry Andric     if (SramEccSupported) {
755e8d8bef9SDimitry Andric       SramEccSetting =
756e8d8bef9SDimitry Andric           *SramEccRequested ? TargetIDSetting::On : TargetIDSetting::Off;
757e8d8bef9SDimitry Andric     } else {
758e8d8bef9SDimitry Andric       // If a specific sramecc setting was requested and this GPU does not
759e8d8bef9SDimitry Andric       // support sramecc emit a warning. Setting will remain set to
760e8d8bef9SDimitry Andric       // "Unsupported".
761e8d8bef9SDimitry Andric       if (*SramEccRequested) {
762e8d8bef9SDimitry Andric         errs() << "warning: sramecc 'On' was requested for a processor that "
763e8d8bef9SDimitry Andric                   "does not support it!\n";
764e8d8bef9SDimitry Andric       } else {
765e8d8bef9SDimitry Andric         errs() << "warning: sramecc 'Off' was requested for a processor that "
766e8d8bef9SDimitry Andric                   "does not support it!\n";
767e8d8bef9SDimitry Andric       }
768e8d8bef9SDimitry Andric     }
769e8d8bef9SDimitry Andric   }
770e8d8bef9SDimitry Andric }
771e8d8bef9SDimitry Andric 
772e8d8bef9SDimitry Andric static TargetIDSetting
getTargetIDSettingFromFeatureString(StringRef FeatureString)773e8d8bef9SDimitry Andric getTargetIDSettingFromFeatureString(StringRef FeatureString) {
7745f757f3fSDimitry Andric   if (FeatureString.ends_with("-"))
775e8d8bef9SDimitry Andric     return TargetIDSetting::Off;
7765f757f3fSDimitry Andric   if (FeatureString.ends_with("+"))
777e8d8bef9SDimitry Andric     return TargetIDSetting::On;
778e8d8bef9SDimitry Andric 
779e8d8bef9SDimitry Andric   llvm_unreachable("Malformed feature string");
780e8d8bef9SDimitry Andric }
781e8d8bef9SDimitry Andric 
setTargetIDFromTargetIDStream(StringRef TargetID)782e8d8bef9SDimitry Andric void AMDGPUTargetID::setTargetIDFromTargetIDStream(StringRef TargetID) {
783e8d8bef9SDimitry Andric   SmallVector<StringRef, 3> TargetIDSplit;
784e8d8bef9SDimitry Andric   TargetID.split(TargetIDSplit, ':');
785e8d8bef9SDimitry Andric 
786e8d8bef9SDimitry Andric   for (const auto &FeatureString : TargetIDSplit) {
7875f757f3fSDimitry Andric     if (FeatureString.starts_with("xnack"))
788e8d8bef9SDimitry Andric       XnackSetting = getTargetIDSettingFromFeatureString(FeatureString);
7895f757f3fSDimitry Andric     if (FeatureString.starts_with("sramecc"))
790e8d8bef9SDimitry Andric       SramEccSetting = getTargetIDSettingFromFeatureString(FeatureString);
791e8d8bef9SDimitry Andric   }
792e8d8bef9SDimitry Andric }
793e8d8bef9SDimitry Andric 
toString() const794fe6060f1SDimitry Andric std::string AMDGPUTargetID::toString() const {
79504eeddc0SDimitry Andric   std::string StringRep;
796fe6060f1SDimitry Andric   raw_string_ostream StreamRep(StringRep);
7970b57cec5SDimitry Andric 
798fe6060f1SDimitry Andric   auto TargetTriple = STI.getTargetTriple();
799fe6060f1SDimitry Andric   auto Version = getIsaVersion(STI.getCPU());
800fe6060f1SDimitry Andric 
801fe6060f1SDimitry Andric   StreamRep << TargetTriple.getArchName() << '-'
8020b57cec5SDimitry Andric             << TargetTriple.getVendorName() << '-'
8030b57cec5SDimitry Andric             << TargetTriple.getOSName() << '-'
804fe6060f1SDimitry Andric             << TargetTriple.getEnvironmentName() << '-';
8050b57cec5SDimitry Andric 
80604eeddc0SDimitry Andric   std::string Processor;
807fe6060f1SDimitry Andric   // TODO: Following else statement is present here because we used various
808fe6060f1SDimitry Andric   // alias names for GPUs up until GFX9 (e.g. 'fiji' is same as 'gfx803').
809fe6060f1SDimitry Andric   // Remove once all aliases are removed from GCNProcessors.td.
810fe6060f1SDimitry Andric   if (Version.Major >= 9)
811fe6060f1SDimitry Andric     Processor = STI.getCPU().str();
812fe6060f1SDimitry Andric   else
813fe6060f1SDimitry Andric     Processor = (Twine("gfx") + Twine(Version.Major) + Twine(Version.Minor) +
814fe6060f1SDimitry Andric                  Twine(Version.Stepping))
815fe6060f1SDimitry Andric                     .str();
8160b57cec5SDimitry Andric 
81704eeddc0SDimitry Andric   std::string Features;
81806c3fb27SDimitry Andric   if (STI.getTargetTriple().getOS() == Triple::AMDHSA) {
819fe6060f1SDimitry Andric     // sramecc.
820fe6060f1SDimitry Andric     if (getSramEccSetting() == TargetIDSetting::Off)
821fe6060f1SDimitry Andric       Features += ":sramecc-";
822fe6060f1SDimitry Andric     else if (getSramEccSetting() == TargetIDSetting::On)
823fe6060f1SDimitry Andric       Features += ":sramecc+";
824fe6060f1SDimitry Andric     // xnack.
825fe6060f1SDimitry Andric     if (getXnackSetting() == TargetIDSetting::Off)
826fe6060f1SDimitry Andric       Features += ":xnack-";
827fe6060f1SDimitry Andric     else if (getXnackSetting() == TargetIDSetting::On)
828fe6060f1SDimitry Andric       Features += ":xnack+";
829fe6060f1SDimitry Andric   }
830fe6060f1SDimitry Andric 
831fe6060f1SDimitry Andric   StreamRep << Processor << Features;
832fe6060f1SDimitry Andric 
833fe6060f1SDimitry Andric   StreamRep.flush();
834fe6060f1SDimitry Andric   return StringRep;
8350b57cec5SDimitry Andric }
8360b57cec5SDimitry Andric 
getWavefrontSize(const MCSubtargetInfo * STI)8370b57cec5SDimitry Andric unsigned getWavefrontSize(const MCSubtargetInfo *STI) {
8380b57cec5SDimitry Andric   if (STI->getFeatureBits().test(FeatureWavefrontSize16))
8390b57cec5SDimitry Andric     return 16;
8400b57cec5SDimitry Andric   if (STI->getFeatureBits().test(FeatureWavefrontSize32))
8410b57cec5SDimitry Andric     return 32;
8420b57cec5SDimitry Andric 
8430b57cec5SDimitry Andric   return 64;
8440b57cec5SDimitry Andric }
8450b57cec5SDimitry Andric 
getLocalMemorySize(const MCSubtargetInfo * STI)8460b57cec5SDimitry Andric unsigned getLocalMemorySize(const MCSubtargetInfo *STI) {
847bdd1243dSDimitry Andric   unsigned BytesPerCU = 0;
848bdd1243dSDimitry Andric   if (STI->getFeatureBits().test(FeatureLocalMemorySize32768))
849bdd1243dSDimitry Andric     BytesPerCU = 32768;
850bdd1243dSDimitry Andric   if (STI->getFeatureBits().test(FeatureLocalMemorySize65536))
851bdd1243dSDimitry Andric     BytesPerCU = 65536;
852bdd1243dSDimitry Andric 
853bdd1243dSDimitry Andric   // "Per CU" really means "per whatever functional block the waves of a
854bdd1243dSDimitry Andric   // workgroup must share". So the effective local memory size is doubled in
855bdd1243dSDimitry Andric   // WGP mode on gfx10.
856bdd1243dSDimitry Andric   if (isGFX10Plus(*STI) && !STI->getFeatureBits().test(FeatureCuMode))
857bdd1243dSDimitry Andric     BytesPerCU *= 2;
858bdd1243dSDimitry Andric 
859bdd1243dSDimitry Andric   return BytesPerCU;
860bdd1243dSDimitry Andric }
861bdd1243dSDimitry Andric 
getAddressableLocalMemorySize(const MCSubtargetInfo * STI)862bdd1243dSDimitry Andric unsigned getAddressableLocalMemorySize(const MCSubtargetInfo *STI) {
8630b57cec5SDimitry Andric   if (STI->getFeatureBits().test(FeatureLocalMemorySize32768))
8640b57cec5SDimitry Andric     return 32768;
8650b57cec5SDimitry Andric   if (STI->getFeatureBits().test(FeatureLocalMemorySize65536))
8660b57cec5SDimitry Andric     return 65536;
8670b57cec5SDimitry Andric   return 0;
8680b57cec5SDimitry Andric }
8690b57cec5SDimitry Andric 
getEUsPerCU(const MCSubtargetInfo * STI)8700b57cec5SDimitry Andric unsigned getEUsPerCU(const MCSubtargetInfo *STI) {
8715ffd83dbSDimitry Andric   // "Per CU" really means "per whatever functional block the waves of a
8725ffd83dbSDimitry Andric   // workgroup must share". For gfx10 in CU mode this is the CU, which contains
8735ffd83dbSDimitry Andric   // two SIMDs.
874e8d8bef9SDimitry Andric   if (isGFX10Plus(*STI) && STI->getFeatureBits().test(FeatureCuMode))
8755ffd83dbSDimitry Andric     return 2;
8765ffd83dbSDimitry Andric   // Pre-gfx10 a CU contains four SIMDs. For gfx10 in WGP mode the WGP contains
8775ffd83dbSDimitry Andric   // two CUs, so a total of four SIMDs.
8780b57cec5SDimitry Andric   return 4;
8790b57cec5SDimitry Andric }
8800b57cec5SDimitry Andric 
getMaxWorkGroupsPerCU(const MCSubtargetInfo * STI,unsigned FlatWorkGroupSize)8810b57cec5SDimitry Andric unsigned getMaxWorkGroupsPerCU(const MCSubtargetInfo *STI,
8820b57cec5SDimitry Andric                                unsigned FlatWorkGroupSize) {
8830b57cec5SDimitry Andric   assert(FlatWorkGroupSize != 0);
8840b57cec5SDimitry Andric   if (STI->getTargetTriple().getArch() != Triple::amdgcn)
8850b57cec5SDimitry Andric     return 8;
886bdd1243dSDimitry Andric   unsigned MaxWaves = getMaxWavesPerEU(STI) * getEUsPerCU(STI);
8870b57cec5SDimitry Andric   unsigned N = getWavesPerWorkGroup(STI, FlatWorkGroupSize);
888bdd1243dSDimitry Andric   if (N == 1) {
889bdd1243dSDimitry Andric     // Single-wave workgroups don't consume barrier resources.
890bdd1243dSDimitry Andric     return MaxWaves;
891bdd1243dSDimitry Andric   }
892bdd1243dSDimitry Andric 
893bdd1243dSDimitry Andric   unsigned MaxBarriers = 16;
894bdd1243dSDimitry Andric   if (isGFX10Plus(*STI) && !STI->getFeatureBits().test(FeatureCuMode))
895bdd1243dSDimitry Andric     MaxBarriers = 32;
896bdd1243dSDimitry Andric 
897bdd1243dSDimitry Andric   return std::min(MaxWaves / N, MaxBarriers);
8980b57cec5SDimitry Andric }
8990b57cec5SDimitry Andric 
getMinWavesPerEU(const MCSubtargetInfo * STI)9000b57cec5SDimitry Andric unsigned getMinWavesPerEU(const MCSubtargetInfo *STI) {
9010b57cec5SDimitry Andric   return 1;
9020b57cec5SDimitry Andric }
9030b57cec5SDimitry Andric 
getMaxWavesPerEU(const MCSubtargetInfo * STI)9048bcb0991SDimitry Andric unsigned getMaxWavesPerEU(const MCSubtargetInfo *STI) {
9050b57cec5SDimitry Andric   // FIXME: Need to take scratch memory into account.
906fe6060f1SDimitry Andric   if (isGFX90A(*STI))
907fe6060f1SDimitry Andric     return 8;
908e8d8bef9SDimitry Andric   if (!isGFX10Plus(*STI))
9090b57cec5SDimitry Andric     return 10;
9105ffd83dbSDimitry Andric   return hasGFX10_3Insts(*STI) ? 16 : 20;
9110b57cec5SDimitry Andric }
9120b57cec5SDimitry Andric 
getWavesPerEUForWorkGroup(const MCSubtargetInfo * STI,unsigned FlatWorkGroupSize)9135ffd83dbSDimitry Andric unsigned getWavesPerEUForWorkGroup(const MCSubtargetInfo *STI,
9140b57cec5SDimitry Andric                                    unsigned FlatWorkGroupSize) {
9155ffd83dbSDimitry Andric   return divideCeil(getWavesPerWorkGroup(STI, FlatWorkGroupSize),
9165ffd83dbSDimitry Andric                     getEUsPerCU(STI));
9170b57cec5SDimitry Andric }
9180b57cec5SDimitry Andric 
getMinFlatWorkGroupSize(const MCSubtargetInfo * STI)9190b57cec5SDimitry Andric unsigned getMinFlatWorkGroupSize(const MCSubtargetInfo *STI) {
9200b57cec5SDimitry Andric   return 1;
9210b57cec5SDimitry Andric }
9220b57cec5SDimitry Andric 
getMaxFlatWorkGroupSize(const MCSubtargetInfo * STI)9230b57cec5SDimitry Andric unsigned getMaxFlatWorkGroupSize(const MCSubtargetInfo *STI) {
924480093f4SDimitry Andric   // Some subtargets allow encoding 2048, but this isn't tested or supported.
925480093f4SDimitry Andric   return 1024;
9260b57cec5SDimitry Andric }
9270b57cec5SDimitry Andric 
getWavesPerWorkGroup(const MCSubtargetInfo * STI,unsigned FlatWorkGroupSize)9280b57cec5SDimitry Andric unsigned getWavesPerWorkGroup(const MCSubtargetInfo *STI,
9290b57cec5SDimitry Andric                               unsigned FlatWorkGroupSize) {
9305ffd83dbSDimitry Andric   return divideCeil(FlatWorkGroupSize, getWavefrontSize(STI));
9310b57cec5SDimitry Andric }
9320b57cec5SDimitry Andric 
getSGPRAllocGranule(const MCSubtargetInfo * STI)9330b57cec5SDimitry Andric unsigned getSGPRAllocGranule(const MCSubtargetInfo *STI) {
9340b57cec5SDimitry Andric   IsaVersion Version = getIsaVersion(STI->getCPU());
9350b57cec5SDimitry Andric   if (Version.Major >= 10)
9360b57cec5SDimitry Andric     return getAddressableNumSGPRs(STI);
9370b57cec5SDimitry Andric   if (Version.Major >= 8)
9380b57cec5SDimitry Andric     return 16;
9390b57cec5SDimitry Andric   return 8;
9400b57cec5SDimitry Andric }
9410b57cec5SDimitry Andric 
getSGPREncodingGranule(const MCSubtargetInfo * STI)9420b57cec5SDimitry Andric unsigned getSGPREncodingGranule(const MCSubtargetInfo *STI) {
9430b57cec5SDimitry Andric   return 8;
9440b57cec5SDimitry Andric }
9450b57cec5SDimitry Andric 
getTotalNumSGPRs(const MCSubtargetInfo * STI)9460b57cec5SDimitry Andric unsigned getTotalNumSGPRs(const MCSubtargetInfo *STI) {
9470b57cec5SDimitry Andric   IsaVersion Version = getIsaVersion(STI->getCPU());
9480b57cec5SDimitry Andric   if (Version.Major >= 8)
9490b57cec5SDimitry Andric     return 800;
9500b57cec5SDimitry Andric   return 512;
9510b57cec5SDimitry Andric }
9520b57cec5SDimitry Andric 
getAddressableNumSGPRs(const MCSubtargetInfo * STI)9530b57cec5SDimitry Andric unsigned getAddressableNumSGPRs(const MCSubtargetInfo *STI) {
9540b57cec5SDimitry Andric   if (STI->getFeatureBits().test(FeatureSGPRInitBug))
9550b57cec5SDimitry Andric     return FIXED_NUM_SGPRS_FOR_INIT_BUG;
9560b57cec5SDimitry Andric 
9570b57cec5SDimitry Andric   IsaVersion Version = getIsaVersion(STI->getCPU());
9580b57cec5SDimitry Andric   if (Version.Major >= 10)
9590b57cec5SDimitry Andric     return 106;
9600b57cec5SDimitry Andric   if (Version.Major >= 8)
9610b57cec5SDimitry Andric     return 102;
9620b57cec5SDimitry Andric   return 104;
9630b57cec5SDimitry Andric }
9640b57cec5SDimitry Andric 
getMinNumSGPRs(const MCSubtargetInfo * STI,unsigned WavesPerEU)9650b57cec5SDimitry Andric unsigned getMinNumSGPRs(const MCSubtargetInfo *STI, unsigned WavesPerEU) {
9660b57cec5SDimitry Andric   assert(WavesPerEU != 0);
9670b57cec5SDimitry Andric 
9680b57cec5SDimitry Andric   IsaVersion Version = getIsaVersion(STI->getCPU());
9690b57cec5SDimitry Andric   if (Version.Major >= 10)
9700b57cec5SDimitry Andric     return 0;
9710b57cec5SDimitry Andric 
9728bcb0991SDimitry Andric   if (WavesPerEU >= getMaxWavesPerEU(STI))
9730b57cec5SDimitry Andric     return 0;
9740b57cec5SDimitry Andric 
9750b57cec5SDimitry Andric   unsigned MinNumSGPRs = getTotalNumSGPRs(STI) / (WavesPerEU + 1);
9760b57cec5SDimitry Andric   if (STI->getFeatureBits().test(FeatureTrapHandler))
9770b57cec5SDimitry Andric     MinNumSGPRs -= std::min(MinNumSGPRs, (unsigned)TRAP_NUM_SGPRS);
9780b57cec5SDimitry Andric   MinNumSGPRs = alignDown(MinNumSGPRs, getSGPRAllocGranule(STI)) + 1;
9790b57cec5SDimitry Andric   return std::min(MinNumSGPRs, getAddressableNumSGPRs(STI));
9800b57cec5SDimitry Andric }
9810b57cec5SDimitry Andric 
getMaxNumSGPRs(const MCSubtargetInfo * STI,unsigned WavesPerEU,bool Addressable)9820b57cec5SDimitry Andric unsigned getMaxNumSGPRs(const MCSubtargetInfo *STI, unsigned WavesPerEU,
9830b57cec5SDimitry Andric                         bool Addressable) {
9840b57cec5SDimitry Andric   assert(WavesPerEU != 0);
9850b57cec5SDimitry Andric 
9860b57cec5SDimitry Andric   unsigned AddressableNumSGPRs = getAddressableNumSGPRs(STI);
9870b57cec5SDimitry Andric   IsaVersion Version = getIsaVersion(STI->getCPU());
9880b57cec5SDimitry Andric   if (Version.Major >= 10)
9890b57cec5SDimitry Andric     return Addressable ? AddressableNumSGPRs : 108;
9900b57cec5SDimitry Andric   if (Version.Major >= 8 && !Addressable)
9910b57cec5SDimitry Andric     AddressableNumSGPRs = 112;
9920b57cec5SDimitry Andric   unsigned MaxNumSGPRs = getTotalNumSGPRs(STI) / WavesPerEU;
9930b57cec5SDimitry Andric   if (STI->getFeatureBits().test(FeatureTrapHandler))
9940b57cec5SDimitry Andric     MaxNumSGPRs -= std::min(MaxNumSGPRs, (unsigned)TRAP_NUM_SGPRS);
9950b57cec5SDimitry Andric   MaxNumSGPRs = alignDown(MaxNumSGPRs, getSGPRAllocGranule(STI));
9960b57cec5SDimitry Andric   return std::min(MaxNumSGPRs, AddressableNumSGPRs);
9970b57cec5SDimitry Andric }
9980b57cec5SDimitry Andric 
getNumExtraSGPRs(const MCSubtargetInfo * STI,bool VCCUsed,bool FlatScrUsed,bool XNACKUsed)9990b57cec5SDimitry Andric unsigned getNumExtraSGPRs(const MCSubtargetInfo *STI, bool VCCUsed,
10000b57cec5SDimitry Andric                           bool FlatScrUsed, bool XNACKUsed) {
10010b57cec5SDimitry Andric   unsigned ExtraSGPRs = 0;
10020b57cec5SDimitry Andric   if (VCCUsed)
10030b57cec5SDimitry Andric     ExtraSGPRs = 2;
10040b57cec5SDimitry Andric 
10050b57cec5SDimitry Andric   IsaVersion Version = getIsaVersion(STI->getCPU());
10060b57cec5SDimitry Andric   if (Version.Major >= 10)
10070b57cec5SDimitry Andric     return ExtraSGPRs;
10080b57cec5SDimitry Andric 
10090b57cec5SDimitry Andric   if (Version.Major < 8) {
10100b57cec5SDimitry Andric     if (FlatScrUsed)
10110b57cec5SDimitry Andric       ExtraSGPRs = 4;
10120b57cec5SDimitry Andric   } else {
10130b57cec5SDimitry Andric     if (XNACKUsed)
10140b57cec5SDimitry Andric       ExtraSGPRs = 4;
10150b57cec5SDimitry Andric 
1016349cc55cSDimitry Andric     if (FlatScrUsed ||
1017349cc55cSDimitry Andric         STI->getFeatureBits().test(AMDGPU::FeatureArchitectedFlatScratch))
10180b57cec5SDimitry Andric       ExtraSGPRs = 6;
10190b57cec5SDimitry Andric   }
10200b57cec5SDimitry Andric 
10210b57cec5SDimitry Andric   return ExtraSGPRs;
10220b57cec5SDimitry Andric }
10230b57cec5SDimitry Andric 
getNumExtraSGPRs(const MCSubtargetInfo * STI,bool VCCUsed,bool FlatScrUsed)10240b57cec5SDimitry Andric unsigned getNumExtraSGPRs(const MCSubtargetInfo *STI, bool VCCUsed,
10250b57cec5SDimitry Andric                           bool FlatScrUsed) {
10260b57cec5SDimitry Andric   return getNumExtraSGPRs(STI, VCCUsed, FlatScrUsed,
10270b57cec5SDimitry Andric                           STI->getFeatureBits().test(AMDGPU::FeatureXNACK));
10280b57cec5SDimitry Andric }
10290b57cec5SDimitry Andric 
getNumSGPRBlocks(const MCSubtargetInfo * STI,unsigned NumSGPRs)10300b57cec5SDimitry Andric unsigned getNumSGPRBlocks(const MCSubtargetInfo *STI, unsigned NumSGPRs) {
10310b57cec5SDimitry Andric   NumSGPRs = alignTo(std::max(1u, NumSGPRs), getSGPREncodingGranule(STI));
10320b57cec5SDimitry Andric   // SGPRBlocks is actual number of SGPR blocks minus 1.
10330b57cec5SDimitry Andric   return NumSGPRs / getSGPREncodingGranule(STI) - 1;
10340b57cec5SDimitry Andric }
10350b57cec5SDimitry Andric 
getVGPRAllocGranule(const MCSubtargetInfo * STI,std::optional<bool> EnableWavefrontSize32)10360b57cec5SDimitry Andric unsigned getVGPRAllocGranule(const MCSubtargetInfo *STI,
1037bdd1243dSDimitry Andric                              std::optional<bool> EnableWavefrontSize32) {
1038fe6060f1SDimitry Andric   if (STI->getFeatureBits().test(FeatureGFX90AInsts))
1039fe6060f1SDimitry Andric     return 8;
1040fe6060f1SDimitry Andric 
10410b57cec5SDimitry Andric   bool IsWave32 = EnableWavefrontSize32 ?
10420b57cec5SDimitry Andric       *EnableWavefrontSize32 :
10430b57cec5SDimitry Andric       STI->getFeatureBits().test(FeatureWavefrontSize32);
10445ffd83dbSDimitry Andric 
1045bdd1243dSDimitry Andric   if (STI->getFeatureBits().test(FeatureGFX11FullVGPRs))
1046bdd1243dSDimitry Andric     return IsWave32 ? 24 : 12;
1047bdd1243dSDimitry Andric 
10485ffd83dbSDimitry Andric   if (hasGFX10_3Insts(*STI))
10495ffd83dbSDimitry Andric     return IsWave32 ? 16 : 8;
10505ffd83dbSDimitry Andric 
10510b57cec5SDimitry Andric   return IsWave32 ? 8 : 4;
10520b57cec5SDimitry Andric }
10530b57cec5SDimitry Andric 
getVGPREncodingGranule(const MCSubtargetInfo * STI,std::optional<bool> EnableWavefrontSize32)10540b57cec5SDimitry Andric unsigned getVGPREncodingGranule(const MCSubtargetInfo *STI,
1055bdd1243dSDimitry Andric                                 std::optional<bool> EnableWavefrontSize32) {
1056fe6060f1SDimitry Andric   if (STI->getFeatureBits().test(FeatureGFX90AInsts))
1057fe6060f1SDimitry Andric     return 8;
10585ffd83dbSDimitry Andric 
10595ffd83dbSDimitry Andric   bool IsWave32 = EnableWavefrontSize32 ?
10605ffd83dbSDimitry Andric       *EnableWavefrontSize32 :
10615ffd83dbSDimitry Andric       STI->getFeatureBits().test(FeatureWavefrontSize32);
10625ffd83dbSDimitry Andric 
10635ffd83dbSDimitry Andric   return IsWave32 ? 8 : 4;
10640b57cec5SDimitry Andric }
10650b57cec5SDimitry Andric 
getTotalNumVGPRs(const MCSubtargetInfo * STI)10660b57cec5SDimitry Andric unsigned getTotalNumVGPRs(const MCSubtargetInfo *STI) {
1067fe6060f1SDimitry Andric   if (STI->getFeatureBits().test(FeatureGFX90AInsts))
1068fe6060f1SDimitry Andric     return 512;
1069e8d8bef9SDimitry Andric   if (!isGFX10Plus(*STI))
10700b57cec5SDimitry Andric     return 256;
1071bdd1243dSDimitry Andric   bool IsWave32 = STI->getFeatureBits().test(FeatureWavefrontSize32);
1072bdd1243dSDimitry Andric   if (STI->getFeatureBits().test(FeatureGFX11FullVGPRs))
1073bdd1243dSDimitry Andric     return IsWave32 ? 1536 : 768;
1074bdd1243dSDimitry Andric   return IsWave32 ? 1024 : 512;
10750b57cec5SDimitry Andric }
10760b57cec5SDimitry Andric 
getAddressableNumVGPRs(const MCSubtargetInfo * STI)10770b57cec5SDimitry Andric unsigned getAddressableNumVGPRs(const MCSubtargetInfo *STI) {
1078fe6060f1SDimitry Andric   if (STI->getFeatureBits().test(FeatureGFX90AInsts))
1079fe6060f1SDimitry Andric     return 512;
10808bcb0991SDimitry Andric   return 256;
10810b57cec5SDimitry Andric }
10820b57cec5SDimitry Andric 
getNumWavesPerEUWithNumVGPRs(const MCSubtargetInfo * STI,unsigned NumVGPRs)1083bdd1243dSDimitry Andric unsigned getNumWavesPerEUWithNumVGPRs(const MCSubtargetInfo *STI,
1084bdd1243dSDimitry Andric                                       unsigned NumVGPRs) {
1085bdd1243dSDimitry Andric   unsigned MaxWaves = getMaxWavesPerEU(STI);
1086bdd1243dSDimitry Andric   unsigned Granule = getVGPRAllocGranule(STI);
1087bdd1243dSDimitry Andric   if (NumVGPRs < Granule)
1088bdd1243dSDimitry Andric     return MaxWaves;
1089bdd1243dSDimitry Andric   unsigned RoundedRegs = alignTo(NumVGPRs, Granule);
1090bdd1243dSDimitry Andric   return std::min(std::max(getTotalNumVGPRs(STI) / RoundedRegs, 1u), MaxWaves);
1091bdd1243dSDimitry Andric }
1092bdd1243dSDimitry Andric 
getMinNumVGPRs(const MCSubtargetInfo * STI,unsigned WavesPerEU)10930b57cec5SDimitry Andric unsigned getMinNumVGPRs(const MCSubtargetInfo *STI, unsigned WavesPerEU) {
10940b57cec5SDimitry Andric   assert(WavesPerEU != 0);
10950b57cec5SDimitry Andric 
1096bdd1243dSDimitry Andric   unsigned MaxWavesPerEU = getMaxWavesPerEU(STI);
1097bdd1243dSDimitry Andric   if (WavesPerEU >= MaxWavesPerEU)
10980b57cec5SDimitry Andric     return 0;
1099bdd1243dSDimitry Andric 
1100bdd1243dSDimitry Andric   unsigned TotNumVGPRs = getTotalNumVGPRs(STI);
1101bdd1243dSDimitry Andric   unsigned AddrsableNumVGPRs = getAddressableNumVGPRs(STI);
1102bdd1243dSDimitry Andric   unsigned Granule = getVGPRAllocGranule(STI);
1103bdd1243dSDimitry Andric   unsigned MaxNumVGPRs = alignDown(TotNumVGPRs / WavesPerEU, Granule);
1104bdd1243dSDimitry Andric 
1105bdd1243dSDimitry Andric   if (MaxNumVGPRs == alignDown(TotNumVGPRs / MaxWavesPerEU, Granule))
1106bdd1243dSDimitry Andric     return 0;
1107bdd1243dSDimitry Andric 
1108bdd1243dSDimitry Andric   unsigned MinWavesPerEU = getNumWavesPerEUWithNumVGPRs(STI, AddrsableNumVGPRs);
1109bdd1243dSDimitry Andric   if (WavesPerEU < MinWavesPerEU)
1110bdd1243dSDimitry Andric     return getMinNumVGPRs(STI, MinWavesPerEU);
1111bdd1243dSDimitry Andric 
1112bdd1243dSDimitry Andric   unsigned MaxNumVGPRsNext = alignDown(TotNumVGPRs / (WavesPerEU + 1), Granule);
1113bdd1243dSDimitry Andric   unsigned MinNumVGPRs = 1 + std::min(MaxNumVGPRs - Granule, MaxNumVGPRsNext);
1114bdd1243dSDimitry Andric   return std::min(MinNumVGPRs, AddrsableNumVGPRs);
11150b57cec5SDimitry Andric }
11160b57cec5SDimitry Andric 
getMaxNumVGPRs(const MCSubtargetInfo * STI,unsigned WavesPerEU)11170b57cec5SDimitry Andric unsigned getMaxNumVGPRs(const MCSubtargetInfo *STI, unsigned WavesPerEU) {
11180b57cec5SDimitry Andric   assert(WavesPerEU != 0);
11190b57cec5SDimitry Andric 
11200b57cec5SDimitry Andric   unsigned MaxNumVGPRs = alignDown(getTotalNumVGPRs(STI) / WavesPerEU,
11210b57cec5SDimitry Andric                                    getVGPRAllocGranule(STI));
11220b57cec5SDimitry Andric   unsigned AddressableNumVGPRs = getAddressableNumVGPRs(STI);
11230b57cec5SDimitry Andric   return std::min(MaxNumVGPRs, AddressableNumVGPRs);
11240b57cec5SDimitry Andric }
11250b57cec5SDimitry Andric 
getNumVGPRBlocks(const MCSubtargetInfo * STI,unsigned NumVGPRs,std::optional<bool> EnableWavefrontSize32)11260b57cec5SDimitry Andric unsigned getNumVGPRBlocks(const MCSubtargetInfo *STI, unsigned NumVGPRs,
1127bdd1243dSDimitry Andric                           std::optional<bool> EnableWavefrontSize32) {
11280b57cec5SDimitry Andric   NumVGPRs = alignTo(std::max(1u, NumVGPRs),
11290b57cec5SDimitry Andric                      getVGPREncodingGranule(STI, EnableWavefrontSize32));
11300b57cec5SDimitry Andric   // VGPRBlocks is actual number of VGPR blocks minus 1.
11310b57cec5SDimitry Andric   return NumVGPRs / getVGPREncodingGranule(STI, EnableWavefrontSize32) - 1;
11320b57cec5SDimitry Andric }
11330b57cec5SDimitry Andric 
11340b57cec5SDimitry Andric } // end namespace IsaInfo
11350b57cec5SDimitry Andric 
initDefaultAMDKernelCodeT(amd_kernel_code_t & Header,const MCSubtargetInfo * STI)11360b57cec5SDimitry Andric void initDefaultAMDKernelCodeT(amd_kernel_code_t &Header,
11370b57cec5SDimitry Andric                                const MCSubtargetInfo *STI) {
11380b57cec5SDimitry Andric   IsaVersion Version = getIsaVersion(STI->getCPU());
11390b57cec5SDimitry Andric 
11400b57cec5SDimitry Andric   memset(&Header, 0, sizeof(Header));
11410b57cec5SDimitry Andric 
11420b57cec5SDimitry Andric   Header.amd_kernel_code_version_major = 1;
11430b57cec5SDimitry Andric   Header.amd_kernel_code_version_minor = 2;
11440b57cec5SDimitry Andric   Header.amd_machine_kind = 1; // AMD_MACHINE_KIND_AMDGPU
11450b57cec5SDimitry Andric   Header.amd_machine_version_major = Version.Major;
11460b57cec5SDimitry Andric   Header.amd_machine_version_minor = Version.Minor;
11470b57cec5SDimitry Andric   Header.amd_machine_version_stepping = Version.Stepping;
11480b57cec5SDimitry Andric   Header.kernel_code_entry_byte_offset = sizeof(Header);
11490b57cec5SDimitry Andric   Header.wavefront_size = 6;
11500b57cec5SDimitry Andric 
11510b57cec5SDimitry Andric   // If the code object does not support indirect functions, then the value must
11520b57cec5SDimitry Andric   // be 0xffffffff.
11530b57cec5SDimitry Andric   Header.call_convention = -1;
11540b57cec5SDimitry Andric 
11550b57cec5SDimitry Andric   // These alignment values are specified in powers of two, so alignment =
11560b57cec5SDimitry Andric   // 2^n.  The minimum alignment is 2^4 = 16.
11570b57cec5SDimitry Andric   Header.kernarg_segment_alignment = 4;
11580b57cec5SDimitry Andric   Header.group_segment_alignment = 4;
11590b57cec5SDimitry Andric   Header.private_segment_alignment = 4;
11600b57cec5SDimitry Andric 
11610b57cec5SDimitry Andric   if (Version.Major >= 10) {
11620b57cec5SDimitry Andric     if (STI->getFeatureBits().test(FeatureWavefrontSize32)) {
11630b57cec5SDimitry Andric       Header.wavefront_size = 5;
11640b57cec5SDimitry Andric       Header.code_properties |= AMD_CODE_PROPERTY_ENABLE_WAVEFRONT_SIZE32;
11650b57cec5SDimitry Andric     }
11660b57cec5SDimitry Andric     Header.compute_pgm_resource_registers |=
11670b57cec5SDimitry Andric       S_00B848_WGP_MODE(STI->getFeatureBits().test(FeatureCuMode) ? 0 : 1) |
11680b57cec5SDimitry Andric       S_00B848_MEM_ORDERED(1);
11690b57cec5SDimitry Andric   }
11700b57cec5SDimitry Andric }
11710b57cec5SDimitry Andric 
getDefaultAmdhsaKernelDescriptor(const MCSubtargetInfo * STI)11720b57cec5SDimitry Andric amdhsa::kernel_descriptor_t getDefaultAmdhsaKernelDescriptor(
11730b57cec5SDimitry Andric     const MCSubtargetInfo *STI) {
11740b57cec5SDimitry Andric   IsaVersion Version = getIsaVersion(STI->getCPU());
11750b57cec5SDimitry Andric 
11760b57cec5SDimitry Andric   amdhsa::kernel_descriptor_t KD;
11770b57cec5SDimitry Andric   memset(&KD, 0, sizeof(KD));
11780b57cec5SDimitry Andric 
11790b57cec5SDimitry Andric   AMDHSA_BITS_SET(KD.compute_pgm_rsrc1,
11800b57cec5SDimitry Andric                   amdhsa::COMPUTE_PGM_RSRC1_FLOAT_DENORM_MODE_16_64,
11810b57cec5SDimitry Andric                   amdhsa::FLOAT_DENORM_MODE_FLUSH_NONE);
11825f757f3fSDimitry Andric   if (Version.Major >= 12) {
11830b57cec5SDimitry Andric     AMDHSA_BITS_SET(KD.compute_pgm_rsrc1,
11845f757f3fSDimitry Andric                     amdhsa::COMPUTE_PGM_RSRC1_GFX12_PLUS_ENABLE_WG_RR_EN, 0);
11850b57cec5SDimitry Andric     AMDHSA_BITS_SET(KD.compute_pgm_rsrc1,
11865f757f3fSDimitry Andric                     amdhsa::COMPUTE_PGM_RSRC1_GFX12_PLUS_DISABLE_PERF, 0);
11875f757f3fSDimitry Andric   } else {
11885f757f3fSDimitry Andric     AMDHSA_BITS_SET(KD.compute_pgm_rsrc1,
11895f757f3fSDimitry Andric                     amdhsa::COMPUTE_PGM_RSRC1_GFX6_GFX11_ENABLE_DX10_CLAMP, 1);
11905f757f3fSDimitry Andric     AMDHSA_BITS_SET(KD.compute_pgm_rsrc1,
11915f757f3fSDimitry Andric                     amdhsa::COMPUTE_PGM_RSRC1_GFX6_GFX11_ENABLE_IEEE_MODE, 1);
11925f757f3fSDimitry Andric   }
11930b57cec5SDimitry Andric   AMDHSA_BITS_SET(KD.compute_pgm_rsrc2,
11940b57cec5SDimitry Andric                   amdhsa::COMPUTE_PGM_RSRC2_ENABLE_SGPR_WORKGROUP_ID_X, 1);
11950b57cec5SDimitry Andric   if (Version.Major >= 10) {
11960b57cec5SDimitry Andric     AMDHSA_BITS_SET(KD.kernel_code_properties,
11970b57cec5SDimitry Andric                     amdhsa::KERNEL_CODE_PROPERTY_ENABLE_WAVEFRONT_SIZE32,
11980b57cec5SDimitry Andric                     STI->getFeatureBits().test(FeatureWavefrontSize32) ? 1 : 0);
11990b57cec5SDimitry Andric     AMDHSA_BITS_SET(KD.compute_pgm_rsrc1,
12005f757f3fSDimitry Andric                     amdhsa::COMPUTE_PGM_RSRC1_GFX10_PLUS_WGP_MODE,
12010b57cec5SDimitry Andric                     STI->getFeatureBits().test(FeatureCuMode) ? 0 : 1);
12020b57cec5SDimitry Andric     AMDHSA_BITS_SET(KD.compute_pgm_rsrc1,
12035f757f3fSDimitry Andric                     amdhsa::COMPUTE_PGM_RSRC1_GFX10_PLUS_MEM_ORDERED, 1);
12040b57cec5SDimitry Andric   }
1205fe6060f1SDimitry Andric   if (AMDGPU::isGFX90A(*STI)) {
1206fe6060f1SDimitry Andric     AMDHSA_BITS_SET(KD.compute_pgm_rsrc3,
1207fe6060f1SDimitry Andric                     amdhsa::COMPUTE_PGM_RSRC3_GFX90A_TG_SPLIT,
1208fe6060f1SDimitry Andric                     STI->getFeatureBits().test(FeatureTgSplit) ? 1 : 0);
1209fe6060f1SDimitry Andric   }
12100b57cec5SDimitry Andric   return KD;
12110b57cec5SDimitry Andric }
12120b57cec5SDimitry Andric 
isGroupSegment(const GlobalValue * GV)12130b57cec5SDimitry Andric bool isGroupSegment(const GlobalValue *GV) {
1214480093f4SDimitry Andric   return GV->getAddressSpace() == AMDGPUAS::LOCAL_ADDRESS;
12150b57cec5SDimitry Andric }
12160b57cec5SDimitry Andric 
isGlobalSegment(const GlobalValue * GV)12170b57cec5SDimitry Andric bool isGlobalSegment(const GlobalValue *GV) {
1218480093f4SDimitry Andric   return GV->getAddressSpace() == AMDGPUAS::GLOBAL_ADDRESS;
12190b57cec5SDimitry Andric }
12200b57cec5SDimitry Andric 
isReadOnlySegment(const GlobalValue * GV)12210b57cec5SDimitry Andric bool isReadOnlySegment(const GlobalValue *GV) {
1222480093f4SDimitry Andric   unsigned AS = GV->getAddressSpace();
1223480093f4SDimitry Andric   return AS == AMDGPUAS::CONSTANT_ADDRESS ||
1224480093f4SDimitry Andric          AS == AMDGPUAS::CONSTANT_ADDRESS_32BIT;
12250b57cec5SDimitry Andric }
12260b57cec5SDimitry Andric 
shouldEmitConstantsToTextSection(const Triple & TT)12270b57cec5SDimitry Andric bool shouldEmitConstantsToTextSection(const Triple &TT) {
1228e8d8bef9SDimitry Andric   return TT.getArch() == Triple::r600;
12290b57cec5SDimitry Andric }
12300b57cec5SDimitry Andric 
123106c3fb27SDimitry Andric std::pair<unsigned, unsigned>
getIntegerPairAttribute(const Function & F,StringRef Name,std::pair<unsigned,unsigned> Default,bool OnlyFirstRequired)123206c3fb27SDimitry Andric getIntegerPairAttribute(const Function &F, StringRef Name,
123306c3fb27SDimitry Andric                         std::pair<unsigned, unsigned> Default,
12340b57cec5SDimitry Andric                         bool OnlyFirstRequired) {
12350b57cec5SDimitry Andric   Attribute A = F.getFnAttribute(Name);
12360b57cec5SDimitry Andric   if (!A.isStringAttribute())
12370b57cec5SDimitry Andric     return Default;
12380b57cec5SDimitry Andric 
12390b57cec5SDimitry Andric   LLVMContext &Ctx = F.getContext();
124006c3fb27SDimitry Andric   std::pair<unsigned, unsigned> Ints = Default;
12410b57cec5SDimitry Andric   std::pair<StringRef, StringRef> Strs = A.getValueAsString().split(',');
12420b57cec5SDimitry Andric   if (Strs.first.trim().getAsInteger(0, Ints.first)) {
12430b57cec5SDimitry Andric     Ctx.emitError("can't parse first integer attribute " + Name);
12440b57cec5SDimitry Andric     return Default;
12450b57cec5SDimitry Andric   }
12460b57cec5SDimitry Andric   if (Strs.second.trim().getAsInteger(0, Ints.second)) {
12470b57cec5SDimitry Andric     if (!OnlyFirstRequired || !Strs.second.trim().empty()) {
12480b57cec5SDimitry Andric       Ctx.emitError("can't parse second integer attribute " + Name);
12490b57cec5SDimitry Andric       return Default;
12500b57cec5SDimitry Andric     }
12510b57cec5SDimitry Andric   }
12520b57cec5SDimitry Andric 
12530b57cec5SDimitry Andric   return Ints;
12540b57cec5SDimitry Andric }
12550b57cec5SDimitry Andric 
getVmcntBitMask(const IsaVersion & Version)12560b57cec5SDimitry Andric unsigned getVmcntBitMask(const IsaVersion &Version) {
125781ad6265SDimitry Andric   return (1 << (getVmcntBitWidthLo(Version.Major) +
125881ad6265SDimitry Andric                 getVmcntBitWidthHi(Version.Major))) -
125981ad6265SDimitry Andric          1;
12600b57cec5SDimitry Andric }
12610b57cec5SDimitry Andric 
getLoadcntBitMask(const IsaVersion & Version)12627a6dacacSDimitry Andric unsigned getLoadcntBitMask(const IsaVersion &Version) {
12637a6dacacSDimitry Andric   return (1 << getLoadcntBitWidth(Version.Major)) - 1;
12647a6dacacSDimitry Andric }
12657a6dacacSDimitry Andric 
getSamplecntBitMask(const IsaVersion & Version)12667a6dacacSDimitry Andric unsigned getSamplecntBitMask(const IsaVersion &Version) {
12677a6dacacSDimitry Andric   return (1 << getSamplecntBitWidth(Version.Major)) - 1;
12687a6dacacSDimitry Andric }
12697a6dacacSDimitry Andric 
getBvhcntBitMask(const IsaVersion & Version)12707a6dacacSDimitry Andric unsigned getBvhcntBitMask(const IsaVersion &Version) {
12717a6dacacSDimitry Andric   return (1 << getBvhcntBitWidth(Version.Major)) - 1;
12727a6dacacSDimitry Andric }
12737a6dacacSDimitry Andric 
getExpcntBitMask(const IsaVersion & Version)12740b57cec5SDimitry Andric unsigned getExpcntBitMask(const IsaVersion &Version) {
127581ad6265SDimitry Andric   return (1 << getExpcntBitWidth(Version.Major)) - 1;
12760b57cec5SDimitry Andric }
12770b57cec5SDimitry Andric 
getLgkmcntBitMask(const IsaVersion & Version)12780b57cec5SDimitry Andric unsigned getLgkmcntBitMask(const IsaVersion &Version) {
12790b57cec5SDimitry Andric   return (1 << getLgkmcntBitWidth(Version.Major)) - 1;
12800b57cec5SDimitry Andric }
12810b57cec5SDimitry Andric 
getDscntBitMask(const IsaVersion & Version)12827a6dacacSDimitry Andric unsigned getDscntBitMask(const IsaVersion &Version) {
12837a6dacacSDimitry Andric   return (1 << getDscntBitWidth(Version.Major)) - 1;
12847a6dacacSDimitry Andric }
12857a6dacacSDimitry Andric 
getKmcntBitMask(const IsaVersion & Version)12867a6dacacSDimitry Andric unsigned getKmcntBitMask(const IsaVersion &Version) {
12877a6dacacSDimitry Andric   return (1 << getKmcntBitWidth(Version.Major)) - 1;
12887a6dacacSDimitry Andric }
12897a6dacacSDimitry Andric 
getStorecntBitMask(const IsaVersion & Version)12907a6dacacSDimitry Andric unsigned getStorecntBitMask(const IsaVersion &Version) {
12917a6dacacSDimitry Andric   return (1 << getStorecntBitWidth(Version.Major)) - 1;
12927a6dacacSDimitry Andric }
12937a6dacacSDimitry Andric 
getWaitcntBitMask(const IsaVersion & Version)12940b57cec5SDimitry Andric unsigned getWaitcntBitMask(const IsaVersion &Version) {
129581ad6265SDimitry Andric   unsigned VmcntLo = getBitMask(getVmcntBitShiftLo(Version.Major),
129681ad6265SDimitry Andric                                 getVmcntBitWidthLo(Version.Major));
129781ad6265SDimitry Andric   unsigned Expcnt = getBitMask(getExpcntBitShift(Version.Major),
129881ad6265SDimitry Andric                                getExpcntBitWidth(Version.Major));
129981ad6265SDimitry Andric   unsigned Lgkmcnt = getBitMask(getLgkmcntBitShift(Version.Major),
13000b57cec5SDimitry Andric                                 getLgkmcntBitWidth(Version.Major));
130181ad6265SDimitry Andric   unsigned VmcntHi = getBitMask(getVmcntBitShiftHi(Version.Major),
130281ad6265SDimitry Andric                                 getVmcntBitWidthHi(Version.Major));
130381ad6265SDimitry Andric   return VmcntLo | Expcnt | Lgkmcnt | VmcntHi;
13040b57cec5SDimitry Andric }
13050b57cec5SDimitry Andric 
decodeVmcnt(const IsaVersion & Version,unsigned Waitcnt)13060b57cec5SDimitry Andric unsigned decodeVmcnt(const IsaVersion &Version, unsigned Waitcnt) {
130781ad6265SDimitry Andric   unsigned VmcntLo = unpackBits(Waitcnt, getVmcntBitShiftLo(Version.Major),
130881ad6265SDimitry Andric                                 getVmcntBitWidthLo(Version.Major));
130981ad6265SDimitry Andric   unsigned VmcntHi = unpackBits(Waitcnt, getVmcntBitShiftHi(Version.Major),
131081ad6265SDimitry Andric                                 getVmcntBitWidthHi(Version.Major));
131181ad6265SDimitry Andric   return VmcntLo | VmcntHi << getVmcntBitWidthLo(Version.Major);
13120b57cec5SDimitry Andric }
13130b57cec5SDimitry Andric 
decodeExpcnt(const IsaVersion & Version,unsigned Waitcnt)13140b57cec5SDimitry Andric unsigned decodeExpcnt(const IsaVersion &Version, unsigned Waitcnt) {
131581ad6265SDimitry Andric   return unpackBits(Waitcnt, getExpcntBitShift(Version.Major),
131681ad6265SDimitry Andric                     getExpcntBitWidth(Version.Major));
13170b57cec5SDimitry Andric }
13180b57cec5SDimitry Andric 
decodeLgkmcnt(const IsaVersion & Version,unsigned Waitcnt)13190b57cec5SDimitry Andric unsigned decodeLgkmcnt(const IsaVersion &Version, unsigned Waitcnt) {
132081ad6265SDimitry Andric   return unpackBits(Waitcnt, getLgkmcntBitShift(Version.Major),
13210b57cec5SDimitry Andric                     getLgkmcntBitWidth(Version.Major));
13220b57cec5SDimitry Andric }
13230b57cec5SDimitry Andric 
decodeWaitcnt(const IsaVersion & Version,unsigned Waitcnt,unsigned & Vmcnt,unsigned & Expcnt,unsigned & Lgkmcnt)13240b57cec5SDimitry Andric void decodeWaitcnt(const IsaVersion &Version, unsigned Waitcnt,
13250b57cec5SDimitry Andric                    unsigned &Vmcnt, unsigned &Expcnt, unsigned &Lgkmcnt) {
13260b57cec5SDimitry Andric   Vmcnt = decodeVmcnt(Version, Waitcnt);
13270b57cec5SDimitry Andric   Expcnt = decodeExpcnt(Version, Waitcnt);
13280b57cec5SDimitry Andric   Lgkmcnt = decodeLgkmcnt(Version, Waitcnt);
13290b57cec5SDimitry Andric }
13300b57cec5SDimitry Andric 
decodeWaitcnt(const IsaVersion & Version,unsigned Encoded)13310b57cec5SDimitry Andric Waitcnt decodeWaitcnt(const IsaVersion &Version, unsigned Encoded) {
13320b57cec5SDimitry Andric   Waitcnt Decoded;
13337a6dacacSDimitry Andric   Decoded.LoadCnt = decodeVmcnt(Version, Encoded);
13340b57cec5SDimitry Andric   Decoded.ExpCnt = decodeExpcnt(Version, Encoded);
13357a6dacacSDimitry Andric   Decoded.DsCnt = decodeLgkmcnt(Version, Encoded);
13360b57cec5SDimitry Andric   return Decoded;
13370b57cec5SDimitry Andric }
13380b57cec5SDimitry Andric 
encodeVmcnt(const IsaVersion & Version,unsigned Waitcnt,unsigned Vmcnt)13390b57cec5SDimitry Andric unsigned encodeVmcnt(const IsaVersion &Version, unsigned Waitcnt,
13400b57cec5SDimitry Andric                      unsigned Vmcnt) {
134181ad6265SDimitry Andric   Waitcnt = packBits(Vmcnt, Waitcnt, getVmcntBitShiftLo(Version.Major),
134281ad6265SDimitry Andric                      getVmcntBitWidthLo(Version.Major));
134381ad6265SDimitry Andric   return packBits(Vmcnt >> getVmcntBitWidthLo(Version.Major), Waitcnt,
134481ad6265SDimitry Andric                   getVmcntBitShiftHi(Version.Major),
134581ad6265SDimitry Andric                   getVmcntBitWidthHi(Version.Major));
13460b57cec5SDimitry Andric }
13470b57cec5SDimitry Andric 
encodeExpcnt(const IsaVersion & Version,unsigned Waitcnt,unsigned Expcnt)13480b57cec5SDimitry Andric unsigned encodeExpcnt(const IsaVersion &Version, unsigned Waitcnt,
13490b57cec5SDimitry Andric                       unsigned Expcnt) {
135081ad6265SDimitry Andric   return packBits(Expcnt, Waitcnt, getExpcntBitShift(Version.Major),
135181ad6265SDimitry Andric                   getExpcntBitWidth(Version.Major));
13520b57cec5SDimitry Andric }
13530b57cec5SDimitry Andric 
encodeLgkmcnt(const IsaVersion & Version,unsigned Waitcnt,unsigned Lgkmcnt)13540b57cec5SDimitry Andric unsigned encodeLgkmcnt(const IsaVersion &Version, unsigned Waitcnt,
13550b57cec5SDimitry Andric                        unsigned Lgkmcnt) {
135681ad6265SDimitry Andric   return packBits(Lgkmcnt, Waitcnt, getLgkmcntBitShift(Version.Major),
13570b57cec5SDimitry Andric                   getLgkmcntBitWidth(Version.Major));
13580b57cec5SDimitry Andric }
13590b57cec5SDimitry Andric 
encodeWaitcnt(const IsaVersion & Version,unsigned Vmcnt,unsigned Expcnt,unsigned Lgkmcnt)13600b57cec5SDimitry Andric unsigned encodeWaitcnt(const IsaVersion &Version,
13610b57cec5SDimitry Andric                        unsigned Vmcnt, unsigned Expcnt, unsigned Lgkmcnt) {
13620b57cec5SDimitry Andric   unsigned Waitcnt = getWaitcntBitMask(Version);
13630b57cec5SDimitry Andric   Waitcnt = encodeVmcnt(Version, Waitcnt, Vmcnt);
13640b57cec5SDimitry Andric   Waitcnt = encodeExpcnt(Version, Waitcnt, Expcnt);
13650b57cec5SDimitry Andric   Waitcnt = encodeLgkmcnt(Version, Waitcnt, Lgkmcnt);
13660b57cec5SDimitry Andric   return Waitcnt;
13670b57cec5SDimitry Andric }
13680b57cec5SDimitry Andric 
encodeWaitcnt(const IsaVersion & Version,const Waitcnt & Decoded)13690b57cec5SDimitry Andric unsigned encodeWaitcnt(const IsaVersion &Version, const Waitcnt &Decoded) {
13707a6dacacSDimitry Andric   return encodeWaitcnt(Version, Decoded.LoadCnt, Decoded.ExpCnt, Decoded.DsCnt);
13717a6dacacSDimitry Andric }
13727a6dacacSDimitry Andric 
getCombinedCountBitMask(const IsaVersion & Version,bool IsStore)13737a6dacacSDimitry Andric static unsigned getCombinedCountBitMask(const IsaVersion &Version,
13747a6dacacSDimitry Andric                                         bool IsStore) {
13757a6dacacSDimitry Andric   unsigned Dscnt = getBitMask(getDscntBitShift(Version.Major),
13767a6dacacSDimitry Andric                               getDscntBitWidth(Version.Major));
13777a6dacacSDimitry Andric   if (IsStore) {
13787a6dacacSDimitry Andric     unsigned Storecnt = getBitMask(getLoadcntStorecntBitShift(Version.Major),
13797a6dacacSDimitry Andric                                    getStorecntBitWidth(Version.Major));
13807a6dacacSDimitry Andric     return Dscnt | Storecnt;
13817a6dacacSDimitry Andric   } else {
13827a6dacacSDimitry Andric     unsigned Loadcnt = getBitMask(getLoadcntStorecntBitShift(Version.Major),
13837a6dacacSDimitry Andric                                   getLoadcntBitWidth(Version.Major));
13847a6dacacSDimitry Andric     return Dscnt | Loadcnt;
13857a6dacacSDimitry Andric   }
13867a6dacacSDimitry Andric }
13877a6dacacSDimitry Andric 
decodeLoadcntDscnt(const IsaVersion & Version,unsigned LoadcntDscnt)13887a6dacacSDimitry Andric Waitcnt decodeLoadcntDscnt(const IsaVersion &Version, unsigned LoadcntDscnt) {
13897a6dacacSDimitry Andric   Waitcnt Decoded;
13907a6dacacSDimitry Andric   Decoded.LoadCnt =
13917a6dacacSDimitry Andric       unpackBits(LoadcntDscnt, getLoadcntStorecntBitShift(Version.Major),
13927a6dacacSDimitry Andric                  getLoadcntBitWidth(Version.Major));
13937a6dacacSDimitry Andric   Decoded.DsCnt = unpackBits(LoadcntDscnt, getDscntBitShift(Version.Major),
13947a6dacacSDimitry Andric                              getDscntBitWidth(Version.Major));
13957a6dacacSDimitry Andric   return Decoded;
13967a6dacacSDimitry Andric }
13977a6dacacSDimitry Andric 
decodeStorecntDscnt(const IsaVersion & Version,unsigned StorecntDscnt)13987a6dacacSDimitry Andric Waitcnt decodeStorecntDscnt(const IsaVersion &Version, unsigned StorecntDscnt) {
13997a6dacacSDimitry Andric   Waitcnt Decoded;
14007a6dacacSDimitry Andric   Decoded.StoreCnt =
14017a6dacacSDimitry Andric       unpackBits(StorecntDscnt, getLoadcntStorecntBitShift(Version.Major),
14027a6dacacSDimitry Andric                  getStorecntBitWidth(Version.Major));
14037a6dacacSDimitry Andric   Decoded.DsCnt = unpackBits(StorecntDscnt, getDscntBitShift(Version.Major),
14047a6dacacSDimitry Andric                              getDscntBitWidth(Version.Major));
14057a6dacacSDimitry Andric   return Decoded;
14067a6dacacSDimitry Andric }
14077a6dacacSDimitry Andric 
encodeLoadcnt(const IsaVersion & Version,unsigned Waitcnt,unsigned Loadcnt)14087a6dacacSDimitry Andric static unsigned encodeLoadcnt(const IsaVersion &Version, unsigned Waitcnt,
14097a6dacacSDimitry Andric                               unsigned Loadcnt) {
14107a6dacacSDimitry Andric   return packBits(Loadcnt, Waitcnt, getLoadcntStorecntBitShift(Version.Major),
14117a6dacacSDimitry Andric                   getLoadcntBitWidth(Version.Major));
14127a6dacacSDimitry Andric }
14137a6dacacSDimitry Andric 
encodeStorecnt(const IsaVersion & Version,unsigned Waitcnt,unsigned Storecnt)14147a6dacacSDimitry Andric static unsigned encodeStorecnt(const IsaVersion &Version, unsigned Waitcnt,
14157a6dacacSDimitry Andric                                unsigned Storecnt) {
14167a6dacacSDimitry Andric   return packBits(Storecnt, Waitcnt, getLoadcntStorecntBitShift(Version.Major),
14177a6dacacSDimitry Andric                   getStorecntBitWidth(Version.Major));
14187a6dacacSDimitry Andric }
14197a6dacacSDimitry Andric 
encodeDscnt(const IsaVersion & Version,unsigned Waitcnt,unsigned Dscnt)14207a6dacacSDimitry Andric static unsigned encodeDscnt(const IsaVersion &Version, unsigned Waitcnt,
14217a6dacacSDimitry Andric                             unsigned Dscnt) {
14227a6dacacSDimitry Andric   return packBits(Dscnt, Waitcnt, getDscntBitShift(Version.Major),
14237a6dacacSDimitry Andric                   getDscntBitWidth(Version.Major));
14247a6dacacSDimitry Andric }
14257a6dacacSDimitry Andric 
encodeLoadcntDscnt(const IsaVersion & Version,unsigned Loadcnt,unsigned Dscnt)14267a6dacacSDimitry Andric static unsigned encodeLoadcntDscnt(const IsaVersion &Version, unsigned Loadcnt,
14277a6dacacSDimitry Andric                                    unsigned Dscnt) {
14287a6dacacSDimitry Andric   unsigned Waitcnt = getCombinedCountBitMask(Version, false);
14297a6dacacSDimitry Andric   Waitcnt = encodeLoadcnt(Version, Waitcnt, Loadcnt);
14307a6dacacSDimitry Andric   Waitcnt = encodeDscnt(Version, Waitcnt, Dscnt);
14317a6dacacSDimitry Andric   return Waitcnt;
14327a6dacacSDimitry Andric }
14337a6dacacSDimitry Andric 
encodeLoadcntDscnt(const IsaVersion & Version,const Waitcnt & Decoded)14347a6dacacSDimitry Andric unsigned encodeLoadcntDscnt(const IsaVersion &Version, const Waitcnt &Decoded) {
14357a6dacacSDimitry Andric   return encodeLoadcntDscnt(Version, Decoded.LoadCnt, Decoded.DsCnt);
14367a6dacacSDimitry Andric }
14377a6dacacSDimitry Andric 
encodeStorecntDscnt(const IsaVersion & Version,unsigned Storecnt,unsigned Dscnt)14387a6dacacSDimitry Andric static unsigned encodeStorecntDscnt(const IsaVersion &Version,
14397a6dacacSDimitry Andric                                     unsigned Storecnt, unsigned Dscnt) {
14407a6dacacSDimitry Andric   unsigned Waitcnt = getCombinedCountBitMask(Version, true);
14417a6dacacSDimitry Andric   Waitcnt = encodeStorecnt(Version, Waitcnt, Storecnt);
14427a6dacacSDimitry Andric   Waitcnt = encodeDscnt(Version, Waitcnt, Dscnt);
14437a6dacacSDimitry Andric   return Waitcnt;
14447a6dacacSDimitry Andric }
14457a6dacacSDimitry Andric 
encodeStorecntDscnt(const IsaVersion & Version,const Waitcnt & Decoded)14467a6dacacSDimitry Andric unsigned encodeStorecntDscnt(const IsaVersion &Version,
14477a6dacacSDimitry Andric                              const Waitcnt &Decoded) {
14487a6dacacSDimitry Andric   return encodeStorecntDscnt(Version, Decoded.StoreCnt, Decoded.DsCnt);
14490b57cec5SDimitry Andric }
14500b57cec5SDimitry Andric 
14510b57cec5SDimitry Andric //===----------------------------------------------------------------------===//
145281ad6265SDimitry Andric // Custom Operands.
145381ad6265SDimitry Andric //
145481ad6265SDimitry Andric // A table of custom operands shall describe "primary" operand names
145581ad6265SDimitry Andric // first followed by aliases if any. It is not required but recommended
145681ad6265SDimitry Andric // to arrange operands so that operand encoding match operand position
145781ad6265SDimitry Andric // in the table. This will make disassembly a bit more efficient.
145881ad6265SDimitry Andric // Unused slots in the table shall have an empty name.
145981ad6265SDimitry Andric //
146081ad6265SDimitry Andric //===----------------------------------------------------------------------===//
146181ad6265SDimitry Andric 
146281ad6265SDimitry Andric template <class T>
isValidOpr(int Idx,const CustomOperand<T> OpInfo[],int OpInfoSize,T Context)146381ad6265SDimitry Andric static bool isValidOpr(int Idx, const CustomOperand<T> OpInfo[], int OpInfoSize,
146481ad6265SDimitry Andric                        T Context) {
146581ad6265SDimitry Andric   return 0 <= Idx && Idx < OpInfoSize && !OpInfo[Idx].Name.empty() &&
146681ad6265SDimitry Andric          (!OpInfo[Idx].Cond || OpInfo[Idx].Cond(Context));
146781ad6265SDimitry Andric }
146881ad6265SDimitry Andric 
146981ad6265SDimitry Andric template <class T>
getOprIdx(std::function<bool (const CustomOperand<T> &)> Test,const CustomOperand<T> OpInfo[],int OpInfoSize,T Context)147081ad6265SDimitry Andric static int getOprIdx(std::function<bool(const CustomOperand<T> &)> Test,
147181ad6265SDimitry Andric                      const CustomOperand<T> OpInfo[], int OpInfoSize,
147281ad6265SDimitry Andric                      T Context) {
147381ad6265SDimitry Andric   int InvalidIdx = OPR_ID_UNKNOWN;
147481ad6265SDimitry Andric   for (int Idx = 0; Idx < OpInfoSize; ++Idx) {
147581ad6265SDimitry Andric     if (Test(OpInfo[Idx])) {
147681ad6265SDimitry Andric       if (!OpInfo[Idx].Cond || OpInfo[Idx].Cond(Context))
147781ad6265SDimitry Andric         return Idx;
147881ad6265SDimitry Andric       InvalidIdx = OPR_ID_UNSUPPORTED;
147981ad6265SDimitry Andric     }
148081ad6265SDimitry Andric   }
148181ad6265SDimitry Andric   return InvalidIdx;
148281ad6265SDimitry Andric }
148381ad6265SDimitry Andric 
148481ad6265SDimitry Andric template <class T>
getOprIdx(const StringRef Name,const CustomOperand<T> OpInfo[],int OpInfoSize,T Context)148581ad6265SDimitry Andric static int getOprIdx(const StringRef Name, const CustomOperand<T> OpInfo[],
148681ad6265SDimitry Andric                      int OpInfoSize, T Context) {
148781ad6265SDimitry Andric   auto Test = [=](const CustomOperand<T> &Op) { return Op.Name == Name; };
148881ad6265SDimitry Andric   return getOprIdx<T>(Test, OpInfo, OpInfoSize, Context);
148981ad6265SDimitry Andric }
149081ad6265SDimitry Andric 
149181ad6265SDimitry Andric template <class T>
getOprIdx(int Id,const CustomOperand<T> OpInfo[],int OpInfoSize,T Context,bool QuickCheck=true)149281ad6265SDimitry Andric static int getOprIdx(int Id, const CustomOperand<T> OpInfo[], int OpInfoSize,
149381ad6265SDimitry Andric                      T Context, bool QuickCheck = true) {
149481ad6265SDimitry Andric   auto Test = [=](const CustomOperand<T> &Op) {
149581ad6265SDimitry Andric     return Op.Encoding == Id && !Op.Name.empty();
149681ad6265SDimitry Andric   };
149781ad6265SDimitry Andric   // This is an optimization that should work in most cases.
149881ad6265SDimitry Andric   // As a side effect, it may cause selection of an alias
149981ad6265SDimitry Andric   // instead of a primary operand name in case of sparse tables.
150081ad6265SDimitry Andric   if (QuickCheck && isValidOpr<T>(Id, OpInfo, OpInfoSize, Context) &&
150181ad6265SDimitry Andric       OpInfo[Id].Encoding == Id) {
150281ad6265SDimitry Andric     return Id;
150381ad6265SDimitry Andric   }
150481ad6265SDimitry Andric   return getOprIdx<T>(Test, OpInfo, OpInfoSize, Context);
150581ad6265SDimitry Andric }
150681ad6265SDimitry Andric 
150781ad6265SDimitry Andric //===----------------------------------------------------------------------===//
150881ad6265SDimitry Andric // Custom Operand Values
150981ad6265SDimitry Andric //===----------------------------------------------------------------------===//
151081ad6265SDimitry Andric 
getDefaultCustomOperandEncoding(const CustomOperandVal * Opr,int Size,const MCSubtargetInfo & STI)151181ad6265SDimitry Andric static unsigned getDefaultCustomOperandEncoding(const CustomOperandVal *Opr,
151281ad6265SDimitry Andric                                                 int Size,
151381ad6265SDimitry Andric                                                 const MCSubtargetInfo &STI) {
151481ad6265SDimitry Andric   unsigned Enc = 0;
151581ad6265SDimitry Andric   for (int Idx = 0; Idx < Size; ++Idx) {
151681ad6265SDimitry Andric     const auto &Op = Opr[Idx];
151781ad6265SDimitry Andric     if (Op.isSupported(STI))
151881ad6265SDimitry Andric       Enc |= Op.encode(Op.Default);
151981ad6265SDimitry Andric   }
152081ad6265SDimitry Andric   return Enc;
152181ad6265SDimitry Andric }
152281ad6265SDimitry Andric 
isSymbolicCustomOperandEncoding(const CustomOperandVal * Opr,int Size,unsigned Code,bool & HasNonDefaultVal,const MCSubtargetInfo & STI)152381ad6265SDimitry Andric static bool isSymbolicCustomOperandEncoding(const CustomOperandVal *Opr,
152481ad6265SDimitry Andric                                             int Size, unsigned Code,
152581ad6265SDimitry Andric                                             bool &HasNonDefaultVal,
152681ad6265SDimitry Andric                                             const MCSubtargetInfo &STI) {
152781ad6265SDimitry Andric   unsigned UsedOprMask = 0;
152881ad6265SDimitry Andric   HasNonDefaultVal = false;
152981ad6265SDimitry Andric   for (int Idx = 0; Idx < Size; ++Idx) {
153081ad6265SDimitry Andric     const auto &Op = Opr[Idx];
153181ad6265SDimitry Andric     if (!Op.isSupported(STI))
153281ad6265SDimitry Andric       continue;
153381ad6265SDimitry Andric     UsedOprMask |= Op.getMask();
153481ad6265SDimitry Andric     unsigned Val = Op.decode(Code);
153581ad6265SDimitry Andric     if (!Op.isValid(Val))
153681ad6265SDimitry Andric       return false;
153781ad6265SDimitry Andric     HasNonDefaultVal |= (Val != Op.Default);
153881ad6265SDimitry Andric   }
153981ad6265SDimitry Andric   return (Code & ~UsedOprMask) == 0;
154081ad6265SDimitry Andric }
154181ad6265SDimitry Andric 
decodeCustomOperand(const CustomOperandVal * Opr,int Size,unsigned Code,int & Idx,StringRef & Name,unsigned & Val,bool & IsDefault,const MCSubtargetInfo & STI)154281ad6265SDimitry Andric static bool decodeCustomOperand(const CustomOperandVal *Opr, int Size,
154381ad6265SDimitry Andric                                 unsigned Code, int &Idx, StringRef &Name,
154481ad6265SDimitry Andric                                 unsigned &Val, bool &IsDefault,
154581ad6265SDimitry Andric                                 const MCSubtargetInfo &STI) {
154681ad6265SDimitry Andric   while (Idx < Size) {
154781ad6265SDimitry Andric     const auto &Op = Opr[Idx++];
154881ad6265SDimitry Andric     if (Op.isSupported(STI)) {
154981ad6265SDimitry Andric       Name = Op.Name;
155081ad6265SDimitry Andric       Val = Op.decode(Code);
155181ad6265SDimitry Andric       IsDefault = (Val == Op.Default);
155281ad6265SDimitry Andric       return true;
155381ad6265SDimitry Andric     }
155481ad6265SDimitry Andric   }
155581ad6265SDimitry Andric 
155681ad6265SDimitry Andric   return false;
155781ad6265SDimitry Andric }
155881ad6265SDimitry Andric 
encodeCustomOperandVal(const CustomOperandVal & Op,int64_t InputVal)155981ad6265SDimitry Andric static int encodeCustomOperandVal(const CustomOperandVal &Op,
156081ad6265SDimitry Andric                                   int64_t InputVal) {
156181ad6265SDimitry Andric   if (InputVal < 0 || InputVal > Op.Max)
156281ad6265SDimitry Andric     return OPR_VAL_INVALID;
156381ad6265SDimitry Andric   return Op.encode(InputVal);
156481ad6265SDimitry Andric }
156581ad6265SDimitry Andric 
encodeCustomOperand(const CustomOperandVal * Opr,int Size,const StringRef Name,int64_t InputVal,unsigned & UsedOprMask,const MCSubtargetInfo & STI)156681ad6265SDimitry Andric static int encodeCustomOperand(const CustomOperandVal *Opr, int Size,
156781ad6265SDimitry Andric                                const StringRef Name, int64_t InputVal,
156881ad6265SDimitry Andric                                unsigned &UsedOprMask,
156981ad6265SDimitry Andric                                const MCSubtargetInfo &STI) {
157081ad6265SDimitry Andric   int InvalidId = OPR_ID_UNKNOWN;
157181ad6265SDimitry Andric   for (int Idx = 0; Idx < Size; ++Idx) {
157281ad6265SDimitry Andric     const auto &Op = Opr[Idx];
157381ad6265SDimitry Andric     if (Op.Name == Name) {
157481ad6265SDimitry Andric       if (!Op.isSupported(STI)) {
157581ad6265SDimitry Andric         InvalidId = OPR_ID_UNSUPPORTED;
157681ad6265SDimitry Andric         continue;
157781ad6265SDimitry Andric       }
157881ad6265SDimitry Andric       auto OprMask = Op.getMask();
157981ad6265SDimitry Andric       if (OprMask & UsedOprMask)
158081ad6265SDimitry Andric         return OPR_ID_DUPLICATE;
158181ad6265SDimitry Andric       UsedOprMask |= OprMask;
158281ad6265SDimitry Andric       return encodeCustomOperandVal(Op, InputVal);
158381ad6265SDimitry Andric     }
158481ad6265SDimitry Andric   }
158581ad6265SDimitry Andric   return InvalidId;
158681ad6265SDimitry Andric }
158781ad6265SDimitry Andric 
158881ad6265SDimitry Andric //===----------------------------------------------------------------------===//
158981ad6265SDimitry Andric // DepCtr
159081ad6265SDimitry Andric //===----------------------------------------------------------------------===//
159181ad6265SDimitry Andric 
159281ad6265SDimitry Andric namespace DepCtr {
159381ad6265SDimitry Andric 
getDefaultDepCtrEncoding(const MCSubtargetInfo & STI)159481ad6265SDimitry Andric int getDefaultDepCtrEncoding(const MCSubtargetInfo &STI) {
159581ad6265SDimitry Andric   static int Default = -1;
159681ad6265SDimitry Andric   if (Default == -1)
159781ad6265SDimitry Andric     Default = getDefaultCustomOperandEncoding(DepCtrInfo, DEP_CTR_SIZE, STI);
159881ad6265SDimitry Andric   return Default;
159981ad6265SDimitry Andric }
160081ad6265SDimitry Andric 
isSymbolicDepCtrEncoding(unsigned Code,bool & HasNonDefaultVal,const MCSubtargetInfo & STI)160181ad6265SDimitry Andric bool isSymbolicDepCtrEncoding(unsigned Code, bool &HasNonDefaultVal,
160281ad6265SDimitry Andric                               const MCSubtargetInfo &STI) {
160381ad6265SDimitry Andric   return isSymbolicCustomOperandEncoding(DepCtrInfo, DEP_CTR_SIZE, Code,
160481ad6265SDimitry Andric                                          HasNonDefaultVal, STI);
160581ad6265SDimitry Andric }
160681ad6265SDimitry Andric 
decodeDepCtr(unsigned Code,int & Id,StringRef & Name,unsigned & Val,bool & IsDefault,const MCSubtargetInfo & STI)160781ad6265SDimitry Andric bool decodeDepCtr(unsigned Code, int &Id, StringRef &Name, unsigned &Val,
160881ad6265SDimitry Andric                   bool &IsDefault, const MCSubtargetInfo &STI) {
160981ad6265SDimitry Andric   return decodeCustomOperand(DepCtrInfo, DEP_CTR_SIZE, Code, Id, Name, Val,
161081ad6265SDimitry Andric                              IsDefault, STI);
161181ad6265SDimitry Andric }
161281ad6265SDimitry Andric 
encodeDepCtr(const StringRef Name,int64_t Val,unsigned & UsedOprMask,const MCSubtargetInfo & STI)161381ad6265SDimitry Andric int encodeDepCtr(const StringRef Name, int64_t Val, unsigned &UsedOprMask,
161481ad6265SDimitry Andric                  const MCSubtargetInfo &STI) {
161581ad6265SDimitry Andric   return encodeCustomOperand(DepCtrInfo, DEP_CTR_SIZE, Name, Val, UsedOprMask,
161681ad6265SDimitry Andric                              STI);
161781ad6265SDimitry Andric }
161881ad6265SDimitry Andric 
decodeFieldVmVsrc(unsigned Encoded)161906c3fb27SDimitry Andric unsigned decodeFieldVmVsrc(unsigned Encoded) {
162006c3fb27SDimitry Andric   return unpackBits(Encoded, getVmVsrcBitShift(), getVmVsrcBitWidth());
162106c3fb27SDimitry Andric }
162206c3fb27SDimitry Andric 
decodeFieldVaVdst(unsigned Encoded)162306c3fb27SDimitry Andric unsigned decodeFieldVaVdst(unsigned Encoded) {
162406c3fb27SDimitry Andric   return unpackBits(Encoded, getVaVdstBitShift(), getVaVdstBitWidth());
162506c3fb27SDimitry Andric }
162606c3fb27SDimitry Andric 
decodeFieldSaSdst(unsigned Encoded)162706c3fb27SDimitry Andric unsigned decodeFieldSaSdst(unsigned Encoded) {
162806c3fb27SDimitry Andric   return unpackBits(Encoded, getSaSdstBitShift(), getSaSdstBitWidth());
162906c3fb27SDimitry Andric }
163006c3fb27SDimitry Andric 
encodeFieldVmVsrc(unsigned Encoded,unsigned VmVsrc)163106c3fb27SDimitry Andric unsigned encodeFieldVmVsrc(unsigned Encoded, unsigned VmVsrc) {
163206c3fb27SDimitry Andric   return packBits(VmVsrc, Encoded, getVmVsrcBitShift(), getVmVsrcBitWidth());
163306c3fb27SDimitry Andric }
163406c3fb27SDimitry Andric 
encodeFieldVmVsrc(unsigned VmVsrc)163506c3fb27SDimitry Andric unsigned encodeFieldVmVsrc(unsigned VmVsrc) {
163606c3fb27SDimitry Andric   return encodeFieldVmVsrc(0xffff, VmVsrc);
163706c3fb27SDimitry Andric }
163806c3fb27SDimitry Andric 
encodeFieldVaVdst(unsigned Encoded,unsigned VaVdst)163906c3fb27SDimitry Andric unsigned encodeFieldVaVdst(unsigned Encoded, unsigned VaVdst) {
164006c3fb27SDimitry Andric   return packBits(VaVdst, Encoded, getVaVdstBitShift(), getVaVdstBitWidth());
164106c3fb27SDimitry Andric }
164206c3fb27SDimitry Andric 
encodeFieldVaVdst(unsigned VaVdst)164306c3fb27SDimitry Andric unsigned encodeFieldVaVdst(unsigned VaVdst) {
164406c3fb27SDimitry Andric   return encodeFieldVaVdst(0xffff, VaVdst);
164506c3fb27SDimitry Andric }
164606c3fb27SDimitry Andric 
encodeFieldSaSdst(unsigned Encoded,unsigned SaSdst)164706c3fb27SDimitry Andric unsigned encodeFieldSaSdst(unsigned Encoded, unsigned SaSdst) {
164806c3fb27SDimitry Andric   return packBits(SaSdst, Encoded, getSaSdstBitShift(), getSaSdstBitWidth());
164906c3fb27SDimitry Andric }
165006c3fb27SDimitry Andric 
encodeFieldSaSdst(unsigned SaSdst)165106c3fb27SDimitry Andric unsigned encodeFieldSaSdst(unsigned SaSdst) {
165206c3fb27SDimitry Andric   return encodeFieldSaSdst(0xffff, SaSdst);
165306c3fb27SDimitry Andric }
165406c3fb27SDimitry Andric 
165581ad6265SDimitry Andric } // namespace DepCtr
165681ad6265SDimitry Andric 
165781ad6265SDimitry Andric //===----------------------------------------------------------------------===//
16580b57cec5SDimitry Andric // hwreg
16590b57cec5SDimitry Andric //===----------------------------------------------------------------------===//
16600b57cec5SDimitry Andric 
16610b57cec5SDimitry Andric namespace Hwreg {
16620b57cec5SDimitry Andric 
getHwregId(const StringRef Name,const MCSubtargetInfo & STI)166381ad6265SDimitry Andric int64_t getHwregId(const StringRef Name, const MCSubtargetInfo &STI) {
166481ad6265SDimitry Andric   int Idx = getOprIdx<const MCSubtargetInfo &>(Name, Opr, OPR_SIZE, STI);
166581ad6265SDimitry Andric   return (Idx < 0) ? Idx : Opr[Idx].Encoding;
16660b57cec5SDimitry Andric }
16670b57cec5SDimitry Andric 
isValidHwreg(int64_t Id)16680b57cec5SDimitry Andric bool isValidHwreg(int64_t Id) {
16690b57cec5SDimitry Andric   return 0 <= Id && isUInt<ID_WIDTH_>(Id);
16700b57cec5SDimitry Andric }
16710b57cec5SDimitry Andric 
isValidHwregOffset(int64_t Offset)16720b57cec5SDimitry Andric bool isValidHwregOffset(int64_t Offset) {
16730b57cec5SDimitry Andric   return 0 <= Offset && isUInt<OFFSET_WIDTH_>(Offset);
16740b57cec5SDimitry Andric }
16750b57cec5SDimitry Andric 
isValidHwregWidth(int64_t Width)16760b57cec5SDimitry Andric bool isValidHwregWidth(int64_t Width) {
16770b57cec5SDimitry Andric   return 0 <= (Width - 1) && isUInt<WIDTH_M1_WIDTH_>(Width - 1);
16780b57cec5SDimitry Andric }
16790b57cec5SDimitry Andric 
encodeHwreg(uint64_t Id,uint64_t Offset,uint64_t Width)16800b57cec5SDimitry Andric uint64_t encodeHwreg(uint64_t Id, uint64_t Offset, uint64_t Width) {
16810b57cec5SDimitry Andric   return (Id << ID_SHIFT_) |
16820b57cec5SDimitry Andric          (Offset << OFFSET_SHIFT_) |
16830b57cec5SDimitry Andric          ((Width - 1) << WIDTH_M1_SHIFT_);
16840b57cec5SDimitry Andric }
16850b57cec5SDimitry Andric 
getHwreg(unsigned Id,const MCSubtargetInfo & STI)16860b57cec5SDimitry Andric StringRef getHwreg(unsigned Id, const MCSubtargetInfo &STI) {
168781ad6265SDimitry Andric   int Idx = getOprIdx<const MCSubtargetInfo &>(Id, Opr, OPR_SIZE, STI);
168881ad6265SDimitry Andric   return (Idx < 0) ? "" : Opr[Idx].Name;
16890b57cec5SDimitry Andric }
16900b57cec5SDimitry Andric 
decodeHwreg(unsigned Val,unsigned & Id,unsigned & Offset,unsigned & Width)16910b57cec5SDimitry Andric void decodeHwreg(unsigned Val, unsigned &Id, unsigned &Offset, unsigned &Width) {
16920b57cec5SDimitry Andric   Id = (Val & ID_MASK_) >> ID_SHIFT_;
16930b57cec5SDimitry Andric   Offset = (Val & OFFSET_MASK_) >> OFFSET_SHIFT_;
16940b57cec5SDimitry Andric   Width = ((Val & WIDTH_M1_MASK_) >> WIDTH_M1_SHIFT_) + 1;
16950b57cec5SDimitry Andric }
16960b57cec5SDimitry Andric 
16970b57cec5SDimitry Andric } // namespace Hwreg
16980b57cec5SDimitry Andric 
16990b57cec5SDimitry Andric //===----------------------------------------------------------------------===//
1700e8d8bef9SDimitry Andric // exp tgt
1701e8d8bef9SDimitry Andric //===----------------------------------------------------------------------===//
1702e8d8bef9SDimitry Andric 
1703e8d8bef9SDimitry Andric namespace Exp {
1704e8d8bef9SDimitry Andric 
1705e8d8bef9SDimitry Andric struct ExpTgt {
1706e8d8bef9SDimitry Andric   StringLiteral Name;
1707e8d8bef9SDimitry Andric   unsigned Tgt;
1708e8d8bef9SDimitry Andric   unsigned MaxIndex;
1709e8d8bef9SDimitry Andric };
1710e8d8bef9SDimitry Andric 
1711e8d8bef9SDimitry Andric static constexpr ExpTgt ExpTgtInfo[] = {
1712e8d8bef9SDimitry Andric   {{"null"},           ET_NULL,            ET_NULL_MAX_IDX},
1713e8d8bef9SDimitry Andric   {{"mrtz"},           ET_MRTZ,            ET_MRTZ_MAX_IDX},
1714e8d8bef9SDimitry Andric   {{"prim"},           ET_PRIM,            ET_PRIM_MAX_IDX},
1715e8d8bef9SDimitry Andric   {{"mrt"},            ET_MRT0,            ET_MRT_MAX_IDX},
1716e8d8bef9SDimitry Andric   {{"pos"},            ET_POS0,            ET_POS_MAX_IDX},
171781ad6265SDimitry Andric   {{"dual_src_blend"}, ET_DUAL_SRC_BLEND0, ET_DUAL_SRC_BLEND_MAX_IDX},
1718e8d8bef9SDimitry Andric   {{"param"},          ET_PARAM0,          ET_PARAM_MAX_IDX},
1719e8d8bef9SDimitry Andric };
1720e8d8bef9SDimitry Andric 
getTgtName(unsigned Id,StringRef & Name,int & Index)1721e8d8bef9SDimitry Andric bool getTgtName(unsigned Id, StringRef &Name, int &Index) {
1722e8d8bef9SDimitry Andric   for (const ExpTgt &Val : ExpTgtInfo) {
1723e8d8bef9SDimitry Andric     if (Val.Tgt <= Id && Id <= Val.Tgt + Val.MaxIndex) {
1724e8d8bef9SDimitry Andric       Index = (Val.MaxIndex == 0) ? -1 : (Id - Val.Tgt);
1725e8d8bef9SDimitry Andric       Name = Val.Name;
1726e8d8bef9SDimitry Andric       return true;
1727e8d8bef9SDimitry Andric     }
1728e8d8bef9SDimitry Andric   }
1729e8d8bef9SDimitry Andric   return false;
1730e8d8bef9SDimitry Andric }
1731e8d8bef9SDimitry Andric 
getTgtId(const StringRef Name)1732e8d8bef9SDimitry Andric unsigned getTgtId(const StringRef Name) {
1733e8d8bef9SDimitry Andric 
1734e8d8bef9SDimitry Andric   for (const ExpTgt &Val : ExpTgtInfo) {
1735e8d8bef9SDimitry Andric     if (Val.MaxIndex == 0 && Name == Val.Name)
1736e8d8bef9SDimitry Andric       return Val.Tgt;
1737e8d8bef9SDimitry Andric 
17385f757f3fSDimitry Andric     if (Val.MaxIndex > 0 && Name.starts_with(Val.Name)) {
1739e8d8bef9SDimitry Andric       StringRef Suffix = Name.drop_front(Val.Name.size());
1740e8d8bef9SDimitry Andric 
1741e8d8bef9SDimitry Andric       unsigned Id;
1742e8d8bef9SDimitry Andric       if (Suffix.getAsInteger(10, Id) || Id > Val.MaxIndex)
1743e8d8bef9SDimitry Andric         return ET_INVALID;
1744e8d8bef9SDimitry Andric 
1745e8d8bef9SDimitry Andric       // Disable leading zeroes
1746e8d8bef9SDimitry Andric       if (Suffix.size() > 1 && Suffix[0] == '0')
1747e8d8bef9SDimitry Andric         return ET_INVALID;
1748e8d8bef9SDimitry Andric 
1749e8d8bef9SDimitry Andric       return Val.Tgt + Id;
1750e8d8bef9SDimitry Andric     }
1751e8d8bef9SDimitry Andric   }
1752e8d8bef9SDimitry Andric   return ET_INVALID;
1753e8d8bef9SDimitry Andric }
1754e8d8bef9SDimitry Andric 
isSupportedTgtId(unsigned Id,const MCSubtargetInfo & STI)1755e8d8bef9SDimitry Andric bool isSupportedTgtId(unsigned Id, const MCSubtargetInfo &STI) {
175681ad6265SDimitry Andric   switch (Id) {
175781ad6265SDimitry Andric   case ET_NULL:
175881ad6265SDimitry Andric     return !isGFX11Plus(STI);
175981ad6265SDimitry Andric   case ET_POS4:
176081ad6265SDimitry Andric   case ET_PRIM:
176181ad6265SDimitry Andric     return isGFX10Plus(STI);
176281ad6265SDimitry Andric   case ET_DUAL_SRC_BLEND0:
176381ad6265SDimitry Andric   case ET_DUAL_SRC_BLEND1:
176481ad6265SDimitry Andric     return isGFX11Plus(STI);
176581ad6265SDimitry Andric   default:
176681ad6265SDimitry Andric     if (Id >= ET_PARAM0 && Id <= ET_PARAM31)
176781ad6265SDimitry Andric       return !isGFX11Plus(STI);
176881ad6265SDimitry Andric     return true;
176981ad6265SDimitry Andric   }
1770e8d8bef9SDimitry Andric }
1771e8d8bef9SDimitry Andric 
1772e8d8bef9SDimitry Andric } // namespace Exp
1773e8d8bef9SDimitry Andric 
1774e8d8bef9SDimitry Andric //===----------------------------------------------------------------------===//
1775e8d8bef9SDimitry Andric // MTBUF Format
1776e8d8bef9SDimitry Andric //===----------------------------------------------------------------------===//
1777e8d8bef9SDimitry Andric 
1778e8d8bef9SDimitry Andric namespace MTBUFFormat {
1779e8d8bef9SDimitry Andric 
getDfmt(const StringRef Name)1780e8d8bef9SDimitry Andric int64_t getDfmt(const StringRef Name) {
1781e8d8bef9SDimitry Andric   for (int Id = DFMT_MIN; Id <= DFMT_MAX; ++Id) {
1782e8d8bef9SDimitry Andric     if (Name == DfmtSymbolic[Id])
1783e8d8bef9SDimitry Andric       return Id;
1784e8d8bef9SDimitry Andric   }
1785e8d8bef9SDimitry Andric   return DFMT_UNDEF;
1786e8d8bef9SDimitry Andric }
1787e8d8bef9SDimitry Andric 
getDfmtName(unsigned Id)1788e8d8bef9SDimitry Andric StringRef getDfmtName(unsigned Id) {
1789e8d8bef9SDimitry Andric   assert(Id <= DFMT_MAX);
1790e8d8bef9SDimitry Andric   return DfmtSymbolic[Id];
1791e8d8bef9SDimitry Andric }
1792e8d8bef9SDimitry Andric 
getNfmtLookupTable(const MCSubtargetInfo & STI)1793e8d8bef9SDimitry Andric static StringLiteral const *getNfmtLookupTable(const MCSubtargetInfo &STI) {
1794e8d8bef9SDimitry Andric   if (isSI(STI) || isCI(STI))
1795e8d8bef9SDimitry Andric     return NfmtSymbolicSICI;
1796e8d8bef9SDimitry Andric   if (isVI(STI) || isGFX9(STI))
1797e8d8bef9SDimitry Andric     return NfmtSymbolicVI;
1798e8d8bef9SDimitry Andric   return NfmtSymbolicGFX10;
1799e8d8bef9SDimitry Andric }
1800e8d8bef9SDimitry Andric 
getNfmt(const StringRef Name,const MCSubtargetInfo & STI)1801e8d8bef9SDimitry Andric int64_t getNfmt(const StringRef Name, const MCSubtargetInfo &STI) {
1802e8d8bef9SDimitry Andric   auto lookupTable = getNfmtLookupTable(STI);
1803e8d8bef9SDimitry Andric   for (int Id = NFMT_MIN; Id <= NFMT_MAX; ++Id) {
1804e8d8bef9SDimitry Andric     if (Name == lookupTable[Id])
1805e8d8bef9SDimitry Andric       return Id;
1806e8d8bef9SDimitry Andric   }
1807e8d8bef9SDimitry Andric   return NFMT_UNDEF;
1808e8d8bef9SDimitry Andric }
1809e8d8bef9SDimitry Andric 
getNfmtName(unsigned Id,const MCSubtargetInfo & STI)1810e8d8bef9SDimitry Andric StringRef getNfmtName(unsigned Id, const MCSubtargetInfo &STI) {
1811e8d8bef9SDimitry Andric   assert(Id <= NFMT_MAX);
1812e8d8bef9SDimitry Andric   return getNfmtLookupTable(STI)[Id];
1813e8d8bef9SDimitry Andric }
1814e8d8bef9SDimitry Andric 
isValidDfmtNfmt(unsigned Id,const MCSubtargetInfo & STI)1815e8d8bef9SDimitry Andric bool isValidDfmtNfmt(unsigned Id, const MCSubtargetInfo &STI) {
1816e8d8bef9SDimitry Andric   unsigned Dfmt;
1817e8d8bef9SDimitry Andric   unsigned Nfmt;
1818e8d8bef9SDimitry Andric   decodeDfmtNfmt(Id, Dfmt, Nfmt);
1819e8d8bef9SDimitry Andric   return isValidNfmt(Nfmt, STI);
1820e8d8bef9SDimitry Andric }
1821e8d8bef9SDimitry Andric 
isValidNfmt(unsigned Id,const MCSubtargetInfo & STI)1822e8d8bef9SDimitry Andric bool isValidNfmt(unsigned Id, const MCSubtargetInfo &STI) {
1823e8d8bef9SDimitry Andric   return !getNfmtName(Id, STI).empty();
1824e8d8bef9SDimitry Andric }
1825e8d8bef9SDimitry Andric 
encodeDfmtNfmt(unsigned Dfmt,unsigned Nfmt)1826e8d8bef9SDimitry Andric int64_t encodeDfmtNfmt(unsigned Dfmt, unsigned Nfmt) {
1827e8d8bef9SDimitry Andric   return (Dfmt << DFMT_SHIFT) | (Nfmt << NFMT_SHIFT);
1828e8d8bef9SDimitry Andric }
1829e8d8bef9SDimitry Andric 
decodeDfmtNfmt(unsigned Format,unsigned & Dfmt,unsigned & Nfmt)1830e8d8bef9SDimitry Andric void decodeDfmtNfmt(unsigned Format, unsigned &Dfmt, unsigned &Nfmt) {
1831e8d8bef9SDimitry Andric   Dfmt = (Format >> DFMT_SHIFT) & DFMT_MASK;
1832e8d8bef9SDimitry Andric   Nfmt = (Format >> NFMT_SHIFT) & NFMT_MASK;
1833e8d8bef9SDimitry Andric }
1834e8d8bef9SDimitry Andric 
getUnifiedFormat(const StringRef Name,const MCSubtargetInfo & STI)183581ad6265SDimitry Andric int64_t getUnifiedFormat(const StringRef Name, const MCSubtargetInfo &STI) {
183681ad6265SDimitry Andric   if (isGFX11Plus(STI)) {
183781ad6265SDimitry Andric     for (int Id = UfmtGFX11::UFMT_FIRST; Id <= UfmtGFX11::UFMT_LAST; ++Id) {
183881ad6265SDimitry Andric       if (Name == UfmtSymbolicGFX11[Id])
1839e8d8bef9SDimitry Andric         return Id;
1840e8d8bef9SDimitry Andric     }
184181ad6265SDimitry Andric   } else {
184281ad6265SDimitry Andric     for (int Id = UfmtGFX10::UFMT_FIRST; Id <= UfmtGFX10::UFMT_LAST; ++Id) {
184381ad6265SDimitry Andric       if (Name == UfmtSymbolicGFX10[Id])
184481ad6265SDimitry Andric         return Id;
184581ad6265SDimitry Andric     }
184681ad6265SDimitry Andric   }
1847e8d8bef9SDimitry Andric   return UFMT_UNDEF;
1848e8d8bef9SDimitry Andric }
1849e8d8bef9SDimitry Andric 
getUnifiedFormatName(unsigned Id,const MCSubtargetInfo & STI)185081ad6265SDimitry Andric StringRef getUnifiedFormatName(unsigned Id, const MCSubtargetInfo &STI) {
185181ad6265SDimitry Andric   if(isValidUnifiedFormat(Id, STI))
185281ad6265SDimitry Andric     return isGFX10(STI) ? UfmtSymbolicGFX10[Id] : UfmtSymbolicGFX11[Id];
185381ad6265SDimitry Andric   return "";
1854e8d8bef9SDimitry Andric }
1855e8d8bef9SDimitry Andric 
isValidUnifiedFormat(unsigned Id,const MCSubtargetInfo & STI)185681ad6265SDimitry Andric bool isValidUnifiedFormat(unsigned Id, const MCSubtargetInfo &STI) {
185781ad6265SDimitry Andric   return isGFX10(STI) ? Id <= UfmtGFX10::UFMT_LAST : Id <= UfmtGFX11::UFMT_LAST;
1858e8d8bef9SDimitry Andric }
1859e8d8bef9SDimitry Andric 
convertDfmtNfmt2Ufmt(unsigned Dfmt,unsigned Nfmt,const MCSubtargetInfo & STI)186081ad6265SDimitry Andric int64_t convertDfmtNfmt2Ufmt(unsigned Dfmt, unsigned Nfmt,
186181ad6265SDimitry Andric                              const MCSubtargetInfo &STI) {
1862e8d8bef9SDimitry Andric   int64_t Fmt = encodeDfmtNfmt(Dfmt, Nfmt);
186381ad6265SDimitry Andric   if (isGFX11Plus(STI)) {
186481ad6265SDimitry Andric     for (int Id = UfmtGFX11::UFMT_FIRST; Id <= UfmtGFX11::UFMT_LAST; ++Id) {
186581ad6265SDimitry Andric       if (Fmt == DfmtNfmt2UFmtGFX11[Id])
1866e8d8bef9SDimitry Andric         return Id;
1867e8d8bef9SDimitry Andric     }
186881ad6265SDimitry Andric   } else {
186981ad6265SDimitry Andric     for (int Id = UfmtGFX10::UFMT_FIRST; Id <= UfmtGFX10::UFMT_LAST; ++Id) {
187081ad6265SDimitry Andric       if (Fmt == DfmtNfmt2UFmtGFX10[Id])
187181ad6265SDimitry Andric         return Id;
187281ad6265SDimitry Andric     }
187381ad6265SDimitry Andric   }
1874e8d8bef9SDimitry Andric   return UFMT_UNDEF;
1875e8d8bef9SDimitry Andric }
1876e8d8bef9SDimitry Andric 
isValidFormatEncoding(unsigned Val,const MCSubtargetInfo & STI)1877e8d8bef9SDimitry Andric bool isValidFormatEncoding(unsigned Val, const MCSubtargetInfo &STI) {
1878e8d8bef9SDimitry Andric   return isGFX10Plus(STI) ? (Val <= UFMT_MAX) : (Val <= DFMT_NFMT_MAX);
1879e8d8bef9SDimitry Andric }
1880e8d8bef9SDimitry Andric 
getDefaultFormatEncoding(const MCSubtargetInfo & STI)1881e8d8bef9SDimitry Andric unsigned getDefaultFormatEncoding(const MCSubtargetInfo &STI) {
1882e8d8bef9SDimitry Andric   if (isGFX10Plus(STI))
1883e8d8bef9SDimitry Andric     return UFMT_DEFAULT;
1884e8d8bef9SDimitry Andric   return DFMT_NFMT_DEFAULT;
1885e8d8bef9SDimitry Andric }
1886e8d8bef9SDimitry Andric 
1887e8d8bef9SDimitry Andric } // namespace MTBUFFormat
1888e8d8bef9SDimitry Andric 
1889e8d8bef9SDimitry Andric //===----------------------------------------------------------------------===//
18900b57cec5SDimitry Andric // SendMsg
18910b57cec5SDimitry Andric //===----------------------------------------------------------------------===//
18920b57cec5SDimitry Andric 
18930b57cec5SDimitry Andric namespace SendMsg {
18940b57cec5SDimitry Andric 
getMsgIdMask(const MCSubtargetInfo & STI)189581ad6265SDimitry Andric static uint64_t getMsgIdMask(const MCSubtargetInfo &STI) {
189681ad6265SDimitry Andric   return isGFX11Plus(STI) ? ID_MASK_GFX11Plus_ : ID_MASK_PreGFX11_;
18970b57cec5SDimitry Andric }
18980b57cec5SDimitry Andric 
getMsgId(const StringRef Name,const MCSubtargetInfo & STI)189981ad6265SDimitry Andric int64_t getMsgId(const StringRef Name, const MCSubtargetInfo &STI) {
190081ad6265SDimitry Andric   int Idx = getOprIdx<const MCSubtargetInfo &>(Name, Msg, MSG_SIZE, STI);
190181ad6265SDimitry Andric   return (Idx < 0) ? Idx : Msg[Idx].Encoding;
19020b57cec5SDimitry Andric }
19030b57cec5SDimitry Andric 
isValidMsgId(int64_t MsgId,const MCSubtargetInfo & STI)190481ad6265SDimitry Andric bool isValidMsgId(int64_t MsgId, const MCSubtargetInfo &STI) {
190581ad6265SDimitry Andric   return (MsgId & ~(getMsgIdMask(STI))) == 0;
190681ad6265SDimitry Andric }
190781ad6265SDimitry Andric 
getMsgName(int64_t MsgId,const MCSubtargetInfo & STI)190881ad6265SDimitry Andric StringRef getMsgName(int64_t MsgId, const MCSubtargetInfo &STI) {
190981ad6265SDimitry Andric   int Idx = getOprIdx<const MCSubtargetInfo &>(MsgId, Msg, MSG_SIZE, STI);
191081ad6265SDimitry Andric   return (Idx < 0) ? "" : Msg[Idx].Name;
19110b57cec5SDimitry Andric }
19120b57cec5SDimitry Andric 
getMsgOpId(int64_t MsgId,const StringRef Name)19130b57cec5SDimitry Andric int64_t getMsgOpId(int64_t MsgId, const StringRef Name) {
19140b57cec5SDimitry Andric   const char* const *S = (MsgId == ID_SYSMSG) ? OpSysSymbolic : OpGsSymbolic;
19150b57cec5SDimitry Andric   const int F = (MsgId == ID_SYSMSG) ? OP_SYS_FIRST_ : OP_GS_FIRST_;
19160b57cec5SDimitry Andric   const int L = (MsgId == ID_SYSMSG) ? OP_SYS_LAST_ : OP_GS_LAST_;
19170b57cec5SDimitry Andric   for (int i = F; i < L; ++i) {
19180b57cec5SDimitry Andric     if (Name == S[i]) {
19190b57cec5SDimitry Andric       return i;
19200b57cec5SDimitry Andric     }
19210b57cec5SDimitry Andric   }
19220b57cec5SDimitry Andric   return OP_UNKNOWN_;
19230b57cec5SDimitry Andric }
19240b57cec5SDimitry Andric 
isValidMsgOp(int64_t MsgId,int64_t OpId,const MCSubtargetInfo & STI,bool Strict)1925fe6060f1SDimitry Andric bool isValidMsgOp(int64_t MsgId, int64_t OpId, const MCSubtargetInfo &STI,
1926fe6060f1SDimitry Andric                   bool Strict) {
192781ad6265SDimitry Andric   assert(isValidMsgId(MsgId, STI));
19280b57cec5SDimitry Andric 
19290b57cec5SDimitry Andric   if (!Strict)
19300b57cec5SDimitry Andric     return 0 <= OpId && isUInt<OP_WIDTH_>(OpId);
19310b57cec5SDimitry Andric 
193281ad6265SDimitry Andric   if (MsgId == ID_SYSMSG)
19330b57cec5SDimitry Andric     return OP_SYS_FIRST_ <= OpId && OpId < OP_SYS_LAST_;
193481ad6265SDimitry Andric   if (!isGFX11Plus(STI)) {
193581ad6265SDimitry Andric     switch (MsgId) {
193681ad6265SDimitry Andric     case ID_GS_PreGFX11:
193781ad6265SDimitry Andric       return (OP_GS_FIRST_ <= OpId && OpId < OP_GS_LAST_) && OpId != OP_GS_NOP;
193881ad6265SDimitry Andric     case ID_GS_DONE_PreGFX11:
193981ad6265SDimitry Andric       return OP_GS_FIRST_ <= OpId && OpId < OP_GS_LAST_;
194081ad6265SDimitry Andric     }
194181ad6265SDimitry Andric   }
19420b57cec5SDimitry Andric   return OpId == OP_NONE_;
19430b57cec5SDimitry Andric }
19440b57cec5SDimitry Andric 
getMsgOpName(int64_t MsgId,int64_t OpId,const MCSubtargetInfo & STI)194581ad6265SDimitry Andric StringRef getMsgOpName(int64_t MsgId, int64_t OpId,
194681ad6265SDimitry Andric                        const MCSubtargetInfo &STI) {
194781ad6265SDimitry Andric   assert(msgRequiresOp(MsgId, STI));
19480b57cec5SDimitry Andric   return (MsgId == ID_SYSMSG)? OpSysSymbolic[OpId] : OpGsSymbolic[OpId];
19490b57cec5SDimitry Andric }
19500b57cec5SDimitry Andric 
isValidMsgStream(int64_t MsgId,int64_t OpId,int64_t StreamId,const MCSubtargetInfo & STI,bool Strict)1951fe6060f1SDimitry Andric bool isValidMsgStream(int64_t MsgId, int64_t OpId, int64_t StreamId,
1952fe6060f1SDimitry Andric                       const MCSubtargetInfo &STI, bool Strict) {
1953fe6060f1SDimitry Andric   assert(isValidMsgOp(MsgId, OpId, STI, Strict));
19540b57cec5SDimitry Andric 
19550b57cec5SDimitry Andric   if (!Strict)
19560b57cec5SDimitry Andric     return 0 <= StreamId && isUInt<STREAM_ID_WIDTH_>(StreamId);
19570b57cec5SDimitry Andric 
195881ad6265SDimitry Andric   if (!isGFX11Plus(STI)) {
195981ad6265SDimitry Andric     switch (MsgId) {
196081ad6265SDimitry Andric     case ID_GS_PreGFX11:
19610b57cec5SDimitry Andric       return STREAM_ID_FIRST_ <= StreamId && StreamId < STREAM_ID_LAST_;
196281ad6265SDimitry Andric     case ID_GS_DONE_PreGFX11:
19630b57cec5SDimitry Andric       return (OpId == OP_GS_NOP) ?
19640b57cec5SDimitry Andric           (StreamId == STREAM_ID_NONE_) :
19650b57cec5SDimitry Andric           (STREAM_ID_FIRST_ <= StreamId && StreamId < STREAM_ID_LAST_);
196681ad6265SDimitry Andric     }
196781ad6265SDimitry Andric   }
19680b57cec5SDimitry Andric   return StreamId == STREAM_ID_NONE_;
19690b57cec5SDimitry Andric }
197081ad6265SDimitry Andric 
msgRequiresOp(int64_t MsgId,const MCSubtargetInfo & STI)197181ad6265SDimitry Andric bool msgRequiresOp(int64_t MsgId, const MCSubtargetInfo &STI) {
197281ad6265SDimitry Andric   return MsgId == ID_SYSMSG ||
197381ad6265SDimitry Andric       (!isGFX11Plus(STI) &&
197481ad6265SDimitry Andric        (MsgId == ID_GS_PreGFX11 || MsgId == ID_GS_DONE_PreGFX11));
19750b57cec5SDimitry Andric }
19760b57cec5SDimitry Andric 
msgSupportsStream(int64_t MsgId,int64_t OpId,const MCSubtargetInfo & STI)197781ad6265SDimitry Andric bool msgSupportsStream(int64_t MsgId, int64_t OpId,
197881ad6265SDimitry Andric                        const MCSubtargetInfo &STI) {
197981ad6265SDimitry Andric   return !isGFX11Plus(STI) &&
198081ad6265SDimitry Andric       (MsgId == ID_GS_PreGFX11 || MsgId == ID_GS_DONE_PreGFX11) &&
198181ad6265SDimitry Andric       OpId != OP_GS_NOP;
19820b57cec5SDimitry Andric }
19830b57cec5SDimitry Andric 
decodeMsg(unsigned Val,uint16_t & MsgId,uint16_t & OpId,uint16_t & StreamId,const MCSubtargetInfo & STI)198481ad6265SDimitry Andric void decodeMsg(unsigned Val, uint16_t &MsgId, uint16_t &OpId,
198581ad6265SDimitry Andric                uint16_t &StreamId, const MCSubtargetInfo &STI) {
198681ad6265SDimitry Andric   MsgId = Val & getMsgIdMask(STI);
198781ad6265SDimitry Andric   if (isGFX11Plus(STI)) {
198881ad6265SDimitry Andric     OpId = 0;
198981ad6265SDimitry Andric     StreamId = 0;
199081ad6265SDimitry Andric   } else {
19910b57cec5SDimitry Andric     OpId = (Val & OP_MASK_) >> OP_SHIFT_;
19920b57cec5SDimitry Andric     StreamId = (Val & STREAM_ID_MASK_) >> STREAM_ID_SHIFT_;
19930b57cec5SDimitry Andric   }
199481ad6265SDimitry Andric }
19950b57cec5SDimitry Andric 
encodeMsg(uint64_t MsgId,uint64_t OpId,uint64_t StreamId)19960b57cec5SDimitry Andric uint64_t encodeMsg(uint64_t MsgId,
19970b57cec5SDimitry Andric                    uint64_t OpId,
19980b57cec5SDimitry Andric                    uint64_t StreamId) {
199981ad6265SDimitry Andric   return MsgId | (OpId << OP_SHIFT_) | (StreamId << STREAM_ID_SHIFT_);
20000b57cec5SDimitry Andric }
20010b57cec5SDimitry Andric 
20020b57cec5SDimitry Andric } // namespace SendMsg
20030b57cec5SDimitry Andric 
20040b57cec5SDimitry Andric //===----------------------------------------------------------------------===//
20050b57cec5SDimitry Andric //
20060b57cec5SDimitry Andric //===----------------------------------------------------------------------===//
20070b57cec5SDimitry Andric 
getInitialPSInputAddr(const Function & F)20080b57cec5SDimitry Andric unsigned getInitialPSInputAddr(const Function &F) {
2009bdd1243dSDimitry Andric   return F.getFnAttributeAsParsedInteger("InitialPSInputAddr", 0);
20100b57cec5SDimitry Andric }
20110b57cec5SDimitry Andric 
getHasColorExport(const Function & F)2012fe6060f1SDimitry Andric bool getHasColorExport(const Function &F) {
2013fe6060f1SDimitry Andric   // As a safe default always respond as if PS has color exports.
2014bdd1243dSDimitry Andric   return F.getFnAttributeAsParsedInteger(
2015bdd1243dSDimitry Andric              "amdgpu-color-export",
2016fe6060f1SDimitry Andric              F.getCallingConv() == CallingConv::AMDGPU_PS ? 1 : 0) != 0;
2017fe6060f1SDimitry Andric }
2018fe6060f1SDimitry Andric 
getHasDepthExport(const Function & F)2019fe6060f1SDimitry Andric bool getHasDepthExport(const Function &F) {
2020bdd1243dSDimitry Andric   return F.getFnAttributeAsParsedInteger("amdgpu-depth-export", 0) != 0;
2021fe6060f1SDimitry Andric }
2022fe6060f1SDimitry Andric 
isShader(CallingConv::ID cc)20230b57cec5SDimitry Andric bool isShader(CallingConv::ID cc) {
20240b57cec5SDimitry Andric   switch(cc) {
20250b57cec5SDimitry Andric     case CallingConv::AMDGPU_VS:
20260b57cec5SDimitry Andric     case CallingConv::AMDGPU_LS:
20270b57cec5SDimitry Andric     case CallingConv::AMDGPU_HS:
20280b57cec5SDimitry Andric     case CallingConv::AMDGPU_ES:
20290b57cec5SDimitry Andric     case CallingConv::AMDGPU_GS:
20300b57cec5SDimitry Andric     case CallingConv::AMDGPU_PS:
20315f757f3fSDimitry Andric     case CallingConv::AMDGPU_CS_Chain:
20325f757f3fSDimitry Andric     case CallingConv::AMDGPU_CS_ChainPreserve:
20330b57cec5SDimitry Andric     case CallingConv::AMDGPU_CS:
20340b57cec5SDimitry Andric       return true;
20350b57cec5SDimitry Andric     default:
20360b57cec5SDimitry Andric       return false;
20370b57cec5SDimitry Andric   }
20380b57cec5SDimitry Andric }
20390b57cec5SDimitry Andric 
isGraphics(CallingConv::ID cc)2040e8d8bef9SDimitry Andric bool isGraphics(CallingConv::ID cc) {
2041e8d8bef9SDimitry Andric   return isShader(cc) || cc == CallingConv::AMDGPU_Gfx;
2042e8d8bef9SDimitry Andric }
2043e8d8bef9SDimitry Andric 
isCompute(CallingConv::ID cc)20440b57cec5SDimitry Andric bool isCompute(CallingConv::ID cc) {
2045e8d8bef9SDimitry Andric   return !isGraphics(cc) || cc == CallingConv::AMDGPU_CS;
20460b57cec5SDimitry Andric }
20470b57cec5SDimitry Andric 
isEntryFunctionCC(CallingConv::ID CC)20480b57cec5SDimitry Andric bool isEntryFunctionCC(CallingConv::ID CC) {
20490b57cec5SDimitry Andric   switch (CC) {
20500b57cec5SDimitry Andric   case CallingConv::AMDGPU_KERNEL:
20510b57cec5SDimitry Andric   case CallingConv::SPIR_KERNEL:
20520b57cec5SDimitry Andric   case CallingConv::AMDGPU_VS:
20530b57cec5SDimitry Andric   case CallingConv::AMDGPU_GS:
20540b57cec5SDimitry Andric   case CallingConv::AMDGPU_PS:
20550b57cec5SDimitry Andric   case CallingConv::AMDGPU_CS:
20560b57cec5SDimitry Andric   case CallingConv::AMDGPU_ES:
20570b57cec5SDimitry Andric   case CallingConv::AMDGPU_HS:
20580b57cec5SDimitry Andric   case CallingConv::AMDGPU_LS:
20590b57cec5SDimitry Andric     return true;
20600b57cec5SDimitry Andric   default:
20610b57cec5SDimitry Andric     return false;
20620b57cec5SDimitry Andric   }
20630b57cec5SDimitry Andric }
20640b57cec5SDimitry Andric 
isModuleEntryFunctionCC(CallingConv::ID CC)2065e8d8bef9SDimitry Andric bool isModuleEntryFunctionCC(CallingConv::ID CC) {
2066e8d8bef9SDimitry Andric   switch (CC) {
2067e8d8bef9SDimitry Andric   case CallingConv::AMDGPU_Gfx:
2068e8d8bef9SDimitry Andric     return true;
2069e8d8bef9SDimitry Andric   default:
20705f757f3fSDimitry Andric     return isEntryFunctionCC(CC) || isChainCC(CC);
20715f757f3fSDimitry Andric   }
20725f757f3fSDimitry Andric }
20735f757f3fSDimitry Andric 
isChainCC(CallingConv::ID CC)20745f757f3fSDimitry Andric bool isChainCC(CallingConv::ID CC) {
20755f757f3fSDimitry Andric   switch (CC) {
20765f757f3fSDimitry Andric   case CallingConv::AMDGPU_CS_Chain:
20775f757f3fSDimitry Andric   case CallingConv::AMDGPU_CS_ChainPreserve:
20785f757f3fSDimitry Andric     return true;
20795f757f3fSDimitry Andric   default:
20805f757f3fSDimitry Andric     return false;
2081e8d8bef9SDimitry Andric   }
2082e8d8bef9SDimitry Andric }
2083e8d8bef9SDimitry Andric 
isKernelCC(const Function * Func)208481ad6265SDimitry Andric bool isKernelCC(const Function *Func) {
208581ad6265SDimitry Andric   return AMDGPU::isModuleEntryFunctionCC(Func->getCallingConv());
208681ad6265SDimitry Andric }
208781ad6265SDimitry Andric 
hasXNACK(const MCSubtargetInfo & STI)20880b57cec5SDimitry Andric bool hasXNACK(const MCSubtargetInfo &STI) {
208906c3fb27SDimitry Andric   return STI.hasFeature(AMDGPU::FeatureXNACK);
20900b57cec5SDimitry Andric }
20910b57cec5SDimitry Andric 
hasSRAMECC(const MCSubtargetInfo & STI)20920b57cec5SDimitry Andric bool hasSRAMECC(const MCSubtargetInfo &STI) {
209306c3fb27SDimitry Andric   return STI.hasFeature(AMDGPU::FeatureSRAMECC);
20940b57cec5SDimitry Andric }
20950b57cec5SDimitry Andric 
hasMIMG_R128(const MCSubtargetInfo & STI)20960b57cec5SDimitry Andric bool hasMIMG_R128(const MCSubtargetInfo &STI) {
209706c3fb27SDimitry Andric   return STI.hasFeature(AMDGPU::FeatureMIMG_R128) && !STI.hasFeature(AMDGPU::FeatureR128A16);
20985ffd83dbSDimitry Andric }
20995ffd83dbSDimitry Andric 
hasA16(const MCSubtargetInfo & STI)2100bdd1243dSDimitry Andric bool hasA16(const MCSubtargetInfo &STI) {
210106c3fb27SDimitry Andric   return STI.hasFeature(AMDGPU::FeatureA16);
21025ffd83dbSDimitry Andric }
21035ffd83dbSDimitry Andric 
hasG16(const MCSubtargetInfo & STI)21045ffd83dbSDimitry Andric bool hasG16(const MCSubtargetInfo &STI) {
210506c3fb27SDimitry Andric   return STI.hasFeature(AMDGPU::FeatureG16);
21060b57cec5SDimitry Andric }
21070b57cec5SDimitry Andric 
hasPackedD16(const MCSubtargetInfo & STI)21080b57cec5SDimitry Andric bool hasPackedD16(const MCSubtargetInfo &STI) {
210906c3fb27SDimitry Andric   return !STI.hasFeature(AMDGPU::FeatureUnpackedD16VMem) && !isCI(STI) &&
211081ad6265SDimitry Andric          !isSI(STI);
21110b57cec5SDimitry Andric }
21120b57cec5SDimitry Andric 
hasGDS(const MCSubtargetInfo & STI)21135f757f3fSDimitry Andric bool hasGDS(const MCSubtargetInfo &STI) {
21145f757f3fSDimitry Andric   return STI.hasFeature(AMDGPU::FeatureGDS);
21155f757f3fSDimitry Andric }
21165f757f3fSDimitry Andric 
getNSAMaxSize(const MCSubtargetInfo & STI,bool HasSampler)21175f757f3fSDimitry Andric unsigned getNSAMaxSize(const MCSubtargetInfo &STI, bool HasSampler) {
211806c3fb27SDimitry Andric   auto Version = getIsaVersion(STI.getCPU());
211906c3fb27SDimitry Andric   if (Version.Major == 10)
212006c3fb27SDimitry Andric     return Version.Minor >= 3 ? 13 : 5;
212106c3fb27SDimitry Andric   if (Version.Major == 11)
212206c3fb27SDimitry Andric     return 5;
21235f757f3fSDimitry Andric   if (Version.Major >= 12)
21245f757f3fSDimitry Andric     return HasSampler ? 4 : 5;
212506c3fb27SDimitry Andric   return 0;
212606c3fb27SDimitry Andric }
212706c3fb27SDimitry Andric 
getMaxNumUserSGPRs(const MCSubtargetInfo & STI)21285f757f3fSDimitry Andric unsigned getMaxNumUserSGPRs(const MCSubtargetInfo &STI) { return 16; }
21295f757f3fSDimitry Andric 
isSI(const MCSubtargetInfo & STI)21300b57cec5SDimitry Andric bool isSI(const MCSubtargetInfo &STI) {
213106c3fb27SDimitry Andric   return STI.hasFeature(AMDGPU::FeatureSouthernIslands);
21320b57cec5SDimitry Andric }
21330b57cec5SDimitry Andric 
isCI(const MCSubtargetInfo & STI)21340b57cec5SDimitry Andric bool isCI(const MCSubtargetInfo &STI) {
213506c3fb27SDimitry Andric   return STI.hasFeature(AMDGPU::FeatureSeaIslands);
21360b57cec5SDimitry Andric }
21370b57cec5SDimitry Andric 
isVI(const MCSubtargetInfo & STI)21380b57cec5SDimitry Andric bool isVI(const MCSubtargetInfo &STI) {
213906c3fb27SDimitry Andric   return STI.hasFeature(AMDGPU::FeatureVolcanicIslands);
21400b57cec5SDimitry Andric }
21410b57cec5SDimitry Andric 
isGFX9(const MCSubtargetInfo & STI)21420b57cec5SDimitry Andric bool isGFX9(const MCSubtargetInfo &STI) {
214306c3fb27SDimitry Andric   return STI.hasFeature(AMDGPU::FeatureGFX9);
21440b57cec5SDimitry Andric }
21450b57cec5SDimitry Andric 
isGFX9_GFX10(const MCSubtargetInfo & STI)214681ad6265SDimitry Andric bool isGFX9_GFX10(const MCSubtargetInfo &STI) {
214781ad6265SDimitry Andric   return isGFX9(STI) || isGFX10(STI);
214881ad6265SDimitry Andric }
214981ad6265SDimitry Andric 
isGFX9_GFX10_GFX11(const MCSubtargetInfo & STI)21505f757f3fSDimitry Andric bool isGFX9_GFX10_GFX11(const MCSubtargetInfo &STI) {
21515f757f3fSDimitry Andric   return isGFX9(STI) || isGFX10(STI) || isGFX11(STI);
21525f757f3fSDimitry Andric }
21535f757f3fSDimitry Andric 
isGFX8_GFX9_GFX10(const MCSubtargetInfo & STI)215481ad6265SDimitry Andric bool isGFX8_GFX9_GFX10(const MCSubtargetInfo &STI) {
215581ad6265SDimitry Andric   return isVI(STI) || isGFX9(STI) || isGFX10(STI);
215681ad6265SDimitry Andric }
215781ad6265SDimitry Andric 
isGFX8Plus(const MCSubtargetInfo & STI)215881ad6265SDimitry Andric bool isGFX8Plus(const MCSubtargetInfo &STI) {
215981ad6265SDimitry Andric   return isVI(STI) || isGFX9Plus(STI);
216081ad6265SDimitry Andric }
216181ad6265SDimitry Andric 
isGFX9Plus(const MCSubtargetInfo & STI)2162e8d8bef9SDimitry Andric bool isGFX9Plus(const MCSubtargetInfo &STI) {
2163e8d8bef9SDimitry Andric   return isGFX9(STI) || isGFX10Plus(STI);
2164e8d8bef9SDimitry Andric }
2165e8d8bef9SDimitry Andric 
isGFX10(const MCSubtargetInfo & STI)21660b57cec5SDimitry Andric bool isGFX10(const MCSubtargetInfo &STI) {
216706c3fb27SDimitry Andric   return STI.hasFeature(AMDGPU::FeatureGFX10);
21680b57cec5SDimitry Andric }
21690b57cec5SDimitry Andric 
isGFX10_GFX11(const MCSubtargetInfo & STI)21705f757f3fSDimitry Andric bool isGFX10_GFX11(const MCSubtargetInfo &STI) {
21715f757f3fSDimitry Andric   return isGFX10(STI) || isGFX11(STI);
21725f757f3fSDimitry Andric }
21735f757f3fSDimitry Andric 
isGFX10Plus(const MCSubtargetInfo & STI)217481ad6265SDimitry Andric bool isGFX10Plus(const MCSubtargetInfo &STI) {
217581ad6265SDimitry Andric   return isGFX10(STI) || isGFX11Plus(STI);
217681ad6265SDimitry Andric }
217781ad6265SDimitry Andric 
isGFX11(const MCSubtargetInfo & STI)217881ad6265SDimitry Andric bool isGFX11(const MCSubtargetInfo &STI) {
217906c3fb27SDimitry Andric   return STI.hasFeature(AMDGPU::FeatureGFX11);
218081ad6265SDimitry Andric }
218181ad6265SDimitry Andric 
isGFX11Plus(const MCSubtargetInfo & STI)218281ad6265SDimitry Andric bool isGFX11Plus(const MCSubtargetInfo &STI) {
21835f757f3fSDimitry Andric   return isGFX11(STI) || isGFX12Plus(STI);
218481ad6265SDimitry Andric }
218581ad6265SDimitry Andric 
isGFX12(const MCSubtargetInfo & STI)21865f757f3fSDimitry Andric bool isGFX12(const MCSubtargetInfo &STI) {
21875f757f3fSDimitry Andric   return STI.getFeatureBits()[AMDGPU::FeatureGFX12];
21885f757f3fSDimitry Andric }
21895f757f3fSDimitry Andric 
isGFX12Plus(const MCSubtargetInfo & STI)21905f757f3fSDimitry Andric bool isGFX12Plus(const MCSubtargetInfo &STI) { return isGFX12(STI); }
21915f757f3fSDimitry Andric 
isNotGFX12Plus(const MCSubtargetInfo & STI)21925f757f3fSDimitry Andric bool isNotGFX12Plus(const MCSubtargetInfo &STI) { return !isGFX12Plus(STI); }
21935f757f3fSDimitry Andric 
isNotGFX11Plus(const MCSubtargetInfo & STI)219481ad6265SDimitry Andric bool isNotGFX11Plus(const MCSubtargetInfo &STI) {
219581ad6265SDimitry Andric   return !isGFX11Plus(STI);
219681ad6265SDimitry Andric }
219781ad6265SDimitry Andric 
isNotGFX10Plus(const MCSubtargetInfo & STI)219881ad6265SDimitry Andric bool isNotGFX10Plus(const MCSubtargetInfo &STI) {
219981ad6265SDimitry Andric   return isSI(STI) || isCI(STI) || isVI(STI) || isGFX9(STI);
220081ad6265SDimitry Andric }
220181ad6265SDimitry Andric 
isGFX10Before1030(const MCSubtargetInfo & STI)220281ad6265SDimitry Andric bool isGFX10Before1030(const MCSubtargetInfo &STI) {
220381ad6265SDimitry Andric   return isGFX10(STI) && !AMDGPU::isGFX10_BEncoding(STI);
220481ad6265SDimitry Andric }
2205e8d8bef9SDimitry Andric 
isGCN3Encoding(const MCSubtargetInfo & STI)22060b57cec5SDimitry Andric bool isGCN3Encoding(const MCSubtargetInfo &STI) {
220706c3fb27SDimitry Andric   return STI.hasFeature(AMDGPU::FeatureGCN3Encoding);
22080b57cec5SDimitry Andric }
22090b57cec5SDimitry Andric 
isGFX10_AEncoding(const MCSubtargetInfo & STI)2210fe6060f1SDimitry Andric bool isGFX10_AEncoding(const MCSubtargetInfo &STI) {
221106c3fb27SDimitry Andric   return STI.hasFeature(AMDGPU::FeatureGFX10_AEncoding);
2212fe6060f1SDimitry Andric }
2213fe6060f1SDimitry Andric 
isGFX10_BEncoding(const MCSubtargetInfo & STI)22145ffd83dbSDimitry Andric bool isGFX10_BEncoding(const MCSubtargetInfo &STI) {
221506c3fb27SDimitry Andric   return STI.hasFeature(AMDGPU::FeatureGFX10_BEncoding);
22165ffd83dbSDimitry Andric }
22175ffd83dbSDimitry Andric 
hasGFX10_3Insts(const MCSubtargetInfo & STI)22185ffd83dbSDimitry Andric bool hasGFX10_3Insts(const MCSubtargetInfo &STI) {
221906c3fb27SDimitry Andric   return STI.hasFeature(AMDGPU::FeatureGFX10_3Insts);
22205ffd83dbSDimitry Andric }
22215ffd83dbSDimitry Andric 
isGFX10_3_GFX11(const MCSubtargetInfo & STI)22225f757f3fSDimitry Andric bool isGFX10_3_GFX11(const MCSubtargetInfo &STI) {
22235f757f3fSDimitry Andric   return isGFX10_BEncoding(STI) && !isGFX12Plus(STI);
22245f757f3fSDimitry Andric }
22255f757f3fSDimitry Andric 
isGFX90A(const MCSubtargetInfo & STI)2226fe6060f1SDimitry Andric bool isGFX90A(const MCSubtargetInfo &STI) {
222706c3fb27SDimitry Andric   return STI.hasFeature(AMDGPU::FeatureGFX90AInsts);
2228fe6060f1SDimitry Andric }
2229fe6060f1SDimitry Andric 
isGFX940(const MCSubtargetInfo & STI)223081ad6265SDimitry Andric bool isGFX940(const MCSubtargetInfo &STI) {
223106c3fb27SDimitry Andric   return STI.hasFeature(AMDGPU::FeatureGFX940Insts);
223281ad6265SDimitry Andric }
223381ad6265SDimitry Andric 
hasArchitectedFlatScratch(const MCSubtargetInfo & STI)2234fe6060f1SDimitry Andric bool hasArchitectedFlatScratch(const MCSubtargetInfo &STI) {
223506c3fb27SDimitry Andric   return STI.hasFeature(AMDGPU::FeatureArchitectedFlatScratch);
2236fe6060f1SDimitry Andric }
2237fe6060f1SDimitry Andric 
hasMAIInsts(const MCSubtargetInfo & STI)223881ad6265SDimitry Andric bool hasMAIInsts(const MCSubtargetInfo &STI) {
223906c3fb27SDimitry Andric   return STI.hasFeature(AMDGPU::FeatureMAIInsts);
224081ad6265SDimitry Andric }
224181ad6265SDimitry Andric 
hasVOPD(const MCSubtargetInfo & STI)224281ad6265SDimitry Andric bool hasVOPD(const MCSubtargetInfo &STI) {
224306c3fb27SDimitry Andric   return STI.hasFeature(AMDGPU::FeatureVOPD);
224481ad6265SDimitry Andric }
224581ad6265SDimitry Andric 
hasDPPSrc1SGPR(const MCSubtargetInfo & STI)22465f757f3fSDimitry Andric bool hasDPPSrc1SGPR(const MCSubtargetInfo &STI) {
22475f757f3fSDimitry Andric   return STI.hasFeature(AMDGPU::FeatureDPPSrc1SGPR);
22485f757f3fSDimitry Andric }
22495f757f3fSDimitry Andric 
hasKernargPreload(const MCSubtargetInfo & STI)22505f757f3fSDimitry Andric unsigned hasKernargPreload(const MCSubtargetInfo &STI) {
22515f757f3fSDimitry Andric   return STI.hasFeature(AMDGPU::FeatureKernargPreload);
22525f757f3fSDimitry Andric }
22535f757f3fSDimitry Andric 
getTotalNumVGPRs(bool has90AInsts,int32_t ArgNumAGPR,int32_t ArgNumVGPR)225481ad6265SDimitry Andric int32_t getTotalNumVGPRs(bool has90AInsts, int32_t ArgNumAGPR,
225581ad6265SDimitry Andric                          int32_t ArgNumVGPR) {
225681ad6265SDimitry Andric   if (has90AInsts && ArgNumAGPR)
225781ad6265SDimitry Andric     return alignTo(ArgNumVGPR, 4) + ArgNumAGPR;
225881ad6265SDimitry Andric   return std::max(ArgNumVGPR, ArgNumAGPR);
225981ad6265SDimitry Andric }
226081ad6265SDimitry Andric 
isSGPR(unsigned Reg,const MCRegisterInfo * TRI)22610b57cec5SDimitry Andric bool isSGPR(unsigned Reg, const MCRegisterInfo* TRI) {
22620b57cec5SDimitry Andric   const MCRegisterClass SGPRClass = TRI->getRegClass(AMDGPU::SReg_32RegClassID);
22635ffd83dbSDimitry Andric   const unsigned FirstSubReg = TRI->getSubReg(Reg, AMDGPU::sub0);
22640b57cec5SDimitry Andric   return SGPRClass.contains(FirstSubReg != 0 ? FirstSubReg : Reg) ||
22650b57cec5SDimitry Andric     Reg == AMDGPU::SCC;
22660b57cec5SDimitry Andric }
22670b57cec5SDimitry Andric 
isHi(unsigned Reg,const MCRegisterInfo & MRI)22685f757f3fSDimitry Andric bool isHi(unsigned Reg, const MCRegisterInfo &MRI) {
22695f757f3fSDimitry Andric   return MRI.getEncodingValue(Reg) & AMDGPU::HWEncoding::IS_HI;
22705f757f3fSDimitry Andric }
22715f757f3fSDimitry Andric 
22720b57cec5SDimitry Andric #define MAP_REG2REG \
22730b57cec5SDimitry Andric   using namespace AMDGPU; \
22740b57cec5SDimitry Andric   switch(Reg) { \
22750b57cec5SDimitry Andric   default: return Reg; \
22760b57cec5SDimitry Andric   CASE_CI_VI(FLAT_SCR) \
22770b57cec5SDimitry Andric   CASE_CI_VI(FLAT_SCR_LO) \
22780b57cec5SDimitry Andric   CASE_CI_VI(FLAT_SCR_HI) \
2279e8d8bef9SDimitry Andric   CASE_VI_GFX9PLUS(TTMP0) \
2280e8d8bef9SDimitry Andric   CASE_VI_GFX9PLUS(TTMP1) \
2281e8d8bef9SDimitry Andric   CASE_VI_GFX9PLUS(TTMP2) \
2282e8d8bef9SDimitry Andric   CASE_VI_GFX9PLUS(TTMP3) \
2283e8d8bef9SDimitry Andric   CASE_VI_GFX9PLUS(TTMP4) \
2284e8d8bef9SDimitry Andric   CASE_VI_GFX9PLUS(TTMP5) \
2285e8d8bef9SDimitry Andric   CASE_VI_GFX9PLUS(TTMP6) \
2286e8d8bef9SDimitry Andric   CASE_VI_GFX9PLUS(TTMP7) \
2287e8d8bef9SDimitry Andric   CASE_VI_GFX9PLUS(TTMP8) \
2288e8d8bef9SDimitry Andric   CASE_VI_GFX9PLUS(TTMP9) \
2289e8d8bef9SDimitry Andric   CASE_VI_GFX9PLUS(TTMP10) \
2290e8d8bef9SDimitry Andric   CASE_VI_GFX9PLUS(TTMP11) \
2291e8d8bef9SDimitry Andric   CASE_VI_GFX9PLUS(TTMP12) \
2292e8d8bef9SDimitry Andric   CASE_VI_GFX9PLUS(TTMP13) \
2293e8d8bef9SDimitry Andric   CASE_VI_GFX9PLUS(TTMP14) \
2294e8d8bef9SDimitry Andric   CASE_VI_GFX9PLUS(TTMP15) \
2295e8d8bef9SDimitry Andric   CASE_VI_GFX9PLUS(TTMP0_TTMP1) \
2296e8d8bef9SDimitry Andric   CASE_VI_GFX9PLUS(TTMP2_TTMP3) \
2297e8d8bef9SDimitry Andric   CASE_VI_GFX9PLUS(TTMP4_TTMP5) \
2298e8d8bef9SDimitry Andric   CASE_VI_GFX9PLUS(TTMP6_TTMP7) \
2299e8d8bef9SDimitry Andric   CASE_VI_GFX9PLUS(TTMP8_TTMP9) \
2300e8d8bef9SDimitry Andric   CASE_VI_GFX9PLUS(TTMP10_TTMP11) \
2301e8d8bef9SDimitry Andric   CASE_VI_GFX9PLUS(TTMP12_TTMP13) \
2302e8d8bef9SDimitry Andric   CASE_VI_GFX9PLUS(TTMP14_TTMP15) \
2303e8d8bef9SDimitry Andric   CASE_VI_GFX9PLUS(TTMP0_TTMP1_TTMP2_TTMP3) \
2304e8d8bef9SDimitry Andric   CASE_VI_GFX9PLUS(TTMP4_TTMP5_TTMP6_TTMP7) \
2305e8d8bef9SDimitry Andric   CASE_VI_GFX9PLUS(TTMP8_TTMP9_TTMP10_TTMP11) \
2306e8d8bef9SDimitry Andric   CASE_VI_GFX9PLUS(TTMP12_TTMP13_TTMP14_TTMP15) \
2307e8d8bef9SDimitry Andric   CASE_VI_GFX9PLUS(TTMP0_TTMP1_TTMP2_TTMP3_TTMP4_TTMP5_TTMP6_TTMP7) \
2308e8d8bef9SDimitry Andric   CASE_VI_GFX9PLUS(TTMP4_TTMP5_TTMP6_TTMP7_TTMP8_TTMP9_TTMP10_TTMP11) \
2309e8d8bef9SDimitry Andric   CASE_VI_GFX9PLUS(TTMP8_TTMP9_TTMP10_TTMP11_TTMP12_TTMP13_TTMP14_TTMP15) \
2310e8d8bef9SDimitry Andric   CASE_VI_GFX9PLUS(TTMP0_TTMP1_TTMP2_TTMP3_TTMP4_TTMP5_TTMP6_TTMP7_TTMP8_TTMP9_TTMP10_TTMP11_TTMP12_TTMP13_TTMP14_TTMP15) \
231181ad6265SDimitry Andric   CASE_GFXPRE11_GFX11PLUS(M0) \
231281ad6265SDimitry Andric   CASE_GFXPRE11_GFX11PLUS(SGPR_NULL) \
231381ad6265SDimitry Andric   CASE_GFXPRE11_GFX11PLUS_TO(SGPR_NULL64, SGPR_NULL) \
23140b57cec5SDimitry Andric   }
23150b57cec5SDimitry Andric 
23160b57cec5SDimitry Andric #define CASE_CI_VI(node) \
23170b57cec5SDimitry Andric   assert(!isSI(STI)); \
23180b57cec5SDimitry Andric   case node: return isCI(STI) ? node##_ci : node##_vi;
23190b57cec5SDimitry Andric 
2320e8d8bef9SDimitry Andric #define CASE_VI_GFX9PLUS(node) \
2321e8d8bef9SDimitry Andric   case node: return isGFX9Plus(STI) ? node##_gfx9plus : node##_vi;
23220b57cec5SDimitry Andric 
232381ad6265SDimitry Andric #define CASE_GFXPRE11_GFX11PLUS(node) \
232481ad6265SDimitry Andric   case node: return isGFX11Plus(STI) ? node##_gfx11plus : node##_gfxpre11;
232581ad6265SDimitry Andric 
232681ad6265SDimitry Andric #define CASE_GFXPRE11_GFX11PLUS_TO(node, result) \
232781ad6265SDimitry Andric   case node: return isGFX11Plus(STI) ? result##_gfx11plus : result##_gfxpre11;
232881ad6265SDimitry Andric 
getMCReg(unsigned Reg,const MCSubtargetInfo & STI)23290b57cec5SDimitry Andric unsigned getMCReg(unsigned Reg, const MCSubtargetInfo &STI) {
23300b57cec5SDimitry Andric   if (STI.getTargetTriple().getArch() == Triple::r600)
23310b57cec5SDimitry Andric     return Reg;
23320b57cec5SDimitry Andric   MAP_REG2REG
23330b57cec5SDimitry Andric }
23340b57cec5SDimitry Andric 
23350b57cec5SDimitry Andric #undef CASE_CI_VI
2336e8d8bef9SDimitry Andric #undef CASE_VI_GFX9PLUS
233781ad6265SDimitry Andric #undef CASE_GFXPRE11_GFX11PLUS
233881ad6265SDimitry Andric #undef CASE_GFXPRE11_GFX11PLUS_TO
23390b57cec5SDimitry Andric 
23400b57cec5SDimitry Andric #define CASE_CI_VI(node)   case node##_ci: case node##_vi:   return node;
2341e8d8bef9SDimitry Andric #define CASE_VI_GFX9PLUS(node) case node##_vi: case node##_gfx9plus: return node;
234281ad6265SDimitry Andric #define CASE_GFXPRE11_GFX11PLUS(node) case node##_gfx11plus: case node##_gfxpre11: return node;
234381ad6265SDimitry Andric #define CASE_GFXPRE11_GFX11PLUS_TO(node, result)
23440b57cec5SDimitry Andric 
mc2PseudoReg(unsigned Reg)23450b57cec5SDimitry Andric unsigned mc2PseudoReg(unsigned Reg) {
23460b57cec5SDimitry Andric   MAP_REG2REG
23470b57cec5SDimitry Andric }
23480b57cec5SDimitry Andric 
isInlineValue(unsigned Reg)2349bdd1243dSDimitry Andric bool isInlineValue(unsigned Reg) {
2350bdd1243dSDimitry Andric   switch (Reg) {
2351bdd1243dSDimitry Andric   case AMDGPU::SRC_SHARED_BASE_LO:
2352bdd1243dSDimitry Andric   case AMDGPU::SRC_SHARED_BASE:
2353bdd1243dSDimitry Andric   case AMDGPU::SRC_SHARED_LIMIT_LO:
2354bdd1243dSDimitry Andric   case AMDGPU::SRC_SHARED_LIMIT:
2355bdd1243dSDimitry Andric   case AMDGPU::SRC_PRIVATE_BASE_LO:
2356bdd1243dSDimitry Andric   case AMDGPU::SRC_PRIVATE_BASE:
2357bdd1243dSDimitry Andric   case AMDGPU::SRC_PRIVATE_LIMIT_LO:
2358bdd1243dSDimitry Andric   case AMDGPU::SRC_PRIVATE_LIMIT:
2359bdd1243dSDimitry Andric   case AMDGPU::SRC_POPS_EXITING_WAVE_ID:
2360bdd1243dSDimitry Andric     return true;
2361bdd1243dSDimitry Andric   case AMDGPU::SRC_VCCZ:
2362bdd1243dSDimitry Andric   case AMDGPU::SRC_EXECZ:
2363bdd1243dSDimitry Andric   case AMDGPU::SRC_SCC:
2364bdd1243dSDimitry Andric     return true;
2365bdd1243dSDimitry Andric   case AMDGPU::SGPR_NULL:
2366bdd1243dSDimitry Andric     return true;
2367bdd1243dSDimitry Andric   default:
2368bdd1243dSDimitry Andric     return false;
2369bdd1243dSDimitry Andric   }
2370bdd1243dSDimitry Andric }
2371bdd1243dSDimitry Andric 
23720b57cec5SDimitry Andric #undef CASE_CI_VI
2373e8d8bef9SDimitry Andric #undef CASE_VI_GFX9PLUS
237481ad6265SDimitry Andric #undef CASE_GFXPRE11_GFX11PLUS
237581ad6265SDimitry Andric #undef CASE_GFXPRE11_GFX11PLUS_TO
23760b57cec5SDimitry Andric #undef MAP_REG2REG
23770b57cec5SDimitry Andric 
isSISrcOperand(const MCInstrDesc & Desc,unsigned OpNo)23780b57cec5SDimitry Andric bool isSISrcOperand(const MCInstrDesc &Desc, unsigned OpNo) {
23790b57cec5SDimitry Andric   assert(OpNo < Desc.NumOperands);
2380bdd1243dSDimitry Andric   unsigned OpType = Desc.operands()[OpNo].OperandType;
23810b57cec5SDimitry Andric   return OpType >= AMDGPU::OPERAND_SRC_FIRST &&
23820b57cec5SDimitry Andric          OpType <= AMDGPU::OPERAND_SRC_LAST;
23830b57cec5SDimitry Andric }
23840b57cec5SDimitry Andric 
isKImmOperand(const MCInstrDesc & Desc,unsigned OpNo)2385bdd1243dSDimitry Andric bool isKImmOperand(const MCInstrDesc &Desc, unsigned OpNo) {
2386bdd1243dSDimitry Andric   assert(OpNo < Desc.NumOperands);
2387bdd1243dSDimitry Andric   unsigned OpType = Desc.operands()[OpNo].OperandType;
2388bdd1243dSDimitry Andric   return OpType >= AMDGPU::OPERAND_KIMM_FIRST &&
2389bdd1243dSDimitry Andric          OpType <= AMDGPU::OPERAND_KIMM_LAST;
2390bdd1243dSDimitry Andric }
2391bdd1243dSDimitry Andric 
isSISrcFPOperand(const MCInstrDesc & Desc,unsigned OpNo)23920b57cec5SDimitry Andric bool isSISrcFPOperand(const MCInstrDesc &Desc, unsigned OpNo) {
23930b57cec5SDimitry Andric   assert(OpNo < Desc.NumOperands);
2394bdd1243dSDimitry Andric   unsigned OpType = Desc.operands()[OpNo].OperandType;
23950b57cec5SDimitry Andric   switch (OpType) {
23960b57cec5SDimitry Andric   case AMDGPU::OPERAND_REG_IMM_FP32:
2397349cc55cSDimitry Andric   case AMDGPU::OPERAND_REG_IMM_FP32_DEFERRED:
23980b57cec5SDimitry Andric   case AMDGPU::OPERAND_REG_IMM_FP64:
23990b57cec5SDimitry Andric   case AMDGPU::OPERAND_REG_IMM_FP16:
2400349cc55cSDimitry Andric   case AMDGPU::OPERAND_REG_IMM_FP16_DEFERRED:
24010b57cec5SDimitry Andric   case AMDGPU::OPERAND_REG_IMM_V2FP16:
24020b57cec5SDimitry Andric   case AMDGPU::OPERAND_REG_INLINE_C_FP32:
24030b57cec5SDimitry Andric   case AMDGPU::OPERAND_REG_INLINE_C_FP64:
24040b57cec5SDimitry Andric   case AMDGPU::OPERAND_REG_INLINE_C_FP16:
24050b57cec5SDimitry Andric   case AMDGPU::OPERAND_REG_INLINE_C_V2FP16:
24060b57cec5SDimitry Andric   case AMDGPU::OPERAND_REG_INLINE_AC_FP32:
24070b57cec5SDimitry Andric   case AMDGPU::OPERAND_REG_INLINE_AC_FP16:
24080b57cec5SDimitry Andric   case AMDGPU::OPERAND_REG_INLINE_AC_V2FP16:
2409fe6060f1SDimitry Andric   case AMDGPU::OPERAND_REG_IMM_V2FP32:
2410fe6060f1SDimitry Andric   case AMDGPU::OPERAND_REG_INLINE_C_V2FP32:
2411fe6060f1SDimitry Andric   case AMDGPU::OPERAND_REG_INLINE_AC_FP64:
24120b57cec5SDimitry Andric     return true;
24130b57cec5SDimitry Andric   default:
24140b57cec5SDimitry Andric     return false;
24150b57cec5SDimitry Andric   }
24160b57cec5SDimitry Andric }
24170b57cec5SDimitry Andric 
isSISrcInlinableOperand(const MCInstrDesc & Desc,unsigned OpNo)24180b57cec5SDimitry Andric bool isSISrcInlinableOperand(const MCInstrDesc &Desc, unsigned OpNo) {
24190b57cec5SDimitry Andric   assert(OpNo < Desc.NumOperands);
2420bdd1243dSDimitry Andric   unsigned OpType = Desc.operands()[OpNo].OperandType;
24215f757f3fSDimitry Andric   return (OpType >= AMDGPU::OPERAND_REG_INLINE_C_FIRST &&
24225f757f3fSDimitry Andric           OpType <= AMDGPU::OPERAND_REG_INLINE_C_LAST) ||
24235f757f3fSDimitry Andric          (OpType >= AMDGPU::OPERAND_REG_INLINE_AC_FIRST &&
24245f757f3fSDimitry Andric           OpType <= AMDGPU::OPERAND_REG_INLINE_AC_LAST);
24250b57cec5SDimitry Andric }
24260b57cec5SDimitry Andric 
24270b57cec5SDimitry Andric // Avoid using MCRegisterClass::getSize, since that function will go away
24280b57cec5SDimitry Andric // (move from MC* level to Target* level). Return size in bits.
getRegBitWidth(unsigned RCID)24290b57cec5SDimitry Andric unsigned getRegBitWidth(unsigned RCID) {
24300b57cec5SDimitry Andric   switch (RCID) {
24315ffd83dbSDimitry Andric   case AMDGPU::SGPR_LO16RegClassID:
24325ffd83dbSDimitry Andric   case AMDGPU::AGPR_LO16RegClassID:
24335ffd83dbSDimitry Andric     return 16;
24340b57cec5SDimitry Andric   case AMDGPU::SGPR_32RegClassID:
24350b57cec5SDimitry Andric   case AMDGPU::VGPR_32RegClassID:
24360b57cec5SDimitry Andric   case AMDGPU::VRegOrLds_32RegClassID:
24370b57cec5SDimitry Andric   case AMDGPU::AGPR_32RegClassID:
24380b57cec5SDimitry Andric   case AMDGPU::VS_32RegClassID:
24390b57cec5SDimitry Andric   case AMDGPU::AV_32RegClassID:
24400b57cec5SDimitry Andric   case AMDGPU::SReg_32RegClassID:
24410b57cec5SDimitry Andric   case AMDGPU::SReg_32_XM0RegClassID:
24420b57cec5SDimitry Andric   case AMDGPU::SRegOrLds_32RegClassID:
24430b57cec5SDimitry Andric     return 32;
24440b57cec5SDimitry Andric   case AMDGPU::SGPR_64RegClassID:
24450b57cec5SDimitry Andric   case AMDGPU::VS_64RegClassID:
24460b57cec5SDimitry Andric   case AMDGPU::SReg_64RegClassID:
24470b57cec5SDimitry Andric   case AMDGPU::VReg_64RegClassID:
24480b57cec5SDimitry Andric   case AMDGPU::AReg_64RegClassID:
24490b57cec5SDimitry Andric   case AMDGPU::SReg_64_XEXECRegClassID:
2450fe6060f1SDimitry Andric   case AMDGPU::VReg_64_Align2RegClassID:
2451fe6060f1SDimitry Andric   case AMDGPU::AReg_64_Align2RegClassID:
24524824e7fdSDimitry Andric   case AMDGPU::AV_64RegClassID:
24534824e7fdSDimitry Andric   case AMDGPU::AV_64_Align2RegClassID:
24540b57cec5SDimitry Andric     return 64;
24550b57cec5SDimitry Andric   case AMDGPU::SGPR_96RegClassID:
24560b57cec5SDimitry Andric   case AMDGPU::SReg_96RegClassID:
24570b57cec5SDimitry Andric   case AMDGPU::VReg_96RegClassID:
24585ffd83dbSDimitry Andric   case AMDGPU::AReg_96RegClassID:
2459fe6060f1SDimitry Andric   case AMDGPU::VReg_96_Align2RegClassID:
2460fe6060f1SDimitry Andric   case AMDGPU::AReg_96_Align2RegClassID:
2461fe6060f1SDimitry Andric   case AMDGPU::AV_96RegClassID:
24624824e7fdSDimitry Andric   case AMDGPU::AV_96_Align2RegClassID:
24630b57cec5SDimitry Andric     return 96;
24640b57cec5SDimitry Andric   case AMDGPU::SGPR_128RegClassID:
24650b57cec5SDimitry Andric   case AMDGPU::SReg_128RegClassID:
24660b57cec5SDimitry Andric   case AMDGPU::VReg_128RegClassID:
24670b57cec5SDimitry Andric   case AMDGPU::AReg_128RegClassID:
2468fe6060f1SDimitry Andric   case AMDGPU::VReg_128_Align2RegClassID:
2469fe6060f1SDimitry Andric   case AMDGPU::AReg_128_Align2RegClassID:
2470fe6060f1SDimitry Andric   case AMDGPU::AV_128RegClassID:
24714824e7fdSDimitry Andric   case AMDGPU::AV_128_Align2RegClassID:
24720b57cec5SDimitry Andric     return 128;
24730b57cec5SDimitry Andric   case AMDGPU::SGPR_160RegClassID:
24740b57cec5SDimitry Andric   case AMDGPU::SReg_160RegClassID:
24750b57cec5SDimitry Andric   case AMDGPU::VReg_160RegClassID:
24765ffd83dbSDimitry Andric   case AMDGPU::AReg_160RegClassID:
2477fe6060f1SDimitry Andric   case AMDGPU::VReg_160_Align2RegClassID:
2478fe6060f1SDimitry Andric   case AMDGPU::AReg_160_Align2RegClassID:
2479fe6060f1SDimitry Andric   case AMDGPU::AV_160RegClassID:
24804824e7fdSDimitry Andric   case AMDGPU::AV_160_Align2RegClassID:
24810b57cec5SDimitry Andric     return 160;
24825ffd83dbSDimitry Andric   case AMDGPU::SGPR_192RegClassID:
24835ffd83dbSDimitry Andric   case AMDGPU::SReg_192RegClassID:
24845ffd83dbSDimitry Andric   case AMDGPU::VReg_192RegClassID:
24855ffd83dbSDimitry Andric   case AMDGPU::AReg_192RegClassID:
2486fe6060f1SDimitry Andric   case AMDGPU::VReg_192_Align2RegClassID:
2487fe6060f1SDimitry Andric   case AMDGPU::AReg_192_Align2RegClassID:
24884824e7fdSDimitry Andric   case AMDGPU::AV_192RegClassID:
24894824e7fdSDimitry Andric   case AMDGPU::AV_192_Align2RegClassID:
24905ffd83dbSDimitry Andric     return 192;
2491fe6060f1SDimitry Andric   case AMDGPU::SGPR_224RegClassID:
2492fe6060f1SDimitry Andric   case AMDGPU::SReg_224RegClassID:
2493fe6060f1SDimitry Andric   case AMDGPU::VReg_224RegClassID:
2494fe6060f1SDimitry Andric   case AMDGPU::AReg_224RegClassID:
2495fe6060f1SDimitry Andric   case AMDGPU::VReg_224_Align2RegClassID:
2496fe6060f1SDimitry Andric   case AMDGPU::AReg_224_Align2RegClassID:
24974824e7fdSDimitry Andric   case AMDGPU::AV_224RegClassID:
24984824e7fdSDimitry Andric   case AMDGPU::AV_224_Align2RegClassID:
2499fe6060f1SDimitry Andric     return 224;
25005ffd83dbSDimitry Andric   case AMDGPU::SGPR_256RegClassID:
25010b57cec5SDimitry Andric   case AMDGPU::SReg_256RegClassID:
25020b57cec5SDimitry Andric   case AMDGPU::VReg_256RegClassID:
25035ffd83dbSDimitry Andric   case AMDGPU::AReg_256RegClassID:
2504fe6060f1SDimitry Andric   case AMDGPU::VReg_256_Align2RegClassID:
2505fe6060f1SDimitry Andric   case AMDGPU::AReg_256_Align2RegClassID:
25064824e7fdSDimitry Andric   case AMDGPU::AV_256RegClassID:
25074824e7fdSDimitry Andric   case AMDGPU::AV_256_Align2RegClassID:
25080b57cec5SDimitry Andric     return 256;
2509bdd1243dSDimitry Andric   case AMDGPU::SGPR_288RegClassID:
2510bdd1243dSDimitry Andric   case AMDGPU::SReg_288RegClassID:
2511bdd1243dSDimitry Andric   case AMDGPU::VReg_288RegClassID:
2512bdd1243dSDimitry Andric   case AMDGPU::AReg_288RegClassID:
2513bdd1243dSDimitry Andric   case AMDGPU::VReg_288_Align2RegClassID:
2514bdd1243dSDimitry Andric   case AMDGPU::AReg_288_Align2RegClassID:
2515bdd1243dSDimitry Andric   case AMDGPU::AV_288RegClassID:
2516bdd1243dSDimitry Andric   case AMDGPU::AV_288_Align2RegClassID:
2517bdd1243dSDimitry Andric     return 288;
2518bdd1243dSDimitry Andric   case AMDGPU::SGPR_320RegClassID:
2519bdd1243dSDimitry Andric   case AMDGPU::SReg_320RegClassID:
2520bdd1243dSDimitry Andric   case AMDGPU::VReg_320RegClassID:
2521bdd1243dSDimitry Andric   case AMDGPU::AReg_320RegClassID:
2522bdd1243dSDimitry Andric   case AMDGPU::VReg_320_Align2RegClassID:
2523bdd1243dSDimitry Andric   case AMDGPU::AReg_320_Align2RegClassID:
2524bdd1243dSDimitry Andric   case AMDGPU::AV_320RegClassID:
2525bdd1243dSDimitry Andric   case AMDGPU::AV_320_Align2RegClassID:
2526bdd1243dSDimitry Andric     return 320;
2527bdd1243dSDimitry Andric   case AMDGPU::SGPR_352RegClassID:
2528bdd1243dSDimitry Andric   case AMDGPU::SReg_352RegClassID:
2529bdd1243dSDimitry Andric   case AMDGPU::VReg_352RegClassID:
2530bdd1243dSDimitry Andric   case AMDGPU::AReg_352RegClassID:
2531bdd1243dSDimitry Andric   case AMDGPU::VReg_352_Align2RegClassID:
2532bdd1243dSDimitry Andric   case AMDGPU::AReg_352_Align2RegClassID:
2533bdd1243dSDimitry Andric   case AMDGPU::AV_352RegClassID:
2534bdd1243dSDimitry Andric   case AMDGPU::AV_352_Align2RegClassID:
2535bdd1243dSDimitry Andric     return 352;
2536bdd1243dSDimitry Andric   case AMDGPU::SGPR_384RegClassID:
2537bdd1243dSDimitry Andric   case AMDGPU::SReg_384RegClassID:
2538bdd1243dSDimitry Andric   case AMDGPU::VReg_384RegClassID:
2539bdd1243dSDimitry Andric   case AMDGPU::AReg_384RegClassID:
2540bdd1243dSDimitry Andric   case AMDGPU::VReg_384_Align2RegClassID:
2541bdd1243dSDimitry Andric   case AMDGPU::AReg_384_Align2RegClassID:
2542bdd1243dSDimitry Andric   case AMDGPU::AV_384RegClassID:
2543bdd1243dSDimitry Andric   case AMDGPU::AV_384_Align2RegClassID:
2544bdd1243dSDimitry Andric     return 384;
25455ffd83dbSDimitry Andric   case AMDGPU::SGPR_512RegClassID:
25460b57cec5SDimitry Andric   case AMDGPU::SReg_512RegClassID:
25470b57cec5SDimitry Andric   case AMDGPU::VReg_512RegClassID:
25480b57cec5SDimitry Andric   case AMDGPU::AReg_512RegClassID:
2549fe6060f1SDimitry Andric   case AMDGPU::VReg_512_Align2RegClassID:
2550fe6060f1SDimitry Andric   case AMDGPU::AReg_512_Align2RegClassID:
25514824e7fdSDimitry Andric   case AMDGPU::AV_512RegClassID:
25524824e7fdSDimitry Andric   case AMDGPU::AV_512_Align2RegClassID:
25530b57cec5SDimitry Andric     return 512;
25545ffd83dbSDimitry Andric   case AMDGPU::SGPR_1024RegClassID:
25550b57cec5SDimitry Andric   case AMDGPU::SReg_1024RegClassID:
25560b57cec5SDimitry Andric   case AMDGPU::VReg_1024RegClassID:
25570b57cec5SDimitry Andric   case AMDGPU::AReg_1024RegClassID:
2558fe6060f1SDimitry Andric   case AMDGPU::VReg_1024_Align2RegClassID:
2559fe6060f1SDimitry Andric   case AMDGPU::AReg_1024_Align2RegClassID:
25604824e7fdSDimitry Andric   case AMDGPU::AV_1024RegClassID:
25614824e7fdSDimitry Andric   case AMDGPU::AV_1024_Align2RegClassID:
25620b57cec5SDimitry Andric     return 1024;
25630b57cec5SDimitry Andric   default:
25640b57cec5SDimitry Andric     llvm_unreachable("Unexpected register class");
25650b57cec5SDimitry Andric   }
25660b57cec5SDimitry Andric }
25670b57cec5SDimitry Andric 
getRegBitWidth(const MCRegisterClass & RC)25680b57cec5SDimitry Andric unsigned getRegBitWidth(const MCRegisterClass &RC) {
25690b57cec5SDimitry Andric   return getRegBitWidth(RC.getID());
25700b57cec5SDimitry Andric }
25710b57cec5SDimitry Andric 
getRegOperandSize(const MCRegisterInfo * MRI,const MCInstrDesc & Desc,unsigned OpNo)25720b57cec5SDimitry Andric unsigned getRegOperandSize(const MCRegisterInfo *MRI, const MCInstrDesc &Desc,
25730b57cec5SDimitry Andric                            unsigned OpNo) {
25740b57cec5SDimitry Andric   assert(OpNo < Desc.NumOperands);
2575bdd1243dSDimitry Andric   unsigned RCID = Desc.operands()[OpNo].RegClass;
257606c3fb27SDimitry Andric   return getRegBitWidth(RCID) / 8;
25770b57cec5SDimitry Andric }
25780b57cec5SDimitry Andric 
isInlinableLiteral64(int64_t Literal,bool HasInv2Pi)25790b57cec5SDimitry Andric bool isInlinableLiteral64(int64_t Literal, bool HasInv2Pi) {
25805ffd83dbSDimitry Andric   if (isInlinableIntLiteral(Literal))
25810b57cec5SDimitry Andric     return true;
25820b57cec5SDimitry Andric 
25830b57cec5SDimitry Andric   uint64_t Val = static_cast<uint64_t>(Literal);
258406c3fb27SDimitry Andric   return (Val == llvm::bit_cast<uint64_t>(0.0)) ||
258506c3fb27SDimitry Andric          (Val == llvm::bit_cast<uint64_t>(1.0)) ||
258606c3fb27SDimitry Andric          (Val == llvm::bit_cast<uint64_t>(-1.0)) ||
258706c3fb27SDimitry Andric          (Val == llvm::bit_cast<uint64_t>(0.5)) ||
258806c3fb27SDimitry Andric          (Val == llvm::bit_cast<uint64_t>(-0.5)) ||
258906c3fb27SDimitry Andric          (Val == llvm::bit_cast<uint64_t>(2.0)) ||
259006c3fb27SDimitry Andric          (Val == llvm::bit_cast<uint64_t>(-2.0)) ||
259106c3fb27SDimitry Andric          (Val == llvm::bit_cast<uint64_t>(4.0)) ||
259206c3fb27SDimitry Andric          (Val == llvm::bit_cast<uint64_t>(-4.0)) ||
25930b57cec5SDimitry Andric          (Val == 0x3fc45f306dc9c882 && HasInv2Pi);
25940b57cec5SDimitry Andric }
25950b57cec5SDimitry Andric 
isInlinableLiteral32(int32_t Literal,bool HasInv2Pi)25960b57cec5SDimitry Andric bool isInlinableLiteral32(int32_t Literal, bool HasInv2Pi) {
25975ffd83dbSDimitry Andric   if (isInlinableIntLiteral(Literal))
25980b57cec5SDimitry Andric     return true;
25990b57cec5SDimitry Andric 
26000b57cec5SDimitry Andric   // The actual type of the operand does not seem to matter as long
26010b57cec5SDimitry Andric   // as the bits match one of the inline immediate values.  For example:
26020b57cec5SDimitry Andric   //
26030b57cec5SDimitry Andric   // -nan has the hexadecimal encoding of 0xfffffffe which is -2 in decimal,
26040b57cec5SDimitry Andric   // so it is a legal inline immediate.
26050b57cec5SDimitry Andric   //
26060b57cec5SDimitry Andric   // 1065353216 has the hexadecimal encoding 0x3f800000 which is 1.0f in
26070b57cec5SDimitry Andric   // floating-point, so it is a legal inline immediate.
26080b57cec5SDimitry Andric 
26090b57cec5SDimitry Andric   uint32_t Val = static_cast<uint32_t>(Literal);
261006c3fb27SDimitry Andric   return (Val == llvm::bit_cast<uint32_t>(0.0f)) ||
261106c3fb27SDimitry Andric          (Val == llvm::bit_cast<uint32_t>(1.0f)) ||
261206c3fb27SDimitry Andric          (Val == llvm::bit_cast<uint32_t>(-1.0f)) ||
261306c3fb27SDimitry Andric          (Val == llvm::bit_cast<uint32_t>(0.5f)) ||
261406c3fb27SDimitry Andric          (Val == llvm::bit_cast<uint32_t>(-0.5f)) ||
261506c3fb27SDimitry Andric          (Val == llvm::bit_cast<uint32_t>(2.0f)) ||
261606c3fb27SDimitry Andric          (Val == llvm::bit_cast<uint32_t>(-2.0f)) ||
261706c3fb27SDimitry Andric          (Val == llvm::bit_cast<uint32_t>(4.0f)) ||
261806c3fb27SDimitry Andric          (Val == llvm::bit_cast<uint32_t>(-4.0f)) ||
26190b57cec5SDimitry Andric          (Val == 0x3e22f983 && HasInv2Pi);
26200b57cec5SDimitry Andric }
26210b57cec5SDimitry Andric 
isInlinableLiteral16(int16_t Literal,bool HasInv2Pi)26220b57cec5SDimitry Andric bool isInlinableLiteral16(int16_t Literal, bool HasInv2Pi) {
26230b57cec5SDimitry Andric   if (!HasInv2Pi)
26240b57cec5SDimitry Andric     return false;
26250b57cec5SDimitry Andric 
26265ffd83dbSDimitry Andric   if (isInlinableIntLiteral(Literal))
26270b57cec5SDimitry Andric     return true;
26280b57cec5SDimitry Andric 
26290b57cec5SDimitry Andric   uint16_t Val = static_cast<uint16_t>(Literal);
26300b57cec5SDimitry Andric   return Val == 0x3C00 || // 1.0
26310b57cec5SDimitry Andric          Val == 0xBC00 || // -1.0
26320b57cec5SDimitry Andric          Val == 0x3800 || // 0.5
26330b57cec5SDimitry Andric          Val == 0xB800 || // -0.5
26340b57cec5SDimitry Andric          Val == 0x4000 || // 2.0
26350b57cec5SDimitry Andric          Val == 0xC000 || // -2.0
26360b57cec5SDimitry Andric          Val == 0x4400 || // 4.0
26370b57cec5SDimitry Andric          Val == 0xC400 || // -4.0
26380b57cec5SDimitry Andric          Val == 0x3118;   // 1/2pi
26390b57cec5SDimitry Andric }
26400b57cec5SDimitry Andric 
getInlineEncodingV216(bool IsFloat,uint32_t Literal)26411db9f3b2SDimitry Andric std::optional<unsigned> getInlineEncodingV216(bool IsFloat, uint32_t Literal) {
26421db9f3b2SDimitry Andric   // Unfortunately, the Instruction Set Architecture Reference Guide is
26431db9f3b2SDimitry Andric   // misleading about how the inline operands work for (packed) 16-bit
26441db9f3b2SDimitry Andric   // instructions. In a nutshell, the actual HW behavior is:
26451db9f3b2SDimitry Andric   //
26461db9f3b2SDimitry Andric   //  - integer encodings (-16 .. 64) are always produced as sign-extended
26471db9f3b2SDimitry Andric   //    32-bit values
26481db9f3b2SDimitry Andric   //  - float encodings are produced as:
26491db9f3b2SDimitry Andric   //    - for F16 instructions: corresponding half-precision float values in
26501db9f3b2SDimitry Andric   //      the LSBs, 0 in the MSBs
26511db9f3b2SDimitry Andric   //    - for UI16 instructions: corresponding single-precision float value
26521db9f3b2SDimitry Andric   int32_t Signed = static_cast<int32_t>(Literal);
26531db9f3b2SDimitry Andric   if (Signed >= 0 && Signed <= 64)
26541db9f3b2SDimitry Andric     return 128 + Signed;
26550b57cec5SDimitry Andric 
26561db9f3b2SDimitry Andric   if (Signed >= -16 && Signed <= -1)
26571db9f3b2SDimitry Andric     return 192 + std::abs(Signed);
26581db9f3b2SDimitry Andric 
26591db9f3b2SDimitry Andric   if (IsFloat) {
26601db9f3b2SDimitry Andric     // clang-format off
26611db9f3b2SDimitry Andric     switch (Literal) {
26621db9f3b2SDimitry Andric     case 0x3800: return 240; // 0.5
26631db9f3b2SDimitry Andric     case 0xB800: return 241; // -0.5
26641db9f3b2SDimitry Andric     case 0x3C00: return 242; // 1.0
26651db9f3b2SDimitry Andric     case 0xBC00: return 243; // -1.0
26661db9f3b2SDimitry Andric     case 0x4000: return 244; // 2.0
26671db9f3b2SDimitry Andric     case 0xC000: return 245; // -2.0
26681db9f3b2SDimitry Andric     case 0x4400: return 246; // 4.0
26691db9f3b2SDimitry Andric     case 0xC400: return 247; // -4.0
26701db9f3b2SDimitry Andric     case 0x3118: return 248; // 1.0 / (2.0 * pi)
26711db9f3b2SDimitry Andric     default: break;
26720b57cec5SDimitry Andric     }
26731db9f3b2SDimitry Andric     // clang-format on
26741db9f3b2SDimitry Andric   } else {
26751db9f3b2SDimitry Andric     // clang-format off
26761db9f3b2SDimitry Andric     switch (Literal) {
26771db9f3b2SDimitry Andric     case 0x3F000000: return 240; // 0.5
26781db9f3b2SDimitry Andric     case 0xBF000000: return 241; // -0.5
26791db9f3b2SDimitry Andric     case 0x3F800000: return 242; // 1.0
26801db9f3b2SDimitry Andric     case 0xBF800000: return 243; // -1.0
26811db9f3b2SDimitry Andric     case 0x40000000: return 244; // 2.0
26821db9f3b2SDimitry Andric     case 0xC0000000: return 245; // -2.0
26831db9f3b2SDimitry Andric     case 0x40800000: return 246; // 4.0
26841db9f3b2SDimitry Andric     case 0xC0800000: return 247; // -4.0
26851db9f3b2SDimitry Andric     case 0x3E22F983: return 248; // 1.0 / (2.0 * pi)
26861db9f3b2SDimitry Andric     default: break;
26871db9f3b2SDimitry Andric     }
26881db9f3b2SDimitry Andric     // clang-format on
26890b57cec5SDimitry Andric   }
26900b57cec5SDimitry Andric 
26911db9f3b2SDimitry Andric   return {};
26925ffd83dbSDimitry Andric }
26935ffd83dbSDimitry Andric 
26941db9f3b2SDimitry Andric // Encoding of the literal as an inline constant for a V_PK_*_IU16 instruction
26951db9f3b2SDimitry Andric // or nullopt.
getInlineEncodingV2I16(uint32_t Literal)26961db9f3b2SDimitry Andric std::optional<unsigned> getInlineEncodingV2I16(uint32_t Literal) {
26971db9f3b2SDimitry Andric   return getInlineEncodingV216(false, Literal);
26981db9f3b2SDimitry Andric }
26991db9f3b2SDimitry Andric 
27001db9f3b2SDimitry Andric // Encoding of the literal as an inline constant for a V_PK_*_F16 instruction
27011db9f3b2SDimitry Andric // or nullopt.
getInlineEncodingV2F16(uint32_t Literal)27021db9f3b2SDimitry Andric std::optional<unsigned> getInlineEncodingV2F16(uint32_t Literal) {
27031db9f3b2SDimitry Andric   return getInlineEncodingV216(true, Literal);
27041db9f3b2SDimitry Andric }
27051db9f3b2SDimitry Andric 
27061db9f3b2SDimitry Andric // Whether the given literal can be inlined for a V_PK_* instruction.
isInlinableLiteralV216(uint32_t Literal,uint8_t OpType)27071db9f3b2SDimitry Andric bool isInlinableLiteralV216(uint32_t Literal, uint8_t OpType) {
27085f757f3fSDimitry Andric   switch (OpType) {
27091db9f3b2SDimitry Andric   case AMDGPU::OPERAND_REG_IMM_V2INT16:
27101db9f3b2SDimitry Andric   case AMDGPU::OPERAND_REG_INLINE_C_V2INT16:
27111db9f3b2SDimitry Andric   case AMDGPU::OPERAND_REG_INLINE_AC_V2INT16:
27121db9f3b2SDimitry Andric     return getInlineEncodingV216(false, Literal).has_value();
27135f757f3fSDimitry Andric   case AMDGPU::OPERAND_REG_IMM_V2FP16:
27145f757f3fSDimitry Andric   case AMDGPU::OPERAND_REG_INLINE_C_V2FP16:
27151db9f3b2SDimitry Andric   case AMDGPU::OPERAND_REG_INLINE_AC_V2FP16:
27161db9f3b2SDimitry Andric     return getInlineEncodingV216(true, Literal).has_value();
27175f757f3fSDimitry Andric   default:
27181db9f3b2SDimitry Andric     llvm_unreachable("bad packed operand type");
27195f757f3fSDimitry Andric   }
27205f757f3fSDimitry Andric }
27215f757f3fSDimitry Andric 
27221db9f3b2SDimitry Andric // Whether the given literal can be inlined for a V_PK_*_IU16 instruction.
isInlinableLiteralV2I16(uint32_t Literal)27231db9f3b2SDimitry Andric bool isInlinableLiteralV2I16(uint32_t Literal) {
27241db9f3b2SDimitry Andric   return getInlineEncodingV2I16(Literal).has_value();
27251db9f3b2SDimitry Andric }
272616d6b3b3SDimitry Andric 
27271db9f3b2SDimitry Andric // Whether the given literal can be inlined for a V_PK_*_F16 instruction.
isInlinableLiteralV2F16(uint32_t Literal)27281db9f3b2SDimitry Andric bool isInlinableLiteralV2F16(uint32_t Literal) {
27291db9f3b2SDimitry Andric   return getInlineEncodingV2F16(Literal).has_value();
273016d6b3b3SDimitry Andric }
273116d6b3b3SDimitry Andric 
isValid32BitLiteral(uint64_t Val,bool IsFP64)27325f757f3fSDimitry Andric bool isValid32BitLiteral(uint64_t Val, bool IsFP64) {
27335f757f3fSDimitry Andric   if (IsFP64)
27345f757f3fSDimitry Andric     return !(Val & 0xffffffffu);
27355f757f3fSDimitry Andric 
27365f757f3fSDimitry Andric   return isUInt<32>(Val) || isInt<32>(Val);
27375f757f3fSDimitry Andric }
27385f757f3fSDimitry Andric 
isArgPassedInSGPR(const Argument * A)27390b57cec5SDimitry Andric bool isArgPassedInSGPR(const Argument *A) {
27400b57cec5SDimitry Andric   const Function *F = A->getParent();
27410b57cec5SDimitry Andric 
27420b57cec5SDimitry Andric   // Arguments to compute shaders are never a source of divergence.
27430b57cec5SDimitry Andric   CallingConv::ID CC = F->getCallingConv();
27440b57cec5SDimitry Andric   switch (CC) {
27450b57cec5SDimitry Andric   case CallingConv::AMDGPU_KERNEL:
27460b57cec5SDimitry Andric   case CallingConv::SPIR_KERNEL:
27470b57cec5SDimitry Andric     return true;
27480b57cec5SDimitry Andric   case CallingConv::AMDGPU_VS:
27490b57cec5SDimitry Andric   case CallingConv::AMDGPU_LS:
27500b57cec5SDimitry Andric   case CallingConv::AMDGPU_HS:
27510b57cec5SDimitry Andric   case CallingConv::AMDGPU_ES:
27520b57cec5SDimitry Andric   case CallingConv::AMDGPU_GS:
27530b57cec5SDimitry Andric   case CallingConv::AMDGPU_PS:
27540b57cec5SDimitry Andric   case CallingConv::AMDGPU_CS:
2755e8d8bef9SDimitry Andric   case CallingConv::AMDGPU_Gfx:
27565f757f3fSDimitry Andric   case CallingConv::AMDGPU_CS_Chain:
27575f757f3fSDimitry Andric   case CallingConv::AMDGPU_CS_ChainPreserve:
275806c3fb27SDimitry Andric     // For non-compute shaders, SGPR inputs are marked with either inreg or
275906c3fb27SDimitry Andric     // byval. Everything else is in VGPRs.
276006c3fb27SDimitry Andric     return A->hasAttribute(Attribute::InReg) ||
276106c3fb27SDimitry Andric            A->hasAttribute(Attribute::ByVal);
276206c3fb27SDimitry Andric   default:
27635f757f3fSDimitry Andric     // TODO: treat i1 as divergent?
27645f757f3fSDimitry Andric     return A->hasAttribute(Attribute::InReg);
276506c3fb27SDimitry Andric   }
276606c3fb27SDimitry Andric }
276706c3fb27SDimitry Andric 
isArgPassedInSGPR(const CallBase * CB,unsigned ArgNo)276806c3fb27SDimitry Andric bool isArgPassedInSGPR(const CallBase *CB, unsigned ArgNo) {
276906c3fb27SDimitry Andric   // Arguments to compute shaders are never a source of divergence.
277006c3fb27SDimitry Andric   CallingConv::ID CC = CB->getCallingConv();
277106c3fb27SDimitry Andric   switch (CC) {
277206c3fb27SDimitry Andric   case CallingConv::AMDGPU_KERNEL:
277306c3fb27SDimitry Andric   case CallingConv::SPIR_KERNEL:
277406c3fb27SDimitry Andric     return true;
277506c3fb27SDimitry Andric   case CallingConv::AMDGPU_VS:
277606c3fb27SDimitry Andric   case CallingConv::AMDGPU_LS:
277706c3fb27SDimitry Andric   case CallingConv::AMDGPU_HS:
277806c3fb27SDimitry Andric   case CallingConv::AMDGPU_ES:
277906c3fb27SDimitry Andric   case CallingConv::AMDGPU_GS:
278006c3fb27SDimitry Andric   case CallingConv::AMDGPU_PS:
278106c3fb27SDimitry Andric   case CallingConv::AMDGPU_CS:
278206c3fb27SDimitry Andric   case CallingConv::AMDGPU_Gfx:
27835f757f3fSDimitry Andric   case CallingConv::AMDGPU_CS_Chain:
27845f757f3fSDimitry Andric   case CallingConv::AMDGPU_CS_ChainPreserve:
278506c3fb27SDimitry Andric     // For non-compute shaders, SGPR inputs are marked with either inreg or
278606c3fb27SDimitry Andric     // byval. Everything else is in VGPRs.
278706c3fb27SDimitry Andric     return CB->paramHasAttr(ArgNo, Attribute::InReg) ||
278806c3fb27SDimitry Andric            CB->paramHasAttr(ArgNo, Attribute::ByVal);
27890b57cec5SDimitry Andric   default:
27905f757f3fSDimitry Andric     return CB->paramHasAttr(ArgNo, Attribute::InReg);
27910b57cec5SDimitry Andric   }
27920b57cec5SDimitry Andric }
27930b57cec5SDimitry Andric 
hasSMEMByteOffset(const MCSubtargetInfo & ST)27940b57cec5SDimitry Andric static bool hasSMEMByteOffset(const MCSubtargetInfo &ST) {
2795e8d8bef9SDimitry Andric   return isGCN3Encoding(ST) || isGFX10Plus(ST);
27960b57cec5SDimitry Andric }
27970b57cec5SDimitry Andric 
hasSMRDSignedImmOffset(const MCSubtargetInfo & ST)27985ffd83dbSDimitry Andric static bool hasSMRDSignedImmOffset(const MCSubtargetInfo &ST) {
2799e8d8bef9SDimitry Andric   return isGFX9Plus(ST);
28005ffd83dbSDimitry Andric }
28015ffd83dbSDimitry Andric 
isLegalSMRDEncodedUnsignedOffset(const MCSubtargetInfo & ST,int64_t EncodedOffset)28025ffd83dbSDimitry Andric bool isLegalSMRDEncodedUnsignedOffset(const MCSubtargetInfo &ST,
28035ffd83dbSDimitry Andric                                       int64_t EncodedOffset) {
28045f757f3fSDimitry Andric   if (isGFX12Plus(ST))
28055f757f3fSDimitry Andric     return isUInt<23>(EncodedOffset);
28065f757f3fSDimitry Andric 
28075ffd83dbSDimitry Andric   return hasSMEMByteOffset(ST) ? isUInt<20>(EncodedOffset)
28085ffd83dbSDimitry Andric                                : isUInt<8>(EncodedOffset);
28095ffd83dbSDimitry Andric }
28105ffd83dbSDimitry Andric 
isLegalSMRDEncodedSignedOffset(const MCSubtargetInfo & ST,int64_t EncodedOffset,bool IsBuffer)28115ffd83dbSDimitry Andric bool isLegalSMRDEncodedSignedOffset(const MCSubtargetInfo &ST,
28125ffd83dbSDimitry Andric                                     int64_t EncodedOffset,
28135ffd83dbSDimitry Andric                                     bool IsBuffer) {
28145f757f3fSDimitry Andric   if (isGFX12Plus(ST))
28155f757f3fSDimitry Andric     return isInt<24>(EncodedOffset);
28165f757f3fSDimitry Andric 
28175ffd83dbSDimitry Andric   return !IsBuffer &&
28185ffd83dbSDimitry Andric          hasSMRDSignedImmOffset(ST) &&
28195ffd83dbSDimitry Andric          isInt<21>(EncodedOffset);
28205ffd83dbSDimitry Andric }
28215ffd83dbSDimitry Andric 
isDwordAligned(uint64_t ByteOffset)28225ffd83dbSDimitry Andric static bool isDwordAligned(uint64_t ByteOffset) {
28235ffd83dbSDimitry Andric   return (ByteOffset & 3) == 0;
28245ffd83dbSDimitry Andric }
28255ffd83dbSDimitry Andric 
convertSMRDOffsetUnits(const MCSubtargetInfo & ST,uint64_t ByteOffset)28265ffd83dbSDimitry Andric uint64_t convertSMRDOffsetUnits(const MCSubtargetInfo &ST,
28275ffd83dbSDimitry Andric                                 uint64_t ByteOffset) {
28280b57cec5SDimitry Andric   if (hasSMEMByteOffset(ST))
28290b57cec5SDimitry Andric     return ByteOffset;
28305ffd83dbSDimitry Andric 
28315ffd83dbSDimitry Andric   assert(isDwordAligned(ByteOffset));
28320b57cec5SDimitry Andric   return ByteOffset >> 2;
28330b57cec5SDimitry Andric }
28340b57cec5SDimitry Andric 
getSMRDEncodedOffset(const MCSubtargetInfo & ST,int64_t ByteOffset,bool IsBuffer)2835bdd1243dSDimitry Andric std::optional<int64_t> getSMRDEncodedOffset(const MCSubtargetInfo &ST,
28365ffd83dbSDimitry Andric                                             int64_t ByteOffset, bool IsBuffer) {
28375f757f3fSDimitry Andric   if (isGFX12Plus(ST)) // 24 bit signed offsets
28385f757f3fSDimitry Andric     return isInt<24>(ByteOffset) ? std::optional<int64_t>(ByteOffset)
28395f757f3fSDimitry Andric                                  : std::nullopt;
28405f757f3fSDimitry Andric 
28415ffd83dbSDimitry Andric   // The signed version is always a byte offset.
28425ffd83dbSDimitry Andric   if (!IsBuffer && hasSMRDSignedImmOffset(ST)) {
28435ffd83dbSDimitry Andric     assert(hasSMEMByteOffset(ST));
2844bdd1243dSDimitry Andric     return isInt<20>(ByteOffset) ? std::optional<int64_t>(ByteOffset)
2845bdd1243dSDimitry Andric                                  : std::nullopt;
28465ffd83dbSDimitry Andric   }
28475ffd83dbSDimitry Andric 
28485ffd83dbSDimitry Andric   if (!isDwordAligned(ByteOffset) && !hasSMEMByteOffset(ST))
2849bdd1243dSDimitry Andric     return std::nullopt;
28505ffd83dbSDimitry Andric 
28515ffd83dbSDimitry Andric   int64_t EncodedOffset = convertSMRDOffsetUnits(ST, ByteOffset);
28525ffd83dbSDimitry Andric   return isLegalSMRDEncodedUnsignedOffset(ST, EncodedOffset)
2853bdd1243dSDimitry Andric              ? std::optional<int64_t>(EncodedOffset)
2854bdd1243dSDimitry Andric              : std::nullopt;
28555ffd83dbSDimitry Andric }
28565ffd83dbSDimitry Andric 
getSMRDEncodedLiteralOffset32(const MCSubtargetInfo & ST,int64_t ByteOffset)2857bdd1243dSDimitry Andric std::optional<int64_t> getSMRDEncodedLiteralOffset32(const MCSubtargetInfo &ST,
28585ffd83dbSDimitry Andric                                                      int64_t ByteOffset) {
28595ffd83dbSDimitry Andric   if (!isCI(ST) || !isDwordAligned(ByteOffset))
2860bdd1243dSDimitry Andric     return std::nullopt;
28615ffd83dbSDimitry Andric 
28625ffd83dbSDimitry Andric   int64_t EncodedOffset = convertSMRDOffsetUnits(ST, ByteOffset);
2863bdd1243dSDimitry Andric   return isUInt<32>(EncodedOffset) ? std::optional<int64_t>(EncodedOffset)
2864bdd1243dSDimitry Andric                                    : std::nullopt;
28650b57cec5SDimitry Andric }
28660b57cec5SDimitry Andric 
getNumFlatOffsetBits(const MCSubtargetInfo & ST)2867bdd1243dSDimitry Andric unsigned getNumFlatOffsetBits(const MCSubtargetInfo &ST) {
2868e8d8bef9SDimitry Andric   if (AMDGPU::isGFX10(ST))
2869bdd1243dSDimitry Andric     return 12;
2870e8d8bef9SDimitry Andric 
28715f757f3fSDimitry Andric   if (AMDGPU::isGFX12(ST))
28725f757f3fSDimitry Andric     return 24;
2873bdd1243dSDimitry Andric   return 13;
2874e8d8bef9SDimitry Andric }
2875e8d8bef9SDimitry Andric 
28760b57cec5SDimitry Andric namespace {
28770b57cec5SDimitry Andric 
28780b57cec5SDimitry Andric struct SourceOfDivergence {
28790b57cec5SDimitry Andric   unsigned Intr;
28800b57cec5SDimitry Andric };
28810b57cec5SDimitry Andric const SourceOfDivergence *lookupSourceOfDivergence(unsigned Intr);
28820b57cec5SDimitry Andric 
288306c3fb27SDimitry Andric struct AlwaysUniform {
288406c3fb27SDimitry Andric   unsigned Intr;
288506c3fb27SDimitry Andric };
288606c3fb27SDimitry Andric const AlwaysUniform *lookupAlwaysUniform(unsigned Intr);
288706c3fb27SDimitry Andric 
28880b57cec5SDimitry Andric #define GET_SourcesOfDivergence_IMPL
288906c3fb27SDimitry Andric #define GET_UniformIntrinsics_IMPL
2890480093f4SDimitry Andric #define GET_Gfx9BufferFormat_IMPL
289181ad6265SDimitry Andric #define GET_Gfx10BufferFormat_IMPL
289281ad6265SDimitry Andric #define GET_Gfx11PlusBufferFormat_IMPL
28930b57cec5SDimitry Andric #include "AMDGPUGenSearchableTables.inc"
28940b57cec5SDimitry Andric 
28950b57cec5SDimitry Andric } // end anonymous namespace
28960b57cec5SDimitry Andric 
isIntrinsicSourceOfDivergence(unsigned IntrID)28970b57cec5SDimitry Andric bool isIntrinsicSourceOfDivergence(unsigned IntrID) {
28980b57cec5SDimitry Andric   return lookupSourceOfDivergence(IntrID);
28990b57cec5SDimitry Andric }
29000b57cec5SDimitry Andric 
isIntrinsicAlwaysUniform(unsigned IntrID)290106c3fb27SDimitry Andric bool isIntrinsicAlwaysUniform(unsigned IntrID) {
290206c3fb27SDimitry Andric   return lookupAlwaysUniform(IntrID);
290306c3fb27SDimitry Andric }
290406c3fb27SDimitry Andric 
getGcnBufferFormatInfo(uint8_t BitsPerComp,uint8_t NumComponents,uint8_t NumFormat,const MCSubtargetInfo & STI)2905480093f4SDimitry Andric const GcnBufferFormatInfo *getGcnBufferFormatInfo(uint8_t BitsPerComp,
2906480093f4SDimitry Andric                                                   uint8_t NumComponents,
2907480093f4SDimitry Andric                                                   uint8_t NumFormat,
2908480093f4SDimitry Andric                                                   const MCSubtargetInfo &STI) {
290981ad6265SDimitry Andric   return isGFX11Plus(STI)
291081ad6265SDimitry Andric              ? getGfx11PlusBufferFormatInfo(BitsPerComp, NumComponents,
2911480093f4SDimitry Andric                                             NumFormat)
291281ad6265SDimitry Andric              : isGFX10(STI) ? getGfx10BufferFormatInfo(BitsPerComp,
291381ad6265SDimitry Andric                                                        NumComponents, NumFormat)
291481ad6265SDimitry Andric                             : getGfx9BufferFormatInfo(BitsPerComp,
291581ad6265SDimitry Andric                                                       NumComponents, NumFormat);
2916480093f4SDimitry Andric }
2917480093f4SDimitry Andric 
getGcnBufferFormatInfo(uint8_t Format,const MCSubtargetInfo & STI)2918480093f4SDimitry Andric const GcnBufferFormatInfo *getGcnBufferFormatInfo(uint8_t Format,
2919480093f4SDimitry Andric                                                   const MCSubtargetInfo &STI) {
292081ad6265SDimitry Andric   return isGFX11Plus(STI) ? getGfx11PlusBufferFormatInfo(Format)
292181ad6265SDimitry Andric                           : isGFX10(STI) ? getGfx10BufferFormatInfo(Format)
2922480093f4SDimitry Andric                                          : getGfx9BufferFormatInfo(Format);
2923480093f4SDimitry Andric }
2924480093f4SDimitry Andric 
hasAny64BitVGPROperands(const MCInstrDesc & OpDesc)29255f757f3fSDimitry Andric bool hasAny64BitVGPROperands(const MCInstrDesc &OpDesc) {
29265f757f3fSDimitry Andric   for (auto OpName : { OpName::vdst, OpName::src0, OpName::src1,
29275f757f3fSDimitry Andric                        OpName::src2 }) {
29285f757f3fSDimitry Andric     int Idx = getNamedOperandIdx(OpDesc.getOpcode(), OpName);
29295f757f3fSDimitry Andric     if (Idx == -1)
29305f757f3fSDimitry Andric       continue;
29315f757f3fSDimitry Andric 
29325f757f3fSDimitry Andric     if (OpDesc.operands()[Idx].RegClass == AMDGPU::VReg_64RegClassID ||
29335f757f3fSDimitry Andric         OpDesc.operands()[Idx].RegClass == AMDGPU::VReg_64_Align2RegClassID)
29345f757f3fSDimitry Andric       return true;
29355f757f3fSDimitry Andric   }
29365f757f3fSDimitry Andric 
29375f757f3fSDimitry Andric   return false;
29385f757f3fSDimitry Andric }
29395f757f3fSDimitry Andric 
isDPALU_DPP(const MCInstrDesc & OpDesc)29405f757f3fSDimitry Andric bool isDPALU_DPP(const MCInstrDesc &OpDesc) {
29415f757f3fSDimitry Andric   return hasAny64BitVGPROperands(OpDesc);
29425f757f3fSDimitry Andric }
29435f757f3fSDimitry Andric 
29440b57cec5SDimitry Andric } // namespace AMDGPU
2945e8d8bef9SDimitry Andric 
operator <<(raw_ostream & OS,const AMDGPU::IsaInfo::TargetIDSetting S)2946e8d8bef9SDimitry Andric raw_ostream &operator<<(raw_ostream &OS,
2947e8d8bef9SDimitry Andric                         const AMDGPU::IsaInfo::TargetIDSetting S) {
2948e8d8bef9SDimitry Andric   switch (S) {
2949e8d8bef9SDimitry Andric   case (AMDGPU::IsaInfo::TargetIDSetting::Unsupported):
2950e8d8bef9SDimitry Andric     OS << "Unsupported";
2951e8d8bef9SDimitry Andric     break;
2952e8d8bef9SDimitry Andric   case (AMDGPU::IsaInfo::TargetIDSetting::Any):
2953e8d8bef9SDimitry Andric     OS << "Any";
2954e8d8bef9SDimitry Andric     break;
2955e8d8bef9SDimitry Andric   case (AMDGPU::IsaInfo::TargetIDSetting::Off):
2956e8d8bef9SDimitry Andric     OS << "Off";
2957e8d8bef9SDimitry Andric     break;
2958e8d8bef9SDimitry Andric   case (AMDGPU::IsaInfo::TargetIDSetting::On):
2959e8d8bef9SDimitry Andric     OS << "On";
2960e8d8bef9SDimitry Andric     break;
2961e8d8bef9SDimitry Andric   }
2962e8d8bef9SDimitry Andric   return OS;
2963e8d8bef9SDimitry Andric }
2964e8d8bef9SDimitry Andric 
29650b57cec5SDimitry Andric } // namespace llvm
2966