p47068915
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Add log which indicates whether the func_graph has control-flow nodes.

This commit is contained in:
liangzelang 2021-12-15 11:28:11 +08:00
parent d81ae6cd09
commit 7866ffc8be
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
mindspore/ccsrc/pipeline/jit/action.cc
View File

@ -757,6 +757,9 @@ bool TaskEmitAction(const ResourcePtr &res) {
std::string device_target = context_ptr->get_param<std::string>(MS_CTX_DEVICE_TARGET);
auto manager = func_graph->manager();
auto graphs = manager->func_graphs();
if (graphs.size() > 1 && device_target == kAscendDevice) {
MS_LOG(INFO) << "This func_graph has control flow nodes, owns " << graphs.size() << " subgraphs.";
}
bool exist_while =
std::any_of(graphs.cbegin(), graphs.cend(), [](const FuncGraphPtr &fg) { return fg->recursive(); });
if (device_target == kAscendDevice && backend != kMsVm && !exist_while) {