Dan Gohman | 10e730a | 2015-06-29 23:51:55 +0000 | [diff] [blame] | 1 | //=- WebAssemblyISelLowering.cpp - WebAssembly DAG Lowering Implementation -==// |
| 2 | // |
| 3 | // The LLVM Compiler Infrastructure |
| 4 | // |
| 5 | // This file is distributed under the University of Illinois Open Source |
| 6 | // License. See LICENSE.TXT for details. |
| 7 | // |
| 8 | //===----------------------------------------------------------------------===// |
| 9 | /// |
| 10 | /// \file |
Adrian Prantl | 5f8f34e4 | 2018-05-01 15:54:18 +0000 | [diff] [blame] | 11 | /// This file implements the WebAssemblyTargetLowering class. |
Dan Gohman | 10e730a | 2015-06-29 23:51:55 +0000 | [diff] [blame] | 12 | /// |
| 13 | //===----------------------------------------------------------------------===// |
| 14 | |
| 15 | #include "WebAssemblyISelLowering.h" |
| 16 | #include "MCTargetDesc/WebAssemblyMCTargetDesc.h" |
| 17 | #include "WebAssemblyMachineFunctionInfo.h" |
| 18 | #include "WebAssemblySubtarget.h" |
| 19 | #include "WebAssemblyTargetMachine.h" |
Dan Gohman | 10e730a | 2015-06-29 23:51:55 +0000 | [diff] [blame] | 20 | #include "llvm/CodeGen/Analysis.h" |
JF Bastien | af111db | 2015-08-24 22:16:48 +0000 | [diff] [blame] | 21 | #include "llvm/CodeGen/CallingConvLower.h" |
Dan Gohman | cdd48b8 | 2017-11-28 01:13:40 +0000 | [diff] [blame] | 22 | #include "llvm/CodeGen/MachineInstrBuilder.h" |
Dan Gohman | 950a13c | 2015-09-16 16:51:30 +0000 | [diff] [blame] | 23 | #include "llvm/CodeGen/MachineJumpTableInfo.h" |
Dan Gohman | 10e730a | 2015-06-29 23:51:55 +0000 | [diff] [blame] | 24 | #include "llvm/CodeGen/MachineRegisterInfo.h" |
| 25 | #include "llvm/CodeGen/SelectionDAG.h" |
Oliver Stannard | 02fa1c8 | 2016-01-28 13:19:47 +0000 | [diff] [blame] | 26 | #include "llvm/IR/DiagnosticInfo.h" |
JF Bastien | b9073fb | 2015-07-22 21:28:15 +0000 | [diff] [blame] | 27 | #include "llvm/IR/DiagnosticPrinter.h" |
Dan Gohman | 10e730a | 2015-06-29 23:51:55 +0000 | [diff] [blame] | 28 | #include "llvm/IR/Function.h" |
| 29 | #include "llvm/IR/Intrinsics.h" |
Dan Gohman | 10e730a | 2015-06-29 23:51:55 +0000 | [diff] [blame] | 30 | #include "llvm/Support/Debug.h" |
| 31 | #include "llvm/Support/ErrorHandling.h" |
| 32 | #include "llvm/Support/raw_ostream.h" |
| 33 | #include "llvm/Target/TargetOptions.h" |
| 34 | using namespace llvm; |
| 35 | |
| 36 | #define DEBUG_TYPE "wasm-lower" |
| 37 | |
Heejin Ahn | 5831e9c | 2018-08-09 23:58:51 +0000 | [diff] [blame] | 38 | // Emit proposed instructions that may not have been implemented in engines |
| 39 | cl::opt<bool> EnableUnimplementedWasmSIMDInstrs( |
| 40 | "wasm-enable-unimplemented-simd", |
| 41 | cl::desc("Emit potentially-unimplemented WebAssembly SIMD instructions"), |
| 42 | cl::init(false)); |
| 43 | |
Dan Gohman | 10e730a | 2015-06-29 23:51:55 +0000 | [diff] [blame] | 44 | WebAssemblyTargetLowering::WebAssemblyTargetLowering( |
| 45 | const TargetMachine &TM, const WebAssemblySubtarget &STI) |
Dan Gohman | bfaf7e1 | 2015-07-02 21:36:25 +0000 | [diff] [blame] | 46 | : TargetLowering(TM), Subtarget(&STI) { |
JF Bastien | af111db | 2015-08-24 22:16:48 +0000 | [diff] [blame] | 47 | auto MVTPtr = Subtarget->hasAddr64() ? MVT::i64 : MVT::i32; |
| 48 | |
JF Bastien | 71d29ac | 2015-08-12 17:53:29 +0000 | [diff] [blame] | 49 | // Booleans always contain 0 or 1. |
| 50 | setBooleanContents(ZeroOrOneBooleanContent); |
Thomas Lively | 5ea17d4 | 2018-10-20 01:35:23 +0000 | [diff] [blame] | 51 | // Except in SIMD vectors |
| 52 | setBooleanVectorContents(ZeroOrNegativeOneBooleanContent); |
Dan Gohman | bfaf7e1 | 2015-07-02 21:36:25 +0000 | [diff] [blame] | 53 | // WebAssembly does not produce floating-point exceptions on normal floating |
| 54 | // point operations. |
| 55 | setHasFloatingPointExceptions(false); |
Dan Gohman | 489abd7 | 2015-07-07 22:38:06 +0000 | [diff] [blame] | 56 | // We don't know the microarchitecture here, so just reduce register pressure. |
| 57 | setSchedulingPreference(Sched::RegPressure); |
JF Bastien | b9073fb | 2015-07-22 21:28:15 +0000 | [diff] [blame] | 58 | // Tell ISel that we have a stack pointer. |
| 59 | setStackPointerRegisterToSaveRestore( |
| 60 | Subtarget->hasAddr64() ? WebAssembly::SP64 : WebAssembly::SP32); |
| 61 | // Set up the register classes. |
Dan Gohman | d0bf981 | 2015-09-26 01:09:44 +0000 | [diff] [blame] | 62 | addRegisterClass(MVT::i32, &WebAssembly::I32RegClass); |
| 63 | addRegisterClass(MVT::i64, &WebAssembly::I64RegClass); |
| 64 | addRegisterClass(MVT::f32, &WebAssembly::F32RegClass); |
| 65 | addRegisterClass(MVT::f64, &WebAssembly::F64RegClass); |
Derek Schuff | 39bf39f | 2016-08-02 23:16:09 +0000 | [diff] [blame] | 66 | if (Subtarget->hasSIMD128()) { |
| 67 | addRegisterClass(MVT::v16i8, &WebAssembly::V128RegClass); |
| 68 | addRegisterClass(MVT::v8i16, &WebAssembly::V128RegClass); |
| 69 | addRegisterClass(MVT::v4i32, &WebAssembly::V128RegClass); |
| 70 | addRegisterClass(MVT::v4f32, &WebAssembly::V128RegClass); |
Heejin Ahn | 5831e9c | 2018-08-09 23:58:51 +0000 | [diff] [blame] | 71 | if (EnableUnimplementedWasmSIMDInstrs) { |
| 72 | addRegisterClass(MVT::v2i64, &WebAssembly::V128RegClass); |
| 73 | addRegisterClass(MVT::v2f64, &WebAssembly::V128RegClass); |
| 74 | } |
Derek Schuff | 39bf39f | 2016-08-02 23:16:09 +0000 | [diff] [blame] | 75 | } |
JF Bastien | b9073fb | 2015-07-22 21:28:15 +0000 | [diff] [blame] | 76 | // Compute derived properties from the register classes. |
| 77 | computeRegisterProperties(Subtarget->getRegisterInfo()); |
| 78 | |
JF Bastien | af111db | 2015-08-24 22:16:48 +0000 | [diff] [blame] | 79 | setOperationAction(ISD::GlobalAddress, MVTPtr, Custom); |
Dan Gohman | 2c8fe6a | 2015-11-25 16:44:29 +0000 | [diff] [blame] | 80 | setOperationAction(ISD::ExternalSymbol, MVTPtr, Custom); |
Dan Gohman | 950a13c | 2015-09-16 16:51:30 +0000 | [diff] [blame] | 81 | setOperationAction(ISD::JumpTable, MVTPtr, Custom); |
Derek Schuff | 51699a8 | 2016-02-12 22:56:03 +0000 | [diff] [blame] | 82 | setOperationAction(ISD::BlockAddress, MVTPtr, Custom); |
| 83 | setOperationAction(ISD::BRIND, MVT::Other, Custom); |
JF Bastien | af111db | 2015-08-24 22:16:48 +0000 | [diff] [blame] | 84 | |
Dan Gohman | 35bfb24 | 2015-12-04 23:22:35 +0000 | [diff] [blame] | 85 | // Take the default expansion for va_arg, va_copy, and va_end. There is no |
| 86 | // default action for va_start, so we do that custom. |
| 87 | setOperationAction(ISD::VASTART, MVT::Other, Custom); |
| 88 | setOperationAction(ISD::VAARG, MVT::Other, Expand); |
| 89 | setOperationAction(ISD::VACOPY, MVT::Other, Expand); |
| 90 | setOperationAction(ISD::VAEND, MVT::Other, Expand); |
| 91 | |
Thomas Lively | ebd4c90 | 2018-09-12 17:56:00 +0000 | [diff] [blame] | 92 | for (auto T : {MVT::f32, MVT::f64, MVT::v4f32, MVT::v2f64}) { |
JF Bastien | da06bce | 2015-08-11 21:02:46 +0000 | [diff] [blame] | 93 | // Don't expand the floating-point types to constant pools. |
| 94 | setOperationAction(ISD::ConstantFP, T, Legal); |
| 95 | // Expand floating-point comparisons. |
| 96 | for (auto CC : {ISD::SETO, ISD::SETUO, ISD::SETUEQ, ISD::SETONE, |
| 97 | ISD::SETULT, ISD::SETULE, ISD::SETUGT, ISD::SETUGE}) |
| 98 | setCondCodeAction(CC, T, Expand); |
Dan Gohman | 32907a6 | 2015-08-20 22:57:13 +0000 | [diff] [blame] | 99 | // Expand floating-point library function operators. |
Heejin Ahn | f208f63 | 2018-09-05 01:27:38 +0000 | [diff] [blame] | 100 | for (auto Op : |
| 101 | {ISD::FSIN, ISD::FCOS, ISD::FSINCOS, ISD::FPOW, ISD::FREM, ISD::FMA}) |
Dan Gohman | 32907a6 | 2015-08-20 22:57:13 +0000 | [diff] [blame] | 102 | setOperationAction(Op, T, Expand); |
Dan Gohman | 896e53f | 2015-08-24 18:23:13 +0000 | [diff] [blame] | 103 | // Note supported floating-point library function operators that otherwise |
| 104 | // default to expand. |
Dan Gohman | 7a6b982 | 2015-11-29 22:32:02 +0000 | [diff] [blame] | 105 | for (auto Op : |
| 106 | {ISD::FCEIL, ISD::FFLOOR, ISD::FTRUNC, ISD::FNEARBYINT, ISD::FRINT}) |
Dan Gohman | 896e53f | 2015-08-24 18:23:13 +0000 | [diff] [blame] | 107 | setOperationAction(Op, T, Legal); |
Thomas Lively | 30f1d69 | 2018-10-24 22:49:55 +0000 | [diff] [blame^] | 108 | // Support minimum and maximum, which otherwise default to expand. |
| 109 | setOperationAction(ISD::FMINIMUM, T, Legal); |
| 110 | setOperationAction(ISD::FMAXIMUM, T, Legal); |
Dan Gohman | a63e8eb | 2017-02-22 16:28:00 +0000 | [diff] [blame] | 111 | // WebAssembly currently has no builtin f16 support. |
| 112 | setOperationAction(ISD::FP16_TO_FP, T, Expand); |
| 113 | setOperationAction(ISD::FP_TO_FP16, T, Expand); |
| 114 | setLoadExtAction(ISD::EXTLOAD, T, MVT::f16, Expand); |
| 115 | setTruncStoreAction(T, MVT::f16, Expand); |
JF Bastien | da06bce | 2015-08-11 21:02:46 +0000 | [diff] [blame] | 116 | } |
Dan Gohman | 32907a6 | 2015-08-20 22:57:13 +0000 | [diff] [blame] | 117 | |
| 118 | for (auto T : {MVT::i32, MVT::i64}) { |
| 119 | // Expand unavailable integer operations. |
Dan Gohman | 7a6b982 | 2015-11-29 22:32:02 +0000 | [diff] [blame] | 120 | for (auto Op : |
Heejin Ahn | f208f63 | 2018-09-05 01:27:38 +0000 | [diff] [blame] | 121 | {ISD::BSWAP, ISD::SMUL_LOHI, ISD::UMUL_LOHI, ISD::MULHS, ISD::MULHU, |
| 122 | ISD::SDIVREM, ISD::UDIVREM, ISD::SHL_PARTS, ISD::SRA_PARTS, |
| 123 | ISD::SRL_PARTS, ISD::ADDC, ISD::ADDE, ISD::SUBC, ISD::SUBE}) { |
Dan Gohman | 32907a6 | 2015-08-20 22:57:13 +0000 | [diff] [blame] | 124 | setOperationAction(Op, T, Expand); |
| 125 | } |
| 126 | } |
| 127 | |
Thomas Lively | 2ee686d | 2018-08-22 23:06:27 +0000 | [diff] [blame] | 128 | // There is no i64x2.mul instruction |
| 129 | setOperationAction(ISD::MUL, MVT::v2i64, Expand); |
| 130 | |
Thomas Lively | a0d2581 | 2018-09-07 21:54:46 +0000 | [diff] [blame] | 131 | // We have custom shuffle lowering to expose the shuffle mask |
| 132 | if (Subtarget->hasSIMD128()) { |
| 133 | for (auto T : {MVT::v16i8, MVT::v8i16, MVT::v4i32, MVT::v4f32}) { |
| 134 | setOperationAction(ISD::VECTOR_SHUFFLE, T, Custom); |
| 135 | } |
| 136 | if (EnableUnimplementedWasmSIMDInstrs) { |
| 137 | setOperationAction(ISD::VECTOR_SHUFFLE, MVT::v2i64, Custom); |
| 138 | setOperationAction(ISD::VECTOR_SHUFFLE, MVT::v2f64, Custom); |
| 139 | } |
| 140 | } |
| 141 | |
Thomas Lively | 55735d5 | 2018-10-20 01:31:18 +0000 | [diff] [blame] | 142 | // Custom lowering to avoid having to emit a wrap for 2xi64 constant shifts |
| 143 | if (Subtarget->hasSIMD128() && EnableUnimplementedWasmSIMDInstrs) |
| 144 | for (auto Op : {ISD::SHL, ISD::SRA, ISD::SRL}) |
| 145 | setOperationAction(Op, MVT::v2i64, Custom); |
| 146 | |
Dan Gohman | 32907a6 | 2015-08-20 22:57:13 +0000 | [diff] [blame] | 147 | // As a special case, these operators use the type to mean the type to |
| 148 | // sign-extend from. |
Derek Schuff | a519fe5 | 2017-09-13 00:29:06 +0000 | [diff] [blame] | 149 | setOperationAction(ISD::SIGN_EXTEND_INREG, MVT::i1, Expand); |
Dan Gohman | 5d2b935 | 2018-01-19 17:16:24 +0000 | [diff] [blame] | 150 | if (!Subtarget->hasSignExt()) { |
Derek Schuff | a519fe5 | 2017-09-13 00:29:06 +0000 | [diff] [blame] | 151 | for (auto T : {MVT::i8, MVT::i16, MVT::i32}) |
| 152 | setOperationAction(ISD::SIGN_EXTEND_INREG, T, Expand); |
| 153 | } |
Thomas Lively | 5ea17d4 | 2018-10-20 01:35:23 +0000 | [diff] [blame] | 154 | for (auto T : MVT::integer_vector_valuetypes()) |
| 155 | setOperationAction(ISD::SIGN_EXTEND_INREG, T, Expand); |
Dan Gohman | 32907a6 | 2015-08-20 22:57:13 +0000 | [diff] [blame] | 156 | |
| 157 | // Dynamic stack allocation: use the default expansion. |
| 158 | setOperationAction(ISD::STACKSAVE, MVT::Other, Expand); |
| 159 | setOperationAction(ISD::STACKRESTORE, MVT::Other, Expand); |
Dan Gohman | 2683a55 | 2015-08-24 22:31:52 +0000 | [diff] [blame] | 160 | setOperationAction(ISD::DYNAMIC_STACKALLOC, MVTPtr, Expand); |
JF Bastien | 73ff6af | 2015-08-31 22:24:11 +0000 | [diff] [blame] | 161 | |
Derek Schuff | 9769deb | 2015-12-11 23:49:46 +0000 | [diff] [blame] | 162 | setOperationAction(ISD::FrameIndex, MVT::i32, Custom); |
Derek Schuff | aadc89c | 2016-02-16 18:18:36 +0000 | [diff] [blame] | 163 | setOperationAction(ISD::CopyToReg, MVT::Other, Custom); |
Derek Schuff | 9769deb | 2015-12-11 23:49:46 +0000 | [diff] [blame] | 164 | |
Dan Gohman | 950a13c | 2015-09-16 16:51:30 +0000 | [diff] [blame] | 165 | // Expand these forms; we pattern-match the forms that we can handle in isel. |
| 166 | for (auto T : {MVT::i32, MVT::i64, MVT::f32, MVT::f64}) |
| 167 | for (auto Op : {ISD::BR_CC, ISD::SELECT_CC}) |
| 168 | setOperationAction(Op, T, Expand); |
| 169 | |
| 170 | // We have custom switch handling. |
| 171 | setOperationAction(ISD::BR_JT, MVT::Other, Custom); |
| 172 | |
JF Bastien | 73ff6af | 2015-08-31 22:24:11 +0000 | [diff] [blame] | 173 | // WebAssembly doesn't have: |
| 174 | // - Floating-point extending loads. |
| 175 | // - Floating-point truncating stores. |
| 176 | // - i1 extending loads. |
Dan Gohman | 60bddf1 | 2015-12-10 02:07:53 +0000 | [diff] [blame] | 177 | setLoadExtAction(ISD::EXTLOAD, MVT::f64, MVT::f32, Expand); |
JF Bastien | 73ff6af | 2015-08-31 22:24:11 +0000 | [diff] [blame] | 178 | setTruncStoreAction(MVT::f64, MVT::f32, Expand); |
| 179 | for (auto T : MVT::integer_valuetypes()) |
| 180 | for (auto Ext : {ISD::EXTLOAD, ISD::ZEXTLOAD, ISD::SEXTLOAD}) |
| 181 | setLoadExtAction(Ext, T, MVT::i1, Promote); |
Derek Schuff | ffa143c | 2015-11-10 00:30:57 +0000 | [diff] [blame] | 182 | |
| 183 | // Trap lowers to wasm unreachable |
| 184 | setOperationAction(ISD::TRAP, MVT::Other, Legal); |
Derek Schuff | 18ba192 | 2017-08-30 18:07:45 +0000 | [diff] [blame] | 185 | |
Heejin Ahn | 5ef4d5f | 2018-05-31 22:25:54 +0000 | [diff] [blame] | 186 | // Exception handling intrinsics |
| 187 | setOperationAction(ISD::INTRINSIC_WO_CHAIN, MVT::Other, Custom); |
| 188 | |
Derek Schuff | 18ba192 | 2017-08-30 18:07:45 +0000 | [diff] [blame] | 189 | setMaxAtomicSizeInBitsSupported(64); |
Dan Gohman | bfaf7e1 | 2015-07-02 21:36:25 +0000 | [diff] [blame] | 190 | } |
Dan Gohman | 10e730a | 2015-06-29 23:51:55 +0000 | [diff] [blame] | 191 | |
Heejin Ahn | e8653bb | 2018-08-07 00:22:22 +0000 | [diff] [blame] | 192 | TargetLowering::AtomicExpansionKind |
| 193 | WebAssemblyTargetLowering::shouldExpandAtomicRMWInIR(AtomicRMWInst *AI) const { |
| 194 | // We have wasm instructions for these |
| 195 | switch (AI->getOperation()) { |
| 196 | case AtomicRMWInst::Add: |
| 197 | case AtomicRMWInst::Sub: |
| 198 | case AtomicRMWInst::And: |
| 199 | case AtomicRMWInst::Or: |
| 200 | case AtomicRMWInst::Xor: |
| 201 | case AtomicRMWInst::Xchg: |
| 202 | return AtomicExpansionKind::None; |
| 203 | default: |
| 204 | break; |
| 205 | } |
| 206 | return AtomicExpansionKind::CmpXChg; |
| 207 | } |
| 208 | |
Dan Gohman | 7b63484 | 2015-08-24 18:44:37 +0000 | [diff] [blame] | 209 | FastISel *WebAssemblyTargetLowering::createFastISel( |
| 210 | FunctionLoweringInfo &FuncInfo, const TargetLibraryInfo *LibInfo) const { |
| 211 | return WebAssembly::createFastISel(FuncInfo, LibInfo); |
| 212 | } |
| 213 | |
JF Bastien | af111db | 2015-08-24 22:16:48 +0000 | [diff] [blame] | 214 | bool WebAssemblyTargetLowering::isOffsetFoldingLegal( |
Dan Gohman | 7a6b982 | 2015-11-29 22:32:02 +0000 | [diff] [blame] | 215 | const GlobalAddressSDNode * /*GA*/) const { |
Dan Gohman | a4b710a | 2015-12-06 19:33:32 +0000 | [diff] [blame] | 216 | // All offsets can be folded. |
| 217 | return true; |
JF Bastien | af111db | 2015-08-24 22:16:48 +0000 | [diff] [blame] | 218 | } |
| 219 | |
Dan Gohman | 7a6b982 | 2015-11-29 22:32:02 +0000 | [diff] [blame] | 220 | MVT WebAssemblyTargetLowering::getScalarShiftAmountTy(const DataLayout & /*DL*/, |
JF Bastien | fda5337 | 2015-08-03 00:00:11 +0000 | [diff] [blame] | 221 | EVT VT) const { |
Dan Gohman | a848375 | 2015-12-10 00:26:26 +0000 | [diff] [blame] | 222 | unsigned BitWidth = NextPowerOf2(VT.getSizeInBits() - 1); |
Heejin Ahn | f208f63 | 2018-09-05 01:27:38 +0000 | [diff] [blame] | 223 | if (BitWidth > 1 && BitWidth < 8) |
| 224 | BitWidth = 8; |
Dan Gohman | 4172953 | 2015-12-16 23:25:51 +0000 | [diff] [blame] | 225 | |
| 226 | if (BitWidth > 64) { |
Dan Gohman | a01e8bd | 2016-05-14 02:15:47 +0000 | [diff] [blame] | 227 | // The shift will be lowered to a libcall, and compiler-rt libcalls expect |
| 228 | // the count to be an i32. |
| 229 | BitWidth = 32; |
Dan Gohman | 4172953 | 2015-12-16 23:25:51 +0000 | [diff] [blame] | 230 | assert(BitWidth >= Log2_32_Ceil(VT.getSizeInBits()) && |
Dan Gohman | a01e8bd | 2016-05-14 02:15:47 +0000 | [diff] [blame] | 231 | "32-bit shift counts ought to be enough for anyone"); |
Dan Gohman | 4172953 | 2015-12-16 23:25:51 +0000 | [diff] [blame] | 232 | } |
| 233 | |
Dan Gohman | a848375 | 2015-12-10 00:26:26 +0000 | [diff] [blame] | 234 | MVT Result = MVT::getIntegerVT(BitWidth); |
| 235 | assert(Result != MVT::INVALID_SIMPLE_VALUE_TYPE && |
| 236 | "Unable to represent scalar shift amount type"); |
| 237 | return Result; |
JF Bastien | fda5337 | 2015-08-03 00:00:11 +0000 | [diff] [blame] | 238 | } |
| 239 | |
Dan Gohman | cdd48b8 | 2017-11-28 01:13:40 +0000 | [diff] [blame] | 240 | // Lower an fp-to-int conversion operator from the LLVM opcode, which has an |
| 241 | // undefined result on invalid/overflow, to the WebAssembly opcode, which |
| 242 | // traps on invalid/overflow. |
Heejin Ahn | f208f63 | 2018-09-05 01:27:38 +0000 | [diff] [blame] | 243 | static MachineBasicBlock *LowerFPToInt(MachineInstr &MI, DebugLoc DL, |
| 244 | MachineBasicBlock *BB, |
| 245 | const TargetInstrInfo &TII, |
| 246 | bool IsUnsigned, bool Int64, |
| 247 | bool Float64, unsigned LoweredOpcode) { |
Dan Gohman | cdd48b8 | 2017-11-28 01:13:40 +0000 | [diff] [blame] | 248 | MachineRegisterInfo &MRI = BB->getParent()->getRegInfo(); |
| 249 | |
| 250 | unsigned OutReg = MI.getOperand(0).getReg(); |
| 251 | unsigned InReg = MI.getOperand(1).getReg(); |
| 252 | |
| 253 | unsigned Abs = Float64 ? WebAssembly::ABS_F64 : WebAssembly::ABS_F32; |
| 254 | unsigned FConst = Float64 ? WebAssembly::CONST_F64 : WebAssembly::CONST_F32; |
| 255 | unsigned LT = Float64 ? WebAssembly::LT_F64 : WebAssembly::LT_F32; |
Dan Gohman | 580c102 | 2017-11-29 20:20:11 +0000 | [diff] [blame] | 256 | unsigned GE = Float64 ? WebAssembly::GE_F64 : WebAssembly::GE_F32; |
Dan Gohman | cdd48b8 | 2017-11-28 01:13:40 +0000 | [diff] [blame] | 257 | unsigned IConst = Int64 ? WebAssembly::CONST_I64 : WebAssembly::CONST_I32; |
Dan Gohman | 580c102 | 2017-11-29 20:20:11 +0000 | [diff] [blame] | 258 | unsigned Eqz = WebAssembly::EQZ_I32; |
| 259 | unsigned And = WebAssembly::AND_I32; |
Dan Gohman | cdd48b8 | 2017-11-28 01:13:40 +0000 | [diff] [blame] | 260 | int64_t Limit = Int64 ? INT64_MIN : INT32_MIN; |
| 261 | int64_t Substitute = IsUnsigned ? 0 : Limit; |
| 262 | double CmpVal = IsUnsigned ? -(double)Limit * 2.0 : -(double)Limit; |
David Blaikie | 2110924 | 2017-12-15 23:52:06 +0000 | [diff] [blame] | 263 | auto &Context = BB->getParent()->getFunction().getContext(); |
Dan Gohman | cdd48b8 | 2017-11-28 01:13:40 +0000 | [diff] [blame] | 264 | Type *Ty = Float64 ? Type::getDoubleTy(Context) : Type::getFloatTy(Context); |
| 265 | |
| 266 | const BasicBlock *LLVM_BB = BB->getBasicBlock(); |
| 267 | MachineFunction *F = BB->getParent(); |
| 268 | MachineBasicBlock *TrueMBB = F->CreateMachineBasicBlock(LLVM_BB); |
| 269 | MachineBasicBlock *FalseMBB = F->CreateMachineBasicBlock(LLVM_BB); |
| 270 | MachineBasicBlock *DoneMBB = F->CreateMachineBasicBlock(LLVM_BB); |
| 271 | |
| 272 | MachineFunction::iterator It = ++BB->getIterator(); |
| 273 | F->insert(It, FalseMBB); |
| 274 | F->insert(It, TrueMBB); |
| 275 | F->insert(It, DoneMBB); |
| 276 | |
| 277 | // Transfer the remainder of BB and its successor edges to DoneMBB. |
| 278 | DoneMBB->splice(DoneMBB->begin(), BB, |
Heejin Ahn | f208f63 | 2018-09-05 01:27:38 +0000 | [diff] [blame] | 279 | std::next(MachineBasicBlock::iterator(MI)), BB->end()); |
Dan Gohman | cdd48b8 | 2017-11-28 01:13:40 +0000 | [diff] [blame] | 280 | DoneMBB->transferSuccessorsAndUpdatePHIs(BB); |
| 281 | |
| 282 | BB->addSuccessor(TrueMBB); |
| 283 | BB->addSuccessor(FalseMBB); |
| 284 | TrueMBB->addSuccessor(DoneMBB); |
| 285 | FalseMBB->addSuccessor(DoneMBB); |
| 286 | |
Dan Gohman | 580c102 | 2017-11-29 20:20:11 +0000 | [diff] [blame] | 287 | unsigned Tmp0, Tmp1, CmpReg, EqzReg, FalseReg, TrueReg; |
Dan Gohman | cdd48b8 | 2017-11-28 01:13:40 +0000 | [diff] [blame] | 288 | Tmp0 = MRI.createVirtualRegister(MRI.getRegClass(InReg)); |
| 289 | Tmp1 = MRI.createVirtualRegister(MRI.getRegClass(InReg)); |
Dan Gohman | 580c102 | 2017-11-29 20:20:11 +0000 | [diff] [blame] | 290 | CmpReg = MRI.createVirtualRegister(&WebAssembly::I32RegClass); |
| 291 | EqzReg = MRI.createVirtualRegister(&WebAssembly::I32RegClass); |
| 292 | FalseReg = MRI.createVirtualRegister(MRI.getRegClass(OutReg)); |
| 293 | TrueReg = MRI.createVirtualRegister(MRI.getRegClass(OutReg)); |
Dan Gohman | cdd48b8 | 2017-11-28 01:13:40 +0000 | [diff] [blame] | 294 | |
| 295 | MI.eraseFromParent(); |
Dan Gohman | 580c102 | 2017-11-29 20:20:11 +0000 | [diff] [blame] | 296 | // For signed numbers, we can do a single comparison to determine whether |
| 297 | // fabs(x) is within range. |
Dan Gohman | cdd48b8 | 2017-11-28 01:13:40 +0000 | [diff] [blame] | 298 | if (IsUnsigned) { |
| 299 | Tmp0 = InReg; |
| 300 | } else { |
Heejin Ahn | f208f63 | 2018-09-05 01:27:38 +0000 | [diff] [blame] | 301 | BuildMI(BB, DL, TII.get(Abs), Tmp0).addReg(InReg); |
Dan Gohman | cdd48b8 | 2017-11-28 01:13:40 +0000 | [diff] [blame] | 302 | } |
| 303 | BuildMI(BB, DL, TII.get(FConst), Tmp1) |
| 304 | .addFPImm(cast<ConstantFP>(ConstantFP::get(Ty, CmpVal))); |
Heejin Ahn | f208f63 | 2018-09-05 01:27:38 +0000 | [diff] [blame] | 305 | BuildMI(BB, DL, TII.get(LT), CmpReg).addReg(Tmp0).addReg(Tmp1); |
Dan Gohman | 580c102 | 2017-11-29 20:20:11 +0000 | [diff] [blame] | 306 | |
| 307 | // For unsigned numbers, we have to do a separate comparison with zero. |
| 308 | if (IsUnsigned) { |
| 309 | Tmp1 = MRI.createVirtualRegister(MRI.getRegClass(InReg)); |
Heejin Ahn | f208f63 | 2018-09-05 01:27:38 +0000 | [diff] [blame] | 310 | unsigned SecondCmpReg = |
| 311 | MRI.createVirtualRegister(&WebAssembly::I32RegClass); |
Dan Gohman | 580c102 | 2017-11-29 20:20:11 +0000 | [diff] [blame] | 312 | unsigned AndReg = MRI.createVirtualRegister(&WebAssembly::I32RegClass); |
| 313 | BuildMI(BB, DL, TII.get(FConst), Tmp1) |
| 314 | .addFPImm(cast<ConstantFP>(ConstantFP::get(Ty, 0.0))); |
Heejin Ahn | f208f63 | 2018-09-05 01:27:38 +0000 | [diff] [blame] | 315 | BuildMI(BB, DL, TII.get(GE), SecondCmpReg).addReg(Tmp0).addReg(Tmp1); |
| 316 | BuildMI(BB, DL, TII.get(And), AndReg).addReg(CmpReg).addReg(SecondCmpReg); |
Dan Gohman | 580c102 | 2017-11-29 20:20:11 +0000 | [diff] [blame] | 317 | CmpReg = AndReg; |
| 318 | } |
| 319 | |
Heejin Ahn | f208f63 | 2018-09-05 01:27:38 +0000 | [diff] [blame] | 320 | BuildMI(BB, DL, TII.get(Eqz), EqzReg).addReg(CmpReg); |
Dan Gohman | 580c102 | 2017-11-29 20:20:11 +0000 | [diff] [blame] | 321 | |
| 322 | // Create the CFG diamond to select between doing the conversion or using |
| 323 | // the substitute value. |
Heejin Ahn | f208f63 | 2018-09-05 01:27:38 +0000 | [diff] [blame] | 324 | BuildMI(BB, DL, TII.get(WebAssembly::BR_IF)).addMBB(TrueMBB).addReg(EqzReg); |
| 325 | BuildMI(FalseMBB, DL, TII.get(LoweredOpcode), FalseReg).addReg(InReg); |
| 326 | BuildMI(FalseMBB, DL, TII.get(WebAssembly::BR)).addMBB(DoneMBB); |
| 327 | BuildMI(TrueMBB, DL, TII.get(IConst), TrueReg).addImm(Substitute); |
Dan Gohman | cdd48b8 | 2017-11-28 01:13:40 +0000 | [diff] [blame] | 328 | BuildMI(*DoneMBB, DoneMBB->begin(), DL, TII.get(TargetOpcode::PHI), OutReg) |
Dan Gohman | 580c102 | 2017-11-29 20:20:11 +0000 | [diff] [blame] | 329 | .addReg(FalseReg) |
Dan Gohman | cdd48b8 | 2017-11-28 01:13:40 +0000 | [diff] [blame] | 330 | .addMBB(FalseMBB) |
Dan Gohman | 580c102 | 2017-11-29 20:20:11 +0000 | [diff] [blame] | 331 | .addReg(TrueReg) |
Dan Gohman | cdd48b8 | 2017-11-28 01:13:40 +0000 | [diff] [blame] | 332 | .addMBB(TrueMBB); |
| 333 | |
| 334 | return DoneMBB; |
| 335 | } |
| 336 | |
Heejin Ahn | f208f63 | 2018-09-05 01:27:38 +0000 | [diff] [blame] | 337 | MachineBasicBlock *WebAssemblyTargetLowering::EmitInstrWithCustomInserter( |
| 338 | MachineInstr &MI, MachineBasicBlock *BB) const { |
Dan Gohman | cdd48b8 | 2017-11-28 01:13:40 +0000 | [diff] [blame] | 339 | const TargetInstrInfo &TII = *Subtarget->getInstrInfo(); |
| 340 | DebugLoc DL = MI.getDebugLoc(); |
| 341 | |
| 342 | switch (MI.getOpcode()) { |
Heejin Ahn | f208f63 | 2018-09-05 01:27:38 +0000 | [diff] [blame] | 343 | default: |
| 344 | llvm_unreachable("Unexpected instr type to insert"); |
Dan Gohman | cdd48b8 | 2017-11-28 01:13:40 +0000 | [diff] [blame] | 345 | case WebAssembly::FP_TO_SINT_I32_F32: |
| 346 | return LowerFPToInt(MI, DL, BB, TII, false, false, false, |
| 347 | WebAssembly::I32_TRUNC_S_F32); |
| 348 | case WebAssembly::FP_TO_UINT_I32_F32: |
| 349 | return LowerFPToInt(MI, DL, BB, TII, true, false, false, |
| 350 | WebAssembly::I32_TRUNC_U_F32); |
| 351 | case WebAssembly::FP_TO_SINT_I64_F32: |
| 352 | return LowerFPToInt(MI, DL, BB, TII, false, true, false, |
| 353 | WebAssembly::I64_TRUNC_S_F32); |
| 354 | case WebAssembly::FP_TO_UINT_I64_F32: |
| 355 | return LowerFPToInt(MI, DL, BB, TII, true, true, false, |
| 356 | WebAssembly::I64_TRUNC_U_F32); |
| 357 | case WebAssembly::FP_TO_SINT_I32_F64: |
| 358 | return LowerFPToInt(MI, DL, BB, TII, false, false, true, |
| 359 | WebAssembly::I32_TRUNC_S_F64); |
| 360 | case WebAssembly::FP_TO_UINT_I32_F64: |
| 361 | return LowerFPToInt(MI, DL, BB, TII, true, false, true, |
| 362 | WebAssembly::I32_TRUNC_U_F64); |
| 363 | case WebAssembly::FP_TO_SINT_I64_F64: |
| 364 | return LowerFPToInt(MI, DL, BB, TII, false, true, true, |
| 365 | WebAssembly::I64_TRUNC_S_F64); |
| 366 | case WebAssembly::FP_TO_UINT_I64_F64: |
| 367 | return LowerFPToInt(MI, DL, BB, TII, true, true, true, |
| 368 | WebAssembly::I64_TRUNC_U_F64); |
Heejin Ahn | f208f63 | 2018-09-05 01:27:38 +0000 | [diff] [blame] | 369 | llvm_unreachable("Unexpected instruction to emit with custom inserter"); |
Dan Gohman | cdd48b8 | 2017-11-28 01:13:40 +0000 | [diff] [blame] | 370 | } |
| 371 | } |
| 372 | |
Heejin Ahn | f208f63 | 2018-09-05 01:27:38 +0000 | [diff] [blame] | 373 | const char * |
| 374 | WebAssemblyTargetLowering::getTargetNodeName(unsigned Opcode) const { |
JF Bastien | 480c840 | 2015-08-11 20:13:18 +0000 | [diff] [blame] | 375 | switch (static_cast<WebAssemblyISD::NodeType>(Opcode)) { |
Heejin Ahn | f208f63 | 2018-09-05 01:27:38 +0000 | [diff] [blame] | 376 | case WebAssemblyISD::FIRST_NUMBER: |
| 377 | break; |
| 378 | #define HANDLE_NODETYPE(NODE) \ |
| 379 | case WebAssemblyISD::NODE: \ |
JF Bastien | af111db | 2015-08-24 22:16:48 +0000 | [diff] [blame] | 380 | return "WebAssemblyISD::" #NODE; |
| 381 | #include "WebAssemblyISD.def" |
| 382 | #undef HANDLE_NODETYPE |
JF Bastien | 480c840 | 2015-08-11 20:13:18 +0000 | [diff] [blame] | 383 | } |
| 384 | return nullptr; |
| 385 | } |
| 386 | |
Dan Gohman | f19ed56 | 2015-11-13 01:42:29 +0000 | [diff] [blame] | 387 | std::pair<unsigned, const TargetRegisterClass *> |
| 388 | WebAssemblyTargetLowering::getRegForInlineAsmConstraint( |
| 389 | const TargetRegisterInfo *TRI, StringRef Constraint, MVT VT) const { |
| 390 | // First, see if this is a constraint that directly corresponds to a |
| 391 | // WebAssembly register class. |
| 392 | if (Constraint.size() == 1) { |
| 393 | switch (Constraint[0]) { |
Heejin Ahn | f208f63 | 2018-09-05 01:27:38 +0000 | [diff] [blame] | 394 | case 'r': |
| 395 | assert(VT != MVT::iPTR && "Pointer MVT not expected here"); |
| 396 | if (Subtarget->hasSIMD128() && VT.isVector()) { |
| 397 | if (VT.getSizeInBits() == 128) |
| 398 | return std::make_pair(0U, &WebAssembly::V128RegClass); |
| 399 | } |
| 400 | if (VT.isInteger() && !VT.isVector()) { |
| 401 | if (VT.getSizeInBits() <= 32) |
| 402 | return std::make_pair(0U, &WebAssembly::I32RegClass); |
| 403 | if (VT.getSizeInBits() <= 64) |
| 404 | return std::make_pair(0U, &WebAssembly::I64RegClass); |
| 405 | } |
| 406 | break; |
| 407 | default: |
| 408 | break; |
Dan Gohman | f19ed56 | 2015-11-13 01:42:29 +0000 | [diff] [blame] | 409 | } |
| 410 | } |
| 411 | |
| 412 | return TargetLowering::getRegForInlineAsmConstraint(TRI, Constraint, VT); |
| 413 | } |
| 414 | |
Dan Gohman | 3192ddf | 2015-11-19 23:04:59 +0000 | [diff] [blame] | 415 | bool WebAssemblyTargetLowering::isCheapToSpeculateCttz() const { |
| 416 | // Assume ctz is a relatively cheap operation. |
| 417 | return true; |
| 418 | } |
| 419 | |
| 420 | bool WebAssemblyTargetLowering::isCheapToSpeculateCtlz() const { |
| 421 | // Assume clz is a relatively cheap operation. |
| 422 | return true; |
| 423 | } |
| 424 | |
Dan Gohman | 4b9d791 | 2015-12-15 22:01:29 +0000 | [diff] [blame] | 425 | bool WebAssemblyTargetLowering::isLegalAddressingMode(const DataLayout &DL, |
| 426 | const AddrMode &AM, |
Heejin Ahn | f208f63 | 2018-09-05 01:27:38 +0000 | [diff] [blame] | 427 | Type *Ty, unsigned AS, |
Jonas Paulsson | 024e319 | 2017-07-21 11:59:37 +0000 | [diff] [blame] | 428 | Instruction *I) const { |
Dan Gohman | 4b9d791 | 2015-12-15 22:01:29 +0000 | [diff] [blame] | 429 | // WebAssembly offsets are added as unsigned without wrapping. The |
| 430 | // isLegalAddressingMode gives us no way to determine if wrapping could be |
| 431 | // happening, so we approximate this by accepting only non-negative offsets. |
Heejin Ahn | f208f63 | 2018-09-05 01:27:38 +0000 | [diff] [blame] | 432 | if (AM.BaseOffs < 0) |
| 433 | return false; |
Dan Gohman | 4b9d791 | 2015-12-15 22:01:29 +0000 | [diff] [blame] | 434 | |
| 435 | // WebAssembly has no scale register operands. |
Heejin Ahn | f208f63 | 2018-09-05 01:27:38 +0000 | [diff] [blame] | 436 | if (AM.Scale != 0) |
| 437 | return false; |
Dan Gohman | 4b9d791 | 2015-12-15 22:01:29 +0000 | [diff] [blame] | 438 | |
| 439 | // Everything else is legal. |
| 440 | return true; |
| 441 | } |
| 442 | |
Dan Gohman | bb37224 | 2016-01-26 03:39:31 +0000 | [diff] [blame] | 443 | bool WebAssemblyTargetLowering::allowsMisalignedMemoryAccesses( |
Derek Schuff | 3f06329 | 2016-02-11 20:57:09 +0000 | [diff] [blame] | 444 | EVT /*VT*/, unsigned /*AddrSpace*/, unsigned /*Align*/, bool *Fast) const { |
Dan Gohman | bb37224 | 2016-01-26 03:39:31 +0000 | [diff] [blame] | 445 | // WebAssembly supports unaligned accesses, though it should be declared |
| 446 | // with the p2align attribute on loads and stores which do so, and there |
| 447 | // may be a performance impact. We tell LLVM they're "fast" because |
Dan Gohman | fb619e9 | 2016-01-26 14:55:17 +0000 | [diff] [blame] | 448 | // for the kinds of things that LLVM uses this for (merging adjacent stores |
Dan Gohman | bb37224 | 2016-01-26 03:39:31 +0000 | [diff] [blame] | 449 | // of constants, etc.), WebAssembly implementations will either want the |
| 450 | // unaligned access or they'll split anyway. |
Heejin Ahn | f208f63 | 2018-09-05 01:27:38 +0000 | [diff] [blame] | 451 | if (Fast) |
| 452 | *Fast = true; |
Dan Gohman | bb37224 | 2016-01-26 03:39:31 +0000 | [diff] [blame] | 453 | return true; |
| 454 | } |
| 455 | |
Reid Kleckner | b518054 | 2017-03-21 16:57:19 +0000 | [diff] [blame] | 456 | bool WebAssemblyTargetLowering::isIntDivCheap(EVT VT, |
| 457 | AttributeList Attr) const { |
Dan Gohman | b4c3c38 | 2016-05-18 14:29:42 +0000 | [diff] [blame] | 458 | // The current thinking is that wasm engines will perform this optimization, |
| 459 | // so we can save on code size. |
| 460 | return true; |
| 461 | } |
| 462 | |
Simon Pilgrim | 99f7016 | 2018-06-28 17:27:09 +0000 | [diff] [blame] | 463 | EVT WebAssemblyTargetLowering::getSetCCResultType(const DataLayout &DL, |
| 464 | LLVMContext &C, |
| 465 | EVT VT) const { |
| 466 | if (VT.isVector()) |
| 467 | return VT.changeVectorElementTypeToInteger(); |
| 468 | |
| 469 | return TargetLowering::getSetCCResultType(DL, C, VT); |
| 470 | } |
| 471 | |
Heejin Ahn | 4128cb0 | 2018-08-02 21:44:24 +0000 | [diff] [blame] | 472 | bool WebAssemblyTargetLowering::getTgtMemIntrinsic(IntrinsicInfo &Info, |
| 473 | const CallInst &I, |
| 474 | MachineFunction &MF, |
| 475 | unsigned Intrinsic) const { |
| 476 | switch (Intrinsic) { |
| 477 | case Intrinsic::wasm_atomic_notify: |
| 478 | Info.opc = ISD::INTRINSIC_W_CHAIN; |
| 479 | Info.memVT = MVT::i32; |
| 480 | Info.ptrVal = I.getArgOperand(0); |
| 481 | Info.offset = 0; |
| 482 | Info.align = 4; |
| 483 | // atomic.notify instruction does not really load the memory specified with |
| 484 | // this argument, but MachineMemOperand should either be load or store, so |
| 485 | // we set this to a load. |
| 486 | // FIXME Volatile isn't really correct, but currently all LLVM atomic |
| 487 | // instructions are treated as volatiles in the backend, so we should be |
| 488 | // consistent. The same applies for wasm_atomic_wait intrinsics too. |
| 489 | Info.flags = MachineMemOperand::MOVolatile | MachineMemOperand::MOLoad; |
| 490 | return true; |
| 491 | case Intrinsic::wasm_atomic_wait_i32: |
| 492 | Info.opc = ISD::INTRINSIC_W_CHAIN; |
| 493 | Info.memVT = MVT::i32; |
| 494 | Info.ptrVal = I.getArgOperand(0); |
| 495 | Info.offset = 0; |
| 496 | Info.align = 4; |
| 497 | Info.flags = MachineMemOperand::MOVolatile | MachineMemOperand::MOLoad; |
| 498 | return true; |
| 499 | case Intrinsic::wasm_atomic_wait_i64: |
| 500 | Info.opc = ISD::INTRINSIC_W_CHAIN; |
| 501 | Info.memVT = MVT::i64; |
| 502 | Info.ptrVal = I.getArgOperand(0); |
| 503 | Info.offset = 0; |
| 504 | Info.align = 8; |
| 505 | Info.flags = MachineMemOperand::MOVolatile | MachineMemOperand::MOLoad; |
| 506 | return true; |
| 507 | default: |
| 508 | return false; |
| 509 | } |
| 510 | } |
| 511 | |
Dan Gohman | 10e730a | 2015-06-29 23:51:55 +0000 | [diff] [blame] | 512 | //===----------------------------------------------------------------------===// |
| 513 | // WebAssembly Lowering private implementation. |
| 514 | //===----------------------------------------------------------------------===// |
| 515 | |
| 516 | //===----------------------------------------------------------------------===// |
| 517 | // Lowering Code |
| 518 | //===----------------------------------------------------------------------===// |
| 519 | |
Benjamin Kramer | bdc4956 | 2016-06-12 15:39:02 +0000 | [diff] [blame] | 520 | static void fail(const SDLoc &DL, SelectionDAG &DAG, const char *msg) { |
JF Bastien | b9073fb | 2015-07-22 21:28:15 +0000 | [diff] [blame] | 521 | MachineFunction &MF = DAG.getMachineFunction(); |
| 522 | DAG.getContext()->diagnose( |
David Blaikie | 2110924 | 2017-12-15 23:52:06 +0000 | [diff] [blame] | 523 | DiagnosticInfoUnsupported(MF.getFunction(), msg, DL.getDebugLoc())); |
JF Bastien | b9073fb | 2015-07-22 21:28:15 +0000 | [diff] [blame] | 524 | } |
| 525 | |
Dan Gohman | 85dbdda | 2015-12-04 17:16:07 +0000 | [diff] [blame] | 526 | // Test whether the given calling convention is supported. |
Dan Gohman | a3f5ce5 | 2015-12-04 17:18:32 +0000 | [diff] [blame] | 527 | static bool CallingConvSupported(CallingConv::ID CallConv) { |
Dan Gohman | 85dbdda | 2015-12-04 17:16:07 +0000 | [diff] [blame] | 528 | // We currently support the language-independent target-independent |
Dan Gohman | 1ce2b1a | 2015-12-04 18:27:03 +0000 | [diff] [blame] | 529 | // conventions. We don't yet have a way to annotate calls with properties like |
| 530 | // "cold", and we don't have any call-clobbered registers, so these are mostly |
| 531 | // all handled the same. |
Dan Gohman | a3f5ce5 | 2015-12-04 17:18:32 +0000 | [diff] [blame] | 532 | return CallConv == CallingConv::C || CallConv == CallingConv::Fast || |
Dan Gohman | 1ce2b1a | 2015-12-04 18:27:03 +0000 | [diff] [blame] | 533 | CallConv == CallingConv::Cold || |
| 534 | CallConv == CallingConv::PreserveMost || |
| 535 | CallConv == CallingConv::PreserveAll || |
| 536 | CallConv == CallingConv::CXX_FAST_TLS; |
Dan Gohman | 85dbdda | 2015-12-04 17:16:07 +0000 | [diff] [blame] | 537 | } |
| 538 | |
Heejin Ahn | f208f63 | 2018-09-05 01:27:38 +0000 | [diff] [blame] | 539 | SDValue |
| 540 | WebAssemblyTargetLowering::LowerCall(CallLoweringInfo &CLI, |
| 541 | SmallVectorImpl<SDValue> &InVals) const { |
JF Bastien | d8a9d66 | 2015-08-24 21:59:51 +0000 | [diff] [blame] | 542 | SelectionDAG &DAG = CLI.DAG; |
| 543 | SDLoc DL = CLI.DL; |
| 544 | SDValue Chain = CLI.Chain; |
| 545 | SDValue Callee = CLI.Callee; |
| 546 | MachineFunction &MF = DAG.getMachineFunction(); |
Derek Schuff | 992d83f | 2016-02-10 20:14:15 +0000 | [diff] [blame] | 547 | auto Layout = MF.getDataLayout(); |
JF Bastien | d8a9d66 | 2015-08-24 21:59:51 +0000 | [diff] [blame] | 548 | |
| 549 | CallingConv::ID CallConv = CLI.CallConv; |
Dan Gohman | 85dbdda | 2015-12-04 17:16:07 +0000 | [diff] [blame] | 550 | if (!CallingConvSupported(CallConv)) |
Dan Gohman | 9cc692b | 2015-10-02 20:54:23 +0000 | [diff] [blame] | 551 | fail(DL, DAG, |
| 552 | "WebAssembly doesn't support language-specific or target-specific " |
| 553 | "calling conventions yet"); |
JF Bastien | d8a9d66 | 2015-08-24 21:59:51 +0000 | [diff] [blame] | 554 | if (CLI.IsPatchPoint) |
| 555 | fail(DL, DAG, "WebAssembly doesn't support patch point yet"); |
| 556 | |
Dan Gohman | 9cc692b | 2015-10-02 20:54:23 +0000 | [diff] [blame] | 557 | // WebAssembly doesn't currently support explicit tail calls. If they are |
| 558 | // required, fail. Otherwise, just disable them. |
| 559 | if ((CallConv == CallingConv::Fast && CLI.IsTailCall && |
| 560 | MF.getTarget().Options.GuaranteedTailCallOpt) || |
Peter Collingbourne | 081ffe2 | 2017-07-26 19:15:29 +0000 | [diff] [blame] | 561 | (CLI.CS && CLI.CS.isMustTailCall())) |
Dan Gohman | 9cc692b | 2015-10-02 20:54:23 +0000 | [diff] [blame] | 562 | fail(DL, DAG, "WebAssembly doesn't support tail call yet"); |
| 563 | CLI.IsTailCall = false; |
| 564 | |
JF Bastien | d8a9d66 | 2015-08-24 21:59:51 +0000 | [diff] [blame] | 565 | SmallVectorImpl<ISD::InputArg> &Ins = CLI.Ins; |
Dan Gohman | e590b33 | 2015-09-09 01:52:45 +0000 | [diff] [blame] | 566 | if (Ins.size() > 1) |
| 567 | fail(DL, DAG, "WebAssembly doesn't support more than 1 returned value yet"); |
| 568 | |
Dan Gohman | 2d822e7 | 2015-12-04 17:12:52 +0000 | [diff] [blame] | 569 | SmallVectorImpl<ISD::OutputArg> &Outs = CLI.Outs; |
Derek Schuff | 4dd6778 | 2016-01-27 21:17:39 +0000 | [diff] [blame] | 570 | SmallVectorImpl<SDValue> &OutVals = CLI.OutVals; |
Dan Gohman | 910ba33 | 2018-06-26 03:18:38 +0000 | [diff] [blame] | 571 | unsigned NumFixedArgs = 0; |
Derek Schuff | 4dd6778 | 2016-01-27 21:17:39 +0000 | [diff] [blame] | 572 | for (unsigned i = 0; i < Outs.size(); ++i) { |
| 573 | const ISD::OutputArg &Out = Outs[i]; |
| 574 | SDValue &OutVal = OutVals[i]; |
Dan Gohman | 7935fa3 | 2015-12-10 00:22:40 +0000 | [diff] [blame] | 575 | if (Out.Flags.isNest()) |
| 576 | fail(DL, DAG, "WebAssembly hasn't implemented nest arguments"); |
Dan Gohman | 2d822e7 | 2015-12-04 17:12:52 +0000 | [diff] [blame] | 577 | if (Out.Flags.isInAlloca()) |
Dan Gohman | 7935fa3 | 2015-12-10 00:22:40 +0000 | [diff] [blame] | 578 | fail(DL, DAG, "WebAssembly hasn't implemented inalloca arguments"); |
Dan Gohman | 2d822e7 | 2015-12-04 17:12:52 +0000 | [diff] [blame] | 579 | if (Out.Flags.isInConsecutiveRegs()) |
Dan Gohman | 7935fa3 | 2015-12-10 00:22:40 +0000 | [diff] [blame] | 580 | fail(DL, DAG, "WebAssembly hasn't implemented cons regs arguments"); |
Dan Gohman | 2d822e7 | 2015-12-04 17:12:52 +0000 | [diff] [blame] | 581 | if (Out.Flags.isInConsecutiveRegsLast()) |
Dan Gohman | 7935fa3 | 2015-12-10 00:22:40 +0000 | [diff] [blame] | 582 | fail(DL, DAG, "WebAssembly hasn't implemented cons regs last arguments"); |
Dan Gohman | a6771b3 | 2016-02-12 21:30:18 +0000 | [diff] [blame] | 583 | if (Out.Flags.isByVal() && Out.Flags.getByValSize() != 0) { |
Matthias Braun | 941a705 | 2016-07-28 18:40:00 +0000 | [diff] [blame] | 584 | auto &MFI = MF.getFrameInfo(); |
| 585 | int FI = MFI.CreateStackObject(Out.Flags.getByValSize(), |
| 586 | Out.Flags.getByValAlign(), |
| 587 | /*isSS=*/false); |
Derek Schuff | 4dd6778 | 2016-01-27 21:17:39 +0000 | [diff] [blame] | 588 | SDValue SizeNode = |
| 589 | DAG.getConstant(Out.Flags.getByValSize(), DL, MVT::i32); |
Derek Schuff | 992d83f | 2016-02-10 20:14:15 +0000 | [diff] [blame] | 590 | SDValue FINode = DAG.getFrameIndex(FI, getPointerTy(Layout)); |
Derek Schuff | 4dd6778 | 2016-01-27 21:17:39 +0000 | [diff] [blame] | 591 | Chain = DAG.getMemcpy( |
| 592 | Chain, DL, FINode, OutVal, SizeNode, Out.Flags.getByValAlign(), |
Dan Gohman | 476ffce | 2016-02-17 01:43:37 +0000 | [diff] [blame] | 593 | /*isVolatile*/ false, /*AlwaysInline=*/false, |
Derek Schuff | 4dd6778 | 2016-01-27 21:17:39 +0000 | [diff] [blame] | 594 | /*isTailCall*/ false, MachinePointerInfo(), MachinePointerInfo()); |
| 595 | OutVal = FINode; |
| 596 | } |
Dan Gohman | 910ba33 | 2018-06-26 03:18:38 +0000 | [diff] [blame] | 597 | // Count the number of fixed args *after* legalization. |
| 598 | NumFixedArgs += Out.IsFixed; |
Dan Gohman | 2d822e7 | 2015-12-04 17:12:52 +0000 | [diff] [blame] | 599 | } |
| 600 | |
JF Bastien | d8a9d66 | 2015-08-24 21:59:51 +0000 | [diff] [blame] | 601 | bool IsVarArg = CLI.IsVarArg; |
Derek Schuff | 992d83f | 2016-02-10 20:14:15 +0000 | [diff] [blame] | 602 | auto PtrVT = getPointerTy(Layout); |
Dan Gohman | e590b33 | 2015-09-09 01:52:45 +0000 | [diff] [blame] | 603 | |
JF Bastien | d8a9d66 | 2015-08-24 21:59:51 +0000 | [diff] [blame] | 604 | // Analyze operands of the call, assigning locations to each operand. |
| 605 | SmallVector<CCValAssign, 16> ArgLocs; |
| 606 | CCState CCInfo(CallConv, IsVarArg, MF, ArgLocs, *DAG.getContext()); |
JF Bastien | d8a9d66 | 2015-08-24 21:59:51 +0000 | [diff] [blame] | 607 | |
Dan Gohman | 35bfb24 | 2015-12-04 23:22:35 +0000 | [diff] [blame] | 608 | if (IsVarArg) { |
Derek Schuff | 27501e2 | 2016-02-10 19:51:04 +0000 | [diff] [blame] | 609 | // Outgoing non-fixed arguments are placed in a buffer. First |
| 610 | // compute their offsets and the total amount of buffer space needed. |
Dan Gohman | 35bfb24 | 2015-12-04 23:22:35 +0000 | [diff] [blame] | 611 | for (SDValue Arg : |
| 612 | make_range(OutVals.begin() + NumFixedArgs, OutVals.end())) { |
| 613 | EVT VT = Arg.getValueType(); |
| 614 | assert(VT != MVT::iPTR && "Legalized args should be concrete"); |
| 615 | Type *Ty = VT.getTypeForEVT(*DAG.getContext()); |
Derek Schuff | 992d83f | 2016-02-10 20:14:15 +0000 | [diff] [blame] | 616 | unsigned Offset = CCInfo.AllocateStack(Layout.getTypeAllocSize(Ty), |
| 617 | Layout.getABITypeAlignment(Ty)); |
Dan Gohman | 35bfb24 | 2015-12-04 23:22:35 +0000 | [diff] [blame] | 618 | CCInfo.addLoc(CCValAssign::getMem(ArgLocs.size(), VT.getSimpleVT(), |
| 619 | Offset, VT.getSimpleVT(), |
| 620 | CCValAssign::Full)); |
| 621 | } |
| 622 | } |
| 623 | |
| 624 | unsigned NumBytes = CCInfo.getAlignedCallFrameSize(); |
| 625 | |
Derek Schuff | 27501e2 | 2016-02-10 19:51:04 +0000 | [diff] [blame] | 626 | SDValue FINode; |
| 627 | if (IsVarArg && NumBytes) { |
Dan Gohman | 35bfb24 | 2015-12-04 23:22:35 +0000 | [diff] [blame] | 628 | // For non-fixed arguments, next emit stores to store the argument values |
Derek Schuff | 27501e2 | 2016-02-10 19:51:04 +0000 | [diff] [blame] | 629 | // to the stack buffer at the offsets computed above. |
Matthias Braun | 941a705 | 2016-07-28 18:40:00 +0000 | [diff] [blame] | 630 | int FI = MF.getFrameInfo().CreateStackObject(NumBytes, |
| 631 | Layout.getStackAlignment(), |
| 632 | /*isSS=*/false); |
Dan Gohman | 35bfb24 | 2015-12-04 23:22:35 +0000 | [diff] [blame] | 633 | unsigned ValNo = 0; |
| 634 | SmallVector<SDValue, 8> Chains; |
| 635 | for (SDValue Arg : |
| 636 | make_range(OutVals.begin() + NumFixedArgs, OutVals.end())) { |
| 637 | assert(ArgLocs[ValNo].getValNo() == ValNo && |
| 638 | "ArgLocs should remain in order and only hold varargs args"); |
| 639 | unsigned Offset = ArgLocs[ValNo++].getLocMemOffset(); |
Derek Schuff | 992d83f | 2016-02-10 20:14:15 +0000 | [diff] [blame] | 640 | FINode = DAG.getFrameIndex(FI, getPointerTy(Layout)); |
Derek Schuff | 27501e2 | 2016-02-10 19:51:04 +0000 | [diff] [blame] | 641 | SDValue Add = DAG.getNode(ISD::ADD, DL, PtrVT, FINode, |
Dan Gohman | 35bfb24 | 2015-12-04 23:22:35 +0000 | [diff] [blame] | 642 | DAG.getConstant(Offset, DL, PtrVT)); |
Heejin Ahn | f208f63 | 2018-09-05 01:27:38 +0000 | [diff] [blame] | 643 | Chains.push_back( |
| 644 | DAG.getStore(Chain, DL, Arg, Add, |
| 645 | MachinePointerInfo::getFixedStack(MF, FI, Offset), 0)); |
Dan Gohman | 35bfb24 | 2015-12-04 23:22:35 +0000 | [diff] [blame] | 646 | } |
| 647 | if (!Chains.empty()) |
| 648 | Chain = DAG.getNode(ISD::TokenFactor, DL, MVT::Other, Chains); |
Derek Schuff | 27501e2 | 2016-02-10 19:51:04 +0000 | [diff] [blame] | 649 | } else if (IsVarArg) { |
| 650 | FINode = DAG.getIntPtrConstant(0, DL); |
Dan Gohman | 35bfb24 | 2015-12-04 23:22:35 +0000 | [diff] [blame] | 651 | } |
| 652 | |
| 653 | // Compute the operands for the CALLn node. |
JF Bastien | d8a9d66 | 2015-08-24 21:59:51 +0000 | [diff] [blame] | 654 | SmallVector<SDValue, 16> Ops; |
| 655 | Ops.push_back(Chain); |
JF Bastien | af111db | 2015-08-24 22:16:48 +0000 | [diff] [blame] | 656 | Ops.push_back(Callee); |
Dan Gohman | 35bfb24 | 2015-12-04 23:22:35 +0000 | [diff] [blame] | 657 | |
| 658 | // Add all fixed arguments. Note that for non-varargs calls, NumFixedArgs |
| 659 | // isn't reliable. |
| 660 | Ops.append(OutVals.begin(), |
| 661 | IsVarArg ? OutVals.begin() + NumFixedArgs : OutVals.end()); |
Derek Schuff | 27501e2 | 2016-02-10 19:51:04 +0000 | [diff] [blame] | 662 | // Add a pointer to the vararg buffer. |
Heejin Ahn | f208f63 | 2018-09-05 01:27:38 +0000 | [diff] [blame] | 663 | if (IsVarArg) |
| 664 | Ops.push_back(FINode); |
JF Bastien | d8a9d66 | 2015-08-24 21:59:51 +0000 | [diff] [blame] | 665 | |
Derek Schuff | 27501e2 | 2016-02-10 19:51:04 +0000 | [diff] [blame] | 666 | SmallVector<EVT, 8> InTys; |
Dan Gohman | 2d822e7 | 2015-12-04 17:12:52 +0000 | [diff] [blame] | 667 | for (const auto &In : Ins) { |
Dan Gohman | 7935fa3 | 2015-12-10 00:22:40 +0000 | [diff] [blame] | 668 | assert(!In.Flags.isByVal() && "byval is not valid for return values"); |
| 669 | assert(!In.Flags.isNest() && "nest is not valid for return values"); |
Dan Gohman | 2d822e7 | 2015-12-04 17:12:52 +0000 | [diff] [blame] | 670 | if (In.Flags.isInAlloca()) |
Dan Gohman | 7935fa3 | 2015-12-10 00:22:40 +0000 | [diff] [blame] | 671 | fail(DL, DAG, "WebAssembly hasn't implemented inalloca return values"); |
Dan Gohman | 2d822e7 | 2015-12-04 17:12:52 +0000 | [diff] [blame] | 672 | if (In.Flags.isInConsecutiveRegs()) |
Dan Gohman | 7935fa3 | 2015-12-10 00:22:40 +0000 | [diff] [blame] | 673 | fail(DL, DAG, "WebAssembly hasn't implemented cons regs return values"); |
Dan Gohman | 2d822e7 | 2015-12-04 17:12:52 +0000 | [diff] [blame] | 674 | if (In.Flags.isInConsecutiveRegsLast()) |
Dan Gohman | 4b9d791 | 2015-12-15 22:01:29 +0000 | [diff] [blame] | 675 | fail(DL, DAG, |
| 676 | "WebAssembly hasn't implemented cons regs last return values"); |
Dan Gohman | 2d822e7 | 2015-12-04 17:12:52 +0000 | [diff] [blame] | 677 | // Ignore In.getOrigAlign() because all our arguments are passed in |
| 678 | // registers. |
Derek Schuff | 27501e2 | 2016-02-10 19:51:04 +0000 | [diff] [blame] | 679 | InTys.push_back(In.VT); |
Dan Gohman | 2d822e7 | 2015-12-04 17:12:52 +0000 | [diff] [blame] | 680 | } |
Derek Schuff | 27501e2 | 2016-02-10 19:51:04 +0000 | [diff] [blame] | 681 | InTys.push_back(MVT::Other); |
| 682 | SDVTList InTyList = DAG.getVTList(InTys); |
Dan Gohman | f71abef | 2015-09-09 16:13:47 +0000 | [diff] [blame] | 683 | SDValue Res = |
| 684 | DAG.getNode(Ins.empty() ? WebAssemblyISD::CALL0 : WebAssemblyISD::CALL1, |
Derek Schuff | 27501e2 | 2016-02-10 19:51:04 +0000 | [diff] [blame] | 685 | DL, InTyList, Ops); |
JF Bastien | af111db | 2015-08-24 22:16:48 +0000 | [diff] [blame] | 686 | if (Ins.empty()) { |
| 687 | Chain = Res; |
| 688 | } else { |
| 689 | InVals.push_back(Res); |
| 690 | Chain = Res.getValue(1); |
| 691 | } |
JF Bastien | d8a9d66 | 2015-08-24 21:59:51 +0000 | [diff] [blame] | 692 | |
JF Bastien | d8a9d66 | 2015-08-24 21:59:51 +0000 | [diff] [blame] | 693 | return Chain; |
| 694 | } |
| 695 | |
JF Bastien | b9073fb | 2015-07-22 21:28:15 +0000 | [diff] [blame] | 696 | bool WebAssemblyTargetLowering::CanLowerReturn( |
Dan Gohman | 7a6b982 | 2015-11-29 22:32:02 +0000 | [diff] [blame] | 697 | CallingConv::ID /*CallConv*/, MachineFunction & /*MF*/, bool /*IsVarArg*/, |
| 698 | const SmallVectorImpl<ISD::OutputArg> &Outs, |
| 699 | LLVMContext & /*Context*/) const { |
JF Bastien | b9073fb | 2015-07-22 21:28:15 +0000 | [diff] [blame] | 700 | // WebAssembly can't currently handle returning tuples. |
| 701 | return Outs.size() <= 1; |
| 702 | } |
| 703 | |
| 704 | SDValue WebAssemblyTargetLowering::LowerReturn( |
Dan Gohman | 35bfb24 | 2015-12-04 23:22:35 +0000 | [diff] [blame] | 705 | SDValue Chain, CallingConv::ID CallConv, bool /*IsVarArg*/, |
JF Bastien | b9073fb | 2015-07-22 21:28:15 +0000 | [diff] [blame] | 706 | const SmallVectorImpl<ISD::OutputArg> &Outs, |
Benjamin Kramer | bdc4956 | 2016-06-12 15:39:02 +0000 | [diff] [blame] | 707 | const SmallVectorImpl<SDValue> &OutVals, const SDLoc &DL, |
JF Bastien | b9073fb | 2015-07-22 21:28:15 +0000 | [diff] [blame] | 708 | SelectionDAG &DAG) const { |
JF Bastien | b9073fb | 2015-07-22 21:28:15 +0000 | [diff] [blame] | 709 | assert(Outs.size() <= 1 && "WebAssembly can only return up to one value"); |
Dan Gohman | 85dbdda | 2015-12-04 17:16:07 +0000 | [diff] [blame] | 710 | if (!CallingConvSupported(CallConv)) |
JF Bastien | b9073fb | 2015-07-22 21:28:15 +0000 | [diff] [blame] | 711 | fail(DL, DAG, "WebAssembly doesn't support non-C calling conventions"); |
| 712 | |
JF Bastien | 600aee9 | 2015-07-31 17:53:38 +0000 | [diff] [blame] | 713 | SmallVector<SDValue, 4> RetOps(1, Chain); |
| 714 | RetOps.append(OutVals.begin(), OutVals.end()); |
JF Bastien | 4a2d560 | 2015-07-31 21:04:18 +0000 | [diff] [blame] | 715 | Chain = DAG.getNode(WebAssemblyISD::RETURN, DL, MVT::Other, RetOps); |
JF Bastien | b9073fb | 2015-07-22 21:28:15 +0000 | [diff] [blame] | 716 | |
Dan Gohman | 754cd11 | 2015-11-11 01:33:02 +0000 | [diff] [blame] | 717 | // Record the number and types of the return values. |
| 718 | for (const ISD::OutputArg &Out : Outs) { |
Dan Gohman | ac132e9 | 2015-12-02 23:40:03 +0000 | [diff] [blame] | 719 | assert(!Out.Flags.isByVal() && "byval is not valid for return values"); |
| 720 | assert(!Out.Flags.isNest() && "nest is not valid for return values"); |
Dan Gohman | 35bfb24 | 2015-12-04 23:22:35 +0000 | [diff] [blame] | 721 | assert(Out.IsFixed && "non-fixed return value is not valid"); |
Dan Gohman | 754cd11 | 2015-11-11 01:33:02 +0000 | [diff] [blame] | 722 | if (Out.Flags.isInAlloca()) |
| 723 | fail(DL, DAG, "WebAssembly hasn't implemented inalloca results"); |
Dan Gohman | 754cd11 | 2015-11-11 01:33:02 +0000 | [diff] [blame] | 724 | if (Out.Flags.isInConsecutiveRegs()) |
| 725 | fail(DL, DAG, "WebAssembly hasn't implemented cons regs results"); |
| 726 | if (Out.Flags.isInConsecutiveRegsLast()) |
| 727 | fail(DL, DAG, "WebAssembly hasn't implemented cons regs last results"); |
Dan Gohman | 754cd11 | 2015-11-11 01:33:02 +0000 | [diff] [blame] | 728 | } |
| 729 | |
JF Bastien | b9073fb | 2015-07-22 21:28:15 +0000 | [diff] [blame] | 730 | return Chain; |
| 731 | } |
| 732 | |
| 733 | SDValue WebAssemblyTargetLowering::LowerFormalArguments( |
Derek Schuff | 27501e2 | 2016-02-10 19:51:04 +0000 | [diff] [blame] | 734 | SDValue Chain, CallingConv::ID CallConv, bool IsVarArg, |
Benjamin Kramer | bdc4956 | 2016-06-12 15:39:02 +0000 | [diff] [blame] | 735 | const SmallVectorImpl<ISD::InputArg> &Ins, const SDLoc &DL, |
| 736 | SelectionDAG &DAG, SmallVectorImpl<SDValue> &InVals) const { |
Dan Gohman | 85dbdda | 2015-12-04 17:16:07 +0000 | [diff] [blame] | 737 | if (!CallingConvSupported(CallConv)) |
JF Bastien | b9073fb | 2015-07-22 21:28:15 +0000 | [diff] [blame] | 738 | fail(DL, DAG, "WebAssembly doesn't support non-C calling conventions"); |
JF Bastien | b9073fb | 2015-07-22 21:28:15 +0000 | [diff] [blame] | 739 | |
Dan Gohman | 2726b88 | 2016-10-06 22:29:32 +0000 | [diff] [blame] | 740 | MachineFunction &MF = DAG.getMachineFunction(); |
| 741 | auto *MFI = MF.getInfo<WebAssemblyFunctionInfo>(); |
| 742 | |
Dan Gohman | fb3e059 | 2015-11-25 19:36:19 +0000 | [diff] [blame] | 743 | // Set up the incoming ARGUMENTS value, which serves to represent the liveness |
| 744 | // of the incoming values before they're represented by virtual registers. |
| 745 | MF.getRegInfo().addLiveIn(WebAssembly::ARGUMENTS); |
| 746 | |
JF Bastien | 600aee9 | 2015-07-31 17:53:38 +0000 | [diff] [blame] | 747 | for (const ISD::InputArg &In : Ins) { |
JF Bastien | 600aee9 | 2015-07-31 17:53:38 +0000 | [diff] [blame] | 748 | if (In.Flags.isInAlloca()) |
| 749 | fail(DL, DAG, "WebAssembly hasn't implemented inalloca arguments"); |
| 750 | if (In.Flags.isNest()) |
| 751 | fail(DL, DAG, "WebAssembly hasn't implemented nest arguments"); |
JF Bastien | 600aee9 | 2015-07-31 17:53:38 +0000 | [diff] [blame] | 752 | if (In.Flags.isInConsecutiveRegs()) |
| 753 | fail(DL, DAG, "WebAssembly hasn't implemented cons regs arguments"); |
| 754 | if (In.Flags.isInConsecutiveRegsLast()) |
| 755 | fail(DL, DAG, "WebAssembly hasn't implemented cons regs last arguments"); |
Dan Gohman | 9c54d3b | 2015-11-25 18:13:18 +0000 | [diff] [blame] | 756 | // Ignore In.getOrigAlign() because all our arguments are passed in |
| 757 | // registers. |
Heejin Ahn | f208f63 | 2018-09-05 01:27:38 +0000 | [diff] [blame] | 758 | InVals.push_back(In.Used ? DAG.getNode(WebAssemblyISD::ARGUMENT, DL, In.VT, |
| 759 | DAG.getTargetConstant(InVals.size(), |
| 760 | DL, MVT::i32)) |
| 761 | : DAG.getUNDEF(In.VT)); |
Dan Gohman | 754cd11 | 2015-11-11 01:33:02 +0000 | [diff] [blame] | 762 | |
| 763 | // Record the number and types of arguments. |
Derek Schuff | 27501e2 | 2016-02-10 19:51:04 +0000 | [diff] [blame] | 764 | MFI->addParam(In.VT); |
JF Bastien | 600aee9 | 2015-07-31 17:53:38 +0000 | [diff] [blame] | 765 | } |
JF Bastien | b9073fb | 2015-07-22 21:28:15 +0000 | [diff] [blame] | 766 | |
Derek Schuff | 27501e2 | 2016-02-10 19:51:04 +0000 | [diff] [blame] | 767 | // Varargs are copied into a buffer allocated by the caller, and a pointer to |
| 768 | // the buffer is passed as an argument. |
| 769 | if (IsVarArg) { |
| 770 | MVT PtrVT = getPointerTy(MF.getDataLayout()); |
| 771 | unsigned VarargVreg = |
| 772 | MF.getRegInfo().createVirtualRegister(getRegClassFor(PtrVT)); |
| 773 | MFI->setVarargBufferVreg(VarargVreg); |
| 774 | Chain = DAG.getCopyToReg( |
| 775 | Chain, DL, VarargVreg, |
| 776 | DAG.getNode(WebAssemblyISD::ARGUMENT, DL, PtrVT, |
| 777 | DAG.getTargetConstant(Ins.size(), DL, MVT::i32))); |
| 778 | MFI->addParam(PtrVT); |
| 779 | } |
Dan Gohman | 35bfb24 | 2015-12-04 23:22:35 +0000 | [diff] [blame] | 780 | |
Derek Schuff | 77a7a38 | 2018-10-03 22:22:48 +0000 | [diff] [blame] | 781 | // Record the number and types of arguments and results. |
Dan Gohman | 2726b88 | 2016-10-06 22:29:32 +0000 | [diff] [blame] | 782 | SmallVector<MVT, 4> Params; |
| 783 | SmallVector<MVT, 4> Results; |
Derek Schuff | 77a7a38 | 2018-10-03 22:22:48 +0000 | [diff] [blame] | 784 | ComputeSignatureVTs(MF.getFunction().getFunctionType(), MF.getFunction(), |
| 785 | DAG.getTarget(), Params, Results); |
Dan Gohman | 2726b88 | 2016-10-06 22:29:32 +0000 | [diff] [blame] | 786 | for (MVT VT : Results) |
| 787 | MFI->addResult(VT); |
Derek Schuff | 77a7a38 | 2018-10-03 22:22:48 +0000 | [diff] [blame] | 788 | // TODO: Use signatures in WebAssemblyMachineFunctionInfo too and unify |
| 789 | // the param logic here with ComputeSignatureVTs |
| 790 | assert(MFI->getParams().size() == Params.size() && |
| 791 | std::equal(MFI->getParams().begin(), MFI->getParams().end(), |
| 792 | Params.begin())); |
Dan Gohman | 2726b88 | 2016-10-06 22:29:32 +0000 | [diff] [blame] | 793 | |
JF Bastien | b9073fb | 2015-07-22 21:28:15 +0000 | [diff] [blame] | 794 | return Chain; |
| 795 | } |
| 796 | |
Dan Gohman | 10e730a | 2015-06-29 23:51:55 +0000 | [diff] [blame] | 797 | //===----------------------------------------------------------------------===// |
JF Bastien | af111db | 2015-08-24 22:16:48 +0000 | [diff] [blame] | 798 | // Custom lowering hooks. |
Dan Gohman | 10e730a | 2015-06-29 23:51:55 +0000 | [diff] [blame] | 799 | //===----------------------------------------------------------------------===// |
| 800 | |
JF Bastien | af111db | 2015-08-24 22:16:48 +0000 | [diff] [blame] | 801 | SDValue WebAssemblyTargetLowering::LowerOperation(SDValue Op, |
| 802 | SelectionDAG &DAG) const { |
Derek Schuff | 51699a8 | 2016-02-12 22:56:03 +0000 | [diff] [blame] | 803 | SDLoc DL(Op); |
JF Bastien | af111db | 2015-08-24 22:16:48 +0000 | [diff] [blame] | 804 | switch (Op.getOpcode()) { |
Heejin Ahn | f208f63 | 2018-09-05 01:27:38 +0000 | [diff] [blame] | 805 | default: |
| 806 | llvm_unreachable("unimplemented operation lowering"); |
| 807 | return SDValue(); |
| 808 | case ISD::FrameIndex: |
| 809 | return LowerFrameIndex(Op, DAG); |
| 810 | case ISD::GlobalAddress: |
| 811 | return LowerGlobalAddress(Op, DAG); |
| 812 | case ISD::ExternalSymbol: |
| 813 | return LowerExternalSymbol(Op, DAG); |
| 814 | case ISD::JumpTable: |
| 815 | return LowerJumpTable(Op, DAG); |
| 816 | case ISD::BR_JT: |
| 817 | return LowerBR_JT(Op, DAG); |
| 818 | case ISD::VASTART: |
| 819 | return LowerVASTART(Op, DAG); |
| 820 | case ISD::BlockAddress: |
| 821 | case ISD::BRIND: |
| 822 | fail(DL, DAG, "WebAssembly hasn't implemented computed gotos"); |
| 823 | return SDValue(); |
| 824 | case ISD::RETURNADDR: // Probably nothing meaningful can be returned here. |
| 825 | fail(DL, DAG, "WebAssembly hasn't implemented __builtin_return_address"); |
| 826 | return SDValue(); |
| 827 | case ISD::FRAMEADDR: |
| 828 | return LowerFRAMEADDR(Op, DAG); |
| 829 | case ISD::CopyToReg: |
| 830 | return LowerCopyToReg(Op, DAG); |
| 831 | case ISD::INTRINSIC_WO_CHAIN: |
| 832 | return LowerINTRINSIC_WO_CHAIN(Op, DAG); |
Thomas Lively | a0d2581 | 2018-09-07 21:54:46 +0000 | [diff] [blame] | 833 | case ISD::VECTOR_SHUFFLE: |
| 834 | return LowerVECTOR_SHUFFLE(Op, DAG); |
Thomas Lively | 55735d5 | 2018-10-20 01:31:18 +0000 | [diff] [blame] | 835 | case ISD::SHL: |
| 836 | case ISD::SRA: |
| 837 | case ISD::SRL: |
| 838 | return LowerShift(Op, DAG); |
JF Bastien | af111db | 2015-08-24 22:16:48 +0000 | [diff] [blame] | 839 | } |
| 840 | } |
| 841 | |
Derek Schuff | aadc89c | 2016-02-16 18:18:36 +0000 | [diff] [blame] | 842 | SDValue WebAssemblyTargetLowering::LowerCopyToReg(SDValue Op, |
| 843 | SelectionDAG &DAG) const { |
| 844 | SDValue Src = Op.getOperand(2); |
| 845 | if (isa<FrameIndexSDNode>(Src.getNode())) { |
| 846 | // CopyToReg nodes don't support FrameIndex operands. Other targets select |
| 847 | // the FI to some LEA-like instruction, but since we don't have that, we |
| 848 | // need to insert some kind of instruction that can take an FI operand and |
| 849 | // produces a value usable by CopyToReg (i.e. in a vreg). So insert a dummy |
| 850 | // copy_local between Op and its FI operand. |
Dan Gohman | 02c0871 | 2016-02-20 23:09:44 +0000 | [diff] [blame] | 851 | SDValue Chain = Op.getOperand(0); |
Derek Schuff | aadc89c | 2016-02-16 18:18:36 +0000 | [diff] [blame] | 852 | SDLoc DL(Op); |
Dan Gohman | 02c0871 | 2016-02-20 23:09:44 +0000 | [diff] [blame] | 853 | unsigned Reg = cast<RegisterSDNode>(Op.getOperand(1))->getReg(); |
Derek Schuff | aadc89c | 2016-02-16 18:18:36 +0000 | [diff] [blame] | 854 | EVT VT = Src.getValueType(); |
Heejin Ahn | f208f63 | 2018-09-05 01:27:38 +0000 | [diff] [blame] | 855 | SDValue Copy(DAG.getMachineNode(VT == MVT::i32 ? WebAssembly::COPY_I32 |
| 856 | : WebAssembly::COPY_I64, |
| 857 | DL, VT, Src), |
| 858 | 0); |
Dan Gohman | 02c0871 | 2016-02-20 23:09:44 +0000 | [diff] [blame] | 859 | return Op.getNode()->getNumValues() == 1 |
| 860 | ? DAG.getCopyToReg(Chain, DL, Reg, Copy) |
Heejin Ahn | f208f63 | 2018-09-05 01:27:38 +0000 | [diff] [blame] | 861 | : DAG.getCopyToReg(Chain, DL, Reg, Copy, |
| 862 | Op.getNumOperands() == 4 ? Op.getOperand(3) |
| 863 | : SDValue()); |
Derek Schuff | aadc89c | 2016-02-16 18:18:36 +0000 | [diff] [blame] | 864 | } |
| 865 | return SDValue(); |
| 866 | } |
| 867 | |
Derek Schuff | 9769deb | 2015-12-11 23:49:46 +0000 | [diff] [blame] | 868 | SDValue WebAssemblyTargetLowering::LowerFrameIndex(SDValue Op, |
| 869 | SelectionDAG &DAG) const { |
| 870 | int FI = cast<FrameIndexSDNode>(Op)->getIndex(); |
| 871 | return DAG.getTargetFrameIndex(FI, Op.getValueType()); |
| 872 | } |
| 873 | |
Dan Gohman | 94c6566 | 2016-02-16 23:48:04 +0000 | [diff] [blame] | 874 | SDValue WebAssemblyTargetLowering::LowerFRAMEADDR(SDValue Op, |
| 875 | SelectionDAG &DAG) const { |
| 876 | // Non-zero depths are not supported by WebAssembly currently. Use the |
| 877 | // legalizer's default expansion, which is to return 0 (what this function is |
| 878 | // documented to do). |
Dan Gohman | 1d547bf | 2016-02-17 00:14:03 +0000 | [diff] [blame] | 879 | if (Op.getConstantOperandVal(0) > 0) |
Dan Gohman | 94c6566 | 2016-02-16 23:48:04 +0000 | [diff] [blame] | 880 | return SDValue(); |
| 881 | |
Matthias Braun | 941a705 | 2016-07-28 18:40:00 +0000 | [diff] [blame] | 882 | DAG.getMachineFunction().getFrameInfo().setFrameAddressIsTaken(true); |
Dan Gohman | 94c6566 | 2016-02-16 23:48:04 +0000 | [diff] [blame] | 883 | EVT VT = Op.getValueType(); |
| 884 | unsigned FP = |
| 885 | Subtarget->getRegisterInfo()->getFrameRegister(DAG.getMachineFunction()); |
| 886 | return DAG.getCopyFromReg(DAG.getEntryNode(), SDLoc(Op), FP, VT); |
| 887 | } |
| 888 | |
JF Bastien | af111db | 2015-08-24 22:16:48 +0000 | [diff] [blame] | 889 | SDValue WebAssemblyTargetLowering::LowerGlobalAddress(SDValue Op, |
| 890 | SelectionDAG &DAG) const { |
| 891 | SDLoc DL(Op); |
| 892 | const auto *GA = cast<GlobalAddressSDNode>(Op); |
| 893 | EVT VT = Op.getValueType(); |
Dan Gohman | 26c6765 | 2016-01-11 23:38:05 +0000 | [diff] [blame] | 894 | assert(GA->getTargetFlags() == 0 && |
| 895 | "Unexpected target flags on generic GlobalAddressSDNode"); |
JF Bastien | af111db | 2015-08-24 22:16:48 +0000 | [diff] [blame] | 896 | if (GA->getAddressSpace() != 0) |
| 897 | fail(DL, DAG, "WebAssembly only expects the 0 address space"); |
Dan Gohman | 4b9d791 | 2015-12-15 22:01:29 +0000 | [diff] [blame] | 898 | return DAG.getNode( |
| 899 | WebAssemblyISD::Wrapper, DL, VT, |
| 900 | DAG.getTargetGlobalAddress(GA->getGlobal(), DL, VT, GA->getOffset())); |
JF Bastien | af111db | 2015-08-24 22:16:48 +0000 | [diff] [blame] | 901 | } |
| 902 | |
Heejin Ahn | f208f63 | 2018-09-05 01:27:38 +0000 | [diff] [blame] | 903 | SDValue |
| 904 | WebAssemblyTargetLowering::LowerExternalSymbol(SDValue Op, |
| 905 | SelectionDAG &DAG) const { |
Dan Gohman | 2c8fe6a | 2015-11-25 16:44:29 +0000 | [diff] [blame] | 906 | SDLoc DL(Op); |
| 907 | const auto *ES = cast<ExternalSymbolSDNode>(Op); |
| 908 | EVT VT = Op.getValueType(); |
Dan Gohman | 26c6765 | 2016-01-11 23:38:05 +0000 | [diff] [blame] | 909 | assert(ES->getTargetFlags() == 0 && |
| 910 | "Unexpected target flags on generic ExternalSymbolSDNode"); |
| 911 | // Set the TargetFlags to 0x1 which indicates that this is a "function" |
| 912 | // symbol rather than a data symbol. We do this unconditionally even though |
| 913 | // we don't know anything about the symbol other than its name, because all |
| 914 | // external symbols used in target-independent SelectionDAG code are for |
| 915 | // functions. |
Heejin Ahn | f208f63 | 2018-09-05 01:27:38 +0000 | [diff] [blame] | 916 | return DAG.getNode( |
| 917 | WebAssemblyISD::Wrapper, DL, VT, |
| 918 | DAG.getTargetExternalSymbol(ES->getSymbol(), VT, |
| 919 | WebAssemblyII::MO_SYMBOL_FUNCTION)); |
Dan Gohman | 2c8fe6a | 2015-11-25 16:44:29 +0000 | [diff] [blame] | 920 | } |
| 921 | |
Dan Gohman | 950a13c | 2015-09-16 16:51:30 +0000 | [diff] [blame] | 922 | SDValue WebAssemblyTargetLowering::LowerJumpTable(SDValue Op, |
| 923 | SelectionDAG &DAG) const { |
| 924 | // There's no need for a Wrapper node because we always incorporate a jump |
Dan Gohman | 1402606 | 2016-03-08 03:18:12 +0000 | [diff] [blame] | 925 | // table operand into a BR_TABLE instruction, rather than ever |
Dan Gohman | bb7ce8e | 2015-11-20 03:02:49 +0000 | [diff] [blame] | 926 | // materializing it in a register. |
Dan Gohman | 950a13c | 2015-09-16 16:51:30 +0000 | [diff] [blame] | 927 | const JumpTableSDNode *JT = cast<JumpTableSDNode>(Op); |
| 928 | return DAG.getTargetJumpTable(JT->getIndex(), Op.getValueType(), |
| 929 | JT->getTargetFlags()); |
| 930 | } |
| 931 | |
| 932 | SDValue WebAssemblyTargetLowering::LowerBR_JT(SDValue Op, |
| 933 | SelectionDAG &DAG) const { |
| 934 | SDLoc DL(Op); |
| 935 | SDValue Chain = Op.getOperand(0); |
| 936 | const auto *JT = cast<JumpTableSDNode>(Op.getOperand(1)); |
| 937 | SDValue Index = Op.getOperand(2); |
| 938 | assert(JT->getTargetFlags() == 0 && "WebAssembly doesn't set target flags"); |
| 939 | |
| 940 | SmallVector<SDValue, 8> Ops; |
| 941 | Ops.push_back(Chain); |
| 942 | Ops.push_back(Index); |
| 943 | |
| 944 | MachineJumpTableInfo *MJTI = DAG.getMachineFunction().getJumpTableInfo(); |
| 945 | const auto &MBBs = MJTI->getJumpTables()[JT->getIndex()].MBBs; |
| 946 | |
Dan Gohman | 1402606 | 2016-03-08 03:18:12 +0000 | [diff] [blame] | 947 | // Add an operand for each case. |
Heejin Ahn | f208f63 | 2018-09-05 01:27:38 +0000 | [diff] [blame] | 948 | for (auto MBB : MBBs) |
| 949 | Ops.push_back(DAG.getBasicBlock(MBB)); |
Dan Gohman | 1402606 | 2016-03-08 03:18:12 +0000 | [diff] [blame] | 950 | |
Dan Gohman | 950a13c | 2015-09-16 16:51:30 +0000 | [diff] [blame] | 951 | // TODO: For now, we just pick something arbitrary for a default case for now. |
| 952 | // We really want to sniff out the guard and put in the real default case (and |
| 953 | // delete the guard). |
| 954 | Ops.push_back(DAG.getBasicBlock(MBBs[0])); |
| 955 | |
Dan Gohman | 1402606 | 2016-03-08 03:18:12 +0000 | [diff] [blame] | 956 | return DAG.getNode(WebAssemblyISD::BR_TABLE, DL, MVT::Other, Ops); |
Dan Gohman | 950a13c | 2015-09-16 16:51:30 +0000 | [diff] [blame] | 957 | } |
| 958 | |
Dan Gohman | 35bfb24 | 2015-12-04 23:22:35 +0000 | [diff] [blame] | 959 | SDValue WebAssemblyTargetLowering::LowerVASTART(SDValue Op, |
| 960 | SelectionDAG &DAG) const { |
| 961 | SDLoc DL(Op); |
| 962 | EVT PtrVT = getPointerTy(DAG.getMachineFunction().getDataLayout()); |
| 963 | |
Derek Schuff | 27501e2 | 2016-02-10 19:51:04 +0000 | [diff] [blame] | 964 | auto *MFI = DAG.getMachineFunction().getInfo<WebAssemblyFunctionInfo>(); |
Dan Gohman | 35bfb24 | 2015-12-04 23:22:35 +0000 | [diff] [blame] | 965 | const Value *SV = cast<SrcValueSDNode>(Op.getOperand(2))->getValue(); |
Derek Schuff | 27501e2 | 2016-02-10 19:51:04 +0000 | [diff] [blame] | 966 | |
| 967 | SDValue ArgN = DAG.getCopyFromReg(DAG.getEntryNode(), DL, |
| 968 | MFI->getVarargBufferVreg(), PtrVT); |
| 969 | return DAG.getStore(Op.getOperand(0), DL, ArgN, Op.getOperand(1), |
Derek Schuff | 1a946e4 | 2016-07-15 19:35:43 +0000 | [diff] [blame] | 970 | MachinePointerInfo(SV), 0); |
Dan Gohman | 35bfb24 | 2015-12-04 23:22:35 +0000 | [diff] [blame] | 971 | } |
| 972 | |
Heejin Ahn | 5ef4d5f | 2018-05-31 22:25:54 +0000 | [diff] [blame] | 973 | SDValue |
| 974 | WebAssemblyTargetLowering::LowerINTRINSIC_WO_CHAIN(SDValue Op, |
| 975 | SelectionDAG &DAG) const { |
| 976 | unsigned IntNo = cast<ConstantSDNode>(Op.getOperand(0))->getZExtValue(); |
| 977 | SDLoc DL(Op); |
| 978 | switch (IntNo) { |
| 979 | default: |
| 980 | return {}; // Don't custom lower most intrinsics. |
Thomas Lively | 5d461c9 | 2018-10-03 23:02:23 +0000 | [diff] [blame] | 981 | |
Krasimir Georgiev | 547d824 | 2018-10-16 18:50:09 +0000 | [diff] [blame] | 982 | case Intrinsic::wasm_lsda: |
| 983 | // TODO For now, just return 0 not to crash |
| 984 | return DAG.getConstant(0, DL, Op.getValueType()); |
Heejin Ahn | 5ef4d5f | 2018-05-31 22:25:54 +0000 | [diff] [blame] | 985 | } |
| 986 | } |
| 987 | |
Thomas Lively | a0d2581 | 2018-09-07 21:54:46 +0000 | [diff] [blame] | 988 | SDValue |
| 989 | WebAssemblyTargetLowering::LowerVECTOR_SHUFFLE(SDValue Op, |
| 990 | SelectionDAG &DAG) const { |
| 991 | SDLoc DL(Op); |
| 992 | ArrayRef<int> Mask = cast<ShuffleVectorSDNode>(Op.getNode())->getMask(); |
| 993 | MVT VecType = Op.getOperand(0).getSimpleValueType(); |
| 994 | assert(VecType.is128BitVector() && "Unexpected shuffle vector type"); |
| 995 | size_t LaneBytes = VecType.getVectorElementType().getSizeInBits() / 8; |
| 996 | |
| 997 | // Space for two vector args and sixteen mask indices |
| 998 | SDValue Ops[18]; |
| 999 | size_t OpIdx = 0; |
| 1000 | Ops[OpIdx++] = Op.getOperand(0); |
| 1001 | Ops[OpIdx++] = Op.getOperand(1); |
| 1002 | |
| 1003 | // Expand mask indices to byte indices and materialize them as operands |
| 1004 | for (size_t I = 0, Lanes = Mask.size(); I < Lanes; ++I) { |
| 1005 | for (size_t J = 0; J < LaneBytes; ++J) { |
Thomas Lively | 11a332d0 | 2018-10-19 19:08:06 +0000 | [diff] [blame] | 1006 | // Lower undefs (represented by -1 in mask) to zero |
| 1007 | uint64_t ByteIndex = |
| 1008 | Mask[I] == -1 ? 0 : (uint64_t)Mask[I] * LaneBytes + J; |
| 1009 | Ops[OpIdx++] = DAG.getConstant(ByteIndex, DL, MVT::i32); |
Thomas Lively | a0d2581 | 2018-09-07 21:54:46 +0000 | [diff] [blame] | 1010 | } |
| 1011 | } |
| 1012 | |
| 1013 | return DAG.getNode(WebAssemblyISD::SHUFFLE, DL, MVT::v16i8, Ops); |
| 1014 | } |
| 1015 | |
Thomas Lively | 55735d5 | 2018-10-20 01:31:18 +0000 | [diff] [blame] | 1016 | SDValue WebAssemblyTargetLowering::LowerShift(SDValue Op, |
| 1017 | SelectionDAG &DAG) const { |
| 1018 | SDLoc DL(Op); |
| 1019 | auto *ShiftVec = dyn_cast<BuildVectorSDNode>(Op.getOperand(1).getNode()); |
| 1020 | APInt SplatValue, SplatUndef; |
| 1021 | unsigned SplatBitSize; |
| 1022 | bool HasAnyUndefs; |
| 1023 | if (!ShiftVec || !ShiftVec->isConstantSplat(SplatValue, SplatUndef, |
| 1024 | SplatBitSize, HasAnyUndefs)) |
| 1025 | return Op; |
| 1026 | unsigned Opcode; |
| 1027 | switch (Op.getOpcode()) { |
| 1028 | case ISD::SHL: |
| 1029 | Opcode = WebAssemblyISD::VEC_SHL; |
| 1030 | break; |
| 1031 | case ISD::SRA: |
| 1032 | Opcode = WebAssemblyISD::VEC_SHR_S; |
| 1033 | break; |
| 1034 | case ISD::SRL: |
| 1035 | Opcode = WebAssemblyISD::VEC_SHR_U; |
| 1036 | break; |
| 1037 | default: |
| 1038 | llvm_unreachable("unexpected opcode"); |
| 1039 | return Op; |
| 1040 | } |
| 1041 | return DAG.getNode(Opcode, DL, Op.getValueType(), Op.getOperand(0), |
| 1042 | DAG.getConstant(SplatValue.trunc(32), DL, MVT::i32)); |
| 1043 | } |
| 1044 | |
Dan Gohman | 10e730a | 2015-06-29 23:51:55 +0000 | [diff] [blame] | 1045 | //===----------------------------------------------------------------------===// |
| 1046 | // WebAssembly Optimization Hooks |
| 1047 | //===----------------------------------------------------------------------===// |