Skip to content

Commit d7f499d

Browse files
Fix ROCm5.4.2 builds breakage (#1421)
* Use MIOpen db logic for ROCm5.5 or later * Remove moved lines * Fix bash logic to add elements of array to existing array
1 parent 13164b4 commit d7f499d

File tree

1 file changed

+7
-7
lines changed

1 file changed

+7
-7
lines changed

manywheel/build_rocm.sh

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -150,11 +150,6 @@ ARCH_SPECIFIC_FILES=$(ls $ROCBLAS_LIB_SRC | grep -E $ARCH)
150150
OTHER_FILES=$(ls $ROCBLAS_LIB_SRC | grep -v gfx)
151151
ROCBLAS_LIB_FILES=($ARCH_SPECIFIC_FILES $OTHER_FILES)
152152

153-
# MIOpen library files
154-
MIOPEN_SHARE_SRC=$ROCM_HOME/share/miopen/db
155-
MIOPEN_SHARE_DST=share/miopen/db
156-
MIOPEN_SHARE_FILES=($(ls $MIOPEN_SHARE_SRC | grep -E $ARCH))
157-
158153
# ROCm library files
159154
ROCM_SO_PATHS=()
160155
for lib in "${ROCM_SO_FILES[@]}"
@@ -196,8 +191,13 @@ DEPS_AUX_DSTLIST=(
196191
)
197192

198193
if [[ $ROCM_INT -ge 50500 ]]; then
199-
DEPS_AUX_SRCLIST[${#DEPS_AUX_SRCLIST[@]}]="${MIOPEN_SHARE_FILES[@]/#/$MIOPEN_SHARE_SRC/}"
200-
DEPS_AUX_DSTLIST[${#DEPS_AUX_DSTLIST[@]}]="${MIOPEN_SHARE_FILES[@]/#/$MIOPEN_SHARE_DST/}"
194+
# MIOpen library files
195+
MIOPEN_SHARE_SRC=$ROCM_HOME/share/miopen/db
196+
MIOPEN_SHARE_DST=share/miopen/db
197+
MIOPEN_SHARE_FILES=($(ls $MIOPEN_SHARE_SRC | grep -E $ARCH))
198+
199+
DEPS_AUX_SRCLIST+=(${MIOPEN_SHARE_FILES[@]/#/$MIOPEN_SHARE_SRC/})
200+
DEPS_AUX_DSTLIST+=(${MIOPEN_SHARE_FILES[@]/#/$MIOPEN_SHARE_DST/})
201201
fi
202202

203203
echo "PYTORCH_ROCM_ARCH: ${PYTORCH_ROCM_ARCH}"

0 commit comments

Comments
 (0)