[unroll] Extract the code to enqueue operansd for the worklist in the
unroll analysis into a lambda and call it. That's much simpler than
duplicating all the code.
llvm-svn: 229053
diff --git a/llvm/lib/Transforms/Scalar/LoopUnrollPass.cpp b/llvm/lib/Transforms/Scalar/LoopUnrollPass.cpp
index e4771db..63d3cc6 100644
--- a/llvm/lib/Transforms/Scalar/LoopUnrollPass.cpp
+++ b/llvm/lib/Transforms/Scalar/LoopUnrollPass.cpp
@@ -509,6 +509,15 @@
// worklist.
SmallPtrSet<Instruction *, 4> OperandSet;
+ // Lambda to enque operands onto the worklist.
+ auto EnqueueOperands = [&](Instruction &I) {
+ OperandSet.clear();
+ for (auto *Op : I.operand_values())
+ if (auto *OpI = dyn_cast<Instruction>(Op))
+ if (OperandSet.insert(OpI).second)
+ Worklist.push_back(OpI);
+ };
+
// Start by initializing worklist with simplified instructions.
for (auto &FoldedKeyValue : SimplifiedValues)
if (auto *FoldedInst = dyn_cast<Instruction>(FoldedKeyValue.first)) {
@@ -516,11 +525,7 @@
// Add each instruction operand of this dead instruction to the
// worklist.
- OperandSet.clear();
- for (auto *Op : FoldedInst->operand_values())
- if (auto *OpI = dyn_cast<Instruction>(Op))
- if (OperandSet.insert(OpI).second)
- Worklist.push_back(OpI);
+ EnqueueOperands(*FoldedInst);
}
// If a definition of an insn is only used by simplified or dead
@@ -545,11 +550,7 @@
if (AllUsersFolded) {
NumberOfOptimizedInstructions += TTI.getUserCost(I);
DeadInstructions.insert(I);
- OperandSet.clear();
- for (auto *Op : I->operand_values())
- if (auto *OpI = dyn_cast<Instruction>(Op))
- if (OperandSet.insert(OpI).second)
- Worklist.push_back(OpI);
+ EnqueueOperands(*I);
}
}
return NumberOfOptimizedInstructions;