lib/Target/BPF/BPFISelLowering.h

   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
  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
  83   bool shouldConvertConstantLoadToIntImm(const APInt &Imm,
  84                                          Type *Ty) const override {
  85     return true;
  86   }
  87 };
  88 }
  89
  90 #endif