lib/Target/AMDGPU/AMDGPUHSATargetObjectFile.h

   1 //===-- AMDGPUHSATargetObjectFile.h - AMDGPU HSA 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 use for targeting the HSA-runtime.
  13 ///
  14 //===----------------------------------------------------------------------===//
  15
  16 #ifndef LLVM_LIB_TARGET_AMDGPU_AMDGPUHSATARGETOBJECTFILE_H
  17 #define LLVM_LIB_TARGET_AMDGPU_AMDGPUHSATARGETOBJECTFILE_H
  18
  19 #include "llvm/CodeGen/TargetLoweringObjectFileImpl.h"
  20 #include "llvm/Target/TargetMachine.h"
  21
  22 namespace llvm {
  23
  24 class AMDGPUHSATargetObjectFile final : public TargetLoweringObjectFileELF {
  25 private:
  26   MCSection *DataGlobalAgentSection;
  27   MCSection *DataGlobalProgramSection;
  28
  29   bool isAgentAllocationSection(const char *SectionName) const;
  30   bool isAgentAllocation(const GlobalValue *GV) const;
  31   bool isProgramAllocation(const GlobalValue *GV) const;
  32
  33 public:
  34   void Initialize(MCContext &Ctx, const TargetMachine &TM) override;
  35
  36   MCSection *SelectSectionForGlobal(const GlobalValue *GV, SectionKind Kind,
  37                                     Mangler &Mang,
  38                                     const TargetMachine &TM) const override;
  39 };
  40
  41 } // end namespace llvm
  42
  43 #endif