Skip to content

[xegpu] Elementwise schedule and refactoring#200

Open
tkarna wants to merge 3 commits into
llvm:mainfrom
tkarna:xegpu-elemwise-schedule
Open

[xegpu] Elementwise schedule and refactoring#200
tkarna wants to merge 3 commits into
llvm:mainfrom
tkarna:xegpu-elemwise-schedule

Conversation

@tkarna

@tkarna tkarna commented Jun 25, 2026

Copy link
Copy Markdown
Contributor
  • Adds a simple linalg-to-binary schedule for elementwise kernels.
  • Refactors common parts with the mlp schedule to sub-schedules (python functions in lowering_common.py).
    • These sub-schedules could be outlined into separate schedules if needed.
  • The mlp and elementwise schedules now take the payload function name for more robust matching.
    • GPU specific transformations should not be applied to the entire module nor all functions.

@tkarna tkarna requested review from adam-smnk and rengolin June 25, 2026 19:50
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