Sam Gross
bb72ccf1a5
Support CUDA IPC in Python 3 ( #203 )
...
CUDA IPC only works with Python 3 using the "spawn" start method. You
can select the start method using the get_context method:
import torch.multiprocessing as mp
ctx = mp.get_context('spawn')
queue = ctx.Queue()
event = ctx.Event()
2016-12-19 20:42:53 -05:00
Sam Gross
1af9a9637f
Refactor copy and release GIL during copy ( #286 )
2016-12-11 21:54:58 +01: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
Adam Paszke
2c89ae4e8a
Rename getDevice to get_device
2016-09-27 15:12:48 -07:00
Sam Gross
779a460030
Add cuDNN support for convolutions ( #36 )
2016-09-27 17:55:04 -04:00
Adam Paszke
4a8a185aa4
Preserve storage view sharing in torch.save and torch.load
2016-09-25 12:24:10 -07:00
Adam Paszke
e71204b52f
Improve error messages in storage and tensor C functions
2016-09-23 17:17:35 -07:00
Adam Paszke
06ab3f962f
Refactor _C extension to export some utilities
2016-09-21 08:36:54 -07:00
Adam Paszke
8fdec15a55
Codemod to remove camel case method naming
2016-09-20 08:40:28 -07:00
Adam Paszke
95d545e75b
Fix a multiprocessing GIL issue
2016-09-15 18:49:20 -07:00
Adam Paszke
491fbfdc8c
Improve error messages of tensor methods
2016-09-15 18:49:20 -07:00
Adam Paszke
da5bb373e6
Type conversions now use auto gpu
2016-09-15 18:48:27 -07:00
Soumith Chintala
96cd92a0a9
fix OSX build
2016-09-13 10:34:13 -07:00
Sam Gross
8a76dc8b59
Add missing PyBuffer_Release calls
2016-09-09 11:23:18 -07:00
Adam Paszke
58f507f9e3
Add file descriptor sharing mode to multiprocessing
2016-09-08 11:23:33 -07:00
Sam Gross
1486d880b0
Add Storage.from_buffer
...
The from_buffer is similar to numpy's frombuffer. It decodes a Python
buffer object into a Storage object. For byte and char storages, it
simply copies the bytes.
2016-09-07 15:32:33 -07:00
Adam Paszke
f9d186d33a
Add initial version of multiprocessing module
2016-08-31 19:46:08 -07:00
Adam Paszke
686e8d32e2
Add torch.save and torch.load
2016-08-23 07:51:55 -07:00
Adam Paszke
e9f9fd3727
Major refactor
2016-08-10 09:24:53 -07:00
Adam Paszke
3a44259b32
Add support for CUDA
2016-07-19 10:45:59 -04:00
Soumith Chintala
5ee3358a92
python 2 support
2016-06-08 19:14:57 -04:00
Adam Paszke
c3b3df9f22
Add utilities and clenup Tensor wrappers
2016-05-06 15:04:57 +02:00
Adam Paszke
4a5b66de9a
Add copy and type methods to Tensors
2016-05-05 22:44:43 +02:00
Adam Paszke
842e1b6358
Add exception handling
2016-05-05 20:58:13 +02:00