lib/Target/R600/AMDGPUMachineFunction.h

   1 //===-- R600MachineFunctionInfo.h - R600 Machine Function Info ----*- 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 //===----------------------------------------------------------------------===//
  12
  13 #ifndef AMDGPUMACHINEFUNCTION_H
  14 #define AMDGPUMACHINEFUNCTION_H
  15
  16 #include "llvm/CodeGen/MachineFunction.h"
  17 #include <map>
  18
  19 namespace llvm {
  20
  21 class AMDGPUMachineFunction : public MachineFunctionInfo {
  22 public:
  23   AMDGPUMachineFunction(const MachineFunction &MF);
  24   unsigned ShaderType;
  25   /// A map to keep track of local memory objects and their offsets within
  26   /// the local memory space.
  27   std::map<const GlobalValue *, unsigned> LocalMemoryObjects;
  28   /// Number of bytes in the LDS that are being used.
  29   unsigned LDSSize;
  30 };
  31
  32 }
  33 #endif // AMDGPUMACHINEFUNCTION_H