mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-21 05:34:18 +08:00
6ac2b3ae35
This PR reorganizes CUDA and XPU API documentation with additional aliases pages. Multiple entries of APIs under torch.cuda are thus removed. Pull Request resolved: https://github.com/pytorch/pytorch/pull/162984 Approved by: https://github.com/janeyx99
47 lines
1.0 KiB
Markdown
47 lines
1.0 KiB
Markdown
# Aliases in torch.cuda
|
|
|
|
The following are aliases to their counterparts in ``torch.cuda`` in the nested namespaces in which they are defined. For any of these APIs, feel free to use the top-level version in ``torch.cuda`` like ``torch.cuda.seed`` or the nested version ``torch.cuda.random.seed``.
|
|
|
|
```{eval-rst}
|
|
.. automodule:: torch.cuda.random
|
|
.. currentmodule:: torch.cuda.random
|
|
.. autosummary::
|
|
:toctree: generated
|
|
:nosignatures:
|
|
|
|
get_rng_state
|
|
get_rng_state_all
|
|
set_rng_state
|
|
set_rng_state_all
|
|
manual_seed
|
|
manual_seed_all
|
|
seed
|
|
seed_all
|
|
initial_seed
|
|
```
|
|
|
|
```{eval-rst}
|
|
.. automodule:: torch.cuda.graphs
|
|
.. currentmodule:: torch.cuda.graphs
|
|
.. autosummary::
|
|
:toctree: generated
|
|
:nosignatures:
|
|
|
|
is_current_stream_capturing
|
|
graph_pool_handle
|
|
CUDAGraph
|
|
graph
|
|
make_graphed_callables
|
|
```
|
|
|
|
```{eval-rst}
|
|
.. automodule:: torch.cuda.streams
|
|
.. currentmodule:: torch.cuda.streams
|
|
.. autosummary::
|
|
:toctree: generated
|
|
:nosignatures:
|
|
|
|
Stream
|
|
ExternalStream
|
|
Event
|
|
``` |