Galantsev, Dmitrii
874d8edb3d
Also modify README to include: export PATH="$PATH:/opt/rocm/bin" Change-Id: Ib4513d61fd3e6b23cc789491c11d83e6a6de947a Signed-off-by: Galantsev, Dmitrii <dmitrii.galantsev@amd.com>
52 lines
1.7 KiB
Bash
Executable File
52 lines
1.7 KiB
Bash
Executable File
#!/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" | 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]"
|