frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 21:14:14 +08:00
Code Issues Packages Projects Releases Wiki Activity
Files
main
pytorch/docs/source/cuda.aliases.md
Jiannan Wang 6ac2b3ae35 [BE] Adding aliases for CUDA and XPU API documentation (#162984) 
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
2025-09-21 22:28:27 +00:00

1.0 KiB
Raw Permalink Blame History

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.

.. 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

.. automodule:: torch.cuda.graphs
.. currentmodule:: torch.cuda.graphs
.. autosummary::
    :toctree: generated
    :nosignatures:

    is_current_stream_capturing
    graph_pool_handle
    CUDAGraph
    graph
    make_graphed_callables

.. automodule:: torch.cuda.streams
.. currentmodule:: torch.cuda.streams
.. autosummary::
    :toctree: generated
    :nosignatures:

    Stream
    ExternalStream
    Event