From 5bf567f5f6ad6250ef0e3b5101fafe0c0bb5e98f Mon Sep 17 00:00:00 2001
From: Victoria Mitchell <victoria_m@apple.com>
Date: Fri, 11 Dec 2020 08:30:32 -0700
Subject: [PATCH] remove DiagPrinter directly from the CompilerInvocation

---
 tools/driver/swift_symbolgraph_extract_main.cpp | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tools/driver/swift_symbolgraph_extract_main.cpp b/tools/driver/swift_symbolgraph_extract_main.cpp
index fab2ba47a6c..f8829a0aea7 100644
--- a/tools/driver/swift_symbolgraph_extract_main.cpp
+++ b/tools/driver/swift_symbolgraph_extract_main.cpp
@@ -250,7 +250,7 @@ int swift_symbolgraph_extract_main(ArrayRef<const char *> Args, const char *Argv
   // since some SDK modules can cause errors in the getModuleByName() call. The call
   // itself will properly return nullptr after this failure, so for our purposes we
   // don't need to print these errors.
-  CI.getDiags().removeConsumer(DiagPrinter);
+  CI.removeDiagnosticConsumer(&DiagPrinter);
   
   for (const auto &ModuleName : VisibleModuleNames) {
     if (ModuleName.str().startswith("_")) {