Skip to content

P
Paddle

BaiXuePrincess / Paddle
与 Fork 源项目一致

Fork自 PaddlePaddle / Paddle

通知 1
Star 1
Fork 0
P
Paddle
未验证 提交 b95eb38b 编写于 作者: J JamesLim 提交者： GitHub
浏览文件
操作

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

上级 6b3371e0
显示空白变更内容
内联 并排
Showing with 1 addition and 1 deletion
paddle/fluid/operators/index_sample_op.cu
浏览文件 @ b95eb38b
...@@ -55,7 +55,7 @@ __global__ void IndexSampleGrad(const IndexT* index, T* in_grad, ...@@ -55,7 +55,7 @@ __global__ void IndexSampleGrad(const IndexT* index, T* in_grad,
platform::CudaAtomicAdd(&(in_grad[in_idx - index_i + sample_idx]), platform::CudaAtomicAdd(&(in_grad[in_idx - index_i + sample_idx]),
out_grad[sample_idx]); out_grad[sample_idx]);
} else { } 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];
} }
} }
} }
......
Markdown is supported
0% 或 .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑！
想要评论请 注册 或