From ee5c7662d8a5add3bacb0c945a85e1570d1a0218 Mon Sep 17 00:00:00 2001 From: Yuting Jiang Date: Mon, 13 Sep 2021 18:07:44 +0800 Subject: [PATCH] Benchmarks: Build Pipeline - Restore rocblas build logic (#197) **Description** restore rocblas build logic to cancel support of rocblas build in rocm4.0_ubuntu18.04_py3.6_pytorch_1.7.0 base image. **Major Revision** - restore rocblas build logic, remove gpu target limit and other resource limit for rocm4.0. --- third_party/Makefile | 8 +------- 1 file changed, 1 insertion(+), 7 deletions(-) diff --git a/third_party/Makefile b/third_party/Makefile index 5531cabb7..0a80e4dce 100755 --- a/third_party/Makefile +++ b/third_party/Makefile @@ -8,7 +8,6 @@ MPI_HOME ?= /usr/local/mpi HIP_HOME ?= /opt/rocm/hip RCCL_HOME ?= /opt/rocm/rccl ROCM_VERSION ?= rocm-$(shell dpkg -l | grep 'rocm-dev ' | awk '{print $$3}' | cut -d '.' -f1-3) -ROCM_ARCH ?= $(shell rocminfo | grep " gfx" | uniq | awk '{print $$2}') .PHONY: all cuda rocm common cuda_cutlass cuda_bandwidthTest cuda_nccl_tests cuda_perftest rocm_perftest fio rocm_rccl_tests rocm_rocblas rocm_bandwidthTest @@ -81,12 +80,7 @@ rocm_rocblas: sb_micro_path ifeq (, $(wildcard $(SB_MICRO_PATH)/bin/rocblas-bench)) if [ -d rocBLAS ]; then rm -rf rocBLAS; fi git clone -b ${ROCM_VERSION} https://github.com/ROCmSoftwarePlatform/rocBLAS.git ./rocBLAS -ifeq (${ROCM_VERSION}, rocm-4.0.0) - sed -i '/CMAKE_MATCH_1/a\ get_filename_component(HIP_CLANG_ROOT "$${HIP_CLANG_ROOT}" DIRECTORY)' /opt/rocm/hip/lib/cmake/hip/hip-config.cmake - cd ./rocBLAS && HIPCC_COMPILE_FLAGS_APPEND="-D_OPENMP=201811 -O3 -Wno-format-nonliteral -DCMAKE_HAVE_LIBC_PTHREAD -parallel-jobs=2" HIPCC_LINK_FLAGS_APPEND="-lpthread -O3 -parallel-jobs=2" ./install.sh -idc -a ${ROCM_ARCH} -else - cd ./rocBLAS && ./install.sh -idc -endif + cd ./rocBLAS && ./install.sh --dependencies --clients-only cp -v ./rocBLAS/build/release/clients/staging/rocblas-bench $(SB_MICRO_PATH)/bin/ endif