未验证 提交 d0459ac8 编写于 作者: X Xin Pan 提交者: GitHub

Merge pull request #14223 from panyx0718/fix5

add more debug info.
...@@ -38,29 +38,35 @@ void CheckProgram(const ProgramDesc &program) { ...@@ -38,29 +38,35 @@ void CheckProgram(const ProgramDesc &program) {
visit[role_id] = true; visit[role_id] = true;
switch (role_id) { switch (role_id) {
case _INT(OpRole::kForward): case _INT(OpRole::kForward):
PADDLE_ENFORCE( if (visit.find(_INT(OpRole::kBackward)) != visit.end()) {
visit.find(_INT(OpRole::kBackward)) == visit.end(), LOG(ERROR)
"Cannot add forward operator before backward operator."); << "Cannot add backward operator before forward operator %s."
<< op->Type();
}
break; break;
case _INT(OpRole::kBackward): case _INT(OpRole::kBackward):
case _INT(OpRole::kBackward) | _INT(OpRole::kLoss): case _INT(OpRole::kBackward) | _INT(OpRole::kLoss):
PADDLE_ENFORCE( PADDLE_ENFORCE(
visit.find(_INT(OpRole::kOptimize)) == visit.end(), visit.find(_INT(OpRole::kOptimize)) == visit.end(),
"Cannot add backward operator before optimize operator."); "Cannot add backward operator %s before optimize operator.",
op->Type());
break; break;
case _INT(OpRole::kForward) | _INT(OpRole::kLoss): case _INT(OpRole::kForward) | _INT(OpRole::kLoss):
PADDLE_ENFORCE(visit.find(_INT(OpRole::kBackward) | PADDLE_ENFORCE(visit.find(_INT(OpRole::kBackward) |
_INT(OpRole::kLoss)) == visit.end(), _INT(OpRole::kLoss)) == visit.end(),
"Cannot add backward|loss operator before " "Cannot add backward|loss operator before "
"forward|loss operator."); "forward|loss operator %s.",
op->Type());
PADDLE_ENFORCE( PADDLE_ENFORCE(
visit.find(_INT(OpRole::kOptimize)) == visit.end(), visit.find(_INT(OpRole::kOptimize)) == visit.end(),
"Cannot add backward operator before optimize operator."); "Cannot add forward|loss operator %s after optimize operator.",
op->Type());
break; break;
case _INT(OpRole::kOptimize): case _INT(OpRole::kOptimize):
case _INT(OpRole::kOptimize) | _INT(OpRole::kLRSched): case _INT(OpRole::kOptimize) | _INT(OpRole::kLRSched):
PADDLE_ENFORCE(visit.find(_INT(OpRole::kBackward)) != visit.end(), PADDLE_ENFORCE(visit.find(_INT(OpRole::kBackward)) != visit.end(),
"Optimize operators must follow backward operator."); "Optimize operators %s must follow backward operator.",
op->Type());
break; break;
case _INT(OpRole::kLRSched): case _INT(OpRole::kLRSched):
case _INT(OpRole::kDist): case _INT(OpRole::kDist):
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册