diff --git a/mindspore/ccsrc/pipeline/pynative/pynative_execute.cc b/mindspore/ccsrc/pipeline/pynative/pynative_execute.cc index 6407ca986f561c112c7362edea07ff8c33dee011..03399586ce45ff72482c3d9b88e91a7888555f4e 100644 --- a/mindspore/ccsrc/pipeline/pynative/pynative_execute.cc +++ b/mindspore/ccsrc/pipeline/pynative/pynative_execute.cc @@ -522,6 +522,9 @@ void ConstructInputTensor(const OpExecInfoPtr &op_run_info, std::vector *te opt::ConstInputToAttrInfoRegister reg; bool reg_exist = opt::ConstInputToAttrInfoRegistry::Instance().GetRegisterByOpName(op_run_info->op_name, ®); + if (op_run_info->op_name == prim::kPrimEmbeddingLookup->name()) { + reg_exist = false; + } op_prim->BeginRecordAddAttr(); size_t input_num = op_run_info->op_inputs.size();