SsnL
300dcc3b96
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
2019-01-14 07:31:51 -08:00
SsnL
e4477feb15
Update cuda.get/set_rng_state doc ( #14324 )
...
Summary:
Now that `cuda.get/set_rng_state` accept `device` objects, the default value should be an device object, and doc should mention so.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/14324
Reviewed By: ezyang
Differential Revision: D13528707
Pulled By: soumith
fbshipit-source-id: 32fdac467dfea6d5b96b7e2a42dc8cfd42ba11ee
2018-12-27 14:09:25 -08:00
Sam Gross
f1c616418d
Fix Python docs for broadcast and braodcast_coalesced ( #4727 )
2018-01-19 10:57:20 -05:00
Tongzhou Wang
5918243b0c
Methods for checking CUDA memory usage ( #4511 )
...
* gpu mem allocated
* add test
* addressed some of @apaszke 's comments
* cache stats
* add more comments about test
2018-01-09 11:47:48 -05:00
SsnL
bb1b826cdc
Exposing emptyCache from allocator ( #3518 )
...
* Add empty_cache binding
* cuda.empty_cache document
* update docs
2017-11-07 17:00:38 -05:00
Soumith Chintala
cf7e28de8e
add CUDA RNG docs
2017-09-21 19:36:41 -04:00
Edward Z. Yang
ba690d5607
Add support for NVTX functions. ( #1748 )
2017-06-10 18:26:58 +02:00
Adam Paszke
15c1dad340
Minor fixes and torch.cuda docs
2017-01-16 20:38:14 -05:00
Sam Gross
126a1cc398
Add Sphinx docs
2016-12-28 00:03:39 +01:00
