lib/Target/AMDGPU/AMDGPUTargetObjectFile.h

   1 //===-- AMDGPUTargetObjectFile.h - AMDGPU  Object Info ----*- C++ -*-===//
   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 /// \file
  11 /// \brief This file declares the AMDGPU-specific subclass of
  12 /// TargetLoweringObjectFile.
  13 ///
  14 //===----------------------------------------------------------------------===//
  15
  16 #ifndef LLVM_LIB_TARGET_AMDGPU_AMDGPUTARGETOBJECTFILE_H
  17 #define LLVM_LIB_TARGET_AMDGPU_AMDGPUTARGETOBJECTFILE_H
  18
  19 #include "llvm/CodeGen/TargetLoweringObjectFileImpl.h"
  20 #include "llvm/Target/TargetMachine.h"
  21
  22 namespace llvm {
  23
  24 class AMDGPUTargetObjectFile : public TargetLoweringObjectFileELF {
  25   public:
  26     MCSection *SelectSectionForGlobal(const GlobalValue *GV, SectionKind Kind,
  27                                       Mangler &Mang,
  28                                       const TargetMachine &TM) const override;
  29 };
  30
  31 class AMDGPUHSATargetObjectFile final : public AMDGPUTargetObjectFile {
  32 private:
  33   MCSection *DataGlobalAgentSection;
  34   MCSection *DataGlobalProgramSection;
  35   MCSection *RodataReadonlyAgentSection;
  36
  37   bool isAgentAllocationSection(const char *SectionName) const;
  38   bool isAgentAllocation(const GlobalValue *GV) const;
  39   bool isProgramAllocation(const GlobalValue *GV) const;
  40
  41 public:
  42   void Initialize(MCContext &Ctx, const TargetMachine &TM) override;
  43
  44   MCSection *SelectSectionForGlobal(const GlobalValue *GV, SectionKind Kind,
  45                                     Mangler &Mang,
  46                                     const TargetMachine &TM) const override;
  47 };
  48
  49 } // end namespace llvm
  50
  51 #endif