forked from OSchip/llvm-project
68 lines
1.8 KiB
C++
68 lines
1.8 KiB
C++
//===-- AMDGPUAsmPrinter.h - Print AMDGPU assembly code ---------*- C++ -*-===//
|
|
//
|
|
// The LLVM Compiler Infrastructure
|
|
//
|
|
// This file is distributed under the University of Illinois Open Source
|
|
// License. See LICENSE.TXT for details.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
//
|
|
/// \file
|
|
/// \brief AMDGPU Assembly printer class.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#ifndef AMDGPU_ASMPRINTER_H
|
|
#define AMDGPU_ASMPRINTER_H
|
|
|
|
#include "llvm/CodeGen/AsmPrinter.h"
|
|
#include <string>
|
|
#include <vector>
|
|
|
|
namespace llvm {
|
|
|
|
class AMDGPUAsmPrinter : public AsmPrinter {
|
|
private:
|
|
struct SIProgramInfo {
|
|
SIProgramInfo() :
|
|
CodeLen(0),
|
|
NumSGPR(0),
|
|
NumVGPR(0) {}
|
|
|
|
uint64_t CodeLen;
|
|
unsigned NumSGPR;
|
|
unsigned NumVGPR;
|
|
};
|
|
|
|
void getSIProgramInfo(SIProgramInfo &Out, MachineFunction &MF) const;
|
|
void findNumUsedRegistersSI(MachineFunction &MF,
|
|
unsigned &NumSGPR,
|
|
unsigned &NumVGPR) const;
|
|
|
|
/// \brief Emit register usage information so that the GPU driver
|
|
/// can correctly setup the GPU state.
|
|
void EmitProgramInfoR600(MachineFunction &MF);
|
|
void EmitProgramInfoSI(MachineFunction &MF, const SIProgramInfo &KernelInfo);
|
|
|
|
public:
|
|
explicit AMDGPUAsmPrinter(TargetMachine &TM, MCStreamer &Streamer);
|
|
|
|
bool runOnMachineFunction(MachineFunction &MF) override;
|
|
|
|
const char *getPassName() const override {
|
|
return "AMDGPU Assembly Printer";
|
|
}
|
|
|
|
/// Implemented in AMDGPUMCInstLower.cpp
|
|
void EmitInstruction(const MachineInstr *MI) override;
|
|
|
|
protected:
|
|
bool DisasmEnabled;
|
|
std::vector<std::string> DisasmLines, HexLines;
|
|
size_t DisasmLineMaxLen;
|
|
};
|
|
|
|
} // End anonymous llvm
|
|
|
|
#endif //AMDGPU_ASMPRINTER_H
|