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