Add R580 support for precompiled driver

.common-ci.yml

@@ -99,15 +99,15 @@ trigger-pipeline:
 .driver-versions-precompiled-ubuntu22.04:
   parallel:
     matrix:
-      - DRIVER_BRANCH: [535, 570]
+      - DRIVER_BRANCH: [535, 570, 580]
         KERNEL_FLAVOR: [aws, azure, generic, nvidia, oracle]
         LTS_KERNEL: ["5.15", "6.8"]
 
 # Define the matrix of precompiled jobs that can be run in parallel for ubuntu24.04
 .driver-versions-precompiled-ubuntu24.04:
   parallel:
     matrix:
-      - DRIVER_BRANCH: [570]
+      - DRIVER_BRANCH: [570, 580]
         KERNEL_FLAVOR: [aws, azure, generic, nvidia, oracle]
         LTS_KERNEL: ["6.8"]
 

.github/workflows/image.yaml

@@ -107,6 +107,7 @@ jobs:
         driver: 
           - 535
           - 570
+          - 580
         flavor:
           - aws
           - azure

.github/workflows/precompiled.yaml

@@ -34,7 +34,7 @@ jobs:
         id: extract_driver_branch
         run: |
           # get driver_branch
-          DRIVER_BRANCH=("535" "570")
+          DRIVER_BRANCH=("535" "570" "580")
           driver_branch_json=$(printf '%s\n' "${DRIVER_BRANCH[@]}" | jq -R . | jq -cs .)
           echo "driver_branch=$driver_branch_json" >> $GITHUB_OUTPUT
 

ubuntu22.04/precompiled/Dockerfile

@@ -51,11 +51,6 @@ RUN if [ -n "${CVE_UPDATES}" ]; then \
         rm -rf /var/lib/apt/lists/*; \
     fi
 
-RUN if [ "$DRIVER_BRANCH" -ge "550" ]; then \
-    apt-get update && \
-    apt-get install -y --no-install-recommends nvidia-imex-${DRIVER_BRANCH}=${DRIVER_VERSION}-1 && \
-    rm -rf /var/lib/apt/lists/*; fi
-
 COPY nvidia-driver /usr/local/bin
 
 ADD local-repo.sh /tmp

ubuntu22.04/precompiled/local-repo.sh

@@ -22,6 +22,48 @@ download_apt_with_dep () {
   fi
 }
 
+nvlink5_pkgs_download() {
+  if [ "$DRIVER_BRANCH" -ge "570" ]; then
+    download_apt_with_dep nvlsm
+    download_apt_with_dep infiniband-diags
+  fi
+}
+
+# libnvsdm packages are not available for arm64
+nvsdm_download() {
+  if [ "$TARGETARCH" = "amd64" ]; then
+    if [ "$DRIVER_BRANCH" -ge "580" ]; then
+       download_apt_with_dep libnvsdm ${DRIVER_VERSION}-1
+    elif [ "$DRIVER_BRANCH" -ge "560" ]; then
+       download_apt_with_dep libnvsdm-${DRIVER_BRANCH} ${DRIVER_VERSION}-1
+    fi
+  fi
+}
+
+fabricmanager_download() {
+  if [ "$DRIVER_BRANCH" -ge "580" ]; then
+    download_apt_with_dep nvidia-fabricmanager ${DRIVER_VERSION}-1
+  else
+    download_apt_with_dep nvidia-fabricmanager-${DRIVER_BRANCH} ${DRIVER_VERSION}-1
+  fi
+}
+
+nscq_download() {
+  if [ "$DRIVER_BRANCH" -ge "580" ]; then
+     download_apt_with_dep libnvidia-nscq ${DRIVER_VERSION}-1
+  else
+    download_apt_with_dep libnvidia-nscq-${DRIVER_BRANCH} ${DRIVER_VERSION}-1
+  fi
+}
+
+imex_download() {
+  if [ "$DRIVER_BRANCH" -ge "580" ]; then
+    download_apt_with_dep nvidia-imex ${DRIVER_VERSION}-1
+  elif [ "$DRIVER_BRANCH" -ge "550" ]; then
+    download_apt_with_dep nvidia-imex-${DRIVER_BRANCH} ${DRIVER_VERSION}-1
+  fi
+}
+
 download_driver_package_deps () {
   apt-get update
   pushd ${LOCAL_REPO_DIR}
@@ -37,18 +79,11 @@ download_driver_package_deps () {
   download_apt_with_dep libnvidia-encode-${DRIVER_BRANCH}-server
   download_apt_with_dep libnvidia-fbc1-${DRIVER_BRANCH}-server
 
-  download_apt_with_dep nvidia-fabricmanager-${DRIVER_BRANCH} ${DRIVER_VERSION}-1
-  download_apt_with_dep libnvidia-nscq-${DRIVER_BRANCH} ${DRIVER_VERSION}-1
-
-  if [ "$DRIVER_BRANCH" -ge "550" ]; then
-      download_apt_with_dep nvlsm
-      download_apt_with_dep infiniband-diags
-      download_apt_with_dep nvidia-imex-${DRIVER_BRANCH} ${DRIVER_VERSION}-1
-  fi
-
-  if [ "$DRIVER_BRANCH" -ge "560" ]; then
-      download_apt_with_dep libnvsdm-${DRIVER_BRANCH} ${DRIVER_VERSION}-1
-  fi
+  fabricmanager_download
+  nscq_download
+  nvlink5_pkgs_download
+  imex_download
+  nvsdm_download
 
   ls -al .
   popd

ubuntu24.04/precompiled/Dockerfile

@@ -46,11 +46,6 @@ RUN if [ -n "${CVE_UPDATES}" ]; then \
         rm -rf /var/lib/apt/lists/*; \
     fi
 
-RUN if [ "$DRIVER_BRANCH" -ge "550" ]; then \
-    apt-get update && \
-    apt-get install -y --no-install-recommends nvidia-imex-${DRIVER_BRANCH}=${DRIVER_VERSION}-1 && \
-    rm -rf /var/lib/apt/lists/*; fi
-
 COPY nvidia-driver /usr/local/bin
 
 ADD local-repo.sh /tmp

ubuntu24.04/precompiled/local-repo.sh

@@ -22,6 +22,47 @@ download_apt_with_dep () {
   fi
 }
 
+nvlink5_pkgs_download() {
+  if [ "$DRIVER_BRANCH" -ge "550" ]; then
+    download_apt_with_dep nvlsm
+    download_apt_with_dep infiniband-diags
+  fi
+}
+
+nvsdm_download() {
+  if [ "$TARGETARCH" = "amd64" ]; then
+    if [ "$DRIVER_BRANCH" -ge "580" ]; then
+       download_apt_with_dep libnvsdm ${DRIVER_VERSION}-1
+    elif [ "$DRIVER_BRANCH" -ge "560" ]; then
+       download_apt_with_dep libnvsdm-${DRIVER_BRANCH} ${DRIVER_VERSION}-1
+    fi
+  fi
+}
+
+fabricmanager_download() {
+  if [ "$DRIVER_BRANCH" -ge "580" ]; then
+    download_apt_with_dep nvidia-fabricmanager ${DRIVER_VERSION}-1
+  else
+    download_apt_with_dep nvidia-fabricmanager-${DRIVER_BRANCH} ${DRIVER_VERSION}-1
+  fi
+}
+
+nscq_download() {
+  if [ "$DRIVER_BRANCH" -ge "580" ]; then
+     download_apt_with_dep libnvidia-nscq ${DRIVER_VERSION}-1
+  else
+    download_apt_with_dep libnvidia-nscq-${DRIVER_BRANCH} ${DRIVER_VERSION}-1
+  fi
+}
+
+imex_download() {
+  if [ "$DRIVER_BRANCH" -ge "580" ]; then
+    download_apt_with_dep nvidia-imex ${DRIVER_VERSION}-1
+  elif [ "$DRIVER_BRANCH" -ge "550" ]; then
+    download_apt_with_dep nvidia-imex-${DRIVER_BRANCH} ${DRIVER_VERSION}-1
+  fi
+}
+
 download_driver_package_deps () {
   apt-get update
   pushd ${LOCAL_REPO_DIR}
@@ -37,18 +78,11 @@ download_driver_package_deps () {
   download_apt_with_dep libnvidia-encode-${DRIVER_BRANCH}-server
   download_apt_with_dep libnvidia-fbc1-${DRIVER_BRANCH}-server
 
-  download_apt_with_dep nvidia-fabricmanager-${DRIVER_BRANCH} ${DRIVER_VERSION}-1
-  download_apt_with_dep libnvidia-nscq-${DRIVER_BRANCH} ${DRIVER_VERSION}-1
-
-  if [ "$DRIVER_BRANCH" -ge "550" ]; then
-      download_apt_with_dep nvlsm
-      download_apt_with_dep infiniband-diags
-      download_apt_with_dep nvidia-imex-${DRIVER_BRANCH} ${DRIVER_VERSION}-1
-  fi
-
-  if [ "$DRIVER_BRANCH" -ge "560" ]; then
-      download_apt_with_dep libnvsdm-${DRIVER_BRANCH} ${DRIVER_VERSION}-1
-  fi
+  fabricmanager_download
+  nscq_download
+  nvlink5_pkgs_download
+  imex_download
+  nvsdm_download
 
   ls -al .
   popd