Bug Summary

File:llvm/include/llvm/CodeGen/SelectionDAGNodes.h
Warning:line 1114, column 10
Called C++ object pointer is null

Annotated Source Code

Press '?' to see keyboard shortcuts

clang -cc1 -cc1 -triple x86_64-pc-linux-gnu -analyze -disable-free -disable-llvm-verifier -discard-value-names -main-file-name WebAssemblyISelLowering.cpp -analyzer-store=region -analyzer-opt-analyze-nested-blocks -analyzer-checker=core -analyzer-checker=apiModeling -analyzer-checker=unix -analyzer-checker=deadcode -analyzer-checker=cplusplus -analyzer-checker=security.insecureAPI.UncheckedReturn -analyzer-checker=security.insecureAPI.getpw -analyzer-checker=security.insecureAPI.gets -analyzer-checker=security.insecureAPI.mktemp -analyzer-checker=security.insecureAPI.mkstemp -analyzer-checker=security.insecureAPI.vfork -analyzer-checker=nullability.NullPassedToNonnull -analyzer-checker=nullability.NullReturnedFromNonnull -analyzer-output plist -w -setup-static-analyzer -analyzer-config-compatibility-mode=true -mrelocation-model pic -pic-level 2 -mframe-pointer=none -fmath-errno -fno-rounding-math -mconstructor-aliases -munwind-tables -target-cpu x86-64 -tune-cpu generic -debugger-tuning=gdb -ffunction-sections -fdata-sections -fcoverage-compilation-dir=/build/llvm-toolchain-snapshot-13~++20210726100616+dead50d4427c/build-llvm/lib/Target/WebAssembly -resource-dir /usr/lib/llvm-13/lib/clang/13.0.0 -D _DEBUG -D _GNU_SOURCE -D __STDC_CONSTANT_MACROS -D __STDC_FORMAT_MACROS -D __STDC_LIMIT_MACROS -I /build/llvm-toolchain-snapshot-13~++20210726100616+dead50d4427c/build-llvm/lib/Target/WebAssembly -I /build/llvm-toolchain-snapshot-13~++20210726100616+dead50d4427c/llvm/lib/Target/WebAssembly -I /build/llvm-toolchain-snapshot-13~++20210726100616+dead50d4427c/build-llvm/include -I /build/llvm-toolchain-snapshot-13~++20210726100616+dead50d4427c/llvm/include -D NDEBUG -U NDEBUG -internal-isystem /usr/lib/gcc/x86_64-linux-gnu/10/../../../../include/c++/10 -internal-isystem /usr/lib/gcc/x86_64-linux-gnu/10/../../../../include/x86_64-linux-gnu/c++/10 -internal-isystem /usr/lib/gcc/x86_64-linux-gnu/10/../../../../include/c++/10/backward -internal-isystem /usr/lib/llvm-13/lib/clang/13.0.0/include -internal-isystem /usr/local/include -internal-isystem /usr/lib/gcc/x86_64-linux-gnu/10/../../../../x86_64-linux-gnu/include -internal-externc-isystem /usr/include/x86_64-linux-gnu -internal-externc-isystem /include -internal-externc-isystem /usr/include -O2 -Wno-unused-parameter -Wwrite-strings -Wno-missing-field-initializers -Wno-long-long -Wno-maybe-uninitialized -Wno-class-memaccess -Wno-redundant-move -Wno-pessimizing-move -Wno-noexcept-type -Wno-comment -std=c++14 -fdeprecated-macro -fdebug-compilation-dir=/build/llvm-toolchain-snapshot-13~++20210726100616+dead50d4427c/build-llvm/lib/Target/WebAssembly -fdebug-prefix-map=/build/llvm-toolchain-snapshot-13~++20210726100616+dead50d4427c=. -ferror-limit 19 -fvisibility hidden -fvisibility-inlines-hidden -stack-protector 2 -fgnuc-version=4.2.1 -vectorize-loops -vectorize-slp -analyzer-output=html -analyzer-config stable-report-filename=true -faddrsig -D__GCC_HAVE_DWARF2_CFI_ASM=1 -o /tmp/scan-build-2021-07-26-235520-9401-1 -x c++ /build/llvm-toolchain-snapshot-13~++20210726100616+dead50d4427c/llvm/lib/Target/WebAssembly/WebAssemblyISelLowering.cpp

/build/llvm-toolchain-snapshot-13~++20210726100616+dead50d4427c/llvm/lib/Target/WebAssembly/WebAssemblyISelLowering.cpp

1//=- WebAssemblyISelLowering.cpp - WebAssembly DAG Lowering Implementation -==//
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/// \file
10/// This file implements the WebAssemblyTargetLowering class.
11///
12//===----------------------------------------------------------------------===//
13
14#include "WebAssemblyISelLowering.h"
15#include "MCTargetDesc/WebAssemblyMCTargetDesc.h"
16#include "Utils/WebAssemblyTypeUtilities.h"
17#include "Utils/WebAssemblyUtilities.h"
18#include "WebAssemblyMachineFunctionInfo.h"
19#include "WebAssemblySubtarget.h"
20#include "WebAssemblyTargetMachine.h"
21#include "llvm/CodeGen/CallingConvLower.h"
22#include "llvm/CodeGen/MachineInstrBuilder.h"
23#include "llvm/CodeGen/MachineJumpTableInfo.h"
24#include "llvm/CodeGen/MachineModuleInfo.h"
25#include "llvm/CodeGen/MachineRegisterInfo.h"
26#include "llvm/CodeGen/SelectionDAG.h"
27#include "llvm/CodeGen/SelectionDAGNodes.h"
28#include "llvm/CodeGen/WasmEHFuncInfo.h"
29#include "llvm/IR/DiagnosticInfo.h"
30#include "llvm/IR/DiagnosticPrinter.h"
31#include "llvm/IR/Function.h"
32#include "llvm/IR/Intrinsics.h"
33#include "llvm/IR/IntrinsicsWebAssembly.h"
34#include "llvm/Support/Debug.h"
35#include "llvm/Support/ErrorHandling.h"
36#include "llvm/Support/MathExtras.h"
37#include "llvm/Support/raw_ostream.h"
38#include "llvm/Target/TargetOptions.h"
39using namespace llvm;
40
41#define DEBUG_TYPE"wasm-lower" "wasm-lower"
42
43WebAssemblyTargetLowering::WebAssemblyTargetLowering(
44 const TargetMachine &TM, const WebAssemblySubtarget &STI)
45 : TargetLowering(TM), Subtarget(&STI) {
46 auto MVTPtr = Subtarget->hasAddr64() ? MVT::i64 : MVT::i32;
47
48 // Booleans always contain 0 or 1.
49 setBooleanContents(ZeroOrOneBooleanContent);
50 // Except in SIMD vectors
51 setBooleanVectorContents(ZeroOrNegativeOneBooleanContent);
52 // We don't know the microarchitecture here, so just reduce register pressure.
53 setSchedulingPreference(Sched::RegPressure);
54 // Tell ISel that we have a stack pointer.
55 setStackPointerRegisterToSaveRestore(
56 Subtarget->hasAddr64() ? WebAssembly::SP64 : WebAssembly::SP32);
57 // Set up the register classes.
58 addRegisterClass(MVT::i32, &WebAssembly::I32RegClass);
59 addRegisterClass(MVT::i64, &WebAssembly::I64RegClass);
60 addRegisterClass(MVT::f32, &WebAssembly::F32RegClass);
61 addRegisterClass(MVT::f64, &WebAssembly::F64RegClass);
62 if (Subtarget->hasSIMD128()) {
63 addRegisterClass(MVT::v16i8, &WebAssembly::V128RegClass);
64 addRegisterClass(MVT::v8i16, &WebAssembly::V128RegClass);
65 addRegisterClass(MVT::v4i32, &WebAssembly::V128RegClass);
66 addRegisterClass(MVT::v4f32, &WebAssembly::V128RegClass);
67 addRegisterClass(MVT::v2i64, &WebAssembly::V128RegClass);
68 addRegisterClass(MVT::v2f64, &WebAssembly::V128RegClass);
69 }
70 if (Subtarget->hasReferenceTypes()) {
71 addRegisterClass(MVT::externref, &WebAssembly::EXTERNREFRegClass);
72 addRegisterClass(MVT::funcref, &WebAssembly::FUNCREFRegClass);
73 }
74 // Compute derived properties from the register classes.
75 computeRegisterProperties(Subtarget->getRegisterInfo());
76
77 // Transform loads and stores to pointers in address space 1 to loads and
78 // stores to WebAssembly global variables, outside linear memory.
79 for (auto T : {MVT::i32, MVT::i64, MVT::f32, MVT::f64}) {
80 setOperationAction(ISD::LOAD, T, Custom);
81 setOperationAction(ISD::STORE, T, Custom);
82 }
83 if (Subtarget->hasSIMD128()) {
84 for (auto T : {MVT::v16i8, MVT::v8i16, MVT::v4i32, MVT::v4f32, MVT::v2i64,
85 MVT::v2f64}) {
86 setOperationAction(ISD::LOAD, T, Custom);
87 setOperationAction(ISD::STORE, T, Custom);
88 }
89 }
90 if (Subtarget->hasReferenceTypes()) {
91 for (auto T : {MVT::externref, MVT::funcref}) {
92 setOperationAction(ISD::LOAD, T, Custom);
93 setOperationAction(ISD::STORE, T, Custom);
94 }
95 }
96
97 setOperationAction(ISD::GlobalAddress, MVTPtr, Custom);
98 setOperationAction(ISD::GlobalTLSAddress, MVTPtr, Custom);
99 setOperationAction(ISD::ExternalSymbol, MVTPtr, Custom);
100 setOperationAction(ISD::JumpTable, MVTPtr, Custom);
101 setOperationAction(ISD::BlockAddress, MVTPtr, Custom);
102 setOperationAction(ISD::BRIND, MVT::Other, Custom);
103
104 // Take the default expansion for va_arg, va_copy, and va_end. There is no
105 // default action for va_start, so we do that custom.
106 setOperationAction(ISD::VASTART, MVT::Other, Custom);
107 setOperationAction(ISD::VAARG, MVT::Other, Expand);
108 setOperationAction(ISD::VACOPY, MVT::Other, Expand);
109 setOperationAction(ISD::VAEND, MVT::Other, Expand);
110
111 for (auto T : {MVT::f32, MVT::f64, MVT::v4f32, MVT::v2f64}) {
112 // Don't expand the floating-point types to constant pools.
113 setOperationAction(ISD::ConstantFP, T, Legal);
114 // Expand floating-point comparisons.
115 for (auto CC : {ISD::SETO, ISD::SETUO, ISD::SETUEQ, ISD::SETONE,
116 ISD::SETULT, ISD::SETULE, ISD::SETUGT, ISD::SETUGE})
117 setCondCodeAction(CC, T, Expand);
118 // Expand floating-point library function operators.
119 for (auto Op :
120 {ISD::FSIN, ISD::FCOS, ISD::FSINCOS, ISD::FPOW, ISD::FREM, ISD::FMA})
121 setOperationAction(Op, T, Expand);
122 // Note supported floating-point library function operators that otherwise
123 // default to expand.
124 for (auto Op :
125 {ISD::FCEIL, ISD::FFLOOR, ISD::FTRUNC, ISD::FNEARBYINT, ISD::FRINT})
126 setOperationAction(Op, T, Legal);
127 // Support minimum and maximum, which otherwise default to expand.
128 setOperationAction(ISD::FMINIMUM, T, Legal);
129 setOperationAction(ISD::FMAXIMUM, T, Legal);
130 // WebAssembly currently has no builtin f16 support.
131 setOperationAction(ISD::FP16_TO_FP, T, Expand);
132 setOperationAction(ISD::FP_TO_FP16, T, Expand);
133 setLoadExtAction(ISD::EXTLOAD, T, MVT::f16, Expand);
134 setTruncStoreAction(T, MVT::f16, Expand);
135 }
136
137 // Expand unavailable integer operations.
138 for (auto Op :
139 {ISD::BSWAP, ISD::SMUL_LOHI, ISD::UMUL_LOHI, ISD::MULHS, ISD::MULHU,
140 ISD::SDIVREM, ISD::UDIVREM, ISD::SHL_PARTS, ISD::SRA_PARTS,
141 ISD::SRL_PARTS, ISD::ADDC, ISD::ADDE, ISD::SUBC, ISD::SUBE}) {
142 for (auto T : {MVT::i32, MVT::i64})
143 setOperationAction(Op, T, Expand);
144 if (Subtarget->hasSIMD128())
145 for (auto T : {MVT::v16i8, MVT::v8i16, MVT::v4i32, MVT::v2i64})
146 setOperationAction(Op, T, Expand);
147 }
148
149 if (Subtarget->hasNontrappingFPToInt())
150 for (auto Op : {ISD::FP_TO_SINT_SAT, ISD::FP_TO_UINT_SAT})
151 for (auto T : {MVT::i32, MVT::i64})
152 setOperationAction(Op, T, Custom);
153
154 // SIMD-specific configuration
155 if (Subtarget->hasSIMD128()) {
156 // Hoist bitcasts out of shuffles
157 setTargetDAGCombine(ISD::VECTOR_SHUFFLE);
158
159 // Combine extends of extract_subvectors into widening ops
160 setTargetDAGCombine(ISD::SIGN_EXTEND);
161 setTargetDAGCombine(ISD::ZERO_EXTEND);
162
163 // Combine int_to_fp or fp_extend of extract_vectors and vice versa into
164 // conversions ops
165 setTargetDAGCombine(ISD::SINT_TO_FP);
166 setTargetDAGCombine(ISD::UINT_TO_FP);
167 setTargetDAGCombine(ISD::FP_EXTEND);
168 setTargetDAGCombine(ISD::EXTRACT_SUBVECTOR);
169
170 // Combine fp_to_{s,u}int_sat or fp_round of concat_vectors or vice versa
171 // into conversion ops
172 setTargetDAGCombine(ISD::FP_TO_SINT_SAT);
173 setTargetDAGCombine(ISD::FP_TO_UINT_SAT);
174 setTargetDAGCombine(ISD::FP_ROUND);
175 setTargetDAGCombine(ISD::CONCAT_VECTORS);
176
177 // Support saturating add for i8x16 and i16x8
178 for (auto Op : {ISD::SADDSAT, ISD::UADDSAT})
179 for (auto T : {MVT::v16i8, MVT::v8i16})
180 setOperationAction(Op, T, Legal);
181
182 // Support integer abs
183 for (auto T : {MVT::v16i8, MVT::v8i16, MVT::v4i32, MVT::v2i64})
184 setOperationAction(ISD::ABS, T, Legal);
185
186 // Custom lower BUILD_VECTORs to minimize number of replace_lanes
187 for (auto T : {MVT::v16i8, MVT::v8i16, MVT::v4i32, MVT::v4f32, MVT::v2i64,
188 MVT::v2f64})
189 setOperationAction(ISD::BUILD_VECTOR, T, Custom);
190
191 // We have custom shuffle lowering to expose the shuffle mask
192 for (auto T : {MVT::v16i8, MVT::v8i16, MVT::v4i32, MVT::v4f32, MVT::v2i64,
193 MVT::v2f64})
194 setOperationAction(ISD::VECTOR_SHUFFLE, T, Custom);
195
196 // Custom lowering since wasm shifts must have a scalar shift amount
197 for (auto Op : {ISD::SHL, ISD::SRA, ISD::SRL})
198 for (auto T : {MVT::v16i8, MVT::v8i16, MVT::v4i32, MVT::v2i64})
199 setOperationAction(Op, T, Custom);
200
201 // Custom lower lane accesses to expand out variable indices
202 for (auto Op : {ISD::EXTRACT_VECTOR_ELT, ISD::INSERT_VECTOR_ELT})
203 for (auto T : {MVT::v16i8, MVT::v8i16, MVT::v4i32, MVT::v4f32, MVT::v2i64,
204 MVT::v2f64})
205 setOperationAction(Op, T, Custom);
206
207 // There is no i8x16.mul instruction
208 setOperationAction(ISD::MUL, MVT::v16i8, Expand);
209
210 // There is no vector conditional select instruction
211 for (auto T : {MVT::v16i8, MVT::v8i16, MVT::v4i32, MVT::v4f32, MVT::v2i64,
212 MVT::v2f64})
213 setOperationAction(ISD::SELECT_CC, T, Expand);
214
215 // Expand integer operations supported for scalars but not SIMD
216 for (auto Op : {ISD::CTLZ, ISD::CTTZ, ISD::CTPOP, ISD::SDIV, ISD::UDIV,
217 ISD::SREM, ISD::UREM, ISD::ROTL, ISD::ROTR})
218 for (auto T : {MVT::v16i8, MVT::v8i16, MVT::v4i32, MVT::v2i64})
219 setOperationAction(Op, T, Expand);
220
221 // But we do have integer min and max operations
222 for (auto Op : {ISD::SMIN, ISD::SMAX, ISD::UMIN, ISD::UMAX})
223 for (auto T : {MVT::v16i8, MVT::v8i16, MVT::v4i32})
224 setOperationAction(Op, T, Legal);
225
226 // And we have popcnt for i8x16
227 setOperationAction(ISD::CTPOP, MVT::v16i8, Legal);
228
229 // Expand float operations supported for scalars but not SIMD
230 for (auto Op : {ISD::FCOPYSIGN, ISD::FLOG, ISD::FLOG2, ISD::FLOG10,
231 ISD::FEXP, ISD::FEXP2, ISD::FRINT})
232 for (auto T : {MVT::v4f32, MVT::v2f64})
233 setOperationAction(Op, T, Expand);
234
235 // Unsigned comparison operations are unavailable for i64x2 vectors.
236 for (auto CC : {ISD::SETUGT, ISD::SETUGE, ISD::SETULT, ISD::SETULE})
237 setCondCodeAction(CC, MVT::v2i64, Custom);
238
239 // 64x2 conversions are not in the spec
240 for (auto Op :
241 {ISD::SINT_TO_FP, ISD::UINT_TO_FP, ISD::FP_TO_SINT, ISD::FP_TO_UINT})
242 for (auto T : {MVT::v2i64, MVT::v2f64})
243 setOperationAction(Op, T, Expand);
244
245 // But saturating fp_to_int converstions are
246 for (auto Op : {ISD::FP_TO_SINT_SAT, ISD::FP_TO_UINT_SAT})
247 setOperationAction(Op, MVT::v4i32, Custom);
248 }
249
250 // As a special case, these operators use the type to mean the type to
251 // sign-extend from.
252 setOperationAction(ISD::SIGN_EXTEND_INREG, MVT::i1, Expand);
253 if (!Subtarget->hasSignExt()) {
254 // Sign extends are legal only when extending a vector extract
255 auto Action = Subtarget->hasSIMD128() ? Custom : Expand;
256 for (auto T : {MVT::i8, MVT::i16, MVT::i32})
257 setOperationAction(ISD::SIGN_EXTEND_INREG, T, Action);
258 }
259 for (auto T : MVT::integer_fixedlen_vector_valuetypes())
260 setOperationAction(ISD::SIGN_EXTEND_INREG, T, Expand);
261
262 // Dynamic stack allocation: use the default expansion.
263 setOperationAction(ISD::STACKSAVE, MVT::Other, Expand);
264 setOperationAction(ISD::STACKRESTORE, MVT::Other, Expand);
265 setOperationAction(ISD::DYNAMIC_STACKALLOC, MVTPtr, Expand);
266
267 setOperationAction(ISD::FrameIndex, MVT::i32, Custom);
268 setOperationAction(ISD::FrameIndex, MVT::i64, Custom);
269 setOperationAction(ISD::CopyToReg, MVT::Other, Custom);
270
271 // Expand these forms; we pattern-match the forms that we can handle in isel.
272 for (auto T : {MVT::i32, MVT::i64, MVT::f32, MVT::f64})
273 for (auto Op : {ISD::BR_CC, ISD::SELECT_CC})
274 setOperationAction(Op, T, Expand);
275
276 // We have custom switch handling.
277 setOperationAction(ISD::BR_JT, MVT::Other, Custom);
278
279 // WebAssembly doesn't have:
280 // - Floating-point extending loads.
281 // - Floating-point truncating stores.
282 // - i1 extending loads.
283 // - truncating SIMD stores and most extending loads
284 setLoadExtAction(ISD::EXTLOAD, MVT::f64, MVT::f32, Expand);
285 setTruncStoreAction(MVT::f64, MVT::f32, Expand);
286 for (auto T : MVT::integer_valuetypes())
287 for (auto Ext : {ISD::EXTLOAD, ISD::ZEXTLOAD, ISD::SEXTLOAD})
288 setLoadExtAction(Ext, T, MVT::i1, Promote);
289 if (Subtarget->hasSIMD128()) {
290 for (auto T : {MVT::v16i8, MVT::v8i16, MVT::v4i32, MVT::v2i64, MVT::v4f32,
291 MVT::v2f64}) {
292 for (auto MemT : MVT::fixedlen_vector_valuetypes()) {
293 if (MVT(T) != MemT) {
294 setTruncStoreAction(T, MemT, Expand);
295 for (auto Ext : {ISD::EXTLOAD, ISD::ZEXTLOAD, ISD::SEXTLOAD})
296 setLoadExtAction(Ext, T, MemT, Expand);
297 }
298 }
299 }
300 // But some vector extending loads are legal
301 for (auto Ext : {ISD::EXTLOAD, ISD::SEXTLOAD, ISD::ZEXTLOAD}) {
302 setLoadExtAction(Ext, MVT::v8i16, MVT::v8i8, Legal);
303 setLoadExtAction(Ext, MVT::v4i32, MVT::v4i16, Legal);
304 setLoadExtAction(Ext, MVT::v2i64, MVT::v2i32, Legal);
305 }
306 // And some truncating stores are legal as well
307 setTruncStoreAction(MVT::v8i16, MVT::v8i8, Legal);
308 setTruncStoreAction(MVT::v4i32, MVT::v4i16, Legal);
309 }
310
311 // Don't do anything clever with build_pairs
312 setOperationAction(ISD::BUILD_PAIR, MVT::i64, Expand);
313
314 // Trap lowers to wasm unreachable
315 setOperationAction(ISD::TRAP, MVT::Other, Legal);
316 setOperationAction(ISD::DEBUGTRAP, MVT::Other, Legal);
317
318 // Exception handling intrinsics
319 setOperationAction(ISD::INTRINSIC_WO_CHAIN, MVT::Other, Custom);
320 setOperationAction(ISD::INTRINSIC_W_CHAIN, MVT::Other, Custom);
321 setOperationAction(ISD::INTRINSIC_VOID, MVT::Other, Custom);
322
323 setMaxAtomicSizeInBitsSupported(64);
324
325 // Override the __gnu_f2h_ieee/__gnu_h2f_ieee names so that the f32 name is
326 // consistent with the f64 and f128 names.
327 setLibcallName(RTLIB::FPEXT_F16_F32, "__extendhfsf2");
328 setLibcallName(RTLIB::FPROUND_F32_F16, "__truncsfhf2");
329
330 // Define the emscripten name for return address helper.
331 // TODO: when implementing other Wasm backends, make this generic or only do
332 // this on emscripten depending on what they end up doing.
333 setLibcallName(RTLIB::RETURN_ADDRESS, "emscripten_return_address");
334
335 // Always convert switches to br_tables unless there is only one case, which
336 // is equivalent to a simple branch. This reduces code size for wasm, and we
337 // defer possible jump table optimizations to the VM.
338 setMinimumJumpTableEntries(2);
339}
340
341TargetLowering::AtomicExpansionKind
342WebAssemblyTargetLowering::shouldExpandAtomicRMWInIR(AtomicRMWInst *AI) const {
343 // We have wasm instructions for these
344 switch (AI->getOperation()) {
345 case AtomicRMWInst::Add:
346 case AtomicRMWInst::Sub:
347 case AtomicRMWInst::And:
348 case AtomicRMWInst::Or:
349 case AtomicRMWInst::Xor:
350 case AtomicRMWInst::Xchg:
351 return AtomicExpansionKind::None;
352 default:
353 break;
354 }
355 return AtomicExpansionKind::CmpXChg;
356}
357
358bool WebAssemblyTargetLowering::shouldScalarizeBinop(SDValue VecOp) const {
359 // Implementation copied from X86TargetLowering.
360 unsigned Opc = VecOp.getOpcode();
361
362 // Assume target opcodes can't be scalarized.
363 // TODO - do we have any exceptions?
364 if (Opc >= ISD::BUILTIN_OP_END)
365 return false;
366
367 // If the vector op is not supported, try to convert to scalar.
368 EVT VecVT = VecOp.getValueType();
369 if (!isOperationLegalOrCustomOrPromote(Opc, VecVT))
370 return true;
371
372 // If the vector op is supported, but the scalar op is not, the transform may
373 // not be worthwhile.
374 EVT ScalarVT = VecVT.getScalarType();
375 return isOperationLegalOrCustomOrPromote(Opc, ScalarVT);
376}
377
378FastISel *WebAssemblyTargetLowering::createFastISel(
379 FunctionLoweringInfo &FuncInfo, const TargetLibraryInfo *LibInfo) const {
380 return WebAssembly::createFastISel(FuncInfo, LibInfo);
381}
382
383MVT WebAssemblyTargetLowering::getScalarShiftAmountTy(const DataLayout & /*DL*/,
384 EVT VT) const {
385 unsigned BitWidth = NextPowerOf2(VT.getSizeInBits() - 1);
386 if (BitWidth > 1 && BitWidth < 8)
387 BitWidth = 8;
388
389 if (BitWidth > 64) {
390 // The shift will be lowered to a libcall, and compiler-rt libcalls expect
391 // the count to be an i32.
392 BitWidth = 32;
393 assert(BitWidth >= Log2_32_Ceil(VT.getSizeInBits()) &&(static_cast <bool> (BitWidth >= Log2_32_Ceil(VT.getSizeInBits
()) && "32-bit shift counts ought to be enough for anyone"
) ? void (0) : __assert_fail ("BitWidth >= Log2_32_Ceil(VT.getSizeInBits()) && \"32-bit shift counts ought to be enough for anyone\""
, "/build/llvm-toolchain-snapshot-13~++20210726100616+dead50d4427c/llvm/lib/Target/WebAssembly/WebAssemblyISelLowering.cpp"
, 394, __extension__ __PRETTY_FUNCTION__))
394 "32-bit shift counts ought to be enough for anyone")(static_cast <bool> (BitWidth >= Log2_32_Ceil(VT.getSizeInBits
()) && "32-bit shift counts ought to be enough for anyone"
) ? void (0) : __assert_fail ("BitWidth >= Log2_32_Ceil(VT.getSizeInBits()) && \"32-bit shift counts ought to be enough for anyone\""
, "/build/llvm-toolchain-snapshot-13~++20210726100616+dead50d4427c/llvm/lib/Target/WebAssembly/WebAssemblyISelLowering.cpp"
, 394, __extension__ __PRETTY_FUNCTION__))
;
395 }
396
397 MVT Result = MVT::getIntegerVT(BitWidth);
398 assert(Result != MVT::INVALID_SIMPLE_VALUE_TYPE &&(static_cast <bool> (Result != MVT::INVALID_SIMPLE_VALUE_TYPE
&& "Unable to represent scalar shift amount type") ?
void (0) : __assert_fail ("Result != MVT::INVALID_SIMPLE_VALUE_TYPE && \"Unable to represent scalar shift amount type\""
, "/build/llvm-toolchain-snapshot-13~++20210726100616+dead50d4427c/llvm/lib/Target/WebAssembly/WebAssemblyISelLowering.cpp"
, 399, __extension__ __PRETTY_FUNCTION__))
399 "Unable to represent scalar shift amount type")(static_cast <bool> (Result != MVT::INVALID_SIMPLE_VALUE_TYPE
&& "Unable to represent scalar shift amount type") ?
void (0) : __assert_fail ("Result != MVT::INVALID_SIMPLE_VALUE_TYPE && \"Unable to represent scalar shift amount type\""
, "/build/llvm-toolchain-snapshot-13~++20210726100616+dead50d4427c/llvm/lib/Target/WebAssembly/WebAssemblyISelLowering.cpp"
, 399, __extension__ __PRETTY_FUNCTION__))
;
400 return Result;
401}
402
403// Lower an fp-to-int conversion operator from the LLVM opcode, which has an
404// undefined result on invalid/overflow, to the WebAssembly opcode, which
405// traps on invalid/overflow.
406static MachineBasicBlock *LowerFPToInt(MachineInstr &MI, DebugLoc DL,
407 MachineBasicBlock *BB,
408 const TargetInstrInfo &TII,
409 bool IsUnsigned, bool Int64,
410 bool Float64, unsigned LoweredOpcode) {
411 MachineRegisterInfo &MRI = BB->getParent()->getRegInfo();
412
413 Register OutReg = MI.getOperand(0).getReg();
414 Register InReg = MI.getOperand(1).getReg();
415
416 unsigned Abs = Float64 ? WebAssembly::ABS_F64 : WebAssembly::ABS_F32;
417 unsigned FConst = Float64 ? WebAssembly::CONST_F64 : WebAssembly::CONST_F32;
418 unsigned LT = Float64 ? WebAssembly::LT_F64 : WebAssembly::LT_F32;
419 unsigned GE = Float64 ? WebAssembly::GE_F64 : WebAssembly::GE_F32;
420 unsigned IConst = Int64 ? WebAssembly::CONST_I64 : WebAssembly::CONST_I32;
421 unsigned Eqz = WebAssembly::EQZ_I32;
422 unsigned And = WebAssembly::AND_I32;
423 int64_t Limit = Int64 ? INT64_MIN(-9223372036854775807L -1) : INT32_MIN(-2147483647-1);
424 int64_t Substitute = IsUnsigned ? 0 : Limit;
425 double CmpVal = IsUnsigned ? -(double)Limit * 2.0 : -(double)Limit;
426 auto &Context = BB->getParent()->getFunction().getContext();
427 Type *Ty = Float64 ? Type::getDoubleTy(Context) : Type::getFloatTy(Context);
428
429 const BasicBlock *LLVMBB = BB->getBasicBlock();
430 MachineFunction *F = BB->getParent();
431 MachineBasicBlock *TrueMBB = F->CreateMachineBasicBlock(LLVMBB);
432 MachineBasicBlock *FalseMBB = F->CreateMachineBasicBlock(LLVMBB);
433 MachineBasicBlock *DoneMBB = F->CreateMachineBasicBlock(LLVMBB);
434
435 MachineFunction::iterator It = ++BB->getIterator();
436 F->insert(It, FalseMBB);
437 F->insert(It, TrueMBB);
438 F->insert(It, DoneMBB);
439
440 // Transfer the remainder of BB and its successor edges to DoneMBB.
441 DoneMBB->splice(DoneMBB->begin(), BB, std::next(MI.getIterator()), BB->end());
442 DoneMBB->transferSuccessorsAndUpdatePHIs(BB);
443
444 BB->addSuccessor(TrueMBB);
445 BB->addSuccessor(FalseMBB);
446 TrueMBB->addSuccessor(DoneMBB);
447 FalseMBB->addSuccessor(DoneMBB);
448
449 unsigned Tmp0, Tmp1, CmpReg, EqzReg, FalseReg, TrueReg;
450 Tmp0 = MRI.createVirtualRegister(MRI.getRegClass(InReg));
451 Tmp1 = MRI.createVirtualRegister(MRI.getRegClass(InReg));
452 CmpReg = MRI.createVirtualRegister(&WebAssembly::I32RegClass);
453 EqzReg = MRI.createVirtualRegister(&WebAssembly::I32RegClass);
454 FalseReg = MRI.createVirtualRegister(MRI.getRegClass(OutReg));
455 TrueReg = MRI.createVirtualRegister(MRI.getRegClass(OutReg));
456
457 MI.eraseFromParent();
458 // For signed numbers, we can do a single comparison to determine whether
459 // fabs(x) is within range.
460 if (IsUnsigned) {
461 Tmp0 = InReg;
462 } else {
463 BuildMI(BB, DL, TII.get(Abs), Tmp0).addReg(InReg);
464 }
465 BuildMI(BB, DL, TII.get(FConst), Tmp1)
466 .addFPImm(cast<ConstantFP>(ConstantFP::get(Ty, CmpVal)));
467 BuildMI(BB, DL, TII.get(LT), CmpReg).addReg(Tmp0).addReg(Tmp1);
468
469 // For unsigned numbers, we have to do a separate comparison with zero.
470 if (IsUnsigned) {
471 Tmp1 = MRI.createVirtualRegister(MRI.getRegClass(InReg));
472 Register SecondCmpReg =
473 MRI.createVirtualRegister(&WebAssembly::I32RegClass);
474 Register AndReg = MRI.createVirtualRegister(&WebAssembly::I32RegClass);
475 BuildMI(BB, DL, TII.get(FConst), Tmp1)
476 .addFPImm(cast<ConstantFP>(ConstantFP::get(Ty, 0.0)));
477 BuildMI(BB, DL, TII.get(GE), SecondCmpReg).addReg(Tmp0).addReg(Tmp1);
478 BuildMI(BB, DL, TII.get(And), AndReg).addReg(CmpReg).addReg(SecondCmpReg);
479 CmpReg = AndReg;
480 }
481
482 BuildMI(BB, DL, TII.get(Eqz), EqzReg).addReg(CmpReg);
483
484 // Create the CFG diamond to select between doing the conversion or using
485 // the substitute value.
486 BuildMI(BB, DL, TII.get(WebAssembly::BR_IF)).addMBB(TrueMBB).addReg(EqzReg);
487 BuildMI(FalseMBB, DL, TII.get(LoweredOpcode), FalseReg).addReg(InReg);
488 BuildMI(FalseMBB, DL, TII.get(WebAssembly::BR)).addMBB(DoneMBB);
489 BuildMI(TrueMBB, DL, TII.get(IConst), TrueReg).addImm(Substitute);
490 BuildMI(*DoneMBB, DoneMBB->begin(), DL, TII.get(TargetOpcode::PHI), OutReg)
491 .addReg(FalseReg)
492 .addMBB(FalseMBB)
493 .addReg(TrueReg)
494 .addMBB(TrueMBB);
495
496 return DoneMBB;
497}
498
499static MachineBasicBlock *
500LowerCallResults(MachineInstr &CallResults, DebugLoc DL, MachineBasicBlock *BB,
501 const WebAssemblySubtarget *Subtarget,
502 const TargetInstrInfo &TII) {
503 MachineInstr &CallParams = *CallResults.getPrevNode();
504 assert(CallParams.getOpcode() == WebAssembly::CALL_PARAMS)(static_cast <bool> (CallParams.getOpcode() == WebAssembly
::CALL_PARAMS) ? void (0) : __assert_fail ("CallParams.getOpcode() == WebAssembly::CALL_PARAMS"
, "/build/llvm-toolchain-snapshot-13~++20210726100616+dead50d4427c/llvm/lib/Target/WebAssembly/WebAssemblyISelLowering.cpp"
, 504, __extension__ __PRETTY_FUNCTION__))
;
505 assert(CallResults.getOpcode() == WebAssembly::CALL_RESULTS ||(static_cast <bool> (CallResults.getOpcode() == WebAssembly
::CALL_RESULTS || CallResults.getOpcode() == WebAssembly::RET_CALL_RESULTS
) ? void (0) : __assert_fail ("CallResults.getOpcode() == WebAssembly::CALL_RESULTS || CallResults.getOpcode() == WebAssembly::RET_CALL_RESULTS"
, "/build/llvm-toolchain-snapshot-13~++20210726100616+dead50d4427c/llvm/lib/Target/WebAssembly/WebAssemblyISelLowering.cpp"
, 506, __extension__ __PRETTY_FUNCTION__))
506 CallResults.getOpcode() == WebAssembly::RET_CALL_RESULTS)(static_cast <bool> (CallResults.getOpcode() == WebAssembly
::CALL_RESULTS || CallResults.getOpcode() == WebAssembly::RET_CALL_RESULTS
) ? void (0) : __assert_fail ("CallResults.getOpcode() == WebAssembly::CALL_RESULTS || CallResults.getOpcode() == WebAssembly::RET_CALL_RESULTS"
, "/build/llvm-toolchain-snapshot-13~++20210726100616+dead50d4427c/llvm/lib/Target/WebAssembly/WebAssemblyISelLowering.cpp"
, 506, __extension__ __PRETTY_FUNCTION__))
;
507
508 bool IsIndirect = CallParams.getOperand(0).isReg();
509 bool IsRetCall = CallResults.getOpcode() == WebAssembly::RET_CALL_RESULTS;
510
511 bool IsFuncrefCall = false;
512 if (IsIndirect) {
513 Register Reg = CallParams.getOperand(0).getReg();
514 const MachineFunction *MF = BB->getParent();
515 const MachineRegisterInfo &MRI = MF->getRegInfo();
516 const TargetRegisterClass *TRC = MRI.getRegClass(Reg);
517 IsFuncrefCall = (TRC == &WebAssembly::FUNCREFRegClass);
518 assert(!IsFuncrefCall || Subtarget->hasReferenceTypes())(static_cast <bool> (!IsFuncrefCall || Subtarget->hasReferenceTypes
()) ? void (0) : __assert_fail ("!IsFuncrefCall || Subtarget->hasReferenceTypes()"
, "/build/llvm-toolchain-snapshot-13~++20210726100616+dead50d4427c/llvm/lib/Target/WebAssembly/WebAssemblyISelLowering.cpp"
, 518, __extension__ __PRETTY_FUNCTION__))
;
519 }
520
521 unsigned CallOp;
522 if (IsIndirect && IsRetCall) {
523 CallOp = WebAssembly::RET_CALL_INDIRECT;
524 } else if (IsIndirect) {
525 CallOp = WebAssembly::CALL_INDIRECT;
526 } else if (IsRetCall) {
527 CallOp = WebAssembly::RET_CALL;
528 } else {
529 CallOp = WebAssembly::CALL;
530 }
531
532 MachineFunction &MF = *BB->getParent();
533 const MCInstrDesc &MCID = TII.get(CallOp);
534 MachineInstrBuilder MIB(MF, MF.CreateMachineInstr(MCID, DL));
535
536 // See if we must truncate the function pointer.
537 // CALL_INDIRECT takes an i32, but in wasm64 we represent function pointers
538 // as 64-bit for uniformity with other pointer types.
539 // See also: WebAssemblyFastISel::selectCall
540 if (IsIndirect && MF.getSubtarget<WebAssemblySubtarget>().hasAddr64()) {
541 Register Reg32 =
542 MF.getRegInfo().createVirtualRegister(&WebAssembly::I32RegClass);
543 auto &FnPtr = CallParams.getOperand(0);
544 BuildMI(*BB, CallResults.getIterator(), DL,
545 TII.get(WebAssembly::I32_WRAP_I64), Reg32)
546 .addReg(FnPtr.getReg());
547 FnPtr.setReg(Reg32);
548 }
549
550 // Move the function pointer to the end of the arguments for indirect calls
551 if (IsIndirect) {
552 auto FnPtr = CallParams.getOperand(0);
553 CallParams.RemoveOperand(0);
554 CallParams.addOperand(FnPtr);
555 }
556
557 for (auto Def : CallResults.defs())
558 MIB.add(Def);
559
560 if (IsIndirect) {
561 // Placeholder for the type index.
562 MIB.addImm(0);
563 // The table into which this call_indirect indexes.
564 MCSymbolWasm *Table = IsFuncrefCall
565 ? WebAssembly::getOrCreateFuncrefCallTableSymbol(
566 MF.getContext(), Subtarget)
567 : WebAssembly::getOrCreateFunctionTableSymbol(
568 MF.getContext(), Subtarget);
569 if (Subtarget->hasReferenceTypes()) {
570 MIB.addSym(Table);
571 } else {
572 // For the MVP there is at most one table whose number is 0, but we can't
573 // write a table symbol or issue relocations. Instead we just ensure the
574 // table is live and write a zero.
575 Table->setNoStrip();
576 MIB.addImm(0);
577 }
578 }
579
580 for (auto Use : CallParams.uses())
581 MIB.add(Use);
582
583 BB->insert(CallResults.getIterator(), MIB);
584 CallParams.eraseFromParent();
585 CallResults.eraseFromParent();
586
587 // If this is a funcref call, to avoid hidden GC roots, we need to clear the
588 // table slot with ref.null upon call_indirect return.
589 //
590 // This generates the following code, which comes right after a call_indirect
591 // of a funcref:
592 //
593 // i32.const 0
594 // ref.null func
595 // table.set __funcref_call_table
596 if (IsIndirect && IsFuncrefCall) {
597 MCSymbolWasm *Table = WebAssembly::getOrCreateFuncrefCallTableSymbol(
598 MF.getContext(), Subtarget);
599 Register RegZero =
600 MF.getRegInfo().createVirtualRegister(&WebAssembly::I32RegClass);
601 MachineInstr *Const0 =
602 BuildMI(MF, DL, TII.get(WebAssembly::CONST_I32), RegZero).addImm(0);
603 BB->insertAfter(MIB.getInstr()->getIterator(), Const0);
604
605 Register RegFuncref =
606 MF.getRegInfo().createVirtualRegister(&WebAssembly::FUNCREFRegClass);
607 MachineInstr *RefNull =
608 BuildMI(MF, DL, TII.get(WebAssembly::REF_NULL_FUNCREF), RegFuncref)
609 .addImm(static_cast<int32_t>(WebAssembly::HeapType::Funcref));
610 BB->insertAfter(Const0->getIterator(), RefNull);
611
612 MachineInstr *TableSet =
613 BuildMI(MF, DL, TII.get(WebAssembly::TABLE_SET_FUNCREF))
614 .addSym(Table)
615 .addReg(RegZero)
616 .addReg(RegFuncref);
617 BB->insertAfter(RefNull->getIterator(), TableSet);
618 }
619
620 return BB;
621}
622
623MachineBasicBlock *WebAssemblyTargetLowering::EmitInstrWithCustomInserter(
624 MachineInstr &MI, MachineBasicBlock *BB) const {
625 const TargetInstrInfo &TII = *Subtarget->getInstrInfo();
626 DebugLoc DL = MI.getDebugLoc();
627
628 switch (MI.getOpcode()) {
629 default:
630 llvm_unreachable("Unexpected instr type to insert")::llvm::llvm_unreachable_internal("Unexpected instr type to insert"
, "/build/llvm-toolchain-snapshot-13~++20210726100616+dead50d4427c/llvm/lib/Target/WebAssembly/WebAssemblyISelLowering.cpp"
, 630)
;
631 case WebAssembly::FP_TO_SINT_I32_F32:
632 return LowerFPToInt(MI, DL, BB, TII, false, false, false,
633 WebAssembly::I32_TRUNC_S_F32);
634 case WebAssembly::FP_TO_UINT_I32_F32:
635 return LowerFPToInt(MI, DL, BB, TII, true, false, false,
636 WebAssembly::I32_TRUNC_U_F32);
637 case WebAssembly::FP_TO_SINT_I64_F32:
638 return LowerFPToInt(MI, DL, BB, TII, false, true, false,
639 WebAssembly::I64_TRUNC_S_F32);
640 case WebAssembly::FP_TO_UINT_I64_F32:
641 return LowerFPToInt(MI, DL, BB, TII, true, true, false,
642 WebAssembly::I64_TRUNC_U_F32);
643 case WebAssembly::FP_TO_SINT_I32_F64:
644 return LowerFPToInt(MI, DL, BB, TII, false, false, true,
645 WebAssembly::I32_TRUNC_S_F64);
646 case WebAssembly::FP_TO_UINT_I32_F64:
647 return LowerFPToInt(MI, DL, BB, TII, true, false, true,
648 WebAssembly::I32_TRUNC_U_F64);
649 case WebAssembly::FP_TO_SINT_I64_F64:
650 return LowerFPToInt(MI, DL, BB, TII, false, true, true,
651 WebAssembly::I64_TRUNC_S_F64);
652 case WebAssembly::FP_TO_UINT_I64_F64:
653 return LowerFPToInt(MI, DL, BB, TII, true, true, true,
654 WebAssembly::I64_TRUNC_U_F64);
655 case WebAssembly::CALL_RESULTS:
656 case WebAssembly::RET_CALL_RESULTS:
657 return LowerCallResults(MI, DL, BB, Subtarget, TII);
658 }
659}
660
661const char *
662WebAssemblyTargetLowering::getTargetNodeName(unsigned Opcode) const {
663 switch (static_cast<WebAssemblyISD::NodeType>(Opcode)) {
664 case WebAssemblyISD::FIRST_NUMBER:
665 case WebAssemblyISD::FIRST_MEM_OPCODE:
666 break;
667#define HANDLE_NODETYPE(NODE) \
668 case WebAssemblyISD::NODE: \
669 return "WebAssemblyISD::" #NODE;
670#define HANDLE_MEM_NODETYPE(NODE) HANDLE_NODETYPE(NODE)
671#include "WebAssemblyISD.def"
672#undef HANDLE_MEM_NODETYPE
673#undef HANDLE_NODETYPE
674 }
675 return nullptr;
676}
677
678std::pair<unsigned, const TargetRegisterClass *>
679WebAssemblyTargetLowering::getRegForInlineAsmConstraint(
680 const TargetRegisterInfo *TRI, StringRef Constraint, MVT VT) const {
681 // First, see if this is a constraint that directly corresponds to a
682 // WebAssembly register class.
683 if (Constraint.size() == 1) {
684 switch (Constraint[0]) {
685 case 'r':
686 assert(VT != MVT::iPTR && "Pointer MVT not expected here")(static_cast <bool> (VT != MVT::iPTR && "Pointer MVT not expected here"
) ? void (0) : __assert_fail ("VT != MVT::iPTR && \"Pointer MVT not expected here\""
, "/build/llvm-toolchain-snapshot-13~++20210726100616+dead50d4427c/llvm/lib/Target/WebAssembly/WebAssemblyISelLowering.cpp"
, 686, __extension__ __PRETTY_FUNCTION__))
;
687 if (Subtarget->hasSIMD128() && VT.isVector()) {
688 if (VT.getSizeInBits() == 128)
689 return std::make_pair(0U, &WebAssembly::V128RegClass);
690 }
691 if (VT.isInteger() && !VT.isVector()) {
692 if (VT.getSizeInBits() <= 32)
693 return std::make_pair(0U, &WebAssembly::I32RegClass);
694 if (VT.getSizeInBits() <= 64)
695 return std::make_pair(0U, &WebAssembly::I64RegClass);
696 }
697 if (VT.isFloatingPoint() && !VT.isVector()) {
698 switch (VT.getSizeInBits()) {
699 case 32:
700 return std::make_pair(0U, &WebAssembly::F32RegClass);
701 case 64:
702 return std::make_pair(0U, &WebAssembly::F64RegClass);
703 default:
704 break;
705 }
706 }
707 break;
708 default:
709 break;
710 }
711 }
712
713 return TargetLowering::getRegForInlineAsmConstraint(TRI, Constraint, VT);
714}
715
716bool WebAssemblyTargetLowering::isCheapToSpeculateCttz() const {
717 // Assume ctz is a relatively cheap operation.
718 return true;
719}
720
721bool WebAssemblyTargetLowering::isCheapToSpeculateCtlz() const {
722 // Assume clz is a relatively cheap operation.
723 return true;
724}
725
726bool WebAssemblyTargetLowering::isLegalAddressingMode(const DataLayout &DL,
727 const AddrMode &AM,
728 Type *Ty, unsigned AS,
729 Instruction *I) const {
730 // WebAssembly offsets are added as unsigned without wrapping. The
731 // isLegalAddressingMode gives us no way to determine if wrapping could be
732 // happening, so we approximate this by accepting only non-negative offsets.
733 if (AM.BaseOffs < 0)
734 return false;
735
736 // WebAssembly has no scale register operands.
737 if (AM.Scale != 0)
738 return false;
739
740 // Everything else is legal.
741 return true;
742}
743
744bool WebAssemblyTargetLowering::allowsMisalignedMemoryAccesses(
745 EVT /*VT*/, unsigned /*AddrSpace*/, Align /*Align*/,
746 MachineMemOperand::Flags /*Flags*/, bool *Fast) const {
747 // WebAssembly supports unaligned accesses, though it should be declared
748 // with the p2align attribute on loads and stores which do so, and there
749 // may be a performance impact. We tell LLVM they're "fast" because
750 // for the kinds of things that LLVM uses this for (merging adjacent stores
751 // of constants, etc.), WebAssembly implementations will either want the
752 // unaligned access or they'll split anyway.
753 if (Fast)
754 *Fast = true;
755 return true;
756}
757
758bool WebAssemblyTargetLowering::isIntDivCheap(EVT VT,
759 AttributeList Attr) const {
760 // The current thinking is that wasm engines will perform this optimization,
761 // so we can save on code size.
762 return true;
763}
764
765bool WebAssemblyTargetLowering::isVectorLoadExtDesirable(SDValue ExtVal) const {
766 EVT ExtT = ExtVal.getValueType();
767 EVT MemT = cast<LoadSDNode>(ExtVal->getOperand(0))->getValueType(0);
768 return (ExtT == MVT::v8i16 && MemT == MVT::v8i8) ||
769 (ExtT == MVT::v4i32 && MemT == MVT::v4i16) ||
770 (ExtT == MVT::v2i64 && MemT == MVT::v2i32);
771}
772
773EVT WebAssemblyTargetLowering::getSetCCResultType(const DataLayout &DL,
774 LLVMContext &C,
775 EVT VT) const {
776 if (VT.isVector())
777 return VT.changeVectorElementTypeToInteger();
778
779 // So far, all branch instructions in Wasm take an I32 condition.
780 // The default TargetLowering::getSetCCResultType returns the pointer size,
781 // which would be useful to reduce instruction counts when testing
782 // against 64-bit pointers/values if at some point Wasm supports that.
783 return EVT::getIntegerVT(C, 32);
784}
785
786bool WebAssemblyTargetLowering::getTgtMemIntrinsic(IntrinsicInfo &Info,
787 const CallInst &I,
788 MachineFunction &MF,
789 unsigned Intrinsic) const {
790 switch (Intrinsic) {
791 case Intrinsic::wasm_memory_atomic_notify:
792 Info.opc = ISD::INTRINSIC_W_CHAIN;
793 Info.memVT = MVT::i32;
794 Info.ptrVal = I.getArgOperand(0);
795 Info.offset = 0;
796 Info.align = Align(4);
797 // atomic.notify instruction does not really load the memory specified with
798 // this argument, but MachineMemOperand should either be load or store, so
799 // we set this to a load.
800 // FIXME Volatile isn't really correct, but currently all LLVM atomic
801 // instructions are treated as volatiles in the backend, so we should be
802 // consistent. The same applies for wasm_atomic_wait intrinsics too.
803 Info.flags = MachineMemOperand::MOVolatile | MachineMemOperand::MOLoad;
804 return true;
805 case Intrinsic::wasm_memory_atomic_wait32:
806 Info.opc = ISD::INTRINSIC_W_CHAIN;
807 Info.memVT = MVT::i32;
808 Info.ptrVal = I.getArgOperand(0);
809 Info.offset = 0;
810 Info.align = Align(4);
811 Info.flags = MachineMemOperand::MOVolatile | MachineMemOperand::MOLoad;
812 return true;
813 case Intrinsic::wasm_memory_atomic_wait64:
814 Info.opc = ISD::INTRINSIC_W_CHAIN;
815 Info.memVT = MVT::i64;
816 Info.ptrVal = I.getArgOperand(0);
817 Info.offset = 0;
818 Info.align = Align(8);
819 Info.flags = MachineMemOperand::MOVolatile | MachineMemOperand::MOLoad;
820 return true;
821 default:
822 return false;
823 }
824}
825
826//===----------------------------------------------------------------------===//
827// WebAssembly Lowering private implementation.
828//===----------------------------------------------------------------------===//
829
830//===----------------------------------------------------------------------===//
831// Lowering Code
832//===----------------------------------------------------------------------===//
833
834static void fail(const SDLoc &DL, SelectionDAG &DAG, const char *Msg) {
835 MachineFunction &MF = DAG.getMachineFunction();
836 DAG.getContext()->diagnose(
837 DiagnosticInfoUnsupported(MF.getFunction(), Msg, DL.getDebugLoc()));
838}
839
840// Test whether the given calling convention is supported.
841static bool callingConvSupported(CallingConv::ID CallConv) {
842 // We currently support the language-independent target-independent
843 // conventions. We don't yet have a way to annotate calls with properties like
844 // "cold", and we don't have any call-clobbered registers, so these are mostly
845 // all handled the same.
846 return CallConv == CallingConv::C || CallConv == CallingConv::Fast ||
847 CallConv == CallingConv::Cold ||
848 CallConv == CallingConv::PreserveMost ||
849 CallConv == CallingConv::PreserveAll ||
850 CallConv == CallingConv::CXX_FAST_TLS ||
851 CallConv == CallingConv::WASM_EmscriptenInvoke ||
852 CallConv == CallingConv::Swift;
853}
854
855SDValue
856WebAssemblyTargetLowering::LowerCall(CallLoweringInfo &CLI,
857 SmallVectorImpl<SDValue> &InVals) const {
858 SelectionDAG &DAG = CLI.DAG;
859 SDLoc DL = CLI.DL;
860 SDValue Chain = CLI.Chain;
861 SDValue Callee = CLI.Callee;
862 MachineFunction &MF = DAG.getMachineFunction();
863 auto Layout = MF.getDataLayout();
864
865 CallingConv::ID CallConv = CLI.CallConv;
866 if (!callingConvSupported(CallConv))
867 fail(DL, DAG,
868 "WebAssembly doesn't support language-specific or target-specific "
869 "calling conventions yet");
870 if (CLI.IsPatchPoint)
871 fail(DL, DAG, "WebAssembly doesn't support patch point yet");
872
873 if (CLI.IsTailCall) {
874 auto NoTail = [&](const char *Msg) {
875 if (CLI.CB && CLI.CB->isMustTailCall())
876 fail(DL, DAG, Msg);
877 CLI.IsTailCall = false;
878 };
879
880 if (!Subtarget->hasTailCall())
881 NoTail("WebAssembly 'tail-call' feature not enabled");
882
883 // Varargs calls cannot be tail calls because the buffer is on the stack
884 if (CLI.IsVarArg)
885 NoTail("WebAssembly does not support varargs tail calls");
886
887 // Do not tail call unless caller and callee return types match
888 const Function &F = MF.getFunction();
889 const TargetMachine &TM = getTargetMachine();
890 Type *RetTy = F.getReturnType();
891 SmallVector<MVT, 4> CallerRetTys;
892 SmallVector<MVT, 4> CalleeRetTys;
893 computeLegalValueVTs(F, TM, RetTy, CallerRetTys);
894 computeLegalValueVTs(F, TM, CLI.RetTy, CalleeRetTys);
895 bool TypesMatch = CallerRetTys.size() == CalleeRetTys.size() &&
896 std::equal(CallerRetTys.begin(), CallerRetTys.end(),
897 CalleeRetTys.begin());
898 if (!TypesMatch)
899 NoTail("WebAssembly tail call requires caller and callee return types to "
900 "match");
901
902 // If pointers to local stack values are passed, we cannot tail call
903 if (CLI.CB) {
904 for (auto &Arg : CLI.CB->args()) {
905 Value *Val = Arg.get();
906 // Trace the value back through pointer operations
907 while (true) {
908 Value *Src = Val->stripPointerCastsAndAliases();
909 if (auto *GEP = dyn_cast<GetElementPtrInst>(Src))
910 Src = GEP->getPointerOperand();
911 if (Val == Src)
912 break;
913 Val = Src;
914 }
915 if (isa<AllocaInst>(Val)) {
916 NoTail(
917 "WebAssembly does not support tail calling with stack arguments");
918 break;
919 }
920 }
921 }
922 }
923
924 SmallVectorImpl<ISD::InputArg> &Ins = CLI.Ins;
925 SmallVectorImpl<ISD::OutputArg> &Outs = CLI.Outs;
926 SmallVectorImpl<SDValue> &OutVals = CLI.OutVals;
927
928 // The generic code may have added an sret argument. If we're lowering an
929 // invoke function, the ABI requires that the function pointer be the first
930 // argument, so we may have to swap the arguments.
931 if (CallConv == CallingConv::WASM_EmscriptenInvoke && Outs.size() >= 2 &&
932 Outs[0].Flags.isSRet()) {
933 std::swap(Outs[0], Outs[1]);
934 std::swap(OutVals[0], OutVals[1]);
935 }
936
937 bool HasSwiftSelfArg = false;
938 bool HasSwiftErrorArg = false;
939 unsigned NumFixedArgs = 0;
940 for (unsigned I = 0; I < Outs.size(); ++I) {
941 const ISD::OutputArg &Out = Outs[I];
942 SDValue &OutVal = OutVals[I];
943 HasSwiftSelfArg |= Out.Flags.isSwiftSelf();
944 HasSwiftErrorArg |= Out.Flags.isSwiftError();
945 if (Out.Flags.isNest())
946 fail(DL, DAG, "WebAssembly hasn't implemented nest arguments");
947 if (Out.Flags.isInAlloca())
948 fail(DL, DAG, "WebAssembly hasn't implemented inalloca arguments");
949 if (Out.Flags.isInConsecutiveRegs())
950 fail(DL, DAG, "WebAssembly hasn't implemented cons regs arguments");
951 if (Out.Flags.isInConsecutiveRegsLast())
952 fail(DL, DAG, "WebAssembly hasn't implemented cons regs last arguments");
953 if (Out.Flags.isByVal() && Out.Flags.getByValSize() != 0) {
954 auto &MFI = MF.getFrameInfo();
955 int FI = MFI.CreateStackObject(Out.Flags.getByValSize(),
956 Out.Flags.getNonZeroByValAlign(),
957 /*isSS=*/false);
958 SDValue SizeNode =
959 DAG.getConstant(Out.Flags.getByValSize(), DL, MVT::i32);
960 SDValue FINode = DAG.getFrameIndex(FI, getPointerTy(Layout));
961 Chain = DAG.getMemcpy(
962 Chain, DL, FINode, OutVal, SizeNode, Out.Flags.getNonZeroByValAlign(),
963 /*isVolatile*/ false, /*AlwaysInline=*/false,
964 /*isTailCall*/ false, MachinePointerInfo(), MachinePointerInfo());
965 OutVal = FINode;
966 }
967 // Count the number of fixed args *after* legalization.
968 NumFixedArgs += Out.IsFixed;
969 }
970
971 bool IsVarArg = CLI.IsVarArg;
972 auto PtrVT = getPointerTy(Layout);
973
974 // For swiftcc, emit additional swiftself and swifterror arguments
975 // if there aren't. These additional arguments are also added for callee
976 // signature They are necessary to match callee and caller signature for
977 // indirect call.
978 if (CallConv == CallingConv::Swift) {
979 if (!HasSwiftSelfArg) {
980 NumFixedArgs++;
981 ISD::OutputArg Arg;
982 Arg.Flags.setSwiftSelf();
983 CLI.Outs.push_back(Arg);
984 SDValue ArgVal = DAG.getUNDEF(PtrVT);
985 CLI.OutVals.push_back(ArgVal);
986 }
987 if (!HasSwiftErrorArg) {
988 NumFixedArgs++;
989 ISD::OutputArg Arg;
990 Arg.Flags.setSwiftError();
991 CLI.Outs.push_back(Arg);
992 SDValue ArgVal = DAG.getUNDEF(PtrVT);
993 CLI.OutVals.push_back(ArgVal);
994 }
995 }
996
997 // Analyze operands of the call, assigning locations to each operand.
998 SmallVector<CCValAssign, 16> ArgLocs;
999 CCState CCInfo(CallConv, IsVarArg, MF, ArgLocs, *DAG.getContext());
1000
1001 if (IsVarArg) {
1002 // Outgoing non-fixed arguments are placed in a buffer. First
1003 // compute their offsets and the total amount of buffer space needed.
1004 for (unsigned I = NumFixedArgs; I < Outs.size(); ++I) {
1005 const ISD::OutputArg &Out = Outs[I];
1006 SDValue &Arg = OutVals[I];
1007 EVT VT = Arg.getValueType();
1008 assert(VT != MVT::iPTR && "Legalized args should be concrete")(static_cast <bool> (VT != MVT::iPTR && "Legalized args should be concrete"
) ? void (0) : __assert_fail ("VT != MVT::iPTR && \"Legalized args should be concrete\""
, "/build/llvm-toolchain-snapshot-13~++20210726100616+dead50d4427c/llvm/lib/Target/WebAssembly/WebAssemblyISelLowering.cpp"
, 1008, __extension__ __PRETTY_FUNCTION__))
;
1009 Type *Ty = VT.getTypeForEVT(*DAG.getContext());
1010 Align Alignment =
1011 std::max(Out.Flags.getNonZeroOrigAlign(), Layout.getABITypeAlign(Ty));
1012 unsigned Offset =
1013 CCInfo.AllocateStack(Layout.getTypeAllocSize(Ty), Alignment);
1014 CCInfo.addLoc(CCValAssign::getMem(ArgLocs.size(), VT.getSimpleVT(),
1015 Offset, VT.getSimpleVT(),
1016 CCValAssign::Full));
1017 }
1018 }
1019
1020 unsigned NumBytes = CCInfo.getAlignedCallFrameSize();
1021
1022 SDValue FINode;
1023 if (IsVarArg && NumBytes) {
1024 // For non-fixed arguments, next emit stores to store the argument values
1025 // to the stack buffer at the offsets computed above.
1026 int FI = MF.getFrameInfo().CreateStackObject(NumBytes,
1027 Layout.getStackAlignment(),
1028 /*isSS=*/false);
1029 unsigned ValNo = 0;
1030 SmallVector<SDValue, 8> Chains;
1031 for (SDValue Arg : drop_begin(OutVals, NumFixedArgs)) {
1032 assert(ArgLocs[ValNo].getValNo() == ValNo &&(static_cast <bool> (ArgLocs[ValNo].getValNo() == ValNo
&& "ArgLocs should remain in order and only hold varargs args"
) ? void (0) : __assert_fail ("ArgLocs[ValNo].getValNo() == ValNo && \"ArgLocs should remain in order and only hold varargs args\""
, "/build/llvm-toolchain-snapshot-13~++20210726100616+dead50d4427c/llvm/lib/Target/WebAssembly/WebAssemblyISelLowering.cpp"
, 1033, __extension__ __PRETTY_FUNCTION__))
1033 "ArgLocs should remain in order and only hold varargs args")(static_cast <bool> (ArgLocs[ValNo].getValNo() == ValNo
&& "ArgLocs should remain in order and only hold varargs args"
) ? void (0) : __assert_fail ("ArgLocs[ValNo].getValNo() == ValNo && \"ArgLocs should remain in order and only hold varargs args\""
, "/build/llvm-toolchain-snapshot-13~++20210726100616+dead50d4427c/llvm/lib/Target/WebAssembly/WebAssemblyISelLowering.cpp"
, 1033, __extension__ __PRETTY_FUNCTION__))
;
1034 unsigned Offset = ArgLocs[ValNo++].getLocMemOffset();
1035 FINode = DAG.getFrameIndex(FI, getPointerTy(Layout));
1036 SDValue Add = DAG.getNode(ISD::ADD, DL, PtrVT, FINode,
1037 DAG.getConstant(Offset, DL, PtrVT));
1038 Chains.push_back(
1039 DAG.getStore(Chain, DL, Arg, Add,
1040 MachinePointerInfo::getFixedStack(MF, FI, Offset)));
1041 }
1042 if (!Chains.empty())
1043 Chain = DAG.getNode(ISD::TokenFactor, DL, MVT::Other, Chains);
1044 } else if (IsVarArg) {
1045 FINode = DAG.getIntPtrConstant(0, DL);
1046 }
1047
1048 if (Callee->getOpcode() == ISD::GlobalAddress) {
1049 // If the callee is a GlobalAddress node (quite common, every direct call
1050 // is) turn it into a TargetGlobalAddress node so that LowerGlobalAddress
1051 // doesn't at MO_GOT which is not needed for direct calls.
1052 GlobalAddressSDNode* GA = cast<GlobalAddressSDNode>(Callee);
1053 Callee = DAG.getTargetGlobalAddress(GA->getGlobal(), DL,
1054 getPointerTy(DAG.getDataLayout()),
1055 GA->getOffset());
1056 Callee = DAG.getNode(WebAssemblyISD::Wrapper, DL,
1057 getPointerTy(DAG.getDataLayout()), Callee);
1058 }
1059
1060 // Compute the operands for the CALLn node.
1061 SmallVector<SDValue, 16> Ops;
1062 Ops.push_back(Chain);
1063 Ops.push_back(Callee);
1064
1065 // Add all fixed arguments. Note that for non-varargs calls, NumFixedArgs
1066 // isn't reliable.
1067 Ops.append(OutVals.begin(),
1068 IsVarArg ? OutVals.begin() + NumFixedArgs : OutVals.end());
1069 // Add a pointer to the vararg buffer.
1070 if (IsVarArg)
1071 Ops.push_back(FINode);
1072
1073 SmallVector<EVT, 8> InTys;
1074 for (const auto &In : Ins) {
1075 assert(!In.Flags.isByVal() && "byval is not valid for return values")(static_cast <bool> (!In.Flags.isByVal() && "byval is not valid for return values"
) ? void (0) : __assert_fail ("!In.Flags.isByVal() && \"byval is not valid for return values\""
, "/build/llvm-toolchain-snapshot-13~++20210726100616+dead50d4427c/llvm/lib/Target/WebAssembly/WebAssemblyISelLowering.cpp"
, 1075, __extension__ __PRETTY_FUNCTION__))
;
1076 assert(!In.Flags.isNest() && "nest is not valid for return values")(static_cast <bool> (!In.Flags.isNest() && "nest is not valid for return values"
) ? void (0) : __assert_fail ("!In.Flags.isNest() && \"nest is not valid for return values\""
, "/build/llvm-toolchain-snapshot-13~++20210726100616+dead50d4427c/llvm/lib/Target/WebAssembly/WebAssemblyISelLowering.cpp"
, 1076, __extension__ __PRETTY_FUNCTION__))
;
1077 if (In.Flags.isInAlloca())
1078 fail(DL, DAG, "WebAssembly hasn't implemented inalloca return values");
1079 if (In.Flags.isInConsecutiveRegs())
1080 fail(DL, DAG, "WebAssembly hasn't implemented cons regs return values");
1081 if (In.Flags.isInConsecutiveRegsLast())
1082 fail(DL, DAG,
1083 "WebAssembly hasn't implemented cons regs last return values");
1084 // Ignore In.getNonZeroOrigAlign() because all our arguments are passed in
1085 // registers.
1086 InTys.push_back(In.VT);
1087 }
1088
1089 // Lastly, if this is a call to a funcref we need to add an instruction
1090 // table.set to the chain and transform the call.
1091 if (CLI.CB && isFuncrefType(CLI.CB->getCalledOperand()->getType())) {
1092 // In the absence of function references proposal where a funcref call is
1093 // lowered to call_ref, using reference types we generate a table.set to set
1094 // the funcref to a special table used solely for this purpose, followed by
1095 // a call_indirect. Here we just generate the table set, and return the
1096 // SDValue of the table.set so that LowerCall can finalize the lowering by
1097 // generating the call_indirect.
1098 SDValue Chain = Ops[0];
1099
1100 MCSymbolWasm *Table = WebAssembly::getOrCreateFuncrefCallTableSymbol(
1101 MF.getContext(), Subtarget);
1102 SDValue Sym = DAG.getMCSymbol(Table, PtrVT);
1103 SDValue TableSlot = DAG.getConstant(0, DL, MVT::i32);
1104 SDValue TableSetOps[] = {Chain, Sym, TableSlot, Callee};
1105 SDValue TableSet = DAG.getMemIntrinsicNode(
1106 WebAssemblyISD::TABLE_SET, DL, DAG.getVTList(MVT::Other), TableSetOps,
1107 MVT::funcref,
1108 // Machine Mem Operand args
1109 MachinePointerInfo(WasmAddressSpace::FUNCREF),
1110 CLI.CB->getCalledOperand()->getPointerAlignment(DAG.getDataLayout()),
1111 MachineMemOperand::MOStore);
1112
1113 Ops[0] = TableSet; // The new chain is the TableSet itself
1114 }
1115
1116 if (CLI.IsTailCall) {
1117 // ret_calls do not return values to the current frame
1118 SDVTList NodeTys = DAG.getVTList(MVT::Other, MVT::Glue);
1119 return DAG.getNode(WebAssemblyISD::RET_CALL, DL, NodeTys, Ops);
1120 }
1121
1122 InTys.push_back(MVT::Other);
1123 SDVTList InTyList = DAG.getVTList(InTys);
1124 SDValue Res = DAG.getNode(WebAssemblyISD::CALL, DL, InTyList, Ops);
1125
1126 for (size_t I = 0; I < Ins.size(); ++I)
1127 InVals.push_back(Res.getValue(I));
1128
1129 // Return the chain
1130 return Res.getValue(Ins.size());
1131}
1132
1133bool WebAssemblyTargetLowering::CanLowerReturn(
1134 CallingConv::ID /*CallConv*/, MachineFunction & /*MF*/, bool /*IsVarArg*/,
1135 const SmallVectorImpl<ISD::OutputArg> &Outs,
1136 LLVMContext & /*Context*/) const {
1137 // WebAssembly can only handle returning tuples with multivalue enabled
1138 return Subtarget->hasMultivalue() || Outs.size() <= 1;
1139}
1140
1141SDValue WebAssemblyTargetLowering::LowerReturn(
1142 SDValue Chain, CallingConv::ID CallConv, bool /*IsVarArg*/,
1143 const SmallVectorImpl<ISD::OutputArg> &Outs,
1144 const SmallVectorImpl<SDValue> &OutVals, const SDLoc &DL,
1145 SelectionDAG &DAG) const {
1146 assert((Subtarget->hasMultivalue() || Outs.size() <= 1) &&(static_cast <bool> ((Subtarget->hasMultivalue() || Outs
.size() <= 1) && "MVP WebAssembly can only return up to one value"
) ? void (0) : __assert_fail ("(Subtarget->hasMultivalue() || Outs.size() <= 1) && \"MVP WebAssembly can only return up to one value\""
, "/build/llvm-toolchain-snapshot-13~++20210726100616+dead50d4427c/llvm/lib/Target/WebAssembly/WebAssemblyISelLowering.cpp"
, 1147, __extension__ __PRETTY_FUNCTION__))
1147 "MVP WebAssembly can only return up to one value")(static_cast <bool> ((Subtarget->hasMultivalue() || Outs
.size() <= 1) && "MVP WebAssembly can only return up to one value"
) ? void (0) : __assert_fail ("(Subtarget->hasMultivalue() || Outs.size() <= 1) && \"MVP WebAssembly can only return up to one value\""
, "/build/llvm-toolchain-snapshot-13~++20210726100616+dead50d4427c/llvm/lib/Target/WebAssembly/WebAssemblyISelLowering.cpp"
, 1147, __extension__ __PRETTY_FUNCTION__))
;
1148 if (!callingConvSupported(CallConv))
1149 fail(DL, DAG, "WebAssembly doesn't support non-C calling conventions");
1150
1151 SmallVector<SDValue, 4> RetOps(1, Chain);
1152 RetOps.append(OutVals.begin(), OutVals.end());
1153 Chain = DAG.getNode(WebAssemblyISD::RETURN, DL, MVT::Other, RetOps);
1154
1155 // Record the number and types of the return values.
1156 for (const ISD::OutputArg &Out : Outs) {
1157 assert(!Out.Flags.isByVal() && "byval is not valid for return values")(static_cast <bool> (!Out.Flags.isByVal() && "byval is not valid for return values"
) ? void (0) : __assert_fail ("!Out.Flags.isByVal() && \"byval is not valid for return values\""
, "/build/llvm-toolchain-snapshot-13~++20210726100616+dead50d4427c/llvm/lib/Target/WebAssembly/WebAssemblyISelLowering.cpp"
, 1157, __extension__ __PRETTY_FUNCTION__))
;
1158 assert(!Out.Flags.isNest() && "nest is not valid for return values")(static_cast <bool> (!Out.Flags.isNest() && "nest is not valid for return values"
) ? void (0) : __assert_fail ("!Out.Flags.isNest() && \"nest is not valid for return values\""
, "/build/llvm-toolchain-snapshot-13~++20210726100616+dead50d4427c/llvm/lib/Target/WebAssembly/WebAssemblyISelLowering.cpp"
, 1158, __extension__ __PRETTY_FUNCTION__))
;
1159 assert(Out.IsFixed && "non-fixed return value is not valid")(static_cast <bool> (Out.IsFixed && "non-fixed return value is not valid"
) ? void (0) : __assert_fail ("Out.IsFixed && \"non-fixed return value is not valid\""
, "/build/llvm-toolchain-snapshot-13~++20210726100616+dead50d4427c/llvm/lib/Target/WebAssembly/WebAssemblyISelLowering.cpp"
, 1159, __extension__ __PRETTY_FUNCTION__))
;
1160 if (Out.Flags.isInAlloca())
1161 fail(DL, DAG, "WebAssembly hasn't implemented inalloca results");
1162 if (Out.Flags.isInConsecutiveRegs())
1163 fail(DL, DAG, "WebAssembly hasn't implemented cons regs results");
1164 if (Out.Flags.isInConsecutiveRegsLast())
1165 fail(DL, DAG, "WebAssembly hasn't implemented cons regs last results");
1166 }
1167
1168 return Chain;
1169}
1170
1171SDValue WebAssemblyTargetLowering::LowerFormalArguments(
1172 SDValue Chain, CallingConv::ID CallConv, bool IsVarArg,
1173 const SmallVectorImpl<ISD::InputArg> &Ins, const SDLoc &DL,
1174 SelectionDAG &DAG, SmallVectorImpl<SDValue> &InVals) const {
1175 if (!callingConvSupported(CallConv))
1176 fail(DL, DAG, "WebAssembly doesn't support non-C calling conventions");
1177
1178 MachineFunction &MF = DAG.getMachineFunction();
1179 auto *MFI = MF.getInfo<WebAssemblyFunctionInfo>();
1180
1181 // Set up the incoming ARGUMENTS value, which serves to represent the liveness
1182 // of the incoming values before they're represented by virtual registers.
1183 MF.getRegInfo().addLiveIn(WebAssembly::ARGUMENTS);
1184
1185 bool HasSwiftErrorArg = false;
1186 bool HasSwiftSelfArg = false;
1187 for (const ISD::InputArg &In : Ins) {
1188 HasSwiftSelfArg |= In.Flags.isSwiftSelf();
1189 HasSwiftErrorArg |= In.Flags.isSwiftError();
1190 if (In.Flags.isInAlloca())
1191 fail(DL, DAG, "WebAssembly hasn't implemented inalloca arguments");
1192 if (In.Flags.isNest())
1193 fail(DL, DAG, "WebAssembly hasn't implemented nest arguments");
1194 if (In.Flags.isInConsecutiveRegs())
1195 fail(DL, DAG, "WebAssembly hasn't implemented cons regs arguments");
1196 if (In.Flags.isInConsecutiveRegsLast())
1197 fail(DL, DAG, "WebAssembly hasn't implemented cons regs last arguments");
1198 // Ignore In.getNonZeroOrigAlign() because all our arguments are passed in
1199 // registers.
1200 InVals.push_back(In.Used ? DAG.getNode(WebAssemblyISD::ARGUMENT, DL, In.VT,
1201 DAG.getTargetConstant(InVals.size(),
1202 DL, MVT::i32))
1203 : DAG.getUNDEF(In.VT));
1204
1205 // Record the number and types of arguments.
1206 MFI->addParam(In.VT);
1207 }
1208
1209 // For swiftcc, emit additional swiftself and swifterror arguments
1210 // if there aren't. These additional arguments are also added for callee
1211 // signature They are necessary to match callee and caller signature for
1212 // indirect call.
1213 auto PtrVT = getPointerTy(MF.getDataLayout());
1214 if (CallConv == CallingConv::Swift) {
1215 if (!HasSwiftSelfArg) {
1216 MFI->addParam(PtrVT);
1217 }
1218 if (!HasSwiftErrorArg) {
1219 MFI->addParam(PtrVT);
1220 }
1221 }
1222 // Varargs are copied into a buffer allocated by the caller, and a pointer to
1223 // the buffer is passed as an argument.
1224 if (IsVarArg) {
1225 MVT PtrVT = getPointerTy(MF.getDataLayout());
1226 Register VarargVreg =
1227 MF.getRegInfo().createVirtualRegister(getRegClassFor(PtrVT));
1228 MFI->setVarargBufferVreg(VarargVreg);
1229 Chain = DAG.getCopyToReg(
1230 Chain, DL, VarargVreg,
1231 DAG.getNode(WebAssemblyISD::ARGUMENT, DL, PtrVT,
1232 DAG.getTargetConstant(Ins.size(), DL, MVT::i32)));
1233 MFI->addParam(PtrVT);
1234 }
1235
1236 // Record the number and types of arguments and results.
1237 SmallVector<MVT, 4> Params;
1238 SmallVector<MVT, 4> Results;
1239 computeSignatureVTs(MF.getFunction().getFunctionType(), &MF.getFunction(),
1240 MF.getFunction(), DAG.getTarget(), Params, Results);
1241 for (MVT VT : Results)
1242 MFI->addResult(VT);
1243 // TODO: Use signatures in WebAssemblyMachineFunctionInfo too and unify
1244 // the param logic here with ComputeSignatureVTs
1245 assert(MFI->getParams().size() == Params.size() &&(static_cast <bool> (MFI->getParams().size() == Params
.size() && std::equal(MFI->getParams().begin(), MFI
->getParams().end(), Params.begin())) ? void (0) : __assert_fail
("MFI->getParams().size() == Params.size() && std::equal(MFI->getParams().begin(), MFI->getParams().end(), Params.begin())"
, "/build/llvm-toolchain-snapshot-13~++20210726100616+dead50d4427c/llvm/lib/Target/WebAssembly/WebAssemblyISelLowering.cpp"
, 1247, __extension__ __PRETTY_FUNCTION__))
1246 std::equal(MFI->getParams().begin(), MFI->getParams().end(),(static_cast <bool> (MFI->getParams().size() == Params
.size() && std::equal(MFI->getParams().begin(), MFI
->getParams().end(), Params.begin())) ? void (0) : __assert_fail
("MFI->getParams().size() == Params.size() && std::equal(MFI->getParams().begin(), MFI->getParams().end(), Params.begin())"
, "/build/llvm-toolchain-snapshot-13~++20210726100616+dead50d4427c/llvm/lib/Target/WebAssembly/WebAssemblyISelLowering.cpp"
, 1247, __extension__ __PRETTY_FUNCTION__))
1247 Params.begin()))(static_cast <bool> (MFI->getParams().size() == Params
.size() && std::equal(MFI->getParams().begin(), MFI
->getParams().end(), Params.begin())) ? void (0) : __assert_fail
("MFI->getParams().size() == Params.size() && std::equal(MFI->getParams().begin(), MFI->getParams().end(), Params.begin())"
, "/build/llvm-toolchain-snapshot-13~++20210726100616+dead50d4427c/llvm/lib/Target/WebAssembly/WebAssemblyISelLowering.cpp"
, 1247, __extension__ __PRETTY_FUNCTION__))
;
1248
1249 return Chain;
1250}
1251
1252void WebAssemblyTargetLowering::ReplaceNodeResults(
1253 SDNode *N, SmallVectorImpl<SDValue> &Results, SelectionDAG &DAG) const {
1254 switch (N->getOpcode()) {
1255 case ISD::SIGN_EXTEND_INREG:
1256 // Do not add any results, signifying that N should not be custom lowered
1257 // after all. This happens because simd128 turns on custom lowering for
1258 // SIGN_EXTEND_INREG, but for non-vector sign extends the result might be an
1259 // illegal type.
1260 break;
1261 default:
1262 llvm_unreachable(::llvm::llvm_unreachable_internal("ReplaceNodeResults not implemented for this op for WebAssembly!"
, "/build/llvm-toolchain-snapshot-13~++20210726100616+dead50d4427c/llvm/lib/Target/WebAssembly/WebAssemblyISelLowering.cpp"
, 1263)
1263 "ReplaceNodeResults not implemented for this op for WebAssembly!")::llvm::llvm_unreachable_internal("ReplaceNodeResults not implemented for this op for WebAssembly!"
, "/build/llvm-toolchain-snapshot-13~++20210726100616+dead50d4427c/llvm/lib/Target/WebAssembly/WebAssemblyISelLowering.cpp"
, 1263)
;
1264 }
1265}
1266
1267//===----------------------------------------------------------------------===//
1268// Custom lowering hooks.
1269//===----------------------------------------------------------------------===//
1270
1271SDValue WebAssemblyTargetLowering::LowerOperation(SDValue Op,
1272 SelectionDAG &DAG) const {
1273 SDLoc DL(Op);
1274 switch (Op.getOpcode()) {
1
Control jumps to 'case BUILD_VECTOR:' at line 1311
1275 default:
1276 llvm_unreachable("unimplemented operation lowering")::llvm::llvm_unreachable_internal("unimplemented operation lowering"
, "/build/llvm-toolchain-snapshot-13~++20210726100616+dead50d4427c/llvm/lib/Target/WebAssembly/WebAssemblyISelLowering.cpp"
, 1276)
;
1277 return SDValue();
1278 case ISD::FrameIndex:
1279 return LowerFrameIndex(Op, DAG);
1280 case ISD::GlobalAddress:
1281 return LowerGlobalAddress(Op, DAG);
1282 case ISD::GlobalTLSAddress:
1283 return LowerGlobalTLSAddress(Op, DAG);
1284 case ISD::ExternalSymbol:
1285 return LowerExternalSymbol(Op, DAG);
1286 case ISD::JumpTable:
1287 return LowerJumpTable(Op, DAG);
1288 case ISD::BR_JT:
1289 return LowerBR_JT(Op, DAG);
1290 case ISD::VASTART:
1291 return LowerVASTART(Op, DAG);
1292 case ISD::BlockAddress:
1293 case ISD::BRIND:
1294 fail(DL, DAG, "WebAssembly hasn't implemented computed gotos");
1295 return SDValue();
1296 case ISD::RETURNADDR:
1297 return LowerRETURNADDR(Op, DAG);
1298 case ISD::FRAMEADDR:
1299 return LowerFRAMEADDR(Op, DAG);
1300 case ISD::CopyToReg:
1301 return LowerCopyToReg(Op, DAG);
1302 case ISD::EXTRACT_VECTOR_ELT:
1303 case ISD::INSERT_VECTOR_ELT:
1304 return LowerAccessVectorElement(Op, DAG);
1305 case ISD::INTRINSIC_VOID:
1306 case ISD::INTRINSIC_WO_CHAIN:
1307 case ISD::INTRINSIC_W_CHAIN:
1308 return LowerIntrinsic(Op, DAG);
1309 case ISD::SIGN_EXTEND_INREG:
1310 return LowerSIGN_EXTEND_INREG(Op, DAG);
1311 case ISD::BUILD_VECTOR:
1312 return LowerBUILD_VECTOR(Op, DAG);
2
Calling 'WebAssemblyTargetLowering::LowerBUILD_VECTOR'
1313 case ISD::VECTOR_SHUFFLE:
1314 return LowerVECTOR_SHUFFLE(Op, DAG);
1315 case ISD::SETCC:
1316 return LowerSETCC(Op, DAG);
1317 case ISD::SHL:
1318 case ISD::SRA:
1319 case ISD::SRL:
1320 return LowerShift(Op, DAG);
1321 case ISD::FP_TO_SINT_SAT:
1322 case ISD::FP_TO_UINT_SAT:
1323 return LowerFP_TO_INT_SAT(Op, DAG);
1324 case ISD::LOAD:
1325 return LowerLoad(Op, DAG);
1326 case ISD::STORE:
1327 return LowerStore(Op, DAG);
1328 }
1329}
1330
1331static bool IsWebAssemblyGlobal(SDValue Op) {
1332 if (const GlobalAddressSDNode *GA = dyn_cast<GlobalAddressSDNode>(Op))
1333 return WebAssembly::isWasmVarAddressSpace(GA->getAddressSpace());
1334
1335 return false;
1336}
1337
1338static Optional<unsigned> IsWebAssemblyLocal(SDValue Op, SelectionDAG &DAG) {
1339 const FrameIndexSDNode *FI = dyn_cast<FrameIndexSDNode>(Op);
1340 if (!FI)
1341 return None;
1342
1343 auto &MF = DAG.getMachineFunction();
1344 return WebAssemblyFrameLowering::getLocalForStackObject(MF, FI->getIndex());
1345}
1346
1347bool WebAssemblyTargetLowering::isFuncrefType(const Type *Ty) {
1348 return isa<PointerType>(Ty) &&
1349 Ty->getPointerAddressSpace() == WasmAddressSpace::FUNCREF;
1350}
1351
1352bool WebAssemblyTargetLowering::isExternrefType(const Type *Ty) {
1353 return isa<PointerType>(Ty) &&
1354 Ty->getPointerAddressSpace() == WasmAddressSpace::EXTERNREF;
1355}
1356
1357SDValue WebAssemblyTargetLowering::LowerStore(SDValue Op,
1358 SelectionDAG &DAG) const {
1359 SDLoc DL(Op);
1360 StoreSDNode *SN = cast<StoreSDNode>(Op.getNode());
1361 const SDValue &Value = SN->getValue();
1362 const SDValue &Base = SN->getBasePtr();
1363 const SDValue &Offset = SN->getOffset();
1364
1365 if (IsWebAssemblyGlobal(Base)) {
1366 if (!Offset->isUndef())
1367 report_fatal_error("unexpected offset when storing to webassembly global",
1368 false);
1369
1370 SDVTList Tys = DAG.getVTList(MVT::Other);
1371 SDValue Ops[] = {SN->getChain(), Value, Base};
1372 return DAG.getMemIntrinsicNode(WebAssemblyISD::GLOBAL_SET, DL, Tys, Ops,
1373 SN->getMemoryVT(), SN->getMemOperand());
1374 }
1375
1376 if (Optional<unsigned> Local = IsWebAssemblyLocal(Base, DAG)) {
1377 if (!Offset->isUndef())
1378 report_fatal_error("unexpected offset when storing to webassembly local",
1379 false);
1380
1381 SDValue Idx = DAG.getTargetConstant(*Local, Base, MVT::i32);
1382 SDVTList Tys = DAG.getVTList(MVT::Other); // The chain.
1383 SDValue Ops[] = {SN->getChain(), Idx, Value};
1384 return DAG.getNode(WebAssemblyISD::LOCAL_SET, DL, Tys, Ops);
1385 }
1386
1387 return Op;
1388}
1389
1390SDValue WebAssemblyTargetLowering::LowerLoad(SDValue Op,
1391 SelectionDAG &DAG) const {
1392 SDLoc DL(Op);
1393 LoadSDNode *LN = cast<LoadSDNode>(Op.getNode());
1394 const SDValue &Base = LN->getBasePtr();
1395 const SDValue &Offset = LN->getOffset();
1396
1397 if (IsWebAssemblyGlobal(Base)) {
1398 if (!Offset->isUndef())
1399 report_fatal_error(
1400 "unexpected offset when loading from webassembly global", false);
1401
1402 SDVTList Tys = DAG.getVTList(LN->getValueType(0), MVT::Other);
1403 SDValue Ops[] = {LN->getChain(), Base};
1404 return DAG.getMemIntrinsicNode(WebAssemblyISD::GLOBAL_GET, DL, Tys, Ops,
1405 LN->getMemoryVT(), LN->getMemOperand());
1406 }
1407
1408 if (Optional<unsigned> Local = IsWebAssemblyLocal(Base, DAG)) {
1409 if (!Offset->isUndef())
1410 report_fatal_error(
1411 "unexpected offset when loading from webassembly local", false);
1412
1413 SDValue Idx = DAG.getTargetConstant(*Local, Base, MVT::i32);
1414 EVT LocalVT = LN->getValueType(0);
1415 SDValue LocalGet = DAG.getNode(WebAssemblyISD::LOCAL_GET, DL, LocalVT,
1416 {LN->getChain(), Idx});
1417 SDValue Result = DAG.getMergeValues({LocalGet, LN->getChain()}, DL);
1418 assert(Result->getNumValues() == 2 && "Loads must carry a chain!")(static_cast <bool> (Result->getNumValues() == 2 &&
"Loads must carry a chain!") ? void (0) : __assert_fail ("Result->getNumValues() == 2 && \"Loads must carry a chain!\""
, "/build/llvm-toolchain-snapshot-13~++20210726100616+dead50d4427c/llvm/lib/Target/WebAssembly/WebAssemblyISelLowering.cpp"
, 1418, __extension__ __PRETTY_FUNCTION__))
;
1419 return Result;
1420 }
1421
1422 return Op;
1423}
1424
1425SDValue WebAssemblyTargetLowering::LowerCopyToReg(SDValue Op,
1426 SelectionDAG &DAG) const {
1427 SDValue Src = Op.getOperand(2);
1428 if (isa<FrameIndexSDNode>(Src.getNode())) {
1429 // CopyToReg nodes don't support FrameIndex operands. Other targets select
1430 // the FI to some LEA-like instruction, but since we don't have that, we
1431 // need to insert some kind of instruction that can take an FI operand and
1432 // produces a value usable by CopyToReg (i.e. in a vreg). So insert a dummy
1433 // local.copy between Op and its FI operand.
1434 SDValue Chain = Op.getOperand(0);
1435 SDLoc DL(Op);
1436 unsigned Reg = cast<RegisterSDNode>(Op.getOperand(1))->getReg();
1437 EVT VT = Src.getValueType();
1438 SDValue Copy(DAG.getMachineNode(VT == MVT::i32 ? WebAssembly::COPY_I32
1439 : WebAssembly::COPY_I64,
1440 DL, VT, Src),
1441 0);
1442 return Op.getNode()->getNumValues() == 1
1443 ? DAG.getCopyToReg(Chain, DL, Reg, Copy)
1444 : DAG.getCopyToReg(Chain, DL, Reg, Copy,
1445 Op.getNumOperands() == 4 ? Op.getOperand(3)
1446 : SDValue());
1447 }
1448 return SDValue();
1449}
1450
1451SDValue WebAssemblyTargetLowering::LowerFrameIndex(SDValue Op,
1452 SelectionDAG &DAG) const {
1453 int FI = cast<FrameIndexSDNode>(Op)->getIndex();
1454 return DAG.getTargetFrameIndex(FI, Op.getValueType());
1455}
1456
1457SDValue WebAssemblyTargetLowering::LowerRETURNADDR(SDValue Op,
1458 SelectionDAG &DAG) const {
1459 SDLoc DL(Op);
1460
1461 if (!Subtarget->getTargetTriple().isOSEmscripten()) {
1462 fail(DL, DAG,
1463 "Non-Emscripten WebAssembly hasn't implemented "
1464 "__builtin_return_address");
1465 return SDValue();
1466 }
1467
1468 if (verifyReturnAddressArgumentIsConstant(Op, DAG))
1469 return SDValue();
1470
1471 unsigned Depth = cast<ConstantSDNode>(Op.getOperand(0))->getZExtValue();
1472 MakeLibCallOptions CallOptions;
1473 return makeLibCall(DAG, RTLIB::RETURN_ADDRESS, Op.getValueType(),
1474 {DAG.getConstant(Depth, DL, MVT::i32)}, CallOptions, DL)
1475 .first;
1476}
1477
1478SDValue WebAssemblyTargetLowering::LowerFRAMEADDR(SDValue Op,
1479 SelectionDAG &DAG) const {
1480 // Non-zero depths are not supported by WebAssembly currently. Use the
1481 // legalizer's default expansion, which is to return 0 (what this function is
1482 // documented to do).
1483 if (Op.getConstantOperandVal(0) > 0)
1484 return SDValue();
1485
1486 DAG.getMachineFunction().getFrameInfo().setFrameAddressIsTaken(true);
1487 EVT VT = Op.getValueType();
1488 Register FP =
1489 Subtarget->getRegisterInfo()->getFrameRegister(DAG.getMachineFunction());
1490 return DAG.getCopyFromReg(DAG.getEntryNode(), SDLoc(Op), FP, VT);
1491}
1492
1493SDValue
1494WebAssemblyTargetLowering::LowerGlobalTLSAddress(SDValue Op,
1495 SelectionDAG &DAG) const {
1496 SDLoc DL(Op);
1497 const auto *GA = cast<GlobalAddressSDNode>(Op);
1498 MVT PtrVT = getPointerTy(DAG.getDataLayout());
1499
1500 MachineFunction &MF = DAG.getMachineFunction();
1501 if (!MF.getSubtarget<WebAssemblySubtarget>().hasBulkMemory())
1502 report_fatal_error("cannot use thread-local storage without bulk memory",
1503 false);
1504
1505 const GlobalValue *GV = GA->getGlobal();
1506
1507 // Currently Emscripten does not support dynamic linking with threads.
1508 // Therefore, if we have thread-local storage, only the local-exec model
1509 // is possible.
1510 // TODO: remove this and implement proper TLS models once Emscripten
1511 // supports dynamic linking with threads.
1512 if (GV->getThreadLocalMode() != GlobalValue::LocalExecTLSModel &&
1513 !Subtarget->getTargetTriple().isOSEmscripten()) {
1514 report_fatal_error("only -ftls-model=local-exec is supported for now on "
1515 "non-Emscripten OSes: variable " +
1516 GV->getName(),
1517 false);
1518 }
1519
1520 auto GlobalGet = PtrVT == MVT::i64 ? WebAssembly::GLOBAL_GET_I64
1521 : WebAssembly::GLOBAL_GET_I32;
1522 const char *BaseName = MF.createExternalSymbolName("__tls_base");
1523
1524 SDValue BaseAddr(
1525 DAG.getMachineNode(GlobalGet, DL, PtrVT,
1526 DAG.getTargetExternalSymbol(BaseName, PtrVT)),
1527 0);
1528
1529 SDValue TLSOffset = DAG.getTargetGlobalAddress(
1530 GV, DL, PtrVT, GA->getOffset(), WebAssemblyII::MO_TLS_BASE_REL);
1531 SDValue SymAddr = DAG.getNode(WebAssemblyISD::Wrapper, DL, PtrVT, TLSOffset);
1532
1533 return DAG.getNode(ISD::ADD, DL, PtrVT, BaseAddr, SymAddr);
1534}
1535
1536SDValue WebAssemblyTargetLowering::LowerGlobalAddress(SDValue Op,
1537 SelectionDAG &DAG) const {
1538 SDLoc DL(Op);
1539 const auto *GA = cast<GlobalAddressSDNode>(Op);
1540 EVT VT = Op.getValueType();
1541 assert(GA->getTargetFlags() == 0 &&(static_cast <bool> (GA->getTargetFlags() == 0 &&
"Unexpected target flags on generic GlobalAddressSDNode") ? void
(0) : __assert_fail ("GA->getTargetFlags() == 0 && \"Unexpected target flags on generic GlobalAddressSDNode\""
, "/build/llvm-toolchain-snapshot-13~++20210726100616+dead50d4427c/llvm/lib/Target/WebAssembly/WebAssemblyISelLowering.cpp"
, 1542, __extension__ __PRETTY_FUNCTION__))
1542 "Unexpected target flags on generic GlobalAddressSDNode")(static_cast <bool> (GA->getTargetFlags() == 0 &&
"Unexpected target flags on generic GlobalAddressSDNode") ? void
(0) : __assert_fail ("GA->getTargetFlags() == 0 && \"Unexpected target flags on generic GlobalAddressSDNode\""
, "/build/llvm-toolchain-snapshot-13~++20210726100616+dead50d4427c/llvm/lib/Target/WebAssembly/WebAssemblyISelLowering.cpp"
, 1542, __extension__ __PRETTY_FUNCTION__))
;
1543 if (!WebAssembly::isValidAddressSpace(GA->getAddressSpace()))
1544 fail(DL, DAG, "Invalid address space for WebAssembly target");
1545
1546 unsigned OperandFlags = 0;
1547 if (isPositionIndependent()) {
1548 const GlobalValue *GV = GA->getGlobal();
1549 if (getTargetMachine().shouldAssumeDSOLocal(*GV->getParent(), GV)) {
1550 MachineFunction &MF = DAG.getMachineFunction();
1551 MVT PtrVT = getPointerTy(MF.getDataLayout());
1552 const char *BaseName;
1553 if (GV->getValueType()->isFunctionTy()) {
1554 BaseName = MF.createExternalSymbolName("__table_base");
1555 OperandFlags = WebAssemblyII::MO_TABLE_BASE_REL;
1556 }
1557 else {
1558 BaseName = MF.createExternalSymbolName("__memory_base");
1559 OperandFlags = WebAssemblyII::MO_MEMORY_BASE_REL;
1560 }
1561 SDValue BaseAddr =
1562 DAG.getNode(WebAssemblyISD::Wrapper, DL, PtrVT,
1563 DAG.getTargetExternalSymbol(BaseName, PtrVT));
1564
1565 SDValue SymAddr = DAG.getNode(
1566 WebAssemblyISD::WrapperPIC, DL, VT,
1567 DAG.getTargetGlobalAddress(GA->getGlobal(), DL, VT, GA->getOffset(),
1568 OperandFlags));
1569
1570 return DAG.getNode(ISD::ADD, DL, VT, BaseAddr, SymAddr);
1571 } else {
1572 OperandFlags = WebAssemblyII::MO_GOT;
1573 }
1574 }
1575
1576 return DAG.getNode(WebAssemblyISD::Wrapper, DL, VT,
1577 DAG.getTargetGlobalAddress(GA->getGlobal(), DL, VT,
1578 GA->getOffset(), OperandFlags));
1579}
1580
1581SDValue
1582WebAssemblyTargetLowering::LowerExternalSymbol(SDValue Op,
1583 SelectionDAG &DAG) const {
1584 SDLoc DL(Op);
1585 const auto *ES = cast<ExternalSymbolSDNode>(Op);
1586 EVT VT = Op.getValueType();
1587 assert(ES->getTargetFlags() == 0 &&(static_cast <bool> (ES->getTargetFlags() == 0 &&
"Unexpected target flags on generic ExternalSymbolSDNode") ?
void (0) : __assert_fail ("ES->getTargetFlags() == 0 && \"Unexpected target flags on generic ExternalSymbolSDNode\""
, "/build/llvm-toolchain-snapshot-13~++20210726100616+dead50d4427c/llvm/lib/Target/WebAssembly/WebAssemblyISelLowering.cpp"
, 1588, __extension__ __PRETTY_FUNCTION__))
1588 "Unexpected target flags on generic ExternalSymbolSDNode")(static_cast <bool> (ES->getTargetFlags() == 0 &&
"Unexpected target flags on generic ExternalSymbolSDNode") ?
void (0) : __assert_fail ("ES->getTargetFlags() == 0 && \"Unexpected target flags on generic ExternalSymbolSDNode\""
, "/build/llvm-toolchain-snapshot-13~++20210726100616+dead50d4427c/llvm/lib/Target/WebAssembly/WebAssemblyISelLowering.cpp"
, 1588, __extension__ __PRETTY_FUNCTION__))
;
1589 return DAG.getNode(WebAssemblyISD::Wrapper, DL, VT,
1590 DAG.getTargetExternalSymbol(ES->getSymbol(), VT));
1591}
1592
1593SDValue WebAssemblyTargetLowering::LowerJumpTable(SDValue Op,
1594 SelectionDAG &DAG) const {
1595 // There's no need for a Wrapper node because we always incorporate a jump
1596 // table operand into a BR_TABLE instruction, rather than ever
1597 // materializing it in a register.
1598 const JumpTableSDNode *JT = cast<JumpTableSDNode>(Op);
1599 return DAG.getTargetJumpTable(JT->getIndex(), Op.getValueType(),
1600 JT->getTargetFlags());
1601}
1602
1603SDValue WebAssemblyTargetLowering::LowerBR_JT(SDValue Op,
1604 SelectionDAG &DAG) const {
1605 SDLoc DL(Op);
1606 SDValue Chain = Op.getOperand(0);
1607 const auto *JT = cast<JumpTableSDNode>(Op.getOperand(1));
1608 SDValue Index = Op.getOperand(2);
1609 assert(JT->getTargetFlags() == 0 && "WebAssembly doesn't set target flags")(static_cast <bool> (JT->getTargetFlags() == 0 &&
"WebAssembly doesn't set target flags") ? void (0) : __assert_fail
("JT->getTargetFlags() == 0 && \"WebAssembly doesn't set target flags\""
, "/build/llvm-toolchain-snapshot-13~++20210726100616+dead50d4427c/llvm/lib/Target/WebAssembly/WebAssemblyISelLowering.cpp"
, 1609, __extension__ __PRETTY_FUNCTION__))
;
1610
1611 SmallVector<SDValue, 8> Ops;
1612 Ops.push_back(Chain);
1613 Ops.push_back(Index);
1614
1615 MachineJumpTableInfo *MJTI = DAG.getMachineFunction().getJumpTableInfo();
1616 const auto &MBBs = MJTI->getJumpTables()[JT->getIndex()].MBBs;
1617
1618 // Add an operand for each case.
1619 for (auto MBB : MBBs)
1620 Ops.push_back(DAG.getBasicBlock(MBB));
1621
1622 // Add the first MBB as a dummy default target for now. This will be replaced
1623 // with the proper default target (and the preceding range check eliminated)
1624 // if possible by WebAssemblyFixBrTableDefaults.
1625 Ops.push_back(DAG.getBasicBlock(*MBBs.begin()));
1626 return DAG.getNode(WebAssemblyISD::BR_TABLE, DL, MVT::Other, Ops);
1627}
1628
1629SDValue WebAssemblyTargetLowering::LowerVASTART(SDValue Op,
1630 SelectionDAG &DAG) const {
1631 SDLoc DL(Op);
1632 EVT PtrVT = getPointerTy(DAG.getMachineFunction().getDataLayout());
1633
1634 auto *MFI = DAG.getMachineFunction().getInfo<WebAssemblyFunctionInfo>();
1635 const Value *SV = cast<SrcValueSDNode>(Op.getOperand(2))->getValue();
1636
1637 SDValue ArgN = DAG.getCopyFromReg(DAG.getEntryNode(), DL,
1638 MFI->getVarargBufferVreg(), PtrVT);
1639 return DAG.getStore(Op.getOperand(0), DL, ArgN, Op.getOperand(1),
1640 MachinePointerInfo(SV));
1641}
1642
1643static SDValue getCppExceptionSymNode(SDValue Op, unsigned TagIndex,
1644 SelectionDAG &DAG) {
1645 // We only support C++ exceptions for now
1646 int Tag =
1647 cast<ConstantSDNode>(Op.getOperand(TagIndex).getNode())->getZExtValue();
1648 if (Tag != WebAssembly::CPP_EXCEPTION)
1649 llvm_unreachable("Invalid tag: We only support C++ exceptions for now")::llvm::llvm_unreachable_internal("Invalid tag: We only support C++ exceptions for now"
, "/build/llvm-toolchain-snapshot-13~++20210726100616+dead50d4427c/llvm/lib/Target/WebAssembly/WebAssemblyISelLowering.cpp"
, 1649)
;
1650 auto &MF = DAG.getMachineFunction();
1651 const auto &TLI = DAG.getTargetLoweringInfo();
1652 MVT PtrVT = TLI.getPointerTy(DAG.getDataLayout());
1653 const char *SymName = MF.createExternalSymbolName("__cpp_exception");
1654 return DAG.getNode(WebAssemblyISD::Wrapper, SDLoc(Op), PtrVT,
1655 DAG.getTargetExternalSymbol(SymName, PtrVT));
1656}
1657
1658SDValue WebAssemblyTargetLowering::LowerIntrinsic(SDValue Op,
1659 SelectionDAG &DAG) const {
1660 MachineFunction &MF = DAG.getMachineFunction();
1661 unsigned IntNo;
1662 switch (Op.getOpcode()) {
1663 case ISD::INTRINSIC_VOID:
1664 case ISD::INTRINSIC_W_CHAIN:
1665 IntNo = cast<ConstantSDNode>(Op.getOperand(1))->getZExtValue();
1666 break;
1667 case ISD::INTRINSIC_WO_CHAIN:
1668 IntNo = cast<ConstantSDNode>(Op.getOperand(0))->getZExtValue();
1669 break;
1670 default:
1671 llvm_unreachable("Invalid intrinsic")::llvm::llvm_unreachable_internal("Invalid intrinsic", "/build/llvm-toolchain-snapshot-13~++20210726100616+dead50d4427c/llvm/lib/Target/WebAssembly/WebAssemblyISelLowering.cpp"
, 1671)
;
1672 }
1673 SDLoc DL(Op);
1674
1675 switch (IntNo) {
1676 default:
1677 return SDValue(); // Don't custom lower most intrinsics.
1678
1679 case Intrinsic::wasm_lsda: {
1680 EVT VT = Op.getValueType();
1681 const TargetLowering &TLI = DAG.getTargetLoweringInfo();
1682 MVT PtrVT = TLI.getPointerTy(DAG.getDataLayout());
1683 auto &Context = MF.getMMI().getContext();
1684 MCSymbol *S = Context.getOrCreateSymbol(Twine("GCC_except_table") +
1685 Twine(MF.getFunctionNumber()));
1686 return DAG.getNode(WebAssemblyISD::Wrapper, DL, VT,
1687 DAG.getMCSymbol(S, PtrVT));
1688 }
1689
1690 case Intrinsic::wasm_throw: {
1691 SDValue SymNode = getCppExceptionSymNode(Op, 2, DAG);
1692 return DAG.getNode(WebAssemblyISD::THROW, DL,
1693 MVT::Other, // outchain type
1694 {
1695 Op.getOperand(0), // inchain
1696 SymNode, // exception symbol
1697 Op.getOperand(3) // thrown value
1698 });
1699 }
1700
1701 case Intrinsic::wasm_catch: {
1702 SDValue SymNode = getCppExceptionSymNode(Op, 2, DAG);
1703 return DAG.getNode(WebAssemblyISD::CATCH, DL,
1704 {
1705 MVT::i32, // outchain type
1706 MVT::Other // return value
1707 },
1708 {
1709 Op.getOperand(0), // inchain
1710 SymNode // exception symbol
1711 });
1712 }
1713
1714 case Intrinsic::wasm_shuffle: {
1715 // Drop in-chain and replace undefs, but otherwise pass through unchanged
1716 SDValue Ops[18];
1717 size_t OpIdx = 0;
1718 Ops[OpIdx++] = Op.getOperand(1);
1719 Ops[OpIdx++] = Op.getOperand(2);
1720 while (OpIdx < 18) {
1721 const SDValue &MaskIdx = Op.getOperand(OpIdx + 1);
1722 if (MaskIdx.isUndef() ||
1723 cast<ConstantSDNode>(MaskIdx.getNode())->getZExtValue() >= 32) {
1724 Ops[OpIdx++] = DAG.getConstant(0, DL, MVT::i32);
1725 } else {
1726 Ops[OpIdx++] = MaskIdx;
1727 }
1728 }
1729 return DAG.getNode(WebAssemblyISD::SHUFFLE, DL, Op.getValueType(), Ops);
1730 }
1731 }
1732}
1733
1734SDValue
1735WebAssemblyTargetLowering::LowerSIGN_EXTEND_INREG(SDValue Op,
1736 SelectionDAG &DAG) const {
1737 SDLoc DL(Op);
1738 // If sign extension operations are disabled, allow sext_inreg only if operand
1739 // is a vector extract of an i8 or i16 lane. SIMD does not depend on sign
1740 // extension operations, but allowing sext_inreg in this context lets us have
1741 // simple patterns to select extract_lane_s instructions. Expanding sext_inreg
1742 // everywhere would be simpler in this file, but would necessitate large and
1743 // brittle patterns to undo the expansion and select extract_lane_s
1744 // instructions.
1745 assert(!Subtarget->hasSignExt() && Subtarget->hasSIMD128())(static_cast <bool> (!Subtarget->hasSignExt() &&
Subtarget->hasSIMD128()) ? void (0) : __assert_fail ("!Subtarget->hasSignExt() && Subtarget->hasSIMD128()"
, "/build/llvm-toolchain-snapshot-13~++20210726100616+dead50d4427c/llvm/lib/Target/WebAssembly/WebAssemblyISelLowering.cpp"
, 1745, __extension__ __PRETTY_FUNCTION__))
;
1746 if (Op.getOperand(0).getOpcode() != ISD::EXTRACT_VECTOR_ELT)
1747 return SDValue();
1748
1749 const SDValue &Extract = Op.getOperand(0);
1750 MVT VecT = Extract.getOperand(0).getSimpleValueType();
1751 if (VecT.getVectorElementType().getSizeInBits() > 32)
1752 return SDValue();
1753 MVT ExtractedLaneT =
1754 cast<VTSDNode>(Op.getOperand(1).getNode())->getVT().getSimpleVT();
1755 MVT ExtractedVecT =
1756 MVT::getVectorVT(ExtractedLaneT, 128 / ExtractedLaneT.getSizeInBits());
1757 if (ExtractedVecT == VecT)
1758 return Op;
1759
1760 // Bitcast vector to appropriate type to ensure ISel pattern coverage
1761 const SDNode *Index = Extract.getOperand(1).getNode();
1762 if (!isa<ConstantSDNode>(Index))
1763 return SDValue();
1764 unsigned IndexVal = cast<ConstantSDNode>(Index)->getZExtValue();
1765 unsigned Scale =
1766 ExtractedVecT.getVectorNumElements() / VecT.getVectorNumElements();
1767 assert(Scale > 1)(static_cast <bool> (Scale > 1) ? void (0) : __assert_fail
("Scale > 1", "/build/llvm-toolchain-snapshot-13~++20210726100616+dead50d4427c/llvm/lib/Target/WebAssembly/WebAssemblyISelLowering.cpp"
, 1767, __extension__ __PRETTY_FUNCTION__))
;
1768 SDValue NewIndex =
1769 DAG.getConstant(IndexVal * Scale, DL, Index->getValueType(0));
1770 SDValue NewExtract = DAG.getNode(
1771 ISD::EXTRACT_VECTOR_ELT, DL, Extract.getValueType(),
1772 DAG.getBitcast(ExtractedVecT, Extract.getOperand(0)), NewIndex);
1773 return DAG.getNode(ISD::SIGN_EXTEND_INREG, DL, Op.getValueType(), NewExtract,
1774 Op.getOperand(1));
1775}
1776
1777SDValue WebAssemblyTargetLowering::LowerBUILD_VECTOR(SDValue Op,
1778 SelectionDAG &DAG) const {
1779 SDLoc DL(Op);
1780 const EVT VecT = Op.getValueType();
1781 const EVT LaneT = Op.getOperand(0).getValueType();
1782 const size_t Lanes = Op.getNumOperands();
1783 bool CanSwizzle = VecT == MVT::v16i8;
1784
1785 // BUILD_VECTORs are lowered to the instruction that initializes the highest
1786 // possible number of lanes at once followed by a sequence of replace_lane
1787 // instructions to individually initialize any remaining lanes.
1788
1789 // TODO: Tune this. For example, lanewise swizzling is very expensive, so
1790 // swizzled lanes should be given greater weight.
1791
1792 // TODO: Investigate looping rather than always extracting/replacing specific
1793 // lanes to fill gaps.
1794
1795 auto IsConstant = [](const SDValue &V) {
1796 return V.getOpcode() == ISD::Constant || V.getOpcode() == ISD::ConstantFP;
1797 };
1798
1799 // Returns the source vector and index vector pair if they exist. Checks for:
1800 // (extract_vector_elt
1801 // $src,
1802 // (sign_extend_inreg (extract_vector_elt $indices, $i))
1803 // )
1804 auto GetSwizzleSrcs = [](size_t I, const SDValue &Lane) {
1805 auto Bail = std::make_pair(SDValue(), SDValue());
1806 if (Lane->getOpcode() != ISD::EXTRACT_VECTOR_ELT)
1807 return Bail;
1808 const SDValue &SwizzleSrc = Lane->getOperand(0);
1809 const SDValue &IndexExt = Lane->getOperand(1);
1810 if (IndexExt->getOpcode() != ISD::SIGN_EXTEND_INREG)
1811 return Bail;
1812 const SDValue &Index = IndexExt->getOperand(0);
1813 if (Index->getOpcode() != ISD::EXTRACT_VECTOR_ELT)
1814 return Bail;
1815 const SDValue &SwizzleIndices = Index->getOperand(0);
1816 if (SwizzleSrc.getValueType() != MVT::v16i8 ||
1817 SwizzleIndices.getValueType() != MVT::v16i8 ||
1818 Index->getOperand(1)->getOpcode() != ISD::Constant ||
1819 Index->getConstantOperandVal(1) != I)
1820 return Bail;
1821 return std::make_pair(SwizzleSrc, SwizzleIndices);
1822 };
1823
1824 // If the lane is extracted from another vector at a constant index, return
1825 // that vector. The source vector must not have more lanes than the dest
1826 // because the shufflevector indices are in terms of the destination lanes and
1827 // would not be able to address the smaller individual source lanes.
1828 auto GetShuffleSrc = [&](const SDValue &Lane) {
1829 if (Lane->getOpcode() != ISD::EXTRACT_VECTOR_ELT)
1830 return SDValue();
1831 if (!isa<ConstantSDNode>(Lane->getOperand(1).getNode()))
1832 return SDValue();
1833 if (Lane->getOperand(0).getValueType().getVectorNumElements() >
1834 VecT.getVectorNumElements())
1835 return SDValue();
1836 return Lane->getOperand(0);
1837 };
1838
1839 using ValueEntry = std::pair<SDValue, size_t>;
1840 SmallVector<ValueEntry, 16> SplatValueCounts;
1841
1842 using SwizzleEntry = std::pair<std::pair<SDValue, SDValue>, size_t>;
1843 SmallVector<SwizzleEntry, 16> SwizzleCounts;
1844
1845 using ShuffleEntry = std::pair<SDValue, size_t>;
1846 SmallVector<ShuffleEntry, 16> ShuffleCounts;
1847
1848 auto AddCount = [](auto &Counts, const auto &Val) {
1849 auto CountIt =
1850 llvm::find_if(Counts, [&Val](auto E) { return E.first == Val; });
1851 if (CountIt == Counts.end()) {
1852 Counts.emplace_back(Val, 1);
1853 } else {
1854 CountIt->second++;
1855 }
1856 };
1857
1858 auto GetMostCommon = [](auto &Counts) {
1859 auto CommonIt =
1860 std::max_element(Counts.begin(), Counts.end(),
1861 [](auto A, auto B) { return A.second < B.second; });
1862 assert(CommonIt != Counts.end() && "Unexpected all-undef build_vector")(static_cast <bool> (CommonIt != Counts.end() &&
"Unexpected all-undef build_vector") ? void (0) : __assert_fail
("CommonIt != Counts.end() && \"Unexpected all-undef build_vector\""
, "/build/llvm-toolchain-snapshot-13~++20210726100616+dead50d4427c/llvm/lib/Target/WebAssembly/WebAssemblyISelLowering.cpp"
, 1862, __extension__ __PRETTY_FUNCTION__))
;
1863 return *CommonIt;
1864 };
1865
1866 size_t NumConstantLanes = 0;
1867
1868 // Count eligible lanes for each type of vector creation op
1869 for (size_t I = 0; I
2.1
'I' is < 'Lanes'
2.1
'I' is < 'Lanes'
< Lanes
; ++I) {
3
Loop condition is true. Entering loop body
9
Assuming 'I' is >= 'Lanes'
10
Loop condition is false. Execution continues on line 1887
1870 const SDValue &Lane = Op->getOperand(I);
1871 if (Lane.isUndef())
4
Taking false branch
1872 continue;
1873
1874 AddCount(SplatValueCounts, Lane);
1875
1876 if (IsConstant(Lane))
5
Taking false branch
1877 NumConstantLanes++;
1878 if (auto ShuffleSrc = GetShuffleSrc(Lane))
6
Taking false branch
1879 AddCount(ShuffleCounts, ShuffleSrc);
1880 if (CanSwizzle
6.1
'CanSwizzle' is true
6.1
'CanSwizzle' is true
) {
7
Taking true branch
1881 auto SwizzleSrcs = GetSwizzleSrcs(I, Lane);
1882 if (SwizzleSrcs.first)
8
Taking false branch
1883 AddCount(SwizzleCounts, SwizzleSrcs);
1884 }
1885 }
1886
1887 SDValue SplatValue;
1888 size_t NumSplatLanes;
1889 std::tie(SplatValue, NumSplatLanes) = GetMostCommon(SplatValueCounts);
1890
1891 SDValue SwizzleSrc;
1892 SDValue SwizzleIndices;
1893 size_t NumSwizzleLanes = 0;
1894 if (SwizzleCounts.size())
11
Assuming the condition is true
12
Taking true branch
1895 std::forward_as_tuple(std::tie(SwizzleSrc, SwizzleIndices),
1896 NumSwizzleLanes) = GetMostCommon(SwizzleCounts);
1897
1898 // Shuffles can draw from up to two vectors, so find the two most common
1899 // sources.
1900 SDValue ShuffleSrc1, ShuffleSrc2;
1901 size_t NumShuffleLanes = 0;
1902 if (ShuffleCounts.size()) {
13
Assuming the condition is false
14
Taking false branch
1903 std::tie(ShuffleSrc1, NumShuffleLanes) = GetMostCommon(ShuffleCounts);
1904 ShuffleCounts.erase(std::remove_if(ShuffleCounts.begin(),
1905 ShuffleCounts.end(),
1906 [&](const auto &Pair) {
1907 return Pair.first == ShuffleSrc1;
1908 }),
1909 ShuffleCounts.end());
1910 }
1911 if (ShuffleCounts.size()) {
15
Taking false branch
1912 size_t AdditionalShuffleLanes;
1913 std::tie(ShuffleSrc2, AdditionalShuffleLanes) =
1914 GetMostCommon(ShuffleCounts);
1915 NumShuffleLanes += AdditionalShuffleLanes;
1916 }
1917
1918 // Predicate returning true if the lane is properly initialized by the
1919 // original instruction
1920 std::function<bool(size_t, const SDValue &)> IsLaneConstructed;
1921 SDValue Result;
1922 // Prefer swizzles over shuffles over vector consts over splats
1923 if (NumSwizzleLanes
15.1
'NumSwizzleLanes' is >= 'NumShuffleLanes'
15.1
'NumSwizzleLanes' is >= 'NumShuffleLanes'
>= NumShuffleLanes &&
17
Taking false branch
1924 NumSwizzleLanes
15.2
'NumSwizzleLanes' is >= 'NumConstantLanes'
15.2
'NumSwizzleLanes' is >= 'NumConstantLanes'
>= NumConstantLanes && NumSwizzleLanes >= NumSplatLanes) {
16
Assuming 'NumSwizzleLanes' is < 'NumSplatLanes'
1925 Result = DAG.getNode(WebAssemblyISD::SWIZZLE, DL, VecT, SwizzleSrc,
1926 SwizzleIndices);
1927 auto Swizzled = std::make_pair(SwizzleSrc, SwizzleIndices);
1928 IsLaneConstructed = [&, Swizzled](size_t I, const SDValue &Lane) {
1929 return Swizzled == GetSwizzleSrcs(I, Lane);
1930 };
1931 } else if (NumShuffleLanes
17.1
'NumShuffleLanes' is >= 'NumConstantLanes'
17.1
'NumShuffleLanes' is >= 'NumConstantLanes'
>= NumConstantLanes &&
19
Taking true branch
1932 NumShuffleLanes >= NumSplatLanes) {
18
Assuming 'NumShuffleLanes' is >= 'NumSplatLanes'
1933 size_t DestLaneSize = VecT.getVectorElementType().getFixedSizeInBits() / 8;
1934 size_t DestLaneCount = VecT.getVectorNumElements();
1935 size_t Scale1 = 1;
1936 size_t Scale2 = 1;
1937 SDValue Src1 = ShuffleSrc1;
20
Null pointer value stored to 'Src1.Node'
1938 SDValue Src2 = ShuffleSrc2 ? ShuffleSrc2 : DAG.getUNDEF(VecT);
21
'?' condition is false
1939 if (Src1.getValueType() != VecT) {
22
Calling 'SDValue::getValueType'
1940 size_t LaneSize =
1941 Src1.getValueType().getVectorElementType().getFixedSizeInBits() / 8;
1942 assert(LaneSize > DestLaneSize)(static_cast <bool> (LaneSize > DestLaneSize) ? void
(0) : __assert_fail ("LaneSize > DestLaneSize", "/build/llvm-toolchain-snapshot-13~++20210726100616+dead50d4427c/llvm/lib/Target/WebAssembly/WebAssemblyISelLowering.cpp"
, 1942, __extension__ __PRETTY_FUNCTION__))
;
1943 Scale1 = LaneSize / DestLaneSize;
1944 Src1 = DAG.getBitcast(VecT, Src1);
1945 }
1946 if (Src2.getValueType() != VecT) {
1947 size_t LaneSize =
1948 Src2.getValueType().getVectorElementType().getFixedSizeInBits() / 8;
1949 assert(LaneSize > DestLaneSize)(static_cast <bool> (LaneSize > DestLaneSize) ? void
(0) : __assert_fail ("LaneSize > DestLaneSize", "/build/llvm-toolchain-snapshot-13~++20210726100616+dead50d4427c/llvm/lib/Target/WebAssembly/WebAssemblyISelLowering.cpp"
, 1949, __extension__ __PRETTY_FUNCTION__))
;
1950 Scale2 = LaneSize / DestLaneSize;
1951 Src2 = DAG.getBitcast(VecT, Src2);
1952 }
1953
1954 int Mask[16];
1955 assert(DestLaneCount <= 16)(static_cast <bool> (DestLaneCount <= 16) ? void (0)
: __assert_fail ("DestLaneCount <= 16", "/build/llvm-toolchain-snapshot-13~++20210726100616+dead50d4427c/llvm/lib/Target/WebAssembly/WebAssemblyISelLowering.cpp"
, 1955, __extension__ __PRETTY_FUNCTION__))
;
1956 for (size_t I = 0; I < DestLaneCount; ++I) {
1957 const SDValue &Lane = Op->getOperand(I);
1958 SDValue Src = GetShuffleSrc(Lane);
1959 if (Src == ShuffleSrc1) {
1960 Mask[I] = Lane->getConstantOperandVal(1) * Scale1;
1961 } else if (Src && Src == ShuffleSrc2) {
1962 Mask[I] = DestLaneCount + Lane->getConstantOperandVal(1) * Scale2;
1963 } else {
1964 Mask[I] = -1;
1965 }
1966 }
1967 ArrayRef<int> MaskRef(Mask, DestLaneCount);
1968 Result = DAG.getVectorShuffle(VecT, DL, Src1, Src2, MaskRef);
1969 IsLaneConstructed = [&](size_t, const SDValue &Lane) {
1970 auto Src = GetShuffleSrc(Lane);
1971 return Src == ShuffleSrc1 || (Src && Src == ShuffleSrc2);
1972 };
1973 } else if (NumConstantLanes >= NumSplatLanes) {
1974 SmallVector<SDValue, 16> ConstLanes;
1975 for (const SDValue &Lane : Op->op_values()) {
1976 if (IsConstant(Lane)) {
1977 ConstLanes.push_back(Lane);
1978 } else if (LaneT.isFloatingPoint()) {
1979 ConstLanes.push_back(DAG.getConstantFP(0, DL, LaneT));
1980 } else {
1981 ConstLanes.push_back(DAG.getConstant(0, DL, LaneT));
1982 }
1983 }
1984 Result = DAG.getBuildVector(VecT, DL, ConstLanes);
1985 IsLaneConstructed = [&IsConstant](size_t _, const SDValue &Lane) {
1986 return IsConstant(Lane);
1987 };
1988 } else {
1989 // Use a splat, but possibly a load_splat
1990 LoadSDNode *SplattedLoad;
1991 if ((SplattedLoad = dyn_cast<LoadSDNode>(SplatValue)) &&
1992 SplattedLoad->getMemoryVT() == VecT.getVectorElementType()) {
1993 Result = DAG.getMemIntrinsicNode(
1994 WebAssemblyISD::LOAD_SPLAT, DL, DAG.getVTList(VecT),
1995 {SplattedLoad->getChain(), SplattedLoad->getBasePtr(),
1996 SplattedLoad->getOffset()},
1997 SplattedLoad->getMemoryVT(), SplattedLoad->getMemOperand());
1998 } else {
1999 Result = DAG.getSplatBuildVector(VecT, DL, SplatValue);
2000 }
2001 IsLaneConstructed = [&SplatValue](size_t _, const SDValue &Lane) {
2002 return Lane == SplatValue;
2003 };
2004 }
2005
2006 assert(Result)(static_cast <bool> (Result) ? void (0) : __assert_fail
("Result", "/build/llvm-toolchain-snapshot-13~++20210726100616+dead50d4427c/llvm/lib/Target/WebAssembly/WebAssemblyISelLowering.cpp"
, 2006, __extension__ __PRETTY_FUNCTION__))
;
2007 assert(IsLaneConstructed)(static_cast <bool> (IsLaneConstructed) ? void (0) : __assert_fail
("IsLaneConstructed", "/build/llvm-toolchain-snapshot-13~++20210726100616+dead50d4427c/llvm/lib/Target/WebAssembly/WebAssemblyISelLowering.cpp"
, 2007, __extension__ __PRETTY_FUNCTION__))
;
2008
2009 // Add replace_lane instructions for any unhandled values
2010 for (size_t I = 0; I < Lanes; ++I) {
2011 const SDValue &Lane = Op->getOperand(I);
2012 if (!Lane.isUndef() && !IsLaneConstructed(I, Lane))
2013 Result = DAG.getNode(ISD::INSERT_VECTOR_ELT, DL, VecT, Result, Lane,
2014 DAG.getConstant(I, DL, MVT::i32));
2015 }
2016
2017 return Result;
2018}
2019
2020SDValue
2021WebAssemblyTargetLowering::LowerVECTOR_SHUFFLE(SDValue Op,
2022 SelectionDAG &DAG) const {
2023 SDLoc DL(Op);
2024 ArrayRef<int> Mask = cast<ShuffleVectorSDNode>(Op.getNode())->getMask();
2025 MVT VecType = Op.getOperand(0).getSimpleValueType();
2026 assert(VecType.is128BitVector() && "Unexpected shuffle vector type")(static_cast <bool> (VecType.is128BitVector() &&
"Unexpected shuffle vector type") ? void (0) : __assert_fail
("VecType.is128BitVector() && \"Unexpected shuffle vector type\""
, "/build/llvm-toolchain-snapshot-13~++20210726100616+dead50d4427c/llvm/lib/Target/WebAssembly/WebAssemblyISelLowering.cpp"
, 2026, __extension__ __PRETTY_FUNCTION__))
;
2027 size_t LaneBytes = VecType.getVectorElementType().getSizeInBits() / 8;
2028
2029 // Space for two vector args and sixteen mask indices
2030 SDValue Ops[18];
2031 size_t OpIdx = 0;
2032 Ops[OpIdx++] = Op.getOperand(0);
2033 Ops[OpIdx++] = Op.getOperand(1);
2034
2035 // Expand mask indices to byte indices and materialize them as operands
2036 for (int M : Mask) {
2037 for (size_t J = 0; J < LaneBytes; ++J) {
2038 // Lower undefs (represented by -1 in mask) to zero
2039 uint64_t ByteIndex = M == -1 ? 0 : (uint64_t)M * LaneBytes + J;
2040 Ops[OpIdx++] = DAG.getConstant(ByteIndex, DL, MVT::i32);
2041 }
2042 }
2043
2044 return DAG.getNode(WebAssemblyISD::SHUFFLE, DL, Op.getValueType(), Ops);
2045}
2046
2047SDValue WebAssemblyTargetLowering::LowerSETCC(SDValue Op,
2048 SelectionDAG &DAG) const {
2049 SDLoc DL(Op);
2050 // The legalizer does not know how to expand the unsupported comparison modes
2051 // of i64x2 vectors, so we manually unroll them here.
2052 assert(Op->getOperand(0)->getSimpleValueType(0) == MVT::v2i64)(static_cast <bool> (Op->getOperand(0)->getSimpleValueType
(0) == MVT::v2i64) ? void (0) : __assert_fail ("Op->getOperand(0)->getSimpleValueType(0) == MVT::v2i64"
, "/build/llvm-toolchain-snapshot-13~++20210726100616+dead50d4427c/llvm/lib/Target/WebAssembly/WebAssemblyISelLowering.cpp"
, 2052, __extension__ __PRETTY_FUNCTION__))
;
2053 SmallVector<SDValue, 2> LHS, RHS;
2054 DAG.ExtractVectorElements(Op->getOperand(0), LHS);
2055 DAG.ExtractVectorElements(Op->getOperand(1), RHS);
2056 const SDValue &CC = Op->getOperand(2);
2057 auto MakeLane = [&](unsigned I) {
2058 return DAG.getNode(ISD::SELECT_CC, DL, MVT::i64, LHS[I], RHS[I],
2059 DAG.getConstant(uint64_t(-1), DL, MVT::i64),
2060 DAG.getConstant(uint64_t(0), DL, MVT::i64), CC);
2061 };
2062 return DAG.getBuildVector(Op->getValueType(0), DL,
2063 {MakeLane(0), MakeLane(1)});
2064}
2065
2066SDValue
2067WebAssemblyTargetLowering::LowerAccessVectorElement(SDValue Op,
2068 SelectionDAG &DAG) const {
2069 // Allow constant lane indices, expand variable lane indices
2070 SDNode *IdxNode = Op.getOperand(Op.getNumOperands() - 1).getNode();
2071 if (isa<ConstantSDNode>(IdxNode) || IdxNode->isUndef())
2072 return Op;
2073 else
2074 // Perform default expansion
2075 return SDValue();
2076}
2077
2078static SDValue unrollVectorShift(SDValue Op, SelectionDAG &DAG) {
2079 EVT LaneT = Op.getSimpleValueType().getVectorElementType();
2080 // 32-bit and 64-bit unrolled shifts will have proper semantics
2081 if (LaneT.bitsGE(MVT::i32))
2082 return DAG.UnrollVectorOp(Op.getNode());
2083 // Otherwise mask the shift value to get proper semantics from 32-bit shift
2084 SDLoc DL(Op);
2085 size_t NumLanes = Op.getSimpleValueType().getVectorNumElements();
2086 SDValue Mask = DAG.getConstant(LaneT.getSizeInBits() - 1, DL, MVT::i32);
2087 unsigned ShiftOpcode = Op.getOpcode();
2088 SmallVector<SDValue, 16> ShiftedElements;
2089 DAG.ExtractVectorElements(Op.getOperand(0), ShiftedElements, 0, 0, MVT::i32);
2090 SmallVector<SDValue, 16> ShiftElements;
2091 DAG.ExtractVectorElements(Op.getOperand(1), ShiftElements, 0, 0, MVT::i32);
2092 SmallVector<SDValue, 16> UnrolledOps;
2093 for (size_t i = 0; i < NumLanes; ++i) {
2094 SDValue MaskedShiftValue =
2095 DAG.getNode(ISD::AND, DL, MVT::i32, ShiftElements[i], Mask);
2096 SDValue ShiftedValue = ShiftedElements[i];
2097 if (ShiftOpcode == ISD::SRA)
2098 ShiftedValue = DAG.getNode(ISD::SIGN_EXTEND_INREG, DL, MVT::i32,
2099 ShiftedValue, DAG.getValueType(LaneT));
2100 UnrolledOps.push_back(
2101 DAG.getNode(ShiftOpcode, DL, MVT::i32, ShiftedValue, MaskedShiftValue));
2102 }
2103 return DAG.getBuildVector(Op.getValueType(), DL, UnrolledOps);
2104}
2105
2106SDValue WebAssemblyTargetLowering::LowerShift(SDValue Op,
2107 SelectionDAG &DAG) const {
2108 SDLoc DL(Op);
2109
2110 // Only manually lower vector shifts
2111 assert(Op.getSimpleValueType().isVector())(static_cast <bool> (Op.getSimpleValueType().isVector()
) ? void (0) : __assert_fail ("Op.getSimpleValueType().isVector()"
, "/build/llvm-toolchain-snapshot-13~++20210726100616+dead50d4427c/llvm/lib/Target/WebAssembly/WebAssemblyISelLowering.cpp"
, 2111, __extension__ __PRETTY_FUNCTION__))
;
2112
2113 auto ShiftVal = DAG.getSplatValue(Op.getOperand(1));
2114 if (!ShiftVal)
2115 return unrollVectorShift(Op, DAG);
2116
2117 // Use anyext because none of the high bits can affect the shift
2118 ShiftVal = DAG.getAnyExtOrTrunc(ShiftVal, DL, MVT::i32);
2119
2120 unsigned Opcode;
2121 switch (Op.getOpcode()) {
2122 case ISD::SHL:
2123 Opcode = WebAssemblyISD::VEC_SHL;
2124 break;
2125 case ISD::SRA:
2126 Opcode = WebAssemblyISD::VEC_SHR_S;
2127 break;
2128 case ISD::SRL:
2129 Opcode = WebAssemblyISD::VEC_SHR_U;
2130 break;
2131 default:
2132 llvm_unreachable("unexpected opcode")::llvm::llvm_unreachable_internal("unexpected opcode", "/build/llvm-toolchain-snapshot-13~++20210726100616+dead50d4427c/llvm/lib/Target/WebAssembly/WebAssemblyISelLowering.cpp"
, 2132)
;
2133 }
2134
2135 return DAG.getNode(Opcode, DL, Op.getValueType(), Op.getOperand(0), ShiftVal);
2136}
2137
2138SDValue WebAssemblyTargetLowering::LowerFP_TO_INT_SAT(SDValue Op,
2139 SelectionDAG &DAG) const {
2140 SDLoc DL(Op);
2141 EVT ResT = Op.getValueType();
2142 EVT SatVT = cast<VTSDNode>(Op.getOperand(1))->getVT();
2143
2144 if ((ResT == MVT::i32 || ResT == MVT::i64) &&
2145 (SatVT == MVT::i32 || SatVT == MVT::i64))
2146 return Op;
2147
2148 if (ResT == MVT::v4i32 && SatVT == MVT::i32)
2149 return Op;
2150
2151 return SDValue();
2152}
2153
2154//===----------------------------------------------------------------------===//
2155// Custom DAG combine hooks
2156//===----------------------------------------------------------------------===//
2157static SDValue
2158performVECTOR_SHUFFLECombine(SDNode *N, TargetLowering::DAGCombinerInfo &DCI) {
2159 auto &DAG = DCI.DAG;
2160 auto Shuffle = cast<ShuffleVectorSDNode>(N);
2161
2162 // Hoist vector bitcasts that don't change the number of lanes out of unary
2163 // shuffles, where they are less likely to get in the way of other combines.
2164 // (shuffle (vNxT1 (bitcast (vNxT0 x))), undef, mask) ->
2165 // (vNxT1 (bitcast (vNxT0 (shuffle x, undef, mask))))
2166 SDValue Bitcast = N->getOperand(0);
2167 if (Bitcast.getOpcode() != ISD::BITCAST)
2168 return SDValue();
2169 if (!N->getOperand(1).isUndef())
2170 return SDValue();
2171 SDValue CastOp = Bitcast.getOperand(0);
2172 MVT SrcType = CastOp.getSimpleValueType();
2173 MVT DstType = Bitcast.getSimpleValueType();
2174 if (!SrcType.is128BitVector() ||
2175 SrcType.getVectorNumElements() != DstType.getVectorNumElements())
2176 return SDValue();
2177 SDValue NewShuffle = DAG.getVectorShuffle(
2178 SrcType, SDLoc(N), CastOp, DAG.getUNDEF(SrcType), Shuffle->getMask());
2179 return DAG.getBitcast(DstType, NewShuffle);
2180}
2181
2182static SDValue
2183performVectorExtendCombine(SDNode *N, TargetLowering::DAGCombinerInfo &DCI) {
2184 auto &DAG = DCI.DAG;
2185 assert(N->getOpcode() == ISD::SIGN_EXTEND ||(static_cast <bool> (N->getOpcode() == ISD::SIGN_EXTEND
|| N->getOpcode() == ISD::ZERO_EXTEND) ? void (0) : __assert_fail
("N->getOpcode() == ISD::SIGN_EXTEND || N->getOpcode() == ISD::ZERO_EXTEND"
, "/build/llvm-toolchain-snapshot-13~++20210726100616+dead50d4427c/llvm/lib/Target/WebAssembly/WebAssemblyISelLowering.cpp"
, 2186, __extension__ __PRETTY_FUNCTION__))
2186 N->getOpcode() == ISD::ZERO_EXTEND)(static_cast <bool> (N->getOpcode() == ISD::SIGN_EXTEND
|| N->getOpcode() == ISD::ZERO_EXTEND) ? void (0) : __assert_fail
("N->getOpcode() == ISD::SIGN_EXTEND || N->getOpcode() == ISD::ZERO_EXTEND"
, "/build/llvm-toolchain-snapshot-13~++20210726100616+dead50d4427c/llvm/lib/Target/WebAssembly/WebAssemblyISelLowering.cpp"
, 2186, __extension__ __PRETTY_FUNCTION__))
;
2187
2188 // Combine ({s,z}ext (extract_subvector src, i)) into a widening operation if
2189 // possible before the extract_subvector can be expanded.
2190 auto Extract = N->getOperand(0);
2191 if (Extract.getOpcode() != ISD::EXTRACT_SUBVECTOR)
2192 return SDValue();
2193 auto Source = Extract.getOperand(0);
2194 auto *IndexNode = dyn_cast<ConstantSDNode>(Extract.getOperand(1));
2195 if (IndexNode == nullptr)
2196 return SDValue();
2197 auto Index = IndexNode->getZExtValue();
2198
2199 // Only v8i8, v4i16, and v2i32 extracts can be widened, and only if the
2200 // extracted subvector is the low or high half of its source.
2201 EVT ResVT = N->getValueType(0);
2202 if (ResVT == MVT::v8i16) {
2203 if (Extract.getValueType() != MVT::v8i8 ||
2204 Source.getValueType() != MVT::v16i8 || (Index != 0 && Index != 8))
2205 return SDValue();
2206 } else if (ResVT == MVT::v4i32) {
2207 if (Extract.getValueType() != MVT::v4i16 ||
2208 Source.getValueType() != MVT::v8i16 || (Index != 0 && Index != 4))
2209 return SDValue();
2210 } else if (ResVT == MVT::v2i64) {
2211 if (Extract.getValueType() != MVT::v2i32 ||
2212 Source.getValueType() != MVT::v4i32 || (Index != 0 && Index != 2))
2213 return SDValue();
2214 } else {
2215 return SDValue();
2216 }
2217
2218 bool IsSext = N->getOpcode() == ISD::SIGN_EXTEND;
2219 bool IsLow = Index == 0;
2220
2221 unsigned Op = IsSext ? (IsLow ? WebAssemblyISD::EXTEND_LOW_S
2222 : WebAssemblyISD::EXTEND_HIGH_S)
2223 : (IsLow ? WebAssemblyISD::EXTEND_LOW_U
2224 : WebAssemblyISD::EXTEND_HIGH_U);
2225
2226 return DAG.getNode(Op, SDLoc(N), ResVT, Source);
2227}
2228
2229static SDValue
2230performVectorConvertLowCombine(SDNode *N,
2231 TargetLowering::DAGCombinerInfo &DCI) {
2232 auto &DAG = DCI.DAG;
2233
2234 EVT ResVT = N->getValueType(0);
2235 if (ResVT != MVT::v2f64)
2236 return SDValue();
2237
2238 auto GetWasmConversionOp = [](unsigned Op) {
2239 switch (Op) {
2240 case ISD::SINT_TO_FP:
2241 return WebAssemblyISD::CONVERT_LOW_S;
2242 case ISD::UINT_TO_FP:
2243 return WebAssemblyISD::CONVERT_LOW_U;
2244 case ISD::FP_EXTEND:
2245 return WebAssemblyISD::PROMOTE_LOW;
2246 }
2247 llvm_unreachable("unexpected op")::llvm::llvm_unreachable_internal("unexpected op", "/build/llvm-toolchain-snapshot-13~++20210726100616+dead50d4427c/llvm/lib/Target/WebAssembly/WebAssemblyISelLowering.cpp"
, 2247)
;
2248 };
2249
2250 if (N->getOpcode() == ISD::EXTRACT_SUBVECTOR) {
2251 // Combine this:
2252 //
2253 // (v2f64 (extract_subvector
2254 // (v4f64 ({s,u}int_to_fp (v4i32 $x))), 0))
2255 //
2256 // into (f64x2.convert_low_i32x4_{s,u} $x).
2257 //
2258 // Or this:
2259 //
2260 // (v2f64 (extract_subvector
2261 // (v4f64 (fp_extend (v4f32 $x))), 0))
2262 //
2263 // into (f64x2.promote_low_f32x4 $x).
2264 auto Conversion = N->getOperand(0);
2265 auto ConversionOp = Conversion.getOpcode();
2266 MVT ExpectedSourceType;
2267 switch (ConversionOp) {
2268 case ISD::SINT_TO_FP:
2269 case ISD::UINT_TO_FP:
2270 ExpectedSourceType = MVT::v4i32;
2271 break;
2272 case ISD::FP_EXTEND:
2273 ExpectedSourceType = MVT::v4f32;
2274 break;
2275 default:
2276 return SDValue();
2277 }
2278
2279 if (Conversion.getValueType() != MVT::v4f64)
2280 return SDValue();
2281
2282 auto Source = Conversion.getOperand(0);
2283 if (Source.getValueType() != ExpectedSourceType)
2284 return SDValue();
2285
2286 auto IndexNode = dyn_cast<ConstantSDNode>(N->getOperand(1));
2287 if (IndexNode == nullptr || IndexNode->getZExtValue() != 0)
2288 return SDValue();
2289
2290 auto Op = GetWasmConversionOp(ConversionOp);
2291 return DAG.getNode(Op, SDLoc(N), ResVT, Source);
2292 }
2293
2294 // Combine this:
2295 //
2296 // (v2f64 ({s,u}int_to_fp
2297 // (v2i32 (extract_subvector (v4i32 $x), 0))))
2298 //
2299 // into (f64x2.convert_low_i32x4_{s,u} $x).
2300 //
2301 // Or this:
2302 //
2303 // (v2f64 (fp_extend
2304 // (v2f32 (extract_subvector (v4f32 $x), 0))))
2305 //
2306 // into (f64x2.promote_low_f32x4 $x).
2307 auto ConversionOp = N->getOpcode();
2308 MVT ExpectedExtractType;
2309 MVT ExpectedSourceType;
2310 switch (ConversionOp) {
2311 case ISD::SINT_TO_FP:
2312 case ISD::UINT_TO_FP:
2313 ExpectedExtractType = MVT::v2i32;
2314 ExpectedSourceType = MVT::v4i32;
2315 break;
2316 case ISD::FP_EXTEND:
2317 ExpectedExtractType = MVT::v2f32;
2318 ExpectedSourceType = MVT::v4f32;
2319 break;
2320 default:
2321 llvm_unreachable("unexpected opcode")::llvm::llvm_unreachable_internal("unexpected opcode", "/build/llvm-toolchain-snapshot-13~++20210726100616+dead50d4427c/llvm/lib/Target/WebAssembly/WebAssemblyISelLowering.cpp"
, 2321)
;
2322 }
2323
2324 auto Extract = N->getOperand(0);
2325 if (Extract.getOpcode() != ISD::EXTRACT_SUBVECTOR)
2326 return SDValue();
2327
2328 if (Extract.getValueType() != ExpectedExtractType)
2329 return SDValue();
2330
2331 auto Source = Extract.getOperand(0);
2332 if (Source.getValueType() != ExpectedSourceType)
2333 return SDValue();
2334
2335 auto *IndexNode = dyn_cast<ConstantSDNode>(Extract.getOperand(1));
2336 if (IndexNode == nullptr || IndexNode->getZExtValue() != 0)
2337 return SDValue();
2338
2339 unsigned Op = GetWasmConversionOp(ConversionOp);
2340 return DAG.getNode(Op, SDLoc(N), ResVT, Source);
2341}
2342
2343static SDValue
2344performVectorTruncZeroCombine(SDNode *N, TargetLowering::DAGCombinerInfo &DCI) {
2345 auto &DAG = DCI.DAG;
2346
2347 auto GetWasmConversionOp = [](unsigned Op) {
2348 switch (Op) {
2349 case ISD::FP_TO_SINT_SAT:
2350 return WebAssemblyISD::TRUNC_SAT_ZERO_S;
2351 case ISD::FP_TO_UINT_SAT:
2352 return WebAssemblyISD::TRUNC_SAT_ZERO_U;
2353 case ISD::FP_ROUND:
2354 return WebAssemblyISD::DEMOTE_ZERO;
2355 }
2356 llvm_unreachable("unexpected op")::llvm::llvm_unreachable_internal("unexpected op", "/build/llvm-toolchain-snapshot-13~++20210726100616+dead50d4427c/llvm/lib/Target/WebAssembly/WebAssemblyISelLowering.cpp"
, 2356)
;
2357 };
2358
2359 auto IsZeroSplat = [](SDValue SplatVal) {
2360 auto *Splat = dyn_cast<BuildVectorSDNode>(SplatVal.getNode());
2361 APInt SplatValue, SplatUndef;
2362 unsigned SplatBitSize;
2363 bool HasAnyUndefs;
2364 return Splat &&
2365 Splat->isConstantSplat(SplatValue, SplatUndef, SplatBitSize,
2366 HasAnyUndefs) &&
2367 SplatValue == 0;
2368 };
2369
2370 if (N->getOpcode() == ISD::CONCAT_VECTORS) {
2371 // Combine this:
2372 //
2373 // (concat_vectors (v2i32 (fp_to_{s,u}int_sat $x, 32)), (v2i32 (splat 0)))
2374 //
2375 // into (i32x4.trunc_sat_f64x2_zero_{s,u} $x).
2376 //
2377 // Or this:
2378 //
2379 // (concat_vectors (v2f32 (fp_round (v2f64 $x))), (v2f32 (splat 0)))
2380 //
2381 // into (f32x4.demote_zero_f64x2 $x).
2382 EVT ResVT;
2383 EVT ExpectedConversionType;
2384 auto Conversion = N->getOperand(0);
2385 auto ConversionOp = Conversion.getOpcode();
2386 switch (ConversionOp) {
2387 case ISD::FP_TO_SINT_SAT:
2388 case ISD::FP_TO_UINT_SAT:
2389 ResVT = MVT::v4i32;
2390 ExpectedConversionType = MVT::v2i32;
2391 break;
2392 case ISD::FP_ROUND:
2393 ResVT = MVT::v4f32;
2394 ExpectedConversionType = MVT::v2f32;
2395 break;
2396 default:
2397 return SDValue();
2398 }
2399
2400 if (N->getValueType(0) != ResVT)
2401 return SDValue();
2402
2403 if (Conversion.getValueType() != ExpectedConversionType)
2404 return SDValue();
2405
2406 auto Source = Conversion.getOperand(0);
2407 if (Source.getValueType() != MVT::v2f64)
2408 return SDValue();
2409
2410 if (!IsZeroSplat(N->getOperand(1)) ||
2411 N->getOperand(1).getValueType() != ExpectedConversionType)
2412 return SDValue();
2413
2414 unsigned Op = GetWasmConversionOp(ConversionOp);
2415 return DAG.getNode(Op, SDLoc(N), ResVT, Source);
2416 }
2417
2418 // Combine this:
2419 //
2420 // (fp_to_{s,u}int_sat (concat_vectors $x, (v2f64 (splat 0))), 32)
2421 //
2422 // into (i32x4.trunc_sat_f64x2_zero_{s,u} $x).
2423 //
2424 // Or this:
2425 //
2426 // (v4f32 (fp_round (concat_vectors $x, (v2f64 (splat 0)))))
2427 //
2428 // into (f32x4.demote_zero_f64x2 $x).
2429 EVT ResVT;
2430 auto ConversionOp = N->getOpcode();
2431 switch (ConversionOp) {
2432 case ISD::FP_TO_SINT_SAT:
2433 case ISD::FP_TO_UINT_SAT:
2434 ResVT = MVT::v4i32;
2435 break;
2436 case ISD::FP_ROUND:
2437 ResVT = MVT::v4f32;
2438 break;
2439 default:
2440 llvm_unreachable("unexpected op")::llvm::llvm_unreachable_internal("unexpected op", "/build/llvm-toolchain-snapshot-13~++20210726100616+dead50d4427c/llvm/lib/Target/WebAssembly/WebAssemblyISelLowering.cpp"
, 2440)
;
2441 }
2442
2443 if (N->getValueType(0) != ResVT)
2444 return SDValue();
2445
2446 auto Concat = N->getOperand(0);
2447 if (Concat.getValueType() != MVT::v4f64)
2448 return SDValue();
2449
2450 auto Source = Concat.getOperand(0);
2451 if (Source.getValueType() != MVT::v2f64)
2452 return SDValue();
2453
2454 if (!IsZeroSplat(Concat.getOperand(1)) ||
2455 Concat.getOperand(1).getValueType() != MVT::v2f64)
2456 return SDValue();
2457
2458 unsigned Op = GetWasmConversionOp(ConversionOp);
2459 return DAG.getNode(Op, SDLoc(N), ResVT, Source);
2460}
2461
2462SDValue
2463WebAssemblyTargetLowering::PerformDAGCombine(SDNode *N,
2464 DAGCombinerInfo &DCI) const {
2465 switch (N->getOpcode()) {
2466 default:
2467 return SDValue();
2468 case ISD::VECTOR_SHUFFLE:
2469 return performVECTOR_SHUFFLECombine(N, DCI);
2470 case ISD::SIGN_EXTEND:
2471 case ISD::ZERO_EXTEND:
2472 return performVectorExtendCombine(N, DCI);
2473 case ISD::SINT_TO_FP:
2474 case ISD::UINT_TO_FP:
2475 case ISD::FP_EXTEND:
2476 case ISD::EXTRACT_SUBVECTOR:
2477 return performVectorConvertLowCombine(N, DCI);
2478 case ISD::FP_TO_SINT_SAT:
2479 case ISD::FP_TO_UINT_SAT:
2480 case ISD::FP_ROUND:
2481 case ISD::CONCAT_VECTORS:
2482 return performVectorTruncZeroCombine(N, DCI);
2483 }
2484}

/build/llvm-toolchain-snapshot-13~++20210726100616+dead50d4427c/llvm/include/llvm/CodeGen/SelectionDAGNodes.h

1//===- llvm/CodeGen/SelectionDAGNodes.h - SelectionDAG Nodes ----*- C++ -*-===//
2//
3// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
4// See https://llvm.org/LICENSE.txt for license information.
5// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
6//
7//===----------------------------------------------------------------------===//
8//
9// This file declares the SDNode class and derived classes, which are used to
10// represent the nodes and operations present in a SelectionDAG. These nodes
11// and operations are machine code level operations, with some similarities to
12// the GCC RTL representation.
13//
14// Clients should include the SelectionDAG.h file instead of this file directly.
15//
16//===----------------------------------------------------------------------===//
17
18#ifndef LLVM_CODEGEN_SELECTIONDAGNODES_H
19#define LLVM_CODEGEN_SELECTIONDAGNODES_H
20
21#include "llvm/ADT/APFloat.h"
22#include "llvm/ADT/ArrayRef.h"
23#include "llvm/ADT/BitVector.h"
24#include "llvm/ADT/FoldingSet.h"
25#include "llvm/ADT/GraphTraits.h"
26#include "llvm/ADT/SmallPtrSet.h"
27#include "llvm/ADT/SmallVector.h"
28#include "llvm/ADT/ilist_node.h"
29#include "llvm/ADT/iterator.h"
30#include "llvm/ADT/iterator_range.h"
31#include "llvm/CodeGen/ISDOpcodes.h"
32#include "llvm/CodeGen/MachineMemOperand.h"
33#include "llvm/CodeGen/Register.h"
34#include "llvm/CodeGen/ValueTypes.h"
35#include "llvm/IR/Constants.h"
36#include "llvm/IR/DebugLoc.h"
37#include "llvm/IR/Instruction.h"
38#include "llvm/IR/Instructions.h"
39#include "llvm/IR/Metadata.h"
40#include "llvm/IR/Operator.h"
41#include "llvm/Support/AlignOf.h"
42#include "llvm/Support/AtomicOrdering.h"
43#include "llvm/Support/Casting.h"
44#include "llvm/Support/ErrorHandling.h"
45#include "llvm/Support/MachineValueType.h"
46#include "llvm/Support/TypeSize.h"
47#include <algorithm>
48#include <cassert>
49#include <climits>
50#include <cstddef>
51#include <cstdint>
52#include <cstring>
53#include <iterator>
54#include <string>
55#include <tuple>
56
57namespace llvm {
58
59class APInt;
60class Constant;
61template <typename T> struct DenseMapInfo;
62class GlobalValue;
63class MachineBasicBlock;
64class MachineConstantPoolValue;
65class MCSymbol;
66class raw_ostream;
67class SDNode;
68class SelectionDAG;
69class Type;
70class Value;
71
72void checkForCycles(const SDNode *N, const SelectionDAG *DAG = nullptr,
73 bool force = false);
74
75/// This represents a list of ValueType's that has been intern'd by
76/// a SelectionDAG. Instances of this simple value class are returned by
77/// SelectionDAG::getVTList(...).
78///
79struct SDVTList {
80 const EVT *VTs;
81 unsigned int NumVTs;
82};
83
84namespace ISD {
85
86 /// Node predicates
87
88/// If N is a BUILD_VECTOR or SPLAT_VECTOR node whose elements are all the
89/// same constant or undefined, return true and return the constant value in
90/// \p SplatValue.
91bool isConstantSplatVector(const SDNode *N, APInt &SplatValue);
92
93/// Return true if the specified node is a BUILD_VECTOR or SPLAT_VECTOR where
94/// all of the elements are ~0 or undef. If \p BuildVectorOnly is set to
95/// true, it only checks BUILD_VECTOR.
96bool isConstantSplatVectorAllOnes(const SDNode *N,
97 bool BuildVectorOnly = false);
98
99/// Return true if the specified node is a BUILD_VECTOR or SPLAT_VECTOR where
100/// all of the elements are 0 or undef. If \p BuildVectorOnly is set to true, it
101/// only checks BUILD_VECTOR.
102bool isConstantSplatVectorAllZeros(const SDNode *N,
103 bool BuildVectorOnly = false);
104
105/// Return true if the specified node is a BUILD_VECTOR where all of the
106/// elements are ~0 or undef.
107bool isBuildVectorAllOnes(const SDNode *N);
108
109/// Return true if the specified node is a BUILD_VECTOR where all of the
110/// elements are 0 or undef.
111bool isBuildVectorAllZeros(const SDNode *N);
112
113/// Return true if the specified node is a BUILD_VECTOR node of all
114/// ConstantSDNode or undef.
115bool isBuildVectorOfConstantSDNodes(const SDNode *N);
116
117/// Return true if the specified node is a BUILD_VECTOR node of all
118/// ConstantFPSDNode or undef.
119bool isBuildVectorOfConstantFPSDNodes(const SDNode *N);
120
121/// Return true if the node has at least one operand and all operands of the
122/// specified node are ISD::UNDEF.
123bool allOperandsUndef(const SDNode *N);
124
125} // end namespace ISD
126
127//===----------------------------------------------------------------------===//
128/// Unlike LLVM values, Selection DAG nodes may return multiple
129/// values as the result of a computation. Many nodes return multiple values,
130/// from loads (which define a token and a return value) to ADDC (which returns
131/// a result and a carry value), to calls (which may return an arbitrary number
132/// of values).
133///
134/// As such, each use of a SelectionDAG computation must indicate the node that
135/// computes it as well as which return value to use from that node. This pair
136/// of information is represented with the SDValue value type.
137///
138class SDValue {
139 friend struct DenseMapInfo<SDValue>;
140
141 SDNode *Node = nullptr; // The node defining the value we are using.
142 unsigned ResNo = 0; // Which return value of the node we are using.
143
144public:
145 SDValue() = default;
146 SDValue(SDNode *node, unsigned resno);
147
148 /// get the index which selects a specific result in the SDNode
149 unsigned getResNo() const { return ResNo; }
150
151 /// get the SDNode which holds the desired result
152 SDNode *getNode() const { return Node; }
153
154 /// set the SDNode
155 void setNode(SDNode *N) { Node = N; }
156
157 inline SDNode *operator->() const { return Node; }
158
159 bool operator==(const SDValue &O) const {
160 return Node == O.Node && ResNo == O.ResNo;
161 }
162 bool operator!=(const SDValue &O) const {
163 return !operator==(O);
164 }
165 bool operator<(const SDValue &O) const {
166 return std::tie(Node, ResNo) < std::tie(O.Node, O.ResNo);
167 }
168 explicit operator bool() const {
169 return Node != nullptr;
170 }
171
172 SDValue getValue(unsigned R) const {
173 return SDValue(Node, R);
174 }
175
176 /// Return true if this node is an operand of N.
177 bool isOperandOf(const SDNode *N) const;
178
179 /// Return the ValueType of the referenced return value.
180 inline EVT getValueType() const;
181
182 /// Return the simple ValueType of the referenced return value.
183 MVT getSimpleValueType() const {
184 return getValueType().getSimpleVT();
185 }
186
187 /// Returns the size of the value in bits.
188 ///
189 /// If the value type is a scalable vector type, the scalable property will
190 /// be set and the runtime size will be a positive integer multiple of the
191 /// base size.
192 TypeSize getValueSizeInBits() const {
193 return getValueType().getSizeInBits();
194 }
195
196 uint64_t getScalarValueSizeInBits() const {
197 return getValueType().getScalarType().getFixedSizeInBits();
198 }
199
200 // Forwarding methods - These forward to the corresponding methods in SDNode.
201 inline unsigned getOpcode() const;
202 inline unsigned getNumOperands() const;
203 inline const SDValue &getOperand(unsigned i) const;
204 inline uint64_t getConstantOperandVal(unsigned i) const;
205 inline const APInt &getConstantOperandAPInt(unsigned i) const;
206 inline bool isTargetMemoryOpcode() const;
207 inline bool isTargetOpcode() const;
208 inline bool isMachineOpcode() const;
209 inline bool isUndef() const;
210 inline unsigned getMachineOpcode() const;
211 inline const DebugLoc &getDebugLoc() const;
212 inline void dump() const;
213 inline void dump(const SelectionDAG *G) const;
214 inline void dumpr() const;
215 inline void dumpr(const SelectionDAG *G) const;
216
217 /// Return true if this operand (which must be a chain) reaches the
218 /// specified operand without crossing any side-effecting instructions.
219 /// In practice, this looks through token factors and non-volatile loads.
220 /// In order to remain efficient, this only
221 /// looks a couple of nodes in, it does not do an exhaustive search.
222 bool reachesChainWithoutSideEffects(SDValue Dest,
223 unsigned Depth = 2) const;
224
225 /// Return true if there are no nodes using value ResNo of Node.
226 inline bool use_empty() const;
227
228 /// Return true if there is exactly one node using value ResNo of Node.
229 inline bool hasOneUse() const;
230};
231
232template<> struct DenseMapInfo<SDValue> {
233 static inline SDValue getEmptyKey() {
234 SDValue V;
235 V.ResNo = -1U;
236 return V;
237 }
238
239 static inline SDValue getTombstoneKey() {
240 SDValue V;
241 V.ResNo = -2U;
242 return V;
243 }
244
245 static unsigned getHashValue(const SDValue &Val) {
246 return ((unsigned)((uintptr_t)Val.getNode() >> 4) ^
247 (unsigned)((uintptr_t)Val.getNode() >> 9)) + Val.getResNo();
248 }
249
250 static bool isEqual(const SDValue &LHS, const SDValue &RHS) {
251 return LHS == RHS;
252 }
253};
254
255/// Allow casting operators to work directly on
256/// SDValues as if they were SDNode*'s.
257template<> struct simplify_type<SDValue> {
258 using SimpleType = SDNode *;
259
260 static SimpleType getSimplifiedValue(SDValue &Val) {
261 return Val.getNode();
262 }
263};
264template<> struct simplify_type<const SDValue> {
265 using SimpleType = /*const*/ SDNode *;
266
267 static SimpleType getSimplifiedValue(const SDValue &Val) {
268 return Val.getNode();
269 }
270};
271
272/// Represents a use of a SDNode. This class holds an SDValue,
273/// which records the SDNode being used and the result number, a
274/// pointer to the SDNode using the value, and Next and Prev pointers,
275/// which link together all the uses of an SDNode.
276///
277class SDUse {
278 /// Val - The value being used.
279 SDValue Val;
280 /// User - The user of this value.
281 SDNode *User = nullptr;
282 /// Prev, Next - Pointers to the uses list of the SDNode referred by
283 /// this operand.
284 SDUse **Prev = nullptr;
285 SDUse *Next = nullptr;
286
287public:
288 SDUse() = default;
289 SDUse(const SDUse &U) = delete;
290 SDUse &operator=(const SDUse &) = delete;
291
292 /// Normally SDUse will just implicitly convert to an SDValue that it holds.
293 operator const SDValue&() const { return Val; }
294
295 /// If implicit conversion to SDValue doesn't work, the get() method returns
296 /// the SDValue.
297 const SDValue &get() const { return Val; }
298
299 /// This returns the SDNode that contains this Use.
300 SDNode *getUser() { return User; }
301
302 /// Get the next SDUse in the use list.
303 SDUse *getNext() const { return Next; }
304
305 /// Convenience function for get().getNode().
306 SDNode *getNode() const { return Val.getNode(); }
307 /// Convenience function for get().getResNo().
308 unsigned getResNo() const { return Val.getResNo(); }
309 /// Convenience function for get().getValueType().
310 EVT getValueType() const { return Val.getValueType(); }
311
312 /// Convenience function for get().operator==
313 bool operator==(const SDValue &V) const {
314 return Val == V;
315 }
316
317 /// Convenience function for get().operator!=
318 bool operator!=(const SDValue &V) const {
319 return Val != V;
320 }
321
322 /// Convenience function for get().operator<
323 bool operator<(const SDValue &V) const {
324 return Val < V;
325 }
326
327private:
328 friend class SelectionDAG;
329 friend class SDNode;
330 // TODO: unfriend HandleSDNode once we fix its operand handling.
331 friend class HandleSDNode;
332
333 void setUser(SDNode *p) { User = p; }
334
335 /// Remove this use from its existing use list, assign it the
336 /// given value, and add it to the new value's node's use list.
337 inline void set(const SDValue &V);
338 /// Like set, but only supports initializing a newly-allocated
339 /// SDUse with a non-null value.
340 inline void setInitial(const SDValue &V);
341 /// Like set, but only sets the Node portion of the value,
342 /// leaving the ResNo portion unmodified.
343 inline void setNode(SDNode *N);
344
345 void addToList(SDUse **List) {
346 Next = *List;
347 if (Next) Next->Prev = &Next;
348 Prev = List;
349 *List = this;
350 }
351
352 void removeFromList() {
353 *Prev = Next;
354 if (Next) Next->Prev = Prev;
355 }
356};
357
358/// simplify_type specializations - Allow casting operators to work directly on
359/// SDValues as if they were SDNode*'s.
360template<> struct simplify_type<SDUse> {
361 using SimpleType = SDNode *;
362
363 static SimpleType getSimplifiedValue(SDUse &Val) {
364 return Val.getNode();
365 }
366};
367
368/// These are IR-level optimization flags that may be propagated to SDNodes.
369/// TODO: This data structure should be shared by the IR optimizer and the
370/// the backend.
371struct SDNodeFlags {
372private:
373 bool NoUnsignedWrap : 1;
374 bool NoSignedWrap : 1;
375 bool Exact : 1;
376 bool NoNaNs : 1;
377 bool NoInfs : 1;
378 bool NoSignedZeros : 1;
379 bool AllowReciprocal : 1;
380 bool AllowContract : 1;
381 bool ApproximateFuncs : 1;
382 bool AllowReassociation : 1;
383
384 // We assume instructions do not raise floating-point exceptions by default,
385 // and only those marked explicitly may do so. We could choose to represent
386 // this via a positive "FPExcept" flags like on the MI level, but having a
387 // negative "NoFPExcept" flag here (that defaults to true) makes the flag
388 // intersection logic more straightforward.
389 bool NoFPExcept : 1;
390
391public:
392 /// Default constructor turns off all optimization flags.
393 SDNodeFlags()
394 : NoUnsignedWrap(false), NoSignedWrap(false), Exact(false), NoNaNs(false),
395 NoInfs(false), NoSignedZeros(false), AllowReciprocal(false),
396 AllowContract(false), ApproximateFuncs(false),
397 AllowReassociation(false), NoFPExcept(false) {}
398
399 /// Propagate the fast-math-flags from an IR FPMathOperator.
400 void copyFMF(const FPMathOperator &FPMO) {
401 setNoNaNs(FPMO.hasNoNaNs());
402 setNoInfs(FPMO.hasNoInfs());
403 setNoSignedZeros(FPMO.hasNoSignedZeros());
404 setAllowReciprocal(FPMO.hasAllowReciprocal());
405 setAllowContract(FPMO.hasAllowContract());
406 setApproximateFuncs(FPMO.hasApproxFunc());
407 setAllowReassociation(FPMO.hasAllowReassoc());
408 }
409
410 // These are mutators for each flag.
411 void setNoUnsignedWrap(bool b) { NoUnsignedWrap = b; }
412 void setNoSignedWrap(bool b) { NoSignedWrap = b; }
413 void setExact(bool b) { Exact = b; }
414 void setNoNaNs(bool b) { NoNaNs = b; }
415 void setNoInfs(bool b) { NoInfs = b; }
416 void setNoSignedZeros(bool b) { NoSignedZeros = b; }
417 void setAllowReciprocal(bool b) { AllowReciprocal = b; }
418 void setAllowContract(bool b) { AllowContract = b; }
419 void setApproximateFuncs(bool b) { ApproximateFuncs = b; }
420 void setAllowReassociation(bool b) { AllowReassociation = b; }
421 void setNoFPExcept(bool b) { NoFPExcept = b; }
422
423 // These are accessors for each flag.
424 bool hasNoUnsignedWrap() const { return NoUnsignedWrap; }
425 bool hasNoSignedWrap() const { return NoSignedWrap; }
426 bool hasExact() const { return Exact; }
427 bool hasNoNaNs() const { return NoNaNs; }
428 bool hasNoInfs() const { return NoInfs; }
429 bool hasNoSignedZeros() const { return NoSignedZeros; }
430 bool hasAllowReciprocal() const { return AllowReciprocal; }
431 bool hasAllowContract() const { return AllowContract; }
432 bool hasApproximateFuncs() const { return ApproximateFuncs; }
433 bool hasAllowReassociation() const { return AllowReassociation; }
434 bool hasNoFPExcept() const { return NoFPExcept; }
435
436 /// Clear any flags in this flag set that aren't also set in Flags. All
437 /// flags will be cleared if Flags are undefined.
438 void intersectWith(const SDNodeFlags Flags) {
439 NoUnsignedWrap &= Flags.NoUnsignedWrap;
440 NoSignedWrap &= Flags.NoSignedWrap;
441 Exact &= Flags.Exact;
442 NoNaNs &= Flags.NoNaNs;
443 NoInfs &= Flags.NoInfs;
444 NoSignedZeros &= Flags.NoSignedZeros;
445 AllowReciprocal &= Flags.AllowReciprocal;
446 AllowContract &= Flags.AllowContract;
447 ApproximateFuncs &= Flags.ApproximateFuncs;
448 AllowReassociation &= Flags.AllowReassociation;
449 NoFPExcept &= Flags.NoFPExcept;
450 }
451};
452
453/// Represents one node in the SelectionDAG.
454///
455class SDNode : public FoldingSetNode, public ilist_node<SDNode> {
456private:
457 /// The operation that this node performs.
458 int16_t NodeType;
459
460protected:
461 // We define a set of mini-helper classes to help us interpret the bits in our
462 // SubclassData. These are designed to fit within a uint16_t so they pack
463 // with NodeType.
464
465#if defined(_AIX) && (!defined(__GNUC__4) || defined(__clang__1))
466// Except for GCC; by default, AIX compilers store bit-fields in 4-byte words
467// and give the `pack` pragma push semantics.
468#define BEGIN_TWO_BYTE_PACK() _Pragma("pack(2)")pack(2)
469#define END_TWO_BYTE_PACK() _Pragma("pack(pop)")pack(pop)
470#else
471#define BEGIN_TWO_BYTE_PACK()
472#define END_TWO_BYTE_PACK()
473#endif
474
475BEGIN_TWO_BYTE_PACK()
476 class SDNodeBitfields {
477 friend class SDNode;
478 friend class MemIntrinsicSDNode;
479 friend class MemSDNode;
480 friend class SelectionDAG;
481
482 uint16_t HasDebugValue : 1;
483 uint16_t IsMemIntrinsic : 1;
484 uint16_t IsDivergent : 1;
485 };
486 enum { NumSDNodeBits = 3 };
487
488 class ConstantSDNodeBitfields {
489 friend class ConstantSDNode;
490
491 uint16_t : NumSDNodeBits;
492
493 uint16_t IsOpaque : 1;
494 };
495
496 class MemSDNodeBitfields {
497 friend class MemSDNode;
498 friend class MemIntrinsicSDNode;
499 friend class AtomicSDNode;
500
501 uint16_t : NumSDNodeBits;
502
503 uint16_t IsVolatile : 1;
504 uint16_t IsNonTemporal : 1;
505 uint16_t IsDereferenceable : 1;
506 uint16_t IsInvariant : 1;
507 };
508 enum { NumMemSDNodeBits = NumSDNodeBits + 4 };
509
510 class LSBaseSDNodeBitfields {
511 friend class LSBaseSDNode;
512 friend class MaskedLoadStoreSDNode;
513 friend class MaskedGatherScatterSDNode;
514
515 uint16_t : NumMemSDNodeBits;
516
517 // This storage is shared between disparate class hierarchies to hold an
518 // enumeration specific to the class hierarchy in use.
519 // LSBaseSDNode => enum ISD::MemIndexedMode
520 // MaskedLoadStoreBaseSDNode => enum ISD::MemIndexedMode
521 // MaskedGatherScatterSDNode => enum ISD::MemIndexType
522 uint16_t AddressingMode : 3;
523 };
524 enum { NumLSBaseSDNodeBits = NumMemSDNodeBits + 3 };
525
526 class LoadSDNodeBitfields {
527 friend class LoadSDNode;
528 friend class MaskedLoadSDNode;
529 friend class MaskedGatherSDNode;
530
531 uint16_t : NumLSBaseSDNodeBits;
532
533 uint16_t ExtTy : 2; // enum ISD::LoadExtType
534 uint16_t IsExpanding : 1;
535 };
536
537 class StoreSDNodeBitfields {
538 friend class StoreSDNode;
539 friend class MaskedStoreSDNode;
540 friend class MaskedScatterSDNode;
541
542 uint16_t : NumLSBaseSDNodeBits;
543
544 uint16_t IsTruncating : 1;
545 uint16_t IsCompressing : 1;
546 };
547
548 union {
549 char RawSDNodeBits[sizeof(uint16_t)];
550 SDNodeBitfields SDNodeBits;
551 ConstantSDNodeBitfields ConstantSDNodeBits;
552 MemSDNodeBitfields MemSDNodeBits;
553 LSBaseSDNodeBitfields LSBaseSDNodeBits;
554 LoadSDNodeBitfields LoadSDNodeBits;
555 StoreSDNodeBitfields StoreSDNodeBits;
556 };
557END_TWO_BYTE_PACK()
558#undef BEGIN_TWO_BYTE_PACK
559#undef END_TWO_BYTE_PACK
560
561 // RawSDNodeBits must cover the entirety of the union. This means that all of
562 // the union's members must have size <= RawSDNodeBits. We write the RHS as
563 // "2" instead of sizeof(RawSDNodeBits) because MSVC can't handle the latter.
564 static_assert(sizeof(SDNodeBitfields) <= 2, "field too wide");
565 static_assert(sizeof(ConstantSDNodeBitfields) <= 2, "field too wide");
566 static_assert(sizeof(MemSDNodeBitfields) <= 2, "field too wide");
567 static_assert(sizeof(LSBaseSDNodeBitfields) <= 2, "field too wide");
568 static_assert(sizeof(LoadSDNodeBitfields) <= 2, "field too wide");
569 static_assert(sizeof(StoreSDNodeBitfields) <= 2, "field too wide");
570
571private:
572 friend class SelectionDAG;
573 // TODO: unfriend HandleSDNode once we fix its operand handling.
574 friend class HandleSDNode;
575
576 /// Unique id per SDNode in the DAG.
577 int NodeId = -1;
578
579 /// The values that are used by this operation.
580 SDUse *OperandList = nullptr;
581
582 /// The types of the values this node defines. SDNode's may
583 /// define multiple values simultaneously.
584 const EVT *ValueList;
585
586 /// List of uses for this SDNode.
587 SDUse *UseList = nullptr;
588
589 /// The number of entries in the Operand/Value list.
590 unsigned short NumOperands = 0;
591 unsigned short NumValues;
592
593 // The ordering of the SDNodes. It roughly corresponds to the ordering of the
594 // original LLVM instructions.
595 // This is used for turning off scheduling, because we'll forgo
596 // the normal scheduling algorithms and output the instructions according to
597 // this ordering.
598 unsigned IROrder;
599
600 /// Source line information.
601 DebugLoc debugLoc;
602
603 /// Return a pointer to the specified value type.
604 static const EVT *getValueTypeList(EVT VT);
605
606 SDNodeFlags Flags;
607
608public:
609 /// Unique and persistent id per SDNode in the DAG.
610 /// Used for debug printing.
611 uint16_t PersistentId;
612
613 //===--------------------------------------------------------------------===//
614 // Accessors
615 //
616
617 /// Return the SelectionDAG opcode value for this node. For
618 /// pre-isel nodes (those for which isMachineOpcode returns false), these
619 /// are the opcode values in the ISD and <target>ISD namespaces. For
620 /// post-isel opcodes, see getMachineOpcode.
621 unsigned getOpcode() const { return (unsigned short)NodeType; }
622
623 /// Test if this node has a target-specific opcode (in the
624 /// \<target\>ISD namespace).
625 bool isTargetOpcode() const { return NodeType >= ISD::BUILTIN_OP_END; }
626
627 /// Test if this node has a target-specific opcode that may raise
628 /// FP exceptions (in the \<target\>ISD namespace and greater than
629 /// FIRST_TARGET_STRICTFP_OPCODE). Note that all target memory
630 /// opcode are currently automatically considered to possibly raise
631 /// FP exceptions as well.
632 bool isTargetStrictFPOpcode() const {
633 return NodeType >= ISD::FIRST_TARGET_STRICTFP_OPCODE;
634 }
635
636 /// Test if this node has a target-specific
637 /// memory-referencing opcode (in the \<target\>ISD namespace and
638 /// greater than FIRST_TARGET_MEMORY_OPCODE).
639 bool isTargetMemoryOpcode() const {
640 return NodeType >= ISD::FIRST_TARGET_MEMORY_OPCODE;
641 }
642
643 /// Return true if the type of the node type undefined.
644 bool isUndef() const { return NodeType == ISD::UNDEF; }
645
646 /// Test if this node is a memory intrinsic (with valid pointer information).
647 /// INTRINSIC_W_CHAIN and INTRINSIC_VOID nodes are sometimes created for
648 /// non-memory intrinsics (with chains) that are not really instances of
649 /// MemSDNode. For such nodes, we need some extra state to determine the
650 /// proper classof relationship.
651 bool isMemIntrinsic() const {
652 return (NodeType == ISD::INTRINSIC_W_CHAIN ||
653 NodeType == ISD::INTRINSIC_VOID) &&
654 SDNodeBits.IsMemIntrinsic;
655 }
656
657 /// Test if this node is a strict floating point pseudo-op.
658 bool isStrictFPOpcode() {
659 switch (NodeType) {
660 default:
661 return false;
662 case ISD::STRICT_FP16_TO_FP:
663 case ISD::STRICT_FP_TO_FP16:
664#define DAG_INSTRUCTION(NAME, NARG, ROUND_MODE, INTRINSIC, DAGN) \
665 case ISD::STRICT_##DAGN:
666#include "llvm/IR/ConstrainedOps.def"
667 return true;
668 }
669 }
670
671 /// Test if this node has a post-isel opcode, directly
672 /// corresponding to a MachineInstr opcode.
673 bool isMachineOpcode() const { return NodeType < 0; }
674
675 /// This may only be called if isMachineOpcode returns
676 /// true. It returns the MachineInstr opcode value that the node's opcode
677 /// corresponds to.
678 unsigned getMachineOpcode() const {
679 assert(isMachineOpcode() && "Not a MachineInstr opcode!")(static_cast <bool> (isMachineOpcode() && "Not a MachineInstr opcode!"
) ? void (0) : __assert_fail ("isMachineOpcode() && \"Not a MachineInstr opcode!\""
, "/build/llvm-toolchain-snapshot-13~++20210726100616+dead50d4427c/llvm/include/llvm/CodeGen/SelectionDAGNodes.h"
, 679, __extension__ __PRETTY_FUNCTION__))
;
680 return ~NodeType;
681 }
682
683 bool getHasDebugValue() const { return SDNodeBits.HasDebugValue; }
684 void setHasDebugValue(bool b) { SDNodeBits.HasDebugValue = b; }
685
686 bool isDivergent() const { return SDNodeBits.IsDivergent; }
687
688 /// Return true if there are no uses of this node.
689 bool use_empty() const { return UseList == nullptr; }
690
691 /// Return true if there is exactly one use of this node.
692 bool hasOneUse() const { return hasSingleElement(uses()); }
693
694 /// Return the number of uses of this node. This method takes
695 /// time proportional to the number of uses.
696 size_t use_size() const { return std::distance(use_begin(), use_end()); }
697
698 /// Return the unique node id.
699 int getNodeId() const { return NodeId; }
700
701 /// Set unique node id.
702 void setNodeId(int Id) { NodeId = Id; }
703
704 /// Return the node ordering.
705 unsigned getIROrder() const { return IROrder; }
706
707 /// Set the node ordering.
708 void setIROrder(unsigned Order) { IROrder = Order; }
709
710 /// Return the source location info.
711 const DebugLoc &getDebugLoc() const { return debugLoc; }
712
713 /// Set source location info. Try to avoid this, putting
714 /// it in the constructor is preferable.
715 void setDebugLoc(DebugLoc dl) { debugLoc = std::move(dl); }
716
717 /// This class provides iterator support for SDUse
718 /// operands that use a specific SDNode.
719 class use_iterator {
720 friend class SDNode;
721
722 SDUse *Op = nullptr;
723
724 explicit use_iterator(SDUse *op) : Op(op) {}
725
726 public:
727 using iterator_category = std::forward_iterator_tag;
728 using value_type = SDUse;
729 using difference_type = std::ptrdiff_t;
730 using pointer = value_type *;
731 using reference = value_type &;
732
733 use_iterator() = default;
734 use_iterator(const use_iterator &I) : Op(I.Op) {}
735
736 bool operator==(const use_iterator &x) const {
737 return Op == x.Op;
738 }
739 bool operator!=(const use_iterator &x) const {
740 return !operator==(x);
741 }
742
743 /// Return true if this iterator is at the end of uses list.
744 bool atEnd() const { return Op == nullptr; }
745
746 // Iterator traversal: forward iteration only.
747 use_iterator &operator++() { // Preincrement
748 assert(Op && "Cannot increment end iterator!")(static_cast <bool> (Op && "Cannot increment end iterator!"
) ? void (0) : __assert_fail ("Op && \"Cannot increment end iterator!\""
, "/build/llvm-toolchain-snapshot-13~++20210726100616+dead50d4427c/llvm/include/llvm/CodeGen/SelectionDAGNodes.h"
, 748, __extension__ __PRETTY_FUNCTION__))
;
749 Op = Op->getNext();
750 return *this;
751 }
752
753 use_iterator operator++(int) { // Postincrement
754 use_iterator tmp = *this; ++*this; return tmp;
755 }
756
757 /// Retrieve a pointer to the current user node.
758 SDNode *operator*() const {
759 assert(Op && "Cannot dereference end iterator!")(static_cast <bool> (Op && "Cannot dereference end iterator!"
) ? void (0) : __assert_fail ("Op && \"Cannot dereference end iterator!\""
, "/build/llvm-toolchain-snapshot-13~++20210726100616+dead50d4427c/llvm/include/llvm/CodeGen/SelectionDAGNodes.h"
, 759, __extension__ __PRETTY_FUNCTION__))
;
760 return Op->getUser();
761 }
762
763 SDNode *operator->() const { return operator*(); }
764
765 SDUse &getUse() const { return *Op; }
766
767 /// Retrieve the operand # of this use in its user.
768 unsigned getOperandNo() const {
769 assert(Op && "Cannot dereference end iterator!")(static_cast <bool> (Op && "Cannot dereference end iterator!"
) ? void (0) : __assert_fail ("Op && \"Cannot dereference end iterator!\""
, "/build/llvm-toolchain-snapshot-13~++20210726100616+dead50d4427c/llvm/include/llvm/CodeGen/SelectionDAGNodes.h"
, 769, __extension__ __PRETTY_FUNCTION__))
;
770 return (unsigned)(Op - Op->getUser()->OperandList);
771 }
772 };
773
774 /// Provide iteration support to walk over all uses of an SDNode.
775 use_iterator use_begin() const {
776 return use_iterator(UseList);
777 }
778
779 static use_iterator use_end() { return use_iterator(nullptr); }
780
781 inline iterator_range<use_iterator> uses() {
782 return make_range(use_begin(), use_end());
783 }
784 inline iterator_range<use_iterator> uses() const {
785 return make_range(use_begin(), use_end());
786 }
787
788 /// Return true if there are exactly NUSES uses of the indicated value.
789 /// This method ignores uses of other values defined by this operation.
790 bool hasNUsesOfValue(unsigned NUses, unsigned Value) const;
791
792 /// Return true if there are any use of the indicated value.
793 /// This method ignores uses of other values defined by this operation.
794 bool hasAnyUseOfValue(unsigned Value) const;
795
796 /// Return true if this node is the only use of N.
797 bool isOnlyUserOf(const SDNode *N) const;
798
799 /// Return true if this node is an operand of N.
800 bool isOperandOf(const SDNode *N) const;
801
802 /// Return true if this node is a predecessor of N.
803 /// NOTE: Implemented on top of hasPredecessor and every bit as
804 /// expensive. Use carefully.
805 bool isPredecessorOf(const SDNode *N) const {
806 return N->hasPredecessor(this);
807 }
808
809 /// Return true if N is a predecessor of this node.
810 /// N is either an operand of this node, or can be reached by recursively
811 /// traversing up the operands.
812 /// NOTE: This is an expensive method. Use it carefully.
813 bool hasPredecessor(const SDNode *N) const;
814
815 /// Returns true if N is a predecessor of any node in Worklist. This
816 /// helper keeps Visited and Worklist sets externally to allow unions
817 /// searches to be performed in parallel, caching of results across
818 /// queries and incremental addition to Worklist. Stops early if N is
819 /// found but will resume. Remember to clear Visited and Worklists
820 /// if DAG changes. MaxSteps gives a maximum number of nodes to visit before
821 /// giving up. The TopologicalPrune flag signals that positive NodeIds are
822 /// topologically ordered (Operands have strictly smaller node id) and search
823 /// can be pruned leveraging this.
824 static bool hasPredecessorHelper(const SDNode *N,
825 SmallPtrSetImpl<const SDNode *> &Visited,
826 SmallVectorImpl<const SDNode *> &Worklist,
827 unsigned int MaxSteps = 0,
828 bool TopologicalPrune = false) {
829 SmallVector<const SDNode *, 8> DeferredNodes;
830 if (Visited.count(N))
831 return true;
832
833 // Node Id's are assigned in three places: As a topological
834 // ordering (> 0), during legalization (results in values set to
835 // 0), new nodes (set to -1). If N has a topolgical id then we
836 // know that all nodes with ids smaller than it cannot be
837 // successors and we need not check them. Filter out all node
838 // that can't be matches. We add them to the worklist before exit
839 // in case of multiple calls. Note that during selection the topological id
840 // may be violated if a node's predecessor is selected before it. We mark
841 // this at selection negating the id of unselected successors and
842 // restricting topological pruning to positive ids.
843
844 int NId = N->getNodeId();
845 // If we Invalidated the Id, reconstruct original NId.
846 if (NId < -1)
847 NId = -(NId + 1);
848
849 bool Found = false;
850 while (!Worklist.empty()) {
851 const SDNode *M = Worklist.pop_back_val();
852 int MId = M->getNodeId();
853 if (TopologicalPrune && M->getOpcode() != ISD::TokenFactor && (NId > 0) &&
854 (MId > 0) && (MId < NId)) {
855 DeferredNodes.push_back(M);
856 continue;
857 }
858 for (const SDValue &OpV : M->op_values()) {
859 SDNode *Op = OpV.getNode();
860 if (Visited.insert(Op).second)
861 Worklist.push_back(Op);
862 if (Op == N)
863 Found = true;
864 }
865 if (Found)
866 break;
867 if (MaxSteps != 0 && Visited.size() >= MaxSteps)
868 break;
869 }
870 // Push deferred nodes back on worklist.
871 Worklist.append(DeferredNodes.begin(), DeferredNodes.end());
872 // If we bailed early, conservatively return found.
873 if (MaxSteps != 0 && Visited.size() >= MaxSteps)
874 return true;
875 return Found;
876 }
877
878 /// Return true if all the users of N are contained in Nodes.
879 /// NOTE: Requires at least one match, but doesn't require them all.
880 static bool areOnlyUsersOf(ArrayRef<const SDNode *> Nodes, const SDNode *N);
881
882 /// Return the number of values used by this operation.
883 unsigned getNumOperands() const { return NumOperands; }
884
885 /// Return the maximum number of operands that a SDNode can hold.
886 static constexpr size_t getMaxNumOperands() {
887 return std::numeric_limits<decltype(SDNode::NumOperands)>::max();
888 }
889
890 /// Helper method returns the integer value of a ConstantSDNode operand.
891 inline uint64_t getConstantOperandVal(unsigned Num) const;
892
893 /// Helper method returns the APInt of a ConstantSDNode operand.
894 inline const APInt &getConstantOperandAPInt(unsigned Num) const;
895
896 const SDValue &getOperand(unsigned Num) const {
897 assert(Num < NumOperands && "Invalid child # of SDNode!")(static_cast <bool> (Num < NumOperands && "Invalid child # of SDNode!"
) ? void (0) : __assert_fail ("Num < NumOperands && \"Invalid child # of SDNode!\""
, "/build/llvm-toolchain-snapshot-13~++20210726100616+dead50d4427c/llvm/include/llvm/CodeGen/SelectionDAGNodes.h"
, 897, __extension__ __PRETTY_FUNCTION__))
;
898 return OperandList[Num];
899 }
900
901 using op_iterator = SDUse *;
902
903 op_iterator op_begin() const { return OperandList; }
904 op_iterator op_end() const { return OperandList+NumOperands; }
905 ArrayRef<SDUse> ops() const { return makeArrayRef(op_begin(), op_end()); }
906
907 /// Iterator for directly iterating over the operand SDValue's.
908 struct value_op_iterator
909 : iterator_adaptor_base<value_op_iterator, op_iterator,
910 std::random_access_iterator_tag, SDValue,
911 ptrdiff_t, value_op_iterator *,
912 value_op_iterator *> {
913 explicit value_op_iterator(SDUse *U = nullptr)
914 : iterator_adaptor_base(U) {}
915
916 const SDValue &operator*() const { return I->get(); }
917 };
918
919 iterator_range<value_op_iterator> op_values() const {
920 return make_range(value_op_iterator(op_begin()),
921 value_op_iterator(op_end()));
922 }
923
924 SDVTList getVTList() const {
925 SDVTList X = { ValueList, NumValues };
926 return X;
927 }
928
929 /// If this node has a glue operand, return the node
930 /// to which the glue operand points. Otherwise return NULL.
931 SDNode *getGluedNode() const {
932 if (getNumOperands() != 0 &&
933 getOperand(getNumOperands()-1).getValueType() == MVT::Glue)
934 return getOperand(getNumOperands()-1).getNode();
935 return nullptr;
936 }
937
938 /// If this node has a glue value with a user, return
939 /// the user (there is at most one). Otherwise return NULL.
940 SDNode *getGluedUser() const {
941 for (use_iterator UI = use_begin(), UE = use_end(); UI != UE; ++UI)
942 if (UI.getUse().get().getValueType() == MVT::Glue)
943 return *UI;
944 return nullptr;
945 }
946
947 SDNodeFlags getFlags() const { return Flags; }
948 void setFlags(SDNodeFlags NewFlags) { Flags = NewFlags; }
949
950 /// Clear any flags in this node that aren't also set in Flags.
951 /// If Flags is not in a defined state then this has no effect.
952 void intersectFlagsWith(const SDNodeFlags Flags);
953
954 /// Return the number of values defined/returned by this operator.
955 unsigned getNumValues() const { return NumValues; }
956
957 /// Return the type of a specified result.
958 EVT getValueType(unsigned ResNo) const {
959 assert(ResNo < NumValues && "Illegal result number!")(static_cast <bool> (ResNo < NumValues && "Illegal result number!"
) ? void (0) : __assert_fail ("ResNo < NumValues && \"Illegal result number!\""
, "/build/llvm-toolchain-snapshot-13~++20210726100616+dead50d4427c/llvm/include/llvm/CodeGen/SelectionDAGNodes.h"
, 959, __extension__ __PRETTY_FUNCTION__))
;
960 return ValueList[ResNo];
961 }
962
963 /// Return the type of a specified result as a simple type.
964 MVT getSimpleValueType(unsigned ResNo) const {
965 return getValueType(ResNo).getSimpleVT();
966 }
967
968 /// Returns MVT::getSizeInBits(getValueType(ResNo)).
969 ///
970 /// If the value type is a scalable vector type, the scalable property will
971 /// be set and the runtime size will be a positive integer multiple of the
972 /// base size.
973 TypeSize getValueSizeInBits(unsigned ResNo) const {
974 return getValueType(ResNo).getSizeInBits();
975 }
976
977 using value_iterator = const EVT *;
978
979 value_iterator value_begin() const { return ValueList; }
980 value_iterator value_end() const { return ValueList+NumValues; }
981 iterator_range<value_iterator> values() const {
982 return llvm::make_range(value_begin(), value_end());
983 }
984
985 /// Return the opcode of this operation for printing.
986 std::string getOperationName(const SelectionDAG *G = nullptr) const;
987 static const char* getIndexedModeName(ISD::MemIndexedMode AM);
988 void print_types(raw_ostream &OS, const SelectionDAG *G) const;
989 void print_details(raw_ostream &OS, const SelectionDAG *G) const;
990 void print(raw_ostream &OS, const SelectionDAG *G = nullptr) const;
991 void printr(raw_ostream &OS, const SelectionDAG *G = nullptr) const;
992
993 /// Print a SelectionDAG node and all children down to
994 /// the leaves. The given SelectionDAG allows target-specific nodes
995 /// to be printed in human-readable form. Unlike printr, this will
996 /// print the whole DAG, including children that appear multiple
997 /// times.
998 ///
999 void printrFull(raw_ostream &O, const SelectionDAG *G = nullptr) const;
1000
1001 /// Print a SelectionDAG node and children up to
1002 /// depth "depth." The given SelectionDAG allows target-specific
1003 /// nodes to be printed in human-readable form. Unlike printr, this
1004 /// will print children that appear multiple times wherever they are
1005 /// used.
1006 ///
1007 void printrWithDepth(raw_ostream &O, const SelectionDAG *G = nullptr,
1008 unsigned depth = 100) const;
1009
1010 /// Dump this node, for debugging.
1011 void dump() const;
1012
1013 /// Dump (recursively) this node and its use-def subgraph.
1014 void dumpr() const;
1015
1016 /// Dump this node, for debugging.
1017 /// The given SelectionDAG allows target-specific nodes to be printed
1018 /// in human-readable form.
1019 void dump(const SelectionDAG *G) const;
1020
1021 /// Dump (recursively) this node and its use-def subgraph.
1022 /// The given SelectionDAG allows target-specific nodes to be printed
1023 /// in human-readable form.
1024 void dumpr(const SelectionDAG *G) const;
1025
1026 /// printrFull to dbgs(). The given SelectionDAG allows
1027 /// target-specific nodes to be printed in human-readable form.
1028 /// Unlike dumpr, this will print the whole DAG, including children
1029 /// that appear multiple times.
1030 void dumprFull(const SelectionDAG *G = nullptr) const;
1031
1032 /// printrWithDepth to dbgs(). The given
1033 /// SelectionDAG allows target-specific nodes to be printed in
1034 /// human-readable form. Unlike dumpr, this will print children
1035 /// that appear multiple times wherever they are used.
1036 ///
1037 void dumprWithDepth(const SelectionDAG *G = nullptr,
1038 unsigned depth = 100) const;
1039
1040 /// Gather unique data for the node.
1041 void Profile(FoldingSetNodeID &ID) const;
1042
1043 /// This method should only be used by the SDUse class.
1044 void addUse(SDUse &U) { U.addToList(&UseList); }
1045
1046protected:
1047 static SDVTList getSDVTList(EVT VT) {
1048 SDVTList Ret = { getValueTypeList(VT), 1 };
1049 return Ret;
1050 }
1051
1052 /// Create an SDNode.
1053 ///
1054 /// SDNodes are created without any operands, and never own the operand
1055 /// storage. To add operands, see SelectionDAG::createOperands.
1056 SDNode(unsigned Opc, unsigned Order, DebugLoc dl, SDVTList VTs)
1057 : NodeType(Opc), ValueList(VTs.VTs), NumValues(VTs.NumVTs),
1058 IROrder(Order), debugLoc(std::move(dl)) {
1059 memset(&RawSDNodeBits, 0, sizeof(RawSDNodeBits));
1060 assert(debugLoc.hasTrivialDestructor() && "Expected trivial destructor")(static_cast <bool> (debugLoc.hasTrivialDestructor() &&
"Expected trivial destructor") ? void (0) : __assert_fail ("debugLoc.hasTrivialDestructor() && \"Expected trivial destructor\""
, "/build/llvm-toolchain-snapshot-13~++20210726100616+dead50d4427c/llvm/include/llvm/CodeGen/SelectionDAGNodes.h"
, 1060, __extension__ __PRETTY_FUNCTION__))
;
1061 assert(NumValues == VTs.NumVTs &&(static_cast <bool> (NumValues == VTs.NumVTs &&
"NumValues wasn't wide enough for its operands!") ? void (0)
: __assert_fail ("NumValues == VTs.NumVTs && \"NumValues wasn't wide enough for its operands!\""
, "/build/llvm-toolchain-snapshot-13~++20210726100616+dead50d4427c/llvm/include/llvm/CodeGen/SelectionDAGNodes.h"
, 1062, __extension__ __PRETTY_FUNCTION__))
1062 "NumValues wasn't wide enough for its operands!")(static_cast <bool> (NumValues == VTs.NumVTs &&
"NumValues wasn't wide enough for its operands!") ? void (0)
: __assert_fail ("NumValues == VTs.NumVTs && \"NumValues wasn't wide enough for its operands!\""
, "/build/llvm-toolchain-snapshot-13~++20210726100616+dead50d4427c/llvm/include/llvm/CodeGen/SelectionDAGNodes.h"
, 1062, __extension__ __PRETTY_FUNCTION__))
;
1063 }
1064
1065 /// Release the operands and set this node to have zero operands.
1066 void DropOperands();
1067};
1068
1069/// Wrapper class for IR location info (IR ordering and DebugLoc) to be passed
1070/// into SDNode creation functions.
1071/// When an SDNode is created from the DAGBuilder, the DebugLoc is extracted
1072/// from the original Instruction, and IROrder is the ordinal position of
1073/// the instruction.
1074/// When an SDNode is created after the DAG is being built, both DebugLoc and
1075/// the IROrder are propagated from the original SDNode.
1076/// So SDLoc class provides two constructors besides the default one, one to
1077/// be used by the DAGBuilder, the other to be used by others.
1078class SDLoc {
1079private:
1080 DebugLoc DL;
1081 int IROrder = 0;
1082
1083public:
1084 SDLoc() = default;
1085 SDLoc(const SDNode *N) : DL(N->getDebugLoc()), IROrder(N->getIROrder()) {}
1086 SDLoc(const SDValue V) : SDLoc(V.getNode()) {}
1087 SDLoc(const Instruction *I, int Order) : IROrder(Order) {
1088 assert(Order >= 0 && "bad IROrder")(static_cast <bool> (Order >= 0 && "bad IROrder"
) ? void (0) : __assert_fail ("Order >= 0 && \"bad IROrder\""
, "/build/llvm-toolchain-snapshot-13~++20210726100616+dead50d4427c/llvm/include/llvm/CodeGen/SelectionDAGNodes.h"
, 1088, __extension__ __PRETTY_FUNCTION__))
;
1089 if (I)
1090 DL = I->getDebugLoc();
1091 }
1092
1093 unsigned getIROrder() const { return IROrder; }
1094 const DebugLoc &getDebugLoc() const { return DL; }
1095};
1096
1097// Define inline functions from the SDValue class.
1098
1099inline SDValue::SDValue(SDNode *node, unsigned resno)
1100 : Node(node), ResNo(resno) {
1101 // Explicitly check for !ResNo to avoid use-after-free, because there are
1102 // callers that use SDValue(N, 0) with a deleted N to indicate successful
1103 // combines.
1104 assert((!Node || !ResNo || ResNo < Node->getNumValues()) &&(static_cast <bool> ((!Node || !ResNo || ResNo < Node
->getNumValues()) && "Invalid result number for the given node!"
) ? void (0) : __assert_fail ("(!Node || !ResNo || ResNo < Node->getNumValues()) && \"Invalid result number for the given node!\""
, "/build/llvm-toolchain-snapshot-13~++20210726100616+dead50d4427c/llvm/include/llvm/CodeGen/SelectionDAGNodes.h"
, 1105, __extension__ __PRETTY_FUNCTION__))
1105 "Invalid result number for the given node!")(static_cast <bool> ((!Node || !ResNo || ResNo < Node
->getNumValues()) && "Invalid result number for the given node!"
) ? void (0) : __assert_fail ("(!Node || !ResNo || ResNo < Node->getNumValues()) && \"Invalid result number for the given node!\""
, "/build/llvm-toolchain-snapshot-13~++20210726100616+dead50d4427c/llvm/include/llvm/CodeGen/SelectionDAGNodes.h"
, 1105, __extension__ __PRETTY_FUNCTION__))
;
1106 assert(ResNo < -2U && "Cannot use result numbers reserved for DenseMaps.")(static_cast <bool> (ResNo < -2U && "Cannot use result numbers reserved for DenseMaps."
) ? void (0) : __assert_fail ("ResNo < -2U && \"Cannot use result numbers reserved for DenseMaps.\""
, "/build/llvm-toolchain-snapshot-13~++20210726100616+dead50d4427c/llvm/include/llvm/CodeGen/SelectionDAGNodes.h"
, 1106, __extension__ __PRETTY_FUNCTION__))
;
1107}
1108
1109inline unsigned SDValue::getOpcode() const {
1110 return Node->getOpcode();
1111}
1112
1113inline EVT SDValue::getValueType() const {
1114 return Node->getValueType(ResNo);
23
Called C++ object pointer is null
1115}
1116
1117inline unsigned SDValue::getNumOperands() const {
1118 return Node->getNumOperands();
1119}
1120
1121inline const SDValue &SDValue::getOperand(unsigned i) const {
1122 return Node->getOperand(i);
1123}
1124
1125inline uint64_t SDValue::getConstantOperandVal(unsigned i) const {
1126 return Node->getConstantOperandVal(i);
1127}
1128
1129inline const APInt &SDValue::getConstantOperandAPInt(unsigned i) const {
1130 return Node->getConstantOperandAPInt(i);
1131}
1132
1133inline bool SDValue::isTargetOpcode() const {
1134 return Node->isTargetOpcode();
1135}
1136
1137inline bool SDValue::isTargetMemoryOpcode() const {
1138 return Node->isTargetMemoryOpcode();
1139}
1140
1141inline bool SDValue::isMachineOpcode() const {
1142 return Node->isMachineOpcode();
1143}
1144
1145inline unsigned SDValue::getMachineOpcode() const {
1146 return Node->getMachineOpcode();
1147}
1148
1149inline bool SDValue::isUndef() const {
1150 return Node->isUndef();
1151}
1152
1153inline bool SDValue::use_empty() const {
1154 return !Node->hasAnyUseOfValue(ResNo);
1155}
1156
1157inline bool SDValue::hasOneUse() const {
1158 return Node->hasNUsesOfValue(1, ResNo);
1159}
1160
1161inline const DebugLoc &SDValue::getDebugLoc() const {
1162 return Node->getDebugLoc();
1163}
1164
1165inline void SDValue::dump() const {
1166 return Node->dump();
1167}
1168
1169inline void SDValue::dump(const SelectionDAG *G) const {
1170 return Node->dump(G);
1171}
1172
1173inline void SDValue::dumpr() const {
1174 return Node->dumpr();
1175}
1176
1177inline void SDValue::dumpr(const SelectionDAG *G) const {
1178 return Node->dumpr(G);
1179}
1180
1181// Define inline functions from the SDUse class.
1182
1183inline void SDUse::set(const SDValue &V) {
1184 if (Val.getNode()) removeFromList();
1185 Val = V;
1186 if (V.getNode()) V.getNode()->addUse(*this);
1187}
1188
1189inline void SDUse::setInitial(const SDValue &V) {
1190 Val = V;
1191 V.getNode()->addUse(*this);
1192}
1193
1194inline void SDUse::setNode(SDNode *N) {
1195 if (Val.getNode()) removeFromList();
1196 Val.setNode(N);
1197 if (N) N->addUse(*this);
1198}
1199
1200/// This class is used to form a handle around another node that
1201/// is persistent and is updated across invocations of replaceAllUsesWith on its
1202/// operand. This node should be directly created by end-users and not added to
1203/// the AllNodes list.
1204class HandleSDNode : public SDNode {
1205 SDUse Op;
1206
1207public:
1208 explicit HandleSDNode(SDValue X)
1209 : SDNode(ISD::HANDLENODE, 0, DebugLoc(), getSDVTList(MVT::Other)) {
1210 // HandleSDNodes are never inserted into the DAG, so they won't be
1211 // auto-numbered. Use ID 65535 as a sentinel.
1212 PersistentId = 0xffff;
1213
1214 // Manually set up the operand list. This node type is special in that it's
1215 // always stack allocated and SelectionDAG does not manage its operands.
1216 // TODO: This should either (a) not be in the SDNode hierarchy, or (b) not
1217 // be so special.
1218 Op.setUser(this);
1219 Op.setInitial(X);
1220 NumOperands = 1;
1221 OperandList = &Op;
1222 }
1223 ~HandleSDNode();
1224
1225 const SDValue &getValue() const { return Op; }
1226};
1227
1228class AddrSpaceCastSDNode : public SDNode {
1229private:
1230 unsigned SrcAddrSpace;
1231 unsigned DestAddrSpace;
1232
1233public:
1234 AddrSpaceCastSDNode(unsigned Order, const DebugLoc &dl, EVT VT,
1235 unsigned SrcAS, unsigned DestAS);
1236
1237 unsigned getSrcAddressSpace() const { return SrcAddrSpace; }
1238 unsigned getDestAddressSpace() const { return DestAddrSpace; }
1239
1240 static bool classof(const SDNode *N) {
1241 return N->getOpcode() == ISD::ADDRSPACECAST;
1242 }
1243};
1244
1245/// This is an abstract virtual class for memory operations.
1246class MemSDNode : public SDNode {
1247private:
1248 // VT of in-memory value.
1249 EVT MemoryVT;
1250
1251protected:
1252 /// Memory reference information.
1253 MachineMemOperand *MMO;
1254
1255public:
1256 MemSDNode(unsigned Opc, unsigned Order, const DebugLoc &dl, SDVTList VTs,
1257 EVT memvt, MachineMemOperand *MMO);
1258
1259 bool readMem() const { return MMO->isLoad(); }
1260 bool writeMem() const { return MMO->isStore(); }
1261
1262 /// Returns alignment and volatility of the memory access
1263 Align getOriginalAlign() const { return MMO->getBaseAlign(); }
1264 Align getAlign() const { return MMO->getAlign(); }
1265 // FIXME: Remove once transition to getAlign is over.
1266 unsigned getAlignment() const { return MMO->getAlign().value(); }
1267
1268 /// Return the SubclassData value, without HasDebugValue. This contains an
1269 /// encoding of the volatile flag, as well as bits used by subclasses. This
1270 /// function should only be used to compute a FoldingSetNodeID value.
1271 /// The HasDebugValue bit is masked out because CSE map needs to match
1272 /// nodes with debug info with nodes without debug info. Same is about
1273 /// isDivergent bit.
1274 unsigned getRawSubclassData() const {
1275 uint16_t Data;
1276 union {
1277 char RawSDNodeBits[sizeof(uint16_t)];
1278 SDNodeBitfields SDNodeBits;
1279 };
1280 memcpy(&RawSDNodeBits, &this->RawSDNodeBits, sizeof(this->RawSDNodeBits));
1281 SDNodeBits.HasDebugValue = 0;
1282 SDNodeBits.IsDivergent = false;
1283 memcpy(&Data, &RawSDNodeBits, sizeof(RawSDNodeBits));
1284 return Data;
1285 }
1286
1287 bool isVolatile() const { return MemSDNodeBits.IsVolatile; }
1288 bool isNonTemporal() const { return MemSDNodeBits.IsNonTemporal; }
1289 bool isDereferenceable() const { return MemSDNodeBits.IsDereferenceable; }
1290 bool isInvariant() const { return MemSDNodeBits.IsInvariant; }
1291
1292 // Returns the offset from the location of the access.
1293 int64_t getSrcValueOffset() const { return MMO->getOffset(); }
1294
1295 /// Returns the AA info that describes the dereference.
1296 AAMDNodes getAAInfo() const { return MMO->getAAInfo(); }
1297
1298 /// Returns the Ranges that describes the dereference.
1299 const MDNode *getRanges() const { return MMO->getRanges(); }
1300
1301 /// Returns the synchronization scope ID for this memory operation.
1302 SyncScope::ID getSyncScopeID() const { return MMO->getSyncScopeID(); }
1303
1304 /// Return the atomic ordering requirements for this memory operation. For
1305 /// cmpxchg atomic operations, return the atomic ordering requirements when
1306 /// store occurs.
1307 AtomicOrdering getSuccessOrdering() const {
1308 return MMO->getSuccessOrdering();
1309 }
1310
1311 /// Return a single atomic ordering that is at least as strong as both the
1312 /// success and failure orderings for an atomic operation. (For operations
1313 /// other than cmpxchg, this is equivalent to getSuccessOrdering().)
1314 AtomicOrdering getMergedOrdering() const { return MMO->getMergedOrdering(); }
1315
1316 /// Return true if the memory operation ordering is Unordered or higher.
1317 bool isAtomic() const { return MMO->isAtomic(); }
1318
1319 /// Returns true if the memory operation doesn't imply any ordering
1320 /// constraints on surrounding memory operations beyond the normal memory
1321 /// aliasing rules.
1322 bool isUnordered() const { return MMO->isUnordered(); }
1323
1324 /// Returns true if the memory operation is neither atomic or volatile.
1325 bool isSimple() const { return !isAtomic() && !isVolatile(); }
1326
1327 /// Return the type of the in-memory value.
1328 EVT getMemoryVT() const { return MemoryVT; }
1329
1330 /// Return a MachineMemOperand object describing the memory
1331 /// reference performed by operation.
1332 MachineMemOperand *getMemOperand() const { return MMO; }
1333
1334 const MachinePointerInfo &getPointerInfo() const {
1335 return MMO->getPointerInfo();
1336 }
1337
1338 /// Return the address space for the associated pointer
1339 unsigned getAddressSpace() const {
1340 return getPointerInfo().getAddrSpace();
1341 }
1342
1343 /// Update this MemSDNode's MachineMemOperand information
1344 /// to reflect the alignment of NewMMO, if it has a greater alignment.
1345 /// This must only be used when the new alignment applies to all users of
1346 /// this MachineMemOperand.
1347 void refineAlignment(const MachineMemOperand *NewMMO) {
1348 MMO->refineAlignment(NewMMO);
1349 }
1350
1351 const SDValue &getChain() const { return getOperand(0); }
1352
1353 const SDValue &getBasePtr() const {
1354 switch (getOpcode()) {
1355 case ISD::STORE:
1356 case ISD::MSTORE:
1357 return getOperand(2);
1358 case ISD::MGATHER:
1359 case ISD::MSCATTER:
1360 return getOperand(3);
1361 default:
1362 return getOperand(1);
1363 }
1364 }
1365
1366 // Methods to support isa and dyn_cast
1367 static bool classof(const SDNode *N) {
1368 // For some targets, we lower some target intrinsics to a MemIntrinsicNode
1369 // with either an intrinsic or a target opcode.
1370 switch (N->getOpcode()) {
1371 case ISD::LOAD:
1372 case ISD::STORE:
1373 case ISD::PREFETCH:
1374 case ISD::ATOMIC_CMP_SWAP:
1375 case ISD::ATOMIC_CMP_SWAP_WITH_SUCCESS:
1376 case ISD::ATOMIC_SWAP:
1377 case ISD::ATOMIC_LOAD_ADD:
1378 case ISD::ATOMIC_LOAD_SUB:
1379 case ISD::ATOMIC_LOAD_AND:
1380 case ISD::ATOMIC_LOAD_CLR:
1381 case ISD::ATOMIC_LOAD_OR:
1382 case ISD::ATOMIC_LOAD_XOR:
1383 case ISD::ATOMIC_LOAD_NAND:
1384 case ISD::ATOMIC_LOAD_MIN:
1385 case ISD::ATOMIC_LOAD_MAX:
1386 case ISD::ATOMIC_LOAD_UMIN:
1387 case ISD::ATOMIC_LOAD_UMAX:
1388 case ISD::ATOMIC_LOAD_FADD:
1389 case ISD::ATOMIC_LOAD_FSUB:
1390 case ISD::ATOMIC_LOAD:
1391 case ISD::ATOMIC_STORE:
1392 case ISD::MLOAD:
1393 case ISD::MSTORE:
1394 case ISD::MGATHER:
1395 case ISD::MSCATTER:
1396 return true;
1397 default:
1398 return N->isMemIntrinsic() || N->isTargetMemoryOpcode();
1399 }
1400 }
1401};
1402
1403/// This is an SDNode representing atomic operations.
1404class AtomicSDNode : public MemSDNode {
1405public:
1406 AtomicSDNode(unsigned Opc, unsigned Order, const DebugLoc &dl, SDVTList VTL,
1407 EVT MemVT, MachineMemOperand *MMO)
1408 : MemSDNode(Opc, Order, dl, VTL, MemVT, MMO) {
1409 assert(((Opc != ISD::ATOMIC_LOAD && Opc != ISD::ATOMIC_STORE) ||(static_cast <bool> (((Opc != ISD::ATOMIC_LOAD &&
Opc != ISD::ATOMIC_STORE) || MMO->isAtomic()) && "then why are we using an AtomicSDNode?"
) ? void (0) : __assert_fail ("((Opc != ISD::ATOMIC_LOAD && Opc != ISD::ATOMIC_STORE) || MMO->isAtomic()) && \"then why are we using an AtomicSDNode?\""
, "/build/llvm-toolchain-snapshot-13~++20210726100616+dead50d4427c/llvm/include/llvm/CodeGen/SelectionDAGNodes.h"
, 1410, __extension__ __PRETTY_FUNCTION__))
1410 MMO->isAtomic()) && "then why are we using an AtomicSDNode?")(static_cast <bool> (((Opc != ISD::ATOMIC_LOAD &&
Opc != ISD::ATOMIC_STORE) || MMO->isAtomic()) && "then why are we using an AtomicSDNode?"
) ? void (0) : __assert_fail ("((Opc != ISD::ATOMIC_LOAD && Opc != ISD::ATOMIC_STORE) || MMO->isAtomic()) && \"then why are we using an AtomicSDNode?\""
, "/build/llvm-toolchain-snapshot-13~++20210726100616+dead50d4427c/llvm/include/llvm/CodeGen/SelectionDAGNodes.h"
, 1410, __extension__ __PRETTY_FUNCTION__))
;
1411 }
1412
1413 const SDValue &getBasePtr() const { return getOperand(1); }
1414 const SDValue &getVal() const { return getOperand(2); }
1415
1416 /// Returns true if this SDNode represents cmpxchg atomic operation, false
1417 /// otherwise.
1418 bool isCompareAndSwap() const {
1419 unsigned Op = getOpcode();
1420 return Op == ISD::ATOMIC_CMP_SWAP ||
1421 Op == ISD::ATOMIC_CMP_SWAP_WITH_SUCCESS;
1422 }
1423
1424 /// For cmpxchg atomic operations, return the atomic ordering requirements
1425 /// when store does not occur.
1426 AtomicOrdering getFailureOrdering() const {
1427 assert(isCompareAndSwap() && "Must be cmpxchg operation")(static_cast <bool> (isCompareAndSwap() && "Must be cmpxchg operation"
) ? void (0) : __assert_fail ("isCompareAndSwap() && \"Must be cmpxchg operation\""
, "/build/llvm-toolchain-snapshot-13~++20210726100616+dead50d4427c/llvm/include/llvm/CodeGen/SelectionDAGNodes.h"
, 1427, __extension__ __PRETTY_FUNCTION__))
;
1428 return MMO->getFailureOrdering();
1429 }
1430
1431 // Methods to support isa and dyn_cast
1432 static bool classof(const SDNode *N) {
1433 return N->getOpcode() == ISD::ATOMIC_CMP_SWAP ||
1434 N->getOpcode() == ISD::ATOMIC_CMP_SWAP_WITH_SUCCESS ||
1435 N->getOpcode() == ISD::ATOMIC_SWAP ||
1436 N->getOpcode() == ISD::ATOMIC_LOAD_ADD ||
1437 N->getOpcode() == ISD::ATOMIC_LOAD_SUB ||
1438 N->getOpcode() == ISD::ATOMIC_LOAD_AND ||
1439 N->getOpcode() == ISD::ATOMIC_LOAD_CLR ||
1440 N->getOpcode() == ISD::ATOMIC_LOAD_OR ||
1441 N->getOpcode() == ISD::ATOMIC_LOAD_XOR ||
1442 N->getOpcode() == ISD::ATOMIC_LOAD_NAND ||
1443 N->getOpcode() == ISD::ATOMIC_LOAD_MIN ||
1444 N->getOpcode() == ISD::ATOMIC_LOAD_MAX ||
1445 N->getOpcode() == ISD::ATOMIC_LOAD_UMIN ||
1446 N->getOpcode() == ISD::ATOMIC_LOAD_UMAX ||
1447 N->getOpcode() == ISD::ATOMIC_LOAD_FADD ||
1448 N->getOpcode() == ISD::ATOMIC_LOAD_FSUB ||
1449 N->getOpcode() == ISD::ATOMIC_LOAD ||
1450 N->getOpcode() == ISD::ATOMIC_STORE;
1451 }
1452};
1453
1454/// This SDNode is used for target intrinsics that touch
1455/// memory and need an associated MachineMemOperand. Its opcode may be
1456/// INTRINSIC_VOID, INTRINSIC_W_CHAIN, PREFETCH, or a target-specific opcode
1457/// with a value not less than FIRST_TARGET_MEMORY_OPCODE.
1458class MemIntrinsicSDNode : public MemSDNode {
1459public:
1460 MemIntrinsicSDNode(unsigned Opc, unsigned Order, const DebugLoc &dl,
1461 SDVTList VTs, EVT MemoryVT, MachineMemOperand *MMO)
1462 : MemSDNode(Opc, Order, dl, VTs, MemoryVT, MMO) {
1463 SDNodeBits.IsMemIntrinsic = true;
1464 }
1465
1466 // Methods to support isa and dyn_cast
1467 static bool classof(const SDNode *N) {
1468 // We lower some target intrinsics to their target opcode
1469 // early a node with a target opcode can be of this class
1470 return N->isMemIntrinsic() ||
1471 N->getOpcode() == ISD::PREFETCH ||
1472 N->isTargetMemoryOpcode();
1473 }
1474};
1475
1476/// This SDNode is used to implement the code generator
1477/// support for the llvm IR shufflevector instruction. It combines elements
1478/// from two input vectors into a new input vector, with the selection and
1479/// ordering of elements determined by an array of integers, referred to as
1480/// the shuffle mask. For input vectors of width N, mask indices of 0..N-1
1481/// refer to elements from the LHS input, and indices from N to 2N-1 the RHS.
1482/// An index of -1 is treated as undef, such that the code generator may put
1483/// any value in the corresponding element of the result.
1484class ShuffleVectorSDNode : public SDNode {
1485 // The memory for Mask is owned by the SelectionDAG's OperandAllocator, and
1486 // is freed when the SelectionDAG object is destroyed.
1487 const int *Mask;
1488
1489protected:
1490 friend class SelectionDAG;
1491
1492 ShuffleVectorSDNode(EVT VT, unsigned Order, const DebugLoc &dl, const int *M)
1493 : SDNode(ISD::VECTOR_SHUFFLE, Order, dl, getSDVTList(VT)), Mask(M) {}
1494
1495public:
1496 ArrayRef<int> getMask() const {
1497 EVT VT = getValueType(0);
1498 return makeArrayRef(Mask, VT.getVectorNumElements());
1499 }
1500
1501 int getMaskElt(unsigned Idx) const {
1502 assert(Idx < getValueType(0).getVectorNumElements() && "Idx out of range!")(static_cast <bool> (Idx < getValueType(0).getVectorNumElements
() && "Idx out of range!") ? void (0) : __assert_fail
("Idx < getValueType(0).getVectorNumElements() && \"Idx out of range!\""
, "/build/llvm-toolchain-snapshot-13~++20210726100616+dead50d4427c/llvm/include/llvm/CodeGen/SelectionDAGNodes.h"
, 1502, __extension__ __PRETTY_FUNCTION__))
;
1503 return Mask[Idx];
1504 }
1505
1506 bool isSplat() const { return isSplatMask(Mask, getValueType(0)); }
1507
1508 int getSplatIndex() const {
1509 assert(isSplat() && "Cannot get splat index for non-splat!")(static_cast <bool> (isSplat() && "Cannot get splat index for non-splat!"
) ? void (0) : __assert_fail ("isSplat() && \"Cannot get splat index for non-splat!\""
, "/build/llvm-toolchain-snapshot-13~++20210726100616+dead50d4427c/llvm/include/llvm/CodeGen/SelectionDAGNodes.h"
, 1509, __extension__ __PRETTY_FUNCTION__))
;
1510 EVT VT = getValueType(0);
1511 for (unsigned i = 0, e = VT.getVectorNumElements(); i != e; ++i)
1512 if (Mask[i] >= 0)
1513 return Mask[i];
1514
1515 // We can choose any index value here and be correct because all elements
1516 // are undefined. Return 0 for better potential for callers to simplify.
1517 return 0;
1518 }
1519
1520 static bool isSplatMask(const int *Mask, EVT VT);
1521
1522 /// Change values in a shuffle permute mask assuming
1523 /// the two vector operands have swapped position.
1524 static void commuteMask(MutableArrayRef<int> Mask) {
1525 unsigned NumElems = Mask.size();
1526 for (unsigned i = 0; i != NumElems; ++i) {
1527 int idx = Mask[i];
1528 if (idx < 0)
1529 continue;
1530 else if (idx < (int)NumElems)
1531 Mask[i] = idx + NumElems;
1532 else
1533 Mask[i] = idx - NumElems;
1534 }
1535 }
1536
1537 static bool classof(const SDNode *N) {
1538 return N->getOpcode() == ISD::VECTOR_SHUFFLE;
1539 }
1540};
1541
1542class ConstantSDNode : public SDNode {
1543 friend class SelectionDAG;
1544
1545 const ConstantInt *Value;
1546
1547 ConstantSDNode(bool isTarget, bool isOpaque, const ConstantInt *val, EVT VT)
1548 : SDNode(isTarget ? ISD::TargetConstant : ISD::Constant, 0, DebugLoc(),
1549 getSDVTList(VT)),
1550 Value(val) {
1551 ConstantSDNodeBits.IsOpaque = isOpaque;
1552 }
1553
1554public:
1555 const ConstantInt *getConstantIntValue() const { return Value; }
1556 const APInt &getAPIntValue() const { return Value->getValue(); }
1557 uint64_t getZExtValue() const { return Value->getZExtValue(); }
1558 int64_t getSExtValue() const { return Value->getSExtValue(); }
1559 uint64_t getLimitedValue(uint64_t Limit = UINT64_MAX(18446744073709551615UL)) {
1560 return Value->getLimitedValue(Limit);
1561 }
1562 MaybeAlign getMaybeAlignValue() const { return Value->getMaybeAlignValue(); }
1563 Align getAlignValue() const { return Value->getAlignValue(); }
1564
1565 bool isOne() const { return Value->isOne(); }
1566 bool isNullValue() const { return Value->isZero(); }
1567 bool isAllOnesValue() const { return Value->isMinusOne(); }
1568
1569 bool isOpaque() const { return ConstantSDNodeBits.IsOpaque; }
1570
1571 static bool classof(const SDNode *N) {
1572 return N->getOpcode() == ISD::Constant ||
1573 N->getOpcode() == ISD::TargetConstant;
1574 }
1575};
1576
1577uint64_t SDNode::getConstantOperandVal(unsigned Num) const {
1578 return cast<ConstantSDNode>(getOperand(Num))->getZExtValue();
1579}
1580
1581const APInt &SDNode::getConstantOperandAPInt(unsigned Num) const {
1582 return cast<ConstantSDNode>(getOperand(Num))->getAPIntValue();
1583}
1584
1585class ConstantFPSDNode : public SDNode {
1586 friend class SelectionDAG;
1587
1588 const ConstantFP *Value;
1589
1590 ConstantFPSDNode(bool isTarget, const ConstantFP *val, EVT VT)
1591 : SDNode(isTarget ? ISD::TargetConstantFP : ISD::ConstantFP, 0,
1592 DebugLoc(), getSDVTList(VT)),
1593 Value(val) {}
1594
1595public:
1596 const APFloat& getValueAPF() const { return Value->getValueAPF(); }
1597 const ConstantFP *getConstantFPValue() const { return Value; }
1598
1599 /// Return true if the value is positive or negative zero.
1600 bool isZero() const { return Value->isZero(); }
1601
1602 /// Return true if the value is a NaN.
1603 bool isNaN() const { return Value->isNaN(); }
1604
1605 /// Return true if the value is an infinity
1606 bool isInfinity() const { return Value->isInfinity(); }
1607
1608 /// Return true if the value is negative.
1609 bool isNegative() const { return Value->isNegative(); }
1610
1611 /// We don't rely on operator== working on double values, as
1612 /// it returns true for things that are clearly not equal, like -0.0 and 0.0.
1613 /// As such, this method can be used to do an exact bit-for-bit comparison of
1614 /// two floating point values.
1615
1616 /// We leave the version with the double argument here because it's just so
1617 /// convenient to write "2.0" and the like. Without this function we'd
1618 /// have to duplicate its logic everywhere it's called.
1619 bool isExactlyValue(double V) const {
1620 return Value->getValueAPF().isExactlyValue(V);
1621 }
1622 bool isExactlyValue(const APFloat& V) const;
1623
1624 static bool isValueValidForType(EVT VT, const APFloat& Val);
1625
1626 static bool classof(const SDNode *N) {
1627 return N->getOpcode() == ISD::ConstantFP ||
1628 N->getOpcode() == ISD::TargetConstantFP;
1629 }
1630};
1631
1632/// Returns true if \p V is a constant integer zero.
1633bool isNullConstant(SDValue V);
1634
1635/// Returns true if \p V is an FP constant with a value of positive zero.
1636bool isNullFPConstant(SDValue V);
1637
1638/// Returns true if \p V is an integer constant with all bits set.
1639bool isAllOnesConstant(SDValue V);
1640
1641/// Returns true if \p V is a constant integer one.
1642bool isOneConstant(SDValue V);
1643
1644/// Return the non-bitcasted source operand of \p V if it exists.
1645/// If \p V is not a bitcasted value, it is returned as-is.
1646SDValue peekThroughBitcasts(SDValue V);
1647
1648/// Return the non-bitcasted and one-use source operand of \p V if it exists.
1649/// If \p V is not a bitcasted one-use value, it is returned as-is.
1650SDValue peekThroughOneUseBitcasts(SDValue V);
1651
1652/// Return the non-extracted vector source operand of \p V if it exists.
1653/// If \p V is not an extracted subvector, it is returned as-is.
1654SDValue peekThroughExtractSubvectors(SDValue V);
1655
1656/// Returns true if \p V is a bitwise not operation. Assumes that an all ones
1657/// constant is canonicalized to be operand 1.
1658bool isBitwiseNot(SDValue V, bool AllowUndefs = false);
1659
1660/// Returns the SDNode if it is a constant splat BuildVector or constant int.
1661ConstantSDNode *isConstOrConstSplat(SDValue N, bool AllowUndefs = false,
1662 bool AllowTruncation = false);
1663
1664/// Returns the SDNode if it is a demanded constant splat BuildVector or
1665/// constant int.
1666ConstantSDNode *isConstOrConstSplat(SDValue N, const APInt &DemandedElts,
1667 bool AllowUndefs = false,
1668 bool AllowTruncation = false);
1669
1670/// Returns the SDNode if it is a constant splat BuildVector or constant float.
1671ConstantFPSDNode *isConstOrConstSplatFP(SDValue N, bool AllowUndefs = false);
1672
1673/// Returns the SDNode if it is a demanded constant splat BuildVector or
1674/// constant float.
1675ConstantFPSDNode *isConstOrConstSplatFP(SDValue N, const APInt &DemandedElts,
1676 bool AllowUndefs = false);
1677
1678/// Return true if the value is a constant 0 integer or a splatted vector of
1679/// a constant 0 integer (with no undefs by default).
1680/// Build vector implicit truncation is not an issue for null values.
1681bool isNullOrNullSplat(SDValue V, bool AllowUndefs = false);
1682
1683/// Return true if the value is a constant 1 integer or a splatted vector of a
1684/// constant 1 integer (with no undefs).
1685/// Does not permit build vector implicit truncation.
1686bool isOneOrOneSplat(SDValue V, bool AllowUndefs = false);
1687
1688/// Return true if the value is a constant -1 integer or a splatted vector of a
1689/// constant -1 integer (with no undefs).
1690/// Does not permit build vector implicit truncation.
1691bool isAllOnesOrAllOnesSplat(SDValue V, bool AllowUndefs = false);
1692
1693/// Return true if \p V is either a integer or FP constant.
1694inline bool isIntOrFPConstant(SDValue V) {
1695 return isa<ConstantSDNode>(V) || isa<ConstantFPSDNode>(V);
1696}
1697
1698class GlobalAddressSDNode : public SDNode {
1699 friend class SelectionDAG;
1700
1701 const GlobalValue *TheGlobal;
1702 int64_t Offset;
1703 unsigned TargetFlags;
1704
1705 GlobalAddressSDNode(unsigned Opc, unsigned Order, const DebugLoc &DL,
1706 const GlobalValue *GA, EVT VT, int64_t o,
1707 unsigned TF);
1708
1709public:
1710 const GlobalValue *getGlobal() const { return TheGlobal; }
1711 int64_t getOffset() const { return Offset; }
1712 unsigned getTargetFlags() const { return TargetFlags; }
1713 // Return the address space this GlobalAddress belongs to.
1714 unsigned getAddressSpace() const;
1715
1716 static bool classof(const SDNode *N) {
1717 return N->getOpcode() == ISD::GlobalAddress ||
1718 N->getOpcode() == ISD::TargetGlobalAddress ||
1719 N->getOpcode() == ISD::GlobalTLSAddress ||
1720 N->getOpcode() == ISD::TargetGlobalTLSAddress;
1721 }
1722};
1723
1724class FrameIndexSDNode : public SDNode {
1725 friend class SelectionDAG;
1726
1727 int FI;
1728
1729 FrameIndexSDNode(int fi, EVT VT, bool isTarg)
1730 : SDNode(isTarg ? ISD::TargetFrameIndex : ISD::FrameIndex,
1731 0, DebugLoc(), getSDVTList(VT)), FI(fi) {
1732 }
1733
1734public:
1735 int getIndex() const { return FI; }
1736
1737 static bool classof(const SDNode *N) {
1738 return N->getOpcode() == ISD::FrameIndex ||
1739 N->getOpcode() == ISD::TargetFrameIndex;
1740 }
1741};
1742
1743/// This SDNode is used for LIFETIME_START/LIFETIME_END values, which indicate
1744/// the offet and size that are started/ended in the underlying FrameIndex.
1745class LifetimeSDNode : public SDNode {
1746 friend class SelectionDAG;
1747 int64_t Size;
1748 int64_t Offset; // -1 if offset is unknown.
1749
1750 LifetimeSDNode(unsigned Opcode, unsigned Order, const DebugLoc &dl,
1751 SDVTList VTs, int64_t Size, int64_t Offset)
1752 : SDNode(Opcode, Order, dl, VTs), Size(Size), Offset(Offset) {}
1753public:
1754 int64_t getFrameIndex() const {
1755 return cast<FrameIndexSDNode>(getOperand(1))->getIndex();
1756 }
1757
1758 bool hasOffset() const { return Offset >= 0; }
1759 int64_t getOffset() const {
1760 assert(hasOffset() && "offset is unknown")(static_cast <bool> (hasOffset() && "offset is unknown"
) ? void (0) : __assert_fail ("hasOffset() && \"offset is unknown\""
, "/build/llvm-toolchain-snapshot-13~++20210726100616+dead50d4427c/llvm/include/llvm/CodeGen/SelectionDAGNodes.h"
, 1760, __extension__ __PRETTY_FUNCTION__))
;
1761 return Offset;
1762 }
1763 int64_t getSize() const {
1764 assert(hasOffset() && "offset is unknown")(static_cast <bool> (hasOffset() && "offset is unknown"
) ? void (0) : __assert_fail ("hasOffset() && \"offset is unknown\""
, "/build/llvm-toolchain-snapshot-13~++20210726100616+dead50d4427c/llvm/include/llvm/CodeGen/SelectionDAGNodes.h"
, 1764, __extension__ __PRETTY_FUNCTION__))
;
1765 return Size;
1766 }
1767
1768 // Methods to support isa and dyn_cast
1769 static bool classof(const SDNode *N) {
1770 return N->getOpcode() == ISD::LIFETIME_START ||
1771 N->getOpcode() == ISD::LIFETIME_END;
1772 }
1773};
1774
1775/// This SDNode is used for PSEUDO_PROBE values, which are the function guid and
1776/// the index of the basic block being probed. A pseudo probe serves as a place
1777/// holder and will be removed at the end of compilation. It does not have any
1778/// operand because we do not want the instruction selection to deal with any.
1779class PseudoProbeSDNode : public SDNode {
1780 friend class SelectionDAG;
1781 uint64_t Guid;
1782 uint64_t Index;
1783 uint32_t Attributes;
1784
1785 PseudoProbeSDNode(unsigned Opcode, unsigned Order, const DebugLoc &Dl,
1786 SDVTList VTs, uint64_t Guid, uint64_t Index, uint32_t Attr)
1787 : SDNode(Opcode, Order, Dl, VTs), Guid(Guid), Index(Index),
1788 Attributes(Attr) {}
1789
1790public:
1791 uint64_t getGuid() const { return Guid; }
1792 uint64_t getIndex() const { return Index; }
1793 uint32_t getAttributes() const { return Attributes; }
1794
1795 // Methods to support isa and dyn_cast
1796 static bool classof(const SDNode *N) {
1797 return N->getOpcode() == ISD::PSEUDO_PROBE;
1798 }
1799};
1800
1801class JumpTableSDNode : public SDNode {
1802 friend class SelectionDAG;
1803
1804 int JTI;
1805 unsigned TargetFlags;
1806
1807 JumpTableSDNode(int jti, EVT VT, bool isTarg, unsigned TF)
1808 : SDNode(isTarg ? ISD::TargetJumpTable : ISD::JumpTable,
1809 0, DebugLoc(), getSDVTList(VT)), JTI(jti), TargetFlags(TF) {
1810 }
1811
1812public:
1813 int getIndex() const { return JTI; }
1814 unsigned getTargetFlags() const { return TargetFlags; }
1815
1816 static bool classof(const SDNode *N) {
1817 return N->getOpcode() == ISD::JumpTable ||
1818 N->getOpcode() == ISD::TargetJumpTable;
1819 }
1820};
1821
1822class ConstantPoolSDNode : public SDNode {
1823 friend class SelectionDAG;
1824
1825 union {
1826 const Constant *ConstVal;
1827 MachineConstantPoolValue *MachineCPVal;
1828 } Val;
1829 int Offset; // It's a MachineConstantPoolValue if top bit is set.
1830 Align Alignment; // Minimum alignment requirement of CP.
1831 unsigned TargetFlags;
1832
1833 ConstantPoolSDNode(bool isTarget, const Constant *c, EVT VT, int o,
1834 Align Alignment, unsigned TF)
1835 : SDNode(isTarget ? ISD::TargetConstantPool : ISD::ConstantPool, 0,
1836 DebugLoc(), getSDVTList(VT)),
1837 Offset(o), Alignment(Alignment), TargetFlags(TF) {
1838 assert(Offset >= 0 && "Offset is too large")(static_cast <bool> (Offset >= 0 && "Offset is too large"
) ? void (0) : __assert_fail ("Offset >= 0 && \"Offset is too large\""
, "/build/llvm-toolchain-snapshot-13~++20210726100616+dead50d4427c/llvm/include/llvm/CodeGen/SelectionDAGNodes.h"
, 1838, __extension__ __PRETTY_FUNCTION__))
;
1839 Val.ConstVal = c;
1840 }
1841
1842 ConstantPoolSDNode(bool isTarget, MachineConstantPoolValue *v, EVT VT, int o,
1843 Align Alignment, unsigned TF)
1844 : SDNode(isTarget ? ISD::TargetConstantPool : ISD::ConstantPool, 0,
1845 DebugLoc(), getSDVTList(VT)),
1846 Offset(o), Alignment(Alignment), TargetFlags(TF) {
1847 assert(Offset >= 0 && "Offset is too large")(static_cast <bool> (Offset >= 0 && "Offset is too large"
) ? void (0) : __assert_fail ("Offset >= 0 && \"Offset is too large\""
, "/build/llvm-toolchain-snapshot-13~++20210726100616+dead50d4427c/llvm/include/llvm/CodeGen/SelectionDAGNodes.h"
, 1847, __extension__ __PRETTY_FUNCTION__))
;
1848 Val.MachineCPVal = v;
1849 Offset |= 1 << (sizeof(unsigned)*CHAR_BIT8-1);
1850 }
1851
1852public:
1853 bool isMachineConstantPoolEntry() const {
1854 return Offset < 0;
1855 }
1856
1857 const Constant *getConstVal() const {
1858 assert(!isMachineConstantPoolEntry() && "Wrong constantpool type")(static_cast <bool> (!isMachineConstantPoolEntry() &&
"Wrong constantpool type") ? void (0) : __assert_fail ("!isMachineConstantPoolEntry() && \"Wrong constantpool type\""
, "/build/llvm-toolchain-snapshot-13~++20210726100616+dead50d4427c/llvm/include/llvm/CodeGen/SelectionDAGNodes.h"
, 1858, __extension__ __PRETTY_FUNCTION__))
;
1859 return Val.ConstVal;
1860 }
1861
1862 MachineConstantPoolValue *getMachineCPVal() const {
1863 assert(isMachineConstantPoolEntry() && "Wrong constantpool type")(static_cast <bool> (isMachineConstantPoolEntry() &&
"Wrong constantpool type") ? void (0) : __assert_fail ("isMachineConstantPoolEntry() && \"Wrong constantpool type\""
, "/build/llvm-toolchain-snapshot-13~++20210726100616+dead50d4427c/llvm/include/llvm/CodeGen/SelectionDAGNodes.h"
, 1863, __extension__ __PRETTY_FUNCTION__))
;
1864 return Val.MachineCPVal;
1865 }
1866
1867 int getOffset() const {
1868 return Offset & ~(1 << (sizeof(unsigned)*CHAR_BIT8-1));
1869 }
1870
1871 // Return the alignment of this constant pool object, which is either 0 (for
1872 // default alignment) or the desired value.
1873 Align getAlign() const { return Alignment; }
1874 unsigned getTargetFlags() const { return TargetFlags; }
1875
1876 Type *getType() const;
1877
1878 static bool classof(const SDNode *N) {
1879 return N->getOpcode() == ISD::ConstantPool ||
1880 N->getOpcode() == ISD::TargetConstantPool;
1881 }
1882};
1883
1884/// Completely target-dependent object reference.
1885class TargetIndexSDNode : public SDNode {
1886 friend class SelectionDAG;
1887
1888 unsigned TargetFlags;
1889 int Index;
1890 int64_t Offset;
1891
1892public:
1893 TargetIndexSDNode(int Idx, EVT VT, int64_t Ofs, unsigned TF)
1894 : SDNode(ISD::TargetIndex, 0, DebugLoc(), getSDVTList(VT)),
1895 TargetFlags(TF), Index(Idx), Offset(Ofs) {}
1896
1897 unsigned getTargetFlags() const { return TargetFlags; }
1898 int getIndex() const { return Index; }
1899 int64_t getOffset() const { return Offset; }
1900
1901 static bool classof(const SDNode *N) {
1902 return N->getOpcode() == ISD::TargetIndex;
1903 }
1904};
1905
1906class BasicBlockSDNode : public SDNode {
1907 friend class SelectionDAG;
1908
1909 MachineBasicBlock *MBB;
1910
1911 /// Debug info is meaningful and potentially useful here, but we create
1912 /// blocks out of order when they're jumped to, which makes it a bit
1913 /// harder. Let's see if we need it first.
1914 explicit BasicBlockSDNode(MachineBasicBlock *mbb)
1915 : SDNode(ISD::BasicBlock, 0, DebugLoc(), getSDVTList(MVT::Other)), MBB(mbb)
1916 {}
1917
1918public:
1919 MachineBasicBlock *getBasicBlock() const { return MBB; }
1920
1921 static bool classof(const SDNode *N) {
1922 return N->getOpcode() == ISD::BasicBlock;
1923 }
1924};
1925
1926/// A "pseudo-class" with methods for operating on BUILD_VECTORs.
1927class BuildVectorSDNode : public SDNode {
1928public:
1929 // These are constructed as SDNodes and then cast to BuildVectorSDNodes.
1930 explicit BuildVectorSDNode() = delete;
1931
1932 /// Check if this is a constant splat, and if so, find the
1933 /// smallest element size that splats the vector. If MinSplatBits is
1934 /// nonzero, the element size must be at least that large. Note that the
1935 /// splat element may be the entire vector (i.e., a one element vector).
1936 /// Returns the splat element value in SplatValue. Any undefined bits in
1937 /// that value are zero, and the corresponding bits in the SplatUndef mask
1938 /// are set. The SplatBitSize value is set to the splat element size in
1939 /// bits. HasAnyUndefs is set to true if any bits in the vector are
1940 /// undefined. isBigEndian describes the endianness of the target.
1941 bool isConstantSplat(APInt &SplatValue, APInt &SplatUndef,
1942 unsigned &SplatBitSize, bool &HasAnyUndefs,
1943 unsigned MinSplatBits = 0,
1944 bool isBigEndian = false) const;
1945
1946 /// Returns the demanded splatted value or a null value if this is not a
1947 /// splat.
1948 ///
1949 /// The DemandedElts mask indicates the elements that must be in the splat.
1950 /// If passed a non-null UndefElements bitvector, it will resize it to match
1951 /// the vector width and set the bits where elements are undef.
1952 SDValue getSplatValue(const APInt &DemandedElts,
1953 BitVector *UndefElements = nullptr) const;
1954
1955 /// Returns the splatted value or a null value if this is not a splat.
1956 ///
1957 /// If passed a non-null UndefElements bitvector, it will resize it to match
1958 /// the vector width and set the bits where elements are undef.
1959 SDValue getSplatValue(BitVector *UndefElements = nullptr) const;
1960
1961 /// Find the shortest repeating sequence of values in the build vector.
1962 ///
1963 /// e.g. { u, X, u, X, u, u, X, u } -> { X }
1964 /// { X, Y, u, Y, u, u, X, u } -> { X, Y }
1965 ///
1966 /// Currently this must be a power-of-2 build vector.
1967 /// The DemandedElts mask indicates the elements that must be present,
1968 /// undemanded elements in Sequence may be null (SDValue()). If passed a
1969 /// non-null UndefElements bitvector, it will resize it to match the original
1970 /// vector width and set the bits where elements are undef. If result is
1971 /// false, Sequence will be empty.
1972 bool getRepeatedSequence(const APInt &DemandedElts,
1973 SmallVectorImpl<SDValue> &Sequence,
1974 BitVector *UndefElements = nullptr) const;
1975
1976 /// Find the shortest repeating sequence of values in the build vector.
1977 ///
1978 /// e.g. { u, X, u, X, u, u, X, u } -> { X }
1979 /// { X, Y, u, Y, u, u, X, u } -> { X, Y }
1980 ///
1981 /// Currently this must be a power-of-2 build vector.
1982 /// If passed a non-null UndefElements bitvector, it will resize it to match
1983 /// the original vector width and set the bits where elements are undef.
1984 /// If result is false, Sequence will be empty.
1985 bool getRepeatedSequence(SmallVectorImpl<SDValue> &Sequence,
1986 BitVector *UndefElements = nullptr) const;
1987
1988 /// Returns the demanded splatted constant or null if this is not a constant
1989 /// splat.
1990 ///
1991 /// The DemandedElts mask indicates the elements that must be in the splat.
1992 /// If passed a non-null UndefElements bitvector, it will resize it to match
1993 /// the vector width and set the bits where elements are undef.
1994 ConstantSDNode *
1995 getConstantSplatNode(const APInt &DemandedElts,
1996 BitVector *UndefElements = nullptr) const;
1997
1998 /// Returns the splatted constant or null if this is not a constant
1999 /// splat.
2000 ///
2001 /// If passed a non-null UndefElements bitvector, it will resize it to match
2002 /// the vector width and set the bits where elements are undef.
2003 ConstantSDNode *
2004 getConstantSplatNode(BitVector *UndefElements = nullptr) const;
2005
2006 /// Returns the demanded splatted constant FP or null if this is not a
2007 /// constant FP splat.
2008 ///
2009 /// The DemandedElts mask indicates the elements that must be in the splat.
2010 /// If passed a non-null UndefElements bitvector, it will resize it to match
2011 /// the vector width and set the bits where elements are undef.
2012 ConstantFPSDNode *
2013 getConstantFPSplatNode(const APInt &DemandedElts,
2014 BitVector *UndefElements = nullptr) const;
2015
2016 /// Returns the splatted constant FP or null if this is not a constant
2017 /// FP splat.
2018 ///
2019 /// If passed a non-null UndefElements bitvector, it will resize it to match
2020 /// the vector width and set the bits where elements are undef.
2021 ConstantFPSDNode *
2022 getConstantFPSplatNode(BitVector *UndefElements = nullptr) const;
2023
2024 /// If this is a constant FP splat and the splatted constant FP is an
2025 /// exact power or 2, return the log base 2 integer value. Otherwise,
2026 /// return -1.
2027 ///
2028 /// The BitWidth specifies the necessary bit precision.
2029 int32_t getConstantFPSplatPow2ToLog2Int(BitVector *UndefElements,
2030 uint32_t BitWidth) const;
2031
2032 bool isConstant() const;
2033
2034 static bool classof(const SDNode *N) {
2035 return N->getOpcode() == ISD::BUILD_VECTOR;
2036 }
2037};
2038
2039/// An SDNode that holds an arbitrary LLVM IR Value. This is
2040/// used when the SelectionDAG needs to make a simple reference to something
2041/// in the LLVM IR representation.
2042///
2043class SrcValueSDNode : public SDNode {
2044 friend class SelectionDAG;
2045
2046 const Value *V;
2047
2048 /// Create a SrcValue for a general value.
2049 explicit SrcValueSDNode(const Value *v)
2050 : SDNode(ISD::SRCVALUE, 0, DebugLoc(), getSDVTList(MVT::Other)), V(v) {}
2051
2052public:
2053 /// Return the contained Value.
2054 const Value *getValue() const { return V; }
2055
2056 static bool classof(const SDNode *N) {
2057 return N->getOpcode() == ISD::SRCVALUE;
2058 }
2059};
2060
2061class MDNodeSDNode : public SDNode {
2062 friend class SelectionDAG;
2063
2064 const MDNode *MD;
2065
2066 explicit MDNodeSDNode(const MDNode *md)
2067 : SDNode(ISD::MDNODE_SDNODE, 0, DebugLoc(), getSDVTList(MVT::Other)), MD(md)
2068 {}
2069
2070public:
2071 const MDNode *getMD() const { return MD; }
2072
2073 static bool classof(const SDNode *N) {
2074 return N->getOpcode() == ISD::MDNODE_SDNODE;
2075 }
2076};
2077
2078class RegisterSDNode : public SDNode {
2079 friend class SelectionDAG;
2080
2081 Register Reg;
2082
2083 RegisterSDNode(Register reg, EVT VT)
2084 : SDNode(ISD::Register, 0, DebugLoc(), getSDVTList(VT)), Reg(reg) {}
2085
2086public:
2087 Register getReg() const { return Reg; }
2088
2089 static bool classof(const SDNode *N) {
2090 return N->getOpcode() == ISD::Register;
2091 }
2092};
2093
2094class RegisterMaskSDNode : public SDNode {
2095 friend class SelectionDAG;
2096
2097 // The memory for RegMask is not owned by the node.
2098 const uint32_t *RegMask;
2099
2100 RegisterMaskSDNode(const uint32_t *mask)
2101 : SDNode(ISD::RegisterMask, 0, DebugLoc(), getSDVTList(MVT::Untyped)),
2102 RegMask(mask) {}
2103
2104public:
2105 const uint32_t *getRegMask() const { return RegMask; }
2106
2107 static bool classof(const SDNode *N) {
2108 return N->getOpcode() == ISD::RegisterMask;
2109 }
2110};
2111
2112class BlockAddressSDNode : public SDNode {
2113 friend class SelectionDAG;
2114
2115 const BlockAddress *BA;
2116 int64_t Offset;
2117 unsigned TargetFlags;
2118
2119 BlockAddressSDNode(unsigned NodeTy, EVT VT, const BlockAddress *ba,
2120 int64_t o, unsigned Flags)
2121 : SDNode(NodeTy, 0, DebugLoc(), getSDVTList(VT)),
2122 BA(ba), Offset(o), TargetFlags(Flags) {}
2123
2124public:
2125 const BlockAddress *getBlockAddress() const { return BA; }
2126 int64_t getOffset() const { return Offset; }
2127 unsigned getTargetFlags() const { return TargetFlags; }
2128
2129 static bool classof(const SDNode *N) {
2130 return N->getOpcode() == ISD::BlockAddress ||
2131 N->getOpcode() == ISD::TargetBlockAddress;
2132 }
2133};
2134
2135class LabelSDNode : public SDNode {
2136 friend class SelectionDAG;
2137
2138 MCSymbol *Label;
2139
2140 LabelSDNode(unsigned Opcode, unsigned Order, const DebugLoc &dl, MCSymbol *L)
2141 : SDNode(Opcode, Order, dl, getSDVTList(MVT::Other)), Label(L) {
2142 assert(LabelSDNode::classof(this) && "not a label opcode")(static_cast <bool> (LabelSDNode::classof(this) &&
"not a label opcode") ? void (0) : __assert_fail ("LabelSDNode::classof(this) && \"not a label opcode\""
, "/build/llvm-toolchain-snapshot-13~++20210726100616+dead50d4427c/llvm/include/llvm/CodeGen/SelectionDAGNodes.h"
, 2142, __extension__ __PRETTY_FUNCTION__))
;
2143 }
2144
2145public:
2146 MCSymbol *getLabel() const { return Label; }
2147
2148 static bool classof(const SDNode *N) {
2149 return N->getOpcode() == ISD::EH_LABEL ||
2150 N->getOpcode() == ISD::ANNOTATION_LABEL;
2151 }
2152};
2153
2154class ExternalSymbolSDNode : public SDNode {
2155 friend class SelectionDAG;
2156
2157 const char *Symbol;
2158 unsigned TargetFlags;
2159
2160 ExternalSymbolSDNode(bool isTarget, const char *Sym, unsigned TF, EVT VT)
2161 : SDNode(isTarget ? ISD::TargetExternalSymbol : ISD::ExternalSymbol, 0,
2162 DebugLoc(), getSDVTList(VT)),
2163 Symbol(Sym), TargetFlags(TF) {}
2164
2165public:
2166 const char *getSymbol() const { return Symbol; }
2167 unsigned getTargetFlags() const { return TargetFlags; }
2168
2169 static bool classof(const SDNode *N) {
2170 return N->getOpcode() == ISD::ExternalSymbol ||
2171 N->getOpcode() == ISD::TargetExternalSymbol;
2172 }
2173};
2174
2175class MCSymbolSDNode : public SDNode {
2176 friend class SelectionDAG;
2177
2178 MCSymbol *Symbol;
2179
2180 MCSymbolSDNode(MCSymbol *Symbol, EVT VT)
2181 : SDNode(ISD::MCSymbol, 0, DebugLoc(), getSDVTList(VT)), Symbol(Symbol) {}
2182
2183public:
2184 MCSymbol *getMCSymbol() const { return Symbol; }
2185
2186 static bool classof(const SDNode *N) {
2187 return N->getOpcode() == ISD::MCSymbol;
2188 }
2189};
2190
2191class CondCodeSDNode : public SDNode {
2192 friend class SelectionDAG;
2193
2194 ISD::CondCode Condition;
2195
2196 explicit CondCodeSDNode(ISD::CondCode Cond)
2197 : SDNode(ISD::CONDCODE, 0, DebugLoc(), getSDVTList(MVT::Other)),
2198 Condition(Cond) {}
2199
2200public:
2201 ISD::CondCode get() const { return Condition; }
2202
2203 static bool classof(const SDNode *N) {
2204 return N->getOpcode() == ISD::CONDCODE;
2205 }
2206};
2207
2208/// This class is used to represent EVT's, which are used
2209/// to parameterize some operations.
2210class VTSDNode : public SDNode {
2211 friend class SelectionDAG;
2212
2213 EVT ValueType;
2214
2215 explicit VTSDNode(EVT VT)
2216 : SDNode(ISD::VALUETYPE, 0, DebugLoc(), getSDVTList(MVT::Other)),
2217 ValueType(VT) {}
2218
2219public:
2220 EVT getVT() const { return ValueType; }
2221
2222 static bool classof(const SDNode *N) {
2223 return N->getOpcode() == ISD::VALUETYPE;
2224 }
2225};
2226
2227/// Base class for LoadSDNode and StoreSDNode
2228class LSBaseSDNode : public MemSDNode {
2229public:
2230 LSBaseSDNode(ISD::NodeType NodeTy, unsigned Order, const DebugLoc &dl,
2231 SDVTList VTs, ISD::MemIndexedMode AM, EVT MemVT,
2232 MachineMemOperand *MMO)
2233 : MemSDNode(NodeTy, Order, dl, VTs, MemVT, MMO) {
2234 LSBaseSDNodeBits.AddressingMode = AM;
2235 assert(getAddressingMode() == AM && "Value truncated")(static_cast <bool> (getAddressingMode() == AM &&
"Value truncated") ? void (0) : __assert_fail ("getAddressingMode() == AM && \"Value truncated\""
, "/build/llvm-toolchain-snapshot-13~++20210726100616+dead50d4427c/llvm/include/llvm/CodeGen/SelectionDAGNodes.h"
, 2235, __extension__ __PRETTY_FUNCTION__))
;
2236 }
2237
2238 const SDValue &getOffset() const {
2239 return getOperand(getOpcode() == ISD::LOAD ? 2 : 3);
2240 }
2241
2242 /// Return the addressing mode for this load or store:
2243 /// unindexed, pre-inc, pre-dec, post-inc, or post-dec.
2244 ISD::MemIndexedMode getAddressingMode() const {
2245 return static_cast<ISD::MemIndexedMode>(LSBaseSDNodeBits.AddressingMode);
2246 }
2247
2248 /// Return true if this is a pre/post inc/dec load/store.
2249 bool isIndexed() const { return getAddressingMode() != ISD::UNINDEXED; }
2250
2251 /// Return true if this is NOT a pre/post inc/dec load/store.
2252 bool isUnindexed() const { return getAddressingMode() == ISD::UNINDEXED; }
2253
2254 static bool classof(const SDNode *N) {
2255 return N->getOpcode() == ISD::LOAD ||
2256 N->getOpcode() == ISD::STORE;
2257 }
2258};
2259
2260/// This class is used to represent ISD::LOAD nodes.
2261class LoadSDNode : public LSBaseSDNode {
2262 friend class SelectionDAG;
2263
2264 LoadSDNode(unsigned Order, const DebugLoc &dl, SDVTList VTs,
2265 ISD::MemIndexedMode AM, ISD::LoadExtType ETy, EVT MemVT,
2266 MachineMemOperand *MMO)
2267 : LSBaseSDNode(ISD::LOAD, Order, dl, VTs, AM, MemVT, MMO) {
2268 LoadSDNodeBits.ExtTy = ETy;
2269 assert(readMem() && "Load MachineMemOperand is not a load!")(static_cast <bool> (readMem() && "Load MachineMemOperand is not a load!"
) ? void (0) : __assert_fail ("readMem() && \"Load MachineMemOperand is not a load!\""
, "/build/llvm-toolchain-snapshot-13~++20210726100616+dead50d4427c/llvm/include/llvm/CodeGen/SelectionDAGNodes.h"
, 2269, __extension__ __PRETTY_FUNCTION__))
;
2270 assert(!writeMem() && "Load MachineMemOperand is a store!")(static_cast <bool> (!writeMem() && "Load MachineMemOperand is a store!"
) ? void (0) : __assert_fail ("!writeMem() && \"Load MachineMemOperand is a store!\""
, "/build/llvm-toolchain-snapshot-13~++20210726100616+dead50d4427c/llvm/include/llvm/CodeGen/SelectionDAGNodes.h"
, 2270, __extension__ __PRETTY_FUNCTION__))
;
2271 }
2272
2273public:
2274 /// Return whether this is a plain node,
2275 /// or one of the varieties of value-extending loads.
2276 ISD::LoadExtType getExtensionType() const {
2277 return static_cast<ISD::LoadExtType>(LoadSDNodeBits.ExtTy);
2278 }
2279
2280 const SDValue &getBasePtr() const { return getOperand(1); }
2281 const SDValue &getOffset() const { return getOperand(2); }
2282
2283 static bool classof(const SDNode *N) {
2284 return N->getOpcode() == ISD::LOAD;
2285 }
2286};
2287
2288/// This class is used to represent ISD::STORE nodes.
2289class StoreSDNode : public LSBaseSDNode {
2290 friend class SelectionDAG;
2291
2292 StoreSDNode(unsigned Order, const DebugLoc &dl, SDVTList VTs,
2293 ISD::MemIndexedMode AM, bool isTrunc, EVT MemVT,
2294 MachineMemOperand *MMO)
2295 : LSBaseSDNode(ISD::STORE, Order, dl, VTs, AM, MemVT, MMO) {
2296 StoreSDNodeBits.IsTruncating = isTrunc;
2297 assert(!readMem() && "Store MachineMemOperand is a load!")(static_cast <bool> (!readMem() && "Store MachineMemOperand is a load!"
) ? void (0) : __assert_fail ("!readMem() && \"Store MachineMemOperand is a load!\""
, "/build/llvm-toolchain-snapshot-13~++20210726100616+dead50d4427c/llvm/include/llvm/CodeGen/SelectionDAGNodes.h"
, 2297, __extension__ __PRETTY_FUNCTION__))
;
2298 assert(writeMem() && "Store MachineMemOperand is not a store!")(static_cast <bool> (writeMem() && "Store MachineMemOperand is not a store!"
) ? void (0) : __assert_fail ("writeMem() && \"Store MachineMemOperand is not a store!\""
, "/build/llvm-toolchain-snapshot-13~++20210726100616+dead50d4427c/llvm/include/llvm/CodeGen/SelectionDAGNodes.h"
, 2298, __extension__ __PRETTY_FUNCTION__))
;
2299 }
2300
2301public:
2302 /// Return true if the op does a truncation before store.
2303 /// For integers this is the same as doing a TRUNCATE and storing the result.
2304 /// For floats, it is the same as doing an FP_ROUND and storing the result.
2305 bool isTruncatingStore() const { return StoreSDNodeBits.IsTruncating; }
2306 void setTruncatingStore(bool Truncating) {
2307 StoreSDNodeBits.IsTruncating = Truncating;
2308 }
2309
2310 const SDValue &getValue() const { return getOperand(1); }
2311 const SDValue &getBasePtr() const { return getOperand(2); }
2312 const SDValue &getOffset() const { return getOperand(3); }
2313
2314 static bool classof(const SDNode *N) {
2315 return N->getOpcode() == ISD::STORE;
2316 }
2317};
2318
2319/// This base class is used to represent MLOAD and MSTORE nodes
2320class MaskedLoadStoreSDNode : public MemSDNode {
2321public:
2322 friend class SelectionDAG;
2323
2324 MaskedLoadStoreSDNode(ISD::NodeType NodeTy, unsigned Order,
2325 const DebugLoc &dl, SDVTList VTs,
2326 ISD::MemIndexedMode AM, EVT MemVT,
2327 MachineMemOperand *MMO)
2328 : MemSDNode(NodeTy, Order, dl, VTs, MemVT, MMO) {
2329 LSBaseSDNodeBits.AddressingMode = AM;
2330 assert(getAddressingMode() == AM && "Value truncated")(static_cast <bool> (getAddressingMode() == AM &&
"Value truncated") ? void (0) : __assert_fail ("getAddressingMode() == AM && \"Value truncated\""
, "/build/llvm-toolchain-snapshot-13~++20210726100616+dead50d4427c/llvm/include/llvm/CodeGen/SelectionDAGNodes.h"
, 2330, __extension__ __PRETTY_FUNCTION__))
;
2331 }
2332
2333 // MaskedLoadSDNode (Chain, ptr, offset, mask, passthru)
2334 // MaskedStoreSDNode (Chain, data, ptr, offset, mask)
2335 // Mask is a vector of i1 elements
2336 const SDValue &getOffset() const {
2337 return getOperand(getOpcode() == ISD::MLOAD ? 2 : 3);
2338 }
2339 const SDValue &getMask() const {
2340 return getOperand(getOpcode() == ISD::MLOAD ? 3 : 4);
2341 }
2342
2343 /// Return the addressing mode for this load or store:
2344 /// unindexed, pre-inc, pre-dec, post-inc, or post-dec.
2345 ISD::MemIndexedMode getAddressingMode() const {
2346 return static_cast<ISD::MemIndexedMode>(LSBaseSDNodeBits.AddressingMode);
2347 }
2348
2349 /// Return true if this is a pre/post inc/dec load/store.
2350 bool isIndexed() const { return getAddressingMode() != ISD::UNINDEXED; }
2351
2352 /// Return true if this is NOT a pre/post inc/dec load/store.
2353 bool isUnindexed() const { return getAddressingMode() == ISD::UNINDEXED; }
2354
2355 static bool classof(const SDNode *N) {
2356 return N->getOpcode() == ISD::MLOAD ||
2357 N->getOpcode() == ISD::MSTORE;
2358 }
2359};
2360
2361/// This class is used to represent an MLOAD node
2362class MaskedLoadSDNode : public MaskedLoadStoreSDNode {
2363public:
2364 friend class SelectionDAG;
2365
2366 MaskedLoadSDNode(unsigned Order, const DebugLoc &dl, SDVTList VTs,
2367 ISD::MemIndexedMode AM, ISD::LoadExtType ETy,
2368 bool IsExpanding, EVT MemVT, MachineMemOperand *MMO)
2369 : MaskedLoadStoreSDNode(ISD::MLOAD, Order, dl, VTs, AM, MemVT, MMO) {
2370 LoadSDNodeBits.ExtTy = ETy;
2371 LoadSDNodeBits.IsExpanding = IsExpanding;
2372 }
2373
2374 ISD::LoadExtType getExtensionType() const {
2375 return static_cast<ISD::LoadExtType>(LoadSDNodeBits.ExtTy);
2376 }
2377
2378 const SDValue &getBasePtr() const { return getOperand(1); }
2379 const SDValue &getOffset() const { return getOperand(2); }
2380 const SDValue &getMask() const { return getOperand(3); }
2381 const SDValue &getPassThru() const { return getOperand(4); }
2382
2383 static bool classof(const SDNode *N) {
2384 return N->getOpcode() == ISD::MLOAD;
2385 }
2386
2387 bool isExpandingLoad() const { return LoadSDNodeBits.IsExpanding; }
2388};
2389
2390/// This class is used to represent an MSTORE node
2391class MaskedStoreSDNode : public MaskedLoadStoreSDNode {
2392public:
2393 friend class SelectionDAG;
2394
2395 MaskedStoreSDNode(unsigned Order, const DebugLoc &dl, SDVTList VTs,
2396 ISD::MemIndexedMode AM, bool isTrunc, bool isCompressing,
2397 EVT MemVT, MachineMemOperand *MMO)
2398 : MaskedLoadStoreSDNode(ISD::MSTORE, Order, dl, VTs, AM, MemVT, MMO) {
2399 StoreSDNodeBits.IsTruncating = isTrunc;
2400 StoreSDNodeBits.IsCompressing = isCompressing;
2401 }
2402
2403 /// Return true if the op does a truncation before store.
2404 /// For integers this is the same as doing a TRUNCATE and storing the result.
2405 /// For floats, it is the same as doing an FP_ROUND and storing the result.
2406 bool isTruncatingStore() const { return StoreSDNodeBits.IsTruncating; }
2407
2408 /// Returns true if the op does a compression to the vector before storing.
2409 /// The node contiguously stores the active elements (integers or floats)
2410 /// in src (those with their respective bit set in writemask k) to unaligned
2411 /// memory at base_addr.
2412 bool isCompressingStore() const { return StoreSDNodeBits.IsCompressing; }
2413
2414 const SDValue &getValue() const { return getOperand(1); }
2415 const SDValue &getBasePtr() const { return getOperand(2); }
2416 const SDValue &getOffset() const { return getOperand(3); }
2417 const SDValue &getMask() const { return getOperand(4); }
2418
2419 static bool classof(const SDNode *N) {
2420 return N->getOpcode() == ISD::MSTORE;
2421 }
2422};
2423
2424/// This is a base class used to represent
2425/// MGATHER and MSCATTER nodes
2426///
2427class MaskedGatherScatterSDNode : public MemSDNode {
2428public:
2429 friend class SelectionDAG;
2430
2431 MaskedGatherScatterSDNode(ISD::NodeType NodeTy, unsigned Order,
2432 const DebugLoc &dl, SDVTList VTs, EVT MemVT,
2433 MachineMemOperand *MMO, ISD::MemIndexType IndexType)
2434 : MemSDNode(NodeTy, Order, dl, VTs, MemVT, MMO) {
2435 LSBaseSDNodeBits.AddressingMode = IndexType;
2436 assert(getIndexType() == IndexType && "Value truncated")(static_cast <bool> (getIndexType() == IndexType &&
"Value truncated") ? void (0) : __assert_fail ("getIndexType() == IndexType && \"Value truncated\""
, "/build/llvm-toolchain-snapshot-13~++20210726100616+dead50d4427c/llvm/include/llvm/CodeGen/SelectionDAGNodes.h"
, 2436, __extension__ __PRETTY_FUNCTION__))
;
2437 }
2438
2439 /// How is Index applied to BasePtr when computing addresses.
2440 ISD::MemIndexType getIndexType() const {
2441 return static_cast<ISD::MemIndexType>(LSBaseSDNodeBits.AddressingMode);
2442 }
2443 void setIndexType(ISD::MemIndexType IndexType) {
2444 LSBaseSDNodeBits.AddressingMode = IndexType;
2445 }
2446 bool isIndexScaled() const {
2447 return (getIndexType() == ISD::SIGNED_SCALED) ||
2448 (getIndexType() == ISD::UNSIGNED_SCALED);
2449 }
2450 bool isIndexSigned() const {
2451 return (getIndexType() == ISD::SIGNED_SCALED) ||
2452 (getIndexType() == ISD::SIGNED_UNSCALED);
2453 }
2454
2455 // In the both nodes address is Op1, mask is Op2:
2456 // MaskedGatherSDNode (Chain, passthru, mask, base, index, scale)
2457 // MaskedScatterSDNode (Chain, value, mask, base, index, scale)
2458 // Mask is a vector of i1 elements
2459 const SDValue &getBasePtr() const { return getOperand(3); }
2460 const SDValue &getIndex() const { return getOperand(4); }
2461 const SDValue &getMask() const { return getOperand(2); }
2462 const SDValue &getScale() const { return getOperand(5); }
2463
2464 static bool classof(const SDNode *N) {
2465 return N->getOpcode() == ISD::MGATHER ||
2466 N->getOpcode() == ISD::MSCATTER;
2467 }
2468};
2469
2470/// This class is used to represent an MGATHER node
2471///
2472class MaskedGatherSDNode : public MaskedGatherScatterSDNode {
2473public:
2474 friend class SelectionDAG;
2475
2476 MaskedGatherSDNode(unsigned Order, const DebugLoc &dl, SDVTList VTs,
2477 EVT MemVT, MachineMemOperand *MMO,
2478 ISD::MemIndexType IndexType, ISD::LoadExtType ETy)
2479 : MaskedGatherScatterSDNode(ISD::MGATHER, Order, dl, VTs, MemVT, MMO,
2480 IndexType) {
2481 LoadSDNodeBits.ExtTy = ETy;
2482 }
2483
2484 const SDValue &getPassThru() const { return getOperand(1); }
2485
2486 ISD::LoadExtType getExtensionType() const {
2487 return ISD::LoadExtType(LoadSDNodeBits.ExtTy);
2488 }
2489
2490 static bool classof(const SDNode *N) {
2491 return N->getOpcode() == ISD::MGATHER;
2492 }
2493};
2494
2495/// This class is used to represent an MSCATTER node
2496///
2497class MaskedScatterSDNode : public MaskedGatherScatterSDNode {
2498public:
2499 friend class SelectionDAG;
2500
2501 MaskedScatterSDNode(unsigned Order, const DebugLoc &dl, SDVTList VTs,
2502 EVT MemVT, MachineMemOperand *MMO,
2503 ISD::MemIndexType IndexType, bool IsTrunc)
2504 : MaskedGatherScatterSDNode(ISD::MSCATTER, Order, dl, VTs, MemVT, MMO,
2505 IndexType) {
2506 StoreSDNodeBits.IsTruncating = IsTrunc;
2507 }
2508
2509 /// Return true if the op does a truncation before store.
2510 /// For integers this is the same as doing a TRUNCATE and storing the result.
2511 /// For floats, it is the same as doing an FP_ROUND and storing the result.
2512 bool isTruncatingStore() const { return StoreSDNodeBits.IsTruncating; }
2513
2514 const SDValue &getValue() const { return getOperand(1); }
2515
2516 static bool classof(const SDNode *N) {
2517 return N->getOpcode() == ISD::MSCATTER;
2518 }
2519};
2520
2521/// An SDNode that represents everything that will be needed
2522/// to construct a MachineInstr. These nodes are created during the
2523/// instruction selection proper phase.
2524///
2525/// Note that the only supported way to set the `memoperands` is by calling the
2526/// `SelectionDAG::setNodeMemRefs` function as the memory management happens
2527/// inside the DAG rather than in the node.
2528class MachineSDNode : public SDNode {
2529private:
2530 friend class SelectionDAG;
2531
2532 MachineSDNode(unsigned Opc, unsigned Order, const DebugLoc &DL, SDVTList VTs)
2533 : SDNode(Opc, Order, DL, VTs) {}
2534
2535 // We use a pointer union between a single `MachineMemOperand` pointer and
2536 // a pointer to an array of `MachineMemOperand` pointers. This is null when
2537 // the number of these is zero, the single pointer variant used when the
2538 // number is one, and the array is used for larger numbers.
2539 //
2540 // The array is allocated via the `SelectionDAG`'s allocator and so will
2541 // always live until the DAG is cleaned up and doesn't require ownership here.
2542 //
2543 // We can't use something simpler like `TinyPtrVector` here because `SDNode`
2544 // subclasses aren't managed in a conforming C++ manner. See the comments on
2545 // `SelectionDAG::MorphNodeTo` which details what all goes on, but the
2546 // constraint here is that these don't manage memory with their constructor or
2547 // destructor and can be initialized to a good state even if they start off
2548 // uninitialized.
2549 PointerUnion<MachineMemOperand *, MachineMemOperand **> MemRefs = {};
2550
2551 // Note that this could be folded into the above `MemRefs` member if doing so
2552 // is advantageous at some point. We don't need to store this in most cases.
2553 // However, at the moment this doesn't appear to make the allocation any
2554 // smaller and makes the code somewhat simpler to read.
2555 int NumMemRefs = 0;
2556
2557public:
2558 using mmo_iterator = ArrayRef<MachineMemOperand *>::const_iterator;
2559
2560 ArrayRef<MachineMemOperand *> memoperands() const {
2561 // Special case the common cases.
2562 if (NumMemRefs == 0)
2563 return {};
2564 if (NumMemRefs == 1)
2565 return makeArrayRef(MemRefs.getAddrOfPtr1(), 1);
2566
2567 // Otherwise we have an actual array.
2568 return makeArrayRef(MemRefs.get<MachineMemOperand **>(), NumMemRefs);
2569 }
2570 mmo_iterator memoperands_begin() const { return memoperands().begin(); }
2571 mmo_iterator memoperands_end() const { return memoperands().end(); }
2572 bool memoperands_empty() const { return memoperands().empty(); }
2573
2574 /// Clear out the memory reference descriptor list.
2575 void clearMemRefs() {
2576 MemRefs = nullptr;
2577 NumMemRefs = 0;
2578 }
2579
2580 static bool classof(const SDNode *N) {
2581 return N->isMachineOpcode();
2582 }
2583};
2584
2585/// An SDNode that records if a register contains a value that is guaranteed to
2586/// be aligned accordingly.
2587class AssertAlignSDNode : public SDNode {
2588 Align Alignment;
2589
2590public:
2591 AssertAlignSDNode(unsigned Order, const DebugLoc &DL, EVT VT, Align A)
2592 : SDNode(ISD::AssertAlign, Order, DL, getSDVTList(VT)), Alignment(A) {}
2593
2594 Align getAlign() const { return Alignment; }
2595
2596 static bool classof(const SDNode *N) {
2597 return N->getOpcode() == ISD::AssertAlign;
2598 }
2599};
2600
2601class SDNodeIterator {
2602 const SDNode *Node;
2603 unsigned Operand;
2604
2605 SDNodeIterator(const SDNode *N, unsigned Op) : Node(N), Operand(Op) {}
2606
2607public:
2608 using iterator_category = std::forward_iterator_tag;
2609 using value_type = SDNode;
2610 using difference_type = std::ptrdiff_t;
2611 using pointer = value_type *;
2612 using reference = value_type &;
2613
2614 bool operator==(const SDNodeIterator& x) const {
2615 return Operand == x.Operand;
2616 }
2617 bool operator!=(const SDNodeIterator& x) const { return !operator==(x); }
2618
2619 pointer operator*() const {
2620 return Node->getOperand(Operand).getNode();
2621 }
2622 pointer operator->() const { return operator*(); }
2623
2624 SDNodeIterator& operator++() { // Preincrement
2625 ++Operand;
2626 return *this;
2627 }
2628 SDNodeIterator operator++(int) { // Postincrement
2629 SDNodeIterator tmp = *this; ++*this; return tmp;
2630 }
2631 size_t operator-(SDNodeIterator Other) const {
2632 assert(Node == Other.Node &&(static_cast <bool> (Node == Other.Node && "Cannot compare iterators of two different nodes!"
) ? void (0) : __assert_fail ("Node == Other.Node && \"Cannot compare iterators of two different nodes!\""
, "/build/llvm-toolchain-snapshot-13~++20210726100616+dead50d4427c/llvm/include/llvm/CodeGen/SelectionDAGNodes.h"
, 2633, __extension__ __PRETTY_FUNCTION__))
2633 "Cannot compare iterators of two different nodes!")(static_cast <bool> (Node == Other.Node && "Cannot compare iterators of two different nodes!"
) ? void (0) : __assert_fail ("Node == Other.Node && \"Cannot compare iterators of two different nodes!\""
, "/build/llvm-toolchain-snapshot-13~++20210726100616+dead50d4427c/llvm/include/llvm/CodeGen/SelectionDAGNodes.h"
, 2633, __extension__ __PRETTY_FUNCTION__))
;
2634 return Operand - Other.Operand;
2635 }
2636
2637 static SDNodeIterator begin(const SDNode *N) { return SDNodeIterator(N, 0); }
2638 static SDNodeIterator end (const SDNode *N) {
2639 return SDNodeIterator(N, N->getNumOperands());
2640 }
2641
2642 unsigned getOperand() const { return Operand; }
2643 const SDNode *getNode() const { return Node; }
2644};
2645
2646template <> struct GraphTraits<SDNode*> {
2647 using NodeRef = SDNode *;
2648 using ChildIteratorType = SDNodeIterator;
2649
2650 static NodeRef getEntryNode(SDNode *N) { return N; }
2651
2652 static ChildIteratorType child_begin(NodeRef N) {
2653 return SDNodeIterator::begin(N);
2654 }
2655
2656 static ChildIteratorType child_end(NodeRef N) {
2657 return SDNodeIterator::end(N);
2658 }
2659};
2660
2661/// A representation of the largest SDNode, for use in sizeof().
2662///
2663/// This needs to be a union because the largest node differs on 32 bit systems
2664/// with 4 and 8 byte pointer alignment, respectively.
2665using LargestSDNode = AlignedCharArrayUnion<AtomicSDNode, TargetIndexSDNode,
2666 BlockAddressSDNode,
2667 GlobalAddressSDNode,
2668 PseudoProbeSDNode>;
2669
2670/// The SDNode class with the greatest alignment requirement.
2671using MostAlignedSDNode = GlobalAddressSDNode;
2672
2673namespace ISD {
2674
2675 /// Returns true if the specified node is a non-extending and unindexed load.
2676 inline bool isNormalLoad(const SDNode *N) {
2677 const LoadSDNode *Ld = dyn_cast<LoadSDNode>(N);
2678 return Ld && Ld->getExtensionType() == ISD::NON_EXTLOAD &&
2679 Ld->getAddressingMode() == ISD::UNINDEXED;
2680 }
2681
2682 /// Returns true if the specified node is a non-extending load.
2683 inline bool isNON_EXTLoad(const SDNode *N) {
2684 return isa<LoadSDNode>(N) &&
2685 cast<LoadSDNode>(N)->getExtensionType() == ISD::NON_EXTLOAD;
2686 }
2687
2688 /// Returns true if the specified node is a EXTLOAD.
2689 inline bool isEXTLoad(const SDNode *N) {
2690 return isa<LoadSDNode>(N) &&
2691 cast<LoadSDNode>(N)->getExtensionType() == ISD::EXTLOAD;
2692 }
2693
2694 /// Returns true if the specified node is a SEXTLOAD.
2695 inline bool isSEXTLoad(const SDNode *N) {
2696 return isa<LoadSDNode>(N) &&
2697 cast<LoadSDNode>(N)->getExtensionType() == ISD::SEXTLOAD;
2698 }
2699
2700 /// Returns true if the specified node is a ZEXTLOAD.
2701 inline bool isZEXTLoad(const SDNode *N) {
2702 return isa<LoadSDNode>(N) &&
2703 cast<LoadSDNode>(N)->getExtensionType() == ISD::ZEXTLOAD;
2704 }
2705
2706 /// Returns true if the specified node is an unindexed load.
2707 inline bool isUNINDEXEDLoad(const SDNode *N) {
2708 return isa<LoadSDNode>(N) &&
2709 cast<LoadSDNode>(N)->getAddressingMode() == ISD::UNINDEXED;
2710 }
2711
2712 /// Returns true if the specified node is a non-truncating
2713 /// and unindexed store.
2714 inline bool isNormalStore(const SDNode *N) {
2715 const StoreSDNode *St = dyn_cast<StoreSDNode>(N);
2716 return St && !St->isTruncatingStore() &&
2717 St->getAddressingMode() == ISD::UNINDEXED;
2718 }
2719
2720 /// Returns true if the specified node is an unindexed store.
2721 inline bool isUNINDEXEDStore(const SDNode *N) {
2722 return isa<StoreSDNode>(N) &&
2723 cast<StoreSDNode>(N)->getAddressingMode() == ISD::UNINDEXED;
2724 }
2725
2726 /// Attempt to match a unary predicate against a scalar/splat constant or
2727 /// every element of a constant BUILD_VECTOR.
2728 /// If AllowUndef is true, then UNDEF elements will pass nullptr to Match.
2729 bool matchUnaryPredicate(SDValue Op,
2730 std::function<bool(ConstantSDNode *)> Match,
2731 bool AllowUndefs = false);
2732
2733 /// Attempt to match a binary predicate against a pair of scalar/splat
2734 /// constants or every element of a pair of constant BUILD_VECTORs.
2735 /// If AllowUndef is true, then UNDEF elements will pass nullptr to Match.
2736 /// If AllowTypeMismatch is true then RetType + ArgTypes don't need to match.
2737 bool matchBinaryPredicate(
2738 SDValue LHS, SDValue RHS,
2739 std::function<bool(ConstantSDNode *, ConstantSDNode *)> Match,
2740 bool AllowUndefs = false, bool AllowTypeMismatch = false);
2741
2742 /// Returns true if the specified value is the overflow result from one
2743 /// of the overflow intrinsic nodes.
2744 inline bool isOverflowIntrOpRes(SDValue Op) {
2745 unsigned Opc = Op.getOpcode();
2746 return (Op.getResNo() == 1 &&
2747 (Opc == ISD::SADDO || Opc == ISD::UADDO || Opc == ISD::SSUBO ||
2748 Opc == ISD::USUBO || Opc == ISD::SMULO || Opc == ISD::UMULO));
2749 }
2750
2751} // end namespace ISD
2752
2753} // end namespace llvm
2754
2755#endif // LLVM_CODEGEN_SELECTIONDAGNODES_H