pytorch/docs/source/mtia.memory.md
2025-10-29 03:01:07 +00:00

21 lines
331 B
Markdown

# torch.mtia.memory
The MTIA backend is implemented out of the tree, only interfaces are defined here.
```{eval-rst}
.. automodule:: torch.mtia.memory
```
```{eval-rst}
.. currentmodule:: torch.mtia.memory
```
```{eval-rst}
.. autosummary::
:toctree: generated
:nosignatures:
memory_stats
memory_allocated
```