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