[mlir][SymbolDCE] Track the number of symbols DCE'd

Added a statistic counting the number of erased symbols. Differential Revision: https://reviews.llvm.org/D121930
llvm · Mar 18, 2022 · 1ebf1af · 1ebf1af
1 parent c975668
commit 1ebf1af
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git a/mlir/include/mlir/Transforms/Passes.td b/mlir/include/mlir/Transforms/Passes.td
@@ -212,6 +212,10 @@ def SymbolDCE : Pass<"symbol-dce"> {
     information on `Symbols`.
   }];
   let constructor = "mlir::createSymbolDCEPass()";
+
+  let statistics = [
+    Statistic<"numDCE", "num-dce'd", "Number of symbols DCE'd">,
+  ];
 }
 
 def SymbolPrivatize : Pass<"symbol-privatize"> {

diff --git a/mlir/lib/Transforms/SymbolDCE.cpp b/mlir/lib/Transforms/SymbolDCE.cpp
@@ -62,8 +62,10 @@ void SymbolDCE::runOnOperation() {
       return;
     for (auto &block : nestedSymbolTable->getRegion(0)) {
       for (Operation &op : llvm::make_early_inc_range(block)) {
-        if (isa<SymbolOpInterface>(&op) && !liveSymbols.count(&op))
+        if (isa<SymbolOpInterface>(&op) && !liveSymbols.count(&op)) {
           op.erase();
+          ++numDCE;
+        }
       }
     }
   });