diff --git a/manywheel/build_rocm.sh b/manywheel/build_rocm.sh index 04768ec8f..1183cfff3 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[@]}" @@ -196,8 +191,13 @@ DEPS_AUX_DSTLIST=( ) if [[ $ROCM_INT -ge 50500 ]]; then - DEPS_AUX_SRCLIST[${#DEPS_AUX_SRCLIST[@]}]="${MIOPEN_SHARE_FILES[@]/#/$MIOPEN_SHARE_SRC/}" - DEPS_AUX_DSTLIST[${#DEPS_AUX_DSTLIST[@]}]="${MIOPEN_SHARE_FILES[@]/#/$MIOPEN_SHARE_DST/}" + # 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+=(${MIOPEN_SHARE_FILES[@]/#/$MIOPEN_SHARE_SRC/}) + DEPS_AUX_DSTLIST+=(${MIOPEN_SHARE_FILES[@]/#/$MIOPEN_SHARE_DST/}) fi echo "PYTORCH_ROCM_ARCH: ${PYTORCH_ROCM_ARCH}"