Browse Source

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

tags/v1.0.0
lvliang 5 years ago
parent
commit
d613b0cfbb
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