LLVM 24.0.0git
IRTranslator.cpp
Go to the documentation of this file.
1//===- llvm/CodeGen/GlobalISel/IRTranslator.cpp - IRTranslator ---*- 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 IRTranslator class.
10//===----------------------------------------------------------------------===//
11
14#include "llvm/ADT/STLExtras.h"
15#include "llvm/ADT/ScopeExit.h"
20#include "llvm/Analysis/Loads.h"
55#include "llvm/IR/Analysis.h"
56#include "llvm/IR/BasicBlock.h"
57#include "llvm/IR/CFG.h"
58#include "llvm/IR/Constant.h"
59#include "llvm/IR/Constants.h"
60#include "llvm/IR/DataLayout.h"
63#include "llvm/IR/Function.h"
65#include "llvm/IR/InlineAsm.h"
66#include "llvm/IR/InstrTypes.h"
69#include "llvm/IR/Intrinsics.h"
70#include "llvm/IR/IntrinsicsAMDGPU.h"
71#include "llvm/IR/LLVMContext.h"
72#include "llvm/IR/Metadata.h"
73#include "llvm/IR/Module.h"
75#include "llvm/IR/Statepoint.h"
76#include "llvm/IR/Type.h"
77#include "llvm/IR/User.h"
78#include "llvm/IR/Value.h"
80#include "llvm/MC/MCContext.h"
81#include "llvm/Pass.h"
84#include "llvm/Support/Debug.h"
91#include <algorithm>
92#include <cassert>
93#include <cstdint>
94#include <iterator>
95#include <optional>
96#include <string>
97#include <utility>
98#include <vector>
99
100#define DEBUG_TYPE "ir-translator"
101
102using namespace llvm;
103
104static cl::opt<bool>
105 EnableCSEInIRTranslator("enable-cse-in-ir-translator",
106 cl::desc("Should enable CSE in ir-translator"),
107 cl::init(false));
108
109namespace llvm {
110
112 /// Interface used to lower the everything related to calls.
113 const CallLowering *CLI = nullptr;
114
115 SSPLayoutInfo *SPInfo = nullptr;
116
117 /// This class contains the mapping between the Values to vreg related data.
118 class ValueToVRegInfo {
119 public:
120 ValueToVRegInfo() = default;
121
122 using VRegListT = SmallVector<Register, 1>;
123 using OffsetListT = SmallVector<uint64_t, 1>;
124
125 using const_vreg_iterator =
127 using const_offset_iterator =
129
130 inline const_vreg_iterator vregs_end() const { return ValToVRegs.end(); }
131
132 VRegListT *getVRegs(const Value &V) {
133 auto [It, Inserted] = ValToVRegs.try_emplace(&V);
134 if (!Inserted)
135 return It->second;
136
137 // We placement new using our fast allocator since we never try to free
138 // the vectors until translation is finished.
139 It->second = new (VRegAlloc.Allocate()) VRegListT();
140 return It->second;
141 }
142
143 OffsetListT *getOffsets(const Value &V) {
144 assert(V.getType()->isAggregateType() &&
145 "Offsets are for aggregate values");
146 auto [It, Inserted] = TypeToOffsets.try_emplace(V.getType());
147 if (!Inserted)
148 return It->second;
149
150 It->second = new (OffsetAlloc.Allocate()) OffsetListT();
151 return It->second;
152 }
153
154 const_vreg_iterator findVRegs(const Value &V) const {
155 return ValToVRegs.find(&V);
156 }
157
158 bool contains(const Value &V) const { return ValToVRegs.contains(&V); }
159
160 void reserveVRegs(unsigned NumValues) { ValToVRegs.reserve(NumValues); }
161
162 void reset() {
163 ValToVRegs.clear();
164 TypeToOffsets.clear();
165 VRegAlloc.DestroyAll();
166 OffsetAlloc.DestroyAll();
167 }
168
169 private:
172
173 // We store pointers to vectors here since references may be invalidated
174 // while we hold them if we stored the vectors directly.
177 };
178
179 /// Mapping of the values of the current LLVM IR function to the related
180 /// virtual registers and offsets.
181 ValueToVRegInfo VMap;
182
183 // One BasicBlock can be translated to multiple MachineBasicBlocks. For such
184 // BasicBlocks translated to multiple MachineBasicBlocks, MachinePreds retains
185 // a mapping between the edges arriving at the BasicBlock to the corresponding
186 // created MachineBasicBlocks. Some BasicBlocks that get translated to a
187 // single MachineBasicBlock may also end up in this Map.
188 using CFGEdge = std::pair<const BasicBlock *, const BasicBlock *>;
190
191 // List of stubbed PHI instructions, for values and basic blocks to be filled
192 // in once all MachineBasicBlocks have been created.
194 PendingPHIs;
195
196 /// Record of what frame index has been allocated to specified allocas for
197 /// this function.
199
200 SwiftErrorValueTracking SwiftError;
201
202 /// \name Methods for translating form LLVM IR to MachineInstr.
203 /// \see ::translate for general information on the translate methods.
204 /// @{
205
206 /// Translate \p Inst into its corresponding MachineInstr instruction(s).
207 /// Insert the newly translated instruction(s) right where the CurBuilder
208 /// is set.
209 ///
210 /// The general algorithm is:
211 /// 1. Look for a virtual register for each operand or
212 /// create one.
213 /// 2 Update the VMap accordingly.
214 /// 2.alt. For constant arguments, if they are compile time constants,
215 /// produce an immediate in the right operand and do not touch
216 /// ValToReg. Actually we will go with a virtual register for each
217 /// constants because it may be expensive to actually materialize the
218 /// constant. Moreover, if the constant spans on several instructions,
219 /// CSE may not catch them.
220 /// => Update ValToVReg and remember that we saw a constant in Constants.
221 /// We will materialize all the constants in finalize.
222 /// Note: we would need to do something so that we can recognize such operand
223 /// as constants.
224 /// 3. Create the generic instruction.
225 ///
226 /// \return true if the translation succeeded.
227 bool translate(const Instruction &Inst);
228
229 /// Materialize \p C into virtual-register \p Reg. The generic instructions
230 /// performing this materialization will be inserted into the entry block of
231 /// the function.
232 ///
233 /// \return true if the materialization succeeded.
234 bool translate(const Constant &C, Register Reg);
235
236 /// Examine any debug-info attached to the instruction (in the form of
237 /// DbgRecords) and translate it.
238 void translateDbgInfo(const Instruction &Inst, MachineIRBuilder &MIRBuilder);
239
240 /// Translate a debug-info record of a dbg.value into a DBG_* instruction.
241 /// Pass in all the contents of the record, rather than relying on how it's
242 /// stored.
243 void translateDbgValueRecord(Value *V, bool HasArgList,
244 const DILocalVariable *Variable,
246 const DebugLoc &DL,
247 MachineIRBuilder &MIRBuilder);
248
249 /// Translate a debug-info record of a dbg.declare into an indirect DBG_*
250 /// instruction. Pass in all the contents of the record, rather than relying
251 /// on how it's stored.
252 void translateDbgDeclareRecord(Value *Address, bool HasArgList,
253 const DILocalVariable *Variable,
255 const DebugLoc &DL,
256 MachineIRBuilder &MIRBuilder);
257
258 // Translate U as a copy of V.
259 bool translateCopy(const User &U, const Value &V,
260 MachineIRBuilder &MIRBuilder);
261 bool translateCopy(const User &U, Register Src, MachineIRBuilder &MIRBuilder);
262
263 /// Translate an LLVM bitcast into generic IR. Either a COPY or a G_BITCAST is
264 /// emitted.
265 bool translateBitCast(const User &U, MachineIRBuilder &MIRBuilder);
266
267 /// Translate an LLVM load instruction into generic IR.
268 bool translateLoad(const User &U, MachineIRBuilder &MIRBuilder);
269
270 /// Translate an LLVM store instruction into generic IR.
271 bool translateStore(const User &U, MachineIRBuilder &MIRBuilder);
272
273 /// Translate an LLVM string intrinsic (memcpy, memset, ...).
274 bool translateMemFunc(const CallInst &CI, MachineIRBuilder &MIRBuilder,
275 unsigned Opcode);
276
277 /// Translate an LLVM trap intrinsic (trap, debugtrap, ubsantrap).
278 bool translateTrap(const CallInst &U, MachineIRBuilder &MIRBuilder,
279 unsigned Opcode);
280
281 // Translate @llvm.vector.interleave2 and
282 // @llvm.vector.deinterleave2 intrinsics for fixed-width vector
283 // types into vector shuffles.
284 bool translateVectorInterleave2Intrinsic(const CallInst &CI,
285 MachineIRBuilder &MIRBuilder);
286 bool translateVectorDeinterleave2Intrinsic(const CallInst &CI,
287 MachineIRBuilder &MIRBuilder);
288
289 void getStackGuard(Register DstReg, MachineIRBuilder &MIRBuilder);
290
291 bool translateOverflowIntrinsic(const CallInst &CI, unsigned Op,
292 MachineIRBuilder &MIRBuilder);
293 bool translateFixedPointIntrinsic(unsigned Op, const CallInst &CI,
294 MachineIRBuilder &MIRBuilder);
295
296 /// Helper function for translateSimpleIntrinsic.
297 /// \return The generic opcode for \p IntrinsicID if \p IntrinsicID is a
298 /// simple intrinsic (ceil, fabs, etc.). Otherwise, returns
299 /// Intrinsic::not_intrinsic.
300 unsigned getSimpleIntrinsicOpcode(Intrinsic::ID ID);
301
302 /// Translates the intrinsics defined in getSimpleIntrinsicOpcode.
303 /// \return true if the translation succeeded.
304 bool translateSimpleIntrinsic(const CallInst &CI, Intrinsic::ID ID,
305 MachineIRBuilder &MIRBuilder);
306
307 bool translateConstrainedFPIntrinsic(const ConstrainedFPIntrinsic &FPI,
308 MachineIRBuilder &MIRBuilder);
309
310 bool translateKnownIntrinsic(const CallInst &CI, Intrinsic::ID ID,
311 MachineIRBuilder &MIRBuilder);
312
313 /// Returns the single livein physical register Arg was lowered to, if
314 /// possible.
315 std::optional<MCRegister> getArgPhysReg(Argument &Arg);
316
317 /// If debug-info targets an Argument and its expression is an EntryValue,
318 /// lower it as either an entry in the MF debug table (dbg.declare), or a
319 /// DBG_VALUE targeting the corresponding livein register for that Argument
320 /// (dbg.value).
321 bool translateIfEntryValueArgument(bool isDeclare, Value *Arg,
322 const DILocalVariable *Var,
323 const DIExpression *Expr,
324 const DebugLoc &DL,
325 MachineIRBuilder &MIRBuilder);
326
327 bool translateInlineAsm(const CallBase &CB, MachineIRBuilder &MIRBuilder);
328
329 /// Common code for translating normal calls or invokes.
330 bool translateCallBase(const CallBase &CB, MachineIRBuilder &MIRBuilder);
331
332 /// Translate call instruction.
333 /// \pre \p U is a call instruction.
334 bool translateCall(const User &U, MachineIRBuilder &MIRBuilder);
335
336 bool translateIntrinsic(
337 const CallBase &CB, Intrinsic::ID ID, MachineIRBuilder &MIRBuilder,
338 ArrayRef<TargetLowering::IntrinsicInfo> TgtMemIntrinsicInfos = {});
339
340 /// Report an intrinsic the subtarget does not support and define its results
341 /// with G_IMPLICIT_DEF. Prevents creating a malformed MIR.
342 bool handleUnsupportedIntrinsic(const CallBase &CB, Intrinsic::ID ID,
343 MachineIRBuilder &MIRBuilder);
344
345 /// When an invoke or a cleanupret unwinds to the next EH pad, there are
346 /// many places it could ultimately go. In the IR, we have a single unwind
347 /// destination, but in the machine CFG, we enumerate all the possible blocks.
348 /// This function skips over imaginary basic blocks that hold catchswitch
349 /// instructions, and finds all the "real" machine
350 /// basic block destinations. As those destinations may not be successors of
351 /// EHPadBB, here we also calculate the edge probability to those
352 /// destinations. The passed-in Prob is the edge probability to EHPadBB.
353 bool findUnwindDestinations(
354 const BasicBlock *EHPadBB, BranchProbability Prob,
355 SmallVectorImpl<std::pair<MachineBasicBlock *, BranchProbability>>
356 &UnwindDests);
357
358 bool translateInvoke(const User &U, MachineIRBuilder &MIRBuilder);
359
360 bool translateCallBr(const User &U, MachineIRBuilder &MIRBuilder);
361
362 bool translateLandingPad(const User &U, MachineIRBuilder &MIRBuilder);
363
364 /// Translate one of LLVM's cast instructions into MachineInstrs, with the
365 /// given generic Opcode.
366 bool translateCast(unsigned Opcode, const User &U,
367 MachineIRBuilder &MIRBuilder);
368
369 /// Translate a phi instruction.
370 bool translatePHI(const User &U, MachineIRBuilder &MIRBuilder);
371
372 /// Translate a comparison (icmp or fcmp) instruction or constant.
373 bool translateCompare(const User &U, MachineIRBuilder &MIRBuilder);
374
375 /// Translate an integer compare instruction (or constant).
376 bool translateICmp(const User &U, MachineIRBuilder &MIRBuilder) {
377 return translateCompare(U, MIRBuilder);
378 }
379
380 /// Translate a floating-point compare instruction (or constant).
381 bool translateFCmp(const User &U, MachineIRBuilder &MIRBuilder) {
382 return translateCompare(U, MIRBuilder);
383 }
384
385 /// Add remaining operands onto phis we've translated. Executed after all
386 /// MachineBasicBlocks for the function have been created.
387 void finishPendingPhis();
388
389 /// Translate \p Inst into a unary operation \p Opcode.
390 /// \pre \p U is a unary operation.
391 bool translateUnaryOp(unsigned Opcode, const User &U,
392 MachineIRBuilder &MIRBuilder);
393
394 /// Translate \p Inst into a binary operation \p Opcode.
395 /// \pre \p U is a binary operation.
396 bool translateBinaryOp(unsigned Opcode, const User &U,
397 MachineIRBuilder &MIRBuilder);
398
399 /// If the set of cases should be emitted as a series of branches, return
400 /// true. If we should emit this as a bunch of and/or'd together conditions,
401 /// return false.
402 bool shouldEmitAsBranches(const std::vector<SwitchCG::CaseBlock> &Cases);
403 /// Helper method for findMergedConditions.
404 /// This function emits a branch and is used at the leaves of an OR or an
405 /// AND operator tree.
406 void emitBranchForMergedCondition(const Value *Cond, MachineBasicBlock *TBB,
408 MachineBasicBlock *CurBB,
409 MachineBasicBlock *SwitchBB,
410 BranchProbability TProb,
411 BranchProbability FProb, bool InvertCond);
412 /// Used during condbr translation to find trees of conditions that can be
413 /// optimized.
414 void findMergedConditions(const Value *Cond, MachineBasicBlock *TBB,
416 MachineBasicBlock *SwitchBB,
418 BranchProbability FProb, bool InvertCond);
419
420 /// Translate branch (br) instruction.
421 /// \pre \p U is a branch instruction.
422 bool translateUncondBr(const User &U, MachineIRBuilder &MIRBuilder);
423 bool translateCondBr(const User &U, MachineIRBuilder &MIRBuilder);
424
425 // Begin switch lowering functions.
426 bool emitJumpTableHeader(SwitchCG::JumpTable &JT,
428 MachineBasicBlock *HeaderBB);
429 void emitJumpTable(SwitchCG::JumpTable &JT, MachineBasicBlock *MBB);
430
431 void emitSwitchCase(SwitchCG::CaseBlock &CB, MachineBasicBlock *SwitchBB,
432 MachineIRBuilder &MIB);
433
434 /// Generate for the BitTest header block, which precedes each sequence of
435 /// BitTestCases.
436 void emitBitTestHeader(SwitchCG::BitTestBlock &BTB,
437 MachineBasicBlock *SwitchMBB);
438 /// Generate code to produces one "bit test" for a given BitTestCase \p B.
439 void emitBitTestCase(SwitchCG::BitTestBlock &BB, MachineBasicBlock *NextMBB,
440 BranchProbability BranchProbToNext, Register Reg,
442
443 void splitWorkItem(SwitchCG::SwitchWorkList &WorkList,
445 MachineBasicBlock *SwitchMBB, MachineIRBuilder &MIB);
446
447 bool lowerJumpTableWorkItem(
449 MachineBasicBlock *CurMBB, MachineBasicBlock *DefaultMBB,
452 MachineBasicBlock *Fallthrough, bool FallthroughUnreachable);
453
454 bool lowerSwitchRangeWorkItem(SwitchCG::CaseClusterIt I, Value *Cond,
455 MachineBasicBlock *Fallthrough,
456 bool FallthroughUnreachable,
457 BranchProbability UnhandledProbs,
458 MachineBasicBlock *CurMBB,
459 MachineIRBuilder &MIB,
460 MachineBasicBlock *SwitchMBB);
461
462 bool lowerBitTestWorkItem(
464 MachineBasicBlock *CurMBB, MachineBasicBlock *DefaultMBB,
466 BranchProbability DefaultProb, BranchProbability UnhandledProbs,
468 bool FallthroughUnreachable);
469
470 bool lowerSwitchWorkItem(SwitchCG::SwitchWorkListItem W, Value *Cond,
471 MachineBasicBlock *SwitchMBB,
472 MachineBasicBlock *DefaultMBB,
473 MachineIRBuilder &MIB);
474
475 bool translateSwitch(const User &U, MachineIRBuilder &MIRBuilder);
476 // End switch lowering section.
477
478 bool translateIndirectBr(const User &U, MachineIRBuilder &MIRBuilder);
479
480 bool translateExtractValue(const User &U, MachineIRBuilder &MIRBuilder);
481
482 bool translateInsertValue(const User &U, MachineIRBuilder &MIRBuilder);
483
484 bool translateSelect(const User &U, MachineIRBuilder &MIRBuilder);
485
486 bool translateGetElementPtr(const User &U, MachineIRBuilder &MIRBuilder);
487
488 bool translateAlloca(const User &U, MachineIRBuilder &MIRBuilder);
489
490 /// Translate return (ret) instruction.
491 /// The target needs to implement CallLowering::lowerReturn for
492 /// this to succeed.
493 /// \pre \p U is a return instruction.
494 bool translateRet(const User &U, MachineIRBuilder &MIRBuilder);
495
496 bool translateFNeg(const User &U, MachineIRBuilder &MIRBuilder);
497
498 bool translateAdd(const User &U, MachineIRBuilder &MIRBuilder) {
499 return translateBinaryOp(TargetOpcode::G_ADD, U, MIRBuilder);
500 }
501 bool translateSub(const User &U, MachineIRBuilder &MIRBuilder) {
502 return translateBinaryOp(TargetOpcode::G_SUB, U, MIRBuilder);
503 }
504 bool translateAnd(const User &U, MachineIRBuilder &MIRBuilder) {
505 return translateBinaryOp(TargetOpcode::G_AND, U, MIRBuilder);
506 }
507 bool translateMul(const User &U, MachineIRBuilder &MIRBuilder) {
508 return translateBinaryOp(TargetOpcode::G_MUL, U, MIRBuilder);
509 }
510 bool translateOr(const User &U, MachineIRBuilder &MIRBuilder) {
511 return translateBinaryOp(TargetOpcode::G_OR, U, MIRBuilder);
512 }
513 bool translateXor(const User &U, MachineIRBuilder &MIRBuilder) {
514 return translateBinaryOp(TargetOpcode::G_XOR, U, MIRBuilder);
515 }
516
517 bool translateUDiv(const User &U, MachineIRBuilder &MIRBuilder) {
518 return translateBinaryOp(TargetOpcode::G_UDIV, U, MIRBuilder);
519 }
520 bool translateSDiv(const User &U, MachineIRBuilder &MIRBuilder) {
521 return translateBinaryOp(TargetOpcode::G_SDIV, U, MIRBuilder);
522 }
523 bool translateURem(const User &U, MachineIRBuilder &MIRBuilder) {
524 return translateBinaryOp(TargetOpcode::G_UREM, U, MIRBuilder);
525 }
526 bool translateSRem(const User &U, MachineIRBuilder &MIRBuilder) {
527 return translateBinaryOp(TargetOpcode::G_SREM, U, MIRBuilder);
528 }
529 bool translateIntToPtr(const User &U, MachineIRBuilder &MIRBuilder) {
530 return translateCast(TargetOpcode::G_INTTOPTR, U, MIRBuilder);
531 }
532 bool translatePtrToInt(const User &U, MachineIRBuilder &MIRBuilder) {
533 return translateCast(TargetOpcode::G_PTRTOINT, U, MIRBuilder);
534 }
535 bool translatePtrToAddr(const User &U, MachineIRBuilder &MIRBuilder) {
536 // FIXME: this is not correct for pointers with addr width != pointer width
537 return translatePtrToInt(U, MIRBuilder);
538 }
539 bool translateTrunc(const User &U, MachineIRBuilder &MIRBuilder) {
540 return translateCast(TargetOpcode::G_TRUNC, U, MIRBuilder);
541 }
542 bool translateFPTrunc(const User &U, MachineIRBuilder &MIRBuilder) {
543 return translateCast(TargetOpcode::G_FPTRUNC, U, MIRBuilder);
544 }
545 bool translateFPExt(const User &U, MachineIRBuilder &MIRBuilder) {
546 return translateCast(TargetOpcode::G_FPEXT, U, MIRBuilder);
547 }
548 bool translateFPToUI(const User &U, MachineIRBuilder &MIRBuilder) {
549 return translateCast(TargetOpcode::G_FPTOUI, U, MIRBuilder);
550 }
551 bool translateFPToSI(const User &U, MachineIRBuilder &MIRBuilder) {
552 return translateCast(TargetOpcode::G_FPTOSI, U, MIRBuilder);
553 }
554 bool translateUIToFP(const User &U, MachineIRBuilder &MIRBuilder) {
555 return translateCast(TargetOpcode::G_UITOFP, U, MIRBuilder);
556 }
557 bool translateSIToFP(const User &U, MachineIRBuilder &MIRBuilder) {
558 return translateCast(TargetOpcode::G_SITOFP, U, MIRBuilder);
559 }
560 bool translateUnreachable(const User &U, MachineIRBuilder &MIRBuilder);
561
562 bool translateSExt(const User &U, MachineIRBuilder &MIRBuilder) {
563 return translateCast(TargetOpcode::G_SEXT, U, MIRBuilder);
564 }
565
566 bool translateZExt(const User &U, MachineIRBuilder &MIRBuilder) {
567 return translateCast(TargetOpcode::G_ZEXT, U, MIRBuilder);
568 }
569
570 bool translateShl(const User &U, MachineIRBuilder &MIRBuilder) {
571 return translateBinaryOp(TargetOpcode::G_SHL, U, MIRBuilder);
572 }
573 bool translateLShr(const User &U, MachineIRBuilder &MIRBuilder) {
574 return translateBinaryOp(TargetOpcode::G_LSHR, U, MIRBuilder);
575 }
576 bool translateAShr(const User &U, MachineIRBuilder &MIRBuilder) {
577 return translateBinaryOp(TargetOpcode::G_ASHR, U, MIRBuilder);
578 }
579
580 bool translateFAdd(const User &U, MachineIRBuilder &MIRBuilder) {
581 return translateBinaryOp(TargetOpcode::G_FADD, U, MIRBuilder);
582 }
583 bool translateFSub(const User &U, MachineIRBuilder &MIRBuilder) {
584 return translateBinaryOp(TargetOpcode::G_FSUB, U, MIRBuilder);
585 }
586 bool translateFMul(const User &U, MachineIRBuilder &MIRBuilder) {
587 return translateBinaryOp(TargetOpcode::G_FMUL, U, MIRBuilder);
588 }
589 bool translateFDiv(const User &U, MachineIRBuilder &MIRBuilder) {
590 return translateBinaryOp(TargetOpcode::G_FDIV, U, MIRBuilder);
591 }
592 bool translateFRem(const User &U, MachineIRBuilder &MIRBuilder) {
593 return translateBinaryOp(TargetOpcode::G_FREM, U, MIRBuilder);
594 }
595
596 bool translateVAArg(const User &U, MachineIRBuilder &MIRBuilder);
597
598 bool translateInsertElement(const User &U, MachineIRBuilder &MIRBuilder);
599 bool translateInsertVector(const User &U, MachineIRBuilder &MIRBuilder);
600
601 bool translateExtractElement(const User &U, MachineIRBuilder &MIRBuilder);
602 bool translateExtractVector(const User &U, MachineIRBuilder &MIRBuilder);
603
604 bool translateShuffleVector(const User &U, MachineIRBuilder &MIRBuilder);
605
606 bool translateAtomicCmpXchg(const User &U, MachineIRBuilder &MIRBuilder);
607 bool translateAtomicRMW(const User &U, MachineIRBuilder &MIRBuilder);
608 bool translateFence(const User &U, MachineIRBuilder &MIRBuilder);
609 bool translateFreeze(const User &U, MachineIRBuilder &MIRBuilder);
610
611 bool translateBitExtract(const User &U, MachineIRBuilder &MIRBuilder);
612 bool translateBitInsert(const User &U, MachineIRBuilder &MIRBuilder);
613
614 // Stubs to keep the compiler happy while we implement the rest of the
615 // translation.
616 bool translateResume(const User &U, MachineIRBuilder &MIRBuilder) {
617 return false;
618 }
619 bool translateCleanupRet(const User &U, MachineIRBuilder &MIRBuilder) {
620 return false;
621 }
622 bool translateCatchRet(const User &U, MachineIRBuilder &MIRBuilder) {
623 return false;
624 }
625 bool translateCatchSwitch(const User &U, MachineIRBuilder &MIRBuilder) {
626 return false;
627 }
628 bool translateAddrSpaceCast(const User &U, MachineIRBuilder &MIRBuilder) {
629 return translateCast(TargetOpcode::G_ADDRSPACE_CAST, U, MIRBuilder);
630 }
631 bool translateCleanupPad(const User &U, MachineIRBuilder &MIRBuilder) {
632 return false;
633 }
634 bool translateCatchPad(const User &U, MachineIRBuilder &MIRBuilder) {
635 return false;
636 }
637 bool translateUserOp1(const User &U, MachineIRBuilder &MIRBuilder) {
638 return false;
639 }
640 bool translateUserOp2(const User &U, MachineIRBuilder &MIRBuilder) {
641 return false;
642 }
643
644 bool translateConvergenceControlIntrinsic(const CallInst &CI,
645 Intrinsic::ID ID,
646 MachineIRBuilder &MIRBuilder);
647
648 /// @}
649
650 // Builder for machine instruction a la IRBuilder.
651 // I.e., compared to regular MIBuilder, this one also inserts the instruction
652 // in the current block, it can creates block, etc., basically a kind of
653 // IRBuilder, but for Machine IR.
654 // CSEMIRBuilder CurBuilder;
655 std::unique_ptr<MachineIRBuilder> CurBuilder;
656
657 // Builder set to the entry block (just after ABI lowering instructions). Used
658 // as a convenient location for Constants.
659 // CSEMIRBuilder EntryBuilder;
660 std::unique_ptr<MachineIRBuilder> EntryBuilder;
661
662 // The MachineFunction currently being translated.
663 MachineFunction *MF = nullptr;
664
665 /// MachineRegisterInfo used to create virtual registers.
666 MachineRegisterInfo *MRI = nullptr;
667
668 const DataLayout *DL = nullptr;
669
670 CodeGenOptLevel OptLevel;
671
672 /// Current optimization remark emitter. Used to report failures.
673 std::unique_ptr<OptimizationRemarkEmitter> ORE;
674
675 AAResults *AA = nullptr;
676 AssumptionCache *AC = nullptr;
677 const TargetLibraryInfo *LibInfo = nullptr;
678 const LibcallLoweringInfo *Libcalls = nullptr;
679 const TargetLowering *TLI = nullptr;
680 FunctionLoweringInfo FuncInfo;
681
682 // True when either the Target Machine specifies no optimizations or the
683 // function has the optnone attribute.
684 bool EnableOpts = false;
685
686 /// True when the block contains a tail call. This allows the IRTranslator to
687 /// stop translating such blocks early.
688 bool HasTailCall = false;
689
690 StackProtectorDescriptor SPDescriptor;
691
692 bool mayTranslateUserTypes(const User &U) const;
693
694 /// Switch analysis and optimization.
695 class GISelSwitchLowering : public SwitchCG::SwitchLowering {
696 public:
697 GISelSwitchLowering(IRTranslatorImpl *irt, FunctionLoweringInfo &funcinfo)
698 : SwitchLowering(funcinfo), IRT(irt) {
699 assert(irt && "irt is null!");
700 }
701
702 void addSuccessorWithProb(
705 IRT->addSuccessorWithProb(Src, Dst, Prob);
706 }
707
708 ~GISelSwitchLowering() override = default;
709
710 private:
711 IRTranslatorImpl *IRT;
712 };
713
714 std::unique_ptr<GISelSwitchLowering> SL;
715
716 // * Insert all the code needed to materialize the constants
717 // at the proper place. E.g., Entry block or dominator block
718 // of each constant depending on how fancy we want to be.
719 // * Clear the different maps.
720 void finalizeFunction();
721
722 // Processing steps done per block. E.g. emitting jump tables, stack
723 // protectors etc. Returns true if no errors, false if there was a problem
724 // that caused an abort.
725 bool finalizeBasicBlock(const BasicBlock &BB, MachineBasicBlock &MBB);
726
727 /// Codegen a new tail for a stack protector check ParentMBB which has had its
728 /// tail spliced into a stack protector check success bb.
729 ///
730 /// For a high level explanation of how this fits into the stack protector
731 /// generation see the comment on the declaration of class
732 /// StackProtectorDescriptor.
733 ///
734 /// \return true if there were no problems.
735 bool emitSPDescriptorParent(StackProtectorDescriptor &SPD,
736 MachineBasicBlock *ParentBB);
737
738 /// Codegen the failure basic block for a stack protector check.
739 ///
740 /// A failure stack protector machine basic block consists simply of a call to
741 /// __stack_chk_fail().
742 ///
743 /// For a high level explanation of how this fits into the stack protector
744 /// generation see the comment on the declaration of class
745 /// StackProtectorDescriptor.
746 ///
747 /// \return true if there were no problems.
748 bool emitSPDescriptorFailure(StackProtectorDescriptor &SPD,
749 MachineBasicBlock *FailureBB);
750
751 /// Get the VRegs that represent \p Val.
752 /// Non-aggregate types have just one corresponding VReg and the list can be
753 /// used as a single "unsigned". Aggregates get flattened. If such VRegs do
754 /// not exist, they are created.
755 ArrayRef<Register> getOrCreateVRegs(const Value &Val);
756
757 Register getOrCreateVReg(const Value &Val) {
758 auto Regs = getOrCreateVRegs(Val);
759 if (Regs.empty())
760 return 0;
761 assert(Regs.size() == 1 &&
762 "attempt to get single VReg for aggregate or void");
763 return Regs[0];
764 }
765
766 Register getOrCreateConvergenceTokenVReg(const Value &Token) {
767 assert(Token.getType()->isTokenTy());
768 auto &Regs = *VMap.getVRegs(Token);
769 if (!Regs.empty()) {
770 assert(Regs.size() == 1 &&
771 "Expected a single register for convergence tokens.");
772 return Regs[0];
773 }
774
775 auto Reg = MRI->createGenericVirtualRegister(LLT::token());
776 Regs.push_back(Reg);
777 return Reg;
778 }
779
780 /// Allocate empty vregs for \p Val. For aggregate values, also populate
781 /// their offsets.
782 ValueToVRegInfo::VRegListT &allocateVRegs(const Value &Val);
783
784 /// Get the frame index that represents \p Val.
785 /// If such VReg does not exist, it is created.
786 int getOrCreateFrameIndex(const AllocaInst &AI);
787
788 /// Get the alignment of the given memory operation instruction. This will
789 /// either be the explicitly specified value or the ABI-required alignment for
790 /// the type being accessed (according to the Module's DataLayout).
791 Align getMemOpAlign(const Instruction &I);
792
793 /// Get the MachineBasicBlock that represents \p BB. Specifically, the block
794 /// returned will be the head of the translated block (suitable for branch
795 /// destinations).
796 MachineBasicBlock &getMBB(const BasicBlock &BB);
797
798 /// Record \p NewPred as a Machine predecessor to `Edge.second`, corresponding
799 /// to `Edge.first` at the IR level. This is used when IRTranslation creates
800 /// multiple MachineBasicBlocks for a given IR block and the CFG is no longer
801 /// represented simply by the IR-level CFG.
802 void addMachineCFGPred(CFGEdge Edge, MachineBasicBlock *NewPred);
803
804 /// Returns the Machine IR predecessors for the given IR CFG edge. Usually
805 /// this is just the single MachineBasicBlock corresponding to the predecessor
806 /// in the IR. More complex lowering can result in multiple MachineBasicBlocks
807 /// preceding the original though (e.g. switch instructions).
808 SmallVector<MachineBasicBlock *, 1> getMachinePredBBs(CFGEdge Edge) {
809 auto RemappedEdge = MachinePreds.find(Edge);
810 if (RemappedEdge != MachinePreds.end())
811 return RemappedEdge->second;
812 return SmallVector<MachineBasicBlock *, 4>(1, &getMBB(*Edge.first));
813 }
814
815 /// Return branch probability calculated by BranchProbabilityInfo for IR
816 /// blocks.
817 BranchProbability getEdgeProbability(const MachineBasicBlock *Src,
818 const MachineBasicBlock *Dst) const;
819
820 void addSuccessorWithProb(
823
824public:
826 : OptLevel(OptLevel) {}
827
828 // Algo:
829 // CallLowering = MF.subtarget.getCallLowering()
830 // F = MF.getParent()
831 // MIRBuilder.reset(MF)
832 // getMBB(F.getEntryBB())
833 // CallLowering->translateArguments(MIRBuilder, F, ValToVReg)
834 // for each bb in F
835 // getMBB(bb)
836 // for each inst in bb
837 // if (!translate(MIRBuilder, inst, ValToVReg, ConstantToSequence))
838 // reportFatalUsageError("Don't know how to translate input");
839 // finalize()
841 function_ref<GISelCSEInfo *()> GetCSEInfo,
842 bool ShouldSkipOpts,
843 function_ref<AAResults *()> GetAAResults,
845 function_ref<AssumptionCache *()> GetAC,
846 TargetLibraryInfo *LibraryInfo,
847 const LibcallLoweringInfo *LibcallInfo,
848 SSPLayoutInfo *StackProtectorInfo);
849};
850
851} // namespace llvm
852
854
856 "IRTranslator LLVM IR -> MI", false, false)
863 "IRTranslator LLVM IR -> MI", false, false)
864
868 MF.getProperties().setFailedISel();
869 bool IsGlobalISelAbortEnabled =
870 MF.getTarget().Options.GlobalISelAbort == GlobalISelAbortMode::Enable;
871
872 // Print the function name explicitly if we don't have a debug location (which
873 // makes the diagnostic less useful) or if we're going to emit a raw error.
874 if (!R.getLocation().isValid() || IsGlobalISelAbortEnabled)
875 R << (" (in function: " + MF.getName() + ")").str();
876
877 if (IsGlobalISelAbortEnabled)
878 report_fatal_error(Twine(R.getMsg()));
879 else
880 ORE.emit(R);
881}
882
884 : MachineFunctionPass(ID), OptLevel(OptLevel),
885 Impl(std::make_unique<IRTranslatorImpl>(OptLevel)) {}
886
888
889#ifndef NDEBUG
890namespace {
891/// Verify that every instruction created has the same DILocation as the
892/// instruction being translated.
893class DILocationVerifier : public GISelChangeObserver {
894 const Instruction *CurrInst = nullptr;
895
896public:
897 DILocationVerifier() = default;
898 ~DILocationVerifier() override = default;
899
900 const Instruction *getCurrentInst() const { return CurrInst; }
901 void setCurrentInst(const Instruction *Inst) { CurrInst = Inst; }
902
903 void erasingInstr(MachineInstr &MI) override {}
904 void changingInstr(MachineInstr &MI) override {}
905 void changedInstr(MachineInstr &MI) override {}
906
907 void createdInstr(MachineInstr &MI) override {
908 assert(getCurrentInst() && "Inserted instruction without a current MI");
909
910 // Only print the check message if we're actually checking it.
911#ifndef NDEBUG
912 LLVM_DEBUG(dbgs() << "Checking DILocation from " << *CurrInst
913 << " was copied to " << MI);
914#endif
915 // We allow insts in the entry block to have no debug loc because
916 // they could have originated from constants, and we don't want a jumpy
917 // debug experience.
918 assert((CurrInst->getDebugLoc() == MI.getDebugLoc() ||
919 (MI.getParent()->isEntryBlock() && !MI.getDebugLoc()) ||
920 (MI.isDebugInstr())) &&
921 "Line info was not transferred to all instructions");
922 }
923};
924} // namespace
925#endif // ifndef NDEBUG
926
943
944IRTranslatorImpl::ValueToVRegInfo::VRegListT &
945IRTranslatorImpl::allocateVRegs(const Value &Val) {
946 auto VRegsIt = VMap.findVRegs(Val);
947 if (VRegsIt != VMap.vregs_end())
948 return *VRegsIt->second;
949 auto *Regs = VMap.getVRegs(Val);
950 if (!Val.getType()->isAggregateType()) {
951 Regs->push_back(0);
952 return *Regs;
953 }
954
955 auto *Offsets = VMap.getOffsets(Val);
956 SmallVector<LLT, 4> SplitTys;
957 computeValueLLTs(*DL, *Val.getType(), SplitTys,
958 Offsets->empty() ? Offsets : nullptr);
959 for (unsigned i = 0; i < SplitTys.size(); ++i)
960 Regs->push_back(0);
961 return *Regs;
962}
963
964ArrayRef<Register> IRTranslatorImpl::getOrCreateVRegs(const Value &Val) {
965 auto VRegsIt = VMap.findVRegs(Val);
966 if (VRegsIt != VMap.vregs_end())
967 return *VRegsIt->second;
968
969 if (Val.getType()->isVoidTy())
970 return *VMap.getVRegs(Val);
971
972 // Create entry for this type.
973 auto *VRegs = VMap.getVRegs(Val);
974
975 if (!Val.getType()->isTokenTy())
976 assert(Val.getType()->isSized() &&
977 "Don't know how to create an empty vreg");
978
979 // Fast-path values that lower to a single vreg.
980 if (!Val.getType()->isAggregateType()) {
981 LLT Ty = getLLTForType(*Val.getType(), *DL);
982 VRegs->push_back(MRI->createGenericVirtualRegister(Ty));
983 if (isa<Constant>(Val)) {
984 bool Success = translate(cast<Constant>(Val), VRegs->front());
985 if (!Success) {
986 OptimizationRemarkMissed R("gisel-ir-translator", "GISelFailure",
987 MF->getFunction().getSubprogram(),
988 &MF->getFunction().getEntryBlock());
989 R << "unable to translate constant: " << ore::NV("Type", Val.getType());
990 reportTranslationError(*MF, *ORE, R);
991 }
992 }
993 return *VRegs;
994 }
995
996 SmallVector<LLT, 4> SplitTys;
997 auto *Offsets = VMap.getOffsets(Val);
998 computeValueLLTs(*DL, *Val.getType(), SplitTys,
999 Offsets->empty() ? Offsets : nullptr);
1000
1001 if (!isa<Constant>(Val)) {
1002 for (auto Ty : SplitTys)
1003 VRegs->push_back(MRI->createGenericVirtualRegister(Ty));
1004 return *VRegs;
1005 }
1006
1007 // UndefValue, ConstantAggregateZero
1008 auto &C = cast<Constant>(Val);
1009 unsigned Idx = 0;
1010 while (auto Elt = C.getAggregateElement(Idx++)) {
1011 auto EltRegs = getOrCreateVRegs(*Elt);
1012 llvm::append_range(*VRegs, EltRegs);
1013 }
1014
1015 return *VRegs;
1016}
1017
1018int IRTranslatorImpl::getOrCreateFrameIndex(const AllocaInst &AI) {
1019 auto [MapEntry, Inserted] = FrameIndices.try_emplace(&AI);
1020 if (!Inserted)
1021 return MapEntry->second;
1022
1023 TypeSize TySize = AI.getAllocationSize(*DL).value_or(TypeSize::getZero());
1024 uint64_t Size = TySize.getKnownMinValue();
1025
1026 // Always allocate at least one byte.
1027 Size = std::max<uint64_t>(Size, 1u);
1028
1029 int &FI = MapEntry->second;
1030 FI = MF->getFrameInfo().CreateStackObject(Size, AI.getAlign(), false, &AI);
1031
1032 // Scalable vectors and structures that contain scalable vectors may
1033 // need a special StackID to distinguish them from other (fixed size)
1034 // stack objects.
1035 if (TySize.isScalable()) {
1036 auto StackID =
1037 MF->getSubtarget().getFrameLowering()->getStackIDForScalableVectors();
1038 MF->getFrameInfo().setStackID(FI, StackID);
1039 }
1040
1041 return FI;
1042}
1043
1044Align IRTranslatorImpl::getMemOpAlign(const Instruction &I) {
1045 if (const StoreInst *SI = dyn_cast<StoreInst>(&I))
1046 return SI->getAlign();
1047 if (const LoadInst *LI = dyn_cast<LoadInst>(&I))
1048 return LI->getAlign();
1049 if (const AtomicCmpXchgInst *AI = dyn_cast<AtomicCmpXchgInst>(&I))
1050 return AI->getAlign();
1051 if (const AtomicRMWInst *AI = dyn_cast<AtomicRMWInst>(&I))
1052 return AI->getAlign();
1053
1054 OptimizationRemarkMissed R("gisel-ir-translator", "", &I);
1055 R << "unable to translate memop: " << ore::NV("Opcode", &I);
1056 reportTranslationError(*MF, *ORE, R);
1057 return Align(1);
1058}
1059
1060MachineBasicBlock &IRTranslatorImpl::getMBB(const BasicBlock &BB) {
1061 MachineBasicBlock *MBB = FuncInfo.getMBB(&BB);
1062 assert(MBB && "BasicBlock was not encountered before");
1063 return *MBB;
1064}
1065
1066void IRTranslatorImpl::addMachineCFGPred(CFGEdge Edge,
1067 MachineBasicBlock *NewPred) {
1068 assert(NewPred && "new predecessor must be a real MachineBasicBlock");
1069 MachinePreds[Edge].push_back(NewPred);
1070}
1071
1072bool IRTranslatorImpl::translateBinaryOp(unsigned Opcode, const User &U,
1073 MachineIRBuilder &MIRBuilder) {
1074 if (!mayTranslateUserTypes(U))
1075 return false;
1076
1077 // Get or create a virtual register for each value.
1078 // Unless the value is a Constant => loadimm cst?
1079 // or inline constant each time?
1080 // Creation of a virtual register needs to have a size.
1081 Register Op0 = getOrCreateVReg(*U.getOperand(0));
1082 Register Op1 = getOrCreateVReg(*U.getOperand(1));
1083 Register Res = getOrCreateVReg(U);
1084 uint32_t Flags = 0;
1085 if (isa<Instruction>(U)) {
1086 const Instruction &I = cast<Instruction>(U);
1088 }
1089
1090 MIRBuilder.buildInstr(Opcode, {Res}, {Op0, Op1}, Flags);
1091 return true;
1092}
1093
1094bool IRTranslatorImpl::translateUnaryOp(unsigned Opcode, const User &U,
1095 MachineIRBuilder &MIRBuilder) {
1096 if (!mayTranslateUserTypes(U))
1097 return false;
1098
1099 Register Op0 = getOrCreateVReg(*U.getOperand(0));
1100 Register Res = getOrCreateVReg(U);
1101 uint32_t Flags = 0;
1102 if (isa<Instruction>(U)) {
1103 const Instruction &I = cast<Instruction>(U);
1105 }
1106 MIRBuilder.buildInstr(Opcode, {Res}, {Op0}, Flags);
1107 return true;
1108}
1109
1110bool IRTranslatorImpl::translateFNeg(const User &U,
1111 MachineIRBuilder &MIRBuilder) {
1112 return translateUnaryOp(TargetOpcode::G_FNEG, U, MIRBuilder);
1113}
1114
1115bool IRTranslatorImpl::translateCompare(const User &U,
1116 MachineIRBuilder &MIRBuilder) {
1117 if (!mayTranslateUserTypes(U))
1118 return false;
1119
1120 auto *CI = cast<CmpInst>(&U);
1121 Register Op0 = getOrCreateVReg(*U.getOperand(0));
1122 Register Op1 = getOrCreateVReg(*U.getOperand(1));
1123 Register Res = getOrCreateVReg(U);
1124 CmpInst::Predicate Pred = CI->getPredicate();
1126 if (CmpInst::isIntPredicate(Pred))
1127 MIRBuilder.buildICmp(Pred, Res, Op0, Op1, Flags);
1128 else if (Pred == CmpInst::FCMP_FALSE)
1129 MIRBuilder.buildCopy(
1130 Res, getOrCreateVReg(*Constant::getNullValue(U.getType())));
1131 else if (Pred == CmpInst::FCMP_TRUE)
1132 MIRBuilder.buildCopy(
1133 Res, getOrCreateVReg(*Constant::getAllOnesValue(U.getType())));
1134 else
1135 MIRBuilder.buildFCmp(Pred, Res, Op0, Op1, Flags);
1136
1137 return true;
1138}
1139
1140bool IRTranslatorImpl::translateRet(const User &U,
1141 MachineIRBuilder &MIRBuilder) {
1142 const ReturnInst &RI = cast<ReturnInst>(U);
1143 const Value *Ret = RI.getReturnValue();
1144 if (Ret && DL->getTypeStoreSize(Ret->getType()).isZero())
1145 Ret = nullptr;
1146
1147 ArrayRef<Register> VRegs;
1148 if (Ret)
1149 VRegs = getOrCreateVRegs(*Ret);
1150
1151 Register SwiftErrorVReg = 0;
1152 if (CLI->supportSwiftError() && SwiftError.getFunctionArg()) {
1153 SwiftErrorVReg = SwiftError.getOrCreateVRegUseAt(
1154 &RI, &MIRBuilder.getMBB(), SwiftError.getFunctionArg());
1155 }
1156
1157 // The target may mess up with the insertion point, but
1158 // this is not important as a return is the last instruction
1159 // of the block anyway.
1160 return CLI->lowerReturn(MIRBuilder, Ret, VRegs, FuncInfo, SwiftErrorVReg);
1161}
1162
1163void IRTranslatorImpl::emitBranchForMergedCondition(
1165 MachineBasicBlock *CurBB, MachineBasicBlock *SwitchBB,
1166 BranchProbability TProb, BranchProbability FProb, bool InvertCond) {
1167 // If the leaf of the tree is a comparison, merge the condition into
1168 // the caseblock.
1169 if (const CmpInst *BOp = dyn_cast<CmpInst>(Cond)) {
1170 CmpInst::Predicate Condition;
1171 if (const ICmpInst *IC = dyn_cast<ICmpInst>(Cond)) {
1172 Condition = InvertCond ? IC->getInversePredicate() : IC->getPredicate();
1173 } else {
1174 const FCmpInst *FC = cast<FCmpInst>(Cond);
1175 Condition = InvertCond ? FC->getInversePredicate() : FC->getPredicate();
1176 }
1177
1178 SwitchCG::CaseBlock CB(Condition, false, BOp->getOperand(0),
1179 BOp->getOperand(1), nullptr, TBB, FBB, CurBB,
1180 CurBuilder->getDebugLoc(), TProb, FProb);
1181 SL->SwitchCases.push_back(CB);
1182 return;
1183 }
1184
1185 // Create a CaseBlock record representing this branch.
1187 SwitchCG::CaseBlock CB(
1188 Pred, false, Cond, ConstantInt::getTrue(MF->getFunction().getContext()),
1189 nullptr, TBB, FBB, CurBB, CurBuilder->getDebugLoc(), TProb, FProb);
1190 SL->SwitchCases.push_back(CB);
1191}
1192
1193static bool isValInBlock(const Value *V, const BasicBlock *BB) {
1194 if (const Instruction *I = dyn_cast<Instruction>(V))
1195 return I->getParent() == BB;
1196 return true;
1197}
1198
1199void IRTranslatorImpl::findMergedConditions(
1201 MachineBasicBlock *CurBB, MachineBasicBlock *SwitchBB,
1203 BranchProbability FProb, bool InvertCond) {
1204 using namespace PatternMatch;
1205 assert((Opc == Instruction::And || Opc == Instruction::Or) &&
1206 "Expected Opc to be AND/OR");
1207 // Skip over not part of the tree and remember to invert op and operands at
1208 // next level.
1209 Value *NotCond;
1210 if (match(Cond, m_OneUse(m_Not(m_Value(NotCond)))) &&
1211 isValInBlock(NotCond, CurBB->getBasicBlock())) {
1212 findMergedConditions(NotCond, TBB, FBB, CurBB, SwitchBB, Opc, TProb, FProb,
1213 !InvertCond);
1214 return;
1215 }
1216
1218 const Value *BOpOp0, *BOpOp1;
1219 // Compute the effective opcode for Cond, taking into account whether it needs
1220 // to be inverted, e.g.
1221 // and (not (or A, B)), C
1222 // gets lowered as
1223 // and (and (not A, not B), C)
1225 if (BOp) {
1226 BOpc = match(BOp, m_LogicalAnd(m_Value(BOpOp0), m_Value(BOpOp1)))
1227 ? Instruction::And
1228 : (match(BOp, m_LogicalOr(m_Value(BOpOp0), m_Value(BOpOp1)))
1229 ? Instruction::Or
1231 if (InvertCond) {
1232 if (BOpc == Instruction::And)
1233 BOpc = Instruction::Or;
1234 else if (BOpc == Instruction::Or)
1235 BOpc = Instruction::And;
1236 }
1237 }
1238
1239 // If this node is not part of the or/and tree, emit it as a branch.
1240 // Note that all nodes in the tree should have same opcode.
1241 bool BOpIsInOrAndTree = BOpc && BOpc == Opc && BOp->hasOneUse();
1242 if (!BOpIsInOrAndTree || BOp->getParent() != CurBB->getBasicBlock() ||
1243 !isValInBlock(BOpOp0, CurBB->getBasicBlock()) ||
1244 !isValInBlock(BOpOp1, CurBB->getBasicBlock())) {
1245 emitBranchForMergedCondition(Cond, TBB, FBB, CurBB, SwitchBB, TProb, FProb,
1246 InvertCond);
1247 return;
1248 }
1249
1250 // Create TmpBB after CurBB.
1251 MachineFunction::iterator BBI(CurBB);
1252 MachineBasicBlock *TmpBB =
1253 MF->CreateMachineBasicBlock(CurBB->getBasicBlock());
1254 CurBB->getParent()->insert(++BBI, TmpBB);
1255
1256 if (Opc == Instruction::Or) {
1257 // Codegen X | Y as:
1258 // BB1:
1259 // jmp_if_X TBB
1260 // jmp TmpBB
1261 // TmpBB:
1262 // jmp_if_Y TBB
1263 // jmp FBB
1264 //
1265
1266 // We have flexibility in setting Prob for BB1 and Prob for TmpBB.
1267 // The requirement is that
1268 // TrueProb for BB1 + (FalseProb for BB1 * TrueProb for TmpBB)
1269 // = TrueProb for original BB.
1270 // Assuming the original probabilities are A and B, one choice is to set
1271 // BB1's probabilities to A/2 and A/2+B, and set TmpBB's probabilities to
1272 // A/(1+B) and 2B/(1+B). This choice assumes that
1273 // TrueProb for BB1 == FalseProb for BB1 * TrueProb for TmpBB.
1274 // Another choice is to assume TrueProb for BB1 equals to TrueProb for
1275 // TmpBB, but the math is more complicated.
1276
1277 auto NewTrueProb = TProb / 2;
1278 auto NewFalseProb = TProb / 2 + FProb;
1279 // Emit the LHS condition.
1280 findMergedConditions(BOpOp0, TBB, TmpBB, CurBB, SwitchBB, Opc, NewTrueProb,
1281 NewFalseProb, InvertCond);
1282
1283 // Normalize A/2 and B to get A/(1+B) and 2B/(1+B).
1284 SmallVector<BranchProbability, 2> Probs{TProb / 2, FProb};
1286 // Emit the RHS condition into TmpBB.
1287 findMergedConditions(BOpOp1, TBB, FBB, TmpBB, SwitchBB, Opc, Probs[0],
1288 Probs[1], InvertCond);
1289 } else {
1290 assert(Opc == Instruction::And && "Unknown merge op!");
1291 // Codegen X & Y as:
1292 // BB1:
1293 // jmp_if_X TmpBB
1294 // jmp FBB
1295 // TmpBB:
1296 // jmp_if_Y TBB
1297 // jmp FBB
1298 //
1299 // This requires creation of TmpBB after CurBB.
1300
1301 // We have flexibility in setting Prob for BB1 and Prob for TmpBB.
1302 // The requirement is that
1303 // FalseProb for BB1 + (TrueProb for BB1 * FalseProb for TmpBB)
1304 // = FalseProb for original BB.
1305 // Assuming the original probabilities are A and B, one choice is to set
1306 // BB1's probabilities to A+B/2 and B/2, and set TmpBB's probabilities to
1307 // 2A/(1+A) and B/(1+A). This choice assumes that FalseProb for BB1 ==
1308 // TrueProb for BB1 * FalseProb for TmpBB.
1309
1310 auto NewTrueProb = TProb + FProb / 2;
1311 auto NewFalseProb = FProb / 2;
1312 // Emit the LHS condition.
1313 findMergedConditions(BOpOp0, TmpBB, FBB, CurBB, SwitchBB, Opc, NewTrueProb,
1314 NewFalseProb, InvertCond);
1315
1316 // Normalize A and B/2 to get 2A/(1+A) and B/(1+A).
1317 SmallVector<BranchProbability, 2> Probs{TProb, FProb / 2};
1319 // Emit the RHS condition into TmpBB.
1320 findMergedConditions(BOpOp1, TBB, FBB, TmpBB, SwitchBB, Opc, Probs[0],
1321 Probs[1], InvertCond);
1322 }
1323}
1324
1325bool IRTranslatorImpl::shouldEmitAsBranches(
1326 const std::vector<SwitchCG::CaseBlock> &Cases) {
1327 // For multiple cases, it's better to emit as branches.
1328 if (Cases.size() != 2)
1329 return true;
1330
1331 // If this is two comparisons of the same values or'd or and'd together, they
1332 // will get folded into a single comparison, so don't emit two blocks.
1333 if ((Cases[0].CmpLHS == Cases[1].CmpLHS &&
1334 Cases[0].CmpRHS == Cases[1].CmpRHS) ||
1335 (Cases[0].CmpRHS == Cases[1].CmpLHS &&
1336 Cases[0].CmpLHS == Cases[1].CmpRHS)) {
1337 return false;
1338 }
1339
1340 // Handle: (X != null) | (Y != null) --> (X|Y) != 0
1341 // Handle: (X == null) & (Y == null) --> (X|Y) == 0
1342 if (Cases[0].CmpRHS == Cases[1].CmpRHS &&
1343 Cases[0].PredInfo.Pred == Cases[1].PredInfo.Pred &&
1344 isa<Constant>(Cases[0].CmpRHS) &&
1345 cast<Constant>(Cases[0].CmpRHS)->isNullValue()) {
1346 if (Cases[0].PredInfo.Pred == CmpInst::ICMP_EQ &&
1347 Cases[0].TrueBB == Cases[1].ThisBB)
1348 return false;
1349 if (Cases[0].PredInfo.Pred == CmpInst::ICMP_NE &&
1350 Cases[0].FalseBB == Cases[1].ThisBB)
1351 return false;
1352 }
1353
1354 return true;
1355}
1356
1357bool IRTranslatorImpl::translateUncondBr(const User &U,
1358 MachineIRBuilder &MIRBuilder) {
1359 const UncondBrInst &BrInst = cast<UncondBrInst>(U);
1360 auto &CurMBB = MIRBuilder.getMBB();
1361 auto *Succ0MBB = &getMBB(*BrInst.getSuccessor(0));
1362
1363 // If the unconditional target is the layout successor, fallthrough.
1364 if (OptLevel == CodeGenOptLevel::None || !CurMBB.isLayoutSuccessor(Succ0MBB))
1365 MIRBuilder.buildBr(*Succ0MBB);
1366
1367 // Link successors.
1368 for (const BasicBlock *Succ : successors(&BrInst))
1369 CurMBB.addSuccessor(&getMBB(*Succ));
1370 return true;
1371}
1372
1373bool IRTranslatorImpl::translateCondBr(const User &U,
1374 MachineIRBuilder &MIRBuilder) {
1375 const CondBrInst &BrInst = cast<CondBrInst>(U);
1376 auto &CurMBB = MIRBuilder.getMBB();
1377 auto *Succ0MBB = &getMBB(*BrInst.getSuccessor(0));
1378
1379 // If this condition is one of the special cases we handle, do special stuff
1380 // now.
1381 const Value *CondVal = BrInst.getCondition();
1382 MachineBasicBlock *Succ1MBB = &getMBB(*BrInst.getSuccessor(1));
1383
1384 // If this is a series of conditions that are or'd or and'd together, emit
1385 // this as a sequence of branches instead of setcc's with and/or operations.
1386 // As long as jumps are not expensive (exceptions for multi-use logic ops,
1387 // unpredictable branches, and vector extracts because those jumps are likely
1388 // expensive for any target), this should improve performance.
1389 // For example, instead of something like:
1390 // cmp A, B
1391 // C = seteq
1392 // cmp D, E
1393 // F = setle
1394 // or C, F
1395 // jnz foo
1396 // Emit:
1397 // cmp A, B
1398 // je foo
1399 // cmp D, E
1400 // jle foo
1401 using namespace PatternMatch;
1402 const Instruction *CondI = dyn_cast<Instruction>(CondVal);
1403 if (!TLI->isJumpExpensive() && CondI && CondI->hasOneUse() &&
1404 !BrInst.hasMetadata(LLVMContext::MD_unpredictable)) {
1406 Value *Vec;
1407 const Value *BOp0, *BOp1;
1408 if (match(CondI, m_LogicalAnd(m_Value(BOp0), m_Value(BOp1))))
1409 Opcode = Instruction::And;
1410 else if (match(CondI, m_LogicalOr(m_Value(BOp0), m_Value(BOp1))))
1411 Opcode = Instruction::Or;
1412
1413 if (Opcode && !(match(BOp0, m_ExtractElt(m_Value(Vec), m_Value())) &&
1414 match(BOp1, m_ExtractElt(m_Specific(Vec), m_Value())))) {
1415 findMergedConditions(CondI, Succ0MBB, Succ1MBB, &CurMBB, &CurMBB, Opcode,
1416 getEdgeProbability(&CurMBB, Succ0MBB),
1417 getEdgeProbability(&CurMBB, Succ1MBB),
1418 /*InvertCond=*/false);
1419 assert(SL->SwitchCases[0].ThisBB == &CurMBB && "Unexpected lowering!");
1420
1421 // Allow some cases to be rejected.
1422 if (shouldEmitAsBranches(SL->SwitchCases)) {
1423 // Emit the branch for this block.
1424 emitSwitchCase(SL->SwitchCases[0], &CurMBB, *CurBuilder);
1425 SL->SwitchCases.erase(SL->SwitchCases.begin());
1426 return true;
1427 }
1428
1429 // Okay, we decided not to do this, remove any inserted MBB's and clear
1430 // SwitchCases.
1431 for (unsigned I = 1, E = SL->SwitchCases.size(); I != E; ++I)
1432 MF->erase(SL->SwitchCases[I].ThisBB);
1433
1434 SL->SwitchCases.clear();
1435 }
1436 }
1437
1438 // Create a CaseBlock record representing this branch.
1439 SwitchCG::CaseBlock CB(CmpInst::ICMP_EQ, false, CondVal,
1440 ConstantInt::getTrue(MF->getFunction().getContext()),
1441 nullptr, Succ0MBB, Succ1MBB, &CurMBB,
1442 CurBuilder->getDebugLoc());
1443
1444 // Use emitSwitchCase to actually insert the fast branch sequence for this
1445 // cond branch.
1446 emitSwitchCase(CB, &CurMBB, *CurBuilder);
1447 return true;
1448}
1449
1450void IRTranslatorImpl::addSuccessorWithProb(MachineBasicBlock *Src,
1451 MachineBasicBlock *Dst,
1452 BranchProbability Prob) {
1453 if (!FuncInfo.BPI) {
1454 Src->addSuccessorWithoutProb(Dst);
1455 return;
1456 }
1457 if (Prob.isUnknown())
1458 Prob = getEdgeProbability(Src, Dst);
1459 Src->addSuccessor(Dst, Prob);
1460}
1461
1463IRTranslatorImpl::getEdgeProbability(const MachineBasicBlock *Src,
1464 const MachineBasicBlock *Dst) const {
1465 const BasicBlock *SrcBB = Src->getBasicBlock();
1466 const BasicBlock *DstBB = Dst->getBasicBlock();
1467 if (!FuncInfo.BPI) {
1468 // If BPI is not available, set the default probability as 1 / N, where N is
1469 // the number of successors.
1470 auto SuccSize = std::max<uint32_t>(succ_size(SrcBB), 1);
1471 return BranchProbability(1, SuccSize);
1472 }
1473 return FuncInfo.BPI->getEdgeProbability(SrcBB, DstBB);
1474}
1475
1476bool IRTranslatorImpl::translateSwitch(const User &U, MachineIRBuilder &MIB) {
1477 using namespace SwitchCG;
1478 // Extract cases from the switch.
1479 const SwitchInst &SI = cast<SwitchInst>(U);
1480 BranchProbabilityInfo *BPI = FuncInfo.BPI;
1481 CaseClusterVector Clusters;
1482 Clusters.reserve(SI.getNumCases());
1483 for (const auto &I : SI.cases()) {
1484 MachineBasicBlock *Succ = &getMBB(*I.getCaseSuccessor());
1485 assert(Succ && "Could not find successor mbb in mapping");
1486 const ConstantInt *CaseVal = I.getCaseValue();
1487 BranchProbability Prob =
1488 BPI ? BPI->getEdgeProbability(SI.getParent(), I.getSuccessorIndex())
1489 : BranchProbability(1, SI.getNumCases() + 1);
1490 Clusters.push_back(CaseCluster::range(CaseVal, CaseVal, Succ, Prob));
1491 }
1492
1493 MachineBasicBlock *DefaultMBB = &getMBB(*SI.getDefaultDest());
1494
1495 // Cluster adjacent cases with the same destination. We do this at all
1496 // optimization levels because it's cheap to do and will make codegen faster
1497 // if there are many clusters.
1498 sortAndRangeify(Clusters);
1499
1500 MachineBasicBlock *SwitchMBB = &getMBB(*SI.getParent());
1501
1502 // If there is only the default destination, jump there directly.
1503 if (Clusters.empty()) {
1504 SwitchMBB->addSuccessor(DefaultMBB);
1505 if (DefaultMBB != SwitchMBB->getNextNode())
1506 MIB.buildBr(*DefaultMBB);
1507 return true;
1508 }
1509
1510 SL->findJumpTables(Clusters, &SI, std::nullopt, DefaultMBB, nullptr, nullptr);
1511 SL->findBitTestClusters(Clusters, &SI);
1512
1513 LLVM_DEBUG({
1514 dbgs() << "Case clusters: ";
1515 for (const CaseCluster &C : Clusters) {
1516 if (C.Kind == CC_JumpTable)
1517 dbgs() << "JT:";
1518 if (C.Kind == CC_BitTests)
1519 dbgs() << "BT:";
1520
1521 C.Low->getValue().print(dbgs(), true);
1522 if (C.Low != C.High) {
1523 dbgs() << '-';
1524 C.High->getValue().print(dbgs(), true);
1525 }
1526 dbgs() << ' ';
1527 }
1528 dbgs() << '\n';
1529 });
1530
1531 assert(!Clusters.empty());
1532 SwitchWorkList WorkList;
1533 CaseClusterIt First = Clusters.begin();
1534 CaseClusterIt Last = Clusters.end() - 1;
1535 auto DefaultProb = getEdgeProbability(SwitchMBB, DefaultMBB);
1536 WorkList.push_back({SwitchMBB, First, Last, nullptr, nullptr, DefaultProb});
1537
1538 while (!WorkList.empty()) {
1539 SwitchWorkListItem W = WorkList.pop_back_val();
1540
1541 unsigned NumClusters = W.LastCluster - W.FirstCluster + 1;
1542 // For optimized builds, lower large range as a balanced binary tree.
1543 if (NumClusters > 3 &&
1544 MF->getTarget().getOptLevel() != CodeGenOptLevel::None &&
1545 !DefaultMBB->getParent()->getFunction().hasMinSize()) {
1546 splitWorkItem(WorkList, W, SI.getCondition(), SwitchMBB, MIB);
1547 continue;
1548 }
1549
1550 if (!lowerSwitchWorkItem(W, SI.getCondition(), SwitchMBB, DefaultMBB, MIB))
1551 return false;
1552 }
1553 return true;
1554}
1555
1556void IRTranslatorImpl::splitWorkItem(SwitchCG::SwitchWorkList &WorkList,
1558 Value *Cond, MachineBasicBlock *SwitchMBB,
1559 MachineIRBuilder &MIB) {
1560 using namespace SwitchCG;
1561 assert(W.FirstCluster->Low->getValue().slt(W.LastCluster->Low->getValue()) &&
1562 "Clusters not sorted?");
1563 assert(W.LastCluster - W.FirstCluster + 1 >= 2 && "Too small to split!");
1564
1565 auto [LastLeft, FirstRight, LeftProb, RightProb] =
1566 SL->computeSplitWorkItemInfo(W);
1567
1568 // Use the first element on the right as pivot since we will make less-than
1569 // comparisons against it.
1570 CaseClusterIt PivotCluster = FirstRight;
1571 assert(PivotCluster > W.FirstCluster);
1572 assert(PivotCluster <= W.LastCluster);
1573
1574 CaseClusterIt FirstLeft = W.FirstCluster;
1575 CaseClusterIt LastRight = W.LastCluster;
1576
1577 const ConstantInt *Pivot = PivotCluster->Low;
1578
1579 // New blocks will be inserted immediately after the current one.
1581 ++BBI;
1582
1583 // We will branch to the LHS if Value < Pivot. If LHS is a single cluster,
1584 // we can branch to its destination directly if it's squeezed exactly in
1585 // between the known lower bound and Pivot - 1.
1586 MachineBasicBlock *LeftMBB;
1587 if (FirstLeft == LastLeft && FirstLeft->Kind == CC_Range &&
1588 FirstLeft->Low == W.GE &&
1589 (FirstLeft->High->getValue() + 1LL) == Pivot->getValue()) {
1590 LeftMBB = FirstLeft->MBB;
1591 } else {
1592 LeftMBB = FuncInfo.MF->CreateMachineBasicBlock(W.MBB->getBasicBlock());
1593 FuncInfo.MF->insert(BBI, LeftMBB);
1594 WorkList.push_back(
1595 {LeftMBB, FirstLeft, LastLeft, W.GE, Pivot, W.DefaultProb / 2});
1596 }
1597
1598 // Similarly, we will branch to the RHS if Value >= Pivot. If RHS is a
1599 // single cluster, RHS.Low == Pivot, and we can branch to its destination
1600 // directly if RHS.High equals the current upper bound.
1601 MachineBasicBlock *RightMBB;
1602 if (FirstRight == LastRight && FirstRight->Kind == CC_Range && W.LT &&
1603 (FirstRight->High->getValue() + 1ULL) == W.LT->getValue()) {
1604 RightMBB = FirstRight->MBB;
1605 } else {
1606 RightMBB = FuncInfo.MF->CreateMachineBasicBlock(W.MBB->getBasicBlock());
1607 FuncInfo.MF->insert(BBI, RightMBB);
1608 WorkList.push_back(
1609 {RightMBB, FirstRight, LastRight, Pivot, W.LT, W.DefaultProb / 2});
1610 }
1611
1612 // Create the CaseBlock record that will be used to lower the branch.
1613 CaseBlock CB(ICmpInst::Predicate::ICMP_SLT, false, Cond, Pivot, nullptr,
1614 LeftMBB, RightMBB, W.MBB, MIB.getDebugLoc(), LeftProb,
1615 RightProb);
1616
1617 if (W.MBB == SwitchMBB)
1618 emitSwitchCase(CB, SwitchMBB, MIB);
1619 else
1620 SL->SwitchCases.push_back(CB);
1621}
1622
1623void IRTranslatorImpl::emitJumpTable(SwitchCG::JumpTable &JT,
1625 // Emit the code for the jump table
1626 assert(JT.Reg && "Should lower JT Header first!");
1627 MachineIRBuilder MIB(*MBB->getParent());
1628 MIB.setMBB(*MBB);
1629 MIB.setDebugLoc(CurBuilder->getDebugLoc());
1630
1631 Type *PtrIRTy = PointerType::getUnqual(MF->getFunction().getContext());
1632 const LLT PtrTy = getLLTForType(*PtrIRTy, *DL);
1633
1634 auto Table = MIB.buildJumpTable(PtrTy, JT.JTI);
1635 MIB.buildBrJT(Table.getReg(0), JT.JTI, JT.Reg);
1636}
1637
1638bool IRTranslatorImpl::emitJumpTableHeader(SwitchCG::JumpTable &JT,
1640 MachineBasicBlock *HeaderBB) {
1641 MachineIRBuilder MIB(*HeaderBB->getParent());
1642 MIB.setMBB(*HeaderBB);
1643 MIB.setDebugLoc(CurBuilder->getDebugLoc());
1644
1645 const Value &SValue = *JTH.SValue;
1646 // Subtract the lowest switch case value from the value being switched on.
1647 const LLT SwitchTy = getLLTForType(*SValue.getType(), *DL);
1648 Register SwitchOpReg = getOrCreateVReg(SValue);
1649 auto FirstCst = MIB.buildConstant(SwitchTy, JTH.First);
1650 auto Sub = MIB.buildSub({SwitchTy}, SwitchOpReg, FirstCst);
1651
1652 // This value may be smaller or larger than the target's pointer type, and
1653 // therefore require extension or truncating.
1654 auto *PtrIRTy = PointerType::getUnqual(SValue.getContext());
1655 const LLT PtrScalarTy = LLT::integer(DL->getTypeSizeInBits(PtrIRTy));
1656 auto Index = MIB.buildZExtOrTrunc(PtrScalarTy, Sub);
1657
1658 JT.Reg = Index.getReg(0);
1659
1660 if (JTH.FallthroughUnreachable) {
1661 if (JT.MBB != HeaderBB->getNextNode())
1662 MIB.buildBr(*JT.MBB);
1663 return true;
1664 }
1665
1666 // Emit the range check for the jump table, and branch to the default block
1667 // for the switch statement if the value being switched on exceeds the
1668 // largest case in the switch.
1669 auto Cst = getOrCreateVReg(
1670 *ConstantInt::get(SValue.getType(), JTH.Last - JTH.First));
1671 auto Cmp = MIB.buildICmp(CmpInst::ICMP_UGT, LLT::integer(1), Sub, Cst);
1672
1673 auto BrCond = MIB.buildBrCond(Cmp.getReg(0), *JT.Default);
1674
1675 // Avoid emitting unnecessary branches to the next block.
1676 if (JT.MBB != HeaderBB->getNextNode())
1677 BrCond = MIB.buildBr(*JT.MBB);
1678 return true;
1679}
1680
1681void IRTranslatorImpl::emitSwitchCase(SwitchCG::CaseBlock &CB,
1682 MachineBasicBlock *SwitchBB,
1683 MachineIRBuilder &MIB) {
1684 Register CondLHS = getOrCreateVReg(*CB.CmpLHS);
1685 Register Cond;
1686 DebugLoc OldDbgLoc = MIB.getDebugLoc();
1687 MIB.setDebugLoc(CB.DbgLoc);
1688 MIB.setMBB(*CB.ThisBB);
1689
1690 if (CB.PredInfo.NoCmp) {
1691 // Branch or fall through to TrueBB.
1692 addSuccessorWithProb(CB.ThisBB, CB.TrueBB, CB.TrueProb);
1693 addMachineCFGPred({SwitchBB->getBasicBlock(), CB.TrueBB->getBasicBlock()},
1694 CB.ThisBB);
1696 if (CB.TrueBB != CB.ThisBB->getNextNode())
1697 MIB.buildBr(*CB.TrueBB);
1698 MIB.setDebugLoc(OldDbgLoc);
1699 return;
1700 }
1701
1702 const LLT i1Ty = LLT::integer(1);
1703 // Build the compare.
1704 if (!CB.CmpMHS) {
1705 const auto *CI = dyn_cast<ConstantInt>(CB.CmpRHS);
1706 // For conditional branch lowering, we might try to do something silly like
1707 // emit an G_ICMP to compare an existing G_ICMP i1 result with true. If so,
1708 // just re-use the existing condition vreg.
1709 if (MRI->getType(CondLHS).getSizeInBits() == 1 && CI && CI->isOne() &&
1711 Cond = CondLHS;
1712 } else {
1713 Register CondRHS = getOrCreateVReg(*CB.CmpRHS);
1715 Cond =
1716 MIB.buildFCmp(CB.PredInfo.Pred, i1Ty, CondLHS, CondRHS).getReg(0);
1717 else
1718 Cond =
1719 MIB.buildICmp(CB.PredInfo.Pred, i1Ty, CondLHS, CondRHS).getReg(0);
1720 }
1721 } else {
1723 "Can only handle SLE ranges");
1724
1725 const APInt& Low = cast<ConstantInt>(CB.CmpLHS)->getValue();
1726 const APInt& High = cast<ConstantInt>(CB.CmpRHS)->getValue();
1727
1728 Register CmpOpReg = getOrCreateVReg(*CB.CmpMHS);
1729 if (cast<ConstantInt>(CB.CmpLHS)->isMinValue(true)) {
1730 Register CondRHS = getOrCreateVReg(*CB.CmpRHS);
1731 Cond =
1732 MIB.buildICmp(CmpInst::ICMP_SLE, i1Ty, CmpOpReg, CondRHS).getReg(0);
1733 } else {
1734 const LLT CmpTy = MRI->getType(CmpOpReg);
1735 auto Sub = MIB.buildSub({CmpTy}, CmpOpReg, CondLHS);
1736 auto Diff = MIB.buildConstant(CmpTy, High - Low);
1737 Cond = MIB.buildICmp(CmpInst::ICMP_ULE, i1Ty, Sub, Diff).getReg(0);
1738 }
1739 }
1740
1741 // Update successor info
1742 addSuccessorWithProb(CB.ThisBB, CB.TrueBB, CB.TrueProb);
1743
1744 addMachineCFGPred({SwitchBB->getBasicBlock(), CB.TrueBB->getBasicBlock()},
1745 CB.ThisBB);
1746
1747 // TrueBB and FalseBB are always different unless the incoming IR is
1748 // degenerate. This only happens when running llc on weird IR.
1749 if (CB.TrueBB != CB.FalseBB)
1750 addSuccessorWithProb(CB.ThisBB, CB.FalseBB, CB.FalseProb);
1752
1753 addMachineCFGPred({SwitchBB->getBasicBlock(), CB.FalseBB->getBasicBlock()},
1754 CB.ThisBB);
1755
1756 MIB.buildBrCond(Cond, *CB.TrueBB);
1757 MIB.buildBr(*CB.FalseBB);
1758 MIB.setDebugLoc(OldDbgLoc);
1759}
1760
1761bool IRTranslatorImpl::lowerJumpTableWorkItem(
1763 MachineBasicBlock *CurMBB, MachineBasicBlock *DefaultMBB,
1766 MachineBasicBlock *Fallthrough, bool FallthroughUnreachable) {
1767 using namespace SwitchCG;
1768 MachineFunction *CurMF = SwitchMBB->getParent();
1769 // FIXME: Optimize away range check based on pivot comparisons.
1770 JumpTableHeader *JTH = &SL->JTCases[I->JTCasesIndex].first;
1771 SwitchCG::JumpTable *JT = &SL->JTCases[I->JTCasesIndex].second;
1772 BranchProbability DefaultProb = W.DefaultProb;
1773
1774 // The jump block hasn't been inserted yet; insert it here.
1775 MachineBasicBlock *JumpMBB = JT->MBB;
1776 CurMF->insert(BBI, JumpMBB);
1777
1778 // Since the jump table block is separate from the switch block, we need
1779 // to keep track of it as a machine predecessor to the default block,
1780 // otherwise we lose the phi edges.
1781 addMachineCFGPred({SwitchMBB->getBasicBlock(), DefaultMBB->getBasicBlock()},
1782 CurMBB);
1783 addMachineCFGPred({SwitchMBB->getBasicBlock(), DefaultMBB->getBasicBlock()},
1784 JumpMBB);
1785
1786 auto JumpProb = I->Prob;
1787 auto FallthroughProb = UnhandledProbs;
1788
1789 // If the default statement is a target of the jump table, we evenly
1790 // distribute the default probability to successors of CurMBB. Also
1791 // update the probability on the edge from JumpMBB to Fallthrough.
1792 for (MachineBasicBlock::succ_iterator SI = JumpMBB->succ_begin(),
1793 SE = JumpMBB->succ_end();
1794 SI != SE; ++SI) {
1795 if (*SI == DefaultMBB) {
1796 JumpProb += DefaultProb / 2;
1797 FallthroughProb -= DefaultProb / 2;
1798 JumpMBB->setSuccProbability(SI, DefaultProb / 2);
1799 JumpMBB->normalizeSuccProbs();
1800 } else {
1801 // Also record edges from the jump table block to it's successors.
1802 addMachineCFGPred({SwitchMBB->getBasicBlock(), (*SI)->getBasicBlock()},
1803 JumpMBB);
1804 }
1805 }
1806
1807 if (FallthroughUnreachable)
1808 JTH->FallthroughUnreachable = true;
1809
1810 if (!JTH->FallthroughUnreachable)
1811 addSuccessorWithProb(CurMBB, Fallthrough, FallthroughProb);
1812 addSuccessorWithProb(CurMBB, JumpMBB, JumpProb);
1813 CurMBB->normalizeSuccProbs();
1814
1815 // The jump table header will be inserted in our current block, do the
1816 // range check, and fall through to our fallthrough block.
1817 JTH->HeaderBB = CurMBB;
1818 JT->Default = Fallthrough; // FIXME: Move Default to JumpTableHeader.
1819
1820 // If we're in the right place, emit the jump table header right now.
1821 if (CurMBB == SwitchMBB) {
1822 if (!emitJumpTableHeader(*JT, *JTH, CurMBB))
1823 return false;
1824 JTH->Emitted = true;
1825 }
1826 return true;
1827}
1828bool IRTranslatorImpl::lowerSwitchRangeWorkItem(
1830 bool FallthroughUnreachable, BranchProbability UnhandledProbs,
1831 MachineBasicBlock *CurMBB, MachineIRBuilder &MIB,
1832 MachineBasicBlock *SwitchMBB) {
1833 using namespace SwitchCG;
1834 const Value *RHS, *LHS, *MHS;
1835 CmpInst::Predicate Pred;
1836 if (I->Low == I->High) {
1837 // Check Cond == I->Low.
1838 Pred = CmpInst::ICMP_EQ;
1839 LHS = Cond;
1840 RHS = I->Low;
1841 MHS = nullptr;
1842 } else {
1843 // Check I->Low <= Cond <= I->High.
1844 Pred = CmpInst::ICMP_SLE;
1845 LHS = I->Low;
1846 MHS = Cond;
1847 RHS = I->High;
1848 }
1849
1850 // If Fallthrough is unreachable, fold away the comparison.
1851 // The false probability is the sum of all unhandled cases.
1852 CaseBlock CB(Pred, FallthroughUnreachable, LHS, RHS, MHS, I->MBB, Fallthrough,
1853 CurMBB, MIB.getDebugLoc(), I->Prob, UnhandledProbs);
1854
1855 emitSwitchCase(CB, SwitchMBB, MIB);
1856 return true;
1857}
1858
1859void IRTranslatorImpl::emitBitTestHeader(SwitchCG::BitTestBlock &B,
1860 MachineBasicBlock *SwitchBB) {
1861 MachineIRBuilder &MIB = *CurBuilder;
1862 MIB.setMBB(*SwitchBB);
1863
1864 // Subtract the minimum value.
1865 Register SwitchOpReg = getOrCreateVReg(*B.SValue);
1866
1867 LLT SwitchOpTy = MRI->getType(SwitchOpReg);
1868 Register MinValReg = MIB.buildConstant(SwitchOpTy, B.First).getReg(0);
1869 auto RangeSub = MIB.buildSub(SwitchOpTy, SwitchOpReg, MinValReg);
1870
1871 Type *PtrIRTy = PointerType::getUnqual(MF->getFunction().getContext());
1872 const LLT PtrTy = getLLTForType(*PtrIRTy, *DL);
1873
1874 LLT MaskTy = SwitchOpTy;
1875 if (MaskTy.getSizeInBits() > PtrTy.getSizeInBits() ||
1877 MaskTy = LLT::integer(PtrTy.getSizeInBits());
1878 else {
1879 // Ensure that the type will fit the mask value.
1880 for (const SwitchCG::BitTestCase &Case : B.Cases) {
1881 if (!isUIntN(SwitchOpTy.getSizeInBits(), Case.Mask)) {
1882 // Switch table case range are encoded into series of masks.
1883 // Just use pointer type, it's guaranteed to fit.
1884 MaskTy = LLT::integer(PtrTy.getSizeInBits());
1885 break;
1886 }
1887 }
1888 }
1889 Register SubReg = RangeSub.getReg(0);
1890 if (SwitchOpTy != MaskTy)
1891 SubReg = MIB.buildZExtOrTrunc(MaskTy, SubReg).getReg(0);
1892
1893 B.RegVT = getMVTForLLT(MaskTy);
1894 B.Reg = SubReg;
1895
1896 MachineBasicBlock *MBB = B.Cases[0].ThisBB;
1897
1898 if (!B.FallthroughUnreachable)
1899 addSuccessorWithProb(SwitchBB, B.Default, B.DefaultProb);
1900 addSuccessorWithProb(SwitchBB, MBB, B.Prob);
1901
1902 SwitchBB->normalizeSuccProbs();
1903
1904 if (!B.FallthroughUnreachable) {
1905 // Conditional branch to the default block.
1906 auto RangeCst = MIB.buildConstant(SwitchOpTy, B.Range);
1907 auto RangeCmp = MIB.buildICmp(CmpInst::Predicate::ICMP_UGT, LLT::integer(1),
1908 RangeSub, RangeCst);
1909 MIB.buildBrCond(RangeCmp, *B.Default);
1910 }
1911
1912 // Avoid emitting unnecessary branches to the next block.
1913 if (MBB != SwitchBB->getNextNode())
1914 MIB.buildBr(*MBB);
1915}
1916
1917void IRTranslatorImpl::emitBitTestCase(SwitchCG::BitTestBlock &BB,
1918 MachineBasicBlock *NextMBB,
1919 BranchProbability BranchProbToNext,
1921 MachineBasicBlock *SwitchBB) {
1922 MachineIRBuilder &MIB = *CurBuilder;
1923 MIB.setMBB(*SwitchBB);
1924
1925 LLT SwitchTy = getLLTForMVT(BB.RegVT);
1926 Register Cmp;
1927 unsigned PopCount = llvm::popcount(B.Mask);
1928 if (PopCount == 1) {
1929 // Testing for a single bit; just compare the shift count with what it
1930 // would need to be to shift a 1 bit in that position.
1931 auto MaskTrailingZeros =
1932 MIB.buildConstant(SwitchTy, llvm::countr_zero(B.Mask));
1934 MaskTrailingZeros)
1935 .getReg(0);
1936 } else if (PopCount == BB.Range) {
1937 // There is only one zero bit in the range, test for it directly.
1938 auto MaskTrailingOnes =
1939 MIB.buildConstant(SwitchTy, llvm::countr_one(B.Mask));
1940 Cmp =
1941 MIB.buildICmp(CmpInst::ICMP_NE, LLT::integer(1), Reg, MaskTrailingOnes)
1942 .getReg(0);
1943 } else {
1944 // Make desired shift.
1945 auto CstOne = MIB.buildConstant(SwitchTy, 1);
1946 auto SwitchVal = MIB.buildShl(SwitchTy, CstOne, Reg);
1947
1948 // Emit bit tests and jumps.
1949 auto CstMask = MIB.buildConstant(SwitchTy, B.Mask);
1950 auto AndOp = MIB.buildAnd(SwitchTy, SwitchVal, CstMask);
1951 auto CstZero = MIB.buildConstant(SwitchTy, 0);
1952 Cmp = MIB.buildICmp(CmpInst::ICMP_NE, LLT::integer(1), AndOp, CstZero)
1953 .getReg(0);
1954 }
1955
1956 // The branch probability from SwitchBB to B.TargetBB is B.ExtraProb.
1957 addSuccessorWithProb(SwitchBB, B.TargetBB, B.ExtraProb);
1958 // The branch probability from SwitchBB to NextMBB is BranchProbToNext.
1959 addSuccessorWithProb(SwitchBB, NextMBB, BranchProbToNext);
1960 // It is not guaranteed that the sum of B.ExtraProb and BranchProbToNext is
1961 // one as they are relative probabilities (and thus work more like weights),
1962 // and hence we need to normalize them to let the sum of them become one.
1963 SwitchBB->normalizeSuccProbs();
1964
1965 // Record the fact that the IR edge from the header to the bit test target
1966 // will go through our new block. Neeeded for PHIs to have nodes added.
1967 addMachineCFGPred({BB.Parent->getBasicBlock(), B.TargetBB->getBasicBlock()},
1968 SwitchBB);
1969
1970 MIB.buildBrCond(Cmp, *B.TargetBB);
1971
1972 // Avoid emitting unnecessary branches to the next block.
1973 if (NextMBB != SwitchBB->getNextNode())
1974 MIB.buildBr(*NextMBB);
1975}
1976
1977bool IRTranslatorImpl::lowerBitTestWorkItem(
1979 MachineBasicBlock *CurMBB, MachineBasicBlock *DefaultMBB,
1981 BranchProbability DefaultProb, BranchProbability UnhandledProbs,
1983 bool FallthroughUnreachable) {
1984 using namespace SwitchCG;
1985 MachineFunction *CurMF = SwitchMBB->getParent();
1986 // FIXME: Optimize away range check based on pivot comparisons.
1987 BitTestBlock *BTB = &SL->BitTestCases[I->BTCasesIndex];
1988 // The bit test blocks haven't been inserted yet; insert them here.
1989 for (BitTestCase &BTC : BTB->Cases)
1990 CurMF->insert(BBI, BTC.ThisBB);
1991
1992 // Fill in fields of the BitTestBlock.
1993 BTB->Parent = CurMBB;
1994 BTB->Default = Fallthrough;
1995
1996 BTB->DefaultProb = UnhandledProbs;
1997 // If the cases in bit test don't form a contiguous range, we evenly
1998 // distribute the probability on the edge to Fallthrough to two
1999 // successors of CurMBB.
2000 if (!BTB->ContiguousRange) {
2001 BTB->Prob += DefaultProb / 2;
2002 BTB->DefaultProb -= DefaultProb / 2;
2003 }
2004
2005 if (FallthroughUnreachable)
2006 BTB->FallthroughUnreachable = true;
2007
2008 // If we're in the right place, emit the bit test header right now.
2009 if (CurMBB == SwitchMBB) {
2010 emitBitTestHeader(*BTB, SwitchMBB);
2011 BTB->Emitted = true;
2012 }
2013 return true;
2014}
2015
2016bool IRTranslatorImpl::lowerSwitchWorkItem(SwitchCG::SwitchWorkListItem W,
2017 Value *Cond,
2018 MachineBasicBlock *SwitchMBB,
2019 MachineBasicBlock *DefaultMBB,
2020 MachineIRBuilder &MIB) {
2021 using namespace SwitchCG;
2022 MachineFunction *CurMF = FuncInfo.MF;
2023 MachineBasicBlock *NextMBB = nullptr;
2025 if (++BBI != FuncInfo.MF->end())
2026 NextMBB = &*BBI;
2027
2028 if (EnableOpts) {
2029 // Here, we order cases by probability so the most likely case will be
2030 // checked first. However, two clusters can have the same probability in
2031 // which case their relative ordering is non-deterministic. So we use Low
2032 // as a tie-breaker as clusters are guaranteed to never overlap.
2033 llvm::sort(W.FirstCluster, W.LastCluster + 1,
2034 [](const CaseCluster &a, const CaseCluster &b) {
2035 return a.Prob != b.Prob
2036 ? a.Prob > b.Prob
2037 : a.Low->getValue().slt(b.Low->getValue());
2038 });
2039
2040 // Rearrange the case blocks so that the last one falls through if possible
2041 // without changing the order of probabilities.
2042 for (CaseClusterIt I = W.LastCluster; I > W.FirstCluster;) {
2043 --I;
2044 if (I->Prob > W.LastCluster->Prob)
2045 break;
2046 if (I->Kind == CC_Range && I->MBB == NextMBB) {
2047 std::swap(*I, *W.LastCluster);
2048 break;
2049 }
2050 }
2051 }
2052
2053 // Compute total probability.
2054 BranchProbability DefaultProb = W.DefaultProb;
2055 BranchProbability UnhandledProbs = DefaultProb;
2056 for (CaseClusterIt I = W.FirstCluster; I <= W.LastCluster; ++I)
2057 UnhandledProbs += I->Prob;
2058
2059 MachineBasicBlock *CurMBB = W.MBB;
2060 for (CaseClusterIt I = W.FirstCluster, E = W.LastCluster; I <= E; ++I) {
2061 bool FallthroughUnreachable = false;
2062 MachineBasicBlock *Fallthrough;
2063 if (I == W.LastCluster) {
2064 // For the last cluster, fall through to the default destination.
2065 Fallthrough = DefaultMBB;
2066 FallthroughUnreachable = isa<UnreachableInst>(
2067 DefaultMBB->getBasicBlock()->getFirstNonPHIOrDbg());
2068 } else {
2069 Fallthrough = CurMF->CreateMachineBasicBlock(CurMBB->getBasicBlock());
2070 CurMF->insert(BBI, Fallthrough);
2071 }
2072 UnhandledProbs -= I->Prob;
2073
2074 switch (I->Kind) {
2075 case CC_BitTests: {
2076 if (!lowerBitTestWorkItem(W, SwitchMBB, CurMBB, DefaultMBB, MIB, BBI,
2077 DefaultProb, UnhandledProbs, I, Fallthrough,
2078 FallthroughUnreachable)) {
2079 LLVM_DEBUG(dbgs() << "Failed to lower bit test for switch");
2080 return false;
2081 }
2082 break;
2083 }
2084
2085 case CC_JumpTable: {
2086 if (!lowerJumpTableWorkItem(W, SwitchMBB, CurMBB, DefaultMBB, MIB, BBI,
2087 UnhandledProbs, I, Fallthrough,
2088 FallthroughUnreachable)) {
2089 LLVM_DEBUG(dbgs() << "Failed to lower jump table");
2090 return false;
2091 }
2092 break;
2093 }
2094 case CC_Range: {
2095 if (!lowerSwitchRangeWorkItem(I, Cond, Fallthrough,
2096 FallthroughUnreachable, UnhandledProbs,
2097 CurMBB, MIB, SwitchMBB)) {
2098 LLVM_DEBUG(dbgs() << "Failed to lower switch range");
2099 return false;
2100 }
2101 break;
2102 }
2103 }
2104 CurMBB = Fallthrough;
2105 }
2106
2107 return true;
2108}
2109
2110bool IRTranslatorImpl::translateIndirectBr(const User &U,
2111 MachineIRBuilder &MIRBuilder) {
2112 const IndirectBrInst &BrInst = cast<IndirectBrInst>(U);
2113
2114 const Register Tgt = getOrCreateVReg(*BrInst.getAddress());
2115 MIRBuilder.buildBrIndirect(Tgt);
2116
2117 // Link successors.
2118 SmallPtrSet<const BasicBlock *, 32> AddedSuccessors;
2119 MachineBasicBlock &CurBB = MIRBuilder.getMBB();
2120 for (const BasicBlock *Succ : successors(&BrInst)) {
2121 // It's legal for indirectbr instructions to have duplicate blocks in the
2122 // destination list. We don't allow this in MIR. Skip anything that's
2123 // already a successor.
2124 if (!AddedSuccessors.insert(Succ).second)
2125 continue;
2126 CurBB.addSuccessor(&getMBB(*Succ));
2127 }
2128
2129 return true;
2130}
2131
2132static bool isSwiftError(const Value *V) {
2133 if (auto Arg = dyn_cast<Argument>(V))
2134 return Arg->hasSwiftErrorAttr();
2135 if (auto AI = dyn_cast<AllocaInst>(V))
2136 return AI->isSwiftError();
2137 return false;
2138}
2139
2140bool IRTranslatorImpl::translateLoad(const User &U,
2141 MachineIRBuilder &MIRBuilder) {
2142 const LoadInst &LI = cast<LoadInst>(U);
2143 TypeSize StoreSize = DL->getTypeStoreSize(LI.getType());
2144 if (StoreSize.isZero())
2145 return true;
2146
2147 ArrayRef<Register> Regs = getOrCreateVRegs(LI);
2148 Register Base = getOrCreateVReg(*LI.getPointerOperand());
2149 AAMDNodes AAInfo = LI.getAAMetadata();
2150
2151 const Value *Ptr = LI.getPointerOperand();
2152
2153 if (CLI->supportSwiftError() && isSwiftError(Ptr)) {
2154 assert(Regs.size() == 1 && "swifterror should be single pointer");
2155 Register VReg =
2156 SwiftError.getOrCreateVRegUseAt(&LI, &MIRBuilder.getMBB(), Ptr);
2157 MIRBuilder.buildCopy(Regs[0], VReg);
2158 return true;
2159 }
2160
2162 TLI->getLoadMemOperandFlags(LI, *DL, AC, LibInfo, OptLevel);
2163 if (AA && !(Flags & MachineMemOperand::MOInvariant)) {
2164 if (AA->pointsToConstantMemory(
2165 MemoryLocation(Ptr, LocationSize::precise(StoreSize), AAInfo))) {
2167 }
2168 }
2169
2170 // Fast-path the common single-register load.
2171 if (Regs.size() == 1) {
2172 auto *MMO = MF->getMachineMemOperand(
2173 MachinePointerInfo(LI.getPointerOperand()), Flags,
2174 MRI->getType(Regs[0]), getMemOpAlign(LI),
2175 MMOMetadata(AAInfo, LI.getMetadata(LLVMContext::MD_range)),
2176 LI.getSyncScopeID(), LI.getOrdering());
2177 MIRBuilder.buildLoad(Regs[0], Base, *MMO);
2178 return true;
2179 }
2180
2181 ArrayRef<uint64_t> Offsets = *VMap.getOffsets(LI);
2182 Type *OffsetIRTy = DL->getIndexType(Ptr->getType());
2183 LLT OffsetTy = getLLTForType(*OffsetIRTy, *DL);
2184 for (unsigned i = 0; i < Regs.size(); ++i) {
2185 Register Addr;
2186 MIRBuilder.materializeObjectPtrOffset(Addr, Base, OffsetTy, Offsets[i]);
2187
2188 MachinePointerInfo Ptr(LI.getPointerOperand(), Offsets[i]);
2189 Align BaseAlign = getMemOpAlign(LI);
2190 auto *MMO =
2191 MF->getMachineMemOperand(Ptr, Flags, MRI->getType(Regs[i]),
2192 commonAlignment(BaseAlign, Offsets[i]), AAInfo,
2193 LI.getSyncScopeID(), LI.getOrdering());
2194 MIRBuilder.buildLoad(Regs[i], Addr, *MMO);
2195 }
2196
2197 return true;
2198}
2199
2200bool IRTranslatorImpl::translateStore(const User &U,
2201 MachineIRBuilder &MIRBuilder) {
2202 const StoreInst &SI = cast<StoreInst>(U);
2203 if (DL->getTypeStoreSize(SI.getValueOperand()->getType()).isZero())
2204 return true;
2205
2206 ArrayRef<Register> Vals = getOrCreateVRegs(*SI.getValueOperand());
2207 Register Base = getOrCreateVReg(*SI.getPointerOperand());
2208
2209 if (CLI->supportSwiftError() && isSwiftError(SI.getPointerOperand())) {
2210 assert(Vals.size() == 1 && "swifterror should be single pointer");
2211
2212 Register VReg = SwiftError.getOrCreateVRegDefAt(&SI, &MIRBuilder.getMBB(),
2213 SI.getPointerOperand());
2214 MIRBuilder.buildCopy(VReg, Vals[0]);
2215 return true;
2216 }
2217
2218 MachineMemOperand::Flags Flags = TLI->getStoreMemOperandFlags(SI, *DL);
2219 // Fast-path the common single-register store.
2220 if (Vals.size() == 1) {
2221 auto *MMO = MF->getMachineMemOperand(
2222 MachinePointerInfo(SI.getPointerOperand()), Flags,
2223 MRI->getType(Vals[0]), getMemOpAlign(SI), SI.getAAMetadata(),
2224 SI.getSyncScopeID(), SI.getOrdering());
2225 MIRBuilder.buildStore(Vals[0], Base, *MMO);
2226 return true;
2227 }
2228
2229 ArrayRef<uint64_t> Offsets = *VMap.getOffsets(*SI.getValueOperand());
2230 Type *OffsetIRTy = DL->getIndexType(SI.getPointerOperandType());
2231 LLT OffsetTy = getLLTForType(*OffsetIRTy, *DL);
2232 for (unsigned i = 0; i < Vals.size(); ++i) {
2233 Register Addr;
2234 MIRBuilder.materializeObjectPtrOffset(Addr, Base, OffsetTy, Offsets[i]);
2235
2236 MachinePointerInfo Ptr(SI.getPointerOperand(), Offsets[i]);
2237 Align BaseAlign = getMemOpAlign(SI);
2238 auto *MMO = MF->getMachineMemOperand(Ptr, Flags, MRI->getType(Vals[i]),
2239 commonAlignment(BaseAlign, Offsets[i]),
2240 SI.getAAMetadata(),
2241 SI.getSyncScopeID(), SI.getOrdering());
2242 MIRBuilder.buildStore(Vals[i], Addr, *MMO);
2243 }
2244 return true;
2245}
2246
2248 const Value *Src = U.getOperand(0);
2249 Type *Int32Ty = Type::getInt32Ty(U.getContext());
2250
2251 // getIndexedOffsetInType is designed for GEPs, so the first index is the
2252 // usual array element rather than looking into the actual aggregate.
2254 Indices.push_back(ConstantInt::get(Int32Ty, 0));
2255
2256 if (const ExtractValueInst *EVI = dyn_cast<ExtractValueInst>(&U)) {
2257 for (auto Idx : EVI->indices())
2258 Indices.push_back(ConstantInt::get(Int32Ty, Idx));
2259 } else if (const InsertValueInst *IVI = dyn_cast<InsertValueInst>(&U)) {
2260 for (auto Idx : IVI->indices())
2261 Indices.push_back(ConstantInt::get(Int32Ty, Idx));
2262 } else {
2263 llvm::append_range(Indices, drop_begin(U.operands()));
2264 }
2265
2266 return static_cast<uint64_t>(
2267 DL.getIndexedOffsetInType(Src->getType(), Indices));
2268}
2269
2270bool IRTranslatorImpl::translateExtractValue(const User &U,
2271 MachineIRBuilder &MIRBuilder) {
2272 const Value *Src = U.getOperand(0);
2274 ArrayRef<Register> SrcRegs = getOrCreateVRegs(*Src);
2275 ArrayRef<uint64_t> Offsets = *VMap.getOffsets(*Src);
2276 unsigned Idx = llvm::lower_bound(Offsets, Offset) - Offsets.begin();
2277 auto &DstRegs = allocateVRegs(U);
2278
2279 for (unsigned i = 0; i < DstRegs.size(); ++i)
2280 DstRegs[i] = SrcRegs[Idx++];
2281
2282 return true;
2283}
2284
2285bool IRTranslatorImpl::translateInsertValue(const User &U,
2286 MachineIRBuilder &MIRBuilder) {
2287 const Value *Src = U.getOperand(0);
2289 auto &DstRegs = allocateVRegs(U);
2290 ArrayRef<uint64_t> DstOffsets = *VMap.getOffsets(U);
2291 ArrayRef<Register> SrcRegs = getOrCreateVRegs(*Src);
2292 ArrayRef<Register> InsertedRegs = getOrCreateVRegs(*U.getOperand(1));
2293 auto *InsertedIt = InsertedRegs.begin();
2294
2295 for (unsigned i = 0; i < DstRegs.size(); ++i) {
2296 if (DstOffsets[i] >= Offset && InsertedIt != InsertedRegs.end())
2297 DstRegs[i] = *InsertedIt++;
2298 else
2299 DstRegs[i] = SrcRegs[i];
2300 }
2301
2302 return true;
2303}
2304
2305bool IRTranslatorImpl::translateSelect(const User &U,
2306 MachineIRBuilder &MIRBuilder) {
2307 Register Tst = getOrCreateVReg(*U.getOperand(0));
2308 ArrayRef<Register> ResRegs = getOrCreateVRegs(U);
2309 ArrayRef<Register> Op0Regs = getOrCreateVRegs(*U.getOperand(1));
2310 ArrayRef<Register> Op1Regs = getOrCreateVRegs(*U.getOperand(2));
2311
2312 uint32_t Flags = 0;
2313 if (const SelectInst *SI = dyn_cast<SelectInst>(&U))
2315
2316 for (unsigned i = 0; i < ResRegs.size(); ++i) {
2317 MIRBuilder.buildSelect(ResRegs[i], Tst, Op0Regs[i], Op1Regs[i], Flags);
2318 }
2319
2320 return true;
2321}
2322
2323bool IRTranslatorImpl::translateCopy(const User &U, const Value &V,
2324 MachineIRBuilder &MIRBuilder) {
2325 return translateCopy(U, getOrCreateVReg(V), MIRBuilder);
2326}
2327
2328bool IRTranslatorImpl::translateCopy(const User &U, Register Src,
2329 MachineIRBuilder &MIRBuilder) {
2330 auto &Regs = *VMap.getVRegs(U);
2331 if (Regs.empty()) {
2332 Regs.push_back(Src);
2333 } else {
2334 // If we already assigned a vreg for this instruction, we can't change that.
2335 // Emit a copy to satisfy the users we already emitted.
2336 MIRBuilder.buildCopy(Regs[0], Src);
2337 }
2338 return true;
2339}
2340
2341bool IRTranslatorImpl::translateBitCast(const User &U,
2342 MachineIRBuilder &MIRBuilder) {
2343 Type *SrcTy = U.getOperand(0)->getType();
2344 Type *DstTy = U.getType();
2345
2346 // If we're bitcasting to the source type, we can reuse the source vreg.
2347 if (getLLTForType(*SrcTy, *DL) == getLLTForType(*DstTy, *DL)) {
2348 // If the source is a ConstantInt then it was probably created by
2349 // ConstantHoisting and we should leave it alone.
2350 if (isa<ConstantInt>(U.getOperand(0)))
2351 return translateCast(TargetOpcode::G_CONSTANT_FOLD_BARRIER, U,
2352 MIRBuilder);
2353 return translateCopy(U, *U.getOperand(0), MIRBuilder);
2354 }
2355
2356 // The IR only allows pointer/non-pointer bitcasts with byte types, but
2357 // G_BITCAST can't convert between pointers and other types. Go through an
2358 // integer with the pointer's shape instead: `bitcast <2 x b32> to ptr`
2359 // becomes a G_BITCAST to i64 and a G_INTTOPTR.
2360 if (SrcTy->isPtrOrPtrVectorTy() != DstTy->isPtrOrPtrVectorTy()) {
2361 assert((SrcTy->isByteOrByteVectorTy() || DstTy->isByteOrByteVectorTy()) &&
2362 "only byte types can be bitcast to or from pointers");
2363 Type *PtrIRTy = SrcTy->isPtrOrPtrVectorTy() ? SrcTy : DstTy;
2364 LLT IntTy = getLLTForType(*DL->getIntPtrType(PtrIRTy), *DL);
2365 Register Src = getOrCreateVReg(*U.getOperand(0));
2366 Register Dst = getOrCreateVReg(U);
2367 if (MRI->getType(Src) != IntTy && MRI->getType(Dst) != IntTy)
2368 Src = MIRBuilder.buildCast(IntTy, Src).getReg(0);
2369 MIRBuilder.buildCast(Dst, Src);
2370 return true;
2371 }
2372
2373 return translateCast(TargetOpcode::G_BITCAST, U, MIRBuilder);
2374}
2375
2376bool IRTranslatorImpl::translateCast(unsigned Opcode, const User &U,
2377 MachineIRBuilder &MIRBuilder) {
2378 if (!mayTranslateUserTypes(U))
2379 return false;
2380
2381 uint32_t Flags = 0;
2382 if (const Instruction *I = dyn_cast<Instruction>(&U))
2384
2385 Register Op = getOrCreateVReg(*U.getOperand(0));
2386 Register Res = getOrCreateVReg(U);
2387 MIRBuilder.buildInstr(Opcode, {Res}, {Op}, Flags);
2388 return true;
2389}
2390
2391bool IRTranslatorImpl::translateGetElementPtr(const User &U,
2392 MachineIRBuilder &MIRBuilder) {
2393 Value &Op0 = *U.getOperand(0);
2394 Register BaseReg = getOrCreateVReg(Op0);
2395 Type *PtrIRTy = Op0.getType();
2396 LLT PtrTy = getLLTForType(*PtrIRTy, *DL);
2397 Type *OffsetIRTy = DL->getIndexType(PtrIRTy);
2398 LLT OffsetTy = getLLTForType(*OffsetIRTy, *DL);
2399
2400 uint32_t PtrAddFlags = 0;
2401 // Each PtrAdd generated to implement the GEP inherits its nuw, nusw, inbounds
2402 // flags.
2403 if (const Instruction *I = dyn_cast<Instruction>(&U))
2405
2406 auto PtrAddFlagsWithConst = [&](int64_t Offset) {
2407 // For nusw/inbounds GEP with an offset that is nonnegative when interpreted
2408 // as signed, assume there is no unsigned overflow.
2409 if (Offset >= 0 && (PtrAddFlags & MachineInstr::MIFlag::NoUSWrap))
2410 return PtrAddFlags | MachineInstr::MIFlag::NoUWrap;
2411 return PtrAddFlags;
2412 };
2413
2414 // Normalize Vector GEP - all scalar operands should be converted to the
2415 // splat vector.
2416 unsigned VectorWidth = 0;
2417
2418 // True if we should use a splat vector; using VectorWidth alone is not
2419 // sufficient.
2420 bool WantSplatVector = false;
2421 if (auto *VT = dyn_cast<VectorType>(U.getType())) {
2422 VectorWidth = cast<FixedVectorType>(VT)->getNumElements();
2423 // We don't produce 1 x N vectors; those are treated as scalars.
2424 WantSplatVector = VectorWidth > 1;
2425 }
2426
2427 if (cast<GEPOperator>(U).hasAllZeroIndices())
2428 return translateCopy(U, BaseReg, MIRBuilder);
2429
2430 // We might need to splat the base pointer into a vector if the offsets
2431 // are vectors.
2432 if (WantSplatVector && !PtrTy.isVector()) {
2433 BaseReg = MIRBuilder
2434 .buildSplatBuildVector(LLT::fixed_vector(VectorWidth, PtrTy),
2435 BaseReg)
2436 .getReg(0);
2437 PtrIRTy = FixedVectorType::get(PtrIRTy, VectorWidth);
2438 PtrTy = getLLTForType(*PtrIRTy, *DL);
2439 OffsetIRTy = DL->getIndexType(PtrIRTy);
2440 OffsetTy = getLLTForType(*OffsetIRTy, *DL);
2441 }
2442
2443 int64_t Offset = 0;
2444 for (gep_type_iterator GTI = gep_type_begin(&U), E = gep_type_end(&U);
2445 GTI != E; ++GTI) {
2446 const Value *Idx = GTI.getOperand();
2447 if (StructType *StTy = GTI.getStructTypeOrNull()) {
2448 unsigned Field = cast<Constant>(Idx)->getUniqueInteger().getZExtValue();
2449 Offset += DL->getStructLayout(StTy)->getElementOffset(Field);
2450 continue;
2451 } else {
2452 uint64_t ElementSize = GTI.getSequentialElementStride(*DL);
2453
2454 // If this is a scalar constant or a splat vector of constants,
2455 // handle it quickly.
2456 if (const auto *CI = dyn_cast<ConstantInt>(Idx)) {
2457 if (std::optional<int64_t> Val = CI->getValue().trySExtValue()) {
2458 Offset += ElementSize * *Val;
2459 continue;
2460 }
2461 }
2462
2463 if (Offset != 0) {
2464 auto OffsetMIB = MIRBuilder.buildConstant({OffsetTy}, Offset);
2465 BaseReg = MIRBuilder
2466 .buildPtrAdd(PtrTy, BaseReg, OffsetMIB.getReg(0),
2467 PtrAddFlagsWithConst(Offset))
2468 .getReg(0);
2469 Offset = 0;
2470 }
2471
2472 Register IdxReg = getOrCreateVReg(*Idx);
2473 LLT IdxTy = MRI->getType(IdxReg);
2474 if (IdxTy != OffsetTy) {
2475 if (!IdxTy.isVector() && WantSplatVector) {
2476 IdxReg = MIRBuilder
2478 IdxReg)
2479 .getReg(0);
2480 }
2481
2482 IdxReg = MIRBuilder.buildSExtOrTrunc(OffsetTy, IdxReg).getReg(0);
2483 }
2484
2485 // N = N + Idx * ElementSize;
2486 // Avoid doing it for ElementSize of 1.
2487 Register GepOffsetReg;
2488 if (ElementSize != 1) {
2489 auto ElementSizeMIB = MIRBuilder.buildConstant(
2490 getLLTForType(*OffsetIRTy, *DL), ElementSize);
2491
2492 // The multiplication is NUW if the GEP is NUW and NSW if the GEP is
2493 // NUSW.
2494 uint32_t ScaleFlags = PtrAddFlags & MachineInstr::MIFlag::NoUWrap;
2495 if (PtrAddFlags & MachineInstr::MIFlag::NoUSWrap)
2496 ScaleFlags |= MachineInstr::MIFlag::NoSWrap;
2497
2498 GepOffsetReg =
2499 MIRBuilder.buildMul(OffsetTy, IdxReg, ElementSizeMIB, ScaleFlags)
2500 .getReg(0);
2501 } else {
2502 GepOffsetReg = IdxReg;
2503 }
2504
2505 BaseReg =
2506 MIRBuilder.buildPtrAdd(PtrTy, BaseReg, GepOffsetReg, PtrAddFlags)
2507 .getReg(0);
2508 }
2509 }
2510
2511 if (Offset != 0) {
2512 auto OffsetMIB =
2513 MIRBuilder.buildConstant(OffsetTy, Offset);
2514
2515 MIRBuilder.buildPtrAdd(getOrCreateVReg(U), BaseReg, OffsetMIB.getReg(0),
2516 PtrAddFlagsWithConst(Offset));
2517 return true;
2518 }
2519
2520 return translateCopy(U, BaseReg, MIRBuilder);
2521}
2522
2523bool IRTranslatorImpl::translateMemFunc(const CallInst &CI,
2524 MachineIRBuilder &MIRBuilder,
2525 unsigned Opcode) {
2526 const Value *SrcPtr = CI.getArgOperand(1);
2527 // If the source is undef, then just emit a nop.
2528 if (isa<UndefValue>(SrcPtr))
2529 return true;
2530
2532
2533 unsigned MinPtrSize = UINT_MAX;
2534 for (auto AI = CI.arg_begin(), AE = CI.arg_end(); std::next(AI) != AE; ++AI) {
2535 Register SrcReg = getOrCreateVReg(**AI);
2536 LLT SrcTy = MRI->getType(SrcReg);
2537 if (SrcTy.isPointer())
2538 MinPtrSize = std::min<unsigned>(SrcTy.getSizeInBits(), MinPtrSize);
2539 SrcRegs.push_back(SrcReg);
2540 }
2541
2542 LLT SizeTy = LLT::integer(MinPtrSize);
2543
2544 // The size operand should be the minimum of the pointer sizes.
2545 Register &SizeOpReg = SrcRegs[SrcRegs.size() - 1];
2546 if (MRI->getType(SizeOpReg) != SizeTy)
2547 SizeOpReg = MIRBuilder.buildZExtOrTrunc(SizeTy, SizeOpReg).getReg(0);
2548
2549 auto ICall = MIRBuilder.buildInstr(Opcode);
2550 for (Register SrcReg : SrcRegs)
2551 ICall.addUse(SrcReg);
2552
2553 Align DstAlign;
2554 Align SrcAlign;
2555 unsigned IsVol =
2556 cast<ConstantInt>(CI.getArgOperand(CI.arg_size() - 1))->getZExtValue();
2557
2558 ConstantInt *CopySize = nullptr;
2559
2560 if (auto *MCI = dyn_cast<MemCpyInst>(&CI)) {
2561 DstAlign = MCI->getDestAlign().valueOrOne();
2562 SrcAlign = MCI->getSourceAlign().valueOrOne();
2563 CopySize = dyn_cast<ConstantInt>(MCI->getArgOperand(2));
2564 } else if (auto *MMI = dyn_cast<MemMoveInst>(&CI)) {
2565 DstAlign = MMI->getDestAlign().valueOrOne();
2566 SrcAlign = MMI->getSourceAlign().valueOrOne();
2567 CopySize = dyn_cast<ConstantInt>(MMI->getArgOperand(2));
2568 } else {
2569 auto *MSI = cast<MemSetInst>(&CI);
2570 DstAlign = MSI->getDestAlign().valueOrOne();
2571 }
2572
2573 if (Opcode != TargetOpcode::G_MEMCPY_INLINE &&
2574 Opcode != TargetOpcode::G_MEMSET_INLINE) {
2575 // We need to propagate the tail call flag from the IR inst as an argument.
2576 // Otherwise, we have to pessimize and assume later that we cannot tail call
2577 // any memory intrinsics.
2578 ICall.addImm(CI.isTailCall() ? 1 : 0);
2579 }
2580
2581 // Create mem operands to store the alignment and volatile info.
2584 if (IsVol) {
2585 LoadFlags |= MachineMemOperand::MOVolatile;
2586 StoreFlags |= MachineMemOperand::MOVolatile;
2587 }
2588
2589 AAMDNodes AAInfo = CI.getAAMetadata();
2590 if (AA && CopySize &&
2591 AA->pointsToConstantMemory(MemoryLocation(
2592 SrcPtr, LocationSize::precise(CopySize->getZExtValue()), AAInfo))) {
2593 LoadFlags |= MachineMemOperand::MOInvariant;
2594
2595 // FIXME: pointsToConstantMemory probably does not imply dereferenceable,
2596 // but the previous usage implied it did. Probably should check
2597 // isDereferenceableAndAlignedPointer.
2599 }
2600
2601 ICall.addMemOperand(
2602 MF->getMachineMemOperand(MachinePointerInfo(CI.getArgOperand(0)),
2603 StoreFlags, 1, DstAlign, AAInfo));
2604 if (Opcode != TargetOpcode::G_MEMSET &&
2605 Opcode != TargetOpcode::G_MEMSET_INLINE)
2606 ICall.addMemOperand(MF->getMachineMemOperand(
2607 MachinePointerInfo(SrcPtr), LoadFlags, 1, SrcAlign, AAInfo));
2608
2609 return true;
2610}
2611
2612bool IRTranslatorImpl::translateTrap(const CallInst &CI,
2613 MachineIRBuilder &MIRBuilder,
2614 unsigned Opcode) {
2615 StringRef TrapFuncName =
2616 CI.getAttributes().getFnAttr("trap-func-name").getValueAsString();
2617 if (TrapFuncName.empty()) {
2618 MachineInstrBuilder MIB;
2619 if (Opcode == TargetOpcode::G_UBSANTRAP) {
2620 uint64_t Code = cast<ConstantInt>(CI.getOperand(0))->getZExtValue();
2621 MIB = MIRBuilder.buildInstr(Opcode, {}, ArrayRef<llvm::SrcOp>{Code});
2622 } else {
2623 MIB = MIRBuilder.buildInstr(Opcode);
2624 }
2625 if (CI.cannotMerge())
2627 return true;
2628 }
2629
2630 CallLowering::CallLoweringInfo Info;
2631 if (Opcode == TargetOpcode::G_UBSANTRAP)
2632 Info.OrigArgs.push_back({getOrCreateVRegs(*CI.getArgOperand(0)),
2633 CI.getArgOperand(0)->getType(), 0});
2634
2635 Info.Callee = MachineOperand::CreateES(TrapFuncName.data());
2636 Info.CB = &CI;
2637 Info.OrigRet = {Register(), Type::getVoidTy(CI.getContext()), 0};
2638 Info.NoMerge = CI.cannotMerge();
2639 return CLI->lowerCall(MIRBuilder, Info);
2640}
2641
2642bool IRTranslatorImpl::translateVectorInterleave2Intrinsic(
2643 const CallInst &CI, MachineIRBuilder &MIRBuilder) {
2644 assert(CI.getIntrinsicID() == Intrinsic::vector_interleave2 &&
2645 "This function can only be called on the interleave2 intrinsic!");
2646 // Canonicalize interleave2 to G_SHUFFLE_VECTOR (similar to SelectionDAG).
2647 Register Op0 = getOrCreateVReg(*CI.getOperand(0));
2648 Register Op1 = getOrCreateVReg(*CI.getOperand(1));
2649 Register Res = getOrCreateVReg(CI);
2650
2651 LLT OpTy = MRI->getType(Op0);
2652 MIRBuilder.buildShuffleVector(Res, Op0, Op1,
2654
2655 return true;
2656}
2657
2658bool IRTranslatorImpl::translateVectorDeinterleave2Intrinsic(
2659 const CallInst &CI, MachineIRBuilder &MIRBuilder) {
2660 assert(CI.getIntrinsicID() == Intrinsic::vector_deinterleave2 &&
2661 "This function can only be called on the deinterleave2 intrinsic!");
2662 // Canonicalize deinterleave2 to shuffles that extract sub-vectors (similar to
2663 // SelectionDAG).
2664 Register Op = getOrCreateVReg(*CI.getOperand(0));
2665 auto Undef = MIRBuilder.buildUndef(MRI->getType(Op));
2666 ArrayRef<Register> Res = getOrCreateVRegs(CI);
2667
2668 LLT ResTy = MRI->getType(Res[0]);
2669 if (ResTy.isScalar()) {
2670 MIRBuilder.buildExtractVectorElementConstant(Res[0], Op, 0);
2671 MIRBuilder.buildExtractVectorElementConstant(Res[1], Op, 1);
2672
2673 return true;
2674 }
2675
2676 assert(ResTy.isVector() && "Expected vector result type");
2677 MIRBuilder.buildShuffleVector(Res[0], Op, Undef,
2678 createStrideMask(0, 2, ResTy.getNumElements()));
2679 MIRBuilder.buildShuffleVector(Res[1], Op, Undef,
2680 createStrideMask(1, 2, ResTy.getNumElements()));
2681
2682 return true;
2683}
2684
2685void IRTranslatorImpl::getStackGuard(Register DstReg,
2686 MachineIRBuilder &MIRBuilder) {
2687 Value *Global =
2688 TLI->getSDagStackGuard(*MF->getFunction().getParent(), *Libcalls);
2689 if (!Global) {
2690 LLVMContext &Ctx = MIRBuilder.getContext();
2691 Ctx.diagnose(DiagnosticInfoGeneric("unable to lower stackguard"));
2692 MIRBuilder.buildUndef(DstReg);
2693 return;
2694 }
2695
2696 const TargetInstrInfo &TII = *MF->getSubtarget().getInstrInfo();
2697 MRI->setRegClass(DstReg,
2698 TII.getRegClass(TII.get(TargetOpcode::LOAD_STACK_GUARD), 0));
2699 auto MIB =
2700 MIRBuilder.buildInstr(TargetOpcode::LOAD_STACK_GUARD, {DstReg}, {});
2701
2702 unsigned AddrSpace = Global->getType()->getPointerAddressSpace();
2703 LLT PtrTy = LLT::pointer(AddrSpace, DL->getPointerSizeInBits(AddrSpace));
2704
2705 MachinePointerInfo MPInfo(Global);
2708 MachineMemOperand *MemRef = MF->getMachineMemOperand(
2709 MPInfo, Flags, PtrTy, DL->getPointerABIAlignment(AddrSpace));
2710 MIB.setMemRefs({MemRef});
2711}
2712
2713bool IRTranslatorImpl::translateOverflowIntrinsic(
2714 const CallInst &CI, unsigned Op, MachineIRBuilder &MIRBuilder) {
2715 ArrayRef<Register> ResRegs = getOrCreateVRegs(CI);
2716 MIRBuilder.buildInstr(
2717 Op, {ResRegs[0], ResRegs[1]},
2718 {getOrCreateVReg(*CI.getOperand(0)), getOrCreateVReg(*CI.getOperand(1))});
2719
2720 return true;
2721}
2722
2723bool IRTranslatorImpl::translateFixedPointIntrinsic(
2724 unsigned Op, const CallInst &CI, MachineIRBuilder &MIRBuilder) {
2725 Register Dst = getOrCreateVReg(CI);
2726 Register Src0 = getOrCreateVReg(*CI.getOperand(0));
2727 Register Src1 = getOrCreateVReg(*CI.getOperand(1));
2728 uint64_t Scale = cast<ConstantInt>(CI.getOperand(2))->getZExtValue();
2729 MIRBuilder.buildInstr(Op, {Dst}, { Src0, Src1, Scale });
2730 return true;
2731}
2732
2733unsigned IRTranslatorImpl::getSimpleIntrinsicOpcode(Intrinsic::ID ID) {
2734 switch (ID) {
2735 default:
2736 break;
2737 case Intrinsic::acos:
2738 return TargetOpcode::G_FACOS;
2739 case Intrinsic::asin:
2740 return TargetOpcode::G_FASIN;
2741 case Intrinsic::atan:
2742 return TargetOpcode::G_FATAN;
2743 case Intrinsic::atan2:
2744 return TargetOpcode::G_FATAN2;
2745 case Intrinsic::bswap:
2746 return TargetOpcode::G_BSWAP;
2747 case Intrinsic::bitreverse:
2748 return TargetOpcode::G_BITREVERSE;
2749 case Intrinsic::clmul:
2750 return TargetOpcode::G_CLMUL;
2751 case Intrinsic::fshl:
2752 return TargetOpcode::G_FSHL;
2753 case Intrinsic::fshr:
2754 return TargetOpcode::G_FSHR;
2755 case Intrinsic::ceil:
2756 return TargetOpcode::G_FCEIL;
2757 case Intrinsic::cos:
2758 return TargetOpcode::G_FCOS;
2759 case Intrinsic::cosh:
2760 return TargetOpcode::G_FCOSH;
2761 case Intrinsic::ctpop:
2762 return TargetOpcode::G_CTPOP;
2763 case Intrinsic::smulh:
2764 return TargetOpcode::G_SMULH;
2765 case Intrinsic::umulh:
2766 return TargetOpcode::G_UMULH;
2767 case Intrinsic::exp:
2768 return TargetOpcode::G_FEXP;
2769 case Intrinsic::exp2:
2770 return TargetOpcode::G_FEXP2;
2771 case Intrinsic::exp10:
2772 return TargetOpcode::G_FEXP10;
2773 case Intrinsic::fabs:
2774 return TargetOpcode::G_FABS;
2775 case Intrinsic::copysign:
2776 return TargetOpcode::G_FCOPYSIGN;
2777 case Intrinsic::minnum:
2778 return TargetOpcode::G_FMINNUM;
2779 case Intrinsic::maxnum:
2780 return TargetOpcode::G_FMAXNUM;
2781 case Intrinsic::minimum:
2782 return TargetOpcode::G_FMINIMUM;
2783 case Intrinsic::maximum:
2784 return TargetOpcode::G_FMAXIMUM;
2785 case Intrinsic::minimumnum:
2786 return TargetOpcode::G_FMINIMUMNUM;
2787 case Intrinsic::maximumnum:
2788 return TargetOpcode::G_FMAXIMUMNUM;
2789 case Intrinsic::canonicalize:
2790 return TargetOpcode::G_FCANONICALIZE;
2791 case Intrinsic::floor:
2792 return TargetOpcode::G_FFLOOR;
2793 case Intrinsic::fma:
2794 return TargetOpcode::G_FMA;
2795 case Intrinsic::log:
2796 return TargetOpcode::G_FLOG;
2797 case Intrinsic::log2:
2798 return TargetOpcode::G_FLOG2;
2799 case Intrinsic::log10:
2800 return TargetOpcode::G_FLOG10;
2801 case Intrinsic::ldexp:
2802 return TargetOpcode::G_FLDEXP;
2803 case Intrinsic::nearbyint:
2804 return TargetOpcode::G_FNEARBYINT;
2805 case Intrinsic::pow:
2806 return TargetOpcode::G_FPOW;
2807 case Intrinsic::powi:
2808 return TargetOpcode::G_FPOWI;
2809 case Intrinsic::rint:
2810 return TargetOpcode::G_FRINT;
2811 case Intrinsic::round:
2812 return TargetOpcode::G_INTRINSIC_ROUND;
2813 case Intrinsic::roundeven:
2814 return TargetOpcode::G_INTRINSIC_ROUNDEVEN;
2815 case Intrinsic::sin:
2816 return TargetOpcode::G_FSIN;
2817 case Intrinsic::sinh:
2818 return TargetOpcode::G_FSINH;
2819 case Intrinsic::sqrt:
2820 return TargetOpcode::G_FSQRT;
2821 case Intrinsic::tan:
2822 return TargetOpcode::G_FTAN;
2823 case Intrinsic::tanh:
2824 return TargetOpcode::G_FTANH;
2825 case Intrinsic::trunc:
2826 return TargetOpcode::G_INTRINSIC_TRUNC;
2827 case Intrinsic::readcyclecounter:
2828 return TargetOpcode::G_READCYCLECOUNTER;
2829 case Intrinsic::readsteadycounter:
2830 return TargetOpcode::G_READSTEADYCOUNTER;
2831 case Intrinsic::ptrmask:
2832 return TargetOpcode::G_PTRMASK;
2833 case Intrinsic::lrint:
2834 return TargetOpcode::G_INTRINSIC_LRINT;
2835 case Intrinsic::llrint:
2836 return TargetOpcode::G_INTRINSIC_LLRINT;
2837 // FADD/FMUL require checking the FMF, so are handled elsewhere.
2838 case Intrinsic::vector_reduce_fmin:
2839 return TargetOpcode::G_VECREDUCE_FMIN;
2840 case Intrinsic::vector_reduce_fmax:
2841 return TargetOpcode::G_VECREDUCE_FMAX;
2842 case Intrinsic::vector_reduce_fminimum:
2843 return TargetOpcode::G_VECREDUCE_FMINIMUM;
2844 case Intrinsic::vector_reduce_fmaximum:
2845 return TargetOpcode::G_VECREDUCE_FMAXIMUM;
2846 case Intrinsic::vector_reduce_fminimumnum:
2847 return TargetOpcode::G_VECREDUCE_FMINIMUMNUM;
2848 case Intrinsic::vector_reduce_fmaximumnum:
2849 return TargetOpcode::G_VECREDUCE_FMAXIMUMNUM;
2850 case Intrinsic::vector_reduce_add:
2851 return TargetOpcode::G_VECREDUCE_ADD;
2852 case Intrinsic::vector_reduce_mul:
2853 return TargetOpcode::G_VECREDUCE_MUL;
2854 case Intrinsic::vector_reduce_and:
2855 return TargetOpcode::G_VECREDUCE_AND;
2856 case Intrinsic::vector_reduce_or:
2857 return TargetOpcode::G_VECREDUCE_OR;
2858 case Intrinsic::vector_reduce_xor:
2859 return TargetOpcode::G_VECREDUCE_XOR;
2860 case Intrinsic::vector_reduce_smax:
2861 return TargetOpcode::G_VECREDUCE_SMAX;
2862 case Intrinsic::vector_reduce_smin:
2863 return TargetOpcode::G_VECREDUCE_SMIN;
2864 case Intrinsic::vector_reduce_umax:
2865 return TargetOpcode::G_VECREDUCE_UMAX;
2866 case Intrinsic::vector_reduce_umin:
2867 return TargetOpcode::G_VECREDUCE_UMIN;
2868 case Intrinsic::experimental_vector_compress:
2869 return TargetOpcode::G_VECTOR_COMPRESS;
2870 case Intrinsic::lround:
2871 return TargetOpcode::G_LROUND;
2872 case Intrinsic::llround:
2873 return TargetOpcode::G_LLROUND;
2874 case Intrinsic::get_fpenv:
2875 return TargetOpcode::G_GET_FPENV;
2876 case Intrinsic::get_fpmode:
2877 return TargetOpcode::G_GET_FPMODE;
2878 }
2880}
2881
2882bool IRTranslatorImpl::translateSimpleIntrinsic(const CallInst &CI,
2883 Intrinsic::ID ID,
2884 MachineIRBuilder &MIRBuilder) {
2885
2886 unsigned Op = getSimpleIntrinsicOpcode(ID);
2887
2888 // Is this a simple intrinsic?
2890 return false;
2891
2892 // Yes. Let's translate it.
2894 for (const auto &Arg : CI.args())
2895 VRegs.push_back(getOrCreateVReg(*Arg));
2896
2897 MIRBuilder.buildInstr(Op, {getOrCreateVReg(CI)}, VRegs,
2899 return true;
2900}
2901
2902// TODO: Include ConstainedOps.def when all strict instructions are defined.
2904 switch (ID) {
2905 case Intrinsic::experimental_constrained_fadd:
2906 return TargetOpcode::G_STRICT_FADD;
2907 case Intrinsic::experimental_constrained_fsub:
2908 return TargetOpcode::G_STRICT_FSUB;
2909 case Intrinsic::experimental_constrained_fmul:
2910 return TargetOpcode::G_STRICT_FMUL;
2911 case Intrinsic::experimental_constrained_fdiv:
2912 return TargetOpcode::G_STRICT_FDIV;
2913 case Intrinsic::experimental_constrained_frem:
2914 return TargetOpcode::G_STRICT_FREM;
2915 case Intrinsic::experimental_constrained_fma:
2916 return TargetOpcode::G_STRICT_FMA;
2917 case Intrinsic::experimental_constrained_sqrt:
2918 return TargetOpcode::G_STRICT_FSQRT;
2919 case Intrinsic::experimental_constrained_ldexp:
2920 return TargetOpcode::G_STRICT_FLDEXP;
2921 case Intrinsic::experimental_constrained_fcmp:
2922 return TargetOpcode::G_STRICT_FCMP;
2923 case Intrinsic::experimental_constrained_fcmps:
2924 return TargetOpcode::G_STRICT_FCMPS;
2925 default:
2926 return 0;
2927 }
2928}
2929
2930bool IRTranslatorImpl::translateConstrainedFPIntrinsic(
2931 const ConstrainedFPIntrinsic &FPI, MachineIRBuilder &MIRBuilder) {
2933
2934 unsigned Opcode = getConstrainedOpcode(FPI.getIntrinsicID());
2935 if (!Opcode)
2936 return false;
2937
2941
2942 if (Opcode == TargetOpcode::G_STRICT_FCMP ||
2943 Opcode == TargetOpcode::G_STRICT_FCMPS) {
2944 auto *FPCmp = cast<ConstrainedFPCmpIntrinsic>(&FPI);
2945 Register Operand0 = getOrCreateVReg(*FPCmp->getArgOperand(0));
2946 Register Operand1 = getOrCreateVReg(*FPCmp->getArgOperand(1));
2947 Register Result = getOrCreateVReg(FPI);
2948 MIRBuilder.buildInstr(Opcode, {Result}, {}, Flags)
2949 .addPredicate(FPCmp->getPredicate())
2950 .addUse(Operand0)
2951 .addUse(Operand1);
2952 return true;
2953 }
2954
2956 for (unsigned I = 0, E = FPI.getNonMetadataArgCount(); I != E; ++I)
2957 VRegs.push_back(getOrCreateVReg(*FPI.getArgOperand(I)));
2958
2959 MIRBuilder.buildInstr(Opcode, {getOrCreateVReg(FPI)}, VRegs, Flags);
2960 return true;
2961}
2962
2963std::optional<MCRegister> IRTranslatorImpl::getArgPhysReg(Argument &Arg) {
2964 auto VRegs = getOrCreateVRegs(Arg);
2965 if (VRegs.size() != 1)
2966 return std::nullopt;
2967
2968 // Arguments are lowered as a copy of a livein physical register.
2969 auto *VRegDef = MF->getRegInfo().getVRegDef(VRegs[0]);
2970 if (!VRegDef || !VRegDef->isCopy())
2971 return std::nullopt;
2972 return VRegDef->getOperand(1).getReg().asMCReg();
2973}
2974
2975bool IRTranslatorImpl::translateIfEntryValueArgument(
2976 bool isDeclare, Value *Val, const DILocalVariable *Var,
2977 const DIExpression *Expr, const DebugLoc &DL,
2978 MachineIRBuilder &MIRBuilder) {
2979 auto *Arg = dyn_cast<Argument>(Val);
2980 if (!Arg)
2981 return false;
2982
2983 if (!Expr->isEntryValue())
2984 return false;
2985
2986 std::optional<MCRegister> PhysReg = getArgPhysReg(*Arg);
2987 if (!PhysReg) {
2988 LLVM_DEBUG(dbgs() << "Dropping dbg." << (isDeclare ? "declare" : "value")
2989 << ": expression is entry_value but "
2990 << "couldn't find a physical register\n");
2991 LLVM_DEBUG(dbgs() << *Var << "\n");
2992 return true;
2993 }
2994
2995 if (isDeclare) {
2996 // Append an op deref to account for the fact that this is a dbg_declare.
2997 Expr = DIExpression::append(Expr, dwarf::DW_OP_deref);
2998 MF->setVariableDbgInfo(Var, Expr, *PhysReg, DL);
2999 } else {
3000 MIRBuilder.buildDirectDbgValue(*PhysReg, Var, Expr);
3001 }
3002
3003 return true;
3004}
3005
3006static unsigned getConvOpcode(Intrinsic::ID ID) {
3007 switch (ID) {
3008 default:
3009 llvm_unreachable("Unexpected intrinsic");
3010 case Intrinsic::experimental_convergence_anchor:
3011 return TargetOpcode::CONVERGENCECTRL_ANCHOR;
3012 case Intrinsic::experimental_convergence_entry:
3013 return TargetOpcode::CONVERGENCECTRL_ENTRY;
3014 case Intrinsic::experimental_convergence_loop:
3015 return TargetOpcode::CONVERGENCECTRL_LOOP;
3016 }
3017}
3018
3019bool IRTranslatorImpl::translateConvergenceControlIntrinsic(
3020 const CallInst &CI, Intrinsic::ID ID, MachineIRBuilder &MIRBuilder) {
3021 MachineInstrBuilder MIB = MIRBuilder.buildInstr(getConvOpcode(ID));
3022 Register OutputReg = getOrCreateConvergenceTokenVReg(CI);
3023 MIB.addDef(OutputReg);
3024
3025 if (ID == Intrinsic::experimental_convergence_loop) {
3027 assert(Bundle && "Expected a convergence control token.");
3028 Register InputReg =
3029 getOrCreateConvergenceTokenVReg(*Bundle->Inputs[0].get());
3030 MIB.addUse(InputReg);
3031 }
3032
3033 return true;
3034}
3035
3036bool IRTranslatorImpl::translateKnownIntrinsic(const CallInst &CI,
3037 Intrinsic::ID ID,
3038 MachineIRBuilder &MIRBuilder) {
3039 if (auto *MI = dyn_cast<AnyMemIntrinsic>(&CI)) {
3040 if (ORE->enabled()) {
3041 if (MemoryOpRemark::canHandle(MI, *LibInfo)) {
3042 MemoryOpRemark R(*ORE, "gisel-ir-translator-memsize", *DL, *LibInfo);
3043 R.visit(MI);
3044 }
3045 }
3046 }
3047
3048 // If this is a simple intrinsic (that is, we just need to add a def of
3049 // a vreg, and uses for each arg operand, then translate it.
3050 if (translateSimpleIntrinsic(CI, ID, MIRBuilder))
3051 return true;
3052
3053 switch (ID) {
3054 default:
3055 break;
3056 case Intrinsic::lifetime_start:
3057 case Intrinsic::lifetime_end: {
3058 // No stack colouring in O0, discard region information.
3059 if (MF->getTarget().getOptLevel() == CodeGenOptLevel::None ||
3060 MF->getFunction().hasOptNone())
3061 return true;
3062
3063 unsigned Op = ID == Intrinsic::lifetime_start ? TargetOpcode::LIFETIME_START
3064 : TargetOpcode::LIFETIME_END;
3065
3066 const AllocaInst *AI = dyn_cast<AllocaInst>(CI.getArgOperand(0));
3067 if (!AI || !AI->isStaticAlloca())
3068 return true;
3069
3070 MIRBuilder.buildInstr(Op).addFrameIndex(getOrCreateFrameIndex(*AI));
3071 return true;
3072 }
3073 case Intrinsic::fake_use: {
3075 for (const auto &Arg : CI.args())
3076 llvm::append_range(VRegs, getOrCreateVRegs(*Arg));
3077 MIRBuilder.buildInstr(TargetOpcode::FAKE_USE, {}, VRegs);
3078 MF->setHasFakeUses(true);
3079 return true;
3080 }
3081 case Intrinsic::vaend:
3082 // No target I know of cares about va_end. Certainly no in-tree target
3083 // does. Simplest intrinsic ever!
3084 return true;
3085 case Intrinsic::vastart: {
3086 Value *Ptr = CI.getArgOperand(0);
3087 unsigned ListSize = TLI->getVaListSizeInBits(*DL) / 8;
3088 Align Alignment = getKnownAlignment(Ptr, *DL);
3089
3090 MIRBuilder.buildInstr(TargetOpcode::G_VASTART, {}, {getOrCreateVReg(*Ptr)})
3091 .addMemOperand(MF->getMachineMemOperand(MachinePointerInfo(Ptr),
3093 ListSize, Alignment));
3094 return true;
3095 }
3096 case Intrinsic::uadd_with_overflow:
3097 return translateOverflowIntrinsic(CI, TargetOpcode::G_UADDO, MIRBuilder);
3098 case Intrinsic::sadd_with_overflow:
3099 return translateOverflowIntrinsic(CI, TargetOpcode::G_SADDO, MIRBuilder);
3100 case Intrinsic::usub_with_overflow:
3101 return translateOverflowIntrinsic(CI, TargetOpcode::G_USUBO, MIRBuilder);
3102 case Intrinsic::ssub_with_overflow:
3103 return translateOverflowIntrinsic(CI, TargetOpcode::G_SSUBO, MIRBuilder);
3104 case Intrinsic::umul_with_overflow:
3105 return translateOverflowIntrinsic(CI, TargetOpcode::G_UMULO, MIRBuilder);
3106 case Intrinsic::smul_with_overflow:
3107 return translateOverflowIntrinsic(CI, TargetOpcode::G_SMULO, MIRBuilder);
3108 case Intrinsic::uadd_sat:
3109 return translateBinaryOp(TargetOpcode::G_UADDSAT, CI, MIRBuilder);
3110 case Intrinsic::sadd_sat:
3111 return translateBinaryOp(TargetOpcode::G_SADDSAT, CI, MIRBuilder);
3112 case Intrinsic::usub_sat:
3113 return translateBinaryOp(TargetOpcode::G_USUBSAT, CI, MIRBuilder);
3114 case Intrinsic::ssub_sat:
3115 return translateBinaryOp(TargetOpcode::G_SSUBSAT, CI, MIRBuilder);
3116 case Intrinsic::ushl_sat:
3117 return translateBinaryOp(TargetOpcode::G_USHLSAT, CI, MIRBuilder);
3118 case Intrinsic::sshl_sat:
3119 return translateBinaryOp(TargetOpcode::G_SSHLSAT, CI, MIRBuilder);
3120 case Intrinsic::umin:
3121 return translateBinaryOp(TargetOpcode::G_UMIN, CI, MIRBuilder);
3122 case Intrinsic::umax:
3123 return translateBinaryOp(TargetOpcode::G_UMAX, CI, MIRBuilder);
3124 case Intrinsic::smin:
3125 return translateBinaryOp(TargetOpcode::G_SMIN, CI, MIRBuilder);
3126 case Intrinsic::smax:
3127 return translateBinaryOp(TargetOpcode::G_SMAX, CI, MIRBuilder);
3128 case Intrinsic::abs:
3129 // TODO: Preserve "int min is poison" arg in GMIR?
3130 return translateUnaryOp(TargetOpcode::G_ABS, CI, MIRBuilder);
3131 case Intrinsic::smul_fix:
3132 return translateFixedPointIntrinsic(TargetOpcode::G_SMULFIX, CI, MIRBuilder);
3133 case Intrinsic::umul_fix:
3134 return translateFixedPointIntrinsic(TargetOpcode::G_UMULFIX, CI, MIRBuilder);
3135 case Intrinsic::smul_fix_sat:
3136 return translateFixedPointIntrinsic(TargetOpcode::G_SMULFIXSAT, CI, MIRBuilder);
3137 case Intrinsic::umul_fix_sat:
3138 return translateFixedPointIntrinsic(TargetOpcode::G_UMULFIXSAT, CI, MIRBuilder);
3139 case Intrinsic::sdiv_fix:
3140 return translateFixedPointIntrinsic(TargetOpcode::G_SDIVFIX, CI, MIRBuilder);
3141 case Intrinsic::udiv_fix:
3142 return translateFixedPointIntrinsic(TargetOpcode::G_UDIVFIX, CI, MIRBuilder);
3143 case Intrinsic::sdiv_fix_sat:
3144 return translateFixedPointIntrinsic(TargetOpcode::G_SDIVFIXSAT, CI, MIRBuilder);
3145 case Intrinsic::udiv_fix_sat:
3146 return translateFixedPointIntrinsic(TargetOpcode::G_UDIVFIXSAT, CI, MIRBuilder);
3147 case Intrinsic::fmuladd: {
3148 Register Dst = getOrCreateVReg(CI);
3149 Register Op0 = getOrCreateVReg(*CI.getArgOperand(0));
3150 Register Op1 = getOrCreateVReg(*CI.getArgOperand(1));
3151 Register Op2 = getOrCreateVReg(*CI.getArgOperand(2));
3152 if (TLI->isFMAFasterThanFMulAndFAdd(*MF,
3153 TLI->getValueType(*DL, CI.getType()))) {
3154 // TODO: Revisit this to see if we should move this part of the
3155 // lowering to the combiner.
3156 MIRBuilder.buildFMA(Dst, Op0, Op1, Op2,
3158 } else {
3159 LLT Ty = getLLTForType(*CI.getType(), *DL);
3160 auto FMul = MIRBuilder.buildFMul(
3161 Ty, Op0, Op1, MachineInstr::copyFlagsFromInstruction(CI));
3162 MIRBuilder.buildFAdd(Dst, FMul, Op2,
3164 }
3165 return true;
3166 }
3167 case Intrinsic::frexp: {
3168 ArrayRef<Register> VRegs = getOrCreateVRegs(CI);
3169 MIRBuilder.buildFFrexp(VRegs[0], VRegs[1],
3170 getOrCreateVReg(*CI.getArgOperand(0)),
3172 return true;
3173 }
3174 case Intrinsic::modf: {
3175 ArrayRef<Register> VRegs = getOrCreateVRegs(CI);
3176 MIRBuilder.buildModf(VRegs[0], VRegs[1],
3177 getOrCreateVReg(*CI.getArgOperand(0)),
3179 return true;
3180 }
3181 case Intrinsic::sincos: {
3182 ArrayRef<Register> VRegs = getOrCreateVRegs(CI);
3183 MIRBuilder.buildFSincos(VRegs[0], VRegs[1],
3184 getOrCreateVReg(*CI.getArgOperand(0)),
3186 return true;
3187 }
3188 case Intrinsic::fptosi_sat:
3189 MIRBuilder.buildFPTOSI_SAT(getOrCreateVReg(CI),
3190 getOrCreateVReg(*CI.getArgOperand(0)));
3191 return true;
3192 case Intrinsic::fptoui_sat:
3193 MIRBuilder.buildFPTOUI_SAT(getOrCreateVReg(CI),
3194 getOrCreateVReg(*CI.getArgOperand(0)));
3195 return true;
3196 case Intrinsic::memcpy_inline:
3197 return translateMemFunc(CI, MIRBuilder, TargetOpcode::G_MEMCPY_INLINE);
3198 case Intrinsic::memcpy:
3199 return translateMemFunc(CI, MIRBuilder, TargetOpcode::G_MEMCPY);
3200 case Intrinsic::memmove:
3201 return translateMemFunc(CI, MIRBuilder, TargetOpcode::G_MEMMOVE);
3202 case Intrinsic::memset:
3203 return translateMemFunc(CI, MIRBuilder, TargetOpcode::G_MEMSET);
3204 case Intrinsic::memset_inline:
3205 return translateMemFunc(CI, MIRBuilder, TargetOpcode::G_MEMSET_INLINE);
3206 case Intrinsic::eh_typeid_for: {
3207 GlobalValue *GV = ExtractTypeInfo(CI.getArgOperand(0));
3208 Register Reg = getOrCreateVReg(CI);
3209 unsigned TypeID = MF->getTypeIDFor(GV);
3210 MIRBuilder.buildConstant(Reg, TypeID);
3211 return true;
3212 }
3213 case Intrinsic::objectsize:
3214 llvm_unreachable("llvm.objectsize.* should have been lowered already");
3215
3216 case Intrinsic::is_constant:
3217 llvm_unreachable("llvm.is.constant.* should have been lowered already");
3218
3219 case Intrinsic::stackguard:
3220 getStackGuard(getOrCreateVReg(CI), MIRBuilder);
3221 return true;
3222 case Intrinsic::stackprotector: {
3223 LLT PtrTy = getLLTForType(*CI.getArgOperand(0)->getType(), *DL);
3224 Register GuardVal;
3225 if (TLI->useLoadStackGuardNode(*CI.getModule())) {
3226 GuardVal = MRI->createGenericVirtualRegister(PtrTy);
3227 getStackGuard(GuardVal, MIRBuilder);
3228 } else
3229 GuardVal = getOrCreateVReg(*CI.getArgOperand(0)); // The guard's value.
3230
3231 AllocaInst *Slot = cast<AllocaInst>(CI.getArgOperand(1));
3232 int FI = getOrCreateFrameIndex(*Slot);
3233 MF->getFrameInfo().setStackProtectorIndex(FI);
3234
3235 MIRBuilder.buildStore(
3236 GuardVal, getOrCreateVReg(*Slot),
3237 *MF->getMachineMemOperand(MachinePointerInfo::getFixedStack(*MF, FI),
3240 PtrTy, Align(8)));
3241 return true;
3242 }
3243 case Intrinsic::stacksave: {
3244 MIRBuilder.buildInstr(TargetOpcode::G_STACKSAVE, {getOrCreateVReg(CI)}, {});
3245 return true;
3246 }
3247 case Intrinsic::stackrestore: {
3248 MIRBuilder.buildInstr(TargetOpcode::G_STACKRESTORE, {},
3249 {getOrCreateVReg(*CI.getArgOperand(0))});
3250 return true;
3251 }
3252 case Intrinsic::cttz:
3253 case Intrinsic::ctlz: {
3254 ConstantInt *Cst = cast<ConstantInt>(CI.getArgOperand(1));
3255 bool isTrailing = ID == Intrinsic::cttz;
3256 unsigned Opcode = isTrailing ? Cst->isZero()
3257 ? TargetOpcode::G_CTTZ
3258 : TargetOpcode::G_CTTZ_ZERO_POISON
3259 : Cst->isZero() ? TargetOpcode::G_CTLZ
3260 : TargetOpcode::G_CTLZ_ZERO_POISON;
3261 MIRBuilder.buildInstr(Opcode, {getOrCreateVReg(CI)},
3262 {getOrCreateVReg(*CI.getArgOperand(0))});
3263 return true;
3264 }
3265 case Intrinsic::invariant_start: {
3266 MIRBuilder.buildUndef(getOrCreateVReg(CI));
3267 return true;
3268 }
3269 case Intrinsic::invariant_end:
3270 return true;
3271 case Intrinsic::expect:
3272 case Intrinsic::expect_with_probability:
3273 case Intrinsic::annotation:
3274 case Intrinsic::ptr_annotation:
3275 case Intrinsic::launder_invariant_group:
3276 case Intrinsic::threadlocal_address: {
3277 // Drop the intrinsic, but forward the value.
3278 MIRBuilder.buildCopy(getOrCreateVReg(CI),
3279 getOrCreateVReg(*CI.getArgOperand(0)));
3280 return true;
3281 }
3282 case Intrinsic::assume:
3283 case Intrinsic::experimental_noalias_scope_decl:
3284 case Intrinsic::var_annotation:
3285 case Intrinsic::sideeffect:
3286 // Discard annotate attributes, assumptions, and artificial side-effects.
3287 return true;
3288 case Intrinsic::read_volatile_register:
3289 case Intrinsic::read_register: {
3290 Value *Arg = CI.getArgOperand(0);
3291 MIRBuilder
3292 .buildInstr(TargetOpcode::G_READ_REGISTER, {getOrCreateVReg(CI)}, {})
3293 .addMetadata(cast<MDNode>(cast<MetadataAsValue>(Arg)->getMetadata()));
3294 return true;
3295 }
3296 case Intrinsic::write_register: {
3297 Value *Arg = CI.getArgOperand(0);
3298 MIRBuilder.buildInstr(TargetOpcode::G_WRITE_REGISTER)
3299 .addMetadata(cast<MDNode>(cast<MetadataAsValue>(Arg)->getMetadata()))
3300 .addUse(getOrCreateVReg(*CI.getArgOperand(1)));
3301 return true;
3302 }
3303 case Intrinsic::localescape: {
3304 MachineBasicBlock &EntryMBB = MF->front();
3305 StringRef EscapedName = GlobalValue::dropLLVMManglingEscape(MF->getName());
3306
3307 // Directly emit some LOCAL_ESCAPE machine instrs. Label assignment emission
3308 // is the same on all targets.
3309 for (unsigned Idx = 0, E = CI.arg_size(); Idx < E; ++Idx) {
3310 Value *Arg = CI.getArgOperand(Idx)->stripPointerCasts();
3311 if (isa<ConstantPointerNull>(Arg))
3312 continue; // Skip null pointers. They represent a hole in index space.
3313
3314 int FI = getOrCreateFrameIndex(*cast<AllocaInst>(Arg));
3315 MCSymbol *FrameAllocSym =
3316 MF->getContext().getOrCreateFrameAllocSymbol(EscapedName, Idx);
3317
3318 // This should be inserted at the start of the entry block.
3319 auto LocalEscape =
3320 MIRBuilder.buildInstrNoInsert(TargetOpcode::LOCAL_ESCAPE)
3321 .addSym(FrameAllocSym)
3322 .addFrameIndex(FI);
3323
3324 EntryMBB.insert(EntryMBB.begin(), LocalEscape);
3325 }
3326
3327 return true;
3328 }
3329 case Intrinsic::vector_reduce_fadd:
3330 case Intrinsic::vector_reduce_fmul: {
3331 // Need to check for the reassoc flag to decide whether we want a
3332 // sequential reduction opcode or not.
3333 Register Dst = getOrCreateVReg(CI);
3334 Register ScalarSrc = getOrCreateVReg(*CI.getArgOperand(0));
3335 Register VecSrc = getOrCreateVReg(*CI.getArgOperand(1));
3336 unsigned Opc = 0;
3337 if (!CI.hasAllowReassoc()) {
3338 // The sequential ordering case.
3339 Opc = ID == Intrinsic::vector_reduce_fadd
3340 ? TargetOpcode::G_VECREDUCE_SEQ_FADD
3341 : TargetOpcode::G_VECREDUCE_SEQ_FMUL;
3342 if (!MRI->getType(VecSrc).isVector())
3343 Opc = ID == Intrinsic::vector_reduce_fadd ? TargetOpcode::G_FADD
3344 : TargetOpcode::G_FMUL;
3345 MIRBuilder.buildInstr(Opc, {Dst}, {ScalarSrc, VecSrc},
3347 return true;
3348 }
3349 // We split the operation into a separate G_FADD/G_FMUL + the reduce,
3350 // since the associativity doesn't matter.
3351 unsigned ScalarOpc;
3352 if (ID == Intrinsic::vector_reduce_fadd) {
3353 Opc = TargetOpcode::G_VECREDUCE_FADD;
3354 ScalarOpc = TargetOpcode::G_FADD;
3355 } else {
3356 Opc = TargetOpcode::G_VECREDUCE_FMUL;
3357 ScalarOpc = TargetOpcode::G_FMUL;
3358 }
3359 LLT DstTy = MRI->getType(Dst);
3360 auto Rdx = MIRBuilder.buildInstr(
3361 Opc, {DstTy}, {VecSrc}, MachineInstr::copyFlagsFromInstruction(CI));
3362 MIRBuilder.buildInstr(ScalarOpc, {Dst}, {ScalarSrc, Rdx},
3364
3365 return true;
3366 }
3367 case Intrinsic::trap:
3368 return translateTrap(CI, MIRBuilder, TargetOpcode::G_TRAP);
3369 case Intrinsic::debugtrap:
3370 return translateTrap(CI, MIRBuilder, TargetOpcode::G_DEBUGTRAP);
3371 case Intrinsic::ubsantrap:
3372 return translateTrap(CI, MIRBuilder, TargetOpcode::G_UBSANTRAP);
3373 case Intrinsic::allow_runtime_check:
3374 case Intrinsic::allow_ubsan_check:
3375 MIRBuilder.buildCopy(getOrCreateVReg(CI),
3376 getOrCreateVReg(*ConstantInt::getTrue(CI.getType())));
3377 return true;
3378 case Intrinsic::amdgcn_cs_chain:
3379 case Intrinsic::amdgcn_call_whole_wave:
3380 return translateCallBase(CI, MIRBuilder);
3381 case Intrinsic::fptrunc_round: {
3383
3384 // Convert the metadata argument to a constant integer
3385 Metadata *MD = cast<MetadataAsValue>(CI.getArgOperand(1))->getMetadata();
3386 std::optional<RoundingMode> RoundMode =
3387 convertStrToRoundingMode(cast<MDString>(MD)->getString());
3388
3389 // Add the Rounding mode as an integer
3390 MIRBuilder
3391 .buildInstr(TargetOpcode::G_INTRINSIC_FPTRUNC_ROUND,
3392 {getOrCreateVReg(CI)},
3393 {getOrCreateVReg(*CI.getArgOperand(0))}, Flags)
3394 .addImm((int)*RoundMode);
3395
3396 return true;
3397 }
3398 case Intrinsic::is_fpclass: {
3399 Value *FpValue = CI.getOperand(0);
3400 ConstantInt *TestMaskValue = cast<ConstantInt>(CI.getOperand(1));
3401
3402 MIRBuilder
3403 .buildInstr(TargetOpcode::G_IS_FPCLASS, {getOrCreateVReg(CI)},
3404 {getOrCreateVReg(*FpValue)})
3405 .addImm(TestMaskValue->getZExtValue());
3406
3407 return true;
3408 }
3409 case Intrinsic::set_fpenv: {
3410 Value *FPEnv = CI.getOperand(0);
3411 MIRBuilder.buildSetFPEnv(getOrCreateVReg(*FPEnv));
3412 return true;
3413 }
3414 case Intrinsic::reset_fpenv:
3415 MIRBuilder.buildResetFPEnv();
3416 return true;
3417 case Intrinsic::set_fpmode: {
3418 Value *FPState = CI.getOperand(0);
3419 MIRBuilder.buildSetFPMode(getOrCreateVReg(*FPState));
3420 return true;
3421 }
3422 case Intrinsic::reset_fpmode:
3423 MIRBuilder.buildResetFPMode();
3424 return true;
3425 case Intrinsic::get_rounding:
3426 MIRBuilder.buildGetRounding(getOrCreateVReg(CI));
3427 return true;
3428 case Intrinsic::set_rounding:
3429 MIRBuilder.buildSetRounding(getOrCreateVReg(*CI.getOperand(0)));
3430 return true;
3431 case Intrinsic::vscale: {
3432 MIRBuilder.buildVScale(getOrCreateVReg(CI), 1);
3433 return true;
3434 }
3435 case Intrinsic::scmp:
3436 MIRBuilder.buildSCmp(getOrCreateVReg(CI),
3437 getOrCreateVReg(*CI.getOperand(0)),
3438 getOrCreateVReg(*CI.getOperand(1)));
3439 return true;
3440 case Intrinsic::ucmp:
3441 MIRBuilder.buildUCmp(getOrCreateVReg(CI),
3442 getOrCreateVReg(*CI.getOperand(0)),
3443 getOrCreateVReg(*CI.getOperand(1)));
3444 return true;
3445 case Intrinsic::vector_extract:
3446 return translateExtractVector(CI, MIRBuilder);
3447 case Intrinsic::vector_insert:
3448 return translateInsertVector(CI, MIRBuilder);
3449 case Intrinsic::stepvector: {
3450 MIRBuilder.buildStepVector(getOrCreateVReg(CI), 1);
3451 return true;
3452 }
3453 case Intrinsic::prefetch: {
3454 Value *Addr = CI.getOperand(0);
3455 unsigned RW = cast<ConstantInt>(CI.getOperand(1))->getZExtValue();
3456 unsigned Locality = cast<ConstantInt>(CI.getOperand(2))->getZExtValue();
3457 unsigned CacheType = cast<ConstantInt>(CI.getOperand(3))->getZExtValue();
3458
3460 auto &MMO = *MF->getMachineMemOperand(MachinePointerInfo(Addr), Flags,
3461 LLT(), Align());
3462
3463 MIRBuilder.buildPrefetch(getOrCreateVReg(*Addr), RW, Locality, CacheType,
3464 MMO);
3465
3466 return true;
3467 }
3468
3469 case Intrinsic::speculative_load: {
3470 // Only the pointer operand is needed at codegen; the remaining arguments
3471 // carry IR-level semantics only.
3472 const Value *Ptr = CI.getArgOperand(0);
3473 Register Dst = getOrCreateVReg(CI);
3475 Flags |= TLI->getTargetMMOFlags(CI);
3476 if (CI.hasMetadata(LLVMContext::MD_nontemporal))
3478 if (CI.hasMetadata(LLVMContext::MD_invariant_load))
3480 auto *MMO = MF->getMachineMemOperand(
3481 MachinePointerInfo(Ptr), Flags, MRI->getType(Dst),
3482 CI.getParamAlign(0).valueOrOne(), MMOMetadata(CI.getAAMetadata()));
3483 MIRBuilder.buildLoad(Dst, getOrCreateVReg(*Ptr), *MMO);
3484 return true;
3485 }
3486
3487 case Intrinsic::vector_interleave2:
3488 case Intrinsic::vector_deinterleave2: {
3489 // Both intrinsics have at least one operand.
3490 Value *Op0 = CI.getOperand(0);
3491 LLT ResTy = getLLTForType(*Op0->getType(), MIRBuilder.getDataLayout());
3492 if (!ResTy.isFixedVector())
3493 return false;
3494
3495 if (CI.getIntrinsicID() == Intrinsic::vector_interleave2)
3496 return translateVectorInterleave2Intrinsic(CI, MIRBuilder);
3497
3498 return translateVectorDeinterleave2Intrinsic(CI, MIRBuilder);
3499 }
3500
3501#define INSTRUCTION(NAME, NARG, ROUND_MODE, INTRINSIC) \
3502 case Intrinsic::INTRINSIC:
3503#include "llvm/IR/ConstrainedOps.def"
3504 return translateConstrainedFPIntrinsic(cast<ConstrainedFPIntrinsic>(CI),
3505 MIRBuilder);
3506 case Intrinsic::experimental_convergence_anchor:
3507 case Intrinsic::experimental_convergence_entry:
3508 case Intrinsic::experimental_convergence_loop:
3509 return translateConvergenceControlIntrinsic(CI, ID, MIRBuilder);
3510 case Intrinsic::reloc_none: {
3511 Metadata *MD = cast<MetadataAsValue>(CI.getArgOperand(0))->getMetadata();
3512 StringRef SymbolName = cast<MDString>(MD)->getString();
3513 MIRBuilder.buildInstr(TargetOpcode::RELOC_NONE)
3515 return true;
3516 }
3517 }
3518 return false;
3519}
3520
3521bool IRTranslatorImpl::translateInlineAsm(const CallBase &CB,
3522 MachineIRBuilder &MIRBuilder) {
3523 if (!mayTranslateUserTypes(CB))
3524 return false;
3525
3526 const InlineAsmLowering *ALI = MF->getSubtarget().getInlineAsmLowering();
3527
3528 if (!ALI) {
3529 LLVM_DEBUG(
3530 dbgs() << "Inline asm lowering is not supported for this target yet\n");
3531 return false;
3532 }
3533
3534 return ALI->lowerInlineAsm(
3535 MIRBuilder, CB, [&](const Value &Val) { return getOrCreateVRegs(Val); });
3536}
3537
3538bool IRTranslatorImpl::translateCallBase(const CallBase &CB,
3539 MachineIRBuilder &MIRBuilder) {
3540 ArrayRef<Register> Res = getOrCreateVRegs(CB);
3541
3543 Register SwiftInVReg = 0;
3544 Register SwiftErrorVReg = 0;
3545 for (const auto &Arg : CB.args()) {
3546 if (CLI->supportSwiftError() && isSwiftError(Arg)) {
3547 assert(SwiftInVReg == 0 && "Expected only one swift error argument");
3548 LLT Ty = getLLTForType(*Arg->getType(), *DL);
3549 SwiftInVReg = MRI->createGenericVirtualRegister(Ty);
3550 MIRBuilder.buildCopy(SwiftInVReg, SwiftError.getOrCreateVRegUseAt(
3551 &CB, &MIRBuilder.getMBB(), Arg));
3552 Args.emplace_back(ArrayRef(SwiftInVReg));
3553 SwiftErrorVReg =
3554 SwiftError.getOrCreateVRegDefAt(&CB, &MIRBuilder.getMBB(), Arg);
3555 continue;
3556 }
3557 Args.push_back(getOrCreateVRegs(*Arg));
3558 }
3559
3560 if (auto *CI = dyn_cast<CallInst>(&CB)) {
3561 if (ORE->enabled()) {
3562 if (MemoryOpRemark::canHandle(CI, *LibInfo)) {
3563 MemoryOpRemark R(*ORE, "gisel-ir-translator-memsize", *DL, *LibInfo);
3564 R.visit(CI);
3565 }
3566 }
3567 }
3568
3569 std::optional<CallLowering::PtrAuthInfo> PAI;
3570 if (auto Bundle = CB.getOperandBundle(LLVMContext::OB_ptrauth)) {
3571 // Functions should never be ptrauth-called directly.
3572 assert(!CB.getCalledFunction() && "invalid direct ptrauth call");
3573
3574 const Value *Key = Bundle->Inputs[0];
3575 const Value *Discriminator = Bundle->Inputs[1];
3576
3577 // Look through ptrauth constants to try to eliminate the matching bundle
3578 // and turn this into a direct call with no ptrauth.
3579 // CallLowering will use the raw pointer if it doesn't find the PAI.
3580 const auto *CalleeCPA = dyn_cast<ConstantPtrAuth>(CB.getCalledOperand());
3581 if (!CalleeCPA || !isa<Function>(CalleeCPA->getPointer()) ||
3582 !CalleeCPA->isKnownCompatibleWith(Key, Discriminator, *DL)) {
3583 // If we can't make it direct, package the bundle into PAI.
3584 Register DiscReg = getOrCreateVReg(*Discriminator);
3585 PAI = CallLowering::PtrAuthInfo{cast<ConstantInt>(Key)->getZExtValue(),
3586 DiscReg};
3587 }
3588 }
3589
3590 Register ConvergenceCtrlToken = 0;
3591 if (auto Bundle = CB.getOperandBundle(LLVMContext::OB_convergencectrl)) {
3592 const auto &Token = *Bundle->Inputs[0].get();
3593 ConvergenceCtrlToken = getOrCreateConvergenceTokenVReg(Token);
3594 }
3595
3596 // We don't set HasCalls on MFI here yet because call lowering may decide to
3597 // optimize into tail calls. Instead, we defer that to selection where a final
3598 // scan is done to check if any instructions are calls.
3599 bool Success = CLI->lowerCall(
3600 MIRBuilder, CB, Res, Args, SwiftErrorVReg, PAI, ConvergenceCtrlToken,
3601 [&]() { return getOrCreateVReg(*CB.getCalledOperand()); });
3602
3603 // Check if we just inserted a tail call.
3604 if (Success) {
3605 assert(!HasTailCall && "Can't tail call return twice from block?");
3606 const TargetInstrInfo *TII = MF->getSubtarget().getInstrInfo();
3607 HasTailCall = TII->isTailCall(*std::prev(MIRBuilder.getInsertPt()));
3608 }
3609
3610 return Success;
3611}
3612
3613bool IRTranslatorImpl::translateCall(const User &U,
3614 MachineIRBuilder &MIRBuilder) {
3615 if (!mayTranslateUserTypes(U))
3616 return false;
3617
3618 const CallInst &CI = cast<CallInst>(U);
3619 const Function *F = CI.getCalledFunction();
3620
3621 // FIXME: support Windows dllimport function calls and calls through
3622 // weak symbols.
3623 if (F && (F->hasDLLImportStorageClass() ||
3624 (MF->getTarget().getTargetTriple().isOSWindows() &&
3625 F->hasExternalWeakLinkage())))
3626 return false;
3627
3628 // FIXME: support control flow guard targets.
3630 return false;
3631
3632 // FIXME: support statepoints and related.
3634 return false;
3635
3636 if (CI.isInlineAsm())
3637 return translateInlineAsm(CI, MIRBuilder);
3638
3639 Intrinsic::ID ID = F ? F->getIntrinsicID() : Intrinsic::not_intrinsic;
3640 if (!F || ID == Intrinsic::not_intrinsic) {
3641 if (translateCallBase(CI, MIRBuilder)) {
3642 diagnoseDontCall(CI);
3643 return true;
3644 }
3645 return false;
3646 }
3647
3648 assert(ID != Intrinsic::not_intrinsic && "unknown intrinsic");
3649
3650 if (!MF->getSubtarget().isIntrinsicSupported(ID))
3651 return handleUnsupportedIntrinsic(CI, ID, MIRBuilder);
3652
3653 if (translateKnownIntrinsic(CI, ID, MIRBuilder))
3654 return true;
3655
3657 TLI->getTgtMemIntrinsic(Infos, CI, *MF, ID);
3658
3659 return translateIntrinsic(CI, ID, MIRBuilder, Infos);
3660}
3661
3662bool IRTranslatorImpl::handleUnsupportedIntrinsic(
3663 const CallBase &CB, Intrinsic::ID ID, MachineIRBuilder &MIRBuilder) {
3664 const Function &F = MF->getFunction();
3665 F.getContext().diagnose(
3666 DiagnosticInfoUnsupportedTargetIntrinsic(F, ID, CB.getDebugLoc()));
3667
3668 if (!CB.getType()->isVoidTy()) {
3669 for (Register Reg : getOrCreateVRegs(CB))
3670 MIRBuilder.buildUndef(Reg);
3671 }
3672
3673 return true;
3674}
3675
3676/// Translate a call or callbr to an intrinsic.
3677bool IRTranslatorImpl::translateIntrinsic(
3678 const CallBase &CB, Intrinsic::ID ID, MachineIRBuilder &MIRBuilder,
3679 ArrayRef<TargetLowering::IntrinsicInfo> TgtMemIntrinsicInfos) {
3680 if (!MF->getSubtarget().isIntrinsicSupported(ID))
3681 return handleUnsupportedIntrinsic(CB, ID, MIRBuilder);
3682
3683 ArrayRef<Register> ResultRegs;
3684 if (!CB.getType()->isVoidTy())
3685 ResultRegs = getOrCreateVRegs(CB);
3686
3687 // Ignore the callsite attributes. Backend code is most likely not expecting
3688 // an intrinsic to sometimes have side effects and sometimes not.
3689 MachineInstrBuilder MIB = MIRBuilder.buildIntrinsic(ID, ResultRegs);
3690 if (isa<FPMathOperator>(CB))
3691 MIB->copyIRFlags(CB);
3692
3693 for (const auto &Arg : enumerate(CB.args())) {
3694 // If this is required to be an immediate, don't materialize it in a
3695 // register.
3696 if (CB.paramHasAttr(Arg.index(), Attribute::ImmArg)) {
3697 if (ConstantInt *CI = dyn_cast<ConstantInt>(Arg.value())) {
3698 // imm arguments are more convenient than cimm (and realistically
3699 // probably sufficient), so use them.
3700 assert(CI->getBitWidth() <= 64 &&
3701 "large intrinsic immediates not handled");
3702 MIB.addImm(CI->getSExtValue());
3703 } else {
3704 MIB.addFPImm(cast<ConstantFP>(Arg.value()));
3705 }
3706 } else if (auto *MDVal = dyn_cast<MetadataAsValue>(Arg.value())) {
3707 auto *MD = MDVal->getMetadata();
3708 auto *MDN = dyn_cast<MDNode>(MD);
3709 if (!MDN) {
3710 if (auto *ConstMD = dyn_cast<ConstantAsMetadata>(MD))
3711 MDN = MDNode::get(MF->getFunction().getContext(), ConstMD);
3712 else // This was probably an MDString.
3713 return false;
3714 }
3715 MIB.addMetadata(MDN);
3716 } else {
3717 ArrayRef<Register> VRegs = getOrCreateVRegs(*Arg.value());
3718 if (VRegs.size() > 1)
3719 return false;
3720 MIB.addUse(VRegs[0]);
3721 }
3722 }
3723
3724 // Add MachineMemOperands for each memory access described by the target.
3725 for (const auto &Info : TgtMemIntrinsicInfos) {
3726 Align Alignment = Info.align.value_or(
3727 DL->getABITypeAlign(Info.memVT.getTypeForEVT(CB.getContext())));
3728 LLT MemTy = Info.memVT.isSimple()
3729 ? getLLTForMVT(Info.memVT.getSimpleVT())
3730 : LLT::scalar(Info.memVT.getStoreSizeInBits());
3731
3732 // TODO: We currently just fallback to address space 0 if
3733 // getTgtMemIntrinsic didn't yield anything useful.
3734 MachinePointerInfo MPI;
3735 if (Info.ptrVal) {
3736 MPI = MachinePointerInfo(Info.ptrVal, Info.offset);
3737 } else if (Info.fallbackAddressSpace) {
3738 MPI = MachinePointerInfo(*Info.fallbackAddressSpace);
3739 }
3740 MIB.addMemOperand(MF->getMachineMemOperand(
3741 MPI, Info.flags, MemTy, Alignment, CB.getAAMetadata(), Info.ssid,
3742 Info.order, Info.failureOrder));
3743 }
3744
3745 if (CB.isConvergent()) {
3746 if (auto Bundle = CB.getOperandBundle(LLVMContext::OB_convergencectrl)) {
3747 auto *Token = Bundle->Inputs[0].get();
3748 Register TokenReg = getOrCreateVReg(*Token);
3749 MIB.addUse(TokenReg, RegState::Implicit);
3750 }
3751 }
3752
3754 MIB->setDeactivationSymbol(*MF, Bundle->Inputs[0].get());
3755
3756 return true;
3757}
3758
3759bool IRTranslatorImpl::findUnwindDestinations(
3760 const BasicBlock *EHPadBB, BranchProbability Prob,
3761 SmallVectorImpl<std::pair<MachineBasicBlock *, BranchProbability>>
3762 &UnwindDests) {
3764 EHPadBB->getParent()->getFunction().getPersonalityFn());
3765 bool IsMSVCCXX = Personality == EHPersonality::MSVC_CXX;
3766 bool IsCoreCLR = Personality == EHPersonality::CoreCLR;
3767 bool IsWasmCXX = Personality == EHPersonality::Wasm_CXX;
3768 bool IsWasmD = Personality == EHPersonality::Wasm_D;
3769 bool IsSEH = isAsynchronousEHPersonality(Personality);
3770
3771 if (IsWasmCXX || IsWasmD) {
3772 // Ignore this for now.
3773 return false;
3774 }
3775
3776 while (EHPadBB) {
3778 BasicBlock *NewEHPadBB = nullptr;
3779 if (isa<LandingPadInst>(Pad)) {
3780 // Stop on landingpads. They are not funclets.
3781 UnwindDests.emplace_back(&getMBB(*EHPadBB), Prob);
3782 break;
3783 }
3784 if (isa<CleanupPadInst>(Pad)) {
3785 // Stop on cleanup pads. Cleanups are always funclet entries for all known
3786 // personalities.
3787 UnwindDests.emplace_back(&getMBB(*EHPadBB), Prob);
3788 UnwindDests.back().first->setIsEHScopeEntry();
3789 UnwindDests.back().first->setIsEHFuncletEntry();
3790 break;
3791 }
3792 if (auto *CatchSwitch = dyn_cast<CatchSwitchInst>(Pad)) {
3793 // Add the catchpad handlers to the possible destinations.
3794 for (const BasicBlock *CatchPadBB : CatchSwitch->handlers()) {
3795 UnwindDests.emplace_back(&getMBB(*CatchPadBB), Prob);
3796 // For MSVC++ and the CLR, catchblocks are funclets and need prologues.
3797 if (IsMSVCCXX || IsCoreCLR)
3798 UnwindDests.back().first->setIsEHFuncletEntry();
3799 if (!IsSEH)
3800 UnwindDests.back().first->setIsEHScopeEntry();
3801 }
3802 NewEHPadBB = CatchSwitch->getUnwindDest();
3803 } else {
3804 continue;
3805 }
3806
3807 BranchProbabilityInfo *BPI = FuncInfo.BPI;
3808 if (BPI && NewEHPadBB)
3809 Prob *= BPI->getEdgeProbability(EHPadBB, NewEHPadBB);
3810 EHPadBB = NewEHPadBB;
3811 }
3812 return true;
3813}
3814
3815bool IRTranslatorImpl::translateInvoke(const User &U,
3816 MachineIRBuilder &MIRBuilder) {
3817 const InvokeInst &I = cast<InvokeInst>(U);
3818 MCContext &Context = MF->getContext();
3819
3820 const BasicBlock *ReturnBB = I.getSuccessor(0);
3821 const BasicBlock *EHPadBB = I.getSuccessor(1);
3822
3823 const Function *Fn = I.getCalledFunction();
3824
3825 // FIXME: support invoking patchpoint and statepoint intrinsics.
3826 if (Fn && Fn->isIntrinsic())
3827 return false;
3828
3829 // FIXME: support whatever these are.
3830 if (I.hasDeoptState())
3831 return false;
3832
3833 // FIXME: support control flow guard targets.
3834 if (I.countOperandBundlesOfType(LLVMContext::OB_cfguardtarget))
3835 return false;
3836
3837 // FIXME: support Windows exception handling.
3838 if (!isa<LandingPadInst>(EHPadBB->getFirstNonPHIIt()))
3839 return false;
3840
3841 // FIXME: support Windows dllimport function calls and calls through
3842 // weak symbols.
3843 if (Fn && (Fn->hasDLLImportStorageClass() ||
3844 (MF->getTarget().getTargetTriple().isOSWindows() &&
3845 Fn->hasExternalWeakLinkage())))
3846 return false;
3847
3848 bool LowerInlineAsm = I.isInlineAsm();
3849 bool NeedEHLabel = true;
3850
3851 // Emit the actual call, bracketed by EH_LABELs so that the MF knows about
3852 // the region covered by the try.
3853 MCSymbol *BeginSymbol = nullptr;
3854 if (NeedEHLabel) {
3855 MIRBuilder.buildInstr(TargetOpcode::G_INVOKE_REGION_START);
3856 BeginSymbol = Context.createTempSymbol();
3857 MIRBuilder.buildInstr(TargetOpcode::EH_LABEL).addSym(BeginSymbol);
3858 }
3859
3860 if (LowerInlineAsm) {
3861 if (!translateInlineAsm(I, MIRBuilder))
3862 return false;
3863 } else if (!translateCallBase(I, MIRBuilder))
3864 return false;
3865
3866 MCSymbol *EndSymbol = nullptr;
3867 if (NeedEHLabel) {
3868 EndSymbol = Context.createTempSymbol();
3869 MIRBuilder.buildInstr(TargetOpcode::EH_LABEL).addSym(EndSymbol);
3870 }
3871
3873 BranchProbabilityInfo *BPI = FuncInfo.BPI;
3874 MachineBasicBlock *InvokeMBB = &MIRBuilder.getMBB();
3875 BranchProbability EHPadBBProb =
3876 BPI ? BPI->getEdgeProbability(InvokeMBB->getBasicBlock(), EHPadBB)
3878
3879 if (!findUnwindDestinations(EHPadBB, EHPadBBProb, UnwindDests))
3880 return false;
3881
3882 MachineBasicBlock &EHPadMBB = getMBB(*EHPadBB),
3883 &ReturnMBB = getMBB(*ReturnBB);
3884 // Update successor info.
3885 addSuccessorWithProb(InvokeMBB, &ReturnMBB);
3886 for (auto &UnwindDest : UnwindDests) {
3887 UnwindDest.first->setIsEHPad();
3888 addSuccessorWithProb(InvokeMBB, UnwindDest.first, UnwindDest.second);
3889 }
3890 InvokeMBB->normalizeSuccProbs();
3891
3892 if (NeedEHLabel) {
3893 assert(BeginSymbol && "Expected a begin symbol!");
3894 assert(EndSymbol && "Expected an end symbol!");
3895 MF->addInvoke(&EHPadMBB, BeginSymbol, EndSymbol);
3896 }
3897
3898 MIRBuilder.buildBr(ReturnMBB);
3899 return true;
3900}
3901
3902/// The intrinsics currently supported by callbr are implicit control flow
3903/// intrinsics such as amdgcn.kill.
3904bool IRTranslatorImpl::translateCallBr(const User &U,
3905 MachineIRBuilder &MIRBuilder) {
3906 if (!mayTranslateUserTypes(U))
3907 return false; // see translateCall
3908
3909 const CallBrInst &I = cast<CallBrInst>(U);
3910 MachineBasicBlock *CallBrMBB = &MIRBuilder.getMBB();
3911
3912 Intrinsic::ID IID = I.getIntrinsicID();
3913 if (I.isInlineAsm()) {
3914 // FIXME: inline asm is not yet supported for callbr in GlobalISel. As soon
3915 // as we add support, we need to handle the indirect asm targets, see
3916 // SelectionDAGBuilder::visitCallBr().
3917 return false;
3918 }
3919 if (!translateIntrinsic(I, IID, MIRBuilder))
3920 return false;
3921
3922 // Retrieve successors.
3923 SmallPtrSet<BasicBlock *, 8> Dests = {I.getDefaultDest()};
3924 MachineBasicBlock *Return = &getMBB(*I.getDefaultDest());
3925
3926 // Update successor info.
3927 addSuccessorWithProb(CallBrMBB, Return, BranchProbability::getOne());
3928
3929 // Add indirect targets as successors. For intrinsic callbr, these represent
3930 // implicit control flow (e.g., the "kill" path for amdgcn.kill). We mark them
3931 // with setIsInlineAsmBrIndirectTarget so the machine verifier accepts them as
3932 // valid successors, even though they're not from inline asm.
3933 for (BasicBlock *Dest : I.getIndirectDests()) {
3934 MachineBasicBlock &Target = getMBB(*Dest);
3935 Target.setIsInlineAsmBrIndirectTarget();
3936 Target.setLabelMustBeEmitted();
3937 // Don't add duplicate machine successors.
3938 if (Dests.insert(Dest).second)
3939 addSuccessorWithProb(CallBrMBB, &Target, BranchProbability::getZero());
3940 }
3941
3942 CallBrMBB->normalizeSuccProbs();
3943
3944 // Drop into default successor.
3945 MIRBuilder.buildBr(*Return);
3946
3947 return true;
3948}
3949
3950bool IRTranslatorImpl::translateLandingPad(const User &U,
3951 MachineIRBuilder &MIRBuilder) {
3952 const LandingPadInst &LP = cast<LandingPadInst>(U);
3953
3954 MachineBasicBlock &MBB = MIRBuilder.getMBB();
3955
3956 MBB.setIsEHPad();
3957
3958 // If there aren't registers to copy the values into (e.g., during SjLj
3959 // exceptions), then don't bother.
3960 const Constant *PersonalityFn = MF->getFunction().getPersonalityFn();
3961 if (TLI->getExceptionPointerRegister(FuncInfo.ExceptionModel,
3962 PersonalityFn) == 0 &&
3963 TLI->getExceptionSelectorRegister(FuncInfo.ExceptionModel,
3964 PersonalityFn) == 0)
3965 return true;
3966
3967 // If landingpad's return type is token type, we don't create DAG nodes
3968 // for its exception pointer and selector value. The extraction of exception
3969 // pointer or selector value from token type landingpads is not currently
3970 // supported.
3971 if (LP.getType()->isTokenTy())
3972 return true;
3973
3975 return false;
3976
3977 // Add a label to mark the beginning of the landing pad. Deletion of the
3978 // landing pad can thus be detected via the MachineModuleInfo.
3979 MIRBuilder.buildInstr(TargetOpcode::EH_LABEL)
3980 .addSym(MF->addLandingPad(&MBB));
3981
3982 // If the unwinder does not preserve all registers, ensure that the
3983 // function marks the clobbered registers as used.
3984 const TargetRegisterInfo &TRI = *MF->getSubtarget().getRegisterInfo();
3985 if (auto *RegMask = TRI.getCustomEHPadPreservedMask(*MF))
3986 MF->getRegInfo().addPhysRegsUsedFromRegMask(RegMask);
3987
3988 LLT Ty = getLLTForType(*LP.getType(), *DL);
3989 Register Undef = MRI->createGenericVirtualRegister(Ty);
3990 MIRBuilder.buildUndef(Undef);
3991
3993 for (Type *Ty : cast<StructType>(LP.getType())->elements())
3994 Tys.push_back(getLLTForType(*Ty, *DL));
3995 assert(Tys.size() == 2 && "Only two-valued landingpads are supported");
3996
3997 // Mark exception register as live in.
3998 Register ExceptionReg =
3999 TLI->getExceptionPointerRegister(FuncInfo.ExceptionModel, PersonalityFn);
4000 if (!ExceptionReg)
4001 return false;
4002
4003 MBB.addLiveIn(ExceptionReg);
4004 ArrayRef<Register> ResRegs = getOrCreateVRegs(LP);
4005 MIRBuilder.buildCopy(ResRegs[0], ExceptionReg);
4006
4007 Register SelectorReg =
4008 TLI->getExceptionSelectorRegister(FuncInfo.ExceptionModel, PersonalityFn);
4009 if (!SelectorReg)
4010 return false;
4011
4012 MBB.addLiveIn(SelectorReg);
4013 Register PtrVReg = MRI->createGenericVirtualRegister(Tys[0]);
4014 MIRBuilder.buildCopy(PtrVReg, SelectorReg);
4015 MIRBuilder.buildCast(ResRegs[1], PtrVReg);
4016
4017 return true;
4018}
4019
4020bool IRTranslatorImpl::translateAlloca(const User &U,
4021 MachineIRBuilder &MIRBuilder) {
4022 auto &AI = cast<AllocaInst>(U);
4023
4024 if (AI.isSwiftError())
4025 return true;
4026
4027 if (AI.isStaticAlloca()) {
4028 Register Res = getOrCreateVReg(AI);
4029 int FI = getOrCreateFrameIndex(AI);
4030 MIRBuilder.buildFrameIndex(Res, FI);
4031 return true;
4032 }
4033
4034 // FIXME: support stack probing for Windows.
4035 if (MF->getTarget().getTargetTriple().isOSWindows())
4036 return false;
4037
4038 // Now we're in the harder dynamic case.
4039 Register NumElts = getOrCreateVReg(*AI.getArraySize());
4040 Type *IntPtrIRTy = DL->getIntPtrType(AI.getType());
4041 LLT IntPtrTy = getLLTForType(*IntPtrIRTy, *DL);
4042 if (MRI->getType(NumElts) != IntPtrTy) {
4043 Register ExtElts = MRI->createGenericVirtualRegister(IntPtrTy);
4044 MIRBuilder.buildZExtOrTrunc(ExtElts, NumElts);
4045 NumElts = ExtElts;
4046 }
4047
4048 TypeSize TySize = AI.getAllocationBaseSize(*DL);
4049
4050 Register AllocSize = MRI->createGenericVirtualRegister(IntPtrTy);
4051 Register TySizeReg;
4052 if (TySize.isScalable()) {
4053 // For scalable types, use vscale * min_value
4054 TySizeReg = MRI->createGenericVirtualRegister(IntPtrTy);
4055 MIRBuilder.buildVScale(TySizeReg, TySize.getKnownMinValue());
4056 } else {
4057 // For fixed types, use a constant
4058 TySizeReg =
4059 getOrCreateVReg(*ConstantInt::get(IntPtrIRTy, TySize.getFixedValue()));
4060 }
4061 MIRBuilder.buildMul(AllocSize, NumElts, TySizeReg);
4062
4063 // Round the size of the allocation up to the stack alignment size
4064 // by add SA-1 to the size. This doesn't overflow because we're computing
4065 // an address inside an alloca.
4066 Align StackAlign = MF->getSubtarget().getFrameLowering()->getStackAlign();
4067 auto SAMinusOne = MIRBuilder.buildConstant(IntPtrTy, StackAlign.value() - 1);
4068 auto AllocAdd = MIRBuilder.buildAdd(IntPtrTy, AllocSize, SAMinusOne,
4070 auto AlignCst =
4071 MIRBuilder.buildConstant(IntPtrTy, ~(uint64_t)(StackAlign.value() - 1));
4072 auto AlignedAlloc = MIRBuilder.buildAnd(IntPtrTy, AllocAdd, AlignCst);
4073
4074 Align Alignment = AI.getAlign();
4075 if (Alignment <= StackAlign)
4076 Alignment = Align(1);
4077 MIRBuilder.buildDynStackAlloc(getOrCreateVReg(AI), AlignedAlloc, Alignment);
4078
4079 MF->getFrameInfo().CreateVariableSizedObject(Alignment, &AI);
4080 assert(MF->getFrameInfo().hasVarSizedObjects());
4081 return true;
4082}
4083
4084bool IRTranslatorImpl::translateVAArg(const User &U,
4085 MachineIRBuilder &MIRBuilder) {
4086 // FIXME: We may need more info about the type. Because of how LLT works,
4087 // we're completely discarding the i64/double distinction here (amongst
4088 // others). Fortunately the ABIs I know of where that matters don't use va_arg
4089 // anyway but that's not guaranteed.
4090 MIRBuilder.buildInstr(TargetOpcode::G_VAARG, {getOrCreateVReg(U)},
4091 {getOrCreateVReg(*U.getOperand(0)),
4092 DL->getABITypeAlign(U.getType()).value()});
4093 return true;
4094}
4095
4096bool IRTranslatorImpl::translateUnreachable(const User &U,
4097 MachineIRBuilder &MIRBuilder) {
4098 auto &UI = cast<UnreachableInst>(U);
4099 if (!UI.shouldLowerToTrap(MF->getTarget().Options.TrapUnreachable,
4100 MF->getTarget().Options.NoTrapAfterNoreturn))
4101 return true;
4102
4103 MIRBuilder.buildTrap();
4104 return true;
4105}
4106
4107bool IRTranslatorImpl::translateInsertElement(const User &U,
4108 MachineIRBuilder &MIRBuilder) {
4109 // If it is a <1 x Ty> vector, use the scalar as it is
4110 // not a legal vector type in LLT.
4111 if (auto *FVT = dyn_cast<FixedVectorType>(U.getType());
4112 FVT && FVT->getNumElements() == 1)
4113 return translateCopy(U, *U.getOperand(1), MIRBuilder);
4114
4115 Register Res = getOrCreateVReg(U);
4116 Register Val = getOrCreateVReg(*U.getOperand(0));
4117 Register Elt = getOrCreateVReg(*U.getOperand(1));
4118 unsigned PreferredVecIdxWidth = TLI->getVectorIdxWidth(*DL);
4119 Register Idx;
4120 if (auto *CI = dyn_cast<ConstantInt>(U.getOperand(2))) {
4121 if (CI->getBitWidth() != PreferredVecIdxWidth) {
4122 APInt NewIdx = CI->getValue().zextOrTrunc(PreferredVecIdxWidth);
4123 auto *NewIdxCI = ConstantInt::get(CI->getContext(), NewIdx);
4124 Idx = getOrCreateVReg(*NewIdxCI);
4125 }
4126 }
4127 if (!Idx)
4128 Idx = getOrCreateVReg(*U.getOperand(2));
4129 if (MRI->getType(Idx).getSizeInBits() != PreferredVecIdxWidth) {
4130 const LLT VecIdxTy =
4131 MRI->getType(Idx).changeElementSize(PreferredVecIdxWidth);
4132 Idx = MIRBuilder.buildZExtOrTrunc(VecIdxTy, Idx).getReg(0);
4133 }
4134 MIRBuilder.buildInsertVectorElement(Res, Val, Elt, Idx);
4135 return true;
4136}
4137
4138bool IRTranslatorImpl::translateInsertVector(const User &U,
4139 MachineIRBuilder &MIRBuilder) {
4140 Register Dst = getOrCreateVReg(U);
4141 Register Vec = getOrCreateVReg(*U.getOperand(0));
4142 Register Elt = getOrCreateVReg(*U.getOperand(1));
4143
4144 ConstantInt *CI = cast<ConstantInt>(U.getOperand(2));
4145 unsigned PreferredVecIdxWidth = TLI->getVectorIdxWidth(*DL);
4146
4147 // Resize Index to preferred index width.
4148 if (CI->getBitWidth() != PreferredVecIdxWidth) {
4149 APInt NewIdx = CI->getValue().zextOrTrunc(PreferredVecIdxWidth);
4150 CI = ConstantInt::get(CI->getContext(), NewIdx);
4151 }
4152
4153 // If it is a <1 x Ty> vector, we have to use other means.
4154 if (auto *ResultType = dyn_cast<FixedVectorType>(U.getOperand(1)->getType());
4155 ResultType && ResultType->getNumElements() == 1) {
4156 if (auto *InputType = dyn_cast<FixedVectorType>(U.getOperand(0)->getType());
4157 InputType && InputType->getNumElements() == 1) {
4158 // We are inserting an illegal fixed vector into an illegal
4159 // fixed vector, use the scalar as it is not a legal vector type
4160 // in LLT.
4161 return translateCopy(U, Vec, MIRBuilder);
4162 }
4163 if (isa<FixedVectorType>(U.getOperand(0)->getType())) {
4164 // We are inserting an illegal fixed vector into a legal fixed
4165 // vector, use the scalar as it is not a legal vector type in
4166 // LLT.
4167 Register Idx = getOrCreateVReg(*CI);
4168 MIRBuilder.buildInsertVectorElement(Dst, Vec, Elt, Idx);
4169 return true;
4170 }
4171 if (isa<ScalableVectorType>(U.getOperand(0)->getType())) {
4172 // We are inserting an illegal fixed vector into a scalable
4173 // vector, use a scalar element insert.
4174 LLT VecIdxTy = LLT::integer(PreferredVecIdxWidth);
4175 Register Idx = getOrCreateVReg(*CI);
4176 auto ScaledIndex = MIRBuilder.buildMul(
4177 VecIdxTy, MIRBuilder.buildVScale(VecIdxTy, 1), Idx);
4178 MIRBuilder.buildInsertVectorElement(Dst, Vec, Elt, ScaledIndex);
4179 return true;
4180 }
4181 }
4182
4183 MIRBuilder.buildInsertSubvector(Dst, Vec, Elt, CI->getZExtValue());
4184 return true;
4185}
4186
4187bool IRTranslatorImpl::translateExtractElement(const User &U,
4188 MachineIRBuilder &MIRBuilder) {
4189 // If it is a <1 x Ty> vector, use the scalar as it is
4190 // not a legal vector type in LLT.
4191 if (const FixedVectorType *FVT =
4192 dyn_cast<FixedVectorType>(U.getOperand(0)->getType()))
4193 if (FVT->getNumElements() == 1)
4194 return translateCopy(U, *U.getOperand(0), MIRBuilder);
4195
4196 Register Res = getOrCreateVReg(U);
4197 Register Val = getOrCreateVReg(*U.getOperand(0));
4198 unsigned PreferredVecIdxWidth = TLI->getVectorIdxWidth(*DL);
4199 Register Idx;
4200 if (auto *CI = dyn_cast<ConstantInt>(U.getOperand(1))) {
4201 if (CI->getBitWidth() != PreferredVecIdxWidth) {
4202 APInt NewIdx = CI->getValue().zextOrTrunc(PreferredVecIdxWidth);
4203 auto *NewIdxCI = ConstantInt::get(CI->getContext(), NewIdx);
4204 Idx = getOrCreateVReg(*NewIdxCI);
4205 }
4206 }
4207 if (!Idx)
4208 Idx = getOrCreateVReg(*U.getOperand(1));
4209 if (MRI->getType(Idx).getSizeInBits() != PreferredVecIdxWidth) {
4210 const LLT VecIdxTy =
4211 MRI->getType(Idx).changeElementSize(PreferredVecIdxWidth);
4212 Idx = MIRBuilder.buildZExtOrTrunc(VecIdxTy, Idx).getReg(0);
4213 }
4214 MIRBuilder.buildExtractVectorElement(Res, Val, Idx);
4215 return true;
4216}
4217
4218bool IRTranslatorImpl::translateExtractVector(const User &U,
4219 MachineIRBuilder &MIRBuilder) {
4220 Register Res = getOrCreateVReg(U);
4221 Register Vec = getOrCreateVReg(*U.getOperand(0));
4222 ConstantInt *CI = cast<ConstantInt>(U.getOperand(1));
4223 unsigned PreferredVecIdxWidth = TLI->getVectorIdxWidth(*DL);
4224
4225 // Resize Index to preferred index width.
4226 if (CI->getBitWidth() != PreferredVecIdxWidth) {
4227 APInt NewIdx = CI->getValue().zextOrTrunc(PreferredVecIdxWidth);
4228 CI = ConstantInt::get(CI->getContext(), NewIdx);
4229 }
4230
4231 // If it is a <1 x Ty> vector, we have to use other means.
4232 if (auto *ResultType = dyn_cast<FixedVectorType>(U.getType());
4233 ResultType && ResultType->getNumElements() == 1) {
4234 if (auto *InputType = dyn_cast<FixedVectorType>(U.getOperand(0)->getType());
4235 InputType && InputType->getNumElements() == 1) {
4236 // We are extracting an illegal fixed vector from an illegal fixed vector,
4237 // use the scalar as it is not a legal vector type in LLT.
4238 return translateCopy(U, Vec, MIRBuilder);
4239 }
4240 if (isa<FixedVectorType>(U.getOperand(0)->getType())) {
4241 // We are extracting an illegal fixed vector from a legal fixed
4242 // vector, use the scalar as it is not a legal vector type in
4243 // LLT.
4244 Register Idx = getOrCreateVReg(*CI);
4245 MIRBuilder.buildExtractVectorElement(Res, Vec, Idx);
4246 return true;
4247 }
4248 if (isa<ScalableVectorType>(U.getOperand(0)->getType())) {
4249 // We are extracting an illegal fixed vector from a scalable
4250 // vector, use a scalar element extract.
4251 LLT VecIdxTy = LLT::integer(PreferredVecIdxWidth);
4252 Register Idx = getOrCreateVReg(*CI);
4253 auto ScaledIndex = MIRBuilder.buildMul(
4254 VecIdxTy, MIRBuilder.buildVScale(VecIdxTy, 1), Idx);
4255 MIRBuilder.buildExtractVectorElement(Res, Vec, ScaledIndex);
4256 return true;
4257 }
4258 }
4259
4260 MIRBuilder.buildExtractSubvector(Res, Vec, CI->getZExtValue());
4261 return true;
4262}
4263
4264bool IRTranslatorImpl::translateShuffleVector(const User &U,
4265 MachineIRBuilder &MIRBuilder) {
4266 // A ShuffleVector that operates on scalable vectors is a splat vector where
4267 // the value of the splat vector is the 0th element of the first operand,
4268 // since the index mask operand is the zeroinitializer (undef and
4269 // poison are treated as zeroinitializer here).
4270 if (U.getOperand(0)->getType()->isScalableTy()) {
4271 Register Val = getOrCreateVReg(*U.getOperand(0));
4272 auto SplatVal = MIRBuilder.buildExtractVectorElementConstant(
4273 MRI->getType(Val).getElementType(), Val, 0);
4274 MIRBuilder.buildSplatVector(getOrCreateVReg(U), SplatVal);
4275 return true;
4276 }
4277
4278 ArrayRef<int> Mask;
4279 if (auto *SVI = dyn_cast<ShuffleVectorInst>(&U))
4280 Mask = SVI->getShuffleMask();
4281 else
4282 Mask = cast<ConstantExpr>(U).getShuffleMask();
4283
4284 // As GISel does not represent <1 x > vectors as a separate type from scalars,
4285 // we transform shuffle_vector with a scalar output to an
4286 // ExtractVectorElement. If the input type is also scalar it becomes a Copy.
4287 unsigned DstElts = cast<FixedVectorType>(U.getType())->getNumElements();
4288 unsigned SrcElts =
4289 cast<FixedVectorType>(U.getOperand(0)->getType())->getNumElements();
4290 if (DstElts == 1) {
4291 unsigned M = Mask[0];
4292 if (SrcElts == 1) {
4293 if (M == 0 || M == 1)
4294 return translateCopy(U, *U.getOperand(M), MIRBuilder);
4295 MIRBuilder.buildUndef(getOrCreateVReg(U));
4296 } else {
4297 Register Dst = getOrCreateVReg(U);
4298 if (M < SrcElts) {
4300 Dst, getOrCreateVReg(*U.getOperand(0)), M);
4301 } else if (M < SrcElts * 2) {
4303 Dst, getOrCreateVReg(*U.getOperand(1)), M - SrcElts);
4304 } else {
4305 MIRBuilder.buildUndef(Dst);
4306 }
4307 }
4308 return true;
4309 }
4310
4311 // A single element src is transformed to a build_vector.
4312 if (SrcElts == 1) {
4315 for (int M : Mask) {
4316 LLT SrcTy = getLLTForType(*U.getOperand(0)->getType(), *DL);
4317 if (M == 0 || M == 1) {
4318 Ops.push_back(getOrCreateVReg(*U.getOperand(M)));
4319 } else {
4320 if (!Undef.isValid()) {
4321 Undef = MRI->createGenericVirtualRegister(SrcTy);
4322 MIRBuilder.buildUndef(Undef);
4323 }
4324 Ops.push_back(Undef);
4325 }
4326 }
4327 MIRBuilder.buildBuildVector(getOrCreateVReg(U), Ops);
4328 return true;
4329 }
4330
4331 ArrayRef<int> MaskAlloc = MF->allocateShuffleMask(Mask);
4332 MIRBuilder
4333 .buildInstr(TargetOpcode::G_SHUFFLE_VECTOR, {getOrCreateVReg(U)},
4334 {getOrCreateVReg(*U.getOperand(0)),
4335 getOrCreateVReg(*U.getOperand(1))})
4336 .addShuffleMask(MaskAlloc);
4337 return true;
4338}
4339
4340bool IRTranslatorImpl::translateBitInsert(const User &U,
4341 MachineIRBuilder &MIRBuilder) {
4342 Register Res = getOrCreateVReg(U);
4343 Register Base = getOrCreateVReg(*U.getOperand(0));
4344 Register Val = getOrCreateVReg(*U.getOperand(1));
4345 Register Offset = getOrCreateVReg(*U.getOperand(2));
4346 MachineRegisterInfo &MRI = *MIRBuilder.getMRI();
4347 LLT BaseTy = MRI.getType(Base);
4348 LLT ValTy = MRI.getType(Val);
4349
4350 assert(BaseTy.getSizeInBits() >= ValTy.getSizeInBits() &&
4351 "bitinsert val wider than base should be rejected by verifier");
4352
4353 // If Val is a floating-point type, bitcast it to an integer of the same
4354 // size so buildZExtOrTrunc can safely extend or truncate it.
4355 if (ValTy.isFloat()) {
4356 ValTy = LLT::scalar(ValTy.getSizeInBits());
4357 Val = MIRBuilder.buildBitcast(ValTy, Val).getReg(0);
4358 } else if (ValTy.isPointer()) {
4359 ValTy = LLT::scalar(ValTy.getSizeInBits());
4360 Val = MIRBuilder.buildPtrToInt(ValTy, Val).getReg(0);
4361 }
4362
4363 // Convert Offset to the target's preferred shift amount type.
4364 LLT ShiftAmtTy = TLI->getPreferredShiftAmountTy(BaseTy);
4365 Register LegalOffset =
4366 MIRBuilder.buildZExtOrTrunc(ShiftAmtTy, Offset).getReg(0);
4367
4368 // Truncate or extend Val to BaseTy so only the inserted bit range remains.
4369 Register ExtVal = MIRBuilder.buildZExtOrTrunc(BaseTy, Val).getReg(0);
4370
4371 unsigned BaseBitWidth = BaseTy.getSizeInBits();
4372 unsigned ValBitWidth = ValTy.getSizeInBits();
4373 APInt InsertMask = APInt::getLowBitsSet(BaseBitWidth, ValBitWidth);
4374 Register MaskConst = MIRBuilder.buildConstant(BaseTy, InsertMask).getReg(0);
4375 Register ShiftedMask =
4376 MIRBuilder.buildShl(BaseTy, MaskConst, LegalOffset).getReg(0);
4377 Register ClearMask = MIRBuilder.buildNot(BaseTy, ShiftedMask).getReg(0);
4378 Register ClearedBase = MIRBuilder.buildAnd(BaseTy, Base, ClearMask).getReg(0);
4379 Register ShiftedVal =
4380 MIRBuilder.buildShl(BaseTy, ExtVal, LegalOffset).getReg(0);
4381 MIRBuilder.buildOr(Res, ClearedBase, ShiftedVal);
4382 return true;
4383}
4384
4385bool IRTranslatorImpl::translateBitExtract(const User &U,
4386 MachineIRBuilder &MIRBuilder) {
4387 Register Res = getOrCreateVReg(U);
4388 Register Src = getOrCreateVReg(*U.getOperand(0));
4389 Register Offset = getOrCreateVReg(*U.getOperand(1));
4390 MachineRegisterInfo &MRI = *MIRBuilder.getMRI();
4391 LLT SrcTy = MRI.getType(Src);
4392 LLT ResTy = MRI.getType(Res);
4393
4394 assert(ResTy.getSizeInBits() <= SrcTy.getSizeInBits() &&
4395 "bitextract result wider than source should be rejected by verifier");
4396
4397 // Convert Offset to the target's preferred shift amount type.
4398 LLT ShiftAmtTy = TLI->getPreferredShiftAmountTy(SrcTy);
4399 Register LegalOffset =
4400 MIRBuilder.buildZExtOrTrunc(ShiftAmtTy, Offset).getReg(0);
4401
4402 // Shift right by Offset to bring the target field down to bit 0.
4403 Register Shifted = MIRBuilder.buildLShr(SrcTy, Src, LegalOffset).getReg(0);
4404
4405 if (ResTy.isFloat()) {
4406 // Drop into the integer domain to safely handle the size conversion
4407 LLT IntResTy = LLT::scalar(ResTy.getSizeInBits());
4408 Register IntRes = MRI.createGenericVirtualRegister(IntResTy);
4409
4410 if (SrcTy == IntResTy)
4411 MIRBuilder.buildCopy(IntRes, Shifted);
4412 else
4413 MIRBuilder.buildTrunc(IntRes, Shifted);
4414
4415 // Bitcast the raw integer bits back into the requested floating-point
4416 // register
4417 MIRBuilder.buildBitcast(Res, IntRes);
4418 } else if (ResTy.isPointer()) {
4419 // Drop into the integer domain to safely handle the size conversion
4420 LLT IntResTy = LLT::scalar(ResTy.getSizeInBits());
4421 Register IntRes = MRI.createGenericVirtualRegister(IntResTy);
4422
4423 if (SrcTy == IntResTy)
4424 MIRBuilder.buildCopy(IntRes, Shifted);
4425 else
4426 MIRBuilder.buildTrunc(IntRes, Shifted);
4427
4428 MIRBuilder.buildIntToPtr(Res, IntRes);
4429 } else {
4430 // Normal integer path
4431 if (SrcTy == ResTy)
4432 MIRBuilder.buildCopy(Res, Shifted);
4433 else
4434 MIRBuilder.buildTrunc(Res, Shifted);
4435 }
4436
4437 return true;
4438}
4439
4440bool IRTranslatorImpl::translatePHI(const User &U,
4441 MachineIRBuilder &MIRBuilder) {
4442 const PHINode &PI = cast<PHINode>(U);
4443
4444 SmallVector<MachineInstr *, 4> Insts;
4445 for (auto Reg : getOrCreateVRegs(PI)) {
4446 auto MIB = MIRBuilder.buildInstr(TargetOpcode::G_PHI, {Reg}, {});
4447 Insts.push_back(MIB.getInstr());
4448 }
4449
4450 PendingPHIs.emplace_back(&PI, std::move(Insts));
4451 return true;
4452}
4453
4454bool IRTranslatorImpl::translateAtomicCmpXchg(const User &U,
4455 MachineIRBuilder &MIRBuilder) {
4456 const AtomicCmpXchgInst &I = cast<AtomicCmpXchgInst>(U);
4457
4458 auto Flags = TLI->getAtomicMemOperandFlags(I, *DL);
4459
4460 auto Res = getOrCreateVRegs(I);
4461 Register OldValRes = Res[0];
4462 Register SuccessRes = Res[1];
4463 Register Addr = getOrCreateVReg(*I.getPointerOperand());
4464 Register Cmp = getOrCreateVReg(*I.getCompareOperand());
4465 Register NewVal = getOrCreateVReg(*I.getNewValOperand());
4466
4468 OldValRes, SuccessRes, Addr, Cmp, NewVal,
4469 *MF->getMachineMemOperand(
4470 MachinePointerInfo(I.getPointerOperand()), Flags, MRI->getType(Cmp),
4471 getMemOpAlign(I), I.getAAMetadata(), I.getSyncScopeID(),
4472 I.getSuccessOrdering(), I.getFailureOrdering()));
4473 return true;
4474}
4475
4476bool IRTranslatorImpl::translateAtomicRMW(const User &U,
4477 MachineIRBuilder &MIRBuilder) {
4478 if (!mayTranslateUserTypes(U))
4479 return false;
4480
4481 const AtomicRMWInst &I = cast<AtomicRMWInst>(U);
4482 auto Flags = TLI->getAtomicMemOperandFlags(I, *DL);
4483
4484 Register Res = getOrCreateVReg(I);
4485 Register Addr = getOrCreateVReg(*I.getPointerOperand());
4486 Register Val = getOrCreateVReg(*I.getValOperand());
4487
4488 unsigned Opcode = 0;
4489 switch (I.getOperation()) {
4490 default:
4491 return false;
4493 Opcode = TargetOpcode::G_ATOMICRMW_XCHG;
4494 break;
4495 case AtomicRMWInst::Add:
4496 Opcode = TargetOpcode::G_ATOMICRMW_ADD;
4497 break;
4498 case AtomicRMWInst::Sub:
4499 Opcode = TargetOpcode::G_ATOMICRMW_SUB;
4500 break;
4501 case AtomicRMWInst::And:
4502 Opcode = TargetOpcode::G_ATOMICRMW_AND;
4503 break;
4505 Opcode = TargetOpcode::G_ATOMICRMW_NAND;
4506 break;
4507 case AtomicRMWInst::Or:
4508 Opcode = TargetOpcode::G_ATOMICRMW_OR;
4509 break;
4510 case AtomicRMWInst::Xor:
4511 Opcode = TargetOpcode::G_ATOMICRMW_XOR;
4512 break;
4513 case AtomicRMWInst::Max:
4514 Opcode = TargetOpcode::G_ATOMICRMW_MAX;
4515 break;
4516 case AtomicRMWInst::Min:
4517 Opcode = TargetOpcode::G_ATOMICRMW_MIN;
4518 break;
4520 Opcode = TargetOpcode::G_ATOMICRMW_UMAX;
4521 break;
4523 Opcode = TargetOpcode::G_ATOMICRMW_UMIN;
4524 break;
4526 Opcode = TargetOpcode::G_ATOMICRMW_FADD;
4527 break;
4529 Opcode = TargetOpcode::G_ATOMICRMW_FSUB;
4530 break;
4532 Opcode = TargetOpcode::G_ATOMICRMW_FMAX;
4533 break;
4535 Opcode = TargetOpcode::G_ATOMICRMW_FMIN;
4536 break;
4538 Opcode = TargetOpcode::G_ATOMICRMW_FMAXIMUM;
4539 break;
4541 Opcode = TargetOpcode::G_ATOMICRMW_FMINIMUM;
4542 break;
4544 Opcode = TargetOpcode::G_ATOMICRMW_FMAXIMUMNUM;
4545 break;
4547 Opcode = TargetOpcode::G_ATOMICRMW_FMINIMUMNUM;
4548 break;
4550 Opcode = TargetOpcode::G_ATOMICRMW_UINC_WRAP;
4551 break;
4553 Opcode = TargetOpcode::G_ATOMICRMW_UDEC_WRAP;
4554 break;
4556 Opcode = TargetOpcode::G_ATOMICRMW_USUB_COND;
4557 break;
4559 Opcode = TargetOpcode::G_ATOMICRMW_USUB_SAT;
4560 break;
4561 }
4562
4563 MIRBuilder.buildAtomicRMW(
4564 Opcode, Res, Addr, Val,
4565 *MF->getMachineMemOperand(MachinePointerInfo(I.getPointerOperand()),
4566 Flags, MRI->getType(Val), getMemOpAlign(I),
4567 I.getAAMetadata(), I.getSyncScopeID(),
4568 I.getOrdering()));
4569 return true;
4570}
4571
4572bool IRTranslatorImpl::translateFence(const User &U,
4573 MachineIRBuilder &MIRBuilder) {
4574 const FenceInst &Fence = cast<FenceInst>(U);
4575 MIRBuilder.buildFence(static_cast<unsigned>(Fence.getOrdering()),
4576 Fence.getSyncScopeID());
4577 return true;
4578}
4579
4580bool IRTranslatorImpl::translateFreeze(const User &U,
4581 MachineIRBuilder &MIRBuilder) {
4582 const ArrayRef<Register> DstRegs = getOrCreateVRegs(U);
4583 const ArrayRef<Register> SrcRegs = getOrCreateVRegs(*U.getOperand(0));
4584
4585 assert(DstRegs.size() == SrcRegs.size() &&
4586 "Freeze with different source and destination type?");
4587
4588 for (unsigned I = 0; I < DstRegs.size(); ++I) {
4589 MIRBuilder.buildFreeze(DstRegs[I], SrcRegs[I]);
4590 }
4591
4592 return true;
4593}
4594
4595void IRTranslatorImpl::finishPendingPhis() {
4596#ifndef NDEBUG
4597 DILocationVerifier Verifier;
4598 GISelObserverWrapper WrapperObserver(&Verifier);
4599 RAIIMFObsDelInstaller ObsInstall(*MF, WrapperObserver);
4600#endif // ifndef NDEBUG
4601 for (auto &Phi : PendingPHIs) {
4602 const PHINode *PI = Phi.first;
4603 if (PI->getType()->isEmptyTy())
4604 continue;
4605 ArrayRef<MachineInstr *> ComponentPHIs = Phi.second;
4606 MachineBasicBlock *PhiMBB = ComponentPHIs[0]->getParent();
4607 EntryBuilder->setDebugLoc(PI->getDebugLoc());
4608#ifndef NDEBUG
4609 Verifier.setCurrentInst(PI);
4610#endif // ifndef NDEBUG
4611
4612 SmallPtrSet<const MachineBasicBlock *, 16> SeenPreds;
4613 for (unsigned i = 0; i < PI->getNumIncomingValues(); ++i) {
4614 auto IRPred = PI->getIncomingBlock(i);
4615 ArrayRef<Register> ValRegs = getOrCreateVRegs(*PI->getIncomingValue(i));
4616 for (auto *Pred : getMachinePredBBs({IRPred, PI->getParent()})) {
4617 if (SeenPreds.count(Pred) || !PhiMBB->isPredecessor(Pred))
4618 continue;
4619 SeenPreds.insert(Pred);
4620 for (unsigned j = 0; j < ValRegs.size(); ++j) {
4621 MachineInstrBuilder MIB(*MF, ComponentPHIs[j]);
4622 MIB.addUse(ValRegs[j]);
4623 MIB.addMBB(Pred);
4624 }
4625 }
4626 }
4627 }
4628}
4629
4630void IRTranslatorImpl::translateDbgValueRecord(Value *V, bool HasArgList,
4631 const DILocalVariable *Variable,
4632 const DIExpression *Expression,
4633 const DebugLoc &DL,
4634 MachineIRBuilder &MIRBuilder) {
4635 assert(Variable->isValidLocationForIntrinsic(DL) &&
4636 "Expected inlined-at fields to agree");
4637 // Act as if we're handling a debug intrinsic.
4638 MIRBuilder.setDebugLoc(DL);
4639
4640 if (!V || HasArgList) {
4641 // DI cannot produce a valid DBG_VALUE, so produce an undef DBG_VALUE to
4642 // terminate any prior location.
4643 MIRBuilder.buildIndirectDbgValue(0, Variable, Expression);
4644 return;
4645 }
4646
4647 if (const auto *CI = dyn_cast<Constant>(V)) {
4648 MIRBuilder.buildConstDbgValue(*CI, Variable, Expression);
4649 return;
4650 }
4651
4652 if (auto *AI = dyn_cast<AllocaInst>(V);
4653 AI && AI->isStaticAlloca() && Expression->startsWithDeref()) {
4654 // If the value is an alloca and the expression starts with a
4655 // dereference, track a stack slot instead of a register, as registers
4656 // may be clobbered.
4657 auto ExprOperands = Expression->getElements();
4658 auto *ExprDerefRemoved =
4659 DIExpression::get(AI->getContext(), ExprOperands.drop_front());
4660 MIRBuilder.buildFIDbgValue(getOrCreateFrameIndex(*AI), Variable,
4661 ExprDerefRemoved);
4662 return;
4663 }
4664 if (translateIfEntryValueArgument(false, V, Variable, Expression, DL,
4665 MIRBuilder))
4666 return;
4667 for (Register Reg : getOrCreateVRegs(*V)) {
4668 // FIXME: This does not handle register-indirect values at offset 0. The
4669 // direct/indirect thing shouldn't really be handled by something as
4670 // implicit as reg+noreg vs reg+imm in the first place, but it seems
4671 // pretty baked in right now.
4672 MIRBuilder.buildDirectDbgValue(Reg, Variable, Expression);
4673 }
4674}
4675
4676void IRTranslatorImpl::translateDbgDeclareRecord(
4677 Value *Address, bool HasArgList, const DILocalVariable *Variable,
4678 const DIExpression *Expression, const DebugLoc &DL,
4679 MachineIRBuilder &MIRBuilder) {
4680 if (!Address || isa<UndefValue>(Address)) {
4681 LLVM_DEBUG(dbgs() << "Dropping debug info for " << *Variable << "\n");
4682 return;
4683 }
4684
4685 assert(Variable->isValidLocationForIntrinsic(DL) &&
4686 "Expected inlined-at fields to agree");
4687 auto AI = dyn_cast<AllocaInst>(Address);
4688 if (AI && AI->isStaticAlloca()) {
4689 // Static allocas are tracked at the MF level, no need for DBG_VALUE
4690 // instructions (in fact, they get ignored if they *do* exist).
4691 MF->setVariableDbgInfo(Variable, Expression,
4692 getOrCreateFrameIndex(*AI), DL);
4693 return;
4694 }
4695
4696 if (translateIfEntryValueArgument(true, Address, Variable,
4697 Expression, DL,
4698 MIRBuilder))
4699 return;
4700
4701 // A dbg.declare describes the address of a source variable, so lower it
4702 // into an indirect DBG_VALUE.
4703 MIRBuilder.setDebugLoc(DL);
4704 MIRBuilder.buildIndirectDbgValue(getOrCreateVReg(*Address), Variable,
4705 Expression);
4706}
4707
4708void IRTranslatorImpl::translateDbgInfo(const Instruction &Inst,
4709 MachineIRBuilder &MIRBuilder) {
4710 for (DbgRecord &DR : Inst.getDbgRecordRange()) {
4711 if (DbgLabelRecord *DLR = dyn_cast<DbgLabelRecord>(&DR)) {
4712 MIRBuilder.setDebugLoc(DLR->getDebugLoc());
4713 assert(DLR->getLabel() && "Missing label");
4714 assert(DLR->getLabel()->isValidLocationForIntrinsic(
4715 MIRBuilder.getDebugLoc()) &&
4716 "Expected inlined-at fields to agree");
4717 MIRBuilder.buildDbgLabel(DLR->getLabel());
4718 continue;
4719 }
4720 DbgVariableRecord &DVR = cast<DbgVariableRecord>(DR);
4721 const DILocalVariable *Variable = DVR.getVariable();
4722 const DIExpression *Expression = DVR.getExpression();
4723 Value *V = DVR.getVariableLocationOp(0);
4724 if (DVR.isDbgDeclare())
4725 translateDbgDeclareRecord(V, DVR.hasArgList(), Variable, Expression,
4726 DVR.getDebugLoc(), MIRBuilder);
4727 else
4728 translateDbgValueRecord(V, DVR.hasArgList(), Variable, Expression,
4729 DVR.getDebugLoc(), MIRBuilder);
4730 }
4731}
4732
4733bool IRTranslatorImpl::translate(const Instruction &Inst) {
4734 CurBuilder->setDebugLoc(Inst.getDebugLoc());
4735 CurBuilder->setPCSections(Inst.getMetadata(LLVMContext::MD_pcsections));
4736 CurBuilder->setMMRAMetadata(Inst.getMetadata(LLVMContext::MD_mmra));
4737
4738 if (TLI->fallBackToDAGISel(Inst))
4739 return false;
4740
4741 switch (Inst.getOpcode()) {
4742#define HANDLE_INST(NUM, OPCODE, CLASS) \
4743 case Instruction::OPCODE: \
4744 return translate##OPCODE(Inst, *CurBuilder.get());
4745#include "llvm/IR/Instruction.def"
4746 default:
4747 return false;
4748 }
4749}
4750
4751bool IRTranslatorImpl::translate(const Constant &C, Register Reg) {
4752 // We only emit constants into the entry block from here. To prevent jumpy
4753 // debug behaviour remove debug line.
4754 if (auto CurrInstDL = CurBuilder->getDL())
4755 EntryBuilder->setDebugLoc(DebugLoc());
4756
4757 if (auto CI = dyn_cast<ConstantInt>(&C)) {
4758 // buildConstant expects a to-be-splatted scalar ConstantInt.
4759 if (isa<VectorType>(CI->getType()))
4760 CI = ConstantInt::get(CI->getContext(), CI->getValue());
4761 EntryBuilder->buildConstant(Reg, *CI);
4762 } else if (auto CB = dyn_cast<ConstantByte>(&C)) {
4763 // Byte constants share G_CONSTANT with integers; the destination Reg's
4764 // LLT (an integer LLT, see getLLTForType) determines vector splatting.
4765 EntryBuilder->buildConstant(Reg, CB->getValue());
4766 } else if (auto CF = dyn_cast<ConstantFP>(&C)) {
4767 // buildFConstant expects a to-be-splatted scalar ConstantFP.
4768 if (isa<VectorType>(CF->getType()))
4769 CF = ConstantFP::get(CF->getContext(), CF->getValue());
4770 EntryBuilder->buildFConstant(Reg, *CF);
4771 } else if (isa<UndefValue>(C))
4772 EntryBuilder->buildUndef(Reg);
4773 else if (isa<ConstantPointerNull>(C))
4774 EntryBuilder->buildConstant(Reg, 0);
4775 else if (auto GV = dyn_cast<GlobalValue>(&C))
4776 EntryBuilder->buildGlobalValue(Reg, GV);
4777 else if (auto CPA = dyn_cast<ConstantPtrAuth>(&C)) {
4778 Register Addr = getOrCreateVReg(*CPA->getPointer());
4779 Register AddrDisc = getOrCreateVReg(*CPA->getAddrDiscriminator());
4780 EntryBuilder->buildConstantPtrAuth(Reg, CPA, Addr, AddrDisc);
4781 } else if (auto CAZ = dyn_cast<ConstantAggregateZero>(&C)) {
4782 Constant &Elt = *CAZ->getElementValue(0u);
4783 if (isa<ScalableVectorType>(CAZ->getType())) {
4784 EntryBuilder->buildSplatVector(Reg, getOrCreateVReg(Elt));
4785 return true;
4786 }
4787 // Return the scalar if it is a <1 x Ty> vector.
4788 unsigned NumElts = CAZ->getElementCount().getFixedValue();
4789 if (NumElts == 1)
4790 return translateCopy(C, Elt, *EntryBuilder);
4791 // All elements are zero so we can just use the first one.
4792 EntryBuilder->buildSplatBuildVector(Reg, getOrCreateVReg(Elt));
4793 } else if (auto CV = dyn_cast<ConstantDataVector>(&C)) {
4794 // Return the scalar if it is a <1 x Ty> vector.
4795 if (CV->getNumElements() == 1)
4796 return translateCopy(C, *CV->getElementAsConstant(0), *EntryBuilder);
4798 for (unsigned i = 0; i < CV->getNumElements(); ++i) {
4799 Constant &Elt = *CV->getElementAsConstant(i);
4800 Ops.push_back(getOrCreateVReg(Elt));
4801 }
4802 EntryBuilder->buildBuildVector(Reg, Ops);
4803 } else if (auto CE = dyn_cast<ConstantExpr>(&C)) {
4804 switch(CE->getOpcode()) {
4805#define HANDLE_INST(NUM, OPCODE, CLASS) \
4806 case Instruction::OPCODE: \
4807 return translate##OPCODE(*CE, *EntryBuilder.get());
4808#include "llvm/IR/Instruction.def"
4809 default:
4810 return false;
4811 }
4812 } else if (auto CV = dyn_cast<ConstantVector>(&C)) {
4813 if (CV->getNumOperands() == 1)
4814 return translateCopy(C, *CV->getOperand(0), *EntryBuilder);
4816 for (unsigned i = 0; i < CV->getNumOperands(); ++i) {
4817 Ops.push_back(getOrCreateVReg(*CV->getOperand(i)));
4818 }
4819 EntryBuilder->buildBuildVector(Reg, Ops);
4820 } else if (auto *BA = dyn_cast<BlockAddress>(&C)) {
4821 EntryBuilder->buildBlockAddress(Reg, BA);
4822 } else
4823 return false;
4824
4825 return true;
4826}
4827
4828bool IRTranslatorImpl::mayTranslateUserTypes(const User &U) const {
4829 const TargetMachine &TM = TLI->getTargetMachine();
4830 if (LLT::getUseExtended())
4831 return true;
4832
4833 // BF16 cannot currently be represented by default LLT. To avoid miscompiles
4834 // we prevent any instructions using them by default in all targets that do
4835 // not explicitly enable it via LLT::setUseExtended(true).
4836 // SPIRV target is exception.
4837 return TM.getTargetTriple().isSPIRV() ||
4838 (!U.getType()->getScalarType()->isBFloatTy() &&
4839 !any_of(U.operands(), [](Value *V) {
4840 return V->getType()->getScalarType()->isBFloatTy();
4841 }));
4842}
4843
4844bool IRTranslatorImpl::finalizeBasicBlock(const BasicBlock &BB,
4846 for (auto &BTB : SL->BitTestCases) {
4847 // Emit header first, if it wasn't already emitted.
4848 if (!BTB.Emitted)
4849 emitBitTestHeader(BTB, BTB.Parent);
4850
4851 BranchProbability UnhandledProb = BTB.Prob;
4852 for (unsigned j = 0, ej = BTB.Cases.size(); j != ej; ++j) {
4853 UnhandledProb -= BTB.Cases[j].ExtraProb;
4854 // Set the current basic block to the mbb we wish to insert the code into
4855 MachineBasicBlock *MBB = BTB.Cases[j].ThisBB;
4856 // If all cases cover a contiguous range, it is not necessary to jump to
4857 // the default block after the last bit test fails. This is because the
4858 // range check during bit test header creation has guaranteed that every
4859 // case here doesn't go outside the range. In this case, there is no need
4860 // to perform the last bit test, as it will always be true. Instead, make
4861 // the second-to-last bit-test fall through to the target of the last bit
4862 // test, and delete the last bit test.
4863
4864 MachineBasicBlock *NextMBB;
4865 if ((BTB.ContiguousRange || BTB.FallthroughUnreachable) && j + 2 == ej) {
4866 // Second-to-last bit-test with contiguous range: fall through to the
4867 // target of the final bit test.
4868 NextMBB = BTB.Cases[j + 1].TargetBB;
4869 } else if (j + 1 == ej) {
4870 // For the last bit test, fall through to Default.
4871 NextMBB = BTB.Default;
4872 } else {
4873 // Otherwise, fall through to the next bit test.
4874 NextMBB = BTB.Cases[j + 1].ThisBB;
4875 }
4876
4877 emitBitTestCase(BTB, NextMBB, UnhandledProb, BTB.Reg, BTB.Cases[j], MBB);
4878
4879 if ((BTB.ContiguousRange || BTB.FallthroughUnreachable) && j + 2 == ej) {
4880 // We need to record the replacement phi edge here that normally
4881 // happens in emitBitTestCase before we delete the case, otherwise the
4882 // phi edge will be lost.
4883 addMachineCFGPred({BTB.Parent->getBasicBlock(),
4884 BTB.Cases[ej - 1].TargetBB->getBasicBlock()},
4885 MBB);
4886 // Since we're not going to use the final bit test, remove it.
4887 BTB.Cases.pop_back();
4888 break;
4889 }
4890 }
4891 // This is "default" BB. We have two jumps to it. From "header" BB and from
4892 // last "case" BB, unless the latter was skipped.
4893 CFGEdge HeaderToDefaultEdge = {BTB.Parent->getBasicBlock(),
4894 BTB.Default->getBasicBlock()};
4895 addMachineCFGPred(HeaderToDefaultEdge, BTB.Parent);
4896 if (!BTB.ContiguousRange) {
4897 addMachineCFGPred(HeaderToDefaultEdge, BTB.Cases.back().ThisBB);
4898 }
4899 }
4900 SL->BitTestCases.clear();
4901
4902 for (auto &JTCase : SL->JTCases) {
4903 // Emit header first, if it wasn't already emitted.
4904 if (!JTCase.first.Emitted)
4905 emitJumpTableHeader(JTCase.second, JTCase.first, JTCase.first.HeaderBB);
4906
4907 emitJumpTable(JTCase.second, JTCase.second.MBB);
4908 }
4909 SL->JTCases.clear();
4910
4911 for (auto &SwCase : SL->SwitchCases)
4912 emitSwitchCase(SwCase, &CurBuilder->getMBB(), *CurBuilder);
4913 SL->SwitchCases.clear();
4914
4915 // Check if we need to generate stack-protector guard checks.
4916 if (SPInfo->shouldEmitSDCheck(BB)) {
4917 bool FunctionBasedInstrumentation =
4918 TLI->getSSPStackGuardCheck(*MF->getFunction().getParent(), *Libcalls);
4919 SPDescriptor.initialize(&BB, &MBB, FunctionBasedInstrumentation);
4920 }
4921 // Handle stack protector.
4922 if (SPDescriptor.shouldEmitFunctionBasedCheckStackProtector()) {
4923 LLVM_DEBUG(dbgs() << "Unimplemented stack protector case\n");
4924 return false;
4925 } else if (SPDescriptor.shouldEmitStackProtector()) {
4926 MachineBasicBlock *ParentMBB = SPDescriptor.getParentMBB();
4927 MachineBasicBlock *SuccessMBB = SPDescriptor.getSuccessMBB();
4928
4929 // Find the split point to split the parent mbb. At the same time copy all
4930 // physical registers used in the tail of parent mbb into virtual registers
4931 // before the split point and back into physical registers after the split
4932 // point. This prevents us needing to deal with Live-ins and many other
4933 // register allocation issues caused by us splitting the parent mbb. The
4934 // register allocator will clean up said virtual copies later on.
4936 ParentMBB, *MF->getSubtarget().getInstrInfo());
4937
4938 // Splice the terminator of ParentMBB into SuccessMBB.
4939 SuccessMBB->splice(SuccessMBB->end(), ParentMBB, SplitPoint,
4940 ParentMBB->end());
4941
4942 // Add compare/jump on neq/jump to the parent BB.
4943 if (!emitSPDescriptorParent(SPDescriptor, ParentMBB))
4944 return false;
4945
4946 // CodeGen Failure MBB if we have not codegened it yet.
4947 MachineBasicBlock *FailureMBB = SPDescriptor.getFailureMBB();
4948 if (FailureMBB->empty()) {
4949 if (!emitSPDescriptorFailure(SPDescriptor, FailureMBB))
4950 return false;
4951 }
4952
4953 // Clear the Per-BB State.
4954 SPDescriptor.resetPerBBState();
4955 }
4956 return true;
4957}
4958
4959bool IRTranslatorImpl::emitSPDescriptorParent(StackProtectorDescriptor &SPD,
4960 MachineBasicBlock *ParentBB) {
4961 CurBuilder->setInsertPt(*ParentBB, ParentBB->end());
4962 // First create the loads to the guard/stack slot for the comparison.
4963 Type *PtrIRTy = PointerType::getUnqual(MF->getFunction().getContext());
4964 const LLT PtrTy = getLLTForType(*PtrIRTy, *DL);
4965 LLT PtrMemTy = getLLTForMVT(TLI->getPointerMemTy(*DL));
4966
4967 MachineFrameInfo &MFI = ParentBB->getParent()->getFrameInfo();
4968 int FI = MFI.getStackProtectorIndex();
4969
4970 Register Guard;
4971 Register StackSlotPtr = CurBuilder->buildFrameIndex(PtrTy, FI).getReg(0);
4972 const Module &M = *ParentBB->getParent()->getFunction().getParent();
4973 Align Align = DL->getPrefTypeAlign(PointerType::getUnqual(M.getContext()));
4974
4975 // Generate code to load the content of the guard slot.
4976 Register GuardVal =
4977 CurBuilder
4978 ->buildLoad(PtrMemTy, StackSlotPtr,
4979 MachinePointerInfo::getFixedStack(*MF, FI), Align,
4981 .getReg(0);
4982
4983 // Retrieve guard check function, nullptr if instrumentation is inlined.
4984 if (const Function *GuardCheckFn = TLI->getSSPStackGuardCheck(M, *Libcalls)) {
4985 // This path is currently untestable on GlobalISel, since the only platform
4986 // that needs this seems to be Windows, and we fall back on that currently.
4987 // The code still lives here in case that changes.
4988 // Silence warning about unused variable until the code below that uses
4989 // 'GuardCheckFn' is enabled.
4990 (void)GuardCheckFn;
4991 return false;
4992#if 0
4993 // The target provides a guard check function to validate the guard value.
4994 // Generate a call to that function with the content of the guard slot as
4995 // argument.
4996 FunctionType *FnTy = GuardCheckFn->getFunctionType();
4997 assert(FnTy->getNumParams() == 1 && "Invalid function signature");
4998 ISD::ArgFlagsTy Flags;
4999 if (GuardCheckFn->hasAttribute(1, Attribute::AttrKind::InReg))
5000 Flags.setInReg();
5001 CallLowering::ArgInfo GuardArgInfo(
5002 {GuardVal, FnTy->getParamType(0), {Flags}});
5003
5004 CallLowering::CallLoweringInfo Info;
5005 Info.OrigArgs.push_back(GuardArgInfo);
5006 Info.CallConv = GuardCheckFn->getCallingConv();
5007 Info.Callee = MachineOperand::CreateGA(GuardCheckFn, 0);
5008 Info.OrigRet = {Register(), FnTy->getReturnType()};
5009 if (!CLI->lowerCall(MIRBuilder, Info)) {
5010 LLVM_DEBUG(dbgs() << "Failed to lower call to stack protector check\n");
5011 return false;
5012 }
5013 return true;
5014#endif
5015 }
5016
5017 // If useLoadStackGuardNode returns true, generate LOAD_STACK_GUARD.
5018 // Otherwise, emit a volatile load to retrieve the stack guard value.
5019 if (TLI->useLoadStackGuardNode(*ParentBB->getBasicBlock()->getModule())) {
5020 Guard = MRI->createGenericVirtualRegister(PtrMemTy);
5021 getStackGuard(Guard, *CurBuilder);
5022 } else {
5023 // TODO: test using android subtarget when we support @llvm.thread.pointer.
5024 const Value *IRGuard = TLI->getSDagStackGuard(M, *Libcalls);
5025 Register GuardPtr = getOrCreateVReg(*IRGuard);
5026
5027 Guard = CurBuilder
5028 ->buildLoad(PtrMemTy, GuardPtr,
5029 MachinePointerInfo::getFixedStack(*MF, FI), Align,
5032 .getReg(0);
5033 }
5034
5035 // Perform the comparison.
5036 auto Cmp =
5037 CurBuilder->buildICmp(CmpInst::ICMP_NE, LLT::integer(1), Guard, GuardVal);
5038 // If the guard/stackslot do not equal, branch to failure MBB.
5039 CurBuilder->buildBrCond(Cmp, *SPD.getFailureMBB());
5040 // Otherwise branch to success MBB.
5041 CurBuilder->buildBr(*SPD.getSuccessMBB());
5042 return true;
5043}
5044
5045bool IRTranslatorImpl::emitSPDescriptorFailure(StackProtectorDescriptor &SPD,
5046 MachineBasicBlock *FailureBB) {
5047 const RTLIB::LibcallImpl LibcallImpl =
5048 Libcalls->getLibcallImpl(RTLIB::STACKPROTECTOR_CHECK_FAIL);
5049 if (LibcallImpl == RTLIB::Unsupported)
5050 return false;
5051
5052 CurBuilder->setInsertPt(*FailureBB, FailureBB->end());
5053
5054 CallLowering::CallLoweringInfo Info;
5055 Info.CallConv = Libcalls->getLibcallImplCallingConv(LibcallImpl);
5056
5057 StringRef LibcallName =
5059 Info.Callee = MachineOperand::CreateES(LibcallName.data());
5060 Info.OrigRet = {Register(), Type::getVoidTy(MF->getFunction().getContext()),
5061 0};
5062 if (!CLI->lowerCall(*CurBuilder, Info)) {
5063 LLVM_DEBUG(dbgs() << "Failed to lower call to stack protector fail\n");
5064 return false;
5065 }
5066
5067 // Emit a trap instruction if we are required to do so.
5068 const TargetOptions &TargetOpts = TLI->getTargetMachine().Options;
5069 if (TargetOpts.TrapUnreachable && !TargetOpts.NoTrapAfterNoreturn)
5070 CurBuilder->buildInstr(TargetOpcode::G_TRAP);
5071
5072 return true;
5073}
5074
5075void IRTranslatorImpl::finalizeFunction() {
5076 // Release the memory used by the different maps we
5077 // needed during the translation.
5078 PendingPHIs.clear();
5079 VMap.reset();
5080 FrameIndices.clear();
5081 MachinePreds.clear();
5082 // MachineIRBuilder::DebugLoc can outlive the DILocation it holds. Clear it
5083 // to avoid accessing free’d memory (in runOnMachineFunction) and to avoid
5084 // destroying it twice (in ~IRTranslator() and ~LLVMContext())
5085 EntryBuilder.reset();
5086 CurBuilder.reset();
5087 FuncInfo.clear();
5088 SPDescriptor.resetPerFunctionState();
5089}
5090
5091/// Returns true if a BasicBlock \p BB within a variadic function contains a
5092/// variadic musttail call.
5093static bool checkForMustTailInVarArgFn(bool IsVarArg, const BasicBlock &BB) {
5094 if (!IsVarArg)
5095 return false;
5096
5097 // Walk the block backwards, because tail calls usually only appear at the end
5098 // of a block.
5099 return llvm::any_of(llvm::reverse(BB), [](const Instruction &I) {
5100 const auto *CI = dyn_cast<CallInst>(&I);
5101 return CI && CI->isMustTailCall();
5102 });
5103}
5104
5106 MachineFunction &CurMF, function_ref<GISelCSEInfo *()> GetCSEInfo,
5107 bool ShouldSkipOpts, function_ref<AAResults *()> GetAAResults,
5109 function_ref<AssumptionCache *()> GetAC, TargetLibraryInfo *LibraryInfo,
5110 const LibcallLoweringInfo *LibcallInfo, SSPLayoutInfo *StackProtectorInfo) {
5111 MF = &CurMF;
5112 const Function &F = MF->getFunction();
5113 ORE = std::make_unique<OptimizationRemarkEmitter>(&F);
5114 CLI = MF->getSubtarget().getCallLowering();
5115 SPInfo = StackProtectorInfo;
5116
5117 if (CLI->fallBackToDAGISel(*MF)) {
5118 OptimizationRemarkMissed R("gisel-ir-translator", "GISelFailure",
5119 F.getSubprogram(), &F.getEntryBlock());
5120 R << "unable to lower function: "
5121 << ore::NV("Prototype", F.getFunctionType());
5122
5123 reportTranslationError(*MF, *ORE, R);
5124 return false;
5125 }
5126
5127 // Set the CSEConfig and run the analysis.
5128 GISelCSEInfo *CSEInfo = nullptr;
5129
5130 bool EnableCSE = EnableCSEInIRTranslator.getNumOccurrences()
5132 : true;
5133
5134 const TargetSubtargetInfo &Subtarget = MF->getSubtarget();
5135 TLI = Subtarget.getTargetLowering();
5136
5137 if (EnableCSE) {
5138 EntryBuilder = std::make_unique<CSEMIRBuilder>(CurMF);
5139 CSEInfo = GetCSEInfo();
5140 EntryBuilder->setCSEInfo(CSEInfo);
5141 CurBuilder = std::make_unique<CSEMIRBuilder>(CurMF);
5142 CurBuilder->setCSEInfo(CSEInfo);
5143 } else {
5144 EntryBuilder = std::make_unique<MachineIRBuilder>();
5145 CurBuilder = std::make_unique<MachineIRBuilder>();
5146 }
5147 CLI = Subtarget.getCallLowering();
5148 CurBuilder->setMF(*MF);
5149 EntryBuilder->setMF(*MF);
5150 MRI = &MF->getRegInfo();
5151 DL = &F.getDataLayout();
5152 const TargetMachine &TM = MF->getTarget();
5153 EnableOpts = OptLevel != CodeGenOptLevel::None && !ShouldSkipOpts;
5154 FuncInfo.MF = MF;
5155 // Prefer the "exception-model" module flag, else the TargetOptions default.
5156 FuncInfo.ExceptionModel = F.getParent()->getExceptionModel();
5157 if (FuncInfo.ExceptionModel == ExceptionHandling::Default)
5158 FuncInfo.ExceptionModel = TM.getExceptionModel();
5159 if (EnableOpts) {
5160 AA = GetAAResults();
5161 FuncInfo.BPI = GetBPI();
5162 AC = GetAC();
5163 } else {
5164 AA = nullptr;
5165 FuncInfo.BPI = nullptr;
5166 AC = nullptr;
5167 }
5168 LibInfo = LibraryInfo;
5169 Libcalls = LibcallInfo;
5170
5171 FuncInfo.CanLowerReturn = CLI->checkReturnTypeForCallConv(*MF);
5172
5173 SL = std::make_unique<GISelSwitchLowering>(this, FuncInfo);
5174 SL->init(*TLI, TM, *DL);
5175
5176 assert(PendingPHIs.empty() && "stale PHIs");
5177
5178 // Targets which want to use big endian can enable it using
5179 // enableBigEndian()
5180 if (!DL->isLittleEndian() && !CLI->enableBigEndian()) {
5181 // Currently we don't properly handle big endian code.
5182 OptimizationRemarkMissed R("gisel-ir-translator", "GISelFailure",
5183 F.getSubprogram(), &F.getEntryBlock());
5184 R << "unable to translate in big endian mode";
5185 reportTranslationError(*MF, *ORE, R);
5186 return false;
5187 }
5188
5189 // Release the per-function state when we return, whether we succeeded or not.
5190 llvm::scope_exit FinalizeOnReturn([this]() { finalizeFunction(); });
5191
5192 // Setup a separate basic-block for the arguments and constants
5193 MachineBasicBlock *EntryBB = MF->CreateMachineBasicBlock();
5194 MF->push_back(EntryBB);
5195 EntryBuilder->setMBB(*EntryBB);
5196
5197 DebugLoc DbgLoc = F.getEntryBlock().getFirstNonPHIIt()->getDebugLoc();
5198 SwiftError.setFunction(CurMF);
5199 SwiftError.createEntriesInEntryBlock(DbgLoc);
5200
5201 bool IsVarArg = F.isVarArg();
5202 bool HasMustTailInVarArgFn = false;
5203 // Use arguments and instructions to estimate the number of mapped values and
5204 // virtual registers.
5205 unsigned NumValues = F.arg_size();
5206
5207 // Create all blocks, in IR order, to preserve the layout.
5208 FuncInfo.MBBMap.resize(F.getMaxBlockNumber());
5209 for (const BasicBlock &BB: F) {
5210 NumValues += BB.size();
5211 auto *&MBB = FuncInfo.MBBMap[BB.getNumber()];
5212
5213 MBB = MF->CreateMachineBasicBlock(&BB);
5214 MF->push_back(MBB);
5215
5216 // Only mark the block if the BlockAddress actually has users. The
5217 // hasAddressTaken flag may be stale if the BlockAddress was optimized away
5218 // but the constant still exists in the uniquing table.
5219 if (BB.hasAddressTaken()) {
5220 if (BlockAddress *BA = BlockAddress::lookup(&BB))
5221 if (!BA->hasZeroLiveUses())
5222 MBB->setAddressTakenIRBlock(const_cast<BasicBlock *>(&BB));
5223 }
5224
5225 if (!HasMustTailInVarArgFn)
5226 HasMustTailInVarArgFn = checkForMustTailInVarArgFn(IsVarArg, BB);
5227 }
5228
5229 VMap.reserveVRegs(NumValues);
5230 MRI->reserveVirtRegs(NumValues);
5231
5232 MF->getFrameInfo().setHasMustTailInVarArgFunc(HasMustTailInVarArgFn);
5233
5234 // Make our arguments/constants entry block fallthrough to the IR entry block.
5235 EntryBB->addSuccessor(&getMBB(F.front()));
5236
5237 // Lower the actual args into this basic block.
5238 SmallVector<ArrayRef<Register>, 8> VRegArgs;
5239 for (const Argument &Arg: F.args()) {
5240 if (DL->getTypeStoreSize(Arg.getType()).isZero())
5241 continue; // Don't handle zero sized types.
5242 ArrayRef<Register> VRegs = getOrCreateVRegs(Arg);
5243 VRegArgs.push_back(VRegs);
5244
5245 if (CLI->supportSwiftError() && Arg.hasSwiftErrorAttr()) {
5246 assert(VRegs.size() == 1 && "Too many vregs for Swift error");
5247 SwiftError.setCurrentVReg(EntryBB, SwiftError.getFunctionArg(), VRegs[0]);
5248 }
5249 }
5250
5251 if (!CLI->lowerFormalArguments(*EntryBuilder, F, VRegArgs, FuncInfo)) {
5252 OptimizationRemarkMissed R("gisel-ir-translator", "GISelFailure",
5253 F.getSubprogram(), &F.getEntryBlock());
5254 R << "unable to lower arguments: "
5255 << ore::NV("Prototype", F.getFunctionType());
5256 reportTranslationError(*MF, *ORE, R);
5257 return false;
5258 }
5259
5260 // Need to visit defs before uses when translating instructions.
5261 GISelObserverWrapper WrapperObserver;
5262 if (EnableCSE && CSEInfo)
5263 WrapperObserver.addObserver(CSEInfo);
5264 {
5266#ifndef NDEBUG
5267 DILocationVerifier Verifier;
5268 WrapperObserver.addObserver(&Verifier);
5269#endif // ifndef NDEBUG
5270 RAIIMFObsDelInstaller ObsInstall(*MF, WrapperObserver);
5271 for (const BasicBlock *BB : RPOT) {
5272 MachineBasicBlock &MBB = getMBB(*BB);
5273 // Set the insertion point of all the following translations to
5274 // the end of this basic block.
5275 CurBuilder->setMBB(MBB);
5276 HasTailCall = false;
5277 for (const Instruction &Inst : *BB) {
5278 // If we translated a tail call in the last step, then we know
5279 // everything after the call is either a return, or something that is
5280 // handled by the call itself. (E.g. a lifetime marker or assume
5281 // intrinsic.) In this case, we should stop translating the block and
5282 // move on.
5283 if (HasTailCall)
5284 break;
5285#ifndef NDEBUG
5286 Verifier.setCurrentInst(&Inst);
5287#endif // ifndef NDEBUG
5288
5289 // Translate any debug-info attached to the instruction.
5290 translateDbgInfo(Inst, *CurBuilder);
5291
5292 if (translate(Inst))
5293 continue;
5294
5295 OptimizationRemarkMissed R("gisel-ir-translator", "GISelFailure",
5296 Inst.getDebugLoc(), BB);
5297 R << "unable to translate instruction: " << ore::NV("Opcode", &Inst);
5298
5299 if (ORE->allowExtraAnalysis("gisel-ir-translator")) {
5300 std::string InstStrStorage;
5301 raw_string_ostream InstStr(InstStrStorage);
5302 InstStr << Inst;
5303
5304 R << ": '" << InstStrStorage << "'";
5305 }
5306
5307 reportTranslationError(*MF, *ORE, R);
5308 return false;
5309 }
5310
5311 if (!finalizeBasicBlock(*BB, MBB)) {
5312 OptimizationRemarkMissed R("gisel-ir-translator", "GISelFailure",
5313 BB->getTerminator()->getDebugLoc(), BB);
5314 R << "unable to translate basic block";
5315 reportTranslationError(*MF, *ORE, R);
5316 return false;
5317 }
5318 }
5319#ifndef NDEBUG
5320 WrapperObserver.removeObserver(&Verifier);
5321#endif
5322 }
5323
5324 finishPendingPhis();
5325
5326 SwiftError.propagateVRegs();
5327
5328 // Merge the argument lowering and constants block with its single
5329 // successor, the LLVM-IR entry block. We want the basic block to
5330 // be maximal.
5331 assert(EntryBB->succ_size() == 1 &&
5332 "Custom BB used for lowering should have only one successor");
5333 // Get the successor of the current entry block.
5334 MachineBasicBlock &NewEntryBB = **EntryBB->succ_begin();
5335 assert(NewEntryBB.pred_size() == 1 &&
5336 "LLVM-IR entry block has a predecessor!?");
5337 // Move all the instruction from the current entry block to the
5338 // new entry block.
5339 NewEntryBB.splice(NewEntryBB.begin(), EntryBB, EntryBB->begin(),
5340 EntryBB->end());
5341
5342 // Update the live-in information for the new entry block.
5343 for (const MachineBasicBlock::RegisterMaskPair &LiveIn : EntryBB->liveins())
5344 NewEntryBB.addLiveIn(LiveIn);
5345 NewEntryBB.sortUniqueLiveIns();
5346
5347 // Get rid of the now empty basic block.
5348 EntryBB->removeSuccessor(&NewEntryBB);
5349 MF->remove(EntryBB);
5350 MF->deleteMachineBasicBlock(EntryBB);
5351
5352 assert(&MF->front() == &NewEntryBB &&
5353 "New entry wasn't next in the list of basic block!");
5354
5355 // Initialize stack protector information.
5356 SPInfo->copyToMachineFrameInfo(MF->getFrameInfo());
5357
5358 return false;
5359}
5360
5362 const TargetSubtargetInfo &Subtarget = MF.getSubtarget();
5363 Function &F = MF.getFunction();
5364
5365 bool ShouldSkipOpts = skipFunction(MF.getFunction());
5366 return Impl->runOnMachineFunction(
5367 MF,
5368 [&]() {
5372 return &Wrapper.get(TPC.getCSEConfig());
5373 },
5374 ShouldSkipOpts,
5375 [&]() { return &getAnalysis<AAResultsWrapperPass>().getAAResults(); },
5376 [&]() {
5378 },
5379 [&]() {
5380 return &getAnalysis<AssumptionCacheTracker>().getAssumptionCache(
5381 MF.getFunction());
5382 },
5384 &getAnalysis<LibcallLoweringInfoWrapper>().getLibcallLowering(
5385 *F.getParent(), Subtarget),
5386 &getAnalysis<StackProtector>().getLayoutInfo());
5387}
5388
5390 : Impl(std::make_unique<IRTranslatorImpl>(OptLevel)) {}
5391
5394
5397 const TargetSubtargetInfo &Subtarget = MF.getSubtarget();
5398 Function &F = MF.getFunction();
5399
5400 bool ShouldSkipOpts = MF.getFunction().hasOptNone() ||
5403 .getManager();
5404 auto &MAMProxy =
5406 const ModuleLibcallLoweringInfo *MLLI =
5407 MAMProxy.getCachedResult<LibcallLoweringModuleAnalysis>(*F.getParent());
5408 if (!MLLI)
5410 "LibcallLoweringModuleAnalysis must be available for IRTranslator");
5411 Impl->runOnMachineFunction(
5412 MF, [&]() { return MFAM.getResult<GISelCSEAnalysis>(MF).get(); },
5413 ShouldSkipOpts, [&]() { return &FAM.getResult<AAManager>(F); },
5414 [&]() { return &FAM.getResult<BranchProbabilityAnalysis>(F); },
5415 [&]() { return &FAM.getResult<AssumptionAnalysis>(F); },
5416 &FAM.getResult<TargetLibraryAnalysis>(F),
5417 &getLibcallLowering(*MLLI, Subtarget),
5418 &FAM.getResult<SSPLayoutAnalysis>(F));
5419
5421}
assert(UImm &&(UImm !=~static_cast< T >(0)) &&"Invalid immediate!")
amdgpu aa AMDGPU Address space based Alias Analysis Wrapper
unsigned uint64_t
MachineBasicBlock & MBB
MachineBasicBlock MachineBasicBlock::iterator DebugLoc DL
static GCRegistry::Add< ShadowStackGC > C("shadow-stack", "Very portable GC for uncooperative code generators")
static GCRegistry::Add< CoreCLRGC > E("coreclr", "CoreCLR-compatible GC")
static GCRegistry::Add< OcamlGC > B("ocaml", "ocaml 3.10-compatible GC")
Provides analysis for continuously CSEing during GISel passes.
This file implements a version of MachineIRBuilder which CSEs insts within a MachineBasicBlock.
This file describes how to lower LLVM calls to machine code calls.
This file contains the declarations for the subclasses of Constant, which represent the different fla...
dxil translate DXIL Translate Metadata
This contains common code to allow clients to notify changes to machine instr.
#define DEBUG_TYPE
const HexagonInstrInfo * TII
static cl::opt< bool > EnableCSEInIRTranslator("enable-cse-in-ir-translator", cl::desc("Should enable CSE in ir-translator"), cl::init(false))
static bool checkForMustTailInVarArgFn(bool IsVarArg, const BasicBlock &BB)
Returns true if a BasicBlock BB within a variadic function contains a variadic musttail call.
static unsigned getConvOpcode(Intrinsic::ID ID)
static uint64_t getOffsetFromIndices(const User &U, const DataLayout &DL)
static unsigned getConstrainedOpcode(Intrinsic::ID ID)
IRTranslator LLVM IR MI
IRTranslator LLVM IR static false void reportTranslationError(MachineFunction &MF, OptimizationRemarkEmitter &ORE, OptimizationRemarkMissed &R)
static bool isValInBlock(const Value *V, const BasicBlock *BB)
static bool isSwiftError(const Value *V)
This file declares the IRTranslator pass.
This file provides various utilities for inspecting and working with the control flow graph in LLVM I...
Module.h This file contains the declarations for the Module class.
This file describes how to lower LLVM inline asm to machine code INLINEASM.
const AbstractManglingParser< Derived, Alloc >::OperatorInfo AbstractManglingParser< Derived, Alloc >::Ops[]
static LVOptions Options
Definition LVOptions.cpp:25
Implement a low-level type suitable for MachineInstr level instruction selection.
Implement a low-level type suitable for MachineInstr level instruction selection.
#define F(x, y, z)
Definition MD5.cpp:54
#define I(x, y, z)
Definition MD5.cpp:57
This file declares the MachineIRBuilder class.
Register Reg
Register const TargetRegisterInfo * TRI
Promote Memory to Register
Definition Mem2Reg.cpp:110
This file contains the declarations for metadata subclasses.
Type::TypeID TypeID
uint64_t High
OptimizedStructLayoutField Field
FunctionAnalysisManager FAM
if(PassOpts->AAPipeline)
#define INITIALIZE_PASS_DEPENDENCY(depName)
Definition PassSupport.h:42
#define INITIALIZE_PASS_END(passName, arg, name, cfg, analysis)
Definition PassSupport.h:44
#define INITIALIZE_PASS_BEGIN(passName, arg, name, cfg, analysis)
Definition PassSupport.h:39
This file builds on the ADT/GraphTraits.h file to build a generic graph post order iterator.
const SmallVectorImpl< MachineOperand > MachineBasicBlock * TBB
const SmallVectorImpl< MachineOperand > & Cond
std::pair< BasicBlock *, BasicBlock * > Edge
This file contains some templates that are useful if you are working with the STL at all.
verify safepoint Safepoint IR Verifier
static bool contains(SmallPtrSetImpl< ConstantExpr * > &Cache, ConstantExpr *Expr, Constant *C)
Definition Value.cpp:484
This file defines the scope_exit class, which executes user-defined cleanup logic at scope exit.
This file defines the SmallVector class.
#define LLVM_DEBUG(...)
Definition Debug.h:119
This file describes how to lower LLVM code to machine code.
Target-Independent Code Generator Pass Configuration Options pass.
Value * RHS
Value * LHS
A manager for alias analyses.
A wrapper pass to provide the legacy pass manager access to a suitably prepared AAResults object.
LLVM_ABI APInt zextOrTrunc(unsigned width) const
Zero extend or truncate to width.
Definition APInt.cpp:1078
static APInt getLowBitsSet(unsigned numBits, unsigned loBitsSet)
Constructs an APInt value that has the bottom loBitsSet bits set.
Definition APInt.h:302
an instruction to allocate memory on the stack
bool isSwiftError() const
Return true if this alloca is used as a swifterror argument to a call.
LLVM_ABI bool isStaticAlloca() const
Return true if this alloca is in the entry block of the function and is a constant size.
Align getAlign() const
Return the alignment of the memory that is being allocated by the instruction.
LLVM_ABI TypeSize getAllocationBaseSize(const DataLayout &DL) const
Get the size of the allocated type.
PointerType * getType() const
Overload to return most specific pointer type.
LLVM_ABI std::optional< TypeSize > getAllocationSize(const DataLayout &DL) const
Get allocation size in bytes.
const Value * getArraySize() const
Get the number of elements allocated.
PassT::Result & getResult(IRUnitT &IR, ExtraArgTs... ExtraArgs)
Get the result of an analysis pass for a given IR unit.
Represent the analysis usage information of a pass.
AnalysisUsage & addRequired()
AnalysisUsage & addPreserved()
Add the specified Pass class to the set of analyses preserved by this pass.
This class represents an incoming formal argument to a Function.
Definition Argument.h:32
LLVM_ABI bool hasSwiftErrorAttr() const
Return true if this argument has the swifterror attribute.
Definition Function.cpp:150
Represent a constant reference to an array (0 or more elements consecutively in memory),...
Definition ArrayRef.h:40
iterator end() const
Definition ArrayRef.h:130
size_t size() const
Get the array size.
Definition ArrayRef.h:141
iterator begin() const
Definition ArrayRef.h:129
bool empty() const
Check if the array is empty.
Definition ArrayRef.h:136
A function analysis which provides an AssumptionCache.
An immutable pass that tracks lazily created AssumptionCache objects.
A cache of @llvm.assume calls within a function.
@ Add
*p = old + v
@ FAdd
*p = old + v
@ USubCond
Subtract only if no unsigned overflow.
@ FMinimum
*p = minimum(old, v) minimum matches the behavior of llvm.minimum.
@ Min
*p = old <signed v ? old : v
@ Sub
*p = old - v
@ And
*p = old & v
@ Xor
*p = old ^ v
@ USubSat
*p = usub.sat(old, v) usub.sat matches the behavior of llvm.usub.sat.
@ FMaximum
*p = maximum(old, v) maximum matches the behavior of llvm.maximum.
@ FSub
*p = old - v
@ UIncWrap
Increment one up to a maximum value.
@ Max
*p = old >signed v ? old : v
@ UMin
*p = old <unsigned v ? old : v
@ FMin
*p = minnum(old, v) minnum matches the behavior of llvm.minnum.
@ UMax
*p = old >unsigned v ? old : v
@ FMaximumNum
*p = maximumnum(old, v) maximumnum matches the behavior of llvm.maximumnum.
@ FMax
*p = maxnum(old, v) maxnum matches the behavior of llvm.maxnum.
@ UDecWrap
Decrement one until a minimum value or zero.
@ FMinimumNum
*p = minimumnum(old, v) minimumnum matches the behavior of llvm.minimumnum.
@ Nand
*p = ~(old & v)
LLVM Basic Block Representation.
Definition BasicBlock.h:62
unsigned getNumber() const
Definition BasicBlock.h:95
const Function * getParent() const
Return the enclosing method, or null if none.
Definition BasicBlock.h:213
bool hasAddressTaken() const
Returns true if there are any uses of this basic block other than direct branches,...
Definition BasicBlock.h:672
LLVM_ABI InstListType::const_iterator getFirstNonPHIIt() const
Returns an iterator to the first instruction in this block that is not a PHINode instruction.
InstListType::const_iterator const_iterator
Definition BasicBlock.h:171
LLVM_ABI InstListType::const_iterator getFirstNonPHIOrDbg(bool SkipPseudoOp=true) const
Returns a pointer to the first instruction in this block that is not a PHINode or a debug intrinsic,...
size_t size() const
Definition BasicBlock.h:467
LLVM_ABI const Module * getModule() const
Return the module owning the function this basic block belongs to, or nullptr if the function does no...
The address of a basic block.
Definition Constants.h:1088
static LLVM_ABI BlockAddress * lookup(const BasicBlock *BB)
Lookup an existing BlockAddress constant for the given BasicBlock.
Legacy analysis pass which computes BlockFrequencyInfo.
Analysis pass which computes BranchProbabilityInfo.
Legacy analysis pass which computes BranchProbabilityInfo.
Analysis providing branch probability information.
LLVM_ABI BranchProbability getEdgeProbability(const BasicBlock *Src, unsigned IndexInSuccessors) const
Get an edge's probability, relative to other out-edges of the Src.
static constexpr BranchProbability getOne()
static constexpr BranchProbability getUnknown()
static constexpr BranchProbability getZero()
static void normalizeProbabilities(ProbabilityIter Begin, ProbabilityIter End)
Base class for all callable instructions (InvokeInst and CallInst) Holds everything related to callin...
bool isInlineAsm() const
Check if this call is an inline asm statement.
bool cannotMerge() const
Determine if the call cannot be tail merged.
std::optional< OperandBundleUse > getOperandBundle(StringRef Name) const
Return an operand bundle by name, if present.
Function * getCalledFunction() const
Returns the function called, or null if this is an indirect function invocation or the function signa...
LLVM_ABI bool paramHasAttr(unsigned ArgNo, Attribute::AttrKind Kind) const
Determine whether the argument or parameter has the given attribute.
User::op_iterator arg_begin()
Return the iterator pointing to the beginning of the argument list.
unsigned countOperandBundlesOfType(StringRef Name) const
Return the number of operand bundles with the tag Name attached to this instruction.
MaybeAlign getParamAlign(unsigned ArgNo) const
Extract the alignment for a call or parameter (0=unknown).
Value * getCalledOperand() const
Value * getArgOperand(unsigned i) const
User::op_iterator arg_end()
Return the iterator pointing to the end of the argument list.
bool isConvergent() const
Determine if the invoke is convergent.
LLVM_ABI Intrinsic::ID getIntrinsicID() const
Returns the intrinsic ID of the intrinsic called or Intrinsic::not_intrinsic if the called function i...
iterator_range< User::op_iterator > args()
Iteration adapter for range-for loops.
unsigned arg_size() const
AttributeList getAttributes() const
Return the attributes for this call.
This class represents a function call, abstracting a target machine's calling convention.
bool isTailCall() const
Predicate
This enumeration lists the possible predicates for CmpInst subclasses.
Definition InstrTypes.h:740
@ FCMP_TRUE
1 1 1 1 Always true (always folded)
Definition InstrTypes.h:757
@ ICMP_SLT
signed less than
Definition InstrTypes.h:769
@ ICMP_SLE
signed less or equal
Definition InstrTypes.h:770
@ ICMP_UGT
unsigned greater than
Definition InstrTypes.h:763
@ ICMP_NE
not equal
Definition InstrTypes.h:762
@ ICMP_ULE
unsigned less or equal
Definition InstrTypes.h:766
@ FCMP_FALSE
0 0 0 0 Always false (always folded)
Definition InstrTypes.h:742
bool isFPPredicate() const
Definition InstrTypes.h:845
bool isIntPredicate() const
Definition InstrTypes.h:846
Value * getCondition() const
BasicBlock * getSuccessor(unsigned i) const
static LLVM_ABI ConstantInt * getTrue(LLVMContext &Context)
bool isZero() const
This is just a convenience method to make client code smaller for a common code.
Definition Constants.h:219
unsigned getBitWidth() const
getBitWidth - Return the scalar bitwidth of this constant.
Definition Constants.h:162
uint64_t getZExtValue() const
Return the constant as a 64-bit unsigned integer value after it has been zero extended as appropriate...
Definition Constants.h:168
const APInt & getValue() const
Return the constant as an APInt value reference.
Definition Constants.h:159
This is an important base class in LLVM.
Definition Constant.h:43
static LLVM_ABI Constant * getAllOnesValue(Type *Ty)
static LLVM_ABI Constant * getNullValue(Type *Ty)
Constructor to create a '0' constant of arbitrary type.
This is the common base class for constrained floating point intrinsics.
LLVM_ABI std::optional< fp::ExceptionBehavior > getExceptionBehavior() const
LLVM_ABI unsigned getNonMetadataArgCount() const
DWARF expression.
LLVM_ABI bool isEntryValue() const
Check if the expression consists of exactly one entry value operand.
static LLVM_ABI DIExpression * append(const DIExpression *Expr, ArrayRef< uint64_t > Ops)
Append the opcodes Ops to DIExpr.
LLVM_ABI bool startsWithDeref() const
Return whether the first element a DW_OP_deref.
ArrayRef< uint64_t > getElements() const
A parsed version of the target data layout string in and methods for querying it.
Definition DataLayout.h:64
DebugLoc getDebugLoc() const
LLVM_ABI Value * getVariableLocationOp(unsigned OpIdx) const
DIExpression * getExpression() const
DILocalVariable * getVariable() const
A debug info location.
Definition DebugLoc.h:126
DenseMapIterator< KeyT, ValueT, KeyInfoT, BucketT, true > const_iterator
Definition DenseMap.h:680
Class representing an expression and its matching format.
This instruction extracts a struct member or array element value from an aggregate value.
static LLVM_ABI FixedVectorType * get(Type *ElementType, unsigned NumElts)
Definition Type.cpp:843
FunctionLoweringInfo - This contains information that is global to a function that is used when lower...
bool skipFunction(const Function &F) const
Optional passes call this function to check whether the pass should be skipped.
Definition Pass.cpp:196
bool hasMinSize() const
Optimize this function for minimum size (-Oz).
Definition Function.h:696
Constant * getPersonalityFn() const
Get the personality function associated with this function.
const Function & getFunction() const
Definition Function.h:167
bool isIntrinsic() const
isIntrinsic - Returns true if the function's name starts with "llvm.".
Definition Function.h:252
bool hasOptNone() const
Do not optimize this function (-O0).
Definition Function.h:686
The actual analysis pass wrapper.
Definition CSEInfo.h:244
Simple wrapper that does the following.
Definition CSEInfo.h:214
The CSE Analysis object.
Definition CSEInfo.h:72
Abstract class that contains various methods for clients to notify about changes.
Simple wrapper observer that takes several observers, and calls each one for each event.
void removeObserver(GISelChangeObserver *O)
void addObserver(GISelChangeObserver *O)
static StringRef dropLLVMManglingEscape(StringRef Name)
If the given string begins with the GlobalValue name mangling escape character '\1',...
bool hasExternalWeakLinkage() const
bool hasDLLImportStorageClass() const
Module * getParent()
Get the module that this global value is contained inside of...
bool isTailCall(const MachineInstr &MI) const override
IRTranslatorImpl(CodeGenOptLevel OptLevel=CodeGenOptLevel::None)
bool runOnMachineFunction(MachineFunction &MF, function_ref< GISelCSEInfo *()> GetCSEInfo, bool ShouldSkipOpts, function_ref< AAResults *()> GetAAResults, function_ref< BranchProbabilityInfo *()> GetBPI, function_ref< AssumptionCache *()> GetAC, TargetLibraryInfo *LibraryInfo, const LibcallLoweringInfo *LibcallInfo, SSPLayoutInfo *StackProtectorInfo)
IRTranslatorLegacy(CodeGenOptLevel OptLevel=CodeGenOptLevel::None)
bool runOnMachineFunction(MachineFunction &MF) override
runOnMachineFunction - This method must be overloaded to perform the desired machine code transformat...
void getAnalysisUsage(AnalysisUsage &AU) const override
getAnalysisUsage - This function should be overriden by passes that need analysis information to do t...
~IRTranslatorLegacy() override
LLVM_ABI PreservedAnalyses run(MachineFunction &MF, MachineFunctionAnalysisManager &MFAM)
LLVM_ABI ~IRTranslatorPass()
LLVM_ABI IRTranslatorPass(CodeGenOptLevel OptLevel)
bool lowerInlineAsm(MachineIRBuilder &MIRBuilder, const CallBase &CB, std::function< ArrayRef< Register >(const Value &Val)> GetOrCreateVRegs) const
Lower the given inline asm call instruction GetOrCreateVRegs is a callback to materialize a register ...
This instruction inserts a struct field of array element value into an aggregate value.
iterator_range< simple_ilist< DbgRecord >::iterator > getDbgRecordRange() const
Return a range over the DbgRecords attached to this instruction.
const DebugLoc & getDebugLoc() const
Return the debug location for this node as a DebugLoc.
LLVM_ABI const Module * getModule() const
Return the module owning the function this instruction belongs to or nullptr it the function does not...
bool hasMetadata() const
Return true if this instruction has any metadata attached to it.
MDNode * getMetadata(unsigned KindID) const
Get the metadata of given kind attached to this Instruction.
LLVM_ABI AAMDNodes getAAMetadata() const
Returns the AA metadata for this instruction.
unsigned getOpcode() const
Returns a member of one of the enums like Instruction::Add.
LLVM_ABI bool hasAllowReassoc() const LLVM_READONLY
Determine whether the allow-reassociation flag is set.
Intrinsic::ID getIntrinsicID() const
Return the intrinsic ID of this intrinsic.
static bool getUseExtended()
constexpr bool isScalar() const
constexpr LLT changeElementType(LLT NewEltTy) const
If this type is a vector, return a vector with the same number of elements but the new element type.
static constexpr LLT scalar(unsigned SizeInBits)
Get a low-level scalar or aggregate "bag of bits".
constexpr uint16_t getNumElements() const
Returns the number of elements in a vector LLT.
constexpr bool isFloat() const
constexpr bool isVector() const
static constexpr LLT pointer(unsigned AddressSpace, unsigned SizeInBits)
Get a low-level pointer in the given address space.
constexpr TypeSize getSizeInBits() const
Returns the total size of the type. Must only be called on sized types.
constexpr bool isPointer() const
static constexpr LLT fixed_vector(unsigned NumElements, unsigned ScalarSizeInBits)
Get a low-level fixed-width vector of some number of elements and element width.
constexpr bool isFixedVector() const
Returns true if the LLT is a fixed vector.
static constexpr LLT token()
Get a low-level token; just a scalar with zero bits (or no size).
static LLT integer(unsigned SizeInBits)
LLT changeElementSize(unsigned NewEltSize) const
If this type is a vector, return a vector with the same number of elements but the new element size.
LLVM_ABI void diagnose(const DiagnosticInfo &DI)
Report a message to the currently installed diagnostic handler.
Tracks which library functions to use for a particular subtarget or function.
Value * getPointerOperand()
AtomicOrdering getOrdering() const
Returns the ordering constraint of this load instruction.
SyncScope::ID getSyncScopeID() const
Returns the synchronization scope ID of this load instruction.
static LocationSize precise(uint64_t Value)
static MDTuple * get(LLVMContext &Context, ArrayRef< Metadata * > MDs)
Definition Metadata.h:1579
void normalizeSuccProbs()
Normalize probabilities of all successors so that the sum of them becomes one.
LLVM_ABI instr_iterator insert(instr_iterator I, MachineInstr *M)
Insert MI into the instruction list before I, possibly inside a bundle.
void push_back(MachineInstr *MI)
const BasicBlock * getBasicBlock() const
Return the LLVM basic block that this instance corresponded to originally.
LLVM_ABI void setSuccProbability(succ_iterator I, BranchProbability Prob)
Set successor probability of a given iterator.
LLVM_ABI void addSuccessor(MachineBasicBlock *Succ, BranchProbability Prob=BranchProbability::getUnknown())
Add Succ as a successor of this MachineBasicBlock.
SmallVectorImpl< MachineBasicBlock * >::iterator succ_iterator
LLVM_ABI void sortUniqueLiveIns()
Sorts and uniques the LiveIns vector.
LLVM_ABI bool isPredecessor(const MachineBasicBlock *MBB) const
Return true if the specified MBB is a predecessor of this block.
LLVM_ABI bool isLayoutSuccessor(const MachineBasicBlock *MBB) const
Return true if the specified MBB will be emitted immediately after this block, such that if this bloc...
void addLiveIn(MCRegister PhysReg, LaneBitmask LaneMask=LaneBitmask::getAll())
Adds the specified register as a live in.
const MachineFunction * getParent() const
Return the MachineFunction containing this basic block.
void splice(iterator Where, MachineBasicBlock *Other, iterator From)
Take an instruction from MBB 'Other' at the position From, and insert it into this MBB right before '...
MachineInstrBundleIterator< MachineInstr > iterator
void setIsEHPad(bool V=true)
Indicates the block is a landing pad.
int getStackProtectorIndex() const
Return the index for the stack protector object.
void getAnalysisUsage(AnalysisUsage &AU) const override
getAnalysisUsage - Subclasses that override getAnalysisUsage must call this.
const TargetSubtargetInfo & getSubtarget() const
getSubtarget - Return the subtarget for which this machine code is being compiled.
MachineFrameInfo & getFrameInfo()
getFrameInfo - Return the frame info object for the current function.
Function & getFunction()
Return the LLVM function that this machine code represents.
BasicBlockListType::iterator iterator
MachineBasicBlock * CreateMachineBasicBlock(const BasicBlock *BB=nullptr, std::optional< UniqueBBID > BBID=std::nullopt)
CreateMachineInstr - Allocate a new MachineInstr.
void insert(iterator MBBI, MachineBasicBlock *MBB)
Helper class to build MachineInstr.
MachineInstrBuilder buildFPTOUI_SAT(const DstOp &Dst, const SrcOp &Src0)
Build and insert Res = G_FPTOUI_SAT Src0.
MachineInstrBuilder buildFMul(const DstOp &Dst, const SrcOp &Src0, const SrcOp &Src1, std::optional< unsigned > Flags=std::nullopt)
MachineInstrBuilder buildFreeze(const DstOp &Dst, const SrcOp &Src)
Build and insert Dst = G_FREEZE Src.
MachineInstrBuilder buildBr(MachineBasicBlock &Dest)
Build and insert G_BR Dest.
MachineInstrBuilder buildModf(const DstOp &Fract, const DstOp &Int, const SrcOp &Src, std::optional< unsigned > Flags=std::nullopt)
Build and insert Fract, Int = G_FMODF Src.
LLVMContext & getContext() const
MachineInstrBuilder buildAdd(const DstOp &Dst, const SrcOp &Src0, const SrcOp &Src1, std::optional< unsigned > Flags=std::nullopt)
Build and insert Res = G_ADD Op0, Op1.
MachineInstrBuilder buildUndef(const DstOp &Res)
Build and insert Res = IMPLICIT_DEF.
MachineInstrBuilder buildResetFPMode()
Build and insert G_RESET_FPMODE.
MachineInstrBuilder buildNot(const DstOp &Dst, const SrcOp &Src0)
Build and insert a bitwise not, NegOne = G_CONSTANT -1 Res = G_OR Op0, NegOne.
MachineInstrBuilder buildFPTOSI_SAT(const DstOp &Dst, const SrcOp &Src0)
Build and insert Res = G_FPTOSI_SAT Src0.
MachineInstrBuilder buildUCmp(const DstOp &Res, const SrcOp &Op0, const SrcOp &Op1)
Build and insert a Res = G_UCMP Op0, Op1.
MachineInstrBuilder buildJumpTable(const LLT PtrTy, unsigned JTI)
Build and insert Res = G_JUMP_TABLE JTI.
MachineInstrBuilder buildGetRounding(const DstOp &Dst)
Build and insert Dst = G_GET_ROUNDING.
MachineInstrBuilder buildSCmp(const DstOp &Res, const SrcOp &Op0, const SrcOp &Op1)
Build and insert a Res = G_SCMP Op0, Op1.
MachineInstrBuilder buildFence(unsigned Ordering, unsigned Scope)
Build and insert G_FENCE Ordering, Scope.
MachineInstrBuilder buildSelect(const DstOp &Res, const SrcOp &Tst, const SrcOp &Op0, const SrcOp &Op1, std::optional< unsigned > Flags=std::nullopt)
Build and insert a Res = G_SELECT Tst, Op0, Op1.
MachineInstrBuilder buildFMA(const DstOp &Dst, const SrcOp &Src0, const SrcOp &Src1, const SrcOp &Src2, std::optional< unsigned > Flags=std::nullopt)
Build and insert Res = G_FMA Op0, Op1, Op2.
MachineInstrBuilder buildMul(const DstOp &Dst, const SrcOp &Src0, const SrcOp &Src1, std::optional< unsigned > Flags=std::nullopt)
Build and insert Res = G_MUL Op0, Op1.
MachineInstrBuilder buildInsertSubvector(const DstOp &Res, const SrcOp &Src0, const SrcOp &Src1, unsigned Index)
Build and insert Res = G_INSERT_SUBVECTOR Src0, Src1, Idx.
MachineInstrBuilder buildAnd(const DstOp &Dst, const SrcOp &Src0, const SrcOp &Src1)
Build and insert Res = G_AND Op0, Op1.
MachineInstrBuilder buildCast(const DstOp &Dst, const SrcOp &Src)
Build and insert an appropriate cast between two registers of equal size.
MachineInstrBuilder buildICmp(CmpInst::Predicate Pred, const DstOp &Res, const SrcOp &Op0, const SrcOp &Op1, std::optional< unsigned > Flags=std::nullopt)
Build and insert a Res = G_ICMP Pred, Op0, Op1.
MachineInstrBuilder buildLShr(const DstOp &Dst, const SrcOp &Src0, const SrcOp &Src1, std::optional< unsigned > Flags=std::nullopt)
MachineBasicBlock::iterator getInsertPt()
Current insertion point for new instructions.
MachineInstrBuilder buildSExtOrTrunc(const DstOp &Res, const SrcOp &Op)
Build and insert Res = G_SEXT Op, Res = G_TRUNC Op, or Res = COPY Op depending on the differing sizes...
MachineInstrBuilder buildAtomicRMW(unsigned Opcode, const DstOp &OldValRes, const SrcOp &Addr, const SrcOp &Val, MachineMemOperand &MMO)
Build and insert OldValRes<def> = G_ATOMICRMW_<Opcode> Addr, Val, MMO.
MachineInstrBuilder buildSub(const DstOp &Dst, const SrcOp &Src0, const SrcOp &Src1, std::optional< unsigned > Flags=std::nullopt)
Build and insert Res = G_SUB Op0, Op1.
MachineInstrBuilder buildIntrinsic(Intrinsic::ID ID, ArrayRef< Register > Res, bool HasSideEffects, bool isConvergent)
Build and insert a G_INTRINSIC instruction.
MachineInstrBuilder buildVScale(const DstOp &Res, unsigned MinElts)
Build and insert Res = G_VSCALE MinElts.
MachineInstrBuilder buildSplatBuildVector(const DstOp &Res, const SrcOp &Src)
Build and insert Res = G_BUILD_VECTOR with Src replicated to fill the number of elements.
MachineInstrBuilder buildSetFPMode(const SrcOp &Src)
Build and insert G_SET_FPMODE Src.
MachineInstrBuilder buildIntToPtr(const DstOp &Dst, const SrcOp &Src)
Build and insert a G_INTTOPTR instruction.
MachineInstrBuilder buildIndirectDbgValue(Register Reg, const MDNode *Variable, const MDNode *Expr)
Build and insert a DBG_VALUE instruction expressing the fact that the associated Variable lives in me...
MachineInstrBuilder buildBuildVector(const DstOp &Res, ArrayRef< Register > Ops)
Build and insert Res = G_BUILD_VECTOR Op0, ...
MachineInstrBuilder buildConstDbgValue(const Constant &C, const MDNode *Variable, const MDNode *Expr)
Build and insert a DBG_VALUE instructions specifying that Variable is given by C (suitably modified b...
MachineInstrBuilder buildBrCond(const SrcOp &Tst, MachineBasicBlock &Dest)
Build and insert G_BRCOND Tst, Dest.
std::optional< MachineInstrBuilder > materializeObjectPtrOffset(Register &Res, Register Op0, const LLT ValueTy, uint64_t Value)
Materialize and insert an instruction with appropriate flags for addressing some offset of an object,...
MachineInstrBuilder buildSetRounding(const SrcOp &Src)
Build and insert G_SET_ROUNDING.
MachineInstrBuilder buildExtractVectorElement(const DstOp &Res, const SrcOp &Val, const SrcOp &Idx)
Build and insert Res = G_EXTRACT_VECTOR_ELT Val, Idx.
MachineInstrBuilder buildLoad(const DstOp &Res, const SrcOp &Addr, MachineMemOperand &MMO)
Build and insert Res = G_LOAD Addr, MMO.
MachineInstrBuilder buildPtrAdd(const DstOp &Res, const SrcOp &Op0, const SrcOp &Op1, std::optional< unsigned > Flags=std::nullopt)
Build and insert Res = G_PTR_ADD Op0, Op1.
MachineInstrBuilder buildZExtOrTrunc(const DstOp &Res, const SrcOp &Op)
Build and insert Res = G_ZEXT Op, Res = G_TRUNC Op, or Res = COPY Op depending on the differing sizes...
MachineInstrBuilder buildExtractVectorElementConstant(const DstOp &Res, const SrcOp &Val, const int Idx)
Build and insert Res = G_EXTRACT_VECTOR_ELT Val, Idx.
MachineInstrBuilder buildShl(const DstOp &Dst, const SrcOp &Src0, const SrcOp &Src1, std::optional< unsigned > Flags=std::nullopt)
MachineInstrBuilder buildStore(const SrcOp &Val, const SrcOp &Addr, MachineMemOperand &MMO)
Build and insert G_STORE Val, Addr, MMO.
MachineInstrBuilder buildInstr(unsigned Opcode)
Build and insert <empty> = Opcode <empty>.
MachineInstrBuilder buildFrameIndex(const DstOp &Res, int Idx)
Build and insert Res = G_FRAME_INDEX Idx.
MachineInstrBuilder buildDirectDbgValue(Register Reg, const MDNode *Variable, const MDNode *Expr)
Build and insert a DBG_VALUE instruction expressing the fact that the associated Variable lives in Re...
MachineInstrBuilder buildDbgLabel(const MDNode *Label)
Build and insert a DBG_LABEL instructions specifying that Label is given.
MachineInstrBuilder buildBrJT(Register TablePtr, unsigned JTI, Register IndexReg)
Build and insert G_BRJT TablePtr, JTI, IndexReg.
MachineInstrBuilder buildDynStackAlloc(const DstOp &Res, const SrcOp &Size, Align Alignment)
Build and insert Res = G_DYN_STACKALLOC Size, Align.
MachineInstrBuilder buildFIDbgValue(int FI, const MDNode *Variable, const MDNode *Expr)
Build and insert a DBG_VALUE instruction expressing the fact that the associated Variable lives in th...
MachineInstrBuilder buildResetFPEnv()
Build and insert G_RESET_FPENV.
MachineInstrBuilder buildTrunc(const DstOp &Res, const SrcOp &Op, std::optional< unsigned > Flags=std::nullopt)
Build and insert Res = G_TRUNC Op.
void setDebugLoc(const DebugLoc &DL)
Set the debug location to DL for all the next build instructions.
const MachineBasicBlock & getMBB() const
Getter for the basic block we currently build.
MachineInstrBuilder buildInsertVectorElement(const DstOp &Res, const SrcOp &Val, const SrcOp &Elt, const SrcOp &Idx)
Build and insert Res = G_INSERT_VECTOR_ELT Val, Elt, Idx.
MachineInstrBuilder buildAtomicCmpXchgWithSuccess(const DstOp &OldValRes, const DstOp &SuccessRes, const SrcOp &Addr, const SrcOp &CmpVal, const SrcOp &NewVal, MachineMemOperand &MMO)
Build and insert OldValRes<def>, SuccessRes<def> = / G_ATOMIC_CMPXCHG_WITH_SUCCESS Addr,...
void setMBB(MachineBasicBlock &MBB)
Set the insertion point to the end of MBB.
MachineInstrBuilder buildBitcast(const DstOp &Dst, const SrcOp &Src)
Build and insert Dst = G_BITCAST Src.
const DebugLoc & getDebugLoc()
Get the current instruction's debug location.
MachineInstrBuilder buildTrap(bool Debug=false)
Build and insert G_TRAP or G_DEBUGTRAP.
MachineInstrBuilder buildFFrexp(const DstOp &Fract, const DstOp &Exp, const SrcOp &Src, std::optional< unsigned > Flags=std::nullopt)
Build and insert Fract, Exp = G_FFREXP Src.
MachineRegisterInfo * getMRI()
Getter for MRI.
MachineInstrBuilder buildOr(const DstOp &Dst, const SrcOp &Src0, const SrcOp &Src1, std::optional< unsigned > Flags=std::nullopt)
Build and insert Res = G_OR Op0, Op1.
MachineInstrBuilder buildFSincos(const DstOp &Sin, const DstOp &Cos, const SrcOp &Src, std::optional< unsigned > Flags=std::nullopt)
Build and insert Sin, Cos = G_FSINCOS Src.
MachineInstrBuilder buildShuffleVector(const DstOp &Res, const SrcOp &Src1, const SrcOp &Src2, ArrayRef< int > Mask)
Build and insert Res = G_SHUFFLE_VECTOR Src1, Src2, Mask.
MachineInstrBuilder buildInstrNoInsert(unsigned Opcode)
Build but don't insert <empty> = Opcode <empty>.
MachineInstrBuilder buildCopy(const DstOp &Res, const SrcOp &Op)
Build and insert Res = COPY Op.
MachineInstrBuilder buildPrefetch(const SrcOp &Addr, unsigned RW, unsigned Locality, unsigned CacheType, MachineMemOperand &MMO)
Build and insert G_PREFETCH Addr, RW, Locality, CacheType.
MachineInstrBuilder buildExtractSubvector(const DstOp &Res, const SrcOp &Src, unsigned Index)
Build and insert Res = G_EXTRACT_SUBVECTOR Src, Idx0.
const DataLayout & getDataLayout() const
MachineInstrBuilder buildBrIndirect(Register Tgt)
Build and insert G_BRINDIRECT Tgt.
MachineInstrBuilder buildSplatVector(const DstOp &Res, const SrcOp &Val)
Build and insert Res = G_SPLAT_VECTOR Val.
MachineInstrBuilder buildStepVector(const DstOp &Res, unsigned Step)
Build and insert Res = G_STEP_VECTOR Step.
virtual MachineInstrBuilder buildConstant(const DstOp &Res, const ConstantInt &Val)
Build and insert Res = G_CONSTANT Val.
MachineInstrBuilder buildFCmp(CmpInst::Predicate Pred, const DstOp &Res, const SrcOp &Op0, const SrcOp &Op1, std::optional< unsigned > Flags=std::nullopt)
Build and insert a Res = G_FCMP PredOp0, Op1.
MachineInstrBuilder buildFAdd(const DstOp &Dst, const SrcOp &Src0, const SrcOp &Src1, std::optional< unsigned > Flags=std::nullopt)
Build and insert Res = G_FADD Op0, Op1.
MachineInstrBuilder buildSetFPEnv(const SrcOp &Src)
Build and insert G_SET_FPENV Src.
MachineInstrBuilder buildPtrToInt(const DstOp &Dst, const SrcOp &Src)
Build and insert a G_PTRTOINT instruction.
Register getReg(unsigned Idx) const
Get the register for the operand index.
const MachineInstrBuilder & setMemRefs(ArrayRef< MachineMemOperand * > MMOs) const
const MachineInstrBuilder & addExternalSymbol(const char *FnName, unsigned TargetFlags=0) const
const MachineInstrBuilder & addUse(Register RegNo, RegState Flags={}, unsigned SubReg=0) const
Add a virtual register use operand.
const MachineInstrBuilder & setMIFlag(MachineInstr::MIFlag Flag) const
const MachineInstrBuilder & addImm(int64_t Val) const
Add a new immediate operand.
const MachineInstrBuilder & addMetadata(const MDNode *MD) const
const MachineInstrBuilder & addSym(MCSymbol *Sym, unsigned char TargetFlags=0) const
const MachineInstrBuilder & addFrameIndex(int Idx) const
const MachineInstrBuilder & addFPImm(const ConstantFP *Val) const
const MachineInstrBuilder & addMBB(MachineBasicBlock *MBB, unsigned TargetFlags=0) const
const MachineInstrBuilder & addDef(Register RegNo, RegState Flags={}, unsigned SubReg=0) const
Add a virtual register definition operand.
const MachineInstrBuilder & addMemOperand(MachineMemOperand *MMO) const
MachineInstr * getInstr() const
If conversion operators fail, use this method to get the MachineInstr explicitly.
LLVM_ABI void copyIRFlags(const Instruction &I)
Copy all flags to MachineInst MIFlags.
static LLVM_ABI uint32_t copyFlagsFromInstruction(const Instruction &I)
LLVM_ABI void setDeactivationSymbol(MachineFunction &MF, Value *DS)
void setDebugLoc(DebugLoc DL)
Replace current source information with new such.
Flags
Flags values. These may be or'd together.
@ MOVolatile
The memory access is volatile.
@ MODereferenceable
The memory access is dereferenceable (i.e., doesn't trap).
@ MOLoad
The memory access reads data.
@ MONonTemporal
The memory access is non-temporal.
@ MOInvariant
The memory access always returns the same value (or traps).
@ MOStore
The memory access writes data.
static MachineOperand CreateES(const char *SymName, unsigned TargetFlags=0)
static MachineOperand CreateGA(const GlobalValue *GV, int64_t Offset, unsigned TargetFlags=0)
MachineRegisterInfo - Keep track of information for virtual and physical registers,...
LLT getType(Register Reg) const
Get the low-level type of Reg or LLT{} if Reg is not a generic (target independent) virtual register.
Records a mapping from an opaque lowering context to its LibcallLoweringInfo.
The optimization diagnostic interface.
Diagnostic information for missed-optimization remarks.
BasicBlock * getIncomingBlock(unsigned i) const
Return incoming basic block number i.
Value * getIncomingValue(unsigned i) const
Return incoming value number x.
unsigned getNumIncomingValues() const
Return the number of incoming edges.
AnalysisType & getAnalysis() const
getAnalysis<AnalysisType>() - This function is used by subclasses to get to the analysis information ...
static PointerType * getUnqual(LLVMContext &C)
This constructs an opaque pointer to an object in the default address space (address space zero).
A set of analyses that are preserved following a run of a transformation pass.
Definition Analysis.h:112
Class to install both of the above.
Wrapper class representing virtual and physical registers.
Definition Register.h:20
Value * getReturnValue() const
Convenience accessor. Returns null if there is no return value.
size_type count(ConstPtrType Ptr) const
count - Return 1 if the specified pointer is in the set, 0 otherwise.
std::pair< iterator, bool > insert(PtrType Ptr)
Inserts Ptr if and only if there is no element in the container equal to Ptr.
This class consists of common code factored out of the SmallVector class to reduce code duplication b...
void push_back(const T &Elt)
This is a 'vector' (really, a variable-sized array), optimized for the case when the array is small.
A BumpPtrAllocator that allows only elements of a specific type to be allocated.
Definition Allocator.h:398
Encapsulates all of the information needed to generate a stack protector check, and signals to isel w...
MachineBasicBlock * getSuccessMBB()
MachineBasicBlock * getFailureMBB()
constexpr bool empty() const
Check if the string is empty.
Definition StringRef.h:141
constexpr const char * data() const
Get a pointer to the start of the string (which may not be null terminated).
Definition StringRef.h:138
SwitchLowering(FunctionLoweringInfo &funcinfo)
Analysis pass providing the TargetLibraryInfo.
Provides information about what library functions are available for the current target.
This class defines information used to lower LLVM code to legal SelectionDAG operators that the targe...
Primary interface to the complete machine description for the target machine.
ExceptionHandling getExceptionModel() const
Return the ExceptionHandling to use.
const Triple & getTargetTriple() const
const Target & getTarget() const
unsigned NoTrapAfterNoreturn
Do not emit a trap instruction for 'unreachable' IR instructions behind noreturn calls,...
unsigned TrapUnreachable
Emit target-specific trap instruction for 'unreachable' IR instructions.
Target-Independent Code Generator Pass Configuration Options.
virtual std::unique_ptr< CSEConfigBase > getCSEConfig() const
Returns the CSEConfig object to use for the current optimization level.
TargetSubtargetInfo - Generic base class for all target subtargets.
virtual const CallLowering * getCallLowering() const
virtual const TargetLowering * getTargetLowering() const
bool isSPIRV() const
Tests whether the target is SPIR-V (32/64-bit/Logical).
Definition Triple.h:979
Twine - A lightweight data structure for efficiently representing the concatenation of temporary valu...
Definition Twine.h:82
static constexpr TypeSize getZero()
Definition TypeSize.h:345
The instances of the Type class are immutable: once they are created, they are never changed.
Definition Type.h:46
LLVM_ABI bool isEmptyTy() const
Return true if this type is empty, that is, it has no elements or all of its elements are empty.
Definition Type.cpp:170
static LLVM_ABI IntegerType * getInt32Ty(LLVMContext &C)
Definition Type.cpp:299
bool isSized() const
Return true if it makes sense to take the size of this type.
Definition Type.h:321
static LLVM_ABI Type * getVoidTy(LLVMContext &C)
Definition Type.cpp:272
bool isByteOrByteVectorTy() const
Return true if this is a byte type or a vector of byte types.
Definition Type.h:243
bool isAggregateType() const
Return true if the type is an aggregate type.
Definition Type.h:314
bool isPtrOrPtrVectorTy() const
Return true if this is a pointer type or a vector of pointer types.
Definition Type.h:280
bool isTokenTy() const
Return true if this is 'token'.
Definition Type.h:231
bool isVoidTy() const
Return true if this is 'void'.
Definition Type.h:141
BasicBlock * getSuccessor(unsigned i=0) const
Value * getOperand(unsigned i) const
Definition User.h:207
LLVM Value Representation.
Definition Value.h:75
Type * getType() const
All values are typed, get the type of this value.
Definition Value.h:257
bool hasOneUse() const
Return true if there is exactly one use of this value.
Definition Value.h:441
LLVMContext & getContext() const
All values hold a context through their type.
Definition Value.h:260
LLVM_ABI const Value * stripPointerCasts() const
Strip off pointer casts, all-zero GEPs and address space casts.
Definition Value.cpp:712
constexpr ScalarTy getFixedValue() const
Definition TypeSize.h:200
constexpr bool isScalable() const
Returns whether the quantity is scaled by a runtime quantity (vscale).
Definition TypeSize.h:168
constexpr ScalarTy getKnownMinValue() const
Returns the minimum value this quantity can represent.
Definition TypeSize.h:165
constexpr bool isZero() const
Definition TypeSize.h:153
An efficient, type-erasing, non-owning reference to a callable.
const ParentTy * getParent() const
Definition ilist_node.h:34
NodeTy * getNextNode()
Get the next node, or nullptr for the list tail.
Definition ilist_node.h:348
A raw_ostream that writes to an std::string.
Pass manager infrastructure for declaring and invalidating analyses.
#define llvm_unreachable(msg)
Marks that the current location is not supposed to be reachable.
constexpr char Align[]
Key for Kernel::Arg::Metadata::mAlign.
constexpr char Args[]
Key for Kernel::Metadata::mArgs.
constexpr char SymbolName[]
Key for Kernel::Metadata::mSymbolName.
constexpr std::underlying_type_t< E > Mask()
Get a bitmask with 1s in all places up to the high-order bit of E's largest value.
@ BasicBlock
Various leaf nodes.
Definition ISDOpcodes.h:83
BinaryOp_match< SrcTy, SpecificConstantMatch, TargetOpcode::G_XOR, true > m_Not(const SrcTy &&Src)
Matches a register not-ed by a G_XOR.
OneUse_match< SubPat > m_OneUse(const SubPat &SP)
bool match(Val *V, const Pattern &P)
specificval_ty m_Specific(const Value *V)
Match if we have a specific specified value.
TwoOps_match< Val_t, Idx_t, Instruction::ExtractElement > m_ExtractElt(const Val_t &Val, const Idx_t &Idx)
Matches ExtractElementInst.
auto m_Value()
Match an arbitrary value and ignore it.
auto m_LogicalOr()
Matches L || R where L and R are arbitrary values.
auto m_LogicalAnd()
Matches L && R where L and R are arbitrary values.
Offsets
Offsets in bytes from the start of the input buffer.
LLVM_ABI void sortAndRangeify(CaseClusterVector &Clusters)
Sort Clusters and merge adjacent cases.
std::vector< CaseCluster > CaseClusterVector
@ CC_Range
A cluster of adjacent case labels with the same destination, or just one case.
@ CC_JumpTable
A cluster of cases suitable for jump table lowering.
@ CC_BitTests
A cluster of cases suitable for bit test lowering.
SmallVector< SwitchWorkListItem, 4 > SwitchWorkList
CaseClusterVector::iterator CaseClusterIt
@ CE
Windows NT (Windows on ARM)
Definition MCAsmInfo.h:51
initializer< Ty > init(const Ty &Val)
ExceptionBehavior
Exception behavior used for floating point operations.
Definition FPEnv.h:39
@ ebIgnore
This corresponds to "fpexcept.ignore".
Definition FPEnv.h:40
DiagnosticInfoOptimizationBase::Argument NV
NodeAddr< PhiNode * > Phi
Definition RDFGraph.h:390
NodeAddr< CodeNode * > Code
Definition RDFGraph.h:388
friend class Instruction
Iterator for Instructions in a `BasicBlock.
Definition BasicBlock.h:73
BaseReg
Stack frame base register. Bit 0 of FREInfo.Info.
Definition SFrame.h:77
This is an optimization pass for GlobalISel generic memory operations.
auto drop_begin(T &&RangeOrContainer, size_t N=1)
Return a range covering RangeOrContainer with the first N elements excluded.
Definition STLExtras.h:316
@ Low
Lower the current thread's priority such that it does not affect foreground tasks significantly.
Definition Threading.h:280
@ Offset
Definition DWP.cpp:577
OuterAnalysisManagerProxy< ModuleAnalysisManager, MachineFunction > ModuleAnalysisManagerMachineFunctionProxy
Provide the ModuleAnalysisManager to Function proxy.
@ Implicit
Not emitted register (e.g. carry, or temporary result).
@ Undef
Value of the register doesn't matter.
auto enumerate(FirstRange &&First, RestRanges &&...Rest)
Given two or more input ranges, returns a new range whose values are tuples (A, B,...
Definition STLExtras.h:2570
decltype(auto) dyn_cast(const From &Val)
dyn_cast<X> - Return the argument parameter cast to the specified type.
Definition Casting.h:643
int countr_one(T Value)
Count the number of ones from the least significant bit to the first zero bit.
Definition bit.h:315
LLVM_ABI void diagnoseDontCall(const CallInst &CI)
auto successors(const MachineBasicBlock *BB)
LLVM_ABI bool isExceptionPointerAndSelectorType(Type *Ty)
Return true if landingpad result type Ty is a struct of an exception pointer (pointer or integer) and...
Definition Analysis.cpp:201
LLVM_ABI MVT getMVTForLLT(LLT Ty)
Get a rough equivalent of an MVT for a given LLT.
void append_range(Container &C, Range &&R)
Wrapper function to append range R to container C.
Definition STLExtras.h:2224
constexpr bool isUIntN(unsigned N, uint64_t x)
Checks if an unsigned integer fits into the given (dynamic) bit width.
Definition MathExtras.h:244
gep_type_iterator gep_type_end(const User *GEP)
LLVM_ABI MachineBasicBlock::iterator findSplitPointForStackProtector(MachineBasicBlock *BB, const TargetInstrInfo &TII)
Find the split point at which to splice the end of BB into its success stack protector check machine ...
LLVM_ABI LLT getLLTForMVT(MVT Ty)
Get a rough equivalent of an LLT for a given MVT.
AnalysisManager< MachineFunction > MachineFunctionAnalysisManager
constexpr int popcount(T Value) noexcept
Count the number of set bits in a value.
Definition bit.h:156
RelativeUniformCounterPtr ValuesPtrExpr VTableAddr Value
Definition InstrProf.h:143
int countr_zero(T Val)
Count number of 0's from the least significant bit to the most stopping at the first 1.
Definition bit.h:204
LLVM_ABI PreservedAnalyses getMachineFunctionPassPreservedAnalyses()
Returns the minimum set of Analyses that all machine function passes must preserve.
constexpr bool has_single_bit(T Value) noexcept
Definition bit.h:149
bool any_of(R &&range, UnaryPredicate P)
Provide wrappers to std::any_of which take ranges instead of having to pass begin/end explicitly.
Definition STLExtras.h:1762
LLVM_ABI llvm::SmallVector< int, 16 > createStrideMask(unsigned Start, unsigned Stride, unsigned VF)
Create a stride shuffle mask.
auto reverse(ContainerTy &&C)
Definition STLExtras.h:408
LLVM_ABI const LibcallLoweringInfo & getLibcallLowering(const ModuleLibcallLoweringInfo &ModuleInfo, const TargetSubtargetInfo &Subtarget)
Resolve the LibcallLoweringInfo for Subtarget from the module-level ModuleInfo, applying the subtarge...
decltype(auto) get(const PointerIntPair< PointerTy, IntBits, IntType, PtrTraits, Info > &Pair)
void sort(IteratorTy Start, IteratorTy End)
Definition STLExtras.h:1652
LLVM_ABI raw_ostream & dbgs()
dbgs() - This returns a reference to a raw_ostream for debugging messages.
Definition Debug.cpp:209
LLVM_ABI void report_fatal_error(Error Err, bool gen_crash_diag=true)
Definition Error.cpp:163
generic_gep_type_iterator<> gep_type_iterator
auto succ_size(const MachineBasicBlock *BB)
LLVM_ABI EHPersonality classifyEHPersonality(const Value *Pers)
See if the given exception handling personality function is one that we understand.
CodeGenOptLevel
Code generation optimization level.
Definition CodeGen.h:227
class LLVM_GSL_OWNER SmallVector
Forward declaration of SmallVector so that calculateSmallVectorDefaultInlinedElements can reference s...
bool isa(const From &Val)
isa<X> - Return true if the parameter to the template is an instance of one of the template type argu...
Definition Casting.h:547
@ Success
The lock was released successfully.
LLVM_ATTRIBUTE_VISIBILITY_DEFAULT AnalysisKey InnerAnalysisManagerProxy< AnalysisManagerT, IRUnitT, ExtraArgTs... >::Key
@ Global
Append to llvm.global_dtors.
@ First
Helpers to iterate all locations in the MemoryEffectsBase class.
Definition ModRef.h:74
LLVM_ABI void getSelectionDAGFallbackAnalysisUsage(AnalysisUsage &AU)
Modify analysis usage so it preserves passes required for the SelectionDAG fallback.
Definition Utils.cpp:1137
auto lower_bound(R &&Range, T &&Value)
Provide wrappers to std::lower_bound which take ranges instead of having to pass begin/end explicitly...
Definition STLExtras.h:2068
LLVM_ABI llvm::SmallVector< int, 16 > createInterleaveMask(unsigned VF, unsigned NumVecs)
Create an interleave shuffle mask.
@ FMul
Product of floats.
@ Sub
Subtraction of integers.
IntPtrTy
Definition InstrProf.h:82
DWARFExpression::Operation Op
ArrayRef(const T &OneElt) -> ArrayRef< T >
bool isAsynchronousEHPersonality(EHPersonality Pers)
Returns true if this personality function catches asynchronous exceptions.
@ Default
Not specified; resolve to the target's default model.
Definition CodeGen.h:55
decltype(auto) cast(const From &Val)
cast<X> - Return the argument parameter cast to the specified type.
Definition Casting.h:559
LLVM_ABI std::optional< RoundingMode > convertStrToRoundingMode(StringRef)
Returns a valid RoundingMode enumerator when given a string that is valid as input in constrained int...
Definition FPEnv.cpp:25
gep_type_iterator gep_type_begin(const User *GEP)
LLVM_ABI void computeValueLLTs(const DataLayout &DL, Type &Ty, SmallVectorImpl< LLT > &ValueLLTs, SmallVectorImpl< TypeSize > *Offsets=nullptr, TypeSize StartingOffset=TypeSize::getZero())
computeValueLLTs - Given an LLVM IR type, compute a sequence of LLTs that represent all the individua...
Definition Analysis.cpp:155
LLVM_ABI GlobalValue * ExtractTypeInfo(Value *V)
ExtractTypeInfo - Returns the type info, possibly bitcast, encoded in V.
Definition Analysis.cpp:183
Align commonAlignment(Align A, uint64_t Offset)
Returns the alignment that satisfies both alignments.
Definition Alignment.h:206
Align getKnownAlignment(Value *V, const DataLayout &DL, const Instruction *CtxI=nullptr, AssumptionCache *AC=nullptr, const DominatorTree *DT=nullptr)
Try to infer an alignment for the specified pointer.
Definition Local.h:240
LLVM_ABI LLT getLLTForType(Type &Ty, const DataLayout &DL)
Construct a low-level type based on an LLVM type.
LLVM_ABI void reportFatalUsageError(Error Err)
Report a fatal error that does not indicate a bug in LLVM.
Definition Error.cpp:177
Implement std::hash so that hash_code can be used in STL containers.
Definition BitVector.h:878
void swap(llvm::BitVector &LHS, llvm::BitVector &RHS)
Implement std::swap in terms of BitVector swap.
Definition BitVector.h:880
This struct is a compact representation of a valid (non-zero power of two) alignment.
Definition Alignment.h:39
constexpr uint64_t value() const
This is a hole in the type system and should not be abused.
Definition Alignment.h:82
Pair of physical register and lane mask.
static LLVM_ABI MachinePointerInfo getFixedStack(MachineFunction &MF, int FI, int64_t Offset=0)
Return a MachinePointerInfo record that refers to the specified FrameIndex.
Align valueOrOne() const
For convenience, returns a valid alignment or 1 if undefined.
Definition Alignment.h:135
static bool canHandle(const Instruction *I, const TargetLibraryInfo &TLI)
static StringRef getLibcallImplName(RTLIB::LibcallImpl CallImpl)
Get the libcall routine name for the specified libcall implementation.
This structure is used to communicate between SelectionDAGBuilder and SDISel for the code generation ...
Register Reg
The virtual register containing the index of the jump table entry to jump to.
MachineBasicBlock * Default
The MBB of the default bb, which is a successor of the range check MBB.
unsigned JTI
The JumpTableIndex for this jump table in the function.
MachineBasicBlock * MBB
The MBB into which to emit the code for the indirect jump.