Skip to content

[MLIR][Linalg] Fix FlattenElementwiseOp for linalg.broadcast#1

Open
nujaa wants to merge 2 commits into
mainfrom
upstreamFlatten
Open

[MLIR][Linalg] Fix FlattenElementwiseOp for linalg.broadcast#1
nujaa wants to merge 2 commits into
mainfrom
upstreamFlatten

Conversation

@nujaa

@nujaa nujaa commented Jul 13, 2026

Copy link
Copy Markdown
Owner

Test 'unsupported_dim_expanding_broadcast' passed with broadcast-like 'linalg.generic' but failed with a 'linalg.broadcast'. Generating an incorrect linalg.collapse_shape raising error op number of elements must be preserved upon flattening.
Limiting the check to only isPermutation broke the linalg.fill usecases, checking for the map number of results allowed it and enables 0-rank broadcasting flattening as well.

@banach, let me know if this added positive / negative test is feels necessary for you.

@nujaa
nujaa force-pushed the upstreamFlatten branch from e5842ef to 91cf238 Compare July 13, 2026 11:44
@nujaa
nujaa force-pushed the upstreamFlatten branch from 91cf238 to 73992bf Compare July 20, 2026 15:13
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