Skip to content

Commit 0feb91a

Browse files
authored
Building hipblaslt including the clients (ROCm#462)
1 parent 27f6c7b commit 0feb91a

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

Dockerfile.rocm_base

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -31,7 +31,7 @@ ENV DEBIAN_FRONTEND=noninteractive
3131

3232
# Install Python and other dependencies
3333
RUN apt-get update -y \
34-
&& apt-get install -y software-properties-common git curl sudo vim less \
34+
&& apt-get install -y software-properties-common git curl sudo vim less libgfortran5 \
3535
&& add-apt-repository ppa:deadsnakes/ppa \
3636
&& apt-get update -y \
3737
&& apt-get install -y python${PYTHON_VERSION} python${PYTHON_VERSION}-dev python${PYTHON_VERSION}-venv \
@@ -60,7 +60,8 @@ RUN cd hipBLAS-common \
6060
RUN git clone https://github.com/ROCm/hipBLASLt
6161
RUN cd hipBLASLt \
6262
&& git checkout ${HIPBLASLT_BRANCH} \
63-
&& ./install.sh -d --architecture ${PYTORCH_ROCM_ARCH} ${LEGACY_HIPBLASLT_OPTION} \
63+
&& apt-get install -y llvm-dev \
64+
&& ./install.sh -dc --architecture ${PYTORCH_ROCM_ARCH} ${LEGACY_HIPBLASLT_OPTION} \
6465
&& cd build/release \
6566
&& make package
6667
RUN mkdir -p /app/install && cp /app/hipBLASLt/build/release/*.deb /app/hipBLAS-common/build/*.deb /app/install

0 commit comments

Comments
 (0)