Skip to content

[Backport to 3.4] Use plus tunings for lookahead scan more widely (#9813) - #9848

Merged
wmaxey merged 2 commits into
NVIDIA:branch/3.4.xfrom
bernhardmgruber:backport_plus_tune
Jul 14, 2026
Merged

[Backport to 3.4] Use plus tunings for lookahead scan more widely (#9813)#9848
wmaxey merged 2 commits into
NVIDIA:branch/3.4.xfrom
bernhardmgruber:backport_plus_tune

Conversation

@bernhardmgruber

@bernhardmgruber bernhardmgruber commented Jul 14, 2026

Copy link
Copy Markdown
Contributor

Fixes: #9811

(cherry picked from commit 22e4e68)

Fixes: NVIDIA#9811

Co-authored-by: Paul Grosse-Bley <pauleonix-dev@pm.me>

(cherry picked from commit 22e4e68)
@bernhardmgruber
bernhardmgruber requested a review from a team as a code owner July 14, 2026 07:17
@bernhardmgruber
bernhardmgruber requested a review from elstehle July 14, 2026 07:17
@github-project-automation github-project-automation Bot moved this to Todo in CCCL Jul 14, 2026
@cccl-authenticator-app cccl-authenticator-app Bot moved this from Todo to In Review in CCCL Jul 14, 2026
@bernhardmgruber bernhardmgruber changed the title Use plus tunings for lookahead scan more widely (#9813) [Backport to 3.4] Use plus tunings for lookahead scan more widely (#9813) Jul 14, 2026
@coderabbitai

coderabbitai Bot commented Jul 14, 2026

Copy link
Copy Markdown
Contributor

Review Change Stack

No actionable comments were generated in the recent review. 🎉

ℹ️ Recent review info
⚙️ Run configuration

Configuration used: Path: .coderabbit.yaml

Review profile: CHILL

Plan: Enterprise

Run ID: 9262d032-79ce-47a2-9ca8-eaf4cc4d02b9

📥 Commits

Reviewing files that changed from the base of the PR and between 305c95b and 03600bc.

📒 Files selected for processing (1)
  • cub/cub/device/dispatch/tuning/tuning_scan.cuh

📝 Walkthrough

Summary by CodeRabbit

  • Performance Improvements
    • Expanded warpspeed scan tuning support to additional scan operations when the accumulator meets the required compatibility conditions.
    • Enabled more input value sizes to use optimized SM10+ scan policies.

Walkthrough

The SM10+ warpspeed tuning gate now depends only on accumulator primitiveness or trivial copy constructibility, allowing matching policies for non-plus scan operations.

Changes

Warpspeed scan tuning

Layer / File(s) Summary
Broaden warpspeed policy selection
cub/cub/device/dispatch/tuning/tuning_scan.cuh
The SM10+ warpspeed policy switch no longer requires operation_t to be plus; the accumulator constraint remains enforced.

Assessment against linked issues

Objective Addressed Explanation
Restore scan_exclusive_custom performance on Blackwell [#9811] The tuning gate now covers non-plus operations, but no benchmark or regression-test evidence is provided to confirm the reported performance regression is resolved.

Suggested labels: backport branch/3.4.x

Suggested reviewers: elstehle, gonidelis, miscco


Comment @coderabbitai help to get the list of available commands.

@github-actions

This comment has been minimized.

@bernhardmgruber

Copy link
Copy Markdown
Contributor Author

pre-commit.ci autofix

@bernhardmgruber

Copy link
Copy Markdown
Contributor Author

/ok to test a72fe92

@github-actions

Copy link
Copy Markdown
Contributor

🥳 CI Workflow Results

🟩 Finished in 1h 13m: Pass: 100%/283 | Total: 1d 21h | Max: 43m 26s | Hits: 100%/196179

See results here.

@wmaxey
wmaxey merged commit 3b5d0e4 into NVIDIA:branch/3.4.x Jul 14, 2026
302 of 305 checks passed
@github-project-automation github-project-automation Bot moved this from In Review to Done in CCCL Jul 14, 2026
@bernhardmgruber
bernhardmgruber deleted the backport_plus_tune branch July 14, 2026 16:15
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

Archived in project

Development

Successfully merging this pull request may close these issues.

3 participants