| //===- AMDGPULegalizerInfo ---------------------------------------*- C++ -*-==// |
| // |
| // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. |
| // See https://llvm.org/LICENSE.txt for license information. |
| // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception |
| // |
| //===----------------------------------------------------------------------===// |
| /// \file |
| /// This file declares the targeting of the Machinelegalizer class for |
| /// AMDGPU. |
| /// \todo This should be generated by TableGen. |
| //===----------------------------------------------------------------------===// |
| |
| #ifndef LLVM_LIB_TARGET_AMDGPU_AMDGPUMACHINELEGALIZER_H |
| #define LLVM_LIB_TARGET_AMDGPU_AMDGPUMACHINELEGALIZER_H |
| |
| #include "llvm/CodeGen/GlobalISel/LegalizerInfo.h" |
| #include "AMDGPUArgumentUsageInfo.h" |
| #include "SIInstrInfo.h" |
| |
| namespace llvm { |
| |
| class GCNTargetMachine; |
| class LLVMContext; |
| class GCNSubtarget; |
| class MachineIRBuilder; |
| |
| namespace AMDGPU { |
| struct ImageDimIntrinsicInfo; |
| } |
| /// This class provides the information for the target register banks. |
| class AMDGPULegalizerInfo final : public LegalizerInfo { |
| const GCNSubtarget &ST; |
| |
| public: |
| AMDGPULegalizerInfo(const GCNSubtarget &ST, |
| const GCNTargetMachine &TM); |
| |
| bool legalizeCustom(LegalizerHelper &Helper, MachineInstr &MI) const override; |
| |
| Register getSegmentAperture(unsigned AddrSpace, |
| MachineRegisterInfo &MRI, |
| MachineIRBuilder &B) const; |
| |
| bool legalizeAddrSpaceCast(MachineInstr &MI, MachineRegisterInfo &MRI, |
| MachineIRBuilder &B) const; |
| bool legalizeFrint(MachineInstr &MI, MachineRegisterInfo &MRI, |
| MachineIRBuilder &B) const; |
| bool legalizeFceil(MachineInstr &MI, MachineRegisterInfo &MRI, |
| MachineIRBuilder &B) const; |
| bool legalizeFrem(MachineInstr &MI, MachineRegisterInfo &MRI, |
| MachineIRBuilder &B) const; |
| bool legalizeIntrinsicTrunc(MachineInstr &MI, MachineRegisterInfo &MRI, |
| MachineIRBuilder &B) const; |
| bool legalizeITOFP(MachineInstr &MI, MachineRegisterInfo &MRI, |
| MachineIRBuilder &B, bool Signed) const; |
| bool legalizeFPTOI(MachineInstr &MI, MachineRegisterInfo &MRI, |
| MachineIRBuilder &B, bool Signed) const; |
| bool legalizeMinNumMaxNum(LegalizerHelper &Helper, MachineInstr &MI) const; |
| bool legalizeExtractVectorElt(MachineInstr &MI, MachineRegisterInfo &MRI, |
| MachineIRBuilder &B) const; |
| bool legalizeInsertVectorElt(MachineInstr &MI, MachineRegisterInfo &MRI, |
| MachineIRBuilder &B) const; |
| bool legalizeShuffleVector(MachineInstr &MI, MachineRegisterInfo &MRI, |
| MachineIRBuilder &B) const; |
| |
| bool legalizeSinCos(MachineInstr &MI, MachineRegisterInfo &MRI, |
| MachineIRBuilder &B) const; |
| |
| bool buildPCRelGlobalAddress(Register DstReg, LLT PtrTy, MachineIRBuilder &B, |
| const GlobalValue *GV, int64_t Offset, |
| unsigned GAFlags = SIInstrInfo::MO_NONE) const; |
| |
| bool legalizeGlobalValue(MachineInstr &MI, MachineRegisterInfo &MRI, |
| MachineIRBuilder &B) const; |
| bool legalizeLoad(LegalizerHelper &Helper, MachineInstr &MI) const; |
| |
| bool legalizeFMad(MachineInstr &MI, MachineRegisterInfo &MRI, |
| MachineIRBuilder &B) const; |
| |
| bool legalizeAtomicCmpXChg(MachineInstr &MI, MachineRegisterInfo &MRI, |
| MachineIRBuilder &B) const; |
| bool legalizeFlog(MachineInstr &MI, MachineIRBuilder &B, |
| double Log2BaseInverted) const; |
| bool legalizeFExp(MachineInstr &MI, MachineIRBuilder &B) const; |
| bool legalizeFPow(MachineInstr &MI, MachineIRBuilder &B) const; |
| bool legalizeFFloor(MachineInstr &MI, MachineRegisterInfo &MRI, |
| MachineIRBuilder &B) const; |
| |
| bool legalizeBuildVector(MachineInstr &MI, MachineRegisterInfo &MRI, |
| MachineIRBuilder &B) const; |
| bool legalizeCTLZ_CTTZ(MachineInstr &MI, MachineRegisterInfo &MRI, |
| MachineIRBuilder &B) const; |
| |
| bool loadInputValue(Register DstReg, MachineIRBuilder &B, |
| const ArgDescriptor *Arg, |
| const TargetRegisterClass *ArgRC, LLT ArgTy) const; |
| bool loadInputValue(Register DstReg, MachineIRBuilder &B, |
| AMDGPUFunctionArgInfo::PreloadedValue ArgType) const; |
| bool legalizePreloadedArgIntrin( |
| MachineInstr &MI, MachineRegisterInfo &MRI, MachineIRBuilder &B, |
| AMDGPUFunctionArgInfo::PreloadedValue ArgType) const; |
| |
| bool legalizeUnsignedDIV_REM(MachineInstr &MI, MachineRegisterInfo &MRI, |
| MachineIRBuilder &B) const; |
| |
| void legalizeUnsignedDIV_REM32Impl(MachineIRBuilder &B, Register DstDivReg, |
| Register DstRemReg, Register Num, |
| Register Den) const; |
| |
| void legalizeUnsignedDIV_REM64Impl(MachineIRBuilder &B, Register DstDivReg, |
| Register DstRemReg, Register Num, |
| Register Den) const; |
| |
| bool legalizeSignedDIV_REM(MachineInstr &MI, MachineRegisterInfo &MRI, |
| MachineIRBuilder &B) const; |
| |
| bool legalizeFDIV(MachineInstr &MI, MachineRegisterInfo &MRI, |
| MachineIRBuilder &B) const; |
| bool legalizeFDIV16(MachineInstr &MI, MachineRegisterInfo &MRI, |
| MachineIRBuilder &B) const; |
| bool legalizeFDIV32(MachineInstr &MI, MachineRegisterInfo &MRI, |
| MachineIRBuilder &B) const; |
| bool legalizeFDIV64(MachineInstr &MI, MachineRegisterInfo &MRI, |
| MachineIRBuilder &B) const; |
| bool legalizeFastUnsafeFDIV(MachineInstr &MI, MachineRegisterInfo &MRI, |
| MachineIRBuilder &B) const; |
| bool legalizeFastUnsafeFDIV64(MachineInstr &MI, MachineRegisterInfo &MRI, |
| MachineIRBuilder &B) const; |
| bool legalizeFDIVFastIntrin(MachineInstr &MI, MachineRegisterInfo &MRI, |
| MachineIRBuilder &B) const; |
| |
| bool legalizeRsqClampIntrinsic(MachineInstr &MI, MachineRegisterInfo &MRI, |
| MachineIRBuilder &B) const; |
| |
| bool legalizeDSAtomicFPIntrinsic(LegalizerHelper &Helper, |
| MachineInstr &MI, Intrinsic::ID IID) const; |
| |
| bool getImplicitArgPtr(Register DstReg, MachineRegisterInfo &MRI, |
| MachineIRBuilder &B) const; |
| |
| bool legalizeImplicitArgPtr(MachineInstr &MI, MachineRegisterInfo &MRI, |
| MachineIRBuilder &B) const; |
| bool legalizeIsAddrSpace(MachineInstr &MI, MachineRegisterInfo &MRI, |
| MachineIRBuilder &B, unsigned AddrSpace) const; |
| |
| std::pair<Register, unsigned> splitBufferOffsets(MachineIRBuilder &B, |
| Register OrigOffset) const; |
| void updateBufferMMO(MachineMemOperand *MMO, Register VOffset, |
| Register SOffset, unsigned ImmOffset, Register VIndex, |
| MachineRegisterInfo &MRI) const; |
| |
| Register handleD16VData(MachineIRBuilder &B, MachineRegisterInfo &MRI, |
| Register Reg, bool ImageStore = false) const; |
| bool legalizeRawBufferStore(MachineInstr &MI, MachineRegisterInfo &MRI, |
| MachineIRBuilder &B, bool IsFormat) const; |
| bool legalizeRawBufferLoad(MachineInstr &MI, MachineRegisterInfo &MRI, |
| MachineIRBuilder &B, bool IsFormat) const; |
| Register fixStoreSourceType(MachineIRBuilder &B, Register VData, |
| bool IsFormat) const; |
| |
| bool legalizeBufferStore(MachineInstr &MI, MachineRegisterInfo &MRI, |
| MachineIRBuilder &B, bool IsTyped, |
| bool IsFormat) const; |
| bool legalizeBufferLoad(MachineInstr &MI, MachineRegisterInfo &MRI, |
| MachineIRBuilder &B, bool IsFormat, |
| bool IsTyped) const; |
| bool legalizeBufferAtomic(MachineInstr &MI, MachineIRBuilder &B, |
| Intrinsic::ID IID) const; |
| |
| bool legalizeBVHIntrinsic(MachineInstr &MI, MachineIRBuilder &B) const; |
| |
| bool legalizeImageIntrinsic( |
| MachineInstr &MI, MachineIRBuilder &B, |
| GISelChangeObserver &Observer, |
| const AMDGPU::ImageDimIntrinsicInfo *ImageDimIntr) const; |
| |
| bool legalizeSBufferLoad(LegalizerHelper &Helper, MachineInstr &MI) const; |
| |
| bool legalizeAtomicIncDec(MachineInstr &MI, MachineIRBuilder &B, |
| bool IsInc) const; |
| |
| bool legalizeTrapIntrinsic(MachineInstr &MI, MachineRegisterInfo &MRI, |
| MachineIRBuilder &B) const; |
| bool legalizeTrapEndpgm(MachineInstr &MI, MachineRegisterInfo &MRI, |
| MachineIRBuilder &B) const; |
| bool legalizeTrapHsaQueuePtr(MachineInstr &MI, MachineRegisterInfo &MRI, |
| MachineIRBuilder &B) const; |
| bool legalizeTrapHsa(MachineInstr &MI, MachineRegisterInfo &MRI, |
| MachineIRBuilder &B) const; |
| bool legalizeDebugTrapIntrinsic(MachineInstr &MI, MachineRegisterInfo &MRI, |
| MachineIRBuilder &B) const; |
| |
| bool legalizeIntrinsic(LegalizerHelper &Helper, |
| MachineInstr &MI) const override; |
| }; |
| } // End llvm namespace. |
| #endif |