From 7ab89f58be479cd63dbe6b25f418033d70cd23a2 Mon Sep 17 00:00:00 2001
From: Natalia Gimelshein <ngimel@fb.com>
Date: Fri, 29 Jan 2021 18:20:17 -0800
Subject: [PATCH] expose memory_fraction and gpu_process docs (#51372)

Summary:
Per title

Pull Request resolved: https://github.com/pytorch/pytorch/pull/51372

Reviewed By: mruberry

Differential Revision: D26157787

Pulled By: ngimel

fbshipit-source-id: 97eac5f12881a2bf62c251f6f7eaf65fdbe34056
---
 docs/source/cuda.rst | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/docs/source/cuda.rst b/docs/source/cuda.rst
index 056ee3da0a5e..9063a9cd7802 100644
--- a/docs/source/cuda.rst
+++ b/docs/source/cuda.rst
@@ -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
@@ -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