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

b95eb38b · JamesLim · GitHub · 6b3371e0 · b95eb38b
显示空白变更内容
内联并排

Showing with 1 addition and 1 deletion

paddle/fluid/operators/index_sample_op.cu paddle/fluid/operators/index_sample_op.cu +1 -1

未找到文件。
--- 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];
    }
  }
 }