pytorch
ee5f09ab - [Feature] storage pin memory support custom device. (#99712)

Commit
1 year ago
[Feature] storage pin memory support custom device. (#99712) Fixes #99326 Support storage pin_memory and is_pinned for custom device, by calling dispatched tensor operations. @ezyang this pr is what we have discussed in issue #99326, would you please take a moment to review it, thanks. Pull Request resolved: https://github.com/pytorch/pytorch/pull/99712 Approved by: https://github.com/ezyang
Author
Committer
Parents
Loading