Skip to content

[WIP] Share NN-descent join and graph-maintenance kernels - #2902

Draft
divyegala wants to merge 5 commits into
codex/dedup-raft-map-ownershipfrom
codex/dedup-nn-descent-implementation
Draft

divyegala wants to merge 5 commits into
codex/dedup-raft-map-ownershipfrom
codex/dedup-nn-descent-implementation

Conversation

@divyegala

@divyegala divyegala commented Oct 22, 2026 •

Copy link
Copy Markdown
Contributor

Split reverse-edge maintenance, dense joins, BBQ joins, norms, conversion and the graph class into independent CUDA owners, with each private kernel beside its host definition.

The existing generate_inst_matrix expands ordinary JSON matrices directly into separate .cu files, with one concrete owning instantiation per generated translation unit. Launcher declarations stay in headers; host definitions and the corresponding device kernels stay in CUDA sources. Shared lower kernels have their own owners.

Prerequisites: #2913.

Diff against this PR’s parent: 21 files, +2563/−1566 lines.

Validation: native JSON/CMake generation checked on this exact branch and its dependency closure; each generated owner has one owning instantiation. Complete-library builds, device-kernel census, and focused runtime checks are tracked in #2888.

Binary savings: the previous figures measured the grouped-owner implementation and are retired for this revised layout. Per-PR CUDA 12.9 and 13.3 complete-library remeasurement is pending; increased fatbin-container overhead means the old figures must not be carried forward.

@copy-pr-bot

copy-pr-bot Bot commented Oct 22, 2026

Copy link
Copy Markdown

Auto-sync is disabled for draft pull requests in this repository. Workflows must be run manually.

Contributors can view more details about this message here.

@divyegala
divyegala force-pushed the codex/dedup-nn-descent-implementation branch from 1ca04c8 to 89fecb0 Compare October 22, 2026 21:49
@divyegala divyegala changed the title [WIP] Share NN-descent internal kernels across data types [WIP] Own NN-descent join and graph-maintenance launchers Oct 22, 2026
@divyegala
divyegala force-pushed the codex/dedup-nn-descent-implementation branch from 89fecb0 to 0ef3d45 Compare October 23, 2026 02:21
@divyegala divyegala changed the title [WIP] Own NN-descent join and graph-maintenance launchers [WIP] Share NN-descent join and graph-maintenance kernels Oct 23, 2026
@divyegala
divyegala force-pushed the codex/dedup-nn-descent-implementation branch from 0ef3d45 to f56f7c1 Compare October 23, 2026 07:46
@divyegala
divyegala changed the base branch from main to codex/dedup-raft-map-ownership October 23, 2026 07:46
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

Status: No status

Development

Successfully merging this pull request may close these issues.

1 participant