Skip to content
Snippets Groups Projects
Unverified Commit 7f4cbbd3 authored by Xiaoyu Zhang's avatar Xiaoyu Zhang Committed by GitHub
Browse files

fix bert backward bug (#5138)


* fix bert backward bug

* set time limit

* add time limit

Co-authored-by: default avataroneflow-ci-bot <69100618+oneflow-ci-bot@users.noreply.github.com>
Co-authored-by: default avatarjackalcooper <jackalcooper@gmail.com>
parent 234e58a3
No related branches found
No related tags found
No related merge requests found
......@@ -213,7 +213,7 @@ class BroadcastDiv : public BroadcastBinaryGrad {
}
if (y->requires_grad()) {
in_grads->at(1) =
JUST(OpInterpUtil::Dispatch<Tensor>(*y_grad_op_, {out_grads.at(0), y, z}, /*attrs=*/{}));
JUST(OpInterpUtil::Dispatch<Tensor>(*y_grad_op_, {out_grads.at(0), z, y}, /*attrs=*/{}));
}
return Maybe<void>::Ok();
}
......
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment