pytorch

3460b2b7 - Add support for pin memory on custom device. (#97621)

Add support for pin memory on custom device. (#97621) Add support for pin memory on custom device. Pull Request resolved: https://github.com/pytorch/pytorch/pull/97621 Approved by: https://github.com/NivekT