Add a wheels-publish job (#25)

This PR adds a job to our release workflow which uploads release wheels to the nightly rapids pypi index. --------- Co-authored-by: Vyas Ramasubramani <[email protected]>
rapidsai · Dec 5, 2023 · 9fbc642 · 9fbc642
1 parent 0e69e72
commit 9fbc642
Show file tree

Hide file tree

Showing 3 changed files with 15 additions and 2 deletions.
diff --git a/.github/workflows/build.yaml b/.github/workflows/build.yaml
@@ -3,6 +3,8 @@ name: build
 
 on:
   push:
+    branches:
+      - main
     tags:
       - 'v*'
 
@@ -28,3 +30,14 @@ jobs:
       build_type: branch
       script: "ci/build_wheel.sh"
       matrix_filter: ${{ needs.compute-matrix.outputs.BUILD_MATRIX }}
+  publish-wheels:
+    needs:
+      - build-wheels
+    secrets: inherit
+    uses: rapidsai/shared-workflows/.github/workflows/[email protected]
+    with:
+      build_type: ${{ inputs.build_type || 'branch' }}
+      branch: ${{ inputs.branch }}
+      sha: ${{ inputs.sha }}
+      date: ${{ inputs.date }}
+      package-name: pynvjitlink
diff --git a/ci/build_wheel.sh b/ci/build_wheel.sh
@@ -16,4 +16,4 @@ mkdir -p ./wheel-build
 pip wheel . --wheel-dir=./wheel-build -vvv
 
 rapids-logger "Upload Wheel"
-rapids-upload-wheels-to-s3 ./wheel-build
+RAPIDS_PY_WHEEL_NAME="pynvjitlink-cu12" rapids-upload-wheels-to-s3 ./wheel-build
diff --git a/ci/test_wheel.sh b/ci/test_wheel.sh
@@ -4,7 +4,7 @@
 set -e
 
 rapids-logger "Download Wheel"
-rapids-download-wheels-from-s3 ./wheel-build/
+RAPIDS_PY_WHEEL_NAME="pynvjitlink-cu12" rapids-download-wheels-from-s3 ./wheel-build/
 
 rapids-logger "Install wheel"
 pip install --find-links ./wheel-build pynvjitlink-cu12