commit 80005613cccd940ba621cea17837ce914695fb11
author River Riddle <riverriddle@google.com> Thu Dec 13 09:00:06 2018 -0800
committer jpienaar <jpienaar@google.com> Fri Mar 29 14:27:14 2019 -0700
tree 28bb6ece3b2f71498695d929e12c502d591e7b60
parent a48bb91132cc614c98c1620e07ec54998852ad9b

Check if the operation is already in the worklist before adding it.

PiperOrigin-RevId: 225379496

lib/Transforms/Utils/GreedyPatternRewriteDriver.cpp

diff --git a/lib/Transforms/Utils/GreedyPatternRewriteDriver.cpp b/lib/Transforms/Utils/GreedyPatternRewriteDriver.cpp
index 518f117..554e3cb 100644
--- a/lib/Transforms/Utils/GreedyPatternRewriteDriver.cpp
+++ b/lib/Transforms/Utils/GreedyPatternRewriteDriver.cpp
@@ -40,6 +40,10 @@
   void simplifyFunction(Function *currentFunction, WorklistRewriter &rewriter);
 
   void addToWorklist(Operation *op) {
+    // Check to see if the worklist already contains this op.
+    if (worklistMap.count(op))
+      return;
+
     worklistMap[op] = worklist.size();
     worklist.push_back(op);
   }