Skip to content

Commit 36c10cc

Browse files
Fix erroneous logic that was skipping msccl files even for ROCm5.6; update msccl path for ROCm5.7
1 parent 2a03668 commit 36c10cc

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
@@ -202,10 +202,17 @@ if [[ $ROCM_INT -ge 50500 ]]; then
202202

203203
DEPS_AUX_SRCLIST+=(${MIOPEN_SHARE_FILES[@]/#/$MIOPEN_SHARE_SRC/})
204204
DEPS_AUX_DSTLIST+=(${MIOPEN_SHARE_FILES[@]/#/$MIOPEN_SHARE_DST/})
205-
elif [[ $ROCM_INT -ge 50600 ]]; then
205+
fi
206+
207+
if [[ $ROCM_INT -ge 50600 ]]; then
206208
# RCCL library files
207-
RCCL_SHARE_SRC=$ROCM_HOME/lib/msccl-algorithms
208-
RCCL_SHARE_DST=lib/msccl-algorithms
209+
if [[ $ROCM_INT -ge 50700 ]]; then
210+
RCCL_SHARE_SRC=$ROCM_HOME/share/rccl/msccl-algorithms
211+
RCCL_SHARE_DST=share/rccl/msccl-algorithms
212+
else
213+
RCCL_SHARE_SRC=$ROCM_HOME/lib/msccl-algorithms
214+
RCCL_SHARE_DST=lib/msccl-algorithms
215+
fi
209216
RCCL_SHARE_FILES=($(ls $RCCL_SHARE_SRC))
210217

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

0 commit comments

Comments
 (0)