From 7a50155227e84d94ae233407050a7f85e7f53199 Mon Sep 17 00:00:00 2001 From: Stepan Dyatkovskiy Date: Wed, 23 May 2012 08:18:26 +0000 Subject: PR1255(case ranges) related changes in Local Transformations. llvm-svn: 157315 --- llvm/lib/Transforms/Utils/Local.cpp | 24 ++++++++++++++---------- 1 file changed, 14 insertions(+), 10 deletions(-) (limited to 'llvm/lib/Transforms/Utils/Local.cpp') diff --git a/llvm/lib/Transforms/Utils/Local.cpp b/llvm/lib/Transforms/Utils/Local.cpp index 4e1a00fa2d3..179f29cdafa 100644 --- a/llvm/lib/Transforms/Utils/Local.cpp +++ b/llvm/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"); - - // Insert the new branch. - Builder.CreateCondBr(Cond, FirstCase.getCaseSuccessor(), - SI->getDefaultDest()); - - // Delete the old switch. - SI->eraseFromParent(); - return true; + 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()); + + // Delete the old switch. + SI->eraseFromParent(); + return true; + + } } return false; } -- cgit v1.2.3