AMDGPUMachineFunction.cpp revision 327952
1//===-- AMDGPUMachineFunctionInfo.cpp ---------------------------------------=// 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#include "AMDGPUMachineFunction.h" 11#include "AMDGPUSubtarget.h" 12 13using namespace llvm; 14 15AMDGPUMachineFunction::AMDGPUMachineFunction(const MachineFunction &MF) : 16 MachineFunctionInfo(), 17 LocalMemoryObjects(), 18 KernArgSize(0), 19 MaxKernArgAlign(0), 20 LDSSize(0), 21 ABIArgOffset(0), 22 IsEntryFunction(AMDGPU::isEntryFunctionCC(MF.getFunction().getCallingConv())), 23 NoSignedZerosFPMath(MF.getTarget().Options.NoSignedZerosFPMath) { 24 // FIXME: Should initialize KernArgSize based on ExplicitKernelArgOffset, 25 // except reserved size is not correctly aligned. 26} 27 28unsigned AMDGPUMachineFunction::allocateLDSGlobal(const DataLayout &DL, 29 const GlobalValue &GV) { 30 auto Entry = LocalMemoryObjects.insert(std::make_pair(&GV, 0)); 31 if (!Entry.second) 32 return Entry.first->second; 33 34 unsigned Align = GV.getAlignment(); 35 if (Align == 0) 36 Align = DL.getABITypeAlignment(GV.getValueType()); 37 38 /// TODO: We should sort these to minimize wasted space due to alignment 39 /// padding. Currently the padding is decided by the first encountered use 40 /// during lowering. 41 unsigned Offset = LDSSize = alignTo(LDSSize, Align); 42 43 Entry.first->second = Offset; 44 LDSSize += DL.getTypeAllocSize(GV.getValueType()); 45 46 return Offset; 47} 48