Browse Source

!1966 fix load input data repeatedly in PyNative mode

Merge pull request !1966 from chujinjin/fix_loadinputdata_for_pynative
tags/v0.5.0-beta
mindspore-ci-bot Gitee 6 years ago
parent
commit
68d760d5d3
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      mindspore/ccsrc/session/session_basic.cc

+ 1
- 1
mindspore/ccsrc/session/session_basic.cc View File

@@ -686,7 +686,7 @@ void SessionBasic::LoadInputData(const std::shared_ptr<KernelGraph> &kernel_grap
}
}
if (need_sync) {
if (AnfAlgo::IsParameterWeight(pk_node)) {
if (ms_context->execution_mode() == kPynativeMode || AnfAlgo::IsParameterWeight(pk_node)) {
tensor->set_device_address(device_address);
}
MS_EXCEPTION_IF_NULL(device_address);


Loading…
Cancel
Save