10b57cec5SDimitry Andric //===-- WebAssemblyFastISel.cpp - WebAssembly FastISel implementation -----===// 20b57cec5SDimitry Andric // 30b57cec5SDimitry Andric // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. 40b57cec5SDimitry Andric // See https://llvm.org/LICENSE.txt for license information. 50b57cec5SDimitry Andric // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception 60b57cec5SDimitry Andric // 70b57cec5SDimitry Andric //===----------------------------------------------------------------------===// 80b57cec5SDimitry Andric /// 90b57cec5SDimitry Andric /// \file 100b57cec5SDimitry Andric /// This file defines the WebAssembly-specific support for the FastISel 110b57cec5SDimitry Andric /// class. Some of the target-specific code is generated by tablegen in the file 120b57cec5SDimitry Andric /// WebAssemblyGenFastISel.inc, which is #included here. 130b57cec5SDimitry Andric /// 140b57cec5SDimitry Andric /// TODO: kill flags 150b57cec5SDimitry Andric /// 160b57cec5SDimitry Andric //===----------------------------------------------------------------------===// 170b57cec5SDimitry Andric 180b57cec5SDimitry Andric #include "MCTargetDesc/WebAssemblyMCTargetDesc.h" 190b57cec5SDimitry Andric #include "WebAssembly.h" 200b57cec5SDimitry Andric #include "WebAssemblyMachineFunctionInfo.h" 210b57cec5SDimitry Andric #include "WebAssemblySubtarget.h" 220b57cec5SDimitry Andric #include "WebAssemblyTargetMachine.h" 230b57cec5SDimitry Andric #include "llvm/Analysis/BranchProbabilityInfo.h" 240b57cec5SDimitry Andric #include "llvm/CodeGen/FastISel.h" 250b57cec5SDimitry Andric #include "llvm/CodeGen/FunctionLoweringInfo.h" 260b57cec5SDimitry Andric #include "llvm/CodeGen/MachineConstantPool.h" 270b57cec5SDimitry Andric #include "llvm/CodeGen/MachineFrameInfo.h" 280b57cec5SDimitry Andric #include "llvm/CodeGen/MachineInstrBuilder.h" 290b57cec5SDimitry Andric #include "llvm/CodeGen/MachineRegisterInfo.h" 300b57cec5SDimitry Andric #include "llvm/IR/DataLayout.h" 310b57cec5SDimitry Andric #include "llvm/IR/DerivedTypes.h" 320b57cec5SDimitry Andric #include "llvm/IR/Function.h" 330b57cec5SDimitry Andric #include "llvm/IR/GetElementPtrTypeIterator.h" 340b57cec5SDimitry Andric #include "llvm/IR/GlobalAlias.h" 350b57cec5SDimitry Andric #include "llvm/IR/GlobalVariable.h" 360b57cec5SDimitry Andric #include "llvm/IR/Instructions.h" 370b57cec5SDimitry Andric #include "llvm/IR/IntrinsicInst.h" 380b57cec5SDimitry Andric #include "llvm/IR/Operator.h" 390b57cec5SDimitry Andric #include "llvm/IR/PatternMatch.h" 400b57cec5SDimitry Andric 410b57cec5SDimitry Andric using namespace llvm; 420b57cec5SDimitry Andric using namespace PatternMatch; 430b57cec5SDimitry Andric 440b57cec5SDimitry Andric #define DEBUG_TYPE "wasm-fastisel" 450b57cec5SDimitry Andric 460b57cec5SDimitry Andric namespace { 470b57cec5SDimitry Andric 480b57cec5SDimitry Andric class WebAssemblyFastISel final : public FastISel { 490b57cec5SDimitry Andric // All possible address modes. 500b57cec5SDimitry Andric class Address { 510b57cec5SDimitry Andric public: 520b57cec5SDimitry Andric using BaseKind = enum { RegBase, FrameIndexBase }; 530b57cec5SDimitry Andric 540b57cec5SDimitry Andric private: 550b57cec5SDimitry Andric BaseKind Kind = RegBase; 560b57cec5SDimitry Andric union { 570b57cec5SDimitry Andric unsigned Reg; 580b57cec5SDimitry Andric int FI; 590b57cec5SDimitry Andric } Base; 600b57cec5SDimitry Andric 610b57cec5SDimitry Andric int64_t Offset = 0; 620b57cec5SDimitry Andric 630b57cec5SDimitry Andric const GlobalValue *GV = nullptr; 640b57cec5SDimitry Andric 650b57cec5SDimitry Andric public: 660b57cec5SDimitry Andric // Innocuous defaults for our address. 670b57cec5SDimitry Andric Address() { Base.Reg = 0; } 680b57cec5SDimitry Andric void setKind(BaseKind K) { 690b57cec5SDimitry Andric assert(!isSet() && "Can't change kind with non-zero base"); 700b57cec5SDimitry Andric Kind = K; 710b57cec5SDimitry Andric } 720b57cec5SDimitry Andric BaseKind getKind() const { return Kind; } 730b57cec5SDimitry Andric bool isRegBase() const { return Kind == RegBase; } 740b57cec5SDimitry Andric bool isFIBase() const { return Kind == FrameIndexBase; } 750b57cec5SDimitry Andric void setReg(unsigned Reg) { 760b57cec5SDimitry Andric assert(isRegBase() && "Invalid base register access!"); 770b57cec5SDimitry Andric assert(Base.Reg == 0 && "Overwriting non-zero register"); 780b57cec5SDimitry Andric Base.Reg = Reg; 790b57cec5SDimitry Andric } 800b57cec5SDimitry Andric unsigned getReg() const { 810b57cec5SDimitry Andric assert(isRegBase() && "Invalid base register access!"); 820b57cec5SDimitry Andric return Base.Reg; 830b57cec5SDimitry Andric } 840b57cec5SDimitry Andric void setFI(unsigned FI) { 850b57cec5SDimitry Andric assert(isFIBase() && "Invalid base frame index access!"); 860b57cec5SDimitry Andric assert(Base.FI == 0 && "Overwriting non-zero frame index"); 870b57cec5SDimitry Andric Base.FI = FI; 880b57cec5SDimitry Andric } 890b57cec5SDimitry Andric unsigned getFI() const { 900b57cec5SDimitry Andric assert(isFIBase() && "Invalid base frame index access!"); 910b57cec5SDimitry Andric return Base.FI; 920b57cec5SDimitry Andric } 930b57cec5SDimitry Andric 940b57cec5SDimitry Andric void setOffset(int64_t NewOffset) { 950b57cec5SDimitry Andric assert(NewOffset >= 0 && "Offsets must be non-negative"); 960b57cec5SDimitry Andric Offset = NewOffset; 970b57cec5SDimitry Andric } 980b57cec5SDimitry Andric int64_t getOffset() const { return Offset; } 990b57cec5SDimitry Andric void setGlobalValue(const GlobalValue *G) { GV = G; } 1000b57cec5SDimitry Andric const GlobalValue *getGlobalValue() const { return GV; } 1010b57cec5SDimitry Andric bool isSet() const { 1020b57cec5SDimitry Andric if (isRegBase()) { 1030b57cec5SDimitry Andric return Base.Reg != 0; 1040b57cec5SDimitry Andric } else { 1050b57cec5SDimitry Andric return Base.FI != 0; 1060b57cec5SDimitry Andric } 1070b57cec5SDimitry Andric } 1080b57cec5SDimitry Andric }; 1090b57cec5SDimitry Andric 1100b57cec5SDimitry Andric /// Keep a pointer to the WebAssemblySubtarget around so that we can make the 1110b57cec5SDimitry Andric /// right decision when generating code for different targets. 1120b57cec5SDimitry Andric const WebAssemblySubtarget *Subtarget; 1130b57cec5SDimitry Andric LLVMContext *Context; 1140b57cec5SDimitry Andric 1150b57cec5SDimitry Andric private: 1160b57cec5SDimitry Andric // Utility helper routines 1170b57cec5SDimitry Andric MVT::SimpleValueType getSimpleType(Type *Ty) { 1180b57cec5SDimitry Andric EVT VT = TLI.getValueType(DL, Ty, /*AllowUnknown=*/true); 1190b57cec5SDimitry Andric return VT.isSimple() ? VT.getSimpleVT().SimpleTy 1200b57cec5SDimitry Andric : MVT::INVALID_SIMPLE_VALUE_TYPE; 1210b57cec5SDimitry Andric } 1220b57cec5SDimitry Andric MVT::SimpleValueType getLegalType(MVT::SimpleValueType VT) { 1230b57cec5SDimitry Andric switch (VT) { 1240b57cec5SDimitry Andric case MVT::i1: 1250b57cec5SDimitry Andric case MVT::i8: 1260b57cec5SDimitry Andric case MVT::i16: 1270b57cec5SDimitry Andric return MVT::i32; 1280b57cec5SDimitry Andric case MVT::i32: 1290b57cec5SDimitry Andric case MVT::i64: 1300b57cec5SDimitry Andric case MVT::f32: 1310b57cec5SDimitry Andric case MVT::f64: 1320b57cec5SDimitry Andric case MVT::exnref: 1330b57cec5SDimitry Andric return VT; 1340b57cec5SDimitry Andric case MVT::f16: 1350b57cec5SDimitry Andric return MVT::f32; 1360b57cec5SDimitry Andric case MVT::v16i8: 1370b57cec5SDimitry Andric case MVT::v8i16: 1380b57cec5SDimitry Andric case MVT::v4i32: 1390b57cec5SDimitry Andric case MVT::v4f32: 1400b57cec5SDimitry Andric if (Subtarget->hasSIMD128()) 1410b57cec5SDimitry Andric return VT; 1420b57cec5SDimitry Andric break; 1430b57cec5SDimitry Andric case MVT::v2i64: 1440b57cec5SDimitry Andric case MVT::v2f64: 1450b57cec5SDimitry Andric if (Subtarget->hasUnimplementedSIMD128()) 1460b57cec5SDimitry Andric return VT; 1470b57cec5SDimitry Andric break; 1480b57cec5SDimitry Andric default: 1490b57cec5SDimitry Andric break; 1500b57cec5SDimitry Andric } 1510b57cec5SDimitry Andric return MVT::INVALID_SIMPLE_VALUE_TYPE; 1520b57cec5SDimitry Andric } 1530b57cec5SDimitry Andric bool computeAddress(const Value *Obj, Address &Addr); 1540b57cec5SDimitry Andric void materializeLoadStoreOperands(Address &Addr); 1550b57cec5SDimitry Andric void addLoadStoreOperands(const Address &Addr, const MachineInstrBuilder &MIB, 1560b57cec5SDimitry Andric MachineMemOperand *MMO); 1570b57cec5SDimitry Andric unsigned maskI1Value(unsigned Reg, const Value *V); 1580b57cec5SDimitry Andric unsigned getRegForI1Value(const Value *V, bool &Not); 1590b57cec5SDimitry Andric unsigned zeroExtendToI32(unsigned Reg, const Value *V, 1600b57cec5SDimitry Andric MVT::SimpleValueType From); 1610b57cec5SDimitry Andric unsigned signExtendToI32(unsigned Reg, const Value *V, 1620b57cec5SDimitry Andric MVT::SimpleValueType From); 1630b57cec5SDimitry Andric unsigned zeroExtend(unsigned Reg, const Value *V, MVT::SimpleValueType From, 1640b57cec5SDimitry Andric MVT::SimpleValueType To); 1650b57cec5SDimitry Andric unsigned signExtend(unsigned Reg, const Value *V, MVT::SimpleValueType From, 1660b57cec5SDimitry Andric MVT::SimpleValueType To); 1670b57cec5SDimitry Andric unsigned getRegForUnsignedValue(const Value *V); 1680b57cec5SDimitry Andric unsigned getRegForSignedValue(const Value *V); 1690b57cec5SDimitry Andric unsigned getRegForPromotedValue(const Value *V, bool IsSigned); 1700b57cec5SDimitry Andric unsigned notValue(unsigned Reg); 1710b57cec5SDimitry Andric unsigned copyValue(unsigned Reg); 1720b57cec5SDimitry Andric 1730b57cec5SDimitry Andric // Backend specific FastISel code. 1740b57cec5SDimitry Andric unsigned fastMaterializeAlloca(const AllocaInst *AI) override; 1750b57cec5SDimitry Andric unsigned fastMaterializeConstant(const Constant *C) override; 1760b57cec5SDimitry Andric bool fastLowerArguments() override; 1770b57cec5SDimitry Andric 1780b57cec5SDimitry Andric // Selection routines. 1790b57cec5SDimitry Andric bool selectCall(const Instruction *I); 1800b57cec5SDimitry Andric bool selectSelect(const Instruction *I); 1810b57cec5SDimitry Andric bool selectTrunc(const Instruction *I); 1820b57cec5SDimitry Andric bool selectZExt(const Instruction *I); 1830b57cec5SDimitry Andric bool selectSExt(const Instruction *I); 1840b57cec5SDimitry Andric bool selectICmp(const Instruction *I); 1850b57cec5SDimitry Andric bool selectFCmp(const Instruction *I); 1860b57cec5SDimitry Andric bool selectBitCast(const Instruction *I); 1870b57cec5SDimitry Andric bool selectLoad(const Instruction *I); 1880b57cec5SDimitry Andric bool selectStore(const Instruction *I); 1890b57cec5SDimitry Andric bool selectBr(const Instruction *I); 1900b57cec5SDimitry Andric bool selectRet(const Instruction *I); 1910b57cec5SDimitry Andric bool selectUnreachable(const Instruction *I); 1920b57cec5SDimitry Andric 1930b57cec5SDimitry Andric public: 1940b57cec5SDimitry Andric // Backend specific FastISel code. 1950b57cec5SDimitry Andric WebAssemblyFastISel(FunctionLoweringInfo &FuncInfo, 1960b57cec5SDimitry Andric const TargetLibraryInfo *LibInfo) 1970b57cec5SDimitry Andric : FastISel(FuncInfo, LibInfo, /*SkipTargetIndependentISel=*/true) { 1980b57cec5SDimitry Andric Subtarget = &FuncInfo.MF->getSubtarget<WebAssemblySubtarget>(); 1990b57cec5SDimitry Andric Context = &FuncInfo.Fn->getContext(); 2000b57cec5SDimitry Andric } 2010b57cec5SDimitry Andric 2020b57cec5SDimitry Andric bool fastSelectInstruction(const Instruction *I) override; 2030b57cec5SDimitry Andric 2040b57cec5SDimitry Andric #include "WebAssemblyGenFastISel.inc" 2050b57cec5SDimitry Andric }; 2060b57cec5SDimitry Andric 2070b57cec5SDimitry Andric } // end anonymous namespace 2080b57cec5SDimitry Andric 2090b57cec5SDimitry Andric bool WebAssemblyFastISel::computeAddress(const Value *Obj, Address &Addr) { 2100b57cec5SDimitry Andric const User *U = nullptr; 2110b57cec5SDimitry Andric unsigned Opcode = Instruction::UserOp1; 2120b57cec5SDimitry Andric if (const auto *I = dyn_cast<Instruction>(Obj)) { 2130b57cec5SDimitry Andric // Don't walk into other basic blocks unless the object is an alloca from 2140b57cec5SDimitry Andric // another block, otherwise it may not have a virtual register assigned. 2150b57cec5SDimitry Andric if (FuncInfo.StaticAllocaMap.count(static_cast<const AllocaInst *>(Obj)) || 2160b57cec5SDimitry Andric FuncInfo.MBBMap[I->getParent()] == FuncInfo.MBB) { 2170b57cec5SDimitry Andric Opcode = I->getOpcode(); 2180b57cec5SDimitry Andric U = I; 2190b57cec5SDimitry Andric } 2200b57cec5SDimitry Andric } else if (const auto *C = dyn_cast<ConstantExpr>(Obj)) { 2210b57cec5SDimitry Andric Opcode = C->getOpcode(); 2220b57cec5SDimitry Andric U = C; 2230b57cec5SDimitry Andric } 2240b57cec5SDimitry Andric 2250b57cec5SDimitry Andric if (auto *Ty = dyn_cast<PointerType>(Obj->getType())) 2260b57cec5SDimitry Andric if (Ty->getAddressSpace() > 255) 2270b57cec5SDimitry Andric // Fast instruction selection doesn't support the special 2280b57cec5SDimitry Andric // address spaces. 2290b57cec5SDimitry Andric return false; 2300b57cec5SDimitry Andric 2310b57cec5SDimitry Andric if (const auto *GV = dyn_cast<GlobalValue>(Obj)) { 2320b57cec5SDimitry Andric if (TLI.isPositionIndependent()) 2330b57cec5SDimitry Andric return false; 2340b57cec5SDimitry Andric if (Addr.getGlobalValue()) 2350b57cec5SDimitry Andric return false; 2360b57cec5SDimitry Andric if (GV->isThreadLocal()) 2370b57cec5SDimitry Andric return false; 2380b57cec5SDimitry Andric Addr.setGlobalValue(GV); 2390b57cec5SDimitry Andric return true; 2400b57cec5SDimitry Andric } 2410b57cec5SDimitry Andric 2420b57cec5SDimitry Andric switch (Opcode) { 2430b57cec5SDimitry Andric default: 2440b57cec5SDimitry Andric break; 2450b57cec5SDimitry Andric case Instruction::BitCast: { 2460b57cec5SDimitry Andric // Look through bitcasts. 2470b57cec5SDimitry Andric return computeAddress(U->getOperand(0), Addr); 2480b57cec5SDimitry Andric } 2490b57cec5SDimitry Andric case Instruction::IntToPtr: { 2500b57cec5SDimitry Andric // Look past no-op inttoptrs. 2510b57cec5SDimitry Andric if (TLI.getValueType(DL, U->getOperand(0)->getType()) == 2520b57cec5SDimitry Andric TLI.getPointerTy(DL)) 2530b57cec5SDimitry Andric return computeAddress(U->getOperand(0), Addr); 2540b57cec5SDimitry Andric break; 2550b57cec5SDimitry Andric } 2560b57cec5SDimitry Andric case Instruction::PtrToInt: { 2570b57cec5SDimitry Andric // Look past no-op ptrtoints. 2580b57cec5SDimitry Andric if (TLI.getValueType(DL, U->getType()) == TLI.getPointerTy(DL)) 2590b57cec5SDimitry Andric return computeAddress(U->getOperand(0), Addr); 2600b57cec5SDimitry Andric break; 2610b57cec5SDimitry Andric } 2620b57cec5SDimitry Andric case Instruction::GetElementPtr: { 2630b57cec5SDimitry Andric Address SavedAddr = Addr; 2640b57cec5SDimitry Andric uint64_t TmpOffset = Addr.getOffset(); 2650b57cec5SDimitry Andric // Non-inbounds geps can wrap; wasm's offsets can't. 2660b57cec5SDimitry Andric if (!cast<GEPOperator>(U)->isInBounds()) 2670b57cec5SDimitry Andric goto unsupported_gep; 2680b57cec5SDimitry Andric // Iterate through the GEP folding the constants into offsets where 2690b57cec5SDimitry Andric // we can. 2700b57cec5SDimitry Andric for (gep_type_iterator GTI = gep_type_begin(U), E = gep_type_end(U); 2710b57cec5SDimitry Andric GTI != E; ++GTI) { 2720b57cec5SDimitry Andric const Value *Op = GTI.getOperand(); 2730b57cec5SDimitry Andric if (StructType *STy = GTI.getStructTypeOrNull()) { 2740b57cec5SDimitry Andric const StructLayout *SL = DL.getStructLayout(STy); 2750b57cec5SDimitry Andric unsigned Idx = cast<ConstantInt>(Op)->getZExtValue(); 2760b57cec5SDimitry Andric TmpOffset += SL->getElementOffset(Idx); 2770b57cec5SDimitry Andric } else { 2780b57cec5SDimitry Andric uint64_t S = DL.getTypeAllocSize(GTI.getIndexedType()); 2790b57cec5SDimitry Andric for (;;) { 2800b57cec5SDimitry Andric if (const auto *CI = dyn_cast<ConstantInt>(Op)) { 2810b57cec5SDimitry Andric // Constant-offset addressing. 2820b57cec5SDimitry Andric TmpOffset += CI->getSExtValue() * S; 2830b57cec5SDimitry Andric break; 2840b57cec5SDimitry Andric } 2850b57cec5SDimitry Andric if (S == 1 && Addr.isRegBase() && Addr.getReg() == 0) { 2860b57cec5SDimitry Andric // An unscaled add of a register. Set it as the new base. 2870b57cec5SDimitry Andric unsigned Reg = getRegForValue(Op); 2880b57cec5SDimitry Andric if (Reg == 0) 2890b57cec5SDimitry Andric return false; 2900b57cec5SDimitry Andric Addr.setReg(Reg); 2910b57cec5SDimitry Andric break; 2920b57cec5SDimitry Andric } 2930b57cec5SDimitry Andric if (canFoldAddIntoGEP(U, Op)) { 2940b57cec5SDimitry Andric // A compatible add with a constant operand. Fold the constant. 2950b57cec5SDimitry Andric auto *CI = cast<ConstantInt>(cast<AddOperator>(Op)->getOperand(1)); 2960b57cec5SDimitry Andric TmpOffset += CI->getSExtValue() * S; 2970b57cec5SDimitry Andric // Iterate on the other operand. 2980b57cec5SDimitry Andric Op = cast<AddOperator>(Op)->getOperand(0); 2990b57cec5SDimitry Andric continue; 3000b57cec5SDimitry Andric } 3010b57cec5SDimitry Andric // Unsupported 3020b57cec5SDimitry Andric goto unsupported_gep; 3030b57cec5SDimitry Andric } 3040b57cec5SDimitry Andric } 3050b57cec5SDimitry Andric } 3060b57cec5SDimitry Andric // Don't fold in negative offsets. 3070b57cec5SDimitry Andric if (int64_t(TmpOffset) >= 0) { 3080b57cec5SDimitry Andric // Try to grab the base operand now. 3090b57cec5SDimitry Andric Addr.setOffset(TmpOffset); 3100b57cec5SDimitry Andric if (computeAddress(U->getOperand(0), Addr)) 3110b57cec5SDimitry Andric return true; 3120b57cec5SDimitry Andric } 3130b57cec5SDimitry Andric // We failed, restore everything and try the other options. 3140b57cec5SDimitry Andric Addr = SavedAddr; 3150b57cec5SDimitry Andric unsupported_gep: 3160b57cec5SDimitry Andric break; 3170b57cec5SDimitry Andric } 3180b57cec5SDimitry Andric case Instruction::Alloca: { 3190b57cec5SDimitry Andric const auto *AI = cast<AllocaInst>(Obj); 3200b57cec5SDimitry Andric DenseMap<const AllocaInst *, int>::iterator SI = 3210b57cec5SDimitry Andric FuncInfo.StaticAllocaMap.find(AI); 3220b57cec5SDimitry Andric if (SI != FuncInfo.StaticAllocaMap.end()) { 3230b57cec5SDimitry Andric if (Addr.isSet()) { 3240b57cec5SDimitry Andric return false; 3250b57cec5SDimitry Andric } 3260b57cec5SDimitry Andric Addr.setKind(Address::FrameIndexBase); 3270b57cec5SDimitry Andric Addr.setFI(SI->second); 3280b57cec5SDimitry Andric return true; 3290b57cec5SDimitry Andric } 3300b57cec5SDimitry Andric break; 3310b57cec5SDimitry Andric } 3320b57cec5SDimitry Andric case Instruction::Add: { 3330b57cec5SDimitry Andric // Adds of constants are common and easy enough. 3340b57cec5SDimitry Andric const Value *LHS = U->getOperand(0); 3350b57cec5SDimitry Andric const Value *RHS = U->getOperand(1); 3360b57cec5SDimitry Andric 3370b57cec5SDimitry Andric if (isa<ConstantInt>(LHS)) 3380b57cec5SDimitry Andric std::swap(LHS, RHS); 3390b57cec5SDimitry Andric 3400b57cec5SDimitry Andric if (const auto *CI = dyn_cast<ConstantInt>(RHS)) { 3410b57cec5SDimitry Andric uint64_t TmpOffset = Addr.getOffset() + CI->getSExtValue(); 3420b57cec5SDimitry Andric if (int64_t(TmpOffset) >= 0) { 3430b57cec5SDimitry Andric Addr.setOffset(TmpOffset); 3440b57cec5SDimitry Andric return computeAddress(LHS, Addr); 3450b57cec5SDimitry Andric } 3460b57cec5SDimitry Andric } 3470b57cec5SDimitry Andric 3480b57cec5SDimitry Andric Address Backup = Addr; 3490b57cec5SDimitry Andric if (computeAddress(LHS, Addr) && computeAddress(RHS, Addr)) 3500b57cec5SDimitry Andric return true; 3510b57cec5SDimitry Andric Addr = Backup; 3520b57cec5SDimitry Andric 3530b57cec5SDimitry Andric break; 3540b57cec5SDimitry Andric } 3550b57cec5SDimitry Andric case Instruction::Sub: { 3560b57cec5SDimitry Andric // Subs of constants are common and easy enough. 3570b57cec5SDimitry Andric const Value *LHS = U->getOperand(0); 3580b57cec5SDimitry Andric const Value *RHS = U->getOperand(1); 3590b57cec5SDimitry Andric 3600b57cec5SDimitry Andric if (const auto *CI = dyn_cast<ConstantInt>(RHS)) { 3610b57cec5SDimitry Andric int64_t TmpOffset = Addr.getOffset() - CI->getSExtValue(); 3620b57cec5SDimitry Andric if (TmpOffset >= 0) { 3630b57cec5SDimitry Andric Addr.setOffset(TmpOffset); 3640b57cec5SDimitry Andric return computeAddress(LHS, Addr); 3650b57cec5SDimitry Andric } 3660b57cec5SDimitry Andric } 3670b57cec5SDimitry Andric break; 3680b57cec5SDimitry Andric } 3690b57cec5SDimitry Andric } 3700b57cec5SDimitry Andric if (Addr.isSet()) { 3710b57cec5SDimitry Andric return false; 3720b57cec5SDimitry Andric } 3730b57cec5SDimitry Andric unsigned Reg = getRegForValue(Obj); 3740b57cec5SDimitry Andric if (Reg == 0) 3750b57cec5SDimitry Andric return false; 3760b57cec5SDimitry Andric Addr.setReg(Reg); 3770b57cec5SDimitry Andric return Addr.getReg() != 0; 3780b57cec5SDimitry Andric } 3790b57cec5SDimitry Andric 3800b57cec5SDimitry Andric void WebAssemblyFastISel::materializeLoadStoreOperands(Address &Addr) { 3810b57cec5SDimitry Andric if (Addr.isRegBase()) { 3820b57cec5SDimitry Andric unsigned Reg = Addr.getReg(); 3830b57cec5SDimitry Andric if (Reg == 0) { 3840b57cec5SDimitry Andric Reg = createResultReg(Subtarget->hasAddr64() ? &WebAssembly::I64RegClass 3850b57cec5SDimitry Andric : &WebAssembly::I32RegClass); 3860b57cec5SDimitry Andric unsigned Opc = Subtarget->hasAddr64() ? WebAssembly::CONST_I64 3870b57cec5SDimitry Andric : WebAssembly::CONST_I32; 3880b57cec5SDimitry Andric BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc, TII.get(Opc), Reg) 3890b57cec5SDimitry Andric .addImm(0); 3900b57cec5SDimitry Andric Addr.setReg(Reg); 3910b57cec5SDimitry Andric } 3920b57cec5SDimitry Andric } 3930b57cec5SDimitry Andric } 3940b57cec5SDimitry Andric 3950b57cec5SDimitry Andric void WebAssemblyFastISel::addLoadStoreOperands(const Address &Addr, 3960b57cec5SDimitry Andric const MachineInstrBuilder &MIB, 3970b57cec5SDimitry Andric MachineMemOperand *MMO) { 3980b57cec5SDimitry Andric // Set the alignment operand (this is rewritten in SetP2AlignOperands). 3990b57cec5SDimitry Andric // TODO: Disable SetP2AlignOperands for FastISel and just do it here. 4000b57cec5SDimitry Andric MIB.addImm(0); 4010b57cec5SDimitry Andric 4020b57cec5SDimitry Andric if (const GlobalValue *GV = Addr.getGlobalValue()) 4030b57cec5SDimitry Andric MIB.addGlobalAddress(GV, Addr.getOffset()); 4040b57cec5SDimitry Andric else 4050b57cec5SDimitry Andric MIB.addImm(Addr.getOffset()); 4060b57cec5SDimitry Andric 4070b57cec5SDimitry Andric if (Addr.isRegBase()) 4080b57cec5SDimitry Andric MIB.addReg(Addr.getReg()); 4090b57cec5SDimitry Andric else 4100b57cec5SDimitry Andric MIB.addFrameIndex(Addr.getFI()); 4110b57cec5SDimitry Andric 4120b57cec5SDimitry Andric MIB.addMemOperand(MMO); 4130b57cec5SDimitry Andric } 4140b57cec5SDimitry Andric 4150b57cec5SDimitry Andric unsigned WebAssemblyFastISel::maskI1Value(unsigned Reg, const Value *V) { 4160b57cec5SDimitry Andric return zeroExtendToI32(Reg, V, MVT::i1); 4170b57cec5SDimitry Andric } 4180b57cec5SDimitry Andric 4190b57cec5SDimitry Andric unsigned WebAssemblyFastISel::getRegForI1Value(const Value *V, bool &Not) { 4200b57cec5SDimitry Andric if (const auto *ICmp = dyn_cast<ICmpInst>(V)) 4210b57cec5SDimitry Andric if (const ConstantInt *C = dyn_cast<ConstantInt>(ICmp->getOperand(1))) 4220b57cec5SDimitry Andric if (ICmp->isEquality() && C->isZero() && C->getType()->isIntegerTy(32)) { 4230b57cec5SDimitry Andric Not = ICmp->isTrueWhenEqual(); 4240b57cec5SDimitry Andric return getRegForValue(ICmp->getOperand(0)); 4250b57cec5SDimitry Andric } 4260b57cec5SDimitry Andric 4270b57cec5SDimitry Andric Value *NotV; 4280b57cec5SDimitry Andric if (match(V, m_Not(m_Value(NotV))) && V->getType()->isIntegerTy(32)) { 4290b57cec5SDimitry Andric Not = true; 4300b57cec5SDimitry Andric return getRegForValue(NotV); 4310b57cec5SDimitry Andric } 4320b57cec5SDimitry Andric 4330b57cec5SDimitry Andric Not = false; 4340b57cec5SDimitry Andric unsigned Reg = getRegForValue(V); 4350b57cec5SDimitry Andric if (Reg == 0) 4360b57cec5SDimitry Andric return 0; 4370b57cec5SDimitry Andric return maskI1Value(Reg, V); 4380b57cec5SDimitry Andric } 4390b57cec5SDimitry Andric 4400b57cec5SDimitry Andric unsigned WebAssemblyFastISel::zeroExtendToI32(unsigned Reg, const Value *V, 4410b57cec5SDimitry Andric MVT::SimpleValueType From) { 4420b57cec5SDimitry Andric if (Reg == 0) 4430b57cec5SDimitry Andric return 0; 4440b57cec5SDimitry Andric 4450b57cec5SDimitry Andric switch (From) { 4460b57cec5SDimitry Andric case MVT::i1: 4470b57cec5SDimitry Andric // If the value is naturally an i1, we don't need to mask it. We only know 4480b57cec5SDimitry Andric // if a value is naturally an i1 if it is definitely lowered by FastISel, 4490b57cec5SDimitry Andric // not a DAG ISel fallback. 4500b57cec5SDimitry Andric if (V != nullptr && isa<Argument>(V) && cast<Argument>(V)->hasZExtAttr()) 4510b57cec5SDimitry Andric return copyValue(Reg); 4520b57cec5SDimitry Andric break; 4530b57cec5SDimitry Andric case MVT::i8: 4540b57cec5SDimitry Andric case MVT::i16: 4550b57cec5SDimitry Andric break; 4560b57cec5SDimitry Andric case MVT::i32: 4570b57cec5SDimitry Andric return copyValue(Reg); 4580b57cec5SDimitry Andric default: 4590b57cec5SDimitry Andric return 0; 4600b57cec5SDimitry Andric } 4610b57cec5SDimitry Andric 4620b57cec5SDimitry Andric unsigned Imm = createResultReg(&WebAssembly::I32RegClass); 4630b57cec5SDimitry Andric BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc, 4640b57cec5SDimitry Andric TII.get(WebAssembly::CONST_I32), Imm) 4650b57cec5SDimitry Andric .addImm(~(~uint64_t(0) << MVT(From).getSizeInBits())); 4660b57cec5SDimitry Andric 4670b57cec5SDimitry Andric unsigned Result = createResultReg(&WebAssembly::I32RegClass); 4680b57cec5SDimitry Andric BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc, 4690b57cec5SDimitry Andric TII.get(WebAssembly::AND_I32), Result) 4700b57cec5SDimitry Andric .addReg(Reg) 4710b57cec5SDimitry Andric .addReg(Imm); 4720b57cec5SDimitry Andric 4730b57cec5SDimitry Andric return Result; 4740b57cec5SDimitry Andric } 4750b57cec5SDimitry Andric 4760b57cec5SDimitry Andric unsigned WebAssemblyFastISel::signExtendToI32(unsigned Reg, const Value *V, 4770b57cec5SDimitry Andric MVT::SimpleValueType From) { 4780b57cec5SDimitry Andric if (Reg == 0) 4790b57cec5SDimitry Andric return 0; 4800b57cec5SDimitry Andric 4810b57cec5SDimitry Andric switch (From) { 4820b57cec5SDimitry Andric case MVT::i1: 4830b57cec5SDimitry Andric case MVT::i8: 4840b57cec5SDimitry Andric case MVT::i16: 4850b57cec5SDimitry Andric break; 4860b57cec5SDimitry Andric case MVT::i32: 4870b57cec5SDimitry Andric return copyValue(Reg); 4880b57cec5SDimitry Andric default: 4890b57cec5SDimitry Andric return 0; 4900b57cec5SDimitry Andric } 4910b57cec5SDimitry Andric 4920b57cec5SDimitry Andric unsigned Imm = createResultReg(&WebAssembly::I32RegClass); 4930b57cec5SDimitry Andric BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc, 4940b57cec5SDimitry Andric TII.get(WebAssembly::CONST_I32), Imm) 4950b57cec5SDimitry Andric .addImm(32 - MVT(From).getSizeInBits()); 4960b57cec5SDimitry Andric 4970b57cec5SDimitry Andric unsigned Left = createResultReg(&WebAssembly::I32RegClass); 4980b57cec5SDimitry Andric BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc, 4990b57cec5SDimitry Andric TII.get(WebAssembly::SHL_I32), Left) 5000b57cec5SDimitry Andric .addReg(Reg) 5010b57cec5SDimitry Andric .addReg(Imm); 5020b57cec5SDimitry Andric 5030b57cec5SDimitry Andric unsigned Right = createResultReg(&WebAssembly::I32RegClass); 5040b57cec5SDimitry Andric BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc, 5050b57cec5SDimitry Andric TII.get(WebAssembly::SHR_S_I32), Right) 5060b57cec5SDimitry Andric .addReg(Left) 5070b57cec5SDimitry Andric .addReg(Imm); 5080b57cec5SDimitry Andric 5090b57cec5SDimitry Andric return Right; 5100b57cec5SDimitry Andric } 5110b57cec5SDimitry Andric 5120b57cec5SDimitry Andric unsigned WebAssemblyFastISel::zeroExtend(unsigned Reg, const Value *V, 5130b57cec5SDimitry Andric MVT::SimpleValueType From, 5140b57cec5SDimitry Andric MVT::SimpleValueType To) { 5150b57cec5SDimitry Andric if (To == MVT::i64) { 5160b57cec5SDimitry Andric if (From == MVT::i64) 5170b57cec5SDimitry Andric return copyValue(Reg); 5180b57cec5SDimitry Andric 5190b57cec5SDimitry Andric Reg = zeroExtendToI32(Reg, V, From); 5200b57cec5SDimitry Andric 5210b57cec5SDimitry Andric unsigned Result = createResultReg(&WebAssembly::I64RegClass); 5220b57cec5SDimitry Andric BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc, 5230b57cec5SDimitry Andric TII.get(WebAssembly::I64_EXTEND_U_I32), Result) 5240b57cec5SDimitry Andric .addReg(Reg); 5250b57cec5SDimitry Andric return Result; 5260b57cec5SDimitry Andric } 5270b57cec5SDimitry Andric 5280b57cec5SDimitry Andric if (To == MVT::i32) 5290b57cec5SDimitry Andric return zeroExtendToI32(Reg, V, From); 5300b57cec5SDimitry Andric 5310b57cec5SDimitry Andric return 0; 5320b57cec5SDimitry Andric } 5330b57cec5SDimitry Andric 5340b57cec5SDimitry Andric unsigned WebAssemblyFastISel::signExtend(unsigned Reg, const Value *V, 5350b57cec5SDimitry Andric MVT::SimpleValueType From, 5360b57cec5SDimitry Andric MVT::SimpleValueType To) { 5370b57cec5SDimitry Andric if (To == MVT::i64) { 5380b57cec5SDimitry Andric if (From == MVT::i64) 5390b57cec5SDimitry Andric return copyValue(Reg); 5400b57cec5SDimitry Andric 5410b57cec5SDimitry Andric Reg = signExtendToI32(Reg, V, From); 5420b57cec5SDimitry Andric 5430b57cec5SDimitry Andric unsigned Result = createResultReg(&WebAssembly::I64RegClass); 5440b57cec5SDimitry Andric BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc, 5450b57cec5SDimitry Andric TII.get(WebAssembly::I64_EXTEND_S_I32), Result) 5460b57cec5SDimitry Andric .addReg(Reg); 5470b57cec5SDimitry Andric return Result; 5480b57cec5SDimitry Andric } 5490b57cec5SDimitry Andric 5500b57cec5SDimitry Andric if (To == MVT::i32) 5510b57cec5SDimitry Andric return signExtendToI32(Reg, V, From); 5520b57cec5SDimitry Andric 5530b57cec5SDimitry Andric return 0; 5540b57cec5SDimitry Andric } 5550b57cec5SDimitry Andric 5560b57cec5SDimitry Andric unsigned WebAssemblyFastISel::getRegForUnsignedValue(const Value *V) { 5570b57cec5SDimitry Andric MVT::SimpleValueType From = getSimpleType(V->getType()); 5580b57cec5SDimitry Andric MVT::SimpleValueType To = getLegalType(From); 5590b57cec5SDimitry Andric unsigned VReg = getRegForValue(V); 5600b57cec5SDimitry Andric if (VReg == 0) 5610b57cec5SDimitry Andric return 0; 5620b57cec5SDimitry Andric return zeroExtend(VReg, V, From, To); 5630b57cec5SDimitry Andric } 5640b57cec5SDimitry Andric 5650b57cec5SDimitry Andric unsigned WebAssemblyFastISel::getRegForSignedValue(const Value *V) { 5660b57cec5SDimitry Andric MVT::SimpleValueType From = getSimpleType(V->getType()); 5670b57cec5SDimitry Andric MVT::SimpleValueType To = getLegalType(From); 5680b57cec5SDimitry Andric unsigned VReg = getRegForValue(V); 5690b57cec5SDimitry Andric if (VReg == 0) 5700b57cec5SDimitry Andric return 0; 5710b57cec5SDimitry Andric return signExtend(VReg, V, From, To); 5720b57cec5SDimitry Andric } 5730b57cec5SDimitry Andric 5740b57cec5SDimitry Andric unsigned WebAssemblyFastISel::getRegForPromotedValue(const Value *V, 5750b57cec5SDimitry Andric bool IsSigned) { 5760b57cec5SDimitry Andric return IsSigned ? getRegForSignedValue(V) : getRegForUnsignedValue(V); 5770b57cec5SDimitry Andric } 5780b57cec5SDimitry Andric 5790b57cec5SDimitry Andric unsigned WebAssemblyFastISel::notValue(unsigned Reg) { 5800b57cec5SDimitry Andric assert(MRI.getRegClass(Reg) == &WebAssembly::I32RegClass); 5810b57cec5SDimitry Andric 5820b57cec5SDimitry Andric unsigned NotReg = createResultReg(&WebAssembly::I32RegClass); 5830b57cec5SDimitry Andric BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc, 5840b57cec5SDimitry Andric TII.get(WebAssembly::EQZ_I32), NotReg) 5850b57cec5SDimitry Andric .addReg(Reg); 5860b57cec5SDimitry Andric return NotReg; 5870b57cec5SDimitry Andric } 5880b57cec5SDimitry Andric 5890b57cec5SDimitry Andric unsigned WebAssemblyFastISel::copyValue(unsigned Reg) { 5900b57cec5SDimitry Andric unsigned ResultReg = createResultReg(MRI.getRegClass(Reg)); 5910b57cec5SDimitry Andric BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc, TII.get(WebAssembly::COPY), 5920b57cec5SDimitry Andric ResultReg) 5930b57cec5SDimitry Andric .addReg(Reg); 5940b57cec5SDimitry Andric return ResultReg; 5950b57cec5SDimitry Andric } 5960b57cec5SDimitry Andric 5970b57cec5SDimitry Andric unsigned WebAssemblyFastISel::fastMaterializeAlloca(const AllocaInst *AI) { 5980b57cec5SDimitry Andric DenseMap<const AllocaInst *, int>::iterator SI = 5990b57cec5SDimitry Andric FuncInfo.StaticAllocaMap.find(AI); 6000b57cec5SDimitry Andric 6010b57cec5SDimitry Andric if (SI != FuncInfo.StaticAllocaMap.end()) { 6020b57cec5SDimitry Andric unsigned ResultReg = 6030b57cec5SDimitry Andric createResultReg(Subtarget->hasAddr64() ? &WebAssembly::I64RegClass 6040b57cec5SDimitry Andric : &WebAssembly::I32RegClass); 6050b57cec5SDimitry Andric unsigned Opc = 6060b57cec5SDimitry Andric Subtarget->hasAddr64() ? WebAssembly::COPY_I64 : WebAssembly::COPY_I32; 6070b57cec5SDimitry Andric BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc, TII.get(Opc), ResultReg) 6080b57cec5SDimitry Andric .addFrameIndex(SI->second); 6090b57cec5SDimitry Andric return ResultReg; 6100b57cec5SDimitry Andric } 6110b57cec5SDimitry Andric 6120b57cec5SDimitry Andric return 0; 6130b57cec5SDimitry Andric } 6140b57cec5SDimitry Andric 6150b57cec5SDimitry Andric unsigned WebAssemblyFastISel::fastMaterializeConstant(const Constant *C) { 6160b57cec5SDimitry Andric if (const GlobalValue *GV = dyn_cast<GlobalValue>(C)) { 6170b57cec5SDimitry Andric if (TLI.isPositionIndependent()) 6180b57cec5SDimitry Andric return 0; 6190b57cec5SDimitry Andric if (GV->isThreadLocal()) 6200b57cec5SDimitry Andric return 0; 6210b57cec5SDimitry Andric unsigned ResultReg = 6220b57cec5SDimitry Andric createResultReg(Subtarget->hasAddr64() ? &WebAssembly::I64RegClass 6230b57cec5SDimitry Andric : &WebAssembly::I32RegClass); 6240b57cec5SDimitry Andric unsigned Opc = Subtarget->hasAddr64() ? WebAssembly::CONST_I64 6250b57cec5SDimitry Andric : WebAssembly::CONST_I32; 6260b57cec5SDimitry Andric BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc, TII.get(Opc), ResultReg) 6270b57cec5SDimitry Andric .addGlobalAddress(GV); 6280b57cec5SDimitry Andric return ResultReg; 6290b57cec5SDimitry Andric } 6300b57cec5SDimitry Andric 6310b57cec5SDimitry Andric // Let target-independent code handle it. 6320b57cec5SDimitry Andric return 0; 6330b57cec5SDimitry Andric } 6340b57cec5SDimitry Andric 6350b57cec5SDimitry Andric bool WebAssemblyFastISel::fastLowerArguments() { 6360b57cec5SDimitry Andric if (!FuncInfo.CanLowerReturn) 6370b57cec5SDimitry Andric return false; 6380b57cec5SDimitry Andric 6390b57cec5SDimitry Andric const Function *F = FuncInfo.Fn; 6400b57cec5SDimitry Andric if (F->isVarArg()) 6410b57cec5SDimitry Andric return false; 6420b57cec5SDimitry Andric 643*5ffd83dbSDimitry Andric if (FuncInfo.Fn->getCallingConv() == CallingConv::Swift) 644*5ffd83dbSDimitry Andric return false; 645*5ffd83dbSDimitry Andric 6460b57cec5SDimitry Andric unsigned I = 0; 6470b57cec5SDimitry Andric for (auto const &Arg : F->args()) { 6480b57cec5SDimitry Andric const AttributeList &Attrs = F->getAttributes(); 6490b57cec5SDimitry Andric if (Attrs.hasParamAttribute(I, Attribute::ByVal) || 6500b57cec5SDimitry Andric Attrs.hasParamAttribute(I, Attribute::SwiftSelf) || 6510b57cec5SDimitry Andric Attrs.hasParamAttribute(I, Attribute::SwiftError) || 6520b57cec5SDimitry Andric Attrs.hasParamAttribute(I, Attribute::InAlloca) || 6530b57cec5SDimitry Andric Attrs.hasParamAttribute(I, Attribute::Nest)) 6540b57cec5SDimitry Andric return false; 6550b57cec5SDimitry Andric 6560b57cec5SDimitry Andric Type *ArgTy = Arg.getType(); 6570b57cec5SDimitry Andric if (ArgTy->isStructTy() || ArgTy->isArrayTy()) 6580b57cec5SDimitry Andric return false; 6590b57cec5SDimitry Andric if (!Subtarget->hasSIMD128() && ArgTy->isVectorTy()) 6600b57cec5SDimitry Andric return false; 6610b57cec5SDimitry Andric 6620b57cec5SDimitry Andric unsigned Opc; 6630b57cec5SDimitry Andric const TargetRegisterClass *RC; 6640b57cec5SDimitry Andric switch (getSimpleType(ArgTy)) { 6650b57cec5SDimitry Andric case MVT::i1: 6660b57cec5SDimitry Andric case MVT::i8: 6670b57cec5SDimitry Andric case MVT::i16: 6680b57cec5SDimitry Andric case MVT::i32: 6690b57cec5SDimitry Andric Opc = WebAssembly::ARGUMENT_i32; 6700b57cec5SDimitry Andric RC = &WebAssembly::I32RegClass; 6710b57cec5SDimitry Andric break; 6720b57cec5SDimitry Andric case MVT::i64: 6730b57cec5SDimitry Andric Opc = WebAssembly::ARGUMENT_i64; 6740b57cec5SDimitry Andric RC = &WebAssembly::I64RegClass; 6750b57cec5SDimitry Andric break; 6760b57cec5SDimitry Andric case MVT::f32: 6770b57cec5SDimitry Andric Opc = WebAssembly::ARGUMENT_f32; 6780b57cec5SDimitry Andric RC = &WebAssembly::F32RegClass; 6790b57cec5SDimitry Andric break; 6800b57cec5SDimitry Andric case MVT::f64: 6810b57cec5SDimitry Andric Opc = WebAssembly::ARGUMENT_f64; 6820b57cec5SDimitry Andric RC = &WebAssembly::F64RegClass; 6830b57cec5SDimitry Andric break; 6840b57cec5SDimitry Andric case MVT::v16i8: 6850b57cec5SDimitry Andric Opc = WebAssembly::ARGUMENT_v16i8; 6860b57cec5SDimitry Andric RC = &WebAssembly::V128RegClass; 6870b57cec5SDimitry Andric break; 6880b57cec5SDimitry Andric case MVT::v8i16: 6890b57cec5SDimitry Andric Opc = WebAssembly::ARGUMENT_v8i16; 6900b57cec5SDimitry Andric RC = &WebAssembly::V128RegClass; 6910b57cec5SDimitry Andric break; 6920b57cec5SDimitry Andric case MVT::v4i32: 6930b57cec5SDimitry Andric Opc = WebAssembly::ARGUMENT_v4i32; 6940b57cec5SDimitry Andric RC = &WebAssembly::V128RegClass; 6950b57cec5SDimitry Andric break; 6960b57cec5SDimitry Andric case MVT::v2i64: 6970b57cec5SDimitry Andric Opc = WebAssembly::ARGUMENT_v2i64; 6980b57cec5SDimitry Andric RC = &WebAssembly::V128RegClass; 6990b57cec5SDimitry Andric break; 7000b57cec5SDimitry Andric case MVT::v4f32: 7010b57cec5SDimitry Andric Opc = WebAssembly::ARGUMENT_v4f32; 7020b57cec5SDimitry Andric RC = &WebAssembly::V128RegClass; 7030b57cec5SDimitry Andric break; 7040b57cec5SDimitry Andric case MVT::v2f64: 7050b57cec5SDimitry Andric Opc = WebAssembly::ARGUMENT_v2f64; 7060b57cec5SDimitry Andric RC = &WebAssembly::V128RegClass; 7070b57cec5SDimitry Andric break; 7080b57cec5SDimitry Andric case MVT::exnref: 7090b57cec5SDimitry Andric Opc = WebAssembly::ARGUMENT_exnref; 7100b57cec5SDimitry Andric RC = &WebAssembly::EXNREFRegClass; 7110b57cec5SDimitry Andric break; 7120b57cec5SDimitry Andric default: 7130b57cec5SDimitry Andric return false; 7140b57cec5SDimitry Andric } 7150b57cec5SDimitry Andric unsigned ResultReg = createResultReg(RC); 7160b57cec5SDimitry Andric BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc, TII.get(Opc), ResultReg) 7170b57cec5SDimitry Andric .addImm(I); 7180b57cec5SDimitry Andric updateValueMap(&Arg, ResultReg); 7190b57cec5SDimitry Andric 7200b57cec5SDimitry Andric ++I; 7210b57cec5SDimitry Andric } 7220b57cec5SDimitry Andric 7230b57cec5SDimitry Andric MRI.addLiveIn(WebAssembly::ARGUMENTS); 7240b57cec5SDimitry Andric 7250b57cec5SDimitry Andric auto *MFI = MF->getInfo<WebAssemblyFunctionInfo>(); 7260b57cec5SDimitry Andric for (auto const &Arg : F->args()) { 7270b57cec5SDimitry Andric MVT::SimpleValueType ArgTy = getLegalType(getSimpleType(Arg.getType())); 7280b57cec5SDimitry Andric if (ArgTy == MVT::INVALID_SIMPLE_VALUE_TYPE) { 7290b57cec5SDimitry Andric MFI->clearParamsAndResults(); 7300b57cec5SDimitry Andric return false; 7310b57cec5SDimitry Andric } 7320b57cec5SDimitry Andric MFI->addParam(ArgTy); 7330b57cec5SDimitry Andric } 7340b57cec5SDimitry Andric 7350b57cec5SDimitry Andric if (!F->getReturnType()->isVoidTy()) { 7360b57cec5SDimitry Andric MVT::SimpleValueType RetTy = 7370b57cec5SDimitry Andric getLegalType(getSimpleType(F->getReturnType())); 7380b57cec5SDimitry Andric if (RetTy == MVT::INVALID_SIMPLE_VALUE_TYPE) { 7390b57cec5SDimitry Andric MFI->clearParamsAndResults(); 7400b57cec5SDimitry Andric return false; 7410b57cec5SDimitry Andric } 7420b57cec5SDimitry Andric MFI->addResult(RetTy); 7430b57cec5SDimitry Andric } 7440b57cec5SDimitry Andric 7450b57cec5SDimitry Andric return true; 7460b57cec5SDimitry Andric } 7470b57cec5SDimitry Andric 7480b57cec5SDimitry Andric bool WebAssemblyFastISel::selectCall(const Instruction *I) { 7490b57cec5SDimitry Andric const auto *Call = cast<CallInst>(I); 7500b57cec5SDimitry Andric 7510b57cec5SDimitry Andric // TODO: Support tail calls in FastISel 7520b57cec5SDimitry Andric if (Call->isMustTailCall() || Call->isInlineAsm() || 7530b57cec5SDimitry Andric Call->getFunctionType()->isVarArg()) 7540b57cec5SDimitry Andric return false; 7550b57cec5SDimitry Andric 7560b57cec5SDimitry Andric Function *Func = Call->getCalledFunction(); 7570b57cec5SDimitry Andric if (Func && Func->isIntrinsic()) 7580b57cec5SDimitry Andric return false; 7590b57cec5SDimitry Andric 760*5ffd83dbSDimitry Andric if (Call->getCallingConv() == CallingConv::Swift) 761*5ffd83dbSDimitry Andric return false; 762*5ffd83dbSDimitry Andric 7630b57cec5SDimitry Andric bool IsDirect = Func != nullptr; 764*5ffd83dbSDimitry Andric if (!IsDirect && isa<ConstantExpr>(Call->getCalledOperand())) 7650b57cec5SDimitry Andric return false; 7660b57cec5SDimitry Andric 7670b57cec5SDimitry Andric FunctionType *FuncTy = Call->getFunctionType(); 768*5ffd83dbSDimitry Andric unsigned Opc = IsDirect ? WebAssembly::CALL : WebAssembly::CALL_INDIRECT; 7690b57cec5SDimitry Andric bool IsVoid = FuncTy->getReturnType()->isVoidTy(); 7700b57cec5SDimitry Andric unsigned ResultReg; 771*5ffd83dbSDimitry Andric if (!IsVoid) { 7720b57cec5SDimitry Andric if (!Subtarget->hasSIMD128() && Call->getType()->isVectorTy()) 7730b57cec5SDimitry Andric return false; 7740b57cec5SDimitry Andric 7750b57cec5SDimitry Andric MVT::SimpleValueType RetTy = getSimpleType(Call->getType()); 7760b57cec5SDimitry Andric switch (RetTy) { 7770b57cec5SDimitry Andric case MVT::i1: 7780b57cec5SDimitry Andric case MVT::i8: 7790b57cec5SDimitry Andric case MVT::i16: 7800b57cec5SDimitry Andric case MVT::i32: 7810b57cec5SDimitry Andric ResultReg = createResultReg(&WebAssembly::I32RegClass); 7820b57cec5SDimitry Andric break; 7830b57cec5SDimitry Andric case MVT::i64: 7840b57cec5SDimitry Andric ResultReg = createResultReg(&WebAssembly::I64RegClass); 7850b57cec5SDimitry Andric break; 7860b57cec5SDimitry Andric case MVT::f32: 7870b57cec5SDimitry Andric ResultReg = createResultReg(&WebAssembly::F32RegClass); 7880b57cec5SDimitry Andric break; 7890b57cec5SDimitry Andric case MVT::f64: 7900b57cec5SDimitry Andric ResultReg = createResultReg(&WebAssembly::F64RegClass); 7910b57cec5SDimitry Andric break; 7920b57cec5SDimitry Andric case MVT::v16i8: 7930b57cec5SDimitry Andric ResultReg = createResultReg(&WebAssembly::V128RegClass); 7940b57cec5SDimitry Andric break; 7950b57cec5SDimitry Andric case MVT::v8i16: 7960b57cec5SDimitry Andric ResultReg = createResultReg(&WebAssembly::V128RegClass); 7970b57cec5SDimitry Andric break; 7980b57cec5SDimitry Andric case MVT::v4i32: 7990b57cec5SDimitry Andric ResultReg = createResultReg(&WebAssembly::V128RegClass); 8000b57cec5SDimitry Andric break; 8010b57cec5SDimitry Andric case MVT::v2i64: 8020b57cec5SDimitry Andric ResultReg = createResultReg(&WebAssembly::V128RegClass); 8030b57cec5SDimitry Andric break; 8040b57cec5SDimitry Andric case MVT::v4f32: 8050b57cec5SDimitry Andric ResultReg = createResultReg(&WebAssembly::V128RegClass); 8060b57cec5SDimitry Andric break; 8070b57cec5SDimitry Andric case MVT::v2f64: 8080b57cec5SDimitry Andric ResultReg = createResultReg(&WebAssembly::V128RegClass); 8090b57cec5SDimitry Andric break; 8100b57cec5SDimitry Andric case MVT::exnref: 8110b57cec5SDimitry Andric ResultReg = createResultReg(&WebAssembly::EXNREFRegClass); 8120b57cec5SDimitry Andric break; 8130b57cec5SDimitry Andric default: 8140b57cec5SDimitry Andric return false; 8150b57cec5SDimitry Andric } 8160b57cec5SDimitry Andric } 8170b57cec5SDimitry Andric 8180b57cec5SDimitry Andric SmallVector<unsigned, 8> Args; 8190b57cec5SDimitry Andric for (unsigned I = 0, E = Call->getNumArgOperands(); I < E; ++I) { 8200b57cec5SDimitry Andric Value *V = Call->getArgOperand(I); 8210b57cec5SDimitry Andric MVT::SimpleValueType ArgTy = getSimpleType(V->getType()); 8220b57cec5SDimitry Andric if (ArgTy == MVT::INVALID_SIMPLE_VALUE_TYPE) 8230b57cec5SDimitry Andric return false; 8240b57cec5SDimitry Andric 8250b57cec5SDimitry Andric const AttributeList &Attrs = Call->getAttributes(); 8260b57cec5SDimitry Andric if (Attrs.hasParamAttribute(I, Attribute::ByVal) || 8270b57cec5SDimitry Andric Attrs.hasParamAttribute(I, Attribute::SwiftSelf) || 8280b57cec5SDimitry Andric Attrs.hasParamAttribute(I, Attribute::SwiftError) || 8290b57cec5SDimitry Andric Attrs.hasParamAttribute(I, Attribute::InAlloca) || 8300b57cec5SDimitry Andric Attrs.hasParamAttribute(I, Attribute::Nest)) 8310b57cec5SDimitry Andric return false; 8320b57cec5SDimitry Andric 8330b57cec5SDimitry Andric unsigned Reg; 8340b57cec5SDimitry Andric 8350b57cec5SDimitry Andric if (Attrs.hasParamAttribute(I, Attribute::SExt)) 8360b57cec5SDimitry Andric Reg = getRegForSignedValue(V); 8370b57cec5SDimitry Andric else if (Attrs.hasParamAttribute(I, Attribute::ZExt)) 8380b57cec5SDimitry Andric Reg = getRegForUnsignedValue(V); 8390b57cec5SDimitry Andric else 8400b57cec5SDimitry Andric Reg = getRegForValue(V); 8410b57cec5SDimitry Andric 8420b57cec5SDimitry Andric if (Reg == 0) 8430b57cec5SDimitry Andric return false; 8440b57cec5SDimitry Andric 8450b57cec5SDimitry Andric Args.push_back(Reg); 8460b57cec5SDimitry Andric } 8470b57cec5SDimitry Andric 8480b57cec5SDimitry Andric unsigned CalleeReg = 0; 8490b57cec5SDimitry Andric if (!IsDirect) { 850*5ffd83dbSDimitry Andric CalleeReg = getRegForValue(Call->getCalledOperand()); 8510b57cec5SDimitry Andric if (!CalleeReg) 8520b57cec5SDimitry Andric return false; 8530b57cec5SDimitry Andric } 8540b57cec5SDimitry Andric 8550b57cec5SDimitry Andric auto MIB = BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc, TII.get(Opc)); 8560b57cec5SDimitry Andric 8570b57cec5SDimitry Andric if (!IsVoid) 8580b57cec5SDimitry Andric MIB.addReg(ResultReg, RegState::Define); 8590b57cec5SDimitry Andric 860*5ffd83dbSDimitry Andric if (IsDirect) { 8610b57cec5SDimitry Andric MIB.addGlobalAddress(Func); 862*5ffd83dbSDimitry Andric } else { 863*5ffd83dbSDimitry Andric // Add placeholders for the type index and immediate flags 864*5ffd83dbSDimitry Andric MIB.addImm(0); 865*5ffd83dbSDimitry Andric MIB.addImm(0); 866*5ffd83dbSDimitry Andric } 8670b57cec5SDimitry Andric 8680b57cec5SDimitry Andric for (unsigned ArgReg : Args) 8690b57cec5SDimitry Andric MIB.addReg(ArgReg); 8700b57cec5SDimitry Andric 871*5ffd83dbSDimitry Andric if (!IsDirect) 872*5ffd83dbSDimitry Andric MIB.addReg(CalleeReg); 873*5ffd83dbSDimitry Andric 8740b57cec5SDimitry Andric if (!IsVoid) 8750b57cec5SDimitry Andric updateValueMap(Call, ResultReg); 8760b57cec5SDimitry Andric return true; 8770b57cec5SDimitry Andric } 8780b57cec5SDimitry Andric 8790b57cec5SDimitry Andric bool WebAssemblyFastISel::selectSelect(const Instruction *I) { 8800b57cec5SDimitry Andric const auto *Select = cast<SelectInst>(I); 8810b57cec5SDimitry Andric 8820b57cec5SDimitry Andric bool Not; 8830b57cec5SDimitry Andric unsigned CondReg = getRegForI1Value(Select->getCondition(), Not); 8840b57cec5SDimitry Andric if (CondReg == 0) 8850b57cec5SDimitry Andric return false; 8860b57cec5SDimitry Andric 8870b57cec5SDimitry Andric unsigned TrueReg = getRegForValue(Select->getTrueValue()); 8880b57cec5SDimitry Andric if (TrueReg == 0) 8890b57cec5SDimitry Andric return false; 8900b57cec5SDimitry Andric 8910b57cec5SDimitry Andric unsigned FalseReg = getRegForValue(Select->getFalseValue()); 8920b57cec5SDimitry Andric if (FalseReg == 0) 8930b57cec5SDimitry Andric return false; 8940b57cec5SDimitry Andric 8950b57cec5SDimitry Andric if (Not) 8960b57cec5SDimitry Andric std::swap(TrueReg, FalseReg); 8970b57cec5SDimitry Andric 8980b57cec5SDimitry Andric unsigned Opc; 8990b57cec5SDimitry Andric const TargetRegisterClass *RC; 9000b57cec5SDimitry Andric switch (getSimpleType(Select->getType())) { 9010b57cec5SDimitry Andric case MVT::i1: 9020b57cec5SDimitry Andric case MVT::i8: 9030b57cec5SDimitry Andric case MVT::i16: 9040b57cec5SDimitry Andric case MVT::i32: 9050b57cec5SDimitry Andric Opc = WebAssembly::SELECT_I32; 9060b57cec5SDimitry Andric RC = &WebAssembly::I32RegClass; 9070b57cec5SDimitry Andric break; 9080b57cec5SDimitry Andric case MVT::i64: 9090b57cec5SDimitry Andric Opc = WebAssembly::SELECT_I64; 9100b57cec5SDimitry Andric RC = &WebAssembly::I64RegClass; 9110b57cec5SDimitry Andric break; 9120b57cec5SDimitry Andric case MVT::f32: 9130b57cec5SDimitry Andric Opc = WebAssembly::SELECT_F32; 9140b57cec5SDimitry Andric RC = &WebAssembly::F32RegClass; 9150b57cec5SDimitry Andric break; 9160b57cec5SDimitry Andric case MVT::f64: 9170b57cec5SDimitry Andric Opc = WebAssembly::SELECT_F64; 9180b57cec5SDimitry Andric RC = &WebAssembly::F64RegClass; 9190b57cec5SDimitry Andric break; 9200b57cec5SDimitry Andric case MVT::exnref: 9210b57cec5SDimitry Andric Opc = WebAssembly::SELECT_EXNREF; 9220b57cec5SDimitry Andric RC = &WebAssembly::EXNREFRegClass; 9230b57cec5SDimitry Andric break; 9240b57cec5SDimitry Andric default: 9250b57cec5SDimitry Andric return false; 9260b57cec5SDimitry Andric } 9270b57cec5SDimitry Andric 9280b57cec5SDimitry Andric unsigned ResultReg = createResultReg(RC); 9290b57cec5SDimitry Andric BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc, TII.get(Opc), ResultReg) 9300b57cec5SDimitry Andric .addReg(TrueReg) 9310b57cec5SDimitry Andric .addReg(FalseReg) 9320b57cec5SDimitry Andric .addReg(CondReg); 9330b57cec5SDimitry Andric 9340b57cec5SDimitry Andric updateValueMap(Select, ResultReg); 9350b57cec5SDimitry Andric return true; 9360b57cec5SDimitry Andric } 9370b57cec5SDimitry Andric 9380b57cec5SDimitry Andric bool WebAssemblyFastISel::selectTrunc(const Instruction *I) { 9390b57cec5SDimitry Andric const auto *Trunc = cast<TruncInst>(I); 9400b57cec5SDimitry Andric 9410b57cec5SDimitry Andric unsigned Reg = getRegForValue(Trunc->getOperand(0)); 9420b57cec5SDimitry Andric if (Reg == 0) 9430b57cec5SDimitry Andric return false; 9440b57cec5SDimitry Andric 9450b57cec5SDimitry Andric if (Trunc->getOperand(0)->getType()->isIntegerTy(64)) { 9460b57cec5SDimitry Andric unsigned Result = createResultReg(&WebAssembly::I32RegClass); 9470b57cec5SDimitry Andric BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc, 9480b57cec5SDimitry Andric TII.get(WebAssembly::I32_WRAP_I64), Result) 9490b57cec5SDimitry Andric .addReg(Reg); 9500b57cec5SDimitry Andric Reg = Result; 9510b57cec5SDimitry Andric } 9520b57cec5SDimitry Andric 9530b57cec5SDimitry Andric updateValueMap(Trunc, Reg); 9540b57cec5SDimitry Andric return true; 9550b57cec5SDimitry Andric } 9560b57cec5SDimitry Andric 9570b57cec5SDimitry Andric bool WebAssemblyFastISel::selectZExt(const Instruction *I) { 9580b57cec5SDimitry Andric const auto *ZExt = cast<ZExtInst>(I); 9590b57cec5SDimitry Andric 9600b57cec5SDimitry Andric const Value *Op = ZExt->getOperand(0); 9610b57cec5SDimitry Andric MVT::SimpleValueType From = getSimpleType(Op->getType()); 9620b57cec5SDimitry Andric MVT::SimpleValueType To = getLegalType(getSimpleType(ZExt->getType())); 9630b57cec5SDimitry Andric unsigned In = getRegForValue(Op); 9640b57cec5SDimitry Andric if (In == 0) 9650b57cec5SDimitry Andric return false; 9660b57cec5SDimitry Andric unsigned Reg = zeroExtend(In, Op, From, To); 9670b57cec5SDimitry Andric if (Reg == 0) 9680b57cec5SDimitry Andric return false; 9690b57cec5SDimitry Andric 9700b57cec5SDimitry Andric updateValueMap(ZExt, Reg); 9710b57cec5SDimitry Andric return true; 9720b57cec5SDimitry Andric } 9730b57cec5SDimitry Andric 9740b57cec5SDimitry Andric bool WebAssemblyFastISel::selectSExt(const Instruction *I) { 9750b57cec5SDimitry Andric const auto *SExt = cast<SExtInst>(I); 9760b57cec5SDimitry Andric 9770b57cec5SDimitry Andric const Value *Op = SExt->getOperand(0); 9780b57cec5SDimitry Andric MVT::SimpleValueType From = getSimpleType(Op->getType()); 9790b57cec5SDimitry Andric MVT::SimpleValueType To = getLegalType(getSimpleType(SExt->getType())); 9800b57cec5SDimitry Andric unsigned In = getRegForValue(Op); 9810b57cec5SDimitry Andric if (In == 0) 9820b57cec5SDimitry Andric return false; 9830b57cec5SDimitry Andric unsigned Reg = signExtend(In, Op, From, To); 9840b57cec5SDimitry Andric if (Reg == 0) 9850b57cec5SDimitry Andric return false; 9860b57cec5SDimitry Andric 9870b57cec5SDimitry Andric updateValueMap(SExt, Reg); 9880b57cec5SDimitry Andric return true; 9890b57cec5SDimitry Andric } 9900b57cec5SDimitry Andric 9910b57cec5SDimitry Andric bool WebAssemblyFastISel::selectICmp(const Instruction *I) { 9920b57cec5SDimitry Andric const auto *ICmp = cast<ICmpInst>(I); 9930b57cec5SDimitry Andric 9940b57cec5SDimitry Andric bool I32 = getSimpleType(ICmp->getOperand(0)->getType()) != MVT::i64; 9950b57cec5SDimitry Andric unsigned Opc; 9960b57cec5SDimitry Andric bool IsSigned = false; 9970b57cec5SDimitry Andric switch (ICmp->getPredicate()) { 9980b57cec5SDimitry Andric case ICmpInst::ICMP_EQ: 9990b57cec5SDimitry Andric Opc = I32 ? WebAssembly::EQ_I32 : WebAssembly::EQ_I64; 10000b57cec5SDimitry Andric break; 10010b57cec5SDimitry Andric case ICmpInst::ICMP_NE: 10020b57cec5SDimitry Andric Opc = I32 ? WebAssembly::NE_I32 : WebAssembly::NE_I64; 10030b57cec5SDimitry Andric break; 10040b57cec5SDimitry Andric case ICmpInst::ICMP_UGT: 10050b57cec5SDimitry Andric Opc = I32 ? WebAssembly::GT_U_I32 : WebAssembly::GT_U_I64; 10060b57cec5SDimitry Andric break; 10070b57cec5SDimitry Andric case ICmpInst::ICMP_UGE: 10080b57cec5SDimitry Andric Opc = I32 ? WebAssembly::GE_U_I32 : WebAssembly::GE_U_I64; 10090b57cec5SDimitry Andric break; 10100b57cec5SDimitry Andric case ICmpInst::ICMP_ULT: 10110b57cec5SDimitry Andric Opc = I32 ? WebAssembly::LT_U_I32 : WebAssembly::LT_U_I64; 10120b57cec5SDimitry Andric break; 10130b57cec5SDimitry Andric case ICmpInst::ICMP_ULE: 10140b57cec5SDimitry Andric Opc = I32 ? WebAssembly::LE_U_I32 : WebAssembly::LE_U_I64; 10150b57cec5SDimitry Andric break; 10160b57cec5SDimitry Andric case ICmpInst::ICMP_SGT: 10170b57cec5SDimitry Andric Opc = I32 ? WebAssembly::GT_S_I32 : WebAssembly::GT_S_I64; 10180b57cec5SDimitry Andric IsSigned = true; 10190b57cec5SDimitry Andric break; 10200b57cec5SDimitry Andric case ICmpInst::ICMP_SGE: 10210b57cec5SDimitry Andric Opc = I32 ? WebAssembly::GE_S_I32 : WebAssembly::GE_S_I64; 10220b57cec5SDimitry Andric IsSigned = true; 10230b57cec5SDimitry Andric break; 10240b57cec5SDimitry Andric case ICmpInst::ICMP_SLT: 10250b57cec5SDimitry Andric Opc = I32 ? WebAssembly::LT_S_I32 : WebAssembly::LT_S_I64; 10260b57cec5SDimitry Andric IsSigned = true; 10270b57cec5SDimitry Andric break; 10280b57cec5SDimitry Andric case ICmpInst::ICMP_SLE: 10290b57cec5SDimitry Andric Opc = I32 ? WebAssembly::LE_S_I32 : WebAssembly::LE_S_I64; 10300b57cec5SDimitry Andric IsSigned = true; 10310b57cec5SDimitry Andric break; 10320b57cec5SDimitry Andric default: 10330b57cec5SDimitry Andric return false; 10340b57cec5SDimitry Andric } 10350b57cec5SDimitry Andric 10360b57cec5SDimitry Andric unsigned LHS = getRegForPromotedValue(ICmp->getOperand(0), IsSigned); 10370b57cec5SDimitry Andric if (LHS == 0) 10380b57cec5SDimitry Andric return false; 10390b57cec5SDimitry Andric 10400b57cec5SDimitry Andric unsigned RHS = getRegForPromotedValue(ICmp->getOperand(1), IsSigned); 10410b57cec5SDimitry Andric if (RHS == 0) 10420b57cec5SDimitry Andric return false; 10430b57cec5SDimitry Andric 10440b57cec5SDimitry Andric unsigned ResultReg = createResultReg(&WebAssembly::I32RegClass); 10450b57cec5SDimitry Andric BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc, TII.get(Opc), ResultReg) 10460b57cec5SDimitry Andric .addReg(LHS) 10470b57cec5SDimitry Andric .addReg(RHS); 10480b57cec5SDimitry Andric updateValueMap(ICmp, ResultReg); 10490b57cec5SDimitry Andric return true; 10500b57cec5SDimitry Andric } 10510b57cec5SDimitry Andric 10520b57cec5SDimitry Andric bool WebAssemblyFastISel::selectFCmp(const Instruction *I) { 10530b57cec5SDimitry Andric const auto *FCmp = cast<FCmpInst>(I); 10540b57cec5SDimitry Andric 10550b57cec5SDimitry Andric unsigned LHS = getRegForValue(FCmp->getOperand(0)); 10560b57cec5SDimitry Andric if (LHS == 0) 10570b57cec5SDimitry Andric return false; 10580b57cec5SDimitry Andric 10590b57cec5SDimitry Andric unsigned RHS = getRegForValue(FCmp->getOperand(1)); 10600b57cec5SDimitry Andric if (RHS == 0) 10610b57cec5SDimitry Andric return false; 10620b57cec5SDimitry Andric 10630b57cec5SDimitry Andric bool F32 = getSimpleType(FCmp->getOperand(0)->getType()) != MVT::f64; 10640b57cec5SDimitry Andric unsigned Opc; 10650b57cec5SDimitry Andric bool Not = false; 10660b57cec5SDimitry Andric switch (FCmp->getPredicate()) { 10670b57cec5SDimitry Andric case FCmpInst::FCMP_OEQ: 10680b57cec5SDimitry Andric Opc = F32 ? WebAssembly::EQ_F32 : WebAssembly::EQ_F64; 10690b57cec5SDimitry Andric break; 10700b57cec5SDimitry Andric case FCmpInst::FCMP_UNE: 10710b57cec5SDimitry Andric Opc = F32 ? WebAssembly::NE_F32 : WebAssembly::NE_F64; 10720b57cec5SDimitry Andric break; 10730b57cec5SDimitry Andric case FCmpInst::FCMP_OGT: 10740b57cec5SDimitry Andric Opc = F32 ? WebAssembly::GT_F32 : WebAssembly::GT_F64; 10750b57cec5SDimitry Andric break; 10760b57cec5SDimitry Andric case FCmpInst::FCMP_OGE: 10770b57cec5SDimitry Andric Opc = F32 ? WebAssembly::GE_F32 : WebAssembly::GE_F64; 10780b57cec5SDimitry Andric break; 10790b57cec5SDimitry Andric case FCmpInst::FCMP_OLT: 10800b57cec5SDimitry Andric Opc = F32 ? WebAssembly::LT_F32 : WebAssembly::LT_F64; 10810b57cec5SDimitry Andric break; 10820b57cec5SDimitry Andric case FCmpInst::FCMP_OLE: 10830b57cec5SDimitry Andric Opc = F32 ? WebAssembly::LE_F32 : WebAssembly::LE_F64; 10840b57cec5SDimitry Andric break; 10850b57cec5SDimitry Andric case FCmpInst::FCMP_UGT: 10860b57cec5SDimitry Andric Opc = F32 ? WebAssembly::LE_F32 : WebAssembly::LE_F64; 10870b57cec5SDimitry Andric Not = true; 10880b57cec5SDimitry Andric break; 10890b57cec5SDimitry Andric case FCmpInst::FCMP_UGE: 10900b57cec5SDimitry Andric Opc = F32 ? WebAssembly::LT_F32 : WebAssembly::LT_F64; 10910b57cec5SDimitry Andric Not = true; 10920b57cec5SDimitry Andric break; 10930b57cec5SDimitry Andric case FCmpInst::FCMP_ULT: 10940b57cec5SDimitry Andric Opc = F32 ? WebAssembly::GE_F32 : WebAssembly::GE_F64; 10950b57cec5SDimitry Andric Not = true; 10960b57cec5SDimitry Andric break; 10970b57cec5SDimitry Andric case FCmpInst::FCMP_ULE: 10980b57cec5SDimitry Andric Opc = F32 ? WebAssembly::GT_F32 : WebAssembly::GT_F64; 10990b57cec5SDimitry Andric Not = true; 11000b57cec5SDimitry Andric break; 11010b57cec5SDimitry Andric default: 11020b57cec5SDimitry Andric return false; 11030b57cec5SDimitry Andric } 11040b57cec5SDimitry Andric 11050b57cec5SDimitry Andric unsigned ResultReg = createResultReg(&WebAssembly::I32RegClass); 11060b57cec5SDimitry Andric BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc, TII.get(Opc), ResultReg) 11070b57cec5SDimitry Andric .addReg(LHS) 11080b57cec5SDimitry Andric .addReg(RHS); 11090b57cec5SDimitry Andric 11100b57cec5SDimitry Andric if (Not) 11110b57cec5SDimitry Andric ResultReg = notValue(ResultReg); 11120b57cec5SDimitry Andric 11130b57cec5SDimitry Andric updateValueMap(FCmp, ResultReg); 11140b57cec5SDimitry Andric return true; 11150b57cec5SDimitry Andric } 11160b57cec5SDimitry Andric 11170b57cec5SDimitry Andric bool WebAssemblyFastISel::selectBitCast(const Instruction *I) { 11180b57cec5SDimitry Andric // Target-independent code can handle this, except it doesn't set the dead 11190b57cec5SDimitry Andric // flag on the ARGUMENTS clobber, so we have to do that manually in order 11200b57cec5SDimitry Andric // to satisfy code that expects this of isBitcast() instructions. 11210b57cec5SDimitry Andric EVT VT = TLI.getValueType(DL, I->getOperand(0)->getType()); 11220b57cec5SDimitry Andric EVT RetVT = TLI.getValueType(DL, I->getType()); 11230b57cec5SDimitry Andric if (!VT.isSimple() || !RetVT.isSimple()) 11240b57cec5SDimitry Andric return false; 11250b57cec5SDimitry Andric 11260b57cec5SDimitry Andric unsigned In = getRegForValue(I->getOperand(0)); 11270b57cec5SDimitry Andric if (In == 0) 11280b57cec5SDimitry Andric return false; 11290b57cec5SDimitry Andric 11300b57cec5SDimitry Andric if (VT == RetVT) { 11310b57cec5SDimitry Andric // No-op bitcast. 11320b57cec5SDimitry Andric updateValueMap(I, In); 11330b57cec5SDimitry Andric return true; 11340b57cec5SDimitry Andric } 11350b57cec5SDimitry Andric 11368bcb0991SDimitry Andric Register Reg = fastEmit_ISD_BITCAST_r(VT.getSimpleVT(), RetVT.getSimpleVT(), 11370b57cec5SDimitry Andric In, I->getOperand(0)->hasOneUse()); 11380b57cec5SDimitry Andric if (!Reg) 11390b57cec5SDimitry Andric return false; 11400b57cec5SDimitry Andric MachineBasicBlock::iterator Iter = FuncInfo.InsertPt; 11410b57cec5SDimitry Andric --Iter; 11420b57cec5SDimitry Andric assert(Iter->isBitcast()); 11438bcb0991SDimitry Andric Iter->setPhysRegsDeadExcept(ArrayRef<Register>(), TRI); 11440b57cec5SDimitry Andric updateValueMap(I, Reg); 11450b57cec5SDimitry Andric return true; 11460b57cec5SDimitry Andric } 11470b57cec5SDimitry Andric 11480b57cec5SDimitry Andric bool WebAssemblyFastISel::selectLoad(const Instruction *I) { 11490b57cec5SDimitry Andric const auto *Load = cast<LoadInst>(I); 11500b57cec5SDimitry Andric if (Load->isAtomic()) 11510b57cec5SDimitry Andric return false; 11520b57cec5SDimitry Andric if (!Subtarget->hasSIMD128() && Load->getType()->isVectorTy()) 11530b57cec5SDimitry Andric return false; 11540b57cec5SDimitry Andric 11550b57cec5SDimitry Andric Address Addr; 11560b57cec5SDimitry Andric if (!computeAddress(Load->getPointerOperand(), Addr)) 11570b57cec5SDimitry Andric return false; 11580b57cec5SDimitry Andric 11590b57cec5SDimitry Andric // TODO: Fold a following sign-/zero-extend into the load instruction. 11600b57cec5SDimitry Andric 11610b57cec5SDimitry Andric unsigned Opc; 11620b57cec5SDimitry Andric const TargetRegisterClass *RC; 1163*5ffd83dbSDimitry Andric bool A64 = Subtarget->hasAddr64(); 11640b57cec5SDimitry Andric switch (getSimpleType(Load->getType())) { 11650b57cec5SDimitry Andric case MVT::i1: 11660b57cec5SDimitry Andric case MVT::i8: 1167*5ffd83dbSDimitry Andric Opc = A64 ? WebAssembly::LOAD8_U_I32_A64 : WebAssembly::LOAD8_U_I32_A32; 11680b57cec5SDimitry Andric RC = &WebAssembly::I32RegClass; 11690b57cec5SDimitry Andric break; 11700b57cec5SDimitry Andric case MVT::i16: 1171*5ffd83dbSDimitry Andric Opc = A64 ? WebAssembly::LOAD16_U_I32_A64 : WebAssembly::LOAD16_U_I32_A32; 11720b57cec5SDimitry Andric RC = &WebAssembly::I32RegClass; 11730b57cec5SDimitry Andric break; 11740b57cec5SDimitry Andric case MVT::i32: 1175*5ffd83dbSDimitry Andric Opc = A64 ? WebAssembly::LOAD_I32_A64 : WebAssembly::LOAD_I32_A32; 11760b57cec5SDimitry Andric RC = &WebAssembly::I32RegClass; 11770b57cec5SDimitry Andric break; 11780b57cec5SDimitry Andric case MVT::i64: 1179*5ffd83dbSDimitry Andric Opc = A64 ? WebAssembly::LOAD_I64_A64 : WebAssembly::LOAD_I64_A32; 11800b57cec5SDimitry Andric RC = &WebAssembly::I64RegClass; 11810b57cec5SDimitry Andric break; 11820b57cec5SDimitry Andric case MVT::f32: 1183*5ffd83dbSDimitry Andric Opc = A64 ? WebAssembly::LOAD_F32_A64 : WebAssembly::LOAD_F32_A32; 11840b57cec5SDimitry Andric RC = &WebAssembly::F32RegClass; 11850b57cec5SDimitry Andric break; 11860b57cec5SDimitry Andric case MVT::f64: 1187*5ffd83dbSDimitry Andric Opc = A64 ? WebAssembly::LOAD_F64_A64 : WebAssembly::LOAD_F64_A32; 11880b57cec5SDimitry Andric RC = &WebAssembly::F64RegClass; 11890b57cec5SDimitry Andric break; 11900b57cec5SDimitry Andric default: 11910b57cec5SDimitry Andric return false; 11920b57cec5SDimitry Andric } 11930b57cec5SDimitry Andric 11940b57cec5SDimitry Andric materializeLoadStoreOperands(Addr); 11950b57cec5SDimitry Andric 11960b57cec5SDimitry Andric unsigned ResultReg = createResultReg(RC); 11970b57cec5SDimitry Andric auto MIB = BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc, TII.get(Opc), 11980b57cec5SDimitry Andric ResultReg); 11990b57cec5SDimitry Andric 12000b57cec5SDimitry Andric addLoadStoreOperands(Addr, MIB, createMachineMemOperandFor(Load)); 12010b57cec5SDimitry Andric 12020b57cec5SDimitry Andric updateValueMap(Load, ResultReg); 12030b57cec5SDimitry Andric return true; 12040b57cec5SDimitry Andric } 12050b57cec5SDimitry Andric 12060b57cec5SDimitry Andric bool WebAssemblyFastISel::selectStore(const Instruction *I) { 12070b57cec5SDimitry Andric const auto *Store = cast<StoreInst>(I); 12080b57cec5SDimitry Andric if (Store->isAtomic()) 12090b57cec5SDimitry Andric return false; 12100b57cec5SDimitry Andric if (!Subtarget->hasSIMD128() && 12110b57cec5SDimitry Andric Store->getValueOperand()->getType()->isVectorTy()) 12120b57cec5SDimitry Andric return false; 12130b57cec5SDimitry Andric 12140b57cec5SDimitry Andric Address Addr; 12150b57cec5SDimitry Andric if (!computeAddress(Store->getPointerOperand(), Addr)) 12160b57cec5SDimitry Andric return false; 12170b57cec5SDimitry Andric 12180b57cec5SDimitry Andric unsigned Opc; 12190b57cec5SDimitry Andric bool VTIsi1 = false; 1220*5ffd83dbSDimitry Andric bool A64 = Subtarget->hasAddr64(); 12210b57cec5SDimitry Andric switch (getSimpleType(Store->getValueOperand()->getType())) { 12220b57cec5SDimitry Andric case MVT::i1: 12230b57cec5SDimitry Andric VTIsi1 = true; 12240b57cec5SDimitry Andric LLVM_FALLTHROUGH; 12250b57cec5SDimitry Andric case MVT::i8: 1226*5ffd83dbSDimitry Andric Opc = A64 ? WebAssembly::STORE8_I32_A64 : WebAssembly::STORE8_I32_A32; 12270b57cec5SDimitry Andric break; 12280b57cec5SDimitry Andric case MVT::i16: 1229*5ffd83dbSDimitry Andric Opc = A64 ? WebAssembly::STORE16_I32_A64 : WebAssembly::STORE16_I32_A32; 12300b57cec5SDimitry Andric break; 12310b57cec5SDimitry Andric case MVT::i32: 1232*5ffd83dbSDimitry Andric Opc = A64 ? WebAssembly::STORE_I32_A64 : WebAssembly::STORE_I32_A32; 12330b57cec5SDimitry Andric break; 12340b57cec5SDimitry Andric case MVT::i64: 1235*5ffd83dbSDimitry Andric Opc = A64 ? WebAssembly::STORE_I64_A64 : WebAssembly::STORE_I64_A32; 12360b57cec5SDimitry Andric break; 12370b57cec5SDimitry Andric case MVT::f32: 1238*5ffd83dbSDimitry Andric Opc = A64 ? WebAssembly::STORE_F32_A64 : WebAssembly::STORE_F32_A32; 12390b57cec5SDimitry Andric break; 12400b57cec5SDimitry Andric case MVT::f64: 1241*5ffd83dbSDimitry Andric Opc = A64 ? WebAssembly::STORE_F64_A64 : WebAssembly::STORE_F64_A32; 12420b57cec5SDimitry Andric break; 12430b57cec5SDimitry Andric default: 12440b57cec5SDimitry Andric return false; 12450b57cec5SDimitry Andric } 12460b57cec5SDimitry Andric 12470b57cec5SDimitry Andric materializeLoadStoreOperands(Addr); 12480b57cec5SDimitry Andric 12490b57cec5SDimitry Andric unsigned ValueReg = getRegForValue(Store->getValueOperand()); 12500b57cec5SDimitry Andric if (ValueReg == 0) 12510b57cec5SDimitry Andric return false; 12520b57cec5SDimitry Andric if (VTIsi1) 12530b57cec5SDimitry Andric ValueReg = maskI1Value(ValueReg, Store->getValueOperand()); 12540b57cec5SDimitry Andric 12550b57cec5SDimitry Andric auto MIB = BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc, TII.get(Opc)); 12560b57cec5SDimitry Andric 12570b57cec5SDimitry Andric addLoadStoreOperands(Addr, MIB, createMachineMemOperandFor(Store)); 12580b57cec5SDimitry Andric 12590b57cec5SDimitry Andric MIB.addReg(ValueReg); 12600b57cec5SDimitry Andric return true; 12610b57cec5SDimitry Andric } 12620b57cec5SDimitry Andric 12630b57cec5SDimitry Andric bool WebAssemblyFastISel::selectBr(const Instruction *I) { 12640b57cec5SDimitry Andric const auto *Br = cast<BranchInst>(I); 12650b57cec5SDimitry Andric if (Br->isUnconditional()) { 12660b57cec5SDimitry Andric MachineBasicBlock *MSucc = FuncInfo.MBBMap[Br->getSuccessor(0)]; 12670b57cec5SDimitry Andric fastEmitBranch(MSucc, Br->getDebugLoc()); 12680b57cec5SDimitry Andric return true; 12690b57cec5SDimitry Andric } 12700b57cec5SDimitry Andric 12710b57cec5SDimitry Andric MachineBasicBlock *TBB = FuncInfo.MBBMap[Br->getSuccessor(0)]; 12720b57cec5SDimitry Andric MachineBasicBlock *FBB = FuncInfo.MBBMap[Br->getSuccessor(1)]; 12730b57cec5SDimitry Andric 12740b57cec5SDimitry Andric bool Not; 12750b57cec5SDimitry Andric unsigned CondReg = getRegForI1Value(Br->getCondition(), Not); 12760b57cec5SDimitry Andric if (CondReg == 0) 12770b57cec5SDimitry Andric return false; 12780b57cec5SDimitry Andric 12790b57cec5SDimitry Andric unsigned Opc = WebAssembly::BR_IF; 12800b57cec5SDimitry Andric if (Not) 12810b57cec5SDimitry Andric Opc = WebAssembly::BR_UNLESS; 12820b57cec5SDimitry Andric 12830b57cec5SDimitry Andric BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc, TII.get(Opc)) 12840b57cec5SDimitry Andric .addMBB(TBB) 12850b57cec5SDimitry Andric .addReg(CondReg); 12860b57cec5SDimitry Andric 12870b57cec5SDimitry Andric finishCondBranch(Br->getParent(), TBB, FBB); 12880b57cec5SDimitry Andric return true; 12890b57cec5SDimitry Andric } 12900b57cec5SDimitry Andric 12910b57cec5SDimitry Andric bool WebAssemblyFastISel::selectRet(const Instruction *I) { 12920b57cec5SDimitry Andric if (!FuncInfo.CanLowerReturn) 12930b57cec5SDimitry Andric return false; 12940b57cec5SDimitry Andric 12950b57cec5SDimitry Andric const auto *Ret = cast<ReturnInst>(I); 12960b57cec5SDimitry Andric 12970b57cec5SDimitry Andric if (Ret->getNumOperands() == 0) { 12980b57cec5SDimitry Andric BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc, 12998bcb0991SDimitry Andric TII.get(WebAssembly::RETURN)); 13000b57cec5SDimitry Andric return true; 13010b57cec5SDimitry Andric } 13020b57cec5SDimitry Andric 13038bcb0991SDimitry Andric // TODO: support multiple return in FastISel 13048bcb0991SDimitry Andric if (Ret->getNumOperands() > 1) 13058bcb0991SDimitry Andric return false; 13068bcb0991SDimitry Andric 13070b57cec5SDimitry Andric Value *RV = Ret->getOperand(0); 13080b57cec5SDimitry Andric if (!Subtarget->hasSIMD128() && RV->getType()->isVectorTy()) 13090b57cec5SDimitry Andric return false; 13100b57cec5SDimitry Andric 13110b57cec5SDimitry Andric switch (getSimpleType(RV->getType())) { 13120b57cec5SDimitry Andric case MVT::i1: 13130b57cec5SDimitry Andric case MVT::i8: 13140b57cec5SDimitry Andric case MVT::i16: 13150b57cec5SDimitry Andric case MVT::i32: 13160b57cec5SDimitry Andric case MVT::i64: 13170b57cec5SDimitry Andric case MVT::f32: 13180b57cec5SDimitry Andric case MVT::f64: 13190b57cec5SDimitry Andric case MVT::v16i8: 13200b57cec5SDimitry Andric case MVT::v8i16: 13210b57cec5SDimitry Andric case MVT::v4i32: 13220b57cec5SDimitry Andric case MVT::v2i64: 13230b57cec5SDimitry Andric case MVT::v4f32: 13240b57cec5SDimitry Andric case MVT::v2f64: 13250b57cec5SDimitry Andric case MVT::exnref: 13260b57cec5SDimitry Andric break; 13270b57cec5SDimitry Andric default: 13280b57cec5SDimitry Andric return false; 13290b57cec5SDimitry Andric } 13300b57cec5SDimitry Andric 13310b57cec5SDimitry Andric unsigned Reg; 13320b57cec5SDimitry Andric if (FuncInfo.Fn->getAttributes().hasAttribute(0, Attribute::SExt)) 13330b57cec5SDimitry Andric Reg = getRegForSignedValue(RV); 13340b57cec5SDimitry Andric else if (FuncInfo.Fn->getAttributes().hasAttribute(0, Attribute::ZExt)) 13350b57cec5SDimitry Andric Reg = getRegForUnsignedValue(RV); 13360b57cec5SDimitry Andric else 13370b57cec5SDimitry Andric Reg = getRegForValue(RV); 13380b57cec5SDimitry Andric 13390b57cec5SDimitry Andric if (Reg == 0) 13400b57cec5SDimitry Andric return false; 13410b57cec5SDimitry Andric 13428bcb0991SDimitry Andric BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc, 13438bcb0991SDimitry Andric TII.get(WebAssembly::RETURN)) 13448bcb0991SDimitry Andric .addReg(Reg); 13450b57cec5SDimitry Andric return true; 13460b57cec5SDimitry Andric } 13470b57cec5SDimitry Andric 13480b57cec5SDimitry Andric bool WebAssemblyFastISel::selectUnreachable(const Instruction *I) { 13490b57cec5SDimitry Andric BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc, 13500b57cec5SDimitry Andric TII.get(WebAssembly::UNREACHABLE)); 13510b57cec5SDimitry Andric return true; 13520b57cec5SDimitry Andric } 13530b57cec5SDimitry Andric 13540b57cec5SDimitry Andric bool WebAssemblyFastISel::fastSelectInstruction(const Instruction *I) { 13550b57cec5SDimitry Andric switch (I->getOpcode()) { 13560b57cec5SDimitry Andric case Instruction::Call: 13570b57cec5SDimitry Andric if (selectCall(I)) 13580b57cec5SDimitry Andric return true; 13590b57cec5SDimitry Andric break; 13600b57cec5SDimitry Andric case Instruction::Select: 13610b57cec5SDimitry Andric return selectSelect(I); 13620b57cec5SDimitry Andric case Instruction::Trunc: 13630b57cec5SDimitry Andric return selectTrunc(I); 13640b57cec5SDimitry Andric case Instruction::ZExt: 13650b57cec5SDimitry Andric return selectZExt(I); 13660b57cec5SDimitry Andric case Instruction::SExt: 13670b57cec5SDimitry Andric return selectSExt(I); 13680b57cec5SDimitry Andric case Instruction::ICmp: 13690b57cec5SDimitry Andric return selectICmp(I); 13700b57cec5SDimitry Andric case Instruction::FCmp: 13710b57cec5SDimitry Andric return selectFCmp(I); 13720b57cec5SDimitry Andric case Instruction::BitCast: 13730b57cec5SDimitry Andric return selectBitCast(I); 13740b57cec5SDimitry Andric case Instruction::Load: 13750b57cec5SDimitry Andric return selectLoad(I); 13760b57cec5SDimitry Andric case Instruction::Store: 13770b57cec5SDimitry Andric return selectStore(I); 13780b57cec5SDimitry Andric case Instruction::Br: 13790b57cec5SDimitry Andric return selectBr(I); 13800b57cec5SDimitry Andric case Instruction::Ret: 13810b57cec5SDimitry Andric return selectRet(I); 13820b57cec5SDimitry Andric case Instruction::Unreachable: 13830b57cec5SDimitry Andric return selectUnreachable(I); 13840b57cec5SDimitry Andric default: 13850b57cec5SDimitry Andric break; 13860b57cec5SDimitry Andric } 13870b57cec5SDimitry Andric 13880b57cec5SDimitry Andric // Fall back to target-independent instruction selection. 13890b57cec5SDimitry Andric return selectOperator(I, I->getOpcode()); 13900b57cec5SDimitry Andric } 13910b57cec5SDimitry Andric 13920b57cec5SDimitry Andric FastISel *WebAssembly::createFastISel(FunctionLoweringInfo &FuncInfo, 13930b57cec5SDimitry Andric const TargetLibraryInfo *LibInfo) { 13940b57cec5SDimitry Andric return new WebAssemblyFastISel(FuncInfo, LibInfo); 13950b57cec5SDimitry Andric } 1396