AMDGPUAsmPrinter.h revision 36b56886974eae4f9c5ebc96befd3e7bfe5de338 
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() : NumSGPR(0), NumVGPR(0) {}
28    unsigned NumSGPR;
29    unsigned NumVGPR;
30  };
31
32  void getSIProgramInfo(SIProgramInfo &Out, MachineFunction &MF) const;
33  void findNumUsedRegistersSI(MachineFunction &MF,
34                              unsigned &NumSGPR,
35                              unsigned &NumVGPR) const;
36
37  /// \brief Emit register usage information so that the GPU driver
38  /// can correctly setup the GPU state.
39  void EmitProgramInfoR600(MachineFunction &MF);
40  void EmitProgramInfoSI(MachineFunction &MF, const SIProgramInfo &KernelInfo);
41
42public:
43  explicit AMDGPUAsmPrinter(TargetMachine &TM, MCStreamer &Streamer);
44
45  virtual bool runOnMachineFunction(MachineFunction &MF);
46
47  virtual const char *getPassName() const {
48    return "AMDGPU Assembly Printer";
49  }
50
51  /// Implemented in AMDGPUMCInstLower.cpp
52  virtual void EmitInstruction(const MachineInstr *MI);
53
54protected:
55  bool DisasmEnabled;
56  std::vector<std::string> DisasmLines, HexLines;
57  size_t DisasmLineMaxLen;
58};
59
60} // End anonymous llvm
61
62#endif //AMDGPU_ASMPRINTER_H
63