diff --git a/mindspore/core/ir/anf.cc b/mindspore/core/ir/anf.cc index 2645910314..4bc1cfc058 100644 --- a/mindspore/core/ir/anf.cc +++ b/mindspore/core/ir/anf.cc @@ -235,14 +235,6 @@ std::string GetMaketupleNodeTarget(const CNodePtr &cnode) { std::string default_target = context_ptr->get_param(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; }