Searched refs:hash_table_addr (Results 1 – 7 of 7) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/ps/ps_cache/gpu/ |
| D | gpu_ps_cache.cc | 85 bool GPUPsCache::HashSwapOut(void *hash_table_addr, void *swap_out_value_addr, void *swap_out_index… in HashSwapOut() argument
87 MS_ERROR_IF_NULL(hash_table_addr); in HashSwapOut()
90 …DoHashSwapOut(reinterpret_cast<float *>(hash_table_addr), reinterpret_cast<float *>(swap_out_value… in HashSwapOut()
96 bool GPUPsCache::HashSwapIn(void *hash_table_addr, void *swap_in_value_addr, void *swap_in_index_ad… in HashSwapIn() argument
98 MS_ERROR_IF_NULL(hash_table_addr); in HashSwapIn()
101 …DoHashSwapIn(reinterpret_cast<float *>(hash_table_addr), reinterpret_cast<float *>(swap_in_value_a… in HashSwapIn()
|
| D | gpu_ps_cache.h | 38 …bool HashSwapOut(void *hash_table_addr, void *swap_out_value_addr, void *swap_out_index_addr, size…
40 …bool HashSwapIn(void *hash_table_addr, void *swap_in_value_addr, void *swap_in_index_addr, size_t …
|
| /third_party/mindspore/mindspore/ccsrc/ps/ps_cache/ |
| D | ps_cache_manager.cc | 711 …ager::LookUpHostHashTable(size_t embedding_size, size_t indices_lens, const float *hash_table_addr, in LookUpHostHashTable() argument
713 MS_ERROR_IF_NULL_W_RET_VAL(hash_table_addr, false); in LookUpHostHashTable()
732 … hash_table_addr, indices_addr + task_offset, output_addr + task_offset * outer_dim_size); in LookUpHostHashTable()
745 const float *insert_data, float *hash_table_addr) { in InsertHostHashTable() argument
748 MS_ERROR_IF_NULL_W_RET_VAL(hash_table_addr, false); in InsertHostHashTable()
758 … const int *insert_indices, const float *insert_data, float *hash_table_addr) { in InsertHostHashTable() argument
766 …memcpy_s(hash_table_addr + index * outer_dim_size, dest_len, insert_data + i * outer_dim_size, cop… in InsertHostHashTable()
782 … insert_indices + task_offset, insert_data + task_offset * embedding_size, hash_table_addr); in InsertHostHashTable()
807 auto hash_table_addr = reinterpret_cast<float *>(hash_info.device_address.addr); in HashSwapHostToDevice() local
821 …hash_table_addr, embedding_device_cache_->hash_swap_value_addr_, embedding_device_cache_->hash_swa… in HashSwapHostToDevice()
[all …]
|
| D | ps_cache_basic.h | 43 …virtual bool HashSwapOut(void *hash_table_addr, void *swap_out_value_addr, void *swap_out_index_ad…
45 virtual bool HashSwapIn(void *hash_table_addr, void *swap_in_value_addr, void *swap_in_index_addr,
|
| D | ps_cache_manager.h | 166 const float *insert_data, float *hash_table_addr);
167 bool LookUpHostHashTable(size_t embedding_size, size_t indices_lens, const float *hash_table_addr,
|
| /third_party/mindspore/mindspore/ccsrc/ps/ps_cache/ascend/ |
| D | ascend_ps_cache.cc | 212 bool AscendPsCache::HashSwapOut(void *hash_table_addr, void *swap_out_value_addr, void *swap_out_in… in HashSwapOut() argument
214 MS_ERROR_IF_NULL(hash_table_addr); in HashSwapOut()
241 std::make_shared<Address>(hash_table_addr, cache_vocab_size * embedding_size * sizeof(float))); in HashSwapOut()
252 bool AscendPsCache::HashSwapIn(void *hash_table_addr, void *swap_in_value_addr, void *swap_in_index… in HashSwapIn() argument
254 MS_ERROR_IF_NULL(hash_table_addr); in HashSwapIn()
281 std::make_shared<Address>(hash_table_addr, cache_vocab_size * embedding_size * sizeof(float))); in HashSwapIn()
|
| D | ascend_ps_cache.h | 61 …bool HashSwapOut(void *hash_table_addr, void *swap_out_value_addr, void *swap_out_index_addr, size…
63 …bool HashSwapIn(void *hash_table_addr, void *swap_in_value_addr, void *swap_in_index_addr, size_t …
|