NVPTXISelDAGToDAG.h revision 360784
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 25namespace llvm { 26 27class 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 40public: 41 explicit NVPTXDAGToDAGISel(NVPTXTargetMachine &tm, 42 CodeGenOpt::Level OptLevel); 43 44 // Pass Name 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; 54private: 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 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