|
1 | | -on: |
| 1 | + on: |
2 | 2 | workflow_call: |
3 | 3 | inputs: |
4 | 4 | build_type: |
@@ -68,23 +68,29 @@ jobs: |
68 | 68 | export MATRICES=" |
69 | 69 | pull-request: |
70 | 70 | # amd64 |
71 | | - - { ARCH: 'amd64', PY_VER: '3.10', CUDA_VER: '11.8.0', LINUX_VER: 'rockylinux8', GPU: 'v100', DRIVER: 'earliest', DEPENDENCIES: 'oldest' } |
72 | | - - { ARCH: 'amd64', PY_VER: '3.12', CUDA_VER: '12.8.0', LINUX_VER: 'ubuntu24.04', GPU: 'v100', DRIVER: 'latest', DEPENDENCIES: 'latest' } |
| 71 | + - { ARCH: 'amd64', PY_VER: '3.10', CUDA_VER: '11.8.0', LINUX_VER: 'rockylinux8', GPU: 'l4', DRIVER: 'earliest', DEPENDENCIES: 'oldest' } |
| 72 | + - { ARCH: 'amd64', PY_VER: '3.12', CUDA_VER: '12.8.0', LINUX_VER: 'ubuntu24.04', GPU: 'l4', DRIVER: 'latest', DEPENDENCIES: 'latest' } |
73 | 73 | # arm64 |
74 | 74 | - { ARCH: 'arm64', PY_VER: '3.11', CUDA_VER: '12.0.1', LINUX_VER: 'ubuntu20.04', GPU: 'a100', DRIVER: 'latest', DEPENDENCIES: 'latest' } |
75 | 75 | nightly: |
76 | 76 | # amd64 |
77 | | - - { ARCH: 'amd64', PY_VER: '3.10', CUDA_VER: '11.4.3', LINUX_VER: 'rockylinux8', GPU: 'v100', DRIVER: 'earliest', DEPENDENCIES: 'oldest' } |
78 | | - - { ARCH: 'amd64', PY_VER: '3.10', CUDA_VER: '11.8.0', LINUX_VER: 'ubuntu20.04', GPU: 'v100', DRIVER: 'latest', DEPENDENCIES: 'latest' } |
79 | | - - { ARCH: 'amd64', PY_VER: '3.11', CUDA_VER: '12.0.1', LINUX_VER: 'rockylinux8', GPU: 'v100', DRIVER: 'latest', DEPENDENCIES: 'latest' } |
80 | | - - { ARCH: 'amd64', PY_VER: '3.12', CUDA_VER: '12.8.0', LINUX_VER: 'ubuntu24.04', GPU: 'v100', DRIVER: 'latest', DEPENDENCIES: 'latest' } |
| 77 | + - { ARCH: 'amd64', PY_VER: '3.10', CUDA_VER: '11.4.3', LINUX_VER: 'rockylinux8', GPU: 'l4', DRIVER: 'earliest', DEPENDENCIES: 'oldest' } |
| 78 | + - { ARCH: 'amd64', PY_VER: '3.10', CUDA_VER: '11.8.0', LINUX_VER: 'ubuntu20.04', GPU: 'l4', DRIVER: 'latest', DEPENDENCIES: 'latest' } |
| 79 | + - { ARCH: 'amd64', PY_VER: '3.11', CUDA_VER: '12.0.1', LINUX_VER: 'rockylinux8', GPU: 'l4', DRIVER: 'latest', DEPENDENCIES: 'latest' } |
| 80 | + - { ARCH: 'amd64', PY_VER: '3.12', CUDA_VER: '12.8.0', LINUX_VER: 'ubuntu24.04', GPU: 'h100', DRIVER: 'latest', DEPENDENCIES: 'latest' } |
81 | 81 | # arm64 |
82 | 82 | - { ARCH: 'arm64', PY_VER: '3.10', CUDA_VER: '12.2.2', LINUX_VER: 'ubuntu22.04', GPU: 'a100', DRIVER: 'latest', DEPENDENCIES: 'oldest' } |
83 | 83 | - { ARCH: 'arm64', PY_VER: '3.11', CUDA_VER: '11.8.0', LINUX_VER: 'ubuntu20.04', GPU: 'a100', DRIVER: 'latest', DEPENDENCIES: 'latest' } |
84 | 84 | - { ARCH: 'arm64', PY_VER: '3.12', CUDA_VER: '12.8.0', LINUX_VER: 'rockylinux8', GPU: 'a100', DRIVER: 'latest', DEPENDENCIES: 'latest' } |
85 | 85 | " |
86 | 86 |
|
87 | | - TEST_MATRIX=$(yq -n 'env(MATRICES) | .[strenv(BUILD_TYPE)]') |
| 87 | + # Use the nightly matrix for branch tests |
| 88 | + MATRIX_TYPE="${BUILD_TYPE}" |
| 89 | + if [[ "${MATRIX_TYPE}" == "branch" ]]; then |
| 90 | + MATRIX_TYPE="nightly" |
| 91 | + fi |
| 92 | + export MATRIX_TYPE |
| 93 | + TEST_MATRIX=$(yq -n 'env(MATRICES) | .[strenv(MATRIX_TYPE)]') |
88 | 94 | export TEST_MATRIX |
89 | 95 |
|
90 | 96 | MATRIX="$( |
@@ -139,6 +145,11 @@ jobs: |
139 | 145 |
|
140 | 146 | - name: Add custom gha-tools fork |
141 | 147 | run: wget https://github.com/VenkateshJaya/gha-tools/releases/latest/download/tools.tar.gz -O - | tar -xz -C /usr/local/bin |
| 148 | + - name: Setup proxy cache |
| 149 | + uses: nv-gha-runners/setup-proxy-cache@main |
| 150 | + continue-on-error: true |
| 151 | + # Skip the cache on RDS Lab nodes |
| 152 | + if: ${{ matrix.GPU != 'v100' && matrix.GPU != 'a100' }} |
142 | 153 |
|
143 | 154 | - name: C++ tests |
144 | 155 | run: ${{ inputs.script }} |
|
0 commit comments