check for half with cutorch.hasHalf

1 files changed, 1 insertions, 1 deletions

diff --git a/init.lua b/init.lua
index 0e3765b..f8dd115 100644
--- a/init.lua
+++ b/init.lua
@@ -37,7 +37,7 @@ cudnn.typemap = {
 }
 
 local sizeofmap = {
-   ['torch.CudaHalfTensor']   = ffi.sizeof('half'),
+   ['torch.CudaHalfTensor']   = cutorch.hasHalf and ffi.sizeof('half') or 2,
    ['torch.CudaTensor']       = ffi.sizeof('float'),
    ['torch.CudaDoubleTensor'] = ffi.sizeof('double'),
 }