Skip to content

Commit 1f8f93e

Browse files
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 failed
1 parent fcda910 commit 1f8f93e

1 file changed

Lines changed: 8 additions & 0 deletions

File tree

docker/helion.Dockerfile

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -50,6 +50,14 @@ RUN sudo uv pip install --system \
5050
torch \
5151
--index-url https://download.pytorch.org/whl/cu130
5252

53+
# nvtriton (Triton with TileIR backend — replaces upstream triton)
54+
RUN curl -L -o /tmp/nvtriton-3.6.0-cp313-cp313-linux_x86_64.whl \
55+
https://github.com/triton-lang/Triton-to-tile-IR/releases/download/v3.6.0-rc1/nvtriton-3.6.0-cp313-cp313-linux_x86_64.whl \
56+
&& sudo uv pip install --system /tmp/nvtriton-3.6.0-cp313-cp313-linux_x86_64.whl \
57+
&& rm /tmp/nvtriton-3.6.0-cp313-cp313-linux_x86_64.whl
58+
59+
ENV ENABLE_TILE=1
60+
5361
# Helion
5462
RUN sudo uv pip install --system helion
5563

0 commit comments

Comments
 (0)