Skip to content

Simplify the CI build and test matrix#249

Merged
kkraus14 merged 4 commits intoNVIDIA:mainfrom
cryos:ci-updates
May 12, 2025
Merged

Simplify the CI build and test matrix#249
kkraus14 merged 4 commits intoNVIDIA:mainfrom
cryos:ci-updates

Conversation

@cryos
Copy link
Contributor

@cryos cryos commented May 9, 2025

Simplifying the CI build and test matrices by moving them to a single file, ci/matrix.yml, that is read in to create input matrices for CI.

@cryos cryos force-pushed the ci-updates branch 2 times, most recently from 442adba to 625eb32 Compare May 9, 2025 20:08
@gmarkall gmarkall added the 2 - In Progress Currently a work in progress label May 9, 2025
@cryos cryos force-pushed the ci-updates branch 2 times, most recently from b2eb14d to 4fcc8d7 Compare May 9, 2025 20:22
@cryos cryos changed the title [WIP]: Simplify the wheels build matrix Simplify the wheels build matrix May 11, 2025
@cryos cryos marked this pull request as ready for review May 11, 2025 15:40
@cryos cryos changed the title Simplify the wheels build matrix Simplify the CI build and test matrix May 12, 2025
Comment on lines +3 to +13
test-matrix:
- { CUDA_VER: '11.8.0', ARCH: 'amd64', PY_VER: '3.9', LINUX_VER: 'rockylinux8', GPU: 'l4', DRIVER: 'earliest' }
- { CUDA_VER: '11.8.0', ARCH: 'amd64', PY_VER: '3.10', LINUX_VER: 'ubuntu20.04', GPU: 'l4', DRIVER: 'latest' }
- { CUDA_VER: '12.0.1', ARCH: 'amd64', PY_VER: '3.11', LINUX_VER: 'ubuntu22.04', GPU: 'l4', DRIVER: 'latest' }
- { CUDA_VER: '12.2.2', ARCH: 'amd64', PY_VER: '3.12', LINUX_VER: 'ubuntu22.04', GPU: 'l4', DRIVER: 'latest' }
- { CUDA_VER: '12.8.0', ARCH: 'amd64', PY_VER: '3.13', LINUX_VER: 'ubuntu24.04', GPU: 'l4', DRIVER: 'latest' }
- { CUDA_VER: '11.8.0', ARCH: 'arm64', PY_VER: '3.9', LINUX_VER: 'rockylinux8', GPU: 'a100', DRIVER: 'earliest' }
- { CUDA_VER: '11.8.0', ARCH: 'arm64', PY_VER: '3.10', LINUX_VER: 'ubuntu20.04', GPU: 'a100', DRIVER: 'latest' }
- { CUDA_VER: '12.0.1', ARCH: 'arm64', PY_VER: '3.11', LINUX_VER: 'ubuntu22.04', GPU: 'a100', DRIVER: 'latest' }
- { CUDA_VER: '12.2.2', ARCH: 'arm64', PY_VER: '3.12', LINUX_VER: 'ubuntu22.04', GPU: 'a100', DRIVER: 'latest' }
- { CUDA_VER: '12.8.0', ARCH: 'arm64', PY_VER: '3.13', LINUX_VER: 'ubuntu24.04', GPU: 'a100', DRIVER: 'latest' }
Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

As far as I know we can't address it now, but we are actually quite sensitive to driver version in Numba as we're generating PTX which has potential compatibility issues across driver versions. It would be great for us to be able to cover:

  • 12.0 driver with newest 12.x toolkit (I think this is a gap today)
  • Newest 12.x driver with 12.0 toolkit (I think this is covered today)

And similar story with 13.x once we it's released and we support it.

Copy link
Contributor Author

@cryos cryos May 12, 2025

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

This is largely carrying through what was already in place, good to know for future work.

@kkraus14 kkraus14 merged commit 811818f into NVIDIA:main May 12, 2025
37 checks passed
@isVoid isVoid mentioned this pull request May 21, 2025
isVoid added a commit that referenced this pull request May 21, 2025
- Allow External Code to Use Cooperative Group (#240)
- Improve debug info for kernel arguments (#242)
- Allow Numba NVRTC Binding Search Additional Paths (#254)
- Add Bfloat16 High Level API, Documentation (#245)
- add a test to use bf16 bindings inside device functions (#244)
- Change CI to only be manually triggered to save on CI runs (#252)
- Simplify the CI build and test matrix (#249)
@gmarkall gmarkall mentioned this pull request May 21, 2025
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

2 - In Progress Currently a work in progress

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants

Comments