mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-07 00:21:07 +01:00
56 lines
1.1 KiB
ReStructuredText
56 lines
1.1 KiB
ReStructuredText
torch.cuda
|
|
===================================
|
|
|
|
.. currentmodule:: torch.cuda
|
|
|
|
.. automodule:: torch.cuda
|
|
:members:
|
|
|
|
Random Number Generator
|
|
-------------------------
|
|
.. autofunction:: get_rng_state
|
|
.. autofunction:: set_rng_state
|
|
.. autofunction:: manual_seed
|
|
.. autofunction:: manual_seed_all
|
|
.. autofunction:: seed
|
|
.. autofunction:: seed_all
|
|
.. autofunction:: initial_seed
|
|
|
|
|
|
Communication collectives
|
|
-------------------------
|
|
|
|
.. autofunction:: torch.cuda.comm.broadcast
|
|
|
|
.. autofunction:: torch.cuda.comm.broadcast_coalesced
|
|
|
|
.. autofunction:: torch.cuda.comm.reduce_add
|
|
|
|
.. autofunction:: torch.cuda.comm.scatter
|
|
|
|
.. autofunction:: torch.cuda.comm.gather
|
|
|
|
Streams and events
|
|
------------------
|
|
|
|
.. autoclass:: Stream
|
|
:members:
|
|
|
|
.. autoclass:: Event
|
|
:members:
|
|
|
|
Memory management
|
|
-----------------
|
|
.. autofunction:: empty_cache
|
|
.. autofunction:: memory_allocated
|
|
.. autofunction:: max_memory_allocated
|
|
.. autofunction:: memory_cached
|
|
.. autofunction:: max_memory_cached
|
|
|
|
NVIDIA Tools Extension (NVTX)
|
|
-----------------------------
|
|
|
|
.. autofunction:: torch.cuda.nvtx.mark
|
|
.. autofunction:: torch.cuda.nvtx.range_push
|
|
.. autofunction:: torch.cuda.nvtx.range_pop
|