|
3 | 3 | set -ex
|
4 | 4 |
|
5 | 5 | # Create the tmp dir to extract into
|
6 |
| -EXTRACTDIR_ROOT=/extract_miopen_rpm |
| 6 | +EXTRACTDIR_ROOT=/extract_miopen_rpms |
7 | 7 | mkdir -p ${EXTRACTDIR_ROOT}
|
8 | 8 | echo "Creating temporary directory for rpm download..."
|
9 | 9 |
|
| 10 | +MIOPEN_RPM_SOURCE_ARRAY=($MIOPEN_RPM_SOURCE) |
10 | 11 | # Fail if rpm source is not available
|
11 |
| -if ! wget -P ${EXTRACTDIR_ROOT} ${MIOPEN_RPM_SOURCE}; then |
| 12 | +#if [[ ! wget -P ${EXTRACTDIR_ROOT} ${MIOPEN_RPM_SOURCE_ARRAY[0]} ]] || [[ ! wget -P ${EXTRACTDIR_ROOT} ${MIOPEN_RPM_SOURCE_ARRAY[1]} ]]; then |
| 13 | +if ! wget -P ${EXTRACTDIR_ROOT} ${MIOPEN_RPM_SOURCE_ARRAY[0]} || ! wget -P ${EXTRACTDIR_ROOT} ${MIOPEN_RPM_SOURCE_ARRAY[1]}; then |
12 | 14 | echo 'ERROR: Failed to download MIOpen package.'
|
13 | 15 | exit 1
|
14 | 16 | fi
|
15 | 17 | echo "MIOpen package download complete..."
|
16 | 18 |
|
17 | 19 | # Extract rpm in EXTRACT_DIR
|
18 | 20 | cd ${EXTRACTDIR_ROOT}
|
19 |
| -miopen_rpm=$(ls *.rpm) |
20 |
| -rpm2cpio ${miopen_rpm} | cpio -idmv |
21 |
| - |
22 |
| -# Copy libMIOpen.so.1 over existing |
23 |
| -source_file=$(ls opt/rocm*/lib/libMIOpen.so.1.0*) |
24 |
| -dest_file=$(ls /opt/rocm-${ROCM_VERSION}*/lib/libMIOpen.so.1.0*) |
25 |
| -if [ -e ${source_file} ] && [ -e ${dest_file} ]; then |
26 |
| - echo "Source .so: ${source_file}" |
27 |
| - echo "Dest .so: ${dest_file}" |
28 |
| - cp $source_file $dest_file |
29 |
| -else |
30 |
| - echo 'ERROR: either the source or destination path for libMIOpen.so.1.0 does not exist' |
31 |
| - exit 1 |
32 |
| -fi |
33 |
| -# Copy db files over existing |
34 |
| -source="opt/rocm*/share/miopen/db/*" |
35 |
| -dest="/opt/rocm-${ROCM_VERSION}*/share/miopen/db/" |
36 |
| -# Use backslash to run cp in (non-aliased) non-interactive mode |
37 |
| -\cp -R $source $dest |
| 21 | +miopen_rpms=$(ls *.rpm) |
| 22 | +for miopen_rpm in ${miopen_rpms}; do |
| 23 | + rpm2cpio ${miopen_rpm} | cpio -idmv |
| 24 | +done |
38 | 25 |
|
| 26 | +# Copy libMIOpen.so, headers and db files over existing |
| 27 | +files="lib/ include/ share/miopen/db/" |
| 28 | +for file in $files; do |
| 29 | + source="opt/rocm*/${file}/*" |
| 30 | + dest="/opt/rocm/${file}" |
| 31 | + # Use backslash to run cp in (non-aliased) non-interactive mode |
| 32 | + \cp -dR $source $dest |
| 33 | +done |
39 | 34 | echo "libMIOpen so file from RPM copied to existing MIOpen install..."
|
40 | 35 |
|
41 | 36 | # Clean up extracted dir
|
|
0 commit comments