|
22 | 22 | branches: |
23 | 23 | - main |
24 | 24 | - "pull-request/[0-9]+" |
| 25 | + - "deploy-release/*" |
| 26 | + |
25 | 27 | env: |
26 | 28 | UV_PROJECT_ENVIRONMENT: "./venv" |
27 | 29 |
|
28 | 30 | jobs: |
| 31 | + pre-flight: |
| 32 | + uses: NVIDIA-NeMo/FW-CI-templates/.github/workflows/_cicd_preflight.yml@v0.53.0 |
| 33 | + |
29 | 34 | pip-test-no-cuda: |
| 35 | + needs: [pre-flight] |
| 36 | + if: | |
| 37 | + !(needs.pre-flight.outputs.docs_only == 'true' |
| 38 | + || needs.pre-flight.outputs.is_deployment_workflow == 'true') |
30 | 39 | runs-on: ${{ matrix.arch }} |
31 | 40 | name: Pip - Python${{ matrix.python-version }} - ${{ matrix.arch == 'ubuntu-latest' && 'AMD64/Linux' || (matrix.arch == 'ubuntu-24.04-arm' && 'ARM64/Linux' || 'ARM64/Darwin') }} - No CUDA |
32 | 41 | strategy: |
|
80 | 89 | python-binary: ./venv/bin/python |
81 | 90 |
|
82 | 91 | pip-test-cuda: |
| 92 | + needs: [pre-flight] |
| 93 | + if: | |
| 94 | + !(needs.pre-flight.outputs.docs_only == 'true' |
| 95 | + || needs.pre-flight.outputs.is_deployment_workflow == 'true') |
83 | 96 | runs-on: linux-amd64-cpu16 |
84 | 97 | name: Pip - Python${{ matrix.python-version }}${{ matrix.extra-groups != '' && format('[{0}]', matrix.extra-groups) || '' }} - AMD64/Linux - NGC CUDA |
85 | 98 | container: |
@@ -140,6 +153,10 @@ jobs: |
140 | 153 | python-binary: ./venv/bin/python |
141 | 154 |
|
142 | 155 | uv-test-cuda: |
| 156 | + needs: [pre-flight] |
| 157 | + if: | |
| 158 | + !(needs.pre-flight.outputs.docs_only == 'true' |
| 159 | + || needs.pre-flight.outputs.is_deployment_workflow == 'true') |
143 | 160 | runs-on: linux-amd64-cpu16 |
144 | 161 | name: UV - AMD64/Linux - NGC ${{ contains(matrix.image, 'cuda') && 'CUDA' || 'PyTorch' }} |
145 | 162 | container: |
@@ -196,6 +213,10 @@ jobs: |
196 | 213 | python-binary: /opt/venv/bin/python |
197 | 214 |
|
198 | 215 | uv-test-no-cuda: |
| 216 | + needs: [pre-flight] |
| 217 | + if: | |
| 218 | + !(needs.pre-flight.outputs.docs_only == 'true' |
| 219 | + || needs.pre-flight.outputs.is_deployment_workflow == 'true') |
199 | 220 | runs-on: ${{ matrix.arch }} |
200 | 221 | name: UV - Python ${{ matrix.python-version }} - ${{ matrix.arch == 'ubuntu-latest' && 'AMD64/Linux' || 'ARM64/Darwin' }} - No CUDA |
201 | 222 | environment: nemo-ci |
@@ -254,9 +275,16 @@ jobs: |
254 | 275 | python-binary: ./venv/bin/python |
255 | 276 |
|
256 | 277 | install-test-summary: |
257 | | - needs: [pip-test-no-cuda, uv-test-cuda, pip-test-cuda, uv-test-no-cuda] |
| 278 | + needs: [pre-flight, pip-test-no-cuda, uv-test-cuda, pip-test-cuda, uv-test-no-cuda] |
258 | 279 | runs-on: ubuntu-latest |
259 | 280 | name: Install test summary |
| 281 | + if: | |
| 282 | + ( |
| 283 | + needs.pre-flight.outputs.docs_only == 'true' |
| 284 | + || needs.pre-flight.outputs.is_deployment_workflow == 'true' |
| 285 | + || success() |
| 286 | + ) |
| 287 | + && !cancelled() |
260 | 288 | steps: |
261 | 289 | - name: Result |
262 | 290 | run: echo Install check successful |
0 commit comments