contrib/llvm/lib/Target/AMDGPU/AMDGPUMachineFunction.cpp

   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
  13 using namespace llvm;
  14
  15 AMDGPUMachineFunction::AMDGPUMachineFunction(const MachineFunction &MF) :
  16   MachineFunctionInfo(),
  17   LocalMemoryObjects(),
  18   KernArgSize(0),
  19   MaxKernArgAlign(0),
  20   LDSSize(0),
  21   ABIArgOffset(0),
  22   IsKernel(MF.getFunction()->getCallingConv() == CallingConv::AMDGPU_KERNEL ||
  23            MF.getFunction()->getCallingConv() == CallingConv::SPIR_KERNEL) {
  24   // FIXME: Should initialize KernArgSize based on ExplicitKernelArgOffset,
  25   // except reserved size is not correctly aligned.
  26 }
  27
  28 unsigned 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 }