diff options
| author | Trevor Killeen <killeentm@gmail.com> | 2017-05-04 16:50:46 +0300 |
|---|---|---|
| committer | Soumith Chintala <soumith@gmail.com> | 2017-05-09 02:09:02 +0300 |
| commit | 8fb354fa08cd98eccbff13bbdc99fc2b3e07824f (patch) | |
| tree | 86abcbf5ca56cf80ba124f2c524e837311537cfe | |
| parent | abc860b7e1bcdba3cc93a6157c91009a2ca02a58 (diff) | |
s/IndexType/long
| -rw-r--r-- | lib/THC/THCTensorScatterGather.cu | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/lib/THC/THCTensorScatterGather.cu b/lib/THC/THCTensorScatterGather.cu index 323aeb0..18c9dee 100644 --- a/lib/THC/THCTensorScatterGather.cu +++ b/lib/THC/THCTensorScatterGather.cu @@ -92,7 +92,7 @@ __global__ void THCudaTensor_gatherKernel( tensor, &tensorOffset, src, &srcOffset); - IndexType indexValue = (IndexType)index.data[indexOffset] - TH_INDEX_BASE; + long indexValue = index.data[indexOffset] - TH_INDEX_BASE; assert(indexValue >= 0 && indexValue < src.sizes[dim]); srcOffset += indexValue * src.strides[dim]; @@ -119,7 +119,7 @@ __global__ void THCudaTensor_scatterKernel( src, &srcOffset, tensor, &tensorOffset); - IndexType indexValue = (IndexType)index.data[indexOffset] - TH_INDEX_BASE; + long indexValue = index.data[indexOffset] - TH_INDEX_BASE; assert(indexValue >= 0 && indexValue < tensor.sizes[dim]); tensorOffset += indexValue * tensor.strides[dim]; @@ -144,7 +144,7 @@ __global__ void THCudaTensor_scatterFillKernel( index, &indexOffset, tensor, &tensorOffset); - IndexType indexValue = (IndexType)index.data[indexOffset] - TH_INDEX_BASE; + long indexValue = index.data[indexOffset] - TH_INDEX_BASE; assert(indexValue >= 0 && indexValue < tensor.sizes[dim]); tensorOffset += indexValue * tensor.strides[dim]; |