From e2469527b1f85802e93ace4ce709bdabd7f1010a Mon Sep 17 00:00:00 2001
From: soumith <soumith@fb.com>
Date: Tue, 11 Apr 2017 14:55:09 -0700
Subject: fix THNN headers

---
 lib/THCUNN/generic/THCUNN.h | 22 +++++++++++-----------
 1 file changed, 11 insertions(+), 11 deletions(-)

diff --git a/lib/THCUNN/generic/THCUNN.h b/lib/THCUNN/generic/THCUNN.h
index 4ab2949..da8e41e 100644
--- a/lib/THCUNN/generic/THCUNN.h
+++ b/lib/THCUNN/generic/THCUNN.h
@@ -417,11 +417,11 @@ TH_API void THNN_(SparseLinear_updateParameters)(
 
 TH_API void THNN_(IndexLinear_updateOutput)(
                   THCState *state,
-                  THCudaLongTensor *keys,
+                  THCIndexTensor *keys,
                   long keysOffset,
                   THCTensor *values,
-                  THCudaLongTensor *sizes,
-                  THCudaLongTensor *cumSumSizes,
+                  THCIndexTensor *sizes,
+                  THCIndexTensor *cumSumSizes,
                   THCTensor *output,
                   THCTensor *weight,
                   THCTensor *bias,
@@ -430,11 +430,11 @@ TH_API void THNN_(IndexLinear_updateOutput)(
 
 TH_API void THNN_(IndexLinear_accGradParameters)(
                   THCState *state,
-                  THCudaLongTensor *keys,
+                  THCIndexTensor *keys,
                   long keysOffset,
                   THCTensor *values,
-                  THCudaLongTensor *sizes,
-                  THCudaLongTensor *cumSumSizes,
+                  THCIndexTensor *sizes,
+                  THCIndexTensor *cumSumSizes,
                   THCTensor *gradOutput,
                   THCTensor *gradWeight,
                   THCTensor *gradBias,
@@ -446,11 +446,11 @@ TH_API void THNN_(IndexLinear_accGradParameters)(
 
 TH_API void THNN_(IndexLinear_accUpdateGradParameters)(
                   THCState *state,
-                  THCudaLongTensor *keys,
+                  THCIndexTensor *keys,
                   long keysOffset,
                   THCTensor *values,
-                  THCudaLongTensor *sizes,
-                  THCudaLongTensor *cumSumSizes,
+                  THCIndexTensor *sizes,
+                  THCIndexTensor *cumSumSizes,
                   THCTensor *gradOutput,
                   THCTensor *weight,
                   THCTensor *bias,
@@ -463,8 +463,8 @@ TH_API void THNN_(IndexLinear_updateParameters)(
                   THCTensor *gradBias,
                   THCTensor *weight,
                   THCTensor *bias,
-                  THCudaLongTensor *runningKeys,
-                  THCudaLongTensor *cumSumSizes,
+                  THCIndexTensor *runningKeys,
+                  THCIndexTensor *cumSumSizes,
                   long keysOffset,
                   accreal weightDecay,
                   accreal learningRate);
-- 
cgit v1.2.3