1 //===-- AMDGPUHSATargetObjectFile.cpp - AMDGPU Object Files ---------------===//
3 // The LLVM Compiler Infrastructure
5 // This file is distributed under the University of Illinois Open Source
6 // License. See LICENSE.TXT for details.
8 //===----------------------------------------------------------------------===//
10 #include "AMDGPUHSATargetObjectFile.h"
12 #include "Utils/AMDGPUBaseInfo.h"
13 #include "llvm/MC/MCContext.h"
14 #include "llvm/MC/MCSectionELF.h"
15 #include "llvm/Support/ELF.h"
19 void AMDGPUHSATargetObjectFile::Initialize(MCContext &Ctx,
20 const TargetMachine &TM){
21 TargetLoweringObjectFileELF::Initialize(Ctx, TM);
22 InitializeELF(TM.Options.UseInitArray);
24 TextSection = AMDGPU::getHSATextSection(Ctx);
26 DataGlobalAgentSection = AMDGPU::getHSADataGlobalAgentSection(Ctx);
27 DataGlobalProgramSection = AMDGPU::getHSADataGlobalProgramSection(Ctx);
29 RodataReadonlyAgentSection = AMDGPU::getHSARodataReadonlyAgentSection(Ctx);
32 bool AMDGPUHSATargetObjectFile::isAgentAllocationSection(
33 const char *SectionName) const {
34 return cast<MCSectionELF>(DataGlobalAgentSection)
39 bool AMDGPUHSATargetObjectFile::isAgentAllocation(const GlobalValue *GV) const {
40 // Read-only segments can only have agent allocation.
41 return AMDGPU::isReadOnlySegment(GV) ||
42 (AMDGPU::isGlobalSegment(GV) && GV->hasSection() &&
43 isAgentAllocationSection(GV->getSection()));
46 bool AMDGPUHSATargetObjectFile::isProgramAllocation(
47 const GlobalValue *GV) const {
48 // The default for global segments is program allocation.
49 return AMDGPU::isGlobalSegment(GV) && !isAgentAllocation(GV);
52 MCSection *AMDGPUHSATargetObjectFile::SelectSectionForGlobal(
53 const GlobalValue *GV, SectionKind Kind,
55 const TargetMachine &TM) const {
56 if (Kind.isText() && !GV->hasComdat())
57 return getTextSection();
59 if (AMDGPU::isGlobalSegment(GV)) {
60 if (isAgentAllocation(GV))
61 return DataGlobalAgentSection;
63 if (isProgramAllocation(GV))
64 return DataGlobalProgramSection;
67 if (Kind.isReadOnly() && AMDGPU::isReadOnlySegment(GV))
68 return RodataReadonlyAgentSection;
70 return TargetLoweringObjectFileELF::SelectSectionForGlobal(GV, Kind, Mang, TM);