Don't reference HalfTensor when it doesn't exist

1 files changed, 2 insertions, 0 deletions

diff --git a/test.lua b/test.lua
index 7cb2854..1fdfcc8 100644
--- a/test.lua
+++ b/test.lua
@@ -26,9 +26,11 @@ local function checkHalf()
 end
 
 -- workarounds for non-existant functions
+if cutorch.hasHalf then
 function torch.CudaHalfTensor:mean()
    return self:cuda():mean()
 end
+end
 
 function torch.CudaDoubleTensor:mean()
    return self:cuda():mean()