diff --git a/mindspore/ccsrc/pipeline/pynative/pynative_execute.cc b/mindspore/ccsrc/pipeline/pynative/pynative_execute.cc index 7a8faa8a2857985c23bce767e0b73b16cb2c2e3f..3a30003ec4c90c8c96a87c136f899dbdc30716d0 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();