From a8708bf380c939335717537c2156e570d52c7dea Mon Sep 17 00:00:00 2001
From: Ronan Collobert <ronan@collobert.com>
Date: Mon, 22 Oct 2012 21:55:03 +0200
Subject: wrap package: fix in IndexTensor

---
 types.lua | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/types.lua b/types.lua
index 052bfe3..059ec12 100644
--- a/types.lua
+++ b/types.lua
@@ -109,9 +109,11 @@ wrap.argtypes.IndexTensor = {
 
    read = function(arg, idx)
              local txt = {}
-             table.insert(txt, string.format("THLongTensor_add(arg%d, arg%d, -1);", arg.i, arg.i));
+             if not arg.noreadadd then
+                table.insert(txt, string.format("THLongTensor_add(arg%d, arg%d, -1);", arg.i, arg.i));
+             end
              if arg.returned then
-                return table.insert(txt, string.format("arg%d_idx = %d;", arg.i, idx))
+                table.insert(txt, string.format("arg%d_idx = %d;", arg.i, idx))
              end
              return table.concat(txt, '\n')
           end,
-- 
cgit v1.2.3