Skip to content

Commit

Permalink
fix rocm5.7 rocblas build error
Browse files Browse the repository at this point in the history
  • Loading branch information
yukirora authored Jul 25, 2024
1 parent fe0d8a1 commit 656506a
Showing 1 changed file with 11 additions and 0 deletions.
11 changes: 11 additions & 0 deletions dockerfile/rocm5.7.x.dockerfile
Original file line number Diff line number Diff line change
Expand Up @@ -167,6 +167,17 @@ ADD third_party third_party
# Apply patch
RUN cd third_party/perftest && \
git apply ../perftest_rocm6.patch

# Update package index and install dependencies of ROCBLAS
RUN apt-get update && \
apt-get install -y software-properties-common && \
add-apt-repository universe && \
apt-get update && \
apt-get install -y --fix-missing \
make python3 python3-yaml python3-venv python3-joblib 'python3*-pip' libmsgpack-dev gfortran libomp-dev && \
apt-get clean && \
rm -rf /var/lib/apt/lists/*

RUN make RCCL_HOME=/opt/rccl/build/ ROCBLAS_BRANCH=release/rocm-rel-5.7.1.1 HIPBLASLT_BRANCH=release/rocm-rel-5.7 ROCM_VER=rocm-5.5.0 -C third_party rocm -o cpu_hpl -o cpu_stream -o megatron_lm

ADD . .
Expand Down

0 comments on commit 656506a

Please sign in to comment.