1 //===-- llvm/CodeGen/GlobalISel/MachineIRBuilder.cpp - MIBuilder--*- C++ -*-==// 2 // 3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. 4 // See https://llvm.org/LICENSE.txt for license information. 5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception 6 // 7 //===----------------------------------------------------------------------===// 8 /// \file 9 /// This file implements the MachineIRBuidler class. 10 //===----------------------------------------------------------------------===// 11 #include "llvm/CodeGen/GlobalISel/MachineIRBuilder.h" 12 #include "llvm/CodeGen/GlobalISel/GISelChangeObserver.h" 13 14 #include "llvm/CodeGen/MachineFunction.h" 15 #include "llvm/CodeGen/MachineInstr.h" 16 #include "llvm/CodeGen/MachineInstrBuilder.h" 17 #include "llvm/CodeGen/MachineRegisterInfo.h" 18 #include "llvm/CodeGen/TargetInstrInfo.h" 19 #include "llvm/CodeGen/TargetLowering.h" 20 #include "llvm/CodeGen/TargetOpcodes.h" 21 #include "llvm/CodeGen/TargetSubtargetInfo.h" 22 #include "llvm/IR/DebugInfo.h" 23 24 using namespace llvm; 25 26 void MachineIRBuilder::setMF(MachineFunction &MF) { 27 State.MF = &MF; 28 State.MBB = nullptr; 29 State.MRI = &MF.getRegInfo(); 30 State.TII = MF.getSubtarget().getInstrInfo(); 31 State.DL = DebugLoc(); 32 State.II = MachineBasicBlock::iterator(); 33 State.Observer = nullptr; 34 } 35 36 void MachineIRBuilder::setMBB(MachineBasicBlock &MBB) { 37 State.MBB = &MBB; 38 State.II = MBB.end(); 39 assert(&getMF() == MBB.getParent() && 40 "Basic block is in a different function"); 41 } 42 43 void MachineIRBuilder::setInstr(MachineInstr &MI) { 44 assert(MI.getParent() && "Instruction is not part of a basic block"); 45 setMBB(*MI.getParent()); 46 State.II = MI.getIterator(); 47 } 48 49 void MachineIRBuilder::setCSEInfo(GISelCSEInfo *Info) { State.CSEInfo = Info; } 50 51 void MachineIRBuilder::setInsertPt(MachineBasicBlock &MBB, 52 MachineBasicBlock::iterator II) { 53 assert(MBB.getParent() == &getMF() && 54 "Basic block is in a different function"); 55 State.MBB = &MBB; 56 State.II = II; 57 } 58 59 void MachineIRBuilder::recordInsertion(MachineInstr *InsertedInstr) const { 60 if (State.Observer) 61 State.Observer->createdInstr(*InsertedInstr); 62 } 63 64 void MachineIRBuilder::setChangeObserver(GISelChangeObserver &Observer) { 65 State.Observer = &Observer; 66 } 67 68 void MachineIRBuilder::stopObservingChanges() { State.Observer = nullptr; } 69 70 //------------------------------------------------------------------------------ 71 // Build instruction variants. 72 //------------------------------------------------------------------------------ 73 74 MachineInstrBuilder MachineIRBuilder::buildInstr(unsigned Opcode) { 75 return insertInstr(buildInstrNoInsert(Opcode)); 76 } 77 78 MachineInstrBuilder MachineIRBuilder::buildInstrNoInsert(unsigned Opcode) { 79 MachineInstrBuilder MIB = BuildMI(getMF(), getDL(), getTII().get(Opcode)); 80 return MIB; 81 } 82 83 MachineInstrBuilder MachineIRBuilder::insertInstr(MachineInstrBuilder MIB) { 84 getMBB().insert(getInsertPt(), MIB); 85 recordInsertion(MIB); 86 return MIB; 87 } 88 89 MachineInstrBuilder 90 MachineIRBuilder::buildDirectDbgValue(Register Reg, const MDNode *Variable, 91 const MDNode *Expr) { 92 assert(isa<DILocalVariable>(Variable) && "not a variable"); 93 assert(cast<DIExpression>(Expr)->isValid() && "not an expression"); 94 assert( 95 cast<DILocalVariable>(Variable)->isValidLocationForIntrinsic(getDL()) && 96 "Expected inlined-at fields to agree"); 97 return insertInstr(BuildMI(getMF(), getDL(), 98 getTII().get(TargetOpcode::DBG_VALUE), 99 /*IsIndirect*/ false, Reg, Variable, Expr)); 100 } 101 102 MachineInstrBuilder 103 MachineIRBuilder::buildIndirectDbgValue(Register Reg, const MDNode *Variable, 104 const MDNode *Expr) { 105 assert(isa<DILocalVariable>(Variable) && "not a variable"); 106 assert(cast<DIExpression>(Expr)->isValid() && "not an expression"); 107 assert( 108 cast<DILocalVariable>(Variable)->isValidLocationForIntrinsic(getDL()) && 109 "Expected inlined-at fields to agree"); 110 return insertInstr(BuildMI(getMF(), getDL(), 111 getTII().get(TargetOpcode::DBG_VALUE), 112 /*IsIndirect*/ true, Reg, Variable, Expr)); 113 } 114 115 MachineInstrBuilder MachineIRBuilder::buildFIDbgValue(int FI, 116 const MDNode *Variable, 117 const MDNode *Expr) { 118 assert(isa<DILocalVariable>(Variable) && "not a variable"); 119 assert(cast<DIExpression>(Expr)->isValid() && "not an expression"); 120 assert( 121 cast<DILocalVariable>(Variable)->isValidLocationForIntrinsic(getDL()) && 122 "Expected inlined-at fields to agree"); 123 return buildInstr(TargetOpcode::DBG_VALUE) 124 .addFrameIndex(FI) 125 .addImm(0) 126 .addMetadata(Variable) 127 .addMetadata(Expr); 128 } 129 130 MachineInstrBuilder MachineIRBuilder::buildConstDbgValue(const Constant &C, 131 const MDNode *Variable, 132 const MDNode *Expr) { 133 assert(isa<DILocalVariable>(Variable) && "not a variable"); 134 assert(cast<DIExpression>(Expr)->isValid() && "not an expression"); 135 assert( 136 cast<DILocalVariable>(Variable)->isValidLocationForIntrinsic(getDL()) && 137 "Expected inlined-at fields to agree"); 138 auto MIB = buildInstr(TargetOpcode::DBG_VALUE); 139 if (auto *CI = dyn_cast<ConstantInt>(&C)) { 140 if (CI->getBitWidth() > 64) 141 MIB.addCImm(CI); 142 else 143 MIB.addImm(CI->getZExtValue()); 144 } else if (auto *CFP = dyn_cast<ConstantFP>(&C)) { 145 MIB.addFPImm(CFP); 146 } else { 147 // Insert %noreg if we didn't find a usable constant and had to drop it. 148 MIB.addReg(0U); 149 } 150 151 return MIB.addImm(0).addMetadata(Variable).addMetadata(Expr); 152 } 153 154 MachineInstrBuilder MachineIRBuilder::buildDbgLabel(const MDNode *Label) { 155 assert(isa<DILabel>(Label) && "not a label"); 156 assert(cast<DILabel>(Label)->isValidLocationForIntrinsic(State.DL) && 157 "Expected inlined-at fields to agree"); 158 auto MIB = buildInstr(TargetOpcode::DBG_LABEL); 159 160 return MIB.addMetadata(Label); 161 } 162 163 MachineInstrBuilder MachineIRBuilder::buildDynStackAlloc(const DstOp &Res, 164 const SrcOp &Size, 165 unsigned Align) { 166 assert(Res.getLLTTy(*getMRI()).isPointer() && "expected ptr dst type"); 167 auto MIB = buildInstr(TargetOpcode::G_DYN_STACKALLOC); 168 Res.addDefToMIB(*getMRI(), MIB); 169 Size.addSrcToMIB(MIB); 170 MIB.addImm(Align); 171 return MIB; 172 } 173 174 MachineInstrBuilder MachineIRBuilder::buildFrameIndex(const DstOp &Res, 175 int Idx) { 176 assert(Res.getLLTTy(*getMRI()).isPointer() && "invalid operand type"); 177 auto MIB = buildInstr(TargetOpcode::G_FRAME_INDEX); 178 Res.addDefToMIB(*getMRI(), MIB); 179 MIB.addFrameIndex(Idx); 180 return MIB; 181 } 182 183 MachineInstrBuilder MachineIRBuilder::buildGlobalValue(const DstOp &Res, 184 const GlobalValue *GV) { 185 assert(Res.getLLTTy(*getMRI()).isPointer() && "invalid operand type"); 186 assert(Res.getLLTTy(*getMRI()).getAddressSpace() == 187 GV->getType()->getAddressSpace() && 188 "address space mismatch"); 189 190 auto MIB = buildInstr(TargetOpcode::G_GLOBAL_VALUE); 191 Res.addDefToMIB(*getMRI(), MIB); 192 MIB.addGlobalAddress(GV); 193 return MIB; 194 } 195 196 MachineInstrBuilder MachineIRBuilder::buildJumpTable(const LLT PtrTy, 197 unsigned JTI) { 198 return buildInstr(TargetOpcode::G_JUMP_TABLE, {PtrTy}, {}) 199 .addJumpTableIndex(JTI); 200 } 201 202 void MachineIRBuilder::validateBinaryOp(const LLT &Res, const LLT &Op0, 203 const LLT &Op1) { 204 assert((Res.isScalar() || Res.isVector()) && "invalid operand type"); 205 assert((Res == Op0 && Res == Op1) && "type mismatch"); 206 } 207 208 void MachineIRBuilder::validateShiftOp(const LLT &Res, const LLT &Op0, 209 const LLT &Op1) { 210 assert((Res.isScalar() || Res.isVector()) && "invalid operand type"); 211 assert((Res == Op0) && "type mismatch"); 212 } 213 214 MachineInstrBuilder MachineIRBuilder::buildPtrAdd(const DstOp &Res, 215 const SrcOp &Op0, 216 const SrcOp &Op1) { 217 assert(Res.getLLTTy(*getMRI()).getScalarType().isPointer() && 218 Res.getLLTTy(*getMRI()) == Op0.getLLTTy(*getMRI()) && "type mismatch"); 219 assert(Op1.getLLTTy(*getMRI()).getScalarType().isScalar() && "invalid offset type"); 220 221 return buildInstr(TargetOpcode::G_PTR_ADD, {Res}, {Op0, Op1}); 222 } 223 224 Optional<MachineInstrBuilder> 225 MachineIRBuilder::materializePtrAdd(Register &Res, Register Op0, 226 const LLT &ValueTy, uint64_t Value) { 227 assert(Res == 0 && "Res is a result argument"); 228 assert(ValueTy.isScalar() && "invalid offset type"); 229 230 if (Value == 0) { 231 Res = Op0; 232 return None; 233 } 234 235 Res = getMRI()->createGenericVirtualRegister(getMRI()->getType(Op0)); 236 auto Cst = buildConstant(ValueTy, Value); 237 return buildPtrAdd(Res, Op0, Cst.getReg(0)); 238 } 239 240 MachineInstrBuilder MachineIRBuilder::buildPtrMask(const DstOp &Res, 241 const SrcOp &Op0, 242 uint32_t NumBits) { 243 assert(Res.getLLTTy(*getMRI()).isPointer() && 244 Res.getLLTTy(*getMRI()) == Op0.getLLTTy(*getMRI()) && "type mismatch"); 245 246 auto MIB = buildInstr(TargetOpcode::G_PTR_MASK); 247 Res.addDefToMIB(*getMRI(), MIB); 248 Op0.addSrcToMIB(MIB); 249 MIB.addImm(NumBits); 250 return MIB; 251 } 252 253 MachineInstrBuilder MachineIRBuilder::buildBr(MachineBasicBlock &Dest) { 254 return buildInstr(TargetOpcode::G_BR).addMBB(&Dest); 255 } 256 257 MachineInstrBuilder MachineIRBuilder::buildBrIndirect(Register Tgt) { 258 assert(getMRI()->getType(Tgt).isPointer() && "invalid branch destination"); 259 return buildInstr(TargetOpcode::G_BRINDIRECT).addUse(Tgt); 260 } 261 262 MachineInstrBuilder MachineIRBuilder::buildBrJT(Register TablePtr, 263 unsigned JTI, 264 Register IndexReg) { 265 assert(getMRI()->getType(TablePtr).isPointer() && 266 "Table reg must be a pointer"); 267 return buildInstr(TargetOpcode::G_BRJT) 268 .addUse(TablePtr) 269 .addJumpTableIndex(JTI) 270 .addUse(IndexReg); 271 } 272 273 MachineInstrBuilder MachineIRBuilder::buildCopy(const DstOp &Res, 274 const SrcOp &Op) { 275 return buildInstr(TargetOpcode::COPY, Res, Op); 276 } 277 278 MachineInstrBuilder MachineIRBuilder::buildConstant(const DstOp &Res, 279 const ConstantInt &Val) { 280 LLT Ty = Res.getLLTTy(*getMRI()); 281 LLT EltTy = Ty.getScalarType(); 282 assert(EltTy.getScalarSizeInBits() == Val.getBitWidth() && 283 "creating constant with the wrong size"); 284 285 if (Ty.isVector()) { 286 auto Const = buildInstr(TargetOpcode::G_CONSTANT) 287 .addDef(getMRI()->createGenericVirtualRegister(EltTy)) 288 .addCImm(&Val); 289 return buildSplatVector(Res, Const); 290 } 291 292 auto Const = buildInstr(TargetOpcode::G_CONSTANT); 293 Res.addDefToMIB(*getMRI(), Const); 294 Const.addCImm(&Val); 295 return Const; 296 } 297 298 MachineInstrBuilder MachineIRBuilder::buildConstant(const DstOp &Res, 299 int64_t Val) { 300 auto IntN = IntegerType::get(getMF().getFunction().getContext(), 301 Res.getLLTTy(*getMRI()).getScalarSizeInBits()); 302 ConstantInt *CI = ConstantInt::get(IntN, Val, true); 303 return buildConstant(Res, *CI); 304 } 305 306 MachineInstrBuilder MachineIRBuilder::buildFConstant(const DstOp &Res, 307 const ConstantFP &Val) { 308 LLT Ty = Res.getLLTTy(*getMRI()); 309 LLT EltTy = Ty.getScalarType(); 310 311 assert(APFloat::getSizeInBits(Val.getValueAPF().getSemantics()) 312 == EltTy.getSizeInBits() && 313 "creating fconstant with the wrong size"); 314 315 assert(!Ty.isPointer() && "invalid operand type"); 316 317 if (Ty.isVector()) { 318 auto Const = buildInstr(TargetOpcode::G_FCONSTANT) 319 .addDef(getMRI()->createGenericVirtualRegister(EltTy)) 320 .addFPImm(&Val); 321 322 return buildSplatVector(Res, Const); 323 } 324 325 auto Const = buildInstr(TargetOpcode::G_FCONSTANT); 326 Res.addDefToMIB(*getMRI(), Const); 327 Const.addFPImm(&Val); 328 return Const; 329 } 330 331 MachineInstrBuilder MachineIRBuilder::buildConstant(const DstOp &Res, 332 const APInt &Val) { 333 ConstantInt *CI = ConstantInt::get(getMF().getFunction().getContext(), Val); 334 return buildConstant(Res, *CI); 335 } 336 337 MachineInstrBuilder MachineIRBuilder::buildFConstant(const DstOp &Res, 338 double Val) { 339 LLT DstTy = Res.getLLTTy(*getMRI()); 340 auto &Ctx = getMF().getFunction().getContext(); 341 auto *CFP = 342 ConstantFP::get(Ctx, getAPFloatFromSize(Val, DstTy.getScalarSizeInBits())); 343 return buildFConstant(Res, *CFP); 344 } 345 346 MachineInstrBuilder MachineIRBuilder::buildFConstant(const DstOp &Res, 347 const APFloat &Val) { 348 auto &Ctx = getMF().getFunction().getContext(); 349 auto *CFP = ConstantFP::get(Ctx, Val); 350 return buildFConstant(Res, *CFP); 351 } 352 353 MachineInstrBuilder MachineIRBuilder::buildBrCond(Register Tst, 354 MachineBasicBlock &Dest) { 355 assert(getMRI()->getType(Tst).isScalar() && "invalid operand type"); 356 357 return buildInstr(TargetOpcode::G_BRCOND).addUse(Tst).addMBB(&Dest); 358 } 359 360 MachineInstrBuilder MachineIRBuilder::buildLoad(const DstOp &Res, 361 const SrcOp &Addr, 362 MachineMemOperand &MMO) { 363 return buildLoadInstr(TargetOpcode::G_LOAD, Res, Addr, MMO); 364 } 365 366 MachineInstrBuilder MachineIRBuilder::buildLoadInstr(unsigned Opcode, 367 const DstOp &Res, 368 const SrcOp &Addr, 369 MachineMemOperand &MMO) { 370 assert(Res.getLLTTy(*getMRI()).isValid() && "invalid operand type"); 371 assert(Addr.getLLTTy(*getMRI()).isPointer() && "invalid operand type"); 372 373 auto MIB = buildInstr(Opcode); 374 Res.addDefToMIB(*getMRI(), MIB); 375 Addr.addSrcToMIB(MIB); 376 MIB.addMemOperand(&MMO); 377 return MIB; 378 } 379 380 MachineInstrBuilder MachineIRBuilder::buildStore(const SrcOp &Val, 381 const SrcOp &Addr, 382 MachineMemOperand &MMO) { 383 assert(Val.getLLTTy(*getMRI()).isValid() && "invalid operand type"); 384 assert(Addr.getLLTTy(*getMRI()).isPointer() && "invalid operand type"); 385 386 auto MIB = buildInstr(TargetOpcode::G_STORE); 387 Val.addSrcToMIB(MIB); 388 Addr.addSrcToMIB(MIB); 389 MIB.addMemOperand(&MMO); 390 return MIB; 391 } 392 393 MachineInstrBuilder MachineIRBuilder::buildUAddo(const DstOp &Res, 394 const DstOp &CarryOut, 395 const SrcOp &Op0, 396 const SrcOp &Op1) { 397 return buildInstr(TargetOpcode::G_UADDO, {Res, CarryOut}, {Op0, Op1}); 398 } 399 400 MachineInstrBuilder MachineIRBuilder::buildUAdde(const DstOp &Res, 401 const DstOp &CarryOut, 402 const SrcOp &Op0, 403 const SrcOp &Op1, 404 const SrcOp &CarryIn) { 405 return buildInstr(TargetOpcode::G_UADDE, {Res, CarryOut}, 406 {Op0, Op1, CarryIn}); 407 } 408 409 MachineInstrBuilder MachineIRBuilder::buildAnyExt(const DstOp &Res, 410 const SrcOp &Op) { 411 return buildInstr(TargetOpcode::G_ANYEXT, Res, Op); 412 } 413 414 MachineInstrBuilder MachineIRBuilder::buildSExt(const DstOp &Res, 415 const SrcOp &Op) { 416 return buildInstr(TargetOpcode::G_SEXT, Res, Op); 417 } 418 419 MachineInstrBuilder MachineIRBuilder::buildZExt(const DstOp &Res, 420 const SrcOp &Op) { 421 return buildInstr(TargetOpcode::G_ZEXT, Res, Op); 422 } 423 424 unsigned MachineIRBuilder::getBoolExtOp(bool IsVec, bool IsFP) const { 425 const auto *TLI = getMF().getSubtarget().getTargetLowering(); 426 switch (TLI->getBooleanContents(IsVec, IsFP)) { 427 case TargetLoweringBase::ZeroOrNegativeOneBooleanContent: 428 return TargetOpcode::G_SEXT; 429 case TargetLoweringBase::ZeroOrOneBooleanContent: 430 return TargetOpcode::G_ZEXT; 431 default: 432 return TargetOpcode::G_ANYEXT; 433 } 434 } 435 436 MachineInstrBuilder MachineIRBuilder::buildBoolExt(const DstOp &Res, 437 const SrcOp &Op, 438 bool IsFP) { 439 unsigned ExtOp = getBoolExtOp(getMRI()->getType(Op.getReg()).isVector(), IsFP); 440 return buildInstr(ExtOp, Res, Op); 441 } 442 443 MachineInstrBuilder MachineIRBuilder::buildExtOrTrunc(unsigned ExtOpc, 444 const DstOp &Res, 445 const SrcOp &Op) { 446 assert((TargetOpcode::G_ANYEXT == ExtOpc || TargetOpcode::G_ZEXT == ExtOpc || 447 TargetOpcode::G_SEXT == ExtOpc) && 448 "Expecting Extending Opc"); 449 assert(Res.getLLTTy(*getMRI()).isScalar() || 450 Res.getLLTTy(*getMRI()).isVector()); 451 assert(Res.getLLTTy(*getMRI()).isScalar() == 452 Op.getLLTTy(*getMRI()).isScalar()); 453 454 unsigned Opcode = TargetOpcode::COPY; 455 if (Res.getLLTTy(*getMRI()).getSizeInBits() > 456 Op.getLLTTy(*getMRI()).getSizeInBits()) 457 Opcode = ExtOpc; 458 else if (Res.getLLTTy(*getMRI()).getSizeInBits() < 459 Op.getLLTTy(*getMRI()).getSizeInBits()) 460 Opcode = TargetOpcode::G_TRUNC; 461 else 462 assert(Res.getLLTTy(*getMRI()) == Op.getLLTTy(*getMRI())); 463 464 return buildInstr(Opcode, Res, Op); 465 } 466 467 MachineInstrBuilder MachineIRBuilder::buildSExtOrTrunc(const DstOp &Res, 468 const SrcOp &Op) { 469 return buildExtOrTrunc(TargetOpcode::G_SEXT, Res, Op); 470 } 471 472 MachineInstrBuilder MachineIRBuilder::buildZExtOrTrunc(const DstOp &Res, 473 const SrcOp &Op) { 474 return buildExtOrTrunc(TargetOpcode::G_ZEXT, Res, Op); 475 } 476 477 MachineInstrBuilder MachineIRBuilder::buildAnyExtOrTrunc(const DstOp &Res, 478 const SrcOp &Op) { 479 return buildExtOrTrunc(TargetOpcode::G_ANYEXT, Res, Op); 480 } 481 482 MachineInstrBuilder MachineIRBuilder::buildCast(const DstOp &Dst, 483 const SrcOp &Src) { 484 LLT SrcTy = Src.getLLTTy(*getMRI()); 485 LLT DstTy = Dst.getLLTTy(*getMRI()); 486 if (SrcTy == DstTy) 487 return buildCopy(Dst, Src); 488 489 unsigned Opcode; 490 if (SrcTy.isPointer() && DstTy.isScalar()) 491 Opcode = TargetOpcode::G_PTRTOINT; 492 else if (DstTy.isPointer() && SrcTy.isScalar()) 493 Opcode = TargetOpcode::G_INTTOPTR; 494 else { 495 assert(!SrcTy.isPointer() && !DstTy.isPointer() && "n G_ADDRCAST yet"); 496 Opcode = TargetOpcode::G_BITCAST; 497 } 498 499 return buildInstr(Opcode, Dst, Src); 500 } 501 502 MachineInstrBuilder MachineIRBuilder::buildExtract(const DstOp &Dst, 503 const SrcOp &Src, 504 uint64_t Index) { 505 LLT SrcTy = Src.getLLTTy(*getMRI()); 506 LLT DstTy = Dst.getLLTTy(*getMRI()); 507 508 #ifndef NDEBUG 509 assert(SrcTy.isValid() && "invalid operand type"); 510 assert(DstTy.isValid() && "invalid operand type"); 511 assert(Index + DstTy.getSizeInBits() <= SrcTy.getSizeInBits() && 512 "extracting off end of register"); 513 #endif 514 515 if (DstTy.getSizeInBits() == SrcTy.getSizeInBits()) { 516 assert(Index == 0 && "insertion past the end of a register"); 517 return buildCast(Dst, Src); 518 } 519 520 auto Extract = buildInstr(TargetOpcode::G_EXTRACT); 521 Dst.addDefToMIB(*getMRI(), Extract); 522 Src.addSrcToMIB(Extract); 523 Extract.addImm(Index); 524 return Extract; 525 } 526 527 void MachineIRBuilder::buildSequence(Register Res, ArrayRef<Register> Ops, 528 ArrayRef<uint64_t> Indices) { 529 #ifndef NDEBUG 530 assert(Ops.size() == Indices.size() && "incompatible args"); 531 assert(!Ops.empty() && "invalid trivial sequence"); 532 assert(std::is_sorted(Indices.begin(), Indices.end()) && 533 "sequence offsets must be in ascending order"); 534 535 assert(getMRI()->getType(Res).isValid() && "invalid operand type"); 536 for (auto Op : Ops) 537 assert(getMRI()->getType(Op).isValid() && "invalid operand type"); 538 #endif 539 540 LLT ResTy = getMRI()->getType(Res); 541 LLT OpTy = getMRI()->getType(Ops[0]); 542 unsigned OpSize = OpTy.getSizeInBits(); 543 bool MaybeMerge = true; 544 for (unsigned i = 0; i < Ops.size(); ++i) { 545 if (getMRI()->getType(Ops[i]) != OpTy || Indices[i] != i * OpSize) { 546 MaybeMerge = false; 547 break; 548 } 549 } 550 551 if (MaybeMerge && Ops.size() * OpSize == ResTy.getSizeInBits()) { 552 buildMerge(Res, Ops); 553 return; 554 } 555 556 Register ResIn = getMRI()->createGenericVirtualRegister(ResTy); 557 buildUndef(ResIn); 558 559 for (unsigned i = 0; i < Ops.size(); ++i) { 560 Register ResOut = i + 1 == Ops.size() 561 ? Res 562 : getMRI()->createGenericVirtualRegister(ResTy); 563 buildInsert(ResOut, ResIn, Ops[i], Indices[i]); 564 ResIn = ResOut; 565 } 566 } 567 568 MachineInstrBuilder MachineIRBuilder::buildUndef(const DstOp &Res) { 569 return buildInstr(TargetOpcode::G_IMPLICIT_DEF, {Res}, {}); 570 } 571 572 MachineInstrBuilder MachineIRBuilder::buildMerge(const DstOp &Res, 573 ArrayRef<Register> Ops) { 574 // Unfortunately to convert from ArrayRef<LLT> to ArrayRef<SrcOp>, 575 // we need some temporary storage for the DstOp objects. Here we use a 576 // sufficiently large SmallVector to not go through the heap. 577 SmallVector<SrcOp, 8> TmpVec(Ops.begin(), Ops.end()); 578 assert(TmpVec.size() > 1); 579 return buildInstr(TargetOpcode::G_MERGE_VALUES, Res, TmpVec); 580 } 581 582 MachineInstrBuilder MachineIRBuilder::buildUnmerge(ArrayRef<LLT> Res, 583 const SrcOp &Op) { 584 // Unfortunately to convert from ArrayRef<LLT> to ArrayRef<DstOp>, 585 // we need some temporary storage for the DstOp objects. Here we use a 586 // sufficiently large SmallVector to not go through the heap. 587 SmallVector<DstOp, 8> TmpVec(Res.begin(), Res.end()); 588 assert(TmpVec.size() > 1); 589 return buildInstr(TargetOpcode::G_UNMERGE_VALUES, TmpVec, Op); 590 } 591 592 MachineInstrBuilder MachineIRBuilder::buildUnmerge(LLT Res, 593 const SrcOp &Op) { 594 unsigned NumReg = Op.getLLTTy(*getMRI()).getSizeInBits() / Res.getSizeInBits(); 595 SmallVector<Register, 8> TmpVec; 596 for (unsigned I = 0; I != NumReg; ++I) 597 TmpVec.push_back(getMRI()->createGenericVirtualRegister(Res)); 598 return buildUnmerge(TmpVec, Op); 599 } 600 601 MachineInstrBuilder MachineIRBuilder::buildUnmerge(ArrayRef<Register> Res, 602 const SrcOp &Op) { 603 // Unfortunately to convert from ArrayRef<Register> to ArrayRef<DstOp>, 604 // we need some temporary storage for the DstOp objects. Here we use a 605 // sufficiently large SmallVector to not go through the heap. 606 SmallVector<DstOp, 8> TmpVec(Res.begin(), Res.end()); 607 assert(TmpVec.size() > 1); 608 return buildInstr(TargetOpcode::G_UNMERGE_VALUES, TmpVec, Op); 609 } 610 611 MachineInstrBuilder MachineIRBuilder::buildBuildVector(const DstOp &Res, 612 ArrayRef<Register> Ops) { 613 // Unfortunately to convert from ArrayRef<Register> to ArrayRef<SrcOp>, 614 // we need some temporary storage for the DstOp objects. Here we use a 615 // sufficiently large SmallVector to not go through the heap. 616 SmallVector<SrcOp, 8> TmpVec(Ops.begin(), Ops.end()); 617 return buildInstr(TargetOpcode::G_BUILD_VECTOR, Res, TmpVec); 618 } 619 620 MachineInstrBuilder MachineIRBuilder::buildSplatVector(const DstOp &Res, 621 const SrcOp &Src) { 622 SmallVector<SrcOp, 8> TmpVec(Res.getLLTTy(*getMRI()).getNumElements(), Src); 623 return buildInstr(TargetOpcode::G_BUILD_VECTOR, Res, TmpVec); 624 } 625 626 MachineInstrBuilder 627 MachineIRBuilder::buildBuildVectorTrunc(const DstOp &Res, 628 ArrayRef<Register> Ops) { 629 // Unfortunately to convert from ArrayRef<Register> to ArrayRef<SrcOp>, 630 // we need some temporary storage for the DstOp objects. Here we use a 631 // sufficiently large SmallVector to not go through the heap. 632 SmallVector<SrcOp, 8> TmpVec(Ops.begin(), Ops.end()); 633 return buildInstr(TargetOpcode::G_BUILD_VECTOR_TRUNC, Res, TmpVec); 634 } 635 636 MachineInstrBuilder 637 MachineIRBuilder::buildConcatVectors(const DstOp &Res, ArrayRef<Register> Ops) { 638 // Unfortunately to convert from ArrayRef<Register> to ArrayRef<SrcOp>, 639 // we need some temporary storage for the DstOp objects. Here we use a 640 // sufficiently large SmallVector to not go through the heap. 641 SmallVector<SrcOp, 8> TmpVec(Ops.begin(), Ops.end()); 642 return buildInstr(TargetOpcode::G_CONCAT_VECTORS, Res, TmpVec); 643 } 644 645 MachineInstrBuilder MachineIRBuilder::buildInsert(const DstOp &Res, 646 const SrcOp &Src, 647 const SrcOp &Op, 648 unsigned Index) { 649 assert(Index + Op.getLLTTy(*getMRI()).getSizeInBits() <= 650 Res.getLLTTy(*getMRI()).getSizeInBits() && 651 "insertion past the end of a register"); 652 653 if (Res.getLLTTy(*getMRI()).getSizeInBits() == 654 Op.getLLTTy(*getMRI()).getSizeInBits()) { 655 return buildCast(Res, Op); 656 } 657 658 return buildInstr(TargetOpcode::G_INSERT, Res, {Src, Op, uint64_t(Index)}); 659 } 660 661 MachineInstrBuilder MachineIRBuilder::buildIntrinsic(Intrinsic::ID ID, 662 ArrayRef<Register> ResultRegs, 663 bool HasSideEffects) { 664 auto MIB = 665 buildInstr(HasSideEffects ? TargetOpcode::G_INTRINSIC_W_SIDE_EFFECTS 666 : TargetOpcode::G_INTRINSIC); 667 for (unsigned ResultReg : ResultRegs) 668 MIB.addDef(ResultReg); 669 MIB.addIntrinsicID(ID); 670 return MIB; 671 } 672 673 MachineInstrBuilder MachineIRBuilder::buildIntrinsic(Intrinsic::ID ID, 674 ArrayRef<DstOp> Results, 675 bool HasSideEffects) { 676 auto MIB = 677 buildInstr(HasSideEffects ? TargetOpcode::G_INTRINSIC_W_SIDE_EFFECTS 678 : TargetOpcode::G_INTRINSIC); 679 for (DstOp Result : Results) 680 Result.addDefToMIB(*getMRI(), MIB); 681 MIB.addIntrinsicID(ID); 682 return MIB; 683 } 684 685 MachineInstrBuilder MachineIRBuilder::buildTrunc(const DstOp &Res, 686 const SrcOp &Op) { 687 return buildInstr(TargetOpcode::G_TRUNC, Res, Op); 688 } 689 690 MachineInstrBuilder MachineIRBuilder::buildFPTrunc(const DstOp &Res, 691 const SrcOp &Op, 692 Optional<unsigned> Flags) { 693 return buildInstr(TargetOpcode::G_FPTRUNC, Res, Op, Flags); 694 } 695 696 MachineInstrBuilder MachineIRBuilder::buildICmp(CmpInst::Predicate Pred, 697 const DstOp &Res, 698 const SrcOp &Op0, 699 const SrcOp &Op1) { 700 return buildInstr(TargetOpcode::G_ICMP, Res, {Pred, Op0, Op1}); 701 } 702 703 MachineInstrBuilder MachineIRBuilder::buildFCmp(CmpInst::Predicate Pred, 704 const DstOp &Res, 705 const SrcOp &Op0, 706 const SrcOp &Op1, 707 Optional<unsigned> Flags) { 708 709 return buildInstr(TargetOpcode::G_FCMP, Res, {Pred, Op0, Op1}, Flags); 710 } 711 712 MachineInstrBuilder MachineIRBuilder::buildSelect(const DstOp &Res, 713 const SrcOp &Tst, 714 const SrcOp &Op0, 715 const SrcOp &Op1, 716 Optional<unsigned> Flags) { 717 718 return buildInstr(TargetOpcode::G_SELECT, {Res}, {Tst, Op0, Op1}, Flags); 719 } 720 721 MachineInstrBuilder 722 MachineIRBuilder::buildInsertVectorElement(const DstOp &Res, const SrcOp &Val, 723 const SrcOp &Elt, const SrcOp &Idx) { 724 return buildInstr(TargetOpcode::G_INSERT_VECTOR_ELT, Res, {Val, Elt, Idx}); 725 } 726 727 MachineInstrBuilder 728 MachineIRBuilder::buildExtractVectorElement(const DstOp &Res, const SrcOp &Val, 729 const SrcOp &Idx) { 730 return buildInstr(TargetOpcode::G_EXTRACT_VECTOR_ELT, Res, {Val, Idx}); 731 } 732 733 MachineInstrBuilder MachineIRBuilder::buildAtomicCmpXchgWithSuccess( 734 Register OldValRes, Register SuccessRes, Register Addr, Register CmpVal, 735 Register NewVal, MachineMemOperand &MMO) { 736 #ifndef NDEBUG 737 LLT OldValResTy = getMRI()->getType(OldValRes); 738 LLT SuccessResTy = getMRI()->getType(SuccessRes); 739 LLT AddrTy = getMRI()->getType(Addr); 740 LLT CmpValTy = getMRI()->getType(CmpVal); 741 LLT NewValTy = getMRI()->getType(NewVal); 742 assert(OldValResTy.isScalar() && "invalid operand type"); 743 assert(SuccessResTy.isScalar() && "invalid operand type"); 744 assert(AddrTy.isPointer() && "invalid operand type"); 745 assert(CmpValTy.isValid() && "invalid operand type"); 746 assert(NewValTy.isValid() && "invalid operand type"); 747 assert(OldValResTy == CmpValTy && "type mismatch"); 748 assert(OldValResTy == NewValTy && "type mismatch"); 749 #endif 750 751 return buildInstr(TargetOpcode::G_ATOMIC_CMPXCHG_WITH_SUCCESS) 752 .addDef(OldValRes) 753 .addDef(SuccessRes) 754 .addUse(Addr) 755 .addUse(CmpVal) 756 .addUse(NewVal) 757 .addMemOperand(&MMO); 758 } 759 760 MachineInstrBuilder 761 MachineIRBuilder::buildAtomicCmpXchg(Register OldValRes, Register Addr, 762 Register CmpVal, Register NewVal, 763 MachineMemOperand &MMO) { 764 #ifndef NDEBUG 765 LLT OldValResTy = getMRI()->getType(OldValRes); 766 LLT AddrTy = getMRI()->getType(Addr); 767 LLT CmpValTy = getMRI()->getType(CmpVal); 768 LLT NewValTy = getMRI()->getType(NewVal); 769 assert(OldValResTy.isScalar() && "invalid operand type"); 770 assert(AddrTy.isPointer() && "invalid operand type"); 771 assert(CmpValTy.isValid() && "invalid operand type"); 772 assert(NewValTy.isValid() && "invalid operand type"); 773 assert(OldValResTy == CmpValTy && "type mismatch"); 774 assert(OldValResTy == NewValTy && "type mismatch"); 775 #endif 776 777 return buildInstr(TargetOpcode::G_ATOMIC_CMPXCHG) 778 .addDef(OldValRes) 779 .addUse(Addr) 780 .addUse(CmpVal) 781 .addUse(NewVal) 782 .addMemOperand(&MMO); 783 } 784 785 MachineInstrBuilder MachineIRBuilder::buildAtomicRMW( 786 unsigned Opcode, const DstOp &OldValRes, 787 const SrcOp &Addr, const SrcOp &Val, 788 MachineMemOperand &MMO) { 789 790 #ifndef NDEBUG 791 LLT OldValResTy = OldValRes.getLLTTy(*getMRI()); 792 LLT AddrTy = Addr.getLLTTy(*getMRI()); 793 LLT ValTy = Val.getLLTTy(*getMRI()); 794 assert(OldValResTy.isScalar() && "invalid operand type"); 795 assert(AddrTy.isPointer() && "invalid operand type"); 796 assert(ValTy.isValid() && "invalid operand type"); 797 assert(OldValResTy == ValTy && "type mismatch"); 798 assert(MMO.isAtomic() && "not atomic mem operand"); 799 #endif 800 801 auto MIB = buildInstr(Opcode); 802 OldValRes.addDefToMIB(*getMRI(), MIB); 803 Addr.addSrcToMIB(MIB); 804 Val.addSrcToMIB(MIB); 805 MIB.addMemOperand(&MMO); 806 return MIB; 807 } 808 809 MachineInstrBuilder 810 MachineIRBuilder::buildAtomicRMWXchg(Register OldValRes, Register Addr, 811 Register Val, MachineMemOperand &MMO) { 812 return buildAtomicRMW(TargetOpcode::G_ATOMICRMW_XCHG, OldValRes, Addr, Val, 813 MMO); 814 } 815 MachineInstrBuilder 816 MachineIRBuilder::buildAtomicRMWAdd(Register OldValRes, Register Addr, 817 Register Val, MachineMemOperand &MMO) { 818 return buildAtomicRMW(TargetOpcode::G_ATOMICRMW_ADD, OldValRes, Addr, Val, 819 MMO); 820 } 821 MachineInstrBuilder 822 MachineIRBuilder::buildAtomicRMWSub(Register OldValRes, Register Addr, 823 Register Val, MachineMemOperand &MMO) { 824 return buildAtomicRMW(TargetOpcode::G_ATOMICRMW_SUB, OldValRes, Addr, Val, 825 MMO); 826 } 827 MachineInstrBuilder 828 MachineIRBuilder::buildAtomicRMWAnd(Register OldValRes, Register Addr, 829 Register Val, MachineMemOperand &MMO) { 830 return buildAtomicRMW(TargetOpcode::G_ATOMICRMW_AND, OldValRes, Addr, Val, 831 MMO); 832 } 833 MachineInstrBuilder 834 MachineIRBuilder::buildAtomicRMWNand(Register OldValRes, Register Addr, 835 Register Val, MachineMemOperand &MMO) { 836 return buildAtomicRMW(TargetOpcode::G_ATOMICRMW_NAND, OldValRes, Addr, Val, 837 MMO); 838 } 839 MachineInstrBuilder MachineIRBuilder::buildAtomicRMWOr(Register OldValRes, 840 Register Addr, 841 Register Val, 842 MachineMemOperand &MMO) { 843 return buildAtomicRMW(TargetOpcode::G_ATOMICRMW_OR, OldValRes, Addr, Val, 844 MMO); 845 } 846 MachineInstrBuilder 847 MachineIRBuilder::buildAtomicRMWXor(Register OldValRes, Register Addr, 848 Register Val, MachineMemOperand &MMO) { 849 return buildAtomicRMW(TargetOpcode::G_ATOMICRMW_XOR, OldValRes, Addr, Val, 850 MMO); 851 } 852 MachineInstrBuilder 853 MachineIRBuilder::buildAtomicRMWMax(Register OldValRes, Register Addr, 854 Register Val, MachineMemOperand &MMO) { 855 return buildAtomicRMW(TargetOpcode::G_ATOMICRMW_MAX, OldValRes, Addr, Val, 856 MMO); 857 } 858 MachineInstrBuilder 859 MachineIRBuilder::buildAtomicRMWMin(Register OldValRes, Register Addr, 860 Register Val, MachineMemOperand &MMO) { 861 return buildAtomicRMW(TargetOpcode::G_ATOMICRMW_MIN, OldValRes, Addr, Val, 862 MMO); 863 } 864 MachineInstrBuilder 865 MachineIRBuilder::buildAtomicRMWUmax(Register OldValRes, Register Addr, 866 Register Val, MachineMemOperand &MMO) { 867 return buildAtomicRMW(TargetOpcode::G_ATOMICRMW_UMAX, OldValRes, Addr, Val, 868 MMO); 869 } 870 MachineInstrBuilder 871 MachineIRBuilder::buildAtomicRMWUmin(Register OldValRes, Register Addr, 872 Register Val, MachineMemOperand &MMO) { 873 return buildAtomicRMW(TargetOpcode::G_ATOMICRMW_UMIN, OldValRes, Addr, Val, 874 MMO); 875 } 876 877 MachineInstrBuilder 878 MachineIRBuilder::buildAtomicRMWFAdd( 879 const DstOp &OldValRes, const SrcOp &Addr, const SrcOp &Val, 880 MachineMemOperand &MMO) { 881 return buildAtomicRMW(TargetOpcode::G_ATOMICRMW_FADD, OldValRes, Addr, Val, 882 MMO); 883 } 884 885 MachineInstrBuilder 886 MachineIRBuilder::buildAtomicRMWFSub(const DstOp &OldValRes, const SrcOp &Addr, const SrcOp &Val, 887 MachineMemOperand &MMO) { 888 return buildAtomicRMW(TargetOpcode::G_ATOMICRMW_FSUB, OldValRes, Addr, Val, 889 MMO); 890 } 891 892 MachineInstrBuilder 893 MachineIRBuilder::buildFence(unsigned Ordering, unsigned Scope) { 894 return buildInstr(TargetOpcode::G_FENCE) 895 .addImm(Ordering) 896 .addImm(Scope); 897 } 898 899 MachineInstrBuilder 900 MachineIRBuilder::buildBlockAddress(Register Res, const BlockAddress *BA) { 901 #ifndef NDEBUG 902 assert(getMRI()->getType(Res).isPointer() && "invalid res type"); 903 #endif 904 905 return buildInstr(TargetOpcode::G_BLOCK_ADDR).addDef(Res).addBlockAddress(BA); 906 } 907 908 void MachineIRBuilder::validateTruncExt(const LLT &DstTy, const LLT &SrcTy, 909 bool IsExtend) { 910 #ifndef NDEBUG 911 if (DstTy.isVector()) { 912 assert(SrcTy.isVector() && "mismatched cast between vector and non-vector"); 913 assert(SrcTy.getNumElements() == DstTy.getNumElements() && 914 "different number of elements in a trunc/ext"); 915 } else 916 assert(DstTy.isScalar() && SrcTy.isScalar() && "invalid extend/trunc"); 917 918 if (IsExtend) 919 assert(DstTy.getSizeInBits() > SrcTy.getSizeInBits() && 920 "invalid narrowing extend"); 921 else 922 assert(DstTy.getSizeInBits() < SrcTy.getSizeInBits() && 923 "invalid widening trunc"); 924 #endif 925 } 926 927 void MachineIRBuilder::validateSelectOp(const LLT &ResTy, const LLT &TstTy, 928 const LLT &Op0Ty, const LLT &Op1Ty) { 929 #ifndef NDEBUG 930 assert((ResTy.isScalar() || ResTy.isVector() || ResTy.isPointer()) && 931 "invalid operand type"); 932 assert((ResTy == Op0Ty && ResTy == Op1Ty) && "type mismatch"); 933 if (ResTy.isScalar() || ResTy.isPointer()) 934 assert(TstTy.isScalar() && "type mismatch"); 935 else 936 assert((TstTy.isScalar() || 937 (TstTy.isVector() && 938 TstTy.getNumElements() == Op0Ty.getNumElements())) && 939 "type mismatch"); 940 #endif 941 } 942 943 MachineInstrBuilder MachineIRBuilder::buildInstr(unsigned Opc, 944 ArrayRef<DstOp> DstOps, 945 ArrayRef<SrcOp> SrcOps, 946 Optional<unsigned> Flags) { 947 switch (Opc) { 948 default: 949 break; 950 case TargetOpcode::G_SELECT: { 951 assert(DstOps.size() == 1 && "Invalid select"); 952 assert(SrcOps.size() == 3 && "Invalid select"); 953 validateSelectOp( 954 DstOps[0].getLLTTy(*getMRI()), SrcOps[0].getLLTTy(*getMRI()), 955 SrcOps[1].getLLTTy(*getMRI()), SrcOps[2].getLLTTy(*getMRI())); 956 break; 957 } 958 case TargetOpcode::G_ADD: 959 case TargetOpcode::G_AND: 960 case TargetOpcode::G_MUL: 961 case TargetOpcode::G_OR: 962 case TargetOpcode::G_SUB: 963 case TargetOpcode::G_XOR: 964 case TargetOpcode::G_UDIV: 965 case TargetOpcode::G_SDIV: 966 case TargetOpcode::G_UREM: 967 case TargetOpcode::G_SREM: 968 case TargetOpcode::G_SMIN: 969 case TargetOpcode::G_SMAX: 970 case TargetOpcode::G_UMIN: 971 case TargetOpcode::G_UMAX: { 972 // All these are binary ops. 973 assert(DstOps.size() == 1 && "Invalid Dst"); 974 assert(SrcOps.size() == 2 && "Invalid Srcs"); 975 validateBinaryOp(DstOps[0].getLLTTy(*getMRI()), 976 SrcOps[0].getLLTTy(*getMRI()), 977 SrcOps[1].getLLTTy(*getMRI())); 978 break; 979 } 980 case TargetOpcode::G_SHL: 981 case TargetOpcode::G_ASHR: 982 case TargetOpcode::G_LSHR: { 983 assert(DstOps.size() == 1 && "Invalid Dst"); 984 assert(SrcOps.size() == 2 && "Invalid Srcs"); 985 validateShiftOp(DstOps[0].getLLTTy(*getMRI()), 986 SrcOps[0].getLLTTy(*getMRI()), 987 SrcOps[1].getLLTTy(*getMRI())); 988 break; 989 } 990 case TargetOpcode::G_SEXT: 991 case TargetOpcode::G_ZEXT: 992 case TargetOpcode::G_ANYEXT: 993 assert(DstOps.size() == 1 && "Invalid Dst"); 994 assert(SrcOps.size() == 1 && "Invalid Srcs"); 995 validateTruncExt(DstOps[0].getLLTTy(*getMRI()), 996 SrcOps[0].getLLTTy(*getMRI()), true); 997 break; 998 case TargetOpcode::G_TRUNC: 999 case TargetOpcode::G_FPTRUNC: { 1000 assert(DstOps.size() == 1 && "Invalid Dst"); 1001 assert(SrcOps.size() == 1 && "Invalid Srcs"); 1002 validateTruncExt(DstOps[0].getLLTTy(*getMRI()), 1003 SrcOps[0].getLLTTy(*getMRI()), false); 1004 break; 1005 } 1006 case TargetOpcode::G_BITCAST: { 1007 assert(DstOps.size() == 1 && "Invalid Dst"); 1008 assert(SrcOps.size() == 1 && "Invalid Srcs"); 1009 assert(DstOps[0].getLLTTy(*getMRI()).getSizeInBits() == 1010 SrcOps[0].getLLTTy(*getMRI()).getSizeInBits() && "invalid bitcast"); 1011 break; 1012 } 1013 case TargetOpcode::COPY: 1014 assert(DstOps.size() == 1 && "Invalid Dst"); 1015 // If the caller wants to add a subreg source it has to be done separately 1016 // so we may not have any SrcOps at this point yet. 1017 break; 1018 case TargetOpcode::G_FCMP: 1019 case TargetOpcode::G_ICMP: { 1020 assert(DstOps.size() == 1 && "Invalid Dst Operands"); 1021 assert(SrcOps.size() == 3 && "Invalid Src Operands"); 1022 // For F/ICMP, the first src operand is the predicate, followed by 1023 // the two comparands. 1024 assert(SrcOps[0].getSrcOpKind() == SrcOp::SrcType::Ty_Predicate && 1025 "Expecting predicate"); 1026 assert([&]() -> bool { 1027 CmpInst::Predicate Pred = SrcOps[0].getPredicate(); 1028 return Opc == TargetOpcode::G_ICMP ? CmpInst::isIntPredicate(Pred) 1029 : CmpInst::isFPPredicate(Pred); 1030 }() && "Invalid predicate"); 1031 assert(SrcOps[1].getLLTTy(*getMRI()) == SrcOps[2].getLLTTy(*getMRI()) && 1032 "Type mismatch"); 1033 assert([&]() -> bool { 1034 LLT Op0Ty = SrcOps[1].getLLTTy(*getMRI()); 1035 LLT DstTy = DstOps[0].getLLTTy(*getMRI()); 1036 if (Op0Ty.isScalar() || Op0Ty.isPointer()) 1037 return DstTy.isScalar(); 1038 else 1039 return DstTy.isVector() && 1040 DstTy.getNumElements() == Op0Ty.getNumElements(); 1041 }() && "Type Mismatch"); 1042 break; 1043 } 1044 case TargetOpcode::G_UNMERGE_VALUES: { 1045 assert(!DstOps.empty() && "Invalid trivial sequence"); 1046 assert(SrcOps.size() == 1 && "Invalid src for Unmerge"); 1047 assert(std::all_of(DstOps.begin(), DstOps.end(), 1048 [&, this](const DstOp &Op) { 1049 return Op.getLLTTy(*getMRI()) == 1050 DstOps[0].getLLTTy(*getMRI()); 1051 }) && 1052 "type mismatch in output list"); 1053 assert(DstOps.size() * DstOps[0].getLLTTy(*getMRI()).getSizeInBits() == 1054 SrcOps[0].getLLTTy(*getMRI()).getSizeInBits() && 1055 "input operands do not cover output register"); 1056 break; 1057 } 1058 case TargetOpcode::G_MERGE_VALUES: { 1059 assert(!SrcOps.empty() && "invalid trivial sequence"); 1060 assert(DstOps.size() == 1 && "Invalid Dst"); 1061 assert(std::all_of(SrcOps.begin(), SrcOps.end(), 1062 [&, this](const SrcOp &Op) { 1063 return Op.getLLTTy(*getMRI()) == 1064 SrcOps[0].getLLTTy(*getMRI()); 1065 }) && 1066 "type mismatch in input list"); 1067 assert(SrcOps.size() * SrcOps[0].getLLTTy(*getMRI()).getSizeInBits() == 1068 DstOps[0].getLLTTy(*getMRI()).getSizeInBits() && 1069 "input operands do not cover output register"); 1070 if (SrcOps.size() == 1) 1071 return buildCast(DstOps[0], SrcOps[0]); 1072 if (DstOps[0].getLLTTy(*getMRI()).isVector()) { 1073 if (SrcOps[0].getLLTTy(*getMRI()).isVector()) 1074 return buildInstr(TargetOpcode::G_CONCAT_VECTORS, DstOps, SrcOps); 1075 return buildInstr(TargetOpcode::G_BUILD_VECTOR, DstOps, SrcOps); 1076 } 1077 break; 1078 } 1079 case TargetOpcode::G_EXTRACT_VECTOR_ELT: { 1080 assert(DstOps.size() == 1 && "Invalid Dst size"); 1081 assert(SrcOps.size() == 2 && "Invalid Src size"); 1082 assert(SrcOps[0].getLLTTy(*getMRI()).isVector() && "Invalid operand type"); 1083 assert((DstOps[0].getLLTTy(*getMRI()).isScalar() || 1084 DstOps[0].getLLTTy(*getMRI()).isPointer()) && 1085 "Invalid operand type"); 1086 assert(SrcOps[1].getLLTTy(*getMRI()).isScalar() && "Invalid operand type"); 1087 assert(SrcOps[0].getLLTTy(*getMRI()).getElementType() == 1088 DstOps[0].getLLTTy(*getMRI()) && 1089 "Type mismatch"); 1090 break; 1091 } 1092 case TargetOpcode::G_INSERT_VECTOR_ELT: { 1093 assert(DstOps.size() == 1 && "Invalid dst size"); 1094 assert(SrcOps.size() == 3 && "Invalid src size"); 1095 assert(DstOps[0].getLLTTy(*getMRI()).isVector() && 1096 SrcOps[0].getLLTTy(*getMRI()).isVector() && "Invalid operand type"); 1097 assert(DstOps[0].getLLTTy(*getMRI()).getElementType() == 1098 SrcOps[1].getLLTTy(*getMRI()) && 1099 "Type mismatch"); 1100 assert(SrcOps[2].getLLTTy(*getMRI()).isScalar() && "Invalid index"); 1101 assert(DstOps[0].getLLTTy(*getMRI()).getNumElements() == 1102 SrcOps[0].getLLTTy(*getMRI()).getNumElements() && 1103 "Type mismatch"); 1104 break; 1105 } 1106 case TargetOpcode::G_BUILD_VECTOR: { 1107 assert((!SrcOps.empty() || SrcOps.size() < 2) && 1108 "Must have at least 2 operands"); 1109 assert(DstOps.size() == 1 && "Invalid DstOps"); 1110 assert(DstOps[0].getLLTTy(*getMRI()).isVector() && 1111 "Res type must be a vector"); 1112 assert(std::all_of(SrcOps.begin(), SrcOps.end(), 1113 [&, this](const SrcOp &Op) { 1114 return Op.getLLTTy(*getMRI()) == 1115 SrcOps[0].getLLTTy(*getMRI()); 1116 }) && 1117 "type mismatch in input list"); 1118 assert(SrcOps.size() * SrcOps[0].getLLTTy(*getMRI()).getSizeInBits() == 1119 DstOps[0].getLLTTy(*getMRI()).getSizeInBits() && 1120 "input scalars do not exactly cover the output vector register"); 1121 break; 1122 } 1123 case TargetOpcode::G_BUILD_VECTOR_TRUNC: { 1124 assert((!SrcOps.empty() || SrcOps.size() < 2) && 1125 "Must have at least 2 operands"); 1126 assert(DstOps.size() == 1 && "Invalid DstOps"); 1127 assert(DstOps[0].getLLTTy(*getMRI()).isVector() && 1128 "Res type must be a vector"); 1129 assert(std::all_of(SrcOps.begin(), SrcOps.end(), 1130 [&, this](const SrcOp &Op) { 1131 return Op.getLLTTy(*getMRI()) == 1132 SrcOps[0].getLLTTy(*getMRI()); 1133 }) && 1134 "type mismatch in input list"); 1135 if (SrcOps[0].getLLTTy(*getMRI()).getSizeInBits() == 1136 DstOps[0].getLLTTy(*getMRI()).getElementType().getSizeInBits()) 1137 return buildInstr(TargetOpcode::G_BUILD_VECTOR, DstOps, SrcOps); 1138 break; 1139 } 1140 case TargetOpcode::G_CONCAT_VECTORS: { 1141 assert(DstOps.size() == 1 && "Invalid DstOps"); 1142 assert((!SrcOps.empty() || SrcOps.size() < 2) && 1143 "Must have at least 2 operands"); 1144 assert(std::all_of(SrcOps.begin(), SrcOps.end(), 1145 [&, this](const SrcOp &Op) { 1146 return (Op.getLLTTy(*getMRI()).isVector() && 1147 Op.getLLTTy(*getMRI()) == 1148 SrcOps[0].getLLTTy(*getMRI())); 1149 }) && 1150 "type mismatch in input list"); 1151 assert(SrcOps.size() * SrcOps[0].getLLTTy(*getMRI()).getSizeInBits() == 1152 DstOps[0].getLLTTy(*getMRI()).getSizeInBits() && 1153 "input vectors do not exactly cover the output vector register"); 1154 break; 1155 } 1156 case TargetOpcode::G_UADDE: { 1157 assert(DstOps.size() == 2 && "Invalid no of dst operands"); 1158 assert(SrcOps.size() == 3 && "Invalid no of src operands"); 1159 assert(DstOps[0].getLLTTy(*getMRI()).isScalar() && "Invalid operand"); 1160 assert((DstOps[0].getLLTTy(*getMRI()) == SrcOps[0].getLLTTy(*getMRI())) && 1161 (DstOps[0].getLLTTy(*getMRI()) == SrcOps[1].getLLTTy(*getMRI())) && 1162 "Invalid operand"); 1163 assert(DstOps[1].getLLTTy(*getMRI()).isScalar() && "Invalid operand"); 1164 assert(DstOps[1].getLLTTy(*getMRI()) == SrcOps[2].getLLTTy(*getMRI()) && 1165 "type mismatch"); 1166 break; 1167 } 1168 } 1169 1170 auto MIB = buildInstr(Opc); 1171 for (const DstOp &Op : DstOps) 1172 Op.addDefToMIB(*getMRI(), MIB); 1173 for (const SrcOp &Op : SrcOps) 1174 Op.addSrcToMIB(MIB); 1175 if (Flags) 1176 MIB->setFlags(*Flags); 1177 return MIB; 1178 } 1179