Add cuda.reset_max_memory_* (#15985)
authorSsnL <tongzhou.wang.1994@gmail.com>
Mon, 14 Jan 2019 15:28:50 +0000 (07:28 -0800)
committerFacebook Github Bot <facebook-github-bot@users.noreply.github.com>
Mon, 14 Jan 2019 15:31:51 +0000 (07:31 -0800)
commit300dcc3b96b2b8672570c4c9e6038cb5cff75afd
treeb0a64eebc9a9730fccf50a19b9a4b4739808666f
parent7c08f1083e56ce7e9799a15b36947c46052f0058
Add cuda.reset_max_memory_* (#15985)

Summary:
Addresses #15968
Pull Request resolved: https://github.com/pytorch/pytorch/pull/15985

Differential Revision: D13649916

Pulled By: soumith

fbshipit-source-id: a207aea5709a79dba7a6fc541d0a70103f49efff
aten/src/THC/THCCachingAllocator.cpp
aten/src/THC/THCCachingAllocator.h
docs/source/cuda.rst
docs/source/notes/cuda.rst
test/test_cuda.py
torch/csrc/cuda/Module.cpp
torch/cuda/__init__.py