CI tests / Python upgrade #18
Workflow file for this run
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| # Copyright 2026 Google LLC | |
| # | |
| # Licensed under the Apache License, Version 2.0 (the "License"); | |
| # you may not use this file except in compliance with the License. | |
| # You may obtain a copy of the License at | |
| # | |
| # http://www.apache.org/licenses/LICENSE-2.0 | |
| # | |
| # Unless required by applicable law or agreed to in writing, software | |
| # distributed under the License is distributed on an "AS IS" BASIS, | |
| # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. | |
| # See the License for the specific language governing permissions and | |
| # limitations under the License. | |
| name: CUDA 12 Linux x86_64 -> Linux x86_64 build | |
| on: | |
| pull_request: | |
| branches: | |
| - main | |
| permissions: | |
| contents: read | |
| pull-requests: read | |
| issues: read | |
| concurrency: | |
| group: ${{ github.workflow }}-${{ github.head_ref || github.ref }} | |
| # Don't cancel in-progress jobs for main branch | |
| cancel-in-progress: ${{ github.ref != 'main' }} | |
| jobs: | |
| run-tests: | |
| name: "CUDA 12 Linux x86_64 -> Linux x86_64 build. Runner ${{ matrix.job_info.runner }}" | |
| defaults: | |
| run: | |
| # Explicitly set the shell to bash | |
| shell: bash | |
| strategy: | |
| matrix: | |
| job_info: [ | |
| { | |
| runner: "linux-x86-n2-16", | |
| container: "us-docker.pkg.dev/ml-oss-artifacts-published/ml-public-container/ml-build:latest", | |
| }, | |
| ] | |
| # Use self-hosted runners. For a Linux runner, a corresponding container image used for the | |
| # workflow is required. | |
| # See go/ml-github-actions/getting_started/using_runners.md#supported-runner-scale-set. | |
| runs-on: ${{ matrix.job_info.runner }} | |
| container: | |
| image: ${{ matrix.job_info.container }} | |
| steps: | |
| - uses: actions/checkout@v5 | |
| - run: echo "Clang 18 building..." | |
| - run: | | |
| bazel build --config=build_cuda_with_clang \ | |
| --config=cuda \ | |
| --config=cuda_libraries_from_stubs \ | |
| --repo_env HERMETIC_CUDA_COMPUTE_CAPABILITIES="sm_50,sm_60,sm_70,sm_80,sm_90" \ | |
| //cc/tests/gpu:all | |
| - run: echo "NVCC building..." | |
| - run: bazel build --config=build_cuda_with_nvcc --config=cuda --config=cuda_libraries_from_stubs //cc/tests/gpu:all | |
| - run: echo "Bzlmod. Clang 18 building..." | |
| - run: | | |
| bazel build --config=build_cuda_with_clang \ | |
| --config=cuda \ | |
| --config=cuda_libraries_from_stubs \ | |
| --repo_env HERMETIC_CUDA_COMPUTE_CAPABILITIES="sm_50,sm_60,sm_70,sm_80,sm_90" \ | |
| --config=bzlmod \ | |
| //cc/tests/gpu:all | |
| - run: echo "Bzlmod. NVCC building..." | |
| - run: bazel build --config=build_cuda_with_nvcc --config=cuda --config=cuda_libraries_from_stubs --config=bzlmod //cc/tests/gpu:all |