fixed memcpy error

4 years ago · 20d881c5be
parent bfcf4562ad
commit 20d881c5be
1 changed files with 5 additions and 0 deletions
--- a/mindspore/ccsrc/runtime/device/kernel_runtime_manager.cc
+++ b/mindspore/ccsrc/runtime/device/kernel_runtime_manager.cc
@ -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);
 }