address comment

Author: Marc Rasi

include/swift/AST/PrettyStackTrace.h

@@ -192,21 +192,19 @@ class PrettyStackTraceSelector : public llvm::PrettyStackTraceEntry {
 /// specific differentiability witness.
 class PrettyStackTraceDifferentiabilityWitness
     : public llvm::PrettyStackTraceEntry {
-  ASTContext &Context;
   const SILDifferentiabilityWitnessKey Key;
   const char *Action;
 
 public:
   PrettyStackTraceDifferentiabilityWitness(
-      ASTContext &C, const char *action,
-      const SILDifferentiabilityWitnessKey key)
-      : Context(C), Key(key), Action(action) {}
+      const char *action, const SILDifferentiabilityWitnessKey key)
+      : Key(key), Action(action) {}
   virtual void print(llvm::raw_ostream &OS) const;
 };
 
 void printDifferentiabilityWitnessDescription(
     llvm::raw_ostream &out, const SILDifferentiabilityWitnessKey key,
-    ASTContext &Context, bool addNewline = true);
+    bool addNewline = true);
 // SWIFT_ENABLE_TENSORFLOW END
 
 } // end namespace swift

lib/AST/PrettyStackTrace.cpp

@@ -267,12 +267,12 @@ void PrettyStackTraceSelector::print(llvm::raw_ostream &out) const {
 void PrettyStackTraceDifferentiabilityWitness::print(
     llvm::raw_ostream &out) const {
   out << "While " << Action << ' ';
-  printDifferentiabilityWitnessDescription(out, Key, Context);
+  printDifferentiabilityWitnessDescription(out, Key);
 }
 
 void swift::printDifferentiabilityWitnessDescription(
     llvm::raw_ostream &out, const SILDifferentiabilityWitnessKey key,
-    ASTContext &Context, bool addNewline) {
+    bool addNewline) {
   out << key.first << " ";
   key.second.print(out);
   if (addNewline)

lib/IRGen/GenProto.cpp

@@ -2170,7 +2170,7 @@ void IRGenModule::emitSILWitnessTable(SILWitnessTable *wt) {
 void IRGenModule::emitSILDifferentiabilityWitness(
     SILDifferentiabilityWitness *dw) {
   PrettyStackTraceDifferentiabilityWitness _st(
-      Context, "emitting differentiability witness for", dw->getKey());
+      "emitting differentiability witness for", dw->getKey());
 
   // Don't emit declarations.
   if (dw->isDeclaration())