Browse Source

!5326 Fixed bug of embedding look up run failed in pynative

Merge pull request !5326 from JoyLvliang/fixed-bug-of-embedding-look-up-run-failed-in-pynative
tags/v1.0.0
mindspore-ci-bot Gitee 5 years ago
parent
commit
e5409647bf
1 changed files with 3 additions and 0 deletions
  1. +3
    -0
      mindspore/ccsrc/pipeline/pynative/pynative_execute.cc

+ 3
- 0
mindspore/ccsrc/pipeline/pynative/pynative_execute.cc View File

@@ -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();


Loading…
Cancel
Save