!19970 Fix cannot find backend node.

Merge pull request !19970 from gaoyong10/runtime1.36
This commit is contained in:
i-robot 2021-07-10 19:05:59 +00:00 committed by Gitee
commit d61f2b3162
1 changed files with 5 additions and 2 deletions

View File

@ -925,8 +925,11 @@ void ControlNodeParser::FetchFrontValueNode(const std::vector<AnfNodePtr> &contr
}
if (front_to_backend_parameters_.find(parameters[i - kCallInputStartPos]) ==
front_to_backend_parameters_.end()) {
MS_LOG(EXCEPTION) << "Cannot find backend parameter for front parameter:"
<< AnfAlgo::GetNodeDebugString(parameters[i - kCallInputStartPos]);
MS_LOG(INFO) << "Cannot find backend parameter for front parameter:"
<< AnfAlgo::GetNodeDebugString(parameters[i - kCallInputStartPos])
<< ", used the default format";
CreateDeviceTensorForFrontParameter(inputs[i], device_contexts[0]);
front_value_nodes_.push_back({inputs[i], device_contexts[0]});
}
const auto &backend_node = front_to_backend_parameters_[parameters[i - kCallInputStartPos]].first;