Skip to content

[WIP] Share COO and sparse spectral helper kernels - #2923

Draft
divyegala wants to merge 5 commits into
codex/dedup-function-deps-2923from
codex/dedup-raft-coo-sort
Draft

divyegala wants to merge 5 commits into
codex/dedup-function-deps-2923from
codex/dedup-raft-coo-sort

Conversation

@divyegala

@divyegala divyegala commented Oct 22, 2026 •

Copy link
Copy Markdown
Contributor

Give COO, Laplacian, symmetrization and sparse helper templates independent CUDA owners, reusing existing map, scan and lower shared dispatch.

Register every sparse helper matrix, generate the COO kernel declarations before their consumers, and give shared masked-matrix, Laplacian-map, and CUB-merge kernels canonical owners.

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: #2937, #2906, #2913, #2921, #2938.

Diff against this PR’s parent: 40 files, +1601/−52 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-raft-coo-sort branch from 712dc48 to a5d0976 Compare October 22, 2026 21:50
@divyegala divyegala changed the title [WIP] Own shared RAFT COO canonicalization dispatch [WIP] Own shared RAFT COO sorting dispatch Oct 22, 2026
@divyegala
divyegala changed the base branch from main to codex/dedup-cuda-radix-sorting October 22, 2026 21:51
@divyegala
divyegala force-pushed the codex/dedup-cuda-radix-sorting branch from 2f352ae to 55a7deb Compare October 23, 2026 02:20
@divyegala
divyegala force-pushed the codex/dedup-raft-coo-sort branch from a5d0976 to a5f00a3 Compare October 23, 2026 02:24
@divyegala divyegala changed the title [WIP] Own shared RAFT COO sorting dispatch [WIP] Share RAFT COO and sparse spectral helpers Oct 23, 2026
@divyegala
divyegala force-pushed the codex/dedup-cuda-radix-sorting branch from 55a7deb to 5d40575 Compare October 23, 2026 07:47
@divyegala
divyegala force-pushed the codex/dedup-raft-coo-sort branch from a5f00a3 to cd348c9 Compare October 23, 2026 07:48
@divyegala divyegala changed the title [WIP] Share RAFT COO and sparse spectral helpers [WIP] Share COO and sparse spectral helper kernels Oct 23, 2026
@divyegala
divyegala changed the base branch from codex/dedup-cuda-radix-sorting to codex/dedup-function-deps-2923 October 23, 2026 07:48
@copy-pr-bot

copy-pr-bot Bot commented Oct 23, 2026

Copy link
Copy Markdown

This pull request requires additional validation before any workflows can run on NVIDIA's runners.

Pull request vetters can view their responsibilities here.

Contributors can view more details about this message here.

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