1 //===- AMDGPULegalizerInfo ---------------------------------------*- C++ -*-==// 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 /// \file 9 /// This file declares the targeting of the Machinelegalizer class for 10 /// AMDGPU. 11 /// \todo This should be generated by TableGen. 12 //===----------------------------------------------------------------------===// 13 14 #ifndef LLVM_LIB_TARGET_AMDGPU_AMDGPUMACHINELEGALIZER_H 15 #define LLVM_LIB_TARGET_AMDGPU_AMDGPUMACHINELEGALIZER_H 16 17 #include "llvm/CodeGen/GlobalISel/LegalizerInfo.h" 18 #include "AMDGPUArgumentUsageInfo.h" 19 #include "SIInstrInfo.h" 20 21 namespace llvm { 22 23 class GCNTargetMachine; 24 class LLVMContext; 25 class GCNSubtarget; 26 27 /// This class provides the information for the target register banks. 28 class AMDGPULegalizerInfo : public LegalizerInfo { 29 const GCNSubtarget &ST; 30 31 public: 32 AMDGPULegalizerInfo(const GCNSubtarget &ST, 33 const GCNTargetMachine &TM); 34 35 bool legalizeCustom(MachineInstr &MI, MachineRegisterInfo &MRI, 36 MachineIRBuilder &B, 37 GISelChangeObserver &Observer) const override; 38 39 Register getSegmentAperture(unsigned AddrSpace, 40 MachineRegisterInfo &MRI, 41 MachineIRBuilder &B) const; 42 43 bool legalizeAddrSpaceCast(MachineInstr &MI, MachineRegisterInfo &MRI, 44 MachineIRBuilder &B) const; 45 bool legalizeFrint(MachineInstr &MI, MachineRegisterInfo &MRI, 46 MachineIRBuilder &B) const; 47 bool legalizeFceil(MachineInstr &MI, MachineRegisterInfo &MRI, 48 MachineIRBuilder &B) const; 49 bool legalizeIntrinsicTrunc(MachineInstr &MI, MachineRegisterInfo &MRI, 50 MachineIRBuilder &B) const; 51 bool legalizeITOFP(MachineInstr &MI, MachineRegisterInfo &MRI, 52 MachineIRBuilder &B, bool Signed) const; 53 bool legalizeFPTOI(MachineInstr &MI, MachineRegisterInfo &MRI, 54 MachineIRBuilder &B, bool Signed) const; 55 bool legalizeMinNumMaxNum(MachineInstr &MI, MachineRegisterInfo &MRI, 56 MachineIRBuilder &B) const; 57 bool legalizeExtractVectorElt(MachineInstr &MI, MachineRegisterInfo &MRI, 58 MachineIRBuilder &B) const; 59 bool legalizeInsertVectorElt(MachineInstr &MI, MachineRegisterInfo &MRI, 60 MachineIRBuilder &B) const; 61 bool legalizeShuffleVector(MachineInstr &MI, MachineRegisterInfo &MRI, 62 MachineIRBuilder &B) const; 63 64 bool legalizeSinCos(MachineInstr &MI, MachineRegisterInfo &MRI, 65 MachineIRBuilder &B) const; 66 67 bool buildPCRelGlobalAddress( 68 Register DstReg, LLT PtrTy, MachineIRBuilder &B, const GlobalValue *GV, 69 unsigned Offset, unsigned GAFlags = SIInstrInfo::MO_NONE) const; 70 71 bool legalizeGlobalValue(MachineInstr &MI, MachineRegisterInfo &MRI, 72 MachineIRBuilder &B) const; 73 bool legalizeLoad(MachineInstr &MI, MachineRegisterInfo &MRI, 74 MachineIRBuilder &B, 75 GISelChangeObserver &Observer) const; 76 77 bool legalizeFMad(MachineInstr &MI, MachineRegisterInfo &MRI, 78 MachineIRBuilder &B) const; 79 80 bool legalizeAtomicCmpXChg(MachineInstr &MI, MachineRegisterInfo &MRI, 81 MachineIRBuilder &B) const; 82 bool legalizeFlog(MachineInstr &MI, MachineIRBuilder &B, 83 double Log2BaseInverted) const; 84 Register getLiveInRegister(MachineRegisterInfo &MRI, 85 Register Reg, LLT Ty) const; 86 87 bool loadInputValue(Register DstReg, MachineIRBuilder &B, 88 const ArgDescriptor *Arg) const; 89 bool legalizePreloadedArgIntrin( 90 MachineInstr &MI, MachineRegisterInfo &MRI, MachineIRBuilder &B, 91 AMDGPUFunctionArgInfo::PreloadedValue ArgType) const; 92 93 bool legalizeFDIV(MachineInstr &MI, MachineRegisterInfo &MRI, 94 MachineIRBuilder &B) const; 95 bool legalizeFDIV16(MachineInstr &MI, MachineRegisterInfo &MRI, 96 MachineIRBuilder &B) const; 97 bool legalizeFDIV32(MachineInstr &MI, MachineRegisterInfo &MRI, 98 MachineIRBuilder &B) const; 99 bool legalizeFDIV64(MachineInstr &MI, MachineRegisterInfo &MRI, 100 MachineIRBuilder &B) const; 101 bool legalizeFastUnsafeFDIV(MachineInstr &MI, MachineRegisterInfo &MRI, 102 MachineIRBuilder &B) const; 103 bool legalizeFDIVFastIntrin(MachineInstr &MI, MachineRegisterInfo &MRI, 104 MachineIRBuilder &B) const; 105 106 bool legalizeImplicitArgPtr(MachineInstr &MI, MachineRegisterInfo &MRI, 107 MachineIRBuilder &B) const; 108 bool legalizeIsAddrSpace(MachineInstr &MI, MachineRegisterInfo &MRI, 109 MachineIRBuilder &B, unsigned AddrSpace) const; 110 111 std::tuple<Register, unsigned, unsigned> 112 splitBufferOffsets(MachineIRBuilder &B, Register OrigOffset) const; 113 114 Register handleD16VData(MachineIRBuilder &B, MachineRegisterInfo &MRI, 115 Register Reg) const; 116 bool legalizeRawBufferStore(MachineInstr &MI, MachineRegisterInfo &MRI, 117 MachineIRBuilder &B, bool IsFormat) const; 118 bool legalizeRawBufferLoad(MachineInstr &MI, MachineRegisterInfo &MRI, 119 MachineIRBuilder &B, bool IsFormat) const; 120 Register fixStoreSourceType(MachineIRBuilder &B, Register VData, 121 bool IsFormat) const; 122 123 bool legalizeBufferStore(MachineInstr &MI, MachineRegisterInfo &MRI, 124 MachineIRBuilder &B, bool IsTyped, 125 bool IsFormat) const; 126 bool legalizeBufferLoad(MachineInstr &MI, MachineRegisterInfo &MRI, 127 MachineIRBuilder &B, bool IsTyped, 128 bool IsFormat) const; 129 bool legalizeBufferAtomic(MachineInstr &MI, MachineIRBuilder &B, 130 Intrinsic::ID IID) const; 131 132 bool legalizeImageIntrinsic( 133 MachineInstr &MI, MachineIRBuilder &B, 134 GISelChangeObserver &Observer, 135 const AMDGPU::ImageDimIntrinsicInfo *ImageDimIntr) const; 136 137 bool legalizeAtomicIncDec(MachineInstr &MI, MachineIRBuilder &B, 138 bool IsInc) const; 139 140 bool legalizeIntrinsic(MachineInstr &MI, MachineIRBuilder &B, 141 GISelChangeObserver &Observer) const override; 142 }; 143 } // End llvm namespace. 144 #endif 145