1e1818af8STom Stellard //===-- AMDGPUDisassembler.cpp - Disassembler for AMDGPU ISA --------------===// 2e1818af8STom Stellard // 3e1818af8STom Stellard // The LLVM Compiler Infrastructure 4e1818af8STom Stellard // 5e1818af8STom Stellard // This file is distributed under the University of Illinois Open Source 6e1818af8STom Stellard // License. See LICENSE.TXT for details. 7e1818af8STom Stellard // 8e1818af8STom Stellard //===----------------------------------------------------------------------===// 9e1818af8STom Stellard // 10e1818af8STom Stellard //===----------------------------------------------------------------------===// 11e1818af8STom Stellard // 12e1818af8STom Stellard /// \file 13e1818af8STom Stellard /// 14e1818af8STom Stellard /// This file contains definition for AMDGPU ISA disassembler 15e1818af8STom Stellard // 16e1818af8STom Stellard //===----------------------------------------------------------------------===// 17e1818af8STom Stellard 18e1818af8STom Stellard // ToDo: What to do with instruction suffixes (v_mov_b32 vs v_mov_b32_e32)? 19e1818af8STom Stellard 20e1818af8STom Stellard #include "AMDGPUDisassembler.h" 21e1818af8STom Stellard #include "AMDGPU.h" 22e1818af8STom Stellard #include "AMDGPURegisterInfo.h" 236bda14b3SChandler Carruth #include "MCTargetDesc/AMDGPUMCTargetDesc.h" 24212a251cSArtem Tamazov #include "SIDefines.h" 25e1818af8STom Stellard #include "Utils/AMDGPUBaseInfo.h" 26e1818af8STom Stellard 27*264b5d9eSZachary Turner #include "llvm/BinaryFormat/ELF.h" 28ac106addSNikolay Haustov #include "llvm/MC/MCContext.h" 29e1818af8STom Stellard #include "llvm/MC/MCFixedLenDisassembler.h" 30e1818af8STom Stellard #include "llvm/MC/MCInst.h" 31e1818af8STom Stellard #include "llvm/MC/MCInstrDesc.h" 32e1818af8STom Stellard #include "llvm/MC/MCSubtargetInfo.h" 336bda14b3SChandler Carruth #include "llvm/Support/Debug.h" 34ac106addSNikolay Haustov #include "llvm/Support/Endian.h" 35e1818af8STom Stellard #include "llvm/Support/TargetRegistry.h" 36e1818af8STom Stellard 37e1818af8STom Stellard using namespace llvm; 38e1818af8STom Stellard 39e1818af8STom Stellard #define DEBUG_TYPE "amdgpu-disassembler" 40e1818af8STom Stellard 41e1818af8STom Stellard typedef llvm::MCDisassembler::DecodeStatus DecodeStatus; 42e1818af8STom Stellard 43e1818af8STom Stellard 44ac106addSNikolay Haustov inline static MCDisassembler::DecodeStatus 45ac106addSNikolay Haustov addOperand(MCInst &Inst, const MCOperand& Opnd) { 46ac106addSNikolay Haustov Inst.addOperand(Opnd); 47ac106addSNikolay Haustov return Opnd.isValid() ? 48ac106addSNikolay Haustov MCDisassembler::Success : 49ac106addSNikolay Haustov MCDisassembler::SoftFail; 50e1818af8STom Stellard } 51e1818af8STom Stellard 523381d7a2SSam Kolton static DecodeStatus decodeSoppBrTarget(MCInst &Inst, unsigned Imm, 533381d7a2SSam Kolton uint64_t Addr, const void *Decoder) { 543381d7a2SSam Kolton auto DAsm = static_cast<const AMDGPUDisassembler*>(Decoder); 553381d7a2SSam Kolton 563381d7a2SSam Kolton APInt SignedOffset(18, Imm * 4, true); 573381d7a2SSam Kolton int64_t Offset = (SignedOffset.sext(64) + 4 + Addr).getSExtValue(); 583381d7a2SSam Kolton 593381d7a2SSam Kolton if (DAsm->tryAddingSymbolicOperand(Inst, Offset, Addr, true, 2, 2)) 603381d7a2SSam Kolton return MCDisassembler::Success; 613381d7a2SSam Kolton return addOperand(Inst, MCOperand::createImm(Imm)); 623381d7a2SSam Kolton } 633381d7a2SSam Kolton 64363f47a2SSam Kolton #define DECODE_OPERAND(StaticDecoderName, DecoderName) \ 65363f47a2SSam Kolton static DecodeStatus StaticDecoderName(MCInst &Inst, \ 66ac106addSNikolay Haustov unsigned Imm, \ 67ac106addSNikolay Haustov uint64_t /*Addr*/, \ 68ac106addSNikolay Haustov const void *Decoder) { \ 69ac106addSNikolay Haustov auto DAsm = static_cast<const AMDGPUDisassembler*>(Decoder); \ 70363f47a2SSam Kolton return addOperand(Inst, DAsm->DecoderName(Imm)); \ 71e1818af8STom Stellard } 72e1818af8STom Stellard 73363f47a2SSam Kolton #define DECODE_OPERAND_REG(RegClass) \ 74363f47a2SSam Kolton DECODE_OPERAND(Decode##RegClass##RegisterClass, decodeOperand_##RegClass) 75e1818af8STom Stellard 76363f47a2SSam Kolton DECODE_OPERAND_REG(VGPR_32) 77363f47a2SSam Kolton DECODE_OPERAND_REG(VS_32) 78363f47a2SSam Kolton DECODE_OPERAND_REG(VS_64) 79e1818af8STom Stellard 80363f47a2SSam Kolton DECODE_OPERAND_REG(VReg_64) 81363f47a2SSam Kolton DECODE_OPERAND_REG(VReg_96) 82363f47a2SSam Kolton DECODE_OPERAND_REG(VReg_128) 83e1818af8STom Stellard 84363f47a2SSam Kolton DECODE_OPERAND_REG(SReg_32) 85363f47a2SSam Kolton DECODE_OPERAND_REG(SReg_32_XM0_XEXEC) 86363f47a2SSam Kolton DECODE_OPERAND_REG(SReg_64) 87363f47a2SSam Kolton DECODE_OPERAND_REG(SReg_64_XEXEC) 88363f47a2SSam Kolton DECODE_OPERAND_REG(SReg_128) 89363f47a2SSam Kolton DECODE_OPERAND_REG(SReg_256) 90363f47a2SSam Kolton DECODE_OPERAND_REG(SReg_512) 91e1818af8STom Stellard 924bd72361SMatt Arsenault 934bd72361SMatt Arsenault static DecodeStatus decodeOperand_VSrc16(MCInst &Inst, 944bd72361SMatt Arsenault unsigned Imm, 954bd72361SMatt Arsenault uint64_t Addr, 964bd72361SMatt Arsenault const void *Decoder) { 974bd72361SMatt Arsenault auto DAsm = static_cast<const AMDGPUDisassembler*>(Decoder); 984bd72361SMatt Arsenault return addOperand(Inst, DAsm->decodeOperand_VSrc16(Imm)); 994bd72361SMatt Arsenault } 1004bd72361SMatt Arsenault 1019be7b0d4SMatt Arsenault static DecodeStatus decodeOperand_VSrcV216(MCInst &Inst, 1029be7b0d4SMatt Arsenault unsigned Imm, 1039be7b0d4SMatt Arsenault uint64_t Addr, 1049be7b0d4SMatt Arsenault const void *Decoder) { 1059be7b0d4SMatt Arsenault auto DAsm = static_cast<const AMDGPUDisassembler*>(Decoder); 1069be7b0d4SMatt Arsenault return addOperand(Inst, DAsm->decodeOperand_VSrcV216(Imm)); 1079be7b0d4SMatt Arsenault } 1089be7b0d4SMatt Arsenault 109363f47a2SSam Kolton #define DECODE_SDWA9(DecName) \ 110363f47a2SSam Kolton DECODE_OPERAND(decodeSDWA9##DecName, decodeSDWA9##DecName) 111363f47a2SSam Kolton 112363f47a2SSam Kolton DECODE_SDWA9(Src32) 113363f47a2SSam Kolton DECODE_SDWA9(Src16) 114363f47a2SSam Kolton DECODE_SDWA9(VopcDst) 115363f47a2SSam Kolton 116e1818af8STom Stellard #include "AMDGPUGenDisassemblerTables.inc" 117e1818af8STom Stellard 118e1818af8STom Stellard //===----------------------------------------------------------------------===// 119e1818af8STom Stellard // 120e1818af8STom Stellard //===----------------------------------------------------------------------===// 121e1818af8STom Stellard 1221048fb18SSam Kolton template <typename T> static inline T eatBytes(ArrayRef<uint8_t>& Bytes) { 1231048fb18SSam Kolton assert(Bytes.size() >= sizeof(T)); 1241048fb18SSam Kolton const auto Res = support::endian::read<T, support::endianness::little>(Bytes.data()); 1251048fb18SSam Kolton Bytes = Bytes.slice(sizeof(T)); 126ac106addSNikolay Haustov return Res; 127ac106addSNikolay Haustov } 128ac106addSNikolay Haustov 129ac106addSNikolay Haustov DecodeStatus AMDGPUDisassembler::tryDecodeInst(const uint8_t* Table, 130ac106addSNikolay Haustov MCInst &MI, 131ac106addSNikolay Haustov uint64_t Inst, 132ac106addSNikolay Haustov uint64_t Address) const { 133ac106addSNikolay Haustov assert(MI.getOpcode() == 0); 134ac106addSNikolay Haustov assert(MI.getNumOperands() == 0); 135ac106addSNikolay Haustov MCInst TmpInst; 136ce941c9cSDmitry Preobrazhensky HasLiteral = false; 137ac106addSNikolay Haustov const auto SavedBytes = Bytes; 138ac106addSNikolay Haustov if (decodeInstruction(Table, TmpInst, Inst, Address, this, STI)) { 139ac106addSNikolay Haustov MI = TmpInst; 140ac106addSNikolay Haustov return MCDisassembler::Success; 141ac106addSNikolay Haustov } 142ac106addSNikolay Haustov Bytes = SavedBytes; 143ac106addSNikolay Haustov return MCDisassembler::Fail; 144ac106addSNikolay Haustov } 145ac106addSNikolay Haustov 146e1818af8STom Stellard DecodeStatus AMDGPUDisassembler::getInstruction(MCInst &MI, uint64_t &Size, 147ac106addSNikolay Haustov ArrayRef<uint8_t> Bytes_, 148e1818af8STom Stellard uint64_t Address, 149e1818af8STom Stellard raw_ostream &WS, 150e1818af8STom Stellard raw_ostream &CS) const { 151e1818af8STom Stellard CommentStream = &CS; 152e1818af8STom Stellard 153e1818af8STom Stellard // ToDo: AMDGPUDisassembler supports only VI ISA. 154d122abeaSMatt Arsenault if (!STI.getFeatureBits()[AMDGPU::FeatureGCN3Encoding]) 155d122abeaSMatt Arsenault report_fatal_error("Disassembly not yet supported for subtarget"); 156e1818af8STom Stellard 157ac106addSNikolay Haustov const unsigned MaxInstBytesNum = (std::min)((size_t)8, Bytes_.size()); 158ac106addSNikolay Haustov Bytes = Bytes_.slice(0, MaxInstBytesNum); 159161a158eSNikolay Haustov 160ac106addSNikolay Haustov DecodeStatus Res = MCDisassembler::Fail; 161ac106addSNikolay Haustov do { 162824e804bSValery Pykhtin // ToDo: better to switch encoding length using some bit predicate 163ac106addSNikolay Haustov // but it is unknown yet, so try all we can 1641048fb18SSam Kolton 165c9bdcb75SSam Kolton // Try to decode DPP and SDWA first to solve conflict with VOP1 and VOP2 166c9bdcb75SSam Kolton // encodings 1671048fb18SSam Kolton if (Bytes.size() >= 8) { 1681048fb18SSam Kolton const uint64_t QW = eatBytes<uint64_t>(Bytes); 1691048fb18SSam Kolton Res = tryDecodeInst(DecoderTableDPP64, MI, QW, Address); 1701048fb18SSam Kolton if (Res) break; 171c9bdcb75SSam Kolton 172c9bdcb75SSam Kolton Res = tryDecodeInst(DecoderTableSDWA64, MI, QW, Address); 173c9bdcb75SSam Kolton if (Res) break; 174363f47a2SSam Kolton 175363f47a2SSam Kolton Res = tryDecodeInst(DecoderTableSDWA964, MI, QW, Address); 176363f47a2SSam Kolton if (Res) break; 1771048fb18SSam Kolton } 1781048fb18SSam Kolton 1791048fb18SSam Kolton // Reinitialize Bytes as DPP64 could have eaten too much 1801048fb18SSam Kolton Bytes = Bytes_.slice(0, MaxInstBytesNum); 1811048fb18SSam Kolton 1821048fb18SSam Kolton // Try decode 32-bit instruction 183ac106addSNikolay Haustov if (Bytes.size() < 4) break; 1841048fb18SSam Kolton const uint32_t DW = eatBytes<uint32_t>(Bytes); 185ac106addSNikolay Haustov Res = tryDecodeInst(DecoderTableVI32, MI, DW, Address); 186ac106addSNikolay Haustov if (Res) break; 187e1818af8STom Stellard 188ac106addSNikolay Haustov Res = tryDecodeInst(DecoderTableAMDGPU32, MI, DW, Address); 189ac106addSNikolay Haustov if (Res) break; 190ac106addSNikolay Haustov 191ac106addSNikolay Haustov if (Bytes.size() < 4) break; 1921048fb18SSam Kolton const uint64_t QW = ((uint64_t)eatBytes<uint32_t>(Bytes) << 32) | DW; 193ac106addSNikolay Haustov Res = tryDecodeInst(DecoderTableVI64, MI, QW, Address); 194ac106addSNikolay Haustov if (Res) break; 195ac106addSNikolay Haustov 196ac106addSNikolay Haustov Res = tryDecodeInst(DecoderTableAMDGPU64, MI, QW, Address); 197ac106addSNikolay Haustov } while (false); 198ac106addSNikolay Haustov 199678e111eSMatt Arsenault if (Res && (MI.getOpcode() == AMDGPU::V_MAC_F32_e64_vi || 200678e111eSMatt Arsenault MI.getOpcode() == AMDGPU::V_MAC_F32_e64_si || 201678e111eSMatt Arsenault MI.getOpcode() == AMDGPU::V_MAC_F16_e64_vi)) { 202678e111eSMatt Arsenault // Insert dummy unused src2_modifiers. 203678e111eSMatt Arsenault int Src2ModIdx = AMDGPU::getNamedOperandIdx(MI.getOpcode(), 204678e111eSMatt Arsenault AMDGPU::OpName::src2_modifiers); 205678e111eSMatt Arsenault auto I = MI.begin(); 206678e111eSMatt Arsenault std::advance(I, Src2ModIdx); 207678e111eSMatt Arsenault MI.insert(I, MCOperand::createImm(0)); 208678e111eSMatt Arsenault } 209678e111eSMatt Arsenault 210ac106addSNikolay Haustov Size = Res ? (MaxInstBytesNum - Bytes.size()) : 0; 211ac106addSNikolay Haustov return Res; 212161a158eSNikolay Haustov } 213e1818af8STom Stellard 214ac106addSNikolay Haustov const char* AMDGPUDisassembler::getRegClassName(unsigned RegClassID) const { 215ac106addSNikolay Haustov return getContext().getRegisterInfo()-> 216ac106addSNikolay Haustov getRegClassName(&AMDGPUMCRegisterClasses[RegClassID]); 217e1818af8STom Stellard } 218e1818af8STom Stellard 219ac106addSNikolay Haustov inline 220ac106addSNikolay Haustov MCOperand AMDGPUDisassembler::errOperand(unsigned V, 221ac106addSNikolay Haustov const Twine& ErrMsg) const { 222ac106addSNikolay Haustov *CommentStream << "Error: " + ErrMsg; 223ac106addSNikolay Haustov 224ac106addSNikolay Haustov // ToDo: add support for error operands to MCInst.h 225ac106addSNikolay Haustov // return MCOperand::createError(V); 226ac106addSNikolay Haustov return MCOperand(); 227ac106addSNikolay Haustov } 228ac106addSNikolay Haustov 229ac106addSNikolay Haustov inline 230ac106addSNikolay Haustov MCOperand AMDGPUDisassembler::createRegOperand(unsigned int RegId) const { 231ac106addSNikolay Haustov return MCOperand::createReg(RegId); 232ac106addSNikolay Haustov } 233ac106addSNikolay Haustov 234ac106addSNikolay Haustov inline 235ac106addSNikolay Haustov MCOperand AMDGPUDisassembler::createRegOperand(unsigned RegClassID, 236ac106addSNikolay Haustov unsigned Val) const { 237ac106addSNikolay Haustov const auto& RegCl = AMDGPUMCRegisterClasses[RegClassID]; 238ac106addSNikolay Haustov if (Val >= RegCl.getNumRegs()) 239ac106addSNikolay Haustov return errOperand(Val, Twine(getRegClassName(RegClassID)) + 240ac106addSNikolay Haustov ": unknown register " + Twine(Val)); 241ac106addSNikolay Haustov return createRegOperand(RegCl.getRegister(Val)); 242ac106addSNikolay Haustov } 243ac106addSNikolay Haustov 244ac106addSNikolay Haustov inline 245ac106addSNikolay Haustov MCOperand AMDGPUDisassembler::createSRegOperand(unsigned SRegClassID, 246ac106addSNikolay Haustov unsigned Val) const { 247ac106addSNikolay Haustov // ToDo: SI/CI have 104 SGPRs, VI - 102 248ac106addSNikolay Haustov // Valery: here we accepting as much as we can, let assembler sort it out 249ac106addSNikolay Haustov int shift = 0; 250ac106addSNikolay Haustov switch (SRegClassID) { 251ac106addSNikolay Haustov case AMDGPU::SGPR_32RegClassID: 252212a251cSArtem Tamazov case AMDGPU::TTMP_32RegClassID: 253212a251cSArtem Tamazov break; 254ac106addSNikolay Haustov case AMDGPU::SGPR_64RegClassID: 255212a251cSArtem Tamazov case AMDGPU::TTMP_64RegClassID: 256212a251cSArtem Tamazov shift = 1; 257212a251cSArtem Tamazov break; 258212a251cSArtem Tamazov case AMDGPU::SGPR_128RegClassID: 259212a251cSArtem Tamazov case AMDGPU::TTMP_128RegClassID: 260ac106addSNikolay Haustov // ToDo: unclear if s[100:104] is available on VI. Can we use VCC as SGPR in 261ac106addSNikolay Haustov // this bundle? 262ac106addSNikolay Haustov case AMDGPU::SReg_256RegClassID: 263ac106addSNikolay Haustov // ToDo: unclear if s[96:104] is available on VI. Can we use VCC as SGPR in 264ac106addSNikolay Haustov // this bundle? 265212a251cSArtem Tamazov case AMDGPU::SReg_512RegClassID: 266212a251cSArtem Tamazov shift = 2; 267212a251cSArtem Tamazov break; 268ac106addSNikolay Haustov // ToDo: unclear if s[88:104] is available on VI. Can we use VCC as SGPR in 269ac106addSNikolay Haustov // this bundle? 270212a251cSArtem Tamazov default: 27192b355b1SMatt Arsenault llvm_unreachable("unhandled register class"); 272ac106addSNikolay Haustov } 27392b355b1SMatt Arsenault 27492b355b1SMatt Arsenault if (Val % (1 << shift)) { 275ac106addSNikolay Haustov *CommentStream << "Warning: " << getRegClassName(SRegClassID) 276ac106addSNikolay Haustov << ": scalar reg isn't aligned " << Val; 27792b355b1SMatt Arsenault } 27892b355b1SMatt Arsenault 279ac106addSNikolay Haustov return createRegOperand(SRegClassID, Val >> shift); 280ac106addSNikolay Haustov } 281ac106addSNikolay Haustov 282ac106addSNikolay Haustov MCOperand AMDGPUDisassembler::decodeOperand_VS_32(unsigned Val) const { 283212a251cSArtem Tamazov return decodeSrcOp(OPW32, Val); 284ac106addSNikolay Haustov } 285ac106addSNikolay Haustov 286ac106addSNikolay Haustov MCOperand AMDGPUDisassembler::decodeOperand_VS_64(unsigned Val) const { 287212a251cSArtem Tamazov return decodeSrcOp(OPW64, Val); 288ac106addSNikolay Haustov } 289ac106addSNikolay Haustov 2904bd72361SMatt Arsenault MCOperand AMDGPUDisassembler::decodeOperand_VSrc16(unsigned Val) const { 2914bd72361SMatt Arsenault return decodeSrcOp(OPW16, Val); 2924bd72361SMatt Arsenault } 2934bd72361SMatt Arsenault 2949be7b0d4SMatt Arsenault MCOperand AMDGPUDisassembler::decodeOperand_VSrcV216(unsigned Val) const { 2959be7b0d4SMatt Arsenault return decodeSrcOp(OPWV216, Val); 2969be7b0d4SMatt Arsenault } 2979be7b0d4SMatt Arsenault 298ac106addSNikolay Haustov MCOperand AMDGPUDisassembler::decodeOperand_VGPR_32(unsigned Val) const { 299cb540bc0SMatt Arsenault // Some instructions have operand restrictions beyond what the encoding 300cb540bc0SMatt Arsenault // allows. Some ordinarily VSrc_32 operands are VGPR_32, so clear the extra 301cb540bc0SMatt Arsenault // high bit. 302cb540bc0SMatt Arsenault Val &= 255; 303cb540bc0SMatt Arsenault 304ac106addSNikolay Haustov return createRegOperand(AMDGPU::VGPR_32RegClassID, Val); 305ac106addSNikolay Haustov } 306ac106addSNikolay Haustov 307ac106addSNikolay Haustov MCOperand AMDGPUDisassembler::decodeOperand_VReg_64(unsigned Val) const { 308ac106addSNikolay Haustov return createRegOperand(AMDGPU::VReg_64RegClassID, Val); 309ac106addSNikolay Haustov } 310ac106addSNikolay Haustov 311ac106addSNikolay Haustov MCOperand AMDGPUDisassembler::decodeOperand_VReg_96(unsigned Val) const { 312ac106addSNikolay Haustov return createRegOperand(AMDGPU::VReg_96RegClassID, Val); 313ac106addSNikolay Haustov } 314ac106addSNikolay Haustov 315ac106addSNikolay Haustov MCOperand AMDGPUDisassembler::decodeOperand_VReg_128(unsigned Val) const { 316ac106addSNikolay Haustov return createRegOperand(AMDGPU::VReg_128RegClassID, Val); 317ac106addSNikolay Haustov } 318ac106addSNikolay Haustov 319ac106addSNikolay Haustov MCOperand AMDGPUDisassembler::decodeOperand_SReg_32(unsigned Val) const { 320ac106addSNikolay Haustov // table-gen generated disassembler doesn't care about operand types 321ac106addSNikolay Haustov // leaving only registry class so SSrc_32 operand turns into SReg_32 322ac106addSNikolay Haustov // and therefore we accept immediates and literals here as well 323212a251cSArtem Tamazov return decodeSrcOp(OPW32, Val); 324ac106addSNikolay Haustov } 325ac106addSNikolay Haustov 326640c44b8SMatt Arsenault MCOperand AMDGPUDisassembler::decodeOperand_SReg_32_XM0_XEXEC( 327640c44b8SMatt Arsenault unsigned Val) const { 328640c44b8SMatt Arsenault // SReg_32_XM0 is SReg_32 without M0 or EXEC_LO/EXEC_HI 32938e496b1SArtem Tamazov return decodeOperand_SReg_32(Val); 33038e496b1SArtem Tamazov } 33138e496b1SArtem Tamazov 332ac106addSNikolay Haustov MCOperand AMDGPUDisassembler::decodeOperand_SReg_64(unsigned Val) const { 333640c44b8SMatt Arsenault return decodeSrcOp(OPW64, Val); 334640c44b8SMatt Arsenault } 335640c44b8SMatt Arsenault 336640c44b8SMatt Arsenault MCOperand AMDGPUDisassembler::decodeOperand_SReg_64_XEXEC(unsigned Val) const { 337212a251cSArtem Tamazov return decodeSrcOp(OPW64, Val); 338ac106addSNikolay Haustov } 339ac106addSNikolay Haustov 340ac106addSNikolay Haustov MCOperand AMDGPUDisassembler::decodeOperand_SReg_128(unsigned Val) const { 341212a251cSArtem Tamazov return decodeSrcOp(OPW128, Val); 342ac106addSNikolay Haustov } 343ac106addSNikolay Haustov 344ac106addSNikolay Haustov MCOperand AMDGPUDisassembler::decodeOperand_SReg_256(unsigned Val) const { 345ac106addSNikolay Haustov return createSRegOperand(AMDGPU::SReg_256RegClassID, Val); 346ac106addSNikolay Haustov } 347ac106addSNikolay Haustov 348ac106addSNikolay Haustov MCOperand AMDGPUDisassembler::decodeOperand_SReg_512(unsigned Val) const { 349ac106addSNikolay Haustov return createSRegOperand(AMDGPU::SReg_512RegClassID, Val); 350ac106addSNikolay Haustov } 351ac106addSNikolay Haustov 352ac106addSNikolay Haustov 353ac106addSNikolay Haustov MCOperand AMDGPUDisassembler::decodeLiteralConstant() const { 354ac106addSNikolay Haustov // For now all literal constants are supposed to be unsigned integer 355ac106addSNikolay Haustov // ToDo: deal with signed/unsigned 64-bit integer constants 356ac106addSNikolay Haustov // ToDo: deal with float/double constants 357ce941c9cSDmitry Preobrazhensky if (!HasLiteral) { 358ce941c9cSDmitry Preobrazhensky if (Bytes.size() < 4) { 359ac106addSNikolay Haustov return errOperand(0, "cannot read literal, inst bytes left " + 360ac106addSNikolay Haustov Twine(Bytes.size())); 361ce941c9cSDmitry Preobrazhensky } 362ce941c9cSDmitry Preobrazhensky HasLiteral = true; 363ce941c9cSDmitry Preobrazhensky Literal = eatBytes<uint32_t>(Bytes); 364ce941c9cSDmitry Preobrazhensky } 365ce941c9cSDmitry Preobrazhensky return MCOperand::createImm(Literal); 366ac106addSNikolay Haustov } 367ac106addSNikolay Haustov 368ac106addSNikolay Haustov MCOperand AMDGPUDisassembler::decodeIntImmed(unsigned Imm) { 369212a251cSArtem Tamazov using namespace AMDGPU::EncValues; 370212a251cSArtem Tamazov assert(Imm >= INLINE_INTEGER_C_MIN && Imm <= INLINE_INTEGER_C_MAX); 371212a251cSArtem Tamazov return MCOperand::createImm((Imm <= INLINE_INTEGER_C_POSITIVE_MAX) ? 372212a251cSArtem Tamazov (static_cast<int64_t>(Imm) - INLINE_INTEGER_C_MIN) : 373212a251cSArtem Tamazov (INLINE_INTEGER_C_POSITIVE_MAX - static_cast<int64_t>(Imm))); 374212a251cSArtem Tamazov // Cast prevents negative overflow. 375ac106addSNikolay Haustov } 376ac106addSNikolay Haustov 3774bd72361SMatt Arsenault static int64_t getInlineImmVal32(unsigned Imm) { 3784bd72361SMatt Arsenault switch (Imm) { 3794bd72361SMatt Arsenault case 240: 3804bd72361SMatt Arsenault return FloatToBits(0.5f); 3814bd72361SMatt Arsenault case 241: 3824bd72361SMatt Arsenault return FloatToBits(-0.5f); 3834bd72361SMatt Arsenault case 242: 3844bd72361SMatt Arsenault return FloatToBits(1.0f); 3854bd72361SMatt Arsenault case 243: 3864bd72361SMatt Arsenault return FloatToBits(-1.0f); 3874bd72361SMatt Arsenault case 244: 3884bd72361SMatt Arsenault return FloatToBits(2.0f); 3894bd72361SMatt Arsenault case 245: 3904bd72361SMatt Arsenault return FloatToBits(-2.0f); 3914bd72361SMatt Arsenault case 246: 3924bd72361SMatt Arsenault return FloatToBits(4.0f); 3934bd72361SMatt Arsenault case 247: 3944bd72361SMatt Arsenault return FloatToBits(-4.0f); 3954bd72361SMatt Arsenault case 248: // 1 / (2 * PI) 3964bd72361SMatt Arsenault return 0x3e22f983; 3974bd72361SMatt Arsenault default: 3984bd72361SMatt Arsenault llvm_unreachable("invalid fp inline imm"); 3994bd72361SMatt Arsenault } 4004bd72361SMatt Arsenault } 4014bd72361SMatt Arsenault 4024bd72361SMatt Arsenault static int64_t getInlineImmVal64(unsigned Imm) { 4034bd72361SMatt Arsenault switch (Imm) { 4044bd72361SMatt Arsenault case 240: 4054bd72361SMatt Arsenault return DoubleToBits(0.5); 4064bd72361SMatt Arsenault case 241: 4074bd72361SMatt Arsenault return DoubleToBits(-0.5); 4084bd72361SMatt Arsenault case 242: 4094bd72361SMatt Arsenault return DoubleToBits(1.0); 4104bd72361SMatt Arsenault case 243: 4114bd72361SMatt Arsenault return DoubleToBits(-1.0); 4124bd72361SMatt Arsenault case 244: 4134bd72361SMatt Arsenault return DoubleToBits(2.0); 4144bd72361SMatt Arsenault case 245: 4154bd72361SMatt Arsenault return DoubleToBits(-2.0); 4164bd72361SMatt Arsenault case 246: 4174bd72361SMatt Arsenault return DoubleToBits(4.0); 4184bd72361SMatt Arsenault case 247: 4194bd72361SMatt Arsenault return DoubleToBits(-4.0); 4204bd72361SMatt Arsenault case 248: // 1 / (2 * PI) 4214bd72361SMatt Arsenault return 0x3fc45f306dc9c882; 4224bd72361SMatt Arsenault default: 4234bd72361SMatt Arsenault llvm_unreachable("invalid fp inline imm"); 4244bd72361SMatt Arsenault } 4254bd72361SMatt Arsenault } 4264bd72361SMatt Arsenault 4274bd72361SMatt Arsenault static int64_t getInlineImmVal16(unsigned Imm) { 4284bd72361SMatt Arsenault switch (Imm) { 4294bd72361SMatt Arsenault case 240: 4304bd72361SMatt Arsenault return 0x3800; 4314bd72361SMatt Arsenault case 241: 4324bd72361SMatt Arsenault return 0xB800; 4334bd72361SMatt Arsenault case 242: 4344bd72361SMatt Arsenault return 0x3C00; 4354bd72361SMatt Arsenault case 243: 4364bd72361SMatt Arsenault return 0xBC00; 4374bd72361SMatt Arsenault case 244: 4384bd72361SMatt Arsenault return 0x4000; 4394bd72361SMatt Arsenault case 245: 4404bd72361SMatt Arsenault return 0xC000; 4414bd72361SMatt Arsenault case 246: 4424bd72361SMatt Arsenault return 0x4400; 4434bd72361SMatt Arsenault case 247: 4444bd72361SMatt Arsenault return 0xC400; 4454bd72361SMatt Arsenault case 248: // 1 / (2 * PI) 4464bd72361SMatt Arsenault return 0x3118; 4474bd72361SMatt Arsenault default: 4484bd72361SMatt Arsenault llvm_unreachable("invalid fp inline imm"); 4494bd72361SMatt Arsenault } 4504bd72361SMatt Arsenault } 4514bd72361SMatt Arsenault 4524bd72361SMatt Arsenault MCOperand AMDGPUDisassembler::decodeFPImmed(OpWidthTy Width, unsigned Imm) { 453212a251cSArtem Tamazov assert(Imm >= AMDGPU::EncValues::INLINE_FLOATING_C_MIN 454212a251cSArtem Tamazov && Imm <= AMDGPU::EncValues::INLINE_FLOATING_C_MAX); 4554bd72361SMatt Arsenault 456e1818af8STom Stellard // ToDo: case 248: 1/(2*PI) - is allowed only on VI 4574bd72361SMatt Arsenault switch (Width) { 4584bd72361SMatt Arsenault case OPW32: 4594bd72361SMatt Arsenault return MCOperand::createImm(getInlineImmVal32(Imm)); 4604bd72361SMatt Arsenault case OPW64: 4614bd72361SMatt Arsenault return MCOperand::createImm(getInlineImmVal64(Imm)); 4624bd72361SMatt Arsenault case OPW16: 4639be7b0d4SMatt Arsenault case OPWV216: 4644bd72361SMatt Arsenault return MCOperand::createImm(getInlineImmVal16(Imm)); 4654bd72361SMatt Arsenault default: 4664bd72361SMatt Arsenault llvm_unreachable("implement me"); 467e1818af8STom Stellard } 468e1818af8STom Stellard } 469e1818af8STom Stellard 470212a251cSArtem Tamazov unsigned AMDGPUDisassembler::getVgprClassId(const OpWidthTy Width) const { 471e1818af8STom Stellard using namespace AMDGPU; 472212a251cSArtem Tamazov assert(OPW_FIRST_ <= Width && Width < OPW_LAST_); 473212a251cSArtem Tamazov switch (Width) { 474212a251cSArtem Tamazov default: // fall 4754bd72361SMatt Arsenault case OPW32: 4764bd72361SMatt Arsenault case OPW16: 4779be7b0d4SMatt Arsenault case OPWV216: 4784bd72361SMatt Arsenault return VGPR_32RegClassID; 479212a251cSArtem Tamazov case OPW64: return VReg_64RegClassID; 480212a251cSArtem Tamazov case OPW128: return VReg_128RegClassID; 481212a251cSArtem Tamazov } 482212a251cSArtem Tamazov } 483212a251cSArtem Tamazov 484212a251cSArtem Tamazov unsigned AMDGPUDisassembler::getSgprClassId(const OpWidthTy Width) const { 485212a251cSArtem Tamazov using namespace AMDGPU; 486212a251cSArtem Tamazov assert(OPW_FIRST_ <= Width && Width < OPW_LAST_); 487212a251cSArtem Tamazov switch (Width) { 488212a251cSArtem Tamazov default: // fall 4894bd72361SMatt Arsenault case OPW32: 4904bd72361SMatt Arsenault case OPW16: 4919be7b0d4SMatt Arsenault case OPWV216: 4924bd72361SMatt Arsenault return SGPR_32RegClassID; 493212a251cSArtem Tamazov case OPW64: return SGPR_64RegClassID; 494212a251cSArtem Tamazov case OPW128: return SGPR_128RegClassID; 495212a251cSArtem Tamazov } 496212a251cSArtem Tamazov } 497212a251cSArtem Tamazov 498212a251cSArtem Tamazov unsigned AMDGPUDisassembler::getTtmpClassId(const OpWidthTy Width) const { 499212a251cSArtem Tamazov using namespace AMDGPU; 500212a251cSArtem Tamazov assert(OPW_FIRST_ <= Width && Width < OPW_LAST_); 501212a251cSArtem Tamazov switch (Width) { 502212a251cSArtem Tamazov default: // fall 5034bd72361SMatt Arsenault case OPW32: 5044bd72361SMatt Arsenault case OPW16: 5059be7b0d4SMatt Arsenault case OPWV216: 5064bd72361SMatt Arsenault return TTMP_32RegClassID; 507212a251cSArtem Tamazov case OPW64: return TTMP_64RegClassID; 508212a251cSArtem Tamazov case OPW128: return TTMP_128RegClassID; 509212a251cSArtem Tamazov } 510212a251cSArtem Tamazov } 511212a251cSArtem Tamazov 512212a251cSArtem Tamazov MCOperand AMDGPUDisassembler::decodeSrcOp(const OpWidthTy Width, unsigned Val) const { 513212a251cSArtem Tamazov using namespace AMDGPU::EncValues; 514ac106addSNikolay Haustov assert(Val < 512); // enum9 515ac106addSNikolay Haustov 516212a251cSArtem Tamazov if (VGPR_MIN <= Val && Val <= VGPR_MAX) { 517212a251cSArtem Tamazov return createRegOperand(getVgprClassId(Width), Val - VGPR_MIN); 518212a251cSArtem Tamazov } 519b49c3361SArtem Tamazov if (Val <= SGPR_MAX) { 520b49c3361SArtem Tamazov assert(SGPR_MIN == 0); // "SGPR_MIN <= Val" is always true and causes compilation warning. 521212a251cSArtem Tamazov return createSRegOperand(getSgprClassId(Width), Val - SGPR_MIN); 522212a251cSArtem Tamazov } 523212a251cSArtem Tamazov if (TTMP_MIN <= Val && Val <= TTMP_MAX) { 524212a251cSArtem Tamazov return createSRegOperand(getTtmpClassId(Width), Val - TTMP_MIN); 525212a251cSArtem Tamazov } 526ac106addSNikolay Haustov 5274bd72361SMatt Arsenault assert(Width == OPW16 || Width == OPW32 || Width == OPW64); 528212a251cSArtem Tamazov 529212a251cSArtem Tamazov if (INLINE_INTEGER_C_MIN <= Val && Val <= INLINE_INTEGER_C_MAX) 530ac106addSNikolay Haustov return decodeIntImmed(Val); 531ac106addSNikolay Haustov 532212a251cSArtem Tamazov if (INLINE_FLOATING_C_MIN <= Val && Val <= INLINE_FLOATING_C_MAX) 5334bd72361SMatt Arsenault return decodeFPImmed(Width, Val); 534ac106addSNikolay Haustov 535212a251cSArtem Tamazov if (Val == LITERAL_CONST) 536ac106addSNikolay Haustov return decodeLiteralConstant(); 537ac106addSNikolay Haustov 5384bd72361SMatt Arsenault switch (Width) { 5394bd72361SMatt Arsenault case OPW32: 5404bd72361SMatt Arsenault case OPW16: 5419be7b0d4SMatt Arsenault case OPWV216: 5424bd72361SMatt Arsenault return decodeSpecialReg32(Val); 5434bd72361SMatt Arsenault case OPW64: 5444bd72361SMatt Arsenault return decodeSpecialReg64(Val); 5454bd72361SMatt Arsenault default: 5464bd72361SMatt Arsenault llvm_unreachable("unexpected immediate type"); 5474bd72361SMatt Arsenault } 548ac106addSNikolay Haustov } 549ac106addSNikolay Haustov 550ac106addSNikolay Haustov MCOperand AMDGPUDisassembler::decodeSpecialReg32(unsigned Val) const { 551ac106addSNikolay Haustov using namespace AMDGPU; 552e1818af8STom Stellard switch (Val) { 553ac106addSNikolay Haustov case 102: return createRegOperand(getMCReg(FLAT_SCR_LO, STI)); 554ac106addSNikolay Haustov case 103: return createRegOperand(getMCReg(FLAT_SCR_HI, STI)); 555e1818af8STom Stellard // ToDo: no support for xnack_mask_lo/_hi register 556e1818af8STom Stellard case 104: 557ac106addSNikolay Haustov case 105: break; 558ac106addSNikolay Haustov case 106: return createRegOperand(VCC_LO); 559ac106addSNikolay Haustov case 107: return createRegOperand(VCC_HI); 560212a251cSArtem Tamazov case 108: return createRegOperand(TBA_LO); 561212a251cSArtem Tamazov case 109: return createRegOperand(TBA_HI); 562212a251cSArtem Tamazov case 110: return createRegOperand(TMA_LO); 563212a251cSArtem Tamazov case 111: return createRegOperand(TMA_HI); 564ac106addSNikolay Haustov case 124: return createRegOperand(M0); 565ac106addSNikolay Haustov case 126: return createRegOperand(EXEC_LO); 566ac106addSNikolay Haustov case 127: return createRegOperand(EXEC_HI); 567a3b3b489SMatt Arsenault case 235: return createRegOperand(SRC_SHARED_BASE); 568a3b3b489SMatt Arsenault case 236: return createRegOperand(SRC_SHARED_LIMIT); 569a3b3b489SMatt Arsenault case 237: return createRegOperand(SRC_PRIVATE_BASE); 570a3b3b489SMatt Arsenault case 238: return createRegOperand(SRC_PRIVATE_LIMIT); 571a3b3b489SMatt Arsenault // TODO: SRC_POPS_EXITING_WAVE_ID 572e1818af8STom Stellard // ToDo: no support for vccz register 573ac106addSNikolay Haustov case 251: break; 574e1818af8STom Stellard // ToDo: no support for execz register 575ac106addSNikolay Haustov case 252: break; 576ac106addSNikolay Haustov case 253: return createRegOperand(SCC); 577ac106addSNikolay Haustov default: break; 578e1818af8STom Stellard } 579ac106addSNikolay Haustov return errOperand(Val, "unknown operand encoding " + Twine(Val)); 580e1818af8STom Stellard } 581e1818af8STom Stellard 582ac106addSNikolay Haustov MCOperand AMDGPUDisassembler::decodeSpecialReg64(unsigned Val) const { 583161a158eSNikolay Haustov using namespace AMDGPU; 584161a158eSNikolay Haustov switch (Val) { 585ac106addSNikolay Haustov case 102: return createRegOperand(getMCReg(FLAT_SCR, STI)); 586ac106addSNikolay Haustov case 106: return createRegOperand(VCC); 587212a251cSArtem Tamazov case 108: return createRegOperand(TBA); 588212a251cSArtem Tamazov case 110: return createRegOperand(TMA); 589ac106addSNikolay Haustov case 126: return createRegOperand(EXEC); 590ac106addSNikolay Haustov default: break; 591161a158eSNikolay Haustov } 592ac106addSNikolay Haustov return errOperand(Val, "unknown operand encoding " + Twine(Val)); 593161a158eSNikolay Haustov } 594161a158eSNikolay Haustov 595363f47a2SSam Kolton MCOperand AMDGPUDisassembler::decodeSDWA9Src(const OpWidthTy Width, 596363f47a2SSam Kolton unsigned Val) const { 597363f47a2SSam Kolton using namespace AMDGPU::SDWA; 598363f47a2SSam Kolton 599363f47a2SSam Kolton if (SDWA9EncValues::SRC_VGPR_MIN <= Val && 600363f47a2SSam Kolton Val <= SDWA9EncValues::SRC_VGPR_MAX) { 601363f47a2SSam Kolton return createRegOperand(getVgprClassId(Width), 602363f47a2SSam Kolton Val - SDWA9EncValues::SRC_VGPR_MIN); 603363f47a2SSam Kolton } 604363f47a2SSam Kolton if (SDWA9EncValues::SRC_SGPR_MIN <= Val && 605363f47a2SSam Kolton Val <= SDWA9EncValues::SRC_SGPR_MAX) { 606363f47a2SSam Kolton return createSRegOperand(getSgprClassId(Width), 607363f47a2SSam Kolton Val - SDWA9EncValues::SRC_SGPR_MIN); 608363f47a2SSam Kolton } 609363f47a2SSam Kolton 610363f47a2SSam Kolton return decodeSpecialReg32(Val - SDWA9EncValues::SRC_SGPR_MIN); 611363f47a2SSam Kolton } 612363f47a2SSam Kolton 613363f47a2SSam Kolton MCOperand AMDGPUDisassembler::decodeSDWA9Src16(unsigned Val) const { 614363f47a2SSam Kolton return decodeSDWA9Src(OPW16, Val); 615363f47a2SSam Kolton } 616363f47a2SSam Kolton 617363f47a2SSam Kolton MCOperand AMDGPUDisassembler::decodeSDWA9Src32(unsigned Val) const { 618363f47a2SSam Kolton return decodeSDWA9Src(OPW32, Val); 619363f47a2SSam Kolton } 620363f47a2SSam Kolton 621363f47a2SSam Kolton 622363f47a2SSam Kolton MCOperand AMDGPUDisassembler::decodeSDWA9VopcDst(unsigned Val) const { 623363f47a2SSam Kolton using namespace AMDGPU::SDWA; 624363f47a2SSam Kolton 625363f47a2SSam Kolton if (Val & SDWA9EncValues::VOPC_DST_VCC_MASK) { 626363f47a2SSam Kolton Val &= SDWA9EncValues::VOPC_DST_SGPR_MASK; 627363f47a2SSam Kolton if (Val > AMDGPU::EncValues::SGPR_MAX) { 628363f47a2SSam Kolton return decodeSpecialReg64(Val); 629363f47a2SSam Kolton } else { 630363f47a2SSam Kolton return createSRegOperand(getSgprClassId(OPW64), Val); 631363f47a2SSam Kolton } 632363f47a2SSam Kolton } else { 633363f47a2SSam Kolton return createRegOperand(AMDGPU::VCC); 634363f47a2SSam Kolton } 635363f47a2SSam Kolton } 636363f47a2SSam Kolton 6373381d7a2SSam Kolton //===----------------------------------------------------------------------===// 6383381d7a2SSam Kolton // AMDGPUSymbolizer 6393381d7a2SSam Kolton //===----------------------------------------------------------------------===// 6403381d7a2SSam Kolton 6413381d7a2SSam Kolton // Try to find symbol name for specified label 6423381d7a2SSam Kolton bool AMDGPUSymbolizer::tryAddingSymbolicOperand(MCInst &Inst, 6433381d7a2SSam Kolton raw_ostream &/*cStream*/, int64_t Value, 6443381d7a2SSam Kolton uint64_t /*Address*/, bool IsBranch, 6453381d7a2SSam Kolton uint64_t /*Offset*/, uint64_t /*InstSize*/) { 6463381d7a2SSam Kolton typedef std::tuple<uint64_t, StringRef, uint8_t> SymbolInfoTy; 6473381d7a2SSam Kolton typedef std::vector<SymbolInfoTy> SectionSymbolsTy; 6483381d7a2SSam Kolton 6493381d7a2SSam Kolton if (!IsBranch) { 6503381d7a2SSam Kolton return false; 6513381d7a2SSam Kolton } 6523381d7a2SSam Kolton 6533381d7a2SSam Kolton auto *Symbols = static_cast<SectionSymbolsTy *>(DisInfo); 6543381d7a2SSam Kolton auto Result = std::find_if(Symbols->begin(), Symbols->end(), 6553381d7a2SSam Kolton [Value](const SymbolInfoTy& Val) { 6563381d7a2SSam Kolton return std::get<0>(Val) == static_cast<uint64_t>(Value) 6573381d7a2SSam Kolton && std::get<2>(Val) == ELF::STT_NOTYPE; 6583381d7a2SSam Kolton }); 6593381d7a2SSam Kolton if (Result != Symbols->end()) { 6603381d7a2SSam Kolton auto *Sym = Ctx.getOrCreateSymbol(std::get<1>(*Result)); 6613381d7a2SSam Kolton const auto *Add = MCSymbolRefExpr::create(Sym, Ctx); 6623381d7a2SSam Kolton Inst.addOperand(MCOperand::createExpr(Add)); 6633381d7a2SSam Kolton return true; 6643381d7a2SSam Kolton } 6653381d7a2SSam Kolton return false; 6663381d7a2SSam Kolton } 6673381d7a2SSam Kolton 66892b355b1SMatt Arsenault void AMDGPUSymbolizer::tryAddingPcLoadReferenceComment(raw_ostream &cStream, 66992b355b1SMatt Arsenault int64_t Value, 67092b355b1SMatt Arsenault uint64_t Address) { 67192b355b1SMatt Arsenault llvm_unreachable("unimplemented"); 67292b355b1SMatt Arsenault } 67392b355b1SMatt Arsenault 6743381d7a2SSam Kolton //===----------------------------------------------------------------------===// 6753381d7a2SSam Kolton // Initialization 6763381d7a2SSam Kolton //===----------------------------------------------------------------------===// 6773381d7a2SSam Kolton 6783381d7a2SSam Kolton static MCSymbolizer *createAMDGPUSymbolizer(const Triple &/*TT*/, 6793381d7a2SSam Kolton LLVMOpInfoCallback /*GetOpInfo*/, 6803381d7a2SSam Kolton LLVMSymbolLookupCallback /*SymbolLookUp*/, 6813381d7a2SSam Kolton void *DisInfo, 6823381d7a2SSam Kolton MCContext *Ctx, 6833381d7a2SSam Kolton std::unique_ptr<MCRelocationInfo> &&RelInfo) { 6843381d7a2SSam Kolton return new AMDGPUSymbolizer(*Ctx, std::move(RelInfo), DisInfo); 6853381d7a2SSam Kolton } 6863381d7a2SSam Kolton 687e1818af8STom Stellard static MCDisassembler *createAMDGPUDisassembler(const Target &T, 688e1818af8STom Stellard const MCSubtargetInfo &STI, 689e1818af8STom Stellard MCContext &Ctx) { 690e1818af8STom Stellard return new AMDGPUDisassembler(STI, Ctx); 691e1818af8STom Stellard } 692e1818af8STom Stellard 693e1818af8STom Stellard extern "C" void LLVMInitializeAMDGPUDisassembler() { 694f42454b9SMehdi Amini TargetRegistry::RegisterMCDisassembler(getTheGCNTarget(), 695f42454b9SMehdi Amini createAMDGPUDisassembler); 696f42454b9SMehdi Amini TargetRegistry::RegisterMCSymbolizer(getTheGCNTarget(), 697f42454b9SMehdi Amini createAMDGPUSymbolizer); 698e1818af8STom Stellard } 699