blob: 9e2112ba45c39afdb3523b35b8fbe64b34b476e2 [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 Grosser637bd632013-05-07 07:31:10 +000020#include "polly/Options.h"
Hongbin Zheng3b11a162012-04-25 13:16:49 +000021#include "polly/Support/GICHelper.h"
Sebastian Pop97cb8132013-03-18 20:21:13 +000022#include "polly/Support/SCEVValidator.h"
Tobias Grosserecfe21b2013-03-20 18:03:18 +000023#include "polly/Support/ScopHelper.h"
Tobias Grossere71c6ab2012-04-27 16:36:14 +000024#include "llvm/Analysis/LoopInfo.h"
Johannes Doerfertf32d6512015-03-01 18:45:58 +000025#include "llvm/Analysis/RegionInfo.h"
Tobias Grossere71c6ab2012-04-27 16:36:14 +000026#include "llvm/Analysis/ScalarEvolution.h"
Tobias Grosser030237d2014-02-21 15:06:05 +000027#include "llvm/IR/IntrinsicInst.h"
Tobias Grosserc9895062015-03-10 15:24:33 +000028#include "llvm/IR/Module.h"
Hongbin Zheng3b11a162012-04-25 13:16:49 +000029#include "llvm/Transforms/Utils/BasicBlockUtils.h"
Johannes Doerfertf32d6512015-03-01 18:45:58 +000030#include "isl/aff.h"
31#include "isl/ast.h"
Johannes Doerfertf32d6512015-03-01 18:45:58 +000032#include "isl/ast_build.h"
Tobias Grosserba0d0922015-05-09 09:13:42 +000033#include "isl/set.h"
Johannes Doerfertf32d6512015-03-01 18:45:58 +000034#include <deque>
35
Hongbin Zheng3b11a162012-04-25 13:16:49 +000036using namespace llvm;
37using namespace polly;
38
Tobias Grosser878aba42014-10-22 23:22:41 +000039static cl::opt<bool> Aligned("enable-polly-aligned",
40 cl::desc("Assumed aligned memory accesses."),
41 cl::Hidden, cl::init(false), cl::ZeroOrMore,
42 cl::cat(PollyCategory));
Hongbin Zheng3b11a162012-04-25 13:16:49 +000043
Tobias Grosserdcc3b432015-08-04 13:54:20 +000044bool polly::canSynthesize(const Value *V, const llvm::LoopInfo *LI,
Tobias Grosserecfe21b2013-03-20 18:03:18 +000045 ScalarEvolution *SE, const Region *R) {
Tobias Grosserdcc3b432015-08-04 13:54:20 +000046 if (!V || !SE->isSCEVable(V->getType()))
Tobias Grosserecfe21b2013-03-20 18:03:18 +000047 return false;
Tobias Grosserecfe21b2013-03-20 18:03:18 +000048
Tobias Grosserdcc3b432015-08-04 13:54:20 +000049 if (const SCEV *Scev = SE->getSCEV(const_cast<Value *>(V)))
Tobias Grosser683b8e42014-11-30 14:33:31 +000050 if (!isa<SCEVCouldNotCompute>(Scev))
51 if (!hasScalarDepsInsideRegion(Scev, R))
52 return true;
53
54 return false;
Tobias Grosserecfe21b2013-03-20 18:03:18 +000055}
56
Johannes Doerfert9e3a5db2015-01-26 15:55:54 +000057bool polly::isIgnoredIntrinsic(const Value *V) {
58 if (auto *IT = dyn_cast<IntrinsicInst>(V)) {
59 switch (IT->getIntrinsicID()) {
60 // Lifetime markers are supported/ignored.
61 case llvm::Intrinsic::lifetime_start:
62 case llvm::Intrinsic::lifetime_end:
63 // Invariant markers are supported/ignored.
64 case llvm::Intrinsic::invariant_start:
65 case llvm::Intrinsic::invariant_end:
66 // Some misc annotations are supported/ignored.
67 case llvm::Intrinsic::var_annotation:
68 case llvm::Intrinsic::ptr_annotation:
69 case llvm::Intrinsic::annotation:
70 case llvm::Intrinsic::donothing:
71 case llvm::Intrinsic::assume:
72 case llvm::Intrinsic::expect:
73 return true;
74 default:
75 break;
76 }
77 }
78 return false;
79}
80
Johannes Doerfertb4f08eb2015-02-23 13:51:35 +000081BlockGenerator::BlockGenerator(PollyIRBuilder &B, LoopInfo &LI,
82 ScalarEvolution &SE, DominatorTree &DT,
Johannes Doerfertecff11d2015-05-22 23:43:58 +000083 ScalarAllocaMapTy &ScalarMap,
84 ScalarAllocaMapTy &PHIOpMap,
85 EscapeUsersAllocaMapTy &EscapeMap,
Johannes Doerfertb4f08eb2015-02-23 13:51:35 +000086 IslExprBuilder *ExprBuilder)
Johannes Doerfertecff11d2015-05-22 23:43:58 +000087 : Builder(B), LI(LI), SE(SE), ExprBuilder(ExprBuilder), DT(DT),
88 EntryBB(nullptr), PHIOpMap(PHIOpMap), ScalarMap(ScalarMap),
89 EscapeMap(EscapeMap) {}
Tobias Grossere71c6ab2012-04-27 16:36:14 +000090
Johannes Doerfertbe9c9112015-02-06 21:39:31 +000091Value *BlockGenerator::getNewValue(ScopStmt &Stmt, const Value *Old,
92 ValueMapT &BBMap, ValueMapT &GlobalMap,
93 LoopToScevMapT &LTS, Loop *L) const {
Hongbin Zheng3b11a162012-04-25 13:16:49 +000094 // We assume constants never change.
95 // This avoids map lookups for many calls to this function.
96 if (isa<Constant>(Old))
Tobias Grosserc14582f2013-02-05 18:01:29 +000097 return const_cast<Value *>(Old);
Hongbin Zheng3b11a162012-04-25 13:16:49 +000098
Hongbin Zhengfe11e282013-06-29 13:22:15 +000099 if (Value *New = GlobalMap.lookup(Old)) {
Tobias Grosserc14582f2013-02-05 18:01:29 +0000100 if (Old->getType()->getScalarSizeInBits() <
Tobias Grosserd7e58642013-04-10 06:55:45 +0000101 New->getType()->getScalarSizeInBits())
Hongbin Zheng3b11a162012-04-25 13:16:49 +0000102 New = Builder.CreateTruncOrBitCast(New, Old->getType());
103
104 return New;
105 }
106
Hongbin Zhengfe11e282013-06-29 13:22:15 +0000107 if (Value *New = BBMap.lookup(Old))
108 return New;
Hongbin Zheng3b11a162012-04-25 13:16:49 +0000109
Tobias Grosser683b8e42014-11-30 14:33:31 +0000110 if (SE.isSCEVable(Old->getType()))
Tobias Grosser369430f2013-03-22 23:42:53 +0000111 if (const SCEV *Scev = SE.getSCEVAtScope(const_cast<Value *>(Old), L)) {
Tobias Grossere71c6ab2012-04-27 16:36:14 +0000112 if (!isa<SCEVCouldNotCompute>(Scev)) {
Sebastian Pop637b23d2013-02-15 20:56:01 +0000113 const SCEV *NewScev = apply(Scev, LTS, SE);
114 ValueToValueMap VTV;
115 VTV.insert(BBMap.begin(), BBMap.end());
116 VTV.insert(GlobalMap.begin(), GlobalMap.end());
Sebastian Pop47d4ee32013-02-15 21:26:53 +0000117 NewScev = SCEVParameterRewriter::rewrite(NewScev, SE, VTV);
Johannes Doerferte69e1142015-08-18 11:56:00 +0000118
119 Scop &S = *Stmt.getParent();
120 const DataLayout &DL =
121 S.getRegion().getEntry()->getParent()->getParent()->getDataLayout();
122 auto IP = Builder.GetInsertPoint();
123
124 assert(IP != Builder.GetInsertBlock()->end() &&
Tobias Grosser45e79442015-08-01 09:07:57 +0000125 "Only instructions can be insert points for SCEVExpander");
Johannes Doerferte69e1142015-08-18 11:56:00 +0000126 Value *Expanded =
127 expandCodeFor(S, SE, DL, "polly", NewScev, Old->getType(), IP);
Tobias Grossere71c6ab2012-04-27 16:36:14 +0000128
129 BBMap[Old] = Expanded;
130 return Expanded;
131 }
Tobias Grosser369430f2013-03-22 23:42:53 +0000132 }
Tobias Grossere71c6ab2012-04-27 16:36:14 +0000133
Tobias Grosser16371ac2014-11-05 20:48:56 +0000134 // A scop-constant value defined by a global or a function parameter.
135 if (isa<GlobalValue>(Old) || isa<Argument>(Old))
136 return const_cast<Value *>(Old);
137
138 // A scop-constant value defined by an instruction executed outside the scop.
139 if (const Instruction *Inst = dyn_cast<Instruction>(Old))
Johannes Doerfertbe9c9112015-02-06 21:39:31 +0000140 if (!Stmt.getParent()->getRegion().contains(Inst->getParent()))
Tobias Grosser16371ac2014-11-05 20:48:56 +0000141 return const_cast<Value *>(Old);
142
143 // The scalar dependence is neither available nor SCEVCodegenable.
Hongbin Zheng5b463ce2013-07-25 09:12:07 +0000144 llvm_unreachable("Unexpected scalar dependence in region!");
Tobias Grosser5a56cbf2014-04-16 07:33:47 +0000145 return nullptr;
Hongbin Zheng3b11a162012-04-25 13:16:49 +0000146}
147
Johannes Doerfertbe9c9112015-02-06 21:39:31 +0000148void BlockGenerator::copyInstScalar(ScopStmt &Stmt, const Instruction *Inst,
149 ValueMapT &BBMap, ValueMapT &GlobalMap,
150 LoopToScevMapT &LTS) {
Tobias Grosser030237d2014-02-21 15:06:05 +0000151 // We do not generate debug intrinsics as we did not investigate how to
152 // copy them correctly. At the current state, they just crash the code
153 // generation as the meta-data operands are not correctly copied.
154 if (isa<DbgInfoIntrinsic>(Inst))
155 return;
156
Hongbin Zheng3b11a162012-04-25 13:16:49 +0000157 Instruction *NewInst = Inst->clone();
158
159 // Replace old operands with the new ones.
Tobias Grosser91f5b262014-06-04 08:06:40 +0000160 for (Value *OldOperand : Inst->operands()) {
Johannes Doerfertbe9c9112015-02-06 21:39:31 +0000161 Value *NewOperand = getNewValue(Stmt, OldOperand, BBMap, GlobalMap, LTS,
162 getLoopForInst(Inst));
Hongbin Zheng3b11a162012-04-25 13:16:49 +0000163
164 if (!NewOperand) {
Tobias Grosserc14582f2013-02-05 18:01:29 +0000165 assert(!isa<StoreInst>(NewInst) &&
166 "Store instructions are always needed!");
Hongbin Zheng3b11a162012-04-25 13:16:49 +0000167 delete NewInst;
168 return;
169 }
170
171 NewInst->replaceUsesOfWith(OldOperand, NewOperand);
172 }
173
174 Builder.Insert(NewInst);
175 BBMap[Inst] = NewInst;
176
177 if (!NewInst->getType()->isVoidTy())
178 NewInst->setName("p_" + Inst->getName());
179}
180
Johannes Doerfertbe9c9112015-02-06 21:39:31 +0000181Value *BlockGenerator::generateLocationAccessed(
182 ScopStmt &Stmt, const Instruction *Inst, const Value *Pointer,
Tobias Grosser2d1ed0b2015-08-27 07:28:16 +0000183 ValueMapT &BBMap, ValueMapT &GlobalMap, LoopToScevMapT &LTS,
184 isl_id_to_ast_expr *NewAccesses) {
Johannes Doerfertbe9c9112015-02-06 21:39:31 +0000185 const MemoryAccess &MA = Stmt.getAccessFor(Inst);
Hongbin Zheng3b11a162012-04-25 13:16:49 +0000186
Tobias Grosser2d1ed0b2015-08-27 07:28:16 +0000187 isl_ast_expr *AccessExpr = isl_id_to_ast_expr_get(NewAccesses, MA.getId());
Hongbin Zheng3b11a162012-04-25 13:16:49 +0000188
Tobias Grosser2d1ed0b2015-08-27 07:28:16 +0000189 if (AccessExpr) {
190 AccessExpr = isl_ast_expr_address_of(AccessExpr);
191 return ExprBuilder->create(AccessExpr);
192 }
193
194 return getNewValue(Stmt, Pointer, BBMap, GlobalMap, LTS,
195 getLoopForInst(Inst));
Hongbin Zheng3b11a162012-04-25 13:16:49 +0000196}
197
Tobias Grosser4d96c8d2013-03-23 01:05:07 +0000198Loop *BlockGenerator::getLoopForInst(const llvm::Instruction *Inst) {
Johannes Doerfert2ef3f4f2014-08-07 17:14:54 +0000199 return LI.getLoopFor(Inst->getParent());
Tobias Grosser369430f2013-03-22 23:42:53 +0000200}
201
Johannes Doerfertbe9c9112015-02-06 21:39:31 +0000202Value *BlockGenerator::generateScalarLoad(ScopStmt &Stmt, const LoadInst *Load,
Tobias Grossere602a072013-05-07 07:30:56 +0000203 ValueMapT &BBMap,
204 ValueMapT &GlobalMap,
Tobias Grosser2d1ed0b2015-08-27 07:28:16 +0000205 LoopToScevMapT &LTS,
206 isl_id_to_ast_expr *NewAccesses) {
Hongbin Zheng3b11a162012-04-25 13:16:49 +0000207 const Value *Pointer = Load->getPointerOperand();
Tobias Grosser2d1ed0b2015-08-27 07:28:16 +0000208 Value *NewPointer = generateLocationAccessed(Stmt, Load, Pointer, BBMap,
209 GlobalMap, LTS, NewAccesses);
Johannes Doerfert87901452014-10-02 16:22:19 +0000210 Value *ScalarLoad = Builder.CreateAlignedLoad(
211 NewPointer, Load->getAlignment(), Load->getName() + "_p_scalar_");
Hongbin Zheng3b11a162012-04-25 13:16:49 +0000212 return ScalarLoad;
213}
214
Tobias Grosserc186ac72015-08-11 08:13:15 +0000215void BlockGenerator::generateScalarStore(ScopStmt &Stmt, const StoreInst *Store,
216 ValueMapT &BBMap, ValueMapT &GlobalMap,
Tobias Grosser2d1ed0b2015-08-27 07:28:16 +0000217 LoopToScevMapT &LTS,
218 isl_id_to_ast_expr *NewAccesses) {
Hongbin Zheng3b11a162012-04-25 13:16:49 +0000219 const Value *Pointer = Store->getPointerOperand();
Tobias Grosser2d1ed0b2015-08-27 07:28:16 +0000220 Value *NewPointer = generateLocationAccessed(Stmt, Store, Pointer, BBMap,
221 GlobalMap, LTS, NewAccesses);
Johannes Doerfertbe9c9112015-02-06 21:39:31 +0000222 Value *ValueOperand = getNewValue(Stmt, Store->getValueOperand(), BBMap,
223 GlobalMap, LTS, getLoopForInst(Store));
Hongbin Zheng3b11a162012-04-25 13:16:49 +0000224
Tobias Grosserc186ac72015-08-11 08:13:15 +0000225 Builder.CreateAlignedStore(ValueOperand, NewPointer, Store->getAlignment());
Hongbin Zheng3b11a162012-04-25 13:16:49 +0000226}
227
Johannes Doerfertbe9c9112015-02-06 21:39:31 +0000228void BlockGenerator::copyInstruction(ScopStmt &Stmt, const Instruction *Inst,
229 ValueMapT &BBMap, ValueMapT &GlobalMap,
Tobias Grosser2d1ed0b2015-08-27 07:28:16 +0000230 LoopToScevMapT &LTS,
231 isl_id_to_ast_expr *NewAccesses) {
Johannes Doerfertecff11d2015-05-22 23:43:58 +0000232
233 // First check for possible scalar dependences for this instruction.
234 generateScalarLoads(Stmt, Inst, BBMap);
235
Hongbin Zheng3b11a162012-04-25 13:16:49 +0000236 // Terminator instructions control the control flow. They are explicitly
237 // expressed in the clast and do not need to be copied.
238 if (Inst->isTerminator())
239 return;
240
Johannes Doerfertecff11d2015-05-22 23:43:58 +0000241 Loop *L = getLoopForInst(Inst);
242 if ((Stmt.isBlockStmt() || !Stmt.getRegion()->contains(L)) &&
243 canSynthesize(Inst, &LI, &SE, &Stmt.getParent()->getRegion())) {
244 Value *NewValue = getNewValue(Stmt, Inst, BBMap, GlobalMap, LTS, L);
245 BBMap[Inst] = NewValue;
Tobias Grossere71c6ab2012-04-27 16:36:14 +0000246 return;
Johannes Doerfertecff11d2015-05-22 23:43:58 +0000247 }
Tobias Grossere71c6ab2012-04-27 16:36:14 +0000248
Hongbin Zheng3b11a162012-04-25 13:16:49 +0000249 if (const LoadInst *Load = dyn_cast<LoadInst>(Inst)) {
Tobias Grosser2d1ed0b2015-08-27 07:28:16 +0000250 Value *NewLoad =
251 generateScalarLoad(Stmt, Load, BBMap, GlobalMap, LTS, NewAccesses);
Sebastian Pop3d94fed2013-05-24 18:46:02 +0000252 // Compute NewLoad before its insertion in BBMap to make the insertion
253 // deterministic.
Sebastian Pop753d43f2013-05-24 17:16:02 +0000254 BBMap[Load] = NewLoad;
Hongbin Zheng3b11a162012-04-25 13:16:49 +0000255 return;
256 }
257
258 if (const StoreInst *Store = dyn_cast<StoreInst>(Inst)) {
Tobias Grosser2d1ed0b2015-08-27 07:28:16 +0000259 generateScalarStore(Stmt, Store, BBMap, GlobalMap, LTS, NewAccesses);
Hongbin Zheng3b11a162012-04-25 13:16:49 +0000260 return;
261 }
262
Johannes Doerfertecff11d2015-05-22 23:43:58 +0000263 if (const PHINode *PHI = dyn_cast<PHINode>(Inst)) {
264 copyPHIInstruction(Stmt, PHI, BBMap, GlobalMap, LTS);
265 return;
266 }
267
Johannes Doerfert3f500fa2015-01-25 18:07:30 +0000268 // Skip some special intrinsics for which we do not adjust the semantics to
269 // the new schedule. All others are handled like every other instruction.
270 if (auto *IT = dyn_cast<IntrinsicInst>(Inst)) {
271 switch (IT->getIntrinsicID()) {
272 // Lifetime markers are ignored.
273 case llvm::Intrinsic::lifetime_start:
274 case llvm::Intrinsic::lifetime_end:
275 // Invariant markers are ignored.
276 case llvm::Intrinsic::invariant_start:
277 case llvm::Intrinsic::invariant_end:
278 // Some misc annotations are ignored.
279 case llvm::Intrinsic::var_annotation:
280 case llvm::Intrinsic::ptr_annotation:
281 case llvm::Intrinsic::annotation:
282 case llvm::Intrinsic::donothing:
283 case llvm::Intrinsic::assume:
284 case llvm::Intrinsic::expect:
285 return;
286 default:
287 // Other intrinsics are copied.
288 break;
289 }
290 }
291
Johannes Doerfertbe9c9112015-02-06 21:39:31 +0000292 copyInstScalar(Stmt, Inst, BBMap, GlobalMap, LTS);
Hongbin Zheng3b11a162012-04-25 13:16:49 +0000293}
294
Johannes Doerfert275a1752015-02-24 16:16:32 +0000295void BlockGenerator::copyStmt(ScopStmt &Stmt, ValueMapT &GlobalMap,
Tobias Grosser2d1ed0b2015-08-27 07:28:16 +0000296 LoopToScevMapT &LTS,
297 isl_id_to_ast_expr *NewAccesses) {
Johannes Doerfert275a1752015-02-24 16:16:32 +0000298 assert(Stmt.isBlockStmt() &&
299 "Only block statements can be copied by the block generator");
300
301 ValueMapT BBMap;
302
Johannes Doerfertbe9c9112015-02-06 21:39:31 +0000303 BasicBlock *BB = Stmt.getBasicBlock();
Tobias Grosser2d1ed0b2015-08-27 07:28:16 +0000304 copyBB(Stmt, BB, BBMap, GlobalMap, LTS, NewAccesses);
Johannes Doerfert275a1752015-02-24 16:16:32 +0000305}
306
Johannes Doerfert514f6ef2015-02-27 18:29:04 +0000307BasicBlock *BlockGenerator::splitBB(BasicBlock *BB) {
Tobias Grosserc14582f2013-02-05 18:01:29 +0000308 BasicBlock *CopyBB =
Johannes Doerfertb4f08eb2015-02-23 13:51:35 +0000309 SplitBlock(Builder.GetInsertBlock(), Builder.GetInsertPoint(), &DT, &LI);
Hongbin Zheng3b11a162012-04-25 13:16:49 +0000310 CopyBB->setName("polly.stmt." + BB->getName());
Johannes Doerfert514f6ef2015-02-27 18:29:04 +0000311 return CopyBB;
312}
Hongbin Zheng3b11a162012-04-25 13:16:49 +0000313
Johannes Doerfert514f6ef2015-02-27 18:29:04 +0000314BasicBlock *BlockGenerator::copyBB(ScopStmt &Stmt, BasicBlock *BB,
315 ValueMapT &BBMap, ValueMapT &GlobalMap,
Tobias Grosser2d1ed0b2015-08-27 07:28:16 +0000316 LoopToScevMapT &LTS,
317 isl_id_to_ast_expr *NewAccesses) {
Johannes Doerfert514f6ef2015-02-27 18:29:04 +0000318 BasicBlock *CopyBB = splitBB(BB);
Tobias Grosser2d1ed0b2015-08-27 07:28:16 +0000319 copyBB(Stmt, BB, CopyBB, BBMap, GlobalMap, LTS, NewAccesses);
Johannes Doerfert514f6ef2015-02-27 18:29:04 +0000320 return CopyBB;
321}
322
323void BlockGenerator::copyBB(ScopStmt &Stmt, BasicBlock *BB, BasicBlock *CopyBB,
324 ValueMapT &BBMap, ValueMapT &GlobalMap,
Tobias Grosser2d1ed0b2015-08-27 07:28:16 +0000325 LoopToScevMapT &LTS,
326 isl_id_to_ast_expr *NewAccesses) {
Johannes Doerfert514f6ef2015-02-27 18:29:04 +0000327 Builder.SetInsertPoint(CopyBB->begin());
Johannes Doerfertecff11d2015-05-22 23:43:58 +0000328 EntryBB = &CopyBB->getParent()->getEntryBlock();
329
Tobias Grosser91f5b262014-06-04 08:06:40 +0000330 for (Instruction &Inst : *BB)
Tobias Grosser2d1ed0b2015-08-27 07:28:16 +0000331 copyInstruction(Stmt, &Inst, BBMap, GlobalMap, LTS, NewAccesses);
Johannes Doerfertecff11d2015-05-22 23:43:58 +0000332
333 // After a basic block was copied store all scalars that escape this block
334 // in their alloca. First the scalars that have dependences inside the SCoP,
335 // then the ones that might escape the SCoP.
336 generateScalarStores(Stmt, BB, BBMap, GlobalMap);
337
338 const Region &R = Stmt.getParent()->getRegion();
339 for (Instruction &Inst : *BB)
340 handleOutsideUsers(R, &Inst, BBMap[&Inst]);
341}
342
Tobias Grosser0164b8f2015-08-13 08:07:39 +0000343AllocaInst *BlockGenerator::getOrCreateAlloca(Value *ScalarBase,
Johannes Doerfertecff11d2015-05-22 23:43:58 +0000344 ScalarAllocaMapTy &Map,
345 const char *NameExt,
346 bool *IsNew) {
347
348 // Check if an alloca was cached for the base instruction.
349 AllocaInst *&Addr = Map[ScalarBase];
350
351 // If needed indicate if it was found already or will be created.
352 if (IsNew)
353 *IsNew = (Addr == nullptr);
354
355 // If no alloca was found create one and insert it in the entry block.
356 if (!Addr) {
357 auto *Ty = ScalarBase->getType();
358 Addr = new AllocaInst(Ty, ScalarBase->getName() + NameExt);
359 Addr->insertBefore(EntryBB->getFirstInsertionPt());
360 }
361
362 return Addr;
363}
364
Tobias Grosserb79a67d2015-08-28 08:23:35 +0000365AllocaInst *BlockGenerator::getOrCreateScalarAlloca(Value *ScalarBase) {
366 return getOrCreateAlloca(ScalarBase, ScalarMap, ".s2a");
367}
368
369AllocaInst *BlockGenerator::getOrCreatePHIAlloca(Value *ScalarBase) {
370 return getOrCreateAlloca(ScalarBase, PHIOpMap, ".phiops");
371}
372
Johannes Doerfertecff11d2015-05-22 23:43:58 +0000373void BlockGenerator::handleOutsideUsers(const Region &R, Instruction *Inst,
374 Value *InstCopy) {
Michael Kruseacb6ade2015-08-18 17:25:48 +0000375 // If there are escape users we get the alloca for this instruction and put
376 // it in the EscapeMap for later finalization. However, if the alloca was not
377 // created by an already handled scalar dependence we have to initialize it
378 // also. Lastly, if the instruction was copied multiple times we already did
379 // this and can exit.
380 if (EscapeMap.count(Inst))
381 return;
Johannes Doerferte69e1142015-08-18 11:56:00 +0000382
Johannes Doerfertecff11d2015-05-22 23:43:58 +0000383 EscapeUserVectorTy EscapeUsers;
384 for (User *U : Inst->users()) {
385
386 // Non-instruction user will never escape.
387 Instruction *UI = dyn_cast<Instruction>(U);
388 if (!UI)
389 continue;
390
Johannes Doerfertddb83d02015-08-16 08:35:40 +0000391 if (R.contains(UI))
Johannes Doerfertecff11d2015-05-22 23:43:58 +0000392 continue;
393
394 EscapeUsers.push_back(UI);
395 }
396
397 // Exit if no escape uses were found.
398 if (EscapeUsers.empty())
399 return;
400
Johannes Doerfertecff11d2015-05-22 23:43:58 +0000401 // Get or create an escape alloca for this instruction.
402 bool IsNew;
403 AllocaInst *ScalarAddr =
404 getOrCreateAlloca(Inst, ScalarMap, ".escape", &IsNew);
405
406 // Remember that this instruction has escape uses and the escape alloca.
407 EscapeMap[Inst] = std::make_pair(ScalarAddr, std::move(EscapeUsers));
408
409 // If the escape alloca was just created store the instruction in there,
410 // otherwise that happened already.
411 if (IsNew) {
412 assert(InstCopy && "Except PHIs every instruction should have a copy!");
413 Builder.CreateStore(InstCopy, ScalarAddr);
414 }
415}
416
417void BlockGenerator::generateScalarLoads(ScopStmt &Stmt,
418 const Instruction *Inst,
419 ValueMapT &BBMap) {
Tobias Grosserd4dd6ec2015-07-27 17:57:58 +0000420 auto *MAL = Stmt.lookupAccessesFor(Inst);
Johannes Doerfertecff11d2015-05-22 23:43:58 +0000421
Tobias Grosserd4dd6ec2015-07-27 17:57:58 +0000422 if (!MAL)
423 return;
Johannes Doerfertecff11d2015-05-22 23:43:58 +0000424
Tobias Grosserd4dd6ec2015-07-27 17:57:58 +0000425 for (MemoryAccess &MA : *MAL) {
426 AllocaInst *Address;
427 if (!MA.isScalar() || !MA.isRead())
428 continue;
Johannes Doerfertecff11d2015-05-22 23:43:58 +0000429
Tobias Grosser0164b8f2015-08-13 08:07:39 +0000430 auto Base = MA.getBaseAddr();
Johannes Doerfertecff11d2015-05-22 23:43:58 +0000431
Tobias Grosser92245222015-07-28 14:53:44 +0000432 if (MA.getScopArrayInfo()->isPHI())
Tobias Grosserb79a67d2015-08-28 08:23:35 +0000433 Address = getOrCreatePHIAlloca(Base);
Tobias Grosserd4dd6ec2015-07-27 17:57:58 +0000434 else
Tobias Grosserb79a67d2015-08-28 08:23:35 +0000435 Address = getOrCreateScalarAlloca(Base);
Tobias Grosserd4dd6ec2015-07-27 17:57:58 +0000436
437 BBMap[Base] = Builder.CreateLoad(Address, Address->getName() + ".reload");
Johannes Doerfertecff11d2015-05-22 23:43:58 +0000438 }
439}
440
441Value *BlockGenerator::getNewScalarValue(Value *ScalarValue, const Region &R,
442 ScalarAllocaMapTy &ReloadMap,
443 ValueMapT &BBMap,
444 ValueMapT &GlobalMap) {
445 // If the value we want to store is an instruction we might have demoted it
446 // in order to make it accessible here. In such a case a reload is
447 // necessary. If it is no instruction it will always be a value that
448 // dominates the current point and we can just use it. In total there are 4
449 // options:
450 // (1) The value is no instruction ==> use the value.
451 // (2) The value is an instruction that was split out of the region prior to
452 // code generation ==> use the instruction as it dominates the region.
453 // (3) The value is an instruction:
454 // (a) The value was defined in the current block, thus a copy is in
455 // the BBMap ==> use the mapped value.
456 // (b) The value was defined in a previous block, thus we demoted it
457 // earlier ==> use the reloaded value.
458 Instruction *ScalarValueInst = dyn_cast<Instruction>(ScalarValue);
459 if (!ScalarValueInst)
460 return ScalarValue;
461
462 if (!R.contains(ScalarValueInst)) {
463 if (Value *ScalarValueCopy = GlobalMap.lookup(ScalarValueInst))
464 return /* Case (3a) */ ScalarValueCopy;
465 else
466 return /* Case 2 */ ScalarValue;
467 }
468
469 if (Value *ScalarValueCopy = BBMap.lookup(ScalarValueInst))
470 return /* Case (3a) */ ScalarValueCopy;
471
472 // Case (3b)
Johannes Doerferte1fa6da2015-08-17 09:38:46 +0000473 Value *ReloadAddr = getOrCreateAlloca(ScalarValueInst, ReloadMap, ".s2a");
Johannes Doerfertecff11d2015-05-22 23:43:58 +0000474 ScalarValue =
475 Builder.CreateLoad(ReloadAddr, ReloadAddr->getName() + ".reload");
476
477 return ScalarValue;
478}
479
480void BlockGenerator::generateScalarStores(ScopStmt &Stmt, BasicBlock *BB,
481 ValueMapT &BBMap,
482 ValueMapT &GlobalMap) {
483 const Region &R = Stmt.getParent()->getRegion();
484
485 assert(Stmt.isBlockStmt() && BB == Stmt.getBasicBlock() &&
486 "Region statements need to use the generateScalarStores() "
487 "function in the RegionGenerator");
488
Johannes Doerfertecff11d2015-05-22 23:43:58 +0000489 for (MemoryAccess *MA : Stmt) {
Johannes Doerfertecff11d2015-05-22 23:43:58 +0000490 if (!MA->isScalar() || MA->isRead())
491 continue;
492
Tobias Grosser92245222015-07-28 14:53:44 +0000493 Instruction *Base = cast<Instruction>(MA->getBaseAddr());
Johannes Doerfertd86f2152015-08-17 10:58:17 +0000494 Value *Val = MA->getAccessValue();
Johannes Doerfertecff11d2015-05-22 23:43:58 +0000495
Tobias Grosser92245222015-07-28 14:53:44 +0000496 AllocaInst *Address = nullptr;
Johannes Doerfertd86f2152015-08-17 10:58:17 +0000497 if (MA->getScopArrayInfo()->isPHI())
Tobias Grosserb79a67d2015-08-28 08:23:35 +0000498 Address = getOrCreatePHIAlloca(Base);
Johannes Doerfertd86f2152015-08-17 10:58:17 +0000499 else
Tobias Grosserb79a67d2015-08-28 08:23:35 +0000500 Address = getOrCreateScalarAlloca(Base);
Johannes Doerfertd86f2152015-08-17 10:58:17 +0000501
Tobias Grosser92245222015-07-28 14:53:44 +0000502 Val = getNewScalarValue(Val, R, ScalarMap, BBMap, GlobalMap);
503 Builder.CreateStore(Val, Address);
Johannes Doerfertecff11d2015-05-22 23:43:58 +0000504 }
505}
506
507void BlockGenerator::createScalarInitialization(Region &R,
508 ValueMapT &GlobalMap) {
509 // The split block __just before__ the region and optimized region.
510 BasicBlock *SplitBB = R.getEnteringBlock();
511 BranchInst *SplitBBTerm = cast<BranchInst>(SplitBB->getTerminator());
512 assert(SplitBBTerm->getNumSuccessors() == 2 && "Bad region entering block!");
513
514 // Get the start block of the __optimized__ region.
515 BasicBlock *StartBB = SplitBBTerm->getSuccessor(0);
516 if (StartBB == R.getEntry())
517 StartBB = SplitBBTerm->getSuccessor(1);
518
519 // For each PHI predecessor outside the region store the incoming operand
520 // value prior to entering the optimized region.
521 Builder.SetInsertPoint(StartBB->getTerminator());
522
523 ScalarAllocaMapTy EmptyMap;
524 for (const auto &PHIOpMapping : PHIOpMap) {
525 const PHINode *PHI = cast<PHINode>(PHIOpMapping.getFirst());
526
527 // Check if this PHI has the split block as predecessor (that is the only
528 // possible predecessor outside the SCoP).
529 int idx = PHI->getBasicBlockIndex(SplitBB);
530 if (idx < 0)
531 continue;
532
533 Value *ScalarValue = PHI->getIncomingValue(idx);
534 ScalarValue =
535 getNewScalarValue(ScalarValue, R, EmptyMap, GlobalMap, GlobalMap);
536
537 // If the split block is the predecessor initialize the PHI operator alloca.
538 Builder.CreateStore(ScalarValue, PHIOpMapping.getSecond());
539 }
540}
541
542void BlockGenerator::createScalarFinalization(Region &R) {
543 // The exit block of the __unoptimized__ region.
544 BasicBlock *ExitBB = R.getExitingBlock();
545 // The merge block __just after__ the region and the optimized region.
546 BasicBlock *MergeBB = R.getExit();
547
548 // The exit block of the __optimized__ region.
549 BasicBlock *OptExitBB = *(pred_begin(MergeBB));
550 if (OptExitBB == ExitBB)
551 OptExitBB = *(++pred_begin(MergeBB));
552
553 Builder.SetInsertPoint(OptExitBB->getTerminator());
554 for (const auto &EscapeMapping : EscapeMap) {
555 // Extract the escaping instruction and the escaping users as well as the
556 // alloca the instruction was demoted to.
557 Instruction *EscapeInst = EscapeMapping.getFirst();
558 const auto &EscapeMappingValue = EscapeMapping.getSecond();
559 const EscapeUserVectorTy &EscapeUsers = EscapeMappingValue.second;
560 AllocaInst *ScalarAddr = EscapeMappingValue.first;
561
562 // Reload the demoted instruction in the optimized version of the SCoP.
563 Instruction *EscapeInstReload =
564 Builder.CreateLoad(ScalarAddr, EscapeInst->getName() + ".final_reload");
565
566 // Create the merge PHI that merges the optimized and unoptimized version.
567 PHINode *MergePHI = PHINode::Create(EscapeInst->getType(), 2,
568 EscapeInst->getName() + ".merge");
569 MergePHI->insertBefore(MergeBB->getFirstInsertionPt());
570
571 // Add the respective values to the merge PHI.
572 MergePHI->addIncoming(EscapeInstReload, OptExitBB);
573 MergePHI->addIncoming(EscapeInst, ExitBB);
574
575 // The information of scalar evolution about the escaping instruction needs
576 // to be revoked so the new merged instruction will be used.
577 if (SE.isSCEVable(EscapeInst->getType()))
578 SE.forgetValue(EscapeInst);
579
580 // Replace all uses of the demoted instruction with the merge PHI.
581 for (Instruction *EUser : EscapeUsers)
582 EUser->replaceUsesOfWith(EscapeInst, MergePHI);
583 }
584}
585
586void BlockGenerator::finalizeSCoP(Scop &S, ValueMapT &GlobalMap) {
587 createScalarInitialization(S.getRegion(), GlobalMap);
588 createScalarFinalization(S.getRegion());
Hongbin Zheng3b11a162012-04-25 13:16:49 +0000589}
590
Johannes Doerfertbe9c9112015-02-06 21:39:31 +0000591VectorBlockGenerator::VectorBlockGenerator(BlockGenerator &BlockGen,
592 VectorValueMapT &GlobalMaps,
593 std::vector<LoopToScevMapT> &VLTS,
594 isl_map *Schedule)
595 : BlockGenerator(BlockGen), GlobalMaps(GlobalMaps), VLTS(VLTS),
596 Schedule(Schedule) {
Sebastian Popa00a0292012-12-18 07:46:06 +0000597 assert(GlobalMaps.size() > 1 && "Only one vector lane found");
598 assert(Schedule && "No statement domain provided");
599}
Hongbin Zheng3b11a162012-04-25 13:16:49 +0000600
Johannes Doerfertbe9c9112015-02-06 21:39:31 +0000601Value *VectorBlockGenerator::getVectorValue(ScopStmt &Stmt, const Value *Old,
Tobias Grossere602a072013-05-07 07:30:56 +0000602 ValueMapT &VectorMap,
603 VectorValueMapT &ScalarMaps,
604 Loop *L) {
Hongbin Zhengfe11e282013-06-29 13:22:15 +0000605 if (Value *NewValue = VectorMap.lookup(Old))
606 return NewValue;
Hongbin Zheng3b11a162012-04-25 13:16:49 +0000607
608 int Width = getVectorWidth();
609
610 Value *Vector = UndefValue::get(VectorType::get(Old->getType(), Width));
611
612 for (int Lane = 0; Lane < Width; Lane++)
Tobias Grosserc14582f2013-02-05 18:01:29 +0000613 Vector = Builder.CreateInsertElement(
Johannes Doerfertbe9c9112015-02-06 21:39:31 +0000614 Vector, getNewValue(Stmt, Old, ScalarMaps[Lane], GlobalMaps[Lane],
615 VLTS[Lane], L),
Tobias Grosser7242ad92013-02-22 08:07:06 +0000616 Builder.getInt32(Lane));
Hongbin Zheng3b11a162012-04-25 13:16:49 +0000617
618 VectorMap[Old] = Vector;
619
620 return Vector;
621}
622
623Type *VectorBlockGenerator::getVectorPtrTy(const Value *Val, int Width) {
624 PointerType *PointerTy = dyn_cast<PointerType>(Val->getType());
625 assert(PointerTy && "PointerType expected");
626
627 Type *ScalarType = PointerTy->getElementType();
628 VectorType *VectorType = VectorType::get(ScalarType, Width);
629
630 return PointerType::getUnqual(VectorType);
631}
632
Johannes Doerfertbe9c9112015-02-06 21:39:31 +0000633Value *VectorBlockGenerator::generateStrideOneLoad(
634 ScopStmt &Stmt, const LoadInst *Load, VectorValueMapT &ScalarMaps,
Tobias Grosser2d1ed0b2015-08-27 07:28:16 +0000635 __isl_keep isl_id_to_ast_expr *NewAccesses, bool NegativeStride = false) {
Tobias Grosser0dd463f2014-03-19 19:27:24 +0000636 unsigned VectorWidth = getVectorWidth();
Hongbin Zheng3b11a162012-04-25 13:16:49 +0000637 const Value *Pointer = Load->getPointerOperand();
Tobias Grosser0dd463f2014-03-19 19:27:24 +0000638 Type *VectorPtrType = getVectorPtrTy(Pointer, VectorWidth);
639 unsigned Offset = NegativeStride ? VectorWidth - 1 : 0;
640
Tobias Grosser5a56cbf2014-04-16 07:33:47 +0000641 Value *NewPointer = nullptr;
Tobias Grosser2d1ed0b2015-08-27 07:28:16 +0000642 NewPointer =
643 generateLocationAccessed(Stmt, Load, Pointer, ScalarMaps[Offset],
644 GlobalMaps[Offset], VLTS[Offset], NewAccesses);
Tobias Grosserc14582f2013-02-05 18:01:29 +0000645 Value *VectorPtr =
646 Builder.CreateBitCast(NewPointer, VectorPtrType, "vector_ptr");
647 LoadInst *VecLoad =
648 Builder.CreateLoad(VectorPtr, Load->getName() + "_p_vec_full");
Hongbin Zheng3b11a162012-04-25 13:16:49 +0000649 if (!Aligned)
650 VecLoad->setAlignment(8);
651
Tobias Grosser0dd463f2014-03-19 19:27:24 +0000652 if (NegativeStride) {
653 SmallVector<Constant *, 16> Indices;
654 for (int i = VectorWidth - 1; i >= 0; i--)
655 Indices.push_back(ConstantInt::get(Builder.getInt32Ty(), i));
656 Constant *SV = llvm::ConstantVector::get(Indices);
657 Value *RevVecLoad = Builder.CreateShuffleVector(
658 VecLoad, VecLoad, SV, Load->getName() + "_reverse");
659 return RevVecLoad;
660 }
661
Hongbin Zheng3b11a162012-04-25 13:16:49 +0000662 return VecLoad;
663}
664
Tobias Grosser2d1ed0b2015-08-27 07:28:16 +0000665Value *VectorBlockGenerator::generateStrideZeroLoad(
666 ScopStmt &Stmt, const LoadInst *Load, ValueMapT &BBMap,
667 __isl_keep isl_id_to_ast_expr *NewAccesses) {
Hongbin Zheng3b11a162012-04-25 13:16:49 +0000668 const Value *Pointer = Load->getPointerOperand();
669 Type *VectorPtrType = getVectorPtrTy(Pointer, 1);
Tobias Grosser2d1ed0b2015-08-27 07:28:16 +0000670 Value *NewPointer = generateLocationAccessed(
671 Stmt, Load, Pointer, BBMap, GlobalMaps[0], VLTS[0], NewAccesses);
Hongbin Zheng3b11a162012-04-25 13:16:49 +0000672 Value *VectorPtr = Builder.CreateBitCast(NewPointer, VectorPtrType,
673 Load->getName() + "_p_vec_p");
Tobias Grosserc14582f2013-02-05 18:01:29 +0000674 LoadInst *ScalarLoad =
675 Builder.CreateLoad(VectorPtr, Load->getName() + "_p_splat_one");
Hongbin Zheng3b11a162012-04-25 13:16:49 +0000676
677 if (!Aligned)
678 ScalarLoad->setAlignment(8);
679
Tobias Grosserc14582f2013-02-05 18:01:29 +0000680 Constant *SplatVector = Constant::getNullValue(
681 VectorType::get(Builder.getInt32Ty(), getVectorWidth()));
Hongbin Zheng3b11a162012-04-25 13:16:49 +0000682
Tobias Grosserc14582f2013-02-05 18:01:29 +0000683 Value *VectorLoad = Builder.CreateShuffleVector(
684 ScalarLoad, ScalarLoad, SplatVector, Load->getName() + "_p_splat");
Hongbin Zheng3b11a162012-04-25 13:16:49 +0000685 return VectorLoad;
686}
687
Johannes Doerfertbe9c9112015-02-06 21:39:31 +0000688Value *VectorBlockGenerator::generateUnknownStrideLoad(
Tobias Grosser2d1ed0b2015-08-27 07:28:16 +0000689 ScopStmt &Stmt, const LoadInst *Load, VectorValueMapT &ScalarMaps,
690 __isl_keep isl_id_to_ast_expr *NewAccesses
691
692 ) {
Hongbin Zheng3b11a162012-04-25 13:16:49 +0000693 int VectorWidth = getVectorWidth();
694 const Value *Pointer = Load->getPointerOperand();
695 VectorType *VectorType = VectorType::get(
Tobias Grosserc14582f2013-02-05 18:01:29 +0000696 dyn_cast<PointerType>(Pointer->getType())->getElementType(), VectorWidth);
Hongbin Zheng3b11a162012-04-25 13:16:49 +0000697
698 Value *Vector = UndefValue::get(VectorType);
699
700 for (int i = 0; i < VectorWidth; i++) {
Tobias Grosser2d1ed0b2015-08-27 07:28:16 +0000701 Value *NewPointer =
702 generateLocationAccessed(Stmt, Load, Pointer, ScalarMaps[i],
703 GlobalMaps[i], VLTS[i], NewAccesses);
Tobias Grosserc14582f2013-02-05 18:01:29 +0000704 Value *ScalarLoad =
705 Builder.CreateLoad(NewPointer, Load->getName() + "_p_scalar_");
706 Vector = Builder.CreateInsertElement(
707 Vector, ScalarLoad, Builder.getInt32(i), Load->getName() + "_p_vec_");
Hongbin Zheng3b11a162012-04-25 13:16:49 +0000708 }
709
710 return Vector;
711}
712
Tobias Grosser2d1ed0b2015-08-27 07:28:16 +0000713void VectorBlockGenerator::generateLoad(
714 ScopStmt &Stmt, const LoadInst *Load, ValueMapT &VectorMap,
715 VectorValueMapT &ScalarMaps, __isl_keep isl_id_to_ast_expr *NewAccesses) {
Tobias Grosser28736452015-03-23 07:00:36 +0000716 if (!VectorType::isValidElementType(Load->getType())) {
Hongbin Zheng3b11a162012-04-25 13:16:49 +0000717 for (int i = 0; i < getVectorWidth(); i++)
Tobias Grosser2d1ed0b2015-08-27 07:28:16 +0000718 ScalarMaps[i][Load] = generateScalarLoad(
719 Stmt, Load, ScalarMaps[i], GlobalMaps[i], VLTS[i], NewAccesses);
Hongbin Zheng3b11a162012-04-25 13:16:49 +0000720 return;
721 }
722
Johannes Doerfertbe9c9112015-02-06 21:39:31 +0000723 const MemoryAccess &Access = Stmt.getAccessFor(Load);
Hongbin Zheng3b11a162012-04-25 13:16:49 +0000724
Tobias Grosser95493982014-04-18 09:46:35 +0000725 // Make sure we have scalar values available to access the pointer to
726 // the data location.
727 extractScalarValues(Load, VectorMap, ScalarMaps);
728
Hongbin Zheng3b11a162012-04-25 13:16:49 +0000729 Value *NewLoad;
Sebastian Popa00a0292012-12-18 07:46:06 +0000730 if (Access.isStrideZero(isl_map_copy(Schedule)))
Tobias Grosser2d1ed0b2015-08-27 07:28:16 +0000731 NewLoad = generateStrideZeroLoad(Stmt, Load, ScalarMaps[0], NewAccesses);
Sebastian Popa00a0292012-12-18 07:46:06 +0000732 else if (Access.isStrideOne(isl_map_copy(Schedule)))
Tobias Grosser2d1ed0b2015-08-27 07:28:16 +0000733 NewLoad = generateStrideOneLoad(Stmt, Load, ScalarMaps, NewAccesses);
Tobias Grosser0dd463f2014-03-19 19:27:24 +0000734 else if (Access.isStrideX(isl_map_copy(Schedule), -1))
Tobias Grosser2d1ed0b2015-08-27 07:28:16 +0000735 NewLoad = generateStrideOneLoad(Stmt, Load, ScalarMaps, NewAccesses, true);
Hongbin Zheng3b11a162012-04-25 13:16:49 +0000736 else
Tobias Grosser2d1ed0b2015-08-27 07:28:16 +0000737 NewLoad = generateUnknownStrideLoad(Stmt, Load, ScalarMaps, NewAccesses);
Hongbin Zheng3b11a162012-04-25 13:16:49 +0000738
739 VectorMap[Load] = NewLoad;
740}
741
Johannes Doerfertbe9c9112015-02-06 21:39:31 +0000742void VectorBlockGenerator::copyUnaryInst(ScopStmt &Stmt,
743 const UnaryInstruction *Inst,
Hongbin Zheng3b11a162012-04-25 13:16:49 +0000744 ValueMapT &VectorMap,
745 VectorValueMapT &ScalarMaps) {
746 int VectorWidth = getVectorWidth();
Johannes Doerfertbe9c9112015-02-06 21:39:31 +0000747 Value *NewOperand = getVectorValue(Stmt, Inst->getOperand(0), VectorMap,
748 ScalarMaps, getLoopForInst(Inst));
Hongbin Zheng3b11a162012-04-25 13:16:49 +0000749
750 assert(isa<CastInst>(Inst) && "Can not generate vector code for instruction");
751
752 const CastInst *Cast = dyn_cast<CastInst>(Inst);
753 VectorType *DestType = VectorType::get(Inst->getType(), VectorWidth);
754 VectorMap[Inst] = Builder.CreateCast(Cast->getOpcode(), NewOperand, DestType);
755}
756
Johannes Doerfertbe9c9112015-02-06 21:39:31 +0000757void VectorBlockGenerator::copyBinaryInst(ScopStmt &Stmt,
758 const BinaryOperator *Inst,
Hongbin Zheng3b11a162012-04-25 13:16:49 +0000759 ValueMapT &VectorMap,
760 VectorValueMapT &ScalarMaps) {
Tobias Grosser369430f2013-03-22 23:42:53 +0000761 Loop *L = getLoopForInst(Inst);
Hongbin Zheng3b11a162012-04-25 13:16:49 +0000762 Value *OpZero = Inst->getOperand(0);
763 Value *OpOne = Inst->getOperand(1);
764
765 Value *NewOpZero, *NewOpOne;
Johannes Doerfertbe9c9112015-02-06 21:39:31 +0000766 NewOpZero = getVectorValue(Stmt, OpZero, VectorMap, ScalarMaps, L);
767 NewOpOne = getVectorValue(Stmt, OpOne, VectorMap, ScalarMaps, L);
Hongbin Zheng3b11a162012-04-25 13:16:49 +0000768
Tobias Grosser1bb59b02012-12-29 23:47:38 +0000769 Value *NewInst = Builder.CreateBinOp(Inst->getOpcode(), NewOpZero, NewOpOne,
Hongbin Zheng3b11a162012-04-25 13:16:49 +0000770 Inst->getName() + "p_vec");
771 VectorMap[Inst] = NewInst;
772}
773
Tobias Grosser2d1ed0b2015-08-27 07:28:16 +0000774void VectorBlockGenerator::copyStore(
775 ScopStmt &Stmt, const StoreInst *Store, ValueMapT &VectorMap,
776 VectorValueMapT &ScalarMaps, __isl_keep isl_id_to_ast_expr *NewAccesses) {
Johannes Doerfertbe9c9112015-02-06 21:39:31 +0000777 const MemoryAccess &Access = Stmt.getAccessFor(Store);
Hongbin Zheng3b11a162012-04-25 13:16:49 +0000778
779 const Value *Pointer = Store->getPointerOperand();
Johannes Doerfertbe9c9112015-02-06 21:39:31 +0000780 Value *Vector = getVectorValue(Stmt, Store->getValueOperand(), VectorMap,
Tobias Grosser369430f2013-03-22 23:42:53 +0000781 ScalarMaps, getLoopForInst(Store));
Hongbin Zheng3b11a162012-04-25 13:16:49 +0000782
Tobias Grosser50fd7012014-04-17 23:13:49 +0000783 // Make sure we have scalar values available to access the pointer to
784 // the data location.
785 extractScalarValues(Store, VectorMap, ScalarMaps);
786
Sebastian Popa00a0292012-12-18 07:46:06 +0000787 if (Access.isStrideOne(isl_map_copy(Schedule))) {
Johannes Doerfert1947f862014-10-08 20:18:32 +0000788 Type *VectorPtrType = getVectorPtrTy(Pointer, getVectorWidth());
Tobias Grosser2d1ed0b2015-08-27 07:28:16 +0000789 Value *NewPointer =
790 generateLocationAccessed(Stmt, Store, Pointer, ScalarMaps[0],
791 GlobalMaps[0], VLTS[0], NewAccesses);
Hongbin Zheng3b11a162012-04-25 13:16:49 +0000792
Tobias Grosserc14582f2013-02-05 18:01:29 +0000793 Value *VectorPtr =
794 Builder.CreateBitCast(NewPointer, VectorPtrType, "vector_ptr");
Hongbin Zheng3b11a162012-04-25 13:16:49 +0000795 StoreInst *Store = Builder.CreateStore(Vector, VectorPtr);
796
797 if (!Aligned)
798 Store->setAlignment(8);
799 } else {
800 for (unsigned i = 0; i < ScalarMaps.size(); i++) {
Tobias Grosser1bb59b02012-12-29 23:47:38 +0000801 Value *Scalar = Builder.CreateExtractElement(Vector, Builder.getInt32(i));
Tobias Grosser2d1ed0b2015-08-27 07:28:16 +0000802 Value *NewPointer =
803 generateLocationAccessed(Stmt, Store, Pointer, ScalarMaps[i],
804 GlobalMaps[i], VLTS[i], NewAccesses);
Hongbin Zheng3b11a162012-04-25 13:16:49 +0000805 Builder.CreateStore(Scalar, NewPointer);
806 }
807 }
808}
809
810bool VectorBlockGenerator::hasVectorOperands(const Instruction *Inst,
811 ValueMapT &VectorMap) {
Tobias Grosser91f5b262014-06-04 08:06:40 +0000812 for (Value *Operand : Inst->operands())
813 if (VectorMap.count(Operand))
Hongbin Zheng3b11a162012-04-25 13:16:49 +0000814 return true;
815 return false;
816}
817
818bool VectorBlockGenerator::extractScalarValues(const Instruction *Inst,
819 ValueMapT &VectorMap,
820 VectorValueMapT &ScalarMaps) {
821 bool HasVectorOperand = false;
822 int VectorWidth = getVectorWidth();
823
Tobias Grosser91f5b262014-06-04 08:06:40 +0000824 for (Value *Operand : Inst->operands()) {
825 ValueMapT::iterator VecOp = VectorMap.find(Operand);
Hongbin Zheng3b11a162012-04-25 13:16:49 +0000826
827 if (VecOp == VectorMap.end())
828 continue;
829
830 HasVectorOperand = true;
831 Value *NewVector = VecOp->second;
832
833 for (int i = 0; i < VectorWidth; ++i) {
834 ValueMapT &SM = ScalarMaps[i];
835
836 // If there is one scalar extracted, all scalar elements should have
837 // already been extracted by the code here. So no need to check for the
838 // existance of all of them.
Tobias Grosser91f5b262014-06-04 08:06:40 +0000839 if (SM.count(Operand))
Hongbin Zheng3b11a162012-04-25 13:16:49 +0000840 break;
841
Tobias Grosser91f5b262014-06-04 08:06:40 +0000842 SM[Operand] =
843 Builder.CreateExtractElement(NewVector, Builder.getInt32(i));
Hongbin Zheng3b11a162012-04-25 13:16:49 +0000844 }
845 }
846
847 return HasVectorOperand;
848}
849
Tobias Grosser2d1ed0b2015-08-27 07:28:16 +0000850void VectorBlockGenerator::copyInstScalarized(
851 ScopStmt &Stmt, const Instruction *Inst, ValueMapT &VectorMap,
852 VectorValueMapT &ScalarMaps, __isl_keep isl_id_to_ast_expr *NewAccesses) {
Hongbin Zheng3b11a162012-04-25 13:16:49 +0000853 bool HasVectorOperand;
854 int VectorWidth = getVectorWidth();
855
856 HasVectorOperand = extractScalarValues(Inst, VectorMap, ScalarMaps);
857
858 for (int VectorLane = 0; VectorLane < getVectorWidth(); VectorLane++)
Johannes Doerfertbe9c9112015-02-06 21:39:31 +0000859 BlockGenerator::copyInstruction(Stmt, Inst, ScalarMaps[VectorLane],
Tobias Grosser2d1ed0b2015-08-27 07:28:16 +0000860 GlobalMaps[VectorLane], VLTS[VectorLane],
861 NewAccesses);
Hongbin Zheng3b11a162012-04-25 13:16:49 +0000862
863 if (!VectorType::isValidElementType(Inst->getType()) || !HasVectorOperand)
864 return;
865
866 // Make the result available as vector value.
867 VectorType *VectorType = VectorType::get(Inst->getType(), VectorWidth);
868 Value *Vector = UndefValue::get(VectorType);
869
870 for (int i = 0; i < VectorWidth; i++)
871 Vector = Builder.CreateInsertElement(Vector, ScalarMaps[i][Inst],
872 Builder.getInt32(i));
873
874 VectorMap[Inst] = Vector;
875}
876
Tobias Grosserc14582f2013-02-05 18:01:29 +0000877int VectorBlockGenerator::getVectorWidth() { return GlobalMaps.size(); }
Hongbin Zheng3b11a162012-04-25 13:16:49 +0000878
Tobias Grosser2d1ed0b2015-08-27 07:28:16 +0000879void VectorBlockGenerator::copyInstruction(
880 ScopStmt &Stmt, const Instruction *Inst, ValueMapT &VectorMap,
881 VectorValueMapT &ScalarMaps, __isl_keep isl_id_to_ast_expr *NewAccesses) {
Hongbin Zheng3b11a162012-04-25 13:16:49 +0000882 // Terminator instructions control the control flow. They are explicitly
883 // expressed in the clast and do not need to be copied.
884 if (Inst->isTerminator())
885 return;
886
Johannes Doerfert1ef52332015-02-08 20:50:42 +0000887 if (canSynthesize(Inst, &LI, &SE, &Stmt.getParent()->getRegion()))
Tobias Grossere71c6ab2012-04-27 16:36:14 +0000888 return;
889
Hongbin Zheng3b11a162012-04-25 13:16:49 +0000890 if (const LoadInst *Load = dyn_cast<LoadInst>(Inst)) {
Tobias Grosser2d1ed0b2015-08-27 07:28:16 +0000891 generateLoad(Stmt, Load, VectorMap, ScalarMaps, NewAccesses);
Hongbin Zheng3b11a162012-04-25 13:16:49 +0000892 return;
893 }
894
895 if (hasVectorOperands(Inst, VectorMap)) {
896 if (const StoreInst *Store = dyn_cast<StoreInst>(Inst)) {
Tobias Grosser2d1ed0b2015-08-27 07:28:16 +0000897 copyStore(Stmt, Store, VectorMap, ScalarMaps, NewAccesses);
Hongbin Zheng3b11a162012-04-25 13:16:49 +0000898 return;
899 }
900
901 if (const UnaryInstruction *Unary = dyn_cast<UnaryInstruction>(Inst)) {
Johannes Doerfertbe9c9112015-02-06 21:39:31 +0000902 copyUnaryInst(Stmt, Unary, VectorMap, ScalarMaps);
Hongbin Zheng3b11a162012-04-25 13:16:49 +0000903 return;
904 }
905
906 if (const BinaryOperator *Binary = dyn_cast<BinaryOperator>(Inst)) {
Johannes Doerfertbe9c9112015-02-06 21:39:31 +0000907 copyBinaryInst(Stmt, Binary, VectorMap, ScalarMaps);
Hongbin Zheng3b11a162012-04-25 13:16:49 +0000908 return;
909 }
910
911 // Falltrough: We generate scalar instructions, if we don't know how to
912 // generate vector code.
913 }
914
Tobias Grosser2d1ed0b2015-08-27 07:28:16 +0000915 copyInstScalarized(Stmt, Inst, VectorMap, ScalarMaps, NewAccesses);
Hongbin Zheng3b11a162012-04-25 13:16:49 +0000916}
917
Tobias Grosser2d1ed0b2015-08-27 07:28:16 +0000918void VectorBlockGenerator::copyStmt(
919 ScopStmt &Stmt, __isl_keep isl_id_to_ast_expr *NewAccesses) {
Johannes Doerfert275a1752015-02-24 16:16:32 +0000920 assert(Stmt.isBlockStmt() && "TODO: Only block statements can be copied by "
921 "the vector block generator");
922
Johannes Doerfertbe9c9112015-02-06 21:39:31 +0000923 BasicBlock *BB = Stmt.getBasicBlock();
Tobias Grosserc14582f2013-02-05 18:01:29 +0000924 BasicBlock *CopyBB =
Johannes Doerfertb4f08eb2015-02-23 13:51:35 +0000925 SplitBlock(Builder.GetInsertBlock(), Builder.GetInsertPoint(), &DT, &LI);
Hongbin Zheng3b11a162012-04-25 13:16:49 +0000926 CopyBB->setName("polly.stmt." + BB->getName());
927 Builder.SetInsertPoint(CopyBB->begin());
928
929 // Create two maps that store the mapping from the original instructions of
930 // the old basic block to their copies in the new basic block. Those maps
931 // are basic block local.
932 //
933 // As vector code generation is supported there is one map for scalar values
934 // and one for vector values.
935 //
936 // In case we just do scalar code generation, the vectorMap is not used and
937 // the scalarMap has just one dimension, which contains the mapping.
938 //
939 // In case vector code generation is done, an instruction may either appear
940 // in the vector map once (as it is calculating >vectorwidth< values at a
941 // time. Or (if the values are calculated using scalar operations), it
942 // appears once in every dimension of the scalarMap.
943 VectorValueMapT ScalarBlockMap(getVectorWidth());
944 ValueMapT VectorBlockMap;
945
Tobias Grosser91f5b262014-06-04 08:06:40 +0000946 for (Instruction &Inst : *BB)
Tobias Grosser2d1ed0b2015-08-27 07:28:16 +0000947 copyInstruction(Stmt, &Inst, VectorBlockMap, ScalarBlockMap, NewAccesses);
Hongbin Zheng3b11a162012-04-25 13:16:49 +0000948}
Johannes Doerfert275a1752015-02-24 16:16:32 +0000949
Johannes Doerfertecff11d2015-05-22 23:43:58 +0000950BasicBlock *RegionGenerator::repairDominance(BasicBlock *BB,
951 BasicBlock *BBCopy) {
Johannes Doerfert514f6ef2015-02-27 18:29:04 +0000952
953 BasicBlock *BBIDom = DT.getNode(BB)->getIDom()->getBlock();
954 BasicBlock *BBCopyIDom = BlockMap.lookup(BBIDom);
955
956 if (BBCopyIDom)
957 DT.changeImmediateDominator(BBCopy, BBCopyIDom);
958
959 return BBCopyIDom;
960}
961
Johannes Doerfert275a1752015-02-24 16:16:32 +0000962void RegionGenerator::copyStmt(ScopStmt &Stmt, ValueMapT &GlobalMap,
Tobias Grosser2d1ed0b2015-08-27 07:28:16 +0000963 LoopToScevMapT &LTS,
964 isl_id_to_ast_expr *IdToAstExp) {
Johannes Doerfert275a1752015-02-24 16:16:32 +0000965 assert(Stmt.isRegionStmt() &&
Tobias Grosserd3f21832015-08-01 06:26:51 +0000966 "Only region statements can be copied by the region generator");
Johannes Doerfert275a1752015-02-24 16:16:32 +0000967
Johannes Doerfertecff11d2015-05-22 23:43:58 +0000968 // Forget all old mappings.
969 BlockMap.clear();
970 RegionMaps.clear();
971 IncompletePHINodeMap.clear();
972
Johannes Doerfert275a1752015-02-24 16:16:32 +0000973 // The region represented by the statement.
974 Region *R = Stmt.getRegion();
975
Johannes Doerfertecff11d2015-05-22 23:43:58 +0000976 // Create a dedicated entry for the region where we can reload all demoted
977 // inputs.
978 BasicBlock *EntryBB = R->getEntry();
979 BasicBlock *EntryBBCopy =
980 SplitBlock(Builder.GetInsertBlock(), Builder.GetInsertPoint(), &DT, &LI);
981 EntryBBCopy->setName("polly.stmt." + EntryBB->getName() + ".entry");
982 Builder.SetInsertPoint(EntryBBCopy->begin());
Johannes Doerfert514f6ef2015-02-27 18:29:04 +0000983
Johannes Doerfertecff11d2015-05-22 23:43:58 +0000984 for (auto PI = pred_begin(EntryBB), PE = pred_end(EntryBB); PI != PE; ++PI)
985 if (!R->contains(*PI))
986 BlockMap[*PI] = EntryBBCopy;
Johannes Doerfert275a1752015-02-24 16:16:32 +0000987
988 // Iterate over all blocks in the region in a breadth-first search.
989 std::deque<BasicBlock *> Blocks;
990 SmallPtrSet<BasicBlock *, 8> SeenBlocks;
Johannes Doerfertecff11d2015-05-22 23:43:58 +0000991 Blocks.push_back(EntryBB);
992 SeenBlocks.insert(EntryBB);
Johannes Doerfert275a1752015-02-24 16:16:32 +0000993
994 while (!Blocks.empty()) {
995 BasicBlock *BB = Blocks.front();
996 Blocks.pop_front();
997
Johannes Doerfert514f6ef2015-02-27 18:29:04 +0000998 // First split the block and update dominance information.
999 BasicBlock *BBCopy = splitBB(BB);
Johannes Doerfertecff11d2015-05-22 23:43:58 +00001000 BasicBlock *BBCopyIDom = repairDominance(BB, BBCopy);
1001
1002 // In order to remap PHI nodes we store also basic block mappings.
1003 BlockMap[BB] = BBCopy;
Johannes Doerfert514f6ef2015-02-27 18:29:04 +00001004
1005 // Get the mapping for this block and initialize it with the mapping
1006 // available at its immediate dominator (in the new region).
1007 ValueMapT &RegionMap = RegionMaps[BBCopy];
1008 RegionMap = RegionMaps[BBCopyIDom];
1009
Johannes Doerfert275a1752015-02-24 16:16:32 +00001010 // Copy the block with the BlockGenerator.
Tobias Grosser2d1ed0b2015-08-27 07:28:16 +00001011 copyBB(Stmt, BB, BBCopy, RegionMap, GlobalMap, LTS, IdToAstExp);
Johannes Doerfert275a1752015-02-24 16:16:32 +00001012
Johannes Doerfertecff11d2015-05-22 23:43:58 +00001013 // In order to remap PHI nodes we store also basic block mappings.
1014 BlockMap[BB] = BBCopy;
1015
1016 // Add values to incomplete PHI nodes waiting for this block to be copied.
1017 for (const PHINodePairTy &PHINodePair : IncompletePHINodeMap[BB])
1018 addOperandToPHI(Stmt, PHINodePair.first, PHINodePair.second, BB,
1019 GlobalMap, LTS);
1020 IncompletePHINodeMap[BB].clear();
1021
Johannes Doerfert275a1752015-02-24 16:16:32 +00001022 // And continue with new successors inside the region.
1023 for (auto SI = succ_begin(BB), SE = succ_end(BB); SI != SE; SI++)
1024 if (R->contains(*SI) && SeenBlocks.insert(*SI).second)
1025 Blocks.push_back(*SI);
Johannes Doerfert275a1752015-02-24 16:16:32 +00001026 }
1027
1028 // Now create a new dedicated region exit block and add it to the region map.
Johannes Doerfert514f6ef2015-02-27 18:29:04 +00001029 BasicBlock *ExitBBCopy =
Johannes Doerfert275a1752015-02-24 16:16:32 +00001030 SplitBlock(Builder.GetInsertBlock(), Builder.GetInsertPoint(), &DT, &LI);
Johannes Doerfertecff11d2015-05-22 23:43:58 +00001031 ExitBBCopy->setName("polly.stmt." + R->getExit()->getName() + ".exit");
Johannes Doerfert514f6ef2015-02-27 18:29:04 +00001032 BlockMap[R->getExit()] = ExitBBCopy;
1033
Johannes Doerfertecff11d2015-05-22 23:43:58 +00001034 repairDominance(R->getExit(), ExitBBCopy);
Johannes Doerfert275a1752015-02-24 16:16:32 +00001035
1036 // As the block generator doesn't handle control flow we need to add the
1037 // region control flow by hand after all blocks have been copied.
1038 for (BasicBlock *BB : SeenBlocks) {
1039
1040 BranchInst *BI = cast<BranchInst>(BB->getTerminator());
1041
Johannes Doerfert514f6ef2015-02-27 18:29:04 +00001042 BasicBlock *BBCopy = BlockMap[BB];
Johannes Doerfert275a1752015-02-24 16:16:32 +00001043 Instruction *BICopy = BBCopy->getTerminator();
1044
Johannes Doerfert514f6ef2015-02-27 18:29:04 +00001045 ValueMapT &RegionMap = RegionMaps[BBCopy];
1046 RegionMap.insert(BlockMap.begin(), BlockMap.end());
1047
Tobias Grosser45e79442015-08-01 09:07:57 +00001048 Builder.SetInsertPoint(BICopy);
Johannes Doerfert275a1752015-02-24 16:16:32 +00001049 copyInstScalar(Stmt, BI, RegionMap, GlobalMap, LTS);
1050 BICopy->eraseFromParent();
1051 }
1052
Johannes Doerfertecff11d2015-05-22 23:43:58 +00001053 // Add counting PHI nodes to all loops in the region that can be used as
1054 // replacement for SCEVs refering to the old loop.
1055 for (BasicBlock *BB : SeenBlocks) {
1056 Loop *L = LI.getLoopFor(BB);
1057 if (L == nullptr || L->getHeader() != BB)
1058 continue;
1059
1060 BasicBlock *BBCopy = BlockMap[BB];
1061 Value *NullVal = Builder.getInt32(0);
1062 PHINode *LoopPHI =
1063 PHINode::Create(Builder.getInt32Ty(), 2, "polly.subregion.iv");
1064 Instruction *LoopPHIInc = BinaryOperator::CreateAdd(
1065 LoopPHI, Builder.getInt32(1), "polly.subregion.iv.inc");
1066 LoopPHI->insertBefore(BBCopy->begin());
1067 LoopPHIInc->insertBefore(BBCopy->getTerminator());
1068
1069 for (auto *PredBB : make_range(pred_begin(BB), pred_end(BB))) {
1070 if (!R->contains(PredBB))
1071 continue;
1072 if (L->contains(PredBB))
1073 LoopPHI->addIncoming(LoopPHIInc, BlockMap[PredBB]);
1074 else
1075 LoopPHI->addIncoming(NullVal, BlockMap[PredBB]);
1076 }
1077
1078 for (auto *PredBBCopy : make_range(pred_begin(BBCopy), pred_end(BBCopy)))
1079 if (LoopPHI->getBasicBlockIndex(PredBBCopy) < 0)
1080 LoopPHI->addIncoming(NullVal, PredBBCopy);
1081
1082 LTS[L] = SE.getUnknown(LoopPHI);
1083 }
1084
1085 // Add all mappings from the region to the global map so outside uses will use
1086 // the copied instructions.
1087 for (auto &BBMap : RegionMaps)
1088 GlobalMap.insert(BBMap.second.begin(), BBMap.second.end());
1089
Johannes Doerfert275a1752015-02-24 16:16:32 +00001090 // Reset the old insert point for the build.
Johannes Doerfert514f6ef2015-02-27 18:29:04 +00001091 Builder.SetInsertPoint(ExitBBCopy->begin());
Johannes Doerfert275a1752015-02-24 16:16:32 +00001092}
Johannes Doerfertecff11d2015-05-22 23:43:58 +00001093
1094void RegionGenerator::generateScalarLoads(ScopStmt &Stmt,
1095 const Instruction *Inst,
1096 ValueMapT &BBMap) {
1097
1098 // Inside a non-affine region PHI nodes are copied not demoted. Once the
1099 // phi is copied it will reload all inputs from outside the region, hence
1100 // we do not need to generate code for the read access of the operands of a
1101 // PHI.
1102 if (isa<PHINode>(Inst))
1103 return;
1104
1105 return BlockGenerator::generateScalarLoads(Stmt, Inst, BBMap);
1106}
1107
1108void RegionGenerator::generateScalarStores(ScopStmt &Stmt, BasicBlock *BB,
1109 ValueMapT &BBMap,
1110 ValueMapT &GlobalMap) {
1111 const Region &R = Stmt.getParent()->getRegion();
1112
Tobias Grosser75296902015-08-21 19:23:21 +00001113 assert(Stmt.getRegion() &&
1114 "Block statements need to use the generateScalarStores() "
1115 "function in the BlockGenerator");
Johannes Doerfertecff11d2015-05-22 23:43:58 +00001116
Johannes Doerfertecff11d2015-05-22 23:43:58 +00001117 for (MemoryAccess *MA : Stmt) {
1118
Johannes Doerfertecff11d2015-05-22 23:43:58 +00001119 if (!MA->isScalar() || MA->isRead())
1120 continue;
1121
1122 Instruction *ScalarBase = cast<Instruction>(MA->getBaseAddr());
1123 Instruction *ScalarInst = MA->getAccessInstruction();
Johannes Doerfertecff11d2015-05-22 23:43:58 +00001124
Tobias Grosser62139132015-08-02 16:17:41 +00001125 // Only generate accesses that belong to this basic block.
1126 if (ScalarInst->getParent() != BB)
1127 continue;
1128
Johannes Doerfertd86f2152015-08-17 10:58:17 +00001129 Value *Val = MA->getAccessValue();
Johannes Doerfertecff11d2015-05-22 23:43:58 +00001130 AllocaInst *ScalarAddr = nullptr;
1131
Johannes Doerfertd86f2152015-08-17 10:58:17 +00001132 if (MA->getScopArrayInfo()->isPHI())
Tobias Grosserb79a67d2015-08-28 08:23:35 +00001133 ScalarAddr = getOrCreatePHIAlloca(ScalarBase);
Johannes Doerfertd86f2152015-08-17 10:58:17 +00001134 else
Tobias Grosserb79a67d2015-08-28 08:23:35 +00001135 ScalarAddr = getOrCreateScalarAlloca(ScalarBase);
Johannes Doerfertecff11d2015-05-22 23:43:58 +00001136
Tobias Grosser92245222015-07-28 14:53:44 +00001137 Val = getNewScalarValue(Val, R, ScalarMap, BBMap, GlobalMap);
1138 Builder.CreateStore(Val, ScalarAddr);
Johannes Doerfertecff11d2015-05-22 23:43:58 +00001139 }
1140}
1141
1142void RegionGenerator::addOperandToPHI(ScopStmt &Stmt, const PHINode *PHI,
1143 PHINode *PHICopy, BasicBlock *IncomingBB,
1144 ValueMapT &GlobalMap,
1145 LoopToScevMapT &LTS) {
1146 Region *StmtR = Stmt.getRegion();
1147
1148 // If the incoming block was not yet copied mark this PHI as incomplete.
1149 // Once the block will be copied the incoming value will be added.
1150 BasicBlock *BBCopy = BlockMap[IncomingBB];
1151 if (!BBCopy) {
1152 assert(StmtR->contains(IncomingBB) &&
1153 "Bad incoming block for PHI in non-affine region");
1154 IncompletePHINodeMap[IncomingBB].push_back(std::make_pair(PHI, PHICopy));
1155 return;
1156 }
1157
1158 Value *OpCopy = nullptr;
1159 if (StmtR->contains(IncomingBB)) {
1160 assert(RegionMaps.count(BBCopy) &&
1161 "Incoming PHI block did not have a BBMap");
1162 ValueMapT &BBCopyMap = RegionMaps[BBCopy];
1163
1164 Value *Op = PHI->getIncomingValueForBlock(IncomingBB);
1165 OpCopy =
1166 getNewValue(Stmt, Op, BBCopyMap, GlobalMap, LTS, getLoopForInst(PHI));
1167 } else {
1168
1169 if (PHICopy->getBasicBlockIndex(BBCopy) >= 0)
1170 return;
1171
Tobias Grosserb79a67d2015-08-28 08:23:35 +00001172 AllocaInst *PHIOpAddr = getOrCreatePHIAlloca(const_cast<PHINode *>(PHI));
Johannes Doerfertecff11d2015-05-22 23:43:58 +00001173 OpCopy = new LoadInst(PHIOpAddr, PHIOpAddr->getName() + ".reload",
1174 BlockMap[IncomingBB]->getTerminator());
1175 }
1176
1177 assert(OpCopy && "Incoming PHI value was not copied properly");
1178 assert(BBCopy && "Incoming PHI block was not copied properly");
1179 PHICopy->addIncoming(OpCopy, BBCopy);
1180}
1181
1182Value *RegionGenerator::copyPHIInstruction(ScopStmt &Stmt, const PHINode *PHI,
1183 ValueMapT &BBMap,
1184 ValueMapT &GlobalMap,
1185 LoopToScevMapT &LTS) {
1186 unsigned NumIncoming = PHI->getNumIncomingValues();
1187 PHINode *PHICopy =
1188 Builder.CreatePHI(PHI->getType(), NumIncoming, "polly." + PHI->getName());
1189 PHICopy->moveBefore(PHICopy->getParent()->getFirstNonPHI());
1190 BBMap[PHI] = PHICopy;
1191
1192 for (unsigned u = 0; u < NumIncoming; u++)
1193 addOperandToPHI(Stmt, PHI, PHICopy, PHI->getIncomingBlock(u), GlobalMap,
1194 LTS);
1195 return PHICopy;
1196}