From af087f15b3bf2a5cfb235cc21069bbeafab9e81b Mon Sep 17 00:00:00 2001
From: Henry Li <lihl@microsoft.com>
Date: Fri, 28 Jun 2024 11:42:09 -0700
Subject: [PATCH] onboard libnvidia-nscq for azurelinux

---
 azurelinux/Dockerfile    | 2 +-
 azurelinux/nvidia-driver | 4 ++--
 2 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/azurelinux/Dockerfile b/azurelinux/Dockerfile
index 0d21b14f..6e2683a0 100644
--- a/azurelinux/Dockerfile
+++ b/azurelinux/Dockerfile
@@ -18,7 +18,7 @@ RUN curl -fsSL -o /etc/yum.repos.d/mariner-nvidia.repo \
 # Create a location to store the pre-downloaded RPMs for installation during container runtime
 RUN mkdir -p /opt/nvidia
 RUN tdnf -y --downloadonly --downloaddir=/opt/nvidia install cuda-${DRIVER_VERSION}-${KERNEL_VERSION} \
-        nvidia-fabric-manager-${DRIVER_VERSION}
+        nvidia-fabric-manager-${DRIVER_VERSION} libnvidia-nscq-${DRIVER_VERSION}
 
 LABEL io.k8s.display-name="NVIDIA Driver Container"
 LABEL name="NVIDIA Driver Container"
diff --git a/azurelinux/nvidia-driver b/azurelinux/nvidia-driver
index 0a341260..19d2b31a 100755
--- a/azurelinux/nvidia-driver
+++ b/azurelinux/nvidia-driver
@@ -93,8 +93,8 @@ _install_driver() {
     echo "Installing the NVIDIA driver modules"
     tdnf install -y ${PACKAGE_DIR}/cuda-${DRIVER_VERSION}*${KERNEL_VERSION//-/.}*.rpm
 
-    echo "Installing the NVIDIA fabricmanager"
-    tdnf install -y ${PACKAGE_DIR}/nvidia-fabric-manager-${DRIVER_VERSION}*.rpm
+    echo "Installing the NVIDIA fabricmanager and NSCQ"
+    tdnf install -y ${PACKAGE_DIR}/nvidia-fabric-manager-${DRIVER_VERSION}*.rpm ${PACKAGE_DIR}/libnvidia-nscq-${DRIVER_VERSION}*.rpm
 }
 
 # Load the kernel modules and start persistenced.