diff --git a/lib/Transforms/Utils/Local.cpp b/lib/Transforms/Utils/Local.cpp index 4e1a00fa2d3..179f29cdafa 100644 --- a/lib/Transforms/Utils/Local.cpp +++ b/lib/Transforms/Utils/Local.cpp @@ -169,16 +169,20 @@ bool llvm::ConstantFoldTerminator(BasicBlock *BB, bool DeleteDeadConditions) { // Otherwise, we can fold this switch into a conditional branch // instruction if it has only one non-default destination. SwitchInst::CaseIt FirstCase = SI->case_begin(); - Value *Cond = Builder.CreateICmpEQ(SI->getCondition(), - FirstCase.getCaseValue(), "cond"); + ConstantRangesSet CRS = FirstCase.getCaseValueEx(); + if (CRS.getNumItems() == 1 && CRS.isSingleNumber(0)) { + Value *Cond = Builder.CreateICmpEQ(SI->getCondition(), + CRS.getItem(0).Low, "cond"); - // Insert the new branch. - Builder.CreateCondBr(Cond, FirstCase.getCaseSuccessor(), - SI->getDefaultDest()); + // Insert the new branch. + Builder.CreateCondBr(Cond, FirstCase.getCaseSuccessor(), + SI->getDefaultDest()); - // Delete the old switch. - SI->eraseFromParent(); - return true; + // Delete the old switch. + SI->eraseFromParent(); + return true; + + } } return false; }