Skip to content

Commit 04c644d

Browse files
committed
[ROCm] reduce disk space used in image
1 parent 0b51144 commit 04c644d

File tree

2 files changed

+14
-10
lines changed

2 files changed

+14
-10
lines changed

common/install_miopen.sh

+10-9
Original file line numberDiff line numberDiff line change
@@ -88,7 +88,7 @@ fi
8888

8989
git clone https://github.com/ROCmSoftwarePlatform/MIOpen -b ${MIOPEN_BRANCH}
9090
pushd MIOpen
91-
# Don't build MLIR to save docker build time
91+
# Don't build MLIR to save docker build time
9292
# since we are disabling MLIR backend for MIOpen anyway
9393
if [[ $ROCM_INT -ge 50400 ]] && [[ $ROCM_INT -lt 50500 ]]; then
9494
sed -i '/rocMLIR/d' requirements.txt
@@ -97,6 +97,15 @@ elif [[ $ROCM_INT -ge 50200 ]] && [[ $ROCM_INT -lt 50400 ]]; then
9797
fi
9898
## MIOpen minimum requirements
9999
cmake -P install_deps.cmake --minimum
100+
101+
# clean up since CI runner was running out of disk space
102+
rm -rf /usr/local/cget
103+
rm -rf /tmp/*
104+
yum clean all
105+
rm -rf /var/cache/yum
106+
rm -rf /var/lib/yum/yumdb
107+
rm -rf /var/lib/yum/history
108+
100109
## Build MIOpen
101110
mkdir -p build
102111
cd build
@@ -107,14 +116,6 @@ PKG_CONFIG_PATH=/usr/local/lib/pkgconfig CXX=${ROCM_INSTALL_PATH}/llvm/bin/clang
107116
make MIOpen -j $(nproc)
108117
make -j $(nproc) package
109118

110-
# clean up since CI runner was running out of disk space
111-
rm -rf /usr/local/cget
112-
rm -rf /tmp/*
113-
yum clean all
114-
rm -rf /var/cache/yum
115-
rm -rf /var/lib/yum/yumdb
116-
rm -rf /var/lib/yum/history
117-
118119
yum install -y miopen-*.rpm
119120
popd
120121
rm -rf MIOpen

common/install_rocm_magma.sh

+4-1
Original file line numberDiff line numberDiff line change
@@ -37,5 +37,8 @@ make -f make.gen.hipMAGMA -j $(nproc)
3737
LANG=C.UTF-8 make lib/libmagma.so -j $(nproc) MKLROOT="${MKLROOT}"
3838
make testing/testing_dgemm -j $(nproc) MKLROOT="${MKLROOT}"
3939
popd
40-
mv magma /opt/rocm
40+
mkdir -p /opt/rocm/magma
41+
mv magma/include /opt/rocm/magma
42+
mv magma/lib /opt/rocm/magma
43+
rm -rf magma
4144

0 commit comments

Comments
 (0)