1d88c1a5aSDimitry Andric //===-- AMDGPUMachineFunctionInfo.cpp ---------------------------------------=// 2d88c1a5aSDimitry Andric // 3d88c1a5aSDimitry Andric // The LLVM Compiler Infrastructure 4d88c1a5aSDimitry Andric // 5d88c1a5aSDimitry Andric // This file is distributed under the University of Illinois Open Source 6d88c1a5aSDimitry Andric // License. See LICENSE.TXT for details. 7d88c1a5aSDimitry Andric // 8d88c1a5aSDimitry Andric //===----------------------------------------------------------------------===// 9d88c1a5aSDimitry Andric 108f0fd8f6SDimitry Andric #include "AMDGPUMachineFunction.h" 11d88c1a5aSDimitry Andric #include "AMDGPUSubtarget.h" 123ca95b02SDimitry Andric 138f0fd8f6SDimitry Andric using namespace llvm; 148f0fd8f6SDimitry Andric 157a7e6055SDimitry Andric static bool isEntryFunctionCC(CallingConv::ID CC) { 167a7e6055SDimitry Andric switch (CC) { 177a7e6055SDimitry Andric case CallingConv::AMDGPU_KERNEL: 187a7e6055SDimitry Andric case CallingConv::SPIR_KERNEL: 197a7e6055SDimitry Andric case CallingConv::AMDGPU_VS: 207a7e6055SDimitry Andric case CallingConv::AMDGPU_GS: 217a7e6055SDimitry Andric case CallingConv::AMDGPU_PS: 227a7e6055SDimitry Andric case CallingConv::AMDGPU_CS: 237a7e6055SDimitry Andric return true; 247a7e6055SDimitry Andric default: 257a7e6055SDimitry Andric return false; 267a7e6055SDimitry Andric } 277a7e6055SDimitry Andric } 287a7e6055SDimitry Andric 298f0fd8f6SDimitry Andric AMDGPUMachineFunction::AMDGPUMachineFunction(const MachineFunction &MF) : 308f0fd8f6SDimitry Andric MachineFunctionInfo(), 31d88c1a5aSDimitry Andric LocalMemoryObjects(), 323ca95b02SDimitry Andric KernArgSize(0), 333ca95b02SDimitry Andric MaxKernArgAlign(0), 348f0fd8f6SDimitry Andric LDSSize(0), 357d523365SDimitry Andric ABIArgOffset(0), 367a7e6055SDimitry Andric IsEntryFunction(isEntryFunctionCC(MF.getFunction()->getCallingConv())), 377a7e6055SDimitry Andric NoSignedZerosFPMath(MF.getTarget().Options.NoSignedZerosFPMath) { 38d88c1a5aSDimitry Andric // FIXME: Should initialize KernArgSize based on ExplicitKernelArgOffset, 39d88c1a5aSDimitry Andric // except reserved size is not correctly aligned. 403ca95b02SDimitry Andric } 418f0fd8f6SDimitry Andric 42d88c1a5aSDimitry Andric unsigned AMDGPUMachineFunction::allocateLDSGlobal(const DataLayout &DL, 43d88c1a5aSDimitry Andric const GlobalValue &GV) { 44d88c1a5aSDimitry Andric auto Entry = LocalMemoryObjects.insert(std::make_pair(&GV, 0)); 45d88c1a5aSDimitry Andric if (!Entry.second) 46d88c1a5aSDimitry Andric return Entry.first->second; 47d88c1a5aSDimitry Andric 48d88c1a5aSDimitry Andric unsigned Align = GV.getAlignment(); 49d88c1a5aSDimitry Andric if (Align == 0) 50d88c1a5aSDimitry Andric Align = DL.getABITypeAlignment(GV.getValueType()); 51d88c1a5aSDimitry Andric 52d88c1a5aSDimitry Andric /// TODO: We should sort these to minimize wasted space due to alignment 53d88c1a5aSDimitry Andric /// padding. Currently the padding is decided by the first encountered use 54d88c1a5aSDimitry Andric /// during lowering. 55d88c1a5aSDimitry Andric unsigned Offset = LDSSize = alignTo(LDSSize, Align); 56d88c1a5aSDimitry Andric 57d88c1a5aSDimitry Andric Entry.first->second = Offset; 58d88c1a5aSDimitry Andric LDSSize += DL.getTypeAllocSize(GV.getValueType()); 59d88c1a5aSDimitry Andric 60d88c1a5aSDimitry Andric return Offset; 618f0fd8f6SDimitry Andric } 62