accelerator.md

torch.accelerator

.. automodule:: torch.accelerator

.. currentmodule:: torch.accelerator

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

    device_count
    is_available
    current_accelerator
    set_device_index
    set_device_idx
    current_device_index
    current_device_idx
    set_stream
    current_stream
    synchronize
    device_index

.. automodule:: torch.accelerator.memory

.. currentmodule:: torch.accelerator.memory

Memory management

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

     empty_cache
     get_memory_info
     max_memory_allocated
     max_memory_reserved
     memory_allocated
     memory_reserved
     memory_stats
     reset_accumulated_memory_stats
     reset_peak_memory_stats