Resolving luajit/lua5.2 conflict

1 files changed, 6 insertions, 1 deletions

diff --git a/RNN.lua b/RNN.lua
index dc8e543..2836228 100644
--- a/RNN.lua
+++ b/RNN.lua
@@ -613,7 +613,12 @@ local function retrieveLinearParams(self, cuDNNMethod)
                 nbDims:data(),
                 filterDimA:data())
 
-            local offset = (tonumber(matrixPointer[0]) - tonumber(self.weight:data()))/self.weight:elementSize()
+            local offset
+	    if type(jit) == 'table' then
+	       offset = matrixPointer[0] - self.weight:data()
+	    else
+	       offset = (tonumber(matrixPointer[0]) - tonumber(self.weight:data()))/self.weight:elementSize()
+	    end
             local params = torch.CudaTensor(self.weight:storage(), offset + self.weight:storageOffset(), filterDimA:prod())
             table.insert(layerInfo, params)
         end