Tom Stellard | ca16621 | 2017-01-30 21:56:46 +0000 | [diff] [blame] | 1 | //===- AMDGPUInstructionSelector --------------------------------*- 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 | /// \file |
| 10 | /// This file declares the targeting of the InstructionSelector class for |
| 11 | /// AMDGPU. |
| 12 | //===----------------------------------------------------------------------===// |
| 13 | |
| 14 | #ifndef LLVM_LIB_TARGET_AMDGPU_AMDGPUINSTRUCTIONSELECTOR_H |
| 15 | #define LLVM_LIB_TARGET_AMDGPU_AMDGPUINSTRUCTIONSELECTOR_H |
| 16 | |
Yaxun Liu | 1a14bfa | 2017-03-27 14:04:01 +0000 | [diff] [blame] | 17 | #include "AMDGPU.h" |
Matt Arsenault | b1cc4f5 | 2018-06-25 16:17:48 +0000 | [diff] [blame] | 18 | #include "AMDGPUArgumentUsageInfo.h" |
Tom Stellard | ca16621 | 2017-01-30 21:56:46 +0000 | [diff] [blame] | 19 | #include "llvm/ADT/ArrayRef.h" |
| 20 | #include "llvm/ADT/SmallVector.h" |
Chandler Carruth | 6bda14b | 2017-06-06 11:49:48 +0000 | [diff] [blame] | 21 | #include "llvm/CodeGen/GlobalISel/InstructionSelector.h" |
Tom Stellard | ca16621 | 2017-01-30 21:56:46 +0000 | [diff] [blame] | 22 | |
Tom Stellard | 1dc9020 | 2018-05-10 20:53:06 +0000 | [diff] [blame] | 23 | namespace { |
| 24 | #define GET_GLOBALISEL_PREDICATE_BITSET |
Tom Stellard | 5bfbae5 | 2018-07-11 20:59:01 +0000 | [diff] [blame] | 25 | #define AMDGPUSubtarget GCNSubtarget |
Tom Stellard | 1dc9020 | 2018-05-10 20:53:06 +0000 | [diff] [blame] | 26 | #include "AMDGPUGenGlobalISel.inc" |
| 27 | #undef GET_GLOBALISEL_PREDICATE_BITSET |
Tom Stellard | 5bfbae5 | 2018-07-11 20:59:01 +0000 | [diff] [blame] | 28 | #undef AMDGPUSubtarget |
Tom Stellard | 1dc9020 | 2018-05-10 20:53:06 +0000 | [diff] [blame] | 29 | } |
| 30 | |
Tom Stellard | ca16621 | 2017-01-30 21:56:46 +0000 | [diff] [blame] | 31 | namespace llvm { |
| 32 | |
| 33 | class AMDGPUInstrInfo; |
| 34 | class AMDGPURegisterBankInfo; |
Tom Stellard | 5bfbae5 | 2018-07-11 20:59:01 +0000 | [diff] [blame] | 35 | class GCNSubtarget; |
Tom Stellard | ca16621 | 2017-01-30 21:56:46 +0000 | [diff] [blame] | 36 | class MachineInstr; |
| 37 | class MachineOperand; |
| 38 | class MachineRegisterInfo; |
| 39 | class SIInstrInfo; |
Matt Arsenault | b1cc4f5 | 2018-06-25 16:17:48 +0000 | [diff] [blame] | 40 | class SIMachineFunctionInfo; |
Tom Stellard | ca16621 | 2017-01-30 21:56:46 +0000 | [diff] [blame] | 41 | class SIRegisterInfo; |
Tom Stellard | ca16621 | 2017-01-30 21:56:46 +0000 | [diff] [blame] | 42 | |
| 43 | class AMDGPUInstructionSelector : public InstructionSelector { |
| 44 | public: |
Tom Stellard | 5bfbae5 | 2018-07-11 20:59:01 +0000 | [diff] [blame] | 45 | AMDGPUInstructionSelector(const GCNSubtarget &STI, |
Tom Stellard | 1dc9020 | 2018-05-10 20:53:06 +0000 | [diff] [blame] | 46 | const AMDGPURegisterBankInfo &RBI, |
| 47 | const AMDGPUTargetMachine &TM); |
Tom Stellard | ca16621 | 2017-01-30 21:56:46 +0000 | [diff] [blame] | 48 | |
Daniel Sanders | f76f315 | 2017-11-16 00:46:35 +0000 | [diff] [blame] | 49 | bool select(MachineInstr &I, CodeGenCoverage &CoverageInfo) const override; |
Tom Stellard | 1dc9020 | 2018-05-10 20:53:06 +0000 | [diff] [blame] | 50 | static const char *getName(); |
Daniel Sanders | f76f315 | 2017-11-16 00:46:35 +0000 | [diff] [blame] | 51 | |
Tom Stellard | ca16621 | 2017-01-30 21:56:46 +0000 | [diff] [blame] | 52 | private: |
| 53 | struct GEPInfo { |
| 54 | const MachineInstr &GEP; |
| 55 | SmallVector<unsigned, 2> SgprParts; |
| 56 | SmallVector<unsigned, 2> VgprParts; |
| 57 | int64_t Imm; |
| 58 | GEPInfo(const MachineInstr &GEP) : GEP(GEP), Imm(0) { } |
| 59 | }; |
| 60 | |
Tom Stellard | 1dc9020 | 2018-05-10 20:53:06 +0000 | [diff] [blame] | 61 | /// tblgen-erated 'select' implementation. |
| 62 | bool selectImpl(MachineInstr &I, CodeGenCoverage &CoverageInfo) const; |
| 63 | |
Tom Stellard | ca16621 | 2017-01-30 21:56:46 +0000 | [diff] [blame] | 64 | MachineOperand getSubOperand64(MachineOperand &MO, unsigned SubIdx) const; |
Tom Stellard | 1e0edad | 2018-05-10 21:20:10 +0000 | [diff] [blame] | 65 | bool selectCOPY(MachineInstr &I) const; |
Tom Stellard | ca16621 | 2017-01-30 21:56:46 +0000 | [diff] [blame] | 66 | bool selectG_CONSTANT(MachineInstr &I) const; |
| 67 | bool selectG_ADD(MachineInstr &I) const; |
| 68 | bool selectG_GEP(MachineInstr &I) const; |
Tom Stellard | 3f1c6fe | 2018-06-21 23:38:20 +0000 | [diff] [blame] | 69 | bool selectG_IMPLICIT_DEF(MachineInstr &I) const; |
Tom Stellard | a928473 | 2018-06-14 19:26:37 +0000 | [diff] [blame] | 70 | bool selectG_INTRINSIC(MachineInstr &I, CodeGenCoverage &CoverageInfo) const; |
Tom Stellard | 390a5f4 | 2018-07-13 21:05:14 +0000 | [diff] [blame] | 71 | bool selectG_INTRINSIC_W_SIDE_EFFECTS(MachineInstr &I, |
| 72 | CodeGenCoverage &CoverageInfo) const; |
Tom Stellard | ca16621 | 2017-01-30 21:56:46 +0000 | [diff] [blame] | 73 | bool hasVgprParts(ArrayRef<GEPInfo> AddrInfo) const; |
| 74 | void getAddrModeInfo(const MachineInstr &Load, const MachineRegisterInfo &MRI, |
| 75 | SmallVectorImpl<GEPInfo> &AddrInfo) const; |
| 76 | bool selectSMRD(MachineInstr &I, ArrayRef<GEPInfo> AddrInfo) const; |
| 77 | bool selectG_LOAD(MachineInstr &I) const; |
| 78 | bool selectG_STORE(MachineInstr &I) const; |
| 79 | |
Tom Stellard | 1dc9020 | 2018-05-10 20:53:06 +0000 | [diff] [blame] | 80 | InstructionSelector::ComplexRendererFns |
Tom Stellard | 26fac0f | 2018-06-22 02:54:57 +0000 | [diff] [blame] | 81 | selectVCSRC(MachineOperand &Root) const; |
| 82 | |
| 83 | InstructionSelector::ComplexRendererFns |
Tom Stellard | 1dc9020 | 2018-05-10 20:53:06 +0000 | [diff] [blame] | 84 | selectVSRC0(MachineOperand &Root) const; |
| 85 | |
Tom Stellard | dcc95e9 | 2018-05-11 05:44:16 +0000 | [diff] [blame] | 86 | InstructionSelector::ComplexRendererFns |
| 87 | selectVOP3Mods0(MachineOperand &Root) const; |
Tom Stellard | 46bbbc3 | 2018-06-13 22:30:47 +0000 | [diff] [blame] | 88 | InstructionSelector::ComplexRendererFns |
Tom Stellard | 9a65357 | 2018-06-22 02:34:29 +0000 | [diff] [blame] | 89 | selectVOP3OMods(MachineOperand &Root) const; |
| 90 | InstructionSelector::ComplexRendererFns |
Tom Stellard | 46bbbc3 | 2018-06-13 22:30:47 +0000 | [diff] [blame] | 91 | selectVOP3Mods(MachineOperand &Root) const; |
Tom Stellard | dcc95e9 | 2018-05-11 05:44:16 +0000 | [diff] [blame] | 92 | |
Tom Stellard | ca16621 | 2017-01-30 21:56:46 +0000 | [diff] [blame] | 93 | const SIInstrInfo &TII; |
| 94 | const SIRegisterInfo &TRI; |
| 95 | const AMDGPURegisterBankInfo &RBI; |
Tom Stellard | 1dc9020 | 2018-05-10 20:53:06 +0000 | [diff] [blame] | 96 | const AMDGPUTargetMachine &TM; |
Tom Stellard | 5bfbae5 | 2018-07-11 20:59:01 +0000 | [diff] [blame] | 97 | const GCNSubtarget &STI; |
Tom Stellard | 1dc9020 | 2018-05-10 20:53:06 +0000 | [diff] [blame] | 98 | bool EnableLateStructurizeCFG; |
| 99 | #define GET_GLOBALISEL_PREDICATES_DECL |
Tom Stellard | 5bfbae5 | 2018-07-11 20:59:01 +0000 | [diff] [blame] | 100 | #define AMDGPUSubtarget GCNSubtarget |
Tom Stellard | 1dc9020 | 2018-05-10 20:53:06 +0000 | [diff] [blame] | 101 | #include "AMDGPUGenGlobalISel.inc" |
| 102 | #undef GET_GLOBALISEL_PREDICATES_DECL |
Tom Stellard | 5bfbae5 | 2018-07-11 20:59:01 +0000 | [diff] [blame] | 103 | #undef AMDGPUSubtarget |
Tom Stellard | 1dc9020 | 2018-05-10 20:53:06 +0000 | [diff] [blame] | 104 | |
| 105 | #define GET_GLOBALISEL_TEMPORARIES_DECL |
| 106 | #include "AMDGPUGenGlobalISel.inc" |
| 107 | #undef GET_GLOBALISEL_TEMPORARIES_DECL |
| 108 | |
Yaxun Liu | 1a14bfa | 2017-03-27 14:04:01 +0000 | [diff] [blame] | 109 | protected: |
| 110 | AMDGPUAS AMDGPUASI; |
Tom Stellard | ca16621 | 2017-01-30 21:56:46 +0000 | [diff] [blame] | 111 | }; |
| 112 | |
| 113 | } // End llvm namespace. |
| 114 | #endif |