Skip to content

Latest commit

 

History

History
20 lines (15 loc) · 331 Bytes

File metadata and controls

20 lines (15 loc) · 331 Bytes

torch.mtia.memory

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:

    memory_stats
    memory_allocated