[Codegen] Forward distribution inner-tile alignment hints in forall tiling - #24852
Draft
egebeysel wants to merge 1 commit into
Conversation
egebeysel
force-pushed
the
users/egebeysel/scalable-dist-5-distribution-tiling-tests
branch
from
August 25, 2026 16:39
8e1c214 to
a990a89
Compare
egebeysel
force-pushed
the
users/egebeysel/scalable-dist-5-distribution-tiling-tests
branch
from
August 25, 2026 16:58
a990a89 to
c9e28a1
Compare
egebeysel
force-pushed
the
users/egebeysel/scalable-dist-5-distribution-tiling-tests
branch
from
August 25, 2026 17:18
c9e28a1 to
4c1273d
Compare
egebeysel
force-pushed
the
users/egebeysel/scalable-dist-5-distribution-tiling-tests
branch
from
August 25, 2026 17:23
4c1273d to
2639828
Compare
…iling Threads the distribution-level inner-tile alignment hints (set on scalable `linalg.pack`/`linalg.unpack` ops during lowering-strategy selection) into `TileAndDistributeToWorkgroupsUsingForall`, via `makeInnerTileAlignmentFn(DistributionTiles)` on both the tiling options and the consumer-fusion step. With a `Multiple` hint the workgroup tiling folds a scalable/dynamic inner tile to a clean `tile ceildiv innerTile` division instead of leaving the pack/unpack untiled. Adds lit coverage: two scalable pack/unpack cases (hint forwarded and preserved through tiling) and a dynamic-inner-tile case where the hint is load-bearing — the pack is only tiled into the forall because the hint asserts the multiple relationship the driver cannot infer from the IR. Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com> Signed-off-by: Ege Beysel <beyselege@gmail.com>
egebeysel
force-pushed
the
users/egebeysel/scalable-dist-5-distribution-tiling-tests
branch
from
August 26, 2026 15:34
2639828 to
76575bc
Compare
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
1 participant
Add this suggestion to a batch that can be applied as a single commit.This suggestion is invalid because no changes were made to the code.Suggestions cannot be applied while the pull request is closed.Suggestions cannot be applied while viewing a subset of changes.Only one suggestion per line can be applied in a batch.Add this suggestion to a batch that can be applied as a single commit.Applying suggestions on deleted lines is not supported.You must change the existing code in this line in order to create a valid suggestion.Outdated suggestions cannot be applied.This suggestion has been applied or marked resolved.Suggestions cannot be applied from pending reviews.Suggestions cannot be applied on multi-line comments.Suggestions cannot be applied while the pull request is queued to merge.Suggestion cannot be applied right now. Please check back later.
Threads the distribution-level inner-tile alignment hints (set on scalable
linalg.pack/linalg.unpackops during lowering-strategy selection) intoTileAndDistributeToWorkgroupsUsingForall, viamakeInnerTileAlignmentFn(DistributionTiles)on both the tiling options and theconsumer-fusion step. With a
Multiplehint the workgroup tiling folds ascalable/dynamic inner tile to a clean
tile ceildiv innerTiledivision insteadof leaving the pack/unpack untiled.
Adds lit coverage: two scalable pack/unpack cases (hint forwarded and preserved
through tiling) and a dynamic-inner-tile case where the hint is load-bearing —
the pack is only tiled into the forall because the hint asserts the multiple
relationship the driver cannot infer from the IR.
Co-Authored-By: Claude Opus 4.8 noreply@anthropic.com
Stack created with GitHub Stacks CLI • Give Feedback 💬