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 |
|