Skip to content

Commit

Permalink
expose memory_fraction and gpu_process docs (#51372)
Browse files Browse the repository at this point in the history
Summary:
Per title

Pull Request resolved: #51372

Reviewed By: mruberry

Differential Revision: D26157787

Pulled By: ngimel

fbshipit-source-id: 97eac5f12881a2bf62c251f6f7eaf65fdbe34056
  • Loading branch information
ngimel authored and facebook-github-bot committed Jan 30, 2021
1 parent 7d30f67 commit 7ab89f5
Showing 1 changed file with 2 additions and 0 deletions.
2 changes: 2 additions & 0 deletions docs/source/cuda.rst
Expand Up @@ -44,6 +44,7 @@ Streams and events
Memory management
-----------------
.. autofunction:: empty_cache
.. autofunction:: list_gpu_processes
.. autofunction:: memory_stats
.. autofunction:: memory_summary
.. autofunction:: memory_snapshot
Expand All @@ -52,6 +53,7 @@ Memory management
.. autofunction:: reset_max_memory_allocated
.. autofunction:: memory_reserved
.. autofunction:: max_memory_reserved
.. autofunction:: set_per_process_memory_fraction
.. FIXME The following doesn't seem to exist. Is it supposed to?
https://github.com/pytorch/pytorch/issues/27785
.. autofunction:: reset_max_memory_reserved
Expand Down

0 comments on commit 7ab89f5

Please sign in to comment.