| //===- ARMFastISel.cpp - ARM FastISel implementation ----------------------===// |
| // |
| // The LLVM Compiler Infrastructure |
| // |
| // This file is distributed under the University of Illinois Open Source |
| // License. See LICENSE.TXT for details. |
| // |
| //===----------------------------------------------------------------------===// |
| // |
| // This file defines the ARM-specific support for the FastISel class. Some |
| // of the target-specific code is generated by tablegen in the file |
| // ARMGenFastISel.inc, which is #included here. |
| // |
| //===----------------------------------------------------------------------===// |
| |
| #include "ARM.h" |
| #include "ARMBaseInstrInfo.h" |
| #include "ARMBaseRegisterInfo.h" |
| #include "ARMCallingConv.h" |
| #include "ARMConstantPoolValue.h" |
| #include "ARMISelLowering.h" |
| #include "ARMMachineFunctionInfo.h" |
| #include "ARMSubtarget.h" |
| #include "MCTargetDesc/ARMAddressingModes.h" |
| #include "MCTargetDesc/ARMBaseInfo.h" |
| #include "Utils/ARMBaseInfo.h" |
| #include "llvm/ADT/APFloat.h" |
| #include "llvm/ADT/APInt.h" |
| #include "llvm/ADT/DenseMap.h" |
| #include "llvm/ADT/SmallVector.h" |
| #include "llvm/CodeGen/CallingConvLower.h" |
| #include "llvm/CodeGen/FastISel.h" |
| #include "llvm/CodeGen/FunctionLoweringInfo.h" |
| #include "llvm/CodeGen/ISDOpcodes.h" |
| #include "llvm/CodeGen/MachineBasicBlock.h" |
| #include "llvm/CodeGen/MachineConstantPool.h" |
| #include "llvm/CodeGen/MachineFrameInfo.h" |
| #include "llvm/CodeGen/MachineFunction.h" |
| #include "llvm/CodeGen/MachineInstr.h" |
| #include "llvm/CodeGen/MachineInstrBuilder.h" |
| #include "llvm/CodeGen/MachineMemOperand.h" |
| #include "llvm/CodeGen/MachineOperand.h" |
| #include "llvm/CodeGen/MachineRegisterInfo.h" |
| #include "llvm/CodeGen/RuntimeLibcalls.h" |
| #include "llvm/CodeGen/TargetInstrInfo.h" |
| #include "llvm/CodeGen/TargetLowering.h" |
| #include "llvm/CodeGen/TargetOpcodes.h" |
| #include "llvm/CodeGen/TargetRegisterInfo.h" |
| #include "llvm/CodeGen/ValueTypes.h" |
| #include "llvm/IR/Argument.h" |
| #include "llvm/IR/Attributes.h" |
| #include "llvm/IR/CallSite.h" |
| #include "llvm/IR/CallingConv.h" |
| #include "llvm/IR/Constant.h" |
| #include "llvm/IR/Constants.h" |
| #include "llvm/IR/DataLayout.h" |
| #include "llvm/IR/DerivedTypes.h" |
| #include "llvm/IR/Function.h" |
| #include "llvm/IR/GetElementPtrTypeIterator.h" |
| #include "llvm/IR/GlobalValue.h" |
| #include "llvm/IR/GlobalVariable.h" |
| #include "llvm/IR/InstrTypes.h" |
| #include "llvm/IR/Instruction.h" |
| #include "llvm/IR/Instructions.h" |
| #include "llvm/IR/IntrinsicInst.h" |
| #include "llvm/IR/Intrinsics.h" |
| #include "llvm/IR/Module.h" |
| #include "llvm/IR/Operator.h" |
| #include "llvm/IR/Type.h" |
| #include "llvm/IR/User.h" |
| #include "llvm/IR/Value.h" |
| #include "llvm/MC/MCInstrDesc.h" |
| #include "llvm/MC/MCRegisterInfo.h" |
| #include "llvm/Support/Casting.h" |
| #include "llvm/Support/Compiler.h" |
| #include "llvm/Support/ErrorHandling.h" |
| #include "llvm/Support/MachineValueType.h" |
| #include "llvm/Support/MathExtras.h" |
| #include "llvm/Target/TargetMachine.h" |
| #include "llvm/Target/TargetOptions.h" |
| #include <cassert> |
| #include <cstdint> |
| #include <utility> |
| |
| using namespace llvm; |
| |
| namespace { |
| |
| // All possible address modes, plus some. |
| struct Address { |
| enum { |
| RegBase, |
| FrameIndexBase |
| } BaseType = RegBase; |
| |
| union { |
| unsigned Reg; |
| int FI; |
| } Base; |
| |
| int Offset = 0; |
| |
| // Innocuous defaults for our address. |
| Address() { |
| Base.Reg = 0; |
| } |
| }; |
| |
| class ARMFastISel final : public FastISel { |
| /// Subtarget - Keep a pointer to the ARMSubtarget around so that we can |
| /// make the right decision when generating code for different targets. |
| const ARMSubtarget *Subtarget; |
| Module &M; |
| const TargetMachine &TM; |
| const TargetInstrInfo &TII; |
| const TargetLowering &TLI; |
| ARMFunctionInfo *AFI; |
| |
| // Convenience variables to avoid some queries. |
| bool isThumb2; |
| LLVMContext *Context; |
| |
| public: |
| explicit ARMFastISel(FunctionLoweringInfo &funcInfo, |
| const TargetLibraryInfo *libInfo) |
| : FastISel(funcInfo, libInfo), |
| Subtarget( |
| &static_cast<const ARMSubtarget &>(funcInfo.MF->getSubtarget())), |
| M(const_cast<Module &>(*funcInfo.Fn->getParent())), |
| TM(funcInfo.MF->getTarget()), TII(*Subtarget->getInstrInfo()), |
| TLI(*Subtarget->getTargetLowering()) { |
| AFI = funcInfo.MF->getInfo<ARMFunctionInfo>(); |
| isThumb2 = AFI->isThumbFunction(); |
| Context = &funcInfo.Fn->getContext(); |
| } |
| |
| private: |
| // Code from FastISel.cpp. |
| |
| unsigned fastEmitInst_r(unsigned MachineInstOpcode, |
| const TargetRegisterClass *RC, |
| unsigned Op0, bool Op0IsKill); |
| unsigned fastEmitInst_rr(unsigned MachineInstOpcode, |
| const TargetRegisterClass *RC, |
| unsigned Op0, bool Op0IsKill, |
| unsigned Op1, bool Op1IsKill); |
| unsigned fastEmitInst_ri(unsigned MachineInstOpcode, |
| const TargetRegisterClass *RC, |
| unsigned Op0, bool Op0IsKill, |
| uint64_t Imm); |
| unsigned fastEmitInst_i(unsigned MachineInstOpcode, |
| const TargetRegisterClass *RC, |
| uint64_t Imm); |
| |
| // Backend specific FastISel code. |
| |
| bool fastSelectInstruction(const Instruction *I) override; |
| unsigned fastMaterializeConstant(const Constant *C) override; |
| unsigned fastMaterializeAlloca(const AllocaInst *AI) override; |
| bool tryToFoldLoadIntoMI(MachineInstr *MI, unsigned OpNo, |
| const LoadInst *LI) override; |
| bool fastLowerArguments() override; |
| |
| #include "ARMGenFastISel.inc" |
| |
| // Instruction selection routines. |
| |
| bool SelectLoad(const Instruction *I); |
| bool SelectStore(const Instruction *I); |
| bool SelectBranch(const Instruction *I); |
| bool SelectIndirectBr(const Instruction *I); |
| bool SelectCmp(const Instruction *I); |
| bool SelectFPExt(const Instruction *I); |
| bool SelectFPTrunc(const Instruction *I); |
| bool SelectBinaryIntOp(const Instruction *I, unsigned ISDOpcode); |
| bool SelectBinaryFPOp(const Instruction *I, unsigned ISDOpcode); |
| bool SelectIToFP(const Instruction *I, bool isSigned); |
| bool SelectFPToI(const Instruction *I, bool isSigned); |
| bool SelectDiv(const Instruction *I, bool isSigned); |
| bool SelectRem(const Instruction *I, bool isSigned); |
| bool SelectCall(const Instruction *I, const char *IntrMemName); |
| bool SelectIntrinsicCall(const IntrinsicInst &I); |
| bool SelectSelect(const Instruction *I); |
| bool SelectRet(const Instruction *I); |
| bool SelectTrunc(const Instruction *I); |
| bool SelectIntExt(const Instruction *I); |
| bool SelectShift(const Instruction *I, ARM_AM::ShiftOpc ShiftTy); |
| |
| // Utility routines. |
| |
| bool isPositionIndependent() const; |
| bool isTypeLegal(Type *Ty, MVT &VT); |
| bool isLoadTypeLegal(Type *Ty, MVT &VT); |
| bool ARMEmitCmp(const Value *Src1Value, const Value *Src2Value, |
| bool isZExt, bool isEquality); |
| bool ARMEmitLoad(MVT VT, unsigned &ResultReg, Address &Addr, |
| unsigned Alignment = 0, bool isZExt = true, |
| bool allocReg = true); |
| bool ARMEmitStore(MVT VT, unsigned SrcReg, Address &Addr, |
| unsigned Alignment = 0); |
| bool ARMComputeAddress(const Value *Obj, Address &Addr); |
| void ARMSimplifyAddress(Address &Addr, MVT VT, bool useAM3); |
| bool ARMIsMemCpySmall(uint64_t Len); |
| bool ARMTryEmitSmallMemCpy(Address Dest, Address Src, uint64_t Len, |
| unsigned Alignment); |
| unsigned ARMEmitIntExt(MVT SrcVT, unsigned SrcReg, MVT DestVT, bool isZExt); |
| unsigned ARMMaterializeFP(const ConstantFP *CFP, MVT VT); |
| unsigned ARMMaterializeInt(const Constant *C, MVT VT); |
| unsigned ARMMaterializeGV(const GlobalValue *GV, MVT VT); |
| unsigned ARMMoveToFPReg(MVT VT, unsigned SrcReg); |
| unsigned ARMMoveToIntReg(MVT VT, unsigned SrcReg); |
| unsigned ARMSelectCallOp(bool UseReg); |
| unsigned ARMLowerPICELF(const GlobalValue *GV, unsigned Align, MVT VT); |
| |
| const TargetLowering *getTargetLowering() { return &TLI; } |
| |
| // Call handling routines. |
| |
| CCAssignFn *CCAssignFnForCall(CallingConv::ID CC, |
| bool Return, |
| bool isVarArg); |
| bool ProcessCallArgs(SmallVectorImpl<Value*> &Args, |
| SmallVectorImpl<unsigned> &ArgRegs, |
| SmallVectorImpl<MVT> &ArgVTs, |
| SmallVectorImpl<ISD::ArgFlagsTy> &ArgFlags, |
| SmallVectorImpl<unsigned> &RegArgs, |
| CallingConv::ID CC, |
| unsigned &NumBytes, |
| bool isVarArg); |
| unsigned getLibcallReg(const Twine &Name); |
| bool FinishCall(MVT RetVT, SmallVectorImpl<unsigned> &UsedRegs, |
| const Instruction *I, CallingConv::ID CC, |
| unsigned &NumBytes, bool isVarArg); |
| bool ARMEmitLibcall(const Instruction *I, RTLIB::Libcall Call); |
| |
| // OptionalDef handling routines. |
| |
| bool isARMNEONPred(const MachineInstr *MI); |
| bool DefinesOptionalPredicate(MachineInstr *MI, bool *CPSR); |
| const MachineInstrBuilder &AddOptionalDefs(const MachineInstrBuilder &MIB); |
| void AddLoadStoreOperands(MVT VT, Address &Addr, |
| const MachineInstrBuilder &MIB, |
| MachineMemOperand::Flags Flags, bool useAM3); |
| }; |
| |
| } // end anonymous namespace |
| |
| #include "ARMGenCallingConv.inc" |
| |
| // DefinesOptionalPredicate - This is different from DefinesPredicate in that |
| // we don't care about implicit defs here, just places we'll need to add a |
| // default CCReg argument. Sets CPSR if we're setting CPSR instead of CCR. |
| bool ARMFastISel::DefinesOptionalPredicate(MachineInstr *MI, bool *CPSR) { |
| if (!MI->hasOptionalDef()) |
| return false; |
| |
| // Look to see if our OptionalDef is defining CPSR or CCR. |
| for (const MachineOperand &MO : MI->operands()) { |
| if (!MO.isReg() || !MO.isDef()) continue; |
| if (MO.getReg() == ARM::CPSR) |
| *CPSR = true; |
| } |
| return true; |
| } |
| |
| bool ARMFastISel::isARMNEONPred(const MachineInstr *MI) { |
| const MCInstrDesc &MCID = MI->getDesc(); |
| |
| // If we're a thumb2 or not NEON function we'll be handled via isPredicable. |
| if ((MCID.TSFlags & ARMII::DomainMask) != ARMII::DomainNEON || |
| AFI->isThumb2Function()) |
| return MI->isPredicable(); |
| |
| for (const MCOperandInfo &opInfo : MCID.operands()) |
| if (opInfo.isPredicate()) |
| return true; |
| |
| return false; |
| } |
| |
| // If the machine is predicable go ahead and add the predicate operands, if |
| // it needs default CC operands add those. |
| // TODO: If we want to support thumb1 then we'll need to deal with optional |
| // CPSR defs that need to be added before the remaining operands. See s_cc_out |
| // for descriptions why. |
| const MachineInstrBuilder & |
| ARMFastISel::AddOptionalDefs(const MachineInstrBuilder &MIB) { |
| MachineInstr *MI = &*MIB; |
| |
| // Do we use a predicate? or... |
| // Are we NEON in ARM mode and have a predicate operand? If so, I know |
| // we're not predicable but add it anyways. |
| if (isARMNEONPred(MI)) |
| MIB.add(predOps(ARMCC::AL)); |
| |
| // Do we optionally set a predicate? Preds is size > 0 iff the predicate |
| // defines CPSR. All other OptionalDefines in ARM are the CCR register. |
| bool CPSR = false; |
| if (DefinesOptionalPredicate(MI, &CPSR)) |
| MIB.add(CPSR ? t1CondCodeOp() : condCodeOp()); |
| return MIB; |
| } |
| |
| unsigned ARMFastISel::fastEmitInst_r(unsigned MachineInstOpcode, |
| const TargetRegisterClass *RC, |
| unsigned Op0, bool Op0IsKill) { |
| unsigned ResultReg = createResultReg(RC); |
| const MCInstrDesc &II = TII.get(MachineInstOpcode); |
| |
| // Make sure the input operand is sufficiently constrained to be legal |
| // for this instruction. |
| Op0 = constrainOperandRegClass(II, Op0, 1); |
| if (II.getNumDefs() >= 1) { |
| AddOptionalDefs(BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc, II, |
| ResultReg).addReg(Op0, Op0IsKill * RegState::Kill)); |
| } else { |
| AddOptionalDefs(BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc, II) |
| .addReg(Op0, Op0IsKill * RegState::Kill)); |
| AddOptionalDefs(BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc, |
| TII.get(TargetOpcode::COPY), ResultReg) |
| .addReg(II.ImplicitDefs[0])); |
| } |
| return ResultReg; |
| } |
| |
| unsigned ARMFastISel::fastEmitInst_rr(unsigned MachineInstOpcode, |
| const TargetRegisterClass *RC, |
| unsigned Op0, bool Op0IsKill, |
| unsigned Op1, bool Op1IsKill) { |
| unsigned ResultReg = createResultReg(RC); |
| const MCInstrDesc &II = TII.get(MachineInstOpcode); |
| |
| // Make sure the input operands are sufficiently constrained to be legal |
| // for this instruction. |
| Op0 = constrainOperandRegClass(II, Op0, 1); |
| Op1 = constrainOperandRegClass(II, Op1, 2); |
| |
| if (II.getNumDefs() >= 1) { |
| AddOptionalDefs( |
| BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc, II, ResultReg) |
| .addReg(Op0, Op0IsKill * RegState::Kill) |
| .addReg(Op1, Op1IsKill * RegState::Kill)); |
| } else { |
| AddOptionalDefs(BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc, II) |
| .addReg(Op0, Op0IsKill * RegState::Kill) |
| .addReg(Op1, Op1IsKill * RegState::Kill)); |
| AddOptionalDefs(BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc, |
| TII.get(TargetOpcode::COPY), ResultReg) |
| .addReg(II.ImplicitDefs[0])); |
| } |
| return ResultReg; |
| } |
| |
| unsigned ARMFastISel::fastEmitInst_ri(unsigned MachineInstOpcode, |
| const TargetRegisterClass *RC, |
| unsigned Op0, bool Op0IsKill, |
| uint64_t Imm) { |
| unsigned ResultReg = createResultReg(RC); |
| const MCInstrDesc &II = TII.get(MachineInstOpcode); |
| |
| // Make sure the input operand is sufficiently constrained to be legal |
| // for this instruction. |
| Op0 = constrainOperandRegClass(II, Op0, 1); |
| if (II.getNumDefs() >= 1) { |
| AddOptionalDefs( |
| BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc, II, ResultReg) |
| .addReg(Op0, Op0IsKill * RegState::Kill) |
| .addImm(Imm)); |
| } else { |
| AddOptionalDefs(BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc, II) |
| .addReg(Op0, Op0IsKill * RegState::Kill) |
| .addImm(Imm)); |
| AddOptionalDefs(BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc, |
| TII.get(TargetOpcode::COPY), ResultReg) |
| .addReg(II.ImplicitDefs[0])); |
| } |
| return ResultReg; |
| } |
| |
| unsigned ARMFastISel::fastEmitInst_i(unsigned MachineInstOpcode, |
| const TargetRegisterClass *RC, |
| uint64_t Imm) { |
| unsigned ResultReg = createResultReg(RC); |
| const MCInstrDesc &II = TII.get(MachineInstOpcode); |
| |
| if (II.getNumDefs() >= 1) { |
| AddOptionalDefs(BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc, II, |
| ResultReg).addImm(Imm)); |
| } else { |
| AddOptionalDefs(BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc, II) |
| .addImm(Imm)); |
| AddOptionalDefs(BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc, |
| TII.get(TargetOpcode::COPY), ResultReg) |
| .addReg(II.ImplicitDefs[0])); |
| } |
| return ResultReg; |
| } |
| |
| // TODO: Don't worry about 64-bit now, but when this is fixed remove the |
| // checks from the various callers. |
| unsigned ARMFastISel::ARMMoveToFPReg(MVT VT, unsigned SrcReg) { |
| if (VT == MVT::f64) return 0; |
| |
| unsigned MoveReg = createResultReg(TLI.getRegClassFor(VT)); |
| AddOptionalDefs(BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc, |
| TII.get(ARM::VMOVSR), MoveReg) |
| .addReg(SrcReg)); |
| return MoveReg; |
| } |
| |
| unsigned ARMFastISel::ARMMoveToIntReg(MVT VT, unsigned SrcReg) { |
| if (VT == MVT::i64) return 0; |
| |
| unsigned MoveReg = createResultReg(TLI.getRegClassFor(VT)); |
| AddOptionalDefs(BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc, |
| TII.get(ARM::VMOVRS), MoveReg) |
| .addReg(SrcReg)); |
| return MoveReg; |
| } |
| |
| // For double width floating point we need to materialize two constants |
| // (the high and the low) into integer registers then use a move to get |
| // the combined constant into an FP reg. |
| unsigned ARMFastISel::ARMMaterializeFP(const ConstantFP *CFP, MVT VT) { |
| const APFloat Val = CFP->getValueAPF(); |
| bool is64bit = VT == MVT::f64; |
| |
| // This checks to see if we can use VFP3 instructions to materialize |
| // a constant, otherwise we have to go through the constant pool. |
| if (TLI.isFPImmLegal(Val, VT)) { |
| int Imm; |
| unsigned Opc; |
| if (is64bit) { |
| Imm = ARM_AM::getFP64Imm(Val); |
| Opc = ARM::FCONSTD; |
| } else { |
| Imm = ARM_AM::getFP32Imm(Val); |
| Opc = ARM::FCONSTS; |
| } |
| unsigned DestReg = createResultReg(TLI.getRegClassFor(VT)); |
| AddOptionalDefs(BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc, |
| TII.get(Opc), DestReg).addImm(Imm)); |
| return DestReg; |
| } |
| |
| // Require VFP2 for loading fp constants. |
| if (!Subtarget->hasVFP2()) return false; |
| |
| // MachineConstantPool wants an explicit alignment. |
| unsigned Align = DL.getPrefTypeAlignment(CFP->getType()); |
| if (Align == 0) { |
| // TODO: Figure out if this is correct. |
| Align = DL.getTypeAllocSize(CFP->getType()); |
| } |
| unsigned Idx = MCP.getConstantPoolIndex(cast<Constant>(CFP), Align); |
| unsigned DestReg = createResultReg(TLI.getRegClassFor(VT)); |
| unsigned Opc = is64bit ? ARM::VLDRD : ARM::VLDRS; |
| |
| // The extra reg is for addrmode5. |
| AddOptionalDefs( |
| BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc, TII.get(Opc), DestReg) |
| .addConstantPoolIndex(Idx) |
| .addReg(0)); |
| return DestReg; |
| } |
| |
| unsigned ARMFastISel::ARMMaterializeInt(const Constant *C, MVT VT) { |
| if (VT != MVT::i32 && VT != MVT::i16 && VT != MVT::i8 && VT != MVT::i1) |
| return 0; |
| |
| // If we can do this in a single instruction without a constant pool entry |
| // do so now. |
| const ConstantInt *CI = cast<ConstantInt>(C); |
| if (Subtarget->hasV6T2Ops() && isUInt<16>(CI->getZExtValue())) { |
| unsigned Opc = isThumb2 ? ARM::t2MOVi16 : ARM::MOVi16; |
| const TargetRegisterClass *RC = isThumb2 ? &ARM::rGPRRegClass : |
| &ARM::GPRRegClass; |
| unsigned ImmReg = createResultReg(RC); |
| AddOptionalDefs(BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc, |
| TII.get(Opc), ImmReg) |
| .addImm(CI->getZExtValue())); |
| return ImmReg; |
| } |
| |
| // Use MVN to emit negative constants. |
| if (VT == MVT::i32 && Subtarget->hasV6T2Ops() && CI->isNegative()) { |
| unsigned Imm = (unsigned)~(CI->getSExtValue()); |
| bool UseImm = isThumb2 ? (ARM_AM::getT2SOImmVal(Imm) != -1) : |
| (ARM_AM::getSOImmVal(Imm) != -1); |
| if (UseImm) { |
| unsigned Opc = isThumb2 ? ARM::t2MVNi : ARM::MVNi; |
| const TargetRegisterClass *RC = isThumb2 ? &ARM::rGPRRegClass : |
| &ARM::GPRRegClass; |
| unsigned ImmReg = createResultReg(RC); |
| AddOptionalDefs(BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc, |
| TII.get(Opc), ImmReg) |
| .addImm(Imm)); |
| return ImmReg; |
| } |
| } |
| |
| unsigned ResultReg = 0; |
| if (Subtarget->useMovt(*FuncInfo.MF)) |
| ResultReg = fastEmit_i(VT, VT, ISD::Constant, CI->getZExtValue()); |
| |
| if (ResultReg) |
| return ResultReg; |
| |
| // Load from constant pool. For now 32-bit only. |
| if (VT != MVT::i32) |
| return 0; |
| |
| // MachineConstantPool wants an explicit alignment. |
| unsigned Align = DL.getPrefTypeAlignment(C->getType()); |
| if (Align == 0) { |
| // TODO: Figure out if this is correct. |
| Align = DL.getTypeAllocSize(C->getType()); |
| } |
| unsigned Idx = MCP.getConstantPoolIndex(C, Align); |
| ResultReg = createResultReg(TLI.getRegClassFor(VT)); |
| if (isThumb2) |
| AddOptionalDefs(BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc, |
| TII.get(ARM::t2LDRpci), ResultReg) |
| .addConstantPoolIndex(Idx)); |
| else { |
| // The extra immediate is for addrmode2. |
| ResultReg = constrainOperandRegClass(TII.get(ARM::LDRcp), ResultReg, 0); |
| AddOptionalDefs(BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc, |
| TII.get(ARM::LDRcp), ResultReg) |
| .addConstantPoolIndex(Idx) |
| .addImm(0)); |
| } |
| return ResultReg; |
| } |
| |
| bool ARMFastISel::isPositionIndependent() const { |
| return TLI.isPositionIndependent(); |
| } |
| |
| unsigned ARMFastISel::ARMMaterializeGV(const GlobalValue *GV, MVT VT) { |
| // For now 32-bit only. |
| if (VT != MVT::i32 || GV->isThreadLocal()) return 0; |
| |
| // ROPI/RWPI not currently supported. |
| if (Subtarget->isROPI() || Subtarget->isRWPI()) |
| return 0; |
| |
| bool IsIndirect = Subtarget->isGVIndirectSymbol(GV); |
| const TargetRegisterClass *RC = isThumb2 ? &ARM::rGPRRegClass |
| : &ARM::GPRRegClass; |
| unsigned DestReg = createResultReg(RC); |
| |
| // FastISel TLS support on non-MachO is broken, punt to SelectionDAG. |
| const GlobalVariable *GVar = dyn_cast<GlobalVariable>(GV); |
| bool IsThreadLocal = GVar && GVar->isThreadLocal(); |
| if (!Subtarget->isTargetMachO() && IsThreadLocal) return 0; |
| |
| bool IsPositionIndependent = isPositionIndependent(); |
| // Use movw+movt when possible, it avoids constant pool entries. |
| // Non-darwin targets only support static movt relocations in FastISel. |
| if (Subtarget->useMovt(*FuncInfo.MF) && |
| (Subtarget->isTargetMachO() || !IsPositionIndependent)) { |
| unsigned Opc; |
| unsigned char TF = 0; |
| if (Subtarget->isTargetMachO()) |
| TF = ARMII::MO_NONLAZY; |
| |
| if (IsPositionIndependent) |
| Opc = isThumb2 ? ARM::t2MOV_ga_pcrel : ARM::MOV_ga_pcrel; |
| else |
| Opc = isThumb2 ? ARM::t2MOVi32imm : ARM::MOVi32imm; |
| AddOptionalDefs(BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc, |
| TII.get(Opc), DestReg).addGlobalAddress(GV, 0, TF)); |
| } else { |
| // MachineConstantPool wants an explicit alignment. |
| unsigned Align = DL.getPrefTypeAlignment(GV->getType()); |
| if (Align == 0) { |
| // TODO: Figure out if this is correct. |
| Align = DL.getTypeAllocSize(GV->getType()); |
| } |
| |
| if (Subtarget->isTargetELF() && IsPositionIndependent) |
| return ARMLowerPICELF(GV, Align, VT); |
| |
| // Grab index. |
| unsigned PCAdj = IsPositionIndependent ? (Subtarget->isThumb() ? 4 : 8) : 0; |
| unsigned Id = AFI->createPICLabelUId(); |
| ARMConstantPoolValue *CPV = ARMConstantPoolConstant::Create(GV, Id, |
| ARMCP::CPValue, |
| PCAdj); |
| unsigned Idx = MCP.getConstantPoolIndex(CPV, Align); |
| |
| // Load value. |
| MachineInstrBuilder MIB; |
| if (isThumb2) { |
| unsigned Opc = IsPositionIndependent ? ARM::t2LDRpci_pic : ARM::t2LDRpci; |
| MIB = BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc, TII.get(Opc), |
| DestReg).addConstantPoolIndex(Idx); |
| if (IsPositionIndependent) |
| MIB.addImm(Id); |
| AddOptionalDefs(MIB); |
| } else { |
| // The extra immediate is for addrmode2. |
| DestReg = constrainOperandRegClass(TII.get(ARM::LDRcp), DestReg, 0); |
| MIB = BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc, |
| TII.get(ARM::LDRcp), DestReg) |
| .addConstantPoolIndex(Idx) |
| .addImm(0); |
| AddOptionalDefs(MIB); |
| |
| if (IsPositionIndependent) { |
| unsigned Opc = IsIndirect ? ARM::PICLDR : ARM::PICADD; |
| unsigned NewDestReg = createResultReg(TLI.getRegClassFor(VT)); |
| |
| MachineInstrBuilder MIB = BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, |
| DbgLoc, TII.get(Opc), NewDestReg) |
| .addReg(DestReg) |
| .addImm(Id); |
| AddOptionalDefs(MIB); |
| return NewDestReg; |
| } |
| } |
| } |
| |
| if (IsIndirect) { |
| MachineInstrBuilder MIB; |
| unsigned NewDestReg = createResultReg(TLI.getRegClassFor(VT)); |
| if (isThumb2) |
| MIB = BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc, |
| TII.get(ARM::t2LDRi12), NewDestReg) |
| .addReg(DestReg) |
| .addImm(0); |
| else |
| MIB = BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc, |
| TII.get(ARM::LDRi12), NewDestReg) |
| .addReg(DestReg) |
| .addImm(0); |
| DestReg = NewDestReg; |
| AddOptionalDefs(MIB); |
| } |
| |
| return DestReg; |
| } |
| |
| unsigned ARMFastISel::fastMaterializeConstant(const Constant *C) { |
| EVT CEVT = TLI.getValueType(DL, C->getType(), true); |
| |
| // Only handle simple types. |
| if (!CEVT.isSimple()) return 0; |
| MVT VT = CEVT.getSimpleVT(); |
| |
| if (const ConstantFP *CFP = dyn_cast<ConstantFP>(C)) |
| return ARMMaterializeFP(CFP, VT); |
| else if (const GlobalValue *GV = dyn_cast<GlobalValue>(C)) |
| return ARMMaterializeGV(GV, VT); |
| else if (isa<ConstantInt>(C)) |
| return ARMMaterializeInt(C, VT); |
| |
| return 0; |
| } |
| |
| // TODO: unsigned ARMFastISel::TargetMaterializeFloatZero(const ConstantFP *CF); |
| |
| unsigned ARMFastISel::fastMaterializeAlloca(const AllocaInst *AI) { |
| // Don't handle dynamic allocas. |
| if (!FuncInfo.StaticAllocaMap.count(AI)) return 0; |
| |
| MVT VT; |
| if (!isLoadTypeLegal(AI->getType(), VT)) return 0; |
| |
| DenseMap<const AllocaInst*, int>::iterator SI = |
| FuncInfo.StaticAllocaMap.find(AI); |
| |
| // This will get lowered later into the correct offsets and registers |
| // via rewriteXFrameIndex. |
| if (SI != FuncInfo.StaticAllocaMap.end()) { |
| unsigned Opc = isThumb2 ? ARM::t2ADDri : ARM::ADDri; |
| const TargetRegisterClass* RC = TLI.getRegClassFor(VT); |
| unsigned ResultReg = createResultReg(RC); |
| ResultReg = constrainOperandRegClass(TII.get(Opc), ResultReg, 0); |
| |
| AddOptionalDefs(BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc, |
| TII.get(Opc), ResultReg) |
| .addFrameIndex(SI->second) |
| .addImm(0)); |
| return ResultReg; |
| } |
| |
| return 0; |
| } |
| |
| bool ARMFastISel::isTypeLegal(Type *Ty, MVT &VT) { |
| EVT evt = TLI.getValueType(DL, Ty, true); |
| |
| // Only handle simple types. |
| if (evt == MVT::Other || !evt.isSimple()) return false; |
| VT = evt.getSimpleVT(); |
| |
| // Handle all legal types, i.e. a register that will directly hold this |
| // value. |
| return TLI.isTypeLegal(VT); |
| } |
| |
| bool ARMFastISel::isLoadTypeLegal(Type *Ty, MVT &VT) { |
| if (isTypeLegal(Ty, VT)) return true; |
| |
| // If this is a type than can be sign or zero-extended to a basic operation |
| // go ahead and accept it now. |
| if (VT == MVT::i1 || VT == MVT::i8 || VT == MVT::i16) |
| return true; |
| |
| return false; |
| } |
| |
| // Computes the address to get to an object. |
| bool ARMFastISel::ARMComputeAddress(const Value *Obj, Address &Addr) { |
| // Some boilerplate from the X86 FastISel. |
| const User *U = nullptr; |
| unsigned Opcode = Instruction::UserOp1; |
| if (const Instruction *I = dyn_cast<Instruction>(Obj)) { |
| // Don't walk into other basic blocks unless the object is an alloca from |
| // another block, otherwise it may not have a virtual register assigned. |
| if (FuncInfo.StaticAllocaMap.count(static_cast<const AllocaInst *>(Obj)) || |
| FuncInfo.MBBMap[I->getParent()] == FuncInfo.MBB) { |
| Opcode = I->getOpcode(); |
| U = I; |
| } |
| } else if (const ConstantExpr *C = dyn_cast<ConstantExpr>(Obj)) { |
| Opcode = C->getOpcode(); |
| U = C; |
| } |
| |
| if (PointerType *Ty = dyn_cast<PointerType>(Obj->getType())) |
| if (Ty->getAddressSpace() > 255) |
| // Fast instruction selection doesn't support the special |
| // address spaces. |
| return false; |
| |
| switch (Opcode) { |
| default: |
| break; |
| case Instruction::BitCast: |
| // Look through bitcasts. |
| return ARMComputeAddress(U->getOperand(0), Addr); |
| case Instruction::IntToPtr: |
| // Look past no-op inttoptrs. |
| if (TLI.getValueType(DL, U->getOperand(0)->getType()) == |
| TLI.getPointerTy(DL)) |
| return ARMComputeAddress(U->getOperand(0), Addr); |
| break; |
| case Instruction::PtrToInt: |
| // Look past no-op ptrtoints. |
| if (TLI.getValueType(DL, U->getType()) == TLI.getPointerTy(DL)) |
| return ARMComputeAddress(U->getOperand(0), Addr); |
| break; |
| case Instruction::GetElementPtr: { |
| Address SavedAddr = Addr; |
| int TmpOffset = Addr.Offset; |
| |
| // Iterate through the GEP folding the constants into offsets where |
| // we can. |
| gep_type_iterator GTI = gep_type_begin(U); |
| for (User::const_op_iterator i = U->op_begin() + 1, e = U->op_end(); |
| i != e; ++i, ++GTI) { |
| const Value *Op = *i; |
| if (StructType *STy = GTI.getStructTypeOrNull()) { |
| const StructLayout *SL = DL.getStructLayout(STy); |
| unsigned Idx = cast<ConstantInt>(Op)->getZExtValue(); |
| TmpOffset += SL->getElementOffset(Idx); |
| } else { |
| uint64_t S = DL.getTypeAllocSize(GTI.getIndexedType()); |
| while (true) { |
| if (const ConstantInt *CI = dyn_cast<ConstantInt>(Op)) { |
| // Constant-offset addressing. |
| TmpOffset += CI->getSExtValue() * S; |
| break; |
| } |
| if (canFoldAddIntoGEP(U, Op)) { |
| // A compatible add with a constant operand. Fold the constant. |
| ConstantInt *CI = |
| cast<ConstantInt>(cast<AddOperator>(Op)->getOperand(1)); |
| TmpOffset += CI->getSExtValue() * S; |
| // Iterate on the other operand. |
| Op = cast<AddOperator>(Op)->getOperand(0); |
| continue; |
| } |
| // Unsupported |
| goto unsupported_gep; |
| } |
| } |
| } |
| |
| // Try to grab the base operand now. |
| Addr.Offset = TmpOffset; |
| if (ARMComputeAddress(U->getOperand(0), Addr)) return true; |
| |
| // We failed, restore everything and try the other options. |
| Addr = SavedAddr; |
| |
| unsupported_gep: |
| break; |
| } |
| case Instruction::Alloca: { |
| const AllocaInst *AI = cast<AllocaInst>(Obj); |
| DenseMap<const AllocaInst*, int>::iterator SI = |
| FuncInfo.StaticAllocaMap.find(AI); |
| if (SI != FuncInfo.StaticAllocaMap.end()) { |
| Addr.BaseType = Address::FrameIndexBase; |
| Addr.Base.FI = SI->second; |
| return true; |
| } |
| break; |
| } |
| } |
| |
| // Try to get this in a register if nothing else has worked. |
| if (Addr.Base.Reg == 0) Addr.Base.Reg = getRegForValue(Obj); |
| return Addr.Base.Reg != 0; |
| } |
| |
| void ARMFastISel::ARMSimplifyAddress(Address &Addr, MVT VT, bool useAM3) { |
| bool needsLowering = false; |
| switch (VT.SimpleTy) { |
| default: llvm_unreachable("Unhandled load/store type!"); |
| case MVT::i1: |
| case MVT::i8: |
| case MVT::i16: |
| case MVT::i32: |
| if (!useAM3) { |
| // Integer loads/stores handle 12-bit offsets. |
| needsLowering = ((Addr.Offset & 0xfff) != Addr.Offset); |
| // Handle negative offsets. |
| if (needsLowering && isThumb2) |
| needsLowering = !(Subtarget->hasV6T2Ops() && Addr.Offset < 0 && |
| Addr.Offset > -256); |
| } else { |
| // ARM halfword load/stores and signed byte loads use +/-imm8 offsets. |
| needsLowering = (Addr.Offset > 255 || Addr.Offset < -255); |
| } |
| break; |
| case MVT::f32: |
| case MVT::f64: |
| // Floating point operands handle 8-bit offsets. |
| needsLowering = ((Addr.Offset & 0xff) != Addr.Offset); |
| break; |
| } |
| |
| // If this is a stack pointer and the offset needs to be simplified then |
| // put the alloca address into a register, set the base type back to |
| // register and continue. This should almost never happen. |
| if (needsLowering && Addr.BaseType == Address::FrameIndexBase) { |
| const TargetRegisterClass *RC = isThumb2 ? &ARM::tGPRRegClass |
| : &ARM::GPRRegClass; |
| unsigned ResultReg = createResultReg(RC); |
| unsigned Opc = isThumb2 ? ARM::t2ADDri : ARM::ADDri; |
| AddOptionalDefs(BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc, |
| TII.get(Opc), ResultReg) |
| .addFrameIndex(Addr.Base.FI) |
| .addImm(0)); |
| Addr.Base.Reg = ResultReg; |
| Addr.BaseType = Address::RegBase; |
| } |
| |
| // Since the offset is too large for the load/store instruction |
| // get the reg+offset into a register. |
| if (needsLowering) { |
| Addr.Base.Reg = fastEmit_ri_(MVT::i32, ISD::ADD, Addr.Base.Reg, |
| /*Op0IsKill*/false, Addr.Offset, MVT::i32); |
| Addr.Offset = 0; |
| } |
| } |
| |
| void ARMFastISel::AddLoadStoreOperands(MVT VT, Address &Addr, |
| const MachineInstrBuilder &MIB, |
| MachineMemOperand::Flags Flags, |
| bool useAM3) { |
| // addrmode5 output depends on the selection dag addressing dividing the |
| // offset by 4 that it then later multiplies. Do this here as well. |
| if (VT.SimpleTy == MVT::f32 || VT.SimpleTy == MVT::f64) |
| Addr.Offset /= 4; |
| |
| // Frame base works a bit differently. Handle it separately. |
| if (Addr.BaseType == Address::FrameIndexBase) { |
| int FI = Addr.Base.FI; |
| int Offset = Addr.Offset; |
| MachineMemOperand *MMO = FuncInfo.MF->getMachineMemOperand( |
| MachinePointerInfo::getFixedStack(*FuncInfo.MF, FI, Offset), Flags, |
| MFI.getObjectSize(FI), MFI.getObjectAlignment(FI)); |
| // Now add the rest of the operands. |
| MIB.addFrameIndex(FI); |
| |
| // ARM halfword load/stores and signed byte loads need an additional |
| // operand. |
| if (useAM3) { |
| int Imm = (Addr.Offset < 0) ? (0x100 | -Addr.Offset) : Addr.Offset; |
| MIB.addReg(0); |
| MIB.addImm(Imm); |
| } else { |
| MIB.addImm(Addr.Offset); |
| } |
| MIB.addMemOperand(MMO); |
| } else { |
| // Now add the rest of the operands. |
| MIB.addReg(Addr.Base.Reg); |
| |
| // ARM halfword load/stores and signed byte loads need an additional |
| // operand. |
| if (useAM3) { |
| int Imm = (Addr.Offset < 0) ? (0x100 | -Addr.Offset) : Addr.Offset; |
| MIB.addReg(0); |
| MIB.addImm(Imm); |
| } else { |
| MIB.addImm(Addr.Offset); |
| } |
| } |
| AddOptionalDefs(MIB); |
| } |
| |
| bool ARMFastISel::ARMEmitLoad(MVT VT, unsigned &ResultReg, Address &Addr, |
| unsigned Alignment, bool isZExt, bool allocReg) { |
| unsigned Opc; |
| bool useAM3 = false; |
| bool needVMOV = false; |
| const TargetRegisterClass *RC; |
| switch (VT.SimpleTy) { |
| // This is mostly going to be Neon/vector support. |
| default: return false; |
| case MVT::i1: |
| case MVT::i8: |
| if (isThumb2) { |
| if (Addr.Offset < 0 && Addr.Offset > -256 && Subtarget->hasV6T2Ops()) |
| Opc = isZExt ? ARM::t2LDRBi8 : ARM::t2LDRSBi8; |
| else |
| Opc = isZExt ? ARM::t2LDRBi12 : ARM::t2LDRSBi12; |
| } else { |
| if (isZExt) { |
| Opc = ARM::LDRBi12; |
| } else { |
| Opc = ARM::LDRSB; |
| useAM3 = true; |
| } |
| } |
| RC = isThumb2 ? &ARM::rGPRRegClass : &ARM::GPRnopcRegClass; |
| break; |
| case MVT::i16: |
| if (Alignment && Alignment < 2 && !Subtarget->allowsUnalignedMem()) |
| return false; |
| |
| if (isThumb2) { |
| if (Addr.Offset < 0 && Addr.Offset > -256 && Subtarget->hasV6T2Ops()) |
| Opc = isZExt ? ARM::t2LDRHi8 : ARM::t2LDRSHi8; |
| else |
| Opc = isZExt ? ARM::t2LDRHi12 : ARM::t2LDRSHi12; |
| } else { |
| Opc = isZExt ? ARM::LDRH : ARM::LDRSH; |
| useAM3 = true; |
| } |
| RC = isThumb2 ? &ARM::rGPRRegClass : &ARM::GPRnopcRegClass; |
| break; |
| case MVT::i32: |
| if (Alignment && Alignment < 4 && !Subtarget->allowsUnalignedMem()) |
| return false; |
| |
| if (isThumb2) { |
| if (Addr.Offset < 0 && Addr.Offset > -256 && Subtarget->hasV6T2Ops()) |
| Opc = ARM::t2LDRi8; |
| else |
| Opc = ARM::t2LDRi12; |
| } else { |
| Opc = ARM::LDRi12; |
| } |
| RC = isThumb2 ? &ARM::rGPRRegClass : &ARM::GPRnopcRegClass; |
| break; |
| case MVT::f32: |
| if (!Subtarget->hasVFP2()) return false; |
| // Unaligned loads need special handling. Floats require word-alignment. |
| if (Alignment && Alignment < 4) { |
| needVMOV = true; |
| VT = MVT::i32; |
| Opc = isThumb2 ? ARM::t2LDRi12 : ARM::LDRi12; |
| RC = isThumb2 ? &ARM::rGPRRegClass : &ARM::GPRnopcRegClass; |
| } else { |
| Opc = ARM::VLDRS; |
| RC = TLI.getRegClassFor(VT); |
| } |
| break; |
| case MVT::f64: |
| if (!Subtarget->hasVFP2()) return false; |
| // FIXME: Unaligned loads need special handling. Doublewords require |
| // word-alignment. |
| if (Alignment && Alignment < 4) |
| return false; |
| |
| Opc = ARM::VLDRD; |
| RC = TLI.getRegClassFor(VT); |
| break; |
| } |
| // Simplify this down to something we can handle. |
| ARMSimplifyAddress(Addr, VT, useAM3); |
| |
| // Create the base instruction, then add the operands. |
| if (allocReg) |
| ResultReg = createResultReg(RC); |
| assert(ResultReg > 255 && "Expected an allocated virtual register."); |
| MachineInstrBuilder MIB = BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc, |
| TII.get(Opc), ResultReg); |
| AddLoadStoreOperands(VT, Addr, MIB, MachineMemOperand::MOLoad, useAM3); |
| |
| // If we had an unaligned load of a float we've converted it to an regular |
| // load. Now we must move from the GRP to the FP register. |
| if (needVMOV) { |
| unsigned MoveReg = createResultReg(TLI.getRegClassFor(MVT::f32)); |
| AddOptionalDefs(BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc, |
| TII.get(ARM::VMOVSR), MoveReg) |
| .addReg(ResultReg)); |
| ResultReg = MoveReg; |
| } |
| return true; |
| } |
| |
| bool ARMFastISel::SelectLoad(const Instruction *I) { |
| // Atomic loads need special handling. |
| if (cast<LoadInst>(I)->isAtomic()) |
| return false; |
| |
| const Value *SV = I->getOperand(0); |
| if (TLI.supportSwiftError()) { |
| // Swifterror values can come from either a function parameter with |
| // swifterror attribute or an alloca with swifterror attribute. |
| if (const Argument *Arg = dyn_cast<Argument>(SV)) { |
| if (Arg->hasSwiftErrorAttr()) |
| return false; |
| } |
| |
| if (const AllocaInst *Alloca = dyn_cast<AllocaInst>(SV)) { |
| if (Alloca->isSwiftError()) |
| return false; |
| } |
| } |
| |
| // Verify we have a legal type before going any further. |
| MVT VT; |
| if (!isLoadTypeLegal(I->getType(), VT)) |
| return false; |
| |
| // See if we can handle this address. |
| Address Addr; |
| if (!ARMComputeAddress(I->getOperand(0), Addr)) return false; |
| |
| unsigned ResultReg; |
| if (!ARMEmitLoad(VT, ResultReg, Addr, cast<LoadInst>(I)->getAlignment())) |
| return false; |
| updateValueMap(I, ResultReg); |
| return true; |
| } |
| |
| bool ARMFastISel::ARMEmitStore(MVT VT, unsigned SrcReg, Address &Addr, |
| unsigned Alignment) { |
| unsigned StrOpc; |
| bool useAM3 = false; |
| switch (VT.SimpleTy) { |
| // This is mostly going to be Neon/vector support. |
| default: return false; |
| case MVT::i1: { |
| unsigned Res = createResultReg(isThumb2 ? &ARM::tGPRRegClass |
| : &ARM::GPRRegClass); |
| unsigned Opc = isThumb2 ? ARM::t2ANDri : ARM::ANDri; |
| SrcReg = constrainOperandRegClass(TII.get(Opc), SrcReg, 1); |
| AddOptionalDefs(BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc, |
| TII.get(Opc), Res) |
| .addReg(SrcReg).addImm(1)); |
| SrcReg = Res; |
| LLVM_FALLTHROUGH; |
| } |
| case MVT::i8: |
| if (isThumb2) { |
| if (Addr.Offset < 0 && Addr.Offset > -256 && Subtarget->hasV6T2Ops()) |
| StrOpc = ARM::t2STRBi8; |
| else |
| StrOpc = ARM::t2STRBi12; |
| } else { |
| StrOpc = ARM::STRBi12; |
| } |
| break; |
| case MVT::i16: |
| if (Alignment && Alignment < 2 && !Subtarget->allowsUnalignedMem()) |
| return false; |
| |
| if (isThumb2) { |
| if (Addr.Offset < 0 && Addr.Offset > -256 && Subtarget->hasV6T2Ops()) |
| StrOpc = ARM::t2STRHi8; |
| else |
| StrOpc = ARM::t2STRHi12; |
| } else { |
| StrOpc = ARM::STRH; |
| useAM3 = true; |
| } |
| break; |
| case MVT::i32: |
| if (Alignment && Alignment < 4 && !Subtarget->allowsUnalignedMem()) |
| return false; |
| |
| if (isThumb2) { |
| if (Addr.Offset < 0 && Addr.Offset > -256 && Subtarget->hasV6T2Ops()) |
| StrOpc = ARM::t2STRi8; |
| else |
| StrOpc = ARM::t2STRi12; |
| } else { |
| StrOpc = ARM::STRi12; |
| } |
| break; |
| case MVT::f32: |
| if (!Subtarget->hasVFP2()) return false; |
| // Unaligned stores need special handling. Floats require word-alignment. |
| if (Alignment && Alignment < 4) { |
| unsigned MoveReg = createResultReg(TLI.getRegClassFor(MVT::i32)); |
| AddOptionalDefs(BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc, |
| TII.get(ARM::VMOVRS), MoveReg) |
| .addReg(SrcReg)); |
| SrcReg = MoveReg; |
| VT = MVT::i32; |
| StrOpc = isThumb2 ? ARM::t2STRi12 : ARM::STRi12; |
| } else { |
| StrOpc = ARM::VSTRS; |
| } |
| break; |
| case MVT::f64: |
| if (!Subtarget->hasVFP2()) return false; |
| // FIXME: Unaligned stores need special handling. Doublewords require |
| // word-alignment. |
| if (Alignment && Alignment < 4) |
| return false; |
| |
| StrOpc = ARM::VSTRD; |
| break; |
| } |
| // Simplify this down to something we can handle. |
| ARMSimplifyAddress(Addr, VT, useAM3); |
| |
| // Create the base instruction, then add the operands. |
| SrcReg = constrainOperandRegClass(TII.get(StrOpc), SrcReg, 0); |
| MachineInstrBuilder MIB = BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc, |
| TII.get(StrOpc)) |
| .addReg(SrcReg); |
| AddLoadStoreOperands(VT, Addr, MIB, MachineMemOperand::MOStore, useAM3); |
| return true; |
| } |
| |
| bool ARMFastISel::SelectStore(const Instruction *I) { |
| Value *Op0 = I->getOperand(0); |
| unsigned SrcReg = 0; |
| |
| // Atomic stores need special handling. |
| if (cast<StoreInst>(I)->isAtomic()) |
| return false; |
| |
| const Value *PtrV = I->getOperand(1); |
| if (TLI.supportSwiftError()) { |
| // Swifterror values can come from either a function parameter with |
| // swifterror attribute or an alloca with swifterror attribute. |
| if (const Argument *Arg = dyn_cast<Argument>(PtrV)) { |
| if (Arg->hasSwiftErrorAttr()) |
| return false; |
| } |
| |
| if (const AllocaInst *Alloca = dyn_cast<AllocaInst>(PtrV)) { |
| if (Alloca->isSwiftError()) |
| return false; |
| } |
| } |
| |
| // Verify we have a legal type before going any further. |
| MVT VT; |
| if (!isLoadTypeLegal(I->getOperand(0)->getType(), VT)) |
| return false; |
| |
| // Get the value to be stored into a register. |
| SrcReg = getRegForValue(Op0); |
| if (SrcReg == 0) return false; |
| |
| // See if we can handle this address. |
| Address Addr; |
| if (!ARMComputeAddress(I->getOperand(1), Addr)) |
| return false; |
| |
| if (!ARMEmitStore(VT, SrcReg, Addr, cast<StoreInst>(I)->getAlignment())) |
| return false; |
| return true; |
| } |
| |
| static ARMCC::CondCodes getComparePred(CmpInst::Predicate Pred) { |
| switch (Pred) { |
| // Needs two compares... |
| case CmpInst::FCMP_ONE: |
| case CmpInst::FCMP_UEQ: |
| default: |
| // AL is our "false" for now. The other two need more compares. |
| return ARMCC::AL; |
| case CmpInst::ICMP_EQ: |
| case CmpInst::FCMP_OEQ: |
| return ARMCC::EQ; |
| case CmpInst::ICMP_SGT: |
| case CmpInst::FCMP_OGT: |
| return ARMCC::GT; |
| case CmpInst::ICMP_SGE: |
| case CmpInst::FCMP_OGE: |
| return ARMCC::GE; |
| case CmpInst::ICMP_UGT: |
| case CmpInst::FCMP_UGT: |
| return ARMCC::HI; |
| case CmpInst::FCMP_OLT: |
| return ARMCC::MI; |
| case CmpInst::ICMP_ULE: |
| case CmpInst::FCMP_OLE: |
| return ARMCC::LS; |
| case CmpInst::FCMP_ORD: |
| return ARMCC::VC; |
| case CmpInst::FCMP_UNO: |
| return ARMCC::VS; |
| case CmpInst::FCMP_UGE: |
| return ARMCC::PL; |
| case CmpInst::ICMP_SLT: |
| case CmpInst::FCMP_ULT: |
| return ARMCC::LT; |
| case CmpInst::ICMP_SLE: |
| case CmpInst::FCMP_ULE: |
| return ARMCC::LE; |
| case CmpInst::FCMP_UNE: |
| case CmpInst::ICMP_NE: |
| return ARMCC::NE; |
| case CmpInst::ICMP_UGE: |
| return ARMCC::HS; |
| case CmpInst::ICMP_ULT: |
| return ARMCC::LO; |
| } |
| } |
| |
| bool ARMFastISel::SelectBranch(const Instruction *I) { |
| const BranchInst *BI = cast<BranchInst>(I); |
| MachineBasicBlock *TBB = FuncInfo.MBBMap[BI->getSuccessor(0)]; |
| MachineBasicBlock *FBB = FuncInfo.MBBMap[BI->getSuccessor(1)]; |
| |
| // Simple branch support. |
| |
| // If we can, avoid recomputing the compare - redoing it could lead to wonky |
| // behavior. |
| if (const CmpInst *CI = dyn_cast<CmpInst>(BI->getCondition())) { |
| if (CI->hasOneUse() && (CI->getParent() == I->getParent())) { |
| // Get the compare predicate. |
| // Try to take advantage of fallthrough opportunities. |
| CmpInst::Predicate Predicate = CI->getPredicate(); |
| if (FuncInfo.MBB->isLayoutSuccessor(TBB)) { |
| std::swap(TBB, FBB); |
| Predicate = CmpInst::getInversePredicate(Predicate); |
| } |
| |
| ARMCC::CondCodes ARMPred = getComparePred(Predicate); |
| |
| // We may not handle every CC for now. |
| if (ARMPred == ARMCC::AL) return false; |
| |
| // Emit the compare. |
| if (!ARMEmitCmp(CI->getOperand(0), CI->getOperand(1), CI->isUnsigned(), |
| CI->isEquality())) |
| return false; |
| |
| unsigned BrOpc = isThumb2 ? ARM::t2Bcc : ARM::Bcc; |
| BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc, TII.get(BrOpc)) |
| .addMBB(TBB).addImm(ARMPred).addReg(ARM::CPSR); |
| finishCondBranch(BI->getParent(), TBB, FBB); |
| return true; |
| } |
| } else if (TruncInst *TI = dyn_cast<TruncInst>(BI->getCondition())) { |
| MVT SourceVT; |
| if (TI->hasOneUse() && TI->getParent() == I->getParent() && |
| (isLoadTypeLegal(TI->getOperand(0)->getType(), SourceVT))) { |
| unsigned TstOpc = isThumb2 ? ARM::t2TSTri : ARM::TSTri; |
| unsigned OpReg = getRegForValue(TI->getOperand(0)); |
| OpReg = constrainOperandRegClass(TII.get(TstOpc), OpReg, 0); |
| AddOptionalDefs(BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc, |
| TII.get(TstOpc)) |
| .addReg(OpReg).addImm(1)); |
| |
| unsigned CCMode = ARMCC::NE; |
| if (FuncInfo.MBB->isLayoutSuccessor(TBB)) { |
| std::swap(TBB, FBB); |
| CCMode = ARMCC::EQ; |
| } |
| |
| unsigned BrOpc = isThumb2 ? ARM::t2Bcc : ARM::Bcc; |
| BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc, TII.get(BrOpc)) |
| .addMBB(TBB).addImm(CCMode).addReg(ARM::CPSR); |
| |
| finishCondBranch(BI->getParent(), TBB, FBB); |
| return true; |
| } |
| } else if (const ConstantInt *CI = |
| dyn_cast<ConstantInt>(BI->getCondition())) { |
| uint64_t Imm = CI->getZExtValue(); |
| MachineBasicBlock *Target = (Imm == 0) ? FBB : TBB; |
| fastEmitBranch(Target, DbgLoc); |
| return true; |
| } |
| |
| unsigned CmpReg = getRegForValue(BI->getCondition()); |
| if (CmpReg == 0) return false; |
| |
| // We've been divorced from our compare! Our block was split, and |
| // now our compare lives in a predecessor block. We musn't |
| // re-compare here, as the children of the compare aren't guaranteed |
| // live across the block boundary (we *could* check for this). |
| // Regardless, the compare has been done in the predecessor block, |
| // and it left a value for us in a virtual register. Ergo, we test |
| // the one-bit value left in the virtual register. |
| unsigned TstOpc = isThumb2 ? ARM::t2TSTri : ARM::TSTri; |
| CmpReg = constrainOperandRegClass(TII.get(TstOpc), CmpReg, 0); |
| AddOptionalDefs( |
| BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc, TII.get(TstOpc)) |
| .addReg(CmpReg) |
| .addImm(1)); |
| |
| unsigned CCMode = ARMCC::NE; |
| if (FuncInfo.MBB->isLayoutSuccessor(TBB)) { |
| std::swap(TBB, FBB); |
| CCMode = ARMCC::EQ; |
| } |
| |
| unsigned BrOpc = isThumb2 ? ARM::t2Bcc : ARM::Bcc; |
| BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc, TII.get(BrOpc)) |
| .addMBB(TBB).addImm(CCMode).addReg(ARM::CPSR); |
| finishCondBranch(BI->getParent(), TBB, FBB); |
| return true; |
| } |
| |
| bool ARMFastISel::SelectIndirectBr(const Instruction *I) { |
| unsigned AddrReg = getRegForValue(I->getOperand(0)); |
| if (AddrReg == 0) return false; |
| |
| unsigned Opc = isThumb2 ? ARM::tBRIND : ARM::BX; |
| assert(isThumb2 || Subtarget->hasV4TOps()); |
| |
| AddOptionalDefs(BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc, |
| TII.get(Opc)).addReg(AddrReg)); |
| |
| const IndirectBrInst *IB = cast<IndirectBrInst>(I); |
| for (const BasicBlock *SuccBB : IB->successors()) |
| FuncInfo.MBB->addSuccessor(FuncInfo.MBBMap[SuccBB]); |
| |
| return true; |
| } |
| |
| bool ARMFastISel::ARMEmitCmp(const Value *Src1Value, const Value *Src2Value, |
| bool isZExt, bool isEquality) { |
| Type *Ty = Src1Value->getType(); |
| EVT SrcEVT = TLI.getValueType(DL, Ty, true); |
| if (!SrcEVT.isSimple()) return false; |
| MVT SrcVT = SrcEVT.getSimpleVT(); |
| |
| if (Ty->isFloatTy() && !Subtarget->hasVFP2()) |
| return false; |
| |
| if (Ty->isDoubleTy() && (!Subtarget->hasVFP2() || Subtarget->isFPOnlySP())) |
| return false; |
| |
| // Check to see if the 2nd operand is a constant that we can encode directly |
| // in the compare. |
| int Imm = 0; |
| bool UseImm = false; |
| bool isNegativeImm = false; |
| // FIXME: At -O0 we don't have anything that canonicalizes operand order. |
| // Thus, Src1Value may be a ConstantInt, but we're missing it. |
| if (const ConstantInt *ConstInt = dyn_cast<ConstantInt>(Src2Value)) { |
| if (SrcVT == MVT::i32 || SrcVT == MVT::i16 || SrcVT == MVT::i8 || |
| SrcVT == MVT::i1) { |
| const APInt &CIVal = ConstInt->getValue(); |
| Imm = (isZExt) ? (int)CIVal.getZExtValue() : (int)CIVal.getSExtValue(); |
| // For INT_MIN/LONG_MIN (i.e., 0x80000000) we need to use a cmp, rather |
| // then a cmn, because there is no way to represent 2147483648 as a |
| // signed 32-bit int. |
| if (Imm < 0 && Imm != (int)0x80000000) { |
| isNegativeImm = true; |
| Imm = -Imm; |
| } |
| UseImm = isThumb2 ? (ARM_AM::getT2SOImmVal(Imm) != -1) : |
| (ARM_AM::getSOImmVal(Imm) != -1); |
| } |
| } else if (const ConstantFP *ConstFP = dyn_cast<ConstantFP>(Src2Value)) { |
| if (SrcVT == MVT::f32 || SrcVT == MVT::f64) |
| if (ConstFP->isZero() && !ConstFP->isNegative()) |
| UseImm = true; |
| } |
| |
| unsigned CmpOpc; |
| bool isICmp = true; |
| bool needsExt = false; |
| switch (SrcVT.SimpleTy) { |
| default: return false; |
| // TODO: Verify compares. |
| case MVT::f32: |
| isICmp = false; |
| // Equality comparisons shouldn't raise Invalid on uordered inputs. |
| if (isEquality) |
| CmpOpc = UseImm ? ARM::VCMPZS : ARM::VCMPS; |
| else |
| CmpOpc = UseImm ? ARM::VCMPEZS : ARM::VCMPES; |
| break; |
| case MVT::f64: |
| isICmp = false; |
| // Equality comparisons shouldn't raise Invalid on uordered inputs. |
| if (isEquality) |
| CmpOpc = UseImm ? ARM::VCMPZD : ARM::VCMPD; |
| else |
| CmpOpc = UseImm ? ARM::VCMPEZD : ARM::VCMPED; |
| break; |
| case MVT::i1: |
| case MVT::i8: |
| case MVT::i16: |
| needsExt = true; |
| LLVM_FALLTHROUGH; |
| case MVT::i32: |
| if (isThumb2) { |
| if (!UseImm) |
| CmpOpc = ARM::t2CMPrr; |
| else |
| CmpOpc = isNegativeImm ? ARM::t2CMNri : ARM::t2CMPri; |
| } else { |
| if (!UseImm) |
| CmpOpc = ARM::CMPrr; |
| else |
| CmpOpc = isNegativeImm ? ARM::CMNri : ARM::CMPri; |
| } |
| break; |
| } |
| |
| unsigned SrcReg1 = getRegForValue(Src1Value); |
| if (SrcReg1 == 0) return false; |
| |
| unsigned SrcReg2 = 0; |
| if (!UseImm) { |
| SrcReg2 = getRegForValue(Src2Value); |
| if (SrcReg2 == 0) return false; |
| } |
| |
| // We have i1, i8, or i16, we need to either zero extend or sign extend. |
| if (needsExt) { |
| SrcReg1 = ARMEmitIntExt(SrcVT, SrcReg1, MVT::i32, isZExt); |
| if (SrcReg1 == 0) return false; |
| if (!UseImm) { |
| SrcReg2 = ARMEmitIntExt(SrcVT, SrcReg2, MVT::i32, isZExt); |
| if (SrcReg2 == 0) return false; |
| } |
| } |
| |
| const MCInstrDesc &II = TII.get(CmpOpc); |
| SrcReg1 = constrainOperandRegClass(II, SrcReg1, 0); |
| if (!UseImm) { |
| SrcReg2 = constrainOperandRegClass(II, SrcReg2, 1); |
| AddOptionalDefs(BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc, II) |
| .addReg(SrcReg1).addReg(SrcReg2)); |
| } else { |
| MachineInstrBuilder MIB; |
| MIB = BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc, II) |
| .addReg(SrcReg1); |
| |
| // Only add immediate for icmp as the immediate for fcmp is an implicit 0.0. |
| if (isICmp) |
| MIB.addImm(Imm); |
| AddOptionalDefs(MIB); |
| } |
| |
| // For floating point we need to move the result to a comparison register |
| // that we can then use for branches. |
| if (Ty->isFloatTy() || Ty->isDoubleTy()) |
| AddOptionalDefs(BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc, |
| TII.get(ARM::FMSTAT))); |
| return true; |
| } |
| |
| bool ARMFastISel::SelectCmp(const Instruction *I) { |
| const CmpInst *CI = cast<CmpInst>(I); |
| |
| // Get the compare predicate. |
| ARMCC::CondCodes ARMPred = getComparePred(CI->getPredicate()); |
| |
| // We may not handle every CC for now. |
| if (ARMPred == ARMCC::AL) return false; |
| |
| // Emit the compare. |
| if (!ARMEmitCmp(CI->getOperand(0), CI->getOperand(1), CI->isUnsigned(), |
| CI->isEquality())) |
| return false; |
| |
| // Now set a register based on the comparison. Explicitly set the predicates |
| // here. |
| unsigned MovCCOpc = isThumb2 ? ARM::t2MOVCCi : ARM::MOVCCi; |
| const TargetRegisterClass *RC = isThumb2 ? &ARM::rGPRRegClass |
| : &ARM::GPRRegClass; |
| unsigned DestReg = createResultReg(RC); |
| Constant *Zero = ConstantInt::get(Type::getInt32Ty(*Context), 0); |
| unsigned ZeroReg = fastMaterializeConstant(Zero); |
| // ARMEmitCmp emits a FMSTAT when necessary, so it's always safe to use CPSR. |
| BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc, TII.get(MovCCOpc), DestReg) |
| .addReg(ZeroReg).addImm(1) |
| .addImm(ARMPred).addReg(ARM::CPSR); |
| |
| updateValueMap(I, DestReg); |
| return true; |
| } |
| |
| bool ARMFastISel::SelectFPExt(const Instruction *I) { |
| // Make sure we have VFP and that we're extending float to double. |
| if (!Subtarget->hasVFP2() || Subtarget->isFPOnlySP()) return false; |
| |
| Value *V = I->getOperand(0); |
| if (!I->getType()->isDoubleTy() || |
| !V->getType()->isFloatTy()) return false; |
| |
| unsigned Op = getRegForValue(V); |
| if (Op == 0) return false; |
| |
| unsigned Result = createResultReg(&ARM::DPRRegClass); |
| AddOptionalDefs(BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc, |
| TII.get(ARM::VCVTDS), Result) |
| .addReg(Op)); |
| updateValueMap(I, Result); |
| return true; |
| } |
| |
| bool ARMFastISel::SelectFPTrunc(const Instruction *I) { |
| // Make sure we have VFP and that we're truncating double to float. |
| if (!Subtarget->hasVFP2() || Subtarget->isFPOnlySP()) return false; |
| |
| Value *V = I->getOperand(0); |
| if (!(I->getType()->isFloatTy() && |
| V->getType()->isDoubleTy())) return false; |
| |
| unsigned Op = getRegForValue(V); |
| if (Op == 0) return false; |
| |
| unsigned Result = createResultReg(&ARM::SPRRegClass); |
| AddOptionalDefs(BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc, |
| TII.get(ARM::VCVTSD), Result) |
| .addReg(Op)); |
| updateValueMap(I, Result); |
| return true; |
| } |
| |
| bool ARMFastISel::SelectIToFP(const Instruction *I, bool isSigned) { |
| // Make sure we have VFP. |
| if (!Subtarget->hasVFP2()) return false; |
| |
| MVT DstVT; |
| Type *Ty = I->getType(); |
| if (!isTypeLegal(Ty, DstVT)) |
| return false; |
| |
| Value *Src = I->getOperand(0); |
| EVT SrcEVT = TLI.getValueType(DL, Src->getType(), true); |
| if (!SrcEVT.isSimple()) |
| return false; |
| MVT SrcVT = SrcEVT.getSimpleVT(); |
| if (SrcVT != MVT::i32 && SrcVT != MVT::i16 && SrcVT != MVT::i8) |
| return false; |
| |
| unsigned SrcReg = getRegForValue(Src); |
| if (SrcReg == 0) return false; |
| |
| // Handle sign-extension. |
| if (SrcVT == MVT::i16 || SrcVT == MVT::i8) { |
| SrcReg = ARMEmitIntExt(SrcVT, SrcReg, MVT::i32, |
| /*isZExt*/!isSigned); |
| if (SrcReg == 0) return false; |
| } |
| |
| // The conversion routine works on fp-reg to fp-reg and the operand above |
| // was an integer, move it to the fp registers if possible. |
| unsigned FP = ARMMoveToFPReg(MVT::f32, SrcReg); |
| if (FP == 0) return false; |
| |
| unsigned Opc; |
| if (Ty->isFloatTy()) Opc = isSigned ? ARM::VSITOS : ARM::VUITOS; |
| else if (Ty->isDoubleTy() && !Subtarget->isFPOnlySP()) |
| Opc = isSigned ? ARM::VSITOD : ARM::VUITOD; |
| else return false; |
| |
| unsigned ResultReg = createResultReg(TLI.getRegClassFor(DstVT)); |
| AddOptionalDefs(BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc, |
| TII.get(Opc), ResultReg).addReg(FP)); |
| updateValueMap(I, ResultReg); |
| return true; |
| } |
| |
| bool ARMFastISel::SelectFPToI(const Instruction *I, bool isSigned) { |
| // Make sure we have VFP. |
| if (!Subtarget->hasVFP2()) return false; |
| |
| MVT DstVT; |
| Type *RetTy = I->getType(); |
| if (!isTypeLegal(RetTy, DstVT)) |
| return false; |
| |
| unsigned Op = getRegForValue(I->getOperand(0)); |
| if (Op == 0) return false; |
| |
| unsigned Opc; |
| Type *OpTy = I->getOperand(0)->getType(); |
| if (OpTy->isFloatTy()) Opc = isSigned ? ARM::VTOSIZS : ARM::VTOUIZS; |
| else if (OpTy->isDoubleTy() && !Subtarget->isFPOnlySP()) |
| Opc = isSigned ? ARM::VTOSIZD : ARM::VTOUIZD; |
| else return false; |
| |
| // f64->s32/u32 or f32->s32/u32 both need an intermediate f32 reg. |
| unsigned ResultReg = createResultReg(TLI.getRegClassFor(MVT::f32)); |
| AddOptionalDefs(BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc, |
| TII.get(Opc), ResultReg).addReg(Op)); |
| |
| // This result needs to be in an integer register, but the conversion only |
| // takes place in fp-regs. |
| unsigned IntReg = ARMMoveToIntReg(DstVT, ResultReg); |
| if (IntReg == 0) return false; |
| |
| updateValueMap(I, IntReg); |
| return true; |
| } |
| |
| bool ARMFastISel::SelectSelect(const Instruction *I) { |
| MVT VT; |
| if (!isTypeLegal(I->getType(), VT)) |
| return false; |
| |
| // Things need to be register sized for register moves. |
| if (VT != MVT::i32) return false; |
| |
| unsigned CondReg = getRegForValue(I->getOperand(0)); |
| if (CondReg == 0) return false; |
| unsigned Op1Reg = getRegForValue(I->getOperand(1)); |
| if (Op1Reg == 0) return false; |
| |
| // Check to see if we can use an immediate in the conditional move. |
| int Imm = 0; |
| bool UseImm = false; |
| bool isNegativeImm = false; |
| if (const ConstantInt *ConstInt = dyn_cast<ConstantInt>(I->getOperand(2))) { |
| assert(VT == MVT::i32 && "Expecting an i32."); |
| Imm = (int)ConstInt->getValue().getZExtValue(); |
| if (Imm < 0) { |
| isNegativeImm = true; |
| Imm = ~Imm; |
| } |
| UseImm = isThumb2 ? (ARM_AM::getT2SOImmVal(Imm) != -1) : |
| (ARM_AM::getSOImmVal(Imm) != -1); |
| } |
| |
| unsigned Op2Reg = 0; |
| if (!UseImm) { |
| Op2Reg = getRegForValue(I->getOperand(2)); |
| if (Op2Reg == 0) return false; |
| } |
| |
| unsigned TstOpc = isThumb2 ? ARM::t2TSTri : ARM::TSTri; |
| CondReg = constrainOperandRegClass(TII.get(TstOpc), CondReg, 0); |
| AddOptionalDefs( |
| BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc, TII.get(TstOpc)) |
| .addReg(CondReg) |
| .addImm(1)); |
| |
| unsigned MovCCOpc; |
| const TargetRegisterClass *RC; |
| if (!UseImm) { |
| RC = isThumb2 ? &ARM::tGPRRegClass : &ARM::GPRRegClass; |
| MovCCOpc = isThumb2 ? ARM::t2MOVCCr : ARM::MOVCCr; |
| } else { |
| RC = isThumb2 ? &ARM::rGPRRegClass : &ARM::GPRRegClass; |
| if (!isNegativeImm) |
| MovCCOpc = isThumb2 ? ARM::t2MOVCCi : ARM::MOVCCi; |
| else |
| MovCCOpc = isThumb2 ? ARM::t2MVNCCi : ARM::MVNCCi; |
| } |
| unsigned ResultReg = createResultReg(RC); |
| if (!UseImm) { |
| Op2Reg = constrainOperandRegClass(TII.get(MovCCOpc), Op2Reg, 1); |
| Op1Reg = constrainOperandRegClass(TII.get(MovCCOpc), Op1Reg, 2); |
| BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc, TII.get(MovCCOpc), |
| ResultReg) |
| .addReg(Op2Reg) |
| .addReg(Op1Reg) |
| .addImm(ARMCC::NE) |
| .addReg(ARM::CPSR); |
| } else { |
| Op1Reg = constrainOperandRegClass(TII.get(MovCCOpc), Op1Reg, 1); |
| BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc, TII.get(MovCCOpc), |
| ResultReg) |
| .addReg(Op1Reg) |
| .addImm(Imm) |
| .addImm(ARMCC::EQ) |
| .addReg(ARM::CPSR); |
| } |
| updateValueMap(I, ResultReg); |
| return true; |
| } |
| |
| bool ARMFastISel::SelectDiv(const Instruction *I, bool isSigned) { |
| MVT VT; |
| Type *Ty = I->getType(); |
| if (!isTypeLegal(Ty, VT)) |
| return false; |
| |
| // If we have integer div support we should have selected this automagically. |
| // In case we have a real miss go ahead and return false and we'll pick |
| // it up later. |
| if (Subtarget->hasDivideInThumbMode()) |
| return false; |
| |
| // Otherwise emit a libcall. |
| RTLIB::Libcall LC = RTLIB::UNKNOWN_LIBCALL; |
| if (VT == MVT::i8) |
| LC = isSigned ? RTLIB::SDIV_I8 : RTLIB::UDIV_I8; |
| else if (VT == MVT::i16) |
| LC = isSigned ? RTLIB::SDIV_I16 : RTLIB::UDIV_I16; |
| else if (VT == MVT::i32) |
| LC = isSigned ? RTLIB::SDIV_I32 : RTLIB::UDIV_I32; |
| else if (VT == MVT::i64) |
| LC = isSigned ? RTLIB::SDIV_I64 : RTLIB::UDIV_I64; |
| else if (VT == MVT::i128) |
| LC = isSigned ? RTLIB::SDIV_I128 : RTLIB::UDIV_I128; |
| assert(LC != RTLIB::UNKNOWN_LIBCALL && "Unsupported SDIV!"); |
| |
| return ARMEmitLibcall(I, LC); |
| } |
| |
| bool ARMFastISel::SelectRem(const Instruction *I, bool isSigned) { |
| MVT VT; |
| Type *Ty = I->getType(); |
| if (!isTypeLegal(Ty, VT)) |
| return false; |
| |
| // Many ABIs do not provide a libcall for standalone remainder, so we need to |
| // use divrem (see the RTABI 4.3.1). Since FastISel can't handle non-double |
| // multi-reg returns, we'll have to bail out. |
| if (!TLI.hasStandaloneRem(VT)) { |
| return false; |
| } |
| |
| RTLIB::Libcall LC = RTLIB::UNKNOWN_LIBCALL; |
| if (VT == MVT::i8) |
| LC = isSigned ? RTLIB::SREM_I8 : RTLIB::UREM_I8; |
| else if (VT == MVT::i16) |
| LC = isSigned ? RTLIB::SREM_I16 : RTLIB::UREM_I16; |
| else if (VT == MVT::i32) |
| LC = isSigned ? RTLIB::SREM_I32 : RTLIB::UREM_I32; |
| else if (VT == MVT::i64) |
| LC = isSigned ? RTLIB::SREM_I64 : RTLIB::UREM_I64; |
| else if (VT == MVT::i128) |
| LC = isSigned ? RTLIB::SREM_I128 : RTLIB::UREM_I128; |
| assert(LC != RTLIB::UNKNOWN_LIBCALL && "Unsupported SREM!"); |
| |
| return ARMEmitLibcall(I, LC); |
| } |
| |
| bool ARMFastISel::SelectBinaryIntOp(const Instruction *I, unsigned ISDOpcode) { |
| EVT DestVT = TLI.getValueType(DL, I->getType(), true); |
| |
| // We can get here in the case when we have a binary operation on a non-legal |
| // type and the target independent selector doesn't know how to handle it. |
| if (DestVT != MVT::i16 && DestVT != MVT::i8 && DestVT != MVT::i1) |
| return false; |
| |
| unsigned Opc; |
| switch (ISDOpcode) { |
| default: return false; |
| case ISD::ADD: |
| Opc = isThumb2 ? ARM::t2ADDrr : ARM::ADDrr; |
| break; |
| case ISD::OR: |
| Opc = isThumb2 ? ARM::t2ORRrr : ARM::ORRrr; |
| break; |
| case ISD::SUB: |
| Opc = isThumb2 ? ARM::t2SUBrr : ARM::SUBrr; |
| break; |
| } |
| |
| unsigned SrcReg1 = getRegForValue(I->getOperand(0)); |
| if (SrcReg1 == 0) return false; |
| |
| // TODO: Often the 2nd operand is an immediate, which can be encoded directly |
| // in the instruction, rather then materializing the value in a register. |
| unsigned SrcReg2 = getRegForValue(I->getOperand(1)); |
| if (SrcReg2 == 0) return false; |
| |
| unsigned ResultReg = createResultReg(&ARM::GPRnopcRegClass); |
| SrcReg1 = constrainOperandRegClass(TII.get(Opc), SrcReg1, 1); |
| SrcReg2 = constrainOperandRegClass(TII.get(Opc), SrcReg2, 2); |
| AddOptionalDefs(BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc, |
| TII.get(Opc), ResultReg) |
| .addReg(SrcReg1).addReg(SrcReg2)); |
| updateValueMap(I, ResultReg); |
| return true; |
| } |
| |
| bool ARMFastISel::SelectBinaryFPOp(const Instruction *I, unsigned ISDOpcode) { |
| EVT FPVT = TLI.getValueType(DL, I->getType(), true); |
| if (!FPVT.isSimple()) return false; |
| MVT VT = FPVT.getSimpleVT(); |
| |
| // FIXME: Support vector types where possible. |
| if (VT.isVector()) |
| return false; |
| |
| // We can get here in the case when we want to use NEON for our fp |
| // operations, but can't figure out how to. Just use the vfp instructions |
| // if we have them. |
| // FIXME: It'd be nice to use NEON instructions. |
| Type *Ty = I->getType(); |
| if (Ty->isFloatTy() && !Subtarget->hasVFP2()) |
| return false; |
| if (Ty->isDoubleTy() && (!Subtarget->hasVFP2() || Subtarget->isFPOnlySP())) |
| return false; |
| |
| unsigned Opc; |
| bool is64bit = VT == MVT::f64 || VT == MVT::i64; |
| switch (ISDOpcode) { |
| default: return false; |
| case ISD::FADD: |
| Opc = is64bit ? ARM::VADDD : ARM::VADDS; |
| break; |
| case ISD::FSUB: |
| Opc = is64bit ? ARM::VSUBD : ARM::VSUBS; |
| break; |
| case ISD::FMUL: |
| Opc = is64bit ? ARM::VMULD : ARM::VMULS; |
| break; |
| } |
| unsigned Op1 = getRegForValue(I->getOperand(0)); |
| if (Op1 == 0) return false; |
| |
| unsigned Op2 = getRegForValue(I->getOperand(1)); |
| if (Op2 == 0) return false; |
| |
| unsigned ResultReg = createResultReg(TLI.getRegClassFor(VT.SimpleTy)); |
| AddOptionalDefs(BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc, |
| TII.get(Opc), ResultReg) |
| .addReg(Op1).addReg(Op2)); |
| updateValueMap(I, ResultReg); |
| return true; |
| } |
| |
| // Call Handling Code |
| |
| // This is largely taken directly from CCAssignFnForNode |
| // TODO: We may not support all of this. |
| CCAssignFn *ARMFastISel::CCAssignFnForCall(CallingConv::ID CC, |
| bool Return, |
| bool isVarArg) { |
| switch (CC) { |
| default: |
| report_fatal_error("Unsupported calling convention"); |
| case CallingConv::Fast: |
| if (Subtarget->hasVFP2() && !isVarArg) { |
| if (!Subtarget->isAAPCS_ABI()) |
| return (Return ? RetFastCC_ARM_APCS : FastCC_ARM_APCS); |
| // For AAPCS ABI targets, just use VFP variant of the calling convention. |
| return (Return ? RetCC_ARM_AAPCS_VFP : CC_ARM_AAPCS_VFP); |
| } |
| LLVM_FALLTHROUGH; |
| case CallingConv::C: |
| case CallingConv::CXX_FAST_TLS: |
| // Use target triple & subtarget features to do actual dispatch. |
| if (Subtarget->isAAPCS_ABI()) { |
| if (Subtarget->hasVFP2() && |
| TM.Options.FloatABIType == FloatABI::Hard && !isVarArg) |
| return (Return ? RetCC_ARM_AAPCS_VFP: CC_ARM_AAPCS_VFP); |
| else |
| return (Return ? RetCC_ARM_AAPCS: CC_ARM_AAPCS); |
| } else { |
| return (Return ? RetCC_ARM_APCS: CC_ARM_APCS); |
| } |
| case CallingConv::ARM_AAPCS_VFP: |
| case CallingConv::Swift: |
| if (!isVarArg) |
| return (Return ? RetCC_ARM_AAPCS_VFP: CC_ARM_AAPCS_VFP); |
| // Fall through to soft float variant, variadic functions don't |
| // use hard floating point ABI. |
| LLVM_FALLTHROUGH; |
| case CallingConv::ARM_AAPCS: |
| return (Return ? RetCC_ARM_AAPCS: CC_ARM_AAPCS); |
| case CallingConv::ARM_APCS: |
| return (Return ? RetCC_ARM_APCS: CC_ARM_APCS); |
| case CallingConv::GHC: |
| if (Return) |
| report_fatal_error("Can't return in GHC call convention"); |
| else |
| return CC_ARM_APCS_GHC; |
| } |
| } |
| |
| bool ARMFastISel::ProcessCallArgs(SmallVectorImpl<Value*> &Args, |
| SmallVectorImpl<unsigned> &ArgRegs, |
| SmallVectorImpl<MVT> &ArgVTs, |
| SmallVectorImpl<ISD::ArgFlagsTy> &ArgFlags, |
| SmallVectorImpl<unsigned> &RegArgs, |
| CallingConv::ID CC, |
| unsigned &NumBytes, |
| bool isVarArg) { |
| SmallVector<CCValAssign, 16> ArgLocs; |
| CCState CCInfo(CC, isVarArg, *FuncInfo.MF, ArgLocs, *Context); |
| CCInfo.AnalyzeCallOperands(ArgVTs, ArgFlags, |
| CCAssignFnForCall(CC, false, isVarArg)); |
| |
| // Check that we can handle all of the arguments. If we can't, then bail out |
| // now before we add code to the MBB. |
| for (unsigned i = 0, e = ArgLocs.size(); i != e; ++i) { |
| CCValAssign &VA = ArgLocs[i]; |
| MVT ArgVT = ArgVTs[VA.getValNo()]; |
| |
| // We don't handle NEON/vector parameters yet. |
| if (ArgVT.isVector() || ArgVT.getSizeInBits() > 64) |
| return false; |
| |
| // Now copy/store arg to correct locations. |
| if (VA.isRegLoc() && !VA.needsCustom()) { |
| continue; |
| } else if (VA.needsCustom()) { |
| // TODO: We need custom lowering for vector (v2f64) args. |
| if (VA.getLocVT() != MVT::f64 || |
| // TODO: Only handle register args for now. |
| !VA.isRegLoc() || !ArgLocs[++i].isRegLoc()) |
| return false; |
| } else { |
| switch (ArgVT.SimpleTy) { |
| default: |
| return false; |
| case MVT::i1: |
| case MVT::i8: |
| case MVT::i16: |
| case MVT::i32: |
| break; |
| case MVT::f32: |
| if (!Subtarget->hasVFP2()) |
| return false; |
| break; |
| case MVT::f64: |
| if (!Subtarget->hasVFP2()) |
| return false; |
| break; |
| } |
| } |
| } |
| |
| // At the point, we are able to handle the call's arguments in fast isel. |
| |
| // Get a count of how many bytes are to be pushed on the stack. |
| NumBytes = CCInfo.getNextStackOffset(); |
| |
| // Issue CALLSEQ_START |
| unsigned AdjStackDown = TII.getCallFrameSetupOpcode(); |
| AddOptionalDefs(BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc, |
| TII.get(AdjStackDown)) |
| .addImm(NumBytes).addImm(0)); |
| |
| // Process the args. |
| for (unsigned i = 0, e = ArgLocs.size(); i != e; ++i) { |
| CCValAssign &VA = ArgLocs[i]; |
| const Value *ArgVal = Args[VA.getValNo()]; |
| unsigned Arg = ArgRegs[VA.getValNo()]; |
| MVT ArgVT = ArgVTs[VA.getValNo()]; |
| |
| assert((!ArgVT.isVector() && ArgVT.getSizeInBits() <= 64) && |
| "We don't handle NEON/vector parameters yet."); |
| |
| // Handle arg promotion, etc. |
| switch (VA.getLocInfo()) { |
| case CCValAssign::Full: break; |
| case CCValAssign::SExt: { |
| MVT DestVT = VA.getLocVT(); |
| Arg = ARMEmitIntExt(ArgVT, Arg, DestVT, /*isZExt*/false); |
| assert(Arg != 0 && "Failed to emit a sext"); |
| ArgVT = DestVT; |
| break; |
| } |
| case CCValAssign::AExt: |
| // Intentional fall-through. Handle AExt and ZExt. |
| case CCValAssign::ZExt: { |
| MVT DestVT = VA.getLocVT(); |
| Arg = ARMEmitIntExt(ArgVT, Arg, DestVT, /*isZExt*/true); |
| assert(Arg != 0 && "Failed to emit a zext"); |
| ArgVT = DestVT; |
| break; |
| } |
| case CCValAssign::BCvt: { |
| unsigned BC = fastEmit_r(ArgVT, VA.getLocVT(), ISD::BITCAST, Arg, |
| /*TODO: Kill=*/false); |
| assert(BC != 0 && "Failed to emit a bitcast!"); |
| Arg = BC; |
| ArgVT = VA.getLocVT(); |
| break; |
| } |
| default: llvm_unreachable("Unknown arg promotion!"); |
| } |
| |
| // Now copy/store arg to correct locations. |
| if (VA.isRegLoc() && !VA.needsCustom()) { |
| BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc, |
| TII.get(TargetOpcode::COPY), VA.getLocReg()).addReg(Arg); |
| RegArgs.push_back(VA.getLocReg()); |
| } else if (VA.needsCustom()) { |
| // TODO: We need custom lowering for vector (v2f64) args. |
| assert(VA.getLocVT() == MVT::f64 && |
| "Custom lowering for v2f64 args not available"); |
| |
| // FIXME: ArgLocs[++i] may extend beyond ArgLocs.size() |
| CCValAssign &NextVA = ArgLocs[++i]; |
| |
| assert(VA.isRegLoc() && NextVA.isRegLoc() && |
| "We only handle register args!"); |
| |
| AddOptionalDefs(BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc, |
| TII.get(ARM::VMOVRRD), VA.getLocReg()) |
| .addReg(NextVA.getLocReg(), RegState::Define) |
| .addReg(Arg)); |
| RegArgs.push_back(VA.getLocReg()); |
| RegArgs.push_back(NextVA.getLocReg()); |
| } else { |
| assert(VA.isMemLoc()); |
| // Need to store on the stack. |
| |
| // Don't emit stores for undef values. |
| if (isa<UndefValue>(ArgVal)) |
| continue; |
| |
| Address Addr; |
| Addr.BaseType = Address::RegBase; |
| Addr.Base.Reg = ARM::SP; |
| Addr.Offset = VA.getLocMemOffset(); |
| |
| bool EmitRet = ARMEmitStore(ArgVT, Arg, Addr); (void)EmitRet; |
| assert(EmitRet && "Could not emit a store for argument!"); |
| } |
| } |
| |
| return true; |
| } |
| |
| bool ARMFastISel::FinishCall(MVT RetVT, SmallVectorImpl<unsigned> &UsedRegs, |
| const Instruction *I, CallingConv::ID CC, |
| unsigned &NumBytes, bool isVarArg) { |
| // Issue CALLSEQ_END |
| unsigned AdjStackUp = TII.getCallFrameDestroyOpcode(); |
| AddOptionalDefs(BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc, |
| TII.get(AdjStackUp)) |
| .addImm(NumBytes).addImm(0)); |
| |
| // Now the return value. |
| if (RetVT != MVT::isVoid) { |
| SmallVector<CCValAssign, 16> RVLocs; |
| CCState CCInfo(CC, isVarArg, *FuncInfo.MF, RVLocs, *Context); |
| CCInfo.AnalyzeCallResult(RetVT, CCAssignFnForCall(CC, true, isVarArg)); |
| |
| // Copy all of the result registers out of their specified physreg. |
| if (RVLocs.size() == 2 && RetVT == MVT::f64) { |
| // For this move we copy into two registers and then move into the |
| // double fp reg we want. |
| MVT DestVT = RVLocs[0].getValVT(); |
| const TargetRegisterClass* DstRC = TLI.getRegClassFor(DestVT); |
| unsigned ResultReg = createResultReg(DstRC); |
| AddOptionalDefs(BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc, |
| TII.get(ARM::VMOVDRR), ResultReg) |
| .addReg(RVLocs[0].getLocReg()) |
| .addReg(RVLocs[1].getLocReg())); |
| |
| UsedRegs.push_back(RVLocs[0].getLocReg()); |
| UsedRegs.push_back(RVLocs[1].getLocReg()); |
| |
| // Finally update the result. |
| updateValueMap(I, ResultReg); |
| } else { |
| assert(RVLocs.size() == 1 &&"Can't handle non-double multi-reg retvals!"); |
| MVT CopyVT = RVLocs[0].getValVT(); |
| |
| // Special handling for extended integers. |
| if (RetVT == MVT::i1 || RetVT == MVT::i8 || RetVT == MVT::i16) |
| CopyVT = MVT::i32; |
| |
| const TargetRegisterClass* DstRC = TLI.getRegClassFor(CopyVT); |
| |
| unsigned ResultReg = createResultReg(DstRC); |
| BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc, |
| TII.get(TargetOpcode::COPY), |
| ResultReg).addReg(RVLocs[0].getLocReg()); |
| UsedRegs.push_back(RVLocs[0].getLocReg()); |
| |
| // Finally update the result. |
| updateValueMap(I, ResultReg); |
| } |
| } |
| |
| return true; |
| } |
| |
| bool ARMFastISel::SelectRet(const Instruction *I) { |
| const ReturnInst *Ret = cast<ReturnInst>(I); |
| const Function &F = *I->getParent()->getParent(); |
| |
| if (!FuncInfo.CanLowerReturn) |
| return false; |
| |
| if (TLI.supportSwiftError() && |
| F.getAttributes().hasAttrSomewhere(Attribute::SwiftError)) |
| return false; |
| |
| if (TLI.supportSplitCSR(FuncInfo.MF)) |
| return false; |
| |
| // Build a list of return value registers. |
| SmallVector<unsigned, 4> RetRegs; |
| |
| CallingConv::ID CC = F.getCallingConv(); |
| if (Ret->getNumOperands() > 0) { |
| SmallVector<ISD::OutputArg, 4> Outs; |
| GetReturnInfo(CC, F.getReturnType(), F.getAttributes(), Outs, TLI, DL); |
| |
| // Analyze operands of the call, assigning locations to each operand. |
| SmallVector<CCValAssign, 16> ValLocs; |
| CCState CCInfo(CC, F.isVarArg(), *FuncInfo.MF, ValLocs, I->getContext()); |
| CCInfo.AnalyzeReturn(Outs, CCAssignFnForCall(CC, true /* is Ret */, |
| F.isVarArg())); |
| |
| const Value *RV = Ret->getOperand(0); |
| unsigned Reg = getRegForValue(RV); |
| if (Reg == 0) |
| return false; |
| |
| // Only handle a single return value for now. |
| if (ValLocs.size() != 1) |
| return false; |
| |
| CCValAssign &VA = ValLocs[0]; |
| |
| // Don't bother handling odd stuff for now. |
| if (VA.getLocInfo() != CCValAssign::Full) |
| return false; |
| // Only handle register returns for now. |
| if (!VA.isRegLoc()) |
| return false; |
| |
| unsigned SrcReg = Reg + VA.getValNo(); |
| EVT RVEVT = TLI.getValueType(DL, RV->getType()); |
| if (!RVEVT.isSimple()) return false; |
| MVT RVVT = RVEVT.getSimpleVT(); |
| MVT DestVT = VA.getValVT(); |
| // Special handling for extended integers. |
| if (RVVT != DestVT) { |
| if (RVVT != MVT::i1 && RVVT != MVT::i8 && RVVT != MVT::i16) |
| return false; |
| |
| assert(DestVT == MVT::i32 && "ARM should always ext to i32"); |
| |
| // Perform extension if flagged as either zext or sext. Otherwise, do |
| // nothing. |
| if (Outs[0].Flags.isZExt() || Outs[0].Flags.isSExt()) { |
| SrcReg = ARMEmitIntExt(RVVT, SrcReg, DestVT, Outs[0].Flags.isZExt()); |
| if (SrcReg == 0) return false; |
| } |
| } |
| |
| // Make the copy. |
| unsigned DstReg = VA.getLocReg(); |
| const TargetRegisterClass* SrcRC = MRI.getRegClass(SrcReg); |
| // Avoid a cross-class copy. This is very unlikely. |
| if (!SrcRC->contains(DstReg)) |
| return false; |
| BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc, |
| TII.get(TargetOpcode::COPY), DstReg).addReg(SrcReg); |
| |
| // Add register to return instruction. |
| RetRegs.push_back(VA.getLocReg()); |
| } |
| |
| MachineInstrBuilder MIB = BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc, |
| TII.get(Subtarget->getReturnOpcode())); |
| AddOptionalDefs(MIB); |
| for (unsigned R : RetRegs) |
| MIB.addReg(R, RegState::Implicit); |
| return true; |
| } |
| |
| unsigned ARMFastISel::ARMSelectCallOp(bool UseReg) { |
| if (UseReg) |
| return isThumb2 ? ARM::tBLXr : ARM::BLX; |
| else |
| return isThumb2 ? ARM::tBL : ARM::BL; |
| } |
| |
| unsigned ARMFastISel::getLibcallReg(const Twine &Name) { |
| // Manually compute the global's type to avoid building it when unnecessary. |
| Type *GVTy = Type::getInt32PtrTy(*Context, /*AS=*/0); |
| EVT LCREVT = TLI.getValueType(DL, GVTy); |
| if (!LCREVT.isSimple()) return 0; |
| |
| GlobalValue *GV = new GlobalVariable(M, Type::getInt32Ty(*Context), false, |
| GlobalValue::ExternalLinkage, nullptr, |
| Name); |
| assert(GV->getType() == GVTy && "We miscomputed the type for the global!"); |
| return ARMMaterializeGV(GV, LCREVT.getSimpleVT()); |
| } |
| |
| // A quick function that will emit a call for a named libcall in F with the |
| // vector of passed arguments for the Instruction in I. We can assume that we |
| // can emit a call for any libcall we can produce. This is an abridged version |
| // of the full call infrastructure since we won't need to worry about things |
| // like computed function pointers or strange arguments at call sites. |
| // TODO: Try to unify this and the normal call bits for ARM, then try to unify |
| // with X86. |
| bool ARMFastISel::ARMEmitLibcall(const Instruction *I, RTLIB::Libcall Call) { |
| CallingConv::ID CC = TLI.getLibcallCallingConv(Call); |
| |
| // Handle *simple* calls for now. |
| Type *RetTy = I->getType(); |
| MVT RetVT; |
| if (RetTy->isVoidTy()) |
| RetVT = MVT::isVoid; |
| else if (!isTypeLegal(RetTy, RetVT)) |
| return false; |
| |
| // Can't handle non-double multi-reg retvals. |
| if (RetVT != MVT::isVoid && RetVT != MVT::i32) { |
| SmallVector<CCValAssign, 16> RVLocs; |
| CCState CCInfo(CC, false, *FuncInfo.MF, RVLocs, *Context); |
| CCInfo.AnalyzeCallResult(RetVT, CCAssignFnForCall(CC, true, false)); |
| if (RVLocs.size() >= 2 && RetVT != MVT::f64) |
| return false; |
| } |
| |
| // Set up the argument vectors. |
| SmallVector<Value*, 8> Args; |
| SmallVector<unsigned, 8> ArgRegs; |
| SmallVector<MVT, 8> ArgVTs; |
| SmallVector<ISD::ArgFlagsTy, 8> ArgFlags; |
| Args.reserve(I->getNumOperands()); |
| ArgRegs.reserve(I->getNumOperands()); |
| ArgVTs.reserve(I->getNumOperands()); |
| ArgFlags.reserve(I->getNumOperands()); |
| for (Value *Op : I->operands()) { |
| unsigned Arg = getRegForValue(Op); |
| if (Arg == 0) return false; |
| |
| Type *ArgTy = Op->getType(); |
| MVT ArgVT; |
| if (!isTypeLegal(ArgTy, ArgVT)) return false; |
| |
| ISD::ArgFlagsTy Flags; |
| unsigned OriginalAlignment = DL.getABITypeAlignment(ArgTy); |
| Flags.setOrigAlign(OriginalAlignment); |
| |
| Args.push_back(Op); |
| ArgRegs.push_back(Arg); |
| ArgVTs.push_back(ArgVT); |
| ArgFlags.push_back(Flags); |
| } |
| |
| // Handle the arguments now that we've gotten them. |
| SmallVector<unsigned, 4> RegArgs; |
| unsigned NumBytes; |
| if (!ProcessCallArgs(Args, ArgRegs, ArgVTs, ArgFlags, |
| RegArgs, CC, NumBytes, false)) |
| return false; |
| |
| unsigned CalleeReg = 0; |
| if (Subtarget->genLongCalls()) { |
| CalleeReg = getLibcallReg(TLI.getLibcallName(Call)); |
| if (CalleeReg == 0) return false; |
| } |
| |
| // Issue the call. |
| unsigned CallOpc = ARMSelectCallOp(Subtarget->genLongCalls()); |
| MachineInstrBuilder MIB = BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, |
| DbgLoc, TII.get(CallOpc)); |
| // BL / BLX don't take a predicate, but tBL / tBLX do. |
| if (isThumb2) |
| MIB.add(predOps(ARMCC::AL)); |
| if (Subtarget->genLongCalls()) |
| MIB.addReg(CalleeReg); |
| else |
| MIB.addExternalSymbol(TLI.getLibcallName(Call)); |
| |
| // Add implicit physical register uses to the call. |
| for (unsigned R : RegArgs) |
| MIB.addReg(R, RegState::Implicit); |
| |
| // Add a register mask with the call-preserved registers. |
| // Proper defs for return values will be added by setPhysRegsDeadExcept(). |
| MIB.addRegMask(TRI.getCallPreservedMask(*FuncInfo.MF, CC)); |
| |
| // Finish off the call including any return values. |
| SmallVector<unsigned, 4> UsedRegs; |
| if (!FinishCall(RetVT, UsedRegs, I, CC, NumBytes, false)) return false; |
| |
| // Set all unused physreg defs as dead. |
| static_cast<MachineInstr *>(MIB)->setPhysRegsDeadExcept(UsedRegs, TRI); |
| |
| return true; |
| } |
| |
| bool ARMFastISel::SelectCall(const Instruction *I, |
| const char *IntrMemName = nullptr) { |
| const CallInst *CI = cast<CallInst>(I); |
| const Value *Callee = CI->getCalledValue(); |
| |
| // Can't handle inline asm. |
| if (isa<InlineAsm>(Callee)) return false; |
| |
| // Allow SelectionDAG isel to handle tail calls. |
| if (CI->isTailCall()) return false; |
| |
| // Check the calling convention. |
| ImmutableCallSite CS(CI); |
| CallingConv::ID CC = CS.getCallingConv(); |
| |
| // TODO: Avoid some calling conventions? |
| |
| FunctionType *FTy = CS.getFunctionType(); |
| bool isVarArg = FTy->isVarArg(); |
| |
| // Handle *simple* calls for now. |
| Type *RetTy = I->getType(); |
| MVT RetVT; |
| if (RetTy->isVoidTy()) |
| RetVT = MVT::isVoid; |
| else if (!isTypeLegal(RetTy, RetVT) && RetVT != MVT::i16 && |
| RetVT != MVT::i8 && RetVT != MVT::i1) |
| return false; |
| |
| // Can't handle non-double multi-reg retvals. |
| if (RetVT != MVT::isVoid && RetVT != MVT::i1 && RetVT != MVT::i8 && |
| RetVT != MVT::i16 && RetVT != MVT::i32) { |
| SmallVector<CCValAssign, 16> RVLocs; |
| CCState CCInfo(CC, isVarArg, *FuncInfo.MF, RVLocs, *Context); |
| CCInfo.AnalyzeCallResult(RetVT, CCAssignFnForCall(CC, true, isVarArg)); |
| if (RVLocs.size() >= 2 && RetVT != MVT::f64) |
| return false; |
| } |
| |
| // Set up the argument vectors. |
| SmallVector<Value*, 8> Args; |
| SmallVector<unsigned, 8> ArgRegs; |
| SmallVector<MVT, 8> ArgVTs; |
| SmallVector<ISD::ArgFlagsTy, 8> ArgFlags; |
| unsigned arg_size = CS.arg_size(); |
| Args.reserve(arg_size); |
| ArgRegs.reserve(arg_size); |
| ArgVTs.reserve(arg_size); |
| ArgFlags.reserve(arg_size); |
| for (ImmutableCallSite::arg_iterator i = CS.arg_begin(), e = CS.arg_end(); |
| i != e; ++i) { |
| // If we're lowering a memory intrinsic instead of a regular call, skip the |
| // last argument, which shouldn't be passed to the underlying function. |
| if (IntrMemName && e - i <= 1) |
| break; |
| |
| ISD::ArgFlagsTy Flags; |
| unsigned ArgIdx = i - CS.arg_begin(); |
| if (CS.paramHasAttr(ArgIdx, Attribute::SExt)) |
| Flags.setSExt(); |
| if (CS.paramHasAttr(ArgIdx, Attribute::ZExt)) |
| Flags.setZExt(); |
| |
| // FIXME: Only handle *easy* calls for now. |
| if (CS.paramHasAttr(ArgIdx, Attribute::InReg) || |
| CS.paramHasAttr(ArgIdx, Attribute::StructRet) || |
| CS.paramHasAttr(ArgIdx, Attribute::SwiftSelf) || |
| CS.paramHasAttr(ArgIdx, Attribute::SwiftError) || |
| CS.paramHasAttr(ArgIdx, Attribute::Nest) || |
| CS.paramHasAttr(ArgIdx, Attribute::ByVal)) |
| return false; |
| |
| Type *ArgTy = (*i)->getType(); |
| MVT ArgVT; |
| if (!isTypeLegal(ArgTy, ArgVT) && ArgVT != MVT::i16 && ArgVT != MVT::i8 && |
| ArgVT != MVT::i1) |
| return false; |
| |
| unsigned Arg = getRegForValue(*i); |
| if (Arg == 0) |
| return false; |
| |
| unsigned OriginalAlignment = DL.getABITypeAlignment(ArgTy); |
| Flags.setOrigAlign(OriginalAlignment); |
| |
| Args.push_back(*i); |
| ArgRegs.push_back(Arg); |
| ArgVTs.push_back(ArgVT); |
| ArgFlags.push_back(Flags); |
| } |
| |
| // Handle the arguments now that we've gotten them. |
| SmallVector<unsigned, 4> RegArgs; |
| unsigned NumBytes; |
| if (!ProcessCallArgs(Args, ArgRegs, ArgVTs, ArgFlags, |
| RegArgs, CC, NumBytes, isVarArg)) |
| return false; |
| |
| bool UseReg = false; |
| const GlobalValue *GV = dyn_cast<GlobalValue>(Callee); |
| if (!GV || Subtarget->genLongCalls()) UseReg = true; |
| |
| unsigned CalleeReg = 0; |
| if (UseReg) { |
| if (IntrMemName) |
| CalleeReg = getLibcallReg(IntrMemName); |
| else |
| CalleeReg = getRegForValue(Callee); |
| |
| if (CalleeReg == 0) return false; |
| } |
| |
| // Issue the call. |
| unsigned CallOpc = ARMSelectCallOp(UseReg); |
| MachineInstrBuilder MIB = BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, |
| DbgLoc, TII.get(CallOpc)); |
| |
| // ARM calls don't take a predicate, but tBL / tBLX do. |
| if(isThumb2) |
| MIB.add(predOps(ARMCC::AL)); |
| if (UseReg) |
| MIB.addReg(CalleeReg); |
| else if (!IntrMemName) |
| MIB.addGlobalAddress(GV, 0, 0); |
| else |
| MIB.addExternalSymbol(IntrMemName, 0); |
| |
| // Add implicit physical register uses to the call. |
| for (unsigned R : RegArgs) |
| MIB.addReg(R, RegState::Implicit); |
| |
| // Add a register mask with the call-preserved registers. |
| // Proper defs for return values will be added by setPhysRegsDeadExcept(). |
| MIB.addRegMask(TRI.getCallPreservedMask(*FuncInfo.MF, CC)); |
| |
| // Finish off the call including any return values. |
| SmallVector<unsigned, 4> UsedRegs; |
| if (!FinishCall(RetVT, UsedRegs, I, CC, NumBytes, isVarArg)) |
| return false; |
| |
| // Set all unused physreg defs as dead. |
| static_cast<MachineInstr *>(MIB)->setPhysRegsDeadExcept(UsedRegs, TRI); |
| |
| return true; |
| } |
| |
| bool ARMFastISel::ARMIsMemCpySmall(uint64_t Len) { |
| return Len <= 16; |
| } |
| |
| bool ARMFastISel::ARMTryEmitSmallMemCpy(Address Dest, Address Src, |
| uint64_t Len, unsigned Alignment) { |
| // Make sure we don't bloat code by inlining very large memcpy's. |
| if (!ARMIsMemCpySmall(Len)) |
| return false; |
| |
| while (Len) { |
| MVT VT; |
| if (!Alignment || Alignment >= 4) { |
| if (Len >= 4) |
| VT = MVT::i32; |
| else if (Len >= 2) |
| VT = MVT::i16; |
| else { |
| assert(Len == 1 && "Expected a length of 1!"); |
| VT = MVT::i8; |
| } |
| } else { |
| // Bound based on alignment. |
| if (Len >= 2 && Alignment == 2) |
| VT = MVT::i16; |
| else { |
| VT = MVT::i8; |
| } |
| } |
| |
| bool RV; |
| unsigned ResultReg; |
| RV = ARMEmitLoad(VT, ResultReg, Src); |
| assert(RV && "Should be able to handle this load."); |
| RV = ARMEmitStore(VT, ResultReg, Dest); |
| assert(RV && "Should be able to handle this store."); |
| (void)RV; |
| |
| unsigned Size = VT.getSizeInBits()/8; |
| Len -= Size; |
| Dest.Offset += Size; |
| Src.Offset += Size; |
| } |
| |
| return true; |
| } |
| |
| bool ARMFastISel::SelectIntrinsicCall(const IntrinsicInst &I) { |
| // FIXME: Handle more intrinsics. |
| switch (I.getIntrinsicID()) { |
| default: return false; |
| case Intrinsic::frameaddress: { |
| MachineFrameInfo &MFI = FuncInfo.MF->getFrameInfo(); |
| MFI.setFrameAddressIsTaken(true); |
| |
| unsigned LdrOpc = isThumb2 ? ARM::t2LDRi12 : ARM::LDRi12; |
| const TargetRegisterClass *RC = isThumb2 ? &ARM::tGPRRegClass |
| : &ARM::GPRRegClass; |
| |
| const ARMBaseRegisterInfo *RegInfo = |
| static_cast<const ARMBaseRegisterInfo *>(Subtarget->getRegisterInfo()); |
| unsigned FramePtr = RegInfo->getFrameRegister(*(FuncInfo.MF)); |
| unsigned SrcReg = FramePtr; |
| |
| // Recursively load frame address |
| // ldr r0 [fp] |
| // ldr r0 [r0] |
| // ldr r0 [r0] |
| // ... |
| unsigned DestReg; |
| unsigned Depth = cast<ConstantInt>(I.getOperand(0))->getZExtValue(); |
| while (Depth--) { |
| DestReg = createResultReg(RC); |
| AddOptionalDefs(BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc, |
| TII.get(LdrOpc), DestReg) |
| .addReg(SrcReg).addImm(0)); |
| SrcReg = DestReg; |
| } |
| updateValueMap(&I, SrcReg); |
| return true; |
| } |
| case Intrinsic::memcpy: |
| case Intrinsic::memmove: { |
| const MemTransferInst &MTI = cast<MemTransferInst>(I); |
| // Don't handle volatile. |
| if (MTI.isVolatile()) |
| return false; |
| |
| // Disable inlining for memmove before calls to ComputeAddress. Otherwise, |
| // we would emit dead code because we don't currently handle memmoves. |
| bool isMemCpy = (I.getIntrinsicID() == Intrinsic::memcpy); |
| if (isa<ConstantInt>(MTI.getLength()) && isMemCpy) { |
| // Small memcpy's are common enough that we want to do them without a call |
| // if possible. |
| uint64_t Len = cast<ConstantInt>(MTI.getLength())->getZExtValue(); |
| if (ARMIsMemCpySmall(Len)) { |
| Address Dest, Src; |
| if (!ARMComputeAddress(MTI.getRawDest(), Dest) || |
| !ARMComputeAddress(MTI.getRawSource(), Src)) |
| return false; |
| unsigned Alignment = MinAlign(MTI.getDestAlignment(), |
| MTI.getSourceAlignment()); |
| if (ARMTryEmitSmallMemCpy(Dest, Src, Len, Alignment)) |
| return true; |
| } |
| } |
| |
| if (!MTI.getLength()->getType()->isIntegerTy(32)) |
| return false; |
| |
| if (MTI.getSourceAddressSpace() > 255 || MTI.getDestAddressSpace() > 255) |
| return false; |
| |
| const char *IntrMemName = isa<MemCpyInst>(I) ? "memcpy" : "memmove"; |
| return SelectCall(&I, IntrMemName); |
| } |
| case Intrinsic::memset: { |
| const MemSetInst &MSI = cast<MemSetInst>(I); |
| // Don't handle volatile. |
| if (MSI.isVolatile()) |
| return false; |
| |
| if (!MSI.getLength()->getType()->isIntegerTy(32)) |
| return false; |
| |
| if (MSI.getDestAddressSpace() > 255) |
| return false; |
| |
| return SelectCall(&I, "memset"); |
| } |
| case Intrinsic::trap: { |
| BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc, TII.get( |
| Subtarget->useNaClTrap() ? ARM::TRAPNaCl : ARM::TRAP)); |
| return true; |
| } |
| } |
| } |
| |
| bool ARMFastISel::SelectTrunc(const Instruction *I) { |
| // The high bits for a type smaller than the register size are assumed to be |
| // undefined. |
| Value *Op = I->getOperand(0); |
| |
| EVT SrcVT, DestVT; |
| SrcVT = TLI.getValueType(DL, Op->getType(), true); |
| DestVT = TLI.getValueType(DL, I->getType(), true); |
| |
| if (SrcVT != MVT::i32 && SrcVT != MVT::i16 && SrcVT != MVT::i8) |
| return false; |
| if (DestVT != MVT::i16 && DestVT != MVT::i8 && DestVT != MVT::i1) |
| return false; |
| |
| unsigned SrcReg = getRegForValue(Op); |
| if (!SrcReg) return false; |
| |
| // Because the high bits are undefined, a truncate doesn't generate |
| // any code. |
| updateValueMap(I, SrcReg); |
| return true; |
| } |
| |
| unsigned ARMFastISel::ARMEmitIntExt(MVT SrcVT, unsigned SrcReg, MVT DestVT, |
| bool isZExt) { |
| if (DestVT != MVT::i32 && DestVT != MVT::i16 && DestVT != MVT::i8) |
| return 0; |
| if (SrcVT != MVT::i16 && SrcVT != MVT::i8 && SrcVT != MVT::i1) |
| return 0; |
| |
| // Table of which combinations can be emitted as a single instruction, |
| // and which will require two. |
| static const uint8_t isSingleInstrTbl[3][2][2][2] = { |
| // ARM Thumb |
| // !hasV6Ops hasV6Ops !hasV6Ops hasV6Ops |
| // ext: s z s z s z s z |
| /* 1 */ { { { 0, 1 }, { 0, 1 } }, { { 0, 0 }, { 0, 1 } } }, |
| /* 8 */ { { { 0, 1 }, { 1, 1 } }, { { 0, 0 }, { 1, 1 } } }, |
| /* 16 */ { { { 0, 0 }, { 1, 1 } }, { { 0, 0 }, { 1, 1 } } } |
| }; |
| |
| // Target registers for: |
| // - For ARM can never be PC. |
| // - For 16-bit Thumb are restricted to lower 8 registers. |
| // - For 32-bit Thumb are restricted to non-SP and non-PC. |
| static const TargetRegisterClass *RCTbl[2][2] = { |
| // Instructions: Two Single |
| /* ARM */ { &ARM::GPRnopcRegClass, &ARM::GPRnopcRegClass }, |
| /* Thumb */ { &ARM::tGPRRegClass, &ARM::rGPRRegClass } |
| }; |
| |
| // Table governing the instruction(s) to be emitted. |
| static const struct InstructionTable { |
| uint32_t Opc : 16; |
| uint32_t hasS : 1; // Some instructions have an S bit, always set it to 0. |
| uint32_t Shift : 7; // For shift operand addressing mode, used by MOVsi. |
| uint32_t Imm : 8; // All instructions have either a shift or a mask. |
| } IT[2][2][3][2] = { |
| { // Two instructions (first is left shift, second is in this table). |
| { // ARM Opc S Shift Imm |
| /* 1 bit sext */ { { ARM::MOVsi , 1, ARM_AM::asr , 31 }, |
| /* 1 bit zext */ { ARM::MOVsi , 1, ARM_AM::lsr , 31 } }, |
| /* 8 bit sext */ { { ARM::MOVsi , 1, ARM_AM::asr , 24 }, |
| /* 8 bit zext */ { ARM::MOVsi , 1, ARM_AM::lsr , 24 } }, |
| /* 16 bit sext */ { { ARM::MOVsi , 1, ARM_AM::asr , 16 }, |
| /* 16 bit zext */ { ARM::MOVsi , 1, ARM_AM::lsr , 16 } } |
| }, |
| { // Thumb Opc S Shift Imm |
| /* 1 bit sext */ { { ARM::tASRri , 0, ARM_AM::no_shift, 31 }, |
| /* 1 bit zext */ { ARM::tLSRri , 0, ARM_AM::no_shift, 31 } }, |
| /* 8 bit sext */ { { ARM::tASRri , 0, ARM_AM::no_shift, 24 }, |
| /* 8 bit zext */ { ARM::tLSRri , 0, ARM_AM::no_shift, 24 } }, |
| /* 16 bit sext */ { { ARM::tASRri , 0, ARM_AM::no_shift, 16 }, |
| /* 16 bit zext */ { ARM::tLSRri , 0, ARM_AM::no_shift, 16 } } |
| } |
| }, |
| { // Single instruction. |
| { // ARM Opc S Shift Imm |
| /* 1 bit sext */ { { ARM::KILL , 0, ARM_AM::no_shift, 0 }, |
| /* 1 bit zext */ { ARM::ANDri , 1, ARM_AM::no_shift, 1 } }, |
| /* 8 bit sext */ { { ARM::SXTB , 0, ARM_AM::no_shift, 0 }, |
| /* 8 bit zext */ { ARM::ANDri , 1, ARM_AM::no_shift, 255 } }, |
| /* 16 bit sext */ { { ARM::SXTH , 0, ARM_AM::no_shift, 0 }, |
| /* 16 bit zext */ { ARM::UXTH , 0, ARM_AM::no_shift, 0 } } |
| }, |
| { // Thumb Opc S Shift Imm |
| /* 1 bit sext */ { { ARM::KILL , 0, ARM_AM::no_shift, 0 }, |
| /* 1 bit zext */ { ARM::t2ANDri, 1, ARM_AM::no_shift, 1 } }, |
| /* 8 bit sext */ { { ARM::t2SXTB , 0, ARM_AM::no_shift, 0 }, |
| /* 8 bit zext */ { ARM::t2ANDri, 1, ARM_AM::no_shift, 255 } }, |
| /* 16 bit sext */ { { ARM::t2SXTH , 0, ARM_AM::no_shift, 0 }, |
| /* 16 bit zext */ { ARM::t2UXTH , 0, ARM_AM::no_shift, 0 } } |
| } |
| } |
| }; |
| |
| unsigned SrcBits = SrcVT.getSizeInBits(); |
| unsigned DestBits = DestVT.getSizeInBits(); |
| (void) DestBits; |
| assert((SrcBits < DestBits) && "can only extend to larger types"); |
| assert((DestBits == 32 || DestBits == 16 || DestBits == 8) && |
| "other sizes unimplemented"); |
| assert((SrcBits == 16 || SrcBits == 8 || SrcBits == 1) && |
| "other sizes unimplemented"); |
| |
| bool hasV6Ops = Subtarget->hasV6Ops(); |
| unsigned Bitness = SrcBits / 8; // {1,8,16}=>{0,1,2} |
| assert((Bitness < 3) && "sanity-check table bounds"); |
| |
| bool isSingleInstr = isSingleInstrTbl[Bitness][isThumb2][hasV6Ops][isZExt]; |
| const TargetRegisterClass *RC = RCTbl[isThumb2][isSingleInstr]; |
| const InstructionTable *ITP = &IT[isSingleInstr][isThumb2][Bitness][isZExt]; |
| unsigned Opc = ITP->Opc; |
| assert(ARM::KILL != Opc && "Invalid table entry"); |
| unsigned hasS = ITP->hasS; |
| ARM_AM::ShiftOpc Shift = (ARM_AM::ShiftOpc) ITP->Shift; |
| assert(((Shift == ARM_AM::no_shift) == (Opc != ARM::MOVsi)) && |
| "only MOVsi has shift operand addressing mode"); |
| unsigned Imm = ITP->Imm; |
| |
| // 16-bit Thumb instructions always set CPSR (unless they're in an IT block). |
| bool setsCPSR = &ARM::tGPRRegClass == RC; |
| unsigned LSLOpc = isThumb2 ? ARM::tLSLri : ARM::MOVsi; |
| unsigned ResultReg; |
| // MOVsi encodes shift and immediate in shift operand addressing mode. |
| // The following condition has the same value when emitting two |
| // instruction sequences: both are shifts. |
| bool ImmIsSO = (Shift != ARM_AM::no_shift); |
| |
| // Either one or two instructions are emitted. |
| // They're always of the form: |
| // dst = in OP imm |
| // CPSR is set only by 16-bit Thumb instructions. |
| // Predicate, if any, is AL. |
| // S bit, if available, is always 0. |
| // When two are emitted the first's result will feed as the second's input, |
| // that value is then dead. |
| unsigned NumInstrsEmitted = isSingleInstr ? 1 : 2; |
| for (unsigned Instr = 0; Instr != NumInstrsEmitted; ++Instr) { |
| ResultReg = createResultReg(RC); |
| bool isLsl = (0 == Instr) && !isSingleInstr; |
| unsigned Opcode = isLsl ? LSLOpc : Opc; |
| ARM_AM::ShiftOpc ShiftAM = isLsl ? ARM_AM::lsl : Shift; |
| unsigned ImmEnc = ImmIsSO ? ARM_AM::getSORegOpc(ShiftAM, Imm) : Imm; |
| bool isKill = 1 == Instr; |
| MachineInstrBuilder MIB = BuildMI( |
| *FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc, TII.get(Opcode), ResultReg); |
| if (setsCPSR) |
| MIB.addReg(ARM::CPSR, RegState::Define); |
| SrcReg = constrainOperandRegClass(TII.get(Opcode), SrcReg, 1 + setsCPSR); |
| MIB.addReg(SrcReg, isKill * RegState::Kill) |
| .addImm(ImmEnc) |
| .add(predOps(ARMCC::AL)); |
| if (hasS) |
| MIB.add(condCodeOp()); |
| // Second instruction consumes the first's result. |
| SrcReg = ResultReg; |
| } |
| |
| return ResultReg; |
| } |
| |
| bool ARMFastISel::SelectIntExt(const Instruction *I) { |
| // On ARM, in general, integer casts don't involve legal types; this code |
| // handles promotable integers. |
| Type *DestTy = I->getType(); |
| Value *Src = I->getOperand(0); |
| Type *SrcTy = Src->getType(); |
| |
| bool isZExt = isa<ZExtInst>(I); |
| unsigned SrcReg = getRegForValue(Src); |
| if (!SrcReg) return false; |
| |
| EVT SrcEVT, DestEVT; |
| SrcEVT = TLI.getValueType(DL, SrcTy, true); |
| DestEVT = TLI.getValueType(DL, DestTy, true); |
| if (!SrcEVT.isSimple()) return false; |
| if (!DestEVT.isSimple()) return false; |
| |
| MVT SrcVT = SrcEVT.getSimpleVT(); |
| MVT DestVT = DestEVT.getSimpleVT(); |
| unsigned ResultReg = ARMEmitIntExt(SrcVT, SrcReg, DestVT, isZExt); |
| if (ResultReg == 0) return false; |
| updateValueMap(I, ResultReg); |
| return true; |
| } |
| |
| bool ARMFastISel::SelectShift(const Instruction *I, |
| ARM_AM::ShiftOpc ShiftTy) { |
| // We handle thumb2 mode by target independent selector |
| // or SelectionDAG ISel. |
| if (isThumb2) |
| return false; |
| |
| // Only handle i32 now. |
| EVT DestVT = TLI.getValueType(DL, I->getType(), true); |
| if (DestVT != MVT::i32) |
| return false; |
| |
| unsigned Opc = ARM::MOVsr; |
| unsigned ShiftImm; |
| Value *Src2Value = I->getOperand(1); |
| if (const ConstantInt *CI = dyn_cast<ConstantInt>(Src2Value)) { |
| ShiftImm = CI->getZExtValue(); |
| |
| // Fall back to selection DAG isel if the shift amount |
| // is zero or greater than the width of the value type. |
| if (ShiftImm == 0 || ShiftImm >=32) |
| return false; |
| |
| Opc = ARM::MOVsi; |
| } |
| |
| Value *Src1Value = I->getOperand(0); |
| unsigned Reg1 = getRegForValue(Src1Value); |
| if (Reg1 == 0) return false; |
| |
| unsigned Reg2 = 0; |
| if (Opc == ARM::MOVsr) { |
| Reg2 = getRegForValue(Src2Value); |
| if (Reg2 == 0) return false; |
| } |
| |
| unsigned ResultReg = createResultReg(&ARM::GPRnopcRegClass); |
| if(ResultReg == 0) return false; |
| |
| MachineInstrBuilder MIB = BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc, |
| TII.get(Opc), ResultReg) |
| .addReg(Reg1); |
| |
| if (Opc == ARM::MOVsi) |
| MIB.addImm(ARM_AM::getSORegOpc(ShiftTy, ShiftImm)); |
| else if (Opc == ARM::MOVsr) { |
| MIB.addReg(Reg2); |
| MIB.addImm(ARM_AM::getSORegOpc(ShiftTy, 0)); |
| } |
| |
| AddOptionalDefs(MIB); |
| updateValueMap(I, ResultReg); |
| return true; |
| } |
| |
| // TODO: SoftFP support. |
| bool ARMFastISel::fastSelectInstruction(const Instruction *I) { |
| switch (I->getOpcode()) { |
| case Instruction::Load: |
| return SelectLoad(I); |
| case Instruction::Store: |
| return SelectStore(I); |
| case Instruction::Br: |
| return SelectBranch(I); |
| case Instruction::IndirectBr: |
| return SelectIndirectBr(I); |
| case Instruction::ICmp: |
| case Instruction::FCmp: |
| return SelectCmp(I); |
| case Instruction::FPExt: |
| return SelectFPExt(I); |
| case Instruction::FPTrunc: |
| return SelectFPTrunc(I); |
| case Instruction::SIToFP: |
| return SelectIToFP(I, /*isSigned*/ true); |
| case Instruction::UIToFP: |
| return SelectIToFP(I, /*isSigned*/ false); |
| case Instruction::FPToSI: |
| return SelectFPToI(I, /*isSigned*/ true); |
| case Instruction::FPToUI: |
| return SelectFPToI(I, /*isSigned*/ false); |
| case Instruction::Add: |
| return SelectBinaryIntOp(I, ISD::ADD); |
| case Instruction::Or: |
| return SelectBinaryIntOp(I, ISD::OR); |
| case Instruction::Sub: |
| return SelectBinaryIntOp(I, ISD::SUB); |
| case Instruction::FAdd: |
| return SelectBinaryFPOp(I, ISD::FADD); |
| case Instruction::FSub: |
| return SelectBinaryFPOp(I, ISD::FSUB); |
| case Instruction::FMul: |
| return SelectBinaryFPOp(I, ISD::FMUL); |
| case Instruction::SDiv: |
| return SelectDiv(I, /*isSigned*/ true); |
| case Instruction::UDiv: |
| return SelectDiv(I, /*isSigned*/ false); |
| case Instruction::SRem: |
| return SelectRem(I, /*isSigned*/ true); |
| case Instruction::URem: |
| return SelectRem(I, /*isSigned*/ false); |
| case Instruction::Call: |
| if (const IntrinsicInst *II = dyn_cast<IntrinsicInst>(I)) |
| return SelectIntrinsicCall(*II); |
| return SelectCall(I); |
| case Instruction::Select: |
| return SelectSelect(I); |
| case Instruction::Ret: |
| return SelectRet(I); |
| case Instruction::Trunc: |
| return SelectTrunc(I); |
| case Instruction::ZExt: |
| case Instruction::SExt: |
| return SelectIntExt(I); |
| case Instruction::Shl: |
| return SelectShift(I, ARM_AM::lsl); |
| case Instruction::LShr: |
| return SelectShift(I, ARM_AM::lsr); |
| case Instruction::AShr: |
| return SelectShift(I, ARM_AM::asr); |
| default: break; |
| } |
| return false; |
| } |
| |
| // This table describes sign- and zero-extend instructions which can be |
| // folded into a preceding load. All of these extends have an immediate |
| // (sometimes a mask and sometimes a shift) that's applied after |
| // extension. |
| static const struct FoldableLoadExtendsStruct { |
| uint16_t Opc[2]; // ARM, Thumb. |
| uint8_t ExpectedImm; |
| uint8_t isZExt : 1; |
| uint8_t ExpectedVT : 7; |
| } FoldableLoadExtends[] = { |
| { { ARM::SXTH, ARM::t2SXTH }, 0, 0, MVT::i16 }, |
| { { ARM::UXTH, ARM::t2UXTH }, 0, 1, MVT::i16 }, |
| { { ARM::ANDri, ARM::t2ANDri }, 255, 1, MVT::i8 }, |
| { { ARM::SXTB, ARM::t2SXTB }, 0, 0, MVT::i8 }, |
| { { ARM::UXTB, ARM::t2UXTB }, 0, 1, MVT::i8 } |
| }; |
| |
| /// The specified machine instr operand is a vreg, and that |
| /// vreg is being provided by the specified load instruction. If possible, |
| /// try to fold the load as an operand to the instruction, returning true if |
| /// successful. |
| bool ARMFastISel::tryToFoldLoadIntoMI(MachineInstr *MI, unsigned OpNo, |
| const LoadInst *LI) { |
| // Verify we have a legal type before going any further. |
| MVT VT; |
| if (!isLoadTypeLegal(LI->getType(), VT)) |
| return false; |
| |
| // Combine load followed by zero- or sign-extend. |
| // ldrb r1, [r0] ldrb r1, [r0] |
| // uxtb r2, r1 => |
| // mov r3, r2 mov r3, r1 |
| if (MI->getNumOperands() < 3 || !MI->getOperand(2).isImm()) |
| return false; |
| const uint64_t Imm = MI->getOperand(2).getImm(); |
| |
| bool Found = false; |
| bool isZExt; |
| for (const FoldableLoadExtendsStruct &FLE : FoldableLoadExtends) { |
| if (FLE.Opc[isThumb2] == MI->getOpcode() && |
| (uint64_t)FLE.ExpectedImm == Imm && |
| MVT((MVT::SimpleValueType)FLE.ExpectedVT) == VT) { |
| Found = true; |
| isZExt = FLE.isZExt; |
| } |
| } |
| if (!Found) return false; |
| |
| // See if we can handle this address. |
| Address Addr; |
| if (!ARMComputeAddress(LI->getOperand(0), Addr)) return false; |
| |
| unsigned ResultReg = MI->getOperand(0).getReg(); |
| if (!ARMEmitLoad(VT, ResultReg, Addr, LI->getAlignment(), isZExt, false)) |
| return false; |
| MI->eraseFromParent(); |
| return true; |
| } |
| |
| unsigned ARMFastISel::ARMLowerPICELF(const GlobalValue *GV, |
| unsigned Align, MVT VT) { |
| bool UseGOT_PREL = !TM.shouldAssumeDSOLocal(*GV->getParent(), GV); |
| |
| LLVMContext *Context = &MF->getFunction().getContext(); |
| unsigned ARMPCLabelIndex = AFI->createPICLabelUId(); |
| unsigned PCAdj = Subtarget->isThumb() ? 4 : 8; |
| ARMConstantPoolValue *CPV = ARMConstantPoolConstant::Create( |
| GV, ARMPCLabelIndex, ARMCP::CPValue, PCAdj, |
| UseGOT_PREL ? ARMCP::GOT_PREL : ARMCP::no_modifier, |
| /*AddCurrentAddress=*/UseGOT_PREL); |
| |
| unsigned ConstAlign = |
| MF->getDataLayout().getPrefTypeAlignment(Type::getInt32PtrTy(*Context)); |
| unsigned Idx = MF->getConstantPool()->getConstantPoolIndex(CPV, ConstAlign); |
| |
| unsigned TempReg = MF->getRegInfo().createVirtualRegister(&ARM::rGPRRegClass); |
| unsigned Opc = isThumb2 ? ARM::t2LDRpci : ARM::LDRcp; |
| MachineInstrBuilder MIB = |
| BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc, TII.get(Opc), TempReg) |
| .addConstantPoolIndex(Idx); |
| if (Opc == ARM::LDRcp) |
| MIB.addImm(0); |
| MIB.add(predOps(ARMCC::AL)); |
| |
| // Fix the address by adding pc. |
| unsigned DestReg = createResultReg(TLI.getRegClassFor(VT)); |
| Opc = Subtarget->isThumb() ? ARM::tPICADD : UseGOT_PREL ? ARM::PICLDR |
| : ARM::PICADD; |
| DestReg = constrainOperandRegClass(TII.get(Opc), DestReg, 0); |
| MIB = BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc, TII.get(Opc), DestReg) |
| .addReg(TempReg) |
| .addImm(ARMPCLabelIndex); |
| if (!Subtarget->isThumb()) |
| MIB.add(predOps(ARMCC::AL)); |
| |
| if (UseGOT_PREL && Subtarget->isThumb()) { |
| unsigned NewDestReg = createResultReg(TLI.getRegClassFor(VT)); |
| MIB = BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc, |
| TII.get(ARM::t2LDRi12), NewDestReg) |
| .addReg(DestReg) |
| .addImm(0); |
| DestReg = NewDestReg; |
| AddOptionalDefs(MIB); |
| } |
| return DestReg; |
| } |
| |
| bool ARMFastISel::fastLowerArguments() { |
| if (!FuncInfo.CanLowerReturn) |
| return false; |
| |
| const Function *F = FuncInfo.Fn; |
| if (F->isVarArg()) |
| return false; |
| |
| CallingConv::ID CC = F->getCallingConv(); |
| switch (CC) { |
| default: |
| return false; |
| case CallingConv::Fast: |
| case CallingConv::C: |
| case CallingConv::ARM_AAPCS_VFP: |
| case CallingConv::ARM_AAPCS: |
| case CallingConv::ARM_APCS: |
| case CallingConv::Swift: |
| break; |
| } |
| |
| // Only handle simple cases. i.e. Up to 4 i8/i16/i32 scalar arguments |
| // which are passed in r0 - r3. |
| for (const Argument &Arg : F->args()) { |
| if (Arg.getArgNo() >= 4) |
| return false; |
| |
| if (Arg.hasAttribute(Attribute::InReg) || |
| Arg.hasAttribute(Attribute::StructRet) || |
| Arg.hasAttribute(Attribute::SwiftSelf) || |
| Arg.hasAttribute(Attribute::SwiftError) || |
| Arg.hasAttribute(Attribute::ByVal)) |
| return false; |
| |
| Type *ArgTy = Arg.getType(); |
| if (ArgTy->isStructTy() || ArgTy->isArrayTy() || ArgTy->isVectorTy()) |
| return false; |
| |
| EVT ArgVT = TLI.getValueType(DL, ArgTy); |
| if (!ArgVT.isSimple()) return false; |
| switch (ArgVT.getSimpleVT().SimpleTy) { |
| case MVT::i8: |
| case MVT::i16: |
| case MVT::i32: |
| break; |
| default: |
| return false; |
| } |
| } |
| |
| static const MCPhysReg GPRArgRegs[] = { |
| ARM::R0, ARM::R1, ARM::R2, ARM::R3 |
| }; |
| |
| const TargetRegisterClass *RC = &ARM::rGPRRegClass; |
| for (const Argument &Arg : F->args()) { |
| unsigned ArgNo = Arg.getArgNo(); |
| unsigned SrcReg = GPRArgRegs[ArgNo]; |
| unsigned DstReg = FuncInfo.MF->addLiveIn(SrcReg, RC); |
| // FIXME: Unfortunately it's necessary to emit a copy from the livein copy. |
| // Without this, EmitLiveInCopies may eliminate the livein if its only |
| // use is a bitcast (which isn't turned into an instruction). |
| unsigned ResultReg = createResultReg(RC); |
| BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc, |
| TII.get(TargetOpcode::COPY), |
| ResultReg).addReg(DstReg, getKillRegState(true)); |
| updateValueMap(&Arg, ResultReg); |
| } |
| |
| return true; |
| } |
| |
| namespace llvm { |
| |
| FastISel *ARM::createFastISel(FunctionLoweringInfo &funcInfo, |
| const TargetLibraryInfo *libInfo) { |
| if (funcInfo.MF->getSubtarget<ARMSubtarget>().useFastISel()) |
| return new ARMFastISel(funcInfo, libInfo); |
| |
| return nullptr; |
| } |
| |
| } // end namespace llvm |