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   virtual void anchor();
  23 public:
  24   AMDGPUMachineFunction(const MachineFunction &MF);
  25   unsigned ShaderType;
  26   /// A map to keep track of local memory objects and their offsets within
  27   /// the local memory space.
  28   std::map<const GlobalValue *, unsigned> LocalMemoryObjects;
  29   /// Number of bytes in the LDS that are being used.
  30   unsigned LDSSize;
  31 };
  32
  33 }
  34 #endif // AMDGPUMACHINEFUNCTION_H