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 (pytorch#1546)

(cherry picked from commit 5e89be5)
  • Loading branch information
jithunnair-amd authored Sep 21, 2023
1 parent adc1112 commit 15eb5d9
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 15eb5d9

Please sign in to comment.