Skip to content

ci: run smoke tests across a python x torch version matrix#32

Open
guru-desh wants to merge 2 commits into
apple:mainfrom
guru-desh:smoke-test-on-different-torch-versions
Open

ci: run smoke tests across a python x torch version matrix#32
guru-desh wants to merge 2 commits into
apple:mainfrom
guru-desh:smoke-test-on-different-torch-versions

Conversation

@guru-desh

Copy link
Copy Markdown
Contributor

Description

Smoke tests previously ran on one implicit torch version (whatever pyproject.toml's unpinned bound resolved to) across all supported python versions. Add torch_2_8/2_9/2_10/2_11 dependency groups (build-matched torchao/torchvision per pytorch/ao issue 2919) and a matrix.torch_group axis on the test-smoke CI job so each of the 4 torch minor versions is smoke tested independently, still covering all 3 python versions per job via the existing nox session.

Adding these new smoke tests doesn't change the total CI time. Because provisioning is fast and we have more Linux machines, each test-smoke run takes 6 minutes. make test-tutorial takes 11 minutes, so stage 1 CI will be max(test-smoke-torch-2.8, test-smoke-2.9, ..., test-tutorials). Since test-tutorials is 11 minutes, stage 1 CI takes 11 minutes, and adding more test-smoke wouldn't affect total CI time.

Smoke tests previously ran on one implicit torch version (whatever
pyproject.toml's unpinned bound resolved to) across all supported
python versions. Add torch_2_8/2_9/2_10/2_11 dependency groups
(build-matched torchao/torchvision per pytorch/ao issue 2919) and a
matrix.torch_group axis on the test-smoke CI job so each of the 4
torch minor versions is smoke tested independently, still covering
all 3 python versions per job via the existing nox session.

lowest_tested_torch/highest_tested_torch are renamed to torch_2_8/
torch_2_11 for a consistent naming scheme; the linux-tests/macos-tests
full-suite jobs and their Makefile targets are unchanged.
setup_env.sh's group discovery used grep -E '^[a-z_-]+ = \[', which
excludes digits and made torch_2_8/2_9/2_10/2_11 invisible to
validate_groups — breaking test-highest-pytorch/test-lowest-pytorch
with 'Unknown dependency group', caught by the fork CI run.
@guru-desh guru-desh requested review from aseemw and dengqiaoyu July 10, 2026 22:46
@guru-desh guru-desh added the enhancement New feature or request label Jul 10, 2026
@guru-desh guru-desh marked this pull request as ready for review July 10, 2026 22:47
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

enhancement New feature or request

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants