未验证 提交 d6747a9a 编写于 作者: C chengduo 提交者: GitHub

make check_graph choosable (#13674)

test=develop
上级 824a82d7
...@@ -156,10 +156,12 @@ ParallelExecutor::ParallelExecutor( ...@@ -156,10 +156,12 @@ ParallelExecutor::ParallelExecutor(
params, member_->local_scopes_, member_->use_cuda_); params, member_->local_scopes_, member_->use_cuda_);
#endif #endif
// If the loss_var_name is given, the number of graph should be only one. if (VLOG_IS_ON(5)) {
if (loss_var_name.size()) { // If the loss_var_name is given, the number of graph should be only one.
PADDLE_ENFORCE_EQ(ir::GraphNum(*graph), 1, if (loss_var_name.size()) {
"The number of graph should be only one"); PADDLE_ENFORCE_EQ(ir::GraphNum(*graph), 1,
"The number of graph should be only one");
}
} }
if (exec_strategy.type_ == ExecutionStrategy::kDefault) { if (exec_strategy.type_ == ExecutionStrategy::kDefault) {
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册