[AutoDiff] Move dumpActivityInfo to DifferentiableActivityInfo class

Author: bgogul

include/swift/SILOptimizer/Analysis/DifferentiableActivityAnalysis.h

@@ -206,6 +206,16 @@ class DifferentiableActivityInfo {
   /// Returns the activity of the given value for the given `SILAutoDiffIndices`
   /// (parameter indices and result index).
   Activity getActivity(SILValue value, const SILAutoDiffIndices &indices) const;
+
+  /// Prints activity information for the `indices` of the given `value`.
+  void dumpActivityInfo(SILValue value,
+                        const SILAutoDiffIndices &indices,
+                        llvm::raw_ostream &s = llvm::dbgs()) const;
+
+  /// Prints activity information for the `indices` of the given `fn`.
+  void dumpActivityInfo(SILFunction &fn, SILAutoDiffIndices indices,
+                             llvm::raw_ostream &s = llvm::dbgs()) const;
+
 };
 
 class DifferentiableActivityCollection {

lib/SILOptimizer/Analysis/DifferentiableActivityAnalysis.cpp

@@ -421,3 +421,32 @@ Activity DifferentiableActivityInfo::getActivity(
     activity |= ActivityFlags::Useful;
   return activity;
 }
+
+void DifferentiableActivityInfo::dumpActivityInfo(
+    SILValue value, const SILAutoDiffIndices &indices,
+    llvm::raw_ostream &s) const {
+  s << '[';
+  auto activity = getActivity(value, indices);
+  switch (activity.toRaw()) {
+  case 0: s << "NONE"; break;
+  case (unsigned)ActivityFlags::Varied: s << "VARIED"; break;
+  case (unsigned)ActivityFlags::Useful: s << "USEFUL"; break;
+  case (unsigned)ActivityFlags::Active: s << "ACTIVE"; break;
+  }
+  s << "] " << value;
+}
+
+void DifferentiableActivityInfo::dumpActivityInfo(SILFunction &fn,
+                                                  SILAutoDiffIndices indices,
+                                                  llvm::raw_ostream &s) const {
+  s << "Activity info for " << fn.getName() << " at " << indices << '\n';
+  for (auto &bb : fn) {
+    s << "bb" << bb.getDebugID() << ":\n";
+    for (auto *arg : bb.getArguments())
+      dumpActivityInfo(arg, indices, s);
+    for (auto &inst : bb)
+      for (auto res : inst.getResults())
+        dumpActivityInfo(res, indices, s);
+    s << '\n';
+  }
+}

lib/SILOptimizer/Mandatory/Differentiation.cpp

@@ -262,36 +262,6 @@ class DifferentiationTransformer {
 
 } // end anonymous namespace
 
-static void dumpActivityInfo(SILValue value,
-                             const SILAutoDiffIndices &indices,
-                             const DifferentiableActivityInfo &activityInfo,
-                             llvm::raw_ostream &s = llvm::dbgs()) {
-  s << '[';
-  auto activity = activityInfo.getActivity(value, indices);
-  switch (activity.toRaw()) {
-  case 0: s << "NONE"; break;
-  case (unsigned)ActivityFlags::Varied: s << "VARIED"; break;
-  case (unsigned)ActivityFlags::Useful: s << "USEFUL"; break;
-  case (unsigned)ActivityFlags::Active: s << "ACTIVE"; break;
-  }
-  s << "] " << value;
-}
-
-static void dumpActivityInfo(SILFunction &fn, SILAutoDiffIndices indices,
-                             const DifferentiableActivityInfo &activityInfo,
-                             llvm::raw_ostream &s = llvm::dbgs()) {
-  s << "Activity info for " << fn.getName() << " at " << indices << '\n';
-  for (auto &bb : fn) {
-    s << "bb" << bb.getDebugID() << ":\n";
-    for (auto *arg : bb.getArguments())
-      dumpActivityInfo(arg, indices, activityInfo, s);
-    for (auto &inst : bb)
-      for (auto res : inst.getResults())
-        dumpActivityInfo(res, indices, activityInfo, s);
-    s << '\n';
-  }
-}
-
 /// If the original function doesn't have a return, it cannot be differentiated.
 /// Returns true if error is emitted.
 static bool diagnoseNoReturn(ADContext &context, SILFunction *original,
@@ -1374,7 +1344,7 @@ class VJPEmitter final
         vjp->getLoweredFunctionType()->getSubstGenericSignature(),
         AutoDiffDerivativeFunctionKind::VJP);
     LLVM_DEBUG(
-        dumpActivityInfo(*original, indices, activityInfo, getADDebugStream()));
+        activityInfo.dumpActivityInfo(*original, indices, getADDebugStream()));
     return activityInfo;
   }
 
@@ -2128,7 +2098,7 @@ class JVPEmitter final
         jvp->getLoweredFunctionType()->getSubstGenericSignature(),
         AutoDiffDerivativeFunctionKind::JVP);
     LLVM_DEBUG(
-        dumpActivityInfo(*original, indices, activityInfo, getADDebugStream()));
+        activityInfo.dumpActivityInfo(*original, indices, getADDebugStream()));
     return activityInfo;
   }