Skip to content
This repository has been archived by the owner on Nov 25, 2024. It is now read-only.

Commit

Permalink
PR: Use top-k from RAFT (#53)
Browse files Browse the repository at this point in the history 
Closes #5 
- fix bugs in `cpp/tests/wholegraph_ops/wholegraph_csr_weighted_sample_without_replacement_tests.cu` and `cpp/tests/graph_ops/csr_add_self_loop_utils.cu`
- use `raft::warp_sort( select_k)` to impl weighted_sampling_without_replacement, when sample_count>256, cub::DeviceSegmentSort is used for the implementation.
- remove `block_radix_topk.cuh`. replace `block_topk` in file `embedding_cache_func.cuh` with `raft::warp_sort`.

Authors:
  - Chuang Zhu (https://github.com/chuangz0)

Approvers:
  - https://github.com/dongxuy04
  - Brad Rees (https://github.com/BradReesWork)

URL: #53
  • Loading branch information
@chuangz0
chuangz0 authored Aug 17, 2023
1 parent a510e3c commit 58ba5a4
Show file tree
Hide file tree
Showing 7 changed files with 482 additions and 822 deletions.
371 changes: 0 additions & 371 deletions cpp/src/wholegraph_ops/block_radix_topk.cuh
View file

This file was deleted.

Loading

0 comments on commit 58ba5a4

Please sign in to comment.