Skip to content

Commit 02a90d0

Browse files
author
Doug Lehr
committed
Update to new RC4 build with updated HIPBLASLT
1 parent 4f43dae commit 02a90d0

File tree

1 file changed

+36
-36
lines changed

1 file changed

+36
-36
lines changed

‎docker/Dockerfile.rocm_base‎

Lines changed: 36 additions & 36 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
ARG BASE_IMAGE=registry-sc-harbor.amd.com/framework/compute-rocm-rel-7.0:38_ubuntu22.04_py3.10_pytorch_release-2.8_b2fb6885
1+
ARG BASE_IMAGE=registry-sc-harbor.amd.com/framework/compute-rocm-dkms-component-baas-rel:25_ubuntu22.04_py3.10_pytorch_rocm7.1_internal_testing_125803b7
22
# ARG HIPBLASLT_BRANCH="aa0bda7b"
33
# ARG HIPBLAS_COMMON_BRANCH="9b80ba8e"
44
# ARG LEGACY_HIPBLASLT_OPTION=
@@ -62,39 +62,39 @@ RUN pip install -U packaging 'cmake<4' ninja wheel 'setuptools<80' pybind11 Cyth
6262
# && make package
6363
# RUN mkdir -p /app/install && cp /app/hipBLASLt/build/release/*.deb /app/hipBLAS-common/build/*.deb /app/install
6464

65-
FROM base AS build_triton
66-
ARG TRITON_BRANCH
67-
ARG TRITON_REPO
68-
RUN git clone ${TRITON_REPO}
69-
RUN cd triton \
70-
&& git checkout ${TRITON_BRANCH} \
71-
&& python3 setup.py bdist_wheel --dist-dir=python/dist
72-
RUN pip install build && cd triton/python/triton_kernels \
73-
&& python3 -m build --wheel && cp dist/*.whl /app/triton/python/dist
74-
RUN mkdir -p /app/install && cp /app/triton/python/dist/*.whl /app/install
65+
#FROM base AS build_triton
66+
#ARG TRITON_BRANCH
67+
#ARG TRITON_REPO
68+
#RUN git clone ${TRITON_REPO}
69+
#RUN cd triton \
70+
# && git checkout ${TRITON_BRANCH} \
71+
# && python3 setup.py bdist_wheel --dist-dir=python/dist
72+
#RUN pip install build && cd triton/python/triton_kernels \
73+
# && python3 -m build --wheel && cp dist/*.whl /app/triton/python/dist
74+
#RUN mkdir -p /app/install && cp /app/triton/python/dist/*.whl /app/install
7575

7676
FROM base AS build_amdsmi
7777
RUN cd /opt/rocm/share/amd_smi \
7878
&& pip wheel . --wheel-dir=dist
7979
RUN mkdir -p /app/install && cp /opt/rocm/share/amd_smi/dist/*.whl /app/install
8080

81-
FROM base AS build_pytorch
82-
ARG PYTORCH_BRANCH
83-
ARG PYTORCH_VISION_BRANCH
84-
ARG PYTORCH_REPO
85-
ARG PYTORCH_VISION_REPO
86-
ARG FA_BRANCH
87-
ARG FA_REPO
88-
RUN git clone ${PYTORCH_REPO} pytorch
89-
RUN cd pytorch && git checkout ${PYTORCH_BRANCH} && \
90-
pip install -r requirements.txt && git submodule update --init --recursive \
91-
&& python3 tools/amd_build/build_amd.py \
92-
&& CMAKE_PREFIX_PATH=$(python3 -c 'import sys; print(sys.prefix)') python3 setup.py bdist_wheel --dist-dir=dist \
93-
&& pip install dist/*.whl
94-
RUN git clone ${PYTORCH_VISION_REPO} vision
95-
RUN cd vision && git checkout ${PYTORCH_VISION_BRANCH} \
96-
&& python3 setup.py bdist_wheel --dist-dir=dist \
97-
&& pip install dist/*.whl
81+
#FROM base AS build_pytorch
82+
#ARG PYTORCH_BRANCH
83+
#ARG PYTORCH_VISION_BRANCH
84+
#ARG PYTORCH_REPO
85+
#ARG PYTORCH_VISION_REPO
86+
#ARG FA_BRANCH
87+
#ARG FA_REPO
88+
#RUN git clone ${PYTORCH_REPO} pytorch
89+
#RUN cd pytorch && git checkout ${PYTORCH_BRANCH} && \
90+
# pip install -r requirements.txt && git submodule update --init --recursive \
91+
# && python3 tools/amd_build/build_amd.py \
92+
# && CMAKE_PREFIX_PATH=$(python3 -c 'import sys; print(sys.prefix)') python3 setup.py bdist_wheel --dist-dir=dist \
93+
# && pip install dist/*.whl
94+
#RUN git clone ${PYTORCH_VISION_REPO} vision
95+
#RUN cd vision && git checkout ${PYTORCH_VISION_BRANCH} \
96+
# && python3 setup.py bdist_wheel --dist-dir=dist \
97+
# && pip install dist/*.whl
9898
RUN git clone ${FA_REPO}
9999
RUN cd flash-attention \
100100
&& git checkout ${FA_BRANCH} \
@@ -123,12 +123,12 @@ FROM base AS debs
123123
RUN mkdir /app/debs
124124
# RUN --mount=type=bind,from=build_hipblaslt,src=/app/install/,target=/install \
125125
# cp /install/*.deb /app/debs
126-
RUN --mount=type=bind,from=build_triton,src=/app/install/,target=/install \
127-
cp /install/*.whl /app/debs
126+
#RUN --mount=type=bind,from=build_triton,src=/app/install/,target=/install \
127+
# cp /install/*.whl /app/debs
128128
RUN --mount=type=bind,from=build_amdsmi,src=/app/install/,target=/install \
129129
cp /install/*.whl /app/debs
130-
RUN --mount=type=bind,from=build_pytorch,src=/app/install/,target=/install \
131-
cp /install/*.whl /app/debs
130+
#RUN --mount=type=bind,from=build_pytorch,src=/app/install/,target=/install \
131+
# cp /install/*.whl /app/debs
132132
# RUN --mount=type=bind,from=build_aiter,src=/app/install/,target=/install \
133133
# cp /install/*.whl /app/debs
134134

@@ -138,12 +138,12 @@ FROM base AS final
138138
# && perl -p -i -e 's/, hipblas-common-dev \([^)]*?\), /, /g' /var/lib/dpkg/status \
139139
# && perl -p -i -e 's/, hipblaslt-dev \([^)]*?\), /, /g' /var/lib/dpkg/status \
140140
# && perl -p -i -e 's/, hipblaslt \([^)]*?\), /, /g' /var/lib/dpkg/status
141-
RUN --mount=type=bind,from=build_triton,src=/app/install/,target=/install \
142-
pip install /install/*.whl
141+
#RUN --mount=type=bind,from=build_triton,src=/app/install/,target=/install \
142+
# pip install /install/*.whl
143143
RUN --mount=type=bind,from=build_amdsmi,src=/app/install/,target=/install \
144144
pip install /install/*.whl
145-
RUN --mount=type=bind,from=build_pytorch,src=/app/install/,target=/install \
146-
pip install /install/*.whl
145+
#RUN --mount=type=bind,from=build_pytorch,src=/app/install/,target=/install \
146+
# pip install /install/*.whl
147147
# RUN --mount=type=bind,from=build_aiter,src=/app/install/,target=/install \
148148
# pip install /install/*.whl
149149

0 commit comments

Comments
 (0)