blob: 8facbc28e404e2628300f3210bae1350930600b5 [file] [log] [blame]
Dan Gohman3b172f12010-04-22 20:06:42 +00001//===-- FastISel.cpp - Implementation of the FastISel class ---------------===//
Dan Gohmanb0cf29c2008-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 Gohman5ec9efd2008-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 Lattner44d2a982008-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 Gohman5ec9efd2008-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 Lattner44d2a982008-10-13 01:59:13 +000022// support. In many cases, this allows us to avoid duplicating a lot of
Dan Gohman5ec9efd2008-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 Lattner44d2a982008-10-13 01:59:13 +000027// weighed against the speed at which the code can be generated. Also,
Dan Gohman5ec9efd2008-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 Lattner44d2a982008-10-13 01:59:13 +000030// time. Despite its limitations, "fast" instruction selection is able to
Dan Gohman5ec9efd2008-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 Lattner44d2a982008-10-13 01:59:13 +000037// from simple operators. More complicated operations currently require
Dan Gohman5ec9efd2008-09-30 20:48:29 +000038// target-specific code.
39//
Dan Gohmanb0cf29c2008-08-13 20:19:35 +000040//===----------------------------------------------------------------------===//
41
Stephen Hines37ed9c12014-12-01 14:51:49 -080042#include "llvm/CodeGen/Analysis.h"
Chandler Carruthd04a8d42012-12-03 16:50:05 +000043#include "llvm/CodeGen/FastISel.h"
David Blaikie6d9dbd52013-06-16 20:34:15 +000044#include "llvm/ADT/Optional.h"
Chandler Carruthd04a8d42012-12-03 16:50:05 +000045#include "llvm/ADT/Statistic.h"
Stephen Hinesc6a4f5e2014-07-21 00:45:20 -070046#include "llvm/Analysis/BranchProbabilityInfo.h"
Chandler Carruthd04a8d42012-12-03 16:50:05 +000047#include "llvm/Analysis/Loads.h"
48#include "llvm/CodeGen/Analysis.h"
49#include "llvm/CodeGen/FunctionLoweringInfo.h"
Stephen Hinesc6a4f5e2014-07-21 00:45:20 -070050#include "llvm/CodeGen/MachineFrameInfo.h"
Chandler Carruthd04a8d42012-12-03 16:50:05 +000051#include "llvm/CodeGen/MachineInstrBuilder.h"
52#include "llvm/CodeGen/MachineModuleInfo.h"
53#include "llvm/CodeGen/MachineRegisterInfo.h"
Stephen Hinesc6a4f5e2014-07-21 00:45:20 -070054#include "llvm/CodeGen/StackMaps.h"
Chandler Carruth0b8c9a82013-01-02 11:36:10 +000055#include "llvm/IR/DataLayout.h"
Stephen Hines36b56882014-04-23 16:57:46 -070056#include "llvm/IR/DebugInfo.h"
Chandler Carruth0b8c9a82013-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 Carruthd04a8d42012-12-03 16:50:05 +000062#include "llvm/Support/Debug.h"
63#include "llvm/Support/ErrorHandling.h"
Dan Gohmanb0cf29c2008-08-13 20:19:35 +000064#include "llvm/Target/TargetInstrInfo.h"
Bob Wilsond49edb72012-08-03 04:06:28 +000065#include "llvm/Target/TargetLibraryInfo.h"
Evan Cheng83785c82008-08-20 22:45:34 +000066#include "llvm/Target/TargetLowering.h"
Dan Gohmanbb466332008-08-20 21:05:57 +000067#include "llvm/Target/TargetMachine.h"
Stephen Hines37ed9c12014-12-01 14:51:49 -080068#include "llvm/Target/TargetSubtargetInfo.h"
Dan Gohmanb0cf29c2008-08-13 20:19:35 +000069using namespace llvm;
70
Stephen Hinesdce4a402014-05-29 02:49:00 -070071#define DEBUG_TYPE "isel"
72
Chad Rosieraa5656c2011-11-28 19:59:09 +000073STATISTIC(NumFastIselSuccessIndependent, "Number of insts selected by "
Stephen Hines37ed9c12014-12-01 14:51:49 -080074 "target-independent selector");
Chad Rosieraa5656c2011-11-28 19:59:09 +000075STATISTIC(NumFastIselSuccessTarget, "Number of insts selected by "
Stephen Hines37ed9c12014-12-01 14:51:49 -080076 "target-specific selector");
Chad Rosierae6f2cb2011-11-29 19:40:47 +000077STATISTIC(NumFastIselDead, "Number of dead insts removed on failure");
Chad Rosier053e69a2011-11-16 21:05:28 +000078
Stephen Hines37ed9c12014-12-01 14:51:49 -080079void FastISel::ArgListEntry::setAttributes(ImmutableCallSite *CS,
80 unsigned AttrIdx) {
81 IsSExt = CS->paramHasAttr(AttrIdx, Attribute::SExt);
82 IsZExt = CS->paramHasAttr(AttrIdx, Attribute::ZExt);
83 IsInReg = CS->paramHasAttr(AttrIdx, Attribute::InReg);
84 IsSRet = CS->paramHasAttr(AttrIdx, Attribute::StructRet);
85 IsNest = CS->paramHasAttr(AttrIdx, Attribute::Nest);
86 IsByVal = CS->paramHasAttr(AttrIdx, Attribute::ByVal);
87 IsInAlloca = CS->paramHasAttr(AttrIdx, Attribute::InAlloca);
88 IsReturned = CS->paramHasAttr(AttrIdx, Attribute::Returned);
89 Alignment = CS->getParamAlignment(AttrIdx);
90}
91
92/// Set the current block to which generated machine instructions will be
93/// appended, and clear the local CSE map.
Dan Gohman84023e02010-07-10 09:00:22 +000094void FastISel::startNewBlock() {
95 LocalValueMap.clear();
96
Jakob Stoklund Olesen1ab111e2013-07-04 04:53:49 +000097 // Instructions are appended to FuncInfo.MBB. If the basic block already
Jakob Stoklund Olesenef22e0e2013-07-04 04:32:39 +000098 // contains labels or copies, use the last instruction as the last local
99 // value.
Stephen Hinesdce4a402014-05-29 02:49:00 -0700100 EmitStartPt = nullptr;
Jakob Stoklund Olesenef22e0e2013-07-04 04:32:39 +0000101 if (!FuncInfo.MBB->empty())
102 EmitStartPt = &FuncInfo.MBB->back();
Ivan Krasin74af88a2011-08-18 22:06:10 +0000103 LastLocalValue = EmitStartPt;
104}
105
Stephen Hines37ed9c12014-12-01 14:51:49 -0800106bool FastISel::lowerArguments() {
Evan Cheng092e5e72013-02-11 01:27:15 +0000107 if (!FuncInfo.CanLowerReturn)
108 // Fallback to SDISel argument lowering code to deal with sret pointer
109 // parameter.
110 return false;
Stephen Lin155615d2013-07-08 00:37:03 +0000111
Stephen Hines37ed9c12014-12-01 14:51:49 -0800112 if (!fastLowerArguments())
Evan Cheng092e5e72013-02-11 01:27:15 +0000113 return false;
114
David Blaikie19489102013-06-21 22:56:30 +0000115 // Enter arguments into ValueMap for uses in non-entry BBs.
Evan Cheng092e5e72013-02-11 01:27:15 +0000116 for (Function::const_arg_iterator I = FuncInfo.Fn->arg_begin(),
Stephen Hines37ed9c12014-12-01 14:51:49 -0800117 E = FuncInfo.Fn->arg_end();
118 I != E; ++I) {
David Blaikie19489102013-06-21 22:56:30 +0000119 DenseMap<const Value *, unsigned>::iterator VI = LocalValueMap.find(I);
120 assert(VI != LocalValueMap.end() && "Missed an argument?");
121 FuncInfo.ValueMap[I] = VI->second;
Evan Cheng092e5e72013-02-11 01:27:15 +0000122 }
123 return true;
124}
125
Ivan Krasin74af88a2011-08-18 22:06:10 +0000126void FastISel::flushLocalValueMap() {
127 LocalValueMap.clear();
128 LastLocalValue = EmitStartPt;
129 recomputeInsertPt();
Stephen Hines37ed9c12014-12-01 14:51:49 -0800130 SavedInsertPt = FuncInfo.InsertPt;
Dan Gohman84023e02010-07-10 09:00:22 +0000131}
132
Stephen Hines37ed9c12014-12-01 14:51:49 -0800133bool FastISel::hasTrivialKill(const Value *V) {
Dan Gohman7f0d6952010-05-14 22:53:18 +0000134 // Don't consider constants or arguments to have trivial kills.
Dan Gohmana6cb6412010-05-11 23:54:07 +0000135 const Instruction *I = dyn_cast<Instruction>(V);
Dan Gohman7f0d6952010-05-14 22:53:18 +0000136 if (!I)
137 return false;
138
139 // No-op casts are trivially coalesced by fast-isel.
Stephen Hines37ed9c12014-12-01 14:51:49 -0800140 if (const auto *Cast = dyn_cast<CastInst>(I))
Stephen Hines36b56882014-04-23 16:57:46 -0700141 if (Cast->isNoopCast(DL.getIntPtrType(Cast->getContext())) &&
Chandler Carruthece6c6b2012-11-01 08:07:29 +0000142 !hasTrivialKill(Cast->getOperand(0)))
Dan Gohman7f0d6952010-05-14 22:53:18 +0000143 return false;
144
Stephen Hines37ed9c12014-12-01 14:51:49 -0800145 // Even the value might have only one use in the LLVM IR, it is possible that
146 // FastISel might fold the use into another instruction and now there is more
147 // than one use at the Machine Instruction level.
148 unsigned Reg = lookUpRegForValue(V);
149 if (Reg && !MRI.use_empty(Reg))
150 return false;
151
Chad Rosier22b34cc2011-11-15 23:34:05 +0000152 // GEPs with all zero indices are trivially coalesced by fast-isel.
Stephen Hines37ed9c12014-12-01 14:51:49 -0800153 if (const auto *GEP = dyn_cast<GetElementPtrInst>(I))
Chad Rosier22b34cc2011-11-15 23:34:05 +0000154 if (GEP->hasAllZeroIndices() && !hasTrivialKill(GEP->getOperand(0)))
155 return false;
156
Dan Gohman7f0d6952010-05-14 22:53:18 +0000157 // Only instructions with a single use in the same basic block are considered
158 // to have trivial kills.
159 return I->hasOneUse() &&
160 !(I->getOpcode() == Instruction::BitCast ||
161 I->getOpcode() == Instruction::PtrToInt ||
162 I->getOpcode() == Instruction::IntToPtr) &&
Stephen Hines36b56882014-04-23 16:57:46 -0700163 cast<Instruction>(*I->user_begin())->getParent() == I->getParent();
Dan Gohmana6cb6412010-05-11 23:54:07 +0000164}
165
Dan Gohman46510a72010-04-15 01:51:59 +0000166unsigned FastISel::getRegForValue(const Value *V) {
Owen Andersone50ed302009-08-10 22:56:29 +0000167 EVT RealVT = TLI.getValueType(V->getType(), /*AllowUnknown=*/true);
Dan Gohman4fd55282009-04-07 20:40:11 +0000168 // Don't handle non-simple values in FastISel.
169 if (!RealVT.isSimple())
170 return 0;
Dan Gohmanc8a1a3c2008-12-08 07:57:47 +0000171
172 // Ignore illegal types. We must do this before looking up the value
173 // in ValueMap because Arguments are given virtual registers regardless
174 // of whether FastISel can handle them.
Owen Anderson825b72b2009-08-11 20:47:22 +0000175 MVT VT = RealVT.getSimpleVT();
Dan Gohmanc8a1a3c2008-12-08 07:57:47 +0000176 if (!TLI.isTypeLegal(VT)) {
Eli Friedman76927d732011-05-25 23:49:02 +0000177 // Handle integer promotions, though, because they're common and easy.
178 if (VT == MVT::i1 || VT == MVT::i8 || VT == MVT::i16)
Owen Anderson23b9b192009-08-12 00:36:31 +0000179 VT = TLI.getTypeToTransformTo(V->getContext(), VT).getSimpleVT();
Dan Gohmanc8a1a3c2008-12-08 07:57:47 +0000180 else
181 return 0;
182 }
183
Eric Christopher4e270272012-03-20 01:07:47 +0000184 // Look up the value to see if we already have a register for it.
185 unsigned Reg = lookUpRegForValue(V);
Stephen Hines37ed9c12014-12-01 14:51:49 -0800186 if (Reg)
Dan Gohman104e4ce2008-09-03 23:32:19 +0000187 return Reg;
Dan Gohmanad368ac2008-08-27 18:10:19 +0000188
Dan Gohman97c94b82010-05-06 00:02:14 +0000189 // In bottom-up mode, just create the virtual register which will be used
190 // to hold the value. It will be materialized later.
Dan Gohman84023e02010-07-10 09:00:22 +0000191 if (isa<Instruction>(V) &&
192 (!isa<AllocaInst>(V) ||
193 !FuncInfo.StaticAllocaMap.count(cast<AllocaInst>(V))))
194 return FuncInfo.InitializeRegForValue(V);
Dan Gohman97c94b82010-05-06 00:02:14 +0000195
Eric Christopher76ad43c2012-10-03 08:10:01 +0000196 SavePoint SaveInsertPt = enterLocalValueArea();
Dan Gohman84023e02010-07-10 09:00:22 +0000197
198 // Materialize the value in a register. Emit any instructions in the
199 // local value area.
200 Reg = materializeRegForValue(V, VT);
201
Eric Christopher76ad43c2012-10-03 08:10:01 +0000202 leaveLocalValueArea(SaveInsertPt);
Dan Gohman84023e02010-07-10 09:00:22 +0000203
204 return Reg;
Dan Gohman1fdc6142010-05-03 23:36:34 +0000205}
206
Stephen Hines37ed9c12014-12-01 14:51:49 -0800207unsigned FastISel::materializeConstant(const Value *V, MVT VT) {
Dan Gohman1fdc6142010-05-03 23:36:34 +0000208 unsigned Reg = 0;
Stephen Hines37ed9c12014-12-01 14:51:49 -0800209 if (const auto *CI = dyn_cast<ConstantInt>(V)) {
Dan Gohman2ff7fd12008-09-19 22:16:54 +0000210 if (CI->getValue().getActiveBits() <= 64)
Stephen Hines37ed9c12014-12-01 14:51:49 -0800211 Reg = fastEmit_i(VT, VT, ISD::Constant, CI->getZExtValue());
212 } else if (isa<AllocaInst>(V))
213 Reg = fastMaterializeAlloca(cast<AllocaInst>(V));
214 else if (isa<ConstantPointerNull>(V))
Dan Gohman1e9e8c32008-10-07 22:03:27 +0000215 // Translate this as an integer zero so that it can be
216 // local-CSE'd with actual integer zeros.
Stephen Hines37ed9c12014-12-01 14:51:49 -0800217 Reg = getRegForValue(
218 Constant::getNullValue(DL.getIntPtrType(V->getContext())));
219 else if (const auto *CF = dyn_cast<ConstantFP>(V)) {
220 if (CF->isNullValue())
221 Reg = fastMaterializeFloatZero(CF);
222 else
Eli Friedman2790ba82011-04-27 22:41:55 +0000223 // Try to emit the constant directly.
Stephen Hines37ed9c12014-12-01 14:51:49 -0800224 Reg = fastEmit_f(VT, VT, ISD::ConstantFP, CF);
Dan Gohmanad368ac2008-08-27 18:10:19 +0000225
226 if (!Reg) {
Dan Gohman4183e312010-04-13 17:07:06 +0000227 // Try to emit the constant by using an integer constant with a cast.
Dan Gohmanad368ac2008-08-27 18:10:19 +0000228 const APFloat &Flt = CF->getValueAPF();
Owen Andersone50ed302009-08-10 22:56:29 +0000229 EVT IntVT = TLI.getPointerTy();
Dan Gohmanad368ac2008-08-27 18:10:19 +0000230
231 uint64_t x[2];
232 uint32_t IntBitWidth = IntVT.getSizeInBits();
Dale Johannesen23a98552008-10-09 23:00:39 +0000233 bool isExact;
Stephen Hines37ed9c12014-12-01 14:51:49 -0800234 (void)Flt.convertToInteger(x, IntBitWidth, /*isSigned=*/true,
235 APFloat::rmTowardZero, &isExact);
Dale Johannesen23a98552008-10-09 23:00:39 +0000236 if (isExact) {
Jeffrey Yasskin3ba292d2011-07-18 21:45:40 +0000237 APInt IntVal(IntBitWidth, x);
Dan Gohmanad368ac2008-08-27 18:10:19 +0000238
Owen Andersone922c022009-07-22 00:24:57 +0000239 unsigned IntegerReg =
Stephen Hines37ed9c12014-12-01 14:51:49 -0800240 getRegForValue(ConstantInt::get(V->getContext(), IntVal));
Dan Gohman2ff7fd12008-09-19 22:16:54 +0000241 if (IntegerReg != 0)
Stephen Hines37ed9c12014-12-01 14:51:49 -0800242 Reg = fastEmit_r(IntVT.getSimpleVT(), VT, ISD::SINT_TO_FP, IntegerReg,
243 /*Kill=*/false);
Dan Gohman2ff7fd12008-09-19 22:16:54 +0000244 }
Dan Gohmanad368ac2008-08-27 18:10:19 +0000245 }
Stephen Hines37ed9c12014-12-01 14:51:49 -0800246 } else if (const auto *Op = dyn_cast<Operator>(V)) {
247 if (!selectOperator(Op, Op->getOpcode()))
Dan Gohman20d4be12010-07-01 02:58:57 +0000248 if (!isa<Instruction>(Op) ||
Stephen Hines37ed9c12014-12-01 14:51:49 -0800249 !fastSelectInstruction(cast<Instruction>(Op)))
Dan Gohman20d4be12010-07-01 02:58:57 +0000250 return 0;
Dan Gohman37db6cd2010-06-21 14:17:46 +0000251 Reg = lookUpRegForValue(Op);
Dan Gohman205d9252008-08-28 21:19:07 +0000252 } else if (isa<UndefValue>(V)) {
Dan Gohman104e4ce2008-09-03 23:32:19 +0000253 Reg = createResultReg(TLI.getRegClassFor(VT));
Stephen Hines36b56882014-04-23 16:57:46 -0700254 BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc,
Dan Gohman84023e02010-07-10 09:00:22 +0000255 TII.get(TargetOpcode::IMPLICIT_DEF), Reg);
Dan Gohmanad368ac2008-08-27 18:10:19 +0000256 }
Stephen Hines37ed9c12014-12-01 14:51:49 -0800257 return Reg;
258}
Wesley Peckbf17cfa2010-11-23 03:31:01 +0000259
Stephen Hines37ed9c12014-12-01 14:51:49 -0800260/// Helper for getRegForValue. This function is called when the value isn't
261/// already available in a register and must be materialized with new
262/// instructions.
263unsigned FastISel::materializeRegForValue(const Value *V, MVT VT) {
264 unsigned Reg = 0;
265 // Give the target-specific code a try first.
266 if (isa<Constant>(V))
267 Reg = fastMaterializeConstant(cast<Constant>(V));
268
269 // If target-specific code couldn't or didn't want to handle the value, then
270 // give target-independent code a try.
271 if (!Reg)
272 Reg = materializeConstant(V, VT);
Wesley Peckbf17cfa2010-11-23 03:31:01 +0000273
Dan Gohman2ff7fd12008-09-19 22:16:54 +0000274 // Don't cache constant materializations in the general ValueMap.
275 // To do so would require tracking what uses they dominate.
Stephen Hines37ed9c12014-12-01 14:51:49 -0800276 if (Reg) {
Dan Gohmandceffe62008-09-25 01:28:51 +0000277 LocalValueMap[V] = Reg;
Dan Gohman84023e02010-07-10 09:00:22 +0000278 LastLocalValue = MRI.getVRegDef(Reg);
279 }
Dan Gohman104e4ce2008-09-03 23:32:19 +0000280 return Reg;
Dan Gohmanad368ac2008-08-27 18:10:19 +0000281}
282
Dan Gohman46510a72010-04-15 01:51:59 +0000283unsigned FastISel::lookUpRegForValue(const Value *V) {
Evan Cheng59fbc802008-09-09 01:26:59 +0000284 // Look up the value to see if we already have a register for it. We
285 // cache values defined by Instructions across blocks, and other values
286 // only locally. This is because Instructions already have the SSA
Dan Gohman1fdc6142010-05-03 23:36:34 +0000287 // def-dominates-use requirement enforced.
Dan Gohmana4160c32010-07-07 16:29:44 +0000288 DenseMap<const Value *, unsigned>::iterator I = FuncInfo.ValueMap.find(V);
289 if (I != FuncInfo.ValueMap.end())
Dan Gohman3193a682010-06-21 14:21:47 +0000290 return I->second;
Eric Christopher76ad43c2012-10-03 08:10:01 +0000291 return LocalValueMap[V];
Evan Cheng59fbc802008-09-09 01:26:59 +0000292}
293
Stephen Hines37ed9c12014-12-01 14:51:49 -0800294void FastISel::updateValueMap(const Value *I, unsigned Reg, unsigned NumRegs) {
Dan Gohman40b189e2008-09-05 18:18:20 +0000295 if (!isa<Instruction>(I)) {
296 LocalValueMap[I] = Reg;
Eli Friedman482feb32011-05-16 21:06:17 +0000297 return;
Dan Gohman40b189e2008-09-05 18:18:20 +0000298 }
Wesley Peckbf17cfa2010-11-23 03:31:01 +0000299
Dan Gohmana4160c32010-07-07 16:29:44 +0000300 unsigned &AssignedReg = FuncInfo.ValueMap[I];
Chris Lattnerc5040ab2009-04-12 07:45:01 +0000301 if (AssignedReg == 0)
Dan Gohman84023e02010-07-10 09:00:22 +0000302 // Use the new register.
Chris Lattnerc5040ab2009-04-12 07:45:01 +0000303 AssignedReg = Reg;
Chris Lattner36e39462009-04-12 07:46:30 +0000304 else if (Reg != AssignedReg) {
Dan Gohman84023e02010-07-10 09:00:22 +0000305 // Arrange for uses of AssignedReg to be replaced by uses of Reg.
Eli Friedman482feb32011-05-16 21:06:17 +0000306 for (unsigned i = 0; i < NumRegs; i++)
Stephen Hines37ed9c12014-12-01 14:51:49 -0800307 FuncInfo.RegFixups[AssignedReg + i] = Reg + i;
Dan Gohman84023e02010-07-10 09:00:22 +0000308
309 AssignedReg = Reg;
Chris Lattnerc5040ab2009-04-12 07:45:01 +0000310 }
Owen Andersoncc54e762008-08-30 00:38:46 +0000311}
312
Dan Gohmana6cb6412010-05-11 23:54:07 +0000313std::pair<unsigned, bool> FastISel::getRegForGEPIndex(const Value *Idx) {
Dan Gohmanc8a1a3c2008-12-08 07:57:47 +0000314 unsigned IdxN = getRegForValue(Idx);
315 if (IdxN == 0)
316 // Unhandled operand. Halt "fast" selection and bail.
Dan Gohmana6cb6412010-05-11 23:54:07 +0000317 return std::pair<unsigned, bool>(0, false);
318
319 bool IdxNIsKill = hasTrivialKill(Idx);
Dan Gohmanc8a1a3c2008-12-08 07:57:47 +0000320
321 // If the index is smaller or larger than intptr_t, truncate or extend it.
Owen Anderson766b5ef2009-08-11 21:59:30 +0000322 MVT PtrVT = TLI.getPointerTy();
Owen Andersone50ed302009-08-10 22:56:29 +0000323 EVT IdxVT = EVT::getEVT(Idx->getType(), /*HandleUnknown=*/false);
Dan Gohmana6cb6412010-05-11 23:54:07 +0000324 if (IdxVT.bitsLT(PtrVT)) {
Stephen Hines37ed9c12014-12-01 14:51:49 -0800325 IdxN = fastEmit_r(IdxVT.getSimpleVT(), PtrVT, ISD::SIGN_EXTEND, IdxN,
326 IdxNIsKill);
Dan Gohmana6cb6412010-05-11 23:54:07 +0000327 IdxNIsKill = true;
Stephen Hines37ed9c12014-12-01 14:51:49 -0800328 } else if (IdxVT.bitsGT(PtrVT)) {
329 IdxN =
330 fastEmit_r(IdxVT.getSimpleVT(), PtrVT, ISD::TRUNCATE, IdxN, IdxNIsKill);
Dan Gohmana6cb6412010-05-11 23:54:07 +0000331 IdxNIsKill = true;
332 }
333 return std::pair<unsigned, bool>(IdxN, IdxNIsKill);
Dan Gohmanc8a1a3c2008-12-08 07:57:47 +0000334}
335
Dan Gohman84023e02010-07-10 09:00:22 +0000336void FastISel::recomputeInsertPt() {
337 if (getLastLocalValue()) {
338 FuncInfo.InsertPt = getLastLocalValue();
Dan Gohmanc6e59b72010-07-19 22:48:56 +0000339 FuncInfo.MBB = FuncInfo.InsertPt->getParent();
Dan Gohman84023e02010-07-10 09:00:22 +0000340 ++FuncInfo.InsertPt;
341 } else
342 FuncInfo.InsertPt = FuncInfo.MBB->getFirstNonPHI();
343
344 // Now skip past any EH_LABELs, which must remain at the beginning.
345 while (FuncInfo.InsertPt != FuncInfo.MBB->end() &&
346 FuncInfo.InsertPt->getOpcode() == TargetOpcode::EH_LABEL)
347 ++FuncInfo.InsertPt;
348}
349
Chad Rosierae6f2cb2011-11-29 19:40:47 +0000350void FastISel::removeDeadCode(MachineBasicBlock::iterator I,
351 MachineBasicBlock::iterator E) {
Stephen Hines37ed9c12014-12-01 14:51:49 -0800352 assert(I && E && std::distance(I, E) > 0 && "Invalid iterator!");
Chad Rosierae6f2cb2011-11-29 19:40:47 +0000353 while (I != E) {
354 MachineInstr *Dead = &*I;
355 ++I;
356 Dead->eraseFromParent();
Jan Wen Voungfa785cb2013-03-08 22:56:31 +0000357 ++NumFastIselDead;
Chad Rosierae6f2cb2011-11-29 19:40:47 +0000358 }
359 recomputeInsertPt();
360}
361
Eric Christopher76ad43c2012-10-03 08:10:01 +0000362FastISel::SavePoint FastISel::enterLocalValueArea() {
Dan Gohman84023e02010-07-10 09:00:22 +0000363 MachineBasicBlock::iterator OldInsertPt = FuncInfo.InsertPt;
Stephen Hines36b56882014-04-23 16:57:46 -0700364 DebugLoc OldDL = DbgLoc;
Dan Gohman84023e02010-07-10 09:00:22 +0000365 recomputeInsertPt();
Stephen Hines36b56882014-04-23 16:57:46 -0700366 DbgLoc = DebugLoc();
Stephen Hines37ed9c12014-12-01 14:51:49 -0800367 SavePoint SP = {OldInsertPt, OldDL};
Eric Christopher76ad43c2012-10-03 08:10:01 +0000368 return SP;
Dan Gohman84023e02010-07-10 09:00:22 +0000369}
370
Eric Christopher76ad43c2012-10-03 08:10:01 +0000371void FastISel::leaveLocalValueArea(SavePoint OldInsertPt) {
Dan Gohman84023e02010-07-10 09:00:22 +0000372 if (FuncInfo.InsertPt != FuncInfo.MBB->begin())
Stephen Hines36b56882014-04-23 16:57:46 -0700373 LastLocalValue = std::prev(FuncInfo.InsertPt);
Dan Gohman84023e02010-07-10 09:00:22 +0000374
375 // Restore the previous insert position.
Eric Christopher76ad43c2012-10-03 08:10:01 +0000376 FuncInfo.InsertPt = OldInsertPt.InsertPt;
Stephen Hines36b56882014-04-23 16:57:46 -0700377 DbgLoc = OldInsertPt.DL;
Dan Gohman84023e02010-07-10 09:00:22 +0000378}
379
Stephen Hines37ed9c12014-12-01 14:51:49 -0800380bool FastISel::selectBinaryOp(const User *I, unsigned ISDOpcode) {
Owen Andersone50ed302009-08-10 22:56:29 +0000381 EVT VT = EVT::getEVT(I->getType(), /*HandleUnknown=*/true);
Owen Anderson825b72b2009-08-11 20:47:22 +0000382 if (VT == MVT::Other || !VT.isSimple())
Dan Gohmanbdedd442008-08-20 00:11:48 +0000383 // Unhandled type. Halt "fast" selection and bail.
384 return false;
Dan Gohman638c6832008-09-05 18:44:22 +0000385
Dan Gohmanb71fea22008-08-26 20:52:40 +0000386 // We only handle legal types. For example, on x86-32 the instruction
387 // selector contains all of the 64-bit instructions from x86-64,
388 // under the assumption that i64 won't be used if the target doesn't
389 // support it.
Dan Gohman638c6832008-09-05 18:44:22 +0000390 if (!TLI.isTypeLegal(VT)) {
Owen Anderson825b72b2009-08-11 20:47:22 +0000391 // MVT::i1 is special. Allow AND, OR, or XOR because they
Dan Gohman638c6832008-09-05 18:44:22 +0000392 // don't require additional zeroing, which makes them easy.
Stephen Hines37ed9c12014-12-01 14:51:49 -0800393 if (VT == MVT::i1 && (ISDOpcode == ISD::AND || ISDOpcode == ISD::OR ||
394 ISDOpcode == ISD::XOR))
Owen Anderson23b9b192009-08-12 00:36:31 +0000395 VT = TLI.getTypeToTransformTo(I->getContext(), VT);
Dan Gohman638c6832008-09-05 18:44:22 +0000396 else
397 return false;
398 }
Dan Gohmanbdedd442008-08-20 00:11:48 +0000399
Chris Lattnerfff65b32011-04-17 01:16:47 +0000400 // Check if the first operand is a constant, and handle it as "ri". At -O0,
401 // we don't have anything that canonicalizes operand order.
Stephen Hines37ed9c12014-12-01 14:51:49 -0800402 if (const auto *CI = dyn_cast<ConstantInt>(I->getOperand(0)))
Chris Lattnerfff65b32011-04-17 01:16:47 +0000403 if (isa<Instruction>(I) && cast<Instruction>(I)->isCommutative()) {
404 unsigned Op1 = getRegForValue(I->getOperand(1));
Stephen Hines37ed9c12014-12-01 14:51:49 -0800405 if (!Op1)
406 return false;
Chris Lattnerfff65b32011-04-17 01:16:47 +0000407 bool Op1IsKill = hasTrivialKill(I->getOperand(1));
Owen Andersond74ea772011-04-22 23:38:06 +0000408
Stephen Hines37ed9c12014-12-01 14:51:49 -0800409 unsigned ResultReg =
410 fastEmit_ri_(VT.getSimpleVT(), ISDOpcode, Op1, Op1IsKill,
411 CI->getZExtValue(), VT.getSimpleVT());
412 if (!ResultReg)
413 return false;
Owen Andersond74ea772011-04-22 23:38:06 +0000414
Chris Lattner602fc062011-04-17 20:23:29 +0000415 // We successfully emitted code for the given LLVM Instruction.
Stephen Hines37ed9c12014-12-01 14:51:49 -0800416 updateValueMap(I, ResultReg);
Chris Lattner602fc062011-04-17 20:23:29 +0000417 return true;
Chris Lattnerfff65b32011-04-17 01:16:47 +0000418 }
Owen Andersond74ea772011-04-22 23:38:06 +0000419
Dan Gohman3df24e62008-09-03 23:12:08 +0000420 unsigned Op0 = getRegForValue(I->getOperand(0));
Stephen Hines37ed9c12014-12-01 14:51:49 -0800421 if (!Op0) // Unhandled operand. Halt "fast" selection and bail.
Dan Gohmand5fe57d2008-08-21 01:41:07 +0000422 return false;
Dan Gohmana6cb6412010-05-11 23:54:07 +0000423 bool Op0IsKill = hasTrivialKill(I->getOperand(0));
424
Dan Gohmand5fe57d2008-08-21 01:41:07 +0000425 // Check if the second operand is a constant and handle it appropriately.
Stephen Hines37ed9c12014-12-01 14:51:49 -0800426 if (const auto *CI = dyn_cast<ConstantInt>(I->getOperand(1))) {
Chris Lattner602fc062011-04-17 20:23:29 +0000427 uint64_t Imm = CI->getZExtValue();
Owen Andersond74ea772011-04-22 23:38:06 +0000428
Chris Lattnerf051c1a2011-04-18 07:00:40 +0000429 // Transform "sdiv exact X, 8" -> "sra X, 3".
430 if (ISDOpcode == ISD::SDIV && isa<BinaryOperator>(I) &&
Stephen Hines37ed9c12014-12-01 14:51:49 -0800431 cast<BinaryOperator>(I)->isExact() && isPowerOf2_64(Imm)) {
Chris Lattnerf051c1a2011-04-18 07:00:40 +0000432 Imm = Log2_64(Imm);
433 ISDOpcode = ISD::SRA;
434 }
Owen Andersond74ea772011-04-22 23:38:06 +0000435
Chad Rosier544b9b42012-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
Stephen Hines37ed9c12014-12-01 14:51:49 -0800443 unsigned ResultReg = fastEmit_ri_(VT.getSimpleVT(), ISDOpcode, Op0,
Chris Lattner602fc062011-04-17 20:23:29 +0000444 Op0IsKill, Imm, VT.getSimpleVT());
Stephen Hines37ed9c12014-12-01 14:51:49 -0800445 if (!ResultReg)
446 return false;
Owen Andersond74ea772011-04-22 23:38:06 +0000447
Chris Lattner602fc062011-04-17 20:23:29 +0000448 // We successfully emitted code for the given LLVM Instruction.
Stephen Hines37ed9c12014-12-01 14:51:49 -0800449 updateValueMap(I, ResultReg);
Chris Lattner602fc062011-04-17 20:23:29 +0000450 return true;
Dan Gohmand5fe57d2008-08-21 01:41:07 +0000451 }
452
Dan Gohman10df0fa2008-08-27 01:09:54 +0000453 // Check if the second operand is a constant float.
Stephen Hines37ed9c12014-12-01 14:51:49 -0800454 if (const auto *CF = dyn_cast<ConstantFP>(I->getOperand(1))) {
455 unsigned ResultReg = fastEmit_rf(VT.getSimpleVT(), VT.getSimpleVT(),
Dan Gohmana6cb6412010-05-11 23:54:07 +0000456 ISDOpcode, Op0, Op0IsKill, CF);
Stephen Hines37ed9c12014-12-01 14:51:49 -0800457 if (ResultReg) {
Dan Gohmanad368ac2008-08-27 18:10:19 +0000458 // We successfully emitted code for the given LLVM Instruction.
Stephen Hines37ed9c12014-12-01 14:51:49 -0800459 updateValueMap(I, ResultReg);
Dan Gohmanad368ac2008-08-27 18:10:19 +0000460 return true;
461 }
Dan Gohman10df0fa2008-08-27 01:09:54 +0000462 }
463
Dan Gohman3df24e62008-09-03 23:12:08 +0000464 unsigned Op1 = getRegForValue(I->getOperand(1));
Stephen Hines37ed9c12014-12-01 14:51:49 -0800465 if (!Op1) // Unhandled operand. Halt "fast" selection and bail.
Dan Gohmand5fe57d2008-08-21 01:41:07 +0000466 return false;
Dan Gohmana6cb6412010-05-11 23:54:07 +0000467 bool Op1IsKill = hasTrivialKill(I->getOperand(1));
468
Dan Gohmanad368ac2008-08-27 18:10:19 +0000469 // Now we have both operands in registers. Emit the instruction.
Stephen Hines37ed9c12014-12-01 14:51:49 -0800470 unsigned ResultReg = fastEmit_rr(VT.getSimpleVT(), VT.getSimpleVT(),
471 ISDOpcode, Op0, Op0IsKill, Op1, Op1IsKill);
472 if (!ResultReg)
Dan Gohmanbdedd442008-08-20 00:11:48 +0000473 // Target-specific code wasn't able to find a machine opcode for
474 // the given ISD opcode and type. Halt "fast" selection and bail.
475 return false;
476
Dan Gohman8014e862008-08-20 00:23:20 +0000477 // We successfully emitted code for the given LLVM Instruction.
Stephen Hines37ed9c12014-12-01 14:51:49 -0800478 updateValueMap(I, ResultReg);
Dan Gohmanbdedd442008-08-20 00:11:48 +0000479 return true;
480}
481
Stephen Hines37ed9c12014-12-01 14:51:49 -0800482bool FastISel::selectGetElementPtr(const User *I) {
Dan Gohman3df24e62008-09-03 23:12:08 +0000483 unsigned N = getRegForValue(I->getOperand(0));
Stephen Hines37ed9c12014-12-01 14:51:49 -0800484 if (!N) // Unhandled operand. Halt "fast" selection and bail.
Evan Cheng83785c82008-08-20 22:45:34 +0000485 return false;
Dan Gohmana6cb6412010-05-11 23:54:07 +0000486 bool NIsKill = hasTrivialKill(I->getOperand(0));
487
Chad Rosier478b06c2011-11-17 07:15:58 +0000488 // Keep a running tab of the total offset to coalesce multiple N = N + Offset
489 // into a single N = N + TotalOffset.
490 uint64_t TotalOffs = 0;
491 // FIXME: What's a good SWAG number for MaxOffs?
492 uint64_t MaxOffs = 2048;
Chris Lattnerdb125cf2011-07-18 04:54:35 +0000493 Type *Ty = I->getOperand(0)->getType();
Owen Anderson825b72b2009-08-11 20:47:22 +0000494 MVT VT = TLI.getPointerTy();
Stephen Hines37ed9c12014-12-01 14:51:49 -0800495 for (GetElementPtrInst::const_op_iterator OI = I->op_begin() + 1,
496 E = I->op_end();
497 OI != E; ++OI) {
Dan Gohman46510a72010-04-15 01:51:59 +0000498 const Value *Idx = *OI;
Stephen Hines37ed9c12014-12-01 14:51:49 -0800499 if (auto *StTy = dyn_cast<StructType>(Ty)) {
Evan Cheng83785c82008-08-20 22:45:34 +0000500 unsigned Field = cast<ConstantInt>(Idx)->getZExtValue();
501 if (Field) {
502 // N = N + Offset
Stephen Hines36b56882014-04-23 16:57:46 -0700503 TotalOffs += DL.getStructLayout(StTy)->getElementOffset(Field);
Chad Rosier478b06c2011-11-17 07:15:58 +0000504 if (TotalOffs >= MaxOffs) {
Stephen Hines37ed9c12014-12-01 14:51:49 -0800505 N = fastEmit_ri_(VT, ISD::ADD, N, NIsKill, TotalOffs, VT);
506 if (!N) // Unhandled operand. Halt "fast" selection and bail.
Chad Rosier478b06c2011-11-17 07:15:58 +0000507 return false;
508 NIsKill = true;
509 TotalOffs = 0;
510 }
Evan Cheng83785c82008-08-20 22:45:34 +0000511 }
512 Ty = StTy->getElementType(Field);
513 } else {
514 Ty = cast<SequentialType>(Ty)->getElementType();
515
516 // If this is a constant subscript, handle it quickly.
Stephen Hines37ed9c12014-12-01 14:51:49 -0800517 if (const auto *CI = dyn_cast<ConstantInt>(Idx)) {
518 if (CI->isZero())
519 continue;
Chad Rosier478b06c2011-11-17 07:15:58 +0000520 // N = N + Offset
Chad Rosier6016a4a2012-07-06 17:44:22 +0000521 TotalOffs +=
Stephen Hines37ed9c12014-12-01 14:51:49 -0800522 DL.getTypeAllocSize(Ty) * cast<ConstantInt>(CI)->getSExtValue();
Chad Rosier478b06c2011-11-17 07:15:58 +0000523 if (TotalOffs >= MaxOffs) {
Stephen Hines37ed9c12014-12-01 14:51:49 -0800524 N = fastEmit_ri_(VT, ISD::ADD, N, NIsKill, TotalOffs, VT);
525 if (!N) // Unhandled operand. Halt "fast" selection and bail.
Chad Rosier478b06c2011-11-17 07:15:58 +0000526 return false;
527 NIsKill = true;
528 TotalOffs = 0;
529 }
530 continue;
531 }
532 if (TotalOffs) {
Stephen Hines37ed9c12014-12-01 14:51:49 -0800533 N = fastEmit_ri_(VT, ISD::ADD, N, NIsKill, TotalOffs, VT);
534 if (!N) // Unhandled operand. Halt "fast" selection and bail.
Evan Cheng83785c82008-08-20 22:45:34 +0000535 return false;
Dan Gohmana6cb6412010-05-11 23:54:07 +0000536 NIsKill = true;
Chad Rosier478b06c2011-11-17 07:15:58 +0000537 TotalOffs = 0;
Evan Cheng83785c82008-08-20 22:45:34 +0000538 }
Wesley Peckbf17cfa2010-11-23 03:31:01 +0000539
Evan Cheng83785c82008-08-20 22:45:34 +0000540 // N = N + Idx * ElementSize;
Stephen Hines36b56882014-04-23 16:57:46 -0700541 uint64_t ElementSize = DL.getTypeAllocSize(Ty);
Dan Gohmana6cb6412010-05-11 23:54:07 +0000542 std::pair<unsigned, bool> Pair = getRegForGEPIndex(Idx);
543 unsigned IdxN = Pair.first;
544 bool IdxNIsKill = Pair.second;
Stephen Hines37ed9c12014-12-01 14:51:49 -0800545 if (!IdxN) // Unhandled operand. Halt "fast" selection and bail.
Evan Cheng83785c82008-08-20 22:45:34 +0000546 return false;
547
Dan Gohman80bc6e22008-08-26 20:57:08 +0000548 if (ElementSize != 1) {
Stephen Hines37ed9c12014-12-01 14:51:49 -0800549 IdxN = fastEmit_ri_(VT, ISD::MUL, IdxN, IdxNIsKill, ElementSize, VT);
550 if (!IdxN) // Unhandled operand. Halt "fast" selection and bail.
Dan Gohman80bc6e22008-08-26 20:57:08 +0000551 return false;
Dan Gohmana6cb6412010-05-11 23:54:07 +0000552 IdxNIsKill = true;
Dan Gohman80bc6e22008-08-26 20:57:08 +0000553 }
Stephen Hines37ed9c12014-12-01 14:51:49 -0800554 N = fastEmit_rr(VT, VT, ISD::ADD, N, NIsKill, IdxN, IdxNIsKill);
555 if (!N) // Unhandled operand. Halt "fast" selection and bail.
Evan Cheng83785c82008-08-20 22:45:34 +0000556 return false;
557 }
558 }
Chad Rosier478b06c2011-11-17 07:15:58 +0000559 if (TotalOffs) {
Stephen Hines37ed9c12014-12-01 14:51:49 -0800560 N = fastEmit_ri_(VT, ISD::ADD, N, NIsKill, TotalOffs, VT);
561 if (!N) // Unhandled operand. Halt "fast" selection and bail.
Chad Rosier478b06c2011-11-17 07:15:58 +0000562 return false;
563 }
Evan Cheng83785c82008-08-20 22:45:34 +0000564
565 // We successfully emitted code for the given LLVM Instruction.
Stephen Hines37ed9c12014-12-01 14:51:49 -0800566 updateValueMap(I, N);
Evan Cheng83785c82008-08-20 22:45:34 +0000567 return true;
Dan Gohmanbdedd442008-08-20 00:11:48 +0000568}
569
Stephen Hinesc6a4f5e2014-07-21 00:45:20 -0700570bool FastISel::addStackMapLiveVars(SmallVectorImpl<MachineOperand> &Ops,
571 const CallInst *CI, unsigned StartIdx) {
572 for (unsigned i = StartIdx, e = CI->getNumArgOperands(); i != e; ++i) {
573 Value *Val = CI->getArgOperand(i);
574 // Check for constants and encode them with a StackMaps::ConstantOp prefix.
Stephen Hines37ed9c12014-12-01 14:51:49 -0800575 if (const auto *C = dyn_cast<ConstantInt>(Val)) {
Stephen Hinesc6a4f5e2014-07-21 00:45:20 -0700576 Ops.push_back(MachineOperand::CreateImm(StackMaps::ConstantOp));
577 Ops.push_back(MachineOperand::CreateImm(C->getSExtValue()));
578 } else if (isa<ConstantPointerNull>(Val)) {
579 Ops.push_back(MachineOperand::CreateImm(StackMaps::ConstantOp));
580 Ops.push_back(MachineOperand::CreateImm(0));
581 } else if (auto *AI = dyn_cast<AllocaInst>(Val)) {
582 // Values coming from a stack location also require a sepcial encoding,
583 // but that is added later on by the target specific frame index
584 // elimination implementation.
585 auto SI = FuncInfo.StaticAllocaMap.find(AI);
586 if (SI != FuncInfo.StaticAllocaMap.end())
587 Ops.push_back(MachineOperand::CreateFI(SI->second));
588 else
589 return false;
590 } else {
591 unsigned Reg = getRegForValue(Val);
Stephen Hines37ed9c12014-12-01 14:51:49 -0800592 if (!Reg)
Stephen Hinesc6a4f5e2014-07-21 00:45:20 -0700593 return false;
594 Ops.push_back(MachineOperand::CreateReg(Reg, /*IsDef=*/false));
595 }
596 }
Stephen Hinesc6a4f5e2014-07-21 00:45:20 -0700597 return true;
598}
599
Stephen Hines37ed9c12014-12-01 14:51:49 -0800600bool FastISel::selectStackmap(const CallInst *I) {
Stephen Hinesc6a4f5e2014-07-21 00:45:20 -0700601 // void @llvm.experimental.stackmap(i64 <id>, i32 <numShadowBytes>,
602 // [live variables...])
603 assert(I->getCalledFunction()->getReturnType()->isVoidTy() &&
604 "Stackmap cannot return a value.");
605
606 // The stackmap intrinsic only records the live variables (the arguments
607 // passed to it) and emits NOPS (if requested). Unlike the patchpoint
608 // intrinsic, this won't be lowered to a function call. This means we don't
609 // have to worry about calling conventions and target-specific lowering code.
610 // Instead we perform the call lowering right here.
611 //
612 // CALLSEQ_START(0)
613 // STACKMAP(id, nbytes, ...)
614 // CALLSEQ_END(0, 0)
615 //
616 SmallVector<MachineOperand, 32> Ops;
617
618 // Add the <id> and <numBytes> constants.
619 assert(isa<ConstantInt>(I->getOperand(PatchPointOpers::IDPos)) &&
620 "Expected a constant integer.");
621 const auto *ID = cast<ConstantInt>(I->getOperand(PatchPointOpers::IDPos));
622 Ops.push_back(MachineOperand::CreateImm(ID->getZExtValue()));
623
624 assert(isa<ConstantInt>(I->getOperand(PatchPointOpers::NBytesPos)) &&
625 "Expected a constant integer.");
626 const auto *NumBytes =
Stephen Hines37ed9c12014-12-01 14:51:49 -0800627 cast<ConstantInt>(I->getOperand(PatchPointOpers::NBytesPos));
Stephen Hinesc6a4f5e2014-07-21 00:45:20 -0700628 Ops.push_back(MachineOperand::CreateImm(NumBytes->getZExtValue()));
629
630 // Push live variables for the stack map (skipping the first two arguments
631 // <id> and <numBytes>).
632 if (!addStackMapLiveVars(Ops, I, 2))
633 return false;
634
635 // We are not adding any register mask info here, because the stackmap doesn't
636 // clobber anything.
637
638 // Add scratch registers as implicit def and early clobber.
639 CallingConv::ID CC = I->getCallingConv();
640 const MCPhysReg *ScratchRegs = TLI.getScratchRegisters(CC);
641 for (unsigned i = 0; ScratchRegs[i]; ++i)
642 Ops.push_back(MachineOperand::CreateReg(
Stephen Hines37ed9c12014-12-01 14:51:49 -0800643 ScratchRegs[i], /*IsDef=*/true, /*IsImp=*/true, /*IsKill=*/false,
644 /*IsDead=*/false, /*IsUndef=*/false, /*IsEarlyClobber=*/true));
Stephen Hinesc6a4f5e2014-07-21 00:45:20 -0700645
646 // Issue CALLSEQ_START
647 unsigned AdjStackDown = TII.getCallFrameSetupOpcode();
648 BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc, TII.get(AdjStackDown))
Stephen Hines37ed9c12014-12-01 14:51:49 -0800649 .addImm(0);
Stephen Hinesc6a4f5e2014-07-21 00:45:20 -0700650
651 // Issue STACKMAP.
652 MachineInstrBuilder MIB = BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc,
653 TII.get(TargetOpcode::STACKMAP));
654 for (auto const &MO : Ops)
655 MIB.addOperand(MO);
656
657 // Issue CALLSEQ_END
658 unsigned AdjStackUp = TII.getCallFrameDestroyOpcode();
659 BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc, TII.get(AdjStackUp))
Stephen Hines37ed9c12014-12-01 14:51:49 -0800660 .addImm(0)
661 .addImm(0);
Stephen Hinesc6a4f5e2014-07-21 00:45:20 -0700662
663 // Inform the Frame Information that we have a stackmap in this function.
664 FuncInfo.MF->getFrameInfo()->setHasStackMap();
665
666 return true;
667}
668
Stephen Hines37ed9c12014-12-01 14:51:49 -0800669/// \brief Lower an argument list according to the target calling convention.
670///
671/// This is a helper for lowering intrinsics that follow a target calling
672/// convention or require stack pointer adjustment. Only a subset of the
673/// intrinsic's operands need to participate in the calling convention.
674bool FastISel::lowerCallOperands(const CallInst *CI, unsigned ArgIdx,
675 unsigned NumArgs, const Value *Callee,
676 bool ForceRetVoidTy, CallLoweringInfo &CLI) {
677 ArgListTy Args;
678 Args.reserve(NumArgs);
679
680 // Populate the argument list.
681 // Attributes for args start at offset 1, after the return attribute.
682 ImmutableCallSite CS(CI);
683 for (unsigned ArgI = ArgIdx, ArgE = ArgIdx + NumArgs, AttrI = ArgIdx + 1;
684 ArgI != ArgE; ++ArgI) {
685 Value *V = CI->getOperand(ArgI);
686
687 assert(!V->getType()->isEmptyTy() && "Empty type passed to intrinsic.");
688
689 ArgListEntry Entry;
690 Entry.Val = V;
691 Entry.Ty = V->getType();
692 Entry.setAttributes(&CS, AttrI);
693 Args.push_back(Entry);
694 }
695
696 Type *RetTy = ForceRetVoidTy ? Type::getVoidTy(CI->getType()->getContext())
697 : CI->getType();
698 CLI.setCallee(CI->getCallingConv(), RetTy, Callee, std::move(Args), NumArgs);
699
700 return lowerCallTo(CLI);
701}
702
703bool FastISel::selectPatchpoint(const CallInst *I) {
704 // void|i64 @llvm.experimental.patchpoint.void|i64(i64 <id>,
705 // i32 <numBytes>,
706 // i8* <target>,
707 // i32 <numArgs>,
708 // [Args...],
709 // [live variables...])
710 CallingConv::ID CC = I->getCallingConv();
711 bool IsAnyRegCC = CC == CallingConv::AnyReg;
712 bool HasDef = !I->getType()->isVoidTy();
713 Value *Callee = I->getOperand(PatchPointOpers::TargetPos);
714
715 // Get the real number of arguments participating in the call <numArgs>
716 assert(isa<ConstantInt>(I->getOperand(PatchPointOpers::NArgPos)) &&
717 "Expected a constant integer.");
718 const auto *NumArgsVal =
719 cast<ConstantInt>(I->getOperand(PatchPointOpers::NArgPos));
720 unsigned NumArgs = NumArgsVal->getZExtValue();
721
722 // Skip the four meta args: <id>, <numNopBytes>, <target>, <numArgs>
723 // This includes all meta-operands up to but not including CC.
724 unsigned NumMetaOpers = PatchPointOpers::CCPos;
725 assert(I->getNumArgOperands() >= NumMetaOpers + NumArgs &&
726 "Not enough arguments provided to the patchpoint intrinsic");
727
728 // For AnyRegCC the arguments are lowered later on manually.
729 unsigned NumCallArgs = IsAnyRegCC ? 0 : NumArgs;
730 CallLoweringInfo CLI;
731 if (!lowerCallOperands(I, NumMetaOpers, NumCallArgs, Callee, IsAnyRegCC, CLI))
732 return false;
733
734 assert(CLI.Call && "No call instruction specified.");
735
736 SmallVector<MachineOperand, 32> Ops;
737
738 // Add an explicit result reg if we use the anyreg calling convention.
739 if (IsAnyRegCC && HasDef) {
740 assert(CLI.NumResultRegs == 0 && "Unexpected result register.");
741 CLI.ResultReg = createResultReg(TLI.getRegClassFor(MVT::i64));
742 CLI.NumResultRegs = 1;
743 Ops.push_back(MachineOperand::CreateReg(CLI.ResultReg, /*IsDef=*/true));
744 }
745
746 // Add the <id> and <numBytes> constants.
747 assert(isa<ConstantInt>(I->getOperand(PatchPointOpers::IDPos)) &&
748 "Expected a constant integer.");
749 const auto *ID = cast<ConstantInt>(I->getOperand(PatchPointOpers::IDPos));
750 Ops.push_back(MachineOperand::CreateImm(ID->getZExtValue()));
751
752 assert(isa<ConstantInt>(I->getOperand(PatchPointOpers::NBytesPos)) &&
753 "Expected a constant integer.");
754 const auto *NumBytes =
755 cast<ConstantInt>(I->getOperand(PatchPointOpers::NBytesPos));
756 Ops.push_back(MachineOperand::CreateImm(NumBytes->getZExtValue()));
757
758 // Assume that the callee is a constant address or null pointer.
759 // FIXME: handle function symbols in the future.
760 uint64_t CalleeAddr;
761 if (const auto *C = dyn_cast<IntToPtrInst>(Callee))
762 CalleeAddr = cast<ConstantInt>(C->getOperand(0))->getZExtValue();
763 else if (const auto *C = dyn_cast<ConstantExpr>(Callee)) {
764 if (C->getOpcode() == Instruction::IntToPtr)
765 CalleeAddr = cast<ConstantInt>(C->getOperand(0))->getZExtValue();
766 else
767 llvm_unreachable("Unsupported ConstantExpr.");
768 } else if (isa<ConstantPointerNull>(Callee))
769 CalleeAddr = 0;
770 else
771 llvm_unreachable("Unsupported callee address.");
772
773 Ops.push_back(MachineOperand::CreateImm(CalleeAddr));
774
775 // Adjust <numArgs> to account for any arguments that have been passed on
776 // the stack instead.
777 unsigned NumCallRegArgs = IsAnyRegCC ? NumArgs : CLI.OutRegs.size();
778 Ops.push_back(MachineOperand::CreateImm(NumCallRegArgs));
779
780 // Add the calling convention
781 Ops.push_back(MachineOperand::CreateImm((unsigned)CC));
782
783 // Add the arguments we omitted previously. The register allocator should
784 // place these in any free register.
785 if (IsAnyRegCC) {
786 for (unsigned i = NumMetaOpers, e = NumMetaOpers + NumArgs; i != e; ++i) {
787 unsigned Reg = getRegForValue(I->getArgOperand(i));
788 if (!Reg)
789 return false;
790 Ops.push_back(MachineOperand::CreateReg(Reg, /*IsDef=*/false));
791 }
792 }
793
794 // Push the arguments from the call instruction.
795 for (auto Reg : CLI.OutRegs)
796 Ops.push_back(MachineOperand::CreateReg(Reg, /*IsDef=*/false));
797
798 // Push live variables for the stack map.
799 if (!addStackMapLiveVars(Ops, I, NumMetaOpers + NumArgs))
800 return false;
801
802 // Push the register mask info.
803 Ops.push_back(MachineOperand::CreateRegMask(TRI.getCallPreservedMask(CC)));
804
805 // Add scratch registers as implicit def and early clobber.
806 const MCPhysReg *ScratchRegs = TLI.getScratchRegisters(CC);
807 for (unsigned i = 0; ScratchRegs[i]; ++i)
808 Ops.push_back(MachineOperand::CreateReg(
809 ScratchRegs[i], /*IsDef=*/true, /*IsImp=*/true, /*IsKill=*/false,
810 /*IsDead=*/false, /*IsUndef=*/false, /*IsEarlyClobber=*/true));
811
812 // Add implicit defs (return values).
813 for (auto Reg : CLI.InRegs)
814 Ops.push_back(MachineOperand::CreateReg(Reg, /*IsDef=*/true,
815 /*IsImpl=*/true));
816
817 // Insert the patchpoint instruction before the call generated by the target.
818 MachineInstrBuilder MIB = BuildMI(*FuncInfo.MBB, CLI.Call, DbgLoc,
819 TII.get(TargetOpcode::PATCHPOINT));
820
821 for (auto &MO : Ops)
822 MIB.addOperand(MO);
823
824 MIB->setPhysRegsDeadExcept(CLI.InRegs, TRI);
825
826 // Delete the original call instruction.
827 CLI.Call->eraseFromParent();
828
829 // Inform the Frame Information that we have a patchpoint in this function.
830 FuncInfo.MF->getFrameInfo()->setHasPatchPoint();
831
832 if (CLI.NumResultRegs)
833 updateValueMap(I, CLI.ResultReg, CLI.NumResultRegs);
834 return true;
835}
836
837/// Returns an AttributeSet representing the attributes applied to the return
838/// value of the given call.
839static AttributeSet getReturnAttrs(FastISel::CallLoweringInfo &CLI) {
840 SmallVector<Attribute::AttrKind, 2> Attrs;
841 if (CLI.RetSExt)
842 Attrs.push_back(Attribute::SExt);
843 if (CLI.RetZExt)
844 Attrs.push_back(Attribute::ZExt);
845 if (CLI.IsInReg)
846 Attrs.push_back(Attribute::InReg);
847
848 return AttributeSet::get(CLI.RetTy->getContext(), AttributeSet::ReturnIndex,
849 Attrs);
850}
851
852bool FastISel::lowerCallTo(const CallInst *CI, const char *SymName,
853 unsigned NumArgs) {
854 ImmutableCallSite CS(CI);
855
856 PointerType *PT = cast<PointerType>(CS.getCalledValue()->getType());
857 FunctionType *FTy = cast<FunctionType>(PT->getElementType());
858 Type *RetTy = FTy->getReturnType();
859
860 ArgListTy Args;
861 Args.reserve(NumArgs);
862
863 // Populate the argument list.
864 // Attributes for args start at offset 1, after the return attribute.
865 for (unsigned ArgI = 0; ArgI != NumArgs; ++ArgI) {
866 Value *V = CI->getOperand(ArgI);
867
868 assert(!V->getType()->isEmptyTy() && "Empty type passed to intrinsic.");
869
870 ArgListEntry Entry;
871 Entry.Val = V;
872 Entry.Ty = V->getType();
873 Entry.setAttributes(&CS, ArgI + 1);
874 Args.push_back(Entry);
875 }
876
877 CallLoweringInfo CLI;
878 CLI.setCallee(RetTy, FTy, SymName, std::move(Args), CS, NumArgs);
879
880 return lowerCallTo(CLI);
881}
882
883bool FastISel::lowerCallTo(CallLoweringInfo &CLI) {
884 // Handle the incoming return values from the call.
885 CLI.clearIns();
886 SmallVector<EVT, 4> RetTys;
887 ComputeValueVTs(TLI, CLI.RetTy, RetTys);
888
889 SmallVector<ISD::OutputArg, 4> Outs;
890 GetReturnInfo(CLI.RetTy, getReturnAttrs(CLI), Outs, TLI);
891
892 bool CanLowerReturn = TLI.CanLowerReturn(
893 CLI.CallConv, *FuncInfo.MF, CLI.IsVarArg, Outs, CLI.RetTy->getContext());
894
895 // FIXME: sret demotion isn't supported yet - bail out.
896 if (!CanLowerReturn)
897 return false;
898
899 for (unsigned I = 0, E = RetTys.size(); I != E; ++I) {
900 EVT VT = RetTys[I];
901 MVT RegisterVT = TLI.getRegisterType(CLI.RetTy->getContext(), VT);
902 unsigned NumRegs = TLI.getNumRegisters(CLI.RetTy->getContext(), VT);
903 for (unsigned i = 0; i != NumRegs; ++i) {
904 ISD::InputArg MyFlags;
905 MyFlags.VT = RegisterVT;
906 MyFlags.ArgVT = VT;
907 MyFlags.Used = CLI.IsReturnValueUsed;
908 if (CLI.RetSExt)
909 MyFlags.Flags.setSExt();
910 if (CLI.RetZExt)
911 MyFlags.Flags.setZExt();
912 if (CLI.IsInReg)
913 MyFlags.Flags.setInReg();
914 CLI.Ins.push_back(MyFlags);
915 }
916 }
917
918 // Handle all of the outgoing arguments.
919 CLI.clearOuts();
920 for (auto &Arg : CLI.getArgs()) {
921 Type *FinalType = Arg.Ty;
922 if (Arg.IsByVal)
923 FinalType = cast<PointerType>(Arg.Ty)->getElementType();
924 bool NeedsRegBlock = TLI.functionArgumentNeedsConsecutiveRegisters(
925 FinalType, CLI.CallConv, CLI.IsVarArg);
926
927 ISD::ArgFlagsTy Flags;
928 if (Arg.IsZExt)
929 Flags.setZExt();
930 if (Arg.IsSExt)
931 Flags.setSExt();
932 if (Arg.IsInReg)
933 Flags.setInReg();
934 if (Arg.IsSRet)
935 Flags.setSRet();
936 if (Arg.IsByVal)
937 Flags.setByVal();
938 if (Arg.IsInAlloca) {
939 Flags.setInAlloca();
940 // Set the byval flag for CCAssignFn callbacks that don't know about
941 // inalloca. This way we can know how many bytes we should've allocated
942 // and how many bytes a callee cleanup function will pop. If we port
943 // inalloca to more targets, we'll have to add custom inalloca handling in
944 // the various CC lowering callbacks.
945 Flags.setByVal();
946 }
947 if (Arg.IsByVal || Arg.IsInAlloca) {
948 PointerType *Ty = cast<PointerType>(Arg.Ty);
949 Type *ElementTy = Ty->getElementType();
950 unsigned FrameSize = DL.getTypeAllocSize(ElementTy);
951 // For ByVal, alignment should come from FE. BE will guess if this info is
952 // not there, but there are cases it cannot get right.
953 unsigned FrameAlign = Arg.Alignment;
954 if (!FrameAlign)
955 FrameAlign = TLI.getByValTypeAlignment(ElementTy);
956 Flags.setByValSize(FrameSize);
957 Flags.setByValAlign(FrameAlign);
958 }
959 if (Arg.IsNest)
960 Flags.setNest();
961 if (NeedsRegBlock)
962 Flags.setInConsecutiveRegs();
963 unsigned OriginalAlignment = DL.getABITypeAlignment(Arg.Ty);
964 Flags.setOrigAlign(OriginalAlignment);
965
966 CLI.OutVals.push_back(Arg.Val);
967 CLI.OutFlags.push_back(Flags);
968 }
969
970 if (!fastLowerCall(CLI))
971 return false;
972
973 // Set all unused physreg defs as dead.
974 assert(CLI.Call && "No call instruction specified.");
975 CLI.Call->setPhysRegsDeadExcept(CLI.InRegs, TRI);
976
977 if (CLI.NumResultRegs && CLI.CS)
978 updateValueMap(CLI.CS->getInstruction(), CLI.ResultReg, CLI.NumResultRegs);
979
980 return true;
981}
982
983bool FastISel::lowerCall(const CallInst *CI) {
984 ImmutableCallSite CS(CI);
985
986 PointerType *PT = cast<PointerType>(CS.getCalledValue()->getType());
987 FunctionType *FuncTy = cast<FunctionType>(PT->getElementType());
988 Type *RetTy = FuncTy->getReturnType();
989
990 ArgListTy Args;
991 ArgListEntry Entry;
992 Args.reserve(CS.arg_size());
993
994 for (ImmutableCallSite::arg_iterator i = CS.arg_begin(), e = CS.arg_end();
995 i != e; ++i) {
996 Value *V = *i;
997
998 // Skip empty types
999 if (V->getType()->isEmptyTy())
1000 continue;
1001
1002 Entry.Val = V;
1003 Entry.Ty = V->getType();
1004
1005 // Skip the first return-type Attribute to get to params.
1006 Entry.setAttributes(&CS, i - CS.arg_begin() + 1);
1007 Args.push_back(Entry);
1008 }
1009
1010 // Check if target-independent constraints permit a tail call here.
1011 // Target-dependent constraints are checked within fastLowerCall.
1012 bool IsTailCall = CI->isTailCall();
1013 if (IsTailCall && !isInTailCallPosition(CS, TM))
1014 IsTailCall = false;
1015
1016 CallLoweringInfo CLI;
1017 CLI.setCallee(RetTy, FuncTy, CI->getCalledValue(), std::move(Args), CS)
1018 .setTailCall(IsTailCall);
1019
1020 return lowerCallTo(CLI);
1021}
1022
1023bool FastISel::selectCall(const User *I) {
Dan Gohmana61e73b2011-04-26 17:18:34 +00001024 const CallInst *Call = cast<CallInst>(I);
1025
1026 // Handle simple inline asms.
Dan Gohman9e15d652011-10-12 15:56:56 +00001027 if (const InlineAsm *IA = dyn_cast<InlineAsm>(Call->getCalledValue())) {
Stephen Hines37ed9c12014-12-01 14:51:49 -08001028 // If the inline asm has side effects, then make sure that no local value
1029 // lives across by flushing the local value map.
1030 if (IA->hasSideEffects())
1031 flushLocalValueMap();
1032
Dan Gohmana61e73b2011-04-26 17:18:34 +00001033 // Don't attempt to handle constraints.
1034 if (!IA->getConstraintString().empty())
1035 return false;
1036
1037 unsigned ExtraInfo = 0;
1038 if (IA->hasSideEffects())
1039 ExtraInfo |= InlineAsm::Extra_HasSideEffects;
1040 if (IA->isAlignStack())
1041 ExtraInfo |= InlineAsm::Extra_IsAlignStack;
1042
Stephen Hines36b56882014-04-23 16:57:46 -07001043 BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc,
Dan Gohmana61e73b2011-04-26 17:18:34 +00001044 TII.get(TargetOpcode::INLINEASM))
Stephen Hines37ed9c12014-12-01 14:51:49 -08001045 .addExternalSymbol(IA->getAsmString().c_str())
1046 .addImm(ExtraInfo);
Dan Gohmana61e73b2011-04-26 17:18:34 +00001047 return true;
1048 }
1049
Michael J. Spencerc9c137b2012-02-22 19:06:13 +00001050 MachineModuleInfo &MMI = FuncInfo.MF->getMMI();
1051 ComputeUsesVAFloatArgument(*Call, &MMI);
1052
Stephen Hines37ed9c12014-12-01 14:51:49 -08001053 // Handle intrinsic function calls.
1054 if (const auto *II = dyn_cast<IntrinsicInst>(Call))
1055 return selectIntrinsicCall(II);
Dan Gohman33134c42008-09-25 17:05:24 +00001056
Stephen Hines37ed9c12014-12-01 14:51:49 -08001057 // Usually, it does not make sense to initialize a value,
1058 // make an unrelated function call and use the value, because
1059 // it tends to be spilled on the stack. So, we move the pointer
1060 // to the last local value to the beginning of the block, so that
1061 // all the values which have already been materialized,
1062 // appear after the call. It also makes sense to skip intrinsics
1063 // since they tend to be inlined.
1064 flushLocalValueMap();
1065
1066 return lowerCall(Call);
1067}
1068
1069bool FastISel::selectIntrinsicCall(const IntrinsicInst *II) {
1070 switch (II->getIntrinsicID()) {
1071 default:
1072 break;
1073 // At -O0 we don't care about the lifetime intrinsics.
Eric Christopher9b5d6b82012-02-17 23:03:39 +00001074 case Intrinsic::lifetime_start:
1075 case Intrinsic::lifetime_end:
Stephen Hines37ed9c12014-12-01 14:51:49 -08001076 // The donothing intrinsic does, well, nothing.
Chad Rosierfd065bb2012-07-06 17:33:39 +00001077 case Intrinsic::donothing:
Eric Christopher9b5d6b82012-02-17 23:03:39 +00001078 return true;
Bill Wendling92c1e122009-02-13 02:16:35 +00001079 case Intrinsic::dbg_declare: {
Stephen Hines37ed9c12014-12-01 14:51:49 -08001080 const DbgDeclareInst *DI = cast<DbgDeclareInst>(II);
Manman Rencbafae62013-06-28 05:43:10 +00001081 DIVariable DIVar(DI->getVariable());
Stephen Lin155615d2013-07-08 00:37:03 +00001082 assert((!DIVar || DIVar.isVariable()) &&
Stephen Hines37ed9c12014-12-01 14:51:49 -08001083 "Variable in DbgDeclareInst should be either null or a DIVariable.");
1084 if (!DIVar || !FuncInfo.MF->getMMI().hasDebugInfo()) {
Eric Christopherbb54d212012-03-15 21:33:44 +00001085 DEBUG(dbgs() << "Dropping debug info for " << *DI << "\n");
Devang Patel7e1e31f2009-07-02 22:43:26 +00001086 return true;
Eric Christopherbb54d212012-03-15 21:33:44 +00001087 }
Devang Patel7e1e31f2009-07-02 22:43:26 +00001088
Dan Gohman46510a72010-04-15 01:51:59 +00001089 const Value *Address = DI->getAddress();
Eric Christopherccaea7d2012-03-15 21:33:47 +00001090 if (!Address || isa<UndefValue>(Address)) {
Eric Christopherbb54d212012-03-15 21:33:44 +00001091 DEBUG(dbgs() << "Dropping debug info for " << *DI << "\n");
Dale Johannesendc918562010-02-06 02:26:02 +00001092 return true;
Eric Christopherbb54d212012-03-15 21:33:44 +00001093 }
Devang Patel6fe75aa2010-09-14 20:29:31 +00001094
Adrian Prantl35176402013-07-09 20:28:37 +00001095 unsigned Offset = 0;
David Blaikie6d9dbd52013-06-16 20:34:15 +00001096 Optional<MachineOperand> Op;
Stephen Hines37ed9c12014-12-01 14:51:49 -08001097 if (const auto *Arg = dyn_cast<Argument>(Address))
Devang Patel9aee3352011-09-08 22:59:09 +00001098 // Some arguments' frame index is recorded during argument lowering.
Adrian Prantl35176402013-07-09 20:28:37 +00001099 Offset = FuncInfo.getArgumentFrameIndex(Arg);
1100 if (Offset)
Stephen Hines37ed9c12014-12-01 14:51:49 -08001101 Op = MachineOperand::CreateFI(Offset);
David Blaikie6d9dbd52013-06-16 20:34:15 +00001102 if (!Op)
1103 if (unsigned Reg = lookUpRegForValue(Address))
1104 Op = MachineOperand::CreateReg(Reg, false);
Eric Christopher8c5293c2012-03-20 01:07:58 +00001105
Bill Wendling84364a42012-03-30 00:02:55 +00001106 // If we have a VLA that has a "use" in a metadata node that's then used
1107 // here but it has no other uses, then we have a problem. E.g.,
1108 //
1109 // int foo (const int *x) {
1110 // char a[*x];
1111 // return 0;
1112 // }
1113 //
1114 // If we assign 'a' a vreg and fast isel later on has to use the selection
1115 // DAG isel, it will want to copy the value to the vreg. However, there are
1116 // no uses, which goes counter to what selection DAG isel expects.
David Blaikie6d9dbd52013-06-16 20:34:15 +00001117 if (!Op && !Address->use_empty() && isa<Instruction>(Address) &&
Eric Christopher8c5293c2012-03-20 01:07:58 +00001118 (!isa<AllocaInst>(Address) ||
1119 !FuncInfo.StaticAllocaMap.count(cast<AllocaInst>(Address))))
David Blaikie6d9dbd52013-06-16 20:34:15 +00001120 Op = MachineOperand::CreateReg(FuncInfo.InitializeRegForValue(Address),
Adrian Prantl0a4371a2013-09-18 22:08:59 +00001121 false);
Wesley Peckbf17cfa2010-11-23 03:31:01 +00001122
Adrian Prantl0a4371a2013-09-18 22:08:59 +00001123 if (Op) {
Adrian Prantl35176402013-07-09 20:28:37 +00001124 if (Op->isReg()) {
1125 Op->setIsDebug(true);
Stephen Hines36b56882014-04-23 16:57:46 -07001126 BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc,
David Blaikie54de36b2013-10-14 20:15:04 +00001127 TII.get(TargetOpcode::DBG_VALUE), false, Op->getReg(), 0,
Stephen Hines37ed9c12014-12-01 14:51:49 -08001128 DI->getVariable(), DI->getExpression());
David Blaikie54de36b2013-10-14 20:15:04 +00001129 } else
Stephen Hines36b56882014-04-23 16:57:46 -07001130 BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc,
David Blaikie54de36b2013-10-14 20:15:04 +00001131 TII.get(TargetOpcode::DBG_VALUE))
1132 .addOperand(*Op)
1133 .addImm(0)
Stephen Hines37ed9c12014-12-01 14:51:49 -08001134 .addMetadata(DI->getVariable())
1135 .addMetadata(DI->getExpression());
Adrian Prantl0a4371a2013-09-18 22:08:59 +00001136 } else {
Eric Christopher4476bae2012-03-20 01:07:53 +00001137 // We can't yet handle anything else here because it would require
1138 // generating code, thus altering codegen because of debug info.
Adrian Prantl5da4e4f2013-05-22 18:02:19 +00001139 DEBUG(dbgs() << "Dropping debug info for " << *DI << "\n");
Adrian Prantl0a4371a2013-09-18 22:08:59 +00001140 }
Dan Gohman33134c42008-09-25 17:05:24 +00001141 return true;
Bill Wendling92c1e122009-02-13 02:16:35 +00001142 }
Dale Johannesen45df7612010-02-26 20:01:55 +00001143 case Intrinsic::dbg_value: {
Dale Johannesen343b42e2010-04-07 01:15:14 +00001144 // This form of DBG_VALUE is target-independent.
Stephen Hines37ed9c12014-12-01 14:51:49 -08001145 const DbgValueInst *DI = cast<DbgValueInst>(II);
Evan Chenge837dea2011-06-28 19:10:37 +00001146 const MCInstrDesc &II = TII.get(TargetOpcode::DBG_VALUE);
Dan Gohman46510a72010-04-15 01:51:59 +00001147 const Value *V = DI->getValue();
Dale Johannesen45df7612010-02-26 20:01:55 +00001148 if (!V) {
1149 // Currently the optimizer can produce this; insert an undef to
1150 // help debugging. Probably the optimizer should not do this.
Stephen Hines36b56882014-04-23 16:57:46 -07001151 BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc, II)
Stephen Hines37ed9c12014-12-01 14:51:49 -08001152 .addReg(0U)
1153 .addImm(DI->getOffset())
1154 .addMetadata(DI->getVariable())
1155 .addMetadata(DI->getExpression());
1156 } else if (const auto *CI = dyn_cast<ConstantInt>(V)) {
Devang Patel8594d422011-06-24 20:46:11 +00001157 if (CI->getBitWidth() > 64)
Stephen Hines36b56882014-04-23 16:57:46 -07001158 BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc, II)
Stephen Hines37ed9c12014-12-01 14:51:49 -08001159 .addCImm(CI)
1160 .addImm(DI->getOffset())
1161 .addMetadata(DI->getVariable())
1162 .addMetadata(DI->getExpression());
Chad Rosier6016a4a2012-07-06 17:44:22 +00001163 else
Stephen Hines36b56882014-04-23 16:57:46 -07001164 BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc, II)
Stephen Hines37ed9c12014-12-01 14:51:49 -08001165 .addImm(CI->getZExtValue())
1166 .addImm(DI->getOffset())
1167 .addMetadata(DI->getVariable())
1168 .addMetadata(DI->getExpression());
1169 } else if (const auto *CF = dyn_cast<ConstantFP>(V)) {
Stephen Hines36b56882014-04-23 16:57:46 -07001170 BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc, II)
Stephen Hines37ed9c12014-12-01 14:51:49 -08001171 .addFPImm(CF)
1172 .addImm(DI->getOffset())
1173 .addMetadata(DI->getVariable())
1174 .addMetadata(DI->getExpression());
Dale Johannesen45df7612010-02-26 20:01:55 +00001175 } else if (unsigned Reg = lookUpRegForValue(V)) {
Adrian Prantl818833f2013-09-16 23:29:03 +00001176 // FIXME: This does not handle register-indirect values at offset 0.
Adrian Prantl35176402013-07-09 20:28:37 +00001177 bool IsIndirect = DI->getOffset() != 0;
Stephen Hines37ed9c12014-12-01 14:51:49 -08001178 BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc, II, IsIndirect, Reg,
1179 DI->getOffset(), DI->getVariable(), DI->getExpression());
Dale Johannesen45df7612010-02-26 20:01:55 +00001180 } else {
1181 // We can't yet handle anything else here because it would require
1182 // generating code, thus altering codegen because of debug info.
Adrian Prantl5da4e4f2013-05-22 18:02:19 +00001183 DEBUG(dbgs() << "Dropping debug info for " << *DI << "\n");
Wesley Peckbf17cfa2010-11-23 03:31:01 +00001184 }
Dale Johannesen45df7612010-02-26 20:01:55 +00001185 return true;
1186 }
Eli Friedmand0118a22011-05-14 00:47:51 +00001187 case Intrinsic::objectsize: {
Stephen Hines37ed9c12014-12-01 14:51:49 -08001188 ConstantInt *CI = cast<ConstantInt>(II->getArgOperand(1));
Eli Friedmand0118a22011-05-14 00:47:51 +00001189 unsigned long long Res = CI->isZero() ? -1ULL : 0;
Stephen Hines37ed9c12014-12-01 14:51:49 -08001190 Constant *ResCI = ConstantInt::get(II->getType(), Res);
Eli Friedmand0118a22011-05-14 00:47:51 +00001191 unsigned ResultReg = getRegForValue(ResCI);
Stephen Hines37ed9c12014-12-01 14:51:49 -08001192 if (!ResultReg)
Eli Friedmand0118a22011-05-14 00:47:51 +00001193 return false;
Stephen Hines37ed9c12014-12-01 14:51:49 -08001194 updateValueMap(II, ResultReg);
Eli Friedmand0118a22011-05-14 00:47:51 +00001195 return true;
1196 }
Chad Rosier33947b42013-03-07 20:42:17 +00001197 case Intrinsic::expect: {
Stephen Hines37ed9c12014-12-01 14:51:49 -08001198 unsigned ResultReg = getRegForValue(II->getArgOperand(0));
1199 if (!ResultReg)
Nick Lewycky33cdfe92013-03-11 21:44:37 +00001200 return false;
Stephen Hines37ed9c12014-12-01 14:51:49 -08001201 updateValueMap(II, ResultReg);
Chad Rosier4fde76d2013-03-07 21:38:33 +00001202 return true;
Chad Rosier33947b42013-03-07 20:42:17 +00001203 }
Stephen Hinesc6a4f5e2014-07-21 00:45:20 -07001204 case Intrinsic::experimental_stackmap:
Stephen Hines37ed9c12014-12-01 14:51:49 -08001205 return selectStackmap(II);
1206 case Intrinsic::experimental_patchpoint_void:
1207 case Intrinsic::experimental_patchpoint_i64:
1208 return selectPatchpoint(II);
Dan Gohman33134c42008-09-25 17:05:24 +00001209 }
Dan Gohman4183e312010-04-13 17:07:06 +00001210
Stephen Hines37ed9c12014-12-01 14:51:49 -08001211 return fastLowerIntrinsicCall(II);
Dan Gohman33134c42008-09-25 17:05:24 +00001212}
1213
Stephen Hines37ed9c12014-12-01 14:51:49 -08001214bool FastISel::selectCast(const User *I, unsigned Opcode) {
Owen Andersone50ed302009-08-10 22:56:29 +00001215 EVT SrcVT = TLI.getValueType(I->getOperand(0)->getType());
1216 EVT DstVT = TLI.getValueType(I->getType());
Wesley Peckbf17cfa2010-11-23 03:31:01 +00001217
Stephen Hines37ed9c12014-12-01 14:51:49 -08001218 if (SrcVT == MVT::Other || !SrcVT.isSimple() || DstVT == MVT::Other ||
1219 !DstVT.isSimple())
Owen Andersond0533c92008-08-26 23:46:32 +00001220 // Unhandled type. Halt "fast" selection and bail.
1221 return false;
Wesley Peckbf17cfa2010-11-23 03:31:01 +00001222
Eli Friedman76927d732011-05-25 23:49:02 +00001223 // Check if the destination type is legal.
Dan Gohman474d3b32009-03-13 23:53:06 +00001224 if (!TLI.isTypeLegal(DstVT))
Eli Friedman76927d732011-05-25 23:49:02 +00001225 return false;
Dan Gohman474d3b32009-03-13 23:53:06 +00001226
Eli Friedman76927d732011-05-25 23:49:02 +00001227 // Check if the source operand is legal.
Dan Gohman474d3b32009-03-13 23:53:06 +00001228 if (!TLI.isTypeLegal(SrcVT))
Eli Friedman76927d732011-05-25 23:49:02 +00001229 return false;
Dan Gohman474d3b32009-03-13 23:53:06 +00001230
Dan Gohman3df24e62008-09-03 23:12:08 +00001231 unsigned InputReg = getRegForValue(I->getOperand(0));
Owen Andersond0533c92008-08-26 23:46:32 +00001232 if (!InputReg)
1233 // Unhandled operand. Halt "fast" selection and bail.
1234 return false;
Dan Gohman14ea1ec2009-03-13 20:42:20 +00001235
Dan Gohmana6cb6412010-05-11 23:54:07 +00001236 bool InputRegIsKill = hasTrivialKill(I->getOperand(0));
1237
Stephen Hines37ed9c12014-12-01 14:51:49 -08001238 unsigned ResultReg = fastEmit_r(SrcVT.getSimpleVT(), DstVT.getSimpleVT(),
1239 Opcode, InputReg, InputRegIsKill);
Owen Andersond0533c92008-08-26 23:46:32 +00001240 if (!ResultReg)
1241 return false;
Wesley Peckbf17cfa2010-11-23 03:31:01 +00001242
Stephen Hines37ed9c12014-12-01 14:51:49 -08001243 updateValueMap(I, ResultReg);
Owen Andersond0533c92008-08-26 23:46:32 +00001244 return true;
1245}
1246
Stephen Hines37ed9c12014-12-01 14:51:49 -08001247bool FastISel::selectBitCast(const User *I) {
Dan Gohmanad368ac2008-08-27 18:10:19 +00001248 // If the bitcast doesn't change the type, just use the operand value.
1249 if (I->getType() == I->getOperand(0)->getType()) {
Dan Gohman3df24e62008-09-03 23:12:08 +00001250 unsigned Reg = getRegForValue(I->getOperand(0));
Stephen Hines37ed9c12014-12-01 14:51:49 -08001251 if (!Reg)
Dan Gohmana318dab2008-08-27 20:41:38 +00001252 return false;
Stephen Hines37ed9c12014-12-01 14:51:49 -08001253 updateValueMap(I, Reg);
Dan Gohmanad368ac2008-08-27 18:10:19 +00001254 return true;
1255 }
1256
Wesley Peckbf17cfa2010-11-23 03:31:01 +00001257 // Bitcasts of other values become reg-reg copies or BITCAST operators.
Patrik Hagglund3d170e62012-12-17 14:30:06 +00001258 EVT SrcEVT = TLI.getValueType(I->getOperand(0)->getType());
1259 EVT DstEVT = TLI.getValueType(I->getType());
1260 if (SrcEVT == MVT::Other || DstEVT == MVT::Other ||
1261 !TLI.isTypeLegal(SrcEVT) || !TLI.isTypeLegal(DstEVT))
Owen Andersond0533c92008-08-26 23:46:32 +00001262 // Unhandled type. Halt "fast" selection and bail.
1263 return false;
Wesley Peckbf17cfa2010-11-23 03:31:01 +00001264
Patrik Hagglund3d170e62012-12-17 14:30:06 +00001265 MVT SrcVT = SrcEVT.getSimpleVT();
1266 MVT DstVT = DstEVT.getSimpleVT();
Dan Gohman3df24e62008-09-03 23:12:08 +00001267 unsigned Op0 = getRegForValue(I->getOperand(0));
Stephen Hines37ed9c12014-12-01 14:51:49 -08001268 if (!Op0) // Unhandled operand. Halt "fast" selection and bail.
Owen Andersond0533c92008-08-26 23:46:32 +00001269 return false;
Dan Gohmana6cb6412010-05-11 23:54:07 +00001270 bool Op0IsKill = hasTrivialKill(I->getOperand(0));
Wesley Peckbf17cfa2010-11-23 03:31:01 +00001271
Dan Gohmanad368ac2008-08-27 18:10:19 +00001272 // First, try to perform the bitcast by inserting a reg-reg copy.
1273 unsigned ResultReg = 0;
Patrik Hagglunda61b17c2012-12-13 06:34:11 +00001274 if (SrcVT == DstVT) {
Stephen Hines37ed9c12014-12-01 14:51:49 -08001275 const TargetRegisterClass *SrcClass = TLI.getRegClassFor(SrcVT);
1276 const TargetRegisterClass *DstClass = TLI.getRegClassFor(DstVT);
Jakob Stoklund Olesene7917bb2010-07-11 05:16:54 +00001277 // Don't attempt a cross-class copy. It will likely fail.
1278 if (SrcClass == DstClass) {
1279 ResultReg = createResultReg(DstClass);
Stephen Hines36b56882014-04-23 16:57:46 -07001280 BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc,
1281 TII.get(TargetOpcode::COPY), ResultReg).addReg(Op0);
Jakob Stoklund Olesene7917bb2010-07-11 05:16:54 +00001282 }
Dan Gohmanad368ac2008-08-27 18:10:19 +00001283 }
Wesley Peckbf17cfa2010-11-23 03:31:01 +00001284
1285 // If the reg-reg copy failed, select a BITCAST opcode.
Dan Gohmanad368ac2008-08-27 18:10:19 +00001286 if (!ResultReg)
Stephen Hines37ed9c12014-12-01 14:51:49 -08001287 ResultReg = fastEmit_r(SrcVT, DstVT, ISD::BITCAST, Op0, Op0IsKill);
Wesley Peckbf17cfa2010-11-23 03:31:01 +00001288
Dan Gohmanad368ac2008-08-27 18:10:19 +00001289 if (!ResultReg)
Owen Andersond0533c92008-08-26 23:46:32 +00001290 return false;
Wesley Peckbf17cfa2010-11-23 03:31:01 +00001291
Stephen Hines37ed9c12014-12-01 14:51:49 -08001292 updateValueMap(I, ResultReg);
Owen Andersond0533c92008-08-26 23:46:32 +00001293 return true;
1294}
1295
Stephen Hines37ed9c12014-12-01 14:51:49 -08001296bool FastISel::selectInstruction(const Instruction *I) {
Dan Gohmane8c92dd2010-04-23 15:29:50 +00001297 // Just before the terminator instruction, insert instructions to
1298 // feed PHI nodes in successor blocks.
1299 if (isa<TerminatorInst>(I))
Stephen Hines37ed9c12014-12-01 14:51:49 -08001300 if (!handlePHINodesInSuccessorBlocks(I->getParent()))
Dan Gohmane8c92dd2010-04-23 15:29:50 +00001301 return false;
1302
Stephen Hines36b56882014-04-23 16:57:46 -07001303 DbgLoc = I->getDebugLoc();
Dan Gohman8ba3aa72010-04-20 00:48:35 +00001304
Stephen Hines37ed9c12014-12-01 14:51:49 -08001305 SavedInsertPt = FuncInfo.InsertPt;
Chad Rosierae6f2cb2011-11-29 19:40:47 +00001306
Stephen Hines37ed9c12014-12-01 14:51:49 -08001307 if (const auto *Call = dyn_cast<CallInst>(I)) {
Bob Wilsond49edb72012-08-03 04:06:28 +00001308 const Function *F = Call->getCalledFunction();
1309 LibFunc::Func Func;
Stephen Hinesdce4a402014-05-29 02:49:00 -07001310
1311 // As a special case, don't handle calls to builtin library functions that
1312 // may be translated directly to target instructions.
Bob Wilsond49edb72012-08-03 04:06:28 +00001313 if (F && !F->hasLocalLinkage() && F->hasName() &&
1314 LibInfo->getLibFunc(F->getName(), Func) &&
Bob Wilson982dc842012-08-03 21:26:24 +00001315 LibInfo->hasOptimizedCodeGen(Func))
Bob Wilsond49edb72012-08-03 04:06:28 +00001316 return false;
Stephen Hinesdce4a402014-05-29 02:49:00 -07001317
1318 // Don't handle Intrinsic::trap if a trap funciton is specified.
1319 if (F && F->getIntrinsicID() == Intrinsic::trap &&
1320 !TM.Options.getTrapFunctionName().empty())
1321 return false;
Bob Wilsond49edb72012-08-03 04:06:28 +00001322 }
1323
Dan Gohman6e3ff372009-12-05 01:27:58 +00001324 // First, try doing target-independent selection.
Stephen Hines37ed9c12014-12-01 14:51:49 -08001325 if (!SkipTargetIndependentISel) {
1326 if (selectOperator(I, I->getOpcode())) {
1327 ++NumFastIselSuccessIndependent;
1328 DbgLoc = DebugLoc();
1329 return true;
1330 }
1331 // Remove dead code.
Chad Rosierae6f2cb2011-11-29 19:40:47 +00001332 recomputeInsertPt();
1333 if (SavedInsertPt != FuncInfo.InsertPt)
1334 removeDeadCode(FuncInfo.InsertPt, SavedInsertPt);
Stephen Hines37ed9c12014-12-01 14:51:49 -08001335 SavedInsertPt = FuncInfo.InsertPt;
Chad Rosierae6f2cb2011-11-29 19:40:47 +00001336 }
Dan Gohman6e3ff372009-12-05 01:27:58 +00001337 // Next, try calling the target to attempt to handle the instruction.
Stephen Hines37ed9c12014-12-01 14:51:49 -08001338 if (fastSelectInstruction(I)) {
Jan Wen Voungfa785cb2013-03-08 22:56:31 +00001339 ++NumFastIselSuccessTarget;
Stephen Hines36b56882014-04-23 16:57:46 -07001340 DbgLoc = DebugLoc();
Dan Gohman6e3ff372009-12-05 01:27:58 +00001341 return true;
Dan Gohman8ba3aa72010-04-20 00:48:35 +00001342 }
Stephen Hines37ed9c12014-12-01 14:51:49 -08001343 // Remove dead code.
Chad Rosierae6f2cb2011-11-29 19:40:47 +00001344 recomputeInsertPt();
1345 if (SavedInsertPt != FuncInfo.InsertPt)
1346 removeDeadCode(FuncInfo.InsertPt, SavedInsertPt);
Dan Gohman6e3ff372009-12-05 01:27:58 +00001347
Stephen Hines36b56882014-04-23 16:57:46 -07001348 DbgLoc = DebugLoc();
Stephen Hines37ed9c12014-12-01 14:51:49 -08001349 // Undo phi node updates, because they will be added again by SelectionDAG.
1350 if (isa<TerminatorInst>(I))
1351 FuncInfo.PHINodesToUpdate.resize(FuncInfo.OrigNumPHINodesToUpdate);
Dan Gohman6e3ff372009-12-05 01:27:58 +00001352 return false;
Dan Gohman40b189e2008-09-05 18:18:20 +00001353}
1354
Stephen Hines37ed9c12014-12-01 14:51:49 -08001355/// Emit an unconditional branch to the given block, unless it is the immediate
1356/// (fall-through) successor, and update the CFG.
1357void FastISel::fastEmitBranch(MachineBasicBlock *MSucc, DebugLoc DbgLoc) {
Evan Cheng092e5e72013-02-11 01:27:15 +00001358 if (FuncInfo.MBB->getBasicBlock()->size() > 1 &&
1359 FuncInfo.MBB->isLayoutSuccessor(MSucc)) {
Eric Christopher18112d82012-04-10 18:18:10 +00001360 // For more accurate line information if this is the only instruction
1361 // in the block then emit it, otherwise we have the unconditional
1362 // fall-through case, which needs no instructions.
Dan Gohmand98d6202008-10-02 22:15:21 +00001363 } else {
1364 // The unconditional branch case.
Stephen Hinesdce4a402014-05-29 02:49:00 -07001365 TII.InsertBranch(*FuncInfo.MBB, MSucc, nullptr,
Stephen Hines36b56882014-04-23 16:57:46 -07001366 SmallVector<MachineOperand, 0>(), DbgLoc);
Dan Gohmand98d6202008-10-02 22:15:21 +00001367 }
Stephen Hinesc6a4f5e2014-07-21 00:45:20 -07001368 uint32_t BranchWeight = 0;
1369 if (FuncInfo.BPI)
1370 BranchWeight = FuncInfo.BPI->getEdgeWeight(FuncInfo.MBB->getBasicBlock(),
1371 MSucc->getBasicBlock());
1372 FuncInfo.MBB->addSuccessor(MSucc, BranchWeight);
Dan Gohmand98d6202008-10-02 22:15:21 +00001373}
1374
Stephen Hines37ed9c12014-12-01 14:51:49 -08001375/// Emit an FNeg operation.
1376bool FastISel::selectFNeg(const User *I) {
Dan Gohman3d45a852009-09-03 22:53:57 +00001377 unsigned OpReg = getRegForValue(BinaryOperator::getFNegArgument(I));
Stephen Hines37ed9c12014-12-01 14:51:49 -08001378 if (!OpReg)
1379 return false;
Dan Gohmana6cb6412010-05-11 23:54:07 +00001380 bool OpRegIsKill = hasTrivialKill(I);
1381
Dan Gohman4a215a12009-09-11 00:36:43 +00001382 // If the target has ISD::FNEG, use it.
1383 EVT VT = TLI.getValueType(I->getType());
Stephen Hines37ed9c12014-12-01 14:51:49 -08001384 unsigned ResultReg = fastEmit_r(VT.getSimpleVT(), VT.getSimpleVT(), ISD::FNEG,
1385 OpReg, OpRegIsKill);
1386 if (ResultReg) {
1387 updateValueMap(I, ResultReg);
Dan Gohman4a215a12009-09-11 00:36:43 +00001388 return true;
1389 }
1390
Dan Gohman5e5abb72009-09-11 00:34:46 +00001391 // Bitcast the value to integer, twiddle the sign bit with xor,
1392 // and then bitcast it back to floating-point.
Stephen Hines37ed9c12014-12-01 14:51:49 -08001393 if (VT.getSizeInBits() > 64)
1394 return false;
Dan Gohman5e5abb72009-09-11 00:34:46 +00001395 EVT IntVT = EVT::getIntegerVT(I->getContext(), VT.getSizeInBits());
1396 if (!TLI.isTypeLegal(IntVT))
1397 return false;
1398
Stephen Hines37ed9c12014-12-01 14:51:49 -08001399 unsigned IntReg = fastEmit_r(VT.getSimpleVT(), IntVT.getSimpleVT(),
Wesley Peckbf17cfa2010-11-23 03:31:01 +00001400 ISD::BITCAST, OpReg, OpRegIsKill);
Stephen Hines37ed9c12014-12-01 14:51:49 -08001401 if (!IntReg)
Dan Gohman5e5abb72009-09-11 00:34:46 +00001402 return false;
1403
Stephen Hines37ed9c12014-12-01 14:51:49 -08001404 unsigned IntResultReg = fastEmit_ri_(
1405 IntVT.getSimpleVT(), ISD::XOR, IntReg, /*IsKill=*/true,
1406 UINT64_C(1) << (VT.getSizeInBits() - 1), IntVT.getSimpleVT());
1407 if (!IntResultReg)
Dan Gohman5e5abb72009-09-11 00:34:46 +00001408 return false;
1409
Stephen Hines37ed9c12014-12-01 14:51:49 -08001410 ResultReg = fastEmit_r(IntVT.getSimpleVT(), VT.getSimpleVT(), ISD::BITCAST,
1411 IntResultReg, /*IsKill=*/true);
1412 if (!ResultReg)
Dan Gohman3d45a852009-09-03 22:53:57 +00001413 return false;
1414
Stephen Hines37ed9c12014-12-01 14:51:49 -08001415 updateValueMap(I, ResultReg);
Dan Gohman3d45a852009-09-03 22:53:57 +00001416 return true;
1417}
1418
Stephen Hines37ed9c12014-12-01 14:51:49 -08001419bool FastISel::selectExtractValue(const User *U) {
Eli Friedman2586b8f2011-05-16 20:27:46 +00001420 const ExtractValueInst *EVI = dyn_cast<ExtractValueInst>(U);
Eli Friedmana4c920d2011-05-16 20:34:53 +00001421 if (!EVI)
Eli Friedman2586b8f2011-05-16 20:27:46 +00001422 return false;
1423
Eli Friedman482feb32011-05-16 21:06:17 +00001424 // Make sure we only try to handle extracts with a legal result. But also
1425 // allow i1 because it's easy.
Eli Friedman2586b8f2011-05-16 20:27:46 +00001426 EVT RealVT = TLI.getValueType(EVI->getType(), /*AllowUnknown=*/true);
1427 if (!RealVT.isSimple())
1428 return false;
1429 MVT VT = RealVT.getSimpleVT();
Eli Friedman482feb32011-05-16 21:06:17 +00001430 if (!TLI.isTypeLegal(VT) && VT != MVT::i1)
Eli Friedman2586b8f2011-05-16 20:27:46 +00001431 return false;
1432
1433 const Value *Op0 = EVI->getOperand(0);
Chris Lattnerdb125cf2011-07-18 04:54:35 +00001434 Type *AggTy = Op0->getType();
Eli Friedman2586b8f2011-05-16 20:27:46 +00001435
1436 // Get the base result register.
1437 unsigned ResultReg;
1438 DenseMap<const Value *, unsigned>::iterator I = FuncInfo.ValueMap.find(Op0);
1439 if (I != FuncInfo.ValueMap.end())
1440 ResultReg = I->second;
Eli Friedman0b4d96b2011-06-06 05:46:34 +00001441 else if (isa<Instruction>(Op0))
Eli Friedman2586b8f2011-05-16 20:27:46 +00001442 ResultReg = FuncInfo.InitializeRegForValue(Op0);
Eli Friedman0b4d96b2011-06-06 05:46:34 +00001443 else
1444 return false; // fast-isel can't handle aggregate constants at the moment
Eli Friedman2586b8f2011-05-16 20:27:46 +00001445
1446 // Get the actual result register, which is an offset from the base register.
Jay Foadfc6d3a42011-07-13 10:26:04 +00001447 unsigned VTIndex = ComputeLinearIndex(AggTy, EVI->getIndices());
Eli Friedman2586b8f2011-05-16 20:27:46 +00001448
1449 SmallVector<EVT, 4> AggValueVTs;
1450 ComputeValueVTs(TLI, AggTy, AggValueVTs);
1451
1452 for (unsigned i = 0; i < VTIndex; i++)
1453 ResultReg += TLI.getNumRegisters(FuncInfo.Fn->getContext(), AggValueVTs[i]);
1454
Stephen Hines37ed9c12014-12-01 14:51:49 -08001455 updateValueMap(EVI, ResultReg);
Eli Friedman2586b8f2011-05-16 20:27:46 +00001456 return true;
1457}
1458
Stephen Hines37ed9c12014-12-01 14:51:49 -08001459bool FastISel::selectOperator(const User *I, unsigned Opcode) {
Dan Gohman40b189e2008-09-05 18:18:20 +00001460 switch (Opcode) {
Dan Gohmanae3a0be2009-06-04 22:49:04 +00001461 case Instruction::Add:
Stephen Hines37ed9c12014-12-01 14:51:49 -08001462 return selectBinaryOp(I, ISD::ADD);
Dan Gohmanae3a0be2009-06-04 22:49:04 +00001463 case Instruction::FAdd:
Stephen Hines37ed9c12014-12-01 14:51:49 -08001464 return selectBinaryOp(I, ISD::FADD);
Dan Gohmanae3a0be2009-06-04 22:49:04 +00001465 case Instruction::Sub:
Stephen Hines37ed9c12014-12-01 14:51:49 -08001466 return selectBinaryOp(I, ISD::SUB);
Dan Gohmanae3a0be2009-06-04 22:49:04 +00001467 case Instruction::FSub:
Dan Gohman3d45a852009-09-03 22:53:57 +00001468 // FNeg is currently represented in LLVM IR as a special case of FSub.
1469 if (BinaryOperator::isFNeg(I))
Stephen Hines37ed9c12014-12-01 14:51:49 -08001470 return selectFNeg(I);
1471 return selectBinaryOp(I, ISD::FSUB);
Dan Gohmanae3a0be2009-06-04 22:49:04 +00001472 case Instruction::Mul:
Stephen Hines37ed9c12014-12-01 14:51:49 -08001473 return selectBinaryOp(I, ISD::MUL);
Dan Gohmanae3a0be2009-06-04 22:49:04 +00001474 case Instruction::FMul:
Stephen Hines37ed9c12014-12-01 14:51:49 -08001475 return selectBinaryOp(I, ISD::FMUL);
Dan Gohman3df24e62008-09-03 23:12:08 +00001476 case Instruction::SDiv:
Stephen Hines37ed9c12014-12-01 14:51:49 -08001477 return selectBinaryOp(I, ISD::SDIV);
Dan Gohman3df24e62008-09-03 23:12:08 +00001478 case Instruction::UDiv:
Stephen Hines37ed9c12014-12-01 14:51:49 -08001479 return selectBinaryOp(I, ISD::UDIV);
Dan Gohman3df24e62008-09-03 23:12:08 +00001480 case Instruction::FDiv:
Stephen Hines37ed9c12014-12-01 14:51:49 -08001481 return selectBinaryOp(I, ISD::FDIV);
Dan Gohman3df24e62008-09-03 23:12:08 +00001482 case Instruction::SRem:
Stephen Hines37ed9c12014-12-01 14:51:49 -08001483 return selectBinaryOp(I, ISD::SREM);
Dan Gohman3df24e62008-09-03 23:12:08 +00001484 case Instruction::URem:
Stephen Hines37ed9c12014-12-01 14:51:49 -08001485 return selectBinaryOp(I, ISD::UREM);
Dan Gohman3df24e62008-09-03 23:12:08 +00001486 case Instruction::FRem:
Stephen Hines37ed9c12014-12-01 14:51:49 -08001487 return selectBinaryOp(I, ISD::FREM);
Dan Gohman3df24e62008-09-03 23:12:08 +00001488 case Instruction::Shl:
Stephen Hines37ed9c12014-12-01 14:51:49 -08001489 return selectBinaryOp(I, ISD::SHL);
Dan Gohman3df24e62008-09-03 23:12:08 +00001490 case Instruction::LShr:
Stephen Hines37ed9c12014-12-01 14:51:49 -08001491 return selectBinaryOp(I, ISD::SRL);
Dan Gohman3df24e62008-09-03 23:12:08 +00001492 case Instruction::AShr:
Stephen Hines37ed9c12014-12-01 14:51:49 -08001493 return selectBinaryOp(I, ISD::SRA);
Dan Gohman3df24e62008-09-03 23:12:08 +00001494 case Instruction::And:
Stephen Hines37ed9c12014-12-01 14:51:49 -08001495 return selectBinaryOp(I, ISD::AND);
Dan Gohman3df24e62008-09-03 23:12:08 +00001496 case Instruction::Or:
Stephen Hines37ed9c12014-12-01 14:51:49 -08001497 return selectBinaryOp(I, ISD::OR);
Dan Gohman3df24e62008-09-03 23:12:08 +00001498 case Instruction::Xor:
Stephen Hines37ed9c12014-12-01 14:51:49 -08001499 return selectBinaryOp(I, ISD::XOR);
Dan Gohmanb0cf29c2008-08-13 20:19:35 +00001500
Dan Gohman3df24e62008-09-03 23:12:08 +00001501 case Instruction::GetElementPtr:
Stephen Hines37ed9c12014-12-01 14:51:49 -08001502 return selectGetElementPtr(I);
Dan Gohmanbdedd442008-08-20 00:11:48 +00001503
Dan Gohman3df24e62008-09-03 23:12:08 +00001504 case Instruction::Br: {
Dan Gohman46510a72010-04-15 01:51:59 +00001505 const BranchInst *BI = cast<BranchInst>(I);
Dan Gohmanbdedd442008-08-20 00:11:48 +00001506
Dan Gohman3df24e62008-09-03 23:12:08 +00001507 if (BI->isUnconditional()) {
Dan Gohman46510a72010-04-15 01:51:59 +00001508 const BasicBlock *LLVMSucc = BI->getSuccessor(0);
Dan Gohmana4160c32010-07-07 16:29:44 +00001509 MachineBasicBlock *MSucc = FuncInfo.MBBMap[LLVMSucc];
Stephen Hines37ed9c12014-12-01 14:51:49 -08001510 fastEmitBranch(MSucc, BI->getDebugLoc());
Dan Gohman3df24e62008-09-03 23:12:08 +00001511 return true;
Owen Anderson9d5b4162008-08-27 00:31:01 +00001512 }
Dan Gohman3df24e62008-09-03 23:12:08 +00001513
1514 // Conditional branches are not handed yet.
1515 // Halt "fast" selection and bail.
1516 return false;
Dan Gohmanb0cf29c2008-08-13 20:19:35 +00001517 }
1518
Dan Gohman087c8502008-09-05 01:08:41 +00001519 case Instruction::Unreachable:
Stephen Hinesdce4a402014-05-29 02:49:00 -07001520 if (TM.Options.TrapUnreachable)
Stephen Hines37ed9c12014-12-01 14:51:49 -08001521 return fastEmit_(MVT::Other, MVT::Other, ISD::TRAP) != 0;
Stephen Hinesdce4a402014-05-29 02:49:00 -07001522 else
1523 return true;
Dan Gohman087c8502008-09-05 01:08:41 +00001524
Dan Gohman0586d912008-09-10 20:11:02 +00001525 case Instruction::Alloca:
1526 // FunctionLowering has the static-sized case covered.
Dan Gohmana4160c32010-07-07 16:29:44 +00001527 if (FuncInfo.StaticAllocaMap.count(cast<AllocaInst>(I)))
Dan Gohman0586d912008-09-10 20:11:02 +00001528 return true;
1529
1530 // Dynamic-sized alloca is not handled yet.
1531 return false;
Wesley Peckbf17cfa2010-11-23 03:31:01 +00001532
Dan Gohman33134c42008-09-25 17:05:24 +00001533 case Instruction::Call:
Stephen Hines37ed9c12014-12-01 14:51:49 -08001534 return selectCall(I);
Wesley Peckbf17cfa2010-11-23 03:31:01 +00001535
Dan Gohman3df24e62008-09-03 23:12:08 +00001536 case Instruction::BitCast:
Stephen Hines37ed9c12014-12-01 14:51:49 -08001537 return selectBitCast(I);
Dan Gohman3df24e62008-09-03 23:12:08 +00001538
1539 case Instruction::FPToSI:
Stephen Hines37ed9c12014-12-01 14:51:49 -08001540 return selectCast(I, ISD::FP_TO_SINT);
Dan Gohman3df24e62008-09-03 23:12:08 +00001541 case Instruction::ZExt:
Stephen Hines37ed9c12014-12-01 14:51:49 -08001542 return selectCast(I, ISD::ZERO_EXTEND);
Dan Gohman3df24e62008-09-03 23:12:08 +00001543 case Instruction::SExt:
Stephen Hines37ed9c12014-12-01 14:51:49 -08001544 return selectCast(I, ISD::SIGN_EXTEND);
Dan Gohman3df24e62008-09-03 23:12:08 +00001545 case Instruction::Trunc:
Stephen Hines37ed9c12014-12-01 14:51:49 -08001546 return selectCast(I, ISD::TRUNCATE);
Dan Gohman3df24e62008-09-03 23:12:08 +00001547 case Instruction::SIToFP:
Stephen Hines37ed9c12014-12-01 14:51:49 -08001548 return selectCast(I, ISD::SINT_TO_FP);
Dan Gohman3df24e62008-09-03 23:12:08 +00001549
1550 case Instruction::IntToPtr: // Deliberate fall-through.
1551 case Instruction::PtrToInt: {
Owen Andersone50ed302009-08-10 22:56:29 +00001552 EVT SrcVT = TLI.getValueType(I->getOperand(0)->getType());
1553 EVT DstVT = TLI.getValueType(I->getType());
Dan Gohman3df24e62008-09-03 23:12:08 +00001554 if (DstVT.bitsGT(SrcVT))
Stephen Hines37ed9c12014-12-01 14:51:49 -08001555 return selectCast(I, ISD::ZERO_EXTEND);
Dan Gohman3df24e62008-09-03 23:12:08 +00001556 if (DstVT.bitsLT(SrcVT))
Stephen Hines37ed9c12014-12-01 14:51:49 -08001557 return selectCast(I, ISD::TRUNCATE);
Dan Gohman3df24e62008-09-03 23:12:08 +00001558 unsigned Reg = getRegForValue(I->getOperand(0));
Stephen Hines37ed9c12014-12-01 14:51:49 -08001559 if (!Reg)
1560 return false;
1561 updateValueMap(I, Reg);
Dan Gohman3df24e62008-09-03 23:12:08 +00001562 return true;
1563 }
Dan Gohmand57dd5f2008-09-23 21:53:34 +00001564
Eli Friedman2586b8f2011-05-16 20:27:46 +00001565 case Instruction::ExtractValue:
Stephen Hines37ed9c12014-12-01 14:51:49 -08001566 return selectExtractValue(I);
Eli Friedman2586b8f2011-05-16 20:27:46 +00001567
Dan Gohmanba5be5c2010-04-20 15:00:41 +00001568 case Instruction::PHI:
1569 llvm_unreachable("FastISel shouldn't visit PHI nodes!");
1570
Dan Gohman3df24e62008-09-03 23:12:08 +00001571 default:
1572 // Unhandled instruction. Halt "fast" selection and bail.
1573 return false;
1574 }
Dan Gohmanb0cf29c2008-08-13 20:19:35 +00001575}
1576
Stephen Hines37ed9c12014-12-01 14:51:49 -08001577FastISel::FastISel(FunctionLoweringInfo &FuncInfo,
1578 const TargetLibraryInfo *LibInfo,
1579 bool SkipTargetIndependentISel)
1580 : FuncInfo(FuncInfo), MF(FuncInfo.MF), MRI(FuncInfo.MF->getRegInfo()),
1581 MFI(*FuncInfo.MF->getFrameInfo()), MCP(*FuncInfo.MF->getConstantPool()),
1582 TM(FuncInfo.MF->getTarget()), DL(*MF->getSubtarget().getDataLayout()),
1583 TII(*MF->getSubtarget().getInstrInfo()),
1584 TLI(*MF->getSubtarget().getTargetLowering()),
1585 TRI(*MF->getSubtarget().getRegisterInfo()), LibInfo(LibInfo),
1586 SkipTargetIndependentISel(SkipTargetIndependentISel) {}
Dan Gohmanbb466332008-08-20 21:05:57 +00001587
Dan Gohmane285a742008-08-14 21:51:29 +00001588FastISel::~FastISel() {}
1589
Stephen Hines37ed9c12014-12-01 14:51:49 -08001590bool FastISel::fastLowerArguments() { return false; }
1591
1592bool FastISel::fastLowerCall(CallLoweringInfo & /*CLI*/) { return false; }
1593
1594bool FastISel::fastLowerIntrinsicCall(const IntrinsicInst * /*II*/) {
Evan Cheng092e5e72013-02-11 01:27:15 +00001595 return false;
1596}
1597
Stephen Hines37ed9c12014-12-01 14:51:49 -08001598unsigned FastISel::fastEmit_(MVT, MVT, unsigned) { return 0; }
1599
1600unsigned FastISel::fastEmit_r(MVT, MVT, unsigned, unsigned /*Op0*/,
1601 bool /*Op0IsKill*/) {
Dan Gohmanb0cf29c2008-08-13 20:19:35 +00001602 return 0;
1603}
1604
Stephen Hines37ed9c12014-12-01 14:51:49 -08001605unsigned FastISel::fastEmit_rr(MVT, MVT, unsigned, unsigned /*Op0*/,
1606 bool /*Op0IsKill*/, unsigned /*Op1*/,
1607 bool /*Op1IsKill*/) {
Dan Gohmanb0cf29c2008-08-13 20:19:35 +00001608 return 0;
1609}
1610
Stephen Hines37ed9c12014-12-01 14:51:49 -08001611unsigned FastISel::fastEmit_i(MVT, MVT, unsigned, uint64_t /*Imm*/) {
Dan Gohmanb0cf29c2008-08-13 20:19:35 +00001612 return 0;
1613}
1614
Stephen Hines37ed9c12014-12-01 14:51:49 -08001615unsigned FastISel::fastEmit_f(MVT, MVT, unsigned,
1616 const ConstantFP * /*FPImm*/) {
Evan Cheng83785c82008-08-20 22:45:34 +00001617 return 0;
1618}
1619
Stephen Hines37ed9c12014-12-01 14:51:49 -08001620unsigned FastISel::fastEmit_ri(MVT, MVT, unsigned, unsigned /*Op0*/,
1621 bool /*Op0IsKill*/, uint64_t /*Imm*/) {
Dan Gohman10df0fa2008-08-27 01:09:54 +00001622 return 0;
1623}
1624
Stephen Hines37ed9c12014-12-01 14:51:49 -08001625unsigned FastISel::fastEmit_rf(MVT, MVT, unsigned, unsigned /*Op0*/,
1626 bool /*Op0IsKill*/,
Dan Gohman46510a72010-04-15 01:51:59 +00001627 const ConstantFP * /*FPImm*/) {
Dan Gohman10df0fa2008-08-27 01:09:54 +00001628 return 0;
1629}
1630
Stephen Hines37ed9c12014-12-01 14:51:49 -08001631unsigned FastISel::fastEmit_rri(MVT, MVT, unsigned, unsigned /*Op0*/,
1632 bool /*Op0IsKill*/, unsigned /*Op1*/,
1633 bool /*Op1IsKill*/, uint64_t /*Imm*/) {
Evan Cheng83785c82008-08-20 22:45:34 +00001634 return 0;
1635}
1636
Stephen Hines37ed9c12014-12-01 14:51:49 -08001637/// This method is a wrapper of fastEmit_ri. It first tries to emit an
1638/// instruction with an immediate operand using fastEmit_ri.
Evan Cheng83785c82008-08-20 22:45:34 +00001639/// If that fails, it materializes the immediate into a register and try
Stephen Hines37ed9c12014-12-01 14:51:49 -08001640/// fastEmit_rr instead.
1641unsigned FastISel::fastEmit_ri_(MVT VT, unsigned Opcode, unsigned Op0,
1642 bool Op0IsKill, uint64_t Imm, MVT ImmType) {
Chris Lattner602fc062011-04-17 20:23:29 +00001643 // If this is a multiply by a power of two, emit this as a shift left.
1644 if (Opcode == ISD::MUL && isPowerOf2_64(Imm)) {
1645 Opcode = ISD::SHL;
1646 Imm = Log2_64(Imm);
Chris Lattner090ca912011-04-18 06:55:51 +00001647 } else if (Opcode == ISD::UDIV && isPowerOf2_64(Imm)) {
1648 // div x, 8 -> srl x, 3
1649 Opcode = ISD::SRL;
1650 Imm = Log2_64(Imm);
Chris Lattner602fc062011-04-17 20:23:29 +00001651 }
Owen Andersond74ea772011-04-22 23:38:06 +00001652
Chris Lattner602fc062011-04-17 20:23:29 +00001653 // Horrible hack (to be removed), check to make sure shift amounts are
1654 // in-range.
1655 if ((Opcode == ISD::SHL || Opcode == ISD::SRA || Opcode == ISD::SRL) &&
1656 Imm >= VT.getSizeInBits())
1657 return 0;
Owen Andersond74ea772011-04-22 23:38:06 +00001658
Evan Cheng83785c82008-08-20 22:45:34 +00001659 // First check if immediate type is legal. If not, we can't use the ri form.
Stephen Hines37ed9c12014-12-01 14:51:49 -08001660 unsigned ResultReg = fastEmit_ri(VT, VT, Opcode, Op0, Op0IsKill, Imm);
1661 if (ResultReg)
Evan Cheng83785c82008-08-20 22:45:34 +00001662 return ResultReg;
Stephen Hines37ed9c12014-12-01 14:51:49 -08001663 unsigned MaterialReg = fastEmit_i(ImmType, ImmType, ISD::Constant, Imm);
1664 if (!MaterialReg) {
Eli Friedmanb2b03fc2011-04-29 23:34:52 +00001665 // This is a bit ugly/slow, but failing here means falling out of
1666 // fast-isel, which would be very slow.
Stephen Hines37ed9c12014-12-01 14:51:49 -08001667 IntegerType *ITy =
1668 IntegerType::get(FuncInfo.Fn->getContext(), VT.getSizeInBits());
Eli Friedmanb2b03fc2011-04-29 23:34:52 +00001669 MaterialReg = getRegForValue(ConstantInt::get(ITy, Imm));
Stephen Hines37ed9c12014-12-01 14:51:49 -08001670 if (!MaterialReg)
1671 return 0;
Eli Friedmanb2b03fc2011-04-29 23:34:52 +00001672 }
Stephen Hines37ed9c12014-12-01 14:51:49 -08001673 return fastEmit_rr(VT, VT, Opcode, Op0, Op0IsKill, MaterialReg,
1674 /*IsKill=*/true);
Dan Gohmand5fe57d2008-08-21 01:41:07 +00001675}
1676
Stephen Hines37ed9c12014-12-01 14:51:49 -08001677unsigned FastISel::createResultReg(const TargetRegisterClass *RC) {
Dan Gohmand5fe57d2008-08-21 01:41:07 +00001678 return MRI.createVirtualRegister(RC);
Evan Cheng83785c82008-08-20 22:45:34 +00001679}
1680
Stephen Hines37ed9c12014-12-01 14:51:49 -08001681unsigned FastISel::constrainOperandRegClass(const MCInstrDesc &II, unsigned Op,
1682 unsigned OpNum) {
Stephen Hinesdce4a402014-05-29 02:49:00 -07001683 if (TargetRegisterInfo::isVirtualRegister(Op)) {
1684 const TargetRegisterClass *RegClass =
1685 TII.getRegClass(II, OpNum, &TRI, *FuncInfo.MF);
1686 if (!MRI.constrainRegClass(Op, RegClass)) {
1687 // If it's not legal to COPY between the register classes, something
1688 // has gone very wrong before we got here.
1689 unsigned NewOp = createResultReg(RegClass);
1690 BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc,
1691 TII.get(TargetOpcode::COPY), NewOp).addReg(Op);
1692 return NewOp;
1693 }
1694 }
1695 return Op;
1696}
1697
Stephen Hines37ed9c12014-12-01 14:51:49 -08001698unsigned FastISel::fastEmitInst_(unsigned MachineInstOpcode,
1699 const TargetRegisterClass *RC) {
Dan Gohmand5fe57d2008-08-21 01:41:07 +00001700 unsigned ResultReg = createResultReg(RC);
Evan Chenge837dea2011-06-28 19:10:37 +00001701 const MCInstrDesc &II = TII.get(MachineInstOpcode);
Dan Gohmanb0cf29c2008-08-13 20:19:35 +00001702
Stephen Hines36b56882014-04-23 16:57:46 -07001703 BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc, II, ResultReg);
Dan Gohmanb0cf29c2008-08-13 20:19:35 +00001704 return ResultReg;
1705}
1706
Stephen Hines37ed9c12014-12-01 14:51:49 -08001707unsigned FastISel::fastEmitInst_r(unsigned MachineInstOpcode,
1708 const TargetRegisterClass *RC, unsigned Op0,
1709 bool Op0IsKill) {
Evan Chenge837dea2011-06-28 19:10:37 +00001710 const MCInstrDesc &II = TII.get(MachineInstOpcode);
Dan Gohmanb0cf29c2008-08-13 20:19:35 +00001711
Stephen Hinesdce4a402014-05-29 02:49:00 -07001712 unsigned ResultReg = createResultReg(RC);
1713 Op0 = constrainOperandRegClass(II, Op0, II.getNumDefs());
1714
Evan Cheng5960e4e2008-09-08 08:38:20 +00001715 if (II.getNumDefs() >= 1)
Stephen Hines36b56882014-04-23 16:57:46 -07001716 BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc, II, ResultReg)
Stephen Hines37ed9c12014-12-01 14:51:49 -08001717 .addReg(Op0, getKillRegState(Op0IsKill));
Evan Cheng5960e4e2008-09-08 08:38:20 +00001718 else {
Stephen Hines36b56882014-04-23 16:57:46 -07001719 BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc, II)
Stephen Hines37ed9c12014-12-01 14:51:49 -08001720 .addReg(Op0, getKillRegState(Op0IsKill));
Stephen Hines36b56882014-04-23 16:57:46 -07001721 BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc,
1722 TII.get(TargetOpcode::COPY), ResultReg).addReg(II.ImplicitDefs[0]);
Evan Cheng5960e4e2008-09-08 08:38:20 +00001723 }
1724
Dan Gohmanb0cf29c2008-08-13 20:19:35 +00001725 return ResultReg;
1726}
1727
Stephen Hines37ed9c12014-12-01 14:51:49 -08001728unsigned FastISel::fastEmitInst_rr(unsigned MachineInstOpcode,
1729 const TargetRegisterClass *RC, unsigned Op0,
1730 bool Op0IsKill, unsigned Op1,
1731 bool Op1IsKill) {
Evan Chenge837dea2011-06-28 19:10:37 +00001732 const MCInstrDesc &II = TII.get(MachineInstOpcode);
Dan Gohmanb0cf29c2008-08-13 20:19:35 +00001733
Stephen Hinesdce4a402014-05-29 02:49:00 -07001734 unsigned ResultReg = createResultReg(RC);
1735 Op0 = constrainOperandRegClass(II, Op0, II.getNumDefs());
1736 Op1 = constrainOperandRegClass(II, Op1, II.getNumDefs() + 1);
1737
Evan Cheng5960e4e2008-09-08 08:38:20 +00001738 if (II.getNumDefs() >= 1)
Stephen Hines36b56882014-04-23 16:57:46 -07001739 BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc, II, ResultReg)
Stephen Hines37ed9c12014-12-01 14:51:49 -08001740 .addReg(Op0, getKillRegState(Op0IsKill))
1741 .addReg(Op1, getKillRegState(Op1IsKill));
Evan Cheng5960e4e2008-09-08 08:38:20 +00001742 else {
Stephen Hines36b56882014-04-23 16:57:46 -07001743 BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc, II)
Stephen Hines37ed9c12014-12-01 14:51:49 -08001744 .addReg(Op0, getKillRegState(Op0IsKill))
1745 .addReg(Op1, getKillRegState(Op1IsKill));
Stephen Hines36b56882014-04-23 16:57:46 -07001746 BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc,
1747 TII.get(TargetOpcode::COPY), ResultReg).addReg(II.ImplicitDefs[0]);
Evan Cheng5960e4e2008-09-08 08:38:20 +00001748 }
Dan Gohmanb0cf29c2008-08-13 20:19:35 +00001749 return ResultReg;
1750}
Dan Gohmand5fe57d2008-08-21 01:41:07 +00001751
Stephen Hines37ed9c12014-12-01 14:51:49 -08001752unsigned FastISel::fastEmitInst_rrr(unsigned MachineInstOpcode,
1753 const TargetRegisterClass *RC, unsigned Op0,
1754 bool Op0IsKill, unsigned Op1,
1755 bool Op1IsKill, unsigned Op2,
1756 bool Op2IsKill) {
Evan Chenge837dea2011-06-28 19:10:37 +00001757 const MCInstrDesc &II = TII.get(MachineInstOpcode);
Owen Andersond71867a2011-05-05 17:59:04 +00001758
Stephen Hinesdce4a402014-05-29 02:49:00 -07001759 unsigned ResultReg = createResultReg(RC);
1760 Op0 = constrainOperandRegClass(II, Op0, II.getNumDefs());
1761 Op1 = constrainOperandRegClass(II, Op1, II.getNumDefs() + 1);
1762 Op2 = constrainOperandRegClass(II, Op2, II.getNumDefs() + 2);
1763
Owen Andersond71867a2011-05-05 17:59:04 +00001764 if (II.getNumDefs() >= 1)
Stephen Hines36b56882014-04-23 16:57:46 -07001765 BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc, II, ResultReg)
Stephen Hines37ed9c12014-12-01 14:51:49 -08001766 .addReg(Op0, getKillRegState(Op0IsKill))
1767 .addReg(Op1, getKillRegState(Op1IsKill))
1768 .addReg(Op2, getKillRegState(Op2IsKill));
Owen Andersond71867a2011-05-05 17:59:04 +00001769 else {
Stephen Hines36b56882014-04-23 16:57:46 -07001770 BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc, II)
Stephen Hines37ed9c12014-12-01 14:51:49 -08001771 .addReg(Op0, getKillRegState(Op0IsKill))
1772 .addReg(Op1, getKillRegState(Op1IsKill))
1773 .addReg(Op2, getKillRegState(Op2IsKill));
Stephen Hines36b56882014-04-23 16:57:46 -07001774 BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc,
1775 TII.get(TargetOpcode::COPY), ResultReg).addReg(II.ImplicitDefs[0]);
Owen Andersond71867a2011-05-05 17:59:04 +00001776 }
1777 return ResultReg;
1778}
1779
Stephen Hines37ed9c12014-12-01 14:51:49 -08001780unsigned FastISel::fastEmitInst_ri(unsigned MachineInstOpcode,
1781 const TargetRegisterClass *RC, unsigned Op0,
1782 bool Op0IsKill, uint64_t Imm) {
Evan Chenge837dea2011-06-28 19:10:37 +00001783 const MCInstrDesc &II = TII.get(MachineInstOpcode);
Owen Anderson2ce5bf12011-03-11 21:33:55 +00001784
Stephen Hinesdce4a402014-05-29 02:49:00 -07001785 unsigned ResultReg = createResultReg(RC);
1786 Op0 = constrainOperandRegClass(II, Op0, II.getNumDefs());
1787
Owen Anderson2ce5bf12011-03-11 21:33:55 +00001788 if (II.getNumDefs() >= 1)
Stephen Hines36b56882014-04-23 16:57:46 -07001789 BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc, II, ResultReg)
Stephen Hines37ed9c12014-12-01 14:51:49 -08001790 .addReg(Op0, getKillRegState(Op0IsKill))
1791 .addImm(Imm);
Owen Anderson2ce5bf12011-03-11 21:33:55 +00001792 else {
Stephen Hines36b56882014-04-23 16:57:46 -07001793 BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc, II)
Stephen Hines37ed9c12014-12-01 14:51:49 -08001794 .addReg(Op0, getKillRegState(Op0IsKill))
1795 .addImm(Imm);
Stephen Hines36b56882014-04-23 16:57:46 -07001796 BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc,
1797 TII.get(TargetOpcode::COPY), ResultReg).addReg(II.ImplicitDefs[0]);
Owen Anderson2ce5bf12011-03-11 21:33:55 +00001798 }
1799 return ResultReg;
1800}
1801
Stephen Hines37ed9c12014-12-01 14:51:49 -08001802unsigned FastISel::fastEmitInst_rii(unsigned MachineInstOpcode,
1803 const TargetRegisterClass *RC, unsigned Op0,
1804 bool Op0IsKill, uint64_t Imm1,
1805 uint64_t Imm2) {
Evan Chenge837dea2011-06-28 19:10:37 +00001806 const MCInstrDesc &II = TII.get(MachineInstOpcode);
Dan Gohman10df0fa2008-08-27 01:09:54 +00001807
Stephen Hinesdce4a402014-05-29 02:49:00 -07001808 unsigned ResultReg = createResultReg(RC);
1809 Op0 = constrainOperandRegClass(II, Op0, II.getNumDefs());
1810
Evan Cheng5960e4e2008-09-08 08:38:20 +00001811 if (II.getNumDefs() >= 1)
Stephen Hines36b56882014-04-23 16:57:46 -07001812 BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc, II, ResultReg)
Stephen Hines37ed9c12014-12-01 14:51:49 -08001813 .addReg(Op0, getKillRegState(Op0IsKill))
1814 .addImm(Imm1)
1815 .addImm(Imm2);
Evan Cheng5960e4e2008-09-08 08:38:20 +00001816 else {
Stephen Hines36b56882014-04-23 16:57:46 -07001817 BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc, II)
Stephen Hines37ed9c12014-12-01 14:51:49 -08001818 .addReg(Op0, getKillRegState(Op0IsKill))
1819 .addImm(Imm1)
1820 .addImm(Imm2);
Stephen Hines36b56882014-04-23 16:57:46 -07001821 BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc,
1822 TII.get(TargetOpcode::COPY), ResultReg).addReg(II.ImplicitDefs[0]);
Evan Cheng5960e4e2008-09-08 08:38:20 +00001823 }
Dan Gohman10df0fa2008-08-27 01:09:54 +00001824 return ResultReg;
1825}
1826
Stephen Hines37ed9c12014-12-01 14:51:49 -08001827unsigned FastISel::fastEmitInst_rf(unsigned MachineInstOpcode,
1828 const TargetRegisterClass *RC, unsigned Op0,
1829 bool Op0IsKill, const ConstantFP *FPImm) {
1830 const MCInstrDesc &II = TII.get(MachineInstOpcode);
1831
1832 unsigned ResultReg = createResultReg(RC);
1833 Op0 = constrainOperandRegClass(II, Op0, II.getNumDefs());
1834
1835 if (II.getNumDefs() >= 1)
1836 BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc, II, ResultReg)
1837 .addReg(Op0, getKillRegState(Op0IsKill))
1838 .addFPImm(FPImm);
1839 else {
1840 BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc, II)
1841 .addReg(Op0, getKillRegState(Op0IsKill))
1842 .addFPImm(FPImm);
1843 BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc,
1844 TII.get(TargetOpcode::COPY), ResultReg).addReg(II.ImplicitDefs[0]);
1845 }
1846 return ResultReg;
1847}
1848
1849unsigned FastISel::fastEmitInst_rri(unsigned MachineInstOpcode,
1850 const TargetRegisterClass *RC, unsigned Op0,
1851 bool Op0IsKill, unsigned Op1,
1852 bool Op1IsKill, uint64_t Imm) {
Evan Chenge837dea2011-06-28 19:10:37 +00001853 const MCInstrDesc &II = TII.get(MachineInstOpcode);
Dan Gohmand5fe57d2008-08-21 01:41:07 +00001854
Stephen Hinesdce4a402014-05-29 02:49:00 -07001855 unsigned ResultReg = createResultReg(RC);
1856 Op0 = constrainOperandRegClass(II, Op0, II.getNumDefs());
1857 Op1 = constrainOperandRegClass(II, Op1, II.getNumDefs() + 1);
1858
Evan Cheng5960e4e2008-09-08 08:38:20 +00001859 if (II.getNumDefs() >= 1)
Stephen Hines36b56882014-04-23 16:57:46 -07001860 BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc, II, ResultReg)
Stephen Hines37ed9c12014-12-01 14:51:49 -08001861 .addReg(Op0, getKillRegState(Op0IsKill))
1862 .addReg(Op1, getKillRegState(Op1IsKill))
1863 .addImm(Imm);
Evan Cheng5960e4e2008-09-08 08:38:20 +00001864 else {
Stephen Hines36b56882014-04-23 16:57:46 -07001865 BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc, II)
Stephen Hines37ed9c12014-12-01 14:51:49 -08001866 .addReg(Op0, getKillRegState(Op0IsKill))
1867 .addReg(Op1, getKillRegState(Op1IsKill))
1868 .addImm(Imm);
Stephen Hines36b56882014-04-23 16:57:46 -07001869 BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc,
1870 TII.get(TargetOpcode::COPY), ResultReg).addReg(II.ImplicitDefs[0]);
Evan Cheng5960e4e2008-09-08 08:38:20 +00001871 }
Dan Gohmand5fe57d2008-08-21 01:41:07 +00001872 return ResultReg;
1873}
Owen Anderson6d0c25e2008-08-25 20:20:32 +00001874
Stephen Hines37ed9c12014-12-01 14:51:49 -08001875unsigned FastISel::fastEmitInst_rrii(unsigned MachineInstOpcode,
Manman Ren68f25572012-06-01 19:33:18 +00001876 const TargetRegisterClass *RC,
Stephen Hines37ed9c12014-12-01 14:51:49 -08001877 unsigned Op0, bool Op0IsKill, unsigned Op1,
1878 bool Op1IsKill, uint64_t Imm1,
1879 uint64_t Imm2) {
Manman Ren68f25572012-06-01 19:33:18 +00001880 const MCInstrDesc &II = TII.get(MachineInstOpcode);
1881
Stephen Hinesdce4a402014-05-29 02:49:00 -07001882 unsigned ResultReg = createResultReg(RC);
1883 Op0 = constrainOperandRegClass(II, Op0, II.getNumDefs());
1884 Op1 = constrainOperandRegClass(II, Op1, II.getNumDefs() + 1);
1885
Manman Ren68f25572012-06-01 19:33:18 +00001886 if (II.getNumDefs() >= 1)
Stephen Hines36b56882014-04-23 16:57:46 -07001887 BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc, II, ResultReg)
Stephen Hines37ed9c12014-12-01 14:51:49 -08001888 .addReg(Op0, getKillRegState(Op0IsKill))
1889 .addReg(Op1, getKillRegState(Op1IsKill))
1890 .addImm(Imm1)
1891 .addImm(Imm2);
Manman Ren68f25572012-06-01 19:33:18 +00001892 else {
Stephen Hines36b56882014-04-23 16:57:46 -07001893 BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc, II)
Stephen Hines37ed9c12014-12-01 14:51:49 -08001894 .addReg(Op0, getKillRegState(Op0IsKill))
1895 .addReg(Op1, getKillRegState(Op1IsKill))
1896 .addImm(Imm1)
1897 .addImm(Imm2);
Stephen Hines36b56882014-04-23 16:57:46 -07001898 BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc,
1899 TII.get(TargetOpcode::COPY), ResultReg).addReg(II.ImplicitDefs[0]);
Manman Ren68f25572012-06-01 19:33:18 +00001900 }
1901 return ResultReg;
1902}
1903
Stephen Hines37ed9c12014-12-01 14:51:49 -08001904unsigned FastISel::fastEmitInst_i(unsigned MachineInstOpcode,
1905 const TargetRegisterClass *RC, uint64_t Imm) {
Owen Anderson6d0c25e2008-08-25 20:20:32 +00001906 unsigned ResultReg = createResultReg(RC);
Evan Chenge837dea2011-06-28 19:10:37 +00001907 const MCInstrDesc &II = TII.get(MachineInstOpcode);
Wesley Peckbf17cfa2010-11-23 03:31:01 +00001908
Evan Cheng5960e4e2008-09-08 08:38:20 +00001909 if (II.getNumDefs() >= 1)
Stephen Hines37ed9c12014-12-01 14:51:49 -08001910 BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc, II, ResultReg)
1911 .addImm(Imm);
Evan Cheng5960e4e2008-09-08 08:38:20 +00001912 else {
Stephen Hines36b56882014-04-23 16:57:46 -07001913 BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc, II).addImm(Imm);
1914 BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc,
1915 TII.get(TargetOpcode::COPY), ResultReg).addReg(II.ImplicitDefs[0]);
Evan Cheng5960e4e2008-09-08 08:38:20 +00001916 }
Owen Anderson6d0c25e2008-08-25 20:20:32 +00001917 return ResultReg;
Evan Chengb41aec52008-08-25 22:20:39 +00001918}
Owen Anderson8970f002008-08-27 22:30:02 +00001919
Stephen Hines37ed9c12014-12-01 14:51:49 -08001920unsigned FastISel::fastEmitInst_ii(unsigned MachineInstOpcode,
1921 const TargetRegisterClass *RC, uint64_t Imm1,
1922 uint64_t Imm2) {
Owen Andersond74ea772011-04-22 23:38:06 +00001923 unsigned ResultReg = createResultReg(RC);
Evan Chenge837dea2011-06-28 19:10:37 +00001924 const MCInstrDesc &II = TII.get(MachineInstOpcode);
Owen Andersond74ea772011-04-22 23:38:06 +00001925
1926 if (II.getNumDefs() >= 1)
Stephen Hines36b56882014-04-23 16:57:46 -07001927 BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc, II, ResultReg)
Stephen Hines37ed9c12014-12-01 14:51:49 -08001928 .addImm(Imm1)
1929 .addImm(Imm2);
Owen Andersond74ea772011-04-22 23:38:06 +00001930 else {
Stephen Hines37ed9c12014-12-01 14:51:49 -08001931 BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc, II).addImm(Imm1)
1932 .addImm(Imm2);
Stephen Hines36b56882014-04-23 16:57:46 -07001933 BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc,
1934 TII.get(TargetOpcode::COPY), ResultReg).addReg(II.ImplicitDefs[0]);
Owen Andersond74ea772011-04-22 23:38:06 +00001935 }
1936 return ResultReg;
1937}
1938
Stephen Hines37ed9c12014-12-01 14:51:49 -08001939unsigned FastISel::fastEmitInst_extractsubreg(MVT RetVT, unsigned Op0,
1940 bool Op0IsKill, uint32_t Idx) {
Evan Cheng536ab132009-01-22 09:10:11 +00001941 unsigned ResultReg = createResultReg(TLI.getRegClassFor(RetVT));
Jakob Stoklund Olesen0bc25f42010-07-08 16:40:22 +00001942 assert(TargetRegisterInfo::isVirtualRegister(Op0) &&
1943 "Cannot yet extract from physregs");
Jakob Stoklund Olesenee0d5d42012-05-20 06:38:37 +00001944 const TargetRegisterClass *RC = MRI.getRegClass(Op0);
1945 MRI.constrainRegClass(Op0, TRI.getSubClassWithSubReg(RC, Idx));
Stephen Hines37ed9c12014-12-01 14:51:49 -08001946 BuildMI(*FuncInfo.MBB, FuncInfo.InsertPt, DbgLoc, TII.get(TargetOpcode::COPY),
1947 ResultReg).addReg(Op0, getKillRegState(Op0IsKill), Idx);
Owen Anderson8970f002008-08-27 22:30:02 +00001948 return ResultReg;
1949}
Dan Gohman14ea1ec2009-03-13 20:42:20 +00001950
Stephen Hines37ed9c12014-12-01 14:51:49 -08001951/// Emit MachineInstrs to compute the value of Op with all but the least
1952/// significant bit set to zero.
1953unsigned FastISel::fastEmitZExtFromI1(MVT VT, unsigned Op0, bool Op0IsKill) {
1954 return fastEmit_ri(VT, VT, ISD::AND, Op0, Op0IsKill, 1);
Dan Gohman14ea1ec2009-03-13 20:42:20 +00001955}
Dan Gohmanf81eca02010-04-22 20:46:50 +00001956
1957/// HandlePHINodesInSuccessorBlocks - Handle PHI nodes in successor blocks.
1958/// Emit code to ensure constants are copied into registers when needed.
1959/// Remember the virtual registers that need to be added to the Machine PHI
1960/// nodes as input. We cannot just directly add them, because expansion
1961/// might result in multiple MBB's for one BB. As such, the start of the
1962/// BB might correspond to a different MBB than the end.
Stephen Hines37ed9c12014-12-01 14:51:49 -08001963bool FastISel::handlePHINodesInSuccessorBlocks(const BasicBlock *LLVMBB) {
Dan Gohmanf81eca02010-04-22 20:46:50 +00001964 const TerminatorInst *TI = LLVMBB->getTerminator();
1965
1966 SmallPtrSet<MachineBasicBlock *, 4> SuccsHandled;
Stephen Hines37ed9c12014-12-01 14:51:49 -08001967 FuncInfo.OrigNumPHINodesToUpdate = FuncInfo.PHINodesToUpdate.size();
Dan Gohmanf81eca02010-04-22 20:46:50 +00001968
1969 // Check successor nodes' PHI nodes that expect a constant to be available
1970 // from this block.
1971 for (unsigned succ = 0, e = TI->getNumSuccessors(); succ != e; ++succ) {
1972 const BasicBlock *SuccBB = TI->getSuccessor(succ);
Stephen Hines37ed9c12014-12-01 14:51:49 -08001973 if (!isa<PHINode>(SuccBB->begin()))
1974 continue;
Dan Gohmana4160c32010-07-07 16:29:44 +00001975 MachineBasicBlock *SuccMBB = FuncInfo.MBBMap[SuccBB];
Dan Gohmanf81eca02010-04-22 20:46:50 +00001976
1977 // If this terminator has multiple identical successors (common for
1978 // switches), only handle each succ once.
Stephen Hines37ed9c12014-12-01 14:51:49 -08001979 if (!SuccsHandled.insert(SuccMBB).second)
1980 continue;
Dan Gohmanf81eca02010-04-22 20:46:50 +00001981
1982 MachineBasicBlock::iterator MBBI = SuccMBB->begin();
1983
1984 // At this point we know that there is a 1-1 correspondence between LLVM PHI
1985 // nodes and Machine PHI nodes, but the incoming operands have not been
1986 // emitted yet.
1987 for (BasicBlock::const_iterator I = SuccBB->begin();
Stephen Hines37ed9c12014-12-01 14:51:49 -08001988 const auto *PN = dyn_cast<PHINode>(I); ++I) {
Dan Gohmanfb95f892010-05-07 01:10:20 +00001989
Dan Gohmanf81eca02010-04-22 20:46:50 +00001990 // Ignore dead phi's.
Stephen Hines37ed9c12014-12-01 14:51:49 -08001991 if (PN->use_empty())
1992 continue;
Dan Gohmanf81eca02010-04-22 20:46:50 +00001993
1994 // Only handle legal types. Two interesting things to note here. First,
1995 // by bailing out early, we may leave behind some dead instructions,
1996 // since SelectionDAG's HandlePHINodesInSuccessorBlocks will insert its
Chris Lattner7a2bdde2011-04-15 05:18:47 +00001997 // own moves. Second, this check is necessary because FastISel doesn't
Dan Gohman89496d02010-07-02 00:10:16 +00001998 // use CreateRegs to create registers, so it always creates
Dan Gohmanf81eca02010-04-22 20:46:50 +00001999 // exactly one register for each non-void instruction.
2000 EVT VT = TLI.getValueType(PN->getType(), /*AllowUnknown=*/true);
2001 if (VT == MVT::Other || !TLI.isTypeLegal(VT)) {
Chad Rosier2f2d1d72012-02-04 00:39:19 +00002002 // Handle integer promotions, though, because they're common and easy.
Stephen Hines37ed9c12014-12-01 14:51:49 -08002003 if (!(VT == MVT::i1 || VT == MVT::i8 || VT == MVT::i16)) {
2004 FuncInfo.PHINodesToUpdate.resize(FuncInfo.OrigNumPHINodesToUpdate);
Dan Gohmanf81eca02010-04-22 20:46:50 +00002005 return false;
2006 }
2007 }
2008
2009 const Value *PHIOp = PN->getIncomingValueForBlock(LLVMBB);
2010
Dan Gohmanfb95f892010-05-07 01:10:20 +00002011 // Set the DebugLoc for the copy. Prefer the location of the operand
2012 // if there is one; use the location of the PHI otherwise.
Stephen Hines36b56882014-04-23 16:57:46 -07002013 DbgLoc = PN->getDebugLoc();
Stephen Hines37ed9c12014-12-01 14:51:49 -08002014 if (const auto *Inst = dyn_cast<Instruction>(PHIOp))
Stephen Hines36b56882014-04-23 16:57:46 -07002015 DbgLoc = Inst->getDebugLoc();
Dan Gohmanfb95f892010-05-07 01:10:20 +00002016
Dan Gohmanf81eca02010-04-22 20:46:50 +00002017 unsigned Reg = getRegForValue(PHIOp);
Stephen Hines37ed9c12014-12-01 14:51:49 -08002018 if (!Reg) {
2019 FuncInfo.PHINodesToUpdate.resize(FuncInfo.OrigNumPHINodesToUpdate);
Dan Gohmanf81eca02010-04-22 20:46:50 +00002020 return false;
2021 }
Dan Gohmana4160c32010-07-07 16:29:44 +00002022 FuncInfo.PHINodesToUpdate.push_back(std::make_pair(MBBI++, Reg));
Stephen Hines36b56882014-04-23 16:57:46 -07002023 DbgLoc = DebugLoc();
Dan Gohmanf81eca02010-04-22 20:46:50 +00002024 }
2025 }
2026
2027 return true;
2028}
Eli Bendersky75299e32013-04-19 22:29:18 +00002029
2030bool FastISel::tryToFoldLoad(const LoadInst *LI, const Instruction *FoldInst) {
Eli Bendersky462123f2013-04-19 23:26:18 +00002031 assert(LI->hasOneUse() &&
Stephen Hines37ed9c12014-12-01 14:51:49 -08002032 "tryToFoldLoad expected a LoadInst with a single use");
Eli Bendersky75299e32013-04-19 22:29:18 +00002033 // We know that the load has a single use, but don't know what it is. If it
2034 // isn't one of the folded instructions, then we can't succeed here. Handle
2035 // this by scanning the single-use users of the load until we get to FoldInst.
Stephen Hines37ed9c12014-12-01 14:51:49 -08002036 unsigned MaxUsers = 6; // Don't scan down huge single-use chains of instrs.
Eli Bendersky75299e32013-04-19 22:29:18 +00002037
Stephen Hines36b56882014-04-23 16:57:46 -07002038 const Instruction *TheUser = LI->user_back();
Stephen Hines37ed9c12014-12-01 14:51:49 -08002039 while (TheUser != FoldInst && // Scan up until we find FoldInst.
Eli Bendersky75299e32013-04-19 22:29:18 +00002040 // Stay in the right block.
2041 TheUser->getParent() == FoldInst->getParent() &&
Stephen Hines37ed9c12014-12-01 14:51:49 -08002042 --MaxUsers) { // Don't scan too far.
Eli Bendersky75299e32013-04-19 22:29:18 +00002043 // If there are multiple or no uses of this instruction, then bail out.
2044 if (!TheUser->hasOneUse())
2045 return false;
2046
Stephen Hines36b56882014-04-23 16:57:46 -07002047 TheUser = TheUser->user_back();
Eli Bendersky75299e32013-04-19 22:29:18 +00002048 }
2049
2050 // If we didn't find the fold instruction, then we failed to collapse the
2051 // sequence.
2052 if (TheUser != FoldInst)
2053 return false;
2054
2055 // Don't try to fold volatile loads. Target has to deal with alignment
2056 // constraints.
Eli Bendersky462123f2013-04-19 23:26:18 +00002057 if (LI->isVolatile())
2058 return false;
Eli Bendersky75299e32013-04-19 22:29:18 +00002059
2060 // Figure out which vreg this is going into. If there is no assigned vreg yet
2061 // then there actually was no reference to it. Perhaps the load is referenced
2062 // by a dead instruction.
2063 unsigned LoadReg = getRegForValue(LI);
Stephen Hines37ed9c12014-12-01 14:51:49 -08002064 if (!LoadReg)
Eli Bendersky75299e32013-04-19 22:29:18 +00002065 return false;
2066
Eli Bendersky462123f2013-04-19 23:26:18 +00002067 // We can't fold if this vreg has no uses or more than one use. Multiple uses
2068 // may mean that the instruction got lowered to multiple MIs, or the use of
2069 // the loaded value ended up being multiple operands of the result.
2070 if (!MRI.hasOneUse(LoadReg))
2071 return false;
2072
Eli Bendersky75299e32013-04-19 22:29:18 +00002073 MachineRegisterInfo::reg_iterator RI = MRI.reg_begin(LoadReg);
Stephen Hines36b56882014-04-23 16:57:46 -07002074 MachineInstr *User = RI->getParent();
Eli Bendersky75299e32013-04-19 22:29:18 +00002075
2076 // Set the insertion point properly. Folding the load can cause generation of
Eli Bendersky462123f2013-04-19 23:26:18 +00002077 // other random instructions (like sign extends) for addressing modes; make
Eli Bendersky75299e32013-04-19 22:29:18 +00002078 // sure they get inserted in a logical place before the new instruction.
2079 FuncInfo.InsertPt = User;
2080 FuncInfo.MBB = User->getParent();
2081
2082 // Ask the target to try folding the load.
2083 return tryToFoldLoadIntoMI(User, RI.getOperandNo(), LI);
2084}
2085
Bob Wilsoncc705232013-11-15 19:09:27 +00002086bool FastISel::canFoldAddIntoGEP(const User *GEP, const Value *Add) {
2087 // Must be an add.
2088 if (!isa<AddOperator>(Add))
2089 return false;
2090 // Type size needs to match.
Stephen Hines36b56882014-04-23 16:57:46 -07002091 if (DL.getTypeSizeInBits(GEP->getType()) !=
2092 DL.getTypeSizeInBits(Add->getType()))
Bob Wilsoncc705232013-11-15 19:09:27 +00002093 return false;
2094 // Must be in the same basic block.
2095 if (isa<Instruction>(Add) &&
2096 FuncInfo.MBBMap[cast<Instruction>(Add)->getParent()] != FuncInfo.MBB)
2097 return false;
2098 // Must have a constant operand.
2099 return isa<ConstantInt>(cast<AddOperator>(Add)->getOperand(1));
2100}
Eli Bendersky75299e32013-04-19 22:29:18 +00002101
Stephen Hinesc6a4f5e2014-07-21 00:45:20 -07002102MachineMemOperand *
2103FastISel::createMachineMemOperandFor(const Instruction *I) const {
2104 const Value *Ptr;
2105 Type *ValTy;
2106 unsigned Alignment;
2107 unsigned Flags;
2108 bool IsVolatile;
2109
2110 if (const auto *LI = dyn_cast<LoadInst>(I)) {
2111 Alignment = LI->getAlignment();
2112 IsVolatile = LI->isVolatile();
2113 Flags = MachineMemOperand::MOLoad;
2114 Ptr = LI->getPointerOperand();
2115 ValTy = LI->getType();
2116 } else if (const auto *SI = dyn_cast<StoreInst>(I)) {
2117 Alignment = SI->getAlignment();
2118 IsVolatile = SI->isVolatile();
2119 Flags = MachineMemOperand::MOStore;
2120 Ptr = SI->getPointerOperand();
2121 ValTy = SI->getValueOperand()->getType();
Stephen Hines37ed9c12014-12-01 14:51:49 -08002122 } else
Stephen Hinesc6a4f5e2014-07-21 00:45:20 -07002123 return nullptr;
Stephen Hinesc6a4f5e2014-07-21 00:45:20 -07002124
Stephen Hines37ed9c12014-12-01 14:51:49 -08002125 bool IsNonTemporal = I->getMetadata(LLVMContext::MD_nontemporal) != nullptr;
2126 bool IsInvariant = I->getMetadata(LLVMContext::MD_invariant_load) != nullptr;
Stephen Hinesc6a4f5e2014-07-21 00:45:20 -07002127 const MDNode *Ranges = I->getMetadata(LLVMContext::MD_range);
2128
Stephen Hines37ed9c12014-12-01 14:51:49 -08002129 AAMDNodes AAInfo;
2130 I->getAAMetadata(AAInfo);
2131
2132 if (Alignment == 0) // Ensure that codegen never sees alignment 0.
Stephen Hinesc6a4f5e2014-07-21 00:45:20 -07002133 Alignment = DL.getABITypeAlignment(ValTy);
2134
Stephen Hines37ed9c12014-12-01 14:51:49 -08002135 unsigned Size = DL.getTypeStoreSize(ValTy);
Stephen Hinesc6a4f5e2014-07-21 00:45:20 -07002136
2137 if (IsVolatile)
2138 Flags |= MachineMemOperand::MOVolatile;
2139 if (IsNonTemporal)
2140 Flags |= MachineMemOperand::MONonTemporal;
2141 if (IsInvariant)
2142 Flags |= MachineMemOperand::MOInvariant;
2143
2144 return FuncInfo.MF->getMachineMemOperand(MachinePointerInfo(Ptr), Flags, Size,
Stephen Hines37ed9c12014-12-01 14:51:49 -08002145 Alignment, AAInfo, Ranges);
2146}
2147
2148CmpInst::Predicate FastISel::optimizeCmpPredicate(const CmpInst *CI) const {
2149 // If both operands are the same, then try to optimize or fold the cmp.
2150 CmpInst::Predicate Predicate = CI->getPredicate();
2151 if (CI->getOperand(0) != CI->getOperand(1))
2152 return Predicate;
2153
2154 switch (Predicate) {
2155 default: llvm_unreachable("Invalid predicate!");
2156 case CmpInst::FCMP_FALSE: Predicate = CmpInst::FCMP_FALSE; break;
2157 case CmpInst::FCMP_OEQ: Predicate = CmpInst::FCMP_ORD; break;
2158 case CmpInst::FCMP_OGT: Predicate = CmpInst::FCMP_FALSE; break;
2159 case CmpInst::FCMP_OGE: Predicate = CmpInst::FCMP_ORD; break;
2160 case CmpInst::FCMP_OLT: Predicate = CmpInst::FCMP_FALSE; break;
2161 case CmpInst::FCMP_OLE: Predicate = CmpInst::FCMP_ORD; break;
2162 case CmpInst::FCMP_ONE: Predicate = CmpInst::FCMP_FALSE; break;
2163 case CmpInst::FCMP_ORD: Predicate = CmpInst::FCMP_ORD; break;
2164 case CmpInst::FCMP_UNO: Predicate = CmpInst::FCMP_UNO; break;
2165 case CmpInst::FCMP_UEQ: Predicate = CmpInst::FCMP_TRUE; break;
2166 case CmpInst::FCMP_UGT: Predicate = CmpInst::FCMP_UNO; break;
2167 case CmpInst::FCMP_UGE: Predicate = CmpInst::FCMP_TRUE; break;
2168 case CmpInst::FCMP_ULT: Predicate = CmpInst::FCMP_UNO; break;
2169 case CmpInst::FCMP_ULE: Predicate = CmpInst::FCMP_TRUE; break;
2170 case CmpInst::FCMP_UNE: Predicate = CmpInst::FCMP_UNO; break;
2171 case CmpInst::FCMP_TRUE: Predicate = CmpInst::FCMP_TRUE; break;
2172
2173 case CmpInst::ICMP_EQ: Predicate = CmpInst::FCMP_TRUE; break;
2174 case CmpInst::ICMP_NE: Predicate = CmpInst::FCMP_FALSE; break;
2175 case CmpInst::ICMP_UGT: Predicate = CmpInst::FCMP_FALSE; break;
2176 case CmpInst::ICMP_UGE: Predicate = CmpInst::FCMP_TRUE; break;
2177 case CmpInst::ICMP_ULT: Predicate = CmpInst::FCMP_FALSE; break;
2178 case CmpInst::ICMP_ULE: Predicate = CmpInst::FCMP_TRUE; break;
2179 case CmpInst::ICMP_SGT: Predicate = CmpInst::FCMP_FALSE; break;
2180 case CmpInst::ICMP_SGE: Predicate = CmpInst::FCMP_TRUE; break;
2181 case CmpInst::ICMP_SLT: Predicate = CmpInst::FCMP_FALSE; break;
2182 case CmpInst::ICMP_SLE: Predicate = CmpInst::FCMP_TRUE; break;
2183 }
2184
2185 return Predicate;
Stephen Hinesc6a4f5e2014-07-21 00:45:20 -07002186}