Merge pull request #69141 from tshortli/silgen-lazy-typecheck

SILGen: Introduce option to skip non-exportable declarations

Author: tshortli

include/swift/AST/Decl.h

@@ -957,6 +957,10 @@ class alignas(1 << DeclAlignInBits) Decl : public ASTAllocated<Decl> {
   unsigned getAttachedMacroDiscriminator(DeclBaseName macroName, MacroRole role,
                                          const CustomAttr *attr) const;
 
+  /// Determines if this declaration is exposed to clients of the module it is
+  /// defined in. For example, `public` declarations are exposed to clients.
+  bool isExposedToClients() const;
+
   /// Returns the innermost enclosing decl with an availability annotation.
   const Decl *getInnermostDeclWithAvailability() const;
 

include/swift/AST/DeclContext.h

@@ -877,15 +877,6 @@ class IterableDeclContext {
   /// The resulting list of members will be stable across translation units.
   ArrayRef<Decl *> getABIMembers() const;
 
-  using DeclsForLowering =
-      OptionalTransformRange<ArrayRef<Decl *>,
-                             AvailableDuringLoweringDeclFilter<Decl>>;
-
-  /// Get all of the members within this context that should be included when
-  /// lowering to SIL/IR, including any implicitly-synthesized members. The
-  /// decls returned by \c getABIMembers() are a superset of these decls.
-  DeclsForLowering getMembersForLowering() const;
-
   /// Get all of the members within this context, including any
   /// implicitly-synthesized members.
   ///

include/swift/SIL/SILVTableVisitor.h

@@ -105,6 +105,9 @@ template <class T> class SILVTableVisitor {
   }
 
   void maybeAddMember(Decl *member) {
+    if (!member->isAvailableDuringLowering())
+      return;
+
     if (isa<AccessorDecl>(member))
       /* handled as part of its storage */;
     else if (auto *fd = dyn_cast<FuncDecl>(member))
@@ -141,7 +144,7 @@ template <class T> class SILVTableVisitor {
     if (!theClass->hasKnownSwiftImplementation())
       return;
 
-    for (Decl *member : theClass->getMembersForLowering()) {
+    for (Decl *member : theClass->getABIMembers()) {
       maybeAddMember(member);
     }
   }

lib/AST/Decl.cpp

@@ -23,6 +23,7 @@
 #include "swift/AST/AccessScope.h"
 #include "swift/AST/Attr.h"
 #include "swift/AST/CaptureInfo.h"
+#include "swift/AST/DeclExportabilityVisitor.h"
 #include "swift/AST/DiagnosticEngine.h"
 #include "swift/AST/DiagnosticsSema.h"
 #include "swift/AST/ExistentialLayout.h"
@@ -492,6 +493,10 @@ unsigned Decl::getAttachedMacroDiscriminator(DeclBaseName macroName,
       macroName);
 }
 
+bool Decl::isExposedToClients() const {
+  return DeclExportabilityVisitor().visit(this);
+}
+
 const Decl *Decl::getInnermostDeclWithAvailability() const {
   if (getAttrs().hasAttribute<AvailableAttr>())
     return this;

lib/AST/DeclContext.cpp

@@ -914,12 +914,6 @@ ArrayRef<Decl *> IterableDeclContext::getABIMembers() const {
       ArrayRef<Decl *>());
 }
 
-IterableDeclContext::DeclsForLowering
-IterableDeclContext::getMembersForLowering() const {
-  return DeclsForLowering(getABIMembers(),
-                          AvailableDuringLoweringDeclFilter<Decl>());
-}
-
 ArrayRef<Decl *> IterableDeclContext::getAllMembers() const {
   ASTContext &ctx = getASTContext();
   return evaluateOrDefault(

lib/Frontend/Frontend.cpp

@@ -1425,11 +1425,6 @@ bool CompilerInstance::performParseAndResolveImportsOnly() {
 void CompilerInstance::performSema() {
   performParseAndResolveImportsOnly();
 
-  // Skip eager type checking. Instead, let later stages of compilation drive
-  // type checking as needed through request evaluation.
-  if (getASTContext().TypeCheckerOpts.EnableLazyTypecheck)
-    return;
-
   FrontendStatsTracer tracer(getStatsReporter(), "perform-sema");
 
   forEachFileToTypeCheck([&](SourceFile &SF) {

lib/SILGen/SILGen.cpp

@@ -797,8 +797,19 @@ bool SILGenModule::hasFunction(SILDeclRef constant) {
   return emittedFunctions.count(constant);
 }
 
-void SILGenModule::visit(Decl *D) {
+bool SILGenModule::shouldSkipDecl(Decl *D) {
   if (!D->isAvailableDuringLowering())
+    return true;
+
+  if (getASTContext().SILOpts.SkipNonExportableDecls &&
+      !D->isExposedToClients())
+    return true;
+
+  return false;
+}
+
+void SILGenModule::visit(Decl *D) {
+  if (shouldSkipDecl(D))
     return;
 
   ASTVisitor::visit(D);
@@ -1412,6 +1423,8 @@ void SILGenModule::emitAbstractFuncDecl(AbstractFunctionDecl *AFD) {
 }
 
 void SILGenModule::emitFunction(FuncDecl *fd) {
+  assert(!shouldSkipDecl(fd));
+
   Types.setCaptureTypeExpansionContext(SILDeclRef(fd), M);
 
   SILDeclRef::Loc decl = fd;

lib/SILGen/SILGen.h

@@ -264,6 +264,9 @@ class LLVM_LIBRARY_VISIBILITY SILGenModule : public ASTVisitor<SILGenModule> {
   // Visitors for top-level forms
   //===--------------------------------------------------------------------===//
 
+  /// Returns true if SILGen should be skipped for the given decl.
+  bool shouldSkipDecl(Decl *d);
+
   void visit(Decl *D);
 
   // These are either not allowed at global scope or don't require

lib/SILGen/SILGenTopLevel.cpp

@@ -294,11 +294,18 @@ void SILGenTopLevel::visitTopLevelCodeDecl(TopLevelCodeDecl *TD) {
 SILGenTopLevel::TypeVisitor::TypeVisitor(SILGenFunction &SGF) : SGF(SGF) {}
 
 void SILGenTopLevel::TypeVisitor::emit(IterableDeclContext *Ctx) {
-  for (auto *Member : Ctx->getMembersForLowering()) {
+  for (auto *Member : Ctx->getABIMembers()) {
     visit(Member);
   }
 }
 
+void SILGenTopLevel::TypeVisitor::visit(Decl *D) {
+  if (SGF.SGM.shouldSkipDecl(D))
+    return;
+
+  TypeMemberVisitor::visit(D);
+}
+
 void SILGenTopLevel::TypeVisitor::visitPatternBindingDecl(
     PatternBindingDecl *PD) {
   for (auto i : range(PD->getNumPatternEntries())) {

lib/SILGen/SILGenTopLevel.h

@@ -49,6 +49,7 @@ class SILGenTopLevel : public ASTVisitor<SILGenTopLevel> {
     /// Emit `mark_function_escape` SIL instructions into `SGF` for encountered
     /// escape points.
     TypeVisitor(SILGenFunction &SGF);
+    void visit(Decl *D);
     void visitDecl(Decl *D) {}
     void emit(IterableDeclContext *Ctx);
     virtual void visitPatternBindingDecl(PatternBindingDecl *PD);