[SWDEV-404135] Add msccl-algorithms directory to PyTorch wheel (#32)

* Bundle msccl-algorithms into wheel * Use correct src path for msccl-algorithms
ROCm · Jun 12, 2023 · 95b5af3 · 95b5af3
1 parent 9e02580
commit 95b5af3
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/manywheel/build_rocm.sh b/manywheel/build_rocm.sh
@@ -154,6 +154,11 @@ MIOPEN_SHARE_SRC=$ROCM_HOME/share/miopen/db
 MIOPEN_SHARE_DST=share/miopen/db
 MIOPEN_SHARE_FILES=($(ls $MIOPEN_SHARE_SRC | grep -E $ARCH))
 
+# RCCL library files
+RCCL_SHARE_SRC=$ROCM_HOME/lib/msccl-algorithms
+RCCL_SHARE_DST=lib/msccl-algorithms
+RCCL_SHARE_FILES=($(ls $RCCL_SHARE_SRC))
+
 # ROCm library files
 ROCM_SO_PATHS=()
 for lib in "${ROCM_SO_FILES[@]}"
@@ -187,12 +192,14 @@ DEPS_SONAME=(
 DEPS_AUX_SRCLIST=(
  "${ROCBLAS_LIB_FILES[@]/#/$ROCBLAS_LIB_SRC/}"
  "${MIOPEN_SHARE_FILES[@]/#/$MIOPEN_SHARE_SRC/}"
+ "${RCCL_SHARE_FILES[@]/#/$RCCL_SHARE_SRC/}"
  "/opt/amdgpu/share/libdrm/amdgpu.ids"
 )
 
 DEPS_AUX_DSTLIST=(
  "${ROCBLAS_LIB_FILES[@]/#/$ROCBLAS_LIB_DST/}"
  "${MIOPEN_SHARE_FILES[@]/#/$MIOPEN_SHARE_DST/}"
+ "${RCCL_SHARE_FILES[@]/#/$RCCL_SHARE_DST/}"
  "share/libdrm/amdgpu.ids"
 )