Browse Source

fix loadinputdata for pynative

tags/v0.5.0-beta
chujinjin 5 years ago
parent
commit
146861f605
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