pytorch

3af6ff98 - [reland] Provide pre-extracted DataPtrs when completing a Future with a Message (#59208)

[reland] Provide pre-extracted DataPtrs when completing a Future with a Message (#59208) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/59208 Reland of 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: 130202845 Test Plan: CI Reviewed By: mrshenli Differential Revision: D28623888 fbshipit-source-id: 1aa4bde8014870c071685ba8f72d5f3f01f0a512