Skip to content

Commit

Permalink
Update base for 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 991c554 + 5f07b53 commit 4492caf
Showing 0 changed files with 0 additions and 0 deletions.

0 comments on commit 4492caf

Please sign in to comment.