[Coroutine] Check indirect uses of alloca when checking lifetime info
authorXun Li <lxfind@gmail.com>
Thu, 25 Feb 2021 02:29:23 +0000 (18:29 -0800)
committerXun Li <lxfind@gmail.com>
Thu, 25 Feb 2021 02:29:23 +0000 (18:29 -0800)
commitc38000a9fb2cd64ad6e72939643e2c0fa4972690
treebaff96219d35cdf1dafb393230af4d27da8cf3ea
parentb950de5c13ef2171c0457a413c3b06aa31047938
[Coroutine] Check indirect uses of alloca when checking lifetime info

In the existing logic, we look at the lifetime.start marker of each alloca, and check all uses of the alloca, to see if any pair of the lifetime marker and an use of alloca crosses suspension point.
This approach is unfortunately incorrect. An use of alloca does not need to be a direct use, but can be an indirect use through alias.
Only checking direct uses can miss cases where indirect uses are crossing suspension point.
This can be demonstrated in the newly added test case 007.
In the test case, both x and y are only directly used prior to suspend, but they are captured into an alias, merged through a PHINode (so they couldn't be materialized), and used after CoroSuspend.
If we only check whether the lifetime starts cross suspension points with direct uses, we will put the allocas to the stack, and then capture their addresses in the frame.

Instead of fixing it in D96441 and D96566, this patch takes a different approach which I think is better.
We still checks the lifetime info in the same way as before, but with two differences:
1. The collection of liftime.start is moved into AllocaUseVisitor to make the logic more concentrated.
2. When looking at lifetime.start and use pairs, we not only checks the direct uses as before, but in this patch we check all uses collected by AllocaUseVisitor, which would include all indirect uses through alias. This will make the analysis more accurate without throwing away the lifetime optimization.

Differential Revision: https://reviews.llvm.org/D96922
llvm/lib/Transforms/Coroutines/CoroFrame.cpp
llvm/test/Transforms/Coroutines/coro-alloca-07.ll [new file with mode: 0644]
llvm/test/Transforms/Coroutines/coro-alloca-08.ll [new file with mode: 0644]