From 980ff15d0e7ebcc7fe0a9bc9459d50bead81032b Mon Sep 17 00:00:00 2001
From: Ben Howe <bhowe@nvidia.com>
Date: Fri, 10 Jan 2025 20:18:09 +0000
Subject: [PATCH] Upload wheel artifacts when running build wheels job

Signed-off-by: Ben Howe <bhowe@nvidia.com>
---
 .github/workflows/build_wheels.yaml | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/.github/workflows/build_wheels.yaml b/.github/workflows/build_wheels.yaml
index 71092f3f..13b95edd 100644
--- a/.github/workflows/build_wheels.yaml
+++ b/.github/workflows/build_wheels.yaml
@@ -20,6 +20,7 @@ jobs:
     strategy:
       fail-fast: false
       matrix:
+        python: ['3.10', '3.11', '3.12']
         toolchain:
           - id: cu12.0-gcc11
             cc: gcc-11
@@ -49,3 +50,8 @@ jobs:
           .github/workflows/scripts/build_wheels.sh \
               --cudaq-prefix $HOME/.cudaq \
 
+      - name: Upload artifact
+        uses: actions/upload-artifact@v4
+        with:
+          name: wheels-py${{ matrix.python }}-amd64
+          path: /wheels/**