Skip to content

[IR][Script] Allow specify cluster dimensions through cluster_blocks#33

Merged
yaoyaoding merged 4 commits intomainfrom
yaoyao/support-block-cluster
Sep 5, 2025
Merged

[IR][Script] Allow specify cluster dimensions through cluster_blocks#33
yaoyaoding merged 4 commits intomainfrom
yaoyao/support-block-cluster

Conversation

@yaoyaoding
Copy link
Copy Markdown
Member

This PR allows Tilus Script to specify the cluster dimensions, a feature occured in Hopper and later GPUs.

class Kernel(tilus.Script):
    def __call__(self):
        self.attrs.blocks = ...
        self.attrs.cluster_blocks = [2, 2, 1]  # must be compilation-time constants
        self.attrs.warps = 4

Signed-off-by: Yaoyao Ding <dingyaoyao.cs@gmail.com>
Signed-off-by: Yaoyao Ding <dingyaoyao.cs@gmail.com>
.
Signed-off-by: Yaoyao Ding <dingyaoyao.cs@gmail.com>
Signed-off-by: Yaoyao Ding <dingyaoyao.cs@gmail.com>
@yaoyaoding yaoyaoding merged commit fe4a9e6 into main Sep 5, 2025
8 checks passed
@yaoyaoding yaoyaoding deleted the yaoyao/support-block-cluster branch September 5, 2025 19:14
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.

1 participant