NVPTXISelDAGToDAG.h revision dce4a407a24b04eebc6a376f8e62b41aaa7b071f
1//===-- NVPTXISelDAGToDAG.h - A dag to dag inst selector for NVPTX --------===// 2// 3// The LLVM Compiler Infrastructure 4// 5// This file is distributed under the University of Illinois Open Source 6// License. See LICENSE.TXT for details. 7// 8//===----------------------------------------------------------------------===// 9// 10// This file defines an instruction selector for the NVPTX target. 11// 12//===----------------------------------------------------------------------===// 13 14#include "NVPTX.h" 15#include "NVPTXISelLowering.h" 16#include "NVPTXRegisterInfo.h" 17#include "NVPTXTargetMachine.h" 18#include "llvm/CodeGen/SelectionDAGISel.h" 19#include "llvm/IR/Intrinsics.h" 20#include "llvm/Support/Compiler.h" 21using namespace llvm; 22 23namespace { 24 25class LLVM_LIBRARY_VISIBILITY NVPTXDAGToDAGISel : public SelectionDAGISel { 26 27 // If true, generate corresponding FPCONTRACT. This is 28 // language dependent (i.e. CUDA and OpenCL works differently). 29 bool doFMAF64; 30 bool doFMAF32; 31 bool doFMAF64AGG; 32 bool doFMAF32AGG; 33 bool allowFMA; 34 35 // If true, generate mul.wide from sext and mul 36 bool doMulWide; 37 38 int getDivF32Level() const; 39 bool usePrecSqrtF32() const; 40 bool useF32FTZ() const; 41 42public: 43 explicit NVPTXDAGToDAGISel(NVPTXTargetMachine &tm, 44 CodeGenOpt::Level OptLevel); 45 46 // Pass Name 47 const char *getPassName() const override { 48 return "NVPTX DAG->DAG Pattern Instruction Selection"; 49 } 50 51 const NVPTXSubtarget &Subtarget; 52 53 bool SelectInlineAsmMemoryOperand(const SDValue &Op, 54 char ConstraintCode, 55 std::vector<SDValue> &OutOps) override; 56private: 57// Include the pieces autogenerated from the target description. 58#include "NVPTXGenDAGISel.inc" 59 60 SDNode *Select(SDNode *N) override; 61 SDNode *SelectIntrinsicNoChain(SDNode *N); 62 SDNode *SelectTexSurfHandle(SDNode *N); 63 SDNode *SelectLoad(SDNode *N); 64 SDNode *SelectLoadVector(SDNode *N); 65 SDNode *SelectLDGLDUVector(SDNode *N); 66 SDNode *SelectStore(SDNode *N); 67 SDNode *SelectStoreVector(SDNode *N); 68 SDNode *SelectLoadParam(SDNode *N); 69 SDNode *SelectStoreRetval(SDNode *N); 70 SDNode *SelectStoreParam(SDNode *N); 71 SDNode *SelectAddrSpaceCast(SDNode *N); 72 SDNode *SelectTextureIntrinsic(SDNode *N); 73 SDNode *SelectSurfaceIntrinsic(SDNode *N); 74 75 inline SDValue getI32Imm(unsigned Imm) { 76 return CurDAG->getTargetConstant(Imm, MVT::i32); 77 } 78 79 // Match direct address complex pattern. 80 bool SelectDirectAddr(SDValue N, SDValue &Address); 81 82 bool SelectADDRri_imp(SDNode *OpNode, SDValue Addr, SDValue &Base, 83 SDValue &Offset, MVT mvt); 84 bool SelectADDRri(SDNode *OpNode, SDValue Addr, SDValue &Base, 85 SDValue &Offset); 86 bool SelectADDRri64(SDNode *OpNode, SDValue Addr, SDValue &Base, 87 SDValue &Offset); 88 89 bool SelectADDRsi_imp(SDNode *OpNode, SDValue Addr, SDValue &Base, 90 SDValue &Offset, MVT mvt); 91 bool SelectADDRsi(SDNode *OpNode, SDValue Addr, SDValue &Base, 92 SDValue &Offset); 93 bool SelectADDRsi64(SDNode *OpNode, SDValue Addr, SDValue &Base, 94 SDValue &Offset); 95 96 bool ChkMemSDNodeAddressSpace(SDNode *N, unsigned int spN) const; 97 98}; 99} 100