diff options
| author | Soumith Chintala <soumith@gmail.com> | 2016-12-16 02:44:42 +0300 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2016-12-16 02:44:42 +0300 |
| commit | 393e72c8a2a26cfae4456f5449173b37d9a16f65 (patch) | |
| tree | cbe11d110a9c04a6002543f85f63a858bdb3da2a | |
| parent | dac0e700f286b223b8b69f4330b44ce87cd7f744 (diff) | |
| parent | 6939652f4caf6eaae09cfeb88163e490b8107ca1 (diff) | |
Merge pull request #88 from gchanan/nonfpSerialize
Add support for non-floating point cuda types.
| -rw-r--r-- | sharedserialize.lua | 10 |
1 files changed, 10 insertions, 0 deletions
diff --git a/sharedserialize.lua b/sharedserialize.lua index 8bd23ff..18612e0 100644 --- a/sharedserialize.lua +++ b/sharedserialize.lua @@ -87,6 +87,11 @@ for _, typename in ipairs{ 'torch.FloatTensor', 'torch.DoubleTensor', 'torch.CudaTensor', + 'torch.CudaByteTensor', + 'torch.CudaCharTensor', + 'torch.CudaShortTensor', + 'torch.CudaIntTensor', + 'torch.CudaLongTensor', 'torch.CudaDoubleTensor', 'torch.CudaHalfTensor', 'torch.ByteStorage', @@ -97,6 +102,11 @@ for _, typename in ipairs{ 'torch.FloatStorage', 'torch.DoubleStorage', 'torch.CudaStorage', + 'torch.CudaByteStorage', + 'torch.CudaCharStorage', + 'torch.CudaShortStorage', + 'torch.CudaIntStorage', + 'torch.CudaLongStorage', 'torch.CudaDoubleStorage', 'torch.CudaHalfStorage', 'torch.Allocator'} do |