From 46c1e283dab3ca5b8aa4f00e9df615fbb85b2d82 Mon Sep 17 00:00:00 2001
From: Gregory Chanan <gchanan@fb.com>
Date: Mon, 12 Dec 2016 11:02:50 -0800
Subject: Add CudaDouble, CudaHalf support in sharedserialize.

---
 sharedserialize.lua | 4 ++++
 1 file changed, 4 insertions(+)

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 = {}
-- 
cgit v1.2.3