frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 21:14:14 +08:00
Code Issues Packages Projects Releases Wiki Activity
Files
main
pytorch/docs/source/mtia.md
jiannanWang b6a48ff69f [BE] Add Documentation for Device APIs (#162834) 
Added documentation for torch.cuda APIs.
Fixed docstring for xpu and mtia is_bf16_supported API.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/162834
Approved by: https://github.com/janeyx99

Co-authored-by: Jane (Yuan) Xu <31798555+janeyx99@users.noreply.github.com>
2025-09-16 17:01:06 +00:00

820 B
Raw Permalink Blame History

torch.mtia

The MTIA backend is implemented out of the tree, only interfaces are be defined here.

.. automodule:: torch.mtia

.. currentmodule:: torch.mtia

.. autosummary::
    :toctree: generated
    :nosignatures:

    StreamContext
    current_device
    current_stream
    default_stream
    device_count
    init
    is_available
    is_bf16_supported
    is_initialized
    memory_stats
    get_device_capability
    empty_cache
    record_memory_history
    snapshot
    attach_out_of_memory_observer
    set_device
    set_stream
    stream
    synchronize
    device
    set_rng_state
    get_rng_state
    DeferredMtiaCallError

Streams and events

.. autosummary::
    :toctree: generated
    :nosignatures:

    Event
    Stream