Skip to content

Linalg/XeGPU Layer norm example#186

Merged
rengolin merged 4 commits into
llvm:mainfrom
nbpatel:layer_norm_xegpu
Jun 16, 2026
Merged

Linalg/XeGPU Layer norm example#186
rengolin merged 4 commits into
llvm:mainfrom
nbpatel:layer_norm_xegpu

Conversation

@nbpatel

@nbpatel nbpatel commented Jun 8, 2026

Copy link
Copy Markdown
Contributor

Adds an end-to-end LayerNorm example that lowers a linalg payload through a transform-dialect schedule (tile rows into forall, tile mean/variance reductions, vectorize, bufferize, lower to XeGPU) for execution on the Intel GPU. Includes the MLIR payload generator, the reusable layer_norm_schedule, and a FileCheck'd example with a numpy reference.

Assisted by Claude.

Comment thread lighthouse/ingress/mlir_gen/gpu_layer_norm_payload.py Outdated
Comment thread lighthouse/ingress/mlir_gen/gpu_layer_norm_payload.py Outdated
@rengolin

Copy link
Copy Markdown
Member

The conflict is just about the previous PR merged added another generator. one-liner.

@nbpatel

nbpatel commented Jun 16, 2026

Copy link
Copy Markdown
Contributor Author

The conflict is just about the previous PR merged added another generator. one-liner.

Thanks. there is some lint failure? is it some formatting check? also I will need help from you or @tkarna for merging it

@rengolin

Copy link
Copy Markdown
Member

Thanks. there is some lint failure? is it some formatting check? also I will need help from you or @tkarna for merging it

Yes, you can reproduce that by running the tests on your machine with:

$ ./precommit.sh

@nbpatel

nbpatel commented Jun 16, 2026

Copy link
Copy Markdown
Contributor Author

Thanks. there is some lint failure? is it some formatting check? also I will need help from you or @tkarna for merging it

Yes, you can reproduce that by running the tests on your machine with:

$ ./precommit.sh

Fixed it. Can you please help me merge it?

@rengolin rengolin merged commit 74bc5b8 into llvm:main Jun 16, 2026
3 checks passed
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.

3 participants