1 //===-- AMDGPUMachineFunctionInfo.h -------------------------------*- 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 #ifndef LLVM_LIB_TARGET_AMDGPU_AMDGPUMACHINEFUNCTION_H
11 #define LLVM_LIB_TARGET_AMDGPU_AMDGPUMACHINEFUNCTION_H
12 
13 #include "llvm/CodeGen/MachineFunction.h"
14 #include <map>
15 
16 namespace llvm {
17 
18 class AMDGPUMachineFunction : public MachineFunctionInfo {
19   uint64_t KernArgSize;
20   unsigned MaxKernArgAlign;
21 
22   virtual void anchor();
23 
24 public:
25   AMDGPUMachineFunction(const MachineFunction &MF);
26 
27   uint64_t allocateKernArg(uint64_t Size, unsigned Align) {
28     assert(isPowerOf2_32(Align));
29     KernArgSize = alignTo(KernArgSize, Align);
30 
31     uint64_t Result = KernArgSize;
32     KernArgSize += Size;
33 
34     MaxKernArgAlign = std::max(Align, MaxKernArgAlign);
35     return Result;
36   }
37 
38   /// A map to keep track of local memory objects and their offsets within
39   /// the local memory space.
40   std::map<const GlobalValue *, unsigned> LocalMemoryObjects;
41   /// Number of bytes in the LDS that are being used.
42   unsigned LDSSize;
43 
44   /// Start of implicit kernel args
45   unsigned ABIArgOffset;
46 
47   bool isKernel() const {
48     // FIXME: Assume everything is a kernel until function calls are supported.
49     return true;
50   }
51 
52   unsigned ScratchSize;
53   bool IsKernel;
54 };
55 
56 }
57 #endif
58