[ROCm]: Bump aiter to 0.1.16.post2 (#46692)
Signed-off-by: Rohan138 <rohanpotdar138@gmail.com>
Rohan Potdar committed
Jun 25, 2026 at 13:53 UTC
e53a17232c3114ab276d7004edb195304ecdad57
1 file changed
+2
-2
docker/Dockerfile.rocm_base
+2
-2
@@ -9,7 +9,7 @@ ARG PYTORCH_AUDIO_BRANCH="v2.9.0"
9
ARG PYTORCH_AUDIO_REPO="https://github.com/pytorch/audio.git"
10
ARG FA_BRANCH="0e60e394"
11
ARG FA_REPO="https://github.com/Dao-AILab/flash-attention.git"
12
-ARG AITER_BRANCH="v0.1.13.post1"
12
+ARG AITER_BRANCH="v0.1.16.post2"
13
ARG AITER_REPO="https://github.com/ROCm/aiter.git"
14
ARG MORI_BRANCH="v1.1.0"
15
ARG MORI_REPO="https://github.com/ROCm/mori.git"
@@ -244,7 +244,7 @@ RUN pip install pyyaml && cd aiter \
244
export HIP_CLANG_PATH=/opt/sccache-wrappers \
245
&& sccache --show-stats; \
246
fi \
247
- && PREBUILD_KERNELS=1 GPU_ARCHS=${AITER_ROCM_ARCH} python3 setup.py bdist_wheel --dist-dir=dist \
247
+ && PREBUILD_KERNELS=1 AITER_USE_SYSTEM_TRITON=1 GPU_ARCHS=${AITER_ROCM_ARCH} python3 setup.py bdist_wheel --dist-dir=dist \
248
&& if [ "$USE_SCCACHE" = "1" ]; then sccache --show-stats; fi \
249
&& ls /app/aiter/dist/*.whl
250
RUN mkdir -p /app/install && cp /app/aiter/dist/*.whl /app/install