pytorch
c981b7e5 - [MPS] Add MPSAllocatorInterface to access methods of MPSAllocator (#94327)

Commit
1 year ago
[MPS] Add MPSAllocatorInterface to access methods of MPSAllocator (#94327) This is a prerequisite for the upcoming PR's for the MPS Modules and Memory Leak Detection features. Also added pragma once to headers. Pull Request resolved: https://github.com/pytorch/pytorch/pull/94327 Approved by: https://github.com/kulinseth
Author
Committer
Parents
Loading