!13835 fixed memcpy error
From: @anancds Reviewed-by: @cristoval,@limingqi107 Signed-off-by: @limingqi107
This commit is contained in:
commit
a59f1814c1
|
@ -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…
Reference in New Issue