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   bool canLowerReturn(MachineFunction &MF, CallingConv::ID CallConv,
32                       SmallVectorImpl<BaseArgInfo> &Outs,
33                       bool IsVarArg) const override;
34 
35   bool lowerReturnVal(MachineIRBuilder &B, const Value *Val,
36                       ArrayRef<Register> VRegs, MachineInstrBuilder &Ret) const;
37 
38 public:
39   AMDGPUCallLowering(const AMDGPUTargetLowering &TLI);
40 
41   bool lowerReturn(MachineIRBuilder &B, const Value *Val,
42                    ArrayRef<Register> VRegs,
43                    FunctionLoweringInfo &FLI) const override;
44 
45   bool lowerFormalArgumentsKernel(MachineIRBuilder &B, const Function &F,
46                                   ArrayRef<ArrayRef<Register>> VRegs) const;
47 
48   bool lowerFormalArguments(MachineIRBuilder &B, const Function &F,
49                             ArrayRef<ArrayRef<Register>> VRegs,
50                             FunctionLoweringInfo &FLI) const override;
51 
52   bool passSpecialInputs(MachineIRBuilder &MIRBuilder,
53                          CCState &CCInfo,
54                          SmallVectorImpl<std::pair<MCRegister, Register>> &ArgRegs,
55                          CallLoweringInfo &Info) const;
56 
57   bool lowerCall(MachineIRBuilder &MIRBuilder,
58                  CallLoweringInfo &Info) const override;
59 
60   static CCAssignFn *CCAssignFnForCall(CallingConv::ID CC, bool IsVarArg);
61   static CCAssignFn *CCAssignFnForReturn(CallingConv::ID CC, bool IsVarArg);
62 };
63 } // End of namespace llvm;
64 #endif
65