Use getSourceFile function

Author: kovdan01

lib/SILOptimizer/Differentiation/ADContext.cpp

@@ -62,10 +62,8 @@ ADContext::ADContext(SILModuleTransform &transform)
 
 /// Get the source file for the given `SILFunction`.
 static SourceFile &getSourceFile(SILFunction *f) {
-  if (f->hasLocation())
-    if (auto *declContext = f->getLocation().getAsDeclContext())
-      if (auto *parentSourceFile = declContext->getParentSourceFile())
-        return *parentSourceFile;
+  if (SourceFile *file = f->getSourceFile())
+    return *file;
   for (auto *file : f->getModule().getSwiftModule()->getFiles())
     if (auto *sourceFile = dyn_cast<SourceFile>(file))
       return *sourceFile;