Skip to content

Commit

Permalink
d1
Browse files Browse the repository at this point in the history
  • Loading branch information
rhdong committed Jul 5, 2023
1 parent 7a8cab2 commit 178ec1d
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion tests/merlin_hashtable_test.cc.cu
Original file line number Diff line number Diff line change
Expand Up @@ -2450,7 +2450,7 @@ void test_bucket_size(bool load_scores = true) {
options.max_capacity = MAX_CAPACITY;
options.dim = DIM;
options.max_hbm_for_vectors = nv::merlin::GB(16);
options.evict_strategy = nv::merlin::EvictStrategy::kCustomized;
using Table = nv::merlin::HashTable<K, V, S, EvictStrategy::kCustomized>;

CUDA_CHECK(cudaMallocHost(&h_keys, KEY_NUM * sizeof(K)));
CUDA_CHECK(cudaMallocHost(&h_scores, KEY_NUM * sizeof(S)));
Expand Down

0 comments on commit 178ec1d

Please sign in to comment.