-
Notifications
You must be signed in to change notification settings - Fork 15
77 lines (74 loc) · 2.18 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
# Copyright (c) 2023, NVIDIA CORPORATION.
name: pr
on:
push:
branches:
- "pull-request/[0-9]+"
concurrency:
group: ${{ github.workflow }}-${{ github.ref }}
cancel-in-progress: true
jobs:
pr-builder:
needs:
- checks
- wheel-build
- wheel-test
secrets: inherit
uses: rapidsai/shared-action-workflows/.github/workflows/pr-builder.yaml@branch-23.12
checks:
secrets: inherit
uses: rapidsai/shared-action-workflows/.github/workflows/checks.yaml@branch-23.12
with:
enable_check_generated_files: false
wheel-build:
runs-on: linux-${{ matrix.arch }}-gpu-v100-latest-1
strategy:
matrix:
python-version: ["3.9", "3.10"]
arch: ["arm64", "amd64"]
container:
# Use CentOS 7 for an old glibc version, so the wheel has broad manylinux compatibility.
image: "rapidsai/ci-conda:cuda12.0.1-centos7-py3.10"
env:
NVIDIA_VISIBLE_DEVICES: ${{ env.NVIDIA_VISIBLE_DEVICES }}
steps:
- name: Checkout code
uses: actions/checkout@v3
with:
fetch-depth: 0
- name: Get PR Info
id: get-pr-info
uses: rapidsai/shared-actions/get-pr-info@main
- name: Run build_wheel.sh
run: ci/build_wheel.sh ${{ matrix.python-version }}
- name: Upload Python Wheel
uses: actions/upload-artifact@v2
with:
path: ./wheel-build-${{ matrix.python-version }}
wheel-test:
needs:
- wheel-build
runs-on: linux-${{ matrix.arch }}-gpu-v100-latest-1
strategy:
matrix:
python-version: ["3.9", "3.10"]
linux-version: ["centos7", "rockylinux8"]
arch: ["arm64", "amd64"]
container:
image: "rapidsai/ci-wheel:cuda12.0.1-${{ matrix.linux-version }}-py${{ matrix.python-version }}"
env:
NVIDIA_VISIBLE_DEVICES: ${{ env.NVIDIA_VISIBLE_DEVICES }}
steps:
- name: Checkout code
uses: actions/checkout@v3
with:
fetch-depth: 0
- name: Get PR Info
id: get-pr-info
uses: rapidsai/shared-actions/get-pr-info@main
- name: Download Python Wheel
uses: actions/download-artifact@v2
with:
path: ./
- name: run test_wheel.sh
run: ci/test_wheel.sh