From 113356d00dc55561f48d0dbaa8935211d7528768 Mon Sep 17 00:00:00 2001 From: Jithun Nair Date: Fri, 9 Jun 2023 16:00:02 +0000 Subject: [PATCH 1/3] Use MIOpen db logic for ROCm5.5 or later --- manywheel/build_rocm.sh | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/manywheel/build_rocm.sh b/manywheel/build_rocm.sh index 04768ec8f..869a10b2d 100755 --- a/manywheel/build_rocm.sh +++ b/manywheel/build_rocm.sh @@ -196,6 +196,11 @@ DEPS_AUX_DSTLIST=( ) if [[ $ROCM_INT -ge 50500 ]]; then + # MIOpen library files + MIOPEN_SHARE_SRC=$ROCM_HOME/share/miopen/db + MIOPEN_SHARE_DST=share/miopen/db + MIOPEN_SHARE_FILES=($(ls $MIOPEN_SHARE_SRC | grep -E $ARCH)) + DEPS_AUX_SRCLIST[${#DEPS_AUX_SRCLIST[@]}]="${MIOPEN_SHARE_FILES[@]/#/$MIOPEN_SHARE_SRC/}" DEPS_AUX_DSTLIST[${#DEPS_AUX_DSTLIST[@]}]="${MIOPEN_SHARE_FILES[@]/#/$MIOPEN_SHARE_DST/}" fi From 45a1b4961853fc0ff7cf97e29012979834007050 Mon Sep 17 00:00:00 2001 From: Jithun Nair Date: Fri, 9 Jun 2023 21:06:45 +0000 Subject: [PATCH 2/3] Remove moved lines --- manywheel/build_rocm.sh | 5 ----- 1 file changed, 5 deletions(-) diff --git a/manywheel/build_rocm.sh b/manywheel/build_rocm.sh index 869a10b2d..400fa8b0c 100755 --- a/manywheel/build_rocm.sh +++ b/manywheel/build_rocm.sh @@ -150,11 +150,6 @@ ARCH_SPECIFIC_FILES=$(ls $ROCBLAS_LIB_SRC | grep -E $ARCH) OTHER_FILES=$(ls $ROCBLAS_LIB_SRC | grep -v gfx) ROCBLAS_LIB_FILES=($ARCH_SPECIFIC_FILES $OTHER_FILES) -# MIOpen library files -MIOPEN_SHARE_SRC=$ROCM_HOME/share/miopen/db -MIOPEN_SHARE_DST=share/miopen/db -MIOPEN_SHARE_FILES=($(ls $MIOPEN_SHARE_SRC | grep -E $ARCH)) - # ROCm library files ROCM_SO_PATHS=() for lib in "${ROCM_SO_FILES[@]}" From 56b8cc1ebdfb2e70683f073b8443313ad9a6c431 Mon Sep 17 00:00:00 2001 From: Jithun Nair Date: Fri, 9 Jun 2023 21:17:40 +0000 Subject: [PATCH 3/3] Fix bash logic to add elements of array to existing array --- manywheel/build_rocm.sh | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/manywheel/build_rocm.sh b/manywheel/build_rocm.sh index 400fa8b0c..1183cfff3 100755 --- a/manywheel/build_rocm.sh +++ b/manywheel/build_rocm.sh @@ -196,8 +196,8 @@ if [[ $ROCM_INT -ge 50500 ]]; then MIOPEN_SHARE_DST=share/miopen/db MIOPEN_SHARE_FILES=($(ls $MIOPEN_SHARE_SRC | grep -E $ARCH)) - DEPS_AUX_SRCLIST[${#DEPS_AUX_SRCLIST[@]}]="${MIOPEN_SHARE_FILES[@]/#/$MIOPEN_SHARE_SRC/}" - DEPS_AUX_DSTLIST[${#DEPS_AUX_DSTLIST[@]}]="${MIOPEN_SHARE_FILES[@]/#/$MIOPEN_SHARE_DST/}" + DEPS_AUX_SRCLIST+=(${MIOPEN_SHARE_FILES[@]/#/$MIOPEN_SHARE_SRC/}) + DEPS_AUX_DSTLIST+=(${MIOPEN_SHARE_FILES[@]/#/$MIOPEN_SHARE_DST/}) fi echo "PYTORCH_ROCM_ARCH: ${PYTORCH_ROCM_ARCH}"