Skip to content

Commit

Permalink
Fix erroneous logic that was skipping msccl files even for ROCm5.6; u…
Browse files Browse the repository at this point in the history
…pdate msccl path for ROCm5.7

(cherry picked from commit 36c10cc)
  • Loading branch information
jithunnair-amd committed Aug 6, 2023
1 parent f4464c9 commit a581835
Showing 1 changed file with 10 additions and 3 deletions.
13 changes: 10 additions & 3 deletions manywheel/build_rocm.sh
Original file line number Diff line number Diff line change
Expand Up @@ -204,10 +204,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/})
Expand Down

0 comments on commit a581835

Please sign in to comment.