Skip duplicate template operations with set

marco-compiler · Oct 25, 2024 · 965fb28 · 965fb28
1 parent 33759b7
commit 965fb28
Showing 1 changed file with 3 additions and 7 deletions.
diff --git a/lib/Dialect/BaseModelica/Transforms/CallCSE.cpp b/lib/Dialect/BaseModelica/Transforms/CallCSE.cpp
@@ -34,18 +34,14 @@ void collectCallOps(ModelOp modelOp, llvm::SmallVectorImpl<CallOp> &callOps) {
   llvm::SmallVector<EquationInstanceOp> dynamicEquationOps;
   modelOp.collectMainEquations(dynamicEquationOps);
 
-  llvm::SmallVector<EquationTemplateOp> templateOps;
-
+  llvm::DenseSet<EquationTemplateOp> visitedTemplateOps;
   for (EquationInstanceOp equationOp : dynamicEquationOps) {
     EquationTemplateOp templateOp = equationOp.getTemplate();
     if (!templateOp.getInductionVariables().empty() ||
-        llvm::is_contained(templateOps, templateOp)) {
+        visitedTemplateOps.contains(templateOp)) {
       continue;
     }
-    templateOps.push_back(templateOp);
-  }
-
-  for (EquationTemplateOp templateOp : templateOps) {
+    visitedTemplateOps.insert(templateOp);
     templateOp->walk([&](CallOp callOp) { callOps.push_back(callOp); });
   }
 }