| //===- AMDGPUDisassembler.hpp - Disassembler for AMDGPU ISA -----*- C++ -*-===// | |
| // | |
| // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. | |
| // See https://llvm.org/LICENSE.txt for license information. | |
| // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception | |
| // | |
| //===----------------------------------------------------------------------===// | |
| // | |
| /// \file | |
| /// | |
| /// This file contains declaration for AMDGPU ISA disassembler | |
| // | |
| //===----------------------------------------------------------------------===// | |
| #ifndef LLVM_LIB_TARGET_AMDGPU_DISASSEMBLER_AMDGPUDISASSEMBLER_H | |
| #define LLVM_LIB_TARGET_AMDGPU_DISASSEMBLER_AMDGPUDISASSEMBLER_H | |
| #include "llvm/ADT/APInt.h" | |
| #include "llvm/MC/MCDisassembler/MCDisassembler.h" | |
| #include "llvm/MC/MCInstrInfo.h" | |
| #include "llvm/MC/MCInst.h" | |
| #include "llvm/Support/DataExtractor.h" | |
| #include <memory> | |
| namespace llvm { | |
| class MCInst; | |
| class MCOperand; | |
| class MCSubtargetInfo; | |
| class Twine; | |
| // Exposes an interface expected by autogenerated code in | |
| // FixedLenDecoderEmitter | |
| class DecoderUInt128 { | |
| private: | |
| uint64_t Lo = 0; | |
| uint64_t Hi = 0; | |
| public: | |
| DecoderUInt128() = default; | |
| DecoderUInt128(uint64_t Lo, uint64_t Hi = 0) : Lo(Lo), Hi(Hi) {} | |
| operator bool() const { return Lo || Hi; } | |
| void insertBits(uint64_t SubBits, unsigned BitPosition, unsigned NumBits) { | |
| assert(NumBits && NumBits <= 64); | |
| assert(SubBits >> 1 >> (NumBits - 1) == 0); | |
| assert(BitPosition < 128); | |
| if (BitPosition < 64) { | |
| Lo |= SubBits << BitPosition; | |
| Hi |= SubBits >> 1 >> (63 - BitPosition); | |
| } else { | |
| Hi |= SubBits << (BitPosition - 64); | |
| } | |
| } | |
| uint64_t extractBitsAsZExtValue(unsigned NumBits, | |
| unsigned BitPosition) const { | |
| assert(NumBits && NumBits <= 64); | |
| assert(BitPosition < 128); | |
| uint64_t Val; | |
| if (BitPosition < 64) | |
| Val = Lo >> BitPosition | Hi << 1 << (63 - BitPosition); | |
| else | |
| Val = Hi >> (BitPosition - 64); | |
| return Val & ((uint64_t(2) << (NumBits - 1)) - 1); | |
| } | |
| DecoderUInt128 operator&(const DecoderUInt128 &RHS) const { | |
| return DecoderUInt128(Lo & RHS.Lo, Hi & RHS.Hi); | |
| } | |
| DecoderUInt128 operator&(const uint64_t &RHS) const { | |
| return *this & DecoderUInt128(RHS); | |
| } | |
| DecoderUInt128 operator~() const { return DecoderUInt128(~Lo, ~Hi); } | |
| bool operator==(const DecoderUInt128 &RHS) { | |
| return Lo == RHS.Lo && Hi == RHS.Hi; | |
| } | |
| bool operator!=(const DecoderUInt128 &RHS) { | |
| return Lo != RHS.Lo || Hi != RHS.Hi; | |
| } | |
| bool operator!=(const int &RHS) { | |
| return *this != DecoderUInt128(RHS); | |
| } | |
| friend raw_ostream &operator<<(raw_ostream &OS, const DecoderUInt128 &RHS) { | |
| return OS << APInt(128, {RHS.Lo, RHS.Hi}); | |
| } | |
| }; | |
| //===----------------------------------------------------------------------===// | |
| // AMDGPUDisassembler | |
| //===----------------------------------------------------------------------===// | |
| class AMDGPUDisassembler : public MCDisassembler { | |
| private: | |
| std::unique_ptr<MCInstrInfo const> const MCII; | |
| const MCRegisterInfo &MRI; | |
| const unsigned TargetMaxInstBytes; | |
| mutable ArrayRef<uint8_t> Bytes; | |
| mutable uint32_t Literal; | |
| mutable bool HasLiteral; | |
| public: | |
| AMDGPUDisassembler(const MCSubtargetInfo &STI, MCContext &Ctx, | |
| MCInstrInfo const *MCII); | |
| ~AMDGPUDisassembler() override = default; | |
| DecodeStatus getInstruction(MCInst &MI, uint64_t &Size, | |
| ArrayRef<uint8_t> Bytes, uint64_t Address, | |
| raw_ostream &CS) const override; | |
| const char* getRegClassName(unsigned RegClassID) const; | |
| MCOperand createRegOperand(unsigned int RegId) const; | |
| MCOperand createRegOperand(unsigned RegClassID, unsigned Val) const; | |
| MCOperand createSRegOperand(unsigned SRegClassID, unsigned Val) const; | |
| MCOperand errOperand(unsigned V, const Twine& ErrMsg) const; | |
| template <typename InsnType> | |
| DecodeStatus tryDecodeInst(const uint8_t *Table, MCInst &MI, InsnType Inst, | |
| uint64_t Address) const { | |
| assert(MI.getOpcode() == 0); | |
| assert(MI.getNumOperands() == 0); | |
| MCInst TmpInst; | |
| HasLiteral = false; | |
| const auto SavedBytes = Bytes; | |
| if (decodeInstruction(Table, TmpInst, Inst, Address, this, STI)) { | |
| MI = TmpInst; | |
| return MCDisassembler::Success; | |
| } | |
| Bytes = SavedBytes; | |
| return MCDisassembler::Fail; | |
| } | |
| std::optional<DecodeStatus> | |
| onSymbolStart(SymbolInfoTy &Symbol, uint64_t &Size, ArrayRef<uint8_t> Bytes, | |
| uint64_t Address, raw_ostream &CStream) const override; | |
| DecodeStatus decodeKernelDescriptor(StringRef KdName, ArrayRef<uint8_t> Bytes, | |
| uint64_t KdAddress) const; | |
| DecodeStatus | |
| decodeKernelDescriptorDirective(DataExtractor::Cursor &Cursor, | |
| ArrayRef<uint8_t> Bytes, | |
| raw_string_ostream &KdStream) const; | |
| /// Decode as directives that handle COMPUTE_PGM_RSRC1. | |
| /// \param FourByteBuffer - Bytes holding contents of COMPUTE_PGM_RSRC1. | |
| /// \param KdStream - Stream to write the disassembled directives to. | |
| // NOLINTNEXTLINE(readability-identifier-naming) | |
| DecodeStatus decodeCOMPUTE_PGM_RSRC1(uint32_t FourByteBuffer, | |
| raw_string_ostream &KdStream) const; | |
| /// Decode as directives that handle COMPUTE_PGM_RSRC2. | |
| /// \param FourByteBuffer - Bytes holding contents of COMPUTE_PGM_RSRC2. | |
| /// \param KdStream - Stream to write the disassembled directives to. | |
| // NOLINTNEXTLINE(readability-identifier-naming) | |
| DecodeStatus decodeCOMPUTE_PGM_RSRC2(uint32_t FourByteBuffer, | |
| raw_string_ostream &KdStream) const; | |
| DecodeStatus convertEXPInst(MCInst &MI) const; | |
| DecodeStatus convertVINTERPInst(MCInst &MI) const; | |
| DecodeStatus convertFMAanyK(MCInst &MI, int ImmLitIdx) const; | |
| DecodeStatus convertSDWAInst(MCInst &MI) const; | |
| DecodeStatus convertDPP8Inst(MCInst &MI) const; | |
| DecodeStatus convertMIMGInst(MCInst &MI) const; | |
| DecodeStatus convertVOP3DPPInst(MCInst &MI) const; | |
| DecodeStatus convertVOP3PDPPInst(MCInst &MI) const; | |
| DecodeStatus convertVOPCDPPInst(MCInst &MI) const; | |
| void convertMacDPPInst(MCInst &MI) const; | |
| MCOperand decodeOperand_VGPR_32(unsigned Val) const; | |
| MCOperand decodeOperand_VGPR_32_Lo128(unsigned Val) const; | |
| MCOperand decodeOperand_VRegOrLds_32(unsigned Val) const; | |
| MCOperand decodeOperand_VS_32(unsigned Val) const; | |
| MCOperand decodeOperand_VS_64(unsigned Val) const; | |
| MCOperand decodeOperand_VS_128(unsigned Val) const; | |
| MCOperand decodeOperand_VSrc16(unsigned Val) const; | |
| MCOperand decodeOperand_VSrcV216(unsigned Val) const; | |
| MCOperand decodeOperand_VSrcV232(unsigned Val) const; | |
| MCOperand decodeOperand_VReg_64(unsigned Val) const; | |
| MCOperand decodeOperand_VReg_96(unsigned Val) const; | |
| MCOperand decodeOperand_VReg_128(unsigned Val) const; | |
| MCOperand decodeOperand_VReg_256(unsigned Val) const; | |
| MCOperand decodeOperand_VReg_288(unsigned Val) const; | |
| MCOperand decodeOperand_VReg_320(unsigned Val) const; | |
| MCOperand decodeOperand_VReg_352(unsigned Val) const; | |
| MCOperand decodeOperand_VReg_384(unsigned Val) const; | |
| MCOperand decodeOperand_VReg_512(unsigned Val) const; | |
| MCOperand decodeOperand_VReg_1024(unsigned Val) const; | |
| MCOperand decodeOperand_SReg_32(unsigned Val) const; | |
| MCOperand decodeOperand_SReg_32_XM0_XEXEC(unsigned Val) const; | |
| MCOperand decodeOperand_SReg_32_XEXEC_HI(unsigned Val) const; | |
| MCOperand decodeOperand_SRegOrLds_32(unsigned Val) const; | |
| MCOperand decodeOperand_SReg_64(unsigned Val) const; | |
| MCOperand decodeOperand_SReg_64_XEXEC(unsigned Val) const; | |
| MCOperand decodeOperand_SReg_128(unsigned Val) const; | |
| MCOperand decodeOperand_SReg_256(unsigned Val) const; | |
| MCOperand decodeOperand_SReg_288(unsigned Val) const; | |
| MCOperand decodeOperand_SReg_320(unsigned Val) const; | |
| MCOperand decodeOperand_SReg_352(unsigned Val) const; | |
| MCOperand decodeOperand_SReg_384(unsigned Val) const; | |
| MCOperand decodeOperand_SReg_512(unsigned Val) const; | |
| MCOperand decodeOperand_AGPR_32(unsigned Val) const; | |
| MCOperand decodeOperand_AReg_64(unsigned Val) const; | |
| MCOperand decodeOperand_AReg_128(unsigned Val) const; | |
| MCOperand decodeOperand_AReg_256(unsigned Val) const; | |
| MCOperand decodeOperand_AReg_288(unsigned Val) const; | |
| MCOperand decodeOperand_AReg_320(unsigned Val) const; | |
| MCOperand decodeOperand_AReg_352(unsigned Val) const; | |
| MCOperand decodeOperand_AReg_384(unsigned Val) const; | |
| MCOperand decodeOperand_AReg_512(unsigned Val) const; | |
| MCOperand decodeOperand_AReg_1024(unsigned Val) const; | |
| MCOperand decodeOperand_AV_32(unsigned Val) const; | |
| MCOperand decodeOperand_AV_64(unsigned Val) const; | |
| MCOperand decodeOperand_AV_128(unsigned Val) const; | |
| MCOperand decodeOperand_AVDst_128(unsigned Val) const; | |
| MCOperand decodeOperand_AVDst_512(unsigned Val) const; | |
| enum OpWidthTy { | |
| OPW32, | |
| OPW64, | |
| OPW96, | |
| OPW128, | |
| OPW160, | |
| OPW256, | |
| OPW288, | |
| OPW320, | |
| OPW352, | |
| OPW384, | |
| OPW512, | |
| OPW1024, | |
| OPW16, | |
| OPWV216, | |
| OPWV232, | |
| OPW_LAST_, | |
| OPW_FIRST_ = OPW32 | |
| }; | |
| unsigned getVgprClassId(const OpWidthTy Width) const; | |
| unsigned getAgprClassId(const OpWidthTy Width) const; | |
| unsigned getSgprClassId(const OpWidthTy Width) const; | |
| unsigned getTtmpClassId(const OpWidthTy Width) const; | |
| static MCOperand decodeIntImmed(unsigned Imm); | |
| static MCOperand decodeFPImmed(OpWidthTy Width, unsigned Imm); | |
| MCOperand decodeMandatoryLiteralConstant(unsigned Imm) const; | |
| MCOperand decodeLiteralConstant() const; | |
| MCOperand decodeSrcOp(const OpWidthTy Width, unsigned Val, | |
| bool MandatoryLiteral = false) const; | |
| MCOperand decodeDstOp(const OpWidthTy Width, unsigned Val) const; | |
| MCOperand decodeVOPDDstYOp(MCInst &Inst, unsigned Val) const; | |
| MCOperand decodeSpecialReg32(unsigned Val) const; | |
| MCOperand decodeSpecialReg64(unsigned Val) const; | |
| MCOperand decodeSDWASrc(const OpWidthTy Width, unsigned Val) const; | |
| MCOperand decodeSDWASrc16(unsigned Val) const; | |
| MCOperand decodeSDWASrc32(unsigned Val) const; | |
| MCOperand decodeSDWAVopcDst(unsigned Val) const; | |
| MCOperand decodeBoolReg(unsigned Val) const; | |
| int getTTmpIdx(unsigned Val) const; | |
| const MCInstrInfo *getMCII() const { return MCII.get(); } | |
| bool isVI() const; | |
| bool isGFX9() const; | |
| bool isGFX90A() const; | |
| bool isGFX9Plus() const; | |
| bool isGFX10() const; | |
| bool isGFX10Plus() const; | |
| bool isGFX11() const; | |
| bool isGFX11Plus() const; | |
| bool hasArchitectedFlatScratch() const; | |
| bool isMacDPP(MCInst &MI) const; | |
| }; | |
| //===----------------------------------------------------------------------===// | |
| // AMDGPUSymbolizer | |
| //===----------------------------------------------------------------------===// | |
| class AMDGPUSymbolizer : public MCSymbolizer { | |
| private: | |
| void *DisInfo; | |
| std::vector<uint64_t> ReferencedAddresses; | |
| public: | |
| AMDGPUSymbolizer(MCContext &Ctx, std::unique_ptr<MCRelocationInfo> &&RelInfo, | |
| void *disInfo) | |
| : MCSymbolizer(Ctx, std::move(RelInfo)), DisInfo(disInfo) {} | |
| bool tryAddingSymbolicOperand(MCInst &Inst, raw_ostream &cStream, | |
| int64_t Value, uint64_t Address, bool IsBranch, | |
| uint64_t Offset, uint64_t OpSize, | |
| uint64_t InstSize) override; | |
| void tryAddingPcLoadReferenceComment(raw_ostream &cStream, | |
| int64_t Value, | |
| uint64_t Address) override; | |
| ArrayRef<uint64_t> getReferencedAddresses() const override { | |
| return ReferencedAddresses; | |
| } | |
| }; | |
| } // end namespace llvm | |
| #endif // LLVM_LIB_TARGET_AMDGPU_DISASSEMBLER_AMDGPUDISASSEMBLER_H |