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