diff options
author | Logan Chien <tzuhsiang.chien@gmail.com> | 2014-03-16 21:00:19 +0800 |
---|---|---|
committer | Logan Chien <tzuhsiang.chien@gmail.com> | 2014-03-16 21:03:24 +0800 |
commit | 6bf46f579f8a936804a99aff5c382eda68a59673 (patch) | |
tree | c7bf3fb89d1995a3b3051206f950d13ba31bd310 | |
parent | 3831cbe99734261bb779e4e8c0e33c94958965df (diff) | |
download | llvm-6bf46f579f8a936804a99aff5c382eda68a59673.tar.gz |
[ndk] Fix createBranchWeights() assertions.
Backport r196845 from the upstream to fix http://b.android.com/57381
The original commit message:
------------------------------------------------------------
commit cb3badc8949fb15ff5f812774c3ff9e698850f16
Author: Justin Bogner <mail justinbogner com>
Date: Tue Dec 10 00:13:41 2013 +0000
Transforms: Don't create bad branch weights when folding a switch
This avoids creating branch weight metadata of length one when we fold
cases into the default of a switch instruction, which was triggering
an assert.
git-svn-id: https://llvm.org/svn/llvm-project/llvm/trunk@196845 91177308-0d3
------------------------------------------------------------
-rw-r--r-- | lib/Transforms/Utils/Local.cpp | 6 | ||||
-rw-r--r-- | test/Transforms/SimplifyCFG/preserve-branchweights.ll | 18 |
2 files changed, 22 insertions, 2 deletions
diff --git a/lib/Transforms/Utils/Local.cpp b/lib/Transforms/Utils/Local.cpp index 2768041fb2b..8a15c7c9a1b 100644 --- a/lib/Transforms/Utils/Local.cpp +++ b/lib/Transforms/Utils/Local.cpp @@ -127,8 +127,10 @@ bool llvm::ConstantFoldTerminator(BasicBlock *BB, bool DeleteDeadConditions, // dest. If so, eliminate it as an explicit compare. if (i.getCaseSuccessor() == DefaultDest) { MDNode* MD = SI->getMetadata(LLVMContext::MD_prof); - // MD should have 2 + NumCases operands. - if (MD && MD->getNumOperands() == 2 + SI->getNumCases()) { + unsigned NCases = SI->getNumCases(); + // Fold the case metadata into the default if there will be any branches + // left, unless the metadata doesn't match the switch. + if (NCases > 1 && MD && MD->getNumOperands() == 2 + NCases) { // Collect branch weights into a vector. SmallVector<uint32_t, 8> Weights; for (unsigned MD_i = 1, MD_e = MD->getNumOperands(); MD_i < MD_e; diff --git a/test/Transforms/SimplifyCFG/preserve-branchweights.ll b/test/Transforms/SimplifyCFG/preserve-branchweights.ll index 4022ed6927e..912544646c0 100644 --- a/test/Transforms/SimplifyCFG/preserve-branchweights.ll +++ b/test/Transforms/SimplifyCFG/preserve-branchweights.ll @@ -293,6 +293,23 @@ c: ret void } +;; test12 - Don't crash if the whole switch is removed +define void @test12(i32 %M, i32 %N) nounwind uwtable { +entry: + switch i32 %N, label %sw.bb [ + i32 1, label %sw.bb + ], !prof !9 +; CHECK: test12 +; CHECK-NOT: switch + +sw.bb: + call void @helper(i32 0) + br label %sw.epilog + +sw.epilog: + ret void +} + !0 = metadata !{metadata !"branch_weights", i32 3, i32 5} !1 = metadata !{metadata !"branch_weights", i32 1, i32 1} !2 = metadata !{metadata !"branch_weights", i32 1, i32 2} @@ -302,6 +319,7 @@ c: !6 = metadata !{metadata !"branch_weights", i32 1, i32 3} !7 = metadata !{metadata !"branch_weights", i32 33, i32 9, i32 8, i32 7} !8 = metadata !{metadata !"branch_weights", i32 33, i32 9, i32 8} +!9 = metadata !{metadata !"branch_weights", i32 7, i32 6} ; CHECK: !0 = metadata !{metadata !"branch_weights", i32 5, i32 11} ; CHECK: !1 = metadata !{metadata !"branch_weights", i32 1, i32 5} |