Skip to content

Commit 050f8ab

Browse files
committed
Add Helion language support
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
1 parent c7386b0 commit 050f8ab

File tree

1 file changed

+15
-0
lines changed

1 file changed

+15
-0
lines changed

bin/yaml/helion.yaml

Lines changed: 15 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,15 @@
1+
compilers:
2+
helion:
3+
depends:
4+
- compilers/python 3.12.1
5+
type: pip
6+
dir: helion/v{{name}}
7+
python: "%DEP0%/bin/python3.12"
8+
package:
9+
- helion=={{name}}
10+
- triton==3.5.0
11+
- torch==2.9.0
12+
- numpy
13+
check_exe: [ "bin/python3.12", "-c", "import helion; print(helion.__name__)" ]
14+
targets:
15+
- 0.2.0

0 commit comments

Comments
 (0)