We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 6ed9c16 commit 6673634Copy full SHA for 6673634
aten/THC/THCGrid.cu
@@ -12,7 +12,7 @@ __global__ void gridKernel(int64_t *self, TensorInfo<T> posInfo, T *size,
12
int64_t coef = 1, value = 0;
13
for (ptrdiff_t d = 0; d < posInfo.size[1]; d += posInfo.stride[1]) {
14
c = THCNumerics<T>::div(pos[d], size[d]);
15
- c = ScalarConvert<int64_t, T>::to(ScalarConvert<T, int64_t>::to(c)); // floor.
+ c = ScalarConvert<int64_t, T>::to(ScalarConvert<T, int64_t>::to(c));
16
value += coef * c;
17
coef *= count[d];
18
}
0 commit comments