forked from OSchip/llvm-project
88 lines
3.1 KiB
C++
88 lines
3.1 KiB
C++
//===-- AMDGPUHSATargetObjectFile.cpp - AMDGPU Object Files ---------------===//
|
|
//
|
|
// The LLVM Compiler Infrastructure
|
|
//
|
|
// This file is distributed under the University of Illinois Open Source
|
|
// License. See LICENSE.TXT for details.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#include "AMDGPUTargetObjectFile.h"
|
|
#include "AMDGPU.h"
|
|
#include "Utils/AMDGPUBaseInfo.h"
|
|
#include "llvm/MC/MCContext.h"
|
|
#include "llvm/MC/MCSectionELF.h"
|
|
#include "llvm/Support/ELF.h"
|
|
|
|
using namespace llvm;
|
|
|
|
//===----------------------------------------------------------------------===//
|
|
// Generic Object File
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
MCSection *AMDGPUTargetObjectFile::SelectSectionForGlobal(const GlobalValue *GV,
|
|
SectionKind Kind,
|
|
Mangler &Mang,
|
|
const TargetMachine &TM) const {
|
|
if (Kind.isReadOnly() && AMDGPU::isReadOnlySegment(GV))
|
|
return TextSection;
|
|
|
|
return TargetLoweringObjectFileELF::SelectSectionForGlobal(GV, Kind, Mang, TM);
|
|
}
|
|
|
|
//===----------------------------------------------------------------------===//
|
|
// HSA Object File
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
|
|
void AMDGPUHSATargetObjectFile::Initialize(MCContext &Ctx,
|
|
const TargetMachine &TM){
|
|
TargetLoweringObjectFileELF::Initialize(Ctx, TM);
|
|
InitializeELF(TM.Options.UseInitArray);
|
|
|
|
TextSection = AMDGPU::getHSATextSection(Ctx);
|
|
|
|
DataGlobalAgentSection = AMDGPU::getHSADataGlobalAgentSection(Ctx);
|
|
DataGlobalProgramSection = AMDGPU::getHSADataGlobalProgramSection(Ctx);
|
|
|
|
RodataReadonlyAgentSection = AMDGPU::getHSARodataReadonlyAgentSection(Ctx);
|
|
}
|
|
|
|
bool AMDGPUHSATargetObjectFile::isAgentAllocationSection(
|
|
const char *SectionName) const {
|
|
return cast<MCSectionELF>(DataGlobalAgentSection)
|
|
->getSectionName()
|
|
.equals(SectionName);
|
|
}
|
|
|
|
bool AMDGPUHSATargetObjectFile::isAgentAllocation(const GlobalValue *GV) const {
|
|
// Read-only segments can only have agent allocation.
|
|
return AMDGPU::isReadOnlySegment(GV) ||
|
|
(AMDGPU::isGlobalSegment(GV) && GV->hasSection() &&
|
|
isAgentAllocationSection(GV->getSection()));
|
|
}
|
|
|
|
bool AMDGPUHSATargetObjectFile::isProgramAllocation(
|
|
const GlobalValue *GV) const {
|
|
// The default for global segments is program allocation.
|
|
return AMDGPU::isGlobalSegment(GV) && !isAgentAllocation(GV);
|
|
}
|
|
|
|
MCSection *AMDGPUHSATargetObjectFile::SelectSectionForGlobal(
|
|
const GlobalValue *GV, SectionKind Kind,
|
|
Mangler &Mang,
|
|
const TargetMachine &TM) const {
|
|
if (Kind.isText() && !GV->hasComdat())
|
|
return getTextSection();
|
|
|
|
if (AMDGPU::isGlobalSegment(GV)) {
|
|
if (isAgentAllocation(GV))
|
|
return DataGlobalAgentSection;
|
|
|
|
if (isProgramAllocation(GV))
|
|
return DataGlobalProgramSection;
|
|
}
|
|
|
|
return AMDGPUTargetObjectFile::SelectSectionForGlobal(GV, Kind, Mang, TM);
|
|
}
|