|
|
|
@@ -856,6 +856,7 @@ void PsCacheManager::SyncEmbeddingTable() { |
|
|
|
|
|
|
|
bool PsCacheManager::SyncHostEmbeddingTable() { |
|
|
|
MS_ERROR_IF_NULL(embedding_host_cache_); |
|
|
|
MS_ERROR_IF_NULL(embedding_host_cache_->host_hash_map_); |
|
|
|
const auto &hash_id_to_index = embedding_host_cache_->host_hash_map_->hash_id_to_index(); |
|
|
|
size_t swap_indices_lens = hash_id_to_index.size(); |
|
|
|
if (swap_indices_lens == 0) { |
|
|
|
@@ -899,6 +900,7 @@ bool PsCacheManager::SyncHostEmbeddingTable() { |
|
|
|
bool PsCacheManager::SyncDeviceEmbeddingTable() { |
|
|
|
MS_ERROR_IF_NULL(embedding_device_cache_); |
|
|
|
const auto &device_hash_map = embedding_device_cache_->device_hash_map_; |
|
|
|
MS_ERROR_IF_NULL(device_hash_map); |
|
|
|
const auto &hash_id_to_index = device_hash_map->hash_id_to_index(); |
|
|
|
size_t swap_indices_lens = hash_id_to_index.size(); |
|
|
|
if (swap_indices_lens == 0) { |
|
|
|
|