Merge pull request #37139 from aschwaighofer/disable_swift_function_merging_for_async

Disable swift function merging for swift async functions

Author: aschwaighofer

lib/LLVMPasses/LLVMMergeFunctions.cpp

@@ -664,6 +664,9 @@ static bool isEligibleFunction(Function *F) {
 
   if (F->getFunctionType()->isVarArg())
     return false;
+
+  if (F->getCallingConv() == CallingConv::SwiftTail)
+    return false;
 
   unsigned Benefit = getBenefit(F);
   if (Benefit < FunctionMergeThreshold)

test/Concurrency/Runtime/checked_continuation.swift

@@ -41,6 +41,30 @@ struct TestError: Error {}
         }
         await task.get()
       }
+
+      tests.test("test withCheckedThrowingContinuation") {
+        let task2 = detach {
+          do {
+            let x: Int = try await withCheckedThrowingContinuation { c in
+              c.resume(returning: 17)
+            }
+            expectEqual(17, x)
+          } catch {
+          }
+        }
+
+        let task = detach {
+          do {
+            let x: Int = try await withCheckedThrowingContinuation { c in
+              c.resume(returning: 17)
+            }
+            expectEqual(17, x)
+          } catch {
+          }
+        }
+        await task.get()
+        await task2.get()
+      }
     }
 
     await runAllTestsAsync()