From f76769f88c0a3b0dd581944684ae0309146e1908 Mon Sep 17 00:00:00 2001 From: Jeremi Piotrowski Date: Mon, 26 Feb 2024 15:22:01 +0000 Subject: [PATCH] x11-drivers/nvidia-drivers: Generate CDI definitions during loading Nvidia CDI definitions are used by Kubernetes to represent the topology of GPUs available to a node and allow workload placement on suitable GPUs. Signed-off-by: Jeremi Piotrowski --- .../x11-drivers/nvidia-drivers/files/bin/setup-nvidia | 2 ++ 1 file changed, 2 insertions(+) diff --git a/sdk_container/src/third_party/coreos-overlay/x11-drivers/nvidia-drivers/files/bin/setup-nvidia b/sdk_container/src/third_party/coreos-overlay/x11-drivers/nvidia-drivers/files/bin/setup-nvidia index c694e40f03..38c5186056 100644 --- a/sdk_container/src/third_party/coreos-overlay/x11-drivers/nvidia-drivers/files/bin/setup-nvidia +++ b/sdk_container/src/third_party/coreos-overlay/x11-drivers/nvidia-drivers/files/bin/setup-nvidia @@ -131,6 +131,8 @@ function install_and_load() { mknod -m 666 /dev/nvidia-uvm c $D 0 fi + nvidia-ctk cdi generate --output=/etc/cdi/nvidia.yaml + if [ "${NVIDIA_PERSISTENCE_MODE}" = "on" ]; then nvidia-persistenced fi