Files
rocm-systems/update_wrapper.sh
T
Galantsev, Dmitrii 6dcb9322f9 update_wrapper.sh - Fix docker
Change-Id: Icb0d80dacfe17222b32bf5616765abc08cafd085

Signed-off-by: Galantsev, Dmitrii <dmitrii.galantsev@amd.com>
2025-01-31 14:55:36 -06:00

56 行
1.8 KiB
Bash
可執行檔

#!/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
}
DOCKER_BUILDKIT=$(docker buildx version >/dev/null 2>&1 && echo 1 || echo 0)
export DOCKER_BUILDKIT
build_docker_image () {
DOCKER_DIR="$DIR/py-interface"
DOCKERFILE="$DOCKER_DIR/Dockerfile"
DOCKERFILE_TIME=$(git log -1 --format=%at -- "$DOCKERFILE")
IMAGE_TIME=$(docker inspect "$DOCKER_NAME" --format '{{.Created}}' 2>/dev/null | date +%s -d- || echo 0)
# Build if Dockerfile is newer than image
if [ "$DOCKERFILE_TIME" -gt "$IMAGE_TIME" ]; then
docker build "$DOCKER_DIR" -f "$DOCKERFILE" -t "$DOCKER_NAME":latest
fi
}
build_docker_image
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]"