Address PR comments on return omission.

Author: nate-chandler

lib/Parse/ParseDecl.cpp

@@ -5649,9 +5649,7 @@ void Parser::parseAbstractFunctionBody(AbstractFunctionDecl *AFD) {
     if (!Body.hasCodeCompletion() && BS->getNumElements() == 1) {
       auto Element = BS->getElement(0);
       if (auto *stmt = Element.dyn_cast<Stmt *>()) {
-        auto kind = AFD->getKind();
-        if (kind == DeclKind::Var || kind == DeclKind::Subscript ||
-            kind == DeclKind::Func ) {
+        if (isa<FuncDecl>(AFD)) {
           if (auto *returnStmt = dyn_cast<ReturnStmt>(stmt)) {
             if (!returnStmt->hasResult()) {
               auto returnExpr = TupleExpr::createEmpty(Context,

lib/SILOptimizer/Mandatory/DataflowDiagnostics.cpp

@@ -63,7 +63,7 @@ static void diagnoseMissingReturn(const UnreachableInst *UI,
   if (numElements > 0) {
     auto element = BS->getElement(numElements - 1);
     if (auto expr = element.dyn_cast<Expr *>()) {
-      if (expr->getType()->getCanonicalType() == ResTy->getCanonicalType()) {
+      if (expr->getType()->isEqual(ResTy)) {
         Context.Diags.diagnose(
           expr->getStartLoc(),
           diag::missing_return_last_expr, ResTy,