From e119277c8de5a44923269acbde00e49f64727552 Mon Sep 17 00:00:00 2001
From: McCloudS <64094529+McCloudS@users.noreply.github.com>
Date: Mon, 3 Feb 2025 18:26:44 -0700
Subject: [PATCH] Update build_GPU_Compute5.yml

---
 .github/workflows/build_GPU_Compute5.yml | 26 +++++++++++++++++-------
 1 file changed, 19 insertions(+), 7 deletions(-)

diff --git a/.github/workflows/build_GPU_Compute5.yml b/.github/workflows/build_GPU_Compute5.yml
index 6b7a876..b22d77d 100644
--- a/.github/workflows/build_GPU_Compute5.yml
+++ b/.github/workflows/build_GPU_Compute5.yml
@@ -1,4 +1,4 @@
-name: Build_Subgen_Dockerfile_GPU_Compute5
+name: Build_Subgen_Dockerfile_Compute5
 
 on:
   push:
@@ -10,12 +10,16 @@ on:
 jobs:
   docker:
     runs-on: ubuntu-latest
+    strategy:
+      fail-fast: false
+      matrix:
+        platform:
+          - linux/amd64
+          - linux/arm64
 
     steps:
-      - name: Checkout code
-        uses: actions/checkout@v2
-        with:
-          fetch-depth: 0
+      - name: Checkout
+        uses: actions/checkout@v4
 
       - name: Get version from subgen.py
         id: get_version
@@ -23,17 +27,25 @@ jobs:
           version=$(grep -oP "subgen_version\s*=\s*'\K[^']+" subgen.py)
           echo "version=$version" >> $GITHUB_ENV
 
+      - name: Set up QEMU
+        uses: docker/setup-qemu-action@v3
+
+      - name: Set up Docker Buildx
+        uses: docker/setup-buildx-action@v3
+
       - name: Login to Docker Hub
         uses: docker/login-action@v3
         with:
           username: ${{ secrets.DOCKERHUB_USERNAME }}
           password: ${{ secrets.DOCKERHUB_TOKEN }}
 
-      - name: Build and push GPU Docker image
+      - name: Build and push CPU Docker image
         uses: docker/build-push-action@v5
         with:
           context: .
+          platforms: linux/amd64,linux/arm64
+          file: ./Dockerfile.compute5
           push: true
           tags: |
             mccloud/subgen:compute5
-            mccloud/subgen:compute5-${{ env.version }}
+            mccloud/subgen:${{ env.version }}-compute5