contrib/llvm/lib/Target/AMDGPU/AMDGPUTargetMachine.h

   1 //===-- AMDGPUTargetMachine.h - AMDGPU TargetMachine Interface --*- 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 The AMDGPU TargetMachine interface definition for hw codgen targets.
  12 //
  13 //===----------------------------------------------------------------------===//
  14
  15 #ifndef LLVM_LIB_TARGET_AMDGPU_AMDGPUTARGETMACHINE_H
  16 #define LLVM_LIB_TARGET_AMDGPU_AMDGPUTARGETMACHINE_H
  17
  18 #include "AMDGPUIntrinsicInfo.h"
  19 #include "AMDGPUSubtarget.h"
  20 #include "llvm/ADT/Optional.h"
  21 #include "llvm/ADT/StringMap.h"
  22 #include "llvm/ADT/StringRef.h"
  23 #include "llvm/Analysis/TargetTransformInfo.h"
  24 #include "llvm/Support/CodeGen.h"
  25 #include "llvm/Target/TargetMachine.h"
  26 #include <memory>
  27
  28 namespace llvm {
  29
  30 //===----------------------------------------------------------------------===//
  31 // AMDGPU Target Machine (R600+)
  32 //===----------------------------------------------------------------------===//
  33
  34 class AMDGPUTargetMachine : public LLVMTargetMachine {
  35 protected:
  36   std::unique_ptr<TargetLoweringObjectFile> TLOF;
  37   AMDGPUIntrinsicInfo IntrinsicInfo;
  38   AMDGPUAS AS;
  39
  40   StringRef getGPUName(const Function &F) const;
  41   StringRef getFeatureString(const Function &F) const;
  42
  43 public:
  44   AMDGPUTargetMachine(const Target &T, const Triple &TT, StringRef CPU,
  45                       StringRef FS, TargetOptions Options,
  46                       Optional<Reloc::Model> RM, CodeModel::Model CM,
  47                       CodeGenOpt::Level OL);
  48   ~AMDGPUTargetMachine() override;
  49
  50   const AMDGPUSubtarget *getSubtargetImpl() const;
  51   const AMDGPUSubtarget *getSubtargetImpl(const Function &) const override = 0;
  52
  53   const AMDGPUIntrinsicInfo *getIntrinsicInfo() const override {
  54     return &IntrinsicInfo;
  55   }
  56   TargetIRAnalysis getTargetIRAnalysis() override;
  57
  58   TargetLoweringObjectFile *getObjFileLowering() const override {
  59     return TLOF.get();
  60   }
  61   AMDGPUAS getAMDGPUAS() const {
  62     return AS;
  63   }
  64
  65   void adjustPassManager(PassManagerBuilder &) override;
  66   /// Get the integer value of a null pointer in the given address space.
  67   uint64_t getNullPointerValue(unsigned AddrSpace) const {
  68     if (AddrSpace == AS.LOCAL_ADDRESS || AddrSpace == AS.REGION_ADDRESS)
  69       return -1;
  70     return 0;
  71   }
  72
  73 };
  74
  75 //===----------------------------------------------------------------------===//
  76 // R600 Target Machine (R600 -> Cayman)
  77 //===----------------------------------------------------------------------===//
  78
  79 class R600TargetMachine final : public AMDGPUTargetMachine {
  80 private:
  81   mutable StringMap<std::unique_ptr<R600Subtarget>> SubtargetMap;
  82
  83 public:
  84   R600TargetMachine(const Target &T, const Triple &TT, StringRef CPU,
  85                     StringRef FS, TargetOptions Options,
  86                     Optional<Reloc::Model> RM, CodeModel::Model CM,
  87                     CodeGenOpt::Level OL);
  88
  89   TargetPassConfig *createPassConfig(PassManagerBase &PM) override;
  90
  91   const R600Subtarget *getSubtargetImpl(const Function &) const override;
  92 };
  93
  94 //===----------------------------------------------------------------------===//
  95 // GCN Target Machine (SI+)
  96 //===----------------------------------------------------------------------===//
  97
  98 class GCNTargetMachine final : public AMDGPUTargetMachine {
  99 private:
 100   mutable StringMap<std::unique_ptr<SISubtarget>> SubtargetMap;
 101
 102 public:
 103   GCNTargetMachine(const Target &T, const Triple &TT, StringRef CPU,
 104                    StringRef FS, TargetOptions Options,
 105                    Optional<Reloc::Model> RM, CodeModel::Model CM,
 106                    CodeGenOpt::Level OL);
 107
 108   TargetPassConfig *createPassConfig(PassManagerBase &PM) override;
 109
 110   const SISubtarget *getSubtargetImpl(const Function &) const override;
 111 };
 112
 113 } // end namespace llvm
 114
 115 #endif // LLVM_LIB_TARGET_AMDGPU_AMDGPUTARGETMACHINE_H