xref: /llvm-project/llvm/lib/Target/LoongArch/LoongArchISelDAGToDAG.h (revision 84220eccb6ce5413f9782590b3877bd689c9b43c)
1 //=- LoongArchISelDAGToDAG.h - A dag to dag inst selector for LoongArch ---===//
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 // This file defines an instruction selector for the LoongArch target.
10 //
11 //===----------------------------------------------------------------------===//
12 
13 #ifndef LLVM_LIB_TARGET_LOONGARCH_LOONGARCHISELDAGTODAG_H
14 #define LLVM_LIB_TARGET_LOONGARCH_LOONGARCHISELDAGTODAG_H
15 
16 #include "LoongArch.h"
17 #include "LoongArchTargetMachine.h"
18 #include "llvm/CodeGen/SelectionDAGISel.h"
19 
20 // LoongArch-specific code to select LoongArch machine instructions for
21 // SelectionDAG operations.
22 namespace llvm {
23 class LoongArchDAGToDAGISel : public SelectionDAGISel {
24   const LoongArchSubtarget *Subtarget = nullptr;
25 
26 public:
27   LoongArchDAGToDAGISel() = delete;
28 
29   explicit LoongArchDAGToDAGISel(LoongArchTargetMachine &TM)
30       : SelectionDAGISel(TM) {}
31 
32   bool runOnMachineFunction(MachineFunction &MF) override {
33     Subtarget = &MF.getSubtarget<LoongArchSubtarget>();
34     return SelectionDAGISel::runOnMachineFunction(MF);
35   }
36 
37   void Select(SDNode *Node) override;
38 
39   bool SelectInlineAsmMemoryOperand(const SDValue &Op,
40                                     InlineAsm::ConstraintCode ConstraintID,
41                                     std::vector<SDValue> &OutOps) override;
42 
43   bool SelectBaseAddr(SDValue Addr, SDValue &Base);
44   bool SelectAddrConstant(SDValue Addr, SDValue &Base, SDValue &Offset);
45   bool selectNonFIBaseAddr(SDValue Addr, SDValue &Base);
46   bool SelectAddrRegImm12(SDValue Addr, SDValue &Base, SDValue &Offset);
47 
48   bool selectShiftMask(SDValue N, unsigned ShiftWidth, SDValue &ShAmt);
49   bool selectShiftMaskGRLen(SDValue N, SDValue &ShAmt) {
50     return selectShiftMask(N, Subtarget->getGRLen(), ShAmt);
51   }
52   bool selectShiftMask32(SDValue N, SDValue &ShAmt) {
53     return selectShiftMask(N, 32, ShAmt);
54   }
55 
56   bool selectSExti32(SDValue N, SDValue &Val);
57   bool selectZExti32(SDValue N, SDValue &Val);
58 
59   bool selectVSplat(SDNode *N, APInt &Imm, unsigned MinSizeInBits) const;
60 
61   template <unsigned ImmSize, bool IsSigned = false>
62   bool selectVSplatImm(SDValue N, SDValue &SplatVal);
63 
64   bool selectVSplatUimmInvPow2(SDValue N, SDValue &SplatImm) const;
65   bool selectVSplatUimmPow2(SDValue N, SDValue &SplatImm) const;
66 
67 // Include the pieces autogenerated from the target description.
68 #include "LoongArchGenDAGISel.inc"
69 };
70 
71 class LoongArchDAGToDAGISelLegacy : public SelectionDAGISelLegacy {
72 public:
73   static char ID;
74   explicit LoongArchDAGToDAGISelLegacy(LoongArchTargetMachine &TM);
75 };
76 
77 } // end namespace llvm
78 
79 #endif // LLVM_LIB_TARGET_LOONGARCH_LOONGARCHISELDAGTODAG_H
80