Merge pull request torch#399 from htwaijry/master

Fix unchecked memory access when indexing a CudaTensor
marsbzp · May 1, 2016 · 0fe1a4c · 0fe1a4c
2 parents c3c8f15 + eb60e79
commit 0fe1a4c
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/lib/THC/THCTensorIndex.cu b/lib/THC/THCTensorIndex.cu
@@ -682,6 +682,8 @@ void THCudaTensor_indexSelect_long(THCState *state, THCudaTensor *dst, THCudaTen
 {
   THAssert(THCudaTensor_checkGPU(state, 2, dst, src));
 
+  THArgCheck(indices->nDimension == 1, 3, "Index is supposed to be a vector");
+
   THCudaTensor *indices_ = THCudaTensor_newWithSize1d(state, indices->size[0]);
   THCudaTensor_copyLong(state, indices_, indices);