diff --git a/manywheel/build_rocm.sh b/manywheel/build_rocm.sh index 0c1650f9b..4b89b1e86 100755 --- a/manywheel/build_rocm.sh +++ b/manywheel/build_rocm.sh @@ -202,10 +202,17 @@ if [[ $ROCM_INT -ge 50500 ]]; then DEPS_AUX_SRCLIST+=(${MIOPEN_SHARE_FILES[@]/#/$MIOPEN_SHARE_SRC/}) DEPS_AUX_DSTLIST+=(${MIOPEN_SHARE_FILES[@]/#/$MIOPEN_SHARE_DST/}) -elif [[ $ROCM_INT -ge 50600 ]]; then +fi + +if [[ $ROCM_INT -ge 50600 ]]; then # RCCL library files - RCCL_SHARE_SRC=$ROCM_HOME/lib/msccl-algorithms - RCCL_SHARE_DST=lib/msccl-algorithms + if [[ $ROCM_INT -ge 50700 ]]; then + RCCL_SHARE_SRC=$ROCM_HOME/share/rccl/msccl-algorithms + RCCL_SHARE_DST=share/rccl/msccl-algorithms + else + RCCL_SHARE_SRC=$ROCM_HOME/lib/msccl-algorithms + RCCL_SHARE_DST=lib/msccl-algorithms + fi RCCL_SHARE_FILES=($(ls $RCCL_SHARE_SRC)) DEPS_AUX_SRCLIST+=(${RCCL_SHARE_FILES[@]/#/$RCCL_SHARE_SRC/})