From f4cf1035c010095093e4ebe2dd9288a4916acd84 Mon Sep 17 00:00:00 2001 From: kswang Date: Mon, 30 Nov 2020 21:45:58 +0800 Subject: [PATCH] fix getitem target error --- mindspore/core/ir/anf.cc | 10 +--------- 1 file changed, 1 insertion(+), 9 deletions(-) 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; }