llvm/lib/Target/SPIRV/SPIRVSubtarget.h

   1 //===-- SPIRVSubtarget.h - SPIR-V Subtarget Information --------*- 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 SPIR-V specific subclass of TargetSubtargetInfo.
  10 //
  11 //===----------------------------------------------------------------------===//
  12
  13 #ifndef LLVM_LIB_TARGET_SPIRV_SPIRVSUBTARGET_H
  14 #define LLVM_LIB_TARGET_SPIRV_SPIRVSUBTARGET_H
  15
  16 #include "SPIRVCallLowering.h"
  17 #include "SPIRVFrameLowering.h"
  18 #include "SPIRVISelLowering.h"
  19 #include "SPIRVInstrInfo.h"
  20 #include "llvm/ADT/SmallSet.h"
  21 #include "llvm/CodeGen/GlobalISel/CallLowering.h"
  22 #include "llvm/CodeGen/GlobalISel/InstructionSelector.h"
  23 #include "llvm/CodeGen/GlobalISel/LegalizerInfo.h"
  24 #include "llvm/CodeGen/SelectionDAGTargetInfo.h"
  25 #include "llvm/CodeGen/TargetSubtargetInfo.h"
  26 #include "llvm/IR/DataLayout.h"
  27 #include "llvm/Target/TargetMachine.h"
  28 #include "llvm/TargetParser/Triple.h"
  29
  30 #define GET_SUBTARGETINFO_HEADER
  31 #include "SPIRVGenSubtargetInfo.inc"
  32
  33 namespace llvm {
  34 class StringRef;
  35 class SPIRVTargetMachine;
  36
  37 class SPIRVSubtarget : public SPIRVGenSubtargetInfo {
  38 private:
  39   const unsigned PointerSize;
  40   VersionTuple SPIRVVersion;
  41   VersionTuple OpenCLVersion;
  42
  43   SmallSet<SPIRV::Extension::Extension, 4> AvailableExtensions;
  44   SmallSet<SPIRV::InstructionSet::InstructionSet, 4> AvailableExtInstSets;
  45   std::unique_ptr<SPIRVGlobalRegistry> GR;
  46
  47   SPIRVInstrInfo InstrInfo;
  48   SPIRVFrameLowering FrameLowering;
  49   SPIRVTargetLowering TLInfo;
  50   Triple TargetTriple;
  51
  52   // GlobalISel related APIs.
  53   std::unique_ptr<CallLowering> CallLoweringInfo;
  54   std::unique_ptr<RegisterBankInfo> RegBankInfo;
  55   std::unique_ptr<LegalizerInfo> Legalizer;
  56   std::unique_ptr<InstructionSelector> InstSelector;
  57
  58   // TODO: Initialise the available extensions, extended instruction sets
  59   // based on the environment settings.
  60   void initAvailableExtensions();
  61   void initAvailableExtInstSets();
  62
  63 public:
  64   // This constructor initializes the data members to match that
  65   // of the specified triple.
  66   SPIRVSubtarget(const Triple &TT, const std::string &CPU,
  67                  const std::string &FS, const SPIRVTargetMachine &TM);
  68   SPIRVSubtarget &initSubtargetDependencies(StringRef CPU, StringRef FS);
  69
  70   // Parses features string setting specified subtarget options.
  71   // The definition of this function is auto generated by tblgen.
  72   void ParseSubtargetFeatures(StringRef CPU, StringRef TuneCPU, StringRef FS);
  73   unsigned getPointerSize() const { return PointerSize; }
  74   unsigned getBound() const { return GR->getBound(); }
  75   bool canDirectlyComparePointers() const;
  76   // TODO: this environment is not implemented in Triple, we need to decide
  77   // how to standardize its support. For now, let's assume SPIR-V with physical
  78   // addressing is OpenCL, and Logical addressing is Vulkan.
  79   bool isOpenCLEnv() const {
  80     return TargetTriple.getArch() == Triple::spirv32 ||
  81            TargetTriple.getArch() == Triple::spirv64;
  82   }
  83   bool isVulkanEnv() const { return TargetTriple.getArch() == Triple::spirv; }
  84   VersionTuple getSPIRVVersion() const { return SPIRVVersion; };
  85   bool isAtLeastSPIRVVer(VersionTuple VerToCompareTo) const;
  86   bool isAtLeastOpenCLVer(VersionTuple VerToCompareTo) const;
  87   // TODO: implement command line args or other ways to determine this.
  88   bool hasOpenCLFullProfile() const { return true; }
  89   bool hasOpenCLImageSupport() const { return true; }
  90   const SmallSet<SPIRV::Extension::Extension, 4> &
  91   getAllAvailableExtensions() const {
  92     return AvailableExtensions;
  93   }
  94   bool canUseExtension(SPIRV::Extension::Extension E) const;
  95   bool canUseExtInstSet(SPIRV::InstructionSet::InstructionSet E) const;
  96
  97   SPIRVGlobalRegistry *getSPIRVGlobalRegistry() const { return GR.get(); }
  98
  99   const CallLowering *getCallLowering() const override {
 100     return CallLoweringInfo.get();
 101   }
 102   const RegisterBankInfo *getRegBankInfo() const override {
 103     return RegBankInfo.get();
 104   }
 105   const LegalizerInfo *getLegalizerInfo() const override {
 106     return Legalizer.get();
 107   }
 108   InstructionSelector *getInstructionSelector() const override {
 109     return InstSelector.get();
 110   }
 111   const SPIRVInstrInfo *getInstrInfo() const override { return &InstrInfo; }
 112   const SPIRVFrameLowering *getFrameLowering() const override {
 113     return &FrameLowering;
 114   }
 115   const SPIRVTargetLowering *getTargetLowering() const override {
 116     return &TLInfo;
 117   }
 118   const SPIRVRegisterInfo *getRegisterInfo() const override {
 119     return &InstrInfo.getRegisterInfo();
 120   }
 121
 122   static bool classof(const TargetSubtargetInfo *ST) {
 123     return ST->getTargetTriple().isSPIRV();
 124   }
 125 };
 126 } // namespace llvm
 127
 128 #endif // LLVM_LIB_TARGET_SPIRV_SPIRVSUBTARGET_H