[AMP] Support AMP-O2 for bfloat16 (#45541)
* support bfloat16 for amp_decorate * add check_finite for bf16 * fix bug * add ut * add ut * refine code
Showing
想要评论请 注册 或 登录
* support bfloat16 for amp_decorate * add check_finite for bf16 * fix bug * add ut * add ut * refine code