pytorch
0c470b17 - Extend storage create for custom storageImpl (#100237)

Commit
2 years ago
Extend storage create for custom storageImpl (#100237) Fixes #ISSUE_NUMBER For the scenario where users inherit storageimpl to implement their own subclasses, the current storage creation method cannot correctly create storage objects. Refer to the registration method of Allocator to expand the creation method of storageimpl, users can register their own custom storageimpl creation. Pull Request resolved: https://github.com/pytorch/pytorch/pull/100237 Approved by: https://github.com/albanD
Author
Committer
Parents
Loading