llvm-project/llvm/lib/Target/R600/AMDGPUMachineFunction.h

43 lines
1.1 KiB
C++

//===-- R600MachineFunctionInfo.h - R600 Machine Function Info ----*- C++ -*-=//
//
// The LLVM Compiler Infrastructure
//
// This file is distributed under the University of Illinois Open Source
// License. See LICENSE.TXT for details.
//
//===----------------------------------------------------------------------===//
//
/// \file
//===----------------------------------------------------------------------===//
#ifndef LLVM_LIB_TARGET_R600_AMDGPUMACHINEFUNCTION_H
#define LLVM_LIB_TARGET_R600_AMDGPUMACHINEFUNCTION_H
#include "llvm/CodeGen/MachineFunction.h"
#include <map>
namespace llvm {
class AMDGPUMachineFunction : public MachineFunctionInfo {
virtual void anchor();
unsigned ShaderType;
public:
AMDGPUMachineFunction(const MachineFunction &MF);
/// A map to keep track of local memory objects and their offsets within
/// the local memory space.
std::map<const GlobalValue *, unsigned> LocalMemoryObjects;
/// Number of bytes in the LDS that are being used.
unsigned LDSSize;
unsigned getShaderType() const {
return ShaderType;
}
unsigned ScratchSize;
bool IsKernel;
};
}
#endif