RequirementMachine: Add a couple of histograms

Author: slavapestov

lib/AST/RequirementMachine/GeneratingConformances.cpp

@@ -64,6 +64,7 @@
 #include "llvm/Support/Debug.h"
 #include "llvm/Support/raw_ostream.h"
 #include <algorithm>
+#include "RewriteContext.h"
 #include "RewriteSystem.h"
 
 using namespace swift;
@@ -681,8 +682,15 @@ void RewriteSystem::computeGeneratingConformances(
     }
   }
 
+  Context.ConformanceRulesHistogram.add(conformanceRules.size());
+
   computeCandidateConformancePaths(conformancePaths);
 
+  for (const auto &pair : conformancePaths) {
+    if (pair.second.size() > 1)
+      Context.GeneratingConformancesHistogram.add(pair.second.size());
+  }
+
   if (Debug.contains(DebugFlags::GeneratingConformances)) {
     llvm::dbgs() << "Initial set of equations:\n";
     for (const auto &pair : conformancePaths) {

lib/AST/RequirementMachine/RewriteContext.cpp

@@ -59,7 +59,9 @@ RewriteContext::RewriteContext(ASTContext &ctx)
       RuleTrieHistogram(16, /*Start=*/1),
       RuleTrieRootHistogram(16),
       PropertyTrieHistogram(16, /*Start=*/1),
-      PropertyTrieRootHistogram(16) {
+      PropertyTrieRootHistogram(16),
+      ConformanceRulesHistogram(16),
+      GeneratingConformancesHistogram(8, /*Start=*/2) {
   auto debugFlags = StringRef(ctx.LangOpts.DebugRequirementMachine);
   if (!debugFlags.empty())
     Debug = parseDebugFlags(debugFlags);
@@ -683,6 +685,10 @@ RewriteContext::~RewriteContext() {
     PropertyTrieHistogram.dump(llvm::dbgs());
     llvm::dbgs() << "\n* Property trie root fanout:\n";
     PropertyTrieRootHistogram.dump(llvm::dbgs());
+    llvm::dbgs() << "\n* Conformance rules:\n";
+    ConformanceRulesHistogram.dump(llvm::dbgs());
+    llvm::dbgs() << "\n* Generating conformance equations:\n";
+    GeneratingConformancesHistogram.dump(llvm::dbgs());
   }
 
   for (const auto &pair : Machines)

lib/AST/RequirementMachine/RewriteContext.h

@@ -130,6 +130,8 @@ class RewriteContext final {
   Histogram RuleTrieRootHistogram;
   Histogram PropertyTrieHistogram;
   Histogram PropertyTrieRootHistogram;
+  Histogram ConformanceRulesHistogram;
+  Histogram GeneratingConformancesHistogram;
 
   explicit RewriteContext(ASTContext &ctx);