Skip to content

Commit

Permalink
Add matrix filtering for wheels-pure-test.yml (#108)
Browse files Browse the repository at this point in the history
This PR adds matrix filtering for the `wheels-pure-test.yml` workflow.
  • Loading branch information
ajschmidt8 authored Jun 21, 2023
1 parent 50d3850 commit 0366832
Showing 1 changed file with 26 additions and 4 deletions.
30 changes: 26 additions & 4 deletions .github/workflows/wheels-pure-test.yml
Original file line number Diff line number Diff line change
Expand Up @@ -28,6 +28,9 @@ on:
required: false
type: string
default: 'true'
matrix_filter:
type: string
default: "."

defaults:
run:
Expand All @@ -49,14 +52,33 @@ permissions:
statuses: none

jobs:
compute-matrix:
runs-on: ubuntu-latest
outputs:
MATRIX: ${{ steps.compute-matrix.outputs.MATRIX }}
steps:
- name: Compute Matrix
id: compute-matrix
run: |
set -eo pipefail
export MATRIX="
- { CUDA_VER: '11.8.0'}
- { CUDA_VER: '12.0.1'}
"
echo "MATRIX=$(
yq -n -o json 'env(MATRIX)' | \
jq -c '${{ inputs.matrix_filter }} | {include: .}' \
)" | tee --append "${GITHUB_OUTPUT}"
wheel-test:
name: wheel test pure
needs: compute-matrix
runs-on: linux-amd64-gpu-v100-latest-1
strategy:
matrix:
ctk: ["11.8.0", "12.0.1"]
matrix: ${{ fromJSON(needs.compute-matrix.outputs.MATRIX) }}
container:
image: "rapidsai/citestwheel:cuda-devel-${{ matrix.ctk }}-ubuntu18.04"
image: "rapidsai/citestwheel:cuda-devel-${{ matrix.CUDA_VER }}-ubuntu18.04"
env:
NVIDIA_VISIBLE_DEVICES: ${{ env.NVIDIA_VISIBLE_DEVICES }} # GPU jobs must set this container env variable
RAPIDS_PY_VERSION: "3.10"
Expand Down Expand Up @@ -94,7 +116,7 @@ jobs:

- name: Run citestwheel
run: |
PIP_CU_VERSION="$(rapids-wheel-ctk-name-gen ${{ matrix.ctk }})"
PIP_CU_VERSION="$(rapids-wheel-ctk-name-gen ${{ matrix.CUDA_VER }})"
# The variable PIP_CU_VERSION needs to be used as an unevaluated string by downstream libraries
# to support pulling wheels built from a previous workflow for different
Expand Down

0 comments on commit 0366832

Please sign in to comment.