Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

[FEA] Batching NN Descent #2403

Merged
merged 49 commits into from
Aug 23, 2024
Merged

[FEA] Batching NN Descent #2403

Show file tree
Hide file tree
Changes from 1 commit
Commits
Show all changes
49 commits
Select commit Hold shift + click to select a range
b7d8d2c
nn descent batch
jinsolp Jul 31, 2024
68e3081
remove print statements
jinsolp Jul 31, 2024
acaed4e
Merge branch 'rapidsai:branch-24.10' into batch-nnd
jinsolp Jul 31, 2024
4175a0f
fix debug log
jinsolp Jul 31, 2024
581db66
trigger CI
jinsolp Jul 31, 2024
39c44f8
optimize memory usage
jinsolp Aug 1, 2024
38f1185
trigger CI
jinsolp Aug 2, 2024
91bb1d2
remove printf
jinsolp Aug 2, 2024
f4ac2d4
change tests
jinsolp Aug 6, 2024
294f493
Trigger CI
jinsolp Aug 6, 2024
5939638
Trigger CI
jinsolp Aug 7, 2024
dd55dd5
Merge branch 'rapidsai:branch-24.10' into batch-nnd
jinsolp Aug 7, 2024
15fcc39
change to merge sort and add DistEpilogue base struct
jinsolp Aug 8, 2024
6155dec
Merge branch 'rapidsai:branch-24.10' into batch-nnd
jinsolp Aug 9, 2024
45f38cd
to check nnd is problem
jinsolp Aug 9, 2024
1c81be9
optimize copying and fix overflow
jinsolp Aug 10, 2024
35e5617
change to raft managed types
jinsolp Aug 10, 2024
3d05246
remove unused code
jinsolp Aug 10, 2024
74cdf6a
for CI debugging
jinsolp Aug 11, 2024
b3a8590
debugging CI purposes
jinsolp Aug 11, 2024
d722997
for CI debugging purposes
jinsolp Aug 11, 2024
7c7817c
for debugging purposes
jinsolp Aug 11, 2024
f0f0909
last check
jinsolp Aug 12, 2024
0ee8553
remove all print statements
jinsolp Aug 12, 2024
53bf47c
remove print
jinsolp Aug 12, 2024
483fabf
Trigger CI
jinsolp Aug 12, 2024
7eb33ac
trigger CI
jinsolp Aug 12, 2024
8768b1c
Trigger CI
jinsolp Aug 12, 2024
320958d
Trigger CI
jinsolp Aug 12, 2024
460eff2
update graph nrow every iter
jinsolp Aug 13, 2024
997e225
Trigger CI
jinsolp Aug 13, 2024
9e984c7
Trigger CI
jinsolp Aug 13, 2024
de63ee3
better usage of memory
jinsolp Aug 18, 2024
9cf61af
Trigger CI
jinsolp Aug 18, 2024
8065dfe
Trigger CI
jinsolp Aug 19, 2024
f85d196
Trigger CI
jinsolp Aug 19, 2024
b8a41b0
Trigger CI
jinsolp Aug 19, 2024
3d29bbc
Trigger CI
jinsolp Aug 19, 2024
d0a4011
Trigger CI
jinsolp Aug 19, 2024
ed29707
Trigger CI
jinsolp Aug 19, 2024
e31d044
shared mem size + mdspan accessor
jinsolp Aug 19, 2024
aa6979d
std algo and rmm::exec_policy_nosync
jinsolp Aug 20, 2024
3e429c1
remove macro
jinsolp Aug 20, 2024
b9903bc
Trigger CI
jinsolp Aug 20, 2024
c7613a3
address comments
jinsolp Aug 22, 2024
0b6a370
Remove public api for testing
jinsolp Aug 22, 2024
b872901
Trigger CI
jinsolp Aug 22, 2024
f08335c
Merge branch 'rapidsai:branch-24.10' into batch-nnd
jinsolp Aug 22, 2024
9b1c509
Merge branch 'rapidsai:branch-24.10' into batch-nnd
jinsolp Aug 22, 2024
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
optimize memory usage
  • Loading branch information
@jinsolp
jinsolp committed Aug 1, 2024
commit 39c44f87392daf6087b27e2d207618bbcd4b772b
Loading
Loading