averello/swift-mirror
1
0
Fork 0
mirror of https://github.com/apple/swift.git synced 2025-12-21 12:14:44 +01:00
Code Issues Packages Projects Releases Wiki Activity

"[NFC] Add tests for LoopUnroll on OSSA""

Browse Source 
Reland #35423
This commit is contained in:
Meghana Gupta
2021-01-20 21:47:43 -08:00
parent 8d217f362f
commit 3b60184f53
2 changed files with 731 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
lib/SILOptimizer/LoopTransforms/LoopUnroll.cpp
View File

@@ -347,10 +347,11 @@ updateSSA(SILModule &M, SILLoop *Loop,
}
/// Try to fully unroll the loop if we can determine the trip count and the trip
/// count lis below a threshold.
/// count is below a threshold.
static bool tryToUnrollLoop(SILLoop *Loop) {
assert(Loop->getSubLoops().empty() && "Expecting innermost loops");
LLVM_DEBUG(llvm::dbgs() << "Trying to unroll loop : \n" << *Loop);
auto *Preheader = Loop->getLoopPreheader();
if (!Preheader)
return false;
@@ -364,11 +365,15 @@ static bool tryToUnrollLoop(SILLoop *Loop) {
Optional<uint64_t> MaxTripCount =
getMaxLoopTripCount(Loop, Preheader, Header, Latch);
if (!MaxTripCount)
if (!MaxTripCount) {
LLVM_DEBUG(llvm::dbgs() << "Not unrolling, did not find trip count\n");
return false;
}
if (!canAndShouldUnrollLoop(Loop, MaxTripCount.getValue()))
if (!canAndShouldUnrollLoop(Loop, MaxTripCount.getValue())) {
LLVM_DEBUG(llvm::dbgs() << "Not unrolling, exceeds cost threshold\n");
return false;
}
// TODO: We need to split edges from non-condbr exits for the SSA updater. For
// now just don't handle loops containing such exits.
@@ -443,7 +448,6 @@ class LoopUnrolling : public SILFunctionTransform {
void run() override {
bool Changed = false;
auto *Fun = getFunction();
SILLoopInfo *LoopInfo = PM->getAnalysis<SILLoopAnalysis>()->get(Fun);
@@ -462,6 +466,12 @@ class LoopUnrolling : public SILFunctionTransform {
}
}
if (InnermostLoops.empty())
return;
LLVM_DEBUG(llvm::dbgs() << "Loop Unroll running on function : "
<< Fun->getName() << "\n");
// Try to unroll innermost loops.
for (auto *Loop : InnermostLoops)
Changed |= tryToUnrollLoop(Loop);