lib/Target/NVPTX/NVPTXSubtarget.h

   1 //=====-- NVPTXSubtarget.h - Define Subtarget for the NVPTX ---*- C++ -*--====//
   2 //
   3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
   4 // See https://llvm.org/LICENSE.txt for license information.
   5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
   6 //
   7 //===----------------------------------------------------------------------===//
   8 //
   9 // This file declares the NVPTX specific subclass of TargetSubtarget.
  10 //
  11 //===----------------------------------------------------------------------===//
  12
  13 #ifndef LLVM_LIB_TARGET_NVPTX_NVPTXSUBTARGET_H
  14 #define LLVM_LIB_TARGET_NVPTX_NVPTXSUBTARGET_H
  15
  16 #include "NVPTX.h"
  17 #include "NVPTXFrameLowering.h"
  18 #include "NVPTXISelLowering.h"
  19 #include "NVPTXInstrInfo.h"
  20 #include "NVPTXRegisterInfo.h"
  21 #include "llvm/CodeGen/SelectionDAGTargetInfo.h"
  22 #include "llvm/CodeGen/TargetSubtargetInfo.h"
  23 #include "llvm/IR/DataLayout.h"
  24 #include <string>
  25
  26 #define GET_SUBTARGETINFO_HEADER
  27 #include "NVPTXGenSubtargetInfo.inc"
  28
  29 namespace llvm {
  30
  31 class NVPTXSubtarget : public NVPTXGenSubtargetInfo {
  32   virtual void anchor();
  33   std::string TargetName;
  34
  35   // PTX version x.y is represented as 10*x+y, e.g. 3.1 == 31
  36   unsigned PTXVersion;
  37
  38   // SM version x.y is represented as 10*x+y, e.g. 3.1 == 31
  39   unsigned int SmVersion;
  40
  41   const NVPTXTargetMachine &TM;
  42   NVPTXInstrInfo InstrInfo;
  43   NVPTXTargetLowering TLInfo;
  44   SelectionDAGTargetInfo TSInfo;
  45
  46   // NVPTX does not have any call stack frame, but need a NVPTX specific
  47   // FrameLowering class because TargetFrameLowering is abstract.
  48   NVPTXFrameLowering FrameLowering;
  49
  50 public:
  51   /// This constructor initializes the data members to match that
  52   /// of the specified module.
  53   ///
  54   NVPTXSubtarget(const Triple &TT, const std::string &CPU,
  55                  const std::string &FS, const NVPTXTargetMachine &TM);
  56
  57   const TargetFrameLowering *getFrameLowering() const override {
  58     return &FrameLowering;
  59   }
  60   const NVPTXInstrInfo *getInstrInfo() const override { return &InstrInfo; }
  61   const NVPTXRegisterInfo *getRegisterInfo() const override {
  62     return &InstrInfo.getRegisterInfo();
  63   }
  64   const NVPTXTargetLowering *getTargetLowering() const override {
  65     return &TLInfo;
  66   }
  67   const SelectionDAGTargetInfo *getSelectionDAGInfo() const override {
  68     return &TSInfo;
  69   }
  70
  71   bool hasAtomAddF64() const { return SmVersion >= 60; }
  72   bool hasAtomScope() const { return SmVersion >= 60; }
  73   bool hasAtomBitwise64() const { return SmVersion >= 32; }
  74   bool hasAtomMinMax64() const { return SmVersion >= 32; }
  75   bool hasLDG() const { return SmVersion >= 32; }
  76   inline bool hasHWROT32() const { return SmVersion >= 32; }
  77   bool hasImageHandles() const;
  78   bool hasFP16Math() const { return SmVersion >= 53; }
  79   bool allowFP16Math() const;
  80   unsigned int getSmVersion() const { return SmVersion; }
  81   std::string getTargetName() const { return TargetName; }
  82
  83   unsigned getPTXVersion() const { return PTXVersion; }
  84
  85   NVPTXSubtarget &initializeSubtargetDependencies(StringRef CPU, StringRef FS);
  86   void ParseSubtargetFeatures(StringRef CPU, StringRef FS);
  87 };
  88
  89 } // End llvm namespace
  90
  91 #endif