LLVM 24.0.0git
SelectionDAG.cpp
Go to the documentation of this file.
1//===- SelectionDAG.cpp - Implement the SelectionDAG data structures ------===//
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//
9// This implements the SelectionDAG class.
10//
11//===----------------------------------------------------------------------===//
12
14#include "SDNodeDbgValue.h"
15#include "llvm/ADT/APFloat.h"
16#include "llvm/ADT/APInt.h"
17#include "llvm/ADT/APSInt.h"
18#include "llvm/ADT/ArrayRef.h"
19#include "llvm/ADT/BitVector.h"
20#include "llvm/ADT/DenseSet.h"
21#include "llvm/ADT/FoldingSet.h"
22#include "llvm/ADT/STLExtras.h"
25#include "llvm/ADT/Twine.h"
52#include "llvm/IR/Constant.h"
53#include "llvm/IR/Constants.h"
54#include "llvm/IR/DataLayout.h"
56#include "llvm/IR/DebugLoc.h"
58#include "llvm/IR/Function.h"
59#include "llvm/IR/GlobalValue.h"
60#include "llvm/IR/Metadata.h"
61#include "llvm/IR/Type.h"
65#include "llvm/Support/Debug.h"
75#include <algorithm>
76#include <cassert>
77#include <cstdint>
78#include <cstdlib>
79#include <limits>
80#include <optional>
81#include <string>
82#include <utility>
83#include <vector>
84
85using namespace llvm;
86using namespace llvm::SDPatternMatch;
87
88/// makeVTList - Return an instance of the SDVTList struct initialized with the
89/// specified members.
90static SDVTList makeVTList(const EVT *VTs, unsigned NumVTs) {
91 SDVTList Res = {VTs, NumVTs};
92 return Res;
93}
94
95// Default null implementations of the callbacks.
99
100void SelectionDAG::DAGNodeDeletedListener::anchor() {}
101void SelectionDAG::DAGNodeInsertedListener::anchor() {}
102
103#define DEBUG_TYPE "selectiondag"
104
105static cl::opt<bool> EnableMemCpyDAGOpt("enable-memcpy-dag-opt",
106 cl::Hidden, cl::init(true),
107 cl::desc("Gang up loads and stores generated by inlining of memcpy"));
108
109static cl::opt<int> MaxLdStGlue("ldstmemcpy-glue-max",
110 cl::desc("Number limit for gluing ld/st of memcpy."),
111 cl::Hidden, cl::init(0));
112
114 MaxSteps("has-predecessor-max-steps", cl::Hidden, cl::init(8192),
115 cl::desc("DAG combiner limit number of steps when searching DAG "
116 "for predecessor nodes"));
117
119 "vscale-unroll-limit",
120 cl::desc("Maximum vscale for which vector unrolling is allowed."),
121 cl::Hidden, cl::init(64));
122
124 LLVM_DEBUG(dbgs() << Msg; V.getNode()->dump(G););
125}
126
128
129//===----------------------------------------------------------------------===//
130// ConstantFPSDNode Class
131//===----------------------------------------------------------------------===//
132
133/// isExactlyValue - We don't rely on operator== working on double values, as
134/// it returns true for things that are clearly not equal, like -0.0 and 0.0.
135/// As such, this method can be used to do an exact bit-for-bit comparison of
136/// two floating point values.
138 return getValueAPF().bitwiseIsEqual(V);
139}
140
142 const APFloat& Val) {
143 assert(VT.isFloatingPoint() && "Can only convert between FP types");
144
145 // convert modifies in place, so make a copy.
146 APFloat Val2 = APFloat(Val);
147 bool losesInfo;
149 &losesInfo);
150 return !losesInfo;
151}
152
153//===----------------------------------------------------------------------===//
154// ISD Namespace
155//===----------------------------------------------------------------------===//
156
157bool ISD::isConstantSplatVector(const SDNode *N, APInt &SplatVal) {
158 if (N->getOpcode() == ISD::SPLAT_VECTOR) {
159 if (auto OptAPInt = N->getOperand(0)->bitcastToAPInt()) {
160 unsigned EltSize =
161 N->getValueType(0).getVectorElementType().getSizeInBits();
162 SplatVal = OptAPInt->trunc(EltSize);
163 return true;
164 }
165 }
166
167 auto *BV = dyn_cast<BuildVectorSDNode>(N);
168 if (!BV)
169 return false;
170
171 APInt SplatUndef;
172 unsigned SplatBitSize;
173 bool HasUndefs;
174 unsigned EltSize = N->getValueType(0).getVectorElementType().getSizeInBits();
175 // Endianness does not matter here. We are checking for a splat given the
176 // element size of the vector, and if we find such a splat for little endian
177 // layout, then that should be valid also for big endian (as the full vector
178 // size is known to be a multiple of the element size).
179 const bool IsBigEndian = false;
180 return BV->isConstantSplat(SplatVal, SplatUndef, SplatBitSize, HasUndefs,
181 EltSize, IsBigEndian) &&
182 EltSize == SplatBitSize;
183}
184
185// FIXME: AllOnes and AllZeros duplicate a lot of code. Could these be
186// specializations of the more general isConstantSplatVector()?
187
188bool ISD::isConstantSplatVectorAllOnes(const SDNode *N, bool BuildVectorOnly) {
189 // Look through a bit convert.
190 while (N->getOpcode() == ISD::BITCAST)
191 N = N->getOperand(0).getNode();
192
193 if (!BuildVectorOnly && N->getOpcode() == ISD::SPLAT_VECTOR) {
194 APInt SplatVal;
195 return isConstantSplatVector(N, SplatVal) && SplatVal.isAllOnes();
196 }
197
198 if (N->getOpcode() != ISD::BUILD_VECTOR) return false;
199
200 unsigned i = 0, e = N->getNumOperands();
201
202 // Skip over all of the undef values.
203 while (i != e && N->getOperand(i).isUndef())
204 ++i;
205
206 // Do not accept an all-undef vector.
207 if (i == e) return false;
208
209 // Do not accept build_vectors that aren't all constants or which have non-~0
210 // elements. We have to be a bit careful here, as the type of the constant
211 // may not be the same as the type of the vector elements due to type
212 // legalization (the elements are promoted to a legal type for the target and
213 // a vector of a type may be legal when the base element type is not).
214 // We only want to check enough bits to cover the vector elements, because
215 // we care if the resultant vector is all ones, not whether the individual
216 // constants are.
217 SDValue NotZero = N->getOperand(i);
218 if (auto OptAPInt = NotZero->bitcastToAPInt()) {
219 unsigned EltSize = N->getValueType(0).getScalarSizeInBits();
220 if (OptAPInt->countr_one() < EltSize)
221 return false;
222 } else
223 return false;
224
225 // Okay, we have at least one ~0 value, check to see if the rest match or are
226 // undefs. Even with the above element type twiddling, this should be OK, as
227 // the same type legalization should have applied to all the elements.
228 for (++i; i != e; ++i)
229 if (N->getOperand(i) != NotZero && !N->getOperand(i).isUndef())
230 return false;
231 return true;
232}
233
234bool ISD::isConstantSplatVectorAllZeros(const SDNode *N, bool BuildVectorOnly) {
235 // Look through a bit convert.
236 while (N->getOpcode() == ISD::BITCAST)
237 N = N->getOperand(0).getNode();
238
239 if (!BuildVectorOnly && N->getOpcode() == ISD::SPLAT_VECTOR) {
240 APInt SplatVal;
241 return isConstantSplatVector(N, SplatVal) && SplatVal.isZero();
242 }
243
244 if (N->getOpcode() != ISD::BUILD_VECTOR) return false;
245
246 bool IsAllUndef = true;
247 for (const SDValue &Op : N->op_values()) {
248 if (Op.isUndef())
249 continue;
250 IsAllUndef = false;
251 // Do not accept build_vectors that aren't all constants or which have non-0
252 // elements. We have to be a bit careful here, as the type of the constant
253 // may not be the same as the type of the vector elements due to type
254 // legalization (the elements are promoted to a legal type for the target
255 // and a vector of a type may be legal when the base element type is not).
256 // We only want to check enough bits to cover the vector elements, because
257 // we care if the resultant vector is all zeros, not whether the individual
258 // constants are.
259 if (auto OptAPInt = Op->bitcastToAPInt()) {
260 unsigned EltSize = N->getValueType(0).getScalarSizeInBits();
261 if (OptAPInt->countr_zero() < EltSize)
262 return false;
263 } else
264 return false;
265 }
266
267 // Do not accept an all-undef vector.
268 if (IsAllUndef)
269 return false;
270 return true;
271}
272
274 return isConstantSplatVectorAllOnes(N, /*BuildVectorOnly*/ true);
275}
276
278 return isConstantSplatVectorAllZeros(N, /*BuildVectorOnly*/ true);
279}
280
282 if (N->getOpcode() != ISD::BUILD_VECTOR)
283 return false;
284
285 for (const SDValue &Op : N->op_values()) {
286 if (Op.isUndef())
287 continue;
289 return false;
290 }
291 return true;
292}
293
295 if (N->getOpcode() != ISD::BUILD_VECTOR)
296 return false;
297
298 for (const SDValue &Op : N->op_values()) {
299 if (Op.isUndef())
300 continue;
302 return false;
303 }
304 return true;
305}
306
307bool ISD::isVectorShrinkable(const SDNode *N, unsigned NewEltSize,
308 bool Signed) {
309 assert(N->getValueType(0).isVector() && "Expected a vector!");
310
311 unsigned EltSize = N->getValueType(0).getScalarSizeInBits();
312 if (EltSize <= NewEltSize)
313 return false;
314
315 if (N->getOpcode() == ISD::ZERO_EXTEND) {
316 return (N->getOperand(0).getValueType().getScalarSizeInBits() <=
317 NewEltSize) &&
318 !Signed;
319 }
320 if (N->getOpcode() == ISD::SIGN_EXTEND) {
321 return (N->getOperand(0).getValueType().getScalarSizeInBits() <=
322 NewEltSize) &&
323 Signed;
324 }
325 if (N->getOpcode() != ISD::BUILD_VECTOR)
326 return false;
327
328 for (const SDValue &Op : N->op_values()) {
329 if (Op.isUndef())
330 continue;
332 return false;
333
334 APInt C = Op->getAsAPIntVal().trunc(EltSize);
335 if (Signed && C.trunc(NewEltSize).sext(EltSize) != C)
336 return false;
337 if (!Signed && C.trunc(NewEltSize).zext(EltSize) != C)
338 return false;
339 }
340
341 return true;
342}
343
345 // Return false if the node has no operands.
346 // This is "logically inconsistent" with the definition of "all" but
347 // is probably the desired behavior.
348 if (N->getNumOperands() == 0)
349 return false;
350 return all_of(N->op_values(), [](SDValue Op) { return Op.isUndef(); });
351}
352
354 return N->getOpcode() == ISD::FREEZE && N->getOperand(0).isUndef();
355}
356
357template <typename ConstNodeType>
359 std::function<bool(ConstNodeType *)> Match,
360 bool AllowUndefs, bool AllowTruncation) {
361 // FIXME: Add support for scalar UNDEF cases?
362 if (auto *C = dyn_cast<ConstNodeType>(Op))
363 return Match(C);
364
365 // FIXME: Add support for vector UNDEF cases?
366 if (ISD::BUILD_VECTOR != Op.getOpcode() &&
367 ISD::SPLAT_VECTOR != Op.getOpcode())
368 return false;
369
370 if (ISD::SPLAT_VECTOR == Op.getOpcode() && !DemandedElts)
371 return true;
372
373 EVT SVT = Op.getValueType().getScalarType();
374 for (unsigned i = 0, e = Op.getNumOperands(); i != e; ++i) {
375 if (ISD::SPLAT_VECTOR != Op.getOpcode() && !DemandedElts[i])
376 continue;
377
378 if (AllowUndefs && Op.getOperand(i).isUndef()) {
379 if (!Match(nullptr))
380 return false;
381 continue;
382 }
383
384 auto *Cst = dyn_cast<ConstNodeType>(Op.getOperand(i));
385 if (!Cst || (!AllowTruncation && Cst->getValueType(0) != SVT) ||
386 !Match(Cst))
387 return false;
388 }
389 return true;
390}
391// Build used template types.
393 SDValue, const APInt &, std::function<bool(ConstantSDNode *)>, bool, bool);
395 SDValue, const APInt &, std::function<bool(ConstantFPSDNode *)>, bool,
396 bool);
397
399 SDValue LHS, SDValue RHS, const APInt &DemandedElts,
400 std::function<bool(ConstantSDNode *, ConstantSDNode *)> Match,
401 bool AllowUndefs, bool AllowTypeMismatch) {
402 if (!AllowTypeMismatch && LHS.getValueType() != RHS.getValueType())
403 return false;
404
405 // TODO: Add support for scalar UNDEF cases?
406 if (auto *LHSCst = dyn_cast<ConstantSDNode>(LHS))
407 if (auto *RHSCst = dyn_cast<ConstantSDNode>(RHS))
408 return Match(LHSCst, RHSCst);
409
410 // TODO: Add support for vector UNDEF cases?
411 if (LHS.getOpcode() != RHS.getOpcode() ||
412 (LHS.getOpcode() != ISD::BUILD_VECTOR &&
413 LHS.getOpcode() != ISD::SPLAT_VECTOR))
414 return false;
415
416 if (ISD::SPLAT_VECTOR == LHS.getOpcode() && !DemandedElts)
417 return true;
418
419 EVT SVT = LHS.getValueType().getScalarType();
420 for (unsigned i = 0, e = LHS.getNumOperands(); i != e; ++i) {
421 if (ISD::SPLAT_VECTOR != LHS.getOpcode() && !DemandedElts[i])
422 continue;
423 SDValue LHSOp = LHS.getOperand(i);
424 SDValue RHSOp = RHS.getOperand(i);
425 bool LHSUndef = AllowUndefs && LHSOp.isUndef();
426 bool RHSUndef = AllowUndefs && RHSOp.isUndef();
427 auto *LHSCst = dyn_cast<ConstantSDNode>(LHSOp);
428 auto *RHSCst = dyn_cast<ConstantSDNode>(RHSOp);
429 if ((!LHSCst && !LHSUndef) || (!RHSCst && !RHSUndef))
430 return false;
431 if (!AllowTypeMismatch && (LHSOp.getValueType() != SVT ||
432 LHSOp.getValueType() != RHSOp.getValueType()))
433 return false;
434 if (!Match(LHSCst, RHSCst))
435 return false;
436 }
437 return true;
438}
439
441 switch (MinMaxOpc) {
442 default:
443 llvm_unreachable("unrecognized opcode");
444 case ISD::UMIN:
445 return ISD::UMAX;
446 case ISD::UMAX:
447 return ISD::UMIN;
448 case ISD::SMIN:
449 return ISD::SMAX;
450 case ISD::SMAX:
451 return ISD::SMIN;
452 }
453}
454
456 switch (MinMaxOpc) {
457 default:
458 llvm_unreachable("unrecognized min/max opcode");
459 case ISD::SMIN:
460 return ISD::UMIN;
461 case ISD::SMAX:
462 return ISD::UMAX;
463 case ISD::UMIN:
464 return ISD::SMIN;
465 case ISD::UMAX:
466 return ISD::SMAX;
467 }
468}
469
471 switch (VecReduceOpcode) {
472 default:
473 llvm_unreachable("Expected VECREDUCE opcode");
476 case ISD::VP_REDUCE_FADD:
477 case ISD::VP_REDUCE_SEQ_FADD:
478 return ISD::FADD;
481 case ISD::VP_REDUCE_FMUL:
482 case ISD::VP_REDUCE_SEQ_FMUL:
483 return ISD::FMUL;
485 case ISD::VP_REDUCE_ADD:
486 return ISD::ADD;
488 case ISD::VP_REDUCE_MUL:
489 return ISD::MUL;
491 case ISD::VP_REDUCE_AND:
492 return ISD::AND;
494 case ISD::VP_REDUCE_OR:
495 return ISD::OR;
497 case ISD::VP_REDUCE_XOR:
498 return ISD::XOR;
500 case ISD::VP_REDUCE_SMAX:
501 return ISD::SMAX;
503 case ISD::VP_REDUCE_SMIN:
504 return ISD::SMIN;
506 case ISD::VP_REDUCE_UMAX:
507 return ISD::UMAX;
509 case ISD::VP_REDUCE_UMIN:
510 return ISD::UMIN;
512 case ISD::VP_REDUCE_FMAX:
513 return ISD::FMAXNUM;
515 case ISD::VP_REDUCE_FMIN:
516 return ISD::FMINNUM;
518 case ISD::VP_REDUCE_FMAXIMUM:
519 return ISD::FMAXIMUM;
521 case ISD::VP_REDUCE_FMINIMUM:
522 return ISD::FMINIMUM;
524 return ISD::FMAXIMUMNUM;
526 return ISD::FMINIMUMNUM;
527 }
528}
529
531 switch (MaskedOpc) {
532 case ISD::MASKED_UDIV:
533 return ISD::UDIV;
534 case ISD::MASKED_SDIV:
535 return ISD::SDIV;
536 case ISD::MASKED_UREM:
537 return ISD::UREM;
538 case ISD::MASKED_SREM:
539 return ISD::SREM;
540 default:
541 llvm_unreachable("Expected masked binop opcode");
542 }
543}
544
545bool ISD::isVPOpcode(unsigned Opcode) {
546 switch (Opcode) {
547 default:
548 return false;
549#define BEGIN_REGISTER_VP_SDNODE(VPSD, ...) \
550 case ISD::VPSD: \
551 return true;
552#include "llvm/IR/VPIntrinsics.def"
553 }
554}
555
556bool ISD::isVPBinaryOp(unsigned Opcode) {
557 switch (Opcode) {
558 default:
559 break;
560#define BEGIN_REGISTER_VP_SDNODE(VPSD, ...) case ISD::VPSD:
561#define VP_PROPERTY_BINARYOP return true;
562#define END_REGISTER_VP_SDNODE(VPSD) break;
563#include "llvm/IR/VPIntrinsics.def"
564 }
565 return false;
566}
567
568bool ISD::isVPReduction(unsigned Opcode) {
569 switch (Opcode) {
570 default:
571 return false;
572 case ISD::VP_REDUCE_ADD:
573 case ISD::VP_REDUCE_MUL:
574 case ISD::VP_REDUCE_AND:
575 case ISD::VP_REDUCE_OR:
576 case ISD::VP_REDUCE_XOR:
577 case ISD::VP_REDUCE_SMAX:
578 case ISD::VP_REDUCE_SMIN:
579 case ISD::VP_REDUCE_UMAX:
580 case ISD::VP_REDUCE_UMIN:
581 case ISD::VP_REDUCE_FMAX:
582 case ISD::VP_REDUCE_FMIN:
583 case ISD::VP_REDUCE_FMAXIMUM:
584 case ISD::VP_REDUCE_FMINIMUM:
585 case ISD::VP_REDUCE_FADD:
586 case ISD::VP_REDUCE_FMUL:
587 case ISD::VP_REDUCE_SEQ_FADD:
588 case ISD::VP_REDUCE_SEQ_FMUL:
589 return true;
590 }
591}
592
593/// The operand position of the vector mask.
594std::optional<unsigned> ISD::getVPMaskIdx(unsigned Opcode) {
595 switch (Opcode) {
596 default:
597 return std::nullopt;
598#define BEGIN_REGISTER_VP_SDNODE(VPSD, LEGALPOS, TDNAME, MASKPOS, ...) \
599 case ISD::VPSD: \
600 return MASKPOS;
601#include "llvm/IR/VPIntrinsics.def"
602 }
603}
604
605/// The operand position of the explicit vector length parameter.
606std::optional<unsigned> ISD::getVPExplicitVectorLengthIdx(unsigned Opcode) {
607 switch (Opcode) {
608 default:
609 return std::nullopt;
610#define BEGIN_REGISTER_VP_SDNODE(VPSD, LEGALPOS, TDNAME, MASKPOS, EVLPOS) \
611 case ISD::VPSD: \
612 return EVLPOS;
613#include "llvm/IR/VPIntrinsics.def"
614 }
615}
616
617std::optional<unsigned> ISD::getBaseOpcodeForVP(unsigned VPOpcode,
618 bool hasFPExcept) {
619 // FIXME: Return strict opcodes in case of fp exceptions.
620 switch (VPOpcode) {
621 default:
622 return std::nullopt;
623#define BEGIN_REGISTER_VP_SDNODE(VPOPC, ...) case ISD::VPOPC:
624#define VP_PROPERTY_FUNCTIONAL_SDOPC(SDOPC) return ISD::SDOPC;
625#define END_REGISTER_VP_SDNODE(VPOPC) break;
626#include "llvm/IR/VPIntrinsics.def"
627 }
628 return std::nullopt;
629}
630
631std::optional<unsigned> ISD::getVPForBaseOpcode(unsigned Opcode) {
632 switch (Opcode) {
633 default:
634 return std::nullopt;
635#define BEGIN_REGISTER_VP_SDNODE(VPOPC, ...) break;
636#define VP_PROPERTY_FUNCTIONAL_SDOPC(SDOPC) case ISD::SDOPC:
637#define END_REGISTER_VP_SDNODE(VPOPC) return ISD::VPOPC;
638#include "llvm/IR/VPIntrinsics.def"
639 }
640}
641
643 switch (ExtType) {
644 case ISD::EXTLOAD:
645 return IsFP ? ISD::FP_EXTEND : ISD::ANY_EXTEND;
646 case ISD::SEXTLOAD:
647 return ISD::SIGN_EXTEND;
648 case ISD::ZEXTLOAD:
649 return ISD::ZERO_EXTEND;
650 default:
651 break;
652 }
653
654 llvm_unreachable("Invalid LoadExtType");
655}
656
658 // To perform this operation, we just need to swap the L and G bits of the
659 // operation.
660 unsigned OldL = (Operation >> 2) & 1;
661 unsigned OldG = (Operation >> 1) & 1;
662 return ISD::CondCode((Operation & ~6) | // Keep the N, U, E bits
663 (OldL << 1) | // New G bit
664 (OldG << 2)); // New L bit.
665}
666
668 unsigned Operation = Op;
669 if (isIntegerLike)
670 Operation ^= 7; // Flip L, G, E bits, but not U.
671 else
672 Operation ^= 15; // Flip all of the condition bits.
673
675 Operation &= ~8; // Don't let N and U bits get set.
676
677 return ISD::CondCode(Operation);
678}
679
683
685 bool isIntegerLike) {
686 return getSetCCInverseImpl(Op, isIntegerLike);
687}
688
689/// For an integer comparison, return 1 if the comparison is a signed operation
690/// and 2 if the result is an unsigned comparison. Return zero if the operation
691/// does not depend on the sign of the input (setne and seteq).
692static int isSignedOp(ISD::CondCode Opcode) {
693 switch (Opcode) {
694 default: llvm_unreachable("Illegal integer setcc operation!");
695 case ISD::SETEQ:
696 case ISD::SETNE: return 0;
697 case ISD::SETLT:
698 case ISD::SETLE:
699 case ISD::SETGT:
700 case ISD::SETGE: return 1;
701 case ISD::SETULT:
702 case ISD::SETULE:
703 case ISD::SETUGT:
704 case ISD::SETUGE: return 2;
705 }
706}
707
709 EVT Type) {
710 bool IsInteger = Type.isInteger();
711 if (IsInteger && (isSignedOp(Op1) | isSignedOp(Op2)) == 3)
712 // Cannot fold a signed integer setcc with an unsigned integer setcc.
713 return ISD::SETCC_INVALID;
714
715 unsigned Op = Op1 | Op2; // Combine all of the condition bits.
716
717 // If the N and U bits get set, then the resultant comparison DOES suddenly
718 // care about orderedness, and it is true when ordered.
719 if (Op > ISD::SETTRUE2)
720 Op &= ~16; // Clear the U bit if the N bit is set.
721
722 // Canonicalize illegal integer setcc's.
723 if (IsInteger && Op == ISD::SETUNE) // e.g. SETUGT | SETULT
724 Op = ISD::SETNE;
725
726 return ISD::CondCode(Op);
727}
728
730 EVT Type) {
731 bool IsInteger = Type.isInteger();
732 if (IsInteger && (isSignedOp(Op1) | isSignedOp(Op2)) == 3)
733 // Cannot fold a signed setcc with an unsigned setcc.
734 return ISD::SETCC_INVALID;
735
736 // Combine all of the condition bits.
737 ISD::CondCode Result = ISD::CondCode(Op1 & Op2);
738
739 // Canonicalize illegal integer setcc's.
740 if (IsInteger) {
741 switch (Result) {
742 default: break;
743 case ISD::SETUO : Result = ISD::SETFALSE; break; // SETUGT & SETULT
744 case ISD::SETOEQ: // SETEQ & SETU[LG]E
745 case ISD::SETUEQ: Result = ISD::SETEQ ; break; // SETUGE & SETULE
746 case ISD::SETOLT: Result = ISD::SETULT ; break; // SETULT & SETNE
747 case ISD::SETOGT: Result = ISD::SETUGT ; break; // SETUGT & SETNE
748 }
749 }
750
751 return Result;
752}
753
754//===----------------------------------------------------------------------===//
755// SDNode Key Support
756//===----------------------------------------------------------------------===//
757
758/// If this is an SDNode with special info, add this info to the NodeID data.
759/// MorphNodeTo passes the opcode \p N is becoming, so \p Opc may differ from
760/// N->getOpcode() and a case may only cast to a class \p N already is.
761static void AddNodeIDCustom(FoldingSetNodeID &ID, const SDNode *N,
762 unsigned Opc) {
763 // Machine nodes are uniqued by opcode, value types and operands alone, as in
764 // getMachineNode; MorphNodeTo's clearMemRefs overwrites the fields below.
766 return;
767
768 switch (Opc) {
771 case ISD::MCSymbol:
772 llvm_unreachable("Should only be used on nodes with operands");
773 default: break; // Normal nodes don't need extra info.
775 case ISD::Constant: {
777 ID.AddPointer(C->getConstantIntValue());
778 ID.AddBoolean(C->isOpaque());
779 break;
780 }
782 case ISD::ConstantFP:
783 ID.AddPointer(cast<ConstantFPSDNode>(N)->getConstantFPValue());
784 break;
790 ID.AddPointer(GA->getGlobal());
791 ID.AddInteger(GA->getOffset());
792 ID.AddInteger(GA->getTargetFlags());
793 break;
794 }
795 case ISD::BasicBlock:
796 ID.AddPointer(cast<BasicBlockSDNode>(N)->getBasicBlock());
797 break;
798 case ISD::EH_LABEL:
800 ID.AddPointer(cast<LabelSDNode>(N)->getLabel());
801 break;
803 ID.AddPointer(cast<DeactivationSymbolSDNode>(N)->getGlobal());
804 break;
805 case ISD::Register:
806 ID.AddInteger(cast<RegisterSDNode>(N)->getReg().id());
807 break;
809 ID.AddPointer(cast<RegisterMaskSDNode>(N)->getRegMask());
810 break;
811 case ISD::SRCVALUE:
812 ID.AddPointer(cast<SrcValueSDNode>(N)->getValue());
813 break;
814 case ISD::FrameIndex:
816 ID.AddInteger(cast<FrameIndexSDNode>(N)->getIndex());
817 break;
819 ID.AddInteger(cast<PseudoProbeSDNode>(N)->getGuid());
820 ID.AddInteger(cast<PseudoProbeSDNode>(N)->getIndex());
821 ID.AddInteger(cast<PseudoProbeSDNode>(N)->getAttributes());
822 break;
823 case ISD::JumpTable:
825 ID.AddInteger(cast<JumpTableSDNode>(N)->getIndex());
826 ID.AddInteger(cast<JumpTableSDNode>(N)->getTargetFlags());
827 break;
831 ID.AddInteger(CP->getAlign().value());
832 ID.AddInteger(CP->getOffset());
835 else
836 ID.AddPointer(CP->getConstVal());
837 ID.AddInteger(CP->getTargetFlags());
838 break;
839 }
840 case ISD::TargetIndex: {
842 ID.AddInteger(TI->getIndex());
843 ID.AddInteger(TI->getOffset());
844 ID.AddInteger(TI->getTargetFlags());
845 break;
846 }
847 case ISD::LOAD: {
848 const LoadSDNode *LD = cast<LoadSDNode>(N);
849 ID.AddInteger(LD->getMemoryVT().getRawBits());
850 ID.AddInteger(LD->getRawSubclassData());
851 ID.AddInteger(LD->getPointerInfo().getAddrSpace());
852 ID.AddInteger(LD->getMemOperand()->getFlags());
853 break;
854 }
855 case ISD::STORE: {
856 const StoreSDNode *ST = cast<StoreSDNode>(N);
857 ID.AddInteger(ST->getMemoryVT().getRawBits());
858 ID.AddInteger(ST->getRawSubclassData());
859 ID.AddInteger(ST->getPointerInfo().getAddrSpace());
860 ID.AddInteger(ST->getMemOperand()->getFlags());
861 break;
862 }
863 case ISD::VP_LOAD: {
864 const VPLoadSDNode *ELD = cast<VPLoadSDNode>(N);
865 ID.AddInteger(ELD->getMemoryVT().getRawBits());
866 ID.AddInteger(ELD->getRawSubclassData());
867 ID.AddInteger(ELD->getPointerInfo().getAddrSpace());
868 ID.AddInteger(ELD->getMemOperand()->getFlags());
869 break;
870 }
871 case ISD::VP_LOAD_FF: {
872 const auto *LD = cast<VPLoadFFSDNode>(N);
873 ID.AddInteger(LD->getMemoryVT().getRawBits());
874 ID.AddInteger(LD->getRawSubclassData());
875 ID.AddInteger(LD->getPointerInfo().getAddrSpace());
876 ID.AddInteger(LD->getMemOperand()->getFlags());
877 break;
878 }
879 case ISD::VP_STORE: {
880 const VPStoreSDNode *EST = cast<VPStoreSDNode>(N);
881 ID.AddInteger(EST->getMemoryVT().getRawBits());
882 ID.AddInteger(EST->getRawSubclassData());
883 ID.AddInteger(EST->getPointerInfo().getAddrSpace());
884 ID.AddInteger(EST->getMemOperand()->getFlags());
885 break;
886 }
887 case ISD::EXPERIMENTAL_VP_STRIDED_LOAD: {
889 ID.AddInteger(SLD->getMemoryVT().getRawBits());
890 ID.AddInteger(SLD->getRawSubclassData());
891 ID.AddInteger(SLD->getPointerInfo().getAddrSpace());
892 break;
893 }
894 case ISD::EXPERIMENTAL_VP_STRIDED_STORE: {
896 ID.AddInteger(SST->getMemoryVT().getRawBits());
897 ID.AddInteger(SST->getRawSubclassData());
898 ID.AddInteger(SST->getPointerInfo().getAddrSpace());
899 break;
900 }
901 case ISD::VP_GATHER: {
903 ID.AddInteger(EG->getMemoryVT().getRawBits());
904 ID.AddInteger(EG->getRawSubclassData());
905 ID.AddInteger(EG->getPointerInfo().getAddrSpace());
906 ID.AddInteger(EG->getMemOperand()->getFlags());
907 break;
908 }
909 case ISD::VP_SCATTER: {
911 ID.AddInteger(ES->getMemoryVT().getRawBits());
912 ID.AddInteger(ES->getRawSubclassData());
913 ID.AddInteger(ES->getPointerInfo().getAddrSpace());
914 ID.AddInteger(ES->getMemOperand()->getFlags());
915 break;
916 }
917 case ISD::MLOAD: {
919 ID.AddInteger(MLD->getMemoryVT().getRawBits());
920 ID.AddInteger(MLD->getRawSubclassData());
921 ID.AddInteger(MLD->getPointerInfo().getAddrSpace());
922 ID.AddInteger(MLD->getMemOperand()->getFlags());
923 break;
924 }
925 case ISD::MSTORE: {
927 ID.AddInteger(MST->getMemoryVT().getRawBits());
928 ID.AddInteger(MST->getRawSubclassData());
929 ID.AddInteger(MST->getPointerInfo().getAddrSpace());
930 ID.AddInteger(MST->getMemOperand()->getFlags());
931 break;
932 }
933 case ISD::MGATHER: {
935 ID.AddInteger(MG->getMemoryVT().getRawBits());
936 ID.AddInteger(MG->getRawSubclassData());
937 ID.AddInteger(MG->getPointerInfo().getAddrSpace());
938 ID.AddInteger(MG->getMemOperand()->getFlags());
939 break;
940 }
941 case ISD::MSCATTER:
945 const MemSDNode *M = cast<MemSDNode>(N);
946 ID.AddInteger(M->getMemoryVT().getRawBits());
947 ID.AddInteger(M->getRawSubclassData());
948 ID.AddInteger(M->getPointerInfo().getAddrSpace());
949 ID.AddInteger(M->getMemOperand()->getFlags());
950 break;
951 }
952 case ISD::VECTOR_SHUFFLE: {
953 ArrayRef<int> Mask = cast<ShuffleVectorSDNode>(N)->getMask();
954 for (int M : Mask)
955 ID.AddInteger(M);
956 break;
957 }
958 case ISD::ADDRSPACECAST: {
960 ID.AddInteger(ASC->getSrcAddressSpace());
961 ID.AddInteger(ASC->getDestAddressSpace());
962 break;
963 }
965 case ISD::BlockAddress: {
967 ID.AddPointer(BA->getBlockAddress());
968 ID.AddInteger(BA->getOffset());
969 ID.AddInteger(BA->getTargetFlags());
970 break;
971 }
972 case ISD::AssertAlign:
973 ID.AddInteger(cast<AssertAlignSDNode>(N)->getAlign().value());
974 break;
975 case ISD::PREFETCH:
978 // Handled by MemIntrinsicSDNode check after the switch.
979 break;
981 ID.AddPointer(cast<MDNodeSDNode>(N)->getMD());
982 break;
983 } // end switch (Opc)
984
985 if (auto *AT = dyn_cast<AtomicSDNode>(N)) {
986 ID.AddInteger(AT->getMemoryVT().getRawBits());
987 ID.AddInteger(AT->getRawSubclassData());
988 ID.AddInteger(AT->getPointerInfo().getAddrSpace());
989 ID.AddInteger(AT->getMemOperand()->getFlags());
990 }
991
992 // MemIntrinsic nodes could also have subclass data, address spaces, and flags
993 // to check.
994 if (auto *MN = dyn_cast<MemIntrinsicSDNode>(N)) {
995 ID.AddInteger(MN->getRawSubclassData());
996 ID.AddInteger(MN->getMemoryVT().getRawBits());
997 for (const MachineMemOperand *MMO : MN->memoperands()) {
998 ID.AddInteger(MMO->getPointerInfo().getAddrSpace());
999 ID.AddInteger(MMO->getFlags());
1000 }
1001 }
1002}
1003
1005 : Opcode(N.getOpcode()), VTs(N.getVTList().VTs),
1006 OpStorage(N.op_begin(), N.op_end()) {
1007 Ops = OpStorage;
1008 AddNodeIDCustom(Tail, &N, N.getOpcode());
1009}
1010
1011#ifndef NDEBUG
1012/// Serialize a node the way SDNodes were uniqued before SDNodeKey, to
1013/// cross-check the typed comparison against. Deliberately reads the node
1014/// rather than going through SDNodeKey, so the key constructor is checked too
1015/// and not used as its own oracle.
1016static void AddNodeIDNode(FoldingSetNodeID &ID, const SDNode *N) {
1017 ID.AddInteger(N->getOpcode());
1018 ID.AddPointer(N->getVTList().VTs);
1019 for (const SDUse &Op : N->ops()) {
1020 ID.AddPointer(Op.getNode());
1021 ID.AddInteger(Op.getResNo());
1022 }
1023 AddNodeIDCustom(ID, N, N->getOpcode());
1024}
1025
1027 ID.AddInteger(Key.Opcode);
1028 ID.AddPointer(Key.VTs);
1029 for (const SDValue &Op : Key.Ops) {
1030 ID.AddPointer(Op.getNode());
1031 ID.AddInteger(Op.getResNo());
1032 }
1033 ID.AddNodeID(Key.Tail);
1034}
1035#endif
1036
1037static bool keyMatches(const SDNodeKey &Key, const SDNode &N) {
1038 if (N.getOpcode() != Key.Opcode || N.getVTList().VTs != Key.VTs)
1039 return false;
1040 if (!llvm::equal(N.ops(), Key.Ops))
1041 return false;
1042 // Build only what AddNodeIDCustom would have added, which is nothing for
1043 // most opcodes, rather than the whole profile.
1044 FoldingSetNodeID Tail;
1045 AddNodeIDCustom(Tail, &N, N.getOpcode());
1046 return Tail == Key.Tail;
1047}
1048
1050 bool Result = keyMatches(Key, N);
1051
1052#ifndef NDEBUG
1053 FoldingSetNodeID NodeProfile, KeyProfile;
1054 AddNodeIDNode(NodeProfile, &N);
1055 AddNodeIDNode(KeyProfile, Key);
1056 assert((NodeProfile == KeyProfile) == Result &&
1057 "SDNodeKey equality disagrees with profile");
1058#endif
1059 return Result;
1060}
1061
1062//===----------------------------------------------------------------------===//
1063// SelectionDAG Class
1064//===----------------------------------------------------------------------===//
1065
1066/// doNotCSE - Return true if CSE should not be performed for this node.
1067static bool doNotCSE(SDNode *N) {
1068 if (N->getValueType(0) == MVT::Glue)
1069 return true; // Never CSE anything that produces a glue result.
1070
1071 switch (N->getOpcode()) {
1072 default: break;
1073 case ISD::HANDLENODE:
1074 case ISD::EH_LABEL:
1075 return true; // Never CSE these nodes.
1076 }
1077
1078 // Check that remaining values produced are not flags.
1079 for (unsigned i = 1, e = N->getNumValues(); i != e; ++i)
1080 if (N->getValueType(i) == MVT::Glue)
1081 return true; // Never CSE anything that produces a glue result.
1082
1083 return false;
1084}
1085
1086/// Construct a DemandedElts mask which demands all elements of \p V.
1087/// If \p V is not a fixed-length vector, then this will return a single bit.
1089 EVT VT = V.getValueType();
1090 // Since the number of lanes in a scalable vector is unknown at compile time,
1091 // we track one bit which is implicitly broadcast to all lanes. This means
1092 // that all lanes in a scalable vector are considered demanded.
1094 : APInt(1, 1);
1095}
1096
1097/// RemoveDeadNodes - This method deletes all unreachable nodes in the
1098/// SelectionDAG.
1100 // Create a dummy node (which is not added to allnodes), that adds a reference
1101 // to the root node, preventing it from being deleted.
1102 HandleSDNode Dummy(getRoot());
1103
1104 SmallVector<SDNode*, 128> DeadNodes;
1105
1106 // Add all obviously-dead nodes to the DeadNodes worklist.
1107 for (SDNode &Node : allnodes())
1108 if (Node.use_empty())
1109 DeadNodes.push_back(&Node);
1110
1111 RemoveDeadNodes(DeadNodes);
1112
1113 // If the root changed (e.g. it was a dead load, update the root).
1114 setRoot(Dummy.getValue());
1115}
1116
1117/// RemoveDeadNodes - This method deletes the unreachable nodes in the
1118/// given list, and any nodes that become unreachable as a result.
1120
1121 // Process the worklist, deleting the nodes and adding their uses to the
1122 // worklist.
1123 while (!DeadNodes.empty()) {
1124 SDNode *N = DeadNodes.pop_back_val();
1125 // Skip to next node if we've already managed to delete the node. This could
1126 // happen if replacing a node causes a node previously added to the node to
1127 // be deleted.
1128 if (N->getOpcode() == ISD::DELETED_NODE)
1129 continue;
1130
1131 for (DAGUpdateListener *DUL = UpdateListeners; DUL; DUL = DUL->Next)
1132 DUL->NodeDeleted(N, nullptr);
1133
1134 // Take the node out of the appropriate CSE map.
1135 RemoveNodeFromCSEMaps(N);
1136
1137 // Next, brutally remove the operand list. This is safe to do, as there are
1138 // no cycles in the graph.
1139 for (SDNode::op_iterator I = N->op_begin(), E = N->op_end(); I != E; ) {
1140 SDUse &Use = *I++;
1141 SDNode *Operand = Use.getNode();
1142 Use.set(SDValue());
1143
1144 // Now that we removed this operand, see if there are no uses of it left.
1145 if (Operand->use_empty())
1146 DeadNodes.push_back(Operand);
1147 }
1148
1149 DeallocateNode(N);
1150 }
1151}
1152
1154 SmallVector<SDNode*, 16> DeadNodes(1, N);
1155
1156 // Create a dummy node that adds a reference to the root node, preventing
1157 // it from being deleted. (This matters if the root is an operand of the
1158 // dead node.)
1159 HandleSDNode Dummy(getRoot());
1160
1161 RemoveDeadNodes(DeadNodes);
1162}
1163
1165 // First take this out of the appropriate CSE map.
1166 RemoveNodeFromCSEMaps(N);
1167
1168 // Finally, remove uses due to operands of this node, remove from the
1169 // AllNodes list, and delete the node.
1170 DeleteNodeNotInCSEMaps(N);
1171}
1172
1173void SelectionDAG::DeleteNodeNotInCSEMaps(SDNode *N) {
1174 assert(N->getIterator() != AllNodes.begin() &&
1175 "Cannot delete the entry node!");
1176 assert(N->use_empty() && "Cannot delete a node that is not dead!");
1177
1178 // Drop all of the operands and decrement used node's use counts.
1179 N->DropOperands();
1180
1181 DeallocateNode(N);
1182}
1183
1184void SDDbgInfo::add(SDDbgValue *V, bool isParameter) {
1185 assert(!(V->isVariadic() && isParameter));
1186 if (isParameter)
1187 ByvalParmDbgValues.push_back(V);
1188 else
1189 DbgValues.push_back(V);
1190 for (const SDNode *Node : V->getSDNodes())
1191 if (Node)
1192 DbgValMap[Node].push_back(V);
1193}
1194
1196 DbgValMapType::iterator I = DbgValMap.find(Node);
1197 if (I == DbgValMap.end())
1198 return;
1199 for (auto &Val: I->second)
1200 Val->setIsInvalidated();
1201 DbgValMap.erase(I);
1202}
1203
1204void SelectionDAG::DeallocateNode(SDNode *N) {
1205 // If we have operands, deallocate them.
1207
1208 NodeAllocator.Deallocate(AllNodes.remove(N));
1209
1210 // Set the opcode to DELETED_NODE to help catch bugs when node
1211 // memory is reallocated.
1212 // FIXME: There are places in SDag that have grown a dependency on the opcode
1213 // value in the released node.
1214 __asan_unpoison_memory_region(&N->NodeType, sizeof(N->NodeType));
1215 N->NodeType = ISD::DELETED_NODE;
1216
1217 // If any of the SDDbgValue nodes refer to this SDNode, invalidate
1218 // them and forget about that node.
1219 DbgInfo->erase(N);
1220
1221 // Invalidate extra info.
1222 SDEI.erase(N);
1223}
1224
1225#ifndef NDEBUG
1226/// VerifySDNode - Check the given SDNode. Aborts if it is invalid.
1227void SelectionDAG::verifyNode(SDNode *N) const {
1228 switch (N->getOpcode()) {
1229 default:
1230 if (N->isTargetOpcode())
1232 break;
1233 case ISD::BUILD_PAIR: {
1234 EVT VT = N->getValueType(0);
1235 assert(N->getNumValues() == 1 && "Too many results!");
1236 assert(!VT.isVector() && (VT.isInteger() || VT.isFloatingPoint()) &&
1237 "Wrong return type!");
1238 assert(N->getNumOperands() == 2 && "Wrong number of operands!");
1239 assert(N->getOperand(0).getValueType() == N->getOperand(1).getValueType() &&
1240 "Mismatched operand types!");
1241 assert(N->getOperand(0).getValueType().isInteger() == VT.isInteger() &&
1242 "Wrong operand type!");
1243 assert(VT.getSizeInBits() == 2 * N->getOperand(0).getValueSizeInBits() &&
1244 "Wrong return type size");
1245 break;
1246 }
1247 case ISD::BUILD_VECTOR: {
1248 assert(N->getNumValues() == 1 && "Too many results!");
1249 assert(N->getValueType(0).isVector() && "Wrong return type!");
1250 assert(N->getNumOperands() == N->getValueType(0).getVectorNumElements() &&
1251 "Wrong number of operands!");
1252 EVT EltVT = N->getValueType(0).getVectorElementType();
1253 for (const SDUse &Op : N->ops()) {
1254 assert((Op.getValueType() == EltVT ||
1255 (EltVT.isInteger() && Op.getValueType().isInteger() &&
1256 EltVT.bitsLE(Op.getValueType()))) &&
1257 "Wrong operand type!");
1258 assert(Op.getValueType() == N->getOperand(0).getValueType() &&
1259 "Operands must all have the same type");
1260 }
1261 break;
1262 }
1263 case ISD::SADDO:
1264 case ISD::UADDO:
1265 case ISD::SSUBO:
1266 case ISD::USUBO:
1267 assert(N->getNumValues() == 2 && "Wrong number of results!");
1268 assert(N->getVTList().NumVTs == 2 && N->getNumOperands() == 2 &&
1269 "Invalid add/sub overflow op!");
1270 assert(N->getVTList().VTs[0].isInteger() &&
1271 N->getVTList().VTs[1].isInteger() &&
1272 N->getOperand(0).getValueType() == N->getOperand(1).getValueType() &&
1273 N->getOperand(0).getValueType() == N->getVTList().VTs[0] &&
1274 "Binary operator types must match!");
1275 break;
1276 }
1277}
1278#endif // NDEBUG
1279
1280/// Insert a newly allocated node into the DAG.
1281///
1282/// Handles insertion into the all nodes list and CSE map, as well as
1283/// verification and other common operations when a new node is allocated.
1284void SelectionDAG::InsertNode(SDNode *N) {
1285 AllNodes.push_back(N);
1286#ifndef NDEBUG
1287 N->PersistentId = NextPersistentId++;
1288 verifyNode(N);
1289#endif
1290 for (DAGUpdateListener *DUL = UpdateListeners; DUL; DUL = DUL->Next)
1291 DUL->NodeInserted(N);
1292}
1293
1294/// RemoveNodeFromCSEMaps - Take the specified node out of the CSE map that
1295/// correspond to it. This is useful when we're about to delete or repurpose
1296/// the node. We don't want future request for structurally identical nodes
1297/// to return N anymore.
1298bool SelectionDAG::RemoveNodeFromCSEMaps(SDNode *N) {
1299 bool Erased = false;
1300 switch (N->getOpcode()) {
1301 case ISD::HANDLENODE: return false; // noop.
1302 case ISD::CONDCODE:
1303 assert(CondCodeNodes[cast<CondCodeSDNode>(N)->get()] &&
1304 "Cond code doesn't exist!");
1305 Erased = CondCodeNodes[cast<CondCodeSDNode>(N)->get()] != nullptr;
1306 CondCodeNodes[cast<CondCodeSDNode>(N)->get()] = nullptr;
1307 break;
1309 Erased = ExternalSymbols.erase(cast<ExternalSymbolSDNode>(N)->getSymbol());
1310 break;
1312 ExternalSymbolSDNode *ESN = cast<ExternalSymbolSDNode>(N);
1313 Erased = TargetExternalSymbols.erase(std::pair<std::string, unsigned>(
1314 ESN->getSymbol(), ESN->getTargetFlags()));
1315 break;
1316 }
1317 case ISD::MCSymbol: {
1318 auto *MCSN = cast<MCSymbolSDNode>(N);
1319 Erased = MCSymbols.erase(MCSN->getMCSymbol());
1320 break;
1321 }
1322 case ISD::VALUETYPE: {
1323 EVT VT = cast<VTSDNode>(N)->getVT();
1324 if (VT.isExtended()) {
1325 Erased = ExtendedValueTypeNodes.erase(VT);
1326 } else {
1327 Erased = ValueTypeNodes[VT.getSimpleVT().SimpleTy] != nullptr;
1328 ValueTypeNodes[VT.getSimpleVT().SimpleTy] = nullptr;
1329 }
1330 break;
1331 }
1332 default:
1333 // Remove it from the CSE Map.
1334 assert(N->getOpcode() != ISD::DELETED_NODE && "DELETED_NODE in CSEMap!");
1335 assert(N->getOpcode() != ISD::EntryToken && "EntryToken in CSEMap!");
1336 Erased = CSEMap.erase(N);
1337 break;
1338 }
1339#ifndef NDEBUG
1340 // Verify that the node was actually in one of the CSE maps, unless it has a
1341 // glue result (which cannot be CSE'd) or is one of the special cases that are
1342 // not subject to CSE.
1343 if (!Erased && N->getValueType(N->getNumValues()-1) != MVT::Glue &&
1344 !N->isMachineOpcode() && !doNotCSE(N)) {
1345 N->dump(this);
1346 dbgs() << "\n";
1347 llvm_unreachable("Node is not in map!");
1348 }
1349#endif
1350 return Erased;
1351}
1352
1353/// AddModifiedNodeToCSEMaps - The specified node has been removed from the CSE
1354/// maps and modified in place. Add it back to the CSE maps, unless an identical
1355/// node already exists, in which case transfer all its users to the existing
1356/// node. This transfer can potentially trigger recursive merging.
1357void
1358SelectionDAG::AddModifiedNodeToCSEMaps(SDNode *N) {
1359 // For node types that aren't CSE'd, just act as if no identical node
1360 // already exists.
1361 if (!doNotCSE(N)) {
1362 SDNode *Existing = CSEMap.getOrInsert(N);
1363 if (Existing != N) {
1364 // If there was already an existing matching node, use ReplaceAllUsesWith
1365 // to replace the dead one with the existing one. This can cause
1366 // recursive merging of other unrelated nodes down the line.
1367 Existing->intersectFlagsWith(N->getFlags());
1368 if (auto *MemNode = dyn_cast<MemSDNode>(Existing)) {
1370 cast<MemSDNode>(N)->memoperands();
1371 // Range and cache hint metadata are not part of the DAG CSE key because
1372 // we prefer to CSE even when metadata does not match. Merge potentially
1373 // differing metadata conservatively.
1374 MemNode->refineMMOMetadata(NewMMOs);
1375 }
1376 ReplaceAllUsesWith(N, Existing);
1377
1378 // N is now dead. Inform the listeners and delete it.
1379 for (DAGUpdateListener *DUL = UpdateListeners; DUL; DUL = DUL->Next)
1380 DUL->NodeDeleted(N, Existing);
1381 DeleteNodeNotInCSEMaps(N);
1382 return;
1383 }
1384 }
1385
1386 // If the node doesn't already exist, we updated it. Inform listeners.
1387 for (DAGUpdateListener *DUL = UpdateListeners; DUL; DUL = DUL->Next)
1388 DUL->NodeUpdated(N);
1389}
1390
1391/// FindModifiedNodeSlot - Find a slot for the specified node if its operands
1392/// were replaced with those specified. If this node is never memoized,
1393/// return null, otherwise return a pointer to the slot it would take. If a
1394/// node already exists with these operands, the slot will be non-null.
1395SDNode *SelectionDAG::FindModifiedNodeSlot(SDNode *N, SDValue Op,
1396 FoldingSetInsertToken &InsertToken) {
1397 if (doNotCSE(N))
1398 return nullptr;
1399
1400 SDValue Ops[] = { Op };
1401 SDNodeKey ID(N->getOpcode(), N->getVTList(), Ops);
1402 AddNodeIDCustom(ID.Tail, N, N->getOpcode());
1403 SDNode *Node = lookupNode(ID, SDLoc(N), InsertToken);
1404 if (Node)
1405 Node->intersectFlagsWith(N->getFlags());
1406 return Node;
1407}
1408
1409/// FindModifiedNodeSlot - Find a slot for the specified node if its operands
1410/// were replaced with those specified. If this node is never memoized,
1411/// return null, otherwise return a pointer to the slot it would take. If a
1412/// node already exists with these operands, the slot will be non-null.
1413SDNode *SelectionDAG::FindModifiedNodeSlot(SDNode *N, SDValue Op1, SDValue Op2,
1414 FoldingSetInsertToken &InsertToken) {
1415 if (doNotCSE(N))
1416 return nullptr;
1417
1418 SDValue Ops[] = { Op1, Op2 };
1419 SDNodeKey ID(N->getOpcode(), N->getVTList(), Ops);
1420 AddNodeIDCustom(ID.Tail, N, N->getOpcode());
1421 SDNode *Node = lookupNode(ID, SDLoc(N), InsertToken);
1422 if (Node)
1423 Node->intersectFlagsWith(N->getFlags());
1424 return Node;
1425}
1426
1427/// FindModifiedNodeSlot - Find a slot for the specified node if its operands
1428/// were replaced with those specified. If this node is never memoized,
1429/// return null, otherwise return a pointer to the slot it would take. If a
1430/// node already exists with these operands, the slot will be non-null.
1431SDNode *SelectionDAG::FindModifiedNodeSlot(SDNode *N, ArrayRef<SDValue> Ops,
1432 FoldingSetInsertToken &InsertToken) {
1433 if (doNotCSE(N))
1434 return nullptr;
1435
1436 SDNodeKey ID(N->getOpcode(), N->getVTList(), Ops);
1437 AddNodeIDCustom(ID.Tail, N, N->getOpcode());
1438 SDNode *Node = lookupNode(ID, SDLoc(N), InsertToken);
1439 if (Node)
1440 Node->intersectFlagsWith(N->getFlags());
1441 return Node;
1442}
1443
1445 Type *Ty = VT == MVT::iPTR ? PointerType::get(*getContext(), 0)
1446 : VT.getTypeForEVT(*getContext());
1447
1448 return getDataLayout().getABITypeAlign(Ty);
1449}
1450
1451// EntryNode could meaningfully have debug info if we can find it...
1453 : TM(tm), OptLevel(OL), EntryNode(ISD::EntryToken, 0, DebugLoc(),
1454 getVTList(MVT::Other, MVT::Glue)),
1455 Root(getEntryNode()) {
1456 InsertNode(&EntryNode);
1457 DbgInfo = new SDDbgInfo();
1458}
1459
1461 const TargetLibraryInfo *LibraryInfo,
1462 const LibcallLoweringInfo *LibcallsInfo,
1463 UniformityInfo *NewUA, ProfileSummaryInfo *PSIin,
1464 BlockFrequencyInfo *BFIin,
1465 FunctionVarLocs const *VarLocs) {
1466 MF = &NewMF;
1469 LibInfo = LibraryInfo;
1470 Libcalls = LibcallsInfo;
1471 Context = &MF->getFunction().getContext();
1472 UA = NewUA;
1473 PSI = PSIin;
1474 BFI = BFIin;
1475 FnVarLocs = VarLocs;
1476}
1477
1479 assert(!UpdateListeners && "Dangling registered DAGUpdateListeners");
1480 allnodes_clear();
1481 OperandRecycler.clear(OperandAllocator);
1482 delete DbgInfo;
1483}
1484
1486 return llvm::shouldOptimizeForSize(FLI->MBB->getBasicBlock(), PSI, BFI);
1487}
1488
1489void SelectionDAG::allnodes_clear() {
1490 assert(&*AllNodes.begin() == &EntryNode);
1491 AllNodes.remove(AllNodes.begin());
1492 while (!AllNodes.empty())
1493 DeallocateNode(&AllNodes.front());
1494#ifndef NDEBUG
1495 NextPersistentId = 0;
1496#endif
1497}
1498
1499SDNode *SelectionDAG::lookupNode(const SDNodeKey &Key,
1500 FoldingSetInsertToken &InsertToken) {
1501 SDNode *N = CSEMap.lookup(Key, InsertToken);
1502 if (N) {
1503 switch (N->getOpcode()) {
1504 default: break;
1505 case ISD::Constant:
1506 case ISD::ConstantFP:
1507 llvm_unreachable("Querying for Constant and ConstantFP nodes requires "
1508 "debug location. Use another overload.");
1509 }
1510 }
1511 return N;
1512}
1513
1514SDNode *SelectionDAG::lookupNode(const SDNodeKey &Key, const SDLoc &DL,
1515 FoldingSetInsertToken &InsertToken) {
1516 SDNode *N = CSEMap.lookup(Key, InsertToken);
1517 if (N) {
1518 switch (N->getOpcode()) {
1519 case ISD::Constant:
1520 case ISD::ConstantFP:
1521 // Erase debug location from the node if the node is used at several
1522 // different places. Do not propagate one location to all uses as it
1523 // will cause a worse single stepping debugging experience.
1524 if (N->getDebugLoc() != DL.getDebugLoc())
1525 N->setDebugLoc(DebugLoc());
1526 break;
1527 default:
1528 // When the node's point of use is located earlier in the instruction
1529 // sequence than its prior point of use, update its debug info to the
1530 // earlier location.
1531 if (DL.getIROrder() && DL.getIROrder() < N->getIROrder())
1532 N->setDebugLoc(DL.getDebugLoc());
1533 break;
1534 }
1535 }
1536 return N;
1537}
1538
1540 allnodes_clear();
1541 OperandRecycler.clear(OperandAllocator);
1542 OperandAllocator.Reset();
1543 CSEMap.clear();
1544
1545 ExtendedValueTypeNodes.clear();
1546 ExternalSymbols.clear();
1547 TargetExternalSymbols.clear();
1548 MCSymbols.clear();
1549 SDEI.clear();
1550 llvm::fill(CondCodeNodes, nullptr);
1551 llvm::fill(ValueTypeNodes, nullptr);
1552
1553 EntryNode.UseList = nullptr;
1554 InsertNode(&EntryNode);
1555 Root = getEntryNode();
1556 DbgInfo->clear();
1557}
1558
1560 return VT.bitsGT(Op.getValueType())
1561 ? getNode(ISD::FP_EXTEND, DL, VT, Op)
1562 : getNode(ISD::FP_ROUND, DL, VT, Op,
1563 getIntPtrConstant(0, DL, /*isTarget=*/true));
1564}
1565
1566std::pair<SDValue, SDValue>
1568 const SDLoc &DL, EVT VT) {
1569 assert(!VT.bitsEq(Op.getValueType()) &&
1570 "Strict no-op FP extend/round not allowed.");
1571 SDValue Res =
1572 VT.bitsGT(Op.getValueType())
1573 ? getNode(ISD::STRICT_FP_EXTEND, DL, {VT, MVT::Other}, {Chain, Op})
1574 : getNode(ISD::STRICT_FP_ROUND, DL, {VT, MVT::Other},
1575 {Chain, Op, getIntPtrConstant(0, DL, /*isTarget=*/true)});
1576
1577 return std::pair<SDValue, SDValue>(Res, SDValue(Res.getNode(), 1));
1578}
1579
1581 return VT.bitsGT(Op.getValueType()) ?
1582 getNode(ISD::ANY_EXTEND, DL, VT, Op) :
1583 getNode(ISD::TRUNCATE, DL, VT, Op);
1584}
1585
1587 return VT.bitsGT(Op.getValueType()) ?
1588 getNode(ISD::SIGN_EXTEND, DL, VT, Op) :
1589 getNode(ISD::TRUNCATE, DL, VT, Op);
1590}
1591
1593 return VT.bitsGT(Op.getValueType()) ?
1594 getNode(ISD::ZERO_EXTEND, DL, VT, Op) :
1595 getNode(ISD::TRUNCATE, DL, VT, Op);
1596}
1597
1599 EVT VT) {
1600 assert(!VT.isVector());
1601 auto Type = Op.getValueType();
1602 SDValue DestOp;
1603 if (Type == VT)
1604 return Op;
1605 auto Size = Op.getValueSizeInBits();
1606 DestOp = getBitcast(EVT::getIntegerVT(*Context, Size), Op);
1607 if (DestOp.getValueType() == VT)
1608 return DestOp;
1609
1610 return getAnyExtOrTrunc(DestOp, DL, VT);
1611}
1612
1614 EVT OpVT) {
1615 if (VT.bitsLE(Op.getValueType()))
1616 return getNode(ISD::TRUNCATE, SL, VT, Op);
1617
1618 TargetLowering::BooleanContent BType = TLI->getBooleanContents(OpVT);
1619 return getNode(TLI->getExtendForContent(BType), SL, VT, Op);
1620}
1621
1623 EVT OpVT = Op.getValueType();
1624 assert(VT.isInteger() && OpVT.isInteger() &&
1625 "Cannot getZeroExtendInReg FP types");
1626 assert(VT.isVector() == OpVT.isVector() &&
1627 "getZeroExtendInReg type should be vector iff the operand "
1628 "type is vector!");
1629 assert((!VT.isVector() ||
1631 "Vector element counts must match in getZeroExtendInReg");
1632 assert(VT.getScalarType().bitsLE(OpVT.getScalarType()) && "Not extending!");
1633 if (OpVT == VT)
1634 return Op;
1635 // TODO: Use computeKnownBits instead of AssertZext.
1636 if (Op.getOpcode() == ISD::AssertZext && cast<VTSDNode>(Op.getOperand(1))
1637 ->getVT()
1638 .getScalarType()
1639 .bitsLE(VT.getScalarType()))
1640 return Op;
1642 VT.getScalarSizeInBits());
1643 return getNode(ISD::AND, DL, OpVT, Op, getConstant(Imm, DL, OpVT));
1644}
1645
1647 // Only unsigned pointer semantics are supported right now. In the future this
1648 // might delegate to TLI to check pointer signedness.
1649 return getZExtOrTrunc(Op, DL, VT);
1650}
1651
1653 // Only unsigned pointer semantics are supported right now. In the future this
1654 // might delegate to TLI to check pointer signedness.
1655 return getZeroExtendInReg(Op, DL, VT);
1656}
1657
1659 return getNode(ISD::SUB, DL, VT, getConstant(0, DL, VT), Val);
1660}
1661
1662/// getNOT - Create a bitwise NOT operation as (XOR Val, -1).
1664 return getNode(ISD::XOR, DL, VT, Val, getAllOnesConstant(DL, VT));
1665}
1666
1668 SDValue TrueValue = getBoolConstant(true, DL, VT, VT);
1669 return getNode(ISD::XOR, DL, VT, Val, TrueValue);
1670}
1671
1673 EVT OpVT) {
1674 if (!V)
1675 return getConstant(0, DL, VT);
1676
1677 switch (TLI->getBooleanContents(OpVT)) {
1680 return getConstant(1, DL, VT);
1682 return getAllOnesConstant(DL, VT);
1683 }
1684 llvm_unreachable("Unexpected boolean content enum!");
1685}
1686
1688 bool isT, bool isO) {
1689 return getConstant(APInt(VT.getScalarSizeInBits(), Val, /*isSigned=*/false),
1690 DL, VT, isT, isO);
1691}
1692
1694 bool isT, bool isO) {
1695 return getConstant(*ConstantInt::get(*Context, Val), DL, VT, isT, isO);
1696}
1697
1699 EVT VT, bool isT, bool isO) {
1700 assert(VT.isInteger() && "Cannot create FP integer constant!");
1701
1702 EVT EltVT = VT.getScalarType();
1703 const ConstantInt *Elt = &Val;
1704
1705 // Vector splats are explicit within the DAG, with ConstantSDNode holding the
1706 // to-be-splatted scalar ConstantInt.
1707 if (isa<VectorType>(Elt->getType()))
1708 Elt = ConstantInt::get(*getContext(), Elt->getValue());
1709
1710 // In some cases the vector type is legal but the element type is illegal and
1711 // needs to be promoted, for example v8i8 on ARM. In this case, promote the
1712 // inserted value (the type does not need to match the vector element type).
1713 // Any extra bits introduced will be truncated away.
1714 if (VT.isVector() && TLI->getTypeAction(*getContext(), EltVT) ==
1716 EltVT = TLI->getTypeToTransformTo(*getContext(), EltVT);
1717 APInt NewVal;
1718 if (TLI->isSExtCheaperThanZExt(VT.getScalarType(), EltVT))
1719 NewVal = Elt->getValue().sextOrTrunc(EltVT.getSizeInBits());
1720 else
1721 NewVal = Elt->getValue().zextOrTrunc(EltVT.getSizeInBits());
1722 Elt = ConstantInt::get(*getContext(), NewVal);
1723 }
1724 // In other cases the element type is illegal and needs to be expanded, for
1725 // example v2i64 on MIPS32. In this case, find the nearest legal type, split
1726 // the value into n parts and use a vector type with n-times the elements.
1727 // Then bitcast to the type requested.
1728 // Legalizing constants too early makes the DAGCombiner's job harder so we
1729 // only legalize if the DAG tells us we must produce legal types.
1730 else if (NewNodesMustHaveLegalTypes && VT.isVector() &&
1731 TLI->getTypeAction(*getContext(), EltVT) ==
1733 const APInt &NewVal = Elt->getValue();
1734 EVT ViaEltVT = TLI->getTypeToTransformTo(*getContext(), EltVT);
1735 unsigned ViaEltSizeInBits = ViaEltVT.getSizeInBits();
1736
1737 // For scalable vectors, try to use a SPLAT_VECTOR_PARTS node.
1738 if (VT.isScalableVector() ||
1739 TLI->isOperationLegal(ISD::SPLAT_VECTOR, VT)) {
1740 assert(EltVT.getSizeInBits() % ViaEltSizeInBits == 0 &&
1741 "Can only handle an even split!");
1742 unsigned Parts = EltVT.getSizeInBits() / ViaEltSizeInBits;
1743
1744 SmallVector<SDValue, 2> ScalarParts;
1745 for (unsigned i = 0; i != Parts; ++i)
1746 ScalarParts.push_back(getConstant(
1747 NewVal.extractBits(ViaEltSizeInBits, i * ViaEltSizeInBits), DL,
1748 ViaEltVT, isT, isO));
1749
1750 return getNode(ISD::SPLAT_VECTOR_PARTS, DL, VT, ScalarParts);
1751 }
1752
1753 unsigned ViaVecNumElts = VT.getSizeInBits() / ViaEltSizeInBits;
1754 EVT ViaVecVT = EVT::getVectorVT(*getContext(), ViaEltVT, ViaVecNumElts);
1755
1756 // Check the temporary vector is the correct size. If this fails then
1757 // getTypeToTransformTo() probably returned a type whose size (in bits)
1758 // isn't a power-of-2 factor of the requested type size.
1759 assert(ViaVecVT.getSizeInBits() == VT.getSizeInBits());
1760
1761 SmallVector<SDValue, 2> EltParts;
1762 for (unsigned i = 0; i < ViaVecNumElts / VT.getVectorNumElements(); ++i)
1763 EltParts.push_back(getConstant(
1764 NewVal.extractBits(ViaEltSizeInBits, i * ViaEltSizeInBits), DL,
1765 ViaEltVT, isT, isO));
1766
1767 // EltParts is currently in little endian order. If we actually want
1768 // big-endian order then reverse it now.
1769 if (getDataLayout().isBigEndian())
1770 std::reverse(EltParts.begin(), EltParts.end());
1771
1772 // The elements must be reversed when the element order is different
1773 // to the endianness of the elements (because the BITCAST is itself a
1774 // vector shuffle in this situation). However, we do not need any code to
1775 // perform this reversal because getConstant() is producing a vector
1776 // splat.
1777 // This situation occurs in MIPS MSA.
1778
1780 for (unsigned i = 0, e = VT.getVectorNumElements(); i != e; ++i)
1781 llvm::append_range(Ops, EltParts);
1782
1783 SDValue V =
1784 getNode(ISD::BITCAST, DL, VT, getBuildVector(ViaVecVT, DL, Ops));
1785 return V;
1786 }
1787
1788 assert(Elt->getBitWidth() == EltVT.getSizeInBits() &&
1789 "APInt size does not match type size!");
1790 unsigned Opc = isT ? ISD::TargetConstant : ISD::Constant;
1791 SDVTList VTs = getVTList(EltVT);
1792 SDNodeKey ID(Opc, VTs, {});
1793 ID.AddPointer(Elt);
1794 ID.AddBoolean(isO);
1795 FoldingSetInsertToken InsertToken;
1796 SDNode *N = nullptr;
1797 if ((N = lookupNode(ID, DL, InsertToken)))
1798 if (!VT.isVector())
1799 return SDValue(N, 0);
1800
1801 if (!N) {
1802 N = newSDNode<ConstantSDNode>(isT, isO, Elt, VTs);
1803 if (!isT)
1804 N->setDebugLoc(DL.getDebugLoc());
1805 CSEMap.insert(N, InsertToken);
1806 InsertNode(N);
1807 NewSDValueDbgMsg(SDValue(N, 0), "Creating constant: ", this);
1808 }
1809
1810 SDValue Result(N, 0);
1811 if (VT.isVector())
1812 Result = getSplat(VT, DL, Result);
1813 return Result;
1814}
1815
1817 bool isT, bool isO) {
1818 unsigned Size = VT.getScalarSizeInBits();
1819 return getConstant(APInt(Size, Val, /*isSigned=*/true), DL, VT, isT, isO);
1820}
1821
1823 bool IsOpaque) {
1825 IsTarget, IsOpaque);
1826}
1827
1829 bool isTarget) {
1830 return getConstant(Val, DL, TLI->getPointerTy(getDataLayout()), isTarget);
1831}
1832
1834 const SDLoc &DL) {
1835 assert(VT.isInteger() && "Shift amount is not an integer type!");
1836 EVT ShiftVT = TLI->getShiftAmountTy(VT, getDataLayout());
1837 return getConstant(Val, DL, ShiftVT);
1838}
1839
1841 const SDLoc &DL) {
1842 assert(Val.ult(VT.getScalarSizeInBits()) && "Out of range shift");
1843 return getShiftAmountConstant(Val.getZExtValue(), VT, DL);
1844}
1845
1847 bool isTarget) {
1848 return getConstant(Val, DL, TLI->getVectorIdxTy(getDataLayout()), isTarget);
1849}
1850
1852 bool isTarget) {
1853 return getConstantFP(*ConstantFP::get(*getContext(), V), DL, VT, isTarget);
1854}
1855
1857 EVT VT, bool isTarget) {
1858 assert(VT.isFloatingPoint() && "Cannot create integer FP constant!");
1859
1860 EVT EltVT = VT.getScalarType();
1861 const ConstantFP *Elt = &V;
1862
1863 // Vector splats are explicit within the DAG, with ConstantFPSDNode holding
1864 // the to-be-splatted scalar ConstantFP.
1865 if (isa<VectorType>(Elt->getType()))
1866 Elt = ConstantFP::get(*getContext(), Elt->getValue());
1867
1868 // Do the map lookup using the actual bit pattern for the floating point
1869 // value, so that we don't have problems with 0.0 comparing equal to -0.0, and
1870 // we don't have issues with SNANs.
1871 unsigned Opc = isTarget ? ISD::TargetConstantFP : ISD::ConstantFP;
1872 SDVTList VTs = getVTList(EltVT);
1873 SDNodeKey ID(Opc, VTs, {});
1874 ID.AddPointer(Elt);
1875 FoldingSetInsertToken InsertToken;
1876 SDNode *N = nullptr;
1877 if ((N = lookupNode(ID, DL, InsertToken)))
1878 if (!VT.isVector())
1879 return SDValue(N, 0);
1880
1881 if (!N) {
1882 N = newSDNode<ConstantFPSDNode>(isTarget, Elt, VTs);
1883 CSEMap.insert(N, InsertToken);
1884 InsertNode(N);
1885 }
1886
1887 SDValue Result(N, 0);
1888 if (VT.isVector())
1889 Result = getSplat(VT, DL, Result);
1890 NewSDValueDbgMsg(Result, "Creating fp constant: ", this);
1891 return Result;
1892}
1893
1895 bool isTarget) {
1896 EVT EltVT = VT.getScalarType();
1897 if (EltVT == MVT::f32)
1898 return getConstantFP(APFloat((float)Val), DL, VT, isTarget);
1899 if (EltVT == MVT::f64)
1900 return getConstantFP(APFloat(Val), DL, VT, isTarget);
1901 if (EltVT == MVT::f80 || EltVT == MVT::f128 || EltVT == MVT::ppcf128 ||
1902 EltVT == MVT::f16 || EltVT == MVT::bf16) {
1903 bool Ignored;
1904 APFloat APF = APFloat(Val);
1906 &Ignored);
1907 return getConstantFP(APF, DL, VT, isTarget);
1908 }
1909 llvm_unreachable("Unsupported type in getConstantFP");
1910}
1911
1913 EVT VT, int64_t Offset, bool isTargetGA,
1914 unsigned TargetFlags) {
1915 assert((TargetFlags == 0 || isTargetGA) &&
1916 "Cannot set target flags on target-independent globals");
1917
1918 // Truncate (with sign-extension) the offset value to the pointer size.
1920 if (BitWidth < 64)
1922
1923 unsigned Opc;
1924 if (GV->isThreadLocal())
1926 else
1928
1929 SDVTList VTs = getVTList(VT);
1930 SDNodeKey ID(Opc, VTs, {});
1931 ID.AddPointer(GV);
1932 ID.AddInteger(Offset);
1933 ID.AddInteger(TargetFlags);
1934 FoldingSetInsertToken InsertToken;
1935 if (SDNode *E = lookupNode(ID, DL, InsertToken))
1936 return SDValue(E, 0);
1937
1938 auto *N = newSDNode<GlobalAddressSDNode>(
1939 Opc, DL.getIROrder(), DL.getDebugLoc(), GV, VTs, Offset, TargetFlags);
1940 CSEMap.insert(N, InsertToken);
1941 InsertNode(N);
1942 return SDValue(N, 0);
1943}
1944
1946 SDVTList VTs = getVTList(MVT::Untyped);
1948 ID.AddPointer(GV);
1949 FoldingSetInsertToken InsertToken;
1950 if (SDNode *E = lookupNode(ID, SDLoc(), InsertToken))
1951 return SDValue(E, 0);
1952
1953 auto *N = newSDNode<DeactivationSymbolSDNode>(GV, VTs);
1954 CSEMap.insert(N, InsertToken);
1955 InsertNode(N);
1956 return SDValue(N, 0);
1957}
1958
1959SDValue SelectionDAG::getFrameIndex(int FI, EVT VT, bool isTarget) {
1960 unsigned Opc = isTarget ? ISD::TargetFrameIndex : ISD::FrameIndex;
1961 SDVTList VTs = getVTList(VT);
1962 SDNodeKey ID(Opc, VTs, {});
1963 ID.AddInteger(FI);
1964 FoldingSetInsertToken InsertToken;
1965 if (SDNode *E = lookupNode(ID, InsertToken))
1966 return SDValue(E, 0);
1967
1968 auto *N = newSDNode<FrameIndexSDNode>(FI, VTs, isTarget);
1969 CSEMap.insert(N, InsertToken);
1970 InsertNode(N);
1971 return SDValue(N, 0);
1972}
1973
1974SDValue SelectionDAG::getJumpTable(int JTI, EVT VT, bool isTarget,
1975 unsigned TargetFlags) {
1976 assert((TargetFlags == 0 || isTarget) &&
1977 "Cannot set target flags on target-independent jump tables");
1978 unsigned Opc = isTarget ? ISD::TargetJumpTable : ISD::JumpTable;
1979 SDVTList VTs = getVTList(VT);
1980 SDNodeKey ID(Opc, VTs, {});
1981 ID.AddInteger(JTI);
1982 ID.AddInteger(TargetFlags);
1983 FoldingSetInsertToken InsertToken;
1984 if (SDNode *E = lookupNode(ID, InsertToken))
1985 return SDValue(E, 0);
1986
1987 auto *N = newSDNode<JumpTableSDNode>(JTI, VTs, isTarget, TargetFlags);
1988 CSEMap.insert(N, InsertToken);
1989 InsertNode(N);
1990 return SDValue(N, 0);
1991}
1992
1994 const SDLoc &DL) {
1996 return getNode(ISD::JUMP_TABLE_DEBUG_INFO, DL, MVT::Other, Chain,
1997 getTargetConstant(static_cast<uint64_t>(JTI), DL, PTy, true));
1998}
1999
2001 MaybeAlign Alignment, int Offset,
2002 bool isTarget, unsigned TargetFlags) {
2003 assert((TargetFlags == 0 || isTarget) &&
2004 "Cannot set target flags on target-independent globals");
2005 if (!Alignment)
2006 Alignment = shouldOptForSize()
2007 ? getDataLayout().getABITypeAlign(C->getType())
2008 : getDataLayout().getPrefTypeAlign(C->getType());
2009 unsigned Opc = isTarget ? ISD::TargetConstantPool : ISD::ConstantPool;
2010 SDVTList VTs = getVTList(VT);
2011 SDNodeKey ID(Opc, VTs, {});
2012 ID.AddInteger(Alignment->value());
2013 ID.AddInteger(Offset);
2014 ID.AddPointer(C);
2015 ID.AddInteger(TargetFlags);
2016 FoldingSetInsertToken InsertToken;
2017 if (SDNode *E = lookupNode(ID, InsertToken))
2018 return SDValue(E, 0);
2019
2020 auto *N = newSDNode<ConstantPoolSDNode>(isTarget, C, VTs, Offset, *Alignment,
2021 TargetFlags);
2022 CSEMap.insert(N, InsertToken);
2023 InsertNode(N);
2024 SDValue V = SDValue(N, 0);
2025 NewSDValueDbgMsg(V, "Creating new constant pool: ", this);
2026 return V;
2027}
2028
2030 MaybeAlign Alignment, int Offset,
2031 bool isTarget, unsigned TargetFlags) {
2032 assert((TargetFlags == 0 || isTarget) &&
2033 "Cannot set target flags on target-independent globals");
2034 if (!Alignment)
2035 Alignment = getDataLayout().getPrefTypeAlign(C->getType());
2036 unsigned Opc = isTarget ? ISD::TargetConstantPool : ISD::ConstantPool;
2037 SDVTList VTs = getVTList(VT);
2038 SDNodeKey ID(Opc, VTs, {});
2039 ID.AddInteger(Alignment->value());
2040 ID.AddInteger(Offset);
2041 C->addSelectionDAGCSEId(ID.Tail);
2042 ID.AddInteger(TargetFlags);
2043 FoldingSetInsertToken InsertToken;
2044 if (SDNode *E = lookupNode(ID, InsertToken))
2045 return SDValue(E, 0);
2046
2047 auto *N = newSDNode<ConstantPoolSDNode>(isTarget, C, VTs, Offset, *Alignment,
2048 TargetFlags);
2049 CSEMap.insert(N, InsertToken);
2050 InsertNode(N);
2051 return SDValue(N, 0);
2052}
2053
2055 SDNodeKey ID(ISD::BasicBlock, getVTList(MVT::Other), {});
2056 ID.AddPointer(MBB);
2057 FoldingSetInsertToken InsertToken;
2058 if (SDNode *E = lookupNode(ID, InsertToken))
2059 return SDValue(E, 0);
2060
2061 auto *N = newSDNode<BasicBlockSDNode>(MBB);
2062 CSEMap.insert(N, InsertToken);
2063 InsertNode(N);
2064 return SDValue(N, 0);
2065}
2066
2068 if (VT.isSimple() && (unsigned)VT.getSimpleVT().SimpleTy >=
2069 ValueTypeNodes.size())
2070 ValueTypeNodes.resize(VT.getSimpleVT().SimpleTy+1);
2071
2072 SDNode *&N = VT.isExtended() ?
2073 ExtendedValueTypeNodes[VT] : ValueTypeNodes[VT.getSimpleVT().SimpleTy];
2074
2075 if (N) return SDValue(N, 0);
2076 N = newSDNode<VTSDNode>(VT);
2077 InsertNode(N);
2078 return SDValue(N, 0);
2079}
2080
2082 SDNode *&N = ExternalSymbols[Sym];
2083 if (N) return SDValue(N, 0);
2084 N = newSDNode<ExternalSymbolSDNode>(false, Sym, 0, getVTList(VT));
2085 InsertNode(N);
2086 return SDValue(N, 0);
2087}
2088
2089SDValue SelectionDAG::getExternalSymbol(RTLIB::LibcallImpl Libcall, EVT VT) {
2091 return getExternalSymbol(SymName.data(), VT);
2092}
2093
2095 SDNode *&N = MCSymbols[Sym];
2096 if (N)
2097 return SDValue(N, 0);
2098 N = newSDNode<MCSymbolSDNode>(Sym, getVTList(VT));
2099 InsertNode(N);
2100 return SDValue(N, 0);
2101}
2102
2104 unsigned TargetFlags) {
2105 SDNode *&N =
2106 TargetExternalSymbols[std::pair<std::string, unsigned>(Sym, TargetFlags)];
2107 if (N) return SDValue(N, 0);
2108 N = newSDNode<ExternalSymbolSDNode>(true, Sym, TargetFlags, getVTList(VT));
2109 InsertNode(N);
2110 return SDValue(N, 0);
2111}
2112
2114 EVT VT, unsigned TargetFlags) {
2116 return getTargetExternalSymbol(SymName.data(), VT, TargetFlags);
2117}
2118
2120 if ((unsigned)Cond >= CondCodeNodes.size())
2121 CondCodeNodes.resize(Cond+1);
2122
2123 if (!CondCodeNodes[Cond]) {
2124 auto *N = newSDNode<CondCodeSDNode>(Cond);
2125 CondCodeNodes[Cond] = N;
2126 InsertNode(N);
2127 }
2128
2129 return SDValue(CondCodeNodes[Cond], 0);
2130}
2131
2133 assert(MulImm.getBitWidth() == VT.getSizeInBits() &&
2134 "APInt size does not match type size!");
2135
2136 if (MulImm == 0)
2137 return getConstant(0, DL, VT);
2138
2139 const MachineFunction &MF = getMachineFunction();
2140 const Function &F = MF.getFunction();
2141 ConstantRange CR = getVScaleRange(&F, 64);
2142 if (const APInt *C = CR.getSingleElement())
2143 return getConstant(MulImm * C->getZExtValue(), DL, VT);
2144
2145 return getNode(ISD::VSCALE, DL, VT, getConstant(MulImm, DL, VT));
2146}
2147
2148/// \returns a value of type \p VT that represents the runtime value of \p
2149/// Quantity, i.e. scaled by vscale if it's scalable, or a fixed constant
2150/// otherwise. Quantity should be a FixedOrScalableQuantity, i.e. ElementCount
2151/// or TypeSize.
2152template <typename Ty>
2154 EVT VT, Ty Quantity) {
2155 if (Quantity.isScalable())
2156 return DAG.getVScale(
2157 DL, VT, APInt(VT.getSizeInBits(), Quantity.getKnownMinValue()));
2158
2159 return DAG.getConstant(Quantity.getKnownMinValue(), DL, VT);
2160}
2161
2163 ElementCount EC) {
2164 return getFixedOrScalableQuantity(*this, DL, VT, EC);
2165}
2166
2168 return getFixedOrScalableQuantity(*this, DL, VT, TS);
2169}
2170
2172 ElementCount EC) {
2173 EVT IdxVT = TLI->getVectorIdxTy(getDataLayout());
2174 EVT MaskVT = TLI->getSetCCResultType(getDataLayout(), *getContext(), DataVT);
2175 return getNode(ISD::GET_ACTIVE_LANE_MASK, DL, MaskVT,
2176 getConstant(0, DL, IdxVT), getElementCount(DL, IdxVT, EC));
2177}
2178
2180 APInt One(ResVT.getScalarSizeInBits(), 1);
2181 return getStepVector(DL, ResVT, One);
2182}
2183
2185 const APInt &StepVal) {
2186 assert(ResVT.getScalarSizeInBits() == StepVal.getBitWidth());
2187 if (ResVT.isScalableVector())
2188 return getNode(
2189 ISD::STEP_VECTOR, DL, ResVT,
2190 getTargetConstant(StepVal, DL, ResVT.getVectorElementType()));
2191
2192 SmallVector<SDValue, 16> OpsStepConstants;
2193 for (uint64_t i = 0; i < ResVT.getVectorNumElements(); i++)
2194 OpsStepConstants.push_back(
2195 getConstant(StepVal * i, DL, ResVT.getVectorElementType()));
2196 return getBuildVector(ResVT, DL, OpsStepConstants);
2197}
2198
2199/// Swaps the values of N1 and N2. Swaps all indices in the shuffle mask M that
2200/// point at N1 to point at N2 and indices that point at N2 to point at N1.
2205
2207 SDValue N2, ArrayRef<int> Mask) {
2208 assert(VT.getVectorNumElements() == Mask.size() &&
2209 "Must have the same number of vector elements as mask elements!");
2210 assert(VT == N1.getValueType() && VT == N2.getValueType() &&
2211 "Invalid VECTOR_SHUFFLE");
2212
2213 // Canonicalize shuffle undef, undef -> undef
2214 if (N1.isUndef() && N2.isUndef()) {
2215 if (N1.getOpcode() == ISD::POISON && N2.getOpcode() == ISD::POISON)
2216 return getPOISON(VT);
2217 return getUNDEF(VT);
2218 }
2219
2220 // Validate that all indices in Mask are within the range of the elements
2221 // input to the shuffle.
2222 int NElts = Mask.size();
2223 assert(llvm::all_of(Mask,
2224 [&](int M) { return M < (NElts * 2) && M >= -1; }) &&
2225 "Index out of range");
2226
2227 // Copy the mask so we can do any needed cleanup.
2228 SmallVector<int, 8> MaskVec(Mask);
2229
2230 // Canonicalize shuffle v, v -> v, poison
2231 if (N1 == N2) {
2232 N2 = getPOISON(VT);
2233 for (int i = 0; i != NElts; ++i)
2234 if (MaskVec[i] >= NElts) MaskVec[i] -= NElts;
2235 }
2236
2237 // Canonicalize shuffle undef, v -> v, undef. Commute the shuffle mask.
2238 if (N1.isUndef())
2239 commuteShuffle(N1, N2, MaskVec);
2240
2241 if (TLI->hasVectorBlend()) {
2242 // If shuffling a splat, try to blend the splat instead. We do this here so
2243 // that even when this arises during lowering we don't have to re-handle it.
2244 auto BlendSplat = [&](BuildVectorSDNode *BV, int Offset) {
2245 BitVector UndefElements;
2246 SDValue Splat = BV->getSplatValue(&UndefElements);
2247 if (!Splat)
2248 return;
2249
2250 for (int i = 0; i < NElts; ++i) {
2251 if (MaskVec[i] < Offset || MaskVec[i] >= (Offset + NElts))
2252 continue;
2253
2254 // If this input comes from undef, mark it as such.
2255 if (UndefElements[MaskVec[i] - Offset]) {
2256 MaskVec[i] = -1;
2257 continue;
2258 }
2259
2260 // If we can blend a non-undef lane, use that instead.
2261 if (!UndefElements[i])
2262 MaskVec[i] = i + Offset;
2263 }
2264 };
2265 if (auto *N1BV = dyn_cast<BuildVectorSDNode>(N1))
2266 BlendSplat(N1BV, 0);
2267 if (auto *N2BV = dyn_cast<BuildVectorSDNode>(N2))
2268 BlendSplat(N2BV, NElts);
2269 }
2270
2271 // Canonicalize all index into lhs, -> shuffle lhs, poison
2272 // Canonicalize all index into rhs, -> shuffle rhs, poison
2273 bool AllLHS = true, AllRHS = true;
2274 bool N2Undef = N2.isUndef();
2275 for (int i = 0; i != NElts; ++i) {
2276 if (MaskVec[i] >= NElts) {
2277 if (N2Undef)
2278 MaskVec[i] = -1;
2279 else
2280 AllLHS = false;
2281 } else if (MaskVec[i] >= 0) {
2282 AllRHS = false;
2283 }
2284 }
2285 if (AllLHS && AllRHS)
2286 return getPOISON(VT);
2287 if (AllLHS && !N2Undef)
2288 N2 = getPOISON(VT);
2289 if (AllRHS) {
2290 N1 = getPOISON(VT);
2291 commuteShuffle(N1, N2, MaskVec);
2292 }
2293 // Reset our undef status after accounting for the mask.
2294 N2Undef = N2.isUndef();
2295 // Re-check whether both sides ended up undef.
2296 if (N1.isUndef() && N2Undef) {
2297 if (N1.getOpcode() == ISD::POISON && N2.getOpcode() == ISD::POISON)
2298 return getPOISON(VT);
2299 return getUNDEF(VT);
2300 }
2301
2302 // If Identity shuffle return that node.
2303 bool Identity = true, AllSame = true;
2304 for (int i = 0; i != NElts; ++i) {
2305 if (MaskVec[i] >= 0 && MaskVec[i] != i) Identity = false;
2306 if (MaskVec[i] != MaskVec[0]) AllSame = false;
2307 }
2308 if (Identity && NElts)
2309 return N1;
2310
2311 // Shuffling a constant splat doesn't change the result.
2312 if (N2Undef) {
2313 SDValue V = N1;
2314
2315 // Look through any bitcasts. We check that these don't change the number
2316 // (and size) of elements and just changes their types.
2317 while (V.getOpcode() == ISD::BITCAST)
2318 V = V->getOperand(0);
2319
2320 // A splat should always show up as a build vector node.
2321 if (auto *BV = dyn_cast<BuildVectorSDNode>(V)) {
2322 BitVector UndefElements;
2323 SDValue Splat = BV->getSplatValue(&UndefElements);
2324 // If this is a splat of an undef, shuffling it is also undef.
2325 if (Splat && Splat.isUndef())
2326 return Splat.getOpcode() == ISD::POISON ? getPOISON(VT) : getUNDEF(VT);
2327
2328 bool SameNumElts =
2329 V.getValueType().getVectorNumElements() == VT.getVectorNumElements();
2330
2331 // We only have a splat which can skip shuffles if there is a splatted
2332 // value and no undef lanes rearranged by the shuffle.
2333 if (Splat && UndefElements.none()) {
2334 // Splat of <x, x, ..., x>, return <x, x, ..., x>, provided that the
2335 // number of elements match or the value splatted is a zero constant.
2336 if (SameNumElts || isNullConstant(Splat))
2337 return N1;
2338 }
2339
2340 // If the shuffle itself creates a splat, build the vector directly.
2341 if (AllSame && SameNumElts) {
2342 EVT BuildVT = BV->getValueType(0);
2343 const SDValue &Splatted = BV->getOperand(MaskVec[0]);
2344 SDValue NewBV = getSplatBuildVector(BuildVT, dl, Splatted);
2345
2346 // We may have jumped through bitcasts, so the type of the
2347 // BUILD_VECTOR may not match the type of the shuffle.
2348 if (BuildVT != VT)
2349 NewBV = getNode(ISD::BITCAST, dl, VT, NewBV);
2350 return NewBV;
2351 }
2352 }
2353 }
2354
2355 SDVTList VTs = getVTList(VT);
2356 SDValue Ops[2] = { N1, N2 };
2358 for (int i = 0; i != NElts; ++i)
2359 ID.AddInteger(MaskVec[i]);
2360
2361 FoldingSetInsertToken InsertToken;
2362 if (SDNode *E = lookupNode(ID, dl, InsertToken))
2363 return SDValue(E, 0);
2364
2365 // Allocate the mask array for the node out of the BumpPtrAllocator, since
2366 // SDNode doesn't have access to it. This memory will be "leaked" when
2367 // the node is deallocated, but recovered when the NodeAllocator is released.
2368 int *MaskAlloc = OperandAllocator.Allocate<int>(NElts);
2369 llvm::copy(MaskVec, MaskAlloc);
2370
2371 auto *N = newSDNode<ShuffleVectorSDNode>(VTs, dl.getIROrder(),
2372 dl.getDebugLoc(), MaskAlloc);
2373 createOperands(N, Ops);
2374
2375 CSEMap.insert(N, InsertToken);
2376 InsertNode(N);
2377 SDValue V = SDValue(N, 0);
2378 NewSDValueDbgMsg(V, "Creating new node: ", this);
2379 return V;
2380}
2381
2383 EVT VT = SV.getValueType(0);
2384 SmallVector<int, 8> MaskVec(SV.getMask());
2386
2387 SDValue Op0 = SV.getOperand(0);
2388 SDValue Op1 = SV.getOperand(1);
2389 return getVectorShuffle(VT, SDLoc(&SV), Op1, Op0, MaskVec);
2390}
2391
2393 SDVTList VTs = getVTList(VT);
2394 SDNodeKey ID(ISD::Register, VTs, {});
2395 ID.AddInteger(Reg.id());
2396 FoldingSetInsertToken InsertToken;
2397 if (SDNode *E = lookupNode(ID, InsertToken))
2398 return SDValue(E, 0);
2399
2400 auto *N = newSDNode<RegisterSDNode>(Reg, VTs);
2401 N->SDNodeBits.IsDivergent = TLI->isSDNodeSourceOfDivergence(N, FLI, UA);
2402 CSEMap.insert(N, InsertToken);
2403 InsertNode(N);
2404 return SDValue(N, 0);
2405}
2406
2408 SDNodeKey ID(ISD::RegisterMask, getVTList(MVT::Untyped), {});
2409 ID.AddPointer(RegMask);
2410 FoldingSetInsertToken InsertToken;
2411 if (SDNode *E = lookupNode(ID, InsertToken))
2412 return SDValue(E, 0);
2413
2414 auto *N = newSDNode<RegisterMaskSDNode>(RegMask);
2415 CSEMap.insert(N, InsertToken);
2416 InsertNode(N);
2417 return SDValue(N, 0);
2418}
2419
2421 MCSymbol *Label) {
2422 return getLabelNode(ISD::EH_LABEL, dl, Root, Label);
2423}
2424
2425SDValue SelectionDAG::getLabelNode(unsigned Opcode, const SDLoc &dl,
2426 SDValue Root, MCSymbol *Label) {
2427 SDValue Ops[] = { Root };
2428 SDNodeKey ID(Opcode, getVTList(MVT::Other), Ops);
2429 ID.AddPointer(Label);
2430 FoldingSetInsertToken InsertToken;
2431 if (SDNode *E = lookupNode(ID, InsertToken))
2432 return SDValue(E, 0);
2433
2434 auto *N =
2435 newSDNode<LabelSDNode>(Opcode, dl.getIROrder(), dl.getDebugLoc(), Label);
2436 createOperands(N, Ops);
2437
2438 CSEMap.insert(N, InsertToken);
2439 InsertNode(N);
2440 return SDValue(N, 0);
2441}
2442
2444 int64_t Offset, bool isTarget,
2445 unsigned TargetFlags) {
2446 unsigned Opc = isTarget ? ISD::TargetBlockAddress : ISD::BlockAddress;
2447 SDVTList VTs = getVTList(VT);
2448
2449 SDNodeKey ID(Opc, VTs, {});
2450 ID.AddPointer(BA);
2451 ID.AddInteger(Offset);
2452 ID.AddInteger(TargetFlags);
2453 FoldingSetInsertToken InsertToken;
2454 if (SDNode *E = lookupNode(ID, InsertToken))
2455 return SDValue(E, 0);
2456
2457 auto *N = newSDNode<BlockAddressSDNode>(Opc, VTs, BA, Offset, TargetFlags);
2458 CSEMap.insert(N, InsertToken);
2459 InsertNode(N);
2460 return SDValue(N, 0);
2461}
2462
2464 SDNodeKey ID(ISD::SRCVALUE, getVTList(MVT::Other), {});
2465 ID.AddPointer(V);
2466
2467 FoldingSetInsertToken InsertToken;
2468 if (SDNode *E = lookupNode(ID, InsertToken))
2469 return SDValue(E, 0);
2470
2471 auto *N = newSDNode<SrcValueSDNode>(V);
2472 CSEMap.insert(N, InsertToken);
2473 InsertNode(N);
2474 return SDValue(N, 0);
2475}
2476
2478 SDNodeKey ID(ISD::MDNODE_SDNODE, getVTList(MVT::Other), {});
2479 ID.AddPointer(MD);
2480
2481 FoldingSetInsertToken InsertToken;
2482 if (SDNode *E = lookupNode(ID, InsertToken))
2483 return SDValue(E, 0);
2484
2485 auto *N = newSDNode<MDNodeSDNode>(MD);
2486 CSEMap.insert(N, InsertToken);
2487 InsertNode(N);
2488 return SDValue(N, 0);
2489}
2490
2492 if (VT == V.getValueType())
2493 return V;
2494
2495 return getNode(ISD::BITCAST, SDLoc(V), VT, V);
2496}
2497
2499 unsigned SrcAS, unsigned DestAS,
2500 const SDNodeFlags Flags) {
2501 SDVTList VTs = getVTList(VT);
2502 SDValue Ops[] = {Ptr};
2504 ID.AddInteger(SrcAS);
2505 ID.AddInteger(DestAS);
2506
2507 FoldingSetInsertToken InsertToken;
2508 if (SDNode *E = lookupNode(ID, dl, InsertToken)) {
2509 E->intersectFlagsWith(Flags);
2510 return SDValue(E, 0);
2511 }
2512
2513 auto *N = newSDNode<AddrSpaceCastSDNode>(dl.getIROrder(), dl.getDebugLoc(),
2514 VTs, SrcAS, DestAS);
2515 N->setFlags(Flags);
2516 createOperands(N, Ops);
2517
2518 CSEMap.insert(N, InsertToken);
2519 InsertNode(N);
2520 return SDValue(N, 0);
2521}
2522
2524 return getNode(ISD::FREEZE, SDLoc(V), V.getValueType(), V);
2525}
2526
2528 UndefPoisonKind Kind) {
2529 if (isGuaranteedNotToBeUndefOrPoison(V, DemandedElts, Kind))
2530 return V;
2531 return getFreeze(V);
2532}
2533
2534/// getShiftAmountOperand - Return the specified value casted to
2535/// the target's desired shift amount type.
2537 EVT OpTy = Op.getValueType();
2538 EVT ShTy = TLI->getShiftAmountTy(LHSTy, getDataLayout());
2539 if (OpTy == ShTy || OpTy.isVector()) return Op;
2540
2541 return getZExtOrTrunc(Op, SDLoc(Op), ShTy);
2542}
2543
2545 SDLoc dl(Node);
2547 const Value *V = cast<SrcValueSDNode>(Node->getOperand(2))->getValue();
2548 EVT VT = Node->getValueType(0);
2549 SDValue Tmp1 = Node->getOperand(0);
2550 SDValue Tmp2 = Node->getOperand(1);
2551 const MaybeAlign MA(Node->getConstantOperandVal(3));
2552
2553 SDValue VAListLoad = getLoad(TLI.getPointerTy(getDataLayout()), dl, Tmp1,
2554 Tmp2, MachinePointerInfo(V));
2555 SDValue VAList = VAListLoad;
2556
2557 if (MA && *MA > TLI.getMinStackArgumentAlignment()) {
2558 VAList = getNode(ISD::ADD, dl, VAList.getValueType(), VAList,
2559 getConstant(MA->value() - 1, dl, VAList.getValueType()));
2560
2561 VAList = getNode(
2562 ISD::AND, dl, VAList.getValueType(), VAList,
2563 getSignedConstant(-(int64_t)MA->value(), dl, VAList.getValueType()));
2564 }
2565
2566 // Increment the pointer, VAList, to the next vaarg
2567 Tmp1 = getNode(ISD::ADD, dl, VAList.getValueType(), VAList,
2568 getConstant(getDataLayout().getTypeAllocSize(
2569 VT.getTypeForEVT(*getContext())),
2570 dl, VAList.getValueType()));
2571 // Store the incremented VAList to the legalized pointer
2572 Tmp1 =
2573 getStore(VAListLoad.getValue(1), dl, Tmp1, Tmp2, MachinePointerInfo(V));
2574 // Load the actual argument out of the pointer VAList
2575 return getLoad(VT, dl, Tmp1, VAList, MachinePointerInfo());
2576}
2577
2579 SDLoc dl(Node);
2581 // This defaults to loading a pointer from the input and storing it to the
2582 // output, returning the chain.
2583 const Value *VD = cast<SrcValueSDNode>(Node->getOperand(3))->getValue();
2584 const Value *VS = cast<SrcValueSDNode>(Node->getOperand(4))->getValue();
2585 SDValue Tmp1 =
2586 getLoad(TLI.getPointerTy(getDataLayout()), dl, Node->getOperand(0),
2587 Node->getOperand(2), MachinePointerInfo(VS));
2588 return getStore(Tmp1.getValue(1), dl, Tmp1, Node->getOperand(1),
2589 MachinePointerInfo(VD));
2590}
2591
2593 const DataLayout &DL = getDataLayout();
2594 Type *Ty = VT.getTypeForEVT(*getContext());
2595 Align RedAlign = UseABI ? DL.getABITypeAlign(Ty) : DL.getPrefTypeAlign(Ty);
2596
2597 if (TLI->isTypeLegal(VT) || !VT.isVector())
2598 return RedAlign;
2599
2600 const TargetFrameLowering *TFI = MF->getSubtarget().getFrameLowering();
2601 const Align StackAlign = TFI->getStackAlign();
2602
2603 // See if we can choose a smaller ABI alignment in cases where it's an
2604 // illegal vector type that will get broken down.
2605 if (RedAlign > StackAlign) {
2606 EVT IntermediateVT;
2607 MVT RegisterVT;
2608 unsigned NumIntermediates;
2609 TLI->getVectorTypeBreakdown(*getContext(), VT, IntermediateVT,
2610 NumIntermediates, RegisterVT);
2611 Ty = IntermediateVT.getTypeForEVT(*getContext());
2612 Align RedAlign2 = UseABI ? DL.getABITypeAlign(Ty) : DL.getPrefTypeAlign(Ty);
2613 if (RedAlign2 < RedAlign)
2614 RedAlign = RedAlign2;
2615
2616 if (!getMachineFunction().getFrameInfo().isStackRealignable())
2617 // If the stack is not realignable, the alignment should be limited to the
2618 // StackAlignment
2619 RedAlign = std::min(RedAlign, StackAlign);
2620 }
2621
2622 return RedAlign;
2623}
2624
2626 MachineFrameInfo &MFI = MF->getFrameInfo();
2627 const TargetFrameLowering *TFI = MF->getSubtarget().getFrameLowering();
2628 int StackID = 0;
2629 if (Bytes.isScalable())
2630 StackID = TFI->getStackIDForScalableVectors();
2631 // The stack id gives an indication of whether the object is scalable or
2632 // not, so it's safe to pass in the minimum size here.
2633 int FrameIdx = MFI.CreateStackObject(Bytes.getKnownMinValue(), Alignment,
2634 false, nullptr, StackID);
2635 return getFrameIndex(FrameIdx, TLI->getFrameIndexTy(getDataLayout()));
2636}
2637
2639 Type *Ty = VT.getTypeForEVT(*getContext());
2640 Align StackAlign =
2641 std::max(getDataLayout().getPrefTypeAlign(Ty), Align(minAlign));
2642 return CreateStackTemporary(VT.getStoreSize(), StackAlign);
2643}
2644
2646 TypeSize VT1Size = VT1.getStoreSize();
2647 TypeSize VT2Size = VT2.getStoreSize();
2648 assert(VT1Size.isScalable() == VT2Size.isScalable() &&
2649 "Don't know how to choose the maximum size when creating a stack "
2650 "temporary");
2651 TypeSize Bytes = VT1Size.getKnownMinValue() > VT2Size.getKnownMinValue()
2652 ? VT1Size
2653 : VT2Size;
2654
2655 Type *Ty1 = VT1.getTypeForEVT(*getContext());
2656 Type *Ty2 = VT2.getTypeForEVT(*getContext());
2657 const DataLayout &DL = getDataLayout();
2658 Align Align = std::max(DL.getPrefTypeAlign(Ty1), DL.getPrefTypeAlign(Ty2));
2659 return CreateStackTemporary(Bytes, Align);
2660}
2661
2663 const SDLoc &DL, SDValue Chain) {
2664 EVT SrcVT = SrcOp.getValueType();
2665 Type *DestType = DestVT.getTypeForEVT(*getContext());
2666 Align DestAlign = getDataLayout().getPrefTypeAlign(DestType);
2667
2668 // Create the stack frame object.
2669 Align SrcAlign =
2671 SDValue FIPtr = CreateStackTemporary(SlotVT.getStoreSize(), SrcAlign);
2672
2673 FrameIndexSDNode *StackPtrFI = cast<FrameIndexSDNode>(FIPtr);
2674 int SPFI = StackPtrFI->getIndex();
2675 MachinePointerInfo PtrInfo =
2677
2678 // Emit a store to the stack slot. Use a truncstore if the input value is
2679 // later than DestVT.
2680 SDValue Store;
2681
2682 if (SrcVT.bitsGT(SlotVT))
2683 Store = getTruncStore(Chain, DL, SrcOp, FIPtr, PtrInfo, SlotVT, SrcAlign);
2684 else {
2685 assert(SrcVT.bitsEq(SlotVT) && "Invalid store");
2686 Store = getStore(Chain, DL, SrcOp, FIPtr, PtrInfo, SrcAlign);
2687 }
2688
2689 // Result is a load from the stack slot.
2690 if (SlotVT.bitsEq(DestVT))
2691 return getLoad(DestVT, DL, Store, FIPtr, PtrInfo, DestAlign);
2692
2693 assert(SlotVT.bitsLT(DestVT) && "Unknown extension!");
2694 return getExtLoad(ISD::EXTLOAD, DL, DestVT, Store, FIPtr, PtrInfo, SlotVT,
2695 DestAlign);
2696}
2697
2699 ISD::CondCode Cond, const SDLoc &dl,
2700 SDNodeFlags Flags) {
2701 EVT OpVT = N1.getValueType();
2702
2703 auto GetUndefBooleanConstant = [&]() {
2704 if (VT.getScalarType() == MVT::i1 ||
2705 TLI->getBooleanContents(OpVT) ==
2707 return getUNDEF(VT);
2708 // ZeroOrOne / ZeroOrNegative require specific values for the high bits,
2709 // so we cannot use getUNDEF(). Return zero instead.
2710 return getConstant(0, dl, VT);
2711 };
2712
2713 // These setcc operations always fold.
2714 switch (Cond) {
2715 default: break;
2716 case ISD::SETFALSE:
2717 case ISD::SETFALSE2: return getBoolConstant(false, dl, VT, OpVT);
2718 case ISD::SETTRUE:
2719 case ISD::SETTRUE2: return getBoolConstant(true, dl, VT, OpVT);
2720
2721 case ISD::SETOEQ:
2722 case ISD::SETOGT:
2723 case ISD::SETOGE:
2724 case ISD::SETOLT:
2725 case ISD::SETOLE:
2726 case ISD::SETONE:
2727 case ISD::SETO:
2728 case ISD::SETUO:
2729 case ISD::SETUEQ:
2730 case ISD::SETUNE:
2731 assert(!OpVT.isInteger() && "Illegal setcc for integer!");
2732 break;
2733 }
2734
2735 if (OpVT.isInteger()) {
2736 // For EQ and NE, we can always pick a value for the undef to make the
2737 // predicate pass or fail, so we can return undef.
2738 // Matches behavior in llvm::ConstantFoldCompareInstruction.
2739 // icmp eq/ne X, undef -> undef.
2740 if ((N1.isUndef() || N2.isUndef()) &&
2741 (Cond == ISD::SETEQ || Cond == ISD::SETNE))
2742 return GetUndefBooleanConstant();
2743
2744 // If both operands are undef, we can return undef for int comparison.
2745 // icmp undef, undef -> undef.
2746 if (N1.isUndef() && N2.isUndef())
2747 return GetUndefBooleanConstant();
2748
2749 // icmp X, X -> true/false
2750 // icmp X, undef -> true/false because undef could be X.
2751 if (N1.isUndef() || N2.isUndef() || N1 == N2)
2752 return getBoolConstant(ISD::isTrueWhenEqual(Cond), dl, VT, OpVT);
2753 }
2754
2756 const APInt &C2 = N2C->getAPIntValue();
2758 const APInt &C1 = N1C->getAPIntValue();
2759
2761 dl, VT, OpVT);
2762 }
2763 }
2764
2765 auto *N1CFP = dyn_cast<ConstantFPSDNode>(N1);
2766 auto *N2CFP = dyn_cast<ConstantFPSDNode>(N2);
2767
2768 if (N1CFP && N2CFP) {
2769 APFloat::cmpResult R = N1CFP->getValueAPF().compare(N2CFP->getValueAPF());
2770 switch (Cond) {
2771 default: break;
2772 case ISD::SETEQ: if (R==APFloat::cmpUnordered)
2773 return GetUndefBooleanConstant();
2774 [[fallthrough]];
2775 case ISD::SETOEQ: return getBoolConstant(R==APFloat::cmpEqual, dl, VT,
2776 OpVT);
2777 case ISD::SETNE: if (R==APFloat::cmpUnordered)
2778 return GetUndefBooleanConstant();
2779 [[fallthrough]];
2781 R==APFloat::cmpLessThan, dl, VT,
2782 OpVT);
2783 case ISD::SETLT: if (R==APFloat::cmpUnordered)
2784 return GetUndefBooleanConstant();
2785 [[fallthrough]];
2786 case ISD::SETOLT: return getBoolConstant(R==APFloat::cmpLessThan, dl, VT,
2787 OpVT);
2788 case ISD::SETGT: if (R==APFloat::cmpUnordered)
2789 return GetUndefBooleanConstant();
2790 [[fallthrough]];
2792 VT, OpVT);
2793 case ISD::SETLE: if (R==APFloat::cmpUnordered)
2794 return GetUndefBooleanConstant();
2795 [[fallthrough]];
2797 R==APFloat::cmpEqual, dl, VT,
2798 OpVT);
2799 case ISD::SETGE: if (R==APFloat::cmpUnordered)
2800 return GetUndefBooleanConstant();
2801 [[fallthrough]];
2803 R==APFloat::cmpEqual, dl, VT, OpVT);
2804 case ISD::SETO: return getBoolConstant(R!=APFloat::cmpUnordered, dl, VT,
2805 OpVT);
2806 case ISD::SETUO: return getBoolConstant(R==APFloat::cmpUnordered, dl, VT,
2807 OpVT);
2809 R==APFloat::cmpEqual, dl, VT,
2810 OpVT);
2811 case ISD::SETUNE: return getBoolConstant(R!=APFloat::cmpEqual, dl, VT,
2812 OpVT);
2814 R==APFloat::cmpLessThan, dl, VT,
2815 OpVT);
2817 R==APFloat::cmpUnordered, dl, VT,
2818 OpVT);
2820 VT, OpVT);
2821 case ISD::SETUGE: return getBoolConstant(R!=APFloat::cmpLessThan, dl, VT,
2822 OpVT);
2823 }
2824 } else if (N1CFP && OpVT.isSimple() && !N2.isUndef()) {
2825 // Ensure that the constant occurs on the RHS.
2827 if (!TLI->isCondCodeLegal(SwappedCond, OpVT.getSimpleVT()))
2828 return SDValue();
2829 return getSetCC(dl, VT, N2, N1, SwappedCond, /*Chain=*/{},
2830 /*IsSignaling=*/false, Flags);
2831 } else if ((N2CFP && N2CFP->getValueAPF().isNaN()) ||
2832 (OpVT.isFloatingPoint() && (N1.isUndef() || N2.isUndef()))) {
2833 // If an operand is known to be a nan (or undef that could be a nan), we can
2834 // fold it.
2835 // Choosing NaN for the undef will always make unordered comparison succeed
2836 // and ordered comparison fails.
2837 // Matches behavior in llvm::ConstantFoldCompareInstruction.
2838 switch (ISD::getUnorderedFlavor(Cond)) {
2839 default:
2840 llvm_unreachable("Unknown flavor!");
2841 case 0: // Known false.
2842 return getBoolConstant(false, dl, VT, OpVT);
2843 case 1: // Known true.
2844 return getBoolConstant(true, dl, VT, OpVT);
2845 case 2: // Undefined.
2846 return GetUndefBooleanConstant();
2847 }
2848 }
2849
2850 // Could not fold it.
2851 return SDValue();
2852}
2853
2854/// SignBitIsZero - Return true if the sign bit of Op is known to be zero. We
2855/// use this predicate to simplify operations downstream.
2857 unsigned BitWidth = Op.getScalarValueSizeInBits();
2859}
2860
2861// TODO: Should have argument to specify if sign bit of nan is ignorable.
2863 if (Depth >= MaxRecursionDepth)
2864 return false; // Limit search depth.
2865
2866 unsigned Opc = Op.getOpcode();
2867 switch (Opc) {
2868 case ISD::FABS:
2869 return true;
2870 case ISD::AssertNoFPClass: {
2871 FPClassTest NoFPClass =
2872 static_cast<FPClassTest>(Op.getConstantOperandVal(1));
2873
2874 const FPClassTest TestMask = fcNan | fcNegative;
2875 return (NoFPClass & TestMask) == TestMask;
2876 }
2877 case ISD::ARITH_FENCE:
2878 return SignBitIsZeroFP(Op.getOperand(0), Depth + 1);
2879 case ISD::FEXP:
2880 case ISD::FEXP2:
2881 case ISD::FEXP10:
2882 return Op->getFlags().hasNoNaNs();
2883 case ISD::FMINNUM:
2884 case ISD::FMINNUM_IEEE:
2885 case ISD::FMINIMUM:
2886 case ISD::FMINIMUMNUM:
2887 return SignBitIsZeroFP(Op.getOperand(1), Depth + 1) &&
2888 SignBitIsZeroFP(Op.getOperand(0), Depth + 1);
2889 case ISD::FMAXNUM:
2890 case ISD::FMAXNUM_IEEE:
2891 case ISD::FMAXIMUM:
2892 case ISD::FMAXIMUMNUM:
2893 // TODO: If we can ignore the sign bit of nans, only one side being known 0
2894 // is sufficient.
2895 return SignBitIsZeroFP(Op.getOperand(1), Depth + 1) &&
2896 SignBitIsZeroFP(Op.getOperand(0), Depth + 1);
2897 default:
2898 return false;
2899 }
2900
2901 llvm_unreachable("covered opcode switch");
2902}
2903
2904/// MaskedValueIsZero - Return true if 'V & Mask' is known to be zero. We use
2905/// this predicate to simplify operations downstream. Mask is known to be zero
2906/// for bits that V cannot have.
2908 unsigned Depth) const {
2909 return Mask.isSubsetOf(computeKnownBits(V, Depth).Zero);
2910}
2911
2912/// MaskedValueIsZero - Return true if 'V & Mask' is known to be zero in
2913/// DemandedElts. We use this predicate to simplify operations downstream.
2914/// Mask is known to be zero for bits that V cannot have.
2916 const APInt &DemandedElts,
2917 unsigned Depth) const {
2918 return Mask.isSubsetOf(computeKnownBits(V, DemandedElts, Depth).Zero);
2919}
2920
2921/// MaskedVectorIsZero - Return true if 'Op' is known to be zero in
2922/// DemandedElts. We use this predicate to simplify operations downstream.
2924 unsigned Depth /* = 0 */) const {
2925 return computeKnownBits(V, DemandedElts, Depth).isZero();
2926}
2927
2928/// MaskedValueIsAllOnes - Return true if '(Op & Mask) == Mask'.
2930 unsigned Depth) const {
2931 return Mask.isSubsetOf(computeKnownBits(V, Depth).One);
2932}
2933
2935 const APInt &DemandedElts,
2936 unsigned Depth) const {
2937 EVT VT = Op.getValueType();
2938 assert(VT.isVector() && !VT.isScalableVector() && "Only for fixed vectors!");
2939
2940 unsigned NumElts = VT.getVectorNumElements();
2941 assert(DemandedElts.getBitWidth() == NumElts && "Unexpected demanded mask.");
2942
2943 APInt KnownZeroElements = APInt::getZero(NumElts);
2944 for (unsigned EltIdx = 0; EltIdx != NumElts; ++EltIdx) {
2945 if (!DemandedElts[EltIdx])
2946 continue; // Don't query elements that are not demanded.
2947 APInt Mask = APInt::getOneBitSet(NumElts, EltIdx);
2948 if (MaskedVectorIsZero(Op, Mask, Depth))
2949 KnownZeroElements.setBit(EltIdx);
2950 }
2951 return KnownZeroElements;
2952}
2953
2954/// isSplatValue - Return true if the vector V has the same value
2955/// across all DemandedElts. For scalable vectors, we don't know the
2956/// number of lanes at compile time. Instead, we use a 1 bit APInt
2957/// to represent a conservative value for all lanes; that is, that
2958/// one bit value is implicitly splatted across all lanes.
2959bool SelectionDAG::isSplatValue(SDValue V, const APInt &DemandedElts,
2960 APInt &UndefElts, unsigned Depth) const {
2961 unsigned Opcode = V.getOpcode();
2962 EVT VT = V.getValueType();
2963 assert(VT.isVector() && "Vector type expected");
2964 assert((!VT.isScalableVector() || DemandedElts.getBitWidth() == 1) &&
2965 "scalable demanded bits are ignored");
2966
2967 if (!DemandedElts)
2968 return false; // No demanded elts, better to assume we don't know anything.
2969
2970 if (Depth >= MaxRecursionDepth)
2971 return false; // Limit search depth.
2972
2973 // Deal with some common cases here that work for both fixed and scalable
2974 // vector types.
2975 switch (Opcode) {
2976 case ISD::SPLAT_VECTOR:
2977 UndefElts = V.getOperand(0).isUndef()
2978 ? APInt::getAllOnes(DemandedElts.getBitWidth())
2979 : APInt(DemandedElts.getBitWidth(), 0);
2980 return true;
2981 case ISD::ADD:
2982 case ISD::SUB:
2983 case ISD::AND:
2984 case ISD::XOR:
2985 case ISD::OR: {
2986 APInt UndefLHS, UndefRHS;
2987 SDValue LHS = V.getOperand(0);
2988 SDValue RHS = V.getOperand(1);
2989 // Only recognize splats with the same demanded undef elements for both
2990 // operands, otherwise we might fail to handle binop-specific undef
2991 // handling.
2992 // e.g. (and undef, 0) -> 0 etc.
2993 if (isSplatValue(LHS, DemandedElts, UndefLHS, Depth + 1) &&
2994 isSplatValue(RHS, DemandedElts, UndefRHS, Depth + 1) &&
2995 (DemandedElts & UndefLHS) == (DemandedElts & UndefRHS)) {
2996 UndefElts = UndefLHS | UndefRHS;
2997 return true;
2998 }
2999 return false;
3000 }
3001 case ISD::ABS:
3003 case ISD::TRUNCATE:
3004 case ISD::SIGN_EXTEND:
3005 case ISD::ZERO_EXTEND:
3006 return isSplatValue(V.getOperand(0), DemandedElts, UndefElts, Depth + 1);
3007 default:
3008 if (Opcode >= ISD::BUILTIN_OP_END || Opcode == ISD::INTRINSIC_WO_CHAIN ||
3009 Opcode == ISD::INTRINSIC_W_CHAIN || Opcode == ISD::INTRINSIC_VOID)
3010 return TLI->isSplatValueForTargetNode(V, DemandedElts, UndefElts, *this,
3011 Depth);
3012 break;
3013 }
3014
3015 // We don't support other cases than those above for scalable vectors at
3016 // the moment.
3017 if (VT.isScalableVector())
3018 return false;
3019
3020 unsigned NumElts = VT.getVectorNumElements();
3021 assert(NumElts == DemandedElts.getBitWidth() && "Vector size mismatch");
3022 UndefElts = APInt::getZero(NumElts);
3023
3024 switch (Opcode) {
3025 case ISD::BUILD_VECTOR: {
3026 SDValue Scl;
3027 for (unsigned i = 0; i != NumElts; ++i) {
3028 SDValue Op = V.getOperand(i);
3029 if (Op.isUndef()) {
3030 UndefElts.setBit(i);
3031 continue;
3032 }
3033 if (!DemandedElts[i])
3034 continue;
3035 if (Scl && Scl != Op)
3036 return false;
3037 Scl = Op;
3038 }
3039 return true;
3040 }
3041 case ISD::VECTOR_SHUFFLE: {
3042 // Check if this is a shuffle node doing a splat or a shuffle of a splat.
3043 APInt DemandedLHS = APInt::getZero(NumElts);
3044 APInt DemandedRHS = APInt::getZero(NumElts);
3045 ArrayRef<int> Mask = cast<ShuffleVectorSDNode>(V)->getMask();
3046 for (int i = 0; i != (int)NumElts; ++i) {
3047 int M = Mask[i];
3048 if (M < 0) {
3049 UndefElts.setBit(i);
3050 continue;
3051 }
3052 if (!DemandedElts[i])
3053 continue;
3054 if (M < (int)NumElts)
3055 DemandedLHS.setBit(M);
3056 else
3057 DemandedRHS.setBit(M - NumElts);
3058 }
3059
3060 // If we aren't demanding either op, assume there's no splat.
3061 // If we are demanding both ops, assume there's no splat.
3062 if ((DemandedLHS.isZero() && DemandedRHS.isZero()) ||
3063 (!DemandedLHS.isZero() && !DemandedRHS.isZero()))
3064 return false;
3065
3066 // See if the demanded elts of the source op is a splat or we only demand
3067 // one element, which should always be a splat.
3068 // TODO: Handle source ops splats with undefs.
3069 auto CheckSplatSrc = [&](SDValue Src, const APInt &SrcElts) {
3070 APInt SrcUndefs;
3071 return (SrcElts.popcount() == 1) ||
3072 (isSplatValue(Src, SrcElts, SrcUndefs, Depth + 1) &&
3073 (SrcElts & SrcUndefs).isZero());
3074 };
3075 if (!DemandedLHS.isZero())
3076 return CheckSplatSrc(V.getOperand(0), DemandedLHS);
3077 return CheckSplatSrc(V.getOperand(1), DemandedRHS);
3078 }
3080 // Offset the demanded elts by the subvector index.
3081 SDValue Src = V.getOperand(0);
3082 // We don't support scalable vectors at the moment.
3083 if (Src.getValueType().isScalableVector())
3084 return false;
3085 uint64_t Idx = V.getConstantOperandVal(1);
3086 unsigned NumSrcElts = Src.getValueType().getVectorNumElements();
3087 APInt UndefSrcElts;
3088 APInt DemandedSrcElts = DemandedElts.zext(NumSrcElts).shl(Idx);
3089 if (isSplatValue(Src, DemandedSrcElts, UndefSrcElts, Depth + 1)) {
3090 UndefElts = UndefSrcElts.extractBits(NumElts, Idx);
3091 return true;
3092 }
3093 break;
3094 }
3098 // Widen the demanded elts by the src element count.
3099 SDValue Src = V.getOperand(0);
3100 // We don't support scalable vectors at the moment.
3101 if (Src.getValueType().isScalableVector())
3102 return false;
3103 unsigned NumSrcElts = Src.getValueType().getVectorNumElements();
3104 APInt UndefSrcElts;
3105 APInt DemandedSrcElts = DemandedElts.zext(NumSrcElts);
3106 if (isSplatValue(Src, DemandedSrcElts, UndefSrcElts, Depth + 1)) {
3107 UndefElts = UndefSrcElts.trunc(NumElts);
3108 return true;
3109 }
3110 break;
3111 }
3112 case ISD::BITCAST: {
3113 SDValue Src = V.getOperand(0);
3114 EVT SrcVT = Src.getValueType();
3115 unsigned SrcBitWidth = SrcVT.getScalarSizeInBits();
3116 unsigned BitWidth = VT.getScalarSizeInBits();
3117
3118 // Ignore bitcasts from unsupported types.
3119 // TODO: Add fp support?
3120 if (!SrcVT.isVector() || !SrcVT.isInteger() || !VT.isInteger())
3121 break;
3122
3123 // Bitcast 'small element' vector to 'large element' vector.
3124 if ((BitWidth % SrcBitWidth) == 0) {
3125 // See if each sub element is a splat.
3126 unsigned Scale = BitWidth / SrcBitWidth;
3127 unsigned NumSrcElts = SrcVT.getVectorNumElements();
3128 APInt ScaledDemandedElts =
3129 APIntOps::ScaleBitMask(DemandedElts, NumSrcElts);
3130 for (unsigned I = 0; I != Scale; ++I) {
3131 APInt SubUndefElts;
3132 APInt SubDemandedElt = APInt::getOneBitSet(Scale, I);
3133 APInt SubDemandedElts = APInt::getSplat(NumSrcElts, SubDemandedElt);
3134 SubDemandedElts &= ScaledDemandedElts;
3135 if (!isSplatValue(Src, SubDemandedElts, SubUndefElts, Depth + 1))
3136 return false;
3137 // TODO: Add support for merging sub undef elements.
3138 if (!SubUndefElts.isZero())
3139 return false;
3140 }
3141 return true;
3142 }
3143 break;
3144 }
3145 }
3146
3147 return false;
3148}
3149
3150/// Helper wrapper to main isSplatValue function.
3151bool SelectionDAG::isSplatValue(SDValue V, bool AllowUndefs) const {
3152 EVT VT = V.getValueType();
3153 assert(VT.isVector() && "Vector type expected");
3154
3155 APInt UndefElts;
3156 // Since the number of lanes in a scalable vector is unknown at compile time,
3157 // we track one bit which is implicitly broadcast to all lanes. This means
3158 // that all lanes in a scalable vector are considered demanded.
3159 APInt DemandedElts
3161 return isSplatValue(V, DemandedElts, UndefElts) &&
3162 (AllowUndefs || !UndefElts);
3163}
3164
3167
3168 EVT VT = V.getValueType();
3169 unsigned Opcode = V.getOpcode();
3170 switch (Opcode) {
3171 default: {
3172 APInt UndefElts;
3173 // Since the number of lanes in a scalable vector is unknown at compile time,
3174 // we track one bit which is implicitly broadcast to all lanes. This means
3175 // that all lanes in a scalable vector are considered demanded.
3176 APInt DemandedElts
3178
3179 if (isSplatValue(V, DemandedElts, UndefElts)) {
3180 if (VT.isScalableVector()) {
3181 // DemandedElts and UndefElts are ignored for scalable vectors, since
3182 // the only supported cases are SPLAT_VECTOR nodes.
3183 SplatIdx = 0;
3184 } else {
3185 // Handle case where all demanded elements are UNDEF.
3186 if (DemandedElts.isSubsetOf(UndefElts)) {
3187 SplatIdx = 0;
3188 return getUNDEF(VT);
3189 }
3190 SplatIdx = (UndefElts & DemandedElts).countr_one();
3191 }
3192 return V;
3193 }
3194 break;
3195 }
3196 case ISD::SPLAT_VECTOR:
3197 SplatIdx = 0;
3198 return V;
3199 case ISD::VECTOR_SHUFFLE: {
3200 assert(!VT.isScalableVector());
3201 // Check if this is a shuffle node doing a splat.
3202 // TODO - remove this and rely purely on SelectionDAG::isSplatValue,
3203 // getTargetVShiftNode currently struggles without the splat source.
3204 auto *SVN = cast<ShuffleVectorSDNode>(V);
3205 if (!SVN->isSplat())
3206 break;
3207 int Idx = SVN->getSplatIndex();
3208 int NumElts = V.getValueType().getVectorNumElements();
3209 SplatIdx = Idx % NumElts;
3210 return V.getOperand(Idx / NumElts);
3211 }
3212 }
3213
3214 return SDValue();
3215}
3216
3218 int SplatIdx;
3219 if (SDValue SrcVector = getSplatSourceVector(V, SplatIdx)) {
3220 EVT SVT = SrcVector.getValueType().getScalarType();
3221 EVT LegalSVT = SVT;
3222 if (LegalTypes && !TLI->isTypeLegal(SVT)) {
3223 if (!SVT.isInteger())
3224 return SDValue();
3225 LegalSVT = TLI->getTypeToTransformTo(*getContext(), LegalSVT);
3226 if (LegalSVT.bitsLT(SVT))
3227 return SDValue();
3228 }
3229 return getExtractVectorElt(SDLoc(V), LegalSVT, SrcVector, SplatIdx);
3230 }
3231 return SDValue();
3232}
3233
3234std::optional<ConstantRange>
3236 unsigned Depth) const {
3237 assert((V.getOpcode() == ISD::SHL || V.getOpcode() == ISD::SRL ||
3238 V.getOpcode() == ISD::SRA) &&
3239 "Unknown shift node");
3240 // Shifting more than the bitwidth is not valid.
3241 unsigned BitWidth = V.getScalarValueSizeInBits();
3242
3243 if (auto *Cst = dyn_cast<ConstantSDNode>(V.getOperand(1))) {
3244 const APInt &ShAmt = Cst->getAPIntValue();
3245 if (ShAmt.uge(BitWidth))
3246 return std::nullopt;
3247 return ConstantRange(ShAmt);
3248 }
3249
3250 if (auto *BV = dyn_cast<BuildVectorSDNode>(V.getOperand(1))) {
3251 const APInt *MinAmt = nullptr, *MaxAmt = nullptr;
3252 for (unsigned i = 0, e = BV->getNumOperands(); i != e; ++i) {
3253 if (!DemandedElts[i])
3254 continue;
3255 auto *SA = dyn_cast<ConstantSDNode>(BV->getOperand(i));
3256 if (!SA) {
3257 MinAmt = MaxAmt = nullptr;
3258 break;
3259 }
3260 const APInt &ShAmt = SA->getAPIntValue();
3261 if (ShAmt.uge(BitWidth))
3262 return std::nullopt;
3263 if (!MinAmt || MinAmt->ugt(ShAmt))
3264 MinAmt = &ShAmt;
3265 if (!MaxAmt || MaxAmt->ult(ShAmt))
3266 MaxAmt = &ShAmt;
3267 }
3268 assert(((!MinAmt && !MaxAmt) || (MinAmt && MaxAmt)) &&
3269 "Failed to find matching min/max shift amounts");
3270 if (MinAmt && MaxAmt)
3271 return ConstantRange(*MinAmt, *MaxAmt + 1);
3272 }
3273
3274 // Use computeKnownBits to find a hidden constant/knownbits (usually type
3275 // legalized). e.g. Hidden behind multiple bitcasts/build_vector/casts etc.
3276 KnownBits KnownAmt = computeKnownBits(V.getOperand(1), DemandedElts, Depth);
3277 if (KnownAmt.getMaxValue().ult(BitWidth))
3278 return ConstantRange::fromKnownBits(KnownAmt, /*IsSigned=*/false);
3279
3280 return std::nullopt;
3281}
3282
3283std::optional<unsigned>
3285 unsigned Depth) const {
3286 assert((V.getOpcode() == ISD::SHL || V.getOpcode() == ISD::SRL ||
3287 V.getOpcode() == ISD::SRA) &&
3288 "Unknown shift node");
3289 if (std::optional<ConstantRange> AmtRange =
3290 getValidShiftAmountRange(V, DemandedElts, Depth))
3291 if (const APInt *ShAmt = AmtRange->getSingleElement())
3292 return ShAmt->getZExtValue();
3293 return std::nullopt;
3294}
3295
3296std::optional<unsigned>
3298 APInt DemandedElts = getDemandAllEltsMask(V);
3299 return getValidShiftAmount(V, DemandedElts, Depth);
3300}
3301
3302std::optional<unsigned>
3304 unsigned Depth) const {
3305 assert((V.getOpcode() == ISD::SHL || V.getOpcode() == ISD::SRL ||
3306 V.getOpcode() == ISD::SRA) &&
3307 "Unknown shift node");
3308 if (std::optional<ConstantRange> AmtRange =
3309 getValidShiftAmountRange(V, DemandedElts, Depth))
3310 return AmtRange->getUnsignedMin().getZExtValue();
3311 return std::nullopt;
3312}
3313
3314std::optional<unsigned>
3316 APInt DemandedElts = getDemandAllEltsMask(V);
3317 return getValidMinimumShiftAmount(V, DemandedElts, Depth);
3318}
3319
3320std::optional<unsigned>
3322 unsigned Depth) const {
3323 assert((V.getOpcode() == ISD::SHL || V.getOpcode() == ISD::SRL ||
3324 V.getOpcode() == ISD::SRA) &&
3325 "Unknown shift node");
3326 if (std::optional<ConstantRange> AmtRange =
3327 getValidShiftAmountRange(V, DemandedElts, Depth))
3328 return AmtRange->getUnsignedMax().getZExtValue();
3329 return std::nullopt;
3330}
3331
3332std::optional<unsigned>
3334 APInt DemandedElts = getDemandAllEltsMask(V);
3335 return getValidMaximumShiftAmount(V, DemandedElts, Depth);
3336}
3337
3338/// Determine which bits of Op are known to be either zero or one and return
3339/// them in Known. For vectors, the known bits are those that are shared by
3340/// every vector element.
3342 APInt DemandedElts = getDemandAllEltsMask(Op);
3343 return computeKnownBits(Op, DemandedElts, Depth);
3344}
3345
3346/// Determine which bits of Op are known to be either zero or one and return
3347/// them in Known. The DemandedElts argument allows us to only collect the known
3348/// bits that are shared by the requested vector elements.
3350 unsigned Depth) const {
3351 unsigned BitWidth = Op.getScalarValueSizeInBits();
3352
3353 KnownBits Known(BitWidth); // Don't know anything.
3354
3355 if (auto OptAPInt = Op->bitcastToAPInt()) {
3356 // We know all of the bits for a constant!
3357 return KnownBits::makeConstant(*std::move(OptAPInt));
3358 }
3359
3360 if (Depth >= MaxRecursionDepth)
3361 return Known; // Limit search depth.
3362
3363 KnownBits Known2;
3364 unsigned NumElts = DemandedElts.getBitWidth();
3365 assert((!Op.getValueType().isScalableVector() || NumElts == 1) &&
3366 "DemandedElts for scalable vectors must be 1 to represent all lanes");
3367 assert((!Op.getValueType().isFixedLengthVector() ||
3368 NumElts == Op.getValueType().getVectorNumElements()) &&
3369 "Unexpected vector size");
3370
3371 if (!DemandedElts)
3372 return Known; // No demanded elts, better to assume we don't know anything.
3373
3374 unsigned Opcode = Op.getOpcode();
3375 switch (Opcode) {
3376 case ISD::FREEZE: {
3377 if (isGuaranteedNotToBeUndefOrPoison(Op.getOperand(0), DemandedElts,
3379 Known = computeKnownBits(Op.getOperand(0), DemandedElts, Depth + 1);
3380 break;
3381 }
3382 case ISD::MERGE_VALUES:
3383 return computeKnownBits(Op.getOperand(Op.getResNo()), DemandedElts,
3384 Depth + 1);
3385 case ISD::SPLAT_VECTOR: {
3386 SDValue SrcOp = Op.getOperand(0);
3387 assert(SrcOp.getValueSizeInBits() >= BitWidth &&
3388 "Expected SPLAT_VECTOR implicit truncation");
3389 // Implicitly truncate the bits to match the official semantics of
3390 // SPLAT_VECTOR.
3392 break;
3393 }
3395 unsigned ScalarSize = Op.getOperand(0).getScalarValueSizeInBits();
3396 assert(ScalarSize * Op.getNumOperands() == BitWidth &&
3397 "Expected SPLAT_VECTOR_PARTS scalars to cover element width");
3398 for (auto [I, SrcOp] : enumerate(Op->ops())) {
3399 Known.insertBits(computeKnownBits(SrcOp, Depth + 1), ScalarSize * I);
3400 }
3401 break;
3402 }
3403 case ISD::STEP_VECTOR: {
3404 const APInt &Step = Op.getConstantOperandAPInt(0);
3405
3406 if (Step.isPowerOf2())
3407 Known.Zero.setLowBits(Step.logBase2());
3408
3410
3411 if (!isUIntN(BitWidth, Op.getValueType().getVectorMinNumElements()))
3412 break;
3413 const APInt MinNumElts =
3414 APInt(BitWidth, Op.getValueType().getVectorMinNumElements());
3415
3416 bool Overflow;
3417 const APInt MaxNumElts = getVScaleRange(&F, BitWidth)
3419 .umul_ov(MinNumElts, Overflow);
3420 if (Overflow)
3421 break;
3422
3423 const APInt MaxValue = (MaxNumElts - 1).umul_ov(Step, Overflow);
3424 if (Overflow)
3425 break;
3426
3427 Known.Zero.setHighBits(MaxValue.countl_zero());
3428 break;
3429 }
3430 case ISD::BUILD_VECTOR:
3431 assert(!Op.getValueType().isScalableVector());
3432 // Collect the known bits that are shared by every demanded vector element.
3433 Known.setAllConflict();
3434 for (unsigned i = 0, e = Op.getNumOperands(); i != e; ++i) {
3435 if (!DemandedElts[i])
3436 continue;
3437
3438 SDValue SrcOp = Op.getOperand(i);
3439 if (SrcOp.getOpcode() == ISD::POISON)
3440 continue;
3441
3442 Known2 = computeKnownBits(SrcOp, Depth + 1);
3443
3444 // BUILD_VECTOR can implicitly truncate sources, we must handle this.
3445 if (SrcOp.getValueSizeInBits() != BitWidth) {
3446 assert(SrcOp.getValueSizeInBits() > BitWidth &&
3447 "Expected BUILD_VECTOR implicit truncation");
3448 Known2 = Known2.trunc(BitWidth);
3449 }
3450
3451 // Known bits are the values that are shared by every demanded element.
3452 Known = Known.intersectWith(Known2);
3453
3454 // If we don't know any bits, early out.
3455 if (Known.isUnknown())
3456 break;
3457 }
3458
3459 // If every demanded element was poison, we know nothing.
3460 if (Known.hasConflict())
3461 Known.resetAll();
3462 break;
3463 case ISD::VECTOR_COMPRESS: {
3464 SDValue Vec = Op.getOperand(0);
3465 SDValue PassThru = Op.getOperand(2);
3466 Known = computeKnownBits(PassThru, DemandedElts, Depth + 1);
3467 // If we don't know any bits, early out.
3468 if (Known.isUnknown())
3469 break;
3470 Known2 = computeKnownBits(Vec, Depth + 1);
3471 Known = Known.intersectWith(Known2);
3472 break;
3473 }
3474 case ISD::VECTOR_SHUFFLE: {
3475 assert(!Op.getValueType().isScalableVector());
3476 // Collect the known bits that are shared by every vector element referenced
3477 // by the shuffle.
3478 APInt DemandedLHS, DemandedRHS;
3480 assert(NumElts == SVN->getMask().size() && "Unexpected vector size");
3481 if (!getShuffleDemandedElts(NumElts, SVN->getMask(), DemandedElts,
3482 DemandedLHS, DemandedRHS))
3483 break;
3484
3485 // Known bits are the values that are shared by every demanded element.
3486 Known.setAllConflict();
3487 if (!!DemandedLHS) {
3488 SDValue LHS = Op.getOperand(0);
3489 Known2 = computeKnownBits(LHS, DemandedLHS, Depth + 1);
3490 Known = Known.intersectWith(Known2);
3491 }
3492 // If we don't know any bits, early out.
3493 if (Known.isUnknown())
3494 break;
3495 if (!!DemandedRHS) {
3496 SDValue RHS = Op.getOperand(1);
3497 Known2 = computeKnownBits(RHS, DemandedRHS, Depth + 1);
3498 Known = Known.intersectWith(Known2);
3499 }
3500 break;
3501 }
3502 case ISD::VSCALE: {
3504 const APInt &Multiplier = Op.getConstantOperandAPInt(0);
3506 break;
3507 }
3508 case ISD::CONCAT_VECTORS: {
3509 if (Op.getValueType().isScalableVector())
3510 break;
3511 // Split DemandedElts and test each of the demanded subvectors.
3512 Known.setAllConflict();
3513 EVT SubVectorVT = Op.getOperand(0).getValueType();
3514 unsigned NumSubVectorElts = SubVectorVT.getVectorNumElements();
3515 unsigned NumSubVectors = Op.getNumOperands();
3516 for (unsigned i = 0; i != NumSubVectors; ++i) {
3517 APInt DemandedSub =
3518 DemandedElts.extractBits(NumSubVectorElts, i * NumSubVectorElts);
3519 if (!!DemandedSub) {
3520 SDValue Sub = Op.getOperand(i);
3521 Known2 = computeKnownBits(Sub, DemandedSub, Depth + 1);
3522 Known = Known.intersectWith(Known2);
3523 }
3524 // If we don't know any bits, early out.
3525 if (Known.isUnknown())
3526 break;
3527 }
3528 break;
3529 }
3530 case ISD::INSERT_SUBVECTOR: {
3531 if (Op.getValueType().isScalableVector())
3532 break;
3533 // Demand any elements from the subvector and the remainder from the src its
3534 // inserted into.
3535 SDValue Src = Op.getOperand(0);
3536 SDValue Sub = Op.getOperand(1);
3537 uint64_t Idx = Op.getConstantOperandVal(2);
3538 unsigned NumSubElts = Sub.getValueType().getVectorNumElements();
3539 APInt DemandedSubElts = DemandedElts.extractBits(NumSubElts, Idx);
3540 APInt DemandedSrcElts = DemandedElts;
3541 DemandedSrcElts.clearBits(Idx, Idx + NumSubElts);
3542
3543 Known.setAllConflict();
3544 if (!!DemandedSubElts) {
3545 Known = computeKnownBits(Sub, DemandedSubElts, Depth + 1);
3546 if (Known.isUnknown())
3547 break; // early-out.
3548 }
3549 if (!!DemandedSrcElts) {
3550 Known2 = computeKnownBits(Src, DemandedSrcElts, Depth + 1);
3551 Known = Known.intersectWith(Known2);
3552 }
3553 break;
3554 }
3556 // Offset the demanded elts by the subvector index.
3557 SDValue Src = Op.getOperand(0);
3558
3559 APInt DemandedSrcElts;
3560 if (Src.getValueType().isScalableVector())
3561 DemandedSrcElts = APInt(1, 1); // <=> 'demand all elements'
3562 else {
3563 uint64_t Idx = Op.getConstantOperandVal(1);
3564 unsigned NumSrcElts = Src.getValueType().getVectorNumElements();
3565 DemandedSrcElts = DemandedElts.zext(NumSrcElts).shl(Idx);
3566 }
3567 Known = computeKnownBits(Src, DemandedSrcElts, Depth + 1);
3568 break;
3569 }
3570 case ISD::SCALAR_TO_VECTOR: {
3571 if (Op.getValueType().isScalableVector())
3572 break;
3573 // We know about scalar_to_vector as much as we know about it source,
3574 // which becomes the first element of otherwise unknown vector.
3575 if (DemandedElts != 1)
3576 break;
3577
3578 SDValue N0 = Op.getOperand(0);
3579 Known = computeKnownBits(N0, Depth + 1);
3580 if (N0.getValueSizeInBits() != BitWidth)
3581 Known = Known.trunc(BitWidth);
3582
3583 break;
3584 }
3585 case ISD::BITCAST: {
3586 if (Op.getValueType().isScalableVector())
3587 break;
3588
3589 SDValue N0 = Op.getOperand(0);
3590 EVT SubVT = N0.getValueType();
3591 unsigned SubBitWidth = SubVT.getScalarSizeInBits();
3592
3593 // Ignore bitcasts from unsupported types.
3594 if (!(SubVT.isInteger() || SubVT.isFloatingPoint()))
3595 break;
3596
3597 // Fast handling of 'identity' bitcasts.
3598 if (BitWidth == SubBitWidth) {
3599 Known = computeKnownBits(N0, DemandedElts, Depth + 1);
3600 break;
3601 }
3602
3603 bool IsLE = getDataLayout().isLittleEndian();
3604
3605 // Bitcast 'small element' vector to 'large element' scalar/vector.
3606 if ((BitWidth % SubBitWidth) == 0) {
3607 assert(N0.getValueType().isVector() && "Expected bitcast from vector");
3608
3609 // Collect known bits for the (larger) output by collecting the known
3610 // bits from each set of sub elements and shift these into place.
3611 // We need to separately call computeKnownBits for each set of
3612 // sub elements as the knownbits for each is likely to be different.
3613 unsigned SubScale = BitWidth / SubBitWidth;
3614 APInt SubDemandedElts(NumElts * SubScale, 0);
3615 for (unsigned i = 0; i != NumElts; ++i)
3616 if (DemandedElts[i])
3617 SubDemandedElts.setBit(i * SubScale);
3618
3619 for (unsigned i = 0; i != SubScale; ++i) {
3620 Known2 = computeKnownBits(N0, SubDemandedElts.shl(i),
3621 Depth + 1);
3622 unsigned Shifts = IsLE ? i : SubScale - 1 - i;
3623 Known.insertBits(Known2, SubBitWidth * Shifts);
3624 }
3625 }
3626
3627 // Bitcast 'large element' scalar/vector to 'small element' vector.
3628 if ((SubBitWidth % BitWidth) == 0) {
3629 assert(Op.getValueType().isVector() && "Expected bitcast to vector");
3630
3631 // Collect known bits for the (smaller) output by collecting the known
3632 // bits from the overlapping larger input elements and extracting the
3633 // sub sections we actually care about.
3634 unsigned SubScale = SubBitWidth / BitWidth;
3635 APInt SubDemandedElts =
3636 APIntOps::ScaleBitMask(DemandedElts, NumElts / SubScale);
3637 Known2 = computeKnownBits(N0, SubDemandedElts, Depth + 1);
3638
3639 Known.setAllConflict();
3640 for (unsigned i = 0; i != NumElts; ++i)
3641 if (DemandedElts[i]) {
3642 unsigned Shifts = IsLE ? i : NumElts - 1 - i;
3643 unsigned Offset = (Shifts % SubScale) * BitWidth;
3644 Known = Known.intersectWith(Known2.extractBits(BitWidth, Offset));
3645 // If we don't know any bits, early out.
3646 if (Known.isUnknown())
3647 break;
3648 }
3649 }
3650 break;
3651 }
3652 case ISD::AND:
3653 Known = computeKnownBits(Op.getOperand(1), DemandedElts, Depth + 1);
3654 Known2 = computeKnownBits(Op.getOperand(0), DemandedElts, Depth + 1);
3655
3656 Known &= Known2;
3657 break;
3658 case ISD::OR:
3659 Known = computeKnownBits(Op.getOperand(1), DemandedElts, Depth + 1);
3660 Known2 = computeKnownBits(Op.getOperand(0), DemandedElts, Depth + 1);
3661
3662 Known |= Known2;
3663 break;
3664 case ISD::XOR:
3665 Known = computeKnownBits(Op.getOperand(1), DemandedElts, Depth + 1);
3666 Known2 = computeKnownBits(Op.getOperand(0), DemandedElts, Depth + 1);
3667
3668 Known ^= Known2;
3669 break;
3670 case ISD::MUL: {
3671 Known = computeKnownBits(Op.getOperand(1), DemandedElts, Depth + 1);
3672 Known2 = computeKnownBits(Op.getOperand(0), DemandedElts, Depth + 1);
3673 bool SelfMultiply = Op.getOperand(0) == Op.getOperand(1);
3674 // TODO: SelfMultiply can be poison, but not undef.
3675 if (SelfMultiply)
3676 SelfMultiply &= isGuaranteedNotToBeUndefOrPoison(
3677 Op.getOperand(0), DemandedElts, UndefPoisonKind::UndefOrPoison,
3678 Depth + 1);
3679 Known = KnownBits::mul(Known, Known2, SelfMultiply);
3680
3681 // If the multiplication is known not to overflow, the product of a number
3682 // with itself is non-negative. Only do this if we didn't already computed
3683 // the opposite value for the sign bit.
3684 if (Op->getFlags().hasNoSignedWrap() &&
3685 Op.getOperand(0) == Op.getOperand(1) &&
3686 !Known.isNegative())
3687 Known.makeNonNegative();
3688 break;
3689 }
3690 case ISD::MULHU: {
3691 Known = computeKnownBits(Op.getOperand(1), DemandedElts, Depth + 1);
3692 Known2 = computeKnownBits(Op.getOperand(0), DemandedElts, Depth + 1);
3693 Known = KnownBits::mulhu(Known, Known2);
3694 break;
3695 }
3696 case ISD::MULHS: {
3697 Known = computeKnownBits(Op.getOperand(1), DemandedElts, Depth + 1);
3698 Known2 = computeKnownBits(Op.getOperand(0), DemandedElts, Depth + 1);
3699 Known = KnownBits::mulhs(Known, Known2);
3700 break;
3701 }
3702 case ISD::ABDU: {
3703 Known = computeKnownBits(Op.getOperand(1), DemandedElts, Depth + 1);
3704 Known2 = computeKnownBits(Op.getOperand(0), DemandedElts, Depth + 1);
3705 Known = KnownBits::abdu(Known, Known2);
3706 break;
3707 }
3708 case ISD::ABDS: {
3709 Known = computeKnownBits(Op.getOperand(1), DemandedElts, Depth + 1);
3710 Known2 = computeKnownBits(Op.getOperand(0), DemandedElts, Depth + 1);
3711 Known = KnownBits::abds(Known, Known2);
3712 unsigned SignBits1 =
3713 ComputeNumSignBits(Op.getOperand(1), DemandedElts, Depth + 1);
3714 if (SignBits1 == 1)
3715 break;
3716 unsigned SignBits0 =
3717 ComputeNumSignBits(Op.getOperand(0), DemandedElts, Depth + 1);
3718 Known.Zero.setHighBits(std::min(SignBits0, SignBits1) - 1);
3719 break;
3720 }
3721 case ISD::UMUL_LOHI: {
3722 assert((Op.getResNo() == 0 || Op.getResNo() == 1) && "Unknown result");
3723 Known = computeKnownBits(Op.getOperand(1), DemandedElts, Depth + 1);
3724 Known2 = computeKnownBits(Op.getOperand(0), DemandedElts, Depth + 1);
3725 bool SelfMultiply = Op.getOperand(0) == Op.getOperand(1);
3726 if (Op.getResNo() == 0)
3727 Known = KnownBits::mul(Known, Known2, SelfMultiply);
3728 else
3729 Known = KnownBits::mulhu(Known, Known2);
3730 break;
3731 }
3732 case ISD::SMUL_LOHI: {
3733 assert((Op.getResNo() == 0 || Op.getResNo() == 1) && "Unknown result");
3734 Known = computeKnownBits(Op.getOperand(1), DemandedElts, Depth + 1);
3735 Known2 = computeKnownBits(Op.getOperand(0), DemandedElts, Depth + 1);
3736 bool SelfMultiply = Op.getOperand(0) == Op.getOperand(1);
3737 if (Op.getResNo() == 0)
3738 Known = KnownBits::mul(Known, Known2, SelfMultiply);
3739 else
3740 Known = KnownBits::mulhs(Known, Known2);
3741 break;
3742 }
3743 case ISD::AVGFLOORU: {
3744 Known = computeKnownBits(Op.getOperand(0), DemandedElts, Depth + 1);
3745 Known2 = computeKnownBits(Op.getOperand(1), DemandedElts, Depth + 1);
3746 Known = KnownBits::avgFloorU(Known, Known2);
3747 break;
3748 }
3749 case ISD::AVGCEILU: {
3750 Known = computeKnownBits(Op.getOperand(0), DemandedElts, Depth + 1);
3751 Known2 = computeKnownBits(Op.getOperand(1), DemandedElts, Depth + 1);
3752 Known = KnownBits::avgCeilU(Known, Known2);
3753 break;
3754 }
3755 case ISD::AVGFLOORS: {
3756 Known = computeKnownBits(Op.getOperand(0), DemandedElts, Depth + 1);
3757 Known2 = computeKnownBits(Op.getOperand(1), DemandedElts, Depth + 1);
3758 Known = KnownBits::avgFloorS(Known, Known2);
3759 break;
3760 }
3761 case ISD::AVGCEILS: {
3762 Known = computeKnownBits(Op.getOperand(0), DemandedElts, Depth + 1);
3763 Known2 = computeKnownBits(Op.getOperand(1), DemandedElts, Depth + 1);
3764 Known = KnownBits::avgCeilS(Known, Known2);
3765 break;
3766 }
3767 case ISD::SELECT:
3768 case ISD::VSELECT:
3769 Known = computeKnownBits(Op.getOperand(2), DemandedElts, Depth+1);
3770 // If we don't know any bits, early out.
3771 if (Known.isUnknown())
3772 break;
3773 Known2 = computeKnownBits(Op.getOperand(1), DemandedElts, Depth+1);
3774
3775 // Only known if known in both the LHS and RHS.
3776 Known = Known.intersectWith(Known2);
3777 break;
3778 case ISD::SELECT_CC:
3779 Known = computeKnownBits(Op.getOperand(3), DemandedElts, Depth+1);
3780 // If we don't know any bits, early out.
3781 if (Known.isUnknown())
3782 break;
3783 Known2 = computeKnownBits(Op.getOperand(2), DemandedElts, Depth+1);
3784
3785 // Only known if known in both the LHS and RHS.
3786 Known = Known.intersectWith(Known2);
3787 break;
3788 case ISD::SMULO:
3789 case ISD::UMULO:
3790 if (Op.getResNo() != 1)
3791 break;
3792 // The boolean result conforms to getBooleanContents.
3793 // If we know the result of a setcc has the top bits zero, use this info.
3794 // We know that we have an integer-based boolean since these operations
3795 // are only available for integer.
3796 if (TLI->getBooleanContents(Op.getValueType().isVector(), false) ==
3798 BitWidth > 1)
3799 Known.Zero.setBitsFrom(1);
3800 break;
3801 case ISD::SETCC:
3802 case ISD::SETCCCARRY:
3803 case ISD::STRICT_FSETCC:
3804 case ISD::STRICT_FSETCCS: {
3805 unsigned OpNo = Op->isStrictFPOpcode() ? 1 : 0;
3806 // If we know the result of a setcc has the top bits zero, use this info.
3807 if (TLI->getBooleanContents(Op.getOperand(OpNo).getValueType()) ==
3809 BitWidth > 1)
3810 Known.Zero.setBitsFrom(1);
3811 break;
3812 }
3813 case ISD::SHL: {
3814 Known = computeKnownBits(Op.getOperand(0), DemandedElts, Depth + 1);
3815 Known2 = computeKnownBits(Op.getOperand(1), DemandedElts, Depth + 1);
3816
3817 bool NUW = Op->getFlags().hasNoUnsignedWrap();
3818 bool NSW = Op->getFlags().hasNoSignedWrap();
3819
3820 bool ShAmtNonZero = Known2.isNonZero();
3821
3822 Known = KnownBits::shl(Known, Known2, NUW, NSW, ShAmtNonZero);
3823
3824 // Minimum shift low bits are known zero.
3825 if (std::optional<unsigned> ShMinAmt =
3826 getValidMinimumShiftAmount(Op, DemandedElts, Depth + 1))
3827 Known.Zero.setLowBits(*ShMinAmt);
3828 break;
3829 }
3830 case ISD::SRL:
3831 Known = computeKnownBits(Op.getOperand(0), DemandedElts, Depth + 1);
3832 Known2 = computeKnownBits(Op.getOperand(1), DemandedElts, Depth + 1);
3833 Known = KnownBits::lshr(Known, Known2, /*ShAmtNonZero=*/false,
3834 Op->getFlags().hasExact());
3835
3836 // Minimum shift high bits are known zero.
3837 if (std::optional<unsigned> ShMinAmt =
3838 getValidMinimumShiftAmount(Op, DemandedElts, Depth + 1))
3839 Known.Zero.setHighBits(*ShMinAmt);
3840 break;
3841 case ISD::SRA:
3842 Known = computeKnownBits(Op.getOperand(0), DemandedElts, Depth + 1);
3843 Known2 = computeKnownBits(Op.getOperand(1), DemandedElts, Depth + 1);
3844 Known = KnownBits::ashr(Known, Known2, /*ShAmtNonZero=*/false,
3845 Op->getFlags().hasExact());
3846 break;
3847 case ISD::ROTL:
3848 case ISD::ROTR:
3849 if (ConstantSDNode *C =
3850 isConstOrConstSplat(Op.getOperand(1), DemandedElts)) {
3851 unsigned Amt = C->getAPIntValue().urem(BitWidth);
3852
3853 Known = computeKnownBits(Op.getOperand(0), DemandedElts, Depth + 1);
3854
3855 // Canonicalize to ROTR.
3856 if (Opcode == ISD::ROTL && Amt != 0)
3857 Amt = BitWidth - Amt;
3858
3859 Known.Zero = Known.Zero.rotr(Amt);
3860 Known.One = Known.One.rotr(Amt);
3861 }
3862 break;
3863 case ISD::FSHL:
3864 case ISD::FSHR:
3865 if (ConstantSDNode *C = isConstOrConstSplat(Op.getOperand(2), DemandedElts)) {
3866 unsigned Amt = C->getAPIntValue().urem(BitWidth);
3867
3868 // For fshl, 0-shift returns the 1st arg.
3869 // For fshr, 0-shift returns the 2nd arg.
3870 if (Amt == 0) {
3871 Known = computeKnownBits(Op.getOperand(Opcode == ISD::FSHL ? 0 : 1),
3872 DemandedElts, Depth + 1);
3873 break;
3874 }
3875
3876 // fshl: (X << (Z % BW)) | (Y >> (BW - (Z % BW)))
3877 // fshr: (X << (BW - (Z % BW))) | (Y >> (Z % BW))
3878 const APInt ShAmt(BitWidth, Amt);
3879 Known = computeKnownBits(Op.getOperand(0), DemandedElts, Depth + 1);
3880 Known2 = computeKnownBits(Op.getOperand(1), DemandedElts, Depth + 1);
3881 Known = Opcode == ISD::FSHL ? KnownBits::fshl(Known, Known2, ShAmt)
3882 : KnownBits::fshr(Known, Known2, ShAmt);
3883 }
3884 break;
3885 case ISD::SHL_PARTS:
3886 case ISD::SRA_PARTS:
3887 case ISD::SRL_PARTS: {
3888 assert((Op.getResNo() == 0 || Op.getResNo() == 1) && "Unknown result");
3889
3890 // Collect lo/hi source values and concatenate.
3891 unsigned LoBits = Op.getOperand(0).getScalarValueSizeInBits();
3892 unsigned HiBits = Op.getOperand(1).getScalarValueSizeInBits();
3893 Known = computeKnownBits(Op.getOperand(0), DemandedElts, Depth + 1);
3894 Known2 = computeKnownBits(Op.getOperand(1), DemandedElts, Depth + 1);
3895 Known = Known2.concat(Known);
3896
3897 // Collect shift amount.
3898 Known2 = computeKnownBits(Op.getOperand(2), DemandedElts, Depth + 1);
3899
3900 if (Opcode == ISD::SHL_PARTS)
3901 Known = KnownBits::shl(Known, Known2);
3902 else if (Opcode == ISD::SRA_PARTS)
3903 Known = KnownBits::ashr(Known, Known2);
3904 else // if (Opcode == ISD::SRL_PARTS)
3905 Known = KnownBits::lshr(Known, Known2);
3906
3907 // TODO: Minimum shift low/high bits are known zero.
3908
3909 if (Op.getResNo() == 0)
3910 Known = Known.extractBits(LoBits, 0);
3911 else
3912 Known = Known.extractBits(HiBits, LoBits);
3913 break;
3914 }
3916 Known = computeKnownBits(Op.getOperand(0), DemandedElts, Depth + 1);
3917 EVT EVT = cast<VTSDNode>(Op.getOperand(1))->getVT();
3918 Known = Known.sextInReg(EVT.getScalarSizeInBits());
3919 break;
3920 }
3921 case ISD::CTTZ:
3922 case ISD::CTTZ_ZERO_POISON: {
3923 Known2 = computeKnownBits(Op.getOperand(0), DemandedElts, Depth + 1);
3924 // If we have a known 1, its position is our upper bound.
3925 unsigned PossibleTZ = Known2.countMaxTrailingZeros();
3926 unsigned LowBits = llvm::bit_width(PossibleTZ);
3927 Known.Zero.setBitsFrom(LowBits);
3928 break;
3929 }
3930 case ISD::CTLZ:
3931 case ISD::CTLZ_ZERO_POISON: {
3932 Known2 = computeKnownBits(Op.getOperand(0), DemandedElts, Depth + 1);
3933 // If we have a known 1, its position is our upper bound.
3934 unsigned PossibleLZ = Known2.countMaxLeadingZeros();
3935 unsigned LowBits = llvm::bit_width(PossibleLZ);
3936 Known.Zero.setBitsFrom(LowBits);
3937 break;
3938 }
3939 case ISD::CTLS: {
3940 unsigned MinRedundantSignBits =
3941 ComputeNumSignBits(Op.getOperand(0), DemandedElts, Depth + 1) - 1;
3942 ConstantRange Range(APInt(BitWidth, MinRedundantSignBits),
3944 Known = Range.toKnownBits();
3945 break;
3946 }
3947 case ISD::CTPOP: {
3948 Known2 = computeKnownBits(Op.getOperand(0), DemandedElts, Depth + 1);
3949 // If we know some of the bits are zero, they can't be one.
3950 unsigned PossibleOnes = Known2.countMaxPopulation();
3951 Known.Zero.setBitsFrom(llvm::bit_width(PossibleOnes));
3952 break;
3953 }
3954 case ISD::PARITY: {
3955 // Parity returns 0 everywhere but the LSB.
3956 Known.Zero.setBitsFrom(1);
3957 break;
3958 }
3959 case ISD::PDEP: {
3960 Known = computeKnownBits(Op.getOperand(1), DemandedElts, Depth + 1);
3961 Known2 = computeKnownBits(Op.getOperand(0), DemandedElts, Depth + 1);
3962 Known = KnownBits::pdep(Known2, Known);
3963 break;
3964 }
3965 case ISD::PEXT: {
3966 Known = computeKnownBits(Op.getOperand(1), DemandedElts, Depth + 1);
3967 Known2 = computeKnownBits(Op.getOperand(0), DemandedElts, Depth + 1);
3968 Known = KnownBits::pext(Known2, Known);
3969 break;
3970 }
3971 case ISD::CLMUL: {
3972 Known = computeKnownBits(Op.getOperand(1), DemandedElts, Depth + 1);
3973 Known2 = computeKnownBits(Op.getOperand(0), DemandedElts, Depth + 1);
3974 Known = KnownBits::clmul(Known, Known2);
3975 break;
3976 }
3977 case ISD::MGATHER:
3978 case ISD::MLOAD: {
3979 ISD::LoadExtType ETy =
3980 (Opcode == ISD::MGATHER)
3981 ? cast<MaskedGatherSDNode>(Op)->getExtensionType()
3982 : cast<MaskedLoadSDNode>(Op)->getExtensionType();
3983 if (ETy == ISD::ZEXTLOAD) {
3984 EVT MemVT = cast<MemSDNode>(Op)->getMemoryVT();
3985 KnownBits Known0(MemVT.getScalarSizeInBits());
3986 return Known0.zext(BitWidth);
3987 }
3988 break;
3989 }
3990 case ISD::LOAD: {
3992 const Constant *Cst = TLI->getTargetConstantFromLoad(LD);
3993 if (ISD::isNON_EXTLoad(LD) && Cst) {
3994 // Determine any common known bits from the loaded constant pool value.
3995 Type *CstTy = Cst->getType();
3996 if ((NumElts * BitWidth) == CstTy->getPrimitiveSizeInBits() &&
3997 !Op.getValueType().isScalableVector()) {
3998 // If its a vector splat, then we can (quickly) reuse the scalar path.
3999 // NOTE: We assume all elements match and none are UNDEF.
4000 if (CstTy->isVectorTy()) {
4001 if (const Constant *Splat = Cst->getSplatValue()) {
4002 Cst = Splat;
4003 CstTy = Cst->getType();
4004 }
4005 }
4006 // TODO - do we need to handle different bitwidths?
4007 if (CstTy->isVectorTy() && BitWidth == CstTy->getScalarSizeInBits()) {
4008 // Iterate across all vector elements finding common known bits.
4009 Known.setAllConflict();
4010 for (unsigned i = 0; i != NumElts; ++i) {
4011 if (!DemandedElts[i])
4012 continue;
4013 if (Constant *Elt = Cst->getAggregateElement(i)) {
4014 if (auto *CInt = dyn_cast<ConstantInt>(Elt)) {
4015 const APInt &Value = CInt->getValue();
4016 Known.One &= Value;
4017 Known.Zero &= ~Value;
4018 continue;
4019 }
4020 if (auto *CFP = dyn_cast<ConstantFP>(Elt)) {
4021 APInt Value = CFP->getValueAPF().bitcastToAPInt();
4022 Known.One &= Value;
4023 Known.Zero &= ~Value;
4024 continue;
4025 }
4026 }
4027 Known.One.clearAllBits();
4028 Known.Zero.clearAllBits();
4029 break;
4030 }
4031 } else if (BitWidth == CstTy->getPrimitiveSizeInBits()) {
4032 if (auto *CInt = dyn_cast<ConstantInt>(Cst)) {
4033 Known = KnownBits::makeConstant(CInt->getValue());
4034 } else if (auto *CFP = dyn_cast<ConstantFP>(Cst)) {
4035 Known =
4036 KnownBits::makeConstant(CFP->getValueAPF().bitcastToAPInt());
4037 }
4038 }
4039 }
4040 } else if (Op.getResNo() == 0) {
4041 unsigned ScalarMemorySize = LD->getMemoryVT().getScalarSizeInBits();
4042 KnownBits KnownScalarMemory(ScalarMemorySize);
4043 if (const MDNode *MD = LD->getRanges())
4044 computeKnownBitsFromRangeMetadata(*MD, KnownScalarMemory);
4045
4046 // Extend the Known bits from memory to the size of the scalar result.
4047 if (ISD::isZEXTLoad(Op.getNode()))
4048 Known = KnownScalarMemory.zext(BitWidth);
4049 else if (ISD::isSEXTLoad(Op.getNode()))
4050 Known = KnownScalarMemory.sext(BitWidth);
4051 else if (ISD::isEXTLoad(Op.getNode()))
4052 Known = KnownScalarMemory.anyext(BitWidth);
4053 else
4054 Known = KnownScalarMemory;
4055 assert(Known.getBitWidth() == BitWidth);
4056 return Known;
4057 }
4058 break;
4059 }
4061 if (Op.getValueType().isScalableVector())
4062 break;
4063 EVT InVT = Op.getOperand(0).getValueType();
4064 APInt InDemandedElts = DemandedElts.zext(InVT.getVectorNumElements());
4065 Known = computeKnownBits(Op.getOperand(0), InDemandedElts, Depth + 1);
4066 Known = Known.zext(BitWidth);
4067 break;
4068 }
4069 case ISD::ZERO_EXTEND: {
4070 Known = computeKnownBits(Op.getOperand(0), DemandedElts, Depth + 1);
4071 Known = Known.zext(BitWidth);
4072 break;
4073 }
4075 if (Op.getValueType().isScalableVector())
4076 break;
4077 EVT InVT = Op.getOperand(0).getValueType();
4078 APInt InDemandedElts = DemandedElts.zext(InVT.getVectorNumElements());
4079 Known = computeKnownBits(Op.getOperand(0), InDemandedElts, Depth + 1);
4080 // If the sign bit is known to be zero or one, then sext will extend
4081 // it to the top bits, else it will just zext.
4082 Known = Known.sext(BitWidth);
4083 break;
4084 }
4085 case ISD::SIGN_EXTEND: {
4086 Known = computeKnownBits(Op.getOperand(0), DemandedElts, Depth + 1);
4087 // If the sign bit is known to be zero or one, then sext will extend
4088 // it to the top bits, else it will just zext.
4089 Known = Known.sext(BitWidth);
4090 break;
4091 }
4093 if (Op.getValueType().isScalableVector())
4094 break;
4095 EVT InVT = Op.getOperand(0).getValueType();
4096 APInt InDemandedElts = DemandedElts.zext(InVT.getVectorNumElements());
4097 Known = computeKnownBits(Op.getOperand(0), InDemandedElts, Depth + 1);
4098 Known = Known.anyext(BitWidth);
4099 break;
4100 }
4101 case ISD::ANY_EXTEND: {
4102 Known = computeKnownBits(Op.getOperand(0), DemandedElts, Depth + 1);
4103 Known = Known.anyext(BitWidth);
4104 break;
4105 }
4106 case ISD::TRUNCATE: {
4107 Known = computeKnownBits(Op.getOperand(0), DemandedElts, Depth + 1);
4108 Known = Known.trunc(BitWidth);
4109 break;
4110 }
4111 case ISD::TRUNCATE_SSAT_S: {
4112 Known = computeKnownBits(Op.getOperand(0), DemandedElts, Depth + 1);
4113 Known = Known.truncSSat(BitWidth);
4114 break;
4115 }
4116 case ISD::TRUNCATE_SSAT_U: {
4117 Known = computeKnownBits(Op.getOperand(0), DemandedElts, Depth + 1);
4118 Known = Known.truncSSatU(BitWidth);
4119 break;
4120 }
4121 case ISD::TRUNCATE_USAT_U: {
4122 Known = computeKnownBits(Op.getOperand(0), DemandedElts, Depth + 1);
4123 Known = Known.truncUSat(BitWidth);
4124 break;
4125 }
4126 case ISD::AssertZext: {
4127 EVT VT = cast<VTSDNode>(Op.getOperand(1))->getVT();
4129 Known = computeKnownBits(Op.getOperand(0), DemandedElts, Depth + 1);
4130 Known.Zero |= (~InMask);
4131 Known.One &= (~Known.Zero);
4132 break;
4133 }
4134 case ISD::AssertAlign: {
4135 unsigned LogOfAlign = Log2(cast<AssertAlignSDNode>(Op)->getAlign());
4136 assert(LogOfAlign != 0);
4137
4138 // TODO: Should use maximum with source
4139 // If a node is guaranteed to be aligned, set low zero bits accordingly as
4140 // well as clearing one bits.
4141 Known.Zero.setLowBits(LogOfAlign);
4142 Known.One.clearLowBits(LogOfAlign);
4143 break;
4144 }
4145 case ISD::AssertNoFPClass: {
4146 Known = computeKnownBits(Op.getOperand(0), DemandedElts, Depth + 1);
4147
4148 FPClassTest NoFPClass =
4149 static_cast<FPClassTest>(Op.getConstantOperandVal(1));
4150 const FPClassTest NegativeTestMask = fcNan | fcNegative;
4151 if ((NoFPClass & NegativeTestMask) == NegativeTestMask) {
4152 // Cannot be negative.
4153 Known.makeNonNegative();
4154 }
4155
4156 const FPClassTest PositiveTestMask = fcNan | fcPositive;
4157 if ((NoFPClass & PositiveTestMask) == PositiveTestMask) {
4158 // Cannot be positive.
4159 Known.makeNegative();
4160 }
4161
4162 break;
4163 }
4164 case ISD::FABS:
4165 // fabs clears the sign bit
4166 Known = computeKnownBits(Op.getOperand(0), DemandedElts, Depth + 1);
4167 Known.makeNonNegative();
4168 break;
4169 case ISD::FGETSIGN:
4170 // All bits are zero except the low bit.
4171 Known.Zero.setBitsFrom(1);
4172 break;
4173 case ISD::ADD: {
4174 SDNodeFlags Flags = Op.getNode()->getFlags();
4175 Known = computeKnownBits(Op.getOperand(0), DemandedElts, Depth + 1);
4176 Known2 = computeKnownBits(Op.getOperand(1), DemandedElts, Depth + 1);
4177 bool SelfAdd = Op.getOperand(0) == Op.getOperand(1) &&
4179 Op.getOperand(0), DemandedElts,
4181 Known = KnownBits::add(Known, Known2, Flags.hasNoSignedWrap(),
4182 Flags.hasNoUnsignedWrap(), SelfAdd);
4183 break;
4184 }
4185 case ISD::SUB: {
4186 SDNodeFlags Flags = Op.getNode()->getFlags();
4187 Known = computeKnownBits(Op.getOperand(0), DemandedElts, Depth + 1);
4188 Known2 = computeKnownBits(Op.getOperand(1), DemandedElts, Depth + 1);
4189 Known = KnownBits::sub(Known, Known2, Flags.hasNoSignedWrap(),
4190 Flags.hasNoUnsignedWrap());
4191 break;
4192 }
4193 case ISD::USUBO:
4194 case ISD::SSUBO:
4195 case ISD::USUBO_CARRY:
4196 case ISD::SSUBO_CARRY:
4197 if (Op.getResNo() == 1) {
4198 // If we know the result of a setcc has the top bits zero, use this info.
4199 if (TLI->getBooleanContents(Op.getOperand(0).getValueType()) ==
4201 BitWidth > 1)
4202 Known.Zero.setBitsFrom(1);
4203 break;
4204 }
4205 [[fallthrough]];
4206 case ISD::SUBC: {
4207 assert(Op.getResNo() == 0 &&
4208 "We only compute knownbits for the difference here.");
4209
4210 // With USUBO_CARRY and SSUBO_CARRY a borrow bit may be added in.
4211 KnownBits Borrow(1);
4212 if (Opcode == ISD::USUBO_CARRY || Opcode == ISD::SSUBO_CARRY) {
4213 Borrow = computeKnownBits(Op.getOperand(2), DemandedElts, Depth + 1);
4214 // Borrow has bit width 1
4215 Borrow = Borrow.trunc(1);
4216 } else {
4217 Borrow.setAllZero();
4218 }
4219
4220 Known = computeKnownBits(Op.getOperand(0), DemandedElts, Depth + 1);
4221 Known2 = computeKnownBits(Op.getOperand(1), DemandedElts, Depth + 1);
4222 Known = KnownBits::computeForSubBorrow(Known, Known2, Borrow);
4223 break;
4224 }
4225 case ISD::UADDO:
4226 case ISD::SADDO:
4227 case ISD::UADDO_CARRY:
4228 case ISD::SADDO_CARRY:
4229 if (Op.getResNo() == 1) {
4230 // If we know the result of a setcc has the top bits zero, use this info.
4231 if (TLI->getBooleanContents(Op.getOperand(0).getValueType()) ==
4233 BitWidth > 1)
4234 Known.Zero.setBitsFrom(1);
4235 break;
4236 }
4237 [[fallthrough]];
4238 case ISD::ADDC:
4239 case ISD::ADDE: {
4240 assert(Op.getResNo() == 0 && "We only compute knownbits for the sum here.");
4241
4242 // With ADDE and UADDO_CARRY, a carry bit may be added in.
4243 KnownBits Carry(1);
4244 if (Opcode == ISD::ADDE)
4245 // Can't track carry from glue, set carry to unknown.
4246 Carry.resetAll();
4247 else if (Opcode == ISD::UADDO_CARRY || Opcode == ISD::SADDO_CARRY) {
4248 Carry = computeKnownBits(Op.getOperand(2), DemandedElts, Depth + 1);
4249 // Carry has bit width 1
4250 Carry = Carry.trunc(1);
4251 } else {
4252 Carry.setAllZero();
4253 }
4254
4255 Known = computeKnownBits(Op.getOperand(0), DemandedElts, Depth + 1);
4256 Known2 = computeKnownBits(Op.getOperand(1), DemandedElts, Depth + 1);
4257 Known = KnownBits::computeForAddCarry(Known, Known2, Carry);
4258 break;
4259 }
4260 case ISD::UDIV: {
4261 Known = computeKnownBits(Op.getOperand(0), DemandedElts, Depth + 1);
4262 Known2 = computeKnownBits(Op.getOperand(1), DemandedElts, Depth + 1);
4263 Known = KnownBits::udiv(Known, Known2, Op->getFlags().hasExact());
4264 break;
4265 }
4266 case ISD::SDIV: {
4267 Known = computeKnownBits(Op.getOperand(0), DemandedElts, Depth + 1);
4268 Known2 = computeKnownBits(Op.getOperand(1), DemandedElts, Depth + 1);
4269 Known = KnownBits::sdiv(Known, Known2, Op->getFlags().hasExact());
4270 break;
4271 }
4272 case ISD::SREM: {
4273 Known = computeKnownBits(Op.getOperand(0), DemandedElts, Depth + 1);
4274 Known2 = computeKnownBits(Op.getOperand(1), DemandedElts, Depth + 1);
4275 Known = KnownBits::srem(Known, Known2);
4276 break;
4277 }
4278 case ISD::UREM: {
4279 Known = computeKnownBits(Op.getOperand(0), DemandedElts, Depth + 1);
4280 Known2 = computeKnownBits(Op.getOperand(1), DemandedElts, Depth + 1);
4281 Known = KnownBits::urem(Known, Known2);
4282 break;
4283 }
4284 case ISD::EXTRACT_ELEMENT: {
4285 Known = computeKnownBits(Op.getOperand(0), Depth+1);
4286 const unsigned Index = Op.getConstantOperandVal(1);
4287 const unsigned EltBitWidth = Op.getValueSizeInBits();
4288
4289 Known = Known.extractBits(EltBitWidth, Index * EltBitWidth);
4290 break;
4291 }
4293 SDValue InVec = Op.getOperand(0);
4294 SDValue EltNo = Op.getOperand(1);
4295 EVT VecVT = InVec.getValueType();
4296 // computeKnownBits not yet implemented for scalable vectors.
4297 if (VecVT.isScalableVector())
4298 break;
4299 const unsigned EltBitWidth = VecVT.getScalarSizeInBits();
4300 const unsigned NumSrcElts = VecVT.getVectorNumElements();
4301
4302 // If BitWidth > EltBitWidth the value is anyext:ed. So we do not know
4303 // anything about the extended bits.
4304 if (BitWidth > EltBitWidth)
4305 Known = Known.trunc(EltBitWidth);
4306
4307 // If we know the element index, just demand that vector element, else for
4308 // an unknown element index, ignore DemandedElts and demand them all.
4309 APInt DemandedSrcElts = APInt::getAllOnes(NumSrcElts);
4310 auto *ConstEltNo = dyn_cast<ConstantSDNode>(EltNo);
4311 if (ConstEltNo && ConstEltNo->getAPIntValue().ult(NumSrcElts))
4312 DemandedSrcElts =
4313 APInt::getOneBitSet(NumSrcElts, ConstEltNo->getZExtValue());
4314
4315 Known = computeKnownBits(InVec, DemandedSrcElts, Depth + 1);
4316 if (BitWidth > EltBitWidth)
4317 Known = Known.anyext(BitWidth);
4318 break;
4319 }
4320 case ISD::BUILD_PAIR: {
4321 // Operand 0 is the low half and operand 1 the high half,
4322 // KnownBits::concat places its argument in the low bits.
4323 Known = computeKnownBits(Op.getOperand(0), Depth + 1);
4324 Known2 = computeKnownBits(Op.getOperand(1), Depth + 1);
4325 Known = Known2.concat(Known);
4326 break;
4327 }
4329 if (Op.getValueType().isScalableVector())
4330 break;
4331
4332 // If we know the element index, split the demand between the
4333 // source vector and the inserted element, otherwise assume we need
4334 // the original demanded vector elements and the value.
4335 SDValue InVec = Op.getOperand(0);
4336 SDValue InVal = Op.getOperand(1);
4337 SDValue EltNo = Op.getOperand(2);
4338 bool DemandedVal = true;
4339 APInt DemandedVecElts = DemandedElts;
4340 auto *CEltNo = dyn_cast<ConstantSDNode>(EltNo);
4341 if (CEltNo && CEltNo->getAPIntValue().ult(NumElts)) {
4342 unsigned EltIdx = CEltNo->getZExtValue();
4343 DemandedVal = !!DemandedElts[EltIdx];
4344 DemandedVecElts.clearBit(EltIdx);
4345 }
4346 Known.setAllConflict();
4347 if (DemandedVal) {
4348 Known2 = computeKnownBits(InVal, Depth + 1);
4349 Known = Known.intersectWith(Known2.zextOrTrunc(BitWidth));
4350 }
4351 if (!!DemandedVecElts) {
4352 Known2 = computeKnownBits(InVec, DemandedVecElts, Depth + 1);
4353 Known = Known.intersectWith(Known2);
4354 }
4355 break;
4356 }
4357 case ISD::BITREVERSE: {
4358 Known2 = computeKnownBits(Op.getOperand(0), DemandedElts, Depth + 1);
4359 Known = Known2.reverseBits();
4360 break;
4361 }
4362 case ISD::BSWAP: {
4363 Known2 = computeKnownBits(Op.getOperand(0), DemandedElts, Depth + 1);
4364 Known = Known2.byteSwap();
4365 break;
4366 }
4367 case ISD::ABS:
4368 case ISD::ABS_MIN_POISON: {
4369 Known2 = computeKnownBits(Op.getOperand(0), DemandedElts, Depth + 1);
4370 Known = Known2.abs();
4371 Known.Zero.setHighBits(
4372 ComputeNumSignBits(Op.getOperand(0), DemandedElts, Depth + 1) - 1);
4373 break;
4374 }
4375 case ISD::USUBSAT: {
4376 Known = computeKnownBits(Op.getOperand(0), DemandedElts, Depth + 1);
4377 Known2 = computeKnownBits(Op.getOperand(1), DemandedElts, Depth + 1);
4378 Known = KnownBits::usub_sat(Known, Known2);
4379 break;
4380 }
4381 case ISD::UMIN: {
4382 Known = computeKnownBits(Op.getOperand(0), DemandedElts, Depth + 1);
4383 Known2 = computeKnownBits(Op.getOperand(1), DemandedElts, Depth + 1);
4384 Known = KnownBits::umin(Known, Known2);
4385 break;
4386 }
4387 case ISD::UMAX: {
4388 Known = computeKnownBits(Op.getOperand(0), DemandedElts, Depth + 1);
4389 Known2 = computeKnownBits(Op.getOperand(1), DemandedElts, Depth + 1);
4390 Known = KnownBits::umax(Known, Known2);
4391 break;
4392 }
4393 case ISD::SMIN:
4394 case ISD::SMAX: {
4395 // If we have a clamp pattern, we know that the number of sign bits will be
4396 // the minimum of the clamp min/max range.
4397 bool IsMax = (Opcode == ISD::SMAX);
4398 ConstantSDNode *CstLow = nullptr, *CstHigh = nullptr;
4399 if ((CstLow = isConstOrConstSplat(Op.getOperand(1), DemandedElts)))
4400 if (Op.getOperand(0).getOpcode() == (IsMax ? ISD::SMIN : ISD::SMAX))
4401 CstHigh =
4402 isConstOrConstSplat(Op.getOperand(0).getOperand(1), DemandedElts);
4403 if (CstLow && CstHigh) {
4404 if (!IsMax)
4405 std::swap(CstLow, CstHigh);
4406
4407 const APInt &ValueLow = CstLow->getAPIntValue();
4408 const APInt &ValueHigh = CstHigh->getAPIntValue();
4409 if (ValueLow.sle(ValueHigh)) {
4410 unsigned LowSignBits = ValueLow.getNumSignBits();
4411 unsigned HighSignBits = ValueHigh.getNumSignBits();
4412 unsigned MinSignBits = std::min(LowSignBits, HighSignBits);
4413 if (ValueLow.isNegative() && ValueHigh.isNegative()) {
4414 Known.One.setHighBits(MinSignBits);
4415 break;
4416 }
4417 if (ValueLow.isNonNegative() && ValueHigh.isNonNegative()) {
4418 Known.Zero.setHighBits(MinSignBits);
4419 break;
4420 }
4421 }
4422 }
4423
4424 Known = computeKnownBits(Op.getOperand(0), DemandedElts, Depth + 1);
4425 Known2 = computeKnownBits(Op.getOperand(1), DemandedElts, Depth + 1);
4426 if (IsMax)
4427 Known = KnownBits::smax(Known, Known2);
4428 else
4429 Known = KnownBits::smin(Known, Known2);
4430
4431 // For SMAX, if CstLow is non-negative we know the result will be
4432 // non-negative and thus all sign bits are 0.
4433 // TODO: There's an equivalent of this for smin with negative constant for
4434 // known ones.
4435 if (IsMax && CstLow) {
4436 const APInt &ValueLow = CstLow->getAPIntValue();
4437 if (ValueLow.isNonNegative()) {
4438 unsigned SignBits = ComputeNumSignBits(Op.getOperand(0), Depth + 1);
4439 Known.Zero.setHighBits(std::min(SignBits, ValueLow.getNumSignBits()));
4440 }
4441 }
4442
4443 break;
4444 }
4445 case ISD::UINT_TO_FP: {
4446 Known.makeNonNegative();
4447 break;
4448 }
4449 case ISD::SINT_TO_FP: {
4450 Known2 = computeKnownBits(Op.getOperand(0), DemandedElts, Depth + 1);
4451 if (Known2.isNonNegative())
4452 Known.makeNonNegative();
4453 else if (Known2.isNegative())
4454 Known.makeNegative();
4455 break;
4456 }
4457 case ISD::FP_TO_UINT_SAT: {
4458 // FP_TO_UINT_SAT produces an unsigned value that fits in the saturating VT.
4459 EVT VT = cast<VTSDNode>(Op.getOperand(1))->getVT();
4461 break;
4462 }
4463 case ISD::ATOMIC_LOAD: {
4464 // If we are looking at the loaded value.
4465 if (Op.getResNo() == 0) {
4466 auto *AT = cast<AtomicSDNode>(Op);
4467 unsigned ScalarMemorySize = AT->getMemoryVT().getScalarSizeInBits();
4468 KnownBits KnownScalarMemory(ScalarMemorySize);
4469 if (const MDNode *MD = AT->getRanges())
4470 computeKnownBitsFromRangeMetadata(*MD, KnownScalarMemory);
4471
4472 switch (AT->getExtensionType()) {
4473 case ISD::ZEXTLOAD:
4474 Known = KnownScalarMemory.zext(BitWidth);
4475 break;
4476 case ISD::SEXTLOAD:
4477 Known = KnownScalarMemory.sext(BitWidth);
4478 break;
4479 case ISD::EXTLOAD:
4480 switch (TLI->getExtendForAtomicOps()) {
4481 case ISD::ZERO_EXTEND:
4482 Known = KnownScalarMemory.zext(BitWidth);
4483 break;
4484 case ISD::SIGN_EXTEND:
4485 Known = KnownScalarMemory.sext(BitWidth);
4486 break;
4487 default:
4488 Known = KnownScalarMemory.anyext(BitWidth);
4489 break;
4490 }
4491 break;
4492 case ISD::NON_EXTLOAD:
4493 Known = KnownScalarMemory;
4494 break;
4495 }
4496 assert(Known.getBitWidth() == BitWidth);
4497 }
4498 break;
4499 }
4501 if (Op.getResNo() == 1) {
4502 // The boolean result conforms to getBooleanContents.
4503 // If we know the result of a setcc has the top bits zero, use this info.
4504 // We know that we have an integer-based boolean since these operations
4505 // are only available for integer.
4506 if (TLI->getBooleanContents(Op.getValueType().isVector(), false) ==
4508 BitWidth > 1)
4509 Known.Zero.setBitsFrom(1);
4510 break;
4511 }
4512 [[fallthrough]];
4514 case ISD::ATOMIC_SWAP:
4525 case ISD::ATOMIC_LOAD_UMAX: {
4526 // If we are looking at the loaded value.
4527 if (Op.getResNo() == 0) {
4528 auto *AT = cast<AtomicSDNode>(Op);
4529 unsigned MemBits = AT->getMemoryVT().getScalarSizeInBits();
4530
4531 if (TLI->getExtendForAtomicOps() == ISD::ZERO_EXTEND)
4532 Known.Zero.setBitsFrom(MemBits);
4533 }
4534 break;
4535 }
4536 case ISD::FrameIndex:
4537 case ISD::TargetFrameIndex: {
4538 const MachineFunction &MF = getMachineFunction();
4539 int FrameIdx = cast<FrameIndexSDNode>(Op)->getIndex();
4540 TLI->computeKnownBitsForStackObjectPointer(
4541 Known, MF, MF.getFrameInfo().getObjectAlign(FrameIdx));
4542 break;
4543 }
4544
4545 default:
4546 if (Opcode < ISD::BUILTIN_OP_END)
4547 break;
4548 [[fallthrough]];
4552 // Allow the target to implement this method for its nodes.
4553 TLI->computeKnownBitsForTargetNode(Op, Known, DemandedElts, *this, Depth);
4554 break;
4555 }
4556
4557 return Known;
4558}
4559
4560/// Convert ConstantRange OverflowResult into SelectionDAG::OverflowKind.
4573
4576 // X + 0 never overflow
4577 if (isNullConstant(N1))
4578 return OFK_Never;
4579
4580 // If both operands each have at least two sign bits, the addition
4581 // cannot overflow.
4582 if (ComputeNumSignBits(N0) > 1 && ComputeNumSignBits(N1) > 1)
4583 return OFK_Never;
4584
4585 // TODO: Add ConstantRange::signedAddMayOverflow handling.
4586 return OFK_Sometime;
4587}
4588
4591 // X + 0 never overflow
4592 if (isNullConstant(N1))
4593 return OFK_Never;
4594
4595 // mulhi + 1 never overflow
4596 KnownBits N1Known = computeKnownBits(N1);
4597 if (N0.getOpcode() == ISD::UMUL_LOHI && N0.getResNo() == 1 &&
4598 N1Known.getMaxValue().ult(2))
4599 return OFK_Never;
4600
4601 KnownBits N0Known = computeKnownBits(N0);
4602 if (N1.getOpcode() == ISD::UMUL_LOHI && N1.getResNo() == 1 &&
4603 N0Known.getMaxValue().ult(2))
4604 return OFK_Never;
4605
4606 // Fallback to ConstantRange::unsignedAddMayOverflow handling.
4607 ConstantRange N0Range = ConstantRange::fromKnownBits(N0Known, false);
4608 ConstantRange N1Range = ConstantRange::fromKnownBits(N1Known, false);
4609 return mapOverflowResult(N0Range.unsignedAddMayOverflow(N1Range));
4610}
4611
4614 // X - 0 never overflow
4615 if (isNullConstant(N1))
4616 return OFK_Never;
4617
4618 // If both operands each have at least two sign bits, the subtraction
4619 // cannot overflow.
4620 if (ComputeNumSignBits(N0) > 1 && ComputeNumSignBits(N1) > 1)
4621 return OFK_Never;
4622
4623 KnownBits N0Known = computeKnownBits(N0);
4624 KnownBits N1Known = computeKnownBits(N1);
4625 ConstantRange N0Range = ConstantRange::fromKnownBits(N0Known, true);
4626 ConstantRange N1Range = ConstantRange::fromKnownBits(N1Known, true);
4627 return mapOverflowResult(N0Range.signedSubMayOverflow(N1Range));
4628}
4629
4632 // X - 0 never overflow
4633 if (isNullConstant(N1))
4634 return OFK_Never;
4635
4636 ConstantRange N0Range =
4637 computeConstantRangeIncludingKnownBits(N0, /*ForSigned=*/false);
4638 ConstantRange N1Range =
4639 computeConstantRangeIncludingKnownBits(N1, /*ForSigned=*/false);
4640 return mapOverflowResult(N0Range.unsignedSubMayOverflow(N1Range));
4641}
4642
4645 // X * 0 and X * 1 never overflow.
4646 if (isNullConstant(N1) || isOneConstant(N1))
4647 return OFK_Never;
4648
4651 return mapOverflowResult(N0Range.unsignedMulMayOverflow(N1Range));
4652}
4653
4656 // X * 0 and X * 1 never overflow.
4657 if (isNullConstant(N1) || isOneConstant(N1))
4658 return OFK_Never;
4659
4660 // Get the size of the result.
4661 unsigned BitWidth = N0.getScalarValueSizeInBits();
4662
4663 // Sum of the sign bits.
4664 unsigned SignBits = ComputeNumSignBits(N0) + ComputeNumSignBits(N1);
4665
4666 // If we have enough sign bits, then there's no overflow.
4667 if (SignBits > BitWidth + 1)
4668 return OFK_Never;
4669
4670 if (SignBits == BitWidth + 1) {
4671 // The overflow occurs when the true multiplication of the
4672 // the operands is the minimum negative number.
4673 KnownBits N0Known = computeKnownBits(N0);
4674 KnownBits N1Known = computeKnownBits(N1);
4675 // If one of the operands is non-negative, then there's no
4676 // overflow.
4677 if (N0Known.isNonNegative() || N1Known.isNonNegative())
4678 return OFK_Never;
4679 }
4680
4681 return OFK_Sometime;
4682}
4683
4685 unsigned Depth) const {
4686 APInt DemandedElts = getDemandAllEltsMask(Op);
4687 return computeConstantRange(Op, DemandedElts, ForSigned, Depth);
4688}
4689
4691 const APInt &DemandedElts,
4692 bool ForSigned,
4693 unsigned Depth) const {
4694 EVT VT = Op.getValueType();
4695 unsigned BitWidth = VT.getScalarSizeInBits();
4696
4697 if (Depth >= MaxRecursionDepth)
4698 return ConstantRange::getFull(BitWidth);
4699
4700 if (ConstantSDNode *C = isConstOrConstSplat(Op, DemandedElts))
4701 return ConstantRange(C->getAPIntValue());
4702
4703 unsigned Opcode = Op.getOpcode();
4704 switch (Opcode) {
4705 case ISD::VSCALE: {
4707 const APInt &Multiplier = Op.getConstantOperandAPInt(0);
4708 return getVScaleRange(&F, BitWidth).multiply(Multiplier);
4709 }
4710 default:
4711 break;
4712 }
4713
4714 return ConstantRange::getFull(BitWidth);
4715}
4716
4719 unsigned Depth) const {
4720 APInt DemandedElts = getDemandAllEltsMask(Op);
4721 return computeConstantRangeIncludingKnownBits(Op, DemandedElts, ForSigned,
4722 Depth);
4723}
4724
4726 SDValue Op, const APInt &DemandedElts, bool ForSigned,
4727 unsigned Depth) const {
4728 KnownBits Known = computeKnownBits(Op, DemandedElts, Depth);
4730 ConstantRange CR2 = computeConstantRange(Op, DemandedElts, ForSigned, Depth);
4733 return CR1.intersectWith(CR2, RangeType);
4734}
4735
4737 unsigned Depth) const {
4738 APInt DemandedElts = getDemandAllEltsMask(Val);
4739 return isKnownToBeAPowerOfTwo(Val, DemandedElts, OrZero, Depth);
4740}
4741
4743 const APInt &DemandedElts,
4744 bool OrZero, unsigned Depth) const {
4745 if (Depth >= MaxRecursionDepth)
4746 return false; // Limit search depth.
4747
4748 EVT OpVT = Val.getValueType();
4749 unsigned BitWidth = OpVT.getScalarSizeInBits();
4750 [[maybe_unused]] unsigned NumElts = DemandedElts.getBitWidth();
4751 assert((!OpVT.isScalableVector() || NumElts == 1) &&
4752 "DemandedElts for scalable vectors must be 1 to represent all lanes");
4753 assert(
4754 (!OpVT.isFixedLengthVector() || NumElts == OpVT.getVectorNumElements()) &&
4755 "Unexpected vector size");
4756
4757 auto IsPowerOfTwoOrZero = [BitWidth, OrZero](const ConstantSDNode *C) {
4758 APInt V = C->getAPIntValue().zextOrTrunc(BitWidth);
4759 return (OrZero && V.isZero()) || V.isPowerOf2();
4760 };
4761
4762 // Is the constant a known power of 2 or zero?
4763 if (ISD::matchUnaryPredicate(Val, DemandedElts, IsPowerOfTwoOrZero,
4764 /*AllowUndefs=*/false, /*AllowTruncation=*/true))
4765 return true;
4766
4767 switch (Val.getOpcode()) {
4769 SDValue InVec = Val.getOperand(0);
4770 SDValue EltNo = Val.getOperand(1);
4771 EVT VecVT = InVec.getValueType();
4772
4773 // Skip scalable vectors or implicit extensions.
4774 if (VecVT.isScalableVector() ||
4775 OpVT.getScalarSizeInBits() != VecVT.getScalarSizeInBits())
4776 break;
4777
4778 // If we know the element index, just demand that vector element, else for
4779 // an unknown element index, ignore DemandedElts and demand them all.
4780 const unsigned NumSrcElts = VecVT.getVectorNumElements();
4781 auto *ConstEltNo = dyn_cast<ConstantSDNode>(EltNo);
4782 APInt DemandedSrcElts =
4783 ConstEltNo && ConstEltNo->getAPIntValue().ult(NumSrcElts)
4784 ? APInt::getOneBitSet(NumSrcElts, ConstEltNo->getZExtValue())
4785 : APInt::getAllOnes(NumSrcElts);
4786 return isKnownToBeAPowerOfTwo(InVec, DemandedSrcElts, OrZero, Depth + 1);
4787 }
4788
4789 case ISD::AND: {
4790 // Looking for `x & -x` pattern:
4791 // If x == 0:
4792 // x & -x -> 0
4793 // If x != 0:
4794 // x & -x -> non-zero pow2
4795 // so if we find the pattern return whether we know `x` is non-zero.
4796 SDValue X, Z;
4797 if (sd_match(Val, m_And(m_Value(X), m_Neg(m_Deferred(X)))) ||
4798 (sd_match(Val, m_And(m_Value(X), m_Sub(m_Value(Z), m_Deferred(X)))) &&
4799 MaskedVectorIsZero(Z, DemandedElts, Depth + 1)))
4800 return OrZero || isKnownNeverZero(X, DemandedElts, Depth);
4801 break;
4802 }
4803
4804 case ISD::SHL: {
4805 // A left-shift of a constant one will have exactly one bit set because
4806 // shifting the bit off the end is undefined.
4807 auto *C = isConstOrConstSplat(Val.getOperand(0), DemandedElts);
4808 if (C && C->getAPIntValue() == 1)
4809 return true;
4810 return (OrZero || isKnownNeverZero(Val, DemandedElts, Depth)) &&
4811 isKnownToBeAPowerOfTwo(Val.getOperand(0), DemandedElts, OrZero,
4812 Depth + 1);
4813 }
4814
4815 case ISD::SRL: {
4816 // A logical right-shift of a constant sign-bit will have exactly
4817 // one bit set.
4818 auto *C = isConstOrConstSplat(Val.getOperand(0), DemandedElts);
4819 if (C && C->getAPIntValue().isSignMask())
4820 return true;
4821 return (OrZero || isKnownNeverZero(Val, DemandedElts, Depth)) &&
4822 isKnownToBeAPowerOfTwo(Val.getOperand(0), DemandedElts, OrZero,
4823 Depth + 1);
4824 }
4825
4826 case ISD::TRUNCATE:
4827 return (OrZero || isKnownNeverZero(Val, DemandedElts, Depth)) &&
4828 isKnownToBeAPowerOfTwo(Val.getOperand(0), DemandedElts, OrZero,
4829 Depth + 1);
4830
4831 case ISD::ROTL:
4832 case ISD::ROTR:
4833 return isKnownToBeAPowerOfTwo(Val.getOperand(0), DemandedElts, OrZero,
4834 Depth + 1);
4835 case ISD::BSWAP:
4836 case ISD::BITREVERSE:
4837 return isKnownToBeAPowerOfTwo(Val.getOperand(0), DemandedElts, OrZero,
4838 Depth + 1);
4839
4840 case ISD::SMIN:
4841 case ISD::SMAX:
4842 case ISD::UMIN:
4843 case ISD::UMAX:
4844 return isKnownToBeAPowerOfTwo(Val.getOperand(1), DemandedElts, OrZero,
4845 Depth + 1) &&
4846 isKnownToBeAPowerOfTwo(Val.getOperand(0), DemandedElts, OrZero,
4847 Depth + 1);
4848
4849 case ISD::SELECT:
4850 case ISD::VSELECT:
4851 return isKnownToBeAPowerOfTwo(Val.getOperand(2), DemandedElts, OrZero,
4852 Depth + 1) &&
4853 isKnownToBeAPowerOfTwo(Val.getOperand(1), DemandedElts, OrZero,
4854 Depth + 1);
4855
4856 case ISD::ZERO_EXTEND:
4857 return isKnownToBeAPowerOfTwo(Val.getOperand(0), DemandedElts, OrZero,
4858 Depth + 1);
4859
4860 case ISD::VSCALE:
4861 // vscale(power-of-two) is a power-of-two
4862 return isKnownToBeAPowerOfTwo(Val.getOperand(0), /*OrZero=*/false,
4863 Depth + 1);
4864
4865 case ISD::VECTOR_SHUFFLE: {
4867 // Demanded elements with undef shuffle mask elements are unknown
4868 // - we cannot guarantee they are a power of two, so return false.
4869 APInt DemandedLHS, DemandedRHS;
4871 assert(NumElts == SVN->getMask().size() && "Unexpected vector size");
4872 if (!getShuffleDemandedElts(NumElts, SVN->getMask(), DemandedElts,
4873 DemandedLHS, DemandedRHS))
4874 return false;
4875
4876 // All demanded elements from LHS must be known power of two.
4877 if (!!DemandedLHS && !isKnownToBeAPowerOfTwo(Val.getOperand(0), DemandedLHS,
4878 OrZero, Depth + 1))
4879 return false;
4880
4881 // All demanded elements from RHS must be known power of two.
4882 if (!!DemandedRHS && !isKnownToBeAPowerOfTwo(Val.getOperand(1), DemandedRHS,
4883 OrZero, Depth + 1))
4884 return false;
4885
4886 return true;
4887 }
4888 }
4889
4890 // More could be done here, though the above checks are enough
4891 // to handle some common cases.
4892 return false;
4893}
4894
4896 if (ConstantFPSDNode *C1 = isConstOrConstSplatFP(Val, true))
4897 return C1->getValueAPF().getExactLog2Abs() >= 0;
4898
4899 if (Val.getOpcode() == ISD::UINT_TO_FP || Val.getOpcode() == ISD::SINT_TO_FP)
4900 return isKnownToBeAPowerOfTwo(Val.getOperand(0), Depth + 1);
4901
4902 return false;
4903}
4904
4906 APInt DemandedElts = getDemandAllEltsMask(Op);
4907 return ComputeNumSignBits(Op, DemandedElts, Depth);
4908}
4909
4910unsigned SelectionDAG::ComputeNumSignBits(SDValue Op, const APInt &DemandedElts,
4911 unsigned Depth) const {
4912 EVT VT = Op.getValueType();
4913 assert((VT.isInteger() || VT.isFloatingPoint()) && "Invalid VT!");
4914 unsigned VTBits = VT.getScalarSizeInBits();
4915 unsigned NumElts = DemandedElts.getBitWidth();
4916 unsigned Tmp, Tmp2;
4917 unsigned FirstAnswer = 1;
4918
4919 assert((!VT.isScalableVector() || NumElts == 1) &&
4920 "DemandedElts for scalable vectors must be 1 to represent all lanes");
4921
4922 if (auto *C = dyn_cast<ConstantSDNode>(Op)) {
4923 const APInt &Val = C->getAPIntValue();
4924 return Val.getNumSignBits();
4925 }
4926
4927 if (Depth >= MaxRecursionDepth)
4928 return 1; // Limit search depth.
4929
4930 if (!DemandedElts)
4931 return 1; // No demanded elts, better to assume we don't know anything.
4932
4933 unsigned Opcode = Op.getOpcode();
4934 switch (Opcode) {
4935 default: break;
4936 case ISD::AssertSext:
4937 Tmp = cast<VTSDNode>(Op.getOperand(1))->getVT().getSizeInBits();
4938 return VTBits-Tmp+1;
4939 case ISD::AssertZext:
4940 Tmp = cast<VTSDNode>(Op.getOperand(1))->getVT().getSizeInBits();
4941 return VTBits-Tmp;
4942 case ISD::FREEZE:
4943 if (isGuaranteedNotToBeUndefOrPoison(Op.getOperand(0), DemandedElts,
4945 return ComputeNumSignBits(Op.getOperand(0), DemandedElts, Depth + 1);
4946 break;
4947 case ISD::MERGE_VALUES:
4948 return ComputeNumSignBits(Op.getOperand(Op.getResNo()), DemandedElts,
4949 Depth + 1);
4950 case ISD::SPLAT_VECTOR: {
4951 // Check if the sign bits of source go down as far as the truncated value.
4952 unsigned NumSrcBits = Op.getOperand(0).getValueSizeInBits();
4953 unsigned NumSrcSignBits = ComputeNumSignBits(Op.getOperand(0), Depth + 1);
4954 if (NumSrcSignBits > (NumSrcBits - VTBits))
4955 return NumSrcSignBits - (NumSrcBits - VTBits);
4956 break;
4957 }
4958 case ISD::BUILD_VECTOR:
4959 assert(!VT.isScalableVector());
4960 Tmp = VTBits;
4961 for (unsigned i = 0, e = Op.getNumOperands(); (i < e) && (Tmp > 1); ++i) {
4962 if (!DemandedElts[i])
4963 continue;
4964
4965 SDValue SrcOp = Op.getOperand(i);
4966 // BUILD_VECTOR can implicitly truncate sources, we handle this specially
4967 // for constant nodes to ensure we only look at the sign bits.
4969 APInt T = C->getAPIntValue().trunc(VTBits);
4970 Tmp2 = T.getNumSignBits();
4971 } else if (SrcOp.getOpcode() == ISD::EXTRACT_VECTOR_ELT &&
4972 SrcOp.getOperand(0).getScalarValueSizeInBits() >= VTBits) {
4973 // EXTRACT_VECTOR_ELT can extend the value with high bits undefined. If
4974 // this BUILD_VECTOR truncates those undefined bits we can just look
4975 // through the SrcOp and query the vector directly.
4976 SDValue InVec = SrcOp.getOperand(0);
4977 EVT InVecVT = InVec.getValueType();
4978
4979 APInt DemandedSrcElts;
4980 if (InVecVT.isScalableVector())
4981 // Demand all elements.
4982 DemandedSrcElts = APInt(1, 1);
4983 else {
4984 unsigned NumSrcElts = InVecVT.getVectorNumElements();
4985 auto *ConstEltNo = dyn_cast<ConstantSDNode>(SrcOp.getOperand(1));
4986 if (ConstEltNo && ConstEltNo->getAPIntValue().ult(NumSrcElts))
4987 DemandedSrcElts =
4988 APInt::getOneBitSet(NumSrcElts, ConstEltNo->getZExtValue());
4989 else
4990 DemandedSrcElts = APInt::getAllOnes(NumSrcElts);
4991 }
4992
4993 Tmp2 = ComputeNumSignBits(InVec, DemandedSrcElts, Depth + 1);
4994 unsigned ExtraBits = InVec.getScalarValueSizeInBits() - VTBits;
4995 if (ExtraBits)
4996 Tmp2 = (Tmp2 > ExtraBits ? Tmp2 - ExtraBits : 1);
4997 } else {
4998 Tmp2 = ComputeNumSignBits(SrcOp, Depth + 1);
4999
5000 if (SrcOp.getValueSizeInBits() != VTBits) {
5001 assert(SrcOp.getValueSizeInBits() > VTBits &&
5002 "Expected BUILD_VECTOR implicit truncation");
5003 unsigned ExtraBits = SrcOp.getValueSizeInBits() - VTBits;
5004 Tmp2 = (Tmp2 > ExtraBits ? Tmp2 - ExtraBits : 1);
5005 }
5006 }
5007 Tmp = std::min(Tmp, Tmp2);
5008 }
5009 return Tmp;
5010
5011 case ISD::VECTOR_COMPRESS: {
5012 SDValue Vec = Op.getOperand(0);
5013 SDValue PassThru = Op.getOperand(2);
5014 Tmp = ComputeNumSignBits(PassThru, DemandedElts, Depth + 1);
5015 if (Tmp == 1)
5016 return 1;
5017 Tmp2 = ComputeNumSignBits(Vec, Depth + 1);
5018 Tmp = std::min(Tmp, Tmp2);
5019 return Tmp;
5020 }
5021
5022 case ISD::VECTOR_SHUFFLE: {
5023 // Collect the minimum number of sign bits that are shared by every vector
5024 // element referenced by the shuffle.
5025 APInt DemandedLHS, DemandedRHS;
5027 assert(NumElts == SVN->getMask().size() && "Unexpected vector size");
5028 if (!getShuffleDemandedElts(NumElts, SVN->getMask(), DemandedElts,
5029 DemandedLHS, DemandedRHS))
5030 return 1;
5031
5032 Tmp = std::numeric_limits<unsigned>::max();
5033 if (!!DemandedLHS)
5034 Tmp = ComputeNumSignBits(Op.getOperand(0), DemandedLHS, Depth + 1);
5035 if (!!DemandedRHS) {
5036 Tmp2 = ComputeNumSignBits(Op.getOperand(1), DemandedRHS, Depth + 1);
5037 Tmp = std::min(Tmp, Tmp2);
5038 }
5039 // If we don't know anything, early out and try computeKnownBits fall-back.
5040 if (Tmp == 1)
5041 break;
5042 assert(Tmp <= VTBits && "Failed to determine minimum sign bits");
5043 return Tmp;
5044 }
5045
5046 case ISD::BITCAST: {
5047 if (VT.isScalableVector())
5048 break;
5049 SDValue N0 = Op.getOperand(0);
5050 EVT SrcVT = N0.getValueType();
5051 unsigned SrcBits = SrcVT.getScalarSizeInBits();
5052
5053 // Ignore bitcasts from unsupported types..
5054 if (!(SrcVT.isInteger() || SrcVT.isFloatingPoint()))
5055 break;
5056
5057 // Fast handling of 'identity' bitcasts.
5058 if (VTBits == SrcBits)
5059 return ComputeNumSignBits(N0, DemandedElts, Depth + 1);
5060
5061 bool IsLE = getDataLayout().isLittleEndian();
5062
5063 // Bitcast 'large element' scalar/vector to 'small element' vector.
5064 if ((SrcBits % VTBits) == 0) {
5065 assert(VT.isVector() && "Expected bitcast to vector");
5066
5067 unsigned Scale = SrcBits / VTBits;
5068 APInt SrcDemandedElts =
5069 APIntOps::ScaleBitMask(DemandedElts, NumElts / Scale);
5070
5071 // Fast case - sign splat can be simply split across the small elements.
5072 Tmp = ComputeNumSignBits(N0, SrcDemandedElts, Depth + 1);
5073 if (Tmp == SrcBits)
5074 return VTBits;
5075
5076 // Slow case - determine how far the sign extends into each sub-element.
5077 Tmp2 = VTBits;
5078 for (unsigned i = 0; i != NumElts; ++i)
5079 if (DemandedElts[i]) {
5080 unsigned SubOffset = i % Scale;
5081 SubOffset = (IsLE ? ((Scale - 1) - SubOffset) : SubOffset);
5082 SubOffset = SubOffset * VTBits;
5083 if (Tmp <= SubOffset)
5084 return 1;
5085 Tmp2 = std::min(Tmp2, Tmp - SubOffset);
5086 }
5087 return Tmp2;
5088 }
5089 break;
5090 }
5091
5093 // FP_TO_SINT_SAT produces a signed value that fits in the saturating VT.
5094 Tmp = cast<VTSDNode>(Op.getOperand(1))->getVT().getScalarSizeInBits();
5095 return VTBits - Tmp + 1;
5096 case ISD::SIGN_EXTEND:
5097 Tmp = VTBits - Op.getOperand(0).getScalarValueSizeInBits();
5098 return ComputeNumSignBits(Op.getOperand(0), DemandedElts, Depth+1) + Tmp;
5100 // Max of the input and what this extends.
5101 Tmp = cast<VTSDNode>(Op.getOperand(1))->getVT().getScalarSizeInBits();
5102 Tmp = VTBits-Tmp+1;
5103 Tmp2 = ComputeNumSignBits(Op.getOperand(0), DemandedElts, Depth+1);
5104 return std::max(Tmp, Tmp2);
5106 if (VT.isScalableVector())
5107 break;
5108 SDValue Src = Op.getOperand(0);
5109 EVT SrcVT = Src.getValueType();
5110 APInt DemandedSrcElts = DemandedElts.zext(SrcVT.getVectorNumElements());
5111 Tmp = VTBits - SrcVT.getScalarSizeInBits();
5112 return ComputeNumSignBits(Src, DemandedSrcElts, Depth+1) + Tmp;
5113 }
5114 case ISD::SRA:
5115 Tmp = ComputeNumSignBits(Op.getOperand(0), DemandedElts, Depth + 1);
5116 // SRA X, C -> adds C sign bits.
5117 if (std::optional<unsigned> ShAmt =
5118 getValidMinimumShiftAmount(Op, DemandedElts, Depth + 1))
5119 Tmp = std::min(Tmp + *ShAmt, VTBits);
5120 return Tmp;
5121 case ISD::SHL:
5122 if (std::optional<ConstantRange> ShAmtRange =
5123 getValidShiftAmountRange(Op, DemandedElts, Depth + 1)) {
5124 unsigned MaxShAmt = ShAmtRange->getUnsignedMax().getZExtValue();
5125 unsigned MinShAmt = ShAmtRange->getUnsignedMin().getZExtValue();
5126 // Try to look through ZERO/SIGN/ANY_EXTEND. If all extended bits are
5127 // shifted out, then we can compute the number of sign bits for the
5128 // operand being extended. A future improvement could be to pass along the
5129 // "shifted left by" information in the recursive calls to
5130 // ComputeKnownSignBits. Allowing us to handle this more generically.
5131 if (ISD::isExtOpcode(Op.getOperand(0).getOpcode())) {
5132 SDValue Ext = Op.getOperand(0);
5133 EVT ExtVT = Ext.getValueType();
5134 SDValue Extendee = Ext.getOperand(0);
5135 EVT ExtendeeVT = Extendee.getValueType();
5136 unsigned SizeDifference =
5137 ExtVT.getScalarSizeInBits() - ExtendeeVT.getScalarSizeInBits();
5138 if (SizeDifference <= MinShAmt) {
5139 Tmp = SizeDifference +
5140 ComputeNumSignBits(Extendee, DemandedElts, Depth + 1);
5141 if (MaxShAmt < Tmp)
5142 return Tmp - MaxShAmt;
5143 }
5144 }
5145 // shl destroys sign bits, ensure it doesn't shift out all sign bits.
5146 Tmp = ComputeNumSignBits(Op.getOperand(0), DemandedElts, Depth + 1);
5147 if (MaxShAmt < Tmp)
5148 return Tmp - MaxShAmt;
5149 }
5150 break;
5151 case ISD::AND:
5152 case ISD::OR:
5153 case ISD::XOR: // NOT is handled here.
5154 // Logical binary ops preserve the number of sign bits at the worst.
5155 Tmp = ComputeNumSignBits(Op.getOperand(0), DemandedElts, Depth+1);
5156 if (Tmp != 1) {
5157 Tmp2 = ComputeNumSignBits(Op.getOperand(1), DemandedElts, Depth+1);
5158 FirstAnswer = std::min(Tmp, Tmp2);
5159 // We computed what we know about the sign bits as our first
5160 // answer. Now proceed to the generic code that uses
5161 // computeKnownBits, and pick whichever answer is better.
5162 }
5163 break;
5164
5165 case ISD::SELECT:
5166 case ISD::VSELECT:
5167 Tmp = ComputeNumSignBits(Op.getOperand(1), DemandedElts, Depth+1);
5168 if (Tmp == 1) return 1; // Early out.
5169 Tmp2 = ComputeNumSignBits(Op.getOperand(2), DemandedElts, Depth+1);
5170 return std::min(Tmp, Tmp2);
5171 case ISD::SELECT_CC:
5172 Tmp = ComputeNumSignBits(Op.getOperand(2), DemandedElts, Depth+1);
5173 if (Tmp == 1) return 1; // Early out.
5174 Tmp2 = ComputeNumSignBits(Op.getOperand(3), DemandedElts, Depth+1);
5175 return std::min(Tmp, Tmp2);
5176
5177 case ISD::SMIN:
5178 case ISD::SMAX: {
5179 // If we have a clamp pattern, we know that the number of sign bits will be
5180 // the minimum of the clamp min/max range.
5181 bool IsMax = (Opcode == ISD::SMAX);
5182 ConstantSDNode *CstLow = nullptr, *CstHigh = nullptr;
5183 if ((CstLow = isConstOrConstSplat(Op.getOperand(1), DemandedElts)))
5184 if (Op.getOperand(0).getOpcode() == (IsMax ? ISD::SMIN : ISD::SMAX))
5185 CstHigh =
5186 isConstOrConstSplat(Op.getOperand(0).getOperand(1), DemandedElts);
5187 if (CstLow && CstHigh) {
5188 if (!IsMax)
5189 std::swap(CstLow, CstHigh);
5190 if (CstLow->getAPIntValue().sle(CstHigh->getAPIntValue())) {
5191 Tmp = CstLow->getAPIntValue().getNumSignBits();
5192 Tmp2 = CstHigh->getAPIntValue().getNumSignBits();
5193 return std::min(Tmp, Tmp2);
5194 }
5195 }
5196
5197 // Fallback - just get the minimum number of sign bits of the operands.
5198 Tmp = ComputeNumSignBits(Op.getOperand(0), DemandedElts, Depth + 1);
5199 if (Tmp == 1)
5200 return 1; // Early out.
5201 Tmp2 = ComputeNumSignBits(Op.getOperand(1), DemandedElts, Depth + 1);
5202 return std::min(Tmp, Tmp2);
5203 }
5204 case ISD::UMIN:
5205 case ISD::UMAX:
5206 Tmp = ComputeNumSignBits(Op.getOperand(0), DemandedElts, Depth + 1);
5207 if (Tmp == 1)
5208 return 1; // Early out.
5209 Tmp2 = ComputeNumSignBits(Op.getOperand(1), DemandedElts, Depth + 1);
5210 return std::min(Tmp, Tmp2);
5211 case ISD::SSUBO_CARRY:
5212 case ISD::USUBO_CARRY:
5213 // sub_carry(x,x,c) -> 0/-1 (sext carry)
5214 if (Op.getResNo() == 0 && Op.getOperand(0) == Op.getOperand(1))
5215 return VTBits;
5216 [[fallthrough]];
5217 case ISD::SADDO:
5218 case ISD::UADDO:
5219 case ISD::SADDO_CARRY:
5220 case ISD::UADDO_CARRY:
5221 case ISD::SSUBO:
5222 case ISD::USUBO:
5223 case ISD::SMULO:
5224 case ISD::UMULO:
5225 if (Op.getResNo() != 1)
5226 break;
5227 // The boolean result conforms to getBooleanContents. Fall through.
5228 // If setcc returns 0/-1, all bits are sign bits.
5229 // We know that we have an integer-based boolean since these operations
5230 // are only available for integer.
5231 if (TLI->getBooleanContents(VT.isVector(), false) ==
5233 return VTBits;
5234 break;
5235 case ISD::SETCC:
5236 case ISD::SETCCCARRY:
5237 case ISD::STRICT_FSETCC:
5238 case ISD::STRICT_FSETCCS: {
5239 unsigned OpNo = Op->isStrictFPOpcode() ? 1 : 0;
5240 // If setcc returns 0/-1, all bits are sign bits.
5241 if (TLI->getBooleanContents(Op.getOperand(OpNo).getValueType()) ==
5243 return VTBits;
5244 break;
5245 }
5247 // Semantically similar to icmp ult.
5248 if (TLI->getBooleanContents(VT.isVector(), /*isFloat=*/false) ==
5250 return VTBits;
5251 break;
5252 case ISD::ROTL:
5253 case ISD::ROTR: {
5254 Tmp = ComputeNumSignBits(Op.getOperand(0), DemandedElts, Depth + 1);
5255 ConstantSDNode *C = isConstOrConstSplat(Op.getOperand(1), DemandedElts);
5256 FirstAnswer = SignBitsOps::rot(
5257 Tmp, VTBits, C ? std::optional(C->getAPIntValue()) : std::nullopt,
5258 Opcode == ISD::ROTR);
5259 break;
5260 }
5261 case ISD::ADD:
5262 case ISD::ADDC:
5263 // TODO: Move Operand 1 check before Operand 0 check
5264 Tmp = ComputeNumSignBits(Op.getOperand(0), DemandedElts, Depth + 1);
5265 if (Tmp == 1) return 1; // Early out.
5266
5267 // Special case decrementing a value (ADD X, -1):
5268 if (ConstantSDNode *CRHS =
5269 isConstOrConstSplat(Op.getOperand(1), DemandedElts))
5270 if (CRHS->isAllOnes()) {
5272 computeKnownBits(Op.getOperand(0), DemandedElts, Depth + 1);
5273
5274 // If the input is known to be 0 or 1, the output is 0/-1, which is all
5275 // sign bits set.
5276 if ((Known.Zero | 1).isAllOnes())
5277 return VTBits;
5278
5279 // If we are subtracting one from a positive number, there is no carry
5280 // out of the result.
5281 if (Known.isNonNegative())
5282 return Tmp;
5283 }
5284
5285 Tmp2 = ComputeNumSignBits(Op.getOperand(1), DemandedElts, Depth + 1);
5286 if (Tmp2 == 1) return 1; // Early out.
5287
5288 // Add can have at most one carry bit. Thus we know that the output
5289 // is, at worst, one more bit than the inputs.
5290 return std::min(Tmp, Tmp2) - 1;
5291 case ISD::SUB:
5292 Tmp2 = ComputeNumSignBits(Op.getOperand(1), DemandedElts, Depth + 1);
5293 if (Tmp2 == 1) return 1; // Early out.
5294
5295 // Handle NEG.
5296 if (ConstantSDNode *CLHS =
5297 isConstOrConstSplat(Op.getOperand(0), DemandedElts))
5298 if (CLHS->isZero()) {
5300 computeKnownBits(Op.getOperand(1), DemandedElts, Depth + 1);
5301 // If the input is known to be 0 or 1, the output is 0/-1, which is all
5302 // sign bits set.
5303 if ((Known.Zero | 1).isAllOnes())
5304 return VTBits;
5305
5306 // If the input is known to be positive (the sign bit is known clear),
5307 // the output of the NEG has the same number of sign bits as the input.
5308 if (Known.isNonNegative())
5309 return Tmp2;
5310
5311 // Otherwise, we treat this like a SUB.
5312 }
5313
5314 // Sub can have at most one carry bit. Thus we know that the output
5315 // is, at worst, one more bit than the inputs.
5316 Tmp = ComputeNumSignBits(Op.getOperand(0), DemandedElts, Depth + 1);
5317 if (Tmp == 1) return 1; // Early out.
5318 return std::min(Tmp, Tmp2) - 1;
5319 case ISD::MUL: {
5320 // The output of the Mul can be at most twice the valid bits in the inputs.
5321 unsigned SignBitsOp0 = ComputeNumSignBits(Op.getOperand(0), Depth + 1);
5322 if (SignBitsOp0 == 1)
5323 break;
5324 unsigned SignBitsOp1 = ComputeNumSignBits(Op.getOperand(1), Depth + 1);
5325 if (SignBitsOp1 == 1)
5326 break;
5327 unsigned OutValidBits =
5328 (VTBits - SignBitsOp0 + 1) + (VTBits - SignBitsOp1 + 1);
5329 return OutValidBits > VTBits ? 1 : VTBits - OutValidBits + 1;
5330 }
5331 case ISD::AVGCEILS:
5332 case ISD::AVGFLOORS:
5333 Tmp = ComputeNumSignBits(Op.getOperand(0), DemandedElts, Depth + 1);
5334 if (Tmp == 1)
5335 return 1; // Early out.
5336 Tmp2 = ComputeNumSignBits(Op.getOperand(1), DemandedElts, Depth + 1);
5337 return std::min(Tmp, Tmp2);
5338 case ISD::SREM:
5339 // The sign bit is the LHS's sign bit, except when the result of the
5340 // remainder is zero. The magnitude of the result should be less than or
5341 // equal to the magnitude of the LHS. Therefore, the result should have
5342 // at least as many sign bits as the left hand side.
5343 return ComputeNumSignBits(Op.getOperand(0), DemandedElts, Depth + 1);
5344 case ISD::TRUNCATE: {
5345 // Check if the sign bits of source go down as far as the truncated value.
5346 unsigned NumSrcBits = Op.getOperand(0).getScalarValueSizeInBits();
5347 unsigned NumSrcSignBits = ComputeNumSignBits(Op.getOperand(0), Depth + 1);
5348 if (NumSrcSignBits > (NumSrcBits - VTBits))
5349 return NumSrcSignBits - (NumSrcBits - VTBits);
5350 break;
5351 }
5352 case ISD::EXTRACT_ELEMENT: {
5353 if (VT.isScalableVector())
5354 break;
5355 const int KnownSign = ComputeNumSignBits(Op.getOperand(0), Depth+1);
5356 const int BitWidth = Op.getValueSizeInBits();
5357 const int Items = Op.getOperand(0).getValueSizeInBits() / BitWidth;
5358
5359 // Get reverse index (starting from 1), Op1 value indexes elements from
5360 // little end. Sign starts at big end.
5361 const int rIndex = Items - 1 - Op.getConstantOperandVal(1);
5362
5363 // If the sign portion ends in our element the subtraction gives correct
5364 // result. Otherwise it gives either negative or > bitwidth result
5365 return std::clamp(KnownSign - rIndex * BitWidth, 1, BitWidth);
5366 }
5368 if (VT.isScalableVector())
5369 break;
5370 // If we know the element index, split the demand between the
5371 // source vector and the inserted element, otherwise assume we need
5372 // the original demanded vector elements and the value.
5373 SDValue InVec = Op.getOperand(0);
5374 SDValue InVal = Op.getOperand(1);
5375 SDValue EltNo = Op.getOperand(2);
5376 bool DemandedVal = true;
5377 APInt DemandedVecElts = DemandedElts;
5378 auto *CEltNo = dyn_cast<ConstantSDNode>(EltNo);
5379 if (CEltNo && CEltNo->getAPIntValue().ult(NumElts)) {
5380 unsigned EltIdx = CEltNo->getZExtValue();
5381 DemandedVal = !!DemandedElts[EltIdx];
5382 DemandedVecElts.clearBit(EltIdx);
5383 }
5384 Tmp = std::numeric_limits<unsigned>::max();
5385 if (DemandedVal) {
5386 // TODO - handle implicit truncation of inserted elements.
5387 if (InVal.getScalarValueSizeInBits() != VTBits)
5388 break;
5389 Tmp2 = ComputeNumSignBits(InVal, Depth + 1);
5390 Tmp = std::min(Tmp, Tmp2);
5391 }
5392 if (!!DemandedVecElts) {
5393 Tmp2 = ComputeNumSignBits(InVec, DemandedVecElts, Depth + 1);
5394 Tmp = std::min(Tmp, Tmp2);
5395 }
5396 assert(Tmp <= VTBits && "Failed to determine minimum sign bits");
5397 return Tmp;
5398 }
5400 SDValue InVec = Op.getOperand(0);
5401 SDValue EltNo = Op.getOperand(1);
5402 EVT VecVT = InVec.getValueType();
5403 // ComputeNumSignBits not yet implemented for scalable vectors.
5404 if (VecVT.isScalableVector())
5405 break;
5406 const unsigned BitWidth = Op.getValueSizeInBits();
5407 const unsigned EltBitWidth = Op.getOperand(0).getScalarValueSizeInBits();
5408 const unsigned NumSrcElts = VecVT.getVectorNumElements();
5409
5410 // If BitWidth > EltBitWidth the value is anyext:ed, and we do not know
5411 // anything about sign bits. But if the sizes match we can derive knowledge
5412 // about sign bits from the vector operand.
5413 if (BitWidth != EltBitWidth)
5414 break;
5415
5416 // If we know the element index, just demand that vector element, else for
5417 // an unknown element index, ignore DemandedElts and demand them all.
5418 APInt DemandedSrcElts = APInt::getAllOnes(NumSrcElts);
5419 auto *ConstEltNo = dyn_cast<ConstantSDNode>(EltNo);
5420 if (ConstEltNo && ConstEltNo->getAPIntValue().ult(NumSrcElts))
5421 DemandedSrcElts =
5422 APInt::getOneBitSet(NumSrcElts, ConstEltNo->getZExtValue());
5423
5424 return ComputeNumSignBits(InVec, DemandedSrcElts, Depth + 1);
5425 }
5427 // Offset the demanded elts by the subvector index.
5428 SDValue Src = Op.getOperand(0);
5429
5430 APInt DemandedSrcElts;
5431 if (Src.getValueType().isScalableVector())
5432 DemandedSrcElts = APInt(1, 1);
5433 else {
5434 uint64_t Idx = Op.getConstantOperandVal(1);
5435 unsigned NumSrcElts = Src.getValueType().getVectorNumElements();
5436 DemandedSrcElts = DemandedElts.zext(NumSrcElts).shl(Idx);
5437 }
5438 return ComputeNumSignBits(Src, DemandedSrcElts, Depth + 1);
5439 }
5440 case ISD::CONCAT_VECTORS: {
5441 if (VT.isScalableVector())
5442 break;
5443 // Determine the minimum number of sign bits across all demanded
5444 // elts of the input vectors. Early out if the result is already 1.
5445 Tmp = std::numeric_limits<unsigned>::max();
5446 EVT SubVectorVT = Op.getOperand(0).getValueType();
5447 unsigned NumSubVectorElts = SubVectorVT.getVectorNumElements();
5448 unsigned NumSubVectors = Op.getNumOperands();
5449 for (unsigned i = 0; (i < NumSubVectors) && (Tmp > 1); ++i) {
5450 APInt DemandedSub =
5451 DemandedElts.extractBits(NumSubVectorElts, i * NumSubVectorElts);
5452 if (!DemandedSub)
5453 continue;
5454 Tmp2 = ComputeNumSignBits(Op.getOperand(i), DemandedSub, Depth + 1);
5455 Tmp = std::min(Tmp, Tmp2);
5456 }
5457 assert(Tmp <= VTBits && "Failed to determine minimum sign bits");
5458 return Tmp;
5459 }
5460 case ISD::INSERT_SUBVECTOR: {
5461 SDValue Src = Op.getOperand(0);
5462 SDValue Sub = Op.getOperand(1);
5463 if (VT.isScalableVector()) {
5464 Tmp = ComputeNumSignBits(Sub, Depth + 1);
5465 Tmp = std::min(Tmp, ComputeNumSignBits(Src, Depth + 1));
5466 return Tmp;
5467 }
5468 // Demand any elements from the subvector and the remainder from the src its
5469 // inserted into.
5470 uint64_t Idx = Op.getConstantOperandVal(2);
5471 unsigned NumSubElts = Sub.getValueType().getVectorNumElements();
5472 APInt DemandedSubElts = DemandedElts.extractBits(NumSubElts, Idx);
5473 APInt DemandedSrcElts = DemandedElts;
5474 DemandedSrcElts.clearBits(Idx, Idx + NumSubElts);
5475
5476 Tmp = std::numeric_limits<unsigned>::max();
5477 if (!!DemandedSubElts) {
5478 Tmp = ComputeNumSignBits(Sub, DemandedSubElts, Depth + 1);
5479 if (Tmp == 1)
5480 return 1; // early-out
5481 }
5482 if (!!DemandedSrcElts) {
5483 Tmp2 = ComputeNumSignBits(Src, DemandedSrcElts, Depth + 1);
5484 Tmp = std::min(Tmp, Tmp2);
5485 }
5486 assert(Tmp <= VTBits && "Failed to determine minimum sign bits");
5487 return Tmp;
5488 }
5489 case ISD::LOAD: {
5490 // If we are looking at the loaded value of the SDNode.
5491 if (Op.getResNo() != 0)
5492 break;
5493
5495 if (const MDNode *Ranges = LD->getRanges()) {
5496 if (DemandedElts != 1)
5497 break;
5498
5500 if (VTBits > CR.getBitWidth()) {
5501 switch (LD->getExtensionType()) {
5502 case ISD::SEXTLOAD:
5503 CR = CR.signExtend(VTBits);
5504 break;
5505 case ISD::ZEXTLOAD:
5506 CR = CR.zeroExtend(VTBits);
5507 break;
5508 default:
5509 break;
5510 }
5511 }
5512
5513 if (VTBits != CR.getBitWidth())
5514 break;
5515 return std::min(CR.getSignedMin().getNumSignBits(),
5517 }
5518
5519 unsigned ExtType = LD->getExtensionType();
5520 switch (ExtType) {
5521 default:
5522 break;
5523 case ISD::SEXTLOAD: // e.g. i16->i32 = '17' bits known.
5524 Tmp = LD->getMemoryVT().getScalarSizeInBits();
5525 return VTBits - Tmp + 1;
5526 case ISD::ZEXTLOAD: // e.g. i16->i32 = '16' bits known.
5527 Tmp = LD->getMemoryVT().getScalarSizeInBits();
5528 return VTBits - Tmp;
5529 case ISD::NON_EXTLOAD:
5530 if (const Constant *Cst = TLI->getTargetConstantFromLoad(LD)) {
5531 // We only need to handle vectors - computeKnownBits should handle
5532 // scalar cases.
5533 Type *CstTy = Cst->getType();
5534 if (CstTy->isVectorTy() && !VT.isScalableVector() &&
5535 (NumElts * VTBits) == CstTy->getPrimitiveSizeInBits() &&
5536 VTBits == CstTy->getScalarSizeInBits()) {
5537 Tmp = VTBits;
5538 for (unsigned i = 0; i != NumElts; ++i) {
5539 if (!DemandedElts[i])
5540 continue;
5541 if (Constant *Elt = Cst->getAggregateElement(i)) {
5542 if (auto *CInt = dyn_cast<ConstantInt>(Elt)) {
5543 const APInt &Value = CInt->getValue();
5544 Tmp = std::min(Tmp, Value.getNumSignBits());
5545 continue;
5546 }
5547 if (auto *CFP = dyn_cast<ConstantFP>(Elt)) {
5548 APInt Value = CFP->getValueAPF().bitcastToAPInt();
5549 Tmp = std::min(Tmp, Value.getNumSignBits());
5550 continue;
5551 }
5552 }
5553 // Unknown type. Conservatively assume no bits match sign bit.
5554 return 1;
5555 }
5556 return Tmp;
5557 }
5558 }
5559 break;
5560 }
5561
5562 break;
5563 }
5566 case ISD::ATOMIC_SWAP:
5578 case ISD::ATOMIC_LOAD: {
5579 auto *AT = cast<AtomicSDNode>(Op);
5580 // If we are looking at the loaded value.
5581 if (Op.getResNo() == 0) {
5582 Tmp = AT->getMemoryVT().getScalarSizeInBits();
5583 if (Tmp == VTBits)
5584 return 1; // early-out
5585
5586 // For atomic_load, prefer to use the extension type.
5587 if (Op->getOpcode() == ISD::ATOMIC_LOAD) {
5588 switch (AT->getExtensionType()) {
5589 default:
5590 break;
5591 case ISD::SEXTLOAD:
5592 return VTBits - Tmp + 1;
5593 case ISD::ZEXTLOAD:
5594 return VTBits - Tmp;
5595 }
5596 }
5597
5598 if (TLI->getExtendForAtomicOps() == ISD::SIGN_EXTEND)
5599 return VTBits - Tmp + 1;
5600 if (TLI->getExtendForAtomicOps() == ISD::ZERO_EXTEND)
5601 return VTBits - Tmp;
5602 }
5603 break;
5604 }
5605 }
5606
5607 // Allow the target to implement this method for its nodes.
5608 if (Opcode >= ISD::BUILTIN_OP_END ||
5609 Opcode == ISD::INTRINSIC_WO_CHAIN ||
5610 Opcode == ISD::INTRINSIC_W_CHAIN ||
5611 Opcode == ISD::INTRINSIC_VOID) {
5612 // TODO: This can probably be removed once target code is audited. This
5613 // is here purely to reduce patch size and review complexity.
5614 if (!VT.isScalableVector()) {
5615 unsigned NumBits =
5616 TLI->ComputeNumSignBitsForTargetNode(Op, DemandedElts, *this, Depth);
5617 if (NumBits > 1)
5618 FirstAnswer = std::max(FirstAnswer, NumBits);
5619 }
5620 }
5621
5622 // Finally, if we can prove that the top bits of the result are 0's or 1's,
5623 // use this information.
5624 KnownBits Known = computeKnownBits(Op, DemandedElts, Depth);
5625 return std::max(FirstAnswer, Known.countMinSignBits());
5626}
5627
5629 unsigned Depth) const {
5630 unsigned SignBits = ComputeNumSignBits(Op, Depth);
5631 return Op.getScalarValueSizeInBits() - SignBits + 1;
5632}
5633
5635 const APInt &DemandedElts,
5636 unsigned Depth) const {
5637 unsigned SignBits = ComputeNumSignBits(Op, DemandedElts, Depth);
5638 return Op.getScalarValueSizeInBits() - SignBits + 1;
5639}
5640
5642 UndefPoisonKind Kind,
5643 unsigned Depth) const {
5644 // Early out for FREEZE.
5645 if (Op.getOpcode() == ISD::FREEZE)
5646 return true;
5647
5648 APInt DemandedElts = getDemandAllEltsMask(Op);
5649 return isGuaranteedNotToBeUndefOrPoison(Op, DemandedElts, Kind, Depth);
5650}
5651
5653 const APInt &DemandedElts,
5654 UndefPoisonKind Kind,
5655 unsigned Depth) const {
5656 unsigned Opcode = Op.getOpcode();
5657
5658 // Early out for FREEZE.
5659 if (Opcode == ISD::FREEZE)
5660 return true;
5661
5662 if (Depth >= MaxRecursionDepth)
5663 return false; // Limit search depth.
5664
5665 if (isIntOrFPConstant(Op))
5666 return true;
5667
5668 switch (Opcode) {
5669 case ISD::CONDCODE:
5670 case ISD::VALUETYPE:
5671 case ISD::FrameIndex:
5673 case ISD::CopyFromReg:
5674 return true;
5675
5676 case ISD::POISON:
5677 return !includesPoison(Kind);
5678
5679 case ISD::UNDEF:
5680 return !includesUndef(Kind);
5681
5682 case ISD::BITCAST: {
5683 SDValue Src = Op.getOperand(0);
5684 EVT SrcVT = Src.getValueType();
5685 EVT DstVT = Op.getValueType();
5686
5687 if (!SrcVT.isVector() || !DstVT.isVector())
5688 return isGuaranteedNotToBeUndefOrPoison(Src, Kind, Depth + 1);
5689
5690 unsigned SrcEltBits = SrcVT.getScalarSizeInBits();
5691 unsigned DstEltBits = DstVT.getScalarSizeInBits();
5692 ElementCount NumSrcElts = SrcVT.getVectorElementCount();
5693 [[maybe_unused]] ElementCount NumDstElts = DstVT.getVectorElementCount();
5694
5695 if (SrcEltBits == DstEltBits)
5696 return isGuaranteedNotToBeUndefOrPoison(Src, DemandedElts, Kind,
5697 Depth + 1);
5698
5699 if (SrcEltBits < DstEltBits) {
5700 if (DstEltBits % SrcEltBits != 0)
5701 return isGuaranteedNotToBeUndefOrPoison(Src, Kind, Depth + 1);
5702
5703 assert(NumSrcElts == NumDstElts * (DstEltBits / SrcEltBits) &&
5704 "Unexpected vector bitcast");
5705 APInt DemandedSrcElts =
5706 APIntOps::ScaleBitMask(DemandedElts, NumSrcElts.getKnownMinValue());
5707 return isGuaranteedNotToBeUndefOrPoison(Src, DemandedSrcElts, Kind,
5708 Depth + 1);
5709 }
5710
5711 if (SrcEltBits % DstEltBits != 0)
5712 return isGuaranteedNotToBeUndefOrPoison(Src, Kind, Depth + 1);
5713
5714 assert(NumDstElts == NumSrcElts * (SrcEltBits / DstEltBits) &&
5715 "Unexpected vector bitcast");
5716 APInt DemandedSrcElts =
5717 APIntOps::ScaleBitMask(DemandedElts, NumSrcElts.getKnownMinValue());
5718 return isGuaranteedNotToBeUndefOrPoison(Src, DemandedSrcElts, Kind,
5719 Depth + 1);
5720 }
5721
5722 case ISD::BUILD_VECTOR:
5723 // NOTE: BUILD_VECTOR has implicit truncation of wider scalar elements -
5724 // this shouldn't affect the result.
5725 for (unsigned i = 0, e = Op.getNumOperands(); i < e; ++i) {
5726 if (!DemandedElts[i])
5727 continue;
5728 if (!isGuaranteedNotToBeUndefOrPoison(Op.getOperand(i), Kind, Depth + 1))
5729 return false;
5730 }
5731 return true;
5732
5733 case ISD::CONCAT_VECTORS: {
5734 EVT VT = Op.getValueType();
5735 if (!VT.isFixedLengthVector())
5736 break;
5737
5738 EVT SubVT = Op.getOperand(0).getValueType();
5739 unsigned NumSubElts = SubVT.getVectorNumElements();
5740 for (unsigned I = 0, E = Op.getNumOperands(); I != E; ++I) {
5741 APInt DemandedSubElts =
5742 DemandedElts.extractBits(NumSubElts, I * NumSubElts);
5743 if (!!DemandedSubElts &&
5744 !isGuaranteedNotToBeUndefOrPoison(Op.getOperand(I), DemandedSubElts,
5745 Kind, Depth + 1))
5746 return false;
5747 }
5748 return true;
5749 }
5750
5752 SDValue Src = Op.getOperand(0);
5753 if (Src.getValueType().isScalableVector())
5754 break;
5755 uint64_t Idx = Op.getConstantOperandVal(1);
5756 unsigned NumSrcElts = Src.getValueType().getVectorNumElements();
5757 APInt DemandedSrcElts = DemandedElts.zext(NumSrcElts).shl(Idx);
5758 return isGuaranteedNotToBeUndefOrPoison(Src, DemandedSrcElts, Kind,
5759 Depth + 1);
5760 }
5761
5762 case ISD::INSERT_SUBVECTOR: {
5763 if (Op.getValueType().isScalableVector())
5764 break;
5765 SDValue Src = Op.getOperand(0);
5766 SDValue Sub = Op.getOperand(1);
5767 uint64_t Idx = Op.getConstantOperandVal(2);
5768 unsigned NumSubElts = Sub.getValueType().getVectorNumElements();
5769 APInt DemandedSubElts = DemandedElts.extractBits(NumSubElts, Idx);
5770 APInt DemandedSrcElts = DemandedElts;
5771 DemandedSrcElts.clearBits(Idx, Idx + NumSubElts);
5772
5773 if (!!DemandedSubElts && !isGuaranteedNotToBeUndefOrPoison(
5774 Sub, DemandedSubElts, Kind, Depth + 1))
5775 return false;
5776 if (!!DemandedSrcElts && !isGuaranteedNotToBeUndefOrPoison(
5777 Src, DemandedSrcElts, Kind, Depth + 1))
5778 return false;
5779 return true;
5780 }
5781
5783 SDValue Src = Op.getOperand(0);
5784 auto *IndexC = dyn_cast<ConstantSDNode>(Op.getOperand(1));
5785 EVT SrcVT = Src.getValueType();
5786 if (SrcVT.isFixedLengthVector() && IndexC &&
5787 IndexC->getAPIntValue().ult(SrcVT.getVectorNumElements())) {
5788 APInt DemandedSrcElts = APInt::getOneBitSet(SrcVT.getVectorNumElements(),
5789 IndexC->getZExtValue());
5790 return isGuaranteedNotToBeUndefOrPoison(Src, DemandedSrcElts, Kind,
5791 Depth + 1);
5792 }
5793 break;
5794 }
5795
5797 SDValue InVec = Op.getOperand(0);
5798 SDValue InVal = Op.getOperand(1);
5799 SDValue EltNo = Op.getOperand(2);
5800 EVT VT = InVec.getValueType();
5801 auto *IndexC = dyn_cast<ConstantSDNode>(EltNo);
5802 if (IndexC && VT.isFixedLengthVector() &&
5803 IndexC->getAPIntValue().ult(VT.getVectorNumElements())) {
5804 if (DemandedElts[IndexC->getZExtValue()] &&
5805 !isGuaranteedNotToBeUndefOrPoison(InVal, Kind, Depth + 1))
5806 return false;
5807 APInt InVecDemandedElts = DemandedElts;
5808 InVecDemandedElts.clearBit(IndexC->getZExtValue());
5809 if (!!InVecDemandedElts &&
5811 peekThroughInsertVectorElt(InVec, InVecDemandedElts),
5812 InVecDemandedElts, Kind, Depth + 1))
5813 return false;
5814 return true;
5815 }
5816 break;
5817 }
5818
5820 // Check upper (known poison) elements.
5821 if (DemandedElts.ugt(1) && includesPoison(Kind))
5822 return false;
5823 // Check element zero.
5824 if (DemandedElts[0] &&
5825 !isGuaranteedNotToBeUndefOrPoison(Op.getOperand(0), Kind, Depth + 1))
5826 return false;
5827 return true;
5828
5829 case ISD::SPLAT_VECTOR:
5830 return isGuaranteedNotToBeUndefOrPoison(Op.getOperand(0), Kind, Depth + 1);
5831
5832 case ISD::SELECT: {
5833 return !canCreateUndefOrPoison(Op, DemandedElts, Kind,
5834 /*ConsiderFlags*/ true, Depth) &&
5835 isGuaranteedNotToBeUndefOrPoison(Op.getOperand(0), Kind,
5836 Depth + 1) &&
5837 isGuaranteedNotToBeUndefOrPoison(Op.getOperand(1), DemandedElts,
5838 Kind, Depth + 1) &&
5839 isGuaranteedNotToBeUndefOrPoison(Op.getOperand(2), DemandedElts,
5840 Kind, Depth + 1);
5841 }
5842
5843 case ISD::VECTOR_SHUFFLE: {
5844 APInt DemandedLHS, DemandedRHS;
5845 auto *SVN = cast<ShuffleVectorSDNode>(Op);
5846 if (!getShuffleDemandedElts(DemandedElts.getBitWidth(), SVN->getMask(),
5847 DemandedElts, DemandedLHS, DemandedRHS,
5848 /*AllowUndefElts=*/false))
5849 return false;
5850 if (!DemandedLHS.isZero() &&
5851 !isGuaranteedNotToBeUndefOrPoison(Op.getOperand(0), DemandedLHS, Kind,
5852 Depth + 1))
5853 return false;
5854 if (!DemandedRHS.isZero() &&
5855 !isGuaranteedNotToBeUndefOrPoison(Op.getOperand(1), DemandedRHS, Kind,
5856 Depth + 1))
5857 return false;
5858 return true;
5859 }
5860
5861 case ISD::SHL:
5862 case ISD::SRL:
5863 case ISD::SRA:
5864 // Shift amount operand is checked by canCreateUndefOrPoison. So it is
5865 // enough to check operand 0 if Op can't create undef/poison.
5866 return !canCreateUndefOrPoison(Op, DemandedElts, Kind,
5867 /*ConsiderFlags*/ true, Depth) &&
5868 isGuaranteedNotToBeUndefOrPoison(Op.getOperand(0), DemandedElts,
5869 Kind, Depth + 1);
5870
5871 case ISD::BSWAP:
5872 case ISD::CTPOP:
5873 case ISD::BITREVERSE:
5874 case ISD::AND:
5875 case ISD::OR:
5876 case ISD::XOR:
5877 case ISD::ADD:
5878 case ISD::SUB:
5879 case ISD::MUL:
5880 case ISD::SADDSAT:
5881 case ISD::UADDSAT:
5882 case ISD::SSUBSAT:
5883 case ISD::USUBSAT:
5884 case ISD::SSHLSAT:
5885 case ISD::USHLSAT:
5886 case ISD::SMIN:
5887 case ISD::SMAX:
5888 case ISD::UMIN:
5889 case ISD::UMAX:
5890 case ISD::ZERO_EXTEND:
5891 case ISD::SIGN_EXTEND:
5892 case ISD::ANY_EXTEND:
5893 case ISD::TRUNCATE:
5894 case ISD::VSELECT: {
5895 // If Op can't create undef/poison and none of its operands are undef/poison
5896 // then Op is never undef/poison. A difference from the more common check
5897 // below, outside the switch, is that we handle elementwise operations for
5898 // which the DemandedElts mask is valid for all operands here.
5899 return !canCreateUndefOrPoison(Op, DemandedElts, Kind,
5900 /*ConsiderFlags*/ true, Depth) &&
5901 all_of(Op->ops(), [&](SDValue V) {
5902 return isGuaranteedNotToBeUndefOrPoison(V, DemandedElts, Kind,
5903 Depth + 1);
5904 });
5905 }
5906
5907 // TODO: Search for noundef attributes from library functions.
5908
5909 // TODO: Pointers dereferenced by ISD::LOAD/STORE ops are noundef.
5910
5911 default:
5912 // Allow the target to implement this method for its nodes.
5913 if (Opcode >= ISD::BUILTIN_OP_END || Opcode == ISD::INTRINSIC_WO_CHAIN ||
5914 Opcode == ISD::INTRINSIC_W_CHAIN || Opcode == ISD::INTRINSIC_VOID)
5915 return TLI->isGuaranteedNotToBeUndefOrPoisonForTargetNode(
5916 Op, DemandedElts, *this, Kind, Depth);
5917 break;
5918 }
5919
5920 // If Op can't create undef/poison and none of its operands are undef/poison
5921 // then Op is never undef/poison.
5922 // NOTE: TargetNodes can handle this in themselves in
5923 // isGuaranteedNotToBeUndefOrPoisonForTargetNode or let
5924 // TargetLowering::isGuaranteedNotToBeUndefOrPoisonForTargetNode handle it.
5925 return !canCreateUndefOrPoison(Op, Kind, /*ConsiderFlags*/ true, Depth) &&
5926 all_of(Op->ops(), [&](SDValue V) {
5927 return isGuaranteedNotToBeUndefOrPoison(V, Kind, Depth + 1);
5928 });
5929}
5930
5932 bool ConsiderFlags,
5933 unsigned Depth) const {
5934 APInt DemandedElts = getDemandAllEltsMask(Op);
5935 return canCreateUndefOrPoison(Op, DemandedElts, Kind, ConsiderFlags, Depth);
5936}
5937
5939 UndefPoisonKind Kind,
5940 bool ConsiderFlags,
5941 unsigned Depth) const {
5942 if (ConsiderFlags && includesPoison(Kind) && Op->hasPoisonGeneratingFlags())
5943 return true;
5944
5945 unsigned Opcode = Op.getOpcode();
5946 switch (Opcode) {
5947 case ISD::AssertSext:
5948 case ISD::AssertZext:
5949 case ISD::AssertAlign:
5951 // Assertion nodes can create poison if the assertion fails.
5952 return includesPoison(Kind);
5953
5954 case ISD::FREEZE:
5958 case ISD::SADDSAT:
5959 case ISD::UADDSAT:
5960 case ISD::SSUBSAT:
5961 case ISD::USUBSAT:
5962 case ISD::MULHU:
5963 case ISD::MULHS:
5964 case ISD::AVGFLOORS:
5965 case ISD::AVGFLOORU:
5966 case ISD::AVGCEILS:
5967 case ISD::AVGCEILU:
5968 case ISD::ABDU:
5969 case ISD::ABDS:
5970 case ISD::SMIN:
5971 case ISD::SMAX:
5972 case ISD::SCMP:
5973 case ISD::UMIN:
5974 case ISD::UMAX:
5975 case ISD::UCMP:
5976 case ISD::AND:
5977 case ISD::XOR:
5978 case ISD::ROTL:
5979 case ISD::ROTR:
5980 case ISD::FSHL:
5981 case ISD::FSHR:
5982 case ISD::BSWAP:
5983 case ISD::CTTZ:
5984 case ISD::CTLZ:
5985 case ISD::CTLS:
5986 case ISD::CTPOP:
5987 case ISD::BITREVERSE:
5988 case ISD::PARITY:
5989 case ISD::SIGN_EXTEND:
5990 case ISD::TRUNCATE:
5994 case ISD::BITCAST:
5995 case ISD::BUILD_VECTOR:
5996 case ISD::BUILD_PAIR:
5997 case ISD::SPLAT_VECTOR:
5998 case ISD::FABS:
5999 case ISD::FCEIL:
6000 case ISD::FFLOOR:
6001 case ISD::FTRUNC:
6002 case ISD::FRINT:
6003 case ISD::FNEARBYINT:
6004 case ISD::FROUND:
6005 case ISD::FROUNDEVEN:
6006 return false;
6007
6008 case ISD::ABS:
6009 // ISD::ABS defines abs(INT_MIN) -> INT_MIN and never generates poison.
6010 // Different to Intrinsic::abs.
6011 return false;
6013 // ABS_MIN_POISON may produce poison if the input is INT_MIN.
6014 return ComputeNumSignBits(Op.getOperand(0), DemandedElts, Depth + 1) <= 1;
6015
6016 case ISD::ADDC:
6017 case ISD::SUBC:
6018 case ISD::ADDE:
6019 case ISD::SUBE:
6020 case ISD::SADDO:
6021 case ISD::SSUBO:
6022 case ISD::SMULO:
6023 case ISD::SADDO_CARRY:
6024 case ISD::SSUBO_CARRY:
6025 case ISD::UADDO:
6026 case ISD::USUBO:
6027 case ISD::UMULO:
6028 case ISD::UADDO_CARRY:
6029 case ISD::USUBO_CARRY:
6030 // No poison on result or overflow flags.
6031 return false;
6032
6033 case ISD::SELECT_CC:
6034 case ISD::SETCC: {
6035 // Integer setcc cannot create undef or poison.
6036 if (Op.getOperand(0).getValueType().isInteger())
6037 return false;
6038
6039 // FP compares are more complicated. They can create poison for nan/infinity
6040 // based on options and flags. The options and flags also cause special
6041 // nonan condition codes to be used. Those condition codes may be preserved
6042 // even if the nonan flag is dropped somewhere.
6043 unsigned CCOp = Opcode == ISD::SETCC ? 2 : 4;
6044 ISD::CondCode CCCode = cast<CondCodeSDNode>(Op.getOperand(CCOp))->get();
6045 return (unsigned)CCCode & 0x10U;
6046 }
6047
6048 case ISD::OR:
6049 case ISD::ZERO_EXTEND:
6050 case ISD::SELECT:
6051 case ISD::VSELECT:
6052 case ISD::ADD:
6053 case ISD::SUB:
6054 case ISD::MUL:
6055 case ISD::FNEG:
6056 case ISD::FADD:
6057 case ISD::FSUB:
6058 case ISD::FMUL:
6059 case ISD::FDIV:
6060 case ISD::FREM:
6061 case ISD::FCOPYSIGN:
6062 case ISD::FMA:
6063 case ISD::FMAD:
6064 case ISD::FMULADD:
6065 case ISD::FP_EXTEND:
6066 case ISD::FMINNUM:
6067 case ISD::FMAXNUM:
6068 case ISD::FMINNUM_IEEE:
6069 case ISD::FMAXNUM_IEEE:
6070 case ISD::FMINIMUM:
6071 case ISD::FMAXIMUM:
6072 case ISD::FMINIMUMNUM:
6073 case ISD::FMAXIMUMNUM:
6079 // No poison except from flags (which is handled above)
6080 return false;
6081
6082 case ISD::SHL:
6083 case ISD::SRL:
6084 case ISD::SRA:
6085 // If the max shift amount isn't in range, then the shift can
6086 // create poison.
6087 return includesPoison(Kind) &&
6088 !getValidMaximumShiftAmount(Op, DemandedElts, Depth + 1);
6089
6092 // If the amount is zero then the result will be poison.
6093 // TODO: Add isKnownNeverZero DemandedElts handling.
6094 return includesPoison(Kind) &&
6095 !isKnownNeverZero(Op.getOperand(0), Depth + 1);
6096
6098 // Check if we demand any upper (poison) elements.
6099 return includesPoison(Kind) && DemandedElts.ugt(1);
6100
6103 // Ensure that the element index is in bounds.
6104 if (includesPoison(Kind)) {
6105 EVT VecVT = Op.getOperand(0).getValueType();
6106 SDValue Idx = Op.getOperand(Opcode == ISD::INSERT_VECTOR_ELT ? 2 : 1);
6107 KnownBits KnownIdx = computeKnownBits(Idx, Depth + 1);
6108 return KnownIdx.getMaxValue().uge(VecVT.getVectorMinNumElements());
6109 }
6110 return false;
6111 }
6112
6113 case ISD::VECTOR_SHUFFLE: {
6114 // Check for any demanded shuffle element that is undef.
6115 auto *SVN = cast<ShuffleVectorSDNode>(Op);
6116 for (auto [Idx, Elt] : enumerate(SVN->getMask()))
6117 if (Elt < 0 && DemandedElts[Idx])
6118 return true;
6119 return false;
6120 }
6121
6123 return false;
6124
6125 default:
6126 // Allow the target to implement this method for its nodes.
6127 if (Opcode >= ISD::BUILTIN_OP_END || Opcode == ISD::INTRINSIC_WO_CHAIN ||
6128 Opcode == ISD::INTRINSIC_W_CHAIN || Opcode == ISD::INTRINSIC_VOID)
6129 return TLI->canCreateUndefOrPoisonForTargetNode(
6130 Op, DemandedElts, *this, Kind, ConsiderFlags, Depth);
6131 break;
6132 }
6133
6134 // Be conservative and return true.
6135 return true;
6136}
6137
6138bool SelectionDAG::isADDLike(SDValue Op, bool NoWrap) const {
6139 unsigned Opcode = Op.getOpcode();
6140 if (Opcode == ISD::OR)
6141 return Op->getFlags().hasDisjoint() ||
6142 haveNoCommonBitsSet(Op.getOperand(0), Op.getOperand(1));
6143 if (Opcode == ISD::XOR)
6144 return !NoWrap && isMinSignedConstant(Op.getOperand(1));
6145 return false;
6146}
6147
6149 return Op.getNumOperands() == 2 && isa<ConstantSDNode>(Op.getOperand(1)) &&
6150 (Op.isAnyAdd() || isADDLike(Op));
6151}
6152
6154 FPClassTest InterestedClasses,
6155 unsigned Depth) const {
6156 APInt DemandedElts = getDemandAllEltsMask(Op);
6157 return computeKnownFPClass(Op, DemandedElts, InterestedClasses, Depth);
6158}
6159
6161 const APInt &DemandedElts,
6162 FPClassTest InterestedClasses,
6163 unsigned Depth) const {
6165
6166 if (const auto *CFP = dyn_cast<ConstantFPSDNode>(Op))
6167 return KnownFPClass(CFP->getValueAPF());
6168
6169 if (Depth >= MaxRecursionDepth)
6170 return Known;
6171
6172 if (Op.getOpcode() == ISD::UNDEF)
6173 return Known;
6174
6175 EVT VT = Op.getValueType();
6176 assert(VT.isFloatingPoint() && "Computing KnownFPClass on non-FP op!");
6177 assert((!VT.isFixedLengthVector() ||
6178 DemandedElts.getBitWidth() == VT.getVectorNumElements()) &&
6179 "Unexpected vector size");
6180
6181 if (!DemandedElts)
6182 return Known;
6183
6184 unsigned Opcode = Op.getOpcode();
6185 switch (Opcode) {
6186 case ISD::POISON: {
6187 Known.setKnownFPClasses(fcNone);
6188 Known.setSignBit(false);
6189 break;
6190 }
6191 case ISD::FNEG: {
6192 Known = computeKnownFPClass(Op.getOperand(0), DemandedElts,
6193 InterestedClasses, Depth + 1);
6194 Known.fneg();
6195 break;
6196 }
6197 case ISD::BUILD_VECTOR: {
6198 assert(!VT.isScalableVector());
6199 bool First = true;
6200 for (unsigned I = 0, E = Op.getNumOperands(); I != E; ++I) {
6201 if (!DemandedElts[I])
6202 continue;
6203
6204 if (First) {
6205 Known =
6206 computeKnownFPClass(Op.getOperand(I), InterestedClasses, Depth + 1);
6207 First = false;
6208 } else {
6209 Known |=
6210 computeKnownFPClass(Op.getOperand(I), InterestedClasses, Depth + 1);
6211 }
6212
6213 if (Known.isUnknown())
6214 break;
6215 }
6216 break;
6217 }
6219 SDValue Src = Op.getOperand(0);
6220 auto *CIdx = dyn_cast<ConstantSDNode>(Op.getOperand(1));
6221 EVT SrcVT = Src.getValueType();
6222 if (SrcVT.isFixedLengthVector() && CIdx) {
6223 if (CIdx->getAPIntValue().ult(SrcVT.getVectorNumElements())) {
6224 APInt DemandedSrcElts = APInt::getOneBitSet(
6225 SrcVT.getVectorNumElements(), CIdx->getZExtValue());
6226 Known = computeKnownFPClass(Src, DemandedSrcElts, InterestedClasses,
6227 Depth + 1);
6228 } else {
6229 // Out of bounds index is poison.
6230 Known.setKnownFPClasses(fcNone);
6231 }
6232 } else {
6233 Known = computeKnownFPClass(Src, InterestedClasses, Depth + 1);
6234 }
6235 break;
6236 }
6237 case ISD::SPLAT_VECTOR: {
6238 Known = computeKnownFPClass(Op.getOperand(0), InterestedClasses, Depth + 1);
6239 break;
6240 }
6241 case ISD::BITCAST: {
6242 // FIXME: It should not be necessary to check for an elementwise bitcast.
6243 // If a bitcast is not elementwise between vector / scalar types,
6244 // computeKnownBits already splices the known bits of the source elements
6245 // appropriately so as to line up with the bits of the result's demanded
6246 // elements.
6247 EVT SrcVT = Op.getOperand(0).getValueType();
6248 if (VT.isScalableVector() || SrcVT.isScalableVector())
6249 break;
6250 unsigned VTNumElts = VT.isVector() ? VT.getVectorNumElements() : 1;
6251 unsigned SrcVTNumElts = SrcVT.isVector() ? SrcVT.getVectorNumElements() : 1;
6252 if (VTNumElts != SrcVTNumElts)
6253 break;
6254
6255 KnownBits Bits = computeKnownBits(Op, DemandedElts, Depth + 1);
6257 break;
6258 }
6259 case ISD::FABS: {
6260 Known = computeKnownFPClass(Op.getOperand(0), DemandedElts,
6261 InterestedClasses, Depth + 1);
6262 Known.fabs();
6263 break;
6264 }
6265 case ISD::FCOPYSIGN: {
6266 Known = computeKnownFPClass(Op.getOperand(0), DemandedElts,
6267 InterestedClasses, Depth + 1);
6268 KnownFPClass KnownSign = computeKnownFPClass(Op.getOperand(1), DemandedElts,
6269 InterestedClasses, Depth + 1);
6270 Known.copysign(KnownSign);
6271 break;
6272 }
6273 case ISD::AssertNoFPClass: {
6274 Known = computeKnownFPClass(Op.getOperand(0), DemandedElts,
6275 InterestedClasses, Depth + 1);
6276 FPClassTest AssertedClasses =
6277 static_cast<FPClassTest>(Op->getConstantOperandVal(1));
6278 Known.setKnownFPClasses(Known.getKnownFPClasses() & ~AssertedClasses);
6279 break;
6280 }
6282 SDValue Src = Op.getOperand(0);
6283 EVT SrcVT = Src.getValueType();
6284 if (SrcVT.isFixedLengthVector()) {
6285 unsigned Idx = Op.getConstantOperandVal(1);
6286 unsigned NumSrcElts = SrcVT.getVectorNumElements();
6287
6288 APInt DemandedSrcElts = DemandedElts.zextOrTrunc(NumSrcElts).shl(Idx);
6289 Known = computeKnownFPClass(Src, DemandedSrcElts, InterestedClasses,
6290 Depth + 1);
6291 } else {
6292 Known = computeKnownFPClass(Src, InterestedClasses, Depth + 1);
6293 }
6294 break;
6295 }
6296 case ISD::INSERT_SUBVECTOR: {
6297 SDValue BaseVector = Op.getOperand(0);
6298 SDValue SubVector = Op.getOperand(1);
6299 EVT BaseVT = BaseVector.getValueType();
6300 if (BaseVT.isFixedLengthVector()) {
6301 unsigned Idx = Op.getConstantOperandVal(2);
6302 unsigned NumBaseElts = BaseVT.getVectorNumElements();
6303 unsigned NumSubElts = SubVector.getValueType().getVectorNumElements();
6304
6305 APInt DemandedMask =
6306 APInt::getBitsSet(NumBaseElts, Idx, Idx + NumSubElts);
6307 APInt DemandedSrcElts = DemandedElts & ~DemandedMask;
6308 APInt DemandedSubElts = DemandedElts.extractBits(NumSubElts, Idx);
6309
6310 if (!DemandedSrcElts.isZero())
6311 Known = computeKnownFPClass(BaseVector, DemandedSrcElts,
6312 InterestedClasses, Depth + 1);
6313 if (!DemandedSubElts.isZero()) {
6315 SubVector, DemandedSubElts, InterestedClasses, Depth + 1);
6316 Known = DemandedSrcElts.isZero() ? SubKnown : (Known | SubKnown);
6317 }
6318 } else {
6319 Known = computeKnownFPClass(SubVector, InterestedClasses, Depth + 1);
6320 if (!Known.isUnknown())
6321 Known |= computeKnownFPClass(BaseVector, InterestedClasses, Depth + 1);
6322 }
6323 break;
6324 }
6325 case ISD::SELECT:
6326 case ISD::VSELECT: {
6327 // TODO: Add adjustKnownFPClassForSelectArm clamp recognition as in
6328 // IR-level ValueTracking.
6329 KnownFPClass KnownFalseClass = computeKnownFPClass(
6330 Op.getOperand(2), DemandedElts, InterestedClasses, Depth + 1);
6331 if (KnownFalseClass.isUnknown())
6332 break;
6333 KnownFPClass KnownTrueClass = computeKnownFPClass(
6334 Op.getOperand(1), DemandedElts, InterestedClasses, Depth + 1);
6335 Known = KnownTrueClass.intersectWith(KnownFalseClass);
6336 break;
6337 }
6338 default:
6339 if (Opcode >= ISD::BUILTIN_OP_END || Opcode == ISD::INTRINSIC_WO_CHAIN ||
6340 Opcode == ISD::INTRINSIC_W_CHAIN || Opcode == ISD::INTRINSIC_VOID) {
6341 TLI->computeKnownFPClassForTargetNode(Op, Known, DemandedElts, *this,
6342 Depth);
6343 }
6344 break;
6345 }
6346
6347 return Known;
6348}
6349
6351 unsigned Depth) const {
6352 APInt DemandedElts = getDemandAllEltsMask(Op);
6353 return isKnownNeverNaN(Op, DemandedElts, SNaN, Depth);
6354}
6355
6357 bool SNaN, unsigned Depth) const {
6358 assert(!DemandedElts.isZero() && "No demanded elements");
6359
6360 // If we're told that NaNs won't happen, assume they won't.
6361 if (Op->getFlags().hasNoNaNs())
6362 return true;
6363
6364 if (Depth >= MaxRecursionDepth)
6365 return false; // Limit search depth.
6366
6367 unsigned Opcode = Op.getOpcode();
6368 switch (Opcode) {
6369 case ISD::FADD:
6370 case ISD::FSUB:
6371 case ISD::FMUL:
6372 case ISD::FDIV:
6373 case ISD::FREM:
6374 case ISD::FSIN:
6375 case ISD::FCOS:
6376 case ISD::FTAN:
6377 case ISD::FASIN:
6378 case ISD::FACOS:
6379 case ISD::FATAN:
6380 case ISD::FATAN2:
6381 case ISD::FSINH:
6382 case ISD::FCOSH:
6383 case ISD::FTANH:
6384 case ISD::FMA:
6385 case ISD::FMULADD:
6386 case ISD::FMAD: {
6387 if (SNaN)
6388 return true;
6389 // TODO: Need isKnownNeverInfinity
6390 return false;
6391 }
6392 case ISD::FCANONICALIZE:
6393 case ISD::FEXP:
6394 case ISD::FEXP2:
6395 case ISD::FEXP10:
6396 case ISD::FTRUNC:
6397 case ISD::FFLOOR:
6398 case ISD::FCEIL:
6399 case ISD::FROUND:
6400 case ISD::FROUNDEVEN:
6401 case ISD::LROUND:
6402 case ISD::LLROUND:
6403 case ISD::FRINT:
6404 case ISD::LRINT:
6405 case ISD::LLRINT:
6406 case ISD::FNEARBYINT:
6407 case ISD::FLDEXP: {
6408 if (SNaN)
6409 return true;
6410 return isKnownNeverNaN(Op.getOperand(0), DemandedElts, SNaN, Depth + 1);
6411 }
6412 case ISD::FABS:
6413 case ISD::FNEG:
6414 case ISD::FCOPYSIGN: {
6415 return isKnownNeverNaN(Op.getOperand(0), DemandedElts, SNaN, Depth + 1);
6416 }
6417 case ISD::SELECT:
6418 return isKnownNeverNaN(Op.getOperand(1), DemandedElts, SNaN, Depth + 1) &&
6419 isKnownNeverNaN(Op.getOperand(2), DemandedElts, SNaN, Depth + 1);
6420 case ISD::FP_EXTEND:
6421 case ISD::FP_ROUND: {
6422 if (SNaN)
6423 return true;
6424 return isKnownNeverNaN(Op.getOperand(0), DemandedElts, SNaN, Depth + 1);
6425 }
6426 case ISD::SINT_TO_FP:
6427 case ISD::UINT_TO_FP:
6428 return true;
6429 case ISD::FSQRT: // Need is known positive
6430 case ISD::FLOG:
6431 case ISD::FLOG2:
6432 case ISD::FLOG10:
6433 case ISD::FPOWI:
6434 case ISD::FPOW: {
6435 if (SNaN)
6436 return true;
6437 // TODO: Refine on operand
6438 return false;
6439 }
6440 case ISD::FMINNUM:
6441 case ISD::FMAXNUM:
6442 case ISD::FMINIMUMNUM:
6443 case ISD::FMAXIMUMNUM: {
6444 // Only one needs to be known not-nan, since it will be returned if the
6445 // other ends up being one.
6446 return isKnownNeverNaN(Op.getOperand(0), DemandedElts, SNaN, Depth + 1) ||
6447 isKnownNeverNaN(Op.getOperand(1), DemandedElts, SNaN, Depth + 1);
6448 }
6449 case ISD::FMINNUM_IEEE:
6450 case ISD::FMAXNUM_IEEE: {
6451 if (SNaN)
6452 return true;
6453 // This can return a NaN if either operand is an sNaN, or if both operands
6454 // are NaN.
6455 return (isKnownNeverNaN(Op.getOperand(0), DemandedElts, false, Depth + 1) &&
6456 isKnownNeverSNaN(Op.getOperand(1), DemandedElts, Depth + 1)) ||
6457 (isKnownNeverNaN(Op.getOperand(1), DemandedElts, false, Depth + 1) &&
6458 isKnownNeverSNaN(Op.getOperand(0), DemandedElts, Depth + 1));
6459 }
6460 case ISD::FMINIMUM:
6461 case ISD::FMAXIMUM: {
6462 // TODO: Does this quiet or return the origina NaN as-is?
6463 return isKnownNeverNaN(Op.getOperand(0), DemandedElts, SNaN, Depth + 1) &&
6464 isKnownNeverNaN(Op.getOperand(1), DemandedElts, SNaN, Depth + 1);
6465 }
6467 SDValue Src = Op.getOperand(0);
6468 auto *Idx = dyn_cast<ConstantSDNode>(Op.getOperand(1));
6469 EVT SrcVT = Src.getValueType();
6470 if (SrcVT.isFixedLengthVector() && Idx &&
6471 Idx->getAPIntValue().ult(SrcVT.getVectorNumElements())) {
6472 APInt DemandedSrcElts = APInt::getOneBitSet(SrcVT.getVectorNumElements(),
6473 Idx->getZExtValue());
6474 return isKnownNeverNaN(Src, DemandedSrcElts, SNaN, Depth + 1);
6475 }
6476 return isKnownNeverNaN(Src, SNaN, Depth + 1);
6477 }
6479 SDValue Src = Op.getOperand(0);
6480 if (Src.getValueType().isFixedLengthVector()) {
6481 unsigned Idx = Op.getConstantOperandVal(1);
6482 unsigned NumSrcElts = Src.getValueType().getVectorNumElements();
6483 APInt DemandedSrcElts = DemandedElts.zext(NumSrcElts).shl(Idx);
6484 return isKnownNeverNaN(Src, DemandedSrcElts, SNaN, Depth + 1);
6485 }
6486 return isKnownNeverNaN(Src, SNaN, Depth + 1);
6487 }
6488 case ISD::INSERT_SUBVECTOR: {
6489 SDValue BaseVector = Op.getOperand(0);
6490 SDValue SubVector = Op.getOperand(1);
6491 EVT BaseVectorVT = BaseVector.getValueType();
6492 if (BaseVectorVT.isFixedLengthVector()) {
6493 unsigned Idx = Op.getConstantOperandVal(2);
6494 unsigned NumBaseElts = BaseVectorVT.getVectorNumElements();
6495 unsigned NumSubElts = SubVector.getValueType().getVectorNumElements();
6496
6497 // Clear/Extract the bits at the position where the subvector will be
6498 // inserted.
6499 APInt DemandedMask =
6500 APInt::getBitsSet(NumBaseElts, Idx, Idx + NumSubElts);
6501 APInt DemandedSrcElts = DemandedElts & ~DemandedMask;
6502 APInt DemandedSubElts = DemandedElts.extractBits(NumSubElts, Idx);
6503
6504 bool NeverNaN = true;
6505 if (!DemandedSrcElts.isZero())
6506 NeverNaN &=
6507 isKnownNeverNaN(BaseVector, DemandedSrcElts, SNaN, Depth + 1);
6508 if (NeverNaN && !DemandedSubElts.isZero())
6509 NeverNaN &=
6510 isKnownNeverNaN(SubVector, DemandedSubElts, SNaN, Depth + 1);
6511 return NeverNaN;
6512 }
6513 return isKnownNeverNaN(BaseVector, SNaN, Depth + 1) &&
6514 isKnownNeverNaN(SubVector, SNaN, Depth + 1);
6515 }
6516 case ISD::BUILD_VECTOR: {
6517 unsigned NumElts = Op.getNumOperands();
6518 for (unsigned I = 0; I != NumElts; ++I)
6519 if (DemandedElts[I] &&
6520 !isKnownNeverNaN(Op.getOperand(I), SNaN, Depth + 1))
6521 return false;
6522 return true;
6523 }
6524 case ISD::SPLAT_VECTOR:
6525 return isKnownNeverNaN(Op.getOperand(0), SNaN, Depth + 1);
6526 case ISD::AssertNoFPClass: {
6527 FPClassTest NoFPClass =
6528 static_cast<FPClassTest>(Op.getConstantOperandVal(1));
6529 if ((NoFPClass & fcNan) == fcNan)
6530 return true;
6531 if (SNaN && (NoFPClass & fcSNan) == fcSNan)
6532 return true;
6533 return isKnownNeverNaN(Op.getOperand(0), DemandedElts, SNaN, Depth + 1);
6534 }
6535 default:
6536 if (Opcode >= ISD::BUILTIN_OP_END || Opcode == ISD::INTRINSIC_WO_CHAIN ||
6537 Opcode == ISD::INTRINSIC_W_CHAIN || Opcode == ISD::INTRINSIC_VOID) {
6538 return TLI->isKnownNeverNaNForTargetNode(Op, DemandedElts, *this, SNaN,
6539 Depth);
6540 }
6541 break;
6542 }
6543
6544 FPClassTest NanMask = SNaN ? fcSNan : fcNan;
6545 KnownFPClass Known = computeKnownFPClass(Op, DemandedElts, NanMask, Depth);
6546 return Known.isKnownNever(NanMask);
6547}
6548
6550 APInt DemandedElts = getDemandAllEltsMask(Op);
6551 return isKnownNeverLogicalZero(Op, DemandedElts, Depth);
6552}
6553
6555 const APInt &DemandedElts,
6556 unsigned Depth) const {
6557 assert(!DemandedElts.isZero() && "No demanded elements");
6558 EVT VT = Op.getValueType();
6560 computeKnownFPClass(Op, DemandedElts, fcZero | fcSubnormal, Depth);
6561 return Known.isKnownNeverLogicalZero(getDenormalMode(VT));
6562}
6563
6565 APInt DemandedElts = getDemandAllEltsMask(Op);
6566 return isKnownNeverZero(Op, DemandedElts, Depth);
6567}
6568
6570 unsigned Depth) const {
6571 if (Depth >= MaxRecursionDepth)
6572 return false; // Limit search depth.
6573
6574 EVT OpVT = Op.getValueType();
6575 unsigned BitWidth = OpVT.getScalarSizeInBits();
6576
6577 assert(!Op.getValueType().isFloatingPoint() &&
6578 "Floating point types unsupported - use isKnownNeverLogicalZero");
6579
6580 // If the value is a constant, we can obviously see if it is a zero or not.
6581 auto IsNeverZero = [BitWidth](const ConstantSDNode *C) {
6582 APInt V = C->getAPIntValue().zextOrTrunc(BitWidth);
6583 return !V.isZero();
6584 };
6585
6586 if (ISD::matchUnaryPredicate(Op, DemandedElts, IsNeverZero,
6587 /*AllowUndefs=*/false, /*AllowTruncation=*/true))
6588 return true;
6589
6590 // TODO: Recognize more cases here. Most of the cases are also incomplete to
6591 // some degree.
6592 switch (Op.getOpcode()) {
6593 default:
6594 break;
6595
6597 SDValue InVec = Op.getOperand(0);
6598 SDValue EltNo = Op.getOperand(1);
6599 EVT VecVT = InVec.getValueType();
6600
6601 // Skip scalable vectors or implicit extensions.
6602 if (VecVT.isScalableVector() ||
6603 OpVT.getScalarSizeInBits() != VecVT.getScalarSizeInBits())
6604 break;
6605
6606 // If we know the element index, just demand that vector element, else for
6607 // an unknown element index, ignore DemandedElts and demand them all.
6608 const unsigned NumSrcElts = VecVT.getVectorNumElements();
6609 APInt DemandedSrcElts = APInt::getAllOnes(NumSrcElts);
6610 auto *ConstEltNo = dyn_cast<ConstantSDNode>(EltNo);
6611 if (ConstEltNo && ConstEltNo->getAPIntValue().ult(NumSrcElts))
6612 DemandedSrcElts =
6613 APInt::getOneBitSet(NumSrcElts, ConstEltNo->getZExtValue());
6614
6615 return isKnownNeverZero(InVec, DemandedSrcElts, Depth + 1);
6616 }
6617
6618 case ISD::OR:
6619 return isKnownNeverZero(Op.getOperand(1), DemandedElts, Depth + 1) ||
6620 isKnownNeverZero(Op.getOperand(0), DemandedElts, Depth + 1);
6621
6622 case ISD::VSELECT:
6623 case ISD::SELECT:
6624 return isKnownNeverZero(Op.getOperand(1), DemandedElts, Depth + 1) &&
6625 isKnownNeverZero(Op.getOperand(2), DemandedElts, Depth + 1);
6626
6627 case ISD::SHL: {
6628 if (Op->getFlags().hasNoSignedWrap() || Op->getFlags().hasNoUnsignedWrap())
6629 return isKnownNeverZero(Op.getOperand(0), DemandedElts, Depth + 1);
6630 KnownBits ValKnown =
6631 computeKnownBits(Op.getOperand(0), DemandedElts, Depth + 1);
6632 // 1 << X is never zero.
6633 if (ValKnown.One[0])
6634 return true;
6635 // If max shift cnt of known ones is non-zero, result is non-zero.
6636 APInt MaxCnt = computeKnownBits(Op.getOperand(1), DemandedElts, Depth + 1)
6637 .getMaxValue();
6638 if (MaxCnt.ult(ValKnown.getBitWidth()) &&
6639 !ValKnown.One.shl(MaxCnt).isZero())
6640 return true;
6641 break;
6642 }
6643
6644 case ISD::VECTOR_SHUFFLE: {
6645 if (Op.getValueType().isScalableVector())
6646 return false;
6647
6648 unsigned NumElts = DemandedElts.getBitWidth();
6649
6650 // All demanded elements from LHS and RHS must be known non-zero.
6651 // Demanded elements with undef shuffle mask elements are unknown.
6652
6653 APInt DemandedLHS, DemandedRHS;
6654 auto *SVN = cast<ShuffleVectorSDNode>(Op);
6655 assert(NumElts == SVN->getMask().size() && "Unexpected vector size");
6656 if (!getShuffleDemandedElts(NumElts, SVN->getMask(), DemandedElts,
6657 DemandedLHS, DemandedRHS))
6658 return false;
6659
6660 return (!DemandedLHS ||
6661 isKnownNeverZero(Op.getOperand(0), DemandedLHS, Depth + 1)) &&
6662 (!DemandedRHS ||
6663 isKnownNeverZero(Op.getOperand(1), DemandedRHS, Depth + 1));
6664 }
6665
6666 case ISD::UADDSAT:
6667 case ISD::UMAX:
6668 return isKnownNeverZero(Op.getOperand(1), DemandedElts, Depth + 1) ||
6669 isKnownNeverZero(Op.getOperand(0), DemandedElts, Depth + 1);
6670
6671 case ISD::UMIN:
6672 return isKnownNeverZero(Op.getOperand(1), DemandedElts, Depth + 1) &&
6673 isKnownNeverZero(Op.getOperand(0), DemandedElts, Depth + 1);
6674
6675 // For smin/smax: If either operand is known negative/positive
6676 // respectively we don't need the other to be known at all.
6677 case ISD::SMAX: {
6678 KnownBits Op1 = computeKnownBits(Op.getOperand(1), DemandedElts, Depth + 1);
6679 if (Op1.isStrictlyPositive())
6680 return true;
6681
6682 KnownBits Op0 = computeKnownBits(Op.getOperand(0), DemandedElts, Depth + 1);
6683 if (Op0.isStrictlyPositive())
6684 return true;
6685
6686 if (Op1.isNonZero() && Op0.isNonZero())
6687 return true;
6688
6689 return isKnownNeverZero(Op.getOperand(1), DemandedElts, Depth + 1) &&
6690 isKnownNeverZero(Op.getOperand(0), DemandedElts, Depth + 1);
6691 }
6692 case ISD::SMIN: {
6693 KnownBits Op1 = computeKnownBits(Op.getOperand(1), DemandedElts, Depth + 1);
6694 if (Op1.isNegative())
6695 return true;
6696
6697 KnownBits Op0 = computeKnownBits(Op.getOperand(0), DemandedElts, Depth + 1);
6698 if (Op0.isNegative())
6699 return true;
6700
6701 if (Op1.isNonZero() && Op0.isNonZero())
6702 return true;
6703
6704 return isKnownNeverZero(Op.getOperand(1), DemandedElts, Depth + 1) &&
6705 isKnownNeverZero(Op.getOperand(0), DemandedElts, Depth + 1);
6706 }
6707
6708 case ISD::ROTL:
6709 case ISD::ROTR:
6710 case ISD::BITREVERSE:
6711 case ISD::BSWAP:
6712 case ISD::CTPOP:
6713 case ISD::ABS:
6715 return isKnownNeverZero(Op.getOperand(0), DemandedElts, Depth + 1);
6716
6717 case ISD::SRA:
6718 case ISD::SRL: {
6719 if (Op->getFlags().hasExact())
6720 return isKnownNeverZero(Op.getOperand(0), DemandedElts, Depth + 1);
6721 KnownBits ValKnown =
6722 computeKnownBits(Op.getOperand(0), DemandedElts, Depth + 1);
6723 if (ValKnown.isNegative())
6724 return true;
6725 // If max shift cnt of known ones is non-zero, result is non-zero.
6726 APInt MaxCnt = computeKnownBits(Op.getOperand(1), DemandedElts, Depth + 1)
6727 .getMaxValue();
6728 if (MaxCnt.ult(ValKnown.getBitWidth()) &&
6729 !ValKnown.One.lshr(MaxCnt).isZero())
6730 return true;
6731 break;
6732 }
6733 case ISD::UDIV:
6734 case ISD::SDIV:
6735 // div exact can only produce a zero if the dividend is zero.
6736 // TODO: For udiv this is also true if Op1 u<= Op0
6737 if (Op->getFlags().hasExact())
6738 return isKnownNeverZero(Op.getOperand(0), DemandedElts, Depth + 1);
6739 break;
6740
6741 case ISD::ADD:
6742 if (Op->getFlags().hasNoUnsignedWrap())
6743 if (isKnownNeverZero(Op.getOperand(1), DemandedElts, Depth + 1) ||
6744 isKnownNeverZero(Op.getOperand(0), DemandedElts, Depth + 1))
6745 return true;
6746 // TODO: There are a lot more cases we can prove for add.
6747 break;
6748
6749 case ISD::SUB: {
6750 if (isNullConstant(Op.getOperand(0)))
6751 return isKnownNeverZero(Op.getOperand(1), DemandedElts, Depth + 1);
6752
6753 std::optional<bool> ne = KnownBits::ne(
6754 computeKnownBits(Op.getOperand(0), DemandedElts, Depth + 1),
6755 computeKnownBits(Op.getOperand(1), DemandedElts, Depth + 1));
6756 return ne && *ne;
6757 }
6758
6759 case ISD::MUL:
6760 if (Op->getFlags().hasNoSignedWrap() || Op->getFlags().hasNoUnsignedWrap())
6761 if (isKnownNeverZero(Op.getOperand(1), Depth + 1) &&
6762 isKnownNeverZero(Op.getOperand(0), Depth + 1))
6763 return true;
6764 break;
6765
6766 case ISD::ZERO_EXTEND:
6767 case ISD::SIGN_EXTEND:
6768 return isKnownNeverZero(Op.getOperand(0), DemandedElts, Depth + 1);
6769 case ISD::VSCALE: {
6771 const APInt &Multiplier = Op.getConstantOperandAPInt(0);
6772 ConstantRange CR =
6773 getVScaleRange(&F, Op.getScalarValueSizeInBits()).multiply(Multiplier);
6774 if (!CR.contains(APInt(CR.getBitWidth(), 0)))
6775 return true;
6776 break;
6777 }
6778 }
6779
6780 return computeKnownBits(Op, DemandedElts, Depth).isNonZero();
6781}
6782
6784 if (ConstantFPSDNode *C1 = isConstOrConstSplatFP(Op, true))
6785 return !C1->isNegative();
6786
6787 switch (Op.getOpcode()) {
6788 case ISD::FABS:
6789 case ISD::FEXP:
6790 case ISD::FEXP2:
6791 case ISD::FEXP10:
6792 return true;
6793 default:
6794 return false;
6795 }
6796
6797 llvm_unreachable("covered opcode switch");
6798}
6799
6801 assert(Use.getValueType().isFloatingPoint());
6802 const SDNode *User = Use.getUser();
6803 if (User->getFlags().hasNoSignedZeros())
6804 return true;
6805
6806 unsigned OperandNo = Use.getOperandNo();
6807 // Check if this use is insensitive to the sign of zero
6808 switch (User->getOpcode()) {
6809 case ISD::SETCC:
6810 // Comparisons: IEEE-754 specifies +0.0 == -0.0.
6811 case ISD::FABS:
6812 // fabs always produces +0.0.
6813 return true;
6814 case ISD::FCOPYSIGN:
6815 // copysign overwrites the sign bit of the first operand.
6816 return OperandNo == 0;
6817 case ISD::FADD:
6818 case ISD::FSUB: {
6819 // Arithmetic with non-zero constants fixes the uncertainty around the
6820 // sign bit.
6821 SDValue Other = User->getOperand(1 - OperandNo);
6823 }
6824 case ISD::FP_TO_SINT:
6825 case ISD::FP_TO_UINT:
6826 // fp-to-int conversions normalize signed zeros.
6827 return true;
6828 default:
6829 return false;
6830 }
6831}
6832
6834 if (Op->getFlags().hasNoSignedZeros())
6835 return true;
6836 // FIXME: Limit the amount of checked uses to not introduce a compile-time
6837 // regression. Ideally, this should be implemented as a demanded-bits
6838 // optimization that stems from the users.
6839 if (Op->use_size() > 2)
6840 return false;
6841 return all_of(Op->uses(),
6842 [&](const SDUse &Use) { return canIgnoreSignBitOfZero(Use); });
6843}
6844
6846 // Check the obvious case.
6847 if (A == B) return true;
6848
6849 // For negative and positive zero.
6852 if (CA->isZero() && CB->isZero()) return true;
6853
6854 // Otherwise they may not be equal.
6855 return false;
6856}
6857
6858// Only bits set in Mask must be negated, other bits may be arbitrary.
6860 if (isBitwiseNot(V, AllowUndefs))
6861 return V.getOperand(0);
6862
6863 // Handle any_extend (not (truncate X)) pattern, where Mask only sets
6864 // bits in the non-extended part.
6865 ConstantSDNode *MaskC = isConstOrConstSplat(Mask);
6866 if (!MaskC || V.getOpcode() != ISD::ANY_EXTEND)
6867 return SDValue();
6868 SDValue ExtArg = V.getOperand(0);
6869 if (ExtArg.getScalarValueSizeInBits() >=
6870 MaskC->getAPIntValue().getActiveBits() &&
6871 isBitwiseNot(ExtArg, AllowUndefs) &&
6872 ExtArg.getOperand(0).getOpcode() == ISD::TRUNCATE &&
6873 ExtArg.getOperand(0).getOperand(0).getValueType() == V.getValueType())
6874 return ExtArg.getOperand(0).getOperand(0);
6875 return SDValue();
6876}
6877
6879 // Match masked merge pattern (X & ~M) op (Y & M)
6880 // Including degenerate case (X & ~M) op M
6881 auto MatchNoCommonBitsPattern = [&](SDValue Not, SDValue Mask,
6882 SDValue Other) {
6883 if (SDValue NotOperand =
6884 getBitwiseNotOperand(Not, Mask, /* AllowUndefs */ true)) {
6885 if (NotOperand->getOpcode() == ISD::ZERO_EXTEND ||
6886 NotOperand->getOpcode() == ISD::TRUNCATE)
6887 NotOperand = NotOperand->getOperand(0);
6888
6889 if (Other == NotOperand)
6890 return true;
6891 if (Other->getOpcode() == ISD::AND)
6892 return NotOperand == Other->getOperand(0) ||
6893 NotOperand == Other->getOperand(1);
6894 }
6895 return false;
6896 };
6897
6898 if (A->getOpcode() == ISD::ZERO_EXTEND || A->getOpcode() == ISD::TRUNCATE)
6899 A = A->getOperand(0);
6900
6901 if (B->getOpcode() == ISD::ZERO_EXTEND || B->getOpcode() == ISD::TRUNCATE)
6902 B = B->getOperand(0);
6903
6904 if (A->getOpcode() == ISD::AND)
6905 return MatchNoCommonBitsPattern(A->getOperand(0), A->getOperand(1), B) ||
6906 MatchNoCommonBitsPattern(A->getOperand(1), A->getOperand(0), B);
6907 return false;
6908}
6909
6910// FIXME: unify with llvm::haveNoCommonBitsSet.
6912 assert(A.getValueType() == B.getValueType() &&
6913 "Values must have the same type");
6916 return true;
6919}
6920
6921static SDValue FoldSTEP_VECTOR(const SDLoc &DL, EVT VT, SDValue Step,
6922 SelectionDAG &DAG) {
6923 if (cast<ConstantSDNode>(Step)->isZero())
6924 return DAG.getConstant(0, DL, VT);
6925
6926 return SDValue();
6927}
6928
6931 SelectionDAG &DAG) {
6932 int NumOps = Ops.size();
6933 assert(NumOps != 0 && "Can't build an empty vector!");
6934 assert(!VT.isScalableVector() &&
6935 "BUILD_VECTOR cannot be used with scalable types");
6936 assert(VT.getVectorNumElements() == (unsigned)NumOps &&
6937 "Incorrect element count in BUILD_VECTOR!");
6938
6939 // BUILD_VECTOR of UNDEFs is UNDEF.
6940 bool AllPoison = true;
6941 if (llvm::all_of(Ops, [&AllPoison](SDValue Op) {
6942 AllPoison &= Op.getOpcode() == ISD::POISON;
6943 return Op.isUndef();
6944 }))
6945 return AllPoison ? DAG.getPOISON(VT) : DAG.getUNDEF(VT);
6946
6947 // BUILD_VECTOR of seq extract/insert from the same vector + type is Identity.
6948 SDValue IdentitySrc;
6949 bool IsIdentity = true;
6950 for (int i = 0; i != NumOps; ++i) {
6951 if (Ops[i].getOpcode() != ISD::EXTRACT_VECTOR_ELT ||
6952 Ops[i].getOperand(0).getValueType() != VT ||
6953 (IdentitySrc && Ops[i].getOperand(0) != IdentitySrc) ||
6954 !isa<ConstantSDNode>(Ops[i].getOperand(1)) ||
6955 Ops[i].getConstantOperandAPInt(1) != i) {
6956 IsIdentity = false;
6957 break;
6958 }
6959 IdentitySrc = Ops[i].getOperand(0);
6960 }
6961 if (IsIdentity)
6962 return IdentitySrc;
6963
6964 return SDValue();
6965}
6966
6967/// Try to simplify vector concatenation to an input value, undef, or build
6968/// vector.
6971 SelectionDAG &DAG) {
6972 assert(!Ops.empty() && "Can't concatenate an empty list of vectors!");
6974 [Ops](SDValue Op) {
6975 return Ops[0].getValueType() == Op.getValueType();
6976 }) &&
6977 "Concatenation of vectors with inconsistent value types!");
6978 assert((Ops[0].getValueType().getVectorElementCount() * Ops.size()) ==
6979 VT.getVectorElementCount() &&
6980 "Incorrect element count in vector concatenation!");
6981
6982 if (Ops.size() == 1)
6983 return Ops[0];
6984
6985 // Concat of UNDEFs is UNDEF.
6986 bool AllPoison = true;
6987 if (llvm::all_of(Ops, [&AllPoison](SDValue Op) {
6988 AllPoison &= Op.getOpcode() == ISD::POISON;
6989 return Op.isUndef();
6990 }))
6991 return AllPoison ? DAG.getPOISON(VT) : DAG.getUNDEF(VT);
6992
6993 // Scan the operands and look for extract operations from a single source
6994 // that correspond to insertion at the same location via this concatenation:
6995 // concat (extract X, 0*subvec_elts), (extract X, 1*subvec_elts), ...
6996 SDValue IdentitySrc;
6997 bool IsIdentity = true;
6998 for (unsigned i = 0, e = Ops.size(); i != e; ++i) {
6999 SDValue Op = Ops[i];
7000 unsigned IdentityIndex = i * Op.getValueType().getVectorMinNumElements();
7001 if (Op.getOpcode() != ISD::EXTRACT_SUBVECTOR ||
7002 Op.getOperand(0).getValueType() != VT ||
7003 (IdentitySrc && Op.getOperand(0) != IdentitySrc) ||
7004 Op.getConstantOperandVal(1) != IdentityIndex) {
7005 IsIdentity = false;
7006 break;
7007 }
7008 assert((!IdentitySrc || IdentitySrc == Op.getOperand(0)) &&
7009 "Unexpected identity source vector for concat of extracts");
7010 IdentitySrc = Op.getOperand(0);
7011 }
7012 if (IsIdentity) {
7013 assert(IdentitySrc && "Failed to set source vector of extracts");
7014 return IdentitySrc;
7015 }
7016
7017 // The code below this point is only designed to work for fixed width
7018 // vectors, so we bail out for now.
7019 if (VT.isScalableVector())
7020 return SDValue();
7021
7022 // A CONCAT_VECTOR of scalar sources, such as UNDEF, BUILD_VECTOR and
7023 // single-element INSERT_VECTOR_ELT operands can be simplified to one big
7024 // BUILD_VECTOR.
7025 // FIXME: Add support for SCALAR_TO_VECTOR as well.
7026 EVT SVT = VT.getScalarType();
7028 for (SDValue Op : Ops) {
7029 EVT OpVT = Op.getValueType();
7030 if (Op.getOpcode() == ISD::POISON)
7031 Elts.append(OpVT.getVectorNumElements(), DAG.getPOISON(SVT));
7032 else if (Op.getOpcode() == ISD::UNDEF)
7033 Elts.append(OpVT.getVectorNumElements(), DAG.getUNDEF(SVT));
7034 else if (Op.getOpcode() == ISD::BUILD_VECTOR)
7035 Elts.append(Op->op_begin(), Op->op_end());
7036 else if (Op.getOpcode() == ISD::INSERT_VECTOR_ELT &&
7037 OpVT.getVectorNumElements() == 1 &&
7038 isNullConstant(Op.getOperand(2)))
7039 Elts.push_back(Op.getOperand(1));
7040 else
7041 return SDValue();
7042 }
7043
7044 // BUILD_VECTOR requires all inputs to be of the same type, find the
7045 // maximum type and extend them all.
7046 for (SDValue Op : Elts)
7047 SVT = (SVT.bitsLT(Op.getValueType()) ? Op.getValueType() : SVT);
7048
7049 if (SVT.bitsGT(VT.getScalarType())) {
7050 for (SDValue &Op : Elts) {
7051 if (Op.getOpcode() == ISD::POISON)
7052 Op = DAG.getPOISON(SVT);
7053 else if (Op.getOpcode() == ISD::UNDEF)
7054 Op = DAG.getUNDEF(SVT);
7055 else
7056 Op = DAG.getTargetLoweringInfo().isZExtFree(Op.getValueType(), SVT)
7057 ? DAG.getZExtOrTrunc(Op, DL, SVT)
7058 : DAG.getSExtOrTrunc(Op, DL, SVT);
7059 }
7060 }
7061
7062 SDValue V = DAG.getBuildVector(VT, DL, Elts);
7063 NewSDValueDbgMsg(V, "New node fold concat vectors: ", &DAG);
7064 return V;
7065}
7066
7067/// Gets or creates the specified node.
7068SDValue SelectionDAG::getNode(unsigned Opcode, const SDLoc &DL, EVT VT) {
7069 SDVTList VTs = getVTList(VT);
7070 SDNodeKey ID(Opcode, VTs, {});
7071 FoldingSetInsertToken InsertToken;
7072 if (SDNode *E = lookupNode(ID, DL, InsertToken))
7073 return SDValue(E, 0);
7074
7075 auto *N = newSDNode<SDNode>(Opcode, DL.getIROrder(), DL.getDebugLoc(), VTs);
7076 CSEMap.insert(N, InsertToken);
7077
7078 InsertNode(N);
7079 SDValue V = SDValue(N, 0);
7080 NewSDValueDbgMsg(V, "Creating new node: ", this);
7081 return V;
7082}
7083
7084SDValue SelectionDAG::getNode(unsigned Opcode, const SDLoc &DL, EVT VT,
7085 SDValue N1) {
7086 SDNodeFlags Flags;
7087 if (Inserter)
7088 Flags = Inserter->getFlags();
7089 return getNode(Opcode, DL, VT, N1, Flags);
7090}
7091
7092SDValue SelectionDAG::getNode(unsigned Opcode, const SDLoc &DL, EVT VT,
7093 SDValue N1, const SDNodeFlags Flags) {
7094 assert(N1.getOpcode() != ISD::DELETED_NODE && "Operand is DELETED_NODE!");
7095
7096 // Constant fold unary operations with a vector integer or float operand.
7097 switch (Opcode) {
7098 default:
7099 // FIXME: Entirely reasonable to perform folding of other unary
7100 // operations here as the need arises.
7101 break;
7102 case ISD::FNEG:
7103 case ISD::FABS:
7104 case ISD::FCEIL:
7105 case ISD::FTRUNC:
7106 case ISD::FFLOOR:
7107 case ISD::FP_EXTEND:
7108 case ISD::FP_TO_SINT:
7109 case ISD::FP_TO_UINT:
7110 case ISD::FP_TO_FP16:
7111 case ISD::FP_TO_BF16:
7112 case ISD::TRUNCATE:
7113 case ISD::ANY_EXTEND:
7114 case ISD::ZERO_EXTEND:
7115 case ISD::SIGN_EXTEND:
7116 case ISD::UINT_TO_FP:
7117 case ISD::SINT_TO_FP:
7118 case ISD::FP16_TO_FP:
7119 case ISD::BF16_TO_FP:
7120 case ISD::BITCAST:
7121 case ISD::ABS:
7123 case ISD::BITREVERSE:
7124 case ISD::BSWAP:
7125 case ISD::CTLZ:
7127 case ISD::CTTZ:
7129 case ISD::CTPOP:
7130 case ISD::PARITY:
7131 case ISD::CTLS:
7132 case ISD::VECREDUCE_ADD:
7137 case ISD::VECREDUCE_MUL:
7138 case ISD::VECREDUCE_AND:
7139 case ISD::VECREDUCE_OR:
7140 case ISD::VECREDUCE_XOR:
7141 case ISD::STEP_VECTOR: {
7142 SDValue Ops = {N1};
7143 if (SDValue Fold = FoldConstantArithmetic(Opcode, DL, VT, Ops))
7144 return Fold;
7145 }
7146 }
7147
7148 unsigned OpOpcode = N1.getNode()->getOpcode();
7149 switch (Opcode) {
7150 case ISD::STEP_VECTOR:
7151 assert(VT.isScalableVector() &&
7152 "STEP_VECTOR can only be used with scalable types");
7153 assert(OpOpcode == ISD::TargetConstant &&
7154 VT.getVectorElementType() == N1.getValueType() &&
7155 "Unexpected step operand");
7156 break;
7157 case ISD::FREEZE:
7158 assert(VT == N1.getValueType() && "Unexpected VT!");
7160 return N1;
7161 break;
7162 case ISD::TokenFactor:
7163 case ISD::MERGE_VALUES:
7165 return N1; // Factor, merge or concat of one node? No need.
7166 case ISD::BUILD_VECTOR: {
7167 // Attempt to simplify BUILD_VECTOR.
7168 SDValue Ops[] = {N1};
7169 if (SDValue V = FoldBUILD_VECTOR(DL, VT, Ops, *this))
7170 return V;
7171 break;
7172 }
7173 case ISD::FP_ROUND: llvm_unreachable("Invalid method to make FP_ROUND node");
7174 case ISD::FP_EXTEND:
7176 "Invalid FP cast!");
7177 if (N1.getValueType() == VT) return N1; // noop conversion.
7178 assert((!VT.isVector() || VT.getVectorElementCount() ==
7180 "Vector element count mismatch!");
7181 assert(N1.getValueType().bitsLT(VT) && "Invalid fpext node, dst < src!");
7182 if (N1.isUndef())
7183 return getUNDEF(VT);
7184 break;
7185 case ISD::FP_TO_SINT:
7186 case ISD::FP_TO_UINT:
7187 if (N1.isUndef())
7188 return getUNDEF(VT);
7189 break;
7190 case ISD::SINT_TO_FP:
7191 case ISD::UINT_TO_FP:
7192 // [us]itofp(undef) = 0, because the result value is bounded.
7193 if (N1.isUndef())
7194 return getConstantFP(0.0, DL, VT);
7195 break;
7196 case ISD::SIGN_EXTEND:
7197 assert(VT.isInteger() && N1.getValueType().isInteger() &&
7198 "Invalid SIGN_EXTEND!");
7199 assert(VT.isVector() == N1.getValueType().isVector() &&
7200 "SIGN_EXTEND result type type should be vector iff the operand "
7201 "type is vector!");
7202 if (N1.getValueType() == VT) return N1; // noop extension
7203 assert((!VT.isVector() || VT.getVectorElementCount() ==
7205 "Vector element count mismatch!");
7206 assert(N1.getValueType().bitsLT(VT) && "Invalid sext node, dst < src!");
7207 if (OpOpcode == ISD::SIGN_EXTEND || OpOpcode == ISD::ZERO_EXTEND) {
7208 SDNodeFlags Flags;
7209 if (OpOpcode == ISD::ZERO_EXTEND)
7210 Flags.setNonNeg(N1->getFlags().hasNonNeg());
7211 SDValue NewVal = getNode(OpOpcode, DL, VT, N1.getOperand(0), Flags);
7212 transferDbgValues(N1, NewVal);
7213 return NewVal;
7214 }
7215
7216 if (OpOpcode == ISD::POISON)
7217 return getPOISON(VT);
7218
7219 if (N1.isUndef())
7220 // sext(undef) = 0, because the top bits will all be the same.
7221 return getConstant(0, DL, VT);
7222
7223 // Skip unnecessary sext_inreg pattern:
7224 // (sext (trunc x)) -> x iff the upper bits are all signbits.
7225 if (OpOpcode == ISD::TRUNCATE) {
7226 SDValue OpOp = N1.getOperand(0);
7227 if (OpOp.getValueType() == VT) {
7228 unsigned NumSignExtBits =
7230 if (ComputeNumSignBits(OpOp) > NumSignExtBits) {
7231 transferDbgValues(N1, OpOp);
7232 return OpOp;
7233 }
7234 }
7235 }
7236 break;
7237 case ISD::ZERO_EXTEND:
7238 assert(VT.isInteger() && N1.getValueType().isInteger() &&
7239 "Invalid ZERO_EXTEND!");
7240 assert(VT.isVector() == N1.getValueType().isVector() &&
7241 "ZERO_EXTEND result type type should be vector iff the operand "
7242 "type is vector!");
7243 if (N1.getValueType() == VT) return N1; // noop extension
7244 assert((!VT.isVector() || VT.getVectorElementCount() ==
7246 "Vector element count mismatch!");
7247 assert(N1.getValueType().bitsLT(VT) && "Invalid zext node, dst < src!");
7248 if (OpOpcode == ISD::ZERO_EXTEND) { // (zext (zext x)) -> (zext x)
7249 SDNodeFlags Flags;
7250 Flags.setNonNeg(N1->getFlags().hasNonNeg());
7251 SDValue NewVal =
7252 getNode(ISD::ZERO_EXTEND, DL, VT, N1.getOperand(0), Flags);
7253 transferDbgValues(N1, NewVal);
7254 return NewVal;
7255 }
7256
7257 if (OpOpcode == ISD::POISON)
7258 return getPOISON(VT);
7259
7260 if (N1.isUndef())
7261 // zext(undef) = 0, because the top bits will be zero.
7262 return getConstant(0, DL, VT);
7263
7264 // Skip unnecessary zext_inreg pattern:
7265 // (zext (trunc x)) -> x iff the upper bits are known zero.
7266 // TODO: Remove (zext (trunc (and x, c))) exception which some targets
7267 // use to recognise zext_inreg patterns.
7268 if (OpOpcode == ISD::TRUNCATE) {
7269 SDValue OpOp = N1.getOperand(0);
7270 if (OpOp.getValueType() == VT) {
7271 if (OpOp.getOpcode() != ISD::AND) {
7274 if (MaskedValueIsZero(OpOp, HiBits)) {
7275 transferDbgValues(N1, OpOp);
7276 return OpOp;
7277 }
7278 }
7279 }
7280 }
7281 break;
7282 case ISD::ANY_EXTEND:
7283 assert(VT.isInteger() && N1.getValueType().isInteger() &&
7284 "Invalid ANY_EXTEND!");
7285 assert(VT.isVector() == N1.getValueType().isVector() &&
7286 "ANY_EXTEND result type type should be vector iff the operand "
7287 "type is vector!");
7288 if (N1.getValueType() == VT) return N1; // noop extension
7289 assert((!VT.isVector() || VT.getVectorElementCount() ==
7291 "Vector element count mismatch!");
7292 assert(N1.getValueType().bitsLT(VT) && "Invalid anyext node, dst < src!");
7293
7294 if (OpOpcode == ISD::ZERO_EXTEND || OpOpcode == ISD::SIGN_EXTEND ||
7295 OpOpcode == ISD::ANY_EXTEND) {
7296 SDNodeFlags Flags;
7297 if (OpOpcode == ISD::ZERO_EXTEND)
7298 Flags.setNonNeg(N1->getFlags().hasNonNeg());
7299 // (ext (zext x)) -> (zext x) and (ext (sext x)) -> (sext x)
7300 return getNode(OpOpcode, DL, VT, N1.getOperand(0), Flags);
7301 }
7302 if (N1.isUndef())
7303 return getUNDEF(VT);
7304
7305 // (ext (trunc x)) -> x
7306 if (OpOpcode == ISD::TRUNCATE) {
7307 SDValue OpOp = N1.getOperand(0);
7308 if (OpOp.getValueType() == VT) {
7309 transferDbgValues(N1, OpOp);
7310 return OpOp;
7311 }
7312 }
7313 break;
7314 case ISD::TRUNCATE:
7315 assert(VT.isInteger() && N1.getValueType().isInteger() &&
7316 "Invalid TRUNCATE!");
7317 assert(VT.isVector() == N1.getValueType().isVector() &&
7318 "TRUNCATE result type type should be vector iff the operand "
7319 "type is vector!");
7320 if (N1.getValueType() == VT) return N1; // noop truncate
7321 assert((!VT.isVector() || VT.getVectorElementCount() ==
7323 "Vector element count mismatch!");
7324 assert(N1.getValueType().bitsGT(VT) && "Invalid truncate node, src < dst!");
7325 if (OpOpcode == ISD::TRUNCATE)
7326 return getNode(ISD::TRUNCATE, DL, VT, N1.getOperand(0));
7327 if (OpOpcode == ISD::ZERO_EXTEND || OpOpcode == ISD::SIGN_EXTEND ||
7328 OpOpcode == ISD::ANY_EXTEND) {
7329 // If the source is smaller than the dest, we still need an extend.
7331 VT.getScalarType())) {
7332 SDNodeFlags Flags;
7333 if (OpOpcode == ISD::ZERO_EXTEND)
7334 Flags.setNonNeg(N1->getFlags().hasNonNeg());
7335 return getNode(OpOpcode, DL, VT, N1.getOperand(0), Flags);
7336 }
7337 if (N1.getOperand(0).getValueType().bitsGT(VT))
7338 return getNode(ISD::TRUNCATE, DL, VT, N1.getOperand(0));
7339 return N1.getOperand(0);
7340 }
7341 if (N1.isUndef())
7342 return getUNDEF(VT);
7343 if (OpOpcode == ISD::VSCALE && !NewNodesMustHaveLegalTypes)
7344 return getVScale(DL, VT,
7346 break;
7350 assert(VT.isVector() && "This DAG node is restricted to vector types.");
7351 assert(N1.getValueType().bitsLE(VT) &&
7352 "The input must be the same size or smaller than the result.");
7355 "The destination vector type must have fewer lanes than the input.");
7356 break;
7357 case ISD::ABS:
7358 assert(VT.isInteger() && VT == N1.getValueType() && "Invalid ABS!");
7359 if (N1.isUndef())
7360 return getConstant(0, DL, VT);
7361 break;
7363 assert(VT.isInteger() && VT == N1.getValueType() &&
7364 "Invalid ABS_MIN_POISON!");
7365 if (N1.isUndef())
7366 return getConstant(0, DL, VT);
7367 break;
7368 case ISD::BSWAP:
7369 assert(VT.isInteger() && VT == N1.getValueType() && "Invalid BSWAP!");
7370 assert((VT.getScalarSizeInBits() % 16 == 0) &&
7371 "BSWAP types must be a multiple of 16 bits!");
7372 if (N1.isUndef())
7373 return getUNDEF(VT);
7374 // bswap(bswap(X)) -> X.
7375 if (OpOpcode == ISD::BSWAP)
7376 return N1.getOperand(0);
7377 break;
7378 case ISD::BITREVERSE:
7379 assert(VT.isInteger() && VT == N1.getValueType() && "Invalid BITREVERSE!");
7380 if (N1.isUndef())
7381 return getUNDEF(VT);
7382 break;
7383 case ISD::BITCAST:
7385 "Cannot BITCAST between types of different sizes!");
7386 if (VT == N1.getValueType()) return N1; // noop conversion.
7387 if (OpOpcode == ISD::BITCAST) // bitconv(bitconv(x)) -> bitconv(x)
7388 return getNode(ISD::BITCAST, DL, VT, N1.getOperand(0));
7389 if (N1.isUndef())
7390 return getUNDEF(VT);
7391 break;
7393 assert(VT.isVector() && !N1.getValueType().isVector() &&
7394 (VT.getVectorElementType() == N1.getValueType() ||
7396 N1.getValueType().isInteger() &&
7398 "Illegal SCALAR_TO_VECTOR node!");
7399 if (N1.isUndef())
7400 return getUNDEF(VT);
7401 // scalar_to_vector(extract_vector_elt V, 0) -> V, top bits are undefined.
7402 if (OpOpcode == ISD::EXTRACT_VECTOR_ELT &&
7404 N1.getConstantOperandVal(1) == 0 &&
7405 N1.getOperand(0).getValueType() == VT)
7406 return N1.getOperand(0);
7407 break;
7408 case ISD::FNEG:
7409 // Negation of an unknown bag of bits is still completely undefined.
7410 if (N1.isUndef())
7411 return getUNDEF(VT);
7412
7413 if (OpOpcode == ISD::FNEG) // --X -> X
7414 return N1.getOperand(0);
7415 break;
7416 case ISD::FABS:
7417 if (OpOpcode == ISD::FNEG) // abs(-X) -> abs(X)
7418 return getNode(ISD::FABS, DL, VT, N1.getOperand(0));
7419 break;
7420 case ISD::VSCALE:
7421 assert(VT == N1.getValueType() && "Unexpected VT!");
7422 break;
7423 case ISD::CTPOP:
7424 if (N1.getValueType().getScalarType() == MVT::i1)
7425 return N1;
7426 break;
7427 case ISD::CTLZ:
7428 case ISD::CTTZ:
7429 if (N1.getValueType().getScalarType() == MVT::i1)
7430 return getNOT(DL, N1, N1.getValueType());
7431 break;
7432 case ISD::CTLS:
7433 if (N1.getValueType().getScalarType() == MVT::i1)
7434 return getConstant(0, DL, VT);
7435 break;
7436 case ISD::VECREDUCE_ADD:
7437 if (N1.getValueType().getScalarType() == MVT::i1)
7438 return getNode(ISD::VECREDUCE_XOR, DL, VT, N1);
7439 break;
7442 if (N1.getValueType().getScalarType() == MVT::i1)
7443 return getNode(ISD::VECREDUCE_OR, DL, VT, N1);
7444 break;
7447 if (N1.getValueType().getScalarType() == MVT::i1)
7448 return getNode(ISD::VECREDUCE_AND, DL, VT, N1);
7449 break;
7450 case ISD::SPLAT_VECTOR:
7451 assert(VT.isVector() && "Wrong return type!");
7452 // FIXME: Hexagon uses i32 scalar for a floating point zero vector so allow
7453 // that for now.
7455 (VT.isFloatingPoint() && N1.getValueType() == MVT::i32) ||
7457 N1.getValueType().isInteger() &&
7459 "Wrong operand type!");
7460 break;
7461 }
7462
7463 SDNode *N;
7464 SDVTList VTs = getVTList(VT);
7465 SDValue Ops[] = {N1};
7466 if (VT != MVT::Glue) { // Don't CSE glue producing nodes
7467 SDNodeKey ID(Opcode, VTs, Ops);
7468 FoldingSetInsertToken InsertToken;
7469 if (SDNode *E = lookupNode(ID, DL, InsertToken)) {
7470 E->intersectFlagsWith(Flags);
7471 return SDValue(E, 0);
7472 }
7473
7474 N = newSDNode<SDNode>(Opcode, DL.getIROrder(), DL.getDebugLoc(), VTs);
7475 N->setFlags(Flags);
7476 createOperands(N, Ops);
7477 CSEMap.insert(N, InsertToken);
7478 } else {
7479 N = newSDNode<SDNode>(Opcode, DL.getIROrder(), DL.getDebugLoc(), VTs);
7480 createOperands(N, Ops);
7481 }
7482
7483 InsertNode(N);
7484 SDValue V = SDValue(N, 0);
7485 NewSDValueDbgMsg(V, "Creating new node: ", this);
7486 return V;
7487}
7488
7489static APInt getIntegerIdentity(unsigned Opcode, unsigned BitWidth) {
7490 switch (Opcode) {
7491 default:
7492 llvm_unreachable("Unexpected integer identity opcode");
7493 case ISD::ADD:
7494 case ISD::OR:
7495 case ISD::XOR:
7496 case ISD::UMAX:
7497 return APInt::getZero(BitWidth);
7498 case ISD::MUL:
7499 return APInt(BitWidth, 1);
7500 case ISD::AND:
7501 case ISD::UMIN:
7503 case ISD::SMAX:
7505 case ISD::SMIN:
7507 }
7508}
7509
7510static std::optional<APInt> FoldValue(unsigned Opcode, const APInt &C1,
7511 const APInt &C2) {
7512 switch (Opcode) {
7513 case ISD::ADD: return C1 + C2;
7514 case ISD::SUB: return C1 - C2;
7515 case ISD::MUL: return C1 * C2;
7516 case ISD::AND: return C1 & C2;
7517 case ISD::OR: return C1 | C2;
7518 case ISD::XOR: return C1 ^ C2;
7519 case ISD::SHL: return C1 << C2;
7520 case ISD::SRL: return C1.lshr(C2);
7521 case ISD::SRA: return C1.ashr(C2);
7522 case ISD::ROTL: return C1.rotl(C2);
7523 case ISD::ROTR: return C1.rotr(C2);
7524 case ISD::SMIN: return C1.sle(C2) ? C1 : C2;
7525 case ISD::SMAX: return C1.sge(C2) ? C1 : C2;
7526 case ISD::UMIN: return C1.ule(C2) ? C1 : C2;
7527 case ISD::UMAX: return C1.uge(C2) ? C1 : C2;
7528 case ISD::SADDSAT: return C1.sadd_sat(C2);
7529 case ISD::UADDSAT: return C1.uadd_sat(C2);
7530 case ISD::SSUBSAT: return C1.ssub_sat(C2);
7531 case ISD::USUBSAT: return C1.usub_sat(C2);
7532 case ISD::SSHLSAT: return C1.sshl_sat(C2);
7533 case ISD::USHLSAT: return C1.ushl_sat(C2);
7534 case ISD::UDIV:
7535 if (!C2.getBoolValue())
7536 break;
7537 return C1.udiv(C2);
7538 case ISD::UREM:
7539 if (!C2.getBoolValue())
7540 break;
7541 return C1.urem(C2);
7542 case ISD::SDIV:
7543 if (!C2.getBoolValue())
7544 break;
7545 return C1.sdiv(C2);
7546 case ISD::SREM:
7547 if (!C2.getBoolValue())
7548 break;
7549 return C1.srem(C2);
7550 case ISD::AVGFLOORS:
7551 return APIntOps::avgFloorS(C1, C2);
7552 case ISD::AVGFLOORU:
7553 return APIntOps::avgFloorU(C1, C2);
7554 case ISD::AVGCEILS:
7555 return APIntOps::avgCeilS(C1, C2);
7556 case ISD::AVGCEILU:
7557 return APIntOps::avgCeilU(C1, C2);
7558 case ISD::ABDS:
7559 return APIntOps::abds(C1, C2);
7560 case ISD::ABDU:
7561 return APIntOps::abdu(C1, C2);
7562 case ISD::MULHS:
7563 return APIntOps::mulhs(C1, C2);
7564 case ISD::MULHU:
7565 return APIntOps::mulhu(C1, C2);
7566 case ISD::CLMUL:
7567 return APIntOps::clmul(C1, C2);
7568 case ISD::CLMULR:
7569 return APIntOps::clmulr(C1, C2);
7570 case ISD::CLMULH:
7571 return APIntOps::clmulh(C1, C2);
7572 case ISD::PEXT:
7573 return APIntOps::pext(C1, C2);
7574 case ISD::PDEP:
7575 return APIntOps::pdep(C1, C2);
7576 }
7577 return std::nullopt;
7578}
7579// Handle constant folding with UNDEF.
7580// TODO: Handle more cases.
7581static std::optional<APInt> FoldValueWithUndef(unsigned Opcode, const APInt &C1,
7582 bool IsUndef1, const APInt &C2,
7583 bool IsUndef2) {
7584 if (!(IsUndef1 || IsUndef2))
7585 return FoldValue(Opcode, C1, C2);
7586
7587 // Fold and(x, undef) -> 0
7588 // Fold mul(x, undef) -> 0
7589 if (Opcode == ISD::AND || Opcode == ISD::MUL)
7590 return APInt::getZero(C1.getBitWidth());
7591
7592 return std::nullopt;
7593}
7594
7596 const GlobalAddressSDNode *GA,
7597 const SDNode *N2) {
7598 if (GA->getOpcode() != ISD::GlobalAddress)
7599 return SDValue();
7600 if (!TLI->isOffsetFoldingLegal(GA))
7601 return SDValue();
7602 auto *C2 = dyn_cast<ConstantSDNode>(N2);
7603 if (!C2)
7604 return SDValue();
7605 int64_t Offset = C2->getSExtValue();
7606 switch (Opcode) {
7607 case ISD::ADD:
7608 case ISD::PTRADD:
7609 break;
7610 case ISD::SUB: Offset = -uint64_t(Offset); break;
7611 default: return SDValue();
7612 }
7613 return getGlobalAddress(GA->getGlobal(), SDLoc(C2), VT,
7614 GA->getOffset() + uint64_t(Offset));
7615}
7616
7618 switch (Opcode) {
7619 case ISD::SDIV:
7620 case ISD::UDIV:
7621 case ISD::SREM:
7622 case ISD::UREM: {
7623 // If a divisor is zero/undef or any element of a divisor vector is
7624 // zero/undef, the whole op is undef.
7625 assert(Ops.size() == 2 && "Div/rem should have 2 operands");
7626 SDValue Divisor = Ops[1];
7627 if (Divisor.isUndef() || isNullConstant(Divisor))
7628 return true;
7629
7630 return ISD::isBuildVectorOfConstantSDNodes(Divisor.getNode()) &&
7631 llvm::any_of(Divisor->op_values(),
7632 [](SDValue V) { return V.isUndef() ||
7633 isNullConstant(V); });
7634 // TODO: Handle signed overflow.
7635 }
7636 // TODO: Handle oversized shifts.
7637 default:
7638 return false;
7639 }
7640}
7641
7644 SDNodeFlags Flags) {
7645 // If the opcode is a target-specific ISD node, there's nothing we can
7646 // do here and the operand rules may not line up with the below, so
7647 // bail early.
7648 // We can't create a scalar CONCAT_VECTORS so skip it. It will break
7649 // for concats involving SPLAT_VECTOR. Concats of BUILD_VECTORS are handled by
7650 // foldCONCAT_VECTORS in getNode before this is called.
7651 if (Opcode >= ISD::BUILTIN_OP_END || Opcode == ISD::CONCAT_VECTORS)
7652 return SDValue();
7653
7654 unsigned NumOps = Ops.size();
7655 if (NumOps == 0)
7656 return SDValue();
7657
7658 if (isUndef(Opcode, Ops))
7659 return getUNDEF(VT);
7660
7661 // Handle unary special cases.
7662 if (NumOps == 1) {
7663 SDValue N1 = Ops[0];
7664
7665 // Constant fold unary operations with an integer constant operand. Even
7666 // opaque constant will be folded, because the folding of unary operations
7667 // doesn't create new constants with different values. Nevertheless, the
7668 // opaque flag is preserved during folding to prevent future folding with
7669 // other constants.
7670 if (auto *C = dyn_cast<ConstantSDNode>(N1)) {
7671 const APInt &Val = C->getAPIntValue();
7672 switch (Opcode) {
7673 case ISD::SIGN_EXTEND:
7674 return getConstant(Val.sextOrTrunc(VT.getSizeInBits()), DL, VT,
7675 C->isTargetOpcode(), C->isOpaque());
7676 case ISD::TRUNCATE:
7677 if (C->isOpaque())
7678 break;
7679 [[fallthrough]];
7680 case ISD::ZERO_EXTEND:
7681 return getConstant(Val.zextOrTrunc(VT.getSizeInBits()), DL, VT,
7682 C->isTargetOpcode(), C->isOpaque());
7683 case ISD::ANY_EXTEND:
7684 // Some targets like RISCV prefer to sign extend some types.
7685 if (TLI->isSExtCheaperThanZExt(N1.getValueType(), VT))
7686 return getConstant(Val.sextOrTrunc(VT.getSizeInBits()), DL, VT,
7687 C->isTargetOpcode(), C->isOpaque());
7688 return getConstant(Val.zextOrTrunc(VT.getSizeInBits()), DL, VT,
7689 C->isTargetOpcode(), C->isOpaque());
7690 case ISD::ABS:
7691 return getConstant(Val.abs(), DL, VT, C->isTargetOpcode(),
7692 C->isOpaque());
7694 if (Val.isMinSignedValue())
7695 return getPOISON(VT);
7696 return getConstant(Val.abs(), DL, VT, C->isTargetOpcode(),
7697 C->isOpaque());
7698 case ISD::BITREVERSE:
7699 return getConstant(Val.reverseBits(), DL, VT, C->isTargetOpcode(),
7700 C->isOpaque());
7701 case ISD::BSWAP:
7702 return getConstant(Val.byteSwap(), DL, VT, C->isTargetOpcode(),
7703 C->isOpaque());
7704 case ISD::CTPOP:
7705 return getConstant(Val.popcount(), DL, VT, C->isTargetOpcode(),
7706 C->isOpaque());
7707 case ISD::PARITY:
7708 return getConstant(Val.popcount() & 1, DL, VT, C->isTargetOpcode(),
7709 C->isOpaque());
7710 case ISD::CTLZ:
7712 return getConstant(Val.countl_zero(), DL, VT, C->isTargetOpcode(),
7713 C->isOpaque());
7714 case ISD::CTTZ:
7716 return getConstant(Val.countr_zero(), DL, VT, C->isTargetOpcode(),
7717 C->isOpaque());
7718 case ISD::CTLS:
7719 // CTLS returns the number of extra sign bits so subtract one.
7720 return getConstant(Val.getNumSignBits() - 1, DL, VT,
7721 C->isTargetOpcode(), C->isOpaque());
7722 case ISD::UINT_TO_FP:
7723 case ISD::SINT_TO_FP: {
7725 (void)FPV.convertFromAPInt(Val, Opcode == ISD::SINT_TO_FP,
7727 return getConstantFP(FPV, DL, VT);
7728 }
7729 case ISD::FP16_TO_FP:
7730 case ISD::BF16_TO_FP: {
7731 bool Ignored;
7732 APFloat FPV(Opcode == ISD::FP16_TO_FP ? APFloat::IEEEhalf()
7733 : APFloat::BFloat(),
7734 (Val.getBitWidth() == 16) ? Val : Val.trunc(16));
7735
7736 // This can return overflow, underflow, or inexact; we don't care.
7737 // FIXME need to be more flexible about rounding mode.
7739 &Ignored);
7740 return getConstantFP(FPV, DL, VT);
7741 }
7742 case ISD::STEP_VECTOR:
7743 if (SDValue V = FoldSTEP_VECTOR(DL, VT, N1, *this))
7744 return V;
7745 break;
7746 case ISD::BITCAST:
7747 if (VT == MVT::f16 && C->getValueType(0) == MVT::i16)
7748 return getConstantFP(APFloat(APFloat::IEEEhalf(), Val), DL, VT);
7749 if (VT == MVT::f32 && C->getValueType(0) == MVT::i32)
7750 return getConstantFP(APFloat(APFloat::IEEEsingle(), Val), DL, VT);
7751 if (VT == MVT::f64 && C->getValueType(0) == MVT::i64)
7752 return getConstantFP(APFloat(APFloat::IEEEdouble(), Val), DL, VT);
7753 if (VT == MVT::f128 && C->getValueType(0) == MVT::i128)
7754 return getConstantFP(APFloat(APFloat::IEEEquad(), Val), DL, VT);
7755 break;
7756 }
7757 }
7758
7759 // Constant fold unary operations with a floating point constant operand.
7760 if (auto *C = dyn_cast<ConstantFPSDNode>(N1)) {
7761 APFloat V = C->getValueAPF(); // make copy
7762 switch (Opcode) {
7763 case ISD::FNEG:
7764 V.changeSign();
7765 return getConstantFP(V, DL, VT);
7766 case ISD::FABS:
7767 V.clearSign();
7768 return getConstantFP(V, DL, VT);
7769 case ISD::FCEIL: {
7770 APFloat::opStatus fs = V.roundToIntegral(APFloat::rmTowardPositive);
7772 return getConstantFP(V, DL, VT);
7773 return SDValue();
7774 }
7775 case ISD::FTRUNC: {
7776 APFloat::opStatus fs = V.roundToIntegral(APFloat::rmTowardZero);
7778 return getConstantFP(V, DL, VT);
7779 return SDValue();
7780 }
7781 case ISD::FFLOOR: {
7782 APFloat::opStatus fs = V.roundToIntegral(APFloat::rmTowardNegative);
7784 return getConstantFP(V, DL, VT);
7785 return SDValue();
7786 }
7787 case ISD::FP_EXTEND: {
7788 bool ignored;
7789 // This can return overflow, underflow, or inexact; we don't care.
7790 // FIXME need to be more flexible about rounding mode.
7791 (void)V.convert(VT.getFltSemantics(), APFloat::rmNearestTiesToEven,
7792 &ignored);
7793 return getConstantFP(V, DL, VT);
7794 }
7795 case ISD::FP_TO_SINT:
7796 case ISD::FP_TO_UINT: {
7797 bool ignored;
7798 APSInt IntVal(VT.getSizeInBits(), Opcode == ISD::FP_TO_UINT);
7799 // FIXME need to be more flexible about rounding mode.
7801 V.convertToInteger(IntVal, APFloat::rmTowardZero, &ignored);
7802 if (s == APFloat::opInvalidOp) // inexact is OK, in fact usual
7803 break;
7804 return getConstant(IntVal, DL, VT);
7805 }
7806 case ISD::FP_TO_FP16:
7807 case ISD::FP_TO_BF16: {
7808 bool Ignored;
7809 // This can return overflow, underflow, or inexact; we don't care.
7810 // FIXME need to be more flexible about rounding mode.
7811 (void)V.convert(Opcode == ISD::FP_TO_FP16 ? APFloat::IEEEhalf()
7812 : APFloat::BFloat(),
7814 return getConstant(V.bitcastToAPInt().getZExtValue(), DL, VT);
7815 }
7816 case ISD::BITCAST:
7817 if (VT == MVT::i16 && C->getValueType(0) == MVT::f16)
7818 return getConstant((uint16_t)V.bitcastToAPInt().getZExtValue(), DL,
7819 VT);
7820 if (VT == MVT::i16 && C->getValueType(0) == MVT::bf16)
7821 return getConstant((uint16_t)V.bitcastToAPInt().getZExtValue(), DL,
7822 VT);
7823 if (VT == MVT::i32 && C->getValueType(0) == MVT::f32)
7824 return getConstant((uint32_t)V.bitcastToAPInt().getZExtValue(), DL,
7825 VT);
7826 if (VT == MVT::i64 && C->getValueType(0) == MVT::f64)
7827 return getConstant(V.bitcastToAPInt().getZExtValue(), DL, VT);
7828 break;
7829 }
7830 }
7831
7832 // Early-out if we failed to constant fold a bitcast.
7833 if (Opcode == ISD::BITCAST)
7834 return SDValue();
7835
7836 // Constant fold integer vector reductions with constant BUILD_VECTORs.
7837 if ((Opcode == ISD::VECREDUCE_ADD || Opcode == ISD::VECREDUCE_SMAX ||
7838 Opcode == ISD::VECREDUCE_SMIN || Opcode == ISD::VECREDUCE_UMAX ||
7839 Opcode == ISD::VECREDUCE_UMIN || Opcode == ISD::VECREDUCE_MUL ||
7840 Opcode == ISD::VECREDUCE_OR || Opcode == ISD::VECREDUCE_XOR ||
7841 Opcode == ISD::VECREDUCE_AND) &&
7843 unsigned EltBits = N1.getValueType().getScalarSizeInBits();
7844 unsigned BaseOpcode = ISD::getVecReduceBaseOpcode(Opcode);
7845 APInt Acc = getIntegerIdentity(BaseOpcode, EltBits);
7846 for (SDValue Elt : N1->op_values()) {
7847 if (Elt.getOpcode() == ISD::POISON)
7848 return getPOISON(VT);
7849 if (Elt.isUndef() || cast<ConstantSDNode>(Elt)->isOpaque())
7850 return SDValue();
7851 APInt Value = cast<ConstantSDNode>(Elt)->getAPIntValue().trunc(EltBits);
7852 std::optional<APInt> Folded = FoldValue(BaseOpcode, Acc, Value);
7853 assert(Folded &&
7854 "Expected vector reduction base opcode to be foldable");
7855 Acc = *Folded;
7856 }
7857 EVT EltVT = N1.getValueType().getScalarType();
7858 return getAnyExtOrTrunc(getConstant(Acc, DL, EltVT), DL, VT);
7859 }
7860 }
7861
7862 // Handle binops special cases.
7863 if (NumOps == 2) {
7864 if (SDValue CFP = foldConstantFPMath(Opcode, DL, VT, Ops))
7865 return CFP;
7866
7867 if (auto *C1 = dyn_cast<ConstantSDNode>(Ops[0])) {
7868 if (auto *C2 = dyn_cast<ConstantSDNode>(Ops[1])) {
7869 if (C1->isOpaque() || C2->isOpaque())
7870 return SDValue();
7871
7872 std::optional<APInt> FoldAttempt =
7873 FoldValue(Opcode, C1->getAPIntValue(), C2->getAPIntValue());
7874 if (!FoldAttempt)
7875 return SDValue();
7876
7877 SDValue Folded = getConstant(*FoldAttempt, DL, VT);
7878 assert((!Folded || !VT.isVector()) &&
7879 "Can't fold vectors ops with scalar operands");
7880 return Folded;
7881 }
7882 }
7883
7884 // fold (add Sym, c) -> Sym+c
7886 return FoldSymbolOffset(Opcode, VT, GA, Ops[1].getNode());
7887 if (TLI->isCommutativeBinOp(Opcode))
7889 return FoldSymbolOffset(Opcode, VT, GA, Ops[0].getNode());
7890
7891 // fold (sext_in_reg c1) -> c2
7892 if (Opcode == ISD::SIGN_EXTEND_INREG) {
7893 EVT EVT = cast<VTSDNode>(Ops[1])->getVT();
7894
7895 auto SignExtendInReg = [&](APInt Val, llvm::EVT ConstantVT) {
7896 unsigned FromBits = EVT.getScalarSizeInBits();
7897 Val <<= Val.getBitWidth() - FromBits;
7898 Val.ashrInPlace(Val.getBitWidth() - FromBits);
7899 return getConstant(Val, DL, ConstantVT);
7900 };
7901
7902 if (auto *C1 = dyn_cast<ConstantSDNode>(Ops[0])) {
7903 const APInt &Val = C1->getAPIntValue();
7904 return SignExtendInReg(Val, VT);
7905 }
7906
7908 SmallVector<SDValue, 8> ScalarOps;
7909 llvm::EVT OpVT = Ops[0].getOperand(0).getValueType();
7910 for (int I = 0, E = VT.getVectorNumElements(); I != E; ++I) {
7911 SDValue Op = Ops[0].getOperand(I);
7912 if (Op.isUndef()) {
7913 ScalarOps.push_back(getUNDEF(OpVT));
7914 continue;
7915 }
7916 const APInt &Val = cast<ConstantSDNode>(Op)->getAPIntValue();
7917 ScalarOps.push_back(SignExtendInReg(Val, OpVT));
7918 }
7919 return getBuildVector(VT, DL, ScalarOps);
7920 }
7921
7922 if (Ops[0].getOpcode() == ISD::SPLAT_VECTOR &&
7923 isa<ConstantSDNode>(Ops[0].getOperand(0)))
7924 return getNode(ISD::SPLAT_VECTOR, DL, VT,
7925 SignExtendInReg(Ops[0].getConstantOperandAPInt(0),
7926 Ops[0].getOperand(0).getValueType()));
7927 }
7928 }
7929
7930 // Handle fshl/fshr special cases.
7931 if (Opcode == ISD::FSHL || Opcode == ISD::FSHR) {
7932 auto *C1 = dyn_cast<ConstantSDNode>(Ops[0]);
7933 auto *C2 = dyn_cast<ConstantSDNode>(Ops[1]);
7934 auto *C3 = dyn_cast<ConstantSDNode>(Ops[2]);
7935
7936 if (C1 && C2 && C3) {
7937 if (C1->isOpaque() || C2->isOpaque() || C3->isOpaque())
7938 return SDValue();
7939 const APInt &V1 = C1->getAPIntValue(), &V2 = C2->getAPIntValue(),
7940 &V3 = C3->getAPIntValue();
7941
7942 APInt FoldedVal = Opcode == ISD::FSHL ? APIntOps::fshl(V1, V2, V3)
7943 : APIntOps::fshr(V1, V2, V3);
7944 return getConstant(FoldedVal, DL, VT);
7945 }
7946 }
7947
7948 // Handle fma/fmad special cases.
7949 if (Opcode == ISD::FMA || Opcode == ISD::FMAD || Opcode == ISD::FMULADD) {
7950 assert(VT.isFloatingPoint() && "This operator only applies to FP types!");
7951 assert(Ops[0].getValueType() == VT && Ops[1].getValueType() == VT &&
7952 Ops[2].getValueType() == VT && "FMA types must match!");
7956 if (C1 && C2 && C3) {
7957 APFloat V1 = C1->getValueAPF();
7958 const APFloat &V2 = C2->getValueAPF();
7959 const APFloat &V3 = C3->getValueAPF();
7960 if (Opcode == ISD::FMAD || Opcode == ISD::FMULADD) {
7961 V1.multiply(V2, APFloat::rmNearestTiesToEven);
7963 } else
7964 V1.fusedMultiplyAdd(V2, V3, APFloat::rmNearestTiesToEven);
7965 return getConstantFP(V1, DL, VT);
7966 }
7967 }
7968
7969 // This is for vector folding only from here on.
7970 if (!VT.isVector())
7971 return SDValue();
7972
7973 // Constant fold integer partial reductions with constant BUILD_VECTOR
7974 // operands. The reduction order is deliberately unspecified. Use the same
7975 // subvector layout as TargetLowering::expandPartialReduceMLA(), where input
7976 // lane I contributes to accumulator lane I % NumAccElts.
7977 if (Opcode == ISD::PARTIAL_REDUCE_SMLA ||
7978 Opcode == ISD::PARTIAL_REDUCE_UMLA ||
7979 Opcode == ISD::PARTIAL_REDUCE_SUMLA) {
7980 // These nodes have no scalar form, so unsupported cases must not fall
7981 // through to generic per-lane vector folding.
7982 if (!llvm::all_of(Ops, [](SDValue Op) {
7983 return ISD::isBuildVectorOfConstantSDNodes(Op.getNode());
7984 }))
7985 return SDValue();
7986
7987 unsigned AccEltBits = VT.getScalarSizeInBits();
7988 unsigned InputEltBits = Ops[1].getScalarValueSizeInBits();
7989 unsigned NumAccElts = VT.getVectorNumElements();
7990 unsigned NumInputElts = Ops[1].getValueType().getVectorNumElements();
7991 SmallVector<APInt, 8> Results(NumAccElts, APInt::getZero(AccEltBits));
7992 BitVector PoisonElts(NumAccElts);
7993
7994 for (unsigned I = 0; I != NumAccElts; ++I) {
7995 SDValue Elt = Ops[0].getOperand(I);
7996 if (Elt.getOpcode() == ISD::POISON) {
7997 PoisonElts.set(I);
7998 continue;
7999 }
8000 auto *C = dyn_cast<ConstantSDNode>(Elt);
8001 if (!C || C->isOpaque())
8002 return SDValue();
8003 Results[I] = C->getAPIntValue().trunc(AccEltBits);
8004 }
8005
8006 bool IsLHSSigned = Opcode != ISD::PARTIAL_REDUCE_UMLA;
8007 bool IsRHSSigned = Opcode == ISD::PARTIAL_REDUCE_SMLA;
8008 for (unsigned I = 0; I != NumInputElts; ++I) {
8009 const unsigned AccIdx = I % NumAccElts;
8010 SDValue LHSElt = Ops[1].getOperand(I);
8011 SDValue RHSElt = Ops[2].getOperand(I);
8012 if (LHSElt.getOpcode() == ISD::POISON ||
8013 RHSElt.getOpcode() == ISD::POISON) {
8014 PoisonElts.set(AccIdx);
8015 continue;
8016 }
8017
8018 auto *LHS = dyn_cast<ConstantSDNode>(LHSElt);
8019 auto *RHS = dyn_cast<ConstantSDNode>(RHSElt);
8020 if (!LHS || !RHS || LHS->isOpaque() || RHS->isOpaque())
8021 return SDValue();
8022
8023 APInt LHSVal = LHS->getAPIntValue().trunc(InputEltBits);
8024 APInt RHSVal = RHS->getAPIntValue().trunc(InputEltBits);
8025 LHSVal = IsLHSSigned ? LHSVal.sext(AccEltBits) : LHSVal.zext(AccEltBits);
8026 RHSVal = IsRHSSigned ? RHSVal.sext(AccEltBits) : RHSVal.zext(AccEltBits);
8027 Results[AccIdx] += LHSVal * RHSVal;
8028 }
8029
8030 // After type legalization the vector element type may not be a legal
8031 // scalar type (e.g. i16 on AArch64). Create the folded constants in the
8032 // promoted legal scalar type instead, matching the generic per-lane path
8033 // below. Bail out if legalization would narrow the type, since the lane
8034 // value would not fit.
8035 EVT AccEltVT = VT.getVectorElementType();
8036 EVT LegalSVT = AccEltVT;
8037 if (NewNodesMustHaveLegalTypes && LegalSVT.isInteger()) {
8038 LegalSVT = TLI->getTypeToTransformTo(*getContext(), LegalSVT);
8039 if (LegalSVT.bitsLT(AccEltVT))
8040 return SDValue();
8041 }
8042
8043 SmallVector<SDValue, 8> ResultOps;
8044 for (unsigned I = 0; I != NumAccElts; ++I)
8045 ResultOps.push_back(
8046 PoisonElts[I] ? getPOISON(LegalSVT)
8047 : getConstant(Results[I].sext(LegalSVT.getSizeInBits()),
8048 DL, LegalSVT));
8049 return getBuildVector(VT, DL, ResultOps);
8050 }
8051
8052 ElementCount NumElts = VT.getVectorElementCount();
8053
8054 // See if we can fold through any bitcasted integer ops.
8055 if (NumOps == 2 && VT.isFixedLengthVector() && VT.isInteger() &&
8056 Ops[0].getValueType() == VT && Ops[1].getValueType() == VT &&
8057 (Ops[0].getOpcode() == ISD::BITCAST ||
8058 Ops[1].getOpcode() == ISD::BITCAST)) {
8061 auto *BV1 = dyn_cast<BuildVectorSDNode>(N1);
8062 auto *BV2 = dyn_cast<BuildVectorSDNode>(N2);
8063 if (BV1 && BV2 && N1.getValueType().isInteger() &&
8064 N2.getValueType().isInteger()) {
8065 bool IsLE = getDataLayout().isLittleEndian();
8066 unsigned EltBits = VT.getScalarSizeInBits();
8067 SmallVector<APInt> RawBits1, RawBits2;
8068 BitVector UndefElts1, UndefElts2;
8069 if (BV1->getConstantRawBits(IsLE, EltBits, RawBits1, UndefElts1) &&
8070 BV2->getConstantRawBits(IsLE, EltBits, RawBits2, UndefElts2)) {
8071 SmallVector<APInt> RawBits;
8072 for (unsigned I = 0, E = NumElts.getFixedValue(); I != E; ++I) {
8073 std::optional<APInt> Fold = FoldValueWithUndef(
8074 Opcode, RawBits1[I], UndefElts1[I], RawBits2[I], UndefElts2[I]);
8075 if (!Fold)
8076 break;
8077 RawBits.push_back(*Fold);
8078 }
8079 if (RawBits.size() == NumElts.getFixedValue()) {
8080 // We have constant folded, but we might need to cast this again back
8081 // to the original (possibly legalized) type.
8082 EVT BVVT, BVEltVT;
8083 if (N1.getValueType() == VT) {
8084 BVVT = N1.getValueType();
8085 BVEltVT = BV1->getOperand(0).getValueType();
8086 } else {
8087 BVVT = N2.getValueType();
8088 BVEltVT = BV2->getOperand(0).getValueType();
8089 }
8090 unsigned BVEltBits = BVEltVT.getSizeInBits();
8091 SmallVector<APInt> DstBits;
8092 BitVector DstUndefs;
8094 DstBits, RawBits, DstUndefs,
8095 BitVector(RawBits.size(), false));
8096 SmallVector<SDValue> Ops(DstBits.size(), getUNDEF(BVEltVT));
8097 for (unsigned I = 0, E = DstBits.size(); I != E; ++I) {
8098 if (DstUndefs[I])
8099 continue;
8100 Ops[I] = getConstant(DstBits[I].sext(BVEltBits), DL, BVEltVT);
8101 }
8102 return getBitcast(VT, getBuildVector(BVVT, DL, Ops));
8103 }
8104 }
8105 }
8106 // Logic ops can be folded from raw integer bits - mainly for AVX512 masks.
8107 if (ISD::isBitwiseLogicOp(Opcode) && isa<ConstantSDNode>(N1) &&
8108 isa<ConstantSDNode>(N2)) {
8109 if (SDValue Res = FoldConstantArithmetic(Opcode, DL, N1.getValueType(),
8110 {N1, N2}, Flags))
8111 return getBitcast(VT, Res);
8112 }
8113 }
8114
8115 // Fold (mul step_vector(C0), C1) to (step_vector(C0 * C1)).
8116 // (shl step_vector(C0), C1) -> (step_vector(C0 << C1))
8117 if ((Opcode == ISD::MUL || Opcode == ISD::SHL) &&
8118 Ops[0].getOpcode() == ISD::STEP_VECTOR) {
8119 APInt RHSVal;
8120 if (ISD::isConstantSplatVector(Ops[1].getNode(), RHSVal)) {
8121 APInt NewStep = Opcode == ISD::MUL
8122 ? Ops[0].getConstantOperandAPInt(0) * RHSVal
8123 : Ops[0].getConstantOperandAPInt(0) << RHSVal;
8124 return getStepVector(DL, VT, NewStep);
8125 }
8126 }
8127
8128 auto IsScalarOrSameVectorSize = [NumElts](const SDValue &Op) {
8129 return !Op.getValueType().isVector() ||
8130 Op.getValueType().getVectorElementCount() == NumElts;
8131 };
8132
8133 auto IsBuildVectorSplatVectorOrUndef = [](const SDValue &Op) {
8134 return Op.isUndef() || Op.getOpcode() == ISD::CONDCODE ||
8135 Op.getOpcode() == ISD::BUILD_VECTOR ||
8136 Op.getOpcode() == ISD::SPLAT_VECTOR;
8137 };
8138
8139 // All operands must be vector types with the same number of elements as
8140 // the result type and must be either UNDEF or a build/splat vector
8141 // or UNDEF scalars.
8142 if (!llvm::all_of(Ops, IsBuildVectorSplatVectorOrUndef) ||
8143 !llvm::all_of(Ops, IsScalarOrSameVectorSize))
8144 return SDValue();
8145
8146 // If we are comparing vectors, then the result needs to be a i1 boolean that
8147 // is then extended back to the legal result type depending on how booleans
8148 // are represented.
8149 EVT SVT = (Opcode == ISD::SETCC ? MVT::i1 : VT.getScalarType());
8150 ISD::NodeType ExtendCode =
8151 (Opcode == ISD::SETCC && SVT != VT.getScalarType())
8152 ? TargetLowering::getExtendForContent(TLI->getBooleanContents(VT))
8154
8155 // Find legal integer scalar type for constant promotion and
8156 // ensure that its scalar size is at least as large as source.
8157 EVT LegalSVT = VT.getScalarType();
8158 if (NewNodesMustHaveLegalTypes && LegalSVT.isInteger()) {
8159 LegalSVT = TLI->getTypeToTransformTo(*getContext(), LegalSVT);
8160 if (LegalSVT.bitsLT(VT.getScalarType()))
8161 return SDValue();
8162 }
8163
8164 // For scalable vector types we know we're dealing with SPLAT_VECTORs. We
8165 // only have one operand to check. For fixed-length vector types we may have
8166 // a combination of BUILD_VECTOR and SPLAT_VECTOR.
8167 unsigned NumVectorElts = NumElts.isScalable() ? 1 : NumElts.getFixedValue();
8168
8169 // Constant fold each scalar lane separately.
8170 SmallVector<SDValue, 4> ScalarResults;
8171 for (unsigned I = 0; I != NumVectorElts; I++) {
8172 SmallVector<SDValue, 4> ScalarOps;
8173 for (SDValue Op : Ops) {
8174 EVT InSVT = Op.getValueType().getScalarType();
8175 if (Op.getOpcode() != ISD::BUILD_VECTOR &&
8176 Op.getOpcode() != ISD::SPLAT_VECTOR) {
8177 if (Op.isUndef())
8178 ScalarOps.push_back(getUNDEF(InSVT));
8179 else
8180 ScalarOps.push_back(Op);
8181 continue;
8182 }
8183
8184 SDValue ScalarOp =
8185 Op.getOperand(Op.getOpcode() == ISD::SPLAT_VECTOR ? 0 : I);
8186 EVT ScalarVT = ScalarOp.getValueType();
8187
8188 // Build vector (integer) scalar operands may need implicit
8189 // truncation - do this before constant folding.
8190 if (ScalarVT.isInteger() && ScalarVT.bitsGT(InSVT)) {
8191 // Don't create illegally-typed nodes unless they're constants or undef
8192 // - if we fail to constant fold we can't guarantee the (dead) nodes
8193 // we're creating will be cleaned up before being visited for
8194 // legalization.
8195 if (NewNodesMustHaveLegalTypes && !ScalarOp.isUndef() &&
8196 !isa<ConstantSDNode>(ScalarOp) &&
8197 TLI->getTypeAction(*getContext(), InSVT) !=
8199 return SDValue();
8200 ScalarOp = getNode(ISD::TRUNCATE, DL, InSVT, ScalarOp);
8201 }
8202
8203 ScalarOps.push_back(ScalarOp);
8204 }
8205
8206 // Constant fold the scalar operands.
8207 SDValue ScalarResult = getNode(Opcode, DL, SVT, ScalarOps, Flags);
8208
8209 // Scalar folding only succeeded if the result is a constant or UNDEF.
8210 if (!ScalarResult.isUndef() && ScalarResult.getOpcode() != ISD::Constant &&
8211 ScalarResult.getOpcode() != ISD::ConstantFP)
8212 return SDValue();
8213
8214 // Legalize the (integer) scalar constant if necessary. We only do
8215 // this once we know the folding succeeded, since otherwise we would
8216 // get a node with illegal type which has a user.
8217 if (LegalSVT != SVT)
8218 ScalarResult = getNode(ExtendCode, DL, LegalSVT, ScalarResult);
8219
8220 ScalarResults.push_back(ScalarResult);
8221 }
8222
8223 SDValue V = NumElts.isScalable() ? getSplatVector(VT, DL, ScalarResults[0])
8224 : getBuildVector(VT, DL, ScalarResults);
8225 NewSDValueDbgMsg(V, "New node fold constant vector: ", this);
8226 return V;
8227}
8228
8231 // TODO: Add support for unary/ternary fp opcodes.
8232 if (Ops.size() != 2)
8233 return SDValue();
8234
8235 // TODO: We don't do any constant folding for strict FP opcodes here, but we
8236 // should. That will require dealing with a potentially non-default
8237 // rounding mode, checking the "opStatus" return value from the APFloat
8238 // math calculations, and possibly other variations.
8239 SDValue N1 = Ops[0];
8240 SDValue N2 = Ops[1];
8241 ConstantFPSDNode *N1CFP = isConstOrConstSplatFP(N1, /*AllowUndefs*/ false);
8242 ConstantFPSDNode *N2CFP = isConstOrConstSplatFP(N2, /*AllowUndefs*/ false);
8243 if (N1CFP && N2CFP) {
8244 APFloat C1 = N1CFP->getValueAPF(); // make copy
8245 const APFloat &C2 = N2CFP->getValueAPF();
8246 switch (Opcode) {
8247 case ISD::FADD:
8249 return getConstantFP(C1, DL, VT);
8250 case ISD::FSUB:
8252 return getConstantFP(C1, DL, VT);
8253 case ISD::FMUL:
8255 return getConstantFP(C1, DL, VT);
8256 case ISD::FDIV:
8258 return getConstantFP(C1, DL, VT);
8259 case ISD::FREM:
8260 C1.mod(C2);
8261 return getConstantFP(C1, DL, VT);
8262 case ISD::FCOPYSIGN:
8263 C1.copySign(C2);
8264 return getConstantFP(C1, DL, VT);
8265 case ISD::FMINNUM:
8266 return getConstantFP(minnum(C1, C2), DL, VT);
8267 case ISD::FMAXNUM:
8268 return getConstantFP(maxnum(C1, C2), DL, VT);
8269 case ISD::FMINIMUM:
8270 return getConstantFP(minimum(C1, C2), DL, VT);
8271 case ISD::FMAXIMUM:
8272 return getConstantFP(maximum(C1, C2), DL, VT);
8273 case ISD::FMINIMUMNUM:
8274 return getConstantFP(minimumnum(C1, C2), DL, VT);
8275 case ISD::FMAXIMUMNUM:
8276 return getConstantFP(maximumnum(C1, C2), DL, VT);
8277 default: break;
8278 }
8279 }
8280 if (N1CFP && Opcode == ISD::FP_ROUND) {
8281 APFloat C1 = N1CFP->getValueAPF(); // make copy
8282 bool Unused;
8283 // This can return overflow, underflow, or inexact; we don't care.
8284 // FIXME need to be more flexible about rounding mode.
8286 &Unused);
8287 return getConstantFP(C1, DL, VT);
8288 }
8289
8290 switch (Opcode) {
8291 case ISD::FSUB:
8292 // -0.0 - undef --> undef (consistent with "fneg undef")
8293 if (ConstantFPSDNode *N1C = isConstOrConstSplatFP(N1, /*AllowUndefs*/ true))
8294 if (N1C && N1C->getValueAPF().isNegZero() && N2.isUndef())
8295 return getUNDEF(VT);
8296 [[fallthrough]];
8297
8298 case ISD::FADD:
8299 case ISD::FMUL:
8300 case ISD::FDIV:
8301 case ISD::FREM:
8302 // If both operands are undef, the result is undef. If 1 operand is undef,
8303 // the result is NaN. This should match the behavior of the IR optimizer.
8304 if (N1.isUndef() && N2.isUndef())
8305 return getUNDEF(VT);
8306 if (N1.isUndef() || N2.isUndef())
8308 }
8309 return SDValue();
8310}
8311
8313 const SDLoc &DL, EVT DstEltVT) {
8314 EVT SrcEltVT = BV->getValueType(0).getVectorElementType();
8315
8316 // If this is already the right type, we're done.
8317 if (SrcEltVT == DstEltVT)
8318 return SDValue(BV, 0);
8319
8320 unsigned SrcBitSize = SrcEltVT.getSizeInBits();
8321 unsigned DstBitSize = DstEltVT.getSizeInBits();
8322
8323 // If this is a conversion of N elements of one type to N elements of another
8324 // type, convert each element. This handles FP<->INT cases.
8325 if (SrcBitSize == DstBitSize) {
8327 for (SDValue Op : BV->op_values()) {
8328 // If the vector element type is not legal, the BUILD_VECTOR operands
8329 // are promoted and implicitly truncated. Make that explicit here.
8330 if (Op.getValueType() != SrcEltVT)
8331 Op = getNode(ISD::TRUNCATE, DL, SrcEltVT, Op);
8332 Ops.push_back(getBitcast(DstEltVT, Op));
8333 }
8334 EVT VT = EVT::getVectorVT(*getContext(), DstEltVT,
8336 return getBuildVector(VT, DL, Ops);
8337 }
8338
8339 // Otherwise, we're growing or shrinking the elements. To avoid having to
8340 // handle annoying details of growing/shrinking FP values, we convert them to
8341 // int first.
8342 if (SrcEltVT.isFloatingPoint()) {
8343 // Convert the input float vector to a int vector where the elements are the
8344 // same sizes.
8345 EVT IntEltVT = EVT::getIntegerVT(*getContext(), SrcEltVT.getSizeInBits());
8346 if (SDValue Tmp = FoldConstantBuildVector(BV, DL, IntEltVT))
8348 DstEltVT);
8349 return SDValue();
8350 }
8351
8352 // Now we know the input is an integer vector. If the output is a FP type,
8353 // convert to integer first, then to FP of the right size.
8354 if (DstEltVT.isFloatingPoint()) {
8355 EVT IntEltVT = EVT::getIntegerVT(*getContext(), DstEltVT.getSizeInBits());
8356 if (SDValue Tmp = FoldConstantBuildVector(BV, DL, IntEltVT))
8358 DstEltVT);
8359 return SDValue();
8360 }
8361
8362 // Okay, we know the src/dst types are both integers of differing types.
8363 assert(SrcEltVT.isInteger() && DstEltVT.isInteger());
8364
8365 // Extract the constant raw bit data.
8366 BitVector UndefElements;
8367 SmallVector<APInt> RawBits;
8368 bool IsLE = getDataLayout().isLittleEndian();
8369 if (!BV->getConstantRawBits(IsLE, DstBitSize, RawBits, UndefElements))
8370 return SDValue();
8371
8373 for (unsigned I = 0, E = RawBits.size(); I != E; ++I) {
8374 if (UndefElements[I])
8375 Ops.push_back(getUNDEF(DstEltVT));
8376 else
8377 Ops.push_back(getConstant(RawBits[I], DL, DstEltVT));
8378 }
8379
8380 EVT VT = EVT::getVectorVT(*getContext(), DstEltVT, Ops.size());
8381 return getBuildVector(VT, DL, Ops);
8382}
8383
8385 assert(Val.getValueType().isInteger() && "Invalid AssertAlign!");
8386
8387 // There's no need to assert on a byte-aligned pointer. All pointers are at
8388 // least byte aligned.
8389 if (A == Align(1))
8390 return Val;
8391
8392 SDVTList VTs = getVTList(Val.getValueType());
8393 SDValue Ops[] = {Val};
8394 SDNodeKey ID(ISD::AssertAlign, VTs, Ops);
8395 ID.AddInteger(A.value());
8396
8397 FoldingSetInsertToken InsertToken;
8398 if (SDNode *E = lookupNode(ID, DL, InsertToken))
8399 return SDValue(E, 0);
8400
8401 auto *N =
8402 newSDNode<AssertAlignSDNode>(DL.getIROrder(), DL.getDebugLoc(), VTs, A);
8403 createOperands(N, {Val});
8404
8405 CSEMap.insert(N, InsertToken);
8406 InsertNode(N);
8407
8408 SDValue V(N, 0);
8409 NewSDValueDbgMsg(V, "Creating new node: ", this);
8410 return V;
8411}
8412
8413SDValue SelectionDAG::getNode(unsigned Opcode, const SDLoc &DL, EVT VT,
8414 SDValue N1, SDValue N2) {
8415 SDNodeFlags Flags;
8416 if (Inserter)
8417 Flags = Inserter->getFlags();
8418 return getNode(Opcode, DL, VT, N1, N2, Flags);
8419}
8420
8422 SDValue &N2) const {
8423 if (!TLI->isCommutativeBinOp(Opcode))
8424 return;
8425
8426 // Canonicalize:
8427 // binop(const, nonconst) -> binop(nonconst, const)
8430 bool N1CFP = isConstantFPBuildVectorOrConstantFP(N1);
8431 bool N2CFP = isConstantFPBuildVectorOrConstantFP(N2);
8432 if ((N1C && !N2C) || (N1CFP && !N2CFP))
8433 std::swap(N1, N2);
8434
8435 // Canonicalize:
8436 // binop(splat(x), step_vector) -> binop(step_vector, splat(x))
8437 else if (N1.getOpcode() == ISD::SPLAT_VECTOR &&
8439 std::swap(N1, N2);
8440}
8441
8442SDValue SelectionDAG::getNode(unsigned Opcode, const SDLoc &DL, EVT VT,
8443 SDValue N1, SDValue N2, const SDNodeFlags Flags) {
8445 N2.getOpcode() != ISD::DELETED_NODE &&
8446 "Operand is DELETED_NODE!");
8447
8448 canonicalizeCommutativeBinop(Opcode, N1, N2);
8449
8450 auto *N1C = dyn_cast<ConstantSDNode>(N1);
8451 auto *N2C = dyn_cast<ConstantSDNode>(N2);
8452
8453 // Don't allow undefs in vector splats - we might be returning N2 when folding
8454 // to zero etc.
8455 ConstantSDNode *N2CV =
8456 isConstOrConstSplat(N2, /*AllowUndefs*/ false, /*AllowTruncation*/ true);
8457
8458 switch (Opcode) {
8459 default: break;
8460 case ISD::TokenFactor:
8461 assert(VT == MVT::Other && N1.getValueType() == MVT::Other &&
8462 N2.getValueType() == MVT::Other && "Invalid token factor!");
8463 // Fold trivial token factors.
8464 if (N1.getOpcode() == ISD::EntryToken) return N2;
8465 if (N2.getOpcode() == ISD::EntryToken) return N1;
8466 if (N1 == N2) return N1;
8467 break;
8468 case ISD::BUILD_VECTOR: {
8469 // Attempt to simplify BUILD_VECTOR.
8470 SDValue Ops[] = {N1, N2};
8471 if (SDValue V = FoldBUILD_VECTOR(DL, VT, Ops, *this))
8472 return V;
8473 break;
8474 }
8475 case ISD::CONCAT_VECTORS: {
8476 SDValue Ops[] = {N1, N2};
8477 if (SDValue V = foldCONCAT_VECTORS(DL, VT, Ops, *this))
8478 return V;
8479 break;
8480 }
8481 case ISD::AND:
8482 assert(VT.isInteger() && "This operator does not apply to FP types!");
8483 assert(N1.getValueType() == N2.getValueType() &&
8484 N1.getValueType() == VT && "Binary operator types must match!");
8485 // (X & 0) -> 0. This commonly occurs when legalizing i64 values, so it's
8486 // worth handling here.
8487 if (N2CV && N2CV->isZero())
8488 return N2;
8489 if (N2CV && N2CV->isAllOnes()) // X & -1 -> X
8490 return N1;
8491 break;
8492 case ISD::OR:
8493 case ISD::XOR:
8494 case ISD::ADD:
8495 case ISD::PTRADD:
8496 case ISD::SUB:
8497 assert(VT.isInteger() && "This operator does not apply to FP types!");
8498 assert(N1.getValueType() == N2.getValueType() &&
8499 N1.getValueType() == VT && "Binary operator types must match!");
8500 // The equal operand types requirement is unnecessarily strong for PTRADD.
8501 // However, the SelectionDAGBuilder does not generate PTRADDs with different
8502 // operand types, and we'd need to re-implement GEP's non-standard wrapping
8503 // logic everywhere where PTRADDs may be folded or combined to properly
8504 // support them. If/when we introduce pointer types to the SDAG, we will
8505 // need to relax this constraint.
8506
8507 // (X ^|+- 0) -> X. This commonly occurs when legalizing i64 values, so
8508 // it's worth handling here.
8509 if (N2CV && N2CV->isZero())
8510 return N1;
8511 if ((Opcode == ISD::ADD || Opcode == ISD::SUB) &&
8512 VT.getScalarType() == MVT::i1)
8513 return getNode(ISD::XOR, DL, VT, N1, N2);
8514 // Fold (add (vscale * C0), (vscale * C1)) to (vscale * (C0 + C1)).
8515 if (Opcode == ISD::ADD && N1.getOpcode() == ISD::VSCALE &&
8516 N2.getOpcode() == ISD::VSCALE) {
8517 const APInt &C1 = N1->getConstantOperandAPInt(0);
8518 const APInt &C2 = N2->getConstantOperandAPInt(0);
8519 return getVScale(DL, VT, C1 + C2);
8520 }
8521 break;
8522 case ISD::MUL:
8523 assert(VT.isInteger() && "This operator does not apply to FP types!");
8524 assert(N1.getValueType() == N2.getValueType() &&
8525 N1.getValueType() == VT && "Binary operator types must match!");
8526 if (VT.getScalarType() == MVT::i1)
8527 return getNode(ISD::AND, DL, VT, N1, N2);
8528 if (N2CV && N2CV->isZero())
8529 return N2;
8530 if (N2C && (N1.getOpcode() == ISD::VSCALE) && Flags.hasNoSignedWrap()) {
8531 const APInt &MulImm = N1->getConstantOperandAPInt(0);
8532 const APInt &N2CImm = N2C->getAPIntValue();
8533 return getVScale(DL, VT, MulImm * N2CImm);
8534 }
8535 break;
8536 case ISD::UDIV:
8537 case ISD::UREM:
8538 case ISD::MULHU:
8539 case ISD::MULHS:
8540 case ISD::SDIV:
8541 case ISD::SREM:
8542 case ISD::SADDSAT:
8543 case ISD::SSUBSAT:
8544 case ISD::UADDSAT:
8545 case ISD::USUBSAT:
8546 assert(VT.isInteger() && "This operator does not apply to FP types!");
8547 assert(N1.getValueType() == N2.getValueType() &&
8548 N1.getValueType() == VT && "Binary operator types must match!");
8549 if (VT.getScalarType() == MVT::i1) {
8550 // fold (add_sat x, y) -> (or x, y) for bool types.
8551 if (Opcode == ISD::SADDSAT || Opcode == ISD::UADDSAT)
8552 return getNode(ISD::OR, DL, VT, N1, N2);
8553 // fold (sub_sat x, y) -> (and x, ~y) for bool types.
8554 if (Opcode == ISD::SSUBSAT || Opcode == ISD::USUBSAT)
8555 return getNode(ISD::AND, DL, VT, N1, getNOT(DL, N2, VT));
8556 }
8557 break;
8558 case ISD::SCMP:
8559 case ISD::UCMP:
8560 assert(N1.getValueType() == N2.getValueType() &&
8561 "Types of operands of UCMP/SCMP must match");
8562 assert(N1.getValueType().isVector() == VT.isVector() &&
8563 "Operands and return type of must both be scalars or vectors");
8564 if (VT.isVector())
8567 "Result and operands must have the same number of elements");
8568 break;
8569 case ISD::AVGFLOORS:
8570 case ISD::AVGFLOORU:
8571 case ISD::AVGCEILS:
8572 case ISD::AVGCEILU:
8573 assert(VT.isInteger() && "This operator does not apply to FP types!");
8574 assert(N1.getValueType() == N2.getValueType() &&
8575 N1.getValueType() == VT && "Binary operator types must match!");
8576 break;
8577 case ISD::ABDS:
8578 case ISD::ABDU:
8579 assert(VT.isInteger() && "This operator does not apply to FP types!");
8580 assert(N1.getValueType() == N2.getValueType() &&
8581 N1.getValueType() == VT && "Binary operator types must match!");
8582 if (VT.getScalarType() == MVT::i1)
8583 return getNode(ISD::XOR, DL, VT, N1, N2);
8584 break;
8585 case ISD::SMIN:
8586 case ISD::UMAX:
8587 assert(VT.isInteger() && "This operator does not apply to FP types!");
8588 assert(N1.getValueType() == N2.getValueType() &&
8589 N1.getValueType() == VT && "Binary operator types must match!");
8590 if (VT.getScalarType() == MVT::i1)
8591 return getNode(ISD::OR, DL, VT, N1, N2);
8592 break;
8593 case ISD::SMAX:
8594 case ISD::UMIN:
8595 assert(VT.isInteger() && "This operator does not apply to FP types!");
8596 assert(N1.getValueType() == N2.getValueType() &&
8597 N1.getValueType() == VT && "Binary operator types must match!");
8598 if (VT.getScalarType() == MVT::i1)
8599 return getNode(ISD::AND, DL, VT, N1, N2);
8600 break;
8601 case ISD::FADD:
8602 case ISD::FSUB:
8603 case ISD::FMUL:
8604 case ISD::FDIV:
8605 case ISD::FREM:
8606 assert(VT.isFloatingPoint() && "This operator only applies to FP types!");
8607 assert(N1.getValueType() == N2.getValueType() &&
8608 N1.getValueType() == VT && "Binary operator types must match!");
8609 if (SDValue V = simplifyFPBinop(Opcode, N1, N2, Flags))
8610 return V;
8611 break;
8612 case ISD::FCOPYSIGN: // N1 and result must match. N1/N2 need not match.
8613 assert(N1.getValueType() == VT &&
8616 "Invalid FCOPYSIGN!");
8617 break;
8618 case ISD::SHL:
8619 if (N2C && (N1.getOpcode() == ISD::VSCALE) && Flags.hasNoSignedWrap()) {
8620 const APInt &MulImm = N1->getConstantOperandAPInt(0);
8621 const APInt &ShiftImm = N2C->getAPIntValue();
8622 return getVScale(DL, VT, MulImm << ShiftImm);
8623 }
8624 [[fallthrough]];
8625 case ISD::SRA:
8626 case ISD::SRL:
8627 if (SDValue V = simplifyShift(N1, N2))
8628 return V;
8629 [[fallthrough]];
8630 case ISD::ROTL:
8631 case ISD::ROTR:
8632 case ISD::SSHLSAT:
8633 case ISD::USHLSAT:
8634 assert(VT == N1.getValueType() &&
8635 "Shift operators return type must be the same as their first arg");
8636 assert(VT.isInteger() && N2.getValueType().isInteger() &&
8637 "Shifts only work on integers");
8638 assert((!VT.isVector() || VT == N2.getValueType()) &&
8639 "Vector shift amounts must be in the same as their first arg");
8640 // Verify that the shift amount VT is big enough to hold valid shift
8641 // amounts. This catches things like trying to shift an i1024 value by an
8642 // i8, which is easy to fall into in generic code that uses
8643 // TLI.getShiftAmount().
8646 "Invalid use of small shift amount with oversized value!");
8647
8648 // Always fold shifts of i1 values so the code generator doesn't need to
8649 // handle them. Since we know the size of the shift has to be less than the
8650 // size of the value, the shift/rotate count is guaranteed to be zero.
8651 if (VT == MVT::i1)
8652 return N1;
8653 if (N2CV && N2CV->isZero())
8654 return N1;
8655 break;
8656 case ISD::FP_ROUND:
8658 VT.bitsLE(N1.getValueType()) && N2C &&
8659 (N2C->getZExtValue() == 0 || N2C->getZExtValue() == 1) &&
8660 N2.getOpcode() == ISD::TargetConstant && "Invalid FP_ROUND!");
8661 if (N1.getValueType() == VT) return N1; // noop conversion.
8662 break;
8663 case ISD::IS_FPCLASS: {
8665 "IS_FPCLASS is used for a non-floating type");
8666 assert(isa<ConstantSDNode>(N2) && "FPClassTest is not Constant");
8667 // is.fpclass(poison, mask) -> poison
8668 if (N1.getOpcode() == ISD::POISON)
8669 return getPOISON(VT);
8670 FPClassTest Mask = static_cast<FPClassTest>(N2->getAsZExtVal());
8671 // If all tests are made, it doesn't matter what the value is.
8672 if ((Mask & fcAllFlags) == fcAllFlags)
8673 return getBoolConstant(true, DL, VT, N1.getValueType());
8674 if ((Mask & fcAllFlags) == 0)
8675 return getBoolConstant(false, DL, VT, N1.getValueType());
8676 break;
8677 }
8678 case ISD::AssertNoFPClass: {
8680 "AssertNoFPClass is used for a non-floating type");
8681 assert(isa<ConstantSDNode>(N2) && "NoFPClass is not Constant");
8682 FPClassTest NoFPClass = static_cast<FPClassTest>(N2->getAsZExtVal());
8683 assert(llvm::to_underlying(NoFPClass) <=
8685 "FPClassTest value too large");
8686 (void)NoFPClass;
8687 break;
8688 }
8689 case ISD::AssertSext:
8690 case ISD::AssertZext: {
8691 EVT EVT = cast<VTSDNode>(N2)->getVT();
8692 assert(VT == N1.getValueType() && "Not an inreg extend!");
8693 assert(VT.isInteger() && EVT.isInteger() &&
8694 "Cannot *_EXTEND_INREG FP types");
8695 assert(!EVT.isVector() &&
8696 "AssertSExt/AssertZExt type should be the vector element type "
8697 "rather than the vector type!");
8698 assert(EVT.bitsLE(VT.getScalarType()) && "Not extending!");
8699 if (VT.getScalarType() == EVT) return N1; // noop assertion.
8700 break;
8701 }
8703 EVT EVT = cast<VTSDNode>(N2)->getVT();
8704 assert(VT == N1.getValueType() && "Not an inreg extend!");
8705 assert(VT.isInteger() && EVT.isInteger() &&
8706 "Cannot *_EXTEND_INREG FP types");
8707 assert(EVT.isVector() == VT.isVector() &&
8708 "SIGN_EXTEND_INREG type should be vector iff the operand "
8709 "type is vector!");
8710 assert((!EVT.isVector() ||
8712 "Vector element counts must match in SIGN_EXTEND_INREG");
8713 assert(EVT.getScalarType().bitsLE(VT.getScalarType()) && "Not extending!");
8714 if (EVT == VT) return N1; // Not actually extending
8715 break;
8716 }
8718 case ISD::FP_TO_UINT_SAT: {
8719 assert(VT.isInteger() && cast<VTSDNode>(N2)->getVT().isInteger() &&
8720 N1.getValueType().isFloatingPoint() && "Invalid FP_TO_*INT_SAT");
8721 assert(N1.getValueType().isVector() == VT.isVector() &&
8722 "FP_TO_*INT_SAT type should be vector iff the operand type is "
8723 "vector!");
8724 assert((!VT.isVector() || VT.getVectorElementCount() ==
8726 "Vector element counts must match in FP_TO_*INT_SAT");
8727 assert(!cast<VTSDNode>(N2)->getVT().isVector() &&
8728 "Type to saturate to must be a scalar.");
8729 assert(cast<VTSDNode>(N2)->getVT().bitsLE(VT.getScalarType()) &&
8730 "Not extending!");
8731 break;
8732 }
8735 "The result of EXTRACT_VECTOR_ELT must be at least as wide as the \
8736 element type of the vector.");
8737
8738 // Extract from an undefined value or using an undefined index is undefined.
8739 if (N1.isUndef() || N2.isUndef())
8740 return getUNDEF(VT);
8741
8742 // EXTRACT_VECTOR_ELT of out-of-bounds element is POISON for fixed length
8743 // vectors. For scalable vectors we will provide appropriate support for
8744 // dealing with arbitrary indices.
8745 if (N2C && N1.getValueType().isFixedLengthVector() &&
8746 N2C->getAPIntValue().uge(N1.getValueType().getVectorNumElements()))
8747 return getPOISON(VT);
8748
8749 // EXTRACT_VECTOR_ELT of CONCAT_VECTORS is often formed while lowering is
8750 // expanding copies of large vectors from registers. This only works for
8751 // fixed length vectors, since we need to know the exact number of
8752 // elements.
8753 if (N2C && N1.getOpcode() == ISD::CONCAT_VECTORS &&
8755 unsigned Factor = N1.getOperand(0).getValueType().getVectorNumElements();
8756 return getExtractVectorElt(DL, VT,
8757 N1.getOperand(N2C->getZExtValue() / Factor),
8758 N2C->getZExtValue() % Factor);
8759 }
8760
8761 // EXTRACT_VECTOR_ELT of BUILD_VECTOR or SPLAT_VECTOR is often formed while
8762 // lowering is expanding large vector constants.
8763 if (N2C && (N1.getOpcode() == ISD::BUILD_VECTOR ||
8764 N1.getOpcode() == ISD::SPLAT_VECTOR)) {
8767 "BUILD_VECTOR used for scalable vectors");
8768 unsigned Index =
8769 N1.getOpcode() == ISD::BUILD_VECTOR ? N2C->getZExtValue() : 0;
8770 SDValue Elt = N1.getOperand(Index);
8771
8772 if (VT != Elt.getValueType())
8773 // If the vector element type is not legal, the BUILD_VECTOR operands
8774 // are promoted and implicitly truncated, and the result implicitly
8775 // extended. Make that explicit here.
8776 Elt = getAnyExtOrTrunc(Elt, DL, VT);
8777
8778 return Elt;
8779 }
8780
8781 // EXTRACT_VECTOR_ELT of INSERT_VECTOR_ELT is often formed when vector
8782 // operations are lowered to scalars.
8783 if (N1.getOpcode() == ISD::INSERT_VECTOR_ELT) {
8784 // If the indices are the same, return the inserted element else
8785 // if the indices are known different, extract the element from
8786 // the original vector.
8787 SDValue N1Op2 = N1.getOperand(2);
8789
8790 if (N1Op2C && N2C) {
8791 if (N1Op2C->getZExtValue() == N2C->getZExtValue()) {
8792 if (VT == N1.getOperand(1).getValueType())
8793 return N1.getOperand(1);
8794 if (VT.isFloatingPoint()) {
8796 return getFPExtendOrRound(N1.getOperand(1), DL, VT);
8797 }
8798 return getSExtOrTrunc(N1.getOperand(1), DL, VT);
8799 }
8800 return getNode(ISD::EXTRACT_VECTOR_ELT, DL, VT, N1.getOperand(0), N2);
8801 }
8802 }
8803
8804 // EXTRACT_VECTOR_ELT of v1iX EXTRACT_SUBVECTOR could be formed
8805 // when vector types are scalarized and v1iX is legal.
8806 // vextract (v1iX extract_subvector(vNiX, Idx)) -> vextract(vNiX,Idx).
8807 // Here we are completely ignoring the extract element index (N2),
8808 // which is fine for fixed width vectors, since any index other than 0
8809 // is undefined anyway. However, this cannot be ignored for scalable
8810 // vectors - in theory we could support this, but we don't want to do this
8811 // without a profitability check.
8812 if (N1.getOpcode() == ISD::EXTRACT_SUBVECTOR &&
8814 N1.getValueType().getVectorNumElements() == 1) {
8815 return getNode(ISD::EXTRACT_VECTOR_ELT, DL, VT, N1.getOperand(0),
8816 N1.getOperand(1));
8817 }
8818 break;
8820 assert(N2C && (unsigned)N2C->getZExtValue() < 2 && "Bad EXTRACT_ELEMENT!");
8821 assert(!N1.getValueType().isVector() && !VT.isVector() &&
8822 (N1.getValueType().isInteger() == VT.isInteger()) &&
8823 N1.getValueType() != VT &&
8824 "Wrong types for EXTRACT_ELEMENT!");
8825
8826 // EXTRACT_ELEMENT of BUILD_PAIR is often formed while legalize is expanding
8827 // 64-bit integers into 32-bit parts. Instead of building the extract of
8828 // the BUILD_PAIR, only to have legalize rip it apart, just do it now.
8829 if (N1.getOpcode() == ISD::BUILD_PAIR)
8830 return N1.getOperand(N2C->getZExtValue());
8831
8832 // EXTRACT_ELEMENT of a constant int is also very common.
8833 if (N1C) {
8834 unsigned ElementSize = VT.getSizeInBits();
8835 unsigned Shift = ElementSize * N2C->getZExtValue();
8836 const APInt &Val = N1C->getAPIntValue();
8837 return getConstant(Val.extractBits(ElementSize, Shift), DL, VT);
8838 }
8839 break;
8841 EVT N1VT = N1.getValueType();
8842 assert(VT.isVector() && N1VT.isVector() &&
8843 "Extract subvector VTs must be vectors!");
8845 "Extract subvector VTs must have the same element type!");
8846 assert((VT.isFixedLengthVector() || N1VT.isScalableVector()) &&
8847 "Cannot extract a scalable vector from a fixed length vector!");
8848 assert((VT.isScalableVector() != N1VT.isScalableVector() ||
8850 "Extract subvector must be from larger vector to smaller vector!");
8851 assert(N2C && "Extract subvector index must be a constant");
8852 assert((VT.isScalableVector() != N1VT.isScalableVector() ||
8853 (VT.getVectorMinNumElements() + N2C->getZExtValue()) <=
8854 N1VT.getVectorMinNumElements()) &&
8855 "Extract subvector overflow!");
8856 assert(N2C->getAPIntValue().getBitWidth() ==
8857 TLI->getVectorIdxWidth(getDataLayout()) &&
8858 "Constant index for EXTRACT_SUBVECTOR has an invalid size");
8859 assert(N2C->getZExtValue() % VT.getVectorMinNumElements() == 0 &&
8860 "Extract index is not a multiple of the output vector length");
8861
8862 // Trivial extraction.
8863 if (VT == N1VT)
8864 return N1;
8865
8866 // EXTRACT_SUBVECTOR of an UNDEF is an UNDEF.
8867 if (N1.isUndef())
8868 return getUNDEF(VT);
8869
8870 // EXTRACT_SUBVECTOR of CONCAT_VECTOR can be simplified if the pieces of
8871 // the concat have the same type as the extract.
8872 if (N1.getOpcode() == ISD::CONCAT_VECTORS &&
8873 VT == N1.getOperand(0).getValueType()) {
8874 unsigned Factor = VT.getVectorMinNumElements();
8875 return N1.getOperand(N2C->getZExtValue() / Factor);
8876 }
8877
8878 // EXTRACT_SUBVECTOR of INSERT_SUBVECTOR is often created
8879 // during shuffle legalization.
8880 if (N1.getOpcode() == ISD::INSERT_SUBVECTOR && N2 == N1.getOperand(2) &&
8881 VT == N1.getOperand(1).getValueType())
8882 return N1.getOperand(1);
8883 break;
8884 }
8885 }
8886
8887 if (N1.getOpcode() == ISD::POISON || N2.getOpcode() == ISD::POISON) {
8888 switch (Opcode) {
8889 case ISD::XOR:
8890 case ISD::ADD:
8891 case ISD::PTRADD:
8892 case ISD::SUB:
8894 case ISD::UDIV:
8895 case ISD::SDIV:
8896 case ISD::UREM:
8897 case ISD::SREM:
8898 case ISD::MUL:
8899 case ISD::AND:
8900 case ISD::SSUBSAT:
8901 case ISD::USUBSAT:
8902 case ISD::UMIN:
8903 case ISD::OR:
8904 case ISD::SADDSAT:
8905 case ISD::UADDSAT:
8906 case ISD::UMAX:
8907 case ISD::SMAX:
8908 case ISD::SMIN:
8909 // fold op(arg1, poison) -> poison, fold op(poison, arg2) -> poison.
8910 return N2.getOpcode() == ISD::POISON ? N2 : N1;
8911 }
8912 }
8913
8914 // Canonicalize an UNDEF to the RHS, even over a constant.
8915 if (N1.getOpcode() == ISD::UNDEF && N2.getOpcode() != ISD::UNDEF) {
8916 if (TLI->isCommutativeBinOp(Opcode)) {
8917 std::swap(N1, N2);
8918 } else {
8919 switch (Opcode) {
8920 case ISD::PTRADD:
8921 case ISD::SUB:
8922 // fold op(undef, non_undef_arg2) -> undef.
8923 return N1;
8925 case ISD::UDIV:
8926 case ISD::SDIV:
8927 case ISD::UREM:
8928 case ISD::SREM:
8929 case ISD::SSUBSAT:
8930 case ISD::USUBSAT:
8931 // fold op(undef, non_undef_arg2) -> 0.
8932 return getConstant(0, DL, VT);
8933 }
8934 }
8935 }
8936
8937 // Fold a bunch of operators when the RHS is undef.
8938 if (N2.getOpcode() == ISD::UNDEF) {
8939 switch (Opcode) {
8940 case ISD::XOR:
8941 if (N1.getOpcode() == ISD::UNDEF)
8942 // Handle undef ^ undef -> 0 special case. This is a common
8943 // idiom (misuse).
8944 return getConstant(0, DL, VT);
8945 [[fallthrough]];
8946 case ISD::ADD:
8947 case ISD::PTRADD:
8948 case ISD::SUB:
8949 // fold op(arg1, undef) -> undef.
8950 return N2;
8951 case ISD::UDIV:
8952 case ISD::SDIV:
8953 case ISD::UREM:
8954 case ISD::SREM:
8955 // fold op(arg1, undef) -> poison.
8956 return getPOISON(VT);
8957 case ISD::MUL:
8958 case ISD::AND:
8959 case ISD::SSUBSAT:
8960 case ISD::USUBSAT:
8961 case ISD::UMIN:
8962 // fold op(undef, undef) -> undef, fold op(arg1, undef) -> 0.
8963 return N1.getOpcode() == ISD::UNDEF ? N2 : getConstant(0, DL, VT);
8964 case ISD::OR:
8965 case ISD::SADDSAT:
8966 case ISD::UADDSAT:
8967 case ISD::UMAX:
8968 // fold op(undef, undef) -> undef, fold op(arg1, undef) -> -1.
8969 return N1.getOpcode() == ISD::UNDEF ? N2 : getAllOnesConstant(DL, VT);
8970 case ISD::SMAX:
8971 // fold op(undef, undef) -> undef, fold op(arg1, undef) -> MAX_INT.
8972 return N1.getOpcode() == ISD::UNDEF
8973 ? N2
8974 : getConstant(
8976 VT);
8977 case ISD::SMIN:
8978 // fold op(undef, undef) -> undef, fold op(arg1, undef) -> MIN_INT.
8979 return N1.getOpcode() == ISD::UNDEF
8980 ? N2
8981 : getConstant(
8983 VT);
8984 }
8985 }
8986
8987 // Perform trivial constant folding.
8988 if (SDValue SV = FoldConstantArithmetic(Opcode, DL, VT, {N1, N2}, Flags))
8989 return SV;
8990
8991 // Memoize this node if possible.
8992 SDNode *N;
8993 SDVTList VTs = getVTList(VT);
8994 SDValue Ops[] = {N1, N2};
8995 if (VT != MVT::Glue) {
8996 SDNodeKey ID(Opcode, VTs, Ops);
8997 FoldingSetInsertToken InsertToken;
8998 if (SDNode *E = lookupNode(ID, DL, InsertToken)) {
8999 E->intersectFlagsWith(Flags);
9000 return SDValue(E, 0);
9001 }
9002
9003 N = newSDNode<SDNode>(Opcode, DL.getIROrder(), DL.getDebugLoc(), VTs);
9004 N->setFlags(Flags);
9005 createOperands(N, Ops);
9006 CSEMap.insert(N, InsertToken);
9007 } else {
9008 N = newSDNode<SDNode>(Opcode, DL.getIROrder(), DL.getDebugLoc(), VTs);
9009 createOperands(N, Ops);
9010 }
9011
9012 InsertNode(N);
9013 SDValue V = SDValue(N, 0);
9014 NewSDValueDbgMsg(V, "Creating new node: ", this);
9015 return V;
9016}
9017
9018SDValue SelectionDAG::getNode(unsigned Opcode, const SDLoc &DL, EVT VT,
9019 SDValue N1, SDValue N2, SDValue N3) {
9020 SDNodeFlags Flags;
9021 if (Inserter)
9022 Flags = Inserter->getFlags();
9023 return getNode(Opcode, DL, VT, N1, N2, N3, Flags);
9024}
9025
9026SDValue SelectionDAG::getNode(unsigned Opcode, const SDLoc &DL, EVT VT,
9027 SDValue N1, SDValue N2, SDValue N3,
9028 const SDNodeFlags Flags) {
9030 N2.getOpcode() != ISD::DELETED_NODE &&
9031 N3.getOpcode() != ISD::DELETED_NODE &&
9032 "Operand is DELETED_NODE!");
9033 // Perform various simplifications.
9034 switch (Opcode) {
9035 case ISD::BUILD_VECTOR: {
9036 // Attempt to simplify BUILD_VECTOR.
9037 SDValue Ops[] = {N1, N2, N3};
9038 if (SDValue V = FoldBUILD_VECTOR(DL, VT, Ops, *this))
9039 return V;
9040 break;
9041 }
9042 case ISD::CONCAT_VECTORS: {
9043 SDValue Ops[] = {N1, N2, N3};
9044 if (SDValue V = foldCONCAT_VECTORS(DL, VT, Ops, *this))
9045 return V;
9046 break;
9047 }
9048 case ISD::SETCC: {
9049 assert(VT.isInteger() && "SETCC result type must be an integer!");
9050 assert(N1.getValueType() == N2.getValueType() &&
9051 "SETCC operands must have the same type!");
9052 assert(VT.isVector() == N1.getValueType().isVector() &&
9053 "SETCC type should be vector iff the operand type is vector!");
9054 assert((!VT.isVector() || VT.getVectorElementCount() ==
9056 "SETCC vector element counts must match!");
9057 // Use FoldSetCC to simplify SETCC's.
9058 if (SDValue V =
9059 FoldSetCC(VT, N1, N2, cast<CondCodeSDNode>(N3)->get(), DL, Flags))
9060 return V;
9061 break;
9062 }
9063 case ISD::SELECT:
9064 case ISD::VSELECT:
9065 if (SDValue V = simplifySelect(N1, N2, N3))
9066 return V;
9067 break;
9069 llvm_unreachable("should use getVectorShuffle constructor!");
9071 if (isNullConstant(N3))
9072 return N1;
9073 break;
9075 if (isNullConstant(N3))
9076 return N2;
9077 break;
9079 assert(VT.isVector() && VT == N1.getValueType() &&
9080 "INSERT_VECTOR_ELT vector type mismatch");
9082 "INSERT_VECTOR_ELT scalar fp/int mismatch");
9083 assert((!VT.isFloatingPoint() ||
9084 VT.getVectorElementType() == N2.getValueType()) &&
9085 "INSERT_VECTOR_ELT fp scalar type mismatch");
9086 assert((!VT.isInteger() ||
9088 "INSERT_VECTOR_ELT int scalar size mismatch");
9089
9090 auto *N3C = dyn_cast<ConstantSDNode>(N3);
9091 // INSERT_VECTOR_ELT into out-of-bounds element is an UNDEF, except
9092 // for scalable vectors where we will generate appropriate code to
9093 // deal with out-of-bounds cases correctly.
9094 if (N3C && VT.isFixedLengthVector() &&
9095 N3C->getZExtValue() >= VT.getVectorNumElements())
9096 return getUNDEF(VT);
9097
9098 // Undefined index can be assumed out-of-bounds, so that's UNDEF too.
9099 if (N3.isUndef())
9100 return getUNDEF(VT);
9101
9102 // If inserting poison, just use the input vector.
9103 if (N2.getOpcode() == ISD::POISON)
9104 return N1;
9105
9106 // Inserting undef into undef/poison is still undef.
9107 if (N2.getOpcode() == ISD::UNDEF && N1.isUndef())
9108 return getUNDEF(VT);
9109
9110 // If the inserted element is an UNDEF, just use the input vector.
9111 // But not if skipping the insert could make the result more poisonous.
9112 if (N2.isUndef()) {
9113 if (N3C && VT.isFixedLengthVector()) {
9114 APInt EltMask =
9115 APInt::getOneBitSet(VT.getVectorNumElements(), N3C->getZExtValue());
9116 if (isGuaranteedNotToBePoison(N1, EltMask))
9117 return N1;
9118 } else if (isGuaranteedNotToBePoison(N1))
9119 return N1;
9120 }
9121 break;
9122 }
9123 case ISD::INSERT_SUBVECTOR: {
9124 // If inserting poison, just use the input vector,
9125 if (N2.getOpcode() == ISD::POISON)
9126 return N1;
9127
9128 // Inserting undef into undef/poison is still undef.
9129 if (N2.getOpcode() == ISD::UNDEF && N1.isUndef())
9130 return getUNDEF(VT);
9131
9132 EVT N2VT = N2.getValueType();
9133 assert(VT == N1.getValueType() &&
9134 "Dest and insert subvector source types must match!");
9135 assert(VT.isVector() && N2VT.isVector() &&
9136 "Insert subvector VTs must be vectors!");
9138 "Insert subvector VTs must have the same element type!");
9139 assert((VT.isScalableVector() || N2VT.isFixedLengthVector()) &&
9140 "Cannot insert a scalable vector into a fixed length vector!");
9141 assert((VT.isScalableVector() != N2VT.isScalableVector() ||
9143 "Insert subvector must be from smaller vector to larger vector!");
9145 "Insert subvector index must be constant");
9146 assert((VT.isScalableVector() != N2VT.isScalableVector() ||
9147 (N2VT.getVectorMinNumElements() + N3->getAsZExtVal()) <=
9149 "Insert subvector overflow!");
9151 TLI->getVectorIdxWidth(getDataLayout()) &&
9152 "Constant index for INSERT_SUBVECTOR has an invalid size");
9153
9154 // Trivial insertion.
9155 if (VT == N2VT)
9156 return N2;
9157
9158 // If this is an insert of an extracted vector into an undef/poison vector,
9159 // we can just use the input to the extract. But not if skipping the
9160 // extract+insert could make the result more poisonous.
9161 if (N1.isUndef() && N2.getOpcode() == ISD::EXTRACT_SUBVECTOR &&
9162 N2.getOperand(1) == N3 && N2.getOperand(0).getValueType() == VT) {
9163 if (N1.getOpcode() == ISD::POISON)
9164 return N2.getOperand(0);
9165 if (VT.isFixedLengthVector() && N2VT.isFixedLengthVector()) {
9166 unsigned LoBit = N3->getAsZExtVal();
9167 unsigned HiBit = LoBit + N2VT.getVectorNumElements();
9168 APInt EltMask =
9169 APInt::getBitsSet(VT.getVectorNumElements(), LoBit, HiBit);
9170 if (isGuaranteedNotToBePoison(N2.getOperand(0), ~EltMask))
9171 return N2.getOperand(0);
9172 } else if (isGuaranteedNotToBePoison(N2.getOperand(0)))
9173 return N2.getOperand(0);
9174 }
9175
9176 // If the inserted subvector is UNDEF, just use the input vector.
9177 // But not if skipping the insert could make the result more poisonous.
9178 if (N2.isUndef()) {
9179 if (VT.isFixedLengthVector()) {
9180 unsigned LoBit = N3->getAsZExtVal();
9181 unsigned HiBit = LoBit + N2VT.getVectorNumElements();
9182 APInt EltMask =
9183 APInt::getBitsSet(VT.getVectorNumElements(), LoBit, HiBit);
9184 if (isGuaranteedNotToBePoison(N1, EltMask))
9185 return N1;
9186 } else if (isGuaranteedNotToBePoison(N1))
9187 return N1;
9188 }
9189 break;
9190 }
9191 case ISD::BITCAST:
9192 // Fold bit_convert nodes from a type to themselves.
9193 if (N1.getValueType() == VT)
9194 return N1;
9195 break;
9196 case ISD::VECTOR_COMPRESS: {
9197 [[maybe_unused]] EVT VecVT = N1.getValueType();
9198 [[maybe_unused]] EVT MaskVT = N2.getValueType();
9199 [[maybe_unused]] EVT PassthruVT = N3.getValueType();
9200 assert(VT == VecVT && "Vector and result type don't match.");
9201 assert(VecVT.isVector() && MaskVT.isVector() && PassthruVT.isVector() &&
9202 "All inputs must be vectors.");
9203 assert(VecVT == PassthruVT && "Vector and passthru types don't match.");
9205 "Vector and mask must have same number of elements.");
9206
9207 if (N1.isUndef() || N2.isUndef())
9208 return N3;
9209
9210 break;
9211 }
9216 [[maybe_unused]] EVT AccVT = N1.getValueType();
9217 [[maybe_unused]] EVT Input1VT = N2.getValueType();
9218 [[maybe_unused]] EVT Input2VT = N3.getValueType();
9219 assert(Input1VT.isVector() && Input1VT == Input2VT &&
9220 "Expected the second and third operands of the PARTIAL_REDUCE_MLA "
9221 "node to have the same type!");
9222 assert(VT.isVector() && VT == AccVT &&
9223 "Expected the first operand of the PARTIAL_REDUCE_MLA node to have "
9224 "the same type as its result!");
9226 AccVT.getVectorElementCount()) &&
9227 "Expected the element count of the second and third operands of the "
9228 "PARTIAL_REDUCE_MLA node to be a positive integer multiple of the "
9229 "element count of the first operand and the result!");
9231 "Expected the second and third operands of the PARTIAL_REDUCE_MLA "
9232 "node to have an element type which is the same as or smaller than "
9233 "the element type of the first operand and result!");
9234 break;
9235 }
9236 }
9237
9238 // Perform trivial constant folding for arithmetic operators.
9239 switch (Opcode) {
9243 case ISD::FMA:
9244 case ISD::FMAD:
9245 case ISD::SETCC:
9246 case ISD::FSHL:
9247 case ISD::FSHR:
9248 if (SDValue SV =
9249 FoldConstantArithmetic(Opcode, DL, VT, {N1, N2, N3}, Flags))
9250 return SV;
9251 break;
9252 }
9253
9254 // Memoize node if it doesn't produce a glue result.
9255 SDNode *N;
9256 SDVTList VTs = getVTList(VT);
9257 SDValue Ops[] = {N1, N2, N3};
9258 if (VT != MVT::Glue) {
9259 SDNodeKey ID(Opcode, VTs, Ops);
9260 FoldingSetInsertToken InsertToken;
9261 if (SDNode *E = lookupNode(ID, DL, InsertToken)) {
9262 E->intersectFlagsWith(Flags);
9263 return SDValue(E, 0);
9264 }
9265
9266 N = newSDNode<SDNode>(Opcode, DL.getIROrder(), DL.getDebugLoc(), VTs);
9267 N->setFlags(Flags);
9268 createOperands(N, Ops);
9269 CSEMap.insert(N, InsertToken);
9270 } else {
9271 N = newSDNode<SDNode>(Opcode, DL.getIROrder(), DL.getDebugLoc(), VTs);
9272 createOperands(N, Ops);
9273 }
9274
9275 InsertNode(N);
9276 SDValue V = SDValue(N, 0);
9277 NewSDValueDbgMsg(V, "Creating new node: ", this);
9278 return V;
9279}
9280
9281SDValue SelectionDAG::getNode(unsigned Opcode, const SDLoc &DL, EVT VT,
9282 SDValue N1, SDValue N2, SDValue N3, SDValue N4,
9283 const SDNodeFlags Flags) {
9284 SDValue Ops[] = { N1, N2, N3, N4 };
9285 return getNode(Opcode, DL, VT, Ops, Flags);
9286}
9287
9288SDValue SelectionDAG::getNode(unsigned Opcode, const SDLoc &DL, EVT VT,
9289 SDValue N1, SDValue N2, SDValue N3, SDValue N4) {
9290 SDNodeFlags Flags;
9291 if (Inserter)
9292 Flags = Inserter->getFlags();
9293 return getNode(Opcode, DL, VT, N1, N2, N3, N4, Flags);
9294}
9295
9296SDValue SelectionDAG::getNode(unsigned Opcode, const SDLoc &DL, EVT VT,
9297 SDValue N1, SDValue N2, SDValue N3, SDValue N4,
9298 SDValue N5, const SDNodeFlags Flags) {
9299 SDValue Ops[] = { N1, N2, N3, N4, N5 };
9300 return getNode(Opcode, DL, VT, Ops, Flags);
9301}
9302
9303SDValue SelectionDAG::getNode(unsigned Opcode, const SDLoc &DL, EVT VT,
9304 SDValue N1, SDValue N2, SDValue N3, SDValue N4,
9305 SDValue N5) {
9306 SDNodeFlags Flags;
9307 if (Inserter)
9308 Flags = Inserter->getFlags();
9309 return getNode(Opcode, DL, VT, N1, N2, N3, N4, N5, Flags);
9310}
9311
9312/// getStackArgumentTokenFactor - Compute a TokenFactor to force all
9313/// the incoming stack arguments to be loaded from the stack.
9315 SmallVector<SDValue, 8> ArgChains;
9316
9317 // Include the original chain at the beginning of the list. When this is
9318 // used by target LowerCall hooks, this helps legalize find the
9319 // CALLSEQ_BEGIN node.
9320 ArgChains.push_back(Chain);
9321
9322 // Add a chain value for each stack argument.
9323 for (SDNode *U : getEntryNode().getNode()->users())
9324 if (LoadSDNode *L = dyn_cast<LoadSDNode>(U))
9325 if (FrameIndexSDNode *FI = dyn_cast<FrameIndexSDNode>(L->getBasePtr()))
9326 if (FI->getIndex() < 0)
9327 ArgChains.push_back(SDValue(L, 1));
9328
9329 // Build a tokenfactor for all the chains.
9330 return getNode(ISD::TokenFactor, SDLoc(Chain), MVT::Other, ArgChains);
9331}
9332
9333/// getMemsetValue - Vectorized representation of the memset value
9334/// operand.
9336 const SDLoc &dl) {
9337 assert(!Value.isUndef());
9338
9339 unsigned NumBits = VT.getScalarSizeInBits();
9341 assert(C->getAPIntValue().getBitWidth() == 8);
9342 APInt Val = APInt::getSplat(NumBits, C->getAPIntValue());
9343 if (VT.isInteger()) {
9344 bool IsOpaque = VT.getSizeInBits() > 64 ||
9345 !DAG.getTargetLoweringInfo().isLegalStoreImmediate(C->getSExtValue());
9346 return DAG.getConstant(Val, dl, VT, false, IsOpaque);
9347 }
9348 return DAG.getConstantFP(APFloat(VT.getFltSemantics(), Val), dl, VT);
9349 }
9350
9351 assert(Value.getValueType() == MVT::i8 && "memset with non-byte fill value?");
9352 EVT IntVT = VT.getScalarType();
9353 if (!IntVT.isInteger())
9354 IntVT = EVT::getIntegerVT(*DAG.getContext(), IntVT.getSizeInBits());
9355
9356 Value = DAG.getNode(ISD::ZERO_EXTEND, dl, IntVT, Value);
9357 if (NumBits > 8) {
9358 // Use a multiplication with 0x010101... to extend the input to the
9359 // required length.
9360 APInt Magic = APInt::getSplat(NumBits, APInt(8, 0x01));
9361 Value = DAG.getNode(ISD::MUL, dl, IntVT, Value,
9362 DAG.getConstant(Magic, dl, IntVT));
9363 }
9364
9365 if (VT != Value.getValueType() && !VT.isInteger())
9366 Value = DAG.getBitcast(VT.getScalarType(), Value);
9367 if (VT != Value.getValueType())
9368 Value = DAG.getSplatBuildVector(VT, dl, Value);
9369
9370 return Value;
9371}
9372
9373/// getMemsetStringVal - Similar to getMemsetValue. Except this is only
9374/// used when a memcpy is turned into a memset when the source is a constant
9375/// string ptr.
9377 const TargetLowering &TLI,
9378 const ConstantDataArraySlice &Slice) {
9379 // Handle vector with all elements zero.
9380 if (Slice.Array == nullptr) {
9381 if (VT.isInteger())
9382 return DAG.getConstant(0, dl, VT);
9383 return DAG.getNode(ISD::BITCAST, dl, VT,
9384 DAG.getConstant(0, dl, VT.changeTypeToInteger()));
9385 }
9386
9387 assert(!VT.isVector() && "Can't handle vector type here!");
9388 unsigned NumVTBits = VT.getSizeInBits();
9389 unsigned NumVTBytes = NumVTBits / 8;
9390 unsigned NumBytes = std::min(NumVTBytes, unsigned(Slice.Length));
9391
9392 APInt Val(NumVTBits, 0);
9393 if (DAG.getDataLayout().isLittleEndian()) {
9394 for (unsigned i = 0; i != NumBytes; ++i)
9395 Val |= (uint64_t)(unsigned char)Slice[i] << i*8;
9396 } else {
9397 for (unsigned i = 0; i != NumBytes; ++i)
9398 Val |= (uint64_t)(unsigned char)Slice[i] << (NumVTBytes-i-1)*8;
9399 }
9400
9401 // If the "cost" of materializing the integer immediate is less than the cost
9402 // of a load, then it is cost effective to turn the load into the immediate.
9403 Type *Ty = VT.getTypeForEVT(*DAG.getContext());
9404 if (TLI.shouldConvertConstantLoadToIntImm(Val, Ty))
9405 return DAG.getConstant(Val, dl, VT);
9406 return SDValue();
9407}
9408
9410 const SDLoc &DL,
9411 const SDNodeFlags Flags) {
9412 SDValue Index = getTypeSize(DL, Base.getValueType(), Offset);
9413 return getMemBasePlusOffset(Base, Index, DL, Flags);
9414}
9415
9417 const SDLoc &DL,
9418 const SDNodeFlags Flags) {
9419 assert(Offset.getValueType().isInteger());
9420 EVT BasePtrVT = Ptr.getValueType();
9421 if (TLI->shouldPreservePtrArith(this->getMachineFunction().getFunction(),
9422 BasePtrVT))
9423 return getNode(ISD::PTRADD, DL, BasePtrVT, Ptr, Offset, Flags);
9424 // InBounds only applies to PTRADD, don't set it if we generate ADD.
9425 SDNodeFlags AddFlags = Flags;
9426 AddFlags.setInBounds(false);
9427 return getNode(ISD::ADD, DL, BasePtrVT, Ptr, Offset, AddFlags);
9428}
9429
9430/// Returns true if memcpy source is constant data.
9432 uint64_t SrcDelta = 0;
9433 GlobalAddressSDNode *G = nullptr;
9434 if (Src.getOpcode() == ISD::GlobalAddress)
9436 else if (Src->isAnyAdd() &&
9437 Src.getOperand(0).getOpcode() == ISD::GlobalAddress &&
9438 Src.getOperand(1).getOpcode() == ISD::Constant) {
9439 G = cast<GlobalAddressSDNode>(Src.getOperand(0));
9440 SrcDelta = Src.getConstantOperandVal(1);
9441 }
9442 if (!G)
9443 return false;
9444
9445 return getConstantDataArrayInfo(G->getGlobal(), Slice, 8,
9446 SrcDelta + G->getOffset());
9447}
9448
9450 SelectionDAG &DAG) {
9451 // On Darwin, -Os means optimize for size without hurting performance, so
9452 // only really optimize for size when -Oz (MinSize) is used.
9454 return MF.getFunction().hasMinSize();
9455 return DAG.shouldOptForSize();
9456}
9457
9459 SmallVector<SDValue, 32> &OutChains, unsigned From,
9460 unsigned To, SmallVector<SDValue, 16> &OutLoadChains,
9461 SmallVector<SDValue, 16> &OutStoreChains) {
9462 assert(OutLoadChains.size() && "Missing loads in memcpy inlining");
9463 assert(OutStoreChains.size() && "Missing stores in memcpy inlining");
9464 SmallVector<SDValue, 16> GluedLoadChains;
9465 for (unsigned i = From; i < To; ++i) {
9466 OutChains.push_back(OutLoadChains[i]);
9467 GluedLoadChains.push_back(OutLoadChains[i]);
9468 }
9469
9470 // Chain for all loads.
9471 SDValue LoadToken = DAG.getNode(ISD::TokenFactor, dl, MVT::Other,
9472 GluedLoadChains);
9473
9474 for (unsigned i = From; i < To; ++i) {
9475 StoreSDNode *ST = dyn_cast<StoreSDNode>(OutStoreChains[i]);
9476 SDValue NewStore = DAG.getTruncStore(LoadToken, dl, ST->getValue(),
9477 ST->getBasePtr(), ST->getMemoryVT(),
9478 ST->getMemOperand());
9479 OutChains.push_back(NewStore);
9480 }
9481}
9482
9483static SDValue
9485 SDValue Dst, SDValue Src, uint64_t Size, Align DstAlign,
9486 Align SrcAlign, bool isVol, bool AlwaysInline,
9487 MachinePointerInfo DstPtrInfo,
9488 MachinePointerInfo SrcPtrInfo, const AAMDNodes &AAInfo,
9489 BatchAAResults *BatchAA, const MDNode *DstMemCacheHint,
9490 const MDNode *SrcMemCacheHint) {
9491 // Turn a memcpy of undef to nop.
9492 // FIXME: We need to honor volatile even is Src is undef.
9493 if (Src.isUndef())
9494 return Chain;
9495
9496 // Expand memcpy to a series of load and store ops if the size operand falls
9497 // below a certain threshold.
9498 // TODO: In the AlwaysInline case, if the size is big then generate a loop
9499 // rather than maybe a humongous number of loads and stores.
9500 const TargetLowering &TLI = DAG.getTargetLoweringInfo();
9501 const DataLayout &DL = DAG.getDataLayout();
9502 LLVMContext &C = *DAG.getContext();
9503 std::vector<EVT> MemOps;
9504 bool DstAlignCanChange = false;
9506 MachineFrameInfo &MFI = MF.getFrameInfo();
9507 bool OptSize = shouldLowerMemFuncForSize(MF, DAG);
9509 if (FI && !MFI.isFixedObjectIndex(FI->getIndex()))
9510 DstAlignCanChange = true;
9511 SrcAlign = std::max(SrcAlign, DAG.InferPtrAlign(Src).valueOrOne());
9513 // If marked as volatile, perform a copy even when marked as constant.
9514 bool CopyFromConstant = !isVol && isMemSrcFromConstant(Src, Slice);
9515 bool isZeroConstant = CopyFromConstant && Slice.Array == nullptr;
9516 unsigned Limit = AlwaysInline ? ~0U : TLI.getMaxStoresPerMemcpy(OptSize);
9517 const MemOp Op = isZeroConstant
9518 ? MemOp::Set(Size, DstAlignCanChange, DstAlign,
9519 /*IsZeroMemset*/ true, isVol)
9520 : MemOp::Copy(Size, DstAlignCanChange, DstAlign,
9521 SrcAlign, isVol, CopyFromConstant);
9522 if (!TLI.findOptimalMemOpLowering(
9523 C, MemOps, Limit, Op, DstPtrInfo.getAddrSpace(),
9524 SrcPtrInfo.getAddrSpace(), MF.getFunction().getAttributes(), nullptr))
9525 return SDValue();
9526
9527 if (DstAlignCanChange) {
9528 Type *Ty = MemOps[0].getTypeForEVT(C);
9529 Align NewDstAlign = DL.getABITypeAlign(Ty);
9530
9531 // Don't promote to an alignment that would require dynamic stack
9532 // realignment which may conflict with optimizations such as tail call
9533 // optimization.
9535 if (!TRI->hasStackRealignment(MF))
9536 if (MaybeAlign StackAlign = DL.getStackAlignment())
9537 NewDstAlign = std::min(NewDstAlign, *StackAlign);
9538
9539 if (NewDstAlign > DstAlign) {
9540 // Give the stack frame object a larger alignment if needed.
9541 if (MFI.getObjectAlign(FI->getIndex()) < NewDstAlign)
9542 MFI.setObjectAlignment(FI->getIndex(), NewDstAlign);
9543 DstAlign = NewDstAlign;
9544 }
9545 }
9546
9547 // Prepare AAInfo for loads/stores after lowering this memcpy.
9548 AAMDNodes NewAAInfo = AAInfo;
9549 NewAAInfo.TBAA = NewAAInfo.TBAAStruct = nullptr;
9550
9551 const Value *SrcVal = dyn_cast_if_present<const Value *>(SrcPtrInfo.V);
9552 bool isConstant =
9553 BatchAA && SrcVal &&
9554 BatchAA->pointsToConstantMemory(MemoryLocation(SrcVal, Size, AAInfo));
9555
9556 MachineMemOperand::Flags MMOFlags =
9558 SmallVector<SDValue, 16> OutLoadChains;
9559 SmallVector<SDValue, 16> OutStoreChains;
9560 SmallVector<SDValue, 32> OutChains;
9561 unsigned NumMemOps = MemOps.size();
9562 uint64_t SrcOff = 0, DstOff = 0;
9563 for (unsigned i = 0; i != NumMemOps; ++i) {
9564 EVT VT = MemOps[i];
9565 unsigned VTSize = VT.getSizeInBits() / 8;
9567
9568 if (VTSize > Size) {
9569 // Issuing an unaligned load / store pair that overlaps with the previous
9570 // pair. Adjust the offset accordingly.
9571 assert(i == NumMemOps-1 && i != 0);
9572 SrcOff -= VTSize - Size;
9573 DstOff -= VTSize - Size;
9574 }
9575
9576 if (CopyFromConstant &&
9577 (isZeroConstant || (VT.isInteger() && !VT.isVector()))) {
9578 // It's unlikely a store of a vector immediate can be done in a single
9579 // instruction. It would require a load from a constantpool first.
9580 // We only handle zero vectors here.
9581 // FIXME: Handle other cases where store of vector immediate is done in
9582 // a single instruction.
9583 ConstantDataArraySlice SubSlice;
9584 if (SrcOff < Slice.Length) {
9585 SubSlice = Slice;
9586 SubSlice.move(SrcOff);
9587 } else {
9588 // This is an out-of-bounds access and hence UB. Pretend we read zero.
9589 SubSlice.Array = nullptr;
9590 SubSlice.Offset = 0;
9591 SubSlice.Length = VTSize;
9592 }
9593 Value = getMemsetStringVal(VT, dl, DAG, TLI, SubSlice);
9594 if (Value.getNode()) {
9595 Store = DAG.getStore(
9596 Chain, dl, Value,
9597 DAG.getObjectPtrOffset(dl, Dst, TypeSize::getFixed(DstOff)),
9598 DstPtrInfo.getWithOffset(DstOff), DstAlign, MMOFlags,
9599 MMOMetadata(NewAAInfo, /*Ranges=*/nullptr, DstMemCacheHint));
9600 OutChains.push_back(Store);
9601 }
9602 }
9603
9604 if (!Store.getNode()) {
9605 // The type might not be legal for the target. This should only happen
9606 // if the type is smaller than a legal type, as on PPC, so the right
9607 // thing to do is generate a LoadExt/StoreTrunc pair. These simplify
9608 // to Load/Store if NVT==VT.
9609 // FIXME does the case above also need this?
9610 EVT NVT = TLI.getTypeToTransformTo(C, VT);
9611 assert(NVT.bitsGE(VT));
9612
9613 bool isDereferenceable =
9614 SrcPtrInfo.getWithOffset(SrcOff).isDereferenceable(VTSize, C, DL);
9615 MachineMemOperand::Flags SrcMMOFlags = MMOFlags;
9616 if (isDereferenceable)
9618 if (isConstant)
9619 SrcMMOFlags |= MachineMemOperand::MOInvariant;
9620
9621 Value = DAG.getExtLoad(
9622 ISD::EXTLOAD, dl, NVT, Chain,
9623 DAG.getObjectPtrOffset(dl, Src, TypeSize::getFixed(SrcOff)),
9624 SrcPtrInfo.getWithOffset(SrcOff), VT,
9625 commonAlignment(SrcAlign, SrcOff), SrcMMOFlags,
9626 MMOMetadata(NewAAInfo, /*Ranges=*/nullptr, SrcMemCacheHint));
9627 OutLoadChains.push_back(Value.getValue(1));
9628
9629 Store = DAG.getTruncStore(
9630 Chain, dl, Value,
9631 DAG.getObjectPtrOffset(dl, Dst, TypeSize::getFixed(DstOff)),
9632 DstPtrInfo.getWithOffset(DstOff), VT, DstAlign, MMOFlags,
9633 MMOMetadata(NewAAInfo, /*Ranges=*/nullptr, DstMemCacheHint));
9634 OutStoreChains.push_back(Store);
9635 }
9636 SrcOff += VTSize;
9637 DstOff += VTSize;
9638 Size -= VTSize;
9639 }
9640
9641 unsigned GluedLdStLimit = MaxLdStGlue == 0 ?
9643 unsigned NumLdStInMemcpy = OutStoreChains.size();
9644
9645 if (NumLdStInMemcpy) {
9646 // It may be that memcpy might be converted to memset if it's memcpy
9647 // of constants. In such a case, we won't have loads and stores, but
9648 // just stores. In the absence of loads, there is nothing to gang up.
9649 if ((GluedLdStLimit <= 1) || !EnableMemCpyDAGOpt) {
9650 // If target does not care, just leave as it.
9651 for (unsigned i = 0; i < NumLdStInMemcpy; ++i) {
9652 OutChains.push_back(OutLoadChains[i]);
9653 OutChains.push_back(OutStoreChains[i]);
9654 }
9655 } else {
9656 // Ld/St less than/equal limit set by target.
9657 if (NumLdStInMemcpy <= GluedLdStLimit) {
9658 chainLoadsAndStoresForMemcpy(DAG, dl, OutChains, 0,
9659 NumLdStInMemcpy, OutLoadChains,
9660 OutStoreChains);
9661 } else {
9662 unsigned NumberLdChain = NumLdStInMemcpy / GluedLdStLimit;
9663 unsigned RemainingLdStInMemcpy = NumLdStInMemcpy % GluedLdStLimit;
9664 unsigned GlueIter = 0;
9665
9666 // Residual ld/st.
9667 if (RemainingLdStInMemcpy) {
9669 DAG, dl, OutChains, NumLdStInMemcpy - RemainingLdStInMemcpy,
9670 NumLdStInMemcpy, OutLoadChains, OutStoreChains);
9671 }
9672
9673 for (unsigned cnt = 0; cnt < NumberLdChain; ++cnt) {
9674 unsigned IndexFrom = NumLdStInMemcpy - RemainingLdStInMemcpy -
9675 GlueIter - GluedLdStLimit;
9676 unsigned IndexTo = NumLdStInMemcpy - RemainingLdStInMemcpy - GlueIter;
9677 chainLoadsAndStoresForMemcpy(DAG, dl, OutChains, IndexFrom, IndexTo,
9678 OutLoadChains, OutStoreChains);
9679 GlueIter += GluedLdStLimit;
9680 }
9681 }
9682 }
9683 }
9684 return DAG.getTokenFactor(dl, OutChains);
9685}
9686
9688 SelectionDAG &DAG, const SDLoc &dl, SDValue Chain, SDValue Dst, SDValue Src,
9689 uint64_t Size, Align DstAlign, Align SrcAlign, bool isVol,
9690 bool AlwaysInline, MachinePointerInfo DstPtrInfo,
9691 MachinePointerInfo SrcPtrInfo, const AAMDNodes &AAInfo) {
9692 // Turn a memmove of undef to nop.
9693 // FIXME: We need to honor volatile even is Src is undef.
9694 if (Src.isUndef())
9695 return Chain;
9696
9697 // Expand memmove to a series of load and store ops if the size operand falls
9698 // below a certain threshold.
9699 const TargetLowering &TLI = DAG.getTargetLoweringInfo();
9700 const DataLayout &DL = DAG.getDataLayout();
9701 LLVMContext &C = *DAG.getContext();
9702 std::vector<EVT> MemOps;
9703 bool DstAlignCanChange = false;
9705 MachineFrameInfo &MFI = MF.getFrameInfo();
9706 bool OptSize = shouldLowerMemFuncForSize(MF, DAG);
9708 if (FI && !MFI.isFixedObjectIndex(FI->getIndex()))
9709 DstAlignCanChange = true;
9710 SrcAlign = std::max(SrcAlign, DAG.InferPtrAlign(Src).valueOrOne());
9711 unsigned Limit = AlwaysInline ? ~0U : TLI.getMaxStoresPerMemmove(OptSize);
9712 if (!TLI.findOptimalMemOpLowering(
9713 C, MemOps, Limit,
9714 MemOp::Move(Size, DstAlignCanChange, DstAlign, SrcAlign, isVol),
9715 DstPtrInfo.getAddrSpace(), SrcPtrInfo.getAddrSpace(),
9716 MF.getFunction().getAttributes(), nullptr))
9717 return SDValue();
9718
9719 if (DstAlignCanChange) {
9720 Type *Ty = MemOps[0].getTypeForEVT(C);
9721 Align NewDstAlign = DL.getABITypeAlign(Ty);
9722
9723 // Don't promote to an alignment that would require dynamic stack
9724 // realignment which may conflict with optimizations such as tail call
9725 // optimization.
9727 if (!TRI->hasStackRealignment(MF))
9728 if (MaybeAlign StackAlign = DL.getStackAlignment())
9729 NewDstAlign = std::min(NewDstAlign, *StackAlign);
9730
9731 if (NewDstAlign > DstAlign) {
9732 // Give the stack frame object a larger alignment if needed.
9733 if (MFI.getObjectAlign(FI->getIndex()) < NewDstAlign)
9734 MFI.setObjectAlignment(FI->getIndex(), NewDstAlign);
9735 DstAlign = NewDstAlign;
9736 }
9737 }
9738
9739 // Prepare AAInfo for loads/stores after lowering this memmove.
9740 AAMDNodes NewAAInfo = AAInfo;
9741 NewAAInfo.TBAA = NewAAInfo.TBAAStruct = nullptr;
9742
9743 MachineMemOperand::Flags MMOFlags =
9745 uint64_t SrcOff = 0;
9746 SmallVector<SDValue, 8> LoadValues;
9747 SmallVector<SDValue, 8> LoadChains;
9748 SmallVector<SDValue, 8> OutChains;
9749 unsigned NumMemOps = MemOps.size();
9750 for (unsigned i = 0; i < NumMemOps; i++) {
9751 EVT VT = MemOps[i];
9752 unsigned VTSize = VT.getSizeInBits() / 8;
9753 SDValue Value;
9754 bool IsOverlapping = false;
9755
9756 if (i == NumMemOps - 1 && i != 0 && VTSize > Size - SrcOff) {
9757 // Issuing an unaligned load / store pair that overlaps with the previous
9758 // pair. Adjust the offset accordingly.
9759 SrcOff = Size - VTSize;
9760 IsOverlapping = true;
9761 }
9762
9763 // Calculate the actual alignment at the current offset. The alignment at
9764 // SrcOff may be lower than the base alignment, especially when using
9765 // overlapping loads.
9766 Align SrcAlignAtOffset = commonAlignment(SrcAlign, SrcOff);
9767 if (IsOverlapping) {
9768 // Verify that the target allows misaligned memory accesses at the
9769 // adjusted offset when using overlapping loads.
9770 unsigned Fast;
9771 if (!TLI.allowsMisalignedMemoryAccesses(VT, SrcPtrInfo.getAddrSpace(),
9772 SrcAlignAtOffset, MMOFlags,
9773 &Fast) ||
9774 !Fast) {
9775 // This should have been caught by findOptimalMemOpLowering, but verify
9776 // here for safety.
9777 return SDValue();
9778 }
9779 }
9780
9781 bool isDereferenceable =
9782 SrcPtrInfo.getWithOffset(SrcOff).isDereferenceable(VTSize, C, DL);
9783 MachineMemOperand::Flags SrcMMOFlags = MMOFlags;
9784 if (isDereferenceable)
9786 Value =
9787 DAG.getLoad(VT, dl, Chain,
9788 DAG.getObjectPtrOffset(dl, Src, TypeSize::getFixed(SrcOff)),
9789 SrcPtrInfo.getWithOffset(SrcOff), SrcAlignAtOffset,
9790 SrcMMOFlags, NewAAInfo);
9791 LoadValues.push_back(Value);
9792 LoadChains.push_back(Value.getValue(1));
9793 SrcOff += VTSize;
9794 }
9795 Chain = DAG.getTokenFactor(dl, LoadChains);
9796 OutChains.clear();
9797 uint64_t DstOff = 0;
9798 for (unsigned i = 0; i < NumMemOps; i++) {
9799 EVT VT = MemOps[i];
9800 unsigned VTSize = VT.getSizeInBits() / 8;
9801 SDValue Store;
9802 bool IsOverlapping = false;
9803
9804 if (i == NumMemOps - 1 && i != 0 && VTSize > Size - DstOff) {
9805 // Issuing an unaligned load / store pair that overlaps with the previous
9806 // pair. Adjust the offset accordingly.
9807 DstOff = Size - VTSize;
9808 IsOverlapping = true;
9809 }
9810
9811 // Calculate the actual alignment at the current offset. The alignment at
9812 // DstOff may be lower than the base alignment, especially when using
9813 // overlapping stores.
9814 Align DstAlignAtOffset = commonAlignment(DstAlign, DstOff);
9815 if (IsOverlapping) {
9816 // Verify that the target allows misaligned memory accesses at the
9817 // adjusted offset when using overlapping stores.
9818 unsigned Fast;
9819 if (!TLI.allowsMisalignedMemoryAccesses(VT, DstPtrInfo.getAddrSpace(),
9820 DstAlignAtOffset, MMOFlags,
9821 &Fast) ||
9822 !Fast) {
9823 // This should have been caught by findOptimalMemOpLowering, but verify
9824 // here for safety.
9825 return SDValue();
9826 }
9827 }
9828 Store = DAG.getStore(
9829 Chain, dl, LoadValues[i],
9830 DAG.getObjectPtrOffset(dl, Dst, TypeSize::getFixed(DstOff)),
9831 DstPtrInfo.getWithOffset(DstOff), DstAlignAtOffset, MMOFlags,
9832 NewAAInfo);
9833 OutChains.push_back(Store);
9834 DstOff += VTSize;
9835 }
9836
9837 return DAG.getTokenFactor(dl, OutChains);
9838}
9839
9840/// Lower the call to 'memset' intrinsic function into a series of store
9841/// operations.
9842///
9843/// \param DAG Selection DAG where lowered code is placed.
9844/// \param dl Link to corresponding IR location.
9845/// \param Chain Control flow dependency.
9846/// \param Dst Pointer to destination memory location.
9847/// \param Src Value of byte to write into the memory.
9848/// \param Size Number of bytes to write.
9849/// \param Alignment Alignment of the destination in bytes.
9850/// \param isVol True if destination is volatile.
9851/// \param AlwaysInline Makes sure no function call is generated.
9852/// \param DstPtrInfo IR information on the memory pointer.
9853/// \returns New head in the control flow, if lowering was successful, empty
9854/// SDValue otherwise.
9855///
9856/// The function tries to replace 'llvm.memset' intrinsic with several store
9857/// operations and value calculation code. This is usually profitable for small
9858/// memory size or when the semantic requires inlining.
9860 SDValue Chain, SDValue Dst, SDValue Src,
9861 uint64_t Size, Align Alignment, bool isVol,
9862 bool AlwaysInline, MachinePointerInfo DstPtrInfo,
9863 const AAMDNodes &AAInfo) {
9864 // Turn a memset of undef to nop.
9865 // FIXME: We need to honor volatile even is Src is undef.
9866 if (Src.isUndef())
9867 return Chain;
9868
9869 // Expand memset to a series of load/store ops if the size operand
9870 // falls below a certain threshold.
9871 const TargetLowering &TLI = DAG.getTargetLoweringInfo();
9872 std::vector<EVT> MemOps;
9873 bool DstAlignCanChange = false;
9874 LLVMContext &C = *DAG.getContext();
9876 MachineFrameInfo &MFI = MF.getFrameInfo();
9877 bool OptSize = shouldLowerMemFuncForSize(MF, DAG);
9879 if (FI && !MFI.isFixedObjectIndex(FI->getIndex()))
9880 DstAlignCanChange = true;
9881 bool IsZeroVal = isNullConstant(Src);
9882 unsigned Limit = AlwaysInline ? ~0 : TLI.getMaxStoresPerMemset(OptSize);
9883
9884 EVT LargestVT;
9885 if (!TLI.findOptimalMemOpLowering(
9886 C, MemOps, Limit,
9887 MemOp::Set(Size, DstAlignCanChange, Alignment, IsZeroVal, isVol),
9888 DstPtrInfo.getAddrSpace(), ~0u, MF.getFunction().getAttributes(),
9889 &LargestVT))
9890 return SDValue();
9891
9892 if (DstAlignCanChange) {
9893 Type *Ty = MemOps[0].getTypeForEVT(*DAG.getContext());
9894 const DataLayout &DL = DAG.getDataLayout();
9895 Align NewAlign = DL.getABITypeAlign(Ty);
9896
9897 // Don't promote to an alignment that would require dynamic stack
9898 // realignment which may conflict with optimizations such as tail call
9899 // optimization.
9901 if (!TRI->hasStackRealignment(MF))
9902 if (MaybeAlign StackAlign = DL.getStackAlignment())
9903 NewAlign = std::min(NewAlign, *StackAlign);
9904
9905 if (NewAlign > Alignment) {
9906 // Give the stack frame object a larger alignment if needed.
9907 if (MFI.getObjectAlign(FI->getIndex()) < NewAlign)
9908 MFI.setObjectAlignment(FI->getIndex(), NewAlign);
9909 Alignment = NewAlign;
9910 }
9911 }
9912
9913 SmallVector<SDValue, 8> OutChains;
9914 uint64_t DstOff = 0;
9915 unsigned NumMemOps = MemOps.size();
9916
9917 // Find the largest store and generate the bit pattern for it.
9918 // If target didn't set LargestVT, compute it from MemOps.
9919 if (!LargestVT.isSimple()) {
9920 LargestVT = MemOps[0];
9921 for (unsigned i = 1; i < NumMemOps; i++)
9922 if (MemOps[i].bitsGT(LargestVT))
9923 LargestVT = MemOps[i];
9924 }
9925 SDValue MemSetValue = getMemsetValue(Src, LargestVT, DAG, dl);
9926
9927 // Prepare AAInfo for loads/stores after lowering this memset.
9928 AAMDNodes NewAAInfo = AAInfo;
9929 NewAAInfo.TBAA = NewAAInfo.TBAAStruct = nullptr;
9930
9931 for (unsigned i = 0; i < NumMemOps; i++) {
9932 EVT VT = MemOps[i];
9933 unsigned VTSize = VT.getSizeInBits() / 8;
9934 // The target should specify store types that exactly cover the memset size
9935 // (with the last store potentially being oversized for overlapping stores).
9936 assert(Size > 0 && "Target specified more stores than needed in "
9937 "findOptimalMemOpLowering");
9938 if (VTSize > Size) {
9939 // Issuing an unaligned load / store pair that overlaps with the previous
9940 // pair. Adjust the offset accordingly.
9941 assert(i == NumMemOps-1 && i != 0);
9942 DstOff -= VTSize - Size;
9943 }
9944
9945 // If this store is smaller than the largest store see whether we can get
9946 // the smaller value for free with a truncate or extract vector element and
9947 // then store.
9948 SDValue Value = MemSetValue;
9949 if (VT.bitsLT(LargestVT)) {
9950 unsigned Index;
9951 unsigned NElts = LargestVT.getSizeInBits() / VT.getSizeInBits();
9952 EVT SVT = EVT::getVectorVT(*DAG.getContext(), VT.getScalarType(), NElts);
9953 if (!LargestVT.isVector() && !VT.isVector() &&
9954 TLI.isTruncateFree(LargestVT, VT))
9955 Value = DAG.getNode(ISD::TRUNCATE, dl, VT, MemSetValue);
9956 else if (LargestVT.isVector() && !VT.isVector() &&
9958 LargestVT.getTypeForEVT(*DAG.getContext()),
9959 VT.getSizeInBits(), Index) &&
9960 TLI.isTypeLegal(SVT) &&
9961 LargestVT.getSizeInBits() == SVT.getSizeInBits()) {
9962 // Target which can combine store(extractelement VectorTy, Idx) can get
9963 // the smaller value for free.
9964 SDValue TailValue = DAG.getNode(ISD::BITCAST, dl, SVT, MemSetValue);
9965 Value = DAG.getExtractVectorElt(dl, VT, TailValue, Index);
9966 } else
9967 Value = getMemsetValue(Src, VT, DAG, dl);
9968 }
9969 assert(Value.getValueType() == VT && "Value with wrong type.");
9970 SDValue Store = DAG.getStore(
9971 Chain, dl, Value,
9972 DAG.getObjectPtrOffset(dl, Dst, TypeSize::getFixed(DstOff)),
9973 DstPtrInfo.getWithOffset(DstOff), Alignment,
9975 NewAAInfo);
9976 OutChains.push_back(Store);
9977 DstOff += VT.getSizeInBits() / 8;
9978 // For oversized overlapping stores, only subtract the remaining bytes.
9979 // For normal stores, subtract the full store size.
9980 if (VTSize > Size) {
9981 Size = 0;
9982 } else {
9983 Size -= VTSize;
9984 }
9985 }
9986
9987 // After processing all stores, Size should be exactly 0. Any remaining bytes
9988 // indicate a bug in the target's findOptimalMemOpLowering implementation.
9989 assert(Size == 0 && "Target's findOptimalMemOpLowering did not specify "
9990 "stores that exactly cover the memset size");
9991
9992 return DAG.getTokenFactor(dl, OutChains);
9993}
9994
9996 unsigned AS) {
9997 // Lowering memcpy / memset / memmove intrinsics to calls is only valid if all
9998 // pointer operands can be losslessly bitcasted to pointers of address space 0
9999 if (AS != 0 && !TLI->getTargetMachine().isNoopAddrSpaceCast(AS, 0)) {
10000 report_fatal_error("cannot lower memory intrinsic in address space " +
10001 Twine(AS));
10002 }
10003}
10004
10006 const SelectionDAG *SelDAG,
10007 bool AllowReturnsFirstArg) {
10008 if (!CI || !CI->isTailCall())
10009 return false;
10010 // TODO: Fix "returns-first-arg" determination so it doesn't depend on which
10011 // helper symbol we lower to.
10012 return isInTailCallPosition(*CI, SelDAG->getTarget(),
10013 AllowReturnsFirstArg &&
10015}
10016
10017static std::pair<SDValue, SDValue>
10020 const CallInst *CI, RTLIB::Libcall Call,
10021 SelectionDAG *DAG, const TargetLowering *TLI) {
10022 RTLIB::LibcallImpl LCImpl = DAG->getLibcalls().getLibcallImpl(Call);
10023
10024 if (LCImpl == RTLIB::Unsupported)
10025 return {};
10026
10028 bool IsTailCall =
10029 isInTailCallPositionWrapper(CI, DAG, /*AllowReturnsFirstArg=*/true) &&
10030 // Lowering doesn't support tail calling inside a function with
10031 // a swifterror argument yet.
10032 !DAG->hasSwiftErrorArg();
10033 SDValue Callee =
10034 DAG->getExternalSymbol(LCImpl, TLI->getPointerTy(DAG->getDataLayout()));
10035
10036 CLI.setDebugLoc(dl)
10037 .setChain(Chain)
10039 CI->getType(), Callee, std::move(Args))
10040 .setTailCall(IsTailCall);
10041
10042 return TLI->LowerCallTo(CLI);
10043}
10044
10045std::pair<SDValue, SDValue> SelectionDAG::getStrcmp(SDValue Chain,
10046 const SDLoc &dl, SDValue S1,
10047 SDValue S2,
10048 const CallInst *CI) {
10050 TargetLowering::ArgListTy Args = {{S1, PT}, {S2, PT}};
10051 return getRuntimeCallSDValueHelper(Chain, dl, std::move(Args), CI,
10052 RTLIB::STRCMP, this, TLI);
10053}
10054
10055std::pair<SDValue, SDValue> SelectionDAG::getStrstr(SDValue Chain,
10056 const SDLoc &dl, SDValue S1,
10057 SDValue S2,
10058 const CallInst *CI) {
10060 TargetLowering::ArgListTy Args = {{S1, PT}, {S2, PT}};
10061 return getRuntimeCallSDValueHelper(Chain, dl, std::move(Args), CI,
10062 RTLIB::STRSTR, this, TLI);
10063}
10064
10065std::pair<SDValue, SDValue> SelectionDAG::getMemccpy(SDValue Chain,
10066 const SDLoc &dl,
10067 SDValue Dst, SDValue Src,
10069 const CallInst *CI) {
10071
10073 {Dst, PT},
10074 {Src, PT},
10077 return getRuntimeCallSDValueHelper(Chain, dl, std::move(Args), CI,
10078 RTLIB::MEMCCPY, this, TLI);
10079}
10080
10081std::pair<SDValue, SDValue>
10083 SDValue Mem1, SDValue Size, const CallInst *CI) {
10086 {Mem0, PT},
10087 {Mem1, PT},
10089 return getRuntimeCallSDValueHelper(Chain, dl, std::move(Args), CI,
10090 RTLIB::MEMCMP, this, TLI);
10091}
10092
10093std::pair<SDValue, SDValue> SelectionDAG::getStrcpy(SDValue Chain,
10094 const SDLoc &dl,
10095 SDValue Dst, SDValue Src,
10096 const CallInst *CI) {
10098 TargetLowering::ArgListTy Args = {{Dst, PT}, {Src, PT}};
10099 return getRuntimeCallSDValueHelper(Chain, dl, std::move(Args), CI,
10100 RTLIB::STRCPY, this, TLI);
10101}
10102
10103std::pair<SDValue, SDValue> SelectionDAG::getStrlen(SDValue Chain,
10104 const SDLoc &dl,
10105 SDValue Src,
10106 const CallInst *CI) {
10107 // Emit a library call.
10110 return getRuntimeCallSDValueHelper(Chain, dl, std::move(Args), CI,
10111 RTLIB::STRLEN, this, TLI);
10112}
10113
10115 return TLI->supportSwiftError() &&
10116 MF->getFunction().getAttributes().hasAttrSomewhere(
10117 Attribute::SwiftError);
10118}
10119
10121 SDValue Chain, const SDLoc &dl, SDValue Dst, SDValue Src, SDValue Size,
10122 Align DstAlign, Align SrcAlign, bool isVol, bool AlwaysInline,
10123 const CallInst *CI, std::optional<bool> OverrideTailCall,
10124 MachinePointerInfo DstPtrInfo, MachinePointerInfo SrcPtrInfo,
10125 const AAMDNodes &AAInfo, BatchAAResults *BatchAA) {
10126 // Check to see if we should lower the memcpy to loads and stores first.
10127 // For cases within the target-specified limits, this is the best choice.
10128 const MDNode *DstMemCacheHint =
10129 CI ? getMemCacheHintMetadata(*CI, /*OperandNo=*/0) : nullptr;
10130 const MDNode *SrcMemCacheHint =
10131 CI ? getMemCacheHintMetadata(*CI, /*OperandNo=*/1) : nullptr;
10132
10134 if (ConstantSize) {
10135 // Memcpy with size zero? Just return the original chain.
10136 if (ConstantSize->isZero())
10137 return Chain;
10138
10140 *this, dl, Chain, Dst, Src, ConstantSize->getZExtValue(), DstAlign,
10141 SrcAlign, isVol, false, DstPtrInfo, SrcPtrInfo, AAInfo, BatchAA,
10142 DstMemCacheHint, SrcMemCacheHint);
10143 if (Result.getNode())
10144 return Result;
10145 }
10146
10147 // Then check to see if we should lower the memcpy with target-specific
10148 // code. If the target chooses to do this, this is the next best.
10149 if (TSI) {
10150 SDValue Result = TSI->EmitTargetCodeForMemcpy(
10151 *this, dl, Chain, Dst, Src, Size, DstAlign, SrcAlign, isVol,
10152 AlwaysInline, DstPtrInfo, SrcPtrInfo);
10153 if (Result.getNode())
10154 return Result;
10155 }
10156
10157 // If we really need inline code and the target declined to provide it,
10158 // use a (potentially long) sequence of loads and stores.
10159 if (AlwaysInline) {
10160 assert(ConstantSize && "AlwaysInline requires a constant size!");
10162 *this, dl, Chain, Dst, Src, ConstantSize->getZExtValue(), DstAlign,
10163 SrcAlign, isVol, true, DstPtrInfo, SrcPtrInfo, AAInfo, BatchAA,
10164 DstMemCacheHint, SrcMemCacheHint);
10165 }
10166
10169
10170 // FIXME: If the memcpy is volatile (isVol), lowering it to a plain libc
10171 // memcpy is not guaranteed to be safe. libc memcpys aren't required to
10172 // respect volatile, so they may do things like read or write memory
10173 // beyond the given memory regions. But fixing this isn't easy, and most
10174 // people don't care.
10175
10176 // Emit a library call.
10179 Args.emplace_back(Dst, PtrTy);
10180 Args.emplace_back(Src, PtrTy);
10181 Args.emplace_back(Size, getDataLayout().getIntPtrType(*getContext()));
10182 // FIXME: pass in SDLoc
10184 bool IsTailCall = false;
10185 RTLIB::LibcallImpl MemCpyImpl = TLI->getMemcpyImpl();
10186
10187 if (OverrideTailCall.has_value()) {
10188 IsTailCall = *OverrideTailCall;
10189 } else {
10190 bool LowersToMemcpy = MemCpyImpl == RTLIB::impl_memcpy;
10191 IsTailCall = isInTailCallPositionWrapper(CI, this, LowersToMemcpy);
10192 }
10193 // Lowering doesn't support tail calling inside a function with a
10194 // swifterror argument yet.
10195 IsTailCall &= !hasSwiftErrorArg();
10196
10197 CLI.setDebugLoc(dl)
10198 .setChain(Chain)
10199 .setLibCallee(
10200 Libcalls->getLibcallImplCallingConv(MemCpyImpl),
10201 Dst.getValueType().getTypeForEVT(*getContext()),
10202 getExternalSymbol(MemCpyImpl, TLI->getPointerTy(getDataLayout())),
10203 std::move(Args))
10205 .setTailCall(IsTailCall);
10206
10207 std::pair<SDValue,SDValue> CallResult = TLI->LowerCallTo(CLI);
10208 return CallResult.second;
10209}
10210
10212 SDValue Dst, SDValue Src, SDValue Size,
10213 Type *SizeTy, unsigned ElemSz,
10214 bool isTailCall,
10215 MachinePointerInfo DstPtrInfo,
10216 MachinePointerInfo SrcPtrInfo) {
10217 // Lowering doesn't support tail calling inside a function with a
10218 // swifterror argument yet.
10219 isTailCall &= !hasSwiftErrorArg();
10220
10221 // Emit a library call.
10224 Args.emplace_back(Dst, ArgTy);
10225 Args.emplace_back(Src, ArgTy);
10226 Args.emplace_back(Size, SizeTy);
10227
10228 RTLIB::Libcall LibraryCall =
10230 RTLIB::LibcallImpl LibcallImpl = Libcalls->getLibcallImpl(LibraryCall);
10231 if (LibcallImpl == RTLIB::Unsupported)
10232 report_fatal_error("Unsupported element size");
10233
10235 CLI.setDebugLoc(dl)
10236 .setChain(Chain)
10237 .setLibCallee(
10238 Libcalls->getLibcallImplCallingConv(LibcallImpl),
10240 getExternalSymbol(LibcallImpl, TLI->getPointerTy(getDataLayout())),
10241 std::move(Args))
10243 .setTailCall(isTailCall);
10244
10245 std::pair<SDValue, SDValue> CallResult = TLI->LowerCallTo(CLI);
10246 return CallResult.second;
10247}
10248
10250 SDValue Src, SDValue Size, Align DstAlign,
10251 Align SrcAlign, bool isVol, const CallInst *CI,
10252 std::optional<bool> OverrideTailCall,
10253 MachinePointerInfo DstPtrInfo,
10254 MachinePointerInfo SrcPtrInfo,
10255 const AAMDNodes &AAInfo,
10256 BatchAAResults *BatchAA) {
10257 // Check to see if we should lower the memmove to loads and stores first.
10258 // For cases within the target-specified limits, this is the best choice.
10260 if (ConstantSize) {
10261 // Memmove with size zero? Just return the original chain.
10262 if (ConstantSize->isZero())
10263 return Chain;
10264
10266 *this, dl, Chain, Dst, Src, ConstantSize->getZExtValue(), DstAlign,
10267 SrcAlign, isVol, false, DstPtrInfo, SrcPtrInfo, AAInfo);
10268 if (Result.getNode())
10269 return Result;
10270 }
10271
10272 // Then check to see if we should lower the memmove with target-specific
10273 // code. If the target chooses to do this, this is the next best.
10274 if (TSI) {
10275 SDValue Result = TSI->EmitTargetCodeForMemmove(
10276 *this, dl, Chain, Dst, Src, Size, DstAlign, SrcAlign, isVol, DstPtrInfo,
10277 SrcPtrInfo);
10278 if (Result.getNode())
10279 return Result;
10280 }
10281
10284
10285 // FIXME: If the memmove is volatile, lowering it to plain libc memmove may
10286 // not be safe. See memcpy above for more details.
10287
10288 // Emit a library call.
10291 Args.emplace_back(Dst, PtrTy);
10292 Args.emplace_back(Src, PtrTy);
10293 Args.emplace_back(Size, getDataLayout().getIntPtrType(*getContext()));
10294 // FIXME: pass in SDLoc
10296
10297 RTLIB::LibcallImpl MemmoveImpl = Libcalls->getLibcallImpl(RTLIB::MEMMOVE);
10298
10299 bool IsTailCall = false;
10300 if (OverrideTailCall.has_value()) {
10301 IsTailCall = *OverrideTailCall;
10302 } else {
10303 bool LowersToMemmove = MemmoveImpl == RTLIB::impl_memmove;
10304 IsTailCall = isInTailCallPositionWrapper(CI, this, LowersToMemmove);
10305 }
10306 // Lowering doesn't support tail calling inside a function with a
10307 // swifterror argument yet.
10308 IsTailCall &= !hasSwiftErrorArg();
10309
10310 CLI.setDebugLoc(dl)
10311 .setChain(Chain)
10312 .setLibCallee(
10313 Libcalls->getLibcallImplCallingConv(MemmoveImpl),
10314 Dst.getValueType().getTypeForEVT(*getContext()),
10315 getExternalSymbol(MemmoveImpl, TLI->getPointerTy(getDataLayout())),
10316 std::move(Args))
10318 .setTailCall(IsTailCall);
10319
10320 std::pair<SDValue,SDValue> CallResult = TLI->LowerCallTo(CLI);
10321 return CallResult.second;
10322}
10323
10325 SDValue Dst, SDValue Src, SDValue Size,
10326 Type *SizeTy, unsigned ElemSz,
10327 bool isTailCall,
10328 MachinePointerInfo DstPtrInfo,
10329 MachinePointerInfo SrcPtrInfo) {
10330 // Lowering doesn't support tail calling inside a function with a
10331 // swifterror argument yet.
10332 isTailCall &= !hasSwiftErrorArg();
10333
10334 // Emit a library call.
10337 Args.emplace_back(Dst, IntPtrTy);
10338 Args.emplace_back(Src, IntPtrTy);
10339 Args.emplace_back(Size, SizeTy);
10340
10341 RTLIB::Libcall LibraryCall =
10343 RTLIB::LibcallImpl LibcallImpl = Libcalls->getLibcallImpl(LibraryCall);
10344 if (LibcallImpl == RTLIB::Unsupported)
10345 report_fatal_error("Unsupported element size");
10346
10348 CLI.setDebugLoc(dl)
10349 .setChain(Chain)
10350 .setLibCallee(
10351 Libcalls->getLibcallImplCallingConv(LibcallImpl),
10353 getExternalSymbol(LibcallImpl, TLI->getPointerTy(getDataLayout())),
10354 std::move(Args))
10356 .setTailCall(isTailCall);
10357
10358 std::pair<SDValue, SDValue> CallResult = TLI->LowerCallTo(CLI);
10359 return CallResult.second;
10360}
10361
10363 SDValue Src, SDValue Size, Align Alignment,
10364 bool isVol, bool AlwaysInline,
10365 const CallInst *CI,
10366 MachinePointerInfo DstPtrInfo,
10367 const AAMDNodes &AAInfo) {
10368 // Check to see if we should lower the memset to stores first.
10369 // For cases within the target-specified limits, this is the best choice.
10371 if (ConstantSize) {
10372 // Memset with size zero? Just return the original chain.
10373 if (ConstantSize->isZero())
10374 return Chain;
10375
10376 SDValue Result = getMemsetStores(*this, dl, Chain, Dst, Src,
10377 ConstantSize->getZExtValue(), Alignment,
10378 isVol, false, DstPtrInfo, AAInfo);
10379
10380 if (Result.getNode())
10381 return Result;
10382 }
10383
10384 // Then check to see if we should lower the memset with target-specific
10385 // code. If the target chooses to do this, this is the next best.
10386 if (TSI) {
10387 SDValue Result = TSI->EmitTargetCodeForMemset(
10388 *this, dl, Chain, Dst, Src, Size, Alignment, isVol, AlwaysInline, DstPtrInfo);
10389 if (Result.getNode())
10390 return Result;
10391 }
10392
10393 // If we really need inline code and the target declined to provide it,
10394 // use a (potentially long) sequence of loads and stores.
10395 if (AlwaysInline) {
10396 assert(ConstantSize && "AlwaysInline requires a constant size!");
10397 SDValue Result = getMemsetStores(*this, dl, Chain, Dst, Src,
10398 ConstantSize->getZExtValue(), Alignment,
10399 isVol, true, DstPtrInfo, AAInfo);
10400 assert(Result &&
10401 "getMemsetStores must return a valid sequence when AlwaysInline");
10402 return Result;
10403 }
10404
10406
10407 // Emit a library call.
10408 auto &Ctx = *getContext();
10409 const auto& DL = getDataLayout();
10410
10412 // FIXME: pass in SDLoc
10413 CLI.setDebugLoc(dl).setChain(Chain);
10414
10415 RTLIB::LibcallImpl BzeroImpl = Libcalls->getLibcallImpl(RTLIB::BZERO);
10416 bool UseBZero = BzeroImpl != RTLIB::Unsupported && isNullConstant(Src);
10417
10418 // If zeroing out and bzero is present, use it.
10419 if (UseBZero) {
10421 Args.emplace_back(Dst, PointerType::getUnqual(Ctx));
10422 Args.emplace_back(Size, DL.getIntPtrType(Ctx));
10423 CLI.setLibCallee(
10424 Libcalls->getLibcallImplCallingConv(BzeroImpl), Type::getVoidTy(Ctx),
10425 getExternalSymbol(BzeroImpl, TLI->getPointerTy(DL)), std::move(Args));
10426 } else {
10427 RTLIB::LibcallImpl MemsetImpl = Libcalls->getLibcallImpl(RTLIB::MEMSET);
10428
10430 Args.emplace_back(Dst, PointerType::getUnqual(Ctx));
10431 Args.emplace_back(Src, Src.getValueType().getTypeForEVT(Ctx));
10432 Args.emplace_back(Size, DL.getIntPtrType(Ctx));
10433 CLI.setLibCallee(Libcalls->getLibcallImplCallingConv(MemsetImpl),
10434 Dst.getValueType().getTypeForEVT(Ctx),
10435 getExternalSymbol(MemsetImpl, TLI->getPointerTy(DL)),
10436 std::move(Args));
10437 }
10438
10439 RTLIB::LibcallImpl MemsetImpl = Libcalls->getLibcallImpl(RTLIB::MEMSET);
10440 bool LowersToMemset = MemsetImpl == RTLIB::impl_memset;
10441
10442 // If we're going to use bzero, make sure not to tail call unless the
10443 // subsequent return doesn't need a value, as bzero doesn't return the first
10444 // arg unlike memset.
10445 bool ReturnsFirstArg = CI && funcReturnsFirstArgOfCall(*CI) && !UseBZero;
10446 bool IsTailCall = CI && CI->isTailCall() &&
10448 ReturnsFirstArg && LowersToMemset) &&
10449 // Lowering doesn't support tail calling inside a function
10450 // with a swifterror argument yet.
10452 CLI.setDiscardResult().setTailCall(IsTailCall);
10453
10454 std::pair<SDValue, SDValue> CallResult = TLI->LowerCallTo(CLI);
10455 return CallResult.second;
10456}
10457
10460 Type *SizeTy, unsigned ElemSz,
10461 bool isTailCall,
10462 MachinePointerInfo DstPtrInfo) {
10463 // Lowering doesn't support tail calling inside a function with a
10464 // swifterror argument yet.
10465 isTailCall &= !hasSwiftErrorArg();
10466
10467 // Emit a library call.
10469 Args.emplace_back(Dst, getDataLayout().getIntPtrType(*getContext()));
10470 Args.emplace_back(Value, Type::getInt8Ty(*getContext()));
10471 Args.emplace_back(Size, SizeTy);
10472
10473 RTLIB::Libcall LibraryCall =
10475 RTLIB::LibcallImpl LibcallImpl = Libcalls->getLibcallImpl(LibraryCall);
10476 if (LibcallImpl == RTLIB::Unsupported)
10477 report_fatal_error("Unsupported element size");
10478
10480 CLI.setDebugLoc(dl)
10481 .setChain(Chain)
10482 .setLibCallee(
10483 Libcalls->getLibcallImplCallingConv(LibcallImpl),
10485 getExternalSymbol(LibcallImpl, TLI->getPointerTy(getDataLayout())),
10486 std::move(Args))
10488 .setTailCall(isTailCall);
10489
10490 std::pair<SDValue, SDValue> CallResult = TLI->LowerCallTo(CLI);
10491 return CallResult.second;
10492}
10493
10494SDValue SelectionDAG::getAtomic(unsigned Opcode, const SDLoc &dl, EVT MemVT,
10496 MachineMemOperand *MMO,
10497 ISD::LoadExtType ExtType) {
10498 SDNodeKey ID(Opcode, VTList, Ops);
10499 ID.AddInteger(MemVT.getRawBits());
10500 ID.AddInteger(getSyntheticNodeSubclassData<AtomicSDNode>(
10501 dl.getIROrder(), Opcode, VTList, MemVT, MMO, ExtType));
10502 ID.AddInteger(MMO->getPointerInfo().getAddrSpace());
10503 ID.AddInteger(MMO->getFlags());
10504 FoldingSetInsertToken InsertToken;
10505 if (auto *E = cast_or_null<AtomicSDNode>(lookupNode(ID, dl, InsertToken))) {
10506 E->refineAlignment(MMO);
10507 E->refineMMOMetadata(MMO);
10508 return SDValue(E, 0);
10509 }
10510
10511 auto *N = newSDNode<AtomicSDNode>(dl.getIROrder(), dl.getDebugLoc(), Opcode,
10512 VTList, MemVT, MMO, ExtType);
10513 createOperands(N, Ops);
10514
10515 CSEMap.insert(N, InsertToken);
10516 InsertNode(N);
10517 SDValue V(N, 0);
10518 NewSDValueDbgMsg(V, "Creating new node: ", this);
10519 return V;
10520}
10521
10523 EVT MemVT, SDVTList VTs, SDValue Chain,
10524 SDValue Ptr, SDValue Cmp, SDValue Swp,
10525 MachineMemOperand *MMO) {
10526 assert(Opcode == ISD::ATOMIC_CMP_SWAP ||
10528 assert(Cmp.getValueType() == Swp.getValueType() && "Invalid Atomic Op Types");
10529
10530 SDValue Ops[] = {Chain, Ptr, Cmp, Swp};
10531 return getAtomic(Opcode, dl, MemVT, VTs, Ops, MMO);
10532}
10533
10534SDValue SelectionDAG::getAtomic(unsigned Opcode, const SDLoc &dl, EVT MemVT,
10535 SDValue Chain, SDValue Ptr, SDValue Val,
10536 MachineMemOperand *MMO) {
10537 assert((Opcode == ISD::ATOMIC_LOAD_ADD || Opcode == ISD::ATOMIC_LOAD_SUB ||
10538 Opcode == ISD::ATOMIC_LOAD_AND || Opcode == ISD::ATOMIC_LOAD_CLR ||
10539 Opcode == ISD::ATOMIC_LOAD_OR || Opcode == ISD::ATOMIC_LOAD_XOR ||
10540 Opcode == ISD::ATOMIC_LOAD_NAND || Opcode == ISD::ATOMIC_LOAD_MIN ||
10541 Opcode == ISD::ATOMIC_LOAD_MAX || Opcode == ISD::ATOMIC_LOAD_UMIN ||
10542 Opcode == ISD::ATOMIC_LOAD_UMAX || Opcode == ISD::ATOMIC_LOAD_FADD ||
10543 Opcode == ISD::ATOMIC_LOAD_FSUB || Opcode == ISD::ATOMIC_LOAD_FMAX ||
10544 Opcode == ISD::ATOMIC_LOAD_FMIN ||
10545 Opcode == ISD::ATOMIC_LOAD_FMINIMUM ||
10546 Opcode == ISD::ATOMIC_LOAD_FMAXIMUM ||
10547 Opcode == ISD::ATOMIC_LOAD_UINC_WRAP ||
10548 Opcode == ISD::ATOMIC_LOAD_UDEC_WRAP ||
10549 Opcode == ISD::ATOMIC_LOAD_USUB_COND ||
10550 Opcode == ISD::ATOMIC_LOAD_USUB_SAT || Opcode == ISD::ATOMIC_SWAP ||
10551 Opcode == ISD::ATOMIC_STORE) &&
10552 "Invalid Atomic Op");
10553
10554 EVT VT = Val.getValueType();
10555
10556 SDVTList VTs = Opcode == ISD::ATOMIC_STORE ? getVTList(MVT::Other) :
10557 getVTList(VT, MVT::Other);
10558 SDValue Ops[] = {Chain, Ptr, Val};
10559 return getAtomic(Opcode, dl, MemVT, VTs, Ops, MMO);
10560}
10561
10563 EVT MemVT, EVT VT, SDValue Chain,
10564 SDValue Ptr, MachineMemOperand *MMO) {
10565 SDVTList VTs = getVTList(VT, MVT::Other);
10566 SDValue Ops[] = {Chain, Ptr};
10567 return getAtomic(ISD::ATOMIC_LOAD, dl, MemVT, VTs, Ops, MMO, ExtType);
10568}
10569
10570/// getMergeValues - Create a MERGE_VALUES node from the given operands.
10572 if (Ops.size() == 1)
10573 return Ops[0];
10574
10576 VTs.reserve(Ops.size());
10577 for (const SDValue &Op : Ops)
10578 VTs.push_back(Op.getValueType());
10579 return getNode(ISD::MERGE_VALUES, dl, getVTList(VTs), Ops);
10580}
10581
10583 SDValue Chain, const SDLoc &dl) {
10584 SmallVector<SDValue, 4> RetValues;
10585 RetValues.reserve(ResultTypes.size());
10586 for (EVT VT : ResultTypes)
10587 RetValues.push_back(VT == MVT::Other ? Chain : getPOISON(VT));
10588 return getMergeValues(RetValues, dl);
10589}
10590
10592 unsigned Opcode, const SDLoc &dl, SDVTList VTList, ArrayRef<SDValue> Ops,
10593 EVT MemVT, MachinePointerInfo PtrInfo, Align Alignment,
10595 const AAMDNodes &AAInfo) {
10596 if (Size.hasValue() && !Size.getValue())
10598
10600 MachineMemOperand *MMO =
10601 MF.getMachineMemOperand(PtrInfo, Flags, Size, Alignment, AAInfo);
10602
10603 return getMemIntrinsicNode(Opcode, dl, VTList, Ops, MemVT, MMO);
10604}
10605
10607 SDVTList VTList,
10608 ArrayRef<SDValue> Ops, EVT MemVT,
10609 MachineMemOperand *MMO) {
10610 return getMemIntrinsicNode(Opcode, dl, VTList, Ops, MemVT, ArrayRef(MMO));
10611}
10612
10614 SDVTList VTList,
10615 ArrayRef<SDValue> Ops, EVT MemVT,
10617 assert(!MMOs.empty() && "Must have at least one MMO");
10618 assert(
10619 (Opcode == ISD::INTRINSIC_VOID || Opcode == ISD::INTRINSIC_W_CHAIN ||
10620 Opcode == ISD::PREFETCH ||
10621 (Opcode <= (unsigned)std::numeric_limits<int>::max() &&
10622 Opcode >= ISD::BUILTIN_OP_END && TSI->isTargetMemoryOpcode(Opcode))) &&
10623 "Opcode is not a memory-accessing opcode!");
10624
10626 if (MMOs.size() == 1) {
10627 MemRefs = MMOs[0];
10628 } else {
10629 // Allocate: [size_t count][MMO*][MMO*]...
10630 size_t AllocSize =
10631 sizeof(size_t) + MMOs.size() * sizeof(MachineMemOperand *);
10632 void *Buffer = Allocator.Allocate(AllocSize, alignof(size_t));
10633 size_t *CountPtr = static_cast<size_t *>(Buffer);
10634 *CountPtr = MMOs.size();
10635 MachineMemOperand **Array =
10636 reinterpret_cast<MachineMemOperand **>(CountPtr + 1);
10637 llvm::copy(MMOs, Array);
10638 MemRefs = Array;
10639 }
10640
10641 // Memoize the node unless it returns a glue result.
10643 if (VTList.VTs[VTList.NumVTs-1] != MVT::Glue) {
10644 SDNodeKey ID(Opcode, VTList, Ops);
10645 ID.AddInteger(getSyntheticNodeSubclassData<MemIntrinsicSDNode>(
10646 Opcode, dl.getIROrder(), VTList, MemVT, MemRefs));
10647 ID.AddInteger(MemVT.getRawBits());
10648 for (const MachineMemOperand *MMO : MMOs) {
10649 ID.AddInteger(MMO->getPointerInfo().getAddrSpace());
10650 ID.AddInteger(MMO->getFlags());
10651 }
10652 FoldingSetInsertToken InsertToken;
10653 if (SDNode *E = lookupNode(ID, dl, InsertToken)) {
10654 cast<MemIntrinsicSDNode>(E)->refineAlignment(MMOs);
10655 return SDValue(E, 0);
10656 }
10657
10658 N = newSDNode<MemIntrinsicSDNode>(Opcode, dl.getIROrder(), dl.getDebugLoc(),
10659 VTList, MemVT, MemRefs);
10660 createOperands(N, Ops);
10661 CSEMap.insert(N, InsertToken);
10662 } else {
10663 N = newSDNode<MemIntrinsicSDNode>(Opcode, dl.getIROrder(), dl.getDebugLoc(),
10664 VTList, MemVT, MemRefs);
10665 createOperands(N, Ops);
10666 }
10667 InsertNode(N);
10668 SDValue V(N, 0);
10669 NewSDValueDbgMsg(V, "Creating new node: ", this);
10670 return V;
10671}
10672
10674 SDValue Chain, int FrameIndex) {
10675 const unsigned Opcode = IsStart ? ISD::LIFETIME_START : ISD::LIFETIME_END;
10676 const auto VTs = getVTList(MVT::Other);
10677 SDValue Ops[2] = {
10678 Chain,
10679 getFrameIndex(FrameIndex,
10680 getTargetLoweringInfo().getFrameIndexTy(getDataLayout()),
10681 true)};
10682
10683 SDNodeKey ID(Opcode, VTs, Ops);
10684 FoldingSetInsertToken InsertToken;
10685 if (SDNode *E = lookupNode(ID, dl, InsertToken))
10686 return SDValue(E, 0);
10687
10688 LifetimeSDNode *N =
10689 newSDNode<LifetimeSDNode>(Opcode, dl.getIROrder(), dl.getDebugLoc(), VTs);
10690 createOperands(N, Ops);
10691 CSEMap.insert(N, InsertToken);
10692 InsertNode(N);
10693 SDValue V(N, 0);
10694 NewSDValueDbgMsg(V, "Creating new node: ", this);
10695 return V;
10696}
10697
10699 uint64_t Guid, uint64_t Index,
10700 uint32_t Attr) {
10701 const unsigned Opcode = ISD::PSEUDO_PROBE;
10702 const auto VTs = getVTList(MVT::Other);
10703 SDValue Ops[] = {Chain};
10704 SDNodeKey ID(Opcode, VTs, Ops);
10705 ID.AddInteger(Guid);
10706 ID.AddInteger(Index);
10707 ID.AddInteger(Attr);
10708 FoldingSetInsertToken InsertToken;
10709 if (SDNode *E = lookupNode(ID, Dl, InsertToken))
10710 return SDValue(E, 0);
10711
10712 auto *N = newSDNode<PseudoProbeSDNode>(
10713 Opcode, Dl.getIROrder(), Dl.getDebugLoc(), VTs, Guid, Index, Attr);
10714 createOperands(N, Ops);
10715 CSEMap.insert(N, InsertToken);
10716 InsertNode(N);
10717 SDValue V(N, 0);
10718 NewSDValueDbgMsg(V, "Creating new node: ", this);
10719 return V;
10720}
10721
10722/// InferPointerInfo - If the specified ptr/offset is a frame index, infer a
10723/// MachinePointerInfo record from it. This is particularly useful because the
10724/// code generator has many cases where it doesn't bother passing in a
10725/// MachinePointerInfo to getLoad or getStore when it has "FI+Cst".
10727 SelectionDAG &DAG, SDValue Ptr,
10728 int64_t Offset = 0) {
10729 // If this is FI+Offset, we can model it.
10730 if (const FrameIndexSDNode *FI = dyn_cast<FrameIndexSDNode>(Ptr))
10732 FI->getIndex(), Offset);
10733
10734 // If this is (FI+Offset1)+Offset2, we can model it.
10735 if (Ptr.getOpcode() != ISD::ADD ||
10738 return Info;
10739
10740 int FI = cast<FrameIndexSDNode>(Ptr.getOperand(0))->getIndex();
10742 DAG.getMachineFunction(), FI,
10743 Offset + cast<ConstantSDNode>(Ptr.getOperand(1))->getSExtValue());
10744}
10745
10746/// InferPointerInfo - If the specified ptr/offset is a frame index, infer a
10747/// MachinePointerInfo record from it. This is particularly useful because the
10748/// code generator has many cases where it doesn't bother passing in a
10749/// MachinePointerInfo to getLoad or getStore when it has "FI+Cst".
10751 SelectionDAG &DAG, SDValue Ptr,
10752 SDValue OffsetOp) {
10753 // If the 'Offset' value isn't a constant, we can't handle this.
10755 return InferPointerInfo(Info, DAG, Ptr, OffsetNode->getSExtValue());
10756 if (OffsetOp.isUndef())
10757 return InferPointerInfo(Info, DAG, Ptr);
10758 return Info;
10759}
10760
10762 EVT VT, const SDLoc &dl, SDValue Chain,
10763 SDValue Ptr, SDValue Offset,
10764 MachinePointerInfo PtrInfo, EVT MemVT,
10765 Align Alignment,
10766 MachineMemOperand::Flags MMOFlags,
10767 const MMOMetadata &Metadata) {
10768 assert(Chain.getValueType() == MVT::Other &&
10769 "Invalid chain type");
10770
10771 MMOFlags |= MachineMemOperand::MOLoad;
10772 assert((MMOFlags & MachineMemOperand::MOStore) == 0);
10773 // If we don't have a PtrInfo, infer the trivial frame index case to simplify
10774 // clients.
10775 if (PtrInfo.V.isNull())
10776 PtrInfo = InferPointerInfo(PtrInfo, *this, Ptr, Offset);
10777
10778 TypeSize Size = MemVT.getStoreSize();
10780 MachineMemOperand *MMO =
10781 MF.getMachineMemOperand(PtrInfo, MMOFlags, Size, Alignment, Metadata);
10782 return getLoad(AM, ExtType, VT, dl, Chain, Ptr, Offset, MemVT, MMO);
10783}
10784
10786 EVT VT, const SDLoc &dl, SDValue Chain,
10787 SDValue Ptr, SDValue Offset, EVT MemVT,
10788 MachineMemOperand *MMO) {
10789 if (VT == MemVT) {
10790 ExtType = ISD::NON_EXTLOAD;
10791 } else if (ExtType == ISD::NON_EXTLOAD) {
10792 assert(VT == MemVT && "Non-extending load from different memory type!");
10793 } else {
10794 // Extending load.
10795 assert(MemVT.getScalarType().bitsLT(VT.getScalarType()) &&
10796 "Should only be an extending load, not truncating!");
10797 assert(VT.isInteger() == MemVT.isInteger() &&
10798 "Cannot convert from FP to Int or Int -> FP!");
10799 assert(VT.isVector() == MemVT.isVector() &&
10800 "Cannot use an ext load to convert to or from a vector!");
10801 assert((!VT.isVector() ||
10803 "Cannot use an ext load to change the number of vector elements!");
10804 }
10805
10806 assert((!MMO->getRanges() ||
10808 ->getBitWidth() == MemVT.getScalarSizeInBits() &&
10809 MemVT.isInteger())) &&
10810 "Range metadata and load type must match!");
10811
10812 bool Indexed = AM != ISD::UNINDEXED;
10813 assert((Indexed || Offset.getOpcode() == ISD::POISON) &&
10814 "Unindexed load with an offset!");
10815
10816 SDVTList VTs = Indexed ?
10817 getVTList(VT, Ptr.getValueType(), MVT::Other) : getVTList(VT, MVT::Other);
10818 SDValue Ops[] = { Chain, Ptr, Offset };
10819 SDNodeKey ID(ISD::LOAD, VTs, Ops);
10820 ID.AddInteger(MemVT.getRawBits());
10821 ID.AddInteger(getSyntheticNodeSubclassData<LoadSDNode>(
10822 dl.getIROrder(), VTs, AM, ExtType, MemVT, MMO));
10823 ID.AddInteger(MMO->getPointerInfo().getAddrSpace());
10824 ID.AddInteger(MMO->getFlags());
10825 FoldingSetInsertToken InsertToken;
10826 if (auto *E = cast_or_null<LoadSDNode>(lookupNode(ID, dl, InsertToken))) {
10827 E->refineAlignment(MMO);
10828 E->refineMMOMetadata(MMO);
10829 return SDValue(E, 0);
10830 }
10831 auto *N = newSDNode<LoadSDNode>(dl.getIROrder(), dl.getDebugLoc(), VTs, AM,
10832 ExtType, MemVT, MMO);
10833 createOperands(N, Ops);
10834
10835 CSEMap.insert(N, InsertToken);
10836 InsertNode(N);
10837 SDValue V(N, 0);
10838 NewSDValueDbgMsg(V, "Creating new node: ", this);
10839 return V;
10840}
10841
10843 SDValue Ptr, MachinePointerInfo PtrInfo,
10844 MaybeAlign Alignment,
10845 MachineMemOperand::Flags MMOFlags,
10846 const MMOMetadata &Metadata) {
10848 return getLoad(ISD::UNINDEXED, ISD::NON_EXTLOAD, VT, dl, Chain, Ptr, Undef,
10849 PtrInfo, VT, Alignment, MMOFlags, Metadata);
10850}
10851
10853 SDValue Ptr, MachineMemOperand *MMO) {
10855 return getLoad(ISD::UNINDEXED, ISD::NON_EXTLOAD, VT, dl, Chain, Ptr, Undef,
10856 VT, MMO);
10857}
10858
10860 EVT VT, SDValue Chain, SDValue Ptr,
10861 MachinePointerInfo PtrInfo, EVT MemVT,
10862 MaybeAlign Alignment,
10863 MachineMemOperand::Flags MMOFlags,
10864 const MMOMetadata &Metadata) {
10866 return getLoad(ISD::UNINDEXED, ExtType, VT, dl, Chain, Ptr, Undef, PtrInfo,
10867 MemVT, Alignment, MMOFlags, Metadata);
10868}
10869
10871 EVT VT, SDValue Chain, SDValue Ptr, EVT MemVT,
10872 MachineMemOperand *MMO) {
10874 return getLoad(ISD::UNINDEXED, ExtType, VT, dl, Chain, Ptr, Undef,
10875 MemVT, MMO);
10876}
10877
10881 LoadSDNode *LD = cast<LoadSDNode>(OrigLoad);
10882 assert(LD->getOffset().getOpcode() == ISD::POISON &&
10883 "Load is already a indexed load!");
10884 // Don't propagate the invariant or dereferenceable flags.
10885 auto MMOFlags =
10886 LD->getMemOperand()->getFlags() &
10888 return getLoad(
10889 AM, LD->getExtensionType(), OrigLoad.getValueType(), dl, LD->getChain(),
10890 Base, Offset, LD->getPointerInfo(), LD->getMemoryVT(), LD->getAlign(),
10891 MMOFlags,
10892 MMOMetadata(LD->getAAInfo(), LD->getRanges(), LD->getMemCacheHint()));
10893}
10894
10896 SDValue Ptr, MachinePointerInfo PtrInfo,
10897 Align Alignment,
10898 MachineMemOperand::Flags MMOFlags,
10899 const MMOMetadata &Metadata) {
10900 assert(Chain.getValueType() == MVT::Other && "Invalid chain type");
10901
10902 MMOFlags |= MachineMemOperand::MOStore;
10903 assert((MMOFlags & MachineMemOperand::MOLoad) == 0);
10904 assert(!Metadata.Ranges && "range metadata is invalid for stores");
10905
10906 if (PtrInfo.V.isNull())
10907 PtrInfo = InferPointerInfo(PtrInfo, *this, Ptr);
10908
10911 MachineMemOperand *MMO =
10912 MF.getMachineMemOperand(PtrInfo, MMOFlags, Size, Alignment, Metadata);
10913 return getStore(Chain, dl, Val, Ptr, MMO);
10914}
10915
10917 SDValue Ptr, MachineMemOperand *MMO) {
10919 return getStore(Chain, dl, Val, Ptr, Undef, Val.getValueType(), MMO,
10921}
10922
10924 SDValue Ptr, SDValue Offset, EVT SVT,
10926 bool IsTruncating) {
10927 assert(Chain.getValueType() == MVT::Other && "Invalid chain type");
10928 EVT VT = Val.getValueType();
10929 if (VT == SVT) {
10930 IsTruncating = false;
10931 } else if (!IsTruncating) {
10932 assert(VT == SVT && "No-truncating store from different memory type!");
10933 } else {
10935 "Should only be a truncating store, not extending!");
10936 assert(VT.isInteger() == SVT.isInteger() && "Can't do FP-INT conversion!");
10937 assert(VT.isVector() == SVT.isVector() &&
10938 "Cannot use trunc store to convert to or from a vector!");
10939 assert((!VT.isVector() ||
10941 "Cannot use trunc store to change the number of vector elements!");
10942 }
10943
10944 bool Indexed = AM != ISD::UNINDEXED;
10945 assert((Indexed || Offset.getOpcode() == ISD::POISON) &&
10946 "Unindexed store with an offset!");
10947 SDVTList VTs = Indexed ? getVTList(Ptr.getValueType(), MVT::Other)
10948 : getVTList(MVT::Other);
10949 SDValue Ops[] = {Chain, Val, Ptr, Offset};
10950 SDNodeKey ID(ISD::STORE, VTs, Ops);
10951 ID.AddInteger(SVT.getRawBits());
10952 ID.AddInteger(getSyntheticNodeSubclassData<StoreSDNode>(
10953 dl.getIROrder(), VTs, AM, IsTruncating, SVT, MMO));
10954 ID.AddInteger(MMO->getPointerInfo().getAddrSpace());
10955 ID.AddInteger(MMO->getFlags());
10956 FoldingSetInsertToken InsertToken;
10957 if (SDNode *E = lookupNode(ID, dl, InsertToken)) {
10958 cast<StoreSDNode>(E)->refineAlignment(MMO);
10959 cast<StoreSDNode>(E)->refineMMOMetadata(MMO);
10960 return SDValue(E, 0);
10961 }
10962 auto *N = newSDNode<StoreSDNode>(dl.getIROrder(), dl.getDebugLoc(), VTs, AM,
10963 IsTruncating, SVT, MMO);
10964 createOperands(N, Ops);
10965
10966 CSEMap.insert(N, InsertToken);
10967 InsertNode(N);
10968 SDValue V(N, 0);
10969 NewSDValueDbgMsg(V, "Creating new node: ", this);
10970 return V;
10971}
10972
10974 SDValue Ptr, SDValue Offset,
10975 MachinePointerInfo PtrInfo, EVT SVT,
10976 Align Alignment,
10977 MachineMemOperand::Flags MMOFlags,
10978 const MMOMetadata &Metadata) {
10979 assert(Chain.getValueType() == MVT::Other &&
10980 "Invalid chain type");
10981
10982 MMOFlags |= MachineMemOperand::MOStore;
10983 assert((MMOFlags & MachineMemOperand::MOLoad) == 0);
10984 assert(!Metadata.Ranges && "range metadata is invalid for stores");
10985
10986 if (PtrInfo.V.isNull())
10987 PtrInfo = InferPointerInfo(PtrInfo, *this, Ptr);
10988
10990 MachineMemOperand *MMO = MF.getMachineMemOperand(
10991 PtrInfo, MMOFlags, SVT.getStoreSize(), Alignment, Metadata);
10992 return getTruncStore(Chain, dl, Val, Ptr, Offset, SVT, MMO);
10993}
10994
10996 SDValue Ptr, MachinePointerInfo PtrInfo,
10997 EVT SVT, Align Alignment,
10998 MachineMemOperand::Flags MMOFlags,
10999 const MMOMetadata &Metadata) {
11000 return getTruncStore(Chain, dl, Val, Ptr, getPOISON(Ptr.getValueType()),
11001 PtrInfo, SVT, Alignment, MMOFlags, Metadata);
11002}
11003
11005 SDValue Ptr, SDValue Offset, EVT SVT,
11006 MachineMemOperand *MMO) {
11007 return getStore(Chain, dl, Val, Ptr, Offset, SVT, MMO, ISD::UNINDEXED, true);
11008}
11009
11011 SDValue Ptr, EVT SVT,
11012 MachineMemOperand *MMO) {
11013 return getStore(Chain, dl, Val, Ptr, getPOISON(Ptr.getValueType()), SVT, MMO,
11014 ISD::UNINDEXED, true);
11015}
11016
11020 StoreSDNode *ST = cast<StoreSDNode>(OrigStore);
11021 assert(ST->getOffset().getOpcode() == ISD::POISON &&
11022 "Store is already a indexed store!");
11023 return getStore(ST->getChain(), dl, ST->getValue(), Base, Offset,
11024 ST->getMemoryVT(), ST->getMemOperand(), AM,
11025 ST->isTruncatingStore());
11026}
11027
11029 ISD::MemIndexedMode AM, ISD::LoadExtType ExtType, EVT VT, const SDLoc &dl,
11030 SDValue Chain, SDValue Ptr, SDValue Offset, SDValue Mask, SDValue EVL,
11031 MachinePointerInfo PtrInfo, EVT MemVT, Align Alignment,
11032 MachineMemOperand::Flags MMOFlags, const AAMDNodes &AAInfo,
11033 const MDNode *Ranges, bool IsExpanding) {
11034 MMOFlags |= MachineMemOperand::MOLoad;
11035 assert((MMOFlags & MachineMemOperand::MOStore) == 0);
11036 // If we don't have a PtrInfo, infer the trivial frame index case to simplify
11037 // clients.
11038 if (PtrInfo.V.isNull())
11039 PtrInfo = InferPointerInfo(PtrInfo, *this, Ptr, Offset);
11040
11041 TypeSize Size = MemVT.getStoreSize();
11043 MachineMemOperand *MMO = MF.getMachineMemOperand(
11044 PtrInfo, MMOFlags, Size, Alignment, MMOMetadata(AAInfo, Ranges));
11045 return getLoadVP(AM, ExtType, VT, dl, Chain, Ptr, Offset, Mask, EVL, MemVT,
11046 MMO, IsExpanding);
11047}
11048
11050 ISD::LoadExtType ExtType, EVT VT,
11051 const SDLoc &dl, SDValue Chain, SDValue Ptr,
11052 SDValue Offset, SDValue Mask, SDValue EVL,
11053 EVT MemVT, MachineMemOperand *MMO,
11054 bool IsExpanding) {
11055 assert(Chain.getValueType() == MVT::Other && "Invalid chain type");
11056 assert(Mask.getValueType().getVectorElementCount() ==
11057 VT.getVectorElementCount() &&
11058 "Vector width mismatch between mask and data");
11059
11060 bool Indexed = AM != ISD::UNINDEXED;
11061 assert((Indexed || Offset.getOpcode() == ISD::POISON) &&
11062 "Unindexed load with an offset!");
11063
11064 SDVTList VTs = Indexed ? getVTList(VT, Ptr.getValueType(), MVT::Other)
11065 : getVTList(VT, MVT::Other);
11066 SDValue Ops[] = {Chain, Ptr, Offset, Mask, EVL};
11067 SDNodeKey ID(ISD::VP_LOAD, VTs, Ops);
11068 ID.AddInteger(MemVT.getRawBits());
11069 ID.AddInteger(getSyntheticNodeSubclassData<VPLoadSDNode>(
11070 dl.getIROrder(), VTs, AM, ExtType, IsExpanding, MemVT, MMO));
11071 ID.AddInteger(MMO->getPointerInfo().getAddrSpace());
11072 ID.AddInteger(MMO->getFlags());
11073 FoldingSetInsertToken InsertToken;
11074 if (auto *E = cast_or_null<VPLoadSDNode>(lookupNode(ID, dl, InsertToken))) {
11075 E->refineAlignment(MMO);
11076 E->refineMMOMetadata(MMO);
11077 return SDValue(E, 0);
11078 }
11079 auto *N = newSDNode<VPLoadSDNode>(dl.getIROrder(), dl.getDebugLoc(), VTs, AM,
11080 ExtType, IsExpanding, MemVT, MMO);
11081 createOperands(N, Ops);
11082
11083 CSEMap.insert(N, InsertToken);
11084 InsertNode(N);
11085 SDValue V(N, 0);
11086 NewSDValueDbgMsg(V, "Creating new node: ", this);
11087 return V;
11088}
11089
11091 SDValue Ptr, SDValue Mask, SDValue EVL,
11092 MachinePointerInfo PtrInfo,
11093 MaybeAlign Alignment,
11094 MachineMemOperand::Flags MMOFlags,
11095 const AAMDNodes &AAInfo, const MDNode *Ranges,
11096 bool IsExpanding) {
11098 return getLoadVP(ISD::UNINDEXED, ISD::NON_EXTLOAD, VT, dl, Chain, Ptr, Undef,
11099 Mask, EVL, PtrInfo, VT, Alignment, MMOFlags, AAInfo, Ranges,
11100 IsExpanding);
11101}
11102
11104 SDValue Ptr, SDValue Mask, SDValue EVL,
11105 MachineMemOperand *MMO, bool IsExpanding) {
11107 return getLoadVP(ISD::UNINDEXED, ISD::NON_EXTLOAD, VT, dl, Chain, Ptr, Undef,
11108 Mask, EVL, VT, MMO, IsExpanding);
11109}
11110
11112 EVT VT, SDValue Chain, SDValue Ptr,
11113 SDValue Mask, SDValue EVL,
11114 MachinePointerInfo PtrInfo, EVT MemVT,
11115 MaybeAlign Alignment,
11116 MachineMemOperand::Flags MMOFlags,
11117 const AAMDNodes &AAInfo, bool IsExpanding) {
11119 return getLoadVP(ISD::UNINDEXED, ExtType, VT, dl, Chain, Ptr, Undef, Mask,
11120 EVL, PtrInfo, MemVT, Alignment, MMOFlags, AAInfo, nullptr,
11121 IsExpanding);
11122}
11123
11125 EVT VT, SDValue Chain, SDValue Ptr,
11126 SDValue Mask, SDValue EVL, EVT MemVT,
11127 MachineMemOperand *MMO, bool IsExpanding) {
11129 return getLoadVP(ISD::UNINDEXED, ExtType, VT, dl, Chain, Ptr, Undef, Mask,
11130 EVL, MemVT, MMO, IsExpanding);
11131}
11132
11134 SDValue Ptr, SDValue Offset, SDValue Mask,
11135 SDValue EVL, EVT MemVT, MachineMemOperand *MMO,
11136 ISD::MemIndexedMode AM, bool IsTruncating,
11137 bool IsCompressing) {
11138 assert(Chain.getValueType() == MVT::Other && "Invalid chain type");
11139 assert(Mask.getValueType().getVectorElementCount() ==
11141 "Vector width mismatch between mask and data");
11142
11143 bool Indexed = AM != ISD::UNINDEXED;
11144 assert((Indexed || Offset.getOpcode() == ISD::POISON) &&
11145 "Unindexed vp_store with an offset!");
11146 SDVTList VTs = Indexed ? getVTList(Ptr.getValueType(), MVT::Other)
11147 : getVTList(MVT::Other);
11148 SDValue Ops[] = {Chain, Val, Ptr, Offset, Mask, EVL};
11149 SDNodeKey ID(ISD::VP_STORE, VTs, Ops);
11150 ID.AddInteger(MemVT.getRawBits());
11151 ID.AddInteger(getSyntheticNodeSubclassData<VPStoreSDNode>(
11152 dl.getIROrder(), VTs, AM, IsTruncating, IsCompressing, MemVT, MMO));
11153 ID.AddInteger(MMO->getPointerInfo().getAddrSpace());
11154 ID.AddInteger(MMO->getFlags());
11155 FoldingSetInsertToken InsertToken;
11156 if (SDNode *E = lookupNode(ID, dl, InsertToken)) {
11157 cast<VPStoreSDNode>(E)->refineAlignment(MMO);
11158 return SDValue(E, 0);
11159 }
11160 auto *N = newSDNode<VPStoreSDNode>(dl.getIROrder(), dl.getDebugLoc(), VTs, AM,
11161 IsTruncating, IsCompressing, MemVT, MMO);
11162 createOperands(N, Ops);
11163
11164 CSEMap.insert(N, InsertToken);
11165 InsertNode(N);
11166 SDValue V(N, 0);
11167 NewSDValueDbgMsg(V, "Creating new node: ", this);
11168 return V;
11169}
11170
11172 SDValue Val, SDValue Ptr, SDValue Mask,
11173 SDValue EVL, MachinePointerInfo PtrInfo,
11174 EVT SVT, Align Alignment,
11175 MachineMemOperand::Flags MMOFlags,
11176 const AAMDNodes &AAInfo,
11177 bool IsCompressing) {
11178 assert(Chain.getValueType() == MVT::Other && "Invalid chain type");
11179
11180 MMOFlags |= MachineMemOperand::MOStore;
11181 assert((MMOFlags & MachineMemOperand::MOLoad) == 0);
11182
11183 if (PtrInfo.V.isNull())
11184 PtrInfo = InferPointerInfo(PtrInfo, *this, Ptr);
11185
11187 MachineMemOperand *MMO = MF.getMachineMemOperand(
11188 PtrInfo, MMOFlags, SVT.getStoreSize(), Alignment, AAInfo);
11189 return getTruncStoreVP(Chain, dl, Val, Ptr, Mask, EVL, SVT, MMO,
11190 IsCompressing);
11191}
11192
11194 SDValue Val, SDValue Ptr, SDValue Mask,
11195 SDValue EVL, EVT SVT,
11196 MachineMemOperand *MMO,
11197 bool IsCompressing) {
11198 EVT VT = Val.getValueType();
11199
11200 assert(Chain.getValueType() == MVT::Other && "Invalid chain type");
11201 if (VT == SVT)
11202 return getStoreVP(Chain, dl, Val, Ptr, getPOISON(Ptr.getValueType()), Mask,
11203 EVL, VT, MMO, ISD::UNINDEXED,
11204 /*IsTruncating*/ false, IsCompressing);
11205
11207 "Should only be a truncating store, not extending!");
11208 assert(VT.isInteger() == SVT.isInteger() && "Can't do FP-INT conversion!");
11209 assert(VT.isVector() == SVT.isVector() &&
11210 "Cannot use trunc store to convert to or from a vector!");
11211 assert((!VT.isVector() ||
11213 "Cannot use trunc store to change the number of vector elements!");
11214
11215 SDVTList VTs = getVTList(MVT::Other);
11217 SDValue Ops[] = {Chain, Val, Ptr, Undef, Mask, EVL};
11218 SDNodeKey ID(ISD::VP_STORE, VTs, Ops);
11219 ID.AddInteger(SVT.getRawBits());
11220 ID.AddInteger(getSyntheticNodeSubclassData<VPStoreSDNode>(
11221 dl.getIROrder(), VTs, ISD::UNINDEXED, true, IsCompressing, SVT, MMO));
11222 ID.AddInteger(MMO->getPointerInfo().getAddrSpace());
11223 ID.AddInteger(MMO->getFlags());
11224 FoldingSetInsertToken InsertToken;
11225 if (SDNode *E = lookupNode(ID, dl, InsertToken)) {
11226 cast<VPStoreSDNode>(E)->refineAlignment(MMO);
11227 return SDValue(E, 0);
11228 }
11229 auto *N =
11230 newSDNode<VPStoreSDNode>(dl.getIROrder(), dl.getDebugLoc(), VTs,
11231 ISD::UNINDEXED, true, IsCompressing, SVT, MMO);
11232 createOperands(N, Ops);
11233
11234 CSEMap.insert(N, InsertToken);
11235 InsertNode(N);
11236 SDValue V(N, 0);
11237 NewSDValueDbgMsg(V, "Creating new node: ", this);
11238 return V;
11239}
11240
11242 ISD::MemIndexedMode AM, ISD::LoadExtType ExtType, EVT VT, const SDLoc &DL,
11243 SDValue Chain, SDValue Ptr, SDValue Offset, SDValue Stride, SDValue Mask,
11244 SDValue EVL, EVT MemVT, MachineMemOperand *MMO, bool IsExpanding) {
11245 bool Indexed = AM != ISD::UNINDEXED;
11246 assert((Indexed || Offset.getOpcode() == ISD::POISON) &&
11247 "Unindexed load with an offset!");
11248
11249 SDValue Ops[] = {Chain, Ptr, Offset, Stride, Mask, EVL};
11250 SDVTList VTs = Indexed ? getVTList(VT, Ptr.getValueType(), MVT::Other)
11251 : getVTList(VT, MVT::Other);
11252 SDNodeKey ID(ISD::EXPERIMENTAL_VP_STRIDED_LOAD, VTs, Ops);
11253 ID.AddInteger(MemVT.getRawBits());
11254 ID.AddInteger(getSyntheticNodeSubclassData<VPStridedLoadSDNode>(
11255 DL.getIROrder(), VTs, AM, ExtType, IsExpanding, MemVT, MMO));
11256 ID.AddInteger(MMO->getPointerInfo().getAddrSpace());
11257
11258 FoldingSetInsertToken InsertToken;
11259 if (SDNode *E = lookupNode(ID, DL, InsertToken)) {
11260 cast<VPStridedLoadSDNode>(E)->refineAlignment(MMO);
11261 return SDValue(E, 0);
11262 }
11263
11264 auto *N =
11265 newSDNode<VPStridedLoadSDNode>(DL.getIROrder(), DL.getDebugLoc(), VTs, AM,
11266 ExtType, IsExpanding, MemVT, MMO);
11267 createOperands(N, Ops);
11268 CSEMap.insert(N, InsertToken);
11269 InsertNode(N);
11270 SDValue V(N, 0);
11271 NewSDValueDbgMsg(V, "Creating new node: ", this);
11272 return V;
11273}
11274
11276 SDValue Ptr, SDValue Stride,
11277 SDValue Mask, SDValue EVL,
11278 MachineMemOperand *MMO,
11279 bool IsExpanding) {
11281 return getStridedLoadVP(ISD::UNINDEXED, ISD::NON_EXTLOAD, VT, DL, Chain, Ptr,
11282 Undef, Stride, Mask, EVL, VT, MMO, IsExpanding);
11283}
11284
11286 ISD::LoadExtType ExtType, const SDLoc &DL, EVT VT, SDValue Chain,
11287 SDValue Ptr, SDValue Stride, SDValue Mask, SDValue EVL, EVT MemVT,
11288 MachineMemOperand *MMO, bool IsExpanding) {
11290 return getStridedLoadVP(ISD::UNINDEXED, ExtType, VT, DL, Chain, Ptr, Undef,
11291 Stride, Mask, EVL, MemVT, MMO, IsExpanding);
11292}
11293
11295 SDValue Val, SDValue Ptr,
11296 SDValue Offset, SDValue Stride,
11297 SDValue Mask, SDValue EVL, EVT MemVT,
11298 MachineMemOperand *MMO,
11300 bool IsTruncating, bool IsCompressing) {
11301 assert(Chain.getValueType() == MVT::Other && "Invalid chain type");
11302 bool Indexed = AM != ISD::UNINDEXED;
11303 assert((Indexed || Offset.getOpcode() == ISD::POISON) &&
11304 "Unindexed vp_store with an offset!");
11305 SDVTList VTs = Indexed ? getVTList(Ptr.getValueType(), MVT::Other)
11306 : getVTList(MVT::Other);
11307 SDValue Ops[] = {Chain, Val, Ptr, Offset, Stride, Mask, EVL};
11308 SDNodeKey ID(ISD::EXPERIMENTAL_VP_STRIDED_STORE, VTs, Ops);
11309 ID.AddInteger(MemVT.getRawBits());
11310 ID.AddInteger(getSyntheticNodeSubclassData<VPStridedStoreSDNode>(
11311 DL.getIROrder(), VTs, AM, IsTruncating, IsCompressing, MemVT, MMO));
11312 ID.AddInteger(MMO->getPointerInfo().getAddrSpace());
11313 FoldingSetInsertToken InsertToken;
11314 if (SDNode *E = lookupNode(ID, DL, InsertToken)) {
11315 cast<VPStridedStoreSDNode>(E)->refineAlignment(MMO);
11316 return SDValue(E, 0);
11317 }
11318 auto *N = newSDNode<VPStridedStoreSDNode>(DL.getIROrder(), DL.getDebugLoc(),
11319 VTs, AM, IsTruncating,
11320 IsCompressing, MemVT, MMO);
11321 createOperands(N, Ops);
11322
11323 CSEMap.insert(N, InsertToken);
11324 InsertNode(N);
11325 SDValue V(N, 0);
11326 NewSDValueDbgMsg(V, "Creating new node: ", this);
11327 return V;
11328}
11329
11332 ISD::MemIndexType IndexType) {
11333 assert(Ops.size() == 6 && "Incompatible number of operands");
11334
11335 SDNodeKey ID(ISD::VP_GATHER, VTs, Ops);
11336 ID.AddInteger(VT.getRawBits());
11337 ID.AddInteger(getSyntheticNodeSubclassData<VPGatherSDNode>(
11338 dl.getIROrder(), VTs, VT, MMO, IndexType));
11339 ID.AddInteger(MMO->getPointerInfo().getAddrSpace());
11340 ID.AddInteger(MMO->getFlags());
11341 FoldingSetInsertToken InsertToken;
11342 if (SDNode *E = lookupNode(ID, dl, InsertToken)) {
11343 cast<VPGatherSDNode>(E)->refineAlignment(MMO);
11344 return SDValue(E, 0);
11345 }
11346
11347 auto *N = newSDNode<VPGatherSDNode>(dl.getIROrder(), dl.getDebugLoc(), VTs,
11348 VT, MMO, IndexType);
11349 createOperands(N, Ops);
11350
11351 assert(N->getMask().getValueType().getVectorElementCount() ==
11352 N->getValueType(0).getVectorElementCount() &&
11353 "Vector width mismatch between mask and data");
11354 assert(N->getIndex().getValueType().getVectorElementCount().isScalable() ==
11355 N->getValueType(0).getVectorElementCount().isScalable() &&
11356 "Scalable flags of index and data do not match");
11358 N->getIndex().getValueType().getVectorElementCount(),
11359 N->getValueType(0).getVectorElementCount()) &&
11360 "Vector width mismatch between index and data");
11361 assert(isa<ConstantSDNode>(N->getScale()) &&
11362 N->getScale()->getAsAPIntVal().isPowerOf2() &&
11363 "Scale should be a constant power of 2");
11364
11365 CSEMap.insert(N, InsertToken);
11366 InsertNode(N);
11367 SDValue V(N, 0);
11368 NewSDValueDbgMsg(V, "Creating new node: ", this);
11369 return V;
11370}
11371
11374 MachineMemOperand *MMO,
11375 ISD::MemIndexType IndexType) {
11376 assert(Ops.size() == 7 && "Incompatible number of operands");
11377
11378 SDNodeKey ID(ISD::VP_SCATTER, VTs, Ops);
11379 ID.AddInteger(VT.getRawBits());
11380 ID.AddInteger(getSyntheticNodeSubclassData<VPScatterSDNode>(
11381 dl.getIROrder(), VTs, VT, MMO, IndexType));
11382 ID.AddInteger(MMO->getPointerInfo().getAddrSpace());
11383 ID.AddInteger(MMO->getFlags());
11384 FoldingSetInsertToken InsertToken;
11385 if (SDNode *E = lookupNode(ID, dl, InsertToken)) {
11386 cast<VPScatterSDNode>(E)->refineAlignment(MMO);
11387 return SDValue(E, 0);
11388 }
11389 auto *N = newSDNode<VPScatterSDNode>(dl.getIROrder(), dl.getDebugLoc(), VTs,
11390 VT, MMO, IndexType);
11391 createOperands(N, Ops);
11392
11393 assert(N->getMask().getValueType().getVectorElementCount() ==
11394 N->getValue().getValueType().getVectorElementCount() &&
11395 "Vector width mismatch between mask and data");
11396 assert(
11397 N->getIndex().getValueType().getVectorElementCount().isScalable() ==
11398 N->getValue().getValueType().getVectorElementCount().isScalable() &&
11399 "Scalable flags of index and data do not match");
11401 N->getIndex().getValueType().getVectorElementCount(),
11402 N->getValue().getValueType().getVectorElementCount()) &&
11403 "Vector width mismatch between index and data");
11404 assert(isa<ConstantSDNode>(N->getScale()) &&
11405 N->getScale()->getAsAPIntVal().isPowerOf2() &&
11406 "Scale should be a constant power of 2");
11407
11408 CSEMap.insert(N, InsertToken);
11409 InsertNode(N);
11410 SDValue V(N, 0);
11411 NewSDValueDbgMsg(V, "Creating new node: ", this);
11412 return V;
11413}
11414
11417 SDValue PassThru, EVT MemVT,
11418 MachineMemOperand *MMO,
11420 ISD::LoadExtType ExtTy, bool isExpanding) {
11421 bool Indexed = AM != ISD::UNINDEXED;
11422 assert((Indexed || Offset.getOpcode() == ISD::POISON) &&
11423 "Unindexed masked load with an offset!");
11424 SDVTList VTs = Indexed ? getVTList(VT, Base.getValueType(), MVT::Other)
11425 : getVTList(VT, MVT::Other);
11426 SDValue Ops[] = {Chain, Base, Offset, Mask, PassThru};
11427 SDNodeKey ID(ISD::MLOAD, VTs, Ops);
11428 ID.AddInteger(MemVT.getRawBits());
11429 ID.AddInteger(getSyntheticNodeSubclassData<MaskedLoadSDNode>(
11430 dl.getIROrder(), VTs, AM, ExtTy, isExpanding, MemVT, MMO));
11431 ID.AddInteger(MMO->getPointerInfo().getAddrSpace());
11432 ID.AddInteger(MMO->getFlags());
11433 FoldingSetInsertToken InsertToken;
11434 if (SDNode *E = lookupNode(ID, dl, InsertToken)) {
11435 cast<MaskedLoadSDNode>(E)->refineAlignment(MMO);
11436 return SDValue(E, 0);
11437 }
11438 auto *N = newSDNode<MaskedLoadSDNode>(dl.getIROrder(), dl.getDebugLoc(), VTs,
11439 AM, ExtTy, isExpanding, MemVT, MMO);
11440 createOperands(N, Ops);
11441
11442 CSEMap.insert(N, InsertToken);
11443 InsertNode(N);
11444 SDValue V(N, 0);
11445 NewSDValueDbgMsg(V, "Creating new node: ", this);
11446 return V;
11447}
11448
11453 assert(LD->getOffset().getOpcode() == ISD::POISON &&
11454 "Masked load is already a indexed load!");
11455 return getMaskedLoad(OrigLoad.getValueType(), dl, LD->getChain(), Base,
11456 Offset, LD->getMask(), LD->getPassThru(),
11457 LD->getMemoryVT(), LD->getMemOperand(), AM,
11458 LD->getExtensionType(), LD->isExpandingLoad());
11459}
11460
11463 SDValue Mask, EVT MemVT,
11464 MachineMemOperand *MMO,
11465 ISD::MemIndexedMode AM, bool IsTruncating,
11466 bool IsCompressing) {
11467 assert(Chain.getValueType() == MVT::Other &&
11468 "Invalid chain type");
11469 bool Indexed = AM != ISD::UNINDEXED;
11470 assert((Indexed || Offset.getOpcode() == ISD::POISON) &&
11471 "Unindexed masked store with an offset!");
11472 SDVTList VTs = Indexed ? getVTList(Base.getValueType(), MVT::Other)
11473 : getVTList(MVT::Other);
11474 SDValue Ops[] = {Chain, Val, Base, Offset, Mask};
11475 SDNodeKey ID(ISD::MSTORE, VTs, Ops);
11476 ID.AddInteger(MemVT.getRawBits());
11477 ID.AddInteger(getSyntheticNodeSubclassData<MaskedStoreSDNode>(
11478 dl.getIROrder(), VTs, AM, IsTruncating, IsCompressing, MemVT, MMO));
11479 ID.AddInteger(MMO->getPointerInfo().getAddrSpace());
11480 ID.AddInteger(MMO->getFlags());
11481 FoldingSetInsertToken InsertToken;
11482 if (SDNode *E = lookupNode(ID, dl, InsertToken)) {
11483 cast<MaskedStoreSDNode>(E)->refineAlignment(MMO);
11484 return SDValue(E, 0);
11485 }
11486 auto *N =
11487 newSDNode<MaskedStoreSDNode>(dl.getIROrder(), dl.getDebugLoc(), VTs, AM,
11488 IsTruncating, IsCompressing, MemVT, MMO);
11489 createOperands(N, Ops);
11490
11491 CSEMap.insert(N, InsertToken);
11492 InsertNode(N);
11493 SDValue V(N, 0);
11494 NewSDValueDbgMsg(V, "Creating new node: ", this);
11495 return V;
11496}
11497
11502 assert(ST->getOffset().getOpcode() == ISD::POISON &&
11503 "Masked store is already a indexed store!");
11504 return getMaskedStore(ST->getChain(), dl, ST->getValue(), Base, Offset,
11505 ST->getMask(), ST->getMemoryVT(), ST->getMemOperand(),
11506 AM, ST->isTruncatingStore(), ST->isCompressingStore());
11507}
11508
11511 MachineMemOperand *MMO,
11512 ISD::MemIndexType IndexType,
11513 ISD::LoadExtType ExtTy) {
11514 assert(Ops.size() == 6 && "Incompatible number of operands");
11515
11516 SDNodeKey ID(ISD::MGATHER, VTs, Ops);
11517 ID.AddInteger(MemVT.getRawBits());
11518 ID.AddInteger(getSyntheticNodeSubclassData<MaskedGatherSDNode>(
11519 dl.getIROrder(), VTs, MemVT, MMO, IndexType, ExtTy));
11520 ID.AddInteger(MMO->getPointerInfo().getAddrSpace());
11521 ID.AddInteger(MMO->getFlags());
11522 FoldingSetInsertToken InsertToken;
11523 if (SDNode *E = lookupNode(ID, dl, InsertToken)) {
11524 cast<MaskedGatherSDNode>(E)->refineAlignment(MMO);
11525 return SDValue(E, 0);
11526 }
11527
11528 auto *N = newSDNode<MaskedGatherSDNode>(dl.getIROrder(), dl.getDebugLoc(),
11529 VTs, MemVT, MMO, IndexType, ExtTy);
11530 createOperands(N, Ops);
11531
11532 assert(N->getPassThru().getValueType() == N->getValueType(0) &&
11533 "Incompatible type of the PassThru value in MaskedGatherSDNode");
11534 assert(N->getMask().getValueType().getVectorElementCount() ==
11535 N->getValueType(0).getVectorElementCount() &&
11536 "Vector width mismatch between mask and data");
11537 assert(N->getIndex().getValueType().getVectorElementCount().isScalable() ==
11538 N->getValueType(0).getVectorElementCount().isScalable() &&
11539 "Scalable flags of index and data do not match");
11541 N->getIndex().getValueType().getVectorElementCount(),
11542 N->getValueType(0).getVectorElementCount()) &&
11543 "Vector width mismatch between index and data");
11544 assert(isa<ConstantSDNode>(N->getScale()) &&
11545 N->getScale()->getAsAPIntVal().isPowerOf2() &&
11546 "Scale should be a constant power of 2");
11547
11548 CSEMap.insert(N, InsertToken);
11549 InsertNode(N);
11550 SDValue V(N, 0);
11551 NewSDValueDbgMsg(V, "Creating new node: ", this);
11552 return V;
11553}
11554
11557 MachineMemOperand *MMO,
11558 ISD::MemIndexType IndexType,
11559 bool IsTrunc) {
11560 assert(Ops.size() == 6 && "Incompatible number of operands");
11561
11562 SDNodeKey ID(ISD::MSCATTER, VTs, Ops);
11563 ID.AddInteger(MemVT.getRawBits());
11564 ID.AddInteger(getSyntheticNodeSubclassData<MaskedScatterSDNode>(
11565 dl.getIROrder(), VTs, MemVT, MMO, IndexType, IsTrunc));
11566 ID.AddInteger(MMO->getPointerInfo().getAddrSpace());
11567 ID.AddInteger(MMO->getFlags());
11568 FoldingSetInsertToken InsertToken;
11569 if (SDNode *E = lookupNode(ID, dl, InsertToken)) {
11570 cast<MaskedScatterSDNode>(E)->refineAlignment(MMO);
11571 return SDValue(E, 0);
11572 }
11573
11574 auto *N = newSDNode<MaskedScatterSDNode>(dl.getIROrder(), dl.getDebugLoc(),
11575 VTs, MemVT, MMO, IndexType, IsTrunc);
11576 createOperands(N, Ops);
11577
11578 assert(N->getMask().getValueType().getVectorElementCount() ==
11579 N->getValue().getValueType().getVectorElementCount() &&
11580 "Vector width mismatch between mask and data");
11581 assert(
11582 N->getIndex().getValueType().getVectorElementCount().isScalable() ==
11583 N->getValue().getValueType().getVectorElementCount().isScalable() &&
11584 "Scalable flags of index and data do not match");
11586 N->getIndex().getValueType().getVectorElementCount(),
11587 N->getValue().getValueType().getVectorElementCount()) &&
11588 "Vector width mismatch between index and data");
11589 assert(isa<ConstantSDNode>(N->getScale()) &&
11590 N->getScale()->getAsAPIntVal().isPowerOf2() &&
11591 "Scale should be a constant power of 2");
11592
11593 CSEMap.insert(N, InsertToken);
11594 InsertNode(N);
11595 SDValue V(N, 0);
11596 NewSDValueDbgMsg(V, "Creating new node: ", this);
11597 return V;
11598}
11599
11601 const SDLoc &dl, ArrayRef<SDValue> Ops,
11602 MachineMemOperand *MMO,
11603 ISD::MemIndexType IndexType) {
11604 assert(Ops.size() == 7 && "Incompatible number of operands");
11605
11607 ID.AddInteger(MemVT.getRawBits());
11608 ID.AddInteger(getSyntheticNodeSubclassData<MaskedHistogramSDNode>(
11609 dl.getIROrder(), VTs, MemVT, MMO, IndexType));
11610 ID.AddInteger(MMO->getPointerInfo().getAddrSpace());
11611 ID.AddInteger(MMO->getFlags());
11612 FoldingSetInsertToken InsertToken;
11613 if (SDNode *E = lookupNode(ID, dl, InsertToken)) {
11614 cast<MaskedGatherSDNode>(E)->refineAlignment(MMO);
11615 return SDValue(E, 0);
11616 }
11617
11618 auto *N = newSDNode<MaskedHistogramSDNode>(dl.getIROrder(), dl.getDebugLoc(),
11619 VTs, MemVT, MMO, IndexType);
11620 createOperands(N, Ops);
11621
11622 assert(N->getMask().getValueType().getVectorElementCount() ==
11623 N->getIndex().getValueType().getVectorElementCount() &&
11624 "Vector width mismatch between mask and data");
11625 assert(isa<ConstantSDNode>(N->getScale()) &&
11626 N->getScale()->getAsAPIntVal().isPowerOf2() &&
11627 "Scale should be a constant power of 2");
11628 assert(N->getInc().getValueType().isInteger() && "Non integer update value");
11629
11630 CSEMap.insert(N, InsertToken);
11631 InsertNode(N);
11632 SDValue V(N, 0);
11633 NewSDValueDbgMsg(V, "Creating new node: ", this);
11634 return V;
11635}
11636
11638 SDValue Ptr, SDValue Mask, SDValue EVL,
11639 MachineMemOperand *MMO) {
11640 SDVTList VTs = getVTList(VT, EVL.getValueType(), MVT::Other);
11641 SDValue Ops[] = {Chain, Ptr, Mask, EVL};
11642 SDNodeKey ID(ISD::VP_LOAD_FF, VTs, Ops);
11643 ID.AddInteger(VT.getRawBits());
11644 ID.AddInteger(getSyntheticNodeSubclassData<VPLoadFFSDNode>(DL.getIROrder(),
11645 VTs, VT, MMO));
11646 ID.AddInteger(MMO->getPointerInfo().getAddrSpace());
11647 ID.AddInteger(MMO->getFlags());
11648 FoldingSetInsertToken InsertToken;
11649 if (SDNode *E = lookupNode(ID, DL, InsertToken)) {
11650 cast<VPLoadFFSDNode>(E)->refineAlignment(MMO);
11651 return SDValue(E, 0);
11652 }
11653 auto *N = newSDNode<VPLoadFFSDNode>(DL.getIROrder(), DL.getDebugLoc(), VTs,
11654 VT, MMO);
11655 createOperands(N, Ops);
11656
11657 CSEMap.insert(N, InsertToken);
11658 InsertNode(N);
11659 SDValue V(N, 0);
11660 NewSDValueDbgMsg(V, "Creating new node: ", this);
11661 return V;
11662}
11663
11665 EVT MemVT, MachineMemOperand *MMO) {
11666 assert(Chain.getValueType() == MVT::Other && "Invalid chain type");
11667 SDVTList VTs = getVTList(MVT::Other);
11668 SDValue Ops[] = {Chain, Ptr};
11670 ID.AddInteger(MemVT.getRawBits());
11671 ID.AddInteger(getSyntheticNodeSubclassData<FPStateAccessSDNode>(
11672 ISD::GET_FPENV_MEM, dl.getIROrder(), VTs, MemVT, MMO));
11673 ID.AddInteger(MMO->getPointerInfo().getAddrSpace());
11674 ID.AddInteger(MMO->getFlags());
11675 FoldingSetInsertToken InsertToken;
11676 if (SDNode *E = lookupNode(ID, dl, InsertToken))
11677 return SDValue(E, 0);
11678
11679 auto *N = newSDNode<FPStateAccessSDNode>(ISD::GET_FPENV_MEM, dl.getIROrder(),
11680 dl.getDebugLoc(), VTs, MemVT, MMO);
11681 createOperands(N, Ops);
11682
11683 CSEMap.insert(N, InsertToken);
11684 InsertNode(N);
11685 SDValue V(N, 0);
11686 NewSDValueDbgMsg(V, "Creating new node: ", this);
11687 return V;
11688}
11689
11691 EVT MemVT, MachineMemOperand *MMO) {
11692 assert(Chain.getValueType() == MVT::Other && "Invalid chain type");
11693 SDVTList VTs = getVTList(MVT::Other);
11694 SDValue Ops[] = {Chain, Ptr};
11696 ID.AddInteger(MemVT.getRawBits());
11697 ID.AddInteger(getSyntheticNodeSubclassData<FPStateAccessSDNode>(
11698 ISD::SET_FPENV_MEM, dl.getIROrder(), VTs, MemVT, MMO));
11699 ID.AddInteger(MMO->getPointerInfo().getAddrSpace());
11700 ID.AddInteger(MMO->getFlags());
11701 FoldingSetInsertToken InsertToken;
11702 if (SDNode *E = lookupNode(ID, dl, InsertToken))
11703 return SDValue(E, 0);
11704
11705 auto *N = newSDNode<FPStateAccessSDNode>(ISD::SET_FPENV_MEM, dl.getIROrder(),
11706 dl.getDebugLoc(), VTs, MemVT, MMO);
11707 createOperands(N, Ops);
11708
11709 CSEMap.insert(N, InsertToken);
11710 InsertNode(N);
11711 SDValue V(N, 0);
11712 NewSDValueDbgMsg(V, "Creating new node: ", this);
11713 return V;
11714}
11715
11717 // select undef, T, F --> T (if T is a constant), otherwise F
11718 // select, ?, undef, F --> F
11719 // select, ?, T, undef --> T
11720 if (Cond.isUndef())
11721 return isConstantValueOfAnyType(T) ? T : F;
11722 if (T.isUndef())
11724 if (F.isUndef())
11726
11727 // select true, T, F --> T
11728 // select false, T, F --> F
11729 if (auto C = isBoolConstant(Cond))
11730 return *C ? T : F;
11731
11732 // select ?, T, T --> T
11733 if (T == F)
11734 return T;
11735
11736 return SDValue();
11737}
11738
11740 // shift undef, Y --> 0 (can always assume that the undef value is 0)
11741 if (X.isUndef())
11742 return getConstant(0, SDLoc(X.getNode()), X.getValueType());
11743 // shift X, undef --> undef (because it may shift by the bitwidth)
11744 if (Y.isUndef())
11745 return getUNDEF(X.getValueType());
11746
11747 // shift 0, Y --> 0
11748 // shift X, 0 --> X
11750 return X;
11751
11752 // shift X, C >= bitwidth(X) --> undef
11753 // All vector elements must be too big (or undef) to avoid partial undefs.
11754 auto isShiftTooBig = [X](ConstantSDNode *Val) {
11755 return !Val || Val->getAPIntValue().uge(X.getScalarValueSizeInBits());
11756 };
11757 if (ISD::matchUnaryPredicate(Y, isShiftTooBig, true))
11758 return getUNDEF(X.getValueType());
11759
11760 // shift i1/vXi1 X, Y --> X (any non-zero shift amount is undefined).
11761 if (X.getValueType().getScalarType() == MVT::i1)
11762 return X;
11763
11764 return SDValue();
11765}
11766
11768 SDNodeFlags Flags) {
11769 // If this operation has 'nnan' or 'ninf' and at least 1 disallowed operand
11770 // (an undef operand can be chosen to be Nan/Inf), then the result of this
11771 // operation is poison. That result can be relaxed to undef.
11772 ConstantFPSDNode *XC = isConstOrConstSplatFP(X, /* AllowUndefs */ true);
11773 ConstantFPSDNode *YC = isConstOrConstSplatFP(Y, /* AllowUndefs */ true);
11774 bool HasNan = (XC && XC->getValueAPF().isNaN()) ||
11775 (YC && YC->getValueAPF().isNaN());
11776 bool HasInf = (XC && XC->getValueAPF().isInfinity()) ||
11777 (YC && YC->getValueAPF().isInfinity());
11778
11779 if (Flags.hasNoNaNs() && (HasNan || X.isUndef() || Y.isUndef()))
11780 return getUNDEF(X.getValueType());
11781
11782 if (Flags.hasNoInfs() && (HasInf || X.isUndef() || Y.isUndef()))
11783 return getUNDEF(X.getValueType());
11784
11785 if (!YC)
11786 return SDValue();
11787
11788 // X + -0.0 --> X
11789 if (Opcode == ISD::FADD)
11790 if (YC->getValueAPF().isNegZero())
11791 return X;
11792
11793 // X - +0.0 --> X
11794 if (Opcode == ISD::FSUB)
11795 if (YC->getValueAPF().isPosZero())
11796 return X;
11797
11798 // X * 1.0 --> X
11799 // X / 1.0 --> X
11800 if (Opcode == ISD::FMUL || Opcode == ISD::FDIV)
11801 if (YC->getValueAPF().isOne())
11802 return X;
11803
11804 // X * 0.0 --> 0.0
11805 if (Opcode == ISD::FMUL && Flags.hasNoNaNs() && Flags.hasNoSignedZeros())
11806 if (YC->getValueAPF().isZero())
11807 return getConstantFP(0.0, SDLoc(Y), Y.getValueType());
11808
11809 return SDValue();
11810}
11811
11813 SDValue Ptr, SDValue SV, unsigned Align) {
11814 SDValue Ops[] = { Chain, Ptr, SV, getTargetConstant(Align, dl, MVT::i32) };
11815 return getNode(ISD::VAARG, dl, getVTList(VT, MVT::Other), Ops);
11816}
11817
11818SDValue SelectionDAG::getNode(unsigned Opcode, const SDLoc &DL, EVT VT,
11820 switch (Ops.size()) {
11821 case 0: return getNode(Opcode, DL, VT);
11822 case 1: return getNode(Opcode, DL, VT, Ops[0].get());
11823 case 2: return getNode(Opcode, DL, VT, Ops[0], Ops[1]);
11824 case 3: return getNode(Opcode, DL, VT, Ops[0], Ops[1], Ops[2]);
11825 default: break;
11826 }
11827
11828 // Copy from an SDUse array into an SDValue array for use with
11829 // the regular getNode logic.
11831 return getNode(Opcode, DL, VT, NewOps);
11832}
11833
11834SDValue SelectionDAG::getNode(unsigned Opcode, const SDLoc &DL, EVT VT,
11836 SDNodeFlags Flags;
11837 if (Inserter)
11838 Flags = Inserter->getFlags();
11839 return getNode(Opcode, DL, VT, Ops, Flags);
11840}
11841
11842SDValue SelectionDAG::getNode(unsigned Opcode, const SDLoc &DL, EVT VT,
11843 ArrayRef<SDValue> Ops, const SDNodeFlags Flags) {
11844 unsigned NumOps = Ops.size();
11845 switch (NumOps) {
11846 case 0: return getNode(Opcode, DL, VT);
11847 case 1: return getNode(Opcode, DL, VT, Ops[0], Flags);
11848 case 2: return getNode(Opcode, DL, VT, Ops[0], Ops[1], Flags);
11849 case 3: return getNode(Opcode, DL, VT, Ops[0], Ops[1], Ops[2], Flags);
11850 default: break;
11851 }
11852
11853#ifndef NDEBUG
11854 for (const auto &Op : Ops)
11855 assert(Op.getOpcode() != ISD::DELETED_NODE &&
11856 "Operand is DELETED_NODE!");
11857#endif
11858
11859 switch (Opcode) {
11860 default: break;
11861 case ISD::BUILD_VECTOR:
11862 // Attempt to simplify BUILD_VECTOR.
11863 if (SDValue V = FoldBUILD_VECTOR(DL, VT, Ops, *this))
11864 return V;
11865 break;
11867 if (SDValue V = foldCONCAT_VECTORS(DL, VT, Ops, *this))
11868 return V;
11869 break;
11870 case ISD::SELECT_CC:
11871 assert(NumOps == 5 && "SELECT_CC takes 5 operands!");
11872 assert(Ops[0].getValueType() == Ops[1].getValueType() &&
11873 "LHS and RHS of condition must have same type!");
11874 assert(Ops[2].getValueType() == Ops[3].getValueType() &&
11875 "True and False arms of SelectCC must have same type!");
11876 assert(Ops[2].getValueType() == VT &&
11877 "select_cc node must be of same type as true and false value!");
11878 assert((!Ops[0].getValueType().isVector() ||
11879 Ops[0].getValueType().getVectorElementCount() ==
11880 VT.getVectorElementCount()) &&
11881 "Expected select_cc with vector result to have the same sized "
11882 "comparison type!");
11883 break;
11884 case ISD::BR_CC:
11885 assert(NumOps == 5 && "BR_CC takes 5 operands!");
11886 assert(Ops[2].getValueType() == Ops[3].getValueType() &&
11887 "LHS/RHS of comparison should match types!");
11888 break;
11889 case ISD::VP_REDUCE_MUL:
11890 // If it is VP_REDUCE_MUL mask operation then turn it to VP_REDUCE_AND
11891 if (VT == MVT::i1)
11892 Opcode = ISD::VP_REDUCE_AND;
11893 break;
11894 case ISD::VP_REDUCE_ADD:
11895 // If it is VP_REDUCE_ADD mask operation then turn it to VP_REDUCE_XOR
11896 if (VT == MVT::i1)
11897 Opcode = ISD::VP_REDUCE_XOR;
11898 break;
11899 case ISD::VP_REDUCE_SMAX:
11900 case ISD::VP_REDUCE_UMIN:
11901 // If it is VP_REDUCE_SMAX/VP_REDUCE_UMIN mask operation then turn it to
11902 // VP_REDUCE_AND.
11903 if (VT == MVT::i1)
11904 Opcode = ISD::VP_REDUCE_AND;
11905 break;
11906 case ISD::VP_REDUCE_SMIN:
11907 case ISD::VP_REDUCE_UMAX:
11908 // If it is VP_REDUCE_SMIN/VP_REDUCE_UMAX mask operation then turn it to
11909 // VP_REDUCE_OR.
11910 if (VT == MVT::i1)
11911 Opcode = ISD::VP_REDUCE_OR;
11912 break;
11913 }
11914
11915 // Memoize nodes.
11916 SDNode *N;
11917 SDVTList VTs = getVTList(VT);
11918
11919 if (VT != MVT::Glue) {
11920 SDNodeKey ID(Opcode, VTs, Ops);
11921 FoldingSetInsertToken InsertToken;
11922
11923 if (SDNode *E = lookupNode(ID, DL, InsertToken)) {
11924 E->intersectFlagsWith(Flags);
11925 return SDValue(E, 0);
11926 }
11927
11928 N = newSDNode<SDNode>(Opcode, DL.getIROrder(), DL.getDebugLoc(), VTs);
11929 createOperands(N, Ops);
11930
11931 CSEMap.insert(N, InsertToken);
11932 } else {
11933 N = newSDNode<SDNode>(Opcode, DL.getIROrder(), DL.getDebugLoc(), VTs);
11934 createOperands(N, Ops);
11935 }
11936
11937 N->setFlags(Flags);
11938 InsertNode(N);
11939 SDValue V(N, 0);
11940 NewSDValueDbgMsg(V, "Creating new node: ", this);
11941 return V;
11942}
11943
11944SDValue SelectionDAG::getNode(unsigned Opcode, const SDLoc &DL,
11945 ArrayRef<EVT> ResultTys, ArrayRef<SDValue> Ops) {
11946 SDNodeFlags Flags;
11947 if (Inserter)
11948 Flags = Inserter->getFlags();
11949 return getNode(Opcode, DL, getVTList(ResultTys), Ops, Flags);
11950}
11951
11952SDValue SelectionDAG::getNode(unsigned Opcode, const SDLoc &DL,
11954 const SDNodeFlags Flags) {
11955 return getNode(Opcode, DL, getVTList(ResultTys), Ops, Flags);
11956}
11957
11958SDValue SelectionDAG::getNode(unsigned Opcode, const SDLoc &DL, SDVTList VTList,
11960 SDNodeFlags Flags;
11961 if (Inserter)
11962 Flags = Inserter->getFlags();
11963 return getNode(Opcode, DL, VTList, Ops, Flags);
11964}
11965
11966SDValue SelectionDAG::getNode(unsigned Opcode, const SDLoc &DL, SDVTList VTList,
11967 ArrayRef<SDValue> Ops, const SDNodeFlags Flags) {
11968 if (VTList.NumVTs == 1)
11969 return getNode(Opcode, DL, VTList.VTs[0], Ops, Flags);
11970
11971#ifndef NDEBUG
11972 for (const auto &Op : Ops)
11973 assert(Op.getOpcode() != ISD::DELETED_NODE &&
11974 "Operand is DELETED_NODE!");
11975#endif
11976
11977 switch (Opcode) {
11978 case ISD::SADDO:
11979 case ISD::UADDO:
11980 case ISD::SSUBO:
11981 case ISD::USUBO: {
11982 assert(VTList.NumVTs == 2 && Ops.size() == 2 &&
11983 "Invalid add/sub overflow op!");
11984 assert(VTList.VTs[0].isInteger() && VTList.VTs[1].isInteger() &&
11985 Ops[0].getValueType() == Ops[1].getValueType() &&
11986 Ops[0].getValueType() == VTList.VTs[0] &&
11987 "Binary operator types must match!");
11988 SDValue N1 = Ops[0], N2 = Ops[1];
11989 canonicalizeCommutativeBinop(Opcode, N1, N2);
11990
11991 // (X +- 0) -> X with zero-overflow.
11992 ConstantSDNode *N2CV = isConstOrConstSplat(N2, /*AllowUndefs*/ false,
11993 /*AllowTruncation*/ true);
11994 if (N2CV && N2CV->isZero()) {
11995 SDValue ZeroOverFlow = getConstant(0, DL, VTList.VTs[1]);
11996 return getNode(ISD::MERGE_VALUES, DL, VTList, {N1, ZeroOverFlow}, Flags);
11997 }
11998
11999 if (VTList.VTs[0].getScalarType() == MVT::i1 &&
12000 VTList.VTs[1].getScalarType() == MVT::i1) {
12001 SDValue F1 = getFreeze(N1);
12002 SDValue F2 = getFreeze(N2);
12003 // {vXi1,vXi1} (u/s)addo(vXi1 x, vXi1y) -> {xor(x,y),and(x,y)}
12004 if (Opcode == ISD::UADDO || Opcode == ISD::SADDO)
12005 return getNode(ISD::MERGE_VALUES, DL, VTList,
12006 {getNode(ISD::XOR, DL, VTList.VTs[0], F1, F2),
12007 getNode(ISD::AND, DL, VTList.VTs[1], F1, F2)},
12008 Flags);
12009 // {vXi1,vXi1} (u/s)subo(vXi1 x, vXi1y) -> {xor(x,y),and(~x,y)}
12010 if (Opcode == ISD::USUBO || Opcode == ISD::SSUBO) {
12011 SDValue NotF1 = getNOT(DL, F1, VTList.VTs[0]);
12012 return getNode(ISD::MERGE_VALUES, DL, VTList,
12013 {getNode(ISD::XOR, DL, VTList.VTs[0], F1, F2),
12014 getNode(ISD::AND, DL, VTList.VTs[1], NotF1, F2)},
12015 Flags);
12016 }
12017 }
12018 break;
12019 }
12020 case ISD::SADDO_CARRY:
12021 case ISD::UADDO_CARRY:
12022 case ISD::SSUBO_CARRY:
12023 case ISD::USUBO_CARRY:
12024 assert(VTList.NumVTs == 2 && Ops.size() == 3 &&
12025 "Invalid add/sub overflow op!");
12026 assert(VTList.VTs[0].isInteger() && VTList.VTs[1].isInteger() &&
12027 Ops[0].getValueType() == Ops[1].getValueType() &&
12028 Ops[0].getValueType() == VTList.VTs[0] &&
12029 Ops[2].getValueType() == VTList.VTs[1] &&
12030 "Binary operator types must match!");
12031 break;
12032 case ISD::SMUL_LOHI:
12033 case ISD::UMUL_LOHI: {
12034 assert(VTList.NumVTs == 2 && Ops.size() == 2 && "Invalid mul lo/hi op!");
12035 assert(VTList.VTs[0].isInteger() && VTList.VTs[0] == VTList.VTs[1] &&
12036 VTList.VTs[0] == Ops[0].getValueType() &&
12037 VTList.VTs[0] == Ops[1].getValueType() &&
12038 "Binary operator types must match!");
12039 // Constant fold.
12042 if (LHS && RHS) {
12043 unsigned Width = VTList.VTs[0].getScalarSizeInBits();
12044 unsigned OutWidth = Width * 2;
12045 APInt Val = LHS->getAPIntValue();
12046 APInt Mul = RHS->getAPIntValue();
12047 if (Opcode == ISD::SMUL_LOHI) {
12048 Val = Val.sext(OutWidth);
12049 Mul = Mul.sext(OutWidth);
12050 } else {
12051 Val = Val.zext(OutWidth);
12052 Mul = Mul.zext(OutWidth);
12053 }
12054 Val *= Mul;
12055
12056 SDValue Hi =
12057 getConstant(Val.extractBits(Width, Width), DL, VTList.VTs[0]);
12058 SDValue Lo = getConstant(Val.trunc(Width), DL, VTList.VTs[0]);
12059 return getNode(ISD::MERGE_VALUES, DL, VTList, {Lo, Hi}, Flags);
12060 }
12061 break;
12062 }
12063 case ISD::FFREXP: {
12064 assert(VTList.NumVTs == 2 && Ops.size() == 1 && "Invalid ffrexp op!");
12065 assert(VTList.VTs[0].isFloatingPoint() && VTList.VTs[1].isInteger() &&
12066 VTList.VTs[0] == Ops[0].getValueType() && "frexp type mismatch");
12067
12069 int FrexpExp;
12070 APFloat FrexpMant =
12071 frexp(C->getValueAPF(), FrexpExp, APFloat::rmNearestTiesToEven);
12072 SDValue Result0 = getConstantFP(FrexpMant, DL, VTList.VTs[0]);
12073 SDValue Result1 = getSignedConstant(FrexpMant.isFinite() ? FrexpExp : 0,
12074 DL, VTList.VTs[1]);
12075 return getNode(ISD::MERGE_VALUES, DL, VTList, {Result0, Result1}, Flags);
12076 }
12077
12078 break;
12079 }
12081 assert(VTList.NumVTs == 2 && Ops.size() == 2 &&
12082 "Invalid STRICT_FP_EXTEND!");
12083 assert(VTList.VTs[0].isFloatingPoint() &&
12084 Ops[1].getValueType().isFloatingPoint() && "Invalid FP cast!");
12085 assert(VTList.VTs[0].isVector() == Ops[1].getValueType().isVector() &&
12086 "STRICT_FP_EXTEND result type should be vector iff the operand "
12087 "type is vector!");
12088 assert((!VTList.VTs[0].isVector() ||
12089 VTList.VTs[0].getVectorElementCount() ==
12090 Ops[1].getValueType().getVectorElementCount()) &&
12091 "Vector element count mismatch!");
12092 assert(Ops[1].getValueType().bitsLT(VTList.VTs[0]) &&
12093 "Invalid fpext node, dst <= src!");
12094 break;
12096 assert(VTList.NumVTs == 2 && Ops.size() == 3 && "Invalid STRICT_FP_ROUND!");
12097 assert(VTList.VTs[0].isVector() == Ops[1].getValueType().isVector() &&
12098 "STRICT_FP_ROUND result type should be vector iff the operand "
12099 "type is vector!");
12100 assert((!VTList.VTs[0].isVector() ||
12101 VTList.VTs[0].getVectorElementCount() ==
12102 Ops[1].getValueType().getVectorElementCount()) &&
12103 "Vector element count mismatch!");
12104 assert(VTList.VTs[0].isFloatingPoint() &&
12105 Ops[1].getValueType().isFloatingPoint() &&
12106 VTList.VTs[0].bitsLT(Ops[1].getValueType()) &&
12107 Ops[2].getOpcode() == ISD::TargetConstant &&
12108 (Ops[2]->getAsZExtVal() == 0 || Ops[2]->getAsZExtVal() == 1) &&
12109 "Invalid STRICT_FP_ROUND!");
12110 break;
12111 }
12112
12113 // Memoize the node unless it returns a glue result.
12114 SDNode *N;
12115 if (VTList.VTs[VTList.NumVTs-1] != MVT::Glue) {
12116 SDNodeKey ID(Opcode, VTList, Ops);
12117 FoldingSetInsertToken InsertToken;
12118 if (SDNode *E = lookupNode(ID, DL, InsertToken)) {
12119 E->intersectFlagsWith(Flags);
12120 return SDValue(E, 0);
12121 }
12122
12123 N = newSDNode<SDNode>(Opcode, DL.getIROrder(), DL.getDebugLoc(), VTList);
12124 createOperands(N, Ops);
12125 CSEMap.insert(N, InsertToken);
12126 } else {
12127 N = newSDNode<SDNode>(Opcode, DL.getIROrder(), DL.getDebugLoc(), VTList);
12128 createOperands(N, Ops);
12129 }
12130
12131 N->setFlags(Flags);
12132 InsertNode(N);
12133 SDValue V(N, 0);
12134 NewSDValueDbgMsg(V, "Creating new node: ", this);
12135 return V;
12136}
12137
12138SDValue SelectionDAG::getNode(unsigned Opcode, const SDLoc &DL,
12139 SDVTList VTList) {
12140 return getNode(Opcode, DL, VTList, ArrayRef<SDValue>());
12141}
12142
12143SDValue SelectionDAG::getNode(unsigned Opcode, const SDLoc &DL, SDVTList VTList,
12144 SDValue N1) {
12145 SDValue Ops[] = { N1 };
12146 return getNode(Opcode, DL, VTList, Ops);
12147}
12148
12149SDValue SelectionDAG::getNode(unsigned Opcode, const SDLoc &DL, SDVTList VTList,
12150 SDValue N1, SDValue N2) {
12151 SDValue Ops[] = { N1, N2 };
12152 return getNode(Opcode, DL, VTList, Ops);
12153}
12154
12155SDValue SelectionDAG::getNode(unsigned Opcode, const SDLoc &DL, SDVTList VTList,
12156 SDValue N1, SDValue N2, SDValue N3) {
12157 SDValue Ops[] = { N1, N2, N3 };
12158 return getNode(Opcode, DL, VTList, Ops);
12159}
12160
12161SDValue SelectionDAG::getNode(unsigned Opcode, const SDLoc &DL, SDVTList VTList,
12162 SDValue N1, SDValue N2, SDValue N3, SDValue N4) {
12163 SDValue Ops[] = { N1, N2, N3, N4 };
12164 return getNode(Opcode, DL, VTList, Ops);
12165}
12166
12167SDValue SelectionDAG::getNode(unsigned Opcode, const SDLoc &DL, SDVTList VTList,
12168 SDValue N1, SDValue N2, SDValue N3, SDValue N4,
12169 SDValue N5) {
12170 SDValue Ops[] = { N1, N2, N3, N4, N5 };
12171 return getNode(Opcode, DL, VTList, Ops);
12172}
12173
12175 if (!VT.isExtended())
12176 return makeVTList(SDNode::getValueTypeList(VT.getSimpleVT()), 1);
12177
12178 EVT VTs[] = {VT};
12179 return getVTList(VTs);
12180}
12181
12183 EVT VTs[] = {VT1, VT2};
12184 return getVTList(VTs);
12185}
12186
12188 EVT VTs[] = {VT1, VT2, VT3};
12189 return getVTList(VTs);
12190}
12191
12193 EVT VTs[] = {VT1, VT2, VT3, VT4};
12194 return getVTList(VTs);
12195}
12196
12198 auto It = VTLists.find(VTs);
12199 if (It == VTLists.end()) {
12200 EVT *Array = Allocator.Allocate<EVT>(VTs.size());
12201 llvm::copy(VTs, Array);
12202 It = VTLists.insert(ArrayRef(Array, VTs.size())).first;
12203 }
12204 return makeVTList(It->data(), It->size());
12205}
12206
12207/// UpdateNodeOperands - *Mutate* the specified node in-place to have the
12208/// specified operands. If the resultant node already exists in the DAG,
12209/// this does not modify the specified node, instead it returns the node that
12210/// already exists. If the resultant node does not exist in the DAG, the
12211/// input node is returned. As a degenerate case, if you specify the same
12212/// input operands as the node already has, the input node is returned.
12214 assert(N->getNumOperands() == 1 && "Update with wrong number of operands");
12215
12216 // Check to see if there is no change.
12217 if (Op == N->getOperand(0)) return N;
12218
12219 // See if the modified node already exists.
12220 FoldingSetInsertToken InsertToken;
12221 if (SDNode *Existing = FindModifiedNodeSlot(N, Op, InsertToken))
12222 return Existing;
12223
12224 // Nope it doesn't. Remove the node from its current place in the maps.
12225 if (InsertToken)
12226 if (!RemoveNodeFromCSEMaps(N))
12227 InsertToken = {};
12228
12229 // Now we update the operands.
12230 N->OperandList[0].set(Op);
12231
12233 // If this gets put into a CSE map, add it.
12234 if (InsertToken)
12235 CSEMap.insert(N, InsertToken);
12236 return N;
12237}
12238
12240 assert(N->getNumOperands() == 2 && "Update with wrong number of operands");
12241
12242 // Check to see if there is no change.
12243 if (Op1 == N->getOperand(0) && Op2 == N->getOperand(1))
12244 return N; // No operands changed, just return the input node.
12245
12246 // See if the modified node already exists.
12247 FoldingSetInsertToken InsertToken;
12248 if (SDNode *Existing = FindModifiedNodeSlot(N, Op1, Op2, InsertToken))
12249 return Existing;
12250
12251 // Nope it doesn't. Remove the node from its current place in the maps.
12252 if (InsertToken)
12253 if (!RemoveNodeFromCSEMaps(N))
12254 InsertToken = {};
12255
12256 // Now we update the operands.
12257 if (N->OperandList[0] != Op1)
12258 N->OperandList[0].set(Op1);
12259 if (N->OperandList[1] != Op2)
12260 N->OperandList[1].set(Op2);
12261
12263 // If this gets put into a CSE map, add it.
12264 if (InsertToken)
12265 CSEMap.insert(N, InsertToken);
12266 return N;
12267}
12268
12271 SDValue Ops[] = { Op1, Op2, Op3 };
12272 return UpdateNodeOperands(N, Ops);
12273}
12274
12277 SDValue Op3, SDValue Op4) {
12278 SDValue Ops[] = { Op1, Op2, Op3, Op4 };
12279 return UpdateNodeOperands(N, Ops);
12280}
12281
12284 SDValue Op3, SDValue Op4, SDValue Op5) {
12285 SDValue Ops[] = { Op1, Op2, Op3, Op4, Op5 };
12286 return UpdateNodeOperands(N, Ops);
12287}
12288
12291 unsigned NumOps = Ops.size();
12292 assert(N->getNumOperands() == NumOps &&
12293 "Update with wrong number of operands");
12294
12295 // If no operands changed just return the input node.
12296 if (std::equal(Ops.begin(), Ops.end(), N->op_begin()))
12297 return N;
12298
12299 // See if the modified node already exists.
12300 FoldingSetInsertToken InsertToken;
12301 if (SDNode *Existing = FindModifiedNodeSlot(N, Ops, InsertToken))
12302 return Existing;
12303
12304 // Nope it doesn't. Remove the node from its current place in the maps.
12305 if (InsertToken)
12306 if (!RemoveNodeFromCSEMaps(N))
12307 InsertToken = {};
12308
12309 // Now we update the operands.
12310 for (unsigned i = 0; i != NumOps; ++i)
12311 if (N->OperandList[i] != Ops[i])
12312 N->OperandList[i].set(Ops[i]);
12313
12315 // If this gets put into a CSE map, add it.
12316 if (InsertToken)
12317 CSEMap.insert(N, InsertToken);
12318 return N;
12319}
12320
12321/// DropOperands - Release the operands and set this node to have
12322/// zero operands.
12324 // Unlike the code in MorphNodeTo that does this, we don't need to
12325 // watch for dead nodes here.
12326 for (op_iterator I = op_begin(), E = op_end(); I != E; ) {
12327 SDUse &Use = *I++;
12328 Use.set(SDValue());
12329 }
12330}
12331
12333 ArrayRef<MachineMemOperand *> NewMemRefs) {
12334 if (NewMemRefs.empty()) {
12335 N->clearMemRefs();
12336 return;
12337 }
12338
12339 // Check if we can avoid allocating by storing a single reference directly.
12340 if (NewMemRefs.size() == 1) {
12341 N->MemRefs = NewMemRefs[0];
12342 N->NumMemRefs = 1;
12343 return;
12344 }
12345
12346 MachineMemOperand **MemRefsBuffer =
12347 Allocator.template Allocate<MachineMemOperand *>(NewMemRefs.size());
12348 llvm::copy(NewMemRefs, MemRefsBuffer);
12349 N->MemRefs = MemRefsBuffer;
12350 N->NumMemRefs = static_cast<int>(NewMemRefs.size());
12351}
12352
12353/// SelectNodeTo - These are wrappers around MorphNodeTo that accept a
12354/// machine opcode.
12355///
12357 EVT VT) {
12358 SDVTList VTs = getVTList(VT);
12359 return SelectNodeTo(N, MachineOpc, VTs, {});
12360}
12361
12363 EVT VT, SDValue Op1) {
12364 SDVTList VTs = getVTList(VT);
12365 SDValue Ops[] = { Op1 };
12366 return SelectNodeTo(N, MachineOpc, VTs, Ops);
12367}
12368
12370 EVT VT, SDValue Op1,
12371 SDValue Op2) {
12372 SDVTList VTs = getVTList(VT);
12373 SDValue Ops[] = { Op1, Op2 };
12374 return SelectNodeTo(N, MachineOpc, VTs, Ops);
12375}
12376
12378 EVT VT, SDValue Op1,
12379 SDValue Op2, SDValue Op3) {
12380 SDVTList VTs = getVTList(VT);
12381 SDValue Ops[] = { Op1, Op2, Op3 };
12382 return SelectNodeTo(N, MachineOpc, VTs, Ops);
12383}
12384
12387 SDVTList VTs = getVTList(VT);
12388 return SelectNodeTo(N, MachineOpc, VTs, Ops);
12389}
12390
12392 EVT VT1, EVT VT2, ArrayRef<SDValue> Ops) {
12393 SDVTList VTs = getVTList(VT1, VT2);
12394 return SelectNodeTo(N, MachineOpc, VTs, Ops);
12395}
12396
12398 EVT VT1, EVT VT2) {
12399 SDVTList VTs = getVTList(VT1, VT2);
12400 return SelectNodeTo(N, MachineOpc, VTs, {});
12401}
12402
12404 EVT VT1, EVT VT2, EVT VT3,
12406 SDVTList VTs = getVTList(VT1, VT2, VT3);
12407 return SelectNodeTo(N, MachineOpc, VTs, Ops);
12408}
12409
12411 EVT VT1, EVT VT2,
12412 SDValue Op1, SDValue Op2) {
12413 SDVTList VTs = getVTList(VT1, VT2);
12414 SDValue Ops[] = { Op1, Op2 };
12415 return SelectNodeTo(N, MachineOpc, VTs, Ops);
12416}
12417
12420 SDNode *New = MorphNodeTo(N, ~MachineOpc, VTs, Ops);
12421 // Reset the NodeID to -1.
12422 New->setNodeId(-1);
12423 if (New != N) {
12424 ReplaceAllUsesWith(N, New);
12426 }
12427 return New;
12428}
12429
12430/// UpdateSDLocOnMergeSDNode - If the opt level is -O0 then it throws away
12431/// the line number information on the merged node since it is not possible to
12432/// preserve the information that operation is associated with multiple lines.
12433/// This will make the debugger working better at -O0, were there is a higher
12434/// probability having other instructions associated with that line.
12435///
12436/// For IROrder, we keep the smaller of the two
12437SDNode *SelectionDAG::UpdateSDLocOnMergeSDNode(SDNode *N, const SDLoc &OLoc) {
12438 DebugLoc NLoc = N->getDebugLoc();
12439 if (NLoc && OptLevel == CodeGenOptLevel::None && OLoc.getDebugLoc() != NLoc) {
12440 N->setDebugLoc(DebugLoc());
12441 }
12442 unsigned Order = std::min(N->getIROrder(), OLoc.getIROrder());
12443 N->setIROrder(Order);
12444 return N;
12445}
12446
12447/// MorphNodeTo - This *mutates* the specified node to have the specified
12448/// return type, opcode, and operands.
12449///
12450/// Note that MorphNodeTo returns the resultant node. If there is already a
12451/// node of the specified opcode and operands, it returns that node instead of
12452/// the current one. Note that the SDLoc need not be the same.
12453///
12454/// Using MorphNodeTo is faster than creating a new node and swapping it in
12455/// with ReplaceAllUsesWith both because it often avoids allocating a new
12456/// node, and because it doesn't require CSE recalculation for any of
12457/// the node's users.
12458///
12459/// However, note that MorphNodeTo recursively deletes dead nodes from the DAG.
12460/// As a consequence it isn't appropriate to use from within the DAG combiner or
12461/// the legalizer which maintain worklists that would need to be updated when
12462/// deleting things.
12465 // If an identical node already exists, use it.
12466 FoldingSetInsertToken InsertToken;
12467 if (VTs.VTs[VTs.NumVTs-1] != MVT::Glue) {
12468 SDNodeKey ID(Opc, VTs, Ops);
12469 AddNodeIDCustom(ID.Tail, N, Opc);
12470 if (SDNode *ON = lookupNode(ID, SDLoc(N), InsertToken))
12471 return UpdateSDLocOnMergeSDNode(ON, SDLoc(N));
12472 }
12473
12474 if (!RemoveNodeFromCSEMaps(N))
12475 InsertToken = {};
12476
12477 // Start the morphing.
12478 N->NodeType = Opc;
12479 N->ValueList = VTs.VTs;
12480 N->NumValues = VTs.NumVTs;
12481
12482 // Clear the operands list, updating used nodes to remove this from their
12483 // use list. Keep track of any operands that become dead as a result.
12484 SmallPtrSet<SDNode*, 16> DeadNodeSet;
12485 for (SDNode::op_iterator I = N->op_begin(), E = N->op_end(); I != E; ) {
12486 SDUse &Use = *I++;
12487 SDNode *Used = Use.getNode();
12488 Use.set(SDValue());
12489 if (Used->use_empty())
12490 DeadNodeSet.insert(Used);
12491 }
12492
12493 // For MachineNode, initialize the memory references information.
12495 MN->clearMemRefs();
12496
12497 // Swap for an appropriately sized array from the recycler.
12498 removeOperands(N);
12499 createOperands(N, Ops);
12500
12501 // Delete any nodes that are still dead after adding the uses for the
12502 // new operands.
12503 if (!DeadNodeSet.empty()) {
12504 SmallVector<SDNode *, 16> DeadNodes;
12505 for (SDNode *N : DeadNodeSet)
12506 if (N->use_empty())
12507 DeadNodes.push_back(N);
12508 RemoveDeadNodes(DeadNodes);
12509 }
12510
12511 if (InsertToken)
12512 CSEMap.insert(N, InsertToken); // Memoize the new node.
12513 return N;
12514}
12515
12517 unsigned OrigOpc = Node->getOpcode();
12518 unsigned NewOpc;
12519 switch (OrigOpc) {
12520 default:
12521 llvm_unreachable("mutateStrictFPToFP called with unexpected opcode!");
12522#define DAG_INSTRUCTION(NAME, NARG, ROUND_MODE, INTRINSIC, DAGN) \
12523 case ISD::STRICT_##DAGN: NewOpc = ISD::DAGN; break;
12524#define CMP_INSTRUCTION(NAME, NARG, ROUND_MODE, INTRINSIC, DAGN) \
12525 case ISD::STRICT_##DAGN: NewOpc = ISD::SETCC; break;
12526#include "llvm/IR/ConstrainedOps.def"
12527 }
12528
12529 assert(Node->getNumValues() == 2 && "Unexpected number of results!");
12530
12531 // We're taking this node out of the chain, so we need to re-link things.
12532 SDValue InputChain = Node->getOperand(0);
12533 SDValue OutputChain = SDValue(Node, 1);
12534 ReplaceAllUsesOfValueWith(OutputChain, InputChain);
12535
12537 for (unsigned i = 1, e = Node->getNumOperands(); i != e; ++i)
12538 Ops.push_back(Node->getOperand(i));
12539
12540 SDVTList VTs = getVTList(Node->getValueType(0));
12541 SDNode *Res = MorphNodeTo(Node, NewOpc, VTs, Ops);
12542
12543 // MorphNodeTo can operate in two ways: if an existing node with the
12544 // specified operands exists, it can just return it. Otherwise, it
12545 // updates the node in place to have the requested operands.
12546 if (Res == Node) {
12547 // If we updated the node in place, reset the node ID. To the isel,
12548 // this should be just like a newly allocated machine node.
12549 Res->setNodeId(-1);
12550 } else {
12553 }
12554
12555 return Res;
12556}
12557
12558/// getMachineNode - These are used for target selectors to create a new node
12559/// with specified return type(s), MachineInstr opcode, and operands.
12560///
12561/// Note that getMachineNode returns the resultant node. If there is already a
12562/// node of the specified opcode and operands, it returns that node instead of
12563/// the current one.
12565 EVT VT) {
12566 SDVTList VTs = getVTList(VT);
12567 return getMachineNode(Opcode, dl, VTs, {});
12568}
12569
12571 EVT VT, SDValue Op1) {
12572 SDVTList VTs = getVTList(VT);
12573 SDValue Ops[] = { Op1 };
12574 return getMachineNode(Opcode, dl, VTs, Ops);
12575}
12576
12578 EVT VT, SDValue Op1, SDValue Op2) {
12579 SDVTList VTs = getVTList(VT);
12580 SDValue Ops[] = { Op1, Op2 };
12581 return getMachineNode(Opcode, dl, VTs, Ops);
12582}
12583
12585 EVT VT, SDValue Op1, SDValue Op2,
12586 SDValue Op3) {
12587 SDVTList VTs = getVTList(VT);
12588 SDValue Ops[] = { Op1, Op2, Op3 };
12589 return getMachineNode(Opcode, dl, VTs, Ops);
12590}
12591
12594 SDVTList VTs = getVTList(VT);
12595 return getMachineNode(Opcode, dl, VTs, Ops);
12596}
12597
12599 EVT VT1, EVT VT2, SDValue Op1,
12600 SDValue Op2) {
12601 SDVTList VTs = getVTList(VT1, VT2);
12602 SDValue Ops[] = { Op1, Op2 };
12603 return getMachineNode(Opcode, dl, VTs, Ops);
12604}
12605
12607 EVT VT1, EVT VT2, SDValue Op1,
12608 SDValue Op2, SDValue Op3) {
12609 SDVTList VTs = getVTList(VT1, VT2);
12610 SDValue Ops[] = { Op1, Op2, Op3 };
12611 return getMachineNode(Opcode, dl, VTs, Ops);
12612}
12613
12615 EVT VT1, EVT VT2,
12617 SDVTList VTs = getVTList(VT1, VT2);
12618 return getMachineNode(Opcode, dl, VTs, Ops);
12619}
12620
12622 EVT VT1, EVT VT2, EVT VT3,
12623 SDValue Op1, SDValue Op2) {
12624 SDVTList VTs = getVTList(VT1, VT2, VT3);
12625 SDValue Ops[] = { Op1, Op2 };
12626 return getMachineNode(Opcode, dl, VTs, Ops);
12627}
12628
12630 EVT VT1, EVT VT2, EVT VT3,
12631 SDValue Op1, SDValue Op2,
12632 SDValue Op3) {
12633 SDVTList VTs = getVTList(VT1, VT2, VT3);
12634 SDValue Ops[] = { Op1, Op2, Op3 };
12635 return getMachineNode(Opcode, dl, VTs, Ops);
12636}
12637
12639 EVT VT1, EVT VT2, EVT VT3,
12641 SDVTList VTs = getVTList(VT1, VT2, VT3);
12642 return getMachineNode(Opcode, dl, VTs, Ops);
12643}
12644
12646 ArrayRef<EVT> ResultTys,
12648 SDVTList VTs = getVTList(ResultTys);
12649 return getMachineNode(Opcode, dl, VTs, Ops);
12650}
12651
12653 SDVTList VTs,
12655 bool DoCSE = VTs.VTs[VTs.NumVTs-1] != MVT::Glue;
12657 FoldingSetInsertToken InsertToken;
12658
12659 if (DoCSE) {
12660 SDNodeKey ID(~Opcode, VTs, Ops);
12661 if (SDNode *E = lookupNode(ID, DL, InsertToken)) {
12662 return cast<MachineSDNode>(UpdateSDLocOnMergeSDNode(E, DL));
12663 }
12664 }
12665
12666 // Allocate a new MachineSDNode.
12667 N = newSDNode<MachineSDNode>(~Opcode, DL.getIROrder(), DL.getDebugLoc(), VTs);
12668 createOperands(N, Ops);
12669
12670 if (DoCSE)
12671 CSEMap.insert(N, InsertToken);
12672
12673 InsertNode(N);
12674 NewSDValueDbgMsg(SDValue(N, 0), "Creating new machine node: ", this);
12675 return N;
12676}
12677
12678/// getTargetExtractSubreg - A convenience function for creating
12679/// TargetOpcode::EXTRACT_SUBREG nodes.
12681 SDValue Operand) {
12682 SDValue SRIdxVal = getTargetConstant(SRIdx, DL, MVT::i32);
12683 SDNode *Subreg = getMachineNode(TargetOpcode::EXTRACT_SUBREG, DL,
12684 VT, Operand, SRIdxVal);
12685 return SDValue(Subreg, 0);
12686}
12687
12688/// getTargetInsertSubreg - A convenience function for creating
12689/// TargetOpcode::INSERT_SUBREG nodes.
12691 SDValue Operand, SDValue Subreg) {
12692 SDValue SRIdxVal = getTargetConstant(SRIdx, DL, MVT::i32);
12693 SDNode *Result = getMachineNode(TargetOpcode::INSERT_SUBREG, DL,
12694 VT, Operand, Subreg, SRIdxVal);
12695 return SDValue(Result, 0);
12696}
12697
12698/// getNodeIfExists - Get the specified node if it's already available, or
12699/// else return NULL.
12702 bool AllowCommute) {
12703 SDNodeFlags Flags;
12704 if (Inserter)
12705 Flags = Inserter->getFlags();
12706 return getNodeIfExists(Opcode, VTList, Ops, Flags, AllowCommute);
12707}
12708
12711 const SDNodeFlags Flags,
12712 bool AllowCommute) {
12713 if (VTList.VTs[VTList.NumVTs - 1] == MVT::Glue)
12714 return nullptr;
12715
12716 auto Lookup = [&](ArrayRef<SDValue> LookupOps) -> SDNode * {
12717 SDNodeKey ID(Opcode, VTList, LookupOps);
12718 FoldingSetInsertToken InsertToken;
12719 if (SDNode *E = lookupNode(ID, InsertToken)) {
12720 E->intersectFlagsWith(Flags);
12721 return E;
12722 }
12723 return nullptr;
12724 };
12725
12726 if (SDNode *Existing = Lookup(Ops))
12727 return Existing;
12728
12729 if (AllowCommute && TLI->isCommutativeBinOp(Opcode))
12730 return Lookup({Ops[1], Ops[0]});
12731
12732 return nullptr;
12733}
12734
12735/// doesNodeExist - Check if a node exists without modifying its flags.
12736bool SelectionDAG::doesNodeExist(unsigned Opcode, SDVTList VTList,
12738 if (VTList.VTs[VTList.NumVTs - 1] != MVT::Glue) {
12739 SDNodeKey ID(Opcode, VTList, Ops);
12740 FoldingSetInsertToken InsertToken;
12741 if (lookupNode(ID, SDLoc(), InsertToken))
12742 return true;
12743 }
12744 return false;
12745}
12746
12747/// getDbgValue - Creates a SDDbgValue node.
12748///
12749/// SDNode
12751 SDNode *N, unsigned R, bool IsIndirect,
12752 const DebugLoc &DL, unsigned O) {
12753 assert(cast<DILocalVariable>(Var)->isValidLocationForIntrinsic(DL) &&
12754 "Expected inlined-at fields to agree");
12755 return new (DbgInfo->getAlloc())
12756 SDDbgValue(DbgInfo->getAlloc(), Var, Expr, SDDbgOperand::fromNode(N, R),
12757 {}, IsIndirect, DL, O,
12758 /*IsVariadic=*/false);
12759}
12760
12761/// Constant
12763 DIExpression *Expr,
12764 const Value *C,
12765 const DebugLoc &DL, unsigned O) {
12766 assert(cast<DILocalVariable>(Var)->isValidLocationForIntrinsic(DL) &&
12767 "Expected inlined-at fields to agree");
12768 return new (DbgInfo->getAlloc())
12769 SDDbgValue(DbgInfo->getAlloc(), Var, Expr, SDDbgOperand::fromConst(C), {},
12770 /*IsIndirect=*/false, DL, O,
12771 /*IsVariadic=*/false);
12772}
12773
12774/// FrameIndex
12776 DIExpression *Expr, unsigned FI,
12777 bool IsIndirect,
12778 const DebugLoc &DL,
12779 unsigned O) {
12780 assert(cast<DILocalVariable>(Var)->isValidLocationForIntrinsic(DL) &&
12781 "Expected inlined-at fields to agree");
12782 return getFrameIndexDbgValue(Var, Expr, FI, {}, IsIndirect, DL, O);
12783}
12784
12785/// FrameIndex with dependencies
12787 DIExpression *Expr, unsigned FI,
12788 ArrayRef<SDNode *> Dependencies,
12789 bool IsIndirect,
12790 const DebugLoc &DL,
12791 unsigned O) {
12792 assert(cast<DILocalVariable>(Var)->isValidLocationForIntrinsic(DL) &&
12793 "Expected inlined-at fields to agree");
12794 return new (DbgInfo->getAlloc())
12795 SDDbgValue(DbgInfo->getAlloc(), Var, Expr, SDDbgOperand::fromFrameIdx(FI),
12796 Dependencies, IsIndirect, DL, O,
12797 /*IsVariadic=*/false);
12798}
12799
12800/// VReg
12802 Register VReg, bool IsIndirect,
12803 const DebugLoc &DL, unsigned O) {
12804 assert(cast<DILocalVariable>(Var)->isValidLocationForIntrinsic(DL) &&
12805 "Expected inlined-at fields to agree");
12806 return new (DbgInfo->getAlloc())
12807 SDDbgValue(DbgInfo->getAlloc(), Var, Expr, SDDbgOperand::fromVReg(VReg),
12808 {}, IsIndirect, DL, O,
12809 /*IsVariadic=*/false);
12810}
12811
12814 ArrayRef<SDNode *> Dependencies,
12815 bool IsIndirect, const DebugLoc &DL,
12816 unsigned O, bool IsVariadic) {
12817 assert(cast<DILocalVariable>(Var)->isValidLocationForIntrinsic(DL) &&
12818 "Expected inlined-at fields to agree");
12819 return new (DbgInfo->getAlloc())
12820 SDDbgValue(DbgInfo->getAlloc(), Var, Expr, Locs, Dependencies, IsIndirect,
12821 DL, O, IsVariadic);
12822}
12823
12825 unsigned OffsetInBits, unsigned SizeInBits,
12826 bool InvalidateDbg) {
12827 SDNode *FromNode = From.getNode();
12828 SDNode *ToNode = To.getNode();
12829 assert(FromNode && ToNode && "Can't modify dbg values");
12830
12831 // PR35338
12832 // TODO: assert(From != To && "Redundant dbg value transfer");
12833 // TODO: assert(FromNode != ToNode && "Intranode dbg value transfer");
12834 if (From == To || FromNode == ToNode)
12835 return;
12836
12837 if (!FromNode->getHasDebugValue())
12838 return;
12839
12840 SDDbgOperand FromLocOp =
12841 SDDbgOperand::fromNode(From.getNode(), From.getResNo());
12843
12845 for (SDDbgValue *Dbg : GetDbgValues(FromNode)) {
12846 if (Dbg->isInvalidated())
12847 continue;
12848
12849 // TODO: assert(!Dbg->isInvalidated() && "Transfer of invalid dbg value");
12850
12851 // Create a new location ops vector that is equal to the old vector, but
12852 // with each instance of FromLocOp replaced with ToLocOp.
12853 bool Changed = false;
12854 auto NewLocOps = Dbg->copyLocationOps();
12855 std::replace_if(
12856 NewLocOps.begin(), NewLocOps.end(),
12857 [&Changed, FromLocOp](const SDDbgOperand &Op) {
12858 bool Match = Op == FromLocOp;
12859 Changed |= Match;
12860 return Match;
12861 },
12862 ToLocOp);
12863 // Ignore this SDDbgValue if we didn't find a matching location.
12864 if (!Changed)
12865 continue;
12866
12867 DIVariable *Var = Dbg->getVariable();
12868 auto *Expr = Dbg->getExpression();
12869 // If a fragment is requested, update the expression.
12870 if (SizeInBits) {
12871 // When splitting a larger (e.g., sign-extended) value whose
12872 // lower bits are described with an SDDbgValue, do not attempt
12873 // to transfer the SDDbgValue to the upper bits.
12874 if (auto FI = Expr->getFragmentInfo())
12875 if (OffsetInBits + SizeInBits > FI->SizeInBits)
12876 continue;
12877 auto Fragment = DIExpression::createFragmentExpression(Expr, OffsetInBits,
12878 SizeInBits);
12879 if (!Fragment)
12880 continue;
12881 Expr = *Fragment;
12882 }
12883
12884 auto AdditionalDependencies = Dbg->getAdditionalDependencies();
12885 // Clone the SDDbgValue and move it to To.
12886 SDDbgValue *Clone = getDbgValueList(
12887 Var, Expr, NewLocOps, AdditionalDependencies, Dbg->isIndirect(),
12888 Dbg->getDebugLoc(), std::max(ToNode->getIROrder(), Dbg->getOrder()),
12889 Dbg->isVariadic());
12890 ClonedDVs.push_back(Clone);
12891
12892 if (InvalidateDbg) {
12893 // Invalidate value and indicate the SDDbgValue should not be emitted.
12894 Dbg->setIsInvalidated();
12895 Dbg->setIsEmitted();
12896 }
12897 }
12898
12899 for (SDDbgValue *Dbg : ClonedDVs) {
12900 assert(is_contained(Dbg->getSDNodes(), ToNode) &&
12901 "Transferred DbgValues should depend on the new SDNode");
12902 AddDbgValue(Dbg, false);
12903 }
12904}
12905
12907 if (!N.getHasDebugValue())
12908 return;
12909
12910 auto GetLocationOperand = [](SDNode *Node, unsigned ResNo) {
12911 if (auto *FISDN = dyn_cast<FrameIndexSDNode>(Node))
12912 return SDDbgOperand::fromFrameIdx(FISDN->getIndex());
12913 return SDDbgOperand::fromNode(Node, ResNo);
12914 };
12915
12917 for (auto *DV : GetDbgValues(&N)) {
12918 if (DV->isInvalidated())
12919 continue;
12920 switch (N.getOpcode()) {
12921 default:
12922 break;
12923 case ISD::ADD: {
12924 SDValue N0 = N.getOperand(0);
12925 SDValue N1 = N.getOperand(1);
12926 if (!isa<ConstantSDNode>(N0)) {
12927 bool RHSConstant = isa<ConstantSDNode>(N1);
12928 uint64_t Offset;
12929 if (RHSConstant)
12930 Offset = N.getConstantOperandVal(1);
12931 // We are not allowed to turn indirect debug values variadic, so
12932 // don't salvage those.
12933 if (!RHSConstant && DV->isIndirect())
12934 continue;
12935
12936 // Rewrite an ADD constant node into a DIExpression. Since we are
12937 // performing arithmetic to compute the variable's *value* in the
12938 // DIExpression, we need to mark the expression with a
12939 // DW_OP_stack_value.
12940 auto *DIExpr = DV->getExpression();
12941 auto NewLocOps = DV->copyLocationOps();
12942 bool Changed = false;
12943 size_t OrigLocOpsSize = NewLocOps.size();
12944 for (size_t i = 0; i < OrigLocOpsSize; ++i) {
12945 // We're not given a ResNo to compare against because the whole
12946 // node is going away. We know that any ISD::ADD only has one
12947 // result, so we can assume any node match is using the result.
12948 if (NewLocOps[i].getKind() != SDDbgOperand::SDNODE ||
12949 NewLocOps[i].getSDNode() != &N)
12950 continue;
12951 NewLocOps[i] = GetLocationOperand(N0.getNode(), N0.getResNo());
12952 if (RHSConstant) {
12955 DIExpr = DIExpression::appendOpsToArg(DIExpr, ExprOps, i, true);
12956 } else {
12957 // Convert to a variadic expression (if not already).
12958 // convertToVariadicExpression() returns a const pointer, so we use
12959 // a temporary const variable here.
12960 const auto *TmpDIExpr =
12964 ExprOps.push_back(NewLocOps.size());
12965 ExprOps.push_back(dwarf::DW_OP_plus);
12966 SDDbgOperand RHS =
12968 NewLocOps.push_back(RHS);
12969 DIExpr = DIExpression::appendOpsToArg(TmpDIExpr, ExprOps, i, true);
12970 }
12971 Changed = true;
12972 }
12973 (void)Changed;
12974 assert(Changed && "Salvage target doesn't use N");
12975
12976 bool IsVariadic =
12977 DV->isVariadic() || OrigLocOpsSize != NewLocOps.size();
12978
12979 auto AdditionalDependencies = DV->getAdditionalDependencies();
12980 SDDbgValue *Clone = getDbgValueList(
12981 DV->getVariable(), DIExpr, NewLocOps, AdditionalDependencies,
12982 DV->isIndirect(), DV->getDebugLoc(), DV->getOrder(), IsVariadic);
12983 ClonedDVs.push_back(Clone);
12984 DV->setIsInvalidated();
12985 DV->setIsEmitted();
12986 LLVM_DEBUG(dbgs() << "SALVAGE: Rewriting";
12987 N0.getNode()->dumprFull(this);
12988 dbgs() << " into " << *DIExpr << '\n');
12989 }
12990 break;
12991 }
12992 case ISD::TRUNCATE: {
12993 SDValue N0 = N.getOperand(0);
12994 TypeSize FromSize = N0.getValueSizeInBits();
12995 TypeSize ToSize = N.getValueSizeInBits(0);
12996
12997 DIExpression *DbgExpression = DV->getExpression();
12998 auto ExtOps = DIExpression::getExtOps(FromSize, ToSize, false);
12999 auto NewLocOps = DV->copyLocationOps();
13000 bool Changed = false;
13001 for (size_t i = 0; i < NewLocOps.size(); ++i) {
13002 if (NewLocOps[i].getKind() != SDDbgOperand::SDNODE ||
13003 NewLocOps[i].getSDNode() != &N)
13004 continue;
13005
13006 NewLocOps[i] = GetLocationOperand(N0.getNode(), N0.getResNo());
13007 DbgExpression = DIExpression::appendOpsToArg(DbgExpression, ExtOps, i);
13008 Changed = true;
13009 }
13010 assert(Changed && "Salvage target doesn't use N");
13011 (void)Changed;
13012
13013 SDDbgValue *Clone =
13014 getDbgValueList(DV->getVariable(), DbgExpression, NewLocOps,
13015 DV->getAdditionalDependencies(), DV->isIndirect(),
13016 DV->getDebugLoc(), DV->getOrder(), DV->isVariadic());
13017
13018 ClonedDVs.push_back(Clone);
13019 DV->setIsInvalidated();
13020 DV->setIsEmitted();
13021 LLVM_DEBUG(dbgs() << "SALVAGE: Rewriting"; N0.getNode()->dumprFull(this);
13022 dbgs() << " into " << *DbgExpression << '\n');
13023 break;
13024 }
13025 }
13026 }
13027
13028 for (SDDbgValue *Dbg : ClonedDVs) {
13029 assert((!Dbg->getSDNodes().empty() ||
13030 llvm::any_of(Dbg->getLocationOps(),
13031 [&](const SDDbgOperand &Op) {
13032 return Op.getKind() == SDDbgOperand::FRAMEIX;
13033 })) &&
13034 "Salvaged DbgValue should depend on a new SDNode");
13035 AddDbgValue(Dbg, false);
13036 }
13037}
13038
13039/// Creates a SDDbgLabel node.
13041 const DebugLoc &DL, unsigned O) {
13042 assert(cast<DILabel>(Label)->isValidLocationForIntrinsic(DL) &&
13043 "Expected inlined-at fields to agree");
13044 return new (DbgInfo->getAlloc()) SDDbgLabel(Label, DL, O);
13045}
13046
13047namespace {
13048
13049/// RAUWUpdateListener - Helper for ReplaceAllUsesWith - When the node
13050/// pointed to by a use iterator is deleted, increment the use iterator
13051/// so that it doesn't dangle.
13052///
13053class RAUWUpdateListener : public SelectionDAG::DAGUpdateListener {
13056
13057 void NodeDeleted(SDNode *N, SDNode *E) override {
13058 // Increment the iterator as needed.
13059 while (UI != UE && N == UI->getUser())
13060 ++UI;
13061 }
13062
13063public:
13064 RAUWUpdateListener(SelectionDAG &d,
13067 : SelectionDAG::DAGUpdateListener(d), UI(ui), UE(ue) {}
13068};
13069
13070} // end anonymous namespace
13071
13072/// ReplaceAllUsesWith - Modify anything using 'From' to use 'To' instead.
13073/// This can cause recursive merging of nodes in the DAG.
13074///
13075/// This version assumes From has a single result value.
13076///
13078 SDNode *From = FromN.getNode();
13079 assert(From->getNumValues() == 1 && FromN.getResNo() == 0 &&
13080 "Cannot replace with this method!");
13081 assert(From != To.getNode() && "Cannot replace uses of with self");
13082
13083 // Preserve Debug Values
13084 transferDbgValues(FromN, To);
13085 // Preserve extra info.
13086 copyExtraInfo(From, To.getNode());
13087
13088 // Iterate over all the existing uses of From. New uses will be added
13089 // to the beginning of the use list, which we avoid visiting.
13090 // This specifically avoids visiting uses of From that arise while the
13091 // replacement is happening, because any such uses would be the result
13092 // of CSE: If an existing node looks like From after one of its operands
13093 // is replaced by To, we don't want to replace of all its users with To
13094 // too. See PR3018 for more info.
13095 SDNode::use_iterator UI = From->use_begin(), UE = From->use_end();
13096 RAUWUpdateListener Listener(*this, UI, UE);
13097 while (UI != UE) {
13098 SDNode *User = UI->getUser();
13099
13100 // This node is about to morph, remove its old self from the CSE maps.
13101 RemoveNodeFromCSEMaps(User);
13102
13103 // A user can appear in a use list multiple times, and when this
13104 // happens the uses are usually next to each other in the list.
13105 // To help reduce the number of CSE recomputations, process all
13106 // the uses of this user that we can find this way.
13107 do {
13108 SDUse &Use = *UI;
13109 ++UI;
13110 Use.set(To);
13111 if (To->isDivergent() != From->isDivergent())
13113 } while (UI != UE && UI->getUser() == User);
13114 // Now that we have modified User, add it back to the CSE maps. If it
13115 // already exists there, recursively merge the results together.
13116 AddModifiedNodeToCSEMaps(User);
13117 }
13118
13119 // If we just RAUW'd the root, take note.
13120 if (FromN == getRoot())
13121 setRoot(To);
13122}
13123
13124/// ReplaceAllUsesWith - Modify anything using 'From' to use 'To' instead.
13125/// This can cause recursive merging of nodes in the DAG.
13126///
13127/// This version assumes that for each value of From, there is a
13128/// corresponding value in To in the same position with the same type.
13129///
13131#ifndef NDEBUG
13132 for (unsigned i = 0, e = From->getNumValues(); i != e; ++i)
13133 assert((!From->hasAnyUseOfValue(i) ||
13134 From->getValueType(i) == To->getValueType(i)) &&
13135 "Cannot use this version of ReplaceAllUsesWith!");
13136#endif
13137
13138 // Handle the trivial case.
13139 if (From == To)
13140 return;
13141
13142 // Preserve Debug Info. Only do this if there's a use.
13143 for (unsigned i = 0, e = From->getNumValues(); i != e; ++i)
13144 if (From->hasAnyUseOfValue(i)) {
13145 assert((i < To->getNumValues()) && "Invalid To location");
13146 transferDbgValues(SDValue(From, i), SDValue(To, i));
13147 }
13148 // Preserve extra info.
13149 copyExtraInfo(From, To);
13150
13151 // Iterate over just the existing users of From. See the comments in
13152 // the ReplaceAllUsesWith above.
13153 SDNode::use_iterator UI = From->use_begin(), UE = From->use_end();
13154 RAUWUpdateListener Listener(*this, UI, UE);
13155 while (UI != UE) {
13156 SDNode *User = UI->getUser();
13157
13158 // This node is about to morph, remove its old self from the CSE maps.
13159 RemoveNodeFromCSEMaps(User);
13160
13161 // A user can appear in a use list multiple times, and when this
13162 // happens the uses are usually next to each other in the list.
13163 // To help reduce the number of CSE recomputations, process all
13164 // the uses of this user that we can find this way.
13165 do {
13166 SDUse &Use = *UI;
13167 ++UI;
13168 Use.setNode(To);
13169 if (To->isDivergent() != From->isDivergent())
13171 } while (UI != UE && UI->getUser() == User);
13172
13173 // Now that we have modified User, add it back to the CSE maps. If it
13174 // already exists there, recursively merge the results together.
13175 AddModifiedNodeToCSEMaps(User);
13176 }
13177
13178 // If we just RAUW'd the root, take note.
13179 if (From == getRoot().getNode())
13180 setRoot(SDValue(To, getRoot().getResNo()));
13181}
13182
13183/// ReplaceAllUsesWith - Modify anything using 'From' to use 'To' instead.
13184/// This can cause recursive merging of nodes in the DAG.
13185///
13186/// This version can replace From with any result values. To must match the
13187/// number and types of values returned by From.
13189 if (From->getNumValues() == 1) // Handle the simple case efficiently.
13190 return ReplaceAllUsesWith(SDValue(From, 0), To[0]);
13191
13192 for (unsigned i = 0, e = From->getNumValues(); i != e; ++i) {
13193 // Preserve Debug Info.
13194 transferDbgValues(SDValue(From, i), To[i]);
13195 // Preserve extra info.
13196 copyExtraInfo(From, To[i].getNode());
13197 }
13198
13199 // Iterate over just the existing users of From. See the comments in
13200 // the ReplaceAllUsesWith above.
13201 SDNode::use_iterator UI = From->use_begin(), UE = From->use_end();
13202 RAUWUpdateListener Listener(*this, UI, UE);
13203 while (UI != UE) {
13204 SDNode *User = UI->getUser();
13205
13206 // This node is about to morph, remove its old self from the CSE maps.
13207 RemoveNodeFromCSEMaps(User);
13208
13209 // A user can appear in a use list multiple times, and when this happens the
13210 // uses are usually next to each other in the list. To help reduce the
13211 // number of CSE and divergence recomputations, process all the uses of this
13212 // user that we can find this way.
13213 bool To_IsDivergent = false;
13214 do {
13215 SDUse &Use = *UI;
13216 const SDValue &ToOp = To[Use.getResNo()];
13217 ++UI;
13218 Use.set(ToOp);
13219 if (ToOp.getValueType() != MVT::Other)
13220 To_IsDivergent |= ToOp->isDivergent();
13221 } while (UI != UE && UI->getUser() == User);
13222
13223 if (To_IsDivergent != From->isDivergent())
13225
13226 // Now that we have modified User, add it back to the CSE maps. If it
13227 // already exists there, recursively merge the results together.
13228 AddModifiedNodeToCSEMaps(User);
13229 }
13230
13231 // If we just RAUW'd the root, take note.
13232 if (From == getRoot().getNode())
13233 setRoot(SDValue(To[getRoot().getResNo()]));
13234}
13235
13236/// ReplaceAllUsesOfValueWith - Replace any uses of From with To, leaving
13237/// uses of other values produced by From.getNode() alone. The Deleted
13238/// vector is handled the same way as for ReplaceAllUsesWith.
13240 // Handle the really simple, really trivial case efficiently.
13241 if (From == To) return;
13242
13243 // Handle the simple, trivial, case efficiently.
13244 if (From.getNode()->getNumValues() == 1) {
13245 ReplaceAllUsesWith(From, To);
13246 return;
13247 }
13248
13249 // Preserve Debug Info.
13250 transferDbgValues(From, To);
13251 copyExtraInfo(From.getNode(), To.getNode());
13252
13253 // Iterate over just the existing users of From. See the comments in
13254 // the ReplaceAllUsesWith above.
13255 SDNode::use_iterator UI = From.getNode()->use_begin(),
13256 UE = From.getNode()->use_end();
13257 RAUWUpdateListener Listener(*this, UI, UE);
13258 while (UI != UE) {
13259 SDNode *User = UI->getUser();
13260 bool UserRemovedFromCSEMaps = false;
13261
13262 // A user can appear in a use list multiple times, and when this
13263 // happens the uses are usually next to each other in the list.
13264 // To help reduce the number of CSE recomputations, process all
13265 // the uses of this user that we can find this way.
13266 do {
13267 SDUse &Use = *UI;
13268
13269 // Skip uses of different values from the same node.
13270 if (Use.getResNo() != From.getResNo()) {
13271 ++UI;
13272 continue;
13273 }
13274
13275 // If this node hasn't been modified yet, it's still in the CSE maps,
13276 // so remove its old self from the CSE maps.
13277 if (!UserRemovedFromCSEMaps) {
13278 RemoveNodeFromCSEMaps(User);
13279 UserRemovedFromCSEMaps = true;
13280 }
13281
13282 ++UI;
13283 Use.set(To);
13284 if (To->isDivergent() != From->isDivergent())
13286 } while (UI != UE && UI->getUser() == User);
13287 // We are iterating over all uses of the From node, so if a use
13288 // doesn't use the specific value, no changes are made.
13289 if (!UserRemovedFromCSEMaps)
13290 continue;
13291
13292 // Now that we have modified User, add it back to the CSE maps. If it
13293 // already exists there, recursively merge the results together.
13294 AddModifiedNodeToCSEMaps(User);
13295 }
13296
13297 // If we just RAUW'd the root, take note.
13298 if (From == getRoot())
13299 setRoot(To);
13300}
13301
13302namespace {
13303
13304/// UseMemo - This class is used by SelectionDAG::ReplaceAllUsesOfValuesWith
13305/// to record information about a use.
13306struct UseMemo {
13307 SDNode *User;
13308 unsigned Index;
13309 SDUse *Use;
13310};
13311
13312/// operator< - Sort Memos by User.
13313bool operator<(const UseMemo &L, const UseMemo &R) {
13314 return (intptr_t)L.User < (intptr_t)R.User;
13315}
13316
13317/// RAUOVWUpdateListener - Helper for ReplaceAllUsesOfValuesWith - When the node
13318/// pointed to by a UseMemo is deleted, set the User to nullptr to indicate that
13319/// the node already has been taken care of recursively.
13320class RAUOVWUpdateListener : public SelectionDAG::DAGUpdateListener {
13321 SmallVectorImpl<UseMemo> &Uses;
13322
13323 void NodeDeleted(SDNode *N, SDNode *E) override {
13324 for (UseMemo &Memo : Uses)
13325 if (Memo.User == N)
13326 Memo.User = nullptr;
13327 }
13328
13329public:
13330 RAUOVWUpdateListener(SelectionDAG &d, SmallVectorImpl<UseMemo> &uses)
13331 : SelectionDAG::DAGUpdateListener(d), Uses(uses) {}
13332};
13333
13334} // end anonymous namespace
13335
13336/// Return true if a glue output should propagate divergence information.
13338 switch (Node->getOpcode()) {
13339 case ISD::CopyFromReg:
13340 case ISD::CopyToReg:
13341 return false;
13342 default:
13343 return true;
13344 }
13345
13346 llvm_unreachable("covered opcode switch");
13347}
13348
13350 if (TLI->isSDNodeAlwaysUniform(N)) {
13351 assert(!TLI->isSDNodeSourceOfDivergence(N, FLI, UA) &&
13352 "Conflicting divergence information!");
13353 return false;
13354 }
13355 if (TLI->isSDNodeSourceOfDivergence(N, FLI, UA))
13356 return true;
13357 for (const auto &Op : N->ops()) {
13358 EVT VT = Op.getValueType();
13359
13360 // Skip Chain. It does not carry divergence.
13361 if (VT != MVT::Other && Op.getNode()->isDivergent() &&
13362 (VT != MVT::Glue || gluePropagatesDivergence(Op.getNode())))
13363 return true;
13364 }
13365 return false;
13366}
13367
13369 SmallVector<SDNode *, 16> Worklist(1, N);
13370 do {
13371 N = Worklist.pop_back_val();
13372 bool IsDivergent = calculateDivergence(N);
13373 if (N->SDNodeBits.IsDivergent != IsDivergent) {
13374 N->SDNodeBits.IsDivergent = IsDivergent;
13375 llvm::append_range(Worklist, N->users());
13376 }
13377 } while (!Worklist.empty());
13378}
13379
13380void SelectionDAG::CreateTopologicalOrder(std::vector<SDNode *> &Order) {
13382 Order.reserve(AllNodes.size());
13383 for (auto &N : allnodes()) {
13384 unsigned NOps = N.getNumOperands();
13385 Degree[&N] = NOps;
13386 if (0 == NOps)
13387 Order.push_back(&N);
13388 }
13389 for (size_t I = 0; I != Order.size(); ++I) {
13390 SDNode *N = Order[I];
13391 for (auto *U : N->users()) {
13392 unsigned &UnsortedOps = Degree[U];
13393 if (0 == --UnsortedOps)
13394 Order.push_back(U);
13395 }
13396 }
13397}
13398
13399#if !defined(NDEBUG) && LLVM_ENABLE_ABI_BREAKING_CHECKS
13400void SelectionDAG::VerifyDAGDivergence() {
13401 std::vector<SDNode *> TopoOrder;
13402 CreateTopologicalOrder(TopoOrder);
13403 for (auto *N : TopoOrder) {
13404 assert(calculateDivergence(N) == N->isDivergent() &&
13405 "Divergence bit inconsistency detected");
13406 }
13407}
13408#endif
13409
13410/// ReplaceAllUsesOfValuesWith - Replace any uses of From with To, leaving
13411/// uses of other values produced by From.getNode() alone. The same value
13412/// may appear in both the From and To list. The Deleted vector is
13413/// handled the same way as for ReplaceAllUsesWith.
13415 const SDValue *To,
13416 unsigned Num){
13417 // Handle the simple, trivial case efficiently.
13418 if (Num == 1)
13419 return ReplaceAllUsesOfValueWith(*From, *To);
13420
13421 transferDbgValues(*From, *To);
13422 copyExtraInfo(From->getNode(), To->getNode());
13423
13424 // Read up all the uses and make records of them. This helps
13425 // processing new uses that are introduced during the
13426 // replacement process.
13428 for (unsigned i = 0; i != Num; ++i) {
13429 unsigned FromResNo = From[i].getResNo();
13430 SDNode *FromNode = From[i].getNode();
13431 for (SDUse &Use : FromNode->uses()) {
13432 if (Use.getResNo() == FromResNo) {
13433 UseMemo Memo = {Use.getUser(), i, &Use};
13434 Uses.push_back(Memo);
13435 }
13436 }
13437 }
13438
13439 // Sort the uses, so that all the uses from a given User are together.
13441 RAUOVWUpdateListener Listener(*this, Uses);
13442
13443 for (unsigned UseIndex = 0, UseIndexEnd = Uses.size();
13444 UseIndex != UseIndexEnd; ) {
13445 // We know that this user uses some value of From. If it is the right
13446 // value, update it.
13447 SDNode *User = Uses[UseIndex].User;
13448 // If the node has been deleted by recursive CSE updates when updating
13449 // another node, then just skip this entry.
13450 if (User == nullptr) {
13451 ++UseIndex;
13452 continue;
13453 }
13454
13455 // This node is about to morph, remove its old self from the CSE maps.
13456 RemoveNodeFromCSEMaps(User);
13457
13458 // The Uses array is sorted, so all the uses for a given User
13459 // are next to each other in the list.
13460 // To help reduce the number of CSE recomputations, process all
13461 // the uses of this user that we can find this way.
13462 do {
13463 unsigned i = Uses[UseIndex].Index;
13464 SDUse &Use = *Uses[UseIndex].Use;
13465 ++UseIndex;
13466
13467 Use.set(To[i]);
13468 } while (UseIndex != UseIndexEnd && Uses[UseIndex].User == User);
13469
13470 // Now that we have modified User, add it back to the CSE maps. If it
13471 // already exists there, recursively merge the results together.
13472 AddModifiedNodeToCSEMaps(User);
13473 }
13474}
13475
13476/// AssignTopologicalOrder - Assign a unique node id for each node in the DAG
13477/// based on their topological order. It returns the maximum id and a vector
13478/// of the SDNodes* in assigned order by reference.
13480 unsigned DAGSize = 0;
13481
13482 // SortedPos tracks the progress of the algorithm. Nodes before it are
13483 // sorted, nodes after it are unsorted. When the algorithm completes
13484 // it is at the end of the list.
13485 allnodes_iterator SortedPos = allnodes_begin();
13486
13487 // Visit all the nodes. Move nodes with no operands to the front of
13488 // the list immediately. Annotate nodes that do have operands with their
13489 // operand count. Before we do this, the Node Id fields of the nodes
13490 // may contain arbitrary values. After, the Node Id fields for nodes
13491 // before SortedPos will contain the topological sort index, and the
13492 // Node Id fields for nodes At SortedPos and after will contain the
13493 // count of outstanding operands.
13495 checkForCycles(&N, this);
13496 unsigned Degree = N.getNumOperands();
13497 if (Degree == 0) {
13498 // A node with no uses, add it to the result array immediately.
13499 N.setNodeId(DAGSize++);
13500 allnodes_iterator Q(&N);
13501 if (Q != SortedPos)
13502 SortedPos = AllNodes.insert(SortedPos, AllNodes.remove(Q));
13503 assert(SortedPos != AllNodes.end() && "Overran node list");
13504 ++SortedPos;
13505 } else {
13506 // Temporarily use the Node Id as scratch space for the degree count.
13507 N.setNodeId(Degree);
13508 }
13509 }
13510
13511 // Visit all the nodes. As we iterate, move nodes into sorted order,
13512 // such that by the time the end is reached all nodes will be sorted.
13513 for (SDNode &Node : allnodes()) {
13514 SDNode *N = &Node;
13515 checkForCycles(N, this);
13516 // N is in sorted position, so all its uses have one less operand
13517 // that needs to be sorted.
13518 for (SDNode *P : N->users()) {
13519 unsigned Degree = P->getNodeId();
13520 assert(Degree != 0 && "Invalid node degree");
13521 --Degree;
13522 if (Degree == 0) {
13523 // All of P's operands are sorted, so P may sorted now.
13524 P->setNodeId(DAGSize++);
13525 if (P->getIterator() != SortedPos)
13526 SortedPos = AllNodes.insert(SortedPos, AllNodes.remove(P));
13527 assert(SortedPos != AllNodes.end() && "Overran node list");
13528 ++SortedPos;
13529 } else {
13530 // Update P's outstanding operand count.
13531 P->setNodeId(Degree);
13532 }
13533 }
13534 if (Node.getIterator() == SortedPos) {
13535#ifndef NDEBUG
13537 SDNode *S = &*++I;
13538 dbgs() << "Overran sorted position:\n";
13539 S->dumprFull(this); dbgs() << "\n";
13540 dbgs() << "Checking if this is due to cycles\n";
13541 checkForCycles(this, true);
13542#endif
13543 llvm_unreachable(nullptr);
13544 }
13545 }
13546
13547 assert(SortedPos == AllNodes.end() &&
13548 "Topological sort incomplete!");
13549 assert(AllNodes.front().getOpcode() == ISD::EntryToken &&
13550 "First node in topological sort is not the entry token!");
13551 assert(AllNodes.front().getNodeId() == 0 &&
13552 "First node in topological sort has non-zero id!");
13553 assert(AllNodes.front().getNumOperands() == 0 &&
13554 "First node in topological sort has operands!");
13555 assert(AllNodes.back().getNodeId() == (int)DAGSize-1 &&
13556 "Last node in topologic sort has unexpected id!");
13557 assert(AllNodes.back().use_empty() &&
13558 "Last node in topologic sort has users!");
13559 assert(DAGSize == allnodes_size() && "Node count mismatch!");
13560 return DAGSize;
13561}
13562
13564 SmallVectorImpl<const SDNode *> &SortedNodes) const {
13565 SortedNodes.clear();
13566 // Node -> remaining number of outstanding operands.
13567 DenseMap<const SDNode *, unsigned> RemainingOperands;
13568
13569 // Put nodes without any operands into SortedNodes first.
13570 for (const SDNode &N : allnodes()) {
13571 checkForCycles(&N, this);
13572 unsigned NumOperands = N.getNumOperands();
13573 if (NumOperands == 0)
13574 SortedNodes.push_back(&N);
13575 else
13576 // Record their total number of outstanding operands.
13577 RemainingOperands[&N] = NumOperands;
13578 }
13579
13580 // A node is pushed into SortedNodes when all of its operands (predecessors in
13581 // the graph) are also in SortedNodes.
13582 for (unsigned i = 0U; i < SortedNodes.size(); ++i) {
13583 const SDNode *N = SortedNodes[i];
13584 for (const SDNode *U : N->users()) {
13585 // HandleSDNode is never part of a DAG and therefore has no entry in
13586 // RemainingOperands.
13587 if (U->getOpcode() == ISD::HANDLENODE)
13588 continue;
13589 unsigned &NumRemOperands = RemainingOperands[U];
13590 assert(NumRemOperands && "Invalid number of remaining operands");
13591 --NumRemOperands;
13592 if (!NumRemOperands)
13593 SortedNodes.push_back(U);
13594 }
13595 }
13596
13597 assert(SortedNodes.size() == AllNodes.size() && "Node count mismatch");
13598 assert(SortedNodes.front()->getOpcode() == ISD::EntryToken &&
13599 "First node in topological sort is not the entry token");
13600 assert(SortedNodes.front()->getNumOperands() == 0 &&
13601 "First node in topological sort has operands");
13602}
13603
13604/// AddDbgValue - Add a dbg_value SDNode. If SD is non-null that means the
13605/// value is produced by SD.
13606void SelectionDAG::AddDbgValue(SDDbgValue *DB, bool isParameter) {
13607 for (SDNode *SD : DB->getSDNodes()) {
13608 if (!SD)
13609 continue;
13610 assert(DbgInfo->getSDDbgValues(SD).empty() || SD->getHasDebugValue());
13611 SD->setHasDebugValue(true);
13612 }
13613 DbgInfo->add(DB, isParameter);
13614}
13615
13616void SelectionDAG::AddDbgLabel(SDDbgLabel *DB) { DbgInfo->add(DB); }
13617
13619 SDValue NewMemOpChain) {
13620 assert(isa<MemSDNode>(NewMemOpChain) && "Expected a memop node");
13621 assert(NewMemOpChain.getValueType() == MVT::Other && "Expected a token VT");
13622 // The new memory operation must have the same position as the old load in
13623 // terms of memory dependency. Create a TokenFactor for the old load and new
13624 // memory operation and update uses of the old load's output chain to use that
13625 // TokenFactor.
13626 if (OldChain == NewMemOpChain || OldChain.use_empty())
13627 return NewMemOpChain;
13628
13629 SDValue TokenFactor = getNode(ISD::TokenFactor, SDLoc(OldChain), MVT::Other,
13630 OldChain, NewMemOpChain);
13631 ReplaceAllUsesOfValueWith(OldChain, TokenFactor);
13632 UpdateNodeOperands(TokenFactor.getNode(), OldChain, NewMemOpChain);
13633 return TokenFactor;
13634}
13635
13637 SDValue NewMemOp) {
13638 assert(isa<MemSDNode>(NewMemOp.getNode()) && "Expected a memop node");
13639 SDValue OldChain = SDValue(OldLoad, 1);
13640 SDValue NewMemOpChain = NewMemOp.getValue(1);
13641 return makeEquivalentMemoryOrdering(OldChain, NewMemOpChain);
13642}
13643
13645 Function **OutFunction) {
13646 assert(isa<ExternalSymbolSDNode>(Op) && "Node should be an ExternalSymbol");
13647
13648 auto *Symbol = cast<ExternalSymbolSDNode>(Op)->getSymbol();
13649 auto *Module = MF->getFunction().getParent();
13650 auto *Function = Module->getFunction(Symbol);
13651
13652 if (OutFunction != nullptr)
13653 *OutFunction = Function;
13654
13655 if (Function != nullptr) {
13656 auto PtrTy = TLI->getPointerTy(getDataLayout(), Function->getAddressSpace());
13657 return getGlobalAddress(Function, SDLoc(Op), PtrTy);
13658 }
13659
13660 std::string ErrorStr;
13661 raw_string_ostream ErrorFormatter(ErrorStr);
13662 ErrorFormatter << "Undefined external symbol ";
13663 ErrorFormatter << '"' << Symbol << '"';
13664 report_fatal_error(Twine(ErrorStr));
13665}
13666
13667//===----------------------------------------------------------------------===//
13668// SDNode Class
13669//===----------------------------------------------------------------------===//
13670
13673 return Const != nullptr && Const->isZero();
13674}
13675
13677 return V.isUndef() || isNullConstant(V);
13678}
13679
13682 return Const != nullptr && Const->isZero() && !Const->isNegative();
13683}
13684
13687 return Const != nullptr && Const->isAllOnes();
13688}
13689
13692 return Const != nullptr && Const->isOne();
13693}
13694
13697 return Const != nullptr && Const->isMinSignedValue();
13698}
13699
13701 SDValue V, unsigned OperandNo,
13702 unsigned Depth) const {
13703 APInt DemandedElts = getDemandAllEltsMask(V);
13704 return isIdentityElement(Opcode, Flags, V, DemandedElts, OperandNo, Depth);
13705}
13706
13708 SDValue V, const APInt &DemandedElts,
13709 unsigned OperandNo, unsigned Depth) const {
13710 // NOTE: The cases should match with IR's ConstantExpr::getBinOpIdentity().
13711 // TODO: Target-specific opcodes could be added.
13712 if (V.getValueType().isInteger()) {
13713 KnownBits Known = computeKnownBits(V, DemandedElts, Depth);
13714 if (Known.isConstant()) {
13715 const APInt &Const = Known.getConstant();
13716 switch (Opcode) {
13717 case ISD::ADD:
13718 case ISD::OR:
13719 case ISD::XOR:
13720 case ISD::UMAX:
13721 return Const.isZero();
13722 case ISD::MUL:
13723 return Const.isOne();
13724 case ISD::AND:
13725 case ISD::UMIN:
13726 return Const.isAllOnes();
13727 case ISD::SMAX:
13728 return Const.isMinSignedValue();
13729 case ISD::SMIN:
13730 return Const.isMaxSignedValue();
13731 case ISD::SUB:
13732 case ISD::SHL:
13733 case ISD::SRA:
13734 case ISD::SRL:
13735 return OperandNo == 1 && Const.isZero();
13736 case ISD::UDIV:
13737 case ISD::SDIV:
13738 return OperandNo == 1 && Const.isOne();
13739 }
13740 }
13741 } else if (auto *ConstFP = isConstOrConstSplatFP(V, DemandedElts)) {
13742 switch (Opcode) {
13743 case ISD::FADD:
13744 return ConstFP->isZero() &&
13745 (Flags.hasNoSignedZeros() || ConstFP->isNegative());
13746 case ISD::FSUB:
13747 return OperandNo == 1 && ConstFP->isZero() &&
13748 (Flags.hasNoSignedZeros() || !ConstFP->isNegative());
13749 case ISD::FMUL:
13750 return ConstFP->isOne();
13751 case ISD::FDIV:
13752 return OperandNo == 1 && ConstFP->isOne();
13753 case ISD::FMINNUM:
13754 case ISD::FMAXNUM:
13755 case ISD::FMINIMUMNUM:
13756 case ISD::FMAXIMUMNUM: {
13757 // Neutral element for fminnum/fminimumnum is NaN, Inf or FLT_MAX,
13758 // depending on fast-math flags (FMF).
13759 EVT VT = V.getValueType();
13760 const fltSemantics &Semantics = VT.getFltSemantics();
13761 APFloat NeutralAF = !Flags.hasNoNaNs() ? APFloat::getQNaN(Semantics)
13762 : !Flags.hasNoInfs() ? APFloat::getInf(Semantics)
13763 : APFloat::getLargest(Semantics);
13764 if (Opcode == ISD::FMAXNUM || Opcode == ISD::FMAXIMUMNUM)
13765 NeutralAF.changeSign();
13766
13767 return ConstFP->isExactlyValue(NeutralAF);
13768 }
13769 case ISD::FMINIMUM:
13770 case ISD::FMAXIMUM: {
13771 // Neutral element for fminimum is Inf or FLT_MAX, depending on FMF.
13772 const APFloat &VAPF = ConstFP->getValueAPF();
13773 bool NeutralNegative = (Opcode == ISD::FMAXIMUM);
13774 if (Flags.hasNoInfs())
13775 return VAPF.isLargest() && VAPF.isNegative() == NeutralNegative;
13776 return VAPF.isInfinity() && VAPF.isNegative() == NeutralNegative;
13777 }
13778 }
13779 }
13780 return false;
13781}
13782
13784 while (V.getOpcode() == ISD::BITCAST)
13785 V = V.getOperand(0);
13786 return V;
13787}
13788
13790 while (V.getOpcode() == ISD::BITCAST && V.getOperand(0).hasOneUse())
13791 V = V.getOperand(0);
13792 return V;
13793}
13794
13796 while (V.getOpcode() == ISD::EXTRACT_SUBVECTOR)
13797 V = V.getOperand(0);
13798 return V;
13799}
13800
13802 while (V.getOpcode() == ISD::INSERT_VECTOR_ELT) {
13803 SDValue InVec = V.getOperand(0);
13804 SDValue EltNo = V.getOperand(2);
13805 EVT VT = InVec.getValueType();
13806 auto *IndexC = dyn_cast<ConstantSDNode>(EltNo);
13807 if (IndexC && VT.isFixedLengthVector() &&
13808 IndexC->getAPIntValue().ult(VT.getVectorNumElements()) &&
13809 !DemandedElts[IndexC->getZExtValue()]) {
13810 V = InVec;
13811 continue;
13812 }
13813 break;
13814 }
13815 return V;
13816}
13817
13819 while (V.getOpcode() == ISD::TRUNCATE)
13820 V = V.getOperand(0);
13821 return V;
13822}
13823
13824bool llvm::isBitwiseNot(SDValue V, bool AllowUndefs) {
13825 if (V.getOpcode() != ISD::XOR)
13826 return false;
13827 V = peekThroughBitcasts(V.getOperand(1));
13828 unsigned NumBits = V.getScalarValueSizeInBits();
13829 ConstantSDNode *C =
13830 isConstOrConstSplat(V, AllowUndefs, /*AllowTruncation*/ true);
13831 return C && (C->getAPIntValue().countr_one() >= NumBits);
13832}
13833
13835 bool AllowTruncation) {
13836 APInt DemandedElts = getDemandAllEltsMask(N);
13837 return isConstOrConstSplat(N, DemandedElts, AllowUndefs, AllowTruncation);
13838}
13839
13841 bool AllowUndefs,
13842 bool AllowTruncation) {
13844 return CN;
13845
13846 // SplatVectors can truncate their operands. Ignore that case here unless
13847 // AllowTruncation is set.
13848 if (N->getOpcode() == ISD::SPLAT_VECTOR) {
13849 EVT VecEltVT = N->getValueType(0).getVectorElementType();
13850 if (auto *CN = dyn_cast<ConstantSDNode>(N->getOperand(0))) {
13851 EVT CVT = CN->getValueType(0);
13852 assert(CVT.bitsGE(VecEltVT) && "Illegal splat_vector element extension");
13853 if (AllowTruncation || CVT == VecEltVT)
13854 return CN;
13855 }
13856 }
13857
13859 BitVector UndefElements;
13860 ConstantSDNode *CN = BV->getConstantSplatNode(DemandedElts, &UndefElements);
13861
13862 // BuildVectors can truncate their operands. Ignore that case here unless
13863 // AllowTruncation is set.
13864 // TODO: Look into whether we should allow UndefElements in non-DemandedElts
13865 if (CN && (UndefElements.none() || AllowUndefs)) {
13866 EVT CVT = CN->getValueType(0);
13867 EVT NSVT = N.getValueType().getScalarType();
13868 assert(CVT.bitsGE(NSVT) && "Illegal build vector element extension");
13869 if (AllowTruncation || (CVT == NSVT))
13870 return CN;
13871 }
13872 }
13873
13874 return nullptr;
13875}
13876
13878 APInt DemandedElts = getDemandAllEltsMask(N);
13879 return isConstOrConstSplatFP(N, DemandedElts, AllowUndefs);
13880}
13881
13883 const APInt &DemandedElts,
13884 bool AllowUndefs) {
13886 return CN;
13887
13889 BitVector UndefElements;
13890 ConstantFPSDNode *CN =
13891 BV->getConstantFPSplatNode(DemandedElts, &UndefElements);
13892 // TODO: Look into whether we should allow UndefElements in non-DemandedElts
13893 if (CN && (UndefElements.none() || AllowUndefs))
13894 return CN;
13895 }
13896
13897 if (N.getOpcode() == ISD::SPLAT_VECTOR)
13898 if (ConstantFPSDNode *CN = dyn_cast<ConstantFPSDNode>(N.getOperand(0)))
13899 return CN;
13900
13901 return nullptr;
13902}
13903
13904bool llvm::isNullOrNullSplat(SDValue N, bool AllowUndefs) {
13905 // TODO: may want to use peekThroughBitcast() here.
13906 ConstantSDNode *C =
13907 isConstOrConstSplat(N, AllowUndefs, /*AllowTruncation=*/true);
13908 return C && C->isZero();
13909}
13910
13911bool llvm::isOneOrOneSplat(SDValue N, bool AllowUndefs) {
13912 ConstantSDNode *C =
13913 isConstOrConstSplat(N, AllowUndefs, /*AllowTruncation*/ true);
13914 return C && C->isOne();
13915}
13916
13917bool llvm::isOneOrOneSplatFP(SDValue N, bool AllowUndefs) {
13918 ConstantFPSDNode *C = isConstOrConstSplatFP(N, AllowUndefs);
13919 return C && C->isOne();
13920}
13921
13922bool llvm::isAllOnesOrAllOnesSplat(SDValue N, bool AllowUndefs) {
13924 unsigned BitWidth = N.getScalarValueSizeInBits();
13925 ConstantSDNode *C =
13926 isConstOrConstSplat(N, AllowUndefs, /*AllowTruncation=*/true);
13927 return C && C->getAPIntValue().countTrailingOnes() >= BitWidth;
13928}
13929
13930bool llvm::isOnesOrOnesSplat(SDValue N, bool AllowUndefs) {
13931 ConstantSDNode *C = isConstOrConstSplat(N, AllowUndefs);
13932 return C && APInt::isSameValue(C->getAPIntValue(),
13933 APInt(C->getAPIntValue().getBitWidth(), 1));
13934}
13935
13936bool llvm::isZeroOrZeroSplat(SDValue N, bool AllowUndefs) {
13938 ConstantSDNode *C = isConstOrConstSplat(N, AllowUndefs, true);
13939 return C && C->isZero();
13940}
13941
13942bool llvm::isZeroOrZeroSplatFP(SDValue N, bool AllowUndefs) {
13943 ConstantFPSDNode *C = isConstOrConstSplatFP(N, AllowUndefs);
13944 return C && C->isZero();
13945}
13946
13950
13952 unsigned Opc, unsigned Order, const DebugLoc &dl, SDVTList VTs, EVT memvt,
13954 : SDNode(Opc, Order, dl, VTs), MemoryVT(memvt), MemRefs(memrefs) {
13955 bool IsVolatile = false;
13956 bool IsNonTemporal = false;
13957 bool IsDereferenceable = true;
13958 bool IsInvariant = true;
13959 for (const MachineMemOperand *MMO : memoperands()) {
13960 IsVolatile |= MMO->isVolatile();
13961 IsNonTemporal |= MMO->isNonTemporal();
13962 IsDereferenceable &= MMO->isDereferenceable();
13963 IsInvariant &= MMO->isInvariant();
13964 }
13965 MemSDNodeBits.IsVolatile = IsVolatile;
13966 MemSDNodeBits.IsNonTemporal = IsNonTemporal;
13967 MemSDNodeBits.IsDereferenceable = IsDereferenceable;
13968 MemSDNodeBits.IsInvariant = IsInvariant;
13969
13970 // For the single-MMO case, we check here that the size of the memory operand
13971 // fits within the size of the MMO. This is because the MMO might indicate
13972 // only a possible address range instead of specifying the affected memory
13973 // addresses precisely.
13976 getMemOperand()->getSize().getValue())) &&
13977 "Size mismatch!");
13978}
13979
13980namespace {
13981
13982 struct EVTArray {
13983 std::vector<EVT> VTs;
13984
13985 EVTArray() {
13986 VTs.reserve(MVT::VALUETYPE_SIZE);
13987 for (unsigned i = 0; i < MVT::VALUETYPE_SIZE; ++i)
13988 VTs.push_back(MVT((MVT::SimpleValueType)i));
13989 }
13990 };
13991
13992} // end anonymous namespace
13993
13994/// getValueTypeList - Return a pointer to the specified value type.
13995///
13996const EVT *SDNode::getValueTypeList(MVT VT) {
13997 static EVTArray SimpleVTArray;
13998
13999 assert(VT < MVT::VALUETYPE_SIZE && "Value type out of range!");
14000 return &SimpleVTArray.VTs[VT.SimpleTy];
14001}
14002
14003/// hasAnyUseOfValue - Return true if there are any use of the indicated
14004/// value. This method ignores uses of other values defined by this operation.
14005bool SDNode::hasAnyUseOfValue(unsigned Value) const {
14006 assert(Value < getNumValues() && "Bad value!");
14007
14008 for (SDUse &U : uses())
14009 if (U.getResNo() == Value)
14010 return true;
14011
14012 return false;
14013}
14014
14015/// isOnlyUserOf - Return true if this node is the only use of N.
14016bool SDNode::isOnlyUserOf(const SDNode *N) const {
14017 bool Seen = false;
14018 for (const SDNode *User : N->users()) {
14019 if (User == this)
14020 Seen = true;
14021 else
14022 return false;
14023 }
14024
14025 return Seen;
14026}
14027
14028/// Return true if the only users of N are contained in Nodes.
14030 bool Seen = false;
14031 for (const SDNode *User : N->users()) {
14032 if (llvm::is_contained(Nodes, User))
14033 Seen = true;
14034 else
14035 return false;
14036 }
14037
14038 return Seen;
14039}
14040
14041/// Return true if the referenced return value is an operand of N.
14042bool SDValue::isOperandOf(const SDNode *N) const {
14043 return is_contained(N->op_values(), *this);
14044}
14045
14046bool SDNode::isOperandOf(const SDNode *N) const {
14047 return any_of(N->op_values(),
14048 [this](SDValue Op) { return this == Op.getNode(); });
14049}
14050
14051/// reachesChainWithoutSideEffects - Return true if this operand (which must
14052/// be a chain) reaches the specified operand without crossing any
14053/// side-effecting instructions on any chain path. In practice, this looks
14054/// through token factors and non-volatile loads. In order to remain efficient,
14055/// this only looks a couple of nodes in, it does not do an exhaustive search.
14056///
14057/// Note that we only need to examine chains when we're searching for
14058/// side-effects; SelectionDAG requires that all side-effects are represented
14059/// by chains, even if another operand would force a specific ordering. This
14060/// constraint is necessary to allow transformations like splitting loads.
14062 unsigned Depth) const {
14063 if (*this == Dest) return true;
14064
14065 // Don't search too deeply, we just want to be able to see through
14066 // TokenFactor's etc.
14067 if (Depth == 0) return false;
14068
14069 // If this is a token factor, all inputs to the TF happen in parallel.
14070 if (getOpcode() == ISD::TokenFactor) {
14071 // First, try a shallow search.
14072 if (is_contained((*this)->ops(), Dest)) {
14073 // We found the chain we want as an operand of this TokenFactor.
14074 // Essentially, we reach the chain without side-effects if we could
14075 // serialize the TokenFactor into a simple chain of operations with
14076 // Dest as the last operation. This is automatically true if the
14077 // chain has one use: there are no other ordering constraints.
14078 // If the chain has more than one use, we give up: some other
14079 // use of Dest might force a side-effect between Dest and the current
14080 // node.
14081 if (Dest.hasOneUse())
14082 return true;
14083 }
14084 // Next, try a deep search: check whether every operand of the TokenFactor
14085 // reaches Dest.
14086 return llvm::all_of((*this)->ops(), [=](SDValue Op) {
14087 return Op.reachesChainWithoutSideEffects(Dest, Depth - 1);
14088 });
14089 }
14090
14091 // Loads don't have side effects, look through them.
14092 if (LoadSDNode *Ld = dyn_cast<LoadSDNode>(*this)) {
14093 if (Ld->isUnordered())
14094 return Ld->getChain().reachesChainWithoutSideEffects(Dest, Depth-1);
14095 }
14096 return false;
14097}
14098
14099bool SDNode::hasPredecessor(const SDNode *N) const {
14102 Worklist.push_back(this);
14103 return hasPredecessorHelper(N, Visited, Worklist);
14104}
14105
14107 this->Flags &= Flags;
14108}
14109
14110SDValue
14112 ArrayRef<ISD::NodeType> CandidateBinOps,
14113 bool AllowPartials) {
14114 // The pattern must end in an extract from index 0.
14115 if (Extract->getOpcode() != ISD::EXTRACT_VECTOR_ELT ||
14116 !isNullConstant(Extract->getOperand(1)))
14117 return SDValue();
14118
14119 // Match against one of the candidate binary ops.
14120 SDValue Op = Extract->getOperand(0);
14121 if (llvm::none_of(CandidateBinOps, [Op](ISD::NodeType BinOp) {
14122 return Op.getOpcode() == unsigned(BinOp);
14123 }))
14124 return SDValue();
14125
14126 // Floating-point reductions may require relaxed constraints on the final step
14127 // of the reduction because they may reorder intermediate operations.
14128 unsigned CandidateBinOp = Op.getOpcode();
14129 if (Op.getValueType().isFloatingPoint()) {
14130 SDNodeFlags Flags = Op->getFlags();
14131 switch (CandidateBinOp) {
14132 case ISD::FADD:
14133 if (!Flags.hasNoSignedZeros() || !Flags.hasAllowReassociation())
14134 return SDValue();
14135 break;
14136 default:
14137 llvm_unreachable("Unhandled FP opcode for binop reduction");
14138 }
14139 }
14140
14141 // Matching failed - attempt to see if we did enough stages that a partial
14142 // reduction from a subvector is possible.
14143 auto PartialReduction = [&](SDValue Op, unsigned NumSubElts) {
14144 if (!AllowPartials || !Op)
14145 return SDValue();
14146 EVT OpVT = Op.getValueType();
14147 EVT OpSVT = OpVT.getScalarType();
14148 EVT SubVT = EVT::getVectorVT(*getContext(), OpSVT, NumSubElts);
14149 if (TLI->getExtractSubvectorCost(SubVT, OpVT, 0) >
14151 return SDValue();
14152 BinOp = (ISD::NodeType)CandidateBinOp;
14153 return getExtractSubvector(SDLoc(Op), SubVT, Op, 0);
14154 };
14155
14156 // At each stage, we're looking for something that looks like:
14157 // %s = shufflevector <8 x i32> %op, <8 x i32> undef,
14158 // <8 x i32> <i32 2, i32 3, i32 undef, i32 undef,
14159 // i32 undef, i32 undef, i32 undef, i32 undef>
14160 // %a = binop <8 x i32> %op, %s
14161 // Where the mask changes according to the stage. E.g. for a 3-stage pyramid,
14162 // we expect something like:
14163 // <4,5,6,7,u,u,u,u>
14164 // <2,3,u,u,u,u,u,u>
14165 // <1,u,u,u,u,u,u,u>
14166 // While a partial reduction match would be:
14167 // <2,3,u,u,u,u,u,u>
14168 // <1,u,u,u,u,u,u,u>
14169 unsigned Stages = Log2_32(Op.getValueType().getVectorNumElements());
14170 SDValue PrevOp;
14171 for (unsigned i = 0; i < Stages; ++i) {
14172 unsigned MaskEnd = (1 << i);
14173
14174 if (Op.getOpcode() != CandidateBinOp)
14175 return PartialReduction(PrevOp, MaskEnd);
14176
14177 SDValue Op0 = Op.getOperand(0);
14178 SDValue Op1 = Op.getOperand(1);
14179
14181 if (Shuffle) {
14182 Op = Op1;
14183 } else {
14184 Shuffle = dyn_cast<ShuffleVectorSDNode>(Op1);
14185 Op = Op0;
14186 }
14187
14188 // The first operand of the shuffle should be the same as the other operand
14189 // of the binop.
14190 if (!Shuffle || Shuffle->getOperand(0) != Op)
14191 return PartialReduction(PrevOp, MaskEnd);
14192
14193 // Verify the shuffle has the expected (at this stage of the pyramid) mask.
14194 for (int Index = 0; Index < (int)MaskEnd; ++Index)
14195 if (Shuffle->getMaskElt(Index) != (int)(MaskEnd + Index))
14196 return PartialReduction(PrevOp, MaskEnd);
14197
14198 PrevOp = Op;
14199 }
14200
14201 // Handle subvector reductions, which tend to appear after the shuffle
14202 // reduction stages.
14203 while (Op.getOpcode() == CandidateBinOp) {
14204 unsigned NumElts = Op.getValueType().getVectorNumElements();
14205 SDValue Op0 = Op.getOperand(0);
14206 SDValue Op1 = Op.getOperand(1);
14207 if (Op0.getOpcode() != ISD::EXTRACT_SUBVECTOR ||
14209 Op0.getOperand(0) != Op1.getOperand(0))
14210 break;
14211 SDValue Src = Op0.getOperand(0);
14212 unsigned NumSrcElts = Src.getValueType().getVectorNumElements();
14213 if (NumSrcElts != (2 * NumElts))
14214 break;
14215 if (!(Op0.getConstantOperandAPInt(1) == 0 &&
14216 Op1.getConstantOperandAPInt(1) == NumElts) &&
14217 !(Op1.getConstantOperandAPInt(1) == 0 &&
14218 Op0.getConstantOperandAPInt(1) == NumElts))
14219 break;
14220 Op = Src;
14221 }
14222
14223 BinOp = (ISD::NodeType)CandidateBinOp;
14224 return Op;
14225}
14226
14228 EVT VT = N->getValueType(0);
14229 EVT EltVT = VT.getVectorElementType();
14230 unsigned NE = getMaxRuntimeNumElements(VT);
14231
14232 if (VT.isScalableVector() && (NE == 0 || ResNE != 0))
14233 reportFatalUsageError("Cannot unroll scalable vector!");
14234
14235 assert(NE && "Nothing to unroll!");
14236 SDLoc dl(N);
14237
14238 // If ResNE is 0, fully unroll the vector op.
14239 if (ResNE == 0)
14240 ResNE = NE;
14241 else if (NE > ResNE)
14242 NE = ResNE;
14243
14244 if (N->getNumValues() == 2) {
14245 SmallVector<SDValue, 8> Scalars0, Scalars1;
14246 SmallVector<SDValue, 4> Operands(N->getNumOperands());
14247 EVT VT1 = N->getValueType(1);
14248 EVT EltVT1 = VT1.getVectorElementType();
14249
14250 unsigned i;
14251 for (i = 0; i != NE; ++i) {
14252 for (unsigned j = 0, e = N->getNumOperands(); j != e; ++j) {
14253 SDValue Operand = N->getOperand(j);
14254 EVT OperandVT = Operand.getValueType();
14255
14256 // A vector operand; extract a single element.
14257 EVT OperandEltVT = OperandVT.getVectorElementType();
14258 Operands[j] = getExtractVectorElt(dl, OperandEltVT, Operand, i);
14259 }
14260
14261 SDValue EltOp = getNode(N->getOpcode(), dl, {EltVT, EltVT1}, Operands);
14262 Scalars0.push_back(EltOp);
14263 Scalars1.push_back(EltOp.getValue(1));
14264 }
14265
14266 for (; i < ResNE; ++i) {
14267 Scalars0.push_back(getUNDEF(EltVT));
14268 Scalars1.push_back(getUNDEF(EltVT1));
14269 }
14270
14272 : ElementCount::getFixed(ResNE);
14273 EVT VecVT = EVT::getVectorVT(*getContext(), EltVT, ResEC);
14274 EVT VecVT1 = EVT::getVectorVT(*getContext(), EltVT1, ResEC);
14275 SDValue Vec0 = buildVectorFromUnrolledParts(VecVT, dl, Scalars0);
14276 SDValue Vec1 = buildVectorFromUnrolledParts(VecVT1, dl, Scalars1);
14277 return getMergeValues({Vec0, Vec1}, dl);
14278 }
14279
14280 assert(N->getNumValues() == 1 &&
14281 "Can't unroll a vector with multiple results!");
14282
14284 SmallVector<SDValue, 4> Operands(N->getNumOperands());
14285
14286 unsigned i;
14287 for (i= 0; i != NE; ++i) {
14288 for (unsigned j = 0, e = N->getNumOperands(); j != e; ++j) {
14289 SDValue Operand = N->getOperand(j);
14290 EVT OperandVT = Operand.getValueType();
14291 if (OperandVT.isVector()) {
14292 // A vector operand; extract a single element.
14293 EVT OperandEltVT = OperandVT.getVectorElementType();
14294 Operands[j] = getExtractVectorElt(dl, OperandEltVT, Operand, i);
14295 } else {
14296 // A scalar operand; just use it as is.
14297 Operands[j] = Operand;
14298 }
14299 }
14300
14301 switch (N->getOpcode()) {
14302 default: {
14303 Scalars.push_back(getNode(N->getOpcode(), dl, EltVT, Operands,
14304 N->getFlags()));
14305 break;
14306 }
14307 case ISD::VSELECT:
14308 Scalars.push_back(
14309 getNode(ISD::SELECT, dl, EltVT, Operands, N->getFlags()));
14310 break;
14311 case ISD::SHL:
14312 case ISD::SRA:
14313 case ISD::SRL:
14314 case ISD::ROTL:
14315 case ISD::ROTR:
14316 Scalars.push_back(getNode(N->getOpcode(), dl, EltVT, Operands[0],
14318 Operands[1])));
14319 break;
14321 EVT ExtVT = cast<VTSDNode>(Operands[1])->getVT().getVectorElementType();
14322 Scalars.push_back(getNode(N->getOpcode(), dl, EltVT,
14323 Operands[0],
14324 getValueType(ExtVT)));
14325 break;
14326 }
14327 case ISD::ADDRSPACECAST: {
14328 const auto *ASC = cast<AddrSpaceCastSDNode>(N);
14329 Scalars.push_back(
14330 getAddrSpaceCast(dl, EltVT, Operands[0], ASC->getSrcAddressSpace(),
14331 ASC->getDestAddressSpace(), ASC->getFlags()));
14332 break;
14333 }
14334 }
14335 }
14336
14337 for (; i < ResNE; ++i)
14338 Scalars.push_back(getUNDEF(EltVT));
14339
14340 EVT VecVT = VT.isScalableVector()
14341 ? VT
14342 : EVT::getVectorVT(*getContext(), EltVT, ResNE);
14343 return buildVectorFromUnrolledParts(VecVT, dl, Scalars);
14344}
14345
14346std::pair<SDValue, SDValue> SelectionDAG::UnrollVectorOverflowOp(
14347 SDNode *N, unsigned ResNE) {
14348 unsigned Opcode = N->getOpcode();
14349 assert((Opcode == ISD::UADDO || Opcode == ISD::SADDO ||
14350 Opcode == ISD::USUBO || Opcode == ISD::SSUBO ||
14351 Opcode == ISD::UMULO || Opcode == ISD::SMULO) &&
14352 "Expected an overflow opcode");
14353
14354 EVT ResVT = N->getValueType(0);
14355 EVT OvVT = N->getValueType(1);
14356 EVT ResEltVT = ResVT.getVectorElementType();
14357 EVT OvEltVT = OvVT.getVectorElementType();
14358 SDLoc dl(N);
14359
14360 // If ResNE is 0, fully unroll the vector op.
14361 unsigned NE = ResVT.getVectorNumElements();
14362 if (ResNE == 0)
14363 ResNE = NE;
14364 else if (NE > ResNE)
14365 NE = ResNE;
14366
14367 SmallVector<SDValue, 8> LHSScalars;
14368 SmallVector<SDValue, 8> RHSScalars;
14369 ExtractVectorElements(N->getOperand(0), LHSScalars, 0, NE);
14370 ExtractVectorElements(N->getOperand(1), RHSScalars, 0, NE);
14371
14372 EVT SVT = TLI->getSetCCResultType(getDataLayout(), *getContext(), ResEltVT);
14373 SDVTList VTs = getVTList(ResEltVT, SVT);
14374 SmallVector<SDValue, 8> ResScalars;
14375 SmallVector<SDValue, 8> OvScalars;
14376 for (unsigned i = 0; i < NE; ++i) {
14377 SDValue Res = getNode(Opcode, dl, VTs, LHSScalars[i], RHSScalars[i]);
14378 SDValue Ov =
14379 getSelect(dl, OvEltVT, Res.getValue(1),
14380 getBoolConstant(true, dl, OvEltVT, ResVT),
14381 getConstant(0, dl, OvEltVT));
14382
14383 ResScalars.push_back(Res);
14384 OvScalars.push_back(Ov);
14385 }
14386
14387 ResScalars.append(ResNE - NE, getUNDEF(ResEltVT));
14388 OvScalars.append(ResNE - NE, getUNDEF(OvEltVT));
14389
14390 EVT NewResVT = EVT::getVectorVT(*getContext(), ResEltVT, ResNE);
14391 EVT NewOvVT = EVT::getVectorVT(*getContext(), OvEltVT, ResNE);
14392 return std::make_pair(getBuildVector(NewResVT, dl, ResScalars),
14393 getBuildVector(NewOvVT, dl, OvScalars));
14394}
14395
14398 unsigned Bytes, int Dist,
14399 const SelectionDAG &DAG) {
14400 if (LS->isVolatile() || Base->isVolatile())
14401 return false;
14402 // TODO: probably too restrictive for atomics, revisit
14403 if (!LS->isSimple())
14404 return false;
14405 if (LS->isIndexed() || Base->isIndexed())
14406 return false;
14407 if (LS->getChain() != Base->getChain())
14408 return false;
14409 EVT VT = LS->getMemoryVT();
14410 if (VT.getSizeInBits() / 8 != Bytes)
14411 return false;
14412
14413 auto BaseLocDecomp = BaseIndexOffset::match(Base, DAG);
14414 auto LocDecomp = BaseIndexOffset::match(LS, DAG);
14415
14416 int64_t Offset = 0;
14417 if (BaseLocDecomp.equalBaseIndex(LocDecomp, DAG, Offset))
14418 return (Dist * (int64_t)Bytes == Offset);
14419 return false;
14420}
14421
14424 unsigned Bytes,
14425 int Dist) const {
14426 return areNonVolatileConsecutiveLoadsOrStores(LD, Base, Bytes, Dist, *this);
14427}
14428
14431 unsigned Bytes,
14432 int Dist) const {
14433 return areNonVolatileConsecutiveLoadsOrStores(ST, Base, Bytes, Dist, *this);
14434}
14435
14436/// InferPtrAlignment - Infer alignment of a load / store address. Return
14437/// std::nullopt if it cannot be inferred.
14439 // If this is a GlobalAddress + cst, return the alignment.
14440 const GlobalValue *GV = nullptr;
14441 int64_t GVOffset = 0;
14442 if (TLI->isGAPlusOffset(Ptr.getNode(), GV, GVOffset)) {
14443 unsigned PtrWidth = getDataLayout().getPointerTypeSizeInBits(GV->getType());
14444 KnownBits Known(PtrWidth);
14446 unsigned AlignBits = Known.countMinTrailingZeros();
14447 if (AlignBits)
14448 return commonAlignment(Align(1ull << std::min(31U, AlignBits)), GVOffset);
14449 }
14450
14451 // If this is a direct reference to a stack slot, use information about the
14452 // stack slot's alignment.
14453 int FrameIdx = INT_MIN;
14454 int64_t FrameOffset = 0;
14456 FrameIdx = FI->getIndex();
14457 } else if (isBaseWithConstantOffset(Ptr) &&
14459 // Handle FI+Cst
14460 FrameIdx = cast<FrameIndexSDNode>(Ptr.getOperand(0))->getIndex();
14461 FrameOffset = Ptr.getConstantOperandVal(1);
14462 }
14463
14464 if (FrameIdx != INT_MIN) {
14466 return commonAlignment(MFI.getObjectAlign(FrameIdx), FrameOffset);
14467 }
14468
14469 return std::nullopt;
14470}
14471
14472/// Split the scalar node with EXTRACT_ELEMENT using the provided
14473/// VTs and return the low/high part.
14474std::pair<SDValue, SDValue> SelectionDAG::SplitScalar(const SDValue &N,
14475 const SDLoc &DL,
14476 const EVT &LoVT,
14477 const EVT &HiVT) {
14478 assert(!LoVT.isVector() && !HiVT.isVector() && !N.getValueType().isVector() &&
14479 "Split node must be a scalar type");
14480 SDValue Lo =
14482 SDValue Hi =
14484 return std::make_pair(Lo, Hi);
14485}
14486
14487/// GetSplitDestVTs - Compute the VTs needed for the low/hi parts of a type
14488/// which is split (or expanded) into two not necessarily identical pieces.
14489std::pair<EVT, EVT> SelectionDAG::GetSplitDestVTs(const EVT &VT) const {
14490 // Currently all types are split in half.
14491 EVT LoVT, HiVT;
14492 if (!VT.isVector())
14493 LoVT = HiVT = TLI->getTypeToTransformTo(*getContext(), VT);
14494 else
14495 LoVT = HiVT = VT.getHalfNumVectorElementsVT(*getContext());
14496
14497 return std::make_pair(LoVT, HiVT);
14498}
14499
14500/// GetDependentSplitDestVTs - Compute the VTs needed for the low/hi parts of a
14501/// type, dependent on an enveloping VT that has been split into two identical
14502/// pieces. Sets the HiIsEmpty flag when hi type has zero storage size.
14503std::pair<EVT, EVT>
14505 bool *HiIsEmpty) const {
14506 EVT EltTp = VT.getVectorElementType();
14507 // Examples:
14508 // custom VL=8 with enveloping VL=8/8 yields 8/0 (hi empty)
14509 // custom VL=9 with enveloping VL=8/8 yields 8/1
14510 // custom VL=10 with enveloping VL=8/8 yields 8/2
14511 // etc.
14512 ElementCount VTNumElts = VT.getVectorElementCount();
14513 ElementCount EnvNumElts = EnvVT.getVectorElementCount();
14514 assert(VTNumElts.isScalable() == EnvNumElts.isScalable() &&
14515 "Mixing fixed width and scalable vectors when enveloping a type");
14516 EVT LoVT, HiVT;
14517 if (VTNumElts.getKnownMinValue() > EnvNumElts.getKnownMinValue()) {
14518 LoVT = EVT::getVectorVT(*getContext(), EltTp, EnvNumElts);
14519 HiVT = EVT::getVectorVT(*getContext(), EltTp, VTNumElts - EnvNumElts);
14520 *HiIsEmpty = false;
14521 } else {
14522 // Flag that hi type has zero storage size, but return split envelop type
14523 // (this would be easier if vector types with zero elements were allowed).
14524 LoVT = EVT::getVectorVT(*getContext(), EltTp, VTNumElts);
14525 HiVT = EVT::getVectorVT(*getContext(), EltTp, EnvNumElts);
14526 *HiIsEmpty = true;
14527 }
14528 return std::make_pair(LoVT, HiVT);
14529}
14530
14531/// SplitVector - Split the vector with EXTRACT_SUBVECTOR and return the
14532/// low/high part.
14533std::pair<SDValue, SDValue>
14534SelectionDAG::SplitVector(const SDValue &N, const SDLoc &DL, const EVT &LoVT,
14535 const EVT &HiVT) {
14536 assert(LoVT.isScalableVector() == HiVT.isScalableVector() &&
14537 LoVT.isScalableVector() == N.getValueType().isScalableVector() &&
14538 "Splitting vector with an invalid mixture of fixed and scalable "
14539 "vector types");
14541 N.getValueType().getVectorMinNumElements() &&
14542 "More vector elements requested than available!");
14543 SDValue Lo, Hi;
14544 Lo = getExtractSubvector(DL, LoVT, N, 0);
14545 // For scalable vectors it is safe to use LoVT.getVectorMinNumElements()
14546 // (rather than having to use ElementCount), because EXTRACT_SUBVECTOR scales
14547 // IDX with the runtime scaling factor of the result vector type. For
14548 // fixed-width result vectors, that runtime scaling factor is 1.
14550 return std::make_pair(Lo, Hi);
14551}
14552
14553std::pair<SDValue, SDValue> SelectionDAG::SplitEVL(SDValue N, EVT VecVT,
14554 const SDLoc &DL) {
14555 // Split the vector length parameter.
14556 // %evl -> umin(%evl, %halfnumelts) and usubsat(%evl - %halfnumelts).
14557 EVT VT = N.getValueType();
14559 "Expecting the mask to be an evenly-sized vector");
14560 SDValue HalfNumElts = getElementCount(
14562 SDValue Lo = getNode(ISD::UMIN, DL, VT, N, HalfNumElts);
14563 SDValue Hi = getNode(ISD::USUBSAT, DL, VT, N, HalfNumElts);
14564 return std::make_pair(Lo, Hi);
14565}
14566
14567/// Widen the vector up to the next power of two using INSERT_SUBVECTOR.
14569 EVT VT = N.getValueType();
14572 return getInsertSubvector(DL, getPOISON(WideVT), N, 0);
14573}
14574
14577 unsigned Start, unsigned Count,
14578 EVT EltVT) {
14579 EVT VT = Op.getValueType();
14580 if (Count == 0)
14582 if (EltVT == EVT())
14583 EltVT = VT.getVectorElementType();
14584 SDLoc SL(Op);
14585 for (unsigned i = Start, e = Start + Count; i != e; ++i) {
14586 Args.push_back(getExtractVectorElt(SL, EltVT, Op, i));
14587 }
14588}
14589
14590// getAddressSpace - Return the address space this GlobalAddress belongs to.
14592 return getGlobal()->getType()->getAddressSpace();
14593}
14594
14597 return Val.MachineCPVal->getType();
14598 return Val.ConstVal->getType();
14599}
14600
14601bool BuildVectorSDNode::isConstantSplat(APInt &SplatValue, APInt &SplatUndef,
14602 unsigned &SplatBitSize,
14603 bool &HasAnyUndefs,
14604 unsigned MinSplatBits,
14605 bool IsBigEndian) const {
14606 EVT VT = getValueType(0);
14607 assert(VT.isVector() && "Expected a vector type");
14608 unsigned VecWidth = VT.getSizeInBits();
14609 if (MinSplatBits > VecWidth)
14610 return false;
14611
14612 // FIXME: The widths are based on this node's type, but build vectors can
14613 // truncate their operands.
14614 SplatValue = APInt(VecWidth, 0);
14615 SplatUndef = APInt(VecWidth, 0);
14616
14617 // Get the bits. Bits with undefined values (when the corresponding element
14618 // of the vector is an ISD::UNDEF value) are set in SplatUndef and cleared
14619 // in SplatValue. If any of the values are not constant, give up and return
14620 // false.
14621 unsigned int NumOps = getNumOperands();
14622 assert(NumOps > 0 && "isConstantSplat has 0-size build vector");
14623 unsigned EltWidth = VT.getScalarSizeInBits();
14624
14625 for (unsigned j = 0; j < NumOps; ++j) {
14626 unsigned i = IsBigEndian ? NumOps - 1 - j : j;
14627 SDValue OpVal = getOperand(i);
14628 unsigned BitPos = j * EltWidth;
14629
14630 if (OpVal.isUndef())
14631 SplatUndef.setBits(BitPos, BitPos + EltWidth);
14632 else if (auto *CN = dyn_cast<ConstantSDNode>(OpVal))
14633 SplatValue.insertBits(CN->getAPIntValue().zextOrTrunc(EltWidth), BitPos);
14634 else if (auto *CN = dyn_cast<ConstantFPSDNode>(OpVal))
14635 SplatValue.insertBits(CN->getValueAPF().bitcastToAPInt(), BitPos);
14636 else
14637 return false;
14638 }
14639
14640 // The build_vector is all constants or undefs. Find the smallest element
14641 // size that splats the vector.
14642 HasAnyUndefs = (SplatUndef != 0);
14643
14644 // FIXME: This does not work for vectors with elements less than 8 bits.
14645 while (VecWidth > 8) {
14646 // If we can't split in half, stop here.
14647 if (VecWidth & 1)
14648 break;
14649
14650 unsigned HalfSize = VecWidth / 2;
14651 APInt HighValue = SplatValue.extractBits(HalfSize, HalfSize);
14652 APInt LowValue = SplatValue.extractBits(HalfSize, 0);
14653 APInt HighUndef = SplatUndef.extractBits(HalfSize, HalfSize);
14654 APInt LowUndef = SplatUndef.extractBits(HalfSize, 0);
14655
14656 // If the two halves do not match (ignoring undef bits), stop here.
14657 if ((HighValue & ~LowUndef) != (LowValue & ~HighUndef) ||
14658 MinSplatBits > HalfSize)
14659 break;
14660
14661 SplatValue = HighValue | LowValue;
14662 SplatUndef = HighUndef & LowUndef;
14663
14664 VecWidth = HalfSize;
14665 }
14666
14667 // FIXME: The loop above only tries to split in halves. But if the input
14668 // vector for example is <3 x i16> it wouldn't be able to detect a
14669 // SplatBitSize of 16. No idea if that is a design flaw currently limiting
14670 // optimizations. I guess that back in the days when this helper was created
14671 // vectors normally was power-of-2 sized.
14672
14673 SplatBitSize = VecWidth;
14674 return true;
14675}
14676
14678 BitVector *UndefElements) const {
14679 unsigned NumOps = getNumOperands();
14680 if (UndefElements) {
14681 UndefElements->clear();
14682 UndefElements->resize(NumOps);
14683 }
14684 assert(NumOps == DemandedElts.getBitWidth() && "Unexpected vector size");
14685 if (!DemandedElts)
14686 return SDValue();
14687 SDValue Splatted;
14688 for (unsigned i = 0; i != NumOps; ++i) {
14689 if (!DemandedElts[i])
14690 continue;
14691 SDValue Op = getOperand(i);
14692 if (Op.isUndef()) {
14693 if (UndefElements)
14694 (*UndefElements)[i] = true;
14695 } else if (!Splatted) {
14696 Splatted = Op;
14697 } else if (Splatted != Op) {
14698 return SDValue();
14699 }
14700 }
14701
14702 if (!Splatted) {
14703 unsigned FirstDemandedIdx = DemandedElts.countr_zero();
14704 assert(getOperand(FirstDemandedIdx).isUndef() &&
14705 "Can only have a splat without a constant for all undefs.");
14706 return getOperand(FirstDemandedIdx);
14707 }
14708
14709 return Splatted;
14710}
14711
14713 APInt DemandedElts = APInt::getAllOnes(getNumOperands());
14714 return getSplatValue(DemandedElts, UndefElements);
14715}
14716
14718 SmallVectorImpl<SDValue> &Sequence,
14719 BitVector *UndefElements) const {
14720 unsigned NumOps = getNumOperands();
14721 Sequence.clear();
14722 if (UndefElements) {
14723 UndefElements->clear();
14724 UndefElements->resize(NumOps);
14725 }
14726 assert(NumOps == DemandedElts.getBitWidth() && "Unexpected vector size");
14727 if (!DemandedElts || NumOps < 2 || !isPowerOf2_32(NumOps))
14728 return false;
14729
14730 // Set the undefs even if we don't find a sequence (like getSplatValue).
14731 if (UndefElements)
14732 for (unsigned I = 0; I != NumOps; ++I)
14733 if (DemandedElts[I] && getOperand(I).isUndef())
14734 (*UndefElements)[I] = true;
14735
14736 // Iteratively widen the sequence length looking for repetitions.
14737 for (unsigned SeqLen = 1; SeqLen < NumOps; SeqLen *= 2) {
14738 Sequence.append(SeqLen, SDValue());
14739 for (unsigned I = 0; I != NumOps; ++I) {
14740 if (!DemandedElts[I])
14741 continue;
14742 SDValue &SeqOp = Sequence[I % SeqLen];
14744 if (Op.isUndef()) {
14745 if (!SeqOp)
14746 SeqOp = Op;
14747 continue;
14748 }
14749 if (SeqOp && !SeqOp.isUndef() && SeqOp != Op) {
14750 Sequence.clear();
14751 break;
14752 }
14753 SeqOp = Op;
14754 }
14755 if (!Sequence.empty())
14756 return true;
14757 }
14758
14759 assert(Sequence.empty() && "Failed to empty non-repeating sequence pattern");
14760 return false;
14761}
14762
14764 BitVector *UndefElements) const {
14765 APInt DemandedElts = APInt::getAllOnes(getNumOperands());
14766 return getRepeatedSequence(DemandedElts, Sequence, UndefElements);
14767}
14768
14771 BitVector *UndefElements) const {
14773 getSplatValue(DemandedElts, UndefElements));
14774}
14775
14778 return dyn_cast_or_null<ConstantSDNode>(getSplatValue(UndefElements));
14779}
14780
14783 BitVector *UndefElements) const {
14785 getSplatValue(DemandedElts, UndefElements));
14786}
14787
14792
14793int32_t
14795 uint32_t BitWidth) const {
14796 if (ConstantFPSDNode *CN =
14798 bool IsExact;
14799 APSInt IntVal(BitWidth);
14800 const APFloat &APF = CN->getValueAPF();
14801 if (APF.convertToInteger(IntVal, APFloat::rmTowardZero, &IsExact) !=
14802 APFloat::opOK ||
14803 !IsExact)
14804 return -1;
14805
14806 return IntVal.exactLogBase2();
14807 }
14808 return -1;
14809}
14810
14812 bool IsLittleEndian, unsigned DstEltSizeInBits,
14813 SmallVectorImpl<APInt> &RawBitElements, BitVector &UndefElements) const {
14814 // Early-out if this contains anything but Undef/Constant/ConstantFP.
14815 if (!isConstant())
14816 return false;
14817
14818 unsigned NumSrcOps = getNumOperands();
14819 unsigned SrcEltSizeInBits = getValueType(0).getScalarSizeInBits();
14820 assert(((NumSrcOps * SrcEltSizeInBits) % DstEltSizeInBits) == 0 &&
14821 "Invalid bitcast scale");
14822
14823 // Extract raw src bits.
14824 SmallVector<APInt> SrcBitElements(NumSrcOps,
14825 APInt::getZero(SrcEltSizeInBits));
14826 BitVector SrcUndeElements(NumSrcOps, false);
14827
14828 for (unsigned I = 0; I != NumSrcOps; ++I) {
14830 if (Op.isUndef()) {
14831 SrcUndeElements.set(I);
14832 continue;
14833 }
14834 auto *CInt = dyn_cast<ConstantSDNode>(Op);
14835 auto *CFP = dyn_cast<ConstantFPSDNode>(Op);
14836 assert((CInt || CFP) && "Unknown constant");
14837 SrcBitElements[I] = CInt ? CInt->getAPIntValue().trunc(SrcEltSizeInBits)
14838 : CFP->getValueAPF().bitcastToAPInt();
14839 }
14840
14841 // Recast to dst width.
14842 recastRawBits(IsLittleEndian, DstEltSizeInBits, RawBitElements,
14843 SrcBitElements, UndefElements, SrcUndeElements);
14844 return true;
14845}
14846
14847void BuildVectorSDNode::recastRawBits(bool IsLittleEndian,
14848 unsigned DstEltSizeInBits,
14849 SmallVectorImpl<APInt> &DstBitElements,
14850 ArrayRef<APInt> SrcBitElements,
14851 BitVector &DstUndefElements,
14852 const BitVector &SrcUndefElements) {
14853 unsigned NumSrcOps = SrcBitElements.size();
14854 unsigned SrcEltSizeInBits = SrcBitElements[0].getBitWidth();
14855 assert(((NumSrcOps * SrcEltSizeInBits) % DstEltSizeInBits) == 0 &&
14856 "Invalid bitcast scale");
14857 assert(NumSrcOps == SrcUndefElements.size() &&
14858 "Vector size mismatch");
14859
14860 unsigned NumDstOps = (NumSrcOps * SrcEltSizeInBits) / DstEltSizeInBits;
14861 DstUndefElements.clear();
14862 DstUndefElements.resize(NumDstOps, false);
14863 DstBitElements.assign(NumDstOps, APInt::getZero(DstEltSizeInBits));
14864
14865 // Concatenate src elements constant bits together into dst element.
14866 if (SrcEltSizeInBits <= DstEltSizeInBits) {
14867 unsigned Scale = DstEltSizeInBits / SrcEltSizeInBits;
14868 for (unsigned I = 0; I != NumDstOps; ++I) {
14869 DstUndefElements.set(I);
14870 APInt &DstBits = DstBitElements[I];
14871 for (unsigned J = 0; J != Scale; ++J) {
14872 unsigned Idx = (I * Scale) + (IsLittleEndian ? J : (Scale - J - 1));
14873 if (SrcUndefElements[Idx])
14874 continue;
14875 DstUndefElements.reset(I);
14876 const APInt &SrcBits = SrcBitElements[Idx];
14877 assert(SrcBits.getBitWidth() == SrcEltSizeInBits &&
14878 "Illegal constant bitwidths");
14879 DstBits.insertBits(SrcBits, J * SrcEltSizeInBits);
14880 }
14881 }
14882 return;
14883 }
14884
14885 // Split src element constant bits into dst elements.
14886 unsigned Scale = SrcEltSizeInBits / DstEltSizeInBits;
14887 for (unsigned I = 0; I != NumSrcOps; ++I) {
14888 if (SrcUndefElements[I]) {
14889 DstUndefElements.set(I * Scale, (I + 1) * Scale);
14890 continue;
14891 }
14892 const APInt &SrcBits = SrcBitElements[I];
14893 for (unsigned J = 0; J != Scale; ++J) {
14894 unsigned Idx = (I * Scale) + (IsLittleEndian ? J : (Scale - J - 1));
14895 APInt &DstBits = DstBitElements[Idx];
14896 DstBits = SrcBits.extractBits(DstEltSizeInBits, J * DstEltSizeInBits);
14897 }
14898 }
14899}
14900
14902 for (const SDValue &Op : op_values()) {
14903 unsigned Opc = Op.getOpcode();
14904 if (!Op.isUndef() && Opc != ISD::Constant && Opc != ISD::ConstantFP)
14905 return false;
14906 }
14907 return true;
14908}
14909
14910std::optional<std::pair<APInt, APInt>>
14912 unsigned NumOps = getNumOperands();
14913 if (NumOps < 2)
14914 return std::nullopt;
14915
14916 unsigned EltSize = getValueType(0).getScalarSizeInBits();
14917 APInt Start, Stride;
14918 int FirstIdx = -1, SecondIdx = -1;
14919
14920 // Find the first two non-undef constant elements to determine Start and
14921 // Stride, then verify all remaining elements match the sequence.
14922 for (unsigned I = 0; I < NumOps; ++I) {
14924 if (Op->isUndef())
14925 continue;
14926 if (!isa<ConstantSDNode>(Op))
14927 return std::nullopt;
14928
14929 APInt Val = getConstantOperandAPInt(I).trunc(EltSize);
14930 if (FirstIdx < 0) {
14931 FirstIdx = I;
14932 Start = Val;
14933 } else if (SecondIdx < 0) {
14934 SecondIdx = I;
14935 // Compute stride using modular arithmetic. Simple division would handle
14936 // common strides (1, 2, -1, etc.), but modular inverse maximizes matches.
14937 // Example: <0, poison, poison, 0xFF> has stride 0x55 since 3*0x55 = 0xFF
14938 // Note that modular arithmetic is agnostic to signed/unsigned.
14939 unsigned IdxDiff = I - FirstIdx;
14940 APInt ValDiff = Val - Start;
14941
14942 // Step 1: Factor out common powers of 2 from IdxDiff and ValDiff.
14943 unsigned CommonPow2Bits = llvm::countr_zero(IdxDiff);
14944 if (ValDiff.countr_zero() < CommonPow2Bits)
14945 return std::nullopt; // ValDiff not divisible by 2^CommonPow2Bits
14946 IdxDiff >>= CommonPow2Bits;
14947 ValDiff.lshrInPlace(CommonPow2Bits);
14948
14949 // Step 2: IdxDiff is now odd, so its inverse mod 2^EltSize exists.
14950 // TODO: There are 2^CommonPow2Bits valid strides; currently we only try
14951 // one, but we could try all candidates to handle more cases.
14952 Stride = ValDiff * APInt(EltSize, IdxDiff).multiplicativeInverse();
14953 if (Stride.isZero())
14954 return std::nullopt;
14955
14956 // Step 3: Adjust Start based on the first defined element's index.
14957 Start -= Stride * FirstIdx;
14958 } else {
14959 // Verify this element matches the sequence.
14960 if (Val != Start + Stride * I)
14961 return std::nullopt;
14962 }
14963 }
14964
14965 // Need at least two defined elements.
14966 if (SecondIdx < 0)
14967 return std::nullopt;
14968
14969 return std::make_pair(Start, Stride);
14970}
14971
14973 // Find the first non-undef value in the shuffle mask.
14974 unsigned i, e;
14975 for (i = 0, e = Mask.size(); i != e && Mask[i] < 0; ++i)
14976 /* search */;
14977
14978 // If all elements are undefined, this shuffle can be considered a splat
14979 // (although it should eventually get simplified away completely).
14980 if (i == e)
14981 return true;
14982
14983 // Make sure all remaining elements are either undef or the same as the first
14984 // non-undef value.
14985 for (int Idx = Mask[i]; i != e; ++i)
14986 if (Mask[i] >= 0 && Mask[i] != Idx)
14987 return false;
14988 return true;
14989}
14990
14991// Returns true if it is a constant integer BuildVector or constant integer,
14992// possibly hidden by a bitcast.
14994 SDValue N, bool AllowOpaques) const {
14996
14997 if (auto *C = dyn_cast<ConstantSDNode>(N))
14998 return AllowOpaques || !C->isOpaque();
14999
15001 return true;
15002
15003 // Treat a GlobalAddress supporting constant offset folding as a
15004 // constant integer.
15005 if (auto *GA = dyn_cast<GlobalAddressSDNode>(N))
15006 if (GA->getOpcode() == ISD::GlobalAddress &&
15007 TLI->isOffsetFoldingLegal(GA))
15008 return true;
15009
15010 if ((N.getOpcode() == ISD::SPLAT_VECTOR) &&
15011 isa<ConstantSDNode>(N.getOperand(0)))
15012 return true;
15013 return false;
15014}
15015
15016// Returns true if it is a constant float BuildVector or constant float.
15019 return true;
15020
15022 return true;
15023
15024 if ((N.getOpcode() == ISD::SPLAT_VECTOR) &&
15025 isa<ConstantFPSDNode>(N.getOperand(0)))
15026 return true;
15027
15028 return false;
15029}
15030
15031std::optional<bool> SelectionDAG::isBoolConstant(SDValue N) const {
15032 ConstantSDNode *Const =
15033 isConstOrConstSplat(N, false, /*AllowTruncation=*/true);
15034 if (!Const)
15035 return std::nullopt;
15036
15037 EVT VT = N->getValueType(0);
15038 const APInt CVal = Const->getAPIntValue().trunc(VT.getScalarSizeInBits());
15039 switch (TLI->getBooleanContents(N.getValueType())) {
15041 if (CVal.isOne())
15042 return true;
15043 if (CVal.isZero())
15044 return false;
15045 return std::nullopt;
15047 if (CVal.isAllOnes())
15048 return true;
15049 if (CVal.isZero())
15050 return false;
15051 return std::nullopt;
15053 return CVal[0];
15054 }
15055 llvm_unreachable("Unknown BooleanContent enum");
15056}
15057
15058void SelectionDAG::createOperands(SDNode *Node, ArrayRef<SDValue> Vals) {
15059 assert(!Node->OperandList && "Node already has operands");
15061 "too many operands to fit into SDNode");
15062 SDUse *Ops = OperandRecycler.allocate(
15063 ArrayRecycler<SDUse>::Capacity::get(Vals.size()), OperandAllocator);
15064
15065 bool IsDivergent = false;
15066 for (unsigned I = 0; I != Vals.size(); ++I) {
15067 Ops[I].setUser(Node);
15068 Ops[I].setInitial(Vals[I]);
15069 EVT VT = Ops[I].getValueType();
15070
15071 // Skip Chain. It does not carry divergence.
15072 if (VT != MVT::Other &&
15073 (VT != MVT::Glue || gluePropagatesDivergence(Ops[I].getNode())) &&
15074 Ops[I].getNode()->isDivergent()) {
15075 IsDivergent = true;
15076 }
15077 }
15078 Node->NumOperands = Vals.size();
15079 Node->OperandList = Ops;
15080 if (!TLI->isSDNodeAlwaysUniform(Node)) {
15081 IsDivergent |= TLI->isSDNodeSourceOfDivergence(Node, FLI, UA);
15082 Node->SDNodeBits.IsDivergent = IsDivergent;
15083 }
15084 checkForCycles(Node);
15085}
15086
15089 size_t Limit = SDNode::getMaxNumOperands();
15090 while (Vals.size() > Limit) {
15091 unsigned SliceIdx = Vals.size() - Limit;
15092 auto ExtractedTFs = ArrayRef<SDValue>(Vals).slice(SliceIdx, Limit);
15093 SDValue NewTF = getNode(ISD::TokenFactor, DL, MVT::Other, ExtractedTFs);
15094 Vals.erase(Vals.begin() + SliceIdx, Vals.end());
15095 Vals.emplace_back(NewTF);
15096 }
15097 return getNode(ISD::TokenFactor, DL, MVT::Other, Vals);
15098}
15099
15101 EVT VT, SDNodeFlags Flags) {
15102 switch (Opcode) {
15103 default:
15104 return SDValue();
15105 case ISD::ADD:
15106 case ISD::OR:
15107 case ISD::XOR:
15108 case ISD::UMAX:
15109 case ISD::MUL:
15110 case ISD::AND:
15111 case ISD::UMIN:
15112 case ISD::SMAX:
15113 case ISD::SMIN:
15115 VT);
15116 case ISD::FADD:
15117 // If flags allow, prefer positive zero since it's generally cheaper
15118 // to materialize on most targets.
15119 return getConstantFP(Flags.hasNoSignedZeros() ? 0.0 : -0.0, DL, VT);
15120 case ISD::FMUL:
15121 return getConstantFP(1.0, DL, VT);
15122 case ISD::FMINNUM:
15123 case ISD::FMAXNUM:
15124 case ISD::FMINIMUMNUM:
15125 case ISD::FMAXIMUMNUM: {
15126 // Neutral element for fminnum/fminimumnum is NaN, Inf or FLT_MAX,
15127 // depending on fast-math flags (FMF).
15128 const fltSemantics &Semantics = VT.getFltSemantics();
15129 APFloat NeutralAF = !Flags.hasNoNaNs() ? APFloat::getQNaN(Semantics) :
15130 !Flags.hasNoInfs() ? APFloat::getInf(Semantics) :
15131 APFloat::getLargest(Semantics);
15132 if (Opcode == ISD::FMAXNUM || Opcode == ISD::FMAXIMUMNUM)
15133 NeutralAF.changeSign();
15134
15135 return getConstantFP(NeutralAF, DL, VT);
15136 }
15137 case ISD::FMINIMUM:
15138 case ISD::FMAXIMUM: {
15139 // Neutral element for fminimum is Inf or FLT_MAX, depending on FMF.
15140 const fltSemantics &Semantics = VT.getFltSemantics();
15141 APFloat NeutralAF = !Flags.hasNoInfs() ? APFloat::getInf(Semantics)
15142 : APFloat::getLargest(Semantics);
15143 if (Opcode == ISD::FMAXIMUM)
15144 NeutralAF.changeSign();
15145
15146 return getConstantFP(NeutralAF, DL, VT);
15147 }
15148
15149 }
15150}
15151
15153 SDValue Acc, SDValue LHS,
15154 SDValue RHS) {
15155 EVT AccVT = Acc.getValueType();
15156 if (AccVT.isFloatingPoint()) {
15157 assert(Opc == ISD::PARTIAL_REDUCE_FMLA && "Unexpected opcode");
15158 SDValue NegRHS = getNode(ISD::FNEG, DL, RHS.getValueType(), RHS);
15159 return getNode(Opc, DL, AccVT, Acc, LHS, NegRHS);
15160 }
15163 "Unexpected opcode");
15164 SDValue NegAcc = getNegative(Acc, DL, AccVT);
15165 SDValue MLA = getNode(Opc, DL, AccVT, NegAcc, LHS, RHS);
15166 return getNegative(MLA, DL, AccVT);
15167}
15168
15169/// Helper used to make a call to a library function that has one argument of
15170/// pointer type.
15171///
15172/// Such functions include 'fegetmode', 'fesetenv' and some others, which are
15173/// used to get or set floating-point state. They have one argument of pointer
15174/// type, which points to the memory region containing bits of the
15175/// floating-point state. The value returned by such function is ignored in the
15176/// created call.
15177///
15178/// \param LibFunc Reference to library function (value of RTLIB::Libcall).
15179/// \param Ptr Pointer used to save/load state.
15180/// \param InChain Ingoing token chain.
15181/// \returns Outgoing chain token.
15183 SDValue InChain,
15184 const SDLoc &DLoc) {
15185 assert(InChain.getValueType() == MVT::Other && "Expected token chain");
15187 Args.emplace_back(Ptr, Ptr.getValueType().getTypeForEVT(*getContext()));
15188 RTLIB::LibcallImpl LibcallImpl =
15189 Libcalls->getLibcallImpl(static_cast<RTLIB::Libcall>(LibFunc));
15190 if (LibcallImpl == RTLIB::Unsupported)
15191 reportFatalUsageError("emitting call to unsupported libcall");
15192
15193 SDValue Callee =
15194 getExternalSymbol(LibcallImpl, TLI->getPointerTy(getDataLayout()));
15196 CLI.setDebugLoc(DLoc).setChain(InChain).setLibCallee(
15197 Libcalls->getLibcallImplCallingConv(LibcallImpl),
15198 Type::getVoidTy(*getContext()), Callee, std::move(Args));
15199 return TLI->LowerCallTo(CLI).second;
15200}
15201
15203 assert(From && To && "Invalid SDNode; empty source SDValue?");
15204 auto I = SDEI.find(From);
15205 if (I == SDEI.end())
15206 return;
15207
15208 // Use of operator[] on the DenseMap may cause an insertion, which invalidates
15209 // the iterator, hence the need to make a copy to prevent a use-after-free.
15210 NodeExtraInfo NEI = I->second;
15211 if (LLVM_LIKELY(!NEI.PCSections)) {
15212 // No deep copy required for the types of extra info set.
15213 //
15214 // FIXME: Investigate if other types of extra info also need deep copy. This
15215 // depends on the types of nodes they can be attached to: if some extra info
15216 // is only ever attached to nodes where a replacement To node is always the
15217 // node where later use and propagation of the extra info has the intended
15218 // semantics, no deep copy is required.
15219 SDEI[To] = std::move(NEI);
15220 return;
15221 }
15222
15223 const SDNode *EntrySDN = getEntryNode().getNode();
15224
15225 // We need to copy NodeExtraInfo to all _new_ nodes that are being introduced
15226 // through the replacement of From with To. Otherwise, replacements of a node
15227 // (From) with more complex nodes (To and its operands) may result in lost
15228 // extra info where the root node (To) is insignificant in further propagating
15229 // and using extra info when further lowering to MIR.
15230 //
15231 // In the first step pre-populate the visited set with the nodes reachable
15232 // from the old From node. This avoids copying NodeExtraInfo to parts of the
15233 // DAG that is not new and should be left untouched.
15234 SmallVector<const SDNode *> Leafs{From}; // Leafs reachable with VisitFrom.
15235 DenseSet<const SDNode *> FromReach; // The set of nodes reachable from From.
15236 auto VisitFrom = [&](auto &&Self, const SDNode *N, int MaxDepth) {
15237 if (MaxDepth == 0) {
15238 // Remember this node in case we need to increase MaxDepth and continue
15239 // populating FromReach from this node.
15240 Leafs.emplace_back(N);
15241 return;
15242 }
15243 if (!FromReach.insert(N).second)
15244 return;
15245 for (const SDValue &Op : N->op_values())
15246 Self(Self, Op.getNode(), MaxDepth - 1);
15247 };
15248
15249 // Copy extra info to To and all its transitive operands (that are new).
15251 auto DeepCopyTo = [&](auto &&Self, const SDNode *N) {
15252 if (FromReach.contains(N))
15253 return true;
15254 if (!Visited.insert(N).second)
15255 return true;
15256 if (EntrySDN == N)
15257 return false;
15258 for (const SDValue &Op : N->op_values()) {
15259 if (N == To && Op.getNode() == EntrySDN) {
15260 // Special case: New node's operand is the entry node; just need to
15261 // copy extra info to new node.
15262 break;
15263 }
15264 if (!Self(Self, Op.getNode()))
15265 return false;
15266 }
15267 // Copy only if entry node was not reached.
15268 SDEI[N] = std::move(NEI);
15269 return true;
15270 };
15271
15272 // We first try with a lower MaxDepth, assuming that the path to common
15273 // operands between From and To is relatively short. This significantly
15274 // improves performance in the common case. The initial MaxDepth is big
15275 // enough to avoid retry in the common case; the last MaxDepth is large
15276 // enough to avoid having to use the fallback below (and protects from
15277 // potential stack exhaustion from recursion).
15278 for (int PrevDepth = 0, MaxDepth = 16; MaxDepth <= 1024;
15279 PrevDepth = MaxDepth, MaxDepth *= 2, Visited.clear()) {
15280 // StartFrom is the previous (or initial) set of leafs reachable at the
15281 // previous maximum depth.
15283 std::swap(StartFrom, Leafs);
15284 for (const SDNode *N : StartFrom)
15285 VisitFrom(VisitFrom, N, MaxDepth - PrevDepth);
15286 if (LLVM_LIKELY(DeepCopyTo(DeepCopyTo, To)))
15287 return;
15288 // This should happen very rarely (reached the entry node).
15289 LLVM_DEBUG(dbgs() << __func__ << ": MaxDepth=" << MaxDepth << " too low\n");
15290 assert(!Leafs.empty());
15291 }
15292
15293 // This should not happen - but if it did, that means the subgraph reachable
15294 // from From has depth greater or equal to maximum MaxDepth, and VisitFrom()
15295 // could not visit all reachable common operands. Consequently, we were able
15296 // to reach the entry node.
15297 errs() << "warning: incomplete propagation of SelectionDAG::NodeExtraInfo\n";
15298 assert(false && "From subgraph too complex - increase max. MaxDepth?");
15299 // Best-effort fallback if assertions disabled.
15300 SDEI[To] = std::move(NEI);
15301}
15302
15304 assert(VT.isVector() && "Can only unroll vector types!");
15305 if (VT.isFixedLengthVector())
15306 return VT.getVectorNumElements();
15307
15308 const MachineFunction &MF = getMachineFunction();
15309 const Function &F = MF.getFunction();
15310
15311 APInt MaxVScale = getVScaleRange(&F, sizeof(unsigned) * 8).getUnsignedMax();
15312 if (MaxVScale.ugt(VScaleUnrollLimit))
15313 return 0;
15314
15315 bool Overflow;
15316 APInt MinNElts(sizeof(unsigned) * 8, VT.getVectorMinNumElements());
15317 APInt MaxNElts = MinNElts.umul_ov(MaxVScale, Overflow);
15318 if (Overflow)
15319 return 0;
15320
15321 return MaxNElts.getZExtValue();
15322}
15323
15325 ArrayRef<SDValue> Scalars) {
15326 assert(Scalars.size() == getMaxRuntimeNumElements(VT) &&
15327 "Element count mismatch!");
15328 if (VT.isFixedLengthVector())
15329 return getBuildVector(VT, DL, Scalars);
15330
15331 SDValue Vec = getPOISON(VT);
15332 // Iterate in reverse so result remains poison until we encounter a lane that
15333 // exists, after which all lower-numbered lanes must also exist.
15334 for (unsigned IdxVal : reverse(seq(Scalars.size())))
15335 Vec = getInsertVectorElt(DL, Vec, Scalars[IdxVal], IdxVal);
15336
15337 return Vec;
15338}
15339
15340#ifndef NDEBUG
15341static void checkForCyclesHelper(const SDNode *N,
15344 const llvm::SelectionDAG *DAG) {
15345 // If this node has already been checked, don't check it again.
15346 if (Checked.count(N))
15347 return;
15348
15349 // If a node has already been visited on this depth-first walk, reject it as
15350 // a cycle.
15351 if (!Visited.insert(N).second) {
15352 errs() << "Detected cycle in SelectionDAG\n";
15353 dbgs() << "Offending node:\n";
15354 N->dumprFull(DAG); dbgs() << "\n";
15355 abort();
15356 }
15357
15358 for (const SDValue &Op : N->op_values())
15359 checkForCyclesHelper(Op.getNode(), Visited, Checked, DAG);
15360
15361 Checked.insert(N);
15362 Visited.erase(N);
15363}
15364#endif
15365
15367 const llvm::SelectionDAG *DAG,
15368 bool force) {
15369#ifndef NDEBUG
15370 bool check = force;
15371#ifdef EXPENSIVE_CHECKS
15372 check = true;
15373#endif // EXPENSIVE_CHECKS
15374 if (check) {
15375 assert(N && "Checking nonexistent SDNode");
15378 checkForCyclesHelper(N, visited, checked, DAG);
15379 }
15380#endif // !NDEBUG
15381}
15382
15383void llvm::checkForCycles(const llvm::SelectionDAG *DAG, bool force) {
15384 checkForCycles(DAG->getRoot().getNode(), DAG, force);
15385}
assert(UImm &&(UImm !=~static_cast< T >(0)) &&"Invalid immediate!")
unsigned Imm
unsigned uint64_t
static bool isConstant(const MachineInstr &MI)
constexpr LLT S1
This file declares a class to represent arbitrary precision floating point values and provide a varie...
This file implements a class to represent arbitrary precision integral constant values and operations...
This file implements the APSInt class, which is a simple class that represents an arbitrary sized int...
MachineBasicBlock & MBB
MachineBasicBlock MachineBasicBlock::iterator DebugLoc DL
Function Alias Analysis Results
#define X(NUM, ENUM, NAME)
Definition ELF.h:857
This file implements the BitVector class.
static GCRegistry::Add< ShadowStackGC > C("shadow-stack", "Very portable GC for uncooperative code generators")
static GCRegistry::Add< ErlangGC > A("erlang", "erlang-compatible garbage collector")
static GCRegistry::Add< CoreCLRGC > E("coreclr", "CoreCLR-compatible GC")
static GCRegistry::Add< OcamlGC > B("ocaml", "ocaml 3.10-compatible GC")
static std::optional< bool > isBigEndian(const SmallDenseMap< int64_t, int64_t, 8 > &MemOffset2Idx, int64_t LowestIdx)
Given a map from byte offsets in memory to indices in a load/store, determine if that map corresponds...
#define __asan_unpoison_memory_region(p, size)
Definition Compiler.h:609
#define LLVM_LIKELY(EXPR)
Definition Compiler.h:343
This file contains the declarations for the subclasses of Constant, which represent the different fla...
This file defines the DenseSet and SmallDenseSet classes.
This file contains constants used for implementing Dwarf debug support.
This file defines a hash set that can be used to remove duplication of nodes in a graph.
static MaybeAlign getAlign(Value *Ptr)
iv users
Definition IVUsers.cpp:48
std::pair< Instruction::BinaryOps, Value * > OffsetOp
Find all possible pairs (BinOp, RHS) that BinOp V, RHS can be simplified.
static constexpr Value * getValue(Ty &ValueOrUse)
const size_t AbstractManglingParser< Derived, Alloc >::NumOps
const AbstractManglingParser< Derived, Alloc >::OperatorInfo AbstractManglingParser< Derived, Alloc >::Ops[]
static Register getMemsetValue(Register Val, LLT Ty, MachineIRBuilder &MIB)
static bool isZero(Value *V, const DataLayout &DL, DominatorTree *DT, AssumptionCache *AC)
Definition Lint.cpp:540
static Align getPrefTypeAlign(EVT VT, SelectionDAG &DAG)
static bool isConstantSplatVector(SDValue N, APInt &SplatValue, unsigned MinSizeInBits)
#define F(x, y, z)
Definition MD5.cpp:54
#define I(x, y, z)
Definition MD5.cpp:57
#define G(x, y, z)
Definition MD5.cpp:55
This file declares the MachineConstantPool class which is an abstract constant pool to keep track of ...
Register const TargetRegisterInfo * TRI
This file provides utility analysis objects describing memory locations.
This file contains the declarations for metadata subclasses.
#define T
static MCRegister getReg(const MCDisassembler *D, unsigned RC, unsigned RegNo)
ConstantRange Range(APInt(BitWidth, Low), APInt(BitWidth, High))
#define P(N)
PowerPC Reduce CR logical Operation
const SmallVectorImpl< MachineOperand > & Cond
Remove Loads Into Fake Uses
static bool isValid(const char C)
Returns true if C is a valid mangled character: <0-9a-zA-Z_>.
Contains matchers for matching SelectionDAG nodes and values.
SI Fold Operands
const char * Msg
This file contains some templates that are useful if you are working with the STL at all.
static uint64_t umul_ov(uint64_t i, uint64_t j, bool &Overflow)
static bool shouldLowerMemFuncForSize(const MachineFunction &MF, SelectionDAG &DAG)
static SDValue getFixedOrScalableQuantity(SelectionDAG &DAG, const SDLoc &DL, EVT VT, Ty Quantity)
static std::pair< SDValue, SDValue > getRuntimeCallSDValueHelper(SDValue Chain, const SDLoc &dl, TargetLowering::ArgListTy &&Args, const CallInst *CI, RTLIB::Libcall Call, SelectionDAG *DAG, const TargetLowering *TLI)
static SDValue getMemsetStores(SelectionDAG &DAG, const SDLoc &dl, SDValue Chain, SDValue Dst, SDValue Src, uint64_t Size, Align Alignment, bool isVol, bool AlwaysInline, MachinePointerInfo DstPtrInfo, const AAMDNodes &AAInfo)
Lower the call to 'memset' intrinsic function into a series of store operations.
static std::optional< APInt > FoldValueWithUndef(unsigned Opcode, const APInt &C1, bool IsUndef1, const APInt &C2, bool IsUndef2)
static SDValue FoldSTEP_VECTOR(const SDLoc &DL, EVT VT, SDValue Step, SelectionDAG &DAG)
static cl::opt< int > VScaleUnrollLimit("vscale-unroll-limit", cl::desc("Maximum vscale for which vector unrolling is allowed."), cl::Hidden, cl::init(64))
static SDValue getMemsetStringVal(EVT VT, const SDLoc &dl, SelectionDAG &DAG, const TargetLowering &TLI, const ConstantDataArraySlice &Slice)
getMemsetStringVal - Similar to getMemsetValue.
static cl::opt< bool > EnableMemCpyDAGOpt("enable-memcpy-dag-opt", cl::Hidden, cl::init(true), cl::desc("Gang up loads and stores generated by inlining of memcpy"))
static bool haveNoCommonBitsSetCommutative(SDValue A, SDValue B)
static void AddNodeIDNode(FoldingSetNodeID &ID, const SDNode *N)
Serialize a node the way SDNodes were uniqued before SDNodeKey, to cross-check the typed comparison a...
static void commuteShuffle(SDValue &N1, SDValue &N2, MutableArrayRef< int > M)
Swaps the values of N1 and N2.
static bool isMemSrcFromConstant(SDValue Src, ConstantDataArraySlice &Slice)
Returns true if memcpy source is constant data.
static ISD::CondCode getSetCCInverseImpl(ISD::CondCode Op, bool isIntegerLike)
static bool doNotCSE(SDNode *N)
doNotCSE - Return true if CSE should not be performed for this node.
static cl::opt< int > MaxLdStGlue("ldstmemcpy-glue-max", cl::desc("Number limit for gluing ld/st of memcpy."), cl::Hidden, cl::init(0))
static APInt getIntegerIdentity(unsigned Opcode, unsigned BitWidth)
static SDValue foldCONCAT_VECTORS(const SDLoc &DL, EVT VT, ArrayRef< SDValue > Ops, SelectionDAG &DAG)
Try to simplify vector concatenation to an input value, undef, or build vector.
static MachinePointerInfo InferPointerInfo(const MachinePointerInfo &Info, SelectionDAG &DAG, SDValue Ptr, int64_t Offset=0)
InferPointerInfo - If the specified ptr/offset is a frame index, infer a MachinePointerInfo record fr...
static bool isInTailCallPositionWrapper(const CallInst *CI, const SelectionDAG *SelDAG, bool AllowReturnsFirstArg)
static bool keyMatches(const SDNodeKey &Key, const SDNode &N)
static bool areNonVolatileConsecutiveLoadsOrStores(LSBaseSDNode *LS, LSBaseSDNode *Base, unsigned Bytes, int Dist, const SelectionDAG &DAG)
static bool gluePropagatesDivergence(const SDNode *Node)
Return true if a glue output should propagate divergence information.
static void AddNodeIDCustom(FoldingSetNodeID &ID, const SDNode *N, unsigned Opc)
If this is an SDNode with special info, add this info to the NodeID data.
static void NewSDValueDbgMsg(SDValue V, StringRef Msg, SelectionDAG *G)
static SDVTList makeVTList(const EVT *VTs, unsigned NumVTs)
makeVTList - Return an instance of the SDVTList struct initialized with the specified members.
static void checkForCyclesHelper(const SDNode *N, SmallPtrSetImpl< const SDNode * > &Visited, SmallPtrSetImpl< const SDNode * > &Checked, const llvm::SelectionDAG *DAG)
static void chainLoadsAndStoresForMemcpy(SelectionDAG &DAG, const SDLoc &dl, SmallVector< SDValue, 32 > &OutChains, unsigned From, unsigned To, SmallVector< SDValue, 16 > &OutLoadChains, SmallVector< SDValue, 16 > &OutStoreChains)
static int isSignedOp(ISD::CondCode Opcode)
For an integer comparison, return 1 if the comparison is a signed operation and 2 if the result is an...
static std::optional< APInt > FoldValue(unsigned Opcode, const APInt &C1, const APInt &C2)
static SDValue FoldBUILD_VECTOR(const SDLoc &DL, EVT VT, ArrayRef< SDValue > Ops, SelectionDAG &DAG)
static void checkAddrSpaceIsValidForLibcall(const TargetLowering *TLI, unsigned AS)
static cl::opt< unsigned > MaxSteps("has-predecessor-max-steps", cl::Hidden, cl::init(8192), cl::desc("DAG combiner limit number of steps when searching DAG " "for predecessor nodes"))
static APInt getDemandAllEltsMask(SDValue V)
Construct a DemandedElts mask which demands all elements of V.
static SDValue getMemcpyLoadsAndStores(SelectionDAG &DAG, const SDLoc &dl, SDValue Chain, SDValue Dst, SDValue Src, uint64_t Size, Align DstAlign, Align SrcAlign, bool isVol, bool AlwaysInline, MachinePointerInfo DstPtrInfo, MachinePointerInfo SrcPtrInfo, const AAMDNodes &AAInfo, BatchAAResults *BatchAA, const MDNode *DstMemCacheHint, const MDNode *SrcMemCacheHint)
static SDValue getMemmoveLoadsAndStores(SelectionDAG &DAG, const SDLoc &dl, SDValue Chain, SDValue Dst, SDValue Src, uint64_t Size, Align DstAlign, Align SrcAlign, bool isVol, bool AlwaysInline, MachinePointerInfo DstPtrInfo, MachinePointerInfo SrcPtrInfo, const AAMDNodes &AAInfo)
This file defines the SmallPtrSet class.
This file defines the SmallVector class.
#define LLVM_DEBUG(...)
Definition Debug.h:119
static TableGen::Emitter::Opt Y("gen-skeleton-entry", EmitSkeleton, "Generate example skeleton entry")
static SymbolRef::Type getType(const Symbol *Sym)
Definition TapiFile.cpp:39
This file describes how to lower LLVM code to machine code.
static void removeOperands(MachineInstr &MI, unsigned i)
static OverflowResult mapOverflowResult(ConstantRange::OverflowResult OR)
Convert ConstantRange OverflowResult into ValueTracking OverflowResult.
static int Lookup(ArrayRef< TableEntry > Table, unsigned Opcode)
static unsigned getSize(unsigned Kind)
static const fltSemantics & IEEEsingle()
Definition APFloat.h:304
cmpResult
IEEE-754R 5.11: Floating Point Comparison Relations.
Definition APFloat.h:351
static constexpr roundingMode rmTowardZero
Definition APFloat.h:365
static const fltSemantics & BFloat()
Definition APFloat.h:303
static const fltSemantics & IEEEquad()
Definition APFloat.h:306
static const fltSemantics & IEEEdouble()
Definition APFloat.h:305
static constexpr roundingMode rmTowardNegative
Definition APFloat.h:364
static constexpr roundingMode rmNearestTiesToEven
Definition APFloat.h:361
static constexpr roundingMode rmTowardPositive
Definition APFloat.h:363
static const fltSemantics & IEEEhalf()
Definition APFloat.h:302
opStatus
IEEE-754R 7: Default exception handling.
Definition APFloat.h:377
static APFloat getQNaN(const fltSemantics &Sem, bool Negative=false, const APInt *payload=nullptr)
Factory for QNaN values.
Definition APFloat.h:1224
opStatus divide(const APFloat &RHS, roundingMode RM)
Definition APFloat.h:1312
void copySign(const APFloat &RHS)
Definition APFloat.h:1406
LLVM_ABI opStatus convert(const fltSemantics &ToSemantics, roundingMode RM, bool *losesInfo)
Definition APFloat.cpp:6034
opStatus subtract(const APFloat &RHS, roundingMode RM)
Definition APFloat.h:1294
bool isNegative() const
Definition APFloat.h:1583
opStatus add(const APFloat &RHS, roundingMode RM)
Definition APFloat.h:1285
bool isFinite() const
Definition APFloat.h:1588
opStatus convertFromAPInt(const APInt &Input, bool IsSigned, roundingMode RM)
Definition APFloat.h:1451
opStatus multiply(const APFloat &RHS, roundingMode RM)
Definition APFloat.h:1303
bool isZero() const
Definition APFloat.h:1579
LLVM_READONLY bool isOne() const
Definition APFloat.h:1661
bool isLargest() const
Definition APFloat.h:1599
static APFloat getLargest(const fltSemantics &Sem, bool Negative=false)
Returns the largest finite number in the given semantics.
Definition APFloat.h:1242
opStatus convertToInteger(MutableArrayRef< integerPart > Input, unsigned int Width, bool IsSigned, roundingMode RM, bool *IsExact) const
Definition APFloat.h:1436
static APFloat getInf(const fltSemantics &Sem, bool Negative=false)
Factory for Positive and Negative Infinity.
Definition APFloat.h:1202
opStatus mod(const APFloat &RHS)
Definition APFloat.h:1330
bool isPosZero() const
Definition APFloat.h:1594
bool isNegZero() const
Definition APFloat.h:1595
void changeSign()
Definition APFloat.h:1401
static APFloat getNaN(const fltSemantics &Sem, bool Negative=false, uint64_t payload=0)
Factory for NaN values.
Definition APFloat.h:1213
bool isInfinity() const
Definition APFloat.h:1580
Class for arbitrary precision integers.
Definition APInt.h:78
LLVM_ABI APInt umul_ov(const APInt &RHS, bool &Overflow) const
Definition APInt.cpp:2009
LLVM_ABI APInt usub_sat(const APInt &RHS) const
Definition APInt.cpp:2093
LLVM_ABI APInt udiv(const APInt &RHS) const
Unsigned division operation.
Definition APInt.cpp:1602
static APInt getAllOnes(unsigned numBits)
Return an APInt of a specified width with all bits set.
Definition APInt.h:230
void clearBit(unsigned BitPosition)
Set a given bit to 0.
Definition APInt.h:1426
LLVM_ABI APInt zext(unsigned width) const
Zero extend to a new width.
Definition APInt.cpp:1057
static APInt getSignMask(unsigned BitWidth)
Get the SignMask for a specific bit width.
Definition APInt.h:225
bool isMinSignedValue() const
Determine if this is the smallest signed value.
Definition APInt.h:419
uint64_t getZExtValue() const
Get zero extended value.
Definition APInt.h:1560
unsigned popcount() const
Count the number of bits set.
Definition APInt.h:1690
LLVM_ABI APInt zextOrTrunc(unsigned width) const
Zero extend or truncate to width.
Definition APInt.cpp:1078
unsigned getActiveBits() const
Compute the number of active bits in the value.
Definition APInt.h:1532
LLVM_ABI APInt trunc(unsigned width) const
Truncate to new width.
Definition APInt.cpp:970
void setBit(unsigned BitPosition)
Set the given bit to 1 whose position is given as "bitPosition".
Definition APInt.h:1350
APInt abs() const
Get the absolute value.
Definition APInt.h:1815
LLVM_ABI APInt sadd_sat(const APInt &RHS) const
Definition APInt.cpp:2064
bool isAllOnes() const
Determine if all bits are set. This is true for zero-width values.
Definition APInt.h:367
bool ugt(const APInt &RHS) const
Unsigned greater than comparison.
Definition APInt.h:1186
static APInt getBitsSet(unsigned numBits, unsigned loBit, unsigned hiBit)
Get a value with a block of bits set.
Definition APInt.h:254
bool isZero() const
Determine if this value is zero, i.e. all bits are clear.
Definition APInt.h:376
LLVM_ABI APInt urem(const APInt &RHS) const
Unsigned remainder operation.
Definition APInt.cpp:1695
unsigned getBitWidth() const
Return the number of bits in the APInt.
Definition APInt.h:1508
bool ult(const APInt &RHS) const
Unsigned less than comparison.
Definition APInt.h:1115
static APInt getSignedMaxValue(unsigned numBits)
Gets maximum signed value of APInt for a specific bit width.
Definition APInt.h:205
bool isNegative() const
Determine sign of this APInt.
Definition APInt.h:325
LLVM_ABI APInt sdiv(const APInt &RHS) const
Signed division function for APInt.
Definition APInt.cpp:1673
LLVM_ABI APInt rotr(unsigned rotateAmt) const
Rotate right by rotateAmt.
Definition APInt.cpp:1199
LLVM_ABI APInt reverseBits() const
Definition APInt.cpp:786
void ashrInPlace(unsigned ShiftAmt)
Arithmetic right-shift this APInt by ShiftAmt in place.
Definition APInt.h:836
bool sle(const APInt &RHS) const
Signed less or equal comparison.
Definition APInt.h:1170
unsigned countr_zero() const
Count the number of trailing zero bits.
Definition APInt.h:1659
unsigned getNumSignBits() const
Computes the number of leading bits of this APInt that are equal to its sign bit.
Definition APInt.h:1648
unsigned countl_zero() const
The APInt version of std::countl_zero.
Definition APInt.h:1618
static LLVM_ABI APInt getSplat(unsigned NewLen, const APInt &V)
Return a value containing V broadcasted over NewLen bits.
Definition APInt.cpp:648
static APInt getSignedMinValue(unsigned numBits)
Gets minimum signed value of APInt for a specific bit width.
Definition APInt.h:215
LLVM_ABI APInt sshl_sat(const APInt &RHS) const
Definition APInt.cpp:2124
LLVM_ABI APInt ushl_sat(const APInt &RHS) const
Definition APInt.cpp:2138
LLVM_ABI APInt sextOrTrunc(unsigned width) const
Sign extend or truncate to width.
Definition APInt.cpp:1086
static bool isSameValue(const APInt &I1, const APInt &I2, bool SignedCompare=false)
Determine if two APInts have the same value, after zero-extending or sign-extending (if SignedCompare...
Definition APInt.h:550
LLVM_ABI APInt rotl(unsigned rotateAmt) const
Rotate left by rotateAmt.
Definition APInt.cpp:1186
LLVM_ABI void insertBits(const APInt &SubBits, unsigned bitPosition)
Insert the bits from a smaller APInt starting at bitPosition.
Definition APInt.cpp:393
unsigned logBase2() const
Definition APInt.h:1781
LLVM_ABI APInt uadd_sat(const APInt &RHS) const
Definition APInt.cpp:2074
APInt ashr(unsigned ShiftAmt) const
Arithmetic right-shift function.
Definition APInt.h:829
LLVM_ABI APInt multiplicativeInverse() const
Definition APInt.cpp:1303
LLVM_ABI APInt srem(const APInt &RHS) const
Function for signed remainder operation.
Definition APInt.cpp:1774
bool isNonNegative() const
Determine if this APInt Value is non-negative (>= 0)
Definition APInt.h:330
bool ule(const APInt &RHS) const
Unsigned less or equal comparison.
Definition APInt.h:1154
LLVM_ABI APInt sext(unsigned width) const
Sign extend to a new width.
Definition APInt.cpp:1030
void setBits(unsigned loBit, unsigned hiBit)
Set the bits from loBit (inclusive) to hiBit (exclusive) to 1.
Definition APInt.h:1387
APInt shl(unsigned shiftAmt) const
Left-shift function.
Definition APInt.h:875
LLVM_ABI APInt byteSwap() const
Definition APInt.cpp:764
bool isSubsetOf(const APInt &RHS) const
This operation checks that all bits set in this APInt are also set in RHS.
Definition APInt.h:1261
bool isPowerOf2() const
Check if this APInt's value is a power of two greater than zero.
Definition APInt.h:436
static APInt getLowBitsSet(unsigned numBits, unsigned loBitsSet)
Constructs an APInt value that has the bottom loBitsSet bits set.
Definition APInt.h:302
void clearBits(unsigned LoBit, unsigned HiBit)
Clear the bits from LoBit (inclusive) to HiBit (exclusive) to 0.
Definition APInt.h:1437
static APInt getZero(unsigned numBits)
Get the '0' value for the specified bit-width.
Definition APInt.h:196
LLVM_ABI APInt extractBits(unsigned numBits, unsigned bitPosition) const
Return an APInt with the extracted bits [bitPosition,bitPosition+numBits).
Definition APInt.cpp:478
bool sge(const APInt &RHS) const
Signed greater or equal comparison.
Definition APInt.h:1241
bool isOne() const
Determine if this is a value of 1.
Definition APInt.h:385
static APInt getBitsSetFrom(unsigned numBits, unsigned loBit)
Constructs an APInt value that has a contiguous range of bits set.
Definition APInt.h:282
static APInt getOneBitSet(unsigned numBits, unsigned BitNo)
Return an APInt with exactly one bit set in the result.
Definition APInt.h:235
void lshrInPlace(unsigned ShiftAmt)
Logical right-shift this APInt by ShiftAmt in place.
Definition APInt.h:860
APInt lshr(unsigned shiftAmt) const
Logical right-shift function.
Definition APInt.h:853
bool uge(const APInt &RHS) const
Unsigned greater or equal comparison.
Definition APInt.h:1225
LLVM_ABI APInt ssub_sat(const APInt &RHS) const
Definition APInt.cpp:2083
An arbitrary precision integer that knows its signedness.
Definition APSInt.h:24
unsigned getSrcAddressSpace() const
unsigned getDestAddressSpace() const
static Capacity get(size_t N)
Get the capacity of an array that can hold at least N elements.
Represent a constant reference to an array (0 or more elements consecutively in memory),...
Definition ArrayRef.h:40
size_t size() const
Get the array size.
Definition ArrayRef.h:141
bool empty() const
Check if the array is empty.
Definition ArrayRef.h:136
static LLVM_ABI BaseIndexOffset match(const SDNode *N, const SelectionDAG &DAG)
Parses tree in N for base, index, offset addresses.
This class is a wrapper over an AAResults, and it is intended to be used only when there are no IR ch...
bool pointsToConstantMemory(const MemoryLocation &Loc, bool OrLocal=false)
BitVector & reset()
Reset all bits in the bitvector.
Definition BitVector.h:409
void resize(unsigned N, bool t=false)
Grow or shrink the bitvector.
Definition BitVector.h:355
void clear()
Removes all bits from the bitvector.
Definition BitVector.h:349
BitVector & set()
Set all bits in the bitvector.
Definition BitVector.h:366
bool none() const
Returns true if none of the bits are set.
Definition BitVector.h:207
size_type size() const
Returns the number of bits in this bitvector.
Definition BitVector.h:178
const BlockAddress * getBlockAddress() const
The address of a basic block.
Definition Constants.h:1088
BlockFrequencyInfo pass uses BlockFrequencyInfoImpl implementation to estimate IR basic block frequen...
A "pseudo-class" with methods for operating on BUILD_VECTORs.
LLVM_ABI bool getConstantRawBits(bool IsLittleEndian, unsigned DstEltSizeInBits, SmallVectorImpl< APInt > &RawBitElements, BitVector &UndefElements) const
Extract the raw bit data from a build vector of Undef, Constant or ConstantFP node elements.
static LLVM_ABI void recastRawBits(bool IsLittleEndian, unsigned DstEltSizeInBits, SmallVectorImpl< APInt > &DstBitElements, ArrayRef< APInt > SrcBitElements, BitVector &DstUndefElements, const BitVector &SrcUndefElements)
Recast bit data SrcBitElements to DstEltSizeInBits wide elements.
LLVM_ABI bool getRepeatedSequence(const APInt &DemandedElts, SmallVectorImpl< SDValue > &Sequence, BitVector *UndefElements=nullptr) const
Find the shortest repeating sequence of values in the build vector.
LLVM_ABI ConstantFPSDNode * getConstantFPSplatNode(const APInt &DemandedElts, BitVector *UndefElements=nullptr) const
Returns the demanded splatted constant FP or null if this is not a constant FP splat.
LLVM_ABI SDValue getSplatValue(const APInt &DemandedElts, BitVector *UndefElements=nullptr) const
Returns the demanded splatted value or a null value if this is not a splat.
LLVM_ABI bool isConstantSplat(APInt &SplatValue, APInt &SplatUndef, unsigned &SplatBitSize, bool &HasAnyUndefs, unsigned MinSplatBits=0, bool isBigEndian=false) const
Check if this is a constant splat, and if so, find the smallest element size that splats the vector.
LLVM_ABI ConstantSDNode * getConstantSplatNode(const APInt &DemandedElts, BitVector *UndefElements=nullptr) const
Returns the demanded splatted constant or null if this is not a constant splat.
LLVM_ABI int32_t getConstantFPSplatPow2ToLog2Int(BitVector *UndefElements, uint32_t BitWidth) const
If this is a constant FP splat and the splatted constant FP is an exact power or 2,...
LLVM_ABI std::optional< std::pair< APInt, APInt > > isArithmeticSequence() const
If this BuildVector is constant and represents an arithmetic sequence "<a, a+n, a+2n,...
LLVM_ABI bool isConstant() const
This class represents a function call, abstracting a target machine's calling convention.
bool isTailCall() const
static LLVM_ABI bool isValueValidForType(EVT VT, const APFloat &Val)
const APFloat & getValueAPF() const
bool isExactlyValue(double V) const
We don't rely on operator== working on double values, as it returns true for things that are clearly ...
ConstantFP - Floating Point Values [float, double].
Definition Constants.h:420
const APFloat & getValue() const
Definition Constants.h:464
This is the shared class of boolean and integer constants.
Definition Constants.h:87
unsigned getBitWidth() const
getBitWidth - Return the scalar bitwidth of this constant.
Definition Constants.h:162
const APInt & getValue() const
Return the constant as an APInt value reference.
Definition Constants.h:159
MachineConstantPoolValue * getMachineCPVal() const
const Constant * getConstVal() const
LLVM_ABI Type * getType() const
This class represents a range of values.
PreferredRangeType
If represented precisely, the result of some range operations may consist of multiple disjoint ranges...
const APInt * getSingleElement() const
If this set contains a single element, return it, otherwise return null.
static LLVM_ABI ConstantRange fromKnownBits(const KnownBits &Known, bool IsSigned)
Initialize a range based on a known bits constraint.
LLVM_ABI OverflowResult unsignedSubMayOverflow(const ConstantRange &Other) const
Return whether unsigned sub of the two ranges always/never overflows.
LLVM_ABI OverflowResult unsignedAddMayOverflow(const ConstantRange &Other) const
Return whether unsigned add of the two ranges always/never overflows.
LLVM_ABI KnownBits toKnownBits() const
Return known bits for values in this range.
LLVM_ABI ConstantRange zeroExtend(uint32_t BitWidth) const
Return a new range in the specified integer type, which must be strictly larger than the current type...
LLVM_ABI APInt getSignedMin() const
Return the smallest signed value contained in the ConstantRange.
LLVM_ABI OverflowResult unsignedMulMayOverflow(const ConstantRange &Other) const
Return whether unsigned mul of the two ranges always/never overflows.
LLVM_ABI ConstantRange signExtend(uint32_t BitWidth) const
Return a new range in the specified integer type, which must be strictly larger than the current type...
LLVM_ABI ConstantRange multiply(const ConstantRange &Other, unsigned NoWrapKind=0) const
Return a new range representing the possible values resulting from a multiplication of a value in thi...
LLVM_ABI bool contains(const APInt &Val) const
Return true if the specified value is in the set.
LLVM_ABI APInt getUnsignedMax() const
Return the largest unsigned value contained in the ConstantRange.
LLVM_ABI ConstantRange intersectWith(const ConstantRange &CR, PreferredRangeType Type=Smallest) const
Return the range that results from the intersection of this range with another range.
LLVM_ABI APInt getSignedMax() const
Return the largest signed value contained in the ConstantRange.
OverflowResult
Represents whether an operation on the given constant range is known to always or never overflow.
@ AlwaysOverflowsHigh
Always overflows in the direction of signed/unsigned max value.
@ AlwaysOverflowsLow
Always overflows in the direction of signed/unsigned min value.
@ MayOverflow
May or may not overflow.
uint32_t getBitWidth() const
Get the bit width of this ConstantRange.
LLVM_ABI OverflowResult signedSubMayOverflow(const ConstantRange &Other) const
Return whether signed sub of the two ranges always/never overflows.
uint64_t getZExtValue() const
const APInt & getAPIntValue() const
This is an important base class in LLVM.
Definition Constant.h:43
LLVM_ABI Constant * getSplatValue(bool AllowPoison=false) const
If all elements of the vector constant have the same value, return that value.
LLVM_ABI Constant * getAggregateElement(unsigned Elt) const
For aggregates (struct/array/vector) return the constant that corresponds to the specified element if...
DWARF expression.
static LLVM_ABI ExtOps getExtOps(unsigned FromSize, unsigned ToSize, bool Signed)
Returns the ops for a zero- or sign-extension in a DIExpression.
static LLVM_ABI void appendOffset(SmallVectorImpl< uint64_t > &Ops, int64_t Offset)
Append Ops with operations to apply the Offset.
static LLVM_ABI DIExpression * appendOpsToArg(const DIExpression *Expr, ArrayRef< uint64_t > Ops, unsigned ArgNo, bool StackValue=false)
Create a copy of Expr by appending the given list of Ops to each instance of the operand DW_OP_LLVM_a...
static LLVM_ABI const DIExpression * convertToVariadicExpression(const DIExpression *Expr)
If Expr is a non-variadic expression (i.e.
static LLVM_ABI std::optional< DIExpression * > createFragmentExpression(const DIExpression *Expr, unsigned OffsetInBits, unsigned SizeInBits)
Create a DIExpression to describe one part of an aggregate variable that is fragmented across multipl...
Base class for variables.
A parsed version of the target data layout string in and methods for querying it.
Definition DataLayout.h:64
bool isLittleEndian() const
Layout endianness...
Definition DataLayout.h:217
LLVM_ABI IntegerType * getIntPtrType(LLVMContext &C, unsigned AddressSpace=0) const
Returns an integer type with size at least as big as that of a pointer in the given address space.
LLVM_ABI Align getABITypeAlign(Type *Ty) const
Returns the minimum ABI-required alignment for the specified type.
LLVM_ABI unsigned getPointerTypeSizeInBits(Type *) const
The pointer representation size in bits for this type.
LLVM_ABI Align getPrefTypeAlign(Type *Ty) const
Returns the preferred stack/global alignment for the specified type.
A debug info location.
Definition DebugLoc.h:126
Implements a dense probed hash-table based set.
Definition DenseSet.h:281
static constexpr ElementCount getFixed(ScalarTy MinVal)
Definition TypeSize.h:305
const char * getSymbol() const
Insertion token: a failed lookup fills it in, the matching insert consumes it.
Definition FoldingSet.h:284
This class is used to gather all the unique data bits of a node.
Definition FoldingSet.h:162
Data structure describing the variable locations in a function.
bool hasMinSize() const
Optimize this function for minimum size (-Oz).
Definition Function.h:696
AttributeList getAttributes() const
Return the attribute list for this Function.
Definition Function.h:329
LLVM_ABI unsigned getAddressSpace() const
const GlobalValue * getGlobal() const
bool isThreadLocal() const
If the value is "Thread Local", its value isn't shared by the threads.
unsigned getAddressSpace() const
Module * getParent()
Get the module that this global value is contained inside of...
PointerType * getType() const
Global values are always pointers.
This class is used to form a handle around another node that is persistent and is updated across invo...
const SDValue & getValue() const
static LLVM_ABI bool compare(const APInt &LHS, const APInt &RHS, ICmpInst::Predicate Pred)
Return result of LHS Pred RHS comparison.
This is an important class for using LLVM in a threaded context.
Definition LLVMContext.h:68
Base class for LoadSDNode and StoreSDNode.
Tracks which library functions to use for a particular subtarget or function.
CallingConv::ID getLibcallImplCallingConv(RTLIB::LibcallImpl Call) const
Get the CallingConv that should be used for the specified libcall.
RTLIB::LibcallImpl getLibcallImpl(RTLIB::Libcall Call) const
Return the lowering's selection of implementation call for Call.
This SDNode is used for LIFETIME_START/LIFETIME_END values.
This class is used to represent ISD::LOAD nodes.
static LocationSize precise(uint64_t Value)
MCSymbol - Instances of this class represent a symbol name in the MC file, and MCSymbols are created ...
Definition MCSymbol.h:42
Metadata node.
Definition Metadata.h:1081
const MDOperand & getOperand(unsigned I) const
Definition Metadata.h:1437
Machine Value Type.
SimpleValueType SimpleTy
Abstract base class for all machine specific constantpool value subclasses.
virtual void addSelectionDAGCSEId(FoldingSetNodeID &ID)=0
The MachineFrameInfo class represents an abstract stack frame until prolog/epilog code is inserted.
LLVM_ABI int CreateStackObject(uint64_t Size, Align Alignment, bool isSpillSlot, const AllocaInst *Alloca=nullptr, uint8_t ID=0)
Create a new statically sized stack object, returning a nonnegative identifier to represent it.
Align getObjectAlign(int ObjectIdx) const
Return the alignment of the specified stack object.
bool isFixedObjectIndex(int ObjectIdx) const
Returns true if the specified index corresponds to a fixed stack object.
void setObjectAlignment(int ObjectIdx, Align Alignment)
setObjectAlignment - Change the alignment of the specified stack object.
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.
const TargetMachine & getTarget() const
getTarget - Return the target machine this machine code is compiled with
A description of a memory reference used in the backend.
const MDNode * getRanges() const
Return the range tag for the memory reference.
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.
@ MOInvariant
The memory access always returns the same value (or traps).
@ MOStore
The memory access writes data.
const MachinePointerInfo & getPointerInfo() const
Flags getFlags() const
Return the raw flags of the source value,.
An SDNode that represents everything that will be needed to construct a MachineInstr.
This class is used to represent an MGATHER node.
This class is used to represent an MLOAD node.
This class is used to represent an MSTORE node.
This SDNode is used for target intrinsics that touch memory and need an associated MachineMemOperand.
This is an abstract virtual class for memory operations.
size_t getNumMemOperands() const
Return the number of memory operands.
LLVM_ABI MemSDNode(unsigned Opc, unsigned Order, const DebugLoc &dl, SDVTList VTs, EVT memvt, PointerUnion< MachineMemOperand *, MachineMemOperand ** > memrefs)
Constructor that supports single or multiple MMOs.
PointerUnion< MachineMemOperand *, MachineMemOperand ** > MemRefs
Memory reference information.
MachineMemOperand * getMemOperand() const
Return the unique MachineMemOperand object describing the memory reference performed by operation.
const MachinePointerInfo & getPointerInfo() const
ArrayRef< MachineMemOperand * > memoperands() const
Return the memory operands for this node.
unsigned getRawSubclassData() const
Return the SubclassData value, without HasDebugValue.
EVT getMemoryVT() const
Return the type of the in-memory value.
Representation for a specific memory location.
Root of the metadata hierarchy.
Definition Metadata.h:64
A Module instance is used to store all the information related to an LLVM module.
Definition Module.h:68
Function * getFunction(StringRef Name) const
Look up the specified function in the module symbol table.
Definition Module.cpp:235
Represent a mutable reference to an array (0 or more elements consecutively in memory),...
Definition ArrayRef.h:294
Class to represent pointers.
static PointerType * getUnqual(LLVMContext &C)
This constructs an opaque pointer to an object in the default address space (address space zero).
static LLVM_ABI PointerType * get(LLVMContext &C, unsigned AddressSpace)
This constructs an opaque pointer to an object in a numbered address space.
Definition Type.cpp:887
unsigned getAddressSpace() const
Return the address space of the Pointer type.
A discriminated union of two or more pointer types, with the discriminator in the low bits of the poi...
bool isNull() const
Test if the pointer held in the union is null, regardless of which type it is.
Analysis providing profile information.
void Deallocate(SubClass *E)
Deallocate - Release storage for the pointed-to object.
Wrapper class representing virtual and physical registers.
Definition Register.h:20
Keeps track of dbg_value information through SDISel.
LLVM_ABI void add(SDDbgValue *V, bool isParameter)
LLVM_ABI void erase(const SDNode *Node)
Invalidate all DbgValues attached to the node and remove it from the Node-to-DbgValues map.
Holds the information from a dbg_label node through SDISel.
Holds the information for a single machine location through SDISel; either an SDNode,...
static SDDbgOperand fromNode(SDNode *Node, unsigned ResNo)
static SDDbgOperand fromFrameIdx(unsigned FrameIdx)
static SDDbgOperand fromVReg(Register VReg)
static SDDbgOperand fromConst(const Value *Const)
@ SDNODE
Value is the result of an expression.
Holds the information from a dbg_value node through SDISel.
Wrapper class for IR location info (IR ordering and DebugLoc) to be passed into SDNode creation funct...
const DebugLoc & getDebugLoc() const
unsigned getIROrder() const
This class provides iterator support for SDUse operands that use a specific SDNode.
Represents one node in the SelectionDAG.
const APInt & getAsAPIntVal() const
Helper method returns the APInt value of a ConstantSDNode.
bool isMachineOpcode() const
Test if this node has a post-isel opcode, directly corresponding to a MachineInstr opcode.
LLVM_ABI void dumprFull(const SelectionDAG *G=nullptr) const
printrFull to dbgs().
unsigned getOpcode() const
Return the SelectionDAG opcode value for this node.
bool isDivergent() const
LLVM_ABI bool isOnlyUserOf(const SDNode *N) const
Return true if this node is the only use of N.
iterator_range< value_op_iterator > op_values() const
unsigned getIROrder() const
Return the node ordering.
static constexpr size_t getMaxNumOperands()
Return the maximum number of operands that a SDNode can hold.
iterator_range< use_iterator > uses()
MemSDNodeBitfields MemSDNodeBits
bool getHasDebugValue() const
SDNodeFlags getFlags() const
void setNodeId(int Id)
Set unique node id.
LLVM_ABI void intersectFlagsWith(const SDNodeFlags Flags)
Clear any flags in this node that aren't also set in Flags.
static bool hasPredecessorHelper(const SDNode *N, SmallPtrSetImpl< const SDNode * > &Visited, SmallVectorImpl< const SDNode * > &Worklist, unsigned int MaxSteps=0, bool TopologicalPrune=false)
Returns true if N is a predecessor of any node in Worklist.
uint64_t getAsZExtVal() const
Helper method returns the zero-extended integer value of a ConstantSDNode.
bool use_empty() const
Return true if there are no uses of this node.
unsigned getNumValues() const
Return the number of values defined/returned by this operator.
unsigned getNumOperands() const
Return the number of values used by this operation.
const SDValue & getOperand(unsigned Num) const
static LLVM_ABI bool areOnlyUsersOf(ArrayRef< const SDNode * > Nodes, const SDNode *N)
Return true if all the users of N are contained in Nodes.
use_iterator use_begin() const
Provide iteration support to walk over all uses of an SDNode.
LLVM_ABI bool isOperandOf(const SDNode *N) const
Return true if this node is an operand of N.
const APInt & getConstantOperandAPInt(unsigned Num) const
Helper method returns the APInt of a ConstantSDNode operand.
std::optional< APInt > bitcastToAPInt() const
LLVM_ABI bool hasPredecessor(const SDNode *N) const
Return true if N is a predecessor of this node.
LLVM_ABI bool hasAnyUseOfValue(unsigned Value) const
Return true if there are any use of the indicated value.
EVT getValueType(unsigned ResNo) const
Return the type of a specified result.
bool isUndef() const
Returns true if the node type is UNDEF or POISON.
op_iterator op_end() const
op_iterator op_begin() const
static use_iterator use_end()
LLVM_ABI void DropOperands()
Release the operands and set this node to have zero operands.
SDNode(unsigned Opc, unsigned Order, DebugLoc dl, SDVTList VTs)
Create an SDNode.
Represents a use of a SDNode.
SDNode * getUser()
This returns the SDNode that contains this Use.
Unlike LLVM values, Selection DAG nodes may return multiple values as the result of a computation.
bool isUndef() const
SDNode * getNode() const
get the SDNode which holds the desired result
bool hasOneUse() const
Return true if there is exactly one node using value ResNo of Node, in exactly one operand.
LLVM_ABI bool isOperandOf(const SDNode *N) const
Return true if the referenced return value is an operand of N.
SDValue()=default
LLVM_ABI bool reachesChainWithoutSideEffects(SDValue Dest, unsigned Depth=2) const
Return true if this operand (which must be a chain) reaches the specified operand without crossing an...
SDValue getValue(unsigned R) const
EVT getValueType() const
Return the ValueType of the referenced return value.
TypeSize getValueSizeInBits() const
Returns the size of the value in bits.
const SDValue & getOperand(unsigned i) const
bool use_empty() const
Return true if there are no nodes using value ResNo of Node.
const APInt & getConstantOperandAPInt(unsigned i) const
uint64_t getScalarValueSizeInBits() const
unsigned getResNo() const
get the index which selects a specific result in the SDNode
uint64_t getConstantOperandVal(unsigned i) const
unsigned getOpcode() const
virtual void verifyTargetNode(const SelectionDAG &DAG, const SDNode *N) const
Checks that the given target-specific node is valid. Aborts if it is not.
This is used to represent a portion of an LLVM function in a low-level Data Dependence DAG representa...
LLVM_ABI SDValue getElementCount(const SDLoc &DL, EVT VT, ElementCount EC)
LLVM_ABI Align getReducedAlign(EVT VT, bool UseABI)
In most cases this function returns the ABI alignment for a given type, except for illegal vector typ...
LLVM_ABI SDValue getShiftAmountOperand(EVT LHSTy, SDValue Op)
Return the specified value casted to the target's desired shift amount type.
LLVM_ABI std::pair< SDValue, SDValue > getMemccpy(SDValue Chain, const SDLoc &dl, SDValue Dst, SDValue Src, SDValue C, SDValue Size, const CallInst *CI)
Lower a memccpy operation into a target library call and return the resulting chain and call result a...
LLVM_ABI SDValue emitStackConvert(SDValue SrcOp, EVT SlotVT, EVT DestVT, const SDLoc &DL, SDValue Chain)
Emit a store/load combination to the stack.
LLVM_ABI bool isKnownNeverLogicalZero(SDValue Op, const APInt &DemandedElts, unsigned Depth=0) const
Test whether the given floating point SDValue (or all elements of it, if it is a vector) is known to ...
LLVM_ABI SDValue getExtLoadVP(ISD::LoadExtType ExtType, const SDLoc &dl, EVT VT, SDValue Chain, SDValue Ptr, SDValue Mask, SDValue EVL, MachinePointerInfo PtrInfo, EVT MemVT, MaybeAlign Alignment, MachineMemOperand::Flags MMOFlags, const AAMDNodes &AAInfo, bool IsExpanding=false)
SDValue getExtractVectorElt(const SDLoc &DL, EVT VT, SDValue Vec, unsigned Idx)
Extract element at Idx from Vec.
LLVM_ABI SDValue getSplatSourceVector(SDValue V, int &SplatIndex)
If V is a splatted value, return the source vector and its splat index.
LLVM_ABI SDValue getLabelNode(unsigned Opcode, const SDLoc &dl, SDValue Root, MCSymbol *Label)
LLVM_ABI OverflowKind computeOverflowForUnsignedSub(SDValue N0, SDValue N1) const
Determine if the result of the unsigned sub of 2 nodes can overflow.
LLVM_ABI unsigned ComputeMaxSignificantBits(SDValue Op, unsigned Depth=0) const
Get the upper bound on bit size for this Value Op as a signed integer.
const SDValue & getRoot() const
Return the root tag of the SelectionDAG.
LLVM_ABI std::pair< SDValue, SDValue > getStrlen(SDValue Chain, const SDLoc &dl, SDValue Src, const CallInst *CI)
Lower a strlen operation into a target library call and return the resulting chain and call result as...
LLVM_ABI SDValue getMaskedGather(SDVTList VTs, EVT MemVT, const SDLoc &dl, ArrayRef< SDValue > Ops, MachineMemOperand *MMO, ISD::MemIndexType IndexType, ISD::LoadExtType ExtTy)
LLVM_ABI SDValue FoldSetCC(EVT VT, SDValue N1, SDValue N2, ISD::CondCode Cond, const SDLoc &dl, SDNodeFlags Flags={})
Constant fold a setcc to true or false.
bool isKnownNeverSNaN(SDValue Op, const APInt &DemandedElts, unsigned Depth=0) const
LLVM_ABI std::optional< bool > isBoolConstant(SDValue N) const
Check if a value \op N is a constant using the target's BooleanContent for its type.
LLVM_ABI SDValue getStackArgumentTokenFactor(SDValue Chain)
Compute a TokenFactor to force all the incoming stack arguments to be loaded from the stack.
const TargetSubtargetInfo & getSubtarget() const
LLVM_ABI ConstantRange computeConstantRange(SDValue Op, bool ForSigned, unsigned Depth=0) const
Determine the possible constant range of an integer or vector of integers.
LLVM_ABI SDValue getMergeValues(ArrayRef< SDValue > Ops, const SDLoc &dl)
Create a MERGE_VALUES node from the given operands.
LLVM_ABI SDVTList getVTList(EVT VT)
Return an SDVTList that represents the list of values specified.
LLVM_ABI SDValue getShiftAmountConstant(uint64_t Val, EVT VT, const SDLoc &DL)
LLVM_ABI void updateDivergence(SDNode *N)
LLVM_ABI SDValue getSplatValue(SDValue V, bool LegalTypes=false)
If V is a splat vector, return its scalar source operand by extracting that element from the source v...
LLVM_ABI SDValue getAllOnesConstant(const SDLoc &DL, EVT VT, bool IsTarget=false, bool IsOpaque=false)
LLVM_ABI MachineSDNode * getMachineNode(unsigned Opcode, const SDLoc &dl, EVT VT)
These are used for target selectors to create a new node with specified return type(s),...
LLVM_ABI void ExtractVectorElements(SDValue Op, SmallVectorImpl< SDValue > &Args, unsigned Start=0, unsigned Count=0, EVT EltVT=EVT())
Append the extracted elements from Start to Count out of the vector Op in Args.
LLVM_ABI SDValue getAtomicMemset(SDValue Chain, const SDLoc &dl, SDValue Dst, SDValue Value, SDValue Size, Type *SizeTy, unsigned ElemSz, bool isTailCall, MachinePointerInfo DstPtrInfo)
LLVM_ABI SDValue getAtomicLoad(ISD::LoadExtType ExtType, const SDLoc &dl, EVT MemVT, EVT VT, SDValue Chain, SDValue Ptr, MachineMemOperand *MMO)
LLVM_ABI SDNode * getNodeIfExists(unsigned Opcode, SDVTList VTList, ArrayRef< SDValue > Ops, const SDNodeFlags Flags, bool AllowCommute=false)
Get the specified node if it's already available, or else return NULL.
LLVM_ABI SDValue getPseudoProbeNode(const SDLoc &Dl, SDValue Chain, uint64_t Guid, uint64_t Index, uint32_t Attr)
Creates a PseudoProbeSDNode with function GUID Guid and the index of the block Index it is probing,...
LLVM_ABI SDValue getFreeze(SDValue V)
Return a freeze using the SDLoc of the value operand.
LLVM_ABI SDNode * SelectNodeTo(SDNode *N, unsigned MachineOpc, EVT VT)
These are used for target selectors to mutate the specified node to have the specified return type,...
LLVM_ABI SelectionDAG(const TargetMachine &TM, CodeGenOptLevel)
LLVM_ABI SDValue getMemset(SDValue Chain, const SDLoc &dl, SDValue Dst, SDValue Src, SDValue Size, Align Alignment, bool isVol, bool AlwaysInline, const CallInst *CI, MachinePointerInfo DstPtrInfo, const AAMDNodes &AAInfo=AAMDNodes())
LLVM_ABI SDValue getConstantPool(const Constant *C, EVT VT, MaybeAlign Align=std::nullopt, int Offs=0, bool isT=false, unsigned TargetFlags=0)
LLVM_ABI SDValue getStridedLoadVP(ISD::MemIndexedMode AM, ISD::LoadExtType ExtType, EVT VT, const SDLoc &DL, SDValue Chain, SDValue Ptr, SDValue Offset, SDValue Stride, SDValue Mask, SDValue EVL, EVT MemVT, MachineMemOperand *MMO, bool IsExpanding=false)
LLVM_ABI SDValue getAtomicCmpSwap(unsigned Opcode, const SDLoc &dl, EVT MemVT, SDVTList VTs, SDValue Chain, SDValue Ptr, SDValue Cmp, SDValue Swp, MachineMemOperand *MMO)
Gets a node for an atomic cmpxchg op.
LLVM_ABI SDValue makeEquivalentMemoryOrdering(SDValue OldChain, SDValue NewMemOpChain)
If an existing load has uses of its chain, create a token factor node with that chain and the new mem...
LLVM_ABI bool isConstantIntBuildVectorOrConstantInt(SDValue N, bool AllowOpaques=true) const
Test whether the given value is a constant int or similar node.
LLVM_ABI void ReplaceAllUsesOfValuesWith(const SDValue *From, const SDValue *To, unsigned Num)
Like ReplaceAllUsesOfValueWith, but for multiple values at once.
LLVM_ABI SDValue getJumpTableDebugInfo(int JTI, SDValue Chain, const SDLoc &DL)
LLVM_ABI SDValue getSymbolFunctionGlobalAddress(SDValue Op, Function **TargetFunction=nullptr)
Return a GlobalAddress of the function from the current module with name matching the given ExternalS...
LLVM_ABI std::optional< unsigned > getValidMaximumShiftAmount(SDValue V, const APInt &DemandedElts, unsigned Depth=0) const
If a SHL/SRA/SRL node V has shift amounts that are all less than the element bit-width of the shift n...
LLVM_ABI SDValue UnrollVectorOp(SDNode *N, unsigned ResNE=0)
Utility function used by legalize and lowering to "unroll" a vector operation by splitting out the sc...
LLVM_ABI SDValue getVScale(const SDLoc &DL, EVT VT, APInt MulImm)
Return a node that represents the runtime scaling 'MulImm * RuntimeVL'.
LLVM_ABI SDValue getConstantFP(double Val, const SDLoc &DL, EVT VT, bool isTarget=false)
Create a ConstantFPSDNode wrapping a constant value.
OverflowKind
Used to represent the possible overflow behavior of an operation.
static LLVM_ABI unsigned getHasPredecessorMaxSteps()
LLVM_ABI bool haveNoCommonBitsSet(SDValue A, SDValue B) const
Return true if A and B have no common bits set.
LLVM_ABI SDValue getAddrSpaceCast(const SDLoc &dl, EVT VT, SDValue Ptr, unsigned SrcAS, unsigned DestAS, const SDNodeFlags Flags=SDNodeFlags())
Return an AddrSpaceCastSDNode.
SDValue getExtractSubvector(const SDLoc &DL, EVT VT, SDValue Vec, unsigned Idx)
Return the VT typed sub-vector of Vec at Idx.
LLVM_ABI bool cannotBeOrderedNegativeFP(SDValue Op) const
Test whether the given float value is known to be positive.
LLVM_ABI SDValue getRegister(Register Reg, EVT VT)
LLVM_ABI bool calculateDivergence(SDNode *N)
LLVM_ABI std::pair< SDValue, SDValue > getStrcmp(SDValue Chain, const SDLoc &dl, SDValue S0, SDValue S1, const CallInst *CI)
Lower a strcmp operation into a target library call and return the resulting chain and call result as...
LLVM_ABI SDValue getGetFPEnv(SDValue Chain, const SDLoc &dl, SDValue Ptr, EVT MemVT, MachineMemOperand *MMO)
LLVM_ABI SDValue getAssertAlign(const SDLoc &DL, SDValue V, Align A)
Return an AssertAlignSDNode.
LLVM_ABI SDNode * mutateStrictFPToFP(SDNode *Node)
Mutate the specified strict FP node to its non-strict equivalent, unlinking the node from its chain a...
LLVM_ABI bool canIgnoreSignBitOfZero(const SDUse &Use) const
Check if a use of a float value is insensitive to signed zeros.
LLVM_ABI bool SignBitIsZeroFP(SDValue Op, unsigned Depth=0) const
Return true if the sign bit of Op is known to be zero, for a floating-point value.
LLVM_ABI SDValue getMemIntrinsicNode(unsigned Opcode, const SDLoc &dl, SDVTList VTList, ArrayRef< SDValue > Ops, EVT MemVT, MachinePointerInfo PtrInfo, Align Alignment, MachineMemOperand::Flags Flags=MachineMemOperand::MOLoad|MachineMemOperand::MOStore, LocationSize Size=LocationSize::precise(0), const AAMDNodes &AAInfo=AAMDNodes())
Creates a MemIntrinsicNode that may produce a result and takes a list of operands.
SDValue getInsertSubvector(const SDLoc &DL, SDValue Vec, SDValue SubVec, unsigned Idx)
Insert SubVec at the Idx element of Vec.
LLVM_ABI SDValue getStepVector(const SDLoc &DL, EVT ResVT, const APInt &StepVal)
Returns a vector of type ResVT whose elements contain the linear sequence <0, Step,...
SDValue getSetCC(const SDLoc &DL, EVT VT, SDValue LHS, SDValue RHS, ISD::CondCode Cond, SDValue Chain=SDValue(), bool IsSignaling=false, SDNodeFlags Flags={})
Helper function to make it easier to build SetCC's if you just have an ISD::CondCode instead of an SD...
LLVM_ABI SDValue getAtomic(unsigned Opcode, const SDLoc &dl, EVT MemVT, SDValue Chain, SDValue Ptr, SDValue Val, MachineMemOperand *MMO)
Gets a node for an atomic op, produces result (if relevant) and chain and takes 2 operands.
LLVM_ABI Align getEVTAlign(EVT MemoryVT) const
Compute the default alignment value for the given type.
LLVM_ABI bool shouldOptForSize() const
LLVM_ABI bool hasSwiftErrorArg() const
LLVM_ABI SDValue buildVectorFromUnrolledParts(EVT VT, const SDLoc &DL, ArrayRef< SDValue > Scalars)
Returns a vector constructed from the scalar values in order.
LLVM_ABI SDValue getNOT(const SDLoc &DL, SDValue Val, EVT VT)
Create a bitwise NOT operation as (XOR Val, -1).
LLVM_ABI SDValue getMemcpy(SDValue Chain, const SDLoc &dl, SDValue Dst, SDValue Src, SDValue Size, Align DstAlign, Align SrcAlign, bool isVol, bool AlwaysInline, const CallInst *CI, std::optional< bool > OverrideTailCall, MachinePointerInfo DstPtrInfo, MachinePointerInfo SrcPtrInfo, const AAMDNodes &AAInfo=AAMDNodes(), BatchAAResults *BatchAA=nullptr)
const TargetLowering & getTargetLoweringInfo() const
LLVM_ABI bool isEqualTo(SDValue A, SDValue B) const
Test whether two SDValues are known to compare equal.
static constexpr unsigned MaxRecursionDepth
LLVM_ABI unsigned getMaxRuntimeNumElements(EVT VT) const
Returns the maximum runtime number of elements in VT if known, or 0 otherwise.
LLVM_ABI SDValue getStridedStoreVP(SDValue Chain, const SDLoc &DL, SDValue Val, SDValue Ptr, SDValue Offset, SDValue Stride, SDValue Mask, SDValue EVL, EVT MemVT, MachineMemOperand *MMO, ISD::MemIndexedMode AM, bool IsTruncating=false, bool IsCompressing=false)
bool isGuaranteedNotToBePoison(SDValue Op, unsigned Depth=0) const
Return true if this function can prove that Op is never poison.
LLVM_ABI SDValue getIdentityElement(unsigned Opcode, const SDLoc &DL, EVT VT, SDNodeFlags Flags)
Get the (commutative) identity element for the given opcode, if it exists.
LLVM_ABI SDValue expandVACopy(SDNode *Node)
Expand the specified ISD::VACOPY node as the Legalize pass would.
LLVM_ABI SDValue getIndexedMaskedLoad(SDValue OrigLoad, const SDLoc &dl, SDValue Base, SDValue Offset, ISD::MemIndexedMode AM)
LLVM_ABI APInt computeVectorKnownZeroElements(SDValue Op, const APInt &DemandedElts, unsigned Depth=0) const
For each demanded element of a vector, see if it is known to be zero.
LLVM_ABI void AddDbgValue(SDDbgValue *DB, bool isParameter)
Add a dbg_value SDNode.
bool NewNodesMustHaveLegalTypes
When true, additional steps are taken to ensure that getConstant() and similar functions return DAG n...
LLVM_ABI std::pair< EVT, EVT > GetSplitDestVTs(const EVT &VT) const
Compute the VTs needed for the low/hi parts of a type which is split (or expanded) into two not neces...
LLVM_ABI void salvageDebugInfo(SDNode &N)
To be invoked on an SDNode that is slated to be erased.
LLVM_ABI SDNode * MorphNodeTo(SDNode *N, unsigned Opc, SDVTList VTs, ArrayRef< SDValue > Ops)
This mutates the specified node to have the specified return type, opcode, and operands.
LLVM_ABI std::pair< SDValue, SDValue > UnrollVectorOverflowOp(SDNode *N, unsigned ResNE=0)
Like UnrollVectorOp(), but for the [US](ADD|SUB|MUL)O family of opcodes.
allnodes_const_iterator allnodes_begin() const
SDValue getUNDEF(EVT VT)
Return an UNDEF node. UNDEF does not have a useful SDLoc.
LLVM_ABI SDValue getGatherVP(SDVTList VTs, EVT VT, const SDLoc &dl, ArrayRef< SDValue > Ops, MachineMemOperand *MMO, ISD::MemIndexType IndexType)
SDValue getBuildVector(EVT VT, const SDLoc &DL, ArrayRef< SDValue > Ops)
Return an ISD::BUILD_VECTOR node.
LLVM_ABI SDValue getBitcastedAnyExtOrTrunc(SDValue Op, const SDLoc &DL, EVT VT)
Convert Op, which must be of integer type, to the integer type VT, by first bitcasting (from potentia...
LLVM_ABI bool isSplatValue(SDValue V, const APInt &DemandedElts, APInt &UndefElts, unsigned Depth=0) const
Test whether V has a splatted value for all the demanded elements.
LLVM_ABI void DeleteNode(SDNode *N)
Remove the specified node from the system.
LLVM_ABI SDValue getTruncStore(SDValue Chain, const SDLoc &dl, SDValue Val, SDValue Ptr, SDValue Offset, MachinePointerInfo PtrInfo, EVT SVT, Align Alignment, MachineMemOperand::Flags MMOFlags=MachineMemOperand::MONone, const MMOMetadata &Metadata=MMOMetadata())
LLVM_ABI SDValue getBitcast(EVT VT, SDValue V)
Return a bitcast using the SDLoc of the value operand, and casting to the provided type.
LLVM_ABI SDDbgValue * getDbgValueList(DIVariable *Var, DIExpression *Expr, ArrayRef< SDDbgOperand > Locs, ArrayRef< SDNode * > Dependencies, bool IsIndirect, const DebugLoc &DL, unsigned O, bool IsVariadic)
Creates a SDDbgValue node from a list of locations.
LLVM_ABI std::pair< SDValue, SDValue > getStrcpy(SDValue Chain, const SDLoc &dl, SDValue Dst, SDValue Src, const CallInst *CI)
Lower a strcpy operation into a target library call and return the resulting chain and call result as...
SDValue getSelect(const SDLoc &DL, EVT VT, SDValue Cond, SDValue LHS, SDValue RHS, SDNodeFlags Flags=SDNodeFlags())
Helper function to make it easier to build Select's if you just have operands and don't want to check...
LLVM_ABI SDValue getNegative(SDValue Val, const SDLoc &DL, EVT VT)
Create negative operation as (SUB 0, Val).
LLVM_ABI std::optional< unsigned > getValidShiftAmount(SDValue V, const APInt &DemandedElts, unsigned Depth=0) const
If a SHL/SRA/SRL node V has a uniform shift amount that is less than the element bit-width of the shi...
LLVM_ABI void setNodeMemRefs(MachineSDNode *N, ArrayRef< MachineMemOperand * > NewMemRefs)
Mutate the specified machine node's memory references to the provided list.
LLVM_ABI SDValue simplifySelect(SDValue Cond, SDValue TVal, SDValue FVal)
Try to simplify a select/vselect into 1 of its operands or a constant.
LLVM_ABI SDValue getZeroExtendInReg(SDValue Op, const SDLoc &DL, EVT VT)
Return the expression required to zero extend the Op value assuming it was the smaller SrcTy value.
LLVM_ABI bool isConstantFPBuildVectorOrConstantFP(SDValue N) const
Test whether the given value is a constant FP or similar node.
const DataLayout & getDataLayout() const
LLVM_ABI SDValue getPartialReduceMLS(unsigned Opc, const SDLoc &DL, SDValue Acc, SDValue LHS, SDValue RHS)
Get an expression that implements a partial multiply-subtract reduction.
LLVM_ABI SDValue expandVAArg(SDNode *Node)
Expand the specified ISD::VAARG node as the Legalize pass would.
LLVM_ABI SDValue getTokenFactor(const SDLoc &DL, SmallVectorImpl< SDValue > &Vals)
Creates a new TokenFactor containing Vals.
LLVM_ABI SDValue getStore(SDValue Chain, const SDLoc &dl, SDValue Val, SDValue Ptr, MachinePointerInfo PtrInfo, Align Alignment, MachineMemOperand::Flags MMOFlags=MachineMemOperand::MONone, const MMOMetadata &Metadata=MMOMetadata())
Helper function to build ISD::STORE nodes.
LLVM_ABI bool doesNodeExist(unsigned Opcode, SDVTList VTList, ArrayRef< SDValue > Ops)
Check if a node exists without modifying its flags.
LLVM_ABI ConstantRange computeConstantRangeIncludingKnownBits(SDValue Op, bool ForSigned, unsigned Depth=0) const
Combine constant ranges from computeConstantRange() and computeKnownBits().
const SelectionDAGTargetInfo & getSelectionDAGInfo() const
LLVM_ABI bool areNonVolatileConsecutiveLoads(LoadSDNode *LD, LoadSDNode *Base, unsigned Bytes, int Dist) const
Return true if loads are next to each other and can be merged.
LLVM_ABI SDValue getMaskedHistogram(SDVTList VTs, EVT MemVT, const SDLoc &dl, ArrayRef< SDValue > Ops, MachineMemOperand *MMO, ISD::MemIndexType IndexType)
LLVM_ABI SDDbgLabel * getDbgLabel(DILabel *Label, const DebugLoc &DL, unsigned O)
Creates a SDDbgLabel node.
LLVM_ABI SDValue getStoreVP(SDValue Chain, const SDLoc &dl, SDValue Val, SDValue Ptr, SDValue Offset, SDValue Mask, SDValue EVL, EVT MemVT, MachineMemOperand *MMO, ISD::MemIndexedMode AM, bool IsTruncating=false, bool IsCompressing=false)
LLVM_ABI OverflowKind computeOverflowForUnsignedMul(SDValue N0, SDValue N1) const
Determine if the result of the unsigned mul of 2 nodes can overflow.
LLVM_ABI void copyExtraInfo(SDNode *From, SDNode *To)
Copy extra info associated with one node to another.
LLVM_ABI SDValue getConstant(uint64_t Val, const SDLoc &DL, EVT VT, bool isTarget=false, bool isOpaque=false)
Create a ConstantSDNode wrapping a constant value.
LLVM_ABI SDValue getMemBasePlusOffset(SDValue Base, TypeSize Offset, const SDLoc &DL, const SDNodeFlags Flags=SDNodeFlags())
Returns sum of the base pointer and offset.
LLVM_ABI SDValue getGlobalAddress(const GlobalValue *GV, const SDLoc &DL, EVT VT, int64_t offset=0, bool isTargetGA=false, unsigned TargetFlags=0)
LLVM_ABI SDValue getVAArg(EVT VT, const SDLoc &dl, SDValue Chain, SDValue Ptr, SDValue SV, unsigned Align)
VAArg produces a result and token chain, and takes a pointer and a source value as input.
LLVM_ABI SDValue getLoadFFVP(EVT VT, const SDLoc &DL, SDValue Chain, SDValue Ptr, SDValue Mask, SDValue EVL, MachineMemOperand *MMO)
LLVM_ABI SDValue getTypeSize(const SDLoc &DL, EVT VT, TypeSize TS)
LLVM_ABI SDValue getMDNode(const MDNode *MD)
Return an MDNodeSDNode which holds an MDNode.
LLVM_ABI void clear()
Clear state and free memory necessary to make this SelectionDAG ready to process a new block.
LLVM_ABI std::pair< SDValue, SDValue > getMemcmp(SDValue Chain, const SDLoc &dl, SDValue Dst, SDValue Src, SDValue Size, const CallInst *CI)
Lower a memcmp operation into a target library call and return the resulting chain and call result as...
LLVM_ABI void ReplaceAllUsesWith(SDValue From, SDValue To)
Modify anything using 'From' to use 'To' instead.
LLVM_ABI SDValue getCommutedVectorShuffle(const ShuffleVectorSDNode &SV)
Returns an ISD::VECTOR_SHUFFLE node semantically equivalent to the shuffle node in input but with swa...
LLVM_ABI SDValue getExtLoad(ISD::LoadExtType ExtType, const SDLoc &dl, EVT VT, SDValue Chain, SDValue Ptr, MachinePointerInfo PtrInfo, EVT MemVT, MaybeAlign Alignment=MaybeAlign(), MachineMemOperand::Flags MMOFlags=MachineMemOperand::MONone, const MMOMetadata &Metadata=MMOMetadata())
LLVM_ABI std::pair< SDValue, SDValue > SplitVector(const SDValue &N, const SDLoc &DL, const EVT &LoVT, const EVT &HiVT)
Split the vector with EXTRACT_SUBVECTOR using the provided VTs and return the low/high part.
LLVM_ABI SDValue makeStateFunctionCall(unsigned LibFunc, SDValue Ptr, SDValue InChain, const SDLoc &DLoc)
Helper used to make a call to a library function that has one argument of pointer type.
LLVM_ABI SDValue getSignedConstant(int64_t Val, const SDLoc &DL, EVT VT, bool isTarget=false, bool isOpaque=false)
LLVM_ABI SDValue getSrcValue(const Value *v)
Construct a node to track a Value* through the backend.
SDValue getSplatVector(EVT VT, const SDLoc &DL, SDValue Op)
LLVM_ABI SDValue getAtomicMemcpy(SDValue Chain, const SDLoc &dl, SDValue Dst, SDValue Src, SDValue Size, Type *SizeTy, unsigned ElemSz, bool isTailCall, MachinePointerInfo DstPtrInfo, MachinePointerInfo SrcPtrInfo)
LLVM_ABI OverflowKind computeOverflowForSignedMul(SDValue N0, SDValue N1) const
Determine if the result of the signed mul of 2 nodes can overflow.
LLVM_ABI MaybeAlign InferPtrAlign(SDValue Ptr) const
Infer alignment of a load / store address.
LLVM_ABI void dump() const
Dump the textual format of this DAG.
LLVM_ABI bool MaskedValueIsAllOnes(SDValue Op, const APInt &Mask, unsigned Depth=0) const
Return true if '(Op & Mask) == Mask'.
LLVM_ABI bool SignBitIsZero(SDValue Op, unsigned Depth=0) const
Return true if the sign bit of Op is known to be zero.
LLVM_ABI void RemoveDeadNodes()
This method deletes all unreachable nodes in the SelectionDAG.
LLVM_ABI void RemoveDeadNode(SDNode *N)
Remove the specified node from the system.
LLVM_ABI void AddDbgLabel(SDDbgLabel *DB)
Add a dbg_label SDNode.
bool isConstantValueOfAnyType(SDValue N) const
LLVM_ABI bool canCreateUndefOrPoison(SDValue Op, const APInt &DemandedElts, UndefPoisonKind Kind=UndefPoisonKind::UndefOrPoison, bool ConsiderFlags=true, unsigned Depth=0) const
Return true if Op can create undef or poison from non-undef & non-poison operands.
SDValue getInsertVectorElt(const SDLoc &DL, SDValue Vec, SDValue Elt, unsigned Idx)
Insert Elt into Vec at offset Idx.
LLVM_ABI SDValue getTargetExtractSubreg(int SRIdx, const SDLoc &DL, EVT VT, SDValue Operand)
A convenience function for creating TargetInstrInfo::EXTRACT_SUBREG nodes.
LLVM_ABI SDValue getBasicBlock(MachineBasicBlock *MBB)
LLVM_ABI SDValue getSExtOrTrunc(SDValue Op, const SDLoc &DL, EVT VT)
Convert Op, which must be of integer type, to the integer type VT, by either sign-extending or trunca...
LLVM_ABI SDDbgValue * getVRegDbgValue(DIVariable *Var, DIExpression *Expr, Register VReg, bool IsIndirect, const DebugLoc &DL, unsigned O)
Creates a VReg SDDbgValue node.
LLVM_ABI SDValue getLoad(EVT VT, const SDLoc &dl, SDValue Chain, SDValue Ptr, MachinePointerInfo PtrInfo, MaybeAlign Alignment=MaybeAlign(), MachineMemOperand::Flags MMOFlags=MachineMemOperand::MONone, const MMOMetadata &Metadata=MMOMetadata())
Loads are not normal binary operators: their result type is not determined by their operands,...
LLVM_ABI KnownFPClass computeKnownFPClass(SDValue Op, FPClassTest InterestedClasses, unsigned Depth=0) const
Determine floating-point class information about Op.
LLVM_ABI bool isIdentityElement(unsigned Opc, SDNodeFlags Flags, SDValue V, unsigned OperandNo, unsigned Depth=0) const
Returns true if V is an identity element of Opc with Flags.
LLVM_ABI SDValue getEHLabel(const SDLoc &dl, SDValue Root, MCSymbol *Label)
LLVM_ABI bool isGuaranteedNotToBeUndefOrPoison(SDValue Op, UndefPoisonKind Kind=UndefPoisonKind::UndefOrPoison, unsigned Depth=0) const
Return true if this function can prove that Op is never poison and, Kind can be used to track poison ...
LLVM_ABI bool isKnownNeverZero(SDValue Op, unsigned Depth=0) const
Test whether the given SDValue is known to contain non-zero value(s).
LLVM_ABI SDValue getIndexedStore(SDValue OrigStore, const SDLoc &dl, SDValue Base, SDValue Offset, ISD::MemIndexedMode AM)
LLVM_ABI SDValue FoldConstantArithmetic(unsigned Opcode, const SDLoc &DL, EVT VT, ArrayRef< SDValue > Ops, SDNodeFlags Flags=SDNodeFlags())
LLVM_ABI std::optional< unsigned > getValidMinimumShiftAmount(SDValue V, const APInt &DemandedElts, unsigned Depth=0) const
If a SHL/SRA/SRL node V has shift amounts that are all less than the element bit-width of the shift n...
LLVM_ABI SDValue getSetFPEnv(SDValue Chain, const SDLoc &dl, SDValue Ptr, EVT MemVT, MachineMemOperand *MMO)
LLVM_ABI SDValue getBoolExtOrTrunc(SDValue Op, const SDLoc &SL, EVT VT, EVT OpVT)
Convert Op, which must be of integer type, to the integer type VT, by using an extension appropriate ...
LLVM_ABI SDValue getMaskedStore(SDValue Chain, const SDLoc &dl, SDValue Val, SDValue Base, SDValue Offset, SDValue Mask, EVT MemVT, MachineMemOperand *MMO, ISD::MemIndexedMode AM, bool IsTruncating=false, bool IsCompressing=false)
LLVM_ABI SDValue getExternalSymbol(const char *Sym, EVT VT)
const TargetMachine & getTarget() const
LLVM_ABI std::pair< SDValue, SDValue > getStrictFPExtendOrRound(SDValue Op, SDValue Chain, const SDLoc &DL, EVT VT)
Convert Op, which must be a STRICT operation of float type, to the float type VT, by either extending...
LLVM_ABI std::pair< SDValue, SDValue > SplitEVL(SDValue N, EVT VecVT, const SDLoc &DL)
Split the explicit vector length parameter of a VP operation.
LLVM_ABI SDValue getPtrExtOrTrunc(SDValue Op, const SDLoc &DL, EVT VT)
Convert Op, which must be of integer type, to the integer type VT, by either truncating it or perform...
LLVM_ABI SDValue getMaskFromElementCount(const SDLoc &DL, EVT VT, ElementCount Len)
Return a vector with the first 'Len' lanes set to true and remaining lanes set to false.
LLVM_ABI SDValue getAnyExtOrTrunc(SDValue Op, const SDLoc &DL, EVT VT)
Convert Op, which must be of integer type, to the integer type VT, by either any-extending or truncat...
iterator_range< allnodes_iterator > allnodes()
LLVM_ABI SDValue getBlockAddress(const BlockAddress *BA, EVT VT, int64_t Offset=0, bool isTarget=false, unsigned TargetFlags=0)
LLVM_ABI SDValue WidenVector(const SDValue &N, const SDLoc &DL)
Widen the vector up to the next power of two using INSERT_SUBVECTOR.
const LibcallLoweringInfo & getLibcalls() const
LLVM_ABI SDValue getLoadVP(ISD::MemIndexedMode AM, ISD::LoadExtType ExtType, EVT VT, const SDLoc &dl, SDValue Chain, SDValue Ptr, SDValue Offset, SDValue Mask, SDValue EVL, MachinePointerInfo PtrInfo, EVT MemVT, Align Alignment, MachineMemOperand::Flags MMOFlags, const AAMDNodes &AAInfo, const MDNode *Ranges=nullptr, bool IsExpanding=false)
LLVM_ABI SDValue getIntPtrConstant(uint64_t Val, const SDLoc &DL, bool isTarget=false)
LLVM_ABI SDDbgValue * getConstantDbgValue(DIVariable *Var, DIExpression *Expr, const Value *C, const DebugLoc &DL, unsigned O)
Creates a constant SDDbgValue node.
LLVM_ABI SDValue getScatterVP(SDVTList VTs, EVT VT, const SDLoc &dl, ArrayRef< SDValue > Ops, MachineMemOperand *MMO, ISD::MemIndexType IndexType)
LLVM_ABI SDValue getValueType(EVT)
LLVM_ABI SDValue getLifetimeNode(bool IsStart, const SDLoc &dl, SDValue Chain, int FrameIndex)
Creates a LifetimeSDNode that starts (IsStart==true) or ends (IsStart==false) the lifetime of the Fra...
ArrayRef< SDDbgValue * > GetDbgValues(const SDNode *SD) const
Get the debug values which reference the given SDNode.
LLVM_ABI SDValue getNode(unsigned Opcode, const SDLoc &DL, EVT VT, ArrayRef< SDUse > Ops)
Gets or creates the specified node.
LLVM_ABI OverflowKind computeOverflowForSignedAdd(SDValue N0, SDValue N1) const
Determine if the result of the signed addition of 2 nodes can overflow.
LLVM_ABI SDValue getFPExtendOrRound(SDValue Op, const SDLoc &DL, EVT VT)
Convert Op, which must be of float type, to the float type VT, by either extending or rounding (by tr...
LLVM_ABI unsigned AssignTopologicalOrder()
Topological-sort the AllNodes list and a assign a unique node id for each node in the DAG based on th...
ilist< SDNode >::size_type allnodes_size() const
LLVM_ABI bool isKnownNeverNaN(SDValue Op, const APInt &DemandedElts, bool SNaN=false, unsigned Depth=0) const
Test whether the given SDValue (or all elements of it, if it is a vector) is known to never be NaN in...
LLVM_ABI SDValue FoldConstantBuildVector(BuildVectorSDNode *BV, const SDLoc &DL, EVT DstEltVT)
Fold BUILD_VECTOR of constants/undefs to the destination type BUILD_VECTOR of constants/undefs elemen...
LLVM_ABI SDValue getAtomicMemmove(SDValue Chain, const SDLoc &dl, SDValue Dst, SDValue Src, SDValue Size, Type *SizeTy, unsigned ElemSz, bool isTailCall, MachinePointerInfo DstPtrInfo, MachinePointerInfo SrcPtrInfo)
LLVM_ABI SDValue getIndexedMaskedStore(SDValue OrigStore, const SDLoc &dl, SDValue Base, SDValue Offset, ISD::MemIndexedMode AM)
LLVM_ABI SDValue getTruncStoreVP(SDValue Chain, const SDLoc &dl, SDValue Val, SDValue Ptr, SDValue Mask, SDValue EVL, MachinePointerInfo PtrInfo, EVT SVT, Align Alignment, MachineMemOperand::Flags MMOFlags, const AAMDNodes &AAInfo, bool IsCompressing=false)
SDValue getTargetConstant(uint64_t Val, const SDLoc &DL, EVT VT, bool isOpaque=false)
LLVM_ABI unsigned ComputeNumSignBits(SDValue Op, unsigned Depth=0) const
Return the number of times the sign bit of the register is replicated into the other bits.
LLVM_ABI bool MaskedVectorIsZero(SDValue Op, const APInt &DemandedElts, unsigned Depth=0) const
Return true if 'Op' is known to be zero in DemandedElts.
LLVM_ABI SDValue getBoolConstant(bool V, const SDLoc &DL, EVT VT, EVT OpVT)
Create a true or false constant of type VT using the target's BooleanContent for type OpVT.
LLVM_ABI SDDbgValue * getFrameIndexDbgValue(DIVariable *Var, DIExpression *Expr, unsigned FI, bool IsIndirect, const DebugLoc &DL, unsigned O)
Creates a FrameIndex SDDbgValue node.
LLVM_ABI SDValue getExtStridedLoadVP(ISD::LoadExtType ExtType, const SDLoc &DL, EVT VT, SDValue Chain, SDValue Ptr, SDValue Stride, SDValue Mask, SDValue EVL, EVT MemVT, MachineMemOperand *MMO, bool IsExpanding=false)
LLVM_ABI SDValue getMemmove(SDValue Chain, const SDLoc &dl, SDValue Dst, SDValue Src, SDValue Size, Align DstAlign, Align SrcAlign, bool isVol, const CallInst *CI, std::optional< bool > OverrideTailCall, MachinePointerInfo DstPtrInfo, MachinePointerInfo SrcPtrInfo, const AAMDNodes &AAInfo=AAMDNodes(), BatchAAResults *BatchAA=nullptr)
LLVM_ABI SDValue getJumpTable(int JTI, EVT VT, bool isTarget=false, unsigned TargetFlags=0)
LLVM_ABI bool isBaseWithConstantOffset(SDValue Op) const
Return true if the specified operand is an ISD::ADD with a ConstantSDNode on the right-hand side,...
LLVM_ABI SDValue getVectorIdxConstant(uint64_t Val, const SDLoc &DL, bool isTarget=false)
LLVM_ABI void getTopologicallyOrderedNodes(SmallVectorImpl< const SDNode * > &SortedNodes) const
Get all the nodes in their topological order without modifying any states.
LLVM_ABI void ReplaceAllUsesOfValueWith(SDValue From, SDValue To)
Replace any uses of From with To, leaving uses of other values produced by From.getNode() alone.
MachineFunction & getMachineFunction() const
LLVM_ABI std::pair< SDValue, SDValue > getStrstr(SDValue Chain, const SDLoc &dl, SDValue S0, SDValue S1, const CallInst *CI)
Lower a strstr operation into a target library call and return the resulting chain and call result as...
LLVM_ABI SDValue getPtrExtendInReg(SDValue Op, const SDLoc &DL, EVT VT)
Return the expression required to extend the Op as a pointer value assuming it was the smaller SrcTy ...
LLVM_ABI OverflowKind computeOverflowForUnsignedAdd(SDValue N0, SDValue N1) const
Determine if the result of the unsigned addition of 2 nodes can overflow.
SDValue getPOISON(EVT VT)
Return a POISON node. POISON does not have a useful SDLoc.
SDValue getSplatBuildVector(EVT VT, const SDLoc &DL, SDValue Op)
Return a splat ISD::BUILD_VECTOR node, consisting of Op splatted to all elements.
LLVM_ABI SDValue getErrorMergeValues(ArrayRef< EVT > ResultTypes, SDValue Chain, const SDLoc &dl)
Return poison values for each of ResultTypes, substituting Chain for any result of type MVT::Other,...
LLVM_ABI SDValue getFrameIndex(int FI, EVT VT, bool isTarget=false)
LLVM_ABI void canonicalizeCommutativeBinop(unsigned Opcode, SDValue &N1, SDValue &N2) const
Swap N1 and N2 if Opcode is a commutative binary opcode and the canonical form expects the opposite o...
LLVM_ABI KnownBits computeKnownBits(SDValue Op, unsigned Depth=0) const
Determine which bits of Op are known to be either zero or one and return them in Known.
LLVM_ABI SDValue getRegisterMask(const uint32_t *RegMask)
LLVM_ABI SDValue getZExtOrTrunc(SDValue Op, const SDLoc &DL, EVT VT)
Convert Op, which must be of integer type, to the integer type VT, by either zero-extending or trunca...
LLVM_ABI SDValue getCondCode(ISD::CondCode Cond)
LLVM_ABI bool MaskedValueIsZero(SDValue Op, const APInt &Mask, unsigned Depth=0) const
Return true if 'Op & Mask' is known to be zero.
LLVM_ABI bool isKnownToBeAPowerOfTwoFP(SDValue Val, unsigned Depth=0) const
Test if the given fp value is known to be an integer power-of-2, either positive or negative.
LLVM_ABI OverflowKind computeOverflowForSignedSub(SDValue N0, SDValue N1) const
Determine if the result of the signed sub of 2 nodes can overflow.
LLVM_ABI void init(MachineFunction &NewMF, const TargetLibraryInfo *LibraryInfo, const LibcallLoweringInfo *LibcallsInfo, UniformityInfo *UA, ProfileSummaryInfo *PSIin, BlockFrequencyInfo *BFIin, FunctionVarLocs const *FnVarLocs)
Prepare this SelectionDAG to process code in the given MachineFunction.
SDValue getObjectPtrOffset(const SDLoc &SL, SDValue Ptr, TypeSize Offset)
Create an add instruction with appropriate flags when used for addressing some offset of an object.
LLVMContext * getContext() const
LLVM_ABI SDValue simplifyFPBinop(unsigned Opcode, SDValue X, SDValue Y, SDNodeFlags Flags)
Try to simplify a floating-point binary operation into 1 of its operands or a constant.
const SDValue & setRoot(SDValue N)
Set the current root tag of the SelectionDAG.
LLVM_ABI bool isKnownToBeAPowerOfTwo(SDValue Val, bool OrZero=false, unsigned Depth=0) const
Test if the given value is known to have exactly one bit set.
LLVM_ABI SDValue getDeactivationSymbol(const GlobalValue *GV)
LLVM_ABI SDValue getTargetExternalSymbol(const char *Sym, EVT VT, unsigned TargetFlags=0)
LLVM_ABI SDValue getMCSymbol(MCSymbol *Sym, EVT VT)
LLVM_ABI bool isUndef(unsigned Opcode, ArrayRef< SDValue > Ops)
Return true if the result of this operation is always undefined.
LLVM_ABI SDValue CreateStackTemporary(TypeSize Bytes, Align Alignment)
Create a stack temporary based on the size in bytes and the alignment.
LLVM_ABI SDNode * UpdateNodeOperands(SDNode *N, SDValue Op)
Mutate the specified node in-place to have the specified operands.
LLVM_ABI std::pair< EVT, EVT > GetDependentSplitDestVTs(const EVT &VT, const EVT &EnvVT, bool *HiIsEmpty) const
Compute the VTs needed for the low/hi parts of a type, dependent on an enveloping VT that has been sp...
LLVM_ABI SDValue foldConstantFPMath(unsigned Opcode, const SDLoc &DL, EVT VT, ArrayRef< SDValue > Ops)
Fold floating-point operations when all operands are constants and/or undefined.
LLVM_ABI std::optional< ConstantRange > getValidShiftAmountRange(SDValue V, const APInt &DemandedElts, unsigned Depth) const
If a SHL/SRA/SRL node V has shift amounts that are all less than the element bit-width of the shift n...
LLVM_ABI SDValue FoldSymbolOffset(unsigned Opcode, EVT VT, const GlobalAddressSDNode *GA, const SDNode *N2)
LLVM_ABI SDValue getIndexedLoad(SDValue OrigLoad, const SDLoc &dl, SDValue Base, SDValue Offset, ISD::MemIndexedMode AM)
LLVM_ABI SDValue getTargetInsertSubreg(int SRIdx, const SDLoc &DL, EVT VT, SDValue Operand, SDValue Subreg)
A convenience function for creating TargetInstrInfo::INSERT_SUBREG nodes.
SDValue getEntryNode() const
Return the token chain corresponding to the entry of the function.
LLVM_ABI SDDbgValue * getDbgValue(DIVariable *Var, DIExpression *Expr, SDNode *N, unsigned R, bool IsIndirect, const DebugLoc &DL, unsigned O)
Creates a SDDbgValue node.
LLVM_ABI SDValue getMaskedLoad(EVT VT, const SDLoc &dl, SDValue Chain, SDValue Base, SDValue Offset, SDValue Mask, SDValue Src0, EVT MemVT, MachineMemOperand *MMO, ISD::MemIndexedMode AM, ISD::LoadExtType, bool IsExpanding=false)
DenormalMode getDenormalMode(EVT VT) const
Return the current function's default denormal handling kind for the given floating point type.
SDValue getSplat(EVT VT, const SDLoc &DL, SDValue Op)
Returns a node representing a splat of one value into all lanes of the provided vector type.
LLVM_ABI std::pair< SDValue, SDValue > SplitScalar(const SDValue &N, const SDLoc &DL, const EVT &LoVT, const EVT &HiVT)
Split the scalar node with EXTRACT_ELEMENT using the provided VTs and return the low/high part.
LLVM_ABI SDValue matchBinOpReduction(SDNode *Extract, ISD::NodeType &BinOp, ArrayRef< ISD::NodeType > CandidateBinOps, bool AllowPartials=false)
Match a binop + shuffle pyramid that represents a horizontal reduction over the elements of a vector ...
LLVM_ABI bool isADDLike(SDValue Op, bool NoWrap=false) const
Return true if the specified operand is an ISD::OR or ISD::XOR node that can be treated as an ISD::AD...
LLVM_ABI SDValue getVectorShuffle(EVT VT, const SDLoc &dl, SDValue N1, SDValue N2, ArrayRef< int > Mask)
Return an ISD::VECTOR_SHUFFLE node.
LLVM_ABI SDValue simplifyShift(SDValue X, SDValue Y)
Try to simplify a shift into 1 of its operands or a constant.
LLVM_ABI bool areNonVolatileConsecutiveStores(StoreSDNode *ST, StoreSDNode *Base, unsigned Bytes, int Dist) const
Return true if stores are next to each other and can be merged.
LLVM_ABI void transferDbgValues(SDValue From, SDValue To, unsigned OffsetInBits=0, unsigned SizeInBits=0, bool InvalidateDbg=true)
Transfer debug values from one node to another, while optionally generating fragment expressions for ...
LLVM_ABI SDValue getLogicalNOT(const SDLoc &DL, SDValue Val, EVT VT)
Create a logical NOT operation as (XOR Val, BooleanOne).
LLVM_ABI SDValue getMaskedScatter(SDVTList VTs, EVT MemVT, const SDLoc &dl, ArrayRef< SDValue > Ops, MachineMemOperand *MMO, ISD::MemIndexType IndexType, bool IsTruncating=false)
ilist< SDNode >::iterator allnodes_iterator
This SDNode is used to implement the code generator support for the llvm IR shufflevector instruction...
int getMaskElt(unsigned Idx) const
ArrayRef< int > getMask() const
static void commuteMask(MutableArrayRef< int > Mask)
Change values in a shuffle permute mask assuming the two vector operands have swapped position.
static LLVM_ABI bool isSplatMask(ArrayRef< int > Mask)
A templated base class for SmallPtrSet which provides the typesafe interface that is common across al...
bool erase(PtrType Ptr)
Remove pointer from the set.
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.
SmallPtrSet - This class implements a set which is optimized for holding SmallSize or less elements.
This class consists of common code factored out of the SmallVector class to reduce code duplication b...
void assign(size_type NumElts, ValueParamT Elt)
reference emplace_back(ArgTypes &&... Args)
void reserve(size_type N)
iterator erase(const_iterator CI)
void append(ItTy in_start, ItTy in_end)
Add the specified range to the end of the SmallVector.
void push_back(const T &Elt)
This is a 'vector' (really, a variable-sized array), optimized for the case when the array is small.
This class is used to represent ISD::STORE nodes.
Represent a constant reference to a string, i.e.
Definition StringRef.h:56
constexpr const char * data() const
Get a pointer to the start of the string (which may not be null terminated).
Definition StringRef.h:138
Information about stack frame layout on the target.
virtual TargetStackID::Value getStackIDForScalableVectors() const
Returns the StackID that scalable vectors should be associated with.
Align getStackAlign() const
getStackAlignment - This method returns the number of bytes to which the stack pointer must be aligne...
Completely target-dependent object reference.
unsigned getTargetFlags() const
Provides information about what library functions are available for the current target.
virtual bool shouldConvertConstantLoadToIntImm(const APInt &Imm, Type *Ty) const
Return true if it is beneficial to convert a load of a constant to just the constant itself.
const TargetMachine & getTargetMachine() const
virtual bool isZExtFree(Type *FromTy, Type *ToTy) const
Return true if any actual instruction that defines a value of type FromTy implicitly zero-extends the...
unsigned getMaxStoresPerMemcpy(bool OptSize) const
Get maximum # of store operations permitted for llvm.memcpy.
unsigned getMaxStoresPerMemset(bool OptSize) const
Get maximum # of store operations permitted for llvm.memset.
virtual bool allowsMisalignedMemoryAccesses(EVT, unsigned AddrSpace=0, Align Alignment=Align(1), MachineMemOperand::Flags Flags=MachineMemOperand::MONone, unsigned *=nullptr) const
Determine if the target supports unaligned memory accesses.
virtual bool shallExtractConstSplatVectorElementToStore(Type *VectorTy, unsigned ElemSizeInBits, unsigned &Index) const
Return true if the target shall perform extract vector element and store given that the vector is kno...
virtual bool isTruncateFree(Type *FromTy, Type *ToTy) const
Return true if it's free to truncate a value of type FromTy to type ToTy.
virtual EVT getTypeToTransformTo(LLVMContext &Context, EVT VT) const
For types supported by the target, this is an identity function.
bool isTypeLegal(EVT VT) const
Return true if the target has native support for the specified value type.
virtual MVT getPointerTy(const DataLayout &DL, uint32_t AS=0) const
Return the pointer type for the given address space, defaults to the pointer type from the data layou...
BooleanContent
Enum that describes how the target represents true/false values.
virtual unsigned getMaxGluedStoresPerMemcpy() const
Get maximum # of store operations to be glued together.
std::vector< ArgListEntry > ArgListTy
unsigned getMaxStoresPerMemmove(bool OptSize) const
Get maximum # of store operations permitted for llvm.memmove.
virtual bool isLegalStoreImmediate(int64_t Value) const
Return true if the specified immediate is legal for the value input of a store instruction.
static ISD::NodeType getExtendForContent(BooleanContent Content)
This class defines information used to lower LLVM code to legal SelectionDAG operators that the targe...
virtual bool findOptimalMemOpLowering(LLVMContext &Context, std::vector< EVT > &MemOps, unsigned Limit, const MemOp &Op, unsigned DstAS, unsigned SrcAS, const AttributeList &FuncAttributes, EVT *LargestVT=nullptr) const
Determines the optimal series of memory ops to replace the memset / memcpy.
std::pair< SDValue, SDValue > LowerCallTo(CallLoweringInfo &CLI) const
This function lowers an abstract call to a function into an actual call.
Primary interface to the complete machine description for the target machine.
virtual bool isNoopAddrSpaceCast(unsigned SrcAS, unsigned DestAS) const
Returns true if a cast between SrcAS and DestAS is a noop.
const Triple & getTargetTriple() const
TargetRegisterInfo base class - We assume that the target defines a static array of TargetRegisterDes...
virtual const SelectionDAGTargetInfo * getSelectionDAGInfo() const
virtual const TargetRegisterInfo * getRegisterInfo() const =0
Return the target's register information.
virtual const TargetLowering * getTargetLowering() const
bool isOSDarwin() const
Is this a "Darwin" OS (macOS, iOS, tvOS, watchOS, DriverKit, XROS, or bridgeOS).
Definition Triple.h:723
Twine - A lightweight data structure for efficiently representing the concatenation of temporary valu...
Definition Twine.h:82
static constexpr TypeSize getFixed(ScalarTy ExactSize)
Definition TypeSize.h:339
The instances of the Type class are immutable: once they are created, they are never changed.
Definition Type.h:46
bool isVectorTy() const
True if this is an instance of VectorType.
Definition Type.h:283
static LLVM_ABI IntegerType * getInt32Ty(LLVMContext &C)
Definition Type.cpp:299
static LLVM_ABI Type * getVoidTy(LLVMContext &C)
Definition Type.cpp:272
static LLVM_ABI IntegerType * getInt8Ty(LLVMContext &C)
Definition Type.cpp:297
LLVM_ABI TypeSize getPrimitiveSizeInBits() const LLVM_READONLY
Return the basic size of this type if it is a primitive type.
Definition Type.cpp:187
LLVM_ABI unsigned getScalarSizeInBits() const LLVM_READONLY
If this is a vector type, return the getPrimitiveSizeInBits value for the element type.
Definition Type.cpp:222
A Use represents the edge between a Value definition and its users.
Definition Use.h:35
LLVM_ABI unsigned getOperandNo() const
Return the operand # of this use in its User.
Definition Use.cpp:35
LLVM_ABI void set(Value *Val)
Definition Value.h:876
User * getUser() const
Returns the User that contains this Use.
Definition Use.h:61
Value * getOperand(unsigned i) const
Definition User.h:207
This class is used to represent an VP_GATHER node.
This class is used to represent a VP_LOAD node.
This class is used to represent an VP_SCATTER node.
This class is used to represent a VP_STORE node.
This class is used to represent an EXPERIMENTAL_VP_STRIDED_LOAD node.
This class is used to represent an EXPERIMENTAL_VP_STRIDED_STORE node.
LLVM Value Representation.
Definition Value.h:75
Type * getType() const
All values are typed, get the type of this value.
Definition Value.h:257
std::pair< iterator, bool > insert(const ValueT &V)
Definition DenseSet.h:209
bool contains(const_arg_type_t< ValueT > V) const
Check if the set contains the given element.
Definition DenseSet.h:182
constexpr bool hasKnownScalarFactor(const FixedOrScalableQuantity &RHS) const
Returns true if there exists a value X where RHS*X will result in a value whose quantity matches our ...
Definition TypeSize.h:265
constexpr ScalarTy getFixedValue() const
Definition TypeSize.h:200
static constexpr bool isKnownLE(const FixedOrScalableQuantity &LHS, const FixedOrScalableQuantity &RHS)
Definition TypeSize.h:230
constexpr bool isScalable() const
Returns whether the quantity is scaled by a runtime quantity (vscale).
Definition TypeSize.h:168
constexpr bool isKnownEven() const
A return value of true indicates we know at compile time that the number of elements (vscale * Min) i...
Definition TypeSize.h:176
constexpr ScalarTy getKnownMinValue() const
Returns the minimum value this quantity can represent.
Definition TypeSize.h:165
constexpr LeafTy divideCoefficientBy(ScalarTy RHS) const
We do not provide the '/' operator here because division for polynomial types does not work in the sa...
Definition TypeSize.h:252
static constexpr bool isKnownGE(const FixedOrScalableQuantity &LHS, const FixedOrScalableQuantity &RHS)
Definition TypeSize.h:237
A raw_ostream that writes to an std::string.
CallInst * Call
Changed
#define llvm_unreachable(msg)
Marks that the current location is not supposed to be reachable.
LLVM_ABI APInt clmulr(const APInt &LHS, const APInt &RHS)
Perform a reversed carry-less multiply.
Definition APInt.cpp:3235
LLVM_ABI APInt mulhu(const APInt &C1, const APInt &C2)
Performs (2*N)-bit multiplication on zero-extended operands.
Definition APInt.cpp:3165
LLVM_ABI APInt avgCeilU(const APInt &C1, const APInt &C2)
Compute the ceil of the unsigned average of C1 and C2.
Definition APInt.cpp:3152
LLVM_ABI APInt avgFloorU(const APInt &C1, const APInt &C2)
Compute the floor of the unsigned average of C1 and C2.
Definition APInt.cpp:3142
LLVM_ABI APInt pext(const APInt &Val, const APInt &Mask)
Perform a "compress" operation, also known as pext or bext.
Definition APInt.cpp:3245
LLVM_ABI APInt fshr(const APInt &Hi, const APInt &Lo, const APInt &Shift)
Perform a funnel shift right.
Definition APInt.cpp:3216
LLVM_ABI APInt mulhs(const APInt &C1, const APInt &C2)
Performs (2*N)-bit multiplication on sign-extended operands.
Definition APInt.cpp:3157
LLVM_ABI APInt clmul(const APInt &LHS, const APInt &RHS)
Perform a carry-less multiply, also known as XOR multiplication, and return low-bits.
Definition APInt.cpp:3225
LLVM_ABI APInt pdep(const APInt &Val, const APInt &Mask)
Perform an "expand" operation, also known as pdep or bdep.
Definition APInt.cpp:3255
APInt abds(const APInt &A, const APInt &B)
Determine the absolute difference of two APInts considered to be signed.
Definition APInt.h:2294
LLVM_ABI APInt fshl(const APInt &Hi, const APInt &Lo, const APInt &Shift)
Perform a funnel shift left.
Definition APInt.cpp:3207
LLVM_ABI APInt ScaleBitMask(const APInt &A, unsigned NewBitWidth, bool MatchAllBits=false)
Splat/Merge neighboring bits to widen/narrow the bitmask represented by.
Definition APInt.cpp:3043
LLVM_ABI APInt clmulh(const APInt &LHS, const APInt &RHS)
Perform a carry-less multiply, and return high-bits.
Definition APInt.cpp:3240
APInt abdu(const APInt &A, const APInt &B)
Determine the absolute difference of two APInts considered to be unsigned.
Definition APInt.h:2299
LLVM_ABI APInt avgFloorS(const APInt &C1, const APInt &C2)
Compute the floor of the signed average of C1 and C2.
Definition APInt.cpp:3137
LLVM_ABI APInt avgCeilS(const APInt &C1, const APInt &C2)
Compute the ceil of the signed average of C1 and C2.
Definition APInt.cpp:3147
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.
LLVM_ABI CondCode getSetCCInverse(CondCode Operation, bool isIntegerLike)
Return the operation corresponding to !(X op Y), where 'op' is a valid SetCC operation.
ISD namespace - This namespace contains an enum which represents all of the SelectionDAG node types a...
Definition ISDOpcodes.h:24
LLVM_ABI CondCode getSetCCAndOperation(CondCode Op1, CondCode Op2, EVT Type)
Return the result of a logical AND between different comparisons of identical values: ((X op1 Y) & (X...
LLVM_ABI bool isConstantSplatVectorAllOnes(const SDNode *N, bool BuildVectorOnly=false)
Return true if the specified node is a BUILD_VECTOR or SPLAT_VECTOR where all of the elements are ~0 ...
bool isNON_EXTLoad(const SDNode *N)
Returns true if the specified node is a non-extending load.
bool matchUnaryPredicateImpl(SDValue Op, const APInt &DemandedElts, std::function< bool(ConstNodeType *)> Match, bool AllowUndefs=false, bool AllowTruncation=false)
Attempt to match a unary predicate against a scalar/splat constant or every element of a constant BUI...
NodeType
ISD::NodeType enum - This enum defines the target-independent operators for a SelectionDAG.
Definition ISDOpcodes.h:41
@ SETCC
SetCC operator - This evaluates to a true value iff the condition is true.
Definition ISDOpcodes.h:830
@ MERGE_VALUES
MERGE_VALUES - This node takes multiple discrete operands and returns them all as its individual resu...
Definition ISDOpcodes.h:261
@ TargetConstantPool
Definition ISDOpcodes.h:189
@ MDNODE_SDNODE
MDNODE_SDNODE - This is a node that holdes an MDNode*, which is used to reference metadata in the IR.
@ STRICT_FSETCC
STRICT_FSETCC/STRICT_FSETCCS - Constrained versions of SETCC, used for floating-point operands only.
Definition ISDOpcodes.h:514
@ PTRADD
PTRADD represents pointer arithmetic semantics, for targets that opt in using shouldPreservePtrArith(...
@ DELETED_NODE
DELETED_NODE - This is an illegal value that is used to catch errors.
Definition ISDOpcodes.h:45
@ POISON
POISON - A poison node.
Definition ISDOpcodes.h:236
@ PARTIAL_REDUCE_SMLA
PARTIAL_REDUCE_[U|S]MLA(Accumulator, Input1, Input2) The partial reduction nodes sign or zero extend ...
@ VECREDUCE_SEQ_FADD
Generic reduction nodes.
@ MLOAD
Masked load and store - consecutive vector load and store operations with additional mask operand tha...
@ VECREDUCE_FMINIMUMNUM
@ FGETSIGN
INT = FGETSIGN(FP) - Return the sign bit of the specified floating point value as an integer 0/1 valu...
Definition ISDOpcodes.h:541
@ SMUL_LOHI
SMUL_LOHI/UMUL_LOHI - Multiply two integers of type iN, producing a signed/unsigned value of type i[2...
Definition ISDOpcodes.h:275
@ INSERT_SUBVECTOR
INSERT_SUBVECTOR(VECTOR1, VECTOR2, IDX) - Returns a vector with VECTOR2 inserted into VECTOR1.
Definition ISDOpcodes.h:603
@ JUMP_TABLE_DEBUG_INFO
JUMP_TABLE_DEBUG_INFO - Jumptable debug info.
@ BSWAP
Byte Swap and Counting operators.
Definition ISDOpcodes.h:790
@ TargetBlockAddress
Definition ISDOpcodes.h:191
@ DEACTIVATION_SYMBOL
Untyped node storing deactivation symbol reference (DeactivationSymbolSDNode).
@ ATOMIC_STORE
OUTCHAIN = ATOMIC_STORE(INCHAIN, val, ptr) This corresponds to "store atomic" instruction.
@ ADDC
Carry-setting nodes for multiple precision addition and subtraction.
Definition ISDOpcodes.h:294
@ FMAD
FMAD - Perform a * b + c, while getting the same result as the separately rounded operations.
Definition ISDOpcodes.h:525
@ ADD
Simple integer binary arithmetic operators.
Definition ISDOpcodes.h:264
@ LOAD
LOAD and STORE have token chains as their first operand, then the same operands as an LLVM load/store...
@ ANY_EXTEND
ANY_EXTEND - Used for integer types. The high bits are undefined.
Definition ISDOpcodes.h:864
@ ATOMIC_LOAD_USUB_COND
@ FMA
FMA - Perform a * b + c with no intermediate rounding step.
Definition ISDOpcodes.h:521
@ FATAN2
FATAN2 - atan2, inspired by libm.
@ INTRINSIC_VOID
OUTCHAIN = INTRINSIC_VOID(INCHAIN, INTRINSICID, arg1, arg2, ...) This node represents a target intrin...
Definition ISDOpcodes.h:220
@ GlobalAddress
Definition ISDOpcodes.h:88
@ ATOMIC_CMP_SWAP_WITH_SUCCESS
Val, Success, OUTCHAIN = ATOMIC_CMP_SWAP_WITH_SUCCESS(INCHAIN, ptr, cmp, swap) N.b.
@ SINT_TO_FP
[SU]INT_TO_FP - These operators convert integers (whose interpreted sign depends on the first letter)...
Definition ISDOpcodes.h:891
@ CONCAT_VECTORS
CONCAT_VECTORS(VECTOR0, VECTOR1, ...) - Given a number of values of vector type with the same length ...
Definition ISDOpcodes.h:587
@ VECREDUCE_FMAX
FMIN/FMAX nodes can have flags, for NaN/NoNaN variants.
@ FADD
Simple binary floating point operators.
Definition ISDOpcodes.h:418
@ VECREDUCE_FMAXIMUM
FMINIMUM/FMAXIMUM nodes propatate NaNs and signed zeroes using the llvm.minimum and llvm....
@ ABS
ABS - Determine the unsigned absolute value of a signed integer value of the same bitwidth.
Definition ISDOpcodes.h:750
@ SIGN_EXTEND_VECTOR_INREG
SIGN_EXTEND_VECTOR_INREG(Vector) - This operator represents an in-register sign-extension of the low ...
Definition ISDOpcodes.h:921
@ FP16_TO_FP
FP16_TO_FP, FP_TO_FP16 - These operators are used to perform promotions and truncation for half-preci...
@ FMULADD
FMULADD - Performs a * b + c, with, or without, intermediate rounding.
Definition ISDOpcodes.h:531
@ BITCAST
BITCAST - This operator converts between integer, vector and FP values, as if the value was stored to...
@ BUILD_PAIR
BUILD_PAIR - This is the opposite of EXTRACT_ELEMENT in some ways.
Definition ISDOpcodes.h:254
@ CLMUL
Carry-less multiplication operations.
Definition ISDOpcodes.h:781
@ FLDEXP
FLDEXP - ldexp, inspired by libm (op0 * 2**op1).
@ BUILTIN_OP_END
BUILTIN_OP_END - This must be the last enum value in this list.
@ GlobalTLSAddress
Definition ISDOpcodes.h:89
@ SRCVALUE
SRCVALUE - This is a node type that holds a Value* that is used to make reference to a value in the L...
@ EH_LABEL
EH_LABEL - Represents a label in mid basic block used to track locations needed for debug and excepti...
@ ATOMIC_LOAD_USUB_SAT
@ CTLZ_ZERO_POISON
Definition ISDOpcodes.h:799
@ ANNOTATION_LABEL
ANNOTATION_LABEL - Represents a mid basic block label used by annotations.
@ PARTIAL_REDUCE_UMLA
@ SIGN_EXTEND
Conversion operators.
Definition ISDOpcodes.h:855
@ AVGCEILS
AVGCEILS/AVGCEILU - Rounding averaging add - Add two integers using an integer of type i[N+2],...
Definition ISDOpcodes.h:718
@ SCALAR_TO_VECTOR
SCALAR_TO_VECTOR(VAL) - This represents the operation of loading a scalar value into element 0 of the...
Definition ISDOpcodes.h:668
@ TargetExternalSymbol
Definition ISDOpcodes.h:190
@ VECREDUCE_FADD
These reductions have relaxed evaluation order semantics, and have a single vector operand.
@ TargetJumpTable
Definition ISDOpcodes.h:188
@ TargetIndex
TargetIndex - Like a constant pool entry, but with completely target-dependent semantics.
Definition ISDOpcodes.h:198
@ PARTIAL_REDUCE_FMLA
@ PREFETCH
PREFETCH - This corresponds to a prefetch intrinsic.
@ TRUNCATE_SSAT_U
Definition ISDOpcodes.h:884
@ VECREDUCE_FMAXIMUMNUM
FMINIMUMNUM/FMAXIMUMNUM nodes do not propagate NaNs and order signed zeroes using the llvm....
@ SETCCCARRY
Like SetCC, ops #0 and #1 are the LHS and RHS operands to compare, but op #2 is a boolean indicating ...
Definition ISDOpcodes.h:838
@ FNEG
Perform various unary floating-point operations inspired by libm.
@ BR_CC
BR_CC - Conditional branch.
@ SSUBO
Same for subtraction.
Definition ISDOpcodes.h:353
@ STEP_VECTOR
STEP_VECTOR(IMM) - Returns a scalable vector whose lanes are comprised of a linear sequence of unsign...
Definition ISDOpcodes.h:694
@ FCANONICALIZE
Returns platform specific canonical encoding of a floating point number.
Definition ISDOpcodes.h:544
@ IS_FPCLASS
Performs a check of floating point class property, defined by IEEE-754.
Definition ISDOpcodes.h:551
@ SSUBSAT
RESULT = [US]SUBSAT(LHS, RHS) - Perform saturation subtraction on 2 integers with the same bit width ...
Definition ISDOpcodes.h:375
@ SELECT
Select(COND, TRUEVAL, FALSEVAL).
Definition ISDOpcodes.h:807
@ ATOMIC_LOAD
Val, OUTCHAIN = ATOMIC_LOAD(INCHAIN, ptr) This corresponds to "load atomic" instruction.
@ UNDEF
UNDEF - An undefined node.
Definition ISDOpcodes.h:233
@ EXTRACT_ELEMENT
EXTRACT_ELEMENT - This is used to get the lower or upper (determined by a Constant,...
Definition ISDOpcodes.h:247
@ SPLAT_VECTOR
SPLAT_VECTOR(VAL) - Returns a vector with the scalar value VAL duplicated in all lanes.
Definition ISDOpcodes.h:675
@ AssertAlign
AssertAlign - These nodes record if a register contains a value that has a known alignment and the tr...
Definition ISDOpcodes.h:69
@ GET_ACTIVE_LANE_MASK
GET_ACTIVE_LANE_MASK - this corrosponds to the llvm.get.active.lane.mask intrinsic.
@ BasicBlock
Various leaf nodes.
Definition ISDOpcodes.h:81
@ CopyFromReg
CopyFromReg - This node indicates that the input value is a virtual or physical register that is defi...
Definition ISDOpcodes.h:230
@ SADDO
RESULT, BOOL = [SU]ADDO(LHS, RHS) - Overflow-aware nodes for addition.
Definition ISDOpcodes.h:349
@ TargetGlobalAddress
TargetGlobalAddress - Like GlobalAddress, but the DAG does no folding or anything else with this node...
Definition ISDOpcodes.h:185
@ ARITH_FENCE
ARITH_FENCE - This corresponds to a arithmetic fence intrinsic.
@ CTLS
Count leading redundant sign bits.
Definition ISDOpcodes.h:803
@ VECREDUCE_ADD
Integer reductions may have a result type larger than the vector element type.
@ MULHU
MULHU/MULHS - Multiply high - Multiply two integers of type iN, producing an unsigned/signed value of...
Definition ISDOpcodes.h:707
@ ATOMIC_LOAD_FMAXIMUM
@ SHL
Shift and rotation operations.
Definition ISDOpcodes.h:772
@ AssertNoFPClass
AssertNoFPClass - These nodes record if a register contains a float value that is known to be not som...
Definition ISDOpcodes.h:78
@ VECTOR_SHUFFLE
VECTOR_SHUFFLE(VEC1, VEC2) - Returns a vector, of the same type as VEC1/VEC2.
Definition ISDOpcodes.h:652
@ EXTRACT_SUBVECTOR
EXTRACT_SUBVECTOR(VECTOR, IDX) - Returns a subvector from VECTOR.
Definition ISDOpcodes.h:617
@ FMINNUM_IEEE
FMINNUM_IEEE/FMAXNUM_IEEE - Perform floating-point minimumNumber or maximumNumber on two values,...
@ EntryToken
EntryToken - This is the marker used to indicate the start of a region.
Definition ISDOpcodes.h:48
@ EXTRACT_VECTOR_ELT
EXTRACT_VECTOR_ELT(VECTOR, IDX) - Returns a single element from VECTOR identified by the (potentially...
Definition ISDOpcodes.h:579
@ CopyToReg
CopyToReg - This node has three operands: a chain, a register number to set to this value,...
Definition ISDOpcodes.h:224
@ ZERO_EXTEND
ZERO_EXTEND - Used for integer types, zeroing the new bits.
Definition ISDOpcodes.h:861
@ TargetConstantFP
Definition ISDOpcodes.h:180
@ SELECT_CC
Select with condition operator - This selects between a true value and a false value (ops #2 and #3) ...
Definition ISDOpcodes.h:822
@ VSCALE
VSCALE(IMM) - Returns the runtime scaling factor used to calculate the number of elements within a sc...
@ ATOMIC_CMP_SWAP
Val, OUTCHAIN = ATOMIC_CMP_SWAP(INCHAIN, ptr, cmp, swap) For double-word atomic operations: ValLo,...
@ FMINNUM
FMINNUM/FMAXNUM - Perform floating-point minimum maximum on two values, following IEEE-754 definition...
@ SSHLSAT
RESULT = [US]SHLSAT(LHS, RHS) - Perform saturation left shift.
Definition ISDOpcodes.h:387
@ SMULO
Same for multiplication.
Definition ISDOpcodes.h:357
@ ATOMIC_LOAD_FMINIMUM
@ TargetFrameIndex
Definition ISDOpcodes.h:187
@ VECTOR_SPLICE_LEFT
VECTOR_SPLICE_LEFT(VEC1, VEC2, OFFSET) - Shifts CONCAT_VECTORS(VEC1, VEC2) left by OFFSET elements an...
Definition ISDOpcodes.h:656
@ ANY_EXTEND_VECTOR_INREG
ANY_EXTEND_VECTOR_INREG(Vector) - This operator represents an in-register any-extension of the low la...
Definition ISDOpcodes.h:910
@ SIGN_EXTEND_INREG
SIGN_EXTEND_INREG - This operator atomically performs a SHL/SRA pair to sign extend a small value in ...
Definition ISDOpcodes.h:899
@ SMIN
[US]{MIN/MAX} - Binary minimum or maximum of signed or unsigned integers.
Definition ISDOpcodes.h:730
@ MASKED_UDIV
Masked vector arithmetic that returns poison on disabled lanes.
@ LIFETIME_START
This corresponds to the llvm.lifetime.
@ FP_EXTEND
X = FP_EXTEND(Y) - Extend a smaller FP type into a larger FP type.
Definition ISDOpcodes.h:989
@ VSELECT
Select with a vector condition (op #0) and two vector operands (ops #1 and #2), returning a vector re...
Definition ISDOpcodes.h:816
@ UADDO_CARRY
Carry-using nodes for multiple precision addition and subtraction.
Definition ISDOpcodes.h:329
@ MGATHER
Masked gather and scatter - load and store operations for a vector of random addresses with additiona...
@ HANDLENODE
HANDLENODE node - Used as a handle for various purposes.
@ BF16_TO_FP
BF16_TO_FP, FP_TO_BF16 - These operators are used to perform promotions and truncation for bfloat16.
@ ATOMIC_LOAD_UDEC_WRAP
@ PEXT
Parallel bit extract (compress) and parallel bit deposit (expand).
Definition ISDOpcodes.h:786
@ STRICT_FP_ROUND
X = STRICT_FP_ROUND(Y, TRUNC) - Rounding 'Y' from a larger floating point type down to the precision ...
Definition ISDOpcodes.h:503
@ FMINIMUM
FMINIMUM/FMAXIMUM - NaN-propagating minimum/maximum that also treat -0.0 as less than 0....
@ FP_TO_SINT
FP_TO_[US]INT - Convert a floating point value to a signed or unsigned integer.
Definition ISDOpcodes.h:937
@ TargetConstant
TargetConstant* - Like Constant*, but the DAG does not do any folding, simplification,...
Definition ISDOpcodes.h:179
@ STRICT_FP_EXTEND
X = STRICT_FP_EXTEND(Y) - Extend a smaller FP type into a larger FP type.
Definition ISDOpcodes.h:508
@ AND
Bitwise operators - logical and, logical or, logical xor.
Definition ISDOpcodes.h:742
@ INTRINSIC_WO_CHAIN
RESULT = INTRINSIC_WO_CHAIN(INTRINSICID, arg1, arg2, ...) This node represents a target intrinsic fun...
Definition ISDOpcodes.h:205
@ GET_FPENV_MEM
Gets the current floating-point environment.
@ PSEUDO_PROBE
Pseudo probe for AutoFDO, as a place holder in a basic block to improve the sample counts quality.
@ SCMP
[US]CMP - 3-way comparison of signed or unsigned integers.
Definition ISDOpcodes.h:738
@ AVGFLOORS
AVGFLOORS/AVGFLOORU - Averaging add - Add two integers using an integer of type i[N+1],...
Definition ISDOpcodes.h:713
@ VECTOR_SPLICE_RIGHT
VECTOR_SPLICE_RIGHT(VEC1, VEC2, OFFSET) - Shifts CONCAT_VECTORS(VEC1,VEC2) right by OFFSET elements a...
Definition ISDOpcodes.h:660
@ ADDE
Carry-using nodes for multiple precision addition and subtraction.
Definition ISDOpcodes.h:304
@ SPLAT_VECTOR_PARTS
SPLAT_VECTOR_PARTS(SCALAR1, SCALAR2, ...) - Returns a vector with the scalar values joined together a...
Definition ISDOpcodes.h:684
@ FREEZE
FREEZE - FREEZE(VAL) returns an arbitrary value if VAL is UNDEF (or is evaluated to UNDEF),...
Definition ISDOpcodes.h:241
@ INSERT_VECTOR_ELT
INSERT_VECTOR_ELT(VECTOR, VAL, IDX) - Returns VECTOR with the element at IDX replaced with VAL.
Definition ISDOpcodes.h:568
@ TokenFactor
TokenFactor - This node takes multiple tokens as input and produces a single token result.
Definition ISDOpcodes.h:53
@ ATOMIC_SWAP
Val, OUTCHAIN = ATOMIC_SWAP(INCHAIN, ptr, amt) Val, OUTCHAIN = ATOMIC_LOAD_[OpName](INCHAIN,...
@ CTTZ_ZERO_POISON
Bit counting operators with a poisoned result for zero inputs.
Definition ISDOpcodes.h:798
@ ExternalSymbol
Definition ISDOpcodes.h:93
@ FFREXP
FFREXP - frexp, extract fractional and exponent component of a floating-point value.
@ FP_ROUND
X = FP_ROUND(Y, TRUNC) - Rounding 'Y' from a larger floating point type down to the precision of the ...
Definition ISDOpcodes.h:970
@ VECTOR_COMPRESS
VECTOR_COMPRESS(Vec, Mask, Passthru) consecutively place vector elements based on mask e....
Definition ISDOpcodes.h:702
@ ZERO_EXTEND_VECTOR_INREG
ZERO_EXTEND_VECTOR_INREG(Vector) - This operator represents an in-register zero-extension of the low ...
Definition ISDOpcodes.h:932
@ ADDRSPACECAST
ADDRSPACECAST - This operator converts between pointers of different address spaces.
@ EXPERIMENTAL_VECTOR_HISTOGRAM
Experimental vector histogram intrinsic Operands: Input Chain, Inc, Mask, Base, Index,...
@ FP_TO_SINT_SAT
FP_TO_[US]INT_SAT - Convert floating point value in operand 0 to a signed or unsigned scalar integer ...
Definition ISDOpcodes.h:956
@ VECREDUCE_FMINIMUM
@ TRUNCATE
TRUNCATE - Completely drop the high bits.
Definition ISDOpcodes.h:867
@ VAARG
VAARG - VAARG has four operands: an input chain, a pointer, a SRCVALUE, and the alignment.
@ VECREDUCE_SEQ_FMUL
@ SHL_PARTS
SHL_PARTS/SRA_PARTS/SRL_PARTS - These operators are used for expanded integer shift operations.
Definition ISDOpcodes.h:844
@ AssertSext
AssertSext, AssertZext - These nodes record if a register contains a value that has already been zero...
Definition ISDOpcodes.h:62
@ ATOMIC_LOAD_UINC_WRAP
@ FCOPYSIGN
FCOPYSIGN(X, Y) - Return the value of X with the sign of Y.
Definition ISDOpcodes.h:537
@ PARTIAL_REDUCE_SUMLA
@ SADDSAT
RESULT = [US]ADDSAT(LHS, RHS) - Perform saturation addition on 2 integers with the same bit width (W)...
Definition ISDOpcodes.h:366
@ SET_FPENV_MEM
Sets the current floating point environment.
@ FMINIMUMNUM
FMINIMUMNUM/FMAXIMUMNUM - minimumnum/maximumnum that is same with FMINNUM_IEEE and FMAXNUM_IEEE besid...
@ TRUNCATE_SSAT_S
TRUNCATE_[SU]SAT_[SU] - Truncate for saturated operand [SU] located in middle, prefix for SAT means i...
Definition ISDOpcodes.h:882
@ ABDS
ABDS/ABDU - Absolute difference - Return the absolute difference between two numbers interpreted as s...
Definition ISDOpcodes.h:725
@ TRUNCATE_USAT_U
Definition ISDOpcodes.h:886
@ SADDO_CARRY
Carry-using overflow-aware nodes for multiple precision addition and subtraction.
Definition ISDOpcodes.h:339
@ INTRINSIC_W_CHAIN
RESULT,OUTCHAIN = INTRINSIC_W_CHAIN(INCHAIN, INTRINSICID, arg1, ...) This node represents a target in...
Definition ISDOpcodes.h:213
@ TargetGlobalTLSAddress
Definition ISDOpcodes.h:186
@ ABS_MIN_POISON
ABS with a poison result for INT_MIN.
Definition ISDOpcodes.h:754
@ BUILD_VECTOR
BUILD_VECTOR(ELT0, ELT1, ELT2, ELT3,...) - Return a fixed-width vector with the specified,...
Definition ISDOpcodes.h:559
LLVM_ABI NodeType getOppositeSignednessMinMaxOpcode(unsigned MinMaxOpc)
Given a MinMaxOpc of ISD::(U|S)MIN or ISD::(U|S)MAX, returns the corresponding opcode with the opposi...
LLVM_ABI bool isBuildVectorOfConstantSDNodes(const SDNode *N)
Return true if the specified node is a BUILD_VECTOR node of all ConstantSDNode or undef.
LLVM_ABI NodeType getExtForLoadExtType(bool IsFP, LoadExtType)
bool isZEXTLoad(const SDNode *N)
Returns true if the specified node is a ZEXTLOAD.
bool isExtOpcode(unsigned Opcode)
LLVM_ABI bool isConstantSplatVectorAllZeros(const SDNode *N, bool BuildVectorOnly=false)
Return true if the specified node is a BUILD_VECTOR or SPLAT_VECTOR where all of the elements are 0 o...
LLVM_ABI NodeType getUnmaskedBinOpOpcode(unsigned MaskedOpc)
Given a MaskedOpc of ISD::MASKED_(U|S)(DIV|REM), returns the unmasked ISD::(U|S)(DIV|REM).
LLVM_ABI bool isVectorShrinkable(const SDNode *N, unsigned NewEltSize, bool Signed)
Returns true if the specified node is a vector where all elements can be truncated to the specified e...
LLVM_ABI bool isVPBinaryOp(unsigned Opcode)
Whether this is a vector-predicated binary operation opcode.
LLVM_ABI CondCode getSetCCInverse(CondCode Operation, EVT Type)
Return the operation corresponding to !(X op Y), where 'op' is a valid SetCC operation.
LLVM_ABI std::optional< unsigned > getBaseOpcodeForVP(unsigned Opcode, bool hasFPExcept)
Translate this VP Opcode to its corresponding non-VP Opcode.
bool isBitwiseLogicOp(unsigned Opcode)
Whether this is bitwise logic opcode.
bool isTrueWhenEqual(CondCode Cond)
Return true if the specified condition returns true if the two operands to the condition are equal.
LLVM_ABI std::optional< unsigned > getVPMaskIdx(unsigned Opcode)
The operand position of the vector mask.
unsigned getUnorderedFlavor(CondCode Cond)
This function returns 0 if the condition is always false if an operand is a NaN, 1 if the condition i...
LLVM_ABI std::optional< unsigned > getVPExplicitVectorLengthIdx(unsigned Opcode)
The operand position of the explicit vector length parameter.
LLVM_ABI bool matchBinaryPredicate(SDValue LHS, SDValue RHS, const APInt &DemandedElts, std::function< bool(ConstantSDNode *, ConstantSDNode *)> Match, bool AllowUndefs=false, bool AllowTypeMismatch=false)
Attempt to match a binary predicate against a pair of scalar/splat constants or every element of a pa...
bool isEXTLoad(const SDNode *N)
Returns true if the specified node is a EXTLOAD.
LLVM_ABI bool allOperandsUndef(const SDNode *N)
Return true if the node has at least one operand and all operands of the specified node are ISD::UNDE...
LLVM_ABI bool isFreezeUndef(const SDNode *N)
Return true if the specified node is FREEZE(UNDEF).
LLVM_ABI CondCode getSetCCSwappedOperands(CondCode Operation)
Return the operation corresponding to (Y op X) when given the operation for (X op Y).
LLVM_ABI std::optional< unsigned > getVPForBaseOpcode(unsigned Opcode)
Translate this non-VP Opcode to its corresponding VP Opcode.
MemIndexType
MemIndexType enum - This enum defines how to interpret MGATHER/SCATTER's index parameter when calcula...
LLVM_ABI bool isBuildVectorAllZeros(const SDNode *N)
Return true if the specified node is a BUILD_VECTOR where all of the elements are 0 or undef.
LLVM_ABI bool isConstantSplatVector(const SDNode *N, APInt &SplatValue)
Node predicates.
LLVM_ABI NodeType getInverseMinMaxOpcode(unsigned MinMaxOpc)
Given a MinMaxOpc of ISD::(U|S)MIN or ISD::(U|S)MAX, returns ISD::(U|S)MAX and ISD::(U|S)MIN,...
LLVM_ABI bool isVPReduction(unsigned Opcode)
Whether this is a vector-predicated reduction opcode.
MemIndexedMode
MemIndexedMode enum - This enum defines the load / store indexed addressing modes.
LLVM_ABI bool isBuildVectorOfConstantFPSDNodes(const SDNode *N)
Return true if the specified node is a BUILD_VECTOR node of all ConstantFPSDNode or undef.
bool isSEXTLoad(const SDNode *N)
Returns true if the specified node is a SEXTLOAD.
CondCode
ISD::CondCode enum - These are ordered carefully to make the bitfields below work out,...
LLVM_ABI bool isBuildVectorAllOnes(const SDNode *N)
Return true if the specified node is a BUILD_VECTOR where all of the elements are ~0 or undef.
LLVM_ABI NodeType getVecReduceBaseOpcode(unsigned VecReduceOpcode)
Get underlying scalar opcode for VECREDUCE opcode.
LoadExtType
LoadExtType enum - This enum defines the three variants of LOADEXT (load with extension).
LLVM_ABI bool isVPOpcode(unsigned Opcode)
Whether this is a vector-predicated Opcode.
bool matchUnaryPredicate(SDValue Op, const APInt &DemandedElts, std::function< bool(ConstantSDNode *)> Match, bool AllowUndefs=false, bool AllowTruncation=false)
Hook for matching ConstantSDNode predicate.
LLVM_ABI CondCode getSetCCOrOperation(CondCode Op1, CondCode Op2, EVT Type)
Return the result of a logical OR between different comparisons of identical values: ((X op1 Y) | (X ...
BinaryOp_match< SpecificConstantMatch, SrcTy, TargetOpcode::G_SUB > m_Neg(const SrcTy &&Src)
Matches a register negated by a G_SUB.
BinaryOp_match< LHS, RHS, Instruction::And > m_And(const LHS &L, const RHS &R)
match_deferred< Value > m_Deferred(Value *const &V)
Like m_Specific(), but works if the specific value to match is determined as part of the same match()...
auto m_Value()
Match an arbitrary value and ignore it.
BinaryOp_match< LHS, RHS, Instruction::Sub > m_Sub(const LHS &L, const RHS &R)
LLVM_ABI Libcall getMEMCPY_ELEMENT_UNORDERED_ATOMIC(uint64_t ElementSize)
getMEMCPY_ELEMENT_UNORDERED_ATOMIC - Return MEMCPY_ELEMENT_UNORDERED_ATOMIC_* value for the given ele...
LLVM_ABI Libcall getMEMSET_ELEMENT_UNORDERED_ATOMIC(uint64_t ElementSize)
getMEMSET_ELEMENT_UNORDERED_ATOMIC - Return MEMSET_ELEMENT_UNORDERED_ATOMIC_* value for the given ele...
LLVM_ABI Libcall getMEMMOVE_ELEMENT_UNORDERED_ATOMIC(uint64_t ElementSize)
getMEMMOVE_ELEMENT_UNORDERED_ATOMIC - Return MEMMOVE_ELEMENT_UNORDERED_ATOMIC_* value for the given e...
bool sd_match(SDNode *N, const SelectionDAG *DAG, Pattern &&P)
LLVM_ABI unsigned rot(unsigned SrcSignBits, unsigned BitWidth, std::optional< APInt > RotAmt, bool IsRotateRight)
Compute the number of sign bits after rotating a value.
initializer< Ty > init(const Ty &Val)
@ DW_OP_LLVM_arg
Only used in LLVM metadata.
Definition Dwarf.h:149
std::enable_if_t< detail::IsValidPointer< X, Y >::value, X * > extract(Y &&MD)
Extract a Value from Metadata.
Definition Metadata.h:679
NodeAddr< NodeBase * > Node
Definition RDFGraph.h:381
This is an optimization pass for GlobalISel generic memory operations.
GenericUniformityInfo< SSAContext > UniformityInfo
unsigned Log2_32_Ceil(uint32_t Value)
Return the ceil log base 2 of the specified value, 32 if the value is zero.
Definition MathExtras.h:339
@ Offset
Definition DWP.cpp:577
bool operator<(int64_t V1, const APSInt &V2)
Definition APSInt.h:360
LLVM_ABI ISD::CondCode getICmpCondCode(ICmpInst::Predicate Pred)
getICmpCondCode - Return the ISD condition code corresponding to the given LLVM IR integer condition ...
Definition Analysis.cpp:237
void fill(R &&Range, T &&Value)
Provide wrappers to std::fill which take ranges instead of having to pass begin/end explicitly.
Definition STLExtras.h:1775
LLVM_ABI SDValue peekThroughExtractSubvectors(SDValue V)
Return the non-extracted vector source operand of V if it exists.
bool all_of(R &&range, UnaryPredicate P)
Provide wrappers to std::all_of which take ranges instead of having to pass begin/end explicitly.
Definition STLExtras.h:1755
LLVM_ABI bool isNullConstant(SDValue V)
Returns true if V is a constant integer zero.
LLVM_ABI bool isAllOnesOrAllOnesSplat(const MachineInstr &MI, const MachineRegisterInfo &MRI, bool AllowUndefs=false)
Return true if the value is a constant -1 integer or a splatted vector of a constant -1 integer (with...
Definition Utils.cpp:1557
LLVM_ABI SDValue getBitwiseNotOperand(SDValue V, SDValue Mask, bool AllowUndefs)
If V is a bitwise not, returns the inverted operand.
@ Known
Known to have no common set bits.
@ Undef
Value of the register doesn't matter.
LLVM_ABI SDValue peekThroughBitcasts(SDValue V)
Return the non-bitcasted source operand of V if it exists.
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
@ Store
The extracted value is stored (ExtractElement only).
bool isIntOrFPConstant(SDValue V)
Return true if V is either a integer or FP constant.
auto dyn_cast_if_present(const Y &Val)
dyn_cast_if_present<X> - Functionally identical to dyn_cast, except that a null (or none in the case ...
Definition Casting.h:732
LLVM_ABI bool getConstantDataArrayInfo(const Value *V, ConstantDataArraySlice &Slice, unsigned ElementSize, uint64_t Offset=0)
Returns true if the value V is a pointer into a ConstantDataArray.
LLVM_ABI bool isOneOrOneSplatFP(SDValue V, bool AllowUndefs=false)
Return true if the value is a constant floating-point value, or a splatted vector of a constant float...
int bit_width(T Value)
Returns the number of bits needed to represent Value if Value is nonzero.
Definition bit.h:325
LLVM_READONLY APFloat maximum(const APFloat &A, const APFloat &B)
Implements IEEE 754-2019 maximum semantics.
Definition APFloat.h:1801
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
LLVM_ABI bool shouldOptimizeForSize(const MachineFunction *MF, ProfileSummaryInfo *PSI, const MachineBlockFrequencyInfo *BFI, PGSOQueryType QueryType=PGSOQueryType::Other)
Returns true if machine function MF is suggested to be size-optimized based on the profile.
iterator_range< early_inc_iterator_impl< detail::IterOfRange< RangeT > > > make_early_inc_range(RangeT &&Range)
Make a range that does early increment to allow mutation of the underlying range without disrupting i...
Definition STLExtras.h:649
auto cast_or_null(const Y &Val)
Definition Casting.h:714
LLVM_ABI bool isNullOrNullSplat(const MachineInstr &MI, const MachineRegisterInfo &MRI, bool AllowUndefs=false)
Return true if the value is a constant 0 integer or a splatted vector of a constant 0 integer (with n...
Definition Utils.cpp:1539
LLVM_ABI bool isMinSignedConstant(SDValue V)
Returns true if V is a constant min signed integer value.
LLVM_ABI ConstantFPSDNode * isConstOrConstSplatFP(SDValue N, bool AllowUndefs=false)
Returns the SDNode if it is a constant splat BuildVector or constant float.
LLVM_ABI ConstantRange getConstantRangeFromMetadata(const MDNode &RangeMD)
Parse out a conservative ConstantRange from !range metadata.
APFloat frexp(const APFloat &X, int &Exp, APFloat::roundingMode RM)
Equivalent of C standard library function.
Definition APFloat.h:1713
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
auto dyn_cast_or_null(const Y &Val)
Definition Casting.h:753
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 bool getShuffleDemandedElts(int SrcWidth, ArrayRef< int > Mask, const APInt &DemandedElts, APInt &DemandedLHS, APInt &DemandedRHS, bool AllowUndefElts=false)
Transform a shuffle mask's output demanded element mask into demanded element masks for the 2 operand...
LLVM_READONLY APFloat maxnum(const APFloat &A, const APFloat &B)
Implements IEEE-754 2008 maxNum semantics.
Definition APFloat.h:1756
unsigned Log2_32(uint32_t Value)
Return the floor log base 2 of the specified value, -1 if the value is zero.
Definition MathExtras.h:326
LLVM_ABI bool isBitwiseNot(SDValue V, bool AllowUndefs=false)
Returns true if V is a bitwise not operation.
auto reverse(ContainerTy &&C)
Definition STLExtras.h:408
LLVM_ABI SDValue peekThroughInsertVectorElt(SDValue V, const APInt &DemandedElts)
Recursively peek through INSERT_VECTOR_ELT nodes, returning the source vector operand of V,...
constexpr bool isPowerOf2_32(uint32_t Value)
Return true if the argument is a power of two > 0.
Definition MathExtras.h:280
decltype(auto) get(const PointerIntPair< PointerTy, IntBits, IntType, PtrTraits, Info > &Pair)
LLVM_ABI void checkForCycles(const SelectionDAG *DAG, bool force=false)
void sort(IteratorTy Start, IteratorTy End)
Definition STLExtras.h:1652
LLVM_READONLY APFloat minimumnum(const APFloat &A, const APFloat &B)
Implements IEEE 754-2019 minimumNumber semantics.
Definition APFloat.h:1787
FPClassTest
Floating-point class tests, supported by 'is_fpclass' intrinsic.
LLVM_ABI void computeKnownBits(const Value *V, KnownBits &Known, const DataLayout &DL, AssumptionCache *AC=nullptr, const Instruction *CxtI=nullptr, const DominatorTree *DT=nullptr, bool UseInstrInfo=true, unsigned Depth=0)
Determine which bits of V are known to be either zero or one and return them in the KnownZero/KnownOn...
LLVM_ABI const MDNode * getMemCacheHintMetadata(const Instruction &I, unsigned OperandNo=0)
Return the cache hint metadata node for memory operand OperandNo on I, or nullptr when the instructio...
LLVM_ABI raw_ostream & dbgs()
dbgs() - This returns a reference to a raw_ostream for debugging messages.
Definition Debug.cpp:209
LLVM_ABI SDValue peekThroughTruncates(SDValue V)
Return the non-truncated source operand of V if it exists.
bool none_of(R &&Range, UnaryPredicate P)
Provide wrappers to std::none_of which take ranges instead of having to pass begin/end explicitly.
Definition STLExtras.h:1769
LLVM_ABI void report_fatal_error(Error Err, bool gen_crash_diag=true)
Definition Error.cpp:163
constexpr std::underlying_type_t< Enum > to_underlying(Enum E)
Returns underlying integer value of an enum.
LLVM_ABI ConstantRange getVScaleRange(const Function *F, unsigned BitWidth)
Determine the possible constant range of vscale with the given bit width, based on the vscale_range f...
LLVM_ABI SDValue peekThroughOneUseBitcasts(SDValue V)
Return the non-bitcasted and one-use source operand of V if it exists.
CodeGenOptLevel
Code generation optimization level.
Definition CodeGen.h:227
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
LLVM_ATTRIBUTE_VISIBILITY_DEFAULT AnalysisKey InnerAnalysisManagerProxy< AnalysisManagerT, IRUnitT, ExtraArgTs... >::Key
bool includesPoison(UndefPoisonKind Kind)
Returns true if Kind includes the Poison bit.
Definition UndefPoison.h:27
LLVM_ABI bool isOneOrOneSplat(SDValue V, bool AllowUndefs=false)
Return true if the value is a constant 1 integer or a splatted vector of a constant 1 integer (with n...
LLVM_ABI raw_fd_ostream & errs()
This returns a reference to a raw_ostream for standard error.
@ Other
Any other memory.
Definition ModRef.h:68
@ First
Helpers to iterate all locations in the MemoryEffectsBase class.
Definition ModRef.h:74
bool includesUndef(UndefPoisonKind Kind)
Returns true if Kind includes the Undef bit.
Definition UndefPoison.h:33
LLVM_READONLY APFloat minnum(const APFloat &A, const APFloat &B)
Implements IEEE-754 2008 minNum semantics.
Definition APFloat.h:1737
@ Mul
Product of integers.
@ Sub
Subtraction of integers.
@ Fast
Assign the register banks as fast as possible (default).
RelativeUniformCounterPtr ValuesPtrExpr VTableAddr Count
Definition InstrProf.h:145
LLVM_ABI bool isNullConstantOrUndef(SDValue V)
Returns true if V is a constant integer zero or an UNDEF node.
IntPtrTy
Definition InstrProf.h:82
LLVM_ABI bool isInTailCallPosition(const CallBase &Call, const TargetMachine &TM, bool ReturnsFirstArg=false)
Test if the given instruction is in a position to be optimized with a tail-call.
Definition Analysis.cpp:539
DWARFExpression::Operation Op
ArrayRef(const T &OneElt) -> ArrayRef< T >
LLVM_ABI ConstantSDNode * isConstOrConstSplat(SDValue N, bool AllowUndefs=false, bool AllowTruncation=false)
Returns the SDNode if it is a constant splat BuildVector or constant int.
OutputIt copy(R &&Range, OutputIt Out)
Definition STLExtras.h:1901
constexpr unsigned BitWidth
LLVM_ABI bool funcReturnsFirstArgOfCall(const CallInst &CI)
Returns true if the parent of CI returns CI's first argument after calling CI.
Definition Analysis.cpp:719
decltype(auto) cast(const From &Val)
cast<X> - Return the argument parameter cast to the specified type.
Definition Casting.h:559
LLVM_ABI bool isZeroOrZeroSplat(SDValue N, bool AllowUndefs=false)
Return true if the value is a constant 0 integer or a splatted vector of a constant 0 integer (with n...
constexpr auto seq(T Begin, T End)
Iterate over an integral type from Begin up to - but not including - End.
Definition Sequence.h:341
LLVM_ABI bool isOneConstant(SDValue V)
Returns true if V is a constant integer one.
UndefPoisonKind
Enumeration to track whether we are interested in Undef, Poison, or both.
Definition UndefPoison.h:20
bool is_contained(R &&Range, const E &Element)
Returns true if Element is found in Range.
Definition STLExtras.h:1963
Align commonAlignment(Align A, uint64_t Offset)
Returns the alignment that satisfies both alignments.
Definition Alignment.h:201
LLVM_ABI bool isNullFPConstant(SDValue V)
Returns true if V is an FP constant with a value of positive zero.
constexpr int64_t SignExtend64(uint64_t x)
Sign-extend the number in the bottom B bits of X to a 64-bit integer.
Definition MathExtras.h:567
unsigned Log2(Align A)
Returns the log2 of the alignment.
Definition Alignment.h:197
LLVM_ABI bool isZeroOrZeroSplatFP(SDValue N, bool AllowUndefs=false)
Return true if the value is a constant (+/-)0.0 floating-point value or a splatted vector thereof (wi...
bool equal(L &&LRange, R &&RRange)
Wrapper function around std::equal to detect if pair-wise elements between two ranges are the same.
Definition STLExtras.h:2162
LLVM_ABI void computeKnownBitsFromRangeMetadata(const MDNode &Ranges, KnownBits &Known)
Compute known bits from the range metadata.
LLVM_READONLY APFloat minimum(const APFloat &A, const APFloat &B)
Implements IEEE 754-2019 minimum semantics.
Definition APFloat.h:1774
LLVM_READONLY APFloat maximumnum(const APFloat &A, const APFloat &B)
Implements IEEE 754-2019 maximumNumber semantics.
Definition APFloat.h:1814
LLVM_ABI bool isOnesOrOnesSplat(SDValue N, bool AllowUndefs=false)
Return true if the value is a constant 1 integer or a splatted vector of a constant 1 integer (with n...
LLVM_ABI bool isAllOnesConstant(SDValue V)
Returns true if V is an integer constant with all bits set.
constexpr uint64_t NextPowerOf2(uint64_t A)
Returns the next power of two (in 64-bits) that is strictly greater than A.
Definition MathExtras.h:368
LLVM_ABI void reportFatalUsageError(Error Err)
Report a fatal error that does not indicate a bug in LLVM.
Definition Error.cpp:177
void swap(llvm::BitVector &LHS, llvm::BitVector &RHS)
Implement std::swap in terms of BitVector swap.
Definition BitVector.h:880
#define N
A collection of metadata nodes that might be associated with a memory access used by the alias-analys...
Definition Metadata.h:774
MDNode * TBAAStruct
The tag for type-based alias analysis (tbaa struct).
Definition Metadata.h:794
MDNode * TBAA
The tag for type-based alias analysis.
Definition Metadata.h:791
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:77
Represents offset+length into a ConstantDataArray.
uint64_t Length
Length of the slice.
uint64_t Offset
Slice starts at this Offset.
void move(uint64_t Delta)
Moves the Offset and adjusts Length accordingly.
const ConstantDataArray * Array
ConstantDataArray pointer.
Extended Value Type.
Definition ValueTypes.h:35
TypeSize getStoreSize() const
Return the number of bytes overwritten by a store of the specified value type.
Definition ValueTypes.h:418
bool isSimple() const
Test if the given EVT is simple (as opposed to being extended).
Definition ValueTypes.h:145
intptr_t getRawBits() const
Definition ValueTypes.h:543
static EVT getVectorVT(LLVMContext &Context, EVT VT, unsigned NumElements, bool IsScalable=false)
Returns the EVT that represents a vector NumElements in length, where each element is of type VT.
Definition ValueTypes.h:70
EVT changeTypeToInteger() const
Return the type converted to an equivalently sized integer or vector with integer element type.
Definition ValueTypes.h:129
bool bitsGT(EVT VT) const
Return true if this has more bits than VT.
Definition ValueTypes.h:307
bool bitsLT(EVT VT) const
Return true if this has less bits than VT.
Definition ValueTypes.h:323
bool isFloatingPoint() const
Return true if this is a FP or a vector FP type.
Definition ValueTypes.h:155
ElementCount getVectorElementCount() const
Definition ValueTypes.h:373
TypeSize getSizeInBits() const
Return the size of the specified value type in bits.
Definition ValueTypes.h:396
unsigned getVectorMinNumElements() const
Given a vector type, return the minimum number of elements it contains.
Definition ValueTypes.h:382
uint64_t getScalarSizeInBits() const
Definition ValueTypes.h:408
MVT getSimpleVT() const
Return the SimpleValueType held in the specified simple EVT.
Definition ValueTypes.h:339
static EVT getIntegerVT(LLVMContext &Context, unsigned BitWidth)
Returns the EVT that represents an integer with the given number of bits.
Definition ValueTypes.h:61
bool isFixedLengthVector() const
Definition ValueTypes.h:199
bool isVector() const
Return true if this is a vector value type.
Definition ValueTypes.h:176
EVT getScalarType() const
If this is a vector type, return the element type, otherwise return this.
Definition ValueTypes.h:346
bool bitsGE(EVT VT) const
Return true if this has no less bits than VT.
Definition ValueTypes.h:315
bool bitsEq(EVT VT) const
Return true if this has the same number of bits as VT.
Definition ValueTypes.h:279
LLVM_ABI Type * getTypeForEVT(LLVMContext &Context) const
This method returns an LLVM type corresponding to the specified EVT.
bool isScalableVector() const
Return true if this is a vector type where the runtime length is machine dependent.
Definition ValueTypes.h:187
EVT getVectorElementType() const
Given a vector type, return the type of each element.
Definition ValueTypes.h:351
bool isExtended() const
Test if the given EVT is extended (as opposed to being simple).
Definition ValueTypes.h:150
LLVM_ABI const fltSemantics & getFltSemantics() const
Returns an APFloat semantics tag appropriate for the value type.
unsigned getVectorNumElements() const
Given a vector type, return the number of elements it contains.
Definition ValueTypes.h:359
bool bitsLE(EVT VT) const
Return true if this has no more bits than VT.
Definition ValueTypes.h:331
EVT getHalfNumVectorElementsVT(LLVMContext &Context) const
Definition ValueTypes.h:484
bool isInteger() const
Return true if this is an integer or a vector integer type.
Definition ValueTypes.h:160
static KnownBits makeConstant(const APInt &C)
Create known bits from a known constant.
Definition KnownBits.h:315
static LLVM_ABI KnownBits mulhu(const KnownBits &LHS, const KnownBits &RHS)
Compute known bits from zero-extended multiply-hi.
static LLVM_ABI KnownBits smax(const KnownBits &LHS, const KnownBits &RHS)
Compute known bits for smax(LHS, RHS).
bool isNonNegative() const
Returns true if this value is known to be non-negative.
Definition KnownBits.h:106
bool isZero() const
Returns true if value is all zero.
Definition KnownBits.h:78
static LLVM_ABI KnownBits usub_sat(const KnownBits &LHS, const KnownBits &RHS)
Compute knownbits resulting from llvm.usub.sat(LHS, RHS)
static LLVM_ABI KnownBits ashr(const KnownBits &LHS, const KnownBits &RHS, bool ShAmtNonZero=false, bool Exact=false)
Compute known bits for ashr(LHS, RHS).
static LLVM_ABI KnownBits urem(const KnownBits &LHS, const KnownBits &RHS)
Compute known bits for urem(LHS, RHS).
unsigned countMaxTrailingZeros() const
Returns the maximum number of trailing zero bits possible.
Definition KnownBits.h:288
static LLVM_ABI std::optional< bool > ne(const KnownBits &LHS, const KnownBits &RHS)
Determine if these known bits always give the same ICMP_NE result.
KnownBits trunc(unsigned BitWidth) const
Return known bits for a truncation of the value we're tracking.
Definition KnownBits.h:165
KnownBits byteSwap() const
Definition KnownBits.h:559
static LLVM_ABI KnownBits fshl(const KnownBits &LHS, const KnownBits &RHS, const APInt &Amt)
Compute known bits for fshl(LHS, RHS, Amt).
unsigned countMaxPopulation() const
Returns the maximum number of bits that could be one.
Definition KnownBits.h:303
void setAllZero()
Make all bits known to be zero and discard any previous information.
Definition KnownBits.h:84
KnownBits reverseBits() const
Definition KnownBits.h:563
KnownBits concat(const KnownBits &Lo) const
Concatenate the bits from Lo onto the bottom of *this.
Definition KnownBits.h:247
unsigned getBitWidth() const
Get the bit width of this value.
Definition KnownBits.h:44
static LLVM_ABI KnownBits umax(const KnownBits &LHS, const KnownBits &RHS)
Compute known bits for umax(LHS, RHS).
KnownBits zext(unsigned BitWidth) const
Return known bits for a zero extension of the value we're tracking.
Definition KnownBits.h:176
void resetAll()
Resets the known state of all bits.
Definition KnownBits.h:72
static KnownBits add(const KnownBits &LHS, const KnownBits &RHS, bool NSW=false, bool NUW=false, bool SelfAdd=false)
Compute knownbits resulting from addition of LHS and RHS.
Definition KnownBits.h:361
static LLVM_ABI KnownBits lshr(const KnownBits &LHS, const KnownBits &RHS, bool ShAmtNonZero=false, bool Exact=false)
Compute known bits for lshr(LHS, RHS).
bool isNonZero() const
Returns true if this value is known to be non-zero.
Definition KnownBits.h:109
static LLVM_ABI KnownBits abdu(const KnownBits &LHS, const KnownBits &RHS)
Compute known bits for abdu(LHS, RHS).
KnownBits extractBits(unsigned NumBits, unsigned BitPosition) const
Return a subset of the known bits from [bitPosition,bitPosition+numBits).
Definition KnownBits.h:239
static LLVM_ABI KnownBits pdep(const KnownBits &Val, const KnownBits &Mask)
Compute known bits for pdep(Val, Mask).
static LLVM_ABI KnownBits avgFloorU(const KnownBits &LHS, const KnownBits &RHS)
Compute knownbits resulting from APIntOps::avgFloorU.
KnownBits sext(unsigned BitWidth) const
Return known bits for a sign extension of the value we're tracking.
Definition KnownBits.h:184
static LLVM_ABI KnownBits computeForSubBorrow(const KnownBits &LHS, KnownBits RHS, const KnownBits &Borrow)
Compute known bits results from subtracting RHS from LHS with 1-bit Borrow.
KnownBits zextOrTrunc(unsigned BitWidth) const
Return known bits for a zero extension or truncation of the value we're tracking.
Definition KnownBits.h:200
APInt getMaxValue() const
Return the maximal unsigned value possible given these KnownBits.
Definition KnownBits.h:146
static LLVM_ABI KnownBits fshr(const KnownBits &LHS, const KnownBits &RHS, const APInt &Amt)
Compute known bits for fshr(LHS, RHS, Amt).
static LLVM_ABI KnownBits abds(KnownBits LHS, KnownBits RHS)
Compute known bits for abds(LHS, RHS).
static LLVM_ABI KnownBits smin(const KnownBits &LHS, const KnownBits &RHS)
Compute known bits for smin(LHS, RHS).
static LLVM_ABI KnownBits mulhs(const KnownBits &LHS, const KnownBits &RHS)
Compute known bits from sign-extended multiply-hi.
static LLVM_ABI KnownBits srem(const KnownBits &LHS, const KnownBits &RHS)
Compute known bits for srem(LHS, RHS).
static LLVM_ABI KnownBits udiv(const KnownBits &LHS, const KnownBits &RHS, bool Exact=false)
Compute known bits for udiv(LHS, RHS).
bool isStrictlyPositive() const
Returns true if this value is known to be positive.
Definition KnownBits.h:112
static LLVM_ABI KnownBits sdiv(const KnownBits &LHS, const KnownBits &RHS, bool Exact=false)
Compute known bits for sdiv(LHS, RHS).
static LLVM_ABI KnownBits avgFloorS(const KnownBits &LHS, const KnownBits &RHS)
Compute knownbits resulting from APIntOps::avgFloorS.
static bool haveNoCommonBitsSet(const KnownBits &LHS, const KnownBits &RHS)
Return true if LHS and RHS have no common bits set.
Definition KnownBits.h:340
bool isNegative() const
Returns true if this value is known to be negative.
Definition KnownBits.h:103
static LLVM_ABI KnownBits computeForAddCarry(const KnownBits &LHS, const KnownBits &RHS, const KnownBits &Carry)
Compute known bits resulting from adding LHS, RHS and a 1-bit Carry.
Definition KnownBits.cpp:54
static KnownBits sub(const KnownBits &LHS, const KnownBits &RHS, bool NSW=false, bool NUW=false)
Compute knownbits resulting from subtraction of LHS and RHS.
Definition KnownBits.h:376
unsigned countMaxLeadingZeros() const
Returns the maximum number of leading zero bits possible.
Definition KnownBits.h:294
static LLVM_ABI KnownBits avgCeilU(const KnownBits &LHS, const KnownBits &RHS)
Compute knownbits resulting from APIntOps::avgCeilU.
static LLVM_ABI KnownBits mul(const KnownBits &LHS, const KnownBits &RHS, bool NoUndefSelfMultiply=false)
Compute known bits resulting from multiplying LHS and RHS.
KnownBits anyext(unsigned BitWidth) const
Return known bits for an "any" extension of the value we're tracking, where we don't know anything ab...
Definition KnownBits.h:171
static LLVM_ABI KnownBits clmul(const KnownBits &LHS, const KnownBits &RHS)
Compute known bits for clmul(LHS, RHS).
LLVM_ABI KnownBits abs(bool IntMinIsPoison=false) const
Compute known bits for the absolute value.
static LLVM_ABI KnownBits shl(const KnownBits &LHS, const KnownBits &RHS, bool NUW=false, bool NSW=false, bool ShAmtNonZero=false)
Compute known bits for shl(LHS, RHS).
static LLVM_ABI KnownBits umin(const KnownBits &LHS, const KnownBits &RHS)
Compute known bits for umin(LHS, RHS).
static LLVM_ABI KnownBits pext(const KnownBits &Val, const KnownBits &Mask)
Compute known bits for pext(Val, Mask).
static LLVM_ABI KnownBits avgCeilS(const KnownBits &LHS, const KnownBits &RHS)
Compute knownbits resulting from APIntOps::avgCeilS.
bool isUnknown() const
KnownFPClass intersectWith(const KnownFPClass &RHS) const
static LLVM_ABI KnownFPClass bitcast(const fltSemantics &FltSemantics, const KnownBits &Bits)
Report known values for a bitcast into a float with provided semantics.
LLVM IR metadata carried by a MachineMemOperand.
This class contains a discriminated union of information about pointers in memory operands,...
LLVM_ABI bool isDereferenceable(unsigned Size, LLVMContext &C, const DataLayout &DL) const
Return true if memory region [V, V+Offset+Size) is known to be dereferenceable.
LLVM_ABI unsigned getAddrSpace() const
Return the LLVM IR address space number that this pointer points into.
PointerUnion< const Value *, const PseudoSourceValue * > V
This is the IR pointer value for the access, or it is null if unknown.
MachinePointerInfo getWithOffset(int64_t O) const
static LLVM_ABI MachinePointerInfo getFixedStack(MachineFunction &MF, int FI, int64_t Offset=0)
Return a MachinePointerInfo record that refers to the specified FrameIndex.
This struct is a compact representation of a valid (power of two) or undefined (0) alignment.
Definition Alignment.h:106
Align valueOrOne() const
For convenience, returns a valid alignment or 1 if undefined.
Definition Alignment.h:130
static MemOp Set(uint64_t Size, bool DstAlignCanChange, Align DstAlign, bool IsZeroMemset, bool IsVolatile)
static MemOp Copy(uint64_t Size, bool DstAlignCanChange, Align DstAlign, Align SrcAlign, bool IsVolatile, bool MemcpyStrSrc=false)
static MemOp Move(uint64_t Size, bool DstAlignCanChange, Align DstAlign, Align SrcAlign, bool IsVolatile)
static StringRef getLibcallImplName(RTLIB::LibcallImpl CallImpl)
Get the libcall routine name for the specified libcall implementation.
These are IR-level optimization flags that may be propagated to SDNodes.
static LLVM_ABI bool isEqual(const KeyTy &Key, const SDNode &N)
The key SelectionDAG uniques SDNodes by.
void AddPointer(const void *P)
SmallVector< SDValue, 0 > OpStorage
Backs Ops when the key is built from a node; empty otherwise.
void AddInteger(T I)
const EVT * VTs
ArrayRef< SDValue > Ops
FoldingSetNodeID Tail
SDNodeKey(unsigned Opcode, SDVTList VTList, ArrayRef< SDValue > Ops)
This represents a list of ValueType's that has been intern'd by a SelectionDAG.
unsigned int NumVTs
Clients of various APIs that cause global effects on the DAG can optionally implement this interface.
virtual void NodeDeleted(SDNode *N, SDNode *E)
The node N that was deleted and, if E is not null, an equivalent node E that replaced it.
virtual void NodeInserted(SDNode *N)
The node N that was inserted.
virtual void NodeUpdated(SDNode *N)
The node N that was updated.
This structure contains all information that is necessary for lowering calls.
CallLoweringInfo & setLibCallee(CallingConv::ID CC, Type *ResultType, SDValue Target, ArgListTy &&ArgsList)
CallLoweringInfo & setDiscardResult(bool Value=true)
CallLoweringInfo & setDebugLoc(const SDLoc &dl)
CallLoweringInfo & setTailCall(bool Value=true)
CallLoweringInfo & setChain(SDValue InChain)