[flang] Set proper source location for the main function.

[vzakhari]

Summary

Take the source position for the anonymous program from its scope.
If the first evaluation is a construct or directive, then it has
null source position.

Author: vdonaldson

flang/lib/Lower/PFTBuilder.cpp

if (auto *par = parent.getIf<FunctionLikeUnit>())
     return !par->hostAssociations.empty();
   return false;
 }
 
 parser::CharBlock
 Fortran::lower::pft::FunctionLikeUnit::getStartingSourceLoc() const {
   if (beginStmt)
     return stmtSourceLoc(*beginStmt);
-  if (!evaluationList.empty())
-    return evaluationList.front().position;
-  return stmtSourceLoc(endStmt);
+  return scope->sourceRange();
 }
 
 //===----------------------------------------------------------------------===//
 // ModuleLikeUnit implementation
 //===----------------------------------------------------------------------===//
 
 Fortran::lower::pft::ModuleLikeUnit::ModuleLikeUnit(
     const parser::Module &m, const lower::pft::PftNode &parent)

flang/test/Lower/main_location.f90

+! RUN: split-file %s %t
+! RUN: bbc %t/test1.f90 -o - --emit-fir --mlir-print-debuginfo | FileCheck %s --check-prefix=TEST1
+! RUN: bbc %t/test2.f90 -o - --emit-fir --mlir-print-debuginfo | FileCheck %s --check-prefix=TEST2
+
+! Check that the missing optional program-stmt (R1401)
+! does not result in unknown source location of the corresponding
+! function.
+
+!--- test1.f90
+if (.false.) then
+endif
+end
+
+! TEST1: func.func @_QQmain() {
+! TEST1-NEXT: return loc("{{.*}}test1.f90":3:1)
+! TEST1-NEXT: } loc("{{.*}}test1.f90":1:1)
+
+!--- test2.f90
+!!! keep me here
+if (.true.) then
+endif
+end program
+
+! TEST2: func.func @_QQmain() {
+! TEST2-NEXT: return loc("{{.*}}test2.f90":4:1)
+! TEST2-NEXT: } loc("{{.*}}test2.f90":2:1)