pytorch
bdf6a4bf - Provide pre-extracted DataPtrs when completing a Future with a Message (#58425)

Commit
4 years ago
Provide pre-extracted DataPtrs when completing a Future with a Message (#58425) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/58425 Now that callbacks can provide pre-extracted DataPtrs, let's do so. This will become of crucial importance in the next PR, where some of these futures will become CUDA-aware, and thus they will try to extract DataPtrs on their own, but they would fail to do so here because Message isn't "inspectable". ghstack-source-id: 129567057 Test Plan: CI Reviewed By: mrshenli Differential Revision: D28474877 fbshipit-source-id: e68d7d45f1c1dc6daa5e05cf984cfc93d2dce0d0
Author
lw lw
Parents
Loading