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
AMDGPUMachineFunction(const MachineFunction & MF)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
allocateLDSGlobal(const DataLayout & DL,const GlobalValue & GV)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 }
66