xref: /freebsd/contrib/llvm-project/llvm/lib/CodeGen/GlobalISel/MachineIRBuilder.cpp (revision 13138422bc354a1ec35f53a27c4efeccdffc5639)
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"
120b57cec5SDimitry Andric #include "llvm/CodeGen/GlobalISel/GISelChangeObserver.h"
130b57cec5SDimitry Andric 
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 void MachineIRBuilder::setMBB(MachineBasicBlock &MBB) {
370b57cec5SDimitry Andric   State.MBB = &MBB;
380b57cec5SDimitry Andric   State.II = MBB.end();
390b57cec5SDimitry Andric   assert(&getMF() == MBB.getParent() &&
400b57cec5SDimitry Andric          "Basic block is in a different function");
410b57cec5SDimitry Andric }
420b57cec5SDimitry Andric 
430b57cec5SDimitry Andric void MachineIRBuilder::setInstr(MachineInstr &MI) {
440b57cec5SDimitry Andric   assert(MI.getParent() && "Instruction is not part of a basic block");
450b57cec5SDimitry Andric   setMBB(*MI.getParent());
460b57cec5SDimitry Andric   State.II = MI.getIterator();
470b57cec5SDimitry Andric }
480b57cec5SDimitry Andric 
490b57cec5SDimitry Andric void MachineIRBuilder::setCSEInfo(GISelCSEInfo *Info) { State.CSEInfo = Info; }
500b57cec5SDimitry Andric 
510b57cec5SDimitry Andric void MachineIRBuilder::setInsertPt(MachineBasicBlock &MBB,
520b57cec5SDimitry Andric                                    MachineBasicBlock::iterator II) {
530b57cec5SDimitry Andric   assert(MBB.getParent() == &getMF() &&
540b57cec5SDimitry Andric          "Basic block is in a different function");
550b57cec5SDimitry Andric   State.MBB = &MBB;
560b57cec5SDimitry Andric   State.II = II;
570b57cec5SDimitry Andric }
580b57cec5SDimitry Andric 
590b57cec5SDimitry Andric void MachineIRBuilder::recordInsertion(MachineInstr *InsertedInstr) const {
600b57cec5SDimitry Andric   if (State.Observer)
610b57cec5SDimitry Andric     State.Observer->createdInstr(*InsertedInstr);
620b57cec5SDimitry Andric }
630b57cec5SDimitry Andric 
640b57cec5SDimitry Andric void MachineIRBuilder::setChangeObserver(GISelChangeObserver &Observer) {
650b57cec5SDimitry Andric   State.Observer = &Observer;
660b57cec5SDimitry Andric }
670b57cec5SDimitry Andric 
680b57cec5SDimitry Andric void MachineIRBuilder::stopObservingChanges() { State.Observer = nullptr; }
690b57cec5SDimitry Andric 
700b57cec5SDimitry Andric //------------------------------------------------------------------------------
710b57cec5SDimitry Andric // Build instruction variants.
720b57cec5SDimitry Andric //------------------------------------------------------------------------------
730b57cec5SDimitry Andric 
740b57cec5SDimitry Andric MachineInstrBuilder MachineIRBuilder::buildInstr(unsigned Opcode) {
750b57cec5SDimitry Andric   return insertInstr(buildInstrNoInsert(Opcode));
760b57cec5SDimitry Andric }
770b57cec5SDimitry Andric 
780b57cec5SDimitry Andric MachineInstrBuilder MachineIRBuilder::buildInstrNoInsert(unsigned Opcode) {
790b57cec5SDimitry Andric   MachineInstrBuilder MIB = BuildMI(getMF(), getDL(), getTII().get(Opcode));
800b57cec5SDimitry Andric   return MIB;
810b57cec5SDimitry Andric }
820b57cec5SDimitry Andric 
830b57cec5SDimitry Andric MachineInstrBuilder MachineIRBuilder::insertInstr(MachineInstrBuilder MIB) {
840b57cec5SDimitry Andric   getMBB().insert(getInsertPt(), MIB);
850b57cec5SDimitry Andric   recordInsertion(MIB);
860b57cec5SDimitry Andric   return MIB;
870b57cec5SDimitry Andric }
880b57cec5SDimitry Andric 
890b57cec5SDimitry Andric MachineInstrBuilder
900b57cec5SDimitry Andric MachineIRBuilder::buildDirectDbgValue(Register Reg, const MDNode *Variable,
910b57cec5SDimitry Andric                                       const MDNode *Expr) {
920b57cec5SDimitry Andric   assert(isa<DILocalVariable>(Variable) && "not a variable");
930b57cec5SDimitry Andric   assert(cast<DIExpression>(Expr)->isValid() && "not an expression");
940b57cec5SDimitry Andric   assert(
950b57cec5SDimitry Andric       cast<DILocalVariable>(Variable)->isValidLocationForIntrinsic(getDL()) &&
960b57cec5SDimitry Andric       "Expected inlined-at fields to agree");
970b57cec5SDimitry Andric   return insertInstr(BuildMI(getMF(), getDL(),
980b57cec5SDimitry Andric                              getTII().get(TargetOpcode::DBG_VALUE),
990b57cec5SDimitry Andric                              /*IsIndirect*/ false, Reg, Variable, Expr));
1000b57cec5SDimitry Andric }
1010b57cec5SDimitry Andric 
1020b57cec5SDimitry Andric MachineInstrBuilder
1030b57cec5SDimitry Andric MachineIRBuilder::buildIndirectDbgValue(Register Reg, const MDNode *Variable,
1040b57cec5SDimitry Andric                                         const MDNode *Expr) {
1050b57cec5SDimitry Andric   assert(isa<DILocalVariable>(Variable) && "not a variable");
1060b57cec5SDimitry Andric   assert(cast<DIExpression>(Expr)->isValid() && "not an expression");
1070b57cec5SDimitry Andric   assert(
1080b57cec5SDimitry Andric       cast<DILocalVariable>(Variable)->isValidLocationForIntrinsic(getDL()) &&
1090b57cec5SDimitry Andric       "Expected inlined-at fields to agree");
1100b57cec5SDimitry Andric   return insertInstr(BuildMI(getMF(), getDL(),
1110b57cec5SDimitry Andric                              getTII().get(TargetOpcode::DBG_VALUE),
112*13138422SDimitry Andric                              /*IsIndirect*/ true, Reg, Variable, Expr));
1130b57cec5SDimitry Andric }
1140b57cec5SDimitry Andric 
1150b57cec5SDimitry Andric MachineInstrBuilder MachineIRBuilder::buildFIDbgValue(int FI,
1160b57cec5SDimitry Andric                                                       const MDNode *Variable,
1170b57cec5SDimitry Andric                                                       const MDNode *Expr) {
1180b57cec5SDimitry Andric   assert(isa<DILocalVariable>(Variable) && "not a variable");
1190b57cec5SDimitry Andric   assert(cast<DIExpression>(Expr)->isValid() && "not an expression");
1200b57cec5SDimitry Andric   assert(
1210b57cec5SDimitry Andric       cast<DILocalVariable>(Variable)->isValidLocationForIntrinsic(getDL()) &&
1220b57cec5SDimitry Andric       "Expected inlined-at fields to agree");
1230b57cec5SDimitry Andric   return buildInstr(TargetOpcode::DBG_VALUE)
1240b57cec5SDimitry Andric       .addFrameIndex(FI)
125*13138422SDimitry Andric       .addImm(0)
1260b57cec5SDimitry Andric       .addMetadata(Variable)
127*13138422SDimitry Andric       .addMetadata(Expr);
1280b57cec5SDimitry Andric }
1290b57cec5SDimitry Andric 
1300b57cec5SDimitry Andric MachineInstrBuilder MachineIRBuilder::buildConstDbgValue(const Constant &C,
1310b57cec5SDimitry Andric                                                          const MDNode *Variable,
1320b57cec5SDimitry Andric                                                          const MDNode *Expr) {
1330b57cec5SDimitry Andric   assert(isa<DILocalVariable>(Variable) && "not a variable");
1340b57cec5SDimitry Andric   assert(cast<DIExpression>(Expr)->isValid() && "not an expression");
1350b57cec5SDimitry Andric   assert(
1360b57cec5SDimitry Andric       cast<DILocalVariable>(Variable)->isValidLocationForIntrinsic(getDL()) &&
1370b57cec5SDimitry Andric       "Expected inlined-at fields to agree");
1380b57cec5SDimitry Andric   auto MIB = buildInstr(TargetOpcode::DBG_VALUE);
1390b57cec5SDimitry Andric   if (auto *CI = dyn_cast<ConstantInt>(&C)) {
1400b57cec5SDimitry Andric     if (CI->getBitWidth() > 64)
1410b57cec5SDimitry Andric       MIB.addCImm(CI);
1420b57cec5SDimitry Andric     else
1430b57cec5SDimitry Andric       MIB.addImm(CI->getZExtValue());
1440b57cec5SDimitry Andric   } else if (auto *CFP = dyn_cast<ConstantFP>(&C)) {
1450b57cec5SDimitry Andric     MIB.addFPImm(CFP);
1460b57cec5SDimitry Andric   } else {
1470b57cec5SDimitry Andric     // Insert %noreg if we didn't find a usable constant and had to drop it.
1480b57cec5SDimitry Andric     MIB.addReg(0U);
1490b57cec5SDimitry Andric   }
1500b57cec5SDimitry Andric 
151*13138422SDimitry Andric   return MIB.addImm(0).addMetadata(Variable).addMetadata(Expr);
1520b57cec5SDimitry Andric }
1530b57cec5SDimitry Andric 
1540b57cec5SDimitry Andric MachineInstrBuilder MachineIRBuilder::buildDbgLabel(const MDNode *Label) {
1550b57cec5SDimitry Andric   assert(isa<DILabel>(Label) && "not a label");
1560b57cec5SDimitry Andric   assert(cast<DILabel>(Label)->isValidLocationForIntrinsic(State.DL) &&
1570b57cec5SDimitry Andric          "Expected inlined-at fields to agree");
1580b57cec5SDimitry Andric   auto MIB = buildInstr(TargetOpcode::DBG_LABEL);
1590b57cec5SDimitry Andric 
1600b57cec5SDimitry Andric   return MIB.addMetadata(Label);
1610b57cec5SDimitry Andric }
1620b57cec5SDimitry Andric 
1638bcb0991SDimitry Andric MachineInstrBuilder MachineIRBuilder::buildDynStackAlloc(const DstOp &Res,
1648bcb0991SDimitry Andric                                                          const SrcOp &Size,
1658bcb0991SDimitry Andric                                                          unsigned Align) {
1668bcb0991SDimitry Andric   assert(Res.getLLTTy(*getMRI()).isPointer() && "expected ptr dst type");
1678bcb0991SDimitry Andric   auto MIB = buildInstr(TargetOpcode::G_DYN_STACKALLOC);
1688bcb0991SDimitry Andric   Res.addDefToMIB(*getMRI(), MIB);
1698bcb0991SDimitry Andric   Size.addSrcToMIB(MIB);
1708bcb0991SDimitry Andric   MIB.addImm(Align);
1718bcb0991SDimitry Andric   return MIB;
1728bcb0991SDimitry Andric }
1738bcb0991SDimitry Andric 
1740b57cec5SDimitry Andric MachineInstrBuilder MachineIRBuilder::buildFrameIndex(const DstOp &Res,
1750b57cec5SDimitry Andric                                                       int Idx) {
1760b57cec5SDimitry Andric   assert(Res.getLLTTy(*getMRI()).isPointer() && "invalid operand type");
1770b57cec5SDimitry Andric   auto MIB = buildInstr(TargetOpcode::G_FRAME_INDEX);
1780b57cec5SDimitry Andric   Res.addDefToMIB(*getMRI(), MIB);
1790b57cec5SDimitry Andric   MIB.addFrameIndex(Idx);
1800b57cec5SDimitry Andric   return MIB;
1810b57cec5SDimitry Andric }
1820b57cec5SDimitry Andric 
1830b57cec5SDimitry Andric MachineInstrBuilder MachineIRBuilder::buildGlobalValue(const DstOp &Res,
1840b57cec5SDimitry Andric                                                        const GlobalValue *GV) {
1850b57cec5SDimitry Andric   assert(Res.getLLTTy(*getMRI()).isPointer() && "invalid operand type");
1860b57cec5SDimitry Andric   assert(Res.getLLTTy(*getMRI()).getAddressSpace() ==
1870b57cec5SDimitry Andric              GV->getType()->getAddressSpace() &&
1880b57cec5SDimitry Andric          "address space mismatch");
1890b57cec5SDimitry Andric 
1900b57cec5SDimitry Andric   auto MIB = buildInstr(TargetOpcode::G_GLOBAL_VALUE);
1910b57cec5SDimitry Andric   Res.addDefToMIB(*getMRI(), MIB);
1920b57cec5SDimitry Andric   MIB.addGlobalAddress(GV);
1930b57cec5SDimitry Andric   return MIB;
1940b57cec5SDimitry Andric }
1950b57cec5SDimitry Andric 
1960b57cec5SDimitry Andric MachineInstrBuilder MachineIRBuilder::buildJumpTable(const LLT PtrTy,
1970b57cec5SDimitry Andric                                                      unsigned JTI) {
1980b57cec5SDimitry Andric   return buildInstr(TargetOpcode::G_JUMP_TABLE, {PtrTy}, {})
1990b57cec5SDimitry Andric       .addJumpTableIndex(JTI);
2000b57cec5SDimitry Andric }
2010b57cec5SDimitry Andric 
2020b57cec5SDimitry Andric void MachineIRBuilder::validateBinaryOp(const LLT &Res, const LLT &Op0,
2030b57cec5SDimitry Andric                                         const LLT &Op1) {
2040b57cec5SDimitry Andric   assert((Res.isScalar() || Res.isVector()) && "invalid operand type");
2050b57cec5SDimitry Andric   assert((Res == Op0 && Res == Op1) && "type mismatch");
2060b57cec5SDimitry Andric }
2070b57cec5SDimitry Andric 
2080b57cec5SDimitry Andric void MachineIRBuilder::validateShiftOp(const LLT &Res, const LLT &Op0,
2090b57cec5SDimitry Andric                                        const LLT &Op1) {
2100b57cec5SDimitry Andric   assert((Res.isScalar() || Res.isVector()) && "invalid operand type");
2110b57cec5SDimitry Andric   assert((Res == Op0) && "type mismatch");
2120b57cec5SDimitry Andric }
2130b57cec5SDimitry Andric 
214480093f4SDimitry Andric MachineInstrBuilder MachineIRBuilder::buildPtrAdd(const DstOp &Res,
2150b57cec5SDimitry Andric                                                   const SrcOp &Op0,
2160b57cec5SDimitry Andric                                                   const SrcOp &Op1) {
2170b57cec5SDimitry Andric   assert(Res.getLLTTy(*getMRI()).isPointer() &&
2180b57cec5SDimitry Andric          Res.getLLTTy(*getMRI()) == Op0.getLLTTy(*getMRI()) && "type mismatch");
2190b57cec5SDimitry Andric   assert(Op1.getLLTTy(*getMRI()).isScalar() && "invalid offset type");
2200b57cec5SDimitry Andric 
221480093f4SDimitry Andric   return buildInstr(TargetOpcode::G_PTR_ADD, {Res}, {Op0, Op1});
2220b57cec5SDimitry Andric }
2230b57cec5SDimitry Andric 
2240b57cec5SDimitry Andric Optional<MachineInstrBuilder>
225480093f4SDimitry Andric MachineIRBuilder::materializePtrAdd(Register &Res, Register Op0,
2260b57cec5SDimitry Andric                                     const LLT &ValueTy, uint64_t Value) {
2270b57cec5SDimitry Andric   assert(Res == 0 && "Res is a result argument");
2280b57cec5SDimitry Andric   assert(ValueTy.isScalar()  && "invalid offset type");
2290b57cec5SDimitry Andric 
2300b57cec5SDimitry Andric   if (Value == 0) {
2310b57cec5SDimitry Andric     Res = Op0;
2320b57cec5SDimitry Andric     return None;
2330b57cec5SDimitry Andric   }
2340b57cec5SDimitry Andric 
2350b57cec5SDimitry Andric   Res = getMRI()->createGenericVirtualRegister(getMRI()->getType(Op0));
2360b57cec5SDimitry Andric   auto Cst = buildConstant(ValueTy, Value);
237480093f4SDimitry Andric   return buildPtrAdd(Res, Op0, Cst.getReg(0));
2380b57cec5SDimitry Andric }
2390b57cec5SDimitry Andric 
2400b57cec5SDimitry Andric MachineInstrBuilder MachineIRBuilder::buildPtrMask(const DstOp &Res,
2410b57cec5SDimitry Andric                                                    const SrcOp &Op0,
2420b57cec5SDimitry Andric                                                    uint32_t NumBits) {
2430b57cec5SDimitry Andric   assert(Res.getLLTTy(*getMRI()).isPointer() &&
2440b57cec5SDimitry Andric          Res.getLLTTy(*getMRI()) == Op0.getLLTTy(*getMRI()) && "type mismatch");
2450b57cec5SDimitry Andric 
2460b57cec5SDimitry Andric   auto MIB = buildInstr(TargetOpcode::G_PTR_MASK);
2470b57cec5SDimitry Andric   Res.addDefToMIB(*getMRI(), MIB);
2480b57cec5SDimitry Andric   Op0.addSrcToMIB(MIB);
2490b57cec5SDimitry Andric   MIB.addImm(NumBits);
2500b57cec5SDimitry Andric   return MIB;
2510b57cec5SDimitry Andric }
2520b57cec5SDimitry Andric 
2530b57cec5SDimitry Andric MachineInstrBuilder MachineIRBuilder::buildBr(MachineBasicBlock &Dest) {
2540b57cec5SDimitry Andric   return buildInstr(TargetOpcode::G_BR).addMBB(&Dest);
2550b57cec5SDimitry Andric }
2560b57cec5SDimitry Andric 
2570b57cec5SDimitry Andric MachineInstrBuilder MachineIRBuilder::buildBrIndirect(Register Tgt) {
2580b57cec5SDimitry Andric   assert(getMRI()->getType(Tgt).isPointer() && "invalid branch destination");
2590b57cec5SDimitry Andric   return buildInstr(TargetOpcode::G_BRINDIRECT).addUse(Tgt);
2600b57cec5SDimitry Andric }
2610b57cec5SDimitry Andric 
2620b57cec5SDimitry Andric MachineInstrBuilder MachineIRBuilder::buildBrJT(Register TablePtr,
2630b57cec5SDimitry Andric                                                 unsigned JTI,
2640b57cec5SDimitry Andric                                                 Register IndexReg) {
2650b57cec5SDimitry Andric   assert(getMRI()->getType(TablePtr).isPointer() &&
2660b57cec5SDimitry Andric          "Table reg must be a pointer");
2670b57cec5SDimitry Andric   return buildInstr(TargetOpcode::G_BRJT)
2680b57cec5SDimitry Andric       .addUse(TablePtr)
2690b57cec5SDimitry Andric       .addJumpTableIndex(JTI)
2700b57cec5SDimitry Andric       .addUse(IndexReg);
2710b57cec5SDimitry Andric }
2720b57cec5SDimitry Andric 
2730b57cec5SDimitry Andric MachineInstrBuilder MachineIRBuilder::buildCopy(const DstOp &Res,
2740b57cec5SDimitry Andric                                                 const SrcOp &Op) {
2750b57cec5SDimitry Andric   return buildInstr(TargetOpcode::COPY, Res, Op);
2760b57cec5SDimitry Andric }
2770b57cec5SDimitry Andric 
2780b57cec5SDimitry Andric MachineInstrBuilder MachineIRBuilder::buildConstant(const DstOp &Res,
2790b57cec5SDimitry Andric                                                     const ConstantInt &Val) {
2800b57cec5SDimitry Andric   LLT Ty = Res.getLLTTy(*getMRI());
2810b57cec5SDimitry Andric   LLT EltTy = Ty.getScalarType();
2820b57cec5SDimitry Andric   assert(EltTy.getScalarSizeInBits() == Val.getBitWidth() &&
2830b57cec5SDimitry Andric          "creating constant with the wrong size");
2840b57cec5SDimitry Andric 
2850b57cec5SDimitry Andric   if (Ty.isVector()) {
2860b57cec5SDimitry Andric     auto Const = buildInstr(TargetOpcode::G_CONSTANT)
2870b57cec5SDimitry Andric     .addDef(getMRI()->createGenericVirtualRegister(EltTy))
2880b57cec5SDimitry Andric     .addCImm(&Val);
2890b57cec5SDimitry Andric     return buildSplatVector(Res, Const);
2900b57cec5SDimitry Andric   }
2910b57cec5SDimitry Andric 
2920b57cec5SDimitry Andric   auto Const = buildInstr(TargetOpcode::G_CONSTANT);
2930b57cec5SDimitry Andric   Res.addDefToMIB(*getMRI(), Const);
2940b57cec5SDimitry Andric   Const.addCImm(&Val);
2950b57cec5SDimitry Andric   return Const;
2960b57cec5SDimitry Andric }
2970b57cec5SDimitry Andric 
2980b57cec5SDimitry Andric MachineInstrBuilder MachineIRBuilder::buildConstant(const DstOp &Res,
2990b57cec5SDimitry Andric                                                     int64_t Val) {
3000b57cec5SDimitry Andric   auto IntN = IntegerType::get(getMF().getFunction().getContext(),
3010b57cec5SDimitry Andric                                Res.getLLTTy(*getMRI()).getScalarSizeInBits());
3020b57cec5SDimitry Andric   ConstantInt *CI = ConstantInt::get(IntN, Val, true);
3030b57cec5SDimitry Andric   return buildConstant(Res, *CI);
3040b57cec5SDimitry Andric }
3050b57cec5SDimitry Andric 
3060b57cec5SDimitry Andric MachineInstrBuilder MachineIRBuilder::buildFConstant(const DstOp &Res,
3070b57cec5SDimitry Andric                                                      const ConstantFP &Val) {
3080b57cec5SDimitry Andric   LLT Ty = Res.getLLTTy(*getMRI());
3090b57cec5SDimitry Andric   LLT EltTy = Ty.getScalarType();
3100b57cec5SDimitry Andric 
3110b57cec5SDimitry Andric   assert(APFloat::getSizeInBits(Val.getValueAPF().getSemantics())
3120b57cec5SDimitry Andric          == EltTy.getSizeInBits() &&
3130b57cec5SDimitry Andric          "creating fconstant with the wrong size");
3140b57cec5SDimitry Andric 
3150b57cec5SDimitry Andric   assert(!Ty.isPointer() && "invalid operand type");
3160b57cec5SDimitry Andric 
3170b57cec5SDimitry Andric   if (Ty.isVector()) {
3180b57cec5SDimitry Andric     auto Const = buildInstr(TargetOpcode::G_FCONSTANT)
3190b57cec5SDimitry Andric     .addDef(getMRI()->createGenericVirtualRegister(EltTy))
3200b57cec5SDimitry Andric     .addFPImm(&Val);
3210b57cec5SDimitry Andric 
3220b57cec5SDimitry Andric     return buildSplatVector(Res, Const);
3230b57cec5SDimitry Andric   }
3240b57cec5SDimitry Andric 
3250b57cec5SDimitry Andric   auto Const = buildInstr(TargetOpcode::G_FCONSTANT);
3260b57cec5SDimitry Andric   Res.addDefToMIB(*getMRI(), Const);
3270b57cec5SDimitry Andric   Const.addFPImm(&Val);
3280b57cec5SDimitry Andric   return Const;
3290b57cec5SDimitry Andric }
3300b57cec5SDimitry Andric 
3310b57cec5SDimitry Andric MachineInstrBuilder MachineIRBuilder::buildConstant(const DstOp &Res,
3320b57cec5SDimitry Andric                                                     const APInt &Val) {
3330b57cec5SDimitry Andric   ConstantInt *CI = ConstantInt::get(getMF().getFunction().getContext(), Val);
3340b57cec5SDimitry Andric   return buildConstant(Res, *CI);
3350b57cec5SDimitry Andric }
3360b57cec5SDimitry Andric 
3370b57cec5SDimitry Andric MachineInstrBuilder MachineIRBuilder::buildFConstant(const DstOp &Res,
3380b57cec5SDimitry Andric                                                      double Val) {
3390b57cec5SDimitry Andric   LLT DstTy = Res.getLLTTy(*getMRI());
3400b57cec5SDimitry Andric   auto &Ctx = getMF().getFunction().getContext();
3410b57cec5SDimitry Andric   auto *CFP =
3420b57cec5SDimitry Andric       ConstantFP::get(Ctx, getAPFloatFromSize(Val, DstTy.getScalarSizeInBits()));
3430b57cec5SDimitry Andric   return buildFConstant(Res, *CFP);
3440b57cec5SDimitry Andric }
3450b57cec5SDimitry Andric 
3460b57cec5SDimitry Andric MachineInstrBuilder MachineIRBuilder::buildFConstant(const DstOp &Res,
3470b57cec5SDimitry Andric                                                      const APFloat &Val) {
3480b57cec5SDimitry Andric   auto &Ctx = getMF().getFunction().getContext();
3490b57cec5SDimitry Andric   auto *CFP = ConstantFP::get(Ctx, Val);
3500b57cec5SDimitry Andric   return buildFConstant(Res, *CFP);
3510b57cec5SDimitry Andric }
3520b57cec5SDimitry Andric 
3530b57cec5SDimitry Andric MachineInstrBuilder MachineIRBuilder::buildBrCond(Register Tst,
3540b57cec5SDimitry Andric                                                   MachineBasicBlock &Dest) {
3550b57cec5SDimitry Andric   assert(getMRI()->getType(Tst).isScalar() && "invalid operand type");
3560b57cec5SDimitry Andric 
3570b57cec5SDimitry Andric   return buildInstr(TargetOpcode::G_BRCOND).addUse(Tst).addMBB(&Dest);
3580b57cec5SDimitry Andric }
3590b57cec5SDimitry Andric 
3600b57cec5SDimitry Andric MachineInstrBuilder MachineIRBuilder::buildLoad(const DstOp &Res,
3610b57cec5SDimitry Andric                                                 const SrcOp &Addr,
3620b57cec5SDimitry Andric                                                 MachineMemOperand &MMO) {
3630b57cec5SDimitry Andric   return buildLoadInstr(TargetOpcode::G_LOAD, Res, Addr, MMO);
3640b57cec5SDimitry Andric }
3650b57cec5SDimitry Andric 
3660b57cec5SDimitry Andric MachineInstrBuilder MachineIRBuilder::buildLoadInstr(unsigned Opcode,
3670b57cec5SDimitry Andric                                                      const DstOp &Res,
3680b57cec5SDimitry Andric                                                      const SrcOp &Addr,
3690b57cec5SDimitry Andric                                                      MachineMemOperand &MMO) {
3700b57cec5SDimitry Andric   assert(Res.getLLTTy(*getMRI()).isValid() && "invalid operand type");
3710b57cec5SDimitry Andric   assert(Addr.getLLTTy(*getMRI()).isPointer() && "invalid operand type");
3720b57cec5SDimitry Andric 
3730b57cec5SDimitry Andric   auto MIB = buildInstr(Opcode);
3740b57cec5SDimitry Andric   Res.addDefToMIB(*getMRI(), MIB);
3750b57cec5SDimitry Andric   Addr.addSrcToMIB(MIB);
3760b57cec5SDimitry Andric   MIB.addMemOperand(&MMO);
3770b57cec5SDimitry Andric   return MIB;
3780b57cec5SDimitry Andric }
3790b57cec5SDimitry Andric 
3800b57cec5SDimitry Andric MachineInstrBuilder MachineIRBuilder::buildStore(const SrcOp &Val,
3810b57cec5SDimitry Andric                                                  const SrcOp &Addr,
3820b57cec5SDimitry Andric                                                  MachineMemOperand &MMO) {
3830b57cec5SDimitry Andric   assert(Val.getLLTTy(*getMRI()).isValid() && "invalid operand type");
3840b57cec5SDimitry Andric   assert(Addr.getLLTTy(*getMRI()).isPointer() && "invalid operand type");
3850b57cec5SDimitry Andric 
3860b57cec5SDimitry Andric   auto MIB = buildInstr(TargetOpcode::G_STORE);
3870b57cec5SDimitry Andric   Val.addSrcToMIB(MIB);
3880b57cec5SDimitry Andric   Addr.addSrcToMIB(MIB);
3890b57cec5SDimitry Andric   MIB.addMemOperand(&MMO);
3900b57cec5SDimitry Andric   return MIB;
3910b57cec5SDimitry Andric }
3920b57cec5SDimitry Andric 
3930b57cec5SDimitry Andric MachineInstrBuilder MachineIRBuilder::buildUAddo(const DstOp &Res,
3940b57cec5SDimitry Andric                                                  const DstOp &CarryOut,
3950b57cec5SDimitry Andric                                                  const SrcOp &Op0,
3960b57cec5SDimitry Andric                                                  const SrcOp &Op1) {
3970b57cec5SDimitry Andric   return buildInstr(TargetOpcode::G_UADDO, {Res, CarryOut}, {Op0, Op1});
3980b57cec5SDimitry Andric }
3990b57cec5SDimitry Andric 
4000b57cec5SDimitry Andric MachineInstrBuilder MachineIRBuilder::buildUAdde(const DstOp &Res,
4010b57cec5SDimitry Andric                                                  const DstOp &CarryOut,
4020b57cec5SDimitry Andric                                                  const SrcOp &Op0,
4030b57cec5SDimitry Andric                                                  const SrcOp &Op1,
4040b57cec5SDimitry Andric                                                  const SrcOp &CarryIn) {
4050b57cec5SDimitry Andric   return buildInstr(TargetOpcode::G_UADDE, {Res, CarryOut},
4060b57cec5SDimitry Andric                     {Op0, Op1, CarryIn});
4070b57cec5SDimitry Andric }
4080b57cec5SDimitry Andric 
4090b57cec5SDimitry Andric MachineInstrBuilder MachineIRBuilder::buildAnyExt(const DstOp &Res,
4100b57cec5SDimitry Andric                                                   const SrcOp &Op) {
4110b57cec5SDimitry Andric   return buildInstr(TargetOpcode::G_ANYEXT, Res, Op);
4120b57cec5SDimitry Andric }
4130b57cec5SDimitry Andric 
4140b57cec5SDimitry Andric MachineInstrBuilder MachineIRBuilder::buildSExt(const DstOp &Res,
4150b57cec5SDimitry Andric                                                 const SrcOp &Op) {
4160b57cec5SDimitry Andric   return buildInstr(TargetOpcode::G_SEXT, Res, Op);
4170b57cec5SDimitry Andric }
4180b57cec5SDimitry Andric 
4190b57cec5SDimitry Andric MachineInstrBuilder MachineIRBuilder::buildZExt(const DstOp &Res,
4200b57cec5SDimitry Andric                                                 const SrcOp &Op) {
4210b57cec5SDimitry Andric   return buildInstr(TargetOpcode::G_ZEXT, Res, Op);
4220b57cec5SDimitry Andric }
4230b57cec5SDimitry Andric 
4240b57cec5SDimitry Andric unsigned MachineIRBuilder::getBoolExtOp(bool IsVec, bool IsFP) const {
4250b57cec5SDimitry Andric   const auto *TLI = getMF().getSubtarget().getTargetLowering();
4260b57cec5SDimitry Andric   switch (TLI->getBooleanContents(IsVec, IsFP)) {
4270b57cec5SDimitry Andric   case TargetLoweringBase::ZeroOrNegativeOneBooleanContent:
4280b57cec5SDimitry Andric     return TargetOpcode::G_SEXT;
4290b57cec5SDimitry Andric   case TargetLoweringBase::ZeroOrOneBooleanContent:
4300b57cec5SDimitry Andric     return TargetOpcode::G_ZEXT;
4310b57cec5SDimitry Andric   default:
4320b57cec5SDimitry Andric     return TargetOpcode::G_ANYEXT;
4330b57cec5SDimitry Andric   }
4340b57cec5SDimitry Andric }
4350b57cec5SDimitry Andric 
4360b57cec5SDimitry Andric MachineInstrBuilder MachineIRBuilder::buildBoolExt(const DstOp &Res,
4370b57cec5SDimitry Andric                                                    const SrcOp &Op,
4380b57cec5SDimitry Andric                                                    bool IsFP) {
4390b57cec5SDimitry Andric   unsigned ExtOp = getBoolExtOp(getMRI()->getType(Op.getReg()).isVector(), IsFP);
4400b57cec5SDimitry Andric   return buildInstr(ExtOp, Res, Op);
4410b57cec5SDimitry Andric }
4420b57cec5SDimitry Andric 
4430b57cec5SDimitry Andric MachineInstrBuilder MachineIRBuilder::buildExtOrTrunc(unsigned ExtOpc,
4440b57cec5SDimitry Andric                                                       const DstOp &Res,
4450b57cec5SDimitry Andric                                                       const SrcOp &Op) {
4460b57cec5SDimitry Andric   assert((TargetOpcode::G_ANYEXT == ExtOpc || TargetOpcode::G_ZEXT == ExtOpc ||
4470b57cec5SDimitry Andric           TargetOpcode::G_SEXT == ExtOpc) &&
4480b57cec5SDimitry Andric          "Expecting Extending Opc");
4490b57cec5SDimitry Andric   assert(Res.getLLTTy(*getMRI()).isScalar() ||
4500b57cec5SDimitry Andric          Res.getLLTTy(*getMRI()).isVector());
4510b57cec5SDimitry Andric   assert(Res.getLLTTy(*getMRI()).isScalar() ==
4520b57cec5SDimitry Andric          Op.getLLTTy(*getMRI()).isScalar());
4530b57cec5SDimitry Andric 
4540b57cec5SDimitry Andric   unsigned Opcode = TargetOpcode::COPY;
4550b57cec5SDimitry Andric   if (Res.getLLTTy(*getMRI()).getSizeInBits() >
4560b57cec5SDimitry Andric       Op.getLLTTy(*getMRI()).getSizeInBits())
4570b57cec5SDimitry Andric     Opcode = ExtOpc;
4580b57cec5SDimitry Andric   else if (Res.getLLTTy(*getMRI()).getSizeInBits() <
4590b57cec5SDimitry Andric            Op.getLLTTy(*getMRI()).getSizeInBits())
4600b57cec5SDimitry Andric     Opcode = TargetOpcode::G_TRUNC;
4610b57cec5SDimitry Andric   else
4620b57cec5SDimitry Andric     assert(Res.getLLTTy(*getMRI()) == Op.getLLTTy(*getMRI()));
4630b57cec5SDimitry Andric 
4640b57cec5SDimitry Andric   return buildInstr(Opcode, Res, Op);
4650b57cec5SDimitry Andric }
4660b57cec5SDimitry Andric 
4670b57cec5SDimitry Andric MachineInstrBuilder MachineIRBuilder::buildSExtOrTrunc(const DstOp &Res,
4680b57cec5SDimitry Andric                                                        const SrcOp &Op) {
4690b57cec5SDimitry Andric   return buildExtOrTrunc(TargetOpcode::G_SEXT, Res, Op);
4700b57cec5SDimitry Andric }
4710b57cec5SDimitry Andric 
4720b57cec5SDimitry Andric MachineInstrBuilder MachineIRBuilder::buildZExtOrTrunc(const DstOp &Res,
4730b57cec5SDimitry Andric                                                        const SrcOp &Op) {
4740b57cec5SDimitry Andric   return buildExtOrTrunc(TargetOpcode::G_ZEXT, Res, Op);
4750b57cec5SDimitry Andric }
4760b57cec5SDimitry Andric 
4770b57cec5SDimitry Andric MachineInstrBuilder MachineIRBuilder::buildAnyExtOrTrunc(const DstOp &Res,
4780b57cec5SDimitry Andric                                                          const SrcOp &Op) {
4790b57cec5SDimitry Andric   return buildExtOrTrunc(TargetOpcode::G_ANYEXT, Res, Op);
4800b57cec5SDimitry Andric }
4810b57cec5SDimitry Andric 
4820b57cec5SDimitry Andric MachineInstrBuilder MachineIRBuilder::buildCast(const DstOp &Dst,
4830b57cec5SDimitry Andric                                                 const SrcOp &Src) {
4840b57cec5SDimitry Andric   LLT SrcTy = Src.getLLTTy(*getMRI());
4850b57cec5SDimitry Andric   LLT DstTy = Dst.getLLTTy(*getMRI());
4860b57cec5SDimitry Andric   if (SrcTy == DstTy)
4870b57cec5SDimitry Andric     return buildCopy(Dst, Src);
4880b57cec5SDimitry Andric 
4890b57cec5SDimitry Andric   unsigned Opcode;
4900b57cec5SDimitry Andric   if (SrcTy.isPointer() && DstTy.isScalar())
4910b57cec5SDimitry Andric     Opcode = TargetOpcode::G_PTRTOINT;
4920b57cec5SDimitry Andric   else if (DstTy.isPointer() && SrcTy.isScalar())
4930b57cec5SDimitry Andric     Opcode = TargetOpcode::G_INTTOPTR;
4940b57cec5SDimitry Andric   else {
4950b57cec5SDimitry Andric     assert(!SrcTy.isPointer() && !DstTy.isPointer() && "n G_ADDRCAST yet");
4960b57cec5SDimitry Andric     Opcode = TargetOpcode::G_BITCAST;
4970b57cec5SDimitry Andric   }
4980b57cec5SDimitry Andric 
4990b57cec5SDimitry Andric   return buildInstr(Opcode, Dst, Src);
5000b57cec5SDimitry Andric }
5010b57cec5SDimitry Andric 
5020b57cec5SDimitry Andric MachineInstrBuilder MachineIRBuilder::buildExtract(const DstOp &Dst,
5030b57cec5SDimitry Andric                                                    const SrcOp &Src,
5040b57cec5SDimitry Andric                                                    uint64_t Index) {
5050b57cec5SDimitry Andric   LLT SrcTy = Src.getLLTTy(*getMRI());
5060b57cec5SDimitry Andric   LLT DstTy = Dst.getLLTTy(*getMRI());
5070b57cec5SDimitry Andric 
5080b57cec5SDimitry Andric #ifndef NDEBUG
5090b57cec5SDimitry Andric   assert(SrcTy.isValid() && "invalid operand type");
5100b57cec5SDimitry Andric   assert(DstTy.isValid() && "invalid operand type");
5110b57cec5SDimitry Andric   assert(Index + DstTy.getSizeInBits() <= SrcTy.getSizeInBits() &&
5120b57cec5SDimitry Andric          "extracting off end of register");
5130b57cec5SDimitry Andric #endif
5140b57cec5SDimitry Andric 
5150b57cec5SDimitry Andric   if (DstTy.getSizeInBits() == SrcTy.getSizeInBits()) {
5160b57cec5SDimitry Andric     assert(Index == 0 && "insertion past the end of a register");
5170b57cec5SDimitry Andric     return buildCast(Dst, Src);
5180b57cec5SDimitry Andric   }
5190b57cec5SDimitry Andric 
5200b57cec5SDimitry Andric   auto Extract = buildInstr(TargetOpcode::G_EXTRACT);
5210b57cec5SDimitry Andric   Dst.addDefToMIB(*getMRI(), Extract);
5220b57cec5SDimitry Andric   Src.addSrcToMIB(Extract);
5230b57cec5SDimitry Andric   Extract.addImm(Index);
5240b57cec5SDimitry Andric   return Extract;
5250b57cec5SDimitry Andric }
5260b57cec5SDimitry Andric 
5270b57cec5SDimitry Andric void MachineIRBuilder::buildSequence(Register Res, ArrayRef<Register> Ops,
5280b57cec5SDimitry Andric                                      ArrayRef<uint64_t> Indices) {
5290b57cec5SDimitry Andric #ifndef NDEBUG
5300b57cec5SDimitry Andric   assert(Ops.size() == Indices.size() && "incompatible args");
5310b57cec5SDimitry Andric   assert(!Ops.empty() && "invalid trivial sequence");
5320b57cec5SDimitry Andric   assert(std::is_sorted(Indices.begin(), Indices.end()) &&
5330b57cec5SDimitry Andric          "sequence offsets must be in ascending order");
5340b57cec5SDimitry Andric 
5350b57cec5SDimitry Andric   assert(getMRI()->getType(Res).isValid() && "invalid operand type");
5360b57cec5SDimitry Andric   for (auto Op : Ops)
5370b57cec5SDimitry Andric     assert(getMRI()->getType(Op).isValid() && "invalid operand type");
5380b57cec5SDimitry Andric #endif
5390b57cec5SDimitry Andric 
5400b57cec5SDimitry Andric   LLT ResTy = getMRI()->getType(Res);
5410b57cec5SDimitry Andric   LLT OpTy = getMRI()->getType(Ops[0]);
5420b57cec5SDimitry Andric   unsigned OpSize = OpTy.getSizeInBits();
5430b57cec5SDimitry Andric   bool MaybeMerge = true;
5440b57cec5SDimitry Andric   for (unsigned i = 0; i < Ops.size(); ++i) {
5450b57cec5SDimitry Andric     if (getMRI()->getType(Ops[i]) != OpTy || Indices[i] != i * OpSize) {
5460b57cec5SDimitry Andric       MaybeMerge = false;
5470b57cec5SDimitry Andric       break;
5480b57cec5SDimitry Andric     }
5490b57cec5SDimitry Andric   }
5500b57cec5SDimitry Andric 
5510b57cec5SDimitry Andric   if (MaybeMerge && Ops.size() * OpSize == ResTy.getSizeInBits()) {
5520b57cec5SDimitry Andric     buildMerge(Res, Ops);
5530b57cec5SDimitry Andric     return;
5540b57cec5SDimitry Andric   }
5550b57cec5SDimitry Andric 
5560b57cec5SDimitry Andric   Register ResIn = getMRI()->createGenericVirtualRegister(ResTy);
5570b57cec5SDimitry Andric   buildUndef(ResIn);
5580b57cec5SDimitry Andric 
5590b57cec5SDimitry Andric   for (unsigned i = 0; i < Ops.size(); ++i) {
5600b57cec5SDimitry Andric     Register ResOut = i + 1 == Ops.size()
5610b57cec5SDimitry Andric                           ? Res
5620b57cec5SDimitry Andric                           : getMRI()->createGenericVirtualRegister(ResTy);
5630b57cec5SDimitry Andric     buildInsert(ResOut, ResIn, Ops[i], Indices[i]);
5640b57cec5SDimitry Andric     ResIn = ResOut;
5650b57cec5SDimitry Andric   }
5660b57cec5SDimitry Andric }
5670b57cec5SDimitry Andric 
5680b57cec5SDimitry Andric MachineInstrBuilder MachineIRBuilder::buildUndef(const DstOp &Res) {
5690b57cec5SDimitry Andric   return buildInstr(TargetOpcode::G_IMPLICIT_DEF, {Res}, {});
5700b57cec5SDimitry Andric }
5710b57cec5SDimitry Andric 
5720b57cec5SDimitry Andric MachineInstrBuilder MachineIRBuilder::buildMerge(const DstOp &Res,
5730b57cec5SDimitry Andric                                                  ArrayRef<Register> Ops) {
5740b57cec5SDimitry Andric   // Unfortunately to convert from ArrayRef<LLT> to ArrayRef<SrcOp>,
5750b57cec5SDimitry Andric   // we need some temporary storage for the DstOp objects. Here we use a
5760b57cec5SDimitry Andric   // sufficiently large SmallVector to not go through the heap.
5770b57cec5SDimitry Andric   SmallVector<SrcOp, 8> TmpVec(Ops.begin(), Ops.end());
5780b57cec5SDimitry Andric   assert(TmpVec.size() > 1);
5790b57cec5SDimitry Andric   return buildInstr(TargetOpcode::G_MERGE_VALUES, Res, TmpVec);
5800b57cec5SDimitry Andric }
5810b57cec5SDimitry Andric 
5820b57cec5SDimitry Andric MachineInstrBuilder MachineIRBuilder::buildUnmerge(ArrayRef<LLT> Res,
5830b57cec5SDimitry Andric                                                    const SrcOp &Op) {
5840b57cec5SDimitry Andric   // Unfortunately to convert from ArrayRef<LLT> to ArrayRef<DstOp>,
5850b57cec5SDimitry Andric   // we need some temporary storage for the DstOp objects. Here we use a
5860b57cec5SDimitry Andric   // sufficiently large SmallVector to not go through the heap.
5870b57cec5SDimitry Andric   SmallVector<DstOp, 8> TmpVec(Res.begin(), Res.end());
5880b57cec5SDimitry Andric   assert(TmpVec.size() > 1);
5890b57cec5SDimitry Andric   return buildInstr(TargetOpcode::G_UNMERGE_VALUES, TmpVec, Op);
5900b57cec5SDimitry Andric }
5910b57cec5SDimitry Andric 
5920b57cec5SDimitry Andric MachineInstrBuilder MachineIRBuilder::buildUnmerge(LLT Res,
5930b57cec5SDimitry Andric                                                    const SrcOp &Op) {
5940b57cec5SDimitry Andric   unsigned NumReg = Op.getLLTTy(*getMRI()).getSizeInBits() / Res.getSizeInBits();
5950b57cec5SDimitry Andric   SmallVector<Register, 8> TmpVec;
5960b57cec5SDimitry Andric   for (unsigned I = 0; I != NumReg; ++I)
5970b57cec5SDimitry Andric     TmpVec.push_back(getMRI()->createGenericVirtualRegister(Res));
5980b57cec5SDimitry Andric   return buildUnmerge(TmpVec, Op);
5990b57cec5SDimitry Andric }
6000b57cec5SDimitry Andric 
6010b57cec5SDimitry Andric MachineInstrBuilder MachineIRBuilder::buildUnmerge(ArrayRef<Register> Res,
6020b57cec5SDimitry Andric                                                    const SrcOp &Op) {
6030b57cec5SDimitry Andric   // Unfortunately to convert from ArrayRef<Register> to ArrayRef<DstOp>,
6040b57cec5SDimitry Andric   // we need some temporary storage for the DstOp objects. Here we use a
6050b57cec5SDimitry Andric   // sufficiently large SmallVector to not go through the heap.
6060b57cec5SDimitry Andric   SmallVector<DstOp, 8> TmpVec(Res.begin(), Res.end());
6070b57cec5SDimitry Andric   assert(TmpVec.size() > 1);
6080b57cec5SDimitry Andric   return buildInstr(TargetOpcode::G_UNMERGE_VALUES, TmpVec, Op);
6090b57cec5SDimitry Andric }
6100b57cec5SDimitry Andric 
6110b57cec5SDimitry Andric MachineInstrBuilder MachineIRBuilder::buildBuildVector(const DstOp &Res,
6120b57cec5SDimitry Andric                                                        ArrayRef<Register> Ops) {
6130b57cec5SDimitry Andric   // Unfortunately to convert from ArrayRef<Register> to ArrayRef<SrcOp>,
6140b57cec5SDimitry Andric   // we need some temporary storage for the DstOp objects. Here we use a
6150b57cec5SDimitry Andric   // sufficiently large SmallVector to not go through the heap.
6160b57cec5SDimitry Andric   SmallVector<SrcOp, 8> TmpVec(Ops.begin(), Ops.end());
6170b57cec5SDimitry Andric   return buildInstr(TargetOpcode::G_BUILD_VECTOR, Res, TmpVec);
6180b57cec5SDimitry Andric }
6190b57cec5SDimitry Andric 
6200b57cec5SDimitry Andric MachineInstrBuilder MachineIRBuilder::buildSplatVector(const DstOp &Res,
6210b57cec5SDimitry Andric                                                        const SrcOp &Src) {
6220b57cec5SDimitry Andric   SmallVector<SrcOp, 8> TmpVec(Res.getLLTTy(*getMRI()).getNumElements(), Src);
6230b57cec5SDimitry Andric   return buildInstr(TargetOpcode::G_BUILD_VECTOR, Res, TmpVec);
6240b57cec5SDimitry Andric }
6250b57cec5SDimitry Andric 
6260b57cec5SDimitry Andric MachineInstrBuilder
6270b57cec5SDimitry Andric MachineIRBuilder::buildBuildVectorTrunc(const DstOp &Res,
6280b57cec5SDimitry Andric                                         ArrayRef<Register> Ops) {
6290b57cec5SDimitry Andric   // Unfortunately to convert from ArrayRef<Register> to ArrayRef<SrcOp>,
6300b57cec5SDimitry Andric   // we need some temporary storage for the DstOp objects. Here we use a
6310b57cec5SDimitry Andric   // sufficiently large SmallVector to not go through the heap.
6320b57cec5SDimitry Andric   SmallVector<SrcOp, 8> TmpVec(Ops.begin(), Ops.end());
6330b57cec5SDimitry Andric   return buildInstr(TargetOpcode::G_BUILD_VECTOR_TRUNC, Res, TmpVec);
6340b57cec5SDimitry Andric }
6350b57cec5SDimitry Andric 
6360b57cec5SDimitry Andric MachineInstrBuilder
6370b57cec5SDimitry Andric MachineIRBuilder::buildConcatVectors(const DstOp &Res, ArrayRef<Register> Ops) {
6380b57cec5SDimitry Andric   // Unfortunately to convert from ArrayRef<Register> to ArrayRef<SrcOp>,
6390b57cec5SDimitry Andric   // we need some temporary storage for the DstOp objects. Here we use a
6400b57cec5SDimitry Andric   // sufficiently large SmallVector to not go through the heap.
6410b57cec5SDimitry Andric   SmallVector<SrcOp, 8> TmpVec(Ops.begin(), Ops.end());
6420b57cec5SDimitry Andric   return buildInstr(TargetOpcode::G_CONCAT_VECTORS, Res, TmpVec);
6430b57cec5SDimitry Andric }
6440b57cec5SDimitry Andric 
6450b57cec5SDimitry Andric MachineInstrBuilder MachineIRBuilder::buildInsert(Register Res, Register Src,
6460b57cec5SDimitry Andric                                                   Register Op, unsigned Index) {
6470b57cec5SDimitry Andric   assert(Index + getMRI()->getType(Op).getSizeInBits() <=
6480b57cec5SDimitry Andric              getMRI()->getType(Res).getSizeInBits() &&
6490b57cec5SDimitry Andric          "insertion past the end of a register");
6500b57cec5SDimitry Andric 
6510b57cec5SDimitry Andric   if (getMRI()->getType(Res).getSizeInBits() ==
6520b57cec5SDimitry Andric       getMRI()->getType(Op).getSizeInBits()) {
6530b57cec5SDimitry Andric     return buildCast(Res, Op);
6540b57cec5SDimitry Andric   }
6550b57cec5SDimitry Andric 
6560b57cec5SDimitry Andric   return buildInstr(TargetOpcode::G_INSERT)
6570b57cec5SDimitry Andric       .addDef(Res)
6580b57cec5SDimitry Andric       .addUse(Src)
6590b57cec5SDimitry Andric       .addUse(Op)
6600b57cec5SDimitry Andric       .addImm(Index);
6610b57cec5SDimitry Andric }
6620b57cec5SDimitry Andric 
6630b57cec5SDimitry Andric MachineInstrBuilder MachineIRBuilder::buildIntrinsic(Intrinsic::ID ID,
6640b57cec5SDimitry Andric                                                      ArrayRef<Register> ResultRegs,
6650b57cec5SDimitry Andric                                                      bool HasSideEffects) {
6660b57cec5SDimitry Andric   auto MIB =
6670b57cec5SDimitry Andric       buildInstr(HasSideEffects ? TargetOpcode::G_INTRINSIC_W_SIDE_EFFECTS
6680b57cec5SDimitry Andric                                 : TargetOpcode::G_INTRINSIC);
6690b57cec5SDimitry Andric   for (unsigned ResultReg : ResultRegs)
6700b57cec5SDimitry Andric     MIB.addDef(ResultReg);
6710b57cec5SDimitry Andric   MIB.addIntrinsicID(ID);
6720b57cec5SDimitry Andric   return MIB;
6730b57cec5SDimitry Andric }
6740b57cec5SDimitry Andric 
6750b57cec5SDimitry Andric MachineInstrBuilder MachineIRBuilder::buildIntrinsic(Intrinsic::ID ID,
6760b57cec5SDimitry Andric                                                      ArrayRef<DstOp> Results,
6770b57cec5SDimitry Andric                                                      bool HasSideEffects) {
6780b57cec5SDimitry Andric   auto MIB =
6790b57cec5SDimitry Andric       buildInstr(HasSideEffects ? TargetOpcode::G_INTRINSIC_W_SIDE_EFFECTS
6800b57cec5SDimitry Andric                                 : TargetOpcode::G_INTRINSIC);
6810b57cec5SDimitry Andric   for (DstOp Result : Results)
6820b57cec5SDimitry Andric     Result.addDefToMIB(*getMRI(), MIB);
6830b57cec5SDimitry Andric   MIB.addIntrinsicID(ID);
6840b57cec5SDimitry Andric   return MIB;
6850b57cec5SDimitry Andric }
6860b57cec5SDimitry Andric 
6870b57cec5SDimitry Andric MachineInstrBuilder MachineIRBuilder::buildTrunc(const DstOp &Res,
6880b57cec5SDimitry Andric                                                  const SrcOp &Op) {
6890b57cec5SDimitry Andric   return buildInstr(TargetOpcode::G_TRUNC, Res, Op);
6900b57cec5SDimitry Andric }
6910b57cec5SDimitry Andric 
6920b57cec5SDimitry Andric MachineInstrBuilder MachineIRBuilder::buildFPTrunc(const DstOp &Res,
693480093f4SDimitry Andric                                                    const SrcOp &Op,
694480093f4SDimitry Andric                                                    Optional<unsigned> Flags) {
695480093f4SDimitry Andric   return buildInstr(TargetOpcode::G_FPTRUNC, Res, Op, Flags);
6960b57cec5SDimitry Andric }
6970b57cec5SDimitry Andric 
6980b57cec5SDimitry Andric MachineInstrBuilder MachineIRBuilder::buildICmp(CmpInst::Predicate Pred,
6990b57cec5SDimitry Andric                                                 const DstOp &Res,
7000b57cec5SDimitry Andric                                                 const SrcOp &Op0,
7010b57cec5SDimitry Andric                                                 const SrcOp &Op1) {
7020b57cec5SDimitry Andric   return buildInstr(TargetOpcode::G_ICMP, Res, {Pred, Op0, Op1});
7030b57cec5SDimitry Andric }
7040b57cec5SDimitry Andric 
7050b57cec5SDimitry Andric MachineInstrBuilder MachineIRBuilder::buildFCmp(CmpInst::Predicate Pred,
7060b57cec5SDimitry Andric                                                 const DstOp &Res,
7070b57cec5SDimitry Andric                                                 const SrcOp &Op0,
7088bcb0991SDimitry Andric                                                 const SrcOp &Op1,
7098bcb0991SDimitry Andric                                                 Optional<unsigned> Flags) {
7100b57cec5SDimitry Andric 
7118bcb0991SDimitry Andric   return buildInstr(TargetOpcode::G_FCMP, Res, {Pred, Op0, Op1}, Flags);
7120b57cec5SDimitry Andric }
7130b57cec5SDimitry Andric 
7140b57cec5SDimitry Andric MachineInstrBuilder MachineIRBuilder::buildSelect(const DstOp &Res,
7150b57cec5SDimitry Andric                                                   const SrcOp &Tst,
7160b57cec5SDimitry Andric                                                   const SrcOp &Op0,
7178bcb0991SDimitry Andric                                                   const SrcOp &Op1,
7188bcb0991SDimitry Andric                                                   Optional<unsigned> Flags) {
7190b57cec5SDimitry Andric 
7208bcb0991SDimitry Andric   return buildInstr(TargetOpcode::G_SELECT, {Res}, {Tst, Op0, Op1}, Flags);
7210b57cec5SDimitry Andric }
7220b57cec5SDimitry Andric 
7230b57cec5SDimitry Andric MachineInstrBuilder
7240b57cec5SDimitry Andric MachineIRBuilder::buildInsertVectorElement(const DstOp &Res, const SrcOp &Val,
7250b57cec5SDimitry Andric                                            const SrcOp &Elt, const SrcOp &Idx) {
7260b57cec5SDimitry Andric   return buildInstr(TargetOpcode::G_INSERT_VECTOR_ELT, Res, {Val, Elt, Idx});
7270b57cec5SDimitry Andric }
7280b57cec5SDimitry Andric 
7290b57cec5SDimitry Andric MachineInstrBuilder
7300b57cec5SDimitry Andric MachineIRBuilder::buildExtractVectorElement(const DstOp &Res, const SrcOp &Val,
7310b57cec5SDimitry Andric                                             const SrcOp &Idx) {
7320b57cec5SDimitry Andric   return buildInstr(TargetOpcode::G_EXTRACT_VECTOR_ELT, Res, {Val, Idx});
7330b57cec5SDimitry Andric }
7340b57cec5SDimitry Andric 
7350b57cec5SDimitry Andric MachineInstrBuilder MachineIRBuilder::buildAtomicCmpXchgWithSuccess(
7360b57cec5SDimitry Andric     Register OldValRes, Register SuccessRes, Register Addr, Register CmpVal,
7370b57cec5SDimitry Andric     Register NewVal, MachineMemOperand &MMO) {
7380b57cec5SDimitry Andric #ifndef NDEBUG
7390b57cec5SDimitry Andric   LLT OldValResTy = getMRI()->getType(OldValRes);
7400b57cec5SDimitry Andric   LLT SuccessResTy = getMRI()->getType(SuccessRes);
7410b57cec5SDimitry Andric   LLT AddrTy = getMRI()->getType(Addr);
7420b57cec5SDimitry Andric   LLT CmpValTy = getMRI()->getType(CmpVal);
7430b57cec5SDimitry Andric   LLT NewValTy = getMRI()->getType(NewVal);
7440b57cec5SDimitry Andric   assert(OldValResTy.isScalar() && "invalid operand type");
7450b57cec5SDimitry Andric   assert(SuccessResTy.isScalar() && "invalid operand type");
7460b57cec5SDimitry Andric   assert(AddrTy.isPointer() && "invalid operand type");
7470b57cec5SDimitry Andric   assert(CmpValTy.isValid() && "invalid operand type");
7480b57cec5SDimitry Andric   assert(NewValTy.isValid() && "invalid operand type");
7490b57cec5SDimitry Andric   assert(OldValResTy == CmpValTy && "type mismatch");
7500b57cec5SDimitry Andric   assert(OldValResTy == NewValTy && "type mismatch");
7510b57cec5SDimitry Andric #endif
7520b57cec5SDimitry Andric 
7530b57cec5SDimitry Andric   return buildInstr(TargetOpcode::G_ATOMIC_CMPXCHG_WITH_SUCCESS)
7540b57cec5SDimitry Andric       .addDef(OldValRes)
7550b57cec5SDimitry Andric       .addDef(SuccessRes)
7560b57cec5SDimitry Andric       .addUse(Addr)
7570b57cec5SDimitry Andric       .addUse(CmpVal)
7580b57cec5SDimitry Andric       .addUse(NewVal)
7590b57cec5SDimitry Andric       .addMemOperand(&MMO);
7600b57cec5SDimitry Andric }
7610b57cec5SDimitry Andric 
7620b57cec5SDimitry Andric MachineInstrBuilder
7630b57cec5SDimitry Andric MachineIRBuilder::buildAtomicCmpXchg(Register OldValRes, Register Addr,
7640b57cec5SDimitry Andric                                      Register CmpVal, Register NewVal,
7650b57cec5SDimitry Andric                                      MachineMemOperand &MMO) {
7660b57cec5SDimitry Andric #ifndef NDEBUG
7670b57cec5SDimitry Andric   LLT OldValResTy = getMRI()->getType(OldValRes);
7680b57cec5SDimitry Andric   LLT AddrTy = getMRI()->getType(Addr);
7690b57cec5SDimitry Andric   LLT CmpValTy = getMRI()->getType(CmpVal);
7700b57cec5SDimitry Andric   LLT NewValTy = getMRI()->getType(NewVal);
7710b57cec5SDimitry Andric   assert(OldValResTy.isScalar() && "invalid operand type");
7720b57cec5SDimitry Andric   assert(AddrTy.isPointer() && "invalid operand type");
7730b57cec5SDimitry Andric   assert(CmpValTy.isValid() && "invalid operand type");
7740b57cec5SDimitry Andric   assert(NewValTy.isValid() && "invalid operand type");
7750b57cec5SDimitry Andric   assert(OldValResTy == CmpValTy && "type mismatch");
7760b57cec5SDimitry Andric   assert(OldValResTy == NewValTy && "type mismatch");
7770b57cec5SDimitry Andric #endif
7780b57cec5SDimitry Andric 
7790b57cec5SDimitry Andric   return buildInstr(TargetOpcode::G_ATOMIC_CMPXCHG)
7800b57cec5SDimitry Andric       .addDef(OldValRes)
7810b57cec5SDimitry Andric       .addUse(Addr)
7820b57cec5SDimitry Andric       .addUse(CmpVal)
7830b57cec5SDimitry Andric       .addUse(NewVal)
7840b57cec5SDimitry Andric       .addMemOperand(&MMO);
7850b57cec5SDimitry Andric }
7860b57cec5SDimitry Andric 
7878bcb0991SDimitry Andric MachineInstrBuilder MachineIRBuilder::buildAtomicRMW(
7888bcb0991SDimitry Andric   unsigned Opcode, const DstOp &OldValRes,
7898bcb0991SDimitry Andric   const SrcOp &Addr, const SrcOp &Val,
7900b57cec5SDimitry Andric   MachineMemOperand &MMO) {
7918bcb0991SDimitry Andric 
7920b57cec5SDimitry Andric #ifndef NDEBUG
7938bcb0991SDimitry Andric   LLT OldValResTy = OldValRes.getLLTTy(*getMRI());
7948bcb0991SDimitry Andric   LLT AddrTy = Addr.getLLTTy(*getMRI());
7958bcb0991SDimitry Andric   LLT ValTy = Val.getLLTTy(*getMRI());
7960b57cec5SDimitry Andric   assert(OldValResTy.isScalar() && "invalid operand type");
7970b57cec5SDimitry Andric   assert(AddrTy.isPointer() && "invalid operand type");
7980b57cec5SDimitry Andric   assert(ValTy.isValid() && "invalid operand type");
7990b57cec5SDimitry Andric   assert(OldValResTy == ValTy && "type mismatch");
8008bcb0991SDimitry Andric   assert(MMO.isAtomic() && "not atomic mem operand");
8010b57cec5SDimitry Andric #endif
8020b57cec5SDimitry Andric 
8038bcb0991SDimitry Andric   auto MIB = buildInstr(Opcode);
8048bcb0991SDimitry Andric   OldValRes.addDefToMIB(*getMRI(), MIB);
8058bcb0991SDimitry Andric   Addr.addSrcToMIB(MIB);
8068bcb0991SDimitry Andric   Val.addSrcToMIB(MIB);
8078bcb0991SDimitry Andric   MIB.addMemOperand(&MMO);
8088bcb0991SDimitry Andric   return MIB;
8090b57cec5SDimitry Andric }
8100b57cec5SDimitry Andric 
8110b57cec5SDimitry Andric MachineInstrBuilder
8120b57cec5SDimitry Andric MachineIRBuilder::buildAtomicRMWXchg(Register OldValRes, Register Addr,
8130b57cec5SDimitry Andric                                      Register Val, MachineMemOperand &MMO) {
8140b57cec5SDimitry Andric   return buildAtomicRMW(TargetOpcode::G_ATOMICRMW_XCHG, OldValRes, Addr, Val,
8150b57cec5SDimitry Andric                         MMO);
8160b57cec5SDimitry Andric }
8170b57cec5SDimitry Andric MachineInstrBuilder
8180b57cec5SDimitry Andric MachineIRBuilder::buildAtomicRMWAdd(Register OldValRes, Register Addr,
8190b57cec5SDimitry Andric                                     Register Val, MachineMemOperand &MMO) {
8200b57cec5SDimitry Andric   return buildAtomicRMW(TargetOpcode::G_ATOMICRMW_ADD, OldValRes, Addr, Val,
8210b57cec5SDimitry Andric                         MMO);
8220b57cec5SDimitry Andric }
8230b57cec5SDimitry Andric MachineInstrBuilder
8240b57cec5SDimitry Andric MachineIRBuilder::buildAtomicRMWSub(Register OldValRes, Register Addr,
8250b57cec5SDimitry Andric                                     Register Val, MachineMemOperand &MMO) {
8260b57cec5SDimitry Andric   return buildAtomicRMW(TargetOpcode::G_ATOMICRMW_SUB, OldValRes, Addr, Val,
8270b57cec5SDimitry Andric                         MMO);
8280b57cec5SDimitry Andric }
8290b57cec5SDimitry Andric MachineInstrBuilder
8300b57cec5SDimitry Andric MachineIRBuilder::buildAtomicRMWAnd(Register OldValRes, Register Addr,
8310b57cec5SDimitry Andric                                     Register Val, MachineMemOperand &MMO) {
8320b57cec5SDimitry Andric   return buildAtomicRMW(TargetOpcode::G_ATOMICRMW_AND, OldValRes, Addr, Val,
8330b57cec5SDimitry Andric                         MMO);
8340b57cec5SDimitry Andric }
8350b57cec5SDimitry Andric MachineInstrBuilder
8360b57cec5SDimitry Andric MachineIRBuilder::buildAtomicRMWNand(Register OldValRes, Register Addr,
8370b57cec5SDimitry Andric                                      Register Val, MachineMemOperand &MMO) {
8380b57cec5SDimitry Andric   return buildAtomicRMW(TargetOpcode::G_ATOMICRMW_NAND, OldValRes, Addr, Val,
8390b57cec5SDimitry Andric                         MMO);
8400b57cec5SDimitry Andric }
8410b57cec5SDimitry Andric MachineInstrBuilder MachineIRBuilder::buildAtomicRMWOr(Register OldValRes,
8420b57cec5SDimitry Andric                                                        Register Addr,
8430b57cec5SDimitry Andric                                                        Register Val,
8440b57cec5SDimitry Andric                                                        MachineMemOperand &MMO) {
8450b57cec5SDimitry Andric   return buildAtomicRMW(TargetOpcode::G_ATOMICRMW_OR, OldValRes, Addr, Val,
8460b57cec5SDimitry Andric                         MMO);
8470b57cec5SDimitry Andric }
8480b57cec5SDimitry Andric MachineInstrBuilder
8490b57cec5SDimitry Andric MachineIRBuilder::buildAtomicRMWXor(Register OldValRes, Register Addr,
8500b57cec5SDimitry Andric                                     Register Val, MachineMemOperand &MMO) {
8510b57cec5SDimitry Andric   return buildAtomicRMW(TargetOpcode::G_ATOMICRMW_XOR, OldValRes, Addr, Val,
8520b57cec5SDimitry Andric                         MMO);
8530b57cec5SDimitry Andric }
8540b57cec5SDimitry Andric MachineInstrBuilder
8550b57cec5SDimitry Andric MachineIRBuilder::buildAtomicRMWMax(Register OldValRes, Register Addr,
8560b57cec5SDimitry Andric                                     Register Val, MachineMemOperand &MMO) {
8570b57cec5SDimitry Andric   return buildAtomicRMW(TargetOpcode::G_ATOMICRMW_MAX, OldValRes, Addr, Val,
8580b57cec5SDimitry Andric                         MMO);
8590b57cec5SDimitry Andric }
8600b57cec5SDimitry Andric MachineInstrBuilder
8610b57cec5SDimitry Andric MachineIRBuilder::buildAtomicRMWMin(Register OldValRes, Register Addr,
8620b57cec5SDimitry Andric                                     Register Val, MachineMemOperand &MMO) {
8630b57cec5SDimitry Andric   return buildAtomicRMW(TargetOpcode::G_ATOMICRMW_MIN, OldValRes, Addr, Val,
8640b57cec5SDimitry Andric                         MMO);
8650b57cec5SDimitry Andric }
8660b57cec5SDimitry Andric MachineInstrBuilder
8670b57cec5SDimitry Andric MachineIRBuilder::buildAtomicRMWUmax(Register OldValRes, Register Addr,
8680b57cec5SDimitry Andric                                      Register Val, MachineMemOperand &MMO) {
8690b57cec5SDimitry Andric   return buildAtomicRMW(TargetOpcode::G_ATOMICRMW_UMAX, OldValRes, Addr, Val,
8700b57cec5SDimitry Andric                         MMO);
8710b57cec5SDimitry Andric }
8720b57cec5SDimitry Andric MachineInstrBuilder
8730b57cec5SDimitry Andric MachineIRBuilder::buildAtomicRMWUmin(Register OldValRes, Register Addr,
8740b57cec5SDimitry Andric                                      Register Val, MachineMemOperand &MMO) {
8750b57cec5SDimitry Andric   return buildAtomicRMW(TargetOpcode::G_ATOMICRMW_UMIN, OldValRes, Addr, Val,
8760b57cec5SDimitry Andric                         MMO);
8770b57cec5SDimitry Andric }
8780b57cec5SDimitry Andric 
8790b57cec5SDimitry Andric MachineInstrBuilder
8808bcb0991SDimitry Andric MachineIRBuilder::buildAtomicRMWFAdd(
8818bcb0991SDimitry Andric   const DstOp &OldValRes, const SrcOp &Addr, const SrcOp &Val,
8828bcb0991SDimitry Andric   MachineMemOperand &MMO) {
8838bcb0991SDimitry Andric   return buildAtomicRMW(TargetOpcode::G_ATOMICRMW_FADD, OldValRes, Addr, Val,
8848bcb0991SDimitry Andric                         MMO);
8858bcb0991SDimitry Andric }
8868bcb0991SDimitry Andric 
8878bcb0991SDimitry Andric MachineInstrBuilder
8888bcb0991SDimitry Andric MachineIRBuilder::buildAtomicRMWFSub(const DstOp &OldValRes, const SrcOp &Addr, const SrcOp &Val,
8898bcb0991SDimitry Andric                                      MachineMemOperand &MMO) {
8908bcb0991SDimitry Andric   return buildAtomicRMW(TargetOpcode::G_ATOMICRMW_FSUB, OldValRes, Addr, Val,
8918bcb0991SDimitry Andric                         MMO);
8928bcb0991SDimitry Andric }
8938bcb0991SDimitry Andric 
8948bcb0991SDimitry Andric MachineInstrBuilder
8950b57cec5SDimitry Andric MachineIRBuilder::buildFence(unsigned Ordering, unsigned Scope) {
8960b57cec5SDimitry Andric   return buildInstr(TargetOpcode::G_FENCE)
8970b57cec5SDimitry Andric     .addImm(Ordering)
8980b57cec5SDimitry Andric     .addImm(Scope);
8990b57cec5SDimitry Andric }
9000b57cec5SDimitry Andric 
9010b57cec5SDimitry Andric MachineInstrBuilder
9020b57cec5SDimitry Andric MachineIRBuilder::buildBlockAddress(Register Res, const BlockAddress *BA) {
9030b57cec5SDimitry Andric #ifndef NDEBUG
9040b57cec5SDimitry Andric   assert(getMRI()->getType(Res).isPointer() && "invalid res type");
9050b57cec5SDimitry Andric #endif
9060b57cec5SDimitry Andric 
9070b57cec5SDimitry Andric   return buildInstr(TargetOpcode::G_BLOCK_ADDR).addDef(Res).addBlockAddress(BA);
9080b57cec5SDimitry Andric }
9090b57cec5SDimitry Andric 
9100b57cec5SDimitry Andric void MachineIRBuilder::validateTruncExt(const LLT &DstTy, const LLT &SrcTy,
9110b57cec5SDimitry Andric                                         bool IsExtend) {
9120b57cec5SDimitry Andric #ifndef NDEBUG
9130b57cec5SDimitry Andric   if (DstTy.isVector()) {
9140b57cec5SDimitry Andric     assert(SrcTy.isVector() && "mismatched cast between vector and non-vector");
9150b57cec5SDimitry Andric     assert(SrcTy.getNumElements() == DstTy.getNumElements() &&
9160b57cec5SDimitry Andric            "different number of elements in a trunc/ext");
9170b57cec5SDimitry Andric   } else
9180b57cec5SDimitry Andric     assert(DstTy.isScalar() && SrcTy.isScalar() && "invalid extend/trunc");
9190b57cec5SDimitry Andric 
9200b57cec5SDimitry Andric   if (IsExtend)
9210b57cec5SDimitry Andric     assert(DstTy.getSizeInBits() > SrcTy.getSizeInBits() &&
9220b57cec5SDimitry Andric            "invalid narrowing extend");
9230b57cec5SDimitry Andric   else
9240b57cec5SDimitry Andric     assert(DstTy.getSizeInBits() < SrcTy.getSizeInBits() &&
9250b57cec5SDimitry Andric            "invalid widening trunc");
9260b57cec5SDimitry Andric #endif
9270b57cec5SDimitry Andric }
9280b57cec5SDimitry Andric 
9290b57cec5SDimitry Andric void MachineIRBuilder::validateSelectOp(const LLT &ResTy, const LLT &TstTy,
9300b57cec5SDimitry Andric                                         const LLT &Op0Ty, const LLT &Op1Ty) {
9310b57cec5SDimitry Andric #ifndef NDEBUG
9320b57cec5SDimitry Andric   assert((ResTy.isScalar() || ResTy.isVector() || ResTy.isPointer()) &&
9330b57cec5SDimitry Andric          "invalid operand type");
9340b57cec5SDimitry Andric   assert((ResTy == Op0Ty && ResTy == Op1Ty) && "type mismatch");
9350b57cec5SDimitry Andric   if (ResTy.isScalar() || ResTy.isPointer())
9360b57cec5SDimitry Andric     assert(TstTy.isScalar() && "type mismatch");
9370b57cec5SDimitry Andric   else
9380b57cec5SDimitry Andric     assert((TstTy.isScalar() ||
9390b57cec5SDimitry Andric             (TstTy.isVector() &&
9400b57cec5SDimitry Andric              TstTy.getNumElements() == Op0Ty.getNumElements())) &&
9410b57cec5SDimitry Andric            "type mismatch");
9420b57cec5SDimitry Andric #endif
9430b57cec5SDimitry Andric }
9440b57cec5SDimitry Andric 
9450b57cec5SDimitry Andric MachineInstrBuilder MachineIRBuilder::buildInstr(unsigned Opc,
9460b57cec5SDimitry Andric                                                  ArrayRef<DstOp> DstOps,
9470b57cec5SDimitry Andric                                                  ArrayRef<SrcOp> SrcOps,
9480b57cec5SDimitry Andric                                                  Optional<unsigned> Flags) {
9490b57cec5SDimitry Andric   switch (Opc) {
9500b57cec5SDimitry Andric   default:
9510b57cec5SDimitry Andric     break;
9520b57cec5SDimitry Andric   case TargetOpcode::G_SELECT: {
9530b57cec5SDimitry Andric     assert(DstOps.size() == 1 && "Invalid select");
9540b57cec5SDimitry Andric     assert(SrcOps.size() == 3 && "Invalid select");
9550b57cec5SDimitry Andric     validateSelectOp(
9560b57cec5SDimitry Andric         DstOps[0].getLLTTy(*getMRI()), SrcOps[0].getLLTTy(*getMRI()),
9570b57cec5SDimitry Andric         SrcOps[1].getLLTTy(*getMRI()), SrcOps[2].getLLTTy(*getMRI()));
9580b57cec5SDimitry Andric     break;
9590b57cec5SDimitry Andric   }
9600b57cec5SDimitry Andric   case TargetOpcode::G_ADD:
9610b57cec5SDimitry Andric   case TargetOpcode::G_AND:
9620b57cec5SDimitry Andric   case TargetOpcode::G_MUL:
9630b57cec5SDimitry Andric   case TargetOpcode::G_OR:
9640b57cec5SDimitry Andric   case TargetOpcode::G_SUB:
9650b57cec5SDimitry Andric   case TargetOpcode::G_XOR:
9660b57cec5SDimitry Andric   case TargetOpcode::G_UDIV:
9670b57cec5SDimitry Andric   case TargetOpcode::G_SDIV:
9680b57cec5SDimitry Andric   case TargetOpcode::G_UREM:
9690b57cec5SDimitry Andric   case TargetOpcode::G_SREM:
9700b57cec5SDimitry Andric   case TargetOpcode::G_SMIN:
9710b57cec5SDimitry Andric   case TargetOpcode::G_SMAX:
9720b57cec5SDimitry Andric   case TargetOpcode::G_UMIN:
9730b57cec5SDimitry Andric   case TargetOpcode::G_UMAX: {
9740b57cec5SDimitry Andric     // All these are binary ops.
9750b57cec5SDimitry Andric     assert(DstOps.size() == 1 && "Invalid Dst");
9760b57cec5SDimitry Andric     assert(SrcOps.size() == 2 && "Invalid Srcs");
9770b57cec5SDimitry Andric     validateBinaryOp(DstOps[0].getLLTTy(*getMRI()),
9780b57cec5SDimitry Andric                      SrcOps[0].getLLTTy(*getMRI()),
9790b57cec5SDimitry Andric                      SrcOps[1].getLLTTy(*getMRI()));
9800b57cec5SDimitry Andric     break;
9810b57cec5SDimitry Andric   }
9820b57cec5SDimitry Andric   case TargetOpcode::G_SHL:
9830b57cec5SDimitry Andric   case TargetOpcode::G_ASHR:
9840b57cec5SDimitry Andric   case TargetOpcode::G_LSHR: {
9850b57cec5SDimitry Andric     assert(DstOps.size() == 1 && "Invalid Dst");
9860b57cec5SDimitry Andric     assert(SrcOps.size() == 2 && "Invalid Srcs");
9870b57cec5SDimitry Andric     validateShiftOp(DstOps[0].getLLTTy(*getMRI()),
9880b57cec5SDimitry Andric                     SrcOps[0].getLLTTy(*getMRI()),
9890b57cec5SDimitry Andric                     SrcOps[1].getLLTTy(*getMRI()));
9900b57cec5SDimitry Andric     break;
9910b57cec5SDimitry Andric   }
9920b57cec5SDimitry Andric   case TargetOpcode::G_SEXT:
9930b57cec5SDimitry Andric   case TargetOpcode::G_ZEXT:
9940b57cec5SDimitry Andric   case TargetOpcode::G_ANYEXT:
9950b57cec5SDimitry Andric     assert(DstOps.size() == 1 && "Invalid Dst");
9960b57cec5SDimitry Andric     assert(SrcOps.size() == 1 && "Invalid Srcs");
9970b57cec5SDimitry Andric     validateTruncExt(DstOps[0].getLLTTy(*getMRI()),
9980b57cec5SDimitry Andric                      SrcOps[0].getLLTTy(*getMRI()), true);
9990b57cec5SDimitry Andric     break;
10000b57cec5SDimitry Andric   case TargetOpcode::G_TRUNC:
10010b57cec5SDimitry Andric   case TargetOpcode::G_FPTRUNC: {
10020b57cec5SDimitry Andric     assert(DstOps.size() == 1 && "Invalid Dst");
10030b57cec5SDimitry Andric     assert(SrcOps.size() == 1 && "Invalid Srcs");
10040b57cec5SDimitry Andric     validateTruncExt(DstOps[0].getLLTTy(*getMRI()),
10050b57cec5SDimitry Andric                      SrcOps[0].getLLTTy(*getMRI()), false);
10060b57cec5SDimitry Andric     break;
10070b57cec5SDimitry Andric   }
10080b57cec5SDimitry Andric   case TargetOpcode::COPY:
10090b57cec5SDimitry Andric     assert(DstOps.size() == 1 && "Invalid Dst");
10100b57cec5SDimitry Andric     // If the caller wants to add a subreg source it has to be done separately
10110b57cec5SDimitry Andric     // so we may not have any SrcOps at this point yet.
10120b57cec5SDimitry Andric     break;
10130b57cec5SDimitry Andric   case TargetOpcode::G_FCMP:
10140b57cec5SDimitry Andric   case TargetOpcode::G_ICMP: {
10150b57cec5SDimitry Andric     assert(DstOps.size() == 1 && "Invalid Dst Operands");
10160b57cec5SDimitry Andric     assert(SrcOps.size() == 3 && "Invalid Src Operands");
10170b57cec5SDimitry Andric     // For F/ICMP, the first src operand is the predicate, followed by
10180b57cec5SDimitry Andric     // the two comparands.
10190b57cec5SDimitry Andric     assert(SrcOps[0].getSrcOpKind() == SrcOp::SrcType::Ty_Predicate &&
10200b57cec5SDimitry Andric            "Expecting predicate");
10210b57cec5SDimitry Andric     assert([&]() -> bool {
10220b57cec5SDimitry Andric       CmpInst::Predicate Pred = SrcOps[0].getPredicate();
10230b57cec5SDimitry Andric       return Opc == TargetOpcode::G_ICMP ? CmpInst::isIntPredicate(Pred)
10240b57cec5SDimitry Andric                                          : CmpInst::isFPPredicate(Pred);
10250b57cec5SDimitry Andric     }() && "Invalid predicate");
10260b57cec5SDimitry Andric     assert(SrcOps[1].getLLTTy(*getMRI()) == SrcOps[2].getLLTTy(*getMRI()) &&
10270b57cec5SDimitry Andric            "Type mismatch");
10280b57cec5SDimitry Andric     assert([&]() -> bool {
10290b57cec5SDimitry Andric       LLT Op0Ty = SrcOps[1].getLLTTy(*getMRI());
10300b57cec5SDimitry Andric       LLT DstTy = DstOps[0].getLLTTy(*getMRI());
10310b57cec5SDimitry Andric       if (Op0Ty.isScalar() || Op0Ty.isPointer())
10320b57cec5SDimitry Andric         return DstTy.isScalar();
10330b57cec5SDimitry Andric       else
10340b57cec5SDimitry Andric         return DstTy.isVector() &&
10350b57cec5SDimitry Andric                DstTy.getNumElements() == Op0Ty.getNumElements();
10360b57cec5SDimitry Andric     }() && "Type Mismatch");
10370b57cec5SDimitry Andric     break;
10380b57cec5SDimitry Andric   }
10390b57cec5SDimitry Andric   case TargetOpcode::G_UNMERGE_VALUES: {
10400b57cec5SDimitry Andric     assert(!DstOps.empty() && "Invalid trivial sequence");
10410b57cec5SDimitry Andric     assert(SrcOps.size() == 1 && "Invalid src for Unmerge");
10420b57cec5SDimitry Andric     assert(std::all_of(DstOps.begin(), DstOps.end(),
10430b57cec5SDimitry Andric                        [&, this](const DstOp &Op) {
10440b57cec5SDimitry Andric                          return Op.getLLTTy(*getMRI()) ==
10450b57cec5SDimitry Andric                                 DstOps[0].getLLTTy(*getMRI());
10460b57cec5SDimitry Andric                        }) &&
10470b57cec5SDimitry Andric            "type mismatch in output list");
10480b57cec5SDimitry Andric     assert(DstOps.size() * DstOps[0].getLLTTy(*getMRI()).getSizeInBits() ==
10490b57cec5SDimitry Andric                SrcOps[0].getLLTTy(*getMRI()).getSizeInBits() &&
10500b57cec5SDimitry Andric            "input operands do not cover output register");
10510b57cec5SDimitry Andric     break;
10520b57cec5SDimitry Andric   }
10530b57cec5SDimitry Andric   case TargetOpcode::G_MERGE_VALUES: {
10540b57cec5SDimitry Andric     assert(!SrcOps.empty() && "invalid trivial sequence");
10550b57cec5SDimitry Andric     assert(DstOps.size() == 1 && "Invalid Dst");
10560b57cec5SDimitry Andric     assert(std::all_of(SrcOps.begin(), SrcOps.end(),
10570b57cec5SDimitry Andric                        [&, this](const SrcOp &Op) {
10580b57cec5SDimitry Andric                          return Op.getLLTTy(*getMRI()) ==
10590b57cec5SDimitry Andric                                 SrcOps[0].getLLTTy(*getMRI());
10600b57cec5SDimitry Andric                        }) &&
10610b57cec5SDimitry Andric            "type mismatch in input list");
10620b57cec5SDimitry Andric     assert(SrcOps.size() * SrcOps[0].getLLTTy(*getMRI()).getSizeInBits() ==
10630b57cec5SDimitry Andric                DstOps[0].getLLTTy(*getMRI()).getSizeInBits() &&
10640b57cec5SDimitry Andric            "input operands do not cover output register");
10650b57cec5SDimitry Andric     if (SrcOps.size() == 1)
10660b57cec5SDimitry Andric       return buildCast(DstOps[0], SrcOps[0]);
10678bcb0991SDimitry Andric     if (DstOps[0].getLLTTy(*getMRI()).isVector()) {
10688bcb0991SDimitry Andric       if (SrcOps[0].getLLTTy(*getMRI()).isVector())
10690b57cec5SDimitry Andric         return buildInstr(TargetOpcode::G_CONCAT_VECTORS, DstOps, SrcOps);
10708bcb0991SDimitry Andric       return buildInstr(TargetOpcode::G_BUILD_VECTOR, DstOps, SrcOps);
10718bcb0991SDimitry Andric     }
10720b57cec5SDimitry Andric     break;
10730b57cec5SDimitry Andric   }
10740b57cec5SDimitry Andric   case TargetOpcode::G_EXTRACT_VECTOR_ELT: {
10750b57cec5SDimitry Andric     assert(DstOps.size() == 1 && "Invalid Dst size");
10760b57cec5SDimitry Andric     assert(SrcOps.size() == 2 && "Invalid Src size");
10770b57cec5SDimitry Andric     assert(SrcOps[0].getLLTTy(*getMRI()).isVector() && "Invalid operand type");
10780b57cec5SDimitry Andric     assert((DstOps[0].getLLTTy(*getMRI()).isScalar() ||
10790b57cec5SDimitry Andric             DstOps[0].getLLTTy(*getMRI()).isPointer()) &&
10800b57cec5SDimitry Andric            "Invalid operand type");
10810b57cec5SDimitry Andric     assert(SrcOps[1].getLLTTy(*getMRI()).isScalar() && "Invalid operand type");
10820b57cec5SDimitry Andric     assert(SrcOps[0].getLLTTy(*getMRI()).getElementType() ==
10830b57cec5SDimitry Andric                DstOps[0].getLLTTy(*getMRI()) &&
10840b57cec5SDimitry Andric            "Type mismatch");
10850b57cec5SDimitry Andric     break;
10860b57cec5SDimitry Andric   }
10870b57cec5SDimitry Andric   case TargetOpcode::G_INSERT_VECTOR_ELT: {
10880b57cec5SDimitry Andric     assert(DstOps.size() == 1 && "Invalid dst size");
10890b57cec5SDimitry Andric     assert(SrcOps.size() == 3 && "Invalid src size");
10900b57cec5SDimitry Andric     assert(DstOps[0].getLLTTy(*getMRI()).isVector() &&
10910b57cec5SDimitry Andric            SrcOps[0].getLLTTy(*getMRI()).isVector() && "Invalid operand type");
10920b57cec5SDimitry Andric     assert(DstOps[0].getLLTTy(*getMRI()).getElementType() ==
10930b57cec5SDimitry Andric                SrcOps[1].getLLTTy(*getMRI()) &&
10940b57cec5SDimitry Andric            "Type mismatch");
10950b57cec5SDimitry Andric     assert(SrcOps[2].getLLTTy(*getMRI()).isScalar() && "Invalid index");
10960b57cec5SDimitry Andric     assert(DstOps[0].getLLTTy(*getMRI()).getNumElements() ==
10970b57cec5SDimitry Andric                SrcOps[0].getLLTTy(*getMRI()).getNumElements() &&
10980b57cec5SDimitry Andric            "Type mismatch");
10990b57cec5SDimitry Andric     break;
11000b57cec5SDimitry Andric   }
11010b57cec5SDimitry Andric   case TargetOpcode::G_BUILD_VECTOR: {
11020b57cec5SDimitry Andric     assert((!SrcOps.empty() || SrcOps.size() < 2) &&
11030b57cec5SDimitry Andric            "Must have at least 2 operands");
11040b57cec5SDimitry Andric     assert(DstOps.size() == 1 && "Invalid DstOps");
11050b57cec5SDimitry Andric     assert(DstOps[0].getLLTTy(*getMRI()).isVector() &&
11060b57cec5SDimitry Andric            "Res type must be a vector");
11070b57cec5SDimitry Andric     assert(std::all_of(SrcOps.begin(), SrcOps.end(),
11080b57cec5SDimitry Andric                        [&, this](const SrcOp &Op) {
11090b57cec5SDimitry Andric                          return Op.getLLTTy(*getMRI()) ==
11100b57cec5SDimitry Andric                                 SrcOps[0].getLLTTy(*getMRI());
11110b57cec5SDimitry Andric                        }) &&
11120b57cec5SDimitry Andric            "type mismatch in input list");
11130b57cec5SDimitry Andric     assert(SrcOps.size() * SrcOps[0].getLLTTy(*getMRI()).getSizeInBits() ==
11140b57cec5SDimitry Andric                DstOps[0].getLLTTy(*getMRI()).getSizeInBits() &&
11150b57cec5SDimitry Andric            "input scalars do not exactly cover the output vector register");
11160b57cec5SDimitry Andric     break;
11170b57cec5SDimitry Andric   }
11180b57cec5SDimitry Andric   case TargetOpcode::G_BUILD_VECTOR_TRUNC: {
11190b57cec5SDimitry Andric     assert((!SrcOps.empty() || SrcOps.size() < 2) &&
11200b57cec5SDimitry Andric            "Must have at least 2 operands");
11210b57cec5SDimitry Andric     assert(DstOps.size() == 1 && "Invalid DstOps");
11220b57cec5SDimitry Andric     assert(DstOps[0].getLLTTy(*getMRI()).isVector() &&
11230b57cec5SDimitry Andric            "Res type must be a vector");
11240b57cec5SDimitry Andric     assert(std::all_of(SrcOps.begin(), SrcOps.end(),
11250b57cec5SDimitry Andric                        [&, this](const SrcOp &Op) {
11260b57cec5SDimitry Andric                          return Op.getLLTTy(*getMRI()) ==
11270b57cec5SDimitry Andric                                 SrcOps[0].getLLTTy(*getMRI());
11280b57cec5SDimitry Andric                        }) &&
11290b57cec5SDimitry Andric            "type mismatch in input list");
11300b57cec5SDimitry Andric     if (SrcOps[0].getLLTTy(*getMRI()).getSizeInBits() ==
11310b57cec5SDimitry Andric         DstOps[0].getLLTTy(*getMRI()).getElementType().getSizeInBits())
11320b57cec5SDimitry Andric       return buildInstr(TargetOpcode::G_BUILD_VECTOR, DstOps, SrcOps);
11330b57cec5SDimitry Andric     break;
11340b57cec5SDimitry Andric   }
11350b57cec5SDimitry Andric   case TargetOpcode::G_CONCAT_VECTORS: {
11360b57cec5SDimitry Andric     assert(DstOps.size() == 1 && "Invalid DstOps");
11370b57cec5SDimitry Andric     assert((!SrcOps.empty() || SrcOps.size() < 2) &&
11380b57cec5SDimitry Andric            "Must have at least 2 operands");
11390b57cec5SDimitry Andric     assert(std::all_of(SrcOps.begin(), SrcOps.end(),
11400b57cec5SDimitry Andric                        [&, this](const SrcOp &Op) {
11410b57cec5SDimitry Andric                          return (Op.getLLTTy(*getMRI()).isVector() &&
11420b57cec5SDimitry Andric                                  Op.getLLTTy(*getMRI()) ==
11430b57cec5SDimitry Andric                                      SrcOps[0].getLLTTy(*getMRI()));
11440b57cec5SDimitry Andric                        }) &&
11450b57cec5SDimitry Andric            "type mismatch in input list");
11460b57cec5SDimitry Andric     assert(SrcOps.size() * SrcOps[0].getLLTTy(*getMRI()).getSizeInBits() ==
11470b57cec5SDimitry Andric                DstOps[0].getLLTTy(*getMRI()).getSizeInBits() &&
11480b57cec5SDimitry Andric            "input vectors do not exactly cover the output vector register");
11490b57cec5SDimitry Andric     break;
11500b57cec5SDimitry Andric   }
11510b57cec5SDimitry Andric   case TargetOpcode::G_UADDE: {
11520b57cec5SDimitry Andric     assert(DstOps.size() == 2 && "Invalid no of dst operands");
11530b57cec5SDimitry Andric     assert(SrcOps.size() == 3 && "Invalid no of src operands");
11540b57cec5SDimitry Andric     assert(DstOps[0].getLLTTy(*getMRI()).isScalar() && "Invalid operand");
11550b57cec5SDimitry Andric     assert((DstOps[0].getLLTTy(*getMRI()) == SrcOps[0].getLLTTy(*getMRI())) &&
11560b57cec5SDimitry Andric            (DstOps[0].getLLTTy(*getMRI()) == SrcOps[1].getLLTTy(*getMRI())) &&
11570b57cec5SDimitry Andric            "Invalid operand");
11580b57cec5SDimitry Andric     assert(DstOps[1].getLLTTy(*getMRI()).isScalar() && "Invalid operand");
11590b57cec5SDimitry Andric     assert(DstOps[1].getLLTTy(*getMRI()) == SrcOps[2].getLLTTy(*getMRI()) &&
11600b57cec5SDimitry Andric            "type mismatch");
11610b57cec5SDimitry Andric     break;
11620b57cec5SDimitry Andric   }
11630b57cec5SDimitry Andric   }
11640b57cec5SDimitry Andric 
11650b57cec5SDimitry Andric   auto MIB = buildInstr(Opc);
11660b57cec5SDimitry Andric   for (const DstOp &Op : DstOps)
11670b57cec5SDimitry Andric     Op.addDefToMIB(*getMRI(), MIB);
11680b57cec5SDimitry Andric   for (const SrcOp &Op : SrcOps)
11690b57cec5SDimitry Andric     Op.addSrcToMIB(MIB);
11700b57cec5SDimitry Andric   if (Flags)
11710b57cec5SDimitry Andric     MIB->setFlags(*Flags);
11720b57cec5SDimitry Andric   return MIB;
11730b57cec5SDimitry Andric }
1174