Index: lib/Transforms/Scalar/Reassociate.cpp =================================================================== --- lib/Transforms/Scalar/Reassociate.cpp +++ lib/Transforms/Scalar/Reassociate.cpp @@ -842,6 +842,11 @@ if (TheNeg->getParent()->getParent() != BI->getParent()->getParent()) continue; + // Don't move the negate if it's in the block we're currently optimizing as + // this may invalidate our iterator. + if (TheNeg->getParent() == BI->getParent()) + continue; + BasicBlock::iterator InsertPt; if (Instruction *InstInput = dyn_cast(V)) { if (InvokeInst *II = dyn_cast(InstInput)) { @@ -1863,6 +1868,8 @@ /// Zap the given instruction, adding interesting operands to the work list. void ReassociatePass::EraseInst(Instruction *I) { assert(isInstructionTriviallyDead(I) && "Trivially dead instructions only!"); + DEBUG(dbgs() << "Erasing dead inst: "; I->dump()); + SmallVector Ops(I->op_begin(), I->op_end()); // Erase the dead instruction. ValueRankMap.erase(I); Index: test/Transforms/Reassociate/invalid-iterator.ll =================================================================== --- /dev/null +++ test/Transforms/Reassociate/invalid-iterator.ll @@ -0,0 +1,30 @@ +; RUN: opt < %s -reassociate -die -S | FileCheck %s + +; PR28367 +; Check to make sure %sub2 isn't moved from for.body. Doing so would invalidate +; the iterator. + +; CHECK-LABEL: @fn1 +; CHECK: for.body: +; CHECK-NEXT: %sub2 = sub nsw i32 0, %d.0 +; CHECK-NEXT: store i32 %sub2, i32* %ptr, align 4 +define void @fn1(i32 %a, i1 %c, i32* %ptr) { +entry: + br label %for.cond + +for.cond: + %d.0 = phi i32 [ 1, %entry ], [ 2, %for.body ] + br i1 %c, label %for.end, label %for.body + +for.body: + %sub1 = sub i32 %a, %d.0 + %dead1 = add i32 %sub1, 1 + %dead2 = mul i32 %dead1, 3 + %dead3 = mul i32 %dead2, %sub1 + %sub2 = sub nsw i32 0, %d.0 + store i32 %sub2, i32* %ptr, align 4 + br label %for.cond + +for.end: + ret void +}