SILGen: Record the declarations that need a #_hasSymbol query function generated.

The query functions will be fully defined by IRGen.

Author: tshortli

include/swift/SIL/SILModule.h

@@ -315,6 +315,9 @@ class SILModule {
   /// This is the set of undef values we've created, for uniquing purposes.
   llvm::DenseMap<SILType, SILUndef *> UndefValues;
 
+  /// The list of decls that require query functions for #_hasSymbol conditions.
+  llvm::SetVector<ValueDecl *> hasSymbolDecls;
+
   llvm::DenseMap<std::pair<Decl *, VarDecl *>, unsigned> fieldIndices;
   llvm::DenseMap<EnumElementDecl *, unsigned> enumCaseIndices;
 
@@ -700,6 +703,12 @@ class SILModule {
     return externallyVisible.count(decl) != 0;
   }
 
+  void addHasSymbolDecl(ValueDecl *decl) { hasSymbolDecls.insert(decl); }
+
+  ArrayRef<ValueDecl *> getHasSymbolDecls() {
+    return hasSymbolDecls.getArrayRef();
+  }
+
   using sil_global_iterator = GlobalListType::iterator;
   using sil_global_const_iterator = GlobalListType::const_iterator;
   GlobalListType &getSILGlobalList() { return silGlobals; }

lib/SILGen/SILGenDecl.cpp

@@ -1606,9 +1606,12 @@ void SILGenFunction::emitStmtCondition(StmtCondition Cond, JumpDest FalseDest,
       auto declRef = info->getReferencedDecl();
       assert(declRef);
 
-      auto queryFunc = declRef.getDecl()->getHasSymbolQueryDecl();
+      auto decl = declRef.getDecl();
+      getModule().addHasSymbolDecl(decl);
+
       SILFunction *silFn = SGM.getFunction(
-          SILDeclRef(queryFunc, SILDeclRef::Kind::Func), NotForDefinition);
+          SILDeclRef(decl->getHasSymbolQueryDecl(), SILDeclRef::Kind::Func),
+          NotForDefinition);
       SILValue fnRef = B.createFunctionRefFor(loc, silFn);
       booleanTestValue = B.createApply(loc, fnRef, {}, {});
       booleanTestValue = emitUnwrapIntegerResult(expr, booleanTestValue);