async_execution and RRef helpers do not work together #41123
Labels
module: rpc
Related to RPC, distributed autograd, RRef, and distributed optimizer
triaged
This issue has been looked at a team member, and triaged and prioritized into an appropriate module
Uh oh!
There was an error while loading. Please reload this page.
The following code does not work
This is because the
_local_invoke
function ignored the attribute installed by the decorator. To fix this, we will need to have another version of_local_invoke
which is decorated by@rpc.functions.async_execution
and returns theFuture
object from the wrapped function.pytorch/torch/distributed/rpc/rref_proxy.py
Lines 8 to 13 in 03eec07
cc @pietern @mrshenli @pritamdamania87 @zhaojuanmao @satgera @gqchen @aazzolini @rohan-varma @xush6528 @jjlilley @osalpekar @jiayisuse
The text was updated successfully, but these errors were encountered: