From bbec1572127c3deca45e8924e29d550934459eb2 Mon Sep 17 00:00:00 2001
From: SeanNaren <seannaren@hotmail.com>
Date: Tue, 9 Aug 2016 09:35:03 +0000
Subject: [PATCH] Mask changed to cuda byte tensor for cutorch api

---
 MaskZero.lua | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/MaskZero.lua b/MaskZero.lua
index bdb5e8e..d2ea82b 100644
--- a/MaskZero.lua
+++ b/MaskZero.lua
@@ -68,7 +68,7 @@ function MaskZero:updateOutput(input)
    local vectorDim = rmi:dim() 
    self._zeroMask = self._zeroMask or rmi.new()
    self._zeroMask:norm(rmi, 2, vectorDim)
-   self.zeroMask = self.zeroMask or ((torch.type(rmi) == 'torch.CudaTensor') and torch.CudaTensor() or torch.ByteTensor())
+   self.zeroMask = self.zeroMask or ((torch.type(rmi) == 'torch.CudaTensor') and torch.CudaByteTensor() or torch.ByteTensor())
    self._zeroMask.eq(self.zeroMask, self._zeroMask, 0)
    
    -- forward through decorated module