Skip to content

Commit

Permalink
Merge branch 'fix-hastable-dump-junzhang' into 'main'
Browse files Browse the repository at this point in the history
Fix HashTable::dump memset

See merge request dl/hugectr/hugectr!1473
  • Loading branch information
minseokl committed Sep 25, 2023
2 parents 3bd94e3 + f0a15ef commit 9dcbe75
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion HugeCTR/src/hashtable/nv_hashtable.cu
Original file line number Diff line number Diff line change
Expand Up @@ -271,7 +271,7 @@ void HashTable<KeyType, ValType>::dump(KeyType* d_key, ValType* d_val, size_t* d
cudaStream_t stream) const {
size_t search_length = static_cast<size_t>(capacity_ / LOAD_FACTOR);
// Before we call the kernel, set the global counter to 0
HCTR_LIB_THROW(cudaMemset(d_dump_counter, 0, sizeof(size_t)));
HCTR_LIB_THROW(cudaMemsetAsync(d_dump_counter, 0, sizeof(size_t), stream));
// grid size according to the searching length.
const int grid_size = (search_length - 1) / BLOCK_SIZE_ + 1;
// dump_kernel: dump bucket container_[0, search_length) to d_key and d_val, and report
Expand Down

0 comments on commit 9dcbe75

Please sign in to comment.