Skip to content

Preserve input memory location for NN Descent #1928

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

Merged
rapids-bot merged 25 commits into from
May 20, 2026
+232 −171
Merged

Preserve input memory location for NN Descent #1928

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
25 commits
Select commit Hold shift + click to select a range
666a688
keeping input data mem location
jinsolp Mar 18, 2026
f4cd395
Merge branch 'rapidsai:main' into nnd-keep-input-data-mem
jinsolp Mar 18, 2026
2855e58
compute norms in fp32 if data is fp32
jinsolp Mar 18, 2026
fbb9023
change padding for uint8 and int8
jinsolp Mar 19, 2026
0c1ff7d
Merge branch 'main' into nnd-keep-input-data-mem
jinsolp Mar 20, 2026
157fe30
rm print and revert test
jinsolp Mar 20, 2026
ace6cae
Merge branch 'main' into nnd-keep-input-data-mem
jinsolp Apr 2, 2026
686eb4b
Merge branch 'main' into nnd-keep-input-data-mem
jinsolp Apr 6, 2026
e59e6c7
fix c abi breakages
jinsolp Apr 8, 2026
632c24b
Merge branch 'main' into nnd-keep-input-data-mem
jinsolp Apr 8, 2026
836ef94
Merge branch 'main' into nnd-keep-input-data-mem
cjnolet Apr 9, 2026
298943e
Merge branch 'main' into nnd-keep-input-data-mem
jinsolp Apr 28, 2026
81fdb69
Merge branch 'main' into nnd-keep-input-data-mem
jinsolp May 13, 2026
be9f30b
rename to use_fp16_dist_comp
jinsolp May 13, 2026
1c7589f
change to match ivfpq style with internal_distance_dtype
jinsolp May 13, 2026
080a32c
raft::ceildiv
jinsolp May 13, 2026
27c918e
merge commit
jinsolp May 13, 2026
9a04366
deprecation logs and CUVS_EXPORT
jinsolp May 13, 2026
1a04d6e
rm comments
jinsolp May 13, 2026
4903f8a
Merge branch 'main' into nnd-keep-input-data-mem
jinsolp May 13, 2026
8d9844c
comment with issues
jinsolp May 14, 2026
931ca58
revert to focusing only on mem allocations
jinsolp May 14, 2026
2add75a
constexpr half
jinsolp May 20, 2026
da392c1
unified struct for bytes
jinsolp May 20, 2026
60e9780
Merge branch 'release/26.06' into nnd-keep-input-data-mem
jinsolp May 20, 2026
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
Loading
Loading