Skip to content

Commit b4aa1cf

Browse files
[SWDEV-404135] Add msccl-algorithms directory to PyTorch wheel (#32)
* Bundle msccl-algorithms into wheel * Use correct src path for msccl-algorithms (cherry picked from commit 95b5af3)
1 parent 9e02580 commit b4aa1cf

File tree

1 file changed

+7
-0
lines changed

1 file changed

+7
-0
lines changed

manywheel/build_rocm.sh

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -154,6 +154,11 @@ MIOPEN_SHARE_SRC=$ROCM_HOME/share/miopen/db
154154
MIOPEN_SHARE_DST=share/miopen/db
155155
MIOPEN_SHARE_FILES=($(ls $MIOPEN_SHARE_SRC | grep -E $ARCH))
156156

157+
# RCCL library files
158+
RCCL_SHARE_SRC=$ROCM_HOME/lib/msccl-algorithms
159+
RCCL_SHARE_DST=lib/msccl-algorithms
160+
RCCL_SHARE_FILES=($(ls $RCCL_SHARE_SRC))
161+
157162
# ROCm library files
158163
ROCM_SO_PATHS=()
159164
for lib in "${ROCM_SO_FILES[@]}"
@@ -187,12 +192,14 @@ DEPS_SONAME=(
187192
DEPS_AUX_SRCLIST=(
188193
"${ROCBLAS_LIB_FILES[@]/#/$ROCBLAS_LIB_SRC/}"
189194
"${MIOPEN_SHARE_FILES[@]/#/$MIOPEN_SHARE_SRC/}"
195+
"${RCCL_SHARE_FILES[@]/#/$RCCL_SHARE_SRC/}"
190196
"/opt/amdgpu/share/libdrm/amdgpu.ids"
191197
)
192198

193199
DEPS_AUX_DSTLIST=(
194200
"${ROCBLAS_LIB_FILES[@]/#/$ROCBLAS_LIB_DST/}"
195201
"${MIOPEN_SHARE_FILES[@]/#/$MIOPEN_SHARE_DST/}"
202+
"${RCCL_SHARE_FILES[@]/#/$RCCL_SHARE_DST/}"
196203
"share/libdrm/amdgpu.ids"
197204
)
198205

0 commit comments

Comments
 (0)