Remove tensor copy in the update_loss_scaling op. (#29426)
* remove tensor copy in the update_loss_scaling op * not use thrust. * fix some cuda memory access error.
Showing
想要评论请 注册 或 登录
Fork自 PaddlePaddle / Paddle
* remove tensor copy in the update_loss_scaling op * not use thrust. * fix some cuda memory access error.