perf: add arm64 support to az_acr_release

Author: andhreljaKern

.github/workflows/az_acr_push.yml

@@ -76,6 +76,7 @@ jobs:
           tags: ${{ env.AZURE_CONTAINER_REGISTRY }}/${{ github.event.repository.name }}:${{ matrix.platform }}-gpu
           cache-from: type=gha
           cache-to: type=gha,mode=max
+          platforms: linux/${{ matrix.platform }}
           build-args: |
             platform=linux/${{ matrix.platform }}
             label=dockerfile-path=https://github.com/${{ github.repository }}/blob/${{ github.sha }}/gpu.Dockerfile

.github/workflows/az_acr_release.yml

@@ -20,22 +20,30 @@ jobs:
         with:
           submodules: 'true'
 
+      - name: Set up QEMU
+        uses: docker/setup-qemu-action@v3
+        with:
+          platforms: arm64,arm
+
+      - name: Set up Docker Buildx
+        uses: docker/setup-buildx-action@v3
+        with:
+          platforms: linux/amd64,linux/arm64
+
       - name: Log into registry
         uses: docker/login-action@v3
         with:
           registry: "${{ env.AZURE_CONTAINER_REGISTRY }}"
           username: "${{ env.ACR_LOGIN_USERNAME }}"
           password: "${{ env.ACR_LOGIN_PASSWORD }}"
 
-      - name: Set up Docker Buildx
-        uses: docker/setup-buildx-action@v3
-
       - name: Build & Push ${{ github.event.repository.name }}:amd64
         if: ${{ github.event_name != 'release' }}
         uses: docker/build-push-action@v5
         with:
           cache-from: type=gha
           cache-to: type=gha,mode=max
+          platforms: linux/amd64
           file: ${{ env.DOCKERFILE }}
           tags: ${{ env.AZURE_CONTAINER_REGISTRY }}/${{ github.event.repository.name }}:amd64
           push: true
@@ -49,6 +57,7 @@ jobs:
         with:
           cache-from: type=gha
           cache-to: type=gha,mode=max
+          platforms: linux/arm64
           file: ${{ env.DOCKERFILE }}
           tags: ${{ env.AZURE_CONTAINER_REGISTRY }}/${{ github.event.repository.name }}:arm64
           push: true
@@ -61,6 +70,7 @@ jobs:
         with:
           cache-from: type=gha
           cache-to: type=gha,mode=max
+          platforms: linux/amd64
           file: ${{ env.DOCKERFILE }}
           tags: ${{ env.AZURE_CONTAINER_REGISTRY }}/${{ github.event.repository.name }}:${{ env.IMAGE_TAG }}
           push: true
@@ -73,6 +83,7 @@ jobs:
         with:
           cache-from: type=gha
           cache-to: type=gha,mode=max
+          platforms: linux/amd64
           file: ${{ env.DOCKERFILE }}
           tags: ${{ env.AZURE_CONTAINER_REGISTRY }}/${{ github.event.repository.name }}:latest
           push: true