.. automodule:: torch.accelerator
.. currentmodule:: torch.accelerator
.. autosummary::
:toctree: generated
:nosignatures:
device_count
is_available
current_accelerator
set_device_idx
current_device_idx
set_stream
current_stream
synchronize