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.

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

Dockerfile - Add rocm6.0 dockerfile #602

Merged
merged 16 commits into from
Dec 27, 2023
Prev Previous commit
Next Next commit
reivse hipblaslt build
  • Loading branch information
yukirora authored Dec 25, 2023
commit f34c44168de983e2945180f50e797e16bb0a62c1
5 changes: 5 additions & 0 deletions third_party/Makefile
Original file line number Diff line number Diff line change
Expand Up @@ -7,6 +7,7 @@ MPI_HOME ?= /usr/local/mpi
HIP_HOME ?= /opt/rocm/hip
RCCL_HOME ?= /opt/rocm/rccl
HPCX_HOME ?= /opt/hpcx
ROCM_PATH ?= /opt/rocm

CUDA_VER ?= $(shell nvcc --version | grep 'release' | awk '{print $$6}' | cut -c2- | cut -d '.' -f1-2)
ROCBLAS_BRANCH ?= rocm-$(shell dpkg -l | grep 'rocm-dev ' | awk '{print $$3}' | cut -d '.' -f1-3)
Expand Down Expand Up @@ -113,6 +114,10 @@ rocm_rocblas: sb_micro_path
rocm_hipblaslt: sb_micro_path
@if [ ! -e $(SB_MICRO_PATH)/bin/hipblaslt-bench ] && [ -z `which hipblaslt-bench` ]; then \
if [ -d hipBLASLt ]; then rm -rf hipBLASLt; fi; \
rm -rf ${ROCM_PATH}/lib/hipblaslt \
rm -rf ${ROCM_PATH}/lib/cmake/hipblaslt \
rm -rf ${ROCM_PATH}/include/hipblaslt \
rm -rf ${ROCM_PATH}/share/doc/hipblaslt \
git clone -b ${HIPBLASLT_BRANCH} https://github.com/ROCmSoftwarePlatform/hipBLASLt.git ./hipBLASLt; \
cd ./hipBLASLt && ./install.sh -dc; \
cp -v $(SB_MICRO_PATH)/third_party/hipBLASLt/build/release/clients/staging/hipblaslt-bench $(SB_MICRO_PATH)/bin/; \
Expand Down
Loading