Update CI for arm64 build

Signed-off-by: Ben Howe <[email protected]>

.github/actions/build-lib/action.yaml

@@ -12,6 +12,10 @@ inputs:
     description: 'Indicates whether to save the compilation cache'
     default: 'false'
     required: false
+  platform:
+    description: 'Platform (amd64 or arm64)'
+    default: ''
+    required: true
 outputs:
   build-dir:
     description: 'Build dir.'
@@ -32,7 +36,7 @@ runs:
     - name: Compilation cache key
       id: ccache-key
       run: |
-        echo "main=ccache-${{ inputs.lib }}-cu12.0-gcc11" >> $GITHUB_OUTPUT
+        echo "main=ccache-${{ inputs.lib }}-cu12.0-gcc11-${{ inputs.platform }}" >> $GITHUB_OUTPUT
         if [[ -n "${{ inputs.pr-number }}" ]]; then
           echo "pr=-pr${{ inputs.pr-number }}" >> $GITHUB_OUTPUT
         fi

.github/actions/get-cudaq-build/action.yaml

@@ -28,6 +28,10 @@ inputs:
     description: 'Check if a cache entry exists without downloading the cache'
     default: 'false'
     required: false
+  platform:
+    description: 'Platform (amd64 or arm64)'
+    default: ''
+    required: true
 outputs:
   found-cache:
     description: 'A boolean value to indicate that a cache entry was found.'
@@ -49,7 +53,7 @@ runs:
           .cudaq_version
       run: |
         hash=${{ hashFiles(format('{0}', env.to_hash)) }}
-        echo "main=cudaq-${{ inputs.ref }}-$hash" >> $GITHUB_OUTPUT
+        echo "main=cudaq-${{ matrix.platform }}-${{ inputs.ref }}-$hash" >> $GITHUB_OUTPUT
         if [[ -n "${{ inputs.pr-number }}" ]]; then
           echo "pr=-pr${{ inputs.pr-number }}" >> $GITHUB_OUTPUT
         fi

.github/workflows/all_libs.yaml

@@ -7,7 +7,11 @@ jobs:
   pr-build:
     name: Build and test
     if: startsWith(github.ref, 'refs/heads/pull-request/')
-    runs-on: ${{ startsWith(github.repository, 'NVIDIA/cudaqx') && 'linux-amd64-cpu8' || 'ubuntu-latest' }}
+    strategy:
+      fail-fast: false
+      matrix:
+        platform: ['amd64', 'arm64']
+    runs-on: ${{ startsWith(github.repository, 'NVIDIA/cudaqx') && format('linux-{0}-cpu8', matrix.platform) || 'ubuntu-latest' }}
     container: ghcr.io/nvidia/cuda-quantum-devdeps:ext-cu12.0-gcc11-main
     permissions:
       actions: write
@@ -61,6 +65,7 @@ jobs:
           lib: "all"
           pr-number: ${{ steps.export-pr-info.outputs.pr_number }}
           save-ccache: true
+          platform: ${{ matrix.platform }}
 
       # ========================================================================
       # Run tests

.github/workflows/cudaq_cache.yaml

@@ -20,7 +20,11 @@ concurrency:
 jobs:
   build-cudaq:
     name: Build CUDAQ
-    runs-on: ${{ startsWith(github.repository, 'NVIDIA/cudaqx') && 'linux-amd64-cpu32' || 'ubuntu-latest' }}
+    strategy:
+      fail-fast: false
+      matrix:
+        platform: ['amd64', 'arm64']
+    runs-on: ${{ startsWith(github.repository, 'NVIDIA/cudaqx') && format('linux-{0}-cpu32', matrix.platform) || 'ubuntu-latest' }}
     container: ghcr.io/nvidia/cuda-quantum-devdeps:ext-cu12.0-gcc11-main
     permissions:
       actions: write

.github/workflows/lib_qec.yaml

@@ -7,7 +7,11 @@ jobs:
   pr-build:
     name: Build and test
     if: startsWith(github.ref, 'refs/heads/pull-request/')
-    runs-on: ${{ startsWith(github.repository, 'NVIDIA/cudaqx') && 'linux-amd64-cpu8' || 'ubuntu-latest' }}
+    strategy:
+      fail-fast: false
+      matrix:
+        platform: ['amd64', 'arm64']
+    runs-on: ${{ startsWith(github.repository, 'NVIDIA/cudaqx') && format('linux-{0}-cpu8', matrix.platform) || 'ubuntu-latest' }}
     container: ghcr.io/nvidia/cuda-quantum-devdeps:ext-cu12.0-gcc11-main
     permissions:
       actions: write
@@ -57,6 +61,7 @@ jobs:
           lib: "qec"
           pr-number: ${{ steps.export-pr-info.outputs.pr_number }}
           save-ccache: true
+          platform: ${{ matrix.platform }}
 
       # ========================================================================
       # Run tests

.github/workflows/lib_solvers.yaml

@@ -7,7 +7,11 @@ jobs:
   pr-build:
     name: Build and test
     if: startsWith(github.ref, 'refs/heads/pull-request/')
-    runs-on: ${{ startsWith(github.repository, 'NVIDIA/cudaqx') && 'linux-amd64-cpu8' || 'ubuntu-latest' }}
+    strategy:
+      fail-fast: false
+      matrix:
+        platform: ['amd64', 'arm64']
+    runs-on: ${{ startsWith(github.repository, 'NVIDIA/cudaqx') && format('linux-{0}-cpu8', matrix.platform) || 'ubuntu-latest' }}
     container: ghcr.io/nvidia/cuda-quantum-devdeps:ext-cu12.0-gcc11-main
     permissions:
       actions: write
@@ -61,6 +65,7 @@ jobs:
           lib: "solvers"
           pr-number: ${{ steps.export-pr-info.outputs.pr_number }}
           save-ccache: true
+          platform: ${{ matrix.platform }}
 
       # ========================================================================
       # Run tests

.github/workflows/pr_workflow.yaml

@@ -85,7 +85,11 @@ jobs:
     name: Build CUDAQ
     needs: [check-changes]
     if: needs.check-changes.outputs.build-cudaq == 'true'
-    runs-on: ${{ startsWith(github.repository, 'NVIDIA/cudaqx') && 'linux-amd64-cpu32' || 'ubuntu-latest' }}
+    strategy:
+      fail-fast: false
+      matrix:
+        platform: ['amd64', 'arm64']
+    runs-on: ${{ startsWith(github.repository, 'NVIDIA/cudaqx') && format('linux-{0}-cpu32', matrix.platform) || 'ubuntu-latest' }}
     container: ghcr.io/nvidia/cuda-quantum-devdeps:ext-cu12.0-gcc11-main
     permissions:
       actions: write

.github/workflows/sync.yml

@@ -8,7 +8,7 @@ name: "Sync with upstream repository"
 jobs:
   sync:
     name: Get Updates from Upstream
-    if: ${{ github.repository != 'NVIDIA/cudaqx' }}
+    if: ${{ startsWith(github.repository, 'NVIDIA/cudaqx') && github.repository != 'NVIDIA/cudaqx' }}
     runs-on: 'ubuntu-latest'
 
     steps: