-
Notifications
You must be signed in to change notification settings - Fork 2
91 lines (82 loc) · 2.45 KB
/
pr.yaml
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
name: pr
concurrency:
group: ci-on-${{ github.event_name }}-from-${{ github.ref_name }}
cancel-in-progress: true
on:
# run on pushes to certain branches
push:
branches:
- "pull-request/[0-9]+"
env:
# CUDA architectures to build for
CUDAARCHS: "RAPIDS"
# where jobs that download conda packages store the local channel
RAPIDS_LOCAL_CONDA_CHANNEL: /tmp/local-conda-packages
jobs:
# group together all jobs that must pass for a PR to be merged
# (for use by branch protections)
pr-builder:
needs:
- pre-commit
- conda-python-build
- conda-python-tests
- docs-build
uses: rapidsai/shared-workflows/.github/workflows/[email protected]
pre-commit:
runs-on: ubuntu-latest
steps:
- uses: actions/checkout@v4
- uses: pre-commit/[email protected]
conda-python-build:
needs:
- pre-commit
uses: ./.github/workflows/conda-python-build.yaml
with:
script: "ci/build_python.sh"
secrets: inherit
conda-python-tests:
needs:
- pre-commit
- conda-python-build
strategy:
fail-fast: false
# As of the last time this was updated, legate / cupynumeric packages were published for only:
#
# * architectures: amd64 only
# * CUDA: >=12.2
# * Python: 3.10, 3.11, 3.12
#
# Valid set of RAPIDS ci-conda image tags: https://hub.docker.com/r/rapidsai/ci-conda/tags
matrix:
include:
- ARCH: "amd64"
CUDA_VER: "12.5.1"
PY_VER: "3.10"
runs-on: linux-${{ matrix.ARCH }}-gpu-v100-latest-1
container:
image: "rapidsai/ci-conda:cuda${{ matrix.CUDA_VER }}-ubuntu22.04-py${{ matrix.PY_VER }}"
env:
NVIDIA_VISIBLE_DEVICES: ${{ env.NVIDIA_VISIBLE_DEVICES }}
steps:
- uses: actions/checkout@v4
with:
fetch-depth: 0
- name: download conda packages
uses: actions/download-artifact@v4
with:
name: legate-raft-conda-cuda${{ matrix.CUDA_VER }}-${{ matrix.ARCH }}-py${{ matrix.PY_VER }}
path: ${{ env.RAPIDS_LOCAL_CONDA_CHANNEL }}
github-token: ${{ secrets.GITHUB_TOKEN }}
repository: ${{ github.repository }}
run-id: ${{ github.run_id }}
- name: test
run: |
ci/test_python.sh
docs-build:
needs:
- conda-python-build
uses: ./.github/workflows/docs-build.yaml
with:
script: "ci/build_docs.sh"
deploy: false
secrets: inherit