Files
pytorch/docs/source/cpu.rst
Angel Li fa95882093 [BE] document distributed apis (#165194)
This PR documents some `torch.distributed.distributed_c10d` APIs. Below are some screenshots of the rendered docs.

<img width="909" height="527" alt="Screenshot 2025-10-10 at 10 18 40 PM" src="https://github.com/user-attachments/assets/555ae886-bead-47f3-8c67-9bc91c14bd11" />
<img width="885" height="548" alt="Screenshot 2025-10-10 at 10 18 47 PM" src="https://github.com/user-attachments/assets/1d6f7af1-db28-40f9-927e-5c47668a1a88" />

Pull Request resolved: https://github.com/pytorch/pytorch/pull/165194
Approved by: https://github.com/janeyx99
2025-10-13 20:13:59 +00:00

27 lines
425 B
ReStructuredText

torch.cpu
===================================
.. automodule:: torch.cpu
.. currentmodule:: torch.cpu
.. autosummary::
:toctree: generated
:nosignatures:
current_device
current_stream
is_available
is_initialized
synchronize
stream
set_device
device_count
StreamContext
Streams and events
------------------
.. autosummary::
:toctree: generated
:nosignatures:
Stream