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