Skip to content

Commit ea9fcd5

Browse files
ngimelsoumith
authored andcommitted
fix copy-paste error in #3263 (#3476)
I have no idea how it worked on cuda 8, but apparently this fixes failures on cuda 9. cc @colesbury
1 parent f7a459b commit ea9fcd5

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

aten/src/THC/generic/THCTensorIndex.cu

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -183,7 +183,7 @@ void THCTensor_(put)(THCState *state, THCTensor *dst, THCudaLongTensor *index, T
183183
THCTensor* sorted_src = THCTensor_(newClone)(state, src);
184184

185185
THCTensor_(sort_indices)(state, sorted_index, sorted_src);
186-
dispatchTakePut<real, TensorPutAccumulateOp>(state, dst, src, index);
186+
dispatchTakePut<real, TensorPutAccumulateOp>(state, dst, sorted_src, sorted_index);
187187

188188
THCTensor_(free)(state, sorted_src);
189189
THCudaLongTensor_free(state, sorted_index);

0 commit comments

Comments
 (0)