From 56d7dd294b2fdbab1014028441a0d4a164dcdc5e Mon Sep 17 00:00:00 2001 From: laiyongqiang Date: Sat, 9 Jan 2021 15:30:02 +0800 Subject: [PATCH] adapt to None Type Input --- mindspore/ccsrc/backend/optimizer/somas/somas.cc | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/mindspore/ccsrc/backend/optimizer/somas/somas.cc b/mindspore/ccsrc/backend/optimizer/somas/somas.cc index 5dbe6d1441..5e6a922666 100644 --- a/mindspore/ccsrc/backend/optimizer/somas/somas.cc +++ b/mindspore/ccsrc/backend/optimizer/somas/somas.cc @@ -214,6 +214,11 @@ void Somas::InitCommonNodeInputs(bool is_all_nop_node, const CNodePtr &kernel) { } if (!AnfAlgo::IsRealCNodeKernel(prenode_index.first)) { + auto op_name = AnfAlgo::GetCNodeName(kernel); + TypeId input_origin_type = AnfAlgo::GetPrevNodeOutputInferDataType(kernel, i); + if ((op_name == kDynamicRNNOpName || op_name == kDynamicGRUV2OpName) && input_origin_type == kMetaTypeNone) { + continue; + } auto parameter = GetSomasParameters(prenode_index.first, prenode_index.second); node->input_parameters_map_[i] = parameter; MS_LOG(DEBUG) << "Input [" << prenode_index.first->fullname_with_scope() << "] is not a real cnode kernel.";