Skip to content

Commit 15eb5d9

Browse files
Fix erroneous logic that was skipping msccl files even for ROCm5.6; update msccl path for ROCm5.7 (#1546)
(cherry picked from commit 5e89be5)
1 parent adc1112 commit 15eb5d9

File tree

1 file changed

+10
-3
lines changed

1 file changed

+10
-3
lines changed

manywheel/build_rocm.sh

Lines changed: 10 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -204,10 +204,17 @@ if [[ $ROCM_INT -ge 50500 ]]; then
204204

205205
DEPS_AUX_SRCLIST+=(${MIOPEN_SHARE_FILES[@]/#/$MIOPEN_SHARE_SRC/})
206206
DEPS_AUX_DSTLIST+=(${MIOPEN_SHARE_FILES[@]/#/$MIOPEN_SHARE_DST/})
207-
elif [[ $ROCM_INT -ge 50600 ]]; then
207+
fi
208+
209+
if [[ $ROCM_INT -ge 50600 ]]; then
208210
# RCCL library files
209-
RCCL_SHARE_SRC=$ROCM_HOME/lib/msccl-algorithms
210-
RCCL_SHARE_DST=lib/msccl-algorithms
211+
if [[ $ROCM_INT -ge 50700 ]]; then
212+
RCCL_SHARE_SRC=$ROCM_HOME/share/rccl/msccl-algorithms
213+
RCCL_SHARE_DST=share/rccl/msccl-algorithms
214+
else
215+
RCCL_SHARE_SRC=$ROCM_HOME/lib/msccl-algorithms
216+
RCCL_SHARE_DST=lib/msccl-algorithms
217+
fi
211218
RCCL_SHARE_FILES=($(ls $RCCL_SHARE_SRC))
212219

213220
DEPS_AUX_SRCLIST+=(${RCCL_SHARE_FILES[@]/#/$RCCL_SHARE_SRC/})

0 commit comments

Comments
 (0)