pytorch
b26eafec - [MPS] Specialized memory pool for scalar values. (#85817)

Commit
2 years ago
[MPS] Specialized memory pool for scalar values. (#85817) - Add buffer usage and debug verbosity flags to MPSAllocator - Add high_watermark_ration to limit the memory allocation Pull Request resolved: https://github.com/pytorch/pytorch/pull/85817 Approved by: https://github.com/razarmehr
Author
Committer
Parents
Loading