13b11a16aSHongbin Zheng //===--- BlockGenerators.cpp - Generate code for statements -----*- C++ -*-===//
23b11a16aSHongbin Zheng //
32946cd70SChandler Carruth // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
42946cd70SChandler Carruth // See https://llvm.org/LICENSE.txt for license information.
52946cd70SChandler Carruth // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
63b11a16aSHongbin Zheng //
73b11a16aSHongbin Zheng //===----------------------------------------------------------------------===//
83b11a16aSHongbin Zheng //
93b11a16aSHongbin Zheng // This file implements the BlockGenerator and VectorBlockGenerator classes,
103b11a16aSHongbin Zheng // which generate sequential code and vectorized code for a polyhedral
113b11a16aSHongbin Zheng // statement, respectively.
123b11a16aSHongbin Zheng //
133b11a16aSHongbin Zheng //===----------------------------------------------------------------------===//
143b11a16aSHongbin Zheng 
158a846610SHongbin Zheng #include "polly/CodeGen/BlockGenerators.h"
16a63b2579SJohannes Doerfert #include "polly/CodeGen/IslExprBuilder.h"
1786bc93a9STobias Grosser #include "polly/CodeGen/RuntimeDebugBuilder.h"
18637bd631STobias Grosser #include "polly/Options.h"
195624d3c9STobias Grosser #include "polly/ScopInfo.h"
20ecfe21b7STobias Grosser #include "polly/Support/ScopHelper.h"
21eedae763SMichael Kruse #include "polly/Support/VirtualInstruction.h"
22e71c6ab5STobias Grosser #include "llvm/Analysis/LoopInfo.h"
23f32d651dSJohannes Doerfert #include "llvm/Analysis/RegionInfo.h"
24e71c6ab5STobias Grosser #include "llvm/Analysis/ScalarEvolution.h"
253b11a16aSHongbin Zheng #include "llvm/Transforms/Utils/BasicBlockUtils.h"
26a998f98bSTobias Grosser #include "llvm/Transforms/Utils/Local.h"
27f32d651dSJohannes Doerfert #include "isl/ast.h"
28f32d651dSJohannes Doerfert #include <deque>
29f32d651dSJohannes Doerfert 
303b11a16aSHongbin Zheng using namespace llvm;
313b11a16aSHongbin Zheng using namespace polly;
323b11a16aSHongbin Zheng 
33878aba49STobias Grosser static cl::opt<bool> Aligned("enable-polly-aligned",
34878aba49STobias Grosser                              cl::desc("Assumed aligned memory accesses."),
35878aba49STobias Grosser                              cl::Hidden, cl::init(false), cl::ZeroOrMore,
36878aba49STobias Grosser                              cl::cat(PollyCategory));
373b11a16aSHongbin Zheng 
381be726a4STobias Grosser bool PollyDebugPrinting;
391be726a4STobias Grosser static cl::opt<bool, true> DebugPrintingX(
4086bc93a9STobias Grosser     "polly-codegen-add-debug-printing",
4186bc93a9STobias Grosser     cl::desc("Add printf calls that show the values loaded/stored."),
421be726a4STobias Grosser     cl::location(PollyDebugPrinting), cl::Hidden, cl::init(false),
431be726a4STobias Grosser     cl::ZeroOrMore, cl::cat(PollyCategory));
4486bc93a9STobias Grosser 
45e819fffeSMichael Kruse static cl::opt<bool> TraceStmts(
46e819fffeSMichael Kruse     "polly-codegen-trace-stmts",
47e819fffeSMichael Kruse     cl::desc("Add printf calls that print the statement being executed"),
48e819fffeSMichael Kruse     cl::Hidden, cl::init(false), cl::ZeroOrMore, cl::cat(PollyCategory));
49e819fffeSMichael Kruse 
50e819fffeSMichael Kruse static cl::opt<bool> TraceScalars(
51e819fffeSMichael Kruse     "polly-codegen-trace-scalars",
52e819fffeSMichael Kruse     cl::desc("Add printf calls that print the values of all scalar values "
53e819fffeSMichael Kruse              "used in a statement. Requires -polly-codegen-trace-stmts."),
54e819fffeSMichael Kruse     cl::Hidden, cl::init(false), cl::ZeroOrMore, cl::cat(PollyCategory));
55e819fffeSMichael Kruse 
56acf80064SEli Friedman BlockGenerator::BlockGenerator(
57acf80064SEli Friedman     PollyIRBuilder &B, LoopInfo &LI, ScalarEvolution &SE, DominatorTree &DT,
58587f1f57STobias Grosser     AllocaMapTy &ScalarMap, EscapeUsersAllocaMapTy &EscapeMap,
59587f1f57STobias Grosser     ValueMapT &GlobalMap, IslExprBuilder *ExprBuilder, BasicBlock *StartBlock)
60ecff11dcSJohannes Doerfert     : Builder(B), LI(LI), SE(SE), ExprBuilder(ExprBuilder), DT(DT),
61587f1f57STobias Grosser       EntryBB(nullptr), ScalarMap(ScalarMap), EscapeMap(EscapeMap),
62587f1f57STobias Grosser       GlobalMap(GlobalMap), StartBlock(StartBlock) {}
63e71c6ab5STobias Grosser 
642f1acac6STobias Grosser Value *BlockGenerator::trySynthesizeNewValue(ScopStmt &Stmt, Value *Old,
6533cb9f94STobias Grosser                                              ValueMapT &BBMap,
6633cb9f94STobias Grosser                                              LoopToScevMapT &LTS,
67bc132607STobias Grosser                                              Loop *L) const {
6842df8d1dSJohannes Doerfert   if (!SE.isSCEVable(Old->getType()))
6942df8d1dSJohannes Doerfert     return nullptr;
7042df8d1dSJohannes Doerfert 
7142df8d1dSJohannes Doerfert   const SCEV *Scev = SE.getSCEVAtScope(Old, L);
7242df8d1dSJohannes Doerfert   if (!Scev)
7342df8d1dSJohannes Doerfert     return nullptr;
7442df8d1dSJohannes Doerfert 
7542df8d1dSJohannes Doerfert   if (isa<SCEVCouldNotCompute>(Scev))
7642df8d1dSJohannes Doerfert     return nullptr;
7742df8d1dSJohannes Doerfert 
7803bcb910SSanjoy Das   const SCEV *NewScev = SCEVLoopAddRecRewriter::rewrite(Scev, LTS, SE);
7909e3697fSJohannes Doerfert   ValueMapT VTV;
80637b23dcSSebastian Pop   VTV.insert(BBMap.begin(), BBMap.end());
81637b23dcSSebastian Pop   VTV.insert(GlobalMap.begin(), GlobalMap.end());
82e69e1141SJohannes Doerfert 
83e69e1141SJohannes Doerfert   Scop &S = *Stmt.getParent();
843f52e354SJohannes Doerfert   const DataLayout &DL = S.getFunction().getParent()->getDataLayout();
85e69e1141SJohannes Doerfert   auto IP = Builder.GetInsertPoint();
86e69e1141SJohannes Doerfert 
87e69e1141SJohannes Doerfert   assert(IP != Builder.GetInsertBlock()->end() &&
8845e7944bSTobias Grosser          "Only instructions can be insert points for SCEVExpander");
8942df8d1dSJohannes Doerfert   Value *Expanded =
90acf80064SEli Friedman       expandCodeFor(S, SE, DL, "polly", NewScev, Old->getType(), &*IP, &VTV,
91acf80064SEli Friedman                     StartBlock->getSinglePredecessor());
92e71c6ab5STobias Grosser 
93e71c6ab5STobias Grosser   BBMap[Old] = Expanded;
94e71c6ab5STobias Grosser   return Expanded;
95e71c6ab5STobias Grosser }
9633cb9f94STobias Grosser 
972f1acac6STobias Grosser Value *BlockGenerator::getNewValue(ScopStmt &Stmt, Value *Old, ValueMapT &BBMap,
982f1acac6STobias Grosser                                    LoopToScevMapT &LTS, Loop *L) const {
9933cb9f94STobias Grosser 
100eedae763SMichael Kruse   auto lookupGlobally = [this](Value *Old) -> Value * {
101eedae763SMichael Kruse     Value *New = GlobalMap.lookup(Old);
102eedae763SMichael Kruse     if (!New)
103eedae763SMichael Kruse       return nullptr;
10428f8ac1dSJohannes Doerfert 
105eedae763SMichael Kruse     // Required by:
106eedae763SMichael Kruse     // * Isl/CodeGen/OpenMP/invariant_base_pointer_preloaded.ll
107eedae763SMichael Kruse     // * Isl/CodeGen/OpenMP/invariant_base_pointer_preloaded_different_bb.ll
108eedae763SMichael Kruse     // * Isl/CodeGen/OpenMP/invariant_base_pointer_preloaded_pass_only_needed.ll
109eedae763SMichael Kruse     // * Isl/CodeGen/OpenMP/invariant_base_pointers_preloaded.ll
110eedae763SMichael Kruse     // * Isl/CodeGen/OpenMP/loop-body-references-outer-values-3.ll
111eedae763SMichael Kruse     // * Isl/CodeGen/OpenMP/single_loop_with_loop_invariant_baseptr.ll
112eedae763SMichael Kruse     // GlobalMap should be a mapping from (value in original SCoP) to (copied
113eedae763SMichael Kruse     // value in generated SCoP), without intermediate mappings, which might
114eedae763SMichael Kruse     // easily require transitiveness as well.
11533cb9f94STobias Grosser     if (Value *NewRemapped = GlobalMap.lookup(New))
11633cb9f94STobias Grosser       New = NewRemapped;
117eedae763SMichael Kruse 
118eedae763SMichael Kruse     // No test case for this code.
11933cb9f94STobias Grosser     if (Old->getType()->getScalarSizeInBits() <
12033cb9f94STobias Grosser         New->getType()->getScalarSizeInBits())
12133cb9f94STobias Grosser       New = Builder.CreateTruncOrBitCast(New, Old->getType());
12233cb9f94STobias Grosser 
12333cb9f94STobias Grosser     return New;
124eedae763SMichael Kruse   };
125eedae763SMichael Kruse 
126eedae763SMichael Kruse   Value *New = nullptr;
127eedae763SMichael Kruse   auto VUse = VirtualUse::create(&Stmt, L, Old, true);
128eedae763SMichael Kruse   switch (VUse.getKind()) {
129eedae763SMichael Kruse   case VirtualUse::Block:
130eedae763SMichael Kruse     // BasicBlock are constants, but the BlockGenerator copies them.
131eedae763SMichael Kruse     New = BBMap.lookup(Old);
132eedae763SMichael Kruse     break;
133eedae763SMichael Kruse 
134eedae763SMichael Kruse   case VirtualUse::Constant:
135eedae763SMichael Kruse     // Used by:
136eedae763SMichael Kruse     // * Isl/CodeGen/OpenMP/reference-argument-from-non-affine-region.ll
137eedae763SMichael Kruse     // Constants should not be redefined. In this case, the GlobalMap just
138eedae763SMichael Kruse     // contains a mapping to the same constant, which is unnecessary, but
139eedae763SMichael Kruse     // harmless.
140eedae763SMichael Kruse     if ((New = lookupGlobally(Old)))
141eedae763SMichael Kruse       break;
142eedae763SMichael Kruse 
143eedae763SMichael Kruse     assert(!BBMap.count(Old));
144eedae763SMichael Kruse     New = Old;
145eedae763SMichael Kruse     break;
146eedae763SMichael Kruse 
147eedae763SMichael Kruse   case VirtualUse::ReadOnly:
148eedae763SMichael Kruse     assert(!GlobalMap.count(Old));
149eedae763SMichael Kruse 
150eedae763SMichael Kruse     // Required for:
151eedae763SMichael Kruse     // * Isl/CodeGen/MemAccess/create_arrays.ll
152eedae763SMichael Kruse     // * Isl/CodeGen/read-only-scalars.ll
153eedae763SMichael Kruse     // * ScheduleOptimizer/pattern-matching-based-opts_10.ll
154eedae763SMichael Kruse     // For some reason these reload a read-only value. The reloaded value ends
155eedae763SMichael Kruse     // up in BBMap, buts its value should be identical.
156eedae763SMichael Kruse     //
157eedae763SMichael Kruse     // Required for:
158eedae763SMichael Kruse     // * Isl/CodeGen/OpenMP/single_loop_with_param.ll
159eedae763SMichael Kruse     // The parallel subfunctions need to reference the read-only value from the
160eedae763SMichael Kruse     // parent function, this is done by reloading them locally.
161eedae763SMichael Kruse     if ((New = BBMap.lookup(Old)))
162eedae763SMichael Kruse       break;
163eedae763SMichael Kruse 
164eedae763SMichael Kruse     New = Old;
165eedae763SMichael Kruse     break;
166eedae763SMichael Kruse 
167eedae763SMichael Kruse   case VirtualUse::Synthesizable:
168eedae763SMichael Kruse     // Used by:
169eedae763SMichael Kruse     // * Isl/CodeGen/OpenMP/loop-body-references-outer-values-3.ll
170eedae763SMichael Kruse     // * Isl/CodeGen/OpenMP/recomputed-srem.ll
171eedae763SMichael Kruse     // * Isl/CodeGen/OpenMP/reference-other-bb.ll
172eedae763SMichael Kruse     // * Isl/CodeGen/OpenMP/two-parallel-loops-reference-outer-indvar.ll
173eedae763SMichael Kruse     // For some reason synthesizable values end up in GlobalMap. Their values
174eedae763SMichael Kruse     // are the same as trySynthesizeNewValue would return. The legacy
175eedae763SMichael Kruse     // implementation prioritized GlobalMap, so this is what we do here as well.
176eedae763SMichael Kruse     // Ideally, synthesizable values should not end up in GlobalMap.
177eedae763SMichael Kruse     if ((New = lookupGlobally(Old)))
178eedae763SMichael Kruse       break;
179eedae763SMichael Kruse 
180eedae763SMichael Kruse     // Required for:
181eedae763SMichael Kruse     // * Isl/CodeGen/RuntimeDebugBuilder/combine_different_values.ll
182eedae763SMichael Kruse     // * Isl/CodeGen/getNumberOfIterations.ll
183eedae763SMichael Kruse     // * Isl/CodeGen/non_affine_float_compare.ll
184eedae763SMichael Kruse     // * ScheduleOptimizer/pattern-matching-based-opts_10.ll
185eedae763SMichael Kruse     // Ideally, synthesizable values are synthesized by trySynthesizeNewValue,
186eedae763SMichael Kruse     // not precomputed (SCEVExpander has its own caching mechanism).
187eedae763SMichael Kruse     // These tests fail without this, but I think trySynthesizeNewValue would
188eedae763SMichael Kruse     // just re-synthesize the same instructions.
189eedae763SMichael Kruse     if ((New = BBMap.lookup(Old)))
190eedae763SMichael Kruse       break;
191eedae763SMichael Kruse 
192eedae763SMichael Kruse     New = trySynthesizeNewValue(Stmt, Old, BBMap, LTS, L);
193eedae763SMichael Kruse     break;
194eedae763SMichael Kruse 
195eedae763SMichael Kruse   case VirtualUse::Hoisted:
196eedae763SMichael Kruse     // TODO: Hoisted invariant loads should be found in GlobalMap only, but not
197eedae763SMichael Kruse     // redefined locally (which will be ignored anyway). That is, the following
198eedae763SMichael Kruse     // assertion should apply: assert(!BBMap.count(Old))
199eedae763SMichael Kruse 
200eedae763SMichael Kruse     New = lookupGlobally(Old);
201eedae763SMichael Kruse     break;
202eedae763SMichael Kruse 
203eedae763SMichael Kruse   case VirtualUse::Intra:
204eedae763SMichael Kruse   case VirtualUse::Inter:
205eedae763SMichael Kruse     assert(!GlobalMap.count(Old) &&
206eedae763SMichael Kruse            "Intra and inter-stmt values are never global");
207eedae763SMichael Kruse     New = BBMap.lookup(Old);
208eedae763SMichael Kruse     break;
20933cb9f94STobias Grosser   }
210eedae763SMichael Kruse   assert(New && "Unexpected scalar dependence in region!");
21133cb9f94STobias Grosser   return New;
2123b11a16aSHongbin Zheng }
2133b11a16aSHongbin Zheng 
2142f1acac6STobias Grosser void BlockGenerator::copyInstScalar(ScopStmt &Stmt, Instruction *Inst,
215bc132607STobias Grosser                                     ValueMapT &BBMap, LoopToScevMapT &LTS) {
216030237d0STobias Grosser   // We do not generate debug intrinsics as we did not investigate how to
217030237d0STobias Grosser   // copy them correctly. At the current state, they just crash the code
218030237d0STobias Grosser   // generation as the meta-data operands are not correctly copied.
219030237d0STobias Grosser   if (isa<DbgInfoIntrinsic>(Inst))
220030237d0STobias Grosser     return;
221030237d0STobias Grosser 
2223b11a16aSHongbin Zheng   Instruction *NewInst = Inst->clone();
2233b11a16aSHongbin Zheng 
2243b11a16aSHongbin Zheng   // Replace old operands with the new ones.
22591f5b262STobias Grosser   for (Value *OldOperand : Inst->operands()) {
226bc132607STobias Grosser     Value *NewOperand =
2272c3ffc04SJohannes Doerfert         getNewValue(Stmt, OldOperand, BBMap, LTS, getLoopForStmt(Stmt));
2283b11a16aSHongbin Zheng 
2293b11a16aSHongbin Zheng     if (!NewOperand) {
230c14582f2STobias Grosser       assert(!isa<StoreInst>(NewInst) &&
231c14582f2STobias Grosser              "Store instructions are always needed!");
23296ab8726SReid Kleckner       NewInst->deleteValue();
2333b11a16aSHongbin Zheng       return;
2343b11a16aSHongbin Zheng     }
2353b11a16aSHongbin Zheng 
2363b11a16aSHongbin Zheng     NewInst->replaceUsesOfWith(OldOperand, NewOperand);
2373b11a16aSHongbin Zheng   }
2383b11a16aSHongbin Zheng 
2393b11a16aSHongbin Zheng   Builder.Insert(NewInst);
2403b11a16aSHongbin Zheng   BBMap[Inst] = NewInst;
2413b11a16aSHongbin Zheng 
242188053afSSingapuram Sanjay Srivallabh   // When copying the instruction onto the Module meant for the GPU,
243188053afSSingapuram Sanjay Srivallabh   // debug metadata attached to an instruction causes all related
244188053afSSingapuram Sanjay Srivallabh   // metadata to be pulled into the Module. This includes the DICompileUnit,
245188053afSSingapuram Sanjay Srivallabh   // which will not be listed in llvm.dbg.cu of the Module since the Module
246188053afSSingapuram Sanjay Srivallabh   // doesn't contain one. This fails the verification of the Module and the
247188053afSSingapuram Sanjay Srivallabh   // subsequent generation of the ASM string.
248188053afSSingapuram Sanjay Srivallabh   if (NewInst->getModule() != Inst->getModule())
249188053afSSingapuram Sanjay Srivallabh     NewInst->setDebugLoc(llvm::DebugLoc());
250188053afSSingapuram Sanjay Srivallabh 
2513b11a16aSHongbin Zheng   if (!NewInst->getType()->isVoidTy())
2523b11a16aSHongbin Zheng     NewInst->setName("p_" + Inst->getName());
2533b11a16aSHongbin Zheng }
2543b11a16aSHongbin Zheng 
25570131d34SMichael Kruse Value *
25670131d34SMichael Kruse BlockGenerator::generateLocationAccessed(ScopStmt &Stmt, MemAccInst Inst,
25770131d34SMichael Kruse                                          ValueMapT &BBMap, LoopToScevMapT &LTS,
25870131d34SMichael Kruse                                          isl_id_to_ast_expr *NewAccesses) {
25909214772STobias Grosser   const MemoryAccess &MA = Stmt.getArrayAccessFor(Inst);
2602fa35194SMichael Kruse   return generateLocationAccessed(
2612fa35194SMichael Kruse       Stmt, getLoopForStmt(Stmt),
2622fa35194SMichael Kruse       Inst.isNull() ? nullptr : Inst.getPointerOperand(), BBMap, LTS,
263fe46c3ffSTobias Grosser       NewAccesses, MA.getId().release(), MA.getAccessValue()->getType());
2642fa35194SMichael Kruse }
2653b11a16aSHongbin Zheng 
2662fa35194SMichael Kruse Value *BlockGenerator::generateLocationAccessed(
2672fa35194SMichael Kruse     ScopStmt &Stmt, Loop *L, Value *Pointer, ValueMapT &BBMap,
2682fa35194SMichael Kruse     LoopToScevMapT &LTS, isl_id_to_ast_expr *NewAccesses, __isl_take isl_id *Id,
2692fa35194SMichael Kruse     Type *ExpectedType) {
2702fa35194SMichael Kruse   isl_ast_expr *AccessExpr = isl_id_to_ast_expr_get(NewAccesses, Id);
2713b11a16aSHongbin Zheng 
2722d1ed0bfSTobias Grosser   if (AccessExpr) {
2732d1ed0bfSTobias Grosser     AccessExpr = isl_ast_expr_address_of(AccessExpr);
27498b3ee50STobias Grosser     auto Address = ExprBuilder->create(AccessExpr);
27598b3ee50STobias Grosser 
27698b3ee50STobias Grosser     // Cast the address of this memory access to a pointer type that has the
27798b3ee50STobias Grosser     // same element type as the original access, but uses the address space of
27898b3ee50STobias Grosser     // the newly generated pointer.
2792fa35194SMichael Kruse     auto OldPtrTy = ExpectedType->getPointerTo();
28098b3ee50STobias Grosser     auto NewPtrTy = Address->getType();
28198b3ee50STobias Grosser     OldPtrTy = PointerType::get(OldPtrTy->getElementType(),
28298b3ee50STobias Grosser                                 NewPtrTy->getPointerAddressSpace());
28398b3ee50STobias Grosser 
284d840fc72STobias Grosser     if (OldPtrTy != NewPtrTy)
28598b3ee50STobias Grosser       Address = Builder.CreateBitOrPointerCast(Address, OldPtrTy);
28698b3ee50STobias Grosser     return Address;
2872d1ed0bfSTobias Grosser   }
2882fa35194SMichael Kruse   assert(
2892fa35194SMichael Kruse       Pointer &&
2902fa35194SMichael Kruse       "If expression was not generated, must use the original pointer value");
2912fa35194SMichael Kruse   return getNewValue(Stmt, Pointer, BBMap, LTS, L);
2922fa35194SMichael Kruse }
2932d1ed0bfSTobias Grosser 
2942fa35194SMichael Kruse Value *
2952fa35194SMichael Kruse BlockGenerator::getImplicitAddress(MemoryAccess &Access, Loop *L,
2962fa35194SMichael Kruse                                    LoopToScevMapT &LTS, ValueMapT &BBMap,
2972fa35194SMichael Kruse                                    __isl_keep isl_id_to_ast_expr *NewAccesses) {
2982fa35194SMichael Kruse   if (Access.isLatestArrayKind())
2992fa35194SMichael Kruse     return generateLocationAccessed(*Access.getStatement(), L, nullptr, BBMap,
300fe46c3ffSTobias Grosser                                     LTS, NewAccesses, Access.getId().release(),
3012fa35194SMichael Kruse                                     Access.getAccessValue()->getType());
3022fa35194SMichael Kruse 
303587f1f57STobias Grosser   return getOrCreateAlloca(Access);
3043b11a16aSHongbin Zheng }
3053b11a16aSHongbin Zheng 
3062c3ffc04SJohannes Doerfert Loop *BlockGenerator::getLoopForStmt(const ScopStmt &Stmt) const {
307375cb5feSMichael Kruse   auto *StmtBB = Stmt.getEntryBlock();
3082c3ffc04SJohannes Doerfert   return LI.getLoopFor(StmtBB);
309369430ffSTobias Grosser }
310369430ffSTobias Grosser 
311888ab551SMichael Kruse Value *BlockGenerator::generateArrayLoad(ScopStmt &Stmt, LoadInst *Load,
312bc132607STobias Grosser                                          ValueMapT &BBMap, LoopToScevMapT &LTS,
3132d1ed0bfSTobias Grosser                                          isl_id_to_ast_expr *NewAccesses) {
314c1db67e2SJohannes Doerfert   if (Value *PreloadLoad = GlobalMap.lookup(Load))
315c1db67e2SJohannes Doerfert     return PreloadLoad;
316c1db67e2SJohannes Doerfert 
317bc132607STobias Grosser   Value *NewPointer =
31870131d34SMichael Kruse       generateLocationAccessed(Stmt, Load, BBMap, LTS, NewAccesses);
31907c9d532SGuillaume Chatelet   Value *ScalarLoad = Builder.CreateAlignedLoad(NewPointer, Load->getAlign(),
32007c9d532SGuillaume Chatelet                                                 Load->getName() + "_p_scalar_");
32186bc93a9STobias Grosser 
3221be726a4STobias Grosser   if (PollyDebugPrinting)
32386bc93a9STobias Grosser     RuntimeDebugBuilder::createCPUPrinter(Builder, "Load from ", NewPointer,
32486bc93a9STobias Grosser                                           ": ", ScalarLoad, "\n");
32586bc93a9STobias Grosser 
3263b11a16aSHongbin Zheng   return ScalarLoad;
3273b11a16aSHongbin Zheng }
3283b11a16aSHongbin Zheng 
329888ab551SMichael Kruse void BlockGenerator::generateArrayStore(ScopStmt &Stmt, StoreInst *Store,
330bc132607STobias Grosser                                         ValueMapT &BBMap, LoopToScevMapT &LTS,
3312d1ed0bfSTobias Grosser                                         isl_id_to_ast_expr *NewAccesses) {
332706f79abSMichael Kruse   MemoryAccess &MA = Stmt.getArrayAccessFor(Store);
3331515f6b9STobias Grosser   isl::set AccDom = MA.getAccessRelation().domain();
334fe46c3ffSTobias Grosser   std::string Subject = MA.getId().get_name();
335706f79abSMichael Kruse 
336fe46c3ffSTobias Grosser   generateConditionalExecution(Stmt, AccDom, Subject.c_str(), [&, this]() {
337bc132607STobias Grosser     Value *NewPointer =
33870131d34SMichael Kruse         generateLocationAccessed(Stmt, Store, BBMap, LTS, NewAccesses);
339706f79abSMichael Kruse     Value *ValueOperand = getNewValue(Stmt, Store->getValueOperand(), BBMap,
340706f79abSMichael Kruse                                       LTS, getLoopForStmt(Stmt));
3413b11a16aSHongbin Zheng 
3421be726a4STobias Grosser     if (PollyDebugPrinting)
34386bc93a9STobias Grosser       RuntimeDebugBuilder::createCPUPrinter(Builder, "Store to  ", NewPointer,
34486bc93a9STobias Grosser                                             ": ", ValueOperand, "\n");
34586bc93a9STobias Grosser 
34659f95222SGuillaume Chatelet     Builder.CreateAlignedStore(ValueOperand, NewPointer, Store->getAlign());
347706f79abSMichael Kruse   });
3483b11a16aSHongbin Zheng }
3493b11a16aSHongbin Zheng 
3505dced269SJohannes Doerfert bool BlockGenerator::canSyntheziseInStmt(ScopStmt &Stmt, Instruction *Inst) {
3512c3ffc04SJohannes Doerfert   Loop *L = getLoopForStmt(Stmt);
3525dced269SJohannes Doerfert   return (Stmt.isBlockStmt() || !Stmt.getRegion()->contains(L)) &&
35311c5e079SMichael Kruse          canSynthesize(Inst, *Stmt.getParent(), &SE, L);
3545dced269SJohannes Doerfert }
3555dced269SJohannes Doerfert 
3562f1acac6STobias Grosser void BlockGenerator::copyInstruction(ScopStmt &Stmt, Instruction *Inst,
357bc132607STobias Grosser                                      ValueMapT &BBMap, LoopToScevMapT &LTS,
3582d1ed0bfSTobias Grosser                                      isl_id_to_ast_expr *NewAccesses) {
3593b11a16aSHongbin Zheng   // Terminator instructions control the control flow. They are explicitly
3603b11a16aSHongbin Zheng   // expressed in the clast and do not need to be copied.
3613b11a16aSHongbin Zheng   if (Inst->isTerminator())
3623b11a16aSHongbin Zheng     return;
3633b11a16aSHongbin Zheng 
364aff56c8aSTobias Grosser   // Synthesizable statements will be generated on-demand.
3655dced269SJohannes Doerfert   if (canSyntheziseInStmt(Stmt, Inst))
366e71c6ab5STobias Grosser     return;
367e71c6ab5STobias Grosser 
3689646e3feSTobias Grosser   if (auto *Load = dyn_cast<LoadInst>(Inst)) {
369888ab551SMichael Kruse     Value *NewLoad = generateArrayLoad(Stmt, Load, BBMap, LTS, NewAccesses);
3703d94fedfSSebastian Pop     // Compute NewLoad before its insertion in BBMap to make the insertion
3713d94fedfSSebastian Pop     // deterministic.
372753d43f9SSebastian Pop     BBMap[Load] = NewLoad;
3733b11a16aSHongbin Zheng     return;
3743b11a16aSHongbin Zheng   }
3753b11a16aSHongbin Zheng 
3769646e3feSTobias Grosser   if (auto *Store = dyn_cast<StoreInst>(Inst)) {
3770446d81eSMichael Kruse     // Identified as redundant by -polly-simplify.
3780446d81eSMichael Kruse     if (!Stmt.getArrayAccessOrNULLFor(Store))
3790446d81eSMichael Kruse       return;
3800446d81eSMichael Kruse 
381888ab551SMichael Kruse     generateArrayStore(Stmt, Store, BBMap, LTS, NewAccesses);
3823b11a16aSHongbin Zheng     return;
3833b11a16aSHongbin Zheng   }
3843b11a16aSHongbin Zheng 
3859646e3feSTobias Grosser   if (auto *PHI = dyn_cast<PHINode>(Inst)) {
386bc132607STobias Grosser     copyPHIInstruction(Stmt, PHI, BBMap, LTS);
387ecff11dcSJohannes Doerfert     return;
388ecff11dcSJohannes Doerfert   }
389ecff11dcSJohannes Doerfert 
3903f500fa2SJohannes Doerfert   // Skip some special intrinsics for which we do not adjust the semantics to
3913f500fa2SJohannes Doerfert   // the new schedule. All others are handled like every other instruction.
3929c0ffe3aSTobias Grosser   if (isIgnoredIntrinsic(Inst))
3933f500fa2SJohannes Doerfert     return;
3943f500fa2SJohannes Doerfert 
395bc132607STobias Grosser   copyInstScalar(Stmt, Inst, BBMap, LTS);
3963b11a16aSHongbin Zheng }
3973b11a16aSHongbin Zheng 
3989d12d8adSTobias Grosser void BlockGenerator::removeDeadInstructions(BasicBlock *BB, ValueMapT &BBMap) {
399c59b3ce0STobias Grosser   auto NewBB = Builder.GetInsertBlock();
400c59b3ce0STobias Grosser   for (auto I = NewBB->rbegin(); I != NewBB->rend(); I++) {
401c59b3ce0STobias Grosser     Instruction *NewInst = &*I;
4029d12d8adSTobias Grosser 
4039d12d8adSTobias Grosser     if (!isInstructionTriviallyDead(NewInst))
4049d12d8adSTobias Grosser       continue;
4059d12d8adSTobias Grosser 
406c59b3ce0STobias Grosser     for (auto Pair : BBMap)
407c59b3ce0STobias Grosser       if (Pair.second == NewInst) {
408c59b3ce0STobias Grosser         BBMap.erase(Pair.first);
409c59b3ce0STobias Grosser       }
410c59b3ce0STobias Grosser 
4119d12d8adSTobias Grosser     NewInst->eraseFromParent();
412c59b3ce0STobias Grosser     I = NewBB->rbegin();
4139d12d8adSTobias Grosser   }
4149d12d8adSTobias Grosser }
4159d12d8adSTobias Grosser 
416bc132607STobias Grosser void BlockGenerator::copyStmt(ScopStmt &Stmt, LoopToScevMapT &LTS,
4172d1ed0bfSTobias Grosser                               isl_id_to_ast_expr *NewAccesses) {
418275a1756SJohannes Doerfert   assert(Stmt.isBlockStmt() &&
419275a1756SJohannes Doerfert          "Only block statements can be copied by the block generator");
420275a1756SJohannes Doerfert 
421275a1756SJohannes Doerfert   ValueMapT BBMap;
422275a1756SJohannes Doerfert 
423be9c9117SJohannes Doerfert   BasicBlock *BB = Stmt.getBasicBlock();
424bc132607STobias Grosser   copyBB(Stmt, BB, BBMap, LTS, NewAccesses);
4259d12d8adSTobias Grosser   removeDeadInstructions(BB, BBMap);
426275a1756SJohannes Doerfert }
427275a1756SJohannes Doerfert 
428514f6efaSJohannes Doerfert BasicBlock *BlockGenerator::splitBB(BasicBlock *BB) {
429b8f58b53SDuncan P. N. Exon Smith   BasicBlock *CopyBB = SplitBlock(Builder.GetInsertBlock(),
430b8f58b53SDuncan P. N. Exon Smith                                   &*Builder.GetInsertPoint(), &DT, &LI);
4313b11a16aSHongbin Zheng   CopyBB->setName("polly.stmt." + BB->getName());
432514f6efaSJohannes Doerfert   return CopyBB;
433514f6efaSJohannes Doerfert }
4343b11a16aSHongbin Zheng 
435514f6efaSJohannes Doerfert BasicBlock *BlockGenerator::copyBB(ScopStmt &Stmt, BasicBlock *BB,
436bc132607STobias Grosser                                    ValueMapT &BBMap, LoopToScevMapT &LTS,
4372d1ed0bfSTobias Grosser                                    isl_id_to_ast_expr *NewAccesses) {
438514f6efaSJohannes Doerfert   BasicBlock *CopyBB = splitBB(BB);
439b8f58b53SDuncan P. N. Exon Smith   Builder.SetInsertPoint(&CopyBB->front());
4402fa35194SMichael Kruse   generateScalarLoads(Stmt, LTS, BBMap, NewAccesses);
441e819fffeSMichael Kruse   generateBeginStmtTrace(Stmt, LTS, BBMap);
442225f0d1eSMichael Kruse 
443bc132607STobias Grosser   copyBB(Stmt, BB, CopyBB, BBMap, LTS, NewAccesses);
444225f0d1eSMichael Kruse 
445225f0d1eSMichael Kruse   // After a basic block was copied store all scalars that escape this block in
446225f0d1eSMichael Kruse   // their alloca.
4472fa35194SMichael Kruse   generateScalarStores(Stmt, LTS, BBMap, NewAccesses);
448514f6efaSJohannes Doerfert   return CopyBB;
449514f6efaSJohannes Doerfert }
450514f6efaSJohannes Doerfert 
451514f6efaSJohannes Doerfert void BlockGenerator::copyBB(ScopStmt &Stmt, BasicBlock *BB, BasicBlock *CopyBB,
452bc132607STobias Grosser                             ValueMapT &BBMap, LoopToScevMapT &LTS,
4532d1ed0bfSTobias Grosser                             isl_id_to_ast_expr *NewAccesses) {
454ecff11dcSJohannes Doerfert   EntryBB = &CopyBB->getParent()->getEntryBlock();
455ecff11dcSJohannes Doerfert 
456c43d0360STobias Grosser   // Block statements and the entry blocks of region statement are code
457c43d0360STobias Grosser   // generated from instruction lists. This allow us to optimize the
458c43d0360STobias Grosser   // instructions that belong to a certain scop statement. As the code
459c43d0360STobias Grosser   // structure of region statements might be arbitrary complex, optimizing the
460c43d0360STobias Grosser   // instruction list is not yet supported.
461c43d0360STobias Grosser   if (Stmt.isBlockStmt() || (Stmt.isRegionStmt() && Stmt.getEntryBlock() == BB))
4623bb48299SMichael Kruse     for (Instruction *Inst : Stmt.getInstructions())
4633bb48299SMichael Kruse       copyInstruction(Stmt, Inst, BBMap, LTS, NewAccesses);
4643bb48299SMichael Kruse   else
46591f5b262STobias Grosser     for (Instruction &Inst : *BB)
466bc132607STobias Grosser       copyInstruction(Stmt, &Inst, BBMap, LTS, NewAccesses);
467ecff11dcSJohannes Doerfert }
468ecff11dcSJohannes Doerfert 
469800e17a7SJohannes Doerfert Value *BlockGenerator::getOrCreateAlloca(const MemoryAccess &Access) {
470587f1f57STobias Grosser   assert(!Access.isLatestArrayKind() && "Trying to get alloca for array kind");
4713216f854STobias Grosser 
472587f1f57STobias Grosser   return getOrCreateAlloca(Access.getLatestScopArrayInfo());
473a4f0988dSTobias Grosser }
474a4f0988dSTobias Grosser 
475a4f0988dSTobias Grosser Value *BlockGenerator::getOrCreateAlloca(const ScopArrayInfo *Array) {
4763216f854STobias Grosser   assert(!Array->isArrayKind() && "Trying to get alloca for array kind");
4773216f854STobias Grosser 
478587f1f57STobias Grosser   auto &Addr = ScalarMap[Array];
479587f1f57STobias Grosser 
480587f1f57STobias Grosser   if (Addr) {
481587f1f57STobias Grosser     // Allow allocas to be (temporarily) redirected once by adding a new
482a6d48f59SMichael Kruse     // old-alloca-addr to new-addr mapping to GlobalMap. This functionality
483587f1f57STobias Grosser     // is used for example by the OpenMP code generation where a first use
484587f1f57STobias Grosser     // of a scalar while still in the host code allocates a normal alloca with
485587f1f57STobias Grosser     // getOrCreateAlloca. When the values of this scalar are accessed during
486587f1f57STobias Grosser     // the generation of the parallel subfunction, these values are copied over
487587f1f57STobias Grosser     // to the parallel subfunction and each request for a scalar alloca slot
488a6d48f59SMichael Kruse     // must be forwarded to the temporary in-subfunction slot. This mapping is
489587f1f57STobias Grosser     // removed when the subfunction has been generated and again normal host
490682c5114STobias Grosser     // code is generated. Due to the following reasons it is not possible to
491682c5114STobias Grosser     // perform the GlobalMap lookup right after creating the alloca below, but
492682c5114STobias Grosser     // instead we need to check GlobalMap at each call to getOrCreateAlloca:
493682c5114STobias Grosser     //
494682c5114STobias Grosser     //   1) GlobalMap may be changed multiple times (for each parallel loop),
495682c5114STobias Grosser     //   2) The temporary mapping is commonly only known after the initial
496682c5114STobias Grosser     //      alloca has already been generated, and
497682c5114STobias Grosser     //   3) The original alloca value must be restored after leaving the
498682c5114STobias Grosser     //      sub-function.
499587f1f57STobias Grosser     if (Value *NewAddr = GlobalMap.lookup(&*Addr))
500587f1f57STobias Grosser       return NewAddr;
501587f1f57STobias Grosser     return Addr;
502587f1f57STobias Grosser   }
503587f1f57STobias Grosser 
504587f1f57STobias Grosser   Type *Ty = Array->getElementType();
505587f1f57STobias Grosser   Value *ScalarBase = Array->getBasePtr();
506587f1f57STobias Grosser   std::string NameExt;
507a535dff4STobias Grosser   if (Array->isPHIKind())
508587f1f57STobias Grosser     NameExt = ".phiops";
509f8d55f7eSTobias Grosser   else
510587f1f57STobias Grosser     NameExt = ".s2a";
511587f1f57STobias Grosser 
5127b5a4dfdSTobias Grosser   const DataLayout &DL = Builder.GetInsertBlock()->getModule()->getDataLayout();
513b3e30c32SMatt Arsenault 
514*4f04db4bSEli Friedman   Addr =
515*4f04db4bSEli Friedman       new AllocaInst(Ty, DL.getAllocaAddrSpace(), nullptr,
516*4f04db4bSEli Friedman                      DL.getPrefTypeAlign(Ty), ScalarBase->getName() + NameExt);
517587f1f57STobias Grosser   EntryBB = &Builder.GetInsertBlock()->getParent()->getEntryBlock();
518587f1f57STobias Grosser   Addr->insertBefore(&*EntryBB->getFirstInsertionPt());
519587f1f57STobias Grosser 
520587f1f57STobias Grosser   return Addr;
521f8d55f7eSTobias Grosser }
522f8d55f7eSTobias Grosser 
523587f1f57STobias Grosser void BlockGenerator::handleOutsideUsers(const Scop &S, ScopArrayInfo *Array) {
524587f1f57STobias Grosser   Instruction *Inst = cast<Instruction>(Array->getBasePtr());
525b79a67dfSTobias Grosser 
5262985400aSTobias Grosser   // If there are escape users we get the alloca for this instruction and put it
5272985400aSTobias Grosser   // in the EscapeMap for later finalization. Lastly, if the instruction was
5282985400aSTobias Grosser   // copied multiple times we already did this and can exit.
529acb6ade7SMichael Kruse   if (EscapeMap.count(Inst))
530acb6ade7SMichael Kruse     return;
531e69e1141SJohannes Doerfert 
532ecff11dcSJohannes Doerfert   EscapeUserVectorTy EscapeUsers;
533ecff11dcSJohannes Doerfert   for (User *U : Inst->users()) {
534ecff11dcSJohannes Doerfert 
535ecff11dcSJohannes Doerfert     // Non-instruction user will never escape.
536ecff11dcSJohannes Doerfert     Instruction *UI = dyn_cast<Instruction>(U);
537ecff11dcSJohannes Doerfert     if (!UI)
538ecff11dcSJohannes Doerfert       continue;
539ecff11dcSJohannes Doerfert 
540952b5304SJohannes Doerfert     if (S.contains(UI))
541ecff11dcSJohannes Doerfert       continue;
542ecff11dcSJohannes Doerfert 
543ecff11dcSJohannes Doerfert     EscapeUsers.push_back(UI);
544ecff11dcSJohannes Doerfert   }
545ecff11dcSJohannes Doerfert 
546ecff11dcSJohannes Doerfert   // Exit if no escape uses were found.
547ecff11dcSJohannes Doerfert   if (EscapeUsers.empty())
548ecff11dcSJohannes Doerfert     return;
549ecff11dcSJohannes Doerfert 
550ecff11dcSJohannes Doerfert   // Get or create an escape alloca for this instruction.
551587f1f57STobias Grosser   auto *ScalarAddr = getOrCreateAlloca(Array);
552ecff11dcSJohannes Doerfert 
553ecff11dcSJohannes Doerfert   // Remember that this instruction has escape uses and the escape alloca.
554ecff11dcSJohannes Doerfert   EscapeMap[Inst] = std::make_pair(ScalarAddr, std::move(EscapeUsers));
555ecff11dcSJohannes Doerfert }
556ecff11dcSJohannes Doerfert 
5572fa35194SMichael Kruse void BlockGenerator::generateScalarLoads(
5582fa35194SMichael Kruse     ScopStmt &Stmt, LoopToScevMapT &LTS, ValueMapT &BBMap,
5592fa35194SMichael Kruse     __isl_keep isl_id_to_ast_expr *NewAccesses) {
560225f0d1eSMichael Kruse   for (MemoryAccess *MA : Stmt) {
5612fa35194SMichael Kruse     if (MA->isOriginalArrayKind() || MA->isWrite())
562ecff11dcSJohannes Doerfert       continue;
563ecff11dcSJohannes Doerfert 
56477394f13SMichael Kruse #ifndef NDEBUG
5654485ae08SMichael Kruse     auto StmtDom =
5664485ae08SMichael Kruse         Stmt.getDomain().intersect_params(Stmt.getParent()->getContext());
567388730c9SMichael Kruse     auto AccDom = MA->getAccessRelation().domain();
568388730c9SMichael Kruse     assert(!StmtDom.is_subset(AccDom).is_false() &&
56977394f13SMichael Kruse            "Scalar must be loaded in all statement instances");
57077394f13SMichael Kruse #endif
57177394f13SMichael Kruse 
5722fa35194SMichael Kruse     auto *Address =
5732fa35194SMichael Kruse         getImplicitAddress(*MA, getLoopForStmt(Stmt), LTS, BBMap, NewAccesses);
574eac9726eSMichael Kruse     assert((!isa<Instruction>(Address) ||
575eac9726eSMichael Kruse             DT.dominates(cast<Instruction>(Address)->getParent(),
576eac9726eSMichael Kruse                          Builder.GetInsertBlock())) &&
577eac9726eSMichael Kruse            "Domination violation");
578583be06fSTobias Grosser     BBMap[MA->getAccessValue()] =
5792fc50df9STobias Grosser         Builder.CreateLoad(Address, Address->getName() + ".reload");
580ecff11dcSJohannes Doerfert   }
581ecff11dcSJohannes Doerfert }
582ecff11dcSJohannes Doerfert 
583706f79abSMichael Kruse Value *BlockGenerator::buildContainsCondition(ScopStmt &Stmt,
584706f79abSMichael Kruse                                               const isl::set &Subdomain) {
585132860afSTobias Grosser   isl::ast_build AstBuild = Stmt.getAstBuild();
586dcf8d696STobias Grosser   isl::set Domain = Stmt.getDomain();
587706f79abSMichael Kruse 
5886b6ac900STobias Grosser   isl::union_map USchedule = AstBuild.get_schedule();
5896b6ac900STobias Grosser   USchedule = USchedule.intersect_domain(Domain);
590706f79abSMichael Kruse 
5916b6ac900STobias Grosser   assert(!USchedule.is_empty());
5926b6ac900STobias Grosser   isl::map Schedule = isl::map::from_union_map(USchedule);
593706f79abSMichael Kruse 
5946b6ac900STobias Grosser   isl::set ScheduledDomain = Schedule.range();
5956b6ac900STobias Grosser   isl::set ScheduledSet = Subdomain.apply(Schedule);
596706f79abSMichael Kruse 
5976b6ac900STobias Grosser   isl::ast_build RestrictedBuild = AstBuild.restrict(ScheduledDomain);
5986b6ac900STobias Grosser 
5996b6ac900STobias Grosser   isl::ast_expr IsInSet = RestrictedBuild.expr_from(ScheduledSet);
600706f79abSMichael Kruse   Value *IsInSetExpr = ExprBuilder->create(IsInSet.copy());
601706f79abSMichael Kruse   IsInSetExpr = Builder.CreateICmpNE(
602706f79abSMichael Kruse       IsInSetExpr, ConstantInt::get(IsInSetExpr->getType(), 0));
603706f79abSMichael Kruse 
604706f79abSMichael Kruse   return IsInSetExpr;
605706f79abSMichael Kruse }
606706f79abSMichael Kruse 
607706f79abSMichael Kruse void BlockGenerator::generateConditionalExecution(
608706f79abSMichael Kruse     ScopStmt &Stmt, const isl::set &Subdomain, StringRef Subject,
609706f79abSMichael Kruse     const std::function<void()> &GenThenFunc) {
610dcf8d696STobias Grosser   isl::set StmtDom = Stmt.getDomain();
611706f79abSMichael Kruse 
612706f79abSMichael Kruse   // If the condition is a tautology, don't generate a condition around the
613706f79abSMichael Kruse   // code.
614f51decb5STobias Grosser   bool IsPartialWrite =
6158ea1fc19STobias Grosser       !StmtDom.intersect_params(Stmt.getParent()->getContext())
616f51decb5STobias Grosser            .is_subset(Subdomain);
617f51decb5STobias Grosser   if (!IsPartialWrite) {
618706f79abSMichael Kruse     GenThenFunc();
619706f79abSMichael Kruse     return;
620706f79abSMichael Kruse   }
621706f79abSMichael Kruse 
622706f79abSMichael Kruse   // Generate the condition.
623706f79abSMichael Kruse   Value *Cond = buildContainsCondition(Stmt, Subdomain);
624b795bfc0SMichael Kruse 
625b795bfc0SMichael Kruse   // Don't call GenThenFunc if it is never executed. An ast index expression
626b795bfc0SMichael Kruse   // might not be defined in this case.
627b795bfc0SMichael Kruse   if (auto *Const = dyn_cast<ConstantInt>(Cond))
628b795bfc0SMichael Kruse     if (Const->isZero())
629b795bfc0SMichael Kruse       return;
630b795bfc0SMichael Kruse 
631706f79abSMichael Kruse   BasicBlock *HeadBlock = Builder.GetInsertBlock();
632706f79abSMichael Kruse   StringRef BlockName = HeadBlock->getName();
633706f79abSMichael Kruse 
634706f79abSMichael Kruse   // Generate the conditional block.
635706f79abSMichael Kruse   SplitBlockAndInsertIfThen(Cond, &*Builder.GetInsertPoint(), false, nullptr,
636706f79abSMichael Kruse                             &DT, &LI);
637706f79abSMichael Kruse   BranchInst *Branch = cast<BranchInst>(HeadBlock->getTerminator());
638706f79abSMichael Kruse   BasicBlock *ThenBlock = Branch->getSuccessor(0);
639706f79abSMichael Kruse   BasicBlock *TailBlock = Branch->getSuccessor(1);
640706f79abSMichael Kruse 
641706f79abSMichael Kruse   // Assign descriptive names.
642706f79abSMichael Kruse   if (auto *CondInst = dyn_cast<Instruction>(Cond))
643706f79abSMichael Kruse     CondInst->setName("polly." + Subject + ".cond");
644706f79abSMichael Kruse   ThenBlock->setName(BlockName + "." + Subject + ".partial");
645706f79abSMichael Kruse   TailBlock->setName(BlockName + ".cont");
646706f79abSMichael Kruse 
647706f79abSMichael Kruse   // Put the client code into the conditional block and continue in the merge
648706f79abSMichael Kruse   // block afterwards.
649706f79abSMichael Kruse   Builder.SetInsertPoint(ThenBlock, ThenBlock->getFirstInsertionPt());
650706f79abSMichael Kruse   GenThenFunc();
651706f79abSMichael Kruse   Builder.SetInsertPoint(TailBlock, TailBlock->getFirstInsertionPt());
652706f79abSMichael Kruse }
653706f79abSMichael Kruse 
654e819fffeSMichael Kruse static std::string getInstName(Value *Val) {
655e819fffeSMichael Kruse   std::string Result;
656e819fffeSMichael Kruse   raw_string_ostream OS(Result);
657e819fffeSMichael Kruse   Val->printAsOperand(OS, false);
658e819fffeSMichael Kruse   return OS.str();
659e819fffeSMichael Kruse }
660e819fffeSMichael Kruse 
661e819fffeSMichael Kruse void BlockGenerator::generateBeginStmtTrace(ScopStmt &Stmt, LoopToScevMapT &LTS,
662e819fffeSMichael Kruse                                             ValueMapT &BBMap) {
663e819fffeSMichael Kruse   if (!TraceStmts)
664e819fffeSMichael Kruse     return;
665e819fffeSMichael Kruse 
666e819fffeSMichael Kruse   Scop *S = Stmt.getParent();
667e819fffeSMichael Kruse   const char *BaseName = Stmt.getBaseName();
668e819fffeSMichael Kruse 
669e819fffeSMichael Kruse   isl::ast_build AstBuild = Stmt.getAstBuild();
670e819fffeSMichael Kruse   isl::set Domain = Stmt.getDomain();
671e819fffeSMichael Kruse 
672e819fffeSMichael Kruse   isl::union_map USchedule = AstBuild.get_schedule().intersect_domain(Domain);
673e819fffeSMichael Kruse   isl::map Schedule = isl::map::from_union_map(USchedule);
674e819fffeSMichael Kruse   assert(Schedule.is_empty().is_false() &&
675e819fffeSMichael Kruse          "The stmt must have a valid instance");
676e819fffeSMichael Kruse 
677e819fffeSMichael Kruse   isl::multi_pw_aff ScheduleMultiPwAff =
678e819fffeSMichael Kruse       isl::pw_multi_aff::from_map(Schedule.reverse());
679e819fffeSMichael Kruse   isl::ast_build RestrictedBuild = AstBuild.restrict(Schedule.range());
680e819fffeSMichael Kruse 
681e819fffeSMichael Kruse   // Sequence of strings to print.
682e819fffeSMichael Kruse   SmallVector<llvm::Value *, 8> Values;
683e819fffeSMichael Kruse 
684e819fffeSMichael Kruse   // Print the name of the statement.
685e819fffeSMichael Kruse   // TODO: Indent by the depth of the statement instance in the schedule tree.
686e819fffeSMichael Kruse   Values.push_back(RuntimeDebugBuilder::getPrintableString(Builder, BaseName));
687e819fffeSMichael Kruse   Values.push_back(RuntimeDebugBuilder::getPrintableString(Builder, "("));
688e819fffeSMichael Kruse 
689e819fffeSMichael Kruse   // Add the coordinate of the statement instance.
690e819fffeSMichael Kruse   int DomDims = ScheduleMultiPwAff.dim(isl::dim::out);
691e819fffeSMichael Kruse   for (int i = 0; i < DomDims; i += 1) {
692e819fffeSMichael Kruse     if (i > 0)
693e819fffeSMichael Kruse       Values.push_back(RuntimeDebugBuilder::getPrintableString(Builder, ","));
694e819fffeSMichael Kruse 
695e819fffeSMichael Kruse     isl::ast_expr IsInSet =
696e819fffeSMichael Kruse         RestrictedBuild.expr_from(ScheduleMultiPwAff.get_pw_aff(i));
697e819fffeSMichael Kruse     Values.push_back(ExprBuilder->create(IsInSet.copy()));
698e819fffeSMichael Kruse   }
699e819fffeSMichael Kruse 
700e819fffeSMichael Kruse   if (TraceScalars) {
701e819fffeSMichael Kruse     Values.push_back(RuntimeDebugBuilder::getPrintableString(Builder, ")"));
702e819fffeSMichael Kruse     DenseSet<Instruction *> Encountered;
703e819fffeSMichael Kruse 
704e819fffeSMichael Kruse     // Add the value of each scalar (and the result of PHIs) used in the
705e819fffeSMichael Kruse     // statement.
706e819fffeSMichael Kruse     // TODO: Values used in region-statements.
707e819fffeSMichael Kruse     for (Instruction *Inst : Stmt.insts()) {
708e819fffeSMichael Kruse       if (!RuntimeDebugBuilder::isPrintable(Inst->getType()))
709e819fffeSMichael Kruse         continue;
710e819fffeSMichael Kruse 
711e819fffeSMichael Kruse       if (isa<PHINode>(Inst)) {
712e819fffeSMichael Kruse         Values.push_back(RuntimeDebugBuilder::getPrintableString(Builder, " "));
713e819fffeSMichael Kruse         Values.push_back(RuntimeDebugBuilder::getPrintableString(
714e819fffeSMichael Kruse             Builder, getInstName(Inst)));
715e819fffeSMichael Kruse         Values.push_back(RuntimeDebugBuilder::getPrintableString(Builder, "="));
716e819fffeSMichael Kruse         Values.push_back(getNewValue(Stmt, Inst, BBMap, LTS,
717e819fffeSMichael Kruse                                      LI.getLoopFor(Inst->getParent())));
718e819fffeSMichael Kruse       } else {
719e819fffeSMichael Kruse         for (Value *Op : Inst->operand_values()) {
720e819fffeSMichael Kruse           // Do not print values that cannot change during the execution of the
721e819fffeSMichael Kruse           // SCoP.
722e819fffeSMichael Kruse           auto *OpInst = dyn_cast<Instruction>(Op);
723e819fffeSMichael Kruse           if (!OpInst)
724e819fffeSMichael Kruse             continue;
725e819fffeSMichael Kruse           if (!S->contains(OpInst))
726e819fffeSMichael Kruse             continue;
727e819fffeSMichael Kruse 
728e819fffeSMichael Kruse           // Print each scalar at most once, and exclude values defined in the
729e819fffeSMichael Kruse           // statement itself.
730e819fffeSMichael Kruse           if (Encountered.count(OpInst))
731e819fffeSMichael Kruse             continue;
732e819fffeSMichael Kruse 
733e819fffeSMichael Kruse           Values.push_back(
734e819fffeSMichael Kruse               RuntimeDebugBuilder::getPrintableString(Builder, " "));
735e819fffeSMichael Kruse           Values.push_back(RuntimeDebugBuilder::getPrintableString(
736e819fffeSMichael Kruse               Builder, getInstName(OpInst)));
737e819fffeSMichael Kruse           Values.push_back(
738e819fffeSMichael Kruse               RuntimeDebugBuilder::getPrintableString(Builder, "="));
739e819fffeSMichael Kruse           Values.push_back(getNewValue(Stmt, OpInst, BBMap, LTS,
740e819fffeSMichael Kruse                                        LI.getLoopFor(Inst->getParent())));
741e819fffeSMichael Kruse           Encountered.insert(OpInst);
742e819fffeSMichael Kruse         }
743e819fffeSMichael Kruse       }
744e819fffeSMichael Kruse 
745e819fffeSMichael Kruse       Encountered.insert(Inst);
746e819fffeSMichael Kruse     }
747e819fffeSMichael Kruse 
748e819fffeSMichael Kruse     Values.push_back(RuntimeDebugBuilder::getPrintableString(Builder, "\n"));
749e819fffeSMichael Kruse   } else {
750e819fffeSMichael Kruse     Values.push_back(RuntimeDebugBuilder::getPrintableString(Builder, ")\n"));
751e819fffeSMichael Kruse   }
752e819fffeSMichael Kruse 
753e819fffeSMichael Kruse   RuntimeDebugBuilder::createCPUPrinter(Builder, ArrayRef<Value *>(Values));
754e819fffeSMichael Kruse }
755e819fffeSMichael Kruse 
7562fa35194SMichael Kruse void BlockGenerator::generateScalarStores(
7572fa35194SMichael Kruse     ScopStmt &Stmt, LoopToScevMapT &LTS, ValueMapT &BBMap,
7582fa35194SMichael Kruse     __isl_keep isl_id_to_ast_expr *NewAccesses) {
759f2cdd144STobias Grosser   Loop *L = LI.getLoopFor(Stmt.getBasicBlock());
760ecff11dcSJohannes Doerfert 
76121a059afSTobias Grosser   assert(Stmt.isBlockStmt() &&
76221a059afSTobias Grosser          "Region statements need to use the generateScalarStores() function in "
76321a059afSTobias Grosser          "the RegionGenerator");
764ecff11dcSJohannes Doerfert 
765ecff11dcSJohannes Doerfert   for (MemoryAccess *MA : Stmt) {
7662fa35194SMichael Kruse     if (MA->isOriginalArrayKind() || MA->isRead())
767ecff11dcSJohannes Doerfert       continue;
768ecff11dcSJohannes Doerfert 
7691515f6b9STobias Grosser     isl::set AccDom = MA->getAccessRelation().domain();
770fe46c3ffSTobias Grosser     std::string Subject = MA->getId().get_name();
77177394f13SMichael Kruse 
772fe46c3ffSTobias Grosser     generateConditionalExecution(
773fe46c3ffSTobias Grosser         Stmt, AccDom, Subject.c_str(), [&, this, MA]() {
774d86f2157SJohannes Doerfert           Value *Val = MA->getAccessValue();
775ee6a4fc6SMichael Kruse           if (MA->isAnyPHIKind()) {
776fe46c3ffSTobias Grosser             assert(MA->getIncoming().size() >= 1 &&
777fe46c3ffSTobias Grosser                    "Block statements have exactly one exiting block, or "
778fe46c3ffSTobias Grosser                    "multiple but "
779ee6a4fc6SMichael Kruse                    "with same incoming block and value");
780fe46c3ffSTobias Grosser             assert(std::all_of(MA->getIncoming().begin(),
781fe46c3ffSTobias Grosser                                MA->getIncoming().end(),
782ee6a4fc6SMichael Kruse                                [&](std::pair<BasicBlock *, Value *> p) -> bool {
783ee6a4fc6SMichael Kruse                                  return p.first == Stmt.getBasicBlock();
784ee6a4fc6SMichael Kruse                                }) &&
785ee6a4fc6SMichael Kruse                    "Incoming block must be statement's block");
786ee6a4fc6SMichael Kruse             Val = MA->getIncoming()[0].second;
787ee6a4fc6SMichael Kruse           }
788fe46c3ffSTobias Grosser           auto Address = getImplicitAddress(*MA, getLoopForStmt(Stmt), LTS,
789fe46c3ffSTobias Grosser                                             BBMap, NewAccesses);
790d86f2157SJohannes Doerfert 
791f2cdd144STobias Grosser           Val = getNewValue(Stmt, Val, BBMap, LTS, L);
792eac9726eSMichael Kruse           assert((!isa<Instruction>(Val) ||
793eac9726eSMichael Kruse                   DT.dominates(cast<Instruction>(Val)->getParent(),
794eac9726eSMichael Kruse                                Builder.GetInsertBlock())) &&
795eac9726eSMichael Kruse                  "Domination violation");
796eac9726eSMichael Kruse           assert((!isa<Instruction>(Address) ||
797eac9726eSMichael Kruse                   DT.dominates(cast<Instruction>(Address)->getParent(),
798eac9726eSMichael Kruse                                Builder.GetInsertBlock())) &&
799eac9726eSMichael Kruse                  "Domination violation");
8002f5cbc44SMichael Kruse 
8012f5cbc44SMichael Kruse           // The new Val might have a different type than the old Val due to
8022f5cbc44SMichael Kruse           // ScalarEvolution looking through bitcasts.
8032f5cbc44SMichael Kruse           if (Val->getType() != Address->getType()->getPointerElementType())
8042f5cbc44SMichael Kruse             Address = Builder.CreateBitOrPointerCast(
8052f5cbc44SMichael Kruse                 Address, Val->getType()->getPointerTo());
8062f5cbc44SMichael Kruse 
80792245228STobias Grosser           Builder.CreateStore(Val, Address);
808706f79abSMichael Kruse         });
809ecff11dcSJohannes Doerfert   }
810ecff11dcSJohannes Doerfert }
811ecff11dcSJohannes Doerfert 
812c0091a77STobias Grosser void BlockGenerator::createScalarInitialization(Scop &S) {
813ef74443cSJohannes Doerfert   BasicBlock *ExitBB = S.getExit();
814acf80064SEli Friedman   BasicBlock *PreEntryBB = S.getEnteringBlock();
815188542fdSJohannes Doerfert 
816acf80064SEli Friedman   Builder.SetInsertPoint(&*StartBlock->begin());
817ecff11dcSJohannes Doerfert 
818d7754a12SRoman Gareev   for (auto &Array : S.arrays()) {
819c0091a77STobias Grosser     if (Array->getNumberOfDimensions() != 0)
820c0091a77STobias Grosser       continue;
821a535dff4STobias Grosser     if (Array->isPHIKind()) {
822c0091a77STobias Grosser       // For PHI nodes, the only values we need to store are the ones that
823c0091a77STobias Grosser       // reach the PHI node from outside the region. In general there should
824c0091a77STobias Grosser       // only be one such incoming edge and this edge should enter through
825acf80064SEli Friedman       // 'PreEntryBB'.
826c0091a77STobias Grosser       auto PHI = cast<PHINode>(Array->getBasePtr());
827ecff11dcSJohannes Doerfert 
828c0091a77STobias Grosser       for (auto BI = PHI->block_begin(), BE = PHI->block_end(); BI != BE; BI++)
829acf80064SEli Friedman         if (!S.contains(*BI) && *BI != PreEntryBB)
830c0091a77STobias Grosser           llvm_unreachable("Incoming edges from outside the scop should always "
831acf80064SEli Friedman                            "come from PreEntryBB");
832c0091a77STobias Grosser 
833acf80064SEli Friedman       int Idx = PHI->getBasicBlockIndex(PreEntryBB);
834c0091a77STobias Grosser       if (Idx < 0)
835ecff11dcSJohannes Doerfert         continue;
836ecff11dcSJohannes Doerfert 
837c0091a77STobias Grosser       Value *ScalarValue = PHI->getIncomingValue(Idx);
838ecff11dcSJohannes Doerfert 
839587f1f57STobias Grosser       Builder.CreateStore(ScalarValue, getOrCreateAlloca(Array));
840c0091a77STobias Grosser       continue;
841c0091a77STobias Grosser     }
842c0091a77STobias Grosser 
843c0091a77STobias Grosser     auto *Inst = dyn_cast<Instruction>(Array->getBasePtr());
844c0091a77STobias Grosser 
845952b5304SJohannes Doerfert     if (Inst && S.contains(Inst))
846c0091a77STobias Grosser       continue;
847c0091a77STobias Grosser 
848188542fdSJohannes Doerfert     // PHI nodes that are not marked as such in their SAI object are either exit
849188542fdSJohannes Doerfert     // PHI nodes we model as common scalars but without initialization, or
850188542fdSJohannes Doerfert     // incoming phi nodes that need to be initialized. Check if the first is the
851188542fdSJohannes Doerfert     // case for Inst and do not create and initialize memory if so.
852188542fdSJohannes Doerfert     if (auto *PHI = dyn_cast_or_null<PHINode>(Inst))
853188542fdSJohannes Doerfert       if (!S.hasSingleExitEdge() && PHI->getBasicBlockIndex(ExitBB) >= 0)
854717b8667SJohannes Doerfert         continue;
855717b8667SJohannes Doerfert 
856587f1f57STobias Grosser     Builder.CreateStore(Array->getBasePtr(), getOrCreateAlloca(Array));
857ecff11dcSJohannes Doerfert   }
858ecff11dcSJohannes Doerfert }
859ecff11dcSJohannes Doerfert 
860ef74443cSJohannes Doerfert void BlockGenerator::createScalarFinalization(Scop &S) {
861ecff11dcSJohannes Doerfert   // The exit block of the __unoptimized__ region.
862ef74443cSJohannes Doerfert   BasicBlock *ExitBB = S.getExitingBlock();
863ecff11dcSJohannes Doerfert   // The merge block __just after__ the region and the optimized region.
864ef74443cSJohannes Doerfert   BasicBlock *MergeBB = S.getExit();
865ecff11dcSJohannes Doerfert 
866ecff11dcSJohannes Doerfert   // The exit block of the __optimized__ region.
867ecff11dcSJohannes Doerfert   BasicBlock *OptExitBB = *(pred_begin(MergeBB));
868ecff11dcSJohannes Doerfert   if (OptExitBB == ExitBB)
869ecff11dcSJohannes Doerfert     OptExitBB = *(++pred_begin(MergeBB));
870ecff11dcSJohannes Doerfert 
871ecff11dcSJohannes Doerfert   Builder.SetInsertPoint(OptExitBB->getTerminator());
872ecff11dcSJohannes Doerfert   for (const auto &EscapeMapping : EscapeMap) {
873ecff11dcSJohannes Doerfert     // Extract the escaping instruction and the escaping users as well as the
874ecff11dcSJohannes Doerfert     // alloca the instruction was demoted to.
875fbde4355SMichael Kruse     Instruction *EscapeInst = EscapeMapping.first;
876fbde4355SMichael Kruse     const auto &EscapeMappingValue = EscapeMapping.second;
877ecff11dcSJohannes Doerfert     const EscapeUserVectorTy &EscapeUsers = EscapeMappingValue.second;
87864c0ff41STobias Grosser     Value *ScalarAddr = EscapeMappingValue.first;
879ecff11dcSJohannes Doerfert 
880ecff11dcSJohannes Doerfert     // Reload the demoted instruction in the optimized version of the SCoP.
881d8b6ad25SJohannes Doerfert     Value *EscapeInstReload =
882ecff11dcSJohannes Doerfert         Builder.CreateLoad(ScalarAddr, EscapeInst->getName() + ".final_reload");
883d8b6ad25SJohannes Doerfert     EscapeInstReload =
884d8b6ad25SJohannes Doerfert         Builder.CreateBitOrPointerCast(EscapeInstReload, EscapeInst->getType());
885ecff11dcSJohannes Doerfert 
886ecff11dcSJohannes Doerfert     // Create the merge PHI that merges the optimized and unoptimized version.
887ecff11dcSJohannes Doerfert     PHINode *MergePHI = PHINode::Create(EscapeInst->getType(), 2,
888ecff11dcSJohannes Doerfert                                         EscapeInst->getName() + ".merge");
889b8f58b53SDuncan P. N. Exon Smith     MergePHI->insertBefore(&*MergeBB->getFirstInsertionPt());
890ecff11dcSJohannes Doerfert 
891ecff11dcSJohannes Doerfert     // Add the respective values to the merge PHI.
892ecff11dcSJohannes Doerfert     MergePHI->addIncoming(EscapeInstReload, OptExitBB);
893ecff11dcSJohannes Doerfert     MergePHI->addIncoming(EscapeInst, ExitBB);
894ecff11dcSJohannes Doerfert 
895ecff11dcSJohannes Doerfert     // The information of scalar evolution about the escaping instruction needs
896ecff11dcSJohannes Doerfert     // to be revoked so the new merged instruction will be used.
897ecff11dcSJohannes Doerfert     if (SE.isSCEVable(EscapeInst->getType()))
898ecff11dcSJohannes Doerfert       SE.forgetValue(EscapeInst);
899ecff11dcSJohannes Doerfert 
900ecff11dcSJohannes Doerfert     // Replace all uses of the demoted instruction with the merge PHI.
901ecff11dcSJohannes Doerfert     for (Instruction *EUser : EscapeUsers)
902ecff11dcSJohannes Doerfert       EUser->replaceUsesOfWith(EscapeInst, MergePHI);
903ecff11dcSJohannes Doerfert   }
904ecff11dcSJohannes Doerfert }
905ecff11dcSJohannes Doerfert 
906ffa2446fSTobias Grosser void BlockGenerator::findOutsideUsers(Scop &S) {
907d7754a12SRoman Gareev   for (auto &Array : S.arrays()) {
908ffa2446fSTobias Grosser 
909ffa2446fSTobias Grosser     if (Array->getNumberOfDimensions() != 0)
910ffa2446fSTobias Grosser       continue;
911ffa2446fSTobias Grosser 
912a535dff4STobias Grosser     if (Array->isPHIKind())
913ffa2446fSTobias Grosser       continue;
914ffa2446fSTobias Grosser 
915ffa2446fSTobias Grosser     auto *Inst = dyn_cast<Instruction>(Array->getBasePtr());
916ffa2446fSTobias Grosser 
917ffa2446fSTobias Grosser     if (!Inst)
918ffa2446fSTobias Grosser       continue;
919ffa2446fSTobias Grosser 
920ffa2446fSTobias Grosser     // Scop invariant hoisting moves some of the base pointers out of the scop.
921ffa2446fSTobias Grosser     // We can ignore these, as the invariant load hoisting already registers the
922ffa2446fSTobias Grosser     // relevant outside users.
923952b5304SJohannes Doerfert     if (!S.contains(Inst))
924ffa2446fSTobias Grosser       continue;
925ffa2446fSTobias Grosser 
926587f1f57STobias Grosser     handleOutsideUsers(S, Array);
927ffa2446fSTobias Grosser   }
928ffa2446fSTobias Grosser }
929ffa2446fSTobias Grosser 
93027d742daSTobias Grosser void BlockGenerator::createExitPHINodeMerges(Scop &S) {
93127d742daSTobias Grosser   if (S.hasSingleExitEdge())
93227d742daSTobias Grosser     return;
93327d742daSTobias Grosser 
934ef74443cSJohannes Doerfert   auto *ExitBB = S.getExitingBlock();
935ef74443cSJohannes Doerfert   auto *MergeBB = S.getExit();
93627d742daSTobias Grosser   auto *AfterMergeBB = MergeBB->getSingleSuccessor();
93727d742daSTobias Grosser   BasicBlock *OptExitBB = *(pred_begin(MergeBB));
93827d742daSTobias Grosser   if (OptExitBB == ExitBB)
93927d742daSTobias Grosser     OptExitBB = *(++pred_begin(MergeBB));
94027d742daSTobias Grosser 
94127d742daSTobias Grosser   Builder.SetInsertPoint(OptExitBB->getTerminator());
94227d742daSTobias Grosser 
943d7754a12SRoman Gareev   for (auto &SAI : S.arrays()) {
94427d742daSTobias Grosser     auto *Val = SAI->getBasePtr();
94527d742daSTobias Grosser 
946faedfcbfSMichael Kruse     // Only Value-like scalars need a merge PHI. Exit block PHIs receive either
947faedfcbfSMichael Kruse     // the original PHI's value or the reloaded incoming values from the
948faedfcbfSMichael Kruse     // generated code. An llvm::Value is merged between the original code's
949faedfcbfSMichael Kruse     // value or the generated one.
950fa53c86dSMichael Kruse     if (!SAI->isExitPHIKind())
951faedfcbfSMichael Kruse       continue;
952faedfcbfSMichael Kruse 
95327d742daSTobias Grosser     PHINode *PHI = dyn_cast<PHINode>(Val);
95427d742daSTobias Grosser     if (!PHI)
95527d742daSTobias Grosser       continue;
95627d742daSTobias Grosser 
957a3f6edaeSTobias Grosser     if (PHI->getParent() != AfterMergeBB)
95827d742daSTobias Grosser       continue;
95927d742daSTobias Grosser 
9600257a921SEli Friedman     std::string Name = PHI->getName().str();
961587f1f57STobias Grosser     Value *ScalarAddr = getOrCreateAlloca(SAI);
96227d742daSTobias Grosser     Value *Reload = Builder.CreateLoad(ScalarAddr, Name + ".ph.final_reload");
96327d742daSTobias Grosser     Reload = Builder.CreateBitOrPointerCast(Reload, PHI->getType());
96427d742daSTobias Grosser     Value *OriginalValue = PHI->getIncomingValueForBlock(MergeBB);
965faedfcbfSMichael Kruse     assert((!isa<Instruction>(OriginalValue) ||
966faedfcbfSMichael Kruse             cast<Instruction>(OriginalValue)->getParent() != MergeBB) &&
967faedfcbfSMichael Kruse            "Original value must no be one we just generated.");
96827d742daSTobias Grosser     auto *MergePHI = PHINode::Create(PHI->getType(), 2, Name + ".ph.merge");
969b8f58b53SDuncan P. N. Exon Smith     MergePHI->insertBefore(&*MergeBB->getFirstInsertionPt());
97027d742daSTobias Grosser     MergePHI->addIncoming(Reload, OptExitBB);
97127d742daSTobias Grosser     MergePHI->addIncoming(OriginalValue, ExitBB);
97227d742daSTobias Grosser     int Idx = PHI->getBasicBlockIndex(MergeBB);
97327d742daSTobias Grosser     PHI->setIncomingValue(Idx, MergePHI);
97427d742daSTobias Grosser   }
97527d742daSTobias Grosser }
97627d742daSTobias Grosser 
9771c184409STobias Grosser void BlockGenerator::invalidateScalarEvolution(Scop &S) {
9781c184409STobias Grosser   for (auto &Stmt : S)
979b3224adfSRoman Gareev     if (Stmt.isCopyStmt())
980b3224adfSRoman Gareev       continue;
981b3224adfSRoman Gareev     else if (Stmt.isBlockStmt())
9821c184409STobias Grosser       for (auto &Inst : *Stmt.getBasicBlock())
9831c184409STobias Grosser         SE.forgetValue(&Inst);
9841c184409STobias Grosser     else if (Stmt.isRegionStmt())
9851c184409STobias Grosser       for (auto *BB : Stmt.getRegion()->blocks())
9861c184409STobias Grosser         for (auto &Inst : *BB)
9871c184409STobias Grosser           SE.forgetValue(&Inst);
9881c184409STobias Grosser     else
9891c184409STobias Grosser       llvm_unreachable("Unexpected statement type found");
9901aad76c1SMichael Kruse 
9911aad76c1SMichael Kruse   // Invalidate SCEV of loops surrounding the EscapeUsers.
9921aad76c1SMichael Kruse   for (const auto &EscapeMapping : EscapeMap) {
9931aad76c1SMichael Kruse     const EscapeUserVectorTy &EscapeUsers = EscapeMapping.second.second;
9941aad76c1SMichael Kruse     for (Instruction *EUser : EscapeUsers) {
9951aad76c1SMichael Kruse       if (Loop *L = LI.getLoopFor(EUser->getParent()))
9961aad76c1SMichael Kruse         while (L) {
9971aad76c1SMichael Kruse           SE.forgetLoop(L);
9981aad76c1SMichael Kruse           L = L->getParentLoop();
9991aad76c1SMichael Kruse         }
10001aad76c1SMichael Kruse     }
10011aad76c1SMichael Kruse   }
10021c184409STobias Grosser }
10031c184409STobias Grosser 
1004ffa2446fSTobias Grosser void BlockGenerator::finalizeSCoP(Scop &S) {
1005ffa2446fSTobias Grosser   findOutsideUsers(S);
1006c0091a77STobias Grosser   createScalarInitialization(S);
100727d742daSTobias Grosser   createExitPHINodeMerges(S);
1008ef74443cSJohannes Doerfert   createScalarFinalization(S);
10091c184409STobias Grosser   invalidateScalarEvolution(S);
10103b11a16aSHongbin Zheng }
10113b11a16aSHongbin Zheng 
1012be9c9117SJohannes Doerfert VectorBlockGenerator::VectorBlockGenerator(BlockGenerator &BlockGen,
1013be9c9117SJohannes Doerfert                                            std::vector<LoopToScevMapT> &VLTS,
1014be9c9117SJohannes Doerfert                                            isl_map *Schedule)
1015bc132607STobias Grosser     : BlockGenerator(BlockGen), VLTS(VLTS), Schedule(Schedule) {
1016a00a0291SSebastian Pop   assert(Schedule && "No statement domain provided");
10173b11a16aSHongbin Zheng }
10183b11a16aSHongbin Zheng 
10192f1acac6STobias Grosser Value *VectorBlockGenerator::getVectorValue(ScopStmt &Stmt, Value *Old,
1020e602a076STobias Grosser                                             ValueMapT &VectorMap,
1021e602a076STobias Grosser                                             VectorValueMapT &ScalarMaps,
1022e602a076STobias Grosser                                             Loop *L) {
1023fe11e287SHongbin Zheng   if (Value *NewValue = VectorMap.lookup(Old))
1024fe11e287SHongbin Zheng     return NewValue;
10253b11a16aSHongbin Zheng 
10263b11a16aSHongbin Zheng   int Width = getVectorWidth();
10273b11a16aSHongbin Zheng 
10283b11a16aSHongbin Zheng   Value *Vector = UndefValue::get(VectorType::get(Old->getType(), Width));
10293b11a16aSHongbin Zheng 
10303b11a16aSHongbin Zheng   for (int Lane = 0; Lane < Width; Lane++)
1031c14582f2STobias Grosser     Vector = Builder.CreateInsertElement(
1032bc132607STobias Grosser         Vector, getNewValue(Stmt, Old, ScalarMaps[Lane], VLTS[Lane], L),
10337242ad92STobias Grosser         Builder.getInt32(Lane));
10343b11a16aSHongbin Zheng 
10353b11a16aSHongbin Zheng   VectorMap[Old] = Vector;
10363b11a16aSHongbin Zheng 
10373b11a16aSHongbin Zheng   return Vector;
10383b11a16aSHongbin Zheng }
10393b11a16aSHongbin Zheng 
10403b11a16aSHongbin Zheng Type *VectorBlockGenerator::getVectorPtrTy(const Value *Val, int Width) {
10413b11a16aSHongbin Zheng   PointerType *PointerTy = dyn_cast<PointerType>(Val->getType());
10423b11a16aSHongbin Zheng   assert(PointerTy && "PointerType expected");
10433b11a16aSHongbin Zheng 
10443b11a16aSHongbin Zheng   Type *ScalarType = PointerTy->getElementType();
10453b11a16aSHongbin Zheng   VectorType *VectorType = VectorType::get(ScalarType, Width);
10463b11a16aSHongbin Zheng 
10473b11a16aSHongbin Zheng   return PointerType::getUnqual(VectorType);
10483b11a16aSHongbin Zheng }
10493b11a16aSHongbin Zheng 
1050be9c9117SJohannes Doerfert Value *VectorBlockGenerator::generateStrideOneLoad(
10512f1acac6STobias Grosser     ScopStmt &Stmt, LoadInst *Load, VectorValueMapT &ScalarMaps,
10522d1ed0bfSTobias Grosser     __isl_keep isl_id_to_ast_expr *NewAccesses, bool NegativeStride = false) {
10530dd463faSTobias Grosser   unsigned VectorWidth = getVectorWidth();
10549646e3feSTobias Grosser   auto *Pointer = Load->getPointerOperand();
10550dd463faSTobias Grosser   Type *VectorPtrType = getVectorPtrTy(Pointer, VectorWidth);
10560dd463faSTobias Grosser   unsigned Offset = NegativeStride ? VectorWidth - 1 : 0;
10570dd463faSTobias Grosser 
10588f25b0cbSMichael Kruse   Value *NewPointer = generateLocationAccessed(Stmt, Load, ScalarMaps[Offset],
1059bc132607STobias Grosser                                                VLTS[Offset], NewAccesses);
1060c14582f2STobias Grosser   Value *VectorPtr =
1061c14582f2STobias Grosser       Builder.CreateBitCast(NewPointer, VectorPtrType, "vector_ptr");
1062c14582f2STobias Grosser   LoadInst *VecLoad =
1063c14582f2STobias Grosser       Builder.CreateLoad(VectorPtr, Load->getName() + "_p_vec_full");
10643b11a16aSHongbin Zheng   if (!Aligned)
1065725efb35SGuillaume Chatelet     VecLoad->setAlignment(Align(8));
10663b11a16aSHongbin Zheng 
10670dd463faSTobias Grosser   if (NegativeStride) {
10680dd463faSTobias Grosser     SmallVector<Constant *, 16> Indices;
10690dd463faSTobias Grosser     for (int i = VectorWidth - 1; i >= 0; i--)
10700dd463faSTobias Grosser       Indices.push_back(ConstantInt::get(Builder.getInt32Ty(), i));
10710dd463faSTobias Grosser     Constant *SV = llvm::ConstantVector::get(Indices);
10720dd463faSTobias Grosser     Value *RevVecLoad = Builder.CreateShuffleVector(
10730dd463faSTobias Grosser         VecLoad, VecLoad, SV, Load->getName() + "_reverse");
10740dd463faSTobias Grosser     return RevVecLoad;
10750dd463faSTobias Grosser   }
10760dd463faSTobias Grosser 
10773b11a16aSHongbin Zheng   return VecLoad;
10783b11a16aSHongbin Zheng }
10793b11a16aSHongbin Zheng 
10802d1ed0bfSTobias Grosser Value *VectorBlockGenerator::generateStrideZeroLoad(
10812f1acac6STobias Grosser     ScopStmt &Stmt, LoadInst *Load, ValueMapT &BBMap,
10822d1ed0bfSTobias Grosser     __isl_keep isl_id_to_ast_expr *NewAccesses) {
10839646e3feSTobias Grosser   auto *Pointer = Load->getPointerOperand();
10843b11a16aSHongbin Zheng   Type *VectorPtrType = getVectorPtrTy(Pointer, 1);
108570131d34SMichael Kruse   Value *NewPointer =
108670131d34SMichael Kruse       generateLocationAccessed(Stmt, Load, BBMap, VLTS[0], NewAccesses);
10873b11a16aSHongbin Zheng   Value *VectorPtr = Builder.CreateBitCast(NewPointer, VectorPtrType,
10883b11a16aSHongbin Zheng                                            Load->getName() + "_p_vec_p");
1089c14582f2STobias Grosser   LoadInst *ScalarLoad =
1090c14582f2STobias Grosser       Builder.CreateLoad(VectorPtr, Load->getName() + "_p_splat_one");
10913b11a16aSHongbin Zheng 
10923b11a16aSHongbin Zheng   if (!Aligned)
1093725efb35SGuillaume Chatelet     ScalarLoad->setAlignment(Align(8));
10943b11a16aSHongbin Zheng 
1095c14582f2STobias Grosser   Constant *SplatVector = Constant::getNullValue(
1096c14582f2STobias Grosser       VectorType::get(Builder.getInt32Ty(), getVectorWidth()));
10973b11a16aSHongbin Zheng 
1098c14582f2STobias Grosser   Value *VectorLoad = Builder.CreateShuffleVector(
1099c14582f2STobias Grosser       ScalarLoad, ScalarLoad, SplatVector, Load->getName() + "_p_splat");
11003b11a16aSHongbin Zheng   return VectorLoad;
11013b11a16aSHongbin Zheng }
11023b11a16aSHongbin Zheng 
1103be9c9117SJohannes Doerfert Value *VectorBlockGenerator::generateUnknownStrideLoad(
11042f1acac6STobias Grosser     ScopStmt &Stmt, LoadInst *Load, VectorValueMapT &ScalarMaps,
110509e3697fSJohannes Doerfert     __isl_keep isl_id_to_ast_expr *NewAccesses) {
11063b11a16aSHongbin Zheng   int VectorWidth = getVectorWidth();
11079646e3feSTobias Grosser   auto *Pointer = Load->getPointerOperand();
11083b11a16aSHongbin Zheng   VectorType *VectorType = VectorType::get(
11093b11a16aSHongbin Zheng       dyn_cast<PointerType>(Pointer->getType())->getElementType(), VectorWidth);
11103b11a16aSHongbin Zheng 
11113b11a16aSHongbin Zheng   Value *Vector = UndefValue::get(VectorType);
11123b11a16aSHongbin Zheng 
11133b11a16aSHongbin Zheng   for (int i = 0; i < VectorWidth; i++) {
111470131d34SMichael Kruse     Value *NewPointer = generateLocationAccessed(Stmt, Load, ScalarMaps[i],
111570131d34SMichael Kruse                                                  VLTS[i], NewAccesses);
1116c14582f2STobias Grosser     Value *ScalarLoad =
1117c14582f2STobias Grosser         Builder.CreateLoad(NewPointer, Load->getName() + "_p_scalar_");
1118c14582f2STobias Grosser     Vector = Builder.CreateInsertElement(
1119c14582f2STobias Grosser         Vector, ScalarLoad, Builder.getInt32(i), Load->getName() + "_p_vec_");
11203b11a16aSHongbin Zheng   }
11213b11a16aSHongbin Zheng 
11223b11a16aSHongbin Zheng   return Vector;
11233b11a16aSHongbin Zheng }
11243b11a16aSHongbin Zheng 
11252d1ed0bfSTobias Grosser void VectorBlockGenerator::generateLoad(
11262f1acac6STobias Grosser     ScopStmt &Stmt, LoadInst *Load, ValueMapT &VectorMap,
11272d1ed0bfSTobias Grosser     VectorValueMapT &ScalarMaps, __isl_keep isl_id_to_ast_expr *NewAccesses) {
1128c1db67e2SJohannes Doerfert   if (Value *PreloadLoad = GlobalMap.lookup(Load)) {
1129c1db67e2SJohannes Doerfert     VectorMap[Load] = Builder.CreateVectorSplat(getVectorWidth(), PreloadLoad,
1130c1db67e2SJohannes Doerfert                                                 Load->getName() + "_p");
1131c1db67e2SJohannes Doerfert     return;
1132c1db67e2SJohannes Doerfert   }
1133c1db67e2SJohannes Doerfert 
11342873645cSTobias Grosser   if (!VectorType::isValidElementType(Load->getType())) {
11353b11a16aSHongbin Zheng     for (int i = 0; i < getVectorWidth(); i++)
1136bc132607STobias Grosser       ScalarMaps[i][Load] =
1137888ab551SMichael Kruse           generateArrayLoad(Stmt, Load, ScalarMaps[i], VLTS[i], NewAccesses);
11383b11a16aSHongbin Zheng     return;
11393b11a16aSHongbin Zheng   }
11403b11a16aSHongbin Zheng 
1141184a4926STobias Grosser   const MemoryAccess &Access = Stmt.getArrayAccessFor(Load);
11423b11a16aSHongbin Zheng 
114395493984STobias Grosser   // Make sure we have scalar values available to access the pointer to
114495493984STobias Grosser   // the data location.
114595493984STobias Grosser   extractScalarValues(Load, VectorMap, ScalarMaps);
114695493984STobias Grosser 
11473b11a16aSHongbin Zheng   Value *NewLoad;
1148718d04c6STobias Grosser   if (Access.isStrideZero(isl::manage_copy(Schedule)))
11492d1ed0bfSTobias Grosser     NewLoad = generateStrideZeroLoad(Stmt, Load, ScalarMaps[0], NewAccesses);
1150718d04c6STobias Grosser   else if (Access.isStrideOne(isl::manage_copy(Schedule)))
11512d1ed0bfSTobias Grosser     NewLoad = generateStrideOneLoad(Stmt, Load, ScalarMaps, NewAccesses);
1152718d04c6STobias Grosser   else if (Access.isStrideX(isl::manage_copy(Schedule), -1))
11532d1ed0bfSTobias Grosser     NewLoad = generateStrideOneLoad(Stmt, Load, ScalarMaps, NewAccesses, true);
11543b11a16aSHongbin Zheng   else
11552d1ed0bfSTobias Grosser     NewLoad = generateUnknownStrideLoad(Stmt, Load, ScalarMaps, NewAccesses);
11563b11a16aSHongbin Zheng 
11573b11a16aSHongbin Zheng   VectorMap[Load] = NewLoad;
11583b11a16aSHongbin Zheng }
11593b11a16aSHongbin Zheng 
11602f1acac6STobias Grosser void VectorBlockGenerator::copyUnaryInst(ScopStmt &Stmt, UnaryInstruction *Inst,
11613b11a16aSHongbin Zheng                                          ValueMapT &VectorMap,
11623b11a16aSHongbin Zheng                                          VectorValueMapT &ScalarMaps) {
11633b11a16aSHongbin Zheng   int VectorWidth = getVectorWidth();
1164be9c9117SJohannes Doerfert   Value *NewOperand = getVectorValue(Stmt, Inst->getOperand(0), VectorMap,
11652c3ffc04SJohannes Doerfert                                      ScalarMaps, getLoopForStmt(Stmt));
11663b11a16aSHongbin Zheng 
11673b11a16aSHongbin Zheng   assert(isa<CastInst>(Inst) && "Can not generate vector code for instruction");
11683b11a16aSHongbin Zheng 
11693b11a16aSHongbin Zheng   const CastInst *Cast = dyn_cast<CastInst>(Inst);
11703b11a16aSHongbin Zheng   VectorType *DestType = VectorType::get(Inst->getType(), VectorWidth);
11713b11a16aSHongbin Zheng   VectorMap[Inst] = Builder.CreateCast(Cast->getOpcode(), NewOperand, DestType);
11723b11a16aSHongbin Zheng }
11733b11a16aSHongbin Zheng 
11742f1acac6STobias Grosser void VectorBlockGenerator::copyBinaryInst(ScopStmt &Stmt, BinaryOperator *Inst,
11753b11a16aSHongbin Zheng                                           ValueMapT &VectorMap,
11763b11a16aSHongbin Zheng                                           VectorValueMapT &ScalarMaps) {
11772c3ffc04SJohannes Doerfert   Loop *L = getLoopForStmt(Stmt);
11783b11a16aSHongbin Zheng   Value *OpZero = Inst->getOperand(0);
11793b11a16aSHongbin Zheng   Value *OpOne = Inst->getOperand(1);
11803b11a16aSHongbin Zheng 
11813b11a16aSHongbin Zheng   Value *NewOpZero, *NewOpOne;
1182be9c9117SJohannes Doerfert   NewOpZero = getVectorValue(Stmt, OpZero, VectorMap, ScalarMaps, L);
1183be9c9117SJohannes Doerfert   NewOpOne = getVectorValue(Stmt, OpOne, VectorMap, ScalarMaps, L);
11843b11a16aSHongbin Zheng 
11851bb59b0dSTobias Grosser   Value *NewInst = Builder.CreateBinOp(Inst->getOpcode(), NewOpZero, NewOpOne,
11863b11a16aSHongbin Zheng                                        Inst->getName() + "p_vec");
11873b11a16aSHongbin Zheng   VectorMap[Inst] = NewInst;
11883b11a16aSHongbin Zheng }
11893b11a16aSHongbin Zheng 
11902d1ed0bfSTobias Grosser void VectorBlockGenerator::copyStore(
11912f1acac6STobias Grosser     ScopStmt &Stmt, StoreInst *Store, ValueMapT &VectorMap,
11922d1ed0bfSTobias Grosser     VectorValueMapT &ScalarMaps, __isl_keep isl_id_to_ast_expr *NewAccesses) {
1193184a4926STobias Grosser   const MemoryAccess &Access = Stmt.getArrayAccessFor(Store);
11943b11a16aSHongbin Zheng 
11959646e3feSTobias Grosser   auto *Pointer = Store->getPointerOperand();
1196be9c9117SJohannes Doerfert   Value *Vector = getVectorValue(Stmt, Store->getValueOperand(), VectorMap,
11972c3ffc04SJohannes Doerfert                                  ScalarMaps, getLoopForStmt(Stmt));
11983b11a16aSHongbin Zheng 
119950fd7010STobias Grosser   // Make sure we have scalar values available to access the pointer to
120050fd7010STobias Grosser   // the data location.
120150fd7010STobias Grosser   extractScalarValues(Store, VectorMap, ScalarMaps);
120250fd7010STobias Grosser 
1203718d04c6STobias Grosser   if (Access.isStrideOne(isl::manage_copy(Schedule))) {
12041947f863SJohannes Doerfert     Type *VectorPtrType = getVectorPtrTy(Pointer, getVectorWidth());
120570131d34SMichael Kruse     Value *NewPointer = generateLocationAccessed(Stmt, Store, ScalarMaps[0],
120670131d34SMichael Kruse                                                  VLTS[0], NewAccesses);
12073b11a16aSHongbin Zheng 
1208c14582f2STobias Grosser     Value *VectorPtr =
1209c14582f2STobias Grosser         Builder.CreateBitCast(NewPointer, VectorPtrType, "vector_ptr");
12103b11a16aSHongbin Zheng     StoreInst *Store = Builder.CreateStore(Vector, VectorPtr);
12113b11a16aSHongbin Zheng 
12123b11a16aSHongbin Zheng     if (!Aligned)
1213d400d451SGuillaume Chatelet       Store->setAlignment(Align(8));
12143b11a16aSHongbin Zheng   } else {
12153b11a16aSHongbin Zheng     for (unsigned i = 0; i < ScalarMaps.size(); i++) {
12161bb59b0dSTobias Grosser       Value *Scalar = Builder.CreateExtractElement(Vector, Builder.getInt32(i));
121770131d34SMichael Kruse       Value *NewPointer = generateLocationAccessed(Stmt, Store, ScalarMaps[i],
121870131d34SMichael Kruse                                                    VLTS[i], NewAccesses);
12193b11a16aSHongbin Zheng       Builder.CreateStore(Scalar, NewPointer);
12203b11a16aSHongbin Zheng     }
12213b11a16aSHongbin Zheng   }
12223b11a16aSHongbin Zheng }
12233b11a16aSHongbin Zheng 
12243b11a16aSHongbin Zheng bool VectorBlockGenerator::hasVectorOperands(const Instruction *Inst,
12253b11a16aSHongbin Zheng                                              ValueMapT &VectorMap) {
122691f5b262STobias Grosser   for (Value *Operand : Inst->operands())
122791f5b262STobias Grosser     if (VectorMap.count(Operand))
12283b11a16aSHongbin Zheng       return true;
12293b11a16aSHongbin Zheng   return false;
12303b11a16aSHongbin Zheng }
12313b11a16aSHongbin Zheng 
12323b11a16aSHongbin Zheng bool VectorBlockGenerator::extractScalarValues(const Instruction *Inst,
12333b11a16aSHongbin Zheng                                                ValueMapT &VectorMap,
12343b11a16aSHongbin Zheng                                                VectorValueMapT &ScalarMaps) {
12353b11a16aSHongbin Zheng   bool HasVectorOperand = false;
12363b11a16aSHongbin Zheng   int VectorWidth = getVectorWidth();
12373b11a16aSHongbin Zheng 
123891f5b262STobias Grosser   for (Value *Operand : Inst->operands()) {
123991f5b262STobias Grosser     ValueMapT::iterator VecOp = VectorMap.find(Operand);
12403b11a16aSHongbin Zheng 
12413b11a16aSHongbin Zheng     if (VecOp == VectorMap.end())
12423b11a16aSHongbin Zheng       continue;
12433b11a16aSHongbin Zheng 
12443b11a16aSHongbin Zheng     HasVectorOperand = true;
12453b11a16aSHongbin Zheng     Value *NewVector = VecOp->second;
12463b11a16aSHongbin Zheng 
12473b11a16aSHongbin Zheng     for (int i = 0; i < VectorWidth; ++i) {
12483b11a16aSHongbin Zheng       ValueMapT &SM = ScalarMaps[i];
12493b11a16aSHongbin Zheng 
12503b11a16aSHongbin Zheng       // If there is one scalar extracted, all scalar elements should have
12513b11a16aSHongbin Zheng       // already been extracted by the code here. So no need to check for the
12522219d157STobias Grosser       // existence of all of them.
125391f5b262STobias Grosser       if (SM.count(Operand))
12543b11a16aSHongbin Zheng         break;
12553b11a16aSHongbin Zheng 
125691f5b262STobias Grosser       SM[Operand] =
125791f5b262STobias Grosser           Builder.CreateExtractElement(NewVector, Builder.getInt32(i));
12583b11a16aSHongbin Zheng     }
12593b11a16aSHongbin Zheng   }
12603b11a16aSHongbin Zheng 
12613b11a16aSHongbin Zheng   return HasVectorOperand;
12623b11a16aSHongbin Zheng }
12633b11a16aSHongbin Zheng 
12642d1ed0bfSTobias Grosser void VectorBlockGenerator::copyInstScalarized(
12652f1acac6STobias Grosser     ScopStmt &Stmt, Instruction *Inst, ValueMapT &VectorMap,
12662d1ed0bfSTobias Grosser     VectorValueMapT &ScalarMaps, __isl_keep isl_id_to_ast_expr *NewAccesses) {
12673b11a16aSHongbin Zheng   bool HasVectorOperand;
12683b11a16aSHongbin Zheng   int VectorWidth = getVectorWidth();
12693b11a16aSHongbin Zheng 
12703b11a16aSHongbin Zheng   HasVectorOperand = extractScalarValues(Inst, VectorMap, ScalarMaps);
12713b11a16aSHongbin Zheng 
12723b11a16aSHongbin Zheng   for (int VectorLane = 0; VectorLane < getVectorWidth(); VectorLane++)
1273be9c9117SJohannes Doerfert     BlockGenerator::copyInstruction(Stmt, Inst, ScalarMaps[VectorLane],
1274bc132607STobias Grosser                                     VLTS[VectorLane], NewAccesses);
12753b11a16aSHongbin Zheng 
12763b11a16aSHongbin Zheng   if (!VectorType::isValidElementType(Inst->getType()) || !HasVectorOperand)
12773b11a16aSHongbin Zheng     return;
12783b11a16aSHongbin Zheng 
12793b11a16aSHongbin Zheng   // Make the result available as vector value.
12803b11a16aSHongbin Zheng   VectorType *VectorType = VectorType::get(Inst->getType(), VectorWidth);
12813b11a16aSHongbin Zheng   Value *Vector = UndefValue::get(VectorType);
12823b11a16aSHongbin Zheng 
12833b11a16aSHongbin Zheng   for (int i = 0; i < VectorWidth; i++)
12843b11a16aSHongbin Zheng     Vector = Builder.CreateInsertElement(Vector, ScalarMaps[i][Inst],
12853b11a16aSHongbin Zheng                                          Builder.getInt32(i));
12863b11a16aSHongbin Zheng 
12873b11a16aSHongbin Zheng   VectorMap[Inst] = Vector;
12883b11a16aSHongbin Zheng }
12893b11a16aSHongbin Zheng 
1290bc132607STobias Grosser int VectorBlockGenerator::getVectorWidth() { return VLTS.size(); }
12913b11a16aSHongbin Zheng 
12922d1ed0bfSTobias Grosser void VectorBlockGenerator::copyInstruction(
12932f1acac6STobias Grosser     ScopStmt &Stmt, Instruction *Inst, ValueMapT &VectorMap,
12942d1ed0bfSTobias Grosser     VectorValueMapT &ScalarMaps, __isl_keep isl_id_to_ast_expr *NewAccesses) {
12953b11a16aSHongbin Zheng   // Terminator instructions control the control flow. They are explicitly
12963b11a16aSHongbin Zheng   // expressed in the clast and do not need to be copied.
12973b11a16aSHongbin Zheng   if (Inst->isTerminator())
12983b11a16aSHongbin Zheng     return;
12993b11a16aSHongbin Zheng 
13005dced269SJohannes Doerfert   if (canSyntheziseInStmt(Stmt, Inst))
1301e71c6ab5STobias Grosser     return;
1302e71c6ab5STobias Grosser 
13039646e3feSTobias Grosser   if (auto *Load = dyn_cast<LoadInst>(Inst)) {
13042d1ed0bfSTobias Grosser     generateLoad(Stmt, Load, VectorMap, ScalarMaps, NewAccesses);
13053b11a16aSHongbin Zheng     return;
13063b11a16aSHongbin Zheng   }
13073b11a16aSHongbin Zheng 
13083b11a16aSHongbin Zheng   if (hasVectorOperands(Inst, VectorMap)) {
13099646e3feSTobias Grosser     if (auto *Store = dyn_cast<StoreInst>(Inst)) {
13100446d81eSMichael Kruse       // Identified as redundant by -polly-simplify.
13110446d81eSMichael Kruse       if (!Stmt.getArrayAccessOrNULLFor(Store))
13120446d81eSMichael Kruse         return;
13130446d81eSMichael Kruse 
13142d1ed0bfSTobias Grosser       copyStore(Stmt, Store, VectorMap, ScalarMaps, NewAccesses);
13153b11a16aSHongbin Zheng       return;
13163b11a16aSHongbin Zheng     }
13173b11a16aSHongbin Zheng 
13189646e3feSTobias Grosser     if (auto *Unary = dyn_cast<UnaryInstruction>(Inst)) {
1319be9c9117SJohannes Doerfert       copyUnaryInst(Stmt, Unary, VectorMap, ScalarMaps);
13203b11a16aSHongbin Zheng       return;
13213b11a16aSHongbin Zheng     }
13223b11a16aSHongbin Zheng 
13239646e3feSTobias Grosser     if (auto *Binary = dyn_cast<BinaryOperator>(Inst)) {
1324be9c9117SJohannes Doerfert       copyBinaryInst(Stmt, Binary, VectorMap, ScalarMaps);
13253b11a16aSHongbin Zheng       return;
13263b11a16aSHongbin Zheng     }
13273b11a16aSHongbin Zheng 
1328a6d48f59SMichael Kruse     // Fallthrough: We generate scalar instructions, if we don't know how to
13293b11a16aSHongbin Zheng     // generate vector code.
13303b11a16aSHongbin Zheng   }
13313b11a16aSHongbin Zheng 
13322d1ed0bfSTobias Grosser   copyInstScalarized(Stmt, Inst, VectorMap, ScalarMaps, NewAccesses);
13333b11a16aSHongbin Zheng }
13343b11a16aSHongbin Zheng 
1335a69d4f0dSTobias Grosser void VectorBlockGenerator::generateScalarVectorLoads(
1336a69d4f0dSTobias Grosser     ScopStmt &Stmt, ValueMapT &VectorBlockMap) {
1337a69d4f0dSTobias Grosser   for (MemoryAccess *MA : Stmt) {
1338a69d4f0dSTobias Grosser     if (MA->isArrayKind() || MA->isWrite())
1339a69d4f0dSTobias Grosser       continue;
1340a69d4f0dSTobias Grosser 
1341a69d4f0dSTobias Grosser     auto *Address = getOrCreateAlloca(*MA);
1342a69d4f0dSTobias Grosser     Type *VectorPtrType = getVectorPtrTy(Address, 1);
1343a69d4f0dSTobias Grosser     Value *VectorPtr = Builder.CreateBitCast(Address, VectorPtrType,
1344a69d4f0dSTobias Grosser                                              Address->getName() + "_p_vec_p");
1345a69d4f0dSTobias Grosser     auto *Val = Builder.CreateLoad(VectorPtr, Address->getName() + ".reload");
1346a69d4f0dSTobias Grosser     Constant *SplatVector = Constant::getNullValue(
1347a69d4f0dSTobias Grosser         VectorType::get(Builder.getInt32Ty(), getVectorWidth()));
1348a69d4f0dSTobias Grosser 
1349a69d4f0dSTobias Grosser     Value *VectorVal = Builder.CreateShuffleVector(
1350a69d4f0dSTobias Grosser         Val, Val, SplatVector, Address->getName() + "_p_splat");
1351583be06fSTobias Grosser     VectorBlockMap[MA->getAccessValue()] = VectorVal;
1352a69d4f0dSTobias Grosser   }
1353a69d4f0dSTobias Grosser }
1354a69d4f0dSTobias Grosser 
1355a69d4f0dSTobias Grosser void VectorBlockGenerator::verifyNoScalarStores(ScopStmt &Stmt) {
1356a69d4f0dSTobias Grosser   for (MemoryAccess *MA : Stmt) {
1357a69d4f0dSTobias Grosser     if (MA->isArrayKind() || MA->isRead())
1358a69d4f0dSTobias Grosser       continue;
1359a69d4f0dSTobias Grosser 
1360a69d4f0dSTobias Grosser     llvm_unreachable("Scalar stores not expected in vector loop");
1361a69d4f0dSTobias Grosser   }
1362a69d4f0dSTobias Grosser }
1363a69d4f0dSTobias Grosser 
13642d1ed0bfSTobias Grosser void VectorBlockGenerator::copyStmt(
13652d1ed0bfSTobias Grosser     ScopStmt &Stmt, __isl_keep isl_id_to_ast_expr *NewAccesses) {
136621a059afSTobias Grosser   assert(Stmt.isBlockStmt() &&
136721a059afSTobias Grosser          "TODO: Only block statements can be copied by the vector block "
136821a059afSTobias Grosser          "generator");
1369275a1756SJohannes Doerfert 
1370be9c9117SJohannes Doerfert   BasicBlock *BB = Stmt.getBasicBlock();
1371b8f58b53SDuncan P. N. Exon Smith   BasicBlock *CopyBB = SplitBlock(Builder.GetInsertBlock(),
1372b8f58b53SDuncan P. N. Exon Smith                                   &*Builder.GetInsertPoint(), &DT, &LI);
13733b11a16aSHongbin Zheng   CopyBB->setName("polly.stmt." + BB->getName());
1374b8f58b53SDuncan P. N. Exon Smith   Builder.SetInsertPoint(&CopyBB->front());
13753b11a16aSHongbin Zheng 
13763b11a16aSHongbin Zheng   // Create two maps that store the mapping from the original instructions of
13773b11a16aSHongbin Zheng   // the old basic block to their copies in the new basic block. Those maps
13783b11a16aSHongbin Zheng   // are basic block local.
13793b11a16aSHongbin Zheng   //
13803b11a16aSHongbin Zheng   // As vector code generation is supported there is one map for scalar values
13813b11a16aSHongbin Zheng   // and one for vector values.
13823b11a16aSHongbin Zheng   //
13833b11a16aSHongbin Zheng   // In case we just do scalar code generation, the vectorMap is not used and
13843b11a16aSHongbin Zheng   // the scalarMap has just one dimension, which contains the mapping.
13853b11a16aSHongbin Zheng   //
13863b11a16aSHongbin Zheng   // In case vector code generation is done, an instruction may either appear
13873b11a16aSHongbin Zheng   // in the vector map once (as it is calculating >vectorwidth< values at a
13883b11a16aSHongbin Zheng   // time. Or (if the values are calculated using scalar operations), it
13893b11a16aSHongbin Zheng   // appears once in every dimension of the scalarMap.
13903b11a16aSHongbin Zheng   VectorValueMapT ScalarBlockMap(getVectorWidth());
13913b11a16aSHongbin Zheng   ValueMapT VectorBlockMap;
13923b11a16aSHongbin Zheng 
1393a69d4f0dSTobias Grosser   generateScalarVectorLoads(Stmt, VectorBlockMap);
1394a69d4f0dSTobias Grosser 
13950aff3174SMichael Kruse   for (Instruction *Inst : Stmt.getInstructions())
13960aff3174SMichael Kruse     copyInstruction(Stmt, Inst, VectorBlockMap, ScalarBlockMap, NewAccesses);
1397a69d4f0dSTobias Grosser 
1398a69d4f0dSTobias Grosser   verifyNoScalarStores(Stmt);
13993b11a16aSHongbin Zheng }
1400275a1756SJohannes Doerfert 
1401ecff11dcSJohannes Doerfert BasicBlock *RegionGenerator::repairDominance(BasicBlock *BB,
1402ecff11dcSJohannes Doerfert                                              BasicBlock *BBCopy) {
1403514f6efaSJohannes Doerfert 
1404514f6efaSJohannes Doerfert   BasicBlock *BBIDom = DT.getNode(BB)->getIDom()->getBlock();
1405deefbcedSTobias Grosser   BasicBlock *BBCopyIDom = EndBlockMap.lookup(BBIDom);
1406514f6efaSJohannes Doerfert 
1407514f6efaSJohannes Doerfert   if (BBCopyIDom)
1408514f6efaSJohannes Doerfert     DT.changeImmediateDominator(BBCopy, BBCopyIDom);
1409514f6efaSJohannes Doerfert 
1410deefbcedSTobias Grosser   return StartBlockMap.lookup(BBIDom);
1411514f6efaSJohannes Doerfert }
1412514f6efaSJohannes Doerfert 
1413f33c125dSMichael Kruse // This is to determine whether an llvm::Value (defined in @p BB) is usable when
1414f33c125dSMichael Kruse // leaving a subregion. The straight-forward DT.dominates(BB, R->getExitBlock())
1415f33c125dSMichael Kruse // does not work in cases where the exit block has edges from outside the
1416f33c125dSMichael Kruse // region. In that case the llvm::Value would never be usable in in the exit
1417f33c125dSMichael Kruse // block. The RegionGenerator however creates an new exit block ('ExitBBCopy')
1418f33c125dSMichael Kruse // for the subregion's exiting edges only. We need to determine whether an
1419f33c125dSMichael Kruse // llvm::Value is usable in there. We do this by checking whether it dominates
1420f33c125dSMichael Kruse // all exiting blocks individually.
1421f33c125dSMichael Kruse static bool isDominatingSubregionExit(const DominatorTree &DT, Region *R,
1422f33c125dSMichael Kruse                                       BasicBlock *BB) {
1423f33c125dSMichael Kruse   for (auto ExitingBB : predecessors(R->getExit())) {
1424f33c125dSMichael Kruse     // Check for non-subregion incoming edges.
1425f33c125dSMichael Kruse     if (!R->contains(ExitingBB))
1426f33c125dSMichael Kruse       continue;
1427f33c125dSMichael Kruse 
1428f33c125dSMichael Kruse     if (!DT.dominates(BB, ExitingBB))
1429f33c125dSMichael Kruse       return false;
1430f33c125dSMichael Kruse   }
1431f33c125dSMichael Kruse 
1432f33c125dSMichael Kruse   return true;
1433f33c125dSMichael Kruse }
1434f33c125dSMichael Kruse 
1435f33c125dSMichael Kruse // Find the direct dominator of the subregion's exit block if the subregion was
1436f33c125dSMichael Kruse // simplified.
1437f33c125dSMichael Kruse static BasicBlock *findExitDominator(DominatorTree &DT, Region *R) {
1438f33c125dSMichael Kruse   BasicBlock *Common = nullptr;
1439f33c125dSMichael Kruse   for (auto ExitingBB : predecessors(R->getExit())) {
1440f33c125dSMichael Kruse     // Check for non-subregion incoming edges.
1441f33c125dSMichael Kruse     if (!R->contains(ExitingBB))
1442f33c125dSMichael Kruse       continue;
1443f33c125dSMichael Kruse 
1444f33c125dSMichael Kruse     // First exiting edge.
1445f33c125dSMichael Kruse     if (!Common) {
1446f33c125dSMichael Kruse       Common = ExitingBB;
1447f33c125dSMichael Kruse       continue;
1448f33c125dSMichael Kruse     }
1449f33c125dSMichael Kruse 
1450f33c125dSMichael Kruse     Common = DT.findNearestCommonDominator(Common, ExitingBB);
1451f33c125dSMichael Kruse   }
1452f33c125dSMichael Kruse 
1453f33c125dSMichael Kruse   assert(Common && R->contains(Common));
1454f33c125dSMichael Kruse   return Common;
1455f33c125dSMichael Kruse }
1456f33c125dSMichael Kruse 
1457bc132607STobias Grosser void RegionGenerator::copyStmt(ScopStmt &Stmt, LoopToScevMapT &LTS,
14582d1ed0bfSTobias Grosser                                isl_id_to_ast_expr *IdToAstExp) {
1459275a1756SJohannes Doerfert   assert(Stmt.isRegionStmt() &&
1460d3f21833STobias Grosser          "Only region statements can be copied by the region generator");
1461275a1756SJohannes Doerfert 
1462ecff11dcSJohannes Doerfert   // Forget all old mappings.
1463deefbcedSTobias Grosser   StartBlockMap.clear();
1464deefbcedSTobias Grosser   EndBlockMap.clear();
1465ecff11dcSJohannes Doerfert   RegionMaps.clear();
1466ecff11dcSJohannes Doerfert   IncompletePHINodeMap.clear();
1467ecff11dcSJohannes Doerfert 
1468225f0d1eSMichael Kruse   // Collection of all values related to this subregion.
1469225f0d1eSMichael Kruse   ValueMapT ValueMap;
1470225f0d1eSMichael Kruse 
1471275a1756SJohannes Doerfert   // The region represented by the statement.
1472275a1756SJohannes Doerfert   Region *R = Stmt.getRegion();
1473275a1756SJohannes Doerfert 
1474ecff11dcSJohannes Doerfert   // Create a dedicated entry for the region where we can reload all demoted
1475ecff11dcSJohannes Doerfert   // inputs.
1476ecff11dcSJohannes Doerfert   BasicBlock *EntryBB = R->getEntry();
1477b8f58b53SDuncan P. N. Exon Smith   BasicBlock *EntryBBCopy = SplitBlock(Builder.GetInsertBlock(),
1478b8f58b53SDuncan P. N. Exon Smith                                        &*Builder.GetInsertPoint(), &DT, &LI);
1479ecff11dcSJohannes Doerfert   EntryBBCopy->setName("polly.stmt." + EntryBB->getName() + ".entry");
1480b8f58b53SDuncan P. N. Exon Smith   Builder.SetInsertPoint(&EntryBBCopy->front());
1481514f6efaSJohannes Doerfert 
1482c993739eSMichael Kruse   ValueMapT &EntryBBMap = RegionMaps[EntryBBCopy];
14832fa35194SMichael Kruse   generateScalarLoads(Stmt, LTS, EntryBBMap, IdToAstExp);
1484e819fffeSMichael Kruse   generateBeginStmtTrace(Stmt, LTS, EntryBBMap);
1485225f0d1eSMichael Kruse 
1486ecff11dcSJohannes Doerfert   for (auto PI = pred_begin(EntryBB), PE = pred_end(EntryBB); PI != PE; ++PI)
1487deefbcedSTobias Grosser     if (!R->contains(*PI)) {
1488deefbcedSTobias Grosser       StartBlockMap[*PI] = EntryBBCopy;
1489deefbcedSTobias Grosser       EndBlockMap[*PI] = EntryBBCopy;
1490deefbcedSTobias Grosser     }
1491275a1756SJohannes Doerfert 
1492275a1756SJohannes Doerfert   // Iterate over all blocks in the region in a breadth-first search.
1493275a1756SJohannes Doerfert   std::deque<BasicBlock *> Blocks;
14945b1abfc8SMandeep Singh Grang   SmallSetVector<BasicBlock *, 8> SeenBlocks;
1495ecff11dcSJohannes Doerfert   Blocks.push_back(EntryBB);
1496ecff11dcSJohannes Doerfert   SeenBlocks.insert(EntryBB);
1497275a1756SJohannes Doerfert 
1498275a1756SJohannes Doerfert   while (!Blocks.empty()) {
1499275a1756SJohannes Doerfert     BasicBlock *BB = Blocks.front();
1500275a1756SJohannes Doerfert     Blocks.pop_front();
1501275a1756SJohannes Doerfert 
1502514f6efaSJohannes Doerfert     // First split the block and update dominance information.
1503514f6efaSJohannes Doerfert     BasicBlock *BBCopy = splitBB(BB);
1504ecff11dcSJohannes Doerfert     BasicBlock *BBCopyIDom = repairDominance(BB, BBCopy);
1505ecff11dcSJohannes Doerfert 
1506c993739eSMichael Kruse     // Get the mapping for this block and initialize it with either the scalar
1507c993739eSMichael Kruse     // loads from the generated entering block (which dominates all blocks of
1508c993739eSMichael Kruse     // this subregion) or the maps of the immediate dominator, if part of the
1509c993739eSMichael Kruse     // subregion. The latter necessarily includes the former.
1510c993739eSMichael Kruse     ValueMapT *InitBBMap;
1511c993739eSMichael Kruse     if (BBCopyIDom) {
1512c993739eSMichael Kruse       assert(RegionMaps.count(BBCopyIDom));
1513c993739eSMichael Kruse       InitBBMap = &RegionMaps[BBCopyIDom];
1514c993739eSMichael Kruse     } else
1515c993739eSMichael Kruse       InitBBMap = &EntryBBMap;
1516c993739eSMichael Kruse     auto Inserted = RegionMaps.insert(std::make_pair(BBCopy, *InitBBMap));
1517c993739eSMichael Kruse     ValueMapT &RegionMap = Inserted.first->second;
1518514f6efaSJohannes Doerfert 
1519275a1756SJohannes Doerfert     // Copy the block with the BlockGenerator.
1520b8f58b53SDuncan P. N. Exon Smith     Builder.SetInsertPoint(&BBCopy->front());
1521bc132607STobias Grosser     copyBB(Stmt, BB, BBCopy, RegionMap, LTS, IdToAstExp);
1522275a1756SJohannes Doerfert 
1523ecff11dcSJohannes Doerfert     // In order to remap PHI nodes we store also basic block mappings.
1524deefbcedSTobias Grosser     StartBlockMap[BB] = BBCopy;
1525deefbcedSTobias Grosser     EndBlockMap[BB] = Builder.GetInsertBlock();
1526ecff11dcSJohannes Doerfert 
1527ecff11dcSJohannes Doerfert     // Add values to incomplete PHI nodes waiting for this block to be copied.
1528ecff11dcSJohannes Doerfert     for (const PHINodePairTy &PHINodePair : IncompletePHINodeMap[BB])
1529bc132607STobias Grosser       addOperandToPHI(Stmt, PHINodePair.first, PHINodePair.second, BB, LTS);
1530ecff11dcSJohannes Doerfert     IncompletePHINodeMap[BB].clear();
1531ecff11dcSJohannes Doerfert 
1532275a1756SJohannes Doerfert     // And continue with new successors inside the region.
1533275a1756SJohannes Doerfert     for (auto SI = succ_begin(BB), SE = succ_end(BB); SI != SE; SI++)
15345b1abfc8SMandeep Singh Grang       if (R->contains(*SI) && SeenBlocks.insert(*SI))
1535275a1756SJohannes Doerfert         Blocks.push_back(*SI);
1536ebffcbeeSMichael Kruse 
1537ebffcbeeSMichael Kruse     // Remember value in case it is visible after this subregion.
1538f33c125dSMichael Kruse     if (isDominatingSubregionExit(DT, R, BB))
1539ebffcbeeSMichael Kruse       ValueMap.insert(RegionMap.begin(), RegionMap.end());
1540275a1756SJohannes Doerfert   }
1541275a1756SJohannes Doerfert 
1542275a1756SJohannes Doerfert   // Now create a new dedicated region exit block and add it to the region map.
1543b8f58b53SDuncan P. N. Exon Smith   BasicBlock *ExitBBCopy = SplitBlock(Builder.GetInsertBlock(),
1544b8f58b53SDuncan P. N. Exon Smith                                       &*Builder.GetInsertPoint(), &DT, &LI);
1545ecff11dcSJohannes Doerfert   ExitBBCopy->setName("polly.stmt." + R->getExit()->getName() + ".exit");
1546deefbcedSTobias Grosser   StartBlockMap[R->getExit()] = ExitBBCopy;
1547deefbcedSTobias Grosser   EndBlockMap[R->getExit()] = ExitBBCopy;
1548514f6efaSJohannes Doerfert 
1549deefbcedSTobias Grosser   BasicBlock *ExitDomBBCopy = EndBlockMap.lookup(findExitDominator(DT, R));
155021a059afSTobias Grosser   assert(ExitDomBBCopy &&
155121a059afSTobias Grosser          "Common exit dominator must be within region; at least the entry node "
155221a059afSTobias Grosser          "must match");
1553f33c125dSMichael Kruse   DT.changeImmediateDominator(ExitBBCopy, ExitDomBBCopy);
1554275a1756SJohannes Doerfert 
1555275a1756SJohannes Doerfert   // As the block generator doesn't handle control flow we need to add the
1556275a1756SJohannes Doerfert   // region control flow by hand after all blocks have been copied.
1557275a1756SJohannes Doerfert   for (BasicBlock *BB : SeenBlocks) {
1558275a1756SJohannes Doerfert 
1559deefbcedSTobias Grosser     BasicBlock *BBCopyStart = StartBlockMap[BB];
1560deefbcedSTobias Grosser     BasicBlock *BBCopyEnd = EndBlockMap[BB];
1561e303c87eSChandler Carruth     Instruction *TI = BB->getTerminator();
1562e114dc02SJohannes Doerfert     if (isa<UnreachableInst>(TI)) {
1563deefbcedSTobias Grosser       while (!BBCopyEnd->empty())
1564deefbcedSTobias Grosser         BBCopyEnd->begin()->eraseFromParent();
1565deefbcedSTobias Grosser       new UnreachableInst(BBCopyEnd->getContext(), BBCopyEnd);
1566e114dc02SJohannes Doerfert       continue;
1567e114dc02SJohannes Doerfert     }
1568e114dc02SJohannes Doerfert 
1569deefbcedSTobias Grosser     Instruction *BICopy = BBCopyEnd->getTerminator();
1570275a1756SJohannes Doerfert 
1571deefbcedSTobias Grosser     ValueMapT &RegionMap = RegionMaps[BBCopyStart];
1572deefbcedSTobias Grosser     RegionMap.insert(StartBlockMap.begin(), StartBlockMap.end());
1573514f6efaSJohannes Doerfert 
157445e7944bSTobias Grosser     Builder.SetInsertPoint(BICopy);
15759a132f36SJohannes Doerfert     copyInstScalar(Stmt, TI, RegionMap, LTS);
1576275a1756SJohannes Doerfert     BICopy->eraseFromParent();
1577275a1756SJohannes Doerfert   }
1578275a1756SJohannes Doerfert 
1579ecff11dcSJohannes Doerfert   // Add counting PHI nodes to all loops in the region that can be used as
1580a6d48f59SMichael Kruse   // replacement for SCEVs referring to the old loop.
1581ecff11dcSJohannes Doerfert   for (BasicBlock *BB : SeenBlocks) {
1582ecff11dcSJohannes Doerfert     Loop *L = LI.getLoopFor(BB);
1583bc29e0b2STobias Grosser     if (L == nullptr || L->getHeader() != BB || !R->contains(L))
1584ecff11dcSJohannes Doerfert       continue;
1585ecff11dcSJohannes Doerfert 
1586deefbcedSTobias Grosser     BasicBlock *BBCopy = StartBlockMap[BB];
1587ecff11dcSJohannes Doerfert     Value *NullVal = Builder.getInt32(0);
1588ecff11dcSJohannes Doerfert     PHINode *LoopPHI =
1589ecff11dcSJohannes Doerfert         PHINode::Create(Builder.getInt32Ty(), 2, "polly.subregion.iv");
1590ecff11dcSJohannes Doerfert     Instruction *LoopPHIInc = BinaryOperator::CreateAdd(
1591ecff11dcSJohannes Doerfert         LoopPHI, Builder.getInt32(1), "polly.subregion.iv.inc");
1592b8f58b53SDuncan P. N. Exon Smith     LoopPHI->insertBefore(&BBCopy->front());
1593ecff11dcSJohannes Doerfert     LoopPHIInc->insertBefore(BBCopy->getTerminator());
1594ecff11dcSJohannes Doerfert 
1595ecff11dcSJohannes Doerfert     for (auto *PredBB : make_range(pred_begin(BB), pred_end(BB))) {
1596ecff11dcSJohannes Doerfert       if (!R->contains(PredBB))
1597ecff11dcSJohannes Doerfert         continue;
1598ecff11dcSJohannes Doerfert       if (L->contains(PredBB))
1599deefbcedSTobias Grosser         LoopPHI->addIncoming(LoopPHIInc, EndBlockMap[PredBB]);
1600ecff11dcSJohannes Doerfert       else
1601deefbcedSTobias Grosser         LoopPHI->addIncoming(NullVal, EndBlockMap[PredBB]);
1602ecff11dcSJohannes Doerfert     }
1603ecff11dcSJohannes Doerfert 
1604ecff11dcSJohannes Doerfert     for (auto *PredBBCopy : make_range(pred_begin(BBCopy), pred_end(BBCopy)))
1605ecff11dcSJohannes Doerfert       if (LoopPHI->getBasicBlockIndex(PredBBCopy) < 0)
1606ecff11dcSJohannes Doerfert         LoopPHI->addIncoming(NullVal, PredBBCopy);
1607ecff11dcSJohannes Doerfert 
1608ecff11dcSJohannes Doerfert     LTS[L] = SE.getUnknown(LoopPHI);
1609ecff11dcSJohannes Doerfert   }
1610ecff11dcSJohannes Doerfert 
1611225f0d1eSMichael Kruse   // Continue generating code in the exit block.
1612b8f58b53SDuncan P. N. Exon Smith   Builder.SetInsertPoint(&*ExitBBCopy->getFirstInsertionPt());
1613225f0d1eSMichael Kruse 
1614225f0d1eSMichael Kruse   // Write values visible to other statements.
16152fa35194SMichael Kruse   generateScalarStores(Stmt, LTS, ValueMap, IdToAstExp);
1616deefbcedSTobias Grosser   StartBlockMap.clear();
1617deefbcedSTobias Grosser   EndBlockMap.clear();
16183e956020STobias Grosser   RegionMaps.clear();
16193e956020STobias Grosser   IncompletePHINodeMap.clear();
1620275a1756SJohannes Doerfert }
1621ecff11dcSJohannes Doerfert 
1622ee6a4fc6SMichael Kruse PHINode *RegionGenerator::buildExitPHI(MemoryAccess *MA, LoopToScevMapT &LTS,
1623ee6a4fc6SMichael Kruse                                        ValueMapT &BBMap, Loop *L) {
1624ee6a4fc6SMichael Kruse   ScopStmt *Stmt = MA->getStatement();
1625ee6a4fc6SMichael Kruse   Region *SubR = Stmt->getRegion();
1626ee6a4fc6SMichael Kruse   auto Incoming = MA->getIncoming();
1627ee6a4fc6SMichael Kruse 
1628ee6a4fc6SMichael Kruse   PollyIRBuilder::InsertPointGuard IPGuard(Builder);
1629ee6a4fc6SMichael Kruse   PHINode *OrigPHI = cast<PHINode>(MA->getAccessInstruction());
1630ee6a4fc6SMichael Kruse   BasicBlock *NewSubregionExit = Builder.GetInsertBlock();
1631ee6a4fc6SMichael Kruse 
1632ee6a4fc6SMichael Kruse   // This can happen if the subregion is simplified after the ScopStmts
1633ee6a4fc6SMichael Kruse   // have been created; simplification happens as part of CodeGeneration.
1634ee6a4fc6SMichael Kruse   if (OrigPHI->getParent() != SubR->getExit()) {
1635ee6a4fc6SMichael Kruse     BasicBlock *FormerExit = SubR->getExitingBlock();
1636ee6a4fc6SMichael Kruse     if (FormerExit)
1637deefbcedSTobias Grosser       NewSubregionExit = StartBlockMap.lookup(FormerExit);
1638ee6a4fc6SMichael Kruse   }
1639ee6a4fc6SMichael Kruse 
1640ee6a4fc6SMichael Kruse   PHINode *NewPHI = PHINode::Create(OrigPHI->getType(), Incoming.size(),
1641ee6a4fc6SMichael Kruse                                     "polly." + OrigPHI->getName(),
1642ee6a4fc6SMichael Kruse                                     NewSubregionExit->getFirstNonPHI());
1643ee6a4fc6SMichael Kruse 
1644ee6a4fc6SMichael Kruse   // Add the incoming values to the PHI.
1645ee6a4fc6SMichael Kruse   for (auto &Pair : Incoming) {
1646ee6a4fc6SMichael Kruse     BasicBlock *OrigIncomingBlock = Pair.first;
1647deefbcedSTobias Grosser     BasicBlock *NewIncomingBlockStart = StartBlockMap.lookup(OrigIncomingBlock);
1648deefbcedSTobias Grosser     BasicBlock *NewIncomingBlockEnd = EndBlockMap.lookup(OrigIncomingBlock);
1649deefbcedSTobias Grosser     Builder.SetInsertPoint(NewIncomingBlockEnd->getTerminator());
1650deefbcedSTobias Grosser     assert(RegionMaps.count(NewIncomingBlockStart));
1651deefbcedSTobias Grosser     assert(RegionMaps.count(NewIncomingBlockEnd));
1652deefbcedSTobias Grosser     ValueMapT *LocalBBMap = &RegionMaps[NewIncomingBlockStart];
1653ee6a4fc6SMichael Kruse 
1654ee6a4fc6SMichael Kruse     Value *OrigIncomingValue = Pair.second;
1655ee6a4fc6SMichael Kruse     Value *NewIncomingValue =
1656ee6a4fc6SMichael Kruse         getNewValue(*Stmt, OrigIncomingValue, *LocalBBMap, LTS, L);
1657deefbcedSTobias Grosser     NewPHI->addIncoming(NewIncomingValue, NewIncomingBlockEnd);
1658ee6a4fc6SMichael Kruse   }
1659ee6a4fc6SMichael Kruse 
1660ee6a4fc6SMichael Kruse   return NewPHI;
1661ee6a4fc6SMichael Kruse }
1662ee6a4fc6SMichael Kruse 
1663ee6a4fc6SMichael Kruse Value *RegionGenerator::getExitScalar(MemoryAccess *MA, LoopToScevMapT &LTS,
1664ee6a4fc6SMichael Kruse                                       ValueMapT &BBMap) {
1665ee6a4fc6SMichael Kruse   ScopStmt *Stmt = MA->getStatement();
1666ee6a4fc6SMichael Kruse 
1667ee6a4fc6SMichael Kruse   // TODO: Add some test cases that ensure this is really the right choice.
1668ee6a4fc6SMichael Kruse   Loop *L = LI.getLoopFor(Stmt->getRegion()->getExit());
1669ee6a4fc6SMichael Kruse 
1670ee6a4fc6SMichael Kruse   if (MA->isAnyPHIKind()) {
1671ee6a4fc6SMichael Kruse     auto Incoming = MA->getIncoming();
1672ee6a4fc6SMichael Kruse     assert(!Incoming.empty() &&
1673ee6a4fc6SMichael Kruse            "PHI WRITEs must have originate from at least one incoming block");
1674ee6a4fc6SMichael Kruse 
1675ee6a4fc6SMichael Kruse     // If there is only one incoming value, we do not need to create a PHI.
1676ee6a4fc6SMichael Kruse     if (Incoming.size() == 1) {
1677ee6a4fc6SMichael Kruse       Value *OldVal = Incoming[0].second;
1678ee6a4fc6SMichael Kruse       return getNewValue(*Stmt, OldVal, BBMap, LTS, L);
1679ee6a4fc6SMichael Kruse     }
1680ee6a4fc6SMichael Kruse 
1681ee6a4fc6SMichael Kruse     return buildExitPHI(MA, LTS, BBMap, L);
1682ee6a4fc6SMichael Kruse   }
1683ee6a4fc6SMichael Kruse 
16844d5a9172STobias Grosser   // MemoryKind::Value accesses leaving the subregion must dominate the exit
16854d5a9172STobias Grosser   // block; just pass the copied value.
1686ee6a4fc6SMichael Kruse   Value *OldVal = MA->getAccessValue();
1687ee6a4fc6SMichael Kruse   return getNewValue(*Stmt, OldVal, BBMap, LTS, L);
1688ee6a4fc6SMichael Kruse }
1689ee6a4fc6SMichael Kruse 
16902fa35194SMichael Kruse void RegionGenerator::generateScalarStores(
16912fa35194SMichael Kruse     ScopStmt &Stmt, LoopToScevMapT &LTS, ValueMapT &BBMap,
16922fa35194SMichael Kruse     __isl_keep isl_id_to_ast_expr *NewAccesses) {
169375296901STobias Grosser   assert(Stmt.getRegion() &&
169475296901STobias Grosser          "Block statements need to use the generateScalarStores() "
1695ecff11dcSJohannes Doerfert          "function in the BlockGenerator");
1696ecff11dcSJohannes Doerfert 
1697c4c679c2SMichael Kruse   // Get the exit scalar values before generating the writes.
1698c4c679c2SMichael Kruse   // This is necessary because RegionGenerator::getExitScalar may insert
1699c4c679c2SMichael Kruse   // PHINodes that depend on the region's exiting blocks. But
1700c4c679c2SMichael Kruse   // BlockGenerator::generateConditionalExecution may insert a new basic block
1701c4c679c2SMichael Kruse   // such that the current basic block is not a direct successor of the exiting
1702c4c679c2SMichael Kruse   // blocks anymore. Hence, build the PHINodes while the current block is still
1703c4c679c2SMichael Kruse   // the direct successor.
1704c4c679c2SMichael Kruse   SmallDenseMap<MemoryAccess *, Value *> NewExitScalars;
1705c4c679c2SMichael Kruse   for (MemoryAccess *MA : Stmt) {
1706c4c679c2SMichael Kruse     if (MA->isOriginalArrayKind() || MA->isRead())
1707c4c679c2SMichael Kruse       continue;
1708c4c679c2SMichael Kruse 
1709c4c679c2SMichael Kruse     Value *NewVal = getExitScalar(MA, LTS, BBMap);
1710c4c679c2SMichael Kruse     NewExitScalars[MA] = NewVal;
1711c4c679c2SMichael Kruse   }
1712c4c679c2SMichael Kruse 
1713ecff11dcSJohannes Doerfert   for (MemoryAccess *MA : Stmt) {
17142fa35194SMichael Kruse     if (MA->isOriginalArrayKind() || MA->isRead())
1715ecff11dcSJohannes Doerfert       continue;
1716ecff11dcSJohannes Doerfert 
17171515f6b9STobias Grosser     isl::set AccDom = MA->getAccessRelation().domain();
1718fe46c3ffSTobias Grosser     std::string Subject = MA->getId().get_name();
1719fe46c3ffSTobias Grosser     generateConditionalExecution(
1720fe46c3ffSTobias Grosser         Stmt, AccDom, Subject.c_str(), [&, this, MA]() {
1721c4c679c2SMichael Kruse           Value *NewVal = NewExitScalars.lookup(MA);
1722c4c679c2SMichael Kruse           assert(NewVal && "The exit scalar must be determined before");
1723fe46c3ffSTobias Grosser           Value *Address = getImplicitAddress(*MA, getLoopForStmt(Stmt), LTS,
1724fe46c3ffSTobias Grosser                                               BBMap, NewAccesses);
1725eac9726eSMichael Kruse           assert((!isa<Instruction>(NewVal) ||
1726eac9726eSMichael Kruse                   DT.dominates(cast<Instruction>(NewVal)->getParent(),
1727eac9726eSMichael Kruse                                Builder.GetInsertBlock())) &&
1728eac9726eSMichael Kruse                  "Domination violation");
1729eac9726eSMichael Kruse           assert((!isa<Instruction>(Address) ||
1730eac9726eSMichael Kruse                   DT.dominates(cast<Instruction>(Address)->getParent(),
1731eac9726eSMichael Kruse                                Builder.GetInsertBlock())) &&
1732eac9726eSMichael Kruse                  "Domination violation");
1733ee6a4fc6SMichael Kruse           Builder.CreateStore(NewVal, Address);
1734706f79abSMichael Kruse         });
1735ecff11dcSJohannes Doerfert   }
1736ecff11dcSJohannes Doerfert }
1737ecff11dcSJohannes Doerfert 
1738943c369cSTobias Grosser void RegionGenerator::addOperandToPHI(ScopStmt &Stmt, PHINode *PHI,
1739ecff11dcSJohannes Doerfert                                       PHINode *PHICopy, BasicBlock *IncomingBB,
1740ecff11dcSJohannes Doerfert                                       LoopToScevMapT &LTS) {
1741ecff11dcSJohannes Doerfert   // If the incoming block was not yet copied mark this PHI as incomplete.
1742ecff11dcSJohannes Doerfert   // Once the block will be copied the incoming value will be added.
1743deefbcedSTobias Grosser   BasicBlock *BBCopyStart = StartBlockMap[IncomingBB];
1744deefbcedSTobias Grosser   BasicBlock *BBCopyEnd = EndBlockMap[IncomingBB];
1745deefbcedSTobias Grosser   if (!BBCopyStart) {
1746deefbcedSTobias Grosser     assert(!BBCopyEnd);
17478d89179eSMichael Kruse     assert(Stmt.represents(IncomingBB) &&
1748ecff11dcSJohannes Doerfert            "Bad incoming block for PHI in non-affine region");
1749ecff11dcSJohannes Doerfert     IncompletePHINodeMap[IncomingBB].push_back(std::make_pair(PHI, PHICopy));
1750ecff11dcSJohannes Doerfert     return;
1751ecff11dcSJohannes Doerfert   }
1752ecff11dcSJohannes Doerfert 
1753deefbcedSTobias Grosser   assert(RegionMaps.count(BBCopyStart) &&
1754deefbcedSTobias Grosser          "Incoming PHI block did not have a BBMap");
1755deefbcedSTobias Grosser   ValueMapT &BBCopyMap = RegionMaps[BBCopyStart];
1756ecff11dcSJohannes Doerfert 
175775dfaa1dSTobias Grosser   Value *OpCopy = nullptr;
175875dfaa1dSTobias Grosser 
17598d89179eSMichael Kruse   if (Stmt.represents(IncomingBB)) {
1760ecff11dcSJohannes Doerfert     Value *Op = PHI->getIncomingValueForBlock(IncomingBB);
1761dc122222SMichael Kruse 
17626ba92714SJohannes Doerfert     // If the current insert block is different from the PHIs incoming block
17636ba92714SJohannes Doerfert     // change it, otherwise do not.
17646ba92714SJohannes Doerfert     auto IP = Builder.GetInsertPoint();
1765deefbcedSTobias Grosser     if (IP->getParent() != BBCopyEnd)
1766deefbcedSTobias Grosser       Builder.SetInsertPoint(BBCopyEnd->getTerminator());
17672c3ffc04SJohannes Doerfert     OpCopy = getNewValue(Stmt, Op, BBCopyMap, LTS, getLoopForStmt(Stmt));
1768deefbcedSTobias Grosser     if (IP->getParent() != BBCopyEnd)
17696ba92714SJohannes Doerfert       Builder.SetInsertPoint(&*IP);
1770ecff11dcSJohannes Doerfert   } else {
177175dfaa1dSTobias Grosser     // All edges from outside the non-affine region become a single edge
177275dfaa1dSTobias Grosser     // in the new copy of the non-affine region. Make sure to only add the
177375dfaa1dSTobias Grosser     // corresponding edge the first time we encounter a basic block from
177475dfaa1dSTobias Grosser     // outside the non-affine region.
1775deefbcedSTobias Grosser     if (PHICopy->getBasicBlockIndex(BBCopyEnd) >= 0)
1776ecff11dcSJohannes Doerfert       return;
1777ecff11dcSJohannes Doerfert 
177875dfaa1dSTobias Grosser     // Get the reloaded value.
177975dfaa1dSTobias Grosser     OpCopy = getNewValue(Stmt, PHI, BBCopyMap, LTS, getLoopForStmt(Stmt));
1780ecff11dcSJohannes Doerfert   }
1781ecff11dcSJohannes Doerfert 
1782ecff11dcSJohannes Doerfert   assert(OpCopy && "Incoming PHI value was not copied properly");
1783deefbcedSTobias Grosser   PHICopy->addIncoming(OpCopy, BBCopyEnd);
1784ecff11dcSJohannes Doerfert }
1785ecff11dcSJohannes Doerfert 
17862b809d13STobias Grosser void RegionGenerator::copyPHIInstruction(ScopStmt &Stmt, PHINode *PHI,
1787ecff11dcSJohannes Doerfert                                          ValueMapT &BBMap,
1788ecff11dcSJohannes Doerfert                                          LoopToScevMapT &LTS) {
1789ecff11dcSJohannes Doerfert   unsigned NumIncoming = PHI->getNumIncomingValues();
1790ecff11dcSJohannes Doerfert   PHINode *PHICopy =
1791ecff11dcSJohannes Doerfert       Builder.CreatePHI(PHI->getType(), NumIncoming, "polly." + PHI->getName());
1792ecff11dcSJohannes Doerfert   PHICopy->moveBefore(PHICopy->getParent()->getFirstNonPHI());
1793ecff11dcSJohannes Doerfert   BBMap[PHI] = PHICopy;
1794ecff11dcSJohannes Doerfert 
179597b84909STobias Grosser   for (BasicBlock *IncomingBB : PHI->blocks())
179697b84909STobias Grosser     addOperandToPHI(Stmt, PHI, PHICopy, IncomingBB, LTS);
1797ecff11dcSJohannes Doerfert }
1798