fixed-bug-of-embedding-look-up-run-failed-in-pynative

5 years ago · d613b0cfbb
parent 3b7217c4d9
commit d613b0cfbb
1 changed files with 3 additions and 0 deletions
--- 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<int> *te

  opt::ConstInputToAttrInfoRegister reg;
  bool reg_exist = opt::ConstInputToAttrInfoRegistry::Instance().GetRegisterByOpName(op_run_info->op_name, &reg);
+  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();