Skip to content

[WIP] Share CUB histogram launchers - #2897

Draft
divyegala wants to merge 3 commits into
codex/dedup-cub-ptx-queryfrom
codex/dedup-cub-label-histogram
Draft

divyegala wants to merge 3 commits into
codex/dedup-cub-ptx-queryfrom
codex/dedup-cub-label-histogram

Conversation

@divyegala

@divyegala divyegala commented Oct 22, 2026 •

Copy link
Copy Markdown
Contributor

Own CUB histogram dispatch in its CUDA launcher and route label counting through that owner.

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.

Diff against this PR’s parent: 16 files, +959/−103 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 changed the title [WIP] Share CUB label-histogram launchers in balanced k-means [WIP] Own shared CUB histogram dispatch Oct 22, 2026
@divyegala
divyegala force-pushed the codex/dedup-cub-label-histogram branch from 86b5b67 to 098886f Compare October 23, 2026 02:20
@divyegala divyegala changed the title [WIP] Own shared CUB histogram dispatch [WIP] Share CUB histogram kernels Oct 23, 2026
@divyegala
divyegala changed the base branch from main to codex/dedup-cuda-radix-sorting October 23, 2026 02:20
@divyegala
divyegala force-pushed the codex/dedup-cub-label-histogram branch from 098886f to dac4f92 Compare October 23, 2026 07:46
@divyegala divyegala changed the title [WIP] Share CUB histogram kernels [WIP] Share CUB histogram launchers Oct 23, 2026
@divyegala
divyegala changed the base branch from codex/dedup-cuda-radix-sorting to codex/dedup-cub-ptx-query 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