-
Notifications
You must be signed in to change notification settings - Fork 27
/
update_wrapper.sh
executable file
·51 lines (41 loc) · 1.75 KB
/
update_wrapper.sh
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
#!/usr/bin/env bash
# this program generates py-interface/amdsmi_wrapper.py
set -eu
# get current dir
DIR=$( cd -- "$( dirname -- "${BASH_SOURCE[0]}" )" &> /dev/null && pwd )
# override by calling this script with:
# DOCKER_NAME=yourdockername ./update_wrapper.sh
DOCKER_NAME="${DOCKER_NAME:-dmitriigalantsev/amdsmi_wrapper_updater}"
command -v docker &>/dev/null || {
echo "Please install docker!" >&2
exit 1
}
does_image_exist () {
docker images | grep -q "$DOCKER_NAME"
}
if ! does_image_exist; then
# if you prefer to not generate it yourself:
# pull from https://hub.docker.com/r/dmitriigalantsev/amdsmi_wrapper_updater
# using the following command:
# docker pull dmitriigalantsev/amdsmi_wrapper_updater
echo "No docker image found! Generating one"
# set to 0 because it's compatible with more systems
DOCKER_BUILDKIT="${DOCKER_BUILDKIT:=0}" docker build "$DIR/py-interface" -t "$DOCKER_NAME":latest
fi
ENABLE_ESMI_LIB=""
# source ENABLE_ESMI_LIB variable from the previous build if it exists
if [ -e 'build/CMakeCache.txt' ]; then
GREP_RESULT=$(grep "ENABLE_ESMI_LIB.*=" "${DIR}/build/CMakeCache.txt" | tail -n 1 | cut -d = -f 2)
ENABLE_ESMI_LIB="-DENABLE_ESMI_LIB=$GREP_RESULT"
echo "ENABLE_ESMI_LIB: [$ENABLE_ESMI_LIB]"
fi
docker run --rm -ti --volume "$DIR":/src:rw "$DOCKER_NAME":latest bash -c "
cp -r /src /tmp/src \
&& cd /tmp/src \
&& rm -rf build .cache \
&& cmake -B build -DBUILD_WRAPPER=ON $ENABLE_ESMI_LIB \
&& make -C build -j $(nproc) \
&& cp /tmp/src/py-interface/amdsmi_wrapper.py /src/py-interface/amdsmi_wrapper.py \
&& chown --reference /src/py-interface/CMakeLists.txt /src/py-interface/amdsmi_wrapper.py"
echo -e "Generated new wrapper!
[$DIR/py-interface/amdsmi_wrapper.py]"