Call nvidia-ctk cdi generate on apply-exit hook

Signed-off-by: Carlos Eduardo Arango Gutierrez <[email protected]>

Author: ArangoGutierrez

deployments/systemd/hooks-default.yaml

@@ -16,3 +16,6 @@ hooks:
     - workdir: "/etc/nvidia-mig-manager"
       command: "/bin/bash"
       args: ["-x", "-c", "source hooks.sh; apply-exit"]
+    - workdir: "/etc/nvidia-mig-manager"
+      command: "/bin/bash"
+      args: ["-x", "-c", "source hooks.sh; cdi-refresh"]

deployments/systemd/hooks.sh

@@ -122,3 +122,13 @@ function stop_k8s_pods() {
 	fi
 	return 0
 }
+
+cdi-refresh() {
+    # Check if nvidia-cdi-refresh.service exists
+    if systemctl list-unit-files nvidia-cdi-refresh.service --quiet; then
+        echo "Found nvidia-cdi-refresh.service, calling systemctl..."
+         if ! systemctl start nvidia-cdi-refresh.service; then
+            echo "Error: Failed to start nvidia-cdi-refresh.service" >&2
+        fi
+    fi
+}