[llvm] [coro][CoroSplit] Use `llvm.lifetime.end` to compute putting objects on the frame vs the stack (PR #90265)

Chuanqi Xu via llvm-commits llvm-commits at lists.llvm.org
Mon Apr 29 18:50:37 PDT 2024


================
@@ -1614,6 +1621,24 @@ struct AllocaUseVisitor : PtrUseVisitor<AllocaUseVisitor> {
       // suspend point between lifetime markers. This should also cover the
       // case of a single lifetime.start intrinsic in a loop with suspend point.
       if (PI.isEscaped()) {
+        // If there is no explicit lifetime.end, then assume the address can
+        // cross suspension points.
+        if (LifetimeEndBBs.empty())
+          return true;
+
+        // If there is a path from a lifetime.start to a suspend without a
+        // corresponding lifetime.end, then the alloca's lifetime persists
+        // beyond that suspension point and the alloca must go on the frame.
+        for (AnyCoroSuspendInst *SuspendInst : CoroShape.CoroSuspends) {
+          SmallVector<BasicBlock *> LifetimeStartsBB;
+          for (IntrinsicInst *II : LifetimeStarts)
+            LifetimeStartsBB.push_back(II->getParent());
+          if (isPotentiallyReachableFromMany(LifetimeStartsBB,
----------------
ChuanqiXu9 wrote:

>  "Is there a path from BasicBlock A to BasicBlock B without going through BasicBlocks { C, D, E, ...}. The current logic caches whether a path may go through a suspend point - it doesn't answer the reverse question

I don't understand this. The reverse form of it should be: there is not a path may go through a suspend point. Then it looks the same with the previous one.

https://github.com/llvm/llvm-project/pull/90265


More information about the llvm-commits mailing list