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 #include "AMDGPUPerfHintAnalysis.h"
  13 #include "llvm/CodeGen/MachineModuleInfo.h"
  14
  15 using namespace llvm;
  16
  17 AMDGPUMachineFunction::AMDGPUMachineFunction(const MachineFunction &MF) :
  18   MachineFunctionInfo(),
  19   LocalMemoryObjects(),
  20   ExplicitKernArgSize(0),
  21   MaxKernArgAlign(0),
  22   LDSSize(0),
  23   IsEntryFunction(AMDGPU::isEntryFunctionCC(MF.getFunction().getCallingConv())),
  24   NoSignedZerosFPMath(MF.getTarget().Options.NoSignedZerosFPMath),
  25   MemoryBound(false),
  26   WaveLimiter(false) {
  27   const AMDGPUSubtarget &ST = AMDGPUSubtarget::get(MF);
  28
  29   // FIXME: Should initialize KernArgSize based on ExplicitKernelArgOffset,
  30   // except reserved size is not correctly aligned.
  31   const Function &F = MF.getFunction();
  32
  33   if (auto *Resolver = MF.getMMI().getResolver()) {
  34     if (AMDGPUPerfHintAnalysis *PHA = static_cast<AMDGPUPerfHintAnalysis*>(
  35           Resolver->getAnalysisIfAvailable(&AMDGPUPerfHintAnalysisID, true))) {
  36       MemoryBound = PHA->isMemoryBound(&F);
  37       WaveLimiter = PHA->needsWaveLimiter(&F);
  38     }
  39   }
  40
  41   CallingConv::ID CC = F.getCallingConv();
  42   if (CC == CallingConv::AMDGPU_KERNEL || CC == CallingConv::SPIR_KERNEL)
  43     ExplicitKernArgSize = ST.getExplicitKernArgSize(F, MaxKernArgAlign);
  44 }
  45
  46 unsigned AMDGPUMachineFunction::allocateLDSGlobal(const DataLayout &DL,
  47                                                   const GlobalValue &GV) {
  48   auto Entry = LocalMemoryObjects.insert(std::make_pair(&GV, 0));
  49   if (!Entry.second)
  50     return Entry.first->second;
  51
  52   unsigned Align = GV.getAlignment();
  53   if (Align == 0)
  54     Align = DL.getABITypeAlignment(GV.getValueType());
  55
  56   /// TODO: We should sort these to minimize wasted space due to alignment
  57   /// padding. Currently the padding is decided by the first encountered use
  58   /// during lowering.
  59   unsigned Offset = LDSSize = alignTo(LDSSize, Align);
  60
  61   Entry.first->second = Offset;
  62   LDSSize += DL.getTypeAllocSize(GV.getValueType());
  63
  64   return Offset;
  65 }