mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-07 12:21:27 +01:00
23bffc4f14
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/27782 Warnings show up when running `make html` to build documentation. All of the warnings are very reasonable and point to bugs in our docs. This PR attempts to fix most of those warnings. In the future we will add something to the CI that asserts that there are no warnings in our docs. Test Plan: - build and view changes locally Differential Revision: D17887067 Pulled By: zou3519 fbshipit-source-id: 6bf4d08764759133b20983d6cd7f5d27e5ee3166
68 lines
1.6 KiB
ReStructuredText
68 lines
1.6 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_stats
|
|
.. autofunction:: memory_summary
|
|
.. autofunction:: memory_snapshot
|
|
.. autofunction:: memory_allocated
|
|
.. autofunction:: max_memory_allocated
|
|
.. autofunction:: reset_max_memory_allocated
|
|
.. autofunction:: memory_reserved
|
|
.. autofunction:: max_memory_reserved
|
|
.. FIXME The following doesn't seem to exist. Is it supposed to?
|
|
https://github.com/pytorch/pytorch/issues/27785
|
|
.. autofunction:: reset_max_memory_reserved
|
|
.. 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
|