| /third_party/mindspore/mindspore/ccsrc/ps/ps_cache/ |
| D | embedding_hash_map.cc | 28 auto hash_index = FindInsertionPos(data_step, graph_running_step, &need_swap, need_wait_graph); in ParseData() local
29 if (hash_index == INVALID_INDEX_VALUE) { in ParseData()
30 return hash_index; in ParseData()
35 (void)hash_id_to_index_.emplace(id, hash_index); in ParseData()
36 hash_map_elements_[hash_index].set_id(id); in ParseData()
37 hash_map_elements_[hash_index].set_step(data_step); in ParseData()
38 return hash_index; in ParseData()
41 swap_out_index[*swap_out_size] = hash_index; in ParseData()
42 swap_out_ids[*swap_out_size] = hash_map_elements_[hash_index].id_; in ParseData()
44 (void)hash_id_to_index_.erase(hash_map_elements_[hash_index].id_); in ParseData()
[all …]
|
| D | embedding_hash_map.h | 64 size_t hash_step(const int hash_index) const { return hash_map_elements_[hash_index].step_; } in hash_step() argument
65 …void set_hash_step(const int hash_index, const size_t step) { hash_map_elements_[hash_index].set_s… in set_hash_step() argument
|
| D | ps_cache_manager.cc | 385 std::unique_ptr<int[]> hash_index = std::make_unique<int[]>(batch_ids_len); in ProcessData() local
386 MS_ERROR_IF_NULL_W_RET_VAL(hash_index, false); in ProcessData()
392 RETURN_IF_FALSE(ParseData(batch_ids, batch_ids_len, hash_index.get())); in ProcessData()
408 if (memcpy_s(data, dest_len, hash_index.get(), data_size) != EOK) { in ProcessData()
425 …ger::CheckCacheHitOrOutRangeTask(const int *batch_ids, const size_t batch_ids_len, int *hash_index, in CheckCacheHitOrOutRangeTask() argument
428 MS_ERROR_IF_NULL(hash_index); in CheckCacheHitOrOutRangeTask()
438 hash_index[i] = batch_ids[i] - vocab_cache_size_diff_; in CheckCacheHitOrOutRangeTask()
443 hash_index[i] = batch_ids[i] + cache_indices_bounds_.second; in CheckCacheHitOrOutRangeTask()
449 hash_index[i] = iter->second + cache_indices_bounds_.first; in CheckCacheHitOrOutRangeTask()
460 …Manager::CheckCacheHitOrOutRange(const int *batch_ids, const size_t batch_ids_len, int *hash_index, in CheckCacheHitOrOutRange() argument
[all …]
|
| D | ps_cache_manager.h | 154 bool ParseData(const int *batch_ids, const size_t batch_ids_len, int *hash_index);
156 …ceData(size_t id, bool *need_swap_device_to_host, bool *need_swap_host_to_device, int *hash_index);
177 …tOrOutRangeTask(const int *batch_ids, const size_t batch_ids_len, int *hash_index, bool *in_device,
179 …bool CheckCacheHitOrOutRange(const int *batch_ids, const size_t batch_ids_len, int *hash_index, bo…
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/cuda_impl/ |
| D | hash_impl.cu | 24 int hash_index = swap_out_index[i]; in HashSwapOut() local
26 swap_out_value[i * hash_dim + j] = hash_table[hash_index * hash_dim + j]; in HashSwapOut()
36 int hash_index = swap_in_index[i]; in HashSwapIn() local
38 hash_table[hash_index * hash_dim + j] = swap_in_value[i * hash_dim + j]; in HashSwapIn()
|
| /third_party/grpc/src/core/ext/transport/chttp2/transport/ |
| D | hpack_encoder.cc | 125 HpackEncoderSlotHash hash_index) { in HpackIndex() argument
126 return hashtable[hash_index].index; in HpackIndex()
131 HpackEncoderSlotHash hash_index) { in GetEntry() argument
132 return hashtable[hash_index].value; in GetEntry()
137 HpackEncoderSlotHash hash_index) { in TableEmptyAt() argument
138 return Cmp::Equals(hashtable[hash_index].value, Cmp::Null()); in TableEmptyAt()
143 HpackEncoderSlotHash hash_index) { in Matches() argument
144 return Cmp::Equals(value, hashtable[hash_index].value); in Matches()
148 static void UpdateIndex(Hashtable* hashtable, HpackEncoderSlotHash hash_index, in UpdateIndex() argument
150 hashtable[hash_index].index = hpack_index; in UpdateIndex()
[all …]
|
| /third_party/mesa3d/src/gallium/auxiliary/gallivm/ |
| D | lp_bld_format_s3tc.c | 1126 LLVMValueRef hash_index, in s3tc_store_cached_block() argument
1137 indices[2] = hash_index; in s3tc_store_cached_block()
1142 hash_index = LLVMBuildMul(builder, hash_index, in s3tc_store_cached_block()
1145 indices[2] = hash_index; in s3tc_store_cached_block()
1149 hash_index = LLVMBuildAdd(builder, hash_index, in s3tc_store_cached_block()
1919 LLVMValueRef hash_index; in generate_update_cache_one_block() local
1925 hash_index = LLVMGetParam(function, 1); in generate_update_cache_one_block()
1929 lp_build_name(hash_index, "hash_index"); in generate_update_cache_one_block()
1967 s3tc_store_cached_block(gallivm, col, tag_value, hash_index, cache); in generate_update_cache_one_block()
1982 LLVMValueRef hash_index, in update_cached_block() argument
[all …]
|
| /third_party/mesa3d/src/panfrost/bifrost/ |
| D | bi_opt_cse.c | 43 hash_index(uint32_t hash, bi_index index) in hash_index() function
70 hash = hash_index(hash, I->src[s]); in hash_instr()
|