Test arm image build

Author: sd109

.github/workflows/build-push-vllm-cpu.yml

@@ -14,9 +14,37 @@ on:
         required: true
 
 jobs:
-  build_push_image:
+  # build_push_x86_image:
+  #   name: Build and push image
+  #   runs-on: ubuntu-latest
+  #   permissions:
+  #     contents: read
+  #     id-token: write         # needed for signing the images with GitHub OIDC Token
+  #     packages: write         # required for pushing container images
+  #     security-events: write  # required for pushing SARIF files
+  #   steps:
+  #     - name: Check out the vLLM repository
+  #       uses: actions/checkout@v4
+  #       with:
+  #         repository: vllm-project/vllm
+  #         ref: ${{ inputs.vllm_ref }}
+
+  #     - name: Login to GitHub Container Registry
+  #       uses: docker/login-action@v3
+  #       with:
+  #         registry: ghcr.io
+  #         username: ${{ github.actor }}
+  #         password: ${{ secrets.GITHUB_TOKEN }}
+
+  #     - name: Build and push image
+  #       run: |
+  #         IMAGE=ghcr.io/stackhpc/vllm-cpu:${{ inputs.vllm_ref }}
+  #         docker build -f Dockerfile.cpu -t $IMAGE --shm-size=4g .
+  #         docker push $IMAGE
+
+  build_push_arm64_image:
     name: Build and push image
-    runs-on: ubuntu-latest
+    runs-on: ubuntu-24.04-arm
     permissions:
       contents: read
       id-token: write         # needed for signing the images with GitHub OIDC Token
@@ -38,36 +66,6 @@ jobs:
 
       - name: Build and push image
         run: |
-          IMAGE=ghcr.io/stackhpc/vllm-cpu:${{ inputs.vllm_ref }}
-          docker build -f Dockerfile.cpu -t $IMAGE --shm-size=4g .
+          IMAGE=ghcr.io/stackhpc/vllm-cpu:${{ inputs.vllm_ref }}-arm64
+          docker build -f Dockerfile.arm -t $IMAGE --shm-size=4g .
           docker push $IMAGE
-
-      # - name: Calculate metadata for image
-      #   id: image-meta
-      #   uses: docker/metadata-action@v5
-      #   with:
-      #     images: ghcr.io/stackhpc/vllm-cpu
-      #     # Produce the branch name or tag and the SHA as tags
-      #     tags: |
-      #       type=raw,value=${{ inputs.vllm_ref }}
-
-      # - name: Build and push image x86 image
-      #   uses: azimuth-cloud/github-actions/docker-multiarch-build-push@master
-      #   with:
-      #     cache-key: vllm-cpu
-      #     file: Dockerfile.cpu
-      #     platforms: linux/amd64
-      #     push: true
-      #     tags: ${{ steps.image-meta.outputs.tags }}
-      #     labels: ${{ steps.image-meta.outputs.labels }}
-
-      # # TODO: Test whether this works
-      # # - name: Build and push Arm image
-      # #   uses: azimuth-cloud/github-actions/docker-multiarch-build-push@master
-      # #   with:
-      # #     cache-key: vllm-cpu
-      # #     file: Dockerfile.cpu
-      # #     platforms: linux/arm64
-      # #     push: true
-      # #     tags: ${{ steps.image-meta.outputs.tags }}
-      # #     labels: ${{ steps.image-meta.outputs.labels }}