xref: /freebsd/contrib/llvm-project/llvm/lib/Target/LoongArch/LoongArchISelLowering.cpp (revision 1db9f3b21e39176dd5b67cf8ac378633b172463e)
181ad6265SDimitry Andric //=- LoongArchISelLowering.cpp - LoongArch DAG Lowering Implementation  ---===//
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 defines the interfaces that LoongArch uses to lower LLVM code into
1081ad6265SDimitry Andric // a selection DAG.
1181ad6265SDimitry Andric //
1281ad6265SDimitry Andric //===----------------------------------------------------------------------===//
1381ad6265SDimitry Andric 
1481ad6265SDimitry Andric #include "LoongArchISelLowering.h"
1581ad6265SDimitry Andric #include "LoongArch.h"
1681ad6265SDimitry Andric #include "LoongArchMachineFunctionInfo.h"
1781ad6265SDimitry Andric #include "LoongArchRegisterInfo.h"
1881ad6265SDimitry Andric #include "LoongArchSubtarget.h"
1981ad6265SDimitry Andric #include "LoongArchTargetMachine.h"
20bdd1243dSDimitry Andric #include "MCTargetDesc/LoongArchBaseInfo.h"
21753f127fSDimitry Andric #include "MCTargetDesc/LoongArchMCTargetDesc.h"
2281ad6265SDimitry Andric #include "llvm/ADT/Statistic.h"
2306c3fb27SDimitry Andric #include "llvm/ADT/StringExtras.h"
2481ad6265SDimitry Andric #include "llvm/CodeGen/ISDOpcodes.h"
25bdd1243dSDimitry Andric #include "llvm/CodeGen/RuntimeLibcalls.h"
2606c3fb27SDimitry Andric #include "llvm/CodeGen/SelectionDAGNodes.h"
27bdd1243dSDimitry Andric #include "llvm/IR/IRBuilder.h"
28bdd1243dSDimitry Andric #include "llvm/IR/IntrinsicsLoongArch.h"
2906c3fb27SDimitry Andric #include "llvm/Support/CodeGen.h"
3081ad6265SDimitry Andric #include "llvm/Support/Debug.h"
3106c3fb27SDimitry Andric #include "llvm/Support/ErrorHandling.h"
32753f127fSDimitry Andric #include "llvm/Support/KnownBits.h"
33bdd1243dSDimitry Andric #include "llvm/Support/MathExtras.h"
3481ad6265SDimitry Andric 
3581ad6265SDimitry Andric using namespace llvm;
3681ad6265SDimitry Andric 
3781ad6265SDimitry Andric #define DEBUG_TYPE "loongarch-isel-lowering"
3881ad6265SDimitry Andric 
39bdd1243dSDimitry Andric STATISTIC(NumTailCalls, "Number of tail calls");
40bdd1243dSDimitry Andric 
4106c3fb27SDimitry Andric static cl::opt<bool> ZeroDivCheck("loongarch-check-zero-division", cl::Hidden,
42753f127fSDimitry Andric                                   cl::desc("Trap on integer division by zero."),
43753f127fSDimitry Andric                                   cl::init(false));
44753f127fSDimitry Andric 
4581ad6265SDimitry Andric LoongArchTargetLowering::LoongArchTargetLowering(const TargetMachine &TM,
4681ad6265SDimitry Andric                                                  const LoongArchSubtarget &STI)
4781ad6265SDimitry Andric     : TargetLowering(TM), Subtarget(STI) {
4881ad6265SDimitry Andric 
4981ad6265SDimitry Andric   MVT GRLenVT = Subtarget.getGRLenVT();
505f757f3fSDimitry Andric 
5181ad6265SDimitry Andric   // Set up the register classes.
525f757f3fSDimitry Andric 
5381ad6265SDimitry Andric   addRegisterClass(GRLenVT, &LoongArch::GPRRegClass);
5481ad6265SDimitry Andric   if (Subtarget.hasBasicF())
5581ad6265SDimitry Andric     addRegisterClass(MVT::f32, &LoongArch::FPR32RegClass);
5681ad6265SDimitry Andric   if (Subtarget.hasBasicD())
5781ad6265SDimitry Andric     addRegisterClass(MVT::f64, &LoongArch::FPR64RegClass);
585f757f3fSDimitry Andric 
595f757f3fSDimitry Andric   static const MVT::SimpleValueType LSXVTs[] = {
605f757f3fSDimitry Andric       MVT::v16i8, MVT::v8i16, MVT::v4i32, MVT::v2i64, MVT::v4f32, MVT::v2f64};
615f757f3fSDimitry Andric   static const MVT::SimpleValueType LASXVTs[] = {
625f757f3fSDimitry Andric       MVT::v32i8, MVT::v16i16, MVT::v8i32, MVT::v4i64, MVT::v8f32, MVT::v4f64};
635f757f3fSDimitry Andric 
6406c3fb27SDimitry Andric   if (Subtarget.hasExtLSX())
655f757f3fSDimitry Andric     for (MVT VT : LSXVTs)
6606c3fb27SDimitry Andric       addRegisterClass(VT, &LoongArch::LSX128RegClass);
675f757f3fSDimitry Andric 
6806c3fb27SDimitry Andric   if (Subtarget.hasExtLASX())
695f757f3fSDimitry Andric     for (MVT VT : LASXVTs)
7006c3fb27SDimitry Andric       addRegisterClass(VT, &LoongArch::LASX256RegClass);
7181ad6265SDimitry Andric 
725f757f3fSDimitry Andric   // Set operations for LA32 and LA64.
735f757f3fSDimitry Andric 
74753f127fSDimitry Andric   setLoadExtAction({ISD::EXTLOAD, ISD::SEXTLOAD, ISD::ZEXTLOAD}, GRLenVT,
75753f127fSDimitry Andric                    MVT::i1, Promote);
76753f127fSDimitry Andric 
7781ad6265SDimitry Andric   setOperationAction(ISD::SHL_PARTS, GRLenVT, Custom);
7881ad6265SDimitry Andric   setOperationAction(ISD::SRA_PARTS, GRLenVT, Custom);
7981ad6265SDimitry Andric   setOperationAction(ISD::SRL_PARTS, GRLenVT, Custom);
80753f127fSDimitry Andric   setOperationAction(ISD::FP_TO_SINT, GRLenVT, Custom);
81bdd1243dSDimitry Andric   setOperationAction(ISD::ROTL, GRLenVT, Expand);
82bdd1243dSDimitry Andric   setOperationAction(ISD::CTPOP, GRLenVT, Expand);
83753f127fSDimitry Andric 
84bdd1243dSDimitry Andric   setOperationAction({ISD::GlobalAddress, ISD::BlockAddress, ISD::ConstantPool,
855f757f3fSDimitry Andric                       ISD::JumpTable, ISD::GlobalTLSAddress},
86bdd1243dSDimitry Andric                      GRLenVT, Custom);
87bdd1243dSDimitry Andric 
885f757f3fSDimitry Andric   setOperationAction(ISD::EH_DWARF_CFA, GRLenVT, Custom);
89bdd1243dSDimitry Andric 
90bdd1243dSDimitry Andric   setOperationAction(ISD::DYNAMIC_STACKALLOC, GRLenVT, Expand);
91bdd1243dSDimitry Andric   setOperationAction({ISD::STACKSAVE, ISD::STACKRESTORE}, MVT::Other, Expand);
92bdd1243dSDimitry Andric   setOperationAction(ISD::VASTART, MVT::Other, Custom);
93bdd1243dSDimitry Andric   setOperationAction({ISD::VAARG, ISD::VACOPY, ISD::VAEND}, MVT::Other, Expand);
9481ad6265SDimitry Andric 
955f757f3fSDimitry Andric   setOperationAction(ISD::DEBUGTRAP, MVT::Other, Legal);
965f757f3fSDimitry Andric   setOperationAction(ISD::TRAP, MVT::Other, Legal);
975f757f3fSDimitry Andric 
985f757f3fSDimitry Andric   setOperationAction(ISD::INTRINSIC_VOID, MVT::Other, Custom);
995f757f3fSDimitry Andric   setOperationAction(ISD::INTRINSIC_W_CHAIN, MVT::Other, Custom);
1005f757f3fSDimitry Andric   setOperationAction(ISD::INTRINSIC_WO_CHAIN, MVT::Other, Custom);
1015f757f3fSDimitry Andric 
1025f757f3fSDimitry Andric   // Expand bitreverse.i16 with native-width bitrev and shift for now, before
1035f757f3fSDimitry Andric   // we get to know which of sll and revb.2h is faster.
1045f757f3fSDimitry Andric   setOperationAction(ISD::BITREVERSE, MVT::i8, Custom);
1055f757f3fSDimitry Andric   setOperationAction(ISD::BITREVERSE, GRLenVT, Legal);
1065f757f3fSDimitry Andric 
1075f757f3fSDimitry Andric   // LA32 does not have REVB.2W and REVB.D due to the 64-bit operands, and
1085f757f3fSDimitry Andric   // the narrower REVB.W does not exist. But LA32 does have REVB.2H, so i16
1095f757f3fSDimitry Andric   // and i32 could still be byte-swapped relatively cheaply.
1105f757f3fSDimitry Andric   setOperationAction(ISD::BSWAP, MVT::i16, Custom);
1115f757f3fSDimitry Andric 
1125f757f3fSDimitry Andric   setOperationAction(ISD::BR_JT, MVT::Other, Expand);
1135f757f3fSDimitry Andric   setOperationAction(ISD::BR_CC, GRLenVT, Expand);
1145f757f3fSDimitry Andric   setOperationAction(ISD::SELECT_CC, GRLenVT, Expand);
1155f757f3fSDimitry Andric   setOperationAction(ISD::SIGN_EXTEND_INREG, MVT::i1, Expand);
1165f757f3fSDimitry Andric   setOperationAction({ISD::SMUL_LOHI, ISD::UMUL_LOHI}, GRLenVT, Expand);
1175f757f3fSDimitry Andric 
1185f757f3fSDimitry Andric   setOperationAction(ISD::FP_TO_UINT, GRLenVT, Custom);
1195f757f3fSDimitry Andric   setOperationAction(ISD::UINT_TO_FP, GRLenVT, Expand);
1205f757f3fSDimitry Andric 
1215f757f3fSDimitry Andric   // Set operations for LA64 only.
1225f757f3fSDimitry Andric 
12381ad6265SDimitry Andric   if (Subtarget.is64Bit()) {
12481ad6265SDimitry Andric     setOperationAction(ISD::SHL, MVT::i32, Custom);
12581ad6265SDimitry Andric     setOperationAction(ISD::SRA, MVT::i32, Custom);
12681ad6265SDimitry Andric     setOperationAction(ISD::SRL, MVT::i32, Custom);
127753f127fSDimitry Andric     setOperationAction(ISD::FP_TO_SINT, MVT::i32, Custom);
128753f127fSDimitry Andric     setOperationAction(ISD::BITCAST, MVT::i32, Custom);
129bdd1243dSDimitry Andric     setOperationAction(ISD::ROTR, MVT::i32, Custom);
130bdd1243dSDimitry Andric     setOperationAction(ISD::ROTL, MVT::i32, Custom);
131bdd1243dSDimitry Andric     setOperationAction(ISD::CTTZ, MVT::i32, Custom);
132bdd1243dSDimitry Andric     setOperationAction(ISD::CTLZ, MVT::i32, Custom);
1335f757f3fSDimitry Andric     setOperationAction(ISD::EH_DWARF_CFA, MVT::i32, Custom);
134bdd1243dSDimitry Andric     setOperationAction(ISD::READ_REGISTER, MVT::i32, Custom);
135bdd1243dSDimitry Andric     setOperationAction(ISD::WRITE_REGISTER, MVT::i32, Custom);
1365f757f3fSDimitry Andric     setOperationAction(ISD::INTRINSIC_VOID, MVT::i32, Custom);
1375f757f3fSDimitry Andric     setOperationAction(ISD::INTRINSIC_WO_CHAIN, MVT::i32, Custom);
1385f757f3fSDimitry Andric     setOperationAction(ISD::INTRINSIC_W_CHAIN, MVT::i32, Custom);
13981ad6265SDimitry Andric 
1405f757f3fSDimitry Andric     setOperationAction(ISD::BITREVERSE, MVT::i32, Custom);
141bdd1243dSDimitry Andric     setOperationAction(ISD::BSWAP, MVT::i32, Custom);
142bdd1243dSDimitry Andric   }
143bdd1243dSDimitry Andric 
1445f757f3fSDimitry Andric   // Set operations for LA32 only.
1455f757f3fSDimitry Andric 
1465f757f3fSDimitry Andric   if (!Subtarget.is64Bit()) {
147bdd1243dSDimitry Andric     setOperationAction(ISD::READ_REGISTER, MVT::i64, Custom);
148bdd1243dSDimitry Andric     setOperationAction(ISD::WRITE_REGISTER, MVT::i64, Custom);
149bdd1243dSDimitry Andric     setOperationAction(ISD::INTRINSIC_VOID, MVT::i64, Custom);
1505f757f3fSDimitry Andric     setOperationAction(ISD::INTRINSIC_WO_CHAIN, MVT::i64, Custom);
1515f757f3fSDimitry Andric     setOperationAction(ISD::INTRINSIC_W_CHAIN, MVT::i64, Custom);
1525f757f3fSDimitry Andric 
1535f757f3fSDimitry Andric     // Set libcalls.
1545f757f3fSDimitry Andric     setLibcallName(RTLIB::MUL_I128, nullptr);
1555f757f3fSDimitry Andric     // The MULO libcall is not part of libgcc, only compiler-rt.
1565f757f3fSDimitry Andric     setLibcallName(RTLIB::MULO_I64, nullptr);
157bdd1243dSDimitry Andric   }
158bdd1243dSDimitry Andric 
1595f757f3fSDimitry Andric   // The MULO libcall is not part of libgcc, only compiler-rt.
1605f757f3fSDimitry Andric   setLibcallName(RTLIB::MULO_I128, nullptr);
1615f757f3fSDimitry Andric 
1625f757f3fSDimitry Andric   setOperationAction(ISD::ATOMIC_FENCE, MVT::Other, Custom);
1635f757f3fSDimitry Andric 
164bdd1243dSDimitry Andric   static const ISD::CondCode FPCCToExpand[] = {
165bdd1243dSDimitry Andric       ISD::SETOGT, ISD::SETOGE, ISD::SETUGT, ISD::SETUGE,
166bdd1243dSDimitry Andric       ISD::SETGE,  ISD::SETNE,  ISD::SETGT};
16781ad6265SDimitry Andric 
1685f757f3fSDimitry Andric   // Set operations for 'F' feature.
1695f757f3fSDimitry Andric 
17081ad6265SDimitry Andric   if (Subtarget.hasBasicF()) {
17181ad6265SDimitry Andric     setCondCodeAction(FPCCToExpand, MVT::f32, Expand);
1725f757f3fSDimitry Andric 
17381ad6265SDimitry Andric     setOperationAction(ISD::SELECT_CC, MVT::f32, Expand);
174bdd1243dSDimitry Andric     setOperationAction(ISD::BR_CC, MVT::f32, Expand);
175bdd1243dSDimitry Andric     setOperationAction(ISD::FMA, MVT::f32, Legal);
176bdd1243dSDimitry Andric     setOperationAction(ISD::FMINNUM_IEEE, MVT::f32, Legal);
177bdd1243dSDimitry Andric     setOperationAction(ISD::FMAXNUM_IEEE, MVT::f32, Legal);
178bdd1243dSDimitry Andric     setOperationAction(ISD::STRICT_FSETCCS, MVT::f32, Legal);
179bdd1243dSDimitry Andric     setOperationAction(ISD::STRICT_FSETCC, MVT::f32, Legal);
1805f757f3fSDimitry Andric     setOperationAction(ISD::IS_FPCLASS, MVT::f32, Legal);
181bdd1243dSDimitry Andric     setOperationAction(ISD::FSIN, MVT::f32, Expand);
182bdd1243dSDimitry Andric     setOperationAction(ISD::FCOS, MVT::f32, Expand);
183bdd1243dSDimitry Andric     setOperationAction(ISD::FSINCOS, MVT::f32, Expand);
184bdd1243dSDimitry Andric     setOperationAction(ISD::FPOW, MVT::f32, Expand);
185bdd1243dSDimitry Andric     setOperationAction(ISD::FREM, MVT::f32, Expand);
1865f757f3fSDimitry Andric 
1875f757f3fSDimitry Andric     if (Subtarget.is64Bit())
1885f757f3fSDimitry Andric       setOperationAction(ISD::FRINT, MVT::f32, Legal);
1895f757f3fSDimitry Andric 
1905f757f3fSDimitry Andric     if (!Subtarget.hasBasicD()) {
1915f757f3fSDimitry Andric       setOperationAction(ISD::FP_TO_UINT, MVT::i32, Custom);
1925f757f3fSDimitry Andric       if (Subtarget.is64Bit()) {
1935f757f3fSDimitry Andric         setOperationAction(ISD::SINT_TO_FP, MVT::i64, Custom);
1945f757f3fSDimitry Andric         setOperationAction(ISD::UINT_TO_FP, MVT::i64, Custom);
19581ad6265SDimitry Andric       }
1965f757f3fSDimitry Andric     }
1975f757f3fSDimitry Andric   }
1985f757f3fSDimitry Andric 
1995f757f3fSDimitry Andric   // Set operations for 'D' feature.
2005f757f3fSDimitry Andric 
20181ad6265SDimitry Andric   if (Subtarget.hasBasicD()) {
2025f757f3fSDimitry Andric     setLoadExtAction(ISD::EXTLOAD, MVT::f64, MVT::f32, Expand);
2035f757f3fSDimitry Andric     setTruncStoreAction(MVT::f64, MVT::f32, Expand);
20481ad6265SDimitry Andric     setCondCodeAction(FPCCToExpand, MVT::f64, Expand);
2055f757f3fSDimitry Andric 
20681ad6265SDimitry Andric     setOperationAction(ISD::SELECT_CC, MVT::f64, Expand);
207bdd1243dSDimitry Andric     setOperationAction(ISD::BR_CC, MVT::f64, Expand);
208bdd1243dSDimitry Andric     setOperationAction(ISD::STRICT_FSETCCS, MVT::f64, Legal);
209bdd1243dSDimitry Andric     setOperationAction(ISD::STRICT_FSETCC, MVT::f64, Legal);
210bdd1243dSDimitry Andric     setOperationAction(ISD::FMA, MVT::f64, Legal);
211bdd1243dSDimitry Andric     setOperationAction(ISD::FMINNUM_IEEE, MVT::f64, Legal);
212bdd1243dSDimitry Andric     setOperationAction(ISD::FMAXNUM_IEEE, MVT::f64, Legal);
2135f757f3fSDimitry Andric     setOperationAction(ISD::IS_FPCLASS, MVT::f64, Legal);
214bdd1243dSDimitry Andric     setOperationAction(ISD::FSIN, MVT::f64, Expand);
215bdd1243dSDimitry Andric     setOperationAction(ISD::FCOS, MVT::f64, Expand);
216bdd1243dSDimitry Andric     setOperationAction(ISD::FSINCOS, MVT::f64, Expand);
217bdd1243dSDimitry Andric     setOperationAction(ISD::FPOW, MVT::f64, Expand);
218bdd1243dSDimitry Andric     setOperationAction(ISD::FREM, MVT::f64, Expand);
2195f757f3fSDimitry Andric 
2205f757f3fSDimitry Andric     if (Subtarget.is64Bit())
2215f757f3fSDimitry Andric       setOperationAction(ISD::FRINT, MVT::f64, Legal);
22281ad6265SDimitry Andric   }
22381ad6265SDimitry Andric 
2245f757f3fSDimitry Andric   // Set operations for 'LSX' feature.
225bdd1243dSDimitry Andric 
2265f757f3fSDimitry Andric   if (Subtarget.hasExtLSX()) {
2275f757f3fSDimitry Andric     for (MVT VT : MVT::fixedlen_vector_valuetypes()) {
2285f757f3fSDimitry Andric       // Expand all truncating stores and extending loads.
2295f757f3fSDimitry Andric       for (MVT InnerVT : MVT::fixedlen_vector_valuetypes()) {
2305f757f3fSDimitry Andric         setTruncStoreAction(VT, InnerVT, Expand);
2315f757f3fSDimitry Andric         setLoadExtAction(ISD::SEXTLOAD, VT, InnerVT, Expand);
2325f757f3fSDimitry Andric         setLoadExtAction(ISD::ZEXTLOAD, VT, InnerVT, Expand);
2335f757f3fSDimitry Andric         setLoadExtAction(ISD::EXTLOAD, VT, InnerVT, Expand);
234bdd1243dSDimitry Andric       }
2355f757f3fSDimitry Andric       // By default everything must be expanded. Then we will selectively turn
2365f757f3fSDimitry Andric       // on ones that can be effectively codegen'd.
2375f757f3fSDimitry Andric       for (unsigned Op = 0; Op < ISD::BUILTIN_OP_END; ++Op)
2385f757f3fSDimitry Andric         setOperationAction(Op, VT, Expand);
2395f757f3fSDimitry Andric     }
2405f757f3fSDimitry Andric 
2415f757f3fSDimitry Andric     for (MVT VT : LSXVTs) {
2425f757f3fSDimitry Andric       setOperationAction({ISD::LOAD, ISD::STORE}, VT, Legal);
2435f757f3fSDimitry Andric       setOperationAction(ISD::BITCAST, VT, Legal);
2445f757f3fSDimitry Andric       setOperationAction(ISD::UNDEF, VT, Legal);
2455f757f3fSDimitry Andric 
2465f757f3fSDimitry Andric       setOperationAction(ISD::INSERT_VECTOR_ELT, VT, Custom);
2475f757f3fSDimitry Andric       setOperationAction(ISD::EXTRACT_VECTOR_ELT, VT, Legal);
2485f757f3fSDimitry Andric       setOperationAction(ISD::BUILD_VECTOR, VT, Custom);
2495f757f3fSDimitry Andric 
2505f757f3fSDimitry Andric       setOperationAction(ISD::SETCC, VT, Legal);
2515f757f3fSDimitry Andric       setOperationAction(ISD::VSELECT, VT, Legal);
2525f757f3fSDimitry Andric     }
2535f757f3fSDimitry Andric     for (MVT VT : {MVT::v16i8, MVT::v8i16, MVT::v4i32, MVT::v2i64}) {
2545f757f3fSDimitry Andric       setOperationAction(ISD::VECTOR_SHUFFLE, VT, Custom);
2555f757f3fSDimitry Andric       setOperationAction({ISD::ADD, ISD::SUB}, VT, Legal);
2565f757f3fSDimitry Andric       setOperationAction({ISD::UMAX, ISD::UMIN, ISD::SMAX, ISD::SMIN}, VT,
2575f757f3fSDimitry Andric                          Legal);
2585f757f3fSDimitry Andric       setOperationAction({ISD::MUL, ISD::SDIV, ISD::SREM, ISD::UDIV, ISD::UREM},
2595f757f3fSDimitry Andric                          VT, Legal);
2605f757f3fSDimitry Andric       setOperationAction({ISD::AND, ISD::OR, ISD::XOR}, VT, Legal);
2615f757f3fSDimitry Andric       setOperationAction({ISD::SHL, ISD::SRA, ISD::SRL}, VT, Legal);
2625f757f3fSDimitry Andric       setOperationAction({ISD::CTPOP, ISD::CTLZ}, VT, Legal);
2635f757f3fSDimitry Andric       setOperationAction({ISD::MULHS, ISD::MULHU}, VT, Legal);
2645f757f3fSDimitry Andric       setCondCodeAction(
2655f757f3fSDimitry Andric           {ISD::SETNE, ISD::SETGE, ISD::SETGT, ISD::SETUGE, ISD::SETUGT}, VT,
2665f757f3fSDimitry Andric           Expand);
2675f757f3fSDimitry Andric     }
2685f757f3fSDimitry Andric     for (MVT VT : {MVT::v4f32, MVT::v2f64}) {
2695f757f3fSDimitry Andric       setOperationAction({ISD::FADD, ISD::FSUB}, VT, Legal);
2705f757f3fSDimitry Andric       setOperationAction({ISD::FMUL, ISD::FDIV}, VT, Legal);
2715f757f3fSDimitry Andric       setOperationAction(ISD::FMA, VT, Legal);
2725f757f3fSDimitry Andric       setOperationAction(ISD::FSQRT, VT, Legal);
2735f757f3fSDimitry Andric       setOperationAction(ISD::FNEG, VT, Legal);
2745f757f3fSDimitry Andric       setCondCodeAction({ISD::SETGE, ISD::SETGT, ISD::SETOGE, ISD::SETOGT,
2755f757f3fSDimitry Andric                          ISD::SETUGE, ISD::SETUGT},
2765f757f3fSDimitry Andric                         VT, Expand);
2775f757f3fSDimitry Andric     }
2785f757f3fSDimitry Andric   }
2795f757f3fSDimitry Andric 
2805f757f3fSDimitry Andric   // Set operations for 'LASX' feature.
2815f757f3fSDimitry Andric 
2825f757f3fSDimitry Andric   if (Subtarget.hasExtLASX()) {
2835f757f3fSDimitry Andric     for (MVT VT : LASXVTs) {
2845f757f3fSDimitry Andric       setOperationAction({ISD::LOAD, ISD::STORE}, VT, Legal);
2855f757f3fSDimitry Andric       setOperationAction(ISD::BITCAST, VT, Legal);
2865f757f3fSDimitry Andric       setOperationAction(ISD::UNDEF, VT, Legal);
2875f757f3fSDimitry Andric 
2885f757f3fSDimitry Andric       setOperationAction(ISD::INSERT_VECTOR_ELT, VT, Custom);
289647cbc5dSDimitry Andric       setOperationAction(ISD::EXTRACT_VECTOR_ELT, VT, Custom);
2905f757f3fSDimitry Andric       setOperationAction(ISD::BUILD_VECTOR, VT, Custom);
2915f757f3fSDimitry Andric 
2925f757f3fSDimitry Andric       setOperationAction(ISD::SETCC, VT, Legal);
2935f757f3fSDimitry Andric       setOperationAction(ISD::VSELECT, VT, Legal);
2945f757f3fSDimitry Andric     }
2955f757f3fSDimitry Andric     for (MVT VT : {MVT::v4i64, MVT::v8i32, MVT::v16i16, MVT::v32i8}) {
2965f757f3fSDimitry Andric       setOperationAction(ISD::VECTOR_SHUFFLE, VT, Custom);
2975f757f3fSDimitry Andric       setOperationAction({ISD::ADD, ISD::SUB}, VT, Legal);
2985f757f3fSDimitry Andric       setOperationAction({ISD::UMAX, ISD::UMIN, ISD::SMAX, ISD::SMIN}, VT,
2995f757f3fSDimitry Andric                          Legal);
3005f757f3fSDimitry Andric       setOperationAction({ISD::MUL, ISD::SDIV, ISD::SREM, ISD::UDIV, ISD::UREM},
3015f757f3fSDimitry Andric                          VT, Legal);
3025f757f3fSDimitry Andric       setOperationAction({ISD::AND, ISD::OR, ISD::XOR}, VT, Legal);
3035f757f3fSDimitry Andric       setOperationAction({ISD::SHL, ISD::SRA, ISD::SRL}, VT, Legal);
3045f757f3fSDimitry Andric       setOperationAction({ISD::CTPOP, ISD::CTLZ}, VT, Legal);
3055f757f3fSDimitry Andric       setOperationAction({ISD::MULHS, ISD::MULHU}, VT, Legal);
3065f757f3fSDimitry Andric       setCondCodeAction(
3075f757f3fSDimitry Andric           {ISD::SETNE, ISD::SETGE, ISD::SETGT, ISD::SETUGE, ISD::SETUGT}, VT,
3085f757f3fSDimitry Andric           Expand);
3095f757f3fSDimitry Andric     }
3105f757f3fSDimitry Andric     for (MVT VT : {MVT::v8f32, MVT::v4f64}) {
3115f757f3fSDimitry Andric       setOperationAction({ISD::FADD, ISD::FSUB}, VT, Legal);
3125f757f3fSDimitry Andric       setOperationAction({ISD::FMUL, ISD::FDIV}, VT, Legal);
3135f757f3fSDimitry Andric       setOperationAction(ISD::FMA, VT, Legal);
3145f757f3fSDimitry Andric       setOperationAction(ISD::FSQRT, VT, Legal);
3155f757f3fSDimitry Andric       setOperationAction(ISD::FNEG, VT, Legal);
3165f757f3fSDimitry Andric       setCondCodeAction({ISD::SETGE, ISD::SETGT, ISD::SETOGE, ISD::SETOGT,
3175f757f3fSDimitry Andric                          ISD::SETUGE, ISD::SETUGT},
3185f757f3fSDimitry Andric                         VT, Expand);
3195f757f3fSDimitry Andric     }
3205f757f3fSDimitry Andric   }
3215f757f3fSDimitry Andric 
3225f757f3fSDimitry Andric   // Set DAG combine for LA32 and LA64.
3235f757f3fSDimitry Andric 
3245f757f3fSDimitry Andric   setTargetDAGCombine(ISD::AND);
3255f757f3fSDimitry Andric   setTargetDAGCombine(ISD::OR);
3265f757f3fSDimitry Andric   setTargetDAGCombine(ISD::SRL);
3275f757f3fSDimitry Andric 
3285f757f3fSDimitry Andric   // Set DAG combine for 'LSX' feature.
3295f757f3fSDimitry Andric 
3305f757f3fSDimitry Andric   if (Subtarget.hasExtLSX())
3315f757f3fSDimitry Andric     setTargetDAGCombine(ISD::INTRINSIC_WO_CHAIN);
33281ad6265SDimitry Andric 
33381ad6265SDimitry Andric   // Compute derived properties from the register classes.
33406c3fb27SDimitry Andric   computeRegisterProperties(Subtarget.getRegisterInfo());
33581ad6265SDimitry Andric 
33681ad6265SDimitry Andric   setStackPointerRegisterToSaveRestore(LoongArch::R3);
33781ad6265SDimitry Andric 
33881ad6265SDimitry Andric   setBooleanContents(ZeroOrOneBooleanContent);
3395f757f3fSDimitry Andric   setBooleanVectorContents(ZeroOrNegativeOneBooleanContent);
34081ad6265SDimitry Andric 
341753f127fSDimitry Andric   setMaxAtomicSizeInBitsSupported(Subtarget.getGRLen());
342753f127fSDimitry Andric 
343bdd1243dSDimitry Andric   setMinCmpXchgSizeInBits(32);
344bdd1243dSDimitry Andric 
34581ad6265SDimitry Andric   // Function alignments.
34606c3fb27SDimitry Andric   setMinFunctionAlignment(Align(4));
34706c3fb27SDimitry Andric   // Set preferred alignments.
34806c3fb27SDimitry Andric   setPrefFunctionAlignment(Subtarget.getPrefFunctionAlignment());
34906c3fb27SDimitry Andric   setPrefLoopAlignment(Subtarget.getPrefLoopAlignment());
35006c3fb27SDimitry Andric   setMaxBytesForAlignment(Subtarget.getMaxBytesForAlignment());
35181ad6265SDimitry Andric }
35281ad6265SDimitry Andric 
353bdd1243dSDimitry Andric bool LoongArchTargetLowering::isOffsetFoldingLegal(
354bdd1243dSDimitry Andric     const GlobalAddressSDNode *GA) const {
355bdd1243dSDimitry Andric   // In order to maximise the opportunity for common subexpression elimination,
356bdd1243dSDimitry Andric   // keep a separate ADD node for the global address offset instead of folding
357bdd1243dSDimitry Andric   // it in the global address node. Later peephole optimisations may choose to
358bdd1243dSDimitry Andric   // fold it back in when profitable.
359bdd1243dSDimitry Andric   return false;
360bdd1243dSDimitry Andric }
361bdd1243dSDimitry Andric 
36281ad6265SDimitry Andric SDValue LoongArchTargetLowering::LowerOperation(SDValue Op,
36381ad6265SDimitry Andric                                                 SelectionDAG &DAG) const {
36481ad6265SDimitry Andric   switch (Op.getOpcode()) {
3655f757f3fSDimitry Andric   case ISD::ATOMIC_FENCE:
3665f757f3fSDimitry Andric     return lowerATOMIC_FENCE(Op, DAG);
367bdd1243dSDimitry Andric   case ISD::EH_DWARF_CFA:
368bdd1243dSDimitry Andric     return lowerEH_DWARF_CFA(Op, DAG);
369753f127fSDimitry Andric   case ISD::GlobalAddress:
370753f127fSDimitry Andric     return lowerGlobalAddress(Op, DAG);
371bdd1243dSDimitry Andric   case ISD::GlobalTLSAddress:
372bdd1243dSDimitry Andric     return lowerGlobalTLSAddress(Op, DAG);
373bdd1243dSDimitry Andric   case ISD::INTRINSIC_WO_CHAIN:
374bdd1243dSDimitry Andric     return lowerINTRINSIC_WO_CHAIN(Op, DAG);
375bdd1243dSDimitry Andric   case ISD::INTRINSIC_W_CHAIN:
376bdd1243dSDimitry Andric     return lowerINTRINSIC_W_CHAIN(Op, DAG);
377bdd1243dSDimitry Andric   case ISD::INTRINSIC_VOID:
378bdd1243dSDimitry Andric     return lowerINTRINSIC_VOID(Op, DAG);
379bdd1243dSDimitry Andric   case ISD::BlockAddress:
380bdd1243dSDimitry Andric     return lowerBlockAddress(Op, DAG);
381bdd1243dSDimitry Andric   case ISD::JumpTable:
382bdd1243dSDimitry Andric     return lowerJumpTable(Op, DAG);
38381ad6265SDimitry Andric   case ISD::SHL_PARTS:
38481ad6265SDimitry Andric     return lowerShiftLeftParts(Op, DAG);
38581ad6265SDimitry Andric   case ISD::SRA_PARTS:
38681ad6265SDimitry Andric     return lowerShiftRightParts(Op, DAG, true);
38781ad6265SDimitry Andric   case ISD::SRL_PARTS:
38881ad6265SDimitry Andric     return lowerShiftRightParts(Op, DAG, false);
389753f127fSDimitry Andric   case ISD::ConstantPool:
390753f127fSDimitry Andric     return lowerConstantPool(Op, DAG);
391753f127fSDimitry Andric   case ISD::FP_TO_SINT:
392753f127fSDimitry Andric     return lowerFP_TO_SINT(Op, DAG);
393753f127fSDimitry Andric   case ISD::BITCAST:
394753f127fSDimitry Andric     return lowerBITCAST(Op, DAG);
395753f127fSDimitry Andric   case ISD::UINT_TO_FP:
396753f127fSDimitry Andric     return lowerUINT_TO_FP(Op, DAG);
397bdd1243dSDimitry Andric   case ISD::SINT_TO_FP:
398bdd1243dSDimitry Andric     return lowerSINT_TO_FP(Op, DAG);
399bdd1243dSDimitry Andric   case ISD::VASTART:
400bdd1243dSDimitry Andric     return lowerVASTART(Op, DAG);
401bdd1243dSDimitry Andric   case ISD::FRAMEADDR:
402bdd1243dSDimitry Andric     return lowerFRAMEADDR(Op, DAG);
403bdd1243dSDimitry Andric   case ISD::RETURNADDR:
404bdd1243dSDimitry Andric     return lowerRETURNADDR(Op, DAG);
405bdd1243dSDimitry Andric   case ISD::WRITE_REGISTER:
406bdd1243dSDimitry Andric     return lowerWRITE_REGISTER(Op, DAG);
4075f757f3fSDimitry Andric   case ISD::INSERT_VECTOR_ELT:
4085f757f3fSDimitry Andric     return lowerINSERT_VECTOR_ELT(Op, DAG);
409647cbc5dSDimitry Andric   case ISD::EXTRACT_VECTOR_ELT:
410647cbc5dSDimitry Andric     return lowerEXTRACT_VECTOR_ELT(Op, DAG);
4115f757f3fSDimitry Andric   case ISD::BUILD_VECTOR:
4125f757f3fSDimitry Andric     return lowerBUILD_VECTOR(Op, DAG);
4135f757f3fSDimitry Andric   case ISD::VECTOR_SHUFFLE:
4145f757f3fSDimitry Andric     return lowerVECTOR_SHUFFLE(Op, DAG);
41581ad6265SDimitry Andric   }
416bdd1243dSDimitry Andric   return SDValue();
417bdd1243dSDimitry Andric }
418bdd1243dSDimitry Andric 
4195f757f3fSDimitry Andric SDValue LoongArchTargetLowering::lowerVECTOR_SHUFFLE(SDValue Op,
4205f757f3fSDimitry Andric                                                      SelectionDAG &DAG) const {
4215f757f3fSDimitry Andric   // TODO: custom shuffle.
4225f757f3fSDimitry Andric   return SDValue();
4235f757f3fSDimitry Andric }
4245f757f3fSDimitry Andric 
4255f757f3fSDimitry Andric static bool isConstantOrUndef(const SDValue Op) {
4265f757f3fSDimitry Andric   if (Op->isUndef())
4275f757f3fSDimitry Andric     return true;
4285f757f3fSDimitry Andric   if (isa<ConstantSDNode>(Op))
4295f757f3fSDimitry Andric     return true;
4305f757f3fSDimitry Andric   if (isa<ConstantFPSDNode>(Op))
4315f757f3fSDimitry Andric     return true;
4325f757f3fSDimitry Andric   return false;
4335f757f3fSDimitry Andric }
4345f757f3fSDimitry Andric 
4355f757f3fSDimitry Andric static bool isConstantOrUndefBUILD_VECTOR(const BuildVectorSDNode *Op) {
4365f757f3fSDimitry Andric   for (unsigned i = 0; i < Op->getNumOperands(); ++i)
4375f757f3fSDimitry Andric     if (isConstantOrUndef(Op->getOperand(i)))
4385f757f3fSDimitry Andric       return true;
4395f757f3fSDimitry Andric   return false;
4405f757f3fSDimitry Andric }
4415f757f3fSDimitry Andric 
4425f757f3fSDimitry Andric SDValue LoongArchTargetLowering::lowerBUILD_VECTOR(SDValue Op,
4435f757f3fSDimitry Andric                                                    SelectionDAG &DAG) const {
4445f757f3fSDimitry Andric   BuildVectorSDNode *Node = cast<BuildVectorSDNode>(Op);
4455f757f3fSDimitry Andric   EVT ResTy = Op->getValueType(0);
4465f757f3fSDimitry Andric   SDLoc DL(Op);
4475f757f3fSDimitry Andric   APInt SplatValue, SplatUndef;
4485f757f3fSDimitry Andric   unsigned SplatBitSize;
4495f757f3fSDimitry Andric   bool HasAnyUndefs;
4505f757f3fSDimitry Andric   bool Is128Vec = ResTy.is128BitVector();
4515f757f3fSDimitry Andric   bool Is256Vec = ResTy.is256BitVector();
4525f757f3fSDimitry Andric 
4535f757f3fSDimitry Andric   if ((!Subtarget.hasExtLSX() || !Is128Vec) &&
4545f757f3fSDimitry Andric       (!Subtarget.hasExtLASX() || !Is256Vec))
4555f757f3fSDimitry Andric     return SDValue();
4565f757f3fSDimitry Andric 
4575f757f3fSDimitry Andric   if (Node->isConstantSplat(SplatValue, SplatUndef, SplatBitSize, HasAnyUndefs,
4585f757f3fSDimitry Andric                             /*MinSplatBits=*/8) &&
4595f757f3fSDimitry Andric       SplatBitSize <= 64) {
4605f757f3fSDimitry Andric     // We can only cope with 8, 16, 32, or 64-bit elements.
4615f757f3fSDimitry Andric     if (SplatBitSize != 8 && SplatBitSize != 16 && SplatBitSize != 32 &&
4625f757f3fSDimitry Andric         SplatBitSize != 64)
4635f757f3fSDimitry Andric       return SDValue();
4645f757f3fSDimitry Andric 
4655f757f3fSDimitry Andric     EVT ViaVecTy;
4665f757f3fSDimitry Andric 
4675f757f3fSDimitry Andric     switch (SplatBitSize) {
4685f757f3fSDimitry Andric     default:
4695f757f3fSDimitry Andric       return SDValue();
4705f757f3fSDimitry Andric     case 8:
4715f757f3fSDimitry Andric       ViaVecTy = Is128Vec ? MVT::v16i8 : MVT::v32i8;
4725f757f3fSDimitry Andric       break;
4735f757f3fSDimitry Andric     case 16:
4745f757f3fSDimitry Andric       ViaVecTy = Is128Vec ? MVT::v8i16 : MVT::v16i16;
4755f757f3fSDimitry Andric       break;
4765f757f3fSDimitry Andric     case 32:
4775f757f3fSDimitry Andric       ViaVecTy = Is128Vec ? MVT::v4i32 : MVT::v8i32;
4785f757f3fSDimitry Andric       break;
4795f757f3fSDimitry Andric     case 64:
4805f757f3fSDimitry Andric       ViaVecTy = Is128Vec ? MVT::v2i64 : MVT::v4i64;
4815f757f3fSDimitry Andric       break;
4825f757f3fSDimitry Andric     }
4835f757f3fSDimitry Andric 
4845f757f3fSDimitry Andric     // SelectionDAG::getConstant will promote SplatValue appropriately.
4855f757f3fSDimitry Andric     SDValue Result = DAG.getConstant(SplatValue, DL, ViaVecTy);
4865f757f3fSDimitry Andric 
4875f757f3fSDimitry Andric     // Bitcast to the type we originally wanted.
4885f757f3fSDimitry Andric     if (ViaVecTy != ResTy)
4895f757f3fSDimitry Andric       Result = DAG.getNode(ISD::BITCAST, SDLoc(Node), ResTy, Result);
4905f757f3fSDimitry Andric 
4915f757f3fSDimitry Andric     return Result;
4925f757f3fSDimitry Andric   }
4935f757f3fSDimitry Andric 
4945f757f3fSDimitry Andric   if (DAG.isSplatValue(Op, /*AllowUndefs=*/false))
4955f757f3fSDimitry Andric     return Op;
4965f757f3fSDimitry Andric 
4975f757f3fSDimitry Andric   if (!isConstantOrUndefBUILD_VECTOR(Node)) {
4985f757f3fSDimitry Andric     // Use INSERT_VECTOR_ELT operations rather than expand to stores.
4995f757f3fSDimitry Andric     // The resulting code is the same length as the expansion, but it doesn't
5005f757f3fSDimitry Andric     // use memory operations.
5015f757f3fSDimitry Andric     EVT ResTy = Node->getValueType(0);
5025f757f3fSDimitry Andric 
5035f757f3fSDimitry Andric     assert(ResTy.isVector());
5045f757f3fSDimitry Andric 
5055f757f3fSDimitry Andric     unsigned NumElts = ResTy.getVectorNumElements();
5065f757f3fSDimitry Andric     SDValue Vector = DAG.getUNDEF(ResTy);
5075f757f3fSDimitry Andric     for (unsigned i = 0; i < NumElts; ++i) {
5085f757f3fSDimitry Andric       Vector = DAG.getNode(ISD::INSERT_VECTOR_ELT, DL, ResTy, Vector,
5095f757f3fSDimitry Andric                            Node->getOperand(i),
5105f757f3fSDimitry Andric                            DAG.getConstant(i, DL, Subtarget.getGRLenVT()));
5115f757f3fSDimitry Andric     }
5125f757f3fSDimitry Andric     return Vector;
5135f757f3fSDimitry Andric   }
5145f757f3fSDimitry Andric 
5155f757f3fSDimitry Andric   return SDValue();
5165f757f3fSDimitry Andric }
5175f757f3fSDimitry Andric 
5185f757f3fSDimitry Andric SDValue
519647cbc5dSDimitry Andric LoongArchTargetLowering::lowerEXTRACT_VECTOR_ELT(SDValue Op,
520647cbc5dSDimitry Andric                                                  SelectionDAG &DAG) const {
521647cbc5dSDimitry Andric   EVT VecTy = Op->getOperand(0)->getValueType(0);
522647cbc5dSDimitry Andric   SDValue Idx = Op->getOperand(1);
523647cbc5dSDimitry Andric   EVT EltTy = VecTy.getVectorElementType();
524647cbc5dSDimitry Andric   unsigned NumElts = VecTy.getVectorNumElements();
525647cbc5dSDimitry Andric 
526647cbc5dSDimitry Andric   if (isa<ConstantSDNode>(Idx) &&
527647cbc5dSDimitry Andric       (EltTy == MVT::i32 || EltTy == MVT::i64 || EltTy == MVT::f32 ||
528*1db9f3b2SDimitry Andric        EltTy == MVT::f64 || Idx->getAsZExtVal() < NumElts / 2))
529647cbc5dSDimitry Andric     return Op;
530647cbc5dSDimitry Andric 
531647cbc5dSDimitry Andric   return SDValue();
532647cbc5dSDimitry Andric }
533647cbc5dSDimitry Andric 
534647cbc5dSDimitry Andric SDValue
5355f757f3fSDimitry Andric LoongArchTargetLowering::lowerINSERT_VECTOR_ELT(SDValue Op,
5365f757f3fSDimitry Andric                                                 SelectionDAG &DAG) const {
5375f757f3fSDimitry Andric   if (isa<ConstantSDNode>(Op->getOperand(2)))
5385f757f3fSDimitry Andric     return Op;
5395f757f3fSDimitry Andric   return SDValue();
5405f757f3fSDimitry Andric }
5415f757f3fSDimitry Andric 
5425f757f3fSDimitry Andric SDValue LoongArchTargetLowering::lowerATOMIC_FENCE(SDValue Op,
5435f757f3fSDimitry Andric                                                    SelectionDAG &DAG) const {
5445f757f3fSDimitry Andric   SDLoc DL(Op);
5455f757f3fSDimitry Andric   SyncScope::ID FenceSSID =
5465f757f3fSDimitry Andric       static_cast<SyncScope::ID>(Op.getConstantOperandVal(2));
5475f757f3fSDimitry Andric 
5485f757f3fSDimitry Andric   // singlethread fences only synchronize with signal handlers on the same
5495f757f3fSDimitry Andric   // thread and thus only need to preserve instruction order, not actually
5505f757f3fSDimitry Andric   // enforce memory ordering.
5515f757f3fSDimitry Andric   if (FenceSSID == SyncScope::SingleThread)
5525f757f3fSDimitry Andric     // MEMBARRIER is a compiler barrier; it codegens to a no-op.
5535f757f3fSDimitry Andric     return DAG.getNode(ISD::MEMBARRIER, DL, MVT::Other, Op.getOperand(0));
5545f757f3fSDimitry Andric 
5555f757f3fSDimitry Andric   return Op;
5565f757f3fSDimitry Andric }
5575f757f3fSDimitry Andric 
558bdd1243dSDimitry Andric SDValue LoongArchTargetLowering::lowerWRITE_REGISTER(SDValue Op,
559bdd1243dSDimitry Andric                                                      SelectionDAG &DAG) const {
560bdd1243dSDimitry Andric 
561bdd1243dSDimitry Andric   if (Subtarget.is64Bit() && Op.getOperand(2).getValueType() == MVT::i32) {
562bdd1243dSDimitry Andric     DAG.getContext()->emitError(
563bdd1243dSDimitry Andric         "On LA64, only 64-bit registers can be written.");
564bdd1243dSDimitry Andric     return Op.getOperand(0);
565bdd1243dSDimitry Andric   }
566bdd1243dSDimitry Andric 
567bdd1243dSDimitry Andric   if (!Subtarget.is64Bit() && Op.getOperand(2).getValueType() == MVT::i64) {
568bdd1243dSDimitry Andric     DAG.getContext()->emitError(
569bdd1243dSDimitry Andric         "On LA32, only 32-bit registers can be written.");
570bdd1243dSDimitry Andric     return Op.getOperand(0);
571bdd1243dSDimitry Andric   }
572bdd1243dSDimitry Andric 
573bdd1243dSDimitry Andric   return Op;
574bdd1243dSDimitry Andric }
575bdd1243dSDimitry Andric 
576bdd1243dSDimitry Andric SDValue LoongArchTargetLowering::lowerFRAMEADDR(SDValue Op,
577bdd1243dSDimitry Andric                                                 SelectionDAG &DAG) const {
578bdd1243dSDimitry Andric   if (!isa<ConstantSDNode>(Op.getOperand(0))) {
579bdd1243dSDimitry Andric     DAG.getContext()->emitError("argument to '__builtin_frame_address' must "
580bdd1243dSDimitry Andric                                 "be a constant integer");
581bdd1243dSDimitry Andric     return SDValue();
582bdd1243dSDimitry Andric   }
583bdd1243dSDimitry Andric 
584bdd1243dSDimitry Andric   MachineFunction &MF = DAG.getMachineFunction();
585bdd1243dSDimitry Andric   MF.getFrameInfo().setFrameAddressIsTaken(true);
586bdd1243dSDimitry Andric   Register FrameReg = Subtarget.getRegisterInfo()->getFrameRegister(MF);
587bdd1243dSDimitry Andric   EVT VT = Op.getValueType();
588bdd1243dSDimitry Andric   SDLoc DL(Op);
589bdd1243dSDimitry Andric   SDValue FrameAddr = DAG.getCopyFromReg(DAG.getEntryNode(), DL, FrameReg, VT);
590647cbc5dSDimitry Andric   unsigned Depth = Op.getConstantOperandVal(0);
591bdd1243dSDimitry Andric   int GRLenInBytes = Subtarget.getGRLen() / 8;
592bdd1243dSDimitry Andric 
593bdd1243dSDimitry Andric   while (Depth--) {
594bdd1243dSDimitry Andric     int Offset = -(GRLenInBytes * 2);
595bdd1243dSDimitry Andric     SDValue Ptr = DAG.getNode(ISD::ADD, DL, VT, FrameAddr,
596bdd1243dSDimitry Andric                               DAG.getIntPtrConstant(Offset, DL));
597bdd1243dSDimitry Andric     FrameAddr =
598bdd1243dSDimitry Andric         DAG.getLoad(VT, DL, DAG.getEntryNode(), Ptr, MachinePointerInfo());
599bdd1243dSDimitry Andric   }
600bdd1243dSDimitry Andric   return FrameAddr;
601bdd1243dSDimitry Andric }
602bdd1243dSDimitry Andric 
603bdd1243dSDimitry Andric SDValue LoongArchTargetLowering::lowerRETURNADDR(SDValue Op,
604bdd1243dSDimitry Andric                                                  SelectionDAG &DAG) const {
605bdd1243dSDimitry Andric   if (verifyReturnAddressArgumentIsConstant(Op, DAG))
606bdd1243dSDimitry Andric     return SDValue();
607bdd1243dSDimitry Andric 
608bdd1243dSDimitry Andric   // Currently only support lowering return address for current frame.
609647cbc5dSDimitry Andric   if (Op.getConstantOperandVal(0) != 0) {
610bdd1243dSDimitry Andric     DAG.getContext()->emitError(
611bdd1243dSDimitry Andric         "return address can only be determined for the current frame");
612bdd1243dSDimitry Andric     return SDValue();
613bdd1243dSDimitry Andric   }
614bdd1243dSDimitry Andric 
615bdd1243dSDimitry Andric   MachineFunction &MF = DAG.getMachineFunction();
616bdd1243dSDimitry Andric   MF.getFrameInfo().setReturnAddressIsTaken(true);
617bdd1243dSDimitry Andric   MVT GRLenVT = Subtarget.getGRLenVT();
618bdd1243dSDimitry Andric 
619bdd1243dSDimitry Andric   // Return the value of the return address register, marking it an implicit
620bdd1243dSDimitry Andric   // live-in.
621bdd1243dSDimitry Andric   Register Reg = MF.addLiveIn(Subtarget.getRegisterInfo()->getRARegister(),
622bdd1243dSDimitry Andric                               getRegClassFor(GRLenVT));
623bdd1243dSDimitry Andric   return DAG.getCopyFromReg(DAG.getEntryNode(), SDLoc(Op), Reg, GRLenVT);
624bdd1243dSDimitry Andric }
625bdd1243dSDimitry Andric 
626bdd1243dSDimitry Andric SDValue LoongArchTargetLowering::lowerEH_DWARF_CFA(SDValue Op,
627bdd1243dSDimitry Andric                                                    SelectionDAG &DAG) const {
628bdd1243dSDimitry Andric   MachineFunction &MF = DAG.getMachineFunction();
629bdd1243dSDimitry Andric   auto Size = Subtarget.getGRLen() / 8;
630bdd1243dSDimitry Andric   auto FI = MF.getFrameInfo().CreateFixedObject(Size, 0, false);
631bdd1243dSDimitry Andric   return DAG.getFrameIndex(FI, getPointerTy(DAG.getDataLayout()));
632bdd1243dSDimitry Andric }
633bdd1243dSDimitry Andric 
634bdd1243dSDimitry Andric SDValue LoongArchTargetLowering::lowerVASTART(SDValue Op,
635bdd1243dSDimitry Andric                                               SelectionDAG &DAG) const {
636bdd1243dSDimitry Andric   MachineFunction &MF = DAG.getMachineFunction();
637bdd1243dSDimitry Andric   auto *FuncInfo = MF.getInfo<LoongArchMachineFunctionInfo>();
638bdd1243dSDimitry Andric 
639bdd1243dSDimitry Andric   SDLoc DL(Op);
640bdd1243dSDimitry Andric   SDValue FI = DAG.getFrameIndex(FuncInfo->getVarArgsFrameIndex(),
641bdd1243dSDimitry Andric                                  getPointerTy(MF.getDataLayout()));
642bdd1243dSDimitry Andric 
643bdd1243dSDimitry Andric   // vastart just stores the address of the VarArgsFrameIndex slot into the
644bdd1243dSDimitry Andric   // memory location argument.
645bdd1243dSDimitry Andric   const Value *SV = cast<SrcValueSDNode>(Op.getOperand(2))->getValue();
646bdd1243dSDimitry Andric   return DAG.getStore(Op.getOperand(0), DL, FI, Op.getOperand(1),
647bdd1243dSDimitry Andric                       MachinePointerInfo(SV));
64881ad6265SDimitry Andric }
64981ad6265SDimitry Andric 
650753f127fSDimitry Andric SDValue LoongArchTargetLowering::lowerUINT_TO_FP(SDValue Op,
651753f127fSDimitry Andric                                                  SelectionDAG &DAG) const {
652bdd1243dSDimitry Andric   assert(Subtarget.is64Bit() && Subtarget.hasBasicF() &&
653bdd1243dSDimitry Andric          !Subtarget.hasBasicD() && "unexpected target features");
654753f127fSDimitry Andric 
655753f127fSDimitry Andric   SDLoc DL(Op);
656bdd1243dSDimitry Andric   SDValue Op0 = Op.getOperand(0);
657bdd1243dSDimitry Andric   if (Op0->getOpcode() == ISD::AND) {
658bdd1243dSDimitry Andric     auto *C = dyn_cast<ConstantSDNode>(Op0.getOperand(1));
659bdd1243dSDimitry Andric     if (C && C->getZExtValue() < UINT64_C(0xFFFFFFFF))
660753f127fSDimitry Andric       return Op;
661bdd1243dSDimitry Andric   }
662bdd1243dSDimitry Andric 
663bdd1243dSDimitry Andric   if (Op0->getOpcode() == LoongArchISD::BSTRPICK &&
664bdd1243dSDimitry Andric       Op0.getConstantOperandVal(1) < UINT64_C(0X1F) &&
665bdd1243dSDimitry Andric       Op0.getConstantOperandVal(2) == UINT64_C(0))
666bdd1243dSDimitry Andric     return Op;
667bdd1243dSDimitry Andric 
668bdd1243dSDimitry Andric   if (Op0.getOpcode() == ISD::AssertZext &&
669bdd1243dSDimitry Andric       dyn_cast<VTSDNode>(Op0.getOperand(1))->getVT().bitsLT(MVT::i32))
670bdd1243dSDimitry Andric     return Op;
671bdd1243dSDimitry Andric 
672bdd1243dSDimitry Andric   EVT OpVT = Op0.getValueType();
673bdd1243dSDimitry Andric   EVT RetVT = Op.getValueType();
674bdd1243dSDimitry Andric   RTLIB::Libcall LC = RTLIB::getUINTTOFP(OpVT, RetVT);
675bdd1243dSDimitry Andric   MakeLibCallOptions CallOptions;
676bdd1243dSDimitry Andric   CallOptions.setTypeListBeforeSoften(OpVT, RetVT, true);
677bdd1243dSDimitry Andric   SDValue Chain = SDValue();
678bdd1243dSDimitry Andric   SDValue Result;
679bdd1243dSDimitry Andric   std::tie(Result, Chain) =
680bdd1243dSDimitry Andric       makeLibCall(DAG, LC, Op.getValueType(), Op0, CallOptions, DL, Chain);
681bdd1243dSDimitry Andric   return Result;
682bdd1243dSDimitry Andric }
683bdd1243dSDimitry Andric 
684bdd1243dSDimitry Andric SDValue LoongArchTargetLowering::lowerSINT_TO_FP(SDValue Op,
685bdd1243dSDimitry Andric                                                  SelectionDAG &DAG) const {
686bdd1243dSDimitry Andric   assert(Subtarget.is64Bit() && Subtarget.hasBasicF() &&
687bdd1243dSDimitry Andric          !Subtarget.hasBasicD() && "unexpected target features");
688bdd1243dSDimitry Andric 
689bdd1243dSDimitry Andric   SDLoc DL(Op);
690bdd1243dSDimitry Andric   SDValue Op0 = Op.getOperand(0);
691bdd1243dSDimitry Andric 
692bdd1243dSDimitry Andric   if ((Op0.getOpcode() == ISD::AssertSext ||
693bdd1243dSDimitry Andric        Op0.getOpcode() == ISD::SIGN_EXTEND_INREG) &&
694bdd1243dSDimitry Andric       dyn_cast<VTSDNode>(Op0.getOperand(1))->getVT().bitsLE(MVT::i32))
695bdd1243dSDimitry Andric     return Op;
696bdd1243dSDimitry Andric 
697bdd1243dSDimitry Andric   EVT OpVT = Op0.getValueType();
698bdd1243dSDimitry Andric   EVT RetVT = Op.getValueType();
699bdd1243dSDimitry Andric   RTLIB::Libcall LC = RTLIB::getSINTTOFP(OpVT, RetVT);
700bdd1243dSDimitry Andric   MakeLibCallOptions CallOptions;
701bdd1243dSDimitry Andric   CallOptions.setTypeListBeforeSoften(OpVT, RetVT, true);
702bdd1243dSDimitry Andric   SDValue Chain = SDValue();
703bdd1243dSDimitry Andric   SDValue Result;
704bdd1243dSDimitry Andric   std::tie(Result, Chain) =
705bdd1243dSDimitry Andric       makeLibCall(DAG, LC, Op.getValueType(), Op0, CallOptions, DL, Chain);
706bdd1243dSDimitry Andric   return Result;
707753f127fSDimitry Andric }
708753f127fSDimitry Andric 
709753f127fSDimitry Andric SDValue LoongArchTargetLowering::lowerBITCAST(SDValue Op,
710753f127fSDimitry Andric                                               SelectionDAG &DAG) const {
711753f127fSDimitry Andric 
712753f127fSDimitry Andric   SDLoc DL(Op);
713753f127fSDimitry Andric   SDValue Op0 = Op.getOperand(0);
714753f127fSDimitry Andric 
715753f127fSDimitry Andric   if (Op.getValueType() == MVT::f32 && Op0.getValueType() == MVT::i32 &&
716753f127fSDimitry Andric       Subtarget.is64Bit() && Subtarget.hasBasicF()) {
717753f127fSDimitry Andric     SDValue NewOp0 = DAG.getNode(ISD::ANY_EXTEND, DL, MVT::i64, Op0);
718753f127fSDimitry Andric     return DAG.getNode(LoongArchISD::MOVGR2FR_W_LA64, DL, MVT::f32, NewOp0);
719753f127fSDimitry Andric   }
720753f127fSDimitry Andric   return Op;
721753f127fSDimitry Andric }
722753f127fSDimitry Andric 
723753f127fSDimitry Andric SDValue LoongArchTargetLowering::lowerFP_TO_SINT(SDValue Op,
724753f127fSDimitry Andric                                                  SelectionDAG &DAG) const {
725753f127fSDimitry Andric 
726753f127fSDimitry Andric   SDLoc DL(Op);
727753f127fSDimitry Andric 
728753f127fSDimitry Andric   if (Op.getValueSizeInBits() > 32 && Subtarget.hasBasicF() &&
729753f127fSDimitry Andric       !Subtarget.hasBasicD()) {
730753f127fSDimitry Andric     SDValue Dst =
731753f127fSDimitry Andric         DAG.getNode(LoongArchISD::FTINT, DL, MVT::f32, Op.getOperand(0));
732753f127fSDimitry Andric     return DAG.getNode(LoongArchISD::MOVFR2GR_S_LA64, DL, MVT::i64, Dst);
733753f127fSDimitry Andric   }
734753f127fSDimitry Andric 
735753f127fSDimitry Andric   EVT FPTy = EVT::getFloatingPointVT(Op.getValueSizeInBits());
736753f127fSDimitry Andric   SDValue Trunc = DAG.getNode(LoongArchISD::FTINT, DL, FPTy, Op.getOperand(0));
737753f127fSDimitry Andric   return DAG.getNode(ISD::BITCAST, DL, Op.getValueType(), Trunc);
738753f127fSDimitry Andric }
739753f127fSDimitry Andric 
740bdd1243dSDimitry Andric static SDValue getTargetNode(GlobalAddressSDNode *N, SDLoc DL, EVT Ty,
741bdd1243dSDimitry Andric                              SelectionDAG &DAG, unsigned Flags) {
742bdd1243dSDimitry Andric   return DAG.getTargetGlobalAddress(N->getGlobal(), DL, Ty, 0, Flags);
743bdd1243dSDimitry Andric }
744bdd1243dSDimitry Andric 
745bdd1243dSDimitry Andric static SDValue getTargetNode(BlockAddressSDNode *N, SDLoc DL, EVT Ty,
746bdd1243dSDimitry Andric                              SelectionDAG &DAG, unsigned Flags) {
747bdd1243dSDimitry Andric   return DAG.getTargetBlockAddress(N->getBlockAddress(), Ty, N->getOffset(),
748bdd1243dSDimitry Andric                                    Flags);
749bdd1243dSDimitry Andric }
750bdd1243dSDimitry Andric 
751bdd1243dSDimitry Andric static SDValue getTargetNode(ConstantPoolSDNode *N, SDLoc DL, EVT Ty,
752bdd1243dSDimitry Andric                              SelectionDAG &DAG, unsigned Flags) {
753bdd1243dSDimitry Andric   return DAG.getTargetConstantPool(N->getConstVal(), Ty, N->getAlign(),
754bdd1243dSDimitry Andric                                    N->getOffset(), Flags);
755bdd1243dSDimitry Andric }
756bdd1243dSDimitry Andric 
757bdd1243dSDimitry Andric static SDValue getTargetNode(JumpTableSDNode *N, SDLoc DL, EVT Ty,
758bdd1243dSDimitry Andric                              SelectionDAG &DAG, unsigned Flags) {
759bdd1243dSDimitry Andric   return DAG.getTargetJumpTable(N->getIndex(), Ty, Flags);
760bdd1243dSDimitry Andric }
761bdd1243dSDimitry Andric 
762bdd1243dSDimitry Andric template <class NodeTy>
763bdd1243dSDimitry Andric SDValue LoongArchTargetLowering::getAddr(NodeTy *N, SelectionDAG &DAG,
764*1db9f3b2SDimitry Andric                                          CodeModel::Model M,
765bdd1243dSDimitry Andric                                          bool IsLocal) const {
766bdd1243dSDimitry Andric   SDLoc DL(N);
767bdd1243dSDimitry Andric   EVT Ty = getPointerTy(DAG.getDataLayout());
768bdd1243dSDimitry Andric   SDValue Addr = getTargetNode(N, DL, Ty, DAG, 0);
76906c3fb27SDimitry Andric 
770*1db9f3b2SDimitry Andric   switch (M) {
77106c3fb27SDimitry Andric   default:
77206c3fb27SDimitry Andric     report_fatal_error("Unsupported code model");
77306c3fb27SDimitry Andric 
77406c3fb27SDimitry Andric   case CodeModel::Large: {
77506c3fb27SDimitry Andric     assert(Subtarget.is64Bit() && "Large code model requires LA64");
77606c3fb27SDimitry Andric 
77706c3fb27SDimitry Andric     // This is not actually used, but is necessary for successfully matching
77806c3fb27SDimitry Andric     // the PseudoLA_*_LARGE nodes.
77906c3fb27SDimitry Andric     SDValue Tmp = DAG.getConstant(0, DL, Ty);
78006c3fb27SDimitry Andric     if (IsLocal)
78106c3fb27SDimitry Andric       // This generates the pattern (PseudoLA_PCREL_LARGE tmp sym), that
78206c3fb27SDimitry Andric       // eventually becomes the desired 5-insn code sequence.
78306c3fb27SDimitry Andric       return SDValue(DAG.getMachineNode(LoongArch::PseudoLA_PCREL_LARGE, DL, Ty,
78406c3fb27SDimitry Andric                                         Tmp, Addr),
78506c3fb27SDimitry Andric                      0);
78606c3fb27SDimitry Andric 
78706c3fb27SDimitry Andric     // This generates the pattern (PseudoLA_GOT_LARGE tmp sym), that eventually
78806c3fb27SDimitry Andric     // becomes the desired 5-insn code sequence.
78906c3fb27SDimitry Andric     return SDValue(
79006c3fb27SDimitry Andric         DAG.getMachineNode(LoongArch::PseudoLA_GOT_LARGE, DL, Ty, Tmp, Addr),
79106c3fb27SDimitry Andric         0);
79206c3fb27SDimitry Andric   }
79306c3fb27SDimitry Andric 
79406c3fb27SDimitry Andric   case CodeModel::Small:
79506c3fb27SDimitry Andric   case CodeModel::Medium:
796bdd1243dSDimitry Andric     if (IsLocal)
797bdd1243dSDimitry Andric       // This generates the pattern (PseudoLA_PCREL sym), which expands to
798bdd1243dSDimitry Andric       // (addi.w/d (pcalau12i %pc_hi20(sym)) %pc_lo12(sym)).
79906c3fb27SDimitry Andric       return SDValue(
80006c3fb27SDimitry Andric           DAG.getMachineNode(LoongArch::PseudoLA_PCREL, DL, Ty, Addr), 0);
801bdd1243dSDimitry Andric 
802bdd1243dSDimitry Andric     // This generates the pattern (PseudoLA_GOT sym), which expands to (ld.w/d
803bdd1243dSDimitry Andric     // (pcalau12i %got_pc_hi20(sym)) %got_pc_lo12(sym)).
80406c3fb27SDimitry Andric     return SDValue(DAG.getMachineNode(LoongArch::PseudoLA_GOT, DL, Ty, Addr),
80506c3fb27SDimitry Andric                    0);
80606c3fb27SDimitry Andric   }
807bdd1243dSDimitry Andric }
808bdd1243dSDimitry Andric 
809bdd1243dSDimitry Andric SDValue LoongArchTargetLowering::lowerBlockAddress(SDValue Op,
810bdd1243dSDimitry Andric                                                    SelectionDAG &DAG) const {
811*1db9f3b2SDimitry Andric   return getAddr(cast<BlockAddressSDNode>(Op), DAG,
812*1db9f3b2SDimitry Andric                  DAG.getTarget().getCodeModel());
813bdd1243dSDimitry Andric }
814bdd1243dSDimitry Andric 
815bdd1243dSDimitry Andric SDValue LoongArchTargetLowering::lowerJumpTable(SDValue Op,
816bdd1243dSDimitry Andric                                                 SelectionDAG &DAG) const {
817*1db9f3b2SDimitry Andric   return getAddr(cast<JumpTableSDNode>(Op), DAG,
818*1db9f3b2SDimitry Andric                  DAG.getTarget().getCodeModel());
819bdd1243dSDimitry Andric }
820bdd1243dSDimitry Andric 
821753f127fSDimitry Andric SDValue LoongArchTargetLowering::lowerConstantPool(SDValue Op,
822753f127fSDimitry Andric                                                    SelectionDAG &DAG) const {
823*1db9f3b2SDimitry Andric   return getAddr(cast<ConstantPoolSDNode>(Op), DAG,
824*1db9f3b2SDimitry Andric                  DAG.getTarget().getCodeModel());
825753f127fSDimitry Andric }
826753f127fSDimitry Andric 
827753f127fSDimitry Andric SDValue LoongArchTargetLowering::lowerGlobalAddress(SDValue Op,
828753f127fSDimitry Andric                                                     SelectionDAG &DAG) const {
829bdd1243dSDimitry Andric   GlobalAddressSDNode *N = cast<GlobalAddressSDNode>(Op);
830bdd1243dSDimitry Andric   assert(N->getOffset() == 0 && "unexpected offset in global node");
831*1db9f3b2SDimitry Andric   auto CM = DAG.getTarget().getCodeModel();
832*1db9f3b2SDimitry Andric   const GlobalValue *GV = N->getGlobal();
833*1db9f3b2SDimitry Andric 
834*1db9f3b2SDimitry Andric   if (GV->isDSOLocal() && isa<GlobalVariable>(GV)) {
835*1db9f3b2SDimitry Andric     if (auto GCM = dyn_cast<GlobalVariable>(GV)->getCodeModel())
836*1db9f3b2SDimitry Andric       CM = *GCM;
837*1db9f3b2SDimitry Andric   }
838*1db9f3b2SDimitry Andric 
839*1db9f3b2SDimitry Andric   return getAddr(N, DAG, CM, GV->isDSOLocal());
840bdd1243dSDimitry Andric }
841753f127fSDimitry Andric 
842bdd1243dSDimitry Andric SDValue LoongArchTargetLowering::getStaticTLSAddr(GlobalAddressSDNode *N,
843bdd1243dSDimitry Andric                                                   SelectionDAG &DAG,
84406c3fb27SDimitry Andric                                                   unsigned Opc,
84506c3fb27SDimitry Andric                                                   bool Large) const {
846bdd1243dSDimitry Andric   SDLoc DL(N);
847bdd1243dSDimitry Andric   EVT Ty = getPointerTy(DAG.getDataLayout());
848bdd1243dSDimitry Andric   MVT GRLenVT = Subtarget.getGRLenVT();
849bdd1243dSDimitry Andric 
85006c3fb27SDimitry Andric   // This is not actually used, but is necessary for successfully matching the
85106c3fb27SDimitry Andric   // PseudoLA_*_LARGE nodes.
85206c3fb27SDimitry Andric   SDValue Tmp = DAG.getConstant(0, DL, Ty);
853bdd1243dSDimitry Andric   SDValue Addr = DAG.getTargetGlobalAddress(N->getGlobal(), DL, Ty, 0, 0);
85406c3fb27SDimitry Andric   SDValue Offset = Large
85506c3fb27SDimitry Andric                        ? SDValue(DAG.getMachineNode(Opc, DL, Ty, Tmp, Addr), 0)
85606c3fb27SDimitry Andric                        : SDValue(DAG.getMachineNode(Opc, DL, Ty, Addr), 0);
857bdd1243dSDimitry Andric 
858bdd1243dSDimitry Andric   // Add the thread pointer.
859bdd1243dSDimitry Andric   return DAG.getNode(ISD::ADD, DL, Ty, Offset,
860bdd1243dSDimitry Andric                      DAG.getRegister(LoongArch::R2, GRLenVT));
861bdd1243dSDimitry Andric }
862bdd1243dSDimitry Andric 
863bdd1243dSDimitry Andric SDValue LoongArchTargetLowering::getDynamicTLSAddr(GlobalAddressSDNode *N,
864bdd1243dSDimitry Andric                                                    SelectionDAG &DAG,
86506c3fb27SDimitry Andric                                                    unsigned Opc,
86606c3fb27SDimitry Andric                                                    bool Large) const {
867bdd1243dSDimitry Andric   SDLoc DL(N);
868bdd1243dSDimitry Andric   EVT Ty = getPointerTy(DAG.getDataLayout());
869bdd1243dSDimitry Andric   IntegerType *CallTy = Type::getIntNTy(*DAG.getContext(), Ty.getSizeInBits());
870bdd1243dSDimitry Andric 
87106c3fb27SDimitry Andric   // This is not actually used, but is necessary for successfully matching the
87206c3fb27SDimitry Andric   // PseudoLA_*_LARGE nodes.
87306c3fb27SDimitry Andric   SDValue Tmp = DAG.getConstant(0, DL, Ty);
87406c3fb27SDimitry Andric 
875bdd1243dSDimitry Andric   // Use a PC-relative addressing mode to access the dynamic GOT address.
876bdd1243dSDimitry Andric   SDValue Addr = DAG.getTargetGlobalAddress(N->getGlobal(), DL, Ty, 0, 0);
87706c3fb27SDimitry Andric   SDValue Load = Large ? SDValue(DAG.getMachineNode(Opc, DL, Ty, Tmp, Addr), 0)
87806c3fb27SDimitry Andric                        : SDValue(DAG.getMachineNode(Opc, DL, Ty, Addr), 0);
879bdd1243dSDimitry Andric 
880bdd1243dSDimitry Andric   // Prepare argument list to generate call.
881bdd1243dSDimitry Andric   ArgListTy Args;
882bdd1243dSDimitry Andric   ArgListEntry Entry;
883bdd1243dSDimitry Andric   Entry.Node = Load;
884bdd1243dSDimitry Andric   Entry.Ty = CallTy;
885bdd1243dSDimitry Andric   Args.push_back(Entry);
886bdd1243dSDimitry Andric 
887bdd1243dSDimitry Andric   // Setup call to __tls_get_addr.
888bdd1243dSDimitry Andric   TargetLowering::CallLoweringInfo CLI(DAG);
889bdd1243dSDimitry Andric   CLI.setDebugLoc(DL)
890bdd1243dSDimitry Andric       .setChain(DAG.getEntryNode())
891bdd1243dSDimitry Andric       .setLibCallee(CallingConv::C, CallTy,
892bdd1243dSDimitry Andric                     DAG.getExternalSymbol("__tls_get_addr", Ty),
893bdd1243dSDimitry Andric                     std::move(Args));
894bdd1243dSDimitry Andric 
895bdd1243dSDimitry Andric   return LowerCallTo(CLI).first;
896bdd1243dSDimitry Andric }
897bdd1243dSDimitry Andric 
898bdd1243dSDimitry Andric SDValue
899bdd1243dSDimitry Andric LoongArchTargetLowering::lowerGlobalTLSAddress(SDValue Op,
900bdd1243dSDimitry Andric                                                SelectionDAG &DAG) const {
901bdd1243dSDimitry Andric   if (DAG.getMachineFunction().getFunction().getCallingConv() ==
902bdd1243dSDimitry Andric       CallingConv::GHC)
903bdd1243dSDimitry Andric     report_fatal_error("In GHC calling convention TLS is not supported");
904bdd1243dSDimitry Andric 
90506c3fb27SDimitry Andric   bool Large = DAG.getTarget().getCodeModel() == CodeModel::Large;
90606c3fb27SDimitry Andric   assert((!Large || Subtarget.is64Bit()) && "Large code model requires LA64");
90706c3fb27SDimitry Andric 
908bdd1243dSDimitry Andric   GlobalAddressSDNode *N = cast<GlobalAddressSDNode>(Op);
909bdd1243dSDimitry Andric   assert(N->getOffset() == 0 && "unexpected offset in global node");
910bdd1243dSDimitry Andric 
911bdd1243dSDimitry Andric   SDValue Addr;
912bdd1243dSDimitry Andric   switch (getTargetMachine().getTLSModel(N->getGlobal())) {
913bdd1243dSDimitry Andric   case TLSModel::GeneralDynamic:
914bdd1243dSDimitry Andric     // In this model, application code calls the dynamic linker function
915bdd1243dSDimitry Andric     // __tls_get_addr to locate TLS offsets into the dynamic thread vector at
916bdd1243dSDimitry Andric     // runtime.
91706c3fb27SDimitry Andric     Addr = getDynamicTLSAddr(N, DAG,
91806c3fb27SDimitry Andric                              Large ? LoongArch::PseudoLA_TLS_GD_LARGE
91906c3fb27SDimitry Andric                                    : LoongArch::PseudoLA_TLS_GD,
92006c3fb27SDimitry Andric                              Large);
921bdd1243dSDimitry Andric     break;
922bdd1243dSDimitry Andric   case TLSModel::LocalDynamic:
923bdd1243dSDimitry Andric     // Same as GeneralDynamic, except for assembly modifiers and relocation
924bdd1243dSDimitry Andric     // records.
92506c3fb27SDimitry Andric     Addr = getDynamicTLSAddr(N, DAG,
92606c3fb27SDimitry Andric                              Large ? LoongArch::PseudoLA_TLS_LD_LARGE
92706c3fb27SDimitry Andric                                    : LoongArch::PseudoLA_TLS_LD,
92806c3fb27SDimitry Andric                              Large);
929bdd1243dSDimitry Andric     break;
930bdd1243dSDimitry Andric   case TLSModel::InitialExec:
931bdd1243dSDimitry Andric     // This model uses the GOT to resolve TLS offsets.
93206c3fb27SDimitry Andric     Addr = getStaticTLSAddr(N, DAG,
93306c3fb27SDimitry Andric                             Large ? LoongArch::PseudoLA_TLS_IE_LARGE
93406c3fb27SDimitry Andric                                   : LoongArch::PseudoLA_TLS_IE,
93506c3fb27SDimitry Andric                             Large);
936bdd1243dSDimitry Andric     break;
937bdd1243dSDimitry Andric   case TLSModel::LocalExec:
938bdd1243dSDimitry Andric     // This model is used when static linking as the TLS offsets are resolved
939bdd1243dSDimitry Andric     // during program linking.
94006c3fb27SDimitry Andric     //
94106c3fb27SDimitry Andric     // This node doesn't need an extra argument for the large code model.
942bdd1243dSDimitry Andric     Addr = getStaticTLSAddr(N, DAG, LoongArch::PseudoLA_TLS_LE);
943bdd1243dSDimitry Andric     break;
944bdd1243dSDimitry Andric   }
945bdd1243dSDimitry Andric 
946753f127fSDimitry Andric   return Addr;
947753f127fSDimitry Andric }
948bdd1243dSDimitry Andric 
9495f757f3fSDimitry Andric template <unsigned N>
9505f757f3fSDimitry Andric static SDValue checkIntrinsicImmArg(SDValue Op, unsigned ImmOp,
9515f757f3fSDimitry Andric                                     SelectionDAG &DAG, bool IsSigned = false) {
9525f757f3fSDimitry Andric   auto *CImm = cast<ConstantSDNode>(Op->getOperand(ImmOp));
9535f757f3fSDimitry Andric   // Check the ImmArg.
9545f757f3fSDimitry Andric   if ((IsSigned && !isInt<N>(CImm->getSExtValue())) ||
9555f757f3fSDimitry Andric       (!IsSigned && !isUInt<N>(CImm->getZExtValue()))) {
9565f757f3fSDimitry Andric     DAG.getContext()->emitError(Op->getOperationName(0) +
9575f757f3fSDimitry Andric                                 ": argument out of range.");
9585f757f3fSDimitry Andric     return DAG.getNode(ISD::UNDEF, SDLoc(Op), Op.getValueType());
9595f757f3fSDimitry Andric   }
9605f757f3fSDimitry Andric   return SDValue();
9615f757f3fSDimitry Andric }
9625f757f3fSDimitry Andric 
963bdd1243dSDimitry Andric SDValue
964bdd1243dSDimitry Andric LoongArchTargetLowering::lowerINTRINSIC_WO_CHAIN(SDValue Op,
965bdd1243dSDimitry Andric                                                  SelectionDAG &DAG) const {
9665f757f3fSDimitry Andric   SDLoc DL(Op);
967bdd1243dSDimitry Andric   switch (Op.getConstantOperandVal(0)) {
968bdd1243dSDimitry Andric   default:
969bdd1243dSDimitry Andric     return SDValue(); // Don't custom lower most intrinsics.
970bdd1243dSDimitry Andric   case Intrinsic::thread_pointer: {
971bdd1243dSDimitry Andric     EVT PtrVT = getPointerTy(DAG.getDataLayout());
972bdd1243dSDimitry Andric     return DAG.getRegister(LoongArch::R2, PtrVT);
973bdd1243dSDimitry Andric   }
9745f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vpickve2gr_d:
9755f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vpickve2gr_du:
9765f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vreplvei_d:
9775f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvrepl128vei_d:
9785f757f3fSDimitry Andric     return checkIntrinsicImmArg<1>(Op, 2, DAG);
9795f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vreplvei_w:
9805f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvrepl128vei_w:
9815f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvpickve2gr_d:
9825f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvpickve2gr_du:
9835f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvpickve_d:
9845f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvpickve_d_f:
9855f757f3fSDimitry Andric     return checkIntrinsicImmArg<2>(Op, 2, DAG);
9865f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvinsve0_d:
9875f757f3fSDimitry Andric     return checkIntrinsicImmArg<2>(Op, 3, DAG);
9885f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vsat_b:
9895f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vsat_bu:
9905f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vrotri_b:
9915f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vsllwil_h_b:
9925f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vsllwil_hu_bu:
9935f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vsrlri_b:
9945f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vsrari_b:
9955f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vreplvei_h:
9965f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvsat_b:
9975f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvsat_bu:
9985f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvrotri_b:
9995f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvsllwil_h_b:
10005f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvsllwil_hu_bu:
10015f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvsrlri_b:
10025f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvsrari_b:
10035f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvrepl128vei_h:
10045f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvpickve_w:
10055f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvpickve_w_f:
10065f757f3fSDimitry Andric     return checkIntrinsicImmArg<3>(Op, 2, DAG);
10075f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvinsve0_w:
10085f757f3fSDimitry Andric     return checkIntrinsicImmArg<3>(Op, 3, DAG);
10095f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vsat_h:
10105f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vsat_hu:
10115f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vrotri_h:
10125f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vsllwil_w_h:
10135f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vsllwil_wu_hu:
10145f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vsrlri_h:
10155f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vsrari_h:
10165f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vreplvei_b:
10175f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvsat_h:
10185f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvsat_hu:
10195f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvrotri_h:
10205f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvsllwil_w_h:
10215f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvsllwil_wu_hu:
10225f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvsrlri_h:
10235f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvsrari_h:
10245f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvrepl128vei_b:
10255f757f3fSDimitry Andric     return checkIntrinsicImmArg<4>(Op, 2, DAG);
10265f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vsrlni_b_h:
10275f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vsrani_b_h:
10285f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vsrlrni_b_h:
10295f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vsrarni_b_h:
10305f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vssrlni_b_h:
10315f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vssrani_b_h:
10325f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vssrlni_bu_h:
10335f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vssrani_bu_h:
10345f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vssrlrni_b_h:
10355f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vssrarni_b_h:
10365f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vssrlrni_bu_h:
10375f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vssrarni_bu_h:
10385f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvsrlni_b_h:
10395f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvsrani_b_h:
10405f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvsrlrni_b_h:
10415f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvsrarni_b_h:
10425f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvssrlni_b_h:
10435f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvssrani_b_h:
10445f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvssrlni_bu_h:
10455f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvssrani_bu_h:
10465f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvssrlrni_b_h:
10475f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvssrarni_b_h:
10485f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvssrlrni_bu_h:
10495f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvssrarni_bu_h:
10505f757f3fSDimitry Andric     return checkIntrinsicImmArg<4>(Op, 3, DAG);
10515f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vsat_w:
10525f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vsat_wu:
10535f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vrotri_w:
10545f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vsllwil_d_w:
10555f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vsllwil_du_wu:
10565f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vsrlri_w:
10575f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vsrari_w:
10585f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vslei_bu:
10595f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vslei_hu:
10605f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vslei_wu:
10615f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vslei_du:
10625f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vslti_bu:
10635f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vslti_hu:
10645f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vslti_wu:
10655f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vslti_du:
10665f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vbsll_v:
10675f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vbsrl_v:
10685f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvsat_w:
10695f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvsat_wu:
10705f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvrotri_w:
10715f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvsllwil_d_w:
10725f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvsllwil_du_wu:
10735f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvsrlri_w:
10745f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvsrari_w:
10755f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvslei_bu:
10765f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvslei_hu:
10775f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvslei_wu:
10785f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvslei_du:
10795f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvslti_bu:
10805f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvslti_hu:
10815f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvslti_wu:
10825f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvslti_du:
10835f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvbsll_v:
10845f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvbsrl_v:
10855f757f3fSDimitry Andric     return checkIntrinsicImmArg<5>(Op, 2, DAG);
10865f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vseqi_b:
10875f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vseqi_h:
10885f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vseqi_w:
10895f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vseqi_d:
10905f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vslei_b:
10915f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vslei_h:
10925f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vslei_w:
10935f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vslei_d:
10945f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vslti_b:
10955f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vslti_h:
10965f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vslti_w:
10975f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vslti_d:
10985f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvseqi_b:
10995f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvseqi_h:
11005f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvseqi_w:
11015f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvseqi_d:
11025f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvslei_b:
11035f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvslei_h:
11045f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvslei_w:
11055f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvslei_d:
11065f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvslti_b:
11075f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvslti_h:
11085f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvslti_w:
11095f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvslti_d:
11105f757f3fSDimitry Andric     return checkIntrinsicImmArg<5>(Op, 2, DAG, /*IsSigned=*/true);
11115f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vsrlni_h_w:
11125f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vsrani_h_w:
11135f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vsrlrni_h_w:
11145f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vsrarni_h_w:
11155f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vssrlni_h_w:
11165f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vssrani_h_w:
11175f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vssrlni_hu_w:
11185f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vssrani_hu_w:
11195f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vssrlrni_h_w:
11205f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vssrarni_h_w:
11215f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vssrlrni_hu_w:
11225f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vssrarni_hu_w:
11235f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vfrstpi_b:
11245f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vfrstpi_h:
11255f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvsrlni_h_w:
11265f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvsrani_h_w:
11275f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvsrlrni_h_w:
11285f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvsrarni_h_w:
11295f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvssrlni_h_w:
11305f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvssrani_h_w:
11315f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvssrlni_hu_w:
11325f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvssrani_hu_w:
11335f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvssrlrni_h_w:
11345f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvssrarni_h_w:
11355f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvssrlrni_hu_w:
11365f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvssrarni_hu_w:
11375f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvfrstpi_b:
11385f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvfrstpi_h:
11395f757f3fSDimitry Andric     return checkIntrinsicImmArg<5>(Op, 3, DAG);
11405f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vsat_d:
11415f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vsat_du:
11425f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vrotri_d:
11435f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vsrlri_d:
11445f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vsrari_d:
11455f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvsat_d:
11465f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvsat_du:
11475f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvrotri_d:
11485f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvsrlri_d:
11495f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvsrari_d:
11505f757f3fSDimitry Andric     return checkIntrinsicImmArg<6>(Op, 2, DAG);
11515f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vsrlni_w_d:
11525f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vsrani_w_d:
11535f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vsrlrni_w_d:
11545f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vsrarni_w_d:
11555f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vssrlni_w_d:
11565f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vssrani_w_d:
11575f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vssrlni_wu_d:
11585f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vssrani_wu_d:
11595f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vssrlrni_w_d:
11605f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vssrarni_w_d:
11615f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vssrlrni_wu_d:
11625f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vssrarni_wu_d:
11635f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvsrlni_w_d:
11645f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvsrani_w_d:
11655f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvsrlrni_w_d:
11665f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvsrarni_w_d:
11675f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvssrlni_w_d:
11685f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvssrani_w_d:
11695f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvssrlni_wu_d:
11705f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvssrani_wu_d:
11715f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvssrlrni_w_d:
11725f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvssrarni_w_d:
11735f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvssrlrni_wu_d:
11745f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvssrarni_wu_d:
11755f757f3fSDimitry Andric     return checkIntrinsicImmArg<6>(Op, 3, DAG);
11765f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vsrlni_d_q:
11775f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vsrani_d_q:
11785f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vsrlrni_d_q:
11795f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vsrarni_d_q:
11805f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vssrlni_d_q:
11815f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vssrani_d_q:
11825f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vssrlni_du_q:
11835f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vssrani_du_q:
11845f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vssrlrni_d_q:
11855f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vssrarni_d_q:
11865f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vssrlrni_du_q:
11875f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vssrarni_du_q:
11885f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvsrlni_d_q:
11895f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvsrani_d_q:
11905f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvsrlrni_d_q:
11915f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvsrarni_d_q:
11925f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvssrlni_d_q:
11935f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvssrani_d_q:
11945f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvssrlni_du_q:
11955f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvssrani_du_q:
11965f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvssrlrni_d_q:
11975f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvssrarni_d_q:
11985f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvssrlrni_du_q:
11995f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvssrarni_du_q:
12005f757f3fSDimitry Andric     return checkIntrinsicImmArg<7>(Op, 3, DAG);
12015f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vnori_b:
12025f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vshuf4i_b:
12035f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vshuf4i_h:
12045f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vshuf4i_w:
12055f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvnori_b:
12065f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvshuf4i_b:
12075f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvshuf4i_h:
12085f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvshuf4i_w:
12095f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvpermi_d:
12105f757f3fSDimitry Andric     return checkIntrinsicImmArg<8>(Op, 2, DAG);
12115f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vshuf4i_d:
12125f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vpermi_w:
12135f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vbitseli_b:
12145f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vextrins_b:
12155f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vextrins_h:
12165f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vextrins_w:
12175f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vextrins_d:
12185f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvshuf4i_d:
12195f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvpermi_w:
12205f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvpermi_q:
12215f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvbitseli_b:
12225f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvextrins_b:
12235f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvextrins_h:
12245f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvextrins_w:
12255f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvextrins_d:
12265f757f3fSDimitry Andric     return checkIntrinsicImmArg<8>(Op, 3, DAG);
12275f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vrepli_b:
12285f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vrepli_h:
12295f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vrepli_w:
12305f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vrepli_d:
12315f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvrepli_b:
12325f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvrepli_h:
12335f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvrepli_w:
12345f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvrepli_d:
12355f757f3fSDimitry Andric     return checkIntrinsicImmArg<10>(Op, 1, DAG, /*IsSigned=*/true);
12365f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vldi:
12375f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvldi:
12385f757f3fSDimitry Andric     return checkIntrinsicImmArg<13>(Op, 1, DAG, /*IsSigned=*/true);
1239bdd1243dSDimitry Andric   }
1240bdd1243dSDimitry Andric }
1241bdd1243dSDimitry Andric 
124206c3fb27SDimitry Andric // Helper function that emits error message for intrinsics with chain and return
124306c3fb27SDimitry Andric // merge values of a UNDEF and the chain.
1244bdd1243dSDimitry Andric static SDValue emitIntrinsicWithChainErrorMessage(SDValue Op,
1245bdd1243dSDimitry Andric                                                   StringRef ErrorMsg,
1246bdd1243dSDimitry Andric                                                   SelectionDAG &DAG) {
124706c3fb27SDimitry Andric   DAG.getContext()->emitError(Op->getOperationName(0) + ": " + ErrorMsg + ".");
1248bdd1243dSDimitry Andric   return DAG.getMergeValues({DAG.getUNDEF(Op.getValueType()), Op.getOperand(0)},
1249bdd1243dSDimitry Andric                             SDLoc(Op));
1250bdd1243dSDimitry Andric }
1251bdd1243dSDimitry Andric 
1252bdd1243dSDimitry Andric SDValue
1253bdd1243dSDimitry Andric LoongArchTargetLowering::lowerINTRINSIC_W_CHAIN(SDValue Op,
1254bdd1243dSDimitry Andric                                                 SelectionDAG &DAG) const {
1255bdd1243dSDimitry Andric   SDLoc DL(Op);
1256bdd1243dSDimitry Andric   MVT GRLenVT = Subtarget.getGRLenVT();
125706c3fb27SDimitry Andric   EVT VT = Op.getValueType();
125806c3fb27SDimitry Andric   SDValue Chain = Op.getOperand(0);
125906c3fb27SDimitry Andric   const StringRef ErrorMsgOOR = "argument out of range";
126006c3fb27SDimitry Andric   const StringRef ErrorMsgReqLA64 = "requires loongarch64";
126106c3fb27SDimitry Andric   const StringRef ErrorMsgReqF = "requires basic 'f' target feature";
1262bdd1243dSDimitry Andric 
1263bdd1243dSDimitry Andric   switch (Op.getConstantOperandVal(1)) {
1264bdd1243dSDimitry Andric   default:
1265bdd1243dSDimitry Andric     return Op;
1266bdd1243dSDimitry Andric   case Intrinsic::loongarch_crc_w_b_w:
1267bdd1243dSDimitry Andric   case Intrinsic::loongarch_crc_w_h_w:
1268bdd1243dSDimitry Andric   case Intrinsic::loongarch_crc_w_w_w:
1269bdd1243dSDimitry Andric   case Intrinsic::loongarch_crc_w_d_w:
1270bdd1243dSDimitry Andric   case Intrinsic::loongarch_crcc_w_b_w:
1271bdd1243dSDimitry Andric   case Intrinsic::loongarch_crcc_w_h_w:
1272bdd1243dSDimitry Andric   case Intrinsic::loongarch_crcc_w_w_w:
127306c3fb27SDimitry Andric   case Intrinsic::loongarch_crcc_w_d_w:
127406c3fb27SDimitry Andric     return emitIntrinsicWithChainErrorMessage(Op, ErrorMsgReqLA64, DAG);
1275bdd1243dSDimitry Andric   case Intrinsic::loongarch_csrrd_w:
1276bdd1243dSDimitry Andric   case Intrinsic::loongarch_csrrd_d: {
1277647cbc5dSDimitry Andric     unsigned Imm = Op.getConstantOperandVal(2);
127806c3fb27SDimitry Andric     return !isUInt<14>(Imm)
127906c3fb27SDimitry Andric                ? emitIntrinsicWithChainErrorMessage(Op, ErrorMsgOOR, DAG)
128006c3fb27SDimitry Andric                : DAG.getNode(LoongArchISD::CSRRD, DL, {GRLenVT, MVT::Other},
128106c3fb27SDimitry Andric                              {Chain, DAG.getConstant(Imm, DL, GRLenVT)});
1282bdd1243dSDimitry Andric   }
1283bdd1243dSDimitry Andric   case Intrinsic::loongarch_csrwr_w:
1284bdd1243dSDimitry Andric   case Intrinsic::loongarch_csrwr_d: {
1285647cbc5dSDimitry Andric     unsigned Imm = Op.getConstantOperandVal(3);
128606c3fb27SDimitry Andric     return !isUInt<14>(Imm)
128706c3fb27SDimitry Andric                ? emitIntrinsicWithChainErrorMessage(Op, ErrorMsgOOR, DAG)
128806c3fb27SDimitry Andric                : DAG.getNode(LoongArchISD::CSRWR, DL, {GRLenVT, MVT::Other},
128906c3fb27SDimitry Andric                              {Chain, Op.getOperand(2),
129006c3fb27SDimitry Andric                               DAG.getConstant(Imm, DL, GRLenVT)});
1291bdd1243dSDimitry Andric   }
1292bdd1243dSDimitry Andric   case Intrinsic::loongarch_csrxchg_w:
1293bdd1243dSDimitry Andric   case Intrinsic::loongarch_csrxchg_d: {
1294647cbc5dSDimitry Andric     unsigned Imm = Op.getConstantOperandVal(4);
129506c3fb27SDimitry Andric     return !isUInt<14>(Imm)
129606c3fb27SDimitry Andric                ? emitIntrinsicWithChainErrorMessage(Op, ErrorMsgOOR, DAG)
129706c3fb27SDimitry Andric                : DAG.getNode(LoongArchISD::CSRXCHG, DL, {GRLenVT, MVT::Other},
129806c3fb27SDimitry Andric                              {Chain, Op.getOperand(2), Op.getOperand(3),
129906c3fb27SDimitry Andric                               DAG.getConstant(Imm, DL, GRLenVT)});
1300bdd1243dSDimitry Andric   }
1301bdd1243dSDimitry Andric   case Intrinsic::loongarch_iocsrrd_d: {
130206c3fb27SDimitry Andric     return DAG.getNode(
130306c3fb27SDimitry Andric         LoongArchISD::IOCSRRD_D, DL, {GRLenVT, MVT::Other},
130406c3fb27SDimitry Andric         {Chain, DAG.getNode(ISD::ANY_EXTEND, DL, MVT::i64, Op.getOperand(2))});
1305bdd1243dSDimitry Andric   }
1306bdd1243dSDimitry Andric #define IOCSRRD_CASE(NAME, NODE)                                               \
1307bdd1243dSDimitry Andric   case Intrinsic::loongarch_##NAME: {                                          \
130806c3fb27SDimitry Andric     return DAG.getNode(LoongArchISD::NODE, DL, {GRLenVT, MVT::Other},          \
130906c3fb27SDimitry Andric                        {Chain, Op.getOperand(2)});                             \
1310bdd1243dSDimitry Andric   }
1311bdd1243dSDimitry Andric     IOCSRRD_CASE(iocsrrd_b, IOCSRRD_B);
1312bdd1243dSDimitry Andric     IOCSRRD_CASE(iocsrrd_h, IOCSRRD_H);
1313bdd1243dSDimitry Andric     IOCSRRD_CASE(iocsrrd_w, IOCSRRD_W);
1314bdd1243dSDimitry Andric #undef IOCSRRD_CASE
1315bdd1243dSDimitry Andric   case Intrinsic::loongarch_cpucfg: {
131606c3fb27SDimitry Andric     return DAG.getNode(LoongArchISD::CPUCFG, DL, {GRLenVT, MVT::Other},
131706c3fb27SDimitry Andric                        {Chain, Op.getOperand(2)});
1318bdd1243dSDimitry Andric   }
1319bdd1243dSDimitry Andric   case Intrinsic::loongarch_lddir_d: {
1320647cbc5dSDimitry Andric     unsigned Imm = Op.getConstantOperandVal(3);
132106c3fb27SDimitry Andric     return !isUInt<8>(Imm)
132206c3fb27SDimitry Andric                ? emitIntrinsicWithChainErrorMessage(Op, ErrorMsgOOR, DAG)
132306c3fb27SDimitry Andric                : Op;
1324bdd1243dSDimitry Andric   }
1325bdd1243dSDimitry Andric   case Intrinsic::loongarch_movfcsr2gr: {
132606c3fb27SDimitry Andric     if (!Subtarget.hasBasicF())
132706c3fb27SDimitry Andric       return emitIntrinsicWithChainErrorMessage(Op, ErrorMsgReqF, DAG);
1328647cbc5dSDimitry Andric     unsigned Imm = Op.getConstantOperandVal(2);
132906c3fb27SDimitry Andric     return !isUInt<2>(Imm)
133006c3fb27SDimitry Andric                ? emitIntrinsicWithChainErrorMessage(Op, ErrorMsgOOR, DAG)
133106c3fb27SDimitry Andric                : DAG.getNode(LoongArchISD::MOVFCSR2GR, DL, {VT, MVT::Other},
133206c3fb27SDimitry Andric                              {Chain, DAG.getConstant(Imm, DL, GRLenVT)});
1333bdd1243dSDimitry Andric   }
13345f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vld:
13355f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vldrepl_b:
13365f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvld:
13375f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvldrepl_b:
13385f757f3fSDimitry Andric     return !isInt<12>(cast<ConstantSDNode>(Op.getOperand(3))->getSExtValue())
13395f757f3fSDimitry Andric                ? emitIntrinsicWithChainErrorMessage(Op, ErrorMsgOOR, DAG)
13405f757f3fSDimitry Andric                : SDValue();
13415f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vldrepl_h:
13425f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvldrepl_h:
13435f757f3fSDimitry Andric     return !isShiftedInt<11, 1>(
13445f757f3fSDimitry Andric                cast<ConstantSDNode>(Op.getOperand(3))->getSExtValue())
13455f757f3fSDimitry Andric                ? emitIntrinsicWithChainErrorMessage(
13465f757f3fSDimitry Andric                      Op, "argument out of range or not a multiple of 2", DAG)
13475f757f3fSDimitry Andric                : SDValue();
13485f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vldrepl_w:
13495f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvldrepl_w:
13505f757f3fSDimitry Andric     return !isShiftedInt<10, 2>(
13515f757f3fSDimitry Andric                cast<ConstantSDNode>(Op.getOperand(3))->getSExtValue())
13525f757f3fSDimitry Andric                ? emitIntrinsicWithChainErrorMessage(
13535f757f3fSDimitry Andric                      Op, "argument out of range or not a multiple of 4", DAG)
13545f757f3fSDimitry Andric                : SDValue();
13555f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vldrepl_d:
13565f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvldrepl_d:
13575f757f3fSDimitry Andric     return !isShiftedInt<9, 3>(
13585f757f3fSDimitry Andric                cast<ConstantSDNode>(Op.getOperand(3))->getSExtValue())
13595f757f3fSDimitry Andric                ? emitIntrinsicWithChainErrorMessage(
13605f757f3fSDimitry Andric                      Op, "argument out of range or not a multiple of 8", DAG)
13615f757f3fSDimitry Andric                : SDValue();
1362bdd1243dSDimitry Andric   }
1363bdd1243dSDimitry Andric }
1364bdd1243dSDimitry Andric 
1365bdd1243dSDimitry Andric // Helper function that emits error message for intrinsics with void return
136606c3fb27SDimitry Andric // value and return the chain.
1367bdd1243dSDimitry Andric static SDValue emitIntrinsicErrorMessage(SDValue Op, StringRef ErrorMsg,
1368bdd1243dSDimitry Andric                                          SelectionDAG &DAG) {
1369bdd1243dSDimitry Andric 
137006c3fb27SDimitry Andric   DAG.getContext()->emitError(Op->getOperationName(0) + ": " + ErrorMsg + ".");
1371bdd1243dSDimitry Andric   return Op.getOperand(0);
1372bdd1243dSDimitry Andric }
1373bdd1243dSDimitry Andric 
1374bdd1243dSDimitry Andric SDValue LoongArchTargetLowering::lowerINTRINSIC_VOID(SDValue Op,
1375bdd1243dSDimitry Andric                                                      SelectionDAG &DAG) const {
1376bdd1243dSDimitry Andric   SDLoc DL(Op);
1377bdd1243dSDimitry Andric   MVT GRLenVT = Subtarget.getGRLenVT();
137806c3fb27SDimitry Andric   SDValue Chain = Op.getOperand(0);
1379bdd1243dSDimitry Andric   uint64_t IntrinsicEnum = Op.getConstantOperandVal(1);
1380bdd1243dSDimitry Andric   SDValue Op2 = Op.getOperand(2);
138106c3fb27SDimitry Andric   const StringRef ErrorMsgOOR = "argument out of range";
138206c3fb27SDimitry Andric   const StringRef ErrorMsgReqLA64 = "requires loongarch64";
138306c3fb27SDimitry Andric   const StringRef ErrorMsgReqLA32 = "requires loongarch32";
138406c3fb27SDimitry Andric   const StringRef ErrorMsgReqF = "requires basic 'f' target feature";
1385bdd1243dSDimitry Andric 
1386bdd1243dSDimitry Andric   switch (IntrinsicEnum) {
1387bdd1243dSDimitry Andric   default:
1388bdd1243dSDimitry Andric     // TODO: Add more Intrinsics.
1389bdd1243dSDimitry Andric     return SDValue();
1390bdd1243dSDimitry Andric   case Intrinsic::loongarch_cacop_d:
1391bdd1243dSDimitry Andric   case Intrinsic::loongarch_cacop_w: {
139206c3fb27SDimitry Andric     if (IntrinsicEnum == Intrinsic::loongarch_cacop_d && !Subtarget.is64Bit())
139306c3fb27SDimitry Andric       return emitIntrinsicErrorMessage(Op, ErrorMsgReqLA64, DAG);
139406c3fb27SDimitry Andric     if (IntrinsicEnum == Intrinsic::loongarch_cacop_w && Subtarget.is64Bit())
139506c3fb27SDimitry Andric       return emitIntrinsicErrorMessage(Op, ErrorMsgReqLA32, DAG);
1396bdd1243dSDimitry Andric     // call void @llvm.loongarch.cacop.[d/w](uimm5, rj, simm12)
1397*1db9f3b2SDimitry Andric     unsigned Imm1 = Op2->getAsZExtVal();
139806c3fb27SDimitry Andric     int Imm2 = cast<ConstantSDNode>(Op.getOperand(4))->getSExtValue();
139906c3fb27SDimitry Andric     if (!isUInt<5>(Imm1) || !isInt<12>(Imm2))
1400bdd1243dSDimitry Andric       return emitIntrinsicErrorMessage(Op, ErrorMsgOOR, DAG);
1401bdd1243dSDimitry Andric     return Op;
1402bdd1243dSDimitry Andric   }
1403bdd1243dSDimitry Andric   case Intrinsic::loongarch_dbar: {
1404*1db9f3b2SDimitry Andric     unsigned Imm = Op2->getAsZExtVal();
140506c3fb27SDimitry Andric     return !isUInt<15>(Imm)
140606c3fb27SDimitry Andric                ? emitIntrinsicErrorMessage(Op, ErrorMsgOOR, DAG)
140706c3fb27SDimitry Andric                : DAG.getNode(LoongArchISD::DBAR, DL, MVT::Other, Chain,
1408bdd1243dSDimitry Andric                              DAG.getConstant(Imm, DL, GRLenVT));
1409bdd1243dSDimitry Andric   }
1410bdd1243dSDimitry Andric   case Intrinsic::loongarch_ibar: {
1411*1db9f3b2SDimitry Andric     unsigned Imm = Op2->getAsZExtVal();
141206c3fb27SDimitry Andric     return !isUInt<15>(Imm)
141306c3fb27SDimitry Andric                ? emitIntrinsicErrorMessage(Op, ErrorMsgOOR, DAG)
141406c3fb27SDimitry Andric                : DAG.getNode(LoongArchISD::IBAR, DL, MVT::Other, Chain,
1415bdd1243dSDimitry Andric                              DAG.getConstant(Imm, DL, GRLenVT));
1416bdd1243dSDimitry Andric   }
1417bdd1243dSDimitry Andric   case Intrinsic::loongarch_break: {
1418*1db9f3b2SDimitry Andric     unsigned Imm = Op2->getAsZExtVal();
141906c3fb27SDimitry Andric     return !isUInt<15>(Imm)
142006c3fb27SDimitry Andric                ? emitIntrinsicErrorMessage(Op, ErrorMsgOOR, DAG)
142106c3fb27SDimitry Andric                : DAG.getNode(LoongArchISD::BREAK, DL, MVT::Other, Chain,
1422bdd1243dSDimitry Andric                              DAG.getConstant(Imm, DL, GRLenVT));
1423bdd1243dSDimitry Andric   }
1424bdd1243dSDimitry Andric   case Intrinsic::loongarch_movgr2fcsr: {
142506c3fb27SDimitry Andric     if (!Subtarget.hasBasicF())
142606c3fb27SDimitry Andric       return emitIntrinsicErrorMessage(Op, ErrorMsgReqF, DAG);
1427*1db9f3b2SDimitry Andric     unsigned Imm = Op2->getAsZExtVal();
142806c3fb27SDimitry Andric     return !isUInt<2>(Imm)
142906c3fb27SDimitry Andric                ? emitIntrinsicErrorMessage(Op, ErrorMsgOOR, DAG)
143006c3fb27SDimitry Andric                : DAG.getNode(LoongArchISD::MOVGR2FCSR, DL, MVT::Other, Chain,
1431bdd1243dSDimitry Andric                              DAG.getConstant(Imm, DL, GRLenVT),
143206c3fb27SDimitry Andric                              DAG.getNode(ISD::ANY_EXTEND, DL, GRLenVT,
143306c3fb27SDimitry Andric                                          Op.getOperand(3)));
1434bdd1243dSDimitry Andric   }
1435bdd1243dSDimitry Andric   case Intrinsic::loongarch_syscall: {
1436*1db9f3b2SDimitry Andric     unsigned Imm = Op2->getAsZExtVal();
143706c3fb27SDimitry Andric     return !isUInt<15>(Imm)
143806c3fb27SDimitry Andric                ? emitIntrinsicErrorMessage(Op, ErrorMsgOOR, DAG)
143906c3fb27SDimitry Andric                : DAG.getNode(LoongArchISD::SYSCALL, DL, MVT::Other, Chain,
1440bdd1243dSDimitry Andric                              DAG.getConstant(Imm, DL, GRLenVT));
1441bdd1243dSDimitry Andric   }
1442bdd1243dSDimitry Andric #define IOCSRWR_CASE(NAME, NODE)                                               \
1443bdd1243dSDimitry Andric   case Intrinsic::loongarch_##NAME: {                                          \
1444bdd1243dSDimitry Andric     SDValue Op3 = Op.getOperand(3);                                            \
144506c3fb27SDimitry Andric     return Subtarget.is64Bit()                                                 \
144606c3fb27SDimitry Andric                ? DAG.getNode(LoongArchISD::NODE, DL, MVT::Other, Chain,        \
1447bdd1243dSDimitry Andric                              DAG.getNode(ISD::ANY_EXTEND, DL, MVT::i64, Op2),  \
144806c3fb27SDimitry Andric                              DAG.getNode(ISD::ANY_EXTEND, DL, MVT::i64, Op3))  \
144906c3fb27SDimitry Andric                : DAG.getNode(LoongArchISD::NODE, DL, MVT::Other, Chain, Op2,   \
145006c3fb27SDimitry Andric                              Op3);                                             \
1451bdd1243dSDimitry Andric   }
1452bdd1243dSDimitry Andric     IOCSRWR_CASE(iocsrwr_b, IOCSRWR_B);
1453bdd1243dSDimitry Andric     IOCSRWR_CASE(iocsrwr_h, IOCSRWR_H);
1454bdd1243dSDimitry Andric     IOCSRWR_CASE(iocsrwr_w, IOCSRWR_W);
1455bdd1243dSDimitry Andric #undef IOCSRWR_CASE
1456bdd1243dSDimitry Andric   case Intrinsic::loongarch_iocsrwr_d: {
145706c3fb27SDimitry Andric     return !Subtarget.is64Bit()
145806c3fb27SDimitry Andric                ? emitIntrinsicErrorMessage(Op, ErrorMsgReqLA64, DAG)
145906c3fb27SDimitry Andric                : DAG.getNode(LoongArchISD::IOCSRWR_D, DL, MVT::Other, Chain,
146006c3fb27SDimitry Andric                              Op2,
146106c3fb27SDimitry Andric                              DAG.getNode(ISD::ANY_EXTEND, DL, MVT::i64,
146206c3fb27SDimitry Andric                                          Op.getOperand(3)));
1463bdd1243dSDimitry Andric   }
1464bdd1243dSDimitry Andric #define ASRT_LE_GT_CASE(NAME)                                                  \
1465bdd1243dSDimitry Andric   case Intrinsic::loongarch_##NAME: {                                          \
146606c3fb27SDimitry Andric     return !Subtarget.is64Bit()                                                \
146706c3fb27SDimitry Andric                ? emitIntrinsicErrorMessage(Op, ErrorMsgReqLA64, DAG)           \
146806c3fb27SDimitry Andric                : Op;                                                           \
1469bdd1243dSDimitry Andric   }
1470bdd1243dSDimitry Andric     ASRT_LE_GT_CASE(asrtle_d)
1471bdd1243dSDimitry Andric     ASRT_LE_GT_CASE(asrtgt_d)
1472bdd1243dSDimitry Andric #undef ASRT_LE_GT_CASE
1473bdd1243dSDimitry Andric   case Intrinsic::loongarch_ldpte_d: {
1474647cbc5dSDimitry Andric     unsigned Imm = Op.getConstantOperandVal(3);
147506c3fb27SDimitry Andric     return !Subtarget.is64Bit()
147606c3fb27SDimitry Andric                ? emitIntrinsicErrorMessage(Op, ErrorMsgReqLA64, DAG)
147706c3fb27SDimitry Andric            : !isUInt<8>(Imm) ? emitIntrinsicErrorMessage(Op, ErrorMsgOOR, DAG)
147806c3fb27SDimitry Andric                              : Op;
1479bdd1243dSDimitry Andric   }
14805f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vst:
14815f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvst:
14825f757f3fSDimitry Andric     return !isInt<12>(cast<ConstantSDNode>(Op.getOperand(4))->getSExtValue())
14835f757f3fSDimitry Andric                ? emitIntrinsicErrorMessage(Op, ErrorMsgOOR, DAG)
14845f757f3fSDimitry Andric                : SDValue();
14855f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvstelm_b:
14865f757f3fSDimitry Andric     return (!isInt<8>(cast<ConstantSDNode>(Op.getOperand(4))->getSExtValue()) ||
1487647cbc5dSDimitry Andric             !isUInt<5>(Op.getConstantOperandVal(5)))
14885f757f3fSDimitry Andric                ? emitIntrinsicErrorMessage(Op, ErrorMsgOOR, DAG)
14895f757f3fSDimitry Andric                : SDValue();
14905f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vstelm_b:
14915f757f3fSDimitry Andric     return (!isInt<8>(cast<ConstantSDNode>(Op.getOperand(4))->getSExtValue()) ||
1492647cbc5dSDimitry Andric             !isUInt<4>(Op.getConstantOperandVal(5)))
14935f757f3fSDimitry Andric                ? emitIntrinsicErrorMessage(Op, ErrorMsgOOR, DAG)
14945f757f3fSDimitry Andric                : SDValue();
14955f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvstelm_h:
14965f757f3fSDimitry Andric     return (!isShiftedInt<8, 1>(
14975f757f3fSDimitry Andric                 cast<ConstantSDNode>(Op.getOperand(4))->getSExtValue()) ||
1498647cbc5dSDimitry Andric             !isUInt<4>(Op.getConstantOperandVal(5)))
14995f757f3fSDimitry Andric                ? emitIntrinsicErrorMessage(
15005f757f3fSDimitry Andric                      Op, "argument out of range or not a multiple of 2", DAG)
15015f757f3fSDimitry Andric                : SDValue();
15025f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vstelm_h:
15035f757f3fSDimitry Andric     return (!isShiftedInt<8, 1>(
15045f757f3fSDimitry Andric                 cast<ConstantSDNode>(Op.getOperand(4))->getSExtValue()) ||
1505647cbc5dSDimitry Andric             !isUInt<3>(Op.getConstantOperandVal(5)))
15065f757f3fSDimitry Andric                ? emitIntrinsicErrorMessage(
15075f757f3fSDimitry Andric                      Op, "argument out of range or not a multiple of 2", DAG)
15085f757f3fSDimitry Andric                : SDValue();
15095f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvstelm_w:
15105f757f3fSDimitry Andric     return (!isShiftedInt<8, 2>(
15115f757f3fSDimitry Andric                 cast<ConstantSDNode>(Op.getOperand(4))->getSExtValue()) ||
1512647cbc5dSDimitry Andric             !isUInt<3>(Op.getConstantOperandVal(5)))
15135f757f3fSDimitry Andric                ? emitIntrinsicErrorMessage(
15145f757f3fSDimitry Andric                      Op, "argument out of range or not a multiple of 4", DAG)
15155f757f3fSDimitry Andric                : SDValue();
15165f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vstelm_w:
15175f757f3fSDimitry Andric     return (!isShiftedInt<8, 2>(
15185f757f3fSDimitry Andric                 cast<ConstantSDNode>(Op.getOperand(4))->getSExtValue()) ||
1519647cbc5dSDimitry Andric             !isUInt<2>(Op.getConstantOperandVal(5)))
15205f757f3fSDimitry Andric                ? emitIntrinsicErrorMessage(
15215f757f3fSDimitry Andric                      Op, "argument out of range or not a multiple of 4", DAG)
15225f757f3fSDimitry Andric                : SDValue();
15235f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvstelm_d:
15245f757f3fSDimitry Andric     return (!isShiftedInt<8, 3>(
15255f757f3fSDimitry Andric                 cast<ConstantSDNode>(Op.getOperand(4))->getSExtValue()) ||
1526647cbc5dSDimitry Andric             !isUInt<2>(Op.getConstantOperandVal(5)))
15275f757f3fSDimitry Andric                ? emitIntrinsicErrorMessage(
15285f757f3fSDimitry Andric                      Op, "argument out of range or not a multiple of 8", DAG)
15295f757f3fSDimitry Andric                : SDValue();
15305f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vstelm_d:
15315f757f3fSDimitry Andric     return (!isShiftedInt<8, 3>(
15325f757f3fSDimitry Andric                 cast<ConstantSDNode>(Op.getOperand(4))->getSExtValue()) ||
1533647cbc5dSDimitry Andric             !isUInt<1>(Op.getConstantOperandVal(5)))
15345f757f3fSDimitry Andric                ? emitIntrinsicErrorMessage(
15355f757f3fSDimitry Andric                      Op, "argument out of range or not a multiple of 8", DAG)
15365f757f3fSDimitry Andric                : SDValue();
1537bdd1243dSDimitry Andric   }
1538753f127fSDimitry Andric }
1539753f127fSDimitry Andric 
154081ad6265SDimitry Andric SDValue LoongArchTargetLowering::lowerShiftLeftParts(SDValue Op,
154181ad6265SDimitry Andric                                                      SelectionDAG &DAG) const {
154281ad6265SDimitry Andric   SDLoc DL(Op);
154381ad6265SDimitry Andric   SDValue Lo = Op.getOperand(0);
154481ad6265SDimitry Andric   SDValue Hi = Op.getOperand(1);
154581ad6265SDimitry Andric   SDValue Shamt = Op.getOperand(2);
154681ad6265SDimitry Andric   EVT VT = Lo.getValueType();
154781ad6265SDimitry Andric 
154881ad6265SDimitry Andric   // if Shamt-GRLen < 0: // Shamt < GRLen
154981ad6265SDimitry Andric   //   Lo = Lo << Shamt
155081ad6265SDimitry Andric   //   Hi = (Hi << Shamt) | ((Lo >>u 1) >>u (GRLen-1 ^ Shamt))
155181ad6265SDimitry Andric   // else:
155281ad6265SDimitry Andric   //   Lo = 0
155381ad6265SDimitry Andric   //   Hi = Lo << (Shamt-GRLen)
155481ad6265SDimitry Andric 
155581ad6265SDimitry Andric   SDValue Zero = DAG.getConstant(0, DL, VT);
155681ad6265SDimitry Andric   SDValue One = DAG.getConstant(1, DL, VT);
155781ad6265SDimitry Andric   SDValue MinusGRLen = DAG.getConstant(-(int)Subtarget.getGRLen(), DL, VT);
155881ad6265SDimitry Andric   SDValue GRLenMinus1 = DAG.getConstant(Subtarget.getGRLen() - 1, DL, VT);
155981ad6265SDimitry Andric   SDValue ShamtMinusGRLen = DAG.getNode(ISD::ADD, DL, VT, Shamt, MinusGRLen);
156081ad6265SDimitry Andric   SDValue GRLenMinus1Shamt = DAG.getNode(ISD::XOR, DL, VT, Shamt, GRLenMinus1);
156181ad6265SDimitry Andric 
156281ad6265SDimitry Andric   SDValue LoTrue = DAG.getNode(ISD::SHL, DL, VT, Lo, Shamt);
156381ad6265SDimitry Andric   SDValue ShiftRight1Lo = DAG.getNode(ISD::SRL, DL, VT, Lo, One);
156481ad6265SDimitry Andric   SDValue ShiftRightLo =
156581ad6265SDimitry Andric       DAG.getNode(ISD::SRL, DL, VT, ShiftRight1Lo, GRLenMinus1Shamt);
156681ad6265SDimitry Andric   SDValue ShiftLeftHi = DAG.getNode(ISD::SHL, DL, VT, Hi, Shamt);
156781ad6265SDimitry Andric   SDValue HiTrue = DAG.getNode(ISD::OR, DL, VT, ShiftLeftHi, ShiftRightLo);
156881ad6265SDimitry Andric   SDValue HiFalse = DAG.getNode(ISD::SHL, DL, VT, Lo, ShamtMinusGRLen);
156981ad6265SDimitry Andric 
157081ad6265SDimitry Andric   SDValue CC = DAG.getSetCC(DL, VT, ShamtMinusGRLen, Zero, ISD::SETLT);
157181ad6265SDimitry Andric 
157281ad6265SDimitry Andric   Lo = DAG.getNode(ISD::SELECT, DL, VT, CC, LoTrue, Zero);
157381ad6265SDimitry Andric   Hi = DAG.getNode(ISD::SELECT, DL, VT, CC, HiTrue, HiFalse);
157481ad6265SDimitry Andric 
157581ad6265SDimitry Andric   SDValue Parts[2] = {Lo, Hi};
157681ad6265SDimitry Andric   return DAG.getMergeValues(Parts, DL);
157781ad6265SDimitry Andric }
157881ad6265SDimitry Andric 
157981ad6265SDimitry Andric SDValue LoongArchTargetLowering::lowerShiftRightParts(SDValue Op,
158081ad6265SDimitry Andric                                                       SelectionDAG &DAG,
158181ad6265SDimitry Andric                                                       bool IsSRA) const {
158281ad6265SDimitry Andric   SDLoc DL(Op);
158381ad6265SDimitry Andric   SDValue Lo = Op.getOperand(0);
158481ad6265SDimitry Andric   SDValue Hi = Op.getOperand(1);
158581ad6265SDimitry Andric   SDValue Shamt = Op.getOperand(2);
158681ad6265SDimitry Andric   EVT VT = Lo.getValueType();
158781ad6265SDimitry Andric 
158881ad6265SDimitry Andric   // SRA expansion:
158981ad6265SDimitry Andric   //   if Shamt-GRLen < 0: // Shamt < GRLen
159081ad6265SDimitry Andric   //     Lo = (Lo >>u Shamt) | ((Hi << 1) << (ShAmt ^ GRLen-1))
159181ad6265SDimitry Andric   //     Hi = Hi >>s Shamt
159281ad6265SDimitry Andric   //   else:
159381ad6265SDimitry Andric   //     Lo = Hi >>s (Shamt-GRLen);
159481ad6265SDimitry Andric   //     Hi = Hi >>s (GRLen-1)
159581ad6265SDimitry Andric   //
159681ad6265SDimitry Andric   // SRL expansion:
159781ad6265SDimitry Andric   //   if Shamt-GRLen < 0: // Shamt < GRLen
159881ad6265SDimitry Andric   //     Lo = (Lo >>u Shamt) | ((Hi << 1) << (ShAmt ^ GRLen-1))
159981ad6265SDimitry Andric   //     Hi = Hi >>u Shamt
160081ad6265SDimitry Andric   //   else:
160181ad6265SDimitry Andric   //     Lo = Hi >>u (Shamt-GRLen);
160281ad6265SDimitry Andric   //     Hi = 0;
160381ad6265SDimitry Andric 
160481ad6265SDimitry Andric   unsigned ShiftRightOp = IsSRA ? ISD::SRA : ISD::SRL;
160581ad6265SDimitry Andric 
160681ad6265SDimitry Andric   SDValue Zero = DAG.getConstant(0, DL, VT);
160781ad6265SDimitry Andric   SDValue One = DAG.getConstant(1, DL, VT);
160881ad6265SDimitry Andric   SDValue MinusGRLen = DAG.getConstant(-(int)Subtarget.getGRLen(), DL, VT);
160981ad6265SDimitry Andric   SDValue GRLenMinus1 = DAG.getConstant(Subtarget.getGRLen() - 1, DL, VT);
161081ad6265SDimitry Andric   SDValue ShamtMinusGRLen = DAG.getNode(ISD::ADD, DL, VT, Shamt, MinusGRLen);
161181ad6265SDimitry Andric   SDValue GRLenMinus1Shamt = DAG.getNode(ISD::XOR, DL, VT, Shamt, GRLenMinus1);
161281ad6265SDimitry Andric 
161381ad6265SDimitry Andric   SDValue ShiftRightLo = DAG.getNode(ISD::SRL, DL, VT, Lo, Shamt);
161481ad6265SDimitry Andric   SDValue ShiftLeftHi1 = DAG.getNode(ISD::SHL, DL, VT, Hi, One);
161581ad6265SDimitry Andric   SDValue ShiftLeftHi =
161681ad6265SDimitry Andric       DAG.getNode(ISD::SHL, DL, VT, ShiftLeftHi1, GRLenMinus1Shamt);
161781ad6265SDimitry Andric   SDValue LoTrue = DAG.getNode(ISD::OR, DL, VT, ShiftRightLo, ShiftLeftHi);
161881ad6265SDimitry Andric   SDValue HiTrue = DAG.getNode(ShiftRightOp, DL, VT, Hi, Shamt);
161981ad6265SDimitry Andric   SDValue LoFalse = DAG.getNode(ShiftRightOp, DL, VT, Hi, ShamtMinusGRLen);
162081ad6265SDimitry Andric   SDValue HiFalse =
162181ad6265SDimitry Andric       IsSRA ? DAG.getNode(ISD::SRA, DL, VT, Hi, GRLenMinus1) : Zero;
162281ad6265SDimitry Andric 
162381ad6265SDimitry Andric   SDValue CC = DAG.getSetCC(DL, VT, ShamtMinusGRLen, Zero, ISD::SETLT);
162481ad6265SDimitry Andric 
162581ad6265SDimitry Andric   Lo = DAG.getNode(ISD::SELECT, DL, VT, CC, LoTrue, LoFalse);
162681ad6265SDimitry Andric   Hi = DAG.getNode(ISD::SELECT, DL, VT, CC, HiTrue, HiFalse);
162781ad6265SDimitry Andric 
162881ad6265SDimitry Andric   SDValue Parts[2] = {Lo, Hi};
162981ad6265SDimitry Andric   return DAG.getMergeValues(Parts, DL);
163081ad6265SDimitry Andric }
163181ad6265SDimitry Andric 
163281ad6265SDimitry Andric // Returns the opcode of the target-specific SDNode that implements the 32-bit
163381ad6265SDimitry Andric // form of the given Opcode.
163481ad6265SDimitry Andric static LoongArchISD::NodeType getLoongArchWOpcode(unsigned Opcode) {
163581ad6265SDimitry Andric   switch (Opcode) {
163681ad6265SDimitry Andric   default:
163781ad6265SDimitry Andric     llvm_unreachable("Unexpected opcode");
163881ad6265SDimitry Andric   case ISD::SHL:
163981ad6265SDimitry Andric     return LoongArchISD::SLL_W;
164081ad6265SDimitry Andric   case ISD::SRA:
164181ad6265SDimitry Andric     return LoongArchISD::SRA_W;
164281ad6265SDimitry Andric   case ISD::SRL:
164381ad6265SDimitry Andric     return LoongArchISD::SRL_W;
1644bdd1243dSDimitry Andric   case ISD::ROTR:
1645bdd1243dSDimitry Andric     return LoongArchISD::ROTR_W;
1646bdd1243dSDimitry Andric   case ISD::ROTL:
1647bdd1243dSDimitry Andric     return LoongArchISD::ROTL_W;
1648bdd1243dSDimitry Andric   case ISD::CTTZ:
1649bdd1243dSDimitry Andric     return LoongArchISD::CTZ_W;
1650bdd1243dSDimitry Andric   case ISD::CTLZ:
1651bdd1243dSDimitry Andric     return LoongArchISD::CLZ_W;
165281ad6265SDimitry Andric   }
165381ad6265SDimitry Andric }
165481ad6265SDimitry Andric 
165581ad6265SDimitry Andric // Converts the given i8/i16/i32 operation to a target-specific SelectionDAG
165681ad6265SDimitry Andric // node. Because i8/i16/i32 isn't a legal type for LA64, these operations would
165781ad6265SDimitry Andric // otherwise be promoted to i64, making it difficult to select the
165881ad6265SDimitry Andric // SLL_W/.../*W later one because the fact the operation was originally of
165981ad6265SDimitry Andric // type i8/i16/i32 is lost.
1660bdd1243dSDimitry Andric static SDValue customLegalizeToWOp(SDNode *N, SelectionDAG &DAG, int NumOp,
166181ad6265SDimitry Andric                                    unsigned ExtOpc = ISD::ANY_EXTEND) {
166281ad6265SDimitry Andric   SDLoc DL(N);
166381ad6265SDimitry Andric   LoongArchISD::NodeType WOpcode = getLoongArchWOpcode(N->getOpcode());
1664bdd1243dSDimitry Andric   SDValue NewOp0, NewRes;
1665bdd1243dSDimitry Andric 
1666bdd1243dSDimitry Andric   switch (NumOp) {
1667bdd1243dSDimitry Andric   default:
1668bdd1243dSDimitry Andric     llvm_unreachable("Unexpected NumOp");
1669bdd1243dSDimitry Andric   case 1: {
1670bdd1243dSDimitry Andric     NewOp0 = DAG.getNode(ExtOpc, DL, MVT::i64, N->getOperand(0));
1671bdd1243dSDimitry Andric     NewRes = DAG.getNode(WOpcode, DL, MVT::i64, NewOp0);
1672bdd1243dSDimitry Andric     break;
1673bdd1243dSDimitry Andric   }
1674bdd1243dSDimitry Andric   case 2: {
1675bdd1243dSDimitry Andric     NewOp0 = DAG.getNode(ExtOpc, DL, MVT::i64, N->getOperand(0));
167681ad6265SDimitry Andric     SDValue NewOp1 = DAG.getNode(ExtOpc, DL, MVT::i64, N->getOperand(1));
1677bdd1243dSDimitry Andric     NewRes = DAG.getNode(WOpcode, DL, MVT::i64, NewOp0, NewOp1);
1678bdd1243dSDimitry Andric     break;
1679bdd1243dSDimitry Andric   }
1680bdd1243dSDimitry Andric     // TODO:Handle more NumOp.
1681bdd1243dSDimitry Andric   }
1682bdd1243dSDimitry Andric 
1683bdd1243dSDimitry Andric   // ReplaceNodeResults requires we maintain the same type for the return
1684bdd1243dSDimitry Andric   // value.
168581ad6265SDimitry Andric   return DAG.getNode(ISD::TRUNCATE, DL, N->getValueType(0), NewRes);
168681ad6265SDimitry Andric }
168781ad6265SDimitry Andric 
16885f757f3fSDimitry Andric // Helper function that emits error message for intrinsics with/without chain
16895f757f3fSDimitry Andric // and return a UNDEF or and the chain as the results.
16905f757f3fSDimitry Andric static void emitErrorAndReplaceIntrinsicResults(
169106c3fb27SDimitry Andric     SDNode *N, SmallVectorImpl<SDValue> &Results, SelectionDAG &DAG,
16925f757f3fSDimitry Andric     StringRef ErrorMsg, bool WithChain = true) {
169306c3fb27SDimitry Andric   DAG.getContext()->emitError(N->getOperationName(0) + ": " + ErrorMsg + ".");
169406c3fb27SDimitry Andric   Results.push_back(DAG.getUNDEF(N->getValueType(0)));
16955f757f3fSDimitry Andric   if (!WithChain)
16965f757f3fSDimitry Andric     return;
169706c3fb27SDimitry Andric   Results.push_back(N->getOperand(0));
169806c3fb27SDimitry Andric }
169906c3fb27SDimitry Andric 
17005f757f3fSDimitry Andric template <unsigned N>
17015f757f3fSDimitry Andric static void
17025f757f3fSDimitry Andric replaceVPICKVE2GRResults(SDNode *Node, SmallVectorImpl<SDValue> &Results,
17035f757f3fSDimitry Andric                          SelectionDAG &DAG, const LoongArchSubtarget &Subtarget,
17045f757f3fSDimitry Andric                          unsigned ResOp) {
17055f757f3fSDimitry Andric   const StringRef ErrorMsgOOR = "argument out of range";
1706647cbc5dSDimitry Andric   unsigned Imm = Node->getConstantOperandVal(2);
17075f757f3fSDimitry Andric   if (!isUInt<N>(Imm)) {
17085f757f3fSDimitry Andric     emitErrorAndReplaceIntrinsicResults(Node, Results, DAG, ErrorMsgOOR,
17095f757f3fSDimitry Andric                                         /*WithChain=*/false);
17105f757f3fSDimitry Andric     return;
17115f757f3fSDimitry Andric   }
17125f757f3fSDimitry Andric   SDLoc DL(Node);
17135f757f3fSDimitry Andric   SDValue Vec = Node->getOperand(1);
17145f757f3fSDimitry Andric 
17155f757f3fSDimitry Andric   SDValue PickElt =
17165f757f3fSDimitry Andric       DAG.getNode(ResOp, DL, Subtarget.getGRLenVT(), Vec,
17175f757f3fSDimitry Andric                   DAG.getConstant(Imm, DL, Subtarget.getGRLenVT()),
17185f757f3fSDimitry Andric                   DAG.getValueType(Vec.getValueType().getVectorElementType()));
17195f757f3fSDimitry Andric   Results.push_back(DAG.getNode(ISD::TRUNCATE, DL, Node->getValueType(0),
17205f757f3fSDimitry Andric                                 PickElt.getValue(0)));
17215f757f3fSDimitry Andric }
17225f757f3fSDimitry Andric 
17235f757f3fSDimitry Andric static void replaceVecCondBranchResults(SDNode *N,
17245f757f3fSDimitry Andric                                         SmallVectorImpl<SDValue> &Results,
17255f757f3fSDimitry Andric                                         SelectionDAG &DAG,
17265f757f3fSDimitry Andric                                         const LoongArchSubtarget &Subtarget,
17275f757f3fSDimitry Andric                                         unsigned ResOp) {
17285f757f3fSDimitry Andric   SDLoc DL(N);
17295f757f3fSDimitry Andric   SDValue Vec = N->getOperand(1);
17305f757f3fSDimitry Andric 
17315f757f3fSDimitry Andric   SDValue CB = DAG.getNode(ResOp, DL, Subtarget.getGRLenVT(), Vec);
17325f757f3fSDimitry Andric   Results.push_back(
17335f757f3fSDimitry Andric       DAG.getNode(ISD::TRUNCATE, DL, N->getValueType(0), CB.getValue(0)));
17345f757f3fSDimitry Andric }
17355f757f3fSDimitry Andric 
17365f757f3fSDimitry Andric static void
17375f757f3fSDimitry Andric replaceINTRINSIC_WO_CHAINResults(SDNode *N, SmallVectorImpl<SDValue> &Results,
17385f757f3fSDimitry Andric                                  SelectionDAG &DAG,
17395f757f3fSDimitry Andric                                  const LoongArchSubtarget &Subtarget) {
17405f757f3fSDimitry Andric   switch (N->getConstantOperandVal(0)) {
17415f757f3fSDimitry Andric   default:
17425f757f3fSDimitry Andric     llvm_unreachable("Unexpected Intrinsic.");
17435f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vpickve2gr_b:
17445f757f3fSDimitry Andric     replaceVPICKVE2GRResults<4>(N, Results, DAG, Subtarget,
17455f757f3fSDimitry Andric                                 LoongArchISD::VPICK_SEXT_ELT);
17465f757f3fSDimitry Andric     break;
17475f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vpickve2gr_h:
17485f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvpickve2gr_w:
17495f757f3fSDimitry Andric     replaceVPICKVE2GRResults<3>(N, Results, DAG, Subtarget,
17505f757f3fSDimitry Andric                                 LoongArchISD::VPICK_SEXT_ELT);
17515f757f3fSDimitry Andric     break;
17525f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vpickve2gr_w:
17535f757f3fSDimitry Andric     replaceVPICKVE2GRResults<2>(N, Results, DAG, Subtarget,
17545f757f3fSDimitry Andric                                 LoongArchISD::VPICK_SEXT_ELT);
17555f757f3fSDimitry Andric     break;
17565f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vpickve2gr_bu:
17575f757f3fSDimitry Andric     replaceVPICKVE2GRResults<4>(N, Results, DAG, Subtarget,
17585f757f3fSDimitry Andric                                 LoongArchISD::VPICK_ZEXT_ELT);
17595f757f3fSDimitry Andric     break;
17605f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vpickve2gr_hu:
17615f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvpickve2gr_wu:
17625f757f3fSDimitry Andric     replaceVPICKVE2GRResults<3>(N, Results, DAG, Subtarget,
17635f757f3fSDimitry Andric                                 LoongArchISD::VPICK_ZEXT_ELT);
17645f757f3fSDimitry Andric     break;
17655f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vpickve2gr_wu:
17665f757f3fSDimitry Andric     replaceVPICKVE2GRResults<2>(N, Results, DAG, Subtarget,
17675f757f3fSDimitry Andric                                 LoongArchISD::VPICK_ZEXT_ELT);
17685f757f3fSDimitry Andric     break;
17695f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_bz_b:
17705f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_bz_h:
17715f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_bz_w:
17725f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_bz_d:
17735f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xbz_b:
17745f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xbz_h:
17755f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xbz_w:
17765f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xbz_d:
17775f757f3fSDimitry Andric     replaceVecCondBranchResults(N, Results, DAG, Subtarget,
17785f757f3fSDimitry Andric                                 LoongArchISD::VALL_ZERO);
17795f757f3fSDimitry Andric     break;
17805f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_bz_v:
17815f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xbz_v:
17825f757f3fSDimitry Andric     replaceVecCondBranchResults(N, Results, DAG, Subtarget,
17835f757f3fSDimitry Andric                                 LoongArchISD::VANY_ZERO);
17845f757f3fSDimitry Andric     break;
17855f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_bnz_b:
17865f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_bnz_h:
17875f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_bnz_w:
17885f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_bnz_d:
17895f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xbnz_b:
17905f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xbnz_h:
17915f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xbnz_w:
17925f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xbnz_d:
17935f757f3fSDimitry Andric     replaceVecCondBranchResults(N, Results, DAG, Subtarget,
17945f757f3fSDimitry Andric                                 LoongArchISD::VALL_NONZERO);
17955f757f3fSDimitry Andric     break;
17965f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_bnz_v:
17975f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xbnz_v:
17985f757f3fSDimitry Andric     replaceVecCondBranchResults(N, Results, DAG, Subtarget,
17995f757f3fSDimitry Andric                                 LoongArchISD::VANY_NONZERO);
18005f757f3fSDimitry Andric     break;
18015f757f3fSDimitry Andric   }
18025f757f3fSDimitry Andric }
18035f757f3fSDimitry Andric 
180481ad6265SDimitry Andric void LoongArchTargetLowering::ReplaceNodeResults(
180581ad6265SDimitry Andric     SDNode *N, SmallVectorImpl<SDValue> &Results, SelectionDAG &DAG) const {
180681ad6265SDimitry Andric   SDLoc DL(N);
1807bdd1243dSDimitry Andric   EVT VT = N->getValueType(0);
180881ad6265SDimitry Andric   switch (N->getOpcode()) {
180981ad6265SDimitry Andric   default:
181081ad6265SDimitry Andric     llvm_unreachable("Don't know how to legalize this operation");
181181ad6265SDimitry Andric   case ISD::SHL:
181281ad6265SDimitry Andric   case ISD::SRA:
181381ad6265SDimitry Andric   case ISD::SRL:
1814bdd1243dSDimitry Andric   case ISD::ROTR:
1815bdd1243dSDimitry Andric     assert(VT == MVT::i32 && Subtarget.is64Bit() &&
181681ad6265SDimitry Andric            "Unexpected custom legalisation");
181781ad6265SDimitry Andric     if (N->getOperand(1).getOpcode() != ISD::Constant) {
1818bdd1243dSDimitry Andric       Results.push_back(customLegalizeToWOp(N, DAG, 2));
1819bdd1243dSDimitry Andric       break;
1820bdd1243dSDimitry Andric     }
1821bdd1243dSDimitry Andric     break;
1822bdd1243dSDimitry Andric   case ISD::ROTL:
1823bdd1243dSDimitry Andric     ConstantSDNode *CN;
1824bdd1243dSDimitry Andric     if ((CN = dyn_cast<ConstantSDNode>(N->getOperand(1)))) {
1825bdd1243dSDimitry Andric       Results.push_back(customLegalizeToWOp(N, DAG, 2));
182681ad6265SDimitry Andric       break;
182781ad6265SDimitry Andric     }
182881ad6265SDimitry Andric     break;
1829753f127fSDimitry Andric   case ISD::FP_TO_SINT: {
1830bdd1243dSDimitry Andric     assert(VT == MVT::i32 && Subtarget.is64Bit() &&
1831753f127fSDimitry Andric            "Unexpected custom legalisation");
1832753f127fSDimitry Andric     SDValue Src = N->getOperand(0);
1833bdd1243dSDimitry Andric     EVT FVT = EVT::getFloatingPointVT(N->getValueSizeInBits(0));
1834bdd1243dSDimitry Andric     if (getTypeAction(*DAG.getContext(), Src.getValueType()) !=
1835bdd1243dSDimitry Andric         TargetLowering::TypeSoftenFloat) {
1836bdd1243dSDimitry Andric       SDValue Dst = DAG.getNode(LoongArchISD::FTINT, DL, FVT, Src);
1837bdd1243dSDimitry Andric       Results.push_back(DAG.getNode(ISD::BITCAST, DL, VT, Dst));
1838bdd1243dSDimitry Andric       return;
1839bdd1243dSDimitry Andric     }
1840bdd1243dSDimitry Andric     // If the FP type needs to be softened, emit a library call using the 'si'
1841bdd1243dSDimitry Andric     // version. If we left it to default legalization we'd end up with 'di'.
1842bdd1243dSDimitry Andric     RTLIB::Libcall LC;
1843bdd1243dSDimitry Andric     LC = RTLIB::getFPTOSINT(Src.getValueType(), VT);
1844bdd1243dSDimitry Andric     MakeLibCallOptions CallOptions;
1845bdd1243dSDimitry Andric     EVT OpVT = Src.getValueType();
1846bdd1243dSDimitry Andric     CallOptions.setTypeListBeforeSoften(OpVT, VT, true);
1847bdd1243dSDimitry Andric     SDValue Chain = SDValue();
1848bdd1243dSDimitry Andric     SDValue Result;
1849bdd1243dSDimitry Andric     std::tie(Result, Chain) =
1850bdd1243dSDimitry Andric         makeLibCall(DAG, LC, VT, Src, CallOptions, DL, Chain);
1851bdd1243dSDimitry Andric     Results.push_back(Result);
1852753f127fSDimitry Andric     break;
1853753f127fSDimitry Andric   }
1854753f127fSDimitry Andric   case ISD::BITCAST: {
1855753f127fSDimitry Andric     SDValue Src = N->getOperand(0);
1856753f127fSDimitry Andric     EVT SrcVT = Src.getValueType();
1857753f127fSDimitry Andric     if (VT == MVT::i32 && SrcVT == MVT::f32 && Subtarget.is64Bit() &&
1858753f127fSDimitry Andric         Subtarget.hasBasicF()) {
1859753f127fSDimitry Andric       SDValue Dst =
1860753f127fSDimitry Andric           DAG.getNode(LoongArchISD::MOVFR2GR_S_LA64, DL, MVT::i64, Src);
1861753f127fSDimitry Andric       Results.push_back(DAG.getNode(ISD::TRUNCATE, DL, MVT::i32, Dst));
1862753f127fSDimitry Andric     }
1863753f127fSDimitry Andric     break;
1864753f127fSDimitry Andric   }
1865753f127fSDimitry Andric   case ISD::FP_TO_UINT: {
1866bdd1243dSDimitry Andric     assert(VT == MVT::i32 && Subtarget.is64Bit() &&
1867753f127fSDimitry Andric            "Unexpected custom legalisation");
1868753f127fSDimitry Andric     auto &TLI = DAG.getTargetLoweringInfo();
1869753f127fSDimitry Andric     SDValue Tmp1, Tmp2;
1870753f127fSDimitry Andric     TLI.expandFP_TO_UINT(N, Tmp1, Tmp2, DAG);
1871753f127fSDimitry Andric     Results.push_back(DAG.getNode(ISD::TRUNCATE, DL, MVT::i32, Tmp1));
1872753f127fSDimitry Andric     break;
1873753f127fSDimitry Andric   }
1874bdd1243dSDimitry Andric   case ISD::BSWAP: {
1875bdd1243dSDimitry Andric     SDValue Src = N->getOperand(0);
1876bdd1243dSDimitry Andric     assert((VT == MVT::i16 || VT == MVT::i32) &&
1877bdd1243dSDimitry Andric            "Unexpected custom legalization");
1878bdd1243dSDimitry Andric     MVT GRLenVT = Subtarget.getGRLenVT();
1879bdd1243dSDimitry Andric     SDValue NewSrc = DAG.getNode(ISD::ANY_EXTEND, DL, GRLenVT, Src);
1880bdd1243dSDimitry Andric     SDValue Tmp;
1881bdd1243dSDimitry Andric     switch (VT.getSizeInBits()) {
1882bdd1243dSDimitry Andric     default:
1883bdd1243dSDimitry Andric       llvm_unreachable("Unexpected operand width");
1884bdd1243dSDimitry Andric     case 16:
1885bdd1243dSDimitry Andric       Tmp = DAG.getNode(LoongArchISD::REVB_2H, DL, GRLenVT, NewSrc);
1886bdd1243dSDimitry Andric       break;
1887bdd1243dSDimitry Andric     case 32:
1888bdd1243dSDimitry Andric       // Only LA64 will get to here due to the size mismatch between VT and
1889bdd1243dSDimitry Andric       // GRLenVT, LA32 lowering is directly defined in LoongArchInstrInfo.
1890bdd1243dSDimitry Andric       Tmp = DAG.getNode(LoongArchISD::REVB_2W, DL, GRLenVT, NewSrc);
1891bdd1243dSDimitry Andric       break;
1892bdd1243dSDimitry Andric     }
1893bdd1243dSDimitry Andric     Results.push_back(DAG.getNode(ISD::TRUNCATE, DL, VT, Tmp));
1894bdd1243dSDimitry Andric     break;
1895bdd1243dSDimitry Andric   }
1896bdd1243dSDimitry Andric   case ISD::BITREVERSE: {
1897bdd1243dSDimitry Andric     SDValue Src = N->getOperand(0);
1898bdd1243dSDimitry Andric     assert((VT == MVT::i8 || (VT == MVT::i32 && Subtarget.is64Bit())) &&
1899bdd1243dSDimitry Andric            "Unexpected custom legalization");
1900bdd1243dSDimitry Andric     MVT GRLenVT = Subtarget.getGRLenVT();
1901bdd1243dSDimitry Andric     SDValue NewSrc = DAG.getNode(ISD::ANY_EXTEND, DL, GRLenVT, Src);
1902bdd1243dSDimitry Andric     SDValue Tmp;
1903bdd1243dSDimitry Andric     switch (VT.getSizeInBits()) {
1904bdd1243dSDimitry Andric     default:
1905bdd1243dSDimitry Andric       llvm_unreachable("Unexpected operand width");
1906bdd1243dSDimitry Andric     case 8:
1907bdd1243dSDimitry Andric       Tmp = DAG.getNode(LoongArchISD::BITREV_4B, DL, GRLenVT, NewSrc);
1908bdd1243dSDimitry Andric       break;
1909bdd1243dSDimitry Andric     case 32:
1910bdd1243dSDimitry Andric       Tmp = DAG.getNode(LoongArchISD::BITREV_W, DL, GRLenVT, NewSrc);
1911bdd1243dSDimitry Andric       break;
1912bdd1243dSDimitry Andric     }
1913bdd1243dSDimitry Andric     Results.push_back(DAG.getNode(ISD::TRUNCATE, DL, VT, Tmp));
1914bdd1243dSDimitry Andric     break;
1915bdd1243dSDimitry Andric   }
1916bdd1243dSDimitry Andric   case ISD::CTLZ:
1917bdd1243dSDimitry Andric   case ISD::CTTZ: {
1918bdd1243dSDimitry Andric     assert(VT == MVT::i32 && Subtarget.is64Bit() &&
1919bdd1243dSDimitry Andric            "Unexpected custom legalisation");
1920bdd1243dSDimitry Andric     Results.push_back(customLegalizeToWOp(N, DAG, 1));
1921bdd1243dSDimitry Andric     break;
1922bdd1243dSDimitry Andric   }
1923bdd1243dSDimitry Andric   case ISD::INTRINSIC_W_CHAIN: {
192406c3fb27SDimitry Andric     SDValue Chain = N->getOperand(0);
1925bdd1243dSDimitry Andric     SDValue Op2 = N->getOperand(2);
192606c3fb27SDimitry Andric     MVT GRLenVT = Subtarget.getGRLenVT();
192706c3fb27SDimitry Andric     const StringRef ErrorMsgOOR = "argument out of range";
192806c3fb27SDimitry Andric     const StringRef ErrorMsgReqLA64 = "requires loongarch64";
192906c3fb27SDimitry Andric     const StringRef ErrorMsgReqF = "requires basic 'f' target feature";
1930bdd1243dSDimitry Andric 
193106c3fb27SDimitry Andric     switch (N->getConstantOperandVal(1)) {
1932bdd1243dSDimitry Andric     default:
1933bdd1243dSDimitry Andric       llvm_unreachable("Unexpected Intrinsic.");
193406c3fb27SDimitry Andric     case Intrinsic::loongarch_movfcsr2gr: {
193506c3fb27SDimitry Andric       if (!Subtarget.hasBasicF()) {
19365f757f3fSDimitry Andric         emitErrorAndReplaceIntrinsicResults(N, Results, DAG, ErrorMsgReqF);
193706c3fb27SDimitry Andric         return;
193806c3fb27SDimitry Andric       }
1939*1db9f3b2SDimitry Andric       unsigned Imm = Op2->getAsZExtVal();
194006c3fb27SDimitry Andric       if (!isUInt<2>(Imm)) {
19415f757f3fSDimitry Andric         emitErrorAndReplaceIntrinsicResults(N, Results, DAG, ErrorMsgOOR);
194206c3fb27SDimitry Andric         return;
194306c3fb27SDimitry Andric       }
194406c3fb27SDimitry Andric       SDValue MOVFCSR2GRResults = DAG.getNode(
194506c3fb27SDimitry Andric           LoongArchISD::MOVFCSR2GR, SDLoc(N), {MVT::i64, MVT::Other},
194606c3fb27SDimitry Andric           {Chain, DAG.getConstant(Imm, DL, GRLenVT)});
194706c3fb27SDimitry Andric       Results.push_back(
194806c3fb27SDimitry Andric           DAG.getNode(ISD::TRUNCATE, DL, VT, MOVFCSR2GRResults.getValue(0)));
194906c3fb27SDimitry Andric       Results.push_back(MOVFCSR2GRResults.getValue(1));
195006c3fb27SDimitry Andric       break;
195106c3fb27SDimitry Andric     }
1952bdd1243dSDimitry Andric #define CRC_CASE_EXT_BINARYOP(NAME, NODE)                                      \
1953bdd1243dSDimitry Andric   case Intrinsic::loongarch_##NAME: {                                          \
195406c3fb27SDimitry Andric     SDValue NODE = DAG.getNode(                                                \
195506c3fb27SDimitry Andric         LoongArchISD::NODE, DL, {MVT::i64, MVT::Other},                        \
195606c3fb27SDimitry Andric         {Chain, DAG.getNode(ISD::ANY_EXTEND, DL, MVT::i64, Op2),               \
195706c3fb27SDimitry Andric          DAG.getNode(ISD::ANY_EXTEND, DL, MVT::i64, N->getOperand(3))});       \
195806c3fb27SDimitry Andric     Results.push_back(DAG.getNode(ISD::TRUNCATE, DL, VT, NODE.getValue(0)));   \
195906c3fb27SDimitry Andric     Results.push_back(NODE.getValue(1));                                       \
1960bdd1243dSDimitry Andric     break;                                                                     \
1961bdd1243dSDimitry Andric   }
1962bdd1243dSDimitry Andric       CRC_CASE_EXT_BINARYOP(crc_w_b_w, CRC_W_B_W)
1963bdd1243dSDimitry Andric       CRC_CASE_EXT_BINARYOP(crc_w_h_w, CRC_W_H_W)
1964bdd1243dSDimitry Andric       CRC_CASE_EXT_BINARYOP(crc_w_w_w, CRC_W_W_W)
1965bdd1243dSDimitry Andric       CRC_CASE_EXT_BINARYOP(crcc_w_b_w, CRCC_W_B_W)
1966bdd1243dSDimitry Andric       CRC_CASE_EXT_BINARYOP(crcc_w_h_w, CRCC_W_H_W)
1967bdd1243dSDimitry Andric       CRC_CASE_EXT_BINARYOP(crcc_w_w_w, CRCC_W_W_W)
1968bdd1243dSDimitry Andric #undef CRC_CASE_EXT_BINARYOP
1969bdd1243dSDimitry Andric 
1970bdd1243dSDimitry Andric #define CRC_CASE_EXT_UNARYOP(NAME, NODE)                                       \
1971bdd1243dSDimitry Andric   case Intrinsic::loongarch_##NAME: {                                          \
197206c3fb27SDimitry Andric     SDValue NODE = DAG.getNode(                                                \
197306c3fb27SDimitry Andric         LoongArchISD::NODE, DL, {MVT::i64, MVT::Other},                        \
197406c3fb27SDimitry Andric         {Chain, Op2,                                                           \
197506c3fb27SDimitry Andric          DAG.getNode(ISD::ANY_EXTEND, DL, MVT::i64, N->getOperand(3))});       \
197606c3fb27SDimitry Andric     Results.push_back(DAG.getNode(ISD::TRUNCATE, DL, VT, NODE.getValue(0)));   \
197706c3fb27SDimitry Andric     Results.push_back(NODE.getValue(1));                                       \
1978bdd1243dSDimitry Andric     break;                                                                     \
1979bdd1243dSDimitry Andric   }
1980bdd1243dSDimitry Andric       CRC_CASE_EXT_UNARYOP(crc_w_d_w, CRC_W_D_W)
1981bdd1243dSDimitry Andric       CRC_CASE_EXT_UNARYOP(crcc_w_d_w, CRCC_W_D_W)
1982bdd1243dSDimitry Andric #undef CRC_CASE_EXT_UNARYOP
1983bdd1243dSDimitry Andric #define CSR_CASE(ID)                                                           \
1984bdd1243dSDimitry Andric   case Intrinsic::loongarch_##ID: {                                            \
198506c3fb27SDimitry Andric     if (!Subtarget.is64Bit())                                                  \
19865f757f3fSDimitry Andric       emitErrorAndReplaceIntrinsicResults(N, Results, DAG, ErrorMsgReqLA64);   \
1987bdd1243dSDimitry Andric     break;                                                                     \
1988bdd1243dSDimitry Andric   }
1989bdd1243dSDimitry Andric       CSR_CASE(csrrd_d);
1990bdd1243dSDimitry Andric       CSR_CASE(csrwr_d);
1991bdd1243dSDimitry Andric       CSR_CASE(csrxchg_d);
1992bdd1243dSDimitry Andric       CSR_CASE(iocsrrd_d);
1993bdd1243dSDimitry Andric #undef CSR_CASE
1994bdd1243dSDimitry Andric     case Intrinsic::loongarch_csrrd_w: {
1995*1db9f3b2SDimitry Andric       unsigned Imm = Op2->getAsZExtVal();
1996bdd1243dSDimitry Andric       if (!isUInt<14>(Imm)) {
19975f757f3fSDimitry Andric         emitErrorAndReplaceIntrinsicResults(N, Results, DAG, ErrorMsgOOR);
199806c3fb27SDimitry Andric         return;
1999bdd1243dSDimitry Andric       }
200006c3fb27SDimitry Andric       SDValue CSRRDResults =
200106c3fb27SDimitry Andric           DAG.getNode(LoongArchISD::CSRRD, DL, {GRLenVT, MVT::Other},
200206c3fb27SDimitry Andric                       {Chain, DAG.getConstant(Imm, DL, GRLenVT)});
2003bdd1243dSDimitry Andric       Results.push_back(
200406c3fb27SDimitry Andric           DAG.getNode(ISD::TRUNCATE, DL, VT, CSRRDResults.getValue(0)));
200506c3fb27SDimitry Andric       Results.push_back(CSRRDResults.getValue(1));
2006bdd1243dSDimitry Andric       break;
2007bdd1243dSDimitry Andric     }
2008bdd1243dSDimitry Andric     case Intrinsic::loongarch_csrwr_w: {
2009647cbc5dSDimitry Andric       unsigned Imm = N->getConstantOperandVal(3);
2010bdd1243dSDimitry Andric       if (!isUInt<14>(Imm)) {
20115f757f3fSDimitry Andric         emitErrorAndReplaceIntrinsicResults(N, Results, DAG, ErrorMsgOOR);
201206c3fb27SDimitry Andric         return;
2013bdd1243dSDimitry Andric       }
201406c3fb27SDimitry Andric       SDValue CSRWRResults =
201506c3fb27SDimitry Andric           DAG.getNode(LoongArchISD::CSRWR, DL, {GRLenVT, MVT::Other},
201606c3fb27SDimitry Andric                       {Chain, DAG.getNode(ISD::ANY_EXTEND, DL, MVT::i64, Op2),
201706c3fb27SDimitry Andric                        DAG.getConstant(Imm, DL, GRLenVT)});
201806c3fb27SDimitry Andric       Results.push_back(
201906c3fb27SDimitry Andric           DAG.getNode(ISD::TRUNCATE, DL, VT, CSRWRResults.getValue(0)));
202006c3fb27SDimitry Andric       Results.push_back(CSRWRResults.getValue(1));
2021bdd1243dSDimitry Andric       break;
2022bdd1243dSDimitry Andric     }
2023bdd1243dSDimitry Andric     case Intrinsic::loongarch_csrxchg_w: {
2024647cbc5dSDimitry Andric       unsigned Imm = N->getConstantOperandVal(4);
2025bdd1243dSDimitry Andric       if (!isUInt<14>(Imm)) {
20265f757f3fSDimitry Andric         emitErrorAndReplaceIntrinsicResults(N, Results, DAG, ErrorMsgOOR);
202706c3fb27SDimitry Andric         return;
2028bdd1243dSDimitry Andric       }
202906c3fb27SDimitry Andric       SDValue CSRXCHGResults = DAG.getNode(
203006c3fb27SDimitry Andric           LoongArchISD::CSRXCHG, DL, {GRLenVT, MVT::Other},
203106c3fb27SDimitry Andric           {Chain, DAG.getNode(ISD::ANY_EXTEND, DL, MVT::i64, Op2),
2032bdd1243dSDimitry Andric            DAG.getNode(ISD::ANY_EXTEND, DL, MVT::i64, N->getOperand(3)),
203306c3fb27SDimitry Andric            DAG.getConstant(Imm, DL, GRLenVT)});
203406c3fb27SDimitry Andric       Results.push_back(
203506c3fb27SDimitry Andric           DAG.getNode(ISD::TRUNCATE, DL, VT, CSRXCHGResults.getValue(0)));
203606c3fb27SDimitry Andric       Results.push_back(CSRXCHGResults.getValue(1));
2037bdd1243dSDimitry Andric       break;
2038bdd1243dSDimitry Andric     }
2039bdd1243dSDimitry Andric #define IOCSRRD_CASE(NAME, NODE)                                               \
2040bdd1243dSDimitry Andric   case Intrinsic::loongarch_##NAME: {                                          \
204106c3fb27SDimitry Andric     SDValue IOCSRRDResults =                                                   \
204206c3fb27SDimitry Andric         DAG.getNode(LoongArchISD::NODE, DL, {MVT::i64, MVT::Other},            \
204306c3fb27SDimitry Andric                     {Chain, DAG.getNode(ISD::ANY_EXTEND, DL, MVT::i64, Op2)}); \
204406c3fb27SDimitry Andric     Results.push_back(                                                         \
204506c3fb27SDimitry Andric         DAG.getNode(ISD::TRUNCATE, DL, VT, IOCSRRDResults.getValue(0)));       \
204606c3fb27SDimitry Andric     Results.push_back(IOCSRRDResults.getValue(1));                             \
2047bdd1243dSDimitry Andric     break;                                                                     \
2048bdd1243dSDimitry Andric   }
2049bdd1243dSDimitry Andric       IOCSRRD_CASE(iocsrrd_b, IOCSRRD_B);
2050bdd1243dSDimitry Andric       IOCSRRD_CASE(iocsrrd_h, IOCSRRD_H);
2051bdd1243dSDimitry Andric       IOCSRRD_CASE(iocsrrd_w, IOCSRRD_W);
2052bdd1243dSDimitry Andric #undef IOCSRRD_CASE
2053bdd1243dSDimitry Andric     case Intrinsic::loongarch_cpucfg: {
205406c3fb27SDimitry Andric       SDValue CPUCFGResults =
205506c3fb27SDimitry Andric           DAG.getNode(LoongArchISD::CPUCFG, DL, {GRLenVT, MVT::Other},
205606c3fb27SDimitry Andric                       {Chain, DAG.getNode(ISD::ANY_EXTEND, DL, MVT::i64, Op2)});
205706c3fb27SDimitry Andric       Results.push_back(
205806c3fb27SDimitry Andric           DAG.getNode(ISD::TRUNCATE, DL, VT, CPUCFGResults.getValue(0)));
205906c3fb27SDimitry Andric       Results.push_back(CPUCFGResults.getValue(1));
2060bdd1243dSDimitry Andric       break;
2061bdd1243dSDimitry Andric     }
2062bdd1243dSDimitry Andric     case Intrinsic::loongarch_lddir_d: {
2063bdd1243dSDimitry Andric       if (!Subtarget.is64Bit()) {
20645f757f3fSDimitry Andric         emitErrorAndReplaceIntrinsicResults(N, Results, DAG, ErrorMsgReqLA64);
206506c3fb27SDimitry Andric         return;
2066bdd1243dSDimitry Andric       }
2067bdd1243dSDimitry Andric       break;
2068bdd1243dSDimitry Andric     }
2069bdd1243dSDimitry Andric     }
2070bdd1243dSDimitry Andric     break;
2071bdd1243dSDimitry Andric   }
2072bdd1243dSDimitry Andric   case ISD::READ_REGISTER: {
2073bdd1243dSDimitry Andric     if (Subtarget.is64Bit())
2074bdd1243dSDimitry Andric       DAG.getContext()->emitError(
2075bdd1243dSDimitry Andric           "On LA64, only 64-bit registers can be read.");
2076bdd1243dSDimitry Andric     else
2077bdd1243dSDimitry Andric       DAG.getContext()->emitError(
2078bdd1243dSDimitry Andric           "On LA32, only 32-bit registers can be read.");
2079bdd1243dSDimitry Andric     Results.push_back(DAG.getUNDEF(VT));
2080bdd1243dSDimitry Andric     Results.push_back(N->getOperand(0));
2081bdd1243dSDimitry Andric     break;
2082bdd1243dSDimitry Andric   }
20835f757f3fSDimitry Andric   case ISD::INTRINSIC_WO_CHAIN: {
20845f757f3fSDimitry Andric     replaceINTRINSIC_WO_CHAINResults(N, Results, DAG, Subtarget);
20855f757f3fSDimitry Andric     break;
20865f757f3fSDimitry Andric   }
208781ad6265SDimitry Andric   }
208881ad6265SDimitry Andric }
208981ad6265SDimitry Andric 
209081ad6265SDimitry Andric static SDValue performANDCombine(SDNode *N, SelectionDAG &DAG,
209181ad6265SDimitry Andric                                  TargetLowering::DAGCombinerInfo &DCI,
209281ad6265SDimitry Andric                                  const LoongArchSubtarget &Subtarget) {
209381ad6265SDimitry Andric   if (DCI.isBeforeLegalizeOps())
209481ad6265SDimitry Andric     return SDValue();
209581ad6265SDimitry Andric 
209681ad6265SDimitry Andric   SDValue FirstOperand = N->getOperand(0);
209781ad6265SDimitry Andric   SDValue SecondOperand = N->getOperand(1);
209881ad6265SDimitry Andric   unsigned FirstOperandOpc = FirstOperand.getOpcode();
209981ad6265SDimitry Andric   EVT ValTy = N->getValueType(0);
210081ad6265SDimitry Andric   SDLoc DL(N);
210181ad6265SDimitry Andric   uint64_t lsb, msb;
210281ad6265SDimitry Andric   unsigned SMIdx, SMLen;
210381ad6265SDimitry Andric   ConstantSDNode *CN;
210481ad6265SDimitry Andric   SDValue NewOperand;
210581ad6265SDimitry Andric   MVT GRLenVT = Subtarget.getGRLenVT();
210681ad6265SDimitry Andric 
210781ad6265SDimitry Andric   // Op's second operand must be a shifted mask.
210881ad6265SDimitry Andric   if (!(CN = dyn_cast<ConstantSDNode>(SecondOperand)) ||
210981ad6265SDimitry Andric       !isShiftedMask_64(CN->getZExtValue(), SMIdx, SMLen))
211081ad6265SDimitry Andric     return SDValue();
211181ad6265SDimitry Andric 
211281ad6265SDimitry Andric   if (FirstOperandOpc == ISD::SRA || FirstOperandOpc == ISD::SRL) {
211381ad6265SDimitry Andric     // Pattern match BSTRPICK.
211481ad6265SDimitry Andric     //  $dst = and ((sra or srl) $src , lsb), (2**len - 1)
211581ad6265SDimitry Andric     //  => BSTRPICK $dst, $src, msb, lsb
211681ad6265SDimitry Andric     //  where msb = lsb + len - 1
211781ad6265SDimitry Andric 
211881ad6265SDimitry Andric     // The second operand of the shift must be an immediate.
211981ad6265SDimitry Andric     if (!(CN = dyn_cast<ConstantSDNode>(FirstOperand.getOperand(1))))
212081ad6265SDimitry Andric       return SDValue();
212181ad6265SDimitry Andric 
212281ad6265SDimitry Andric     lsb = CN->getZExtValue();
212381ad6265SDimitry Andric 
212481ad6265SDimitry Andric     // Return if the shifted mask does not start at bit 0 or the sum of its
212581ad6265SDimitry Andric     // length and lsb exceeds the word's size.
212681ad6265SDimitry Andric     if (SMIdx != 0 || lsb + SMLen > ValTy.getSizeInBits())
212781ad6265SDimitry Andric       return SDValue();
212881ad6265SDimitry Andric 
212981ad6265SDimitry Andric     NewOperand = FirstOperand.getOperand(0);
213081ad6265SDimitry Andric   } else {
213181ad6265SDimitry Andric     // Pattern match BSTRPICK.
213281ad6265SDimitry Andric     //  $dst = and $src, (2**len- 1) , if len > 12
213381ad6265SDimitry Andric     //  => BSTRPICK $dst, $src, msb, lsb
213481ad6265SDimitry Andric     //  where lsb = 0 and msb = len - 1
213581ad6265SDimitry Andric 
213681ad6265SDimitry Andric     // If the mask is <= 0xfff, andi can be used instead.
213781ad6265SDimitry Andric     if (CN->getZExtValue() <= 0xfff)
213881ad6265SDimitry Andric       return SDValue();
213981ad6265SDimitry Andric 
214006c3fb27SDimitry Andric     // Return if the MSB exceeds.
214106c3fb27SDimitry Andric     if (SMIdx + SMLen > ValTy.getSizeInBits())
214281ad6265SDimitry Andric       return SDValue();
214381ad6265SDimitry Andric 
214406c3fb27SDimitry Andric     if (SMIdx > 0) {
214506c3fb27SDimitry Andric       // Omit if the constant has more than 2 uses. This a conservative
214606c3fb27SDimitry Andric       // decision. Whether it is a win depends on the HW microarchitecture.
214706c3fb27SDimitry Andric       // However it should always be better for 1 and 2 uses.
214806c3fb27SDimitry Andric       if (CN->use_size() > 2)
214906c3fb27SDimitry Andric         return SDValue();
215006c3fb27SDimitry Andric       // Return if the constant can be composed by a single LU12I.W.
215106c3fb27SDimitry Andric       if ((CN->getZExtValue() & 0xfff) == 0)
215206c3fb27SDimitry Andric         return SDValue();
215306c3fb27SDimitry Andric       // Return if the constand can be composed by a single ADDI with
215406c3fb27SDimitry Andric       // the zero register.
215506c3fb27SDimitry Andric       if (CN->getSExtValue() >= -2048 && CN->getSExtValue() < 0)
215606c3fb27SDimitry Andric         return SDValue();
215706c3fb27SDimitry Andric     }
215806c3fb27SDimitry Andric 
215906c3fb27SDimitry Andric     lsb = SMIdx;
216081ad6265SDimitry Andric     NewOperand = FirstOperand;
216181ad6265SDimitry Andric   }
216206c3fb27SDimitry Andric 
216381ad6265SDimitry Andric   msb = lsb + SMLen - 1;
216406c3fb27SDimitry Andric   SDValue NR0 = DAG.getNode(LoongArchISD::BSTRPICK, DL, ValTy, NewOperand,
216581ad6265SDimitry Andric                             DAG.getConstant(msb, DL, GRLenVT),
216681ad6265SDimitry Andric                             DAG.getConstant(lsb, DL, GRLenVT));
216706c3fb27SDimitry Andric   if (FirstOperandOpc == ISD::SRA || FirstOperandOpc == ISD::SRL || lsb == 0)
216806c3fb27SDimitry Andric     return NR0;
216906c3fb27SDimitry Andric   // Try to optimize to
217006c3fb27SDimitry Andric   //   bstrpick $Rd, $Rs, msb, lsb
217106c3fb27SDimitry Andric   //   slli     $Rd, $Rd, lsb
217206c3fb27SDimitry Andric   return DAG.getNode(ISD::SHL, DL, ValTy, NR0,
217306c3fb27SDimitry Andric                      DAG.getConstant(lsb, DL, GRLenVT));
217481ad6265SDimitry Andric }
217581ad6265SDimitry Andric 
217681ad6265SDimitry Andric static SDValue performSRLCombine(SDNode *N, SelectionDAG &DAG,
217781ad6265SDimitry Andric                                  TargetLowering::DAGCombinerInfo &DCI,
217881ad6265SDimitry Andric                                  const LoongArchSubtarget &Subtarget) {
217981ad6265SDimitry Andric   if (DCI.isBeforeLegalizeOps())
218081ad6265SDimitry Andric     return SDValue();
218181ad6265SDimitry Andric 
218281ad6265SDimitry Andric   // $dst = srl (and $src, Mask), Shamt
218381ad6265SDimitry Andric   // =>
218481ad6265SDimitry Andric   // BSTRPICK $dst, $src, MaskIdx+MaskLen-1, Shamt
218581ad6265SDimitry Andric   // when Mask is a shifted mask, and MaskIdx <= Shamt <= MaskIdx+MaskLen-1
218681ad6265SDimitry Andric   //
218781ad6265SDimitry Andric 
218881ad6265SDimitry Andric   SDValue FirstOperand = N->getOperand(0);
218981ad6265SDimitry Andric   ConstantSDNode *CN;
219081ad6265SDimitry Andric   EVT ValTy = N->getValueType(0);
219181ad6265SDimitry Andric   SDLoc DL(N);
219281ad6265SDimitry Andric   MVT GRLenVT = Subtarget.getGRLenVT();
219381ad6265SDimitry Andric   unsigned MaskIdx, MaskLen;
219481ad6265SDimitry Andric   uint64_t Shamt;
219581ad6265SDimitry Andric 
219681ad6265SDimitry Andric   // The first operand must be an AND and the second operand of the AND must be
219781ad6265SDimitry Andric   // a shifted mask.
219881ad6265SDimitry Andric   if (FirstOperand.getOpcode() != ISD::AND ||
219981ad6265SDimitry Andric       !(CN = dyn_cast<ConstantSDNode>(FirstOperand.getOperand(1))) ||
220081ad6265SDimitry Andric       !isShiftedMask_64(CN->getZExtValue(), MaskIdx, MaskLen))
220181ad6265SDimitry Andric     return SDValue();
220281ad6265SDimitry Andric 
220381ad6265SDimitry Andric   // The second operand (shift amount) must be an immediate.
220481ad6265SDimitry Andric   if (!(CN = dyn_cast<ConstantSDNode>(N->getOperand(1))))
220581ad6265SDimitry Andric     return SDValue();
220681ad6265SDimitry Andric 
220781ad6265SDimitry Andric   Shamt = CN->getZExtValue();
220881ad6265SDimitry Andric   if (MaskIdx <= Shamt && Shamt <= MaskIdx + MaskLen - 1)
220981ad6265SDimitry Andric     return DAG.getNode(LoongArchISD::BSTRPICK, DL, ValTy,
221081ad6265SDimitry Andric                        FirstOperand->getOperand(0),
221181ad6265SDimitry Andric                        DAG.getConstant(MaskIdx + MaskLen - 1, DL, GRLenVT),
221281ad6265SDimitry Andric                        DAG.getConstant(Shamt, DL, GRLenVT));
221381ad6265SDimitry Andric 
221481ad6265SDimitry Andric   return SDValue();
221581ad6265SDimitry Andric }
221681ad6265SDimitry Andric 
2217753f127fSDimitry Andric static SDValue performORCombine(SDNode *N, SelectionDAG &DAG,
2218753f127fSDimitry Andric                                 TargetLowering::DAGCombinerInfo &DCI,
2219753f127fSDimitry Andric                                 const LoongArchSubtarget &Subtarget) {
2220753f127fSDimitry Andric   MVT GRLenVT = Subtarget.getGRLenVT();
2221753f127fSDimitry Andric   EVT ValTy = N->getValueType(0);
2222753f127fSDimitry Andric   SDValue N0 = N->getOperand(0), N1 = N->getOperand(1);
2223753f127fSDimitry Andric   ConstantSDNode *CN0, *CN1;
2224753f127fSDimitry Andric   SDLoc DL(N);
2225753f127fSDimitry Andric   unsigned ValBits = ValTy.getSizeInBits();
2226753f127fSDimitry Andric   unsigned MaskIdx0, MaskLen0, MaskIdx1, MaskLen1;
2227753f127fSDimitry Andric   unsigned Shamt;
2228753f127fSDimitry Andric   bool SwapAndRetried = false;
2229753f127fSDimitry Andric 
2230753f127fSDimitry Andric   if (DCI.isBeforeLegalizeOps())
2231753f127fSDimitry Andric     return SDValue();
2232753f127fSDimitry Andric 
2233753f127fSDimitry Andric   if (ValBits != 32 && ValBits != 64)
2234753f127fSDimitry Andric     return SDValue();
2235753f127fSDimitry Andric 
2236753f127fSDimitry Andric Retry:
2237753f127fSDimitry Andric   // 1st pattern to match BSTRINS:
2238753f127fSDimitry Andric   //  R = or (and X, mask0), (and (shl Y, lsb), mask1)
2239753f127fSDimitry Andric   //  where mask1 = (2**size - 1) << lsb, mask0 = ~mask1
2240753f127fSDimitry Andric   //  =>
2241753f127fSDimitry Andric   //  R = BSTRINS X, Y, msb, lsb (where msb = lsb + size - 1)
2242753f127fSDimitry Andric   if (N0.getOpcode() == ISD::AND &&
2243753f127fSDimitry Andric       (CN0 = dyn_cast<ConstantSDNode>(N0.getOperand(1))) &&
2244753f127fSDimitry Andric       isShiftedMask_64(~CN0->getSExtValue(), MaskIdx0, MaskLen0) &&
2245753f127fSDimitry Andric       N1.getOpcode() == ISD::AND && N1.getOperand(0).getOpcode() == ISD::SHL &&
2246753f127fSDimitry Andric       (CN1 = dyn_cast<ConstantSDNode>(N1.getOperand(1))) &&
2247753f127fSDimitry Andric       isShiftedMask_64(CN1->getZExtValue(), MaskIdx1, MaskLen1) &&
2248753f127fSDimitry Andric       MaskIdx0 == MaskIdx1 && MaskLen0 == MaskLen1 &&
2249753f127fSDimitry Andric       (CN1 = dyn_cast<ConstantSDNode>(N1.getOperand(0).getOperand(1))) &&
2250753f127fSDimitry Andric       (Shamt = CN1->getZExtValue()) == MaskIdx0 &&
2251753f127fSDimitry Andric       (MaskIdx0 + MaskLen0 <= ValBits)) {
2252753f127fSDimitry Andric     LLVM_DEBUG(dbgs() << "Perform OR combine: match pattern 1\n");
2253753f127fSDimitry Andric     return DAG.getNode(LoongArchISD::BSTRINS, DL, ValTy, N0.getOperand(0),
2254753f127fSDimitry Andric                        N1.getOperand(0).getOperand(0),
2255753f127fSDimitry Andric                        DAG.getConstant((MaskIdx0 + MaskLen0 - 1), DL, GRLenVT),
2256753f127fSDimitry Andric                        DAG.getConstant(MaskIdx0, DL, GRLenVT));
2257753f127fSDimitry Andric   }
2258753f127fSDimitry Andric 
2259753f127fSDimitry Andric   // 2nd pattern to match BSTRINS:
2260753f127fSDimitry Andric   //  R = or (and X, mask0), (shl (and Y, mask1), lsb)
2261753f127fSDimitry Andric   //  where mask1 = (2**size - 1), mask0 = ~(mask1 << lsb)
2262753f127fSDimitry Andric   //  =>
2263753f127fSDimitry Andric   //  R = BSTRINS X, Y, msb, lsb (where msb = lsb + size - 1)
2264753f127fSDimitry Andric   if (N0.getOpcode() == ISD::AND &&
2265753f127fSDimitry Andric       (CN0 = dyn_cast<ConstantSDNode>(N0.getOperand(1))) &&
2266753f127fSDimitry Andric       isShiftedMask_64(~CN0->getSExtValue(), MaskIdx0, MaskLen0) &&
2267753f127fSDimitry Andric       N1.getOpcode() == ISD::SHL && N1.getOperand(0).getOpcode() == ISD::AND &&
2268753f127fSDimitry Andric       (CN1 = dyn_cast<ConstantSDNode>(N1.getOperand(1))) &&
2269753f127fSDimitry Andric       (Shamt = CN1->getZExtValue()) == MaskIdx0 &&
2270753f127fSDimitry Andric       (CN1 = dyn_cast<ConstantSDNode>(N1.getOperand(0).getOperand(1))) &&
2271753f127fSDimitry Andric       isShiftedMask_64(CN1->getZExtValue(), MaskIdx1, MaskLen1) &&
2272753f127fSDimitry Andric       MaskLen0 == MaskLen1 && MaskIdx1 == 0 &&
2273753f127fSDimitry Andric       (MaskIdx0 + MaskLen0 <= ValBits)) {
2274753f127fSDimitry Andric     LLVM_DEBUG(dbgs() << "Perform OR combine: match pattern 2\n");
2275753f127fSDimitry Andric     return DAG.getNode(LoongArchISD::BSTRINS, DL, ValTy, N0.getOperand(0),
2276753f127fSDimitry Andric                        N1.getOperand(0).getOperand(0),
2277753f127fSDimitry Andric                        DAG.getConstant((MaskIdx0 + MaskLen0 - 1), DL, GRLenVT),
2278753f127fSDimitry Andric                        DAG.getConstant(MaskIdx0, DL, GRLenVT));
2279753f127fSDimitry Andric   }
2280753f127fSDimitry Andric 
2281753f127fSDimitry Andric   // 3rd pattern to match BSTRINS:
2282753f127fSDimitry Andric   //  R = or (and X, mask0), (and Y, mask1)
2283753f127fSDimitry Andric   //  where ~mask0 = (2**size - 1) << lsb, mask0 & mask1 = 0
2284753f127fSDimitry Andric   //  =>
2285753f127fSDimitry Andric   //  R = BSTRINS X, (shr (and Y, mask1), lsb), msb, lsb
2286753f127fSDimitry Andric   //  where msb = lsb + size - 1
2287753f127fSDimitry Andric   if (N0.getOpcode() == ISD::AND && N1.getOpcode() == ISD::AND &&
2288753f127fSDimitry Andric       (CN0 = dyn_cast<ConstantSDNode>(N0.getOperand(1))) &&
2289753f127fSDimitry Andric       isShiftedMask_64(~CN0->getSExtValue(), MaskIdx0, MaskLen0) &&
2290753f127fSDimitry Andric       (MaskIdx0 + MaskLen0 <= 64) &&
2291753f127fSDimitry Andric       (CN1 = dyn_cast<ConstantSDNode>(N1->getOperand(1))) &&
2292753f127fSDimitry Andric       (CN1->getSExtValue() & CN0->getSExtValue()) == 0) {
2293753f127fSDimitry Andric     LLVM_DEBUG(dbgs() << "Perform OR combine: match pattern 3\n");
2294753f127fSDimitry Andric     return DAG.getNode(LoongArchISD::BSTRINS, DL, ValTy, N0.getOperand(0),
2295753f127fSDimitry Andric                        DAG.getNode(ISD::SRL, DL, N1->getValueType(0), N1,
2296753f127fSDimitry Andric                                    DAG.getConstant(MaskIdx0, DL, GRLenVT)),
2297753f127fSDimitry Andric                        DAG.getConstant(ValBits == 32
2298753f127fSDimitry Andric                                            ? (MaskIdx0 + (MaskLen0 & 31) - 1)
2299753f127fSDimitry Andric                                            : (MaskIdx0 + MaskLen0 - 1),
2300753f127fSDimitry Andric                                        DL, GRLenVT),
2301753f127fSDimitry Andric                        DAG.getConstant(MaskIdx0, DL, GRLenVT));
2302753f127fSDimitry Andric   }
2303753f127fSDimitry Andric 
2304753f127fSDimitry Andric   // 4th pattern to match BSTRINS:
2305753f127fSDimitry Andric   //  R = or (and X, mask), (shl Y, shamt)
2306753f127fSDimitry Andric   //  where mask = (2**shamt - 1)
2307753f127fSDimitry Andric   //  =>
2308753f127fSDimitry Andric   //  R = BSTRINS X, Y, ValBits - 1, shamt
2309753f127fSDimitry Andric   //  where ValBits = 32 or 64
2310753f127fSDimitry Andric   if (N0.getOpcode() == ISD::AND && N1.getOpcode() == ISD::SHL &&
2311753f127fSDimitry Andric       (CN0 = dyn_cast<ConstantSDNode>(N0.getOperand(1))) &&
2312753f127fSDimitry Andric       isShiftedMask_64(CN0->getZExtValue(), MaskIdx0, MaskLen0) &&
2313753f127fSDimitry Andric       MaskIdx0 == 0 && (CN1 = dyn_cast<ConstantSDNode>(N1.getOperand(1))) &&
2314753f127fSDimitry Andric       (Shamt = CN1->getZExtValue()) == MaskLen0 &&
2315753f127fSDimitry Andric       (MaskIdx0 + MaskLen0 <= ValBits)) {
2316753f127fSDimitry Andric     LLVM_DEBUG(dbgs() << "Perform OR combine: match pattern 4\n");
2317753f127fSDimitry Andric     return DAG.getNode(LoongArchISD::BSTRINS, DL, ValTy, N0.getOperand(0),
2318753f127fSDimitry Andric                        N1.getOperand(0),
2319753f127fSDimitry Andric                        DAG.getConstant((ValBits - 1), DL, GRLenVT),
2320753f127fSDimitry Andric                        DAG.getConstant(Shamt, DL, GRLenVT));
2321753f127fSDimitry Andric   }
2322753f127fSDimitry Andric 
2323753f127fSDimitry Andric   // 5th pattern to match BSTRINS:
2324753f127fSDimitry Andric   //  R = or (and X, mask), const
2325753f127fSDimitry Andric   //  where ~mask = (2**size - 1) << lsb, mask & const = 0
2326753f127fSDimitry Andric   //  =>
2327753f127fSDimitry Andric   //  R = BSTRINS X, (const >> lsb), msb, lsb
2328753f127fSDimitry Andric   //  where msb = lsb + size - 1
2329753f127fSDimitry Andric   if (N0.getOpcode() == ISD::AND &&
2330753f127fSDimitry Andric       (CN0 = dyn_cast<ConstantSDNode>(N0.getOperand(1))) &&
2331753f127fSDimitry Andric       isShiftedMask_64(~CN0->getSExtValue(), MaskIdx0, MaskLen0) &&
2332753f127fSDimitry Andric       (CN1 = dyn_cast<ConstantSDNode>(N1)) &&
2333753f127fSDimitry Andric       (CN1->getSExtValue() & CN0->getSExtValue()) == 0) {
2334753f127fSDimitry Andric     LLVM_DEBUG(dbgs() << "Perform OR combine: match pattern 5\n");
2335753f127fSDimitry Andric     return DAG.getNode(
2336753f127fSDimitry Andric         LoongArchISD::BSTRINS, DL, ValTy, N0.getOperand(0),
2337753f127fSDimitry Andric         DAG.getConstant(CN1->getSExtValue() >> MaskIdx0, DL, ValTy),
2338753f127fSDimitry Andric         DAG.getConstant((MaskIdx0 + MaskLen0 - 1), DL, GRLenVT),
2339753f127fSDimitry Andric         DAG.getConstant(MaskIdx0, DL, GRLenVT));
2340753f127fSDimitry Andric   }
2341753f127fSDimitry Andric 
2342753f127fSDimitry Andric   // 6th pattern.
2343753f127fSDimitry Andric   // a = b | ((c & mask) << shamt), where all positions in b to be overwritten
2344753f127fSDimitry Andric   // by the incoming bits are known to be zero.
2345753f127fSDimitry Andric   // =>
2346753f127fSDimitry Andric   // a = BSTRINS b, c, shamt + MaskLen - 1, shamt
2347753f127fSDimitry Andric   //
2348753f127fSDimitry Andric   // Note that the 1st pattern is a special situation of the 6th, i.e. the 6th
2349753f127fSDimitry Andric   // pattern is more common than the 1st. So we put the 1st before the 6th in
2350753f127fSDimitry Andric   // order to match as many nodes as possible.
2351753f127fSDimitry Andric   ConstantSDNode *CNMask, *CNShamt;
2352753f127fSDimitry Andric   unsigned MaskIdx, MaskLen;
2353753f127fSDimitry Andric   if (N1.getOpcode() == ISD::SHL && N1.getOperand(0).getOpcode() == ISD::AND &&
2354753f127fSDimitry Andric       (CNMask = dyn_cast<ConstantSDNode>(N1.getOperand(0).getOperand(1))) &&
2355753f127fSDimitry Andric       isShiftedMask_64(CNMask->getZExtValue(), MaskIdx, MaskLen) &&
2356753f127fSDimitry Andric       MaskIdx == 0 && (CNShamt = dyn_cast<ConstantSDNode>(N1.getOperand(1))) &&
2357753f127fSDimitry Andric       CNShamt->getZExtValue() + MaskLen <= ValBits) {
2358753f127fSDimitry Andric     Shamt = CNShamt->getZExtValue();
2359753f127fSDimitry Andric     APInt ShMask(ValBits, CNMask->getZExtValue() << Shamt);
2360753f127fSDimitry Andric     if (ShMask.isSubsetOf(DAG.computeKnownBits(N0).Zero)) {
2361753f127fSDimitry Andric       LLVM_DEBUG(dbgs() << "Perform OR combine: match pattern 6\n");
2362753f127fSDimitry Andric       return DAG.getNode(LoongArchISD::BSTRINS, DL, ValTy, N0,
2363753f127fSDimitry Andric                          N1.getOperand(0).getOperand(0),
2364753f127fSDimitry Andric                          DAG.getConstant(Shamt + MaskLen - 1, DL, GRLenVT),
2365753f127fSDimitry Andric                          DAG.getConstant(Shamt, DL, GRLenVT));
2366753f127fSDimitry Andric     }
2367753f127fSDimitry Andric   }
2368753f127fSDimitry Andric 
2369753f127fSDimitry Andric   // 7th pattern.
2370753f127fSDimitry Andric   // a = b | ((c << shamt) & shifted_mask), where all positions in b to be
2371753f127fSDimitry Andric   // overwritten by the incoming bits are known to be zero.
2372753f127fSDimitry Andric   // =>
2373753f127fSDimitry Andric   // a = BSTRINS b, c, MaskIdx + MaskLen - 1, MaskIdx
2374753f127fSDimitry Andric   //
2375753f127fSDimitry Andric   // Similarly, the 7th pattern is more common than the 2nd. So we put the 2nd
2376753f127fSDimitry Andric   // before the 7th in order to match as many nodes as possible.
2377753f127fSDimitry Andric   if (N1.getOpcode() == ISD::AND &&
2378753f127fSDimitry Andric       (CNMask = dyn_cast<ConstantSDNode>(N1.getOperand(1))) &&
2379753f127fSDimitry Andric       isShiftedMask_64(CNMask->getZExtValue(), MaskIdx, MaskLen) &&
2380753f127fSDimitry Andric       N1.getOperand(0).getOpcode() == ISD::SHL &&
2381753f127fSDimitry Andric       (CNShamt = dyn_cast<ConstantSDNode>(N1.getOperand(0).getOperand(1))) &&
2382753f127fSDimitry Andric       CNShamt->getZExtValue() == MaskIdx) {
2383753f127fSDimitry Andric     APInt ShMask(ValBits, CNMask->getZExtValue());
2384753f127fSDimitry Andric     if (ShMask.isSubsetOf(DAG.computeKnownBits(N0).Zero)) {
2385753f127fSDimitry Andric       LLVM_DEBUG(dbgs() << "Perform OR combine: match pattern 7\n");
2386753f127fSDimitry Andric       return DAG.getNode(LoongArchISD::BSTRINS, DL, ValTy, N0,
2387753f127fSDimitry Andric                          N1.getOperand(0).getOperand(0),
2388753f127fSDimitry Andric                          DAG.getConstant(MaskIdx + MaskLen - 1, DL, GRLenVT),
2389753f127fSDimitry Andric                          DAG.getConstant(MaskIdx, DL, GRLenVT));
2390753f127fSDimitry Andric     }
2391753f127fSDimitry Andric   }
2392753f127fSDimitry Andric 
2393753f127fSDimitry Andric   // (or a, b) and (or b, a) are equivalent, so swap the operands and retry.
2394753f127fSDimitry Andric   if (!SwapAndRetried) {
2395753f127fSDimitry Andric     std::swap(N0, N1);
2396753f127fSDimitry Andric     SwapAndRetried = true;
2397753f127fSDimitry Andric     goto Retry;
2398753f127fSDimitry Andric   }
2399753f127fSDimitry Andric 
2400753f127fSDimitry Andric   SwapAndRetried = false;
2401753f127fSDimitry Andric Retry2:
2402753f127fSDimitry Andric   // 8th pattern.
2403753f127fSDimitry Andric   // a = b | (c & shifted_mask), where all positions in b to be overwritten by
2404753f127fSDimitry Andric   // the incoming bits are known to be zero.
2405753f127fSDimitry Andric   // =>
2406753f127fSDimitry Andric   // a = BSTRINS b, c >> MaskIdx, MaskIdx + MaskLen - 1, MaskIdx
2407753f127fSDimitry Andric   //
2408753f127fSDimitry Andric   // Similarly, the 8th pattern is more common than the 4th and 5th patterns. So
2409753f127fSDimitry Andric   // we put it here in order to match as many nodes as possible or generate less
2410753f127fSDimitry Andric   // instructions.
2411753f127fSDimitry Andric   if (N1.getOpcode() == ISD::AND &&
2412753f127fSDimitry Andric       (CNMask = dyn_cast<ConstantSDNode>(N1.getOperand(1))) &&
2413753f127fSDimitry Andric       isShiftedMask_64(CNMask->getZExtValue(), MaskIdx, MaskLen)) {
2414753f127fSDimitry Andric     APInt ShMask(ValBits, CNMask->getZExtValue());
2415753f127fSDimitry Andric     if (ShMask.isSubsetOf(DAG.computeKnownBits(N0).Zero)) {
2416753f127fSDimitry Andric       LLVM_DEBUG(dbgs() << "Perform OR combine: match pattern 8\n");
2417753f127fSDimitry Andric       return DAG.getNode(LoongArchISD::BSTRINS, DL, ValTy, N0,
2418753f127fSDimitry Andric                          DAG.getNode(ISD::SRL, DL, N1->getValueType(0),
2419753f127fSDimitry Andric                                      N1->getOperand(0),
2420753f127fSDimitry Andric                                      DAG.getConstant(MaskIdx, DL, GRLenVT)),
2421753f127fSDimitry Andric                          DAG.getConstant(MaskIdx + MaskLen - 1, DL, GRLenVT),
2422753f127fSDimitry Andric                          DAG.getConstant(MaskIdx, DL, GRLenVT));
2423753f127fSDimitry Andric     }
2424753f127fSDimitry Andric   }
2425753f127fSDimitry Andric   // Swap N0/N1 and retry.
2426753f127fSDimitry Andric   if (!SwapAndRetried) {
2427753f127fSDimitry Andric     std::swap(N0, N1);
2428753f127fSDimitry Andric     SwapAndRetried = true;
2429753f127fSDimitry Andric     goto Retry2;
2430753f127fSDimitry Andric   }
2431753f127fSDimitry Andric 
2432753f127fSDimitry Andric   return SDValue();
2433753f127fSDimitry Andric }
2434753f127fSDimitry Andric 
2435bdd1243dSDimitry Andric // Combine (loongarch_bitrev_w (loongarch_revb_2w X)) to loongarch_bitrev_4b.
2436bdd1243dSDimitry Andric static SDValue performBITREV_WCombine(SDNode *N, SelectionDAG &DAG,
2437bdd1243dSDimitry Andric                                       TargetLowering::DAGCombinerInfo &DCI,
2438bdd1243dSDimitry Andric                                       const LoongArchSubtarget &Subtarget) {
2439bdd1243dSDimitry Andric   if (DCI.isBeforeLegalizeOps())
2440bdd1243dSDimitry Andric     return SDValue();
2441bdd1243dSDimitry Andric 
2442bdd1243dSDimitry Andric   SDValue Src = N->getOperand(0);
2443bdd1243dSDimitry Andric   if (Src.getOpcode() != LoongArchISD::REVB_2W)
2444bdd1243dSDimitry Andric     return SDValue();
2445bdd1243dSDimitry Andric 
2446bdd1243dSDimitry Andric   return DAG.getNode(LoongArchISD::BITREV_4B, SDLoc(N), N->getValueType(0),
2447bdd1243dSDimitry Andric                      Src.getOperand(0));
2448bdd1243dSDimitry Andric }
2449bdd1243dSDimitry Andric 
24505f757f3fSDimitry Andric template <unsigned N>
24515f757f3fSDimitry Andric static SDValue legalizeIntrinsicImmArg(SDNode *Node, unsigned ImmOp,
24525f757f3fSDimitry Andric                                        SelectionDAG &DAG,
24535f757f3fSDimitry Andric                                        const LoongArchSubtarget &Subtarget,
24545f757f3fSDimitry Andric                                        bool IsSigned = false) {
24555f757f3fSDimitry Andric   SDLoc DL(Node);
24565f757f3fSDimitry Andric   auto *CImm = cast<ConstantSDNode>(Node->getOperand(ImmOp));
24575f757f3fSDimitry Andric   // Check the ImmArg.
24585f757f3fSDimitry Andric   if ((IsSigned && !isInt<N>(CImm->getSExtValue())) ||
24595f757f3fSDimitry Andric       (!IsSigned && !isUInt<N>(CImm->getZExtValue()))) {
24605f757f3fSDimitry Andric     DAG.getContext()->emitError(Node->getOperationName(0) +
24615f757f3fSDimitry Andric                                 ": argument out of range.");
24625f757f3fSDimitry Andric     return DAG.getNode(ISD::UNDEF, DL, Subtarget.getGRLenVT());
24635f757f3fSDimitry Andric   }
24645f757f3fSDimitry Andric   return DAG.getConstant(CImm->getZExtValue(), DL, Subtarget.getGRLenVT());
24655f757f3fSDimitry Andric }
24665f757f3fSDimitry Andric 
24675f757f3fSDimitry Andric template <unsigned N>
24685f757f3fSDimitry Andric static SDValue lowerVectorSplatImm(SDNode *Node, unsigned ImmOp,
24695f757f3fSDimitry Andric                                    SelectionDAG &DAG, bool IsSigned = false) {
24705f757f3fSDimitry Andric   SDLoc DL(Node);
24715f757f3fSDimitry Andric   EVT ResTy = Node->getValueType(0);
24725f757f3fSDimitry Andric   auto *CImm = cast<ConstantSDNode>(Node->getOperand(ImmOp));
24735f757f3fSDimitry Andric 
24745f757f3fSDimitry Andric   // Check the ImmArg.
24755f757f3fSDimitry Andric   if ((IsSigned && !isInt<N>(CImm->getSExtValue())) ||
24765f757f3fSDimitry Andric       (!IsSigned && !isUInt<N>(CImm->getZExtValue()))) {
24775f757f3fSDimitry Andric     DAG.getContext()->emitError(Node->getOperationName(0) +
24785f757f3fSDimitry Andric                                 ": argument out of range.");
24795f757f3fSDimitry Andric     return DAG.getNode(ISD::UNDEF, DL, ResTy);
24805f757f3fSDimitry Andric   }
24815f757f3fSDimitry Andric   return DAG.getConstant(
24825f757f3fSDimitry Andric       APInt(ResTy.getScalarType().getSizeInBits(),
24835f757f3fSDimitry Andric             IsSigned ? CImm->getSExtValue() : CImm->getZExtValue(), IsSigned),
24845f757f3fSDimitry Andric       DL, ResTy);
24855f757f3fSDimitry Andric }
24865f757f3fSDimitry Andric 
24875f757f3fSDimitry Andric static SDValue truncateVecElts(SDNode *Node, SelectionDAG &DAG) {
24885f757f3fSDimitry Andric   SDLoc DL(Node);
24895f757f3fSDimitry Andric   EVT ResTy = Node->getValueType(0);
24905f757f3fSDimitry Andric   SDValue Vec = Node->getOperand(2);
24915f757f3fSDimitry Andric   SDValue Mask = DAG.getConstant(Vec.getScalarValueSizeInBits() - 1, DL, ResTy);
24925f757f3fSDimitry Andric   return DAG.getNode(ISD::AND, DL, ResTy, Vec, Mask);
24935f757f3fSDimitry Andric }
24945f757f3fSDimitry Andric 
24955f757f3fSDimitry Andric static SDValue lowerVectorBitClear(SDNode *Node, SelectionDAG &DAG) {
24965f757f3fSDimitry Andric   SDLoc DL(Node);
24975f757f3fSDimitry Andric   EVT ResTy = Node->getValueType(0);
24985f757f3fSDimitry Andric   SDValue One = DAG.getConstant(1, DL, ResTy);
24995f757f3fSDimitry Andric   SDValue Bit =
25005f757f3fSDimitry Andric       DAG.getNode(ISD::SHL, DL, ResTy, One, truncateVecElts(Node, DAG));
25015f757f3fSDimitry Andric 
25025f757f3fSDimitry Andric   return DAG.getNode(ISD::AND, DL, ResTy, Node->getOperand(1),
25035f757f3fSDimitry Andric                      DAG.getNOT(DL, Bit, ResTy));
25045f757f3fSDimitry Andric }
25055f757f3fSDimitry Andric 
25065f757f3fSDimitry Andric template <unsigned N>
25075f757f3fSDimitry Andric static SDValue lowerVectorBitClearImm(SDNode *Node, SelectionDAG &DAG) {
25085f757f3fSDimitry Andric   SDLoc DL(Node);
25095f757f3fSDimitry Andric   EVT ResTy = Node->getValueType(0);
25105f757f3fSDimitry Andric   auto *CImm = cast<ConstantSDNode>(Node->getOperand(2));
25115f757f3fSDimitry Andric   // Check the unsigned ImmArg.
25125f757f3fSDimitry Andric   if (!isUInt<N>(CImm->getZExtValue())) {
25135f757f3fSDimitry Andric     DAG.getContext()->emitError(Node->getOperationName(0) +
25145f757f3fSDimitry Andric                                 ": argument out of range.");
25155f757f3fSDimitry Andric     return DAG.getNode(ISD::UNDEF, DL, ResTy);
25165f757f3fSDimitry Andric   }
25175f757f3fSDimitry Andric 
25185f757f3fSDimitry Andric   APInt BitImm = APInt(ResTy.getScalarSizeInBits(), 1) << CImm->getAPIntValue();
25195f757f3fSDimitry Andric   SDValue Mask = DAG.getConstant(~BitImm, DL, ResTy);
25205f757f3fSDimitry Andric 
25215f757f3fSDimitry Andric   return DAG.getNode(ISD::AND, DL, ResTy, Node->getOperand(1), Mask);
25225f757f3fSDimitry Andric }
25235f757f3fSDimitry Andric 
25245f757f3fSDimitry Andric template <unsigned N>
25255f757f3fSDimitry Andric static SDValue lowerVectorBitSetImm(SDNode *Node, SelectionDAG &DAG) {
25265f757f3fSDimitry Andric   SDLoc DL(Node);
25275f757f3fSDimitry Andric   EVT ResTy = Node->getValueType(0);
25285f757f3fSDimitry Andric   auto *CImm = cast<ConstantSDNode>(Node->getOperand(2));
25295f757f3fSDimitry Andric   // Check the unsigned ImmArg.
25305f757f3fSDimitry Andric   if (!isUInt<N>(CImm->getZExtValue())) {
25315f757f3fSDimitry Andric     DAG.getContext()->emitError(Node->getOperationName(0) +
25325f757f3fSDimitry Andric                                 ": argument out of range.");
25335f757f3fSDimitry Andric     return DAG.getNode(ISD::UNDEF, DL, ResTy);
25345f757f3fSDimitry Andric   }
25355f757f3fSDimitry Andric 
25365f757f3fSDimitry Andric   APInt Imm = APInt(ResTy.getScalarSizeInBits(), 1) << CImm->getAPIntValue();
25375f757f3fSDimitry Andric   SDValue BitImm = DAG.getConstant(Imm, DL, ResTy);
25385f757f3fSDimitry Andric   return DAG.getNode(ISD::OR, DL, ResTy, Node->getOperand(1), BitImm);
25395f757f3fSDimitry Andric }
25405f757f3fSDimitry Andric 
25415f757f3fSDimitry Andric template <unsigned N>
25425f757f3fSDimitry Andric static SDValue lowerVectorBitRevImm(SDNode *Node, SelectionDAG &DAG) {
25435f757f3fSDimitry Andric   SDLoc DL(Node);
25445f757f3fSDimitry Andric   EVT ResTy = Node->getValueType(0);
25455f757f3fSDimitry Andric   auto *CImm = cast<ConstantSDNode>(Node->getOperand(2));
25465f757f3fSDimitry Andric   // Check the unsigned ImmArg.
25475f757f3fSDimitry Andric   if (!isUInt<N>(CImm->getZExtValue())) {
25485f757f3fSDimitry Andric     DAG.getContext()->emitError(Node->getOperationName(0) +
25495f757f3fSDimitry Andric                                 ": argument out of range.");
25505f757f3fSDimitry Andric     return DAG.getNode(ISD::UNDEF, DL, ResTy);
25515f757f3fSDimitry Andric   }
25525f757f3fSDimitry Andric 
25535f757f3fSDimitry Andric   APInt Imm = APInt(ResTy.getScalarSizeInBits(), 1) << CImm->getAPIntValue();
25545f757f3fSDimitry Andric   SDValue BitImm = DAG.getConstant(Imm, DL, ResTy);
25555f757f3fSDimitry Andric   return DAG.getNode(ISD::XOR, DL, ResTy, Node->getOperand(1), BitImm);
25565f757f3fSDimitry Andric }
25575f757f3fSDimitry Andric 
25585f757f3fSDimitry Andric static SDValue
25595f757f3fSDimitry Andric performINTRINSIC_WO_CHAINCombine(SDNode *N, SelectionDAG &DAG,
25605f757f3fSDimitry Andric                                  TargetLowering::DAGCombinerInfo &DCI,
25615f757f3fSDimitry Andric                                  const LoongArchSubtarget &Subtarget) {
25625f757f3fSDimitry Andric   SDLoc DL(N);
25635f757f3fSDimitry Andric   switch (N->getConstantOperandVal(0)) {
25645f757f3fSDimitry Andric   default:
25655f757f3fSDimitry Andric     break;
25665f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vadd_b:
25675f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vadd_h:
25685f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vadd_w:
25695f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vadd_d:
25705f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvadd_b:
25715f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvadd_h:
25725f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvadd_w:
25735f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvadd_d:
25745f757f3fSDimitry Andric     return DAG.getNode(ISD::ADD, DL, N->getValueType(0), N->getOperand(1),
25755f757f3fSDimitry Andric                        N->getOperand(2));
25765f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vaddi_bu:
25775f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vaddi_hu:
25785f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vaddi_wu:
25795f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vaddi_du:
25805f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvaddi_bu:
25815f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvaddi_hu:
25825f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvaddi_wu:
25835f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvaddi_du:
25845f757f3fSDimitry Andric     return DAG.getNode(ISD::ADD, DL, N->getValueType(0), N->getOperand(1),
25855f757f3fSDimitry Andric                        lowerVectorSplatImm<5>(N, 2, DAG));
25865f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vsub_b:
25875f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vsub_h:
25885f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vsub_w:
25895f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vsub_d:
25905f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvsub_b:
25915f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvsub_h:
25925f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvsub_w:
25935f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvsub_d:
25945f757f3fSDimitry Andric     return DAG.getNode(ISD::SUB, DL, N->getValueType(0), N->getOperand(1),
25955f757f3fSDimitry Andric                        N->getOperand(2));
25965f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vsubi_bu:
25975f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vsubi_hu:
25985f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vsubi_wu:
25995f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vsubi_du:
26005f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvsubi_bu:
26015f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvsubi_hu:
26025f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvsubi_wu:
26035f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvsubi_du:
26045f757f3fSDimitry Andric     return DAG.getNode(ISD::SUB, DL, N->getValueType(0), N->getOperand(1),
26055f757f3fSDimitry Andric                        lowerVectorSplatImm<5>(N, 2, DAG));
26065f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vneg_b:
26075f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vneg_h:
26085f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vneg_w:
26095f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vneg_d:
26105f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvneg_b:
26115f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvneg_h:
26125f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvneg_w:
26135f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvneg_d:
26145f757f3fSDimitry Andric     return DAG.getNode(
26155f757f3fSDimitry Andric         ISD::SUB, DL, N->getValueType(0),
26165f757f3fSDimitry Andric         DAG.getConstant(
26175f757f3fSDimitry Andric             APInt(N->getValueType(0).getScalarType().getSizeInBits(), 0,
26185f757f3fSDimitry Andric                   /*isSigned=*/true),
26195f757f3fSDimitry Andric             SDLoc(N), N->getValueType(0)),
26205f757f3fSDimitry Andric         N->getOperand(1));
26215f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vmax_b:
26225f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vmax_h:
26235f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vmax_w:
26245f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vmax_d:
26255f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvmax_b:
26265f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvmax_h:
26275f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvmax_w:
26285f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvmax_d:
26295f757f3fSDimitry Andric     return DAG.getNode(ISD::SMAX, DL, N->getValueType(0), N->getOperand(1),
26305f757f3fSDimitry Andric                        N->getOperand(2));
26315f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vmax_bu:
26325f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vmax_hu:
26335f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vmax_wu:
26345f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vmax_du:
26355f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvmax_bu:
26365f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvmax_hu:
26375f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvmax_wu:
26385f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvmax_du:
26395f757f3fSDimitry Andric     return DAG.getNode(ISD::UMAX, DL, N->getValueType(0), N->getOperand(1),
26405f757f3fSDimitry Andric                        N->getOperand(2));
26415f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vmaxi_b:
26425f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vmaxi_h:
26435f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vmaxi_w:
26445f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vmaxi_d:
26455f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvmaxi_b:
26465f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvmaxi_h:
26475f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvmaxi_w:
26485f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvmaxi_d:
26495f757f3fSDimitry Andric     return DAG.getNode(ISD::SMAX, DL, N->getValueType(0), N->getOperand(1),
26505f757f3fSDimitry Andric                        lowerVectorSplatImm<5>(N, 2, DAG, /*IsSigned=*/true));
26515f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vmaxi_bu:
26525f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vmaxi_hu:
26535f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vmaxi_wu:
26545f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vmaxi_du:
26555f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvmaxi_bu:
26565f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvmaxi_hu:
26575f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvmaxi_wu:
26585f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvmaxi_du:
26595f757f3fSDimitry Andric     return DAG.getNode(ISD::UMAX, DL, N->getValueType(0), N->getOperand(1),
26605f757f3fSDimitry Andric                        lowerVectorSplatImm<5>(N, 2, DAG));
26615f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vmin_b:
26625f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vmin_h:
26635f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vmin_w:
26645f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vmin_d:
26655f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvmin_b:
26665f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvmin_h:
26675f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvmin_w:
26685f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvmin_d:
26695f757f3fSDimitry Andric     return DAG.getNode(ISD::SMIN, DL, N->getValueType(0), N->getOperand(1),
26705f757f3fSDimitry Andric                        N->getOperand(2));
26715f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vmin_bu:
26725f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vmin_hu:
26735f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vmin_wu:
26745f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vmin_du:
26755f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvmin_bu:
26765f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvmin_hu:
26775f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvmin_wu:
26785f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvmin_du:
26795f757f3fSDimitry Andric     return DAG.getNode(ISD::UMIN, DL, N->getValueType(0), N->getOperand(1),
26805f757f3fSDimitry Andric                        N->getOperand(2));
26815f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vmini_b:
26825f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vmini_h:
26835f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vmini_w:
26845f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vmini_d:
26855f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvmini_b:
26865f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvmini_h:
26875f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvmini_w:
26885f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvmini_d:
26895f757f3fSDimitry Andric     return DAG.getNode(ISD::SMIN, DL, N->getValueType(0), N->getOperand(1),
26905f757f3fSDimitry Andric                        lowerVectorSplatImm<5>(N, 2, DAG, /*IsSigned=*/true));
26915f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vmini_bu:
26925f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vmini_hu:
26935f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vmini_wu:
26945f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vmini_du:
26955f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvmini_bu:
26965f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvmini_hu:
26975f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvmini_wu:
26985f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvmini_du:
26995f757f3fSDimitry Andric     return DAG.getNode(ISD::UMIN, DL, N->getValueType(0), N->getOperand(1),
27005f757f3fSDimitry Andric                        lowerVectorSplatImm<5>(N, 2, DAG));
27015f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vmul_b:
27025f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vmul_h:
27035f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vmul_w:
27045f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vmul_d:
27055f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvmul_b:
27065f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvmul_h:
27075f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvmul_w:
27085f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvmul_d:
27095f757f3fSDimitry Andric     return DAG.getNode(ISD::MUL, DL, N->getValueType(0), N->getOperand(1),
27105f757f3fSDimitry Andric                        N->getOperand(2));
27115f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vmadd_b:
27125f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vmadd_h:
27135f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vmadd_w:
27145f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vmadd_d:
27155f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvmadd_b:
27165f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvmadd_h:
27175f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvmadd_w:
27185f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvmadd_d: {
27195f757f3fSDimitry Andric     EVT ResTy = N->getValueType(0);
27205f757f3fSDimitry Andric     return DAG.getNode(ISD::ADD, SDLoc(N), ResTy, N->getOperand(1),
27215f757f3fSDimitry Andric                        DAG.getNode(ISD::MUL, SDLoc(N), ResTy, N->getOperand(2),
27225f757f3fSDimitry Andric                                    N->getOperand(3)));
27235f757f3fSDimitry Andric   }
27245f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vmsub_b:
27255f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vmsub_h:
27265f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vmsub_w:
27275f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vmsub_d:
27285f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvmsub_b:
27295f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvmsub_h:
27305f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvmsub_w:
27315f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvmsub_d: {
27325f757f3fSDimitry Andric     EVT ResTy = N->getValueType(0);
27335f757f3fSDimitry Andric     return DAG.getNode(ISD::SUB, SDLoc(N), ResTy, N->getOperand(1),
27345f757f3fSDimitry Andric                        DAG.getNode(ISD::MUL, SDLoc(N), ResTy, N->getOperand(2),
27355f757f3fSDimitry Andric                                    N->getOperand(3)));
27365f757f3fSDimitry Andric   }
27375f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vdiv_b:
27385f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vdiv_h:
27395f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vdiv_w:
27405f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vdiv_d:
27415f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvdiv_b:
27425f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvdiv_h:
27435f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvdiv_w:
27445f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvdiv_d:
27455f757f3fSDimitry Andric     return DAG.getNode(ISD::SDIV, DL, N->getValueType(0), N->getOperand(1),
27465f757f3fSDimitry Andric                        N->getOperand(2));
27475f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vdiv_bu:
27485f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vdiv_hu:
27495f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vdiv_wu:
27505f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vdiv_du:
27515f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvdiv_bu:
27525f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvdiv_hu:
27535f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvdiv_wu:
27545f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvdiv_du:
27555f757f3fSDimitry Andric     return DAG.getNode(ISD::UDIV, DL, N->getValueType(0), N->getOperand(1),
27565f757f3fSDimitry Andric                        N->getOperand(2));
27575f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vmod_b:
27585f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vmod_h:
27595f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vmod_w:
27605f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vmod_d:
27615f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvmod_b:
27625f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvmod_h:
27635f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvmod_w:
27645f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvmod_d:
27655f757f3fSDimitry Andric     return DAG.getNode(ISD::SREM, DL, N->getValueType(0), N->getOperand(1),
27665f757f3fSDimitry Andric                        N->getOperand(2));
27675f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vmod_bu:
27685f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vmod_hu:
27695f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vmod_wu:
27705f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vmod_du:
27715f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvmod_bu:
27725f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvmod_hu:
27735f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvmod_wu:
27745f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvmod_du:
27755f757f3fSDimitry Andric     return DAG.getNode(ISD::UREM, DL, N->getValueType(0), N->getOperand(1),
27765f757f3fSDimitry Andric                        N->getOperand(2));
27775f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vand_v:
27785f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvand_v:
27795f757f3fSDimitry Andric     return DAG.getNode(ISD::AND, DL, N->getValueType(0), N->getOperand(1),
27805f757f3fSDimitry Andric                        N->getOperand(2));
27815f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vor_v:
27825f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvor_v:
27835f757f3fSDimitry Andric     return DAG.getNode(ISD::OR, DL, N->getValueType(0), N->getOperand(1),
27845f757f3fSDimitry Andric                        N->getOperand(2));
27855f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vxor_v:
27865f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvxor_v:
27875f757f3fSDimitry Andric     return DAG.getNode(ISD::XOR, DL, N->getValueType(0), N->getOperand(1),
27885f757f3fSDimitry Andric                        N->getOperand(2));
27895f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vnor_v:
27905f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvnor_v: {
27915f757f3fSDimitry Andric     SDValue Res = DAG.getNode(ISD::OR, DL, N->getValueType(0), N->getOperand(1),
27925f757f3fSDimitry Andric                               N->getOperand(2));
27935f757f3fSDimitry Andric     return DAG.getNOT(DL, Res, Res->getValueType(0));
27945f757f3fSDimitry Andric   }
27955f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vandi_b:
27965f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvandi_b:
27975f757f3fSDimitry Andric     return DAG.getNode(ISD::AND, DL, N->getValueType(0), N->getOperand(1),
27985f757f3fSDimitry Andric                        lowerVectorSplatImm<8>(N, 2, DAG));
27995f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vori_b:
28005f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvori_b:
28015f757f3fSDimitry Andric     return DAG.getNode(ISD::OR, DL, N->getValueType(0), N->getOperand(1),
28025f757f3fSDimitry Andric                        lowerVectorSplatImm<8>(N, 2, DAG));
28035f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vxori_b:
28045f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvxori_b:
28055f757f3fSDimitry Andric     return DAG.getNode(ISD::XOR, DL, N->getValueType(0), N->getOperand(1),
28065f757f3fSDimitry Andric                        lowerVectorSplatImm<8>(N, 2, DAG));
28075f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vsll_b:
28085f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vsll_h:
28095f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vsll_w:
28105f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vsll_d:
28115f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvsll_b:
28125f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvsll_h:
28135f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvsll_w:
28145f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvsll_d:
28155f757f3fSDimitry Andric     return DAG.getNode(ISD::SHL, DL, N->getValueType(0), N->getOperand(1),
28165f757f3fSDimitry Andric                        truncateVecElts(N, DAG));
28175f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vslli_b:
28185f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvslli_b:
28195f757f3fSDimitry Andric     return DAG.getNode(ISD::SHL, DL, N->getValueType(0), N->getOperand(1),
28205f757f3fSDimitry Andric                        lowerVectorSplatImm<3>(N, 2, DAG));
28215f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vslli_h:
28225f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvslli_h:
28235f757f3fSDimitry Andric     return DAG.getNode(ISD::SHL, DL, N->getValueType(0), N->getOperand(1),
28245f757f3fSDimitry Andric                        lowerVectorSplatImm<4>(N, 2, DAG));
28255f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vslli_w:
28265f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvslli_w:
28275f757f3fSDimitry Andric     return DAG.getNode(ISD::SHL, DL, N->getValueType(0), N->getOperand(1),
28285f757f3fSDimitry Andric                        lowerVectorSplatImm<5>(N, 2, DAG));
28295f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vslli_d:
28305f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvslli_d:
28315f757f3fSDimitry Andric     return DAG.getNode(ISD::SHL, DL, N->getValueType(0), N->getOperand(1),
28325f757f3fSDimitry Andric                        lowerVectorSplatImm<6>(N, 2, DAG));
28335f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vsrl_b:
28345f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vsrl_h:
28355f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vsrl_w:
28365f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vsrl_d:
28375f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvsrl_b:
28385f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvsrl_h:
28395f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvsrl_w:
28405f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvsrl_d:
28415f757f3fSDimitry Andric     return DAG.getNode(ISD::SRL, DL, N->getValueType(0), N->getOperand(1),
28425f757f3fSDimitry Andric                        truncateVecElts(N, DAG));
28435f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vsrli_b:
28445f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvsrli_b:
28455f757f3fSDimitry Andric     return DAG.getNode(ISD::SRL, DL, N->getValueType(0), N->getOperand(1),
28465f757f3fSDimitry Andric                        lowerVectorSplatImm<3>(N, 2, DAG));
28475f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vsrli_h:
28485f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvsrli_h:
28495f757f3fSDimitry Andric     return DAG.getNode(ISD::SRL, DL, N->getValueType(0), N->getOperand(1),
28505f757f3fSDimitry Andric                        lowerVectorSplatImm<4>(N, 2, DAG));
28515f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vsrli_w:
28525f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvsrli_w:
28535f757f3fSDimitry Andric     return DAG.getNode(ISD::SRL, DL, N->getValueType(0), N->getOperand(1),
28545f757f3fSDimitry Andric                        lowerVectorSplatImm<5>(N, 2, DAG));
28555f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vsrli_d:
28565f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvsrli_d:
28575f757f3fSDimitry Andric     return DAG.getNode(ISD::SRL, DL, N->getValueType(0), N->getOperand(1),
28585f757f3fSDimitry Andric                        lowerVectorSplatImm<6>(N, 2, DAG));
28595f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vsra_b:
28605f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vsra_h:
28615f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vsra_w:
28625f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vsra_d:
28635f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvsra_b:
28645f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvsra_h:
28655f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvsra_w:
28665f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvsra_d:
28675f757f3fSDimitry Andric     return DAG.getNode(ISD::SRA, DL, N->getValueType(0), N->getOperand(1),
28685f757f3fSDimitry Andric                        truncateVecElts(N, DAG));
28695f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vsrai_b:
28705f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvsrai_b:
28715f757f3fSDimitry Andric     return DAG.getNode(ISD::SRA, DL, N->getValueType(0), N->getOperand(1),
28725f757f3fSDimitry Andric                        lowerVectorSplatImm<3>(N, 2, DAG));
28735f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vsrai_h:
28745f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvsrai_h:
28755f757f3fSDimitry Andric     return DAG.getNode(ISD::SRA, DL, N->getValueType(0), N->getOperand(1),
28765f757f3fSDimitry Andric                        lowerVectorSplatImm<4>(N, 2, DAG));
28775f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vsrai_w:
28785f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvsrai_w:
28795f757f3fSDimitry Andric     return DAG.getNode(ISD::SRA, DL, N->getValueType(0), N->getOperand(1),
28805f757f3fSDimitry Andric                        lowerVectorSplatImm<5>(N, 2, DAG));
28815f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vsrai_d:
28825f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvsrai_d:
28835f757f3fSDimitry Andric     return DAG.getNode(ISD::SRA, DL, N->getValueType(0), N->getOperand(1),
28845f757f3fSDimitry Andric                        lowerVectorSplatImm<6>(N, 2, DAG));
28855f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vclz_b:
28865f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vclz_h:
28875f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vclz_w:
28885f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vclz_d:
28895f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvclz_b:
28905f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvclz_h:
28915f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvclz_w:
28925f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvclz_d:
28935f757f3fSDimitry Andric     return DAG.getNode(ISD::CTLZ, DL, N->getValueType(0), N->getOperand(1));
28945f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vpcnt_b:
28955f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vpcnt_h:
28965f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vpcnt_w:
28975f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vpcnt_d:
28985f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvpcnt_b:
28995f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvpcnt_h:
29005f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvpcnt_w:
29015f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvpcnt_d:
29025f757f3fSDimitry Andric     return DAG.getNode(ISD::CTPOP, DL, N->getValueType(0), N->getOperand(1));
29035f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vbitclr_b:
29045f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vbitclr_h:
29055f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vbitclr_w:
29065f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vbitclr_d:
29075f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvbitclr_b:
29085f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvbitclr_h:
29095f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvbitclr_w:
29105f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvbitclr_d:
29115f757f3fSDimitry Andric     return lowerVectorBitClear(N, DAG);
29125f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vbitclri_b:
29135f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvbitclri_b:
29145f757f3fSDimitry Andric     return lowerVectorBitClearImm<3>(N, DAG);
29155f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vbitclri_h:
29165f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvbitclri_h:
29175f757f3fSDimitry Andric     return lowerVectorBitClearImm<4>(N, DAG);
29185f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vbitclri_w:
29195f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvbitclri_w:
29205f757f3fSDimitry Andric     return lowerVectorBitClearImm<5>(N, DAG);
29215f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vbitclri_d:
29225f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvbitclri_d:
29235f757f3fSDimitry Andric     return lowerVectorBitClearImm<6>(N, DAG);
29245f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vbitset_b:
29255f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vbitset_h:
29265f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vbitset_w:
29275f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vbitset_d:
29285f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvbitset_b:
29295f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvbitset_h:
29305f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvbitset_w:
29315f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvbitset_d: {
29325f757f3fSDimitry Andric     EVT VecTy = N->getValueType(0);
29335f757f3fSDimitry Andric     SDValue One = DAG.getConstant(1, DL, VecTy);
29345f757f3fSDimitry Andric     return DAG.getNode(
29355f757f3fSDimitry Andric         ISD::OR, DL, VecTy, N->getOperand(1),
29365f757f3fSDimitry Andric         DAG.getNode(ISD::SHL, DL, VecTy, One, truncateVecElts(N, DAG)));
29375f757f3fSDimitry Andric   }
29385f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vbitseti_b:
29395f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvbitseti_b:
29405f757f3fSDimitry Andric     return lowerVectorBitSetImm<3>(N, DAG);
29415f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vbitseti_h:
29425f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvbitseti_h:
29435f757f3fSDimitry Andric     return lowerVectorBitSetImm<4>(N, DAG);
29445f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vbitseti_w:
29455f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvbitseti_w:
29465f757f3fSDimitry Andric     return lowerVectorBitSetImm<5>(N, DAG);
29475f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vbitseti_d:
29485f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvbitseti_d:
29495f757f3fSDimitry Andric     return lowerVectorBitSetImm<6>(N, DAG);
29505f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vbitrev_b:
29515f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vbitrev_h:
29525f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vbitrev_w:
29535f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vbitrev_d:
29545f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvbitrev_b:
29555f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvbitrev_h:
29565f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvbitrev_w:
29575f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvbitrev_d: {
29585f757f3fSDimitry Andric     EVT VecTy = N->getValueType(0);
29595f757f3fSDimitry Andric     SDValue One = DAG.getConstant(1, DL, VecTy);
29605f757f3fSDimitry Andric     return DAG.getNode(
29615f757f3fSDimitry Andric         ISD::XOR, DL, VecTy, N->getOperand(1),
29625f757f3fSDimitry Andric         DAG.getNode(ISD::SHL, DL, VecTy, One, truncateVecElts(N, DAG)));
29635f757f3fSDimitry Andric   }
29645f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vbitrevi_b:
29655f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvbitrevi_b:
29665f757f3fSDimitry Andric     return lowerVectorBitRevImm<3>(N, DAG);
29675f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vbitrevi_h:
29685f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvbitrevi_h:
29695f757f3fSDimitry Andric     return lowerVectorBitRevImm<4>(N, DAG);
29705f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vbitrevi_w:
29715f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvbitrevi_w:
29725f757f3fSDimitry Andric     return lowerVectorBitRevImm<5>(N, DAG);
29735f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vbitrevi_d:
29745f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvbitrevi_d:
29755f757f3fSDimitry Andric     return lowerVectorBitRevImm<6>(N, DAG);
29765f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vfadd_s:
29775f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vfadd_d:
29785f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvfadd_s:
29795f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvfadd_d:
29805f757f3fSDimitry Andric     return DAG.getNode(ISD::FADD, DL, N->getValueType(0), N->getOperand(1),
29815f757f3fSDimitry Andric                        N->getOperand(2));
29825f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vfsub_s:
29835f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vfsub_d:
29845f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvfsub_s:
29855f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvfsub_d:
29865f757f3fSDimitry Andric     return DAG.getNode(ISD::FSUB, DL, N->getValueType(0), N->getOperand(1),
29875f757f3fSDimitry Andric                        N->getOperand(2));
29885f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vfmul_s:
29895f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vfmul_d:
29905f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvfmul_s:
29915f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvfmul_d:
29925f757f3fSDimitry Andric     return DAG.getNode(ISD::FMUL, DL, N->getValueType(0), N->getOperand(1),
29935f757f3fSDimitry Andric                        N->getOperand(2));
29945f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vfdiv_s:
29955f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vfdiv_d:
29965f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvfdiv_s:
29975f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvfdiv_d:
29985f757f3fSDimitry Andric     return DAG.getNode(ISD::FDIV, DL, N->getValueType(0), N->getOperand(1),
29995f757f3fSDimitry Andric                        N->getOperand(2));
30005f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vfmadd_s:
30015f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vfmadd_d:
30025f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvfmadd_s:
30035f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvfmadd_d:
30045f757f3fSDimitry Andric     return DAG.getNode(ISD::FMA, DL, N->getValueType(0), N->getOperand(1),
30055f757f3fSDimitry Andric                        N->getOperand(2), N->getOperand(3));
30065f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vinsgr2vr_b:
30075f757f3fSDimitry Andric     return DAG.getNode(ISD::INSERT_VECTOR_ELT, SDLoc(N), N->getValueType(0),
30085f757f3fSDimitry Andric                        N->getOperand(1), N->getOperand(2),
30095f757f3fSDimitry Andric                        legalizeIntrinsicImmArg<4>(N, 3, DAG, Subtarget));
30105f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vinsgr2vr_h:
30115f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvinsgr2vr_w:
30125f757f3fSDimitry Andric     return DAG.getNode(ISD::INSERT_VECTOR_ELT, SDLoc(N), N->getValueType(0),
30135f757f3fSDimitry Andric                        N->getOperand(1), N->getOperand(2),
30145f757f3fSDimitry Andric                        legalizeIntrinsicImmArg<3>(N, 3, DAG, Subtarget));
30155f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vinsgr2vr_w:
30165f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvinsgr2vr_d:
30175f757f3fSDimitry Andric     return DAG.getNode(ISD::INSERT_VECTOR_ELT, SDLoc(N), N->getValueType(0),
30185f757f3fSDimitry Andric                        N->getOperand(1), N->getOperand(2),
30195f757f3fSDimitry Andric                        legalizeIntrinsicImmArg<2>(N, 3, DAG, Subtarget));
30205f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vinsgr2vr_d:
30215f757f3fSDimitry Andric     return DAG.getNode(ISD::INSERT_VECTOR_ELT, SDLoc(N), N->getValueType(0),
30225f757f3fSDimitry Andric                        N->getOperand(1), N->getOperand(2),
30235f757f3fSDimitry Andric                        legalizeIntrinsicImmArg<1>(N, 3, DAG, Subtarget));
30245f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vreplgr2vr_b:
30255f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vreplgr2vr_h:
30265f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vreplgr2vr_w:
30275f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vreplgr2vr_d:
30285f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvreplgr2vr_b:
30295f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvreplgr2vr_h:
30305f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvreplgr2vr_w:
30315f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvreplgr2vr_d: {
30325f757f3fSDimitry Andric     EVT ResTy = N->getValueType(0);
30335f757f3fSDimitry Andric     SmallVector<SDValue> Ops(ResTy.getVectorNumElements(), N->getOperand(1));
30345f757f3fSDimitry Andric     return DAG.getBuildVector(ResTy, DL, Ops);
30355f757f3fSDimitry Andric   }
30365f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vreplve_b:
30375f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vreplve_h:
30385f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vreplve_w:
30395f757f3fSDimitry Andric   case Intrinsic::loongarch_lsx_vreplve_d:
30405f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvreplve_b:
30415f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvreplve_h:
30425f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvreplve_w:
30435f757f3fSDimitry Andric   case Intrinsic::loongarch_lasx_xvreplve_d:
30445f757f3fSDimitry Andric     return DAG.getNode(LoongArchISD::VREPLVE, DL, N->getValueType(0),
30455f757f3fSDimitry Andric                        N->getOperand(1),
30465f757f3fSDimitry Andric                        DAG.getNode(ISD::ANY_EXTEND, DL, Subtarget.getGRLenVT(),
30475f757f3fSDimitry Andric                                    N->getOperand(2)));
30485f757f3fSDimitry Andric   }
30495f757f3fSDimitry Andric   return SDValue();
30505f757f3fSDimitry Andric }
30515f757f3fSDimitry Andric 
305281ad6265SDimitry Andric SDValue LoongArchTargetLowering::PerformDAGCombine(SDNode *N,
305381ad6265SDimitry Andric                                                    DAGCombinerInfo &DCI) const {
305481ad6265SDimitry Andric   SelectionDAG &DAG = DCI.DAG;
305581ad6265SDimitry Andric   switch (N->getOpcode()) {
305681ad6265SDimitry Andric   default:
305781ad6265SDimitry Andric     break;
305881ad6265SDimitry Andric   case ISD::AND:
305981ad6265SDimitry Andric     return performANDCombine(N, DAG, DCI, Subtarget);
3060753f127fSDimitry Andric   case ISD::OR:
3061753f127fSDimitry Andric     return performORCombine(N, DAG, DCI, Subtarget);
306281ad6265SDimitry Andric   case ISD::SRL:
306381ad6265SDimitry Andric     return performSRLCombine(N, DAG, DCI, Subtarget);
3064bdd1243dSDimitry Andric   case LoongArchISD::BITREV_W:
3065bdd1243dSDimitry Andric     return performBITREV_WCombine(N, DAG, DCI, Subtarget);
30665f757f3fSDimitry Andric   case ISD::INTRINSIC_WO_CHAIN:
30675f757f3fSDimitry Andric     return performINTRINSIC_WO_CHAINCombine(N, DAG, DCI, Subtarget);
306881ad6265SDimitry Andric   }
306981ad6265SDimitry Andric   return SDValue();
307081ad6265SDimitry Andric }
307181ad6265SDimitry Andric 
3072753f127fSDimitry Andric static MachineBasicBlock *insertDivByZeroTrap(MachineInstr &MI,
3073bdd1243dSDimitry Andric                                               MachineBasicBlock *MBB) {
3074753f127fSDimitry Andric   if (!ZeroDivCheck)
3075bdd1243dSDimitry Andric     return MBB;
3076753f127fSDimitry Andric 
3077753f127fSDimitry Andric   // Build instructions:
3078bdd1243dSDimitry Andric   // MBB:
3079753f127fSDimitry Andric   //   div(or mod)   $dst, $dividend, $divisor
3080bdd1243dSDimitry Andric   //   bnez          $divisor, SinkMBB
3081bdd1243dSDimitry Andric   // BreakMBB:
3082bdd1243dSDimitry Andric   //   break         7 // BRK_DIVZERO
3083bdd1243dSDimitry Andric   // SinkMBB:
3084753f127fSDimitry Andric   //   fallthrough
3085bdd1243dSDimitry Andric   const BasicBlock *LLVM_BB = MBB->getBasicBlock();
3086bdd1243dSDimitry Andric   MachineFunction::iterator It = ++MBB->getIterator();
3087bdd1243dSDimitry Andric   MachineFunction *MF = MBB->getParent();
3088bdd1243dSDimitry Andric   auto BreakMBB = MF->CreateMachineBasicBlock(LLVM_BB);
3089bdd1243dSDimitry Andric   auto SinkMBB = MF->CreateMachineBasicBlock(LLVM_BB);
3090bdd1243dSDimitry Andric   MF->insert(It, BreakMBB);
3091bdd1243dSDimitry Andric   MF->insert(It, SinkMBB);
3092bdd1243dSDimitry Andric 
3093bdd1243dSDimitry Andric   // Transfer the remainder of MBB and its successor edges to SinkMBB.
3094bdd1243dSDimitry Andric   SinkMBB->splice(SinkMBB->end(), MBB, std::next(MI.getIterator()), MBB->end());
3095bdd1243dSDimitry Andric   SinkMBB->transferSuccessorsAndUpdatePHIs(MBB);
3096bdd1243dSDimitry Andric 
3097bdd1243dSDimitry Andric   const TargetInstrInfo &TII = *MF->getSubtarget().getInstrInfo();
3098bdd1243dSDimitry Andric   DebugLoc DL = MI.getDebugLoc();
3099753f127fSDimitry Andric   MachineOperand &Divisor = MI.getOperand(2);
3100bdd1243dSDimitry Andric   Register DivisorReg = Divisor.getReg();
3101753f127fSDimitry Andric 
3102bdd1243dSDimitry Andric   // MBB:
3103bdd1243dSDimitry Andric   BuildMI(MBB, DL, TII.get(LoongArch::BNEZ))
3104bdd1243dSDimitry Andric       .addReg(DivisorReg, getKillRegState(Divisor.isKill()))
3105bdd1243dSDimitry Andric       .addMBB(SinkMBB);
3106bdd1243dSDimitry Andric   MBB->addSuccessor(BreakMBB);
3107bdd1243dSDimitry Andric   MBB->addSuccessor(SinkMBB);
3108753f127fSDimitry Andric 
3109bdd1243dSDimitry Andric   // BreakMBB:
3110753f127fSDimitry Andric   // See linux header file arch/loongarch/include/uapi/asm/break.h for the
3111753f127fSDimitry Andric   // definition of BRK_DIVZERO.
3112bdd1243dSDimitry Andric   BuildMI(BreakMBB, DL, TII.get(LoongArch::BREAK)).addImm(7 /*BRK_DIVZERO*/);
3113bdd1243dSDimitry Andric   BreakMBB->addSuccessor(SinkMBB);
3114753f127fSDimitry Andric 
3115753f127fSDimitry Andric   // Clear Divisor's kill flag.
3116753f127fSDimitry Andric   Divisor.setIsKill(false);
3117753f127fSDimitry Andric 
3118bdd1243dSDimitry Andric   return SinkMBB;
3119753f127fSDimitry Andric }
3120753f127fSDimitry Andric 
31215f757f3fSDimitry Andric static MachineBasicBlock *
31225f757f3fSDimitry Andric emitVecCondBranchPseudo(MachineInstr &MI, MachineBasicBlock *BB,
31235f757f3fSDimitry Andric                         const LoongArchSubtarget &Subtarget) {
31245f757f3fSDimitry Andric   unsigned CondOpc;
31255f757f3fSDimitry Andric   switch (MI.getOpcode()) {
31265f757f3fSDimitry Andric   default:
31275f757f3fSDimitry Andric     llvm_unreachable("Unexpected opcode");
31285f757f3fSDimitry Andric   case LoongArch::PseudoVBZ:
31295f757f3fSDimitry Andric     CondOpc = LoongArch::VSETEQZ_V;
31305f757f3fSDimitry Andric     break;
31315f757f3fSDimitry Andric   case LoongArch::PseudoVBZ_B:
31325f757f3fSDimitry Andric     CondOpc = LoongArch::VSETANYEQZ_B;
31335f757f3fSDimitry Andric     break;
31345f757f3fSDimitry Andric   case LoongArch::PseudoVBZ_H:
31355f757f3fSDimitry Andric     CondOpc = LoongArch::VSETANYEQZ_H;
31365f757f3fSDimitry Andric     break;
31375f757f3fSDimitry Andric   case LoongArch::PseudoVBZ_W:
31385f757f3fSDimitry Andric     CondOpc = LoongArch::VSETANYEQZ_W;
31395f757f3fSDimitry Andric     break;
31405f757f3fSDimitry Andric   case LoongArch::PseudoVBZ_D:
31415f757f3fSDimitry Andric     CondOpc = LoongArch::VSETANYEQZ_D;
31425f757f3fSDimitry Andric     break;
31435f757f3fSDimitry Andric   case LoongArch::PseudoVBNZ:
31445f757f3fSDimitry Andric     CondOpc = LoongArch::VSETNEZ_V;
31455f757f3fSDimitry Andric     break;
31465f757f3fSDimitry Andric   case LoongArch::PseudoVBNZ_B:
31475f757f3fSDimitry Andric     CondOpc = LoongArch::VSETALLNEZ_B;
31485f757f3fSDimitry Andric     break;
31495f757f3fSDimitry Andric   case LoongArch::PseudoVBNZ_H:
31505f757f3fSDimitry Andric     CondOpc = LoongArch::VSETALLNEZ_H;
31515f757f3fSDimitry Andric     break;
31525f757f3fSDimitry Andric   case LoongArch::PseudoVBNZ_W:
31535f757f3fSDimitry Andric     CondOpc = LoongArch::VSETALLNEZ_W;
31545f757f3fSDimitry Andric     break;
31555f757f3fSDimitry Andric   case LoongArch::PseudoVBNZ_D:
31565f757f3fSDimitry Andric     CondOpc = LoongArch::VSETALLNEZ_D;
31575f757f3fSDimitry Andric     break;
31585f757f3fSDimitry Andric   case LoongArch::PseudoXVBZ:
31595f757f3fSDimitry Andric     CondOpc = LoongArch::XVSETEQZ_V;
31605f757f3fSDimitry Andric     break;
31615f757f3fSDimitry Andric   case LoongArch::PseudoXVBZ_B:
31625f757f3fSDimitry Andric     CondOpc = LoongArch::XVSETANYEQZ_B;
31635f757f3fSDimitry Andric     break;
31645f757f3fSDimitry Andric   case LoongArch::PseudoXVBZ_H:
31655f757f3fSDimitry Andric     CondOpc = LoongArch::XVSETANYEQZ_H;
31665f757f3fSDimitry Andric     break;
31675f757f3fSDimitry Andric   case LoongArch::PseudoXVBZ_W:
31685f757f3fSDimitry Andric     CondOpc = LoongArch::XVSETANYEQZ_W;
31695f757f3fSDimitry Andric     break;
31705f757f3fSDimitry Andric   case LoongArch::PseudoXVBZ_D:
31715f757f3fSDimitry Andric     CondOpc = LoongArch::XVSETANYEQZ_D;
31725f757f3fSDimitry Andric     break;
31735f757f3fSDimitry Andric   case LoongArch::PseudoXVBNZ:
31745f757f3fSDimitry Andric     CondOpc = LoongArch::XVSETNEZ_V;
31755f757f3fSDimitry Andric     break;
31765f757f3fSDimitry Andric   case LoongArch::PseudoXVBNZ_B:
31775f757f3fSDimitry Andric     CondOpc = LoongArch::XVSETALLNEZ_B;
31785f757f3fSDimitry Andric     break;
31795f757f3fSDimitry Andric   case LoongArch::PseudoXVBNZ_H:
31805f757f3fSDimitry Andric     CondOpc = LoongArch::XVSETALLNEZ_H;
31815f757f3fSDimitry Andric     break;
31825f757f3fSDimitry Andric   case LoongArch::PseudoXVBNZ_W:
31835f757f3fSDimitry Andric     CondOpc = LoongArch::XVSETALLNEZ_W;
31845f757f3fSDimitry Andric     break;
31855f757f3fSDimitry Andric   case LoongArch::PseudoXVBNZ_D:
31865f757f3fSDimitry Andric     CondOpc = LoongArch::XVSETALLNEZ_D;
31875f757f3fSDimitry Andric     break;
31885f757f3fSDimitry Andric   }
31895f757f3fSDimitry Andric 
31905f757f3fSDimitry Andric   const TargetInstrInfo *TII = Subtarget.getInstrInfo();
31915f757f3fSDimitry Andric   const BasicBlock *LLVM_BB = BB->getBasicBlock();
31925f757f3fSDimitry Andric   DebugLoc DL = MI.getDebugLoc();
31935f757f3fSDimitry Andric   MachineRegisterInfo &MRI = BB->getParent()->getRegInfo();
31945f757f3fSDimitry Andric   MachineFunction::iterator It = ++BB->getIterator();
31955f757f3fSDimitry Andric 
31965f757f3fSDimitry Andric   MachineFunction *F = BB->getParent();
31975f757f3fSDimitry Andric   MachineBasicBlock *FalseBB = F->CreateMachineBasicBlock(LLVM_BB);
31985f757f3fSDimitry Andric   MachineBasicBlock *TrueBB = F->CreateMachineBasicBlock(LLVM_BB);
31995f757f3fSDimitry Andric   MachineBasicBlock *SinkBB = F->CreateMachineBasicBlock(LLVM_BB);
32005f757f3fSDimitry Andric 
32015f757f3fSDimitry Andric   F->insert(It, FalseBB);
32025f757f3fSDimitry Andric   F->insert(It, TrueBB);
32035f757f3fSDimitry Andric   F->insert(It, SinkBB);
32045f757f3fSDimitry Andric 
32055f757f3fSDimitry Andric   // Transfer the remainder of MBB and its successor edges to Sink.
32065f757f3fSDimitry Andric   SinkBB->splice(SinkBB->end(), BB, std::next(MI.getIterator()), BB->end());
32075f757f3fSDimitry Andric   SinkBB->transferSuccessorsAndUpdatePHIs(BB);
32085f757f3fSDimitry Andric 
32095f757f3fSDimitry Andric   // Insert the real instruction to BB.
32105f757f3fSDimitry Andric   Register FCC = MRI.createVirtualRegister(&LoongArch::CFRRegClass);
32115f757f3fSDimitry Andric   BuildMI(BB, DL, TII->get(CondOpc), FCC).addReg(MI.getOperand(1).getReg());
32125f757f3fSDimitry Andric 
32135f757f3fSDimitry Andric   // Insert branch.
32145f757f3fSDimitry Andric   BuildMI(BB, DL, TII->get(LoongArch::BCNEZ)).addReg(FCC).addMBB(TrueBB);
32155f757f3fSDimitry Andric   BB->addSuccessor(FalseBB);
32165f757f3fSDimitry Andric   BB->addSuccessor(TrueBB);
32175f757f3fSDimitry Andric 
32185f757f3fSDimitry Andric   // FalseBB.
32195f757f3fSDimitry Andric   Register RD1 = MRI.createVirtualRegister(&LoongArch::GPRRegClass);
32205f757f3fSDimitry Andric   BuildMI(FalseBB, DL, TII->get(LoongArch::ADDI_W), RD1)
32215f757f3fSDimitry Andric       .addReg(LoongArch::R0)
32225f757f3fSDimitry Andric       .addImm(0);
32235f757f3fSDimitry Andric   BuildMI(FalseBB, DL, TII->get(LoongArch::PseudoBR)).addMBB(SinkBB);
32245f757f3fSDimitry Andric   FalseBB->addSuccessor(SinkBB);
32255f757f3fSDimitry Andric 
32265f757f3fSDimitry Andric   // TrueBB.
32275f757f3fSDimitry Andric   Register RD2 = MRI.createVirtualRegister(&LoongArch::GPRRegClass);
32285f757f3fSDimitry Andric   BuildMI(TrueBB, DL, TII->get(LoongArch::ADDI_W), RD2)
32295f757f3fSDimitry Andric       .addReg(LoongArch::R0)
32305f757f3fSDimitry Andric       .addImm(1);
32315f757f3fSDimitry Andric   TrueBB->addSuccessor(SinkBB);
32325f757f3fSDimitry Andric 
32335f757f3fSDimitry Andric   // SinkBB: merge the results.
32345f757f3fSDimitry Andric   BuildMI(*SinkBB, SinkBB->begin(), DL, TII->get(LoongArch::PHI),
32355f757f3fSDimitry Andric           MI.getOperand(0).getReg())
32365f757f3fSDimitry Andric       .addReg(RD1)
32375f757f3fSDimitry Andric       .addMBB(FalseBB)
32385f757f3fSDimitry Andric       .addReg(RD2)
32395f757f3fSDimitry Andric       .addMBB(TrueBB);
32405f757f3fSDimitry Andric 
32415f757f3fSDimitry Andric   // The pseudo instruction is gone now.
32425f757f3fSDimitry Andric   MI.eraseFromParent();
32435f757f3fSDimitry Andric   return SinkBB;
32445f757f3fSDimitry Andric }
32455f757f3fSDimitry Andric 
32465f757f3fSDimitry Andric static MachineBasicBlock *
32475f757f3fSDimitry Andric emitPseudoXVINSGR2VR(MachineInstr &MI, MachineBasicBlock *BB,
32485f757f3fSDimitry Andric                      const LoongArchSubtarget &Subtarget) {
32495f757f3fSDimitry Andric   unsigned InsOp;
32505f757f3fSDimitry Andric   unsigned HalfSize;
32515f757f3fSDimitry Andric   switch (MI.getOpcode()) {
32525f757f3fSDimitry Andric   default:
32535f757f3fSDimitry Andric     llvm_unreachable("Unexpected opcode");
32545f757f3fSDimitry Andric   case LoongArch::PseudoXVINSGR2VR_B:
32555f757f3fSDimitry Andric     HalfSize = 16;
32565f757f3fSDimitry Andric     InsOp = LoongArch::VINSGR2VR_B;
32575f757f3fSDimitry Andric     break;
32585f757f3fSDimitry Andric   case LoongArch::PseudoXVINSGR2VR_H:
32595f757f3fSDimitry Andric     HalfSize = 8;
32605f757f3fSDimitry Andric     InsOp = LoongArch::VINSGR2VR_H;
32615f757f3fSDimitry Andric     break;
32625f757f3fSDimitry Andric   }
32635f757f3fSDimitry Andric   const TargetInstrInfo *TII = Subtarget.getInstrInfo();
32645f757f3fSDimitry Andric   const TargetRegisterClass *RC = &LoongArch::LASX256RegClass;
32655f757f3fSDimitry Andric   const TargetRegisterClass *SubRC = &LoongArch::LSX128RegClass;
32665f757f3fSDimitry Andric   DebugLoc DL = MI.getDebugLoc();
32675f757f3fSDimitry Andric   MachineRegisterInfo &MRI = BB->getParent()->getRegInfo();
32685f757f3fSDimitry Andric   // XDst = vector_insert XSrc, Elt, Idx
32695f757f3fSDimitry Andric   Register XDst = MI.getOperand(0).getReg();
32705f757f3fSDimitry Andric   Register XSrc = MI.getOperand(1).getReg();
32715f757f3fSDimitry Andric   Register Elt = MI.getOperand(2).getReg();
32725f757f3fSDimitry Andric   unsigned Idx = MI.getOperand(3).getImm();
32735f757f3fSDimitry Andric 
32745f757f3fSDimitry Andric   Register ScratchReg1 = XSrc;
32755f757f3fSDimitry Andric   if (Idx >= HalfSize) {
32765f757f3fSDimitry Andric     ScratchReg1 = MRI.createVirtualRegister(RC);
32775f757f3fSDimitry Andric     BuildMI(*BB, MI, DL, TII->get(LoongArch::XVPERMI_Q), ScratchReg1)
32785f757f3fSDimitry Andric         .addReg(XSrc)
32795f757f3fSDimitry Andric         .addReg(XSrc)
32805f757f3fSDimitry Andric         .addImm(1);
32815f757f3fSDimitry Andric   }
32825f757f3fSDimitry Andric 
32835f757f3fSDimitry Andric   Register ScratchSubReg1 = MRI.createVirtualRegister(SubRC);
32845f757f3fSDimitry Andric   Register ScratchSubReg2 = MRI.createVirtualRegister(SubRC);
32855f757f3fSDimitry Andric   BuildMI(*BB, MI, DL, TII->get(LoongArch::COPY), ScratchSubReg1)
32865f757f3fSDimitry Andric       .addReg(ScratchReg1, 0, LoongArch::sub_128);
32875f757f3fSDimitry Andric   BuildMI(*BB, MI, DL, TII->get(InsOp), ScratchSubReg2)
32885f757f3fSDimitry Andric       .addReg(ScratchSubReg1)
32895f757f3fSDimitry Andric       .addReg(Elt)
32905f757f3fSDimitry Andric       .addImm(Idx >= HalfSize ? Idx - HalfSize : Idx);
32915f757f3fSDimitry Andric 
32925f757f3fSDimitry Andric   Register ScratchReg2 = XDst;
32935f757f3fSDimitry Andric   if (Idx >= HalfSize)
32945f757f3fSDimitry Andric     ScratchReg2 = MRI.createVirtualRegister(RC);
32955f757f3fSDimitry Andric 
32965f757f3fSDimitry Andric   BuildMI(*BB, MI, DL, TII->get(LoongArch::SUBREG_TO_REG), ScratchReg2)
32975f757f3fSDimitry Andric       .addImm(0)
32985f757f3fSDimitry Andric       .addReg(ScratchSubReg2)
32995f757f3fSDimitry Andric       .addImm(LoongArch::sub_128);
33005f757f3fSDimitry Andric 
33015f757f3fSDimitry Andric   if (Idx >= HalfSize)
33025f757f3fSDimitry Andric     BuildMI(*BB, MI, DL, TII->get(LoongArch::XVPERMI_Q), XDst)
33035f757f3fSDimitry Andric         .addReg(XSrc)
33045f757f3fSDimitry Andric         .addReg(ScratchReg2)
33055f757f3fSDimitry Andric         .addImm(2);
33065f757f3fSDimitry Andric 
33075f757f3fSDimitry Andric   MI.eraseFromParent();
33085f757f3fSDimitry Andric   return BB;
33095f757f3fSDimitry Andric }
33105f757f3fSDimitry Andric 
3311753f127fSDimitry Andric MachineBasicBlock *LoongArchTargetLowering::EmitInstrWithCustomInserter(
3312753f127fSDimitry Andric     MachineInstr &MI, MachineBasicBlock *BB) const {
3313bdd1243dSDimitry Andric   const TargetInstrInfo *TII = Subtarget.getInstrInfo();
3314bdd1243dSDimitry Andric   DebugLoc DL = MI.getDebugLoc();
3315753f127fSDimitry Andric 
3316753f127fSDimitry Andric   switch (MI.getOpcode()) {
3317753f127fSDimitry Andric   default:
3318753f127fSDimitry Andric     llvm_unreachable("Unexpected instr type to insert");
3319753f127fSDimitry Andric   case LoongArch::DIV_W:
3320753f127fSDimitry Andric   case LoongArch::DIV_WU:
3321753f127fSDimitry Andric   case LoongArch::MOD_W:
3322753f127fSDimitry Andric   case LoongArch::MOD_WU:
3323753f127fSDimitry Andric   case LoongArch::DIV_D:
3324753f127fSDimitry Andric   case LoongArch::DIV_DU:
3325753f127fSDimitry Andric   case LoongArch::MOD_D:
3326753f127fSDimitry Andric   case LoongArch::MOD_DU:
3327bdd1243dSDimitry Andric     return insertDivByZeroTrap(MI, BB);
3328753f127fSDimitry Andric     break;
3329bdd1243dSDimitry Andric   case LoongArch::WRFCSR: {
3330bdd1243dSDimitry Andric     BuildMI(*BB, MI, DL, TII->get(LoongArch::MOVGR2FCSR),
3331bdd1243dSDimitry Andric             LoongArch::FCSR0 + MI.getOperand(0).getImm())
3332bdd1243dSDimitry Andric         .addReg(MI.getOperand(1).getReg());
3333bdd1243dSDimitry Andric     MI.eraseFromParent();
3334bdd1243dSDimitry Andric     return BB;
3335bdd1243dSDimitry Andric   }
3336bdd1243dSDimitry Andric   case LoongArch::RDFCSR: {
3337bdd1243dSDimitry Andric     MachineInstr *ReadFCSR =
3338bdd1243dSDimitry Andric         BuildMI(*BB, MI, DL, TII->get(LoongArch::MOVFCSR2GR),
3339bdd1243dSDimitry Andric                 MI.getOperand(0).getReg())
3340bdd1243dSDimitry Andric             .addReg(LoongArch::FCSR0 + MI.getOperand(1).getImm());
3341bdd1243dSDimitry Andric     ReadFCSR->getOperand(1).setIsUndef();
3342bdd1243dSDimitry Andric     MI.eraseFromParent();
3343bdd1243dSDimitry Andric     return BB;
3344bdd1243dSDimitry Andric   }
33455f757f3fSDimitry Andric   case LoongArch::PseudoVBZ:
33465f757f3fSDimitry Andric   case LoongArch::PseudoVBZ_B:
33475f757f3fSDimitry Andric   case LoongArch::PseudoVBZ_H:
33485f757f3fSDimitry Andric   case LoongArch::PseudoVBZ_W:
33495f757f3fSDimitry Andric   case LoongArch::PseudoVBZ_D:
33505f757f3fSDimitry Andric   case LoongArch::PseudoVBNZ:
33515f757f3fSDimitry Andric   case LoongArch::PseudoVBNZ_B:
33525f757f3fSDimitry Andric   case LoongArch::PseudoVBNZ_H:
33535f757f3fSDimitry Andric   case LoongArch::PseudoVBNZ_W:
33545f757f3fSDimitry Andric   case LoongArch::PseudoVBNZ_D:
33555f757f3fSDimitry Andric   case LoongArch::PseudoXVBZ:
33565f757f3fSDimitry Andric   case LoongArch::PseudoXVBZ_B:
33575f757f3fSDimitry Andric   case LoongArch::PseudoXVBZ_H:
33585f757f3fSDimitry Andric   case LoongArch::PseudoXVBZ_W:
33595f757f3fSDimitry Andric   case LoongArch::PseudoXVBZ_D:
33605f757f3fSDimitry Andric   case LoongArch::PseudoXVBNZ:
33615f757f3fSDimitry Andric   case LoongArch::PseudoXVBNZ_B:
33625f757f3fSDimitry Andric   case LoongArch::PseudoXVBNZ_H:
33635f757f3fSDimitry Andric   case LoongArch::PseudoXVBNZ_W:
33645f757f3fSDimitry Andric   case LoongArch::PseudoXVBNZ_D:
33655f757f3fSDimitry Andric     return emitVecCondBranchPseudo(MI, BB, Subtarget);
33665f757f3fSDimitry Andric   case LoongArch::PseudoXVINSGR2VR_B:
33675f757f3fSDimitry Andric   case LoongArch::PseudoXVINSGR2VR_H:
33685f757f3fSDimitry Andric     return emitPseudoXVINSGR2VR(MI, BB, Subtarget);
3369753f127fSDimitry Andric   }
3370753f127fSDimitry Andric }
3371753f127fSDimitry Andric 
337206c3fb27SDimitry Andric bool LoongArchTargetLowering::allowsMisalignedMemoryAccesses(
337306c3fb27SDimitry Andric     EVT VT, unsigned AddrSpace, Align Alignment, MachineMemOperand::Flags Flags,
337406c3fb27SDimitry Andric     unsigned *Fast) const {
337506c3fb27SDimitry Andric   if (!Subtarget.hasUAL())
337606c3fb27SDimitry Andric     return false;
337706c3fb27SDimitry Andric 
337806c3fb27SDimitry Andric   // TODO: set reasonable speed number.
337906c3fb27SDimitry Andric   if (Fast)
338006c3fb27SDimitry Andric     *Fast = 1;
338106c3fb27SDimitry Andric   return true;
338206c3fb27SDimitry Andric }
338306c3fb27SDimitry Andric 
338481ad6265SDimitry Andric const char *LoongArchTargetLowering::getTargetNodeName(unsigned Opcode) const {
338581ad6265SDimitry Andric   switch ((LoongArchISD::NodeType)Opcode) {
338681ad6265SDimitry Andric   case LoongArchISD::FIRST_NUMBER:
338781ad6265SDimitry Andric     break;
338881ad6265SDimitry Andric 
338981ad6265SDimitry Andric #define NODE_NAME_CASE(node)                                                   \
339081ad6265SDimitry Andric   case LoongArchISD::node:                                                     \
339181ad6265SDimitry Andric     return "LoongArchISD::" #node;
339281ad6265SDimitry Andric 
339381ad6265SDimitry Andric     // TODO: Add more target-dependent nodes later.
3394753f127fSDimitry Andric     NODE_NAME_CASE(CALL)
3395*1db9f3b2SDimitry Andric     NODE_NAME_CASE(CALL_MEDIUM)
3396*1db9f3b2SDimitry Andric     NODE_NAME_CASE(CALL_LARGE)
339781ad6265SDimitry Andric     NODE_NAME_CASE(RET)
3398bdd1243dSDimitry Andric     NODE_NAME_CASE(TAIL)
3399*1db9f3b2SDimitry Andric     NODE_NAME_CASE(TAIL_MEDIUM)
3400*1db9f3b2SDimitry Andric     NODE_NAME_CASE(TAIL_LARGE)
340181ad6265SDimitry Andric     NODE_NAME_CASE(SLL_W)
340281ad6265SDimitry Andric     NODE_NAME_CASE(SRA_W)
340381ad6265SDimitry Andric     NODE_NAME_CASE(SRL_W)
3404753f127fSDimitry Andric     NODE_NAME_CASE(BSTRINS)
340581ad6265SDimitry Andric     NODE_NAME_CASE(BSTRPICK)
3406753f127fSDimitry Andric     NODE_NAME_CASE(MOVGR2FR_W_LA64)
3407753f127fSDimitry Andric     NODE_NAME_CASE(MOVFR2GR_S_LA64)
3408753f127fSDimitry Andric     NODE_NAME_CASE(FTINT)
3409bdd1243dSDimitry Andric     NODE_NAME_CASE(REVB_2H)
3410bdd1243dSDimitry Andric     NODE_NAME_CASE(REVB_2W)
3411bdd1243dSDimitry Andric     NODE_NAME_CASE(BITREV_4B)
3412bdd1243dSDimitry Andric     NODE_NAME_CASE(BITREV_W)
3413bdd1243dSDimitry Andric     NODE_NAME_CASE(ROTR_W)
3414bdd1243dSDimitry Andric     NODE_NAME_CASE(ROTL_W)
3415bdd1243dSDimitry Andric     NODE_NAME_CASE(CLZ_W)
3416bdd1243dSDimitry Andric     NODE_NAME_CASE(CTZ_W)
3417bdd1243dSDimitry Andric     NODE_NAME_CASE(DBAR)
3418bdd1243dSDimitry Andric     NODE_NAME_CASE(IBAR)
3419bdd1243dSDimitry Andric     NODE_NAME_CASE(BREAK)
3420bdd1243dSDimitry Andric     NODE_NAME_CASE(SYSCALL)
3421bdd1243dSDimitry Andric     NODE_NAME_CASE(CRC_W_B_W)
3422bdd1243dSDimitry Andric     NODE_NAME_CASE(CRC_W_H_W)
3423bdd1243dSDimitry Andric     NODE_NAME_CASE(CRC_W_W_W)
3424bdd1243dSDimitry Andric     NODE_NAME_CASE(CRC_W_D_W)
3425bdd1243dSDimitry Andric     NODE_NAME_CASE(CRCC_W_B_W)
3426bdd1243dSDimitry Andric     NODE_NAME_CASE(CRCC_W_H_W)
3427bdd1243dSDimitry Andric     NODE_NAME_CASE(CRCC_W_W_W)
3428bdd1243dSDimitry Andric     NODE_NAME_CASE(CRCC_W_D_W)
3429bdd1243dSDimitry Andric     NODE_NAME_CASE(CSRRD)
3430bdd1243dSDimitry Andric     NODE_NAME_CASE(CSRWR)
3431bdd1243dSDimitry Andric     NODE_NAME_CASE(CSRXCHG)
3432bdd1243dSDimitry Andric     NODE_NAME_CASE(IOCSRRD_B)
3433bdd1243dSDimitry Andric     NODE_NAME_CASE(IOCSRRD_H)
3434bdd1243dSDimitry Andric     NODE_NAME_CASE(IOCSRRD_W)
3435bdd1243dSDimitry Andric     NODE_NAME_CASE(IOCSRRD_D)
3436bdd1243dSDimitry Andric     NODE_NAME_CASE(IOCSRWR_B)
3437bdd1243dSDimitry Andric     NODE_NAME_CASE(IOCSRWR_H)
3438bdd1243dSDimitry Andric     NODE_NAME_CASE(IOCSRWR_W)
3439bdd1243dSDimitry Andric     NODE_NAME_CASE(IOCSRWR_D)
3440bdd1243dSDimitry Andric     NODE_NAME_CASE(CPUCFG)
3441bdd1243dSDimitry Andric     NODE_NAME_CASE(MOVGR2FCSR)
3442bdd1243dSDimitry Andric     NODE_NAME_CASE(MOVFCSR2GR)
3443bdd1243dSDimitry Andric     NODE_NAME_CASE(CACOP_D)
3444bdd1243dSDimitry Andric     NODE_NAME_CASE(CACOP_W)
34455f757f3fSDimitry Andric     NODE_NAME_CASE(VPICK_SEXT_ELT)
34465f757f3fSDimitry Andric     NODE_NAME_CASE(VPICK_ZEXT_ELT)
34475f757f3fSDimitry Andric     NODE_NAME_CASE(VREPLVE)
34485f757f3fSDimitry Andric     NODE_NAME_CASE(VALL_ZERO)
34495f757f3fSDimitry Andric     NODE_NAME_CASE(VANY_ZERO)
34505f757f3fSDimitry Andric     NODE_NAME_CASE(VALL_NONZERO)
34515f757f3fSDimitry Andric     NODE_NAME_CASE(VANY_NONZERO)
345281ad6265SDimitry Andric   }
345381ad6265SDimitry Andric #undef NODE_NAME_CASE
345481ad6265SDimitry Andric   return nullptr;
345581ad6265SDimitry Andric }
345681ad6265SDimitry Andric 
345781ad6265SDimitry Andric //===----------------------------------------------------------------------===//
345881ad6265SDimitry Andric //                     Calling Convention Implementation
345981ad6265SDimitry Andric //===----------------------------------------------------------------------===//
3460bdd1243dSDimitry Andric 
3461bdd1243dSDimitry Andric // Eight general-purpose registers a0-a7 used for passing integer arguments,
3462bdd1243dSDimitry Andric // with a0-a1 reused to return values. Generally, the GPRs are used to pass
3463bdd1243dSDimitry Andric // fixed-point arguments, and floating-point arguments when no FPR is available
3464bdd1243dSDimitry Andric // or with soft float ABI.
346581ad6265SDimitry Andric const MCPhysReg ArgGPRs[] = {LoongArch::R4,  LoongArch::R5, LoongArch::R6,
346681ad6265SDimitry Andric                              LoongArch::R7,  LoongArch::R8, LoongArch::R9,
346781ad6265SDimitry Andric                              LoongArch::R10, LoongArch::R11};
3468bdd1243dSDimitry Andric // Eight floating-point registers fa0-fa7 used for passing floating-point
3469bdd1243dSDimitry Andric // arguments, and fa0-fa1 are also used to return values.
347081ad6265SDimitry Andric const MCPhysReg ArgFPR32s[] = {LoongArch::F0, LoongArch::F1, LoongArch::F2,
347181ad6265SDimitry Andric                                LoongArch::F3, LoongArch::F4, LoongArch::F5,
347281ad6265SDimitry Andric                                LoongArch::F6, LoongArch::F7};
3473bdd1243dSDimitry Andric // FPR32 and FPR64 alias each other.
347481ad6265SDimitry Andric const MCPhysReg ArgFPR64s[] = {
347581ad6265SDimitry Andric     LoongArch::F0_64, LoongArch::F1_64, LoongArch::F2_64, LoongArch::F3_64,
347681ad6265SDimitry Andric     LoongArch::F4_64, LoongArch::F5_64, LoongArch::F6_64, LoongArch::F7_64};
347781ad6265SDimitry Andric 
34785f757f3fSDimitry Andric const MCPhysReg ArgVRs[] = {LoongArch::VR0, LoongArch::VR1, LoongArch::VR2,
34795f757f3fSDimitry Andric                             LoongArch::VR3, LoongArch::VR4, LoongArch::VR5,
34805f757f3fSDimitry Andric                             LoongArch::VR6, LoongArch::VR7};
34815f757f3fSDimitry Andric 
34825f757f3fSDimitry Andric const MCPhysReg ArgXRs[] = {LoongArch::XR0, LoongArch::XR1, LoongArch::XR2,
34835f757f3fSDimitry Andric                             LoongArch::XR3, LoongArch::XR4, LoongArch::XR5,
34845f757f3fSDimitry Andric                             LoongArch::XR6, LoongArch::XR7};
34855f757f3fSDimitry Andric 
3486bdd1243dSDimitry Andric // Pass a 2*GRLen argument that has been split into two GRLen values through
3487bdd1243dSDimitry Andric // registers or the stack as necessary.
3488bdd1243dSDimitry Andric static bool CC_LoongArchAssign2GRLen(unsigned GRLen, CCState &State,
3489bdd1243dSDimitry Andric                                      CCValAssign VA1, ISD::ArgFlagsTy ArgFlags1,
3490bdd1243dSDimitry Andric                                      unsigned ValNo2, MVT ValVT2, MVT LocVT2,
3491bdd1243dSDimitry Andric                                      ISD::ArgFlagsTy ArgFlags2) {
3492bdd1243dSDimitry Andric   unsigned GRLenInBytes = GRLen / 8;
3493bdd1243dSDimitry Andric   if (Register Reg = State.AllocateReg(ArgGPRs)) {
3494bdd1243dSDimitry Andric     // At least one half can be passed via register.
3495bdd1243dSDimitry Andric     State.addLoc(CCValAssign::getReg(VA1.getValNo(), VA1.getValVT(), Reg,
3496bdd1243dSDimitry Andric                                      VA1.getLocVT(), CCValAssign::Full));
3497bdd1243dSDimitry Andric   } else {
3498bdd1243dSDimitry Andric     // Both halves must be passed on the stack, with proper alignment.
3499bdd1243dSDimitry Andric     Align StackAlign =
3500bdd1243dSDimitry Andric         std::max(Align(GRLenInBytes), ArgFlags1.getNonZeroOrigAlign());
3501bdd1243dSDimitry Andric     State.addLoc(
3502bdd1243dSDimitry Andric         CCValAssign::getMem(VA1.getValNo(), VA1.getValVT(),
3503bdd1243dSDimitry Andric                             State.AllocateStack(GRLenInBytes, StackAlign),
3504bdd1243dSDimitry Andric                             VA1.getLocVT(), CCValAssign::Full));
3505bdd1243dSDimitry Andric     State.addLoc(CCValAssign::getMem(
3506bdd1243dSDimitry Andric         ValNo2, ValVT2, State.AllocateStack(GRLenInBytes, Align(GRLenInBytes)),
3507bdd1243dSDimitry Andric         LocVT2, CCValAssign::Full));
3508bdd1243dSDimitry Andric     return false;
3509bdd1243dSDimitry Andric   }
3510bdd1243dSDimitry Andric   if (Register Reg = State.AllocateReg(ArgGPRs)) {
3511bdd1243dSDimitry Andric     // The second half can also be passed via register.
3512bdd1243dSDimitry Andric     State.addLoc(
3513bdd1243dSDimitry Andric         CCValAssign::getReg(ValNo2, ValVT2, Reg, LocVT2, CCValAssign::Full));
3514bdd1243dSDimitry Andric   } else {
3515bdd1243dSDimitry Andric     // The second half is passed via the stack, without additional alignment.
3516bdd1243dSDimitry Andric     State.addLoc(CCValAssign::getMem(
3517bdd1243dSDimitry Andric         ValNo2, ValVT2, State.AllocateStack(GRLenInBytes, Align(GRLenInBytes)),
3518bdd1243dSDimitry Andric         LocVT2, CCValAssign::Full));
3519bdd1243dSDimitry Andric   }
352081ad6265SDimitry Andric   return false;
352181ad6265SDimitry Andric }
352281ad6265SDimitry Andric 
3523bdd1243dSDimitry Andric // Implements the LoongArch calling convention. Returns true upon failure.
3524bdd1243dSDimitry Andric static bool CC_LoongArch(const DataLayout &DL, LoongArchABI::ABI ABI,
3525bdd1243dSDimitry Andric                          unsigned ValNo, MVT ValVT,
3526bdd1243dSDimitry Andric                          CCValAssign::LocInfo LocInfo, ISD::ArgFlagsTy ArgFlags,
3527bdd1243dSDimitry Andric                          CCState &State, bool IsFixed, bool IsRet,
3528bdd1243dSDimitry Andric                          Type *OrigTy) {
3529bdd1243dSDimitry Andric   unsigned GRLen = DL.getLargestLegalIntTypeSizeInBits();
3530bdd1243dSDimitry Andric   assert((GRLen == 32 || GRLen == 64) && "Unspport GRLen");
3531bdd1243dSDimitry Andric   MVT GRLenVT = GRLen == 32 ? MVT::i32 : MVT::i64;
3532bdd1243dSDimitry Andric   MVT LocVT = ValVT;
3533bdd1243dSDimitry Andric 
3534bdd1243dSDimitry Andric   // Any return value split into more than two values can't be returned
3535bdd1243dSDimitry Andric   // directly.
3536bdd1243dSDimitry Andric   if (IsRet && ValNo > 1)
353781ad6265SDimitry Andric     return true;
3538bdd1243dSDimitry Andric 
3539bdd1243dSDimitry Andric   // If passing a variadic argument, or if no FPR is available.
3540bdd1243dSDimitry Andric   bool UseGPRForFloat = true;
3541bdd1243dSDimitry Andric 
3542bdd1243dSDimitry Andric   switch (ABI) {
3543bdd1243dSDimitry Andric   default:
3544bdd1243dSDimitry Andric     llvm_unreachable("Unexpected ABI");
3545bdd1243dSDimitry Andric   case LoongArchABI::ABI_ILP32S:
3546bdd1243dSDimitry Andric   case LoongArchABI::ABI_ILP32F:
3547bdd1243dSDimitry Andric   case LoongArchABI::ABI_LP64F:
3548bdd1243dSDimitry Andric     report_fatal_error("Unimplemented ABI");
3549bdd1243dSDimitry Andric     break;
3550bdd1243dSDimitry Andric   case LoongArchABI::ABI_ILP32D:
3551bdd1243dSDimitry Andric   case LoongArchABI::ABI_LP64D:
3552bdd1243dSDimitry Andric     UseGPRForFloat = !IsFixed;
3553bdd1243dSDimitry Andric     break;
355406c3fb27SDimitry Andric   case LoongArchABI::ABI_LP64S:
355506c3fb27SDimitry Andric     break;
3556bdd1243dSDimitry Andric   }
3557bdd1243dSDimitry Andric 
3558bdd1243dSDimitry Andric   // FPR32 and FPR64 alias each other.
3559bdd1243dSDimitry Andric   if (State.getFirstUnallocated(ArgFPR32s) == std::size(ArgFPR32s))
3560bdd1243dSDimitry Andric     UseGPRForFloat = true;
3561bdd1243dSDimitry Andric 
3562bdd1243dSDimitry Andric   if (UseGPRForFloat && ValVT == MVT::f32) {
3563bdd1243dSDimitry Andric     LocVT = GRLenVT;
3564bdd1243dSDimitry Andric     LocInfo = CCValAssign::BCvt;
3565bdd1243dSDimitry Andric   } else if (UseGPRForFloat && GRLen == 64 && ValVT == MVT::f64) {
3566bdd1243dSDimitry Andric     LocVT = MVT::i64;
3567bdd1243dSDimitry Andric     LocInfo = CCValAssign::BCvt;
3568bdd1243dSDimitry Andric   } else if (UseGPRForFloat && GRLen == 32 && ValVT == MVT::f64) {
3569bdd1243dSDimitry Andric     // TODO: Handle passing f64 on LA32 with D feature.
3570bdd1243dSDimitry Andric     report_fatal_error("Passing f64 with GPR on LA32 is undefined");
3571bdd1243dSDimitry Andric   }
3572bdd1243dSDimitry Andric 
3573bdd1243dSDimitry Andric   // If this is a variadic argument, the LoongArch calling convention requires
3574bdd1243dSDimitry Andric   // that it is assigned an 'even' or 'aligned' register if it has (2*GRLen)/8
3575bdd1243dSDimitry Andric   // byte alignment. An aligned register should be used regardless of whether
3576bdd1243dSDimitry Andric   // the original argument was split during legalisation or not. The argument
3577bdd1243dSDimitry Andric   // will not be passed by registers if the original type is larger than
3578bdd1243dSDimitry Andric   // 2*GRLen, so the register alignment rule does not apply.
3579bdd1243dSDimitry Andric   unsigned TwoGRLenInBytes = (2 * GRLen) / 8;
3580bdd1243dSDimitry Andric   if (!IsFixed && ArgFlags.getNonZeroOrigAlign() == TwoGRLenInBytes &&
3581bdd1243dSDimitry Andric       DL.getTypeAllocSize(OrigTy) == TwoGRLenInBytes) {
3582bdd1243dSDimitry Andric     unsigned RegIdx = State.getFirstUnallocated(ArgGPRs);
3583bdd1243dSDimitry Andric     // Skip 'odd' register if necessary.
3584bdd1243dSDimitry Andric     if (RegIdx != std::size(ArgGPRs) && RegIdx % 2 == 1)
3585bdd1243dSDimitry Andric       State.AllocateReg(ArgGPRs);
3586bdd1243dSDimitry Andric   }
3587bdd1243dSDimitry Andric 
3588bdd1243dSDimitry Andric   SmallVectorImpl<CCValAssign> &PendingLocs = State.getPendingLocs();
3589bdd1243dSDimitry Andric   SmallVectorImpl<ISD::ArgFlagsTy> &PendingArgFlags =
3590bdd1243dSDimitry Andric       State.getPendingArgFlags();
3591bdd1243dSDimitry Andric 
3592bdd1243dSDimitry Andric   assert(PendingLocs.size() == PendingArgFlags.size() &&
3593bdd1243dSDimitry Andric          "PendingLocs and PendingArgFlags out of sync");
3594bdd1243dSDimitry Andric 
3595bdd1243dSDimitry Andric   // Split arguments might be passed indirectly, so keep track of the pending
3596bdd1243dSDimitry Andric   // values.
3597bdd1243dSDimitry Andric   if (ValVT.isScalarInteger() && (ArgFlags.isSplit() || !PendingLocs.empty())) {
3598bdd1243dSDimitry Andric     LocVT = GRLenVT;
3599bdd1243dSDimitry Andric     LocInfo = CCValAssign::Indirect;
3600bdd1243dSDimitry Andric     PendingLocs.push_back(
3601bdd1243dSDimitry Andric         CCValAssign::getPending(ValNo, ValVT, LocVT, LocInfo));
3602bdd1243dSDimitry Andric     PendingArgFlags.push_back(ArgFlags);
3603bdd1243dSDimitry Andric     if (!ArgFlags.isSplitEnd()) {
3604bdd1243dSDimitry Andric       return false;
3605bdd1243dSDimitry Andric     }
3606bdd1243dSDimitry Andric   }
3607bdd1243dSDimitry Andric 
3608bdd1243dSDimitry Andric   // If the split argument only had two elements, it should be passed directly
3609bdd1243dSDimitry Andric   // in registers or on the stack.
3610bdd1243dSDimitry Andric   if (ValVT.isScalarInteger() && ArgFlags.isSplitEnd() &&
3611bdd1243dSDimitry Andric       PendingLocs.size() <= 2) {
3612bdd1243dSDimitry Andric     assert(PendingLocs.size() == 2 && "Unexpected PendingLocs.size()");
3613bdd1243dSDimitry Andric     // Apply the normal calling convention rules to the first half of the
3614bdd1243dSDimitry Andric     // split argument.
3615bdd1243dSDimitry Andric     CCValAssign VA = PendingLocs[0];
3616bdd1243dSDimitry Andric     ISD::ArgFlagsTy AF = PendingArgFlags[0];
3617bdd1243dSDimitry Andric     PendingLocs.clear();
3618bdd1243dSDimitry Andric     PendingArgFlags.clear();
3619bdd1243dSDimitry Andric     return CC_LoongArchAssign2GRLen(GRLen, State, VA, AF, ValNo, ValVT, LocVT,
3620bdd1243dSDimitry Andric                                     ArgFlags);
3621bdd1243dSDimitry Andric   }
3622bdd1243dSDimitry Andric 
3623bdd1243dSDimitry Andric   // Allocate to a register if possible, or else a stack slot.
3624bdd1243dSDimitry Andric   Register Reg;
3625bdd1243dSDimitry Andric   unsigned StoreSizeBytes = GRLen / 8;
3626bdd1243dSDimitry Andric   Align StackAlign = Align(GRLen / 8);
3627bdd1243dSDimitry Andric 
3628bdd1243dSDimitry Andric   if (ValVT == MVT::f32 && !UseGPRForFloat)
3629bdd1243dSDimitry Andric     Reg = State.AllocateReg(ArgFPR32s);
3630bdd1243dSDimitry Andric   else if (ValVT == MVT::f64 && !UseGPRForFloat)
3631bdd1243dSDimitry Andric     Reg = State.AllocateReg(ArgFPR64s);
36325f757f3fSDimitry Andric   else if (ValVT.is128BitVector())
36335f757f3fSDimitry Andric     Reg = State.AllocateReg(ArgVRs);
36345f757f3fSDimitry Andric   else if (ValVT.is256BitVector())
36355f757f3fSDimitry Andric     Reg = State.AllocateReg(ArgXRs);
3636bdd1243dSDimitry Andric   else
3637bdd1243dSDimitry Andric     Reg = State.AllocateReg(ArgGPRs);
3638bdd1243dSDimitry Andric 
3639bdd1243dSDimitry Andric   unsigned StackOffset =
3640bdd1243dSDimitry Andric       Reg ? 0 : State.AllocateStack(StoreSizeBytes, StackAlign);
3641bdd1243dSDimitry Andric 
3642bdd1243dSDimitry Andric   // If we reach this point and PendingLocs is non-empty, we must be at the
3643bdd1243dSDimitry Andric   // end of a split argument that must be passed indirectly.
3644bdd1243dSDimitry Andric   if (!PendingLocs.empty()) {
3645bdd1243dSDimitry Andric     assert(ArgFlags.isSplitEnd() && "Expected ArgFlags.isSplitEnd()");
3646bdd1243dSDimitry Andric     assert(PendingLocs.size() > 2 && "Unexpected PendingLocs.size()");
3647bdd1243dSDimitry Andric     for (auto &It : PendingLocs) {
3648bdd1243dSDimitry Andric       if (Reg)
3649bdd1243dSDimitry Andric         It.convertToReg(Reg);
3650bdd1243dSDimitry Andric       else
3651bdd1243dSDimitry Andric         It.convertToMem(StackOffset);
3652bdd1243dSDimitry Andric       State.addLoc(It);
3653bdd1243dSDimitry Andric     }
3654bdd1243dSDimitry Andric     PendingLocs.clear();
3655bdd1243dSDimitry Andric     PendingArgFlags.clear();
3656bdd1243dSDimitry Andric     return false;
3657bdd1243dSDimitry Andric   }
3658bdd1243dSDimitry Andric   assert((!UseGPRForFloat || LocVT == GRLenVT) &&
3659bdd1243dSDimitry Andric          "Expected an GRLenVT at this stage");
3660bdd1243dSDimitry Andric 
3661bdd1243dSDimitry Andric   if (Reg) {
3662bdd1243dSDimitry Andric     State.addLoc(CCValAssign::getReg(ValNo, ValVT, Reg, LocVT, LocInfo));
3663bdd1243dSDimitry Andric     return false;
3664bdd1243dSDimitry Andric   }
3665bdd1243dSDimitry Andric 
3666bdd1243dSDimitry Andric   // When a floating-point value is passed on the stack, no bit-cast is needed.
3667bdd1243dSDimitry Andric   if (ValVT.isFloatingPoint()) {
3668bdd1243dSDimitry Andric     LocVT = ValVT;
3669bdd1243dSDimitry Andric     LocInfo = CCValAssign::Full;
3670bdd1243dSDimitry Andric   }
3671bdd1243dSDimitry Andric 
3672bdd1243dSDimitry Andric   State.addLoc(CCValAssign::getMem(ValNo, ValVT, StackOffset, LocVT, LocInfo));
3673bdd1243dSDimitry Andric   return false;
367481ad6265SDimitry Andric }
367581ad6265SDimitry Andric 
367681ad6265SDimitry Andric void LoongArchTargetLowering::analyzeInputArgs(
3677bdd1243dSDimitry Andric     MachineFunction &MF, CCState &CCInfo,
3678bdd1243dSDimitry Andric     const SmallVectorImpl<ISD::InputArg> &Ins, bool IsRet,
367981ad6265SDimitry Andric     LoongArchCCAssignFn Fn) const {
3680bdd1243dSDimitry Andric   FunctionType *FType = MF.getFunction().getFunctionType();
368181ad6265SDimitry Andric   for (unsigned i = 0, e = Ins.size(); i != e; ++i) {
368281ad6265SDimitry Andric     MVT ArgVT = Ins[i].VT;
3683bdd1243dSDimitry Andric     Type *ArgTy = nullptr;
3684bdd1243dSDimitry Andric     if (IsRet)
3685bdd1243dSDimitry Andric       ArgTy = FType->getReturnType();
3686bdd1243dSDimitry Andric     else if (Ins[i].isOrigArg())
3687bdd1243dSDimitry Andric       ArgTy = FType->getParamType(Ins[i].getOrigArgIndex());
3688bdd1243dSDimitry Andric     LoongArchABI::ABI ABI =
3689bdd1243dSDimitry Andric         MF.getSubtarget<LoongArchSubtarget>().getTargetABI();
3690bdd1243dSDimitry Andric     if (Fn(MF.getDataLayout(), ABI, i, ArgVT, CCValAssign::Full, Ins[i].Flags,
3691bdd1243dSDimitry Andric            CCInfo, /*IsFixed=*/true, IsRet, ArgTy)) {
369206c3fb27SDimitry Andric       LLVM_DEBUG(dbgs() << "InputArg #" << i << " has unhandled type " << ArgVT
369306c3fb27SDimitry Andric                         << '\n');
369481ad6265SDimitry Andric       llvm_unreachable("");
369581ad6265SDimitry Andric     }
369681ad6265SDimitry Andric   }
369781ad6265SDimitry Andric }
369881ad6265SDimitry Andric 
369981ad6265SDimitry Andric void LoongArchTargetLowering::analyzeOutputArgs(
3700bdd1243dSDimitry Andric     MachineFunction &MF, CCState &CCInfo,
3701bdd1243dSDimitry Andric     const SmallVectorImpl<ISD::OutputArg> &Outs, bool IsRet,
3702bdd1243dSDimitry Andric     CallLoweringInfo *CLI, LoongArchCCAssignFn Fn) const {
370381ad6265SDimitry Andric   for (unsigned i = 0, e = Outs.size(); i != e; ++i) {
370481ad6265SDimitry Andric     MVT ArgVT = Outs[i].VT;
3705bdd1243dSDimitry Andric     Type *OrigTy = CLI ? CLI->getArgs()[Outs[i].OrigArgIndex].Ty : nullptr;
3706bdd1243dSDimitry Andric     LoongArchABI::ABI ABI =
3707bdd1243dSDimitry Andric         MF.getSubtarget<LoongArchSubtarget>().getTargetABI();
3708bdd1243dSDimitry Andric     if (Fn(MF.getDataLayout(), ABI, i, ArgVT, CCValAssign::Full, Outs[i].Flags,
3709bdd1243dSDimitry Andric            CCInfo, Outs[i].IsFixed, IsRet, OrigTy)) {
371006c3fb27SDimitry Andric       LLVM_DEBUG(dbgs() << "OutputArg #" << i << " has unhandled type " << ArgVT
371106c3fb27SDimitry Andric                         << "\n");
371281ad6265SDimitry Andric       llvm_unreachable("");
371381ad6265SDimitry Andric     }
371481ad6265SDimitry Andric   }
371581ad6265SDimitry Andric }
371681ad6265SDimitry Andric 
3717bdd1243dSDimitry Andric // Convert Val to a ValVT. Should not be called for CCValAssign::Indirect
3718bdd1243dSDimitry Andric // values.
3719bdd1243dSDimitry Andric static SDValue convertLocVTToValVT(SelectionDAG &DAG, SDValue Val,
3720bdd1243dSDimitry Andric                                    const CCValAssign &VA, const SDLoc &DL) {
3721bdd1243dSDimitry Andric   switch (VA.getLocInfo()) {
3722bdd1243dSDimitry Andric   default:
3723bdd1243dSDimitry Andric     llvm_unreachable("Unexpected CCValAssign::LocInfo");
3724bdd1243dSDimitry Andric   case CCValAssign::Full:
3725bdd1243dSDimitry Andric   case CCValAssign::Indirect:
3726bdd1243dSDimitry Andric     break;
3727bdd1243dSDimitry Andric   case CCValAssign::BCvt:
3728bdd1243dSDimitry Andric     if (VA.getLocVT() == MVT::i64 && VA.getValVT() == MVT::f32)
3729bdd1243dSDimitry Andric       Val = DAG.getNode(LoongArchISD::MOVGR2FR_W_LA64, DL, MVT::f32, Val);
3730bdd1243dSDimitry Andric     else
3731bdd1243dSDimitry Andric       Val = DAG.getNode(ISD::BITCAST, DL, VA.getValVT(), Val);
3732bdd1243dSDimitry Andric     break;
3733bdd1243dSDimitry Andric   }
3734bdd1243dSDimitry Andric   return Val;
3735bdd1243dSDimitry Andric }
3736bdd1243dSDimitry Andric 
373781ad6265SDimitry Andric static SDValue unpackFromRegLoc(SelectionDAG &DAG, SDValue Chain,
373881ad6265SDimitry Andric                                 const CCValAssign &VA, const SDLoc &DL,
373981ad6265SDimitry Andric                                 const LoongArchTargetLowering &TLI) {
374081ad6265SDimitry Andric   MachineFunction &MF = DAG.getMachineFunction();
374181ad6265SDimitry Andric   MachineRegisterInfo &RegInfo = MF.getRegInfo();
374281ad6265SDimitry Andric   EVT LocVT = VA.getLocVT();
3743bdd1243dSDimitry Andric   SDValue Val;
374481ad6265SDimitry Andric   const TargetRegisterClass *RC = TLI.getRegClassFor(LocVT.getSimpleVT());
374581ad6265SDimitry Andric   Register VReg = RegInfo.createVirtualRegister(RC);
374681ad6265SDimitry Andric   RegInfo.addLiveIn(VA.getLocReg(), VReg);
3747bdd1243dSDimitry Andric   Val = DAG.getCopyFromReg(Chain, DL, VReg, LocVT);
374881ad6265SDimitry Andric 
3749bdd1243dSDimitry Andric   return convertLocVTToValVT(DAG, Val, VA, DL);
3750bdd1243dSDimitry Andric }
3751bdd1243dSDimitry Andric 
3752bdd1243dSDimitry Andric // The caller is responsible for loading the full value if the argument is
3753bdd1243dSDimitry Andric // passed with CCValAssign::Indirect.
3754bdd1243dSDimitry Andric static SDValue unpackFromMemLoc(SelectionDAG &DAG, SDValue Chain,
3755bdd1243dSDimitry Andric                                 const CCValAssign &VA, const SDLoc &DL) {
3756bdd1243dSDimitry Andric   MachineFunction &MF = DAG.getMachineFunction();
3757bdd1243dSDimitry Andric   MachineFrameInfo &MFI = MF.getFrameInfo();
3758bdd1243dSDimitry Andric   EVT ValVT = VA.getValVT();
3759bdd1243dSDimitry Andric   int FI = MFI.CreateFixedObject(ValVT.getStoreSize(), VA.getLocMemOffset(),
3760bdd1243dSDimitry Andric                                  /*IsImmutable=*/true);
3761bdd1243dSDimitry Andric   SDValue FIN = DAG.getFrameIndex(
3762bdd1243dSDimitry Andric       FI, MVT::getIntegerVT(DAG.getDataLayout().getPointerSizeInBits(0)));
3763bdd1243dSDimitry Andric 
3764bdd1243dSDimitry Andric   ISD::LoadExtType ExtType;
3765bdd1243dSDimitry Andric   switch (VA.getLocInfo()) {
3766bdd1243dSDimitry Andric   default:
3767bdd1243dSDimitry Andric     llvm_unreachable("Unexpected CCValAssign::LocInfo");
3768bdd1243dSDimitry Andric   case CCValAssign::Full:
3769bdd1243dSDimitry Andric   case CCValAssign::Indirect:
3770bdd1243dSDimitry Andric   case CCValAssign::BCvt:
3771bdd1243dSDimitry Andric     ExtType = ISD::NON_EXTLOAD;
3772bdd1243dSDimitry Andric     break;
3773bdd1243dSDimitry Andric   }
3774bdd1243dSDimitry Andric   return DAG.getExtLoad(
3775bdd1243dSDimitry Andric       ExtType, DL, VA.getLocVT(), Chain, FIN,
3776bdd1243dSDimitry Andric       MachinePointerInfo::getFixedStack(DAG.getMachineFunction(), FI), ValVT);
3777bdd1243dSDimitry Andric }
3778bdd1243dSDimitry Andric 
3779bdd1243dSDimitry Andric static SDValue convertValVTToLocVT(SelectionDAG &DAG, SDValue Val,
3780bdd1243dSDimitry Andric                                    const CCValAssign &VA, const SDLoc &DL) {
3781bdd1243dSDimitry Andric   EVT LocVT = VA.getLocVT();
3782bdd1243dSDimitry Andric 
3783bdd1243dSDimitry Andric   switch (VA.getLocInfo()) {
3784bdd1243dSDimitry Andric   default:
3785bdd1243dSDimitry Andric     llvm_unreachable("Unexpected CCValAssign::LocInfo");
3786bdd1243dSDimitry Andric   case CCValAssign::Full:
3787bdd1243dSDimitry Andric     break;
3788bdd1243dSDimitry Andric   case CCValAssign::BCvt:
3789bdd1243dSDimitry Andric     if (VA.getLocVT() == MVT::i64 && VA.getValVT() == MVT::f32)
3790bdd1243dSDimitry Andric       Val = DAG.getNode(LoongArchISD::MOVFR2GR_S_LA64, DL, MVT::i64, Val);
3791bdd1243dSDimitry Andric     else
3792bdd1243dSDimitry Andric       Val = DAG.getNode(ISD::BITCAST, DL, LocVT, Val);
3793bdd1243dSDimitry Andric     break;
3794bdd1243dSDimitry Andric   }
3795bdd1243dSDimitry Andric   return Val;
3796bdd1243dSDimitry Andric }
3797bdd1243dSDimitry Andric 
3798bdd1243dSDimitry Andric static bool CC_LoongArch_GHC(unsigned ValNo, MVT ValVT, MVT LocVT,
3799bdd1243dSDimitry Andric                              CCValAssign::LocInfo LocInfo,
3800bdd1243dSDimitry Andric                              ISD::ArgFlagsTy ArgFlags, CCState &State) {
3801bdd1243dSDimitry Andric   if (LocVT == MVT::i32 || LocVT == MVT::i64) {
3802bdd1243dSDimitry Andric     // Pass in STG registers: Base, Sp, Hp, R1, R2, R3, R4, R5, SpLim
3803bdd1243dSDimitry Andric     //                        s0    s1  s2  s3  s4  s5  s6  s7  s8
3804bdd1243dSDimitry Andric     static const MCPhysReg GPRList[] = {
380506c3fb27SDimitry Andric         LoongArch::R23, LoongArch::R24, LoongArch::R25,
380606c3fb27SDimitry Andric         LoongArch::R26, LoongArch::R27, LoongArch::R28,
380706c3fb27SDimitry Andric         LoongArch::R29, LoongArch::R30, LoongArch::R31};
3808bdd1243dSDimitry Andric     if (unsigned Reg = State.AllocateReg(GPRList)) {
3809bdd1243dSDimitry Andric       State.addLoc(CCValAssign::getReg(ValNo, ValVT, Reg, LocVT, LocInfo));
3810bdd1243dSDimitry Andric       return false;
3811bdd1243dSDimitry Andric     }
3812bdd1243dSDimitry Andric   }
3813bdd1243dSDimitry Andric 
3814bdd1243dSDimitry Andric   if (LocVT == MVT::f32) {
3815bdd1243dSDimitry Andric     // Pass in STG registers: F1, F2, F3, F4
3816bdd1243dSDimitry Andric     //                        fs0,fs1,fs2,fs3
3817bdd1243dSDimitry Andric     static const MCPhysReg FPR32List[] = {LoongArch::F24, LoongArch::F25,
3818bdd1243dSDimitry Andric                                           LoongArch::F26, LoongArch::F27};
3819bdd1243dSDimitry Andric     if (unsigned Reg = State.AllocateReg(FPR32List)) {
3820bdd1243dSDimitry Andric       State.addLoc(CCValAssign::getReg(ValNo, ValVT, Reg, LocVT, LocInfo));
3821bdd1243dSDimitry Andric       return false;
3822bdd1243dSDimitry Andric     }
3823bdd1243dSDimitry Andric   }
3824bdd1243dSDimitry Andric 
3825bdd1243dSDimitry Andric   if (LocVT == MVT::f64) {
3826bdd1243dSDimitry Andric     // Pass in STG registers: D1, D2, D3, D4
3827bdd1243dSDimitry Andric     //                        fs4,fs5,fs6,fs7
3828bdd1243dSDimitry Andric     static const MCPhysReg FPR64List[] = {LoongArch::F28_64, LoongArch::F29_64,
3829bdd1243dSDimitry Andric                                           LoongArch::F30_64, LoongArch::F31_64};
3830bdd1243dSDimitry Andric     if (unsigned Reg = State.AllocateReg(FPR64List)) {
3831bdd1243dSDimitry Andric       State.addLoc(CCValAssign::getReg(ValNo, ValVT, Reg, LocVT, LocInfo));
3832bdd1243dSDimitry Andric       return false;
3833bdd1243dSDimitry Andric     }
3834bdd1243dSDimitry Andric   }
3835bdd1243dSDimitry Andric 
3836bdd1243dSDimitry Andric   report_fatal_error("No registers left in GHC calling convention");
3837bdd1243dSDimitry Andric   return true;
383881ad6265SDimitry Andric }
383981ad6265SDimitry Andric 
384081ad6265SDimitry Andric // Transform physical registers into virtual registers.
384181ad6265SDimitry Andric SDValue LoongArchTargetLowering::LowerFormalArguments(
384281ad6265SDimitry Andric     SDValue Chain, CallingConv::ID CallConv, bool IsVarArg,
384381ad6265SDimitry Andric     const SmallVectorImpl<ISD::InputArg> &Ins, const SDLoc &DL,
384481ad6265SDimitry Andric     SelectionDAG &DAG, SmallVectorImpl<SDValue> &InVals) const {
384581ad6265SDimitry Andric 
384681ad6265SDimitry Andric   MachineFunction &MF = DAG.getMachineFunction();
384781ad6265SDimitry Andric 
384881ad6265SDimitry Andric   switch (CallConv) {
384981ad6265SDimitry Andric   default:
385081ad6265SDimitry Andric     llvm_unreachable("Unsupported calling convention");
385181ad6265SDimitry Andric   case CallingConv::C:
3852bdd1243dSDimitry Andric   case CallingConv::Fast:
385381ad6265SDimitry Andric     break;
3854bdd1243dSDimitry Andric   case CallingConv::GHC:
385506c3fb27SDimitry Andric     if (!MF.getSubtarget().hasFeature(LoongArch::FeatureBasicF) ||
385606c3fb27SDimitry Andric         !MF.getSubtarget().hasFeature(LoongArch::FeatureBasicD))
3857bdd1243dSDimitry Andric       report_fatal_error(
3858bdd1243dSDimitry Andric           "GHC calling convention requires the F and D extensions");
385981ad6265SDimitry Andric   }
386081ad6265SDimitry Andric 
3861bdd1243dSDimitry Andric   EVT PtrVT = getPointerTy(DAG.getDataLayout());
3862bdd1243dSDimitry Andric   MVT GRLenVT = Subtarget.getGRLenVT();
3863bdd1243dSDimitry Andric   unsigned GRLenInBytes = Subtarget.getGRLen() / 8;
3864bdd1243dSDimitry Andric   // Used with varargs to acumulate store chains.
3865bdd1243dSDimitry Andric   std::vector<SDValue> OutChains;
3866bdd1243dSDimitry Andric 
386781ad6265SDimitry Andric   // Assign locations to all of the incoming arguments.
386881ad6265SDimitry Andric   SmallVector<CCValAssign> ArgLocs;
386981ad6265SDimitry Andric   CCState CCInfo(CallConv, IsVarArg, MF, ArgLocs, *DAG.getContext());
387081ad6265SDimitry Andric 
3871bdd1243dSDimitry Andric   if (CallConv == CallingConv::GHC)
3872bdd1243dSDimitry Andric     CCInfo.AnalyzeFormalArguments(Ins, CC_LoongArch_GHC);
3873bdd1243dSDimitry Andric   else
3874bdd1243dSDimitry Andric     analyzeInputArgs(MF, CCInfo, Ins, /*IsRet=*/false, CC_LoongArch);
387581ad6265SDimitry Andric 
3876bdd1243dSDimitry Andric   for (unsigned i = 0, e = ArgLocs.size(); i != e; ++i) {
3877bdd1243dSDimitry Andric     CCValAssign &VA = ArgLocs[i];
3878bdd1243dSDimitry Andric     SDValue ArgValue;
3879bdd1243dSDimitry Andric     if (VA.isRegLoc())
3880bdd1243dSDimitry Andric       ArgValue = unpackFromRegLoc(DAG, Chain, VA, DL, *this);
3881bdd1243dSDimitry Andric     else
3882bdd1243dSDimitry Andric       ArgValue = unpackFromMemLoc(DAG, Chain, VA, DL);
3883bdd1243dSDimitry Andric     if (VA.getLocInfo() == CCValAssign::Indirect) {
3884bdd1243dSDimitry Andric       // If the original argument was split and passed by reference, we need to
3885bdd1243dSDimitry Andric       // load all parts of it here (using the same address).
3886bdd1243dSDimitry Andric       InVals.push_back(DAG.getLoad(VA.getValVT(), DL, Chain, ArgValue,
3887bdd1243dSDimitry Andric                                    MachinePointerInfo()));
3888bdd1243dSDimitry Andric       unsigned ArgIndex = Ins[i].OrigArgIndex;
3889bdd1243dSDimitry Andric       unsigned ArgPartOffset = Ins[i].PartOffset;
3890bdd1243dSDimitry Andric       assert(ArgPartOffset == 0);
3891bdd1243dSDimitry Andric       while (i + 1 != e && Ins[i + 1].OrigArgIndex == ArgIndex) {
3892bdd1243dSDimitry Andric         CCValAssign &PartVA = ArgLocs[i + 1];
3893bdd1243dSDimitry Andric         unsigned PartOffset = Ins[i + 1].PartOffset - ArgPartOffset;
3894bdd1243dSDimitry Andric         SDValue Offset = DAG.getIntPtrConstant(PartOffset, DL);
3895bdd1243dSDimitry Andric         SDValue Address = DAG.getNode(ISD::ADD, DL, PtrVT, ArgValue, Offset);
3896bdd1243dSDimitry Andric         InVals.push_back(DAG.getLoad(PartVA.getValVT(), DL, Chain, Address,
3897bdd1243dSDimitry Andric                                      MachinePointerInfo()));
3898bdd1243dSDimitry Andric         ++i;
3899bdd1243dSDimitry Andric       }
3900bdd1243dSDimitry Andric       continue;
3901bdd1243dSDimitry Andric     }
3902bdd1243dSDimitry Andric     InVals.push_back(ArgValue);
3903bdd1243dSDimitry Andric   }
3904bdd1243dSDimitry Andric 
3905bdd1243dSDimitry Andric   if (IsVarArg) {
3906bdd1243dSDimitry Andric     ArrayRef<MCPhysReg> ArgRegs = ArrayRef(ArgGPRs);
3907bdd1243dSDimitry Andric     unsigned Idx = CCInfo.getFirstUnallocated(ArgRegs);
3908bdd1243dSDimitry Andric     const TargetRegisterClass *RC = &LoongArch::GPRRegClass;
3909bdd1243dSDimitry Andric     MachineFrameInfo &MFI = MF.getFrameInfo();
3910bdd1243dSDimitry Andric     MachineRegisterInfo &RegInfo = MF.getRegInfo();
3911bdd1243dSDimitry Andric     auto *LoongArchFI = MF.getInfo<LoongArchMachineFunctionInfo>();
3912bdd1243dSDimitry Andric 
3913bdd1243dSDimitry Andric     // Offset of the first variable argument from stack pointer, and size of
3914bdd1243dSDimitry Andric     // the vararg save area. For now, the varargs save area is either zero or
3915bdd1243dSDimitry Andric     // large enough to hold a0-a7.
3916bdd1243dSDimitry Andric     int VaArgOffset, VarArgsSaveSize;
3917bdd1243dSDimitry Andric 
3918bdd1243dSDimitry Andric     // If all registers are allocated, then all varargs must be passed on the
3919bdd1243dSDimitry Andric     // stack and we don't need to save any argregs.
3920bdd1243dSDimitry Andric     if (ArgRegs.size() == Idx) {
392106c3fb27SDimitry Andric       VaArgOffset = CCInfo.getStackSize();
3922bdd1243dSDimitry Andric       VarArgsSaveSize = 0;
3923bdd1243dSDimitry Andric     } else {
3924bdd1243dSDimitry Andric       VarArgsSaveSize = GRLenInBytes * (ArgRegs.size() - Idx);
3925bdd1243dSDimitry Andric       VaArgOffset = -VarArgsSaveSize;
3926bdd1243dSDimitry Andric     }
3927bdd1243dSDimitry Andric 
3928bdd1243dSDimitry Andric     // Record the frame index of the first variable argument
3929bdd1243dSDimitry Andric     // which is a value necessary to VASTART.
3930bdd1243dSDimitry Andric     int FI = MFI.CreateFixedObject(GRLenInBytes, VaArgOffset, true);
3931bdd1243dSDimitry Andric     LoongArchFI->setVarArgsFrameIndex(FI);
3932bdd1243dSDimitry Andric 
3933bdd1243dSDimitry Andric     // If saving an odd number of registers then create an extra stack slot to
3934bdd1243dSDimitry Andric     // ensure that the frame pointer is 2*GRLen-aligned, which in turn ensures
3935bdd1243dSDimitry Andric     // offsets to even-numbered registered remain 2*GRLen-aligned.
3936bdd1243dSDimitry Andric     if (Idx % 2) {
3937bdd1243dSDimitry Andric       MFI.CreateFixedObject(GRLenInBytes, VaArgOffset - (int)GRLenInBytes,
3938bdd1243dSDimitry Andric                             true);
3939bdd1243dSDimitry Andric       VarArgsSaveSize += GRLenInBytes;
3940bdd1243dSDimitry Andric     }
3941bdd1243dSDimitry Andric 
3942bdd1243dSDimitry Andric     // Copy the integer registers that may have been used for passing varargs
3943bdd1243dSDimitry Andric     // to the vararg save area.
3944bdd1243dSDimitry Andric     for (unsigned I = Idx; I < ArgRegs.size();
3945bdd1243dSDimitry Andric          ++I, VaArgOffset += GRLenInBytes) {
3946bdd1243dSDimitry Andric       const Register Reg = RegInfo.createVirtualRegister(RC);
3947bdd1243dSDimitry Andric       RegInfo.addLiveIn(ArgRegs[I], Reg);
3948bdd1243dSDimitry Andric       SDValue ArgValue = DAG.getCopyFromReg(Chain, DL, Reg, GRLenVT);
3949bdd1243dSDimitry Andric       FI = MFI.CreateFixedObject(GRLenInBytes, VaArgOffset, true);
3950bdd1243dSDimitry Andric       SDValue PtrOff = DAG.getFrameIndex(FI, getPointerTy(DAG.getDataLayout()));
3951bdd1243dSDimitry Andric       SDValue Store = DAG.getStore(Chain, DL, ArgValue, PtrOff,
3952bdd1243dSDimitry Andric                                    MachinePointerInfo::getFixedStack(MF, FI));
3953bdd1243dSDimitry Andric       cast<StoreSDNode>(Store.getNode())
3954bdd1243dSDimitry Andric           ->getMemOperand()
3955bdd1243dSDimitry Andric           ->setValue((Value *)nullptr);
3956bdd1243dSDimitry Andric       OutChains.push_back(Store);
3957bdd1243dSDimitry Andric     }
3958bdd1243dSDimitry Andric     LoongArchFI->setVarArgsSaveSize(VarArgsSaveSize);
3959bdd1243dSDimitry Andric   }
3960bdd1243dSDimitry Andric 
3961bdd1243dSDimitry Andric   // All stores are grouped in one node to allow the matching between
3962bdd1243dSDimitry Andric   // the size of Ins and InVals. This only happens for vararg functions.
3963bdd1243dSDimitry Andric   if (!OutChains.empty()) {
3964bdd1243dSDimitry Andric     OutChains.push_back(Chain);
3965bdd1243dSDimitry Andric     Chain = DAG.getNode(ISD::TokenFactor, DL, MVT::Other, OutChains);
3966bdd1243dSDimitry Andric   }
396781ad6265SDimitry Andric 
396881ad6265SDimitry Andric   return Chain;
396981ad6265SDimitry Andric }
397081ad6265SDimitry Andric 
3971bdd1243dSDimitry Andric bool LoongArchTargetLowering::mayBeEmittedAsTailCall(const CallInst *CI) const {
3972bdd1243dSDimitry Andric   return CI->isTailCall();
3973bdd1243dSDimitry Andric }
3974bdd1243dSDimitry Andric 
397506c3fb27SDimitry Andric // Check if the return value is used as only a return value, as otherwise
397606c3fb27SDimitry Andric // we can't perform a tail-call.
397706c3fb27SDimitry Andric bool LoongArchTargetLowering::isUsedByReturnOnly(SDNode *N,
397806c3fb27SDimitry Andric                                                  SDValue &Chain) const {
397906c3fb27SDimitry Andric   if (N->getNumValues() != 1)
398006c3fb27SDimitry Andric     return false;
398106c3fb27SDimitry Andric   if (!N->hasNUsesOfValue(1, 0))
398206c3fb27SDimitry Andric     return false;
398306c3fb27SDimitry Andric 
398406c3fb27SDimitry Andric   SDNode *Copy = *N->use_begin();
398506c3fb27SDimitry Andric   if (Copy->getOpcode() != ISD::CopyToReg)
398606c3fb27SDimitry Andric     return false;
398706c3fb27SDimitry Andric 
398806c3fb27SDimitry Andric   // If the ISD::CopyToReg has a glue operand, we conservatively assume it
398906c3fb27SDimitry Andric   // isn't safe to perform a tail call.
399006c3fb27SDimitry Andric   if (Copy->getGluedNode())
399106c3fb27SDimitry Andric     return false;
399206c3fb27SDimitry Andric 
399306c3fb27SDimitry Andric   // The copy must be used by a LoongArchISD::RET, and nothing else.
399406c3fb27SDimitry Andric   bool HasRet = false;
399506c3fb27SDimitry Andric   for (SDNode *Node : Copy->uses()) {
399606c3fb27SDimitry Andric     if (Node->getOpcode() != LoongArchISD::RET)
399706c3fb27SDimitry Andric       return false;
399806c3fb27SDimitry Andric     HasRet = true;
399906c3fb27SDimitry Andric   }
400006c3fb27SDimitry Andric 
400106c3fb27SDimitry Andric   if (!HasRet)
400206c3fb27SDimitry Andric     return false;
400306c3fb27SDimitry Andric 
400406c3fb27SDimitry Andric   Chain = Copy->getOperand(0);
400506c3fb27SDimitry Andric   return true;
400606c3fb27SDimitry Andric }
400706c3fb27SDimitry Andric 
4008bdd1243dSDimitry Andric // Check whether the call is eligible for tail call optimization.
4009bdd1243dSDimitry Andric bool LoongArchTargetLowering::isEligibleForTailCallOptimization(
4010bdd1243dSDimitry Andric     CCState &CCInfo, CallLoweringInfo &CLI, MachineFunction &MF,
4011bdd1243dSDimitry Andric     const SmallVectorImpl<CCValAssign> &ArgLocs) const {
4012bdd1243dSDimitry Andric 
4013bdd1243dSDimitry Andric   auto CalleeCC = CLI.CallConv;
4014bdd1243dSDimitry Andric   auto &Outs = CLI.Outs;
4015bdd1243dSDimitry Andric   auto &Caller = MF.getFunction();
4016bdd1243dSDimitry Andric   auto CallerCC = Caller.getCallingConv();
4017bdd1243dSDimitry Andric 
4018bdd1243dSDimitry Andric   // Do not tail call opt if the stack is used to pass parameters.
401906c3fb27SDimitry Andric   if (CCInfo.getStackSize() != 0)
4020bdd1243dSDimitry Andric     return false;
4021bdd1243dSDimitry Andric 
4022bdd1243dSDimitry Andric   // Do not tail call opt if any parameters need to be passed indirectly.
4023bdd1243dSDimitry Andric   for (auto &VA : ArgLocs)
4024bdd1243dSDimitry Andric     if (VA.getLocInfo() == CCValAssign::Indirect)
4025bdd1243dSDimitry Andric       return false;
4026bdd1243dSDimitry Andric 
4027bdd1243dSDimitry Andric   // Do not tail call opt if either caller or callee uses struct return
4028bdd1243dSDimitry Andric   // semantics.
4029bdd1243dSDimitry Andric   auto IsCallerStructRet = Caller.hasStructRetAttr();
4030bdd1243dSDimitry Andric   auto IsCalleeStructRet = Outs.empty() ? false : Outs[0].Flags.isSRet();
4031bdd1243dSDimitry Andric   if (IsCallerStructRet || IsCalleeStructRet)
4032bdd1243dSDimitry Andric     return false;
4033bdd1243dSDimitry Andric 
4034bdd1243dSDimitry Andric   // Do not tail call opt if either the callee or caller has a byval argument.
4035bdd1243dSDimitry Andric   for (auto &Arg : Outs)
4036bdd1243dSDimitry Andric     if (Arg.Flags.isByVal())
4037bdd1243dSDimitry Andric       return false;
4038bdd1243dSDimitry Andric 
4039bdd1243dSDimitry Andric   // The callee has to preserve all registers the caller needs to preserve.
4040bdd1243dSDimitry Andric   const LoongArchRegisterInfo *TRI = Subtarget.getRegisterInfo();
4041bdd1243dSDimitry Andric   const uint32_t *CallerPreserved = TRI->getCallPreservedMask(MF, CallerCC);
4042bdd1243dSDimitry Andric   if (CalleeCC != CallerCC) {
4043bdd1243dSDimitry Andric     const uint32_t *CalleePreserved = TRI->getCallPreservedMask(MF, CalleeCC);
4044bdd1243dSDimitry Andric     if (!TRI->regmaskSubsetEqual(CallerPreserved, CalleePreserved))
4045bdd1243dSDimitry Andric       return false;
4046bdd1243dSDimitry Andric   }
4047bdd1243dSDimitry Andric   return true;
4048bdd1243dSDimitry Andric }
4049bdd1243dSDimitry Andric 
4050bdd1243dSDimitry Andric static Align getPrefTypeAlign(EVT VT, SelectionDAG &DAG) {
4051bdd1243dSDimitry Andric   return DAG.getDataLayout().getPrefTypeAlign(
4052bdd1243dSDimitry Andric       VT.getTypeForEVT(*DAG.getContext()));
4053bdd1243dSDimitry Andric }
4054bdd1243dSDimitry Andric 
4055753f127fSDimitry Andric // Lower a call to a callseq_start + CALL + callseq_end chain, and add input
4056753f127fSDimitry Andric // and output parameter nodes.
4057753f127fSDimitry Andric SDValue
4058753f127fSDimitry Andric LoongArchTargetLowering::LowerCall(CallLoweringInfo &CLI,
4059753f127fSDimitry Andric                                    SmallVectorImpl<SDValue> &InVals) const {
4060753f127fSDimitry Andric   SelectionDAG &DAG = CLI.DAG;
4061753f127fSDimitry Andric   SDLoc &DL = CLI.DL;
4062753f127fSDimitry Andric   SmallVectorImpl<ISD::OutputArg> &Outs = CLI.Outs;
4063753f127fSDimitry Andric   SmallVectorImpl<SDValue> &OutVals = CLI.OutVals;
4064753f127fSDimitry Andric   SmallVectorImpl<ISD::InputArg> &Ins = CLI.Ins;
4065753f127fSDimitry Andric   SDValue Chain = CLI.Chain;
4066753f127fSDimitry Andric   SDValue Callee = CLI.Callee;
4067753f127fSDimitry Andric   CallingConv::ID CallConv = CLI.CallConv;
4068753f127fSDimitry Andric   bool IsVarArg = CLI.IsVarArg;
4069753f127fSDimitry Andric   EVT PtrVT = getPointerTy(DAG.getDataLayout());
4070bdd1243dSDimitry Andric   MVT GRLenVT = Subtarget.getGRLenVT();
4071bdd1243dSDimitry Andric   bool &IsTailCall = CLI.IsTailCall;
4072753f127fSDimitry Andric 
4073753f127fSDimitry Andric   MachineFunction &MF = DAG.getMachineFunction();
4074753f127fSDimitry Andric 
4075753f127fSDimitry Andric   // Analyze the operands of the call, assigning locations to each operand.
4076753f127fSDimitry Andric   SmallVector<CCValAssign> ArgLocs;
4077753f127fSDimitry Andric   CCState ArgCCInfo(CallConv, IsVarArg, MF, ArgLocs, *DAG.getContext());
4078753f127fSDimitry Andric 
4079bdd1243dSDimitry Andric   if (CallConv == CallingConv::GHC)
4080bdd1243dSDimitry Andric     ArgCCInfo.AnalyzeCallOperands(Outs, CC_LoongArch_GHC);
4081bdd1243dSDimitry Andric   else
4082bdd1243dSDimitry Andric     analyzeOutputArgs(MF, ArgCCInfo, Outs, /*IsRet=*/false, &CLI, CC_LoongArch);
4083bdd1243dSDimitry Andric 
4084bdd1243dSDimitry Andric   // Check if it's really possible to do a tail call.
4085bdd1243dSDimitry Andric   if (IsTailCall)
4086bdd1243dSDimitry Andric     IsTailCall = isEligibleForTailCallOptimization(ArgCCInfo, CLI, MF, ArgLocs);
4087bdd1243dSDimitry Andric 
4088bdd1243dSDimitry Andric   if (IsTailCall)
4089bdd1243dSDimitry Andric     ++NumTailCalls;
4090bdd1243dSDimitry Andric   else if (CLI.CB && CLI.CB->isMustTailCall())
4091bdd1243dSDimitry Andric     report_fatal_error("failed to perform tail call elimination on a call "
4092bdd1243dSDimitry Andric                        "site marked musttail");
4093753f127fSDimitry Andric 
4094753f127fSDimitry Andric   // Get a count of how many bytes are to be pushed on the stack.
409506c3fb27SDimitry Andric   unsigned NumBytes = ArgCCInfo.getStackSize();
4096753f127fSDimitry Andric 
4097bdd1243dSDimitry Andric   // Create local copies for byval args.
4098bdd1243dSDimitry Andric   SmallVector<SDValue> ByValArgs;
4099bdd1243dSDimitry Andric   for (unsigned i = 0, e = Outs.size(); i != e; ++i) {
4100bdd1243dSDimitry Andric     ISD::ArgFlagsTy Flags = Outs[i].Flags;
4101bdd1243dSDimitry Andric     if (!Flags.isByVal())
4102753f127fSDimitry Andric       continue;
4103bdd1243dSDimitry Andric 
4104bdd1243dSDimitry Andric     SDValue Arg = OutVals[i];
4105bdd1243dSDimitry Andric     unsigned Size = Flags.getByValSize();
4106bdd1243dSDimitry Andric     Align Alignment = Flags.getNonZeroByValAlign();
4107bdd1243dSDimitry Andric 
4108bdd1243dSDimitry Andric     int FI =
4109bdd1243dSDimitry Andric         MF.getFrameInfo().CreateStackObject(Size, Alignment, /*isSS=*/false);
4110bdd1243dSDimitry Andric     SDValue FIPtr = DAG.getFrameIndex(FI, getPointerTy(DAG.getDataLayout()));
4111bdd1243dSDimitry Andric     SDValue SizeNode = DAG.getConstant(Size, DL, GRLenVT);
4112bdd1243dSDimitry Andric 
4113bdd1243dSDimitry Andric     Chain = DAG.getMemcpy(Chain, DL, FIPtr, Arg, SizeNode, Alignment,
4114bdd1243dSDimitry Andric                           /*IsVolatile=*/false,
4115bdd1243dSDimitry Andric                           /*AlwaysInline=*/false, /*isTailCall=*/IsTailCall,
4116bdd1243dSDimitry Andric                           MachinePointerInfo(), MachinePointerInfo());
4117bdd1243dSDimitry Andric     ByValArgs.push_back(FIPtr);
4118753f127fSDimitry Andric   }
4119753f127fSDimitry Andric 
4120bdd1243dSDimitry Andric   if (!IsTailCall)
4121753f127fSDimitry Andric     Chain = DAG.getCALLSEQ_START(Chain, NumBytes, 0, CLI.DL);
4122753f127fSDimitry Andric 
4123753f127fSDimitry Andric   // Copy argument values to their designated locations.
4124753f127fSDimitry Andric   SmallVector<std::pair<Register, SDValue>> RegsToPass;
4125bdd1243dSDimitry Andric   SmallVector<SDValue> MemOpChains;
4126bdd1243dSDimitry Andric   SDValue StackPtr;
4127bdd1243dSDimitry Andric   for (unsigned i = 0, j = 0, e = ArgLocs.size(); i != e; ++i) {
4128753f127fSDimitry Andric     CCValAssign &VA = ArgLocs[i];
4129753f127fSDimitry Andric     SDValue ArgValue = OutVals[i];
4130bdd1243dSDimitry Andric     ISD::ArgFlagsTy Flags = Outs[i].Flags;
4131753f127fSDimitry Andric 
4132753f127fSDimitry Andric     // Promote the value if needed.
4133bdd1243dSDimitry Andric     // For now, only handle fully promoted and indirect arguments.
4134bdd1243dSDimitry Andric     if (VA.getLocInfo() == CCValAssign::Indirect) {
4135bdd1243dSDimitry Andric       // Store the argument in a stack slot and pass its address.
4136bdd1243dSDimitry Andric       Align StackAlign =
4137bdd1243dSDimitry Andric           std::max(getPrefTypeAlign(Outs[i].ArgVT, DAG),
4138bdd1243dSDimitry Andric                    getPrefTypeAlign(ArgValue.getValueType(), DAG));
4139bdd1243dSDimitry Andric       TypeSize StoredSize = ArgValue.getValueType().getStoreSize();
4140bdd1243dSDimitry Andric       // If the original argument was split and passed by reference, we need to
4141bdd1243dSDimitry Andric       // store the required parts of it here (and pass just one address).
4142bdd1243dSDimitry Andric       unsigned ArgIndex = Outs[i].OrigArgIndex;
4143bdd1243dSDimitry Andric       unsigned ArgPartOffset = Outs[i].PartOffset;
4144bdd1243dSDimitry Andric       assert(ArgPartOffset == 0);
4145bdd1243dSDimitry Andric       // Calculate the total size to store. We don't have access to what we're
4146bdd1243dSDimitry Andric       // actually storing other than performing the loop and collecting the
4147bdd1243dSDimitry Andric       // info.
4148bdd1243dSDimitry Andric       SmallVector<std::pair<SDValue, SDValue>> Parts;
4149bdd1243dSDimitry Andric       while (i + 1 != e && Outs[i + 1].OrigArgIndex == ArgIndex) {
4150bdd1243dSDimitry Andric         SDValue PartValue = OutVals[i + 1];
4151bdd1243dSDimitry Andric         unsigned PartOffset = Outs[i + 1].PartOffset - ArgPartOffset;
4152bdd1243dSDimitry Andric         SDValue Offset = DAG.getIntPtrConstant(PartOffset, DL);
4153bdd1243dSDimitry Andric         EVT PartVT = PartValue.getValueType();
4154bdd1243dSDimitry Andric 
4155bdd1243dSDimitry Andric         StoredSize += PartVT.getStoreSize();
4156bdd1243dSDimitry Andric         StackAlign = std::max(StackAlign, getPrefTypeAlign(PartVT, DAG));
4157bdd1243dSDimitry Andric         Parts.push_back(std::make_pair(PartValue, Offset));
4158bdd1243dSDimitry Andric         ++i;
4159bdd1243dSDimitry Andric       }
4160bdd1243dSDimitry Andric       SDValue SpillSlot = DAG.CreateStackTemporary(StoredSize, StackAlign);
4161bdd1243dSDimitry Andric       int FI = cast<FrameIndexSDNode>(SpillSlot)->getIndex();
4162bdd1243dSDimitry Andric       MemOpChains.push_back(
4163bdd1243dSDimitry Andric           DAG.getStore(Chain, DL, ArgValue, SpillSlot,
4164bdd1243dSDimitry Andric                        MachinePointerInfo::getFixedStack(MF, FI)));
4165bdd1243dSDimitry Andric       for (const auto &Part : Parts) {
4166bdd1243dSDimitry Andric         SDValue PartValue = Part.first;
4167bdd1243dSDimitry Andric         SDValue PartOffset = Part.second;
4168bdd1243dSDimitry Andric         SDValue Address =
4169bdd1243dSDimitry Andric             DAG.getNode(ISD::ADD, DL, PtrVT, SpillSlot, PartOffset);
4170bdd1243dSDimitry Andric         MemOpChains.push_back(
4171bdd1243dSDimitry Andric             DAG.getStore(Chain, DL, PartValue, Address,
4172bdd1243dSDimitry Andric                          MachinePointerInfo::getFixedStack(MF, FI)));
4173bdd1243dSDimitry Andric       }
4174bdd1243dSDimitry Andric       ArgValue = SpillSlot;
4175bdd1243dSDimitry Andric     } else {
4176bdd1243dSDimitry Andric       ArgValue = convertValVTToLocVT(DAG, ArgValue, VA, DL);
4177bdd1243dSDimitry Andric     }
4178bdd1243dSDimitry Andric 
4179bdd1243dSDimitry Andric     // Use local copy if it is a byval arg.
4180bdd1243dSDimitry Andric     if (Flags.isByVal())
4181bdd1243dSDimitry Andric       ArgValue = ByValArgs[j++];
4182753f127fSDimitry Andric 
4183753f127fSDimitry Andric     if (VA.isRegLoc()) {
4184753f127fSDimitry Andric       // Queue up the argument copies and emit them at the end.
4185753f127fSDimitry Andric       RegsToPass.push_back(std::make_pair(VA.getLocReg(), ArgValue));
4186753f127fSDimitry Andric     } else {
4187bdd1243dSDimitry Andric       assert(VA.isMemLoc() && "Argument not register or memory");
4188bdd1243dSDimitry Andric       assert(!IsTailCall && "Tail call not allowed if stack is used "
4189bdd1243dSDimitry Andric                             "for passing parameters");
4190bdd1243dSDimitry Andric 
4191bdd1243dSDimitry Andric       // Work out the address of the stack slot.
4192bdd1243dSDimitry Andric       if (!StackPtr.getNode())
4193bdd1243dSDimitry Andric         StackPtr = DAG.getCopyFromReg(Chain, DL, LoongArch::R3, PtrVT);
4194bdd1243dSDimitry Andric       SDValue Address =
4195bdd1243dSDimitry Andric           DAG.getNode(ISD::ADD, DL, PtrVT, StackPtr,
4196bdd1243dSDimitry Andric                       DAG.getIntPtrConstant(VA.getLocMemOffset(), DL));
4197bdd1243dSDimitry Andric 
4198bdd1243dSDimitry Andric       // Emit the store.
4199bdd1243dSDimitry Andric       MemOpChains.push_back(
4200bdd1243dSDimitry Andric           DAG.getStore(Chain, DL, ArgValue, Address, MachinePointerInfo()));
4201753f127fSDimitry Andric     }
4202753f127fSDimitry Andric   }
4203753f127fSDimitry Andric 
4204bdd1243dSDimitry Andric   // Join the stores, which are independent of one another.
4205bdd1243dSDimitry Andric   if (!MemOpChains.empty())
4206bdd1243dSDimitry Andric     Chain = DAG.getNode(ISD::TokenFactor, DL, MVT::Other, MemOpChains);
4207bdd1243dSDimitry Andric 
4208753f127fSDimitry Andric   SDValue Glue;
4209753f127fSDimitry Andric 
4210753f127fSDimitry Andric   // Build a sequence of copy-to-reg nodes, chained and glued together.
4211753f127fSDimitry Andric   for (auto &Reg : RegsToPass) {
4212753f127fSDimitry Andric     Chain = DAG.getCopyToReg(Chain, DL, Reg.first, Reg.second, Glue);
4213753f127fSDimitry Andric     Glue = Chain.getValue(1);
4214753f127fSDimitry Andric   }
4215753f127fSDimitry Andric 
4216753f127fSDimitry Andric   // If the callee is a GlobalAddress/ExternalSymbol node, turn it into a
4217753f127fSDimitry Andric   // TargetGlobalAddress/TargetExternalSymbol node so that legalize won't
4218753f127fSDimitry Andric   // split it and then direct call can be matched by PseudoCALL.
4219bdd1243dSDimitry Andric   if (GlobalAddressSDNode *S = dyn_cast<GlobalAddressSDNode>(Callee)) {
4220bdd1243dSDimitry Andric     const GlobalValue *GV = S->getGlobal();
4221bdd1243dSDimitry Andric     unsigned OpFlags =
4222bdd1243dSDimitry Andric         getTargetMachine().shouldAssumeDSOLocal(*GV->getParent(), GV)
4223bdd1243dSDimitry Andric             ? LoongArchII::MO_CALL
4224bdd1243dSDimitry Andric             : LoongArchII::MO_CALL_PLT;
4225bdd1243dSDimitry Andric     Callee = DAG.getTargetGlobalAddress(S->getGlobal(), DL, PtrVT, 0, OpFlags);
4226bdd1243dSDimitry Andric   } else if (ExternalSymbolSDNode *S = dyn_cast<ExternalSymbolSDNode>(Callee)) {
4227bdd1243dSDimitry Andric     unsigned OpFlags = getTargetMachine().shouldAssumeDSOLocal(
4228bdd1243dSDimitry Andric                            *MF.getFunction().getParent(), nullptr)
4229bdd1243dSDimitry Andric                            ? LoongArchII::MO_CALL
4230bdd1243dSDimitry Andric                            : LoongArchII::MO_CALL_PLT;
4231bdd1243dSDimitry Andric     Callee = DAG.getTargetExternalSymbol(S->getSymbol(), PtrVT, OpFlags);
4232bdd1243dSDimitry Andric   }
4233753f127fSDimitry Andric 
4234753f127fSDimitry Andric   // The first call operand is the chain and the second is the target address.
4235753f127fSDimitry Andric   SmallVector<SDValue> Ops;
4236753f127fSDimitry Andric   Ops.push_back(Chain);
4237753f127fSDimitry Andric   Ops.push_back(Callee);
4238753f127fSDimitry Andric 
4239753f127fSDimitry Andric   // Add argument registers to the end of the list so that they are
4240753f127fSDimitry Andric   // known live into the call.
4241753f127fSDimitry Andric   for (auto &Reg : RegsToPass)
4242753f127fSDimitry Andric     Ops.push_back(DAG.getRegister(Reg.first, Reg.second.getValueType()));
4243753f127fSDimitry Andric 
4244bdd1243dSDimitry Andric   if (!IsTailCall) {
4245753f127fSDimitry Andric     // Add a register mask operand representing the call-preserved registers.
4246753f127fSDimitry Andric     const TargetRegisterInfo *TRI = Subtarget.getRegisterInfo();
4247753f127fSDimitry Andric     const uint32_t *Mask = TRI->getCallPreservedMask(MF, CallConv);
4248753f127fSDimitry Andric     assert(Mask && "Missing call preserved mask for calling convention");
4249753f127fSDimitry Andric     Ops.push_back(DAG.getRegisterMask(Mask));
4250bdd1243dSDimitry Andric   }
4251753f127fSDimitry Andric 
4252753f127fSDimitry Andric   // Glue the call to the argument copies, if any.
4253753f127fSDimitry Andric   if (Glue.getNode())
4254753f127fSDimitry Andric     Ops.push_back(Glue);
4255753f127fSDimitry Andric 
4256753f127fSDimitry Andric   // Emit the call.
4257753f127fSDimitry Andric   SDVTList NodeTys = DAG.getVTList(MVT::Other, MVT::Glue);
4258*1db9f3b2SDimitry Andric   unsigned Op;
4259*1db9f3b2SDimitry Andric   switch (DAG.getTarget().getCodeModel()) {
4260*1db9f3b2SDimitry Andric   default:
4261*1db9f3b2SDimitry Andric     report_fatal_error("Unsupported code model");
4262*1db9f3b2SDimitry Andric   case CodeModel::Small:
4263*1db9f3b2SDimitry Andric     Op = IsTailCall ? LoongArchISD::TAIL : LoongArchISD::CALL;
4264*1db9f3b2SDimitry Andric     break;
4265*1db9f3b2SDimitry Andric   case CodeModel::Medium:
4266*1db9f3b2SDimitry Andric     assert(Subtarget.is64Bit() && "Medium code model requires LA64");
4267*1db9f3b2SDimitry Andric     Op = IsTailCall ? LoongArchISD::TAIL_MEDIUM : LoongArchISD::CALL_MEDIUM;
4268*1db9f3b2SDimitry Andric     break;
4269*1db9f3b2SDimitry Andric   case CodeModel::Large:
4270*1db9f3b2SDimitry Andric     assert(Subtarget.is64Bit() && "Large code model requires LA64");
4271*1db9f3b2SDimitry Andric     Op = IsTailCall ? LoongArchISD::TAIL_LARGE : LoongArchISD::CALL_LARGE;
4272*1db9f3b2SDimitry Andric     break;
4273*1db9f3b2SDimitry Andric   }
4274753f127fSDimitry Andric 
4275bdd1243dSDimitry Andric   if (IsTailCall) {
4276bdd1243dSDimitry Andric     MF.getFrameInfo().setHasTailCall();
4277*1db9f3b2SDimitry Andric     SDValue Ret = DAG.getNode(Op, DL, NodeTys, Ops);
427806c3fb27SDimitry Andric     DAG.addNoMergeSiteInfo(Ret.getNode(), CLI.NoMerge);
427906c3fb27SDimitry Andric     return Ret;
4280bdd1243dSDimitry Andric   }
4281bdd1243dSDimitry Andric 
4282*1db9f3b2SDimitry Andric   Chain = DAG.getNode(Op, DL, NodeTys, Ops);
4283753f127fSDimitry Andric   DAG.addNoMergeSiteInfo(Chain.getNode(), CLI.NoMerge);
4284753f127fSDimitry Andric   Glue = Chain.getValue(1);
4285753f127fSDimitry Andric 
4286753f127fSDimitry Andric   // Mark the end of the call, which is glued to the call itself.
4287bdd1243dSDimitry Andric   Chain = DAG.getCALLSEQ_END(Chain, NumBytes, 0, Glue, DL);
4288753f127fSDimitry Andric   Glue = Chain.getValue(1);
4289753f127fSDimitry Andric 
4290753f127fSDimitry Andric   // Assign locations to each value returned by this call.
4291753f127fSDimitry Andric   SmallVector<CCValAssign> RVLocs;
4292753f127fSDimitry Andric   CCState RetCCInfo(CallConv, IsVarArg, MF, RVLocs, *DAG.getContext());
4293bdd1243dSDimitry Andric   analyzeInputArgs(MF, RetCCInfo, Ins, /*IsRet=*/true, CC_LoongArch);
4294753f127fSDimitry Andric 
4295753f127fSDimitry Andric   // Copy all of the result registers out of their specified physreg.
4296753f127fSDimitry Andric   for (auto &VA : RVLocs) {
4297753f127fSDimitry Andric     // Copy the value out.
4298753f127fSDimitry Andric     SDValue RetValue =
4299753f127fSDimitry Andric         DAG.getCopyFromReg(Chain, DL, VA.getLocReg(), VA.getLocVT(), Glue);
4300bdd1243dSDimitry Andric     // Glue the RetValue to the end of the call sequence.
4301753f127fSDimitry Andric     Chain = RetValue.getValue(1);
4302753f127fSDimitry Andric     Glue = RetValue.getValue(2);
4303753f127fSDimitry Andric 
4304bdd1243dSDimitry Andric     RetValue = convertLocVTToValVT(DAG, RetValue, VA, DL);
4305bdd1243dSDimitry Andric 
4306bdd1243dSDimitry Andric     InVals.push_back(RetValue);
4307753f127fSDimitry Andric   }
4308753f127fSDimitry Andric 
4309753f127fSDimitry Andric   return Chain;
4310753f127fSDimitry Andric }
4311753f127fSDimitry Andric 
431281ad6265SDimitry Andric bool LoongArchTargetLowering::CanLowerReturn(
431381ad6265SDimitry Andric     CallingConv::ID CallConv, MachineFunction &MF, bool IsVarArg,
431481ad6265SDimitry Andric     const SmallVectorImpl<ISD::OutputArg> &Outs, LLVMContext &Context) const {
4315bdd1243dSDimitry Andric   SmallVector<CCValAssign> RVLocs;
4316bdd1243dSDimitry Andric   CCState CCInfo(CallConv, IsVarArg, MF, RVLocs, Context);
4317bdd1243dSDimitry Andric 
4318bdd1243dSDimitry Andric   for (unsigned i = 0, e = Outs.size(); i != e; ++i) {
4319bdd1243dSDimitry Andric     LoongArchABI::ABI ABI =
4320bdd1243dSDimitry Andric         MF.getSubtarget<LoongArchSubtarget>().getTargetABI();
4321bdd1243dSDimitry Andric     if (CC_LoongArch(MF.getDataLayout(), ABI, i, Outs[i].VT, CCValAssign::Full,
4322bdd1243dSDimitry Andric                      Outs[i].Flags, CCInfo, /*IsFixed=*/true, /*IsRet=*/true,
4323bdd1243dSDimitry Andric                      nullptr))
4324bdd1243dSDimitry Andric       return false;
4325bdd1243dSDimitry Andric   }
4326bdd1243dSDimitry Andric   return true;
432781ad6265SDimitry Andric }
432881ad6265SDimitry Andric 
432981ad6265SDimitry Andric SDValue LoongArchTargetLowering::LowerReturn(
433081ad6265SDimitry Andric     SDValue Chain, CallingConv::ID CallConv, bool IsVarArg,
433181ad6265SDimitry Andric     const SmallVectorImpl<ISD::OutputArg> &Outs,
433281ad6265SDimitry Andric     const SmallVectorImpl<SDValue> &OutVals, const SDLoc &DL,
433381ad6265SDimitry Andric     SelectionDAG &DAG) const {
433481ad6265SDimitry Andric   // Stores the assignment of the return value to a location.
433581ad6265SDimitry Andric   SmallVector<CCValAssign> RVLocs;
433681ad6265SDimitry Andric 
433781ad6265SDimitry Andric   // Info about the registers and stack slot.
433881ad6265SDimitry Andric   CCState CCInfo(CallConv, IsVarArg, DAG.getMachineFunction(), RVLocs,
433981ad6265SDimitry Andric                  *DAG.getContext());
434081ad6265SDimitry Andric 
4341bdd1243dSDimitry Andric   analyzeOutputArgs(DAG.getMachineFunction(), CCInfo, Outs, /*IsRet=*/true,
4342bdd1243dSDimitry Andric                     nullptr, CC_LoongArch);
4343bdd1243dSDimitry Andric   if (CallConv == CallingConv::GHC && !RVLocs.empty())
4344bdd1243dSDimitry Andric     report_fatal_error("GHC functions return void only");
434581ad6265SDimitry Andric   SDValue Glue;
434681ad6265SDimitry Andric   SmallVector<SDValue, 4> RetOps(1, Chain);
434781ad6265SDimitry Andric 
434881ad6265SDimitry Andric   // Copy the result values into the output registers.
434981ad6265SDimitry Andric   for (unsigned i = 0, e = RVLocs.size(); i < e; ++i) {
435081ad6265SDimitry Andric     CCValAssign &VA = RVLocs[i];
435181ad6265SDimitry Andric     assert(VA.isRegLoc() && "Can only return in registers!");
435281ad6265SDimitry Andric 
435381ad6265SDimitry Andric     // Handle a 'normal' return.
4354bdd1243dSDimitry Andric     SDValue Val = convertValVTToLocVT(DAG, OutVals[i], VA, DL);
4355bdd1243dSDimitry Andric     Chain = DAG.getCopyToReg(Chain, DL, VA.getLocReg(), Val, Glue);
435681ad6265SDimitry Andric 
435781ad6265SDimitry Andric     // Guarantee that all emitted copies are stuck together.
435881ad6265SDimitry Andric     Glue = Chain.getValue(1);
435981ad6265SDimitry Andric     RetOps.push_back(DAG.getRegister(VA.getLocReg(), VA.getLocVT()));
436081ad6265SDimitry Andric   }
436181ad6265SDimitry Andric 
436281ad6265SDimitry Andric   RetOps[0] = Chain; // Update chain.
436381ad6265SDimitry Andric 
436481ad6265SDimitry Andric   // Add the glue node if we have it.
436581ad6265SDimitry Andric   if (Glue.getNode())
436681ad6265SDimitry Andric     RetOps.push_back(Glue);
436781ad6265SDimitry Andric 
436881ad6265SDimitry Andric   return DAG.getNode(LoongArchISD::RET, DL, MVT::Other, RetOps);
436981ad6265SDimitry Andric }
4370753f127fSDimitry Andric 
4371753f127fSDimitry Andric bool LoongArchTargetLowering::isFPImmLegal(const APFloat &Imm, EVT VT,
4372753f127fSDimitry Andric                                            bool ForCodeSize) const {
4373bdd1243dSDimitry Andric   // TODO: Maybe need more checks here after vector extension is supported.
4374753f127fSDimitry Andric   if (VT == MVT::f32 && !Subtarget.hasBasicF())
4375753f127fSDimitry Andric     return false;
4376753f127fSDimitry Andric   if (VT == MVT::f64 && !Subtarget.hasBasicD())
4377753f127fSDimitry Andric     return false;
4378753f127fSDimitry Andric   return (Imm.isZero() || Imm.isExactlyValue(+1.0));
4379753f127fSDimitry Andric }
4380bdd1243dSDimitry Andric 
4381bdd1243dSDimitry Andric bool LoongArchTargetLowering::isCheapToSpeculateCttz(Type *) const {
4382bdd1243dSDimitry Andric   return true;
4383bdd1243dSDimitry Andric }
4384bdd1243dSDimitry Andric 
4385bdd1243dSDimitry Andric bool LoongArchTargetLowering::isCheapToSpeculateCtlz(Type *) const {
4386bdd1243dSDimitry Andric   return true;
4387bdd1243dSDimitry Andric }
4388bdd1243dSDimitry Andric 
4389bdd1243dSDimitry Andric bool LoongArchTargetLowering::shouldInsertFencesForAtomic(
4390bdd1243dSDimitry Andric     const Instruction *I) const {
4391bdd1243dSDimitry Andric   if (!Subtarget.is64Bit())
4392bdd1243dSDimitry Andric     return isa<LoadInst>(I) || isa<StoreInst>(I);
4393bdd1243dSDimitry Andric 
4394bdd1243dSDimitry Andric   if (isa<LoadInst>(I))
4395bdd1243dSDimitry Andric     return true;
4396bdd1243dSDimitry Andric 
4397bdd1243dSDimitry Andric   // On LA64, atomic store operations with IntegerBitWidth of 32 and 64 do not
4398bdd1243dSDimitry Andric   // require fences beacuse we can use amswap_db.[w/d].
4399bdd1243dSDimitry Andric   if (isa<StoreInst>(I)) {
4400bdd1243dSDimitry Andric     unsigned Size = I->getOperand(0)->getType()->getIntegerBitWidth();
4401bdd1243dSDimitry Andric     return (Size == 8 || Size == 16);
4402bdd1243dSDimitry Andric   }
4403bdd1243dSDimitry Andric 
4404bdd1243dSDimitry Andric   return false;
4405bdd1243dSDimitry Andric }
4406bdd1243dSDimitry Andric 
4407bdd1243dSDimitry Andric EVT LoongArchTargetLowering::getSetCCResultType(const DataLayout &DL,
4408bdd1243dSDimitry Andric                                                 LLVMContext &Context,
4409bdd1243dSDimitry Andric                                                 EVT VT) const {
4410bdd1243dSDimitry Andric   if (!VT.isVector())
4411bdd1243dSDimitry Andric     return getPointerTy(DL);
4412bdd1243dSDimitry Andric   return VT.changeVectorElementTypeToInteger();
4413bdd1243dSDimitry Andric }
4414bdd1243dSDimitry Andric 
4415bdd1243dSDimitry Andric bool LoongArchTargetLowering::hasAndNot(SDValue Y) const {
4416bdd1243dSDimitry Andric   // TODO: Support vectors.
4417bdd1243dSDimitry Andric   return Y.getValueType().isScalarInteger() && !isa<ConstantSDNode>(Y);
4418bdd1243dSDimitry Andric }
4419bdd1243dSDimitry Andric 
4420bdd1243dSDimitry Andric bool LoongArchTargetLowering::getTgtMemIntrinsic(IntrinsicInfo &Info,
4421bdd1243dSDimitry Andric                                                  const CallInst &I,
4422bdd1243dSDimitry Andric                                                  MachineFunction &MF,
4423bdd1243dSDimitry Andric                                                  unsigned Intrinsic) const {
4424bdd1243dSDimitry Andric   switch (Intrinsic) {
4425bdd1243dSDimitry Andric   default:
4426bdd1243dSDimitry Andric     return false;
4427bdd1243dSDimitry Andric   case Intrinsic::loongarch_masked_atomicrmw_xchg_i32:
4428bdd1243dSDimitry Andric   case Intrinsic::loongarch_masked_atomicrmw_add_i32:
4429bdd1243dSDimitry Andric   case Intrinsic::loongarch_masked_atomicrmw_sub_i32:
4430bdd1243dSDimitry Andric   case Intrinsic::loongarch_masked_atomicrmw_nand_i32:
4431bdd1243dSDimitry Andric     Info.opc = ISD::INTRINSIC_W_CHAIN;
4432bdd1243dSDimitry Andric     Info.memVT = MVT::i32;
4433bdd1243dSDimitry Andric     Info.ptrVal = I.getArgOperand(0);
4434bdd1243dSDimitry Andric     Info.offset = 0;
4435bdd1243dSDimitry Andric     Info.align = Align(4);
4436bdd1243dSDimitry Andric     Info.flags = MachineMemOperand::MOLoad | MachineMemOperand::MOStore |
4437bdd1243dSDimitry Andric                  MachineMemOperand::MOVolatile;
4438bdd1243dSDimitry Andric     return true;
4439bdd1243dSDimitry Andric     // TODO: Add more Intrinsics later.
4440bdd1243dSDimitry Andric   }
4441bdd1243dSDimitry Andric }
4442bdd1243dSDimitry Andric 
4443bdd1243dSDimitry Andric TargetLowering::AtomicExpansionKind
4444bdd1243dSDimitry Andric LoongArchTargetLowering::shouldExpandAtomicRMWInIR(AtomicRMWInst *AI) const {
4445bdd1243dSDimitry Andric   // TODO: Add more AtomicRMWInst that needs to be extended.
4446bdd1243dSDimitry Andric 
4447bdd1243dSDimitry Andric   // Since floating-point operation requires a non-trivial set of data
4448bdd1243dSDimitry Andric   // operations, use CmpXChg to expand.
4449bdd1243dSDimitry Andric   if (AI->isFloatingPointOperation() ||
4450bdd1243dSDimitry Andric       AI->getOperation() == AtomicRMWInst::UIncWrap ||
4451bdd1243dSDimitry Andric       AI->getOperation() == AtomicRMWInst::UDecWrap)
4452bdd1243dSDimitry Andric     return AtomicExpansionKind::CmpXChg;
4453bdd1243dSDimitry Andric 
4454bdd1243dSDimitry Andric   unsigned Size = AI->getType()->getPrimitiveSizeInBits();
4455bdd1243dSDimitry Andric   if (Size == 8 || Size == 16)
4456bdd1243dSDimitry Andric     return AtomicExpansionKind::MaskedIntrinsic;
4457bdd1243dSDimitry Andric   return AtomicExpansionKind::None;
4458bdd1243dSDimitry Andric }
4459bdd1243dSDimitry Andric 
4460bdd1243dSDimitry Andric static Intrinsic::ID
4461bdd1243dSDimitry Andric getIntrinsicForMaskedAtomicRMWBinOp(unsigned GRLen,
4462bdd1243dSDimitry Andric                                     AtomicRMWInst::BinOp BinOp) {
4463bdd1243dSDimitry Andric   if (GRLen == 64) {
4464bdd1243dSDimitry Andric     switch (BinOp) {
4465bdd1243dSDimitry Andric     default:
4466bdd1243dSDimitry Andric       llvm_unreachable("Unexpected AtomicRMW BinOp");
4467bdd1243dSDimitry Andric     case AtomicRMWInst::Xchg:
4468bdd1243dSDimitry Andric       return Intrinsic::loongarch_masked_atomicrmw_xchg_i64;
4469bdd1243dSDimitry Andric     case AtomicRMWInst::Add:
4470bdd1243dSDimitry Andric       return Intrinsic::loongarch_masked_atomicrmw_add_i64;
4471bdd1243dSDimitry Andric     case AtomicRMWInst::Sub:
4472bdd1243dSDimitry Andric       return Intrinsic::loongarch_masked_atomicrmw_sub_i64;
4473bdd1243dSDimitry Andric     case AtomicRMWInst::Nand:
4474bdd1243dSDimitry Andric       return Intrinsic::loongarch_masked_atomicrmw_nand_i64;
4475bdd1243dSDimitry Andric     case AtomicRMWInst::UMax:
4476bdd1243dSDimitry Andric       return Intrinsic::loongarch_masked_atomicrmw_umax_i64;
4477bdd1243dSDimitry Andric     case AtomicRMWInst::UMin:
4478bdd1243dSDimitry Andric       return Intrinsic::loongarch_masked_atomicrmw_umin_i64;
4479bdd1243dSDimitry Andric     case AtomicRMWInst::Max:
4480bdd1243dSDimitry Andric       return Intrinsic::loongarch_masked_atomicrmw_max_i64;
4481bdd1243dSDimitry Andric     case AtomicRMWInst::Min:
4482bdd1243dSDimitry Andric       return Intrinsic::loongarch_masked_atomicrmw_min_i64;
4483bdd1243dSDimitry Andric       // TODO: support other AtomicRMWInst.
4484bdd1243dSDimitry Andric     }
4485bdd1243dSDimitry Andric   }
4486bdd1243dSDimitry Andric 
4487bdd1243dSDimitry Andric   if (GRLen == 32) {
4488bdd1243dSDimitry Andric     switch (BinOp) {
4489bdd1243dSDimitry Andric     default:
4490bdd1243dSDimitry Andric       llvm_unreachable("Unexpected AtomicRMW BinOp");
4491bdd1243dSDimitry Andric     case AtomicRMWInst::Xchg:
4492bdd1243dSDimitry Andric       return Intrinsic::loongarch_masked_atomicrmw_xchg_i32;
4493bdd1243dSDimitry Andric     case AtomicRMWInst::Add:
4494bdd1243dSDimitry Andric       return Intrinsic::loongarch_masked_atomicrmw_add_i32;
4495bdd1243dSDimitry Andric     case AtomicRMWInst::Sub:
4496bdd1243dSDimitry Andric       return Intrinsic::loongarch_masked_atomicrmw_sub_i32;
4497bdd1243dSDimitry Andric     case AtomicRMWInst::Nand:
4498bdd1243dSDimitry Andric       return Intrinsic::loongarch_masked_atomicrmw_nand_i32;
4499bdd1243dSDimitry Andric       // TODO: support other AtomicRMWInst.
4500bdd1243dSDimitry Andric     }
4501bdd1243dSDimitry Andric   }
4502bdd1243dSDimitry Andric 
4503bdd1243dSDimitry Andric   llvm_unreachable("Unexpected GRLen\n");
4504bdd1243dSDimitry Andric }
4505bdd1243dSDimitry Andric 
4506bdd1243dSDimitry Andric TargetLowering::AtomicExpansionKind
4507bdd1243dSDimitry Andric LoongArchTargetLowering::shouldExpandAtomicCmpXchgInIR(
4508bdd1243dSDimitry Andric     AtomicCmpXchgInst *CI) const {
4509bdd1243dSDimitry Andric   unsigned Size = CI->getCompareOperand()->getType()->getPrimitiveSizeInBits();
4510bdd1243dSDimitry Andric   if (Size == 8 || Size == 16)
4511bdd1243dSDimitry Andric     return AtomicExpansionKind::MaskedIntrinsic;
4512bdd1243dSDimitry Andric   return AtomicExpansionKind::None;
4513bdd1243dSDimitry Andric }
4514bdd1243dSDimitry Andric 
4515bdd1243dSDimitry Andric Value *LoongArchTargetLowering::emitMaskedAtomicCmpXchgIntrinsic(
4516bdd1243dSDimitry Andric     IRBuilderBase &Builder, AtomicCmpXchgInst *CI, Value *AlignedAddr,
4517bdd1243dSDimitry Andric     Value *CmpVal, Value *NewVal, Value *Mask, AtomicOrdering Ord) const {
45185f757f3fSDimitry Andric   AtomicOrdering FailOrd = CI->getFailureOrdering();
45195f757f3fSDimitry Andric   Value *FailureOrdering =
45205f757f3fSDimitry Andric       Builder.getIntN(Subtarget.getGRLen(), static_cast<uint64_t>(FailOrd));
4521bdd1243dSDimitry Andric 
4522bdd1243dSDimitry Andric   // TODO: Support cmpxchg on LA32.
4523bdd1243dSDimitry Andric   Intrinsic::ID CmpXchgIntrID = Intrinsic::loongarch_masked_cmpxchg_i64;
4524bdd1243dSDimitry Andric   CmpVal = Builder.CreateSExt(CmpVal, Builder.getInt64Ty());
4525bdd1243dSDimitry Andric   NewVal = Builder.CreateSExt(NewVal, Builder.getInt64Ty());
4526bdd1243dSDimitry Andric   Mask = Builder.CreateSExt(Mask, Builder.getInt64Ty());
4527bdd1243dSDimitry Andric   Type *Tys[] = {AlignedAddr->getType()};
4528bdd1243dSDimitry Andric   Function *MaskedCmpXchg =
4529bdd1243dSDimitry Andric       Intrinsic::getDeclaration(CI->getModule(), CmpXchgIntrID, Tys);
4530bdd1243dSDimitry Andric   Value *Result = Builder.CreateCall(
45315f757f3fSDimitry Andric       MaskedCmpXchg, {AlignedAddr, CmpVal, NewVal, Mask, FailureOrdering});
4532bdd1243dSDimitry Andric   Result = Builder.CreateTrunc(Result, Builder.getInt32Ty());
4533bdd1243dSDimitry Andric   return Result;
4534bdd1243dSDimitry Andric }
4535bdd1243dSDimitry Andric 
4536bdd1243dSDimitry Andric Value *LoongArchTargetLowering::emitMaskedAtomicRMWIntrinsic(
4537bdd1243dSDimitry Andric     IRBuilderBase &Builder, AtomicRMWInst *AI, Value *AlignedAddr, Value *Incr,
4538bdd1243dSDimitry Andric     Value *Mask, Value *ShiftAmt, AtomicOrdering Ord) const {
45395f757f3fSDimitry Andric   // In the case of an atomicrmw xchg with a constant 0/-1 operand, replace
45405f757f3fSDimitry Andric   // the atomic instruction with an AtomicRMWInst::And/Or with appropriate
45415f757f3fSDimitry Andric   // mask, as this produces better code than the LL/SC loop emitted by
45425f757f3fSDimitry Andric   // int_loongarch_masked_atomicrmw_xchg.
45435f757f3fSDimitry Andric   if (AI->getOperation() == AtomicRMWInst::Xchg &&
45445f757f3fSDimitry Andric       isa<ConstantInt>(AI->getValOperand())) {
45455f757f3fSDimitry Andric     ConstantInt *CVal = cast<ConstantInt>(AI->getValOperand());
45465f757f3fSDimitry Andric     if (CVal->isZero())
45475f757f3fSDimitry Andric       return Builder.CreateAtomicRMW(AtomicRMWInst::And, AlignedAddr,
45485f757f3fSDimitry Andric                                      Builder.CreateNot(Mask, "Inv_Mask"),
45495f757f3fSDimitry Andric                                      AI->getAlign(), Ord);
45505f757f3fSDimitry Andric     if (CVal->isMinusOne())
45515f757f3fSDimitry Andric       return Builder.CreateAtomicRMW(AtomicRMWInst::Or, AlignedAddr, Mask,
45525f757f3fSDimitry Andric                                      AI->getAlign(), Ord);
45535f757f3fSDimitry Andric   }
45545f757f3fSDimitry Andric 
4555bdd1243dSDimitry Andric   unsigned GRLen = Subtarget.getGRLen();
4556bdd1243dSDimitry Andric   Value *Ordering =
4557bdd1243dSDimitry Andric       Builder.getIntN(GRLen, static_cast<uint64_t>(AI->getOrdering()));
4558bdd1243dSDimitry Andric   Type *Tys[] = {AlignedAddr->getType()};
4559bdd1243dSDimitry Andric   Function *LlwOpScwLoop = Intrinsic::getDeclaration(
4560bdd1243dSDimitry Andric       AI->getModule(),
4561bdd1243dSDimitry Andric       getIntrinsicForMaskedAtomicRMWBinOp(GRLen, AI->getOperation()), Tys);
4562bdd1243dSDimitry Andric 
4563bdd1243dSDimitry Andric   if (GRLen == 64) {
4564bdd1243dSDimitry Andric     Incr = Builder.CreateSExt(Incr, Builder.getInt64Ty());
4565bdd1243dSDimitry Andric     Mask = Builder.CreateSExt(Mask, Builder.getInt64Ty());
4566bdd1243dSDimitry Andric     ShiftAmt = Builder.CreateSExt(ShiftAmt, Builder.getInt64Ty());
4567bdd1243dSDimitry Andric   }
4568bdd1243dSDimitry Andric 
4569bdd1243dSDimitry Andric   Value *Result;
4570bdd1243dSDimitry Andric 
4571bdd1243dSDimitry Andric   // Must pass the shift amount needed to sign extend the loaded value prior
4572bdd1243dSDimitry Andric   // to performing a signed comparison for min/max. ShiftAmt is the number of
4573bdd1243dSDimitry Andric   // bits to shift the value into position. Pass GRLen-ShiftAmt-ValWidth, which
4574bdd1243dSDimitry Andric   // is the number of bits to left+right shift the value in order to
4575bdd1243dSDimitry Andric   // sign-extend.
4576bdd1243dSDimitry Andric   if (AI->getOperation() == AtomicRMWInst::Min ||
4577bdd1243dSDimitry Andric       AI->getOperation() == AtomicRMWInst::Max) {
4578bdd1243dSDimitry Andric     const DataLayout &DL = AI->getModule()->getDataLayout();
4579bdd1243dSDimitry Andric     unsigned ValWidth =
4580bdd1243dSDimitry Andric         DL.getTypeStoreSizeInBits(AI->getValOperand()->getType());
4581bdd1243dSDimitry Andric     Value *SextShamt =
4582bdd1243dSDimitry Andric         Builder.CreateSub(Builder.getIntN(GRLen, GRLen - ValWidth), ShiftAmt);
4583bdd1243dSDimitry Andric     Result = Builder.CreateCall(LlwOpScwLoop,
4584bdd1243dSDimitry Andric                                 {AlignedAddr, Incr, Mask, SextShamt, Ordering});
4585bdd1243dSDimitry Andric   } else {
4586bdd1243dSDimitry Andric     Result =
4587bdd1243dSDimitry Andric         Builder.CreateCall(LlwOpScwLoop, {AlignedAddr, Incr, Mask, Ordering});
4588bdd1243dSDimitry Andric   }
4589bdd1243dSDimitry Andric 
4590bdd1243dSDimitry Andric   if (GRLen == 64)
4591bdd1243dSDimitry Andric     Result = Builder.CreateTrunc(Result, Builder.getInt32Ty());
4592bdd1243dSDimitry Andric   return Result;
4593bdd1243dSDimitry Andric }
4594bdd1243dSDimitry Andric 
4595bdd1243dSDimitry Andric bool LoongArchTargetLowering::isFMAFasterThanFMulAndFAdd(
4596bdd1243dSDimitry Andric     const MachineFunction &MF, EVT VT) const {
4597bdd1243dSDimitry Andric   VT = VT.getScalarType();
4598bdd1243dSDimitry Andric 
4599bdd1243dSDimitry Andric   if (!VT.isSimple())
4600bdd1243dSDimitry Andric     return false;
4601bdd1243dSDimitry Andric 
4602bdd1243dSDimitry Andric   switch (VT.getSimpleVT().SimpleTy) {
4603bdd1243dSDimitry Andric   case MVT::f32:
4604bdd1243dSDimitry Andric   case MVT::f64:
4605bdd1243dSDimitry Andric     return true;
4606bdd1243dSDimitry Andric   default:
4607bdd1243dSDimitry Andric     break;
4608bdd1243dSDimitry Andric   }
4609bdd1243dSDimitry Andric 
4610bdd1243dSDimitry Andric   return false;
4611bdd1243dSDimitry Andric }
4612bdd1243dSDimitry Andric 
4613bdd1243dSDimitry Andric Register LoongArchTargetLowering::getExceptionPointerRegister(
4614bdd1243dSDimitry Andric     const Constant *PersonalityFn) const {
4615bdd1243dSDimitry Andric   return LoongArch::R4;
4616bdd1243dSDimitry Andric }
4617bdd1243dSDimitry Andric 
4618bdd1243dSDimitry Andric Register LoongArchTargetLowering::getExceptionSelectorRegister(
4619bdd1243dSDimitry Andric     const Constant *PersonalityFn) const {
4620bdd1243dSDimitry Andric   return LoongArch::R5;
4621bdd1243dSDimitry Andric }
4622bdd1243dSDimitry Andric 
4623bdd1243dSDimitry Andric //===----------------------------------------------------------------------===//
4624bdd1243dSDimitry Andric //                           LoongArch Inline Assembly Support
4625bdd1243dSDimitry Andric //===----------------------------------------------------------------------===//
4626bdd1243dSDimitry Andric 
4627bdd1243dSDimitry Andric LoongArchTargetLowering::ConstraintType
4628bdd1243dSDimitry Andric LoongArchTargetLowering::getConstraintType(StringRef Constraint) const {
4629bdd1243dSDimitry Andric   // LoongArch specific constraints in GCC: config/loongarch/constraints.md
4630bdd1243dSDimitry Andric   //
4631bdd1243dSDimitry Andric   // 'f':  A floating-point register (if available).
4632bdd1243dSDimitry Andric   // 'k':  A memory operand whose address is formed by a base register and
4633bdd1243dSDimitry Andric   //       (optionally scaled) index register.
4634bdd1243dSDimitry Andric   // 'l':  A signed 16-bit constant.
4635bdd1243dSDimitry Andric   // 'm':  A memory operand whose address is formed by a base register and
4636bdd1243dSDimitry Andric   //       offset that is suitable for use in instructions with the same
4637bdd1243dSDimitry Andric   //       addressing mode as st.w and ld.w.
4638bdd1243dSDimitry Andric   // 'I':  A signed 12-bit constant (for arithmetic instructions).
4639bdd1243dSDimitry Andric   // 'J':  Integer zero.
4640bdd1243dSDimitry Andric   // 'K':  An unsigned 12-bit constant (for logic instructions).
4641bdd1243dSDimitry Andric   // "ZB": An address that is held in a general-purpose register. The offset is
4642bdd1243dSDimitry Andric   //       zero.
4643bdd1243dSDimitry Andric   // "ZC": A memory operand whose address is formed by a base register and
4644bdd1243dSDimitry Andric   //       offset that is suitable for use in instructions with the same
4645bdd1243dSDimitry Andric   //       addressing mode as ll.w and sc.w.
4646bdd1243dSDimitry Andric   if (Constraint.size() == 1) {
4647bdd1243dSDimitry Andric     switch (Constraint[0]) {
4648bdd1243dSDimitry Andric     default:
4649bdd1243dSDimitry Andric       break;
4650bdd1243dSDimitry Andric     case 'f':
4651bdd1243dSDimitry Andric       return C_RegisterClass;
4652bdd1243dSDimitry Andric     case 'l':
4653bdd1243dSDimitry Andric     case 'I':
4654bdd1243dSDimitry Andric     case 'J':
4655bdd1243dSDimitry Andric     case 'K':
4656bdd1243dSDimitry Andric       return C_Immediate;
4657bdd1243dSDimitry Andric     case 'k':
4658bdd1243dSDimitry Andric       return C_Memory;
4659bdd1243dSDimitry Andric     }
4660bdd1243dSDimitry Andric   }
4661bdd1243dSDimitry Andric 
4662bdd1243dSDimitry Andric   if (Constraint == "ZC" || Constraint == "ZB")
4663bdd1243dSDimitry Andric     return C_Memory;
4664bdd1243dSDimitry Andric 
4665bdd1243dSDimitry Andric   // 'm' is handled here.
4666bdd1243dSDimitry Andric   return TargetLowering::getConstraintType(Constraint);
4667bdd1243dSDimitry Andric }
4668bdd1243dSDimitry Andric 
46695f757f3fSDimitry Andric InlineAsm::ConstraintCode LoongArchTargetLowering::getInlineAsmMemConstraint(
4670bdd1243dSDimitry Andric     StringRef ConstraintCode) const {
46715f757f3fSDimitry Andric   return StringSwitch<InlineAsm::ConstraintCode>(ConstraintCode)
46725f757f3fSDimitry Andric       .Case("k", InlineAsm::ConstraintCode::k)
46735f757f3fSDimitry Andric       .Case("ZB", InlineAsm::ConstraintCode::ZB)
46745f757f3fSDimitry Andric       .Case("ZC", InlineAsm::ConstraintCode::ZC)
4675bdd1243dSDimitry Andric       .Default(TargetLowering::getInlineAsmMemConstraint(ConstraintCode));
4676bdd1243dSDimitry Andric }
4677bdd1243dSDimitry Andric 
4678bdd1243dSDimitry Andric std::pair<unsigned, const TargetRegisterClass *>
4679bdd1243dSDimitry Andric LoongArchTargetLowering::getRegForInlineAsmConstraint(
4680bdd1243dSDimitry Andric     const TargetRegisterInfo *TRI, StringRef Constraint, MVT VT) const {
4681bdd1243dSDimitry Andric   // First, see if this is a constraint that directly corresponds to a LoongArch
4682bdd1243dSDimitry Andric   // register class.
4683bdd1243dSDimitry Andric   if (Constraint.size() == 1) {
4684bdd1243dSDimitry Andric     switch (Constraint[0]) {
4685bdd1243dSDimitry Andric     case 'r':
4686bdd1243dSDimitry Andric       // TODO: Support fixed vectors up to GRLen?
4687bdd1243dSDimitry Andric       if (VT.isVector())
4688bdd1243dSDimitry Andric         break;
4689bdd1243dSDimitry Andric       return std::make_pair(0U, &LoongArch::GPRRegClass);
4690bdd1243dSDimitry Andric     case 'f':
4691bdd1243dSDimitry Andric       if (Subtarget.hasBasicF() && VT == MVT::f32)
4692bdd1243dSDimitry Andric         return std::make_pair(0U, &LoongArch::FPR32RegClass);
4693bdd1243dSDimitry Andric       if (Subtarget.hasBasicD() && VT == MVT::f64)
4694bdd1243dSDimitry Andric         return std::make_pair(0U, &LoongArch::FPR64RegClass);
469506c3fb27SDimitry Andric       if (Subtarget.hasExtLSX() &&
469606c3fb27SDimitry Andric           TRI->isTypeLegalForClass(LoongArch::LSX128RegClass, VT))
469706c3fb27SDimitry Andric         return std::make_pair(0U, &LoongArch::LSX128RegClass);
469806c3fb27SDimitry Andric       if (Subtarget.hasExtLASX() &&
469906c3fb27SDimitry Andric           TRI->isTypeLegalForClass(LoongArch::LASX256RegClass, VT))
470006c3fb27SDimitry Andric         return std::make_pair(0U, &LoongArch::LASX256RegClass);
4701bdd1243dSDimitry Andric       break;
4702bdd1243dSDimitry Andric     default:
4703bdd1243dSDimitry Andric       break;
4704bdd1243dSDimitry Andric     }
4705bdd1243dSDimitry Andric   }
4706bdd1243dSDimitry Andric 
4707bdd1243dSDimitry Andric   // TargetLowering::getRegForInlineAsmConstraint uses the name of the TableGen
4708bdd1243dSDimitry Andric   // record (e.g. the "R0" in `def R0`) to choose registers for InlineAsm
4709bdd1243dSDimitry Andric   // constraints while the official register name is prefixed with a '$'. So we
4710bdd1243dSDimitry Andric   // clip the '$' from the original constraint string (e.g. {$r0} to {r0}.)
4711bdd1243dSDimitry Andric   // before it being parsed. And TargetLowering::getRegForInlineAsmConstraint is
4712bdd1243dSDimitry Andric   // case insensitive, so no need to convert the constraint to upper case here.
4713bdd1243dSDimitry Andric   //
4714bdd1243dSDimitry Andric   // For now, no need to support ABI names (e.g. `$a0`) as clang will correctly
4715bdd1243dSDimitry Andric   // decode the usage of register name aliases into their official names. And
4716bdd1243dSDimitry Andric   // AFAIK, the not yet upstreamed `rustc` for LoongArch will always use
4717bdd1243dSDimitry Andric   // official register names.
47185f757f3fSDimitry Andric   if (Constraint.starts_with("{$r") || Constraint.starts_with("{$f") ||
47195f757f3fSDimitry Andric       Constraint.starts_with("{$vr") || Constraint.starts_with("{$xr")) {
4720bdd1243dSDimitry Andric     bool IsFP = Constraint[2] == 'f';
4721bdd1243dSDimitry Andric     std::pair<StringRef, StringRef> Temp = Constraint.split('$');
4722bdd1243dSDimitry Andric     std::pair<unsigned, const TargetRegisterClass *> R;
4723bdd1243dSDimitry Andric     R = TargetLowering::getRegForInlineAsmConstraint(
4724bdd1243dSDimitry Andric         TRI, join_items("", Temp.first, Temp.second), VT);
4725bdd1243dSDimitry Andric     // Match those names to the widest floating point register type available.
4726bdd1243dSDimitry Andric     if (IsFP) {
4727bdd1243dSDimitry Andric       unsigned RegNo = R.first;
4728bdd1243dSDimitry Andric       if (LoongArch::F0 <= RegNo && RegNo <= LoongArch::F31) {
4729bdd1243dSDimitry Andric         if (Subtarget.hasBasicD() && (VT == MVT::f64 || VT == MVT::Other)) {
4730bdd1243dSDimitry Andric           unsigned DReg = RegNo - LoongArch::F0 + LoongArch::F0_64;
4731bdd1243dSDimitry Andric           return std::make_pair(DReg, &LoongArch::FPR64RegClass);
4732bdd1243dSDimitry Andric         }
4733bdd1243dSDimitry Andric       }
4734bdd1243dSDimitry Andric     }
4735bdd1243dSDimitry Andric     return R;
4736bdd1243dSDimitry Andric   }
4737bdd1243dSDimitry Andric 
4738bdd1243dSDimitry Andric   return TargetLowering::getRegForInlineAsmConstraint(TRI, Constraint, VT);
4739bdd1243dSDimitry Andric }
4740bdd1243dSDimitry Andric 
4741bdd1243dSDimitry Andric void LoongArchTargetLowering::LowerAsmOperandForConstraint(
47425f757f3fSDimitry Andric     SDValue Op, StringRef Constraint, std::vector<SDValue> &Ops,
4743bdd1243dSDimitry Andric     SelectionDAG &DAG) const {
4744bdd1243dSDimitry Andric   // Currently only support length 1 constraints.
47455f757f3fSDimitry Andric   if (Constraint.size() == 1) {
4746bdd1243dSDimitry Andric     switch (Constraint[0]) {
4747bdd1243dSDimitry Andric     case 'l':
4748bdd1243dSDimitry Andric       // Validate & create a 16-bit signed immediate operand.
4749bdd1243dSDimitry Andric       if (auto *C = dyn_cast<ConstantSDNode>(Op)) {
4750bdd1243dSDimitry Andric         uint64_t CVal = C->getSExtValue();
4751bdd1243dSDimitry Andric         if (isInt<16>(CVal))
4752bdd1243dSDimitry Andric           Ops.push_back(
4753bdd1243dSDimitry Andric               DAG.getTargetConstant(CVal, SDLoc(Op), Subtarget.getGRLenVT()));
4754bdd1243dSDimitry Andric       }
4755bdd1243dSDimitry Andric       return;
4756bdd1243dSDimitry Andric     case 'I':
4757bdd1243dSDimitry Andric       // Validate & create a 12-bit signed immediate operand.
4758bdd1243dSDimitry Andric       if (auto *C = dyn_cast<ConstantSDNode>(Op)) {
4759bdd1243dSDimitry Andric         uint64_t CVal = C->getSExtValue();
4760bdd1243dSDimitry Andric         if (isInt<12>(CVal))
4761bdd1243dSDimitry Andric           Ops.push_back(
4762bdd1243dSDimitry Andric               DAG.getTargetConstant(CVal, SDLoc(Op), Subtarget.getGRLenVT()));
4763bdd1243dSDimitry Andric       }
4764bdd1243dSDimitry Andric       return;
4765bdd1243dSDimitry Andric     case 'J':
4766bdd1243dSDimitry Andric       // Validate & create an integer zero operand.
4767bdd1243dSDimitry Andric       if (auto *C = dyn_cast<ConstantSDNode>(Op))
4768bdd1243dSDimitry Andric         if (C->getZExtValue() == 0)
4769bdd1243dSDimitry Andric           Ops.push_back(
4770bdd1243dSDimitry Andric               DAG.getTargetConstant(0, SDLoc(Op), Subtarget.getGRLenVT()));
4771bdd1243dSDimitry Andric       return;
4772bdd1243dSDimitry Andric     case 'K':
4773bdd1243dSDimitry Andric       // Validate & create a 12-bit unsigned immediate operand.
4774bdd1243dSDimitry Andric       if (auto *C = dyn_cast<ConstantSDNode>(Op)) {
4775bdd1243dSDimitry Andric         uint64_t CVal = C->getZExtValue();
4776bdd1243dSDimitry Andric         if (isUInt<12>(CVal))
4777bdd1243dSDimitry Andric           Ops.push_back(
4778bdd1243dSDimitry Andric               DAG.getTargetConstant(CVal, SDLoc(Op), Subtarget.getGRLenVT()));
4779bdd1243dSDimitry Andric       }
4780bdd1243dSDimitry Andric       return;
4781bdd1243dSDimitry Andric     default:
4782bdd1243dSDimitry Andric       break;
4783bdd1243dSDimitry Andric     }
4784bdd1243dSDimitry Andric   }
4785bdd1243dSDimitry Andric   TargetLowering::LowerAsmOperandForConstraint(Op, Constraint, Ops, DAG);
4786bdd1243dSDimitry Andric }
4787bdd1243dSDimitry Andric 
4788bdd1243dSDimitry Andric #define GET_REGISTER_MATCHER
4789bdd1243dSDimitry Andric #include "LoongArchGenAsmMatcher.inc"
4790bdd1243dSDimitry Andric 
4791bdd1243dSDimitry Andric Register
4792bdd1243dSDimitry Andric LoongArchTargetLowering::getRegisterByName(const char *RegName, LLT VT,
4793bdd1243dSDimitry Andric                                            const MachineFunction &MF) const {
4794bdd1243dSDimitry Andric   std::pair<StringRef, StringRef> Name = StringRef(RegName).split('$');
4795bdd1243dSDimitry Andric   std::string NewRegName = Name.second.str();
4796bdd1243dSDimitry Andric   Register Reg = MatchRegisterAltName(NewRegName);
4797bdd1243dSDimitry Andric   if (Reg == LoongArch::NoRegister)
4798bdd1243dSDimitry Andric     Reg = MatchRegisterName(NewRegName);
4799bdd1243dSDimitry Andric   if (Reg == LoongArch::NoRegister)
4800bdd1243dSDimitry Andric     report_fatal_error(
4801bdd1243dSDimitry Andric         Twine("Invalid register name \"" + StringRef(RegName) + "\"."));
4802bdd1243dSDimitry Andric   BitVector ReservedRegs = Subtarget.getRegisterInfo()->getReservedRegs(MF);
4803bdd1243dSDimitry Andric   if (!ReservedRegs.test(Reg))
4804bdd1243dSDimitry Andric     report_fatal_error(Twine("Trying to obtain non-reserved register \"" +
4805bdd1243dSDimitry Andric                              StringRef(RegName) + "\"."));
4806bdd1243dSDimitry Andric   return Reg;
4807bdd1243dSDimitry Andric }
4808bdd1243dSDimitry Andric 
4809bdd1243dSDimitry Andric bool LoongArchTargetLowering::decomposeMulByConstant(LLVMContext &Context,
4810bdd1243dSDimitry Andric                                                      EVT VT, SDValue C) const {
4811bdd1243dSDimitry Andric   // TODO: Support vectors.
4812bdd1243dSDimitry Andric   if (!VT.isScalarInteger())
4813bdd1243dSDimitry Andric     return false;
4814bdd1243dSDimitry Andric 
4815bdd1243dSDimitry Andric   // Omit the optimization if the data size exceeds GRLen.
4816bdd1243dSDimitry Andric   if (VT.getSizeInBits() > Subtarget.getGRLen())
4817bdd1243dSDimitry Andric     return false;
4818bdd1243dSDimitry Andric 
4819bdd1243dSDimitry Andric   if (auto *ConstNode = dyn_cast<ConstantSDNode>(C.getNode())) {
4820bdd1243dSDimitry Andric     const APInt &Imm = ConstNode->getAPIntValue();
482106c3fb27SDimitry Andric     // Break MUL into (SLLI + ADD/SUB) or ALSL.
4822bdd1243dSDimitry Andric     if ((Imm + 1).isPowerOf2() || (Imm - 1).isPowerOf2() ||
4823bdd1243dSDimitry Andric         (1 - Imm).isPowerOf2() || (-1 - Imm).isPowerOf2())
4824bdd1243dSDimitry Andric       return true;
482506c3fb27SDimitry Andric     // Break MUL into (ALSL x, (SLLI x, imm0), imm1).
482606c3fb27SDimitry Andric     if (ConstNode->hasOneUse() &&
482706c3fb27SDimitry Andric         ((Imm - 2).isPowerOf2() || (Imm - 4).isPowerOf2() ||
482806c3fb27SDimitry Andric          (Imm - 8).isPowerOf2() || (Imm - 16).isPowerOf2()))
482906c3fb27SDimitry Andric       return true;
483006c3fb27SDimitry Andric     // Break (MUL x, imm) into (ADD (SLLI x, s0), (SLLI x, s1)),
483106c3fb27SDimitry Andric     // in which the immediate has two set bits. Or Break (MUL x, imm)
483206c3fb27SDimitry Andric     // into (SUB (SLLI x, s0), (SLLI x, s1)), in which the immediate
483306c3fb27SDimitry Andric     // equals to (1 << s0) - (1 << s1).
483406c3fb27SDimitry Andric     if (ConstNode->hasOneUse() && !(Imm.sge(-2048) && Imm.sle(4095))) {
483506c3fb27SDimitry Andric       unsigned Shifts = Imm.countr_zero();
483606c3fb27SDimitry Andric       // Reject immediates which can be composed via a single LUI.
483706c3fb27SDimitry Andric       if (Shifts >= 12)
483806c3fb27SDimitry Andric         return false;
483906c3fb27SDimitry Andric       // Reject multiplications can be optimized to
484006c3fb27SDimitry Andric       // (SLLI (ALSL x, x, 1/2/3/4), s).
484106c3fb27SDimitry Andric       APInt ImmPop = Imm.ashr(Shifts);
484206c3fb27SDimitry Andric       if (ImmPop == 3 || ImmPop == 5 || ImmPop == 9 || ImmPop == 17)
484306c3fb27SDimitry Andric         return false;
484406c3fb27SDimitry Andric       // We do not consider the case `(-Imm - ImmSmall).isPowerOf2()`,
484506c3fb27SDimitry Andric       // since it needs one more instruction than other 3 cases.
484606c3fb27SDimitry Andric       APInt ImmSmall = APInt(Imm.getBitWidth(), 1ULL << Shifts, true);
484706c3fb27SDimitry Andric       if ((Imm - ImmSmall).isPowerOf2() || (Imm + ImmSmall).isPowerOf2() ||
484806c3fb27SDimitry Andric           (ImmSmall - Imm).isPowerOf2())
484906c3fb27SDimitry Andric         return true;
485006c3fb27SDimitry Andric     }
4851bdd1243dSDimitry Andric   }
4852bdd1243dSDimitry Andric 
4853bdd1243dSDimitry Andric   return false;
4854bdd1243dSDimitry Andric }
485506c3fb27SDimitry Andric 
485606c3fb27SDimitry Andric bool LoongArchTargetLowering::isLegalAddressingMode(const DataLayout &DL,
485706c3fb27SDimitry Andric                                                     const AddrMode &AM,
485806c3fb27SDimitry Andric                                                     Type *Ty, unsigned AS,
485906c3fb27SDimitry Andric                                                     Instruction *I) const {
486006c3fb27SDimitry Andric   // LoongArch has four basic addressing modes:
486106c3fb27SDimitry Andric   //  1. reg
486206c3fb27SDimitry Andric   //  2. reg + 12-bit signed offset
486306c3fb27SDimitry Andric   //  3. reg + 14-bit signed offset left-shifted by 2
486406c3fb27SDimitry Andric   //  4. reg1 + reg2
486506c3fb27SDimitry Andric   // TODO: Add more checks after support vector extension.
486606c3fb27SDimitry Andric 
486706c3fb27SDimitry Andric   // No global is ever allowed as a base.
486806c3fb27SDimitry Andric   if (AM.BaseGV)
486906c3fb27SDimitry Andric     return false;
487006c3fb27SDimitry Andric 
487106c3fb27SDimitry Andric   // Require a 12 or 14 bit signed offset.
487206c3fb27SDimitry Andric   if (!isInt<12>(AM.BaseOffs) || !isShiftedInt<14, 2>(AM.BaseOffs))
487306c3fb27SDimitry Andric     return false;
487406c3fb27SDimitry Andric 
487506c3fb27SDimitry Andric   switch (AM.Scale) {
487606c3fb27SDimitry Andric   case 0:
487706c3fb27SDimitry Andric     // "i" is not allowed.
487806c3fb27SDimitry Andric     if (!AM.HasBaseReg)
487906c3fb27SDimitry Andric       return false;
488006c3fb27SDimitry Andric     // Otherwise we have "r+i".
488106c3fb27SDimitry Andric     break;
488206c3fb27SDimitry Andric   case 1:
488306c3fb27SDimitry Andric     // "r+r+i" is not allowed.
488406c3fb27SDimitry Andric     if (AM.HasBaseReg && AM.BaseOffs != 0)
488506c3fb27SDimitry Andric       return false;
488606c3fb27SDimitry Andric     // Otherwise we have "r+r" or "r+i".
488706c3fb27SDimitry Andric     break;
488806c3fb27SDimitry Andric   case 2:
488906c3fb27SDimitry Andric     // "2*r+r" or "2*r+i" is not allowed.
489006c3fb27SDimitry Andric     if (AM.HasBaseReg || AM.BaseOffs)
489106c3fb27SDimitry Andric       return false;
489206c3fb27SDimitry Andric     // Otherwise we have "r+r".
489306c3fb27SDimitry Andric     break;
489406c3fb27SDimitry Andric   default:
489506c3fb27SDimitry Andric     return false;
489606c3fb27SDimitry Andric   }
489706c3fb27SDimitry Andric 
489806c3fb27SDimitry Andric   return true;
489906c3fb27SDimitry Andric }
490006c3fb27SDimitry Andric 
490106c3fb27SDimitry Andric bool LoongArchTargetLowering::isLegalICmpImmediate(int64_t Imm) const {
490206c3fb27SDimitry Andric   return isInt<12>(Imm);
490306c3fb27SDimitry Andric }
490406c3fb27SDimitry Andric 
490506c3fb27SDimitry Andric bool LoongArchTargetLowering::isLegalAddImmediate(int64_t Imm) const {
490606c3fb27SDimitry Andric   return isInt<12>(Imm);
490706c3fb27SDimitry Andric }
490806c3fb27SDimitry Andric 
490906c3fb27SDimitry Andric bool LoongArchTargetLowering::isZExtFree(SDValue Val, EVT VT2) const {
491006c3fb27SDimitry Andric   // Zexts are free if they can be combined with a load.
491106c3fb27SDimitry Andric   // Don't advertise i32->i64 zextload as being free for LA64. It interacts
491206c3fb27SDimitry Andric   // poorly with type legalization of compares preferring sext.
491306c3fb27SDimitry Andric   if (auto *LD = dyn_cast<LoadSDNode>(Val)) {
491406c3fb27SDimitry Andric     EVT MemVT = LD->getMemoryVT();
491506c3fb27SDimitry Andric     if ((MemVT == MVT::i8 || MemVT == MVT::i16) &&
491606c3fb27SDimitry Andric         (LD->getExtensionType() == ISD::NON_EXTLOAD ||
491706c3fb27SDimitry Andric          LD->getExtensionType() == ISD::ZEXTLOAD))
491806c3fb27SDimitry Andric       return true;
491906c3fb27SDimitry Andric   }
492006c3fb27SDimitry Andric 
492106c3fb27SDimitry Andric   return TargetLowering::isZExtFree(Val, VT2);
492206c3fb27SDimitry Andric }
492306c3fb27SDimitry Andric 
492406c3fb27SDimitry Andric bool LoongArchTargetLowering::isSExtCheaperThanZExt(EVT SrcVT, EVT DstVT) const {
492506c3fb27SDimitry Andric   return Subtarget.is64Bit() && SrcVT == MVT::i32 && DstVT == MVT::i64;
492606c3fb27SDimitry Andric }
492706c3fb27SDimitry Andric 
492806c3fb27SDimitry Andric bool LoongArchTargetLowering::hasAndNotCompare(SDValue Y) const {
492906c3fb27SDimitry Andric   // TODO: Support vectors.
493006c3fb27SDimitry Andric   if (Y.getValueType().isVector())
493106c3fb27SDimitry Andric     return false;
493206c3fb27SDimitry Andric 
493306c3fb27SDimitry Andric   return !isa<ConstantSDNode>(Y);
493406c3fb27SDimitry Andric }
4935