torch.mtia.memory

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.

empty_cache	Empty the MTIA device cache.
memory_stats	Return a dictionary of MTIA memory allocator statistics for a given device.
max_memory_allocated	Return the maximum memory allocated in bytes for a given device.