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 AMDGPU_ASMPRINTER_H
16#define AMDGPU_ASMPRINTER_H
17
18#include "llvm/CodeGen/AsmPrinter.h"
19#include <string>
20#include <vector>
21
22namespace llvm {
23
24class AMDGPUAsmPrinter : public AsmPrinter {
25private:
26  struct SIProgramInfo {
27    SIProgramInfo() :
28      NumVGPR(0),
29      NumSGPR(0),
30      Priority(0),
31      FloatMode(0),
32      Priv(0),
33      DX10Clamp(0),
34      DebugMode(0),
35      IEEEMode(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
48    // Bonus information for debugging.
49    uint64_t CodeLen;
50  };
51
52  void getSIProgramInfo(SIProgramInfo &Out, MachineFunction &MF) const;
53  void findNumUsedRegistersSI(MachineFunction &MF,
54                              unsigned &NumSGPR,
55                              unsigned &NumVGPR) const;
56
57  /// \brief Emit register usage information so that the GPU driver
58  /// can correctly setup the GPU state.
59  void EmitProgramInfoR600(MachineFunction &MF);
60  void EmitProgramInfoSI(MachineFunction &MF, const SIProgramInfo &KernelInfo);
61
62public:
63  explicit AMDGPUAsmPrinter(TargetMachine &TM, MCStreamer &Streamer);
64
65  bool runOnMachineFunction(MachineFunction &MF) override;
66
67  const char *getPassName() const override {
68    return "AMDGPU Assembly Printer";
69  }
70
71  /// Implemented in AMDGPUMCInstLower.cpp
72  void EmitInstruction(const MachineInstr *MI) override;
73
74protected:
75  bool DisasmEnabled;
76  std::vector<std::string> DisasmLines, HexLines;
77  size_t DisasmLineMaxLen;
78};
79
80} // End anonymous llvm
81
82#endif //AMDGPU_ASMPRINTER_H
83