1 files changed, 4 insertions, 0 deletions
diff --git a/sharedserialize.lua b/sharedserialize.lua
index c34684e..8bd23ff 100644
--- a/sharedserialize.lua
+++ b/sharedserialize.lua
@@ -87,6 +87,8 @@ for _, typename in ipairs{
    'torch.FloatTensor',
    'torch.DoubleTensor',
    'torch.CudaTensor',
+   'torch.CudaDoubleTensor',
+   'torch.CudaHalfTensor',
    'torch.ByteStorage',
    'torch.CharStorage',
    'torch.ShortStorage',
@@ -95,6 +97,8 @@ for _, typename in ipairs{
    'torch.FloatStorage',
    'torch.DoubleStorage',
    'torch.CudaStorage',
+   'torch.CudaDoubleStorage',
+   'torch.CudaHalfStorage',
    'torch.Allocator'} do
 
    local mt = {}