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