diff --git a/docker/helion.Dockerfile b/docker/helion.Dockerfile index 5a9f3664f..178621023 100644 --- a/docker/helion.Dockerfile +++ b/docker/helion.Dockerfile @@ -50,14 +50,16 @@ RUN sudo uv pip install --system \ torch \ --index-url https://download.pytorch.org/whl/cu130 -# nvtriton (Triton with TileIR backend — replaces upstream triton) +# 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 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 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