Commit 1f8f93e
authored
Add nvtriton (Triton TileIR backend) to helion Dockerfile (#461)
Install nvtriton wheel from GitHub Release v3.6.0-rc1 after PyTorch,
replacing the upstream triton==3.6.0 that ships with PyTorch cu130.
Set ENABLE_TILE=1 to activate the TileIR compiler backend by default.
Verified on B200 (sm100):
- TileIRDriver activation: OK
- tileiras embedded discovery: OK
- Triton dot kernel (fp16 64x64): max diff 0.0
- Helion matmul (fp16 256x256): max diff 0.0
- IR pipeline: convert-triton-to-cuda-tile → cuda_tile.module confirmed
- Helion test suite (HELION_BACKEND=tileir): 834 passed, 463 skipped, 0 failed1 parent fcda910 commit 1f8f93e
1 file changed
Lines changed: 8 additions & 0 deletions
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
50 | 50 | | |
51 | 51 | | |
52 | 52 | | |
| 53 | + | |
| 54 | + | |
| 55 | + | |
| 56 | + | |
| 57 | + | |
| 58 | + | |
| 59 | + | |
| 60 | + | |
53 | 61 | | |
54 | 62 | | |
55 | 63 | | |
| |||
0 commit comments