add device asserts in scatter/gather kernels

Author: killeent

THCTensorScatterGather.cu

@@ -93,6 +93,7 @@ __global__ void THCudaTensor_gatherKernel(
  src, &srcOffset);
 
  IndexType indexValue = (IndexType)index.data[indexOffset] - TH_INDEX_BASE;
+ assert(indexValue >= 0 && indexValue < src.sizes[dim]);
  srcOffset += indexValue * src.strides[dim];
 
  tensor.data[tensorOffset] = src.data[srcOffset];
@@ -119,6 +120,7 @@ __global__ void THCudaTensor_scatterKernel(
  tensor, &tensorOffset);
 
  IndexType indexValue = (IndexType)index.data[indexOffset] - TH_INDEX_BASE;
+ assert(indexValue >= 0 && indexValue < tensor.sizes[dim]);
  tensorOffset += indexValue * tensor.strides[dim];
 
  tensor.data[tensorOffset] = src.data[srcOffset];
@@ -143,6 +145,7 @@ __global__ void THCudaTensor_scatterFillKernel(
  tensor, &tensorOffset);
 
  IndexType indexValue = (IndexType)index.data[indexOffset] - TH_INDEX_BASE;
+ assert(indexValue >= 0 && indexValue < tensor.sizes[dim]);
  tensorOffset += indexValue * tensor.strides[dim];
 
  tensor.data[tensorOffset] = value;