10b57cec5SDimitry Andric //===-- llvm/CodeGen/GlobalISel/MachineIRBuilder.cpp - MIBuilder--*- C++ -*-==// 20b57cec5SDimitry Andric // 30b57cec5SDimitry Andric // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. 40b57cec5SDimitry Andric // See https://llvm.org/LICENSE.txt for license information. 50b57cec5SDimitry Andric // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception 60b57cec5SDimitry Andric // 70b57cec5SDimitry Andric //===----------------------------------------------------------------------===// 80b57cec5SDimitry Andric /// \file 90b57cec5SDimitry Andric /// This file implements the MachineIRBuidler class. 100b57cec5SDimitry Andric //===----------------------------------------------------------------------===// 110b57cec5SDimitry Andric #include "llvm/CodeGen/GlobalISel/MachineIRBuilder.h" 12e8d8bef9SDimitry Andric #include "llvm/Analysis/MemoryLocation.h" 130b57cec5SDimitry Andric #include "llvm/CodeGen/GlobalISel/GISelChangeObserver.h" 140b57cec5SDimitry Andric #include "llvm/CodeGen/MachineFunction.h" 150b57cec5SDimitry Andric #include "llvm/CodeGen/MachineInstr.h" 160b57cec5SDimitry Andric #include "llvm/CodeGen/MachineInstrBuilder.h" 170b57cec5SDimitry Andric #include "llvm/CodeGen/MachineRegisterInfo.h" 180b57cec5SDimitry Andric #include "llvm/CodeGen/TargetInstrInfo.h" 190b57cec5SDimitry Andric #include "llvm/CodeGen/TargetLowering.h" 200b57cec5SDimitry Andric #include "llvm/CodeGen/TargetOpcodes.h" 210b57cec5SDimitry Andric #include "llvm/CodeGen/TargetSubtargetInfo.h" 220b57cec5SDimitry Andric #include "llvm/IR/DebugInfo.h" 230b57cec5SDimitry Andric 240b57cec5SDimitry Andric using namespace llvm; 250b57cec5SDimitry Andric 260b57cec5SDimitry Andric void MachineIRBuilder::setMF(MachineFunction &MF) { 270b57cec5SDimitry Andric State.MF = &MF; 280b57cec5SDimitry Andric State.MBB = nullptr; 290b57cec5SDimitry Andric State.MRI = &MF.getRegInfo(); 300b57cec5SDimitry Andric State.TII = MF.getSubtarget().getInstrInfo(); 310b57cec5SDimitry Andric State.DL = DebugLoc(); 320b57cec5SDimitry Andric State.II = MachineBasicBlock::iterator(); 330b57cec5SDimitry Andric State.Observer = nullptr; 340b57cec5SDimitry Andric } 350b57cec5SDimitry Andric 360b57cec5SDimitry Andric //------------------------------------------------------------------------------ 370b57cec5SDimitry Andric // Build instruction variants. 380b57cec5SDimitry Andric //------------------------------------------------------------------------------ 390b57cec5SDimitry Andric 400b57cec5SDimitry Andric MachineInstrBuilder MachineIRBuilder::buildInstrNoInsert(unsigned Opcode) { 410b57cec5SDimitry Andric MachineInstrBuilder MIB = BuildMI(getMF(), getDL(), getTII().get(Opcode)); 420b57cec5SDimitry Andric return MIB; 430b57cec5SDimitry Andric } 440b57cec5SDimitry Andric 450b57cec5SDimitry Andric MachineInstrBuilder MachineIRBuilder::insertInstr(MachineInstrBuilder MIB) { 460b57cec5SDimitry Andric getMBB().insert(getInsertPt(), MIB); 470b57cec5SDimitry Andric recordInsertion(MIB); 480b57cec5SDimitry Andric return MIB; 490b57cec5SDimitry Andric } 500b57cec5SDimitry Andric 510b57cec5SDimitry Andric MachineInstrBuilder 520b57cec5SDimitry Andric MachineIRBuilder::buildDirectDbgValue(Register Reg, const MDNode *Variable, 530b57cec5SDimitry Andric const MDNode *Expr) { 540b57cec5SDimitry Andric assert(isa<DILocalVariable>(Variable) && "not a variable"); 550b57cec5SDimitry Andric assert(cast<DIExpression>(Expr)->isValid() && "not an expression"); 560b57cec5SDimitry Andric assert( 570b57cec5SDimitry Andric cast<DILocalVariable>(Variable)->isValidLocationForIntrinsic(getDL()) && 580b57cec5SDimitry Andric "Expected inlined-at fields to agree"); 590b57cec5SDimitry Andric return insertInstr(BuildMI(getMF(), getDL(), 600b57cec5SDimitry Andric getTII().get(TargetOpcode::DBG_VALUE), 610b57cec5SDimitry Andric /*IsIndirect*/ false, Reg, Variable, Expr)); 620b57cec5SDimitry Andric } 630b57cec5SDimitry Andric 640b57cec5SDimitry Andric MachineInstrBuilder 650b57cec5SDimitry Andric MachineIRBuilder::buildIndirectDbgValue(Register Reg, const MDNode *Variable, 660b57cec5SDimitry Andric const MDNode *Expr) { 670b57cec5SDimitry Andric assert(isa<DILocalVariable>(Variable) && "not a variable"); 680b57cec5SDimitry Andric assert(cast<DIExpression>(Expr)->isValid() && "not an expression"); 690b57cec5SDimitry Andric assert( 700b57cec5SDimitry Andric cast<DILocalVariable>(Variable)->isValidLocationForIntrinsic(getDL()) && 710b57cec5SDimitry Andric "Expected inlined-at fields to agree"); 720b57cec5SDimitry Andric return insertInstr(BuildMI(getMF(), getDL(), 730b57cec5SDimitry Andric getTII().get(TargetOpcode::DBG_VALUE), 7413138422SDimitry Andric /*IsIndirect*/ true, Reg, Variable, Expr)); 750b57cec5SDimitry Andric } 760b57cec5SDimitry Andric 770b57cec5SDimitry Andric MachineInstrBuilder MachineIRBuilder::buildFIDbgValue(int FI, 780b57cec5SDimitry Andric const MDNode *Variable, 790b57cec5SDimitry Andric const MDNode *Expr) { 800b57cec5SDimitry Andric assert(isa<DILocalVariable>(Variable) && "not a variable"); 810b57cec5SDimitry Andric assert(cast<DIExpression>(Expr)->isValid() && "not an expression"); 820b57cec5SDimitry Andric assert( 830b57cec5SDimitry Andric cast<DILocalVariable>(Variable)->isValidLocationForIntrinsic(getDL()) && 840b57cec5SDimitry Andric "Expected inlined-at fields to agree"); 850b57cec5SDimitry Andric return buildInstr(TargetOpcode::DBG_VALUE) 860b57cec5SDimitry Andric .addFrameIndex(FI) 8713138422SDimitry Andric .addImm(0) 880b57cec5SDimitry Andric .addMetadata(Variable) 8913138422SDimitry Andric .addMetadata(Expr); 900b57cec5SDimitry Andric } 910b57cec5SDimitry Andric 920b57cec5SDimitry Andric MachineInstrBuilder MachineIRBuilder::buildConstDbgValue(const Constant &C, 930b57cec5SDimitry Andric const MDNode *Variable, 940b57cec5SDimitry Andric const MDNode *Expr) { 950b57cec5SDimitry Andric assert(isa<DILocalVariable>(Variable) && "not a variable"); 960b57cec5SDimitry Andric assert(cast<DIExpression>(Expr)->isValid() && "not an expression"); 970b57cec5SDimitry Andric assert( 980b57cec5SDimitry Andric cast<DILocalVariable>(Variable)->isValidLocationForIntrinsic(getDL()) && 990b57cec5SDimitry Andric "Expected inlined-at fields to agree"); 1005ffd83dbSDimitry Andric auto MIB = buildInstrNoInsert(TargetOpcode::DBG_VALUE); 1010b57cec5SDimitry Andric if (auto *CI = dyn_cast<ConstantInt>(&C)) { 1020b57cec5SDimitry Andric if (CI->getBitWidth() > 64) 1030b57cec5SDimitry Andric MIB.addCImm(CI); 1040b57cec5SDimitry Andric else 1050b57cec5SDimitry Andric MIB.addImm(CI->getZExtValue()); 1060b57cec5SDimitry Andric } else if (auto *CFP = dyn_cast<ConstantFP>(&C)) { 1070b57cec5SDimitry Andric MIB.addFPImm(CFP); 1080b57cec5SDimitry Andric } else { 109e8d8bef9SDimitry Andric // Insert $noreg if we didn't find a usable constant and had to drop it. 110e8d8bef9SDimitry Andric MIB.addReg(Register()); 1110b57cec5SDimitry Andric } 1120b57cec5SDimitry Andric 1135ffd83dbSDimitry Andric MIB.addImm(0).addMetadata(Variable).addMetadata(Expr); 1145ffd83dbSDimitry Andric return insertInstr(MIB); 1150b57cec5SDimitry Andric } 1160b57cec5SDimitry Andric 1170b57cec5SDimitry Andric MachineInstrBuilder MachineIRBuilder::buildDbgLabel(const MDNode *Label) { 1180b57cec5SDimitry Andric assert(isa<DILabel>(Label) && "not a label"); 1190b57cec5SDimitry Andric assert(cast<DILabel>(Label)->isValidLocationForIntrinsic(State.DL) && 1200b57cec5SDimitry Andric "Expected inlined-at fields to agree"); 1210b57cec5SDimitry Andric auto MIB = buildInstr(TargetOpcode::DBG_LABEL); 1220b57cec5SDimitry Andric 1230b57cec5SDimitry Andric return MIB.addMetadata(Label); 1240b57cec5SDimitry Andric } 1250b57cec5SDimitry Andric 1268bcb0991SDimitry Andric MachineInstrBuilder MachineIRBuilder::buildDynStackAlloc(const DstOp &Res, 1278bcb0991SDimitry Andric const SrcOp &Size, 1285ffd83dbSDimitry Andric Align Alignment) { 1298bcb0991SDimitry Andric assert(Res.getLLTTy(*getMRI()).isPointer() && "expected ptr dst type"); 1308bcb0991SDimitry Andric auto MIB = buildInstr(TargetOpcode::G_DYN_STACKALLOC); 1318bcb0991SDimitry Andric Res.addDefToMIB(*getMRI(), MIB); 1328bcb0991SDimitry Andric Size.addSrcToMIB(MIB); 1335ffd83dbSDimitry Andric MIB.addImm(Alignment.value()); 1348bcb0991SDimitry Andric return MIB; 1358bcb0991SDimitry Andric } 1368bcb0991SDimitry Andric 1370b57cec5SDimitry Andric MachineInstrBuilder MachineIRBuilder::buildFrameIndex(const DstOp &Res, 1380b57cec5SDimitry Andric int Idx) { 1390b57cec5SDimitry Andric assert(Res.getLLTTy(*getMRI()).isPointer() && "invalid operand type"); 1400b57cec5SDimitry Andric auto MIB = buildInstr(TargetOpcode::G_FRAME_INDEX); 1410b57cec5SDimitry Andric Res.addDefToMIB(*getMRI(), MIB); 1420b57cec5SDimitry Andric MIB.addFrameIndex(Idx); 1430b57cec5SDimitry Andric return MIB; 1440b57cec5SDimitry Andric } 1450b57cec5SDimitry Andric 1460b57cec5SDimitry Andric MachineInstrBuilder MachineIRBuilder::buildGlobalValue(const DstOp &Res, 1470b57cec5SDimitry Andric const GlobalValue *GV) { 1480b57cec5SDimitry Andric assert(Res.getLLTTy(*getMRI()).isPointer() && "invalid operand type"); 1490b57cec5SDimitry Andric assert(Res.getLLTTy(*getMRI()).getAddressSpace() == 1500b57cec5SDimitry Andric GV->getType()->getAddressSpace() && 1510b57cec5SDimitry Andric "address space mismatch"); 1520b57cec5SDimitry Andric 1530b57cec5SDimitry Andric auto MIB = buildInstr(TargetOpcode::G_GLOBAL_VALUE); 1540b57cec5SDimitry Andric Res.addDefToMIB(*getMRI(), MIB); 1550b57cec5SDimitry Andric MIB.addGlobalAddress(GV); 1560b57cec5SDimitry Andric return MIB; 1570b57cec5SDimitry Andric } 1580b57cec5SDimitry Andric 1590b57cec5SDimitry Andric MachineInstrBuilder MachineIRBuilder::buildJumpTable(const LLT PtrTy, 1600b57cec5SDimitry Andric unsigned JTI) { 1610b57cec5SDimitry Andric return buildInstr(TargetOpcode::G_JUMP_TABLE, {PtrTy}, {}) 1620b57cec5SDimitry Andric .addJumpTableIndex(JTI); 1630b57cec5SDimitry Andric } 1640b57cec5SDimitry Andric 165e8d8bef9SDimitry Andric void MachineIRBuilder::validateUnaryOp(const LLT Res, const LLT Op0) { 166e8d8bef9SDimitry Andric assert((Res.isScalar() || Res.isVector()) && "invalid operand type"); 167e8d8bef9SDimitry Andric assert((Res == Op0) && "type mismatch"); 168e8d8bef9SDimitry Andric } 169e8d8bef9SDimitry Andric 1705ffd83dbSDimitry Andric void MachineIRBuilder::validateBinaryOp(const LLT Res, const LLT Op0, 1715ffd83dbSDimitry Andric const LLT Op1) { 1720b57cec5SDimitry Andric assert((Res.isScalar() || Res.isVector()) && "invalid operand type"); 1730b57cec5SDimitry Andric assert((Res == Op0 && Res == Op1) && "type mismatch"); 1740b57cec5SDimitry Andric } 1750b57cec5SDimitry Andric 1765ffd83dbSDimitry Andric void MachineIRBuilder::validateShiftOp(const LLT Res, const LLT Op0, 1775ffd83dbSDimitry Andric const LLT Op1) { 1780b57cec5SDimitry Andric assert((Res.isScalar() || Res.isVector()) && "invalid operand type"); 1790b57cec5SDimitry Andric assert((Res == Op0) && "type mismatch"); 1800b57cec5SDimitry Andric } 1810b57cec5SDimitry Andric 182480093f4SDimitry Andric MachineInstrBuilder MachineIRBuilder::buildPtrAdd(const DstOp &Res, 1830b57cec5SDimitry Andric const SrcOp &Op0, 1840b57cec5SDimitry Andric const SrcOp &Op1) { 1855ffd83dbSDimitry Andric assert(Res.getLLTTy(*getMRI()).getScalarType().isPointer() && 1860b57cec5SDimitry Andric Res.getLLTTy(*getMRI()) == Op0.getLLTTy(*getMRI()) && "type mismatch"); 1875ffd83dbSDimitry Andric assert(Op1.getLLTTy(*getMRI()).getScalarType().isScalar() && "invalid offset type"); 1880b57cec5SDimitry Andric 189480093f4SDimitry Andric return buildInstr(TargetOpcode::G_PTR_ADD, {Res}, {Op0, Op1}); 1900b57cec5SDimitry Andric } 1910b57cec5SDimitry Andric 1920b57cec5SDimitry Andric Optional<MachineInstrBuilder> 193480093f4SDimitry Andric MachineIRBuilder::materializePtrAdd(Register &Res, Register Op0, 1945ffd83dbSDimitry Andric const LLT ValueTy, uint64_t Value) { 1950b57cec5SDimitry Andric assert(Res == 0 && "Res is a result argument"); 1960b57cec5SDimitry Andric assert(ValueTy.isScalar() && "invalid offset type"); 1970b57cec5SDimitry Andric 1980b57cec5SDimitry Andric if (Value == 0) { 1990b57cec5SDimitry Andric Res = Op0; 2000b57cec5SDimitry Andric return None; 2010b57cec5SDimitry Andric } 2020b57cec5SDimitry Andric 2030b57cec5SDimitry Andric Res = getMRI()->createGenericVirtualRegister(getMRI()->getType(Op0)); 2040b57cec5SDimitry Andric auto Cst = buildConstant(ValueTy, Value); 205480093f4SDimitry Andric return buildPtrAdd(Res, Op0, Cst.getReg(0)); 2060b57cec5SDimitry Andric } 2070b57cec5SDimitry Andric 2085ffd83dbSDimitry Andric MachineInstrBuilder MachineIRBuilder::buildMaskLowPtrBits(const DstOp &Res, 2090b57cec5SDimitry Andric const SrcOp &Op0, 2100b57cec5SDimitry Andric uint32_t NumBits) { 2115ffd83dbSDimitry Andric LLT PtrTy = Res.getLLTTy(*getMRI()); 2125ffd83dbSDimitry Andric LLT MaskTy = LLT::scalar(PtrTy.getSizeInBits()); 2135ffd83dbSDimitry Andric Register MaskReg = getMRI()->createGenericVirtualRegister(MaskTy); 2145ffd83dbSDimitry Andric buildConstant(MaskReg, maskTrailingZeros<uint64_t>(NumBits)); 2155ffd83dbSDimitry Andric return buildPtrMask(Res, Op0, MaskReg); 2160b57cec5SDimitry Andric } 2170b57cec5SDimitry Andric 218*0eae32dcSDimitry Andric MachineInstrBuilder 219*0eae32dcSDimitry Andric MachineIRBuilder::buildPadVectorWithUndefElements(const DstOp &Res, 220*0eae32dcSDimitry Andric const SrcOp &Op0) { 221*0eae32dcSDimitry Andric LLT ResTy = Res.getLLTTy(*getMRI()); 222*0eae32dcSDimitry Andric LLT Op0Ty = Op0.getLLTTy(*getMRI()); 223*0eae32dcSDimitry Andric 224*0eae32dcSDimitry Andric assert((ResTy.isVector() && Op0Ty.isVector()) && "Non vector type"); 225*0eae32dcSDimitry Andric assert((ResTy.getElementType() == Op0Ty.getElementType()) && 226*0eae32dcSDimitry Andric "Different vector element types"); 227*0eae32dcSDimitry Andric assert((ResTy.getNumElements() > Op0Ty.getNumElements()) && 228*0eae32dcSDimitry Andric "Op0 has more elements"); 229*0eae32dcSDimitry Andric 230*0eae32dcSDimitry Andric auto Unmerge = buildUnmerge(Op0Ty.getElementType(), Op0); 231*0eae32dcSDimitry Andric SmallVector<Register, 8> Regs; 232*0eae32dcSDimitry Andric for (auto Op : Unmerge.getInstr()->defs()) 233*0eae32dcSDimitry Andric Regs.push_back(Op.getReg()); 234*0eae32dcSDimitry Andric Register Undef = buildUndef(Op0Ty.getElementType()).getReg(0); 235*0eae32dcSDimitry Andric unsigned NumberOfPadElts = ResTy.getNumElements() - Regs.size(); 236*0eae32dcSDimitry Andric for (unsigned i = 0; i < NumberOfPadElts; ++i) 237*0eae32dcSDimitry Andric Regs.push_back(Undef); 238*0eae32dcSDimitry Andric return buildMerge(Res, Regs); 239*0eae32dcSDimitry Andric } 240*0eae32dcSDimitry Andric 241*0eae32dcSDimitry Andric MachineInstrBuilder 242*0eae32dcSDimitry Andric MachineIRBuilder::buildDeleteTrailingVectorElements(const DstOp &Res, 243*0eae32dcSDimitry Andric const SrcOp &Op0) { 244*0eae32dcSDimitry Andric LLT ResTy = Res.getLLTTy(*getMRI()); 245*0eae32dcSDimitry Andric LLT Op0Ty = Op0.getLLTTy(*getMRI()); 246*0eae32dcSDimitry Andric 247*0eae32dcSDimitry Andric assert((ResTy.isVector() && Op0Ty.isVector()) && "Non vector type"); 248*0eae32dcSDimitry Andric assert((ResTy.getElementType() == Op0Ty.getElementType()) && 249*0eae32dcSDimitry Andric "Different vector element types"); 250*0eae32dcSDimitry Andric assert((ResTy.getNumElements() < Op0Ty.getNumElements()) && 251*0eae32dcSDimitry Andric "Op0 has fewer elements"); 252*0eae32dcSDimitry Andric 253*0eae32dcSDimitry Andric SmallVector<Register, 8> Regs; 254*0eae32dcSDimitry Andric auto Unmerge = buildUnmerge(Op0Ty.getElementType(), Op0); 255*0eae32dcSDimitry Andric for (unsigned i = 0; i < ResTy.getNumElements(); ++i) 256*0eae32dcSDimitry Andric Regs.push_back(Unmerge.getReg(i)); 257*0eae32dcSDimitry Andric return buildMerge(Res, Regs); 258*0eae32dcSDimitry Andric } 259*0eae32dcSDimitry Andric 2600b57cec5SDimitry Andric MachineInstrBuilder MachineIRBuilder::buildBr(MachineBasicBlock &Dest) { 2610b57cec5SDimitry Andric return buildInstr(TargetOpcode::G_BR).addMBB(&Dest); 2620b57cec5SDimitry Andric } 2630b57cec5SDimitry Andric 2640b57cec5SDimitry Andric MachineInstrBuilder MachineIRBuilder::buildBrIndirect(Register Tgt) { 2650b57cec5SDimitry Andric assert(getMRI()->getType(Tgt).isPointer() && "invalid branch destination"); 2660b57cec5SDimitry Andric return buildInstr(TargetOpcode::G_BRINDIRECT).addUse(Tgt); 2670b57cec5SDimitry Andric } 2680b57cec5SDimitry Andric 2690b57cec5SDimitry Andric MachineInstrBuilder MachineIRBuilder::buildBrJT(Register TablePtr, 2700b57cec5SDimitry Andric unsigned JTI, 2710b57cec5SDimitry Andric Register IndexReg) { 2720b57cec5SDimitry Andric assert(getMRI()->getType(TablePtr).isPointer() && 2730b57cec5SDimitry Andric "Table reg must be a pointer"); 2740b57cec5SDimitry Andric return buildInstr(TargetOpcode::G_BRJT) 2750b57cec5SDimitry Andric .addUse(TablePtr) 2760b57cec5SDimitry Andric .addJumpTableIndex(JTI) 2770b57cec5SDimitry Andric .addUse(IndexReg); 2780b57cec5SDimitry Andric } 2790b57cec5SDimitry Andric 2800b57cec5SDimitry Andric MachineInstrBuilder MachineIRBuilder::buildCopy(const DstOp &Res, 2810b57cec5SDimitry Andric const SrcOp &Op) { 2820b57cec5SDimitry Andric return buildInstr(TargetOpcode::COPY, Res, Op); 2830b57cec5SDimitry Andric } 2840b57cec5SDimitry Andric 285fe6060f1SDimitry Andric MachineInstrBuilder MachineIRBuilder::buildAssertSExt(const DstOp &Res, 286fe6060f1SDimitry Andric const SrcOp &Op, 287fe6060f1SDimitry Andric unsigned Size) { 288fe6060f1SDimitry Andric return buildInstr(TargetOpcode::G_ASSERT_SEXT, Res, Op).addImm(Size); 289fe6060f1SDimitry Andric } 290fe6060f1SDimitry Andric 291fe6060f1SDimitry Andric MachineInstrBuilder MachineIRBuilder::buildAssertZExt(const DstOp &Res, 292fe6060f1SDimitry Andric const SrcOp &Op, 293fe6060f1SDimitry Andric unsigned Size) { 294fe6060f1SDimitry Andric return buildInstr(TargetOpcode::G_ASSERT_ZEXT, Res, Op).addImm(Size); 295fe6060f1SDimitry Andric } 296fe6060f1SDimitry Andric 2970b57cec5SDimitry Andric MachineInstrBuilder MachineIRBuilder::buildConstant(const DstOp &Res, 2980b57cec5SDimitry Andric const ConstantInt &Val) { 2990b57cec5SDimitry Andric LLT Ty = Res.getLLTTy(*getMRI()); 3000b57cec5SDimitry Andric LLT EltTy = Ty.getScalarType(); 3010b57cec5SDimitry Andric assert(EltTy.getScalarSizeInBits() == Val.getBitWidth() && 3020b57cec5SDimitry Andric "creating constant with the wrong size"); 3030b57cec5SDimitry Andric 3040b57cec5SDimitry Andric if (Ty.isVector()) { 3050b57cec5SDimitry Andric auto Const = buildInstr(TargetOpcode::G_CONSTANT) 3060b57cec5SDimitry Andric .addDef(getMRI()->createGenericVirtualRegister(EltTy)) 3070b57cec5SDimitry Andric .addCImm(&Val); 3080b57cec5SDimitry Andric return buildSplatVector(Res, Const); 3090b57cec5SDimitry Andric } 3100b57cec5SDimitry Andric 3110b57cec5SDimitry Andric auto Const = buildInstr(TargetOpcode::G_CONSTANT); 3125ffd83dbSDimitry Andric Const->setDebugLoc(DebugLoc()); 3130b57cec5SDimitry Andric Res.addDefToMIB(*getMRI(), Const); 3140b57cec5SDimitry Andric Const.addCImm(&Val); 3150b57cec5SDimitry Andric return Const; 3160b57cec5SDimitry Andric } 3170b57cec5SDimitry Andric 3180b57cec5SDimitry Andric MachineInstrBuilder MachineIRBuilder::buildConstant(const DstOp &Res, 3190b57cec5SDimitry Andric int64_t Val) { 3200b57cec5SDimitry Andric auto IntN = IntegerType::get(getMF().getFunction().getContext(), 3210b57cec5SDimitry Andric Res.getLLTTy(*getMRI()).getScalarSizeInBits()); 3220b57cec5SDimitry Andric ConstantInt *CI = ConstantInt::get(IntN, Val, true); 3230b57cec5SDimitry Andric return buildConstant(Res, *CI); 3240b57cec5SDimitry Andric } 3250b57cec5SDimitry Andric 3260b57cec5SDimitry Andric MachineInstrBuilder MachineIRBuilder::buildFConstant(const DstOp &Res, 3270b57cec5SDimitry Andric const ConstantFP &Val) { 3280b57cec5SDimitry Andric LLT Ty = Res.getLLTTy(*getMRI()); 3290b57cec5SDimitry Andric LLT EltTy = Ty.getScalarType(); 3300b57cec5SDimitry Andric 3310b57cec5SDimitry Andric assert(APFloat::getSizeInBits(Val.getValueAPF().getSemantics()) 3320b57cec5SDimitry Andric == EltTy.getSizeInBits() && 3330b57cec5SDimitry Andric "creating fconstant with the wrong size"); 3340b57cec5SDimitry Andric 3350b57cec5SDimitry Andric assert(!Ty.isPointer() && "invalid operand type"); 3360b57cec5SDimitry Andric 3370b57cec5SDimitry Andric if (Ty.isVector()) { 3380b57cec5SDimitry Andric auto Const = buildInstr(TargetOpcode::G_FCONSTANT) 3390b57cec5SDimitry Andric .addDef(getMRI()->createGenericVirtualRegister(EltTy)) 3400b57cec5SDimitry Andric .addFPImm(&Val); 3410b57cec5SDimitry Andric 3420b57cec5SDimitry Andric return buildSplatVector(Res, Const); 3430b57cec5SDimitry Andric } 3440b57cec5SDimitry Andric 3450b57cec5SDimitry Andric auto Const = buildInstr(TargetOpcode::G_FCONSTANT); 3465ffd83dbSDimitry Andric Const->setDebugLoc(DebugLoc()); 3470b57cec5SDimitry Andric Res.addDefToMIB(*getMRI(), Const); 3480b57cec5SDimitry Andric Const.addFPImm(&Val); 3490b57cec5SDimitry Andric return Const; 3500b57cec5SDimitry Andric } 3510b57cec5SDimitry Andric 3520b57cec5SDimitry Andric MachineInstrBuilder MachineIRBuilder::buildConstant(const DstOp &Res, 3530b57cec5SDimitry Andric const APInt &Val) { 3540b57cec5SDimitry Andric ConstantInt *CI = ConstantInt::get(getMF().getFunction().getContext(), Val); 3550b57cec5SDimitry Andric return buildConstant(Res, *CI); 3560b57cec5SDimitry Andric } 3570b57cec5SDimitry Andric 3580b57cec5SDimitry Andric MachineInstrBuilder MachineIRBuilder::buildFConstant(const DstOp &Res, 3590b57cec5SDimitry Andric double Val) { 3600b57cec5SDimitry Andric LLT DstTy = Res.getLLTTy(*getMRI()); 3610b57cec5SDimitry Andric auto &Ctx = getMF().getFunction().getContext(); 3620b57cec5SDimitry Andric auto *CFP = 3630b57cec5SDimitry Andric ConstantFP::get(Ctx, getAPFloatFromSize(Val, DstTy.getScalarSizeInBits())); 3640b57cec5SDimitry Andric return buildFConstant(Res, *CFP); 3650b57cec5SDimitry Andric } 3660b57cec5SDimitry Andric 3670b57cec5SDimitry Andric MachineInstrBuilder MachineIRBuilder::buildFConstant(const DstOp &Res, 3680b57cec5SDimitry Andric const APFloat &Val) { 3690b57cec5SDimitry Andric auto &Ctx = getMF().getFunction().getContext(); 3700b57cec5SDimitry Andric auto *CFP = ConstantFP::get(Ctx, Val); 3710b57cec5SDimitry Andric return buildFConstant(Res, *CFP); 3720b57cec5SDimitry Andric } 3730b57cec5SDimitry Andric 374e8d8bef9SDimitry Andric MachineInstrBuilder MachineIRBuilder::buildBrCond(const SrcOp &Tst, 3750b57cec5SDimitry Andric MachineBasicBlock &Dest) { 376e8d8bef9SDimitry Andric assert(Tst.getLLTTy(*getMRI()).isScalar() && "invalid operand type"); 3770b57cec5SDimitry Andric 378e8d8bef9SDimitry Andric auto MIB = buildInstr(TargetOpcode::G_BRCOND); 379e8d8bef9SDimitry Andric Tst.addSrcToMIB(MIB); 380e8d8bef9SDimitry Andric MIB.addMBB(&Dest); 381e8d8bef9SDimitry Andric return MIB; 3820b57cec5SDimitry Andric } 3830b57cec5SDimitry Andric 384e8d8bef9SDimitry Andric MachineInstrBuilder 385e8d8bef9SDimitry Andric MachineIRBuilder::buildLoad(const DstOp &Dst, const SrcOp &Addr, 386e8d8bef9SDimitry Andric MachinePointerInfo PtrInfo, Align Alignment, 387e8d8bef9SDimitry Andric MachineMemOperand::Flags MMOFlags, 388e8d8bef9SDimitry Andric const AAMDNodes &AAInfo) { 389e8d8bef9SDimitry Andric MMOFlags |= MachineMemOperand::MOLoad; 390e8d8bef9SDimitry Andric assert((MMOFlags & MachineMemOperand::MOStore) == 0); 391e8d8bef9SDimitry Andric 392fe6060f1SDimitry Andric LLT Ty = Dst.getLLTTy(*getMRI()); 393e8d8bef9SDimitry Andric MachineMemOperand *MMO = 394fe6060f1SDimitry Andric getMF().getMachineMemOperand(PtrInfo, MMOFlags, Ty, Alignment, AAInfo); 395e8d8bef9SDimitry Andric return buildLoad(Dst, Addr, *MMO); 3960b57cec5SDimitry Andric } 3970b57cec5SDimitry Andric 3980b57cec5SDimitry Andric MachineInstrBuilder MachineIRBuilder::buildLoadInstr(unsigned Opcode, 3990b57cec5SDimitry Andric const DstOp &Res, 4000b57cec5SDimitry Andric const SrcOp &Addr, 4010b57cec5SDimitry Andric MachineMemOperand &MMO) { 4020b57cec5SDimitry Andric assert(Res.getLLTTy(*getMRI()).isValid() && "invalid operand type"); 4030b57cec5SDimitry Andric assert(Addr.getLLTTy(*getMRI()).isPointer() && "invalid operand type"); 4040b57cec5SDimitry Andric 4050b57cec5SDimitry Andric auto MIB = buildInstr(Opcode); 4060b57cec5SDimitry Andric Res.addDefToMIB(*getMRI(), MIB); 4070b57cec5SDimitry Andric Addr.addSrcToMIB(MIB); 4080b57cec5SDimitry Andric MIB.addMemOperand(&MMO); 4090b57cec5SDimitry Andric return MIB; 4100b57cec5SDimitry Andric } 4110b57cec5SDimitry Andric 4125ffd83dbSDimitry Andric MachineInstrBuilder MachineIRBuilder::buildLoadFromOffset( 4135ffd83dbSDimitry Andric const DstOp &Dst, const SrcOp &BasePtr, 4145ffd83dbSDimitry Andric MachineMemOperand &BaseMMO, int64_t Offset) { 4155ffd83dbSDimitry Andric LLT LoadTy = Dst.getLLTTy(*getMRI()); 4165ffd83dbSDimitry Andric MachineMemOperand *OffsetMMO = 417fe6060f1SDimitry Andric getMF().getMachineMemOperand(&BaseMMO, Offset, LoadTy); 4185ffd83dbSDimitry Andric 4195ffd83dbSDimitry Andric if (Offset == 0) // This may be a size or type changing load. 4205ffd83dbSDimitry Andric return buildLoad(Dst, BasePtr, *OffsetMMO); 4215ffd83dbSDimitry Andric 4225ffd83dbSDimitry Andric LLT PtrTy = BasePtr.getLLTTy(*getMRI()); 4235ffd83dbSDimitry Andric LLT OffsetTy = LLT::scalar(PtrTy.getSizeInBits()); 4245ffd83dbSDimitry Andric auto ConstOffset = buildConstant(OffsetTy, Offset); 4255ffd83dbSDimitry Andric auto Ptr = buildPtrAdd(PtrTy, BasePtr, ConstOffset); 4265ffd83dbSDimitry Andric return buildLoad(Dst, Ptr, *OffsetMMO); 4275ffd83dbSDimitry Andric } 4285ffd83dbSDimitry Andric 4290b57cec5SDimitry Andric MachineInstrBuilder MachineIRBuilder::buildStore(const SrcOp &Val, 4300b57cec5SDimitry Andric const SrcOp &Addr, 4310b57cec5SDimitry Andric MachineMemOperand &MMO) { 4320b57cec5SDimitry Andric assert(Val.getLLTTy(*getMRI()).isValid() && "invalid operand type"); 4330b57cec5SDimitry Andric assert(Addr.getLLTTy(*getMRI()).isPointer() && "invalid operand type"); 4340b57cec5SDimitry Andric 4350b57cec5SDimitry Andric auto MIB = buildInstr(TargetOpcode::G_STORE); 4360b57cec5SDimitry Andric Val.addSrcToMIB(MIB); 4370b57cec5SDimitry Andric Addr.addSrcToMIB(MIB); 4380b57cec5SDimitry Andric MIB.addMemOperand(&MMO); 4390b57cec5SDimitry Andric return MIB; 4400b57cec5SDimitry Andric } 4410b57cec5SDimitry Andric 442e8d8bef9SDimitry Andric MachineInstrBuilder 443e8d8bef9SDimitry Andric MachineIRBuilder::buildStore(const SrcOp &Val, const SrcOp &Addr, 444e8d8bef9SDimitry Andric MachinePointerInfo PtrInfo, Align Alignment, 445e8d8bef9SDimitry Andric MachineMemOperand::Flags MMOFlags, 446e8d8bef9SDimitry Andric const AAMDNodes &AAInfo) { 447e8d8bef9SDimitry Andric MMOFlags |= MachineMemOperand::MOStore; 448e8d8bef9SDimitry Andric assert((MMOFlags & MachineMemOperand::MOLoad) == 0); 449e8d8bef9SDimitry Andric 450fe6060f1SDimitry Andric LLT Ty = Val.getLLTTy(*getMRI()); 451e8d8bef9SDimitry Andric MachineMemOperand *MMO = 452fe6060f1SDimitry Andric getMF().getMachineMemOperand(PtrInfo, MMOFlags, Ty, Alignment, AAInfo); 453e8d8bef9SDimitry Andric return buildStore(Val, Addr, *MMO); 454e8d8bef9SDimitry Andric } 455e8d8bef9SDimitry Andric 4560b57cec5SDimitry Andric MachineInstrBuilder MachineIRBuilder::buildAnyExt(const DstOp &Res, 4570b57cec5SDimitry Andric const SrcOp &Op) { 4580b57cec5SDimitry Andric return buildInstr(TargetOpcode::G_ANYEXT, Res, Op); 4590b57cec5SDimitry Andric } 4600b57cec5SDimitry Andric 4610b57cec5SDimitry Andric MachineInstrBuilder MachineIRBuilder::buildSExt(const DstOp &Res, 4620b57cec5SDimitry Andric const SrcOp &Op) { 4630b57cec5SDimitry Andric return buildInstr(TargetOpcode::G_SEXT, Res, Op); 4640b57cec5SDimitry Andric } 4650b57cec5SDimitry Andric 4660b57cec5SDimitry Andric MachineInstrBuilder MachineIRBuilder::buildZExt(const DstOp &Res, 4670b57cec5SDimitry Andric const SrcOp &Op) { 4680b57cec5SDimitry Andric return buildInstr(TargetOpcode::G_ZEXT, Res, Op); 4690b57cec5SDimitry Andric } 4700b57cec5SDimitry Andric 4710b57cec5SDimitry Andric unsigned MachineIRBuilder::getBoolExtOp(bool IsVec, bool IsFP) const { 4720b57cec5SDimitry Andric const auto *TLI = getMF().getSubtarget().getTargetLowering(); 4730b57cec5SDimitry Andric switch (TLI->getBooleanContents(IsVec, IsFP)) { 4740b57cec5SDimitry Andric case TargetLoweringBase::ZeroOrNegativeOneBooleanContent: 4750b57cec5SDimitry Andric return TargetOpcode::G_SEXT; 4760b57cec5SDimitry Andric case TargetLoweringBase::ZeroOrOneBooleanContent: 4770b57cec5SDimitry Andric return TargetOpcode::G_ZEXT; 4780b57cec5SDimitry Andric default: 4790b57cec5SDimitry Andric return TargetOpcode::G_ANYEXT; 4800b57cec5SDimitry Andric } 4810b57cec5SDimitry Andric } 4820b57cec5SDimitry Andric 4830b57cec5SDimitry Andric MachineInstrBuilder MachineIRBuilder::buildBoolExt(const DstOp &Res, 4840b57cec5SDimitry Andric const SrcOp &Op, 4850b57cec5SDimitry Andric bool IsFP) { 4860b57cec5SDimitry Andric unsigned ExtOp = getBoolExtOp(getMRI()->getType(Op.getReg()).isVector(), IsFP); 4870b57cec5SDimitry Andric return buildInstr(ExtOp, Res, Op); 4880b57cec5SDimitry Andric } 4890b57cec5SDimitry Andric 4900b57cec5SDimitry Andric MachineInstrBuilder MachineIRBuilder::buildExtOrTrunc(unsigned ExtOpc, 4910b57cec5SDimitry Andric const DstOp &Res, 4920b57cec5SDimitry Andric const SrcOp &Op) { 4930b57cec5SDimitry Andric assert((TargetOpcode::G_ANYEXT == ExtOpc || TargetOpcode::G_ZEXT == ExtOpc || 4940b57cec5SDimitry Andric TargetOpcode::G_SEXT == ExtOpc) && 4950b57cec5SDimitry Andric "Expecting Extending Opc"); 4960b57cec5SDimitry Andric assert(Res.getLLTTy(*getMRI()).isScalar() || 4970b57cec5SDimitry Andric Res.getLLTTy(*getMRI()).isVector()); 4980b57cec5SDimitry Andric assert(Res.getLLTTy(*getMRI()).isScalar() == 4990b57cec5SDimitry Andric Op.getLLTTy(*getMRI()).isScalar()); 5000b57cec5SDimitry Andric 5010b57cec5SDimitry Andric unsigned Opcode = TargetOpcode::COPY; 5020b57cec5SDimitry Andric if (Res.getLLTTy(*getMRI()).getSizeInBits() > 5030b57cec5SDimitry Andric Op.getLLTTy(*getMRI()).getSizeInBits()) 5040b57cec5SDimitry Andric Opcode = ExtOpc; 5050b57cec5SDimitry Andric else if (Res.getLLTTy(*getMRI()).getSizeInBits() < 5060b57cec5SDimitry Andric Op.getLLTTy(*getMRI()).getSizeInBits()) 5070b57cec5SDimitry Andric Opcode = TargetOpcode::G_TRUNC; 5080b57cec5SDimitry Andric else 5090b57cec5SDimitry Andric assert(Res.getLLTTy(*getMRI()) == Op.getLLTTy(*getMRI())); 5100b57cec5SDimitry Andric 5110b57cec5SDimitry Andric return buildInstr(Opcode, Res, Op); 5120b57cec5SDimitry Andric } 5130b57cec5SDimitry Andric 5140b57cec5SDimitry Andric MachineInstrBuilder MachineIRBuilder::buildSExtOrTrunc(const DstOp &Res, 5150b57cec5SDimitry Andric const SrcOp &Op) { 5160b57cec5SDimitry Andric return buildExtOrTrunc(TargetOpcode::G_SEXT, Res, Op); 5170b57cec5SDimitry Andric } 5180b57cec5SDimitry Andric 5190b57cec5SDimitry Andric MachineInstrBuilder MachineIRBuilder::buildZExtOrTrunc(const DstOp &Res, 5200b57cec5SDimitry Andric const SrcOp &Op) { 5210b57cec5SDimitry Andric return buildExtOrTrunc(TargetOpcode::G_ZEXT, Res, Op); 5220b57cec5SDimitry Andric } 5230b57cec5SDimitry Andric 5240b57cec5SDimitry Andric MachineInstrBuilder MachineIRBuilder::buildAnyExtOrTrunc(const DstOp &Res, 5250b57cec5SDimitry Andric const SrcOp &Op) { 5260b57cec5SDimitry Andric return buildExtOrTrunc(TargetOpcode::G_ANYEXT, Res, Op); 5270b57cec5SDimitry Andric } 5280b57cec5SDimitry Andric 529fe6060f1SDimitry Andric MachineInstrBuilder MachineIRBuilder::buildZExtInReg(const DstOp &Res, 530fe6060f1SDimitry Andric const SrcOp &Op, 531fe6060f1SDimitry Andric int64_t ImmOp) { 532fe6060f1SDimitry Andric LLT ResTy = Res.getLLTTy(*getMRI()); 533fe6060f1SDimitry Andric auto Mask = buildConstant( 534fe6060f1SDimitry Andric ResTy, APInt::getLowBitsSet(ResTy.getScalarSizeInBits(), ImmOp)); 535fe6060f1SDimitry Andric return buildAnd(Res, Op, Mask); 536fe6060f1SDimitry Andric } 537fe6060f1SDimitry Andric 5380b57cec5SDimitry Andric MachineInstrBuilder MachineIRBuilder::buildCast(const DstOp &Dst, 5390b57cec5SDimitry Andric const SrcOp &Src) { 5400b57cec5SDimitry Andric LLT SrcTy = Src.getLLTTy(*getMRI()); 5410b57cec5SDimitry Andric LLT DstTy = Dst.getLLTTy(*getMRI()); 5420b57cec5SDimitry Andric if (SrcTy == DstTy) 5430b57cec5SDimitry Andric return buildCopy(Dst, Src); 5440b57cec5SDimitry Andric 5450b57cec5SDimitry Andric unsigned Opcode; 5460b57cec5SDimitry Andric if (SrcTy.isPointer() && DstTy.isScalar()) 5470b57cec5SDimitry Andric Opcode = TargetOpcode::G_PTRTOINT; 5480b57cec5SDimitry Andric else if (DstTy.isPointer() && SrcTy.isScalar()) 5490b57cec5SDimitry Andric Opcode = TargetOpcode::G_INTTOPTR; 5500b57cec5SDimitry Andric else { 5510b57cec5SDimitry Andric assert(!SrcTy.isPointer() && !DstTy.isPointer() && "n G_ADDRCAST yet"); 5520b57cec5SDimitry Andric Opcode = TargetOpcode::G_BITCAST; 5530b57cec5SDimitry Andric } 5540b57cec5SDimitry Andric 5550b57cec5SDimitry Andric return buildInstr(Opcode, Dst, Src); 5560b57cec5SDimitry Andric } 5570b57cec5SDimitry Andric 5580b57cec5SDimitry Andric MachineInstrBuilder MachineIRBuilder::buildExtract(const DstOp &Dst, 5590b57cec5SDimitry Andric const SrcOp &Src, 5600b57cec5SDimitry Andric uint64_t Index) { 5610b57cec5SDimitry Andric LLT SrcTy = Src.getLLTTy(*getMRI()); 5620b57cec5SDimitry Andric LLT DstTy = Dst.getLLTTy(*getMRI()); 5630b57cec5SDimitry Andric 5640b57cec5SDimitry Andric #ifndef NDEBUG 5650b57cec5SDimitry Andric assert(SrcTy.isValid() && "invalid operand type"); 5660b57cec5SDimitry Andric assert(DstTy.isValid() && "invalid operand type"); 5670b57cec5SDimitry Andric assert(Index + DstTy.getSizeInBits() <= SrcTy.getSizeInBits() && 5680b57cec5SDimitry Andric "extracting off end of register"); 5690b57cec5SDimitry Andric #endif 5700b57cec5SDimitry Andric 5710b57cec5SDimitry Andric if (DstTy.getSizeInBits() == SrcTy.getSizeInBits()) { 5720b57cec5SDimitry Andric assert(Index == 0 && "insertion past the end of a register"); 5730b57cec5SDimitry Andric return buildCast(Dst, Src); 5740b57cec5SDimitry Andric } 5750b57cec5SDimitry Andric 5760b57cec5SDimitry Andric auto Extract = buildInstr(TargetOpcode::G_EXTRACT); 5770b57cec5SDimitry Andric Dst.addDefToMIB(*getMRI(), Extract); 5780b57cec5SDimitry Andric Src.addSrcToMIB(Extract); 5790b57cec5SDimitry Andric Extract.addImm(Index); 5800b57cec5SDimitry Andric return Extract; 5810b57cec5SDimitry Andric } 5820b57cec5SDimitry Andric 5830b57cec5SDimitry Andric void MachineIRBuilder::buildSequence(Register Res, ArrayRef<Register> Ops, 5840b57cec5SDimitry Andric ArrayRef<uint64_t> Indices) { 5850b57cec5SDimitry Andric #ifndef NDEBUG 5860b57cec5SDimitry Andric assert(Ops.size() == Indices.size() && "incompatible args"); 5870b57cec5SDimitry Andric assert(!Ops.empty() && "invalid trivial sequence"); 5885ffd83dbSDimitry Andric assert(llvm::is_sorted(Indices) && 5890b57cec5SDimitry Andric "sequence offsets must be in ascending order"); 5900b57cec5SDimitry Andric 5910b57cec5SDimitry Andric assert(getMRI()->getType(Res).isValid() && "invalid operand type"); 5920b57cec5SDimitry Andric for (auto Op : Ops) 5930b57cec5SDimitry Andric assert(getMRI()->getType(Op).isValid() && "invalid operand type"); 5940b57cec5SDimitry Andric #endif 5950b57cec5SDimitry Andric 5960b57cec5SDimitry Andric LLT ResTy = getMRI()->getType(Res); 5970b57cec5SDimitry Andric LLT OpTy = getMRI()->getType(Ops[0]); 5980b57cec5SDimitry Andric unsigned OpSize = OpTy.getSizeInBits(); 5990b57cec5SDimitry Andric bool MaybeMerge = true; 6000b57cec5SDimitry Andric for (unsigned i = 0; i < Ops.size(); ++i) { 6010b57cec5SDimitry Andric if (getMRI()->getType(Ops[i]) != OpTy || Indices[i] != i * OpSize) { 6020b57cec5SDimitry Andric MaybeMerge = false; 6030b57cec5SDimitry Andric break; 6040b57cec5SDimitry Andric } 6050b57cec5SDimitry Andric } 6060b57cec5SDimitry Andric 6070b57cec5SDimitry Andric if (MaybeMerge && Ops.size() * OpSize == ResTy.getSizeInBits()) { 6080b57cec5SDimitry Andric buildMerge(Res, Ops); 6090b57cec5SDimitry Andric return; 6100b57cec5SDimitry Andric } 6110b57cec5SDimitry Andric 6120b57cec5SDimitry Andric Register ResIn = getMRI()->createGenericVirtualRegister(ResTy); 6130b57cec5SDimitry Andric buildUndef(ResIn); 6140b57cec5SDimitry Andric 6150b57cec5SDimitry Andric for (unsigned i = 0; i < Ops.size(); ++i) { 6160b57cec5SDimitry Andric Register ResOut = i + 1 == Ops.size() 6170b57cec5SDimitry Andric ? Res 6180b57cec5SDimitry Andric : getMRI()->createGenericVirtualRegister(ResTy); 6190b57cec5SDimitry Andric buildInsert(ResOut, ResIn, Ops[i], Indices[i]); 6200b57cec5SDimitry Andric ResIn = ResOut; 6210b57cec5SDimitry Andric } 6220b57cec5SDimitry Andric } 6230b57cec5SDimitry Andric 6240b57cec5SDimitry Andric MachineInstrBuilder MachineIRBuilder::buildUndef(const DstOp &Res) { 6250b57cec5SDimitry Andric return buildInstr(TargetOpcode::G_IMPLICIT_DEF, {Res}, {}); 6260b57cec5SDimitry Andric } 6270b57cec5SDimitry Andric 6280b57cec5SDimitry Andric MachineInstrBuilder MachineIRBuilder::buildMerge(const DstOp &Res, 6290b57cec5SDimitry Andric ArrayRef<Register> Ops) { 6300b57cec5SDimitry Andric // Unfortunately to convert from ArrayRef<LLT> to ArrayRef<SrcOp>, 6310b57cec5SDimitry Andric // we need some temporary storage for the DstOp objects. Here we use a 6320b57cec5SDimitry Andric // sufficiently large SmallVector to not go through the heap. 6330b57cec5SDimitry Andric SmallVector<SrcOp, 8> TmpVec(Ops.begin(), Ops.end()); 6340b57cec5SDimitry Andric assert(TmpVec.size() > 1); 6350b57cec5SDimitry Andric return buildInstr(TargetOpcode::G_MERGE_VALUES, Res, TmpVec); 6360b57cec5SDimitry Andric } 6370b57cec5SDimitry Andric 6385ffd83dbSDimitry Andric MachineInstrBuilder 6395ffd83dbSDimitry Andric MachineIRBuilder::buildMerge(const DstOp &Res, 6405ffd83dbSDimitry Andric std::initializer_list<SrcOp> Ops) { 6415ffd83dbSDimitry Andric assert(Ops.size() > 1); 6425ffd83dbSDimitry Andric return buildInstr(TargetOpcode::G_MERGE_VALUES, Res, Ops); 6435ffd83dbSDimitry Andric } 6445ffd83dbSDimitry Andric 6450b57cec5SDimitry Andric MachineInstrBuilder MachineIRBuilder::buildUnmerge(ArrayRef<LLT> Res, 6460b57cec5SDimitry Andric const SrcOp &Op) { 6470b57cec5SDimitry Andric // Unfortunately to convert from ArrayRef<LLT> to ArrayRef<DstOp>, 6480b57cec5SDimitry Andric // we need some temporary storage for the DstOp objects. Here we use a 6490b57cec5SDimitry Andric // sufficiently large SmallVector to not go through the heap. 6500b57cec5SDimitry Andric SmallVector<DstOp, 8> TmpVec(Res.begin(), Res.end()); 6510b57cec5SDimitry Andric assert(TmpVec.size() > 1); 6520b57cec5SDimitry Andric return buildInstr(TargetOpcode::G_UNMERGE_VALUES, TmpVec, Op); 6530b57cec5SDimitry Andric } 6540b57cec5SDimitry Andric 6550b57cec5SDimitry Andric MachineInstrBuilder MachineIRBuilder::buildUnmerge(LLT Res, 6560b57cec5SDimitry Andric const SrcOp &Op) { 6570b57cec5SDimitry Andric unsigned NumReg = Op.getLLTTy(*getMRI()).getSizeInBits() / Res.getSizeInBits(); 658*0eae32dcSDimitry Andric SmallVector<DstOp, 8> TmpVec(NumReg, Res); 659*0eae32dcSDimitry Andric return buildInstr(TargetOpcode::G_UNMERGE_VALUES, TmpVec, Op); 6600b57cec5SDimitry Andric } 6610b57cec5SDimitry Andric 6620b57cec5SDimitry Andric MachineInstrBuilder MachineIRBuilder::buildUnmerge(ArrayRef<Register> Res, 6630b57cec5SDimitry Andric const SrcOp &Op) { 6640b57cec5SDimitry Andric // Unfortunately to convert from ArrayRef<Register> to ArrayRef<DstOp>, 6650b57cec5SDimitry Andric // we need some temporary storage for the DstOp objects. Here we use a 6660b57cec5SDimitry Andric // sufficiently large SmallVector to not go through the heap. 6670b57cec5SDimitry Andric SmallVector<DstOp, 8> TmpVec(Res.begin(), Res.end()); 6680b57cec5SDimitry Andric assert(TmpVec.size() > 1); 6690b57cec5SDimitry Andric return buildInstr(TargetOpcode::G_UNMERGE_VALUES, TmpVec, Op); 6700b57cec5SDimitry Andric } 6710b57cec5SDimitry Andric 6720b57cec5SDimitry Andric MachineInstrBuilder MachineIRBuilder::buildBuildVector(const DstOp &Res, 6730b57cec5SDimitry Andric ArrayRef<Register> Ops) { 6740b57cec5SDimitry Andric // Unfortunately to convert from ArrayRef<Register> to ArrayRef<SrcOp>, 6750b57cec5SDimitry Andric // we need some temporary storage for the DstOp objects. Here we use a 6760b57cec5SDimitry Andric // sufficiently large SmallVector to not go through the heap. 6770b57cec5SDimitry Andric SmallVector<SrcOp, 8> TmpVec(Ops.begin(), Ops.end()); 6780b57cec5SDimitry Andric return buildInstr(TargetOpcode::G_BUILD_VECTOR, Res, TmpVec); 6790b57cec5SDimitry Andric } 6800b57cec5SDimitry Andric 6810b57cec5SDimitry Andric MachineInstrBuilder MachineIRBuilder::buildSplatVector(const DstOp &Res, 6820b57cec5SDimitry Andric const SrcOp &Src) { 6830b57cec5SDimitry Andric SmallVector<SrcOp, 8> TmpVec(Res.getLLTTy(*getMRI()).getNumElements(), Src); 6840b57cec5SDimitry Andric return buildInstr(TargetOpcode::G_BUILD_VECTOR, Res, TmpVec); 6850b57cec5SDimitry Andric } 6860b57cec5SDimitry Andric 6870b57cec5SDimitry Andric MachineInstrBuilder 6880b57cec5SDimitry Andric MachineIRBuilder::buildBuildVectorTrunc(const DstOp &Res, 6890b57cec5SDimitry Andric ArrayRef<Register> Ops) { 6900b57cec5SDimitry Andric // Unfortunately to convert from ArrayRef<Register> to ArrayRef<SrcOp>, 6910b57cec5SDimitry Andric // we need some temporary storage for the DstOp objects. Here we use a 6920b57cec5SDimitry Andric // sufficiently large SmallVector to not go through the heap. 6930b57cec5SDimitry Andric SmallVector<SrcOp, 8> TmpVec(Ops.begin(), Ops.end()); 6940b57cec5SDimitry Andric return buildInstr(TargetOpcode::G_BUILD_VECTOR_TRUNC, Res, TmpVec); 6950b57cec5SDimitry Andric } 6960b57cec5SDimitry Andric 697e8d8bef9SDimitry Andric MachineInstrBuilder MachineIRBuilder::buildShuffleSplat(const DstOp &Res, 698e8d8bef9SDimitry Andric const SrcOp &Src) { 699e8d8bef9SDimitry Andric LLT DstTy = Res.getLLTTy(*getMRI()); 700e8d8bef9SDimitry Andric assert(Src.getLLTTy(*getMRI()) == DstTy.getElementType() && 701e8d8bef9SDimitry Andric "Expected Src to match Dst elt ty"); 702e8d8bef9SDimitry Andric auto UndefVec = buildUndef(DstTy); 703e8d8bef9SDimitry Andric auto Zero = buildConstant(LLT::scalar(64), 0); 704e8d8bef9SDimitry Andric auto InsElt = buildInsertVectorElement(DstTy, UndefVec, Src, Zero); 705e8d8bef9SDimitry Andric SmallVector<int, 16> ZeroMask(DstTy.getNumElements()); 706e8d8bef9SDimitry Andric return buildShuffleVector(DstTy, InsElt, UndefVec, ZeroMask); 707e8d8bef9SDimitry Andric } 708e8d8bef9SDimitry Andric 709e8d8bef9SDimitry Andric MachineInstrBuilder MachineIRBuilder::buildShuffleVector(const DstOp &Res, 710e8d8bef9SDimitry Andric const SrcOp &Src1, 711e8d8bef9SDimitry Andric const SrcOp &Src2, 712e8d8bef9SDimitry Andric ArrayRef<int> Mask) { 713e8d8bef9SDimitry Andric LLT DstTy = Res.getLLTTy(*getMRI()); 714e8d8bef9SDimitry Andric LLT Src1Ty = Src1.getLLTTy(*getMRI()); 715e8d8bef9SDimitry Andric LLT Src2Ty = Src2.getLLTTy(*getMRI()); 716349cc55cSDimitry Andric assert((size_t)(Src1Ty.getNumElements() + Src2Ty.getNumElements()) >= 717349cc55cSDimitry Andric Mask.size()); 718e8d8bef9SDimitry Andric assert(DstTy.getElementType() == Src1Ty.getElementType() && 719e8d8bef9SDimitry Andric DstTy.getElementType() == Src2Ty.getElementType()); 720fe6060f1SDimitry Andric (void)DstTy; 721e8d8bef9SDimitry Andric (void)Src1Ty; 722e8d8bef9SDimitry Andric (void)Src2Ty; 723e8d8bef9SDimitry Andric ArrayRef<int> MaskAlloc = getMF().allocateShuffleMask(Mask); 724fe6060f1SDimitry Andric return buildInstr(TargetOpcode::G_SHUFFLE_VECTOR, {Res}, {Src1, Src2}) 725e8d8bef9SDimitry Andric .addShuffleMask(MaskAlloc); 726e8d8bef9SDimitry Andric } 727e8d8bef9SDimitry Andric 7280b57cec5SDimitry Andric MachineInstrBuilder 7290b57cec5SDimitry Andric MachineIRBuilder::buildConcatVectors(const DstOp &Res, ArrayRef<Register> Ops) { 7300b57cec5SDimitry Andric // Unfortunately to convert from ArrayRef<Register> to ArrayRef<SrcOp>, 7310b57cec5SDimitry Andric // we need some temporary storage for the DstOp objects. Here we use a 7320b57cec5SDimitry Andric // sufficiently large SmallVector to not go through the heap. 7330b57cec5SDimitry Andric SmallVector<SrcOp, 8> TmpVec(Ops.begin(), Ops.end()); 7340b57cec5SDimitry Andric return buildInstr(TargetOpcode::G_CONCAT_VECTORS, Res, TmpVec); 7350b57cec5SDimitry Andric } 7360b57cec5SDimitry Andric 7375ffd83dbSDimitry Andric MachineInstrBuilder MachineIRBuilder::buildInsert(const DstOp &Res, 7385ffd83dbSDimitry Andric const SrcOp &Src, 7395ffd83dbSDimitry Andric const SrcOp &Op, 7405ffd83dbSDimitry Andric unsigned Index) { 7415ffd83dbSDimitry Andric assert(Index + Op.getLLTTy(*getMRI()).getSizeInBits() <= 7425ffd83dbSDimitry Andric Res.getLLTTy(*getMRI()).getSizeInBits() && 7430b57cec5SDimitry Andric "insertion past the end of a register"); 7440b57cec5SDimitry Andric 7455ffd83dbSDimitry Andric if (Res.getLLTTy(*getMRI()).getSizeInBits() == 7465ffd83dbSDimitry Andric Op.getLLTTy(*getMRI()).getSizeInBits()) { 7470b57cec5SDimitry Andric return buildCast(Res, Op); 7480b57cec5SDimitry Andric } 7490b57cec5SDimitry Andric 7505ffd83dbSDimitry Andric return buildInstr(TargetOpcode::G_INSERT, Res, {Src, Op, uint64_t(Index)}); 7510b57cec5SDimitry Andric } 7520b57cec5SDimitry Andric 7530b57cec5SDimitry Andric MachineInstrBuilder MachineIRBuilder::buildIntrinsic(Intrinsic::ID ID, 7540b57cec5SDimitry Andric ArrayRef<Register> ResultRegs, 7550b57cec5SDimitry Andric bool HasSideEffects) { 7560b57cec5SDimitry Andric auto MIB = 7570b57cec5SDimitry Andric buildInstr(HasSideEffects ? TargetOpcode::G_INTRINSIC_W_SIDE_EFFECTS 7580b57cec5SDimitry Andric : TargetOpcode::G_INTRINSIC); 7590b57cec5SDimitry Andric for (unsigned ResultReg : ResultRegs) 7600b57cec5SDimitry Andric MIB.addDef(ResultReg); 7610b57cec5SDimitry Andric MIB.addIntrinsicID(ID); 7620b57cec5SDimitry Andric return MIB; 7630b57cec5SDimitry Andric } 7640b57cec5SDimitry Andric 7650b57cec5SDimitry Andric MachineInstrBuilder MachineIRBuilder::buildIntrinsic(Intrinsic::ID ID, 7660b57cec5SDimitry Andric ArrayRef<DstOp> Results, 7670b57cec5SDimitry Andric bool HasSideEffects) { 7680b57cec5SDimitry Andric auto MIB = 7690b57cec5SDimitry Andric buildInstr(HasSideEffects ? TargetOpcode::G_INTRINSIC_W_SIDE_EFFECTS 7700b57cec5SDimitry Andric : TargetOpcode::G_INTRINSIC); 7710b57cec5SDimitry Andric for (DstOp Result : Results) 7720b57cec5SDimitry Andric Result.addDefToMIB(*getMRI(), MIB); 7730b57cec5SDimitry Andric MIB.addIntrinsicID(ID); 7740b57cec5SDimitry Andric return MIB; 7750b57cec5SDimitry Andric } 7760b57cec5SDimitry Andric 7770b57cec5SDimitry Andric MachineInstrBuilder MachineIRBuilder::buildTrunc(const DstOp &Res, 7780b57cec5SDimitry Andric const SrcOp &Op) { 7790b57cec5SDimitry Andric return buildInstr(TargetOpcode::G_TRUNC, Res, Op); 7800b57cec5SDimitry Andric } 7810b57cec5SDimitry Andric 7820b57cec5SDimitry Andric MachineInstrBuilder MachineIRBuilder::buildFPTrunc(const DstOp &Res, 783480093f4SDimitry Andric const SrcOp &Op, 784480093f4SDimitry Andric Optional<unsigned> Flags) { 785480093f4SDimitry Andric return buildInstr(TargetOpcode::G_FPTRUNC, Res, Op, Flags); 7860b57cec5SDimitry Andric } 7870b57cec5SDimitry Andric 7880b57cec5SDimitry Andric MachineInstrBuilder MachineIRBuilder::buildICmp(CmpInst::Predicate Pred, 7890b57cec5SDimitry Andric const DstOp &Res, 7900b57cec5SDimitry Andric const SrcOp &Op0, 7910b57cec5SDimitry Andric const SrcOp &Op1) { 7920b57cec5SDimitry Andric return buildInstr(TargetOpcode::G_ICMP, Res, {Pred, Op0, Op1}); 7930b57cec5SDimitry Andric } 7940b57cec5SDimitry Andric 7950b57cec5SDimitry Andric MachineInstrBuilder MachineIRBuilder::buildFCmp(CmpInst::Predicate Pred, 7960b57cec5SDimitry Andric const DstOp &Res, 7970b57cec5SDimitry Andric const SrcOp &Op0, 7988bcb0991SDimitry Andric const SrcOp &Op1, 7998bcb0991SDimitry Andric Optional<unsigned> Flags) { 8000b57cec5SDimitry Andric 8018bcb0991SDimitry Andric return buildInstr(TargetOpcode::G_FCMP, Res, {Pred, Op0, Op1}, Flags); 8020b57cec5SDimitry Andric } 8030b57cec5SDimitry Andric 8040b57cec5SDimitry Andric MachineInstrBuilder MachineIRBuilder::buildSelect(const DstOp &Res, 8050b57cec5SDimitry Andric const SrcOp &Tst, 8060b57cec5SDimitry Andric const SrcOp &Op0, 8078bcb0991SDimitry Andric const SrcOp &Op1, 8088bcb0991SDimitry Andric Optional<unsigned> Flags) { 8090b57cec5SDimitry Andric 8108bcb0991SDimitry Andric return buildInstr(TargetOpcode::G_SELECT, {Res}, {Tst, Op0, Op1}, Flags); 8110b57cec5SDimitry Andric } 8120b57cec5SDimitry Andric 8130b57cec5SDimitry Andric MachineInstrBuilder 8140b57cec5SDimitry Andric MachineIRBuilder::buildInsertVectorElement(const DstOp &Res, const SrcOp &Val, 8150b57cec5SDimitry Andric const SrcOp &Elt, const SrcOp &Idx) { 8160b57cec5SDimitry Andric return buildInstr(TargetOpcode::G_INSERT_VECTOR_ELT, Res, {Val, Elt, Idx}); 8170b57cec5SDimitry Andric } 8180b57cec5SDimitry Andric 8190b57cec5SDimitry Andric MachineInstrBuilder 8200b57cec5SDimitry Andric MachineIRBuilder::buildExtractVectorElement(const DstOp &Res, const SrcOp &Val, 8210b57cec5SDimitry Andric const SrcOp &Idx) { 8220b57cec5SDimitry Andric return buildInstr(TargetOpcode::G_EXTRACT_VECTOR_ELT, Res, {Val, Idx}); 8230b57cec5SDimitry Andric } 8240b57cec5SDimitry Andric 8250b57cec5SDimitry Andric MachineInstrBuilder MachineIRBuilder::buildAtomicCmpXchgWithSuccess( 8260b57cec5SDimitry Andric Register OldValRes, Register SuccessRes, Register Addr, Register CmpVal, 8270b57cec5SDimitry Andric Register NewVal, MachineMemOperand &MMO) { 8280b57cec5SDimitry Andric #ifndef NDEBUG 8290b57cec5SDimitry Andric LLT OldValResTy = getMRI()->getType(OldValRes); 8300b57cec5SDimitry Andric LLT SuccessResTy = getMRI()->getType(SuccessRes); 8310b57cec5SDimitry Andric LLT AddrTy = getMRI()->getType(Addr); 8320b57cec5SDimitry Andric LLT CmpValTy = getMRI()->getType(CmpVal); 8330b57cec5SDimitry Andric LLT NewValTy = getMRI()->getType(NewVal); 8340b57cec5SDimitry Andric assert(OldValResTy.isScalar() && "invalid operand type"); 8350b57cec5SDimitry Andric assert(SuccessResTy.isScalar() && "invalid operand type"); 8360b57cec5SDimitry Andric assert(AddrTy.isPointer() && "invalid operand type"); 8370b57cec5SDimitry Andric assert(CmpValTy.isValid() && "invalid operand type"); 8380b57cec5SDimitry Andric assert(NewValTy.isValid() && "invalid operand type"); 8390b57cec5SDimitry Andric assert(OldValResTy == CmpValTy && "type mismatch"); 8400b57cec5SDimitry Andric assert(OldValResTy == NewValTy && "type mismatch"); 8410b57cec5SDimitry Andric #endif 8420b57cec5SDimitry Andric 8430b57cec5SDimitry Andric return buildInstr(TargetOpcode::G_ATOMIC_CMPXCHG_WITH_SUCCESS) 8440b57cec5SDimitry Andric .addDef(OldValRes) 8450b57cec5SDimitry Andric .addDef(SuccessRes) 8460b57cec5SDimitry Andric .addUse(Addr) 8470b57cec5SDimitry Andric .addUse(CmpVal) 8480b57cec5SDimitry Andric .addUse(NewVal) 8490b57cec5SDimitry Andric .addMemOperand(&MMO); 8500b57cec5SDimitry Andric } 8510b57cec5SDimitry Andric 8520b57cec5SDimitry Andric MachineInstrBuilder 8530b57cec5SDimitry Andric MachineIRBuilder::buildAtomicCmpXchg(Register OldValRes, Register Addr, 8540b57cec5SDimitry Andric Register CmpVal, Register NewVal, 8550b57cec5SDimitry Andric MachineMemOperand &MMO) { 8560b57cec5SDimitry Andric #ifndef NDEBUG 8570b57cec5SDimitry Andric LLT OldValResTy = getMRI()->getType(OldValRes); 8580b57cec5SDimitry Andric LLT AddrTy = getMRI()->getType(Addr); 8590b57cec5SDimitry Andric LLT CmpValTy = getMRI()->getType(CmpVal); 8600b57cec5SDimitry Andric LLT NewValTy = getMRI()->getType(NewVal); 8610b57cec5SDimitry Andric assert(OldValResTy.isScalar() && "invalid operand type"); 8620b57cec5SDimitry Andric assert(AddrTy.isPointer() && "invalid operand type"); 8630b57cec5SDimitry Andric assert(CmpValTy.isValid() && "invalid operand type"); 8640b57cec5SDimitry Andric assert(NewValTy.isValid() && "invalid operand type"); 8650b57cec5SDimitry Andric assert(OldValResTy == CmpValTy && "type mismatch"); 8660b57cec5SDimitry Andric assert(OldValResTy == NewValTy && "type mismatch"); 8670b57cec5SDimitry Andric #endif 8680b57cec5SDimitry Andric 8690b57cec5SDimitry Andric return buildInstr(TargetOpcode::G_ATOMIC_CMPXCHG) 8700b57cec5SDimitry Andric .addDef(OldValRes) 8710b57cec5SDimitry Andric .addUse(Addr) 8720b57cec5SDimitry Andric .addUse(CmpVal) 8730b57cec5SDimitry Andric .addUse(NewVal) 8740b57cec5SDimitry Andric .addMemOperand(&MMO); 8750b57cec5SDimitry Andric } 8760b57cec5SDimitry Andric 8778bcb0991SDimitry Andric MachineInstrBuilder MachineIRBuilder::buildAtomicRMW( 8788bcb0991SDimitry Andric unsigned Opcode, const DstOp &OldValRes, 8798bcb0991SDimitry Andric const SrcOp &Addr, const SrcOp &Val, 8800b57cec5SDimitry Andric MachineMemOperand &MMO) { 8818bcb0991SDimitry Andric 8820b57cec5SDimitry Andric #ifndef NDEBUG 8838bcb0991SDimitry Andric LLT OldValResTy = OldValRes.getLLTTy(*getMRI()); 8848bcb0991SDimitry Andric LLT AddrTy = Addr.getLLTTy(*getMRI()); 8858bcb0991SDimitry Andric LLT ValTy = Val.getLLTTy(*getMRI()); 8860b57cec5SDimitry Andric assert(OldValResTy.isScalar() && "invalid operand type"); 8870b57cec5SDimitry Andric assert(AddrTy.isPointer() && "invalid operand type"); 8880b57cec5SDimitry Andric assert(ValTy.isValid() && "invalid operand type"); 8890b57cec5SDimitry Andric assert(OldValResTy == ValTy && "type mismatch"); 8908bcb0991SDimitry Andric assert(MMO.isAtomic() && "not atomic mem operand"); 8910b57cec5SDimitry Andric #endif 8920b57cec5SDimitry Andric 8938bcb0991SDimitry Andric auto MIB = buildInstr(Opcode); 8948bcb0991SDimitry Andric OldValRes.addDefToMIB(*getMRI(), MIB); 8958bcb0991SDimitry Andric Addr.addSrcToMIB(MIB); 8968bcb0991SDimitry Andric Val.addSrcToMIB(MIB); 8978bcb0991SDimitry Andric MIB.addMemOperand(&MMO); 8988bcb0991SDimitry Andric return MIB; 8990b57cec5SDimitry Andric } 9000b57cec5SDimitry Andric 9010b57cec5SDimitry Andric MachineInstrBuilder 9020b57cec5SDimitry Andric MachineIRBuilder::buildAtomicRMWXchg(Register OldValRes, Register Addr, 9030b57cec5SDimitry Andric Register Val, MachineMemOperand &MMO) { 9040b57cec5SDimitry Andric return buildAtomicRMW(TargetOpcode::G_ATOMICRMW_XCHG, OldValRes, Addr, Val, 9050b57cec5SDimitry Andric MMO); 9060b57cec5SDimitry Andric } 9070b57cec5SDimitry Andric MachineInstrBuilder 9080b57cec5SDimitry Andric MachineIRBuilder::buildAtomicRMWAdd(Register OldValRes, Register Addr, 9090b57cec5SDimitry Andric Register Val, MachineMemOperand &MMO) { 9100b57cec5SDimitry Andric return buildAtomicRMW(TargetOpcode::G_ATOMICRMW_ADD, OldValRes, Addr, Val, 9110b57cec5SDimitry Andric MMO); 9120b57cec5SDimitry Andric } 9130b57cec5SDimitry Andric MachineInstrBuilder 9140b57cec5SDimitry Andric MachineIRBuilder::buildAtomicRMWSub(Register OldValRes, Register Addr, 9150b57cec5SDimitry Andric Register Val, MachineMemOperand &MMO) { 9160b57cec5SDimitry Andric return buildAtomicRMW(TargetOpcode::G_ATOMICRMW_SUB, OldValRes, Addr, Val, 9170b57cec5SDimitry Andric MMO); 9180b57cec5SDimitry Andric } 9190b57cec5SDimitry Andric MachineInstrBuilder 9200b57cec5SDimitry Andric MachineIRBuilder::buildAtomicRMWAnd(Register OldValRes, Register Addr, 9210b57cec5SDimitry Andric Register Val, MachineMemOperand &MMO) { 9220b57cec5SDimitry Andric return buildAtomicRMW(TargetOpcode::G_ATOMICRMW_AND, OldValRes, Addr, Val, 9230b57cec5SDimitry Andric MMO); 9240b57cec5SDimitry Andric } 9250b57cec5SDimitry Andric MachineInstrBuilder 9260b57cec5SDimitry Andric MachineIRBuilder::buildAtomicRMWNand(Register OldValRes, Register Addr, 9270b57cec5SDimitry Andric Register Val, MachineMemOperand &MMO) { 9280b57cec5SDimitry Andric return buildAtomicRMW(TargetOpcode::G_ATOMICRMW_NAND, OldValRes, Addr, Val, 9290b57cec5SDimitry Andric MMO); 9300b57cec5SDimitry Andric } 9310b57cec5SDimitry Andric MachineInstrBuilder MachineIRBuilder::buildAtomicRMWOr(Register OldValRes, 9320b57cec5SDimitry Andric Register Addr, 9330b57cec5SDimitry Andric Register Val, 9340b57cec5SDimitry Andric MachineMemOperand &MMO) { 9350b57cec5SDimitry Andric return buildAtomicRMW(TargetOpcode::G_ATOMICRMW_OR, OldValRes, Addr, Val, 9360b57cec5SDimitry Andric MMO); 9370b57cec5SDimitry Andric } 9380b57cec5SDimitry Andric MachineInstrBuilder 9390b57cec5SDimitry Andric MachineIRBuilder::buildAtomicRMWXor(Register OldValRes, Register Addr, 9400b57cec5SDimitry Andric Register Val, MachineMemOperand &MMO) { 9410b57cec5SDimitry Andric return buildAtomicRMW(TargetOpcode::G_ATOMICRMW_XOR, OldValRes, Addr, Val, 9420b57cec5SDimitry Andric MMO); 9430b57cec5SDimitry Andric } 9440b57cec5SDimitry Andric MachineInstrBuilder 9450b57cec5SDimitry Andric MachineIRBuilder::buildAtomicRMWMax(Register OldValRes, Register Addr, 9460b57cec5SDimitry Andric Register Val, MachineMemOperand &MMO) { 9470b57cec5SDimitry Andric return buildAtomicRMW(TargetOpcode::G_ATOMICRMW_MAX, OldValRes, Addr, Val, 9480b57cec5SDimitry Andric MMO); 9490b57cec5SDimitry Andric } 9500b57cec5SDimitry Andric MachineInstrBuilder 9510b57cec5SDimitry Andric MachineIRBuilder::buildAtomicRMWMin(Register OldValRes, Register Addr, 9520b57cec5SDimitry Andric Register Val, MachineMemOperand &MMO) { 9530b57cec5SDimitry Andric return buildAtomicRMW(TargetOpcode::G_ATOMICRMW_MIN, OldValRes, Addr, Val, 9540b57cec5SDimitry Andric MMO); 9550b57cec5SDimitry Andric } 9560b57cec5SDimitry Andric MachineInstrBuilder 9570b57cec5SDimitry Andric MachineIRBuilder::buildAtomicRMWUmax(Register OldValRes, Register Addr, 9580b57cec5SDimitry Andric Register Val, MachineMemOperand &MMO) { 9590b57cec5SDimitry Andric return buildAtomicRMW(TargetOpcode::G_ATOMICRMW_UMAX, OldValRes, Addr, Val, 9600b57cec5SDimitry Andric MMO); 9610b57cec5SDimitry Andric } 9620b57cec5SDimitry Andric MachineInstrBuilder 9630b57cec5SDimitry Andric MachineIRBuilder::buildAtomicRMWUmin(Register OldValRes, Register Addr, 9640b57cec5SDimitry Andric Register Val, MachineMemOperand &MMO) { 9650b57cec5SDimitry Andric return buildAtomicRMW(TargetOpcode::G_ATOMICRMW_UMIN, OldValRes, Addr, Val, 9660b57cec5SDimitry Andric MMO); 9670b57cec5SDimitry Andric } 9680b57cec5SDimitry Andric 9690b57cec5SDimitry Andric MachineInstrBuilder 9708bcb0991SDimitry Andric MachineIRBuilder::buildAtomicRMWFAdd( 9718bcb0991SDimitry Andric const DstOp &OldValRes, const SrcOp &Addr, const SrcOp &Val, 9728bcb0991SDimitry Andric MachineMemOperand &MMO) { 9738bcb0991SDimitry Andric return buildAtomicRMW(TargetOpcode::G_ATOMICRMW_FADD, OldValRes, Addr, Val, 9748bcb0991SDimitry Andric MMO); 9758bcb0991SDimitry Andric } 9768bcb0991SDimitry Andric 9778bcb0991SDimitry Andric MachineInstrBuilder 9788bcb0991SDimitry Andric MachineIRBuilder::buildAtomicRMWFSub(const DstOp &OldValRes, const SrcOp &Addr, const SrcOp &Val, 9798bcb0991SDimitry Andric MachineMemOperand &MMO) { 9808bcb0991SDimitry Andric return buildAtomicRMW(TargetOpcode::G_ATOMICRMW_FSUB, OldValRes, Addr, Val, 9818bcb0991SDimitry Andric MMO); 9828bcb0991SDimitry Andric } 9838bcb0991SDimitry Andric 9848bcb0991SDimitry Andric MachineInstrBuilder 9850b57cec5SDimitry Andric MachineIRBuilder::buildFence(unsigned Ordering, unsigned Scope) { 9860b57cec5SDimitry Andric return buildInstr(TargetOpcode::G_FENCE) 9870b57cec5SDimitry Andric .addImm(Ordering) 9880b57cec5SDimitry Andric .addImm(Scope); 9890b57cec5SDimitry Andric } 9900b57cec5SDimitry Andric 9910b57cec5SDimitry Andric MachineInstrBuilder 9920b57cec5SDimitry Andric MachineIRBuilder::buildBlockAddress(Register Res, const BlockAddress *BA) { 9930b57cec5SDimitry Andric #ifndef NDEBUG 9940b57cec5SDimitry Andric assert(getMRI()->getType(Res).isPointer() && "invalid res type"); 9950b57cec5SDimitry Andric #endif 9960b57cec5SDimitry Andric 9970b57cec5SDimitry Andric return buildInstr(TargetOpcode::G_BLOCK_ADDR).addDef(Res).addBlockAddress(BA); 9980b57cec5SDimitry Andric } 9990b57cec5SDimitry Andric 10005ffd83dbSDimitry Andric void MachineIRBuilder::validateTruncExt(const LLT DstTy, const LLT SrcTy, 10010b57cec5SDimitry Andric bool IsExtend) { 10020b57cec5SDimitry Andric #ifndef NDEBUG 10030b57cec5SDimitry Andric if (DstTy.isVector()) { 10040b57cec5SDimitry Andric assert(SrcTy.isVector() && "mismatched cast between vector and non-vector"); 10050b57cec5SDimitry Andric assert(SrcTy.getNumElements() == DstTy.getNumElements() && 10060b57cec5SDimitry Andric "different number of elements in a trunc/ext"); 10070b57cec5SDimitry Andric } else 10080b57cec5SDimitry Andric assert(DstTy.isScalar() && SrcTy.isScalar() && "invalid extend/trunc"); 10090b57cec5SDimitry Andric 10100b57cec5SDimitry Andric if (IsExtend) 10110b57cec5SDimitry Andric assert(DstTy.getSizeInBits() > SrcTy.getSizeInBits() && 10120b57cec5SDimitry Andric "invalid narrowing extend"); 10130b57cec5SDimitry Andric else 10140b57cec5SDimitry Andric assert(DstTy.getSizeInBits() < SrcTy.getSizeInBits() && 10150b57cec5SDimitry Andric "invalid widening trunc"); 10160b57cec5SDimitry Andric #endif 10170b57cec5SDimitry Andric } 10180b57cec5SDimitry Andric 10195ffd83dbSDimitry Andric void MachineIRBuilder::validateSelectOp(const LLT ResTy, const LLT TstTy, 10205ffd83dbSDimitry Andric const LLT Op0Ty, const LLT Op1Ty) { 10210b57cec5SDimitry Andric #ifndef NDEBUG 10220b57cec5SDimitry Andric assert((ResTy.isScalar() || ResTy.isVector() || ResTy.isPointer()) && 10230b57cec5SDimitry Andric "invalid operand type"); 10240b57cec5SDimitry Andric assert((ResTy == Op0Ty && ResTy == Op1Ty) && "type mismatch"); 10250b57cec5SDimitry Andric if (ResTy.isScalar() || ResTy.isPointer()) 10260b57cec5SDimitry Andric assert(TstTy.isScalar() && "type mismatch"); 10270b57cec5SDimitry Andric else 10280b57cec5SDimitry Andric assert((TstTy.isScalar() || 10290b57cec5SDimitry Andric (TstTy.isVector() && 10300b57cec5SDimitry Andric TstTy.getNumElements() == Op0Ty.getNumElements())) && 10310b57cec5SDimitry Andric "type mismatch"); 10320b57cec5SDimitry Andric #endif 10330b57cec5SDimitry Andric } 10340b57cec5SDimitry Andric 10350b57cec5SDimitry Andric MachineInstrBuilder MachineIRBuilder::buildInstr(unsigned Opc, 10360b57cec5SDimitry Andric ArrayRef<DstOp> DstOps, 10370b57cec5SDimitry Andric ArrayRef<SrcOp> SrcOps, 10380b57cec5SDimitry Andric Optional<unsigned> Flags) { 10390b57cec5SDimitry Andric switch (Opc) { 10400b57cec5SDimitry Andric default: 10410b57cec5SDimitry Andric break; 10420b57cec5SDimitry Andric case TargetOpcode::G_SELECT: { 10430b57cec5SDimitry Andric assert(DstOps.size() == 1 && "Invalid select"); 10440b57cec5SDimitry Andric assert(SrcOps.size() == 3 && "Invalid select"); 10450b57cec5SDimitry Andric validateSelectOp( 10460b57cec5SDimitry Andric DstOps[0].getLLTTy(*getMRI()), SrcOps[0].getLLTTy(*getMRI()), 10470b57cec5SDimitry Andric SrcOps[1].getLLTTy(*getMRI()), SrcOps[2].getLLTTy(*getMRI())); 10480b57cec5SDimitry Andric break; 10490b57cec5SDimitry Andric } 1050e8d8bef9SDimitry Andric case TargetOpcode::G_FNEG: 1051e8d8bef9SDimitry Andric case TargetOpcode::G_ABS: 1052e8d8bef9SDimitry Andric // All these are unary ops. 1053e8d8bef9SDimitry Andric assert(DstOps.size() == 1 && "Invalid Dst"); 1054e8d8bef9SDimitry Andric assert(SrcOps.size() == 1 && "Invalid Srcs"); 1055e8d8bef9SDimitry Andric validateUnaryOp(DstOps[0].getLLTTy(*getMRI()), 1056e8d8bef9SDimitry Andric SrcOps[0].getLLTTy(*getMRI())); 1057e8d8bef9SDimitry Andric break; 10580b57cec5SDimitry Andric case TargetOpcode::G_ADD: 10590b57cec5SDimitry Andric case TargetOpcode::G_AND: 10600b57cec5SDimitry Andric case TargetOpcode::G_MUL: 10610b57cec5SDimitry Andric case TargetOpcode::G_OR: 10620b57cec5SDimitry Andric case TargetOpcode::G_SUB: 10630b57cec5SDimitry Andric case TargetOpcode::G_XOR: 10640b57cec5SDimitry Andric case TargetOpcode::G_UDIV: 10650b57cec5SDimitry Andric case TargetOpcode::G_SDIV: 10660b57cec5SDimitry Andric case TargetOpcode::G_UREM: 10670b57cec5SDimitry Andric case TargetOpcode::G_SREM: 10680b57cec5SDimitry Andric case TargetOpcode::G_SMIN: 10690b57cec5SDimitry Andric case TargetOpcode::G_SMAX: 10700b57cec5SDimitry Andric case TargetOpcode::G_UMIN: 10715ffd83dbSDimitry Andric case TargetOpcode::G_UMAX: 10725ffd83dbSDimitry Andric case TargetOpcode::G_UADDSAT: 10735ffd83dbSDimitry Andric case TargetOpcode::G_SADDSAT: 10745ffd83dbSDimitry Andric case TargetOpcode::G_USUBSAT: 10755ffd83dbSDimitry Andric case TargetOpcode::G_SSUBSAT: { 10760b57cec5SDimitry Andric // All these are binary ops. 10770b57cec5SDimitry Andric assert(DstOps.size() == 1 && "Invalid Dst"); 10780b57cec5SDimitry Andric assert(SrcOps.size() == 2 && "Invalid Srcs"); 10790b57cec5SDimitry Andric validateBinaryOp(DstOps[0].getLLTTy(*getMRI()), 10800b57cec5SDimitry Andric SrcOps[0].getLLTTy(*getMRI()), 10810b57cec5SDimitry Andric SrcOps[1].getLLTTy(*getMRI())); 10820b57cec5SDimitry Andric break; 10830b57cec5SDimitry Andric } 10840b57cec5SDimitry Andric case TargetOpcode::G_SHL: 10850b57cec5SDimitry Andric case TargetOpcode::G_ASHR: 1086e8d8bef9SDimitry Andric case TargetOpcode::G_LSHR: 1087e8d8bef9SDimitry Andric case TargetOpcode::G_USHLSAT: 1088e8d8bef9SDimitry Andric case TargetOpcode::G_SSHLSAT: { 10890b57cec5SDimitry Andric assert(DstOps.size() == 1 && "Invalid Dst"); 10900b57cec5SDimitry Andric assert(SrcOps.size() == 2 && "Invalid Srcs"); 10910b57cec5SDimitry Andric validateShiftOp(DstOps[0].getLLTTy(*getMRI()), 10920b57cec5SDimitry Andric SrcOps[0].getLLTTy(*getMRI()), 10930b57cec5SDimitry Andric SrcOps[1].getLLTTy(*getMRI())); 10940b57cec5SDimitry Andric break; 10950b57cec5SDimitry Andric } 10960b57cec5SDimitry Andric case TargetOpcode::G_SEXT: 10970b57cec5SDimitry Andric case TargetOpcode::G_ZEXT: 10980b57cec5SDimitry Andric case TargetOpcode::G_ANYEXT: 10990b57cec5SDimitry Andric assert(DstOps.size() == 1 && "Invalid Dst"); 11000b57cec5SDimitry Andric assert(SrcOps.size() == 1 && "Invalid Srcs"); 11010b57cec5SDimitry Andric validateTruncExt(DstOps[0].getLLTTy(*getMRI()), 11020b57cec5SDimitry Andric SrcOps[0].getLLTTy(*getMRI()), true); 11030b57cec5SDimitry Andric break; 11040b57cec5SDimitry Andric case TargetOpcode::G_TRUNC: 11050b57cec5SDimitry Andric case TargetOpcode::G_FPTRUNC: { 11060b57cec5SDimitry Andric assert(DstOps.size() == 1 && "Invalid Dst"); 11070b57cec5SDimitry Andric assert(SrcOps.size() == 1 && "Invalid Srcs"); 11080b57cec5SDimitry Andric validateTruncExt(DstOps[0].getLLTTy(*getMRI()), 11090b57cec5SDimitry Andric SrcOps[0].getLLTTy(*getMRI()), false); 11100b57cec5SDimitry Andric break; 11110b57cec5SDimitry Andric } 11125ffd83dbSDimitry Andric case TargetOpcode::G_BITCAST: { 11135ffd83dbSDimitry Andric assert(DstOps.size() == 1 && "Invalid Dst"); 11145ffd83dbSDimitry Andric assert(SrcOps.size() == 1 && "Invalid Srcs"); 11155ffd83dbSDimitry Andric assert(DstOps[0].getLLTTy(*getMRI()).getSizeInBits() == 11165ffd83dbSDimitry Andric SrcOps[0].getLLTTy(*getMRI()).getSizeInBits() && "invalid bitcast"); 11175ffd83dbSDimitry Andric break; 11185ffd83dbSDimitry Andric } 11190b57cec5SDimitry Andric case TargetOpcode::COPY: 11200b57cec5SDimitry Andric assert(DstOps.size() == 1 && "Invalid Dst"); 11210b57cec5SDimitry Andric // If the caller wants to add a subreg source it has to be done separately 11220b57cec5SDimitry Andric // so we may not have any SrcOps at this point yet. 11230b57cec5SDimitry Andric break; 11240b57cec5SDimitry Andric case TargetOpcode::G_FCMP: 11250b57cec5SDimitry Andric case TargetOpcode::G_ICMP: { 11260b57cec5SDimitry Andric assert(DstOps.size() == 1 && "Invalid Dst Operands"); 11270b57cec5SDimitry Andric assert(SrcOps.size() == 3 && "Invalid Src Operands"); 11280b57cec5SDimitry Andric // For F/ICMP, the first src operand is the predicate, followed by 11290b57cec5SDimitry Andric // the two comparands. 11300b57cec5SDimitry Andric assert(SrcOps[0].getSrcOpKind() == SrcOp::SrcType::Ty_Predicate && 11310b57cec5SDimitry Andric "Expecting predicate"); 11320b57cec5SDimitry Andric assert([&]() -> bool { 11330b57cec5SDimitry Andric CmpInst::Predicate Pred = SrcOps[0].getPredicate(); 11340b57cec5SDimitry Andric return Opc == TargetOpcode::G_ICMP ? CmpInst::isIntPredicate(Pred) 11350b57cec5SDimitry Andric : CmpInst::isFPPredicate(Pred); 11360b57cec5SDimitry Andric }() && "Invalid predicate"); 11370b57cec5SDimitry Andric assert(SrcOps[1].getLLTTy(*getMRI()) == SrcOps[2].getLLTTy(*getMRI()) && 11380b57cec5SDimitry Andric "Type mismatch"); 11390b57cec5SDimitry Andric assert([&]() -> bool { 11400b57cec5SDimitry Andric LLT Op0Ty = SrcOps[1].getLLTTy(*getMRI()); 11410b57cec5SDimitry Andric LLT DstTy = DstOps[0].getLLTTy(*getMRI()); 11420b57cec5SDimitry Andric if (Op0Ty.isScalar() || Op0Ty.isPointer()) 11430b57cec5SDimitry Andric return DstTy.isScalar(); 11440b57cec5SDimitry Andric else 11450b57cec5SDimitry Andric return DstTy.isVector() && 11460b57cec5SDimitry Andric DstTy.getNumElements() == Op0Ty.getNumElements(); 11470b57cec5SDimitry Andric }() && "Type Mismatch"); 11480b57cec5SDimitry Andric break; 11490b57cec5SDimitry Andric } 11500b57cec5SDimitry Andric case TargetOpcode::G_UNMERGE_VALUES: { 11510b57cec5SDimitry Andric assert(!DstOps.empty() && "Invalid trivial sequence"); 11520b57cec5SDimitry Andric assert(SrcOps.size() == 1 && "Invalid src for Unmerge"); 1153e8d8bef9SDimitry Andric assert(llvm::all_of(DstOps, 11540b57cec5SDimitry Andric [&, this](const DstOp &Op) { 11550b57cec5SDimitry Andric return Op.getLLTTy(*getMRI()) == 11560b57cec5SDimitry Andric DstOps[0].getLLTTy(*getMRI()); 11570b57cec5SDimitry Andric }) && 11580b57cec5SDimitry Andric "type mismatch in output list"); 1159fe6060f1SDimitry Andric assert((TypeSize::ScalarTy)DstOps.size() * 1160fe6060f1SDimitry Andric DstOps[0].getLLTTy(*getMRI()).getSizeInBits() == 11610b57cec5SDimitry Andric SrcOps[0].getLLTTy(*getMRI()).getSizeInBits() && 11620b57cec5SDimitry Andric "input operands do not cover output register"); 11630b57cec5SDimitry Andric break; 11640b57cec5SDimitry Andric } 11650b57cec5SDimitry Andric case TargetOpcode::G_MERGE_VALUES: { 11660b57cec5SDimitry Andric assert(!SrcOps.empty() && "invalid trivial sequence"); 11670b57cec5SDimitry Andric assert(DstOps.size() == 1 && "Invalid Dst"); 1168e8d8bef9SDimitry Andric assert(llvm::all_of(SrcOps, 11690b57cec5SDimitry Andric [&, this](const SrcOp &Op) { 11700b57cec5SDimitry Andric return Op.getLLTTy(*getMRI()) == 11710b57cec5SDimitry Andric SrcOps[0].getLLTTy(*getMRI()); 11720b57cec5SDimitry Andric }) && 11730b57cec5SDimitry Andric "type mismatch in input list"); 1174fe6060f1SDimitry Andric assert((TypeSize::ScalarTy)SrcOps.size() * 1175fe6060f1SDimitry Andric SrcOps[0].getLLTTy(*getMRI()).getSizeInBits() == 11760b57cec5SDimitry Andric DstOps[0].getLLTTy(*getMRI()).getSizeInBits() && 11770b57cec5SDimitry Andric "input operands do not cover output register"); 11780b57cec5SDimitry Andric if (SrcOps.size() == 1) 11790b57cec5SDimitry Andric return buildCast(DstOps[0], SrcOps[0]); 11808bcb0991SDimitry Andric if (DstOps[0].getLLTTy(*getMRI()).isVector()) { 11818bcb0991SDimitry Andric if (SrcOps[0].getLLTTy(*getMRI()).isVector()) 11820b57cec5SDimitry Andric return buildInstr(TargetOpcode::G_CONCAT_VECTORS, DstOps, SrcOps); 11838bcb0991SDimitry Andric return buildInstr(TargetOpcode::G_BUILD_VECTOR, DstOps, SrcOps); 11848bcb0991SDimitry Andric } 11850b57cec5SDimitry Andric break; 11860b57cec5SDimitry Andric } 11870b57cec5SDimitry Andric case TargetOpcode::G_EXTRACT_VECTOR_ELT: { 11880b57cec5SDimitry Andric assert(DstOps.size() == 1 && "Invalid Dst size"); 11890b57cec5SDimitry Andric assert(SrcOps.size() == 2 && "Invalid Src size"); 11900b57cec5SDimitry Andric assert(SrcOps[0].getLLTTy(*getMRI()).isVector() && "Invalid operand type"); 11910b57cec5SDimitry Andric assert((DstOps[0].getLLTTy(*getMRI()).isScalar() || 11920b57cec5SDimitry Andric DstOps[0].getLLTTy(*getMRI()).isPointer()) && 11930b57cec5SDimitry Andric "Invalid operand type"); 11940b57cec5SDimitry Andric assert(SrcOps[1].getLLTTy(*getMRI()).isScalar() && "Invalid operand type"); 11950b57cec5SDimitry Andric assert(SrcOps[0].getLLTTy(*getMRI()).getElementType() == 11960b57cec5SDimitry Andric DstOps[0].getLLTTy(*getMRI()) && 11970b57cec5SDimitry Andric "Type mismatch"); 11980b57cec5SDimitry Andric break; 11990b57cec5SDimitry Andric } 12000b57cec5SDimitry Andric case TargetOpcode::G_INSERT_VECTOR_ELT: { 12010b57cec5SDimitry Andric assert(DstOps.size() == 1 && "Invalid dst size"); 12020b57cec5SDimitry Andric assert(SrcOps.size() == 3 && "Invalid src size"); 12030b57cec5SDimitry Andric assert(DstOps[0].getLLTTy(*getMRI()).isVector() && 12040b57cec5SDimitry Andric SrcOps[0].getLLTTy(*getMRI()).isVector() && "Invalid operand type"); 12050b57cec5SDimitry Andric assert(DstOps[0].getLLTTy(*getMRI()).getElementType() == 12060b57cec5SDimitry Andric SrcOps[1].getLLTTy(*getMRI()) && 12070b57cec5SDimitry Andric "Type mismatch"); 12080b57cec5SDimitry Andric assert(SrcOps[2].getLLTTy(*getMRI()).isScalar() && "Invalid index"); 12090b57cec5SDimitry Andric assert(DstOps[0].getLLTTy(*getMRI()).getNumElements() == 12100b57cec5SDimitry Andric SrcOps[0].getLLTTy(*getMRI()).getNumElements() && 12110b57cec5SDimitry Andric "Type mismatch"); 12120b57cec5SDimitry Andric break; 12130b57cec5SDimitry Andric } 12140b57cec5SDimitry Andric case TargetOpcode::G_BUILD_VECTOR: { 12150b57cec5SDimitry Andric assert((!SrcOps.empty() || SrcOps.size() < 2) && 12160b57cec5SDimitry Andric "Must have at least 2 operands"); 12170b57cec5SDimitry Andric assert(DstOps.size() == 1 && "Invalid DstOps"); 12180b57cec5SDimitry Andric assert(DstOps[0].getLLTTy(*getMRI()).isVector() && 12190b57cec5SDimitry Andric "Res type must be a vector"); 1220e8d8bef9SDimitry Andric assert(llvm::all_of(SrcOps, 12210b57cec5SDimitry Andric [&, this](const SrcOp &Op) { 12220b57cec5SDimitry Andric return Op.getLLTTy(*getMRI()) == 12230b57cec5SDimitry Andric SrcOps[0].getLLTTy(*getMRI()); 12240b57cec5SDimitry Andric }) && 12250b57cec5SDimitry Andric "type mismatch in input list"); 1226fe6060f1SDimitry Andric assert((TypeSize::ScalarTy)SrcOps.size() * 1227fe6060f1SDimitry Andric SrcOps[0].getLLTTy(*getMRI()).getSizeInBits() == 12280b57cec5SDimitry Andric DstOps[0].getLLTTy(*getMRI()).getSizeInBits() && 12290b57cec5SDimitry Andric "input scalars do not exactly cover the output vector register"); 12300b57cec5SDimitry Andric break; 12310b57cec5SDimitry Andric } 12320b57cec5SDimitry Andric case TargetOpcode::G_BUILD_VECTOR_TRUNC: { 12330b57cec5SDimitry Andric assert((!SrcOps.empty() || SrcOps.size() < 2) && 12340b57cec5SDimitry Andric "Must have at least 2 operands"); 12350b57cec5SDimitry Andric assert(DstOps.size() == 1 && "Invalid DstOps"); 12360b57cec5SDimitry Andric assert(DstOps[0].getLLTTy(*getMRI()).isVector() && 12370b57cec5SDimitry Andric "Res type must be a vector"); 1238e8d8bef9SDimitry Andric assert(llvm::all_of(SrcOps, 12390b57cec5SDimitry Andric [&, this](const SrcOp &Op) { 12400b57cec5SDimitry Andric return Op.getLLTTy(*getMRI()) == 12410b57cec5SDimitry Andric SrcOps[0].getLLTTy(*getMRI()); 12420b57cec5SDimitry Andric }) && 12430b57cec5SDimitry Andric "type mismatch in input list"); 12440b57cec5SDimitry Andric if (SrcOps[0].getLLTTy(*getMRI()).getSizeInBits() == 12450b57cec5SDimitry Andric DstOps[0].getLLTTy(*getMRI()).getElementType().getSizeInBits()) 12460b57cec5SDimitry Andric return buildInstr(TargetOpcode::G_BUILD_VECTOR, DstOps, SrcOps); 12470b57cec5SDimitry Andric break; 12480b57cec5SDimitry Andric } 12490b57cec5SDimitry Andric case TargetOpcode::G_CONCAT_VECTORS: { 12500b57cec5SDimitry Andric assert(DstOps.size() == 1 && "Invalid DstOps"); 12510b57cec5SDimitry Andric assert((!SrcOps.empty() || SrcOps.size() < 2) && 12520b57cec5SDimitry Andric "Must have at least 2 operands"); 1253e8d8bef9SDimitry Andric assert(llvm::all_of(SrcOps, 12540b57cec5SDimitry Andric [&, this](const SrcOp &Op) { 12550b57cec5SDimitry Andric return (Op.getLLTTy(*getMRI()).isVector() && 12560b57cec5SDimitry Andric Op.getLLTTy(*getMRI()) == 12570b57cec5SDimitry Andric SrcOps[0].getLLTTy(*getMRI())); 12580b57cec5SDimitry Andric }) && 12590b57cec5SDimitry Andric "type mismatch in input list"); 1260fe6060f1SDimitry Andric assert((TypeSize::ScalarTy)SrcOps.size() * 1261fe6060f1SDimitry Andric SrcOps[0].getLLTTy(*getMRI()).getSizeInBits() == 12620b57cec5SDimitry Andric DstOps[0].getLLTTy(*getMRI()).getSizeInBits() && 12630b57cec5SDimitry Andric "input vectors do not exactly cover the output vector register"); 12640b57cec5SDimitry Andric break; 12650b57cec5SDimitry Andric } 12660b57cec5SDimitry Andric case TargetOpcode::G_UADDE: { 12670b57cec5SDimitry Andric assert(DstOps.size() == 2 && "Invalid no of dst operands"); 12680b57cec5SDimitry Andric assert(SrcOps.size() == 3 && "Invalid no of src operands"); 12690b57cec5SDimitry Andric assert(DstOps[0].getLLTTy(*getMRI()).isScalar() && "Invalid operand"); 12700b57cec5SDimitry Andric assert((DstOps[0].getLLTTy(*getMRI()) == SrcOps[0].getLLTTy(*getMRI())) && 12710b57cec5SDimitry Andric (DstOps[0].getLLTTy(*getMRI()) == SrcOps[1].getLLTTy(*getMRI())) && 12720b57cec5SDimitry Andric "Invalid operand"); 12730b57cec5SDimitry Andric assert(DstOps[1].getLLTTy(*getMRI()).isScalar() && "Invalid operand"); 12740b57cec5SDimitry Andric assert(DstOps[1].getLLTTy(*getMRI()) == SrcOps[2].getLLTTy(*getMRI()) && 12750b57cec5SDimitry Andric "type mismatch"); 12760b57cec5SDimitry Andric break; 12770b57cec5SDimitry Andric } 12780b57cec5SDimitry Andric } 12790b57cec5SDimitry Andric 12800b57cec5SDimitry Andric auto MIB = buildInstr(Opc); 12810b57cec5SDimitry Andric for (const DstOp &Op : DstOps) 12820b57cec5SDimitry Andric Op.addDefToMIB(*getMRI(), MIB); 12830b57cec5SDimitry Andric for (const SrcOp &Op : SrcOps) 12840b57cec5SDimitry Andric Op.addSrcToMIB(MIB); 12850b57cec5SDimitry Andric if (Flags) 12860b57cec5SDimitry Andric MIB->setFlags(*Flags); 12870b57cec5SDimitry Andric return MIB; 12880b57cec5SDimitry Andric } 1289