Skip to content

Commit 360b401

Browse files
author
Aleksandr Malyshev
committed
corrected docker to AI days version
Signed-off-by: Aleksandr Malyshev <maleksan@amd.com>
1 parent 353c736 commit 360b401

1 file changed

Lines changed: 39 additions & 39 deletions

File tree

‎docker/Dockerfile.rocm_base‎

Lines changed: 39 additions & 39 deletions
Original file line numberDiff line numberDiff line change
@@ -2,12 +2,12 @@ ARG BASE_IMAGE=compute-artifactory.amd.com:5000/rocm-plus-docker/framework/compu
22
# ARG HIPBLASLT_BRANCH="aa0bda7b"
33
# ARG HIPBLAS_COMMON_BRANCH="9b80ba8e"
44
# ARG LEGACY_HIPBLASLT_OPTION=
5-
ARG TRITON_BRANCH="916969a"
6-
ARG TRITON_REPO="https://github.com/ROCm/triton.git"
7-
ARG PYTORCH_BRANCH="3a7dec2"
8-
ARG PYTORCH_VISION_BRANCH="0.22.0a0"
9-
ARG PYTORCH_REPO="https://github.com/ROCm/pytorch.git"
10-
ARG PYTORCH_VISION_REPO="https://github.com/pytorch/vision.git"
5+
# ARG TRITON_BRANCH="96316ce5"
6+
# ARG TRITON_REPO="https://github.com/ROCm/triton.git"
7+
# ARG PYTORCH_BRANCH="74091a9"
8+
# ARG PYTORCH_VISION_BRANCH="0.22.0a0"
9+
# ARG PYTORCH_REPO="https://github.com/ROCm/pytorch.git"
10+
# ARG PYTORCH_VISION_REPO="https://github.com/pytorch/vision.git"
1111
ARG FA_BRANCH="8ede036"
1212
ARG FA_REPO="https://github.com/Dao-AILab/flash-attention.git"
1313
ARG AITER_BRANCH="498ff21"
@@ -69,15 +69,15 @@ RUN pip install -U packaging 'cmake<4' ninja wheel 'setuptools<80' pybind11 Cyth
6969
# && make package
7070
# RUN mkdir -p /app/install && cp /app/hipBLASLt/build/release/*.deb /app/hipBLAS-common/build/*.deb /app/install
7171

72-
FROM base AS build_triton
73-
ARG TRITON_BRANCH
74-
ARG TRITON_REPO
75-
RUN git clone ${TRITON_REPO}
76-
RUN cd triton \
77-
&& git checkout ${TRITON_BRANCH} \
78-
# && cd python \
79-
&& python3 setup.py bdist_wheel --dist-dir=python/dist
80-
RUN mkdir -p /app/install && cp /app/triton/python/dist/*.whl /app/install
72+
# FROM base AS build_triton
73+
# ARG TRITON_BRANCH
74+
# ARG TRITON_REPO
75+
# RUN git clone ${TRITON_REPO}
76+
# RUN cd triton \
77+
# && git checkout ${TRITON_BRANCH} \
78+
# # && cd python \
79+
# && python3 setup.py bdist_wheel --dist-dir=python/dist
80+
# RUN mkdir -p /app/install && cp /app/triton/python/dist/*.whl /app/install
8181

8282
FROM base AS build_amdsmi
8383
RUN cd /opt/rocm/share/amd_smi \
@@ -91,16 +91,16 @@ ARG PYTORCH_REPO
9191
ARG PYTORCH_VISION_REPO
9292
ARG FA_BRANCH
9393
ARG FA_REPO
94-
RUN git clone ${PYTORCH_REPO} pytorch
95-
RUN cd pytorch && git checkout ${PYTORCH_BRANCH} && \
96-
pip install -r requirements.txt && git submodule update --init --recursive \
97-
&& python3 tools/amd_build/build_amd.py \
98-
&& CMAKE_PREFIX_PATH=$(python3 -c 'import sys; print(sys.prefix)') python3 setup.py bdist_wheel --dist-dir=dist \
99-
&& pip install dist/*.whl
100-
RUN git clone ${PYTORCH_VISION_REPO} vision
101-
RUN cd vision && git checkout ${PYTORCH_VISION_BRANCH} \
102-
&& python3 setup.py bdist_wheel --dist-dir=dist \
103-
&& pip install dist/*.whl
94+
# RUN git clone ${PYTORCH_REPO} pytorch
95+
# RUN cd pytorch && git checkout ${PYTORCH_BRANCH} && \
96+
# pip install -r requirements.txt && git submodule update --init --recursive \
97+
# && python3 tools/amd_build/build_amd.py \
98+
# && CMAKE_PREFIX_PATH=$(python3 -c 'import sys; print(sys.prefix)') python3 setup.py bdist_wheel --dist-dir=dist \
99+
# && pip install dist/*.whl
100+
# RUN git clone ${PYTORCH_VISION_REPO} vision
101+
# RUN cd vision && git checkout ${PYTORCH_VISION_BRANCH} \
102+
# && python3 setup.py bdist_wheel --dist-dir=dist \
103+
# && pip install dist/*.whl
104104
RUN git clone ${FA_REPO}
105105
RUN cd flash-attention \
106106
&& git checkout ${FA_BRANCH} \
@@ -128,12 +128,12 @@ FROM base AS debs
128128
RUN mkdir /app/debs
129129
# RUN --mount=type=bind,from=build_hipblaslt,src=/app/install/,target=/install \
130130
# cp /install/*.deb /app/debs
131-
RUN --mount=type=bind,from=build_triton,src=/app/install/,target=/install \
132-
cp /install/*.whl /app/debs
131+
# RUN --mount=type=bind,from=build_triton,src=/app/install/,target=/install \
132+
# cp /install/*.whl /app/debs
133133
RUN --mount=type=bind,from=build_amdsmi,src=/app/install/,target=/install \
134134
cp /install/*.whl /app/debs
135-
RUN --mount=type=bind,from=build_pytorch,src=/app/install/,target=/install \
136-
cp /install/*.whl /app/debs
135+
# RUN --mount=type=bind,from=build_pytorch,src=/app/install/,target=/install \
136+
# cp /install/*.whl /app/debs
137137
RUN --mount=type=bind,from=build_aiter,src=/app/install/,target=/install \
138138
cp /install/*.whl /app/debs
139139

@@ -143,12 +143,12 @@ FROM base AS final
143143
# && perl -p -i -e 's/, hipblas-common-dev \([^)]*?\), /, /g' /var/lib/dpkg/status \
144144
# && perl -p -i -e 's/, hipblaslt-dev \([^)]*?\), /, /g' /var/lib/dpkg/status \
145145
# && perl -p -i -e 's/, hipblaslt \([^)]*?\), /, /g' /var/lib/dpkg/status
146-
RUN --mount=type=bind,from=build_triton,src=/app/install/,target=/install \
147-
pip install /install/*.whl
146+
# RUN --mount=type=bind,from=build_triton,src=/app/install/,target=/install \
147+
# pip install /install/*.whl
148148
RUN --mount=type=bind,from=build_amdsmi,src=/app/install/,target=/install \
149149
pip install /install/*.whl
150-
RUN --mount=type=bind,from=build_pytorch,src=/app/install/,target=/install \
151-
pip install /install/*.whl
150+
# RUN --mount=type=bind,from=build_pytorch,src=/app/install/,target=/install \
151+
# pip install /install/*.whl
152152
RUN --mount=type=bind,from=build_aiter,src=/app/install/,target=/install \
153153
pip install /install/*.whl
154154

@@ -170,12 +170,12 @@ RUN echo "BASE_IMAGE: ${BASE_IMAGE}" > /app/versions.txt \
170170
# && echo "HIPBLAS_COMMON_BRANCH: ${HIPBLAS_COMMON_BRANCH}" >> /app/versions.txt \
171171
# && echo "HIPBLASLT_BRANCH: ${HIPBLASLT_BRANCH}" >> /app/versions.txt \
172172
# && echo "LEGACY_HIPBLASLT_OPTION: ${LEGACY_HIPBLASLT_OPTION}" >> /app/versions.txt \
173-
&& echo "TRITON_BRANCH: ${TRITON_BRANCH}" >> /app/versions.txt \
174-
&& echo "TRITON_REPO: ${TRITON_REPO}" >> /app/versions.txt \
175-
&& echo "PYTORCH_BRANCH: ${PYTORCH_BRANCH}" >> /app/versions.txt \
176-
&& echo "PYTORCH_VISION_BRANCH: ${PYTORCH_VISION_BRANCH}" >> /app/versions.txt \
177-
&& echo "PYTORCH_REPO: ${PYTORCH_REPO}" >> /app/versions.txt \
178-
&& echo "PYTORCH_VISION_REPO: ${PYTORCH_VISION_REPO}" >> /app/versions.txt \
173+
# && echo "TRITON_BRANCH: ${TRITON_BRANCH}" >> /app/versions.txt \
174+
# && echo "TRITON_REPO: ${TRITON_REPO}" >> /app/versions.txt \
175+
# && echo "PYTORCH_BRANCH: ${PYTORCH_BRANCH}" >> /app/versions.txt \
176+
# && echo "PYTORCH_VISION_BRANCH: ${PYTORCH_VISION_BRANCH}" >> /app/versions.txt \
177+
# && echo "PYTORCH_REPO: ${PYTORCH_REPO}" >> /app/versions.txt \
178+
# && echo "PYTORCH_VISION_REPO: ${PYTORCH_VISION_REPO}" >> /app/versions.txt \
179179
&& echo "FA_BRANCH: ${FA_BRANCH}" >> /app/versions.txt \
180180
&& echo "AITER_BRANCH: ${AITER_BRANCH}" >> /app/versions.txt \
181181
&& echo "AITER_REPO: ${AITER_REPO}" >> /app/versions.txt

0 commit comments

Comments
 (0)