llvm/lib/Target/NVPTX/NVPTXISelDAGToDAG.h

   1 //===-- NVPTXISelDAGToDAG.h - A dag to dag inst selector for NVPTX --------===//
   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 defines an instruction selector for the NVPTX target.
  10 //
  11 //===----------------------------------------------------------------------===//
  12
  13 #ifndef LLVM_LIB_TARGET_NVPTX_NVPTXISELDAGTODAG_H
  14 #define LLVM_LIB_TARGET_NVPTX_NVPTXISELDAGTODAG_H
  15
  16 #include "MCTargetDesc/NVPTXBaseInfo.h"
  17 #include "NVPTX.h"
  18 #include "NVPTXISelLowering.h"
  19 #include "NVPTXRegisterInfo.h"
  20 #include "NVPTXTargetMachine.h"
  21 #include "llvm/CodeGen/SelectionDAGISel.h"
  22 #include "llvm/IR/InlineAsm.h"
  23 #include "llvm/IR/Intrinsics.h"
  24 #include "llvm/Support/Compiler.h"
  25
  26 namespace llvm {
  27
  28 class LLVM_LIBRARY_VISIBILITY NVPTXDAGToDAGISel : public SelectionDAGISel {
  29   const NVPTXTargetMachine &TM;
  30
  31   // If true, generate mul.wide from sext and mul
  32   bool doMulWide;
  33
  34   int getDivF32Level() const;
  35   bool usePrecSqrtF32() const;
  36   bool useF32FTZ() const;
  37   bool allowFMA() const;
  38   bool allowUnsafeFPMath() const;
  39   bool doRsqrtOpt() const;
  40
  41 public:
  42   NVPTXDAGToDAGISel() = delete;
  43
  44   explicit NVPTXDAGToDAGISel(NVPTXTargetMachine &tm, CodeGenOptLevel OptLevel);
  45
  46   bool runOnMachineFunction(MachineFunction &MF) override;
  47   const NVPTXSubtarget *Subtarget = nullptr;
  48
  49   bool SelectInlineAsmMemoryOperand(const SDValue &Op,
  50                                     InlineAsm::ConstraintCode ConstraintID,
  51                                     std::vector<SDValue> &OutOps) override;
  52
  53 private:
  54 // Include the pieces autogenerated from the target description.
  55 #include "NVPTXGenDAGISel.inc"
  56
  57   void Select(SDNode *N) override;
  58   bool tryIntrinsicNoChain(SDNode *N);
  59   bool tryIntrinsicChain(SDNode *N);
  60   void SelectTexSurfHandle(SDNode *N);
  61   bool tryLoad(SDNode *N);
  62   bool tryLoadVector(SDNode *N);
  63   bool tryLDGLDU(SDNode *N);
  64   bool tryStore(SDNode *N);
  65   bool tryStoreVector(SDNode *N);
  66   bool tryLoadParam(SDNode *N);
  67   bool tryStoreRetval(SDNode *N);
  68   bool tryStoreParam(SDNode *N);
  69   void SelectAddrSpaceCast(SDNode *N);
  70   bool tryTextureIntrinsic(SDNode *N);
  71   bool trySurfaceIntrinsic(SDNode *N);
  72   bool tryBFE(SDNode *N);
  73   bool tryConstantFP(SDNode *N);
  74   bool SelectSETP_F16X2(SDNode *N);
  75   bool SelectSETP_BF16X2(SDNode *N);
  76   bool tryEXTRACT_VECTOR_ELEMENT(SDNode *N);
  77   void SelectV2I64toI128(SDNode *N);
  78   void SelectI128toV2I64(SDNode *N);
  79   inline SDValue getI32Imm(unsigned Imm, const SDLoc &DL) {
  80     return CurDAG->getTargetConstant(Imm, DL, MVT::i32);
  81   }
  82
  83   // Match direct address complex pattern.
  84   bool SelectDirectAddr(SDValue N, SDValue &Address);
  85
  86   bool SelectADDRri_imp(SDNode *OpNode, SDValue Addr, SDValue &Base,
  87                         SDValue &Offset, MVT mvt);
  88   bool SelectADDRri(SDNode *OpNode, SDValue Addr, SDValue &Base,
  89                     SDValue &Offset);
  90   bool SelectADDRri64(SDNode *OpNode, SDValue Addr, SDValue &Base,
  91                       SDValue &Offset);
  92   bool SelectADDRsi_imp(SDNode *OpNode, SDValue Addr, SDValue &Base,
  93                         SDValue &Offset, MVT mvt);
  94   bool SelectADDRsi(SDNode *OpNode, SDValue Addr, SDValue &Base,
  95                     SDValue &Offset);
  96   bool SelectADDRsi64(SDNode *OpNode, SDValue Addr, SDValue &Base,
  97                       SDValue &Offset);
  98
  99   bool ChkMemSDNodeAddressSpace(SDNode *N, unsigned int spN) const;
 100
 101   static unsigned GetConvertOpcode(MVT DestTy, MVT SrcTy, LoadSDNode *N);
 102 };
 103
 104 class NVPTXDAGToDAGISelLegacy : public SelectionDAGISelLegacy {
 105 public:
 106   static char ID;
 107   explicit NVPTXDAGToDAGISelLegacy(NVPTXTargetMachine &tm,
 108                                    CodeGenOptLevel OptLevel);
 109 };
 110 } // end namespace llvm
 111
 112 #endif