commit 79533164f72c3b203c829417e6a881619f3a51ca
author nicholas-leonard <nick@nikopia.org> Thu Jun 12 01:04:47 2014 -0400
committer nicholas-leonard <nick@nikopia.org> Thu Jun 12 01:04:47 2014 -0400
tree 0a6d8fcf7c3172b6e263451a01235edf8773133c
parent b26875768eb857a86a7cf450acd458e48bb39918

indexSelect_kernel arguments

THCTensorMath.cu

diff --git a/THCTensorMath.cu b/THCTensorMath.cu
index de05377..7ef2b24 100644
--- a/THCTensorMath.cu
+++ b/THCTensorMath.cu
@@ -1075,8 +1075,8 @@
 }
 
 __global__ void THCudaTensor_kernel_indexSelect(
-   float *tensor, float *src, long* src_stride, long src_nDim, 
-   int dim, long *index, long idx_size, long tensor_size, long size_dim
+   float *tensor, float *src, float* src_stride, float *index, 
+   long src_nDim, int dim, long idx_size, long tensor_size, long size_dim
 )
 {
   int thread_idx = blockIdx.x * blockDim.x * blockDim.y + threadIdx.y * blockDim.x + threadIdx.x;
@@ -1146,7 +1146,7 @@
   THCudaTensor_kernel_indexSelect<<<nblocks, nthreads>>>(
     THCudaTensor_data(res_), THCudaTensor_data(src), 
     THCudaTensor_data(stride_), THCudaTensor_data(indices_), 
-    src->nDimension, dim, 
+    src->nDimension, dim, indices->size[1],
     THCudaTensor_nElement(res_), src->size[dim]
   );