Sam Gross
|
79ead42ade
|
Add CUDA Stream and Event API (#133)
|
2016-10-18 12:15:57 -04:00 |
|
Sam Gross
|
8d39fb4094
|
Use new THC API for device allocator
|
2016-10-17 09:35:41 -07:00 |
|
Sam Gross
|
ee14cf9438
|
Add support for pinned memory: (#127)
torch.Storage/Tensor.pin_memory()
torch.Storage/Tensor.is_pinned()
|
2016-10-15 18:38:26 -04:00 |
|
Sam Gross
|
c20828478e
|
Update Module.cpp for THC changes
|
2016-09-30 11:13:14 -07:00 |
|
Adam Paszke
|
3f7ab95890
|
Finish implementation of prng related functions
|
2016-09-29 11:33:25 -07:00 |
|
Sam Gross
|
4e9f0a8255
|
Use CUDA caching allocator
|
2016-09-26 13:12:39 -07:00 |
|
Adam Paszke
|
06ab3f962f
|
Refactor _C extension to export some utilities
|
2016-09-21 08:36:54 -07:00 |
|
Adam Paszke
|
3ea1da3b2c
|
Minor fix in CUDA module
|
2016-09-14 11:09:03 -04:00 |
|
soumith
|
1f2695e875
|
adding cuda driver check functions for runtime checking
|
2016-09-13 10:34:13 -07:00 |
|
Adam Paszke
|
8d933cbfc4
|
Fixes for OS X
|
2016-08-22 22:45:35 -04:00 |
|
Adam Paszke
|
12bed8dc0d
|
Add CUDA device selection
|
2016-08-12 07:46:46 -07:00 |
|
Adam Paszke
|
92e983a489
|
Fixes for Linux and new cutorch
|
2016-08-02 09:20:18 -07:00 |
|
Adam Paszke
|
c574295012
|
Various fixes
|
2016-07-19 10:45:59 -04:00 |
|
Adam Paszke
|
3a44259b32
|
Add support for CUDA
|
2016-07-19 10:45:59 -04:00 |
|