ubuntu24.04 ci pipeline fix

Signed-off-by: shiva kumar <[email protected]>

Author: shivakunv

.github/workflows/precompiled.yaml

@@ -154,7 +154,14 @@ jobs:
            name: driver-images-${{ matrix.driver_branch }}-${{ env.KERNEL_VERSION }}-${{ env.DIST }}
            path: ./driver-images-${{ matrix.driver_branch }}-${{ env.KERNEL_VERSION }}-${{ env.DIST }}.tar
            retention-days: 1
-
+      
+      - name: Upload kernel version as an artifact
+        uses: actions/upload-artifact@v4
+        with:
+           name: kernel-version-${{ matrix.driver_branch }}-${{ env.KERNEL_VERSION }}-${{ env.DIST }}
+           path: ./kernel-version-${{ matrix.driver_branch }}-${{ env.KERNEL_VERSION }}-${{ env.DIST }}.txt
+           retention-days: 1
+  
   determine-e2e-test-matrix:
     runs-on: linux-amd64-cpu4
     strategy:
@@ -210,6 +217,80 @@ jobs:
           echo "matrix_values=$(cat $GITHUB_WORKSPACE/matrix_values.json | jq -c .)" >> $GITHUB_OUTPUT
           echo "matrix_values_not_empty=1" >> $GITHUB_OUTPUT
 
+  determine-e2e-test-matrix-first-release:
+    runs-on: linux-amd64-cpu4
+    if: ${{ needs.determine-e2e-test-matrix.outputs.matrix_values_not_empty == '0' }}
+    strategy:
+      matrix:
+        dist: ${{ fromJson(needs.set-driver-version-matrix.outputs.dist) }}
+    needs:
+      - precompiled-build-image
+      - set-driver-version-matrix
+    outputs:
+      matrix_values_not_empty: ${{ steps.set_kernel_version.outputs.matrix_values_not_empty }}
+      matrix_values: ${{ steps.set_kernel_version.outputs.matrix_values }}
+      dist: ${{ steps.set-driver-version-matrix.outputs.dist }}
+    steps:
+      - name: Check out code
+        uses: actions/checkout@v4
+      - name: Login to GitHub Container Registry
+        uses: docker/login-action@v3
+        with:
+          registry: ghcr.io
+          username: ${{ github.actor }}
+          password: ${{ secrets.GITHUB_TOKEN }}
+
+      # download kernel version file for first release
+      - name: Download kernel version artifact
+        uses: actions/download-artifact@v4
+        with:
+           name: kernel-version-${{ matrix.driver_branch }}-${{ matrix.kernel_version }}-${{ matrix.dist }}
+           path: ./
+
+      # download kernel version file for first release
+      - name: Download kernel version artifact
+        env:
+          DIST: ${{ matrix.dist }}
+        run: |
+          artifacts=$(gh api -X GET /repos/${{ github.repository }}/actions/artifacts --jq '.artifacts[] | select(.name | endswith(env.DIST)) | .name')
+          for artifact in $artifacts; do
+            echo "Downloading kernel_version: $artifact"
+            gh run download --name "$artifact" --dir ./
+          done
+
+      - name: Set kernel version
+        id: set_kernel_version
+        env:
+          DIST: ${{ matrix.dist }}
+        run: |
+          echo "matrix_values_not_empty=0" >> $GITHUB_OUTPUT
+
+          if [ "$DIST" == "ubuntu22.04" ]; then
+            export BASE_TARGET="jammy"
+            export LTS_KERNEL="5.15"
+          elif [ "$DIST" == "ubuntu24.04" ]; then
+            export BASE_TARGET="noble"
+            export LTS_KERNEL="6.8"
+          fi
+
+          kernel_flavors_json='${{ needs.set-driver-version-matrix.outputs.kernel_flavors }}'
+          KERNEL_FLAVORS=($(echo "$kernel_flavors_json" | jq -r '.[]'))
+          driver_branch_json='${{ needs.set-driver-version-matrix.outputs.driver_branch }}'
+          DRIVER_BRANCHES=($(echo "$driver_branch_json" | jq -r '.[]'))
+
+          source ./tests/scripts/ci-precompiled-helpers.sh
+          KERNEL_VERSIONS=($(get_kernel_versions_to_test $BASE_TARGET KERNEL_FLAVORS[@] DRIVER_BRANCHES[@] $DIST))
+          if [ -z "$KERNEL_VERSIONS" ]; then
+            # no new kernel release
+            echo "Skipping e2e tests"
+            exit 0
+          fi
+          # Convert array to JSON format and assign
+          echo "[]" > $GITHUB_WORKSPACE/matrix_values.json
+          printf '%s\n' "${KERNEL_VERSIONS[@]}" | jq -R . | jq -s . > $GITHUB_WORKSPACE/matrix_values.json
+          echo "matrix_values=$(cat $GITHUB_WORKSPACE/matrix_values.json | jq -c .)" >> $GITHUB_OUTPUT
+          echo "matrix_values_not_empty=1" >> $GITHUB_OUTPUT
+
   e2e-tests-nvidiadriver:
     runs-on: linux-amd64-cpu4
     needs:
@@ -260,6 +341,14 @@ jobs:
           DRIVER_BRANCHES=($(echo "$driver_branch_json" | jq -r '.[]'))
           echo "DRIVER_BRANCHES=${DRIVER_BRANCHES[*]}" >> $GITHUB_ENV
 
+      # step added to skip azure e2e tests and publish the image
+      # FIXME -- remove step once azure kernel upgrade starts working
+      - name: Skip azure e2e
+        if [[ "${KERNEL_VERSION}" == *-azure ]]; then
+            echo "e2e test for azure flavor skipped, as kernel upgrade AWS => azure is not supported"
+            exit 0
+        fi
+
       - name: Install GitHub CLI
         run: |
           sudo apt-get update

tests/scripts/ci-precompiled-helpers.sh

@@ -11,20 +11,22 @@ get_kernel_versions_to_test() {
 
     kernel_versions=()
     for kernel_flavor in "${KERNEL_FLAVORS[@]}"; do
-        # FIXME -- remove if condition, once azure kernel upgrade starts working
-        if [[ "$kernel_flavor" == "azure" ]]; then
-            continue
-        fi
         for DRIVER_BRANCH in "${DRIVER_BRANCHES[@]}"; do
             source ./tests/scripts/findkernelversion.sh "$BASE_TARGET" "${kernel_flavor}" "$DRIVER_BRANCH" "$DIST" >&2
-            if [[ "$should_continue" == true ]]; then
+            if [[ "$should_continue" == true || "$should_continue_first_release" == true ]]; then
                 break
             fi
         done
         if [[ "$should_continue" == true ]]; then
             KERNEL_VERSION=$(echo "$KERNEL_VERSION" | tr -d ' \n')
             kernel_versions+=("$KERNEL_VERSION")
         fi
+        if [[ "$should_continue_first_release" == true ]]; then
+            matching_kernel_version_file=$(ls | grep "$DRIVER_BRANCHES" | grep "$LTS_KERNEL" | grep "$KERNEL_FLAVORS" | grep "$DIST")
+            source $matching_kernel_version_file
+            KERNEL_VERSION=$(echo "$KERNEL_VERSION" | tr -d ' \n')
+            kernel_versions+=("$KERNEL_VERSION")
+        fi
     done
     echo "${kernel_versions[@]}"
 }

tests/scripts/findkernelversion.sh

@@ -20,9 +20,9 @@ export PATH=$(pwd)/bin:${PATH}
 regctl image get-file ghcr.io/nvidia/driver:base-${BASE_TARGET}-${KERNEL_FLAVOR}-${DRIVER_BRANCH} /var/kernel_version.txt ./kernel_version.txt
 export $(grep -oP 'KERNEL_VERSION=[^ ]+' ./kernel_version.txt)
 # first release KERNEL_VERSIONS is null
+export should_continue_first_release=false
 if [ -z "$KERNEL_VERSIONS" ]; then
-    regctl image get-file nvcr.io/nvidia/driver:base-${BASE_TARGET}-${KERNEL_FLAVOR}-${DRIVER_BRANCH} /var/kernel_version.txt ./kernel_version.txt
-    export $(grep -oP 'KERNEL_VERSION=[^ ]+' ./kernel_version.txt)
+    export should_continue_first_release=true
 fi
 
 # calculate driver tag