Fix cannot find backend node.

This commit is contained in:
gaoyong10 2021-07-10 22:22:22 +08:00
parent 6dca76d4eb
commit 04a8f18772
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;