Browse Source

!4164 fix a bug with wrong call SyncDeviceToHost in pynative mode

Merge pull request !4164 from lvchangquan/master
tags/v0.7.0-beta
mindspore-ci-bot Gitee 5 years ago
parent
commit
35b16d8ca4
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      mindspore/ccsrc/backend/session/session_basic.cc

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

@@ -863,7 +863,7 @@ bool TensorNeedSync(const AnfNodePtr &parameter, const tensor::TensorPtr &tensor
auto ms_context = MsContext::GetInstance();
MS_EXCEPTION_IF_NULL(ms_context);
auto device_address = AnfAlgo::GetMutableOutputAddr(parameter, 0);
if (ms_context->enable_pynative_infer()) {
if (ms_context->execution_mode() == kPynativeMode) {
return tensor->device_address().get() == nullptr || tensor->device_address() != device_address;
}
if (tensor->is_dirty()) {


Loading…
Cancel
Save