mirror of
https://github.com/pytorch/pytorch.git
synced 2025-11-13 02:45:19 +08:00
18785c1af9
Pull Request resolved: https://github.com/pytorch/pytorch/pull/140542 Approved by: https://github.com/guangyey, https://github.com/albanD
20 lines
383 B
ReStructuredText
20 lines
383 B
ReStructuredText
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
|