Skip to content

Commit

Permalink
Update on "Fix CUDA RPC Stream Synchronization"
Browse files Browse the repository at this point in the history
When converting RPC Message into Python objects, we were not using
a CUDAFuture for the chained Future. As a result, the streams are
not synchronized when calling `rpc_async(...).wait()`. This commit
uses `Future::then` API to create the chained Future, which will
be creating a CUDAFuture if the existing Future is a CUDA one.

fixes #50881
fixes #50839

[ghstack-poisoned]
  • Loading branch information
mrshenli committed Jan 22, 2021
2 parents 8f14194 + 4492caf commit 76b6714
Showing 0 changed files with 0 additions and 0 deletions.

0 comments on commit 76b6714

Please sign in to comment.