diff options
author | Gregory Chanan <gchanan@fb.com> | 2016-12-16 02:40:21 +0300 |
---|---|---|
committer | Gregory Chanan <gchanan@fb.com> | 2016-12-16 02:40:21 +0300 |
commit | 6939652f4caf6eaae09cfeb88163e490b8107ca1 (patch) | |
tree | cbe11d110a9c04a6002543f85f63a858bdb3da2a | |
parent | 1d8c97124e2fa67bbcf2d7f5832b730cbe1b8de0 (diff) |
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 |