From 76a3520809d0236c2c79844f5a68bceec4ebf6df Mon Sep 17 00:00:00 2001 From: Gregory Shtrasberg Date: Mon, 17 Feb 2025 19:28:18 +0000 Subject: [PATCH 1/2] Advance torch commit to be past https://github.com/pytorch/pytorch/pull/144942 to fix tunable ops --- Dockerfile.rocm_base | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/Dockerfile.rocm_base b/Dockerfile.rocm_base index 9a3adea97101..95a5b6b3e881 100644 --- a/Dockerfile.rocm_base +++ b/Dockerfile.rocm_base @@ -6,8 +6,8 @@ ARG RCCL_BRANCH="648a58d" ARG RCCL_REPO="https://github.com/ROCm/rccl" ARG TRITON_BRANCH="e5be006" ARG TRITON_REPO="https://github.com/triton-lang/triton.git" -ARG PYTORCH_BRANCH="3a585126" -ARG PYTORCH_VISION_BRANCH="v0.19.1" +ARG PYTORCH_BRANCH="6c0e7463" +ARG PYTORCH_VISION_BRANCH="v0.21.0" ARG PYTORCH_REPO="https://github.com/pytorch/pytorch.git" ARG PYTORCH_VISION_REPO="https://github.com/pytorch/vision.git" ARG FA_BRANCH="1a7f4dfa" From 503abb5b649a0bde76c6822feeccdf2227dbdff7 Mon Sep 17 00:00:00 2001 From: Gregory Shtrasberg Date: Mon, 17 Feb 2025 23:58:28 +0000 Subject: [PATCH 2/2] Make sure to use the submodule commit compatible with the main aiter commit --- Dockerfile.rocm_base | 1 + 1 file changed, 1 insertion(+) diff --git a/Dockerfile.rocm_base b/Dockerfile.rocm_base index 95a5b6b3e881..4b511f5aeccf 100644 --- a/Dockerfile.rocm_base +++ b/Dockerfile.rocm_base @@ -126,6 +126,7 @@ RUN --mount=type=bind,from=build_pytorch,src=/app/install/,target=/install \ RUN git clone --recursive ${AITER_REPO} RUN cd aiter \ && git checkout ${AITER_BRANCH} \ + && git submodule update --init --recursive \ && pip install -r requirements.txt \ && PREBUILD_KERNELS=1 GPU_ARCHS=gfx942 python3 setup.py bdist_wheel --dist-dir=dist RUN mkdir -p /app/install && cp /app/aiter/dist/*.whl /app/install