1 //===-- BPFISelLowering.h - BPF DAG Lowering Interface ----------*- C++ -*-===//
2 //
3 //                     The LLVM Compiler Infrastructure
4 //
5 // This file is distributed under the University of Illinois Open Source
6 // License. See LICENSE.TXT for details.
7 //
8 //===----------------------------------------------------------------------===//
9 //
10 // This file defines the interfaces that BPF uses to lower LLVM code into a
11 // selection DAG.
12 //
13 //===----------------------------------------------------------------------===//
14 
15 #ifndef LLVM_LIB_TARGET_BPF_BPFISELLOWERING_H
16 #define LLVM_LIB_TARGET_BPF_BPFISELLOWERING_H
17 
18 #include "BPF.h"
19 #include "llvm/CodeGen/SelectionDAG.h"
20 #include "llvm/Target/TargetLowering.h"
21 
22 namespace llvm {
23 class BPFSubtarget;
24 namespace BPFISD {
25 enum NodeType : unsigned {
26   FIRST_NUMBER = ISD::BUILTIN_OP_END,
27   RET_FLAG,
28   CALL,
29   SELECT_CC,
30   BR_CC,
31   Wrapper
32 };
33 }
34 
35 class BPFTargetLowering : public TargetLowering {
36 public:
37   explicit BPFTargetLowering(const TargetMachine &TM, const BPFSubtarget &STI);
38 
39   // Provide custom lowering hooks for some operations.
40   SDValue LowerOperation(SDValue Op, SelectionDAG &DAG) const override;
41 
42   // This method returns the name of a target specific DAG node.
43   const char *getTargetNodeName(unsigned Opcode) const override;
44 
45   MachineBasicBlock *
46   EmitInstrWithCustomInserter(MachineInstr *MI,
47                               MachineBasicBlock *BB) const override;
48 
49 private:
50   SDValue LowerBR_CC(SDValue Op, SelectionDAG &DAG) const;
51   SDValue LowerSELECT_CC(SDValue Op, SelectionDAG &DAG) const;
52   SDValue LowerGlobalAddress(SDValue Op, SelectionDAG &DAG) const;
53 
54   // Lower the result values of a call, copying them out of physregs into vregs
55   SDValue LowerCallResult(SDValue Chain, SDValue InFlag,
56                           CallingConv::ID CallConv, bool IsVarArg,
57                           const SmallVectorImpl<ISD::InputArg> &Ins, SDLoc DL,
58                           SelectionDAG &DAG,
59                           SmallVectorImpl<SDValue> &InVals) const;
60 
61   // Lower a call into CALLSEQ_START - BPFISD:CALL - CALLSEQ_END chain
62   SDValue LowerCall(TargetLowering::CallLoweringInfo &CLI,
63                     SmallVectorImpl<SDValue> &InVals) const override;
64 
65   // Lower incoming arguments, copy physregs into vregs
66   SDValue LowerFormalArguments(SDValue Chain, CallingConv::ID CallConv,
67                                bool IsVarArg,
68                                const SmallVectorImpl<ISD::InputArg> &Ins,
69                                SDLoc DL, SelectionDAG &DAG,
70                                SmallVectorImpl<SDValue> &InVals) const override;
71 
72   SDValue LowerReturn(SDValue Chain, CallingConv::ID CallConv, bool IsVarArg,
73                       const SmallVectorImpl<ISD::OutputArg> &Outs,
74                       const SmallVectorImpl<SDValue> &OutVals, SDLoc DL,
75                       SelectionDAG &DAG) const override;
76 
getOptimalMemOpType(uint64_t Size,unsigned DstAlign,unsigned SrcAlign,bool IsMemset,bool ZeroMemset,bool MemcpyStrSrc,MachineFunction & MF)77   EVT getOptimalMemOpType(uint64_t Size, unsigned DstAlign, unsigned SrcAlign,
78                           bool IsMemset, bool ZeroMemset, bool MemcpyStrSrc,
79                           MachineFunction &MF) const override {
80     return Size >= 8 ? MVT::i64 : MVT::i32;
81   }
82 
shouldConvertConstantLoadToIntImm(const APInt & Imm,Type * Ty)83   bool shouldConvertConstantLoadToIntImm(const APInt &Imm,
84                                          Type *Ty) const override {
85     return true;
86   }
87 };
88 }
89 
90 #endif
91