blob: 5492dcc268d52a1b68ecd024a50fb5dc8de27f96 [file] [log] [blame]
Dan Gohmane149e982010-04-22 20:06:42 +00001//===-- FastISel.cpp - Implementation of the FastISel class ---------------===//
Dan Gohmanb2226e22008-08-13 20:19:35 +00002//
3// The LLVM Compiler Infrastructure
4//
5// This file is distributed under the University of Illinois Open Source
6// License. See LICENSE.TXT for details.
7//
8//===----------------------------------------------------------------------===//
9//
10// This file contains the implementation of the FastISel class.
11//
Dan Gohmanb4863502008-09-30 20:48:29 +000012// "Fast" instruction selection is designed to emit very poor code quickly.
13// Also, it is not designed to be able to do much lowering, so most illegal
Chris Lattnerc52af452008-10-13 01:59:13 +000014// types (e.g. i64 on 32-bit targets) and operations are not supported. It is
15// also not intended to be able to do much optimization, except in a few cases
16// where doing optimizations reduces overall compile time. For example, folding
17// constants into immediate fields is often done, because it's cheap and it
18// reduces the number of instructions later phases have to examine.
Dan Gohmanb4863502008-09-30 20:48:29 +000019//
20// "Fast" instruction selection is able to fail gracefully and transfer
21// control to the SelectionDAG selector for operations that it doesn't
Chris Lattnerc52af452008-10-13 01:59:13 +000022// support. In many cases, this allows us to avoid duplicating a lot of
Dan Gohmanb4863502008-09-30 20:48:29 +000023// the complicated lowering logic that SelectionDAG currently has.
24//
25// The intended use for "fast" instruction selection is "-O0" mode
26// compilation, where the quality of the generated code is irrelevant when
Chris Lattnerc52af452008-10-13 01:59:13 +000027// weighed against the speed at which the code can be generated. Also,
Dan Gohmanb4863502008-09-30 20:48:29 +000028// at -O0, the LLVM optimizers are not running, and this makes the
29// compile time of codegen a much higher portion of the overall compile
Chris Lattnerc52af452008-10-13 01:59:13 +000030// time. Despite its limitations, "fast" instruction selection is able to
Dan Gohmanb4863502008-09-30 20:48:29 +000031// handle enough code on its own to provide noticeable overall speedups
32// in -O0 compiles.
33//
34// Basic operations are supported in a target-independent way, by reading
35// the same instruction descriptions that the SelectionDAG selector reads,
36// and identifying simple arithmetic operations that can be directly selected
Chris Lattnerc52af452008-10-13 01:59:13 +000037// from simple operators. More complicated operations currently require
Dan Gohmanb4863502008-09-30 20:48:29 +000038// target-specific code.
39//
Dan Gohmanb2226e22008-08-13 20:19:35 +000040//===----------------------------------------------------------------------===//
41
Juergen Ributzka8179e9e2014-07-11 22:01:42 +000042#include "llvm/CodeGen/Analysis.h"
Chandler Carruthed0881b2012-12-03 16:50:05 +000043#include "llvm/CodeGen/FastISel.h"
David Blaikie0252265b2013-06-16 20:34:15 +000044#include "llvm/ADT/Optional.h"
Chandler Carruthed0881b2012-12-03 16:50:05 +000045#include "llvm/ADT/Statistic.h"
Juergen Ributzka454d3742014-06-13 00:45:11 +000046#include "llvm/Analysis/BranchProbabilityInfo.h"
Chandler Carruthed0881b2012-12-03 16:50:05 +000047#include "llvm/Analysis/Loads.h"
48#include "llvm/CodeGen/Analysis.h"
49#include "llvm/CodeGen/FunctionLoweringInfo.h"
Juergen Ributzka04558dc2014-06-12 03:29:26 +000050#include "llvm/CodeGen/MachineFrameInfo.h"
Chandler Carruthed0881b2012-12-03 16:50:05 +000051#include "llvm/CodeGen/MachineInstrBuilder.h"
52#include "llvm/CodeGen/MachineModuleInfo.h"
53#include "llvm/CodeGen/MachineRegisterInfo.h"
Juergen Ributzka04558dc2014-06-12 03:29:26 +000054#include "llvm/CodeGen/StackMaps.h"
Chandler Carruth9fb823b2013-01-02 11:36:10 +000055#include "llvm/IR/DataLayout.h"
Chandler Carruth9a4c9e52014-03-06 00:46:21 +000056#include "llvm/IR/DebugInfo.h"
Chandler Carruth9fb823b2013-01-02 11:36:10 +000057#include "llvm/IR/Function.h"
58#include "llvm/IR/GlobalVariable.h"
59#include "llvm/IR/Instructions.h"
60#include "llvm/IR/IntrinsicInst.h"
61#include "llvm/IR/Operator.h"
Chandler Carruthed0881b2012-12-03 16:50:05 +000062#include "llvm/Support/Debug.h"
63#include "llvm/Support/ErrorHandling.h"
Dan Gohmanb2226e22008-08-13 20:19:35 +000064#include "llvm/Target/TargetInstrInfo.h"
Bob Wilson3e6fa462012-08-03 04:06:28 +000065#include "llvm/Target/TargetLibraryInfo.h"
Evan Cheng864fcc12008-08-20 22:45:34 +000066#include "llvm/Target/TargetLowering.h"
Dan Gohman02c84b82008-08-20 21:05:57 +000067#include "llvm/Target/TargetMachine.h"
Dan Gohmanb2226e22008-08-13 20:19:35 +000068using namespace llvm;
69
Chandler Carruth1b9dde02014-04-22 02:02:50 +000070#define DEBUG_TYPE "isel"
71
Chad Rosier61e8d102011-11-28 19:59:09 +000072STATISTIC(NumFastIselSuccessIndependent, "Number of insts selected by "
73 "target-independent selector");
74STATISTIC(NumFastIselSuccessTarget, "Number of insts selected by "
75 "target-specific selector");
Chad Rosier46addb92011-11-29 19:40:47 +000076STATISTIC(NumFastIselDead, "Number of dead insts removed on failure");
Chad Rosierff40b1e2011-11-16 21:05:28 +000077
Juergen Ributzka8179e9e2014-07-11 22:01:42 +000078/// \brief Set CallLoweringInfo attribute flags based on a call instruction
79/// and called function attributes.
80void FastISel::ArgListEntry::setAttributes(ImmutableCallSite *CS,
81 unsigned AttrIdx) {
82 isSExt = CS->paramHasAttr(AttrIdx, Attribute::SExt);
83 isZExt = CS->paramHasAttr(AttrIdx, Attribute::ZExt);
84 isInReg = CS->paramHasAttr(AttrIdx, Attribute::InReg);
85 isSRet = CS->paramHasAttr(AttrIdx, Attribute::StructRet);
86 isNest = CS->paramHasAttr(AttrIdx, Attribute::Nest);
87 isByVal = CS->paramHasAttr(AttrIdx, Attribute::ByVal);
88 isInAlloca = CS->paramHasAttr(AttrIdx, Attribute::InAlloca);
89 isReturned = CS->paramHasAttr(AttrIdx, Attribute::Returned);
90 Alignment = CS->getParamAlignment(AttrIdx);
91}
92
Dan Gohmand7b5ce32010-07-10 09:00:22 +000093/// startNewBlock - Set the current block to which generated machine
94/// instructions will be appended, and clear the local CSE map.
95///
96void FastISel::startNewBlock() {
97 LocalValueMap.clear();
98
Jakob Stoklund Olesen6a7d6832013-07-04 04:53:49 +000099 // Instructions are appended to FuncInfo.MBB. If the basic block already
Jakob Stoklund Olesen3d8560c2013-07-04 04:32:39 +0000100 // contains labels or copies, use the last instruction as the last local
101 // value.
Craig Topperc0196b12014-04-14 00:51:57 +0000102 EmitStartPt = nullptr;
Jakob Stoklund Olesen3d8560c2013-07-04 04:32:39 +0000103 if (!FuncInfo.MBB->empty())
104 EmitStartPt = &FuncInfo.MBB->back();
Ivan Krasind7cbd4c2011-08-18 22:06:10 +0000105 LastLocalValue = EmitStartPt;
106}
107
Evan Cheng615620c2013-02-11 01:27:15 +0000108bool FastISel::LowerArguments() {
109 if (!FuncInfo.CanLowerReturn)
110 // Fallback to SDISel argument lowering code to deal with sret pointer
111 // parameter.
112 return false;
Stephen Lincfe7f352013-07-08 00:37:03 +0000113
Evan Cheng615620c2013-02-11 01:27:15 +0000114 if (!FastLowerArguments())
115 return false;
116
David Blaikie97c6c5b2013-06-21 22:56:30 +0000117 // Enter arguments into ValueMap for uses in non-entry BBs.
Evan Cheng615620c2013-02-11 01:27:15 +0000118 for (Function::const_arg_iterator I = FuncInfo.Fn->arg_begin(),
119 E = FuncInfo.Fn->arg_end(); I != E; ++I) {
David Blaikie97c6c5b2013-06-21 22:56:30 +0000120 DenseMap<const Value *, unsigned>::iterator VI = LocalValueMap.find(I);
121 assert(VI != LocalValueMap.end() && "Missed an argument?");
122 FuncInfo.ValueMap[I] = VI->second;
Evan Cheng615620c2013-02-11 01:27:15 +0000123 }
124 return true;
125}
126
Ivan Krasind7cbd4c2011-08-18 22:06:10 +0000127void FastISel::flushLocalValueMap() {
128 LocalValueMap.clear();
129 LastLocalValue = EmitStartPt;
130 recomputeInsertPt();
Dan Gohmand7b5ce32010-07-10 09:00:22 +0000131}
132
Dan Gohman1a1b51f2010-05-11 23:54:07 +0000133bool FastISel::hasTrivialKill(const Value *V) const {
Dan Gohman88fb2532010-05-14 22:53:18 +0000134 // Don't consider constants or arguments to have trivial kills.
Dan Gohman1a1b51f2010-05-11 23:54:07 +0000135 const Instruction *I = dyn_cast<Instruction>(V);
Dan Gohman88fb2532010-05-14 22:53:18 +0000136 if (!I)
137 return false;
138
139 // No-op casts are trivially coalesced by fast-isel.
140 if (const CastInst *Cast = dyn_cast<CastInst>(I))
Rafael Espindolaea09c592014-02-18 22:05:46 +0000141 if (Cast->isNoopCast(DL.getIntPtrType(Cast->getContext())) &&
Chandler Carruth7ec50852012-11-01 08:07:29 +0000142 !hasTrivialKill(Cast->getOperand(0)))
Dan Gohman88fb2532010-05-14 22:53:18 +0000143 return false;
144
Chad Rosier291ce472011-11-15 23:34:05 +0000145 // GEPs with all zero indices are trivially coalesced by fast-isel.
146 if (const GetElementPtrInst *GEP = dyn_cast<GetElementPtrInst>(I))
147 if (GEP->hasAllZeroIndices() && !hasTrivialKill(GEP->getOperand(0)))
148 return false;
149
Dan Gohman88fb2532010-05-14 22:53:18 +0000150 // Only instructions with a single use in the same basic block are considered
151 // to have trivial kills.
152 return I->hasOneUse() &&
153 !(I->getOpcode() == Instruction::BitCast ||
154 I->getOpcode() == Instruction::PtrToInt ||
155 I->getOpcode() == Instruction::IntToPtr) &&
Chandler Carruthcdf47882014-03-09 03:16:01 +0000156 cast<Instruction>(*I->user_begin())->getParent() == I->getParent();
Dan Gohman1a1b51f2010-05-11 23:54:07 +0000157}
158
Dan Gohmanbcaf6812010-04-15 01:51:59 +0000159unsigned FastISel::getRegForValue(const Value *V) {
Owen Anderson53aa7a92009-08-10 22:56:29 +0000160 EVT RealVT = TLI.getValueType(V->getType(), /*AllowUnknown=*/true);
Dan Gohmanca93aab2009-04-07 20:40:11 +0000161 // Don't handle non-simple values in FastISel.
162 if (!RealVT.isSimple())
163 return 0;
Dan Gohman4c315242008-12-08 07:57:47 +0000164
165 // Ignore illegal types. We must do this before looking up the value
166 // in ValueMap because Arguments are given virtual registers regardless
167 // of whether FastISel can handle them.
Owen Anderson9f944592009-08-11 20:47:22 +0000168 MVT VT = RealVT.getSimpleVT();
Dan Gohman4c315242008-12-08 07:57:47 +0000169 if (!TLI.isTypeLegal(VT)) {
Eli Friedmanc7035512011-05-25 23:49:02 +0000170 // Handle integer promotions, though, because they're common and easy.
171 if (VT == MVT::i1 || VT == MVT::i8 || VT == MVT::i16)
Owen Anderson117c9e82009-08-12 00:36:31 +0000172 VT = TLI.getTypeToTransformTo(V->getContext(), VT).getSimpleVT();
Dan Gohman4c315242008-12-08 07:57:47 +0000173 else
174 return 0;
175 }
176
Eric Christopher1a06cc92012-03-20 01:07:47 +0000177 // Look up the value to see if we already have a register for it.
178 unsigned Reg = lookUpRegForValue(V);
Dan Gohmane039d552008-09-03 23:32:19 +0000179 if (Reg != 0)
180 return Reg;
Dan Gohmanb0b5a272008-08-27 18:10:19 +0000181
Dan Gohmana7c717d82010-05-06 00:02:14 +0000182 // In bottom-up mode, just create the virtual register which will be used
183 // to hold the value. It will be materialized later.
Dan Gohmand7b5ce32010-07-10 09:00:22 +0000184 if (isa<Instruction>(V) &&
185 (!isa<AllocaInst>(V) ||
186 !FuncInfo.StaticAllocaMap.count(cast<AllocaInst>(V))))
187 return FuncInfo.InitializeRegForValue(V);
Dan Gohmana7c717d82010-05-06 00:02:14 +0000188
Eric Christopherf4fba5c2012-10-03 08:10:01 +0000189 SavePoint SaveInsertPt = enterLocalValueArea();
Dan Gohmand7b5ce32010-07-10 09:00:22 +0000190
191 // Materialize the value in a register. Emit any instructions in the
192 // local value area.
193 Reg = materializeRegForValue(V, VT);
194
Eric Christopherf4fba5c2012-10-03 08:10:01 +0000195 leaveLocalValueArea(SaveInsertPt);
Dan Gohmand7b5ce32010-07-10 09:00:22 +0000196
197 return Reg;
Dan Gohman626b5d82010-05-03 23:36:34 +0000198}
199
Eric Christopher541f8012010-08-17 01:30:33 +0000200/// materializeRegForValue - Helper for getRegForValue. This function is
Dan Gohman626b5d82010-05-03 23:36:34 +0000201/// called when the value isn't already available in a register and must
202/// be materialized with new instructions.
203unsigned FastISel::materializeRegForValue(const Value *V, MVT VT) {
204 unsigned Reg = 0;
205
Dan Gohmanbcaf6812010-04-15 01:51:59 +0000206 if (const ConstantInt *CI = dyn_cast<ConstantInt>(V)) {
Dan Gohman9801ba42008-09-19 22:16:54 +0000207 if (CI->getValue().getActiveBits() <= 64)
208 Reg = FastEmit_i(VT, VT, ISD::Constant, CI->getZExtValue());
Dan Gohman39d82f92008-09-10 20:11:02 +0000209 } else if (isa<AllocaInst>(V)) {
Dan Gohman9801ba42008-09-19 22:16:54 +0000210 Reg = TargetMaterializeAlloca(cast<AllocaInst>(V));
Dan Gohmanc45733f2008-08-28 21:19:07 +0000211 } else if (isa<ConstantPointerNull>(V)) {
Dan Gohmanc1d47c52008-10-07 22:03:27 +0000212 // Translate this as an integer zero so that it can be
213 // local-CSE'd with actual integer zeros.
Owen Anderson55f1c092009-08-13 21:58:54 +0000214 Reg =
Rafael Espindolaea09c592014-02-18 22:05:46 +0000215 getRegForValue(Constant::getNullValue(DL.getIntPtrType(V->getContext())));
Dan Gohmanbcaf6812010-04-15 01:51:59 +0000216 } else if (const ConstantFP *CF = dyn_cast<ConstantFP>(V)) {
Eli Friedman33c13392011-04-28 00:42:03 +0000217 if (CF->isNullValue()) {
Eli Friedman406c4712011-04-27 22:41:55 +0000218 Reg = TargetMaterializeFloatZero(CF);
219 } else {
220 // Try to emit the constant directly.
221 Reg = FastEmit_f(VT, VT, ISD::ConstantFP, CF);
222 }
Dan Gohmanb0b5a272008-08-27 18:10:19 +0000223
224 if (!Reg) {
Dan Gohman8a2dae52010-04-13 17:07:06 +0000225 // Try to emit the constant by using an integer constant with a cast.
Dan Gohmanb0b5a272008-08-27 18:10:19 +0000226 const APFloat &Flt = CF->getValueAPF();
Owen Anderson53aa7a92009-08-10 22:56:29 +0000227 EVT IntVT = TLI.getPointerTy();
Dan Gohmanb0b5a272008-08-27 18:10:19 +0000228
229 uint64_t x[2];
230 uint32_t IntBitWidth = IntVT.getSizeInBits();
Dale Johannesen4f0bd682008-10-09 23:00:39 +0000231 bool isExact;
232 (void) Flt.convertToInteger(x, IntBitWidth, /*isSigned=*/true,
Eric Christopher997aaa92012-03-20 01:07:56 +0000233 APFloat::rmTowardZero, &isExact);
Dale Johannesen4f0bd682008-10-09 23:00:39 +0000234 if (isExact) {
Jeffrey Yasskin7a162882011-07-18 21:45:40 +0000235 APInt IntVal(IntBitWidth, x);
Dan Gohmanb0b5a272008-08-27 18:10:19 +0000236
Owen Anderson47db9412009-07-22 00:24:57 +0000237 unsigned IntegerReg =
Owen Andersonedb4a702009-07-24 23:12:02 +0000238 getRegForValue(ConstantInt::get(V->getContext(), IntVal));
Dan Gohman9801ba42008-09-19 22:16:54 +0000239 if (IntegerReg != 0)
Dan Gohman1a1b51f2010-05-11 23:54:07 +0000240 Reg = FastEmit_r(IntVT.getSimpleVT(), VT, ISD::SINT_TO_FP,
241 IntegerReg, /*Kill=*/false);
Dan Gohman9801ba42008-09-19 22:16:54 +0000242 }
Dan Gohmanb0b5a272008-08-27 18:10:19 +0000243 }
Dan Gohmanbcaf6812010-04-15 01:51:59 +0000244 } else if (const Operator *Op = dyn_cast<Operator>(V)) {
Dan Gohman722f5fc2010-07-01 02:58:57 +0000245 if (!SelectOperator(Op, Op->getOpcode()))
246 if (!isa<Instruction>(Op) ||
247 !TargetSelectInstruction(cast<Instruction>(Op)))
248 return 0;
Dan Gohman7c58cf72010-06-21 14:17:46 +0000249 Reg = lookUpRegForValue(Op);
Dan Gohmanc45733f2008-08-28 21:19:07 +0000250 } else if (isa<UndefValue>(V)) {
Dan Gohmane039d552008-09-03 23:32:19 +0000251 Reg = createResultReg(TLI.getRegClassFor(VT));
Rafael Espindolaea09c592014-02-18 22:05:46 +0000252 BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc,
Dan Gohmand7b5ce32010-07-10 09:00:22 +0000253 TII.get(TargetOpcode::IMPLICIT_DEF), Reg);
Dan Gohmanb0b5a272008-08-27 18:10:19 +0000254 }
Wesley Peck527da1b2010-11-23 03:31:01 +0000255
Dan Gohman3663f152008-09-25 01:28:51 +0000256 // If target-independent code couldn't handle the value, give target-specific
257 // code a try.
Owen Anderson1dd2e402008-09-05 23:36:01 +0000258 if (!Reg && isa<Constant>(V))
Dan Gohman9801ba42008-09-19 22:16:54 +0000259 Reg = TargetMaterializeConstant(cast<Constant>(V));
Wesley Peck527da1b2010-11-23 03:31:01 +0000260
Dan Gohman9801ba42008-09-19 22:16:54 +0000261 // Don't cache constant materializations in the general ValueMap.
262 // To do so would require tracking what uses they dominate.
Dan Gohmand7b5ce32010-07-10 09:00:22 +0000263 if (Reg != 0) {
Dan Gohman3663f152008-09-25 01:28:51 +0000264 LocalValueMap[V] = Reg;
Dan Gohmand7b5ce32010-07-10 09:00:22 +0000265 LastLocalValue = MRI.getVRegDef(Reg);
266 }
Dan Gohmane039d552008-09-03 23:32:19 +0000267 return Reg;
Dan Gohmanb0b5a272008-08-27 18:10:19 +0000268}
269
Dan Gohmanbcaf6812010-04-15 01:51:59 +0000270unsigned FastISel::lookUpRegForValue(const Value *V) {
Evan Cheng1e979012008-09-09 01:26:59 +0000271 // Look up the value to see if we already have a register for it. We
272 // cache values defined by Instructions across blocks, and other values
273 // only locally. This is because Instructions already have the SSA
Dan Gohman626b5d82010-05-03 23:36:34 +0000274 // def-dominates-use requirement enforced.
Dan Gohman87fb4e82010-07-07 16:29:44 +0000275 DenseMap<const Value *, unsigned>::iterator I = FuncInfo.ValueMap.find(V);
276 if (I != FuncInfo.ValueMap.end())
Dan Gohmanf91aff52010-06-21 14:21:47 +0000277 return I->second;
Eric Christopherf4fba5c2012-10-03 08:10:01 +0000278 return LocalValueMap[V];
Evan Cheng1e979012008-09-09 01:26:59 +0000279}
280
Owen Anderson6f0c51d2008-08-30 00:38:46 +0000281/// UpdateValueMap - Update the value map to include the new mapping for this
282/// instruction, or insert an extra copy to get the result in a previous
283/// determined register.
284/// NOTE: This is only necessary because we might select a block that uses
285/// a value before we select the block that defines the value. It might be
286/// possible to fix this by selecting blocks in reverse postorder.
Eli Friedmana4d4a012011-05-16 21:06:17 +0000287void FastISel::UpdateValueMap(const Value *I, unsigned Reg, unsigned NumRegs) {
Dan Gohmanfcf54562008-09-05 18:18:20 +0000288 if (!isa<Instruction>(I)) {
289 LocalValueMap[I] = Reg;
Eli Friedmana4d4a012011-05-16 21:06:17 +0000290 return;
Dan Gohmanfcf54562008-09-05 18:18:20 +0000291 }
Wesley Peck527da1b2010-11-23 03:31:01 +0000292
Dan Gohman87fb4e82010-07-07 16:29:44 +0000293 unsigned &AssignedReg = FuncInfo.ValueMap[I];
Chris Lattnerada5d6c2009-04-12 07:45:01 +0000294 if (AssignedReg == 0)
Dan Gohmand7b5ce32010-07-10 09:00:22 +0000295 // Use the new register.
Chris Lattnerada5d6c2009-04-12 07:45:01 +0000296 AssignedReg = Reg;
Chris Lattnera101f6f2009-04-12 07:46:30 +0000297 else if (Reg != AssignedReg) {
Dan Gohmand7b5ce32010-07-10 09:00:22 +0000298 // Arrange for uses of AssignedReg to be replaced by uses of Reg.
Eli Friedmana4d4a012011-05-16 21:06:17 +0000299 for (unsigned i = 0; i < NumRegs; i++)
300 FuncInfo.RegFixups[AssignedReg+i] = Reg+i;
Dan Gohmand7b5ce32010-07-10 09:00:22 +0000301
302 AssignedReg = Reg;
Chris Lattnerada5d6c2009-04-12 07:45:01 +0000303 }
Owen Anderson6f0c51d2008-08-30 00:38:46 +0000304}
305
Dan Gohman1a1b51f2010-05-11 23:54:07 +0000306std::pair<unsigned, bool> FastISel::getRegForGEPIndex(const Value *Idx) {
Dan Gohman4c315242008-12-08 07:57:47 +0000307 unsigned IdxN = getRegForValue(Idx);
308 if (IdxN == 0)
309 // Unhandled operand. Halt "fast" selection and bail.
Dan Gohman1a1b51f2010-05-11 23:54:07 +0000310 return std::pair<unsigned, bool>(0, false);
311
312 bool IdxNIsKill = hasTrivialKill(Idx);
Dan Gohman4c315242008-12-08 07:57:47 +0000313
314 // If the index is smaller or larger than intptr_t, truncate or extend it.
Owen Andersonc6daf8f2009-08-11 21:59:30 +0000315 MVT PtrVT = TLI.getPointerTy();
Owen Anderson53aa7a92009-08-10 22:56:29 +0000316 EVT IdxVT = EVT::getEVT(Idx->getType(), /*HandleUnknown=*/false);
Dan Gohman1a1b51f2010-05-11 23:54:07 +0000317 if (IdxVT.bitsLT(PtrVT)) {
318 IdxN = FastEmit_r(IdxVT.getSimpleVT(), PtrVT, ISD::SIGN_EXTEND,
319 IdxN, IdxNIsKill);
320 IdxNIsKill = true;
321 }
322 else if (IdxVT.bitsGT(PtrVT)) {
323 IdxN = FastEmit_r(IdxVT.getSimpleVT(), PtrVT, ISD::TRUNCATE,
324 IdxN, IdxNIsKill);
325 IdxNIsKill = true;
326 }
327 return std::pair<unsigned, bool>(IdxN, IdxNIsKill);
Dan Gohman4c315242008-12-08 07:57:47 +0000328}
329
Dan Gohmand7b5ce32010-07-10 09:00:22 +0000330void FastISel::recomputeInsertPt() {
331 if (getLastLocalValue()) {
332 FuncInfo.InsertPt = getLastLocalValue();
Dan Gohmanb5e918d2010-07-19 22:48:56 +0000333 FuncInfo.MBB = FuncInfo.InsertPt->getParent();
Dan Gohmand7b5ce32010-07-10 09:00:22 +0000334 ++FuncInfo.InsertPt;
335 } else
336 FuncInfo.InsertPt = FuncInfo.MBB->getFirstNonPHI();
337
338 // Now skip past any EH_LABELs, which must remain at the beginning.
339 while (FuncInfo.InsertPt != FuncInfo.MBB->end() &&
340 FuncInfo.InsertPt->getOpcode() == TargetOpcode::EH_LABEL)
341 ++FuncInfo.InsertPt;
342}
343
Chad Rosier46addb92011-11-29 19:40:47 +0000344void FastISel::removeDeadCode(MachineBasicBlock::iterator I,
345 MachineBasicBlock::iterator E) {
346 assert (I && E && std::distance(I, E) > 0 && "Invalid iterator!");
347 while (I != E) {
348 MachineInstr *Dead = &*I;
349 ++I;
350 Dead->eraseFromParent();
Jan Wen Voung7857a642013-03-08 22:56:31 +0000351 ++NumFastIselDead;
Chad Rosier46addb92011-11-29 19:40:47 +0000352 }
353 recomputeInsertPt();
354}
355
Eric Christopherf4fba5c2012-10-03 08:10:01 +0000356FastISel::SavePoint FastISel::enterLocalValueArea() {
Dan Gohmand7b5ce32010-07-10 09:00:22 +0000357 MachineBasicBlock::iterator OldInsertPt = FuncInfo.InsertPt;
Rafael Espindolaea09c592014-02-18 22:05:46 +0000358 DebugLoc OldDL = DbgLoc;
Dan Gohmand7b5ce32010-07-10 09:00:22 +0000359 recomputeInsertPt();
Rafael Espindolaea09c592014-02-18 22:05:46 +0000360 DbgLoc = DebugLoc();
Eric Christopherf4fba5c2012-10-03 08:10:01 +0000361 SavePoint SP = { OldInsertPt, OldDL };
362 return SP;
Dan Gohmand7b5ce32010-07-10 09:00:22 +0000363}
364
Eric Christopherf4fba5c2012-10-03 08:10:01 +0000365void FastISel::leaveLocalValueArea(SavePoint OldInsertPt) {
Dan Gohmand7b5ce32010-07-10 09:00:22 +0000366 if (FuncInfo.InsertPt != FuncInfo.MBB->begin())
Benjamin Kramerb6d0bd42014-03-02 12:27:27 +0000367 LastLocalValue = std::prev(FuncInfo.InsertPt);
Dan Gohmand7b5ce32010-07-10 09:00:22 +0000368
369 // Restore the previous insert position.
Eric Christopherf4fba5c2012-10-03 08:10:01 +0000370 FuncInfo.InsertPt = OldInsertPt.InsertPt;
Rafael Espindolaea09c592014-02-18 22:05:46 +0000371 DbgLoc = OldInsertPt.DL;
Dan Gohmand7b5ce32010-07-10 09:00:22 +0000372}
373
Dan Gohmana3e4d5a2008-08-20 00:11:48 +0000374/// SelectBinaryOp - Select and emit code for a binary operator instruction,
375/// which has an opcode which directly corresponds to the given ISD opcode.
376///
Dan Gohmanbcaf6812010-04-15 01:51:59 +0000377bool FastISel::SelectBinaryOp(const User *I, unsigned ISDOpcode) {
Owen Anderson53aa7a92009-08-10 22:56:29 +0000378 EVT VT = EVT::getEVT(I->getType(), /*HandleUnknown=*/true);
Owen Anderson9f944592009-08-11 20:47:22 +0000379 if (VT == MVT::Other || !VT.isSimple())
Dan Gohmana3e4d5a2008-08-20 00:11:48 +0000380 // Unhandled type. Halt "fast" selection and bail.
381 return false;
Dan Gohmanfd634592008-09-05 18:44:22 +0000382
Dan Gohman3bcbbec2008-08-26 20:52:40 +0000383 // We only handle legal types. For example, on x86-32 the instruction
384 // selector contains all of the 64-bit instructions from x86-64,
385 // under the assumption that i64 won't be used if the target doesn't
386 // support it.
Dan Gohmanfd634592008-09-05 18:44:22 +0000387 if (!TLI.isTypeLegal(VT)) {
Owen Anderson9f944592009-08-11 20:47:22 +0000388 // MVT::i1 is special. Allow AND, OR, or XOR because they
Dan Gohmanfd634592008-09-05 18:44:22 +0000389 // don't require additional zeroing, which makes them easy.
Owen Anderson9f944592009-08-11 20:47:22 +0000390 if (VT == MVT::i1 &&
Dan Gohman5e490a72008-09-25 17:22:52 +0000391 (ISDOpcode == ISD::AND || ISDOpcode == ISD::OR ||
392 ISDOpcode == ISD::XOR))
Owen Anderson117c9e82009-08-12 00:36:31 +0000393 VT = TLI.getTypeToTransformTo(I->getContext(), VT);
Dan Gohmanfd634592008-09-05 18:44:22 +0000394 else
395 return false;
396 }
Dan Gohmana3e4d5a2008-08-20 00:11:48 +0000397
Chris Lattnerfba7ca62011-04-17 01:16:47 +0000398 // Check if the first operand is a constant, and handle it as "ri". At -O0,
399 // we don't have anything that canonicalizes operand order.
400 if (ConstantInt *CI = dyn_cast<ConstantInt>(I->getOperand(0)))
401 if (isa<Instruction>(I) && cast<Instruction>(I)->isCommutative()) {
402 unsigned Op1 = getRegForValue(I->getOperand(1));
403 if (Op1 == 0) return false;
404
405 bool Op1IsKill = hasTrivialKill(I->getOperand(1));
Owen Andersondd450b82011-04-22 23:38:06 +0000406
Chris Lattnerb53ccb82011-04-17 20:23:29 +0000407 unsigned ResultReg = FastEmit_ri_(VT.getSimpleVT(), ISDOpcode, Op1,
408 Op1IsKill, CI->getZExtValue(),
409 VT.getSimpleVT());
410 if (ResultReg == 0) return false;
Owen Andersondd450b82011-04-22 23:38:06 +0000411
Chris Lattnerb53ccb82011-04-17 20:23:29 +0000412 // We successfully emitted code for the given LLVM Instruction.
413 UpdateValueMap(I, ResultReg);
414 return true;
Chris Lattnerfba7ca62011-04-17 01:16:47 +0000415 }
Owen Andersondd450b82011-04-22 23:38:06 +0000416
417
Dan Gohman7bda51f2008-09-03 23:12:08 +0000418 unsigned Op0 = getRegForValue(I->getOperand(0));
Chris Lattnerb53ccb82011-04-17 20:23:29 +0000419 if (Op0 == 0) // Unhandled operand. Halt "fast" selection and bail.
Dan Gohmanfe905652008-08-21 01:41:07 +0000420 return false;
421
Dan Gohman1a1b51f2010-05-11 23:54:07 +0000422 bool Op0IsKill = hasTrivialKill(I->getOperand(0));
423
Dan Gohmanfe905652008-08-21 01:41:07 +0000424 // Check if the second operand is a constant and handle it appropriately.
425 if (ConstantInt *CI = dyn_cast<ConstantInt>(I->getOperand(1))) {
Chris Lattnerb53ccb82011-04-17 20:23:29 +0000426 uint64_t Imm = CI->getZExtValue();
Owen Andersondd450b82011-04-22 23:38:06 +0000427
Chris Lattner48f75ad2011-04-18 07:00:40 +0000428 // Transform "sdiv exact X, 8" -> "sra X, 3".
429 if (ISDOpcode == ISD::SDIV && isa<BinaryOperator>(I) &&
430 cast<BinaryOperator>(I)->isExact() &&
431 isPowerOf2_64(Imm)) {
432 Imm = Log2_64(Imm);
433 ISDOpcode = ISD::SRA;
434 }
Owen Andersondd450b82011-04-22 23:38:06 +0000435
Chad Rosier6a63a742012-03-22 00:21:17 +0000436 // Transform "urem x, pow2" -> "and x, pow2-1".
437 if (ISDOpcode == ISD::UREM && isa<BinaryOperator>(I) &&
438 isPowerOf2_64(Imm)) {
439 --Imm;
440 ISDOpcode = ISD::AND;
441 }
442
Chris Lattnerb53ccb82011-04-17 20:23:29 +0000443 unsigned ResultReg = FastEmit_ri_(VT.getSimpleVT(), ISDOpcode, Op0,
444 Op0IsKill, Imm, VT.getSimpleVT());
445 if (ResultReg == 0) return false;
Owen Andersondd450b82011-04-22 23:38:06 +0000446
Chris Lattnerb53ccb82011-04-17 20:23:29 +0000447 // We successfully emitted code for the given LLVM Instruction.
448 UpdateValueMap(I, ResultReg);
449 return true;
Dan Gohmanfe905652008-08-21 01:41:07 +0000450 }
451
Dan Gohman5ca269e2008-08-27 01:09:54 +0000452 // Check if the second operand is a constant float.
453 if (ConstantFP *CF = dyn_cast<ConstantFP>(I->getOperand(1))) {
Dan Gohmanb0b5a272008-08-27 18:10:19 +0000454 unsigned ResultReg = FastEmit_rf(VT.getSimpleVT(), VT.getSimpleVT(),
Dan Gohman1a1b51f2010-05-11 23:54:07 +0000455 ISDOpcode, Op0, Op0IsKill, CF);
Dan Gohmanb0b5a272008-08-27 18:10:19 +0000456 if (ResultReg != 0) {
457 // We successfully emitted code for the given LLVM Instruction.
Dan Gohman7bda51f2008-09-03 23:12:08 +0000458 UpdateValueMap(I, ResultReg);
Dan Gohmanb0b5a272008-08-27 18:10:19 +0000459 return true;
460 }
Dan Gohman5ca269e2008-08-27 01:09:54 +0000461 }
462
Dan Gohman7bda51f2008-09-03 23:12:08 +0000463 unsigned Op1 = getRegForValue(I->getOperand(1));
Dan Gohmanfe905652008-08-21 01:41:07 +0000464 if (Op1 == 0)
465 // Unhandled operand. Halt "fast" selection and bail.
466 return false;
467
Dan Gohman1a1b51f2010-05-11 23:54:07 +0000468 bool Op1IsKill = hasTrivialKill(I->getOperand(1));
469
Dan Gohmanb0b5a272008-08-27 18:10:19 +0000470 // Now we have both operands in registers. Emit the instruction.
Owen Anderson8dd01cc2008-08-25 23:58:18 +0000471 unsigned ResultReg = FastEmit_rr(VT.getSimpleVT(), VT.getSimpleVT(),
Dan Gohman1a1b51f2010-05-11 23:54:07 +0000472 ISDOpcode,
473 Op0, Op0IsKill,
474 Op1, Op1IsKill);
Dan Gohmana3e4d5a2008-08-20 00:11:48 +0000475 if (ResultReg == 0)
476 // Target-specific code wasn't able to find a machine opcode for
477 // the given ISD opcode and type. Halt "fast" selection and bail.
478 return false;
479
Dan Gohmanb16a7782008-08-20 00:23:20 +0000480 // We successfully emitted code for the given LLVM Instruction.
Dan Gohman7bda51f2008-09-03 23:12:08 +0000481 UpdateValueMap(I, ResultReg);
Dan Gohmana3e4d5a2008-08-20 00:11:48 +0000482 return true;
483}
484
Dan Gohmanbcaf6812010-04-15 01:51:59 +0000485bool FastISel::SelectGetElementPtr(const User *I) {
Dan Gohman7bda51f2008-09-03 23:12:08 +0000486 unsigned N = getRegForValue(I->getOperand(0));
Evan Cheng864fcc12008-08-20 22:45:34 +0000487 if (N == 0)
488 // Unhandled operand. Halt "fast" selection and bail.
489 return false;
490
Dan Gohman1a1b51f2010-05-11 23:54:07 +0000491 bool NIsKill = hasTrivialKill(I->getOperand(0));
492
Chad Rosierf83ab702011-11-17 07:15:58 +0000493 // Keep a running tab of the total offset to coalesce multiple N = N + Offset
494 // into a single N = N + TotalOffset.
495 uint64_t TotalOffs = 0;
496 // FIXME: What's a good SWAG number for MaxOffs?
497 uint64_t MaxOffs = 2048;
Chris Lattner229907c2011-07-18 04:54:35 +0000498 Type *Ty = I->getOperand(0)->getType();
Owen Anderson9f944592009-08-11 20:47:22 +0000499 MVT VT = TLI.getPointerTy();
Dan Gohmanbcaf6812010-04-15 01:51:59 +0000500 for (GetElementPtrInst::const_op_iterator OI = I->op_begin()+1,
501 E = I->op_end(); OI != E; ++OI) {
502 const Value *Idx = *OI;
Chris Lattner229907c2011-07-18 04:54:35 +0000503 if (StructType *StTy = dyn_cast<StructType>(Ty)) {
Evan Cheng864fcc12008-08-20 22:45:34 +0000504 unsigned Field = cast<ConstantInt>(Idx)->getZExtValue();
505 if (Field) {
506 // N = N + Offset
Rafael Espindolaea09c592014-02-18 22:05:46 +0000507 TotalOffs += DL.getStructLayout(StTy)->getElementOffset(Field);
Chad Rosierf83ab702011-11-17 07:15:58 +0000508 if (TotalOffs >= MaxOffs) {
509 N = FastEmit_ri_(VT, ISD::ADD, N, NIsKill, TotalOffs, VT);
510 if (N == 0)
511 // Unhandled operand. Halt "fast" selection and bail.
512 return false;
513 NIsKill = true;
514 TotalOffs = 0;
515 }
Evan Cheng864fcc12008-08-20 22:45:34 +0000516 }
517 Ty = StTy->getElementType(Field);
518 } else {
519 Ty = cast<SequentialType>(Ty)->getElementType();
520
521 // If this is a constant subscript, handle it quickly.
Dan Gohmanbcaf6812010-04-15 01:51:59 +0000522 if (const ConstantInt *CI = dyn_cast<ConstantInt>(Idx)) {
Dan Gohmanf1d83042010-06-18 14:22:04 +0000523 if (CI->isZero()) continue;
Chad Rosierf83ab702011-11-17 07:15:58 +0000524 // N = N + Offset
Chad Rosier879c34f2012-07-06 17:44:22 +0000525 TotalOffs +=
Rafael Espindolaea09c592014-02-18 22:05:46 +0000526 DL.getTypeAllocSize(Ty)*cast<ConstantInt>(CI)->getSExtValue();
Chad Rosierf83ab702011-11-17 07:15:58 +0000527 if (TotalOffs >= MaxOffs) {
528 N = FastEmit_ri_(VT, ISD::ADD, N, NIsKill, TotalOffs, VT);
529 if (N == 0)
530 // Unhandled operand. Halt "fast" selection and bail.
531 return false;
532 NIsKill = true;
533 TotalOffs = 0;
534 }
535 continue;
536 }
537 if (TotalOffs) {
538 N = FastEmit_ri_(VT, ISD::ADD, N, NIsKill, TotalOffs, VT);
Evan Cheng864fcc12008-08-20 22:45:34 +0000539 if (N == 0)
540 // Unhandled operand. Halt "fast" selection and bail.
541 return false;
Dan Gohman1a1b51f2010-05-11 23:54:07 +0000542 NIsKill = true;
Chad Rosierf83ab702011-11-17 07:15:58 +0000543 TotalOffs = 0;
Evan Cheng864fcc12008-08-20 22:45:34 +0000544 }
Wesley Peck527da1b2010-11-23 03:31:01 +0000545
Evan Cheng864fcc12008-08-20 22:45:34 +0000546 // N = N + Idx * ElementSize;
Rafael Espindolaea09c592014-02-18 22:05:46 +0000547 uint64_t ElementSize = DL.getTypeAllocSize(Ty);
Dan Gohman1a1b51f2010-05-11 23:54:07 +0000548 std::pair<unsigned, bool> Pair = getRegForGEPIndex(Idx);
549 unsigned IdxN = Pair.first;
550 bool IdxNIsKill = Pair.second;
Evan Cheng864fcc12008-08-20 22:45:34 +0000551 if (IdxN == 0)
552 // Unhandled operand. Halt "fast" selection and bail.
553 return false;
554
Dan Gohmanb5e04bf2008-08-26 20:57:08 +0000555 if (ElementSize != 1) {
Dan Gohman1a1b51f2010-05-11 23:54:07 +0000556 IdxN = FastEmit_ri_(VT, ISD::MUL, IdxN, IdxNIsKill, ElementSize, VT);
Dan Gohmanb5e04bf2008-08-26 20:57:08 +0000557 if (IdxN == 0)
558 // Unhandled operand. Halt "fast" selection and bail.
559 return false;
Dan Gohman1a1b51f2010-05-11 23:54:07 +0000560 IdxNIsKill = true;
Dan Gohmanb5e04bf2008-08-26 20:57:08 +0000561 }
Dan Gohman1a1b51f2010-05-11 23:54:07 +0000562 N = FastEmit_rr(VT, VT, ISD::ADD, N, NIsKill, IdxN, IdxNIsKill);
Evan Cheng864fcc12008-08-20 22:45:34 +0000563 if (N == 0)
564 // Unhandled operand. Halt "fast" selection and bail.
565 return false;
566 }
567 }
Chad Rosierf83ab702011-11-17 07:15:58 +0000568 if (TotalOffs) {
569 N = FastEmit_ri_(VT, ISD::ADD, N, NIsKill, TotalOffs, VT);
570 if (N == 0)
571 // Unhandled operand. Halt "fast" selection and bail.
572 return false;
573 }
Evan Cheng864fcc12008-08-20 22:45:34 +0000574
575 // We successfully emitted code for the given LLVM Instruction.
Dan Gohman7bda51f2008-09-03 23:12:08 +0000576 UpdateValueMap(I, N);
Evan Cheng864fcc12008-08-20 22:45:34 +0000577 return true;
Dan Gohmana3e4d5a2008-08-20 00:11:48 +0000578}
579
Juergen Ributzka190305b2014-07-01 22:25:49 +0000580/// \brief Add a stackmap or patchpoint intrinsic call's live variable operands
581/// to a stackmap or patchpoint machine instruction.
Juergen Ributzka04558dc2014-06-12 03:29:26 +0000582bool FastISel::addStackMapLiveVars(SmallVectorImpl<MachineOperand> &Ops,
583 const CallInst *CI, unsigned StartIdx) {
584 for (unsigned i = StartIdx, e = CI->getNumArgOperands(); i != e; ++i) {
585 Value *Val = CI->getArgOperand(i);
Juergen Ributzka190305b2014-07-01 22:25:49 +0000586 // Check for constants and encode them with a StackMaps::ConstantOp prefix.
Juergen Ributzka04558dc2014-06-12 03:29:26 +0000587 if (auto *C = dyn_cast<ConstantInt>(Val)) {
588 Ops.push_back(MachineOperand::CreateImm(StackMaps::ConstantOp));
589 Ops.push_back(MachineOperand::CreateImm(C->getSExtValue()));
590 } else if (isa<ConstantPointerNull>(Val)) {
591 Ops.push_back(MachineOperand::CreateImm(StackMaps::ConstantOp));
592 Ops.push_back(MachineOperand::CreateImm(0));
593 } else if (auto *AI = dyn_cast<AllocaInst>(Val)) {
Juergen Ributzka190305b2014-07-01 22:25:49 +0000594 // Values coming from a stack location also require a sepcial encoding,
595 // but that is added later on by the target specific frame index
596 // elimination implementation.
Juergen Ributzka04558dc2014-06-12 03:29:26 +0000597 auto SI = FuncInfo.StaticAllocaMap.find(AI);
598 if (SI != FuncInfo.StaticAllocaMap.end())
599 Ops.push_back(MachineOperand::CreateFI(SI->second));
600 else
601 return false;
602 } else {
603 unsigned Reg = getRegForValue(Val);
604 if (Reg == 0)
605 return false;
606 Ops.push_back(MachineOperand::CreateReg(Reg, /*IsDef=*/false));
607 }
608 }
609
610 return true;
611}
612
Juergen Ributzka190305b2014-07-01 22:25:49 +0000613bool FastISel::SelectStackmap(const CallInst *I) {
614 // void @llvm.experimental.stackmap(i64 <id>, i32 <numShadowBytes>,
615 // [live variables...])
616 assert(I->getCalledFunction()->getReturnType()->isVoidTy() &&
617 "Stackmap cannot return a value.");
618
619 // The stackmap intrinsic only records the live variables (the arguments
620 // passed to it) and emits NOPS (if requested). Unlike the patchpoint
621 // intrinsic, this won't be lowered to a function call. This means we don't
622 // have to worry about calling conventions and target-specific lowering code.
623 // Instead we perform the call lowering right here.
624 //
625 // CALLSEQ_START(0)
626 // STACKMAP(id, nbytes, ...)
627 // CALLSEQ_END(0, 0)
628 //
629 SmallVector<MachineOperand, 32> Ops;
630
631 // Add the <id> and <numBytes> constants.
632 assert(isa<ConstantInt>(I->getOperand(PatchPointOpers::IDPos)) &&
633 "Expected a constant integer.");
634 const auto *ID = cast<ConstantInt>(I->getOperand(PatchPointOpers::IDPos));
635 Ops.push_back(MachineOperand::CreateImm(ID->getZExtValue()));
636
637 assert(isa<ConstantInt>(I->getOperand(PatchPointOpers::NBytesPos)) &&
638 "Expected a constant integer.");
639 const auto *NumBytes =
640 cast<ConstantInt>(I->getOperand(PatchPointOpers::NBytesPos));
641 Ops.push_back(MachineOperand::CreateImm(NumBytes->getZExtValue()));
642
643 // Push live variables for the stack map (skipping the first two arguments
644 // <id> and <numBytes>).
645 if (!addStackMapLiveVars(Ops, I, 2))
646 return false;
647
648 // We are not adding any register mask info here, because the stackmap doesn't
649 // clobber anything.
650
651 // Add scratch registers as implicit def and early clobber.
652 CallingConv::ID CC = I->getCallingConv();
653 const MCPhysReg *ScratchRegs = TLI.getScratchRegisters(CC);
654 for (unsigned i = 0; ScratchRegs[i]; ++i)
655 Ops.push_back(MachineOperand::CreateReg(
656 ScratchRegs[i], /*IsDef=*/true, /*IsImp=*/true, /*IsKill=*/false,
657 /*IsDead=*/false, /*IsUndef=*/false, /*IsEarlyClobber=*/true));
658
659 // Issue CALLSEQ_START
660 unsigned AdjStackDown = TII.getCallFrameSetupOpcode();
661 BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc, TII.get(AdjStackDown))
662 .addImm(0);
663
664 // Issue STACKMAP.
665 MachineInstrBuilder MIB = BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc,
666 TII.get(TargetOpcode::STACKMAP));
667 for (auto const &MO : Ops)
668 MIB.addOperand(MO);
669
670 // Issue CALLSEQ_END
671 unsigned AdjStackUp = TII.getCallFrameDestroyOpcode();
672 BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc, TII.get(AdjStackUp))
673 .addImm(0).addImm(0);
674
675 // Inform the Frame Information that we have a stackmap in this function.
676 FuncInfo.MF->getFrameInfo()->setHasStackMap();
677
678 return true;
679}
680
Juergen Ributzka3d9e6752014-07-11 22:19:02 +0000681/// \brief Lower an argument list according to the target calling convention.
682///
683/// This is a helper for lowering intrinsics that follow a target calling
684/// convention or require stack pointer adjustment. Only a subset of the
685/// intrinsic's operands need to participate in the calling convention.
686bool FastISel::lowerCallOperands(const CallInst *CI, unsigned ArgIdx,
687 unsigned NumArgs, const Value *Callee,
688 bool ForceRetVoidTy, CallLoweringInfo &CLI) {
689 ArgListTy Args;
690 Args.reserve(NumArgs);
691
692 // Populate the argument list.
693 // Attributes for args start at offset 1, after the return attribute.
694 ImmutableCallSite CS(CI);
695 for (unsigned ArgI = ArgIdx, ArgE = ArgIdx + NumArgs, AttrI = ArgIdx + 1;
696 ArgI != ArgE; ++ArgI) {
697 Value *V = CI->getOperand(ArgI);
698
699 assert(!V->getType()->isEmptyTy() && "Empty type passed to intrinsic.");
700
701 ArgListEntry Entry;
702 Entry.Val = V;
703 Entry.Ty = V->getType();
704 Entry.setAttributes(&CS, AttrI);
705 Args.push_back(Entry);
706 }
707
708 Type *RetTy = ForceRetVoidTy ? Type::getVoidTy(CI->getType()->getContext())
709 : CI->getType();
710 CLI.setCallee(CI->getCallingConv(), RetTy, Callee, std::move(Args), NumArgs);
711
712 return LowerCallTo(CLI);
713}
714
715bool FastISel::SelectPatchpoint(const CallInst *I) {
716 // void|i64 @llvm.experimental.patchpoint.void|i64(i64 <id>,
717 // i32 <numBytes>,
718 // i8* <target>,
719 // i32 <numArgs>,
720 // [Args...],
721 // [live variables...])
722 CallingConv::ID CC = I->getCallingConv();
723 bool IsAnyRegCC = CC == CallingConv::AnyReg;
724 bool HasDef = !I->getType()->isVoidTy();
725 Value *Callee = I->getOperand(PatchPointOpers::TargetPos);
726
727 // Get the real number of arguments participating in the call <numArgs>
728 assert(isa<ConstantInt>(I->getOperand(PatchPointOpers::NArgPos)) &&
729 "Expected a constant integer.");
730 const auto *NumArgsVal =
731 cast<ConstantInt>(I->getOperand(PatchPointOpers::NArgPos));
732 unsigned NumArgs = NumArgsVal->getZExtValue();
733
734 // Skip the four meta args: <id>, <numNopBytes>, <target>, <numArgs>
735 // This includes all meta-operands up to but not including CC.
736 unsigned NumMetaOpers = PatchPointOpers::CCPos;
737 assert(I->getNumArgOperands() >= NumMetaOpers + NumArgs &&
738 "Not enough arguments provided to the patchpoint intrinsic");
739
740 // For AnyRegCC the arguments are lowered later on manually.
741 unsigned NumCallArgs = IsAnyRegCC ? 0 : NumArgs;
742 CallLoweringInfo CLI;
743 if (!lowerCallOperands(I, NumMetaOpers, NumCallArgs, Callee, IsAnyRegCC, CLI))
744 return false;
745
746 assert(CLI.Call && "No call instruction specified.");
747
748 SmallVector<MachineOperand, 32> Ops;
749
750 // Add an explicit result reg if we use the anyreg calling convention.
Juergen Ributzka3d9e6752014-07-11 22:19:02 +0000751 if (IsAnyRegCC && HasDef) {
Juergen Ributzkaa4159432014-07-15 02:22:43 +0000752 assert(CLI.NumResultRegs == 0 && "Unexpected result register.");
753 CLI.ResultReg = createResultReg(TLI.getRegClassFor(MVT::i64));
754 CLI.NumResultRegs = 1;
755 Ops.push_back(MachineOperand::CreateReg(CLI.ResultReg, /*IsDef=*/true));
Juergen Ributzka3d9e6752014-07-11 22:19:02 +0000756 }
757
758 // Add the <id> and <numBytes> constants.
759 assert(isa<ConstantInt>(I->getOperand(PatchPointOpers::IDPos)) &&
760 "Expected a constant integer.");
761 const auto *ID = cast<ConstantInt>(I->getOperand(PatchPointOpers::IDPos));
762 Ops.push_back(MachineOperand::CreateImm(ID->getZExtValue()));
763
764 assert(isa<ConstantInt>(I->getOperand(PatchPointOpers::NBytesPos)) &&
765 "Expected a constant integer.");
766 const auto *NumBytes =
767 cast<ConstantInt>(I->getOperand(PatchPointOpers::NBytesPos));
768 Ops.push_back(MachineOperand::CreateImm(NumBytes->getZExtValue()));
769
770 // Assume that the callee is a constant address or null pointer.
771 // FIXME: handle function symbols in the future.
Juergen Ributzkae8514fc2014-07-31 00:11:16 +0000772 uint64_t CalleeAddr;
Juergen Ributzka3d9e6752014-07-11 22:19:02 +0000773 if (const auto *C = dyn_cast<IntToPtrInst>(Callee))
774 CalleeAddr = cast<ConstantInt>(C->getOperand(0))->getZExtValue();
775 else if (const auto *C = dyn_cast<ConstantExpr>(Callee)) {
776 if (C->getOpcode() == Instruction::IntToPtr)
777 CalleeAddr = cast<ConstantInt>(C->getOperand(0))->getZExtValue();
778 else
779 llvm_unreachable("Unsupported ConstantExpr.");
780 } else if (isa<ConstantPointerNull>(Callee))
781 CalleeAddr = 0;
782 else
783 llvm_unreachable("Unsupported callee address.");
784
785 Ops.push_back(MachineOperand::CreateImm(CalleeAddr));
786
787 // Adjust <numArgs> to account for any arguments that have been passed on
788 // the stack instead.
789 unsigned NumCallRegArgs = IsAnyRegCC ? NumArgs : CLI.OutRegs.size();
790 Ops.push_back(MachineOperand::CreateImm(NumCallRegArgs));
791
792 // Add the calling convention
793 Ops.push_back(MachineOperand::CreateImm((unsigned)CC));
794
795 // Add the arguments we omitted previously. The register allocator should
796 // place these in any free register.
797 if (IsAnyRegCC) {
798 for (unsigned i = NumMetaOpers, e = NumMetaOpers + NumArgs; i != e; ++i) {
799 unsigned Reg = getRegForValue(I->getArgOperand(i));
800 if (!Reg)
801 return false;
802 Ops.push_back(MachineOperand::CreateReg(Reg, /*IsDef=*/false));
803 }
804 }
805
806 // Push the arguments from the call instruction.
807 for (auto Reg : CLI.OutRegs)
808 Ops.push_back(MachineOperand::CreateReg(Reg, /*IsDef=*/false));
809
810 // Push live variables for the stack map.
811 if (!addStackMapLiveVars(Ops, I, NumMetaOpers + NumArgs))
812 return false;
813
814 // Push the register mask info.
815 Ops.push_back(MachineOperand::CreateRegMask(TRI.getCallPreservedMask(CC)));
816
817 // Add scratch registers as implicit def and early clobber.
818 const MCPhysReg *ScratchRegs = TLI.getScratchRegisters(CC);
819 for (unsigned i = 0; ScratchRegs[i]; ++i)
820 Ops.push_back(MachineOperand::CreateReg(
821 ScratchRegs[i], /*IsDef=*/true, /*IsImp=*/true, /*IsKill=*/false,
822 /*IsDead=*/false, /*IsUndef=*/false, /*IsEarlyClobber=*/true));
823
824 // Add implicit defs (return values).
825 for (auto Reg : CLI.InRegs)
826 Ops.push_back(MachineOperand::CreateReg(Reg, /*IsDef=*/true,
827 /*IsImpl=*/true));
828
Juergen Ributzka718bb712014-07-15 02:22:46 +0000829 // Insert the patchpoint instruction before the call generated by the target.
830 MachineInstrBuilder MIB = BuildMI(*FuncInfo.MBB, CLI.Call, DbgLoc,
Juergen Ributzka3d9e6752014-07-11 22:19:02 +0000831 TII.get(TargetOpcode::PATCHPOINT));
832
833 for (auto &MO : Ops)
834 MIB.addOperand(MO);
835
836 MIB->setPhysRegsDeadExcept(CLI.InRegs, TRI);
837
838 // Delete the original call instruction.
839 CLI.Call->eraseFromParent();
840
841 // Inform the Frame Information that we have a patchpoint in this function.
842 FuncInfo.MF->getFrameInfo()->setHasPatchPoint();
843
Juergen Ributzkaa4159432014-07-15 02:22:43 +0000844 if (CLI.NumResultRegs)
845 UpdateValueMap(I, CLI.ResultReg, CLI.NumResultRegs);
Juergen Ributzka3d9e6752014-07-11 22:19:02 +0000846 return true;
847}
848
Juergen Ributzka8179e9e2014-07-11 22:01:42 +0000849/// Returns an AttributeSet representing the attributes applied to the return
850/// value of the given call.
851static AttributeSet getReturnAttrs(FastISel::CallLoweringInfo &CLI) {
852 SmallVector<Attribute::AttrKind, 2> Attrs;
853 if (CLI.RetSExt)
854 Attrs.push_back(Attribute::SExt);
855 if (CLI.RetZExt)
856 Attrs.push_back(Attribute::ZExt);
857 if (CLI.IsInReg)
858 Attrs.push_back(Attribute::InReg);
859
860 return AttributeSet::get(CLI.RetTy->getContext(), AttributeSet::ReturnIndex,
861 Attrs);
862}
863
864bool FastISel::LowerCallTo(const CallInst *CI, const char *SymName,
865 unsigned NumArgs) {
866 ImmutableCallSite CS(CI);
867
868 PointerType *PT = cast<PointerType>(CS.getCalledValue()->getType());
869 FunctionType *FTy = cast<FunctionType>(PT->getElementType());
870 Type *RetTy = FTy->getReturnType();
871
872 ArgListTy Args;
873 Args.reserve(NumArgs);
874
875 // Populate the argument list.
876 // Attributes for args start at offset 1, after the return attribute.
877 for (unsigned ArgI = 0; ArgI != NumArgs; ++ArgI) {
878 Value *V = CI->getOperand(ArgI);
879
880 assert(!V->getType()->isEmptyTy() && "Empty type passed to intrinsic.");
881
882 ArgListEntry Entry;
883 Entry.Val = V;
884 Entry.Ty = V->getType();
885 Entry.setAttributes(&CS, ArgI + 1);
886 Args.push_back(Entry);
887 }
888
889 CallLoweringInfo CLI;
890 CLI.setCallee(RetTy, FTy, SymName, std::move(Args), CS, NumArgs);
891
892 return LowerCallTo(CLI);
893}
894
895bool FastISel::LowerCallTo(CallLoweringInfo &CLI) {
896 // Handle the incoming return values from the call.
897 CLI.clearIns();
898 SmallVector<EVT, 4> RetTys;
899 ComputeValueVTs(TLI, CLI.RetTy, RetTys);
900
901 SmallVector<ISD::OutputArg, 4> Outs;
902 GetReturnInfo(CLI.RetTy, getReturnAttrs(CLI), Outs, TLI);
903
904 bool CanLowerReturn = TLI.CanLowerReturn(CLI.CallConv, *FuncInfo.MF,
905 CLI.IsVarArg, Outs,
906 CLI.RetTy->getContext());
907
908 // FIXME: sret demotion isn't supported yet - bail out.
909 if (!CanLowerReturn)
910 return false;
911
912 for (unsigned I = 0, E = RetTys.size(); I != E; ++I) {
913 EVT VT = RetTys[I];
914 MVT RegisterVT = TLI.getRegisterType(CLI.RetTy->getContext(), VT);
915 unsigned NumRegs = TLI.getNumRegisters(CLI.RetTy->getContext(), VT);
916 for (unsigned i = 0; i != NumRegs; ++i) {
917 ISD::InputArg MyFlags;
918 MyFlags.VT = RegisterVT;
919 MyFlags.ArgVT = VT;
920 MyFlags.Used = CLI.IsReturnValueUsed;
921 if (CLI.RetSExt)
922 MyFlags.Flags.setSExt();
923 if (CLI.RetZExt)
924 MyFlags.Flags.setZExt();
925 if (CLI.IsInReg)
926 MyFlags.Flags.setInReg();
927 CLI.Ins.push_back(MyFlags);
928 }
929 }
930
931 // Handle all of the outgoing arguments.
932 CLI.clearOuts();
933 for (auto &Arg : CLI.getArgs()) {
934 Type *FinalType = Arg.Ty;
935 if (Arg.isByVal)
936 FinalType = cast<PointerType>(Arg.Ty)->getElementType();
937 bool NeedsRegBlock = TLI.functionArgumentNeedsConsecutiveRegisters(
938 FinalType, CLI.CallConv, CLI.IsVarArg);
939
940 ISD::ArgFlagsTy Flags;
941 if (Arg.isZExt)
942 Flags.setZExt();
943 if (Arg.isSExt)
944 Flags.setSExt();
945 if (Arg.isInReg)
946 Flags.setInReg();
947 if (Arg.isSRet)
948 Flags.setSRet();
949 if (Arg.isByVal)
950 Flags.setByVal();
951 if (Arg.isInAlloca) {
952 Flags.setInAlloca();
953 // Set the byval flag for CCAssignFn callbacks that don't know about
954 // inalloca. This way we can know how many bytes we should've allocated
955 // and how many bytes a callee cleanup function will pop. If we port
956 // inalloca to more targets, we'll have to add custom inalloca handling in
957 // the various CC lowering callbacks.
958 Flags.setByVal();
959 }
960 if (Arg.isByVal || Arg.isInAlloca) {
961 PointerType *Ty = cast<PointerType>(Arg.Ty);
962 Type *ElementTy = Ty->getElementType();
963 unsigned FrameSize = DL.getTypeAllocSize(ElementTy);
964 // For ByVal, alignment should come from FE. BE will guess if this info is
965 // not there, but there are cases it cannot get right.
966 unsigned FrameAlign = Arg.Alignment;
967 if (!FrameAlign)
968 FrameAlign = TLI.getByValTypeAlignment(ElementTy);
969 Flags.setByValSize(FrameSize);
970 Flags.setByValAlign(FrameAlign);
971 }
972 if (Arg.isNest)
973 Flags.setNest();
974 if (NeedsRegBlock)
975 Flags.setInConsecutiveRegs();
976 unsigned OriginalAlignment = DL.getABITypeAlignment(Arg.Ty);
977 Flags.setOrigAlign(OriginalAlignment);
978
979 CLI.OutVals.push_back(Arg.Val);
980 CLI.OutFlags.push_back(Flags);
981 }
982
983 if (!FastLowerCall(CLI))
984 return false;
985
986 // Set all unused physreg defs as dead.
987 assert(CLI.Call && "No call instruction specified.");
988 CLI.Call->setPhysRegsDeadExcept(CLI.InRegs, TRI);
989
990 if (CLI.NumResultRegs && CLI.CS)
991 UpdateValueMap(CLI.CS->getInstruction(), CLI.ResultReg, CLI.NumResultRegs);
992
993 return true;
994}
995
996bool FastISel::LowerCall(const CallInst *CI) {
997 ImmutableCallSite CS(CI);
998
999 PointerType *PT = cast<PointerType>(CS.getCalledValue()->getType());
1000 FunctionType *FuncTy = cast<FunctionType>(PT->getElementType());
1001 Type *RetTy = FuncTy->getReturnType();
1002
1003 ArgListTy Args;
1004 ArgListEntry Entry;
1005 Args.reserve(CS.arg_size());
1006
1007 for (ImmutableCallSite::arg_iterator i = CS.arg_begin(), e = CS.arg_end();
1008 i != e; ++i) {
1009 Value *V = *i;
1010
1011 // Skip empty types
1012 if (V->getType()->isEmptyTy())
1013 continue;
1014
1015 Entry.Val = V;
1016 Entry.Ty = V->getType();
1017
1018 // Skip the first return-type Attribute to get to params.
1019 Entry.setAttributes(&CS, i - CS.arg_begin() + 1);
1020 Args.push_back(Entry);
1021 }
1022
1023 // Check if target-independent constraints permit a tail call here.
1024 // Target-dependent constraints are checked within FastLowerCall.
1025 bool IsTailCall = CI->isTailCall();
Juergen Ributzka480872b2014-07-16 00:01:22 +00001026 if (IsTailCall && !isInTailCallPosition(CS, TM))
Juergen Ributzka8179e9e2014-07-11 22:01:42 +00001027 IsTailCall = false;
1028
1029 CallLoweringInfo CLI;
1030 CLI.setCallee(RetTy, FuncTy, CI->getCalledValue(), std::move(Args), CS)
1031 .setTailCall(IsTailCall);
1032
1033 return LowerCallTo(CLI);
1034}
1035
Dan Gohmanbcaf6812010-04-15 01:51:59 +00001036bool FastISel::SelectCall(const User *I) {
Dan Gohman7da91ae2011-04-26 17:18:34 +00001037 const CallInst *Call = cast<CallInst>(I);
1038
1039 // Handle simple inline asms.
Dan Gohmande239d22011-10-12 15:56:56 +00001040 if (const InlineAsm *IA = dyn_cast<InlineAsm>(Call->getCalledValue())) {
Juergen Ributzka618ce3e2014-07-16 22:20:51 +00001041 // If the inline asm has side effects, then make sure that no local value
1042 // lives across by flushing the local value map.
1043 if (IA->hasSideEffects())
1044 flushLocalValueMap();
1045
Dan Gohman7da91ae2011-04-26 17:18:34 +00001046 // Don't attempt to handle constraints.
1047 if (!IA->getConstraintString().empty())
1048 return false;
1049
1050 unsigned ExtraInfo = 0;
1051 if (IA->hasSideEffects())
1052 ExtraInfo |= InlineAsm::Extra_HasSideEffects;
1053 if (IA->isAlignStack())
1054 ExtraInfo |= InlineAsm::Extra_IsAlignStack;
1055
Rafael Espindolaea09c592014-02-18 22:05:46 +00001056 BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc,
Dan Gohman7da91ae2011-04-26 17:18:34 +00001057 TII.get(TargetOpcode::INLINEASM))
1058 .addExternalSymbol(IA->getAsmString().c_str())
1059 .addImm(ExtraInfo);
1060 return true;
1061 }
1062
Michael J. Spencer8b98bf22012-02-22 19:06:13 +00001063 MachineModuleInfo &MMI = FuncInfo.MF->getMMI();
1064 ComputeUsesVAFloatArgument(*Call, &MMI);
1065
Juergen Ributzka5dd32132014-07-11 20:42:12 +00001066 // Handle intrinsic function calls.
1067 if (const auto *II = dyn_cast<IntrinsicInst>(Call))
1068 return SelectIntrinsicCall(II);
Dan Gohman32a733e2008-09-25 17:05:24 +00001069
Juergen Ributzka5dd32132014-07-11 20:42:12 +00001070 // Usually, it does not make sense to initialize a value,
1071 // make an unrelated function call and use the value, because
1072 // it tends to be spilled on the stack. So, we move the pointer
1073 // to the last local value to the beginning of the block, so that
1074 // all the values which have already been materialized,
1075 // appear after the call. It also makes sense to skip intrinsics
1076 // since they tend to be inlined.
1077 flushLocalValueMap();
1078
Juergen Ributzka8179e9e2014-07-11 22:01:42 +00001079 return LowerCall(Call);
Juergen Ributzka5dd32132014-07-11 20:42:12 +00001080}
1081
1082bool FastISel::SelectIntrinsicCall(const IntrinsicInst *II) {
1083 switch (II->getIntrinsicID()) {
Dan Gohman32a733e2008-09-25 17:05:24 +00001084 default: break;
Juergen Ributzka5dd32132014-07-11 20:42:12 +00001085 // At -O0 we don't care about the lifetime intrinsics.
Eric Christopher81e2bf22012-02-17 23:03:39 +00001086 case Intrinsic::lifetime_start:
1087 case Intrinsic::lifetime_end:
Juergen Ributzka5dd32132014-07-11 20:42:12 +00001088 // The donothing intrinsic does, well, nothing.
Chad Rosier88d53ea2012-07-06 17:33:39 +00001089 case Intrinsic::donothing:
Eric Christopher81e2bf22012-02-17 23:03:39 +00001090 return true;
Bill Wendling65c0fd42009-02-13 02:16:35 +00001091 case Intrinsic::dbg_declare: {
Juergen Ributzka5dd32132014-07-11 20:42:12 +00001092 const DbgDeclareInst *DI = cast<DbgDeclareInst>(II);
Manman Ren983a16c2013-06-28 05:43:10 +00001093 DIVariable DIVar(DI->getVariable());
Stephen Lincfe7f352013-07-08 00:37:03 +00001094 assert((!DIVar || DIVar.isVariable()) &&
Juergen Ributzka5dd32132014-07-11 20:42:12 +00001095 "Variable in DbgDeclareInst should be either null or a DIVariable.");
1096 if (!DIVar || !FuncInfo.MF->getMMI().hasDebugInfo()) {
Eric Christopher142820b2012-03-15 21:33:44 +00001097 DEBUG(dbgs() << "Dropping debug info for " << *DI << "\n");
Devang Patel87127712009-07-02 22:43:26 +00001098 return true;
Eric Christopher142820b2012-03-15 21:33:44 +00001099 }
Devang Patel87127712009-07-02 22:43:26 +00001100
Dan Gohmanbcaf6812010-04-15 01:51:59 +00001101 const Value *Address = DI->getAddress();
Eric Christopher3390a6e2012-03-15 21:33:47 +00001102 if (!Address || isa<UndefValue>(Address)) {
Eric Christopher142820b2012-03-15 21:33:44 +00001103 DEBUG(dbgs() << "Dropping debug info for " << *DI << "\n");
Dale Johannesendb2eb472010-02-06 02:26:02 +00001104 return true;
Eric Christopher142820b2012-03-15 21:33:44 +00001105 }
Devang Patele4682fa2010-09-14 20:29:31 +00001106
Adrian Prantl418d1d12013-07-09 20:28:37 +00001107 unsigned Offset = 0;
David Blaikie0252265b2013-06-16 20:34:15 +00001108 Optional<MachineOperand> Op;
1109 if (const Argument *Arg = dyn_cast<Argument>(Address))
Devang Patel9d904e12011-09-08 22:59:09 +00001110 // Some arguments' frame index is recorded during argument lowering.
Adrian Prantl418d1d12013-07-09 20:28:37 +00001111 Offset = FuncInfo.getArgumentFrameIndex(Arg);
1112 if (Offset)
Juergen Ributzka5dd32132014-07-11 20:42:12 +00001113 Op = MachineOperand::CreateFI(Offset);
David Blaikie0252265b2013-06-16 20:34:15 +00001114 if (!Op)
1115 if (unsigned Reg = lookUpRegForValue(Address))
1116 Op = MachineOperand::CreateReg(Reg, false);
Eric Christopher60e01c52012-03-20 01:07:58 +00001117
Bill Wendling9f829f12012-03-30 00:02:55 +00001118 // If we have a VLA that has a "use" in a metadata node that's then used
1119 // here but it has no other uses, then we have a problem. E.g.,
1120 //
1121 // int foo (const int *x) {
1122 // char a[*x];
1123 // return 0;
1124 // }
1125 //
1126 // If we assign 'a' a vreg and fast isel later on has to use the selection
1127 // DAG isel, it will want to copy the value to the vreg. However, there are
1128 // no uses, which goes counter to what selection DAG isel expects.
David Blaikie0252265b2013-06-16 20:34:15 +00001129 if (!Op && !Address->use_empty() && isa<Instruction>(Address) &&
Eric Christopher60e01c52012-03-20 01:07:58 +00001130 (!isa<AllocaInst>(Address) ||
1131 !FuncInfo.StaticAllocaMap.count(cast<AllocaInst>(Address))))
David Blaikie0252265b2013-06-16 20:34:15 +00001132 Op = MachineOperand::CreateReg(FuncInfo.InitializeRegForValue(Address),
Adrian Prantl262bcf42013-09-18 22:08:59 +00001133 false);
Wesley Peck527da1b2010-11-23 03:31:01 +00001134
Adrian Prantl262bcf42013-09-18 22:08:59 +00001135 if (Op) {
Adrian Prantl418d1d12013-07-09 20:28:37 +00001136 if (Op->isReg()) {
1137 Op->setIsDebug(true);
Rafael Espindolaea09c592014-02-18 22:05:46 +00001138 BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc,
David Blaikie6004dbc2013-10-14 20:15:04 +00001139 TII.get(TargetOpcode::DBG_VALUE), false, Op->getReg(), 0,
1140 DI->getVariable());
1141 } else
Rafael Espindolaea09c592014-02-18 22:05:46 +00001142 BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc,
David Blaikie6004dbc2013-10-14 20:15:04 +00001143 TII.get(TargetOpcode::DBG_VALUE))
Juergen Ributzka5dd32132014-07-11 20:42:12 +00001144 .addOperand(*Op)
1145 .addImm(0)
1146 .addMetadata(DI->getVariable());
Adrian Prantl262bcf42013-09-18 22:08:59 +00001147 } else {
Eric Christophere5e54c82012-03-20 01:07:53 +00001148 // We can't yet handle anything else here because it would require
1149 // generating code, thus altering codegen because of debug info.
Adrian Prantl0d1e5592013-05-22 18:02:19 +00001150 DEBUG(dbgs() << "Dropping debug info for " << *DI << "\n");
Adrian Prantl262bcf42013-09-18 22:08:59 +00001151 }
Dan Gohman32a733e2008-09-25 17:05:24 +00001152 return true;
Bill Wendling65c0fd42009-02-13 02:16:35 +00001153 }
Dale Johannesendd331042010-02-26 20:01:55 +00001154 case Intrinsic::dbg_value: {
Dale Johannesen5d7f0a02010-04-07 01:15:14 +00001155 // This form of DBG_VALUE is target-independent.
Juergen Ributzka5dd32132014-07-11 20:42:12 +00001156 const DbgValueInst *DI = cast<DbgValueInst>(II);
Evan Cheng6cc775f2011-06-28 19:10:37 +00001157 const MCInstrDesc &II = TII.get(TargetOpcode::DBG_VALUE);
Dan Gohmanbcaf6812010-04-15 01:51:59 +00001158 const Value *V = DI->getValue();
Dale Johannesendd331042010-02-26 20:01:55 +00001159 if (!V) {
1160 // Currently the optimizer can produce this; insert an undef to
1161 // help debugging. Probably the optimizer should not do this.
Rafael Espindolaea09c592014-02-18 22:05:46 +00001162 BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc, II)
Dan Gohmand7b5ce32010-07-10 09:00:22 +00001163 .addReg(0U).addImm(DI->getOffset())
1164 .addMetadata(DI->getVariable());
Dan Gohmanbcaf6812010-04-15 01:51:59 +00001165 } else if (const ConstantInt *CI = dyn_cast<ConstantInt>(V)) {
Devang Patelf071d722011-06-24 20:46:11 +00001166 if (CI->getBitWidth() > 64)
Rafael Espindolaea09c592014-02-18 22:05:46 +00001167 BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc, II)
Devang Patelf071d722011-06-24 20:46:11 +00001168 .addCImm(CI).addImm(DI->getOffset())
1169 .addMetadata(DI->getVariable());
Chad Rosier879c34f2012-07-06 17:44:22 +00001170 else
Rafael Espindolaea09c592014-02-18 22:05:46 +00001171 BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc, II)
Devang Patelf071d722011-06-24 20:46:11 +00001172 .addImm(CI->getZExtValue()).addImm(DI->getOffset())
1173 .addMetadata(DI->getVariable());
Dan Gohmanbcaf6812010-04-15 01:51:59 +00001174 } else if (const ConstantFP *CF = dyn_cast<ConstantFP>(V)) {
Rafael Espindolaea09c592014-02-18 22:05:46 +00001175 BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc, II)
Dan Gohmand7b5ce32010-07-10 09:00:22 +00001176 .addFPImm(CF).addImm(DI->getOffset())
1177 .addMetadata(DI->getVariable());
Dale Johannesendd331042010-02-26 20:01:55 +00001178 } else if (unsigned Reg = lookUpRegForValue(V)) {
Adrian Prantldb3e26d2013-09-16 23:29:03 +00001179 // FIXME: This does not handle register-indirect values at offset 0.
Adrian Prantl418d1d12013-07-09 20:28:37 +00001180 bool IsIndirect = DI->getOffset() != 0;
Rafael Espindolaea09c592014-02-18 22:05:46 +00001181 BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc, II, IsIndirect,
Adrian Prantl418d1d12013-07-09 20:28:37 +00001182 Reg, DI->getOffset(), DI->getVariable());
Dale Johannesendd331042010-02-26 20:01:55 +00001183 } else {
1184 // We can't yet handle anything else here because it would require
1185 // generating code, thus altering codegen because of debug info.
Adrian Prantl0d1e5592013-05-22 18:02:19 +00001186 DEBUG(dbgs() << "Dropping debug info for " << *DI << "\n");
Wesley Peck527da1b2010-11-23 03:31:01 +00001187 }
Dale Johannesendd331042010-02-26 20:01:55 +00001188 return true;
1189 }
Eli Friedman8f1e11c2011-05-14 00:47:51 +00001190 case Intrinsic::objectsize: {
Juergen Ributzka5dd32132014-07-11 20:42:12 +00001191 ConstantInt *CI = cast<ConstantInt>(II->getArgOperand(1));
Eli Friedman8f1e11c2011-05-14 00:47:51 +00001192 unsigned long long Res = CI->isZero() ? -1ULL : 0;
Juergen Ributzka5dd32132014-07-11 20:42:12 +00001193 Constant *ResCI = ConstantInt::get(II->getType(), Res);
Eli Friedman8f1e11c2011-05-14 00:47:51 +00001194 unsigned ResultReg = getRegForValue(ResCI);
1195 if (ResultReg == 0)
1196 return false;
Juergen Ributzka5dd32132014-07-11 20:42:12 +00001197 UpdateValueMap(II, ResultReg);
Eli Friedman8f1e11c2011-05-14 00:47:51 +00001198 return true;
1199 }
Chad Rosier9c1796f2013-03-07 20:42:17 +00001200 case Intrinsic::expect: {
Juergen Ributzka5dd32132014-07-11 20:42:12 +00001201 unsigned ResultReg = getRegForValue(II->getArgOperand(0));
Nick Lewycky48beb212013-03-11 21:44:37 +00001202 if (ResultReg == 0)
1203 return false;
Juergen Ributzka5dd32132014-07-11 20:42:12 +00001204 UpdateValueMap(II, ResultReg);
Chad Rosier3a200e12013-03-07 21:38:33 +00001205 return true;
Chad Rosier9c1796f2013-03-07 20:42:17 +00001206 }
Juergen Ributzka190305b2014-07-01 22:25:49 +00001207 case Intrinsic::experimental_stackmap:
Juergen Ributzka5dd32132014-07-11 20:42:12 +00001208 return SelectStackmap(II);
Juergen Ributzka3d9e6752014-07-11 22:19:02 +00001209 case Intrinsic::experimental_patchpoint_void:
1210 case Intrinsic::experimental_patchpoint_i64:
1211 return SelectPatchpoint(II);
Dan Gohman32a733e2008-09-25 17:05:24 +00001212 }
Dan Gohman8a2dae52010-04-13 17:07:06 +00001213
Juergen Ributzka5dd32132014-07-11 20:42:12 +00001214 return FastLowerIntrinsicCall(II);
Dan Gohman32a733e2008-09-25 17:05:24 +00001215}
1216
Dan Gohmanbcaf6812010-04-15 01:51:59 +00001217bool FastISel::SelectCast(const User *I, unsigned Opcode) {
Owen Anderson53aa7a92009-08-10 22:56:29 +00001218 EVT SrcVT = TLI.getValueType(I->getOperand(0)->getType());
1219 EVT DstVT = TLI.getValueType(I->getType());
Wesley Peck527da1b2010-11-23 03:31:01 +00001220
Owen Anderson9f944592009-08-11 20:47:22 +00001221 if (SrcVT == MVT::Other || !SrcVT.isSimple() ||
1222 DstVT == MVT::Other || !DstVT.isSimple())
Owen Andersonca1711a2008-08-26 23:46:32 +00001223 // Unhandled type. Halt "fast" selection and bail.
1224 return false;
Wesley Peck527da1b2010-11-23 03:31:01 +00001225
Eli Friedmanc7035512011-05-25 23:49:02 +00001226 // Check if the destination type is legal.
Dan Gohmana62e4ab2009-03-13 23:53:06 +00001227 if (!TLI.isTypeLegal(DstVT))
Eli Friedmanc7035512011-05-25 23:49:02 +00001228 return false;
Dan Gohmana62e4ab2009-03-13 23:53:06 +00001229
Eli Friedmanc7035512011-05-25 23:49:02 +00001230 // Check if the source operand is legal.
Dan Gohmana62e4ab2009-03-13 23:53:06 +00001231 if (!TLI.isTypeLegal(SrcVT))
Eli Friedmanc7035512011-05-25 23:49:02 +00001232 return false;
Dan Gohmana62e4ab2009-03-13 23:53:06 +00001233
Dan Gohman7bda51f2008-09-03 23:12:08 +00001234 unsigned InputReg = getRegForValue(I->getOperand(0));
Owen Andersonca1711a2008-08-26 23:46:32 +00001235 if (!InputReg)
1236 // Unhandled operand. Halt "fast" selection and bail.
1237 return false;
Dan Gohmanc0bb9592009-03-13 20:42:20 +00001238
Dan Gohman1a1b51f2010-05-11 23:54:07 +00001239 bool InputRegIsKill = hasTrivialKill(I->getOperand(0));
1240
Owen Andersonca1711a2008-08-26 23:46:32 +00001241 unsigned ResultReg = FastEmit_r(SrcVT.getSimpleVT(),
1242 DstVT.getSimpleVT(),
1243 Opcode,
Dan Gohman1a1b51f2010-05-11 23:54:07 +00001244 InputReg, InputRegIsKill);
Owen Andersonca1711a2008-08-26 23:46:32 +00001245 if (!ResultReg)
1246 return false;
Wesley Peck527da1b2010-11-23 03:31:01 +00001247
Dan Gohman7bda51f2008-09-03 23:12:08 +00001248 UpdateValueMap(I, ResultReg);
Owen Andersonca1711a2008-08-26 23:46:32 +00001249 return true;
1250}
1251
Dan Gohmanbcaf6812010-04-15 01:51:59 +00001252bool FastISel::SelectBitCast(const User *I) {
Dan Gohmanb0b5a272008-08-27 18:10:19 +00001253 // If the bitcast doesn't change the type, just use the operand value.
1254 if (I->getType() == I->getOperand(0)->getType()) {
Dan Gohman7bda51f2008-09-03 23:12:08 +00001255 unsigned Reg = getRegForValue(I->getOperand(0));
Dan Gohman61cfa302008-08-27 20:41:38 +00001256 if (Reg == 0)
1257 return false;
Dan Gohman7bda51f2008-09-03 23:12:08 +00001258 UpdateValueMap(I, Reg);
Dan Gohmanb0b5a272008-08-27 18:10:19 +00001259 return true;
1260 }
1261
Wesley Peck527da1b2010-11-23 03:31:01 +00001262 // Bitcasts of other values become reg-reg copies or BITCAST operators.
Patrik Hagglundc494d242012-12-17 14:30:06 +00001263 EVT SrcEVT = TLI.getValueType(I->getOperand(0)->getType());
1264 EVT DstEVT = TLI.getValueType(I->getType());
1265 if (SrcEVT == MVT::Other || DstEVT == MVT::Other ||
1266 !TLI.isTypeLegal(SrcEVT) || !TLI.isTypeLegal(DstEVT))
Owen Andersonca1711a2008-08-26 23:46:32 +00001267 // Unhandled type. Halt "fast" selection and bail.
1268 return false;
Wesley Peck527da1b2010-11-23 03:31:01 +00001269
Patrik Hagglundc494d242012-12-17 14:30:06 +00001270 MVT SrcVT = SrcEVT.getSimpleVT();
1271 MVT DstVT = DstEVT.getSimpleVT();
Dan Gohman7bda51f2008-09-03 23:12:08 +00001272 unsigned Op0 = getRegForValue(I->getOperand(0));
Dan Gohmanb0b5a272008-08-27 18:10:19 +00001273 if (Op0 == 0)
1274 // Unhandled operand. Halt "fast" selection and bail.
Owen Andersonca1711a2008-08-26 23:46:32 +00001275 return false;
Dan Gohman1a1b51f2010-05-11 23:54:07 +00001276
1277 bool Op0IsKill = hasTrivialKill(I->getOperand(0));
Wesley Peck527da1b2010-11-23 03:31:01 +00001278
Dan Gohmanb0b5a272008-08-27 18:10:19 +00001279 // First, try to perform the bitcast by inserting a reg-reg copy.
1280 unsigned ResultReg = 0;
Patrik Hagglund5e6c3612012-12-13 06:34:11 +00001281 if (SrcVT == DstVT) {
Craig Topper760b1342012-02-22 05:59:10 +00001282 const TargetRegisterClass* SrcClass = TLI.getRegClassFor(SrcVT);
1283 const TargetRegisterClass* DstClass = TLI.getRegClassFor(DstVT);
Jakob Stoklund Olesen51642ae2010-07-11 05:16:54 +00001284 // Don't attempt a cross-class copy. It will likely fail.
1285 if (SrcClass == DstClass) {
1286 ResultReg = createResultReg(DstClass);
Rafael Espindolaea09c592014-02-18 22:05:46 +00001287 BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc,
1288 TII.get(TargetOpcode::COPY), ResultReg).addReg(Op0);
Jakob Stoklund Olesen51642ae2010-07-11 05:16:54 +00001289 }
Dan Gohmanb0b5a272008-08-27 18:10:19 +00001290 }
Wesley Peck527da1b2010-11-23 03:31:01 +00001291
1292 // If the reg-reg copy failed, select a BITCAST opcode.
Dan Gohmanb0b5a272008-08-27 18:10:19 +00001293 if (!ResultReg)
Patrik Hagglund5e6c3612012-12-13 06:34:11 +00001294 ResultReg = FastEmit_r(SrcVT, DstVT, ISD::BITCAST, Op0, Op0IsKill);
Wesley Peck527da1b2010-11-23 03:31:01 +00001295
Dan Gohmanb0b5a272008-08-27 18:10:19 +00001296 if (!ResultReg)
Owen Andersonca1711a2008-08-26 23:46:32 +00001297 return false;
Wesley Peck527da1b2010-11-23 03:31:01 +00001298
Dan Gohman7bda51f2008-09-03 23:12:08 +00001299 UpdateValueMap(I, ResultReg);
Owen Andersonca1711a2008-08-26 23:46:32 +00001300 return true;
1301}
1302
Dan Gohman7bda51f2008-09-03 23:12:08 +00001303bool
Dan Gohmanbcaf6812010-04-15 01:51:59 +00001304FastISel::SelectInstruction(const Instruction *I) {
Dan Gohman6e9a8fc2010-04-23 15:29:50 +00001305 // Just before the terminator instruction, insert instructions to
1306 // feed PHI nodes in successor blocks.
1307 if (isa<TerminatorInst>(I))
1308 if (!HandlePHINodesInSuccessorBlocks(I->getParent()))
1309 return false;
1310
Rafael Espindolaea09c592014-02-18 22:05:46 +00001311 DbgLoc = I->getDebugLoc();
Dan Gohmane450d742010-04-20 00:48:35 +00001312
Chad Rosier46addb92011-11-29 19:40:47 +00001313 MachineBasicBlock::iterator SavedInsertPt = FuncInfo.InsertPt;
1314
Bob Wilson3e6fa462012-08-03 04:06:28 +00001315 if (const CallInst *Call = dyn_cast<CallInst>(I)) {
1316 const Function *F = Call->getCalledFunction();
1317 LibFunc::Func Func;
Akira Hatanaka3d90f992014-04-15 21:30:06 +00001318
1319 // As a special case, don't handle calls to builtin library functions that
1320 // may be translated directly to target instructions.
Bob Wilson3e6fa462012-08-03 04:06:28 +00001321 if (F && !F->hasLocalLinkage() && F->hasName() &&
1322 LibInfo->getLibFunc(F->getName(), Func) &&
Bob Wilson871701c2012-08-03 21:26:24 +00001323 LibInfo->hasOptimizedCodeGen(Func))
Bob Wilson3e6fa462012-08-03 04:06:28 +00001324 return false;
Akira Hatanaka3d90f992014-04-15 21:30:06 +00001325
1326 // Don't handle Intrinsic::trap if a trap funciton is specified.
1327 if (F && F->getIntrinsicID() == Intrinsic::trap &&
1328 !TM.Options.getTrapFunctionName().empty())
1329 return false;
Bob Wilson3e6fa462012-08-03 04:06:28 +00001330 }
1331
Dan Gohman18f94462009-12-05 01:27:58 +00001332 // First, try doing target-independent selection.
Michael Ilsemanba8446c2013-02-27 19:54:00 +00001333 if (SelectOperator(I, I->getOpcode())) {
Jan Wen Voung7857a642013-03-08 22:56:31 +00001334 ++NumFastIselSuccessIndependent;
Rafael Espindolaea09c592014-02-18 22:05:46 +00001335 DbgLoc = DebugLoc();
Dan Gohman18f94462009-12-05 01:27:58 +00001336 return true;
Dan Gohmane450d742010-04-20 00:48:35 +00001337 }
Chad Rosier879c34f2012-07-06 17:44:22 +00001338 // Remove dead code. However, ignore call instructions since we've flushed
Chad Rosier46addb92011-11-29 19:40:47 +00001339 // the local value map and recomputed the insert point.
1340 if (!isa<CallInst>(I)) {
1341 recomputeInsertPt();
1342 if (SavedInsertPt != FuncInfo.InsertPt)
1343 removeDeadCode(FuncInfo.InsertPt, SavedInsertPt);
1344 }
Dan Gohman18f94462009-12-05 01:27:58 +00001345
1346 // Next, try calling the target to attempt to handle the instruction.
Chad Rosier46addb92011-11-29 19:40:47 +00001347 SavedInsertPt = FuncInfo.InsertPt;
Dan Gohmane450d742010-04-20 00:48:35 +00001348 if (TargetSelectInstruction(I)) {
Jan Wen Voung7857a642013-03-08 22:56:31 +00001349 ++NumFastIselSuccessTarget;
Rafael Espindolaea09c592014-02-18 22:05:46 +00001350 DbgLoc = DebugLoc();
Dan Gohman18f94462009-12-05 01:27:58 +00001351 return true;
Dan Gohmane450d742010-04-20 00:48:35 +00001352 }
Chad Rosier46addb92011-11-29 19:40:47 +00001353 // Check for dead code and remove as necessary.
1354 recomputeInsertPt();
1355 if (SavedInsertPt != FuncInfo.InsertPt)
1356 removeDeadCode(FuncInfo.InsertPt, SavedInsertPt);
Dan Gohman18f94462009-12-05 01:27:58 +00001357
Rafael Espindolaea09c592014-02-18 22:05:46 +00001358 DbgLoc = DebugLoc();
Dan Gohman18f94462009-12-05 01:27:58 +00001359 return false;
Dan Gohmanfcf54562008-09-05 18:18:20 +00001360}
1361
Dan Gohman1ab1d312008-10-02 22:15:21 +00001362/// FastEmitBranch - Emit an unconditional branch to the given block,
1363/// unless it is the immediate (fall-through) successor, and update
1364/// the CFG.
1365void
Rafael Espindolaea09c592014-02-18 22:05:46 +00001366FastISel::FastEmitBranch(MachineBasicBlock *MSucc, DebugLoc DbgLoc) {
Evan Cheng615620c2013-02-11 01:27:15 +00001367 if (FuncInfo.MBB->getBasicBlock()->size() > 1 &&
1368 FuncInfo.MBB->isLayoutSuccessor(MSucc)) {
Eric Christophere9abba72012-04-10 18:18:10 +00001369 // For more accurate line information if this is the only instruction
1370 // in the block then emit it, otherwise we have the unconditional
1371 // fall-through case, which needs no instructions.
Dan Gohman1ab1d312008-10-02 22:15:21 +00001372 } else {
1373 // The unconditional branch case.
Craig Topperc0196b12014-04-14 00:51:57 +00001374 TII.InsertBranch(*FuncInfo.MBB, MSucc, nullptr,
Rafael Espindolaea09c592014-02-18 22:05:46 +00001375 SmallVector<MachineOperand, 0>(), DbgLoc);
Dan Gohman1ab1d312008-10-02 22:15:21 +00001376 }
Juergen Ributzka454d3742014-06-13 00:45:11 +00001377 uint32_t BranchWeight = 0;
1378 if (FuncInfo.BPI)
1379 BranchWeight = FuncInfo.BPI->getEdgeWeight(FuncInfo.MBB->getBasicBlock(),
1380 MSucc->getBasicBlock());
1381 FuncInfo.MBB->addSuccessor(MSucc, BranchWeight);
Dan Gohman1ab1d312008-10-02 22:15:21 +00001382}
1383
Dan Gohmanaa92dc12009-09-03 22:53:57 +00001384/// SelectFNeg - Emit an FNeg operation.
1385///
1386bool
Dan Gohmanbcaf6812010-04-15 01:51:59 +00001387FastISel::SelectFNeg(const User *I) {
Dan Gohmanaa92dc12009-09-03 22:53:57 +00001388 unsigned OpReg = getRegForValue(BinaryOperator::getFNegArgument(I));
1389 if (OpReg == 0) return false;
1390
Dan Gohman1a1b51f2010-05-11 23:54:07 +00001391 bool OpRegIsKill = hasTrivialKill(I);
1392
Dan Gohman9cbef322009-09-11 00:36:43 +00001393 // If the target has ISD::FNEG, use it.
1394 EVT VT = TLI.getValueType(I->getType());
1395 unsigned ResultReg = FastEmit_r(VT.getSimpleVT(), VT.getSimpleVT(),
Dan Gohman1a1b51f2010-05-11 23:54:07 +00001396 ISD::FNEG, OpReg, OpRegIsKill);
Dan Gohman9cbef322009-09-11 00:36:43 +00001397 if (ResultReg != 0) {
1398 UpdateValueMap(I, ResultReg);
1399 return true;
1400 }
1401
Dan Gohman89b090e2009-09-11 00:34:46 +00001402 // Bitcast the value to integer, twiddle the sign bit with xor,
1403 // and then bitcast it back to floating-point.
Dan Gohmanaa92dc12009-09-03 22:53:57 +00001404 if (VT.getSizeInBits() > 64) return false;
Dan Gohman89b090e2009-09-11 00:34:46 +00001405 EVT IntVT = EVT::getIntegerVT(I->getContext(), VT.getSizeInBits());
1406 if (!TLI.isTypeLegal(IntVT))
1407 return false;
1408
1409 unsigned IntReg = FastEmit_r(VT.getSimpleVT(), IntVT.getSimpleVT(),
Wesley Peck527da1b2010-11-23 03:31:01 +00001410 ISD::BITCAST, OpReg, OpRegIsKill);
Dan Gohman89b090e2009-09-11 00:34:46 +00001411 if (IntReg == 0)
1412 return false;
1413
Dan Gohman1a1b51f2010-05-11 23:54:07 +00001414 unsigned IntResultReg = FastEmit_ri_(IntVT.getSimpleVT(), ISD::XOR,
1415 IntReg, /*Kill=*/true,
Dan Gohman89b090e2009-09-11 00:34:46 +00001416 UINT64_C(1) << (VT.getSizeInBits()-1),
1417 IntVT.getSimpleVT());
1418 if (IntResultReg == 0)
1419 return false;
1420
1421 ResultReg = FastEmit_r(IntVT.getSimpleVT(), VT.getSimpleVT(),
Wesley Peck527da1b2010-11-23 03:31:01 +00001422 ISD::BITCAST, IntResultReg, /*Kill=*/true);
Dan Gohmanaa92dc12009-09-03 22:53:57 +00001423 if (ResultReg == 0)
1424 return false;
1425
1426 UpdateValueMap(I, ResultReg);
1427 return true;
1428}
1429
Dan Gohmanfcf54562008-09-05 18:18:20 +00001430bool
Eli Friedman9ac94472011-05-16 20:27:46 +00001431FastISel::SelectExtractValue(const User *U) {
1432 const ExtractValueInst *EVI = dyn_cast<ExtractValueInst>(U);
Eli Friedman4c08bb42011-05-16 20:34:53 +00001433 if (!EVI)
Eli Friedman9ac94472011-05-16 20:27:46 +00001434 return false;
1435
Eli Friedmana4d4a012011-05-16 21:06:17 +00001436 // Make sure we only try to handle extracts with a legal result. But also
1437 // allow i1 because it's easy.
Eli Friedman9ac94472011-05-16 20:27:46 +00001438 EVT RealVT = TLI.getValueType(EVI->getType(), /*AllowUnknown=*/true);
1439 if (!RealVT.isSimple())
1440 return false;
1441 MVT VT = RealVT.getSimpleVT();
Eli Friedmana4d4a012011-05-16 21:06:17 +00001442 if (!TLI.isTypeLegal(VT) && VT != MVT::i1)
Eli Friedman9ac94472011-05-16 20:27:46 +00001443 return false;
1444
1445 const Value *Op0 = EVI->getOperand(0);
Chris Lattner229907c2011-07-18 04:54:35 +00001446 Type *AggTy = Op0->getType();
Eli Friedman9ac94472011-05-16 20:27:46 +00001447
1448 // Get the base result register.
1449 unsigned ResultReg;
1450 DenseMap<const Value *, unsigned>::iterator I = FuncInfo.ValueMap.find(Op0);
1451 if (I != FuncInfo.ValueMap.end())
1452 ResultReg = I->second;
Eli Friedmanbd375f12011-06-06 05:46:34 +00001453 else if (isa<Instruction>(Op0))
Eli Friedman9ac94472011-05-16 20:27:46 +00001454 ResultReg = FuncInfo.InitializeRegForValue(Op0);
Eli Friedmanbd375f12011-06-06 05:46:34 +00001455 else
1456 return false; // fast-isel can't handle aggregate constants at the moment
Eli Friedman9ac94472011-05-16 20:27:46 +00001457
1458 // Get the actual result register, which is an offset from the base register.
Jay Foad57aa6362011-07-13 10:26:04 +00001459 unsigned VTIndex = ComputeLinearIndex(AggTy, EVI->getIndices());
Eli Friedman9ac94472011-05-16 20:27:46 +00001460
1461 SmallVector<EVT, 4> AggValueVTs;
1462 ComputeValueVTs(TLI, AggTy, AggValueVTs);
1463
1464 for (unsigned i = 0; i < VTIndex; i++)
1465 ResultReg += TLI.getNumRegisters(FuncInfo.Fn->getContext(), AggValueVTs[i]);
1466
1467 UpdateValueMap(EVI, ResultReg);
1468 return true;
1469}
1470
1471bool
Dan Gohmanbcaf6812010-04-15 01:51:59 +00001472FastISel::SelectOperator(const User *I, unsigned Opcode) {
Dan Gohmanfcf54562008-09-05 18:18:20 +00001473 switch (Opcode) {
Dan Gohmana5b96452009-06-04 22:49:04 +00001474 case Instruction::Add:
1475 return SelectBinaryOp(I, ISD::ADD);
1476 case Instruction::FAdd:
1477 return SelectBinaryOp(I, ISD::FADD);
1478 case Instruction::Sub:
1479 return SelectBinaryOp(I, ISD::SUB);
1480 case Instruction::FSub:
Dan Gohmanaa92dc12009-09-03 22:53:57 +00001481 // FNeg is currently represented in LLVM IR as a special case of FSub.
1482 if (BinaryOperator::isFNeg(I))
1483 return SelectFNeg(I);
Dan Gohmana5b96452009-06-04 22:49:04 +00001484 return SelectBinaryOp(I, ISD::FSUB);
1485 case Instruction::Mul:
1486 return SelectBinaryOp(I, ISD::MUL);
1487 case Instruction::FMul:
1488 return SelectBinaryOp(I, ISD::FMUL);
Dan Gohman7bda51f2008-09-03 23:12:08 +00001489 case Instruction::SDiv:
1490 return SelectBinaryOp(I, ISD::SDIV);
1491 case Instruction::UDiv:
1492 return SelectBinaryOp(I, ISD::UDIV);
1493 case Instruction::FDiv:
1494 return SelectBinaryOp(I, ISD::FDIV);
1495 case Instruction::SRem:
1496 return SelectBinaryOp(I, ISD::SREM);
1497 case Instruction::URem:
1498 return SelectBinaryOp(I, ISD::UREM);
1499 case Instruction::FRem:
1500 return SelectBinaryOp(I, ISD::FREM);
1501 case Instruction::Shl:
1502 return SelectBinaryOp(I, ISD::SHL);
1503 case Instruction::LShr:
1504 return SelectBinaryOp(I, ISD::SRL);
1505 case Instruction::AShr:
1506 return SelectBinaryOp(I, ISD::SRA);
1507 case Instruction::And:
1508 return SelectBinaryOp(I, ISD::AND);
1509 case Instruction::Or:
1510 return SelectBinaryOp(I, ISD::OR);
1511 case Instruction::Xor:
1512 return SelectBinaryOp(I, ISD::XOR);
Dan Gohmanb2226e22008-08-13 20:19:35 +00001513
Dan Gohman7bda51f2008-09-03 23:12:08 +00001514 case Instruction::GetElementPtr:
1515 return SelectGetElementPtr(I);
Dan Gohmana3e4d5a2008-08-20 00:11:48 +00001516
Dan Gohman7bda51f2008-09-03 23:12:08 +00001517 case Instruction::Br: {
Dan Gohmanbcaf6812010-04-15 01:51:59 +00001518 const BranchInst *BI = cast<BranchInst>(I);
Dan Gohmana3e4d5a2008-08-20 00:11:48 +00001519
Dan Gohman7bda51f2008-09-03 23:12:08 +00001520 if (BI->isUnconditional()) {
Dan Gohmanbcaf6812010-04-15 01:51:59 +00001521 const BasicBlock *LLVMSucc = BI->getSuccessor(0);
Dan Gohman87fb4e82010-07-07 16:29:44 +00001522 MachineBasicBlock *MSucc = FuncInfo.MBBMap[LLVMSucc];
Stuart Hastings0125b642010-06-17 22:43:56 +00001523 FastEmitBranch(MSucc, BI->getDebugLoc());
Dan Gohman7bda51f2008-09-03 23:12:08 +00001524 return true;
Owen Anderson14054922008-08-27 00:31:01 +00001525 }
Dan Gohman7bda51f2008-09-03 23:12:08 +00001526
1527 // Conditional branches are not handed yet.
1528 // Halt "fast" selection and bail.
1529 return false;
Dan Gohmanb2226e22008-08-13 20:19:35 +00001530 }
1531
Dan Gohmanea56bdd2008-09-05 01:08:41 +00001532 case Instruction::Unreachable:
Yaron Kerend7ba46b2014-04-19 13:47:43 +00001533 if (TM.Options.TrapUnreachable)
1534 return FastEmit_(MVT::Other, MVT::Other, ISD::TRAP) != 0;
1535 else
1536 return true;
Dan Gohmanea56bdd2008-09-05 01:08:41 +00001537
Dan Gohman39d82f92008-09-10 20:11:02 +00001538 case Instruction::Alloca:
1539 // FunctionLowering has the static-sized case covered.
Dan Gohman87fb4e82010-07-07 16:29:44 +00001540 if (FuncInfo.StaticAllocaMap.count(cast<AllocaInst>(I)))
Dan Gohman39d82f92008-09-10 20:11:02 +00001541 return true;
1542
1543 // Dynamic-sized alloca is not handled yet.
1544 return false;
Wesley Peck527da1b2010-11-23 03:31:01 +00001545
Dan Gohman32a733e2008-09-25 17:05:24 +00001546 case Instruction::Call:
1547 return SelectCall(I);
Wesley Peck527da1b2010-11-23 03:31:01 +00001548
Dan Gohman7bda51f2008-09-03 23:12:08 +00001549 case Instruction::BitCast:
1550 return SelectBitCast(I);
1551
1552 case Instruction::FPToSI:
1553 return SelectCast(I, ISD::FP_TO_SINT);
1554 case Instruction::ZExt:
1555 return SelectCast(I, ISD::ZERO_EXTEND);
1556 case Instruction::SExt:
1557 return SelectCast(I, ISD::SIGN_EXTEND);
1558 case Instruction::Trunc:
1559 return SelectCast(I, ISD::TRUNCATE);
1560 case Instruction::SIToFP:
1561 return SelectCast(I, ISD::SINT_TO_FP);
1562
1563 case Instruction::IntToPtr: // Deliberate fall-through.
1564 case Instruction::PtrToInt: {
Owen Anderson53aa7a92009-08-10 22:56:29 +00001565 EVT SrcVT = TLI.getValueType(I->getOperand(0)->getType());
1566 EVT DstVT = TLI.getValueType(I->getType());
Dan Gohman7bda51f2008-09-03 23:12:08 +00001567 if (DstVT.bitsGT(SrcVT))
1568 return SelectCast(I, ISD::ZERO_EXTEND);
1569 if (DstVT.bitsLT(SrcVT))
1570 return SelectCast(I, ISD::TRUNCATE);
1571 unsigned Reg = getRegForValue(I->getOperand(0));
1572 if (Reg == 0) return false;
1573 UpdateValueMap(I, Reg);
1574 return true;
1575 }
Dan Gohman918fe082008-09-23 21:53:34 +00001576
Eli Friedman9ac94472011-05-16 20:27:46 +00001577 case Instruction::ExtractValue:
1578 return SelectExtractValue(I);
1579
Dan Gohmanf41ad472010-04-20 15:00:41 +00001580 case Instruction::PHI:
1581 llvm_unreachable("FastISel shouldn't visit PHI nodes!");
1582
Dan Gohman7bda51f2008-09-03 23:12:08 +00001583 default:
1584 // Unhandled instruction. Halt "fast" selection and bail.
1585 return false;
1586 }
Dan Gohmanb2226e22008-08-13 20:19:35 +00001587}
1588
Bob Wilson3e6fa462012-08-03 04:06:28 +00001589FastISel::FastISel(FunctionLoweringInfo &funcInfo,
1590 const TargetLibraryInfo *libInfo)
Dan Gohmand7b5ce32010-07-10 09:00:22 +00001591 : FuncInfo(funcInfo),
Eric Christopherc1058df2014-07-04 01:55:26 +00001592 MF(funcInfo.MF),
Dan Gohman87fb4e82010-07-07 16:29:44 +00001593 MRI(FuncInfo.MF->getRegInfo()),
1594 MFI(*FuncInfo.MF->getFrameInfo()),
1595 MCP(*FuncInfo.MF->getConstantPool()),
1596 TM(FuncInfo.MF->getTarget()),
Rafael Espindolaea09c592014-02-18 22:05:46 +00001597 DL(*TM.getDataLayout()),
Dan Gohman49e19e92008-08-22 00:20:26 +00001598 TII(*TM.getInstrInfo()),
Dan Gohmanffcb5902010-05-05 23:58:35 +00001599 TLI(*TM.getTargetLowering()),
Bob Wilson3e6fa462012-08-03 04:06:28 +00001600 TRI(*TM.getRegisterInfo()),
1601 LibInfo(libInfo) {
Dan Gohman02c84b82008-08-20 21:05:57 +00001602}
1603
Dan Gohmanc4442382008-08-14 21:51:29 +00001604FastISel::~FastISel() {}
1605
Evan Cheng615620c2013-02-11 01:27:15 +00001606bool FastISel::FastLowerArguments() {
1607 return false;
1608}
1609
Juergen Ributzka8179e9e2014-07-11 22:01:42 +00001610bool FastISel::FastLowerCall(CallLoweringInfo &/*CLI*/) {
1611 return false;
1612}
1613
Reid Klecknerfb951982014-07-12 00:06:46 +00001614bool FastISel::FastLowerIntrinsicCall(const IntrinsicInst * /*II*/) {
Juergen Ributzka5dd32132014-07-11 20:42:12 +00001615 return false;
1616}
1617
Owen Anderson9f944592009-08-11 20:47:22 +00001618unsigned FastISel::FastEmit_(MVT, MVT,
Dan Gohman404a9842010-01-05 22:26:32 +00001619 unsigned) {
Dan Gohmanb2226e22008-08-13 20:19:35 +00001620 return 0;
1621}
1622
Owen Anderson9f944592009-08-11 20:47:22 +00001623unsigned FastISel::FastEmit_r(MVT, MVT,
Dan Gohman1a1b51f2010-05-11 23:54:07 +00001624 unsigned,
1625 unsigned /*Op0*/, bool /*Op0IsKill*/) {
Dan Gohmanb2226e22008-08-13 20:19:35 +00001626 return 0;
1627}
1628
Wesley Peck527da1b2010-11-23 03:31:01 +00001629unsigned FastISel::FastEmit_rr(MVT, MVT,
Dan Gohman1a1b51f2010-05-11 23:54:07 +00001630 unsigned,
1631 unsigned /*Op0*/, bool /*Op0IsKill*/,
1632 unsigned /*Op1*/, bool /*Op1IsKill*/) {
Dan Gohmanb2226e22008-08-13 20:19:35 +00001633 return 0;
1634}
1635
Dan Gohman404a9842010-01-05 22:26:32 +00001636unsigned FastISel::FastEmit_i(MVT, MVT, unsigned, uint64_t /*Imm*/) {
Evan Cheng864fcc12008-08-20 22:45:34 +00001637 return 0;
1638}
1639
Owen Anderson9f944592009-08-11 20:47:22 +00001640unsigned FastISel::FastEmit_f(MVT, MVT,
Dan Gohmanbcaf6812010-04-15 01:51:59 +00001641 unsigned, const ConstantFP * /*FPImm*/) {
Dan Gohman5ca269e2008-08-27 01:09:54 +00001642 return 0;
1643}
1644
Owen Anderson9f944592009-08-11 20:47:22 +00001645unsigned FastISel::FastEmit_ri(MVT, MVT,
Dan Gohman1a1b51f2010-05-11 23:54:07 +00001646 unsigned,
1647 unsigned /*Op0*/, bool /*Op0IsKill*/,
Owen Anderson8dd01cc2008-08-25 23:58:18 +00001648 uint64_t /*Imm*/) {
Dan Gohmanfe905652008-08-21 01:41:07 +00001649 return 0;
1650}
1651
Owen Anderson9f944592009-08-11 20:47:22 +00001652unsigned FastISel::FastEmit_rf(MVT, MVT,
Dan Gohman1a1b51f2010-05-11 23:54:07 +00001653 unsigned,
1654 unsigned /*Op0*/, bool /*Op0IsKill*/,
Dan Gohmanbcaf6812010-04-15 01:51:59 +00001655 const ConstantFP * /*FPImm*/) {
Dan Gohman5ca269e2008-08-27 01:09:54 +00001656 return 0;
1657}
1658
Owen Anderson9f944592009-08-11 20:47:22 +00001659unsigned FastISel::FastEmit_rri(MVT, MVT,
Dan Gohman404a9842010-01-05 22:26:32 +00001660 unsigned,
Dan Gohman1a1b51f2010-05-11 23:54:07 +00001661 unsigned /*Op0*/, bool /*Op0IsKill*/,
1662 unsigned /*Op1*/, bool /*Op1IsKill*/,
Dan Gohmanfe905652008-08-21 01:41:07 +00001663 uint64_t /*Imm*/) {
Evan Cheng864fcc12008-08-20 22:45:34 +00001664 return 0;
1665}
1666
1667/// FastEmit_ri_ - This method is a wrapper of FastEmit_ri. It first tries
1668/// to emit an instruction with an immediate operand using FastEmit_ri.
1669/// If that fails, it materializes the immediate into a register and try
1670/// FastEmit_rr instead.
Dan Gohman404a9842010-01-05 22:26:32 +00001671unsigned FastISel::FastEmit_ri_(MVT VT, unsigned Opcode,
Dan Gohman1a1b51f2010-05-11 23:54:07 +00001672 unsigned Op0, bool Op0IsKill,
1673 uint64_t Imm, MVT ImmType) {
Chris Lattnerb53ccb82011-04-17 20:23:29 +00001674 // If this is a multiply by a power of two, emit this as a shift left.
1675 if (Opcode == ISD::MUL && isPowerOf2_64(Imm)) {
1676 Opcode = ISD::SHL;
1677 Imm = Log2_64(Imm);
Chris Lattner562d6e82011-04-18 06:55:51 +00001678 } else if (Opcode == ISD::UDIV && isPowerOf2_64(Imm)) {
1679 // div x, 8 -> srl x, 3
1680 Opcode = ISD::SRL;
1681 Imm = Log2_64(Imm);
Chris Lattnerb53ccb82011-04-17 20:23:29 +00001682 }
Owen Andersondd450b82011-04-22 23:38:06 +00001683
Chris Lattnerb53ccb82011-04-17 20:23:29 +00001684 // Horrible hack (to be removed), check to make sure shift amounts are
1685 // in-range.
1686 if ((Opcode == ISD::SHL || Opcode == ISD::SRA || Opcode == ISD::SRL) &&
1687 Imm >= VT.getSizeInBits())
1688 return 0;
Owen Andersondd450b82011-04-22 23:38:06 +00001689
Evan Cheng864fcc12008-08-20 22:45:34 +00001690 // First check if immediate type is legal. If not, we can't use the ri form.
Dan Gohman1a1b51f2010-05-11 23:54:07 +00001691 unsigned ResultReg = FastEmit_ri(VT, VT, Opcode, Op0, Op0IsKill, Imm);
Evan Cheng864fcc12008-08-20 22:45:34 +00001692 if (ResultReg != 0)
1693 return ResultReg;
Owen Anderson8dd01cc2008-08-25 23:58:18 +00001694 unsigned MaterialReg = FastEmit_i(ImmType, ImmType, ISD::Constant, Imm);
Eli Friedman4105ed12011-04-29 23:34:52 +00001695 if (MaterialReg == 0) {
1696 // This is a bit ugly/slow, but failing here means falling out of
1697 // fast-isel, which would be very slow.
Chris Lattner229907c2011-07-18 04:54:35 +00001698 IntegerType *ITy = IntegerType::get(FuncInfo.Fn->getContext(),
Eli Friedman4105ed12011-04-29 23:34:52 +00001699 VT.getSizeInBits());
1700 MaterialReg = getRegForValue(ConstantInt::get(ITy, Imm));
Chad Rosierdbac0252013-03-28 23:04:47 +00001701 assert (MaterialReg != 0 && "Unable to materialize imm.");
1702 if (MaterialReg == 0) return 0;
Eli Friedman4105ed12011-04-29 23:34:52 +00001703 }
Dan Gohman1a1b51f2010-05-11 23:54:07 +00001704 return FastEmit_rr(VT, VT, Opcode,
1705 Op0, Op0IsKill,
1706 MaterialReg, /*Kill=*/true);
Dan Gohmanfe905652008-08-21 01:41:07 +00001707}
1708
1709unsigned FastISel::createResultReg(const TargetRegisterClass* RC) {
1710 return MRI.createVirtualRegister(RC);
Evan Cheng864fcc12008-08-20 22:45:34 +00001711}
1712
Tim Northover2f553f32014-04-15 13:59:49 +00001713unsigned FastISel::constrainOperandRegClass(const MCInstrDesc &II,
1714 unsigned Op, unsigned OpNum) {
1715 if (TargetRegisterInfo::isVirtualRegister(Op)) {
1716 const TargetRegisterClass *RegClass =
1717 TII.getRegClass(II, OpNum, &TRI, *FuncInfo.MF);
1718 if (!MRI.constrainRegClass(Op, RegClass)) {
1719 // If it's not legal to COPY between the register classes, something
1720 // has gone very wrong before we got here.
1721 unsigned NewOp = createResultReg(RegClass);
1722 BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc,
1723 TII.get(TargetOpcode::COPY), NewOp).addReg(Op);
1724 return NewOp;
1725 }
1726 }
1727 return Op;
1728}
1729
Dan Gohmanb2226e22008-08-13 20:19:35 +00001730unsigned FastISel::FastEmitInst_(unsigned MachineInstOpcode,
Dan Gohman2471f6c2008-08-20 18:09:38 +00001731 const TargetRegisterClass* RC) {
Dan Gohmanfe905652008-08-21 01:41:07 +00001732 unsigned ResultReg = createResultReg(RC);
Evan Cheng6cc775f2011-06-28 19:10:37 +00001733 const MCInstrDesc &II = TII.get(MachineInstOpcode);
Dan Gohmanb2226e22008-08-13 20:19:35 +00001734
Rafael Espindolaea09c592014-02-18 22:05:46 +00001735 BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc, II, ResultReg);
Dan Gohmanb2226e22008-08-13 20:19:35 +00001736 return ResultReg;
1737}
1738
1739unsigned FastISel::FastEmitInst_r(unsigned MachineInstOpcode,
1740 const TargetRegisterClass *RC,
Dan Gohman1a1b51f2010-05-11 23:54:07 +00001741 unsigned Op0, bool Op0IsKill) {
Evan Cheng6cc775f2011-06-28 19:10:37 +00001742 const MCInstrDesc &II = TII.get(MachineInstOpcode);
Dan Gohmanb2226e22008-08-13 20:19:35 +00001743
Tim Northover2f553f32014-04-15 13:59:49 +00001744 unsigned ResultReg = createResultReg(RC);
1745 Op0 = constrainOperandRegClass(II, Op0, II.getNumDefs());
1746
Evan Chenge775d352008-09-08 08:38:20 +00001747 if (II.getNumDefs() >= 1)
Rafael Espindolaea09c592014-02-18 22:05:46 +00001748 BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc, II, ResultReg)
Dan Gohmand7b5ce32010-07-10 09:00:22 +00001749 .addReg(Op0, Op0IsKill * RegState::Kill);
Evan Chenge775d352008-09-08 08:38:20 +00001750 else {
Rafael Espindolaea09c592014-02-18 22:05:46 +00001751 BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc, II)
Dan Gohmand7b5ce32010-07-10 09:00:22 +00001752 .addReg(Op0, Op0IsKill * RegState::Kill);
Rafael Espindolaea09c592014-02-18 22:05:46 +00001753 BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc,
1754 TII.get(TargetOpcode::COPY), ResultReg).addReg(II.ImplicitDefs[0]);
Evan Chenge775d352008-09-08 08:38:20 +00001755 }
1756
Dan Gohmanb2226e22008-08-13 20:19:35 +00001757 return ResultReg;
1758}
1759
1760unsigned FastISel::FastEmitInst_rr(unsigned MachineInstOpcode,
1761 const TargetRegisterClass *RC,
Dan Gohman1a1b51f2010-05-11 23:54:07 +00001762 unsigned Op0, bool Op0IsKill,
1763 unsigned Op1, bool Op1IsKill) {
Evan Cheng6cc775f2011-06-28 19:10:37 +00001764 const MCInstrDesc &II = TII.get(MachineInstOpcode);
Dan Gohmanb2226e22008-08-13 20:19:35 +00001765
Tim Northover2f553f32014-04-15 13:59:49 +00001766 unsigned ResultReg = createResultReg(RC);
1767 Op0 = constrainOperandRegClass(II, Op0, II.getNumDefs());
1768 Op1 = constrainOperandRegClass(II, Op1, II.getNumDefs() + 1);
1769
Evan Chenge775d352008-09-08 08:38:20 +00001770 if (II.getNumDefs() >= 1)
Rafael Espindolaea09c592014-02-18 22:05:46 +00001771 BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc, II, ResultReg)
Dan Gohman1a1b51f2010-05-11 23:54:07 +00001772 .addReg(Op0, Op0IsKill * RegState::Kill)
1773 .addReg(Op1, Op1IsKill * RegState::Kill);
Evan Chenge775d352008-09-08 08:38:20 +00001774 else {
Rafael Espindolaea09c592014-02-18 22:05:46 +00001775 BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc, II)
Dan Gohman1a1b51f2010-05-11 23:54:07 +00001776 .addReg(Op0, Op0IsKill * RegState::Kill)
1777 .addReg(Op1, Op1IsKill * RegState::Kill);
Rafael Espindolaea09c592014-02-18 22:05:46 +00001778 BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc,
1779 TII.get(TargetOpcode::COPY), ResultReg).addReg(II.ImplicitDefs[0]);
Evan Chenge775d352008-09-08 08:38:20 +00001780 }
Dan Gohmanb2226e22008-08-13 20:19:35 +00001781 return ResultReg;
1782}
Dan Gohmanfe905652008-08-21 01:41:07 +00001783
Owen Anderson68b6b0e2011-05-05 17:59:04 +00001784unsigned FastISel::FastEmitInst_rrr(unsigned MachineInstOpcode,
1785 const TargetRegisterClass *RC,
1786 unsigned Op0, bool Op0IsKill,
1787 unsigned Op1, bool Op1IsKill,
1788 unsigned Op2, bool Op2IsKill) {
Evan Cheng6cc775f2011-06-28 19:10:37 +00001789 const MCInstrDesc &II = TII.get(MachineInstOpcode);
Owen Anderson68b6b0e2011-05-05 17:59:04 +00001790
Tim Northover2f553f32014-04-15 13:59:49 +00001791 unsigned ResultReg = createResultReg(RC);
1792 Op0 = constrainOperandRegClass(II, Op0, II.getNumDefs());
1793 Op1 = constrainOperandRegClass(II, Op1, II.getNumDefs() + 1);
1794 Op2 = constrainOperandRegClass(II, Op2, II.getNumDefs() + 2);
1795
Owen Anderson68b6b0e2011-05-05 17:59:04 +00001796 if (II.getNumDefs() >= 1)
Rafael Espindolaea09c592014-02-18 22:05:46 +00001797 BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc, II, ResultReg)
Owen Anderson68b6b0e2011-05-05 17:59:04 +00001798 .addReg(Op0, Op0IsKill * RegState::Kill)
1799 .addReg(Op1, Op1IsKill * RegState::Kill)
1800 .addReg(Op2, Op2IsKill * RegState::Kill);
1801 else {
Rafael Espindolaea09c592014-02-18 22:05:46 +00001802 BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc, II)
Owen Anderson68b6b0e2011-05-05 17:59:04 +00001803 .addReg(Op0, Op0IsKill * RegState::Kill)
1804 .addReg(Op1, Op1IsKill * RegState::Kill)
1805 .addReg(Op2, Op2IsKill * RegState::Kill);
Rafael Espindolaea09c592014-02-18 22:05:46 +00001806 BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc,
1807 TII.get(TargetOpcode::COPY), ResultReg).addReg(II.ImplicitDefs[0]);
Owen Anderson68b6b0e2011-05-05 17:59:04 +00001808 }
1809 return ResultReg;
1810}
1811
Dan Gohmanfe905652008-08-21 01:41:07 +00001812unsigned FastISel::FastEmitInst_ri(unsigned MachineInstOpcode,
1813 const TargetRegisterClass *RC,
Dan Gohman1a1b51f2010-05-11 23:54:07 +00001814 unsigned Op0, bool Op0IsKill,
1815 uint64_t Imm) {
Evan Cheng6cc775f2011-06-28 19:10:37 +00001816 const MCInstrDesc &II = TII.get(MachineInstOpcode);
Dan Gohmanfe905652008-08-21 01:41:07 +00001817
Tim Northover2f553f32014-04-15 13:59:49 +00001818 unsigned ResultReg = createResultReg(RC);
1819 RC = TII.getRegClass(II, II.getNumDefs(), &TRI, *FuncInfo.MF);
1820 MRI.constrainRegClass(Op0, RC);
1821
Evan Chenge775d352008-09-08 08:38:20 +00001822 if (II.getNumDefs() >= 1)
Rafael Espindolaea09c592014-02-18 22:05:46 +00001823 BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc, II, ResultReg)
Dan Gohman1a1b51f2010-05-11 23:54:07 +00001824 .addReg(Op0, Op0IsKill * RegState::Kill)
1825 .addImm(Imm);
Evan Chenge775d352008-09-08 08:38:20 +00001826 else {
Rafael Espindolaea09c592014-02-18 22:05:46 +00001827 BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc, II)
Dan Gohman1a1b51f2010-05-11 23:54:07 +00001828 .addReg(Op0, Op0IsKill * RegState::Kill)
1829 .addImm(Imm);
Rafael Espindolaea09c592014-02-18 22:05:46 +00001830 BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc,
1831 TII.get(TargetOpcode::COPY), ResultReg).addReg(II.ImplicitDefs[0]);
Evan Chenge775d352008-09-08 08:38:20 +00001832 }
Dan Gohmanfe905652008-08-21 01:41:07 +00001833 return ResultReg;
1834}
1835
Owen Anderson66443c02011-03-11 21:33:55 +00001836unsigned FastISel::FastEmitInst_rii(unsigned MachineInstOpcode,
1837 const TargetRegisterClass *RC,
1838 unsigned Op0, bool Op0IsKill,
1839 uint64_t Imm1, uint64_t Imm2) {
Evan Cheng6cc775f2011-06-28 19:10:37 +00001840 const MCInstrDesc &II = TII.get(MachineInstOpcode);
Owen Anderson66443c02011-03-11 21:33:55 +00001841
Tim Northover2f553f32014-04-15 13:59:49 +00001842 unsigned ResultReg = createResultReg(RC);
1843 Op0 = constrainOperandRegClass(II, Op0, II.getNumDefs());
1844
Owen Anderson66443c02011-03-11 21:33:55 +00001845 if (II.getNumDefs() >= 1)
Rafael Espindolaea09c592014-02-18 22:05:46 +00001846 BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc, II, ResultReg)
Owen Anderson66443c02011-03-11 21:33:55 +00001847 .addReg(Op0, Op0IsKill * RegState::Kill)
1848 .addImm(Imm1)
1849 .addImm(Imm2);
1850 else {
Rafael Espindolaea09c592014-02-18 22:05:46 +00001851 BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc, II)
Owen Anderson66443c02011-03-11 21:33:55 +00001852 .addReg(Op0, Op0IsKill * RegState::Kill)
1853 .addImm(Imm1)
1854 .addImm(Imm2);
Rafael Espindolaea09c592014-02-18 22:05:46 +00001855 BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc,
1856 TII.get(TargetOpcode::COPY), ResultReg).addReg(II.ImplicitDefs[0]);
Owen Anderson66443c02011-03-11 21:33:55 +00001857 }
1858 return ResultReg;
1859}
1860
Dan Gohman5ca269e2008-08-27 01:09:54 +00001861unsigned FastISel::FastEmitInst_rf(unsigned MachineInstOpcode,
1862 const TargetRegisterClass *RC,
Dan Gohman1a1b51f2010-05-11 23:54:07 +00001863 unsigned Op0, bool Op0IsKill,
1864 const ConstantFP *FPImm) {
Evan Cheng6cc775f2011-06-28 19:10:37 +00001865 const MCInstrDesc &II = TII.get(MachineInstOpcode);
Dan Gohman5ca269e2008-08-27 01:09:54 +00001866
Tim Northover2f553f32014-04-15 13:59:49 +00001867 unsigned ResultReg = createResultReg(RC);
1868 Op0 = constrainOperandRegClass(II, Op0, II.getNumDefs());
1869
Evan Chenge775d352008-09-08 08:38:20 +00001870 if (II.getNumDefs() >= 1)
Rafael Espindolaea09c592014-02-18 22:05:46 +00001871 BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc, II, ResultReg)
Dan Gohman1a1b51f2010-05-11 23:54:07 +00001872 .addReg(Op0, Op0IsKill * RegState::Kill)
1873 .addFPImm(FPImm);
Evan Chenge775d352008-09-08 08:38:20 +00001874 else {
Rafael Espindolaea09c592014-02-18 22:05:46 +00001875 BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc, II)
Dan Gohman1a1b51f2010-05-11 23:54:07 +00001876 .addReg(Op0, Op0IsKill * RegState::Kill)
1877 .addFPImm(FPImm);
Rafael Espindolaea09c592014-02-18 22:05:46 +00001878 BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc,
1879 TII.get(TargetOpcode::COPY), ResultReg).addReg(II.ImplicitDefs[0]);
Evan Chenge775d352008-09-08 08:38:20 +00001880 }
Dan Gohman5ca269e2008-08-27 01:09:54 +00001881 return ResultReg;
1882}
1883
Dan Gohmanfe905652008-08-21 01:41:07 +00001884unsigned FastISel::FastEmitInst_rri(unsigned MachineInstOpcode,
1885 const TargetRegisterClass *RC,
Dan Gohman1a1b51f2010-05-11 23:54:07 +00001886 unsigned Op0, bool Op0IsKill,
1887 unsigned Op1, bool Op1IsKill,
1888 uint64_t Imm) {
Evan Cheng6cc775f2011-06-28 19:10:37 +00001889 const MCInstrDesc &II = TII.get(MachineInstOpcode);
Dan Gohmanfe905652008-08-21 01:41:07 +00001890
Tim Northover2f553f32014-04-15 13:59:49 +00001891 unsigned ResultReg = createResultReg(RC);
1892 Op0 = constrainOperandRegClass(II, Op0, II.getNumDefs());
1893 Op1 = constrainOperandRegClass(II, Op1, II.getNumDefs() + 1);
1894
Evan Chenge775d352008-09-08 08:38:20 +00001895 if (II.getNumDefs() >= 1)
Rafael Espindolaea09c592014-02-18 22:05:46 +00001896 BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc, II, ResultReg)
Dan Gohman1a1b51f2010-05-11 23:54:07 +00001897 .addReg(Op0, Op0IsKill * RegState::Kill)
1898 .addReg(Op1, Op1IsKill * RegState::Kill)
1899 .addImm(Imm);
Evan Chenge775d352008-09-08 08:38:20 +00001900 else {
Rafael Espindolaea09c592014-02-18 22:05:46 +00001901 BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc, II)
Dan Gohman1a1b51f2010-05-11 23:54:07 +00001902 .addReg(Op0, Op0IsKill * RegState::Kill)
1903 .addReg(Op1, Op1IsKill * RegState::Kill)
1904 .addImm(Imm);
Rafael Espindolaea09c592014-02-18 22:05:46 +00001905 BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc,
1906 TII.get(TargetOpcode::COPY), ResultReg).addReg(II.ImplicitDefs[0]);
Evan Chenge775d352008-09-08 08:38:20 +00001907 }
Dan Gohmanfe905652008-08-21 01:41:07 +00001908 return ResultReg;
1909}
Owen Anderson32635db2008-08-25 20:20:32 +00001910
Manman Rene8735522012-06-01 19:33:18 +00001911unsigned FastISel::FastEmitInst_rrii(unsigned MachineInstOpcode,
1912 const TargetRegisterClass *RC,
1913 unsigned Op0, bool Op0IsKill,
1914 unsigned Op1, bool Op1IsKill,
1915 uint64_t Imm1, uint64_t Imm2) {
Manman Rene8735522012-06-01 19:33:18 +00001916 const MCInstrDesc &II = TII.get(MachineInstOpcode);
1917
Tim Northover2f553f32014-04-15 13:59:49 +00001918 unsigned ResultReg = createResultReg(RC);
1919 Op0 = constrainOperandRegClass(II, Op0, II.getNumDefs());
1920 Op1 = constrainOperandRegClass(II, Op1, II.getNumDefs() + 1);
1921
Manman Rene8735522012-06-01 19:33:18 +00001922 if (II.getNumDefs() >= 1)
Rafael Espindolaea09c592014-02-18 22:05:46 +00001923 BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc, II, ResultReg)
Manman Rene8735522012-06-01 19:33:18 +00001924 .addReg(Op0, Op0IsKill * RegState::Kill)
1925 .addReg(Op1, Op1IsKill * RegState::Kill)
1926 .addImm(Imm1).addImm(Imm2);
1927 else {
Rafael Espindolaea09c592014-02-18 22:05:46 +00001928 BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc, II)
Manman Rene8735522012-06-01 19:33:18 +00001929 .addReg(Op0, Op0IsKill * RegState::Kill)
1930 .addReg(Op1, Op1IsKill * RegState::Kill)
1931 .addImm(Imm1).addImm(Imm2);
Rafael Espindolaea09c592014-02-18 22:05:46 +00001932 BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc,
1933 TII.get(TargetOpcode::COPY), ResultReg).addReg(II.ImplicitDefs[0]);
Manman Rene8735522012-06-01 19:33:18 +00001934 }
1935 return ResultReg;
1936}
1937
Owen Anderson32635db2008-08-25 20:20:32 +00001938unsigned FastISel::FastEmitInst_i(unsigned MachineInstOpcode,
1939 const TargetRegisterClass *RC,
1940 uint64_t Imm) {
1941 unsigned ResultReg = createResultReg(RC);
Evan Cheng6cc775f2011-06-28 19:10:37 +00001942 const MCInstrDesc &II = TII.get(MachineInstOpcode);
Wesley Peck527da1b2010-11-23 03:31:01 +00001943
Evan Chenge775d352008-09-08 08:38:20 +00001944 if (II.getNumDefs() >= 1)
Rafael Espindolaea09c592014-02-18 22:05:46 +00001945 BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc, II, ResultReg).addImm(Imm);
Evan Chenge775d352008-09-08 08:38:20 +00001946 else {
Rafael Espindolaea09c592014-02-18 22:05:46 +00001947 BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc, II).addImm(Imm);
1948 BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc,
1949 TII.get(TargetOpcode::COPY), ResultReg).addReg(II.ImplicitDefs[0]);
Evan Chenge775d352008-09-08 08:38:20 +00001950 }
Owen Anderson32635db2008-08-25 20:20:32 +00001951 return ResultReg;
Evan Cheng2c067322008-08-25 22:20:39 +00001952}
Owen Anderson5f57bc22008-08-27 22:30:02 +00001953
Owen Andersondd450b82011-04-22 23:38:06 +00001954unsigned FastISel::FastEmitInst_ii(unsigned MachineInstOpcode,
1955 const TargetRegisterClass *RC,
1956 uint64_t Imm1, uint64_t Imm2) {
1957 unsigned ResultReg = createResultReg(RC);
Evan Cheng6cc775f2011-06-28 19:10:37 +00001958 const MCInstrDesc &II = TII.get(MachineInstOpcode);
Owen Andersondd450b82011-04-22 23:38:06 +00001959
1960 if (II.getNumDefs() >= 1)
Rafael Espindolaea09c592014-02-18 22:05:46 +00001961 BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc, II, ResultReg)
Owen Andersondd450b82011-04-22 23:38:06 +00001962 .addImm(Imm1).addImm(Imm2);
1963 else {
Rafael Espindolaea09c592014-02-18 22:05:46 +00001964 BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc, II).addImm(Imm1).addImm(Imm2);
1965 BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc,
1966 TII.get(TargetOpcode::COPY), ResultReg).addReg(II.ImplicitDefs[0]);
Owen Andersondd450b82011-04-22 23:38:06 +00001967 }
1968 return ResultReg;
1969}
1970
Owen Anderson9f944592009-08-11 20:47:22 +00001971unsigned FastISel::FastEmitInst_extractsubreg(MVT RetVT,
Dan Gohman1a1b51f2010-05-11 23:54:07 +00001972 unsigned Op0, bool Op0IsKill,
1973 uint32_t Idx) {
Evan Cheng4a0bf662009-01-22 09:10:11 +00001974 unsigned ResultReg = createResultReg(TLI.getRegClassFor(RetVT));
Jakob Stoklund Olesen00264622010-07-08 16:40:22 +00001975 assert(TargetRegisterInfo::isVirtualRegister(Op0) &&
1976 "Cannot yet extract from physregs");
Jakob Stoklund Olesen1f1c6ad2012-05-20 06:38:37 +00001977 const TargetRegisterClass *RC = MRI.getRegClass(Op0);
1978 MRI.constrainRegClass(Op0, TRI.getSubClassWithSubReg(RC, Idx));
Dan Gohmand7b5ce32010-07-10 09:00:22 +00001979 BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt,
Rafael Espindolaea09c592014-02-18 22:05:46 +00001980 DbgLoc, TII.get(TargetOpcode::COPY), ResultReg)
Jakob Stoklund Olesen00264622010-07-08 16:40:22 +00001981 .addReg(Op0, getKillRegState(Op0IsKill), Idx);
Owen Anderson5f57bc22008-08-27 22:30:02 +00001982 return ResultReg;
1983}
Dan Gohmanc0bb9592009-03-13 20:42:20 +00001984
1985/// FastEmitZExtFromI1 - Emit MachineInstrs to compute the value of Op
1986/// with all but the least significant bit set to zero.
Dan Gohman1a1b51f2010-05-11 23:54:07 +00001987unsigned FastISel::FastEmitZExtFromI1(MVT VT, unsigned Op0, bool Op0IsKill) {
1988 return FastEmit_ri(VT, VT, ISD::AND, Op0, Op0IsKill, 1);
Dan Gohmanc0bb9592009-03-13 20:42:20 +00001989}
Dan Gohmanc594eab2010-04-22 20:46:50 +00001990
1991/// HandlePHINodesInSuccessorBlocks - Handle PHI nodes in successor blocks.
1992/// Emit code to ensure constants are copied into registers when needed.
1993/// Remember the virtual registers that need to be added to the Machine PHI
1994/// nodes as input. We cannot just directly add them, because expansion
1995/// might result in multiple MBB's for one BB. As such, the start of the
1996/// BB might correspond to a different MBB than the end.
1997bool FastISel::HandlePHINodesInSuccessorBlocks(const BasicBlock *LLVMBB) {
1998 const TerminatorInst *TI = LLVMBB->getTerminator();
1999
2000 SmallPtrSet<MachineBasicBlock *, 4> SuccsHandled;
Dan Gohman87fb4e82010-07-07 16:29:44 +00002001 unsigned OrigNumPHINodesToUpdate = FuncInfo.PHINodesToUpdate.size();
Dan Gohmanc594eab2010-04-22 20:46:50 +00002002
2003 // Check successor nodes' PHI nodes that expect a constant to be available
2004 // from this block.
2005 for (unsigned succ = 0, e = TI->getNumSuccessors(); succ != e; ++succ) {
2006 const BasicBlock *SuccBB = TI->getSuccessor(succ);
2007 if (!isa<PHINode>(SuccBB->begin())) continue;
Dan Gohman87fb4e82010-07-07 16:29:44 +00002008 MachineBasicBlock *SuccMBB = FuncInfo.MBBMap[SuccBB];
Dan Gohmanc594eab2010-04-22 20:46:50 +00002009
2010 // If this terminator has multiple identical successors (common for
2011 // switches), only handle each succ once.
2012 if (!SuccsHandled.insert(SuccMBB)) continue;
2013
2014 MachineBasicBlock::iterator MBBI = SuccMBB->begin();
2015
2016 // At this point we know that there is a 1-1 correspondence between LLVM PHI
2017 // nodes and Machine PHI nodes, but the incoming operands have not been
2018 // emitted yet.
2019 for (BasicBlock::const_iterator I = SuccBB->begin();
2020 const PHINode *PN = dyn_cast<PHINode>(I); ++I) {
Dan Gohmane6d40162010-05-07 01:10:20 +00002021
Dan Gohmanc594eab2010-04-22 20:46:50 +00002022 // Ignore dead phi's.
2023 if (PN->use_empty()) continue;
2024
2025 // Only handle legal types. Two interesting things to note here. First,
2026 // by bailing out early, we may leave behind some dead instructions,
2027 // since SelectionDAG's HandlePHINodesInSuccessorBlocks will insert its
Chris Lattner0ab5e2c2011-04-15 05:18:47 +00002028 // own moves. Second, this check is necessary because FastISel doesn't
Dan Gohman93f59202010-07-02 00:10:16 +00002029 // use CreateRegs to create registers, so it always creates
Dan Gohmanc594eab2010-04-22 20:46:50 +00002030 // exactly one register for each non-void instruction.
2031 EVT VT = TLI.getValueType(PN->getType(), /*AllowUnknown=*/true);
2032 if (VT == MVT::Other || !TLI.isTypeLegal(VT)) {
Chad Rosier6d68c7c2012-02-04 00:39:19 +00002033 // Handle integer promotions, though, because they're common and easy.
2034 if (VT == MVT::i1 || VT == MVT::i8 || VT == MVT::i16)
Dan Gohmanc594eab2010-04-22 20:46:50 +00002035 VT = TLI.getTypeToTransformTo(LLVMBB->getContext(), VT);
2036 else {
Dan Gohman87fb4e82010-07-07 16:29:44 +00002037 FuncInfo.PHINodesToUpdate.resize(OrigNumPHINodesToUpdate);
Dan Gohmanc594eab2010-04-22 20:46:50 +00002038 return false;
2039 }
2040 }
2041
2042 const Value *PHIOp = PN->getIncomingValueForBlock(LLVMBB);
2043
Dan Gohmane6d40162010-05-07 01:10:20 +00002044 // Set the DebugLoc for the copy. Prefer the location of the operand
2045 // if there is one; use the location of the PHI otherwise.
Rafael Espindolaea09c592014-02-18 22:05:46 +00002046 DbgLoc = PN->getDebugLoc();
Dan Gohmane6d40162010-05-07 01:10:20 +00002047 if (const Instruction *Inst = dyn_cast<Instruction>(PHIOp))
Rafael Espindolaea09c592014-02-18 22:05:46 +00002048 DbgLoc = Inst->getDebugLoc();
Dan Gohmane6d40162010-05-07 01:10:20 +00002049
Dan Gohmanc594eab2010-04-22 20:46:50 +00002050 unsigned Reg = getRegForValue(PHIOp);
2051 if (Reg == 0) {
Dan Gohman87fb4e82010-07-07 16:29:44 +00002052 FuncInfo.PHINodesToUpdate.resize(OrigNumPHINodesToUpdate);
Dan Gohmanc594eab2010-04-22 20:46:50 +00002053 return false;
2054 }
Dan Gohman87fb4e82010-07-07 16:29:44 +00002055 FuncInfo.PHINodesToUpdate.push_back(std::make_pair(MBBI++, Reg));
Rafael Espindolaea09c592014-02-18 22:05:46 +00002056 DbgLoc = DebugLoc();
Dan Gohmanc594eab2010-04-22 20:46:50 +00002057 }
2058 }
2059
2060 return true;
2061}
Eli Bendersky90dd3e72013-04-19 22:29:18 +00002062
2063bool FastISel::tryToFoldLoad(const LoadInst *LI, const Instruction *FoldInst) {
Eli Benderskye80691d2013-04-19 23:26:18 +00002064 assert(LI->hasOneUse() &&
2065 "tryToFoldLoad expected a LoadInst with a single use");
Eli Bendersky90dd3e72013-04-19 22:29:18 +00002066 // We know that the load has a single use, but don't know what it is. If it
2067 // isn't one of the folded instructions, then we can't succeed here. Handle
2068 // this by scanning the single-use users of the load until we get to FoldInst.
2069 unsigned MaxUsers = 6; // Don't scan down huge single-use chains of instrs.
2070
Chandler Carruthcdf47882014-03-09 03:16:01 +00002071 const Instruction *TheUser = LI->user_back();
Eli Bendersky90dd3e72013-04-19 22:29:18 +00002072 while (TheUser != FoldInst && // Scan up until we find FoldInst.
2073 // Stay in the right block.
2074 TheUser->getParent() == FoldInst->getParent() &&
2075 --MaxUsers) { // Don't scan too far.
2076 // If there are multiple or no uses of this instruction, then bail out.
2077 if (!TheUser->hasOneUse())
2078 return false;
2079
Chandler Carruthcdf47882014-03-09 03:16:01 +00002080 TheUser = TheUser->user_back();
Eli Bendersky90dd3e72013-04-19 22:29:18 +00002081 }
2082
2083 // If we didn't find the fold instruction, then we failed to collapse the
2084 // sequence.
2085 if (TheUser != FoldInst)
2086 return false;
2087
2088 // Don't try to fold volatile loads. Target has to deal with alignment
2089 // constraints.
Eli Benderskye80691d2013-04-19 23:26:18 +00002090 if (LI->isVolatile())
2091 return false;
Eli Bendersky90dd3e72013-04-19 22:29:18 +00002092
2093 // Figure out which vreg this is going into. If there is no assigned vreg yet
2094 // then there actually was no reference to it. Perhaps the load is referenced
2095 // by a dead instruction.
2096 unsigned LoadReg = getRegForValue(LI);
2097 if (LoadReg == 0)
2098 return false;
2099
Eli Benderskye80691d2013-04-19 23:26:18 +00002100 // We can't fold if this vreg has no uses or more than one use. Multiple uses
2101 // may mean that the instruction got lowered to multiple MIs, or the use of
2102 // the loaded value ended up being multiple operands of the result.
2103 if (!MRI.hasOneUse(LoadReg))
2104 return false;
2105
Eli Bendersky90dd3e72013-04-19 22:29:18 +00002106 MachineRegisterInfo::reg_iterator RI = MRI.reg_begin(LoadReg);
Owen Anderson16c6bf42014-03-13 23:12:04 +00002107 MachineInstr *User = RI->getParent();
Eli Bendersky90dd3e72013-04-19 22:29:18 +00002108
2109 // Set the insertion point properly. Folding the load can cause generation of
Eli Benderskye80691d2013-04-19 23:26:18 +00002110 // other random instructions (like sign extends) for addressing modes; make
Eli Bendersky90dd3e72013-04-19 22:29:18 +00002111 // sure they get inserted in a logical place before the new instruction.
2112 FuncInfo.InsertPt = User;
2113 FuncInfo.MBB = User->getParent();
2114
2115 // Ask the target to try folding the load.
2116 return tryToFoldLoadIntoMI(User, RI.getOperandNo(), LI);
2117}
2118
Bob Wilson9f3e6b22013-11-15 19:09:27 +00002119bool FastISel::canFoldAddIntoGEP(const User *GEP, const Value *Add) {
2120 // Must be an add.
2121 if (!isa<AddOperator>(Add))
2122 return false;
2123 // Type size needs to match.
Rafael Espindolaea09c592014-02-18 22:05:46 +00002124 if (DL.getTypeSizeInBits(GEP->getType()) !=
2125 DL.getTypeSizeInBits(Add->getType()))
Bob Wilson9f3e6b22013-11-15 19:09:27 +00002126 return false;
2127 // Must be in the same basic block.
2128 if (isa<Instruction>(Add) &&
2129 FuncInfo.MBBMap[cast<Instruction>(Add)->getParent()] != FuncInfo.MBB)
2130 return false;
2131 // Must have a constant operand.
2132 return isa<ConstantInt>(cast<AddOperator>(Add)->getOperand(1));
2133}
Eli Bendersky90dd3e72013-04-19 22:29:18 +00002134
Juergen Ributzka349777d2014-06-12 23:27:57 +00002135MachineMemOperand *
2136FastISel::createMachineMemOperandFor(const Instruction *I) const {
2137 const Value *Ptr;
2138 Type *ValTy;
2139 unsigned Alignment;
2140 unsigned Flags;
2141 bool IsVolatile;
2142
2143 if (const auto *LI = dyn_cast<LoadInst>(I)) {
2144 Alignment = LI->getAlignment();
2145 IsVolatile = LI->isVolatile();
2146 Flags = MachineMemOperand::MOLoad;
2147 Ptr = LI->getPointerOperand();
2148 ValTy = LI->getType();
2149 } else if (const auto *SI = dyn_cast<StoreInst>(I)) {
2150 Alignment = SI->getAlignment();
2151 IsVolatile = SI->isVolatile();
2152 Flags = MachineMemOperand::MOStore;
2153 Ptr = SI->getPointerOperand();
2154 ValTy = SI->getValueOperand()->getType();
2155 } else {
2156 return nullptr;
2157 }
2158
2159 bool IsNonTemporal = I->getMetadata("nontemporal") != nullptr;
2160 bool IsInvariant = I->getMetadata("invariant.load") != nullptr;
Juergen Ributzka349777d2014-06-12 23:27:57 +00002161 const MDNode *Ranges = I->getMetadata(LLVMContext::MD_range);
2162
Hal Finkelcc39b672014-07-24 12:16:19 +00002163 AAMDNodes AAInfo;
2164 I->getAAMetadata(AAInfo);
2165
Juergen Ributzka349777d2014-06-12 23:27:57 +00002166 if (Alignment == 0) // Ensure that codegen never sees alignment 0.
2167 Alignment = DL.getABITypeAlignment(ValTy);
2168
2169 unsigned Size = TM.getDataLayout()->getTypeStoreSize(ValTy);
2170
2171 if (IsVolatile)
2172 Flags |= MachineMemOperand::MOVolatile;
2173 if (IsNonTemporal)
2174 Flags |= MachineMemOperand::MONonTemporal;
2175 if (IsInvariant)
2176 Flags |= MachineMemOperand::MOInvariant;
2177
2178 return FuncInfo.MF->getMachineMemOperand(MachinePointerInfo(Ptr), Flags, Size,
Hal Finkelcc39b672014-07-24 12:16:19 +00002179 Alignment, AAInfo, Ranges);
Juergen Ributzka349777d2014-06-12 23:27:57 +00002180}