13b11a16aSHongbin Zheng //===--- BlockGenerators.cpp - Generate code for statements -----*- C++ -*-===// 23b11a16aSHongbin Zheng // 3*2946cd70SChandler Carruth // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. 4*2946cd70SChandler Carruth // See https://llvm.org/LICENSE.txt for license information. 5*2946cd70SChandler 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" 1683628182STobias Grosser #include "polly/CodeGen/CodeGeneration.h" 17a63b2579SJohannes Doerfert #include "polly/CodeGen/IslExprBuilder.h" 1886bc93a9STobias Grosser #include "polly/CodeGen/RuntimeDebugBuilder.h" 19637bd631STobias Grosser #include "polly/Options.h" 205624d3c9STobias Grosser #include "polly/ScopInfo.h" 213b11a16aSHongbin Zheng #include "polly/Support/GICHelper.h" 2297cb813cSSebastian Pop #include "polly/Support/SCEVValidator.h" 23ecfe21b7STobias Grosser #include "polly/Support/ScopHelper.h" 24eedae763SMichael Kruse #include "polly/Support/VirtualInstruction.h" 25e71c6ab5STobias Grosser #include "llvm/Analysis/LoopInfo.h" 26f32d651dSJohannes Doerfert #include "llvm/Analysis/RegionInfo.h" 27e71c6ab5STobias Grosser #include "llvm/Analysis/ScalarEvolution.h" 28030237d0STobias Grosser #include "llvm/IR/IntrinsicInst.h" 29c9895067STobias Grosser #include "llvm/IR/Module.h" 303b11a16aSHongbin Zheng #include "llvm/Transforms/Utils/BasicBlockUtils.h" 31a998f98bSTobias Grosser #include "llvm/Transforms/Utils/Local.h" 32f32d651dSJohannes Doerfert #include "isl/aff.h" 33f32d651dSJohannes Doerfert #include "isl/ast.h" 34f32d651dSJohannes Doerfert #include "isl/ast_build.h" 35ba0d0922STobias Grosser #include "isl/set.h" 36f32d651dSJohannes Doerfert #include <deque> 37f32d651dSJohannes Doerfert 383b11a16aSHongbin Zheng using namespace llvm; 393b11a16aSHongbin Zheng using namespace polly; 403b11a16aSHongbin Zheng 41878aba49STobias Grosser static cl::opt<bool> Aligned("enable-polly-aligned", 42878aba49STobias Grosser cl::desc("Assumed aligned memory accesses."), 43878aba49STobias Grosser cl::Hidden, cl::init(false), cl::ZeroOrMore, 44878aba49STobias Grosser cl::cat(PollyCategory)); 453b11a16aSHongbin Zheng 461be726a4STobias Grosser bool PollyDebugPrinting; 471be726a4STobias Grosser static cl::opt<bool, true> DebugPrintingX( 4886bc93a9STobias Grosser "polly-codegen-add-debug-printing", 4986bc93a9STobias Grosser cl::desc("Add printf calls that show the values loaded/stored."), 501be726a4STobias Grosser cl::location(PollyDebugPrinting), cl::Hidden, cl::init(false), 511be726a4STobias Grosser cl::ZeroOrMore, cl::cat(PollyCategory)); 5286bc93a9STobias Grosser 53e819fffeSMichael Kruse static cl::opt<bool> TraceStmts( 54e819fffeSMichael Kruse "polly-codegen-trace-stmts", 55e819fffeSMichael Kruse cl::desc("Add printf calls that print the statement being executed"), 56e819fffeSMichael Kruse cl::Hidden, cl::init(false), cl::ZeroOrMore, cl::cat(PollyCategory)); 57e819fffeSMichael Kruse 58e819fffeSMichael Kruse static cl::opt<bool> TraceScalars( 59e819fffeSMichael Kruse "polly-codegen-trace-scalars", 60e819fffeSMichael Kruse cl::desc("Add printf calls that print the values of all scalar values " 61e819fffeSMichael Kruse "used in a statement. Requires -polly-codegen-trace-stmts."), 62e819fffeSMichael Kruse cl::Hidden, cl::init(false), cl::ZeroOrMore, cl::cat(PollyCategory)); 63e819fffeSMichael Kruse 64acf80064SEli Friedman BlockGenerator::BlockGenerator( 65acf80064SEli Friedman PollyIRBuilder &B, LoopInfo &LI, ScalarEvolution &SE, DominatorTree &DT, 66587f1f57STobias Grosser AllocaMapTy &ScalarMap, EscapeUsersAllocaMapTy &EscapeMap, 67587f1f57STobias Grosser ValueMapT &GlobalMap, IslExprBuilder *ExprBuilder, BasicBlock *StartBlock) 68ecff11dcSJohannes Doerfert : Builder(B), LI(LI), SE(SE), ExprBuilder(ExprBuilder), DT(DT), 69587f1f57STobias Grosser EntryBB(nullptr), ScalarMap(ScalarMap), EscapeMap(EscapeMap), 70587f1f57STobias Grosser GlobalMap(GlobalMap), StartBlock(StartBlock) {} 71e71c6ab5STobias Grosser 722f1acac6STobias Grosser Value *BlockGenerator::trySynthesizeNewValue(ScopStmt &Stmt, Value *Old, 7333cb9f94STobias Grosser ValueMapT &BBMap, 7433cb9f94STobias Grosser LoopToScevMapT <S, 75bc132607STobias Grosser Loop *L) const { 7642df8d1dSJohannes Doerfert if (!SE.isSCEVable(Old->getType())) 7742df8d1dSJohannes Doerfert return nullptr; 7842df8d1dSJohannes Doerfert 7942df8d1dSJohannes Doerfert const SCEV *Scev = SE.getSCEVAtScope(Old, L); 8042df8d1dSJohannes Doerfert if (!Scev) 8142df8d1dSJohannes Doerfert return nullptr; 8242df8d1dSJohannes Doerfert 8342df8d1dSJohannes Doerfert if (isa<SCEVCouldNotCompute>(Scev)) 8442df8d1dSJohannes Doerfert return nullptr; 8542df8d1dSJohannes Doerfert 8603bcb910SSanjoy Das const SCEV *NewScev = SCEVLoopAddRecRewriter::rewrite(Scev, LTS, SE); 8709e3697fSJohannes Doerfert ValueMapT VTV; 88637b23dcSSebastian Pop VTV.insert(BBMap.begin(), BBMap.end()); 89637b23dcSSebastian Pop VTV.insert(GlobalMap.begin(), GlobalMap.end()); 90e69e1141SJohannes Doerfert 91e69e1141SJohannes Doerfert Scop &S = *Stmt.getParent(); 923f52e354SJohannes Doerfert const DataLayout &DL = S.getFunction().getParent()->getDataLayout(); 93e69e1141SJohannes Doerfert auto IP = Builder.GetInsertPoint(); 94e69e1141SJohannes Doerfert 95e69e1141SJohannes Doerfert assert(IP != Builder.GetInsertBlock()->end() && 9645e7944bSTobias Grosser "Only instructions can be insert points for SCEVExpander"); 9742df8d1dSJohannes Doerfert Value *Expanded = 98acf80064SEli Friedman expandCodeFor(S, SE, DL, "polly", NewScev, Old->getType(), &*IP, &VTV, 99acf80064SEli Friedman StartBlock->getSinglePredecessor()); 100e71c6ab5STobias Grosser 101e71c6ab5STobias Grosser BBMap[Old] = Expanded; 102e71c6ab5STobias Grosser return Expanded; 103e71c6ab5STobias Grosser } 10433cb9f94STobias Grosser 1052f1acac6STobias Grosser Value *BlockGenerator::getNewValue(ScopStmt &Stmt, Value *Old, ValueMapT &BBMap, 1062f1acac6STobias Grosser LoopToScevMapT <S, Loop *L) const { 10733cb9f94STobias Grosser 108eedae763SMichael Kruse auto lookupGlobally = [this](Value *Old) -> Value * { 109eedae763SMichael Kruse Value *New = GlobalMap.lookup(Old); 110eedae763SMichael Kruse if (!New) 111eedae763SMichael Kruse return nullptr; 11228f8ac1dSJohannes Doerfert 113eedae763SMichael Kruse // Required by: 114eedae763SMichael Kruse // * Isl/CodeGen/OpenMP/invariant_base_pointer_preloaded.ll 115eedae763SMichael Kruse // * Isl/CodeGen/OpenMP/invariant_base_pointer_preloaded_different_bb.ll 116eedae763SMichael Kruse // * Isl/CodeGen/OpenMP/invariant_base_pointer_preloaded_pass_only_needed.ll 117eedae763SMichael Kruse // * Isl/CodeGen/OpenMP/invariant_base_pointers_preloaded.ll 118eedae763SMichael Kruse // * Isl/CodeGen/OpenMP/loop-body-references-outer-values-3.ll 119eedae763SMichael Kruse // * Isl/CodeGen/OpenMP/single_loop_with_loop_invariant_baseptr.ll 120eedae763SMichael Kruse // GlobalMap should be a mapping from (value in original SCoP) to (copied 121eedae763SMichael Kruse // value in generated SCoP), without intermediate mappings, which might 122eedae763SMichael Kruse // easily require transitiveness as well. 12333cb9f94STobias Grosser if (Value *NewRemapped = GlobalMap.lookup(New)) 12433cb9f94STobias Grosser New = NewRemapped; 125eedae763SMichael Kruse 126eedae763SMichael Kruse // No test case for this code. 12733cb9f94STobias Grosser if (Old->getType()->getScalarSizeInBits() < 12833cb9f94STobias Grosser New->getType()->getScalarSizeInBits()) 12933cb9f94STobias Grosser New = Builder.CreateTruncOrBitCast(New, Old->getType()); 13033cb9f94STobias Grosser 13133cb9f94STobias Grosser return New; 132eedae763SMichael Kruse }; 133eedae763SMichael Kruse 134eedae763SMichael Kruse Value *New = nullptr; 135eedae763SMichael Kruse auto VUse = VirtualUse::create(&Stmt, L, Old, true); 136eedae763SMichael Kruse switch (VUse.getKind()) { 137eedae763SMichael Kruse case VirtualUse::Block: 138eedae763SMichael Kruse // BasicBlock are constants, but the BlockGenerator copies them. 139eedae763SMichael Kruse New = BBMap.lookup(Old); 140eedae763SMichael Kruse break; 141eedae763SMichael Kruse 142eedae763SMichael Kruse case VirtualUse::Constant: 143eedae763SMichael Kruse // Used by: 144eedae763SMichael Kruse // * Isl/CodeGen/OpenMP/reference-argument-from-non-affine-region.ll 145eedae763SMichael Kruse // Constants should not be redefined. In this case, the GlobalMap just 146eedae763SMichael Kruse // contains a mapping to the same constant, which is unnecessary, but 147eedae763SMichael Kruse // harmless. 148eedae763SMichael Kruse if ((New = lookupGlobally(Old))) 149eedae763SMichael Kruse break; 150eedae763SMichael Kruse 151eedae763SMichael Kruse assert(!BBMap.count(Old)); 152eedae763SMichael Kruse New = Old; 153eedae763SMichael Kruse break; 154eedae763SMichael Kruse 155eedae763SMichael Kruse case VirtualUse::ReadOnly: 156eedae763SMichael Kruse assert(!GlobalMap.count(Old)); 157eedae763SMichael Kruse 158eedae763SMichael Kruse // Required for: 159eedae763SMichael Kruse // * Isl/CodeGen/MemAccess/create_arrays.ll 160eedae763SMichael Kruse // * Isl/CodeGen/read-only-scalars.ll 161eedae763SMichael Kruse // * ScheduleOptimizer/pattern-matching-based-opts_10.ll 162eedae763SMichael Kruse // For some reason these reload a read-only value. The reloaded value ends 163eedae763SMichael Kruse // up in BBMap, buts its value should be identical. 164eedae763SMichael Kruse // 165eedae763SMichael Kruse // Required for: 166eedae763SMichael Kruse // * Isl/CodeGen/OpenMP/single_loop_with_param.ll 167eedae763SMichael Kruse // The parallel subfunctions need to reference the read-only value from the 168eedae763SMichael Kruse // parent function, this is done by reloading them locally. 169eedae763SMichael Kruse if ((New = BBMap.lookup(Old))) 170eedae763SMichael Kruse break; 171eedae763SMichael Kruse 172eedae763SMichael Kruse New = Old; 173eedae763SMichael Kruse break; 174eedae763SMichael Kruse 175eedae763SMichael Kruse case VirtualUse::Synthesizable: 176eedae763SMichael Kruse // Used by: 177eedae763SMichael Kruse // * Isl/CodeGen/OpenMP/loop-body-references-outer-values-3.ll 178eedae763SMichael Kruse // * Isl/CodeGen/OpenMP/recomputed-srem.ll 179eedae763SMichael Kruse // * Isl/CodeGen/OpenMP/reference-other-bb.ll 180eedae763SMichael Kruse // * Isl/CodeGen/OpenMP/two-parallel-loops-reference-outer-indvar.ll 181eedae763SMichael Kruse // For some reason synthesizable values end up in GlobalMap. Their values 182eedae763SMichael Kruse // are the same as trySynthesizeNewValue would return. The legacy 183eedae763SMichael Kruse // implementation prioritized GlobalMap, so this is what we do here as well. 184eedae763SMichael Kruse // Ideally, synthesizable values should not end up in GlobalMap. 185eedae763SMichael Kruse if ((New = lookupGlobally(Old))) 186eedae763SMichael Kruse break; 187eedae763SMichael Kruse 188eedae763SMichael Kruse // Required for: 189eedae763SMichael Kruse // * Isl/CodeGen/RuntimeDebugBuilder/combine_different_values.ll 190eedae763SMichael Kruse // * Isl/CodeGen/getNumberOfIterations.ll 191eedae763SMichael Kruse // * Isl/CodeGen/non_affine_float_compare.ll 192eedae763SMichael Kruse // * ScheduleOptimizer/pattern-matching-based-opts_10.ll 193eedae763SMichael Kruse // Ideally, synthesizable values are synthesized by trySynthesizeNewValue, 194eedae763SMichael Kruse // not precomputed (SCEVExpander has its own caching mechanism). 195eedae763SMichael Kruse // These tests fail without this, but I think trySynthesizeNewValue would 196eedae763SMichael Kruse // just re-synthesize the same instructions. 197eedae763SMichael Kruse if ((New = BBMap.lookup(Old))) 198eedae763SMichael Kruse break; 199eedae763SMichael Kruse 200eedae763SMichael Kruse New = trySynthesizeNewValue(Stmt, Old, BBMap, LTS, L); 201eedae763SMichael Kruse break; 202eedae763SMichael Kruse 203eedae763SMichael Kruse case VirtualUse::Hoisted: 204eedae763SMichael Kruse // TODO: Hoisted invariant loads should be found in GlobalMap only, but not 205eedae763SMichael Kruse // redefined locally (which will be ignored anyway). That is, the following 206eedae763SMichael Kruse // assertion should apply: assert(!BBMap.count(Old)) 207eedae763SMichael Kruse 208eedae763SMichael Kruse New = lookupGlobally(Old); 209eedae763SMichael Kruse break; 210eedae763SMichael Kruse 211eedae763SMichael Kruse case VirtualUse::Intra: 212eedae763SMichael Kruse case VirtualUse::Inter: 213eedae763SMichael Kruse assert(!GlobalMap.count(Old) && 214eedae763SMichael Kruse "Intra and inter-stmt values are never global"); 215eedae763SMichael Kruse New = BBMap.lookup(Old); 216eedae763SMichael Kruse break; 21733cb9f94STobias Grosser } 218eedae763SMichael Kruse assert(New && "Unexpected scalar dependence in region!"); 21933cb9f94STobias Grosser return New; 2203b11a16aSHongbin Zheng } 2213b11a16aSHongbin Zheng 2222f1acac6STobias Grosser void BlockGenerator::copyInstScalar(ScopStmt &Stmt, Instruction *Inst, 223bc132607STobias Grosser ValueMapT &BBMap, LoopToScevMapT <S) { 224030237d0STobias Grosser // We do not generate debug intrinsics as we did not investigate how to 225030237d0STobias Grosser // copy them correctly. At the current state, they just crash the code 226030237d0STobias Grosser // generation as the meta-data operands are not correctly copied. 227030237d0STobias Grosser if (isa<DbgInfoIntrinsic>(Inst)) 228030237d0STobias Grosser return; 229030237d0STobias Grosser 2303b11a16aSHongbin Zheng Instruction *NewInst = Inst->clone(); 2313b11a16aSHongbin Zheng 2323b11a16aSHongbin Zheng // Replace old operands with the new ones. 23391f5b262STobias Grosser for (Value *OldOperand : Inst->operands()) { 234bc132607STobias Grosser Value *NewOperand = 2352c3ffc04SJohannes Doerfert getNewValue(Stmt, OldOperand, BBMap, LTS, getLoopForStmt(Stmt)); 2363b11a16aSHongbin Zheng 2373b11a16aSHongbin Zheng if (!NewOperand) { 238c14582f2STobias Grosser assert(!isa<StoreInst>(NewInst) && 239c14582f2STobias Grosser "Store instructions are always needed!"); 24096ab8726SReid Kleckner NewInst->deleteValue(); 2413b11a16aSHongbin Zheng return; 2423b11a16aSHongbin Zheng } 2433b11a16aSHongbin Zheng 2443b11a16aSHongbin Zheng NewInst->replaceUsesOfWith(OldOperand, NewOperand); 2453b11a16aSHongbin Zheng } 2463b11a16aSHongbin Zheng 2473b11a16aSHongbin Zheng Builder.Insert(NewInst); 2483b11a16aSHongbin Zheng BBMap[Inst] = NewInst; 2493b11a16aSHongbin Zheng 250188053afSSingapuram Sanjay Srivallabh // When copying the instruction onto the Module meant for the GPU, 251188053afSSingapuram Sanjay Srivallabh // debug metadata attached to an instruction causes all related 252188053afSSingapuram Sanjay Srivallabh // metadata to be pulled into the Module. This includes the DICompileUnit, 253188053afSSingapuram Sanjay Srivallabh // which will not be listed in llvm.dbg.cu of the Module since the Module 254188053afSSingapuram Sanjay Srivallabh // doesn't contain one. This fails the verification of the Module and the 255188053afSSingapuram Sanjay Srivallabh // subsequent generation of the ASM string. 256188053afSSingapuram Sanjay Srivallabh if (NewInst->getModule() != Inst->getModule()) 257188053afSSingapuram Sanjay Srivallabh NewInst->setDebugLoc(llvm::DebugLoc()); 258188053afSSingapuram Sanjay Srivallabh 2593b11a16aSHongbin Zheng if (!NewInst->getType()->isVoidTy()) 2603b11a16aSHongbin Zheng NewInst->setName("p_" + Inst->getName()); 2613b11a16aSHongbin Zheng } 2623b11a16aSHongbin Zheng 26370131d34SMichael Kruse Value * 26470131d34SMichael Kruse BlockGenerator::generateLocationAccessed(ScopStmt &Stmt, MemAccInst Inst, 26570131d34SMichael Kruse ValueMapT &BBMap, LoopToScevMapT <S, 26670131d34SMichael Kruse isl_id_to_ast_expr *NewAccesses) { 26709214772STobias Grosser const MemoryAccess &MA = Stmt.getArrayAccessFor(Inst); 2682fa35194SMichael Kruse return generateLocationAccessed( 2692fa35194SMichael Kruse Stmt, getLoopForStmt(Stmt), 2702fa35194SMichael Kruse Inst.isNull() ? nullptr : Inst.getPointerOperand(), BBMap, LTS, 271fe46c3ffSTobias Grosser NewAccesses, MA.getId().release(), MA.getAccessValue()->getType()); 2722fa35194SMichael Kruse } 2733b11a16aSHongbin Zheng 2742fa35194SMichael Kruse Value *BlockGenerator::generateLocationAccessed( 2752fa35194SMichael Kruse ScopStmt &Stmt, Loop *L, Value *Pointer, ValueMapT &BBMap, 2762fa35194SMichael Kruse LoopToScevMapT <S, isl_id_to_ast_expr *NewAccesses, __isl_take isl_id *Id, 2772fa35194SMichael Kruse Type *ExpectedType) { 2782fa35194SMichael Kruse isl_ast_expr *AccessExpr = isl_id_to_ast_expr_get(NewAccesses, Id); 2793b11a16aSHongbin Zheng 2802d1ed0bfSTobias Grosser if (AccessExpr) { 2812d1ed0bfSTobias Grosser AccessExpr = isl_ast_expr_address_of(AccessExpr); 28298b3ee50STobias Grosser auto Address = ExprBuilder->create(AccessExpr); 28398b3ee50STobias Grosser 28498b3ee50STobias Grosser // Cast the address of this memory access to a pointer type that has the 28598b3ee50STobias Grosser // same element type as the original access, but uses the address space of 28698b3ee50STobias Grosser // the newly generated pointer. 2872fa35194SMichael Kruse auto OldPtrTy = ExpectedType->getPointerTo(); 28898b3ee50STobias Grosser auto NewPtrTy = Address->getType(); 28998b3ee50STobias Grosser OldPtrTy = PointerType::get(OldPtrTy->getElementType(), 29098b3ee50STobias Grosser NewPtrTy->getPointerAddressSpace()); 29198b3ee50STobias Grosser 292d840fc72STobias Grosser if (OldPtrTy != NewPtrTy) 29398b3ee50STobias Grosser Address = Builder.CreateBitOrPointerCast(Address, OldPtrTy); 29498b3ee50STobias Grosser return Address; 2952d1ed0bfSTobias Grosser } 2962fa35194SMichael Kruse assert( 2972fa35194SMichael Kruse Pointer && 2982fa35194SMichael Kruse "If expression was not generated, must use the original pointer value"); 2992fa35194SMichael Kruse return getNewValue(Stmt, Pointer, BBMap, LTS, L); 3002fa35194SMichael Kruse } 3012d1ed0bfSTobias Grosser 3022fa35194SMichael Kruse Value * 3032fa35194SMichael Kruse BlockGenerator::getImplicitAddress(MemoryAccess &Access, Loop *L, 3042fa35194SMichael Kruse LoopToScevMapT <S, ValueMapT &BBMap, 3052fa35194SMichael Kruse __isl_keep isl_id_to_ast_expr *NewAccesses) { 3062fa35194SMichael Kruse if (Access.isLatestArrayKind()) 3072fa35194SMichael Kruse return generateLocationAccessed(*Access.getStatement(), L, nullptr, BBMap, 308fe46c3ffSTobias Grosser LTS, NewAccesses, Access.getId().release(), 3092fa35194SMichael Kruse Access.getAccessValue()->getType()); 3102fa35194SMichael Kruse 311587f1f57STobias Grosser return getOrCreateAlloca(Access); 3123b11a16aSHongbin Zheng } 3133b11a16aSHongbin Zheng 3142c3ffc04SJohannes Doerfert Loop *BlockGenerator::getLoopForStmt(const ScopStmt &Stmt) const { 315375cb5feSMichael Kruse auto *StmtBB = Stmt.getEntryBlock(); 3162c3ffc04SJohannes Doerfert return LI.getLoopFor(StmtBB); 317369430ffSTobias Grosser } 318369430ffSTobias Grosser 319888ab551SMichael Kruse Value *BlockGenerator::generateArrayLoad(ScopStmt &Stmt, LoadInst *Load, 320bc132607STobias Grosser ValueMapT &BBMap, LoopToScevMapT <S, 3212d1ed0bfSTobias Grosser isl_id_to_ast_expr *NewAccesses) { 322c1db67e2SJohannes Doerfert if (Value *PreloadLoad = GlobalMap.lookup(Load)) 323c1db67e2SJohannes Doerfert return PreloadLoad; 324c1db67e2SJohannes Doerfert 325bc132607STobias Grosser Value *NewPointer = 32670131d34SMichael Kruse generateLocationAccessed(Stmt, Load, BBMap, LTS, NewAccesses); 32787901453SJohannes Doerfert Value *ScalarLoad = Builder.CreateAlignedLoad( 32887901453SJohannes Doerfert NewPointer, Load->getAlignment(), Load->getName() + "_p_scalar_"); 32986bc93a9STobias Grosser 3301be726a4STobias Grosser if (PollyDebugPrinting) 33186bc93a9STobias Grosser RuntimeDebugBuilder::createCPUPrinter(Builder, "Load from ", NewPointer, 33286bc93a9STobias Grosser ": ", ScalarLoad, "\n"); 33386bc93a9STobias Grosser 3343b11a16aSHongbin Zheng return ScalarLoad; 3353b11a16aSHongbin Zheng } 3363b11a16aSHongbin Zheng 337888ab551SMichael Kruse void BlockGenerator::generateArrayStore(ScopStmt &Stmt, StoreInst *Store, 338bc132607STobias Grosser ValueMapT &BBMap, LoopToScevMapT <S, 3392d1ed0bfSTobias Grosser isl_id_to_ast_expr *NewAccesses) { 340706f79abSMichael Kruse MemoryAccess &MA = Stmt.getArrayAccessFor(Store); 3411515f6b9STobias Grosser isl::set AccDom = MA.getAccessRelation().domain(); 342fe46c3ffSTobias Grosser std::string Subject = MA.getId().get_name(); 343706f79abSMichael Kruse 344fe46c3ffSTobias Grosser generateConditionalExecution(Stmt, AccDom, Subject.c_str(), [&, this]() { 345bc132607STobias Grosser Value *NewPointer = 34670131d34SMichael Kruse generateLocationAccessed(Stmt, Store, BBMap, LTS, NewAccesses); 347706f79abSMichael Kruse Value *ValueOperand = getNewValue(Stmt, Store->getValueOperand(), BBMap, 348706f79abSMichael Kruse LTS, getLoopForStmt(Stmt)); 3493b11a16aSHongbin Zheng 3501be726a4STobias Grosser if (PollyDebugPrinting) 35186bc93a9STobias Grosser RuntimeDebugBuilder::createCPUPrinter(Builder, "Store to ", NewPointer, 35286bc93a9STobias Grosser ": ", ValueOperand, "\n"); 35386bc93a9STobias Grosser 354c186ac7aSTobias Grosser Builder.CreateAlignedStore(ValueOperand, NewPointer, Store->getAlignment()); 355706f79abSMichael Kruse }); 3563b11a16aSHongbin Zheng } 3573b11a16aSHongbin Zheng 3585dced269SJohannes Doerfert bool BlockGenerator::canSyntheziseInStmt(ScopStmt &Stmt, Instruction *Inst) { 3592c3ffc04SJohannes Doerfert Loop *L = getLoopForStmt(Stmt); 3605dced269SJohannes Doerfert return (Stmt.isBlockStmt() || !Stmt.getRegion()->contains(L)) && 36111c5e079SMichael Kruse canSynthesize(Inst, *Stmt.getParent(), &SE, L); 3625dced269SJohannes Doerfert } 3635dced269SJohannes Doerfert 3642f1acac6STobias Grosser void BlockGenerator::copyInstruction(ScopStmt &Stmt, Instruction *Inst, 365bc132607STobias Grosser ValueMapT &BBMap, LoopToScevMapT <S, 3662d1ed0bfSTobias Grosser isl_id_to_ast_expr *NewAccesses) { 3673b11a16aSHongbin Zheng // Terminator instructions control the control flow. They are explicitly 3683b11a16aSHongbin Zheng // expressed in the clast and do not need to be copied. 3693b11a16aSHongbin Zheng if (Inst->isTerminator()) 3703b11a16aSHongbin Zheng return; 3713b11a16aSHongbin Zheng 372aff56c8aSTobias Grosser // Synthesizable statements will be generated on-demand. 3735dced269SJohannes Doerfert if (canSyntheziseInStmt(Stmt, Inst)) 374e71c6ab5STobias Grosser return; 375e71c6ab5STobias Grosser 3769646e3feSTobias Grosser if (auto *Load = dyn_cast<LoadInst>(Inst)) { 377888ab551SMichael Kruse Value *NewLoad = generateArrayLoad(Stmt, Load, BBMap, LTS, NewAccesses); 3783d94fedfSSebastian Pop // Compute NewLoad before its insertion in BBMap to make the insertion 3793d94fedfSSebastian Pop // deterministic. 380753d43f9SSebastian Pop BBMap[Load] = NewLoad; 3813b11a16aSHongbin Zheng return; 3823b11a16aSHongbin Zheng } 3833b11a16aSHongbin Zheng 3849646e3feSTobias Grosser if (auto *Store = dyn_cast<StoreInst>(Inst)) { 3850446d81eSMichael Kruse // Identified as redundant by -polly-simplify. 3860446d81eSMichael Kruse if (!Stmt.getArrayAccessOrNULLFor(Store)) 3870446d81eSMichael Kruse return; 3880446d81eSMichael Kruse 389888ab551SMichael Kruse generateArrayStore(Stmt, Store, BBMap, LTS, NewAccesses); 3903b11a16aSHongbin Zheng return; 3913b11a16aSHongbin Zheng } 3923b11a16aSHongbin Zheng 3939646e3feSTobias Grosser if (auto *PHI = dyn_cast<PHINode>(Inst)) { 394bc132607STobias Grosser copyPHIInstruction(Stmt, PHI, BBMap, LTS); 395ecff11dcSJohannes Doerfert return; 396ecff11dcSJohannes Doerfert } 397ecff11dcSJohannes Doerfert 3983f500fa2SJohannes Doerfert // Skip some special intrinsics for which we do not adjust the semantics to 3993f500fa2SJohannes Doerfert // the new schedule. All others are handled like every other instruction. 4009c0ffe3aSTobias Grosser if (isIgnoredIntrinsic(Inst)) 4013f500fa2SJohannes Doerfert return; 4023f500fa2SJohannes Doerfert 403bc132607STobias Grosser copyInstScalar(Stmt, Inst, BBMap, LTS); 4043b11a16aSHongbin Zheng } 4053b11a16aSHongbin Zheng 4069d12d8adSTobias Grosser void BlockGenerator::removeDeadInstructions(BasicBlock *BB, ValueMapT &BBMap) { 407c59b3ce0STobias Grosser auto NewBB = Builder.GetInsertBlock(); 408c59b3ce0STobias Grosser for (auto I = NewBB->rbegin(); I != NewBB->rend(); I++) { 409c59b3ce0STobias Grosser Instruction *NewInst = &*I; 4109d12d8adSTobias Grosser 4119d12d8adSTobias Grosser if (!isInstructionTriviallyDead(NewInst)) 4129d12d8adSTobias Grosser continue; 4139d12d8adSTobias Grosser 414c59b3ce0STobias Grosser for (auto Pair : BBMap) 415c59b3ce0STobias Grosser if (Pair.second == NewInst) { 416c59b3ce0STobias Grosser BBMap.erase(Pair.first); 417c59b3ce0STobias Grosser } 418c59b3ce0STobias Grosser 4199d12d8adSTobias Grosser NewInst->eraseFromParent(); 420c59b3ce0STobias Grosser I = NewBB->rbegin(); 4219d12d8adSTobias Grosser } 4229d12d8adSTobias Grosser } 4239d12d8adSTobias Grosser 424bc132607STobias Grosser void BlockGenerator::copyStmt(ScopStmt &Stmt, LoopToScevMapT <S, 4252d1ed0bfSTobias Grosser isl_id_to_ast_expr *NewAccesses) { 426275a1756SJohannes Doerfert assert(Stmt.isBlockStmt() && 427275a1756SJohannes Doerfert "Only block statements can be copied by the block generator"); 428275a1756SJohannes Doerfert 429275a1756SJohannes Doerfert ValueMapT BBMap; 430275a1756SJohannes Doerfert 431be9c9117SJohannes Doerfert BasicBlock *BB = Stmt.getBasicBlock(); 432bc132607STobias Grosser copyBB(Stmt, BB, BBMap, LTS, NewAccesses); 4339d12d8adSTobias Grosser removeDeadInstructions(BB, BBMap); 434275a1756SJohannes Doerfert } 435275a1756SJohannes Doerfert 436514f6efaSJohannes Doerfert BasicBlock *BlockGenerator::splitBB(BasicBlock *BB) { 437b8f58b53SDuncan P. N. Exon Smith BasicBlock *CopyBB = SplitBlock(Builder.GetInsertBlock(), 438b8f58b53SDuncan P. N. Exon Smith &*Builder.GetInsertPoint(), &DT, &LI); 4393b11a16aSHongbin Zheng CopyBB->setName("polly.stmt." + BB->getName()); 440514f6efaSJohannes Doerfert return CopyBB; 441514f6efaSJohannes Doerfert } 4423b11a16aSHongbin Zheng 443514f6efaSJohannes Doerfert BasicBlock *BlockGenerator::copyBB(ScopStmt &Stmt, BasicBlock *BB, 444bc132607STobias Grosser ValueMapT &BBMap, LoopToScevMapT <S, 4452d1ed0bfSTobias Grosser isl_id_to_ast_expr *NewAccesses) { 446514f6efaSJohannes Doerfert BasicBlock *CopyBB = splitBB(BB); 447b8f58b53SDuncan P. N. Exon Smith Builder.SetInsertPoint(&CopyBB->front()); 4482fa35194SMichael Kruse generateScalarLoads(Stmt, LTS, BBMap, NewAccesses); 449e819fffeSMichael Kruse generateBeginStmtTrace(Stmt, LTS, BBMap); 450225f0d1eSMichael Kruse 451bc132607STobias Grosser copyBB(Stmt, BB, CopyBB, BBMap, LTS, NewAccesses); 452225f0d1eSMichael Kruse 453225f0d1eSMichael Kruse // After a basic block was copied store all scalars that escape this block in 454225f0d1eSMichael Kruse // their alloca. 4552fa35194SMichael Kruse generateScalarStores(Stmt, LTS, BBMap, NewAccesses); 456514f6efaSJohannes Doerfert return CopyBB; 457514f6efaSJohannes Doerfert } 458514f6efaSJohannes Doerfert 459514f6efaSJohannes Doerfert void BlockGenerator::copyBB(ScopStmt &Stmt, BasicBlock *BB, BasicBlock *CopyBB, 460bc132607STobias Grosser ValueMapT &BBMap, LoopToScevMapT <S, 4612d1ed0bfSTobias Grosser isl_id_to_ast_expr *NewAccesses) { 462ecff11dcSJohannes Doerfert EntryBB = &CopyBB->getParent()->getEntryBlock(); 463ecff11dcSJohannes Doerfert 464c43d0360STobias Grosser // Block statements and the entry blocks of region statement are code 465c43d0360STobias Grosser // generated from instruction lists. This allow us to optimize the 466c43d0360STobias Grosser // instructions that belong to a certain scop statement. As the code 467c43d0360STobias Grosser // structure of region statements might be arbitrary complex, optimizing the 468c43d0360STobias Grosser // instruction list is not yet supported. 469c43d0360STobias Grosser if (Stmt.isBlockStmt() || (Stmt.isRegionStmt() && Stmt.getEntryBlock() == BB)) 4703bb48299SMichael Kruse for (Instruction *Inst : Stmt.getInstructions()) 4713bb48299SMichael Kruse copyInstruction(Stmt, Inst, BBMap, LTS, NewAccesses); 4723bb48299SMichael Kruse else 47391f5b262STobias Grosser for (Instruction &Inst : *BB) 474bc132607STobias Grosser copyInstruction(Stmt, &Inst, BBMap, LTS, NewAccesses); 475ecff11dcSJohannes Doerfert } 476ecff11dcSJohannes Doerfert 477800e17a7SJohannes Doerfert Value *BlockGenerator::getOrCreateAlloca(const MemoryAccess &Access) { 478587f1f57STobias Grosser assert(!Access.isLatestArrayKind() && "Trying to get alloca for array kind"); 4793216f854STobias Grosser 480587f1f57STobias Grosser return getOrCreateAlloca(Access.getLatestScopArrayInfo()); 481a4f0988dSTobias Grosser } 482a4f0988dSTobias Grosser 483a4f0988dSTobias Grosser Value *BlockGenerator::getOrCreateAlloca(const ScopArrayInfo *Array) { 4843216f854STobias Grosser assert(!Array->isArrayKind() && "Trying to get alloca for array kind"); 4853216f854STobias Grosser 486587f1f57STobias Grosser auto &Addr = ScalarMap[Array]; 487587f1f57STobias Grosser 488587f1f57STobias Grosser if (Addr) { 489587f1f57STobias Grosser // Allow allocas to be (temporarily) redirected once by adding a new 490a6d48f59SMichael Kruse // old-alloca-addr to new-addr mapping to GlobalMap. This functionality 491587f1f57STobias Grosser // is used for example by the OpenMP code generation where a first use 492587f1f57STobias Grosser // of a scalar while still in the host code allocates a normal alloca with 493587f1f57STobias Grosser // getOrCreateAlloca. When the values of this scalar are accessed during 494587f1f57STobias Grosser // the generation of the parallel subfunction, these values are copied over 495587f1f57STobias Grosser // to the parallel subfunction and each request for a scalar alloca slot 496a6d48f59SMichael Kruse // must be forwarded to the temporary in-subfunction slot. This mapping is 497587f1f57STobias Grosser // removed when the subfunction has been generated and again normal host 498682c5114STobias Grosser // code is generated. Due to the following reasons it is not possible to 499682c5114STobias Grosser // perform the GlobalMap lookup right after creating the alloca below, but 500682c5114STobias Grosser // instead we need to check GlobalMap at each call to getOrCreateAlloca: 501682c5114STobias Grosser // 502682c5114STobias Grosser // 1) GlobalMap may be changed multiple times (for each parallel loop), 503682c5114STobias Grosser // 2) The temporary mapping is commonly only known after the initial 504682c5114STobias Grosser // alloca has already been generated, and 505682c5114STobias Grosser // 3) The original alloca value must be restored after leaving the 506682c5114STobias Grosser // sub-function. 507587f1f57STobias Grosser if (Value *NewAddr = GlobalMap.lookup(&*Addr)) 508587f1f57STobias Grosser return NewAddr; 509587f1f57STobias Grosser return Addr; 510587f1f57STobias Grosser } 511587f1f57STobias Grosser 512587f1f57STobias Grosser Type *Ty = Array->getElementType(); 513587f1f57STobias Grosser Value *ScalarBase = Array->getBasePtr(); 514587f1f57STobias Grosser std::string NameExt; 515a535dff4STobias Grosser if (Array->isPHIKind()) 516587f1f57STobias Grosser NameExt = ".phiops"; 517f8d55f7eSTobias Grosser else 518587f1f57STobias Grosser NameExt = ".s2a"; 519587f1f57STobias Grosser 5207b5a4dfdSTobias Grosser const DataLayout &DL = Builder.GetInsertBlock()->getModule()->getDataLayout(); 521b3e30c32SMatt Arsenault 522b3e30c32SMatt Arsenault Addr = new AllocaInst(Ty, DL.getAllocaAddrSpace(), 523b3e30c32SMatt Arsenault ScalarBase->getName() + NameExt); 524587f1f57STobias Grosser EntryBB = &Builder.GetInsertBlock()->getParent()->getEntryBlock(); 525587f1f57STobias Grosser Addr->insertBefore(&*EntryBB->getFirstInsertionPt()); 526587f1f57STobias Grosser 527587f1f57STobias Grosser return Addr; 528f8d55f7eSTobias Grosser } 529f8d55f7eSTobias Grosser 530587f1f57STobias Grosser void BlockGenerator::handleOutsideUsers(const Scop &S, ScopArrayInfo *Array) { 531587f1f57STobias Grosser Instruction *Inst = cast<Instruction>(Array->getBasePtr()); 532b79a67dfSTobias Grosser 5332985400aSTobias Grosser // If there are escape users we get the alloca for this instruction and put it 5342985400aSTobias Grosser // in the EscapeMap for later finalization. Lastly, if the instruction was 5352985400aSTobias Grosser // copied multiple times we already did this and can exit. 536acb6ade7SMichael Kruse if (EscapeMap.count(Inst)) 537acb6ade7SMichael Kruse return; 538e69e1141SJohannes Doerfert 539ecff11dcSJohannes Doerfert EscapeUserVectorTy EscapeUsers; 540ecff11dcSJohannes Doerfert for (User *U : Inst->users()) { 541ecff11dcSJohannes Doerfert 542ecff11dcSJohannes Doerfert // Non-instruction user will never escape. 543ecff11dcSJohannes Doerfert Instruction *UI = dyn_cast<Instruction>(U); 544ecff11dcSJohannes Doerfert if (!UI) 545ecff11dcSJohannes Doerfert continue; 546ecff11dcSJohannes Doerfert 547952b5304SJohannes Doerfert if (S.contains(UI)) 548ecff11dcSJohannes Doerfert continue; 549ecff11dcSJohannes Doerfert 550ecff11dcSJohannes Doerfert EscapeUsers.push_back(UI); 551ecff11dcSJohannes Doerfert } 552ecff11dcSJohannes Doerfert 553ecff11dcSJohannes Doerfert // Exit if no escape uses were found. 554ecff11dcSJohannes Doerfert if (EscapeUsers.empty()) 555ecff11dcSJohannes Doerfert return; 556ecff11dcSJohannes Doerfert 557ecff11dcSJohannes Doerfert // Get or create an escape alloca for this instruction. 558587f1f57STobias Grosser auto *ScalarAddr = getOrCreateAlloca(Array); 559ecff11dcSJohannes Doerfert 560ecff11dcSJohannes Doerfert // Remember that this instruction has escape uses and the escape alloca. 561ecff11dcSJohannes Doerfert EscapeMap[Inst] = std::make_pair(ScalarAddr, std::move(EscapeUsers)); 562ecff11dcSJohannes Doerfert } 563ecff11dcSJohannes Doerfert 5642fa35194SMichael Kruse void BlockGenerator::generateScalarLoads( 5652fa35194SMichael Kruse ScopStmt &Stmt, LoopToScevMapT <S, ValueMapT &BBMap, 5662fa35194SMichael Kruse __isl_keep isl_id_to_ast_expr *NewAccesses) { 567225f0d1eSMichael Kruse for (MemoryAccess *MA : Stmt) { 5682fa35194SMichael Kruse if (MA->isOriginalArrayKind() || MA->isWrite()) 569ecff11dcSJohannes Doerfert continue; 570ecff11dcSJohannes Doerfert 57177394f13SMichael Kruse #ifndef NDEBUG 5724485ae08SMichael Kruse auto StmtDom = 5734485ae08SMichael Kruse Stmt.getDomain().intersect_params(Stmt.getParent()->getContext()); 574388730c9SMichael Kruse auto AccDom = MA->getAccessRelation().domain(); 575388730c9SMichael Kruse assert(!StmtDom.is_subset(AccDom).is_false() && 57677394f13SMichael Kruse "Scalar must be loaded in all statement instances"); 57777394f13SMichael Kruse #endif 57877394f13SMichael Kruse 5792fa35194SMichael Kruse auto *Address = 5802fa35194SMichael Kruse getImplicitAddress(*MA, getLoopForStmt(Stmt), LTS, BBMap, NewAccesses); 581eac9726eSMichael Kruse assert((!isa<Instruction>(Address) || 582eac9726eSMichael Kruse DT.dominates(cast<Instruction>(Address)->getParent(), 583eac9726eSMichael Kruse Builder.GetInsertBlock())) && 584eac9726eSMichael Kruse "Domination violation"); 585583be06fSTobias Grosser BBMap[MA->getAccessValue()] = 5862fc50df9STobias Grosser Builder.CreateLoad(Address, Address->getName() + ".reload"); 587ecff11dcSJohannes Doerfert } 588ecff11dcSJohannes Doerfert } 589ecff11dcSJohannes Doerfert 590706f79abSMichael Kruse Value *BlockGenerator::buildContainsCondition(ScopStmt &Stmt, 591706f79abSMichael Kruse const isl::set &Subdomain) { 592132860afSTobias Grosser isl::ast_build AstBuild = Stmt.getAstBuild(); 593dcf8d696STobias Grosser isl::set Domain = Stmt.getDomain(); 594706f79abSMichael Kruse 5956b6ac900STobias Grosser isl::union_map USchedule = AstBuild.get_schedule(); 5966b6ac900STobias Grosser USchedule = USchedule.intersect_domain(Domain); 597706f79abSMichael Kruse 5986b6ac900STobias Grosser assert(!USchedule.is_empty()); 5996b6ac900STobias Grosser isl::map Schedule = isl::map::from_union_map(USchedule); 600706f79abSMichael Kruse 6016b6ac900STobias Grosser isl::set ScheduledDomain = Schedule.range(); 6026b6ac900STobias Grosser isl::set ScheduledSet = Subdomain.apply(Schedule); 603706f79abSMichael Kruse 6046b6ac900STobias Grosser isl::ast_build RestrictedBuild = AstBuild.restrict(ScheduledDomain); 6056b6ac900STobias Grosser 6066b6ac900STobias Grosser isl::ast_expr IsInSet = RestrictedBuild.expr_from(ScheduledSet); 607706f79abSMichael Kruse Value *IsInSetExpr = ExprBuilder->create(IsInSet.copy()); 608706f79abSMichael Kruse IsInSetExpr = Builder.CreateICmpNE( 609706f79abSMichael Kruse IsInSetExpr, ConstantInt::get(IsInSetExpr->getType(), 0)); 610706f79abSMichael Kruse 611706f79abSMichael Kruse return IsInSetExpr; 612706f79abSMichael Kruse } 613706f79abSMichael Kruse 614706f79abSMichael Kruse void BlockGenerator::generateConditionalExecution( 615706f79abSMichael Kruse ScopStmt &Stmt, const isl::set &Subdomain, StringRef Subject, 616706f79abSMichael Kruse const std::function<void()> &GenThenFunc) { 617dcf8d696STobias Grosser isl::set StmtDom = Stmt.getDomain(); 618706f79abSMichael Kruse 619706f79abSMichael Kruse // If the condition is a tautology, don't generate a condition around the 620706f79abSMichael Kruse // code. 621f51decb5STobias Grosser bool IsPartialWrite = 6228ea1fc19STobias Grosser !StmtDom.intersect_params(Stmt.getParent()->getContext()) 623f51decb5STobias Grosser .is_subset(Subdomain); 624f51decb5STobias Grosser if (!IsPartialWrite) { 625706f79abSMichael Kruse GenThenFunc(); 626706f79abSMichael Kruse return; 627706f79abSMichael Kruse } 628706f79abSMichael Kruse 629706f79abSMichael Kruse // Generate the condition. 630706f79abSMichael Kruse Value *Cond = buildContainsCondition(Stmt, Subdomain); 631b795bfc0SMichael Kruse 632b795bfc0SMichael Kruse // Don't call GenThenFunc if it is never executed. An ast index expression 633b795bfc0SMichael Kruse // might not be defined in this case. 634b795bfc0SMichael Kruse if (auto *Const = dyn_cast<ConstantInt>(Cond)) 635b795bfc0SMichael Kruse if (Const->isZero()) 636b795bfc0SMichael Kruse return; 637b795bfc0SMichael Kruse 638706f79abSMichael Kruse BasicBlock *HeadBlock = Builder.GetInsertBlock(); 639706f79abSMichael Kruse StringRef BlockName = HeadBlock->getName(); 640706f79abSMichael Kruse 641706f79abSMichael Kruse // Generate the conditional block. 642706f79abSMichael Kruse SplitBlockAndInsertIfThen(Cond, &*Builder.GetInsertPoint(), false, nullptr, 643706f79abSMichael Kruse &DT, &LI); 644706f79abSMichael Kruse BranchInst *Branch = cast<BranchInst>(HeadBlock->getTerminator()); 645706f79abSMichael Kruse BasicBlock *ThenBlock = Branch->getSuccessor(0); 646706f79abSMichael Kruse BasicBlock *TailBlock = Branch->getSuccessor(1); 647706f79abSMichael Kruse 648706f79abSMichael Kruse // Assign descriptive names. 649706f79abSMichael Kruse if (auto *CondInst = dyn_cast<Instruction>(Cond)) 650706f79abSMichael Kruse CondInst->setName("polly." + Subject + ".cond"); 651706f79abSMichael Kruse ThenBlock->setName(BlockName + "." + Subject + ".partial"); 652706f79abSMichael Kruse TailBlock->setName(BlockName + ".cont"); 653706f79abSMichael Kruse 654706f79abSMichael Kruse // Put the client code into the conditional block and continue in the merge 655706f79abSMichael Kruse // block afterwards. 656706f79abSMichael Kruse Builder.SetInsertPoint(ThenBlock, ThenBlock->getFirstInsertionPt()); 657706f79abSMichael Kruse GenThenFunc(); 658706f79abSMichael Kruse Builder.SetInsertPoint(TailBlock, TailBlock->getFirstInsertionPt()); 659706f79abSMichael Kruse } 660706f79abSMichael Kruse 661e819fffeSMichael Kruse static std::string getInstName(Value *Val) { 662e819fffeSMichael Kruse std::string Result; 663e819fffeSMichael Kruse raw_string_ostream OS(Result); 664e819fffeSMichael Kruse Val->printAsOperand(OS, false); 665e819fffeSMichael Kruse return OS.str(); 666e819fffeSMichael Kruse } 667e819fffeSMichael Kruse 668e819fffeSMichael Kruse void BlockGenerator::generateBeginStmtTrace(ScopStmt &Stmt, LoopToScevMapT <S, 669e819fffeSMichael Kruse ValueMapT &BBMap) { 670e819fffeSMichael Kruse if (!TraceStmts) 671e819fffeSMichael Kruse return; 672e819fffeSMichael Kruse 673e819fffeSMichael Kruse Scop *S = Stmt.getParent(); 674e819fffeSMichael Kruse const char *BaseName = Stmt.getBaseName(); 675e819fffeSMichael Kruse 676e819fffeSMichael Kruse isl::ast_build AstBuild = Stmt.getAstBuild(); 677e819fffeSMichael Kruse isl::set Domain = Stmt.getDomain(); 678e819fffeSMichael Kruse 679e819fffeSMichael Kruse isl::union_map USchedule = AstBuild.get_schedule().intersect_domain(Domain); 680e819fffeSMichael Kruse isl::map Schedule = isl::map::from_union_map(USchedule); 681e819fffeSMichael Kruse assert(Schedule.is_empty().is_false() && 682e819fffeSMichael Kruse "The stmt must have a valid instance"); 683e819fffeSMichael Kruse 684e819fffeSMichael Kruse isl::multi_pw_aff ScheduleMultiPwAff = 685e819fffeSMichael Kruse isl::pw_multi_aff::from_map(Schedule.reverse()); 686e819fffeSMichael Kruse isl::ast_build RestrictedBuild = AstBuild.restrict(Schedule.range()); 687e819fffeSMichael Kruse 688e819fffeSMichael Kruse // Sequence of strings to print. 689e819fffeSMichael Kruse SmallVector<llvm::Value *, 8> Values; 690e819fffeSMichael Kruse 691e819fffeSMichael Kruse // Print the name of the statement. 692e819fffeSMichael Kruse // TODO: Indent by the depth of the statement instance in the schedule tree. 693e819fffeSMichael Kruse Values.push_back(RuntimeDebugBuilder::getPrintableString(Builder, BaseName)); 694e819fffeSMichael Kruse Values.push_back(RuntimeDebugBuilder::getPrintableString(Builder, "(")); 695e819fffeSMichael Kruse 696e819fffeSMichael Kruse // Add the coordinate of the statement instance. 697e819fffeSMichael Kruse int DomDims = ScheduleMultiPwAff.dim(isl::dim::out); 698e819fffeSMichael Kruse for (int i = 0; i < DomDims; i += 1) { 699e819fffeSMichael Kruse if (i > 0) 700e819fffeSMichael Kruse Values.push_back(RuntimeDebugBuilder::getPrintableString(Builder, ",")); 701e819fffeSMichael Kruse 702e819fffeSMichael Kruse isl::ast_expr IsInSet = 703e819fffeSMichael Kruse RestrictedBuild.expr_from(ScheduleMultiPwAff.get_pw_aff(i)); 704e819fffeSMichael Kruse Values.push_back(ExprBuilder->create(IsInSet.copy())); 705e819fffeSMichael Kruse } 706e819fffeSMichael Kruse 707e819fffeSMichael Kruse if (TraceScalars) { 708e819fffeSMichael Kruse Values.push_back(RuntimeDebugBuilder::getPrintableString(Builder, ")")); 709e819fffeSMichael Kruse DenseSet<Instruction *> Encountered; 710e819fffeSMichael Kruse 711e819fffeSMichael Kruse // Add the value of each scalar (and the result of PHIs) used in the 712e819fffeSMichael Kruse // statement. 713e819fffeSMichael Kruse // TODO: Values used in region-statements. 714e819fffeSMichael Kruse for (Instruction *Inst : Stmt.insts()) { 715e819fffeSMichael Kruse if (!RuntimeDebugBuilder::isPrintable(Inst->getType())) 716e819fffeSMichael Kruse continue; 717e819fffeSMichael Kruse 718e819fffeSMichael Kruse if (isa<PHINode>(Inst)) { 719e819fffeSMichael Kruse Values.push_back(RuntimeDebugBuilder::getPrintableString(Builder, " ")); 720e819fffeSMichael Kruse Values.push_back(RuntimeDebugBuilder::getPrintableString( 721e819fffeSMichael Kruse Builder, getInstName(Inst))); 722e819fffeSMichael Kruse Values.push_back(RuntimeDebugBuilder::getPrintableString(Builder, "=")); 723e819fffeSMichael Kruse Values.push_back(getNewValue(Stmt, Inst, BBMap, LTS, 724e819fffeSMichael Kruse LI.getLoopFor(Inst->getParent()))); 725e819fffeSMichael Kruse } else { 726e819fffeSMichael Kruse for (Value *Op : Inst->operand_values()) { 727e819fffeSMichael Kruse // Do not print values that cannot change during the execution of the 728e819fffeSMichael Kruse // SCoP. 729e819fffeSMichael Kruse auto *OpInst = dyn_cast<Instruction>(Op); 730e819fffeSMichael Kruse if (!OpInst) 731e819fffeSMichael Kruse continue; 732e819fffeSMichael Kruse if (!S->contains(OpInst)) 733e819fffeSMichael Kruse continue; 734e819fffeSMichael Kruse 735e819fffeSMichael Kruse // Print each scalar at most once, and exclude values defined in the 736e819fffeSMichael Kruse // statement itself. 737e819fffeSMichael Kruse if (Encountered.count(OpInst)) 738e819fffeSMichael Kruse continue; 739e819fffeSMichael Kruse 740e819fffeSMichael Kruse Values.push_back( 741e819fffeSMichael Kruse RuntimeDebugBuilder::getPrintableString(Builder, " ")); 742e819fffeSMichael Kruse Values.push_back(RuntimeDebugBuilder::getPrintableString( 743e819fffeSMichael Kruse Builder, getInstName(OpInst))); 744e819fffeSMichael Kruse Values.push_back( 745e819fffeSMichael Kruse RuntimeDebugBuilder::getPrintableString(Builder, "=")); 746e819fffeSMichael Kruse Values.push_back(getNewValue(Stmt, OpInst, BBMap, LTS, 747e819fffeSMichael Kruse LI.getLoopFor(Inst->getParent()))); 748e819fffeSMichael Kruse Encountered.insert(OpInst); 749e819fffeSMichael Kruse } 750e819fffeSMichael Kruse } 751e819fffeSMichael Kruse 752e819fffeSMichael Kruse Encountered.insert(Inst); 753e819fffeSMichael Kruse } 754e819fffeSMichael Kruse 755e819fffeSMichael Kruse Values.push_back(RuntimeDebugBuilder::getPrintableString(Builder, "\n")); 756e819fffeSMichael Kruse } else { 757e819fffeSMichael Kruse Values.push_back(RuntimeDebugBuilder::getPrintableString(Builder, ")\n")); 758e819fffeSMichael Kruse } 759e819fffeSMichael Kruse 760e819fffeSMichael Kruse RuntimeDebugBuilder::createCPUPrinter(Builder, ArrayRef<Value *>(Values)); 761e819fffeSMichael Kruse } 762e819fffeSMichael Kruse 7632fa35194SMichael Kruse void BlockGenerator::generateScalarStores( 7642fa35194SMichael Kruse ScopStmt &Stmt, LoopToScevMapT <S, ValueMapT &BBMap, 7652fa35194SMichael Kruse __isl_keep isl_id_to_ast_expr *NewAccesses) { 766f2cdd144STobias Grosser Loop *L = LI.getLoopFor(Stmt.getBasicBlock()); 767ecff11dcSJohannes Doerfert 76821a059afSTobias Grosser assert(Stmt.isBlockStmt() && 76921a059afSTobias Grosser "Region statements need to use the generateScalarStores() function in " 77021a059afSTobias Grosser "the RegionGenerator"); 771ecff11dcSJohannes Doerfert 772ecff11dcSJohannes Doerfert for (MemoryAccess *MA : Stmt) { 7732fa35194SMichael Kruse if (MA->isOriginalArrayKind() || MA->isRead()) 774ecff11dcSJohannes Doerfert continue; 775ecff11dcSJohannes Doerfert 7761515f6b9STobias Grosser isl::set AccDom = MA->getAccessRelation().domain(); 777fe46c3ffSTobias Grosser std::string Subject = MA->getId().get_name(); 77877394f13SMichael Kruse 779fe46c3ffSTobias Grosser generateConditionalExecution( 780fe46c3ffSTobias Grosser Stmt, AccDom, Subject.c_str(), [&, this, MA]() { 781d86f2157SJohannes Doerfert Value *Val = MA->getAccessValue(); 782ee6a4fc6SMichael Kruse if (MA->isAnyPHIKind()) { 783fe46c3ffSTobias Grosser assert(MA->getIncoming().size() >= 1 && 784fe46c3ffSTobias Grosser "Block statements have exactly one exiting block, or " 785fe46c3ffSTobias Grosser "multiple but " 786ee6a4fc6SMichael Kruse "with same incoming block and value"); 787fe46c3ffSTobias Grosser assert(std::all_of(MA->getIncoming().begin(), 788fe46c3ffSTobias Grosser MA->getIncoming().end(), 789ee6a4fc6SMichael Kruse [&](std::pair<BasicBlock *, Value *> p) -> bool { 790ee6a4fc6SMichael Kruse return p.first == Stmt.getBasicBlock(); 791ee6a4fc6SMichael Kruse }) && 792ee6a4fc6SMichael Kruse "Incoming block must be statement's block"); 793ee6a4fc6SMichael Kruse Val = MA->getIncoming()[0].second; 794ee6a4fc6SMichael Kruse } 795fe46c3ffSTobias Grosser auto Address = getImplicitAddress(*MA, getLoopForStmt(Stmt), LTS, 796fe46c3ffSTobias Grosser BBMap, NewAccesses); 797d86f2157SJohannes Doerfert 798f2cdd144STobias Grosser Val = getNewValue(Stmt, Val, BBMap, LTS, L); 799eac9726eSMichael Kruse assert((!isa<Instruction>(Val) || 800eac9726eSMichael Kruse DT.dominates(cast<Instruction>(Val)->getParent(), 801eac9726eSMichael Kruse Builder.GetInsertBlock())) && 802eac9726eSMichael Kruse "Domination violation"); 803eac9726eSMichael Kruse assert((!isa<Instruction>(Address) || 804eac9726eSMichael Kruse DT.dominates(cast<Instruction>(Address)->getParent(), 805eac9726eSMichael Kruse Builder.GetInsertBlock())) && 806eac9726eSMichael Kruse "Domination violation"); 8072f5cbc44SMichael Kruse 8082f5cbc44SMichael Kruse // The new Val might have a different type than the old Val due to 8092f5cbc44SMichael Kruse // ScalarEvolution looking through bitcasts. 8102f5cbc44SMichael Kruse if (Val->getType() != Address->getType()->getPointerElementType()) 8112f5cbc44SMichael Kruse Address = Builder.CreateBitOrPointerCast( 8122f5cbc44SMichael Kruse Address, Val->getType()->getPointerTo()); 8132f5cbc44SMichael Kruse 81492245228STobias Grosser Builder.CreateStore(Val, Address); 815706f79abSMichael Kruse }); 816ecff11dcSJohannes Doerfert } 817ecff11dcSJohannes Doerfert } 818ecff11dcSJohannes Doerfert 819c0091a77STobias Grosser void BlockGenerator::createScalarInitialization(Scop &S) { 820ef74443cSJohannes Doerfert BasicBlock *ExitBB = S.getExit(); 821acf80064SEli Friedman BasicBlock *PreEntryBB = S.getEnteringBlock(); 822188542fdSJohannes Doerfert 823acf80064SEli Friedman Builder.SetInsertPoint(&*StartBlock->begin()); 824ecff11dcSJohannes Doerfert 825d7754a12SRoman Gareev for (auto &Array : S.arrays()) { 826c0091a77STobias Grosser if (Array->getNumberOfDimensions() != 0) 827c0091a77STobias Grosser continue; 828a535dff4STobias Grosser if (Array->isPHIKind()) { 829c0091a77STobias Grosser // For PHI nodes, the only values we need to store are the ones that 830c0091a77STobias Grosser // reach the PHI node from outside the region. In general there should 831c0091a77STobias Grosser // only be one such incoming edge and this edge should enter through 832acf80064SEli Friedman // 'PreEntryBB'. 833c0091a77STobias Grosser auto PHI = cast<PHINode>(Array->getBasePtr()); 834ecff11dcSJohannes Doerfert 835c0091a77STobias Grosser for (auto BI = PHI->block_begin(), BE = PHI->block_end(); BI != BE; BI++) 836acf80064SEli Friedman if (!S.contains(*BI) && *BI != PreEntryBB) 837c0091a77STobias Grosser llvm_unreachable("Incoming edges from outside the scop should always " 838acf80064SEli Friedman "come from PreEntryBB"); 839c0091a77STobias Grosser 840acf80064SEli Friedman int Idx = PHI->getBasicBlockIndex(PreEntryBB); 841c0091a77STobias Grosser if (Idx < 0) 842ecff11dcSJohannes Doerfert continue; 843ecff11dcSJohannes Doerfert 844c0091a77STobias Grosser Value *ScalarValue = PHI->getIncomingValue(Idx); 845ecff11dcSJohannes Doerfert 846587f1f57STobias Grosser Builder.CreateStore(ScalarValue, getOrCreateAlloca(Array)); 847c0091a77STobias Grosser continue; 848c0091a77STobias Grosser } 849c0091a77STobias Grosser 850c0091a77STobias Grosser auto *Inst = dyn_cast<Instruction>(Array->getBasePtr()); 851c0091a77STobias Grosser 852952b5304SJohannes Doerfert if (Inst && S.contains(Inst)) 853c0091a77STobias Grosser continue; 854c0091a77STobias Grosser 855188542fdSJohannes Doerfert // PHI nodes that are not marked as such in their SAI object are either exit 856188542fdSJohannes Doerfert // PHI nodes we model as common scalars but without initialization, or 857188542fdSJohannes Doerfert // incoming phi nodes that need to be initialized. Check if the first is the 858188542fdSJohannes Doerfert // case for Inst and do not create and initialize memory if so. 859188542fdSJohannes Doerfert if (auto *PHI = dyn_cast_or_null<PHINode>(Inst)) 860188542fdSJohannes Doerfert if (!S.hasSingleExitEdge() && PHI->getBasicBlockIndex(ExitBB) >= 0) 861717b8667SJohannes Doerfert continue; 862717b8667SJohannes Doerfert 863587f1f57STobias Grosser Builder.CreateStore(Array->getBasePtr(), getOrCreateAlloca(Array)); 864ecff11dcSJohannes Doerfert } 865ecff11dcSJohannes Doerfert } 866ecff11dcSJohannes Doerfert 867ef74443cSJohannes Doerfert void BlockGenerator::createScalarFinalization(Scop &S) { 868ecff11dcSJohannes Doerfert // The exit block of the __unoptimized__ region. 869ef74443cSJohannes Doerfert BasicBlock *ExitBB = S.getExitingBlock(); 870ecff11dcSJohannes Doerfert // The merge block __just after__ the region and the optimized region. 871ef74443cSJohannes Doerfert BasicBlock *MergeBB = S.getExit(); 872ecff11dcSJohannes Doerfert 873ecff11dcSJohannes Doerfert // The exit block of the __optimized__ region. 874ecff11dcSJohannes Doerfert BasicBlock *OptExitBB = *(pred_begin(MergeBB)); 875ecff11dcSJohannes Doerfert if (OptExitBB == ExitBB) 876ecff11dcSJohannes Doerfert OptExitBB = *(++pred_begin(MergeBB)); 877ecff11dcSJohannes Doerfert 878ecff11dcSJohannes Doerfert Builder.SetInsertPoint(OptExitBB->getTerminator()); 879ecff11dcSJohannes Doerfert for (const auto &EscapeMapping : EscapeMap) { 880ecff11dcSJohannes Doerfert // Extract the escaping instruction and the escaping users as well as the 881ecff11dcSJohannes Doerfert // alloca the instruction was demoted to. 882fbde4355SMichael Kruse Instruction *EscapeInst = EscapeMapping.first; 883fbde4355SMichael Kruse const auto &EscapeMappingValue = EscapeMapping.second; 884ecff11dcSJohannes Doerfert const EscapeUserVectorTy &EscapeUsers = EscapeMappingValue.second; 88564c0ff41STobias Grosser Value *ScalarAddr = EscapeMappingValue.first; 886ecff11dcSJohannes Doerfert 887ecff11dcSJohannes Doerfert // Reload the demoted instruction in the optimized version of the SCoP. 888d8b6ad25SJohannes Doerfert Value *EscapeInstReload = 889ecff11dcSJohannes Doerfert Builder.CreateLoad(ScalarAddr, EscapeInst->getName() + ".final_reload"); 890d8b6ad25SJohannes Doerfert EscapeInstReload = 891d8b6ad25SJohannes Doerfert Builder.CreateBitOrPointerCast(EscapeInstReload, EscapeInst->getType()); 892ecff11dcSJohannes Doerfert 893ecff11dcSJohannes Doerfert // Create the merge PHI that merges the optimized and unoptimized version. 894ecff11dcSJohannes Doerfert PHINode *MergePHI = PHINode::Create(EscapeInst->getType(), 2, 895ecff11dcSJohannes Doerfert EscapeInst->getName() + ".merge"); 896b8f58b53SDuncan P. N. Exon Smith MergePHI->insertBefore(&*MergeBB->getFirstInsertionPt()); 897ecff11dcSJohannes Doerfert 898ecff11dcSJohannes Doerfert // Add the respective values to the merge PHI. 899ecff11dcSJohannes Doerfert MergePHI->addIncoming(EscapeInstReload, OptExitBB); 900ecff11dcSJohannes Doerfert MergePHI->addIncoming(EscapeInst, ExitBB); 901ecff11dcSJohannes Doerfert 902ecff11dcSJohannes Doerfert // The information of scalar evolution about the escaping instruction needs 903ecff11dcSJohannes Doerfert // to be revoked so the new merged instruction will be used. 904ecff11dcSJohannes Doerfert if (SE.isSCEVable(EscapeInst->getType())) 905ecff11dcSJohannes Doerfert SE.forgetValue(EscapeInst); 906ecff11dcSJohannes Doerfert 907ecff11dcSJohannes Doerfert // Replace all uses of the demoted instruction with the merge PHI. 908ecff11dcSJohannes Doerfert for (Instruction *EUser : EscapeUsers) 909ecff11dcSJohannes Doerfert EUser->replaceUsesOfWith(EscapeInst, MergePHI); 910ecff11dcSJohannes Doerfert } 911ecff11dcSJohannes Doerfert } 912ecff11dcSJohannes Doerfert 913ffa2446fSTobias Grosser void BlockGenerator::findOutsideUsers(Scop &S) { 914d7754a12SRoman Gareev for (auto &Array : S.arrays()) { 915ffa2446fSTobias Grosser 916ffa2446fSTobias Grosser if (Array->getNumberOfDimensions() != 0) 917ffa2446fSTobias Grosser continue; 918ffa2446fSTobias Grosser 919a535dff4STobias Grosser if (Array->isPHIKind()) 920ffa2446fSTobias Grosser continue; 921ffa2446fSTobias Grosser 922ffa2446fSTobias Grosser auto *Inst = dyn_cast<Instruction>(Array->getBasePtr()); 923ffa2446fSTobias Grosser 924ffa2446fSTobias Grosser if (!Inst) 925ffa2446fSTobias Grosser continue; 926ffa2446fSTobias Grosser 927ffa2446fSTobias Grosser // Scop invariant hoisting moves some of the base pointers out of the scop. 928ffa2446fSTobias Grosser // We can ignore these, as the invariant load hoisting already registers the 929ffa2446fSTobias Grosser // relevant outside users. 930952b5304SJohannes Doerfert if (!S.contains(Inst)) 931ffa2446fSTobias Grosser continue; 932ffa2446fSTobias Grosser 933587f1f57STobias Grosser handleOutsideUsers(S, Array); 934ffa2446fSTobias Grosser } 935ffa2446fSTobias Grosser } 936ffa2446fSTobias Grosser 93727d742daSTobias Grosser void BlockGenerator::createExitPHINodeMerges(Scop &S) { 93827d742daSTobias Grosser if (S.hasSingleExitEdge()) 93927d742daSTobias Grosser return; 94027d742daSTobias Grosser 941ef74443cSJohannes Doerfert auto *ExitBB = S.getExitingBlock(); 942ef74443cSJohannes Doerfert auto *MergeBB = S.getExit(); 94327d742daSTobias Grosser auto *AfterMergeBB = MergeBB->getSingleSuccessor(); 94427d742daSTobias Grosser BasicBlock *OptExitBB = *(pred_begin(MergeBB)); 94527d742daSTobias Grosser if (OptExitBB == ExitBB) 94627d742daSTobias Grosser OptExitBB = *(++pred_begin(MergeBB)); 94727d742daSTobias Grosser 94827d742daSTobias Grosser Builder.SetInsertPoint(OptExitBB->getTerminator()); 94927d742daSTobias Grosser 950d7754a12SRoman Gareev for (auto &SAI : S.arrays()) { 95127d742daSTobias Grosser auto *Val = SAI->getBasePtr(); 95227d742daSTobias Grosser 953faedfcbfSMichael Kruse // Only Value-like scalars need a merge PHI. Exit block PHIs receive either 954faedfcbfSMichael Kruse // the original PHI's value or the reloaded incoming values from the 955faedfcbfSMichael Kruse // generated code. An llvm::Value is merged between the original code's 956faedfcbfSMichael Kruse // value or the generated one. 957fa53c86dSMichael Kruse if (!SAI->isExitPHIKind()) 958faedfcbfSMichael Kruse continue; 959faedfcbfSMichael Kruse 96027d742daSTobias Grosser PHINode *PHI = dyn_cast<PHINode>(Val); 96127d742daSTobias Grosser if (!PHI) 96227d742daSTobias Grosser continue; 96327d742daSTobias Grosser 964a3f6edaeSTobias Grosser if (PHI->getParent() != AfterMergeBB) 96527d742daSTobias Grosser continue; 96627d742daSTobias Grosser 96727d742daSTobias Grosser std::string Name = PHI->getName(); 968587f1f57STobias Grosser Value *ScalarAddr = getOrCreateAlloca(SAI); 96927d742daSTobias Grosser Value *Reload = Builder.CreateLoad(ScalarAddr, Name + ".ph.final_reload"); 97027d742daSTobias Grosser Reload = Builder.CreateBitOrPointerCast(Reload, PHI->getType()); 97127d742daSTobias Grosser Value *OriginalValue = PHI->getIncomingValueForBlock(MergeBB); 972faedfcbfSMichael Kruse assert((!isa<Instruction>(OriginalValue) || 973faedfcbfSMichael Kruse cast<Instruction>(OriginalValue)->getParent() != MergeBB) && 974faedfcbfSMichael Kruse "Original value must no be one we just generated."); 97527d742daSTobias Grosser auto *MergePHI = PHINode::Create(PHI->getType(), 2, Name + ".ph.merge"); 976b8f58b53SDuncan P. N. Exon Smith MergePHI->insertBefore(&*MergeBB->getFirstInsertionPt()); 97727d742daSTobias Grosser MergePHI->addIncoming(Reload, OptExitBB); 97827d742daSTobias Grosser MergePHI->addIncoming(OriginalValue, ExitBB); 97927d742daSTobias Grosser int Idx = PHI->getBasicBlockIndex(MergeBB); 98027d742daSTobias Grosser PHI->setIncomingValue(Idx, MergePHI); 98127d742daSTobias Grosser } 98227d742daSTobias Grosser } 98327d742daSTobias Grosser 9841c184409STobias Grosser void BlockGenerator::invalidateScalarEvolution(Scop &S) { 9851c184409STobias Grosser for (auto &Stmt : S) 986b3224adfSRoman Gareev if (Stmt.isCopyStmt()) 987b3224adfSRoman Gareev continue; 988b3224adfSRoman Gareev else if (Stmt.isBlockStmt()) 9891c184409STobias Grosser for (auto &Inst : *Stmt.getBasicBlock()) 9901c184409STobias Grosser SE.forgetValue(&Inst); 9911c184409STobias Grosser else if (Stmt.isRegionStmt()) 9921c184409STobias Grosser for (auto *BB : Stmt.getRegion()->blocks()) 9931c184409STobias Grosser for (auto &Inst : *BB) 9941c184409STobias Grosser SE.forgetValue(&Inst); 9951c184409STobias Grosser else 9961c184409STobias Grosser llvm_unreachable("Unexpected statement type found"); 9971aad76c1SMichael Kruse 9981aad76c1SMichael Kruse // Invalidate SCEV of loops surrounding the EscapeUsers. 9991aad76c1SMichael Kruse for (const auto &EscapeMapping : EscapeMap) { 10001aad76c1SMichael Kruse const EscapeUserVectorTy &EscapeUsers = EscapeMapping.second.second; 10011aad76c1SMichael Kruse for (Instruction *EUser : EscapeUsers) { 10021aad76c1SMichael Kruse if (Loop *L = LI.getLoopFor(EUser->getParent())) 10031aad76c1SMichael Kruse while (L) { 10041aad76c1SMichael Kruse SE.forgetLoop(L); 10051aad76c1SMichael Kruse L = L->getParentLoop(); 10061aad76c1SMichael Kruse } 10071aad76c1SMichael Kruse } 10081aad76c1SMichael Kruse } 10091c184409STobias Grosser } 10101c184409STobias Grosser 1011ffa2446fSTobias Grosser void BlockGenerator::finalizeSCoP(Scop &S) { 1012ffa2446fSTobias Grosser findOutsideUsers(S); 1013c0091a77STobias Grosser createScalarInitialization(S); 101427d742daSTobias Grosser createExitPHINodeMerges(S); 1015ef74443cSJohannes Doerfert createScalarFinalization(S); 10161c184409STobias Grosser invalidateScalarEvolution(S); 10173b11a16aSHongbin Zheng } 10183b11a16aSHongbin Zheng 1019be9c9117SJohannes Doerfert VectorBlockGenerator::VectorBlockGenerator(BlockGenerator &BlockGen, 1020be9c9117SJohannes Doerfert std::vector<LoopToScevMapT> &VLTS, 1021be9c9117SJohannes Doerfert isl_map *Schedule) 1022bc132607STobias Grosser : BlockGenerator(BlockGen), VLTS(VLTS), Schedule(Schedule) { 1023a00a0291SSebastian Pop assert(Schedule && "No statement domain provided"); 10243b11a16aSHongbin Zheng } 10253b11a16aSHongbin Zheng 10262f1acac6STobias Grosser Value *VectorBlockGenerator::getVectorValue(ScopStmt &Stmt, Value *Old, 1027e602a076STobias Grosser ValueMapT &VectorMap, 1028e602a076STobias Grosser VectorValueMapT &ScalarMaps, 1029e602a076STobias Grosser Loop *L) { 1030fe11e287SHongbin Zheng if (Value *NewValue = VectorMap.lookup(Old)) 1031fe11e287SHongbin Zheng return NewValue; 10323b11a16aSHongbin Zheng 10333b11a16aSHongbin Zheng int Width = getVectorWidth(); 10343b11a16aSHongbin Zheng 10353b11a16aSHongbin Zheng Value *Vector = UndefValue::get(VectorType::get(Old->getType(), Width)); 10363b11a16aSHongbin Zheng 10373b11a16aSHongbin Zheng for (int Lane = 0; Lane < Width; Lane++) 1038c14582f2STobias Grosser Vector = Builder.CreateInsertElement( 1039bc132607STobias Grosser Vector, getNewValue(Stmt, Old, ScalarMaps[Lane], VLTS[Lane], L), 10407242ad92STobias Grosser Builder.getInt32(Lane)); 10413b11a16aSHongbin Zheng 10423b11a16aSHongbin Zheng VectorMap[Old] = Vector; 10433b11a16aSHongbin Zheng 10443b11a16aSHongbin Zheng return Vector; 10453b11a16aSHongbin Zheng } 10463b11a16aSHongbin Zheng 10473b11a16aSHongbin Zheng Type *VectorBlockGenerator::getVectorPtrTy(const Value *Val, int Width) { 10483b11a16aSHongbin Zheng PointerType *PointerTy = dyn_cast<PointerType>(Val->getType()); 10493b11a16aSHongbin Zheng assert(PointerTy && "PointerType expected"); 10503b11a16aSHongbin Zheng 10513b11a16aSHongbin Zheng Type *ScalarType = PointerTy->getElementType(); 10523b11a16aSHongbin Zheng VectorType *VectorType = VectorType::get(ScalarType, Width); 10533b11a16aSHongbin Zheng 10543b11a16aSHongbin Zheng return PointerType::getUnqual(VectorType); 10553b11a16aSHongbin Zheng } 10563b11a16aSHongbin Zheng 1057be9c9117SJohannes Doerfert Value *VectorBlockGenerator::generateStrideOneLoad( 10582f1acac6STobias Grosser ScopStmt &Stmt, LoadInst *Load, VectorValueMapT &ScalarMaps, 10592d1ed0bfSTobias Grosser __isl_keep isl_id_to_ast_expr *NewAccesses, bool NegativeStride = false) { 10600dd463faSTobias Grosser unsigned VectorWidth = getVectorWidth(); 10619646e3feSTobias Grosser auto *Pointer = Load->getPointerOperand(); 10620dd463faSTobias Grosser Type *VectorPtrType = getVectorPtrTy(Pointer, VectorWidth); 10630dd463faSTobias Grosser unsigned Offset = NegativeStride ? VectorWidth - 1 : 0; 10640dd463faSTobias Grosser 10658f25b0cbSMichael Kruse Value *NewPointer = generateLocationAccessed(Stmt, Load, ScalarMaps[Offset], 1066bc132607STobias Grosser VLTS[Offset], NewAccesses); 1067c14582f2STobias Grosser Value *VectorPtr = 1068c14582f2STobias Grosser Builder.CreateBitCast(NewPointer, VectorPtrType, "vector_ptr"); 1069c14582f2STobias Grosser LoadInst *VecLoad = 1070c14582f2STobias Grosser Builder.CreateLoad(VectorPtr, Load->getName() + "_p_vec_full"); 10713b11a16aSHongbin Zheng if (!Aligned) 10723b11a16aSHongbin Zheng VecLoad->setAlignment(8); 10733b11a16aSHongbin Zheng 10740dd463faSTobias Grosser if (NegativeStride) { 10750dd463faSTobias Grosser SmallVector<Constant *, 16> Indices; 10760dd463faSTobias Grosser for (int i = VectorWidth - 1; i >= 0; i--) 10770dd463faSTobias Grosser Indices.push_back(ConstantInt::get(Builder.getInt32Ty(), i)); 10780dd463faSTobias Grosser Constant *SV = llvm::ConstantVector::get(Indices); 10790dd463faSTobias Grosser Value *RevVecLoad = Builder.CreateShuffleVector( 10800dd463faSTobias Grosser VecLoad, VecLoad, SV, Load->getName() + "_reverse"); 10810dd463faSTobias Grosser return RevVecLoad; 10820dd463faSTobias Grosser } 10830dd463faSTobias Grosser 10843b11a16aSHongbin Zheng return VecLoad; 10853b11a16aSHongbin Zheng } 10863b11a16aSHongbin Zheng 10872d1ed0bfSTobias Grosser Value *VectorBlockGenerator::generateStrideZeroLoad( 10882f1acac6STobias Grosser ScopStmt &Stmt, LoadInst *Load, ValueMapT &BBMap, 10892d1ed0bfSTobias Grosser __isl_keep isl_id_to_ast_expr *NewAccesses) { 10909646e3feSTobias Grosser auto *Pointer = Load->getPointerOperand(); 10913b11a16aSHongbin Zheng Type *VectorPtrType = getVectorPtrTy(Pointer, 1); 109270131d34SMichael Kruse Value *NewPointer = 109370131d34SMichael Kruse generateLocationAccessed(Stmt, Load, BBMap, VLTS[0], NewAccesses); 10943b11a16aSHongbin Zheng Value *VectorPtr = Builder.CreateBitCast(NewPointer, VectorPtrType, 10953b11a16aSHongbin Zheng Load->getName() + "_p_vec_p"); 1096c14582f2STobias Grosser LoadInst *ScalarLoad = 1097c14582f2STobias Grosser Builder.CreateLoad(VectorPtr, Load->getName() + "_p_splat_one"); 10983b11a16aSHongbin Zheng 10993b11a16aSHongbin Zheng if (!Aligned) 11003b11a16aSHongbin Zheng ScalarLoad->setAlignment(8); 11013b11a16aSHongbin Zheng 1102c14582f2STobias Grosser Constant *SplatVector = Constant::getNullValue( 1103c14582f2STobias Grosser VectorType::get(Builder.getInt32Ty(), getVectorWidth())); 11043b11a16aSHongbin Zheng 1105c14582f2STobias Grosser Value *VectorLoad = Builder.CreateShuffleVector( 1106c14582f2STobias Grosser ScalarLoad, ScalarLoad, SplatVector, Load->getName() + "_p_splat"); 11073b11a16aSHongbin Zheng return VectorLoad; 11083b11a16aSHongbin Zheng } 11093b11a16aSHongbin Zheng 1110be9c9117SJohannes Doerfert Value *VectorBlockGenerator::generateUnknownStrideLoad( 11112f1acac6STobias Grosser ScopStmt &Stmt, LoadInst *Load, VectorValueMapT &ScalarMaps, 111209e3697fSJohannes Doerfert __isl_keep isl_id_to_ast_expr *NewAccesses) { 11133b11a16aSHongbin Zheng int VectorWidth = getVectorWidth(); 11149646e3feSTobias Grosser auto *Pointer = Load->getPointerOperand(); 11153b11a16aSHongbin Zheng VectorType *VectorType = VectorType::get( 11163b11a16aSHongbin Zheng dyn_cast<PointerType>(Pointer->getType())->getElementType(), VectorWidth); 11173b11a16aSHongbin Zheng 11183b11a16aSHongbin Zheng Value *Vector = UndefValue::get(VectorType); 11193b11a16aSHongbin Zheng 11203b11a16aSHongbin Zheng for (int i = 0; i < VectorWidth; i++) { 112170131d34SMichael Kruse Value *NewPointer = generateLocationAccessed(Stmt, Load, ScalarMaps[i], 112270131d34SMichael Kruse VLTS[i], NewAccesses); 1123c14582f2STobias Grosser Value *ScalarLoad = 1124c14582f2STobias Grosser Builder.CreateLoad(NewPointer, Load->getName() + "_p_scalar_"); 1125c14582f2STobias Grosser Vector = Builder.CreateInsertElement( 1126c14582f2STobias Grosser Vector, ScalarLoad, Builder.getInt32(i), Load->getName() + "_p_vec_"); 11273b11a16aSHongbin Zheng } 11283b11a16aSHongbin Zheng 11293b11a16aSHongbin Zheng return Vector; 11303b11a16aSHongbin Zheng } 11313b11a16aSHongbin Zheng 11322d1ed0bfSTobias Grosser void VectorBlockGenerator::generateLoad( 11332f1acac6STobias Grosser ScopStmt &Stmt, LoadInst *Load, ValueMapT &VectorMap, 11342d1ed0bfSTobias Grosser VectorValueMapT &ScalarMaps, __isl_keep isl_id_to_ast_expr *NewAccesses) { 1135c1db67e2SJohannes Doerfert if (Value *PreloadLoad = GlobalMap.lookup(Load)) { 1136c1db67e2SJohannes Doerfert VectorMap[Load] = Builder.CreateVectorSplat(getVectorWidth(), PreloadLoad, 1137c1db67e2SJohannes Doerfert Load->getName() + "_p"); 1138c1db67e2SJohannes Doerfert return; 1139c1db67e2SJohannes Doerfert } 1140c1db67e2SJohannes Doerfert 11412873645cSTobias Grosser if (!VectorType::isValidElementType(Load->getType())) { 11423b11a16aSHongbin Zheng for (int i = 0; i < getVectorWidth(); i++) 1143bc132607STobias Grosser ScalarMaps[i][Load] = 1144888ab551SMichael Kruse generateArrayLoad(Stmt, Load, ScalarMaps[i], VLTS[i], NewAccesses); 11453b11a16aSHongbin Zheng return; 11463b11a16aSHongbin Zheng } 11473b11a16aSHongbin Zheng 1148184a4926STobias Grosser const MemoryAccess &Access = Stmt.getArrayAccessFor(Load); 11493b11a16aSHongbin Zheng 115095493984STobias Grosser // Make sure we have scalar values available to access the pointer to 115195493984STobias Grosser // the data location. 115295493984STobias Grosser extractScalarValues(Load, VectorMap, ScalarMaps); 115395493984STobias Grosser 11543b11a16aSHongbin Zheng Value *NewLoad; 1155718d04c6STobias Grosser if (Access.isStrideZero(isl::manage_copy(Schedule))) 11562d1ed0bfSTobias Grosser NewLoad = generateStrideZeroLoad(Stmt, Load, ScalarMaps[0], NewAccesses); 1157718d04c6STobias Grosser else if (Access.isStrideOne(isl::manage_copy(Schedule))) 11582d1ed0bfSTobias Grosser NewLoad = generateStrideOneLoad(Stmt, Load, ScalarMaps, NewAccesses); 1159718d04c6STobias Grosser else if (Access.isStrideX(isl::manage_copy(Schedule), -1)) 11602d1ed0bfSTobias Grosser NewLoad = generateStrideOneLoad(Stmt, Load, ScalarMaps, NewAccesses, true); 11613b11a16aSHongbin Zheng else 11622d1ed0bfSTobias Grosser NewLoad = generateUnknownStrideLoad(Stmt, Load, ScalarMaps, NewAccesses); 11633b11a16aSHongbin Zheng 11643b11a16aSHongbin Zheng VectorMap[Load] = NewLoad; 11653b11a16aSHongbin Zheng } 11663b11a16aSHongbin Zheng 11672f1acac6STobias Grosser void VectorBlockGenerator::copyUnaryInst(ScopStmt &Stmt, UnaryInstruction *Inst, 11683b11a16aSHongbin Zheng ValueMapT &VectorMap, 11693b11a16aSHongbin Zheng VectorValueMapT &ScalarMaps) { 11703b11a16aSHongbin Zheng int VectorWidth = getVectorWidth(); 1171be9c9117SJohannes Doerfert Value *NewOperand = getVectorValue(Stmt, Inst->getOperand(0), VectorMap, 11722c3ffc04SJohannes Doerfert ScalarMaps, getLoopForStmt(Stmt)); 11733b11a16aSHongbin Zheng 11743b11a16aSHongbin Zheng assert(isa<CastInst>(Inst) && "Can not generate vector code for instruction"); 11753b11a16aSHongbin Zheng 11763b11a16aSHongbin Zheng const CastInst *Cast = dyn_cast<CastInst>(Inst); 11773b11a16aSHongbin Zheng VectorType *DestType = VectorType::get(Inst->getType(), VectorWidth); 11783b11a16aSHongbin Zheng VectorMap[Inst] = Builder.CreateCast(Cast->getOpcode(), NewOperand, DestType); 11793b11a16aSHongbin Zheng } 11803b11a16aSHongbin Zheng 11812f1acac6STobias Grosser void VectorBlockGenerator::copyBinaryInst(ScopStmt &Stmt, BinaryOperator *Inst, 11823b11a16aSHongbin Zheng ValueMapT &VectorMap, 11833b11a16aSHongbin Zheng VectorValueMapT &ScalarMaps) { 11842c3ffc04SJohannes Doerfert Loop *L = getLoopForStmt(Stmt); 11853b11a16aSHongbin Zheng Value *OpZero = Inst->getOperand(0); 11863b11a16aSHongbin Zheng Value *OpOne = Inst->getOperand(1); 11873b11a16aSHongbin Zheng 11883b11a16aSHongbin Zheng Value *NewOpZero, *NewOpOne; 1189be9c9117SJohannes Doerfert NewOpZero = getVectorValue(Stmt, OpZero, VectorMap, ScalarMaps, L); 1190be9c9117SJohannes Doerfert NewOpOne = getVectorValue(Stmt, OpOne, VectorMap, ScalarMaps, L); 11913b11a16aSHongbin Zheng 11921bb59b0dSTobias Grosser Value *NewInst = Builder.CreateBinOp(Inst->getOpcode(), NewOpZero, NewOpOne, 11933b11a16aSHongbin Zheng Inst->getName() + "p_vec"); 11943b11a16aSHongbin Zheng VectorMap[Inst] = NewInst; 11953b11a16aSHongbin Zheng } 11963b11a16aSHongbin Zheng 11972d1ed0bfSTobias Grosser void VectorBlockGenerator::copyStore( 11982f1acac6STobias Grosser ScopStmt &Stmt, StoreInst *Store, ValueMapT &VectorMap, 11992d1ed0bfSTobias Grosser VectorValueMapT &ScalarMaps, __isl_keep isl_id_to_ast_expr *NewAccesses) { 1200184a4926STobias Grosser const MemoryAccess &Access = Stmt.getArrayAccessFor(Store); 12013b11a16aSHongbin Zheng 12029646e3feSTobias Grosser auto *Pointer = Store->getPointerOperand(); 1203be9c9117SJohannes Doerfert Value *Vector = getVectorValue(Stmt, Store->getValueOperand(), VectorMap, 12042c3ffc04SJohannes Doerfert ScalarMaps, getLoopForStmt(Stmt)); 12053b11a16aSHongbin Zheng 120650fd7010STobias Grosser // Make sure we have scalar values available to access the pointer to 120750fd7010STobias Grosser // the data location. 120850fd7010STobias Grosser extractScalarValues(Store, VectorMap, ScalarMaps); 120950fd7010STobias Grosser 1210718d04c6STobias Grosser if (Access.isStrideOne(isl::manage_copy(Schedule))) { 12111947f863SJohannes Doerfert Type *VectorPtrType = getVectorPtrTy(Pointer, getVectorWidth()); 121270131d34SMichael Kruse Value *NewPointer = generateLocationAccessed(Stmt, Store, ScalarMaps[0], 121370131d34SMichael Kruse VLTS[0], NewAccesses); 12143b11a16aSHongbin Zheng 1215c14582f2STobias Grosser Value *VectorPtr = 1216c14582f2STobias Grosser Builder.CreateBitCast(NewPointer, VectorPtrType, "vector_ptr"); 12173b11a16aSHongbin Zheng StoreInst *Store = Builder.CreateStore(Vector, VectorPtr); 12183b11a16aSHongbin Zheng 12193b11a16aSHongbin Zheng if (!Aligned) 12203b11a16aSHongbin Zheng Store->setAlignment(8); 12213b11a16aSHongbin Zheng } else { 12223b11a16aSHongbin Zheng for (unsigned i = 0; i < ScalarMaps.size(); i++) { 12231bb59b0dSTobias Grosser Value *Scalar = Builder.CreateExtractElement(Vector, Builder.getInt32(i)); 122470131d34SMichael Kruse Value *NewPointer = generateLocationAccessed(Stmt, Store, ScalarMaps[i], 122570131d34SMichael Kruse VLTS[i], NewAccesses); 12263b11a16aSHongbin Zheng Builder.CreateStore(Scalar, NewPointer); 12273b11a16aSHongbin Zheng } 12283b11a16aSHongbin Zheng } 12293b11a16aSHongbin Zheng } 12303b11a16aSHongbin Zheng 12313b11a16aSHongbin Zheng bool VectorBlockGenerator::hasVectorOperands(const Instruction *Inst, 12323b11a16aSHongbin Zheng ValueMapT &VectorMap) { 123391f5b262STobias Grosser for (Value *Operand : Inst->operands()) 123491f5b262STobias Grosser if (VectorMap.count(Operand)) 12353b11a16aSHongbin Zheng return true; 12363b11a16aSHongbin Zheng return false; 12373b11a16aSHongbin Zheng } 12383b11a16aSHongbin Zheng 12393b11a16aSHongbin Zheng bool VectorBlockGenerator::extractScalarValues(const Instruction *Inst, 12403b11a16aSHongbin Zheng ValueMapT &VectorMap, 12413b11a16aSHongbin Zheng VectorValueMapT &ScalarMaps) { 12423b11a16aSHongbin Zheng bool HasVectorOperand = false; 12433b11a16aSHongbin Zheng int VectorWidth = getVectorWidth(); 12443b11a16aSHongbin Zheng 124591f5b262STobias Grosser for (Value *Operand : Inst->operands()) { 124691f5b262STobias Grosser ValueMapT::iterator VecOp = VectorMap.find(Operand); 12473b11a16aSHongbin Zheng 12483b11a16aSHongbin Zheng if (VecOp == VectorMap.end()) 12493b11a16aSHongbin Zheng continue; 12503b11a16aSHongbin Zheng 12513b11a16aSHongbin Zheng HasVectorOperand = true; 12523b11a16aSHongbin Zheng Value *NewVector = VecOp->second; 12533b11a16aSHongbin Zheng 12543b11a16aSHongbin Zheng for (int i = 0; i < VectorWidth; ++i) { 12553b11a16aSHongbin Zheng ValueMapT &SM = ScalarMaps[i]; 12563b11a16aSHongbin Zheng 12573b11a16aSHongbin Zheng // If there is one scalar extracted, all scalar elements should have 12583b11a16aSHongbin Zheng // already been extracted by the code here. So no need to check for the 12592219d157STobias Grosser // existence of all of them. 126091f5b262STobias Grosser if (SM.count(Operand)) 12613b11a16aSHongbin Zheng break; 12623b11a16aSHongbin Zheng 126391f5b262STobias Grosser SM[Operand] = 126491f5b262STobias Grosser Builder.CreateExtractElement(NewVector, Builder.getInt32(i)); 12653b11a16aSHongbin Zheng } 12663b11a16aSHongbin Zheng } 12673b11a16aSHongbin Zheng 12683b11a16aSHongbin Zheng return HasVectorOperand; 12693b11a16aSHongbin Zheng } 12703b11a16aSHongbin Zheng 12712d1ed0bfSTobias Grosser void VectorBlockGenerator::copyInstScalarized( 12722f1acac6STobias Grosser ScopStmt &Stmt, Instruction *Inst, ValueMapT &VectorMap, 12732d1ed0bfSTobias Grosser VectorValueMapT &ScalarMaps, __isl_keep isl_id_to_ast_expr *NewAccesses) { 12743b11a16aSHongbin Zheng bool HasVectorOperand; 12753b11a16aSHongbin Zheng int VectorWidth = getVectorWidth(); 12763b11a16aSHongbin Zheng 12773b11a16aSHongbin Zheng HasVectorOperand = extractScalarValues(Inst, VectorMap, ScalarMaps); 12783b11a16aSHongbin Zheng 12793b11a16aSHongbin Zheng for (int VectorLane = 0; VectorLane < getVectorWidth(); VectorLane++) 1280be9c9117SJohannes Doerfert BlockGenerator::copyInstruction(Stmt, Inst, ScalarMaps[VectorLane], 1281bc132607STobias Grosser VLTS[VectorLane], NewAccesses); 12823b11a16aSHongbin Zheng 12833b11a16aSHongbin Zheng if (!VectorType::isValidElementType(Inst->getType()) || !HasVectorOperand) 12843b11a16aSHongbin Zheng return; 12853b11a16aSHongbin Zheng 12863b11a16aSHongbin Zheng // Make the result available as vector value. 12873b11a16aSHongbin Zheng VectorType *VectorType = VectorType::get(Inst->getType(), VectorWidth); 12883b11a16aSHongbin Zheng Value *Vector = UndefValue::get(VectorType); 12893b11a16aSHongbin Zheng 12903b11a16aSHongbin Zheng for (int i = 0; i < VectorWidth; i++) 12913b11a16aSHongbin Zheng Vector = Builder.CreateInsertElement(Vector, ScalarMaps[i][Inst], 12923b11a16aSHongbin Zheng Builder.getInt32(i)); 12933b11a16aSHongbin Zheng 12943b11a16aSHongbin Zheng VectorMap[Inst] = Vector; 12953b11a16aSHongbin Zheng } 12963b11a16aSHongbin Zheng 1297bc132607STobias Grosser int VectorBlockGenerator::getVectorWidth() { return VLTS.size(); } 12983b11a16aSHongbin Zheng 12992d1ed0bfSTobias Grosser void VectorBlockGenerator::copyInstruction( 13002f1acac6STobias Grosser ScopStmt &Stmt, Instruction *Inst, ValueMapT &VectorMap, 13012d1ed0bfSTobias Grosser VectorValueMapT &ScalarMaps, __isl_keep isl_id_to_ast_expr *NewAccesses) { 13023b11a16aSHongbin Zheng // Terminator instructions control the control flow. They are explicitly 13033b11a16aSHongbin Zheng // expressed in the clast and do not need to be copied. 13043b11a16aSHongbin Zheng if (Inst->isTerminator()) 13053b11a16aSHongbin Zheng return; 13063b11a16aSHongbin Zheng 13075dced269SJohannes Doerfert if (canSyntheziseInStmt(Stmt, Inst)) 1308e71c6ab5STobias Grosser return; 1309e71c6ab5STobias Grosser 13109646e3feSTobias Grosser if (auto *Load = dyn_cast<LoadInst>(Inst)) { 13112d1ed0bfSTobias Grosser generateLoad(Stmt, Load, VectorMap, ScalarMaps, NewAccesses); 13123b11a16aSHongbin Zheng return; 13133b11a16aSHongbin Zheng } 13143b11a16aSHongbin Zheng 13153b11a16aSHongbin Zheng if (hasVectorOperands(Inst, VectorMap)) { 13169646e3feSTobias Grosser if (auto *Store = dyn_cast<StoreInst>(Inst)) { 13170446d81eSMichael Kruse // Identified as redundant by -polly-simplify. 13180446d81eSMichael Kruse if (!Stmt.getArrayAccessOrNULLFor(Store)) 13190446d81eSMichael Kruse return; 13200446d81eSMichael Kruse 13212d1ed0bfSTobias Grosser copyStore(Stmt, Store, VectorMap, ScalarMaps, NewAccesses); 13223b11a16aSHongbin Zheng return; 13233b11a16aSHongbin Zheng } 13243b11a16aSHongbin Zheng 13259646e3feSTobias Grosser if (auto *Unary = dyn_cast<UnaryInstruction>(Inst)) { 1326be9c9117SJohannes Doerfert copyUnaryInst(Stmt, Unary, VectorMap, ScalarMaps); 13273b11a16aSHongbin Zheng return; 13283b11a16aSHongbin Zheng } 13293b11a16aSHongbin Zheng 13309646e3feSTobias Grosser if (auto *Binary = dyn_cast<BinaryOperator>(Inst)) { 1331be9c9117SJohannes Doerfert copyBinaryInst(Stmt, Binary, VectorMap, ScalarMaps); 13323b11a16aSHongbin Zheng return; 13333b11a16aSHongbin Zheng } 13343b11a16aSHongbin Zheng 1335a6d48f59SMichael Kruse // Fallthrough: We generate scalar instructions, if we don't know how to 13363b11a16aSHongbin Zheng // generate vector code. 13373b11a16aSHongbin Zheng } 13383b11a16aSHongbin Zheng 13392d1ed0bfSTobias Grosser copyInstScalarized(Stmt, Inst, VectorMap, ScalarMaps, NewAccesses); 13403b11a16aSHongbin Zheng } 13413b11a16aSHongbin Zheng 1342a69d4f0dSTobias Grosser void VectorBlockGenerator::generateScalarVectorLoads( 1343a69d4f0dSTobias Grosser ScopStmt &Stmt, ValueMapT &VectorBlockMap) { 1344a69d4f0dSTobias Grosser for (MemoryAccess *MA : Stmt) { 1345a69d4f0dSTobias Grosser if (MA->isArrayKind() || MA->isWrite()) 1346a69d4f0dSTobias Grosser continue; 1347a69d4f0dSTobias Grosser 1348a69d4f0dSTobias Grosser auto *Address = getOrCreateAlloca(*MA); 1349a69d4f0dSTobias Grosser Type *VectorPtrType = getVectorPtrTy(Address, 1); 1350a69d4f0dSTobias Grosser Value *VectorPtr = Builder.CreateBitCast(Address, VectorPtrType, 1351a69d4f0dSTobias Grosser Address->getName() + "_p_vec_p"); 1352a69d4f0dSTobias Grosser auto *Val = Builder.CreateLoad(VectorPtr, Address->getName() + ".reload"); 1353a69d4f0dSTobias Grosser Constant *SplatVector = Constant::getNullValue( 1354a69d4f0dSTobias Grosser VectorType::get(Builder.getInt32Ty(), getVectorWidth())); 1355a69d4f0dSTobias Grosser 1356a69d4f0dSTobias Grosser Value *VectorVal = Builder.CreateShuffleVector( 1357a69d4f0dSTobias Grosser Val, Val, SplatVector, Address->getName() + "_p_splat"); 1358583be06fSTobias Grosser VectorBlockMap[MA->getAccessValue()] = VectorVal; 1359a69d4f0dSTobias Grosser } 1360a69d4f0dSTobias Grosser } 1361a69d4f0dSTobias Grosser 1362a69d4f0dSTobias Grosser void VectorBlockGenerator::verifyNoScalarStores(ScopStmt &Stmt) { 1363a69d4f0dSTobias Grosser for (MemoryAccess *MA : Stmt) { 1364a69d4f0dSTobias Grosser if (MA->isArrayKind() || MA->isRead()) 1365a69d4f0dSTobias Grosser continue; 1366a69d4f0dSTobias Grosser 1367a69d4f0dSTobias Grosser llvm_unreachable("Scalar stores not expected in vector loop"); 1368a69d4f0dSTobias Grosser } 1369a69d4f0dSTobias Grosser } 1370a69d4f0dSTobias Grosser 13712d1ed0bfSTobias Grosser void VectorBlockGenerator::copyStmt( 13722d1ed0bfSTobias Grosser ScopStmt &Stmt, __isl_keep isl_id_to_ast_expr *NewAccesses) { 137321a059afSTobias Grosser assert(Stmt.isBlockStmt() && 137421a059afSTobias Grosser "TODO: Only block statements can be copied by the vector block " 137521a059afSTobias Grosser "generator"); 1376275a1756SJohannes Doerfert 1377be9c9117SJohannes Doerfert BasicBlock *BB = Stmt.getBasicBlock(); 1378b8f58b53SDuncan P. N. Exon Smith BasicBlock *CopyBB = SplitBlock(Builder.GetInsertBlock(), 1379b8f58b53SDuncan P. N. Exon Smith &*Builder.GetInsertPoint(), &DT, &LI); 13803b11a16aSHongbin Zheng CopyBB->setName("polly.stmt." + BB->getName()); 1381b8f58b53SDuncan P. N. Exon Smith Builder.SetInsertPoint(&CopyBB->front()); 13823b11a16aSHongbin Zheng 13833b11a16aSHongbin Zheng // Create two maps that store the mapping from the original instructions of 13843b11a16aSHongbin Zheng // the old basic block to their copies in the new basic block. Those maps 13853b11a16aSHongbin Zheng // are basic block local. 13863b11a16aSHongbin Zheng // 13873b11a16aSHongbin Zheng // As vector code generation is supported there is one map for scalar values 13883b11a16aSHongbin Zheng // and one for vector values. 13893b11a16aSHongbin Zheng // 13903b11a16aSHongbin Zheng // In case we just do scalar code generation, the vectorMap is not used and 13913b11a16aSHongbin Zheng // the scalarMap has just one dimension, which contains the mapping. 13923b11a16aSHongbin Zheng // 13933b11a16aSHongbin Zheng // In case vector code generation is done, an instruction may either appear 13943b11a16aSHongbin Zheng // in the vector map once (as it is calculating >vectorwidth< values at a 13953b11a16aSHongbin Zheng // time. Or (if the values are calculated using scalar operations), it 13963b11a16aSHongbin Zheng // appears once in every dimension of the scalarMap. 13973b11a16aSHongbin Zheng VectorValueMapT ScalarBlockMap(getVectorWidth()); 13983b11a16aSHongbin Zheng ValueMapT VectorBlockMap; 13993b11a16aSHongbin Zheng 1400a69d4f0dSTobias Grosser generateScalarVectorLoads(Stmt, VectorBlockMap); 1401a69d4f0dSTobias Grosser 140291f5b262STobias Grosser for (Instruction &Inst : *BB) 14032d1ed0bfSTobias Grosser copyInstruction(Stmt, &Inst, VectorBlockMap, ScalarBlockMap, NewAccesses); 1404a69d4f0dSTobias Grosser 1405a69d4f0dSTobias Grosser verifyNoScalarStores(Stmt); 14063b11a16aSHongbin Zheng } 1407275a1756SJohannes Doerfert 1408ecff11dcSJohannes Doerfert BasicBlock *RegionGenerator::repairDominance(BasicBlock *BB, 1409ecff11dcSJohannes Doerfert BasicBlock *BBCopy) { 1410514f6efaSJohannes Doerfert 1411514f6efaSJohannes Doerfert BasicBlock *BBIDom = DT.getNode(BB)->getIDom()->getBlock(); 1412deefbcedSTobias Grosser BasicBlock *BBCopyIDom = EndBlockMap.lookup(BBIDom); 1413514f6efaSJohannes Doerfert 1414514f6efaSJohannes Doerfert if (BBCopyIDom) 1415514f6efaSJohannes Doerfert DT.changeImmediateDominator(BBCopy, BBCopyIDom); 1416514f6efaSJohannes Doerfert 1417deefbcedSTobias Grosser return StartBlockMap.lookup(BBIDom); 1418514f6efaSJohannes Doerfert } 1419514f6efaSJohannes Doerfert 1420f33c125dSMichael Kruse // This is to determine whether an llvm::Value (defined in @p BB) is usable when 1421f33c125dSMichael Kruse // leaving a subregion. The straight-forward DT.dominates(BB, R->getExitBlock()) 1422f33c125dSMichael Kruse // does not work in cases where the exit block has edges from outside the 1423f33c125dSMichael Kruse // region. In that case the llvm::Value would never be usable in in the exit 1424f33c125dSMichael Kruse // block. The RegionGenerator however creates an new exit block ('ExitBBCopy') 1425f33c125dSMichael Kruse // for the subregion's exiting edges only. We need to determine whether an 1426f33c125dSMichael Kruse // llvm::Value is usable in there. We do this by checking whether it dominates 1427f33c125dSMichael Kruse // all exiting blocks individually. 1428f33c125dSMichael Kruse static bool isDominatingSubregionExit(const DominatorTree &DT, Region *R, 1429f33c125dSMichael Kruse BasicBlock *BB) { 1430f33c125dSMichael Kruse for (auto ExitingBB : predecessors(R->getExit())) { 1431f33c125dSMichael Kruse // Check for non-subregion incoming edges. 1432f33c125dSMichael Kruse if (!R->contains(ExitingBB)) 1433f33c125dSMichael Kruse continue; 1434f33c125dSMichael Kruse 1435f33c125dSMichael Kruse if (!DT.dominates(BB, ExitingBB)) 1436f33c125dSMichael Kruse return false; 1437f33c125dSMichael Kruse } 1438f33c125dSMichael Kruse 1439f33c125dSMichael Kruse return true; 1440f33c125dSMichael Kruse } 1441f33c125dSMichael Kruse 1442f33c125dSMichael Kruse // Find the direct dominator of the subregion's exit block if the subregion was 1443f33c125dSMichael Kruse // simplified. 1444f33c125dSMichael Kruse static BasicBlock *findExitDominator(DominatorTree &DT, Region *R) { 1445f33c125dSMichael Kruse BasicBlock *Common = nullptr; 1446f33c125dSMichael Kruse for (auto ExitingBB : predecessors(R->getExit())) { 1447f33c125dSMichael Kruse // Check for non-subregion incoming edges. 1448f33c125dSMichael Kruse if (!R->contains(ExitingBB)) 1449f33c125dSMichael Kruse continue; 1450f33c125dSMichael Kruse 1451f33c125dSMichael Kruse // First exiting edge. 1452f33c125dSMichael Kruse if (!Common) { 1453f33c125dSMichael Kruse Common = ExitingBB; 1454f33c125dSMichael Kruse continue; 1455f33c125dSMichael Kruse } 1456f33c125dSMichael Kruse 1457f33c125dSMichael Kruse Common = DT.findNearestCommonDominator(Common, ExitingBB); 1458f33c125dSMichael Kruse } 1459f33c125dSMichael Kruse 1460f33c125dSMichael Kruse assert(Common && R->contains(Common)); 1461f33c125dSMichael Kruse return Common; 1462f33c125dSMichael Kruse } 1463f33c125dSMichael Kruse 1464bc132607STobias Grosser void RegionGenerator::copyStmt(ScopStmt &Stmt, LoopToScevMapT <S, 14652d1ed0bfSTobias Grosser isl_id_to_ast_expr *IdToAstExp) { 1466275a1756SJohannes Doerfert assert(Stmt.isRegionStmt() && 1467d3f21833STobias Grosser "Only region statements can be copied by the region generator"); 1468275a1756SJohannes Doerfert 1469ecff11dcSJohannes Doerfert // Forget all old mappings. 1470deefbcedSTobias Grosser StartBlockMap.clear(); 1471deefbcedSTobias Grosser EndBlockMap.clear(); 1472ecff11dcSJohannes Doerfert RegionMaps.clear(); 1473ecff11dcSJohannes Doerfert IncompletePHINodeMap.clear(); 1474ecff11dcSJohannes Doerfert 1475225f0d1eSMichael Kruse // Collection of all values related to this subregion. 1476225f0d1eSMichael Kruse ValueMapT ValueMap; 1477225f0d1eSMichael Kruse 1478275a1756SJohannes Doerfert // The region represented by the statement. 1479275a1756SJohannes Doerfert Region *R = Stmt.getRegion(); 1480275a1756SJohannes Doerfert 1481ecff11dcSJohannes Doerfert // Create a dedicated entry for the region where we can reload all demoted 1482ecff11dcSJohannes Doerfert // inputs. 1483ecff11dcSJohannes Doerfert BasicBlock *EntryBB = R->getEntry(); 1484b8f58b53SDuncan P. N. Exon Smith BasicBlock *EntryBBCopy = SplitBlock(Builder.GetInsertBlock(), 1485b8f58b53SDuncan P. N. Exon Smith &*Builder.GetInsertPoint(), &DT, &LI); 1486ecff11dcSJohannes Doerfert EntryBBCopy->setName("polly.stmt." + EntryBB->getName() + ".entry"); 1487b8f58b53SDuncan P. N. Exon Smith Builder.SetInsertPoint(&EntryBBCopy->front()); 1488514f6efaSJohannes Doerfert 1489c993739eSMichael Kruse ValueMapT &EntryBBMap = RegionMaps[EntryBBCopy]; 14902fa35194SMichael Kruse generateScalarLoads(Stmt, LTS, EntryBBMap, IdToAstExp); 1491e819fffeSMichael Kruse generateBeginStmtTrace(Stmt, LTS, EntryBBMap); 1492225f0d1eSMichael Kruse 1493ecff11dcSJohannes Doerfert for (auto PI = pred_begin(EntryBB), PE = pred_end(EntryBB); PI != PE; ++PI) 1494deefbcedSTobias Grosser if (!R->contains(*PI)) { 1495deefbcedSTobias Grosser StartBlockMap[*PI] = EntryBBCopy; 1496deefbcedSTobias Grosser EndBlockMap[*PI] = EntryBBCopy; 1497deefbcedSTobias Grosser } 1498275a1756SJohannes Doerfert 1499275a1756SJohannes Doerfert // Iterate over all blocks in the region in a breadth-first search. 1500275a1756SJohannes Doerfert std::deque<BasicBlock *> Blocks; 15015b1abfc8SMandeep Singh Grang SmallSetVector<BasicBlock *, 8> SeenBlocks; 1502ecff11dcSJohannes Doerfert Blocks.push_back(EntryBB); 1503ecff11dcSJohannes Doerfert SeenBlocks.insert(EntryBB); 1504275a1756SJohannes Doerfert 1505275a1756SJohannes Doerfert while (!Blocks.empty()) { 1506275a1756SJohannes Doerfert BasicBlock *BB = Blocks.front(); 1507275a1756SJohannes Doerfert Blocks.pop_front(); 1508275a1756SJohannes Doerfert 1509514f6efaSJohannes Doerfert // First split the block and update dominance information. 1510514f6efaSJohannes Doerfert BasicBlock *BBCopy = splitBB(BB); 1511ecff11dcSJohannes Doerfert BasicBlock *BBCopyIDom = repairDominance(BB, BBCopy); 1512ecff11dcSJohannes Doerfert 1513c993739eSMichael Kruse // Get the mapping for this block and initialize it with either the scalar 1514c993739eSMichael Kruse // loads from the generated entering block (which dominates all blocks of 1515c993739eSMichael Kruse // this subregion) or the maps of the immediate dominator, if part of the 1516c993739eSMichael Kruse // subregion. The latter necessarily includes the former. 1517c993739eSMichael Kruse ValueMapT *InitBBMap; 1518c993739eSMichael Kruse if (BBCopyIDom) { 1519c993739eSMichael Kruse assert(RegionMaps.count(BBCopyIDom)); 1520c993739eSMichael Kruse InitBBMap = &RegionMaps[BBCopyIDom]; 1521c993739eSMichael Kruse } else 1522c993739eSMichael Kruse InitBBMap = &EntryBBMap; 1523c993739eSMichael Kruse auto Inserted = RegionMaps.insert(std::make_pair(BBCopy, *InitBBMap)); 1524c993739eSMichael Kruse ValueMapT &RegionMap = Inserted.first->second; 1525514f6efaSJohannes Doerfert 1526275a1756SJohannes Doerfert // Copy the block with the BlockGenerator. 1527b8f58b53SDuncan P. N. Exon Smith Builder.SetInsertPoint(&BBCopy->front()); 1528bc132607STobias Grosser copyBB(Stmt, BB, BBCopy, RegionMap, LTS, IdToAstExp); 1529275a1756SJohannes Doerfert 1530ecff11dcSJohannes Doerfert // In order to remap PHI nodes we store also basic block mappings. 1531deefbcedSTobias Grosser StartBlockMap[BB] = BBCopy; 1532deefbcedSTobias Grosser EndBlockMap[BB] = Builder.GetInsertBlock(); 1533ecff11dcSJohannes Doerfert 1534ecff11dcSJohannes Doerfert // Add values to incomplete PHI nodes waiting for this block to be copied. 1535ecff11dcSJohannes Doerfert for (const PHINodePairTy &PHINodePair : IncompletePHINodeMap[BB]) 1536bc132607STobias Grosser addOperandToPHI(Stmt, PHINodePair.first, PHINodePair.second, BB, LTS); 1537ecff11dcSJohannes Doerfert IncompletePHINodeMap[BB].clear(); 1538ecff11dcSJohannes Doerfert 1539275a1756SJohannes Doerfert // And continue with new successors inside the region. 1540275a1756SJohannes Doerfert for (auto SI = succ_begin(BB), SE = succ_end(BB); SI != SE; SI++) 15415b1abfc8SMandeep Singh Grang if (R->contains(*SI) && SeenBlocks.insert(*SI)) 1542275a1756SJohannes Doerfert Blocks.push_back(*SI); 1543ebffcbeeSMichael Kruse 1544ebffcbeeSMichael Kruse // Remember value in case it is visible after this subregion. 1545f33c125dSMichael Kruse if (isDominatingSubregionExit(DT, R, BB)) 1546ebffcbeeSMichael Kruse ValueMap.insert(RegionMap.begin(), RegionMap.end()); 1547275a1756SJohannes Doerfert } 1548275a1756SJohannes Doerfert 1549275a1756SJohannes Doerfert // Now create a new dedicated region exit block and add it to the region map. 1550b8f58b53SDuncan P. N. Exon Smith BasicBlock *ExitBBCopy = SplitBlock(Builder.GetInsertBlock(), 1551b8f58b53SDuncan P. N. Exon Smith &*Builder.GetInsertPoint(), &DT, &LI); 1552ecff11dcSJohannes Doerfert ExitBBCopy->setName("polly.stmt." + R->getExit()->getName() + ".exit"); 1553deefbcedSTobias Grosser StartBlockMap[R->getExit()] = ExitBBCopy; 1554deefbcedSTobias Grosser EndBlockMap[R->getExit()] = ExitBBCopy; 1555514f6efaSJohannes Doerfert 1556deefbcedSTobias Grosser BasicBlock *ExitDomBBCopy = EndBlockMap.lookup(findExitDominator(DT, R)); 155721a059afSTobias Grosser assert(ExitDomBBCopy && 155821a059afSTobias Grosser "Common exit dominator must be within region; at least the entry node " 155921a059afSTobias Grosser "must match"); 1560f33c125dSMichael Kruse DT.changeImmediateDominator(ExitBBCopy, ExitDomBBCopy); 1561275a1756SJohannes Doerfert 1562275a1756SJohannes Doerfert // As the block generator doesn't handle control flow we need to add the 1563275a1756SJohannes Doerfert // region control flow by hand after all blocks have been copied. 1564275a1756SJohannes Doerfert for (BasicBlock *BB : SeenBlocks) { 1565275a1756SJohannes Doerfert 1566deefbcedSTobias Grosser BasicBlock *BBCopyStart = StartBlockMap[BB]; 1567deefbcedSTobias Grosser BasicBlock *BBCopyEnd = EndBlockMap[BB]; 1568e303c87eSChandler Carruth Instruction *TI = BB->getTerminator(); 1569e114dc02SJohannes Doerfert if (isa<UnreachableInst>(TI)) { 1570deefbcedSTobias Grosser while (!BBCopyEnd->empty()) 1571deefbcedSTobias Grosser BBCopyEnd->begin()->eraseFromParent(); 1572deefbcedSTobias Grosser new UnreachableInst(BBCopyEnd->getContext(), BBCopyEnd); 1573e114dc02SJohannes Doerfert continue; 1574e114dc02SJohannes Doerfert } 1575e114dc02SJohannes Doerfert 1576deefbcedSTobias Grosser Instruction *BICopy = BBCopyEnd->getTerminator(); 1577275a1756SJohannes Doerfert 1578deefbcedSTobias Grosser ValueMapT &RegionMap = RegionMaps[BBCopyStart]; 1579deefbcedSTobias Grosser RegionMap.insert(StartBlockMap.begin(), StartBlockMap.end()); 1580514f6efaSJohannes Doerfert 158145e7944bSTobias Grosser Builder.SetInsertPoint(BICopy); 15829a132f36SJohannes Doerfert copyInstScalar(Stmt, TI, RegionMap, LTS); 1583275a1756SJohannes Doerfert BICopy->eraseFromParent(); 1584275a1756SJohannes Doerfert } 1585275a1756SJohannes Doerfert 1586ecff11dcSJohannes Doerfert // Add counting PHI nodes to all loops in the region that can be used as 1587a6d48f59SMichael Kruse // replacement for SCEVs referring to the old loop. 1588ecff11dcSJohannes Doerfert for (BasicBlock *BB : SeenBlocks) { 1589ecff11dcSJohannes Doerfert Loop *L = LI.getLoopFor(BB); 1590bc29e0b2STobias Grosser if (L == nullptr || L->getHeader() != BB || !R->contains(L)) 1591ecff11dcSJohannes Doerfert continue; 1592ecff11dcSJohannes Doerfert 1593deefbcedSTobias Grosser BasicBlock *BBCopy = StartBlockMap[BB]; 1594ecff11dcSJohannes Doerfert Value *NullVal = Builder.getInt32(0); 1595ecff11dcSJohannes Doerfert PHINode *LoopPHI = 1596ecff11dcSJohannes Doerfert PHINode::Create(Builder.getInt32Ty(), 2, "polly.subregion.iv"); 1597ecff11dcSJohannes Doerfert Instruction *LoopPHIInc = BinaryOperator::CreateAdd( 1598ecff11dcSJohannes Doerfert LoopPHI, Builder.getInt32(1), "polly.subregion.iv.inc"); 1599b8f58b53SDuncan P. N. Exon Smith LoopPHI->insertBefore(&BBCopy->front()); 1600ecff11dcSJohannes Doerfert LoopPHIInc->insertBefore(BBCopy->getTerminator()); 1601ecff11dcSJohannes Doerfert 1602ecff11dcSJohannes Doerfert for (auto *PredBB : make_range(pred_begin(BB), pred_end(BB))) { 1603ecff11dcSJohannes Doerfert if (!R->contains(PredBB)) 1604ecff11dcSJohannes Doerfert continue; 1605ecff11dcSJohannes Doerfert if (L->contains(PredBB)) 1606deefbcedSTobias Grosser LoopPHI->addIncoming(LoopPHIInc, EndBlockMap[PredBB]); 1607ecff11dcSJohannes Doerfert else 1608deefbcedSTobias Grosser LoopPHI->addIncoming(NullVal, EndBlockMap[PredBB]); 1609ecff11dcSJohannes Doerfert } 1610ecff11dcSJohannes Doerfert 1611ecff11dcSJohannes Doerfert for (auto *PredBBCopy : make_range(pred_begin(BBCopy), pred_end(BBCopy))) 1612ecff11dcSJohannes Doerfert if (LoopPHI->getBasicBlockIndex(PredBBCopy) < 0) 1613ecff11dcSJohannes Doerfert LoopPHI->addIncoming(NullVal, PredBBCopy); 1614ecff11dcSJohannes Doerfert 1615ecff11dcSJohannes Doerfert LTS[L] = SE.getUnknown(LoopPHI); 1616ecff11dcSJohannes Doerfert } 1617ecff11dcSJohannes Doerfert 1618225f0d1eSMichael Kruse // Continue generating code in the exit block. 1619b8f58b53SDuncan P. N. Exon Smith Builder.SetInsertPoint(&*ExitBBCopy->getFirstInsertionPt()); 1620225f0d1eSMichael Kruse 1621225f0d1eSMichael Kruse // Write values visible to other statements. 16222fa35194SMichael Kruse generateScalarStores(Stmt, LTS, ValueMap, IdToAstExp); 1623deefbcedSTobias Grosser StartBlockMap.clear(); 1624deefbcedSTobias Grosser EndBlockMap.clear(); 16253e956020STobias Grosser RegionMaps.clear(); 16263e956020STobias Grosser IncompletePHINodeMap.clear(); 1627275a1756SJohannes Doerfert } 1628ecff11dcSJohannes Doerfert 1629ee6a4fc6SMichael Kruse PHINode *RegionGenerator::buildExitPHI(MemoryAccess *MA, LoopToScevMapT <S, 1630ee6a4fc6SMichael Kruse ValueMapT &BBMap, Loop *L) { 1631ee6a4fc6SMichael Kruse ScopStmt *Stmt = MA->getStatement(); 1632ee6a4fc6SMichael Kruse Region *SubR = Stmt->getRegion(); 1633ee6a4fc6SMichael Kruse auto Incoming = MA->getIncoming(); 1634ee6a4fc6SMichael Kruse 1635ee6a4fc6SMichael Kruse PollyIRBuilder::InsertPointGuard IPGuard(Builder); 1636ee6a4fc6SMichael Kruse PHINode *OrigPHI = cast<PHINode>(MA->getAccessInstruction()); 1637ee6a4fc6SMichael Kruse BasicBlock *NewSubregionExit = Builder.GetInsertBlock(); 1638ee6a4fc6SMichael Kruse 1639ee6a4fc6SMichael Kruse // This can happen if the subregion is simplified after the ScopStmts 1640ee6a4fc6SMichael Kruse // have been created; simplification happens as part of CodeGeneration. 1641ee6a4fc6SMichael Kruse if (OrigPHI->getParent() != SubR->getExit()) { 1642ee6a4fc6SMichael Kruse BasicBlock *FormerExit = SubR->getExitingBlock(); 1643ee6a4fc6SMichael Kruse if (FormerExit) 1644deefbcedSTobias Grosser NewSubregionExit = StartBlockMap.lookup(FormerExit); 1645ee6a4fc6SMichael Kruse } 1646ee6a4fc6SMichael Kruse 1647ee6a4fc6SMichael Kruse PHINode *NewPHI = PHINode::Create(OrigPHI->getType(), Incoming.size(), 1648ee6a4fc6SMichael Kruse "polly." + OrigPHI->getName(), 1649ee6a4fc6SMichael Kruse NewSubregionExit->getFirstNonPHI()); 1650ee6a4fc6SMichael Kruse 1651ee6a4fc6SMichael Kruse // Add the incoming values to the PHI. 1652ee6a4fc6SMichael Kruse for (auto &Pair : Incoming) { 1653ee6a4fc6SMichael Kruse BasicBlock *OrigIncomingBlock = Pair.first; 1654deefbcedSTobias Grosser BasicBlock *NewIncomingBlockStart = StartBlockMap.lookup(OrigIncomingBlock); 1655deefbcedSTobias Grosser BasicBlock *NewIncomingBlockEnd = EndBlockMap.lookup(OrigIncomingBlock); 1656deefbcedSTobias Grosser Builder.SetInsertPoint(NewIncomingBlockEnd->getTerminator()); 1657deefbcedSTobias Grosser assert(RegionMaps.count(NewIncomingBlockStart)); 1658deefbcedSTobias Grosser assert(RegionMaps.count(NewIncomingBlockEnd)); 1659deefbcedSTobias Grosser ValueMapT *LocalBBMap = &RegionMaps[NewIncomingBlockStart]; 1660ee6a4fc6SMichael Kruse 1661ee6a4fc6SMichael Kruse Value *OrigIncomingValue = Pair.second; 1662ee6a4fc6SMichael Kruse Value *NewIncomingValue = 1663ee6a4fc6SMichael Kruse getNewValue(*Stmt, OrigIncomingValue, *LocalBBMap, LTS, L); 1664deefbcedSTobias Grosser NewPHI->addIncoming(NewIncomingValue, NewIncomingBlockEnd); 1665ee6a4fc6SMichael Kruse } 1666ee6a4fc6SMichael Kruse 1667ee6a4fc6SMichael Kruse return NewPHI; 1668ee6a4fc6SMichael Kruse } 1669ee6a4fc6SMichael Kruse 1670ee6a4fc6SMichael Kruse Value *RegionGenerator::getExitScalar(MemoryAccess *MA, LoopToScevMapT <S, 1671ee6a4fc6SMichael Kruse ValueMapT &BBMap) { 1672ee6a4fc6SMichael Kruse ScopStmt *Stmt = MA->getStatement(); 1673ee6a4fc6SMichael Kruse 1674ee6a4fc6SMichael Kruse // TODO: Add some test cases that ensure this is really the right choice. 1675ee6a4fc6SMichael Kruse Loop *L = LI.getLoopFor(Stmt->getRegion()->getExit()); 1676ee6a4fc6SMichael Kruse 1677ee6a4fc6SMichael Kruse if (MA->isAnyPHIKind()) { 1678ee6a4fc6SMichael Kruse auto Incoming = MA->getIncoming(); 1679ee6a4fc6SMichael Kruse assert(!Incoming.empty() && 1680ee6a4fc6SMichael Kruse "PHI WRITEs must have originate from at least one incoming block"); 1681ee6a4fc6SMichael Kruse 1682ee6a4fc6SMichael Kruse // If there is only one incoming value, we do not need to create a PHI. 1683ee6a4fc6SMichael Kruse if (Incoming.size() == 1) { 1684ee6a4fc6SMichael Kruse Value *OldVal = Incoming[0].second; 1685ee6a4fc6SMichael Kruse return getNewValue(*Stmt, OldVal, BBMap, LTS, L); 1686ee6a4fc6SMichael Kruse } 1687ee6a4fc6SMichael Kruse 1688ee6a4fc6SMichael Kruse return buildExitPHI(MA, LTS, BBMap, L); 1689ee6a4fc6SMichael Kruse } 1690ee6a4fc6SMichael Kruse 16914d5a9172STobias Grosser // MemoryKind::Value accesses leaving the subregion must dominate the exit 16924d5a9172STobias Grosser // block; just pass the copied value. 1693ee6a4fc6SMichael Kruse Value *OldVal = MA->getAccessValue(); 1694ee6a4fc6SMichael Kruse return getNewValue(*Stmt, OldVal, BBMap, LTS, L); 1695ee6a4fc6SMichael Kruse } 1696ee6a4fc6SMichael Kruse 16972fa35194SMichael Kruse void RegionGenerator::generateScalarStores( 16982fa35194SMichael Kruse ScopStmt &Stmt, LoopToScevMapT <S, ValueMapT &BBMap, 16992fa35194SMichael Kruse __isl_keep isl_id_to_ast_expr *NewAccesses) { 170075296901STobias Grosser assert(Stmt.getRegion() && 170175296901STobias Grosser "Block statements need to use the generateScalarStores() " 1702ecff11dcSJohannes Doerfert "function in the BlockGenerator"); 1703ecff11dcSJohannes Doerfert 1704ecff11dcSJohannes Doerfert for (MemoryAccess *MA : Stmt) { 17052fa35194SMichael Kruse if (MA->isOriginalArrayKind() || MA->isRead()) 1706ecff11dcSJohannes Doerfert continue; 1707ecff11dcSJohannes Doerfert 17081515f6b9STobias Grosser isl::set AccDom = MA->getAccessRelation().domain(); 1709fe46c3ffSTobias Grosser std::string Subject = MA->getId().get_name(); 1710fe46c3ffSTobias Grosser generateConditionalExecution( 1711fe46c3ffSTobias Grosser Stmt, AccDom, Subject.c_str(), [&, this, MA]() { 1712ee6a4fc6SMichael Kruse Value *NewVal = getExitScalar(MA, LTS, BBMap); 1713fe46c3ffSTobias Grosser Value *Address = getImplicitAddress(*MA, getLoopForStmt(Stmt), LTS, 1714fe46c3ffSTobias Grosser BBMap, NewAccesses); 1715eac9726eSMichael Kruse assert((!isa<Instruction>(NewVal) || 1716eac9726eSMichael Kruse DT.dominates(cast<Instruction>(NewVal)->getParent(), 1717eac9726eSMichael Kruse Builder.GetInsertBlock())) && 1718eac9726eSMichael Kruse "Domination violation"); 1719eac9726eSMichael Kruse assert((!isa<Instruction>(Address) || 1720eac9726eSMichael Kruse DT.dominates(cast<Instruction>(Address)->getParent(), 1721eac9726eSMichael Kruse Builder.GetInsertBlock())) && 1722eac9726eSMichael Kruse "Domination violation"); 1723ee6a4fc6SMichael Kruse Builder.CreateStore(NewVal, Address); 1724706f79abSMichael Kruse }); 1725ecff11dcSJohannes Doerfert } 1726ecff11dcSJohannes Doerfert } 1727ecff11dcSJohannes Doerfert 1728943c369cSTobias Grosser void RegionGenerator::addOperandToPHI(ScopStmt &Stmt, PHINode *PHI, 1729ecff11dcSJohannes Doerfert PHINode *PHICopy, BasicBlock *IncomingBB, 1730ecff11dcSJohannes Doerfert LoopToScevMapT <S) { 1731ecff11dcSJohannes Doerfert // If the incoming block was not yet copied mark this PHI as incomplete. 1732ecff11dcSJohannes Doerfert // Once the block will be copied the incoming value will be added. 1733deefbcedSTobias Grosser BasicBlock *BBCopyStart = StartBlockMap[IncomingBB]; 1734deefbcedSTobias Grosser BasicBlock *BBCopyEnd = EndBlockMap[IncomingBB]; 1735deefbcedSTobias Grosser if (!BBCopyStart) { 1736deefbcedSTobias Grosser assert(!BBCopyEnd); 17378d89179eSMichael Kruse assert(Stmt.represents(IncomingBB) && 1738ecff11dcSJohannes Doerfert "Bad incoming block for PHI in non-affine region"); 1739ecff11dcSJohannes Doerfert IncompletePHINodeMap[IncomingBB].push_back(std::make_pair(PHI, PHICopy)); 1740ecff11dcSJohannes Doerfert return; 1741ecff11dcSJohannes Doerfert } 1742ecff11dcSJohannes Doerfert 1743deefbcedSTobias Grosser assert(RegionMaps.count(BBCopyStart) && 1744deefbcedSTobias Grosser "Incoming PHI block did not have a BBMap"); 1745deefbcedSTobias Grosser ValueMapT &BBCopyMap = RegionMaps[BBCopyStart]; 1746ecff11dcSJohannes Doerfert 174775dfaa1dSTobias Grosser Value *OpCopy = nullptr; 174875dfaa1dSTobias Grosser 17498d89179eSMichael Kruse if (Stmt.represents(IncomingBB)) { 1750ecff11dcSJohannes Doerfert Value *Op = PHI->getIncomingValueForBlock(IncomingBB); 1751dc122222SMichael Kruse 17526ba92714SJohannes Doerfert // If the current insert block is different from the PHIs incoming block 17536ba92714SJohannes Doerfert // change it, otherwise do not. 17546ba92714SJohannes Doerfert auto IP = Builder.GetInsertPoint(); 1755deefbcedSTobias Grosser if (IP->getParent() != BBCopyEnd) 1756deefbcedSTobias Grosser Builder.SetInsertPoint(BBCopyEnd->getTerminator()); 17572c3ffc04SJohannes Doerfert OpCopy = getNewValue(Stmt, Op, BBCopyMap, LTS, getLoopForStmt(Stmt)); 1758deefbcedSTobias Grosser if (IP->getParent() != BBCopyEnd) 17596ba92714SJohannes Doerfert Builder.SetInsertPoint(&*IP); 1760ecff11dcSJohannes Doerfert } else { 176175dfaa1dSTobias Grosser // All edges from outside the non-affine region become a single edge 176275dfaa1dSTobias Grosser // in the new copy of the non-affine region. Make sure to only add the 176375dfaa1dSTobias Grosser // corresponding edge the first time we encounter a basic block from 176475dfaa1dSTobias Grosser // outside the non-affine region. 1765deefbcedSTobias Grosser if (PHICopy->getBasicBlockIndex(BBCopyEnd) >= 0) 1766ecff11dcSJohannes Doerfert return; 1767ecff11dcSJohannes Doerfert 176875dfaa1dSTobias Grosser // Get the reloaded value. 176975dfaa1dSTobias Grosser OpCopy = getNewValue(Stmt, PHI, BBCopyMap, LTS, getLoopForStmt(Stmt)); 1770ecff11dcSJohannes Doerfert } 1771ecff11dcSJohannes Doerfert 1772ecff11dcSJohannes Doerfert assert(OpCopy && "Incoming PHI value was not copied properly"); 1773deefbcedSTobias Grosser PHICopy->addIncoming(OpCopy, BBCopyEnd); 1774ecff11dcSJohannes Doerfert } 1775ecff11dcSJohannes Doerfert 17762b809d13STobias Grosser void RegionGenerator::copyPHIInstruction(ScopStmt &Stmt, PHINode *PHI, 1777ecff11dcSJohannes Doerfert ValueMapT &BBMap, 1778ecff11dcSJohannes Doerfert LoopToScevMapT <S) { 1779ecff11dcSJohannes Doerfert unsigned NumIncoming = PHI->getNumIncomingValues(); 1780ecff11dcSJohannes Doerfert PHINode *PHICopy = 1781ecff11dcSJohannes Doerfert Builder.CreatePHI(PHI->getType(), NumIncoming, "polly." + PHI->getName()); 1782ecff11dcSJohannes Doerfert PHICopy->moveBefore(PHICopy->getParent()->getFirstNonPHI()); 1783ecff11dcSJohannes Doerfert BBMap[PHI] = PHICopy; 1784ecff11dcSJohannes Doerfert 178597b84909STobias Grosser for (BasicBlock *IncomingBB : PHI->blocks()) 178697b84909STobias Grosser addOperandToPHI(Stmt, PHI, PHICopy, IncomingBB, LTS); 1787ecff11dcSJohannes Doerfert } 1788