The MTIA backend is implemented out of the tree, only interfaces are defined here.
.. automodule:: torch.mtia.memory
.. currentmodule:: torch.mtia.memory
.. autosummary::
:toctree: generated
:nosignatures:
reset_peak_memory_stats
max_memory_allocated
memory_stats
memory_allocated