xref: /freebsd-src/contrib/llvm-project/llvm/lib/Target/SPIRV/SPIRVCallLowering.h (revision 0fca6ea1d4eea4c934cfff25ac9ee8ad6fe95583)
181ad6265SDimitry Andric //===--- SPIRVCallLowering.h - Call lowering --------------------*- C++ -*-===//
281ad6265SDimitry Andric //
381ad6265SDimitry Andric // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
481ad6265SDimitry Andric // See https://llvm.org/LICENSE.txt for license information.
581ad6265SDimitry Andric // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
681ad6265SDimitry Andric //
781ad6265SDimitry Andric //===----------------------------------------------------------------------===//
881ad6265SDimitry Andric //
981ad6265SDimitry Andric // This file describes how to lower LLVM calls to machine code calls.
1081ad6265SDimitry Andric //
1181ad6265SDimitry Andric //===----------------------------------------------------------------------===//
1281ad6265SDimitry Andric 
1381ad6265SDimitry Andric #ifndef LLVM_LIB_TARGET_SPIRV_SPIRVCALLLOWERING_H
1481ad6265SDimitry Andric #define LLVM_LIB_TARGET_SPIRV_SPIRVCALLLOWERING_H
1581ad6265SDimitry Andric 
16fcaf7f86SDimitry Andric #include "SPIRVGlobalRegistry.h"
1781ad6265SDimitry Andric #include "llvm/CodeGen/GlobalISel/CallLowering.h"
1881ad6265SDimitry Andric 
1981ad6265SDimitry Andric namespace llvm {
2081ad6265SDimitry Andric 
2181ad6265SDimitry Andric class SPIRVGlobalRegistry;
2281ad6265SDimitry Andric class SPIRVTargetLowering;
2381ad6265SDimitry Andric 
2481ad6265SDimitry Andric class SPIRVCallLowering : public CallLowering {
2581ad6265SDimitry Andric private:
2681ad6265SDimitry Andric   // Used to create and assign function, argument, and return type information.
2781ad6265SDimitry Andric   SPIRVGlobalRegistry *GR;
2881ad6265SDimitry Andric 
29*0fca6ea1SDimitry Andric   // Used to postpone producing of indirect function pointer types
30*0fca6ea1SDimitry Andric   // after all indirect calls info is collected
31*0fca6ea1SDimitry Andric   struct SPIRVIndirectCall {
32*0fca6ea1SDimitry Andric     const Type *RetTy = nullptr;
33*0fca6ea1SDimitry Andric     SmallVector<Type *> ArgTys;
34*0fca6ea1SDimitry Andric     SmallVector<Register> ArgRegs;
35*0fca6ea1SDimitry Andric     Register Callee;
36*0fca6ea1SDimitry Andric   };
37*0fca6ea1SDimitry Andric   void produceIndirectPtrTypes(MachineIRBuilder &MIRBuilder) const;
38*0fca6ea1SDimitry Andric   mutable SmallVector<SPIRVIndirectCall> IndirectCalls;
39*0fca6ea1SDimitry Andric 
4081ad6265SDimitry Andric public:
41fcaf7f86SDimitry Andric   SPIRVCallLowering(const SPIRVTargetLowering &TLI, SPIRVGlobalRegistry *GR);
4281ad6265SDimitry Andric 
4381ad6265SDimitry Andric   // Built OpReturn or OpReturnValue.
4481ad6265SDimitry Andric   bool lowerReturn(MachineIRBuilder &MIRBuiler, const Value *Val,
4581ad6265SDimitry Andric                    ArrayRef<Register> VRegs, FunctionLoweringInfo &FLI,
4681ad6265SDimitry Andric                    Register SwiftErrorVReg) const override;
4781ad6265SDimitry Andric 
4881ad6265SDimitry Andric   // Build OpFunction, OpFunctionParameter, and any EntryPoint or Linkage data.
4981ad6265SDimitry Andric   bool lowerFormalArguments(MachineIRBuilder &MIRBuilder, const Function &F,
5081ad6265SDimitry Andric                             ArrayRef<ArrayRef<Register>> VRegs,
5181ad6265SDimitry Andric                             FunctionLoweringInfo &FLI) const override;
5281ad6265SDimitry Andric 
5381ad6265SDimitry Andric   // Build OpCall, or replace with a builtin function.
5481ad6265SDimitry Andric   bool lowerCall(MachineIRBuilder &MIRBuilder,
5581ad6265SDimitry Andric                  CallLoweringInfo &Info) const override;
5681ad6265SDimitry Andric };
5781ad6265SDimitry Andric } // end namespace llvm
5881ad6265SDimitry Andric 
5981ad6265SDimitry Andric #endif // LLVM_LIB_TARGET_SPIRV_SPIRVCALLLOWERING_H
60