mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-07 00:21:07 +01:00
300dcc3b96
Summary: Addresses #15968 Pull Request resolved: https://github.com/pytorch/pytorch/pull/15985 Differential Revision: D13649916 Pulled By: soumith fbshipit-source-id: a207aea5709a79dba7a6fc541d0a70103f49efff
60 lines
1.3 KiB
ReStructuredText
60 lines
1.3 KiB
ReStructuredText
torch.cuda
|
|
===================================
|
|
|
|
.. currentmodule:: torch.cuda
|
|
|
|
.. automodule:: torch.cuda
|
|
:members:
|
|
|
|
Random Number Generator
|
|
-------------------------
|
|
.. autofunction:: get_rng_state
|
|
.. autofunction:: get_rng_state_all
|
|
.. autofunction:: set_rng_state
|
|
.. autofunction:: set_rng_state_all
|
|
.. 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:: reset_max_memory_allocated
|
|
.. autofunction:: memory_cached
|
|
.. autofunction:: max_memory_cached
|
|
.. autofunction:: reset_max_memory_cached
|
|
|
|
NVIDIA Tools Extension (NVTX)
|
|
-----------------------------
|
|
|
|
.. autofunction:: torch.cuda.nvtx.mark
|
|
.. autofunction:: torch.cuda.nvtx.range_push
|
|
.. autofunction:: torch.cuda.nvtx.range_pop
|