1 //===- lib/Target/AMDGPU/AMDGPUCallLowering.h - Call lowering -*- 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 ///
9 /// \file
10 /// This file describes how to lower LLVM calls to machine code calls.
11 ///
12 //===----------------------------------------------------------------------===//
13 
14 #ifndef LLVM_LIB_TARGET_AMDGPU_AMDGPUCALLLOWERING_H
15 #define LLVM_LIB_TARGET_AMDGPU_AMDGPUCALLLOWERING_H
16 
17 #include "llvm/CodeGen/GlobalISel/CallLowering.h"
18 
19 namespace llvm {
20 
21 class AMDGPUTargetLowering;
22 class MachineInstrBuilder;
23 
24 class AMDGPUCallLowering final : public CallLowering {
25   void lowerParameterPtr(Register DstReg, MachineIRBuilder &B, Type *ParamTy,
26                          uint64_t Offset) const;
27 
28   void lowerParameter(MachineIRBuilder &B, Type *ParamTy, uint64_t Offset,
29                       Align Alignment, Register DstReg) const;
30 
31   /// A function of this type is used to perform value split action.
32   using SplitArgTy = std::function<void(ArrayRef<Register>, Register, LLT, LLT, int)>;
33 
34   void processSplitArgs(MachineIRBuilder &B, const ArgInfo &OrigArgInfo,
35                         const SmallVectorImpl<ArgInfo> &SplitArg,
36                         SmallVectorImpl<ArgInfo> &SplitArgs,
37                         const DataLayout &DL, CallingConv::ID CallConv,
38                         bool IsOutgoing,
39                         SplitArgTy PerformArgSplit) const;
40 
41   bool canLowerReturn(MachineFunction &MF, CallingConv::ID CallConv,
42                       SmallVectorImpl<BaseArgInfo> &Outs,
43                       bool IsVarArg) const override;
44 
45   bool lowerReturnVal(MachineIRBuilder &B, const Value *Val,
46                       ArrayRef<Register> VRegs, MachineInstrBuilder &Ret) const;
47 
48 public:
49   AMDGPUCallLowering(const AMDGPUTargetLowering &TLI);
50 
51   bool lowerReturn(MachineIRBuilder &B, const Value *Val,
52                    ArrayRef<Register> VRegs,
53                    FunctionLoweringInfo &FLI) const override;
54 
55   bool lowerFormalArgumentsKernel(MachineIRBuilder &B, const Function &F,
56                                   ArrayRef<ArrayRef<Register>> VRegs) const;
57 
58   bool lowerFormalArguments(MachineIRBuilder &B, const Function &F,
59                             ArrayRef<ArrayRef<Register>> VRegs,
60                             FunctionLoweringInfo &FLI) const override;
61 
62   bool passSpecialInputs(MachineIRBuilder &MIRBuilder,
63                          CCState &CCInfo,
64                          SmallVectorImpl<std::pair<MCRegister, Register>> &ArgRegs,
65                          CallLoweringInfo &Info) const;
66 
67   bool lowerCall(MachineIRBuilder &MIRBuilder,
68                  CallLoweringInfo &Info) const override;
69 
70   static CCAssignFn *CCAssignFnForCall(CallingConv::ID CC, bool IsVarArg);
71   static CCAssignFn *CCAssignFnForReturn(CallingConv::ID CC, bool IsVarArg);
72 };
73 } // End of namespace llvm;
74 #endif
75