2017-01-16 05:17:52 +08:00
|
|
|
//===- LoopInfoTest.cpp - LoopInfo unit tests -----------------------------===//
|
|
|
|
//
|
2019-01-19 16:50:56 +08:00
|
|
|
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
|
|
|
|
// See https://llvm.org/LICENSE.txt for license information.
|
|
|
|
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
|
2017-01-16 05:17:52 +08:00
|
|
|
//
|
|
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
|
|
|
|
#include "llvm/Analysis/LoopInfo.h"
|
[LOOPINFO] Extend Loop object to add utilities to get the loop bounds,
step, and loop induction variable.
Summary: This PR extends the loop object with more utilities to get loop
bounds, step, and loop induction variable. There already exists passes
which try to obtain the loop induction variable in their own pass, e.g.
loop interchange. It would be useful to have a common area to get these
information.
/// Example:
/// for (int i = lb; i < ub; i+=step)
/// <loop body>
/// --- pseudo LLVMIR ---
/// beforeloop:
/// guardcmp = (lb < ub)
/// if (guardcmp) goto preheader; else goto afterloop
/// preheader:
/// loop:
/// i1 = phi[{lb, preheader}, {i2, latch}]
/// <loop body>
/// i2 = i1 + step
/// latch:
/// cmp = (i2 < ub)
/// if (cmp) goto loop
/// exit:
/// afterloop:
///
/// getBounds
/// getInitialIVValue --> lb
/// getStepInst --> i2 = i1 + step
/// getStepValue --> step
/// getFinalIVValue --> ub
/// getCanonicalPredicate --> '<'
/// getDirection --> Increasing
/// getInductionVariable --> i1
/// getAuxiliaryInductionVariable --> {i1}
/// isCanonical --> false
Reviewers: kbarton, hfinkel, dmgreen, Meinersbur, jdoerfert, syzaara,
fhahn
Reviewed By: kbarton
Subscribers: tvvikram, bmahjour, etiotto, fhahn, jsji, hiraditya,
llvm-commits
Tag: LLVM
Differential Revision: https://reviews.llvm.org/D60565
llvm-svn: 362644
2019-06-06 04:42:47 +08:00
|
|
|
#include "llvm/Analysis/AssumptionCache.h"
|
|
|
|
#include "llvm/Analysis/ScalarEvolution.h"
|
|
|
|
#include "llvm/Analysis/TargetLibraryInfo.h"
|
2017-01-16 05:17:52 +08:00
|
|
|
#include "llvm/AsmParser/Parser.h"
|
|
|
|
#include "llvm/IR/Dominators.h"
|
|
|
|
#include "llvm/Support/SourceMgr.h"
|
|
|
|
#include "gtest/gtest.h"
|
|
|
|
|
|
|
|
using namespace llvm;
|
|
|
|
|
2017-01-18 04:24:39 +08:00
|
|
|
/// Build the loop info for the function and run the Test.
|
|
|
|
static void
|
|
|
|
runWithLoopInfo(Module &M, StringRef FuncName,
|
|
|
|
function_ref<void(Function &F, LoopInfo &LI)> Test) {
|
|
|
|
auto *F = M.getFunction(FuncName);
|
|
|
|
ASSERT_NE(F, nullptr) << "Could not find " << FuncName;
|
|
|
|
// Compute the dominator tree and the loop info for the function.
|
|
|
|
DominatorTree DT(*F);
|
|
|
|
LoopInfo LI(DT);
|
|
|
|
Test(*F, LI);
|
|
|
|
}
|
|
|
|
|
[LOOPINFO] Extend Loop object to add utilities to get the loop bounds,
step, and loop induction variable.
Summary: This PR extends the loop object with more utilities to get loop
bounds, step, and loop induction variable. There already exists passes
which try to obtain the loop induction variable in their own pass, e.g.
loop interchange. It would be useful to have a common area to get these
information.
/// Example:
/// for (int i = lb; i < ub; i+=step)
/// <loop body>
/// --- pseudo LLVMIR ---
/// beforeloop:
/// guardcmp = (lb < ub)
/// if (guardcmp) goto preheader; else goto afterloop
/// preheader:
/// loop:
/// i1 = phi[{lb, preheader}, {i2, latch}]
/// <loop body>
/// i2 = i1 + step
/// latch:
/// cmp = (i2 < ub)
/// if (cmp) goto loop
/// exit:
/// afterloop:
///
/// getBounds
/// getInitialIVValue --> lb
/// getStepInst --> i2 = i1 + step
/// getStepValue --> step
/// getFinalIVValue --> ub
/// getCanonicalPredicate --> '<'
/// getDirection --> Increasing
/// getInductionVariable --> i1
/// getAuxiliaryInductionVariable --> {i1}
/// isCanonical --> false
Reviewers: kbarton, hfinkel, dmgreen, Meinersbur, jdoerfert, syzaara,
fhahn
Reviewed By: kbarton
Subscribers: tvvikram, bmahjour, etiotto, fhahn, jsji, hiraditya,
llvm-commits
Tag: LLVM
Differential Revision: https://reviews.llvm.org/D60565
llvm-svn: 362644
2019-06-06 04:42:47 +08:00
|
|
|
/// Build the loop info and scalar evolution for the function and run the Test.
|
|
|
|
static void runWithLoopInfoPlus(
|
|
|
|
Module &M, StringRef FuncName,
|
[LOOPINFO] Introduce the loop guard API.
Summary:
This is the first patch for the loop guard. We introduced
getLoopGuardBranch() and isGuarded().
This currently only works on simplified loop, as it requires a preheader
and a latch to identify the guard.
It will work on loops of the form:
/// GuardBB:
/// br cond1, Preheader, ExitSucc <== GuardBranch
/// Preheader:
/// br Header
/// Header:
/// ...
/// br Latch
/// Latch:
/// br cond2, Header, ExitBlock
/// ExitBlock:
/// br ExitSucc
/// ExitSucc:
Prior discussions leading upto the decision to introduce the loop guard
API: http://lists.llvm.org/pipermail/llvm-dev/2019-May/132607.html
Reviewer: reames, kbarton, hfinkel, jdoerfert, Meinersbur, dmgreen
Reviewed By: reames
Subscribers: wuzish, hiraditya, jsji, llvm-commits, bmahjour, etiotto
Tag: LLVM
Differential Revision: https://reviews.llvm.org/D63885
llvm-svn: 367033
2019-07-26 00:13:18 +08:00
|
|
|
function_ref<void(Function &F, LoopInfo &LI, ScalarEvolution &SE)> Test) {
|
[LOOPINFO] Extend Loop object to add utilities to get the loop bounds,
step, and loop induction variable.
Summary: This PR extends the loop object with more utilities to get loop
bounds, step, and loop induction variable. There already exists passes
which try to obtain the loop induction variable in their own pass, e.g.
loop interchange. It would be useful to have a common area to get these
information.
/// Example:
/// for (int i = lb; i < ub; i+=step)
/// <loop body>
/// --- pseudo LLVMIR ---
/// beforeloop:
/// guardcmp = (lb < ub)
/// if (guardcmp) goto preheader; else goto afterloop
/// preheader:
/// loop:
/// i1 = phi[{lb, preheader}, {i2, latch}]
/// <loop body>
/// i2 = i1 + step
/// latch:
/// cmp = (i2 < ub)
/// if (cmp) goto loop
/// exit:
/// afterloop:
///
/// getBounds
/// getInitialIVValue --> lb
/// getStepInst --> i2 = i1 + step
/// getStepValue --> step
/// getFinalIVValue --> ub
/// getCanonicalPredicate --> '<'
/// getDirection --> Increasing
/// getInductionVariable --> i1
/// getAuxiliaryInductionVariable --> {i1}
/// isCanonical --> false
Reviewers: kbarton, hfinkel, dmgreen, Meinersbur, jdoerfert, syzaara,
fhahn
Reviewed By: kbarton
Subscribers: tvvikram, bmahjour, etiotto, fhahn, jsji, hiraditya,
llvm-commits
Tag: LLVM
Differential Revision: https://reviews.llvm.org/D60565
llvm-svn: 362644
2019-06-06 04:42:47 +08:00
|
|
|
auto *F = M.getFunction(FuncName);
|
|
|
|
ASSERT_NE(F, nullptr) << "Could not find " << FuncName;
|
|
|
|
|
|
|
|
TargetLibraryInfoImpl TLII;
|
|
|
|
TargetLibraryInfo TLI(TLII);
|
|
|
|
AssumptionCache AC(*F);
|
|
|
|
DominatorTree DT(*F);
|
|
|
|
LoopInfo LI(DT);
|
|
|
|
ScalarEvolution SE(*F, TLI, AC, DT, LI);
|
[LOOPINFO] Introduce the loop guard API.
Summary:
This is the first patch for the loop guard. We introduced
getLoopGuardBranch() and isGuarded().
This currently only works on simplified loop, as it requires a preheader
and a latch to identify the guard.
It will work on loops of the form:
/// GuardBB:
/// br cond1, Preheader, ExitSucc <== GuardBranch
/// Preheader:
/// br Header
/// Header:
/// ...
/// br Latch
/// Latch:
/// br cond2, Header, ExitBlock
/// ExitBlock:
/// br ExitSucc
/// ExitSucc:
Prior discussions leading upto the decision to introduce the loop guard
API: http://lists.llvm.org/pipermail/llvm-dev/2019-May/132607.html
Reviewer: reames, kbarton, hfinkel, jdoerfert, Meinersbur, dmgreen
Reviewed By: reames
Subscribers: wuzish, hiraditya, jsji, llvm-commits, bmahjour, etiotto
Tag: LLVM
Differential Revision: https://reviews.llvm.org/D63885
llvm-svn: 367033
2019-07-26 00:13:18 +08:00
|
|
|
Test(*F, LI, SE);
|
[LOOPINFO] Extend Loop object to add utilities to get the loop bounds,
step, and loop induction variable.
Summary: This PR extends the loop object with more utilities to get loop
bounds, step, and loop induction variable. There already exists passes
which try to obtain the loop induction variable in their own pass, e.g.
loop interchange. It would be useful to have a common area to get these
information.
/// Example:
/// for (int i = lb; i < ub; i+=step)
/// <loop body>
/// --- pseudo LLVMIR ---
/// beforeloop:
/// guardcmp = (lb < ub)
/// if (guardcmp) goto preheader; else goto afterloop
/// preheader:
/// loop:
/// i1 = phi[{lb, preheader}, {i2, latch}]
/// <loop body>
/// i2 = i1 + step
/// latch:
/// cmp = (i2 < ub)
/// if (cmp) goto loop
/// exit:
/// afterloop:
///
/// getBounds
/// getInitialIVValue --> lb
/// getStepInst --> i2 = i1 + step
/// getStepValue --> step
/// getFinalIVValue --> ub
/// getCanonicalPredicate --> '<'
/// getDirection --> Increasing
/// getInductionVariable --> i1
/// getAuxiliaryInductionVariable --> {i1}
/// isCanonical --> false
Reviewers: kbarton, hfinkel, dmgreen, Meinersbur, jdoerfert, syzaara,
fhahn
Reviewed By: kbarton
Subscribers: tvvikram, bmahjour, etiotto, fhahn, jsji, hiraditya,
llvm-commits
Tag: LLVM
Differential Revision: https://reviews.llvm.org/D60565
llvm-svn: 362644
2019-06-06 04:42:47 +08:00
|
|
|
}
|
|
|
|
|
2017-01-16 05:17:52 +08:00
|
|
|
static std::unique_ptr<Module> makeLLVMModule(LLVMContext &Context,
|
|
|
|
const char *ModuleStr) {
|
|
|
|
SMDiagnostic Err;
|
|
|
|
return parseAssemblyString(ModuleStr, Err, Context);
|
|
|
|
}
|
|
|
|
|
|
|
|
// This tests that for a loop with a single latch, we get the loop id from
|
|
|
|
// its only latch, even in case the loop may not be in a simplified form.
|
|
|
|
TEST(LoopInfoTest, LoopWithSingleLatch) {
|
|
|
|
const char *ModuleStr =
|
2017-01-18 04:24:39 +08:00
|
|
|
"target datalayout = \"e-m:o-i64:64-f80:128-n8:16:32:64-S128\"\n"
|
|
|
|
"define void @foo(i32 %n) {\n"
|
|
|
|
"entry:\n"
|
|
|
|
" br i1 undef, label %for.cond, label %for.end\n"
|
|
|
|
"for.cond:\n"
|
|
|
|
" %i.0 = phi i32 [ 0, %entry ], [ %inc, %for.inc ]\n"
|
|
|
|
" %cmp = icmp slt i32 %i.0, %n\n"
|
|
|
|
" br i1 %cmp, label %for.inc, label %for.end\n"
|
|
|
|
"for.inc:\n"
|
|
|
|
" %inc = add nsw i32 %i.0, 1\n"
|
|
|
|
" br label %for.cond, !llvm.loop !0\n"
|
|
|
|
"for.end:\n"
|
|
|
|
" ret void\n"
|
|
|
|
"}\n"
|
|
|
|
"!0 = distinct !{!0, !1}\n"
|
|
|
|
"!1 = !{!\"llvm.loop.distribute.enable\", i1 true}\n";
|
|
|
|
|
2017-01-16 05:17:52 +08:00
|
|
|
// Parse the module.
|
|
|
|
LLVMContext Context;
|
|
|
|
std::unique_ptr<Module> M = makeLLVMModule(Context, ModuleStr);
|
|
|
|
|
2017-01-18 04:24:39 +08:00
|
|
|
runWithLoopInfo(*M, "foo", [&](Function &F, LoopInfo &LI) {
|
|
|
|
Function::iterator FI = F.begin();
|
|
|
|
// First basic block is entry - skip it.
|
|
|
|
BasicBlock *Header = &*(++FI);
|
|
|
|
assert(Header->getName() == "for.cond");
|
|
|
|
Loop *L = LI.getLoopFor(Header);
|
2017-01-16 05:17:52 +08:00
|
|
|
|
2017-01-18 04:24:39 +08:00
|
|
|
// This loop is not in simplified form.
|
|
|
|
EXPECT_FALSE(L->isLoopSimplifyForm());
|
2017-01-16 05:17:52 +08:00
|
|
|
|
2017-01-18 04:24:39 +08:00
|
|
|
// Analyze the loop metadata id.
|
|
|
|
bool loopIDFoundAndSet = false;
|
|
|
|
// Try to get and set the metadata id for the loop.
|
|
|
|
if (MDNode *D = L->getLoopID()) {
|
|
|
|
L->setLoopID(D);
|
|
|
|
loopIDFoundAndSet = true;
|
|
|
|
}
|
2017-01-16 05:17:52 +08:00
|
|
|
|
2017-01-18 04:24:39 +08:00
|
|
|
// We must have successfully found and set the loop id in the
|
|
|
|
// only latch the loop has.
|
|
|
|
EXPECT_TRUE(loopIDFoundAndSet);
|
|
|
|
});
|
2017-01-16 05:17:52 +08:00
|
|
|
}
|
2017-01-20 10:41:20 +08:00
|
|
|
|
2018-09-11 19:44:17 +08:00
|
|
|
// Test loop id handling for a loop with multiple latches.
|
|
|
|
TEST(LoopInfoTest, LoopWithMultipleLatches) {
|
|
|
|
const char *ModuleStr =
|
|
|
|
"target datalayout = \"e-m:o-i64:64-f80:128-n8:16:32:64-S128\"\n"
|
|
|
|
"define void @foo(i32 %n) {\n"
|
|
|
|
"entry:\n"
|
|
|
|
" br i1 undef, label %for.cond, label %for.end\n"
|
|
|
|
"for.cond:\n"
|
|
|
|
" %i.0 = phi i32 [ 0, %entry ], [ %inc, %latch.1 ], [ %inc, %latch.2 ]\n"
|
|
|
|
" %inc = add nsw i32 %i.0, 1\n"
|
|
|
|
" %cmp = icmp slt i32 %i.0, %n\n"
|
|
|
|
" br i1 %cmp, label %latch.1, label %for.end\n"
|
|
|
|
"latch.1:\n"
|
|
|
|
" br i1 undef, label %for.cond, label %latch.2, !llvm.loop !0\n"
|
|
|
|
"latch.2:\n"
|
|
|
|
" br label %for.cond, !llvm.loop !0\n"
|
|
|
|
"for.end:\n"
|
|
|
|
" ret void\n"
|
|
|
|
"}\n"
|
|
|
|
"!0 = distinct !{!0, !1}\n"
|
|
|
|
"!1 = !{!\"llvm.loop.distribute.enable\", i1 true}\n";
|
|
|
|
|
|
|
|
// Parse the module.
|
|
|
|
LLVMContext Context;
|
|
|
|
std::unique_ptr<Module> M = makeLLVMModule(Context, ModuleStr);
|
|
|
|
|
|
|
|
runWithLoopInfo(*M, "foo", [&](Function &F, LoopInfo &LI) {
|
|
|
|
Function::iterator FI = F.begin();
|
|
|
|
// First basic block is entry - skip it.
|
|
|
|
BasicBlock *Header = &*(++FI);
|
|
|
|
assert(Header->getName() == "for.cond");
|
|
|
|
Loop *L = LI.getLoopFor(Header);
|
|
|
|
EXPECT_NE(L, nullptr);
|
|
|
|
|
|
|
|
// This loop is not in simplified form.
|
|
|
|
EXPECT_FALSE(L->isLoopSimplifyForm());
|
|
|
|
|
|
|
|
// Try to get and set the metadata id for the loop.
|
|
|
|
MDNode *OldLoopID = L->getLoopID();
|
|
|
|
EXPECT_NE(OldLoopID, nullptr);
|
|
|
|
|
|
|
|
MDNode *NewLoopID = MDNode::get(Context, {nullptr});
|
|
|
|
// Set operand 0 to refer to the loop id itself.
|
|
|
|
NewLoopID->replaceOperandWith(0, NewLoopID);
|
|
|
|
|
|
|
|
L->setLoopID(NewLoopID);
|
|
|
|
EXPECT_EQ(L->getLoopID(), NewLoopID);
|
|
|
|
EXPECT_NE(L->getLoopID(), OldLoopID);
|
|
|
|
|
|
|
|
L->setLoopID(OldLoopID);
|
|
|
|
EXPECT_EQ(L->getLoopID(), OldLoopID);
|
|
|
|
EXPECT_NE(L->getLoopID(), NewLoopID);
|
|
|
|
});
|
|
|
|
}
|
|
|
|
|
2017-01-20 10:41:20 +08:00
|
|
|
TEST(LoopInfoTest, PreorderTraversals) {
|
|
|
|
const char *ModuleStr = "define void @f() {\n"
|
|
|
|
"entry:\n"
|
|
|
|
" br label %loop.0\n"
|
|
|
|
"loop.0:\n"
|
|
|
|
" br i1 undef, label %loop.0.0, label %loop.1\n"
|
|
|
|
"loop.0.0:\n"
|
|
|
|
" br i1 undef, label %loop.0.0, label %loop.0.1\n"
|
|
|
|
"loop.0.1:\n"
|
|
|
|
" br i1 undef, label %loop.0.1, label %loop.0.2\n"
|
|
|
|
"loop.0.2:\n"
|
|
|
|
" br i1 undef, label %loop.0.2, label %loop.0\n"
|
|
|
|
"loop.1:\n"
|
|
|
|
" br i1 undef, label %loop.1.0, label %end\n"
|
|
|
|
"loop.1.0:\n"
|
|
|
|
" br i1 undef, label %loop.1.0, label %loop.1.1\n"
|
|
|
|
"loop.1.1:\n"
|
|
|
|
" br i1 undef, label %loop.1.1, label %loop.1.2\n"
|
|
|
|
"loop.1.2:\n"
|
|
|
|
" br i1 undef, label %loop.1.2, label %loop.1\n"
|
|
|
|
"end:\n"
|
|
|
|
" ret void\n"
|
|
|
|
"}\n";
|
|
|
|
// Parse the module.
|
|
|
|
LLVMContext Context;
|
|
|
|
std::unique_ptr<Module> M = makeLLVMModule(Context, ModuleStr);
|
|
|
|
Function &F = *M->begin();
|
|
|
|
|
|
|
|
DominatorTree DT(F);
|
|
|
|
LoopInfo LI;
|
|
|
|
LI.analyze(DT);
|
|
|
|
|
|
|
|
Function::iterator I = F.begin();
|
|
|
|
ASSERT_EQ("entry", I->getName());
|
|
|
|
++I;
|
|
|
|
Loop &L_0 = *LI.getLoopFor(&*I++);
|
|
|
|
ASSERT_EQ("loop.0", L_0.getHeader()->getName());
|
|
|
|
Loop &L_0_0 = *LI.getLoopFor(&*I++);
|
|
|
|
ASSERT_EQ("loop.0.0", L_0_0.getHeader()->getName());
|
|
|
|
Loop &L_0_1 = *LI.getLoopFor(&*I++);
|
|
|
|
ASSERT_EQ("loop.0.1", L_0_1.getHeader()->getName());
|
|
|
|
Loop &L_0_2 = *LI.getLoopFor(&*I++);
|
|
|
|
ASSERT_EQ("loop.0.2", L_0_2.getHeader()->getName());
|
|
|
|
Loop &L_1 = *LI.getLoopFor(&*I++);
|
|
|
|
ASSERT_EQ("loop.1", L_1.getHeader()->getName());
|
|
|
|
Loop &L_1_0 = *LI.getLoopFor(&*I++);
|
|
|
|
ASSERT_EQ("loop.1.0", L_1_0.getHeader()->getName());
|
|
|
|
Loop &L_1_1 = *LI.getLoopFor(&*I++);
|
|
|
|
ASSERT_EQ("loop.1.1", L_1_1.getHeader()->getName());
|
|
|
|
Loop &L_1_2 = *LI.getLoopFor(&*I++);
|
|
|
|
ASSERT_EQ("loop.1.2", L_1_2.getHeader()->getName());
|
|
|
|
|
|
|
|
auto Preorder = LI.getLoopsInPreorder();
|
|
|
|
ASSERT_EQ(8u, Preorder.size());
|
|
|
|
EXPECT_EQ(&L_0, Preorder[0]);
|
|
|
|
EXPECT_EQ(&L_0_0, Preorder[1]);
|
|
|
|
EXPECT_EQ(&L_0_1, Preorder[2]);
|
|
|
|
EXPECT_EQ(&L_0_2, Preorder[3]);
|
|
|
|
EXPECT_EQ(&L_1, Preorder[4]);
|
|
|
|
EXPECT_EQ(&L_1_0, Preorder[5]);
|
|
|
|
EXPECT_EQ(&L_1_1, Preorder[6]);
|
|
|
|
EXPECT_EQ(&L_1_2, Preorder[7]);
|
|
|
|
|
|
|
|
auto ReverseSiblingPreorder = LI.getLoopsInReverseSiblingPreorder();
|
|
|
|
ASSERT_EQ(8u, ReverseSiblingPreorder.size());
|
|
|
|
EXPECT_EQ(&L_1, ReverseSiblingPreorder[0]);
|
|
|
|
EXPECT_EQ(&L_1_2, ReverseSiblingPreorder[1]);
|
|
|
|
EXPECT_EQ(&L_1_1, ReverseSiblingPreorder[2]);
|
|
|
|
EXPECT_EQ(&L_1_0, ReverseSiblingPreorder[3]);
|
|
|
|
EXPECT_EQ(&L_0, ReverseSiblingPreorder[4]);
|
|
|
|
EXPECT_EQ(&L_0_2, ReverseSiblingPreorder[5]);
|
|
|
|
EXPECT_EQ(&L_0_1, ReverseSiblingPreorder[6]);
|
|
|
|
EXPECT_EQ(&L_0_0, ReverseSiblingPreorder[7]);
|
|
|
|
}
|
[LOOPINFO] Extend Loop object to add utilities to get the loop bounds,
step, and loop induction variable.
Summary: This PR extends the loop object with more utilities to get loop
bounds, step, and loop induction variable. There already exists passes
which try to obtain the loop induction variable in their own pass, e.g.
loop interchange. It would be useful to have a common area to get these
information.
/// Example:
/// for (int i = lb; i < ub; i+=step)
/// <loop body>
/// --- pseudo LLVMIR ---
/// beforeloop:
/// guardcmp = (lb < ub)
/// if (guardcmp) goto preheader; else goto afterloop
/// preheader:
/// loop:
/// i1 = phi[{lb, preheader}, {i2, latch}]
/// <loop body>
/// i2 = i1 + step
/// latch:
/// cmp = (i2 < ub)
/// if (cmp) goto loop
/// exit:
/// afterloop:
///
/// getBounds
/// getInitialIVValue --> lb
/// getStepInst --> i2 = i1 + step
/// getStepValue --> step
/// getFinalIVValue --> ub
/// getCanonicalPredicate --> '<'
/// getDirection --> Increasing
/// getInductionVariable --> i1
/// getAuxiliaryInductionVariable --> {i1}
/// isCanonical --> false
Reviewers: kbarton, hfinkel, dmgreen, Meinersbur, jdoerfert, syzaara,
fhahn
Reviewed By: kbarton
Subscribers: tvvikram, bmahjour, etiotto, fhahn, jsji, hiraditya,
llvm-commits
Tag: LLVM
Differential Revision: https://reviews.llvm.org/D60565
llvm-svn: 362644
2019-06-06 04:42:47 +08:00
|
|
|
|
|
|
|
TEST(LoopInfoTest, CanonicalLoop) {
|
|
|
|
const char *ModuleStr =
|
|
|
|
"define void @foo(i32* %A, i32 %ub) {\n"
|
|
|
|
"entry:\n"
|
|
|
|
" %guardcmp = icmp slt i32 0, %ub\n"
|
|
|
|
" br i1 %guardcmp, label %for.preheader, label %for.end\n"
|
|
|
|
"for.preheader:\n"
|
|
|
|
" br label %for.body\n"
|
|
|
|
"for.body:\n"
|
|
|
|
" %i = phi i32 [ 0, %for.preheader ], [ %inc, %for.body ]\n"
|
|
|
|
" %idxprom = sext i32 %i to i64\n"
|
|
|
|
" %arrayidx = getelementptr inbounds i32, i32* %A, i64 %idxprom\n"
|
|
|
|
" store i32 %i, i32* %arrayidx, align 4\n"
|
|
|
|
" %inc = add nsw i32 %i, 1\n"
|
|
|
|
" %cmp = icmp slt i32 %inc, %ub\n"
|
|
|
|
" br i1 %cmp, label %for.body, label %for.exit\n"
|
|
|
|
"for.exit:\n"
|
|
|
|
" br label %for.end\n"
|
|
|
|
"for.end:\n"
|
|
|
|
" ret void\n"
|
|
|
|
"}\n";
|
|
|
|
|
|
|
|
// Parse the module.
|
|
|
|
LLVMContext Context;
|
|
|
|
std::unique_ptr<Module> M = makeLLVMModule(Context, ModuleStr);
|
|
|
|
|
|
|
|
runWithLoopInfoPlus(
|
|
|
|
*M, "foo",
|
[LOOPINFO] Introduce the loop guard API.
Summary:
This is the first patch for the loop guard. We introduced
getLoopGuardBranch() and isGuarded().
This currently only works on simplified loop, as it requires a preheader
and a latch to identify the guard.
It will work on loops of the form:
/// GuardBB:
/// br cond1, Preheader, ExitSucc <== GuardBranch
/// Preheader:
/// br Header
/// Header:
/// ...
/// br Latch
/// Latch:
/// br cond2, Header, ExitBlock
/// ExitBlock:
/// br ExitSucc
/// ExitSucc:
Prior discussions leading upto the decision to introduce the loop guard
API: http://lists.llvm.org/pipermail/llvm-dev/2019-May/132607.html
Reviewer: reames, kbarton, hfinkel, jdoerfert, Meinersbur, dmgreen
Reviewed By: reames
Subscribers: wuzish, hiraditya, jsji, llvm-commits, bmahjour, etiotto
Tag: LLVM
Differential Revision: https://reviews.llvm.org/D63885
llvm-svn: 367033
2019-07-26 00:13:18 +08:00
|
|
|
[&](Function &F, LoopInfo &LI, ScalarEvolution &SE) {
|
[LOOPINFO] Extend Loop object to add utilities to get the loop bounds,
step, and loop induction variable.
Summary: This PR extends the loop object with more utilities to get loop
bounds, step, and loop induction variable. There already exists passes
which try to obtain the loop induction variable in their own pass, e.g.
loop interchange. It would be useful to have a common area to get these
information.
/// Example:
/// for (int i = lb; i < ub; i+=step)
/// <loop body>
/// --- pseudo LLVMIR ---
/// beforeloop:
/// guardcmp = (lb < ub)
/// if (guardcmp) goto preheader; else goto afterloop
/// preheader:
/// loop:
/// i1 = phi[{lb, preheader}, {i2, latch}]
/// <loop body>
/// i2 = i1 + step
/// latch:
/// cmp = (i2 < ub)
/// if (cmp) goto loop
/// exit:
/// afterloop:
///
/// getBounds
/// getInitialIVValue --> lb
/// getStepInst --> i2 = i1 + step
/// getStepValue --> step
/// getFinalIVValue --> ub
/// getCanonicalPredicate --> '<'
/// getDirection --> Increasing
/// getInductionVariable --> i1
/// getAuxiliaryInductionVariable --> {i1}
/// isCanonical --> false
Reviewers: kbarton, hfinkel, dmgreen, Meinersbur, jdoerfert, syzaara,
fhahn
Reviewed By: kbarton
Subscribers: tvvikram, bmahjour, etiotto, fhahn, jsji, hiraditya,
llvm-commits
Tag: LLVM
Differential Revision: https://reviews.llvm.org/D60565
llvm-svn: 362644
2019-06-06 04:42:47 +08:00
|
|
|
Function::iterator FI = F.begin();
|
[LOOPINFO] Introduce the loop guard API.
Summary:
This is the first patch for the loop guard. We introduced
getLoopGuardBranch() and isGuarded().
This currently only works on simplified loop, as it requires a preheader
and a latch to identify the guard.
It will work on loops of the form:
/// GuardBB:
/// br cond1, Preheader, ExitSucc <== GuardBranch
/// Preheader:
/// br Header
/// Header:
/// ...
/// br Latch
/// Latch:
/// br cond2, Header, ExitBlock
/// ExitBlock:
/// br ExitSucc
/// ExitSucc:
Prior discussions leading upto the decision to introduce the loop guard
API: http://lists.llvm.org/pipermail/llvm-dev/2019-May/132607.html
Reviewer: reames, kbarton, hfinkel, jdoerfert, Meinersbur, dmgreen
Reviewed By: reames
Subscribers: wuzish, hiraditya, jsji, llvm-commits, bmahjour, etiotto
Tag: LLVM
Differential Revision: https://reviews.llvm.org/D63885
llvm-svn: 367033
2019-07-26 00:13:18 +08:00
|
|
|
BasicBlock *Entry = &*(FI);
|
|
|
|
BranchInst *Guard = dyn_cast<BranchInst>(Entry->getTerminator());
|
[LOOPINFO] Extend Loop object to add utilities to get the loop bounds,
step, and loop induction variable.
Summary: This PR extends the loop object with more utilities to get loop
bounds, step, and loop induction variable. There already exists passes
which try to obtain the loop induction variable in their own pass, e.g.
loop interchange. It would be useful to have a common area to get these
information.
/// Example:
/// for (int i = lb; i < ub; i+=step)
/// <loop body>
/// --- pseudo LLVMIR ---
/// beforeloop:
/// guardcmp = (lb < ub)
/// if (guardcmp) goto preheader; else goto afterloop
/// preheader:
/// loop:
/// i1 = phi[{lb, preheader}, {i2, latch}]
/// <loop body>
/// i2 = i1 + step
/// latch:
/// cmp = (i2 < ub)
/// if (cmp) goto loop
/// exit:
/// afterloop:
///
/// getBounds
/// getInitialIVValue --> lb
/// getStepInst --> i2 = i1 + step
/// getStepValue --> step
/// getFinalIVValue --> ub
/// getCanonicalPredicate --> '<'
/// getDirection --> Increasing
/// getInductionVariable --> i1
/// getAuxiliaryInductionVariable --> {i1}
/// isCanonical --> false
Reviewers: kbarton, hfinkel, dmgreen, Meinersbur, jdoerfert, syzaara,
fhahn
Reviewed By: kbarton
Subscribers: tvvikram, bmahjour, etiotto, fhahn, jsji, hiraditya,
llvm-commits
Tag: LLVM
Differential Revision: https://reviews.llvm.org/D60565
llvm-svn: 362644
2019-06-06 04:42:47 +08:00
|
|
|
// First two basic block are entry and for.preheader - skip them.
|
|
|
|
++FI;
|
|
|
|
BasicBlock *Header = &*(++FI);
|
|
|
|
assert(Header->getName() == "for.body");
|
|
|
|
Loop *L = LI.getLoopFor(Header);
|
|
|
|
EXPECT_NE(L, nullptr);
|
|
|
|
|
|
|
|
Optional<Loop::LoopBounds> Bounds = L->getBounds(SE);
|
|
|
|
EXPECT_NE(Bounds, None);
|
|
|
|
ConstantInt *InitialIVValue =
|
|
|
|
dyn_cast<ConstantInt>(&Bounds->getInitialIVValue());
|
|
|
|
EXPECT_TRUE(InitialIVValue && InitialIVValue->isZero());
|
|
|
|
EXPECT_EQ(Bounds->getStepInst().getName(), "inc");
|
|
|
|
ConstantInt *StepValue =
|
|
|
|
dyn_cast_or_null<ConstantInt>(Bounds->getStepValue());
|
|
|
|
EXPECT_TRUE(StepValue && StepValue->isOne());
|
|
|
|
EXPECT_EQ(Bounds->getFinalIVValue().getName(), "ub");
|
|
|
|
EXPECT_EQ(Bounds->getCanonicalPredicate(), ICmpInst::ICMP_SLT);
|
|
|
|
EXPECT_EQ(Bounds->getDirection(),
|
|
|
|
Loop::LoopBounds::Direction::Increasing);
|
|
|
|
EXPECT_EQ(L->getInductionVariable(SE)->getName(), "i");
|
[LOOPINFO] Introduce the loop guard API.
Summary:
This is the first patch for the loop guard. We introduced
getLoopGuardBranch() and isGuarded().
This currently only works on simplified loop, as it requires a preheader
and a latch to identify the guard.
It will work on loops of the form:
/// GuardBB:
/// br cond1, Preheader, ExitSucc <== GuardBranch
/// Preheader:
/// br Header
/// Header:
/// ...
/// br Latch
/// Latch:
/// br cond2, Header, ExitBlock
/// ExitBlock:
/// br ExitSucc
/// ExitSucc:
Prior discussions leading upto the decision to introduce the loop guard
API: http://lists.llvm.org/pipermail/llvm-dev/2019-May/132607.html
Reviewer: reames, kbarton, hfinkel, jdoerfert, Meinersbur, dmgreen
Reviewed By: reames
Subscribers: wuzish, hiraditya, jsji, llvm-commits, bmahjour, etiotto
Tag: LLVM
Differential Revision: https://reviews.llvm.org/D63885
llvm-svn: 367033
2019-07-26 00:13:18 +08:00
|
|
|
EXPECT_EQ(L->getLoopGuardBranch(), Guard);
|
|
|
|
EXPECT_TRUE(L->isGuarded());
|
2019-12-13 03:18:39 +08:00
|
|
|
EXPECT_TRUE(L->isRotatedForm());
|
[LOOPINFO] Extend Loop object to add utilities to get the loop bounds,
step, and loop induction variable.
Summary: This PR extends the loop object with more utilities to get loop
bounds, step, and loop induction variable. There already exists passes
which try to obtain the loop induction variable in their own pass, e.g.
loop interchange. It would be useful to have a common area to get these
information.
/// Example:
/// for (int i = lb; i < ub; i+=step)
/// <loop body>
/// --- pseudo LLVMIR ---
/// beforeloop:
/// guardcmp = (lb < ub)
/// if (guardcmp) goto preheader; else goto afterloop
/// preheader:
/// loop:
/// i1 = phi[{lb, preheader}, {i2, latch}]
/// <loop body>
/// i2 = i1 + step
/// latch:
/// cmp = (i2 < ub)
/// if (cmp) goto loop
/// exit:
/// afterloop:
///
/// getBounds
/// getInitialIVValue --> lb
/// getStepInst --> i2 = i1 + step
/// getStepValue --> step
/// getFinalIVValue --> ub
/// getCanonicalPredicate --> '<'
/// getDirection --> Increasing
/// getInductionVariable --> i1
/// getAuxiliaryInductionVariable --> {i1}
/// isCanonical --> false
Reviewers: kbarton, hfinkel, dmgreen, Meinersbur, jdoerfert, syzaara,
fhahn
Reviewed By: kbarton
Subscribers: tvvikram, bmahjour, etiotto, fhahn, jsji, hiraditya,
llvm-commits
Tag: LLVM
Differential Revision: https://reviews.llvm.org/D60565
llvm-svn: 362644
2019-06-06 04:42:47 +08:00
|
|
|
});
|
|
|
|
}
|
|
|
|
|
|
|
|
TEST(LoopInfoTest, LoopWithInverseGuardSuccs) {
|
|
|
|
const char *ModuleStr =
|
|
|
|
"define void @foo(i32* %A, i32 %ub) {\n"
|
|
|
|
"entry:\n"
|
|
|
|
" %guardcmp = icmp sge i32 0, %ub\n"
|
|
|
|
" br i1 %guardcmp, label %for.end, label %for.preheader\n"
|
|
|
|
"for.preheader:\n"
|
|
|
|
" br label %for.body\n"
|
|
|
|
"for.body:\n"
|
|
|
|
" %i = phi i32 [ 0, %for.preheader ], [ %inc, %for.body ]\n"
|
|
|
|
" %idxprom = sext i32 %i to i64\n"
|
|
|
|
" %arrayidx = getelementptr inbounds i32, i32* %A, i64 %idxprom\n"
|
|
|
|
" store i32 %i, i32* %arrayidx, align 4\n"
|
|
|
|
" %inc = add nsw i32 %i, 1\n"
|
|
|
|
" %cmp = icmp slt i32 %inc, %ub\n"
|
|
|
|
" br i1 %cmp, label %for.body, label %for.exit\n"
|
|
|
|
"for.exit:\n"
|
|
|
|
" br label %for.end\n"
|
|
|
|
"for.end:\n"
|
|
|
|
" ret void\n"
|
|
|
|
"}\n";
|
|
|
|
|
|
|
|
// Parse the module.
|
|
|
|
LLVMContext Context;
|
|
|
|
std::unique_ptr<Module> M = makeLLVMModule(Context, ModuleStr);
|
|
|
|
|
|
|
|
runWithLoopInfoPlus(
|
|
|
|
*M, "foo",
|
[LOOPINFO] Introduce the loop guard API.
Summary:
This is the first patch for the loop guard. We introduced
getLoopGuardBranch() and isGuarded().
This currently only works on simplified loop, as it requires a preheader
and a latch to identify the guard.
It will work on loops of the form:
/// GuardBB:
/// br cond1, Preheader, ExitSucc <== GuardBranch
/// Preheader:
/// br Header
/// Header:
/// ...
/// br Latch
/// Latch:
/// br cond2, Header, ExitBlock
/// ExitBlock:
/// br ExitSucc
/// ExitSucc:
Prior discussions leading upto the decision to introduce the loop guard
API: http://lists.llvm.org/pipermail/llvm-dev/2019-May/132607.html
Reviewer: reames, kbarton, hfinkel, jdoerfert, Meinersbur, dmgreen
Reviewed By: reames
Subscribers: wuzish, hiraditya, jsji, llvm-commits, bmahjour, etiotto
Tag: LLVM
Differential Revision: https://reviews.llvm.org/D63885
llvm-svn: 367033
2019-07-26 00:13:18 +08:00
|
|
|
[&](Function &F, LoopInfo &LI, ScalarEvolution &SE) {
|
[LOOPINFO] Extend Loop object to add utilities to get the loop bounds,
step, and loop induction variable.
Summary: This PR extends the loop object with more utilities to get loop
bounds, step, and loop induction variable. There already exists passes
which try to obtain the loop induction variable in their own pass, e.g.
loop interchange. It would be useful to have a common area to get these
information.
/// Example:
/// for (int i = lb; i < ub; i+=step)
/// <loop body>
/// --- pseudo LLVMIR ---
/// beforeloop:
/// guardcmp = (lb < ub)
/// if (guardcmp) goto preheader; else goto afterloop
/// preheader:
/// loop:
/// i1 = phi[{lb, preheader}, {i2, latch}]
/// <loop body>
/// i2 = i1 + step
/// latch:
/// cmp = (i2 < ub)
/// if (cmp) goto loop
/// exit:
/// afterloop:
///
/// getBounds
/// getInitialIVValue --> lb
/// getStepInst --> i2 = i1 + step
/// getStepValue --> step
/// getFinalIVValue --> ub
/// getCanonicalPredicate --> '<'
/// getDirection --> Increasing
/// getInductionVariable --> i1
/// getAuxiliaryInductionVariable --> {i1}
/// isCanonical --> false
Reviewers: kbarton, hfinkel, dmgreen, Meinersbur, jdoerfert, syzaara,
fhahn
Reviewed By: kbarton
Subscribers: tvvikram, bmahjour, etiotto, fhahn, jsji, hiraditya,
llvm-commits
Tag: LLVM
Differential Revision: https://reviews.llvm.org/D60565
llvm-svn: 362644
2019-06-06 04:42:47 +08:00
|
|
|
Function::iterator FI = F.begin();
|
[LOOPINFO] Introduce the loop guard API.
Summary:
This is the first patch for the loop guard. We introduced
getLoopGuardBranch() and isGuarded().
This currently only works on simplified loop, as it requires a preheader
and a latch to identify the guard.
It will work on loops of the form:
/// GuardBB:
/// br cond1, Preheader, ExitSucc <== GuardBranch
/// Preheader:
/// br Header
/// Header:
/// ...
/// br Latch
/// Latch:
/// br cond2, Header, ExitBlock
/// ExitBlock:
/// br ExitSucc
/// ExitSucc:
Prior discussions leading upto the decision to introduce the loop guard
API: http://lists.llvm.org/pipermail/llvm-dev/2019-May/132607.html
Reviewer: reames, kbarton, hfinkel, jdoerfert, Meinersbur, dmgreen
Reviewed By: reames
Subscribers: wuzish, hiraditya, jsji, llvm-commits, bmahjour, etiotto
Tag: LLVM
Differential Revision: https://reviews.llvm.org/D63885
llvm-svn: 367033
2019-07-26 00:13:18 +08:00
|
|
|
BasicBlock *Entry = &*(FI);
|
|
|
|
BranchInst *Guard = dyn_cast<BranchInst>(Entry->getTerminator());
|
[LOOPINFO] Extend Loop object to add utilities to get the loop bounds,
step, and loop induction variable.
Summary: This PR extends the loop object with more utilities to get loop
bounds, step, and loop induction variable. There already exists passes
which try to obtain the loop induction variable in their own pass, e.g.
loop interchange. It would be useful to have a common area to get these
information.
/// Example:
/// for (int i = lb; i < ub; i+=step)
/// <loop body>
/// --- pseudo LLVMIR ---
/// beforeloop:
/// guardcmp = (lb < ub)
/// if (guardcmp) goto preheader; else goto afterloop
/// preheader:
/// loop:
/// i1 = phi[{lb, preheader}, {i2, latch}]
/// <loop body>
/// i2 = i1 + step
/// latch:
/// cmp = (i2 < ub)
/// if (cmp) goto loop
/// exit:
/// afterloop:
///
/// getBounds
/// getInitialIVValue --> lb
/// getStepInst --> i2 = i1 + step
/// getStepValue --> step
/// getFinalIVValue --> ub
/// getCanonicalPredicate --> '<'
/// getDirection --> Increasing
/// getInductionVariable --> i1
/// getAuxiliaryInductionVariable --> {i1}
/// isCanonical --> false
Reviewers: kbarton, hfinkel, dmgreen, Meinersbur, jdoerfert, syzaara,
fhahn
Reviewed By: kbarton
Subscribers: tvvikram, bmahjour, etiotto, fhahn, jsji, hiraditya,
llvm-commits
Tag: LLVM
Differential Revision: https://reviews.llvm.org/D60565
llvm-svn: 362644
2019-06-06 04:42:47 +08:00
|
|
|
// First two basic block are entry and for.preheader - skip them.
|
|
|
|
++FI;
|
|
|
|
BasicBlock *Header = &*(++FI);
|
|
|
|
assert(Header->getName() == "for.body");
|
|
|
|
Loop *L = LI.getLoopFor(Header);
|
|
|
|
EXPECT_NE(L, nullptr);
|
|
|
|
|
|
|
|
Optional<Loop::LoopBounds> Bounds = L->getBounds(SE);
|
|
|
|
EXPECT_NE(Bounds, None);
|
|
|
|
ConstantInt *InitialIVValue =
|
|
|
|
dyn_cast<ConstantInt>(&Bounds->getInitialIVValue());
|
|
|
|
EXPECT_TRUE(InitialIVValue && InitialIVValue->isZero());
|
|
|
|
EXPECT_EQ(Bounds->getStepInst().getName(), "inc");
|
|
|
|
ConstantInt *StepValue =
|
|
|
|
dyn_cast_or_null<ConstantInt>(Bounds->getStepValue());
|
|
|
|
EXPECT_TRUE(StepValue && StepValue->isOne());
|
|
|
|
EXPECT_EQ(Bounds->getFinalIVValue().getName(), "ub");
|
|
|
|
EXPECT_EQ(Bounds->getCanonicalPredicate(), ICmpInst::ICMP_SLT);
|
|
|
|
EXPECT_EQ(Bounds->getDirection(),
|
|
|
|
Loop::LoopBounds::Direction::Increasing);
|
|
|
|
EXPECT_EQ(L->getInductionVariable(SE)->getName(), "i");
|
[LOOPINFO] Introduce the loop guard API.
Summary:
This is the first patch for the loop guard. We introduced
getLoopGuardBranch() and isGuarded().
This currently only works on simplified loop, as it requires a preheader
and a latch to identify the guard.
It will work on loops of the form:
/// GuardBB:
/// br cond1, Preheader, ExitSucc <== GuardBranch
/// Preheader:
/// br Header
/// Header:
/// ...
/// br Latch
/// Latch:
/// br cond2, Header, ExitBlock
/// ExitBlock:
/// br ExitSucc
/// ExitSucc:
Prior discussions leading upto the decision to introduce the loop guard
API: http://lists.llvm.org/pipermail/llvm-dev/2019-May/132607.html
Reviewer: reames, kbarton, hfinkel, jdoerfert, Meinersbur, dmgreen
Reviewed By: reames
Subscribers: wuzish, hiraditya, jsji, llvm-commits, bmahjour, etiotto
Tag: LLVM
Differential Revision: https://reviews.llvm.org/D63885
llvm-svn: 367033
2019-07-26 00:13:18 +08:00
|
|
|
EXPECT_EQ(L->getLoopGuardBranch(), Guard);
|
|
|
|
EXPECT_TRUE(L->isGuarded());
|
2019-12-13 03:18:39 +08:00
|
|
|
EXPECT_TRUE(L->isRotatedForm());
|
[LOOPINFO] Extend Loop object to add utilities to get the loop bounds,
step, and loop induction variable.
Summary: This PR extends the loop object with more utilities to get loop
bounds, step, and loop induction variable. There already exists passes
which try to obtain the loop induction variable in their own pass, e.g.
loop interchange. It would be useful to have a common area to get these
information.
/// Example:
/// for (int i = lb; i < ub; i+=step)
/// <loop body>
/// --- pseudo LLVMIR ---
/// beforeloop:
/// guardcmp = (lb < ub)
/// if (guardcmp) goto preheader; else goto afterloop
/// preheader:
/// loop:
/// i1 = phi[{lb, preheader}, {i2, latch}]
/// <loop body>
/// i2 = i1 + step
/// latch:
/// cmp = (i2 < ub)
/// if (cmp) goto loop
/// exit:
/// afterloop:
///
/// getBounds
/// getInitialIVValue --> lb
/// getStepInst --> i2 = i1 + step
/// getStepValue --> step
/// getFinalIVValue --> ub
/// getCanonicalPredicate --> '<'
/// getDirection --> Increasing
/// getInductionVariable --> i1
/// getAuxiliaryInductionVariable --> {i1}
/// isCanonical --> false
Reviewers: kbarton, hfinkel, dmgreen, Meinersbur, jdoerfert, syzaara,
fhahn
Reviewed By: kbarton
Subscribers: tvvikram, bmahjour, etiotto, fhahn, jsji, hiraditya,
llvm-commits
Tag: LLVM
Differential Revision: https://reviews.llvm.org/D60565
llvm-svn: 362644
2019-06-06 04:42:47 +08:00
|
|
|
});
|
|
|
|
}
|
|
|
|
|
|
|
|
TEST(LoopInfoTest, LoopWithSwappedGuardCmp) {
|
|
|
|
const char *ModuleStr =
|
|
|
|
"define void @foo(i32* %A, i32 %ub) {\n"
|
|
|
|
"entry:\n"
|
|
|
|
" %guardcmp = icmp sgt i32 %ub, 0\n"
|
|
|
|
" br i1 %guardcmp, label %for.preheader, label %for.end\n"
|
|
|
|
"for.preheader:\n"
|
|
|
|
" br label %for.body\n"
|
|
|
|
"for.body:\n"
|
|
|
|
" %i = phi i32 [ 0, %for.preheader ], [ %inc, %for.body ]\n"
|
|
|
|
" %idxprom = sext i32 %i to i64\n"
|
|
|
|
" %arrayidx = getelementptr inbounds i32, i32* %A, i64 %idxprom\n"
|
|
|
|
" store i32 %i, i32* %arrayidx, align 4\n"
|
|
|
|
" %inc = add nsw i32 %i, 1\n"
|
|
|
|
" %cmp = icmp sge i32 %inc, %ub\n"
|
|
|
|
" br i1 %cmp, label %for.exit, label %for.body\n"
|
|
|
|
"for.exit:\n"
|
|
|
|
" br label %for.end\n"
|
|
|
|
"for.end:\n"
|
|
|
|
" ret void\n"
|
|
|
|
"}\n";
|
|
|
|
|
|
|
|
// Parse the module.
|
|
|
|
LLVMContext Context;
|
|
|
|
std::unique_ptr<Module> M = makeLLVMModule(Context, ModuleStr);
|
|
|
|
|
|
|
|
runWithLoopInfoPlus(
|
|
|
|
*M, "foo",
|
[LOOPINFO] Introduce the loop guard API.
Summary:
This is the first patch for the loop guard. We introduced
getLoopGuardBranch() and isGuarded().
This currently only works on simplified loop, as it requires a preheader
and a latch to identify the guard.
It will work on loops of the form:
/// GuardBB:
/// br cond1, Preheader, ExitSucc <== GuardBranch
/// Preheader:
/// br Header
/// Header:
/// ...
/// br Latch
/// Latch:
/// br cond2, Header, ExitBlock
/// ExitBlock:
/// br ExitSucc
/// ExitSucc:
Prior discussions leading upto the decision to introduce the loop guard
API: http://lists.llvm.org/pipermail/llvm-dev/2019-May/132607.html
Reviewer: reames, kbarton, hfinkel, jdoerfert, Meinersbur, dmgreen
Reviewed By: reames
Subscribers: wuzish, hiraditya, jsji, llvm-commits, bmahjour, etiotto
Tag: LLVM
Differential Revision: https://reviews.llvm.org/D63885
llvm-svn: 367033
2019-07-26 00:13:18 +08:00
|
|
|
[&](Function &F, LoopInfo &LI, ScalarEvolution &SE) {
|
[LOOPINFO] Extend Loop object to add utilities to get the loop bounds,
step, and loop induction variable.
Summary: This PR extends the loop object with more utilities to get loop
bounds, step, and loop induction variable. There already exists passes
which try to obtain the loop induction variable in their own pass, e.g.
loop interchange. It would be useful to have a common area to get these
information.
/// Example:
/// for (int i = lb; i < ub; i+=step)
/// <loop body>
/// --- pseudo LLVMIR ---
/// beforeloop:
/// guardcmp = (lb < ub)
/// if (guardcmp) goto preheader; else goto afterloop
/// preheader:
/// loop:
/// i1 = phi[{lb, preheader}, {i2, latch}]
/// <loop body>
/// i2 = i1 + step
/// latch:
/// cmp = (i2 < ub)
/// if (cmp) goto loop
/// exit:
/// afterloop:
///
/// getBounds
/// getInitialIVValue --> lb
/// getStepInst --> i2 = i1 + step
/// getStepValue --> step
/// getFinalIVValue --> ub
/// getCanonicalPredicate --> '<'
/// getDirection --> Increasing
/// getInductionVariable --> i1
/// getAuxiliaryInductionVariable --> {i1}
/// isCanonical --> false
Reviewers: kbarton, hfinkel, dmgreen, Meinersbur, jdoerfert, syzaara,
fhahn
Reviewed By: kbarton
Subscribers: tvvikram, bmahjour, etiotto, fhahn, jsji, hiraditya,
llvm-commits
Tag: LLVM
Differential Revision: https://reviews.llvm.org/D60565
llvm-svn: 362644
2019-06-06 04:42:47 +08:00
|
|
|
Function::iterator FI = F.begin();
|
[LOOPINFO] Introduce the loop guard API.
Summary:
This is the first patch for the loop guard. We introduced
getLoopGuardBranch() and isGuarded().
This currently only works on simplified loop, as it requires a preheader
and a latch to identify the guard.
It will work on loops of the form:
/// GuardBB:
/// br cond1, Preheader, ExitSucc <== GuardBranch
/// Preheader:
/// br Header
/// Header:
/// ...
/// br Latch
/// Latch:
/// br cond2, Header, ExitBlock
/// ExitBlock:
/// br ExitSucc
/// ExitSucc:
Prior discussions leading upto the decision to introduce the loop guard
API: http://lists.llvm.org/pipermail/llvm-dev/2019-May/132607.html
Reviewer: reames, kbarton, hfinkel, jdoerfert, Meinersbur, dmgreen
Reviewed By: reames
Subscribers: wuzish, hiraditya, jsji, llvm-commits, bmahjour, etiotto
Tag: LLVM
Differential Revision: https://reviews.llvm.org/D63885
llvm-svn: 367033
2019-07-26 00:13:18 +08:00
|
|
|
BasicBlock *Entry = &*(FI);
|
|
|
|
BranchInst *Guard = dyn_cast<BranchInst>(Entry->getTerminator());
|
[LOOPINFO] Extend Loop object to add utilities to get the loop bounds,
step, and loop induction variable.
Summary: This PR extends the loop object with more utilities to get loop
bounds, step, and loop induction variable. There already exists passes
which try to obtain the loop induction variable in their own pass, e.g.
loop interchange. It would be useful to have a common area to get these
information.
/// Example:
/// for (int i = lb; i < ub; i+=step)
/// <loop body>
/// --- pseudo LLVMIR ---
/// beforeloop:
/// guardcmp = (lb < ub)
/// if (guardcmp) goto preheader; else goto afterloop
/// preheader:
/// loop:
/// i1 = phi[{lb, preheader}, {i2, latch}]
/// <loop body>
/// i2 = i1 + step
/// latch:
/// cmp = (i2 < ub)
/// if (cmp) goto loop
/// exit:
/// afterloop:
///
/// getBounds
/// getInitialIVValue --> lb
/// getStepInst --> i2 = i1 + step
/// getStepValue --> step
/// getFinalIVValue --> ub
/// getCanonicalPredicate --> '<'
/// getDirection --> Increasing
/// getInductionVariable --> i1
/// getAuxiliaryInductionVariable --> {i1}
/// isCanonical --> false
Reviewers: kbarton, hfinkel, dmgreen, Meinersbur, jdoerfert, syzaara,
fhahn
Reviewed By: kbarton
Subscribers: tvvikram, bmahjour, etiotto, fhahn, jsji, hiraditya,
llvm-commits
Tag: LLVM
Differential Revision: https://reviews.llvm.org/D60565
llvm-svn: 362644
2019-06-06 04:42:47 +08:00
|
|
|
// First two basic block are entry and for.preheader - skip them.
|
|
|
|
++FI;
|
|
|
|
BasicBlock *Header = &*(++FI);
|
|
|
|
assert(Header->getName() == "for.body");
|
|
|
|
Loop *L = LI.getLoopFor(Header);
|
|
|
|
EXPECT_NE(L, nullptr);
|
|
|
|
|
|
|
|
Optional<Loop::LoopBounds> Bounds = L->getBounds(SE);
|
|
|
|
EXPECT_NE(Bounds, None);
|
|
|
|
ConstantInt *InitialIVValue =
|
|
|
|
dyn_cast<ConstantInt>(&Bounds->getInitialIVValue());
|
|
|
|
EXPECT_TRUE(InitialIVValue && InitialIVValue->isZero());
|
|
|
|
EXPECT_EQ(Bounds->getStepInst().getName(), "inc");
|
|
|
|
ConstantInt *StepValue =
|
|
|
|
dyn_cast_or_null<ConstantInt>(Bounds->getStepValue());
|
|
|
|
EXPECT_TRUE(StepValue && StepValue->isOne());
|
|
|
|
EXPECT_EQ(Bounds->getFinalIVValue().getName(), "ub");
|
|
|
|
EXPECT_EQ(Bounds->getCanonicalPredicate(), ICmpInst::ICMP_SLT);
|
|
|
|
EXPECT_EQ(Bounds->getDirection(),
|
|
|
|
Loop::LoopBounds::Direction::Increasing);
|
|
|
|
EXPECT_EQ(L->getInductionVariable(SE)->getName(), "i");
|
[LOOPINFO] Introduce the loop guard API.
Summary:
This is the first patch for the loop guard. We introduced
getLoopGuardBranch() and isGuarded().
This currently only works on simplified loop, as it requires a preheader
and a latch to identify the guard.
It will work on loops of the form:
/// GuardBB:
/// br cond1, Preheader, ExitSucc <== GuardBranch
/// Preheader:
/// br Header
/// Header:
/// ...
/// br Latch
/// Latch:
/// br cond2, Header, ExitBlock
/// ExitBlock:
/// br ExitSucc
/// ExitSucc:
Prior discussions leading upto the decision to introduce the loop guard
API: http://lists.llvm.org/pipermail/llvm-dev/2019-May/132607.html
Reviewer: reames, kbarton, hfinkel, jdoerfert, Meinersbur, dmgreen
Reviewed By: reames
Subscribers: wuzish, hiraditya, jsji, llvm-commits, bmahjour, etiotto
Tag: LLVM
Differential Revision: https://reviews.llvm.org/D63885
llvm-svn: 367033
2019-07-26 00:13:18 +08:00
|
|
|
EXPECT_EQ(L->getLoopGuardBranch(), Guard);
|
|
|
|
EXPECT_TRUE(L->isGuarded());
|
2019-12-13 03:18:39 +08:00
|
|
|
EXPECT_TRUE(L->isRotatedForm());
|
[LOOPINFO] Extend Loop object to add utilities to get the loop bounds,
step, and loop induction variable.
Summary: This PR extends the loop object with more utilities to get loop
bounds, step, and loop induction variable. There already exists passes
which try to obtain the loop induction variable in their own pass, e.g.
loop interchange. It would be useful to have a common area to get these
information.
/// Example:
/// for (int i = lb; i < ub; i+=step)
/// <loop body>
/// --- pseudo LLVMIR ---
/// beforeloop:
/// guardcmp = (lb < ub)
/// if (guardcmp) goto preheader; else goto afterloop
/// preheader:
/// loop:
/// i1 = phi[{lb, preheader}, {i2, latch}]
/// <loop body>
/// i2 = i1 + step
/// latch:
/// cmp = (i2 < ub)
/// if (cmp) goto loop
/// exit:
/// afterloop:
///
/// getBounds
/// getInitialIVValue --> lb
/// getStepInst --> i2 = i1 + step
/// getStepValue --> step
/// getFinalIVValue --> ub
/// getCanonicalPredicate --> '<'
/// getDirection --> Increasing
/// getInductionVariable --> i1
/// getAuxiliaryInductionVariable --> {i1}
/// isCanonical --> false
Reviewers: kbarton, hfinkel, dmgreen, Meinersbur, jdoerfert, syzaara,
fhahn
Reviewed By: kbarton
Subscribers: tvvikram, bmahjour, etiotto, fhahn, jsji, hiraditya,
llvm-commits
Tag: LLVM
Differential Revision: https://reviews.llvm.org/D60565
llvm-svn: 362644
2019-06-06 04:42:47 +08:00
|
|
|
});
|
|
|
|
}
|
|
|
|
|
|
|
|
TEST(LoopInfoTest, LoopWithInverseLatchSuccs) {
|
|
|
|
const char *ModuleStr =
|
|
|
|
"define void @foo(i32* %A, i32 %ub) {\n"
|
|
|
|
"entry:\n"
|
|
|
|
" %guardcmp = icmp slt i32 0, %ub\n"
|
|
|
|
" br i1 %guardcmp, label %for.preheader, label %for.end\n"
|
|
|
|
"for.preheader:\n"
|
|
|
|
" br label %for.body\n"
|
|
|
|
"for.body:\n"
|
|
|
|
" %i = phi i32 [ 0, %for.preheader ], [ %inc, %for.body ]\n"
|
|
|
|
" %idxprom = sext i32 %i to i64\n"
|
|
|
|
" %arrayidx = getelementptr inbounds i32, i32* %A, i64 %idxprom\n"
|
|
|
|
" store i32 %i, i32* %arrayidx, align 4\n"
|
|
|
|
" %inc = add nsw i32 %i, 1\n"
|
|
|
|
" %cmp = icmp sge i32 %inc, %ub\n"
|
|
|
|
" br i1 %cmp, label %for.exit, label %for.body\n"
|
|
|
|
"for.exit:\n"
|
|
|
|
" br label %for.end\n"
|
|
|
|
"for.end:\n"
|
|
|
|
" ret void\n"
|
|
|
|
"}\n";
|
|
|
|
|
|
|
|
// Parse the module.
|
|
|
|
LLVMContext Context;
|
|
|
|
std::unique_ptr<Module> M = makeLLVMModule(Context, ModuleStr);
|
|
|
|
|
|
|
|
runWithLoopInfoPlus(
|
|
|
|
*M, "foo",
|
[LOOPINFO] Introduce the loop guard API.
Summary:
This is the first patch for the loop guard. We introduced
getLoopGuardBranch() and isGuarded().
This currently only works on simplified loop, as it requires a preheader
and a latch to identify the guard.
It will work on loops of the form:
/// GuardBB:
/// br cond1, Preheader, ExitSucc <== GuardBranch
/// Preheader:
/// br Header
/// Header:
/// ...
/// br Latch
/// Latch:
/// br cond2, Header, ExitBlock
/// ExitBlock:
/// br ExitSucc
/// ExitSucc:
Prior discussions leading upto the decision to introduce the loop guard
API: http://lists.llvm.org/pipermail/llvm-dev/2019-May/132607.html
Reviewer: reames, kbarton, hfinkel, jdoerfert, Meinersbur, dmgreen
Reviewed By: reames
Subscribers: wuzish, hiraditya, jsji, llvm-commits, bmahjour, etiotto
Tag: LLVM
Differential Revision: https://reviews.llvm.org/D63885
llvm-svn: 367033
2019-07-26 00:13:18 +08:00
|
|
|
[&](Function &F, LoopInfo &LI, ScalarEvolution &SE) {
|
[LOOPINFO] Extend Loop object to add utilities to get the loop bounds,
step, and loop induction variable.
Summary: This PR extends the loop object with more utilities to get loop
bounds, step, and loop induction variable. There already exists passes
which try to obtain the loop induction variable in their own pass, e.g.
loop interchange. It would be useful to have a common area to get these
information.
/// Example:
/// for (int i = lb; i < ub; i+=step)
/// <loop body>
/// --- pseudo LLVMIR ---
/// beforeloop:
/// guardcmp = (lb < ub)
/// if (guardcmp) goto preheader; else goto afterloop
/// preheader:
/// loop:
/// i1 = phi[{lb, preheader}, {i2, latch}]
/// <loop body>
/// i2 = i1 + step
/// latch:
/// cmp = (i2 < ub)
/// if (cmp) goto loop
/// exit:
/// afterloop:
///
/// getBounds
/// getInitialIVValue --> lb
/// getStepInst --> i2 = i1 + step
/// getStepValue --> step
/// getFinalIVValue --> ub
/// getCanonicalPredicate --> '<'
/// getDirection --> Increasing
/// getInductionVariable --> i1
/// getAuxiliaryInductionVariable --> {i1}
/// isCanonical --> false
Reviewers: kbarton, hfinkel, dmgreen, Meinersbur, jdoerfert, syzaara,
fhahn
Reviewed By: kbarton
Subscribers: tvvikram, bmahjour, etiotto, fhahn, jsji, hiraditya,
llvm-commits
Tag: LLVM
Differential Revision: https://reviews.llvm.org/D60565
llvm-svn: 362644
2019-06-06 04:42:47 +08:00
|
|
|
Function::iterator FI = F.begin();
|
[LOOPINFO] Introduce the loop guard API.
Summary:
This is the first patch for the loop guard. We introduced
getLoopGuardBranch() and isGuarded().
This currently only works on simplified loop, as it requires a preheader
and a latch to identify the guard.
It will work on loops of the form:
/// GuardBB:
/// br cond1, Preheader, ExitSucc <== GuardBranch
/// Preheader:
/// br Header
/// Header:
/// ...
/// br Latch
/// Latch:
/// br cond2, Header, ExitBlock
/// ExitBlock:
/// br ExitSucc
/// ExitSucc:
Prior discussions leading upto the decision to introduce the loop guard
API: http://lists.llvm.org/pipermail/llvm-dev/2019-May/132607.html
Reviewer: reames, kbarton, hfinkel, jdoerfert, Meinersbur, dmgreen
Reviewed By: reames
Subscribers: wuzish, hiraditya, jsji, llvm-commits, bmahjour, etiotto
Tag: LLVM
Differential Revision: https://reviews.llvm.org/D63885
llvm-svn: 367033
2019-07-26 00:13:18 +08:00
|
|
|
BasicBlock *Entry = &*(FI);
|
|
|
|
BranchInst *Guard = dyn_cast<BranchInst>(Entry->getTerminator());
|
[LOOPINFO] Extend Loop object to add utilities to get the loop bounds,
step, and loop induction variable.
Summary: This PR extends the loop object with more utilities to get loop
bounds, step, and loop induction variable. There already exists passes
which try to obtain the loop induction variable in their own pass, e.g.
loop interchange. It would be useful to have a common area to get these
information.
/// Example:
/// for (int i = lb; i < ub; i+=step)
/// <loop body>
/// --- pseudo LLVMIR ---
/// beforeloop:
/// guardcmp = (lb < ub)
/// if (guardcmp) goto preheader; else goto afterloop
/// preheader:
/// loop:
/// i1 = phi[{lb, preheader}, {i2, latch}]
/// <loop body>
/// i2 = i1 + step
/// latch:
/// cmp = (i2 < ub)
/// if (cmp) goto loop
/// exit:
/// afterloop:
///
/// getBounds
/// getInitialIVValue --> lb
/// getStepInst --> i2 = i1 + step
/// getStepValue --> step
/// getFinalIVValue --> ub
/// getCanonicalPredicate --> '<'
/// getDirection --> Increasing
/// getInductionVariable --> i1
/// getAuxiliaryInductionVariable --> {i1}
/// isCanonical --> false
Reviewers: kbarton, hfinkel, dmgreen, Meinersbur, jdoerfert, syzaara,
fhahn
Reviewed By: kbarton
Subscribers: tvvikram, bmahjour, etiotto, fhahn, jsji, hiraditya,
llvm-commits
Tag: LLVM
Differential Revision: https://reviews.llvm.org/D60565
llvm-svn: 362644
2019-06-06 04:42:47 +08:00
|
|
|
// First two basic block are entry and for.preheader - skip them.
|
|
|
|
++FI;
|
|
|
|
BasicBlock *Header = &*(++FI);
|
|
|
|
assert(Header->getName() == "for.body");
|
|
|
|
Loop *L = LI.getLoopFor(Header);
|
|
|
|
EXPECT_NE(L, nullptr);
|
|
|
|
|
|
|
|
Optional<Loop::LoopBounds> Bounds = L->getBounds(SE);
|
|
|
|
EXPECT_NE(Bounds, None);
|
|
|
|
ConstantInt *InitialIVValue =
|
|
|
|
dyn_cast<ConstantInt>(&Bounds->getInitialIVValue());
|
|
|
|
EXPECT_TRUE(InitialIVValue && InitialIVValue->isZero());
|
|
|
|
EXPECT_EQ(Bounds->getStepInst().getName(), "inc");
|
|
|
|
ConstantInt *StepValue =
|
|
|
|
dyn_cast_or_null<ConstantInt>(Bounds->getStepValue());
|
|
|
|
EXPECT_TRUE(StepValue && StepValue->isOne());
|
|
|
|
EXPECT_EQ(Bounds->getFinalIVValue().getName(), "ub");
|
|
|
|
EXPECT_EQ(Bounds->getCanonicalPredicate(), ICmpInst::ICMP_SLT);
|
|
|
|
EXPECT_EQ(Bounds->getDirection(),
|
|
|
|
Loop::LoopBounds::Direction::Increasing);
|
|
|
|
EXPECT_EQ(L->getInductionVariable(SE)->getName(), "i");
|
[LOOPINFO] Introduce the loop guard API.
Summary:
This is the first patch for the loop guard. We introduced
getLoopGuardBranch() and isGuarded().
This currently only works on simplified loop, as it requires a preheader
and a latch to identify the guard.
It will work on loops of the form:
/// GuardBB:
/// br cond1, Preheader, ExitSucc <== GuardBranch
/// Preheader:
/// br Header
/// Header:
/// ...
/// br Latch
/// Latch:
/// br cond2, Header, ExitBlock
/// ExitBlock:
/// br ExitSucc
/// ExitSucc:
Prior discussions leading upto the decision to introduce the loop guard
API: http://lists.llvm.org/pipermail/llvm-dev/2019-May/132607.html
Reviewer: reames, kbarton, hfinkel, jdoerfert, Meinersbur, dmgreen
Reviewed By: reames
Subscribers: wuzish, hiraditya, jsji, llvm-commits, bmahjour, etiotto
Tag: LLVM
Differential Revision: https://reviews.llvm.org/D63885
llvm-svn: 367033
2019-07-26 00:13:18 +08:00
|
|
|
EXPECT_EQ(L->getLoopGuardBranch(), Guard);
|
|
|
|
EXPECT_TRUE(L->isGuarded());
|
2019-12-13 03:18:39 +08:00
|
|
|
EXPECT_TRUE(L->isRotatedForm());
|
[LOOPINFO] Extend Loop object to add utilities to get the loop bounds,
step, and loop induction variable.
Summary: This PR extends the loop object with more utilities to get loop
bounds, step, and loop induction variable. There already exists passes
which try to obtain the loop induction variable in their own pass, e.g.
loop interchange. It would be useful to have a common area to get these
information.
/// Example:
/// for (int i = lb; i < ub; i+=step)
/// <loop body>
/// --- pseudo LLVMIR ---
/// beforeloop:
/// guardcmp = (lb < ub)
/// if (guardcmp) goto preheader; else goto afterloop
/// preheader:
/// loop:
/// i1 = phi[{lb, preheader}, {i2, latch}]
/// <loop body>
/// i2 = i1 + step
/// latch:
/// cmp = (i2 < ub)
/// if (cmp) goto loop
/// exit:
/// afterloop:
///
/// getBounds
/// getInitialIVValue --> lb
/// getStepInst --> i2 = i1 + step
/// getStepValue --> step
/// getFinalIVValue --> ub
/// getCanonicalPredicate --> '<'
/// getDirection --> Increasing
/// getInductionVariable --> i1
/// getAuxiliaryInductionVariable --> {i1}
/// isCanonical --> false
Reviewers: kbarton, hfinkel, dmgreen, Meinersbur, jdoerfert, syzaara,
fhahn
Reviewed By: kbarton
Subscribers: tvvikram, bmahjour, etiotto, fhahn, jsji, hiraditya,
llvm-commits
Tag: LLVM
Differential Revision: https://reviews.llvm.org/D60565
llvm-svn: 362644
2019-06-06 04:42:47 +08:00
|
|
|
});
|
|
|
|
}
|
|
|
|
|
|
|
|
TEST(LoopInfoTest, LoopWithLatchCmpNE) {
|
|
|
|
const char *ModuleStr =
|
|
|
|
"define void @foo(i32* %A, i32 %ub) {\n"
|
|
|
|
"entry:\n"
|
|
|
|
" %guardcmp = icmp slt i32 0, %ub\n"
|
|
|
|
" br i1 %guardcmp, label %for.preheader, label %for.end\n"
|
|
|
|
"for.preheader:\n"
|
|
|
|
" br label %for.body\n"
|
|
|
|
"for.body:\n"
|
|
|
|
" %i = phi i32 [ 0, %for.preheader ], [ %inc, %for.body ]\n"
|
|
|
|
" %idxprom = sext i32 %i to i64\n"
|
|
|
|
" %arrayidx = getelementptr inbounds i32, i32* %A, i64 %idxprom\n"
|
|
|
|
" store i32 %i, i32* %arrayidx, align 4\n"
|
|
|
|
" %inc = add nsw i32 %i, 1\n"
|
|
|
|
" %cmp = icmp ne i32 %i, %ub\n"
|
|
|
|
" br i1 %cmp, label %for.body, label %for.exit\n"
|
|
|
|
"for.exit:\n"
|
|
|
|
" br label %for.end\n"
|
|
|
|
"for.end:\n"
|
|
|
|
" ret void\n"
|
|
|
|
"}\n";
|
|
|
|
|
|
|
|
// Parse the module.
|
|
|
|
LLVMContext Context;
|
|
|
|
std::unique_ptr<Module> M = makeLLVMModule(Context, ModuleStr);
|
|
|
|
|
|
|
|
runWithLoopInfoPlus(
|
|
|
|
*M, "foo",
|
[LOOPINFO] Introduce the loop guard API.
Summary:
This is the first patch for the loop guard. We introduced
getLoopGuardBranch() and isGuarded().
This currently only works on simplified loop, as it requires a preheader
and a latch to identify the guard.
It will work on loops of the form:
/// GuardBB:
/// br cond1, Preheader, ExitSucc <== GuardBranch
/// Preheader:
/// br Header
/// Header:
/// ...
/// br Latch
/// Latch:
/// br cond2, Header, ExitBlock
/// ExitBlock:
/// br ExitSucc
/// ExitSucc:
Prior discussions leading upto the decision to introduce the loop guard
API: http://lists.llvm.org/pipermail/llvm-dev/2019-May/132607.html
Reviewer: reames, kbarton, hfinkel, jdoerfert, Meinersbur, dmgreen
Reviewed By: reames
Subscribers: wuzish, hiraditya, jsji, llvm-commits, bmahjour, etiotto
Tag: LLVM
Differential Revision: https://reviews.llvm.org/D63885
llvm-svn: 367033
2019-07-26 00:13:18 +08:00
|
|
|
[&](Function &F, LoopInfo &LI, ScalarEvolution &SE) {
|
[LOOPINFO] Extend Loop object to add utilities to get the loop bounds,
step, and loop induction variable.
Summary: This PR extends the loop object with more utilities to get loop
bounds, step, and loop induction variable. There already exists passes
which try to obtain the loop induction variable in their own pass, e.g.
loop interchange. It would be useful to have a common area to get these
information.
/// Example:
/// for (int i = lb; i < ub; i+=step)
/// <loop body>
/// --- pseudo LLVMIR ---
/// beforeloop:
/// guardcmp = (lb < ub)
/// if (guardcmp) goto preheader; else goto afterloop
/// preheader:
/// loop:
/// i1 = phi[{lb, preheader}, {i2, latch}]
/// <loop body>
/// i2 = i1 + step
/// latch:
/// cmp = (i2 < ub)
/// if (cmp) goto loop
/// exit:
/// afterloop:
///
/// getBounds
/// getInitialIVValue --> lb
/// getStepInst --> i2 = i1 + step
/// getStepValue --> step
/// getFinalIVValue --> ub
/// getCanonicalPredicate --> '<'
/// getDirection --> Increasing
/// getInductionVariable --> i1
/// getAuxiliaryInductionVariable --> {i1}
/// isCanonical --> false
Reviewers: kbarton, hfinkel, dmgreen, Meinersbur, jdoerfert, syzaara,
fhahn
Reviewed By: kbarton
Subscribers: tvvikram, bmahjour, etiotto, fhahn, jsji, hiraditya,
llvm-commits
Tag: LLVM
Differential Revision: https://reviews.llvm.org/D60565
llvm-svn: 362644
2019-06-06 04:42:47 +08:00
|
|
|
Function::iterator FI = F.begin();
|
[LOOPINFO] Introduce the loop guard API.
Summary:
This is the first patch for the loop guard. We introduced
getLoopGuardBranch() and isGuarded().
This currently only works on simplified loop, as it requires a preheader
and a latch to identify the guard.
It will work on loops of the form:
/// GuardBB:
/// br cond1, Preheader, ExitSucc <== GuardBranch
/// Preheader:
/// br Header
/// Header:
/// ...
/// br Latch
/// Latch:
/// br cond2, Header, ExitBlock
/// ExitBlock:
/// br ExitSucc
/// ExitSucc:
Prior discussions leading upto the decision to introduce the loop guard
API: http://lists.llvm.org/pipermail/llvm-dev/2019-May/132607.html
Reviewer: reames, kbarton, hfinkel, jdoerfert, Meinersbur, dmgreen
Reviewed By: reames
Subscribers: wuzish, hiraditya, jsji, llvm-commits, bmahjour, etiotto
Tag: LLVM
Differential Revision: https://reviews.llvm.org/D63885
llvm-svn: 367033
2019-07-26 00:13:18 +08:00
|
|
|
BasicBlock *Entry = &*(FI);
|
|
|
|
BranchInst *Guard = dyn_cast<BranchInst>(Entry->getTerminator());
|
[LOOPINFO] Extend Loop object to add utilities to get the loop bounds,
step, and loop induction variable.
Summary: This PR extends the loop object with more utilities to get loop
bounds, step, and loop induction variable. There already exists passes
which try to obtain the loop induction variable in their own pass, e.g.
loop interchange. It would be useful to have a common area to get these
information.
/// Example:
/// for (int i = lb; i < ub; i+=step)
/// <loop body>
/// --- pseudo LLVMIR ---
/// beforeloop:
/// guardcmp = (lb < ub)
/// if (guardcmp) goto preheader; else goto afterloop
/// preheader:
/// loop:
/// i1 = phi[{lb, preheader}, {i2, latch}]
/// <loop body>
/// i2 = i1 + step
/// latch:
/// cmp = (i2 < ub)
/// if (cmp) goto loop
/// exit:
/// afterloop:
///
/// getBounds
/// getInitialIVValue --> lb
/// getStepInst --> i2 = i1 + step
/// getStepValue --> step
/// getFinalIVValue --> ub
/// getCanonicalPredicate --> '<'
/// getDirection --> Increasing
/// getInductionVariable --> i1
/// getAuxiliaryInductionVariable --> {i1}
/// isCanonical --> false
Reviewers: kbarton, hfinkel, dmgreen, Meinersbur, jdoerfert, syzaara,
fhahn
Reviewed By: kbarton
Subscribers: tvvikram, bmahjour, etiotto, fhahn, jsji, hiraditya,
llvm-commits
Tag: LLVM
Differential Revision: https://reviews.llvm.org/D60565
llvm-svn: 362644
2019-06-06 04:42:47 +08:00
|
|
|
// First two basic block are entry and for.preheader - skip them.
|
|
|
|
++FI;
|
|
|
|
BasicBlock *Header = &*(++FI);
|
|
|
|
assert(Header->getName() == "for.body");
|
|
|
|
Loop *L = LI.getLoopFor(Header);
|
|
|
|
EXPECT_NE(L, nullptr);
|
|
|
|
|
|
|
|
Optional<Loop::LoopBounds> Bounds = L->getBounds(SE);
|
|
|
|
EXPECT_NE(Bounds, None);
|
|
|
|
ConstantInt *InitialIVValue =
|
|
|
|
dyn_cast<ConstantInt>(&Bounds->getInitialIVValue());
|
|
|
|
EXPECT_TRUE(InitialIVValue && InitialIVValue->isZero());
|
|
|
|
EXPECT_EQ(Bounds->getStepInst().getName(), "inc");
|
|
|
|
ConstantInt *StepValue =
|
|
|
|
dyn_cast_or_null<ConstantInt>(Bounds->getStepValue());
|
|
|
|
EXPECT_TRUE(StepValue && StepValue->isOne());
|
|
|
|
EXPECT_EQ(Bounds->getFinalIVValue().getName(), "ub");
|
|
|
|
EXPECT_EQ(Bounds->getCanonicalPredicate(), ICmpInst::ICMP_SLT);
|
|
|
|
EXPECT_EQ(Bounds->getDirection(),
|
|
|
|
Loop::LoopBounds::Direction::Increasing);
|
|
|
|
EXPECT_EQ(L->getInductionVariable(SE)->getName(), "i");
|
[LOOPINFO] Introduce the loop guard API.
Summary:
This is the first patch for the loop guard. We introduced
getLoopGuardBranch() and isGuarded().
This currently only works on simplified loop, as it requires a preheader
and a latch to identify the guard.
It will work on loops of the form:
/// GuardBB:
/// br cond1, Preheader, ExitSucc <== GuardBranch
/// Preheader:
/// br Header
/// Header:
/// ...
/// br Latch
/// Latch:
/// br cond2, Header, ExitBlock
/// ExitBlock:
/// br ExitSucc
/// ExitSucc:
Prior discussions leading upto the decision to introduce the loop guard
API: http://lists.llvm.org/pipermail/llvm-dev/2019-May/132607.html
Reviewer: reames, kbarton, hfinkel, jdoerfert, Meinersbur, dmgreen
Reviewed By: reames
Subscribers: wuzish, hiraditya, jsji, llvm-commits, bmahjour, etiotto
Tag: LLVM
Differential Revision: https://reviews.llvm.org/D63885
llvm-svn: 367033
2019-07-26 00:13:18 +08:00
|
|
|
EXPECT_EQ(L->getLoopGuardBranch(), Guard);
|
|
|
|
EXPECT_TRUE(L->isGuarded());
|
2019-12-13 03:18:39 +08:00
|
|
|
EXPECT_TRUE(L->isRotatedForm());
|
[LOOPINFO] Extend Loop object to add utilities to get the loop bounds,
step, and loop induction variable.
Summary: This PR extends the loop object with more utilities to get loop
bounds, step, and loop induction variable. There already exists passes
which try to obtain the loop induction variable in their own pass, e.g.
loop interchange. It would be useful to have a common area to get these
information.
/// Example:
/// for (int i = lb; i < ub; i+=step)
/// <loop body>
/// --- pseudo LLVMIR ---
/// beforeloop:
/// guardcmp = (lb < ub)
/// if (guardcmp) goto preheader; else goto afterloop
/// preheader:
/// loop:
/// i1 = phi[{lb, preheader}, {i2, latch}]
/// <loop body>
/// i2 = i1 + step
/// latch:
/// cmp = (i2 < ub)
/// if (cmp) goto loop
/// exit:
/// afterloop:
///
/// getBounds
/// getInitialIVValue --> lb
/// getStepInst --> i2 = i1 + step
/// getStepValue --> step
/// getFinalIVValue --> ub
/// getCanonicalPredicate --> '<'
/// getDirection --> Increasing
/// getInductionVariable --> i1
/// getAuxiliaryInductionVariable --> {i1}
/// isCanonical --> false
Reviewers: kbarton, hfinkel, dmgreen, Meinersbur, jdoerfert, syzaara,
fhahn
Reviewed By: kbarton
Subscribers: tvvikram, bmahjour, etiotto, fhahn, jsji, hiraditya,
llvm-commits
Tag: LLVM
Differential Revision: https://reviews.llvm.org/D60565
llvm-svn: 362644
2019-06-06 04:42:47 +08:00
|
|
|
});
|
|
|
|
}
|
|
|
|
|
|
|
|
TEST(LoopInfoTest, LoopWithGuardCmpSLE) {
|
|
|
|
const char *ModuleStr =
|
|
|
|
"define void @foo(i32* %A, i32 %ub) {\n"
|
|
|
|
"entry:\n"
|
|
|
|
" %ubPlusOne = add i32 %ub, 1\n"
|
|
|
|
" %guardcmp = icmp sle i32 0, %ub\n"
|
|
|
|
" br i1 %guardcmp, label %for.preheader, label %for.end\n"
|
|
|
|
"for.preheader:\n"
|
|
|
|
" br label %for.body\n"
|
|
|
|
"for.body:\n"
|
|
|
|
" %i = phi i32 [ 0, %for.preheader ], [ %inc, %for.body ]\n"
|
|
|
|
" %idxprom = sext i32 %i to i64\n"
|
|
|
|
" %arrayidx = getelementptr inbounds i32, i32* %A, i64 %idxprom\n"
|
|
|
|
" store i32 %i, i32* %arrayidx, align 4\n"
|
|
|
|
" %inc = add nsw i32 %i, 1\n"
|
|
|
|
" %cmp = icmp ne i32 %i, %ubPlusOne\n"
|
|
|
|
" br i1 %cmp, label %for.body, label %for.exit\n"
|
|
|
|
"for.exit:\n"
|
|
|
|
" br label %for.end\n"
|
|
|
|
"for.end:\n"
|
|
|
|
" ret void\n"
|
|
|
|
"}\n";
|
|
|
|
|
|
|
|
// Parse the module.
|
|
|
|
LLVMContext Context;
|
|
|
|
std::unique_ptr<Module> M = makeLLVMModule(Context, ModuleStr);
|
|
|
|
|
|
|
|
runWithLoopInfoPlus(
|
|
|
|
*M, "foo",
|
[LOOPINFO] Introduce the loop guard API.
Summary:
This is the first patch for the loop guard. We introduced
getLoopGuardBranch() and isGuarded().
This currently only works on simplified loop, as it requires a preheader
and a latch to identify the guard.
It will work on loops of the form:
/// GuardBB:
/// br cond1, Preheader, ExitSucc <== GuardBranch
/// Preheader:
/// br Header
/// Header:
/// ...
/// br Latch
/// Latch:
/// br cond2, Header, ExitBlock
/// ExitBlock:
/// br ExitSucc
/// ExitSucc:
Prior discussions leading upto the decision to introduce the loop guard
API: http://lists.llvm.org/pipermail/llvm-dev/2019-May/132607.html
Reviewer: reames, kbarton, hfinkel, jdoerfert, Meinersbur, dmgreen
Reviewed By: reames
Subscribers: wuzish, hiraditya, jsji, llvm-commits, bmahjour, etiotto
Tag: LLVM
Differential Revision: https://reviews.llvm.org/D63885
llvm-svn: 367033
2019-07-26 00:13:18 +08:00
|
|
|
[&](Function &F, LoopInfo &LI, ScalarEvolution &SE) {
|
[LOOPINFO] Extend Loop object to add utilities to get the loop bounds,
step, and loop induction variable.
Summary: This PR extends the loop object with more utilities to get loop
bounds, step, and loop induction variable. There already exists passes
which try to obtain the loop induction variable in their own pass, e.g.
loop interchange. It would be useful to have a common area to get these
information.
/// Example:
/// for (int i = lb; i < ub; i+=step)
/// <loop body>
/// --- pseudo LLVMIR ---
/// beforeloop:
/// guardcmp = (lb < ub)
/// if (guardcmp) goto preheader; else goto afterloop
/// preheader:
/// loop:
/// i1 = phi[{lb, preheader}, {i2, latch}]
/// <loop body>
/// i2 = i1 + step
/// latch:
/// cmp = (i2 < ub)
/// if (cmp) goto loop
/// exit:
/// afterloop:
///
/// getBounds
/// getInitialIVValue --> lb
/// getStepInst --> i2 = i1 + step
/// getStepValue --> step
/// getFinalIVValue --> ub
/// getCanonicalPredicate --> '<'
/// getDirection --> Increasing
/// getInductionVariable --> i1
/// getAuxiliaryInductionVariable --> {i1}
/// isCanonical --> false
Reviewers: kbarton, hfinkel, dmgreen, Meinersbur, jdoerfert, syzaara,
fhahn
Reviewed By: kbarton
Subscribers: tvvikram, bmahjour, etiotto, fhahn, jsji, hiraditya,
llvm-commits
Tag: LLVM
Differential Revision: https://reviews.llvm.org/D60565
llvm-svn: 362644
2019-06-06 04:42:47 +08:00
|
|
|
Function::iterator FI = F.begin();
|
[LOOPINFO] Introduce the loop guard API.
Summary:
This is the first patch for the loop guard. We introduced
getLoopGuardBranch() and isGuarded().
This currently only works on simplified loop, as it requires a preheader
and a latch to identify the guard.
It will work on loops of the form:
/// GuardBB:
/// br cond1, Preheader, ExitSucc <== GuardBranch
/// Preheader:
/// br Header
/// Header:
/// ...
/// br Latch
/// Latch:
/// br cond2, Header, ExitBlock
/// ExitBlock:
/// br ExitSucc
/// ExitSucc:
Prior discussions leading upto the decision to introduce the loop guard
API: http://lists.llvm.org/pipermail/llvm-dev/2019-May/132607.html
Reviewer: reames, kbarton, hfinkel, jdoerfert, Meinersbur, dmgreen
Reviewed By: reames
Subscribers: wuzish, hiraditya, jsji, llvm-commits, bmahjour, etiotto
Tag: LLVM
Differential Revision: https://reviews.llvm.org/D63885
llvm-svn: 367033
2019-07-26 00:13:18 +08:00
|
|
|
BasicBlock *Entry = &*(FI);
|
|
|
|
BranchInst *Guard = dyn_cast<BranchInst>(Entry->getTerminator());
|
[LOOPINFO] Extend Loop object to add utilities to get the loop bounds,
step, and loop induction variable.
Summary: This PR extends the loop object with more utilities to get loop
bounds, step, and loop induction variable. There already exists passes
which try to obtain the loop induction variable in their own pass, e.g.
loop interchange. It would be useful to have a common area to get these
information.
/// Example:
/// for (int i = lb; i < ub; i+=step)
/// <loop body>
/// --- pseudo LLVMIR ---
/// beforeloop:
/// guardcmp = (lb < ub)
/// if (guardcmp) goto preheader; else goto afterloop
/// preheader:
/// loop:
/// i1 = phi[{lb, preheader}, {i2, latch}]
/// <loop body>
/// i2 = i1 + step
/// latch:
/// cmp = (i2 < ub)
/// if (cmp) goto loop
/// exit:
/// afterloop:
///
/// getBounds
/// getInitialIVValue --> lb
/// getStepInst --> i2 = i1 + step
/// getStepValue --> step
/// getFinalIVValue --> ub
/// getCanonicalPredicate --> '<'
/// getDirection --> Increasing
/// getInductionVariable --> i1
/// getAuxiliaryInductionVariable --> {i1}
/// isCanonical --> false
Reviewers: kbarton, hfinkel, dmgreen, Meinersbur, jdoerfert, syzaara,
fhahn
Reviewed By: kbarton
Subscribers: tvvikram, bmahjour, etiotto, fhahn, jsji, hiraditya,
llvm-commits
Tag: LLVM
Differential Revision: https://reviews.llvm.org/D60565
llvm-svn: 362644
2019-06-06 04:42:47 +08:00
|
|
|
// First two basic block are entry and for.preheader - skip them.
|
|
|
|
++FI;
|
|
|
|
BasicBlock *Header = &*(++FI);
|
|
|
|
assert(Header->getName() == "for.body");
|
|
|
|
Loop *L = LI.getLoopFor(Header);
|
|
|
|
EXPECT_NE(L, nullptr);
|
|
|
|
|
|
|
|
Optional<Loop::LoopBounds> Bounds = L->getBounds(SE);
|
|
|
|
EXPECT_NE(Bounds, None);
|
|
|
|
ConstantInt *InitialIVValue =
|
|
|
|
dyn_cast<ConstantInt>(&Bounds->getInitialIVValue());
|
|
|
|
EXPECT_TRUE(InitialIVValue && InitialIVValue->isZero());
|
|
|
|
EXPECT_EQ(Bounds->getStepInst().getName(), "inc");
|
|
|
|
ConstantInt *StepValue =
|
|
|
|
dyn_cast_or_null<ConstantInt>(Bounds->getStepValue());
|
|
|
|
EXPECT_TRUE(StepValue && StepValue->isOne());
|
|
|
|
EXPECT_EQ(Bounds->getFinalIVValue().getName(), "ubPlusOne");
|
|
|
|
EXPECT_EQ(Bounds->getCanonicalPredicate(), ICmpInst::ICMP_SLT);
|
|
|
|
EXPECT_EQ(Bounds->getDirection(),
|
|
|
|
Loop::LoopBounds::Direction::Increasing);
|
|
|
|
EXPECT_EQ(L->getInductionVariable(SE)->getName(), "i");
|
[LOOPINFO] Introduce the loop guard API.
Summary:
This is the first patch for the loop guard. We introduced
getLoopGuardBranch() and isGuarded().
This currently only works on simplified loop, as it requires a preheader
and a latch to identify the guard.
It will work on loops of the form:
/// GuardBB:
/// br cond1, Preheader, ExitSucc <== GuardBranch
/// Preheader:
/// br Header
/// Header:
/// ...
/// br Latch
/// Latch:
/// br cond2, Header, ExitBlock
/// ExitBlock:
/// br ExitSucc
/// ExitSucc:
Prior discussions leading upto the decision to introduce the loop guard
API: http://lists.llvm.org/pipermail/llvm-dev/2019-May/132607.html
Reviewer: reames, kbarton, hfinkel, jdoerfert, Meinersbur, dmgreen
Reviewed By: reames
Subscribers: wuzish, hiraditya, jsji, llvm-commits, bmahjour, etiotto
Tag: LLVM
Differential Revision: https://reviews.llvm.org/D63885
llvm-svn: 367033
2019-07-26 00:13:18 +08:00
|
|
|
EXPECT_EQ(L->getLoopGuardBranch(), Guard);
|
|
|
|
EXPECT_TRUE(L->isGuarded());
|
2019-12-13 03:18:39 +08:00
|
|
|
EXPECT_TRUE(L->isRotatedForm());
|
[LOOPINFO] Extend Loop object to add utilities to get the loop bounds,
step, and loop induction variable.
Summary: This PR extends the loop object with more utilities to get loop
bounds, step, and loop induction variable. There already exists passes
which try to obtain the loop induction variable in their own pass, e.g.
loop interchange. It would be useful to have a common area to get these
information.
/// Example:
/// for (int i = lb; i < ub; i+=step)
/// <loop body>
/// --- pseudo LLVMIR ---
/// beforeloop:
/// guardcmp = (lb < ub)
/// if (guardcmp) goto preheader; else goto afterloop
/// preheader:
/// loop:
/// i1 = phi[{lb, preheader}, {i2, latch}]
/// <loop body>
/// i2 = i1 + step
/// latch:
/// cmp = (i2 < ub)
/// if (cmp) goto loop
/// exit:
/// afterloop:
///
/// getBounds
/// getInitialIVValue --> lb
/// getStepInst --> i2 = i1 + step
/// getStepValue --> step
/// getFinalIVValue --> ub
/// getCanonicalPredicate --> '<'
/// getDirection --> Increasing
/// getInductionVariable --> i1
/// getAuxiliaryInductionVariable --> {i1}
/// isCanonical --> false
Reviewers: kbarton, hfinkel, dmgreen, Meinersbur, jdoerfert, syzaara,
fhahn
Reviewed By: kbarton
Subscribers: tvvikram, bmahjour, etiotto, fhahn, jsji, hiraditya,
llvm-commits
Tag: LLVM
Differential Revision: https://reviews.llvm.org/D60565
llvm-svn: 362644
2019-06-06 04:42:47 +08:00
|
|
|
});
|
|
|
|
}
|
|
|
|
|
|
|
|
TEST(LoopInfoTest, LoopNonConstantStep) {
|
|
|
|
const char *ModuleStr =
|
|
|
|
"define void @foo(i32* %A, i32 %ub, i32 %step) {\n"
|
|
|
|
"entry:\n"
|
|
|
|
" %guardcmp = icmp slt i32 0, %ub\n"
|
|
|
|
" br i1 %guardcmp, label %for.preheader, label %for.end\n"
|
|
|
|
"for.preheader:\n"
|
|
|
|
" br label %for.body\n"
|
|
|
|
"for.body:\n"
|
|
|
|
" %i = phi i32 [ 0, %for.preheader ], [ %inc, %for.body ]\n"
|
|
|
|
" %idxprom = zext i32 %i to i64\n"
|
|
|
|
" %arrayidx = getelementptr inbounds i32, i32* %A, i64 %idxprom\n"
|
|
|
|
" store i32 %i, i32* %arrayidx, align 4\n"
|
|
|
|
" %inc = add nsw i32 %i, %step\n"
|
|
|
|
" %cmp = icmp slt i32 %inc, %ub\n"
|
|
|
|
" br i1 %cmp, label %for.body, label %for.exit\n"
|
|
|
|
"for.exit:\n"
|
|
|
|
" br label %for.end\n"
|
|
|
|
"for.end:\n"
|
|
|
|
" ret void\n"
|
|
|
|
"}\n";
|
|
|
|
|
|
|
|
// Parse the module.
|
|
|
|
LLVMContext Context;
|
|
|
|
std::unique_ptr<Module> M = makeLLVMModule(Context, ModuleStr);
|
|
|
|
|
|
|
|
runWithLoopInfoPlus(
|
|
|
|
*M, "foo",
|
[LOOPINFO] Introduce the loop guard API.
Summary:
This is the first patch for the loop guard. We introduced
getLoopGuardBranch() and isGuarded().
This currently only works on simplified loop, as it requires a preheader
and a latch to identify the guard.
It will work on loops of the form:
/// GuardBB:
/// br cond1, Preheader, ExitSucc <== GuardBranch
/// Preheader:
/// br Header
/// Header:
/// ...
/// br Latch
/// Latch:
/// br cond2, Header, ExitBlock
/// ExitBlock:
/// br ExitSucc
/// ExitSucc:
Prior discussions leading upto the decision to introduce the loop guard
API: http://lists.llvm.org/pipermail/llvm-dev/2019-May/132607.html
Reviewer: reames, kbarton, hfinkel, jdoerfert, Meinersbur, dmgreen
Reviewed By: reames
Subscribers: wuzish, hiraditya, jsji, llvm-commits, bmahjour, etiotto
Tag: LLVM
Differential Revision: https://reviews.llvm.org/D63885
llvm-svn: 367033
2019-07-26 00:13:18 +08:00
|
|
|
[&](Function &F, LoopInfo &LI, ScalarEvolution &SE) {
|
[LOOPINFO] Extend Loop object to add utilities to get the loop bounds,
step, and loop induction variable.
Summary: This PR extends the loop object with more utilities to get loop
bounds, step, and loop induction variable. There already exists passes
which try to obtain the loop induction variable in their own pass, e.g.
loop interchange. It would be useful to have a common area to get these
information.
/// Example:
/// for (int i = lb; i < ub; i+=step)
/// <loop body>
/// --- pseudo LLVMIR ---
/// beforeloop:
/// guardcmp = (lb < ub)
/// if (guardcmp) goto preheader; else goto afterloop
/// preheader:
/// loop:
/// i1 = phi[{lb, preheader}, {i2, latch}]
/// <loop body>
/// i2 = i1 + step
/// latch:
/// cmp = (i2 < ub)
/// if (cmp) goto loop
/// exit:
/// afterloop:
///
/// getBounds
/// getInitialIVValue --> lb
/// getStepInst --> i2 = i1 + step
/// getStepValue --> step
/// getFinalIVValue --> ub
/// getCanonicalPredicate --> '<'
/// getDirection --> Increasing
/// getInductionVariable --> i1
/// getAuxiliaryInductionVariable --> {i1}
/// isCanonical --> false
Reviewers: kbarton, hfinkel, dmgreen, Meinersbur, jdoerfert, syzaara,
fhahn
Reviewed By: kbarton
Subscribers: tvvikram, bmahjour, etiotto, fhahn, jsji, hiraditya,
llvm-commits
Tag: LLVM
Differential Revision: https://reviews.llvm.org/D60565
llvm-svn: 362644
2019-06-06 04:42:47 +08:00
|
|
|
Function::iterator FI = F.begin();
|
[LOOPINFO] Introduce the loop guard API.
Summary:
This is the first patch for the loop guard. We introduced
getLoopGuardBranch() and isGuarded().
This currently only works on simplified loop, as it requires a preheader
and a latch to identify the guard.
It will work on loops of the form:
/// GuardBB:
/// br cond1, Preheader, ExitSucc <== GuardBranch
/// Preheader:
/// br Header
/// Header:
/// ...
/// br Latch
/// Latch:
/// br cond2, Header, ExitBlock
/// ExitBlock:
/// br ExitSucc
/// ExitSucc:
Prior discussions leading upto the decision to introduce the loop guard
API: http://lists.llvm.org/pipermail/llvm-dev/2019-May/132607.html
Reviewer: reames, kbarton, hfinkel, jdoerfert, Meinersbur, dmgreen
Reviewed By: reames
Subscribers: wuzish, hiraditya, jsji, llvm-commits, bmahjour, etiotto
Tag: LLVM
Differential Revision: https://reviews.llvm.org/D63885
llvm-svn: 367033
2019-07-26 00:13:18 +08:00
|
|
|
BasicBlock *Entry = &*(FI);
|
|
|
|
BranchInst *Guard = dyn_cast<BranchInst>(Entry->getTerminator());
|
[LOOPINFO] Extend Loop object to add utilities to get the loop bounds,
step, and loop induction variable.
Summary: This PR extends the loop object with more utilities to get loop
bounds, step, and loop induction variable. There already exists passes
which try to obtain the loop induction variable in their own pass, e.g.
loop interchange. It would be useful to have a common area to get these
information.
/// Example:
/// for (int i = lb; i < ub; i+=step)
/// <loop body>
/// --- pseudo LLVMIR ---
/// beforeloop:
/// guardcmp = (lb < ub)
/// if (guardcmp) goto preheader; else goto afterloop
/// preheader:
/// loop:
/// i1 = phi[{lb, preheader}, {i2, latch}]
/// <loop body>
/// i2 = i1 + step
/// latch:
/// cmp = (i2 < ub)
/// if (cmp) goto loop
/// exit:
/// afterloop:
///
/// getBounds
/// getInitialIVValue --> lb
/// getStepInst --> i2 = i1 + step
/// getStepValue --> step
/// getFinalIVValue --> ub
/// getCanonicalPredicate --> '<'
/// getDirection --> Increasing
/// getInductionVariable --> i1
/// getAuxiliaryInductionVariable --> {i1}
/// isCanonical --> false
Reviewers: kbarton, hfinkel, dmgreen, Meinersbur, jdoerfert, syzaara,
fhahn
Reviewed By: kbarton
Subscribers: tvvikram, bmahjour, etiotto, fhahn, jsji, hiraditya,
llvm-commits
Tag: LLVM
Differential Revision: https://reviews.llvm.org/D60565
llvm-svn: 362644
2019-06-06 04:42:47 +08:00
|
|
|
// First two basic block are entry and for.preheader - skip them.
|
|
|
|
++FI;
|
|
|
|
BasicBlock *Header = &*(++FI);
|
|
|
|
assert(Header->getName() == "for.body");
|
|
|
|
Loop *L = LI.getLoopFor(Header);
|
|
|
|
EXPECT_NE(L, nullptr);
|
|
|
|
|
|
|
|
Optional<Loop::LoopBounds> Bounds = L->getBounds(SE);
|
|
|
|
EXPECT_NE(Bounds, None);
|
|
|
|
ConstantInt *InitialIVValue =
|
|
|
|
dyn_cast<ConstantInt>(&Bounds->getInitialIVValue());
|
|
|
|
EXPECT_TRUE(InitialIVValue && InitialIVValue->isZero());
|
|
|
|
EXPECT_EQ(Bounds->getStepInst().getName(), "inc");
|
|
|
|
EXPECT_EQ(Bounds->getStepValue()->getName(), "step");
|
|
|
|
EXPECT_EQ(Bounds->getFinalIVValue().getName(), "ub");
|
|
|
|
EXPECT_EQ(Bounds->getCanonicalPredicate(), ICmpInst::ICMP_SLT);
|
|
|
|
EXPECT_EQ(Bounds->getDirection(), Loop::LoopBounds::Direction::Unknown);
|
|
|
|
EXPECT_EQ(L->getInductionVariable(SE)->getName(), "i");
|
[LOOPINFO] Introduce the loop guard API.
Summary:
This is the first patch for the loop guard. We introduced
getLoopGuardBranch() and isGuarded().
This currently only works on simplified loop, as it requires a preheader
and a latch to identify the guard.
It will work on loops of the form:
/// GuardBB:
/// br cond1, Preheader, ExitSucc <== GuardBranch
/// Preheader:
/// br Header
/// Header:
/// ...
/// br Latch
/// Latch:
/// br cond2, Header, ExitBlock
/// ExitBlock:
/// br ExitSucc
/// ExitSucc:
Prior discussions leading upto the decision to introduce the loop guard
API: http://lists.llvm.org/pipermail/llvm-dev/2019-May/132607.html
Reviewer: reames, kbarton, hfinkel, jdoerfert, Meinersbur, dmgreen
Reviewed By: reames
Subscribers: wuzish, hiraditya, jsji, llvm-commits, bmahjour, etiotto
Tag: LLVM
Differential Revision: https://reviews.llvm.org/D63885
llvm-svn: 367033
2019-07-26 00:13:18 +08:00
|
|
|
EXPECT_EQ(L->getLoopGuardBranch(), Guard);
|
|
|
|
EXPECT_TRUE(L->isGuarded());
|
2019-12-13 03:18:39 +08:00
|
|
|
EXPECT_TRUE(L->isRotatedForm());
|
[LOOPINFO] Extend Loop object to add utilities to get the loop bounds,
step, and loop induction variable.
Summary: This PR extends the loop object with more utilities to get loop
bounds, step, and loop induction variable. There already exists passes
which try to obtain the loop induction variable in their own pass, e.g.
loop interchange. It would be useful to have a common area to get these
information.
/// Example:
/// for (int i = lb; i < ub; i+=step)
/// <loop body>
/// --- pseudo LLVMIR ---
/// beforeloop:
/// guardcmp = (lb < ub)
/// if (guardcmp) goto preheader; else goto afterloop
/// preheader:
/// loop:
/// i1 = phi[{lb, preheader}, {i2, latch}]
/// <loop body>
/// i2 = i1 + step
/// latch:
/// cmp = (i2 < ub)
/// if (cmp) goto loop
/// exit:
/// afterloop:
///
/// getBounds
/// getInitialIVValue --> lb
/// getStepInst --> i2 = i1 + step
/// getStepValue --> step
/// getFinalIVValue --> ub
/// getCanonicalPredicate --> '<'
/// getDirection --> Increasing
/// getInductionVariable --> i1
/// getAuxiliaryInductionVariable --> {i1}
/// isCanonical --> false
Reviewers: kbarton, hfinkel, dmgreen, Meinersbur, jdoerfert, syzaara,
fhahn
Reviewed By: kbarton
Subscribers: tvvikram, bmahjour, etiotto, fhahn, jsji, hiraditya,
llvm-commits
Tag: LLVM
Differential Revision: https://reviews.llvm.org/D60565
llvm-svn: 362644
2019-06-06 04:42:47 +08:00
|
|
|
});
|
|
|
|
}
|
|
|
|
|
|
|
|
TEST(LoopInfoTest, LoopUnsignedBounds) {
|
|
|
|
const char *ModuleStr =
|
|
|
|
"define void @foo(i32* %A, i32 %ub) {\n"
|
|
|
|
"entry:\n"
|
|
|
|
" %guardcmp = icmp ult i32 0, %ub\n"
|
|
|
|
" br i1 %guardcmp, label %for.preheader, label %for.end\n"
|
|
|
|
"for.preheader:\n"
|
|
|
|
" br label %for.body\n"
|
|
|
|
"for.body:\n"
|
|
|
|
" %i = phi i32 [ 0, %for.preheader ], [ %inc, %for.body ]\n"
|
|
|
|
" %idxprom = zext i32 %i to i64\n"
|
|
|
|
" %arrayidx = getelementptr inbounds i32, i32* %A, i64 %idxprom\n"
|
|
|
|
" store i32 %i, i32* %arrayidx, align 4\n"
|
|
|
|
" %inc = add i32 %i, 1\n"
|
|
|
|
" %cmp = icmp ult i32 %inc, %ub\n"
|
|
|
|
" br i1 %cmp, label %for.body, label %for.exit\n"
|
|
|
|
"for.exit:\n"
|
|
|
|
" br label %for.end\n"
|
|
|
|
"for.end:\n"
|
|
|
|
" ret void\n"
|
|
|
|
"}\n";
|
|
|
|
|
|
|
|
// Parse the module.
|
|
|
|
LLVMContext Context;
|
|
|
|
std::unique_ptr<Module> M = makeLLVMModule(Context, ModuleStr);
|
|
|
|
|
|
|
|
runWithLoopInfoPlus(
|
|
|
|
*M, "foo",
|
[LOOPINFO] Introduce the loop guard API.
Summary:
This is the first patch for the loop guard. We introduced
getLoopGuardBranch() and isGuarded().
This currently only works on simplified loop, as it requires a preheader
and a latch to identify the guard.
It will work on loops of the form:
/// GuardBB:
/// br cond1, Preheader, ExitSucc <== GuardBranch
/// Preheader:
/// br Header
/// Header:
/// ...
/// br Latch
/// Latch:
/// br cond2, Header, ExitBlock
/// ExitBlock:
/// br ExitSucc
/// ExitSucc:
Prior discussions leading upto the decision to introduce the loop guard
API: http://lists.llvm.org/pipermail/llvm-dev/2019-May/132607.html
Reviewer: reames, kbarton, hfinkel, jdoerfert, Meinersbur, dmgreen
Reviewed By: reames
Subscribers: wuzish, hiraditya, jsji, llvm-commits, bmahjour, etiotto
Tag: LLVM
Differential Revision: https://reviews.llvm.org/D63885
llvm-svn: 367033
2019-07-26 00:13:18 +08:00
|
|
|
[&](Function &F, LoopInfo &LI, ScalarEvolution &SE) {
|
[LOOPINFO] Extend Loop object to add utilities to get the loop bounds,
step, and loop induction variable.
Summary: This PR extends the loop object with more utilities to get loop
bounds, step, and loop induction variable. There already exists passes
which try to obtain the loop induction variable in their own pass, e.g.
loop interchange. It would be useful to have a common area to get these
information.
/// Example:
/// for (int i = lb; i < ub; i+=step)
/// <loop body>
/// --- pseudo LLVMIR ---
/// beforeloop:
/// guardcmp = (lb < ub)
/// if (guardcmp) goto preheader; else goto afterloop
/// preheader:
/// loop:
/// i1 = phi[{lb, preheader}, {i2, latch}]
/// <loop body>
/// i2 = i1 + step
/// latch:
/// cmp = (i2 < ub)
/// if (cmp) goto loop
/// exit:
/// afterloop:
///
/// getBounds
/// getInitialIVValue --> lb
/// getStepInst --> i2 = i1 + step
/// getStepValue --> step
/// getFinalIVValue --> ub
/// getCanonicalPredicate --> '<'
/// getDirection --> Increasing
/// getInductionVariable --> i1
/// getAuxiliaryInductionVariable --> {i1}
/// isCanonical --> false
Reviewers: kbarton, hfinkel, dmgreen, Meinersbur, jdoerfert, syzaara,
fhahn
Reviewed By: kbarton
Subscribers: tvvikram, bmahjour, etiotto, fhahn, jsji, hiraditya,
llvm-commits
Tag: LLVM
Differential Revision: https://reviews.llvm.org/D60565
llvm-svn: 362644
2019-06-06 04:42:47 +08:00
|
|
|
Function::iterator FI = F.begin();
|
[LOOPINFO] Introduce the loop guard API.
Summary:
This is the first patch for the loop guard. We introduced
getLoopGuardBranch() and isGuarded().
This currently only works on simplified loop, as it requires a preheader
and a latch to identify the guard.
It will work on loops of the form:
/// GuardBB:
/// br cond1, Preheader, ExitSucc <== GuardBranch
/// Preheader:
/// br Header
/// Header:
/// ...
/// br Latch
/// Latch:
/// br cond2, Header, ExitBlock
/// ExitBlock:
/// br ExitSucc
/// ExitSucc:
Prior discussions leading upto the decision to introduce the loop guard
API: http://lists.llvm.org/pipermail/llvm-dev/2019-May/132607.html
Reviewer: reames, kbarton, hfinkel, jdoerfert, Meinersbur, dmgreen
Reviewed By: reames
Subscribers: wuzish, hiraditya, jsji, llvm-commits, bmahjour, etiotto
Tag: LLVM
Differential Revision: https://reviews.llvm.org/D63885
llvm-svn: 367033
2019-07-26 00:13:18 +08:00
|
|
|
BasicBlock *Entry = &*(FI);
|
|
|
|
BranchInst *Guard = dyn_cast<BranchInst>(Entry->getTerminator());
|
[LOOPINFO] Extend Loop object to add utilities to get the loop bounds,
step, and loop induction variable.
Summary: This PR extends the loop object with more utilities to get loop
bounds, step, and loop induction variable. There already exists passes
which try to obtain the loop induction variable in their own pass, e.g.
loop interchange. It would be useful to have a common area to get these
information.
/// Example:
/// for (int i = lb; i < ub; i+=step)
/// <loop body>
/// --- pseudo LLVMIR ---
/// beforeloop:
/// guardcmp = (lb < ub)
/// if (guardcmp) goto preheader; else goto afterloop
/// preheader:
/// loop:
/// i1 = phi[{lb, preheader}, {i2, latch}]
/// <loop body>
/// i2 = i1 + step
/// latch:
/// cmp = (i2 < ub)
/// if (cmp) goto loop
/// exit:
/// afterloop:
///
/// getBounds
/// getInitialIVValue --> lb
/// getStepInst --> i2 = i1 + step
/// getStepValue --> step
/// getFinalIVValue --> ub
/// getCanonicalPredicate --> '<'
/// getDirection --> Increasing
/// getInductionVariable --> i1
/// getAuxiliaryInductionVariable --> {i1}
/// isCanonical --> false
Reviewers: kbarton, hfinkel, dmgreen, Meinersbur, jdoerfert, syzaara,
fhahn
Reviewed By: kbarton
Subscribers: tvvikram, bmahjour, etiotto, fhahn, jsji, hiraditya,
llvm-commits
Tag: LLVM
Differential Revision: https://reviews.llvm.org/D60565
llvm-svn: 362644
2019-06-06 04:42:47 +08:00
|
|
|
// First two basic block are entry and for.preheader - skip them.
|
|
|
|
++FI;
|
|
|
|
BasicBlock *Header = &*(++FI);
|
|
|
|
assert(Header->getName() == "for.body");
|
|
|
|
Loop *L = LI.getLoopFor(Header);
|
|
|
|
EXPECT_NE(L, nullptr);
|
|
|
|
|
|
|
|
Optional<Loop::LoopBounds> Bounds = L->getBounds(SE);
|
|
|
|
EXPECT_NE(Bounds, None);
|
|
|
|
ConstantInt *InitialIVValue =
|
|
|
|
dyn_cast<ConstantInt>(&Bounds->getInitialIVValue());
|
|
|
|
EXPECT_TRUE(InitialIVValue && InitialIVValue->isZero());
|
|
|
|
EXPECT_EQ(Bounds->getStepInst().getName(), "inc");
|
|
|
|
ConstantInt *StepValue =
|
|
|
|
dyn_cast_or_null<ConstantInt>(Bounds->getStepValue());
|
|
|
|
EXPECT_TRUE(StepValue && StepValue->isOne());
|
|
|
|
EXPECT_EQ(Bounds->getFinalIVValue().getName(), "ub");
|
|
|
|
EXPECT_EQ(Bounds->getCanonicalPredicate(), ICmpInst::ICMP_ULT);
|
|
|
|
EXPECT_EQ(Bounds->getDirection(),
|
|
|
|
Loop::LoopBounds::Direction::Increasing);
|
|
|
|
EXPECT_EQ(L->getInductionVariable(SE)->getName(), "i");
|
[LOOPINFO] Introduce the loop guard API.
Summary:
This is the first patch for the loop guard. We introduced
getLoopGuardBranch() and isGuarded().
This currently only works on simplified loop, as it requires a preheader
and a latch to identify the guard.
It will work on loops of the form:
/// GuardBB:
/// br cond1, Preheader, ExitSucc <== GuardBranch
/// Preheader:
/// br Header
/// Header:
/// ...
/// br Latch
/// Latch:
/// br cond2, Header, ExitBlock
/// ExitBlock:
/// br ExitSucc
/// ExitSucc:
Prior discussions leading upto the decision to introduce the loop guard
API: http://lists.llvm.org/pipermail/llvm-dev/2019-May/132607.html
Reviewer: reames, kbarton, hfinkel, jdoerfert, Meinersbur, dmgreen
Reviewed By: reames
Subscribers: wuzish, hiraditya, jsji, llvm-commits, bmahjour, etiotto
Tag: LLVM
Differential Revision: https://reviews.llvm.org/D63885
llvm-svn: 367033
2019-07-26 00:13:18 +08:00
|
|
|
EXPECT_EQ(L->getLoopGuardBranch(), Guard);
|
|
|
|
EXPECT_TRUE(L->isGuarded());
|
2019-12-13 03:18:39 +08:00
|
|
|
EXPECT_TRUE(L->isRotatedForm());
|
[LOOPINFO] Extend Loop object to add utilities to get the loop bounds,
step, and loop induction variable.
Summary: This PR extends the loop object with more utilities to get loop
bounds, step, and loop induction variable. There already exists passes
which try to obtain the loop induction variable in their own pass, e.g.
loop interchange. It would be useful to have a common area to get these
information.
/// Example:
/// for (int i = lb; i < ub; i+=step)
/// <loop body>
/// --- pseudo LLVMIR ---
/// beforeloop:
/// guardcmp = (lb < ub)
/// if (guardcmp) goto preheader; else goto afterloop
/// preheader:
/// loop:
/// i1 = phi[{lb, preheader}, {i2, latch}]
/// <loop body>
/// i2 = i1 + step
/// latch:
/// cmp = (i2 < ub)
/// if (cmp) goto loop
/// exit:
/// afterloop:
///
/// getBounds
/// getInitialIVValue --> lb
/// getStepInst --> i2 = i1 + step
/// getStepValue --> step
/// getFinalIVValue --> ub
/// getCanonicalPredicate --> '<'
/// getDirection --> Increasing
/// getInductionVariable --> i1
/// getAuxiliaryInductionVariable --> {i1}
/// isCanonical --> false
Reviewers: kbarton, hfinkel, dmgreen, Meinersbur, jdoerfert, syzaara,
fhahn
Reviewed By: kbarton
Subscribers: tvvikram, bmahjour, etiotto, fhahn, jsji, hiraditya,
llvm-commits
Tag: LLVM
Differential Revision: https://reviews.llvm.org/D60565
llvm-svn: 362644
2019-06-06 04:42:47 +08:00
|
|
|
});
|
|
|
|
}
|
|
|
|
|
|
|
|
TEST(LoopInfoTest, DecreasingLoop) {
|
|
|
|
const char *ModuleStr =
|
|
|
|
"define void @foo(i32* %A, i32 %ub) {\n"
|
|
|
|
"entry:\n"
|
|
|
|
" %guardcmp = icmp slt i32 0, %ub\n"
|
|
|
|
" br i1 %guardcmp, label %for.preheader, label %for.end\n"
|
|
|
|
"for.preheader:\n"
|
|
|
|
" br label %for.body\n"
|
|
|
|
"for.body:\n"
|
|
|
|
" %i = phi i32 [ %ub, %for.preheader ], [ %inc, %for.body ]\n"
|
|
|
|
" %idxprom = sext i32 %i to i64\n"
|
|
|
|
" %arrayidx = getelementptr inbounds i32, i32* %A, i64 %idxprom\n"
|
|
|
|
" store i32 %i, i32* %arrayidx, align 4\n"
|
|
|
|
" %inc = sub nsw i32 %i, 1\n"
|
|
|
|
" %cmp = icmp sgt i32 %inc, 0\n"
|
|
|
|
" br i1 %cmp, label %for.body, label %for.exit\n"
|
|
|
|
"for.exit:\n"
|
|
|
|
" br label %for.end\n"
|
|
|
|
"for.end:\n"
|
|
|
|
" ret void\n"
|
|
|
|
"}\n";
|
|
|
|
|
|
|
|
// Parse the module.
|
|
|
|
LLVMContext Context;
|
|
|
|
std::unique_ptr<Module> M = makeLLVMModule(Context, ModuleStr);
|
|
|
|
|
|
|
|
runWithLoopInfoPlus(
|
|
|
|
*M, "foo",
|
[LOOPINFO] Introduce the loop guard API.
Summary:
This is the first patch for the loop guard. We introduced
getLoopGuardBranch() and isGuarded().
This currently only works on simplified loop, as it requires a preheader
and a latch to identify the guard.
It will work on loops of the form:
/// GuardBB:
/// br cond1, Preheader, ExitSucc <== GuardBranch
/// Preheader:
/// br Header
/// Header:
/// ...
/// br Latch
/// Latch:
/// br cond2, Header, ExitBlock
/// ExitBlock:
/// br ExitSucc
/// ExitSucc:
Prior discussions leading upto the decision to introduce the loop guard
API: http://lists.llvm.org/pipermail/llvm-dev/2019-May/132607.html
Reviewer: reames, kbarton, hfinkel, jdoerfert, Meinersbur, dmgreen
Reviewed By: reames
Subscribers: wuzish, hiraditya, jsji, llvm-commits, bmahjour, etiotto
Tag: LLVM
Differential Revision: https://reviews.llvm.org/D63885
llvm-svn: 367033
2019-07-26 00:13:18 +08:00
|
|
|
[&](Function &F, LoopInfo &LI, ScalarEvolution &SE) {
|
[LOOPINFO] Extend Loop object to add utilities to get the loop bounds,
step, and loop induction variable.
Summary: This PR extends the loop object with more utilities to get loop
bounds, step, and loop induction variable. There already exists passes
which try to obtain the loop induction variable in their own pass, e.g.
loop interchange. It would be useful to have a common area to get these
information.
/// Example:
/// for (int i = lb; i < ub; i+=step)
/// <loop body>
/// --- pseudo LLVMIR ---
/// beforeloop:
/// guardcmp = (lb < ub)
/// if (guardcmp) goto preheader; else goto afterloop
/// preheader:
/// loop:
/// i1 = phi[{lb, preheader}, {i2, latch}]
/// <loop body>
/// i2 = i1 + step
/// latch:
/// cmp = (i2 < ub)
/// if (cmp) goto loop
/// exit:
/// afterloop:
///
/// getBounds
/// getInitialIVValue --> lb
/// getStepInst --> i2 = i1 + step
/// getStepValue --> step
/// getFinalIVValue --> ub
/// getCanonicalPredicate --> '<'
/// getDirection --> Increasing
/// getInductionVariable --> i1
/// getAuxiliaryInductionVariable --> {i1}
/// isCanonical --> false
Reviewers: kbarton, hfinkel, dmgreen, Meinersbur, jdoerfert, syzaara,
fhahn
Reviewed By: kbarton
Subscribers: tvvikram, bmahjour, etiotto, fhahn, jsji, hiraditya,
llvm-commits
Tag: LLVM
Differential Revision: https://reviews.llvm.org/D60565
llvm-svn: 362644
2019-06-06 04:42:47 +08:00
|
|
|
Function::iterator FI = F.begin();
|
[LOOPINFO] Introduce the loop guard API.
Summary:
This is the first patch for the loop guard. We introduced
getLoopGuardBranch() and isGuarded().
This currently only works on simplified loop, as it requires a preheader
and a latch to identify the guard.
It will work on loops of the form:
/// GuardBB:
/// br cond1, Preheader, ExitSucc <== GuardBranch
/// Preheader:
/// br Header
/// Header:
/// ...
/// br Latch
/// Latch:
/// br cond2, Header, ExitBlock
/// ExitBlock:
/// br ExitSucc
/// ExitSucc:
Prior discussions leading upto the decision to introduce the loop guard
API: http://lists.llvm.org/pipermail/llvm-dev/2019-May/132607.html
Reviewer: reames, kbarton, hfinkel, jdoerfert, Meinersbur, dmgreen
Reviewed By: reames
Subscribers: wuzish, hiraditya, jsji, llvm-commits, bmahjour, etiotto
Tag: LLVM
Differential Revision: https://reviews.llvm.org/D63885
llvm-svn: 367033
2019-07-26 00:13:18 +08:00
|
|
|
BasicBlock *Entry = &*(FI);
|
|
|
|
BranchInst *Guard = dyn_cast<BranchInst>(Entry->getTerminator());
|
[LOOPINFO] Extend Loop object to add utilities to get the loop bounds,
step, and loop induction variable.
Summary: This PR extends the loop object with more utilities to get loop
bounds, step, and loop induction variable. There already exists passes
which try to obtain the loop induction variable in their own pass, e.g.
loop interchange. It would be useful to have a common area to get these
information.
/// Example:
/// for (int i = lb; i < ub; i+=step)
/// <loop body>
/// --- pseudo LLVMIR ---
/// beforeloop:
/// guardcmp = (lb < ub)
/// if (guardcmp) goto preheader; else goto afterloop
/// preheader:
/// loop:
/// i1 = phi[{lb, preheader}, {i2, latch}]
/// <loop body>
/// i2 = i1 + step
/// latch:
/// cmp = (i2 < ub)
/// if (cmp) goto loop
/// exit:
/// afterloop:
///
/// getBounds
/// getInitialIVValue --> lb
/// getStepInst --> i2 = i1 + step
/// getStepValue --> step
/// getFinalIVValue --> ub
/// getCanonicalPredicate --> '<'
/// getDirection --> Increasing
/// getInductionVariable --> i1
/// getAuxiliaryInductionVariable --> {i1}
/// isCanonical --> false
Reviewers: kbarton, hfinkel, dmgreen, Meinersbur, jdoerfert, syzaara,
fhahn
Reviewed By: kbarton
Subscribers: tvvikram, bmahjour, etiotto, fhahn, jsji, hiraditya,
llvm-commits
Tag: LLVM
Differential Revision: https://reviews.llvm.org/D60565
llvm-svn: 362644
2019-06-06 04:42:47 +08:00
|
|
|
// First two basic block are entry and for.preheader - skip them.
|
|
|
|
++FI;
|
|
|
|
BasicBlock *Header = &*(++FI);
|
|
|
|
assert(Header->getName() == "for.body");
|
|
|
|
Loop *L = LI.getLoopFor(Header);
|
|
|
|
EXPECT_NE(L, nullptr);
|
|
|
|
|
|
|
|
Optional<Loop::LoopBounds> Bounds = L->getBounds(SE);
|
|
|
|
EXPECT_NE(Bounds, None);
|
|
|
|
EXPECT_EQ(Bounds->getInitialIVValue().getName(), "ub");
|
|
|
|
EXPECT_EQ(Bounds->getStepInst().getName(), "inc");
|
|
|
|
ConstantInt *StepValue =
|
|
|
|
dyn_cast_or_null<ConstantInt>(Bounds->getStepValue());
|
|
|
|
EXPECT_EQ(StepValue, nullptr);
|
|
|
|
ConstantInt *FinalIVValue =
|
|
|
|
dyn_cast<ConstantInt>(&Bounds->getFinalIVValue());
|
|
|
|
EXPECT_TRUE(FinalIVValue && FinalIVValue->isZero());
|
|
|
|
EXPECT_EQ(Bounds->getCanonicalPredicate(), ICmpInst::ICMP_SGT);
|
|
|
|
EXPECT_EQ(Bounds->getDirection(),
|
|
|
|
Loop::LoopBounds::Direction::Decreasing);
|
|
|
|
EXPECT_EQ(L->getInductionVariable(SE)->getName(), "i");
|
[LOOPINFO] Introduce the loop guard API.
Summary:
This is the first patch for the loop guard. We introduced
getLoopGuardBranch() and isGuarded().
This currently only works on simplified loop, as it requires a preheader
and a latch to identify the guard.
It will work on loops of the form:
/// GuardBB:
/// br cond1, Preheader, ExitSucc <== GuardBranch
/// Preheader:
/// br Header
/// Header:
/// ...
/// br Latch
/// Latch:
/// br cond2, Header, ExitBlock
/// ExitBlock:
/// br ExitSucc
/// ExitSucc:
Prior discussions leading upto the decision to introduce the loop guard
API: http://lists.llvm.org/pipermail/llvm-dev/2019-May/132607.html
Reviewer: reames, kbarton, hfinkel, jdoerfert, Meinersbur, dmgreen
Reviewed By: reames
Subscribers: wuzish, hiraditya, jsji, llvm-commits, bmahjour, etiotto
Tag: LLVM
Differential Revision: https://reviews.llvm.org/D63885
llvm-svn: 367033
2019-07-26 00:13:18 +08:00
|
|
|
EXPECT_EQ(L->getLoopGuardBranch(), Guard);
|
|
|
|
EXPECT_TRUE(L->isGuarded());
|
2019-12-13 03:18:39 +08:00
|
|
|
EXPECT_TRUE(L->isRotatedForm());
|
[LOOPINFO] Extend Loop object to add utilities to get the loop bounds,
step, and loop induction variable.
Summary: This PR extends the loop object with more utilities to get loop
bounds, step, and loop induction variable. There already exists passes
which try to obtain the loop induction variable in their own pass, e.g.
loop interchange. It would be useful to have a common area to get these
information.
/// Example:
/// for (int i = lb; i < ub; i+=step)
/// <loop body>
/// --- pseudo LLVMIR ---
/// beforeloop:
/// guardcmp = (lb < ub)
/// if (guardcmp) goto preheader; else goto afterloop
/// preheader:
/// loop:
/// i1 = phi[{lb, preheader}, {i2, latch}]
/// <loop body>
/// i2 = i1 + step
/// latch:
/// cmp = (i2 < ub)
/// if (cmp) goto loop
/// exit:
/// afterloop:
///
/// getBounds
/// getInitialIVValue --> lb
/// getStepInst --> i2 = i1 + step
/// getStepValue --> step
/// getFinalIVValue --> ub
/// getCanonicalPredicate --> '<'
/// getDirection --> Increasing
/// getInductionVariable --> i1
/// getAuxiliaryInductionVariable --> {i1}
/// isCanonical --> false
Reviewers: kbarton, hfinkel, dmgreen, Meinersbur, jdoerfert, syzaara,
fhahn
Reviewed By: kbarton
Subscribers: tvvikram, bmahjour, etiotto, fhahn, jsji, hiraditya,
llvm-commits
Tag: LLVM
Differential Revision: https://reviews.llvm.org/D60565
llvm-svn: 362644
2019-06-06 04:42:47 +08:00
|
|
|
});
|
|
|
|
}
|
|
|
|
|
|
|
|
TEST(LoopInfoTest, CannotFindDirection) {
|
|
|
|
const char *ModuleStr =
|
|
|
|
"define void @foo(i32* %A, i32 %ub, i32 %step) {\n"
|
|
|
|
"entry:\n"
|
|
|
|
" %guardcmp = icmp slt i32 0, %ub\n"
|
|
|
|
" br i1 %guardcmp, label %for.preheader, label %for.end\n"
|
|
|
|
"for.preheader:\n"
|
|
|
|
" br label %for.body\n"
|
|
|
|
"for.body:\n"
|
|
|
|
" %i = phi i32 [ 0, %for.preheader ], [ %inc, %for.body ]\n"
|
|
|
|
" %idxprom = sext i32 %i to i64\n"
|
|
|
|
" %arrayidx = getelementptr inbounds i32, i32* %A, i64 %idxprom\n"
|
|
|
|
" store i32 %i, i32* %arrayidx, align 4\n"
|
|
|
|
" %inc = add nsw i32 %i, %step\n"
|
|
|
|
" %cmp = icmp ne i32 %i, %ub\n"
|
|
|
|
" br i1 %cmp, label %for.body, label %for.exit\n"
|
|
|
|
"for.exit:\n"
|
|
|
|
" br label %for.end\n"
|
|
|
|
"for.end:\n"
|
|
|
|
" ret void\n"
|
|
|
|
"}\n";
|
|
|
|
|
|
|
|
// Parse the module.
|
|
|
|
LLVMContext Context;
|
|
|
|
std::unique_ptr<Module> M = makeLLVMModule(Context, ModuleStr);
|
|
|
|
|
[LOOPINFO] Introduce the loop guard API.
Summary:
This is the first patch for the loop guard. We introduced
getLoopGuardBranch() and isGuarded().
This currently only works on simplified loop, as it requires a preheader
and a latch to identify the guard.
It will work on loops of the form:
/// GuardBB:
/// br cond1, Preheader, ExitSucc <== GuardBranch
/// Preheader:
/// br Header
/// Header:
/// ...
/// br Latch
/// Latch:
/// br cond2, Header, ExitBlock
/// ExitBlock:
/// br ExitSucc
/// ExitSucc:
Prior discussions leading upto the decision to introduce the loop guard
API: http://lists.llvm.org/pipermail/llvm-dev/2019-May/132607.html
Reviewer: reames, kbarton, hfinkel, jdoerfert, Meinersbur, dmgreen
Reviewed By: reames
Subscribers: wuzish, hiraditya, jsji, llvm-commits, bmahjour, etiotto
Tag: LLVM
Differential Revision: https://reviews.llvm.org/D63885
llvm-svn: 367033
2019-07-26 00:13:18 +08:00
|
|
|
runWithLoopInfoPlus(
|
|
|
|
*M, "foo",
|
|
|
|
[&](Function &F, LoopInfo &LI, ScalarEvolution &SE) {
|
|
|
|
Function::iterator FI = F.begin();
|
|
|
|
BasicBlock *Entry = &*(FI);
|
|
|
|
BranchInst *Guard = dyn_cast<BranchInst>(Entry->getTerminator());
|
|
|
|
// First two basic block are entry and for.preheader
|
|
|
|
// - skip them.
|
|
|
|
++FI;
|
|
|
|
BasicBlock *Header = &*(++FI);
|
|
|
|
assert(Header->getName() == "for.body");
|
|
|
|
Loop *L = LI.getLoopFor(Header);
|
|
|
|
EXPECT_NE(L, nullptr);
|
|
|
|
|
|
|
|
Optional<Loop::LoopBounds> Bounds = L->getBounds(SE);
|
|
|
|
EXPECT_NE(Bounds, None);
|
|
|
|
ConstantInt *InitialIVValue =
|
|
|
|
dyn_cast<ConstantInt>(&Bounds->getInitialIVValue());
|
|
|
|
EXPECT_TRUE(InitialIVValue && InitialIVValue->isZero());
|
|
|
|
EXPECT_EQ(Bounds->getStepInst().getName(), "inc");
|
|
|
|
EXPECT_EQ(Bounds->getStepValue()->getName(), "step");
|
|
|
|
EXPECT_EQ(Bounds->getFinalIVValue().getName(), "ub");
|
|
|
|
EXPECT_EQ(Bounds->getCanonicalPredicate(),
|
|
|
|
ICmpInst::BAD_ICMP_PREDICATE);
|
|
|
|
EXPECT_EQ(Bounds->getDirection(), Loop::LoopBounds::Direction::Unknown);
|
|
|
|
EXPECT_EQ(L->getInductionVariable(SE)->getName(), "i");
|
|
|
|
EXPECT_EQ(L->getLoopGuardBranch(), Guard);
|
|
|
|
EXPECT_TRUE(L->isGuarded());
|
2019-12-13 03:18:39 +08:00
|
|
|
EXPECT_TRUE(L->isRotatedForm());
|
[LOOPINFO] Introduce the loop guard API.
Summary:
This is the first patch for the loop guard. We introduced
getLoopGuardBranch() and isGuarded().
This currently only works on simplified loop, as it requires a preheader
and a latch to identify the guard.
It will work on loops of the form:
/// GuardBB:
/// br cond1, Preheader, ExitSucc <== GuardBranch
/// Preheader:
/// br Header
/// Header:
/// ...
/// br Latch
/// Latch:
/// br cond2, Header, ExitBlock
/// ExitBlock:
/// br ExitSucc
/// ExitSucc:
Prior discussions leading upto the decision to introduce the loop guard
API: http://lists.llvm.org/pipermail/llvm-dev/2019-May/132607.html
Reviewer: reames, kbarton, hfinkel, jdoerfert, Meinersbur, dmgreen
Reviewed By: reames
Subscribers: wuzish, hiraditya, jsji, llvm-commits, bmahjour, etiotto
Tag: LLVM
Differential Revision: https://reviews.llvm.org/D63885
llvm-svn: 367033
2019-07-26 00:13:18 +08:00
|
|
|
});
|
[LOOPINFO] Extend Loop object to add utilities to get the loop bounds,
step, and loop induction variable.
Summary: This PR extends the loop object with more utilities to get loop
bounds, step, and loop induction variable. There already exists passes
which try to obtain the loop induction variable in their own pass, e.g.
loop interchange. It would be useful to have a common area to get these
information.
/// Example:
/// for (int i = lb; i < ub; i+=step)
/// <loop body>
/// --- pseudo LLVMIR ---
/// beforeloop:
/// guardcmp = (lb < ub)
/// if (guardcmp) goto preheader; else goto afterloop
/// preheader:
/// loop:
/// i1 = phi[{lb, preheader}, {i2, latch}]
/// <loop body>
/// i2 = i1 + step
/// latch:
/// cmp = (i2 < ub)
/// if (cmp) goto loop
/// exit:
/// afterloop:
///
/// getBounds
/// getInitialIVValue --> lb
/// getStepInst --> i2 = i1 + step
/// getStepValue --> step
/// getFinalIVValue --> ub
/// getCanonicalPredicate --> '<'
/// getDirection --> Increasing
/// getInductionVariable --> i1
/// getAuxiliaryInductionVariable --> {i1}
/// isCanonical --> false
Reviewers: kbarton, hfinkel, dmgreen, Meinersbur, jdoerfert, syzaara,
fhahn
Reviewed By: kbarton
Subscribers: tvvikram, bmahjour, etiotto, fhahn, jsji, hiraditya,
llvm-commits
Tag: LLVM
Differential Revision: https://reviews.llvm.org/D60565
llvm-svn: 362644
2019-06-06 04:42:47 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
TEST(LoopInfoTest, ZextIndVar) {
|
|
|
|
const char *ModuleStr =
|
|
|
|
"define void @foo(i32* %A, i32 %ub) {\n"
|
|
|
|
"entry:\n"
|
|
|
|
" %guardcmp = icmp slt i32 0, %ub\n"
|
|
|
|
" br i1 %guardcmp, label %for.preheader, label %for.end\n"
|
|
|
|
"for.preheader:\n"
|
|
|
|
" br label %for.body\n"
|
|
|
|
"for.body:\n"
|
|
|
|
" %indvars.iv = phi i64 [ 0, %for.preheader ], [ %indvars.iv.next, %for.body ]\n"
|
|
|
|
" %i = phi i32 [ 0, %for.preheader ], [ %inc, %for.body ]\n"
|
|
|
|
" %idxprom = sext i32 %i to i64\n"
|
|
|
|
" %arrayidx = getelementptr inbounds i32, i32* %A, i64 %idxprom\n"
|
|
|
|
" store i32 %i, i32* %arrayidx, align 4\n"
|
|
|
|
" %indvars.iv.next = add nuw nsw i64 %indvars.iv, 1\n"
|
|
|
|
" %inc = add nsw i32 %i, 1\n"
|
|
|
|
" %wide.trip.count = zext i32 %ub to i64\n"
|
|
|
|
" %exitcond = icmp ne i64 %indvars.iv.next, %wide.trip.count\n"
|
|
|
|
" br i1 %exitcond, label %for.body, label %for.exit\n"
|
|
|
|
"for.exit:\n"
|
|
|
|
" br label %for.end\n"
|
|
|
|
"for.end:\n"
|
|
|
|
" ret void\n"
|
|
|
|
"}\n";
|
|
|
|
|
|
|
|
// Parse the module.
|
|
|
|
LLVMContext Context;
|
|
|
|
std::unique_ptr<Module> M = makeLLVMModule(Context, ModuleStr);
|
|
|
|
|
|
|
|
runWithLoopInfoPlus(
|
|
|
|
*M, "foo",
|
[LOOPINFO] Introduce the loop guard API.
Summary:
This is the first patch for the loop guard. We introduced
getLoopGuardBranch() and isGuarded().
This currently only works on simplified loop, as it requires a preheader
and a latch to identify the guard.
It will work on loops of the form:
/// GuardBB:
/// br cond1, Preheader, ExitSucc <== GuardBranch
/// Preheader:
/// br Header
/// Header:
/// ...
/// br Latch
/// Latch:
/// br cond2, Header, ExitBlock
/// ExitBlock:
/// br ExitSucc
/// ExitSucc:
Prior discussions leading upto the decision to introduce the loop guard
API: http://lists.llvm.org/pipermail/llvm-dev/2019-May/132607.html
Reviewer: reames, kbarton, hfinkel, jdoerfert, Meinersbur, dmgreen
Reviewed By: reames
Subscribers: wuzish, hiraditya, jsji, llvm-commits, bmahjour, etiotto
Tag: LLVM
Differential Revision: https://reviews.llvm.org/D63885
llvm-svn: 367033
2019-07-26 00:13:18 +08:00
|
|
|
[&](Function &F, LoopInfo &LI, ScalarEvolution &SE) {
|
[LOOPINFO] Extend Loop object to add utilities to get the loop bounds,
step, and loop induction variable.
Summary: This PR extends the loop object with more utilities to get loop
bounds, step, and loop induction variable. There already exists passes
which try to obtain the loop induction variable in their own pass, e.g.
loop interchange. It would be useful to have a common area to get these
information.
/// Example:
/// for (int i = lb; i < ub; i+=step)
/// <loop body>
/// --- pseudo LLVMIR ---
/// beforeloop:
/// guardcmp = (lb < ub)
/// if (guardcmp) goto preheader; else goto afterloop
/// preheader:
/// loop:
/// i1 = phi[{lb, preheader}, {i2, latch}]
/// <loop body>
/// i2 = i1 + step
/// latch:
/// cmp = (i2 < ub)
/// if (cmp) goto loop
/// exit:
/// afterloop:
///
/// getBounds
/// getInitialIVValue --> lb
/// getStepInst --> i2 = i1 + step
/// getStepValue --> step
/// getFinalIVValue --> ub
/// getCanonicalPredicate --> '<'
/// getDirection --> Increasing
/// getInductionVariable --> i1
/// getAuxiliaryInductionVariable --> {i1}
/// isCanonical --> false
Reviewers: kbarton, hfinkel, dmgreen, Meinersbur, jdoerfert, syzaara,
fhahn
Reviewed By: kbarton
Subscribers: tvvikram, bmahjour, etiotto, fhahn, jsji, hiraditya,
llvm-commits
Tag: LLVM
Differential Revision: https://reviews.llvm.org/D60565
llvm-svn: 362644
2019-06-06 04:42:47 +08:00
|
|
|
Function::iterator FI = F.begin();
|
[LOOPINFO] Introduce the loop guard API.
Summary:
This is the first patch for the loop guard. We introduced
getLoopGuardBranch() and isGuarded().
This currently only works on simplified loop, as it requires a preheader
and a latch to identify the guard.
It will work on loops of the form:
/// GuardBB:
/// br cond1, Preheader, ExitSucc <== GuardBranch
/// Preheader:
/// br Header
/// Header:
/// ...
/// br Latch
/// Latch:
/// br cond2, Header, ExitBlock
/// ExitBlock:
/// br ExitSucc
/// ExitSucc:
Prior discussions leading upto the decision to introduce the loop guard
API: http://lists.llvm.org/pipermail/llvm-dev/2019-May/132607.html
Reviewer: reames, kbarton, hfinkel, jdoerfert, Meinersbur, dmgreen
Reviewed By: reames
Subscribers: wuzish, hiraditya, jsji, llvm-commits, bmahjour, etiotto
Tag: LLVM
Differential Revision: https://reviews.llvm.org/D63885
llvm-svn: 367033
2019-07-26 00:13:18 +08:00
|
|
|
BasicBlock *Entry = &*(FI);
|
|
|
|
BranchInst *Guard = dyn_cast<BranchInst>(Entry->getTerminator());
|
[LOOPINFO] Extend Loop object to add utilities to get the loop bounds,
step, and loop induction variable.
Summary: This PR extends the loop object with more utilities to get loop
bounds, step, and loop induction variable. There already exists passes
which try to obtain the loop induction variable in their own pass, e.g.
loop interchange. It would be useful to have a common area to get these
information.
/// Example:
/// for (int i = lb; i < ub; i+=step)
/// <loop body>
/// --- pseudo LLVMIR ---
/// beforeloop:
/// guardcmp = (lb < ub)
/// if (guardcmp) goto preheader; else goto afterloop
/// preheader:
/// loop:
/// i1 = phi[{lb, preheader}, {i2, latch}]
/// <loop body>
/// i2 = i1 + step
/// latch:
/// cmp = (i2 < ub)
/// if (cmp) goto loop
/// exit:
/// afterloop:
///
/// getBounds
/// getInitialIVValue --> lb
/// getStepInst --> i2 = i1 + step
/// getStepValue --> step
/// getFinalIVValue --> ub
/// getCanonicalPredicate --> '<'
/// getDirection --> Increasing
/// getInductionVariable --> i1
/// getAuxiliaryInductionVariable --> {i1}
/// isCanonical --> false
Reviewers: kbarton, hfinkel, dmgreen, Meinersbur, jdoerfert, syzaara,
fhahn
Reviewed By: kbarton
Subscribers: tvvikram, bmahjour, etiotto, fhahn, jsji, hiraditya,
llvm-commits
Tag: LLVM
Differential Revision: https://reviews.llvm.org/D60565
llvm-svn: 362644
2019-06-06 04:42:47 +08:00
|
|
|
// First two basic block are entry and for.preheader - skip them.
|
|
|
|
++FI;
|
|
|
|
BasicBlock *Header = &*(++FI);
|
|
|
|
assert(Header->getName() == "for.body");
|
|
|
|
Loop *L = LI.getLoopFor(Header);
|
|
|
|
EXPECT_NE(L, nullptr);
|
|
|
|
|
|
|
|
Optional<Loop::LoopBounds> Bounds = L->getBounds(SE);
|
|
|
|
EXPECT_NE(Bounds, None);
|
|
|
|
ConstantInt *InitialIVValue =
|
|
|
|
dyn_cast<ConstantInt>(&Bounds->getInitialIVValue());
|
|
|
|
EXPECT_TRUE(InitialIVValue && InitialIVValue->isZero());
|
|
|
|
EXPECT_EQ(Bounds->getStepInst().getName(), "indvars.iv.next");
|
|
|
|
ConstantInt *StepValue =
|
|
|
|
dyn_cast_or_null<ConstantInt>(Bounds->getStepValue());
|
|
|
|
EXPECT_TRUE(StepValue && StepValue->isOne());
|
|
|
|
EXPECT_EQ(Bounds->getFinalIVValue().getName(), "wide.trip.count");
|
|
|
|
EXPECT_EQ(Bounds->getCanonicalPredicate(), ICmpInst::ICMP_NE);
|
|
|
|
EXPECT_EQ(Bounds->getDirection(),
|
|
|
|
Loop::LoopBounds::Direction::Increasing);
|
|
|
|
EXPECT_EQ(L->getInductionVariable(SE)->getName(), "indvars.iv");
|
[LOOPINFO] Introduce the loop guard API.
Summary:
This is the first patch for the loop guard. We introduced
getLoopGuardBranch() and isGuarded().
This currently only works on simplified loop, as it requires a preheader
and a latch to identify the guard.
It will work on loops of the form:
/// GuardBB:
/// br cond1, Preheader, ExitSucc <== GuardBranch
/// Preheader:
/// br Header
/// Header:
/// ...
/// br Latch
/// Latch:
/// br cond2, Header, ExitBlock
/// ExitBlock:
/// br ExitSucc
/// ExitSucc:
Prior discussions leading upto the decision to introduce the loop guard
API: http://lists.llvm.org/pipermail/llvm-dev/2019-May/132607.html
Reviewer: reames, kbarton, hfinkel, jdoerfert, Meinersbur, dmgreen
Reviewed By: reames
Subscribers: wuzish, hiraditya, jsji, llvm-commits, bmahjour, etiotto
Tag: LLVM
Differential Revision: https://reviews.llvm.org/D63885
llvm-svn: 367033
2019-07-26 00:13:18 +08:00
|
|
|
EXPECT_EQ(L->getLoopGuardBranch(), Guard);
|
|
|
|
EXPECT_TRUE(L->isGuarded());
|
2019-12-13 03:18:39 +08:00
|
|
|
EXPECT_TRUE(L->isRotatedForm());
|
[LOOPINFO] Extend Loop object to add utilities to get the loop bounds,
step, and loop induction variable.
Summary: This PR extends the loop object with more utilities to get loop
bounds, step, and loop induction variable. There already exists passes
which try to obtain the loop induction variable in their own pass, e.g.
loop interchange. It would be useful to have a common area to get these
information.
/// Example:
/// for (int i = lb; i < ub; i+=step)
/// <loop body>
/// --- pseudo LLVMIR ---
/// beforeloop:
/// guardcmp = (lb < ub)
/// if (guardcmp) goto preheader; else goto afterloop
/// preheader:
/// loop:
/// i1 = phi[{lb, preheader}, {i2, latch}]
/// <loop body>
/// i2 = i1 + step
/// latch:
/// cmp = (i2 < ub)
/// if (cmp) goto loop
/// exit:
/// afterloop:
///
/// getBounds
/// getInitialIVValue --> lb
/// getStepInst --> i2 = i1 + step
/// getStepValue --> step
/// getFinalIVValue --> ub
/// getCanonicalPredicate --> '<'
/// getDirection --> Increasing
/// getInductionVariable --> i1
/// getAuxiliaryInductionVariable --> {i1}
/// isCanonical --> false
Reviewers: kbarton, hfinkel, dmgreen, Meinersbur, jdoerfert, syzaara,
fhahn
Reviewed By: kbarton
Subscribers: tvvikram, bmahjour, etiotto, fhahn, jsji, hiraditya,
llvm-commits
Tag: LLVM
Differential Revision: https://reviews.llvm.org/D60565
llvm-svn: 362644
2019-06-06 04:42:47 +08:00
|
|
|
});
|
|
|
|
}
|
|
|
|
|
[LOOPGUARD] Disable loop with multiple loop exiting blocks.
Summary: As discussed in the loop group meeting. With the current
definition of loop guard, we should not allow multiple loop exiting
blocks. For loops that has multiple loop exiting blocks, we can simply
unable to find the loop guard.
When getUniqueExitBlock() obtains a vector size not equals to one, that
means there is either no exit blocks or there exists more than one
unique block the loop exit to.
If we don't disallow loop with multiple loop exit blocks, then with our
current implementation, there can exist exit blocks don't post dominated
by the non pre-header successor of the guard block.
Reviewer: reames, Meinersbur, kbarton, etiotto, bmahjour
Reviewed By: Meinersbur, kbarton
Subscribers: fhahn, hiraditya, llvm-commits
Tag: LLVM
Differential Revision: https://reviews.llvm.org/D66529
llvm-svn: 373011
2019-09-27 04:20:42 +08:00
|
|
|
TEST(LoopInfoTest, MultiExitingLoop) {
|
|
|
|
const char *ModuleStr =
|
|
|
|
"define void @foo(i32* %A, i32 %ub, i1 %cond) {\n"
|
|
|
|
"entry:\n"
|
|
|
|
" %guardcmp = icmp slt i32 0, %ub\n"
|
|
|
|
" br i1 %guardcmp, label %for.preheader, label %for.end\n"
|
|
|
|
"for.preheader:\n"
|
|
|
|
" br label %for.body\n"
|
|
|
|
"for.body:\n"
|
|
|
|
" %i = phi i32 [ 0, %for.preheader ], [ %inc, %for.body.1 ]\n"
|
|
|
|
" br i1 %cond, label %for.body.1, label %for.exit\n"
|
|
|
|
"for.body.1:\n"
|
|
|
|
" %idxprom = sext i32 %i to i64\n"
|
|
|
|
" %arrayidx = getelementptr inbounds i32, i32* %A, i64 %idxprom\n"
|
|
|
|
" store i32 %i, i32* %arrayidx, align 4\n"
|
|
|
|
" %inc = add nsw i32 %i, 1\n"
|
|
|
|
" %cmp = icmp slt i32 %inc, %ub\n"
|
|
|
|
" br i1 %cmp, label %for.body, label %for.exit\n"
|
|
|
|
"for.exit:\n"
|
|
|
|
" br label %for.end\n"
|
|
|
|
"for.end:\n"
|
|
|
|
" ret void\n"
|
|
|
|
"}\n";
|
|
|
|
|
|
|
|
// Parse the module.
|
|
|
|
LLVMContext Context;
|
|
|
|
std::unique_ptr<Module> M = makeLLVMModule(Context, ModuleStr);
|
|
|
|
|
|
|
|
runWithLoopInfoPlus(
|
|
|
|
*M, "foo",
|
|
|
|
[&](Function &F, LoopInfo &LI, ScalarEvolution &SE) {
|
|
|
|
Function::iterator FI = F.begin();
|
|
|
|
BasicBlock *Entry = &*(FI);
|
|
|
|
BranchInst *Guard = dyn_cast<BranchInst>(Entry->getTerminator());
|
|
|
|
// First two basic block are entry and for.preheader - skip them.
|
|
|
|
++FI;
|
|
|
|
BasicBlock *Header = &*(++FI);
|
|
|
|
assert(Header->getName() == "for.body");
|
|
|
|
Loop *L = LI.getLoopFor(Header);
|
|
|
|
EXPECT_NE(L, nullptr);
|
|
|
|
|
|
|
|
Optional<Loop::LoopBounds> Bounds = L->getBounds(SE);
|
|
|
|
EXPECT_NE(Bounds, None);
|
|
|
|
ConstantInt *InitialIVValue =
|
|
|
|
dyn_cast<ConstantInt>(&Bounds->getInitialIVValue());
|
|
|
|
EXPECT_TRUE(InitialIVValue && InitialIVValue->isZero());
|
|
|
|
EXPECT_EQ(Bounds->getStepInst().getName(), "inc");
|
|
|
|
ConstantInt *StepValue =
|
|
|
|
dyn_cast_or_null<ConstantInt>(Bounds->getStepValue());
|
|
|
|
EXPECT_TRUE(StepValue && StepValue->isOne());
|
|
|
|
EXPECT_EQ(Bounds->getFinalIVValue().getName(), "ub");
|
|
|
|
EXPECT_EQ(Bounds->getCanonicalPredicate(), ICmpInst::ICMP_SLT);
|
|
|
|
EXPECT_EQ(Bounds->getDirection(),
|
|
|
|
Loop::LoopBounds::Direction::Increasing);
|
|
|
|
EXPECT_EQ(L->getInductionVariable(SE)->getName(), "i");
|
|
|
|
EXPECT_EQ(L->getLoopGuardBranch(), Guard);
|
|
|
|
EXPECT_TRUE(L->isGuarded());
|
|
|
|
});
|
|
|
|
}
|
|
|
|
|
|
|
|
TEST(LoopInfoTest, MultiExitLoop) {
|
|
|
|
const char *ModuleStr =
|
|
|
|
"define void @foo(i32* %A, i32 %ub, i1 %cond) {\n"
|
|
|
|
"entry:\n"
|
|
|
|
" %guardcmp = icmp slt i32 0, %ub\n"
|
|
|
|
" br i1 %guardcmp, label %for.preheader, label %for.end\n"
|
|
|
|
"for.preheader:\n"
|
|
|
|
" br label %for.body\n"
|
|
|
|
"for.body:\n"
|
|
|
|
" %i = phi i32 [ 0, %for.preheader ], [ %inc, %for.body.1 ]\n"
|
|
|
|
" br i1 %cond, label %for.body.1, label %for.exit\n"
|
|
|
|
"for.body.1:\n"
|
|
|
|
" %idxprom = sext i32 %i to i64\n"
|
|
|
|
" %arrayidx = getelementptr inbounds i32, i32* %A, i64 %idxprom\n"
|
|
|
|
" store i32 %i, i32* %arrayidx, align 4\n"
|
|
|
|
" %inc = add nsw i32 %i, 1\n"
|
|
|
|
" %cmp = icmp slt i32 %inc, %ub\n"
|
|
|
|
" br i1 %cmp, label %for.body, label %for.exit.1\n"
|
|
|
|
"for.exit:\n"
|
|
|
|
" br label %for.end\n"
|
|
|
|
"for.exit.1:\n"
|
|
|
|
" br label %for.end\n"
|
|
|
|
"for.end:\n"
|
|
|
|
" ret void\n"
|
|
|
|
"}\n";
|
|
|
|
|
|
|
|
// Parse the module.
|
|
|
|
LLVMContext Context;
|
|
|
|
std::unique_ptr<Module> M = makeLLVMModule(Context, ModuleStr);
|
|
|
|
|
|
|
|
runWithLoopInfoPlus(
|
|
|
|
*M, "foo",
|
|
|
|
[&](Function &F, LoopInfo &LI, ScalarEvolution &SE) {
|
|
|
|
Function::iterator FI = F.begin();
|
|
|
|
// First two basic block are entry and for.preheader - skip them.
|
|
|
|
++FI;
|
|
|
|
BasicBlock *Header = &*(++FI);
|
|
|
|
assert(Header->getName() == "for.body");
|
|
|
|
Loop *L = LI.getLoopFor(Header);
|
|
|
|
EXPECT_NE(L, nullptr);
|
|
|
|
|
|
|
|
Optional<Loop::LoopBounds> Bounds = L->getBounds(SE);
|
|
|
|
EXPECT_NE(Bounds, None);
|
|
|
|
ConstantInt *InitialIVValue =
|
|
|
|
dyn_cast<ConstantInt>(&Bounds->getInitialIVValue());
|
|
|
|
EXPECT_TRUE(InitialIVValue && InitialIVValue->isZero());
|
|
|
|
EXPECT_EQ(Bounds->getStepInst().getName(), "inc");
|
|
|
|
ConstantInt *StepValue =
|
|
|
|
dyn_cast_or_null<ConstantInt>(Bounds->getStepValue());
|
|
|
|
EXPECT_TRUE(StepValue && StepValue->isOne());
|
|
|
|
EXPECT_EQ(Bounds->getFinalIVValue().getName(), "ub");
|
|
|
|
EXPECT_EQ(Bounds->getCanonicalPredicate(), ICmpInst::ICMP_SLT);
|
|
|
|
EXPECT_EQ(Bounds->getDirection(),
|
|
|
|
Loop::LoopBounds::Direction::Increasing);
|
|
|
|
EXPECT_EQ(L->getInductionVariable(SE)->getName(), "i");
|
|
|
|
EXPECT_EQ(L->getLoopGuardBranch(), nullptr);
|
|
|
|
EXPECT_FALSE(L->isGuarded());
|
|
|
|
});
|
|
|
|
}
|
|
|
|
|
[LOOPINFO] Extend Loop object to add utilities to get the loop bounds,
step, and loop induction variable.
Summary: This PR extends the loop object with more utilities to get loop
bounds, step, and loop induction variable. There already exists passes
which try to obtain the loop induction variable in their own pass, e.g.
loop interchange. It would be useful to have a common area to get these
information.
/// Example:
/// for (int i = lb; i < ub; i+=step)
/// <loop body>
/// --- pseudo LLVMIR ---
/// beforeloop:
/// guardcmp = (lb < ub)
/// if (guardcmp) goto preheader; else goto afterloop
/// preheader:
/// loop:
/// i1 = phi[{lb, preheader}, {i2, latch}]
/// <loop body>
/// i2 = i1 + step
/// latch:
/// cmp = (i2 < ub)
/// if (cmp) goto loop
/// exit:
/// afterloop:
///
/// getBounds
/// getInitialIVValue --> lb
/// getStepInst --> i2 = i1 + step
/// getStepValue --> step
/// getFinalIVValue --> ub
/// getCanonicalPredicate --> '<'
/// getDirection --> Increasing
/// getInductionVariable --> i1
/// getAuxiliaryInductionVariable --> {i1}
/// isCanonical --> false
Reviewers: kbarton, hfinkel, dmgreen, Meinersbur, jdoerfert, syzaara,
fhahn
Reviewed By: kbarton
Subscribers: tvvikram, bmahjour, etiotto, fhahn, jsji, hiraditya,
llvm-commits
Tag: LLVM
Differential Revision: https://reviews.llvm.org/D60565
llvm-svn: 362644
2019-06-06 04:42:47 +08:00
|
|
|
TEST(LoopInfoTest, UnguardedLoop) {
|
|
|
|
const char *ModuleStr =
|
|
|
|
"define void @foo(i32* %A, i32 %ub) {\n"
|
|
|
|
"entry:\n"
|
|
|
|
" br label %for.body\n"
|
|
|
|
"for.body:\n"
|
|
|
|
" %i = phi i32 [ 0, %entry ], [ %inc, %for.body ]\n"
|
|
|
|
" %idxprom = sext i32 %i to i64\n"
|
|
|
|
" %arrayidx = getelementptr inbounds i32, i32* %A, i64 %idxprom\n"
|
|
|
|
" store i32 %i, i32* %arrayidx, align 4\n"
|
|
|
|
" %inc = add nsw i32 %i, 1\n"
|
|
|
|
" %cmp = icmp slt i32 %inc, %ub\n"
|
|
|
|
" br i1 %cmp, label %for.body, label %for.exit\n"
|
|
|
|
"for.exit:\n"
|
|
|
|
" br label %for.end\n"
|
|
|
|
"for.end:\n"
|
|
|
|
" ret void\n"
|
|
|
|
"}\n";
|
|
|
|
|
|
|
|
// Parse the module.
|
|
|
|
LLVMContext Context;
|
|
|
|
std::unique_ptr<Module> M = makeLLVMModule(Context, ModuleStr);
|
|
|
|
|
|
|
|
runWithLoopInfoPlus(
|
|
|
|
*M, "foo",
|
[LOOPINFO] Introduce the loop guard API.
Summary:
This is the first patch for the loop guard. We introduced
getLoopGuardBranch() and isGuarded().
This currently only works on simplified loop, as it requires a preheader
and a latch to identify the guard.
It will work on loops of the form:
/// GuardBB:
/// br cond1, Preheader, ExitSucc <== GuardBranch
/// Preheader:
/// br Header
/// Header:
/// ...
/// br Latch
/// Latch:
/// br cond2, Header, ExitBlock
/// ExitBlock:
/// br ExitSucc
/// ExitSucc:
Prior discussions leading upto the decision to introduce the loop guard
API: http://lists.llvm.org/pipermail/llvm-dev/2019-May/132607.html
Reviewer: reames, kbarton, hfinkel, jdoerfert, Meinersbur, dmgreen
Reviewed By: reames
Subscribers: wuzish, hiraditya, jsji, llvm-commits, bmahjour, etiotto
Tag: LLVM
Differential Revision: https://reviews.llvm.org/D63885
llvm-svn: 367033
2019-07-26 00:13:18 +08:00
|
|
|
[&](Function &F, LoopInfo &LI, ScalarEvolution &SE) {
|
[LOOPINFO] Extend Loop object to add utilities to get the loop bounds,
step, and loop induction variable.
Summary: This PR extends the loop object with more utilities to get loop
bounds, step, and loop induction variable. There already exists passes
which try to obtain the loop induction variable in their own pass, e.g.
loop interchange. It would be useful to have a common area to get these
information.
/// Example:
/// for (int i = lb; i < ub; i+=step)
/// <loop body>
/// --- pseudo LLVMIR ---
/// beforeloop:
/// guardcmp = (lb < ub)
/// if (guardcmp) goto preheader; else goto afterloop
/// preheader:
/// loop:
/// i1 = phi[{lb, preheader}, {i2, latch}]
/// <loop body>
/// i2 = i1 + step
/// latch:
/// cmp = (i2 < ub)
/// if (cmp) goto loop
/// exit:
/// afterloop:
///
/// getBounds
/// getInitialIVValue --> lb
/// getStepInst --> i2 = i1 + step
/// getStepValue --> step
/// getFinalIVValue --> ub
/// getCanonicalPredicate --> '<'
/// getDirection --> Increasing
/// getInductionVariable --> i1
/// getAuxiliaryInductionVariable --> {i1}
/// isCanonical --> false
Reviewers: kbarton, hfinkel, dmgreen, Meinersbur, jdoerfert, syzaara,
fhahn
Reviewed By: kbarton
Subscribers: tvvikram, bmahjour, etiotto, fhahn, jsji, hiraditya,
llvm-commits
Tag: LLVM
Differential Revision: https://reviews.llvm.org/D60565
llvm-svn: 362644
2019-06-06 04:42:47 +08:00
|
|
|
Function::iterator FI = F.begin();
|
|
|
|
// First basic block is entry - skip it.
|
|
|
|
BasicBlock *Header = &*(++FI);
|
|
|
|
assert(Header->getName() == "for.body");
|
|
|
|
Loop *L = LI.getLoopFor(Header);
|
|
|
|
EXPECT_NE(L, nullptr);
|
|
|
|
|
|
|
|
Optional<Loop::LoopBounds> Bounds = L->getBounds(SE);
|
|
|
|
EXPECT_NE(Bounds, None);
|
|
|
|
ConstantInt *InitialIVValue =
|
|
|
|
dyn_cast<ConstantInt>(&Bounds->getInitialIVValue());
|
|
|
|
EXPECT_TRUE(InitialIVValue && InitialIVValue->isZero());
|
|
|
|
EXPECT_EQ(Bounds->getStepInst().getName(), "inc");
|
|
|
|
ConstantInt *StepValue =
|
|
|
|
dyn_cast_or_null<ConstantInt>(Bounds->getStepValue());
|
|
|
|
EXPECT_TRUE(StepValue && StepValue->isOne());
|
|
|
|
EXPECT_EQ(Bounds->getFinalIVValue().getName(), "ub");
|
|
|
|
EXPECT_EQ(Bounds->getCanonicalPredicate(), ICmpInst::ICMP_SLT);
|
|
|
|
EXPECT_EQ(Bounds->getDirection(),
|
|
|
|
Loop::LoopBounds::Direction::Increasing);
|
|
|
|
EXPECT_EQ(L->getInductionVariable(SE)->getName(), "i");
|
[LOOPINFO] Introduce the loop guard API.
Summary:
This is the first patch for the loop guard. We introduced
getLoopGuardBranch() and isGuarded().
This currently only works on simplified loop, as it requires a preheader
and a latch to identify the guard.
It will work on loops of the form:
/// GuardBB:
/// br cond1, Preheader, ExitSucc <== GuardBranch
/// Preheader:
/// br Header
/// Header:
/// ...
/// br Latch
/// Latch:
/// br cond2, Header, ExitBlock
/// ExitBlock:
/// br ExitSucc
/// ExitSucc:
Prior discussions leading upto the decision to introduce the loop guard
API: http://lists.llvm.org/pipermail/llvm-dev/2019-May/132607.html
Reviewer: reames, kbarton, hfinkel, jdoerfert, Meinersbur, dmgreen
Reviewed By: reames
Subscribers: wuzish, hiraditya, jsji, llvm-commits, bmahjour, etiotto
Tag: LLVM
Differential Revision: https://reviews.llvm.org/D63885
llvm-svn: 367033
2019-07-26 00:13:18 +08:00
|
|
|
EXPECT_EQ(L->getLoopGuardBranch(), nullptr);
|
|
|
|
EXPECT_FALSE(L->isGuarded());
|
2019-12-13 03:18:39 +08:00
|
|
|
EXPECT_TRUE(L->isRotatedForm());
|
[LOOPINFO] Extend Loop object to add utilities to get the loop bounds,
step, and loop induction variable.
Summary: This PR extends the loop object with more utilities to get loop
bounds, step, and loop induction variable. There already exists passes
which try to obtain the loop induction variable in their own pass, e.g.
loop interchange. It would be useful to have a common area to get these
information.
/// Example:
/// for (int i = lb; i < ub; i+=step)
/// <loop body>
/// --- pseudo LLVMIR ---
/// beforeloop:
/// guardcmp = (lb < ub)
/// if (guardcmp) goto preheader; else goto afterloop
/// preheader:
/// loop:
/// i1 = phi[{lb, preheader}, {i2, latch}]
/// <loop body>
/// i2 = i1 + step
/// latch:
/// cmp = (i2 < ub)
/// if (cmp) goto loop
/// exit:
/// afterloop:
///
/// getBounds
/// getInitialIVValue --> lb
/// getStepInst --> i2 = i1 + step
/// getStepValue --> step
/// getFinalIVValue --> ub
/// getCanonicalPredicate --> '<'
/// getDirection --> Increasing
/// getInductionVariable --> i1
/// getAuxiliaryInductionVariable --> {i1}
/// isCanonical --> false
Reviewers: kbarton, hfinkel, dmgreen, Meinersbur, jdoerfert, syzaara,
fhahn
Reviewed By: kbarton
Subscribers: tvvikram, bmahjour, etiotto, fhahn, jsji, hiraditya,
llvm-commits
Tag: LLVM
Differential Revision: https://reviews.llvm.org/D60565
llvm-svn: 362644
2019-06-06 04:42:47 +08:00
|
|
|
});
|
|
|
|
}
|
|
|
|
|
|
|
|
TEST(LoopInfoTest, UnguardedLoopWithControlFlow) {
|
|
|
|
const char *ModuleStr =
|
|
|
|
"define void @foo(i32* %A, i32 %ub, i1 %cond) {\n"
|
|
|
|
"entry:\n"
|
|
|
|
" br i1 %cond, label %for.preheader, label %for.end\n"
|
|
|
|
"for.preheader:\n"
|
|
|
|
" br label %for.body\n"
|
|
|
|
"for.body:\n"
|
|
|
|
" %i = phi i32 [ 0, %for.preheader ], [ %inc, %for.body ]\n"
|
|
|
|
" %idxprom = sext i32 %i to i64\n"
|
|
|
|
" %arrayidx = getelementptr inbounds i32, i32* %A, i64 %idxprom\n"
|
|
|
|
" store i32 %i, i32* %arrayidx, align 4\n"
|
|
|
|
" %inc = add nsw i32 %i, 1\n"
|
|
|
|
" %cmp = icmp slt i32 %inc, %ub\n"
|
|
|
|
" br i1 %cmp, label %for.body, label %for.exit\n"
|
|
|
|
"for.exit:\n"
|
|
|
|
" br label %for.end\n"
|
|
|
|
"for.end:\n"
|
|
|
|
" ret void\n"
|
|
|
|
"}\n";
|
|
|
|
|
|
|
|
// Parse the module.
|
|
|
|
LLVMContext Context;
|
|
|
|
std::unique_ptr<Module> M = makeLLVMModule(Context, ModuleStr);
|
|
|
|
|
|
|
|
runWithLoopInfoPlus(
|
|
|
|
*M, "foo",
|
[LOOPINFO] Introduce the loop guard API.
Summary:
This is the first patch for the loop guard. We introduced
getLoopGuardBranch() and isGuarded().
This currently only works on simplified loop, as it requires a preheader
and a latch to identify the guard.
It will work on loops of the form:
/// GuardBB:
/// br cond1, Preheader, ExitSucc <== GuardBranch
/// Preheader:
/// br Header
/// Header:
/// ...
/// br Latch
/// Latch:
/// br cond2, Header, ExitBlock
/// ExitBlock:
/// br ExitSucc
/// ExitSucc:
Prior discussions leading upto the decision to introduce the loop guard
API: http://lists.llvm.org/pipermail/llvm-dev/2019-May/132607.html
Reviewer: reames, kbarton, hfinkel, jdoerfert, Meinersbur, dmgreen
Reviewed By: reames
Subscribers: wuzish, hiraditya, jsji, llvm-commits, bmahjour, etiotto
Tag: LLVM
Differential Revision: https://reviews.llvm.org/D63885
llvm-svn: 367033
2019-07-26 00:13:18 +08:00
|
|
|
[&](Function &F, LoopInfo &LI, ScalarEvolution &SE) {
|
[LOOPINFO] Extend Loop object to add utilities to get the loop bounds,
step, and loop induction variable.
Summary: This PR extends the loop object with more utilities to get loop
bounds, step, and loop induction variable. There already exists passes
which try to obtain the loop induction variable in their own pass, e.g.
loop interchange. It would be useful to have a common area to get these
information.
/// Example:
/// for (int i = lb; i < ub; i+=step)
/// <loop body>
/// --- pseudo LLVMIR ---
/// beforeloop:
/// guardcmp = (lb < ub)
/// if (guardcmp) goto preheader; else goto afterloop
/// preheader:
/// loop:
/// i1 = phi[{lb, preheader}, {i2, latch}]
/// <loop body>
/// i2 = i1 + step
/// latch:
/// cmp = (i2 < ub)
/// if (cmp) goto loop
/// exit:
/// afterloop:
///
/// getBounds
/// getInitialIVValue --> lb
/// getStepInst --> i2 = i1 + step
/// getStepValue --> step
/// getFinalIVValue --> ub
/// getCanonicalPredicate --> '<'
/// getDirection --> Increasing
/// getInductionVariable --> i1
/// getAuxiliaryInductionVariable --> {i1}
/// isCanonical --> false
Reviewers: kbarton, hfinkel, dmgreen, Meinersbur, jdoerfert, syzaara,
fhahn
Reviewed By: kbarton
Subscribers: tvvikram, bmahjour, etiotto, fhahn, jsji, hiraditya,
llvm-commits
Tag: LLVM
Differential Revision: https://reviews.llvm.org/D60565
llvm-svn: 362644
2019-06-06 04:42:47 +08:00
|
|
|
Function::iterator FI = F.begin();
|
[LOOPINFO] Introduce the loop guard API.
Summary:
This is the first patch for the loop guard. We introduced
getLoopGuardBranch() and isGuarded().
This currently only works on simplified loop, as it requires a preheader
and a latch to identify the guard.
It will work on loops of the form:
/// GuardBB:
/// br cond1, Preheader, ExitSucc <== GuardBranch
/// Preheader:
/// br Header
/// Header:
/// ...
/// br Latch
/// Latch:
/// br cond2, Header, ExitBlock
/// ExitBlock:
/// br ExitSucc
/// ExitSucc:
Prior discussions leading upto the decision to introduce the loop guard
API: http://lists.llvm.org/pipermail/llvm-dev/2019-May/132607.html
Reviewer: reames, kbarton, hfinkel, jdoerfert, Meinersbur, dmgreen
Reviewed By: reames
Subscribers: wuzish, hiraditya, jsji, llvm-commits, bmahjour, etiotto
Tag: LLVM
Differential Revision: https://reviews.llvm.org/D63885
llvm-svn: 367033
2019-07-26 00:13:18 +08:00
|
|
|
BasicBlock *Entry = &*(FI);
|
|
|
|
BranchInst *Guard = dyn_cast<BranchInst>(Entry->getTerminator());
|
[LOOPINFO] Extend Loop object to add utilities to get the loop bounds,
step, and loop induction variable.
Summary: This PR extends the loop object with more utilities to get loop
bounds, step, and loop induction variable. There already exists passes
which try to obtain the loop induction variable in their own pass, e.g.
loop interchange. It would be useful to have a common area to get these
information.
/// Example:
/// for (int i = lb; i < ub; i+=step)
/// <loop body>
/// --- pseudo LLVMIR ---
/// beforeloop:
/// guardcmp = (lb < ub)
/// if (guardcmp) goto preheader; else goto afterloop
/// preheader:
/// loop:
/// i1 = phi[{lb, preheader}, {i2, latch}]
/// <loop body>
/// i2 = i1 + step
/// latch:
/// cmp = (i2 < ub)
/// if (cmp) goto loop
/// exit:
/// afterloop:
///
/// getBounds
/// getInitialIVValue --> lb
/// getStepInst --> i2 = i1 + step
/// getStepValue --> step
/// getFinalIVValue --> ub
/// getCanonicalPredicate --> '<'
/// getDirection --> Increasing
/// getInductionVariable --> i1
/// getAuxiliaryInductionVariable --> {i1}
/// isCanonical --> false
Reviewers: kbarton, hfinkel, dmgreen, Meinersbur, jdoerfert, syzaara,
fhahn
Reviewed By: kbarton
Subscribers: tvvikram, bmahjour, etiotto, fhahn, jsji, hiraditya,
llvm-commits
Tag: LLVM
Differential Revision: https://reviews.llvm.org/D60565
llvm-svn: 362644
2019-06-06 04:42:47 +08:00
|
|
|
// First two basic block are entry and for.preheader - skip them.
|
|
|
|
++FI;
|
|
|
|
BasicBlock *Header = &*(++FI);
|
|
|
|
assert(Header->getName() == "for.body");
|
|
|
|
Loop *L = LI.getLoopFor(Header);
|
|
|
|
EXPECT_NE(L, nullptr);
|
|
|
|
|
|
|
|
Optional<Loop::LoopBounds> Bounds = L->getBounds(SE);
|
|
|
|
EXPECT_NE(Bounds, None);
|
|
|
|
ConstantInt *InitialIVValue =
|
|
|
|
dyn_cast<ConstantInt>(&Bounds->getInitialIVValue());
|
|
|
|
EXPECT_TRUE(InitialIVValue && InitialIVValue->isZero());
|
|
|
|
EXPECT_EQ(Bounds->getStepInst().getName(), "inc");
|
|
|
|
ConstantInt *StepValue =
|
|
|
|
dyn_cast_or_null<ConstantInt>(Bounds->getStepValue());
|
|
|
|
EXPECT_TRUE(StepValue && StepValue->isOne());
|
|
|
|
EXPECT_EQ(Bounds->getFinalIVValue().getName(), "ub");
|
|
|
|
EXPECT_EQ(Bounds->getCanonicalPredicate(), ICmpInst::ICMP_SLT);
|
|
|
|
EXPECT_EQ(Bounds->getDirection(),
|
|
|
|
Loop::LoopBounds::Direction::Increasing);
|
|
|
|
EXPECT_EQ(L->getInductionVariable(SE)->getName(), "i");
|
[LOOPINFO] Introduce the loop guard API.
Summary:
This is the first patch for the loop guard. We introduced
getLoopGuardBranch() and isGuarded().
This currently only works on simplified loop, as it requires a preheader
and a latch to identify the guard.
It will work on loops of the form:
/// GuardBB:
/// br cond1, Preheader, ExitSucc <== GuardBranch
/// Preheader:
/// br Header
/// Header:
/// ...
/// br Latch
/// Latch:
/// br cond2, Header, ExitBlock
/// ExitBlock:
/// br ExitSucc
/// ExitSucc:
Prior discussions leading upto the decision to introduce the loop guard
API: http://lists.llvm.org/pipermail/llvm-dev/2019-May/132607.html
Reviewer: reames, kbarton, hfinkel, jdoerfert, Meinersbur, dmgreen
Reviewed By: reames
Subscribers: wuzish, hiraditya, jsji, llvm-commits, bmahjour, etiotto
Tag: LLVM
Differential Revision: https://reviews.llvm.org/D63885
llvm-svn: 367033
2019-07-26 00:13:18 +08:00
|
|
|
EXPECT_EQ(L->getLoopGuardBranch(), Guard);
|
|
|
|
EXPECT_TRUE(L->isGuarded());
|
2019-12-13 03:18:39 +08:00
|
|
|
EXPECT_TRUE(L->isRotatedForm());
|
[LOOPINFO] Extend Loop object to add utilities to get the loop bounds,
step, and loop induction variable.
Summary: This PR extends the loop object with more utilities to get loop
bounds, step, and loop induction variable. There already exists passes
which try to obtain the loop induction variable in their own pass, e.g.
loop interchange. It would be useful to have a common area to get these
information.
/// Example:
/// for (int i = lb; i < ub; i+=step)
/// <loop body>
/// --- pseudo LLVMIR ---
/// beforeloop:
/// guardcmp = (lb < ub)
/// if (guardcmp) goto preheader; else goto afterloop
/// preheader:
/// loop:
/// i1 = phi[{lb, preheader}, {i2, latch}]
/// <loop body>
/// i2 = i1 + step
/// latch:
/// cmp = (i2 < ub)
/// if (cmp) goto loop
/// exit:
/// afterloop:
///
/// getBounds
/// getInitialIVValue --> lb
/// getStepInst --> i2 = i1 + step
/// getStepValue --> step
/// getFinalIVValue --> ub
/// getCanonicalPredicate --> '<'
/// getDirection --> Increasing
/// getInductionVariable --> i1
/// getAuxiliaryInductionVariable --> {i1}
/// isCanonical --> false
Reviewers: kbarton, hfinkel, dmgreen, Meinersbur, jdoerfert, syzaara,
fhahn
Reviewed By: kbarton
Subscribers: tvvikram, bmahjour, etiotto, fhahn, jsji, hiraditya,
llvm-commits
Tag: LLVM
Differential Revision: https://reviews.llvm.org/D60565
llvm-svn: 362644
2019-06-06 04:42:47 +08:00
|
|
|
});
|
|
|
|
}
|
|
|
|
|
|
|
|
TEST(LoopInfoTest, LoopNest) {
|
|
|
|
const char *ModuleStr =
|
|
|
|
"define void @foo(i32* %A, i32 %ub) {\n"
|
|
|
|
"entry:\n"
|
|
|
|
" %guardcmp = icmp slt i32 0, %ub\n"
|
|
|
|
" br i1 %guardcmp, label %for.outer.preheader, label %for.end\n"
|
|
|
|
"for.outer.preheader:\n"
|
|
|
|
" br label %for.outer\n"
|
|
|
|
"for.outer:\n"
|
|
|
|
" %j = phi i32 [ 0, %for.outer.preheader ], [ %inc.outer, %for.outer.latch ]\n"
|
|
|
|
" br i1 %guardcmp, label %for.inner.preheader, label %for.outer.latch\n"
|
|
|
|
"for.inner.preheader:\n"
|
|
|
|
" br label %for.inner\n"
|
|
|
|
"for.inner:\n"
|
|
|
|
" %i = phi i32 [ 0, %for.inner.preheader ], [ %inc, %for.inner ]\n"
|
|
|
|
" %idxprom = sext i32 %i to i64\n"
|
|
|
|
" %arrayidx = getelementptr inbounds i32, i32* %A, i64 %idxprom\n"
|
|
|
|
" store i32 %i, i32* %arrayidx, align 4\n"
|
|
|
|
" %inc = add nsw i32 %i, 1\n"
|
|
|
|
" %cmp = icmp slt i32 %inc, %ub\n"
|
|
|
|
" br i1 %cmp, label %for.inner, label %for.inner.exit\n"
|
|
|
|
"for.inner.exit:\n"
|
|
|
|
" br label %for.outer.latch\n"
|
|
|
|
"for.outer.latch:\n"
|
|
|
|
" %inc.outer = add nsw i32 %j, 1\n"
|
|
|
|
" %cmp.outer = icmp slt i32 %inc.outer, %ub\n"
|
|
|
|
" br i1 %cmp.outer, label %for.outer, label %for.outer.exit\n"
|
|
|
|
"for.outer.exit:\n"
|
|
|
|
" br label %for.end\n"
|
|
|
|
"for.end:\n"
|
|
|
|
" ret void\n"
|
|
|
|
"}\n";
|
|
|
|
|
|
|
|
// Parse the module.
|
|
|
|
LLVMContext Context;
|
|
|
|
std::unique_ptr<Module> M = makeLLVMModule(Context, ModuleStr);
|
|
|
|
|
|
|
|
runWithLoopInfoPlus(
|
|
|
|
*M, "foo",
|
[LOOPINFO] Introduce the loop guard API.
Summary:
This is the first patch for the loop guard. We introduced
getLoopGuardBranch() and isGuarded().
This currently only works on simplified loop, as it requires a preheader
and a latch to identify the guard.
It will work on loops of the form:
/// GuardBB:
/// br cond1, Preheader, ExitSucc <== GuardBranch
/// Preheader:
/// br Header
/// Header:
/// ...
/// br Latch
/// Latch:
/// br cond2, Header, ExitBlock
/// ExitBlock:
/// br ExitSucc
/// ExitSucc:
Prior discussions leading upto the decision to introduce the loop guard
API: http://lists.llvm.org/pipermail/llvm-dev/2019-May/132607.html
Reviewer: reames, kbarton, hfinkel, jdoerfert, Meinersbur, dmgreen
Reviewed By: reames
Subscribers: wuzish, hiraditya, jsji, llvm-commits, bmahjour, etiotto
Tag: LLVM
Differential Revision: https://reviews.llvm.org/D63885
llvm-svn: 367033
2019-07-26 00:13:18 +08:00
|
|
|
[&](Function &F, LoopInfo &LI, ScalarEvolution &SE) {
|
[LOOPINFO] Extend Loop object to add utilities to get the loop bounds,
step, and loop induction variable.
Summary: This PR extends the loop object with more utilities to get loop
bounds, step, and loop induction variable. There already exists passes
which try to obtain the loop induction variable in their own pass, e.g.
loop interchange. It would be useful to have a common area to get these
information.
/// Example:
/// for (int i = lb; i < ub; i+=step)
/// <loop body>
/// --- pseudo LLVMIR ---
/// beforeloop:
/// guardcmp = (lb < ub)
/// if (guardcmp) goto preheader; else goto afterloop
/// preheader:
/// loop:
/// i1 = phi[{lb, preheader}, {i2, latch}]
/// <loop body>
/// i2 = i1 + step
/// latch:
/// cmp = (i2 < ub)
/// if (cmp) goto loop
/// exit:
/// afterloop:
///
/// getBounds
/// getInitialIVValue --> lb
/// getStepInst --> i2 = i1 + step
/// getStepValue --> step
/// getFinalIVValue --> ub
/// getCanonicalPredicate --> '<'
/// getDirection --> Increasing
/// getInductionVariable --> i1
/// getAuxiliaryInductionVariable --> {i1}
/// isCanonical --> false
Reviewers: kbarton, hfinkel, dmgreen, Meinersbur, jdoerfert, syzaara,
fhahn
Reviewed By: kbarton
Subscribers: tvvikram, bmahjour, etiotto, fhahn, jsji, hiraditya,
llvm-commits
Tag: LLVM
Differential Revision: https://reviews.llvm.org/D60565
llvm-svn: 362644
2019-06-06 04:42:47 +08:00
|
|
|
Function::iterator FI = F.begin();
|
[LOOPINFO] Introduce the loop guard API.
Summary:
This is the first patch for the loop guard. We introduced
getLoopGuardBranch() and isGuarded().
This currently only works on simplified loop, as it requires a preheader
and a latch to identify the guard.
It will work on loops of the form:
/// GuardBB:
/// br cond1, Preheader, ExitSucc <== GuardBranch
/// Preheader:
/// br Header
/// Header:
/// ...
/// br Latch
/// Latch:
/// br cond2, Header, ExitBlock
/// ExitBlock:
/// br ExitSucc
/// ExitSucc:
Prior discussions leading upto the decision to introduce the loop guard
API: http://lists.llvm.org/pipermail/llvm-dev/2019-May/132607.html
Reviewer: reames, kbarton, hfinkel, jdoerfert, Meinersbur, dmgreen
Reviewed By: reames
Subscribers: wuzish, hiraditya, jsji, llvm-commits, bmahjour, etiotto
Tag: LLVM
Differential Revision: https://reviews.llvm.org/D63885
llvm-svn: 367033
2019-07-26 00:13:18 +08:00
|
|
|
BasicBlock *Entry = &*(FI);
|
|
|
|
BranchInst *OuterGuard = dyn_cast<BranchInst>(Entry->getTerminator());
|
[LOOPINFO] Extend Loop object to add utilities to get the loop bounds,
step, and loop induction variable.
Summary: This PR extends the loop object with more utilities to get loop
bounds, step, and loop induction variable. There already exists passes
which try to obtain the loop induction variable in their own pass, e.g.
loop interchange. It would be useful to have a common area to get these
information.
/// Example:
/// for (int i = lb; i < ub; i+=step)
/// <loop body>
/// --- pseudo LLVMIR ---
/// beforeloop:
/// guardcmp = (lb < ub)
/// if (guardcmp) goto preheader; else goto afterloop
/// preheader:
/// loop:
/// i1 = phi[{lb, preheader}, {i2, latch}]
/// <loop body>
/// i2 = i1 + step
/// latch:
/// cmp = (i2 < ub)
/// if (cmp) goto loop
/// exit:
/// afterloop:
///
/// getBounds
/// getInitialIVValue --> lb
/// getStepInst --> i2 = i1 + step
/// getStepValue --> step
/// getFinalIVValue --> ub
/// getCanonicalPredicate --> '<'
/// getDirection --> Increasing
/// getInductionVariable --> i1
/// getAuxiliaryInductionVariable --> {i1}
/// isCanonical --> false
Reviewers: kbarton, hfinkel, dmgreen, Meinersbur, jdoerfert, syzaara,
fhahn
Reviewed By: kbarton
Subscribers: tvvikram, bmahjour, etiotto, fhahn, jsji, hiraditya,
llvm-commits
Tag: LLVM
Differential Revision: https://reviews.llvm.org/D60565
llvm-svn: 362644
2019-06-06 04:42:47 +08:00
|
|
|
// First two basic block are entry and for.outer.preheader - skip them.
|
|
|
|
++FI;
|
|
|
|
BasicBlock *Header = &*(++FI);
|
|
|
|
assert(Header->getName() == "for.outer");
|
[LOOPINFO] Introduce the loop guard API.
Summary:
This is the first patch for the loop guard. We introduced
getLoopGuardBranch() and isGuarded().
This currently only works on simplified loop, as it requires a preheader
and a latch to identify the guard.
It will work on loops of the form:
/// GuardBB:
/// br cond1, Preheader, ExitSucc <== GuardBranch
/// Preheader:
/// br Header
/// Header:
/// ...
/// br Latch
/// Latch:
/// br cond2, Header, ExitBlock
/// ExitBlock:
/// br ExitSucc
/// ExitSucc:
Prior discussions leading upto the decision to introduce the loop guard
API: http://lists.llvm.org/pipermail/llvm-dev/2019-May/132607.html
Reviewer: reames, kbarton, hfinkel, jdoerfert, Meinersbur, dmgreen
Reviewed By: reames
Subscribers: wuzish, hiraditya, jsji, llvm-commits, bmahjour, etiotto
Tag: LLVM
Differential Revision: https://reviews.llvm.org/D63885
llvm-svn: 367033
2019-07-26 00:13:18 +08:00
|
|
|
BranchInst *InnerGuard = dyn_cast<BranchInst>(Header->getTerminator());
|
[LOOPINFO] Extend Loop object to add utilities to get the loop bounds,
step, and loop induction variable.
Summary: This PR extends the loop object with more utilities to get loop
bounds, step, and loop induction variable. There already exists passes
which try to obtain the loop induction variable in their own pass, e.g.
loop interchange. It would be useful to have a common area to get these
information.
/// Example:
/// for (int i = lb; i < ub; i+=step)
/// <loop body>
/// --- pseudo LLVMIR ---
/// beforeloop:
/// guardcmp = (lb < ub)
/// if (guardcmp) goto preheader; else goto afterloop
/// preheader:
/// loop:
/// i1 = phi[{lb, preheader}, {i2, latch}]
/// <loop body>
/// i2 = i1 + step
/// latch:
/// cmp = (i2 < ub)
/// if (cmp) goto loop
/// exit:
/// afterloop:
///
/// getBounds
/// getInitialIVValue --> lb
/// getStepInst --> i2 = i1 + step
/// getStepValue --> step
/// getFinalIVValue --> ub
/// getCanonicalPredicate --> '<'
/// getDirection --> Increasing
/// getInductionVariable --> i1
/// getAuxiliaryInductionVariable --> {i1}
/// isCanonical --> false
Reviewers: kbarton, hfinkel, dmgreen, Meinersbur, jdoerfert, syzaara,
fhahn
Reviewed By: kbarton
Subscribers: tvvikram, bmahjour, etiotto, fhahn, jsji, hiraditya,
llvm-commits
Tag: LLVM
Differential Revision: https://reviews.llvm.org/D60565
llvm-svn: 362644
2019-06-06 04:42:47 +08:00
|
|
|
Loop *L = LI.getLoopFor(Header);
|
|
|
|
EXPECT_NE(L, nullptr);
|
|
|
|
|
|
|
|
Optional<Loop::LoopBounds> Bounds = L->getBounds(SE);
|
|
|
|
EXPECT_NE(Bounds, None);
|
|
|
|
ConstantInt *InitialIVValue =
|
|
|
|
dyn_cast<ConstantInt>(&Bounds->getInitialIVValue());
|
|
|
|
EXPECT_TRUE(InitialIVValue && InitialIVValue->isZero());
|
|
|
|
EXPECT_EQ(Bounds->getStepInst().getName(), "inc.outer");
|
|
|
|
ConstantInt *StepValue =
|
|
|
|
dyn_cast_or_null<ConstantInt>(Bounds->getStepValue());
|
|
|
|
EXPECT_TRUE(StepValue && StepValue->isOne());
|
|
|
|
EXPECT_EQ(Bounds->getFinalIVValue().getName(), "ub");
|
|
|
|
EXPECT_EQ(Bounds->getCanonicalPredicate(), ICmpInst::ICMP_SLT);
|
|
|
|
EXPECT_EQ(Bounds->getDirection(),
|
|
|
|
Loop::LoopBounds::Direction::Increasing);
|
|
|
|
EXPECT_EQ(L->getInductionVariable(SE)->getName(), "j");
|
[LOOPINFO] Introduce the loop guard API.
Summary:
This is the first patch for the loop guard. We introduced
getLoopGuardBranch() and isGuarded().
This currently only works on simplified loop, as it requires a preheader
and a latch to identify the guard.
It will work on loops of the form:
/// GuardBB:
/// br cond1, Preheader, ExitSucc <== GuardBranch
/// Preheader:
/// br Header
/// Header:
/// ...
/// br Latch
/// Latch:
/// br cond2, Header, ExitBlock
/// ExitBlock:
/// br ExitSucc
/// ExitSucc:
Prior discussions leading upto the decision to introduce the loop guard
API: http://lists.llvm.org/pipermail/llvm-dev/2019-May/132607.html
Reviewer: reames, kbarton, hfinkel, jdoerfert, Meinersbur, dmgreen
Reviewed By: reames
Subscribers: wuzish, hiraditya, jsji, llvm-commits, bmahjour, etiotto
Tag: LLVM
Differential Revision: https://reviews.llvm.org/D63885
llvm-svn: 367033
2019-07-26 00:13:18 +08:00
|
|
|
EXPECT_EQ(L->getLoopGuardBranch(), OuterGuard);
|
|
|
|
EXPECT_TRUE(L->isGuarded());
|
2019-12-13 03:18:39 +08:00
|
|
|
EXPECT_TRUE(L->isRotatedForm());
|
[LOOPINFO] Extend Loop object to add utilities to get the loop bounds,
step, and loop induction variable.
Summary: This PR extends the loop object with more utilities to get loop
bounds, step, and loop induction variable. There already exists passes
which try to obtain the loop induction variable in their own pass, e.g.
loop interchange. It would be useful to have a common area to get these
information.
/// Example:
/// for (int i = lb; i < ub; i+=step)
/// <loop body>
/// --- pseudo LLVMIR ---
/// beforeloop:
/// guardcmp = (lb < ub)
/// if (guardcmp) goto preheader; else goto afterloop
/// preheader:
/// loop:
/// i1 = phi[{lb, preheader}, {i2, latch}]
/// <loop body>
/// i2 = i1 + step
/// latch:
/// cmp = (i2 < ub)
/// if (cmp) goto loop
/// exit:
/// afterloop:
///
/// getBounds
/// getInitialIVValue --> lb
/// getStepInst --> i2 = i1 + step
/// getStepValue --> step
/// getFinalIVValue --> ub
/// getCanonicalPredicate --> '<'
/// getDirection --> Increasing
/// getInductionVariable --> i1
/// getAuxiliaryInductionVariable --> {i1}
/// isCanonical --> false
Reviewers: kbarton, hfinkel, dmgreen, Meinersbur, jdoerfert, syzaara,
fhahn
Reviewed By: kbarton
Subscribers: tvvikram, bmahjour, etiotto, fhahn, jsji, hiraditya,
llvm-commits
Tag: LLVM
Differential Revision: https://reviews.llvm.org/D60565
llvm-svn: 362644
2019-06-06 04:42:47 +08:00
|
|
|
|
|
|
|
// Next two basic blocks are for.outer and for.inner.preheader - skip
|
|
|
|
// them.
|
|
|
|
++FI;
|
|
|
|
Header = &*(++FI);
|
|
|
|
assert(Header->getName() == "for.inner");
|
|
|
|
L = LI.getLoopFor(Header);
|
|
|
|
EXPECT_NE(L, nullptr);
|
|
|
|
|
|
|
|
Optional<Loop::LoopBounds> InnerBounds = L->getBounds(SE);
|
|
|
|
EXPECT_NE(InnerBounds, None);
|
|
|
|
InitialIVValue =
|
|
|
|
dyn_cast<ConstantInt>(&InnerBounds->getInitialIVValue());
|
|
|
|
EXPECT_TRUE(InitialIVValue && InitialIVValue->isZero());
|
|
|
|
EXPECT_EQ(InnerBounds->getStepInst().getName(), "inc");
|
|
|
|
StepValue = dyn_cast_or_null<ConstantInt>(InnerBounds->getStepValue());
|
|
|
|
EXPECT_TRUE(StepValue && StepValue->isOne());
|
|
|
|
EXPECT_EQ(InnerBounds->getFinalIVValue().getName(), "ub");
|
|
|
|
EXPECT_EQ(InnerBounds->getCanonicalPredicate(), ICmpInst::ICMP_SLT);
|
|
|
|
EXPECT_EQ(InnerBounds->getDirection(),
|
|
|
|
Loop::LoopBounds::Direction::Increasing);
|
|
|
|
EXPECT_EQ(L->getInductionVariable(SE)->getName(), "i");
|
[LOOPINFO] Introduce the loop guard API.
Summary:
This is the first patch for the loop guard. We introduced
getLoopGuardBranch() and isGuarded().
This currently only works on simplified loop, as it requires a preheader
and a latch to identify the guard.
It will work on loops of the form:
/// GuardBB:
/// br cond1, Preheader, ExitSucc <== GuardBranch
/// Preheader:
/// br Header
/// Header:
/// ...
/// br Latch
/// Latch:
/// br cond2, Header, ExitBlock
/// ExitBlock:
/// br ExitSucc
/// ExitSucc:
Prior discussions leading upto the decision to introduce the loop guard
API: http://lists.llvm.org/pipermail/llvm-dev/2019-May/132607.html
Reviewer: reames, kbarton, hfinkel, jdoerfert, Meinersbur, dmgreen
Reviewed By: reames
Subscribers: wuzish, hiraditya, jsji, llvm-commits, bmahjour, etiotto
Tag: LLVM
Differential Revision: https://reviews.llvm.org/D63885
llvm-svn: 367033
2019-07-26 00:13:18 +08:00
|
|
|
EXPECT_EQ(L->getLoopGuardBranch(), InnerGuard);
|
|
|
|
EXPECT_TRUE(L->isGuarded());
|
2019-12-13 03:18:39 +08:00
|
|
|
EXPECT_TRUE(L->isRotatedForm());
|
[LOOPINFO] Extend Loop object to add utilities to get the loop bounds,
step, and loop induction variable.
Summary: This PR extends the loop object with more utilities to get loop
bounds, step, and loop induction variable. There already exists passes
which try to obtain the loop induction variable in their own pass, e.g.
loop interchange. It would be useful to have a common area to get these
information.
/// Example:
/// for (int i = lb; i < ub; i+=step)
/// <loop body>
/// --- pseudo LLVMIR ---
/// beforeloop:
/// guardcmp = (lb < ub)
/// if (guardcmp) goto preheader; else goto afterloop
/// preheader:
/// loop:
/// i1 = phi[{lb, preheader}, {i2, latch}]
/// <loop body>
/// i2 = i1 + step
/// latch:
/// cmp = (i2 < ub)
/// if (cmp) goto loop
/// exit:
/// afterloop:
///
/// getBounds
/// getInitialIVValue --> lb
/// getStepInst --> i2 = i1 + step
/// getStepValue --> step
/// getFinalIVValue --> ub
/// getCanonicalPredicate --> '<'
/// getDirection --> Increasing
/// getInductionVariable --> i1
/// getAuxiliaryInductionVariable --> {i1}
/// isCanonical --> false
Reviewers: kbarton, hfinkel, dmgreen, Meinersbur, jdoerfert, syzaara,
fhahn
Reviewed By: kbarton
Subscribers: tvvikram, bmahjour, etiotto, fhahn, jsji, hiraditya,
llvm-commits
Tag: LLVM
Differential Revision: https://reviews.llvm.org/D60565
llvm-svn: 362644
2019-06-06 04:42:47 +08:00
|
|
|
});
|
|
|
|
}
|
|
|
|
|
|
|
|
TEST(LoopInfoTest, AuxiliaryIV) {
|
|
|
|
const char *ModuleStr =
|
|
|
|
"define void @foo(i32* %A, i32 %ub) {\n"
|
|
|
|
"entry:\n"
|
|
|
|
" %guardcmp = icmp slt i32 0, %ub\n"
|
|
|
|
" br i1 %guardcmp, label %for.preheader, label %for.end\n"
|
|
|
|
"for.preheader:\n"
|
|
|
|
" br label %for.body\n"
|
|
|
|
"for.body:\n"
|
|
|
|
" %i = phi i32 [ 0, %for.preheader ], [ %inc, %for.body ]\n"
|
|
|
|
" %aux = phi i32 [ 0, %for.preheader ], [ %auxinc, %for.body ]\n"
|
|
|
|
" %loopvariant = phi i32 [ 0, %for.preheader ], [ %loopvariantinc, %for.body ]\n"
|
|
|
|
" %usedoutside = phi i32 [ 0, %for.preheader ], [ %usedoutsideinc, %for.body ]\n"
|
|
|
|
" %mulopcode = phi i32 [ 0, %for.preheader ], [ %mulopcodeinc, %for.body ]\n"
|
|
|
|
" %idxprom = sext i32 %i to i64\n"
|
|
|
|
" %arrayidx = getelementptr inbounds i32, i32* %A, i64 %idxprom\n"
|
|
|
|
" store i32 %i, i32* %arrayidx, align 4\n"
|
|
|
|
" %mulopcodeinc = mul nsw i32 %mulopcode, 5\n"
|
|
|
|
" %usedoutsideinc = add nsw i32 %usedoutside, 5\n"
|
|
|
|
" %loopvariantinc = add nsw i32 %loopvariant, %i\n"
|
|
|
|
" %auxinc = add nsw i32 %aux, 5\n"
|
|
|
|
" %inc = add nsw i32 %i, 1\n"
|
|
|
|
" %cmp = icmp slt i32 %inc, %ub\n"
|
|
|
|
" br i1 %cmp, label %for.body, label %for.exit\n"
|
|
|
|
"for.exit:\n"
|
|
|
|
" %lcssa = phi i32 [ %usedoutside, %for.body ]\n"
|
|
|
|
" br label %for.end\n"
|
|
|
|
"for.end:\n"
|
|
|
|
" ret void\n"
|
|
|
|
"}\n";
|
|
|
|
|
|
|
|
// Parse the module.
|
|
|
|
LLVMContext Context;
|
|
|
|
std::unique_ptr<Module> M = makeLLVMModule(Context, ModuleStr);
|
|
|
|
|
|
|
|
runWithLoopInfoPlus(
|
|
|
|
*M, "foo",
|
[LOOPINFO] Introduce the loop guard API.
Summary:
This is the first patch for the loop guard. We introduced
getLoopGuardBranch() and isGuarded().
This currently only works on simplified loop, as it requires a preheader
and a latch to identify the guard.
It will work on loops of the form:
/// GuardBB:
/// br cond1, Preheader, ExitSucc <== GuardBranch
/// Preheader:
/// br Header
/// Header:
/// ...
/// br Latch
/// Latch:
/// br cond2, Header, ExitBlock
/// ExitBlock:
/// br ExitSucc
/// ExitSucc:
Prior discussions leading upto the decision to introduce the loop guard
API: http://lists.llvm.org/pipermail/llvm-dev/2019-May/132607.html
Reviewer: reames, kbarton, hfinkel, jdoerfert, Meinersbur, dmgreen
Reviewed By: reames
Subscribers: wuzish, hiraditya, jsji, llvm-commits, bmahjour, etiotto
Tag: LLVM
Differential Revision: https://reviews.llvm.org/D63885
llvm-svn: 367033
2019-07-26 00:13:18 +08:00
|
|
|
[&](Function &F, LoopInfo &LI, ScalarEvolution &SE) {
|
[LOOPINFO] Extend Loop object to add utilities to get the loop bounds,
step, and loop induction variable.
Summary: This PR extends the loop object with more utilities to get loop
bounds, step, and loop induction variable. There already exists passes
which try to obtain the loop induction variable in their own pass, e.g.
loop interchange. It would be useful to have a common area to get these
information.
/// Example:
/// for (int i = lb; i < ub; i+=step)
/// <loop body>
/// --- pseudo LLVMIR ---
/// beforeloop:
/// guardcmp = (lb < ub)
/// if (guardcmp) goto preheader; else goto afterloop
/// preheader:
/// loop:
/// i1 = phi[{lb, preheader}, {i2, latch}]
/// <loop body>
/// i2 = i1 + step
/// latch:
/// cmp = (i2 < ub)
/// if (cmp) goto loop
/// exit:
/// afterloop:
///
/// getBounds
/// getInitialIVValue --> lb
/// getStepInst --> i2 = i1 + step
/// getStepValue --> step
/// getFinalIVValue --> ub
/// getCanonicalPredicate --> '<'
/// getDirection --> Increasing
/// getInductionVariable --> i1
/// getAuxiliaryInductionVariable --> {i1}
/// isCanonical --> false
Reviewers: kbarton, hfinkel, dmgreen, Meinersbur, jdoerfert, syzaara,
fhahn
Reviewed By: kbarton
Subscribers: tvvikram, bmahjour, etiotto, fhahn, jsji, hiraditya,
llvm-commits
Tag: LLVM
Differential Revision: https://reviews.llvm.org/D60565
llvm-svn: 362644
2019-06-06 04:42:47 +08:00
|
|
|
Function::iterator FI = F.begin();
|
[LOOPINFO] Introduce the loop guard API.
Summary:
This is the first patch for the loop guard. We introduced
getLoopGuardBranch() and isGuarded().
This currently only works on simplified loop, as it requires a preheader
and a latch to identify the guard.
It will work on loops of the form:
/// GuardBB:
/// br cond1, Preheader, ExitSucc <== GuardBranch
/// Preheader:
/// br Header
/// Header:
/// ...
/// br Latch
/// Latch:
/// br cond2, Header, ExitBlock
/// ExitBlock:
/// br ExitSucc
/// ExitSucc:
Prior discussions leading upto the decision to introduce the loop guard
API: http://lists.llvm.org/pipermail/llvm-dev/2019-May/132607.html
Reviewer: reames, kbarton, hfinkel, jdoerfert, Meinersbur, dmgreen
Reviewed By: reames
Subscribers: wuzish, hiraditya, jsji, llvm-commits, bmahjour, etiotto
Tag: LLVM
Differential Revision: https://reviews.llvm.org/D63885
llvm-svn: 367033
2019-07-26 00:13:18 +08:00
|
|
|
BasicBlock *Entry = &*(FI);
|
|
|
|
BranchInst *Guard = dyn_cast<BranchInst>(Entry->getTerminator());
|
[LOOPINFO] Extend Loop object to add utilities to get the loop bounds,
step, and loop induction variable.
Summary: This PR extends the loop object with more utilities to get loop
bounds, step, and loop induction variable. There already exists passes
which try to obtain the loop induction variable in their own pass, e.g.
loop interchange. It would be useful to have a common area to get these
information.
/// Example:
/// for (int i = lb; i < ub; i+=step)
/// <loop body>
/// --- pseudo LLVMIR ---
/// beforeloop:
/// guardcmp = (lb < ub)
/// if (guardcmp) goto preheader; else goto afterloop
/// preheader:
/// loop:
/// i1 = phi[{lb, preheader}, {i2, latch}]
/// <loop body>
/// i2 = i1 + step
/// latch:
/// cmp = (i2 < ub)
/// if (cmp) goto loop
/// exit:
/// afterloop:
///
/// getBounds
/// getInitialIVValue --> lb
/// getStepInst --> i2 = i1 + step
/// getStepValue --> step
/// getFinalIVValue --> ub
/// getCanonicalPredicate --> '<'
/// getDirection --> Increasing
/// getInductionVariable --> i1
/// getAuxiliaryInductionVariable --> {i1}
/// isCanonical --> false
Reviewers: kbarton, hfinkel, dmgreen, Meinersbur, jdoerfert, syzaara,
fhahn
Reviewed By: kbarton
Subscribers: tvvikram, bmahjour, etiotto, fhahn, jsji, hiraditya,
llvm-commits
Tag: LLVM
Differential Revision: https://reviews.llvm.org/D60565
llvm-svn: 362644
2019-06-06 04:42:47 +08:00
|
|
|
// First two basic block are entry and for.preheader - skip them.
|
|
|
|
++FI;
|
|
|
|
BasicBlock *Header = &*(++FI);
|
|
|
|
assert(Header->getName() == "for.body");
|
|
|
|
Loop *L = LI.getLoopFor(Header);
|
|
|
|
EXPECT_NE(L, nullptr);
|
|
|
|
|
|
|
|
Optional<Loop::LoopBounds> Bounds = L->getBounds(SE);
|
|
|
|
EXPECT_NE(Bounds, None);
|
|
|
|
ConstantInt *InitialIVValue =
|
|
|
|
dyn_cast<ConstantInt>(&Bounds->getInitialIVValue());
|
|
|
|
EXPECT_TRUE(InitialIVValue && InitialIVValue->isZero());
|
|
|
|
EXPECT_EQ(Bounds->getStepInst().getName(), "inc");
|
|
|
|
ConstantInt *StepValue =
|
|
|
|
dyn_cast_or_null<ConstantInt>(Bounds->getStepValue());
|
|
|
|
EXPECT_TRUE(StepValue && StepValue->isOne());
|
|
|
|
EXPECT_EQ(Bounds->getFinalIVValue().getName(), "ub");
|
|
|
|
EXPECT_EQ(Bounds->getCanonicalPredicate(), ICmpInst::ICMP_SLT);
|
|
|
|
EXPECT_EQ(Bounds->getDirection(),
|
|
|
|
Loop::LoopBounds::Direction::Increasing);
|
|
|
|
EXPECT_EQ(L->getInductionVariable(SE)->getName(), "i");
|
|
|
|
BasicBlock::iterator II = Header->begin();
|
|
|
|
PHINode &Instruction_i = cast<PHINode>(*(II));
|
|
|
|
EXPECT_TRUE(L->isAuxiliaryInductionVariable(Instruction_i, SE));
|
|
|
|
PHINode &Instruction_aux = cast<PHINode>(*(++II));
|
|
|
|
EXPECT_TRUE(L->isAuxiliaryInductionVariable(Instruction_aux, SE));
|
|
|
|
PHINode &Instruction_loopvariant = cast<PHINode>(*(++II));
|
|
|
|
EXPECT_FALSE(
|
|
|
|
L->isAuxiliaryInductionVariable(Instruction_loopvariant, SE));
|
|
|
|
PHINode &Instruction_usedoutside = cast<PHINode>(*(++II));
|
|
|
|
EXPECT_FALSE(
|
|
|
|
L->isAuxiliaryInductionVariable(Instruction_usedoutside, SE));
|
|
|
|
PHINode &Instruction_mulopcode = cast<PHINode>(*(++II));
|
|
|
|
EXPECT_FALSE(
|
|
|
|
L->isAuxiliaryInductionVariable(Instruction_mulopcode, SE));
|
[LOOPINFO] Introduce the loop guard API.
Summary:
This is the first patch for the loop guard. We introduced
getLoopGuardBranch() and isGuarded().
This currently only works on simplified loop, as it requires a preheader
and a latch to identify the guard.
It will work on loops of the form:
/// GuardBB:
/// br cond1, Preheader, ExitSucc <== GuardBranch
/// Preheader:
/// br Header
/// Header:
/// ...
/// br Latch
/// Latch:
/// br cond2, Header, ExitBlock
/// ExitBlock:
/// br ExitSucc
/// ExitSucc:
Prior discussions leading upto the decision to introduce the loop guard
API: http://lists.llvm.org/pipermail/llvm-dev/2019-May/132607.html
Reviewer: reames, kbarton, hfinkel, jdoerfert, Meinersbur, dmgreen
Reviewed By: reames
Subscribers: wuzish, hiraditya, jsji, llvm-commits, bmahjour, etiotto
Tag: LLVM
Differential Revision: https://reviews.llvm.org/D63885
llvm-svn: 367033
2019-07-26 00:13:18 +08:00
|
|
|
EXPECT_EQ(L->getLoopGuardBranch(), Guard);
|
|
|
|
EXPECT_TRUE(L->isGuarded());
|
2019-12-13 03:18:39 +08:00
|
|
|
EXPECT_TRUE(L->isRotatedForm());
|
2019-10-07 00:39:43 +08:00
|
|
|
});
|
|
|
|
}
|
|
|
|
|
|
|
|
TEST(LoopInfoTest, LoopNotInSimplifyForm) {
|
|
|
|
const char *ModuleStr =
|
|
|
|
"define void @foo(i32 %n) {\n"
|
|
|
|
"entry:\n"
|
|
|
|
" %guard.cmp = icmp sgt i32 %n, 0\n"
|
|
|
|
" br i1 %guard.cmp, label %for.cond, label %for.end\n"
|
|
|
|
"for.cond:\n"
|
|
|
|
" %i.0 = phi i32 [ 0, %entry ], [ %inc, %latch.1 ], [ %inc, %latch.2 ]\n"
|
|
|
|
" %inc = add nsw i32 %i.0, 1\n"
|
|
|
|
" %cmp = icmp slt i32 %i.0, %n\n"
|
|
|
|
" br i1 %cmp, label %latch.1, label %for.end\n"
|
|
|
|
"latch.1:\n"
|
|
|
|
" br i1 undef, label %for.cond, label %latch.2\n"
|
|
|
|
"latch.2:\n"
|
|
|
|
" br label %for.cond\n"
|
|
|
|
"for.end:\n"
|
|
|
|
" ret void\n"
|
|
|
|
"}\n";
|
|
|
|
|
|
|
|
// Parse the module.
|
|
|
|
LLVMContext Context;
|
|
|
|
std::unique_ptr<Module> M = makeLLVMModule(Context, ModuleStr);
|
|
|
|
|
|
|
|
runWithLoopInfo(*M, "foo", [&](Function &F, LoopInfo &LI) {
|
|
|
|
Function::iterator FI = F.begin();
|
|
|
|
// First basic block is entry - skip it.
|
|
|
|
BasicBlock *Header = &*(++FI);
|
|
|
|
assert(Header && "No header");
|
|
|
|
Loop *L = LI.getLoopFor(Header);
|
|
|
|
EXPECT_NE(L, nullptr);
|
|
|
|
EXPECT_FALSE(L->isLoopSimplifyForm());
|
|
|
|
// No loop guard because loop in not in simplify form.
|
|
|
|
EXPECT_EQ(L->getLoopGuardBranch(), nullptr);
|
|
|
|
EXPECT_FALSE(L->isGuarded());
|
|
|
|
});
|
|
|
|
}
|
|
|
|
|
|
|
|
TEST(LoopInfoTest, LoopLatchNotExiting) {
|
|
|
|
const char *ModuleStr =
|
|
|
|
"define void @foo(i32* %A, i32 %ub) {\n"
|
|
|
|
"entry:\n"
|
|
|
|
" %guardcmp = icmp slt i32 0, %ub\n"
|
|
|
|
" br i1 %guardcmp, label %for.preheader, label %for.end\n"
|
|
|
|
"for.preheader:\n"
|
|
|
|
" br label %for.body\n"
|
|
|
|
"for.body:\n"
|
|
|
|
" %i = phi i32 [ 0, %for.preheader ], [ %inc, %for.body ]\n"
|
|
|
|
" %idxprom = sext i32 %i to i64\n"
|
|
|
|
" %arrayidx = getelementptr inbounds i32, i32* %A, i64 %idxprom\n"
|
|
|
|
" store i32 %i, i32* %arrayidx, align 4\n"
|
|
|
|
" %inc = add nsw i32 %i, 1\n"
|
|
|
|
" %cmp = icmp slt i32 %inc, %ub\n"
|
|
|
|
" br i1 %cmp, label %for.latch, label %for.exit\n"
|
|
|
|
"for.latch:\n"
|
|
|
|
" br label %for.body\n"
|
|
|
|
"for.exit:\n"
|
|
|
|
" br label %for.end\n"
|
|
|
|
"for.end:\n"
|
|
|
|
" ret void\n"
|
|
|
|
"}\n";
|
|
|
|
|
|
|
|
// Parse the module.
|
|
|
|
LLVMContext Context;
|
|
|
|
std::unique_ptr<Module> M = makeLLVMModule(Context, ModuleStr);
|
|
|
|
|
|
|
|
runWithLoopInfoPlus(
|
|
|
|
*M, "foo",
|
|
|
|
[&](Function &F, LoopInfo &LI, ScalarEvolution &SE) {
|
|
|
|
Function::iterator FI = F.begin();
|
|
|
|
// First two basic block are entry and for.preheader - skip them.
|
|
|
|
++FI;
|
|
|
|
BasicBlock *Header = &*(++FI);
|
|
|
|
BasicBlock *Latch = &*(++FI);
|
|
|
|
assert(Header && "No header");
|
|
|
|
Loop *L = LI.getLoopFor(Header);
|
|
|
|
EXPECT_NE(L, nullptr);
|
|
|
|
EXPECT_TRUE(L->isLoopSimplifyForm());
|
|
|
|
EXPECT_EQ(L->getLoopLatch(), Latch);
|
|
|
|
EXPECT_FALSE(L->isLoopExiting(Latch));
|
|
|
|
// No loop guard becuase loop is not exiting on latch.
|
|
|
|
EXPECT_EQ(L->getLoopGuardBranch(), nullptr);
|
|
|
|
EXPECT_FALSE(L->isGuarded());
|
[LOOPINFO] Extend Loop object to add utilities to get the loop bounds,
step, and loop induction variable.
Summary: This PR extends the loop object with more utilities to get loop
bounds, step, and loop induction variable. There already exists passes
which try to obtain the loop induction variable in their own pass, e.g.
loop interchange. It would be useful to have a common area to get these
information.
/// Example:
/// for (int i = lb; i < ub; i+=step)
/// <loop body>
/// --- pseudo LLVMIR ---
/// beforeloop:
/// guardcmp = (lb < ub)
/// if (guardcmp) goto preheader; else goto afterloop
/// preheader:
/// loop:
/// i1 = phi[{lb, preheader}, {i2, latch}]
/// <loop body>
/// i2 = i1 + step
/// latch:
/// cmp = (i2 < ub)
/// if (cmp) goto loop
/// exit:
/// afterloop:
///
/// getBounds
/// getInitialIVValue --> lb
/// getStepInst --> i2 = i1 + step
/// getStepValue --> step
/// getFinalIVValue --> ub
/// getCanonicalPredicate --> '<'
/// getDirection --> Increasing
/// getInductionVariable --> i1
/// getAuxiliaryInductionVariable --> {i1}
/// isCanonical --> false
Reviewers: kbarton, hfinkel, dmgreen, Meinersbur, jdoerfert, syzaara,
fhahn
Reviewed By: kbarton
Subscribers: tvvikram, bmahjour, etiotto, fhahn, jsji, hiraditya,
llvm-commits
Tag: LLVM
Differential Revision: https://reviews.llvm.org/D60565
llvm-svn: 362644
2019-06-06 04:42:47 +08:00
|
|
|
});
|
|
|
|
}
|
2019-07-15 13:51:10 +08:00
|
|
|
|
|
|
|
// Examine getUniqueExitBlocks/getUniqueNonLatchExitBlocks functions.
|
|
|
|
TEST(LoopInfoTest, LoopUniqueExitBlocks) {
|
|
|
|
const char *ModuleStr =
|
|
|
|
"target datalayout = \"e-m:o-i64:64-f80:128-n8:16:32:64-S128\"\n"
|
|
|
|
"define void @foo(i32 %n, i1 %cond) {\n"
|
|
|
|
"entry:\n"
|
|
|
|
" br label %for.cond\n"
|
|
|
|
"for.cond:\n"
|
|
|
|
" %i.0 = phi i32 [ 0, %entry ], [ %inc, %for.inc ]\n"
|
|
|
|
" %cmp = icmp slt i32 %i.0, %n\n"
|
|
|
|
" br i1 %cond, label %for.inc, label %for.end1\n"
|
|
|
|
"for.inc:\n"
|
|
|
|
" %inc = add nsw i32 %i.0, 1\n"
|
|
|
|
" br i1 %cmp, label %for.cond, label %for.end2, !llvm.loop !0\n"
|
|
|
|
"for.end1:\n"
|
|
|
|
" br label %for.end\n"
|
|
|
|
"for.end2:\n"
|
|
|
|
" br label %for.end\n"
|
|
|
|
"for.end:\n"
|
|
|
|
" ret void\n"
|
|
|
|
"}\n"
|
|
|
|
"!0 = distinct !{!0, !1}\n"
|
|
|
|
"!1 = !{!\"llvm.loop.distribute.enable\", i1 true}\n";
|
|
|
|
|
|
|
|
// Parse the module.
|
|
|
|
LLVMContext Context;
|
|
|
|
std::unique_ptr<Module> M = makeLLVMModule(Context, ModuleStr);
|
|
|
|
|
|
|
|
runWithLoopInfo(*M, "foo", [&](Function &F, LoopInfo &LI) {
|
|
|
|
Function::iterator FI = F.begin();
|
|
|
|
// First basic block is entry - skip it.
|
|
|
|
BasicBlock *Header = &*(++FI);
|
|
|
|
assert(Header->getName() == "for.cond");
|
|
|
|
Loop *L = LI.getLoopFor(Header);
|
|
|
|
|
|
|
|
SmallVector<BasicBlock *, 2> Exits;
|
|
|
|
// This loop has 2 unique exits.
|
|
|
|
L->getUniqueExitBlocks(Exits);
|
|
|
|
EXPECT_TRUE(Exits.size() == 2);
|
|
|
|
// And one unique non latch exit.
|
|
|
|
Exits.clear();
|
|
|
|
L->getUniqueNonLatchExitBlocks(Exits);
|
|
|
|
EXPECT_TRUE(Exits.size() == 1);
|
|
|
|
});
|
|
|
|
}
|
2019-07-17 15:09:20 +08:00
|
|
|
|
|
|
|
// Regression test for getUniqueNonLatchExitBlocks functions.
|
|
|
|
// It should detect the exit if it comes from both latch and non-latch blocks.
|
|
|
|
TEST(LoopInfoTest, LoopNonLatchUniqueExitBlocks) {
|
|
|
|
const char *ModuleStr =
|
|
|
|
"target datalayout = \"e-m:o-i64:64-f80:128-n8:16:32:64-S128\"\n"
|
|
|
|
"define void @foo(i32 %n, i1 %cond) {\n"
|
|
|
|
"entry:\n"
|
|
|
|
" br label %for.cond\n"
|
|
|
|
"for.cond:\n"
|
|
|
|
" %i.0 = phi i32 [ 0, %entry ], [ %inc, %for.inc ]\n"
|
|
|
|
" %cmp = icmp slt i32 %i.0, %n\n"
|
|
|
|
" br i1 %cond, label %for.inc, label %for.end\n"
|
|
|
|
"for.inc:\n"
|
|
|
|
" %inc = add nsw i32 %i.0, 1\n"
|
|
|
|
" br i1 %cmp, label %for.cond, label %for.end, !llvm.loop !0\n"
|
|
|
|
"for.end:\n"
|
|
|
|
" ret void\n"
|
|
|
|
"}\n"
|
|
|
|
"!0 = distinct !{!0, !1}\n"
|
|
|
|
"!1 = !{!\"llvm.loop.distribute.enable\", i1 true}\n";
|
|
|
|
|
|
|
|
// Parse the module.
|
|
|
|
LLVMContext Context;
|
|
|
|
std::unique_ptr<Module> M = makeLLVMModule(Context, ModuleStr);
|
|
|
|
|
|
|
|
runWithLoopInfo(*M, "foo", [&](Function &F, LoopInfo &LI) {
|
|
|
|
Function::iterator FI = F.begin();
|
|
|
|
// First basic block is entry - skip it.
|
|
|
|
BasicBlock *Header = &*(++FI);
|
|
|
|
assert(Header->getName() == "for.cond");
|
|
|
|
Loop *L = LI.getLoopFor(Header);
|
|
|
|
|
|
|
|
SmallVector<BasicBlock *, 2> Exits;
|
|
|
|
// This loop has 1 unique exit.
|
|
|
|
L->getUniqueExitBlocks(Exits);
|
|
|
|
EXPECT_TRUE(Exits.size() == 1);
|
|
|
|
// And one unique non latch exit.
|
|
|
|
Exits.clear();
|
|
|
|
L->getUniqueNonLatchExitBlocks(Exits);
|
|
|
|
EXPECT_TRUE(Exits.size() == 1);
|
|
|
|
});
|
|
|
|
}
|
2019-12-04 03:20:08 +08:00
|
|
|
|
|
|
|
// Test that a pointer-chasing loop is not rotated.
|
|
|
|
TEST(LoopInfoTest, LoopNotRotated) {
|
|
|
|
const char *ModuleStr =
|
|
|
|
"target datalayout = \"e-m:o-i64:64-f80:128-n8:16:32:64-S128\"\n"
|
|
|
|
"define void @foo(i32* %elem) {\n"
|
|
|
|
"entry:\n"
|
|
|
|
" br label %while.cond\n"
|
|
|
|
"while.cond:\n"
|
|
|
|
" %elem.addr.0 = phi i32* [ %elem, %entry ], [ %incdec.ptr, %while.body "
|
|
|
|
"]\n"
|
|
|
|
" %tobool = icmp eq i32* %elem.addr.0, null\n"
|
|
|
|
" br i1 %tobool, label %while.end, label %while.body\n"
|
|
|
|
"while.body:\n"
|
|
|
|
" %incdec.ptr = getelementptr inbounds i32, i32* %elem.addr.0, i64 1\n"
|
|
|
|
" br label %while.cond\n"
|
|
|
|
"while.end:\n"
|
|
|
|
" ret void\n"
|
|
|
|
"}\n";
|
|
|
|
|
|
|
|
// Parse the module.
|
|
|
|
LLVMContext Context;
|
|
|
|
std::unique_ptr<Module> M = makeLLVMModule(Context, ModuleStr);
|
|
|
|
|
|
|
|
runWithLoopInfo(*M, "foo", [&](Function &F, LoopInfo &LI) {
|
|
|
|
Function::iterator FI = F.begin();
|
|
|
|
// First basic block is entry - skip it.
|
|
|
|
BasicBlock *Header = &*(++FI);
|
|
|
|
assert(Header->getName() == "while.cond");
|
|
|
|
Loop *L = LI.getLoopFor(Header);
|
|
|
|
EXPECT_NE(L, nullptr);
|
|
|
|
|
|
|
|
// This loop is in simplified form.
|
|
|
|
EXPECT_TRUE(L->isLoopSimplifyForm());
|
|
|
|
|
|
|
|
// This loop is not rotated.
|
2019-12-13 03:18:39 +08:00
|
|
|
EXPECT_FALSE(L->isRotatedForm());
|
2019-12-04 03:20:08 +08:00
|
|
|
});
|
|
|
|
}
|