Merge pull request #34681 from rxwei/71191415-debug-scope

Author: swift-ci

lib/AST/Decl.cpp

@@ -654,7 +654,9 @@ static_assert(sizeof(checkSourceLocType(&ID##Decl::getLoc)) == 2, \
   // When the decl is context-free, we should get loc from source buffer.
   if (!getDeclContext())
     return getLocFromSource();
-  auto *File = cast<FileUnit>(getDeclContext()->getModuleScopeContext());
+  FileUnit *File = dyn_cast<FileUnit>(getDeclContext()->getModuleScopeContext());
+  if (!File)
+    return getLocFromSource();
   switch(File->getKind()) {
   case FileUnitKind::Source:
     return getLocFromSource();

test/AutoDiff/compiler_crashers_fixed/rdar71191415-nested-differentiation-of-extension-method-optimized.swift

@@ -0,0 +1,33 @@
+// RUN: %target-build-swift -O %s
+
+// rdar://71191415
+
+import _Differentiation
+
+protocol P {
+    @differentiable
+    func req(_ input: Float) -> Float
+}
+
+extension P {
+    @differentiable
+    func foo(_ input: Float) -> Float {
+        return req(input)
+    }
+}
+
+struct Dummy: P {
+    @differentiable
+    func req(_ input: Float) -> Float {
+        input
+    }
+}
+
+struct DummyComposition: P {
+    var layer = Dummy()
+
+    @differentiable
+    func req(_ input: Float) -> Float {
+        layer.foo(input)
+    }
+}