From fc27c37e302eefe26847d2c2ba2b392e973c420b Mon Sep 17 00:00:00 2001 From: Baihui Jin Date: Thu, 12 Mar 2026 21:39:40 -0700 Subject: [PATCH 1/3] Dual Triton: coexist oait and nvtriton via PYTHONPATH isolation Instead of nvtriton replacing upstream triton, install both side-by-side: - oait triton 3.6.0 in site-packages (default, used by helion) - nvtriton in /opt/nvtriton (activate with PYTHONPATH=/opt/nvtriton ENABLE_TILE=1) This lets kernelbot benchmark kernels against both backends in the same image. Tested: both run GPU kernels end-to-end, helion works with either. --- docker/helion.Dockerfile | 14 ++++++++++---- 1 file changed, 10 insertions(+), 4 deletions(-) diff --git a/docker/helion.Dockerfile b/docker/helion.Dockerfile index 5a9f3664f..9bb0d0bea 100644 --- a/docker/helion.Dockerfile +++ b/docker/helion.Dockerfile @@ -50,14 +50,20 @@ RUN sudo uv pip install --system \ torch \ --index-url https://download.pytorch.org/whl/cu130 -# nvtriton (Triton with TileIR backend — replaces upstream triton) +# Upstream OpenAI Triton (oait) — PyTorch cu130 bundles triton, but pin the +# version explicitly so it survives future PyTorch changes. +RUN sudo uv pip install --system triton==3.6.0 + +# nvtriton (Triton with TileIR backend) — installed to /opt/nvtriton so it +# coexists with the upstream triton in site-packages. +# Activate with: PYTHONPATH=/opt/nvtriton ENABLE_TILE=1 RUN curl -L -o /tmp/nvtriton-3.6.0-cp313-cp313-linux_x86_64.whl \ 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 \ - && sudo uv pip install --system /tmp/nvtriton-3.6.0-cp313-cp313-linux_x86_64.whl \ + && sudo python3 -m pip install --no-cache-dir --no-deps \ + --target /opt/nvtriton \ + /tmp/nvtriton-3.6.0-cp313-cp313-linux_x86_64.whl \ && rm /tmp/nvtriton-3.6.0-cp313-cp313-linux_x86_64.whl -ENV ENABLE_TILE=0 - # Helion RUN sudo uv pip install --system helion From 8fc86d6227cf07ddc62234775ef9d3d69cf3f442 Mon Sep 17 00:00:00 2001 From: Baihui Jin Date: Thu, 12 Mar 2026 23:34:08 -0700 Subject: [PATCH 2/3] =?UTF-8?q?Remove=20redundant=20triton=20install=20?= =?UTF-8?q?=E2=80=94=20PyTorch=20cu130=20already=20bundles=20it?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- docker/helion.Dockerfile | 6 +----- 1 file changed, 1 insertion(+), 5 deletions(-) diff --git a/docker/helion.Dockerfile b/docker/helion.Dockerfile index 9bb0d0bea..564eb1705 100644 --- a/docker/helion.Dockerfile +++ b/docker/helion.Dockerfile @@ -50,12 +50,8 @@ RUN sudo uv pip install --system \ torch \ --index-url https://download.pytorch.org/whl/cu130 -# Upstream OpenAI Triton (oait) — PyTorch cu130 bundles triton, but pin the -# version explicitly so it survives future PyTorch changes. -RUN sudo uv pip install --system triton==3.6.0 - # nvtriton (Triton with TileIR backend) — installed to /opt/nvtriton so it -# coexists with the upstream triton in site-packages. +# coexists with the upstream triton that PyTorch bundles in site-packages. # Activate with: PYTHONPATH=/opt/nvtriton ENABLE_TILE=1 RUN curl -L -o /tmp/nvtriton-3.6.0-cp313-cp313-linux_x86_64.whl \ 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 \ From fcf6f907b1a55836f79c7fcf7b466a16fa36f331 Mon Sep 17 00:00:00 2001 From: Baihui Jin Date: Fri, 13 Mar 2026 04:08:18 -0700 Subject: [PATCH 3/3] Add HELION_BACKEND=tileir to nvtriton activation comment --- docker/helion.Dockerfile | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/docker/helion.Dockerfile b/docker/helion.Dockerfile index 564eb1705..178621023 100644 --- a/docker/helion.Dockerfile +++ b/docker/helion.Dockerfile @@ -52,7 +52,7 @@ RUN sudo uv pip install --system \ # nvtriton (Triton with TileIR backend) — installed to /opt/nvtriton so it # coexists with the upstream triton that PyTorch bundles in site-packages. -# Activate with: PYTHONPATH=/opt/nvtriton ENABLE_TILE=1 +# Activate with: PYTHONPATH=/opt/nvtriton ENABLE_TILE=1 HELION_BACKEND=tileir RUN curl -L -o /tmp/nvtriton-3.6.0-cp313-cp313-linux_x86_64.whl \ 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 \ && sudo python3 -m pip install --no-cache-dir --no-deps \