Skip to content

Add Helion language support #1875

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom
Merged
merged 1 commit into from
Nov 4, 2025
Merged

Add Helion language support #1875

merged 1 commit into from
Nov 4, 2025

Conversation

@oulgen
Copy link
Contributor

@oulgen oulgen commented Oct 19, 2025 •
edited
Loading

Helion is a Python-embedded domain-specific language (DSL) for authoring machine learning kernels, designed to compile down to Triton.

https://github.com/pytorch/helion

I mostly followed the Triton example, please let me know if there's more I need to do or anything I missed.

compiler-explorer PR: compiler-explorer/compiler-explorer#8206

@oulgen oulgen mentioned this pull request Oct 19, 2025
Merged
mattgodbolt
mattgodbolt requested changes Oct 22, 2025
View reviewed changes
Copy link
Member

@mattgodbolt mattgodbolt left a comment

Choose a reason for hiding this comment

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

You should be able to test this locally with bin/ce_install install helion or similar. There's some directories we try to install to (/opt/compiler-explorer) but if you can't make them, then there's CLI flags to set the destination.

bin/yaml/helion.yaml Outdated
@@ -0,0 +1,15 @@
compilers:
triton:
Copy link
Member

@mattgodbolt mattgodbolt Oct 22, 2025

Choose a reason for hiding this comment

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

this should be helion

@oulgen
Add Helion language support
050f8ab 
Helion is a Python-embedded domain-specific language (DSL) for authoring
machine learning kernels, designed to compile down to Triton.

https://github.com/pytorch/helion
@oulgen oulgen requested a review from mattgodbolt October 25, 2025 18:19
@oulgen
Copy link
Contributor Author

oulgen commented Oct 25, 2025

Thanks @mattgodbolt

I've run

(py312) ➜  compiler-explorer-infra git:(helion) ./bin/ce_install install helion     
warning: `VIRTUAL_ENV=/home/oulgen/py312` does not match the project environment path `.venv` and will be ignored; use `--active` to target the active environment instead
2025-10-25 11:15:35,656 lib.ce_install  INFO     Creating thread pool with 8 workers
Installing compilers/helion 0.2.0
2025-10-25 11:15:36,813 compilers/helion 0.2.0 INFO     Installing required dependee S3TarballInstallable(compilers/python 3.12.1, python-3.12.1)
2025-10-25 11:15:37,148 lib.installation_context INFO     Fetching https://s3.amazonaws.com/compiler-explorer/opt/python-3.12.1.tar.xz (21902224 bytes)
2025-10-25 11:15:38,707 lib.installation_context INFO     100% of https://s3.amazonaws.com/compiler-explorer/opt/python-3.12.1.tar.xz
2025-10-25 11:15:38,708 lib.installation_context INFO     Piping to tar Jxf -
2025-10-25 11:15:39,086 lib.installation_context INFO     Moving from staging (/opt/compiler-explorer/staging/df3e98b4-4438-435f-8da3-ceddf5e25d37/python-3.12.1) to final destination (/opt/compiler-explorer/python-3.12.1)

[notice] A new release of pip is available: 23.2.1 -> 25.3
[notice] To update, run: /opt/compiler-explorer/staging/f3e804f0-72d8-42a8-bf71-4412d22378e3/helion/v0.2.0/bin/python3.12 -m pip install --upgrade pip
2025-10-25 11:17:31,262 lib.installation_context INFO     Moving from staging (/opt/compiler-explorer/staging/f3e804f0-72d8-42a8-bf71-4412d22378e3/helion/v0.2.0) to final destination (/opt/compiler-explorer/helion/v0.2.0)
2025-10-25 11:17:34,959 lib.ce_install  INFO     compilers/helion 0.2.0 installed OK
1 packages installed OK, 0 skipped, and 0 failed installation

and was successfully able to install.

I tried to run the GH action: https://github.com/compiler-explorer/infra/actions/workflows/bespoke-build.yaml but I dont seem to have permissions. Is there anything else you'd like me to do on this PR?

@oulgen
Copy link
Contributor Author

oulgen commented Oct 25, 2025

I've also updated the corresponding compiler-explorer PR: compiler-explorer/compiler-explorer#8206

@mattgodbolt
Copy link
Member

mattgodbolt commented Nov 4, 2025

...

and was successfully able to install.

fab!

I tried to run the GH action: https://github.com/compiler-explorer/infra/actions/workflows/bespoke-build.yaml but I dont seem to have permissions. Is there anything else you'd like me to do on this PR?

Right - there's no compiler to build here, so there's no need to run that :)

I'll try and merge now

mattgodbolt
mattgodbolt approved these changes Nov 4, 2025
View reviewed changes
Copy link
Member

@mattgodbolt mattgodbolt left a comment

Choose a reason for hiding this comment

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

LGTM! thanks

@mattgodbolt mattgodbolt merged commit ef0de9b into compiler-explorer:main Nov 4, 2025
1 check passed
@mattgodbolt
Copy link
Member

mattgodbolt commented Nov 4, 2025

installing now...

@mattgodbolt
Copy link
Member

mattgodbolt commented Nov 4, 2025

Yay:

Nov/04 17:04 admin-node~/infra (main|✔) $ sudo ce_install install helion
2025-11-04 17:04:44,562 lib.ce_install  INFO     Creating thread pool with 2 workers
Installing compilers/helion 0.2.0

[notice] A new release of pip is available: 23.2.1 -> 25.3
[notice] To update, run: /tmp/ce-cefs-temp/staging/9fd0138d-c2e0-4680-924d-ce264dd67fd0/helion/v0.2.0/bin/python3.12 -m pip install --upgrade pip
2025-11-04 17:06:45,150 lib.installation_context INFO     Installing via CEFS: helion/v0.2.0 -> helion/v0.2.0
2025-11-04 17:06:45,990 lib.installation_context INFO     Creating squashfs image from /tmp/ce-cefs-temp/staging/9fd0138d-c2e0-4680-924d-ce264dd67fd0/helion/v0.2.0
2025-11-04 17:08:39,682 lib.installation_context INFO     Copying squashfs to CEFS storage: /efs/cefs-images/0d/0d48e9e685ebbd88246c24e5_helion_v0.2.0.sqfs
2025-11-04 17:08:39,682 lib.cefs.deployment INFO     Copying /tmp/ce-cefs-temp/temp_9c8d3a26-a572-4c3f-bcee-8ab31a02179e.img to /efs/cefs-images/0d/0d48e9e685ebbd88246c24e5_helion_v0.2.0.sqfs
2025-11-04 17:09:05,433 lib.cefs.deployment INFO     Created symlink /opt/compiler-explorer/helion/v0.2.0 -> /cefs/0d/0d48e9e685ebbd88246c24e5_helion_v0.2.0
2025-11-04 17:09:31,661 lib.ce_install  INFO     compilers/helion 0.2.0 installed OK
1 packages installed OK, 0 skipped, and 0 failed installation

will review the CE PR soon :)

Hopefully can get this out in today's release

@oulgen
Copy link
Contributor Author

oulgen commented Nov 4, 2025

awesome, thank you @mattgodbolt

mattgodbolt pushed a commit to compiler-explorer/compiler-explorer that referenced this pull request Nov 4, 2025
@oulgen
Add Helion language support (#8206)
f7bc52a 
Helion is a Python-embedded domain-specific language (DSL) for authoring
machine learning kernels, designed to compile down to Triton.

https://github.com/pytorch/helion

I mostly followed the Triton example, please let me know if there's more
I need to do or anything I missed.

infra PR: compiler-explorer/infra#1875
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Reviewers

@mattgodbolt mattgodbolt mattgodbolt approved these changes

Assignees

No one assigned

Labels

None yet

Projects

None yet

Milestone

No milestone

Development

Successfully merging this pull request may close these issues.

2 participants

@oulgen @mattgodbolt