AMDGPUMachineFunction.cpp revision 341825
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
15using namespace llvm;
16
17AMDGPUMachineFunction::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
46unsigned 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}
66