1 //===-- AMDGPUHSATargetObjectFile.cpp - AMDGPU Object Files ---------------===//
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 "AMDGPUTargetObjectFile.h"
11 #include "AMDGPU.h"
12 #include "Utils/AMDGPUBaseInfo.h"
13 #include "llvm/MC/MCContext.h"
14 #include "llvm/MC/MCSectionELF.h"
15 #include "llvm/Support/ELF.h"
16 
17 using namespace llvm;
18 
19 //===----------------------------------------------------------------------===//
20 // Generic Object File
21 //===----------------------------------------------------------------------===//
22 
23 MCSection *AMDGPUTargetObjectFile::SelectSectionForGlobal(const GlobalValue *GV,
24                                                           SectionKind Kind,
25                                                           Mangler &Mang,
26                                                 const TargetMachine &TM) const {
27   if (Kind.isReadOnly() && AMDGPU::isReadOnlySegment(GV))
28     return TextSection;
29 
30   return TargetLoweringObjectFileELF::SelectSectionForGlobal(GV, Kind, Mang, TM);
31 }
32 
33 //===----------------------------------------------------------------------===//
34 // HSA Object File
35 //===----------------------------------------------------------------------===//
36 
37 
38 void AMDGPUHSATargetObjectFile::Initialize(MCContext &Ctx,
39                                            const TargetMachine &TM){
40   TargetLoweringObjectFileELF::Initialize(Ctx, TM);
41   InitializeELF(TM.Options.UseInitArray);
42 
43   TextSection = AMDGPU::getHSATextSection(Ctx);
44 
45   DataGlobalAgentSection = AMDGPU::getHSADataGlobalAgentSection(Ctx);
46   DataGlobalProgramSection = AMDGPU::getHSADataGlobalProgramSection(Ctx);
47 
48   RodataReadonlyAgentSection = AMDGPU::getHSARodataReadonlyAgentSection(Ctx);
49 }
50 
51 bool AMDGPUHSATargetObjectFile::isAgentAllocationSection(
52     const char *SectionName) const {
53   return cast<MCSectionELF>(DataGlobalAgentSection)
54       ->getSectionName()
55       .equals(SectionName);
56 }
57 
58 bool AMDGPUHSATargetObjectFile::isAgentAllocation(const GlobalValue *GV) const {
59   // Read-only segments can only have agent allocation.
60   return AMDGPU::isReadOnlySegment(GV) ||
61          (AMDGPU::isGlobalSegment(GV) && GV->hasSection() &&
62           isAgentAllocationSection(GV->getSection()));
63 }
64 
65 bool AMDGPUHSATargetObjectFile::isProgramAllocation(
66     const GlobalValue *GV) const {
67   // The default for global segments is program allocation.
68   return AMDGPU::isGlobalSegment(GV) && !isAgentAllocation(GV);
69 }
70 
71 MCSection *AMDGPUHSATargetObjectFile::SelectSectionForGlobal(
72                                         const GlobalValue *GV, SectionKind Kind,
73                                         Mangler &Mang,
74                                         const TargetMachine &TM) const {
75   if (Kind.isText() && !GV->hasComdat())
76     return getTextSection();
77 
78   if (AMDGPU::isGlobalSegment(GV)) {
79     if (isAgentAllocation(GV))
80       return DataGlobalAgentSection;
81 
82     if (isProgramAllocation(GV))
83       return DataGlobalProgramSection;
84   }
85 
86   return AMDGPUTargetObjectFile::SelectSectionForGlobal(GV, Kind, Mang, TM);
87 }
88