AMDGPU/SI: Add VI patterns to select FLAT instructions for global memory ops
[oota-llvm.git] / lib / Target / AMDGPU / SIISelLowering.h
1 //===-- SIISelLowering.h - SI DAG Lowering Interface ------------*- C++ -*-===//
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 /// \file
11 /// \brief SI DAG Lowering interface definition
12 //
13 //===----------------------------------------------------------------------===//
14
15 #ifndef LLVM_LIB_TARGET_R600_SIISELLOWERING_H
16 #define LLVM_LIB_TARGET_R600_SIISELLOWERING_H
17
18 #include "AMDGPUISelLowering.h"
19 #include "SIInstrInfo.h"
20
21 namespace llvm {
22
23 class SITargetLowering : public AMDGPUTargetLowering {
24   SDValue LowerParameter(SelectionDAG &DAG, EVT VT, EVT MemVT, SDLoc DL,
25                          SDValue Chain, unsigned Offset, bool Signed) const;
26   SDValue LowerSampleIntrinsic(unsigned Opcode, const SDValue &Op,
27                                SelectionDAG &DAG) const;
28   SDValue LowerGlobalAddress(AMDGPUMachineFunction *MFI, SDValue Op,
29                              SelectionDAG &DAG) const override;
30
31   SDValue LowerINTRINSIC_WO_CHAIN(SDValue Op, SelectionDAG &DAG) const;
32   SDValue LowerINTRINSIC_VOID(SDValue Op, SelectionDAG &DAG) const;
33   SDValue LowerFrameIndex(SDValue Op, SelectionDAG &DAG) const;
34   SDValue LowerLOAD(SDValue Op, SelectionDAG &DAG) const;
35   SDValue LowerSELECT(SDValue Op, SelectionDAG &DAG) const;
36   SDValue LowerFastFDIV(SDValue Op, SelectionDAG &DAG) const;
37   SDValue LowerFDIV32(SDValue Op, SelectionDAG &DAG) const;
38   SDValue LowerFDIV64(SDValue Op, SelectionDAG &DAG) const;
39   SDValue LowerFDIV(SDValue Op, SelectionDAG &DAG) const;
40   SDValue LowerINT_TO_FP(SDValue Op, SelectionDAG &DAG, bool Signed) const;
41   SDValue LowerSTORE(SDValue Op, SelectionDAG &DAG) const;
42   SDValue LowerTrig(SDValue Op, SelectionDAG &DAG) const;
43   SDValue LowerBRCOND(SDValue Op, SelectionDAG &DAG) const;
44
45   void adjustWritemask(MachineSDNode *&N, SelectionDAG &DAG) const;
46
47   SDValue performUCharToFloatCombine(SDNode *N,
48                                      DAGCombinerInfo &DCI) const;
49   SDValue performSHLPtrCombine(SDNode *N,
50                                unsigned AS,
51                                DAGCombinerInfo &DCI) const;
52   SDValue performAndCombine(SDNode *N, DAGCombinerInfo &DCI) const;
53   SDValue performOrCombine(SDNode *N, DAGCombinerInfo &DCI) const;
54   SDValue performClassCombine(SDNode *N, DAGCombinerInfo &DCI) const;
55
56   SDValue performMin3Max3Combine(SDNode *N, DAGCombinerInfo &DCI) const;
57   SDValue performSetCCCombine(SDNode *N, DAGCombinerInfo &DCI) const;
58
59   bool isLegalFlatAddressingMode(const AddrMode &AM) const;
60 public:
61   SITargetLowering(TargetMachine &tm, const AMDGPUSubtarget &STI);
62
63   bool isShuffleMaskLegal(const SmallVectorImpl<int> &/*Mask*/,
64                           EVT /*VT*/) const override;
65
66   bool isLegalAddressingMode(const DataLayout &DL, const AddrMode &AM, Type *Ty,
67                              unsigned AS) const override;
68
69   bool allowsMisalignedMemoryAccesses(EVT VT, unsigned AS,
70                                       unsigned Align,
71                                       bool *IsFast) const override;
72
73   EVT getOptimalMemOpType(uint64_t Size, unsigned DstAlign,
74                           unsigned SrcAlign, bool IsMemset,
75                           bool ZeroMemset,
76                           bool MemcpyStrSrc,
77                           MachineFunction &MF) const override;
78
79   TargetLoweringBase::LegalizeTypeAction
80   getPreferredVectorAction(EVT VT) const override;
81
82   bool shouldConvertConstantLoadToIntImm(const APInt &Imm,
83                                         Type *Ty) const override;
84
85   SDValue LowerFormalArguments(SDValue Chain, CallingConv::ID CallConv,
86                                bool isVarArg,
87                                const SmallVectorImpl<ISD::InputArg> &Ins,
88                                SDLoc DL, SelectionDAG &DAG,
89                                SmallVectorImpl<SDValue> &InVals) const override;
90
91   MachineBasicBlock * EmitInstrWithCustomInserter(MachineInstr * MI,
92                                       MachineBasicBlock * BB) const override;
93   bool enableAggressiveFMAFusion(EVT VT) const override;
94   EVT getSetCCResultType(const DataLayout &DL, LLVMContext &Context,
95                          EVT VT) const override;
96   MVT getScalarShiftAmountTy(const DataLayout &, EVT) const override;
97   bool isFMAFasterThanFMulAndFAdd(EVT VT) const override;
98   SDValue LowerOperation(SDValue Op, SelectionDAG &DAG) const override;
99   SDValue PerformDAGCombine(SDNode *N, DAGCombinerInfo &DCI) const override;
100   SDNode *PostISelFolding(MachineSDNode *N, SelectionDAG &DAG) const override;
101   void AdjustInstrPostInstrSelection(MachineInstr *MI,
102                                      SDNode *Node) const override;
103
104   int32_t analyzeImmediate(const SDNode *N) const;
105   SDValue CreateLiveInRegister(SelectionDAG &DAG, const TargetRegisterClass *RC,
106                                unsigned Reg, EVT VT) const override;
107   void legalizeTargetIndependentNode(SDNode *Node, SelectionDAG &DAG) const;
108
109   MachineSDNode *wrapAddr64Rsrc(SelectionDAG &DAG, SDLoc DL, SDValue Ptr) const;
110   MachineSDNode *buildRSRC(SelectionDAG &DAG,
111                            SDLoc DL,
112                            SDValue Ptr,
113                            uint32_t RsrcDword1,
114                            uint64_t RsrcDword2And3) const;
115   MachineSDNode *buildScratchRSRC(SelectionDAG &DAG,
116                                   SDLoc DL,
117                                   SDValue Ptr) const;
118
119   std::pair<unsigned, const TargetRegisterClass *>
120   getRegForInlineAsmConstraint(const TargetRegisterInfo *TRI,
121                                StringRef Constraint, MVT VT) const override;
122   SDValue copyToM0(SelectionDAG &DAG, SDValue Chain, SDLoc DL, SDValue V) const;
123 };
124
125 } // End namespace llvm
126
127 #endif