@@ -40,27 +40,32 @@ void CheckProgram(const ProgramDesc &program) {
40
40
case _INT (OpRole::kForward ):
41
41
PADDLE_ENFORCE (
42
42
visit.find (_INT (OpRole::kBackward )) == visit.end (),
43
- " Cannot add forward operator before backward operator." );
43
+ " Cannot add backward operator before forward operator %s." ,
44
+ op->Type ());
44
45
break ;
45
46
case _INT (OpRole::kBackward ):
46
47
case _INT (OpRole::kBackward ) | _INT (OpRole::kLoss ):
47
48
PADDLE_ENFORCE (
48
49
visit.find (_INT (OpRole::kOptimize )) == visit.end (),
49
- " Cannot add backward operator before optimize operator." );
50
+ " Cannot add backward operator %s before optimize operator." ,
51
+ op->Type ());
50
52
break ;
51
53
case _INT (OpRole::kForward ) | _INT (OpRole::kLoss ):
52
54
PADDLE_ENFORCE (visit.find (_INT (OpRole::kBackward ) |
53
55
_INT (OpRole::kLoss )) == visit.end (),
54
56
" Cannot add backward|loss operator before "
55
- " forward|loss operator." );
57
+ " forward|loss operator %s." ,
58
+ op->Type ());
56
59
PADDLE_ENFORCE (
57
60
visit.find (_INT (OpRole::kOptimize )) == visit.end (),
58
- " Cannot add backward operator before optimize operator." );
61
+ " Cannot add forward|loss operator %s after optimize operator." ,
62
+ op->Type ());
59
63
break ;
60
64
case _INT (OpRole::kOptimize ):
61
65
case _INT (OpRole::kOptimize ) | _INT (OpRole::kLRSched ):
62
66
PADDLE_ENFORCE (visit.find (_INT (OpRole::kBackward )) != visit.end (),
63
- " Optimize operators must follow backward operator." );
67
+ " Optimize operators %s must follow backward operator." ,
68
+ op->Type ());
64
69
break ;
65
70
case _INT (OpRole::kLRSched ):
66
71
case _INT (OpRole::kDist ):
0 commit comments