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

fix issue I3EAB6: exception while using unique ops

This commit is contained in:
zhangzhaoju 2021-03-31 10:01:46 +08:00 committed by chujinjin
parent 9263c4d63b
commit b82529e26f
2 changed files with 21 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

27
mindspore/ccsrc/pipeline/jit/prim_bprop_optimizer.cc
View File

@ -94,16 +94,24 @@ void PrimBpropOptGraphLevel2Info::AnalysisNodeUsingInfo(
arg_info.using_flg_ = true;
MS_LOG(DEBUG) << "param:" << param->ToString() << " used by node:" << user_node->ToString();
if (!IsPrimitiveCNode(user_node, prim::kPrimTupleGetItem)) {
MS_LOG(EXCEPTION) << "tuple param:" << param->ToString() << " of bp_graph:" << opt_func_graph_->ToString()
<< " unexpect used by node:" << user_node->ToString();
for (auto &sub_info : arg_info.sub_using_info_) {
sub_info.using_flg_ = true;
}
} else {
AalysisForTupleGetItem(node_users, param, arg_info, user_node);
}
auto cnode = user_node->cast<CNodePtr>();
if (cnode->size() != 3) {
MS_LOG(EXCEPTION) << "TupleGetItem Node:" << user_node->ToString() << " of bp_graph:"
<< opt_func_graph_->ToString()
<< "input size is:" << cnode->size();
}
auto idx_node = cnode->input(2);
}
}
void PrimBpropOptGraphLevel2Info::AalysisForTupleGetItem(const NodeUsersMap &node_users,
const std::shared_ptr<AnfNode> &param,
ParamUsingInfo &arg_info,
const AnfNodePtr &user_node) const {
auto cnode = user_node->cast<CNodePtr>();
if (cnode->size() != 3) {
MS_LOG(EXCEPTION) << "TupleGetItem Node:" << user_node->ToString() << " of bp_graph:" << opt_func_graph_->ToString()
<< "input size is:" << cnode->size();
}
auto idx_node = cnode->input(2);
if (!idx_node->isa<ValueNode>()) {
MS_LOG(EXCEPTION) << "tuple :" << param->ToString() << " of bp_graph:" << opt_func_graph_->ToString()
<< " unexpect used by node:" << user_node->ToString() << " TupleGetItem idx node:"
@ -125,7 +133,6 @@ void PrimBpropOptGraphLevel2Info::AnalysisNodeUsingInfo(
if (arg_info.tuple_flg_) {
AnalysisNodeUsingInfo(node_users, cnode, arg_info.sub_using_info_[idx]);
}
}
}
void PrimBpropOptGraphLevel2Info::ArgInfoRefresh(

5
mindspore/ccsrc/pipeline/jit/prim_bprop_optimizer.h
View File

@ -119,7 +119,10 @@ private:
void TryFreeOneValue(const ValuePtrList &op_args, const std::vector<ParamUsingInfo> &param_info_vec);
private:
void AalysisForTupleGetItem(const NodeUsersMap &node_users, const std::shared_ptr<AnfNode> &param,
ParamUsingInfo &arg_info, const AnfNodePtr &user_node) const;
private:
// the level2 opt func_graph
FuncGraphPtr opt_func_graph_;
// to indicate arguments value using or not, if not using should free device memory