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; refresh-cdi"]

deployments/systemd/hooks.sh

@@ -122,3 +122,15 @@ function stop_k8s_pods() {
 	fi
 	return 0
 }
+
+# refresh-cdi triggers the nvidia-cdi-refresh service to regenerate CDI
+# specifications, making updated GPU devices available to container runtimes.
+function refresh-cdi() {
+    # 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..." >&2
+        if ! systemctl restart nvidia-cdi-refresh.service; then
+            echo "Error: Failed to start nvidia-cdi-refresh.service" >&2
+        fi
+    fi
+}