fix bug for duplicated graph id in pynative cases.

5 years ago · 1a997806b3
parent 163eebf7b6
commit 1a997806b3
1 changed files with 3 additions and 1 deletions
--- a/mindspore/ccsrc/pipeline/pipeline.cc
+++ b/mindspore/ccsrc/pipeline/pipeline.cc
@ -761,7 +761,9 @@ bool InitExecDatasetVm(const std::string &queue_name, int64_t size, int64_t batc
  // Convert CNodeList to LinConvertResult.
  ConfigManager::GetInstance().set_iter_num(1);
  auto runner = convert_fn({app_init});
-  backend->Link(runner.graph_id);
+  if (MsContext::GetInstance()->execution_mode() != kPynativeMode) {
+    backend->Link(runner.graph_id);
+  }
  ConfigManager::GetInstance().set_iter_num(size);

  if (!(*runner.run)) {