Skip to content

[Codegen][GPU] Add FuseSubgroupConsumers pass#23484

Merged
qedawkins merged 1 commit intoiree-org:mainfrom
qedawkins:fuse-subgroup-consumers
Feb 19, 2026
Merged

[Codegen][GPU] Add FuseSubgroupConsumers pass#23484
qedawkins merged 1 commit intoiree-org:mainfrom
qedawkins:fuse-subgroup-consumers

Conversation

@qedawkins
Copy link
Contributor

Add a pass that fuses tilable consumers and extract_slice ops into subgroup-scoped pcf.generic/loop ops.

Right now this pass isn't added to any pipeline to separate pipeline changes from pass introductions. Ultimately this pass will be added right around where FuseAndHoistParallelLoops is in the LLVMGPUTileAndFuse pass pipeline.

@qedawkins qedawkins force-pushed the fuse-subgroup-consumers branch 3 times, most recently from dff6fcd to c3c2021 Compare February 17, 2026 03:31
@qedawkins qedawkins marked this pull request as ready for review February 17, 2026 03:31
Copy link
Contributor

@krzysz00 krzysz00 left a comment

Choose a reason for hiding this comment

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

Looks reasonable to me

Add a pass that fuses tilable consumers and extract_slice ops into
subgroup-scoped pcf.generic/loop ops.

Right now this pass isn't added to any pipeline to separate pipeline
changes from pass introductions. Ultimately this pass will be added
right around where FuseAndHoistParallelLoops is in the
LLVMGPUTileAndFuse pass pipeline.
@qedawkins qedawkins force-pushed the fuse-subgroup-consumers branch from c3c2021 to 6aaa4bb Compare February 19, 2026 03:40
@qedawkins qedawkins enabled auto-merge (squash) February 19, 2026 03:41
@qedawkins qedawkins merged commit 038bebb into iree-org:main Feb 19, 2026
56 checks passed
@qedawkins qedawkins deleted the fuse-subgroup-consumers branch February 19, 2026 14:29
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