未验证 提交 5ba30402 编写于 作者: Y Yuang Liu 提交者: GitHub

fix fused bias grad add nan (#53129)

上级 82163d61
...@@ -44,7 +44,7 @@ void FusedLinearParamGradAddImpl(const Context &ctx, ...@@ -44,7 +44,7 @@ void FusedLinearParamGradAddImpl(const Context &ctx,
DenseTensor *dbias_out) { DenseTensor *dbias_out) {
constexpr bool kIsMultiPrecision = !std::is_same<T, MT>::value; constexpr bool kIsMultiPrecision = !std::is_same<T, MT>::value;
const bool fuse_bias_grad = kIsMultiPrecision && dweight_out; const bool fuse_bias_grad = false; // kIsMultiPrecision && dweight_out;
if (dweight_out) { if (dweight_out) {
phi::funcs::ComputeFusedGemmEpilogueBackward<T, T, MT>( phi::funcs::ComputeFusedGemmEpilogueBackward<T, T, MT>(
ctx, ctx,
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册