1c8fbf6ffSEugene Zelenko //===- AMDGPUDisassembler.cpp - Disassembler for AMDGPU ISA ---------------===//
2e1818af8STom Stellard //
32946cd70SChandler Carruth // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
42946cd70SChandler Carruth // See https://llvm.org/LICENSE.txt for license information.
52946cd70SChandler Carruth // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
6e1818af8STom Stellard //
7e1818af8STom Stellard //===----------------------------------------------------------------------===//
8e1818af8STom Stellard //
9e1818af8STom Stellard //===----------------------------------------------------------------------===//
10e1818af8STom Stellard //
11e1818af8STom Stellard /// \file
12e1818af8STom Stellard ///
13e1818af8STom Stellard /// This file contains definition for AMDGPU ISA disassembler
14e1818af8STom Stellard //
15e1818af8STom Stellard //===----------------------------------------------------------------------===//
16e1818af8STom Stellard 
17e1818af8STom Stellard // ToDo: What to do with instruction suffixes (v_mov_b32 vs v_mov_b32_e32)?
18e1818af8STom Stellard 
19c8fbf6ffSEugene Zelenko #include "Disassembler/AMDGPUDisassembler.h"
20c5a154dbSTom Stellard #include "MCTargetDesc/AMDGPUMCTargetDesc.h"
21e8860beeSJoe Nash #include "SIDefines.h"
22e8860beeSJoe Nash #include "SIRegisterInfo.h"
238ce2ee9dSRichard Trieu #include "TargetInfo/AMDGPUTargetInfo.h"
24e1818af8STom Stellard #include "Utils/AMDGPUBaseInfo.h"
256a87e9b0Sdfukalov #include "llvm-c/DisassemblerTypes.h"
26ef736a1cSserge-sans-paille #include "llvm/BinaryFormat/ELF.h"
27ca64ef20SMatt Arsenault #include "llvm/MC/MCAsmInfo.h"
28ac106addSNikolay Haustov #include "llvm/MC/MCContext.h"
29c644488aSSheng #include "llvm/MC/MCDecoderOps.h"
30c8fbf6ffSEugene Zelenko #include "llvm/MC/MCExpr.h"
31b4b7e605SJoe Nash #include "llvm/MC/MCInstrDesc.h"
32ef736a1cSserge-sans-paille #include "llvm/MC/MCRegisterInfo.h"
33ef736a1cSserge-sans-paille #include "llvm/MC/MCSubtargetInfo.h"
34ef736a1cSserge-sans-paille #include "llvm/MC/TargetRegistry.h"
35528057c1SRonak Chauhan #include "llvm/Support/AMDHSAKernelDescriptor.h"
36e1818af8STom Stellard 
37e1818af8STom Stellard using namespace llvm;
38e1818af8STom Stellard 
39e1818af8STom Stellard #define DEBUG_TYPE "amdgpu-disassembler"
40e1818af8STom Stellard 
414f87d30aSJay Foad #define SGPR_MAX                                                               \
424f87d30aSJay Foad   (isGFX10Plus() ? AMDGPU::EncValues::SGPR_MAX_GFX10                           \
4333d806a5SStanislav Mekhanoshin                  : AMDGPU::EncValues::SGPR_MAX_SI)
4433d806a5SStanislav Mekhanoshin 
45c8fbf6ffSEugene Zelenko using DecodeStatus = llvm::MCDisassembler::DecodeStatus;
46e1818af8STom Stellard 
AMDGPUDisassembler(const MCSubtargetInfo & STI,MCContext & Ctx,MCInstrInfo const * MCII)47ca64ef20SMatt Arsenault AMDGPUDisassembler::AMDGPUDisassembler(const MCSubtargetInfo &STI,
48ca64ef20SMatt Arsenault                                        MCContext &Ctx,
49ca64ef20SMatt Arsenault                                        MCInstrInfo const *MCII) :
50ca64ef20SMatt Arsenault   MCDisassembler(STI, Ctx), MCII(MCII), MRI(*Ctx.getRegisterInfo()),
51418e23e3SMatt Arsenault   TargetMaxInstBytes(Ctx.getAsmInfo()->getMaxInstLength(&STI)) {
52418e23e3SMatt Arsenault 
53418e23e3SMatt Arsenault   // ToDo: AMDGPUDisassembler supports only VI ISA.
544f87d30aSJay Foad   if (!STI.getFeatureBits()[AMDGPU::FeatureGCN3Encoding] && !isGFX10Plus())
55418e23e3SMatt Arsenault     report_fatal_error("Disassembly not yet supported for subtarget");
56418e23e3SMatt Arsenault }
57ca64ef20SMatt Arsenault 
58ac106addSNikolay Haustov inline static MCDisassembler::DecodeStatus
addOperand(MCInst & Inst,const MCOperand & Opnd)59ac106addSNikolay Haustov addOperand(MCInst &Inst, const MCOperand& Opnd) {
60ac106addSNikolay Haustov   Inst.addOperand(Opnd);
61ac106addSNikolay Haustov   return Opnd.isValid() ?
62ac106addSNikolay Haustov     MCDisassembler::Success :
63de56a890SStanislav Mekhanoshin     MCDisassembler::Fail;
64e1818af8STom Stellard }
65e1818af8STom Stellard 
insertNamedMCOperand(MCInst & MI,const MCOperand & Op,uint16_t NameIdx)66549c89d2SSam Kolton static int insertNamedMCOperand(MCInst &MI, const MCOperand &Op,
67549c89d2SSam Kolton                                 uint16_t NameIdx) {
68549c89d2SSam Kolton   int OpIdx = AMDGPU::getNamedOperandIdx(MI.getOpcode(), NameIdx);
69549c89d2SSam Kolton   if (OpIdx != -1) {
70549c89d2SSam Kolton     auto I = MI.begin();
71549c89d2SSam Kolton     std::advance(I, OpIdx);
72549c89d2SSam Kolton     MI.insert(I, Op);
73549c89d2SSam Kolton   }
74549c89d2SSam Kolton   return OpIdx;
75549c89d2SSam Kolton }
76549c89d2SSam Kolton 
decodeSoppBrTarget(MCInst & Inst,unsigned Imm,uint64_t Addr,const MCDisassembler * Decoder)773381d7a2SSam Kolton static DecodeStatus decodeSoppBrTarget(MCInst &Inst, unsigned Imm,
784ae9745aSMaksim Panchenko                                        uint64_t Addr,
794ae9745aSMaksim Panchenko                                        const MCDisassembler *Decoder) {
803381d7a2SSam Kolton   auto DAsm = static_cast<const AMDGPUDisassembler*>(Decoder);
813381d7a2SSam Kolton 
82efec1396SScott Linder   // Our branches take a simm16, but we need two extra bits to account for the
83efec1396SScott Linder   // factor of 4.
843381d7a2SSam Kolton   APInt SignedOffset(18, Imm * 4, true);
853381d7a2SSam Kolton   int64_t Offset = (SignedOffset.sext(64) + 4 + Addr).getSExtValue();
863381d7a2SSam Kolton 
87bed9efedSMaksim Panchenko   if (DAsm->tryAddingSymbolicOperand(Inst, Offset, Addr, true, 2, 2, 0))
883381d7a2SSam Kolton     return MCDisassembler::Success;
893381d7a2SSam Kolton   return addOperand(Inst, MCOperand::createImm(Imm));
903381d7a2SSam Kolton }
913381d7a2SSam Kolton 
decodeSMEMOffset(MCInst & Inst,unsigned Imm,uint64_t Addr,const MCDisassembler * Decoder)924ae9745aSMaksim Panchenko static DecodeStatus decodeSMEMOffset(MCInst &Inst, unsigned Imm, uint64_t Addr,
934ae9745aSMaksim Panchenko                                      const MCDisassembler *Decoder) {
945998baccSDmitry Preobrazhensky   auto DAsm = static_cast<const AMDGPUDisassembler*>(Decoder);
955998baccSDmitry Preobrazhensky   int64_t Offset;
965998baccSDmitry Preobrazhensky   if (DAsm->isVI()) {         // VI supports 20-bit unsigned offsets.
975998baccSDmitry Preobrazhensky     Offset = Imm & 0xFFFFF;
985998baccSDmitry Preobrazhensky   } else {                    // GFX9+ supports 21-bit signed offsets.
995998baccSDmitry Preobrazhensky     Offset = SignExtend64<21>(Imm);
1005998baccSDmitry Preobrazhensky   }
1015998baccSDmitry Preobrazhensky   return addOperand(Inst, MCOperand::createImm(Offset));
1025998baccSDmitry Preobrazhensky }
1035998baccSDmitry Preobrazhensky 
decodeBoolReg(MCInst & Inst,unsigned Val,uint64_t Addr,const MCDisassembler * Decoder)1044ae9745aSMaksim Panchenko static DecodeStatus decodeBoolReg(MCInst &Inst, unsigned Val, uint64_t Addr,
1054ae9745aSMaksim Panchenko                                   const MCDisassembler *Decoder) {
1060846c125SStanislav Mekhanoshin   auto DAsm = static_cast<const AMDGPUDisassembler*>(Decoder);
1070846c125SStanislav Mekhanoshin   return addOperand(Inst, DAsm->decodeBoolReg(Val));
1080846c125SStanislav Mekhanoshin }
1090846c125SStanislav Mekhanoshin 
110363f47a2SSam Kolton #define DECODE_OPERAND(StaticDecoderName, DecoderName)                         \
1114ae9745aSMaksim Panchenko   static DecodeStatus StaticDecoderName(MCInst &Inst, unsigned Imm,            \
112ac106addSNikolay Haustov                                         uint64_t /*Addr*/,                     \
1134ae9745aSMaksim Panchenko                                         const MCDisassembler *Decoder) {       \
114ac106addSNikolay Haustov     auto DAsm = static_cast<const AMDGPUDisassembler *>(Decoder);              \
115363f47a2SSam Kolton     return addOperand(Inst, DAsm->DecoderName(Imm));                           \
116e1818af8STom Stellard   }
117e1818af8STom Stellard 
118363f47a2SSam Kolton #define DECODE_OPERAND_REG(RegClass) \
119363f47a2SSam Kolton DECODE_OPERAND(Decode##RegClass##RegisterClass, decodeOperand_##RegClass)
120e1818af8STom Stellard 
121363f47a2SSam Kolton DECODE_OPERAND_REG(VGPR_32)
DECODE_OPERAND_REG(VRegOrLds_32)1226023d599SDmitry Preobrazhensky DECODE_OPERAND_REG(VRegOrLds_32)
123363f47a2SSam Kolton DECODE_OPERAND_REG(VS_32)
124363f47a2SSam Kolton DECODE_OPERAND_REG(VS_64)
12530fc5239SDmitry Preobrazhensky DECODE_OPERAND_REG(VS_128)
126e1818af8STom Stellard 
127363f47a2SSam Kolton DECODE_OPERAND_REG(VReg_64)
128363f47a2SSam Kolton DECODE_OPERAND_REG(VReg_96)
129363f47a2SSam Kolton DECODE_OPERAND_REG(VReg_128)
13091f503c3SStanislav Mekhanoshin DECODE_OPERAND_REG(VReg_256)
13191f503c3SStanislav Mekhanoshin DECODE_OPERAND_REG(VReg_512)
132a8d9d507SStanislav Mekhanoshin DECODE_OPERAND_REG(VReg_1024)
133e1818af8STom Stellard 
134363f47a2SSam Kolton DECODE_OPERAND_REG(SReg_32)
135363f47a2SSam Kolton DECODE_OPERAND_REG(SReg_32_XM0_XEXEC)
136ca7b0a17SMatt Arsenault DECODE_OPERAND_REG(SReg_32_XEXEC_HI)
1376023d599SDmitry Preobrazhensky DECODE_OPERAND_REG(SRegOrLds_32)
138363f47a2SSam Kolton DECODE_OPERAND_REG(SReg_64)
139363f47a2SSam Kolton DECODE_OPERAND_REG(SReg_64_XEXEC)
140363f47a2SSam Kolton DECODE_OPERAND_REG(SReg_128)
141363f47a2SSam Kolton DECODE_OPERAND_REG(SReg_256)
142363f47a2SSam Kolton DECODE_OPERAND_REG(SReg_512)
143e1818af8STom Stellard 
14450d7f464SStanislav Mekhanoshin DECODE_OPERAND_REG(AGPR_32)
145a8d9d507SStanislav Mekhanoshin DECODE_OPERAND_REG(AReg_64)
14650d7f464SStanislav Mekhanoshin DECODE_OPERAND_REG(AReg_128)
147a8d9d507SStanislav Mekhanoshin DECODE_OPERAND_REG(AReg_256)
14850d7f464SStanislav Mekhanoshin DECODE_OPERAND_REG(AReg_512)
14950d7f464SStanislav Mekhanoshin DECODE_OPERAND_REG(AReg_1024)
15050d7f464SStanislav Mekhanoshin DECODE_OPERAND_REG(AV_32)
15150d7f464SStanislav Mekhanoshin DECODE_OPERAND_REG(AV_64)
1526e3e14f6SStanislav Mekhanoshin DECODE_OPERAND_REG(AV_128)
15332ca9bd7SDmitry Preobrazhensky DECODE_OPERAND_REG(AVDst_128)
15432ca9bd7SDmitry Preobrazhensky DECODE_OPERAND_REG(AVDst_512)
15550d7f464SStanislav Mekhanoshin 
1564ae9745aSMaksim Panchenko static DecodeStatus decodeOperand_VSrc16(MCInst &Inst, unsigned Imm,
1574bd72361SMatt Arsenault                                          uint64_t Addr,
1584ae9745aSMaksim Panchenko                                          const MCDisassembler *Decoder) {
1594bd72361SMatt Arsenault   auto DAsm = static_cast<const AMDGPUDisassembler*>(Decoder);
1604bd72361SMatt Arsenault   return addOperand(Inst, DAsm->decodeOperand_VSrc16(Imm));
1614bd72361SMatt Arsenault }
1624bd72361SMatt Arsenault 
decodeOperand_VSrcV216(MCInst & Inst,unsigned Imm,uint64_t Addr,const MCDisassembler * Decoder)1634ae9745aSMaksim Panchenko static DecodeStatus decodeOperand_VSrcV216(MCInst &Inst, unsigned Imm,
1649be7b0d4SMatt Arsenault                                            uint64_t Addr,
1654ae9745aSMaksim Panchenko                                            const MCDisassembler *Decoder) {
1669be7b0d4SMatt Arsenault   auto DAsm = static_cast<const AMDGPUDisassembler*>(Decoder);
1679be7b0d4SMatt Arsenault   return addOperand(Inst, DAsm->decodeOperand_VSrcV216(Imm));
1689be7b0d4SMatt Arsenault }
1699be7b0d4SMatt Arsenault 
decodeOperand_VSrcV232(MCInst & Inst,unsigned Imm,uint64_t Addr,const MCDisassembler * Decoder)1704ae9745aSMaksim Panchenko static DecodeStatus decodeOperand_VSrcV232(MCInst &Inst, unsigned Imm,
171a8d9d507SStanislav Mekhanoshin                                            uint64_t Addr,
1724ae9745aSMaksim Panchenko                                            const MCDisassembler *Decoder) {
173a8d9d507SStanislav Mekhanoshin   auto DAsm = static_cast<const AMDGPUDisassembler*>(Decoder);
174a8d9d507SStanislav Mekhanoshin   return addOperand(Inst, DAsm->decodeOperand_VSrcV232(Imm));
175a8d9d507SStanislav Mekhanoshin }
176a8d9d507SStanislav Mekhanoshin 
decodeOperand_VS_16(MCInst & Inst,unsigned Imm,uint64_t Addr,const MCDisassembler * Decoder)1774ae9745aSMaksim Panchenko static DecodeStatus decodeOperand_VS_16(MCInst &Inst, unsigned Imm,
1789e77d0c6SStanislav Mekhanoshin                                         uint64_t Addr,
1794ae9745aSMaksim Panchenko                                         const MCDisassembler *Decoder) {
1809e77d0c6SStanislav Mekhanoshin   auto DAsm = static_cast<const AMDGPUDisassembler*>(Decoder);
1819e77d0c6SStanislav Mekhanoshin   return addOperand(Inst, DAsm->decodeOperand_VSrc16(Imm));
1829e77d0c6SStanislav Mekhanoshin }
1839e77d0c6SStanislav Mekhanoshin 
decodeOperand_VS_32(MCInst & Inst,unsigned Imm,uint64_t Addr,const MCDisassembler * Decoder)1844ae9745aSMaksim Panchenko static DecodeStatus decodeOperand_VS_32(MCInst &Inst, unsigned Imm,
1859e77d0c6SStanislav Mekhanoshin                                         uint64_t Addr,
1864ae9745aSMaksim Panchenko                                         const MCDisassembler *Decoder) {
1879e77d0c6SStanislav Mekhanoshin   auto DAsm = static_cast<const AMDGPUDisassembler*>(Decoder);
1889e77d0c6SStanislav Mekhanoshin   return addOperand(Inst, DAsm->decodeOperand_VS_32(Imm));
1899e77d0c6SStanislav Mekhanoshin }
1909e77d0c6SStanislav Mekhanoshin 
decodeOperand_AReg_64(MCInst & Inst,unsigned Imm,uint64_t Addr,const MCDisassembler * Decoder)1914ae9745aSMaksim Panchenko static DecodeStatus decodeOperand_AReg_64(MCInst &Inst, unsigned Imm,
192a8d9d507SStanislav Mekhanoshin                                           uint64_t Addr,
1934ae9745aSMaksim Panchenko                                           const MCDisassembler *Decoder) {
194a8d9d507SStanislav Mekhanoshin   auto DAsm = static_cast<const AMDGPUDisassembler*>(Decoder);
195a8d9d507SStanislav Mekhanoshin   return addOperand(Inst, DAsm->decodeSrcOp(AMDGPUDisassembler::OPW64, Imm | 512));
196a8d9d507SStanislav Mekhanoshin }
197a8d9d507SStanislav Mekhanoshin 
decodeOperand_AReg_128(MCInst & Inst,unsigned Imm,uint64_t Addr,const MCDisassembler * Decoder)1984ae9745aSMaksim Panchenko static DecodeStatus decodeOperand_AReg_128(MCInst &Inst, unsigned Imm,
19950d7f464SStanislav Mekhanoshin                                            uint64_t Addr,
2004ae9745aSMaksim Panchenko                                            const MCDisassembler *Decoder) {
20150d7f464SStanislav Mekhanoshin   auto DAsm = static_cast<const AMDGPUDisassembler*>(Decoder);
20250d7f464SStanislav Mekhanoshin   return addOperand(Inst, DAsm->decodeSrcOp(AMDGPUDisassembler::OPW128, Imm | 512));
20350d7f464SStanislav Mekhanoshin }
20450d7f464SStanislav Mekhanoshin 
decodeOperand_AReg_256(MCInst & Inst,unsigned Imm,uint64_t Addr,const MCDisassembler * Decoder)2054ae9745aSMaksim Panchenko static DecodeStatus decodeOperand_AReg_256(MCInst &Inst, unsigned Imm,
206a8d9d507SStanislav Mekhanoshin                                            uint64_t Addr,
2074ae9745aSMaksim Panchenko                                            const MCDisassembler *Decoder) {
208a8d9d507SStanislav Mekhanoshin   auto DAsm = static_cast<const AMDGPUDisassembler*>(Decoder);
209a8d9d507SStanislav Mekhanoshin   return addOperand(Inst, DAsm->decodeSrcOp(AMDGPUDisassembler::OPW256, Imm | 512));
210a8d9d507SStanislav Mekhanoshin }
211a8d9d507SStanislav Mekhanoshin 
decodeOperand_AReg_512(MCInst & Inst,unsigned Imm,uint64_t Addr,const MCDisassembler * Decoder)2124ae9745aSMaksim Panchenko static DecodeStatus decodeOperand_AReg_512(MCInst &Inst, unsigned Imm,
21350d7f464SStanislav Mekhanoshin                                            uint64_t Addr,
2144ae9745aSMaksim Panchenko                                            const MCDisassembler *Decoder) {
21550d7f464SStanislav Mekhanoshin   auto DAsm = static_cast<const AMDGPUDisassembler*>(Decoder);
21650d7f464SStanislav Mekhanoshin   return addOperand(Inst, DAsm->decodeSrcOp(AMDGPUDisassembler::OPW512, Imm | 512));
21750d7f464SStanislav Mekhanoshin }
21850d7f464SStanislav Mekhanoshin 
decodeOperand_AReg_1024(MCInst & Inst,unsigned Imm,uint64_t Addr,const MCDisassembler * Decoder)2194ae9745aSMaksim Panchenko static DecodeStatus decodeOperand_AReg_1024(MCInst &Inst, unsigned Imm,
22050d7f464SStanislav Mekhanoshin                                             uint64_t Addr,
2214ae9745aSMaksim Panchenko                                             const MCDisassembler *Decoder) {
22250d7f464SStanislav Mekhanoshin   auto DAsm = static_cast<const AMDGPUDisassembler*>(Decoder);
22350d7f464SStanislav Mekhanoshin   return addOperand(Inst, DAsm->decodeSrcOp(AMDGPUDisassembler::OPW1024, Imm | 512));
22450d7f464SStanislav Mekhanoshin }
22550d7f464SStanislav Mekhanoshin 
decodeOperand_VReg_64(MCInst & Inst,unsigned Imm,uint64_t Addr,const MCDisassembler * Decoder)2264ae9745aSMaksim Panchenko static DecodeStatus decodeOperand_VReg_64(MCInst &Inst, unsigned Imm,
227a8d9d507SStanislav Mekhanoshin                                           uint64_t Addr,
2284ae9745aSMaksim Panchenko                                           const MCDisassembler *Decoder) {
229a8d9d507SStanislav Mekhanoshin   auto DAsm = static_cast<const AMDGPUDisassembler*>(Decoder);
230a8d9d507SStanislav Mekhanoshin   return addOperand(Inst, DAsm->decodeSrcOp(AMDGPUDisassembler::OPW64, Imm));
231a8d9d507SStanislav Mekhanoshin }
232a8d9d507SStanislav Mekhanoshin 
decodeOperand_VReg_128(MCInst & Inst,unsigned Imm,uint64_t Addr,const MCDisassembler * Decoder)2334ae9745aSMaksim Panchenko static DecodeStatus decodeOperand_VReg_128(MCInst &Inst, unsigned Imm,
234a8d9d507SStanislav Mekhanoshin                                            uint64_t Addr,
2354ae9745aSMaksim Panchenko                                            const MCDisassembler *Decoder) {
236a8d9d507SStanislav Mekhanoshin   auto DAsm = static_cast<const AMDGPUDisassembler*>(Decoder);
237a8d9d507SStanislav Mekhanoshin   return addOperand(Inst, DAsm->decodeSrcOp(AMDGPUDisassembler::OPW128, Imm));
238a8d9d507SStanislav Mekhanoshin }
239a8d9d507SStanislav Mekhanoshin 
decodeOperand_VReg_256(MCInst & Inst,unsigned Imm,uint64_t Addr,const MCDisassembler * Decoder)2404ae9745aSMaksim Panchenko static DecodeStatus decodeOperand_VReg_256(MCInst &Inst, unsigned Imm,
241a8d9d507SStanislav Mekhanoshin                                            uint64_t Addr,
2424ae9745aSMaksim Panchenko                                            const MCDisassembler *Decoder) {
243a8d9d507SStanislav Mekhanoshin   auto DAsm = static_cast<const AMDGPUDisassembler*>(Decoder);
244a8d9d507SStanislav Mekhanoshin   return addOperand(Inst, DAsm->decodeSrcOp(AMDGPUDisassembler::OPW256, Imm));
245a8d9d507SStanislav Mekhanoshin }
246a8d9d507SStanislav Mekhanoshin 
decodeOperand_VReg_512(MCInst & Inst,unsigned Imm,uint64_t Addr,const MCDisassembler * Decoder)2474ae9745aSMaksim Panchenko static DecodeStatus decodeOperand_VReg_512(MCInst &Inst, unsigned Imm,
248a8d9d507SStanislav Mekhanoshin                                            uint64_t Addr,
2494ae9745aSMaksim Panchenko                                            const MCDisassembler *Decoder) {
250a8d9d507SStanislav Mekhanoshin   auto DAsm = static_cast<const AMDGPUDisassembler*>(Decoder);
251a8d9d507SStanislav Mekhanoshin   return addOperand(Inst, DAsm->decodeSrcOp(AMDGPUDisassembler::OPW512, Imm));
252a8d9d507SStanislav Mekhanoshin }
253a8d9d507SStanislav Mekhanoshin 
decodeOperand_VReg_1024(MCInst & Inst,unsigned Imm,uint64_t Addr,const MCDisassembler * Decoder)2544ae9745aSMaksim Panchenko static DecodeStatus decodeOperand_VReg_1024(MCInst &Inst, unsigned Imm,
255a8d9d507SStanislav Mekhanoshin                                             uint64_t Addr,
2564ae9745aSMaksim Panchenko                                             const MCDisassembler *Decoder) {
257a8d9d507SStanislav Mekhanoshin   auto DAsm = static_cast<const AMDGPUDisassembler*>(Decoder);
258a8d9d507SStanislav Mekhanoshin   return addOperand(Inst, DAsm->decodeSrcOp(AMDGPUDisassembler::OPW1024, Imm));
259a8d9d507SStanislav Mekhanoshin }
260a8d9d507SStanislav Mekhanoshin 
decodeOperand_f32kimm(MCInst & Inst,unsigned Imm,uint64_t Addr,const MCDisassembler * Decoder)261b4b7e605SJoe Nash static DecodeStatus decodeOperand_f32kimm(MCInst &Inst, unsigned Imm,
2624ae9745aSMaksim Panchenko                                           uint64_t Addr,
2634ae9745aSMaksim Panchenko                                           const MCDisassembler *Decoder) {
264b4b7e605SJoe Nash   const auto *DAsm = static_cast<const AMDGPUDisassembler *>(Decoder);
265b4b7e605SJoe Nash   return addOperand(Inst, DAsm->decodeMandatoryLiteralConstant(Imm));
266b4b7e605SJoe Nash }
267b4b7e605SJoe Nash 
decodeOperand_f16kimm(MCInst & Inst,unsigned Imm,uint64_t Addr,const MCDisassembler * Decoder)268b4b7e605SJoe Nash static DecodeStatus decodeOperand_f16kimm(MCInst &Inst, unsigned Imm,
2694ae9745aSMaksim Panchenko                                           uint64_t Addr,
2704ae9745aSMaksim Panchenko                                           const MCDisassembler *Decoder) {
271b4b7e605SJoe Nash   const auto *DAsm = static_cast<const AMDGPUDisassembler *>(Decoder);
272b4b7e605SJoe Nash   return addOperand(Inst, DAsm->decodeMandatoryLiteralConstant(Imm));
273b4b7e605SJoe Nash }
274b4b7e605SJoe Nash 
2754ae9745aSMaksim Panchenko static DecodeStatus
decodeOperand_VS_16_Deferred(MCInst & Inst,unsigned Imm,uint64_t Addr,const MCDisassembler * Decoder)2764ae9745aSMaksim Panchenko decodeOperand_VS_16_Deferred(MCInst &Inst, unsigned Imm, uint64_t Addr,
2774ae9745aSMaksim Panchenko                              const MCDisassembler *Decoder) {
278b4b7e605SJoe Nash   const auto *DAsm = static_cast<const AMDGPUDisassembler *>(Decoder);
279b4b7e605SJoe Nash   return addOperand(
280b4b7e605SJoe Nash       Inst, DAsm->decodeSrcOp(llvm::AMDGPUDisassembler::OPW16, Imm, true));
281b4b7e605SJoe Nash }
282b4b7e605SJoe Nash 
2834ae9745aSMaksim Panchenko static DecodeStatus
decodeOperand_VS_32_Deferred(MCInst & Inst,unsigned Imm,uint64_t Addr,const MCDisassembler * Decoder)2844ae9745aSMaksim Panchenko decodeOperand_VS_32_Deferred(MCInst &Inst, unsigned Imm, uint64_t Addr,
2854ae9745aSMaksim Panchenko                              const MCDisassembler *Decoder) {
286b4b7e605SJoe Nash   const auto *DAsm = static_cast<const AMDGPUDisassembler *>(Decoder);
287b4b7e605SJoe Nash   return addOperand(
288b4b7e605SJoe Nash       Inst, DAsm->decodeSrcOp(llvm::AMDGPUDisassembler::OPW32, Imm, true));
289b4b7e605SJoe Nash }
290b4b7e605SJoe Nash 
decodeOperandVOPDDstY(MCInst & Inst,unsigned Val,uint64_t Addr,const void * Decoder)29107b7fadaSJoe Nash static DecodeStatus decodeOperandVOPDDstY(MCInst &Inst, unsigned Val,
29207b7fadaSJoe Nash                                           uint64_t Addr, const void *Decoder) {
29307b7fadaSJoe Nash   const auto *DAsm = static_cast<const AMDGPUDisassembler *>(Decoder);
29407b7fadaSJoe Nash   return addOperand(Inst, DAsm->decodeVOPDDstYOp(Inst, Val));
29507b7fadaSJoe Nash }
29607b7fadaSJoe Nash 
IsAGPROperand(const MCInst & Inst,int OpIdx,const MCRegisterInfo * MRI)297a8d9d507SStanislav Mekhanoshin static bool IsAGPROperand(const MCInst &Inst, int OpIdx,
298a8d9d507SStanislav Mekhanoshin                           const MCRegisterInfo *MRI) {
299a8d9d507SStanislav Mekhanoshin   if (OpIdx < 0)
300a8d9d507SStanislav Mekhanoshin     return false;
301a8d9d507SStanislav Mekhanoshin 
302a8d9d507SStanislav Mekhanoshin   const MCOperand &Op = Inst.getOperand(OpIdx);
303a8d9d507SStanislav Mekhanoshin   if (!Op.isReg())
304a8d9d507SStanislav Mekhanoshin     return false;
305a8d9d507SStanislav Mekhanoshin 
306a8d9d507SStanislav Mekhanoshin   unsigned Sub = MRI->getSubReg(Op.getReg(), AMDGPU::sub0);
307a8d9d507SStanislav Mekhanoshin   auto Reg = Sub ? Sub : Op.getReg();
308a8d9d507SStanislav Mekhanoshin   return Reg >= AMDGPU::AGPR0 && Reg <= AMDGPU::AGPR255;
309a8d9d507SStanislav Mekhanoshin }
310a8d9d507SStanislav Mekhanoshin 
decodeOperand_AVLdSt_Any(MCInst & Inst,unsigned Imm,AMDGPUDisassembler::OpWidthTy Opw,const MCDisassembler * Decoder)3114ae9745aSMaksim Panchenko static DecodeStatus decodeOperand_AVLdSt_Any(MCInst &Inst, unsigned Imm,
312a8d9d507SStanislav Mekhanoshin                                              AMDGPUDisassembler::OpWidthTy Opw,
3134ae9745aSMaksim Panchenko                                              const MCDisassembler *Decoder) {
314a8d9d507SStanislav Mekhanoshin   auto DAsm = static_cast<const AMDGPUDisassembler*>(Decoder);
315a8d9d507SStanislav Mekhanoshin   if (!DAsm->isGFX90A()) {
316a8d9d507SStanislav Mekhanoshin     Imm &= 511;
317a8d9d507SStanislav Mekhanoshin   } else {
318a8d9d507SStanislav Mekhanoshin     // If atomic has both vdata and vdst their register classes are tied.
319a8d9d507SStanislav Mekhanoshin     // The bit is decoded along with the vdst, first operand. We need to
320a8d9d507SStanislav Mekhanoshin     // change register class to AGPR if vdst was AGPR.
321a8d9d507SStanislav Mekhanoshin     // If a DS instruction has both data0 and data1 their register classes
322a8d9d507SStanislav Mekhanoshin     // are also tied.
323a8d9d507SStanislav Mekhanoshin     unsigned Opc = Inst.getOpcode();
324a8d9d507SStanislav Mekhanoshin     uint64_t TSFlags = DAsm->getMCII()->get(Opc).TSFlags;
325a8d9d507SStanislav Mekhanoshin     uint16_t DataNameIdx = (TSFlags & SIInstrFlags::DS) ? AMDGPU::OpName::data0
326a8d9d507SStanislav Mekhanoshin                                                         : AMDGPU::OpName::vdata;
327a8d9d507SStanislav Mekhanoshin     const MCRegisterInfo *MRI = DAsm->getContext().getRegisterInfo();
328a8d9d507SStanislav Mekhanoshin     int DataIdx = AMDGPU::getNamedOperandIdx(Opc, DataNameIdx);
329a8d9d507SStanislav Mekhanoshin     if ((int)Inst.getNumOperands() == DataIdx) {
330a8d9d507SStanislav Mekhanoshin       int DstIdx = AMDGPU::getNamedOperandIdx(Opc, AMDGPU::OpName::vdst);
331a8d9d507SStanislav Mekhanoshin       if (IsAGPROperand(Inst, DstIdx, MRI))
332a8d9d507SStanislav Mekhanoshin         Imm |= 512;
333a8d9d507SStanislav Mekhanoshin     }
334a8d9d507SStanislav Mekhanoshin 
335a8d9d507SStanislav Mekhanoshin     if (TSFlags & SIInstrFlags::DS) {
336a8d9d507SStanislav Mekhanoshin       int Data2Idx = AMDGPU::getNamedOperandIdx(Opc, AMDGPU::OpName::data1);
337a8d9d507SStanislav Mekhanoshin       if ((int)Inst.getNumOperands() == Data2Idx &&
338a8d9d507SStanislav Mekhanoshin           IsAGPROperand(Inst, DataIdx, MRI))
339a8d9d507SStanislav Mekhanoshin         Imm |= 512;
340a8d9d507SStanislav Mekhanoshin     }
341a8d9d507SStanislav Mekhanoshin   }
342a8d9d507SStanislav Mekhanoshin   return addOperand(Inst, DAsm->decodeSrcOp(Opw, Imm | 256));
343a8d9d507SStanislav Mekhanoshin }
344a8d9d507SStanislav Mekhanoshin 
3454ae9745aSMaksim Panchenko static DecodeStatus
DecodeAVLdSt_32RegisterClass(MCInst & Inst,unsigned Imm,uint64_t Addr,const MCDisassembler * Decoder)3464ae9745aSMaksim Panchenko DecodeAVLdSt_32RegisterClass(MCInst &Inst, unsigned Imm, uint64_t Addr,
3474ae9745aSMaksim Panchenko                              const MCDisassembler *Decoder) {
348a8d9d507SStanislav Mekhanoshin   return decodeOperand_AVLdSt_Any(Inst, Imm,
349a8d9d507SStanislav Mekhanoshin                                   AMDGPUDisassembler::OPW32, Decoder);
350a8d9d507SStanislav Mekhanoshin }
351a8d9d507SStanislav Mekhanoshin 
3524ae9745aSMaksim Panchenko static DecodeStatus
DecodeAVLdSt_64RegisterClass(MCInst & Inst,unsigned Imm,uint64_t Addr,const MCDisassembler * Decoder)3534ae9745aSMaksim Panchenko DecodeAVLdSt_64RegisterClass(MCInst &Inst, unsigned Imm, uint64_t Addr,
3544ae9745aSMaksim Panchenko                              const MCDisassembler *Decoder) {
355a8d9d507SStanislav Mekhanoshin   return decodeOperand_AVLdSt_Any(Inst, Imm,
356a8d9d507SStanislav Mekhanoshin                                   AMDGPUDisassembler::OPW64, Decoder);
357a8d9d507SStanislav Mekhanoshin }
358a8d9d507SStanislav Mekhanoshin 
3594ae9745aSMaksim Panchenko static DecodeStatus
DecodeAVLdSt_96RegisterClass(MCInst & Inst,unsigned Imm,uint64_t Addr,const MCDisassembler * Decoder)3604ae9745aSMaksim Panchenko DecodeAVLdSt_96RegisterClass(MCInst &Inst, unsigned Imm, uint64_t Addr,
3614ae9745aSMaksim Panchenko                              const MCDisassembler *Decoder) {
362a8d9d507SStanislav Mekhanoshin   return decodeOperand_AVLdSt_Any(Inst, Imm,
363a8d9d507SStanislav Mekhanoshin                                   AMDGPUDisassembler::OPW96, Decoder);
364a8d9d507SStanislav Mekhanoshin }
365a8d9d507SStanislav Mekhanoshin 
3664ae9745aSMaksim Panchenko static DecodeStatus
DecodeAVLdSt_128RegisterClass(MCInst & Inst,unsigned Imm,uint64_t Addr,const MCDisassembler * Decoder)3674ae9745aSMaksim Panchenko DecodeAVLdSt_128RegisterClass(MCInst &Inst, unsigned Imm, uint64_t Addr,
3684ae9745aSMaksim Panchenko                               const MCDisassembler *Decoder) {
369a8d9d507SStanislav Mekhanoshin   return decodeOperand_AVLdSt_Any(Inst, Imm,
370a8d9d507SStanislav Mekhanoshin                                   AMDGPUDisassembler::OPW128, Decoder);
371a8d9d507SStanislav Mekhanoshin }
372a8d9d507SStanislav Mekhanoshin 
decodeOperand_SReg_32(MCInst & Inst,unsigned Imm,uint64_t Addr,const MCDisassembler * Decoder)3734ae9745aSMaksim Panchenko static DecodeStatus decodeOperand_SReg_32(MCInst &Inst, unsigned Imm,
3749e77d0c6SStanislav Mekhanoshin                                           uint64_t Addr,
3754ae9745aSMaksim Panchenko                                           const MCDisassembler *Decoder) {
3769e77d0c6SStanislav Mekhanoshin   auto DAsm = static_cast<const AMDGPUDisassembler*>(Decoder);
3779e77d0c6SStanislav Mekhanoshin   return addOperand(Inst, DAsm->decodeOperand_SReg_32(Imm));
3789e77d0c6SStanislav Mekhanoshin }
3799e77d0c6SStanislav Mekhanoshin 
380549c89d2SSam Kolton #define DECODE_SDWA(DecName) \
381549c89d2SSam Kolton DECODE_OPERAND(decodeSDWA##DecName, decodeSDWA##DecName)
382363f47a2SSam Kolton 
383549c89d2SSam Kolton DECODE_SDWA(Src32)
DECODE_SDWA(Src16)384549c89d2SSam Kolton DECODE_SDWA(Src16)
385549c89d2SSam Kolton DECODE_SDWA(VopcDst)
386363f47a2SSam Kolton 
387e1818af8STom Stellard #include "AMDGPUGenDisassemblerTables.inc"
388e1818af8STom Stellard 
389e1818af8STom Stellard //===----------------------------------------------------------------------===//
390e1818af8STom Stellard //
391e1818af8STom Stellard //===----------------------------------------------------------------------===//
392e1818af8STom Stellard 
3931048fb18SSam Kolton template <typename T> static inline T eatBytes(ArrayRef<uint8_t>& Bytes) {
3941048fb18SSam Kolton   assert(Bytes.size() >= sizeof(T));
3951048fb18SSam Kolton   const auto Res = support::endian::read<T, support::endianness::little>(Bytes.data());
3961048fb18SSam Kolton   Bytes = Bytes.slice(sizeof(T));
397ac106addSNikolay Haustov   return Res;
398ac106addSNikolay Haustov }
399ac106addSNikolay Haustov 
eat12Bytes(ArrayRef<uint8_t> & Bytes)400e243ead6SJoe Nash static inline DecoderUInt128 eat12Bytes(ArrayRef<uint8_t> &Bytes) {
401e243ead6SJoe Nash   assert(Bytes.size() >= 12);
402e243ead6SJoe Nash   uint64_t Lo = support::endian::read<uint64_t, support::endianness::little>(
403e243ead6SJoe Nash       Bytes.data());
404e243ead6SJoe Nash   Bytes = Bytes.slice(8);
405e243ead6SJoe Nash   uint64_t Hi = support::endian::read<uint32_t, support::endianness::little>(
406e243ead6SJoe Nash       Bytes.data());
407e243ead6SJoe Nash   Bytes = Bytes.slice(4);
408e243ead6SJoe Nash   return DecoderUInt128(Lo, Hi);
409e243ead6SJoe Nash }
410e243ead6SJoe Nash 
411919236e6SJoe Nash // The disassembler is greedy, so we need to check FI operand value to
412919236e6SJoe Nash // not parse a dpp if the correct literal is not set. For dpp16 the
413919236e6SJoe Nash // autogenerated decoder checks the dpp literal
isValidDPP8(const MCInst & MI)414245b5ba3SStanislav Mekhanoshin static bool isValidDPP8(const MCInst &MI) {
415245b5ba3SStanislav Mekhanoshin   using namespace llvm::AMDGPU::DPP;
416245b5ba3SStanislav Mekhanoshin   int FiIdx = AMDGPU::getNamedOperandIdx(MI.getOpcode(), AMDGPU::OpName::fi);
417245b5ba3SStanislav Mekhanoshin   assert(FiIdx != -1);
418245b5ba3SStanislav Mekhanoshin   if ((unsigned)FiIdx >= MI.getNumOperands())
419245b5ba3SStanislav Mekhanoshin     return false;
420245b5ba3SStanislav Mekhanoshin   unsigned Fi = MI.getOperand(FiIdx).getImm();
421245b5ba3SStanislav Mekhanoshin   return Fi == DPP8_FI_0 || Fi == DPP8_FI_1;
422245b5ba3SStanislav Mekhanoshin }
423245b5ba3SStanislav Mekhanoshin 
getInstruction(MCInst & MI,uint64_t & Size,ArrayRef<uint8_t> Bytes_,uint64_t Address,raw_ostream & CS) const424e1818af8STom Stellard DecodeStatus AMDGPUDisassembler::getInstruction(MCInst &MI, uint64_t &Size,
425ac106addSNikolay Haustov                                                 ArrayRef<uint8_t> Bytes_,
426e1818af8STom Stellard                                                 uint64_t Address,
427e1818af8STom Stellard                                                 raw_ostream &CS) const {
428e1818af8STom Stellard   CommentStream = &CS;
429549c89d2SSam Kolton   bool IsSDWA = false;
430e1818af8STom Stellard 
431ca64ef20SMatt Arsenault   unsigned MaxInstBytesNum = std::min((size_t)TargetMaxInstBytes, Bytes_.size());
432ac106addSNikolay Haustov   Bytes = Bytes_.slice(0, MaxInstBytesNum);
433161a158eSNikolay Haustov 
434ac106addSNikolay Haustov   DecodeStatus Res = MCDisassembler::Fail;
435ac106addSNikolay Haustov   do {
436824e804bSValery Pykhtin     // ToDo: better to switch encoding length using some bit predicate
437ac106addSNikolay Haustov     // but it is unknown yet, so try all we can
4381048fb18SSam Kolton 
439c9bdcb75SSam Kolton     // Try to decode DPP and SDWA first to solve conflict with VOP1 and VOP2
440c9bdcb75SSam Kolton     // encodings
441e243ead6SJoe Nash     if (isGFX11Plus() && Bytes.size() >= 12 ) {
442e243ead6SJoe Nash       DecoderUInt128 DecW = eat12Bytes(Bytes);
443e243ead6SJoe Nash       Res = tryDecodeInst(DecoderTableDPP8GFX1196, MI, DecW,
444e243ead6SJoe Nash                                           Address);
445e243ead6SJoe Nash       if (Res && convertDPP8Inst(MI) == MCDisassembler::Success)
446e243ead6SJoe Nash         break;
447e243ead6SJoe Nash       MI = MCInst(); // clear
448e243ead6SJoe Nash       Res = tryDecodeInst(DecoderTableDPPGFX1196, MI, DecW,
449e243ead6SJoe Nash                                           Address);
45040f35cefSJoe Nash       if (Res) {
45140f35cefSJoe Nash         if (MCII->get(MI.getOpcode()).TSFlags & SIInstrFlags::VOP3P)
45240f35cefSJoe Nash           convertVOP3PDPPInst(MI);
453485e8b4fSDmitry Preobrazhensky         else if (AMDGPU::isVOPC64DPP(MI.getOpcode()))
4542a6532d5SDmitry Preobrazhensky           convertVOPCDPPInst(MI); // Special VOP3 case
4552a6532d5SDmitry Preobrazhensky         else {
4562a6532d5SDmitry Preobrazhensky           assert(MCII->get(MI.getOpcode()).TSFlags & SIInstrFlags::VOP3);
4572a6532d5SDmitry Preobrazhensky           convertVOP3DPPInst(MI); // Regular VOP3 case
4582a6532d5SDmitry Preobrazhensky         }
459e243ead6SJoe Nash         break;
460e243ead6SJoe Nash       }
46107b7fadaSJoe Nash       Res = tryDecodeInst(DecoderTableGFX1196, MI, DecW, Address);
46207b7fadaSJoe Nash       if (Res)
46307b7fadaSJoe Nash         break;
46440f35cefSJoe Nash     }
465e243ead6SJoe Nash     // Reinitialize Bytes
466e243ead6SJoe Nash     Bytes = Bytes_.slice(0, MaxInstBytesNum);
467e243ead6SJoe Nash 
4681048fb18SSam Kolton     if (Bytes.size() >= 8) {
4691048fb18SSam Kolton       const uint64_t QW = eatBytes<uint64_t>(Bytes);
470245b5ba3SStanislav Mekhanoshin 
4719ee272f1SStanislav Mekhanoshin       if (STI.getFeatureBits()[AMDGPU::FeatureGFX10_BEncoding]) {
4729ee272f1SStanislav Mekhanoshin         Res = tryDecodeInst(DecoderTableGFX10_B64, MI, QW, Address);
4739ee272f1SStanislav Mekhanoshin         if (Res) {
4749ee272f1SStanislav Mekhanoshin           if (AMDGPU::getNamedOperandIdx(MI.getOpcode(), AMDGPU::OpName::dpp8)
4759ee272f1SStanislav Mekhanoshin               == -1)
4769ee272f1SStanislav Mekhanoshin             break;
4779ee272f1SStanislav Mekhanoshin           if (convertDPP8Inst(MI) == MCDisassembler::Success)
4789ee272f1SStanislav Mekhanoshin             break;
4799ee272f1SStanislav Mekhanoshin           MI = MCInst(); // clear
4809ee272f1SStanislav Mekhanoshin         }
4819ee272f1SStanislav Mekhanoshin       }
4829ee272f1SStanislav Mekhanoshin 
483245b5ba3SStanislav Mekhanoshin       Res = tryDecodeInst(DecoderTableDPP864, MI, QW, Address);
484245b5ba3SStanislav Mekhanoshin       if (Res && convertDPP8Inst(MI) == MCDisassembler::Success)
485245b5ba3SStanislav Mekhanoshin         break;
486086a9c10SJoe Nash       MI = MCInst(); // clear
487245b5ba3SStanislav Mekhanoshin 
488086a9c10SJoe Nash       Res = tryDecodeInst(DecoderTableDPP8GFX1164, MI, QW, Address);
489086a9c10SJoe Nash       if (Res && convertDPP8Inst(MI) == MCDisassembler::Success)
490086a9c10SJoe Nash         break;
491245b5ba3SStanislav Mekhanoshin       MI = MCInst(); // clear
492245b5ba3SStanislav Mekhanoshin 
4931048fb18SSam Kolton       Res = tryDecodeInst(DecoderTableDPP64, MI, QW, Address);
4941048fb18SSam Kolton       if (Res) break;
495c9bdcb75SSam Kolton 
496086a9c10SJoe Nash       Res = tryDecodeInst(DecoderTableDPPGFX1164, MI, QW, Address);
497be1082c6SJoe Nash       if (Res) {
498be1082c6SJoe Nash         if (MCII->get(MI.getOpcode()).TSFlags & SIInstrFlags::VOPC)
499be1082c6SJoe Nash           convertVOPCDPPInst(MI);
500086a9c10SJoe Nash         break;
501be1082c6SJoe Nash       }
502086a9c10SJoe Nash 
503c9bdcb75SSam Kolton       Res = tryDecodeInst(DecoderTableSDWA64, MI, QW, Address);
504549c89d2SSam Kolton       if (Res) { IsSDWA = true;  break; }
505363f47a2SSam Kolton 
506363f47a2SSam Kolton       Res = tryDecodeInst(DecoderTableSDWA964, MI, QW, Address);
507549c89d2SSam Kolton       if (Res) { IsSDWA = true;  break; }
5080905870fSChangpeng Fang 
5098f3da70eSStanislav Mekhanoshin       Res = tryDecodeInst(DecoderTableSDWA1064, MI, QW, Address);
5108f3da70eSStanislav Mekhanoshin       if (Res) { IsSDWA = true;  break; }
5118f3da70eSStanislav Mekhanoshin 
5120905870fSChangpeng Fang       if (STI.getFeatureBits()[AMDGPU::FeatureUnpackedD16VMem]) {
5130905870fSChangpeng Fang         Res = tryDecodeInst(DecoderTableGFX80_UNPACKED64, MI, QW, Address);
5140084adc5SMatt Arsenault         if (Res)
5150084adc5SMatt Arsenault           break;
5160084adc5SMatt Arsenault       }
5170084adc5SMatt Arsenault 
5180084adc5SMatt Arsenault       // Some GFX9 subtargets repurposed the v_mad_mix_f32, v_mad_mixlo_f16 and
5190084adc5SMatt Arsenault       // v_mad_mixhi_f16 for FMA variants. Try to decode using this special
5200084adc5SMatt Arsenault       // table first so we print the correct name.
5210084adc5SMatt Arsenault       if (STI.getFeatureBits()[AMDGPU::FeatureFmaMixInsts]) {
5220084adc5SMatt Arsenault         Res = tryDecodeInst(DecoderTableGFX9_DL64, MI, QW, Address);
5230084adc5SMatt Arsenault         if (Res)
5240084adc5SMatt Arsenault           break;
5250905870fSChangpeng Fang       }
5261048fb18SSam Kolton     }
5271048fb18SSam Kolton 
5281048fb18SSam Kolton     // Reinitialize Bytes as DPP64 could have eaten too much
5291048fb18SSam Kolton     Bytes = Bytes_.slice(0, MaxInstBytesNum);
5301048fb18SSam Kolton 
5311048fb18SSam Kolton     // Try decode 32-bit instruction
532ac106addSNikolay Haustov     if (Bytes.size() < 4) break;
5331048fb18SSam Kolton     const uint32_t DW = eatBytes<uint32_t>(Bytes);
5345182302aSStanislav Mekhanoshin     Res = tryDecodeInst(DecoderTableGFX832, MI, DW, Address);
535ac106addSNikolay Haustov     if (Res) break;
536e1818af8STom Stellard 
537ac106addSNikolay Haustov     Res = tryDecodeInst(DecoderTableAMDGPU32, MI, DW, Address);
538ac106addSNikolay Haustov     if (Res) break;
539ac106addSNikolay Haustov 
540a0342dc9SDmitry Preobrazhensky     Res = tryDecodeInst(DecoderTableGFX932, MI, DW, Address);
541a0342dc9SDmitry Preobrazhensky     if (Res) break;
542a0342dc9SDmitry Preobrazhensky 
543a8d9d507SStanislav Mekhanoshin     if (STI.getFeatureBits()[AMDGPU::FeatureGFX90AInsts]) {
544a8d9d507SStanislav Mekhanoshin       Res = tryDecodeInst(DecoderTableGFX90A32, MI, DW, Address);
545a8d9d507SStanislav Mekhanoshin       if (Res)
546a8d9d507SStanislav Mekhanoshin         break;
547a8d9d507SStanislav Mekhanoshin     }
548a8d9d507SStanislav Mekhanoshin 
5499ee272f1SStanislav Mekhanoshin     if (STI.getFeatureBits()[AMDGPU::FeatureGFX10_BEncoding]) {
5509ee272f1SStanislav Mekhanoshin       Res = tryDecodeInst(DecoderTableGFX10_B32, MI, DW, Address);
5519ee272f1SStanislav Mekhanoshin       if (Res) break;
5529ee272f1SStanislav Mekhanoshin     }
5539ee272f1SStanislav Mekhanoshin 
5548f3da70eSStanislav Mekhanoshin     Res = tryDecodeInst(DecoderTableGFX1032, MI, DW, Address);
5558f3da70eSStanislav Mekhanoshin     if (Res) break;
5568f3da70eSStanislav Mekhanoshin 
557d21b9b49SJoe Nash     Res = tryDecodeInst(DecoderTableGFX1132, MI, DW, Address);
558d21b9b49SJoe Nash     if (Res) break;
559d21b9b49SJoe Nash 
560ac106addSNikolay Haustov     if (Bytes.size() < 4) break;
5611048fb18SSam Kolton     const uint64_t QW = ((uint64_t)eatBytes<uint32_t>(Bytes) << 32) | DW;
562a8d9d507SStanislav Mekhanoshin 
563a8d9d507SStanislav Mekhanoshin     if (STI.getFeatureBits()[AMDGPU::FeatureGFX90AInsts]) {
564a8d9d507SStanislav Mekhanoshin       Res = tryDecodeInst(DecoderTableGFX90A64, MI, QW, Address);
565a8d9d507SStanislav Mekhanoshin       if (Res)
566a8d9d507SStanislav Mekhanoshin         break;
567a8d9d507SStanislav Mekhanoshin     }
568a8d9d507SStanislav Mekhanoshin 
5695182302aSStanislav Mekhanoshin     Res = tryDecodeInst(DecoderTableGFX864, MI, QW, Address);
570ac106addSNikolay Haustov     if (Res) break;
571ac106addSNikolay Haustov 
572ac106addSNikolay Haustov     Res = tryDecodeInst(DecoderTableAMDGPU64, MI, QW, Address);
5731e32550dSDmitry Preobrazhensky     if (Res) break;
5741e32550dSDmitry Preobrazhensky 
5751e32550dSDmitry Preobrazhensky     Res = tryDecodeInst(DecoderTableGFX964, MI, QW, Address);
5768f3da70eSStanislav Mekhanoshin     if (Res) break;
5778f3da70eSStanislav Mekhanoshin 
5788f3da70eSStanislav Mekhanoshin     Res = tryDecodeInst(DecoderTableGFX1064, MI, QW, Address);
579c7025940SJoe Nash     if (Res) break;
580c7025940SJoe Nash 
581c7025940SJoe Nash     Res = tryDecodeInst(DecoderTableGFX1164, MI, QW, Address);
5824874838aSPiotr Sobczak     if (Res)
5834874838aSPiotr Sobczak       break;
5844874838aSPiotr Sobczak 
5854874838aSPiotr Sobczak     Res = tryDecodeInst(DecoderTableWMMAGFX1164, MI, QW, Address);
586ac106addSNikolay Haustov   } while (false);
587ac106addSNikolay Haustov 
588678e111eSMatt Arsenault   if (Res && (MI.getOpcode() == AMDGPU::V_MAC_F32_e64_vi ||
5898f3da70eSStanislav Mekhanoshin               MI.getOpcode() == AMDGPU::V_MAC_F32_e64_gfx6_gfx7 ||
5908f3da70eSStanislav Mekhanoshin               MI.getOpcode() == AMDGPU::V_MAC_F32_e64_gfx10 ||
5917238faa4SJay Foad               MI.getOpcode() == AMDGPU::V_MAC_LEGACY_F32_e64_gfx6_gfx7 ||
5927238faa4SJay Foad               MI.getOpcode() == AMDGPU::V_MAC_LEGACY_F32_e64_gfx10 ||
593603a43fcSKonstantin Zhuravlyov               MI.getOpcode() == AMDGPU::V_MAC_F16_e64_vi ||
594a8d9d507SStanislav Mekhanoshin               MI.getOpcode() == AMDGPU::V_FMAC_F64_e64_gfx90a ||
5958f3da70eSStanislav Mekhanoshin               MI.getOpcode() == AMDGPU::V_FMAC_F32_e64_vi ||
5968f3da70eSStanislav Mekhanoshin               MI.getOpcode() == AMDGPU::V_FMAC_F32_e64_gfx10 ||
597086a9c10SJoe Nash               MI.getOpcode() == AMDGPU::V_FMAC_F32_e64_gfx11 ||
598edc37bacSJay Foad               MI.getOpcode() == AMDGPU::V_FMAC_LEGACY_F32_e64_gfx10 ||
599086a9c10SJoe Nash               MI.getOpcode() == AMDGPU::V_FMAC_DX9_ZERO_F32_e64_gfx11 ||
600086a9c10SJoe Nash               MI.getOpcode() == AMDGPU::V_FMAC_F16_e64_gfx10 ||
601086a9c10SJoe Nash               MI.getOpcode() == AMDGPU::V_FMAC_F16_e64_gfx11)) {
602678e111eSMatt Arsenault     // Insert dummy unused src2_modifiers.
603549c89d2SSam Kolton     insertNamedMCOperand(MI, MCOperand::createImm(0),
604678e111eSMatt Arsenault                          AMDGPU::OpName::src2_modifiers);
605678e111eSMatt Arsenault   }
606678e111eSMatt Arsenault 
607f738aee0SStanislav Mekhanoshin   if (Res && (MCII->get(MI.getOpcode()).TSFlags &
6083bffb1cdSStanislav Mekhanoshin           (SIInstrFlags::MUBUF | SIInstrFlags::FLAT | SIInstrFlags::SMRD))) {
6093bffb1cdSStanislav Mekhanoshin     int CPolPos = AMDGPU::getNamedOperandIdx(MI.getOpcode(),
6103bffb1cdSStanislav Mekhanoshin                                              AMDGPU::OpName::cpol);
6113bffb1cdSStanislav Mekhanoshin     if (CPolPos != -1) {
6123bffb1cdSStanislav Mekhanoshin       unsigned CPol =
6133bffb1cdSStanislav Mekhanoshin           (MCII->get(MI.getOpcode()).TSFlags & SIInstrFlags::IsAtomicRet) ?
6143bffb1cdSStanislav Mekhanoshin               AMDGPU::CPol::GLC : 0;
6153bffb1cdSStanislav Mekhanoshin       if (MI.getNumOperands() <= (unsigned)CPolPos) {
6163bffb1cdSStanislav Mekhanoshin         insertNamedMCOperand(MI, MCOperand::createImm(CPol),
6173bffb1cdSStanislav Mekhanoshin                              AMDGPU::OpName::cpol);
6183bffb1cdSStanislav Mekhanoshin       } else if (CPol) {
6193bffb1cdSStanislav Mekhanoshin         MI.getOperand(CPolPos).setImm(MI.getOperand(CPolPos).getImm() | CPol);
6203bffb1cdSStanislav Mekhanoshin       }
6213bffb1cdSStanislav Mekhanoshin     }
622f738aee0SStanislav Mekhanoshin   }
623f738aee0SStanislav Mekhanoshin 
624a8d9d507SStanislav Mekhanoshin   if (Res && (MCII->get(MI.getOpcode()).TSFlags &
625a8d9d507SStanislav Mekhanoshin               (SIInstrFlags::MTBUF | SIInstrFlags::MUBUF)) &&
626a8d9d507SStanislav Mekhanoshin              (STI.getFeatureBits()[AMDGPU::FeatureGFX90AInsts])) {
627a8d9d507SStanislav Mekhanoshin     // GFX90A lost TFE, its place is occupied by ACC.
628a8d9d507SStanislav Mekhanoshin     int TFEOpIdx =
629a8d9d507SStanislav Mekhanoshin         AMDGPU::getNamedOperandIdx(MI.getOpcode(), AMDGPU::OpName::tfe);
630a8d9d507SStanislav Mekhanoshin     if (TFEOpIdx != -1) {
631a8d9d507SStanislav Mekhanoshin       auto TFEIter = MI.begin();
632a8d9d507SStanislav Mekhanoshin       std::advance(TFEIter, TFEOpIdx);
633a8d9d507SStanislav Mekhanoshin       MI.insert(TFEIter, MCOperand::createImm(0));
634a8d9d507SStanislav Mekhanoshin     }
635a8d9d507SStanislav Mekhanoshin   }
636a8d9d507SStanislav Mekhanoshin 
637a8d9d507SStanislav Mekhanoshin   if (Res && (MCII->get(MI.getOpcode()).TSFlags &
638a8d9d507SStanislav Mekhanoshin               (SIInstrFlags::MTBUF | SIInstrFlags::MUBUF))) {
639a8d9d507SStanislav Mekhanoshin     int SWZOpIdx =
640a8d9d507SStanislav Mekhanoshin         AMDGPU::getNamedOperandIdx(MI.getOpcode(), AMDGPU::OpName::swz);
641a8d9d507SStanislav Mekhanoshin     if (SWZOpIdx != -1) {
642a8d9d507SStanislav Mekhanoshin       auto SWZIter = MI.begin();
643a8d9d507SStanislav Mekhanoshin       std::advance(SWZIter, SWZOpIdx);
644a8d9d507SStanislav Mekhanoshin       MI.insert(SWZIter, MCOperand::createImm(0));
645a8d9d507SStanislav Mekhanoshin     }
646a8d9d507SStanislav Mekhanoshin   }
647a8d9d507SStanislav Mekhanoshin 
648cad7fa85SMatt Arsenault   if (Res && (MCII->get(MI.getOpcode()).TSFlags & SIInstrFlags::MIMG)) {
649692560dcSStanislav Mekhanoshin     int VAddr0Idx =
650692560dcSStanislav Mekhanoshin         AMDGPU::getNamedOperandIdx(MI.getOpcode(), AMDGPU::OpName::vaddr0);
651692560dcSStanislav Mekhanoshin     int RsrcIdx =
652692560dcSStanislav Mekhanoshin         AMDGPU::getNamedOperandIdx(MI.getOpcode(), AMDGPU::OpName::srsrc);
653692560dcSStanislav Mekhanoshin     unsigned NSAArgs = RsrcIdx - VAddr0Idx - 1;
654692560dcSStanislav Mekhanoshin     if (VAddr0Idx >= 0 && NSAArgs > 0) {
655692560dcSStanislav Mekhanoshin       unsigned NSAWords = (NSAArgs + 3) / 4;
656692560dcSStanislav Mekhanoshin       if (Bytes.size() < 4 * NSAWords) {
657692560dcSStanislav Mekhanoshin         Res = MCDisassembler::Fail;
658692560dcSStanislav Mekhanoshin       } else {
659692560dcSStanislav Mekhanoshin         for (unsigned i = 0; i < NSAArgs; ++i) {
660e8860beeSJoe Nash           const unsigned VAddrIdx = VAddr0Idx + 1 + i;
661e8860beeSJoe Nash           auto VAddrRCID = MCII->get(MI.getOpcode()).OpInfo[VAddrIdx].RegClass;
662e8860beeSJoe Nash           MI.insert(MI.begin() + VAddrIdx,
663e8860beeSJoe Nash                     createRegOperand(VAddrRCID, Bytes[i]));
664692560dcSStanislav Mekhanoshin         }
665692560dcSStanislav Mekhanoshin         Bytes = Bytes.slice(4 * NSAWords);
666692560dcSStanislav Mekhanoshin       }
667692560dcSStanislav Mekhanoshin     }
668692560dcSStanislav Mekhanoshin 
669692560dcSStanislav Mekhanoshin     if (Res)
670cad7fa85SMatt Arsenault       Res = convertMIMGInst(MI);
671cad7fa85SMatt Arsenault   }
672cad7fa85SMatt Arsenault 
6731a51ab76SJoe Nash   if (Res && (MCII->get(MI.getOpcode()).TSFlags & SIInstrFlags::EXP))
6741a51ab76SJoe Nash     Res = convertEXPInst(MI);
6751a51ab76SJoe Nash 
676ef1ea5acSJoe Nash   if (Res && (MCII->get(MI.getOpcode()).TSFlags & SIInstrFlags::VINTERP))
677ef1ea5acSJoe Nash     Res = convertVINTERPInst(MI);
678ef1ea5acSJoe Nash 
679549c89d2SSam Kolton   if (Res && IsSDWA)
680549c89d2SSam Kolton     Res = convertSDWAInst(MI);
681549c89d2SSam Kolton 
6828f3da70eSStanislav Mekhanoshin   int VDstIn_Idx = AMDGPU::getNamedOperandIdx(MI.getOpcode(),
6838f3da70eSStanislav Mekhanoshin                                               AMDGPU::OpName::vdst_in);
6848f3da70eSStanislav Mekhanoshin   if (VDstIn_Idx != -1) {
6858f3da70eSStanislav Mekhanoshin     int Tied = MCII->get(MI.getOpcode()).getOperandConstraint(VDstIn_Idx,
6868f3da70eSStanislav Mekhanoshin                            MCOI::OperandConstraint::TIED_TO);
6878f3da70eSStanislav Mekhanoshin     if (Tied != -1 && (MI.getNumOperands() <= (unsigned)VDstIn_Idx ||
6888f3da70eSStanislav Mekhanoshin          !MI.getOperand(VDstIn_Idx).isReg() ||
6898f3da70eSStanislav Mekhanoshin          MI.getOperand(VDstIn_Idx).getReg() != MI.getOperand(Tied).getReg())) {
6908f3da70eSStanislav Mekhanoshin       if (MI.getNumOperands() > (unsigned)VDstIn_Idx)
6918f3da70eSStanislav Mekhanoshin         MI.erase(&MI.getOperand(VDstIn_Idx));
6928f3da70eSStanislav Mekhanoshin       insertNamedMCOperand(MI,
6938f3da70eSStanislav Mekhanoshin         MCOperand::createReg(MI.getOperand(Tied).getReg()),
6948f3da70eSStanislav Mekhanoshin         AMDGPU::OpName::vdst_in);
6958f3da70eSStanislav Mekhanoshin     }
6968f3da70eSStanislav Mekhanoshin   }
6978f3da70eSStanislav Mekhanoshin 
698b4b7e605SJoe Nash   int ImmLitIdx =
699b4b7e605SJoe Nash       AMDGPU::getNamedOperandIdx(MI.getOpcode(), AMDGPU::OpName::imm);
700b4b7e605SJoe Nash   if (Res && ImmLitIdx != -1)
701b4b7e605SJoe Nash     Res = convertFMAanyK(MI, ImmLitIdx);
702b4b7e605SJoe Nash 
7037116e896STim Corringham   // if the opcode was not recognized we'll assume a Size of 4 bytes
7047116e896STim Corringham   // (unless there are fewer bytes left)
7057116e896STim Corringham   Size = Res ? (MaxInstBytesNum - Bytes.size())
7067116e896STim Corringham              : std::min((size_t)4, Bytes_.size());
707ac106addSNikolay Haustov   return Res;
708161a158eSNikolay Haustov }
709e1818af8STom Stellard 
convertEXPInst(MCInst & MI) const7101a51ab76SJoe Nash DecodeStatus AMDGPUDisassembler::convertEXPInst(MCInst &MI) const {
7111a51ab76SJoe Nash   if (STI.getFeatureBits()[AMDGPU::FeatureGFX11]) {
7121a51ab76SJoe Nash     // The MCInst still has these fields even though they are no longer encoded
7131a51ab76SJoe Nash     // in the GFX11 instruction.
7141a51ab76SJoe Nash     insertNamedMCOperand(MI, MCOperand::createImm(0), AMDGPU::OpName::vm);
7151a51ab76SJoe Nash     insertNamedMCOperand(MI, MCOperand::createImm(0), AMDGPU::OpName::compr);
7161a51ab76SJoe Nash   }
7171a51ab76SJoe Nash   return MCDisassembler::Success;
7181a51ab76SJoe Nash }
7191a51ab76SJoe Nash 
convertVINTERPInst(MCInst & MI) const720ef1ea5acSJoe Nash DecodeStatus AMDGPUDisassembler::convertVINTERPInst(MCInst &MI) const {
721ef1ea5acSJoe Nash   if (MI.getOpcode() == AMDGPU::V_INTERP_P10_F16_F32_inreg_gfx11 ||
722ef1ea5acSJoe Nash       MI.getOpcode() == AMDGPU::V_INTERP_P10_RTZ_F16_F32_inreg_gfx11 ||
723ef1ea5acSJoe Nash       MI.getOpcode() == AMDGPU::V_INTERP_P2_F16_F32_inreg_gfx11 ||
724ef1ea5acSJoe Nash       MI.getOpcode() == AMDGPU::V_INTERP_P2_RTZ_F16_F32_inreg_gfx11) {
725ef1ea5acSJoe Nash     // The MCInst has this field that is not directly encoded in the
726ef1ea5acSJoe Nash     // instruction.
727ef1ea5acSJoe Nash     insertNamedMCOperand(MI, MCOperand::createImm(0), AMDGPU::OpName::op_sel);
728ef1ea5acSJoe Nash   }
729ef1ea5acSJoe Nash   return MCDisassembler::Success;
730ef1ea5acSJoe Nash }
731ef1ea5acSJoe Nash 
convertSDWAInst(MCInst & MI) const732549c89d2SSam Kolton DecodeStatus AMDGPUDisassembler::convertSDWAInst(MCInst &MI) const {
7338f3da70eSStanislav Mekhanoshin   if (STI.getFeatureBits()[AMDGPU::FeatureGFX9] ||
7348f3da70eSStanislav Mekhanoshin       STI.getFeatureBits()[AMDGPU::FeatureGFX10]) {
735549c89d2SSam Kolton     if (AMDGPU::getNamedOperandIdx(MI.getOpcode(), AMDGPU::OpName::sdst) != -1)
736549c89d2SSam Kolton       // VOPC - insert clamp
737549c89d2SSam Kolton       insertNamedMCOperand(MI, MCOperand::createImm(0), AMDGPU::OpName::clamp);
738549c89d2SSam Kolton   } else if (STI.getFeatureBits()[AMDGPU::FeatureVolcanicIslands]) {
739549c89d2SSam Kolton     int SDst = AMDGPU::getNamedOperandIdx(MI.getOpcode(), AMDGPU::OpName::sdst);
740549c89d2SSam Kolton     if (SDst != -1) {
741549c89d2SSam Kolton       // VOPC - insert VCC register as sdst
742ac2b0264SDmitry Preobrazhensky       insertNamedMCOperand(MI, createRegOperand(AMDGPU::VCC),
743549c89d2SSam Kolton                            AMDGPU::OpName::sdst);
744549c89d2SSam Kolton     } else {
745549c89d2SSam Kolton       // VOP1/2 - insert omod if present in instruction
746549c89d2SSam Kolton       insertNamedMCOperand(MI, MCOperand::createImm(0), AMDGPU::OpName::omod);
747549c89d2SSam Kolton     }
748549c89d2SSam Kolton   }
749549c89d2SSam Kolton   return MCDisassembler::Success;
750549c89d2SSam Kolton }
751549c89d2SSam Kolton 
7522a6532d5SDmitry Preobrazhensky struct VOPModifiers {
7532a6532d5SDmitry Preobrazhensky   unsigned OpSel = 0;
7542a6532d5SDmitry Preobrazhensky   unsigned OpSelHi = 0;
7552a6532d5SDmitry Preobrazhensky   unsigned NegLo = 0;
7562a6532d5SDmitry Preobrazhensky   unsigned NegHi = 0;
7572a6532d5SDmitry Preobrazhensky };
7582a6532d5SDmitry Preobrazhensky 
7592a6532d5SDmitry Preobrazhensky // Reconstruct values of VOP3/VOP3P operands such as op_sel.
7602a6532d5SDmitry Preobrazhensky // Note that these values do not affect disassembler output,
7612a6532d5SDmitry Preobrazhensky // so this is only necessary for consistency with src_modifiers.
collectVOPModifiers(const MCInst & MI,bool IsVOP3P=false)7622a6532d5SDmitry Preobrazhensky static VOPModifiers collectVOPModifiers(const MCInst &MI,
7632a6532d5SDmitry Preobrazhensky                                         bool IsVOP3P = false) {
7642a6532d5SDmitry Preobrazhensky   VOPModifiers Modifiers;
7652a6532d5SDmitry Preobrazhensky   unsigned Opc = MI.getOpcode();
7662a6532d5SDmitry Preobrazhensky   const int ModOps[] = {AMDGPU::OpName::src0_modifiers,
7672a6532d5SDmitry Preobrazhensky                         AMDGPU::OpName::src1_modifiers,
7682a6532d5SDmitry Preobrazhensky                         AMDGPU::OpName::src2_modifiers};
7692a6532d5SDmitry Preobrazhensky   for (int J = 0; J < 3; ++J) {
7702a6532d5SDmitry Preobrazhensky     int OpIdx = AMDGPU::getNamedOperandIdx(Opc, ModOps[J]);
7712a6532d5SDmitry Preobrazhensky     if (OpIdx == -1)
7722a6532d5SDmitry Preobrazhensky       continue;
7732a6532d5SDmitry Preobrazhensky 
7742a6532d5SDmitry Preobrazhensky     unsigned Val = MI.getOperand(OpIdx).getImm();
7752a6532d5SDmitry Preobrazhensky 
7762a6532d5SDmitry Preobrazhensky     Modifiers.OpSel |= !!(Val & SISrcMods::OP_SEL_0) << J;
7772a6532d5SDmitry Preobrazhensky     if (IsVOP3P) {
7782a6532d5SDmitry Preobrazhensky       Modifiers.OpSelHi |= !!(Val & SISrcMods::OP_SEL_1) << J;
7792a6532d5SDmitry Preobrazhensky       Modifiers.NegLo |= !!(Val & SISrcMods::NEG) << J;
7802a6532d5SDmitry Preobrazhensky       Modifiers.NegHi |= !!(Val & SISrcMods::NEG_HI) << J;
7812a6532d5SDmitry Preobrazhensky     } else if (J == 0) {
7822a6532d5SDmitry Preobrazhensky       Modifiers.OpSel |= !!(Val & SISrcMods::DST_OP_SEL) << 3;
7832a6532d5SDmitry Preobrazhensky     }
7842a6532d5SDmitry Preobrazhensky   }
7852a6532d5SDmitry Preobrazhensky 
7862a6532d5SDmitry Preobrazhensky   return Modifiers;
7872a6532d5SDmitry Preobrazhensky }
7882a6532d5SDmitry Preobrazhensky 
789919236e6SJoe Nash // We must check FI == literal to reject not genuine dpp8 insts, and we must
790919236e6SJoe Nash // first add optional MI operands to check FI
convertDPP8Inst(MCInst & MI) const791245b5ba3SStanislav Mekhanoshin DecodeStatus AMDGPUDisassembler::convertDPP8Inst(MCInst &MI) const {
792245b5ba3SStanislav Mekhanoshin   unsigned Opc = MI.getOpcode();
793245b5ba3SStanislav Mekhanoshin   unsigned DescNumOps = MCII->get(Opc).getNumOperands();
79440f35cefSJoe Nash   if (MCII->get(Opc).TSFlags & SIInstrFlags::VOP3P) {
79540f35cefSJoe Nash     convertVOP3PDPPInst(MI);
796dcb24f93SDmitry Preobrazhensky   } else if ((MCII->get(Opc).TSFlags & SIInstrFlags::VOPC) ||
797dcb24f93SDmitry Preobrazhensky              AMDGPU::isVOPC64DPP(Opc)) {
798be1082c6SJoe Nash     convertVOPCDPPInst(MI);
7992a6532d5SDmitry Preobrazhensky   } else if (MI.getNumOperands() < DescNumOps &&
8002a6532d5SDmitry Preobrazhensky              AMDGPU::getNamedOperandIdx(Opc, AMDGPU::OpName::op_sel) != -1) {
8012a6532d5SDmitry Preobrazhensky     auto Mods = collectVOPModifiers(MI);
8022a6532d5SDmitry Preobrazhensky     insertNamedMCOperand(MI, MCOperand::createImm(Mods.OpSel),
8032a6532d5SDmitry Preobrazhensky                          AMDGPU::OpName::op_sel);
80440f35cefSJoe Nash   } else {
805245b5ba3SStanislav Mekhanoshin     // Insert dummy unused src modifiers.
806245b5ba3SStanislav Mekhanoshin     if (MI.getNumOperands() < DescNumOps &&
807245b5ba3SStanislav Mekhanoshin         AMDGPU::getNamedOperandIdx(Opc, AMDGPU::OpName::src0_modifiers) != -1)
808245b5ba3SStanislav Mekhanoshin       insertNamedMCOperand(MI, MCOperand::createImm(0),
809245b5ba3SStanislav Mekhanoshin                            AMDGPU::OpName::src0_modifiers);
810245b5ba3SStanislav Mekhanoshin 
811245b5ba3SStanislav Mekhanoshin     if (MI.getNumOperands() < DescNumOps &&
812245b5ba3SStanislav Mekhanoshin         AMDGPU::getNamedOperandIdx(Opc, AMDGPU::OpName::src1_modifiers) != -1)
813245b5ba3SStanislav Mekhanoshin       insertNamedMCOperand(MI, MCOperand::createImm(0),
814245b5ba3SStanislav Mekhanoshin                            AMDGPU::OpName::src1_modifiers);
81540f35cefSJoe Nash   }
816245b5ba3SStanislav Mekhanoshin   return isValidDPP8(MI) ? MCDisassembler::Success : MCDisassembler::SoftFail;
817245b5ba3SStanislav Mekhanoshin }
818245b5ba3SStanislav Mekhanoshin 
convertVOP3DPPInst(MCInst & MI) const8192a6532d5SDmitry Preobrazhensky DecodeStatus AMDGPUDisassembler::convertVOP3DPPInst(MCInst &MI) const {
8202a6532d5SDmitry Preobrazhensky   unsigned Opc = MI.getOpcode();
8212a6532d5SDmitry Preobrazhensky   unsigned DescNumOps = MCII->get(Opc).getNumOperands();
8222a6532d5SDmitry Preobrazhensky   if (MI.getNumOperands() < DescNumOps &&
8232a6532d5SDmitry Preobrazhensky       AMDGPU::getNamedOperandIdx(Opc, AMDGPU::OpName::op_sel) != -1) {
8242a6532d5SDmitry Preobrazhensky     auto Mods = collectVOPModifiers(MI);
8252a6532d5SDmitry Preobrazhensky     insertNamedMCOperand(MI, MCOperand::createImm(Mods.OpSel),
8262a6532d5SDmitry Preobrazhensky                          AMDGPU::OpName::op_sel);
8272a6532d5SDmitry Preobrazhensky   }
8282a6532d5SDmitry Preobrazhensky   return MCDisassembler::Success;
8292a6532d5SDmitry Preobrazhensky }
8302a6532d5SDmitry Preobrazhensky 
831692560dcSStanislav Mekhanoshin // Note that before gfx10, the MIMG encoding provided no information about
832692560dcSStanislav Mekhanoshin // VADDR size. Consequently, decoded instructions always show address as if it
833692560dcSStanislav Mekhanoshin // has 1 dword, which could be not really so.
convertMIMGInst(MCInst & MI) const834cad7fa85SMatt Arsenault DecodeStatus AMDGPUDisassembler::convertMIMGInst(MCInst &MI) const {
835da4a7c01SDmitry Preobrazhensky 
8360b4eb1eaSDmitry Preobrazhensky   int VDstIdx = AMDGPU::getNamedOperandIdx(MI.getOpcode(),
8370b4eb1eaSDmitry Preobrazhensky                                            AMDGPU::OpName::vdst);
8380b4eb1eaSDmitry Preobrazhensky 
839cad7fa85SMatt Arsenault   int VDataIdx = AMDGPU::getNamedOperandIdx(MI.getOpcode(),
840cad7fa85SMatt Arsenault                                             AMDGPU::OpName::vdata);
841692560dcSStanislav Mekhanoshin   int VAddr0Idx =
842692560dcSStanislav Mekhanoshin       AMDGPU::getNamedOperandIdx(MI.getOpcode(), AMDGPU::OpName::vaddr0);
843cad7fa85SMatt Arsenault   int DMaskIdx = AMDGPU::getNamedOperandIdx(MI.getOpcode(),
844cad7fa85SMatt Arsenault                                             AMDGPU::OpName::dmask);
8450b4eb1eaSDmitry Preobrazhensky 
8460a1ff464SDmitry Preobrazhensky   int TFEIdx   = AMDGPU::getNamedOperandIdx(MI.getOpcode(),
8470a1ff464SDmitry Preobrazhensky                                             AMDGPU::OpName::tfe);
848f2674319SNicolai Haehnle   int D16Idx   = AMDGPU::getNamedOperandIdx(MI.getOpcode(),
849f2674319SNicolai Haehnle                                             AMDGPU::OpName::d16);
8500a1ff464SDmitry Preobrazhensky 
85199c790dcSCarl Ritson   const AMDGPU::MIMGInfo *Info = AMDGPU::getMIMGInfo(MI.getOpcode());
85299c790dcSCarl Ritson   const AMDGPU::MIMGBaseOpcodeInfo *BaseOpcode =
85399c790dcSCarl Ritson       AMDGPU::getMIMGBaseOpcodeInfo(Info->BaseOpcode);
85499c790dcSCarl Ritson 
8550b4eb1eaSDmitry Preobrazhensky   assert(VDataIdx != -1);
85699c790dcSCarl Ritson   if (BaseOpcode->BVH) {
85799c790dcSCarl Ritson     // Add A16 operand for intersect_ray instructions
85891f503c3SStanislav Mekhanoshin     if (AMDGPU::getNamedOperandIdx(MI.getOpcode(), AMDGPU::OpName::a16) > -1) {
85991f503c3SStanislav Mekhanoshin       addOperand(MI, MCOperand::createImm(1));
86091f503c3SStanislav Mekhanoshin     }
86191f503c3SStanislav Mekhanoshin     return MCDisassembler::Success;
86291f503c3SStanislav Mekhanoshin   }
8630b4eb1eaSDmitry Preobrazhensky 
864da4a7c01SDmitry Preobrazhensky   bool IsAtomic = (VDstIdx != -1);
865f2674319SNicolai Haehnle   bool IsGather4 = MCII->get(MI.getOpcode()).TSFlags & SIInstrFlags::Gather4;
866692560dcSStanislav Mekhanoshin   bool IsNSA = false;
867692560dcSStanislav Mekhanoshin   unsigned AddrSize = Info->VAddrDwords;
868cad7fa85SMatt Arsenault 
869e8860beeSJoe Nash   if (isGFX10Plus()) {
870692560dcSStanislav Mekhanoshin     unsigned DimIdx =
871692560dcSStanislav Mekhanoshin         AMDGPU::getNamedOperandIdx(MI.getOpcode(), AMDGPU::OpName::dim);
87272d570caSDavid Stuttard     int A16Idx =
87372d570caSDavid Stuttard         AMDGPU::getNamedOperandIdx(MI.getOpcode(), AMDGPU::OpName::a16);
874692560dcSStanislav Mekhanoshin     const AMDGPU::MIMGDimInfo *Dim =
875692560dcSStanislav Mekhanoshin         AMDGPU::getMIMGDimInfoByEncoding(MI.getOperand(DimIdx).getImm());
87672d570caSDavid Stuttard     const bool IsA16 = (A16Idx != -1 && MI.getOperand(A16Idx).getImm());
877692560dcSStanislav Mekhanoshin 
87872d570caSDavid Stuttard     AddrSize =
87972d570caSDavid Stuttard         AMDGPU::getAddrSizeMIMGOp(BaseOpcode, Dim, IsA16, AMDGPU::hasG16(STI));
88072d570caSDavid Stuttard 
881e8860beeSJoe Nash     IsNSA = Info->MIMGEncoding == AMDGPU::MIMGEncGfx10NSA ||
882e8860beeSJoe Nash             Info->MIMGEncoding == AMDGPU::MIMGEncGfx11NSA;
883692560dcSStanislav Mekhanoshin     if (!IsNSA) {
884692560dcSStanislav Mekhanoshin       if (AddrSize > 8)
885692560dcSStanislav Mekhanoshin         AddrSize = 16;
886692560dcSStanislav Mekhanoshin     } else {
887692560dcSStanislav Mekhanoshin       if (AddrSize > Info->VAddrDwords) {
888692560dcSStanislav Mekhanoshin         // The NSA encoding does not contain enough operands for the combination
889692560dcSStanislav Mekhanoshin         // of base opcode / dimension. Should this be an error?
8900a1ff464SDmitry Preobrazhensky         return MCDisassembler::Success;
891692560dcSStanislav Mekhanoshin       }
892692560dcSStanislav Mekhanoshin     }
893692560dcSStanislav Mekhanoshin   }
894692560dcSStanislav Mekhanoshin 
895692560dcSStanislav Mekhanoshin   unsigned DMask = MI.getOperand(DMaskIdx).getImm() & 0xf;
896692560dcSStanislav Mekhanoshin   unsigned DstSize = IsGather4 ? 4 : std::max(countPopulation(DMask), 1u);
8970a1ff464SDmitry Preobrazhensky 
898f2674319SNicolai Haehnle   bool D16 = D16Idx >= 0 && MI.getOperand(D16Idx).getImm();
8990a1ff464SDmitry Preobrazhensky   if (D16 && AMDGPU::hasPackedD16(STI)) {
9000a1ff464SDmitry Preobrazhensky     DstSize = (DstSize + 1) / 2;
9010a1ff464SDmitry Preobrazhensky   }
9020a1ff464SDmitry Preobrazhensky 
903a8d9d507SStanislav Mekhanoshin   if (TFEIdx != -1 && MI.getOperand(TFEIdx).getImm())
9044ab704d6SPetar Avramovic     DstSize += 1;
905cad7fa85SMatt Arsenault 
906692560dcSStanislav Mekhanoshin   if (DstSize == Info->VDataDwords && AddrSize == Info->VAddrDwords)
907f2674319SNicolai Haehnle     return MCDisassembler::Success;
908692560dcSStanislav Mekhanoshin 
909692560dcSStanislav Mekhanoshin   int NewOpcode =
910692560dcSStanislav Mekhanoshin       AMDGPU::getMIMGOpcode(Info->BaseOpcode, Info->MIMGEncoding, DstSize, AddrSize);
9110ab200b6SNicolai Haehnle   if (NewOpcode == -1)
9120ab200b6SNicolai Haehnle     return MCDisassembler::Success;
9130b4eb1eaSDmitry Preobrazhensky 
914692560dcSStanislav Mekhanoshin   // Widen the register to the correct number of enabled channels.
915692560dcSStanislav Mekhanoshin   unsigned NewVdata = AMDGPU::NoRegister;
916692560dcSStanislav Mekhanoshin   if (DstSize != Info->VDataDwords) {
917692560dcSStanislav Mekhanoshin     auto DataRCID = MCII->get(NewOpcode).OpInfo[VDataIdx].RegClass;
918cad7fa85SMatt Arsenault 
9190b4eb1eaSDmitry Preobrazhensky     // Get first subregister of VData
920cad7fa85SMatt Arsenault     unsigned Vdata0 = MI.getOperand(VDataIdx).getReg();
9210b4eb1eaSDmitry Preobrazhensky     unsigned VdataSub0 = MRI.getSubReg(Vdata0, AMDGPU::sub0);
9220b4eb1eaSDmitry Preobrazhensky     Vdata0 = (VdataSub0 != 0)? VdataSub0 : Vdata0;
9230b4eb1eaSDmitry Preobrazhensky 
924692560dcSStanislav Mekhanoshin     NewVdata = MRI.getMatchingSuperReg(Vdata0, AMDGPU::sub0,
925692560dcSStanislav Mekhanoshin                                        &MRI.getRegClass(DataRCID));
926cad7fa85SMatt Arsenault     if (NewVdata == AMDGPU::NoRegister) {
927cad7fa85SMatt Arsenault       // It's possible to encode this such that the low register + enabled
928cad7fa85SMatt Arsenault       // components exceeds the register count.
929cad7fa85SMatt Arsenault       return MCDisassembler::Success;
930cad7fa85SMatt Arsenault     }
931692560dcSStanislav Mekhanoshin   }
932692560dcSStanislav Mekhanoshin 
933e8860beeSJoe Nash   // If not using NSA on GFX10+, widen address register to correct size.
934692560dcSStanislav Mekhanoshin   unsigned NewVAddr0 = AMDGPU::NoRegister;
935e8860beeSJoe Nash   if (isGFX10Plus() && !IsNSA && AddrSize != Info->VAddrDwords) {
936692560dcSStanislav Mekhanoshin     unsigned VAddr0 = MI.getOperand(VAddr0Idx).getReg();
937692560dcSStanislav Mekhanoshin     unsigned VAddrSub0 = MRI.getSubReg(VAddr0, AMDGPU::sub0);
938692560dcSStanislav Mekhanoshin     VAddr0 = (VAddrSub0 != 0) ? VAddrSub0 : VAddr0;
939692560dcSStanislav Mekhanoshin 
940692560dcSStanislav Mekhanoshin     auto AddrRCID = MCII->get(NewOpcode).OpInfo[VAddr0Idx].RegClass;
941692560dcSStanislav Mekhanoshin     NewVAddr0 = MRI.getMatchingSuperReg(VAddr0, AMDGPU::sub0,
942692560dcSStanislav Mekhanoshin                                         &MRI.getRegClass(AddrRCID));
943692560dcSStanislav Mekhanoshin     if (NewVAddr0 == AMDGPU::NoRegister)
944692560dcSStanislav Mekhanoshin       return MCDisassembler::Success;
945692560dcSStanislav Mekhanoshin   }
946cad7fa85SMatt Arsenault 
947cad7fa85SMatt Arsenault   MI.setOpcode(NewOpcode);
948692560dcSStanislav Mekhanoshin 
949692560dcSStanislav Mekhanoshin   if (NewVdata != AMDGPU::NoRegister) {
950cad7fa85SMatt Arsenault     MI.getOperand(VDataIdx) = MCOperand::createReg(NewVdata);
9510b4eb1eaSDmitry Preobrazhensky 
952da4a7c01SDmitry Preobrazhensky     if (IsAtomic) {
9530b4eb1eaSDmitry Preobrazhensky       // Atomic operations have an additional operand (a copy of data)
9540b4eb1eaSDmitry Preobrazhensky       MI.getOperand(VDstIdx) = MCOperand::createReg(NewVdata);
9550b4eb1eaSDmitry Preobrazhensky     }
956692560dcSStanislav Mekhanoshin   }
957692560dcSStanislav Mekhanoshin 
958692560dcSStanislav Mekhanoshin   if (NewVAddr0 != AMDGPU::NoRegister) {
959692560dcSStanislav Mekhanoshin     MI.getOperand(VAddr0Idx) = MCOperand::createReg(NewVAddr0);
960692560dcSStanislav Mekhanoshin   } else if (IsNSA) {
961692560dcSStanislav Mekhanoshin     assert(AddrSize <= Info->VAddrDwords);
962692560dcSStanislav Mekhanoshin     MI.erase(MI.begin() + VAddr0Idx + AddrSize,
963692560dcSStanislav Mekhanoshin              MI.begin() + VAddr0Idx + Info->VAddrDwords);
964692560dcSStanislav Mekhanoshin   }
9650b4eb1eaSDmitry Preobrazhensky 
966cad7fa85SMatt Arsenault   return MCDisassembler::Success;
967cad7fa85SMatt Arsenault }
968cad7fa85SMatt Arsenault 
96940f35cefSJoe Nash // Opsel and neg bits are used in src_modifiers and standalone operands. Autogen
97040f35cefSJoe Nash // decoder only adds to src_modifiers, so manually add the bits to the other
97140f35cefSJoe Nash // operands.
convertVOP3PDPPInst(MCInst & MI) const97240f35cefSJoe Nash DecodeStatus AMDGPUDisassembler::convertVOP3PDPPInst(MCInst &MI) const {
97340f35cefSJoe Nash   unsigned Opc = MI.getOpcode();
97440f35cefSJoe Nash   unsigned DescNumOps = MCII->get(Opc).getNumOperands();
9752a6532d5SDmitry Preobrazhensky   auto Mods = collectVOPModifiers(MI, true);
97640f35cefSJoe Nash 
97740f35cefSJoe Nash   if (MI.getNumOperands() < DescNumOps &&
97840f35cefSJoe Nash       AMDGPU::getNamedOperandIdx(Opc, AMDGPU::OpName::vdst_in) != -1)
97940f35cefSJoe Nash     insertNamedMCOperand(MI, MCOperand::createImm(0), AMDGPU::OpName::vdst_in);
98040f35cefSJoe Nash 
98140f35cefSJoe Nash   if (MI.getNumOperands() < DescNumOps &&
98240f35cefSJoe Nash       AMDGPU::getNamedOperandIdx(Opc, AMDGPU::OpName::op_sel) != -1)
9832a6532d5SDmitry Preobrazhensky     insertNamedMCOperand(MI, MCOperand::createImm(Mods.OpSel),
98440f35cefSJoe Nash                          AMDGPU::OpName::op_sel);
98540f35cefSJoe Nash   if (MI.getNumOperands() < DescNumOps &&
98640f35cefSJoe Nash       AMDGPU::getNamedOperandIdx(Opc, AMDGPU::OpName::op_sel_hi) != -1)
9872a6532d5SDmitry Preobrazhensky     insertNamedMCOperand(MI, MCOperand::createImm(Mods.OpSelHi),
98840f35cefSJoe Nash                          AMDGPU::OpName::op_sel_hi);
98940f35cefSJoe Nash   if (MI.getNumOperands() < DescNumOps &&
99040f35cefSJoe Nash       AMDGPU::getNamedOperandIdx(Opc, AMDGPU::OpName::neg_lo) != -1)
9912a6532d5SDmitry Preobrazhensky     insertNamedMCOperand(MI, MCOperand::createImm(Mods.NegLo),
99240f35cefSJoe Nash                          AMDGPU::OpName::neg_lo);
99340f35cefSJoe Nash   if (MI.getNumOperands() < DescNumOps &&
99440f35cefSJoe Nash       AMDGPU::getNamedOperandIdx(Opc, AMDGPU::OpName::neg_hi) != -1)
9952a6532d5SDmitry Preobrazhensky     insertNamedMCOperand(MI, MCOperand::createImm(Mods.NegHi),
99640f35cefSJoe Nash                          AMDGPU::OpName::neg_hi);
99740f35cefSJoe Nash 
99840f35cefSJoe Nash   return MCDisassembler::Success;
99940f35cefSJoe Nash }
100040f35cefSJoe Nash 
1001be1082c6SJoe Nash // Create dummy old operand and insert optional operands
convertVOPCDPPInst(MCInst & MI) const1002be1082c6SJoe Nash DecodeStatus AMDGPUDisassembler::convertVOPCDPPInst(MCInst &MI) const {
1003be1082c6SJoe Nash   unsigned Opc = MI.getOpcode();
1004be1082c6SJoe Nash   unsigned DescNumOps = MCII->get(Opc).getNumOperands();
1005be1082c6SJoe Nash 
1006be1082c6SJoe Nash   if (MI.getNumOperands() < DescNumOps &&
1007be1082c6SJoe Nash       AMDGPU::getNamedOperandIdx(Opc, AMDGPU::OpName::old) != -1)
1008be1082c6SJoe Nash     insertNamedMCOperand(MI, MCOperand::createReg(0), AMDGPU::OpName::old);
1009be1082c6SJoe Nash 
1010be1082c6SJoe Nash   if (MI.getNumOperands() < DescNumOps &&
1011be1082c6SJoe Nash       AMDGPU::getNamedOperandIdx(Opc, AMDGPU::OpName::src0_modifiers) != -1)
1012be1082c6SJoe Nash     insertNamedMCOperand(MI, MCOperand::createImm(0),
1013be1082c6SJoe Nash                          AMDGPU::OpName::src0_modifiers);
1014be1082c6SJoe Nash 
1015be1082c6SJoe Nash   if (MI.getNumOperands() < DescNumOps &&
1016be1082c6SJoe Nash       AMDGPU::getNamedOperandIdx(Opc, AMDGPU::OpName::src1_modifiers) != -1)
1017be1082c6SJoe Nash     insertNamedMCOperand(MI, MCOperand::createImm(0),
1018be1082c6SJoe Nash                          AMDGPU::OpName::src1_modifiers);
1019be1082c6SJoe Nash   return MCDisassembler::Success;
1020be1082c6SJoe Nash }
1021be1082c6SJoe Nash 
convertFMAanyK(MCInst & MI,int ImmLitIdx) const1022b4b7e605SJoe Nash DecodeStatus AMDGPUDisassembler::convertFMAanyK(MCInst &MI,
1023b4b7e605SJoe Nash                                                 int ImmLitIdx) const {
1024b4b7e605SJoe Nash   assert(HasLiteral && "Should have decoded a literal");
1025b4b7e605SJoe Nash   const MCInstrDesc &Desc = MCII->get(MI.getOpcode());
1026b4b7e605SJoe Nash   unsigned DescNumOps = Desc.getNumOperands();
102707b7fadaSJoe Nash   insertNamedMCOperand(MI, MCOperand::createImm(Literal),
102807b7fadaSJoe Nash                        AMDGPU::OpName::immDeferred);
1029b4b7e605SJoe Nash   assert(DescNumOps == MI.getNumOperands());
1030b4b7e605SJoe Nash   for (unsigned I = 0; I < DescNumOps; ++I) {
1031b4b7e605SJoe Nash     auto &Op = MI.getOperand(I);
1032b4b7e605SJoe Nash     auto OpType = Desc.OpInfo[I].OperandType;
1033b4b7e605SJoe Nash     bool IsDeferredOp = (OpType == AMDGPU::OPERAND_REG_IMM_FP32_DEFERRED ||
1034b4b7e605SJoe Nash                          OpType == AMDGPU::OPERAND_REG_IMM_FP16_DEFERRED);
1035b4b7e605SJoe Nash     if (Op.isImm() && Op.getImm() == AMDGPU::EncValues::LITERAL_CONST &&
1036b4b7e605SJoe Nash         IsDeferredOp)
1037b4b7e605SJoe Nash       Op.setImm(Literal);
1038b4b7e605SJoe Nash   }
1039b4b7e605SJoe Nash   return MCDisassembler::Success;
1040b4b7e605SJoe Nash }
1041b4b7e605SJoe Nash 
getRegClassName(unsigned RegClassID) const1042ac106addSNikolay Haustov const char* AMDGPUDisassembler::getRegClassName(unsigned RegClassID) const {
1043ac106addSNikolay Haustov   return getContext().getRegisterInfo()->
1044ac106addSNikolay Haustov     getRegClassName(&AMDGPUMCRegisterClasses[RegClassID]);
1045e1818af8STom Stellard }
1046e1818af8STom Stellard 
1047ac106addSNikolay Haustov inline
errOperand(unsigned V,const Twine & ErrMsg) const1048ac106addSNikolay Haustov MCOperand AMDGPUDisassembler::errOperand(unsigned V,
1049ac106addSNikolay Haustov                                          const Twine& ErrMsg) const {
1050ac106addSNikolay Haustov   *CommentStream << "Error: " + ErrMsg;
1051ac106addSNikolay Haustov 
1052ac106addSNikolay Haustov   // ToDo: add support for error operands to MCInst.h
1053ac106addSNikolay Haustov   // return MCOperand::createError(V);
1054ac106addSNikolay Haustov   return MCOperand();
1055ac106addSNikolay Haustov }
1056ac106addSNikolay Haustov 
1057ac106addSNikolay Haustov inline
createRegOperand(unsigned int RegId) const1058ac106addSNikolay Haustov MCOperand AMDGPUDisassembler::createRegOperand(unsigned int RegId) const {
1059ac2b0264SDmitry Preobrazhensky   return MCOperand::createReg(AMDGPU::getMCReg(RegId, STI));
1060ac106addSNikolay Haustov }
1061ac106addSNikolay Haustov 
1062ac106addSNikolay Haustov inline
createRegOperand(unsigned RegClassID,unsigned Val) const1063ac106addSNikolay Haustov MCOperand AMDGPUDisassembler::createRegOperand(unsigned RegClassID,
1064ac106addSNikolay Haustov                                                unsigned Val) const {
1065ac106addSNikolay Haustov   const auto& RegCl = AMDGPUMCRegisterClasses[RegClassID];
1066ac106addSNikolay Haustov   if (Val >= RegCl.getNumRegs())
1067ac106addSNikolay Haustov     return errOperand(Val, Twine(getRegClassName(RegClassID)) +
1068ac106addSNikolay Haustov                            ": unknown register " + Twine(Val));
1069ac106addSNikolay Haustov   return createRegOperand(RegCl.getRegister(Val));
1070ac106addSNikolay Haustov }
1071ac106addSNikolay Haustov 
1072ac106addSNikolay Haustov inline
createSRegOperand(unsigned SRegClassID,unsigned Val) const1073ac106addSNikolay Haustov MCOperand AMDGPUDisassembler::createSRegOperand(unsigned SRegClassID,
1074ac106addSNikolay Haustov                                                 unsigned Val) const {
1075ac106addSNikolay Haustov   // ToDo: SI/CI have 104 SGPRs, VI - 102
1076ac106addSNikolay Haustov   // Valery: here we accepting as much as we can, let assembler sort it out
1077ac106addSNikolay Haustov   int shift = 0;
1078ac106addSNikolay Haustov   switch (SRegClassID) {
1079ac106addSNikolay Haustov   case AMDGPU::SGPR_32RegClassID:
1080212a251cSArtem Tamazov   case AMDGPU::TTMP_32RegClassID:
1081212a251cSArtem Tamazov     break;
1082ac106addSNikolay Haustov   case AMDGPU::SGPR_64RegClassID:
1083212a251cSArtem Tamazov   case AMDGPU::TTMP_64RegClassID:
1084212a251cSArtem Tamazov     shift = 1;
1085212a251cSArtem Tamazov     break;
1086212a251cSArtem Tamazov   case AMDGPU::SGPR_128RegClassID:
1087212a251cSArtem Tamazov   case AMDGPU::TTMP_128RegClassID:
1088ac106addSNikolay Haustov   // ToDo: unclear if s[100:104] is available on VI. Can we use VCC as SGPR in
1089ac106addSNikolay Haustov   // this bundle?
109027134953SDmitry Preobrazhensky   case AMDGPU::SGPR_256RegClassID:
109127134953SDmitry Preobrazhensky   case AMDGPU::TTMP_256RegClassID:
1092ac106addSNikolay Haustov     // ToDo: unclear if s[96:104] is available on VI. Can we use VCC as SGPR in
1093ac106addSNikolay Haustov   // this bundle?
109427134953SDmitry Preobrazhensky   case AMDGPU::SGPR_512RegClassID:
109527134953SDmitry Preobrazhensky   case AMDGPU::TTMP_512RegClassID:
1096212a251cSArtem Tamazov     shift = 2;
1097212a251cSArtem Tamazov     break;
1098ac106addSNikolay Haustov   // ToDo: unclear if s[88:104] is available on VI. Can we use VCC as SGPR in
1099ac106addSNikolay Haustov   // this bundle?
1100212a251cSArtem Tamazov   default:
110192b355b1SMatt Arsenault     llvm_unreachable("unhandled register class");
1102ac106addSNikolay Haustov   }
110392b355b1SMatt Arsenault 
110492b355b1SMatt Arsenault   if (Val % (1 << shift)) {
1105ac106addSNikolay Haustov     *CommentStream << "Warning: " << getRegClassName(SRegClassID)
1106ac106addSNikolay Haustov                    << ": scalar reg isn't aligned " << Val;
110792b355b1SMatt Arsenault   }
110892b355b1SMatt Arsenault 
1109ac106addSNikolay Haustov   return createRegOperand(SRegClassID, Val >> shift);
1110ac106addSNikolay Haustov }
1111ac106addSNikolay Haustov 
decodeOperand_VS_32(unsigned Val) const1112ac106addSNikolay Haustov MCOperand AMDGPUDisassembler::decodeOperand_VS_32(unsigned Val) const {
1113212a251cSArtem Tamazov   return decodeSrcOp(OPW32, Val);
1114ac106addSNikolay Haustov }
1115ac106addSNikolay Haustov 
decodeOperand_VS_64(unsigned Val) const1116ac106addSNikolay Haustov MCOperand AMDGPUDisassembler::decodeOperand_VS_64(unsigned Val) const {
1117212a251cSArtem Tamazov   return decodeSrcOp(OPW64, Val);
1118ac106addSNikolay Haustov }
1119ac106addSNikolay Haustov 
decodeOperand_VS_128(unsigned Val) const112030fc5239SDmitry Preobrazhensky MCOperand AMDGPUDisassembler::decodeOperand_VS_128(unsigned Val) const {
112130fc5239SDmitry Preobrazhensky   return decodeSrcOp(OPW128, Val);
112230fc5239SDmitry Preobrazhensky }
112330fc5239SDmitry Preobrazhensky 
decodeOperand_VSrc16(unsigned Val) const11244bd72361SMatt Arsenault MCOperand AMDGPUDisassembler::decodeOperand_VSrc16(unsigned Val) const {
11254bd72361SMatt Arsenault   return decodeSrcOp(OPW16, Val);
11264bd72361SMatt Arsenault }
11274bd72361SMatt Arsenault 
decodeOperand_VSrcV216(unsigned Val) const11289be7b0d4SMatt Arsenault MCOperand AMDGPUDisassembler::decodeOperand_VSrcV216(unsigned Val) const {
11299be7b0d4SMatt Arsenault   return decodeSrcOp(OPWV216, Val);
11309be7b0d4SMatt Arsenault }
11319be7b0d4SMatt Arsenault 
decodeOperand_VSrcV232(unsigned Val) const1132a8d9d507SStanislav Mekhanoshin MCOperand AMDGPUDisassembler::decodeOperand_VSrcV232(unsigned Val) const {
1133a8d9d507SStanislav Mekhanoshin   return decodeSrcOp(OPWV232, Val);
1134a8d9d507SStanislav Mekhanoshin }
1135a8d9d507SStanislav Mekhanoshin 
decodeOperand_VGPR_32(unsigned Val) const1136ac106addSNikolay Haustov MCOperand AMDGPUDisassembler::decodeOperand_VGPR_32(unsigned Val) const {
1137cb540bc0SMatt Arsenault   // Some instructions have operand restrictions beyond what the encoding
1138cb540bc0SMatt Arsenault   // allows. Some ordinarily VSrc_32 operands are VGPR_32, so clear the extra
1139cb540bc0SMatt Arsenault   // high bit.
1140cb540bc0SMatt Arsenault   Val &= 255;
1141cb540bc0SMatt Arsenault 
1142ac106addSNikolay Haustov   return createRegOperand(AMDGPU::VGPR_32RegClassID, Val);
1143ac106addSNikolay Haustov }
1144ac106addSNikolay Haustov 
decodeOperand_VRegOrLds_32(unsigned Val) const11456023d599SDmitry Preobrazhensky MCOperand AMDGPUDisassembler::decodeOperand_VRegOrLds_32(unsigned Val) const {
11466023d599SDmitry Preobrazhensky   return decodeSrcOp(OPW32, Val);
11476023d599SDmitry Preobrazhensky }
11486023d599SDmitry Preobrazhensky 
decodeOperand_AGPR_32(unsigned Val) const11499e77d0c6SStanislav Mekhanoshin MCOperand AMDGPUDisassembler::decodeOperand_AGPR_32(unsigned Val) const {
11509e77d0c6SStanislav Mekhanoshin   return createRegOperand(AMDGPU::AGPR_32RegClassID, Val & 255);
11519e77d0c6SStanislav Mekhanoshin }
11529e77d0c6SStanislav Mekhanoshin 
decodeOperand_AReg_64(unsigned Val) const1153a8d9d507SStanislav Mekhanoshin MCOperand AMDGPUDisassembler::decodeOperand_AReg_64(unsigned Val) const {
1154a8d9d507SStanislav Mekhanoshin   return createRegOperand(AMDGPU::AReg_64RegClassID, Val & 255);
1155a8d9d507SStanislav Mekhanoshin }
1156a8d9d507SStanislav Mekhanoshin 
decodeOperand_AReg_128(unsigned Val) const11579e77d0c6SStanislav Mekhanoshin MCOperand AMDGPUDisassembler::decodeOperand_AReg_128(unsigned Val) const {
11589e77d0c6SStanislav Mekhanoshin   return createRegOperand(AMDGPU::AReg_128RegClassID, Val & 255);
11599e77d0c6SStanislav Mekhanoshin }
11609e77d0c6SStanislav Mekhanoshin 
decodeOperand_AReg_256(unsigned Val) const1161a8d9d507SStanislav Mekhanoshin MCOperand AMDGPUDisassembler::decodeOperand_AReg_256(unsigned Val) const {
1162a8d9d507SStanislav Mekhanoshin   return createRegOperand(AMDGPU::AReg_256RegClassID, Val & 255);
1163a8d9d507SStanislav Mekhanoshin }
1164a8d9d507SStanislav Mekhanoshin 
decodeOperand_AReg_512(unsigned Val) const11659e77d0c6SStanislav Mekhanoshin MCOperand AMDGPUDisassembler::decodeOperand_AReg_512(unsigned Val) const {
11669e77d0c6SStanislav Mekhanoshin   return createRegOperand(AMDGPU::AReg_512RegClassID, Val & 255);
11679e77d0c6SStanislav Mekhanoshin }
11689e77d0c6SStanislav Mekhanoshin 
decodeOperand_AReg_1024(unsigned Val) const11699e77d0c6SStanislav Mekhanoshin MCOperand AMDGPUDisassembler::decodeOperand_AReg_1024(unsigned Val) const {
11709e77d0c6SStanislav Mekhanoshin   return createRegOperand(AMDGPU::AReg_1024RegClassID, Val & 255);
11719e77d0c6SStanislav Mekhanoshin }
11729e77d0c6SStanislav Mekhanoshin 
decodeOperand_AV_32(unsigned Val) const11739e77d0c6SStanislav Mekhanoshin MCOperand AMDGPUDisassembler::decodeOperand_AV_32(unsigned Val) const {
11749e77d0c6SStanislav Mekhanoshin   return decodeSrcOp(OPW32, Val);
11759e77d0c6SStanislav Mekhanoshin }
11769e77d0c6SStanislav Mekhanoshin 
decodeOperand_AV_64(unsigned Val) const11779e77d0c6SStanislav Mekhanoshin MCOperand AMDGPUDisassembler::decodeOperand_AV_64(unsigned Val) const {
11789e77d0c6SStanislav Mekhanoshin   return decodeSrcOp(OPW64, Val);
11799e77d0c6SStanislav Mekhanoshin }
11809e77d0c6SStanislav Mekhanoshin 
decodeOperand_AV_128(unsigned Val) const11816e3e14f6SStanislav Mekhanoshin MCOperand AMDGPUDisassembler::decodeOperand_AV_128(unsigned Val) const {
11826e3e14f6SStanislav Mekhanoshin   return decodeSrcOp(OPW128, Val);
11836e3e14f6SStanislav Mekhanoshin }
11846e3e14f6SStanislav Mekhanoshin 
decodeOperand_AVDst_128(unsigned Val) const118532ca9bd7SDmitry Preobrazhensky MCOperand AMDGPUDisassembler::decodeOperand_AVDst_128(unsigned Val) const {
118632ca9bd7SDmitry Preobrazhensky   using namespace AMDGPU::EncValues;
118732ca9bd7SDmitry Preobrazhensky   assert((Val & IS_VGPR) == 0); // Val{8} is not encoded but assumed to be 1.
118832ca9bd7SDmitry Preobrazhensky   return decodeSrcOp(OPW128, Val | IS_VGPR);
118932ca9bd7SDmitry Preobrazhensky }
119032ca9bd7SDmitry Preobrazhensky 
decodeOperand_AVDst_512(unsigned Val) const119132ca9bd7SDmitry Preobrazhensky MCOperand AMDGPUDisassembler::decodeOperand_AVDst_512(unsigned Val) const {
119232ca9bd7SDmitry Preobrazhensky   using namespace AMDGPU::EncValues;
119332ca9bd7SDmitry Preobrazhensky   assert((Val & IS_VGPR) == 0); // Val{8} is not encoded but assumed to be 1.
119432ca9bd7SDmitry Preobrazhensky   return decodeSrcOp(OPW512, Val | IS_VGPR);
11956e3e14f6SStanislav Mekhanoshin }
11966e3e14f6SStanislav Mekhanoshin 
decodeOperand_VReg_64(unsigned Val) const1197ac106addSNikolay Haustov MCOperand AMDGPUDisassembler::decodeOperand_VReg_64(unsigned Val) const {
1198ac106addSNikolay Haustov   return createRegOperand(AMDGPU::VReg_64RegClassID, Val);
1199ac106addSNikolay Haustov }
1200ac106addSNikolay Haustov 
decodeOperand_VReg_96(unsigned Val) const1201ac106addSNikolay Haustov MCOperand AMDGPUDisassembler::decodeOperand_VReg_96(unsigned Val) const {
1202ac106addSNikolay Haustov   return createRegOperand(AMDGPU::VReg_96RegClassID, Val);
1203ac106addSNikolay Haustov }
1204ac106addSNikolay Haustov 
decodeOperand_VReg_128(unsigned Val) const1205ac106addSNikolay Haustov MCOperand AMDGPUDisassembler::decodeOperand_VReg_128(unsigned Val) const {
1206ac106addSNikolay Haustov   return createRegOperand(AMDGPU::VReg_128RegClassID, Val);
1207ac106addSNikolay Haustov }
1208ac106addSNikolay Haustov 
decodeOperand_VReg_256(unsigned Val) const12099e77d0c6SStanislav Mekhanoshin MCOperand AMDGPUDisassembler::decodeOperand_VReg_256(unsigned Val) const {
12109e77d0c6SStanislav Mekhanoshin   return createRegOperand(AMDGPU::VReg_256RegClassID, Val);
12119e77d0c6SStanislav Mekhanoshin }
12129e77d0c6SStanislav Mekhanoshin 
decodeOperand_VReg_512(unsigned Val) const12139e77d0c6SStanislav Mekhanoshin MCOperand AMDGPUDisassembler::decodeOperand_VReg_512(unsigned Val) const {
12149e77d0c6SStanislav Mekhanoshin   return createRegOperand(AMDGPU::VReg_512RegClassID, Val);
12159e77d0c6SStanislav Mekhanoshin }
12169e77d0c6SStanislav Mekhanoshin 
decodeOperand_VReg_1024(unsigned Val) const1217a8d9d507SStanislav Mekhanoshin MCOperand AMDGPUDisassembler::decodeOperand_VReg_1024(unsigned Val) const {
1218a8d9d507SStanislav Mekhanoshin   return createRegOperand(AMDGPU::VReg_1024RegClassID, Val);
1219a8d9d507SStanislav Mekhanoshin }
1220a8d9d507SStanislav Mekhanoshin 
decodeOperand_SReg_32(unsigned Val) const1221ac106addSNikolay Haustov MCOperand AMDGPUDisassembler::decodeOperand_SReg_32(unsigned Val) const {
1222ac106addSNikolay Haustov   // table-gen generated disassembler doesn't care about operand types
1223ac106addSNikolay Haustov   // leaving only registry class so SSrc_32 operand turns into SReg_32
1224ac106addSNikolay Haustov   // and therefore we accept immediates and literals here as well
1225212a251cSArtem Tamazov   return decodeSrcOp(OPW32, Val);
1226ac106addSNikolay Haustov }
1227ac106addSNikolay Haustov 
decodeOperand_SReg_32_XM0_XEXEC(unsigned Val) const1228640c44b8SMatt Arsenault MCOperand AMDGPUDisassembler::decodeOperand_SReg_32_XM0_XEXEC(
1229640c44b8SMatt Arsenault   unsigned Val) const {
1230640c44b8SMatt Arsenault   // SReg_32_XM0 is SReg_32 without M0 or EXEC_LO/EXEC_HI
123138e496b1SArtem Tamazov   return decodeOperand_SReg_32(Val);
123238e496b1SArtem Tamazov }
123338e496b1SArtem Tamazov 
decodeOperand_SReg_32_XEXEC_HI(unsigned Val) const1234ca7b0a17SMatt Arsenault MCOperand AMDGPUDisassembler::decodeOperand_SReg_32_XEXEC_HI(
1235ca7b0a17SMatt Arsenault   unsigned Val) const {
1236ca7b0a17SMatt Arsenault   // SReg_32_XM0 is SReg_32 without EXEC_HI
1237ca7b0a17SMatt Arsenault   return decodeOperand_SReg_32(Val);
1238ca7b0a17SMatt Arsenault }
1239ca7b0a17SMatt Arsenault 
decodeOperand_SRegOrLds_32(unsigned Val) const12406023d599SDmitry Preobrazhensky MCOperand AMDGPUDisassembler::decodeOperand_SRegOrLds_32(unsigned Val) const {
12416023d599SDmitry Preobrazhensky   // table-gen generated disassembler doesn't care about operand types
12426023d599SDmitry Preobrazhensky   // leaving only registry class so SSrc_32 operand turns into SReg_32
12436023d599SDmitry Preobrazhensky   // and therefore we accept immediates and literals here as well
12446023d599SDmitry Preobrazhensky   return decodeSrcOp(OPW32, Val);
12456023d599SDmitry Preobrazhensky }
12466023d599SDmitry Preobrazhensky 
decodeOperand_SReg_64(unsigned Val) const1247ac106addSNikolay Haustov MCOperand AMDGPUDisassembler::decodeOperand_SReg_64(unsigned Val) const {
1248640c44b8SMatt Arsenault   return decodeSrcOp(OPW64, Val);
1249640c44b8SMatt Arsenault }
1250640c44b8SMatt Arsenault 
decodeOperand_SReg_64_XEXEC(unsigned Val) const1251640c44b8SMatt Arsenault MCOperand AMDGPUDisassembler::decodeOperand_SReg_64_XEXEC(unsigned Val) const {
1252212a251cSArtem Tamazov   return decodeSrcOp(OPW64, Val);
1253ac106addSNikolay Haustov }
1254ac106addSNikolay Haustov 
decodeOperand_SReg_128(unsigned Val) const1255ac106addSNikolay Haustov MCOperand AMDGPUDisassembler::decodeOperand_SReg_128(unsigned Val) const {
1256212a251cSArtem Tamazov   return decodeSrcOp(OPW128, Val);
1257ac106addSNikolay Haustov }
1258ac106addSNikolay Haustov 
decodeOperand_SReg_256(unsigned Val) const1259ac106addSNikolay Haustov MCOperand AMDGPUDisassembler::decodeOperand_SReg_256(unsigned Val) const {
126027134953SDmitry Preobrazhensky   return decodeDstOp(OPW256, Val);
1261ac106addSNikolay Haustov }
1262ac106addSNikolay Haustov 
decodeOperand_SReg_512(unsigned Val) const1263ac106addSNikolay Haustov MCOperand AMDGPUDisassembler::decodeOperand_SReg_512(unsigned Val) const {
126427134953SDmitry Preobrazhensky   return decodeDstOp(OPW512, Val);
1265ac106addSNikolay Haustov }
1266ac106addSNikolay Haustov 
1267b4b7e605SJoe Nash // Decode Literals for insts which always have a literal in the encoding
1268b4b7e605SJoe Nash MCOperand
decodeMandatoryLiteralConstant(unsigned Val) const1269b4b7e605SJoe Nash AMDGPUDisassembler::decodeMandatoryLiteralConstant(unsigned Val) const {
1270b4b7e605SJoe Nash   if (HasLiteral) {
127107b7fadaSJoe Nash     assert(
127207b7fadaSJoe Nash         AMDGPU::hasVOPD(STI) &&
127307b7fadaSJoe Nash         "Should only decode multiple kimm with VOPD, check VSrc operand types");
1274b4b7e605SJoe Nash     if (Literal != Val)
1275b4b7e605SJoe Nash       return errOperand(Val, "More than one unique literal is illegal");
1276b4b7e605SJoe Nash   }
1277b4b7e605SJoe Nash   HasLiteral = true;
1278b4b7e605SJoe Nash   Literal = Val;
1279b4b7e605SJoe Nash   return MCOperand::createImm(Literal);
1280b4b7e605SJoe Nash }
1281b4b7e605SJoe Nash 
decodeLiteralConstant() const1282ac106addSNikolay Haustov MCOperand AMDGPUDisassembler::decodeLiteralConstant() const {
1283ac106addSNikolay Haustov   // For now all literal constants are supposed to be unsigned integer
1284ac106addSNikolay Haustov   // ToDo: deal with signed/unsigned 64-bit integer constants
1285ac106addSNikolay Haustov   // ToDo: deal with float/double constants
1286ce941c9cSDmitry Preobrazhensky   if (!HasLiteral) {
1287ce941c9cSDmitry Preobrazhensky     if (Bytes.size() < 4) {
1288ac106addSNikolay Haustov       return errOperand(0, "cannot read literal, inst bytes left " +
1289ac106addSNikolay Haustov                         Twine(Bytes.size()));
1290ce941c9cSDmitry Preobrazhensky     }
1291ce941c9cSDmitry Preobrazhensky     HasLiteral = true;
1292ce941c9cSDmitry Preobrazhensky     Literal = eatBytes<uint32_t>(Bytes);
1293ce941c9cSDmitry Preobrazhensky   }
1294ce941c9cSDmitry Preobrazhensky   return MCOperand::createImm(Literal);
1295ac106addSNikolay Haustov }
1296ac106addSNikolay Haustov 
decodeIntImmed(unsigned Imm)1297ac106addSNikolay Haustov MCOperand AMDGPUDisassembler::decodeIntImmed(unsigned Imm) {
1298212a251cSArtem Tamazov   using namespace AMDGPU::EncValues;
1299c8fbf6ffSEugene Zelenko 
1300212a251cSArtem Tamazov   assert(Imm >= INLINE_INTEGER_C_MIN && Imm <= INLINE_INTEGER_C_MAX);
1301212a251cSArtem Tamazov   return MCOperand::createImm((Imm <= INLINE_INTEGER_C_POSITIVE_MAX) ?
1302212a251cSArtem Tamazov     (static_cast<int64_t>(Imm) - INLINE_INTEGER_C_MIN) :
1303212a251cSArtem Tamazov     (INLINE_INTEGER_C_POSITIVE_MAX - static_cast<int64_t>(Imm)));
1304212a251cSArtem Tamazov       // Cast prevents negative overflow.
1305ac106addSNikolay Haustov }
1306ac106addSNikolay Haustov 
getInlineImmVal32(unsigned Imm)13074bd72361SMatt Arsenault static int64_t getInlineImmVal32(unsigned Imm) {
13084bd72361SMatt Arsenault   switch (Imm) {
13094bd72361SMatt Arsenault   case 240:
13104bd72361SMatt Arsenault     return FloatToBits(0.5f);
13114bd72361SMatt Arsenault   case 241:
13124bd72361SMatt Arsenault     return FloatToBits(-0.5f);
13134bd72361SMatt Arsenault   case 242:
13144bd72361SMatt Arsenault     return FloatToBits(1.0f);
13154bd72361SMatt Arsenault   case 243:
13164bd72361SMatt Arsenault     return FloatToBits(-1.0f);
13174bd72361SMatt Arsenault   case 244:
13184bd72361SMatt Arsenault     return FloatToBits(2.0f);
13194bd72361SMatt Arsenault   case 245:
13204bd72361SMatt Arsenault     return FloatToBits(-2.0f);
13214bd72361SMatt Arsenault   case 246:
13224bd72361SMatt Arsenault     return FloatToBits(4.0f);
13234bd72361SMatt Arsenault   case 247:
13244bd72361SMatt Arsenault     return FloatToBits(-4.0f);
13254bd72361SMatt Arsenault   case 248: // 1 / (2 * PI)
13264bd72361SMatt Arsenault     return 0x3e22f983;
13274bd72361SMatt Arsenault   default:
13284bd72361SMatt Arsenault     llvm_unreachable("invalid fp inline imm");
13294bd72361SMatt Arsenault   }
13304bd72361SMatt Arsenault }
13314bd72361SMatt Arsenault 
getInlineImmVal64(unsigned Imm)13324bd72361SMatt Arsenault static int64_t getInlineImmVal64(unsigned Imm) {
13334bd72361SMatt Arsenault   switch (Imm) {
13344bd72361SMatt Arsenault   case 240:
13354bd72361SMatt Arsenault     return DoubleToBits(0.5);
13364bd72361SMatt Arsenault   case 241:
13374bd72361SMatt Arsenault     return DoubleToBits(-0.5);
13384bd72361SMatt Arsenault   case 242:
13394bd72361SMatt Arsenault     return DoubleToBits(1.0);
13404bd72361SMatt Arsenault   case 243:
13414bd72361SMatt Arsenault     return DoubleToBits(-1.0);
13424bd72361SMatt Arsenault   case 244:
13434bd72361SMatt Arsenault     return DoubleToBits(2.0);
13444bd72361SMatt Arsenault   case 245:
13454bd72361SMatt Arsenault     return DoubleToBits(-2.0);
13464bd72361SMatt Arsenault   case 246:
13474bd72361SMatt Arsenault     return DoubleToBits(4.0);
13484bd72361SMatt Arsenault   case 247:
13494bd72361SMatt Arsenault     return DoubleToBits(-4.0);
13504bd72361SMatt Arsenault   case 248: // 1 / (2 * PI)
13514bd72361SMatt Arsenault     return 0x3fc45f306dc9c882;
13524bd72361SMatt Arsenault   default:
13534bd72361SMatt Arsenault     llvm_unreachable("invalid fp inline imm");
13544bd72361SMatt Arsenault   }
13554bd72361SMatt Arsenault }
13564bd72361SMatt Arsenault 
getInlineImmVal16(unsigned Imm)13574bd72361SMatt Arsenault static int64_t getInlineImmVal16(unsigned Imm) {
13584bd72361SMatt Arsenault   switch (Imm) {
13594bd72361SMatt Arsenault   case 240:
13604bd72361SMatt Arsenault     return 0x3800;
13614bd72361SMatt Arsenault   case 241:
13624bd72361SMatt Arsenault     return 0xB800;
13634bd72361SMatt Arsenault   case 242:
13644bd72361SMatt Arsenault     return 0x3C00;
13654bd72361SMatt Arsenault   case 243:
13664bd72361SMatt Arsenault     return 0xBC00;
13674bd72361SMatt Arsenault   case 244:
13684bd72361SMatt Arsenault     return 0x4000;
13694bd72361SMatt Arsenault   case 245:
13704bd72361SMatt Arsenault     return 0xC000;
13714bd72361SMatt Arsenault   case 246:
13724bd72361SMatt Arsenault     return 0x4400;
13734bd72361SMatt Arsenault   case 247:
13744bd72361SMatt Arsenault     return 0xC400;
13754bd72361SMatt Arsenault   case 248: // 1 / (2 * PI)
13764bd72361SMatt Arsenault     return 0x3118;
13774bd72361SMatt Arsenault   default:
13784bd72361SMatt Arsenault     llvm_unreachable("invalid fp inline imm");
13794bd72361SMatt Arsenault   }
13804bd72361SMatt Arsenault }
13814bd72361SMatt Arsenault 
decodeFPImmed(OpWidthTy Width,unsigned Imm)13824bd72361SMatt Arsenault MCOperand AMDGPUDisassembler::decodeFPImmed(OpWidthTy Width, unsigned Imm) {
1383212a251cSArtem Tamazov   assert(Imm >= AMDGPU::EncValues::INLINE_FLOATING_C_MIN
1384212a251cSArtem Tamazov       && Imm <= AMDGPU::EncValues::INLINE_FLOATING_C_MAX);
13854bd72361SMatt Arsenault 
1386e1818af8STom Stellard   // ToDo: case 248: 1/(2*PI) - is allowed only on VI
13874bd72361SMatt Arsenault   switch (Width) {
13884bd72361SMatt Arsenault   case OPW32:
13899e77d0c6SStanislav Mekhanoshin   case OPW128: // splat constants
13909e77d0c6SStanislav Mekhanoshin   case OPW512:
13919e77d0c6SStanislav Mekhanoshin   case OPW1024:
1392a8d9d507SStanislav Mekhanoshin   case OPWV232:
13934bd72361SMatt Arsenault     return MCOperand::createImm(getInlineImmVal32(Imm));
13944bd72361SMatt Arsenault   case OPW64:
1395a8d9d507SStanislav Mekhanoshin   case OPW256:
13964bd72361SMatt Arsenault     return MCOperand::createImm(getInlineImmVal64(Imm));
13974bd72361SMatt Arsenault   case OPW16:
13989be7b0d4SMatt Arsenault   case OPWV216:
13994bd72361SMatt Arsenault     return MCOperand::createImm(getInlineImmVal16(Imm));
14004bd72361SMatt Arsenault   default:
14014bd72361SMatt Arsenault     llvm_unreachable("implement me");
1402e1818af8STom Stellard   }
1403e1818af8STom Stellard }
1404e1818af8STom Stellard 
getVgprClassId(const OpWidthTy Width) const1405212a251cSArtem Tamazov unsigned AMDGPUDisassembler::getVgprClassId(const OpWidthTy Width) const {
1406e1818af8STom Stellard   using namespace AMDGPU;
1407c8fbf6ffSEugene Zelenko 
1408212a251cSArtem Tamazov   assert(OPW_FIRST_ <= Width && Width < OPW_LAST_);
1409212a251cSArtem Tamazov   switch (Width) {
1410212a251cSArtem Tamazov   default: // fall
14114bd72361SMatt Arsenault   case OPW32:
14124bd72361SMatt Arsenault   case OPW16:
14139be7b0d4SMatt Arsenault   case OPWV216:
14144bd72361SMatt Arsenault     return VGPR_32RegClassID;
1415a8d9d507SStanislav Mekhanoshin   case OPW64:
1416a8d9d507SStanislav Mekhanoshin   case OPWV232: return VReg_64RegClassID;
1417a8d9d507SStanislav Mekhanoshin   case OPW96: return VReg_96RegClassID;
1418212a251cSArtem Tamazov   case OPW128: return VReg_128RegClassID;
1419a8d9d507SStanislav Mekhanoshin   case OPW160: return VReg_160RegClassID;
1420a8d9d507SStanislav Mekhanoshin   case OPW256: return VReg_256RegClassID;
1421a8d9d507SStanislav Mekhanoshin   case OPW512: return VReg_512RegClassID;
1422a8d9d507SStanislav Mekhanoshin   case OPW1024: return VReg_1024RegClassID;
1423212a251cSArtem Tamazov   }
1424212a251cSArtem Tamazov }
1425212a251cSArtem Tamazov 
getAgprClassId(const OpWidthTy Width) const14269e77d0c6SStanislav Mekhanoshin unsigned AMDGPUDisassembler::getAgprClassId(const OpWidthTy Width) const {
14279e77d0c6SStanislav Mekhanoshin   using namespace AMDGPU;
14289e77d0c6SStanislav Mekhanoshin 
14299e77d0c6SStanislav Mekhanoshin   assert(OPW_FIRST_ <= Width && Width < OPW_LAST_);
14309e77d0c6SStanislav Mekhanoshin   switch (Width) {
14319e77d0c6SStanislav Mekhanoshin   default: // fall
14329e77d0c6SStanislav Mekhanoshin   case OPW32:
14339e77d0c6SStanislav Mekhanoshin   case OPW16:
14349e77d0c6SStanislav Mekhanoshin   case OPWV216:
14359e77d0c6SStanislav Mekhanoshin     return AGPR_32RegClassID;
1436a8d9d507SStanislav Mekhanoshin   case OPW64:
1437a8d9d507SStanislav Mekhanoshin   case OPWV232: return AReg_64RegClassID;
1438a8d9d507SStanislav Mekhanoshin   case OPW96: return AReg_96RegClassID;
14399e77d0c6SStanislav Mekhanoshin   case OPW128: return AReg_128RegClassID;
1440a8d9d507SStanislav Mekhanoshin   case OPW160: return AReg_160RegClassID;
1441d625b4b0SJay Foad   case OPW256: return AReg_256RegClassID;
14429e77d0c6SStanislav Mekhanoshin   case OPW512: return AReg_512RegClassID;
14439e77d0c6SStanislav Mekhanoshin   case OPW1024: return AReg_1024RegClassID;
14449e77d0c6SStanislav Mekhanoshin   }
14459e77d0c6SStanislav Mekhanoshin }
14469e77d0c6SStanislav Mekhanoshin 
14479e77d0c6SStanislav Mekhanoshin 
getSgprClassId(const OpWidthTy Width) const1448212a251cSArtem Tamazov unsigned AMDGPUDisassembler::getSgprClassId(const OpWidthTy Width) const {
1449212a251cSArtem Tamazov   using namespace AMDGPU;
1450c8fbf6ffSEugene Zelenko 
1451212a251cSArtem Tamazov   assert(OPW_FIRST_ <= Width && Width < OPW_LAST_);
1452212a251cSArtem Tamazov   switch (Width) {
1453212a251cSArtem Tamazov   default: // fall
14544bd72361SMatt Arsenault   case OPW32:
14554bd72361SMatt Arsenault   case OPW16:
14569be7b0d4SMatt Arsenault   case OPWV216:
14574bd72361SMatt Arsenault     return SGPR_32RegClassID;
1458a8d9d507SStanislav Mekhanoshin   case OPW64:
1459a8d9d507SStanislav Mekhanoshin   case OPWV232: return SGPR_64RegClassID;
1460a8d9d507SStanislav Mekhanoshin   case OPW96: return SGPR_96RegClassID;
1461212a251cSArtem Tamazov   case OPW128: return SGPR_128RegClassID;
1462a8d9d507SStanislav Mekhanoshin   case OPW160: return SGPR_160RegClassID;
146327134953SDmitry Preobrazhensky   case OPW256: return SGPR_256RegClassID;
146427134953SDmitry Preobrazhensky   case OPW512: return SGPR_512RegClassID;
1465212a251cSArtem Tamazov   }
1466212a251cSArtem Tamazov }
1467212a251cSArtem Tamazov 
getTtmpClassId(const OpWidthTy Width) const1468212a251cSArtem Tamazov unsigned AMDGPUDisassembler::getTtmpClassId(const OpWidthTy Width) const {
1469212a251cSArtem Tamazov   using namespace AMDGPU;
1470c8fbf6ffSEugene Zelenko 
1471212a251cSArtem Tamazov   assert(OPW_FIRST_ <= Width && Width < OPW_LAST_);
1472212a251cSArtem Tamazov   switch (Width) {
1473212a251cSArtem Tamazov   default: // fall
14744bd72361SMatt Arsenault   case OPW32:
14754bd72361SMatt Arsenault   case OPW16:
14769be7b0d4SMatt Arsenault   case OPWV216:
14774bd72361SMatt Arsenault     return TTMP_32RegClassID;
1478a8d9d507SStanislav Mekhanoshin   case OPW64:
1479a8d9d507SStanislav Mekhanoshin   case OPWV232: return TTMP_64RegClassID;
1480212a251cSArtem Tamazov   case OPW128: return TTMP_128RegClassID;
148127134953SDmitry Preobrazhensky   case OPW256: return TTMP_256RegClassID;
148227134953SDmitry Preobrazhensky   case OPW512: return TTMP_512RegClassID;
1483212a251cSArtem Tamazov   }
1484212a251cSArtem Tamazov }
1485212a251cSArtem Tamazov 
getTTmpIdx(unsigned Val) const1486ac2b0264SDmitry Preobrazhensky int AMDGPUDisassembler::getTTmpIdx(unsigned Val) const {
1487ac2b0264SDmitry Preobrazhensky   using namespace AMDGPU::EncValues;
1488ac2b0264SDmitry Preobrazhensky 
148918cb7441SJay Foad   unsigned TTmpMin = isGFX9Plus() ? TTMP_GFX9PLUS_MIN : TTMP_VI_MIN;
149018cb7441SJay Foad   unsigned TTmpMax = isGFX9Plus() ? TTMP_GFX9PLUS_MAX : TTMP_VI_MAX;
1491ac2b0264SDmitry Preobrazhensky 
1492ac2b0264SDmitry Preobrazhensky   return (TTmpMin <= Val && Val <= TTmpMax)? Val - TTmpMin : -1;
1493ac2b0264SDmitry Preobrazhensky }
1494ac2b0264SDmitry Preobrazhensky 
decodeSrcOp(const OpWidthTy Width,unsigned Val,bool MandatoryLiteral) const1495b4b7e605SJoe Nash MCOperand AMDGPUDisassembler::decodeSrcOp(const OpWidthTy Width, unsigned Val,
1496b4b7e605SJoe Nash                                           bool MandatoryLiteral) const {
1497212a251cSArtem Tamazov   using namespace AMDGPU::EncValues;
1498c8fbf6ffSEugene Zelenko 
14999e77d0c6SStanislav Mekhanoshin   assert(Val < 1024); // enum10
15009e77d0c6SStanislav Mekhanoshin 
15019e77d0c6SStanislav Mekhanoshin   bool IsAGPR = Val & 512;
15029e77d0c6SStanislav Mekhanoshin   Val &= 511;
1503ac106addSNikolay Haustov 
1504212a251cSArtem Tamazov   if (VGPR_MIN <= Val && Val <= VGPR_MAX) {
15059e77d0c6SStanislav Mekhanoshin     return createRegOperand(IsAGPR ? getAgprClassId(Width)
15069e77d0c6SStanislav Mekhanoshin                                    : getVgprClassId(Width), Val - VGPR_MIN);
1507212a251cSArtem Tamazov   }
1508b49c3361SArtem Tamazov   if (Val <= SGPR_MAX) {
150949231c1fSKazu Hirata     // "SGPR_MIN <= Val" is always true and causes compilation warning.
151049231c1fSKazu Hirata     static_assert(SGPR_MIN == 0, "");
1511212a251cSArtem Tamazov     return createSRegOperand(getSgprClassId(Width), Val - SGPR_MIN);
1512212a251cSArtem Tamazov   }
1513ac2b0264SDmitry Preobrazhensky 
1514ac2b0264SDmitry Preobrazhensky   int TTmpIdx = getTTmpIdx(Val);
1515ac2b0264SDmitry Preobrazhensky   if (TTmpIdx >= 0) {
1516ac2b0264SDmitry Preobrazhensky     return createSRegOperand(getTtmpClassId(Width), TTmpIdx);
1517212a251cSArtem Tamazov   }
1518ac106addSNikolay Haustov 
1519212a251cSArtem Tamazov   if (INLINE_INTEGER_C_MIN <= Val && Val <= INLINE_INTEGER_C_MAX)
1520ac106addSNikolay Haustov     return decodeIntImmed(Val);
1521ac106addSNikolay Haustov 
1522212a251cSArtem Tamazov   if (INLINE_FLOATING_C_MIN <= Val && Val <= INLINE_FLOATING_C_MAX)
15234bd72361SMatt Arsenault     return decodeFPImmed(Width, Val);
1524ac106addSNikolay Haustov 
1525b4b7e605SJoe Nash   if (Val == LITERAL_CONST) {
1526b4b7e605SJoe Nash     if (MandatoryLiteral)
1527b4b7e605SJoe Nash       // Keep a sentinel value for deferred setting
1528b4b7e605SJoe Nash       return MCOperand::createImm(LITERAL_CONST);
1529b4b7e605SJoe Nash     else
1530ac106addSNikolay Haustov       return decodeLiteralConstant();
1531b4b7e605SJoe Nash   }
1532ac106addSNikolay Haustov 
15334bd72361SMatt Arsenault   switch (Width) {
15344bd72361SMatt Arsenault   case OPW32:
15354bd72361SMatt Arsenault   case OPW16:
15369be7b0d4SMatt Arsenault   case OPWV216:
15374bd72361SMatt Arsenault     return decodeSpecialReg32(Val);
15384bd72361SMatt Arsenault   case OPW64:
1539a8d9d507SStanislav Mekhanoshin   case OPWV232:
15404bd72361SMatt Arsenault     return decodeSpecialReg64(Val);
15414bd72361SMatt Arsenault   default:
15424bd72361SMatt Arsenault     llvm_unreachable("unexpected immediate type");
15434bd72361SMatt Arsenault   }
1544ac106addSNikolay Haustov }
1545ac106addSNikolay Haustov 
decodeDstOp(const OpWidthTy Width,unsigned Val) const154627134953SDmitry Preobrazhensky MCOperand AMDGPUDisassembler::decodeDstOp(const OpWidthTy Width, unsigned Val) const {
154727134953SDmitry Preobrazhensky   using namespace AMDGPU::EncValues;
154827134953SDmitry Preobrazhensky 
154927134953SDmitry Preobrazhensky   assert(Val < 128);
155027134953SDmitry Preobrazhensky   assert(Width == OPW256 || Width == OPW512);
155127134953SDmitry Preobrazhensky 
155227134953SDmitry Preobrazhensky   if (Val <= SGPR_MAX) {
155349231c1fSKazu Hirata     // "SGPR_MIN <= Val" is always true and causes compilation warning.
155449231c1fSKazu Hirata     static_assert(SGPR_MIN == 0, "");
155527134953SDmitry Preobrazhensky     return createSRegOperand(getSgprClassId(Width), Val - SGPR_MIN);
155627134953SDmitry Preobrazhensky   }
155727134953SDmitry Preobrazhensky 
155827134953SDmitry Preobrazhensky   int TTmpIdx = getTTmpIdx(Val);
155927134953SDmitry Preobrazhensky   if (TTmpIdx >= 0) {
156027134953SDmitry Preobrazhensky     return createSRegOperand(getTtmpClassId(Width), TTmpIdx);
156127134953SDmitry Preobrazhensky   }
156227134953SDmitry Preobrazhensky 
156327134953SDmitry Preobrazhensky   llvm_unreachable("unknown dst register");
156427134953SDmitry Preobrazhensky }
156527134953SDmitry Preobrazhensky 
156607b7fadaSJoe Nash // Bit 0 of DstY isn't stored in the instruction, because it's always the
156707b7fadaSJoe Nash // opposite of bit 0 of DstX.
decodeVOPDDstYOp(MCInst & Inst,unsigned Val) const156807b7fadaSJoe Nash MCOperand AMDGPUDisassembler::decodeVOPDDstYOp(MCInst &Inst,
156907b7fadaSJoe Nash                                                unsigned Val) const {
157007b7fadaSJoe Nash   int VDstXInd =
157107b7fadaSJoe Nash       AMDGPU::getNamedOperandIdx(Inst.getOpcode(), AMDGPU::OpName::vdstX);
157207b7fadaSJoe Nash   assert(VDstXInd != -1);
157307b7fadaSJoe Nash   assert(Inst.getOperand(VDstXInd).isReg());
157407b7fadaSJoe Nash   unsigned XDstReg = MRI.getEncodingValue(Inst.getOperand(VDstXInd).getReg());
157507b7fadaSJoe Nash   Val |= ~XDstReg & 1;
157607b7fadaSJoe Nash   auto Width = llvm::AMDGPUDisassembler::OPW32;
157707b7fadaSJoe Nash   return createRegOperand(getVgprClassId(Width), Val);
157807b7fadaSJoe Nash }
157907b7fadaSJoe Nash 
decodeSpecialReg32(unsigned Val) const1580ac106addSNikolay Haustov MCOperand AMDGPUDisassembler::decodeSpecialReg32(unsigned Val) const {
1581ac106addSNikolay Haustov   using namespace AMDGPU;
1582c8fbf6ffSEugene Zelenko 
1583e1818af8STom Stellard   switch (Val) {
1584ac2b0264SDmitry Preobrazhensky   case 102: return createRegOperand(FLAT_SCR_LO);
1585ac2b0264SDmitry Preobrazhensky   case 103: return createRegOperand(FLAT_SCR_HI);
15863afbd825SDmitry Preobrazhensky   case 104: return createRegOperand(XNACK_MASK_LO);
15873afbd825SDmitry Preobrazhensky   case 105: return createRegOperand(XNACK_MASK_HI);
1588ac106addSNikolay Haustov   case 106: return createRegOperand(VCC_LO);
1589ac106addSNikolay Haustov   case 107: return createRegOperand(VCC_HI);
1590137976faSDmitry Preobrazhensky   case 108: return createRegOperand(TBA_LO);
1591137976faSDmitry Preobrazhensky   case 109: return createRegOperand(TBA_HI);
1592137976faSDmitry Preobrazhensky   case 110: return createRegOperand(TMA_LO);
1593137976faSDmitry Preobrazhensky   case 111: return createRegOperand(TMA_HI);
1594c7025940SJoe Nash   case 124:
1595c7025940SJoe Nash     return isGFX11Plus() ? createRegOperand(SGPR_NULL) : createRegOperand(M0);
1596c7025940SJoe Nash   case 125:
1597c7025940SJoe Nash     return isGFX11Plus() ? createRegOperand(M0) : createRegOperand(SGPR_NULL);
1598ac106addSNikolay Haustov   case 126: return createRegOperand(EXEC_LO);
1599ac106addSNikolay Haustov   case 127: return createRegOperand(EXEC_HI);
1600a3b3b489SMatt Arsenault   case 235: return createRegOperand(SRC_SHARED_BASE);
1601a3b3b489SMatt Arsenault   case 236: return createRegOperand(SRC_SHARED_LIMIT);
1602a3b3b489SMatt Arsenault   case 237: return createRegOperand(SRC_PRIVATE_BASE);
1603a3b3b489SMatt Arsenault   case 238: return createRegOperand(SRC_PRIVATE_LIMIT);
1604137976faSDmitry Preobrazhensky   case 239: return createRegOperand(SRC_POPS_EXITING_WAVE_ID);
16059111f35fSDmitry Preobrazhensky   case 251: return createRegOperand(SRC_VCCZ);
16069111f35fSDmitry Preobrazhensky   case 252: return createRegOperand(SRC_EXECZ);
16079111f35fSDmitry Preobrazhensky   case 253: return createRegOperand(SRC_SCC);
1608942c273dSDmitry Preobrazhensky   case 254: return createRegOperand(LDS_DIRECT);
1609ac106addSNikolay Haustov   default: break;
1610e1818af8STom Stellard   }
1611ac106addSNikolay Haustov   return errOperand(Val, "unknown operand encoding " + Twine(Val));
1612e1818af8STom Stellard }
1613e1818af8STom Stellard 
decodeSpecialReg64(unsigned Val) const1614ac106addSNikolay Haustov MCOperand AMDGPUDisassembler::decodeSpecialReg64(unsigned Val) const {
1615161a158eSNikolay Haustov   using namespace AMDGPU;
1616c8fbf6ffSEugene Zelenko 
1617161a158eSNikolay Haustov   switch (Val) {
1618ac2b0264SDmitry Preobrazhensky   case 102: return createRegOperand(FLAT_SCR);
16193afbd825SDmitry Preobrazhensky   case 104: return createRegOperand(XNACK_MASK);
1620ac106addSNikolay Haustov   case 106: return createRegOperand(VCC);
1621137976faSDmitry Preobrazhensky   case 108: return createRegOperand(TBA);
1622137976faSDmitry Preobrazhensky   case 110: return createRegOperand(TMA);
1623c7025940SJoe Nash   case 124:
1624c7025940SJoe Nash     if (isGFX11Plus())
1625c7025940SJoe Nash       return createRegOperand(SGPR_NULL);
1626c7025940SJoe Nash     break;
1627c7025940SJoe Nash   case 125:
1628c7025940SJoe Nash     if (!isGFX11Plus())
1629c7025940SJoe Nash       return createRegOperand(SGPR_NULL);
1630c7025940SJoe Nash     break;
1631ac106addSNikolay Haustov   case 126: return createRegOperand(EXEC);
1632137976faSDmitry Preobrazhensky   case 235: return createRegOperand(SRC_SHARED_BASE);
1633137976faSDmitry Preobrazhensky   case 236: return createRegOperand(SRC_SHARED_LIMIT);
1634137976faSDmitry Preobrazhensky   case 237: return createRegOperand(SRC_PRIVATE_BASE);
1635137976faSDmitry Preobrazhensky   case 238: return createRegOperand(SRC_PRIVATE_LIMIT);
1636137976faSDmitry Preobrazhensky   case 239: return createRegOperand(SRC_POPS_EXITING_WAVE_ID);
16379111f35fSDmitry Preobrazhensky   case 251: return createRegOperand(SRC_VCCZ);
16389111f35fSDmitry Preobrazhensky   case 252: return createRegOperand(SRC_EXECZ);
16399111f35fSDmitry Preobrazhensky   case 253: return createRegOperand(SRC_SCC);
1640ac106addSNikolay Haustov   default: break;
1641161a158eSNikolay Haustov   }
1642ac106addSNikolay Haustov   return errOperand(Val, "unknown operand encoding " + Twine(Val));
1643161a158eSNikolay Haustov }
1644161a158eSNikolay Haustov 
decodeSDWASrc(const OpWidthTy Width,const unsigned Val) const1645549c89d2SSam Kolton MCOperand AMDGPUDisassembler::decodeSDWASrc(const OpWidthTy Width,
16466b65f7c3SDmitry Preobrazhensky                                             const unsigned Val) const {
1647363f47a2SSam Kolton   using namespace AMDGPU::SDWA;
16486b65f7c3SDmitry Preobrazhensky   using namespace AMDGPU::EncValues;
1649363f47a2SSam Kolton 
165033d806a5SStanislav Mekhanoshin   if (STI.getFeatureBits()[AMDGPU::FeatureGFX9] ||
165133d806a5SStanislav Mekhanoshin       STI.getFeatureBits()[AMDGPU::FeatureGFX10]) {
1652da644c02SStanislav Mekhanoshin     // XXX: cast to int is needed to avoid stupid warning:
1653a179d25bSSam Kolton     // compare with unsigned is always true
1654da644c02SStanislav Mekhanoshin     if (int(SDWA9EncValues::SRC_VGPR_MIN) <= int(Val) &&
1655363f47a2SSam Kolton         Val <= SDWA9EncValues::SRC_VGPR_MAX) {
1656363f47a2SSam Kolton       return createRegOperand(getVgprClassId(Width),
1657363f47a2SSam Kolton                               Val - SDWA9EncValues::SRC_VGPR_MIN);
1658363f47a2SSam Kolton     }
1659363f47a2SSam Kolton     if (SDWA9EncValues::SRC_SGPR_MIN <= Val &&
16604f87d30aSJay Foad         Val <= (isGFX10Plus() ? SDWA9EncValues::SRC_SGPR_MAX_GFX10
166133d806a5SStanislav Mekhanoshin                               : SDWA9EncValues::SRC_SGPR_MAX_SI)) {
1662363f47a2SSam Kolton       return createSRegOperand(getSgprClassId(Width),
1663363f47a2SSam Kolton                                Val - SDWA9EncValues::SRC_SGPR_MIN);
1664363f47a2SSam Kolton     }
1665ac2b0264SDmitry Preobrazhensky     if (SDWA9EncValues::SRC_TTMP_MIN <= Val &&
1666ac2b0264SDmitry Preobrazhensky         Val <= SDWA9EncValues::SRC_TTMP_MAX) {
1667ac2b0264SDmitry Preobrazhensky       return createSRegOperand(getTtmpClassId(Width),
1668ac2b0264SDmitry Preobrazhensky                                Val - SDWA9EncValues::SRC_TTMP_MIN);
1669ac2b0264SDmitry Preobrazhensky     }
1670363f47a2SSam Kolton 
16716b65f7c3SDmitry Preobrazhensky     const unsigned SVal = Val - SDWA9EncValues::SRC_SGPR_MIN;
16726b65f7c3SDmitry Preobrazhensky 
16736b65f7c3SDmitry Preobrazhensky     if (INLINE_INTEGER_C_MIN <= SVal && SVal <= INLINE_INTEGER_C_MAX)
16746b65f7c3SDmitry Preobrazhensky       return decodeIntImmed(SVal);
16756b65f7c3SDmitry Preobrazhensky 
16766b65f7c3SDmitry Preobrazhensky     if (INLINE_FLOATING_C_MIN <= SVal && SVal <= INLINE_FLOATING_C_MAX)
16776b65f7c3SDmitry Preobrazhensky       return decodeFPImmed(Width, SVal);
16786b65f7c3SDmitry Preobrazhensky 
16796b65f7c3SDmitry Preobrazhensky     return decodeSpecialReg32(SVal);
1680549c89d2SSam Kolton   } else if (STI.getFeatureBits()[AMDGPU::FeatureVolcanicIslands]) {
1681549c89d2SSam Kolton     return createRegOperand(getVgprClassId(Width), Val);
1682549c89d2SSam Kolton   }
1683549c89d2SSam Kolton   llvm_unreachable("unsupported target");
1684363f47a2SSam Kolton }
1685363f47a2SSam Kolton 
decodeSDWASrc16(unsigned Val) const1686549c89d2SSam Kolton MCOperand AMDGPUDisassembler::decodeSDWASrc16(unsigned Val) const {
1687549c89d2SSam Kolton   return decodeSDWASrc(OPW16, Val);
1688363f47a2SSam Kolton }
1689363f47a2SSam Kolton 
decodeSDWASrc32(unsigned Val) const1690549c89d2SSam Kolton MCOperand AMDGPUDisassembler::decodeSDWASrc32(unsigned Val) const {
1691549c89d2SSam Kolton   return decodeSDWASrc(OPW32, Val);
1692363f47a2SSam Kolton }
1693363f47a2SSam Kolton 
decodeSDWAVopcDst(unsigned Val) const1694549c89d2SSam Kolton MCOperand AMDGPUDisassembler::decodeSDWAVopcDst(unsigned Val) const {
1695363f47a2SSam Kolton   using namespace AMDGPU::SDWA;
1696363f47a2SSam Kolton 
169733d806a5SStanislav Mekhanoshin   assert((STI.getFeatureBits()[AMDGPU::FeatureGFX9] ||
169833d806a5SStanislav Mekhanoshin           STI.getFeatureBits()[AMDGPU::FeatureGFX10]) &&
169933d806a5SStanislav Mekhanoshin          "SDWAVopcDst should be present only on GFX9+");
170033d806a5SStanislav Mekhanoshin 
1701ab4f2ea7SStanislav Mekhanoshin   bool IsWave64 = STI.getFeatureBits()[AMDGPU::FeatureWavefrontSize64];
1702ab4f2ea7SStanislav Mekhanoshin 
1703363f47a2SSam Kolton   if (Val & SDWA9EncValues::VOPC_DST_VCC_MASK) {
1704363f47a2SSam Kolton     Val &= SDWA9EncValues::VOPC_DST_SGPR_MASK;
1705ac2b0264SDmitry Preobrazhensky 
1706ac2b0264SDmitry Preobrazhensky     int TTmpIdx = getTTmpIdx(Val);
1707ac2b0264SDmitry Preobrazhensky     if (TTmpIdx >= 0) {
1708434d5925SDmitry Preobrazhensky       auto TTmpClsId = getTtmpClassId(IsWave64 ? OPW64 : OPW32);
1709434d5925SDmitry Preobrazhensky       return createSRegOperand(TTmpClsId, TTmpIdx);
171033d806a5SStanislav Mekhanoshin     } else if (Val > SGPR_MAX) {
1711ab4f2ea7SStanislav Mekhanoshin       return IsWave64 ? decodeSpecialReg64(Val)
1712ab4f2ea7SStanislav Mekhanoshin                       : decodeSpecialReg32(Val);
1713363f47a2SSam Kolton     } else {
1714ab4f2ea7SStanislav Mekhanoshin       return createSRegOperand(getSgprClassId(IsWave64 ? OPW64 : OPW32), Val);
1715363f47a2SSam Kolton     }
1716363f47a2SSam Kolton   } else {
1717ab4f2ea7SStanislav Mekhanoshin     return createRegOperand(IsWave64 ? AMDGPU::VCC : AMDGPU::VCC_LO);
1718363f47a2SSam Kolton   }
1719363f47a2SSam Kolton }
1720363f47a2SSam Kolton 
decodeBoolReg(unsigned Val) const1721ab4f2ea7SStanislav Mekhanoshin MCOperand AMDGPUDisassembler::decodeBoolReg(unsigned Val) const {
1722ab4f2ea7SStanislav Mekhanoshin   return STI.getFeatureBits()[AMDGPU::FeatureWavefrontSize64] ?
1723ab4f2ea7SStanislav Mekhanoshin     decodeOperand_SReg_64(Val) : decodeOperand_SReg_32(Val);
1724ab4f2ea7SStanislav Mekhanoshin }
1725ab4f2ea7SStanislav Mekhanoshin 
isVI() const1726ac2b0264SDmitry Preobrazhensky bool AMDGPUDisassembler::isVI() const {
1727ac2b0264SDmitry Preobrazhensky   return STI.getFeatureBits()[AMDGPU::FeatureVolcanicIslands];
1728ac2b0264SDmitry Preobrazhensky }
1729ac2b0264SDmitry Preobrazhensky 
isGFX9() const17304f87d30aSJay Foad bool AMDGPUDisassembler::isGFX9() const { return AMDGPU::isGFX9(STI); }
1731ac2b0264SDmitry Preobrazhensky 
isGFX90A() const1732a8d9d507SStanislav Mekhanoshin bool AMDGPUDisassembler::isGFX90A() const {
1733a8d9d507SStanislav Mekhanoshin   return STI.getFeatureBits()[AMDGPU::FeatureGFX90AInsts];
1734a8d9d507SStanislav Mekhanoshin }
1735a8d9d507SStanislav Mekhanoshin 
isGFX9Plus() const17364f87d30aSJay Foad bool AMDGPUDisassembler::isGFX9Plus() const { return AMDGPU::isGFX9Plus(STI); }
17374f87d30aSJay Foad 
isGFX10() const17384f87d30aSJay Foad bool AMDGPUDisassembler::isGFX10() const { return AMDGPU::isGFX10(STI); }
17394f87d30aSJay Foad 
isGFX10Plus() const17404f87d30aSJay Foad bool AMDGPUDisassembler::isGFX10Plus() const {
17414f87d30aSJay Foad   return AMDGPU::isGFX10Plus(STI);
174233d806a5SStanislav Mekhanoshin }
174333d806a5SStanislav Mekhanoshin 
isGFX11() const1744c7025940SJoe Nash bool AMDGPUDisassembler::isGFX11() const {
1745c7025940SJoe Nash   return STI.getFeatureBits()[AMDGPU::FeatureGFX11];
1746c7025940SJoe Nash }
1747c7025940SJoe Nash 
isGFX11Plus() const1748c7025940SJoe Nash bool AMDGPUDisassembler::isGFX11Plus() const {
1749c7025940SJoe Nash   return AMDGPU::isGFX11Plus(STI);
1750c7025940SJoe Nash }
1751c7025940SJoe Nash 
1752c7025940SJoe Nash 
hasArchitectedFlatScratch() const17536fb02596SStanislav Mekhanoshin bool AMDGPUDisassembler::hasArchitectedFlatScratch() const {
17546fb02596SStanislav Mekhanoshin   return STI.getFeatureBits()[AMDGPU::FeatureArchitectedFlatScratch];
17556fb02596SStanislav Mekhanoshin }
17566fb02596SStanislav Mekhanoshin 
17573381d7a2SSam Kolton //===----------------------------------------------------------------------===//
1758528057c1SRonak Chauhan // AMDGPU specific symbol handling
1759528057c1SRonak Chauhan //===----------------------------------------------------------------------===//
1760528057c1SRonak Chauhan #define PRINT_DIRECTIVE(DIRECTIVE, MASK)                                       \
1761528057c1SRonak Chauhan   do {                                                                         \
1762528057c1SRonak Chauhan     KdStream << Indent << DIRECTIVE " "                                        \
1763528057c1SRonak Chauhan              << ((FourByteBuffer & MASK) >> (MASK##_SHIFT)) << '\n';           \
1764528057c1SRonak Chauhan   } while (0)
1765528057c1SRonak Chauhan 
1766528057c1SRonak Chauhan // NOLINTNEXTLINE(readability-identifier-naming)
decodeCOMPUTE_PGM_RSRC1(uint32_t FourByteBuffer,raw_string_ostream & KdStream) const1767528057c1SRonak Chauhan MCDisassembler::DecodeStatus AMDGPUDisassembler::decodeCOMPUTE_PGM_RSRC1(
1768528057c1SRonak Chauhan     uint32_t FourByteBuffer, raw_string_ostream &KdStream) const {
1769528057c1SRonak Chauhan   using namespace amdhsa;
1770528057c1SRonak Chauhan   StringRef Indent = "\t";
1771528057c1SRonak Chauhan 
1772528057c1SRonak Chauhan   // We cannot accurately backward compute #VGPRs used from
1773528057c1SRonak Chauhan   // GRANULATED_WORKITEM_VGPR_COUNT. But we are concerned with getting the same
1774528057c1SRonak Chauhan   // value of GRANULATED_WORKITEM_VGPR_COUNT in the reassembled binary. So we
1775528057c1SRonak Chauhan   // simply calculate the inverse of what the assembler does.
1776528057c1SRonak Chauhan 
1777528057c1SRonak Chauhan   uint32_t GranulatedWorkitemVGPRCount =
1778528057c1SRonak Chauhan       (FourByteBuffer & COMPUTE_PGM_RSRC1_GRANULATED_WORKITEM_VGPR_COUNT) >>
1779528057c1SRonak Chauhan       COMPUTE_PGM_RSRC1_GRANULATED_WORKITEM_VGPR_COUNT_SHIFT;
1780528057c1SRonak Chauhan 
1781528057c1SRonak Chauhan   uint32_t NextFreeVGPR = (GranulatedWorkitemVGPRCount + 1) *
1782528057c1SRonak Chauhan                           AMDGPU::IsaInfo::getVGPREncodingGranule(&STI);
1783528057c1SRonak Chauhan 
1784528057c1SRonak Chauhan   KdStream << Indent << ".amdhsa_next_free_vgpr " << NextFreeVGPR << '\n';
1785528057c1SRonak Chauhan 
1786528057c1SRonak Chauhan   // We cannot backward compute values used to calculate
1787528057c1SRonak Chauhan   // GRANULATED_WAVEFRONT_SGPR_COUNT. Hence the original values for following
1788528057c1SRonak Chauhan   // directives can't be computed:
1789528057c1SRonak Chauhan   // .amdhsa_reserve_vcc
1790528057c1SRonak Chauhan   // .amdhsa_reserve_flat_scratch
1791528057c1SRonak Chauhan   // .amdhsa_reserve_xnack_mask
1792528057c1SRonak Chauhan   // They take their respective default values if not specified in the assembly.
1793528057c1SRonak Chauhan   //
1794528057c1SRonak Chauhan   // GRANULATED_WAVEFRONT_SGPR_COUNT
1795528057c1SRonak Chauhan   //    = f(NEXT_FREE_SGPR + VCC + FLAT_SCRATCH + XNACK_MASK)
1796528057c1SRonak Chauhan   //
1797528057c1SRonak Chauhan   // We compute the inverse as though all directives apart from NEXT_FREE_SGPR
1798528057c1SRonak Chauhan   // are set to 0. So while disassembling we consider that:
1799528057c1SRonak Chauhan   //
1800528057c1SRonak Chauhan   // GRANULATED_WAVEFRONT_SGPR_COUNT
1801528057c1SRonak Chauhan   //    = f(NEXT_FREE_SGPR + 0 + 0 + 0)
1802528057c1SRonak Chauhan   //
1803528057c1SRonak Chauhan   // The disassembler cannot recover the original values of those 3 directives.
1804528057c1SRonak Chauhan 
1805528057c1SRonak Chauhan   uint32_t GranulatedWavefrontSGPRCount =
1806528057c1SRonak Chauhan       (FourByteBuffer & COMPUTE_PGM_RSRC1_GRANULATED_WAVEFRONT_SGPR_COUNT) >>
1807528057c1SRonak Chauhan       COMPUTE_PGM_RSRC1_GRANULATED_WAVEFRONT_SGPR_COUNT_SHIFT;
1808528057c1SRonak Chauhan 
18094f87d30aSJay Foad   if (isGFX10Plus() && GranulatedWavefrontSGPRCount)
1810528057c1SRonak Chauhan     return MCDisassembler::Fail;
1811528057c1SRonak Chauhan 
1812528057c1SRonak Chauhan   uint32_t NextFreeSGPR = (GranulatedWavefrontSGPRCount + 1) *
1813528057c1SRonak Chauhan                           AMDGPU::IsaInfo::getSGPREncodingGranule(&STI);
1814528057c1SRonak Chauhan 
1815528057c1SRonak Chauhan   KdStream << Indent << ".amdhsa_reserve_vcc " << 0 << '\n';
18166fb02596SStanislav Mekhanoshin   if (!hasArchitectedFlatScratch())
1817528057c1SRonak Chauhan     KdStream << Indent << ".amdhsa_reserve_flat_scratch " << 0 << '\n';
1818528057c1SRonak Chauhan   KdStream << Indent << ".amdhsa_reserve_xnack_mask " << 0 << '\n';
1819528057c1SRonak Chauhan   KdStream << Indent << ".amdhsa_next_free_sgpr " << NextFreeSGPR << "\n";
1820528057c1SRonak Chauhan 
1821528057c1SRonak Chauhan   if (FourByteBuffer & COMPUTE_PGM_RSRC1_PRIORITY)
1822528057c1SRonak Chauhan     return MCDisassembler::Fail;
1823528057c1SRonak Chauhan 
1824528057c1SRonak Chauhan   PRINT_DIRECTIVE(".amdhsa_float_round_mode_32",
1825528057c1SRonak Chauhan                   COMPUTE_PGM_RSRC1_FLOAT_ROUND_MODE_32);
1826528057c1SRonak Chauhan   PRINT_DIRECTIVE(".amdhsa_float_round_mode_16_64",
1827528057c1SRonak Chauhan                   COMPUTE_PGM_RSRC1_FLOAT_ROUND_MODE_16_64);
1828528057c1SRonak Chauhan   PRINT_DIRECTIVE(".amdhsa_float_denorm_mode_32",
1829528057c1SRonak Chauhan                   COMPUTE_PGM_RSRC1_FLOAT_DENORM_MODE_32);
1830528057c1SRonak Chauhan   PRINT_DIRECTIVE(".amdhsa_float_denorm_mode_16_64",
1831528057c1SRonak Chauhan                   COMPUTE_PGM_RSRC1_FLOAT_DENORM_MODE_16_64);
1832528057c1SRonak Chauhan 
1833528057c1SRonak Chauhan   if (FourByteBuffer & COMPUTE_PGM_RSRC1_PRIV)
1834528057c1SRonak Chauhan     return MCDisassembler::Fail;
1835528057c1SRonak Chauhan 
1836528057c1SRonak Chauhan   PRINT_DIRECTIVE(".amdhsa_dx10_clamp", COMPUTE_PGM_RSRC1_ENABLE_DX10_CLAMP);
1837528057c1SRonak Chauhan 
1838528057c1SRonak Chauhan   if (FourByteBuffer & COMPUTE_PGM_RSRC1_DEBUG_MODE)
1839528057c1SRonak Chauhan     return MCDisassembler::Fail;
1840528057c1SRonak Chauhan 
1841528057c1SRonak Chauhan   PRINT_DIRECTIVE(".amdhsa_ieee_mode", COMPUTE_PGM_RSRC1_ENABLE_IEEE_MODE);
1842528057c1SRonak Chauhan 
1843528057c1SRonak Chauhan   if (FourByteBuffer & COMPUTE_PGM_RSRC1_BULKY)
1844528057c1SRonak Chauhan     return MCDisassembler::Fail;
1845528057c1SRonak Chauhan 
1846528057c1SRonak Chauhan   if (FourByteBuffer & COMPUTE_PGM_RSRC1_CDBG_USER)
1847528057c1SRonak Chauhan     return MCDisassembler::Fail;
1848528057c1SRonak Chauhan 
1849528057c1SRonak Chauhan   PRINT_DIRECTIVE(".amdhsa_fp16_overflow", COMPUTE_PGM_RSRC1_FP16_OVFL);
1850528057c1SRonak Chauhan 
1851528057c1SRonak Chauhan   if (FourByteBuffer & COMPUTE_PGM_RSRC1_RESERVED0)
1852528057c1SRonak Chauhan     return MCDisassembler::Fail;
1853528057c1SRonak Chauhan 
18544f87d30aSJay Foad   if (isGFX10Plus()) {
1855528057c1SRonak Chauhan     PRINT_DIRECTIVE(".amdhsa_workgroup_processor_mode",
1856528057c1SRonak Chauhan                     COMPUTE_PGM_RSRC1_WGP_MODE);
1857528057c1SRonak Chauhan     PRINT_DIRECTIVE(".amdhsa_memory_ordered", COMPUTE_PGM_RSRC1_MEM_ORDERED);
1858528057c1SRonak Chauhan     PRINT_DIRECTIVE(".amdhsa_forward_progress", COMPUTE_PGM_RSRC1_FWD_PROGRESS);
1859528057c1SRonak Chauhan   }
1860528057c1SRonak Chauhan   return MCDisassembler::Success;
1861528057c1SRonak Chauhan }
1862528057c1SRonak Chauhan 
1863528057c1SRonak Chauhan // NOLINTNEXTLINE(readability-identifier-naming)
decodeCOMPUTE_PGM_RSRC2(uint32_t FourByteBuffer,raw_string_ostream & KdStream) const1864528057c1SRonak Chauhan MCDisassembler::DecodeStatus AMDGPUDisassembler::decodeCOMPUTE_PGM_RSRC2(
1865528057c1SRonak Chauhan     uint32_t FourByteBuffer, raw_string_ostream &KdStream) const {
1866528057c1SRonak Chauhan   using namespace amdhsa;
1867528057c1SRonak Chauhan   StringRef Indent = "\t";
18686fb02596SStanislav Mekhanoshin   if (hasArchitectedFlatScratch())
18696fb02596SStanislav Mekhanoshin     PRINT_DIRECTIVE(".amdhsa_enable_private_segment",
18706fb02596SStanislav Mekhanoshin                     COMPUTE_PGM_RSRC2_ENABLE_PRIVATE_SEGMENT);
18716fb02596SStanislav Mekhanoshin   else
18726fb02596SStanislav Mekhanoshin     PRINT_DIRECTIVE(".amdhsa_system_sgpr_private_segment_wavefront_offset",
1873d5ea8f70STony                     COMPUTE_PGM_RSRC2_ENABLE_PRIVATE_SEGMENT);
1874528057c1SRonak Chauhan   PRINT_DIRECTIVE(".amdhsa_system_sgpr_workgroup_id_x",
1875528057c1SRonak Chauhan                   COMPUTE_PGM_RSRC2_ENABLE_SGPR_WORKGROUP_ID_X);
1876528057c1SRonak Chauhan   PRINT_DIRECTIVE(".amdhsa_system_sgpr_workgroup_id_y",
1877528057c1SRonak Chauhan                   COMPUTE_PGM_RSRC2_ENABLE_SGPR_WORKGROUP_ID_Y);
1878528057c1SRonak Chauhan   PRINT_DIRECTIVE(".amdhsa_system_sgpr_workgroup_id_z",
1879528057c1SRonak Chauhan                   COMPUTE_PGM_RSRC2_ENABLE_SGPR_WORKGROUP_ID_Z);
1880528057c1SRonak Chauhan   PRINT_DIRECTIVE(".amdhsa_system_sgpr_workgroup_info",
1881528057c1SRonak Chauhan                   COMPUTE_PGM_RSRC2_ENABLE_SGPR_WORKGROUP_INFO);
1882528057c1SRonak Chauhan   PRINT_DIRECTIVE(".amdhsa_system_vgpr_workitem_id",
1883528057c1SRonak Chauhan                   COMPUTE_PGM_RSRC2_ENABLE_VGPR_WORKITEM_ID);
1884528057c1SRonak Chauhan 
1885528057c1SRonak Chauhan   if (FourByteBuffer & COMPUTE_PGM_RSRC2_ENABLE_EXCEPTION_ADDRESS_WATCH)
1886528057c1SRonak Chauhan     return MCDisassembler::Fail;
1887528057c1SRonak Chauhan 
1888528057c1SRonak Chauhan   if (FourByteBuffer & COMPUTE_PGM_RSRC2_ENABLE_EXCEPTION_MEMORY)
1889528057c1SRonak Chauhan     return MCDisassembler::Fail;
1890528057c1SRonak Chauhan 
1891528057c1SRonak Chauhan   if (FourByteBuffer & COMPUTE_PGM_RSRC2_GRANULATED_LDS_SIZE)
1892528057c1SRonak Chauhan     return MCDisassembler::Fail;
1893528057c1SRonak Chauhan 
1894528057c1SRonak Chauhan   PRINT_DIRECTIVE(
1895528057c1SRonak Chauhan       ".amdhsa_exception_fp_ieee_invalid_op",
1896528057c1SRonak Chauhan       COMPUTE_PGM_RSRC2_ENABLE_EXCEPTION_IEEE_754_FP_INVALID_OPERATION);
1897528057c1SRonak Chauhan   PRINT_DIRECTIVE(".amdhsa_exception_fp_denorm_src",
1898528057c1SRonak Chauhan                   COMPUTE_PGM_RSRC2_ENABLE_EXCEPTION_FP_DENORMAL_SOURCE);
1899528057c1SRonak Chauhan   PRINT_DIRECTIVE(
1900528057c1SRonak Chauhan       ".amdhsa_exception_fp_ieee_div_zero",
1901528057c1SRonak Chauhan       COMPUTE_PGM_RSRC2_ENABLE_EXCEPTION_IEEE_754_FP_DIVISION_BY_ZERO);
1902528057c1SRonak Chauhan   PRINT_DIRECTIVE(".amdhsa_exception_fp_ieee_overflow",
1903528057c1SRonak Chauhan                   COMPUTE_PGM_RSRC2_ENABLE_EXCEPTION_IEEE_754_FP_OVERFLOW);
1904528057c1SRonak Chauhan   PRINT_DIRECTIVE(".amdhsa_exception_fp_ieee_underflow",
1905528057c1SRonak Chauhan                   COMPUTE_PGM_RSRC2_ENABLE_EXCEPTION_IEEE_754_FP_UNDERFLOW);
1906528057c1SRonak Chauhan   PRINT_DIRECTIVE(".amdhsa_exception_fp_ieee_inexact",
1907528057c1SRonak Chauhan                   COMPUTE_PGM_RSRC2_ENABLE_EXCEPTION_IEEE_754_FP_INEXACT);
1908528057c1SRonak Chauhan   PRINT_DIRECTIVE(".amdhsa_exception_int_div_zero",
1909528057c1SRonak Chauhan                   COMPUTE_PGM_RSRC2_ENABLE_EXCEPTION_INT_DIVIDE_BY_ZERO);
1910528057c1SRonak Chauhan 
1911528057c1SRonak Chauhan   if (FourByteBuffer & COMPUTE_PGM_RSRC2_RESERVED0)
1912528057c1SRonak Chauhan     return MCDisassembler::Fail;
1913528057c1SRonak Chauhan 
1914528057c1SRonak Chauhan   return MCDisassembler::Success;
1915528057c1SRonak Chauhan }
1916528057c1SRonak Chauhan 
1917528057c1SRonak Chauhan #undef PRINT_DIRECTIVE
1918528057c1SRonak Chauhan 
1919528057c1SRonak Chauhan MCDisassembler::DecodeStatus
decodeKernelDescriptorDirective(DataExtractor::Cursor & Cursor,ArrayRef<uint8_t> Bytes,raw_string_ostream & KdStream) const1920528057c1SRonak Chauhan AMDGPUDisassembler::decodeKernelDescriptorDirective(
1921528057c1SRonak Chauhan     DataExtractor::Cursor &Cursor, ArrayRef<uint8_t> Bytes,
1922528057c1SRonak Chauhan     raw_string_ostream &KdStream) const {
1923528057c1SRonak Chauhan #define PRINT_DIRECTIVE(DIRECTIVE, MASK)                                       \
1924528057c1SRonak Chauhan   do {                                                                         \
1925528057c1SRonak Chauhan     KdStream << Indent << DIRECTIVE " "                                        \
1926528057c1SRonak Chauhan              << ((TwoByteBuffer & MASK) >> (MASK##_SHIFT)) << '\n';            \
1927528057c1SRonak Chauhan   } while (0)
1928528057c1SRonak Chauhan 
1929528057c1SRonak Chauhan   uint16_t TwoByteBuffer = 0;
1930528057c1SRonak Chauhan   uint32_t FourByteBuffer = 0;
1931528057c1SRonak Chauhan 
1932528057c1SRonak Chauhan   StringRef ReservedBytes;
1933528057c1SRonak Chauhan   StringRef Indent = "\t";
1934528057c1SRonak Chauhan 
1935528057c1SRonak Chauhan   assert(Bytes.size() == 64);
1936528057c1SRonak Chauhan   DataExtractor DE(Bytes, /*IsLittleEndian=*/true, /*AddressSize=*/8);
1937528057c1SRonak Chauhan 
1938528057c1SRonak Chauhan   switch (Cursor.tell()) {
1939528057c1SRonak Chauhan   case amdhsa::GROUP_SEGMENT_FIXED_SIZE_OFFSET:
1940528057c1SRonak Chauhan     FourByteBuffer = DE.getU32(Cursor);
1941528057c1SRonak Chauhan     KdStream << Indent << ".amdhsa_group_segment_fixed_size " << FourByteBuffer
1942528057c1SRonak Chauhan              << '\n';
1943528057c1SRonak Chauhan     return MCDisassembler::Success;
1944528057c1SRonak Chauhan 
1945528057c1SRonak Chauhan   case amdhsa::PRIVATE_SEGMENT_FIXED_SIZE_OFFSET:
1946528057c1SRonak Chauhan     FourByteBuffer = DE.getU32(Cursor);
1947528057c1SRonak Chauhan     KdStream << Indent << ".amdhsa_private_segment_fixed_size "
1948528057c1SRonak Chauhan              << FourByteBuffer << '\n';
1949528057c1SRonak Chauhan     return MCDisassembler::Success;
1950528057c1SRonak Chauhan 
1951f4ace637SKonstantin Zhuravlyov   case amdhsa::KERNARG_SIZE_OFFSET:
1952f4ace637SKonstantin Zhuravlyov     FourByteBuffer = DE.getU32(Cursor);
1953f4ace637SKonstantin Zhuravlyov     KdStream << Indent << ".amdhsa_kernarg_size "
1954f4ace637SKonstantin Zhuravlyov              << FourByteBuffer << '\n';
1955f4ace637SKonstantin Zhuravlyov     return MCDisassembler::Success;
1956f4ace637SKonstantin Zhuravlyov 
1957528057c1SRonak Chauhan   case amdhsa::RESERVED0_OFFSET:
1958f4ace637SKonstantin Zhuravlyov     // 4 reserved bytes, must be 0.
1959f4ace637SKonstantin Zhuravlyov     ReservedBytes = DE.getBytes(Cursor, 4);
1960f4ace637SKonstantin Zhuravlyov     for (int I = 0; I < 4; ++I) {
1961f4ace637SKonstantin Zhuravlyov       if (ReservedBytes[I] != 0) {
1962528057c1SRonak Chauhan         return MCDisassembler::Fail;
1963528057c1SRonak Chauhan       }
1964f4ace637SKonstantin Zhuravlyov     }
1965528057c1SRonak Chauhan     return MCDisassembler::Success;
1966528057c1SRonak Chauhan 
1967528057c1SRonak Chauhan   case amdhsa::KERNEL_CODE_ENTRY_BYTE_OFFSET_OFFSET:
1968528057c1SRonak Chauhan     // KERNEL_CODE_ENTRY_BYTE_OFFSET
1969528057c1SRonak Chauhan     // So far no directive controls this for Code Object V3, so simply skip for
1970528057c1SRonak Chauhan     // disassembly.
1971528057c1SRonak Chauhan     DE.skip(Cursor, 8);
1972528057c1SRonak Chauhan     return MCDisassembler::Success;
1973528057c1SRonak Chauhan 
1974528057c1SRonak Chauhan   case amdhsa::RESERVED1_OFFSET:
1975528057c1SRonak Chauhan     // 20 reserved bytes, must be 0.
1976528057c1SRonak Chauhan     ReservedBytes = DE.getBytes(Cursor, 20);
1977528057c1SRonak Chauhan     for (int I = 0; I < 20; ++I) {
1978528057c1SRonak Chauhan       if (ReservedBytes[I] != 0) {
1979528057c1SRonak Chauhan         return MCDisassembler::Fail;
1980528057c1SRonak Chauhan       }
1981528057c1SRonak Chauhan     }
1982528057c1SRonak Chauhan     return MCDisassembler::Success;
1983528057c1SRonak Chauhan 
1984528057c1SRonak Chauhan   case amdhsa::COMPUTE_PGM_RSRC3_OFFSET:
1985528057c1SRonak Chauhan     // COMPUTE_PGM_RSRC3
1986528057c1SRonak Chauhan     //  - Only set for GFX10, GFX6-9 have this to be 0.
1987528057c1SRonak Chauhan     //  - Currently no directives directly control this.
1988528057c1SRonak Chauhan     FourByteBuffer = DE.getU32(Cursor);
19894f87d30aSJay Foad     if (!isGFX10Plus() && FourByteBuffer) {
1990528057c1SRonak Chauhan       return MCDisassembler::Fail;
1991528057c1SRonak Chauhan     }
1992528057c1SRonak Chauhan     return MCDisassembler::Success;
1993528057c1SRonak Chauhan 
1994528057c1SRonak Chauhan   case amdhsa::COMPUTE_PGM_RSRC1_OFFSET:
1995528057c1SRonak Chauhan     FourByteBuffer = DE.getU32(Cursor);
1996528057c1SRonak Chauhan     if (decodeCOMPUTE_PGM_RSRC1(FourByteBuffer, KdStream) ==
1997528057c1SRonak Chauhan         MCDisassembler::Fail) {
1998528057c1SRonak Chauhan       return MCDisassembler::Fail;
1999528057c1SRonak Chauhan     }
2000528057c1SRonak Chauhan     return MCDisassembler::Success;
2001528057c1SRonak Chauhan 
2002528057c1SRonak Chauhan   case amdhsa::COMPUTE_PGM_RSRC2_OFFSET:
2003528057c1SRonak Chauhan     FourByteBuffer = DE.getU32(Cursor);
2004528057c1SRonak Chauhan     if (decodeCOMPUTE_PGM_RSRC2(FourByteBuffer, KdStream) ==
2005528057c1SRonak Chauhan         MCDisassembler::Fail) {
2006528057c1SRonak Chauhan       return MCDisassembler::Fail;
2007528057c1SRonak Chauhan     }
2008528057c1SRonak Chauhan     return MCDisassembler::Success;
2009528057c1SRonak Chauhan 
2010528057c1SRonak Chauhan   case amdhsa::KERNEL_CODE_PROPERTIES_OFFSET:
2011528057c1SRonak Chauhan     using namespace amdhsa;
2012528057c1SRonak Chauhan     TwoByteBuffer = DE.getU16(Cursor);
2013528057c1SRonak Chauhan 
20146fb02596SStanislav Mekhanoshin     if (!hasArchitectedFlatScratch())
2015528057c1SRonak Chauhan       PRINT_DIRECTIVE(".amdhsa_user_sgpr_private_segment_buffer",
2016528057c1SRonak Chauhan                       KERNEL_CODE_PROPERTY_ENABLE_SGPR_PRIVATE_SEGMENT_BUFFER);
2017528057c1SRonak Chauhan     PRINT_DIRECTIVE(".amdhsa_user_sgpr_dispatch_ptr",
2018528057c1SRonak Chauhan                     KERNEL_CODE_PROPERTY_ENABLE_SGPR_DISPATCH_PTR);
2019528057c1SRonak Chauhan     PRINT_DIRECTIVE(".amdhsa_user_sgpr_queue_ptr",
2020528057c1SRonak Chauhan                     KERNEL_CODE_PROPERTY_ENABLE_SGPR_QUEUE_PTR);
2021528057c1SRonak Chauhan     PRINT_DIRECTIVE(".amdhsa_user_sgpr_kernarg_segment_ptr",
2022528057c1SRonak Chauhan                     KERNEL_CODE_PROPERTY_ENABLE_SGPR_KERNARG_SEGMENT_PTR);
2023528057c1SRonak Chauhan     PRINT_DIRECTIVE(".amdhsa_user_sgpr_dispatch_id",
2024528057c1SRonak Chauhan                     KERNEL_CODE_PROPERTY_ENABLE_SGPR_DISPATCH_ID);
20256fb02596SStanislav Mekhanoshin     if (!hasArchitectedFlatScratch())
2026528057c1SRonak Chauhan       PRINT_DIRECTIVE(".amdhsa_user_sgpr_flat_scratch_init",
2027528057c1SRonak Chauhan                       KERNEL_CODE_PROPERTY_ENABLE_SGPR_FLAT_SCRATCH_INIT);
2028528057c1SRonak Chauhan     PRINT_DIRECTIVE(".amdhsa_user_sgpr_private_segment_size",
2029528057c1SRonak Chauhan                     KERNEL_CODE_PROPERTY_ENABLE_SGPR_PRIVATE_SEGMENT_SIZE);
2030528057c1SRonak Chauhan 
2031528057c1SRonak Chauhan     if (TwoByteBuffer & KERNEL_CODE_PROPERTY_RESERVED0)
2032528057c1SRonak Chauhan       return MCDisassembler::Fail;
2033528057c1SRonak Chauhan 
2034528057c1SRonak Chauhan     // Reserved for GFX9
2035528057c1SRonak Chauhan     if (isGFX9() &&
2036528057c1SRonak Chauhan         (TwoByteBuffer & KERNEL_CODE_PROPERTY_ENABLE_WAVEFRONT_SIZE32)) {
2037528057c1SRonak Chauhan       return MCDisassembler::Fail;
20384f87d30aSJay Foad     } else if (isGFX10Plus()) {
2039528057c1SRonak Chauhan       PRINT_DIRECTIVE(".amdhsa_wavefront_size32",
2040528057c1SRonak Chauhan                       KERNEL_CODE_PROPERTY_ENABLE_WAVEFRONT_SIZE32);
2041528057c1SRonak Chauhan     }
2042528057c1SRonak Chauhan 
2043*d96361d7SAbinav Puthan Purayil     PRINT_DIRECTIVE(".amdhsa_uses_dynamic_stack",
2044*d96361d7SAbinav Puthan Purayil                     KERNEL_CODE_PROPERTY_USES_DYNAMIC_STACK);
2045*d96361d7SAbinav Puthan Purayil 
2046528057c1SRonak Chauhan     if (TwoByteBuffer & KERNEL_CODE_PROPERTY_RESERVED1)
2047528057c1SRonak Chauhan       return MCDisassembler::Fail;
2048528057c1SRonak Chauhan 
2049528057c1SRonak Chauhan     return MCDisassembler::Success;
2050528057c1SRonak Chauhan 
2051528057c1SRonak Chauhan   case amdhsa::RESERVED2_OFFSET:
2052528057c1SRonak Chauhan     // 6 bytes from here are reserved, must be 0.
2053528057c1SRonak Chauhan     ReservedBytes = DE.getBytes(Cursor, 6);
2054528057c1SRonak Chauhan     for (int I = 0; I < 6; ++I) {
2055528057c1SRonak Chauhan       if (ReservedBytes[I] != 0)
2056528057c1SRonak Chauhan         return MCDisassembler::Fail;
2057528057c1SRonak Chauhan     }
2058528057c1SRonak Chauhan     return MCDisassembler::Success;
2059528057c1SRonak Chauhan 
2060528057c1SRonak Chauhan   default:
2061528057c1SRonak Chauhan     llvm_unreachable("Unhandled index. Case statements cover everything.");
2062528057c1SRonak Chauhan     return MCDisassembler::Fail;
2063528057c1SRonak Chauhan   }
2064528057c1SRonak Chauhan #undef PRINT_DIRECTIVE
2065528057c1SRonak Chauhan }
2066528057c1SRonak Chauhan 
decodeKernelDescriptor(StringRef KdName,ArrayRef<uint8_t> Bytes,uint64_t KdAddress) const2067528057c1SRonak Chauhan MCDisassembler::DecodeStatus AMDGPUDisassembler::decodeKernelDescriptor(
2068528057c1SRonak Chauhan     StringRef KdName, ArrayRef<uint8_t> Bytes, uint64_t KdAddress) const {
2069528057c1SRonak Chauhan   // CP microcode requires the kernel descriptor to be 64 aligned.
2070528057c1SRonak Chauhan   if (Bytes.size() != 64 || KdAddress % 64 != 0)
2071528057c1SRonak Chauhan     return MCDisassembler::Fail;
2072528057c1SRonak Chauhan 
2073528057c1SRonak Chauhan   std::string Kd;
2074528057c1SRonak Chauhan   raw_string_ostream KdStream(Kd);
2075528057c1SRonak Chauhan   KdStream << ".amdhsa_kernel " << KdName << '\n';
2076528057c1SRonak Chauhan 
2077528057c1SRonak Chauhan   DataExtractor::Cursor C(0);
2078528057c1SRonak Chauhan   while (C && C.tell() < Bytes.size()) {
2079528057c1SRonak Chauhan     MCDisassembler::DecodeStatus Status =
2080528057c1SRonak Chauhan         decodeKernelDescriptorDirective(C, Bytes, KdStream);
2081528057c1SRonak Chauhan 
2082528057c1SRonak Chauhan     cantFail(C.takeError());
2083528057c1SRonak Chauhan 
2084528057c1SRonak Chauhan     if (Status == MCDisassembler::Fail)
2085528057c1SRonak Chauhan       return MCDisassembler::Fail;
2086528057c1SRonak Chauhan   }
2087528057c1SRonak Chauhan   KdStream << ".end_amdhsa_kernel\n";
2088528057c1SRonak Chauhan   outs() << KdStream.str();
2089528057c1SRonak Chauhan   return MCDisassembler::Success;
2090528057c1SRonak Chauhan }
2091528057c1SRonak Chauhan 
2092528057c1SRonak Chauhan Optional<MCDisassembler::DecodeStatus>
onSymbolStart(SymbolInfoTy & Symbol,uint64_t & Size,ArrayRef<uint8_t> Bytes,uint64_t Address,raw_ostream & CStream) const2093528057c1SRonak Chauhan AMDGPUDisassembler::onSymbolStart(SymbolInfoTy &Symbol, uint64_t &Size,
2094528057c1SRonak Chauhan                                   ArrayRef<uint8_t> Bytes, uint64_t Address,
2095528057c1SRonak Chauhan                                   raw_ostream &CStream) const {
2096528057c1SRonak Chauhan   // Right now only kernel descriptor needs to be handled.
2097528057c1SRonak Chauhan   // We ignore all other symbols for target specific handling.
2098528057c1SRonak Chauhan   // TODO:
2099528057c1SRonak Chauhan   // Fix the spurious symbol issue for AMDGPU kernels. Exists for both Code
2100528057c1SRonak Chauhan   // Object V2 and V3 when symbols are marked protected.
2101528057c1SRonak Chauhan 
2102528057c1SRonak Chauhan   // amd_kernel_code_t for Code Object V2.
2103528057c1SRonak Chauhan   if (Symbol.Type == ELF::STT_AMDGPU_HSA_KERNEL) {
2104528057c1SRonak Chauhan     Size = 256;
2105528057c1SRonak Chauhan     return MCDisassembler::Fail;
2106528057c1SRonak Chauhan   }
2107528057c1SRonak Chauhan 
2108528057c1SRonak Chauhan   // Code Object V3 kernel descriptors.
2109528057c1SRonak Chauhan   StringRef Name = Symbol.Name;
2110528057c1SRonak Chauhan   if (Symbol.Type == ELF::STT_OBJECT && Name.endswith(StringRef(".kd"))) {
2111528057c1SRonak Chauhan     Size = 64; // Size = 64 regardless of success or failure.
2112528057c1SRonak Chauhan     return decodeKernelDescriptor(Name.drop_back(3), Bytes, Address);
2113528057c1SRonak Chauhan   }
2114528057c1SRonak Chauhan   return None;
2115528057c1SRonak Chauhan }
2116528057c1SRonak Chauhan 
2117528057c1SRonak Chauhan //===----------------------------------------------------------------------===//
21183381d7a2SSam Kolton // AMDGPUSymbolizer
21193381d7a2SSam Kolton //===----------------------------------------------------------------------===//
21203381d7a2SSam Kolton 
21213381d7a2SSam Kolton // Try to find symbol name for specified label
tryAddingSymbolicOperand(MCInst & Inst,raw_ostream &,int64_t Value,uint64_t,bool IsBranch,uint64_t,uint64_t,uint64_t)2122bed9efedSMaksim Panchenko bool AMDGPUSymbolizer::tryAddingSymbolicOperand(
2123bed9efedSMaksim Panchenko     MCInst &Inst, raw_ostream & /*cStream*/, int64_t Value,
2124bed9efedSMaksim Panchenko     uint64_t /*Address*/, bool IsBranch, uint64_t /*Offset*/,
2125bed9efedSMaksim Panchenko     uint64_t /*OpSize*/, uint64_t /*InstSize*/) {
21263381d7a2SSam Kolton 
21273381d7a2SSam Kolton   if (!IsBranch) {
21283381d7a2SSam Kolton     return false;
21293381d7a2SSam Kolton   }
21303381d7a2SSam Kolton 
21313381d7a2SSam Kolton   auto *Symbols = static_cast<SectionSymbolsTy *>(DisInfo);
2132b1c3b22bSNicolai Haehnle   if (!Symbols)
2133b1c3b22bSNicolai Haehnle     return false;
2134b1c3b22bSNicolai Haehnle 
2135b934160aSKazu Hirata   auto Result = llvm::find_if(*Symbols, [Value](const SymbolInfoTy &Val) {
2136b934160aSKazu Hirata     return Val.Addr == static_cast<uint64_t>(Value) &&
2137b934160aSKazu Hirata            Val.Type == ELF::STT_NOTYPE;
21383381d7a2SSam Kolton   });
21393381d7a2SSam Kolton   if (Result != Symbols->end()) {
214009d26b79Sdiggerlin     auto *Sym = Ctx.getOrCreateSymbol(Result->Name);
21413381d7a2SSam Kolton     const auto *Add = MCSymbolRefExpr::create(Sym, Ctx);
21423381d7a2SSam Kolton     Inst.addOperand(MCOperand::createExpr(Add));
21433381d7a2SSam Kolton     return true;
21443381d7a2SSam Kolton   }
21458710eff6STim Renouf   // Add to list of referenced addresses, so caller can synthesize a label.
21468710eff6STim Renouf   ReferencedAddresses.push_back(static_cast<uint64_t>(Value));
21473381d7a2SSam Kolton   return false;
21483381d7a2SSam Kolton }
21493381d7a2SSam Kolton 
tryAddingPcLoadReferenceComment(raw_ostream & cStream,int64_t Value,uint64_t Address)215092b355b1SMatt Arsenault void AMDGPUSymbolizer::tryAddingPcLoadReferenceComment(raw_ostream &cStream,
215192b355b1SMatt Arsenault                                                        int64_t Value,
215292b355b1SMatt Arsenault                                                        uint64_t Address) {
215392b355b1SMatt Arsenault   llvm_unreachable("unimplemented");
215492b355b1SMatt Arsenault }
215592b355b1SMatt Arsenault 
21563381d7a2SSam Kolton //===----------------------------------------------------------------------===//
21573381d7a2SSam Kolton // Initialization
21583381d7a2SSam Kolton //===----------------------------------------------------------------------===//
21593381d7a2SSam Kolton 
createAMDGPUSymbolizer(const Triple &,LLVMOpInfoCallback,LLVMSymbolLookupCallback,void * DisInfo,MCContext * Ctx,std::unique_ptr<MCRelocationInfo> && RelInfo)21603381d7a2SSam Kolton static MCSymbolizer *createAMDGPUSymbolizer(const Triple &/*TT*/,
21613381d7a2SSam Kolton                               LLVMOpInfoCallback /*GetOpInfo*/,
21623381d7a2SSam Kolton                               LLVMSymbolLookupCallback /*SymbolLookUp*/,
21633381d7a2SSam Kolton                               void *DisInfo,
21643381d7a2SSam Kolton                               MCContext *Ctx,
21653381d7a2SSam Kolton                               std::unique_ptr<MCRelocationInfo> &&RelInfo) {
21663381d7a2SSam Kolton   return new AMDGPUSymbolizer(*Ctx, std::move(RelInfo), DisInfo);
21673381d7a2SSam Kolton }
21683381d7a2SSam Kolton 
createAMDGPUDisassembler(const Target & T,const MCSubtargetInfo & STI,MCContext & Ctx)2169e1818af8STom Stellard static MCDisassembler *createAMDGPUDisassembler(const Target &T,
2170e1818af8STom Stellard                                                 const MCSubtargetInfo &STI,
2171e1818af8STom Stellard                                                 MCContext &Ctx) {
2172cad7fa85SMatt Arsenault   return new AMDGPUDisassembler(STI, Ctx, T.createMCInstrInfo());
2173e1818af8STom Stellard }
2174e1818af8STom Stellard 
LLVMInitializeAMDGPUDisassembler()21750dbcb363STom Stellard extern "C" LLVM_EXTERNAL_VISIBILITY void LLVMInitializeAMDGPUDisassembler() {
2176f42454b9SMehdi Amini   TargetRegistry::RegisterMCDisassembler(getTheGCNTarget(),
2177f42454b9SMehdi Amini                                          createAMDGPUDisassembler);
2178f42454b9SMehdi Amini   TargetRegistry::RegisterMCSymbolizer(getTheGCNTarget(),
2179f42454b9SMehdi Amini                                        createAMDGPUSymbolizer);
2180e1818af8STom Stellard }
2181