Skip to content

Commit b5a4a37

Browse files
authored
Stable aiter build (#450)
* Using aiter branch that can be built into a whl with PREBUILD_KERNELS=1 * Using fail fast on aiter build to see compilation errors in the log since it fails silently * Check for build success without installing whl
1 parent ba6f019 commit b5a4a37

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

Dockerfile.rocm_base

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -12,7 +12,7 @@ ARG PYTORCH_REPO="https://github.com/pytorch/pytorch.git"
1212
ARG PYTORCH_VISION_REPO="https://github.com/pytorch/vision.git"
1313
ARG FA_BRANCH="1a7f4dfa"
1414
ARG FA_REPO="https://github.com/Dao-AILab/flash-attention.git"
15-
ARG AITER_BRANCH="9633cab3"
15+
ARG AITER_BRANCH="485b4b28"
1616
ARG AITER_REPO="https://github.com/ROCm/aiter.git"
1717

1818
FROM ${BASE_IMAGE} AS base
@@ -128,7 +128,7 @@ RUN cd aiter \
128128
&& git checkout ${AITER_BRANCH} \
129129
&& git submodule update --init --recursive \
130130
&& pip install -r requirements.txt \
131-
&& GPU_ARCHS=gfx942 python3 setup.py bdist_wheel --dist-dir=dist
131+
&& PREBUILD_KERNELS=1 GPU_ARCHS=gfx942 python3 setup.py bdist_wheel --dist-dir=dist && ls /app/aiter/dist/*.whl
132132
RUN mkdir -p /app/install && cp /app/aiter/dist/*.whl /app/install
133133

134134
FROM base AS final

0 commit comments

Comments
 (0)