Browse Source

!13831 fixed memcpy error

From: @anancds
Reviewed-by: @cristoval,@limingqi107
Signed-off-by: @limingqi107
tags/v1.2.0-rc1
mindspore-ci-bot Gitee 5 years ago
parent
commit
f30a0281a5
1 changed files with 5 additions and 0 deletions
  1. +5
    -0
      mindspore/ccsrc/runtime/device/kernel_runtime_manager.cc

+ 5
- 0
mindspore/ccsrc/runtime/device/kernel_runtime_manager.cc View File

@@ -125,6 +125,11 @@ void KernelRuntimeManager::ReleaseKernelRuntime(const std::string &device_name,
if (runtime == nullptr) {
return;
}
#if (ENABLE_CPU && (ENABLE_D || ENABLE_GPU))
if (ps::PSContext::instance()->is_worker() && ps::PsDataPrefetch::GetInstance().cache_enable()) {
ps::ps_cache_instance.SyncEmbeddingTable();
}
#endif
runtime->ReleaseDeviceRes();
runtime_map_.erase(runtime_iter);
}


Loading…
Cancel
Save