[AutoDiff] NFC: Highlights original function name in diagnostic (swiftlang#36259)

Small follow up of [PR-36128](swiftlang#36128)
to be consistent with other diagnostics emitted when type-checking
`@derivative` and `@transpose` attributes.

Author: vguerra

lib/Sema/TypeCheckAttr.cpp

@@ -4843,10 +4843,12 @@ static bool typeCheckDerivativeAttr(ASTContext &Ctx, Decl *D,
   // Diagnose if original function and derivative differ in terms of static declaration.
   if (!compatibleStaticDecls()) {
     bool derivativeMustBeStatic = !derivative->isStatic();
-    diags.diagnose(attr->getOriginalFunctionName().Loc.getBaseNameLoc(),
-                   diag::derivative_attr_static_method_mismatch_original,
-                   originalAFD->getName(), derivative->getName(),
-                   derivativeMustBeStatic);
+    diags
+        .diagnose(attr->getOriginalFunctionName().Loc.getBaseNameLoc(),
+                  diag::derivative_attr_static_method_mismatch_original,
+                  originalAFD->getName(), derivative->getName(),
+                  derivativeMustBeStatic)
+        .highlight(attr->getOriginalFunctionName().Loc.getSourceRange());
     diags.diagnose(originalAFD->getNameLoc(),
                    diag::derivative_attr_static_method_mismatch_original_note,
                    originalAFD->getName(), derivativeMustBeStatic);
@@ -5393,7 +5395,8 @@ void AttributeChecker::visitTransposeAttr(TransposeAttr *attr) {
     diagnose(attr->getOriginalFunctionName().Loc.getBaseNameLoc(),
              diag::transpose_attr_static_method_mismatch_original,
              originalAFD->getName(), transpose->getName(),
-             transposeMustBeStatic);
+             transposeMustBeStatic)
+        .highlight(attr->getOriginalFunctionName().Loc.getSourceRange());
     diagnose(originalAFD->getNameLoc(),
              diag::transpose_attr_static_method_mismatch_original_note,
              originalAFD->getName(), transposeMustBeStatic);