fix getitem target error

pull/9233/head
kswang 4 years ago
parent 4bd9460e45
commit f4cf1035c0

@ -235,14 +235,6 @@ std::string GetMaketupleNodeTarget(const CNodePtr &cnode) {
std::string default_target = context_ptr->get_param<std::string>(MS_CTX_DEVICE_TARGET);
return default_target;
}
std::string GetTupleGetItemTarget(const CNodePtr &cnode, const PrimitivePtr &primitive) {
MS_EXCEPTION_IF_NULL(cnode);
MS_EXCEPTION_IF_NULL(primitive);
auto input_target = GetCNodeTarget(cnode->input(1));
primitive->set_attr("primitive_target", MakeValue(input_target));
return input_target;
}
} // namespace
std::string GetCNodeTarget(const AnfNodePtr &node) {
@ -297,7 +289,7 @@ std::string GetCNodeTarget(const AnfNodePtr &node) {
} else if (IsPrimitiveCNode(node, prim::kPrimMakeTuple)) {
return GetMaketupleNodeTarget(cnode);
} else if (IsPrimitiveCNode(node, prim::kPrimTupleGetItem)) {
return GetTupleGetItemTarget(cnode, primitive);
return GetCNodeTarget(cnode->input(1));
}
return default_target;
}

Loading…
Cancel
Save