torch.mtia.memory#
Created On: Dec 09, 2024 | Last Updated: Dec 21, 2024
The MTIA backend is implemented out of the tree, only interfaces are be defined here.
This package adds support for device memory management implemented in MTIA.
memory_stats |
Return a dictionary of MTIA memory allocator statistics for a given device. |