Jeff Yang 84232b762b docs: add reset_peak_memory_stats in cuda.rst (#54668) ...

Summary:
fixes https://github.com/pytorch/pytorch/issues/41808
https://11812999-65600975-gh.circle-artifacts.com/0/docs/cuda.html

One question: does `reset_peak_stats` exist in `torch.cuda` ?
I can't find anywhere.

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

Reviewed By: ailzhang

Differential Revision: D27328444

Pulled By: zou3519

fbshipit-source-id: 098024d43da98e3249aa9aa71cb10126095504a4

2021-03-29 10:05:20 -07:00

..

amp

Mypy fixes for pytorch master (#52090)

2021-02-17 10:39:51 -08:00

__init__.py

Merge CUDA Streams and Events (#53902)

2021-03-26 14:19:39 -07:00

_utils.py

Merge CUDA Streams and Events (#53902)

2021-03-26 14:19:39 -07:00

comm.py

Add torch.cuda.comm to typechecking CI (#45350)

2020-09-25 12:13:43 -07:00

error.py

remediation of S205607

2020-07-17 17:19:47 -07:00

memory.py

docs: add reset_peak_memory_stats in cuda.rst (#54668)

2021-03-29 10:05:20 -07:00

nccl.py

Clean up usage of torch._six partially (#49785)

2021-02-08 13:58:34 -08:00

nvtx.py

[*.py] Rename "Arguments:" to "Args:" (#49736)

2020-12-28 09:34:47 -08:00

profiler.py

Replace map(lambda constructs (#46462)

2020-10-22 09:50:22 -07:00

random.py

doc string fix for torch.cuda.set_rng_state_all (#40544)

2020-06-25 08:37:14 -07:00

sparse.py

Delete dead Tensor code paths (#5417)

2018-02-27 17:58:09 -05:00

streams.py

[RELAND] [CUDA graphs] Private mempools for CUDA graphs (#54038)

2021-03-16 12:13:33 -07:00