xref: /freebsd/contrib/llvm-project/llvm/lib/CodeGen/GlobalISel/MachineIRBuilder.cpp (revision 0eae32dcef82f6f06de6419a0d623d7def0cc8f6)
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