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 <S, 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 <S, 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 <S) { 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 <S, 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 <S, 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 <S, 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 <S, 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 <S, 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 <S, 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 <S, 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 <S, 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 <S, 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 <S, 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 <S, 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 <S, 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 <S, 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 <S, 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 <S, 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 <S, 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 <S) { 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 <S) { 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