blob: fe2ed7934b0cf4b78970bb8bd6e672f83fed4260 [file] [log] [blame]
Hongbin Zheng3b11a162012-04-25 13:16:49 +00001//===--- BlockGenerators.cpp - Generate code for statements -----*- C++ -*-===//
2//
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 implements the BlockGenerator and VectorBlockGenerator classes,
11// which generate sequential code and vectorized code for a polyhedral
12// statement, respectively.
13//
14//===----------------------------------------------------------------------===//
15
16#include "polly/ScopInfo.h"
Hongbin Zheng8a846612012-04-25 13:18:28 +000017#include "polly/CodeGen/BlockGenerators.h"
Tobias Grosser83628182013-05-07 08:11:54 +000018#include "polly/CodeGen/CodeGeneration.h"
Johannes Doerferta63b2572014-08-03 01:51:59 +000019#include "polly/CodeGen/IslExprBuilder.h"
Tobias Grosser86bc93a2015-09-06 08:47:57 +000020#include "polly/CodeGen/RuntimeDebugBuilder.h"
Tobias Grosser637bd632013-05-07 07:31:10 +000021#include "polly/Options.h"
Hongbin Zheng3b11a162012-04-25 13:16:49 +000022#include "polly/Support/GICHelper.h"
Sebastian Pop97cb8132013-03-18 20:21:13 +000023#include "polly/Support/SCEVValidator.h"
Tobias Grosserecfe21b2013-03-20 18:03:18 +000024#include "polly/Support/ScopHelper.h"
Tobias Grossere71c6ab2012-04-27 16:36:14 +000025#include "llvm/Analysis/LoopInfo.h"
Johannes Doerfertf32d6512015-03-01 18:45:58 +000026#include "llvm/Analysis/RegionInfo.h"
Tobias Grossere71c6ab2012-04-27 16:36:14 +000027#include "llvm/Analysis/ScalarEvolution.h"
Tobias Grosser030237d2014-02-21 15:06:05 +000028#include "llvm/IR/IntrinsicInst.h"
Tobias Grosserc9895062015-03-10 15:24:33 +000029#include "llvm/IR/Module.h"
Hongbin Zheng3b11a162012-04-25 13:16:49 +000030#include "llvm/Transforms/Utils/BasicBlockUtils.h"
Tobias Grosser1b9d25a2015-09-27 11:17:22 +000031#include "llvm/Transforms/Utils/Local.h"
Johannes Doerfertf32d6512015-03-01 18:45:58 +000032#include "isl/aff.h"
33#include "isl/ast.h"
Johannes Doerfertf32d6512015-03-01 18:45:58 +000034#include "isl/ast_build.h"
Tobias Grosserba0d0922015-05-09 09:13:42 +000035#include "isl/set.h"
Johannes Doerfertf32d6512015-03-01 18:45:58 +000036#include <deque>
37
Hongbin Zheng3b11a162012-04-25 13:16:49 +000038using namespace llvm;
39using namespace polly;
40
Tobias Grosser878aba42014-10-22 23:22:41 +000041static cl::opt<bool> Aligned("enable-polly-aligned",
42 cl::desc("Assumed aligned memory accesses."),
43 cl::Hidden, cl::init(false), cl::ZeroOrMore,
44 cl::cat(PollyCategory));
Hongbin Zheng3b11a162012-04-25 13:16:49 +000045
Tobias Grosser86bc93a2015-09-06 08:47:57 +000046static cl::opt<bool> DebugPrinting(
47 "polly-codegen-add-debug-printing",
48 cl::desc("Add printf calls that show the values loaded/stored."),
49 cl::Hidden, cl::init(false), cl::ZeroOrMore, cl::cat(PollyCategory));
50
Johannes Doerfertb4f08eb2015-02-23 13:51:35 +000051BlockGenerator::BlockGenerator(PollyIRBuilder &B, LoopInfo &LI,
52 ScalarEvolution &SE, DominatorTree &DT,
Johannes Doerfertecff11d2015-05-22 23:43:58 +000053 ScalarAllocaMapTy &ScalarMap,
54 ScalarAllocaMapTy &PHIOpMap,
55 EscapeUsersAllocaMapTy &EscapeMap,
Tobias Grosserf4bb7a62015-10-04 10:18:32 +000056 ValueMapT &GlobalMap,
Johannes Doerfertb4f08eb2015-02-23 13:51:35 +000057 IslExprBuilder *ExprBuilder)
Johannes Doerfertecff11d2015-05-22 23:43:58 +000058 : Builder(B), LI(LI), SE(SE), ExprBuilder(ExprBuilder), DT(DT),
59 EntryBB(nullptr), PHIOpMap(PHIOpMap), ScalarMap(ScalarMap),
Tobias Grosserbc132602015-09-05 09:56:54 +000060 EscapeMap(EscapeMap), GlobalMap(GlobalMap) {}
Tobias Grossere71c6ab2012-04-27 16:36:14 +000061
Tobias Grosser2f1acac2015-10-04 10:18:45 +000062Value *BlockGenerator::trySynthesizeNewValue(ScopStmt &Stmt, Value *Old,
Tobias Grosser33cb9f92015-09-30 11:56:19 +000063 ValueMapT &BBMap,
64 LoopToScevMapT &LTS,
65 Loop *L) const {
Tobias Grosser683b8e42014-11-30 14:33:31 +000066 if (SE.isSCEVable(Old->getType()))
Tobias Grosser369430f2013-03-22 23:42:53 +000067 if (const SCEV *Scev = SE.getSCEVAtScope(const_cast<Value *>(Old), L)) {
Tobias Grossere71c6ab2012-04-27 16:36:14 +000068 if (!isa<SCEVCouldNotCompute>(Scev)) {
Sebastian Pop637b23d2013-02-15 20:56:01 +000069 const SCEV *NewScev = apply(Scev, LTS, SE);
Johannes Doerfert09e36972015-10-07 20:17:36 +000070 ValueMapT VTV;
Sebastian Pop637b23d2013-02-15 20:56:01 +000071 VTV.insert(BBMap.begin(), BBMap.end());
72 VTV.insert(GlobalMap.begin(), GlobalMap.end());
Johannes Doerferte69e1142015-08-18 11:56:00 +000073
74 Scop &S = *Stmt.getParent();
75 const DataLayout &DL =
76 S.getRegion().getEntry()->getParent()->getParent()->getDataLayout();
77 auto IP = Builder.GetInsertPoint();
78
79 assert(IP != Builder.GetInsertBlock()->end() &&
Tobias Grosser45e79442015-08-01 09:07:57 +000080 "Only instructions can be insert points for SCEVExpander");
Johannes Doerfertc0729a32015-09-30 16:52:03 +000081 Value *Expanded = expandCodeFor(S, SE, DL, "polly", NewScev,
82 Old->getType(), IP, &VTV);
Tobias Grossere71c6ab2012-04-27 16:36:14 +000083
84 BBMap[Old] = Expanded;
85 return Expanded;
86 }
Tobias Grosser369430f2013-03-22 23:42:53 +000087 }
Tobias Grossere71c6ab2012-04-27 16:36:14 +000088
Tobias Grosser33cb9f92015-09-30 11:56:19 +000089 return nullptr;
90}
91
Tobias Grosser2f1acac2015-10-04 10:18:45 +000092Value *BlockGenerator::getNewValue(ScopStmt &Stmt, Value *Old, ValueMapT &BBMap,
93 LoopToScevMapT &LTS, Loop *L) const {
Tobias Grosser33cb9f92015-09-30 11:56:19 +000094 // We assume constants never change.
95 // This avoids map lookups for many calls to this function.
96 if (isa<Constant>(Old))
97 return const_cast<Value *>(Old);
98
99 if (Value *New = GlobalMap.lookup(Old)) {
100 if (Value *NewRemapped = GlobalMap.lookup(New))
101 New = NewRemapped;
102 if (Old->getType()->getScalarSizeInBits() <
103 New->getType()->getScalarSizeInBits())
104 New = Builder.CreateTruncOrBitCast(New, Old->getType());
105
106 return New;
107 }
108
109 if (Value *New = BBMap.lookup(Old))
110 return New;
111
112 if (Value *New = trySynthesizeNewValue(Stmt, Old, BBMap, LTS, L))
113 return New;
114
Tobias Grosser16371ac2014-11-05 20:48:56 +0000115 // A scop-constant value defined by a global or a function parameter.
116 if (isa<GlobalValue>(Old) || isa<Argument>(Old))
117 return const_cast<Value *>(Old);
118
119 // A scop-constant value defined by an instruction executed outside the scop.
120 if (const Instruction *Inst = dyn_cast<Instruction>(Old))
Johannes Doerfertbe9c9112015-02-06 21:39:31 +0000121 if (!Stmt.getParent()->getRegion().contains(Inst->getParent()))
Tobias Grosser16371ac2014-11-05 20:48:56 +0000122 return const_cast<Value *>(Old);
123
124 // The scalar dependence is neither available nor SCEVCodegenable.
Hongbin Zheng5b463ce2013-07-25 09:12:07 +0000125 llvm_unreachable("Unexpected scalar dependence in region!");
Tobias Grosser5a56cbf2014-04-16 07:33:47 +0000126 return nullptr;
Hongbin Zheng3b11a162012-04-25 13:16:49 +0000127}
128
Tobias Grosser2f1acac2015-10-04 10:18:45 +0000129void BlockGenerator::copyInstScalar(ScopStmt &Stmt, Instruction *Inst,
Tobias Grosserbc132602015-09-05 09:56:54 +0000130 ValueMapT &BBMap, LoopToScevMapT &LTS) {
Tobias Grosser030237d2014-02-21 15:06:05 +0000131 // We do not generate debug intrinsics as we did not investigate how to
132 // copy them correctly. At the current state, they just crash the code
133 // generation as the meta-data operands are not correctly copied.
134 if (isa<DbgInfoIntrinsic>(Inst))
135 return;
136
Hongbin Zheng3b11a162012-04-25 13:16:49 +0000137 Instruction *NewInst = Inst->clone();
138
139 // Replace old operands with the new ones.
Tobias Grosser91f5b262014-06-04 08:06:40 +0000140 for (Value *OldOperand : Inst->operands()) {
Tobias Grosserbc132602015-09-05 09:56:54 +0000141 Value *NewOperand =
142 getNewValue(Stmt, OldOperand, BBMap, LTS, getLoopForInst(Inst));
Hongbin Zheng3b11a162012-04-25 13:16:49 +0000143
144 if (!NewOperand) {
Tobias Grosserc14582f2013-02-05 18:01:29 +0000145 assert(!isa<StoreInst>(NewInst) &&
146 "Store instructions are always needed!");
Hongbin Zheng3b11a162012-04-25 13:16:49 +0000147 delete NewInst;
148 return;
149 }
150
151 NewInst->replaceUsesOfWith(OldOperand, NewOperand);
152 }
153
154 Builder.Insert(NewInst);
155 BBMap[Inst] = NewInst;
156
157 if (!NewInst->getType()->isVoidTy())
158 NewInst->setName("p_" + Inst->getName());
159}
160
Johannes Doerfertbe9c9112015-02-06 21:39:31 +0000161Value *BlockGenerator::generateLocationAccessed(
Tobias Grosser2f1acac2015-10-04 10:18:45 +0000162 ScopStmt &Stmt, const Instruction *Inst, Value *Pointer, ValueMapT &BBMap,
163 LoopToScevMapT &LTS, isl_id_to_ast_expr *NewAccesses) {
Johannes Doerfertbe9c9112015-02-06 21:39:31 +0000164 const MemoryAccess &MA = Stmt.getAccessFor(Inst);
Hongbin Zheng3b11a162012-04-25 13:16:49 +0000165
Tobias Grosser2d1ed0b2015-08-27 07:28:16 +0000166 isl_ast_expr *AccessExpr = isl_id_to_ast_expr_get(NewAccesses, MA.getId());
Hongbin Zheng3b11a162012-04-25 13:16:49 +0000167
Tobias Grosser2d1ed0b2015-08-27 07:28:16 +0000168 if (AccessExpr) {
169 AccessExpr = isl_ast_expr_address_of(AccessExpr);
Tobias Grosser98b3ee52015-09-29 06:44:38 +0000170 auto Address = ExprBuilder->create(AccessExpr);
171
172 // Cast the address of this memory access to a pointer type that has the
173 // same element type as the original access, but uses the address space of
174 // the newly generated pointer.
175 auto OldPtrTy = MA.getAccessValue()->getType()->getPointerTo();
176 auto NewPtrTy = Address->getType();
177 OldPtrTy = PointerType::get(OldPtrTy->getElementType(),
178 NewPtrTy->getPointerAddressSpace());
179
180 if (OldPtrTy != NewPtrTy) {
181 assert(OldPtrTy->getPointerElementType()->getPrimitiveSizeInBits() ==
182 NewPtrTy->getPointerElementType()->getPrimitiveSizeInBits() &&
183 "Pointer types to elements with different size found");
184 Address = Builder.CreateBitOrPointerCast(Address, OldPtrTy);
185 }
186 return Address;
Tobias Grosser2d1ed0b2015-08-27 07:28:16 +0000187 }
188
Tobias Grosserbc132602015-09-05 09:56:54 +0000189 return getNewValue(Stmt, Pointer, BBMap, LTS, getLoopForInst(Inst));
Hongbin Zheng3b11a162012-04-25 13:16:49 +0000190}
191
Tobias Grosser4d96c8d2013-03-23 01:05:07 +0000192Loop *BlockGenerator::getLoopForInst(const llvm::Instruction *Inst) {
Johannes Doerfert2ef3f4f2014-08-07 17:14:54 +0000193 return LI.getLoopFor(Inst->getParent());
Tobias Grosser369430f2013-03-22 23:42:53 +0000194}
195
Tobias Grosser2f1acac2015-10-04 10:18:45 +0000196Value *BlockGenerator::generateScalarLoad(ScopStmt &Stmt, LoadInst *Load,
Tobias Grosserbc132602015-09-05 09:56:54 +0000197 ValueMapT &BBMap, LoopToScevMapT &LTS,
Tobias Grosser2d1ed0b2015-08-27 07:28:16 +0000198 isl_id_to_ast_expr *NewAccesses) {
Johannes Doerfertc1db67e2015-09-29 23:47:21 +0000199 if (Value *PreloadLoad = GlobalMap.lookup(Load))
200 return PreloadLoad;
201
Tobias Grosser9646e3f2015-10-04 10:18:39 +0000202 auto *Pointer = Load->getPointerOperand();
Tobias Grosserbc132602015-09-05 09:56:54 +0000203 Value *NewPointer =
204 generateLocationAccessed(Stmt, Load, Pointer, BBMap, LTS, NewAccesses);
Johannes Doerfert87901452014-10-02 16:22:19 +0000205 Value *ScalarLoad = Builder.CreateAlignedLoad(
206 NewPointer, Load->getAlignment(), Load->getName() + "_p_scalar_");
Tobias Grosser86bc93a2015-09-06 08:47:57 +0000207
208 if (DebugPrinting)
209 RuntimeDebugBuilder::createCPUPrinter(Builder, "Load from ", NewPointer,
210 ": ", ScalarLoad, "\n");
211
Hongbin Zheng3b11a162012-04-25 13:16:49 +0000212 return ScalarLoad;
213}
214
Tobias Grosser2f1acac2015-10-04 10:18:45 +0000215void BlockGenerator::generateScalarStore(ScopStmt &Stmt, StoreInst *Store,
Tobias Grosserbc132602015-09-05 09:56:54 +0000216 ValueMapT &BBMap, LoopToScevMapT &LTS,
Tobias Grosser2d1ed0b2015-08-27 07:28:16 +0000217 isl_id_to_ast_expr *NewAccesses) {
Tobias Grosser9646e3f2015-10-04 10:18:39 +0000218 auto *Pointer = Store->getPointerOperand();
Tobias Grosserbc132602015-09-05 09:56:54 +0000219 Value *NewPointer =
220 generateLocationAccessed(Stmt, Store, Pointer, BBMap, LTS, NewAccesses);
221 Value *ValueOperand = getNewValue(Stmt, Store->getValueOperand(), BBMap, LTS,
222 getLoopForInst(Store));
Hongbin Zheng3b11a162012-04-25 13:16:49 +0000223
Tobias Grosser86bc93a2015-09-06 08:47:57 +0000224 if (DebugPrinting)
225 RuntimeDebugBuilder::createCPUPrinter(Builder, "Store to ", NewPointer,
226 ": ", ValueOperand, "\n");
227
Tobias Grosserc186ac72015-08-11 08:13:15 +0000228 Builder.CreateAlignedStore(ValueOperand, NewPointer, Store->getAlignment());
Hongbin Zheng3b11a162012-04-25 13:16:49 +0000229}
230
Tobias Grosser2f1acac2015-10-04 10:18:45 +0000231void BlockGenerator::copyInstruction(ScopStmt &Stmt, Instruction *Inst,
Tobias Grosserbc132602015-09-05 09:56:54 +0000232 ValueMapT &BBMap, LoopToScevMapT &LTS,
Tobias Grosser2d1ed0b2015-08-27 07:28:16 +0000233 isl_id_to_ast_expr *NewAccesses) {
Hongbin Zheng3b11a162012-04-25 13:16:49 +0000234 // Terminator instructions control the control flow. They are explicitly
235 // expressed in the clast and do not need to be copied.
236 if (Inst->isTerminator())
237 return;
238
Johannes Doerfertecff11d2015-05-22 23:43:58 +0000239 Loop *L = getLoopForInst(Inst);
240 if ((Stmt.isBlockStmt() || !Stmt.getRegion()->contains(L)) &&
241 canSynthesize(Inst, &LI, &SE, &Stmt.getParent()->getRegion())) {
Tobias Grosseraff56c82015-09-30 13:36:54 +0000242 // Synthesizable statements will be generated on-demand.
Tobias Grossere71c6ab2012-04-27 16:36:14 +0000243 return;
Johannes Doerfertecff11d2015-05-22 23:43:58 +0000244 }
Tobias Grossere71c6ab2012-04-27 16:36:14 +0000245
Tobias Grosser9646e3f2015-10-04 10:18:39 +0000246 if (auto *Load = dyn_cast<LoadInst>(Inst)) {
Tobias Grosserbc132602015-09-05 09:56:54 +0000247 Value *NewLoad = generateScalarLoad(Stmt, Load, BBMap, LTS, NewAccesses);
Sebastian Pop3d94fed2013-05-24 18:46:02 +0000248 // Compute NewLoad before its insertion in BBMap to make the insertion
249 // deterministic.
Sebastian Pop753d43f2013-05-24 17:16:02 +0000250 BBMap[Load] = NewLoad;
Hongbin Zheng3b11a162012-04-25 13:16:49 +0000251 return;
252 }
253
Tobias Grosser9646e3f2015-10-04 10:18:39 +0000254 if (auto *Store = dyn_cast<StoreInst>(Inst)) {
Tobias Grosserbc132602015-09-05 09:56:54 +0000255 generateScalarStore(Stmt, Store, BBMap, LTS, NewAccesses);
Hongbin Zheng3b11a162012-04-25 13:16:49 +0000256 return;
257 }
258
Tobias Grosser9646e3f2015-10-04 10:18:39 +0000259 if (auto *PHI = dyn_cast<PHINode>(Inst)) {
Tobias Grosserbc132602015-09-05 09:56:54 +0000260 copyPHIInstruction(Stmt, PHI, BBMap, LTS);
Johannes Doerfertecff11d2015-05-22 23:43:58 +0000261 return;
262 }
263
Johannes Doerfert3f500fa2015-01-25 18:07:30 +0000264 // Skip some special intrinsics for which we do not adjust the semantics to
265 // the new schedule. All others are handled like every other instruction.
Tobias Grosser9c0ffe32015-08-30 16:57:15 +0000266 if (isIgnoredIntrinsic(Inst))
267 return;
Johannes Doerfert3f500fa2015-01-25 18:07:30 +0000268
Tobias Grosserbc132602015-09-05 09:56:54 +0000269 copyInstScalar(Stmt, Inst, BBMap, LTS);
Hongbin Zheng3b11a162012-04-25 13:16:49 +0000270}
271
Tobias Grosserbc132602015-09-05 09:56:54 +0000272void BlockGenerator::copyStmt(ScopStmt &Stmt, LoopToScevMapT &LTS,
Tobias Grosser2d1ed0b2015-08-27 07:28:16 +0000273 isl_id_to_ast_expr *NewAccesses) {
Johannes Doerfert275a1752015-02-24 16:16:32 +0000274 assert(Stmt.isBlockStmt() &&
275 "Only block statements can be copied by the block generator");
276
277 ValueMapT BBMap;
278
Johannes Doerfertbe9c9112015-02-06 21:39:31 +0000279 BasicBlock *BB = Stmt.getBasicBlock();
Tobias Grosserbc132602015-09-05 09:56:54 +0000280 copyBB(Stmt, BB, BBMap, LTS, NewAccesses);
Johannes Doerfert275a1752015-02-24 16:16:32 +0000281}
282
Johannes Doerfert514f6ef2015-02-27 18:29:04 +0000283BasicBlock *BlockGenerator::splitBB(BasicBlock *BB) {
Tobias Grosserc14582f2013-02-05 18:01:29 +0000284 BasicBlock *CopyBB =
Johannes Doerfertb4f08eb2015-02-23 13:51:35 +0000285 SplitBlock(Builder.GetInsertBlock(), Builder.GetInsertPoint(), &DT, &LI);
Hongbin Zheng3b11a162012-04-25 13:16:49 +0000286 CopyBB->setName("polly.stmt." + BB->getName());
Johannes Doerfert514f6ef2015-02-27 18:29:04 +0000287 return CopyBB;
288}
Hongbin Zheng3b11a162012-04-25 13:16:49 +0000289
Johannes Doerfert514f6ef2015-02-27 18:29:04 +0000290BasicBlock *BlockGenerator::copyBB(ScopStmt &Stmt, BasicBlock *BB,
Tobias Grosserbc132602015-09-05 09:56:54 +0000291 ValueMapT &BBMap, LoopToScevMapT &LTS,
Tobias Grosser2d1ed0b2015-08-27 07:28:16 +0000292 isl_id_to_ast_expr *NewAccesses) {
Johannes Doerfert514f6ef2015-02-27 18:29:04 +0000293 BasicBlock *CopyBB = splitBB(BB);
Michael Kruse225f0d12015-10-17 21:36:00 +0000294 Builder.SetInsertPoint(CopyBB->begin());
295 generateScalarLoads(Stmt, BBMap);
296
Tobias Grosserbc132602015-09-05 09:56:54 +0000297 copyBB(Stmt, BB, CopyBB, BBMap, LTS, NewAccesses);
Michael Kruse225f0d12015-10-17 21:36:00 +0000298
299 // After a basic block was copied store all scalars that escape this block in
300 // their alloca.
301 generateScalarStores(Stmt, LTS, BBMap);
Johannes Doerfert514f6ef2015-02-27 18:29:04 +0000302 return CopyBB;
303}
304
305void BlockGenerator::copyBB(ScopStmt &Stmt, BasicBlock *BB, BasicBlock *CopyBB,
Tobias Grosserbc132602015-09-05 09:56:54 +0000306 ValueMapT &BBMap, LoopToScevMapT &LTS,
Tobias Grosser2d1ed0b2015-08-27 07:28:16 +0000307 isl_id_to_ast_expr *NewAccesses) {
Johannes Doerfertecff11d2015-05-22 23:43:58 +0000308 EntryBB = &CopyBB->getParent()->getEntryBlock();
309
Tobias Grosser91f5b262014-06-04 08:06:40 +0000310 for (Instruction &Inst : *BB)
Tobias Grosserbc132602015-09-05 09:56:54 +0000311 copyInstruction(Stmt, &Inst, BBMap, LTS, NewAccesses);
Johannes Doerfertecff11d2015-05-22 23:43:58 +0000312}
313
Tobias Grosser64c0ff42015-08-31 05:52:24 +0000314Value *BlockGenerator::getOrCreateAlloca(Value *ScalarBase,
315 ScalarAllocaMapTy &Map,
Tobias Grosser64c0ff42015-08-31 05:52:24 +0000316 const char *NameExt) {
Johannes Doerfertecff11d2015-05-22 23:43:58 +0000317 // If no alloca was found create one and insert it in the entry block.
Tobias Grossere9cb5a02015-10-03 17:19:49 +0000318 if (!Map.count(ScalarBase)) {
Johannes Doerfertecff11d2015-05-22 23:43:58 +0000319 auto *Ty = ScalarBase->getType();
Tobias Grosserb09455d2015-09-18 06:01:11 +0000320 auto NewAddr = new AllocaInst(Ty, ScalarBase->getName() + NameExt);
Tobias Grosser64c0ff42015-08-31 05:52:24 +0000321 EntryBB = &Builder.GetInsertBlock()->getParent()->getEntryBlock();
Tobias Grosserb09455d2015-09-18 06:01:11 +0000322 NewAddr->insertBefore(EntryBB->getFirstInsertionPt());
Tobias Grossere9cb5a02015-10-03 17:19:49 +0000323 Map[ScalarBase] = NewAddr;
Johannes Doerfertecff11d2015-05-22 23:43:58 +0000324 }
325
Tobias Grossere9cb5a02015-10-03 17:19:49 +0000326 auto Addr = Map[ScalarBase];
327
Tobias Grosserbc132602015-09-05 09:56:54 +0000328 if (GlobalMap.count(Addr))
329 return GlobalMap[Addr];
Tobias Grosser64c0ff42015-08-31 05:52:24 +0000330
Johannes Doerfertecff11d2015-05-22 23:43:58 +0000331 return Addr;
332}
333
Tobias Grosserbc132602015-09-05 09:56:54 +0000334Value *BlockGenerator::getOrCreateAlloca(MemoryAccess &Access) {
Tobias Grosserb8d27aa2015-10-18 00:51:13 +0000335 if (Access.isPHI())
336 return getOrCreatePHIAlloca(Access.getBaseAddr());
337 else
338 return getOrCreateScalarAlloca(Access.getBaseAddr());
Tobias Grossera4f09882015-10-17 22:16:00 +0000339}
340
341Value *BlockGenerator::getOrCreateAlloca(const ScopArrayInfo *Array) {
342 if (Array->isPHI())
343 return getOrCreatePHIAlloca(Array->getBasePtr());
Tobias Grosserf8d55f72015-08-29 18:12:03 +0000344 else
Tobias Grossera4f09882015-10-17 22:16:00 +0000345 return getOrCreateScalarAlloca(Array->getBasePtr());
Tobias Grosserf8d55f72015-08-29 18:12:03 +0000346}
347
Tobias Grosserbc132602015-09-05 09:56:54 +0000348Value *BlockGenerator::getOrCreateScalarAlloca(Value *ScalarBase) {
349 return getOrCreateAlloca(ScalarBase, ScalarMap, ".s2a");
Tobias Grosserb79a67d2015-08-28 08:23:35 +0000350}
351
Tobias Grosserbc132602015-09-05 09:56:54 +0000352Value *BlockGenerator::getOrCreatePHIAlloca(Value *ScalarBase) {
353 return getOrCreateAlloca(ScalarBase, PHIOpMap, ".phiops");
Tobias Grosserb79a67d2015-08-28 08:23:35 +0000354}
355
Tobias Grosserbc132602015-09-05 09:56:54 +0000356void BlockGenerator::handleOutsideUsers(const Region &R, Instruction *Inst,
Tobias Grosser26e59ee2015-10-17 08:25:54 +0000357 Value *Address) {
Tobias Grosser29854002015-08-30 15:03:59 +0000358 // If there are escape users we get the alloca for this instruction and put it
359 // in the EscapeMap for later finalization. Lastly, if the instruction was
360 // copied multiple times we already did this and can exit.
Michael Kruseacb6ade2015-08-18 17:25:48 +0000361 if (EscapeMap.count(Inst))
362 return;
Johannes Doerferte69e1142015-08-18 11:56:00 +0000363
Johannes Doerfertecff11d2015-05-22 23:43:58 +0000364 EscapeUserVectorTy EscapeUsers;
365 for (User *U : Inst->users()) {
366
367 // Non-instruction user will never escape.
368 Instruction *UI = dyn_cast<Instruction>(U);
369 if (!UI)
370 continue;
371
Johannes Doerfertddb83d02015-08-16 08:35:40 +0000372 if (R.contains(UI))
Johannes Doerfertecff11d2015-05-22 23:43:58 +0000373 continue;
374
375 EscapeUsers.push_back(UI);
376 }
377
378 // Exit if no escape uses were found.
379 if (EscapeUsers.empty())
380 return;
381
Johannes Doerfertecff11d2015-05-22 23:43:58 +0000382 // Get or create an escape alloca for this instruction.
Tobias Grosserb09455d2015-09-18 06:01:11 +0000383 auto *ScalarAddr = Address ? Address : getOrCreateScalarAlloca(Inst);
Johannes Doerfertecff11d2015-05-22 23:43:58 +0000384
385 // Remember that this instruction has escape uses and the escape alloca.
386 EscapeMap[Inst] = std::make_pair(ScalarAddr, std::move(EscapeUsers));
Johannes Doerfertecff11d2015-05-22 23:43:58 +0000387}
388
Michael Kruse225f0d12015-10-17 21:36:00 +0000389void BlockGenerator::generateScalarLoads(ScopStmt &Stmt, ValueMapT &BBMap) {
390 for (MemoryAccess *MA : Stmt) {
391 if (MA->isExplicit() || MA->isWrite())
Tobias Grosserd4dd6ec2015-07-27 17:57:58 +0000392 continue;
Johannes Doerfertecff11d2015-05-22 23:43:58 +0000393
Michael Krusee2bccbb2015-09-18 19:59:43 +0000394 auto *Address = getOrCreateAlloca(*MA);
395 BBMap[MA->getBaseAddr()] =
Tobias Grosser2fc50df2015-08-30 19:51:01 +0000396 Builder.CreateLoad(Address, Address->getName() + ".reload");
Johannes Doerfertecff11d2015-05-22 23:43:58 +0000397 }
398}
399
400Value *BlockGenerator::getNewScalarValue(Value *ScalarValue, const Region &R,
Tobias Grosseraff56c82015-09-30 13:36:54 +0000401 ScopStmt &Stmt, LoopToScevMapT &LTS,
Tobias Grosserbc132602015-09-05 09:56:54 +0000402 ValueMapT &BBMap) {
Johannes Doerfertecff11d2015-05-22 23:43:58 +0000403 // If the value we want to store is an instruction we might have demoted it
404 // in order to make it accessible here. In such a case a reload is
405 // necessary. If it is no instruction it will always be a value that
406 // dominates the current point and we can just use it. In total there are 4
407 // options:
408 // (1) The value is no instruction ==> use the value.
409 // (2) The value is an instruction that was split out of the region prior to
410 // code generation ==> use the instruction as it dominates the region.
411 // (3) The value is an instruction:
412 // (a) The value was defined in the current block, thus a copy is in
413 // the BBMap ==> use the mapped value.
414 // (b) The value was defined in a previous block, thus we demoted it
415 // earlier ==> use the reloaded value.
416 Instruction *ScalarValueInst = dyn_cast<Instruction>(ScalarValue);
417 if (!ScalarValueInst)
418 return ScalarValue;
419
420 if (!R.contains(ScalarValueInst)) {
421 if (Value *ScalarValueCopy = GlobalMap.lookup(ScalarValueInst))
422 return /* Case (3a) */ ScalarValueCopy;
423 else
424 return /* Case 2 */ ScalarValue;
425 }
426
427 if (Value *ScalarValueCopy = BBMap.lookup(ScalarValueInst))
428 return /* Case (3a) */ ScalarValueCopy;
429
Tobias Grosseraff56c82015-09-30 13:36:54 +0000430 if ((Stmt.isBlockStmt() &&
431 Stmt.getBasicBlock() == ScalarValueInst->getParent()) ||
432 (Stmt.isRegionStmt() && Stmt.getRegion()->contains(ScalarValueInst))) {
433 auto SynthesizedValue = trySynthesizeNewValue(
434 Stmt, ScalarValueInst, BBMap, LTS, getLoopForInst(ScalarValueInst));
435
436 if (SynthesizedValue)
437 return SynthesizedValue;
438 }
439
Johannes Doerfertecff11d2015-05-22 23:43:58 +0000440 // Case (3b)
Tobias Grosserbc132602015-09-05 09:56:54 +0000441 Value *Address = getOrCreateScalarAlloca(ScalarValueInst);
Tobias Grosserb649e262015-08-30 17:37:55 +0000442 ScalarValue = Builder.CreateLoad(Address, Address->getName() + ".reload");
Johannes Doerfertecff11d2015-05-22 23:43:58 +0000443
444 return ScalarValue;
445}
446
Michael Kruse225f0d12015-10-17 21:36:00 +0000447void BlockGenerator::generateScalarStores(ScopStmt &Stmt, LoopToScevMapT &LTS,
Tobias Grosserbc132602015-09-05 09:56:54 +0000448 ValueMapT &BBMap) {
Johannes Doerfertecff11d2015-05-22 23:43:58 +0000449 const Region &R = Stmt.getParent()->getRegion();
450
Michael Kruse225f0d12015-10-17 21:36:00 +0000451 assert(Stmt.isBlockStmt() && "Region statements need to use the "
452 "generateScalarStores() function in the "
453 "RegionGenerator");
Johannes Doerfertecff11d2015-05-22 23:43:58 +0000454
Johannes Doerfertecff11d2015-05-22 23:43:58 +0000455 for (MemoryAccess *MA : Stmt) {
Michael Kruse8d0b7342015-09-25 21:21:00 +0000456 if (MA->isExplicit() || MA->isRead())
Johannes Doerfertecff11d2015-05-22 23:43:58 +0000457 continue;
458
Johannes Doerfertd86f2152015-08-17 10:58:17 +0000459 Value *Val = MA->getAccessValue();
Tobias Grosserbc132602015-09-05 09:56:54 +0000460 auto *Address = getOrCreateAlloca(*MA);
Johannes Doerfertd86f2152015-08-17 10:58:17 +0000461
Tobias Grosseraff56c82015-09-30 13:36:54 +0000462 Val = getNewScalarValue(Val, R, Stmt, LTS, BBMap);
Tobias Grosser92245222015-07-28 14:53:44 +0000463 Builder.CreateStore(Val, Address);
Johannes Doerfertecff11d2015-05-22 23:43:58 +0000464 }
465}
466
Tobias Grosserc0091a72015-08-30 19:19:34 +0000467void BlockGenerator::createScalarInitialization(Scop &S) {
468 Region &R = S.getRegion();
Johannes Doerfertecff11d2015-05-22 23:43:58 +0000469 // The split block __just before__ the region and optimized region.
470 BasicBlock *SplitBB = R.getEnteringBlock();
471 BranchInst *SplitBBTerm = cast<BranchInst>(SplitBB->getTerminator());
472 assert(SplitBBTerm->getNumSuccessors() == 2 && "Bad region entering block!");
473
474 // Get the start block of the __optimized__ region.
475 BasicBlock *StartBB = SplitBBTerm->getSuccessor(0);
476 if (StartBB == R.getEntry())
477 StartBB = SplitBBTerm->getSuccessor(1);
478
Johannes Doerfertfb19dd62015-09-26 20:57:59 +0000479 Builder.SetInsertPoint(StartBB->getTerminator());
Johannes Doerfertecff11d2015-05-22 23:43:58 +0000480
Tobias Grosserc0091a72015-08-30 19:19:34 +0000481 for (auto &Pair : S.arrays()) {
482 auto &Array = Pair.second;
483 if (Array->getNumberOfDimensions() != 0)
484 continue;
485 if (Array->isPHI()) {
486 // For PHI nodes, the only values we need to store are the ones that
487 // reach the PHI node from outside the region. In general there should
488 // only be one such incoming edge and this edge should enter through
489 // 'SplitBB'.
490 auto PHI = cast<PHINode>(Array->getBasePtr());
Johannes Doerfertecff11d2015-05-22 23:43:58 +0000491
Tobias Grosserc0091a72015-08-30 19:19:34 +0000492 for (auto BI = PHI->block_begin(), BE = PHI->block_end(); BI != BE; BI++)
493 if (!R.contains(*BI) && *BI != SplitBB)
494 llvm_unreachable("Incoming edges from outside the scop should always "
495 "come from SplitBB");
496
497 int Idx = PHI->getBasicBlockIndex(SplitBB);
498 if (Idx < 0)
499 continue;
500
501 Value *ScalarValue = PHI->getIncomingValue(Idx);
502
Tobias Grosserbc132602015-09-05 09:56:54 +0000503 Builder.CreateStore(ScalarValue, getOrCreatePHIAlloca(PHI));
Tobias Grosserc0091a72015-08-30 19:19:34 +0000504 continue;
505 }
506
507 auto *Inst = dyn_cast<Instruction>(Array->getBasePtr());
508
509 if (Inst && R.contains(Inst))
Johannes Doerfertecff11d2015-05-22 23:43:58 +0000510 continue;
511
Johannes Doerfert717b8662015-09-08 21:44:27 +0000512 // PHI nodes that are not marked as such in their SAI object are exit PHI
513 // nodes we model as common scalars but do not need to initialize.
514 if (Inst && isa<PHINode>(Inst))
515 continue;
516
Tobias Grosser64c0ff42015-08-31 05:52:24 +0000517 ValueMapT EmptyMap;
Tobias Grosserc0091a72015-08-30 19:19:34 +0000518 Builder.CreateStore(Array->getBasePtr(),
Tobias Grosserbc132602015-09-05 09:56:54 +0000519 getOrCreateScalarAlloca(Array->getBasePtr()));
Johannes Doerfertecff11d2015-05-22 23:43:58 +0000520 }
521}
522
523void BlockGenerator::createScalarFinalization(Region &R) {
524 // The exit block of the __unoptimized__ region.
525 BasicBlock *ExitBB = R.getExitingBlock();
526 // The merge block __just after__ the region and the optimized region.
527 BasicBlock *MergeBB = R.getExit();
528
529 // The exit block of the __optimized__ region.
530 BasicBlock *OptExitBB = *(pred_begin(MergeBB));
531 if (OptExitBB == ExitBB)
532 OptExitBB = *(++pred_begin(MergeBB));
533
534 Builder.SetInsertPoint(OptExitBB->getTerminator());
535 for (const auto &EscapeMapping : EscapeMap) {
536 // Extract the escaping instruction and the escaping users as well as the
537 // alloca the instruction was demoted to.
538 Instruction *EscapeInst = EscapeMapping.getFirst();
539 const auto &EscapeMappingValue = EscapeMapping.getSecond();
540 const EscapeUserVectorTy &EscapeUsers = EscapeMappingValue.second;
Tobias Grosser64c0ff42015-08-31 05:52:24 +0000541 Value *ScalarAddr = EscapeMappingValue.first;
Johannes Doerfertecff11d2015-05-22 23:43:58 +0000542
543 // Reload the demoted instruction in the optimized version of the SCoP.
Johannes Doerfertd8b6ad22015-10-18 12:36:42 +0000544 Value *EscapeInstReload =
Johannes Doerfertecff11d2015-05-22 23:43:58 +0000545 Builder.CreateLoad(ScalarAddr, EscapeInst->getName() + ".final_reload");
Johannes Doerfertd8b6ad22015-10-18 12:36:42 +0000546 EscapeInstReload =
547 Builder.CreateBitOrPointerCast(EscapeInstReload, EscapeInst->getType());
Johannes Doerfertecff11d2015-05-22 23:43:58 +0000548
549 // Create the merge PHI that merges the optimized and unoptimized version.
550 PHINode *MergePHI = PHINode::Create(EscapeInst->getType(), 2,
551 EscapeInst->getName() + ".merge");
552 MergePHI->insertBefore(MergeBB->getFirstInsertionPt());
553
554 // Add the respective values to the merge PHI.
555 MergePHI->addIncoming(EscapeInstReload, OptExitBB);
556 MergePHI->addIncoming(EscapeInst, ExitBB);
557
558 // The information of scalar evolution about the escaping instruction needs
559 // to be revoked so the new merged instruction will be used.
560 if (SE.isSCEVable(EscapeInst->getType()))
561 SE.forgetValue(EscapeInst);
562
563 // Replace all uses of the demoted instruction with the merge PHI.
564 for (Instruction *EUser : EscapeUsers)
565 EUser->replaceUsesOfWith(EscapeInst, MergePHI);
566 }
567}
568
Tobias Grosserffa24462015-10-17 08:54:13 +0000569void BlockGenerator::findOutsideUsers(Scop &S) {
570 auto &R = S.getRegion();
Johannes Doerfert717b8662015-09-08 21:44:27 +0000571
572 // Handle PHI nodes that were in the original exit and are now
573 // moved into the region exiting block.
574 if (!S.hasSingleExitEdge()) {
575 for (Instruction &I : *S.getRegion().getExitingBlock()) {
576 PHINode *PHI = dyn_cast<PHINode>(&I);
577 if (!PHI)
578 break;
579
580 assert(PHI->getNumUses() == 1);
581 assert(ScalarMap.count(PHI->user_back()));
582
Tobias Grosser26e59ee2015-10-17 08:25:54 +0000583 handleOutsideUsers(S.getRegion(), PHI, ScalarMap[PHI->user_back()]);
Johannes Doerfert717b8662015-09-08 21:44:27 +0000584 }
585 }
586
Tobias Grosserffa24462015-10-17 08:54:13 +0000587 for (auto &Pair : S.arrays()) {
588 auto &Array = Pair.second;
589
590 if (Array->getNumberOfDimensions() != 0)
591 continue;
592
593 if (Array->isPHI())
594 continue;
595
596 auto *Inst = dyn_cast<Instruction>(Array->getBasePtr());
597
598 if (!Inst)
599 continue;
600
601 // Scop invariant hoisting moves some of the base pointers out of the scop.
602 // We can ignore these, as the invariant load hoisting already registers the
603 // relevant outside users.
604 if (!R.contains(Inst))
605 continue;
606
607 handleOutsideUsers(R, Inst, nullptr);
608 }
609}
610
611void BlockGenerator::finalizeSCoP(Scop &S) {
612 findOutsideUsers(S);
Tobias Grosserc0091a72015-08-30 19:19:34 +0000613 createScalarInitialization(S);
Johannes Doerfertecff11d2015-05-22 23:43:58 +0000614 createScalarFinalization(S.getRegion());
Hongbin Zheng3b11a162012-04-25 13:16:49 +0000615}
616
Johannes Doerfertbe9c9112015-02-06 21:39:31 +0000617VectorBlockGenerator::VectorBlockGenerator(BlockGenerator &BlockGen,
Johannes Doerfertbe9c9112015-02-06 21:39:31 +0000618 std::vector<LoopToScevMapT> &VLTS,
619 isl_map *Schedule)
Tobias Grosserbc132602015-09-05 09:56:54 +0000620 : BlockGenerator(BlockGen), VLTS(VLTS), Schedule(Schedule) {
Sebastian Popa00a0292012-12-18 07:46:06 +0000621 assert(Schedule && "No statement domain provided");
622}
Hongbin Zheng3b11a162012-04-25 13:16:49 +0000623
Tobias Grosser2f1acac2015-10-04 10:18:45 +0000624Value *VectorBlockGenerator::getVectorValue(ScopStmt &Stmt, Value *Old,
Tobias Grossere602a072013-05-07 07:30:56 +0000625 ValueMapT &VectorMap,
626 VectorValueMapT &ScalarMaps,
627 Loop *L) {
Hongbin Zhengfe11e282013-06-29 13:22:15 +0000628 if (Value *NewValue = VectorMap.lookup(Old))
629 return NewValue;
Hongbin Zheng3b11a162012-04-25 13:16:49 +0000630
631 int Width = getVectorWidth();
632
633 Value *Vector = UndefValue::get(VectorType::get(Old->getType(), Width));
634
635 for (int Lane = 0; Lane < Width; Lane++)
Tobias Grosserc14582f2013-02-05 18:01:29 +0000636 Vector = Builder.CreateInsertElement(
Tobias Grosserbc132602015-09-05 09:56:54 +0000637 Vector, getNewValue(Stmt, Old, ScalarMaps[Lane], VLTS[Lane], L),
Tobias Grosser7242ad92013-02-22 08:07:06 +0000638 Builder.getInt32(Lane));
Hongbin Zheng3b11a162012-04-25 13:16:49 +0000639
640 VectorMap[Old] = Vector;
641
642 return Vector;
643}
644
645Type *VectorBlockGenerator::getVectorPtrTy(const Value *Val, int Width) {
646 PointerType *PointerTy = dyn_cast<PointerType>(Val->getType());
647 assert(PointerTy && "PointerType expected");
648
649 Type *ScalarType = PointerTy->getElementType();
650 VectorType *VectorType = VectorType::get(ScalarType, Width);
651
652 return PointerType::getUnqual(VectorType);
653}
654
Johannes Doerfertbe9c9112015-02-06 21:39:31 +0000655Value *VectorBlockGenerator::generateStrideOneLoad(
Tobias Grosser2f1acac2015-10-04 10:18:45 +0000656 ScopStmt &Stmt, LoadInst *Load, VectorValueMapT &ScalarMaps,
Tobias Grosser2d1ed0b2015-08-27 07:28:16 +0000657 __isl_keep isl_id_to_ast_expr *NewAccesses, bool NegativeStride = false) {
Tobias Grosser0dd463f2014-03-19 19:27:24 +0000658 unsigned VectorWidth = getVectorWidth();
Tobias Grosser9646e3f2015-10-04 10:18:39 +0000659 auto *Pointer = Load->getPointerOperand();
Tobias Grosser0dd463f2014-03-19 19:27:24 +0000660 Type *VectorPtrType = getVectorPtrTy(Pointer, VectorWidth);
661 unsigned Offset = NegativeStride ? VectorWidth - 1 : 0;
662
Tobias Grosser5a56cbf2014-04-16 07:33:47 +0000663 Value *NewPointer = nullptr;
Tobias Grosserbc132602015-09-05 09:56:54 +0000664 NewPointer = generateLocationAccessed(Stmt, Load, Pointer, ScalarMaps[Offset],
665 VLTS[Offset], NewAccesses);
Tobias Grosserc14582f2013-02-05 18:01:29 +0000666 Value *VectorPtr =
667 Builder.CreateBitCast(NewPointer, VectorPtrType, "vector_ptr");
668 LoadInst *VecLoad =
669 Builder.CreateLoad(VectorPtr, Load->getName() + "_p_vec_full");
Hongbin Zheng3b11a162012-04-25 13:16:49 +0000670 if (!Aligned)
671 VecLoad->setAlignment(8);
672
Tobias Grosser0dd463f2014-03-19 19:27:24 +0000673 if (NegativeStride) {
674 SmallVector<Constant *, 16> Indices;
675 for (int i = VectorWidth - 1; i >= 0; i--)
676 Indices.push_back(ConstantInt::get(Builder.getInt32Ty(), i));
677 Constant *SV = llvm::ConstantVector::get(Indices);
678 Value *RevVecLoad = Builder.CreateShuffleVector(
679 VecLoad, VecLoad, SV, Load->getName() + "_reverse");
680 return RevVecLoad;
681 }
682
Hongbin Zheng3b11a162012-04-25 13:16:49 +0000683 return VecLoad;
684}
685
Tobias Grosser2d1ed0b2015-08-27 07:28:16 +0000686Value *VectorBlockGenerator::generateStrideZeroLoad(
Tobias Grosser2f1acac2015-10-04 10:18:45 +0000687 ScopStmt &Stmt, LoadInst *Load, ValueMapT &BBMap,
Tobias Grosser2d1ed0b2015-08-27 07:28:16 +0000688 __isl_keep isl_id_to_ast_expr *NewAccesses) {
Tobias Grosser9646e3f2015-10-04 10:18:39 +0000689 auto *Pointer = Load->getPointerOperand();
Hongbin Zheng3b11a162012-04-25 13:16:49 +0000690 Type *VectorPtrType = getVectorPtrTy(Pointer, 1);
Tobias Grosserbc132602015-09-05 09:56:54 +0000691 Value *NewPointer = generateLocationAccessed(Stmt, Load, Pointer, BBMap,
692 VLTS[0], NewAccesses);
Hongbin Zheng3b11a162012-04-25 13:16:49 +0000693 Value *VectorPtr = Builder.CreateBitCast(NewPointer, VectorPtrType,
694 Load->getName() + "_p_vec_p");
Tobias Grosserc14582f2013-02-05 18:01:29 +0000695 LoadInst *ScalarLoad =
696 Builder.CreateLoad(VectorPtr, Load->getName() + "_p_splat_one");
Hongbin Zheng3b11a162012-04-25 13:16:49 +0000697
698 if (!Aligned)
699 ScalarLoad->setAlignment(8);
700
Tobias Grosserc14582f2013-02-05 18:01:29 +0000701 Constant *SplatVector = Constant::getNullValue(
702 VectorType::get(Builder.getInt32Ty(), getVectorWidth()));
Hongbin Zheng3b11a162012-04-25 13:16:49 +0000703
Tobias Grosserc14582f2013-02-05 18:01:29 +0000704 Value *VectorLoad = Builder.CreateShuffleVector(
705 ScalarLoad, ScalarLoad, SplatVector, Load->getName() + "_p_splat");
Hongbin Zheng3b11a162012-04-25 13:16:49 +0000706 return VectorLoad;
707}
708
Johannes Doerfertbe9c9112015-02-06 21:39:31 +0000709Value *VectorBlockGenerator::generateUnknownStrideLoad(
Tobias Grosser2f1acac2015-10-04 10:18:45 +0000710 ScopStmt &Stmt, LoadInst *Load, VectorValueMapT &ScalarMaps,
Johannes Doerfert09e36972015-10-07 20:17:36 +0000711 __isl_keep isl_id_to_ast_expr *NewAccesses) {
Hongbin Zheng3b11a162012-04-25 13:16:49 +0000712 int VectorWidth = getVectorWidth();
Tobias Grosser9646e3f2015-10-04 10:18:39 +0000713 auto *Pointer = Load->getPointerOperand();
Hongbin Zheng3b11a162012-04-25 13:16:49 +0000714 VectorType *VectorType = VectorType::get(
Tobias Grosserc14582f2013-02-05 18:01:29 +0000715 dyn_cast<PointerType>(Pointer->getType())->getElementType(), VectorWidth);
Hongbin Zheng3b11a162012-04-25 13:16:49 +0000716
717 Value *Vector = UndefValue::get(VectorType);
718
719 for (int i = 0; i < VectorWidth; i++) {
Tobias Grosserbc132602015-09-05 09:56:54 +0000720 Value *NewPointer = generateLocationAccessed(
721 Stmt, Load, Pointer, ScalarMaps[i], VLTS[i], NewAccesses);
Tobias Grosserc14582f2013-02-05 18:01:29 +0000722 Value *ScalarLoad =
723 Builder.CreateLoad(NewPointer, Load->getName() + "_p_scalar_");
724 Vector = Builder.CreateInsertElement(
725 Vector, ScalarLoad, Builder.getInt32(i), Load->getName() + "_p_vec_");
Hongbin Zheng3b11a162012-04-25 13:16:49 +0000726 }
727
728 return Vector;
729}
730
Tobias Grosser2d1ed0b2015-08-27 07:28:16 +0000731void VectorBlockGenerator::generateLoad(
Tobias Grosser2f1acac2015-10-04 10:18:45 +0000732 ScopStmt &Stmt, LoadInst *Load, ValueMapT &VectorMap,
Tobias Grosser2d1ed0b2015-08-27 07:28:16 +0000733 VectorValueMapT &ScalarMaps, __isl_keep isl_id_to_ast_expr *NewAccesses) {
Johannes Doerfertc1db67e2015-09-29 23:47:21 +0000734 if (Value *PreloadLoad = GlobalMap.lookup(Load)) {
735 VectorMap[Load] = Builder.CreateVectorSplat(getVectorWidth(), PreloadLoad,
736 Load->getName() + "_p");
737 return;
738 }
739
Tobias Grosser28736452015-03-23 07:00:36 +0000740 if (!VectorType::isValidElementType(Load->getType())) {
Hongbin Zheng3b11a162012-04-25 13:16:49 +0000741 for (int i = 0; i < getVectorWidth(); i++)
Tobias Grosserbc132602015-09-05 09:56:54 +0000742 ScalarMaps[i][Load] =
743 generateScalarLoad(Stmt, Load, ScalarMaps[i], VLTS[i], NewAccesses);
Hongbin Zheng3b11a162012-04-25 13:16:49 +0000744 return;
745 }
746
Johannes Doerfertbe9c9112015-02-06 21:39:31 +0000747 const MemoryAccess &Access = Stmt.getAccessFor(Load);
Hongbin Zheng3b11a162012-04-25 13:16:49 +0000748
Tobias Grosser95493982014-04-18 09:46:35 +0000749 // Make sure we have scalar values available to access the pointer to
750 // the data location.
751 extractScalarValues(Load, VectorMap, ScalarMaps);
752
Hongbin Zheng3b11a162012-04-25 13:16:49 +0000753 Value *NewLoad;
Sebastian Popa00a0292012-12-18 07:46:06 +0000754 if (Access.isStrideZero(isl_map_copy(Schedule)))
Tobias Grosser2d1ed0b2015-08-27 07:28:16 +0000755 NewLoad = generateStrideZeroLoad(Stmt, Load, ScalarMaps[0], NewAccesses);
Sebastian Popa00a0292012-12-18 07:46:06 +0000756 else if (Access.isStrideOne(isl_map_copy(Schedule)))
Tobias Grosser2d1ed0b2015-08-27 07:28:16 +0000757 NewLoad = generateStrideOneLoad(Stmt, Load, ScalarMaps, NewAccesses);
Tobias Grosser0dd463f2014-03-19 19:27:24 +0000758 else if (Access.isStrideX(isl_map_copy(Schedule), -1))
Tobias Grosser2d1ed0b2015-08-27 07:28:16 +0000759 NewLoad = generateStrideOneLoad(Stmt, Load, ScalarMaps, NewAccesses, true);
Hongbin Zheng3b11a162012-04-25 13:16:49 +0000760 else
Tobias Grosser2d1ed0b2015-08-27 07:28:16 +0000761 NewLoad = generateUnknownStrideLoad(Stmt, Load, ScalarMaps, NewAccesses);
Hongbin Zheng3b11a162012-04-25 13:16:49 +0000762
763 VectorMap[Load] = NewLoad;
764}
765
Tobias Grosser2f1acac2015-10-04 10:18:45 +0000766void VectorBlockGenerator::copyUnaryInst(ScopStmt &Stmt, UnaryInstruction *Inst,
Hongbin Zheng3b11a162012-04-25 13:16:49 +0000767 ValueMapT &VectorMap,
768 VectorValueMapT &ScalarMaps) {
769 int VectorWidth = getVectorWidth();
Johannes Doerfertbe9c9112015-02-06 21:39:31 +0000770 Value *NewOperand = getVectorValue(Stmt, Inst->getOperand(0), VectorMap,
771 ScalarMaps, getLoopForInst(Inst));
Hongbin Zheng3b11a162012-04-25 13:16:49 +0000772
773 assert(isa<CastInst>(Inst) && "Can not generate vector code for instruction");
774
775 const CastInst *Cast = dyn_cast<CastInst>(Inst);
776 VectorType *DestType = VectorType::get(Inst->getType(), VectorWidth);
777 VectorMap[Inst] = Builder.CreateCast(Cast->getOpcode(), NewOperand, DestType);
778}
779
Tobias Grosser2f1acac2015-10-04 10:18:45 +0000780void VectorBlockGenerator::copyBinaryInst(ScopStmt &Stmt, BinaryOperator *Inst,
Hongbin Zheng3b11a162012-04-25 13:16:49 +0000781 ValueMapT &VectorMap,
782 VectorValueMapT &ScalarMaps) {
Tobias Grosser369430f2013-03-22 23:42:53 +0000783 Loop *L = getLoopForInst(Inst);
Hongbin Zheng3b11a162012-04-25 13:16:49 +0000784 Value *OpZero = Inst->getOperand(0);
785 Value *OpOne = Inst->getOperand(1);
786
787 Value *NewOpZero, *NewOpOne;
Johannes Doerfertbe9c9112015-02-06 21:39:31 +0000788 NewOpZero = getVectorValue(Stmt, OpZero, VectorMap, ScalarMaps, L);
789 NewOpOne = getVectorValue(Stmt, OpOne, VectorMap, ScalarMaps, L);
Hongbin Zheng3b11a162012-04-25 13:16:49 +0000790
Tobias Grosser1bb59b02012-12-29 23:47:38 +0000791 Value *NewInst = Builder.CreateBinOp(Inst->getOpcode(), NewOpZero, NewOpOne,
Hongbin Zheng3b11a162012-04-25 13:16:49 +0000792 Inst->getName() + "p_vec");
793 VectorMap[Inst] = NewInst;
794}
795
Tobias Grosser2d1ed0b2015-08-27 07:28:16 +0000796void VectorBlockGenerator::copyStore(
Tobias Grosser2f1acac2015-10-04 10:18:45 +0000797 ScopStmt &Stmt, StoreInst *Store, ValueMapT &VectorMap,
Tobias Grosser2d1ed0b2015-08-27 07:28:16 +0000798 VectorValueMapT &ScalarMaps, __isl_keep isl_id_to_ast_expr *NewAccesses) {
Johannes Doerfertbe9c9112015-02-06 21:39:31 +0000799 const MemoryAccess &Access = Stmt.getAccessFor(Store);
Hongbin Zheng3b11a162012-04-25 13:16:49 +0000800
Tobias Grosser9646e3f2015-10-04 10:18:39 +0000801 auto *Pointer = Store->getPointerOperand();
Johannes Doerfertbe9c9112015-02-06 21:39:31 +0000802 Value *Vector = getVectorValue(Stmt, Store->getValueOperand(), VectorMap,
Tobias Grosser369430f2013-03-22 23:42:53 +0000803 ScalarMaps, getLoopForInst(Store));
Hongbin Zheng3b11a162012-04-25 13:16:49 +0000804
Tobias Grosser50fd7012014-04-17 23:13:49 +0000805 // Make sure we have scalar values available to access the pointer to
806 // the data location.
807 extractScalarValues(Store, VectorMap, ScalarMaps);
808
Sebastian Popa00a0292012-12-18 07:46:06 +0000809 if (Access.isStrideOne(isl_map_copy(Schedule))) {
Johannes Doerfert1947f862014-10-08 20:18:32 +0000810 Type *VectorPtrType = getVectorPtrTy(Pointer, getVectorWidth());
Tobias Grosserbc132602015-09-05 09:56:54 +0000811 Value *NewPointer = generateLocationAccessed(
812 Stmt, Store, Pointer, ScalarMaps[0], VLTS[0], NewAccesses);
Hongbin Zheng3b11a162012-04-25 13:16:49 +0000813
Tobias Grosserc14582f2013-02-05 18:01:29 +0000814 Value *VectorPtr =
815 Builder.CreateBitCast(NewPointer, VectorPtrType, "vector_ptr");
Hongbin Zheng3b11a162012-04-25 13:16:49 +0000816 StoreInst *Store = Builder.CreateStore(Vector, VectorPtr);
817
818 if (!Aligned)
819 Store->setAlignment(8);
820 } else {
821 for (unsigned i = 0; i < ScalarMaps.size(); i++) {
Tobias Grosser1bb59b02012-12-29 23:47:38 +0000822 Value *Scalar = Builder.CreateExtractElement(Vector, Builder.getInt32(i));
Tobias Grosserbc132602015-09-05 09:56:54 +0000823 Value *NewPointer = generateLocationAccessed(
824 Stmt, Store, Pointer, ScalarMaps[i], VLTS[i], NewAccesses);
Hongbin Zheng3b11a162012-04-25 13:16:49 +0000825 Builder.CreateStore(Scalar, NewPointer);
826 }
827 }
828}
829
830bool VectorBlockGenerator::hasVectorOperands(const Instruction *Inst,
831 ValueMapT &VectorMap) {
Tobias Grosser91f5b262014-06-04 08:06:40 +0000832 for (Value *Operand : Inst->operands())
833 if (VectorMap.count(Operand))
Hongbin Zheng3b11a162012-04-25 13:16:49 +0000834 return true;
835 return false;
836}
837
838bool VectorBlockGenerator::extractScalarValues(const Instruction *Inst,
839 ValueMapT &VectorMap,
840 VectorValueMapT &ScalarMaps) {
841 bool HasVectorOperand = false;
842 int VectorWidth = getVectorWidth();
843
Tobias Grosser91f5b262014-06-04 08:06:40 +0000844 for (Value *Operand : Inst->operands()) {
845 ValueMapT::iterator VecOp = VectorMap.find(Operand);
Hongbin Zheng3b11a162012-04-25 13:16:49 +0000846
847 if (VecOp == VectorMap.end())
848 continue;
849
850 HasVectorOperand = true;
851 Value *NewVector = VecOp->second;
852
853 for (int i = 0; i < VectorWidth; ++i) {
854 ValueMapT &SM = ScalarMaps[i];
855
856 // If there is one scalar extracted, all scalar elements should have
857 // already been extracted by the code here. So no need to check for the
858 // existance of all of them.
Tobias Grosser91f5b262014-06-04 08:06:40 +0000859 if (SM.count(Operand))
Hongbin Zheng3b11a162012-04-25 13:16:49 +0000860 break;
861
Tobias Grosser91f5b262014-06-04 08:06:40 +0000862 SM[Operand] =
863 Builder.CreateExtractElement(NewVector, Builder.getInt32(i));
Hongbin Zheng3b11a162012-04-25 13:16:49 +0000864 }
865 }
866
867 return HasVectorOperand;
868}
869
Tobias Grosser2d1ed0b2015-08-27 07:28:16 +0000870void VectorBlockGenerator::copyInstScalarized(
Tobias Grosser2f1acac2015-10-04 10:18:45 +0000871 ScopStmt &Stmt, Instruction *Inst, ValueMapT &VectorMap,
Tobias Grosser2d1ed0b2015-08-27 07:28:16 +0000872 VectorValueMapT &ScalarMaps, __isl_keep isl_id_to_ast_expr *NewAccesses) {
Hongbin Zheng3b11a162012-04-25 13:16:49 +0000873 bool HasVectorOperand;
874 int VectorWidth = getVectorWidth();
875
876 HasVectorOperand = extractScalarValues(Inst, VectorMap, ScalarMaps);
877
878 for (int VectorLane = 0; VectorLane < getVectorWidth(); VectorLane++)
Johannes Doerfertbe9c9112015-02-06 21:39:31 +0000879 BlockGenerator::copyInstruction(Stmt, Inst, ScalarMaps[VectorLane],
Tobias Grosserbc132602015-09-05 09:56:54 +0000880 VLTS[VectorLane], NewAccesses);
Hongbin Zheng3b11a162012-04-25 13:16:49 +0000881
882 if (!VectorType::isValidElementType(Inst->getType()) || !HasVectorOperand)
883 return;
884
885 // Make the result available as vector value.
886 VectorType *VectorType = VectorType::get(Inst->getType(), VectorWidth);
887 Value *Vector = UndefValue::get(VectorType);
888
889 for (int i = 0; i < VectorWidth; i++)
890 Vector = Builder.CreateInsertElement(Vector, ScalarMaps[i][Inst],
891 Builder.getInt32(i));
892
893 VectorMap[Inst] = Vector;
894}
895
Tobias Grosserbc132602015-09-05 09:56:54 +0000896int VectorBlockGenerator::getVectorWidth() { return VLTS.size(); }
Hongbin Zheng3b11a162012-04-25 13:16:49 +0000897
Tobias Grosser2d1ed0b2015-08-27 07:28:16 +0000898void VectorBlockGenerator::copyInstruction(
Tobias Grosser2f1acac2015-10-04 10:18:45 +0000899 ScopStmt &Stmt, Instruction *Inst, ValueMapT &VectorMap,
Tobias Grosser2d1ed0b2015-08-27 07:28:16 +0000900 VectorValueMapT &ScalarMaps, __isl_keep isl_id_to_ast_expr *NewAccesses) {
Hongbin Zheng3b11a162012-04-25 13:16:49 +0000901 // Terminator instructions control the control flow. They are explicitly
902 // expressed in the clast and do not need to be copied.
903 if (Inst->isTerminator())
904 return;
905
Johannes Doerfert1ef52332015-02-08 20:50:42 +0000906 if (canSynthesize(Inst, &LI, &SE, &Stmt.getParent()->getRegion()))
Tobias Grossere71c6ab2012-04-27 16:36:14 +0000907 return;
908
Tobias Grosser9646e3f2015-10-04 10:18:39 +0000909 if (auto *Load = dyn_cast<LoadInst>(Inst)) {
Tobias Grosser2d1ed0b2015-08-27 07:28:16 +0000910 generateLoad(Stmt, Load, VectorMap, ScalarMaps, NewAccesses);
Hongbin Zheng3b11a162012-04-25 13:16:49 +0000911 return;
912 }
913
914 if (hasVectorOperands(Inst, VectorMap)) {
Tobias Grosser9646e3f2015-10-04 10:18:39 +0000915 if (auto *Store = dyn_cast<StoreInst>(Inst)) {
Tobias Grosser2d1ed0b2015-08-27 07:28:16 +0000916 copyStore(Stmt, Store, VectorMap, ScalarMaps, NewAccesses);
Hongbin Zheng3b11a162012-04-25 13:16:49 +0000917 return;
918 }
919
Tobias Grosser9646e3f2015-10-04 10:18:39 +0000920 if (auto *Unary = dyn_cast<UnaryInstruction>(Inst)) {
Johannes Doerfertbe9c9112015-02-06 21:39:31 +0000921 copyUnaryInst(Stmt, Unary, VectorMap, ScalarMaps);
Hongbin Zheng3b11a162012-04-25 13:16:49 +0000922 return;
923 }
924
Tobias Grosser9646e3f2015-10-04 10:18:39 +0000925 if (auto *Binary = dyn_cast<BinaryOperator>(Inst)) {
Johannes Doerfertbe9c9112015-02-06 21:39:31 +0000926 copyBinaryInst(Stmt, Binary, VectorMap, ScalarMaps);
Hongbin Zheng3b11a162012-04-25 13:16:49 +0000927 return;
928 }
929
930 // Falltrough: We generate scalar instructions, if we don't know how to
931 // generate vector code.
932 }
933
Tobias Grosser2d1ed0b2015-08-27 07:28:16 +0000934 copyInstScalarized(Stmt, Inst, VectorMap, ScalarMaps, NewAccesses);
Hongbin Zheng3b11a162012-04-25 13:16:49 +0000935}
936
Tobias Grosser2d1ed0b2015-08-27 07:28:16 +0000937void VectorBlockGenerator::copyStmt(
938 ScopStmt &Stmt, __isl_keep isl_id_to_ast_expr *NewAccesses) {
Johannes Doerfert275a1752015-02-24 16:16:32 +0000939 assert(Stmt.isBlockStmt() && "TODO: Only block statements can be copied by "
940 "the vector block generator");
941
Johannes Doerfertbe9c9112015-02-06 21:39:31 +0000942 BasicBlock *BB = Stmt.getBasicBlock();
Tobias Grosserc14582f2013-02-05 18:01:29 +0000943 BasicBlock *CopyBB =
Johannes Doerfertb4f08eb2015-02-23 13:51:35 +0000944 SplitBlock(Builder.GetInsertBlock(), Builder.GetInsertPoint(), &DT, &LI);
Hongbin Zheng3b11a162012-04-25 13:16:49 +0000945 CopyBB->setName("polly.stmt." + BB->getName());
946 Builder.SetInsertPoint(CopyBB->begin());
947
948 // Create two maps that store the mapping from the original instructions of
949 // the old basic block to their copies in the new basic block. Those maps
950 // are basic block local.
951 //
952 // As vector code generation is supported there is one map for scalar values
953 // and one for vector values.
954 //
955 // In case we just do scalar code generation, the vectorMap is not used and
956 // the scalarMap has just one dimension, which contains the mapping.
957 //
958 // In case vector code generation is done, an instruction may either appear
959 // in the vector map once (as it is calculating >vectorwidth< values at a
960 // time. Or (if the values are calculated using scalar operations), it
961 // appears once in every dimension of the scalarMap.
962 VectorValueMapT ScalarBlockMap(getVectorWidth());
963 ValueMapT VectorBlockMap;
964
Tobias Grosser91f5b262014-06-04 08:06:40 +0000965 for (Instruction &Inst : *BB)
Tobias Grosser2d1ed0b2015-08-27 07:28:16 +0000966 copyInstruction(Stmt, &Inst, VectorBlockMap, ScalarBlockMap, NewAccesses);
Hongbin Zheng3b11a162012-04-25 13:16:49 +0000967}
Johannes Doerfert275a1752015-02-24 16:16:32 +0000968
Johannes Doerfertecff11d2015-05-22 23:43:58 +0000969BasicBlock *RegionGenerator::repairDominance(BasicBlock *BB,
970 BasicBlock *BBCopy) {
Johannes Doerfert514f6ef2015-02-27 18:29:04 +0000971
972 BasicBlock *BBIDom = DT.getNode(BB)->getIDom()->getBlock();
973 BasicBlock *BBCopyIDom = BlockMap.lookup(BBIDom);
974
975 if (BBCopyIDom)
976 DT.changeImmediateDominator(BBCopy, BBCopyIDom);
977
978 return BBCopyIDom;
979}
980
Tobias Grosserbc132602015-09-05 09:56:54 +0000981void RegionGenerator::copyStmt(ScopStmt &Stmt, LoopToScevMapT &LTS,
Tobias Grosser2d1ed0b2015-08-27 07:28:16 +0000982 isl_id_to_ast_expr *IdToAstExp) {
Johannes Doerfert275a1752015-02-24 16:16:32 +0000983 assert(Stmt.isRegionStmt() &&
Tobias Grosserd3f21832015-08-01 06:26:51 +0000984 "Only region statements can be copied by the region generator");
Johannes Doerfert275a1752015-02-24 16:16:32 +0000985
Johannes Doerfertecff11d2015-05-22 23:43:58 +0000986 // Forget all old mappings.
987 BlockMap.clear();
988 RegionMaps.clear();
989 IncompletePHINodeMap.clear();
990
Michael Kruse225f0d12015-10-17 21:36:00 +0000991 // Collection of all values related to this subregion.
992 ValueMapT ValueMap;
993
Johannes Doerfert275a1752015-02-24 16:16:32 +0000994 // The region represented by the statement.
995 Region *R = Stmt.getRegion();
996
Johannes Doerfertecff11d2015-05-22 23:43:58 +0000997 // Create a dedicated entry for the region where we can reload all demoted
998 // inputs.
999 BasicBlock *EntryBB = R->getEntry();
1000 BasicBlock *EntryBBCopy =
1001 SplitBlock(Builder.GetInsertBlock(), Builder.GetInsertPoint(), &DT, &LI);
1002 EntryBBCopy->setName("polly.stmt." + EntryBB->getName() + ".entry");
1003 Builder.SetInsertPoint(EntryBBCopy->begin());
Johannes Doerfert514f6ef2015-02-27 18:29:04 +00001004
Michael Kruse225f0d12015-10-17 21:36:00 +00001005 generateScalarLoads(Stmt, RegionMaps[EntryBBCopy]);
1006
Johannes Doerfertecff11d2015-05-22 23:43:58 +00001007 for (auto PI = pred_begin(EntryBB), PE = pred_end(EntryBB); PI != PE; ++PI)
1008 if (!R->contains(*PI))
1009 BlockMap[*PI] = EntryBBCopy;
Johannes Doerfert275a1752015-02-24 16:16:32 +00001010
1011 // Iterate over all blocks in the region in a breadth-first search.
1012 std::deque<BasicBlock *> Blocks;
1013 SmallPtrSet<BasicBlock *, 8> SeenBlocks;
Johannes Doerfertecff11d2015-05-22 23:43:58 +00001014 Blocks.push_back(EntryBB);
1015 SeenBlocks.insert(EntryBB);
Johannes Doerfert275a1752015-02-24 16:16:32 +00001016
1017 while (!Blocks.empty()) {
1018 BasicBlock *BB = Blocks.front();
1019 Blocks.pop_front();
1020
Johannes Doerfert514f6ef2015-02-27 18:29:04 +00001021 // First split the block and update dominance information.
1022 BasicBlock *BBCopy = splitBB(BB);
Johannes Doerfertecff11d2015-05-22 23:43:58 +00001023 BasicBlock *BBCopyIDom = repairDominance(BB, BBCopy);
1024
1025 // In order to remap PHI nodes we store also basic block mappings.
1026 BlockMap[BB] = BBCopy;
Johannes Doerfert514f6ef2015-02-27 18:29:04 +00001027
1028 // Get the mapping for this block and initialize it with the mapping
1029 // available at its immediate dominator (in the new region).
1030 ValueMapT &RegionMap = RegionMaps[BBCopy];
Michael Kruse225f0d12015-10-17 21:36:00 +00001031 if (BBCopy != EntryBBCopy)
1032 RegionMap = RegionMaps[BBCopyIDom];
Johannes Doerfert514f6ef2015-02-27 18:29:04 +00001033
Johannes Doerfert275a1752015-02-24 16:16:32 +00001034 // Copy the block with the BlockGenerator.
Michael Kruse225f0d12015-10-17 21:36:00 +00001035 Builder.SetInsertPoint(BBCopy->begin());
Tobias Grosserbc132602015-09-05 09:56:54 +00001036 copyBB(Stmt, BB, BBCopy, RegionMap, LTS, IdToAstExp);
Johannes Doerfert275a1752015-02-24 16:16:32 +00001037
Johannes Doerfertecff11d2015-05-22 23:43:58 +00001038 // In order to remap PHI nodes we store also basic block mappings.
1039 BlockMap[BB] = BBCopy;
1040
1041 // Add values to incomplete PHI nodes waiting for this block to be copied.
1042 for (const PHINodePairTy &PHINodePair : IncompletePHINodeMap[BB])
Tobias Grosserbc132602015-09-05 09:56:54 +00001043 addOperandToPHI(Stmt, PHINodePair.first, PHINodePair.second, BB, LTS);
Johannes Doerfertecff11d2015-05-22 23:43:58 +00001044 IncompletePHINodeMap[BB].clear();
1045
Johannes Doerfert275a1752015-02-24 16:16:32 +00001046 // And continue with new successors inside the region.
1047 for (auto SI = succ_begin(BB), SE = succ_end(BB); SI != SE; SI++)
1048 if (R->contains(*SI) && SeenBlocks.insert(*SI).second)
1049 Blocks.push_back(*SI);
Michael Kruse225f0d12015-10-17 21:36:00 +00001050
1051 // Remember value in case it is visible after this subregion.
1052 ValueMap.insert(RegionMap.begin(), RegionMap.end());
Johannes Doerfert275a1752015-02-24 16:16:32 +00001053 }
1054
1055 // Now create a new dedicated region exit block and add it to the region map.
Johannes Doerfert514f6ef2015-02-27 18:29:04 +00001056 BasicBlock *ExitBBCopy =
Johannes Doerfert275a1752015-02-24 16:16:32 +00001057 SplitBlock(Builder.GetInsertBlock(), Builder.GetInsertPoint(), &DT, &LI);
Johannes Doerfertecff11d2015-05-22 23:43:58 +00001058 ExitBBCopy->setName("polly.stmt." + R->getExit()->getName() + ".exit");
Johannes Doerfert514f6ef2015-02-27 18:29:04 +00001059 BlockMap[R->getExit()] = ExitBBCopy;
1060
Johannes Doerfertecff11d2015-05-22 23:43:58 +00001061 repairDominance(R->getExit(), ExitBBCopy);
Johannes Doerfert275a1752015-02-24 16:16:32 +00001062
1063 // As the block generator doesn't handle control flow we need to add the
1064 // region control flow by hand after all blocks have been copied.
1065 for (BasicBlock *BB : SeenBlocks) {
1066
Johannes Doerfert514f6ef2015-02-27 18:29:04 +00001067 BasicBlock *BBCopy = BlockMap[BB];
Johannes Doerferte114dc02015-09-14 11:15:58 +00001068 TerminatorInst *TI = BB->getTerminator();
1069 if (isa<UnreachableInst>(TI)) {
1070 while (!BBCopy->empty())
1071 BBCopy->begin()->eraseFromParent();
1072 new UnreachableInst(BBCopy->getContext(), BBCopy);
1073 continue;
1074 }
1075
Johannes Doerfert275a1752015-02-24 16:16:32 +00001076 Instruction *BICopy = BBCopy->getTerminator();
1077
Johannes Doerfert514f6ef2015-02-27 18:29:04 +00001078 ValueMapT &RegionMap = RegionMaps[BBCopy];
1079 RegionMap.insert(BlockMap.begin(), BlockMap.end());
1080
Tobias Grosser45e79442015-08-01 09:07:57 +00001081 Builder.SetInsertPoint(BICopy);
Johannes Doerfert9a132f32015-09-28 09:33:22 +00001082 copyInstScalar(Stmt, TI, RegionMap, LTS);
Johannes Doerfert275a1752015-02-24 16:16:32 +00001083 BICopy->eraseFromParent();
1084 }
1085
Johannes Doerfertecff11d2015-05-22 23:43:58 +00001086 // Add counting PHI nodes to all loops in the region that can be used as
1087 // replacement for SCEVs refering to the old loop.
1088 for (BasicBlock *BB : SeenBlocks) {
1089 Loop *L = LI.getLoopFor(BB);
1090 if (L == nullptr || L->getHeader() != BB)
1091 continue;
1092
1093 BasicBlock *BBCopy = BlockMap[BB];
1094 Value *NullVal = Builder.getInt32(0);
1095 PHINode *LoopPHI =
1096 PHINode::Create(Builder.getInt32Ty(), 2, "polly.subregion.iv");
1097 Instruction *LoopPHIInc = BinaryOperator::CreateAdd(
1098 LoopPHI, Builder.getInt32(1), "polly.subregion.iv.inc");
1099 LoopPHI->insertBefore(BBCopy->begin());
1100 LoopPHIInc->insertBefore(BBCopy->getTerminator());
1101
1102 for (auto *PredBB : make_range(pred_begin(BB), pred_end(BB))) {
1103 if (!R->contains(PredBB))
1104 continue;
1105 if (L->contains(PredBB))
1106 LoopPHI->addIncoming(LoopPHIInc, BlockMap[PredBB]);
1107 else
1108 LoopPHI->addIncoming(NullVal, BlockMap[PredBB]);
1109 }
1110
1111 for (auto *PredBBCopy : make_range(pred_begin(BBCopy), pred_end(BBCopy)))
1112 if (LoopPHI->getBasicBlockIndex(PredBBCopy) < 0)
1113 LoopPHI->addIncoming(NullVal, PredBBCopy);
1114
1115 LTS[L] = SE.getUnknown(LoopPHI);
1116 }
1117
Michael Kruse225f0d12015-10-17 21:36:00 +00001118 // Continue generating code in the exit block.
1119 Builder.SetInsertPoint(ExitBBCopy->getFirstInsertionPt());
1120
1121 // Write values visible to other statements.
1122 generateScalarStores(Stmt, LTS, ValueMap);
Johannes Doerfert275a1752015-02-24 16:16:32 +00001123}
Johannes Doerfertecff11d2015-05-22 23:43:58 +00001124
Michael Kruse225f0d12015-10-17 21:36:00 +00001125void RegionGenerator::generateScalarStores(ScopStmt &Stmt, LoopToScevMapT &LTS,
Tobias Grosserbc132602015-09-05 09:56:54 +00001126 ValueMapT &BBMap) {
Johannes Doerfertecff11d2015-05-22 23:43:58 +00001127 const Region &R = Stmt.getParent()->getRegion();
1128
Tobias Grosser75296902015-08-21 19:23:21 +00001129 assert(Stmt.getRegion() &&
1130 "Block statements need to use the generateScalarStores() "
1131 "function in the BlockGenerator");
Johannes Doerfertecff11d2015-05-22 23:43:58 +00001132
Johannes Doerfertecff11d2015-05-22 23:43:58 +00001133 for (MemoryAccess *MA : Stmt) {
Michael Kruse8d0b7342015-09-25 21:21:00 +00001134 if (MA->isExplicit() || MA->isRead())
Johannes Doerfertecff11d2015-05-22 23:43:58 +00001135 continue;
1136
Johannes Doerfertecff11d2015-05-22 23:43:58 +00001137 Instruction *ScalarInst = MA->getAccessInstruction();
Johannes Doerfertd86f2152015-08-17 10:58:17 +00001138 Value *Val = MA->getAccessValue();
Johannes Doerfertecff11d2015-05-22 23:43:58 +00001139
Michael Kruse225f0d12015-10-17 21:36:00 +00001140 // In case we add the store into an exiting block, we need to restore the
1141 // position for stores in the exit node.
1142 auto SavedInsertionPoint = Builder.GetInsertPoint();
1143
1144 // Implicit writes induced by PHIs must be written in the incoming blocks.
1145 if (isa<TerminatorInst>(ScalarInst)) {
1146 BasicBlock *ExitingBB = ScalarInst->getParent();
1147 BasicBlock *ExitingBBCopy = BlockMap[ExitingBB];
1148 Builder.SetInsertPoint(ExitingBBCopy->getTerminator());
1149 }
1150
Tobias Grosserbc132602015-09-05 09:56:54 +00001151 auto Address = getOrCreateAlloca(*MA);
Johannes Doerfertecff11d2015-05-22 23:43:58 +00001152
Tobias Grosseraff56c82015-09-30 13:36:54 +00001153 Val = getNewScalarValue(Val, R, Stmt, LTS, BBMap);
Tobias Grosserf8d55f72015-08-29 18:12:03 +00001154 Builder.CreateStore(Val, Address);
Michael Kruse225f0d12015-10-17 21:36:00 +00001155
1156 // Restore the insertion point if necessary.
1157 if (isa<TerminatorInst>(ScalarInst))
1158 Builder.SetInsertPoint(SavedInsertionPoint);
Johannes Doerfertecff11d2015-05-22 23:43:58 +00001159 }
1160}
1161
1162void RegionGenerator::addOperandToPHI(ScopStmt &Stmt, const PHINode *PHI,
1163 PHINode *PHICopy, BasicBlock *IncomingBB,
Johannes Doerfertecff11d2015-05-22 23:43:58 +00001164 LoopToScevMapT &LTS) {
1165 Region *StmtR = Stmt.getRegion();
1166
1167 // If the incoming block was not yet copied mark this PHI as incomplete.
1168 // Once the block will be copied the incoming value will be added.
1169 BasicBlock *BBCopy = BlockMap[IncomingBB];
1170 if (!BBCopy) {
1171 assert(StmtR->contains(IncomingBB) &&
1172 "Bad incoming block for PHI in non-affine region");
1173 IncompletePHINodeMap[IncomingBB].push_back(std::make_pair(PHI, PHICopy));
1174 return;
1175 }
1176
1177 Value *OpCopy = nullptr;
1178 if (StmtR->contains(IncomingBB)) {
1179 assert(RegionMaps.count(BBCopy) &&
1180 "Incoming PHI block did not have a BBMap");
1181 ValueMapT &BBCopyMap = RegionMaps[BBCopy];
1182
1183 Value *Op = PHI->getIncomingValueForBlock(IncomingBB);
Tobias Grosserbc132602015-09-05 09:56:54 +00001184 OpCopy = getNewValue(Stmt, Op, BBCopyMap, LTS, getLoopForInst(PHI));
Johannes Doerfertecff11d2015-05-22 23:43:58 +00001185 } else {
1186
1187 if (PHICopy->getBasicBlockIndex(BBCopy) >= 0)
1188 return;
1189
Tobias Grosserbc132602015-09-05 09:56:54 +00001190 Value *PHIOpAddr = getOrCreatePHIAlloca(const_cast<PHINode *>(PHI));
Johannes Doerfertecff11d2015-05-22 23:43:58 +00001191 OpCopy = new LoadInst(PHIOpAddr, PHIOpAddr->getName() + ".reload",
1192 BlockMap[IncomingBB]->getTerminator());
1193 }
1194
1195 assert(OpCopy && "Incoming PHI value was not copied properly");
1196 assert(BBCopy && "Incoming PHI block was not copied properly");
1197 PHICopy->addIncoming(OpCopy, BBCopy);
1198}
1199
Tobias Grosser2f1acac2015-10-04 10:18:45 +00001200Value *RegionGenerator::copyPHIInstruction(ScopStmt &Stmt, PHINode *PHI,
Johannes Doerfertecff11d2015-05-22 23:43:58 +00001201 ValueMapT &BBMap,
Johannes Doerfertecff11d2015-05-22 23:43:58 +00001202 LoopToScevMapT &LTS) {
1203 unsigned NumIncoming = PHI->getNumIncomingValues();
1204 PHINode *PHICopy =
1205 Builder.CreatePHI(PHI->getType(), NumIncoming, "polly." + PHI->getName());
1206 PHICopy->moveBefore(PHICopy->getParent()->getFirstNonPHI());
1207 BBMap[PHI] = PHICopy;
1208
1209 for (unsigned u = 0; u < NumIncoming; u++)
Tobias Grosserbc132602015-09-05 09:56:54 +00001210 addOperandToPHI(Stmt, PHI, PHICopy, PHI->getIncomingBlock(u), LTS);
Johannes Doerfertecff11d2015-05-22 23:43:58 +00001211 return PHICopy;
1212}