diff options
author | Florian Hahn <florian.hahn@arm.com> | 2018-07-10 10:45:46 +0000 |
---|---|---|
committer | Florian Hahn <florian.hahn@arm.com> | 2018-07-10 10:45:46 +0000 |
commit | 20b07351273d8883ad8141d88f8bd0ac43a6d854 (patch) | |
tree | 503ebd011a21f9619dd35b9e57230e45d1cebccc /unittests | |
parent | 9cec565a79f1e99838702e8ed19f8e0eea1bc2e6 (diff) |
[VPlan] Add VPlanTestBase.h with helper class to build VPlan for tests.
Reviewers: dcaballe, hsaito, rengolin
Reviewed By: dcaballe
Differential Revision: https://reviews.llvm.org/D49032
git-svn-id: https://llvm.org/svn/llvm-project/llvm/trunk@336653 91177308-0d34-0410-b5e6-96231b3b80d8
Diffstat (limited to 'unittests')
-rw-r--r-- | unittests/Transforms/Vectorize/VPlanHCFGTest.cpp | 38 | ||||
-rw-r--r-- | unittests/Transforms/Vectorize/VPlanTestBase.h | 68 |
2 files changed, 76 insertions, 30 deletions
diff --git a/unittests/Transforms/Vectorize/VPlanHCFGTest.cpp b/unittests/Transforms/Vectorize/VPlanHCFGTest.cpp index e2b8a8ab5d8..215c3ef40da 100644 --- a/unittests/Transforms/Vectorize/VPlanHCFGTest.cpp +++ b/unittests/Transforms/Vectorize/VPlanHCFGTest.cpp @@ -8,35 +8,16 @@ //===----------------------------------------------------------------------===// #include "../lib/Transforms/Vectorize/VPlan.h" -#include "../lib/Transforms/Vectorize/VPlanHCFGBuilder.h" #include "../lib/Transforms/Vectorize/VPlanHCFGTransforms.h" -#include "llvm/AsmParser/Parser.h" -#include "llvm/IR/Dominators.h" +#include "VPlanTestBase.h" #include "gtest/gtest.h" namespace llvm { namespace { -class VPlanHCFGTest : public testing::Test { -protected: - std::unique_ptr<DominatorTree> DT; - std::unique_ptr<LoopInfo> LI; - - VPlanHCFGTest() {} - - VPlanPtr doBuildPlan(BasicBlock *LoopHeader) { - DT.reset(new DominatorTree(*LoopHeader->getParent())); - LI.reset(new LoopInfo(*DT)); - - auto Plan = llvm::make_unique<VPlan>(); - VPlanHCFGBuilder HCFGBuilder(LI->getLoopFor(LoopHeader), LI.get()); - HCFGBuilder.buildHierarchicalCFG(*Plan.get()); - return Plan; - } -}; +class VPlanHCFGTest : public VPlanTestBase {}; TEST_F(VPlanHCFGTest, testBuildHCFGInnerLoop) { - LLVMContext Ctx; const char *ModuleString = "define void @f(i32* %A, i64 %N) {\n" "entry:\n" @@ -54,12 +35,11 @@ TEST_F(VPlanHCFGTest, testBuildHCFGInnerLoop) { " ret void\n" "}\n"; - SMDiagnostic Err; - std::unique_ptr<Module> M = parseAssemblyString(ModuleString, Err, Ctx); + Module &M = parseModule(ModuleString); - Function *F = M->getFunction("f"); + Function *F = M.getFunction("f"); BasicBlock *LoopHeader = F->getEntryBlock().getSingleSuccessor(); - auto Plan = doBuildPlan(LoopHeader); + auto Plan = buildHCFG(LoopHeader); VPBasicBlock *Entry = Plan->getEntry()->getEntryBasicBlock(); EXPECT_NE(nullptr, Entry->getSingleSuccessor()); @@ -115,7 +95,6 @@ TEST_F(VPlanHCFGTest, testBuildHCFGInnerLoop) { } TEST_F(VPlanHCFGTest, testVPInstructionToVPRecipesInner) { - LLVMContext Ctx; const char *ModuleString = "define void @f(i32* %A, i64 %N) {\n" "entry:\n" @@ -133,12 +112,11 @@ TEST_F(VPlanHCFGTest, testVPInstructionToVPRecipesInner) { " ret void\n" "}\n"; - SMDiagnostic Err; - std::unique_ptr<Module> M = parseAssemblyString(ModuleString, Err, Ctx); + Module &M = parseModule(ModuleString); - Function *F = M->getFunction("f"); + Function *F = M.getFunction("f"); BasicBlock *LoopHeader = F->getEntryBlock().getSingleSuccessor(); - auto Plan = doBuildPlan(LoopHeader); + auto Plan = buildHCFG(LoopHeader); LoopVectorizationLegality::InductionList Inductions; SmallPtrSet<Instruction *, 1> DeadInstructions; diff --git a/unittests/Transforms/Vectorize/VPlanTestBase.h b/unittests/Transforms/Vectorize/VPlanTestBase.h new file mode 100644 index 00000000000..3e0fb5d6786 --- /dev/null +++ b/unittests/Transforms/Vectorize/VPlanTestBase.h @@ -0,0 +1,68 @@ +//===- llvm/unittest/Transforms/Vectorize/VPlanTestBase.h -----------------===// +// +// The LLVM Compiler Infrastructure +// +// This file is distributed under the University of Illinois Open Source +// License. See LICENSE.TXT for details. +// +//===----------------------------------------------------------------------===// +/// \file +/// This file defines a VPlanTestBase class, which provides helpers to parse +/// a LLVM IR string and create VPlans given a loop entry block. +//===----------------------------------------------------------------------===// +#ifndef LLVM_UNITTESTS_TRANSFORMS_VECTORIZE_VPLANTESTBASE_H +#define LLVM_UNITTESTS_TRANSFORMS_VECTORIZE_VPLANTESTBASE_H + +#include "../lib/Transforms/Vectorize/VPlan.h" +#include "../lib/Transforms/Vectorize/VPlanHCFGBuilder.h" +#include "llvm/Analysis/LoopInfo.h" +#include "llvm/AsmParser/Parser.h" +#include "llvm/IR/Dominators.h" +#include "llvm/Support/SourceMgr.h" +#include "gtest/gtest.h" + +namespace llvm { + +/// Helper class to create a module from an assembly string and VPlans for a +/// given loop entry block. +class VPlanTestBase : public testing::Test { +protected: + std::unique_ptr<DominatorTree> DT; + std::unique_ptr<LoopInfo> LI; + std::unique_ptr<Module> M; + std::unique_ptr<LLVMContext> Ctx; + + VPlanTestBase() : Ctx(new LLVMContext) {} + ~VPlanTestBase() { + LI.release(); + DT.release(); + M.release(); + // We need to release objects depending on Ctx first. + Ctx.release(); + } + + Module &parseModule(const char *ModuleString) { + SMDiagnostic Err; + M = parseAssemblyString(ModuleString, Err, *Ctx); + EXPECT_TRUE(M); + return *M; + } + + void doAnalysis(Function &F) { + DT.reset(new DominatorTree(F)); + LI.reset(new LoopInfo(*DT)); + } + + VPlanPtr buildHCFG(BasicBlock *LoopHeader) { + doAnalysis(*LoopHeader->getParent()); + + auto Plan = llvm::make_unique<VPlan>(); + VPlanHCFGBuilder HCFGBuilder(LI->getLoopFor(LoopHeader), LI.get()); + HCFGBuilder.buildHierarchicalCFG(*Plan.get()); + return Plan; + } +}; + +} // namespace llvm + +#endif // LLVM_UNITTESTS_TRANSFORMS_VECTORIZE_VPLANTESTBASE_H |