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 49 unsigned ScratchSize; 50 bool IsKernel; 51 }; 52 53 } 54 #endif 55
