From 6939652f4caf6eaae09cfeb88163e490b8107ca1 Mon Sep 17 00:00:00 2001
From: Gregory Chanan <gchanan@fb.com>
Date: Thu, 15 Dec 2016 15:40:21 -0800
Subject: Add support for non-floating point cuda types.

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

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