Index: llvm/trunk/lib/Transforms/Utils/SimplifyCFG.cpp =================================================================== --- llvm/trunk/lib/Transforms/Utils/SimplifyCFG.cpp +++ llvm/trunk/lib/Transforms/Utils/SimplifyCFG.cpp @@ -4218,14 +4218,15 @@ } } } else if (auto *SI = dyn_cast(TI)) { - for (auto i = SI->case_begin(), e = SI->case_end(); i != e;) { + SwitchInstProfUpdateWrapper SU(*SI); + for (auto i = SU->case_begin(), e = SU->case_end(); i != e;) { if (i->getCaseSuccessor() != BB) { ++i; continue; } - BB->removePredecessor(SI->getParent()); - i = SI->removeCase(i); - e = SI->case_end(); + BB->removePredecessor(SU->getParent()); + i = SU.removeCase(i); + e = SU->case_end(); Changed = true; } } else if (auto *II = dyn_cast(TI)) { Index: llvm/trunk/test/Transforms/SimplifyCFG/switch-profmd.ll =================================================================== --- llvm/trunk/test/Transforms/SimplifyCFG/switch-profmd.ll +++ llvm/trunk/test/Transforms/SimplifyCFG/switch-profmd.ll @@ -0,0 +1,35 @@ +; RUN: opt -S -simplifycfg < %s | FileCheck %s + +declare i32 @f(i32 %val) + +; Check that eliminating cases with unreachable branches keeps +; prof branch_weights metadata consistent with switch instruction. +define i32 @test_switch_to_unreachable(i32 %val) { +; CHECK-LABEL: test_switch_to_unreachable +; CHECK: switch +; CHECK-NOT: i32 0, label %on0 + switch i32 %val, label %otherwise [ + i32 0, label %on0 + i32 1, label %on1 + i32 2, label %on2 + ], !prof !{!"branch_weights", i32 99, i32 0, i32 1, i32 2} +; CHECK: !prof ![[MD0:[0-9]+]] + +otherwise: + %result = call i32 @f(i32 -1) + ret i32 %result + +on0: + unreachable + ret i32 125 + +on1: + %result1 = call i32 @f(i32 -2) + ret i32 %result1 + +on2: + %result2 = call i32 @f(i32 -3) + ret i32 %result2 +} + +; CHECK: ![[MD0]] = !{!"branch_weights", i32 99, i32 2, i32 1}