diff options
Diffstat (limited to 'sharedserialize.lua')
-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 |