[ROCm]: Bump triton 3.7 commit (#52819)
Signed-off-by: Rohan138 <rohanpotdar138@gmail.com>
Rohan Potdar committed
Aug 19, 2026 at 17:39 UTC
823ec22b78b639f6794df3e5ee16ce4429f2c269
1 file changed
+2
-2
docker/Dockerfile.rocm_base
+2
-2
@@ -1,5 +1,5 @@
1
ARG BASE_IMAGE=rocm/dev-ubuntu-22.04:7.2.3-complete
2
-ARG TRITON_BRANCH="532137f" # release/internal/3.7.x as of 08/07
2
+ARG TRITON_BRANCH="f0b55c0" # release/internal/3.7.x as of 08/18
3
ARG TRITON_REPO="https://github.com/ROCm/triton.git"
4
ARG PYTORCH_BRANCH="6bbd260" # release/2.12 as of 08/02
5
ARG PYTORCH_REPO="https://github.com/ROCm/pytorch.git"
@@ -58,7 +58,7 @@ RUN apt-get update -y \
58
&& curl -sS https://bootstrap.pypa.io/get-pip.py | python${PYTHON_VERSION} \
59
&& python3 --version && python3 -m pip --version
60
61
-RUN pip install -U packaging 'cmake<4' ninja wheel 'setuptools<80' pybind11 Cython
61
+RUN pip install -U packaging 'cmake<4' ninja wheel 'setuptools<80' 'pybind11<3.1.0' Cython
62
RUN apt-get update && apt-get install -y libjpeg-dev libsox-dev libsox-fmt-all sox && rm -rf /var/lib/apt/lists/*
63
64
# Install sccache if USE_SCCACHE is enabled (for release builds)