OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-07 12:21:27 +01:00
Code Issues Actions 38 Packages Projects Releases Wiki Activity
master
pytorch/docs/source/mtia.memory.md
linhaifeng 1764f3a9c8 [Fix] fix gramma error in PyTorch docs (#166158) 
Fix several gramma errors in PyTorch docs.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/166158
Approved by: https://github.com/yewentao256, https://github.com/cyyever, https://github.com/ezyang
2025-10-29 03:01:07 +00:00

331 B
Raw Permalink Blame History

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