1*0b57cec5SDimitry Andric //===-- WebAssemblyFastISel.cpp - WebAssembly FastISel implementation -----===// 2*0b57cec5SDimitry Andric // 3*0b57cec5SDimitry Andric // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. 4*0b57cec5SDimitry Andric // See https://llvm.org/LICENSE.txt for license information. 5*0b57cec5SDimitry Andric // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception 6*0b57cec5SDimitry Andric // 7*0b57cec5SDimitry Andric //===----------------------------------------------------------------------===// 8*0b57cec5SDimitry Andric /// 9*0b57cec5SDimitry Andric /// \file 10*0b57cec5SDimitry Andric /// This file defines the WebAssembly-specific support for the FastISel 11*0b57cec5SDimitry Andric /// class. Some of the target-specific code is generated by tablegen in the file 12*0b57cec5SDimitry Andric /// WebAssemblyGenFastISel.inc, which is #included here. 13*0b57cec5SDimitry Andric /// 14*0b57cec5SDimitry Andric /// TODO: kill flags 15*0b57cec5SDimitry Andric /// 16*0b57cec5SDimitry Andric //===----------------------------------------------------------------------===// 17*0b57cec5SDimitry Andric 18*0b57cec5SDimitry Andric #include "MCTargetDesc/WebAssemblyMCTargetDesc.h" 19fe6060f1SDimitry Andric #include "Utils/WebAssemblyUtilities.h" 20*0b57cec5SDimitry Andric #include "WebAssembly.h" 21*0b57cec5SDimitry Andric #include "WebAssemblyMachineFunctionInfo.h" 22*0b57cec5SDimitry Andric #include "WebAssemblySubtarget.h" 23*0b57cec5SDimitry Andric #include "WebAssemblyTargetMachine.h" 24*0b57cec5SDimitry Andric #include "llvm/Analysis/BranchProbabilityInfo.h" 25*0b57cec5SDimitry Andric #include "llvm/CodeGen/FastISel.h" 26*0b57cec5SDimitry Andric #include "llvm/CodeGen/FunctionLoweringInfo.h" 27*0b57cec5SDimitry Andric #include "llvm/CodeGen/MachineConstantPool.h" 28*0b57cec5SDimitry Andric #include "llvm/CodeGen/MachineFrameInfo.h" 29*0b57cec5SDimitry Andric #include "llvm/CodeGen/MachineInstrBuilder.h" 30e8d8bef9SDimitry Andric #include "llvm/CodeGen/MachineModuleInfo.h" 31*0b57cec5SDimitry Andric #include "llvm/CodeGen/MachineRegisterInfo.h" 32*0b57cec5SDimitry Andric #include "llvm/IR/DataLayout.h" 33*0b57cec5SDimitry Andric #include "llvm/IR/DerivedTypes.h" 34*0b57cec5SDimitry Andric #include "llvm/IR/Function.h" 35*0b57cec5SDimitry Andric #include "llvm/IR/GetElementPtrTypeIterator.h" 36*0b57cec5SDimitry Andric #include "llvm/IR/GlobalAlias.h" 37*0b57cec5SDimitry Andric #include "llvm/IR/GlobalVariable.h" 38*0b57cec5SDimitry Andric #include "llvm/IR/Instructions.h" 39*0b57cec5SDimitry Andric #include "llvm/IR/IntrinsicInst.h" 40*0b57cec5SDimitry Andric #include "llvm/IR/Operator.h" 41*0b57cec5SDimitry Andric #include "llvm/IR/PatternMatch.h" 42*0b57cec5SDimitry Andric 43*0b57cec5SDimitry Andric using namespace llvm; 44*0b57cec5SDimitry Andric using namespace PatternMatch; 45*0b57cec5SDimitry Andric 46*0b57cec5SDimitry Andric #define DEBUG_TYPE "wasm-fastisel" 47*0b57cec5SDimitry Andric 48*0b57cec5SDimitry Andric namespace { 49*0b57cec5SDimitry Andric 50*0b57cec5SDimitry Andric class WebAssemblyFastISel final : public FastISel { 51*0b57cec5SDimitry Andric // All possible address modes. 52*0b57cec5SDimitry Andric class Address { 53*0b57cec5SDimitry Andric public: 54*0b57cec5SDimitry Andric using BaseKind = enum { RegBase, FrameIndexBase }; 55*0b57cec5SDimitry Andric 56*0b57cec5SDimitry Andric private: 57*0b57cec5SDimitry Andric BaseKind Kind = RegBase; 58*0b57cec5SDimitry Andric union { 59*0b57cec5SDimitry Andric unsigned Reg; 60*0b57cec5SDimitry Andric int FI; 61*0b57cec5SDimitry Andric } Base; 62*0b57cec5SDimitry Andric 63eaeb601bSDimitry Andric // Whether the base has been determined yet 64eaeb601bSDimitry Andric bool IsBaseSet = false; 65eaeb601bSDimitry Andric 66*0b57cec5SDimitry Andric int64_t Offset = 0; 67*0b57cec5SDimitry Andric 68*0b57cec5SDimitry Andric const GlobalValue *GV = nullptr; 69*0b57cec5SDimitry Andric 70*0b57cec5SDimitry Andric public: 71*0b57cec5SDimitry Andric // Innocuous defaults for our address. 72*0b57cec5SDimitry Andric Address() { Base.Reg = 0; } 73*0b57cec5SDimitry Andric void setKind(BaseKind K) { 74*0b57cec5SDimitry Andric assert(!isSet() && "Can't change kind with non-zero base"); 75*0b57cec5SDimitry Andric Kind = K; 76*0b57cec5SDimitry Andric } 77*0b57cec5SDimitry Andric BaseKind getKind() const { return Kind; } 78*0b57cec5SDimitry Andric bool isRegBase() const { return Kind == RegBase; } 79*0b57cec5SDimitry Andric bool isFIBase() const { return Kind == FrameIndexBase; } 80*0b57cec5SDimitry Andric void setReg(unsigned Reg) { 81*0b57cec5SDimitry Andric assert(isRegBase() && "Invalid base register access!"); 82eaeb601bSDimitry Andric assert(!IsBaseSet && "Base cannot be reset"); 83*0b57cec5SDimitry Andric Base.Reg = Reg; 84eaeb601bSDimitry Andric IsBaseSet = true; 85*0b57cec5SDimitry Andric } 86*0b57cec5SDimitry Andric unsigned getReg() const { 87*0b57cec5SDimitry Andric assert(isRegBase() && "Invalid base register access!"); 88*0b57cec5SDimitry Andric return Base.Reg; 89*0b57cec5SDimitry Andric } 90*0b57cec5SDimitry Andric void setFI(unsigned FI) { 91*0b57cec5SDimitry Andric assert(isFIBase() && "Invalid base frame index access!"); 92eaeb601bSDimitry Andric assert(!IsBaseSet && "Base cannot be reset"); 93*0b57cec5SDimitry Andric Base.FI = FI; 94eaeb601bSDimitry Andric IsBaseSet = true; 95*0b57cec5SDimitry Andric } 96*0b57cec5SDimitry Andric unsigned getFI() const { 97*0b57cec5SDimitry Andric assert(isFIBase() && "Invalid base frame index access!"); 98*0b57cec5SDimitry Andric return Base.FI; 99*0b57cec5SDimitry Andric } 100*0b57cec5SDimitry Andric 101*0b57cec5SDimitry Andric void setOffset(int64_t NewOffset) { 102*0b57cec5SDimitry Andric assert(NewOffset >= 0 && "Offsets must be non-negative"); 103*0b57cec5SDimitry Andric Offset = NewOffset; 104*0b57cec5SDimitry Andric } 105*0b57cec5SDimitry Andric int64_t getOffset() const { return Offset; } 106*0b57cec5SDimitry Andric void setGlobalValue(const GlobalValue *G) { GV = G; } 107*0b57cec5SDimitry Andric const GlobalValue *getGlobalValue() const { return GV; } 108eaeb601bSDimitry Andric bool isSet() const { return IsBaseSet; } 109*0b57cec5SDimitry Andric }; 110*0b57cec5SDimitry Andric 111*0b57cec5SDimitry Andric /// Keep a pointer to the WebAssemblySubtarget around so that we can make the 112*0b57cec5SDimitry Andric /// right decision when generating code for different targets. 113*0b57cec5SDimitry Andric const WebAssemblySubtarget *Subtarget; 114*0b57cec5SDimitry Andric LLVMContext *Context; 115*0b57cec5SDimitry Andric 116*0b57cec5SDimitry Andric private: 117*0b57cec5SDimitry Andric // Utility helper routines 118*0b57cec5SDimitry Andric MVT::SimpleValueType getSimpleType(Type *Ty) { 119*0b57cec5SDimitry Andric EVT VT = TLI.getValueType(DL, Ty, /*AllowUnknown=*/true); 120*0b57cec5SDimitry Andric return VT.isSimple() ? VT.getSimpleVT().SimpleTy 121*0b57cec5SDimitry Andric : MVT::INVALID_SIMPLE_VALUE_TYPE; 122*0b57cec5SDimitry Andric } 123*0b57cec5SDimitry Andric MVT::SimpleValueType getLegalType(MVT::SimpleValueType VT) { 124*0b57cec5SDimitry Andric switch (VT) { 125*0b57cec5SDimitry Andric case MVT::i1: 126*0b57cec5SDimitry Andric case MVT::i8: 127*0b57cec5SDimitry Andric case MVT::i16: 128*0b57cec5SDimitry Andric return MVT::i32; 129*0b57cec5SDimitry Andric case MVT::i32: 130*0b57cec5SDimitry Andric case MVT::i64: 131*0b57cec5SDimitry Andric case MVT::f32: 132*0b57cec5SDimitry Andric case MVT::f64: 133fe6060f1SDimitry Andric return VT; 134e8d8bef9SDimitry Andric case MVT::funcref: 135e8d8bef9SDimitry Andric case MVT::externref: 136fe6060f1SDimitry Andric if (Subtarget->hasReferenceTypes()) 137*0b57cec5SDimitry Andric return VT; 138fe6060f1SDimitry Andric break; 139*0b57cec5SDimitry Andric case MVT::f16: 140*0b57cec5SDimitry Andric return MVT::f32; 141*0b57cec5SDimitry Andric case MVT::v16i8: 142*0b57cec5SDimitry Andric case MVT::v8i16: 143*0b57cec5SDimitry Andric case MVT::v4i32: 144*0b57cec5SDimitry Andric case MVT::v4f32: 145*0b57cec5SDimitry Andric case MVT::v2i64: 146*0b57cec5SDimitry Andric case MVT::v2f64: 147fe6060f1SDimitry Andric if (Subtarget->hasSIMD128()) 148*0b57cec5SDimitry Andric return VT; 149*0b57cec5SDimitry Andric break; 150*0b57cec5SDimitry Andric default: 151*0b57cec5SDimitry Andric break; 152*0b57cec5SDimitry Andric } 153*0b57cec5SDimitry Andric return MVT::INVALID_SIMPLE_VALUE_TYPE; 154*0b57cec5SDimitry Andric } 155*0b57cec5SDimitry Andric bool computeAddress(const Value *Obj, Address &Addr); 156*0b57cec5SDimitry Andric void materializeLoadStoreOperands(Address &Addr); 157*0b57cec5SDimitry Andric void addLoadStoreOperands(const Address &Addr, const MachineInstrBuilder &MIB, 158*0b57cec5SDimitry Andric MachineMemOperand *MMO); 159*0b57cec5SDimitry Andric unsigned maskI1Value(unsigned Reg, const Value *V); 16069ade1e0SDimitry Andric unsigned getRegForI1Value(const Value *V, const BasicBlock *BB, bool &Not); 161*0b57cec5SDimitry Andric unsigned zeroExtendToI32(unsigned Reg, const Value *V, 162*0b57cec5SDimitry Andric MVT::SimpleValueType From); 163*0b57cec5SDimitry Andric unsigned signExtendToI32(unsigned Reg, const Value *V, 164*0b57cec5SDimitry Andric MVT::SimpleValueType From); 165*0b57cec5SDimitry Andric unsigned zeroExtend(unsigned Reg, const Value *V, MVT::SimpleValueType From, 166*0b57cec5SDimitry Andric MVT::SimpleValueType To); 167*0b57cec5SDimitry Andric unsigned signExtend(unsigned Reg, const Value *V, MVT::SimpleValueType From, 168*0b57cec5SDimitry Andric MVT::SimpleValueType To); 169*0b57cec5SDimitry Andric unsigned getRegForUnsignedValue(const Value *V); 170*0b57cec5SDimitry Andric unsigned getRegForSignedValue(const Value *V); 171*0b57cec5SDimitry Andric unsigned getRegForPromotedValue(const Value *V, bool IsSigned); 172*0b57cec5SDimitry Andric unsigned notValue(unsigned Reg); 173*0b57cec5SDimitry Andric unsigned copyValue(unsigned Reg); 174*0b57cec5SDimitry Andric 175*0b57cec5SDimitry Andric // Backend specific FastISel code. 176*0b57cec5SDimitry Andric unsigned fastMaterializeAlloca(const AllocaInst *AI) override; 177*0b57cec5SDimitry Andric unsigned fastMaterializeConstant(const Constant *C) override; 178*0b57cec5SDimitry Andric bool fastLowerArguments() override; 179*0b57cec5SDimitry Andric 180*0b57cec5SDimitry Andric // Selection routines. 181*0b57cec5SDimitry Andric bool selectCall(const Instruction *I); 182*0b57cec5SDimitry Andric bool selectSelect(const Instruction *I); 183*0b57cec5SDimitry Andric bool selectTrunc(const Instruction *I); 184*0b57cec5SDimitry Andric bool selectZExt(const Instruction *I); 185*0b57cec5SDimitry Andric bool selectSExt(const Instruction *I); 186*0b57cec5SDimitry Andric bool selectICmp(const Instruction *I); 187*0b57cec5SDimitry Andric bool selectFCmp(const Instruction *I); 188*0b57cec5SDimitry Andric bool selectBitCast(const Instruction *I); 189*0b57cec5SDimitry Andric bool selectLoad(const Instruction *I); 190*0b57cec5SDimitry Andric bool selectStore(const Instruction *I); 191*0b57cec5SDimitry Andric bool selectBr(const Instruction *I); 192*0b57cec5SDimitry Andric bool selectRet(const Instruction *I); 193*0b57cec5SDimitry Andric bool selectUnreachable(const Instruction *I); 194*0b57cec5SDimitry Andric 195*0b57cec5SDimitry Andric public: 196*0b57cec5SDimitry Andric // Backend specific FastISel code. 197*0b57cec5SDimitry Andric WebAssemblyFastISel(FunctionLoweringInfo &FuncInfo, 198*0b57cec5SDimitry Andric const TargetLibraryInfo *LibInfo) 199*0b57cec5SDimitry Andric : FastISel(FuncInfo, LibInfo, /*SkipTargetIndependentISel=*/true) { 200*0b57cec5SDimitry Andric Subtarget = &FuncInfo.MF->getSubtarget<WebAssemblySubtarget>(); 201*0b57cec5SDimitry Andric Context = &FuncInfo.Fn->getContext(); 202*0b57cec5SDimitry Andric } 203*0b57cec5SDimitry Andric 204*0b57cec5SDimitry Andric bool fastSelectInstruction(const Instruction *I) override; 205*0b57cec5SDimitry Andric 206*0b57cec5SDimitry Andric #include "WebAssemblyGenFastISel.inc" 207*0b57cec5SDimitry Andric }; 208*0b57cec5SDimitry Andric 209*0b57cec5SDimitry Andric } // end anonymous namespace 210*0b57cec5SDimitry Andric 211*0b57cec5SDimitry Andric bool WebAssemblyFastISel::computeAddress(const Value *Obj, Address &Addr) { 212*0b57cec5SDimitry Andric const User *U = nullptr; 213*0b57cec5SDimitry Andric unsigned Opcode = Instruction::UserOp1; 214*0b57cec5SDimitry Andric if (const auto *I = dyn_cast<Instruction>(Obj)) { 215*0b57cec5SDimitry Andric // Don't walk into other basic blocks unless the object is an alloca from 216*0b57cec5SDimitry Andric // another block, otherwise it may not have a virtual register assigned. 217*0b57cec5SDimitry Andric if (FuncInfo.StaticAllocaMap.count(static_cast<const AllocaInst *>(Obj)) || 218*0b57cec5SDimitry Andric FuncInfo.MBBMap[I->getParent()] == FuncInfo.MBB) { 219*0b57cec5SDimitry Andric Opcode = I->getOpcode(); 220*0b57cec5SDimitry Andric U = I; 221*0b57cec5SDimitry Andric } 222*0b57cec5SDimitry Andric } else if (const auto *C = dyn_cast<ConstantExpr>(Obj)) { 223*0b57cec5SDimitry Andric Opcode = C->getOpcode(); 224*0b57cec5SDimitry Andric U = C; 225*0b57cec5SDimitry Andric } 226*0b57cec5SDimitry Andric 227*0b57cec5SDimitry Andric if (auto *Ty = dyn_cast<PointerType>(Obj->getType())) 228*0b57cec5SDimitry Andric if (Ty->getAddressSpace() > 255) 229*0b57cec5SDimitry Andric // Fast instruction selection doesn't support the special 230*0b57cec5SDimitry Andric // address spaces. 231*0b57cec5SDimitry Andric return false; 232*0b57cec5SDimitry Andric 233*0b57cec5SDimitry Andric if (const auto *GV = dyn_cast<GlobalValue>(Obj)) { 234*0b57cec5SDimitry Andric if (TLI.isPositionIndependent()) 235*0b57cec5SDimitry Andric return false; 236*0b57cec5SDimitry Andric if (Addr.getGlobalValue()) 237*0b57cec5SDimitry Andric return false; 238*0b57cec5SDimitry Andric if (GV->isThreadLocal()) 239*0b57cec5SDimitry Andric return false; 240*0b57cec5SDimitry Andric Addr.setGlobalValue(GV); 241*0b57cec5SDimitry Andric return true; 242*0b57cec5SDimitry Andric } 243*0b57cec5SDimitry Andric 244*0b57cec5SDimitry Andric switch (Opcode) { 245*0b57cec5SDimitry Andric default: 246*0b57cec5SDimitry Andric break; 247*0b57cec5SDimitry Andric case Instruction::BitCast: { 248*0b57cec5SDimitry Andric // Look through bitcasts. 249*0b57cec5SDimitry Andric return computeAddress(U->getOperand(0), Addr); 250*0b57cec5SDimitry Andric } 251*0b57cec5SDimitry Andric case Instruction::IntToPtr: { 252*0b57cec5SDimitry Andric // Look past no-op inttoptrs. 253*0b57cec5SDimitry Andric if (TLI.getValueType(DL, U->getOperand(0)->getType()) == 254*0b57cec5SDimitry Andric TLI.getPointerTy(DL)) 255*0b57cec5SDimitry Andric return computeAddress(U->getOperand(0), Addr); 256*0b57cec5SDimitry Andric break; 257*0b57cec5SDimitry Andric } 258*0b57cec5SDimitry Andric case Instruction::PtrToInt: { 259*0b57cec5SDimitry Andric // Look past no-op ptrtoints. 260*0b57cec5SDimitry Andric if (TLI.getValueType(DL, U->getType()) == TLI.getPointerTy(DL)) 261*0b57cec5SDimitry Andric return computeAddress(U->getOperand(0), Addr); 262*0b57cec5SDimitry Andric break; 263*0b57cec5SDimitry Andric } 264*0b57cec5SDimitry Andric case Instruction::GetElementPtr: { 265*0b57cec5SDimitry Andric Address SavedAddr = Addr; 266*0b57cec5SDimitry Andric uint64_t TmpOffset = Addr.getOffset(); 267*0b57cec5SDimitry Andric // Non-inbounds geps can wrap; wasm's offsets can't. 268*0b57cec5SDimitry Andric if (!cast<GEPOperator>(U)->isInBounds()) 269*0b57cec5SDimitry Andric goto unsupported_gep; 270*0b57cec5SDimitry Andric // Iterate through the GEP folding the constants into offsets where 271*0b57cec5SDimitry Andric // we can. 272*0b57cec5SDimitry Andric for (gep_type_iterator GTI = gep_type_begin(U), E = gep_type_end(U); 273*0b57cec5SDimitry Andric GTI != E; ++GTI) { 274*0b57cec5SDimitry Andric const Value *Op = GTI.getOperand(); 275*0b57cec5SDimitry Andric if (StructType *STy = GTI.getStructTypeOrNull()) { 276*0b57cec5SDimitry Andric const StructLayout *SL = DL.getStructLayout(STy); 277*0b57cec5SDimitry Andric unsigned Idx = cast<ConstantInt>(Op)->getZExtValue(); 278*0b57cec5SDimitry Andric TmpOffset += SL->getElementOffset(Idx); 279*0b57cec5SDimitry Andric } else { 280*0b57cec5SDimitry Andric uint64_t S = DL.getTypeAllocSize(GTI.getIndexedType()); 281*0b57cec5SDimitry Andric for (;;) { 282*0b57cec5SDimitry Andric if (const auto *CI = dyn_cast<ConstantInt>(Op)) { 283*0b57cec5SDimitry Andric // Constant-offset addressing. 284*0b57cec5SDimitry Andric TmpOffset += CI->getSExtValue() * S; 285*0b57cec5SDimitry Andric break; 286*0b57cec5SDimitry Andric } 287*0b57cec5SDimitry Andric if (S == 1 && Addr.isRegBase() && Addr.getReg() == 0) { 288*0b57cec5SDimitry Andric // An unscaled add of a register. Set it as the new base. 289*0b57cec5SDimitry Andric unsigned Reg = getRegForValue(Op); 290*0b57cec5SDimitry Andric if (Reg == 0) 291*0b57cec5SDimitry Andric return false; 292*0b57cec5SDimitry Andric Addr.setReg(Reg); 293*0b57cec5SDimitry Andric break; 294*0b57cec5SDimitry Andric } 295*0b57cec5SDimitry Andric if (canFoldAddIntoGEP(U, Op)) { 296*0b57cec5SDimitry Andric // A compatible add with a constant operand. Fold the constant. 297*0b57cec5SDimitry Andric auto *CI = cast<ConstantInt>(cast<AddOperator>(Op)->getOperand(1)); 298*0b57cec5SDimitry Andric TmpOffset += CI->getSExtValue() * S; 299*0b57cec5SDimitry Andric // Iterate on the other operand. 300*0b57cec5SDimitry Andric Op = cast<AddOperator>(Op)->getOperand(0); 301*0b57cec5SDimitry Andric continue; 302*0b57cec5SDimitry Andric } 303*0b57cec5SDimitry Andric // Unsupported 304*0b57cec5SDimitry Andric goto unsupported_gep; 305*0b57cec5SDimitry Andric } 306*0b57cec5SDimitry Andric } 307*0b57cec5SDimitry Andric } 308*0b57cec5SDimitry Andric // Don't fold in negative offsets. 309*0b57cec5SDimitry Andric if (int64_t(TmpOffset) >= 0) { 310*0b57cec5SDimitry Andric // Try to grab the base operand now. 311*0b57cec5SDimitry Andric Addr.setOffset(TmpOffset); 312*0b57cec5SDimitry Andric if (computeAddress(U->getOperand(0), Addr)) 313*0b57cec5SDimitry Andric return true; 314*0b57cec5SDimitry Andric } 315*0b57cec5SDimitry Andric // We failed, restore everything and try the other options. 316*0b57cec5SDimitry Andric Addr = SavedAddr; 317*0b57cec5SDimitry Andric unsupported_gep: 318*0b57cec5SDimitry Andric break; 319*0b57cec5SDimitry Andric } 320*0b57cec5SDimitry Andric case Instruction::Alloca: { 321*0b57cec5SDimitry Andric const auto *AI = cast<AllocaInst>(Obj); 322*0b57cec5SDimitry Andric DenseMap<const AllocaInst *, int>::iterator SI = 323*0b57cec5SDimitry Andric FuncInfo.StaticAllocaMap.find(AI); 324*0b57cec5SDimitry Andric if (SI != FuncInfo.StaticAllocaMap.end()) { 325*0b57cec5SDimitry Andric if (Addr.isSet()) { 326*0b57cec5SDimitry Andric return false; 327*0b57cec5SDimitry Andric } 328*0b57cec5SDimitry Andric Addr.setKind(Address::FrameIndexBase); 329*0b57cec5SDimitry Andric Addr.setFI(SI->second); 330*0b57cec5SDimitry Andric return true; 331*0b57cec5SDimitry Andric } 332*0b57cec5SDimitry Andric break; 333*0b57cec5SDimitry Andric } 334*0b57cec5SDimitry Andric case Instruction::Add: { 335*0b57cec5SDimitry Andric // Adds of constants are common and easy enough. 336*0b57cec5SDimitry Andric const Value *LHS = U->getOperand(0); 337*0b57cec5SDimitry Andric const Value *RHS = U->getOperand(1); 338*0b57cec5SDimitry Andric 339*0b57cec5SDimitry Andric if (isa<ConstantInt>(LHS)) 340*0b57cec5SDimitry Andric std::swap(LHS, RHS); 341*0b57cec5SDimitry Andric 342*0b57cec5SDimitry Andric if (const auto *CI = dyn_cast<ConstantInt>(RHS)) { 343*0b57cec5SDimitry Andric uint64_t TmpOffset = Addr.getOffset() + CI->getSExtValue(); 344*0b57cec5SDimitry Andric if (int64_t(TmpOffset) >= 0) { 345*0b57cec5SDimitry Andric Addr.setOffset(TmpOffset); 346*0b57cec5SDimitry Andric return computeAddress(LHS, Addr); 347*0b57cec5SDimitry Andric } 348*0b57cec5SDimitry Andric } 349*0b57cec5SDimitry Andric 350*0b57cec5SDimitry Andric Address Backup = Addr; 351*0b57cec5SDimitry Andric if (computeAddress(LHS, Addr) && computeAddress(RHS, Addr)) 352*0b57cec5SDimitry Andric return true; 353*0b57cec5SDimitry Andric Addr = Backup; 354*0b57cec5SDimitry Andric 355*0b57cec5SDimitry Andric break; 356*0b57cec5SDimitry Andric } 357*0b57cec5SDimitry Andric case Instruction::Sub: { 358*0b57cec5SDimitry Andric // Subs of constants are common and easy enough. 359*0b57cec5SDimitry Andric const Value *LHS = U->getOperand(0); 360*0b57cec5SDimitry Andric const Value *RHS = U->getOperand(1); 361*0b57cec5SDimitry Andric 362*0b57cec5SDimitry Andric if (const auto *CI = dyn_cast<ConstantInt>(RHS)) { 363*0b57cec5SDimitry Andric int64_t TmpOffset = Addr.getOffset() - CI->getSExtValue(); 364*0b57cec5SDimitry Andric if (TmpOffset >= 0) { 365*0b57cec5SDimitry Andric Addr.setOffset(TmpOffset); 366*0b57cec5SDimitry Andric return computeAddress(LHS, Addr); 367*0b57cec5SDimitry Andric } 368*0b57cec5SDimitry Andric } 369*0b57cec5SDimitry Andric break; 370*0b57cec5SDimitry Andric } 371*0b57cec5SDimitry Andric } 372*0b57cec5SDimitry Andric if (Addr.isSet()) { 373*0b57cec5SDimitry Andric return false; 374*0b57cec5SDimitry Andric } 375*0b57cec5SDimitry Andric unsigned Reg = getRegForValue(Obj); 376*0b57cec5SDimitry Andric if (Reg == 0) 377*0b57cec5SDimitry Andric return false; 378*0b57cec5SDimitry Andric Addr.setReg(Reg); 379*0b57cec5SDimitry Andric return Addr.getReg() != 0; 380*0b57cec5SDimitry Andric } 381*0b57cec5SDimitry Andric 382*0b57cec5SDimitry Andric void WebAssemblyFastISel::materializeLoadStoreOperands(Address &Addr) { 383*0b57cec5SDimitry Andric if (Addr.isRegBase()) { 384*0b57cec5SDimitry Andric unsigned Reg = Addr.getReg(); 385*0b57cec5SDimitry Andric if (Reg == 0) { 386*0b57cec5SDimitry Andric Reg = createResultReg(Subtarget->hasAddr64() ? &WebAssembly::I64RegClass 387*0b57cec5SDimitry Andric : &WebAssembly::I32RegClass); 388*0b57cec5SDimitry Andric unsigned Opc = Subtarget->hasAddr64() ? WebAssembly::CONST_I64 389*0b57cec5SDimitry Andric : WebAssembly::CONST_I32; 390*0b57cec5SDimitry Andric BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc, TII.get(Opc), Reg) 391*0b57cec5SDimitry Andric .addImm(0); 392*0b57cec5SDimitry Andric Addr.setReg(Reg); 393*0b57cec5SDimitry Andric } 394*0b57cec5SDimitry Andric } 395*0b57cec5SDimitry Andric } 396*0b57cec5SDimitry Andric 397*0b57cec5SDimitry Andric void WebAssemblyFastISel::addLoadStoreOperands(const Address &Addr, 398*0b57cec5SDimitry Andric const MachineInstrBuilder &MIB, 399*0b57cec5SDimitry Andric MachineMemOperand *MMO) { 400*0b57cec5SDimitry Andric // Set the alignment operand (this is rewritten in SetP2AlignOperands). 401*0b57cec5SDimitry Andric // TODO: Disable SetP2AlignOperands for FastISel and just do it here. 402*0b57cec5SDimitry Andric MIB.addImm(0); 403*0b57cec5SDimitry Andric 404*0b57cec5SDimitry Andric if (const GlobalValue *GV = Addr.getGlobalValue()) 405*0b57cec5SDimitry Andric MIB.addGlobalAddress(GV, Addr.getOffset()); 406*0b57cec5SDimitry Andric else 407*0b57cec5SDimitry Andric MIB.addImm(Addr.getOffset()); 408*0b57cec5SDimitry Andric 409*0b57cec5SDimitry Andric if (Addr.isRegBase()) 410*0b57cec5SDimitry Andric MIB.addReg(Addr.getReg()); 411*0b57cec5SDimitry Andric else 412*0b57cec5SDimitry Andric MIB.addFrameIndex(Addr.getFI()); 413*0b57cec5SDimitry Andric 414*0b57cec5SDimitry Andric MIB.addMemOperand(MMO); 415*0b57cec5SDimitry Andric } 416*0b57cec5SDimitry Andric 417*0b57cec5SDimitry Andric unsigned WebAssemblyFastISel::maskI1Value(unsigned Reg, const Value *V) { 418*0b57cec5SDimitry Andric return zeroExtendToI32(Reg, V, MVT::i1); 419*0b57cec5SDimitry Andric } 420*0b57cec5SDimitry Andric 42169ade1e0SDimitry Andric unsigned WebAssemblyFastISel::getRegForI1Value(const Value *V, 42269ade1e0SDimitry Andric const BasicBlock *BB, 42369ade1e0SDimitry Andric bool &Not) { 424*0b57cec5SDimitry Andric if (const auto *ICmp = dyn_cast<ICmpInst>(V)) 425*0b57cec5SDimitry Andric if (const ConstantInt *C = dyn_cast<ConstantInt>(ICmp->getOperand(1))) 42669ade1e0SDimitry Andric if (ICmp->isEquality() && C->isZero() && C->getType()->isIntegerTy(32) && 42769ade1e0SDimitry Andric ICmp->getParent() == BB) { 428*0b57cec5SDimitry Andric Not = ICmp->isTrueWhenEqual(); 429*0b57cec5SDimitry Andric return getRegForValue(ICmp->getOperand(0)); 430*0b57cec5SDimitry Andric } 431*0b57cec5SDimitry Andric 432*0b57cec5SDimitry Andric Not = false; 433*0b57cec5SDimitry Andric unsigned Reg = getRegForValue(V); 434*0b57cec5SDimitry Andric if (Reg == 0) 435*0b57cec5SDimitry Andric return 0; 436*0b57cec5SDimitry Andric return maskI1Value(Reg, V); 437*0b57cec5SDimitry Andric } 438*0b57cec5SDimitry Andric 439*0b57cec5SDimitry Andric unsigned WebAssemblyFastISel::zeroExtendToI32(unsigned Reg, const Value *V, 440*0b57cec5SDimitry Andric MVT::SimpleValueType From) { 441*0b57cec5SDimitry Andric if (Reg == 0) 442*0b57cec5SDimitry Andric return 0; 443*0b57cec5SDimitry Andric 444*0b57cec5SDimitry Andric switch (From) { 445*0b57cec5SDimitry Andric case MVT::i1: 446*0b57cec5SDimitry Andric // If the value is naturally an i1, we don't need to mask it. We only know 447*0b57cec5SDimitry Andric // if a value is naturally an i1 if it is definitely lowered by FastISel, 448*0b57cec5SDimitry Andric // not a DAG ISel fallback. 449*0b57cec5SDimitry Andric if (V != nullptr && isa<Argument>(V) && cast<Argument>(V)->hasZExtAttr()) 450*0b57cec5SDimitry Andric return copyValue(Reg); 451*0b57cec5SDimitry Andric break; 452*0b57cec5SDimitry Andric case MVT::i8: 453*0b57cec5SDimitry Andric case MVT::i16: 454*0b57cec5SDimitry Andric break; 455*0b57cec5SDimitry Andric case MVT::i32: 456*0b57cec5SDimitry Andric return copyValue(Reg); 457*0b57cec5SDimitry Andric default: 458*0b57cec5SDimitry Andric return 0; 459*0b57cec5SDimitry Andric } 460*0b57cec5SDimitry Andric 461*0b57cec5SDimitry Andric unsigned Imm = createResultReg(&WebAssembly::I32RegClass); 462*0b57cec5SDimitry Andric BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc, 463*0b57cec5SDimitry Andric TII.get(WebAssembly::CONST_I32), Imm) 464*0b57cec5SDimitry Andric .addImm(~(~uint64_t(0) << MVT(From).getSizeInBits())); 465*0b57cec5SDimitry Andric 466*0b57cec5SDimitry Andric unsigned Result = createResultReg(&WebAssembly::I32RegClass); 467*0b57cec5SDimitry Andric BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc, 468*0b57cec5SDimitry Andric TII.get(WebAssembly::AND_I32), Result) 469*0b57cec5SDimitry Andric .addReg(Reg) 470*0b57cec5SDimitry Andric .addReg(Imm); 471*0b57cec5SDimitry Andric 472*0b57cec5SDimitry Andric return Result; 473*0b57cec5SDimitry Andric } 474*0b57cec5SDimitry Andric 475*0b57cec5SDimitry Andric unsigned WebAssemblyFastISel::signExtendToI32(unsigned Reg, const Value *V, 476*0b57cec5SDimitry Andric MVT::SimpleValueType From) { 477*0b57cec5SDimitry Andric if (Reg == 0) 478*0b57cec5SDimitry Andric return 0; 479*0b57cec5SDimitry Andric 480*0b57cec5SDimitry Andric switch (From) { 481*0b57cec5SDimitry Andric case MVT::i1: 482*0b57cec5SDimitry Andric case MVT::i8: 483*0b57cec5SDimitry Andric case MVT::i16: 484*0b57cec5SDimitry Andric break; 485*0b57cec5SDimitry Andric case MVT::i32: 486*0b57cec5SDimitry Andric return copyValue(Reg); 487*0b57cec5SDimitry Andric default: 488*0b57cec5SDimitry Andric return 0; 489*0b57cec5SDimitry Andric } 490*0b57cec5SDimitry Andric 491*0b57cec5SDimitry Andric unsigned Imm = createResultReg(&WebAssembly::I32RegClass); 492*0b57cec5SDimitry Andric BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc, 493*0b57cec5SDimitry Andric TII.get(WebAssembly::CONST_I32), Imm) 494*0b57cec5SDimitry Andric .addImm(32 - MVT(From).getSizeInBits()); 495*0b57cec5SDimitry Andric 496*0b57cec5SDimitry Andric unsigned Left = createResultReg(&WebAssembly::I32RegClass); 497*0b57cec5SDimitry Andric BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc, 498*0b57cec5SDimitry Andric TII.get(WebAssembly::SHL_I32), Left) 499*0b57cec5SDimitry Andric .addReg(Reg) 500*0b57cec5SDimitry Andric .addReg(Imm); 501*0b57cec5SDimitry Andric 502*0b57cec5SDimitry Andric unsigned Right = createResultReg(&WebAssembly::I32RegClass); 503*0b57cec5SDimitry Andric BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc, 504*0b57cec5SDimitry Andric TII.get(WebAssembly::SHR_S_I32), Right) 505*0b57cec5SDimitry Andric .addReg(Left) 506*0b57cec5SDimitry Andric .addReg(Imm); 507*0b57cec5SDimitry Andric 508*0b57cec5SDimitry Andric return Right; 509*0b57cec5SDimitry Andric } 510*0b57cec5SDimitry Andric 511*0b57cec5SDimitry Andric unsigned WebAssemblyFastISel::zeroExtend(unsigned Reg, const Value *V, 512*0b57cec5SDimitry Andric MVT::SimpleValueType From, 513*0b57cec5SDimitry Andric MVT::SimpleValueType To) { 514*0b57cec5SDimitry Andric if (To == MVT::i64) { 515*0b57cec5SDimitry Andric if (From == MVT::i64) 516*0b57cec5SDimitry Andric return copyValue(Reg); 517*0b57cec5SDimitry Andric 518*0b57cec5SDimitry Andric Reg = zeroExtendToI32(Reg, V, From); 519*0b57cec5SDimitry Andric 520*0b57cec5SDimitry Andric unsigned Result = createResultReg(&WebAssembly::I64RegClass); 521*0b57cec5SDimitry Andric BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc, 522*0b57cec5SDimitry Andric TII.get(WebAssembly::I64_EXTEND_U_I32), Result) 523*0b57cec5SDimitry Andric .addReg(Reg); 524*0b57cec5SDimitry Andric return Result; 525*0b57cec5SDimitry Andric } 526*0b57cec5SDimitry Andric 527*0b57cec5SDimitry Andric if (To == MVT::i32) 528*0b57cec5SDimitry Andric return zeroExtendToI32(Reg, V, From); 529*0b57cec5SDimitry Andric 530*0b57cec5SDimitry Andric return 0; 531*0b57cec5SDimitry Andric } 532*0b57cec5SDimitry Andric 533*0b57cec5SDimitry Andric unsigned WebAssemblyFastISel::signExtend(unsigned Reg, const Value *V, 534*0b57cec5SDimitry Andric MVT::SimpleValueType From, 535*0b57cec5SDimitry Andric MVT::SimpleValueType To) { 536*0b57cec5SDimitry Andric if (To == MVT::i64) { 537*0b57cec5SDimitry Andric if (From == MVT::i64) 538*0b57cec5SDimitry Andric return copyValue(Reg); 539*0b57cec5SDimitry Andric 540*0b57cec5SDimitry Andric Reg = signExtendToI32(Reg, V, From); 541*0b57cec5SDimitry Andric 542*0b57cec5SDimitry Andric unsigned Result = createResultReg(&WebAssembly::I64RegClass); 543*0b57cec5SDimitry Andric BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc, 544*0b57cec5SDimitry Andric TII.get(WebAssembly::I64_EXTEND_S_I32), Result) 545*0b57cec5SDimitry Andric .addReg(Reg); 546*0b57cec5SDimitry Andric return Result; 547*0b57cec5SDimitry Andric } 548*0b57cec5SDimitry Andric 549*0b57cec5SDimitry Andric if (To == MVT::i32) 550*0b57cec5SDimitry Andric return signExtendToI32(Reg, V, From); 551*0b57cec5SDimitry Andric 552*0b57cec5SDimitry Andric return 0; 553*0b57cec5SDimitry Andric } 554*0b57cec5SDimitry Andric 555*0b57cec5SDimitry Andric unsigned WebAssemblyFastISel::getRegForUnsignedValue(const Value *V) { 556*0b57cec5SDimitry Andric MVT::SimpleValueType From = getSimpleType(V->getType()); 557*0b57cec5SDimitry Andric MVT::SimpleValueType To = getLegalType(From); 558*0b57cec5SDimitry Andric unsigned VReg = getRegForValue(V); 559*0b57cec5SDimitry Andric if (VReg == 0) 560*0b57cec5SDimitry Andric return 0; 561*0b57cec5SDimitry Andric return zeroExtend(VReg, V, From, To); 562*0b57cec5SDimitry Andric } 563*0b57cec5SDimitry Andric 564*0b57cec5SDimitry Andric unsigned WebAssemblyFastISel::getRegForSignedValue(const Value *V) { 565*0b57cec5SDimitry Andric MVT::SimpleValueType From = getSimpleType(V->getType()); 566*0b57cec5SDimitry Andric MVT::SimpleValueType To = getLegalType(From); 567*0b57cec5SDimitry Andric unsigned VReg = getRegForValue(V); 568*0b57cec5SDimitry Andric if (VReg == 0) 569*0b57cec5SDimitry Andric return 0; 570*0b57cec5SDimitry Andric return signExtend(VReg, V, From, To); 571*0b57cec5SDimitry Andric } 572*0b57cec5SDimitry Andric 573*0b57cec5SDimitry Andric unsigned WebAssemblyFastISel::getRegForPromotedValue(const Value *V, 574*0b57cec5SDimitry Andric bool IsSigned) { 575*0b57cec5SDimitry Andric return IsSigned ? getRegForSignedValue(V) : getRegForUnsignedValue(V); 576*0b57cec5SDimitry Andric } 577*0b57cec5SDimitry Andric 578*0b57cec5SDimitry Andric unsigned WebAssemblyFastISel::notValue(unsigned Reg) { 579*0b57cec5SDimitry Andric assert(MRI.getRegClass(Reg) == &WebAssembly::I32RegClass); 580*0b57cec5SDimitry Andric 581*0b57cec5SDimitry Andric unsigned NotReg = createResultReg(&WebAssembly::I32RegClass); 582*0b57cec5SDimitry Andric BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc, 583*0b57cec5SDimitry Andric TII.get(WebAssembly::EQZ_I32), NotReg) 584*0b57cec5SDimitry Andric .addReg(Reg); 585*0b57cec5SDimitry Andric return NotReg; 586*0b57cec5SDimitry Andric } 587*0b57cec5SDimitry Andric 588*0b57cec5SDimitry Andric unsigned WebAssemblyFastISel::copyValue(unsigned Reg) { 589*0b57cec5SDimitry Andric unsigned ResultReg = createResultReg(MRI.getRegClass(Reg)); 590*0b57cec5SDimitry Andric BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc, TII.get(WebAssembly::COPY), 591*0b57cec5SDimitry Andric ResultReg) 592*0b57cec5SDimitry Andric .addReg(Reg); 593*0b57cec5SDimitry Andric return ResultReg; 594*0b57cec5SDimitry Andric } 595*0b57cec5SDimitry Andric 596*0b57cec5SDimitry Andric unsigned WebAssemblyFastISel::fastMaterializeAlloca(const AllocaInst *AI) { 597*0b57cec5SDimitry Andric DenseMap<const AllocaInst *, int>::iterator SI = 598*0b57cec5SDimitry Andric FuncInfo.StaticAllocaMap.find(AI); 599*0b57cec5SDimitry Andric 600*0b57cec5SDimitry Andric if (SI != FuncInfo.StaticAllocaMap.end()) { 601*0b57cec5SDimitry Andric unsigned ResultReg = 602*0b57cec5SDimitry Andric createResultReg(Subtarget->hasAddr64() ? &WebAssembly::I64RegClass 603*0b57cec5SDimitry Andric : &WebAssembly::I32RegClass); 604*0b57cec5SDimitry Andric unsigned Opc = 605*0b57cec5SDimitry Andric Subtarget->hasAddr64() ? WebAssembly::COPY_I64 : WebAssembly::COPY_I32; 606*0b57cec5SDimitry Andric BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc, TII.get(Opc), ResultReg) 607*0b57cec5SDimitry Andric .addFrameIndex(SI->second); 608*0b57cec5SDimitry Andric return ResultReg; 609*0b57cec5SDimitry Andric } 610*0b57cec5SDimitry Andric 611*0b57cec5SDimitry Andric return 0; 612*0b57cec5SDimitry Andric } 613*0b57cec5SDimitry Andric 614*0b57cec5SDimitry Andric unsigned WebAssemblyFastISel::fastMaterializeConstant(const Constant *C) { 615*0b57cec5SDimitry Andric if (const GlobalValue *GV = dyn_cast<GlobalValue>(C)) { 616*0b57cec5SDimitry Andric if (TLI.isPositionIndependent()) 617*0b57cec5SDimitry Andric return 0; 618*0b57cec5SDimitry Andric if (GV->isThreadLocal()) 619*0b57cec5SDimitry Andric return 0; 620*0b57cec5SDimitry Andric unsigned ResultReg = 621*0b57cec5SDimitry Andric createResultReg(Subtarget->hasAddr64() ? &WebAssembly::I64RegClass 622*0b57cec5SDimitry Andric : &WebAssembly::I32RegClass); 623*0b57cec5SDimitry Andric unsigned Opc = Subtarget->hasAddr64() ? WebAssembly::CONST_I64 624*0b57cec5SDimitry Andric : WebAssembly::CONST_I32; 625*0b57cec5SDimitry Andric BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc, TII.get(Opc), ResultReg) 626*0b57cec5SDimitry Andric .addGlobalAddress(GV); 627*0b57cec5SDimitry Andric return ResultReg; 628*0b57cec5SDimitry Andric } 629*0b57cec5SDimitry Andric 630*0b57cec5SDimitry Andric // Let target-independent code handle it. 631*0b57cec5SDimitry Andric return 0; 632*0b57cec5SDimitry Andric } 633*0b57cec5SDimitry Andric 634*0b57cec5SDimitry Andric bool WebAssemblyFastISel::fastLowerArguments() { 635*0b57cec5SDimitry Andric if (!FuncInfo.CanLowerReturn) 636*0b57cec5SDimitry Andric return false; 637*0b57cec5SDimitry Andric 638*0b57cec5SDimitry Andric const Function *F = FuncInfo.Fn; 639*0b57cec5SDimitry Andric if (F->isVarArg()) 640*0b57cec5SDimitry Andric return false; 641*0b57cec5SDimitry Andric 6425ffd83dbSDimitry Andric if (FuncInfo.Fn->getCallingConv() == CallingConv::Swift) 6435ffd83dbSDimitry Andric return false; 6445ffd83dbSDimitry Andric 645*0b57cec5SDimitry Andric unsigned I = 0; 646*0b57cec5SDimitry Andric for (auto const &Arg : F->args()) { 647*0b57cec5SDimitry Andric const AttributeList &Attrs = F->getAttributes(); 648349cc55cSDimitry Andric if (Attrs.hasParamAttr(I, Attribute::ByVal) || 649349cc55cSDimitry Andric Attrs.hasParamAttr(I, Attribute::SwiftSelf) || 650349cc55cSDimitry Andric Attrs.hasParamAttr(I, Attribute::SwiftError) || 651349cc55cSDimitry Andric Attrs.hasParamAttr(I, Attribute::InAlloca) || 652349cc55cSDimitry Andric Attrs.hasParamAttr(I, Attribute::Nest)) 653*0b57cec5SDimitry Andric return false; 654*0b57cec5SDimitry Andric 655*0b57cec5SDimitry Andric Type *ArgTy = Arg.getType(); 656*0b57cec5SDimitry Andric if (ArgTy->isStructTy() || ArgTy->isArrayTy()) 657*0b57cec5SDimitry Andric return false; 658*0b57cec5SDimitry Andric if (!Subtarget->hasSIMD128() && ArgTy->isVectorTy()) 659*0b57cec5SDimitry Andric return false; 660*0b57cec5SDimitry Andric 661*0b57cec5SDimitry Andric unsigned Opc; 662*0b57cec5SDimitry Andric const TargetRegisterClass *RC; 663*0b57cec5SDimitry Andric switch (getSimpleType(ArgTy)) { 664*0b57cec5SDimitry Andric case MVT::i1: 665*0b57cec5SDimitry Andric case MVT::i8: 666*0b57cec5SDimitry Andric case MVT::i16: 667*0b57cec5SDimitry Andric case MVT::i32: 668*0b57cec5SDimitry Andric Opc = WebAssembly::ARGUMENT_i32; 669*0b57cec5SDimitry Andric RC = &WebAssembly::I32RegClass; 670*0b57cec5SDimitry Andric break; 671*0b57cec5SDimitry Andric case MVT::i64: 672*0b57cec5SDimitry Andric Opc = WebAssembly::ARGUMENT_i64; 673*0b57cec5SDimitry Andric RC = &WebAssembly::I64RegClass; 674*0b57cec5SDimitry Andric break; 675*0b57cec5SDimitry Andric case MVT::f32: 676*0b57cec5SDimitry Andric Opc = WebAssembly::ARGUMENT_f32; 677*0b57cec5SDimitry Andric RC = &WebAssembly::F32RegClass; 678*0b57cec5SDimitry Andric break; 679*0b57cec5SDimitry Andric case MVT::f64: 680*0b57cec5SDimitry Andric Opc = WebAssembly::ARGUMENT_f64; 681*0b57cec5SDimitry Andric RC = &WebAssembly::F64RegClass; 682*0b57cec5SDimitry Andric break; 683*0b57cec5SDimitry Andric case MVT::v16i8: 684*0b57cec5SDimitry Andric Opc = WebAssembly::ARGUMENT_v16i8; 685*0b57cec5SDimitry Andric RC = &WebAssembly::V128RegClass; 686*0b57cec5SDimitry Andric break; 687*0b57cec5SDimitry Andric case MVT::v8i16: 688*0b57cec5SDimitry Andric Opc = WebAssembly::ARGUMENT_v8i16; 689*0b57cec5SDimitry Andric RC = &WebAssembly::V128RegClass; 690*0b57cec5SDimitry Andric break; 691*0b57cec5SDimitry Andric case MVT::v4i32: 692*0b57cec5SDimitry Andric Opc = WebAssembly::ARGUMENT_v4i32; 693*0b57cec5SDimitry Andric RC = &WebAssembly::V128RegClass; 694*0b57cec5SDimitry Andric break; 695*0b57cec5SDimitry Andric case MVT::v2i64: 696*0b57cec5SDimitry Andric Opc = WebAssembly::ARGUMENT_v2i64; 697*0b57cec5SDimitry Andric RC = &WebAssembly::V128RegClass; 698*0b57cec5SDimitry Andric break; 699*0b57cec5SDimitry Andric case MVT::v4f32: 700*0b57cec5SDimitry Andric Opc = WebAssembly::ARGUMENT_v4f32; 701*0b57cec5SDimitry Andric RC = &WebAssembly::V128RegClass; 702*0b57cec5SDimitry Andric break; 703*0b57cec5SDimitry Andric case MVT::v2f64: 704*0b57cec5SDimitry Andric Opc = WebAssembly::ARGUMENT_v2f64; 705*0b57cec5SDimitry Andric RC = &WebAssembly::V128RegClass; 706*0b57cec5SDimitry Andric break; 707e8d8bef9SDimitry Andric case MVT::funcref: 708e8d8bef9SDimitry Andric Opc = WebAssembly::ARGUMENT_funcref; 709e8d8bef9SDimitry Andric RC = &WebAssembly::FUNCREFRegClass; 710e8d8bef9SDimitry Andric break; 711e8d8bef9SDimitry Andric case MVT::externref: 712e8d8bef9SDimitry Andric Opc = WebAssembly::ARGUMENT_externref; 713e8d8bef9SDimitry Andric RC = &WebAssembly::EXTERNREFRegClass; 714*0b57cec5SDimitry Andric break; 715*0b57cec5SDimitry Andric default: 716*0b57cec5SDimitry Andric return false; 717*0b57cec5SDimitry Andric } 718*0b57cec5SDimitry Andric unsigned ResultReg = createResultReg(RC); 719*0b57cec5SDimitry Andric BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc, TII.get(Opc), ResultReg) 720*0b57cec5SDimitry Andric .addImm(I); 721*0b57cec5SDimitry Andric updateValueMap(&Arg, ResultReg); 722*0b57cec5SDimitry Andric 723*0b57cec5SDimitry Andric ++I; 724*0b57cec5SDimitry Andric } 725*0b57cec5SDimitry Andric 726*0b57cec5SDimitry Andric MRI.addLiveIn(WebAssembly::ARGUMENTS); 727*0b57cec5SDimitry Andric 728*0b57cec5SDimitry Andric auto *MFI = MF->getInfo<WebAssemblyFunctionInfo>(); 729*0b57cec5SDimitry Andric for (auto const &Arg : F->args()) { 730*0b57cec5SDimitry Andric MVT::SimpleValueType ArgTy = getLegalType(getSimpleType(Arg.getType())); 731*0b57cec5SDimitry Andric if (ArgTy == MVT::INVALID_SIMPLE_VALUE_TYPE) { 732*0b57cec5SDimitry Andric MFI->clearParamsAndResults(); 733*0b57cec5SDimitry Andric return false; 734*0b57cec5SDimitry Andric } 735*0b57cec5SDimitry Andric MFI->addParam(ArgTy); 736*0b57cec5SDimitry Andric } 737*0b57cec5SDimitry Andric 738*0b57cec5SDimitry Andric if (!F->getReturnType()->isVoidTy()) { 739*0b57cec5SDimitry Andric MVT::SimpleValueType RetTy = 740*0b57cec5SDimitry Andric getLegalType(getSimpleType(F->getReturnType())); 741*0b57cec5SDimitry Andric if (RetTy == MVT::INVALID_SIMPLE_VALUE_TYPE) { 742*0b57cec5SDimitry Andric MFI->clearParamsAndResults(); 743*0b57cec5SDimitry Andric return false; 744*0b57cec5SDimitry Andric } 745*0b57cec5SDimitry Andric MFI->addResult(RetTy); 746*0b57cec5SDimitry Andric } 747*0b57cec5SDimitry Andric 748*0b57cec5SDimitry Andric return true; 749*0b57cec5SDimitry Andric } 750*0b57cec5SDimitry Andric 751*0b57cec5SDimitry Andric bool WebAssemblyFastISel::selectCall(const Instruction *I) { 752*0b57cec5SDimitry Andric const auto *Call = cast<CallInst>(I); 753*0b57cec5SDimitry Andric 754*0b57cec5SDimitry Andric // TODO: Support tail calls in FastISel 755*0b57cec5SDimitry Andric if (Call->isMustTailCall() || Call->isInlineAsm() || 756*0b57cec5SDimitry Andric Call->getFunctionType()->isVarArg()) 757*0b57cec5SDimitry Andric return false; 758*0b57cec5SDimitry Andric 759*0b57cec5SDimitry Andric Function *Func = Call->getCalledFunction(); 760*0b57cec5SDimitry Andric if (Func && Func->isIntrinsic()) 761*0b57cec5SDimitry Andric return false; 762*0b57cec5SDimitry Andric 7635ffd83dbSDimitry Andric if (Call->getCallingConv() == CallingConv::Swift) 7645ffd83dbSDimitry Andric return false; 7655ffd83dbSDimitry Andric 766*0b57cec5SDimitry Andric bool IsDirect = Func != nullptr; 7675ffd83dbSDimitry Andric if (!IsDirect && isa<ConstantExpr>(Call->getCalledOperand())) 768*0b57cec5SDimitry Andric return false; 769*0b57cec5SDimitry Andric 770*0b57cec5SDimitry Andric FunctionType *FuncTy = Call->getFunctionType(); 7715ffd83dbSDimitry Andric unsigned Opc = IsDirect ? WebAssembly::CALL : WebAssembly::CALL_INDIRECT; 772*0b57cec5SDimitry Andric bool IsVoid = FuncTy->getReturnType()->isVoidTy(); 773*0b57cec5SDimitry Andric unsigned ResultReg; 7745ffd83dbSDimitry Andric if (!IsVoid) { 775*0b57cec5SDimitry Andric if (!Subtarget->hasSIMD128() && Call->getType()->isVectorTy()) 776*0b57cec5SDimitry Andric return false; 777*0b57cec5SDimitry Andric 778*0b57cec5SDimitry Andric MVT::SimpleValueType RetTy = getSimpleType(Call->getType()); 779*0b57cec5SDimitry Andric switch (RetTy) { 780*0b57cec5SDimitry Andric case MVT::i1: 781*0b57cec5SDimitry Andric case MVT::i8: 782*0b57cec5SDimitry Andric case MVT::i16: 783*0b57cec5SDimitry Andric case MVT::i32: 784*0b57cec5SDimitry Andric ResultReg = createResultReg(&WebAssembly::I32RegClass); 785*0b57cec5SDimitry Andric break; 786*0b57cec5SDimitry Andric case MVT::i64: 787*0b57cec5SDimitry Andric ResultReg = createResultReg(&WebAssembly::I64RegClass); 788*0b57cec5SDimitry Andric break; 789*0b57cec5SDimitry Andric case MVT::f32: 790*0b57cec5SDimitry Andric ResultReg = createResultReg(&WebAssembly::F32RegClass); 791*0b57cec5SDimitry Andric break; 792*0b57cec5SDimitry Andric case MVT::f64: 793*0b57cec5SDimitry Andric ResultReg = createResultReg(&WebAssembly::F64RegClass); 794*0b57cec5SDimitry Andric break; 795*0b57cec5SDimitry Andric case MVT::v16i8: 796*0b57cec5SDimitry Andric ResultReg = createResultReg(&WebAssembly::V128RegClass); 797*0b57cec5SDimitry Andric break; 798*0b57cec5SDimitry Andric case MVT::v8i16: 799*0b57cec5SDimitry Andric ResultReg = createResultReg(&WebAssembly::V128RegClass); 800*0b57cec5SDimitry Andric break; 801*0b57cec5SDimitry Andric case MVT::v4i32: 802*0b57cec5SDimitry Andric ResultReg = createResultReg(&WebAssembly::V128RegClass); 803*0b57cec5SDimitry Andric break; 804*0b57cec5SDimitry Andric case MVT::v2i64: 805*0b57cec5SDimitry Andric ResultReg = createResultReg(&WebAssembly::V128RegClass); 806*0b57cec5SDimitry Andric break; 807*0b57cec5SDimitry Andric case MVT::v4f32: 808*0b57cec5SDimitry Andric ResultReg = createResultReg(&WebAssembly::V128RegClass); 809*0b57cec5SDimitry Andric break; 810*0b57cec5SDimitry Andric case MVT::v2f64: 811*0b57cec5SDimitry Andric ResultReg = createResultReg(&WebAssembly::V128RegClass); 812*0b57cec5SDimitry Andric break; 813e8d8bef9SDimitry Andric case MVT::funcref: 814e8d8bef9SDimitry Andric ResultReg = createResultReg(&WebAssembly::FUNCREFRegClass); 815e8d8bef9SDimitry Andric break; 816e8d8bef9SDimitry Andric case MVT::externref: 817e8d8bef9SDimitry Andric ResultReg = createResultReg(&WebAssembly::EXTERNREFRegClass); 818*0b57cec5SDimitry Andric break; 819*0b57cec5SDimitry Andric default: 820*0b57cec5SDimitry Andric return false; 821*0b57cec5SDimitry Andric } 822*0b57cec5SDimitry Andric } 823*0b57cec5SDimitry Andric 824*0b57cec5SDimitry Andric SmallVector<unsigned, 8> Args; 825349cc55cSDimitry Andric for (unsigned I = 0, E = Call->arg_size(); I < E; ++I) { 826*0b57cec5SDimitry Andric Value *V = Call->getArgOperand(I); 827*0b57cec5SDimitry Andric MVT::SimpleValueType ArgTy = getSimpleType(V->getType()); 828*0b57cec5SDimitry Andric if (ArgTy == MVT::INVALID_SIMPLE_VALUE_TYPE) 829*0b57cec5SDimitry Andric return false; 830*0b57cec5SDimitry Andric 831*0b57cec5SDimitry Andric const AttributeList &Attrs = Call->getAttributes(); 832349cc55cSDimitry Andric if (Attrs.hasParamAttr(I, Attribute::ByVal) || 833349cc55cSDimitry Andric Attrs.hasParamAttr(I, Attribute::SwiftSelf) || 834349cc55cSDimitry Andric Attrs.hasParamAttr(I, Attribute::SwiftError) || 835349cc55cSDimitry Andric Attrs.hasParamAttr(I, Attribute::InAlloca) || 836349cc55cSDimitry Andric Attrs.hasParamAttr(I, Attribute::Nest)) 837*0b57cec5SDimitry Andric return false; 838*0b57cec5SDimitry Andric 839*0b57cec5SDimitry Andric unsigned Reg; 840*0b57cec5SDimitry Andric 841349cc55cSDimitry Andric if (Attrs.hasParamAttr(I, Attribute::SExt)) 842*0b57cec5SDimitry Andric Reg = getRegForSignedValue(V); 843349cc55cSDimitry Andric else if (Attrs.hasParamAttr(I, Attribute::ZExt)) 844*0b57cec5SDimitry Andric Reg = getRegForUnsignedValue(V); 845*0b57cec5SDimitry Andric else 846*0b57cec5SDimitry Andric Reg = getRegForValue(V); 847*0b57cec5SDimitry Andric 848*0b57cec5SDimitry Andric if (Reg == 0) 849*0b57cec5SDimitry Andric return false; 850*0b57cec5SDimitry Andric 851*0b57cec5SDimitry Andric Args.push_back(Reg); 852*0b57cec5SDimitry Andric } 853*0b57cec5SDimitry Andric 854*0b57cec5SDimitry Andric unsigned CalleeReg = 0; 855*0b57cec5SDimitry Andric if (!IsDirect) { 8565ffd83dbSDimitry Andric CalleeReg = getRegForValue(Call->getCalledOperand()); 857*0b57cec5SDimitry Andric if (!CalleeReg) 858*0b57cec5SDimitry Andric return false; 859*0b57cec5SDimitry Andric } 860*0b57cec5SDimitry Andric 861*0b57cec5SDimitry Andric auto MIB = BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc, TII.get(Opc)); 862*0b57cec5SDimitry Andric 863*0b57cec5SDimitry Andric if (!IsVoid) 864*0b57cec5SDimitry Andric MIB.addReg(ResultReg, RegState::Define); 865*0b57cec5SDimitry Andric 8665ffd83dbSDimitry Andric if (IsDirect) { 867*0b57cec5SDimitry Andric MIB.addGlobalAddress(Func); 8685ffd83dbSDimitry Andric } else { 869fe6060f1SDimitry Andric // Placeholder for the type index. 8705ffd83dbSDimitry Andric MIB.addImm(0); 871fe6060f1SDimitry Andric // The table into which this call_indirect indexes. 872fe6060f1SDimitry Andric MCSymbolWasm *Table = WebAssembly::getOrCreateFunctionTableSymbol( 873fe6060f1SDimitry Andric MF->getMMI().getContext(), Subtarget); 874fe6060f1SDimitry Andric if (Subtarget->hasReferenceTypes()) { 875fe6060f1SDimitry Andric MIB.addSym(Table); 876fe6060f1SDimitry Andric } else { 877fe6060f1SDimitry Andric // Otherwise for the MVP there is at most one table whose number is 0, but 878fe6060f1SDimitry Andric // we can't write a table symbol or issue relocations. Instead we just 879fe6060f1SDimitry Andric // ensure the table is live. 880fe6060f1SDimitry Andric Table->setNoStrip(); 8815ffd83dbSDimitry Andric MIB.addImm(0); 882fe6060f1SDimitry Andric } 883fe6060f1SDimitry Andric // See if we must truncate the function pointer. 884fe6060f1SDimitry Andric // CALL_INDIRECT takes an i32, but in wasm64 we represent function pointers 885fe6060f1SDimitry Andric // as 64-bit for uniformity with other pointer types. 886fe6060f1SDimitry Andric // See also: WebAssemblyISelLowering.cpp: LowerCallResults 887fe6060f1SDimitry Andric if (Subtarget->hasAddr64()) { 888fe6060f1SDimitry Andric auto Wrap = BuildMI(*FuncInfo.MBB, std::prev(FuncInfo.InsertPt), DbgLoc, 889fe6060f1SDimitry Andric TII.get(WebAssembly::I32_WRAP_I64)); 890fe6060f1SDimitry Andric unsigned Reg32 = createResultReg(&WebAssembly::I32RegClass); 891fe6060f1SDimitry Andric Wrap.addReg(Reg32, RegState::Define); 892fe6060f1SDimitry Andric Wrap.addReg(CalleeReg); 893fe6060f1SDimitry Andric CalleeReg = Reg32; 894fe6060f1SDimitry Andric } 8955ffd83dbSDimitry Andric } 896*0b57cec5SDimitry Andric 897*0b57cec5SDimitry Andric for (unsigned ArgReg : Args) 898*0b57cec5SDimitry Andric MIB.addReg(ArgReg); 899*0b57cec5SDimitry Andric 9005ffd83dbSDimitry Andric if (!IsDirect) 9015ffd83dbSDimitry Andric MIB.addReg(CalleeReg); 9025ffd83dbSDimitry Andric 903*0b57cec5SDimitry Andric if (!IsVoid) 904*0b57cec5SDimitry Andric updateValueMap(Call, ResultReg); 905*0b57cec5SDimitry Andric return true; 906*0b57cec5SDimitry Andric } 907*0b57cec5SDimitry Andric 908*0b57cec5SDimitry Andric bool WebAssemblyFastISel::selectSelect(const Instruction *I) { 909*0b57cec5SDimitry Andric const auto *Select = cast<SelectInst>(I); 910*0b57cec5SDimitry Andric 911*0b57cec5SDimitry Andric bool Not; 91269ade1e0SDimitry Andric unsigned CondReg = 91369ade1e0SDimitry Andric getRegForI1Value(Select->getCondition(), I->getParent(), Not); 914*0b57cec5SDimitry Andric if (CondReg == 0) 915*0b57cec5SDimitry Andric return false; 916*0b57cec5SDimitry Andric 917*0b57cec5SDimitry Andric unsigned TrueReg = getRegForValue(Select->getTrueValue()); 918*0b57cec5SDimitry Andric if (TrueReg == 0) 919*0b57cec5SDimitry Andric return false; 920*0b57cec5SDimitry Andric 921*0b57cec5SDimitry Andric unsigned FalseReg = getRegForValue(Select->getFalseValue()); 922*0b57cec5SDimitry Andric if (FalseReg == 0) 923*0b57cec5SDimitry Andric return false; 924*0b57cec5SDimitry Andric 925*0b57cec5SDimitry Andric if (Not) 926*0b57cec5SDimitry Andric std::swap(TrueReg, FalseReg); 927*0b57cec5SDimitry Andric 928*0b57cec5SDimitry Andric unsigned Opc; 929*0b57cec5SDimitry Andric const TargetRegisterClass *RC; 930*0b57cec5SDimitry Andric switch (getSimpleType(Select->getType())) { 931*0b57cec5SDimitry Andric case MVT::i1: 932*0b57cec5SDimitry Andric case MVT::i8: 933*0b57cec5SDimitry Andric case MVT::i16: 934*0b57cec5SDimitry Andric case MVT::i32: 935*0b57cec5SDimitry Andric Opc = WebAssembly::SELECT_I32; 936*0b57cec5SDimitry Andric RC = &WebAssembly::I32RegClass; 937*0b57cec5SDimitry Andric break; 938*0b57cec5SDimitry Andric case MVT::i64: 939*0b57cec5SDimitry Andric Opc = WebAssembly::SELECT_I64; 940*0b57cec5SDimitry Andric RC = &WebAssembly::I64RegClass; 941*0b57cec5SDimitry Andric break; 942*0b57cec5SDimitry Andric case MVT::f32: 943*0b57cec5SDimitry Andric Opc = WebAssembly::SELECT_F32; 944*0b57cec5SDimitry Andric RC = &WebAssembly::F32RegClass; 945*0b57cec5SDimitry Andric break; 946*0b57cec5SDimitry Andric case MVT::f64: 947*0b57cec5SDimitry Andric Opc = WebAssembly::SELECT_F64; 948*0b57cec5SDimitry Andric RC = &WebAssembly::F64RegClass; 949*0b57cec5SDimitry Andric break; 950e8d8bef9SDimitry Andric case MVT::funcref: 951e8d8bef9SDimitry Andric Opc = WebAssembly::SELECT_FUNCREF; 952e8d8bef9SDimitry Andric RC = &WebAssembly::FUNCREFRegClass; 953e8d8bef9SDimitry Andric break; 954e8d8bef9SDimitry Andric case MVT::externref: 955e8d8bef9SDimitry Andric Opc = WebAssembly::SELECT_EXTERNREF; 956e8d8bef9SDimitry Andric RC = &WebAssembly::EXTERNREFRegClass; 957*0b57cec5SDimitry Andric break; 958*0b57cec5SDimitry Andric default: 959*0b57cec5SDimitry Andric return false; 960*0b57cec5SDimitry Andric } 961*0b57cec5SDimitry Andric 962*0b57cec5SDimitry Andric unsigned ResultReg = createResultReg(RC); 963*0b57cec5SDimitry Andric BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc, TII.get(Opc), ResultReg) 964*0b57cec5SDimitry Andric .addReg(TrueReg) 965*0b57cec5SDimitry Andric .addReg(FalseReg) 966*0b57cec5SDimitry Andric .addReg(CondReg); 967*0b57cec5SDimitry Andric 968*0b57cec5SDimitry Andric updateValueMap(Select, ResultReg); 969*0b57cec5SDimitry Andric return true; 970*0b57cec5SDimitry Andric } 971*0b57cec5SDimitry Andric 972*0b57cec5SDimitry Andric bool WebAssemblyFastISel::selectTrunc(const Instruction *I) { 973*0b57cec5SDimitry Andric const auto *Trunc = cast<TruncInst>(I); 974*0b57cec5SDimitry Andric 975*0b57cec5SDimitry Andric unsigned Reg = getRegForValue(Trunc->getOperand(0)); 976*0b57cec5SDimitry Andric if (Reg == 0) 977*0b57cec5SDimitry Andric return false; 978*0b57cec5SDimitry Andric 979*0b57cec5SDimitry Andric if (Trunc->getOperand(0)->getType()->isIntegerTy(64)) { 980*0b57cec5SDimitry Andric unsigned Result = createResultReg(&WebAssembly::I32RegClass); 981*0b57cec5SDimitry Andric BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc, 982*0b57cec5SDimitry Andric TII.get(WebAssembly::I32_WRAP_I64), Result) 983*0b57cec5SDimitry Andric .addReg(Reg); 984*0b57cec5SDimitry Andric Reg = Result; 985*0b57cec5SDimitry Andric } 986*0b57cec5SDimitry Andric 987*0b57cec5SDimitry Andric updateValueMap(Trunc, Reg); 988*0b57cec5SDimitry Andric return true; 989*0b57cec5SDimitry Andric } 990*0b57cec5SDimitry Andric 991*0b57cec5SDimitry Andric bool WebAssemblyFastISel::selectZExt(const Instruction *I) { 992*0b57cec5SDimitry Andric const auto *ZExt = cast<ZExtInst>(I); 993*0b57cec5SDimitry Andric 994*0b57cec5SDimitry Andric const Value *Op = ZExt->getOperand(0); 995*0b57cec5SDimitry Andric MVT::SimpleValueType From = getSimpleType(Op->getType()); 996*0b57cec5SDimitry Andric MVT::SimpleValueType To = getLegalType(getSimpleType(ZExt->getType())); 997*0b57cec5SDimitry Andric unsigned In = getRegForValue(Op); 998*0b57cec5SDimitry Andric if (In == 0) 999*0b57cec5SDimitry Andric return false; 1000*0b57cec5SDimitry Andric unsigned Reg = zeroExtend(In, Op, From, To); 1001*0b57cec5SDimitry Andric if (Reg == 0) 1002*0b57cec5SDimitry Andric return false; 1003*0b57cec5SDimitry Andric 1004*0b57cec5SDimitry Andric updateValueMap(ZExt, Reg); 1005*0b57cec5SDimitry Andric return true; 1006*0b57cec5SDimitry Andric } 1007*0b57cec5SDimitry Andric 1008*0b57cec5SDimitry Andric bool WebAssemblyFastISel::selectSExt(const Instruction *I) { 1009*0b57cec5SDimitry Andric const auto *SExt = cast<SExtInst>(I); 1010*0b57cec5SDimitry Andric 1011*0b57cec5SDimitry Andric const Value *Op = SExt->getOperand(0); 1012*0b57cec5SDimitry Andric MVT::SimpleValueType From = getSimpleType(Op->getType()); 1013*0b57cec5SDimitry Andric MVT::SimpleValueType To = getLegalType(getSimpleType(SExt->getType())); 1014*0b57cec5SDimitry Andric unsigned In = getRegForValue(Op); 1015*0b57cec5SDimitry Andric if (In == 0) 1016*0b57cec5SDimitry Andric return false; 1017*0b57cec5SDimitry Andric unsigned Reg = signExtend(In, Op, From, To); 1018*0b57cec5SDimitry Andric if (Reg == 0) 1019*0b57cec5SDimitry Andric return false; 1020*0b57cec5SDimitry Andric 1021*0b57cec5SDimitry Andric updateValueMap(SExt, Reg); 1022*0b57cec5SDimitry Andric return true; 1023*0b57cec5SDimitry Andric } 1024*0b57cec5SDimitry Andric 1025*0b57cec5SDimitry Andric bool WebAssemblyFastISel::selectICmp(const Instruction *I) { 1026*0b57cec5SDimitry Andric const auto *ICmp = cast<ICmpInst>(I); 1027*0b57cec5SDimitry Andric 1028*0b57cec5SDimitry Andric bool I32 = getSimpleType(ICmp->getOperand(0)->getType()) != MVT::i64; 1029*0b57cec5SDimitry Andric unsigned Opc; 1030*0b57cec5SDimitry Andric bool IsSigned = false; 1031*0b57cec5SDimitry Andric switch (ICmp->getPredicate()) { 1032*0b57cec5SDimitry Andric case ICmpInst::ICMP_EQ: 1033*0b57cec5SDimitry Andric Opc = I32 ? WebAssembly::EQ_I32 : WebAssembly::EQ_I64; 1034*0b57cec5SDimitry Andric break; 1035*0b57cec5SDimitry Andric case ICmpInst::ICMP_NE: 1036*0b57cec5SDimitry Andric Opc = I32 ? WebAssembly::NE_I32 : WebAssembly::NE_I64; 1037*0b57cec5SDimitry Andric break; 1038*0b57cec5SDimitry Andric case ICmpInst::ICMP_UGT: 1039*0b57cec5SDimitry Andric Opc = I32 ? WebAssembly::GT_U_I32 : WebAssembly::GT_U_I64; 1040*0b57cec5SDimitry Andric break; 1041*0b57cec5SDimitry Andric case ICmpInst::ICMP_UGE: 1042*0b57cec5SDimitry Andric Opc = I32 ? WebAssembly::GE_U_I32 : WebAssembly::GE_U_I64; 1043*0b57cec5SDimitry Andric break; 1044*0b57cec5SDimitry Andric case ICmpInst::ICMP_ULT: 1045*0b57cec5SDimitry Andric Opc = I32 ? WebAssembly::LT_U_I32 : WebAssembly::LT_U_I64; 1046*0b57cec5SDimitry Andric break; 1047*0b57cec5SDimitry Andric case ICmpInst::ICMP_ULE: 1048*0b57cec5SDimitry Andric Opc = I32 ? WebAssembly::LE_U_I32 : WebAssembly::LE_U_I64; 1049*0b57cec5SDimitry Andric break; 1050*0b57cec5SDimitry Andric case ICmpInst::ICMP_SGT: 1051*0b57cec5SDimitry Andric Opc = I32 ? WebAssembly::GT_S_I32 : WebAssembly::GT_S_I64; 1052*0b57cec5SDimitry Andric IsSigned = true; 1053*0b57cec5SDimitry Andric break; 1054*0b57cec5SDimitry Andric case ICmpInst::ICMP_SGE: 1055*0b57cec5SDimitry Andric Opc = I32 ? WebAssembly::GE_S_I32 : WebAssembly::GE_S_I64; 1056*0b57cec5SDimitry Andric IsSigned = true; 1057*0b57cec5SDimitry Andric break; 1058*0b57cec5SDimitry Andric case ICmpInst::ICMP_SLT: 1059*0b57cec5SDimitry Andric Opc = I32 ? WebAssembly::LT_S_I32 : WebAssembly::LT_S_I64; 1060*0b57cec5SDimitry Andric IsSigned = true; 1061*0b57cec5SDimitry Andric break; 1062*0b57cec5SDimitry Andric case ICmpInst::ICMP_SLE: 1063*0b57cec5SDimitry Andric Opc = I32 ? WebAssembly::LE_S_I32 : WebAssembly::LE_S_I64; 1064*0b57cec5SDimitry Andric IsSigned = true; 1065*0b57cec5SDimitry Andric break; 1066*0b57cec5SDimitry Andric default: 1067*0b57cec5SDimitry Andric return false; 1068*0b57cec5SDimitry Andric } 1069*0b57cec5SDimitry Andric 1070*0b57cec5SDimitry Andric unsigned LHS = getRegForPromotedValue(ICmp->getOperand(0), IsSigned); 1071*0b57cec5SDimitry Andric if (LHS == 0) 1072*0b57cec5SDimitry Andric return false; 1073*0b57cec5SDimitry Andric 1074*0b57cec5SDimitry Andric unsigned RHS = getRegForPromotedValue(ICmp->getOperand(1), IsSigned); 1075*0b57cec5SDimitry Andric if (RHS == 0) 1076*0b57cec5SDimitry Andric return false; 1077*0b57cec5SDimitry Andric 1078*0b57cec5SDimitry Andric unsigned ResultReg = createResultReg(&WebAssembly::I32RegClass); 1079*0b57cec5SDimitry Andric BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc, TII.get(Opc), ResultReg) 1080*0b57cec5SDimitry Andric .addReg(LHS) 1081*0b57cec5SDimitry Andric .addReg(RHS); 1082*0b57cec5SDimitry Andric updateValueMap(ICmp, ResultReg); 1083*0b57cec5SDimitry Andric return true; 1084*0b57cec5SDimitry Andric } 1085*0b57cec5SDimitry Andric 1086*0b57cec5SDimitry Andric bool WebAssemblyFastISel::selectFCmp(const Instruction *I) { 1087*0b57cec5SDimitry Andric const auto *FCmp = cast<FCmpInst>(I); 1088*0b57cec5SDimitry Andric 1089*0b57cec5SDimitry Andric unsigned LHS = getRegForValue(FCmp->getOperand(0)); 1090*0b57cec5SDimitry Andric if (LHS == 0) 1091*0b57cec5SDimitry Andric return false; 1092*0b57cec5SDimitry Andric 1093*0b57cec5SDimitry Andric unsigned RHS = getRegForValue(FCmp->getOperand(1)); 1094*0b57cec5SDimitry Andric if (RHS == 0) 1095*0b57cec5SDimitry Andric return false; 1096*0b57cec5SDimitry Andric 1097*0b57cec5SDimitry Andric bool F32 = getSimpleType(FCmp->getOperand(0)->getType()) != MVT::f64; 1098*0b57cec5SDimitry Andric unsigned Opc; 1099*0b57cec5SDimitry Andric bool Not = false; 1100*0b57cec5SDimitry Andric switch (FCmp->getPredicate()) { 1101*0b57cec5SDimitry Andric case FCmpInst::FCMP_OEQ: 1102*0b57cec5SDimitry Andric Opc = F32 ? WebAssembly::EQ_F32 : WebAssembly::EQ_F64; 1103*0b57cec5SDimitry Andric break; 1104*0b57cec5SDimitry Andric case FCmpInst::FCMP_UNE: 1105*0b57cec5SDimitry Andric Opc = F32 ? WebAssembly::NE_F32 : WebAssembly::NE_F64; 1106*0b57cec5SDimitry Andric break; 1107*0b57cec5SDimitry Andric case FCmpInst::FCMP_OGT: 1108*0b57cec5SDimitry Andric Opc = F32 ? WebAssembly::GT_F32 : WebAssembly::GT_F64; 1109*0b57cec5SDimitry Andric break; 1110*0b57cec5SDimitry Andric case FCmpInst::FCMP_OGE: 1111*0b57cec5SDimitry Andric Opc = F32 ? WebAssembly::GE_F32 : WebAssembly::GE_F64; 1112*0b57cec5SDimitry Andric break; 1113*0b57cec5SDimitry Andric case FCmpInst::FCMP_OLT: 1114*0b57cec5SDimitry Andric Opc = F32 ? WebAssembly::LT_F32 : WebAssembly::LT_F64; 1115*0b57cec5SDimitry Andric break; 1116*0b57cec5SDimitry Andric case FCmpInst::FCMP_OLE: 1117*0b57cec5SDimitry Andric Opc = F32 ? WebAssembly::LE_F32 : WebAssembly::LE_F64; 1118*0b57cec5SDimitry Andric break; 1119*0b57cec5SDimitry Andric case FCmpInst::FCMP_UGT: 1120*0b57cec5SDimitry Andric Opc = F32 ? WebAssembly::LE_F32 : WebAssembly::LE_F64; 1121*0b57cec5SDimitry Andric Not = true; 1122*0b57cec5SDimitry Andric break; 1123*0b57cec5SDimitry Andric case FCmpInst::FCMP_UGE: 1124*0b57cec5SDimitry Andric Opc = F32 ? WebAssembly::LT_F32 : WebAssembly::LT_F64; 1125*0b57cec5SDimitry Andric Not = true; 1126*0b57cec5SDimitry Andric break; 1127*0b57cec5SDimitry Andric case FCmpInst::FCMP_ULT: 1128*0b57cec5SDimitry Andric Opc = F32 ? WebAssembly::GE_F32 : WebAssembly::GE_F64; 1129*0b57cec5SDimitry Andric Not = true; 1130*0b57cec5SDimitry Andric break; 1131*0b57cec5SDimitry Andric case FCmpInst::FCMP_ULE: 1132*0b57cec5SDimitry Andric Opc = F32 ? WebAssembly::GT_F32 : WebAssembly::GT_F64; 1133*0b57cec5SDimitry Andric Not = true; 1134*0b57cec5SDimitry Andric break; 1135*0b57cec5SDimitry Andric default: 1136*0b57cec5SDimitry Andric return false; 1137*0b57cec5SDimitry Andric } 1138*0b57cec5SDimitry Andric 1139*0b57cec5SDimitry Andric unsigned ResultReg = createResultReg(&WebAssembly::I32RegClass); 1140*0b57cec5SDimitry Andric BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc, TII.get(Opc), ResultReg) 1141*0b57cec5SDimitry Andric .addReg(LHS) 1142*0b57cec5SDimitry Andric .addReg(RHS); 1143*0b57cec5SDimitry Andric 1144*0b57cec5SDimitry Andric if (Not) 1145*0b57cec5SDimitry Andric ResultReg = notValue(ResultReg); 1146*0b57cec5SDimitry Andric 1147*0b57cec5SDimitry Andric updateValueMap(FCmp, ResultReg); 1148*0b57cec5SDimitry Andric return true; 1149*0b57cec5SDimitry Andric } 1150*0b57cec5SDimitry Andric 1151*0b57cec5SDimitry Andric bool WebAssemblyFastISel::selectBitCast(const Instruction *I) { 1152*0b57cec5SDimitry Andric // Target-independent code can handle this, except it doesn't set the dead 1153*0b57cec5SDimitry Andric // flag on the ARGUMENTS clobber, so we have to do that manually in order 1154*0b57cec5SDimitry Andric // to satisfy code that expects this of isBitcast() instructions. 1155*0b57cec5SDimitry Andric EVT VT = TLI.getValueType(DL, I->getOperand(0)->getType()); 1156*0b57cec5SDimitry Andric EVT RetVT = TLI.getValueType(DL, I->getType()); 1157*0b57cec5SDimitry Andric if (!VT.isSimple() || !RetVT.isSimple()) 1158*0b57cec5SDimitry Andric return false; 1159*0b57cec5SDimitry Andric 1160*0b57cec5SDimitry Andric unsigned In = getRegForValue(I->getOperand(0)); 1161*0b57cec5SDimitry Andric if (In == 0) 1162*0b57cec5SDimitry Andric return false; 1163*0b57cec5SDimitry Andric 1164*0b57cec5SDimitry Andric if (VT == RetVT) { 1165*0b57cec5SDimitry Andric // No-op bitcast. 1166*0b57cec5SDimitry Andric updateValueMap(I, In); 1167*0b57cec5SDimitry Andric return true; 1168*0b57cec5SDimitry Andric } 1169*0b57cec5SDimitry Andric 11708bcb0991SDimitry Andric Register Reg = fastEmit_ISD_BITCAST_r(VT.getSimpleVT(), RetVT.getSimpleVT(), 1171fe6060f1SDimitry Andric In); 1172*0b57cec5SDimitry Andric if (!Reg) 1173*0b57cec5SDimitry Andric return false; 1174*0b57cec5SDimitry Andric MachineBasicBlock::iterator Iter = FuncInfo.InsertPt; 1175*0b57cec5SDimitry Andric --Iter; 1176*0b57cec5SDimitry Andric assert(Iter->isBitcast()); 11778bcb0991SDimitry Andric Iter->setPhysRegsDeadExcept(ArrayRef<Register>(), TRI); 1178*0b57cec5SDimitry Andric updateValueMap(I, Reg); 1179*0b57cec5SDimitry Andric return true; 1180*0b57cec5SDimitry Andric } 1181*0b57cec5SDimitry Andric 1182*0b57cec5SDimitry Andric bool WebAssemblyFastISel::selectLoad(const Instruction *I) { 1183*0b57cec5SDimitry Andric const auto *Load = cast<LoadInst>(I); 1184*0b57cec5SDimitry Andric if (Load->isAtomic()) 1185*0b57cec5SDimitry Andric return false; 1186fe6060f1SDimitry Andric if (!WebAssembly::isDefaultAddressSpace(Load->getPointerAddressSpace())) 1187fe6060f1SDimitry Andric return false; 1188*0b57cec5SDimitry Andric if (!Subtarget->hasSIMD128() && Load->getType()->isVectorTy()) 1189*0b57cec5SDimitry Andric return false; 1190*0b57cec5SDimitry Andric 1191*0b57cec5SDimitry Andric Address Addr; 1192*0b57cec5SDimitry Andric if (!computeAddress(Load->getPointerOperand(), Addr)) 1193*0b57cec5SDimitry Andric return false; 1194*0b57cec5SDimitry Andric 1195*0b57cec5SDimitry Andric // TODO: Fold a following sign-/zero-extend into the load instruction. 1196*0b57cec5SDimitry Andric 1197*0b57cec5SDimitry Andric unsigned Opc; 1198*0b57cec5SDimitry Andric const TargetRegisterClass *RC; 11995ffd83dbSDimitry Andric bool A64 = Subtarget->hasAddr64(); 1200*0b57cec5SDimitry Andric switch (getSimpleType(Load->getType())) { 1201*0b57cec5SDimitry Andric case MVT::i1: 1202*0b57cec5SDimitry Andric case MVT::i8: 12035ffd83dbSDimitry Andric Opc = A64 ? WebAssembly::LOAD8_U_I32_A64 : WebAssembly::LOAD8_U_I32_A32; 1204*0b57cec5SDimitry Andric RC = &WebAssembly::I32RegClass; 1205*0b57cec5SDimitry Andric break; 1206*0b57cec5SDimitry Andric case MVT::i16: 12075ffd83dbSDimitry Andric Opc = A64 ? WebAssembly::LOAD16_U_I32_A64 : WebAssembly::LOAD16_U_I32_A32; 1208*0b57cec5SDimitry Andric RC = &WebAssembly::I32RegClass; 1209*0b57cec5SDimitry Andric break; 1210*0b57cec5SDimitry Andric case MVT::i32: 12115ffd83dbSDimitry Andric Opc = A64 ? WebAssembly::LOAD_I32_A64 : WebAssembly::LOAD_I32_A32; 1212*0b57cec5SDimitry Andric RC = &WebAssembly::I32RegClass; 1213*0b57cec5SDimitry Andric break; 1214*0b57cec5SDimitry Andric case MVT::i64: 12155ffd83dbSDimitry Andric Opc = A64 ? WebAssembly::LOAD_I64_A64 : WebAssembly::LOAD_I64_A32; 1216*0b57cec5SDimitry Andric RC = &WebAssembly::I64RegClass; 1217*0b57cec5SDimitry Andric break; 1218*0b57cec5SDimitry Andric case MVT::f32: 12195ffd83dbSDimitry Andric Opc = A64 ? WebAssembly::LOAD_F32_A64 : WebAssembly::LOAD_F32_A32; 1220*0b57cec5SDimitry Andric RC = &WebAssembly::F32RegClass; 1221*0b57cec5SDimitry Andric break; 1222*0b57cec5SDimitry Andric case MVT::f64: 12235ffd83dbSDimitry Andric Opc = A64 ? WebAssembly::LOAD_F64_A64 : WebAssembly::LOAD_F64_A32; 1224*0b57cec5SDimitry Andric RC = &WebAssembly::F64RegClass; 1225*0b57cec5SDimitry Andric break; 1226*0b57cec5SDimitry Andric default: 1227*0b57cec5SDimitry Andric return false; 1228*0b57cec5SDimitry Andric } 1229*0b57cec5SDimitry Andric 1230*0b57cec5SDimitry Andric materializeLoadStoreOperands(Addr); 1231*0b57cec5SDimitry Andric 1232*0b57cec5SDimitry Andric unsigned ResultReg = createResultReg(RC); 1233*0b57cec5SDimitry Andric auto MIB = BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc, TII.get(Opc), 1234*0b57cec5SDimitry Andric ResultReg); 1235*0b57cec5SDimitry Andric 1236*0b57cec5SDimitry Andric addLoadStoreOperands(Addr, MIB, createMachineMemOperandFor(Load)); 1237*0b57cec5SDimitry Andric 1238*0b57cec5SDimitry Andric updateValueMap(Load, ResultReg); 1239*0b57cec5SDimitry Andric return true; 1240*0b57cec5SDimitry Andric } 1241*0b57cec5SDimitry Andric 1242*0b57cec5SDimitry Andric bool WebAssemblyFastISel::selectStore(const Instruction *I) { 1243*0b57cec5SDimitry Andric const auto *Store = cast<StoreInst>(I); 1244*0b57cec5SDimitry Andric if (Store->isAtomic()) 1245*0b57cec5SDimitry Andric return false; 1246fe6060f1SDimitry Andric if (!WebAssembly::isDefaultAddressSpace(Store->getPointerAddressSpace())) 1247fe6060f1SDimitry Andric return false; 1248*0b57cec5SDimitry Andric if (!Subtarget->hasSIMD128() && 1249*0b57cec5SDimitry Andric Store->getValueOperand()->getType()->isVectorTy()) 1250*0b57cec5SDimitry Andric return false; 1251*0b57cec5SDimitry Andric 1252*0b57cec5SDimitry Andric Address Addr; 1253*0b57cec5SDimitry Andric if (!computeAddress(Store->getPointerOperand(), Addr)) 1254*0b57cec5SDimitry Andric return false; 1255*0b57cec5SDimitry Andric 1256*0b57cec5SDimitry Andric unsigned Opc; 1257*0b57cec5SDimitry Andric bool VTIsi1 = false; 12585ffd83dbSDimitry Andric bool A64 = Subtarget->hasAddr64(); 1259*0b57cec5SDimitry Andric switch (getSimpleType(Store->getValueOperand()->getType())) { 1260*0b57cec5SDimitry Andric case MVT::i1: 1261*0b57cec5SDimitry Andric VTIsi1 = true; 1262*0b57cec5SDimitry Andric LLVM_FALLTHROUGH; 1263*0b57cec5SDimitry Andric case MVT::i8: 12645ffd83dbSDimitry Andric Opc = A64 ? WebAssembly::STORE8_I32_A64 : WebAssembly::STORE8_I32_A32; 1265*0b57cec5SDimitry Andric break; 1266*0b57cec5SDimitry Andric case MVT::i16: 12675ffd83dbSDimitry Andric Opc = A64 ? WebAssembly::STORE16_I32_A64 : WebAssembly::STORE16_I32_A32; 1268*0b57cec5SDimitry Andric break; 1269*0b57cec5SDimitry Andric case MVT::i32: 12705ffd83dbSDimitry Andric Opc = A64 ? WebAssembly::STORE_I32_A64 : WebAssembly::STORE_I32_A32; 1271*0b57cec5SDimitry Andric break; 1272*0b57cec5SDimitry Andric case MVT::i64: 12735ffd83dbSDimitry Andric Opc = A64 ? WebAssembly::STORE_I64_A64 : WebAssembly::STORE_I64_A32; 1274*0b57cec5SDimitry Andric break; 1275*0b57cec5SDimitry Andric case MVT::f32: 12765ffd83dbSDimitry Andric Opc = A64 ? WebAssembly::STORE_F32_A64 : WebAssembly::STORE_F32_A32; 1277*0b57cec5SDimitry Andric break; 1278*0b57cec5SDimitry Andric case MVT::f64: 12795ffd83dbSDimitry Andric Opc = A64 ? WebAssembly::STORE_F64_A64 : WebAssembly::STORE_F64_A32; 1280*0b57cec5SDimitry Andric break; 1281*0b57cec5SDimitry Andric default: 1282*0b57cec5SDimitry Andric return false; 1283*0b57cec5SDimitry Andric } 1284*0b57cec5SDimitry Andric 1285*0b57cec5SDimitry Andric materializeLoadStoreOperands(Addr); 1286*0b57cec5SDimitry Andric 1287*0b57cec5SDimitry Andric unsigned ValueReg = getRegForValue(Store->getValueOperand()); 1288*0b57cec5SDimitry Andric if (ValueReg == 0) 1289*0b57cec5SDimitry Andric return false; 1290*0b57cec5SDimitry Andric if (VTIsi1) 1291*0b57cec5SDimitry Andric ValueReg = maskI1Value(ValueReg, Store->getValueOperand()); 1292*0b57cec5SDimitry Andric 1293*0b57cec5SDimitry Andric auto MIB = BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc, TII.get(Opc)); 1294*0b57cec5SDimitry Andric 1295*0b57cec5SDimitry Andric addLoadStoreOperands(Addr, MIB, createMachineMemOperandFor(Store)); 1296*0b57cec5SDimitry Andric 1297*0b57cec5SDimitry Andric MIB.addReg(ValueReg); 1298*0b57cec5SDimitry Andric return true; 1299*0b57cec5SDimitry Andric } 1300*0b57cec5SDimitry Andric 1301*0b57cec5SDimitry Andric bool WebAssemblyFastISel::selectBr(const Instruction *I) { 1302*0b57cec5SDimitry Andric const auto *Br = cast<BranchInst>(I); 1303*0b57cec5SDimitry Andric if (Br->isUnconditional()) { 1304*0b57cec5SDimitry Andric MachineBasicBlock *MSucc = FuncInfo.MBBMap[Br->getSuccessor(0)]; 1305*0b57cec5SDimitry Andric fastEmitBranch(MSucc, Br->getDebugLoc()); 1306*0b57cec5SDimitry Andric return true; 1307*0b57cec5SDimitry Andric } 1308*0b57cec5SDimitry Andric 1309*0b57cec5SDimitry Andric MachineBasicBlock *TBB = FuncInfo.MBBMap[Br->getSuccessor(0)]; 1310*0b57cec5SDimitry Andric MachineBasicBlock *FBB = FuncInfo.MBBMap[Br->getSuccessor(1)]; 1311*0b57cec5SDimitry Andric 1312*0b57cec5SDimitry Andric bool Not; 131369ade1e0SDimitry Andric unsigned CondReg = getRegForI1Value(Br->getCondition(), Br->getParent(), Not); 1314*0b57cec5SDimitry Andric if (CondReg == 0) 1315*0b57cec5SDimitry Andric return false; 1316*0b57cec5SDimitry Andric 1317*0b57cec5SDimitry Andric unsigned Opc = WebAssembly::BR_IF; 1318*0b57cec5SDimitry Andric if (Not) 1319*0b57cec5SDimitry Andric Opc = WebAssembly::BR_UNLESS; 1320*0b57cec5SDimitry Andric 1321*0b57cec5SDimitry Andric BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc, TII.get(Opc)) 1322*0b57cec5SDimitry Andric .addMBB(TBB) 1323*0b57cec5SDimitry Andric .addReg(CondReg); 1324*0b57cec5SDimitry Andric 1325*0b57cec5SDimitry Andric finishCondBranch(Br->getParent(), TBB, FBB); 1326*0b57cec5SDimitry Andric return true; 1327*0b57cec5SDimitry Andric } 1328*0b57cec5SDimitry Andric 1329*0b57cec5SDimitry Andric bool WebAssemblyFastISel::selectRet(const Instruction *I) { 1330*0b57cec5SDimitry Andric if (!FuncInfo.CanLowerReturn) 1331*0b57cec5SDimitry Andric return false; 1332*0b57cec5SDimitry Andric 1333*0b57cec5SDimitry Andric const auto *Ret = cast<ReturnInst>(I); 1334*0b57cec5SDimitry Andric 1335*0b57cec5SDimitry Andric if (Ret->getNumOperands() == 0) { 1336*0b57cec5SDimitry Andric BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc, 13378bcb0991SDimitry Andric TII.get(WebAssembly::RETURN)); 1338*0b57cec5SDimitry Andric return true; 1339*0b57cec5SDimitry Andric } 1340*0b57cec5SDimitry Andric 13418bcb0991SDimitry Andric // TODO: support multiple return in FastISel 13428bcb0991SDimitry Andric if (Ret->getNumOperands() > 1) 13438bcb0991SDimitry Andric return false; 13448bcb0991SDimitry Andric 1345*0b57cec5SDimitry Andric Value *RV = Ret->getOperand(0); 1346*0b57cec5SDimitry Andric if (!Subtarget->hasSIMD128() && RV->getType()->isVectorTy()) 1347*0b57cec5SDimitry Andric return false; 1348*0b57cec5SDimitry Andric 1349*0b57cec5SDimitry Andric switch (getSimpleType(RV->getType())) { 1350*0b57cec5SDimitry Andric case MVT::i1: 1351*0b57cec5SDimitry Andric case MVT::i8: 1352*0b57cec5SDimitry Andric case MVT::i16: 1353*0b57cec5SDimitry Andric case MVT::i32: 1354*0b57cec5SDimitry Andric case MVT::i64: 1355*0b57cec5SDimitry Andric case MVT::f32: 1356*0b57cec5SDimitry Andric case MVT::f64: 1357*0b57cec5SDimitry Andric case MVT::v16i8: 1358*0b57cec5SDimitry Andric case MVT::v8i16: 1359*0b57cec5SDimitry Andric case MVT::v4i32: 1360*0b57cec5SDimitry Andric case MVT::v2i64: 1361*0b57cec5SDimitry Andric case MVT::v4f32: 1362*0b57cec5SDimitry Andric case MVT::v2f64: 1363e8d8bef9SDimitry Andric case MVT::funcref: 1364e8d8bef9SDimitry Andric case MVT::externref: 1365*0b57cec5SDimitry Andric break; 1366*0b57cec5SDimitry Andric default: 1367*0b57cec5SDimitry Andric return false; 1368*0b57cec5SDimitry Andric } 1369*0b57cec5SDimitry Andric 1370*0b57cec5SDimitry Andric unsigned Reg; 1371349cc55cSDimitry Andric if (FuncInfo.Fn->getAttributes().hasRetAttr(Attribute::SExt)) 1372*0b57cec5SDimitry Andric Reg = getRegForSignedValue(RV); 1373349cc55cSDimitry Andric else if (FuncInfo.Fn->getAttributes().hasRetAttr(Attribute::ZExt)) 1374*0b57cec5SDimitry Andric Reg = getRegForUnsignedValue(RV); 1375*0b57cec5SDimitry Andric else 1376*0b57cec5SDimitry Andric Reg = getRegForValue(RV); 1377*0b57cec5SDimitry Andric 1378*0b57cec5SDimitry Andric if (Reg == 0) 1379*0b57cec5SDimitry Andric return false; 1380*0b57cec5SDimitry Andric 13818bcb0991SDimitry Andric BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc, 13828bcb0991SDimitry Andric TII.get(WebAssembly::RETURN)) 13838bcb0991SDimitry Andric .addReg(Reg); 1384*0b57cec5SDimitry Andric return true; 1385*0b57cec5SDimitry Andric } 1386*0b57cec5SDimitry Andric 1387*0b57cec5SDimitry Andric bool WebAssemblyFastISel::selectUnreachable(const Instruction *I) { 1388*0b57cec5SDimitry Andric BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc, 1389*0b57cec5SDimitry Andric TII.get(WebAssembly::UNREACHABLE)); 1390*0b57cec5SDimitry Andric return true; 1391*0b57cec5SDimitry Andric } 1392*0b57cec5SDimitry Andric 1393*0b57cec5SDimitry Andric bool WebAssemblyFastISel::fastSelectInstruction(const Instruction *I) { 1394*0b57cec5SDimitry Andric switch (I->getOpcode()) { 1395*0b57cec5SDimitry Andric case Instruction::Call: 1396*0b57cec5SDimitry Andric if (selectCall(I)) 1397*0b57cec5SDimitry Andric return true; 1398*0b57cec5SDimitry Andric break; 1399*0b57cec5SDimitry Andric case Instruction::Select: 1400*0b57cec5SDimitry Andric return selectSelect(I); 1401*0b57cec5SDimitry Andric case Instruction::Trunc: 1402*0b57cec5SDimitry Andric return selectTrunc(I); 1403*0b57cec5SDimitry Andric case Instruction::ZExt: 1404*0b57cec5SDimitry Andric return selectZExt(I); 1405*0b57cec5SDimitry Andric case Instruction::SExt: 1406*0b57cec5SDimitry Andric return selectSExt(I); 1407*0b57cec5SDimitry Andric case Instruction::ICmp: 1408*0b57cec5SDimitry Andric return selectICmp(I); 1409*0b57cec5SDimitry Andric case Instruction::FCmp: 1410*0b57cec5SDimitry Andric return selectFCmp(I); 1411*0b57cec5SDimitry Andric case Instruction::BitCast: 1412*0b57cec5SDimitry Andric return selectBitCast(I); 1413*0b57cec5SDimitry Andric case Instruction::Load: 1414*0b57cec5SDimitry Andric return selectLoad(I); 1415*0b57cec5SDimitry Andric case Instruction::Store: 1416*0b57cec5SDimitry Andric return selectStore(I); 1417*0b57cec5SDimitry Andric case Instruction::Br: 1418*0b57cec5SDimitry Andric return selectBr(I); 1419*0b57cec5SDimitry Andric case Instruction::Ret: 1420*0b57cec5SDimitry Andric return selectRet(I); 1421*0b57cec5SDimitry Andric case Instruction::Unreachable: 1422*0b57cec5SDimitry Andric return selectUnreachable(I); 1423*0b57cec5SDimitry Andric default: 1424*0b57cec5SDimitry Andric break; 1425*0b57cec5SDimitry Andric } 1426*0b57cec5SDimitry Andric 1427*0b57cec5SDimitry Andric // Fall back to target-independent instruction selection. 1428*0b57cec5SDimitry Andric return selectOperator(I, I->getOpcode()); 1429*0b57cec5SDimitry Andric } 1430*0b57cec5SDimitry Andric 1431*0b57cec5SDimitry Andric FastISel *WebAssembly::createFastISel(FunctionLoweringInfo &FuncInfo, 1432*0b57cec5SDimitry Andric const TargetLibraryInfo *LibInfo) { 1433*0b57cec5SDimitry Andric return new WebAssemblyFastISel(FuncInfo, LibInfo); 1434*0b57cec5SDimitry Andric } 1435