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);
31 bool AMDGPUHSATargetObjectFile::isAgentAllocationSection(
32 const char *SectionName) const {
33 return cast<MCSectionELF>(DataGlobalAgentSection)
38 bool AMDGPUHSATargetObjectFile::isAgentAllocation(const GlobalValue *GV) const {
39 // Read-only segments can only have agent allocation.
40 return AMDGPU::isReadOnlySegment(GV) ||
41 (AMDGPU::isGlobalSegment(GV) && GV->hasSection() &&
42 isAgentAllocationSection(GV->getSection()));
45 bool AMDGPUHSATargetObjectFile::isProgramAllocation(
46 const GlobalValue *GV) const {
47 // The default for global segments is program allocation.
48 return AMDGPU::isGlobalSegment(GV) && !isAgentAllocation(GV);
51 MCSection *AMDGPUHSATargetObjectFile::SelectSectionForGlobal(
52 const GlobalValue *GV, SectionKind Kind,
54 const TargetMachine &TM) const {
55 if (Kind.isText() && !GV->hasComdat())
56 return getTextSection();
58 if (AMDGPU::isGlobalSegment(GV)) {
59 if (isAgentAllocation(GV))
60 return DataGlobalAgentSection;
62 if (isProgramAllocation(GV))
63 return DataGlobalProgramSection;
66 return TargetLoweringObjectFileELF::SelectSectionForGlobal(GV, Kind, Mang, TM);