From 7f27ca73b7cecbe1aff48ba657e2fb58e91fb679 Mon Sep 17 00:00:00 2001 From: huangbingjian Date: Sat, 27 Feb 2021 18:26:24 +0800 Subject: [PATCH] fix convert_const_input_to_attr --- mindspore/ccsrc/backend/optimizer/common/helper.cc | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/mindspore/ccsrc/backend/optimizer/common/helper.cc b/mindspore/ccsrc/backend/optimizer/common/helper.cc index 5414240d48..5eb80b6c3b 100644 --- a/mindspore/ccsrc/backend/optimizer/common/helper.cc +++ b/mindspore/ccsrc/backend/optimizer/common/helper.cc @@ -516,7 +516,7 @@ void ConstInputToAttr(const CNodePtr &cnode, const std::unordered_set &i input_node = AnfAlgo::VisitKernel(input_node, 0).first; } MS_EXCEPTION_IF_NULL(input_node); - if (input_attrs.find(i) != input_attrs.end() && input_node->isa()) { + if (input_attrs.find(i) != input_attrs.end() && input_node->isa() && !HasAbstractMonad(input_node)) { auto value_node = input_node->cast(); MS_EXCEPTION_IF_NULL(value_node); MS_LOG(DEBUG) << "start erase input[" << i << "] of cnode[" + cnode->DebugString() + "]";