diff --git a/.github/workflows/pr.yaml b/.github/workflows/pr.yaml index 0fe57264..1ead9e51 100644 --- a/.github/workflows/pr.yaml +++ b/.github/workflows/pr.yaml @@ -17,6 +17,7 @@ jobs: - compute-matrix - build-wheels - test-wheels + - publish-wheels secrets: inherit uses: rapidsai/shared-workflows/.github/workflows/pr-builder.yaml@branch-24.02 checks: @@ -51,3 +52,14 @@ jobs: build_type: pull-request script: "ci/test_wheel.sh" matrix_filter: ${{ needs.compute-matrix.outputs.TEST_MATRIX }} + publish-wheels: + needs: + - build-wheels + secrets: inherit + uses: rapidsai/shared-workflows/.github/workflows/wheels-publish.yaml@branch-24.02 + with: + build_type: ${{ inputs.build_type || 'branch' }} + branch: ${{ inputs.branch }} + sha: ${{ inputs.sha }} + date: ${{ inputs.date }} + package-name: pynvjitlink-cu12