Skip to content

Conversation

@yaoyaoding
Copy link
Member

@yaoyaoding yaoyaoding commented Nov 5, 2025

This PR adds the support of cluster launch control instructions in Blackwell architecture.

class ClusterLaunchControlExample(tilus.Script):
    def __call__(self, ...):
        # try to cancel a pending thread block not dispatch to any SMs for execution
        self.clc.try_cancel(
            cancel_response[producer_stage],   # the shared memory slice int32[4] to store the response
            mbarrier=consumer_mbarriers[producer_stage],   # mbarrier to track the completion
            multicast=True   # whether to multicast the result and mbarrier to other blocks in the block cluster
        )

        # wait the mbarrier
        ...

        # query the result of try_cancel
        # is_valid: whether a block is canceled
        # blockIdx: the block index of the canceled block, when is_valid = True
        is_valid, blockIdx = self.clc.query_response(cancel_response[consumer_stage])   

@copy-pr-bot
Copy link

copy-pr-bot bot commented Nov 5, 2025

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.

@yaoyaoding yaoyaoding mentioned this pull request Nov 5, 2025
17 tasks
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>
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>
Signed-off-by: Yaoyao Ding <dingyaoyao.cs@gmail.com>
Signed-off-by: Yaoyao Ding <dingyaoyao.cs@gmail.com>
@yaoyaoding yaoyaoding marked this pull request as ready for review November 7, 2025 23:27
@yaoyaoding
Copy link
Member Author

/ok to test 16a4f09

@yaoyaoding yaoyaoding merged commit 3deb3d2 into main Nov 7, 2025
9 checks passed
@yaoyaoding yaoyaoding deleted the blackwell/clc branch November 7, 2025 23:56
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