diff options
author | Gregory Chanan <gchanan@fb.com> | 2016-12-12 22:02:50 +0300 |
---|---|---|
committer | Gregory Chanan <gchanan@fb.com> | 2016-12-16 00:38:18 +0300 |
commit | 46c1e283dab3ca5b8aa4f00e9df615fbb85b2d82 (patch) | |
tree | 66610f51501315a72739ac58e21107c804376954 | |
parent | 1eaaf59925c4fff44a794e2ba6b3c734f37d6e77 (diff) |
Add CudaDouble, CudaHalf support in sharedserialize.
-rw-r--r-- | sharedserialize.lua | 4 |
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 = {} |