Use GPU runners

Author: bmhowe23

.github/workflows/all_libs_release.yaml

@@ -22,9 +22,15 @@ jobs:
     strategy:
       fail-fast: false
       matrix:
-        platform: ['amd64', 'arm64']
-    runs-on: ${{ startsWith(github.repository, 'NVIDIA/cudaqx') && format('linux-{0}-cpu8', matrix.platform) || 'ubuntu-latest' }}
-    container: ${{ format('ghcr.io/nvidia/cudaqx-dev:{0}-{1}', inputs.release-number, matrix.platform) }}
+        runner: [
+          { arch: arm64, gpu: a100 },
+          { arch: amd64, gpu: v100 },
+        ]
+    runs-on: linux-${{ matrix.runner.arch }}-gpu-${{ matrix.runner.gpu }}-latest-1
+    container: 
+      image: ${{ format('ghcr.io/nvidia/cudaqx-dev:{0}-{1}', inputs.release-number, matrix.runner.arch) }}
+      env:
+        NVIDIA_VISIBLE_DEVICES: ${{ env.NVIDIA_VISIBLE_DEVICES }}
     permissions: write-all
     steps:
       - name: Checkout repository
@@ -64,7 +70,7 @@ jobs:
             gh release view -R ${{ inputs.assets_repo }} ${{ inputs.assets_tag }} >> $GITHUB_STEP_SUMMARY
             # Extract the decoder that needs to be embedded in the release
             mkdir -p tmp
-            unzip -d tmp installed_files-${{ matrix.platform }}.zip
+            unzip -d tmp installed_files-${{ matrix.runner.arch }}.zip
             echo "QEC_EXTERNAL_DECODERS=$(pwd)/tmp/lib/decoder-plugins/libcudaq-qec-nv-qldpc-decoder.so" >> $GITHUB_ENV
           fi
         shell: bash
@@ -78,19 +84,19 @@ jobs:
         uses: ./.github/actions/build-lib
         with:
           lib: "all"
-          platform: ${{ matrix.platform }}
+          platform: ${{ matrix.runner.arch }}
 
       - name: Save build artifacts
         run: |
           cmake --build ${{ steps.build.outputs.build-dir }} --target zip_installed_files
           cd ${{ steps.build.outputs.build-dir }}
-          mv installed_files.zip installed_files-${{ matrix.platform }}.zip
+          mv installed_files.zip installed_files-${{ matrix.runner.arch }}.zip
       
       - name: Upload build artifacts
         uses: actions/upload-artifact@v4
         with:
-          name: installed_files-${{ matrix.platform }}
-          path: ${{ steps.build.outputs.build-dir }}/installed_files-${{ matrix.platform }}.zip
+          name: installed_files-${{ matrix.runner.arch }}
+          path: ${{ steps.build.outputs.build-dir }}/installed_files-${{ matrix.runner.arch }}.zip
 
       # ========================================================================
       # Run tests