emit statistics

Author: jroelofs

llvm/lib/Transforms/Scalar/LowerMatrixIntrinsics.cpp

@@ -22,6 +22,7 @@
 #include "llvm/ADT/ScopeExit.h"
 #include "llvm/ADT/SmallSet.h"
 #include "llvm/ADT/SmallVector.h"
+#include "llvm/ADT/Statistic.h"
 #include "llvm/Analysis/AliasAnalysis.h"
 #include "llvm/Analysis/DomTreeUpdater.h"
 #include "llvm/Analysis/LoopInfo.h"
@@ -53,6 +54,10 @@ using namespace PatternMatch;
 
 #define DEBUG_TYPE "lower-matrix-intrinsics"
 
+STATISTIC(FlattenedMatrices, "Number of matrix flattenings");
+STATISTIC(ReshapedMatrices, "Number of matrix reshapes");
+STATISTIC(SplitMatrices, "Number of matrix splits");
+
 static cl::opt<bool>
     FuseMatrix("fuse-matrix", cl::init(true), cl::Hidden,
                cl::desc("Enable/disable fusing matrix instructions."));
@@ -597,11 +602,13 @@ class LowerMatrixIntrinsics {
         dbgs() << "matrix reshape from " << Found->second.shape() << " to "
                << SI << " using at least " << SplitVecs.size()
                << " shuffles on behalf of:\n" << *Inst << '\n';
+        ReshapedMatrices++;
       } else if (!ShapeMap.contains(MatrixVal)) {
         dbgs() << "splitting a " << SI << " matrix with " << SplitVecs.size()
                << " shuffles beacuse we do not have a shape-aware lowering for "
                   "its def:\n"
                << *Inst << '\n';
+        SplitMatrices++;
       } else {
         // The ShapeMap has it, so it's a case where we're being lowered
         // before the def, and we expect that InstCombine will clean things up
@@ -1428,6 +1435,7 @@ class LowerMatrixIntrinsics {
                 << "\nbecause we do not have a shape-aware lowering for its "
                    "user:\n"
                 << *User << '\n';);
+        FlattenedMatrices++;
       }
       U.set(Flattened);
     }