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

add graph number check (#14025)

test=develop
上级 6b4056bb
...@@ -156,12 +156,10 @@ ParallelExecutor::ParallelExecutor( ...@@ -156,12 +156,10 @@ ParallelExecutor::ParallelExecutor(
params, member_->local_scopes_, member_->use_cuda_); params, member_->local_scopes_, member_->use_cuda_);
#endif #endif
if (VLOG_IS_ON(5)) { // If the loss_var_name is given, the number of graph should be only one.
// If the loss_var_name is given, the number of graph should be only one. if (loss_var_name.size()) {
if (loss_var_name.size()) { PADDLE_ENFORCE_EQ(ir::GraphNum(*graph), 1,
PADDLE_ENFORCE_EQ(ir::GraphNum(*graph), 1, "The number of graph should be only one");
"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.
先完成此消息的编辑!
想要评论请 注册