luajit fix

1 files changed, 6 insertions, 1 deletions

diff --git a/RNN.lua b/RNN.lua
index dc8e543..80ee760 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 jit 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