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); }