synchronization

gpushare-scheduler-extender/deployer/docker/device-plugin-recover/dp-evict/dp-recover-on-host.sh

@@ -0,0 +1,34 @@
+#!/usr/bin/env bash
+
+set -e -x
+
+dir="/etc/kubernetes/manifests"
+backup_dir="/etc/kubernetes/manifests-backup"
+
+public::common::log() {
+	echo $(date +"[%Y%m%d %H:%M:%S]: ") $1
+}
+
+public::recover::gpu-device-plugin() {
+
+    if [  -f $dir/nvidia-device-plugin.yml ]; then
+		public::common::log "Skip recovering nvidia-device-plugin.yml, because it already exist."
+    else
+    	if [  -f $backup_dir/nvidia-device-plugin.yml ]; then
+			cp -f $backup_dir/nvidia-device-plugin.yml $dir/nvidia-device-plugin.yml
+			public::common::log "Finish recovering nvidia-device-plugin.yml."
+		else
+			public::common::log "No nvidia-device-plugin.yml to recover."
+		fi	
+	fi
+}
+
+main() {
+
+	public::recover::gpu-device-plugin
+
+	touch /ready
+	while sleep 3600; do :; done
+}
+
+main "$@"

gpushare-scheduler-extender/deployer/docker/device-plugin-recover/dp-evict/dp-recover.sh

@@ -0,0 +1,11 @@
+#!/usr/bin/env bash
+set -xe
+
+if [ -d "/k8s-host" ]; then
+	rm -rf /k8s-host/usr/local/dp-evict
+	mkdir -p /k8s-host/usr/local/dp-evict
+	cp -r /dp-evict/* /k8s-host/usr/local/dp-evict
+	chmod -R +x /k8s-host/usr/local/dp-evict
+	chroot /k8s-host /usr/local/dp-evict/dp-recover-on-host.sh "$@"
+	while sleep 3600; do :; done
+fi