Add a visited set

llvm · Mar 14, 2024 · 2c7c0bb · 2c7c0bb
1 parent d6d9e35
commit 2c7c0bb
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/lib/Dialect/FIRRTL/Transforms/InferReadWrite.cpp b/lib/Dialect/FIRRTL/Transforms/InferReadWrite.cpp
@@ -363,9 +363,12 @@ struct InferReadWritePass : public InferReadWriteBase<InferReadWritePass> {
       return defOp;
     };
     SmallVector<Value> stack;
+    llvm::SmallDenseSet<Value> visited;
     stack.push_back(wmodeDriver);
     while (!stack.empty()) {
       auto driver = stack.pop_back_val();
+      if (!visited.insert(driver).second)
+        continue;
       auto *defOp = getDriverOp(driver);
       if (!defOp)
         continue;