fixes for multiple cuda types

1 files changed, 2 insertions, 3 deletions

diff --git a/THCUNN.lua b/THCUNN.lua
index 07a8cc8..771cd0c 100644
--- a/THCUNN.lua
+++ b/THCUNN.lua
@@ -25,9 +25,8 @@ local replacements =
 {
    {
       ['THTensor'] = 'THCudaTensor',
-      ['THIndexTensor'] = 'THCudaTensor',
-      ['THIntegerTensor'] = 'THCudaTensor',
-      ['THIndex_t'] = 'float',
+      ['THIndexTensor'] = 'THCudaLongTensor',
+      ['THIndex_t'] = 'long',
       ['THInteger_t'] = 'float'
    }
 }