1 //===-- SPIRVInstPrinter.h - Output SPIR-V MCInsts as ASM -------*- C++ -*-===// 2 // 3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. 4 // See https://llvm.org/LICENSE.txt for license information. 5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception 6 // 7 //===----------------------------------------------------------------------===// 8 // 9 // This class prints a SPIR-V MCInst to a .s file. 10 // 11 //===----------------------------------------------------------------------===// 12 13 #ifndef LLVM_LIB_TARGET_SPIRV_INSTPRINTER_SPIRVINSTPRINTER_H 14 #define LLVM_LIB_TARGET_SPIRV_INSTPRINTER_SPIRVINSTPRINTER_H 15 16 #include "MCTargetDesc/SPIRVBaseInfo.h" 17 #include "llvm/ADT/DenseSet.h" 18 #include "llvm/MC/MCInstPrinter.h" 19 20 namespace llvm { 21 class SPIRVInstPrinter : public MCInstPrinter { 22 private: 23 SmallDenseMap<unsigned, SPIRV::InstructionSet::InstructionSet> ExtInstSetIDs; 24 void recordOpExtInstImport(const MCInst *MI); 25 26 public: 27 using MCInstPrinter::MCInstPrinter; 28 29 void printInst(const MCInst *MI, uint64_t Address, StringRef Annot, 30 const MCSubtargetInfo &STI, raw_ostream &OS) override; 31 void printOperand(const MCInst *MI, unsigned OpNo, raw_ostream &O, 32 const char *Modifier = nullptr); 33 34 void printStringImm(const MCInst *MI, unsigned OpNo, raw_ostream &O); 35 36 void printOpDecorate(const MCInst *MI, raw_ostream &O); 37 void printOpExtInst(const MCInst *MI, raw_ostream &O); 38 void printRemainingVariableOps(const MCInst *MI, unsigned StartIndex, 39 raw_ostream &O, bool SkipFirstSpace = false, 40 bool SkipImmediates = false); 41 void printOpConstantVarOps(const MCInst *MI, unsigned StartIndex, 42 raw_ostream &O); 43 44 void printExtension(const MCInst *MI, unsigned OpNo, raw_ostream &O); 45 template <SPIRV::OperandCategory::OperandCategory category> 46 void printSymbolicOperand(const MCInst *MI, unsigned OpNo, raw_ostream &O); 47 48 // Autogenerated by tblgen. 49 std::pair<const char *, uint64_t> getMnemonic(const MCInst *MI) override; 50 void printInstruction(const MCInst *MI, uint64_t Address, raw_ostream &O); 51 static const char *getRegisterName(MCRegister Reg); 52 }; 53 } // namespace llvm 54 55 #endif // LLVM_LIB_TARGET_SPIRV_INSTPRINTER_SPIRVINSTPRINTER_H 56