Merge pull request swiftlang#87088 from felipepiovezan/felipe/mark_all_async_funclets

[EmitAsyncEntryReturnMetadata] Emit sections for all async funclets

Author: felipepiovezan

lib/LLVMPasses/LLVMEmitAsyncEntryReturnMetadata.cpp

@@ -18,6 +18,7 @@ PreservedAnalyses AsyncEntryReturnMetadataPass::run(Module &M,
 
   SmallVector<llvm::Function *, 16> asyncEntries;
   SmallVector<llvm::Function *, 16> asyncReturns;
+  SmallVector<llvm::Function *, 16> asyncContinuations;
   for (auto &F : M) {
     if (F.isDeclaration())
       continue;
@@ -26,6 +27,8 @@ PreservedAnalyses AsyncEntryReturnMetadataPass::run(Module &M,
       asyncEntries.push_back(&F);
     if (F.hasFnAttribute("async_ret"))
       asyncReturns.push_back(&F);
+    if (F.hasFnAttribute("async_continuation"))
+      asyncContinuations.push_back(&F);
   }
 
   auto &ctxt = M.getContext();
@@ -83,6 +86,8 @@ PreservedAnalyses AsyncEntryReturnMetadataPass::run(Module &M,
   addSection("__TEXT,__swift_as_ret, coalesced, no_dead_strip",
              "__swift_async_ret_functlets",
              asyncReturns);
+  addSection("__TEXT,__swift_as_cont, coalesced, no_dead_strip",
+             "__swift_async_cont_functlets", asyncContinuations);
 
   if (!changed)
     return PreservedAnalyses::all();

test/IRGen/async_frame_entry_return_metadata.swift

@@ -9,14 +9,24 @@
 // ENABLED: @__swift_async_entry_functlets = internal constant [2 x i32] [i32 trunc (i64 sub (i64 ptrtoint (ptr @"$s5async6calleeyyYaF" to i64), i64 ptrtoint (ptr @__swift_async_entry_functlets to i64)) to i32), i32 trunc (i64 sub (i64 ptrtoint (ptr @"$s5async6callerySiSbYaF" to i64), i64 ptrtoint (ptr getelementptr inbounds ([2 x i32], ptr @__swift_async_entry_functlets, i32 0, i32 1) to i64)) to i32)], section "__TEXT,__swift_as_entry, coalesced, no_dead_strip", no_sanitize_address, align 4
 // ENABLED: @__swift_async_ret_functlets = internal constant [1 x i32] [i32 trunc (i64 sub (i64 ptrtoint (ptr @"$s5async6callerySiSbYaFTQ1_" to i64), i64 ptrtoint (ptr @__swift_async_ret_functlets to i64)) to i32)], section "__TEXT,__swift_as_ret, coalesced, no_dead_strip", no_sanitize_address, align 4
 
-// ENABLED: define{{.*}} swifttailcc void @"$s5async6callerySiSbYaF"{{.*}} [[CALLER_FUNCLET_ATTRS:#[0-9]+]]
-// ENABLED: define{{.*}} internal swifttailcc void @"$s5async6callerySiSbYaFTY0_"{{.*}} [[CALLER_FUNCLET_ATTRS2:#[0-9]+]]
+// ENABLED: @__swift_async_cont_functlets = internal constant [3 x i32]
+// ENABLED-SAME: ptr @"$s5async6calleeyyYaFTY0_"
+// ENABLED-SAME: ptr @"$s5async6callerySiSbYaFTY0_"
+// ENABLED-SAME: ptr @"$s5async6callerySiSbYaFTY2_"
+// ENABLED-SAME: section "__TEXT,__swift_as_cont, coalesced, no_dead_strip", no_sanitize_address, align 4
 
-// ENABLED: attributes [[CALLER_FUNCLET_ATTRS2]] = { {{.*}}noinline
-// ENABLED: attributes [[CALLER_FUNCLET_ATTRS]] = { {{.*}}noinline
+// ENABLED: define{{.*}} swifttailcc void @"$s5async6callerySiSbYaF"{{.*}} [[ENTRY_ATTR:#[0-9]+]]
+// ENABLED: define{{.*}} internal swifttailcc void @"$s5async6callerySiSbYaFTY0_"{{.*}} [[CONTINUATION_ATTR:#[0-9]+]]
+// ENABLED: define{{.*}} internal swifttailcc void @"$s5async6callerySiSbYaFTQ1_"{{.*}} [[RET_ATTR:#[0-9]+]]
+// ENABLED: define{{.*}} internal swifttailcc void @"$s5async6callerySiSbYaFTY2_"{{.*}} [[CONTINUATION_ATTR]]
+
+// ENABLED: attributes [[ENTRY_ATTR]] = { {{.*}} "async_entry"
+// ENABLED: attributes [[CONTINUATION_ATTR]] = { {{.*}} "async_continuation"
+// ENABLED: attributes [[RET_ATTR]] = { {{.*}} "async_ret"
 
 // DISABLED-NOT: @__swift_async_entry_functlets
 // DISABLED-NOT: @__swift_async_ret_functlets
+// DISABLED-NOT: @__swift_async_cont_functlets
 // DISABLED-NOT: s5async6calleeyyYaF.0
 // DISABLED-NOT: s5async6callerySiSbYaF.0