Added Pass Statistics

Author: Alwaysproblem

mlir/mhlo-pass-example/Pow2/main.cpp

@@ -94,6 +94,9 @@ int main(int argc, char *argv[]) {
     if (mlir::failed(mlir::applyPassManagerCLOptions(pm)))
       return 4;
 
+    // Enable the statistics (only works on debug mode)
+    // pm.enableStatistics();
+
     // Add a run of the canonicalizer to optimize the mlir module.
     // pm.addNestedPass<mlir::func::FuncOp>(mlir::createCanonicalizerPass());
     pm.addNestedPass<mlir::func::FuncOp>(

mlir/mhlo-pass-example/Pow2/mlir/Pow2.cpp

@@ -89,6 +89,15 @@ namespace {
 struct SubstitutePow2Pass
     : public PassWrapper<SubstitutePow2Pass, OperationPass<func::FuncOp>> {
   MLIR_DEFINE_EXPLICIT_INTERNAL_INLINE_TYPE_ID(SubstitutePow2Pass)
+  // Here we can add the statistic (the statistic is only works on debug mode)
+  // Since the statistic has the atomic value, we can't use the default copy
+  // constructor and assignment operator.
+  // SubstitutePow2Pass() = default;
+  // SubstitutePow2Pass(const SubstitutePow2Pass &other) {
+  //   this->statistic = other.statistic.getValue();
+  // };
+  // mlir::Pass::Statistic statistic{this, "example-statistic", "An example
+  // statistic"};
 
   void runOnOperation() final;
 };
@@ -100,6 +109,8 @@ void SubstitutePow2Pass::runOnOperation() {
   patterns.add<Pow2OptPattern>(&getContext());
   if (failed(applyPatternsAndFoldGreedily(op, std::move(patterns))))
     signalPassFailure();
+  // Here we can add the statistic (the statistic is only works on debug mode)
+  // statistic++;
 }
 
 namespace {

mlir/mhlo-pass-example/Pow2/mlir/transform/pass.td

@@ -20,8 +20,8 @@ def Pow2Pass : Pass<"pow-2", "func::FuncOp"> {
 
   // Specify any statistics.
   let statistics = [
-    // Statistic</*C++ variable name=*/"statistic",
-    //           /*display name*/"example-statistic",
-    //           /*description*/"An example statistic">
+    Statistic</*C++ variable name=*/"statistic",
+              /*display name*/"example-statistic",
+              /*description*/"An example statistic">
   ];
 }