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