Skip to content

Enable setting thread block cluster for Hopper and later - #3168

Merged
zcbenz merged 1 commit into
ml-explore:mainfrom
zcbenz:cuda-cluster
Feb 26, 2026
Merged

zcbenz merged 1 commit into
ml-explore:mainfrom
zcbenz:cuda-cluster

Conversation

@zcbenz

@zcbenz zcbenz commented Feb 25, 2026

Copy link
Copy Markdown
Member

Make add_kernel_node capable of setting thread block cluster which will make some CUTLASS kernels run faster. Also add an overload of add_kernel_node that omits cluster and smem parameters.

@awni awni left a comment

Copy link
Copy Markdown
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM, thanks!

@zcbenz
zcbenz merged commit 6ec0192 into ml-explore:main Feb 26, 2026
16 checks passed
@zcbenz
zcbenz deleted the cuda-cluster branch February 26, 2026 07:15
@zcbenz zcbenz mentioned this pull request Feb 27, 2026
3 of 4 tasks
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants