Support Data Center precompiled driver container for Arm (Ubuntu 24.04)

Signed-off-by: Shiva Kumar (SW-CLOUD) <shivaku@nvidia.com>

Author: Shiva Kumar

.github/workflows/precompiled.yaml

@@ -41,7 +41,8 @@ jobs:
           echo "driver_branch=$driver_branch_json" >> $GITHUB_OUTPUT
 
           # get kernel flavors
-          KERNEL_FLAVORS=("aws" "azure" "azure-fde" "generic" "nvidia" "oracle")
+          # KERNEL_FLAVORS=("aws" "azure" "azure-fde" "generic" "nvidia" "oracle")
+          KERNEL_FLAVORS=("generic")
           kernel_flavors_json=$(printf '%s\n' "${KERNEL_FLAVORS[@]}" | jq -R . | jq -cs .)
           echo "kernel_flavors=$kernel_flavors_json" >> $GITHUB_OUTPUT
 
@@ -51,7 +52,8 @@ jobs:
           echo "dist=$dist_json" >> $GITHUB_OUTPUT
 
           # LTS_KERNEL setup
-          LTS_KERNEL=("5.15" "6.8")
+          # LTS_KERNEL=("5.15" "6.8")
+          LTS_KERNEL=("6.8")
           lts_kernel_json=$(printf '%s\n' "${LTS_KERNEL[@]}" | jq -R . | jq -cs .)
           echo "lts_kernel=$lts_kernel_json" >> $GITHUB_OUTPUT
 
@@ -459,6 +461,8 @@ jobs:
     steps:
       - name: Check out code
         uses: actions/checkout@v6
+      - name: Set up Docker Buildx
+        uses: docker/setup-buildx-action@v3
       - name: Login to GitHub Container Registry
         uses: docker/login-action@v3
         with:
@@ -470,12 +474,12 @@ jobs:
         run: |
           echo "PRIVATE_REGISTRY=ghcr.io" >> $GITHUB_ENV
           KERNEL_VERSION="${{ matrix.kernel_version }}"
-          PLATFORM_SUFFIX="amd64"
+          PLATFORM_NAME="amd64"
           if [[ "${KERNEL_VERSION}" == *-arm64 ]]; then
             KERNEL_VERSION="${KERNEL_VERSION%-arm64}"
-            PLATFORM_SUFFIX="-arm64"
+            PLATFORM_NAME="arm64"
           fi
-          echo "PLATFORM_SUFFIX=$PLATFORM_SUFFIX" >> $GITHUB_ENV
+          echo "PLATFORM_NAME=$PLATFORM_NAME" >> $GITHUB_ENV
           echo "KERNEL_VERSION=$KERNEL_VERSION" >> $GITHUB_ENV
 
       - name: Download base image artifact
@@ -499,7 +503,9 @@ jobs:
           image_path="./base-images-${{ matrix.driver_branch }}-${{ matrix.kernel_version }}.tar"
           echo "uploading  $image_path"
           docker load -i $image_path
-          docker push ${PRIVATE_REGISTRY}/nvidia/driver:base-${BASE_TARGET}-${LTS_KERNEL}-${KERNEL_FLAVOR}-${{ matrix.driver_branch }}
+          BASE_IMAGE_TAG="${PRIVATE_REGISTRY}/nvidia/driver:base-${BASE_TARGET}-${LTS_KERNEL}-${KERNEL_FLAVOR}-${{ matrix.driver_branch }}"
+          docker push "${BASE_IMAGE_TAG}-${{ env.PLATFORM_NAME }}"
+          docker buildx imagetools create -t "${BASE_IMAGE_TAG}" --append "${BASE_IMAGE_TAG}-${{ env.PLATFORM_NAME }}" || docker buildx imagetools create -t "${BASE_IMAGE_TAG}" "${BASE_IMAGE_TAG}-${{ env.PLATFORM_NAME }}"
 
       - name: Download built image artifact
         if: ${{ ! (matrix.driver_branch == 535 && contains(matrix.kernel_version, 'ubuntu24.04')) }}
@@ -514,7 +520,9 @@ jobs:
           image_path="./driver-images-${{ matrix.driver_branch }}-${{ matrix.kernel_version }}.tar"
           echo "uploading  $image_path"
           docker load -i $image_path
-          docker push ${PRIVATE_REGISTRY}/nvidia/driver:${{ matrix.driver_branch }}-${{ env.KERNEL_VERSION }}
+          DRIVER_IMAGE_TAG="${PRIVATE_REGISTRY}/nvidia/driver:${{ matrix.driver_branch }}-${{ env.KERNEL_VERSION }}"
+          docker push "${DRIVER_IMAGE_TAG}-${{ env.PLATFORM_NAME }}"
+          docker buildx imagetools create -t "${DRIVER_IMAGE_TAG}" --append "${DRIVER_IMAGE_TAG}-${{ env.PLATFORM_NAME }}" || docker buildx imagetools create -t "${DRIVER_IMAGE_TAG}" "${DRIVER_IMAGE_TAG}-${{ env.PLATFORM_NAME }}"
 
       - name: Slack notification
         if: ${{ ! (matrix.driver_branch == 535 && contains(matrix.kernel_version, 'ubuntu24.04')) }}
@@ -525,5 +533,5 @@ jobs:
           payload: |
             {
               "channel": "${{ secrets.SLACK_CHANNEL_ID }}",
-              "text": "${{ secrets.PUBLISHED_PRECOMPILED_IMAGE_MESSAGE }} ${{ env.PRIVATE_REGISTRY }}/nvidia/driver:${{ matrix.driver_branch }}-${{ matrix.kernel_version }} for platform ${{ env.PLATFORM_SUFFIX }}"
+              "text": "${{ secrets.PUBLISHED_PRECOMPILED_IMAGE_MESSAGE }} ${{ env.PRIVATE_REGISTRY }}/nvidia/driver:${{ matrix.driver_branch }}-${{ matrix.kernel_version }} for platform ${{ env.PLATFORM_NAME }}"
             }