"server/vscode:/vscode.git/clone" did not exist on "38773453ae0d29fba3dc79a38d589ebdc5451093"
Commit 2f2dd12f authored by rusty1s's avatar rusty1s
Browse files

typo

parent cc6f8862
......@@ -75,7 +75,9 @@ __global__ void indexBackwardKernel(TensorInfo<Real> output, TensorInfo<int64_t>
KERNEL_LOOP(i, n) {
int outputOffset = 0; int indexOffset = 0; int gradOffset = 0; int argOffset = 0;
IndexToScatterOffsets4<Real, Real, int64_t, Dims>::compute(i, dim, index, &indexOffset, output, &outputOffset, grad, &gradOffset, arg, &argOffset);
if (arg.data[argOffset] == (outputOffset / output.stride[dim]) % output.size[dim]) output.data[outputOffset] = grad.data[gradOffset];
if (arg.data[argOffset] == (outputOffset / output.stride[dim]) % output.size[dim]) {
output.data[outputOffset] = grad.data[gradOffset];
}
}
}
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment