|
|
|
@ -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;
|
|
|
|
|
}
|
|
|
|
|