xref: /freebsd/contrib/llvm-project/llvm/lib/CodeGen/GlobalISel/MachineIRBuilder.cpp (revision fe6060f10f634930ff71b7c50291ddc610da2475)
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 
2180b57cec5SDimitry Andric MachineInstrBuilder MachineIRBuilder::buildBr(MachineBasicBlock &Dest) {
2190b57cec5SDimitry Andric   return buildInstr(TargetOpcode::G_BR).addMBB(&Dest);
2200b57cec5SDimitry Andric }
2210b57cec5SDimitry Andric 
2220b57cec5SDimitry Andric MachineInstrBuilder MachineIRBuilder::buildBrIndirect(Register Tgt) {
2230b57cec5SDimitry Andric   assert(getMRI()->getType(Tgt).isPointer() && "invalid branch destination");
2240b57cec5SDimitry Andric   return buildInstr(TargetOpcode::G_BRINDIRECT).addUse(Tgt);
2250b57cec5SDimitry Andric }
2260b57cec5SDimitry Andric 
2270b57cec5SDimitry Andric MachineInstrBuilder MachineIRBuilder::buildBrJT(Register TablePtr,
2280b57cec5SDimitry Andric                                                 unsigned JTI,
2290b57cec5SDimitry Andric                                                 Register IndexReg) {
2300b57cec5SDimitry Andric   assert(getMRI()->getType(TablePtr).isPointer() &&
2310b57cec5SDimitry Andric          "Table reg must be a pointer");
2320b57cec5SDimitry Andric   return buildInstr(TargetOpcode::G_BRJT)
2330b57cec5SDimitry Andric       .addUse(TablePtr)
2340b57cec5SDimitry Andric       .addJumpTableIndex(JTI)
2350b57cec5SDimitry Andric       .addUse(IndexReg);
2360b57cec5SDimitry Andric }
2370b57cec5SDimitry Andric 
2380b57cec5SDimitry Andric MachineInstrBuilder MachineIRBuilder::buildCopy(const DstOp &Res,
2390b57cec5SDimitry Andric                                                 const SrcOp &Op) {
2400b57cec5SDimitry Andric   return buildInstr(TargetOpcode::COPY, Res, Op);
2410b57cec5SDimitry Andric }
2420b57cec5SDimitry Andric 
243*fe6060f1SDimitry Andric MachineInstrBuilder MachineIRBuilder::buildAssertSExt(const DstOp &Res,
244*fe6060f1SDimitry Andric                                                       const SrcOp &Op,
245*fe6060f1SDimitry Andric                                                       unsigned Size) {
246*fe6060f1SDimitry Andric   return buildInstr(TargetOpcode::G_ASSERT_SEXT, Res, Op).addImm(Size);
247*fe6060f1SDimitry Andric }
248*fe6060f1SDimitry Andric 
249*fe6060f1SDimitry Andric MachineInstrBuilder MachineIRBuilder::buildAssertZExt(const DstOp &Res,
250*fe6060f1SDimitry Andric                                                       const SrcOp &Op,
251*fe6060f1SDimitry Andric                                                       unsigned Size) {
252*fe6060f1SDimitry Andric   return buildInstr(TargetOpcode::G_ASSERT_ZEXT, Res, Op).addImm(Size);
253*fe6060f1SDimitry Andric }
254*fe6060f1SDimitry Andric 
2550b57cec5SDimitry Andric MachineInstrBuilder MachineIRBuilder::buildConstant(const DstOp &Res,
2560b57cec5SDimitry Andric                                                     const ConstantInt &Val) {
2570b57cec5SDimitry Andric   LLT Ty = Res.getLLTTy(*getMRI());
2580b57cec5SDimitry Andric   LLT EltTy = Ty.getScalarType();
2590b57cec5SDimitry Andric   assert(EltTy.getScalarSizeInBits() == Val.getBitWidth() &&
2600b57cec5SDimitry Andric          "creating constant with the wrong size");
2610b57cec5SDimitry Andric 
2620b57cec5SDimitry Andric   if (Ty.isVector()) {
2630b57cec5SDimitry Andric     auto Const = buildInstr(TargetOpcode::G_CONSTANT)
2640b57cec5SDimitry Andric     .addDef(getMRI()->createGenericVirtualRegister(EltTy))
2650b57cec5SDimitry Andric     .addCImm(&Val);
2660b57cec5SDimitry Andric     return buildSplatVector(Res, Const);
2670b57cec5SDimitry Andric   }
2680b57cec5SDimitry Andric 
2690b57cec5SDimitry Andric   auto Const = buildInstr(TargetOpcode::G_CONSTANT);
2705ffd83dbSDimitry Andric   Const->setDebugLoc(DebugLoc());
2710b57cec5SDimitry Andric   Res.addDefToMIB(*getMRI(), Const);
2720b57cec5SDimitry Andric   Const.addCImm(&Val);
2730b57cec5SDimitry Andric   return Const;
2740b57cec5SDimitry Andric }
2750b57cec5SDimitry Andric 
2760b57cec5SDimitry Andric MachineInstrBuilder MachineIRBuilder::buildConstant(const DstOp &Res,
2770b57cec5SDimitry Andric                                                     int64_t Val) {
2780b57cec5SDimitry Andric   auto IntN = IntegerType::get(getMF().getFunction().getContext(),
2790b57cec5SDimitry Andric                                Res.getLLTTy(*getMRI()).getScalarSizeInBits());
2800b57cec5SDimitry Andric   ConstantInt *CI = ConstantInt::get(IntN, Val, true);
2810b57cec5SDimitry Andric   return buildConstant(Res, *CI);
2820b57cec5SDimitry Andric }
2830b57cec5SDimitry Andric 
2840b57cec5SDimitry Andric MachineInstrBuilder MachineIRBuilder::buildFConstant(const DstOp &Res,
2850b57cec5SDimitry Andric                                                      const ConstantFP &Val) {
2860b57cec5SDimitry Andric   LLT Ty = Res.getLLTTy(*getMRI());
2870b57cec5SDimitry Andric   LLT EltTy = Ty.getScalarType();
2880b57cec5SDimitry Andric 
2890b57cec5SDimitry Andric   assert(APFloat::getSizeInBits(Val.getValueAPF().getSemantics())
2900b57cec5SDimitry Andric          == EltTy.getSizeInBits() &&
2910b57cec5SDimitry Andric          "creating fconstant with the wrong size");
2920b57cec5SDimitry Andric 
2930b57cec5SDimitry Andric   assert(!Ty.isPointer() && "invalid operand type");
2940b57cec5SDimitry Andric 
2950b57cec5SDimitry Andric   if (Ty.isVector()) {
2960b57cec5SDimitry Andric     auto Const = buildInstr(TargetOpcode::G_FCONSTANT)
2970b57cec5SDimitry Andric     .addDef(getMRI()->createGenericVirtualRegister(EltTy))
2980b57cec5SDimitry Andric     .addFPImm(&Val);
2990b57cec5SDimitry Andric 
3000b57cec5SDimitry Andric     return buildSplatVector(Res, Const);
3010b57cec5SDimitry Andric   }
3020b57cec5SDimitry Andric 
3030b57cec5SDimitry Andric   auto Const = buildInstr(TargetOpcode::G_FCONSTANT);
3045ffd83dbSDimitry Andric   Const->setDebugLoc(DebugLoc());
3050b57cec5SDimitry Andric   Res.addDefToMIB(*getMRI(), Const);
3060b57cec5SDimitry Andric   Const.addFPImm(&Val);
3070b57cec5SDimitry Andric   return Const;
3080b57cec5SDimitry Andric }
3090b57cec5SDimitry Andric 
3100b57cec5SDimitry Andric MachineInstrBuilder MachineIRBuilder::buildConstant(const DstOp &Res,
3110b57cec5SDimitry Andric                                                     const APInt &Val) {
3120b57cec5SDimitry Andric   ConstantInt *CI = ConstantInt::get(getMF().getFunction().getContext(), Val);
3130b57cec5SDimitry Andric   return buildConstant(Res, *CI);
3140b57cec5SDimitry Andric }
3150b57cec5SDimitry Andric 
3160b57cec5SDimitry Andric MachineInstrBuilder MachineIRBuilder::buildFConstant(const DstOp &Res,
3170b57cec5SDimitry Andric                                                      double Val) {
3180b57cec5SDimitry Andric   LLT DstTy = Res.getLLTTy(*getMRI());
3190b57cec5SDimitry Andric   auto &Ctx = getMF().getFunction().getContext();
3200b57cec5SDimitry Andric   auto *CFP =
3210b57cec5SDimitry Andric       ConstantFP::get(Ctx, getAPFloatFromSize(Val, DstTy.getScalarSizeInBits()));
3220b57cec5SDimitry Andric   return buildFConstant(Res, *CFP);
3230b57cec5SDimitry Andric }
3240b57cec5SDimitry Andric 
3250b57cec5SDimitry Andric MachineInstrBuilder MachineIRBuilder::buildFConstant(const DstOp &Res,
3260b57cec5SDimitry Andric                                                      const APFloat &Val) {
3270b57cec5SDimitry Andric   auto &Ctx = getMF().getFunction().getContext();
3280b57cec5SDimitry Andric   auto *CFP = ConstantFP::get(Ctx, Val);
3290b57cec5SDimitry Andric   return buildFConstant(Res, *CFP);
3300b57cec5SDimitry Andric }
3310b57cec5SDimitry Andric 
332e8d8bef9SDimitry Andric MachineInstrBuilder MachineIRBuilder::buildBrCond(const SrcOp &Tst,
3330b57cec5SDimitry Andric                                                   MachineBasicBlock &Dest) {
334e8d8bef9SDimitry Andric   assert(Tst.getLLTTy(*getMRI()).isScalar() && "invalid operand type");
3350b57cec5SDimitry Andric 
336e8d8bef9SDimitry Andric   auto MIB = buildInstr(TargetOpcode::G_BRCOND);
337e8d8bef9SDimitry Andric   Tst.addSrcToMIB(MIB);
338e8d8bef9SDimitry Andric   MIB.addMBB(&Dest);
339e8d8bef9SDimitry Andric   return MIB;
3400b57cec5SDimitry Andric }
3410b57cec5SDimitry Andric 
342e8d8bef9SDimitry Andric MachineInstrBuilder
343e8d8bef9SDimitry Andric MachineIRBuilder::buildLoad(const DstOp &Dst, const SrcOp &Addr,
344e8d8bef9SDimitry Andric                             MachinePointerInfo PtrInfo, Align Alignment,
345e8d8bef9SDimitry Andric                             MachineMemOperand::Flags MMOFlags,
346e8d8bef9SDimitry Andric                             const AAMDNodes &AAInfo) {
347e8d8bef9SDimitry Andric   MMOFlags |= MachineMemOperand::MOLoad;
348e8d8bef9SDimitry Andric   assert((MMOFlags & MachineMemOperand::MOStore) == 0);
349e8d8bef9SDimitry Andric 
350*fe6060f1SDimitry Andric   LLT Ty = Dst.getLLTTy(*getMRI());
351e8d8bef9SDimitry Andric   MachineMemOperand *MMO =
352*fe6060f1SDimitry Andric       getMF().getMachineMemOperand(PtrInfo, MMOFlags, Ty, Alignment, AAInfo);
353e8d8bef9SDimitry Andric   return buildLoad(Dst, Addr, *MMO);
3540b57cec5SDimitry Andric }
3550b57cec5SDimitry Andric 
3560b57cec5SDimitry Andric MachineInstrBuilder MachineIRBuilder::buildLoadInstr(unsigned Opcode,
3570b57cec5SDimitry Andric                                                      const DstOp &Res,
3580b57cec5SDimitry Andric                                                      const SrcOp &Addr,
3590b57cec5SDimitry Andric                                                      MachineMemOperand &MMO) {
3600b57cec5SDimitry Andric   assert(Res.getLLTTy(*getMRI()).isValid() && "invalid operand type");
3610b57cec5SDimitry Andric   assert(Addr.getLLTTy(*getMRI()).isPointer() && "invalid operand type");
3620b57cec5SDimitry Andric 
3630b57cec5SDimitry Andric   auto MIB = buildInstr(Opcode);
3640b57cec5SDimitry Andric   Res.addDefToMIB(*getMRI(), MIB);
3650b57cec5SDimitry Andric   Addr.addSrcToMIB(MIB);
3660b57cec5SDimitry Andric   MIB.addMemOperand(&MMO);
3670b57cec5SDimitry Andric   return MIB;
3680b57cec5SDimitry Andric }
3690b57cec5SDimitry Andric 
3705ffd83dbSDimitry Andric MachineInstrBuilder MachineIRBuilder::buildLoadFromOffset(
3715ffd83dbSDimitry Andric   const DstOp &Dst, const SrcOp &BasePtr,
3725ffd83dbSDimitry Andric   MachineMemOperand &BaseMMO, int64_t Offset) {
3735ffd83dbSDimitry Andric   LLT LoadTy = Dst.getLLTTy(*getMRI());
3745ffd83dbSDimitry Andric   MachineMemOperand *OffsetMMO =
375*fe6060f1SDimitry Andric       getMF().getMachineMemOperand(&BaseMMO, Offset, LoadTy);
3765ffd83dbSDimitry Andric 
3775ffd83dbSDimitry Andric   if (Offset == 0) // This may be a size or type changing load.
3785ffd83dbSDimitry Andric     return buildLoad(Dst, BasePtr, *OffsetMMO);
3795ffd83dbSDimitry Andric 
3805ffd83dbSDimitry Andric   LLT PtrTy = BasePtr.getLLTTy(*getMRI());
3815ffd83dbSDimitry Andric   LLT OffsetTy = LLT::scalar(PtrTy.getSizeInBits());
3825ffd83dbSDimitry Andric   auto ConstOffset = buildConstant(OffsetTy, Offset);
3835ffd83dbSDimitry Andric   auto Ptr = buildPtrAdd(PtrTy, BasePtr, ConstOffset);
3845ffd83dbSDimitry Andric   return buildLoad(Dst, Ptr, *OffsetMMO);
3855ffd83dbSDimitry Andric }
3865ffd83dbSDimitry Andric 
3870b57cec5SDimitry Andric MachineInstrBuilder MachineIRBuilder::buildStore(const SrcOp &Val,
3880b57cec5SDimitry Andric                                                  const SrcOp &Addr,
3890b57cec5SDimitry Andric                                                  MachineMemOperand &MMO) {
3900b57cec5SDimitry Andric   assert(Val.getLLTTy(*getMRI()).isValid() && "invalid operand type");
3910b57cec5SDimitry Andric   assert(Addr.getLLTTy(*getMRI()).isPointer() && "invalid operand type");
3920b57cec5SDimitry Andric 
3930b57cec5SDimitry Andric   auto MIB = buildInstr(TargetOpcode::G_STORE);
3940b57cec5SDimitry Andric   Val.addSrcToMIB(MIB);
3950b57cec5SDimitry Andric   Addr.addSrcToMIB(MIB);
3960b57cec5SDimitry Andric   MIB.addMemOperand(&MMO);
3970b57cec5SDimitry Andric   return MIB;
3980b57cec5SDimitry Andric }
3990b57cec5SDimitry Andric 
400e8d8bef9SDimitry Andric MachineInstrBuilder
401e8d8bef9SDimitry Andric MachineIRBuilder::buildStore(const SrcOp &Val, const SrcOp &Addr,
402e8d8bef9SDimitry Andric                              MachinePointerInfo PtrInfo, Align Alignment,
403e8d8bef9SDimitry Andric                              MachineMemOperand::Flags MMOFlags,
404e8d8bef9SDimitry Andric                              const AAMDNodes &AAInfo) {
405e8d8bef9SDimitry Andric   MMOFlags |= MachineMemOperand::MOStore;
406e8d8bef9SDimitry Andric   assert((MMOFlags & MachineMemOperand::MOLoad) == 0);
407e8d8bef9SDimitry Andric 
408*fe6060f1SDimitry Andric   LLT Ty = Val.getLLTTy(*getMRI());
409e8d8bef9SDimitry Andric   MachineMemOperand *MMO =
410*fe6060f1SDimitry Andric       getMF().getMachineMemOperand(PtrInfo, MMOFlags, Ty, Alignment, AAInfo);
411e8d8bef9SDimitry Andric   return buildStore(Val, Addr, *MMO);
412e8d8bef9SDimitry Andric }
413e8d8bef9SDimitry Andric 
4140b57cec5SDimitry Andric MachineInstrBuilder MachineIRBuilder::buildAnyExt(const DstOp &Res,
4150b57cec5SDimitry Andric                                                   const SrcOp &Op) {
4160b57cec5SDimitry Andric   return buildInstr(TargetOpcode::G_ANYEXT, Res, Op);
4170b57cec5SDimitry Andric }
4180b57cec5SDimitry Andric 
4190b57cec5SDimitry Andric MachineInstrBuilder MachineIRBuilder::buildSExt(const DstOp &Res,
4200b57cec5SDimitry Andric                                                 const SrcOp &Op) {
4210b57cec5SDimitry Andric   return buildInstr(TargetOpcode::G_SEXT, Res, Op);
4220b57cec5SDimitry Andric }
4230b57cec5SDimitry Andric 
4240b57cec5SDimitry Andric MachineInstrBuilder MachineIRBuilder::buildZExt(const DstOp &Res,
4250b57cec5SDimitry Andric                                                 const SrcOp &Op) {
4260b57cec5SDimitry Andric   return buildInstr(TargetOpcode::G_ZEXT, Res, Op);
4270b57cec5SDimitry Andric }
4280b57cec5SDimitry Andric 
4290b57cec5SDimitry Andric unsigned MachineIRBuilder::getBoolExtOp(bool IsVec, bool IsFP) const {
4300b57cec5SDimitry Andric   const auto *TLI = getMF().getSubtarget().getTargetLowering();
4310b57cec5SDimitry Andric   switch (TLI->getBooleanContents(IsVec, IsFP)) {
4320b57cec5SDimitry Andric   case TargetLoweringBase::ZeroOrNegativeOneBooleanContent:
4330b57cec5SDimitry Andric     return TargetOpcode::G_SEXT;
4340b57cec5SDimitry Andric   case TargetLoweringBase::ZeroOrOneBooleanContent:
4350b57cec5SDimitry Andric     return TargetOpcode::G_ZEXT;
4360b57cec5SDimitry Andric   default:
4370b57cec5SDimitry Andric     return TargetOpcode::G_ANYEXT;
4380b57cec5SDimitry Andric   }
4390b57cec5SDimitry Andric }
4400b57cec5SDimitry Andric 
4410b57cec5SDimitry Andric MachineInstrBuilder MachineIRBuilder::buildBoolExt(const DstOp &Res,
4420b57cec5SDimitry Andric                                                    const SrcOp &Op,
4430b57cec5SDimitry Andric                                                    bool IsFP) {
4440b57cec5SDimitry Andric   unsigned ExtOp = getBoolExtOp(getMRI()->getType(Op.getReg()).isVector(), IsFP);
4450b57cec5SDimitry Andric   return buildInstr(ExtOp, Res, Op);
4460b57cec5SDimitry Andric }
4470b57cec5SDimitry Andric 
4480b57cec5SDimitry Andric MachineInstrBuilder MachineIRBuilder::buildExtOrTrunc(unsigned ExtOpc,
4490b57cec5SDimitry Andric                                                       const DstOp &Res,
4500b57cec5SDimitry Andric                                                       const SrcOp &Op) {
4510b57cec5SDimitry Andric   assert((TargetOpcode::G_ANYEXT == ExtOpc || TargetOpcode::G_ZEXT == ExtOpc ||
4520b57cec5SDimitry Andric           TargetOpcode::G_SEXT == ExtOpc) &&
4530b57cec5SDimitry Andric          "Expecting Extending Opc");
4540b57cec5SDimitry Andric   assert(Res.getLLTTy(*getMRI()).isScalar() ||
4550b57cec5SDimitry Andric          Res.getLLTTy(*getMRI()).isVector());
4560b57cec5SDimitry Andric   assert(Res.getLLTTy(*getMRI()).isScalar() ==
4570b57cec5SDimitry Andric          Op.getLLTTy(*getMRI()).isScalar());
4580b57cec5SDimitry Andric 
4590b57cec5SDimitry Andric   unsigned Opcode = TargetOpcode::COPY;
4600b57cec5SDimitry Andric   if (Res.getLLTTy(*getMRI()).getSizeInBits() >
4610b57cec5SDimitry Andric       Op.getLLTTy(*getMRI()).getSizeInBits())
4620b57cec5SDimitry Andric     Opcode = ExtOpc;
4630b57cec5SDimitry Andric   else if (Res.getLLTTy(*getMRI()).getSizeInBits() <
4640b57cec5SDimitry Andric            Op.getLLTTy(*getMRI()).getSizeInBits())
4650b57cec5SDimitry Andric     Opcode = TargetOpcode::G_TRUNC;
4660b57cec5SDimitry Andric   else
4670b57cec5SDimitry Andric     assert(Res.getLLTTy(*getMRI()) == Op.getLLTTy(*getMRI()));
4680b57cec5SDimitry Andric 
4690b57cec5SDimitry Andric   return buildInstr(Opcode, Res, Op);
4700b57cec5SDimitry Andric }
4710b57cec5SDimitry Andric 
4720b57cec5SDimitry Andric MachineInstrBuilder MachineIRBuilder::buildSExtOrTrunc(const DstOp &Res,
4730b57cec5SDimitry Andric                                                        const SrcOp &Op) {
4740b57cec5SDimitry Andric   return buildExtOrTrunc(TargetOpcode::G_SEXT, Res, Op);
4750b57cec5SDimitry Andric }
4760b57cec5SDimitry Andric 
4770b57cec5SDimitry Andric MachineInstrBuilder MachineIRBuilder::buildZExtOrTrunc(const DstOp &Res,
4780b57cec5SDimitry Andric                                                        const SrcOp &Op) {
4790b57cec5SDimitry Andric   return buildExtOrTrunc(TargetOpcode::G_ZEXT, Res, Op);
4800b57cec5SDimitry Andric }
4810b57cec5SDimitry Andric 
4820b57cec5SDimitry Andric MachineInstrBuilder MachineIRBuilder::buildAnyExtOrTrunc(const DstOp &Res,
4830b57cec5SDimitry Andric                                                          const SrcOp &Op) {
4840b57cec5SDimitry Andric   return buildExtOrTrunc(TargetOpcode::G_ANYEXT, Res, Op);
4850b57cec5SDimitry Andric }
4860b57cec5SDimitry Andric 
487*fe6060f1SDimitry Andric MachineInstrBuilder MachineIRBuilder::buildZExtInReg(const DstOp &Res,
488*fe6060f1SDimitry Andric                                                      const SrcOp &Op,
489*fe6060f1SDimitry Andric                                                      int64_t ImmOp) {
490*fe6060f1SDimitry Andric   LLT ResTy = Res.getLLTTy(*getMRI());
491*fe6060f1SDimitry Andric   auto Mask = buildConstant(
492*fe6060f1SDimitry Andric       ResTy, APInt::getLowBitsSet(ResTy.getScalarSizeInBits(), ImmOp));
493*fe6060f1SDimitry Andric   return buildAnd(Res, Op, Mask);
494*fe6060f1SDimitry Andric }
495*fe6060f1SDimitry Andric 
4960b57cec5SDimitry Andric MachineInstrBuilder MachineIRBuilder::buildCast(const DstOp &Dst,
4970b57cec5SDimitry Andric                                                 const SrcOp &Src) {
4980b57cec5SDimitry Andric   LLT SrcTy = Src.getLLTTy(*getMRI());
4990b57cec5SDimitry Andric   LLT DstTy = Dst.getLLTTy(*getMRI());
5000b57cec5SDimitry Andric   if (SrcTy == DstTy)
5010b57cec5SDimitry Andric     return buildCopy(Dst, Src);
5020b57cec5SDimitry Andric 
5030b57cec5SDimitry Andric   unsigned Opcode;
5040b57cec5SDimitry Andric   if (SrcTy.isPointer() && DstTy.isScalar())
5050b57cec5SDimitry Andric     Opcode = TargetOpcode::G_PTRTOINT;
5060b57cec5SDimitry Andric   else if (DstTy.isPointer() && SrcTy.isScalar())
5070b57cec5SDimitry Andric     Opcode = TargetOpcode::G_INTTOPTR;
5080b57cec5SDimitry Andric   else {
5090b57cec5SDimitry Andric     assert(!SrcTy.isPointer() && !DstTy.isPointer() && "n G_ADDRCAST yet");
5100b57cec5SDimitry Andric     Opcode = TargetOpcode::G_BITCAST;
5110b57cec5SDimitry Andric   }
5120b57cec5SDimitry Andric 
5130b57cec5SDimitry Andric   return buildInstr(Opcode, Dst, Src);
5140b57cec5SDimitry Andric }
5150b57cec5SDimitry Andric 
5160b57cec5SDimitry Andric MachineInstrBuilder MachineIRBuilder::buildExtract(const DstOp &Dst,
5170b57cec5SDimitry Andric                                                    const SrcOp &Src,
5180b57cec5SDimitry Andric                                                    uint64_t Index) {
5190b57cec5SDimitry Andric   LLT SrcTy = Src.getLLTTy(*getMRI());
5200b57cec5SDimitry Andric   LLT DstTy = Dst.getLLTTy(*getMRI());
5210b57cec5SDimitry Andric 
5220b57cec5SDimitry Andric #ifndef NDEBUG
5230b57cec5SDimitry Andric   assert(SrcTy.isValid() && "invalid operand type");
5240b57cec5SDimitry Andric   assert(DstTy.isValid() && "invalid operand type");
5250b57cec5SDimitry Andric   assert(Index + DstTy.getSizeInBits() <= SrcTy.getSizeInBits() &&
5260b57cec5SDimitry Andric          "extracting off end of register");
5270b57cec5SDimitry Andric #endif
5280b57cec5SDimitry Andric 
5290b57cec5SDimitry Andric   if (DstTy.getSizeInBits() == SrcTy.getSizeInBits()) {
5300b57cec5SDimitry Andric     assert(Index == 0 && "insertion past the end of a register");
5310b57cec5SDimitry Andric     return buildCast(Dst, Src);
5320b57cec5SDimitry Andric   }
5330b57cec5SDimitry Andric 
5340b57cec5SDimitry Andric   auto Extract = buildInstr(TargetOpcode::G_EXTRACT);
5350b57cec5SDimitry Andric   Dst.addDefToMIB(*getMRI(), Extract);
5360b57cec5SDimitry Andric   Src.addSrcToMIB(Extract);
5370b57cec5SDimitry Andric   Extract.addImm(Index);
5380b57cec5SDimitry Andric   return Extract;
5390b57cec5SDimitry Andric }
5400b57cec5SDimitry Andric 
5410b57cec5SDimitry Andric void MachineIRBuilder::buildSequence(Register Res, ArrayRef<Register> Ops,
5420b57cec5SDimitry Andric                                      ArrayRef<uint64_t> Indices) {
5430b57cec5SDimitry Andric #ifndef NDEBUG
5440b57cec5SDimitry Andric   assert(Ops.size() == Indices.size() && "incompatible args");
5450b57cec5SDimitry Andric   assert(!Ops.empty() && "invalid trivial sequence");
5465ffd83dbSDimitry Andric   assert(llvm::is_sorted(Indices) &&
5470b57cec5SDimitry Andric          "sequence offsets must be in ascending order");
5480b57cec5SDimitry Andric 
5490b57cec5SDimitry Andric   assert(getMRI()->getType(Res).isValid() && "invalid operand type");
5500b57cec5SDimitry Andric   for (auto Op : Ops)
5510b57cec5SDimitry Andric     assert(getMRI()->getType(Op).isValid() && "invalid operand type");
5520b57cec5SDimitry Andric #endif
5530b57cec5SDimitry Andric 
5540b57cec5SDimitry Andric   LLT ResTy = getMRI()->getType(Res);
5550b57cec5SDimitry Andric   LLT OpTy = getMRI()->getType(Ops[0]);
5560b57cec5SDimitry Andric   unsigned OpSize = OpTy.getSizeInBits();
5570b57cec5SDimitry Andric   bool MaybeMerge = true;
5580b57cec5SDimitry Andric   for (unsigned i = 0; i < Ops.size(); ++i) {
5590b57cec5SDimitry Andric     if (getMRI()->getType(Ops[i]) != OpTy || Indices[i] != i * OpSize) {
5600b57cec5SDimitry Andric       MaybeMerge = false;
5610b57cec5SDimitry Andric       break;
5620b57cec5SDimitry Andric     }
5630b57cec5SDimitry Andric   }
5640b57cec5SDimitry Andric 
5650b57cec5SDimitry Andric   if (MaybeMerge && Ops.size() * OpSize == ResTy.getSizeInBits()) {
5660b57cec5SDimitry Andric     buildMerge(Res, Ops);
5670b57cec5SDimitry Andric     return;
5680b57cec5SDimitry Andric   }
5690b57cec5SDimitry Andric 
5700b57cec5SDimitry Andric   Register ResIn = getMRI()->createGenericVirtualRegister(ResTy);
5710b57cec5SDimitry Andric   buildUndef(ResIn);
5720b57cec5SDimitry Andric 
5730b57cec5SDimitry Andric   for (unsigned i = 0; i < Ops.size(); ++i) {
5740b57cec5SDimitry Andric     Register ResOut = i + 1 == Ops.size()
5750b57cec5SDimitry Andric                           ? Res
5760b57cec5SDimitry Andric                           : getMRI()->createGenericVirtualRegister(ResTy);
5770b57cec5SDimitry Andric     buildInsert(ResOut, ResIn, Ops[i], Indices[i]);
5780b57cec5SDimitry Andric     ResIn = ResOut;
5790b57cec5SDimitry Andric   }
5800b57cec5SDimitry Andric }
5810b57cec5SDimitry Andric 
5820b57cec5SDimitry Andric MachineInstrBuilder MachineIRBuilder::buildUndef(const DstOp &Res) {
5830b57cec5SDimitry Andric   return buildInstr(TargetOpcode::G_IMPLICIT_DEF, {Res}, {});
5840b57cec5SDimitry Andric }
5850b57cec5SDimitry Andric 
5860b57cec5SDimitry Andric MachineInstrBuilder MachineIRBuilder::buildMerge(const DstOp &Res,
5870b57cec5SDimitry Andric                                                  ArrayRef<Register> Ops) {
5880b57cec5SDimitry Andric   // Unfortunately to convert from ArrayRef<LLT> to ArrayRef<SrcOp>,
5890b57cec5SDimitry Andric   // we need some temporary storage for the DstOp objects. Here we use a
5900b57cec5SDimitry Andric   // sufficiently large SmallVector to not go through the heap.
5910b57cec5SDimitry Andric   SmallVector<SrcOp, 8> TmpVec(Ops.begin(), Ops.end());
5920b57cec5SDimitry Andric   assert(TmpVec.size() > 1);
5930b57cec5SDimitry Andric   return buildInstr(TargetOpcode::G_MERGE_VALUES, Res, TmpVec);
5940b57cec5SDimitry Andric }
5950b57cec5SDimitry Andric 
5965ffd83dbSDimitry Andric MachineInstrBuilder
5975ffd83dbSDimitry Andric MachineIRBuilder::buildMerge(const DstOp &Res,
5985ffd83dbSDimitry Andric                              std::initializer_list<SrcOp> Ops) {
5995ffd83dbSDimitry Andric   assert(Ops.size() > 1);
6005ffd83dbSDimitry Andric   return buildInstr(TargetOpcode::G_MERGE_VALUES, Res, Ops);
6015ffd83dbSDimitry Andric }
6025ffd83dbSDimitry Andric 
6030b57cec5SDimitry Andric MachineInstrBuilder MachineIRBuilder::buildUnmerge(ArrayRef<LLT> Res,
6040b57cec5SDimitry Andric                                                    const SrcOp &Op) {
6050b57cec5SDimitry Andric   // Unfortunately to convert from ArrayRef<LLT> to ArrayRef<DstOp>,
6060b57cec5SDimitry Andric   // we need some temporary storage for the DstOp objects. Here we use a
6070b57cec5SDimitry Andric   // sufficiently large SmallVector to not go through the heap.
6080b57cec5SDimitry Andric   SmallVector<DstOp, 8> TmpVec(Res.begin(), Res.end());
6090b57cec5SDimitry Andric   assert(TmpVec.size() > 1);
6100b57cec5SDimitry Andric   return buildInstr(TargetOpcode::G_UNMERGE_VALUES, TmpVec, Op);
6110b57cec5SDimitry Andric }
6120b57cec5SDimitry Andric 
6130b57cec5SDimitry Andric MachineInstrBuilder MachineIRBuilder::buildUnmerge(LLT Res,
6140b57cec5SDimitry Andric                                                    const SrcOp &Op) {
6150b57cec5SDimitry Andric   unsigned NumReg = Op.getLLTTy(*getMRI()).getSizeInBits() / Res.getSizeInBits();
6160b57cec5SDimitry Andric   SmallVector<Register, 8> TmpVec;
6170b57cec5SDimitry Andric   for (unsigned I = 0; I != NumReg; ++I)
6180b57cec5SDimitry Andric     TmpVec.push_back(getMRI()->createGenericVirtualRegister(Res));
6190b57cec5SDimitry Andric   return buildUnmerge(TmpVec, Op);
6200b57cec5SDimitry Andric }
6210b57cec5SDimitry Andric 
6220b57cec5SDimitry Andric MachineInstrBuilder MachineIRBuilder::buildUnmerge(ArrayRef<Register> Res,
6230b57cec5SDimitry Andric                                                    const SrcOp &Op) {
6240b57cec5SDimitry Andric   // Unfortunately to convert from ArrayRef<Register> to ArrayRef<DstOp>,
6250b57cec5SDimitry Andric   // we need some temporary storage for the DstOp objects. Here we use a
6260b57cec5SDimitry Andric   // sufficiently large SmallVector to not go through the heap.
6270b57cec5SDimitry Andric   SmallVector<DstOp, 8> TmpVec(Res.begin(), Res.end());
6280b57cec5SDimitry Andric   assert(TmpVec.size() > 1);
6290b57cec5SDimitry Andric   return buildInstr(TargetOpcode::G_UNMERGE_VALUES, TmpVec, Op);
6300b57cec5SDimitry Andric }
6310b57cec5SDimitry Andric 
6320b57cec5SDimitry Andric MachineInstrBuilder MachineIRBuilder::buildBuildVector(const DstOp &Res,
6330b57cec5SDimitry Andric                                                        ArrayRef<Register> Ops) {
6340b57cec5SDimitry Andric   // Unfortunately to convert from ArrayRef<Register> to ArrayRef<SrcOp>,
6350b57cec5SDimitry Andric   // we need some temporary storage for the DstOp objects. Here we use a
6360b57cec5SDimitry Andric   // sufficiently large SmallVector to not go through the heap.
6370b57cec5SDimitry Andric   SmallVector<SrcOp, 8> TmpVec(Ops.begin(), Ops.end());
6380b57cec5SDimitry Andric   return buildInstr(TargetOpcode::G_BUILD_VECTOR, Res, TmpVec);
6390b57cec5SDimitry Andric }
6400b57cec5SDimitry Andric 
6410b57cec5SDimitry Andric MachineInstrBuilder MachineIRBuilder::buildSplatVector(const DstOp &Res,
6420b57cec5SDimitry Andric                                                        const SrcOp &Src) {
6430b57cec5SDimitry Andric   SmallVector<SrcOp, 8> TmpVec(Res.getLLTTy(*getMRI()).getNumElements(), Src);
6440b57cec5SDimitry Andric   return buildInstr(TargetOpcode::G_BUILD_VECTOR, Res, TmpVec);
6450b57cec5SDimitry Andric }
6460b57cec5SDimitry Andric 
6470b57cec5SDimitry Andric MachineInstrBuilder
6480b57cec5SDimitry Andric MachineIRBuilder::buildBuildVectorTrunc(const DstOp &Res,
6490b57cec5SDimitry Andric                                         ArrayRef<Register> Ops) {
6500b57cec5SDimitry Andric   // Unfortunately to convert from ArrayRef<Register> to ArrayRef<SrcOp>,
6510b57cec5SDimitry Andric   // we need some temporary storage for the DstOp objects. Here we use a
6520b57cec5SDimitry Andric   // sufficiently large SmallVector to not go through the heap.
6530b57cec5SDimitry Andric   SmallVector<SrcOp, 8> TmpVec(Ops.begin(), Ops.end());
6540b57cec5SDimitry Andric   return buildInstr(TargetOpcode::G_BUILD_VECTOR_TRUNC, Res, TmpVec);
6550b57cec5SDimitry Andric }
6560b57cec5SDimitry Andric 
657e8d8bef9SDimitry Andric MachineInstrBuilder MachineIRBuilder::buildShuffleSplat(const DstOp &Res,
658e8d8bef9SDimitry Andric                                                         const SrcOp &Src) {
659e8d8bef9SDimitry Andric   LLT DstTy = Res.getLLTTy(*getMRI());
660e8d8bef9SDimitry Andric   assert(Src.getLLTTy(*getMRI()) == DstTy.getElementType() &&
661e8d8bef9SDimitry Andric          "Expected Src to match Dst elt ty");
662e8d8bef9SDimitry Andric   auto UndefVec = buildUndef(DstTy);
663e8d8bef9SDimitry Andric   auto Zero = buildConstant(LLT::scalar(64), 0);
664e8d8bef9SDimitry Andric   auto InsElt = buildInsertVectorElement(DstTy, UndefVec, Src, Zero);
665e8d8bef9SDimitry Andric   SmallVector<int, 16> ZeroMask(DstTy.getNumElements());
666e8d8bef9SDimitry Andric   return buildShuffleVector(DstTy, InsElt, UndefVec, ZeroMask);
667e8d8bef9SDimitry Andric }
668e8d8bef9SDimitry Andric 
669e8d8bef9SDimitry Andric MachineInstrBuilder MachineIRBuilder::buildShuffleVector(const DstOp &Res,
670e8d8bef9SDimitry Andric                                                          const SrcOp &Src1,
671e8d8bef9SDimitry Andric                                                          const SrcOp &Src2,
672e8d8bef9SDimitry Andric                                                          ArrayRef<int> Mask) {
673e8d8bef9SDimitry Andric   LLT DstTy = Res.getLLTTy(*getMRI());
674e8d8bef9SDimitry Andric   LLT Src1Ty = Src1.getLLTTy(*getMRI());
675e8d8bef9SDimitry Andric   LLT Src2Ty = Src2.getLLTTy(*getMRI());
676e8d8bef9SDimitry Andric   assert(Src1Ty.getNumElements() + Src2Ty.getNumElements() >= Mask.size());
677e8d8bef9SDimitry Andric   assert(DstTy.getElementType() == Src1Ty.getElementType() &&
678e8d8bef9SDimitry Andric          DstTy.getElementType() == Src2Ty.getElementType());
679*fe6060f1SDimitry Andric   (void)DstTy;
680e8d8bef9SDimitry Andric   (void)Src1Ty;
681e8d8bef9SDimitry Andric   (void)Src2Ty;
682e8d8bef9SDimitry Andric   ArrayRef<int> MaskAlloc = getMF().allocateShuffleMask(Mask);
683*fe6060f1SDimitry Andric   return buildInstr(TargetOpcode::G_SHUFFLE_VECTOR, {Res}, {Src1, Src2})
684e8d8bef9SDimitry Andric       .addShuffleMask(MaskAlloc);
685e8d8bef9SDimitry Andric }
686e8d8bef9SDimitry Andric 
6870b57cec5SDimitry Andric MachineInstrBuilder
6880b57cec5SDimitry Andric MachineIRBuilder::buildConcatVectors(const DstOp &Res, ArrayRef<Register> Ops) {
6890b57cec5SDimitry Andric   // Unfortunately to convert from ArrayRef<Register> to ArrayRef<SrcOp>,
6900b57cec5SDimitry Andric   // we need some temporary storage for the DstOp objects. Here we use a
6910b57cec5SDimitry Andric   // sufficiently large SmallVector to not go through the heap.
6920b57cec5SDimitry Andric   SmallVector<SrcOp, 8> TmpVec(Ops.begin(), Ops.end());
6930b57cec5SDimitry Andric   return buildInstr(TargetOpcode::G_CONCAT_VECTORS, Res, TmpVec);
6940b57cec5SDimitry Andric }
6950b57cec5SDimitry Andric 
6965ffd83dbSDimitry Andric MachineInstrBuilder MachineIRBuilder::buildInsert(const DstOp &Res,
6975ffd83dbSDimitry Andric                                                   const SrcOp &Src,
6985ffd83dbSDimitry Andric                                                   const SrcOp &Op,
6995ffd83dbSDimitry Andric                                                   unsigned Index) {
7005ffd83dbSDimitry Andric   assert(Index + Op.getLLTTy(*getMRI()).getSizeInBits() <=
7015ffd83dbSDimitry Andric              Res.getLLTTy(*getMRI()).getSizeInBits() &&
7020b57cec5SDimitry Andric          "insertion past the end of a register");
7030b57cec5SDimitry Andric 
7045ffd83dbSDimitry Andric   if (Res.getLLTTy(*getMRI()).getSizeInBits() ==
7055ffd83dbSDimitry Andric       Op.getLLTTy(*getMRI()).getSizeInBits()) {
7060b57cec5SDimitry Andric     return buildCast(Res, Op);
7070b57cec5SDimitry Andric   }
7080b57cec5SDimitry Andric 
7095ffd83dbSDimitry Andric   return buildInstr(TargetOpcode::G_INSERT, Res, {Src, Op, uint64_t(Index)});
7100b57cec5SDimitry Andric }
7110b57cec5SDimitry Andric 
7120b57cec5SDimitry Andric MachineInstrBuilder MachineIRBuilder::buildIntrinsic(Intrinsic::ID ID,
7130b57cec5SDimitry Andric                                                      ArrayRef<Register> ResultRegs,
7140b57cec5SDimitry Andric                                                      bool HasSideEffects) {
7150b57cec5SDimitry Andric   auto MIB =
7160b57cec5SDimitry Andric       buildInstr(HasSideEffects ? TargetOpcode::G_INTRINSIC_W_SIDE_EFFECTS
7170b57cec5SDimitry Andric                                 : TargetOpcode::G_INTRINSIC);
7180b57cec5SDimitry Andric   for (unsigned ResultReg : ResultRegs)
7190b57cec5SDimitry Andric     MIB.addDef(ResultReg);
7200b57cec5SDimitry Andric   MIB.addIntrinsicID(ID);
7210b57cec5SDimitry Andric   return MIB;
7220b57cec5SDimitry Andric }
7230b57cec5SDimitry Andric 
7240b57cec5SDimitry Andric MachineInstrBuilder MachineIRBuilder::buildIntrinsic(Intrinsic::ID ID,
7250b57cec5SDimitry Andric                                                      ArrayRef<DstOp> Results,
7260b57cec5SDimitry Andric                                                      bool HasSideEffects) {
7270b57cec5SDimitry Andric   auto MIB =
7280b57cec5SDimitry Andric       buildInstr(HasSideEffects ? TargetOpcode::G_INTRINSIC_W_SIDE_EFFECTS
7290b57cec5SDimitry Andric                                 : TargetOpcode::G_INTRINSIC);
7300b57cec5SDimitry Andric   for (DstOp Result : Results)
7310b57cec5SDimitry Andric     Result.addDefToMIB(*getMRI(), MIB);
7320b57cec5SDimitry Andric   MIB.addIntrinsicID(ID);
7330b57cec5SDimitry Andric   return MIB;
7340b57cec5SDimitry Andric }
7350b57cec5SDimitry Andric 
7360b57cec5SDimitry Andric MachineInstrBuilder MachineIRBuilder::buildTrunc(const DstOp &Res,
7370b57cec5SDimitry Andric                                                  const SrcOp &Op) {
7380b57cec5SDimitry Andric   return buildInstr(TargetOpcode::G_TRUNC, Res, Op);
7390b57cec5SDimitry Andric }
7400b57cec5SDimitry Andric 
7410b57cec5SDimitry Andric MachineInstrBuilder MachineIRBuilder::buildFPTrunc(const DstOp &Res,
742480093f4SDimitry Andric                                                    const SrcOp &Op,
743480093f4SDimitry Andric                                                    Optional<unsigned> Flags) {
744480093f4SDimitry Andric   return buildInstr(TargetOpcode::G_FPTRUNC, Res, Op, Flags);
7450b57cec5SDimitry Andric }
7460b57cec5SDimitry Andric 
7470b57cec5SDimitry Andric MachineInstrBuilder MachineIRBuilder::buildICmp(CmpInst::Predicate Pred,
7480b57cec5SDimitry Andric                                                 const DstOp &Res,
7490b57cec5SDimitry Andric                                                 const SrcOp &Op0,
7500b57cec5SDimitry Andric                                                 const SrcOp &Op1) {
7510b57cec5SDimitry Andric   return buildInstr(TargetOpcode::G_ICMP, Res, {Pred, Op0, Op1});
7520b57cec5SDimitry Andric }
7530b57cec5SDimitry Andric 
7540b57cec5SDimitry Andric MachineInstrBuilder MachineIRBuilder::buildFCmp(CmpInst::Predicate Pred,
7550b57cec5SDimitry Andric                                                 const DstOp &Res,
7560b57cec5SDimitry Andric                                                 const SrcOp &Op0,
7578bcb0991SDimitry Andric                                                 const SrcOp &Op1,
7588bcb0991SDimitry Andric                                                 Optional<unsigned> Flags) {
7590b57cec5SDimitry Andric 
7608bcb0991SDimitry Andric   return buildInstr(TargetOpcode::G_FCMP, Res, {Pred, Op0, Op1}, Flags);
7610b57cec5SDimitry Andric }
7620b57cec5SDimitry Andric 
7630b57cec5SDimitry Andric MachineInstrBuilder MachineIRBuilder::buildSelect(const DstOp &Res,
7640b57cec5SDimitry Andric                                                   const SrcOp &Tst,
7650b57cec5SDimitry Andric                                                   const SrcOp &Op0,
7668bcb0991SDimitry Andric                                                   const SrcOp &Op1,
7678bcb0991SDimitry Andric                                                   Optional<unsigned> Flags) {
7680b57cec5SDimitry Andric 
7698bcb0991SDimitry Andric   return buildInstr(TargetOpcode::G_SELECT, {Res}, {Tst, Op0, Op1}, Flags);
7700b57cec5SDimitry Andric }
7710b57cec5SDimitry Andric 
7720b57cec5SDimitry Andric MachineInstrBuilder
7730b57cec5SDimitry Andric MachineIRBuilder::buildInsertVectorElement(const DstOp &Res, const SrcOp &Val,
7740b57cec5SDimitry Andric                                            const SrcOp &Elt, const SrcOp &Idx) {
7750b57cec5SDimitry Andric   return buildInstr(TargetOpcode::G_INSERT_VECTOR_ELT, Res, {Val, Elt, Idx});
7760b57cec5SDimitry Andric }
7770b57cec5SDimitry Andric 
7780b57cec5SDimitry Andric MachineInstrBuilder
7790b57cec5SDimitry Andric MachineIRBuilder::buildExtractVectorElement(const DstOp &Res, const SrcOp &Val,
7800b57cec5SDimitry Andric                                             const SrcOp &Idx) {
7810b57cec5SDimitry Andric   return buildInstr(TargetOpcode::G_EXTRACT_VECTOR_ELT, Res, {Val, Idx});
7820b57cec5SDimitry Andric }
7830b57cec5SDimitry Andric 
7840b57cec5SDimitry Andric MachineInstrBuilder MachineIRBuilder::buildAtomicCmpXchgWithSuccess(
7850b57cec5SDimitry Andric     Register OldValRes, Register SuccessRes, Register Addr, Register CmpVal,
7860b57cec5SDimitry Andric     Register NewVal, MachineMemOperand &MMO) {
7870b57cec5SDimitry Andric #ifndef NDEBUG
7880b57cec5SDimitry Andric   LLT OldValResTy = getMRI()->getType(OldValRes);
7890b57cec5SDimitry Andric   LLT SuccessResTy = getMRI()->getType(SuccessRes);
7900b57cec5SDimitry Andric   LLT AddrTy = getMRI()->getType(Addr);
7910b57cec5SDimitry Andric   LLT CmpValTy = getMRI()->getType(CmpVal);
7920b57cec5SDimitry Andric   LLT NewValTy = getMRI()->getType(NewVal);
7930b57cec5SDimitry Andric   assert(OldValResTy.isScalar() && "invalid operand type");
7940b57cec5SDimitry Andric   assert(SuccessResTy.isScalar() && "invalid operand type");
7950b57cec5SDimitry Andric   assert(AddrTy.isPointer() && "invalid operand type");
7960b57cec5SDimitry Andric   assert(CmpValTy.isValid() && "invalid operand type");
7970b57cec5SDimitry Andric   assert(NewValTy.isValid() && "invalid operand type");
7980b57cec5SDimitry Andric   assert(OldValResTy == CmpValTy && "type mismatch");
7990b57cec5SDimitry Andric   assert(OldValResTy == NewValTy && "type mismatch");
8000b57cec5SDimitry Andric #endif
8010b57cec5SDimitry Andric 
8020b57cec5SDimitry Andric   return buildInstr(TargetOpcode::G_ATOMIC_CMPXCHG_WITH_SUCCESS)
8030b57cec5SDimitry Andric       .addDef(OldValRes)
8040b57cec5SDimitry Andric       .addDef(SuccessRes)
8050b57cec5SDimitry Andric       .addUse(Addr)
8060b57cec5SDimitry Andric       .addUse(CmpVal)
8070b57cec5SDimitry Andric       .addUse(NewVal)
8080b57cec5SDimitry Andric       .addMemOperand(&MMO);
8090b57cec5SDimitry Andric }
8100b57cec5SDimitry Andric 
8110b57cec5SDimitry Andric MachineInstrBuilder
8120b57cec5SDimitry Andric MachineIRBuilder::buildAtomicCmpXchg(Register OldValRes, Register Addr,
8130b57cec5SDimitry Andric                                      Register CmpVal, Register NewVal,
8140b57cec5SDimitry Andric                                      MachineMemOperand &MMO) {
8150b57cec5SDimitry Andric #ifndef NDEBUG
8160b57cec5SDimitry Andric   LLT OldValResTy = getMRI()->getType(OldValRes);
8170b57cec5SDimitry Andric   LLT AddrTy = getMRI()->getType(Addr);
8180b57cec5SDimitry Andric   LLT CmpValTy = getMRI()->getType(CmpVal);
8190b57cec5SDimitry Andric   LLT NewValTy = getMRI()->getType(NewVal);
8200b57cec5SDimitry Andric   assert(OldValResTy.isScalar() && "invalid operand type");
8210b57cec5SDimitry Andric   assert(AddrTy.isPointer() && "invalid operand type");
8220b57cec5SDimitry Andric   assert(CmpValTy.isValid() && "invalid operand type");
8230b57cec5SDimitry Andric   assert(NewValTy.isValid() && "invalid operand type");
8240b57cec5SDimitry Andric   assert(OldValResTy == CmpValTy && "type mismatch");
8250b57cec5SDimitry Andric   assert(OldValResTy == NewValTy && "type mismatch");
8260b57cec5SDimitry Andric #endif
8270b57cec5SDimitry Andric 
8280b57cec5SDimitry Andric   return buildInstr(TargetOpcode::G_ATOMIC_CMPXCHG)
8290b57cec5SDimitry Andric       .addDef(OldValRes)
8300b57cec5SDimitry Andric       .addUse(Addr)
8310b57cec5SDimitry Andric       .addUse(CmpVal)
8320b57cec5SDimitry Andric       .addUse(NewVal)
8330b57cec5SDimitry Andric       .addMemOperand(&MMO);
8340b57cec5SDimitry Andric }
8350b57cec5SDimitry Andric 
8368bcb0991SDimitry Andric MachineInstrBuilder MachineIRBuilder::buildAtomicRMW(
8378bcb0991SDimitry Andric   unsigned Opcode, const DstOp &OldValRes,
8388bcb0991SDimitry Andric   const SrcOp &Addr, const SrcOp &Val,
8390b57cec5SDimitry Andric   MachineMemOperand &MMO) {
8408bcb0991SDimitry Andric 
8410b57cec5SDimitry Andric #ifndef NDEBUG
8428bcb0991SDimitry Andric   LLT OldValResTy = OldValRes.getLLTTy(*getMRI());
8438bcb0991SDimitry Andric   LLT AddrTy = Addr.getLLTTy(*getMRI());
8448bcb0991SDimitry Andric   LLT ValTy = Val.getLLTTy(*getMRI());
8450b57cec5SDimitry Andric   assert(OldValResTy.isScalar() && "invalid operand type");
8460b57cec5SDimitry Andric   assert(AddrTy.isPointer() && "invalid operand type");
8470b57cec5SDimitry Andric   assert(ValTy.isValid() && "invalid operand type");
8480b57cec5SDimitry Andric   assert(OldValResTy == ValTy && "type mismatch");
8498bcb0991SDimitry Andric   assert(MMO.isAtomic() && "not atomic mem operand");
8500b57cec5SDimitry Andric #endif
8510b57cec5SDimitry Andric 
8528bcb0991SDimitry Andric   auto MIB = buildInstr(Opcode);
8538bcb0991SDimitry Andric   OldValRes.addDefToMIB(*getMRI(), MIB);
8548bcb0991SDimitry Andric   Addr.addSrcToMIB(MIB);
8558bcb0991SDimitry Andric   Val.addSrcToMIB(MIB);
8568bcb0991SDimitry Andric   MIB.addMemOperand(&MMO);
8578bcb0991SDimitry Andric   return MIB;
8580b57cec5SDimitry Andric }
8590b57cec5SDimitry Andric 
8600b57cec5SDimitry Andric MachineInstrBuilder
8610b57cec5SDimitry Andric MachineIRBuilder::buildAtomicRMWXchg(Register OldValRes, Register Addr,
8620b57cec5SDimitry Andric                                      Register Val, MachineMemOperand &MMO) {
8630b57cec5SDimitry Andric   return buildAtomicRMW(TargetOpcode::G_ATOMICRMW_XCHG, OldValRes, Addr, Val,
8640b57cec5SDimitry Andric                         MMO);
8650b57cec5SDimitry Andric }
8660b57cec5SDimitry Andric MachineInstrBuilder
8670b57cec5SDimitry Andric MachineIRBuilder::buildAtomicRMWAdd(Register OldValRes, Register Addr,
8680b57cec5SDimitry Andric                                     Register Val, MachineMemOperand &MMO) {
8690b57cec5SDimitry Andric   return buildAtomicRMW(TargetOpcode::G_ATOMICRMW_ADD, OldValRes, Addr, Val,
8700b57cec5SDimitry Andric                         MMO);
8710b57cec5SDimitry Andric }
8720b57cec5SDimitry Andric MachineInstrBuilder
8730b57cec5SDimitry Andric MachineIRBuilder::buildAtomicRMWSub(Register OldValRes, Register Addr,
8740b57cec5SDimitry Andric                                     Register Val, MachineMemOperand &MMO) {
8750b57cec5SDimitry Andric   return buildAtomicRMW(TargetOpcode::G_ATOMICRMW_SUB, OldValRes, Addr, Val,
8760b57cec5SDimitry Andric                         MMO);
8770b57cec5SDimitry Andric }
8780b57cec5SDimitry Andric MachineInstrBuilder
8790b57cec5SDimitry Andric MachineIRBuilder::buildAtomicRMWAnd(Register OldValRes, Register Addr,
8800b57cec5SDimitry Andric                                     Register Val, MachineMemOperand &MMO) {
8810b57cec5SDimitry Andric   return buildAtomicRMW(TargetOpcode::G_ATOMICRMW_AND, OldValRes, Addr, Val,
8820b57cec5SDimitry Andric                         MMO);
8830b57cec5SDimitry Andric }
8840b57cec5SDimitry Andric MachineInstrBuilder
8850b57cec5SDimitry Andric MachineIRBuilder::buildAtomicRMWNand(Register OldValRes, Register Addr,
8860b57cec5SDimitry Andric                                      Register Val, MachineMemOperand &MMO) {
8870b57cec5SDimitry Andric   return buildAtomicRMW(TargetOpcode::G_ATOMICRMW_NAND, OldValRes, Addr, Val,
8880b57cec5SDimitry Andric                         MMO);
8890b57cec5SDimitry Andric }
8900b57cec5SDimitry Andric MachineInstrBuilder MachineIRBuilder::buildAtomicRMWOr(Register OldValRes,
8910b57cec5SDimitry Andric                                                        Register Addr,
8920b57cec5SDimitry Andric                                                        Register Val,
8930b57cec5SDimitry Andric                                                        MachineMemOperand &MMO) {
8940b57cec5SDimitry Andric   return buildAtomicRMW(TargetOpcode::G_ATOMICRMW_OR, OldValRes, Addr, Val,
8950b57cec5SDimitry Andric                         MMO);
8960b57cec5SDimitry Andric }
8970b57cec5SDimitry Andric MachineInstrBuilder
8980b57cec5SDimitry Andric MachineIRBuilder::buildAtomicRMWXor(Register OldValRes, Register Addr,
8990b57cec5SDimitry Andric                                     Register Val, MachineMemOperand &MMO) {
9000b57cec5SDimitry Andric   return buildAtomicRMW(TargetOpcode::G_ATOMICRMW_XOR, OldValRes, Addr, Val,
9010b57cec5SDimitry Andric                         MMO);
9020b57cec5SDimitry Andric }
9030b57cec5SDimitry Andric MachineInstrBuilder
9040b57cec5SDimitry Andric MachineIRBuilder::buildAtomicRMWMax(Register OldValRes, Register Addr,
9050b57cec5SDimitry Andric                                     Register Val, MachineMemOperand &MMO) {
9060b57cec5SDimitry Andric   return buildAtomicRMW(TargetOpcode::G_ATOMICRMW_MAX, OldValRes, Addr, Val,
9070b57cec5SDimitry Andric                         MMO);
9080b57cec5SDimitry Andric }
9090b57cec5SDimitry Andric MachineInstrBuilder
9100b57cec5SDimitry Andric MachineIRBuilder::buildAtomicRMWMin(Register OldValRes, Register Addr,
9110b57cec5SDimitry Andric                                     Register Val, MachineMemOperand &MMO) {
9120b57cec5SDimitry Andric   return buildAtomicRMW(TargetOpcode::G_ATOMICRMW_MIN, OldValRes, Addr, Val,
9130b57cec5SDimitry Andric                         MMO);
9140b57cec5SDimitry Andric }
9150b57cec5SDimitry Andric MachineInstrBuilder
9160b57cec5SDimitry Andric MachineIRBuilder::buildAtomicRMWUmax(Register OldValRes, Register Addr,
9170b57cec5SDimitry Andric                                      Register Val, MachineMemOperand &MMO) {
9180b57cec5SDimitry Andric   return buildAtomicRMW(TargetOpcode::G_ATOMICRMW_UMAX, OldValRes, Addr, Val,
9190b57cec5SDimitry Andric                         MMO);
9200b57cec5SDimitry Andric }
9210b57cec5SDimitry Andric MachineInstrBuilder
9220b57cec5SDimitry Andric MachineIRBuilder::buildAtomicRMWUmin(Register OldValRes, Register Addr,
9230b57cec5SDimitry Andric                                      Register Val, MachineMemOperand &MMO) {
9240b57cec5SDimitry Andric   return buildAtomicRMW(TargetOpcode::G_ATOMICRMW_UMIN, OldValRes, Addr, Val,
9250b57cec5SDimitry Andric                         MMO);
9260b57cec5SDimitry Andric }
9270b57cec5SDimitry Andric 
9280b57cec5SDimitry Andric MachineInstrBuilder
9298bcb0991SDimitry Andric MachineIRBuilder::buildAtomicRMWFAdd(
9308bcb0991SDimitry Andric   const DstOp &OldValRes, const SrcOp &Addr, const SrcOp &Val,
9318bcb0991SDimitry Andric   MachineMemOperand &MMO) {
9328bcb0991SDimitry Andric   return buildAtomicRMW(TargetOpcode::G_ATOMICRMW_FADD, OldValRes, Addr, Val,
9338bcb0991SDimitry Andric                         MMO);
9348bcb0991SDimitry Andric }
9358bcb0991SDimitry Andric 
9368bcb0991SDimitry Andric MachineInstrBuilder
9378bcb0991SDimitry Andric MachineIRBuilder::buildAtomicRMWFSub(const DstOp &OldValRes, const SrcOp &Addr, const SrcOp &Val,
9388bcb0991SDimitry Andric                                      MachineMemOperand &MMO) {
9398bcb0991SDimitry Andric   return buildAtomicRMW(TargetOpcode::G_ATOMICRMW_FSUB, OldValRes, Addr, Val,
9408bcb0991SDimitry Andric                         MMO);
9418bcb0991SDimitry Andric }
9428bcb0991SDimitry Andric 
9438bcb0991SDimitry Andric MachineInstrBuilder
9440b57cec5SDimitry Andric MachineIRBuilder::buildFence(unsigned Ordering, unsigned Scope) {
9450b57cec5SDimitry Andric   return buildInstr(TargetOpcode::G_FENCE)
9460b57cec5SDimitry Andric     .addImm(Ordering)
9470b57cec5SDimitry Andric     .addImm(Scope);
9480b57cec5SDimitry Andric }
9490b57cec5SDimitry Andric 
9500b57cec5SDimitry Andric MachineInstrBuilder
9510b57cec5SDimitry Andric MachineIRBuilder::buildBlockAddress(Register Res, const BlockAddress *BA) {
9520b57cec5SDimitry Andric #ifndef NDEBUG
9530b57cec5SDimitry Andric   assert(getMRI()->getType(Res).isPointer() && "invalid res type");
9540b57cec5SDimitry Andric #endif
9550b57cec5SDimitry Andric 
9560b57cec5SDimitry Andric   return buildInstr(TargetOpcode::G_BLOCK_ADDR).addDef(Res).addBlockAddress(BA);
9570b57cec5SDimitry Andric }
9580b57cec5SDimitry Andric 
9595ffd83dbSDimitry Andric void MachineIRBuilder::validateTruncExt(const LLT DstTy, const LLT SrcTy,
9600b57cec5SDimitry Andric                                         bool IsExtend) {
9610b57cec5SDimitry Andric #ifndef NDEBUG
9620b57cec5SDimitry Andric   if (DstTy.isVector()) {
9630b57cec5SDimitry Andric     assert(SrcTy.isVector() && "mismatched cast between vector and non-vector");
9640b57cec5SDimitry Andric     assert(SrcTy.getNumElements() == DstTy.getNumElements() &&
9650b57cec5SDimitry Andric            "different number of elements in a trunc/ext");
9660b57cec5SDimitry Andric   } else
9670b57cec5SDimitry Andric     assert(DstTy.isScalar() && SrcTy.isScalar() && "invalid extend/trunc");
9680b57cec5SDimitry Andric 
9690b57cec5SDimitry Andric   if (IsExtend)
9700b57cec5SDimitry Andric     assert(DstTy.getSizeInBits() > SrcTy.getSizeInBits() &&
9710b57cec5SDimitry Andric            "invalid narrowing extend");
9720b57cec5SDimitry Andric   else
9730b57cec5SDimitry Andric     assert(DstTy.getSizeInBits() < SrcTy.getSizeInBits() &&
9740b57cec5SDimitry Andric            "invalid widening trunc");
9750b57cec5SDimitry Andric #endif
9760b57cec5SDimitry Andric }
9770b57cec5SDimitry Andric 
9785ffd83dbSDimitry Andric void MachineIRBuilder::validateSelectOp(const LLT ResTy, const LLT TstTy,
9795ffd83dbSDimitry Andric                                         const LLT Op0Ty, const LLT Op1Ty) {
9800b57cec5SDimitry Andric #ifndef NDEBUG
9810b57cec5SDimitry Andric   assert((ResTy.isScalar() || ResTy.isVector() || ResTy.isPointer()) &&
9820b57cec5SDimitry Andric          "invalid operand type");
9830b57cec5SDimitry Andric   assert((ResTy == Op0Ty && ResTy == Op1Ty) && "type mismatch");
9840b57cec5SDimitry Andric   if (ResTy.isScalar() || ResTy.isPointer())
9850b57cec5SDimitry Andric     assert(TstTy.isScalar() && "type mismatch");
9860b57cec5SDimitry Andric   else
9870b57cec5SDimitry Andric     assert((TstTy.isScalar() ||
9880b57cec5SDimitry Andric             (TstTy.isVector() &&
9890b57cec5SDimitry Andric              TstTy.getNumElements() == Op0Ty.getNumElements())) &&
9900b57cec5SDimitry Andric            "type mismatch");
9910b57cec5SDimitry Andric #endif
9920b57cec5SDimitry Andric }
9930b57cec5SDimitry Andric 
9940b57cec5SDimitry Andric MachineInstrBuilder MachineIRBuilder::buildInstr(unsigned Opc,
9950b57cec5SDimitry Andric                                                  ArrayRef<DstOp> DstOps,
9960b57cec5SDimitry Andric                                                  ArrayRef<SrcOp> SrcOps,
9970b57cec5SDimitry Andric                                                  Optional<unsigned> Flags) {
9980b57cec5SDimitry Andric   switch (Opc) {
9990b57cec5SDimitry Andric   default:
10000b57cec5SDimitry Andric     break;
10010b57cec5SDimitry Andric   case TargetOpcode::G_SELECT: {
10020b57cec5SDimitry Andric     assert(DstOps.size() == 1 && "Invalid select");
10030b57cec5SDimitry Andric     assert(SrcOps.size() == 3 && "Invalid select");
10040b57cec5SDimitry Andric     validateSelectOp(
10050b57cec5SDimitry Andric         DstOps[0].getLLTTy(*getMRI()), SrcOps[0].getLLTTy(*getMRI()),
10060b57cec5SDimitry Andric         SrcOps[1].getLLTTy(*getMRI()), SrcOps[2].getLLTTy(*getMRI()));
10070b57cec5SDimitry Andric     break;
10080b57cec5SDimitry Andric   }
1009e8d8bef9SDimitry Andric   case TargetOpcode::G_FNEG:
1010e8d8bef9SDimitry Andric   case TargetOpcode::G_ABS:
1011e8d8bef9SDimitry Andric     // All these are unary ops.
1012e8d8bef9SDimitry Andric     assert(DstOps.size() == 1 && "Invalid Dst");
1013e8d8bef9SDimitry Andric     assert(SrcOps.size() == 1 && "Invalid Srcs");
1014e8d8bef9SDimitry Andric     validateUnaryOp(DstOps[0].getLLTTy(*getMRI()),
1015e8d8bef9SDimitry Andric                     SrcOps[0].getLLTTy(*getMRI()));
1016e8d8bef9SDimitry Andric     break;
10170b57cec5SDimitry Andric   case TargetOpcode::G_ADD:
10180b57cec5SDimitry Andric   case TargetOpcode::G_AND:
10190b57cec5SDimitry Andric   case TargetOpcode::G_MUL:
10200b57cec5SDimitry Andric   case TargetOpcode::G_OR:
10210b57cec5SDimitry Andric   case TargetOpcode::G_SUB:
10220b57cec5SDimitry Andric   case TargetOpcode::G_XOR:
10230b57cec5SDimitry Andric   case TargetOpcode::G_UDIV:
10240b57cec5SDimitry Andric   case TargetOpcode::G_SDIV:
10250b57cec5SDimitry Andric   case TargetOpcode::G_UREM:
10260b57cec5SDimitry Andric   case TargetOpcode::G_SREM:
10270b57cec5SDimitry Andric   case TargetOpcode::G_SMIN:
10280b57cec5SDimitry Andric   case TargetOpcode::G_SMAX:
10290b57cec5SDimitry Andric   case TargetOpcode::G_UMIN:
10305ffd83dbSDimitry Andric   case TargetOpcode::G_UMAX:
10315ffd83dbSDimitry Andric   case TargetOpcode::G_UADDSAT:
10325ffd83dbSDimitry Andric   case TargetOpcode::G_SADDSAT:
10335ffd83dbSDimitry Andric   case TargetOpcode::G_USUBSAT:
10345ffd83dbSDimitry Andric   case TargetOpcode::G_SSUBSAT: {
10350b57cec5SDimitry Andric     // All these are binary ops.
10360b57cec5SDimitry Andric     assert(DstOps.size() == 1 && "Invalid Dst");
10370b57cec5SDimitry Andric     assert(SrcOps.size() == 2 && "Invalid Srcs");
10380b57cec5SDimitry Andric     validateBinaryOp(DstOps[0].getLLTTy(*getMRI()),
10390b57cec5SDimitry Andric                      SrcOps[0].getLLTTy(*getMRI()),
10400b57cec5SDimitry Andric                      SrcOps[1].getLLTTy(*getMRI()));
10410b57cec5SDimitry Andric     break;
10420b57cec5SDimitry Andric   }
10430b57cec5SDimitry Andric   case TargetOpcode::G_SHL:
10440b57cec5SDimitry Andric   case TargetOpcode::G_ASHR:
1045e8d8bef9SDimitry Andric   case TargetOpcode::G_LSHR:
1046e8d8bef9SDimitry Andric   case TargetOpcode::G_USHLSAT:
1047e8d8bef9SDimitry Andric   case TargetOpcode::G_SSHLSAT: {
10480b57cec5SDimitry Andric     assert(DstOps.size() == 1 && "Invalid Dst");
10490b57cec5SDimitry Andric     assert(SrcOps.size() == 2 && "Invalid Srcs");
10500b57cec5SDimitry Andric     validateShiftOp(DstOps[0].getLLTTy(*getMRI()),
10510b57cec5SDimitry Andric                     SrcOps[0].getLLTTy(*getMRI()),
10520b57cec5SDimitry Andric                     SrcOps[1].getLLTTy(*getMRI()));
10530b57cec5SDimitry Andric     break;
10540b57cec5SDimitry Andric   }
10550b57cec5SDimitry Andric   case TargetOpcode::G_SEXT:
10560b57cec5SDimitry Andric   case TargetOpcode::G_ZEXT:
10570b57cec5SDimitry Andric   case TargetOpcode::G_ANYEXT:
10580b57cec5SDimitry Andric     assert(DstOps.size() == 1 && "Invalid Dst");
10590b57cec5SDimitry Andric     assert(SrcOps.size() == 1 && "Invalid Srcs");
10600b57cec5SDimitry Andric     validateTruncExt(DstOps[0].getLLTTy(*getMRI()),
10610b57cec5SDimitry Andric                      SrcOps[0].getLLTTy(*getMRI()), true);
10620b57cec5SDimitry Andric     break;
10630b57cec5SDimitry Andric   case TargetOpcode::G_TRUNC:
10640b57cec5SDimitry Andric   case TargetOpcode::G_FPTRUNC: {
10650b57cec5SDimitry Andric     assert(DstOps.size() == 1 && "Invalid Dst");
10660b57cec5SDimitry Andric     assert(SrcOps.size() == 1 && "Invalid Srcs");
10670b57cec5SDimitry Andric     validateTruncExt(DstOps[0].getLLTTy(*getMRI()),
10680b57cec5SDimitry Andric                      SrcOps[0].getLLTTy(*getMRI()), false);
10690b57cec5SDimitry Andric     break;
10700b57cec5SDimitry Andric   }
10715ffd83dbSDimitry Andric   case TargetOpcode::G_BITCAST: {
10725ffd83dbSDimitry Andric     assert(DstOps.size() == 1 && "Invalid Dst");
10735ffd83dbSDimitry Andric     assert(SrcOps.size() == 1 && "Invalid Srcs");
10745ffd83dbSDimitry Andric     assert(DstOps[0].getLLTTy(*getMRI()).getSizeInBits() ==
10755ffd83dbSDimitry Andric            SrcOps[0].getLLTTy(*getMRI()).getSizeInBits() && "invalid bitcast");
10765ffd83dbSDimitry Andric     break;
10775ffd83dbSDimitry Andric   }
10780b57cec5SDimitry Andric   case TargetOpcode::COPY:
10790b57cec5SDimitry Andric     assert(DstOps.size() == 1 && "Invalid Dst");
10800b57cec5SDimitry Andric     // If the caller wants to add a subreg source it has to be done separately
10810b57cec5SDimitry Andric     // so we may not have any SrcOps at this point yet.
10820b57cec5SDimitry Andric     break;
10830b57cec5SDimitry Andric   case TargetOpcode::G_FCMP:
10840b57cec5SDimitry Andric   case TargetOpcode::G_ICMP: {
10850b57cec5SDimitry Andric     assert(DstOps.size() == 1 && "Invalid Dst Operands");
10860b57cec5SDimitry Andric     assert(SrcOps.size() == 3 && "Invalid Src Operands");
10870b57cec5SDimitry Andric     // For F/ICMP, the first src operand is the predicate, followed by
10880b57cec5SDimitry Andric     // the two comparands.
10890b57cec5SDimitry Andric     assert(SrcOps[0].getSrcOpKind() == SrcOp::SrcType::Ty_Predicate &&
10900b57cec5SDimitry Andric            "Expecting predicate");
10910b57cec5SDimitry Andric     assert([&]() -> bool {
10920b57cec5SDimitry Andric       CmpInst::Predicate Pred = SrcOps[0].getPredicate();
10930b57cec5SDimitry Andric       return Opc == TargetOpcode::G_ICMP ? CmpInst::isIntPredicate(Pred)
10940b57cec5SDimitry Andric                                          : CmpInst::isFPPredicate(Pred);
10950b57cec5SDimitry Andric     }() && "Invalid predicate");
10960b57cec5SDimitry Andric     assert(SrcOps[1].getLLTTy(*getMRI()) == SrcOps[2].getLLTTy(*getMRI()) &&
10970b57cec5SDimitry Andric            "Type mismatch");
10980b57cec5SDimitry Andric     assert([&]() -> bool {
10990b57cec5SDimitry Andric       LLT Op0Ty = SrcOps[1].getLLTTy(*getMRI());
11000b57cec5SDimitry Andric       LLT DstTy = DstOps[0].getLLTTy(*getMRI());
11010b57cec5SDimitry Andric       if (Op0Ty.isScalar() || Op0Ty.isPointer())
11020b57cec5SDimitry Andric         return DstTy.isScalar();
11030b57cec5SDimitry Andric       else
11040b57cec5SDimitry Andric         return DstTy.isVector() &&
11050b57cec5SDimitry Andric                DstTy.getNumElements() == Op0Ty.getNumElements();
11060b57cec5SDimitry Andric     }() && "Type Mismatch");
11070b57cec5SDimitry Andric     break;
11080b57cec5SDimitry Andric   }
11090b57cec5SDimitry Andric   case TargetOpcode::G_UNMERGE_VALUES: {
11100b57cec5SDimitry Andric     assert(!DstOps.empty() && "Invalid trivial sequence");
11110b57cec5SDimitry Andric     assert(SrcOps.size() == 1 && "Invalid src for Unmerge");
1112e8d8bef9SDimitry Andric     assert(llvm::all_of(DstOps,
11130b57cec5SDimitry Andric                         [&, this](const DstOp &Op) {
11140b57cec5SDimitry Andric                           return Op.getLLTTy(*getMRI()) ==
11150b57cec5SDimitry Andric                                  DstOps[0].getLLTTy(*getMRI());
11160b57cec5SDimitry Andric                         }) &&
11170b57cec5SDimitry Andric            "type mismatch in output list");
1118*fe6060f1SDimitry Andric     assert((TypeSize::ScalarTy)DstOps.size() *
1119*fe6060f1SDimitry Andric                    DstOps[0].getLLTTy(*getMRI()).getSizeInBits() ==
11200b57cec5SDimitry Andric                SrcOps[0].getLLTTy(*getMRI()).getSizeInBits() &&
11210b57cec5SDimitry Andric            "input operands do not cover output register");
11220b57cec5SDimitry Andric     break;
11230b57cec5SDimitry Andric   }
11240b57cec5SDimitry Andric   case TargetOpcode::G_MERGE_VALUES: {
11250b57cec5SDimitry Andric     assert(!SrcOps.empty() && "invalid trivial sequence");
11260b57cec5SDimitry Andric     assert(DstOps.size() == 1 && "Invalid Dst");
1127e8d8bef9SDimitry Andric     assert(llvm::all_of(SrcOps,
11280b57cec5SDimitry Andric                         [&, this](const SrcOp &Op) {
11290b57cec5SDimitry Andric                           return Op.getLLTTy(*getMRI()) ==
11300b57cec5SDimitry Andric                                  SrcOps[0].getLLTTy(*getMRI());
11310b57cec5SDimitry Andric                         }) &&
11320b57cec5SDimitry Andric            "type mismatch in input list");
1133*fe6060f1SDimitry Andric     assert((TypeSize::ScalarTy)SrcOps.size() *
1134*fe6060f1SDimitry Andric                    SrcOps[0].getLLTTy(*getMRI()).getSizeInBits() ==
11350b57cec5SDimitry Andric                DstOps[0].getLLTTy(*getMRI()).getSizeInBits() &&
11360b57cec5SDimitry Andric            "input operands do not cover output register");
11370b57cec5SDimitry Andric     if (SrcOps.size() == 1)
11380b57cec5SDimitry Andric       return buildCast(DstOps[0], SrcOps[0]);
11398bcb0991SDimitry Andric     if (DstOps[0].getLLTTy(*getMRI()).isVector()) {
11408bcb0991SDimitry Andric       if (SrcOps[0].getLLTTy(*getMRI()).isVector())
11410b57cec5SDimitry Andric         return buildInstr(TargetOpcode::G_CONCAT_VECTORS, DstOps, SrcOps);
11428bcb0991SDimitry Andric       return buildInstr(TargetOpcode::G_BUILD_VECTOR, DstOps, SrcOps);
11438bcb0991SDimitry Andric     }
11440b57cec5SDimitry Andric     break;
11450b57cec5SDimitry Andric   }
11460b57cec5SDimitry Andric   case TargetOpcode::G_EXTRACT_VECTOR_ELT: {
11470b57cec5SDimitry Andric     assert(DstOps.size() == 1 && "Invalid Dst size");
11480b57cec5SDimitry Andric     assert(SrcOps.size() == 2 && "Invalid Src size");
11490b57cec5SDimitry Andric     assert(SrcOps[0].getLLTTy(*getMRI()).isVector() && "Invalid operand type");
11500b57cec5SDimitry Andric     assert((DstOps[0].getLLTTy(*getMRI()).isScalar() ||
11510b57cec5SDimitry Andric             DstOps[0].getLLTTy(*getMRI()).isPointer()) &&
11520b57cec5SDimitry Andric            "Invalid operand type");
11530b57cec5SDimitry Andric     assert(SrcOps[1].getLLTTy(*getMRI()).isScalar() && "Invalid operand type");
11540b57cec5SDimitry Andric     assert(SrcOps[0].getLLTTy(*getMRI()).getElementType() ==
11550b57cec5SDimitry Andric                DstOps[0].getLLTTy(*getMRI()) &&
11560b57cec5SDimitry Andric            "Type mismatch");
11570b57cec5SDimitry Andric     break;
11580b57cec5SDimitry Andric   }
11590b57cec5SDimitry Andric   case TargetOpcode::G_INSERT_VECTOR_ELT: {
11600b57cec5SDimitry Andric     assert(DstOps.size() == 1 && "Invalid dst size");
11610b57cec5SDimitry Andric     assert(SrcOps.size() == 3 && "Invalid src size");
11620b57cec5SDimitry Andric     assert(DstOps[0].getLLTTy(*getMRI()).isVector() &&
11630b57cec5SDimitry Andric            SrcOps[0].getLLTTy(*getMRI()).isVector() && "Invalid operand type");
11640b57cec5SDimitry Andric     assert(DstOps[0].getLLTTy(*getMRI()).getElementType() ==
11650b57cec5SDimitry Andric                SrcOps[1].getLLTTy(*getMRI()) &&
11660b57cec5SDimitry Andric            "Type mismatch");
11670b57cec5SDimitry Andric     assert(SrcOps[2].getLLTTy(*getMRI()).isScalar() && "Invalid index");
11680b57cec5SDimitry Andric     assert(DstOps[0].getLLTTy(*getMRI()).getNumElements() ==
11690b57cec5SDimitry Andric                SrcOps[0].getLLTTy(*getMRI()).getNumElements() &&
11700b57cec5SDimitry Andric            "Type mismatch");
11710b57cec5SDimitry Andric     break;
11720b57cec5SDimitry Andric   }
11730b57cec5SDimitry Andric   case TargetOpcode::G_BUILD_VECTOR: {
11740b57cec5SDimitry Andric     assert((!SrcOps.empty() || SrcOps.size() < 2) &&
11750b57cec5SDimitry Andric            "Must have at least 2 operands");
11760b57cec5SDimitry Andric     assert(DstOps.size() == 1 && "Invalid DstOps");
11770b57cec5SDimitry Andric     assert(DstOps[0].getLLTTy(*getMRI()).isVector() &&
11780b57cec5SDimitry Andric            "Res type must be a vector");
1179e8d8bef9SDimitry Andric     assert(llvm::all_of(SrcOps,
11800b57cec5SDimitry Andric                         [&, this](const SrcOp &Op) {
11810b57cec5SDimitry Andric                           return Op.getLLTTy(*getMRI()) ==
11820b57cec5SDimitry Andric                                  SrcOps[0].getLLTTy(*getMRI());
11830b57cec5SDimitry Andric                         }) &&
11840b57cec5SDimitry Andric            "type mismatch in input list");
1185*fe6060f1SDimitry Andric     assert((TypeSize::ScalarTy)SrcOps.size() *
1186*fe6060f1SDimitry Andric                    SrcOps[0].getLLTTy(*getMRI()).getSizeInBits() ==
11870b57cec5SDimitry Andric                DstOps[0].getLLTTy(*getMRI()).getSizeInBits() &&
11880b57cec5SDimitry Andric            "input scalars do not exactly cover the output vector register");
11890b57cec5SDimitry Andric     break;
11900b57cec5SDimitry Andric   }
11910b57cec5SDimitry Andric   case TargetOpcode::G_BUILD_VECTOR_TRUNC: {
11920b57cec5SDimitry Andric     assert((!SrcOps.empty() || SrcOps.size() < 2) &&
11930b57cec5SDimitry Andric            "Must have at least 2 operands");
11940b57cec5SDimitry Andric     assert(DstOps.size() == 1 && "Invalid DstOps");
11950b57cec5SDimitry Andric     assert(DstOps[0].getLLTTy(*getMRI()).isVector() &&
11960b57cec5SDimitry Andric            "Res type must be a vector");
1197e8d8bef9SDimitry Andric     assert(llvm::all_of(SrcOps,
11980b57cec5SDimitry Andric                         [&, this](const SrcOp &Op) {
11990b57cec5SDimitry Andric                           return Op.getLLTTy(*getMRI()) ==
12000b57cec5SDimitry Andric                                  SrcOps[0].getLLTTy(*getMRI());
12010b57cec5SDimitry Andric                         }) &&
12020b57cec5SDimitry Andric            "type mismatch in input list");
12030b57cec5SDimitry Andric     if (SrcOps[0].getLLTTy(*getMRI()).getSizeInBits() ==
12040b57cec5SDimitry Andric         DstOps[0].getLLTTy(*getMRI()).getElementType().getSizeInBits())
12050b57cec5SDimitry Andric       return buildInstr(TargetOpcode::G_BUILD_VECTOR, DstOps, SrcOps);
12060b57cec5SDimitry Andric     break;
12070b57cec5SDimitry Andric   }
12080b57cec5SDimitry Andric   case TargetOpcode::G_CONCAT_VECTORS: {
12090b57cec5SDimitry Andric     assert(DstOps.size() == 1 && "Invalid DstOps");
12100b57cec5SDimitry Andric     assert((!SrcOps.empty() || SrcOps.size() < 2) &&
12110b57cec5SDimitry Andric            "Must have at least 2 operands");
1212e8d8bef9SDimitry Andric     assert(llvm::all_of(SrcOps,
12130b57cec5SDimitry Andric                         [&, this](const SrcOp &Op) {
12140b57cec5SDimitry Andric                           return (Op.getLLTTy(*getMRI()).isVector() &&
12150b57cec5SDimitry Andric                                   Op.getLLTTy(*getMRI()) ==
12160b57cec5SDimitry Andric                                       SrcOps[0].getLLTTy(*getMRI()));
12170b57cec5SDimitry Andric                         }) &&
12180b57cec5SDimitry Andric            "type mismatch in input list");
1219*fe6060f1SDimitry Andric     assert((TypeSize::ScalarTy)SrcOps.size() *
1220*fe6060f1SDimitry Andric                    SrcOps[0].getLLTTy(*getMRI()).getSizeInBits() ==
12210b57cec5SDimitry Andric                DstOps[0].getLLTTy(*getMRI()).getSizeInBits() &&
12220b57cec5SDimitry Andric            "input vectors do not exactly cover the output vector register");
12230b57cec5SDimitry Andric     break;
12240b57cec5SDimitry Andric   }
12250b57cec5SDimitry Andric   case TargetOpcode::G_UADDE: {
12260b57cec5SDimitry Andric     assert(DstOps.size() == 2 && "Invalid no of dst operands");
12270b57cec5SDimitry Andric     assert(SrcOps.size() == 3 && "Invalid no of src operands");
12280b57cec5SDimitry Andric     assert(DstOps[0].getLLTTy(*getMRI()).isScalar() && "Invalid operand");
12290b57cec5SDimitry Andric     assert((DstOps[0].getLLTTy(*getMRI()) == SrcOps[0].getLLTTy(*getMRI())) &&
12300b57cec5SDimitry Andric            (DstOps[0].getLLTTy(*getMRI()) == SrcOps[1].getLLTTy(*getMRI())) &&
12310b57cec5SDimitry Andric            "Invalid operand");
12320b57cec5SDimitry Andric     assert(DstOps[1].getLLTTy(*getMRI()).isScalar() && "Invalid operand");
12330b57cec5SDimitry Andric     assert(DstOps[1].getLLTTy(*getMRI()) == SrcOps[2].getLLTTy(*getMRI()) &&
12340b57cec5SDimitry Andric            "type mismatch");
12350b57cec5SDimitry Andric     break;
12360b57cec5SDimitry Andric   }
12370b57cec5SDimitry Andric   }
12380b57cec5SDimitry Andric 
12390b57cec5SDimitry Andric   auto MIB = buildInstr(Opc);
12400b57cec5SDimitry Andric   for (const DstOp &Op : DstOps)
12410b57cec5SDimitry Andric     Op.addDefToMIB(*getMRI(), MIB);
12420b57cec5SDimitry Andric   for (const SrcOp &Op : SrcOps)
12430b57cec5SDimitry Andric     Op.addSrcToMIB(MIB);
12440b57cec5SDimitry Andric   if (Flags)
12450b57cec5SDimitry Andric     MIB->setFlags(*Flags);
12460b57cec5SDimitry Andric   return MIB;
12470b57cec5SDimitry Andric }
1248