Pre-compiled image e2e test

Signed-off-by: shiva kumar <[email protected]>
NVIDIA · Sep 11, 2024 · 4a2829e · 4a2829e
1 parent 735d3cd
commit 4a2829e
Showing 1 changed file with 38 additions and 8 deletions.
diff --git a/.github/workflows/precompiled.yaml b/.github/workflows/precompiled.yaml
@@ -53,7 +53,7 @@ jobs:
           kernel_flavors_json=$(printf '%s\n' "${KERNEL_FLAVORS[@]}" | jq -R . | jq -cs .)
           echo "kernel_flavors=$kernel_flavors_json" >> $GITHUB_OUTPUT
 
-  precompiled-image:
+  precompiled-build-image:
     needs: set-driver-version-matrix
     runs-on: ubuntu-latest
     strategy:
@@ -110,7 +110,7 @@ jobs:
           source kernel_version.txt && \
           make DRIVER_VERSIONS=${DRIVER_VERSIONS} DRIVER_BRANCH=${{ matrix.driver-branch }} build-${DIST}-${DRIVER_VERSION}
       
-      - name: Save build image as a tarball
+      - name: Save build image as a tar
         env:
           DIST: "ubuntu22.04"
           PRIVATE_REGISTRY: "ghcr.io"
@@ -120,13 +120,20 @@ jobs:
           docker images  
           docker images "${PRIVATE_REGISTRY}/nvidia/driver:${{ matrix.driver-branch }}-${KERNEL_VERSION}-${DIST}"
           docker save "${PRIVATE_REGISTRY}/nvidia/driver:${{ matrix.driver-branch }}-${KERNEL_VERSION}-${DIST}" \
-            -o driver_images_${{ matrix.driver-branch }}_${KERNEL_VERSION}.tar
+            -o driver_images_${{ matrix.driver-branch }}_${KERNEL_VERSION}_${DIST}.tar
 
-      # - name: Upload build image as an artifact
-      #   uses: actions/upload-artifact@v4
-      #   with:
-      #     name: driver-images-${{ matrix.driver-branch }}_${{ matrix.flavor }}
-      #     path: driver_images_${{ matrix.driver-branch }}_${{ matrix.flavor }}.tar
+      # commit sha for tar artifacts 
+
+      - name: Upload build image as an artifact
+        uses: actions/upload-artifact@v4
+        env:
+          DIST: "ubuntu22.04"
+          PRIVATE_REGISTRY: "ghcr.io"
+          DIST: "ubuntu22.04"
+        with:
+           name: driver_images_${{ matrix.driver-branch }}_${KERNEL_VERSION}_${DIST}
+           path: driver_images_${{ matrix.driver-branch }}_${KERNEL_VERSION}_${DIST}.tar
+           retention-days: 1 
 
   determine-e2e-test-matrix:
     runs-on: ubuntu-latest
@@ -289,3 +296,26 @@ jobs:
           name: nvidiadriver-Precompiled-e2e-test-logs
           path: ./logs/
           retention-days: 15
+
+precompiled-push-image:
+    runs-on: ubuntu-latest
+    needs: 
+    - set-driver-version-matrix
+    - determine-e2e-test-matrix
+    strategy:
+      matrix:
+        kernel_version: ${{ fromJson(needs.determine-e2e-test-matrix.outputs.matrix_values) }}
+        driver-branch: ${{ fromJson(needs.set-driver-version-matrix.outputs.driver_branch) }}
+        DIST: "ubuntu22.04"
+    steps:
+      - name: Download Docker image artifact
+        uses: actions/download-artifact@v4
+        with:
+          name: driver_images_${{ matrix.driver-branch }}_${kernel_version}_${DIST}
+
+      - name: Push built image
+        run: | 
+          ls "driver_images_${{ matrix.driver-branch }}_${kernel_version}_${DIST}.tar"
+
+      - name: Remove built image tar
+        run: rm driver_images_${{ matrix.driver-branch }}_${kernel_version}_${DIST}.tar