fix the bug in backward OP of index_sample. (#31026)

4 years ago · b95eb38b8a
parent 6b3371e0c7
commit b95eb38b8a
1 changed files with 1 additions and 1 deletions
--- a/paddle/fluid/operators/index_sample_op.cu
+++ b/paddle/fluid/operators/index_sample_op.cu
@ -55,7 +55,7 @@ __global__ void IndexSampleGrad(const IndexT* index, T* in_grad,
      platform::CudaAtomicAdd(&(in_grad[in_idx - index_i + sample_idx]),
                              out_grad[sample_idx]);
    } else {
-      in_grad[in_idx - index_i + sample_idx] = out_grad[sample_idx];
+      in_grad[in_idx - index_i + sample_idx] = out_grad[index_idx];
    }
  }
 }