Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

Dockerfile - Add rocm6.0 dockerfile #602

Merged
merged 16 commits into from
Dec 27, 2023
Merged

Dockerfile - Add rocm6.0 dockerfile #602

Changes from 1 commit
Commits
Show all changes
16 commits
Select commit Hold shift + click to select a range
407b12e
add rocm6.0 dockerfile
yukirora Dec 21, 2023
3d00250
add pipeline for rocm6.0
yukirora Dec 21, 2023
55ea25c
bug fix
yukirora Dec 21, 2023
2858d84
update lib version and bugfix
yukirora Dec 22, 2023
dc7a64e
Merge branch 'release/0.10' into yutji/rocm-6.0-dockerfile
yukirora Dec 23, 2023
fabaccb
bugfix
yukirora Dec 25, 2023
f34c441
reivse hipblaslt build
yukirora Dec 25, 2023
b19e774
bugfix
yukirora Dec 25, 2023
28eee19
update
yukirora Dec 25, 2023
0ff39d3
update perftest version
Andyli1007 Dec 25, 2023
692020e
bugfix
yukirora Dec 25, 2023
d5cf4e0
update
yukirora Dec 25, 2023
fd87f1a
update mpi build to replace the original static mpi build in base image
yukirora Dec 25, 2023
6148591
bugfix
Andyli1007 Dec 26, 2023
e6052da
resotre hipblaslt
Dec 27, 2023
1699be0
Merge branch 'release/0.10' into yutji/rocm-6.0-dockerfile
yukirora Dec 27, 2023
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
bugfix
  • Loading branch information
@yukirora
yukirora authored Dec 25, 2023
commit 692020e80b9681b62c7677e02bfa23e3fe89c1f7
8 changes: 5 additions & 3 deletions dockerfile/rocm5.7.x.dockerfile
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -166,11 +166,13 @@ RUN apt install rocm-cmake -y && \
WORKDIR ${SB_HOME}

ADD third_party third_party
RUN make RCCL_HOME=/opt/rccl/build/ MPI_HOME=/usr/local ROCBLAS_BRANCH=release/rocm-rel-5.7.1.1 HIPBLASLT_BRANCH=release-staging/rocm-rel-5.7 ROCM_VER=rocm-5.5.0 -C third_party rocm -o cpu_hpl -o cpu_stream -o megatron_lm
# Apply patch
RUN cd third_party/perftest && \
git apply ../perftest_rocm6.patch
RUN make ROCM_PATH=/opt/rocm-5.7.0 RCCL_HOME=/opt/rccl/build/ MPI_HOME=/usr/local 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 . .
#ENV USE_HIPBLASLT_DATATYPE=1
ENV CXX=/opt/rocm/bin/hipcc
RUN python3 -m pip install .[amdworker] && \
make cppbuild && \
CXX=/opt/rocm/bin/hipcc make cppbuild && \
make postinstall
Loading