Skip to content

[FEA]: Add a large-block segmented-sort kernel using the runtime-width group merge primitive #10949

Closed

Description

@anikaj-eng

Is this a duplicate?

Area

CUB

Is your feature request related to a problem? Please describe.

Fine-grained segmented-sort planning can assign 512 or 1024 cooperating threads to a segment that fits within a CTA-scale tile. A kernel is needed to consume these large-block work-plan configurations.

Describe the solution you'd like

Add a large-block segmented-sort kernel that:

Segments exceeding the supported CTA-scale capacity remain the responsibility of the fallback or separate large-segment load-balancing work.

This issue can be closed once the kernel is implemented in cub::detail and tested as a consumer of the large-block work-plan tier.

Describe alternatives you've considered

Separate sorting implementations for each width are unnecessary because they share the runtime-width primitive.

Additional context

Parent tracking issue: #10946.

Activity

  1. changed the title [-][FEA]: Add a large-segment kernel for fine-grained segmented sort[/-] [+][FEA]: Enable large-block configurations in the shared segmented-sort kernel[/+] on Aug 24, 2026
  2. changed the title [-][FEA]: Enable large-block configurations in the shared segmented-sort kernel[/-] [+][FEA]: Add a large-block segmented-sort kernel using the runtime-width group merge primitive[/+] on Aug 24, 2026
  3. kaatish commented on Aug 28, 2026

    @kaatish
    Contributor

    I will add it after the runtime-width merge sort is done.

  4. anikaj-eng commented on Nov 10, 2026

    @anikaj-eng
    ContributorAuthor

    Addressed in #12236

  5. moved this from Todo to Done in CCCLon Nov 10, 2026
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Metadata

Metadata

Assignees

Labels

No labels
No labels

Type

No type

Projects

  • Status
    Done

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions