File tree Expand file tree Collapse file tree 1 file changed +10
-3
lines changed Expand file tree Collapse file tree 1 file changed +10
-3
lines changed Original file line number Diff line number Diff line change @@ -204,10 +204,17 @@ if [[ $ROCM_INT -ge 50500 ]]; then
204
204
205
205
DEPS_AUX_SRCLIST+=(${MIOPEN_SHARE_FILES[@]/#/ $MIOPEN_SHARE_SRC / } )
206
206
DEPS_AUX_DSTLIST+=(${MIOPEN_SHARE_FILES[@]/#/ $MIOPEN_SHARE_DST / } )
207
- elif [[ $ROCM_INT -ge 50600 ]]; then
207
+ fi
208
+
209
+ if [[ $ROCM_INT -ge 50600 ]]; then
208
210
# RCCL library files
209
- RCCL_SHARE_SRC=$ROCM_HOME /lib/msccl-algorithms
210
- RCCL_SHARE_DST=lib/msccl-algorithms
211
+ if [[ $ROCM_INT -ge 50700 ]]; then
212
+ RCCL_SHARE_SRC=$ROCM_HOME /share/rccl/msccl-algorithms
213
+ RCCL_SHARE_DST=share/rccl/msccl-algorithms
214
+ else
215
+ RCCL_SHARE_SRC=$ROCM_HOME /lib/msccl-algorithms
216
+ RCCL_SHARE_DST=lib/msccl-algorithms
217
+ fi
211
218
RCCL_SHARE_FILES=($( ls $RCCL_SHARE_SRC ) )
212
219
213
220
DEPS_AUX_SRCLIST+=(${RCCL_SHARE_FILES[@]/#/ $RCCL_SHARE_SRC / } )
You can’t perform that action at this time.
0 commit comments