From 20d881c5be2a56717cda3d68e8d1c8db38d07634 Mon Sep 17 00:00:00 2001
From: chendongsheng <chendongsheng1@huawei.com>
Date: Mon, 22 Mar 2021 21:50:25 +0800
Subject: [PATCH] fixed memcpy error

---
 mindspore/ccsrc/runtime/device/kernel_runtime_manager.cc | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/mindspore/ccsrc/runtime/device/kernel_runtime_manager.cc b/mindspore/ccsrc/runtime/device/kernel_runtime_manager.cc
index c19a026860..90b43369c9 100644
--- 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);
 }