Skip to content

Commit

Permalink
put jobs back in pr.yaml
Browse files Browse the repository at this point in the history
  • Loading branch information
brandon-b-miller committed Nov 30, 2023
1 parent 5581ac6 commit 7ef5290
Showing 1 changed file with 64 additions and 3 deletions.
67 changes: 64 additions & 3 deletions .github/workflows/pr.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -13,7 +13,11 @@ concurrency:
jobs:
pr-builder:
needs:
- build-and-test-wheels
- checks
- compute-matrix-build
- compute-matrix-test
- build-wheels
- test-wheels
secrets: inherit
uses: rapidsai/shared-workflows/.github/workflows/[email protected]
checks:
Expand All @@ -22,5 +26,62 @@ jobs:
with:
enable_check_generated_files: false
enable_check_pr_job_dependencies: false
build-and-test-wheels:
uses: ./.github/workflows/build_and_test_wheels.yaml
compute-matrix-build:
runs-on: ubuntu-latest
outputs:
MATRIX: ${{ steps.compute-matrix-build.outputs.MATRIX }}
steps:
- name: Compute Build Matrix
id: compute-matrix-build
run: |
set -eo pipefail
export MATRIX="
- { CUDA_VER: '12.0.1', ARCH: 'amd64', PY_VER: '3.9', LINUX_VER: 'centos7' }
- { CUDA_VER: '12.0.1', ARCH: 'amd64', PY_VER: '3.10', LINUX_VER: 'centos7' }
- { CUDA_VER: '12.0.1', ARCH: 'arm64', PY_VER: '3.9', LINUX_VER: 'rockylinux8' }
- { CUDA_VER: '12.0.1', ARCH: 'arm64', PY_VER: '3.10', LINUX_VER: 'rockylinux8' }
"
echo "MATRIX=$(
yq -n -o json 'env(MATRIX)' | \
jq -c '.' \
)" | tee --append "${GITHUB_OUTPUT}"
compute-matrix-test:
runs-on: ubuntu-latest
outputs:
MATRIX: ${{ steps.compute-matrix-test.outputs.MATRIX }}
steps:
- name: Compute Build Matrix
id: compute-matrix-test
run: |
set -eo pipefail
export MATRIX="
- { CUDA_VER: '12.0.1', ARCH: 'amd64', PY_VER: '3.9', LINUX_VER: 'ubuntu18.04', gpu: 'v100', driver: 'latest' }
- { CUDA_VER: '12.0.1', ARCH: 'amd64', PY_VER: '3.10', LINUX_VER: 'ubuntu18.04', gpu: 'v100', driver: 'latest' }
- { CUDA_VER: '12.0.1', ARCH: 'arm64', PY_VER: '3.9', LINUX_VER: 'ubuntu20.04', gpu: 'a100', driver: 'latest' }
- { CUDA_VER: '12.0.1', ARCH: 'arm64', PY_VER: '3.10', LINUX_VER: 'ubuntu20.04', gpu: 'a100', driver: 'latest' }
"
echo "MATRIX=$(
yq -n -o json 'env(MATRIX)' | \
jq -c '.' \
)" | tee --append "${GITHUB_OUTPUT}"
build-wheels:
needs:
- compute-matrix-build
uses: rapidsai/shared-workflows/.github/workflows/[email protected]
with:
build_type: pull-request
script: "ci/build_wheel.sh"
matrix_filter: ${{ needs.compute-matrix-build.outputs.MATRIX }}
test-wheels:
needs:
- build-wheels
- compute-matrix-test
uses: rapidsai/shared-workflows/.github/workflows/[email protected]
with:
build_type: pull-request
script: "ci/test_wheel.sh"
matrix_filter: ${{ needs.compute-matrix-test.outputs.MATRIX }}

0 comments on commit 7ef5290

Please sign in to comment.