lib/Target/R600/AMDGPUAsmPrinter.h

   1 //===-- AMDGPUAsmPrinter.h - Print AMDGPU assembly code ---------*- C++ -*-===//
   2 //
   3 //                     The LLVM Compiler Infrastructure
   4 //
   5 // This file is distributed under the University of Illinois Open Source
   6 // License. See LICENSE.TXT for details.
   7 //
   8 //===----------------------------------------------------------------------===//
   9 //
  10 /// \file
  11 /// \brief AMDGPU Assembly printer class.
  12 //
  13 //===----------------------------------------------------------------------===//
  14
  15 #ifndef LLVM_LIB_TARGET_R600_AMDGPUASMPRINTER_H
  16 #define LLVM_LIB_TARGET_R600_AMDGPUASMPRINTER_H
  17
  18 #include "llvm/CodeGen/AsmPrinter.h"
  19 #include <vector>
  20
  21 namespace llvm {
  22
  23 class AMDGPUAsmPrinter : public AsmPrinter {
  24 private:
  25   struct SIProgramInfo {
  26     SIProgramInfo() :
  27       NumVGPR(0),
  28       NumSGPR(0),
  29       Priority(0),
  30       FloatMode(0),
  31       Priv(0),
  32       DX10Clamp(0),
  33       DebugMode(0),
  34       IEEEMode(0),
  35       ScratchSize(0),
  36       CodeLen(0) {}
  37
  38     // Fields set in PGM_RSRC1 pm4 packet.
  39     uint32_t NumVGPR;
  40     uint32_t NumSGPR;
  41     uint32_t Priority;
  42     uint32_t FloatMode;
  43     uint32_t Priv;
  44     uint32_t DX10Clamp;
  45     uint32_t DebugMode;
  46     uint32_t IEEEMode;
  47     uint32_t ScratchSize;
  48
  49     // Bonus information for debugging.
  50     uint64_t CodeLen;
  51   };
  52
  53   void getSIProgramInfo(SIProgramInfo &Out, const MachineFunction &MF) const;
  54   void findNumUsedRegistersSI(const MachineFunction &MF,
  55                               unsigned &NumSGPR,
  56                               unsigned &NumVGPR) const;
  57
  58   /// \brief Emit register usage information so that the GPU driver
  59   /// can correctly setup the GPU state.
  60   void EmitProgramInfoR600(const MachineFunction &MF);
  61   void EmitProgramInfoSI(const MachineFunction &MF, const SIProgramInfo &KernelInfo);
  62
  63 public:
  64   explicit AMDGPUAsmPrinter(TargetMachine &TM, MCStreamer &Streamer);
  65
  66   bool runOnMachineFunction(MachineFunction &MF) override;
  67
  68   const char *getPassName() const override {
  69     return "AMDGPU Assembly Printer";
  70   }
  71
  72   /// Implemented in AMDGPUMCInstLower.cpp
  73   void EmitInstruction(const MachineInstr *MI) override;
  74
  75   void EmitEndOfAsmFile(Module &M) override;
  76
  77 protected:
  78   bool DisasmEnabled;
  79   std::vector<std::string> DisasmLines, HexLines;
  80   size_t DisasmLineMaxLen;
  81 };
  82
  83 } // End anonymous llvm
  84
  85 #endif