Skip to content

Latest commit

 

History

History
19 lines (17 loc) · 383 Bytes

accelerator.rst

File metadata and controls

19 lines (17 loc) · 383 Bytes

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