Skip to content

New 1-nn primitive using cutile - #2552

Open
divyegala wants to merge 21 commits into
NVIDIA:mainfrom
divyegala:cutile-1nn
Open

New 1-nn primitive using cutile#2552
divyegala wants to merge 21 commits into
NVIDIA:mainfrom
divyegala:cutile-1nn

Conversation

@divyegala

@divyegala divyegala commented Sep 3, 2026

Copy link
Copy Markdown
Contributor

Relies on #2548. This PR also adds the three ways to do a 1-nn behind a new primitive called top_1_nn.

@divyegala
divyegala requested review from a team as code owners September 3, 2026 21:24
@divyegala
divyegala requested a review from jameslamb September 3, 2026 21:24
@divyegala divyegala added feature request New feature or request non-breaking Introduces a non-breaking change labels Sep 3, 2026
@divyegala
divyegala removed the request for review from a team September 4, 2026 01:01
Comment thread cpp/include/cuvs/detail/jit_lto/fused_distance_nn/fused_1nn_fragments.hpp Outdated
Comment thread cpp/src/distance/fused_distance_nn-inl.cuh Outdated
Comment thread cpp/src/distance/fused_distance_nn-inl.cuh Outdated
Comment thread cpp/src/distance/fused_distance_nn-inl.cuh Outdated
Comment thread cpp/src/distance/fused_distance_nn-inl.cuh Outdated
Comment thread cpp/src/distance/fused_distance_nn-inl.cuh Outdated
Comment thread cpp/tests/neighbors/distance_nn.cu
Comment thread cpp/tests/neighbors/distance_nn_helper.cuh Outdated
Comment thread cpp/src/distance/detail/fused_distance_nn/cutile/fused_1nn_tile.hpp Outdated
Comment thread cpp/src/distance/detail/fused_distance_nn/cutile/fused_1nn_tile.hpp Outdated
Comment thread cpp/CMakeLists.txt Outdated
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

feature request New feature or request non-breaking Introduces a non-breaking change

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants