diff options
author | Matt Arsenault <Matthew.Arsenault@amd.com> | 2018-01-31 22:54:37 +0000 |
---|---|---|
committer | Matt Arsenault <Matthew.Arsenault@amd.com> | 2018-01-31 22:54:37 +0000 |
commit | 0c580ec2a434ba030606e085a23e2268ed0e47b0 (patch) | |
tree | 8152facb01a89261e5d56a36b9977deda70d2d7b /unittests/Transforms/Utils | |
parent | 9a2ee25a73b02dd4cd34d3ee1d398f37d2686c98 (diff) |
Utils: Fix DomTree update for entry block
If SplitBlockPredecessors was used on a function entry block,
it wouldn't update the dominator tree.
git-svn-id: https://llvm.org/svn/llvm-project/llvm/trunk@323928 91177308-0d34-0410-b5e6-96231b3b80d8
Diffstat (limited to 'unittests/Transforms/Utils')
-rw-r--r-- | unittests/Transforms/Utils/BasicBlockUtils.cpp | 52 | ||||
-rw-r--r-- | unittests/Transforms/Utils/CMakeLists.txt | 1 | ||||
-rw-r--r-- | unittests/Transforms/Utils/Local.cpp | 2 |
3 files changed, 54 insertions, 1 deletions
diff --git a/unittests/Transforms/Utils/BasicBlockUtils.cpp b/unittests/Transforms/Utils/BasicBlockUtils.cpp new file mode 100644 index 00000000000..2d0a9302011 --- /dev/null +++ b/unittests/Transforms/Utils/BasicBlockUtils.cpp @@ -0,0 +1,52 @@ +//===- BasicBlockUtils.cpp - Unit tests for BasicBlockUtils ---------------===// +// +// The LLVM Compiler Infrastructure +// +// This file is distributed under the University of Illinois Open Source +// License. See LICENSE.TXT for details. +// +//===----------------------------------------------------------------------===// + +#include "llvm/Transforms/Utils/BasicBlockUtils.h" +#include "llvm/AsmParser/Parser.h" +#include "llvm/IR/BasicBlock.h" +#include "llvm/IR/Dominators.h" +#include "llvm/IR/LLVMContext.h" +#include "llvm/Support/SourceMgr.h" +#include "gtest/gtest.h" + +using namespace llvm; + +static std::unique_ptr<Module> parseIR(LLVMContext &C, const char *IR) { + SMDiagnostic Err; + std::unique_ptr<Module> Mod = parseAssemblyString(IR, Err, C); + if (!Mod) + Err.print("BasicBlockUtilsTests", errs()); + return Mod; +} + +TEST(BasicBlockUtils, SplitBlockPredecessors) { + LLVMContext C; + + std::unique_ptr<Module> M = parseIR( + C, + "define i32 @basic_func(i1 %cond) {\n" + "entry:\n" + " br i1 %cond, label %bb0, label %bb1\n" + "bb0:\n" + " br label %bb1\n" + "bb1:\n" + " %phi = phi i32 [ 0, %entry ], [ 1, %bb0 ]" + " ret i32 %phi\n" + "}\n" + "\n" + ); + + auto *F = M->getFunction("basic_func"); + DominatorTree DT(*F); + + // Make sure the dominator tree is properly updated if calling this on the + // entry block. + SplitBlockPredecessors(&F->getEntryBlock(), {}, "split.entry", &DT); + EXPECT_TRUE(DT.verify()); +} diff --git a/unittests/Transforms/Utils/CMakeLists.txt b/unittests/Transforms/Utils/CMakeLists.txt index e2bb0af0f77..574e66f1d4e 100644 --- a/unittests/Transforms/Utils/CMakeLists.txt +++ b/unittests/Transforms/Utils/CMakeLists.txt @@ -8,6 +8,7 @@ set(LLVM_LINK_COMPONENTS add_llvm_unittest(UtilsTests ASanStackFrameLayoutTest.cpp + BasicBlockUtils.cpp Cloning.cpp CodeExtractor.cpp FunctionComparator.cpp diff --git a/unittests/Transforms/Utils/Local.cpp b/unittests/Transforms/Utils/Local.cpp index ab601222ae8..9b305f739a2 100644 --- a/unittests/Transforms/Utils/Local.cpp +++ b/unittests/Transforms/Utils/Local.cpp @@ -100,7 +100,7 @@ TEST(Local, RemoveDuplicatePHINodes) { EXPECT_EQ(3U, BB->size()); } -std::unique_ptr<Module> parseIR(LLVMContext &C, const char *IR) { +static std::unique_ptr<Module> parseIR(LLVMContext &C, const char *IR) { SMDiagnostic Err; std::unique_ptr<Module> Mod = parseAssemblyString(IR, Err, C); if (!Mod) |