[AutoDiff] NFC: Clean up print/dump utility functions. (#27772)

All `dump` functions now print a newline at the end.

Author: dan-zheng

include/swift/AST/AutoDiff.h

@@ -243,12 +243,9 @@ struct SILAutoDiffIndices {
            parameters->contains(parameterIndex);
   }
 
-  void print(llvm::raw_ostream &s = llvm::outs()) const {
-    s << "(source=" << source << " parameters=(";
-    interleave(parameters->getIndices(),
-               [&s](unsigned p) { s << p; }, [&s]{ s << ' '; });
-    s << "))";
-  }
+  void print(llvm::raw_ostream &s = llvm::outs()) const;
+  LLVM_ATTRIBUTE_DEPRECATED(void dump() const LLVM_ATTRIBUTE_USED,
+                            "only for use within the debugger");
 
   std::string mangle() const {
     std::string result = "src_" + llvm::utostr(source) + "_wrt_";
@@ -279,6 +276,10 @@ struct AutoDiffConfig {
                               GenericSignatureImpl *derivativeGenericSignature)
       : parameterIndices(parameterIndices), resultIndices(resultIndices),
         derivativeGenericSignature(derivativeGenericSignature) {}
+
+  void print(llvm::raw_ostream &s = llvm::outs()) const;
+  LLVM_ATTRIBUTE_DEPRECATED(void dump() const LLVM_ATTRIBUTE_USED,
+                            "only for use within the debugger");
 };
 
 /// In conjunction with the original function declaration, identifies an

include/swift/AST/IndexSubset.h

@@ -201,19 +201,10 @@ class IndexSubset : public llvm::FoldingSetNode {
       id.AddInteger(index);
   }
 
-  void print(llvm::raw_ostream &s = llvm::outs()) const {
-    s << '{';
-    interleave(range(capacity), [this, &s](unsigned i) { s << contains(i); },
-               [&s] { s << ", "; });
-    s << '}';
-  }
-
-  void dump(llvm::raw_ostream &s = llvm::errs()) const {
-    s << "(index_subset capacity=" << capacity << " indices=(";
-    interleave(getIndices(), [&s](unsigned i) { s << i; },
-               [&s] { s << ", "; });
-    s << "))";
-  }
+  void print(llvm::raw_ostream &s = llvm::outs()) const;
+  LLVM_ATTRIBUTE_DEPRECATED(void dump(llvm::raw_ostream &s = llvm::errs())
+                                const LLVM_ATTRIBUTE_USED,
+                            "only for use within the debugger");
 
   int findNext(int startIndex) const;
   int findFirst() const { return findNext(-1); }

lib/AST/AutoDiff.cpp

@@ -88,6 +88,35 @@ LinearDifferentiableFunctionTypeComponent(StringRef string) {
   rawValue = *result;
 }
 
+void SILAutoDiffIndices::print(llvm::raw_ostream &s) const {
+  s << "(source=" << source << " parameters=(";
+  interleave(parameters->getIndices(),
+             [&s](unsigned p) { s << p; }, [&s]{ s << ' '; });
+  s << "))";
+}
+
+void SILAutoDiffIndices::dump() const {
+  print(llvm::errs());
+  llvm::errs() << '\n';
+}
+
+void AutoDiffConfig::print(llvm::raw_ostream &s) const {
+  s << "(parameters=";
+  parameterIndices->print(s);
+  s << " results=";
+  resultIndices->print(s);
+  if (derivativeGenericSignature) {
+    s << " where=";
+    derivativeGenericSignature->print(s);
+  }
+  s << ')';
+}
+
+void AutoDiffConfig::dump() const {
+  print(llvm::errs());
+  llvm::errs() << '\n';
+}
+
 // TODO(TF-874): This helper is inefficient and should be removed. Unwrapping at
 // most once (for curried method types) is sufficient.
 static void unwrapCurryLevels(AnyFunctionType *fnTy,

lib/AST/IndexSubset.cpp

@@ -79,6 +79,20 @@ IndexSubset *IndexSubset::extendingCapacity(
   return IndexSubset::get(ctx, indices);
 }
 
+void IndexSubset::print(llvm::raw_ostream &s) const {
+  s << '{';
+  interleave(range(capacity), [this, &s](unsigned i) { s << contains(i); },
+             [&s] { s << ", "; });
+  s << '}';
+}
+
+void IndexSubset::dump(llvm::raw_ostream &s) const {
+  s << "(index_subset capacity=" << capacity << " indices=(";
+  interleave(getIndices(), [&s](unsigned i) { s << i; },
+             [&s] { s << ", "; });
+  s << "))\n";
+}
+
 int IndexSubset::findNext(int startIndex) const {
   assert(startIndex < (int)capacity && "Start index cannot be past the end");
   unsigned bitWordIndex = 0, offset = 0;