Skip to content

Conversation

@neildhar
Copy link
Contributor

Existing IR traversals for things like dataflow analysis cannot automatically descend into the non-default partitions of a WarpSpecializeOp because we do not encode the edges to these regions through RegionBranchInterface.

Changing the warpspec operations to reflect these edges allows us to clean up code in the dataflow analyses that walks the IR looking for WarpSpecializePartitionsOp, and populates its block args.

Existing IR traversals for things like dataflow analysis cannot
automatically descend into the non-default partitions of a
WarpSpecializeOp because we do not encode the edges to these regions
through RegionBranchInterface.

Changing the warpspec operations to reflect these edges allows us to
clean up code in the dataflow analyses that walks the IR looking for
`WarpSpecializePartitionsOp`, and populates its block args.
@neildhar
Copy link
Contributor Author

cc @Mogball this is a follow-up to your suggestion on #8723. I wanted to see if the desired behaviour could just fall out of setting up WarpSpecializePartitionsOp as a RegionBranchInterface. I imagine you have given this some thought, so I'm keen to get your input on doing this as opposed to explicitly handling the non-default partitions when doing dataflow analysis.

@ThomasRaoux ThomasRaoux requested a review from Mogball November 21, 2025 19:46
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